shithub: scc

ref: f4273538d57487fc60fae6a3c6d44b9b083aace8
dir: /cc1/main.c/

/* See LICENSE file for copyright and license details. */
static char sccsid[] = "@(#) ./cc1/main.c";
#include <setjmp.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>

#include "../inc/arg.h"
#include "../inc/cc.h"
#include "cc1.h"

char *argv0, *infile, *outfile;

int warnings;
jmp_buf recover;

static struct items uflags;
int onlycpp, onlyheader;


extern int failure;

static void
clean(void)
{
	if (failure && outfile)
		remove(outfile);
}

static void
defmacro(char *macro)
{
	char *p = strchr(macro, '=');

	if (p)
		*p++ = '\0';
	else
		p = "1";

	defdefine(macro, p, "command-line");
}

static void
usage(void)
{
	die("usage: cc1 [-Ewd] [-D def[=val]]... [-U def]... "
	    "[-I dir]... [-o output] [input]");
}

int
main(int argc, char *argv[])
{
	char *cp;
	int i;

	atexit(clean);
	ilex();
	icpp();

	ARGBEGIN {
	case 'D':
		defmacro(EARGF(usage()));
		break;
	case 'M':
		onlyheader = 1;
		break;
	case 'E':
		onlycpp = 1;
		break;
	case 'I':
		incdir(EARGF(usage()));
		break;
	case 'U':
		newitem(&uflags, EARGF(usage()));
		break;
	case 'd':
		DBGON();
		break;
	case 'o':
		outfile = EARGF(usage());
		break;
	case 'w':
		warnings = 1;
		break;
	default:
		usage();
	} ARGEND

	if (argc > 1)
		usage();

	if (outfile && !freopen(outfile, "w", stdout))
		die("error opening output: %s", strerror(errno));

	for (i = 0; i < uflags.n; ++i)
		undefmacro(uflags.s[i]);

	infile = (*argv) ? *argv : "<stdin>";
	if (!addinput(*argv, NULL, NULL)) {
		die("error: failed to open input file '%s': %s",
		    *argv, strerror(errno));
	}
	if (onlycpp || onlyheader) {
		outcpp();
	} else {
		for (next(); yytoken != EOFTOK; decl())
			/* nothing */;
	}

	return failure;
}