#include <stdlib.h>
#include <string.h>
#include "cpp.h"

Includelist includelist[NINCLUDE];

void doinclude(Tokenrow * trp)
{
	char fname[256], iname[256];
	Includelist *ip;
	int angled, len, fd, i;

	trp->tp += 1;
	if (trp->tp >= trp->lp)
		goto syntax;
	if (trp->tp->type != STRING && trp->tp->type != LT)
	{
		len = trp->tp - trp->bp;
		expandrow(trp, "<include>");
		trp->tp = trp->bp + len;
	}
	if (trp->tp->type == STRING)
	{
		len = trp->tp->len - 2;
		if (len > (int)sizeof(fname) - 1)
			len = sizeof(fname) - 1;
		strncpy(fname, (char *)trp->tp->t + 1, len);
		angled = 0;
	}
	else if (trp->tp->type == LT)
	{
		len = 0;
		trp->tp++;
		while (trp->tp->type != GT)
		{
			if (trp->tp > trp->lp || len + trp->tp->len + 2 >= sizeof(fname))
				goto syntax;
			strncpy(fname + len, (char *)trp->tp->t, trp->tp->len);
			len += trp->tp->len;
			trp->tp++;
		}
		angled = 1;
	}
	else
		goto syntax;
	trp->tp += 2;
	if (trp->tp < trp->lp || len == 0)
		goto syntax;
	fname[len] = '\0';
	if (fname[0] == '/')
	{
		fd = open(fname, 0);
		strcpy(iname, fname);
	}
	else
	{
		if (!angled)
		{
			char *sp;
			strcpy(iname, cursource->filename);
			sp = strrchr(iname, '/');
			if (sp)
			{
				strcpy(sp + 1, fname);
				if ((fd = open(iname, 0)) >= 0)
					goto found;
			}
		}
		for (fd = -1, i = NINCLUDE - 1; i >= 0; i--)
		{
			ip = &includelist[i];
			if (ip->file == NULL || (angled && ip->always == 0))
				continue;
			if (strlen(fname) + strlen(ip->file) + 2 > sizeof(iname))
				continue;
			strcpy(iname, ip->file);
			strcat(iname, "/");
			strcat(iname, fname);
			if ((fd = open(iname, 0)) >= 0)
				break;
		}
	}
  found:
	if (fd >= 0)
	{
		if (++incdepth > 10)
			error(FATAL, "#include too deeply nested");
		setsource((char *)newstring((uchar *) iname, strlen(iname), 0), fd,
			NULL);
		genline();
	}
	else
	{
		trp->tp = trp->bp + 2;
		error(ERROR, "Could not find include file %r", trp);
	}
	return;
  syntax:
	error(ERROR, "Syntax error in #include");
	return;
}

/*
 * Generate a line directive for cursource
 */
void genline(void)
{
	static Token ta = { UNCLASS };
	static Tokenrow tr = { &ta, &ta, &ta + 1, 1 };
	uchar *p;

	ta.t = p = (uchar *) outp;
	strcpy((char *)p, "#line ");
	p += sizeof("#line ") - 1;
	p = (uchar *) outnum((char *)p, cursource->line);
	*p++ = ' ';
	*p++ = '"';
	if (cursource->filename[0] != '/' && wd[0])
	{
		strcpy((char *)p, wd);
		p += strlen(wd);
		*p++ = '/';
	}
	strcpy((char *)p, cursource->filename);
	p += strlen((char *)p);
	*p++ = '"';
	*p++ = '\n';
	ta.len = (char *)p - outp;
	outp = (char *)p;
	tr.tp = tr.bp;
	puttokens(&tr);
}

