#include <sys/cdefs.h>
#include <sys/types.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
char **args;
u_int cargs, nargs;
void addarg(const char *);
void addlib(const char *);
void combine_and_addarg (const char *, const char *);
void usage(void);
char **undef_args;
u_int undef_cargs, undef_nargs;
void record_undef(const char *);
void add_def(const char *);
void combine_and_addarg (const char *, const char *);
int dash_dash_seen = 0;
int
main(int argc, char *argv[])
{
int ch, i;
int link = 1;
int inputs = 0;
args = NULL;
cargs = nargs = 0;
while ((ch = getopt(argc, argv, "cD:EgI:L:o:O:sU:l:")) != -1) {
if (ch == 'l') {
if (argv[optind - 1][0] == '-')
optind -= 1;
else
optind -= 2;
break;
} else if (ch == '?')
usage();
}
addarg("cc");
addarg("-std=iso9899:1999");
addarg("-pedantic");
addarg("-Wextra-tokens");
addarg("-mlong-double-64");
addarg("-fmath-errno");
addarg("-fno-builtin-pow");
addarg("-fno-builtin-powl");
addarg("-fno-builtin-powf");
for (i = 1; i < optind; i++) {
if (strcmp (argv[i], "--") == 0)
dash_dash_seen = 1;
else if (strcmp (argv[i], "-O") == 0) {
if (i+1 < argc) {
if (strcmp (argv[i+1], "1") == 0 || strcmp (argv[i+1], "0") == 0) {
combine_and_addarg(argv[i], argv[i+1]);
i++;
} else
addarg(argv[i]);
}
} else if (strcmp (argv[i], "-U") == 0) {
record_undef (argv[i+1]);
addarg(argv[i]);
addarg(argv[i+1]);
i++;
} else if (strncmp (argv[i], "-U", 2) == 0) {
record_undef (argv[i]+2);
addarg(argv[i]);
} else if (strcmp (argv[i], "-L") == 0 && i+1 < argc) {
combine_and_addarg(argv[i], argv[i+1]);
i++;
} else if (strcmp (argv[i], "-D") == 0) {
add_def (argv[i+1]);
i++;
} else if (strncmp (argv[i], "-D", 2) == 0)
add_def (argv[i]+2);
else {
addarg(argv[i]);
if (argv[i][0] == '-') {
if (argv[i][1] == 'c' ||
argv[i][1] == 'E')
link = 0;
} else
inputs++;
}
}
while (i < argc) {
if (strncmp(argv[i], "-l", 2) == 0) {
if (argv[i][2] != '\0')
addlib(argv[i++] + 2);
else {
if (argv[++i] == NULL)
usage();
addlib(argv[i++]);
}
}
else if (strcmp (argv[i], "-L") == 0 && i+1 < argc) {
combine_and_addarg(argv[i], argv[i+1]);
i++;
i++;
}
else if (strcmp (argv[i], "--") == 0) {
dash_dash_seen = 1;
i++;
} else {
addarg(argv[i++]);
inputs++;
}
}
if (link && inputs > 0) {
dash_dash_seen = 0;
addlib("iconv");
}
execv("/usr/bin/cc", args);
err(1, "/usr/bin/cc");
}
void
combine_and_addarg (const char *item1, const char *item2)
{
char *item = (char *) malloc (sizeof (char) * (strlen(item1) + strlen(item2) + 1));
strcpy (item, item1);
strcat (item, item2);
addarg (item);
}
void
record_undef (const char *item)
{
if (undef_nargs + 1 > undef_cargs) {
undef_cargs += 16;
if ((undef_args = realloc(undef_args, sizeof(*undef_args) * undef_cargs)) == NULL)
err(1, "malloc");
}
if ((undef_args[undef_nargs++] = strdup(item)) == NULL)
err(1, "strdup");
undef_args[undef_nargs] = NULL;
}
void
add_def (const char *item)
{
u_int i;
char *updated_def;
if (strlen (item) > 2)
for (i = 0; i < undef_nargs; i++)
{
if (strcmp (item, undef_args[i]) == 0)
return;
else {
int undef_len = strlen (undef_args[i]);
int def_len = strlen (item);
if (undef_len < def_len) {
const char *p = item + undef_len;
if (*p == '=') {
if (strncmp (item, undef_args[i], undef_len) == 0)
return;
}
}
}
}
updated_def = (char *) malloc (strlen (item) + 3);
sprintf (updated_def,"-D%s",item);
addarg (updated_def);
}
void
addarg(const char *item)
{
if (nargs + 2 > cargs) {
cargs += 16;
if ((args = realloc(args, sizeof(*args) * cargs)) == NULL)
err(1, "malloc");
}
if (dash_dash_seen && strncmp (item, "-", 1) == 0)
{
char *updated_item = (char *) malloc (strlen (item) + 3);
sprintf (updated_item,"./%s",item);
args[nargs++] = updated_item;
}
else
{
args[nargs++] = strdup (item);
}
if (args[nargs-1] == NULL)
err(1, "strdup");
args[nargs] = NULL;
}
void
addlib(const char *lib)
{
if (strcmp(lib, "pthread") == 0)
;
else if (strcmp(lib, "rt") == 0)
;
else if (strcmp(lib, "xnet") == 0)
;
else {
addarg("-l");
addarg(lib);
}
}
void
usage(void)
{
fprintf(stderr,
"usage: c99 [-cEgs] [-D name[=value]] [-I directory] ... [-L directory] ...\n");
fprintf(stderr,
" [-o outfile] [-O optlevel] [-U name]... operand ...\n");
exit(1);
}