1*7c478bd9Sstevel@tonic-gate /*
2*7c478bd9Sstevel@tonic-gate * CDDL HEADER START
3*7c478bd9Sstevel@tonic-gate *
4*7c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the
5*7c478bd9Sstevel@tonic-gate * Common Development and Distribution License, Version 1.0 only
6*7c478bd9Sstevel@tonic-gate * (the "License"). You may not use this file except in compliance
7*7c478bd9Sstevel@tonic-gate * with the License.
8*7c478bd9Sstevel@tonic-gate *
9*7c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*7c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing.
11*7c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions
12*7c478bd9Sstevel@tonic-gate * and limitations under the License.
13*7c478bd9Sstevel@tonic-gate *
14*7c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each
15*7c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*7c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the
17*7c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying
18*7c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner]
19*7c478bd9Sstevel@tonic-gate *
20*7c478bd9Sstevel@tonic-gate * CDDL HEADER END
21*7c478bd9Sstevel@tonic-gate */
22*7c478bd9Sstevel@tonic-gate /*
23*7c478bd9Sstevel@tonic-gate * Copyright 2004 Sun Microsystems, Inc. All rights reserved.
24*7c478bd9Sstevel@tonic-gate * Use is subject to license terms.
25*7c478bd9Sstevel@tonic-gate */
26*7c478bd9Sstevel@tonic-gate
27*7c478bd9Sstevel@tonic-gate /*
28*7c478bd9Sstevel@tonic-gate * Set bits in the DT_FLAGS_1 member of the .dynamic section of an object.
29*7c478bd9Sstevel@tonic-gate */
30*7c478bd9Sstevel@tonic-gate
31*7c478bd9Sstevel@tonic-gate #include <stdio.h>
32*7c478bd9Sstevel@tonic-gate #include <stdlib.h>
33*7c478bd9Sstevel@tonic-gate #include <stdarg.h>
34*7c478bd9Sstevel@tonic-gate #include <errno.h>
35*7c478bd9Sstevel@tonic-gate #include <libelf.h>
36*7c478bd9Sstevel@tonic-gate #include <gelf.h>
37*7c478bd9Sstevel@tonic-gate #include <string.h>
38*7c478bd9Sstevel@tonic-gate #include <fcntl.h>
39*7c478bd9Sstevel@tonic-gate #include <libgen.h>
40*7c478bd9Sstevel@tonic-gate #include <unistd.h>
41*7c478bd9Sstevel@tonic-gate #include <sys/types.h>
42*7c478bd9Sstevel@tonic-gate #include <sys/stat.h>
43*7c478bd9Sstevel@tonic-gate #include <sys/link.h>
44*7c478bd9Sstevel@tonic-gate
45*7c478bd9Sstevel@tonic-gate #include <util.h>
46*7c478bd9Sstevel@tonic-gate
47*7c478bd9Sstevel@tonic-gate /*
48*7c478bd9Sstevel@tonic-gate * These are here because we can't be sure (yet) that the build machine has a
49*7c478bd9Sstevel@tonic-gate * sys/link.h that includes the following #defines. This tool will be executed
50*7c478bd9Sstevel@tonic-gate * on the build machine, so we have to use its headers (rather than the ones
51*7c478bd9Sstevel@tonic-gate * in $ROOT which will, by definition, be up to date). These #defines can be
52*7c478bd9Sstevel@tonic-gate * removed when we're sure that all build machines have recent copies of
53*7c478bd9Sstevel@tonic-gate * sys/link.h.
54*7c478bd9Sstevel@tonic-gate */
55*7c478bd9Sstevel@tonic-gate #ifndef DF_1_IGNMULDEF
56*7c478bd9Sstevel@tonic-gate #define DF_1_IGNMULDEF 0x00040000
57*7c478bd9Sstevel@tonic-gate #endif
58*7c478bd9Sstevel@tonic-gate #ifndef DF_1_NOKSYMS
59*7c478bd9Sstevel@tonic-gate #define DF_1_NOKSYMS 0x00080000
60*7c478bd9Sstevel@tonic-gate #endif
61*7c478bd9Sstevel@tonic-gate
62*7c478bd9Sstevel@tonic-gate struct dtflagval {
63*7c478bd9Sstevel@tonic-gate char *fv_name;
64*7c478bd9Sstevel@tonic-gate ulong_t fv_val;
65*7c478bd9Sstevel@tonic-gate };
66*7c478bd9Sstevel@tonic-gate
67*7c478bd9Sstevel@tonic-gate static struct dtflagval dtflagvals[] = {
68*7c478bd9Sstevel@tonic-gate { "DF_1_IGNMULDEF", DF_1_IGNMULDEF },
69*7c478bd9Sstevel@tonic-gate { "DF_1_NOKSYMS", DF_1_NOKSYMS },
70*7c478bd9Sstevel@tonic-gate { NULL }
71*7c478bd9Sstevel@tonic-gate };
72*7c478bd9Sstevel@tonic-gate
73*7c478bd9Sstevel@tonic-gate const char *progname;
74*7c478bd9Sstevel@tonic-gate
75*7c478bd9Sstevel@tonic-gate static void
usage(void)76*7c478bd9Sstevel@tonic-gate usage(void)
77*7c478bd9Sstevel@tonic-gate {
78*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr, "Usage: %s -f flag_val file\n", progname);
79*7c478bd9Sstevel@tonic-gate exit(2);
80*7c478bd9Sstevel@tonic-gate }
81*7c478bd9Sstevel@tonic-gate
82*7c478bd9Sstevel@tonic-gate static void
set_flag(char * ifile,ulong_t flval)83*7c478bd9Sstevel@tonic-gate set_flag(char *ifile, ulong_t flval)
84*7c478bd9Sstevel@tonic-gate {
85*7c478bd9Sstevel@tonic-gate Elf *elf;
86*7c478bd9Sstevel@tonic-gate Elf_Scn *scn;
87*7c478bd9Sstevel@tonic-gate Elf_Data *data;
88*7c478bd9Sstevel@tonic-gate GElf_Shdr shdr;
89*7c478bd9Sstevel@tonic-gate GElf_Dyn dyn;
90*7c478bd9Sstevel@tonic-gate int fd, secidx, nent, i;
91*7c478bd9Sstevel@tonic-gate
92*7c478bd9Sstevel@tonic-gate (void) elf_version(EV_CURRENT);
93*7c478bd9Sstevel@tonic-gate
94*7c478bd9Sstevel@tonic-gate if ((fd = open(ifile, O_RDWR)) < 0)
95*7c478bd9Sstevel@tonic-gate die("Can't open %s", ifile);
96*7c478bd9Sstevel@tonic-gate
97*7c478bd9Sstevel@tonic-gate if ((elf = elf_begin(fd, ELF_C_RDWR, NULL)) == NULL)
98*7c478bd9Sstevel@tonic-gate elfdie("Can't start ELF for %s", ifile);
99*7c478bd9Sstevel@tonic-gate
100*7c478bd9Sstevel@tonic-gate if ((secidx = findelfsecidx(elf, ".dynamic")) == -1)
101*7c478bd9Sstevel@tonic-gate die("Can't find .dynamic section in %s\n", ifile);
102*7c478bd9Sstevel@tonic-gate
103*7c478bd9Sstevel@tonic-gate if ((scn = elf_getscn(elf, secidx)) == NULL)
104*7c478bd9Sstevel@tonic-gate elfdie("elf_getscn (%d)", secidx);
105*7c478bd9Sstevel@tonic-gate
106*7c478bd9Sstevel@tonic-gate if (gelf_getshdr(scn, &shdr) == NULL)
107*7c478bd9Sstevel@tonic-gate elfdie("gelf_shdr");
108*7c478bd9Sstevel@tonic-gate
109*7c478bd9Sstevel@tonic-gate if ((data = elf_getdata(scn, NULL)) == NULL)
110*7c478bd9Sstevel@tonic-gate elfdie("elf_getdata");
111*7c478bd9Sstevel@tonic-gate
112*7c478bd9Sstevel@tonic-gate nent = shdr.sh_size / shdr.sh_entsize;
113*7c478bd9Sstevel@tonic-gate for (i = 0; i < nent; i++) {
114*7c478bd9Sstevel@tonic-gate if (gelf_getdyn(data, i, &dyn) == NULL)
115*7c478bd9Sstevel@tonic-gate elfdie("gelf_getdyn");
116*7c478bd9Sstevel@tonic-gate
117*7c478bd9Sstevel@tonic-gate if (dyn.d_tag == DT_FLAGS_1) {
118*7c478bd9Sstevel@tonic-gate dyn.d_un.d_val |= (Elf64_Xword)flval;
119*7c478bd9Sstevel@tonic-gate
120*7c478bd9Sstevel@tonic-gate if (gelf_update_dyn(data, i, &dyn) == 0)
121*7c478bd9Sstevel@tonic-gate elfdie("gelf_update_dyn");
122*7c478bd9Sstevel@tonic-gate
123*7c478bd9Sstevel@tonic-gate break;
124*7c478bd9Sstevel@tonic-gate }
125*7c478bd9Sstevel@tonic-gate }
126*7c478bd9Sstevel@tonic-gate
127*7c478bd9Sstevel@tonic-gate if (i == nent) {
128*7c478bd9Sstevel@tonic-gate die("%s's .dynamic section doesn't have a DT_FLAGS_1 "
129*7c478bd9Sstevel@tonic-gate "field\n", ifile);
130*7c478bd9Sstevel@tonic-gate }
131*7c478bd9Sstevel@tonic-gate
132*7c478bd9Sstevel@tonic-gate if (elf_update(elf, ELF_C_WRITE) == -1)
133*7c478bd9Sstevel@tonic-gate elfdie("Couldn't update %s with changes", ifile);
134*7c478bd9Sstevel@tonic-gate
135*7c478bd9Sstevel@tonic-gate (void) elf_end(elf);
136*7c478bd9Sstevel@tonic-gate (void) close(fd);
137*7c478bd9Sstevel@tonic-gate }
138*7c478bd9Sstevel@tonic-gate
139*7c478bd9Sstevel@tonic-gate static ulong_t
parse_flag(char * optarg)140*7c478bd9Sstevel@tonic-gate parse_flag(char *optarg)
141*7c478bd9Sstevel@tonic-gate {
142*7c478bd9Sstevel@tonic-gate ulong_t flval = 0L;
143*7c478bd9Sstevel@tonic-gate char *arg;
144*7c478bd9Sstevel@tonic-gate int i;
145*7c478bd9Sstevel@tonic-gate
146*7c478bd9Sstevel@tonic-gate for (arg = strtok(optarg, ","); arg != NULL; arg = strtok(NULL, ",")) {
147*7c478bd9Sstevel@tonic-gate for (i = 0; dtflagvals[i].fv_name != NULL; i++) {
148*7c478bd9Sstevel@tonic-gate if (strcmp(dtflagvals[i].fv_name, arg) == 0)
149*7c478bd9Sstevel@tonic-gate flval |= dtflagvals[i].fv_val;
150*7c478bd9Sstevel@tonic-gate }
151*7c478bd9Sstevel@tonic-gate }
152*7c478bd9Sstevel@tonic-gate
153*7c478bd9Sstevel@tonic-gate return (flval);
154*7c478bd9Sstevel@tonic-gate }
155*7c478bd9Sstevel@tonic-gate
156*7c478bd9Sstevel@tonic-gate int
main(int argc,char ** argv)157*7c478bd9Sstevel@tonic-gate main(int argc, char **argv)
158*7c478bd9Sstevel@tonic-gate {
159*7c478bd9Sstevel@tonic-gate ulong_t flval = 0L;
160*7c478bd9Sstevel@tonic-gate int c;
161*7c478bd9Sstevel@tonic-gate
162*7c478bd9Sstevel@tonic-gate progname = basename(argv[0]);
163*7c478bd9Sstevel@tonic-gate
164*7c478bd9Sstevel@tonic-gate while ((c = getopt(argc, argv, "f:")) != EOF) {
165*7c478bd9Sstevel@tonic-gate switch (c) {
166*7c478bd9Sstevel@tonic-gate case 'f':
167*7c478bd9Sstevel@tonic-gate if ((flval = strtoul(optarg, NULL, 0)) == 0 &&
168*7c478bd9Sstevel@tonic-gate (flval = parse_flag(optarg)) == 0)
169*7c478bd9Sstevel@tonic-gate usage();
170*7c478bd9Sstevel@tonic-gate break;
171*7c478bd9Sstevel@tonic-gate default:
172*7c478bd9Sstevel@tonic-gate usage();
173*7c478bd9Sstevel@tonic-gate }
174*7c478bd9Sstevel@tonic-gate }
175*7c478bd9Sstevel@tonic-gate
176*7c478bd9Sstevel@tonic-gate if (flval == 0 || argc - optind != 1)
177*7c478bd9Sstevel@tonic-gate usage();
178*7c478bd9Sstevel@tonic-gate
179*7c478bd9Sstevel@tonic-gate set_flag(argv[optind], flval);
180*7c478bd9Sstevel@tonic-gate
181*7c478bd9Sstevel@tonic-gate return (0);
182*7c478bd9Sstevel@tonic-gate }
183