usr.bin/c99/c99.c

   1 /*-
   2  * Copyright (c) 2002 Tim J. Robbins.
   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
  27 /*
  28  * c99 -- compile standard C programs
  29  *
  30  * This is essentially a wrapper around the system C compiler that forces
  31  * the compiler into C99 mode and handles some of the standard libraries
  32  * specially.
  33  */
  34
  35 #include <sys/cdefs.h>
  36 __FBSDID("$FreeBSD$");
  37
  38 #include <sys/types.h>
  39
  40 #include <err.h>
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45
  46 static char **args;
  47 static u_int cargs, nargs;
  48
  49 static void addarg(const char *);
  50 static void addlib(const char *);
  51 static void usage(void);
  52
  53 int
  54 main(int argc, char *argv[])
  55 {
  56         int ch, i;
  57
  58         args = NULL;
  59         cargs = nargs = 0;
  60
  61         while ((ch = getopt(argc, argv, "cD:EgI:L:o:O:sU:l:")) != -1) {
  62                 if (ch == 'l') {
  63                         /* Gone too far. Back up and get out. */
  64                         if (argv[optind - 1][0] == '-')
  65                                 optind -= 1;
  66                         else
  67                                 optind -= 2;
  68                         break;
  69                 } else if (ch == '?')
  70                         usage();
  71         }
  72
  73         addarg("/usr/bin/cc");
  74         addarg("-std=iso9899:1999");
  75         addarg("-pedantic");
  76         for (i = 1; i < optind; i++)
  77                 addarg(argv[i]);
  78         while (i < argc) {
  79                 if (strncmp(argv[i], "-l", 2) == 0) {
  80                         if (argv[i][2] != '\0')
  81                                 addlib(argv[i++] + 2);
  82                         else {
  83                                 if (argv[++i] == NULL)
  84                                         usage();
  85                                 addlib(argv[i++]);
  86                         }
  87                 } else
  88                         addarg(argv[i++]);
  89         }
  90         execv("/usr/bin/cc", args);
  91         err(1, "/usr/bin/cc");
  92 }
  93
  94 static void
  95 addarg(const char *item)
  96 {
  97         if (nargs + 1 >= cargs) {
  98                 cargs += 16;
  99                 if ((args = realloc(args, sizeof(*args) * cargs)) == NULL)
 100                         err(1, "malloc");
 101         }
 102         if ((args[nargs++] = strdup(item)) == NULL)
 103                 err(1, "strdup");
 104         args[nargs] = NULL;
 105 }
 106
 107 static void
 108 addlib(const char *lib)
 109 {
 110
 111         if (strcmp(lib, "pthread") == 0)
 112                 /* FreeBSD's gcc uses -pthread instead of -lpthread. */
 113                 addarg("-pthread");
 114         else if (strcmp(lib, "rt") == 0)
 115                 /* librt functionality is in libc or unimplemented. */
 116                 ;
 117         else if (strcmp(lib, "xnet") == 0)
 118                 /* xnet functionality is in libc. */
 119                 ;
 120         else {
 121                 addarg("-l");
 122                 addarg(lib);
 123         }
 124 }
 125
 126 static void
 127 usage(void)
 128 {
 129         (void)fprintf(stderr, "%s\n%s\n",
 130 "usage: c99 [-cEgs] [-D name[=value]] ... [-I directory] ... [-L directory] ...",
 131 "       [-o outfile] [-O optlevel] [-U name] ... operand ...");
 132         exit(1);
 133 }