1 /*- 2 * Copyright (c) 2004, The DragonFly Project. All rights reserved. 3 * Copyright (c) 1998, Peter Wemm <peter@netplex.com.au> 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 * 27 * $FreeBSD: src/usr.bin/objformat/objformat.c,v 1.6 1998/10/24 02:01:30 jdp Exp $ 28 */ 29 30 #include <sys/param.h> 31 32 #include <err.h> 33 #include <objformat.h> 34 #include <stdio.h> 35 #include <stdlib.h> 36 #include <string.h> 37 #include <unistd.h> 38 39 #ifndef CCVER_DEFAULT 40 #define CCVER_DEFAULT "gcc44" 41 #endif 42 43 #ifndef BINUTILSVER_DEFAULT 44 #define BINUTILSVER_DEFAULT "binutils221" 45 #endif 46 47 #ifndef OBJFORMAT_PATH_DEFAULT 48 #define OBJFORMAT_PATH_DEFAULT "" 49 #endif 50 51 enum cmd_type { OBJFORMAT, COMPILER, BINUTILS }; 52 53 struct command { 54 const char *cmd; 55 enum cmd_type type; 56 }; 57 58 static struct command commands[] = { 59 {"CC", COMPILER}, 60 {"c++", COMPILER}, 61 {"cc", COMPILER}, 62 {"cpp", COMPILER}, 63 {"g++", COMPILER}, 64 {"gcc", COMPILER}, 65 {"gcov", COMPILER}, 66 {"addr2line", BINUTILS}, 67 {"ar", BINUTILS}, 68 {"as", BINUTILS}, 69 {"c++filt", BINUTILS}, 70 {"elfedit", BINUTILS}, 71 {"gprof", BINUTILS}, 72 {"ld", BINUTILS}, 73 {"nm", BINUTILS}, 74 {"objcopy", BINUTILS}, 75 {"objdump", BINUTILS}, 76 {"ranlib", BINUTILS}, 77 {"readelf", BINUTILS}, 78 {"size", BINUTILS}, 79 {"strings", BINUTILS}, 80 {"strip", BINUTILS}, 81 {"objformat", OBJFORMAT}, 82 {"", -1} 83 }; 84 85 int 86 main(int argc, char **argv) 87 { 88 struct command *cmds; 89 char objformat[32]; 90 char *path, *chunk; 91 char *cmd, *newcmd = NULL; 92 const char *objformat_path; 93 const char *ccver; 94 const char *buver; 95 const char *env_value = NULL; 96 const char *base_path = NULL; 97 int use_objformat = 0; 98 99 if (getobjformat(objformat, sizeof objformat, &argc, argv) == -1) 100 errx(1, "Invalid object format"); 101 102 /* 103 * Get the last path element of the program name being executed 104 */ 105 cmd = strrchr(argv[0], '/'); 106 if (cmd != NULL) 107 cmd++; 108 else 109 cmd = argv[0]; 110 111 for (cmds = commands; cmds < &commands[NELEM(commands) - 1]; ++cmds) { 112 if (strcmp(cmd, cmds->cmd) == 0) 113 break; 114 } 115 116 if ((ccver = getenv("CCVER")) == NULL || ccver[0] == 0) 117 ccver = CCVER_DEFAULT; 118 if ((buver = getenv("BINUTILSVER")) == NULL) { 119 buver = BINUTILSVER_DEFAULT; 120 } 121 122 if (cmds) { 123 switch (cmds->type) { 124 case COMPILER: 125 base_path = "/usr/libexec"; 126 use_objformat = 0; 127 env_value = ccver; 128 break; 129 case BINUTILS: 130 base_path = "/usr/libexec"; 131 use_objformat = 1; 132 env_value = buver; 133 break; 134 case OBJFORMAT: 135 break; 136 default: 137 errx(1, "unknown command type"); 138 break; 139 } 140 } 141 142 /* 143 * The objformat command itself doesn't need another exec 144 */ 145 if (cmds->type == OBJFORMAT) { 146 if (argc != 1) { 147 fprintf(stderr, "Usage: objformat\n"); 148 exit(1); 149 } 150 151 printf("%s\n", objformat); 152 exit(0); 153 } 154 155 /* 156 * make buildworld glue and CCVER overrides. 157 */ 158 objformat_path = getenv("OBJFORMAT_PATH"); 159 if (objformat_path == NULL) 160 objformat_path = OBJFORMAT_PATH_DEFAULT; 161 162 again: 163 path = strdup(objformat_path); 164 165 if (setenv("OBJFORMAT", objformat, 1) == -1) 166 err(1, "setenv: cannot set OBJFORMAT=%s", objformat); 167 168 /* 169 * objformat_path could be sequence of colon-separated paths. 170 */ 171 while ((chunk = strsep(&path, ":")) != NULL) { 172 if (newcmd != NULL) { 173 free(newcmd); 174 newcmd = NULL; 175 } 176 if (use_objformat) { 177 asprintf(&newcmd, "%s%s/%s/%s/%s", 178 chunk, base_path, env_value, objformat, cmd); 179 } else { 180 asprintf(&newcmd, "%s%s/%s/%s", 181 chunk, base_path, env_value, cmd); 182 } 183 if (newcmd == NULL) 184 err(1, "cannot allocate memory"); 185 186 argv[0] = newcmd; 187 execv(newcmd, argv); 188 } 189 190 /* 191 * Fallback: if we're searching for a compiler, but didn't 192 * find any, try again using the custom compiler driver. 193 */ 194 if (cmds && cmds->type == COMPILER && 195 strcmp(env_value, "custom") != 0) { 196 env_value = "custom"; 197 goto again; 198 } 199 200 if (use_objformat) { 201 err(1, "in path [%s]%s/%s/%s/%s", 202 objformat_path, base_path, env_value, objformat, cmd); 203 } else { 204 err(1, "in path [%s]%s/%s/%s", 205 objformat_path, base_path, env_value, cmd); 206 } 207 } 208