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