1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include "cpp.h"
5
6 Includelist includelist[NINCLUDE];
7
8 extern char *objname;
9
appendDirToIncludeList(char * dir)10 void appendDirToIncludeList( char *dir )
11 {
12 int i;
13 char *fqdir;
14
15 fqdir = (char *)newstring( (uchar *)includelist[NINCLUDE-1].file, 256, 0 );
16 strcat( fqdir, "/" );
17 strcat( fqdir, dir );
18
19 //avoid adding it more than once
20 for (i=NINCLUDE-2; i>=0; i--) {
21 if (includelist[i].file &&
22 !strcmp (includelist[i].file, fqdir)) {
23 return;
24 }
25 }
26
27 for (i=NINCLUDE-2; i>=0; i--) {
28 if (includelist[i].file==NULL) {
29 includelist[i].always = 1;
30 includelist[i].file = fqdir;
31 break;
32 }
33 }
34 if (i<0)
35 error(FATAL, "Too many -I directives");
36 }
37
38 void
doinclude(Tokenrow * trp)39 doinclude(Tokenrow *trp)
40 {
41 char fname[256], iname[256];
42 Includelist *ip;
43 int angled, len, fd, i;
44
45 trp->tp += 1;
46 if (trp->tp>=trp->lp)
47 goto syntax;
48 if (trp->tp->type!=STRING && trp->tp->type!=LT) {
49 len = trp->tp - trp->bp;
50 expandrow(trp, "<include>");
51 trp->tp = trp->bp+len;
52 }
53 if (trp->tp->type==STRING) {
54 len = trp->tp->len-2;
55 if (len > sizeof(fname) - 1)
56 len = sizeof(fname) - 1;
57 strncpy(fname, (char*)trp->tp->t+1, len);
58 angled = 0;
59 } else if (trp->tp->type==LT) {
60 len = 0;
61 trp->tp++;
62 while (trp->tp->type!=GT) {
63 if (trp->tp>trp->lp || len+trp->tp->len+2 >= sizeof(fname))
64 goto syntax;
65 strncpy(fname+len, (char*)trp->tp->t, trp->tp->len);
66 len += trp->tp->len;
67 trp->tp++;
68 }
69 angled = 1;
70 } else
71 goto syntax;
72 trp->tp += 2;
73 if (trp->tp < trp->lp || len==0)
74 goto syntax;
75 fname[len] = '\0';
76
77 appendDirToIncludeList( basepath( fname ) );
78
79 if (fname[0]=='/') {
80 fd = open(fname, 0);
81 strcpy(iname, fname);
82 } else for (fd = -1,i=NINCLUDE-1; i>=0; i--) {
83 ip = &includelist[i];
84 if (ip->file==NULL || ip->deleted || (angled && ip->always==0))
85 continue;
86 if (strlen(fname)+strlen(ip->file)+2 > sizeof(iname))
87 continue;
88 strcpy(iname, ip->file);
89 strcat(iname, "/");
90 strcat(iname, fname);
91 if ((fd = open(iname, 0)) >= 0)
92 break;
93 }
94 if ( Mflag>1 || (!angled&&Mflag==1) ) {
95 write(1,objname,strlen(objname));
96 write(1,iname,strlen(iname));
97 write(1,"\n",1);
98 }
99 if (fd >= 0) {
100 if (++incdepth > 10)
101 error(FATAL, "#include too deeply nested");
102 setsource((char*)newstring((uchar*)iname, strlen(iname), 0), fd, NULL);
103 genline();
104 } else {
105 trp->tp = trp->bp+2;
106 error(ERROR, "Could not find include file %r", trp);
107 }
108 return;
109 syntax:
110 error(ERROR, "Syntax error in #include");
111 return;
112 }
113
114 /*
115 * Generate a line directive for cursource
116 */
117 void
genline(void)118 genline(void)
119 {
120 static Token ta = { UNCLASS };
121 static Tokenrow tr = { &ta, &ta, &ta+1, 1 };
122 uchar *p;
123
124 ta.t = p = (uchar*)outp;
125 strcpy((char*)p, "#line ");
126 p += sizeof("#line ")-1;
127 p = (uchar*)outnum((char*)p, cursource->line);
128 *p++ = ' '; *p++ = '"';
129 if (cursource->filename[0]!='/' && wd[0]) {
130 strcpy((char*)p, wd);
131 p += strlen(wd);
132 *p++ = '/';
133 }
134 strcpy((char*)p, cursource->filename);
135 p += strlen((char*)p);
136 *p++ = '"'; *p++ = '\n';
137 ta.len = (char*)p-outp;
138 outp = (char*)p;
139 tr.tp = tr.bp;
140 puttokens(&tr);
141 }
142
143 void
setobjname(char * f)144 setobjname(char *f)
145 {
146 int n = strlen(f);
147 objname = (char*)domalloc(n+5);
148 strcpy(objname,f);
149 if(objname[n-2]=='.'){
150 strcpy(objname+n-1,"$O: ");
151 }else{
152 strcpy(objname+n,"$O: ");
153 }
154 }
155