lib/libcr/gen/getobjformat.c

   1 /*-
   2  * Copyright (c) 1998 John D. Polstra
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  * $FreeBSD: src/lib/libc/gen/getobjformat.c,v 1.3 1999/12/18 04:47:43 obrien Exp $
  27  * $DragonFly: src/lib/libcr/gen/Attic/getobjformat.c,v 1.2 2003/06/17 04:26:42 dillon Exp $
  28  */
  29
  30 #include <sys/param.h>
  31 #include <objformat.h>
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <string.h>
  35
  36 #define PATH_OBJFORMAT  "/etc/objformat"
  37
  38 static int copyformat(char *, const char *, size_t);
  39
  40 static const char *known_formats[] = { OBJFORMAT_NAMES, NULL };
  41
  42 static int
  43 copyformat(char *buf, const char *fmt, size_t bufsize)
  44 {
  45         size_t           len;
  46
  47         len = strlen(fmt);
  48         if (len > bufsize - 1)
  49                 return -1;
  50         strcpy(buf, fmt);
  51         return len;
  52 }
  53
  54 int
  55 getobjformat(char *buf, size_t bufsize, int *argcp, char **argv)
  56 {
  57         const char       *fmt;
  58         char            **src, **dst;
  59         const char       *env;
  60         FILE             *fp;
  61
  62         fmt = NULL;
  63
  64         if (argv != NULL) {
  65                 /*
  66                  * Scan for arguments setting known formats, e.g., "-elf".
  67                  * If "argcp" is non-NULL, delete these arguments from the
  68                  * list and update the argument count in "*argcp".
  69                  */
  70                 for (dst = src = argv;  *src != NULL;  src++) {
  71                         if ((*src)[0] == '-') {
  72                                 const char **p;
  73
  74                                 for (p = known_formats;  *p != NULL;  p++)
  75                                         if (strcmp(*src + 1, *p) == 0)
  76                                                 break;
  77                                 if (*p != NULL) {
  78                                         fmt = *p;
  79                                         if (argcp == NULL)  /* Don't delete */
  80                                                 *dst++ = *src;
  81                                 } else
  82                                         *dst++ = *src;
  83                         } else
  84                                 *dst++ = *src;
  85                 }
  86                 *dst = NULL;
  87                 if (argcp != NULL)
  88                         *argcp -= src - dst;
  89                 if (fmt != NULL)
  90                         return copyformat(buf, fmt, bufsize);
  91         }
  92
  93         /* Check the OBJFORMAT environment variable. */
  94         if ((env = getenv("OBJFORMAT")) != NULL)
  95                 return copyformat(buf, env, bufsize);
  96
  97         /* Take a look at "/etc/objformat". */
  98         if ((fp = fopen(PATH_OBJFORMAT, "r")) != NULL) {
  99                 char line[1024];
 100                 int found;
 101                 int len;
 102
 103                 found = len = 0;
 104                 while (fgets(line, sizeof line, fp) != NULL) {
 105                         if (strncmp(line, "OBJFORMAT=", 10) == 0) {
 106                                 char *p = &line[10];
 107
 108                                 p[strcspn(p, " \t\n")] = '\0';
 109                                 len = copyformat(buf, p, bufsize);
 110                                 found = 1;
 111                         }
 112                 }
 113                 fclose(fp);
 114                 if (found)
 115                         return len;
 116         }
 117
 118         /* As a last resort, use the compiled in default. */
 119         return copyformat(buf, OBJFORMAT_DEFAULT, bufsize);
 120 }