Commit | Line | Data |
---|---|---|
ae320d11 TL |
1 | #include "stdio.h" |
2 | #include "awk.def" | |
3 | #include "awk.h" | |
4 | #include "ctype.h" | |
5 | ||
6 | FILE *infile = NULL; | |
7 | char *file; | |
8 | #define RECSIZE 512 | |
9 | char record[RECSIZE]; | |
10 | char fields[RECSIZE]; | |
11 | ||
12 | #define MAXFLD 50 | |
13 | int donefld; /* 1 = implies rec broken into fields */ | |
14 | int donerec; /* 1 = record is valid (no flds have changed) */ | |
15 | int mustfld; /* 1 = NF seen, so always break*/ | |
16 | ||
17 | #define FINIT {0, NULL, 0.0, FLD|STR} | |
18 | cell fldtab[MAXFLD] = { /*room for fields */ | |
19 | { "$record", record, 0.0, STR|FLD}, | |
20 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, | |
21 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, | |
22 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, | |
23 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, | |
24 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, | |
25 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, | |
26 | FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT | |
27 | }; | |
28 | int maxfld = 0; /* last used field */ | |
29 | ||
30 | ||
31 | getrec() | |
32 | { | |
33 | register char *rr; | |
34 | extern int svargc; | |
35 | extern char **svargv; | |
36 | register c, sep; | |
37 | ||
38 | dprintf("**RS=%o, **FS=%o\n", **RS, **FS, NULL); | |
39 | donefld = 0; | |
40 | donerec = 1; | |
41 | record[0] = 0; | |
42 | while (svargc > 0) { | |
43 | dprintf("svargc=%d, *svargv=%s\n", svargc, *svargv, NULL); | |
44 | if (infile == NULL) { /* have to open a new file */ | |
45 | if (member('=', *svargv)) { /* it's a var=value argument */ | |
46 | setclvar(*svargv); | |
47 | svargv++; | |
48 | svargc--; | |
49 | continue; | |
50 | } | |
51 | *FILENAME = file = *svargv; | |
52 | dprintf("opening file %s\n", file, NULL, NULL); | |
53 | if (*file == '-') | |
54 | infile = stdin; | |
55 | else if ((infile = fopen(file, "r")) == NULL) | |
56 | error(FATAL, "can't open %s", file); | |
57 | } | |
58 | if ((sep = **RS) == 0) | |
59 | sep = '\n'; | |
60 | for (rr = record; ; ) { | |
61 | for (; (c=getc(infile)) != sep && c != EOF; *rr++ = c) | |
62 | ; | |
63 | if (**RS == sep || c == EOF) | |
64 | break; | |
65 | if ((c = getc(infile)) == '\n' || c == EOF) /* 2 in a row */ | |
66 | break; | |
67 | *rr++ = '\n'; | |
68 | *rr++ = c; | |
69 | } | |
70 | if (rr > record+RECSIZE) | |
71 | error(FATAL, "record `%.20s...' too long", record); | |
72 | *rr = 0; | |
73 | if (mustfld) | |
74 | fldbld(); | |
75 | if (c != EOF) /* normal record */ | |
76 | return(1); | |
77 | /* EOF arrived on this file; set up next */ | |
78 | if (infile != stdin) | |
79 | fclose(infile); | |
80 | infile = NULL; | |
81 | svargc--; | |
82 | svargv++; | |
83 | } | |
84 | return(0); /* true end of file */ | |
85 | } | |
86 | ||
87 | setclvar(s) /* set var=value from s */ | |
88 | char *s; | |
89 | { | |
90 | char *p; | |
91 | cell *q; | |
92 | ||
93 | for (p=s; *p != '='; p++) | |
94 | ; | |
95 | *p++ = 0; | |
96 | q = setsymtab(s, tostring(p), 0.0, STR, symtab); | |
97 | setsval(q, p); | |
98 | dprintf("command line set %s to |%s|\n", s, p, NULL); | |
99 | } | |
100 | ||
101 | fldbld() | |
102 | { | |
103 | register char *r, *fr, sep; | |
104 | int i, j; | |
105 | ||
106 | r = record; | |
107 | fr = fields; | |
108 | if ((sep = **FS) == ' ') | |
109 | for (i = 0; ; ) { | |
110 | while (*r == ' ' || *r == '\t' || *r == '\n') | |
111 | r++; | |
112 | if (*r == 0) | |
113 | break; | |
114 | i++; | |
115 | if (i >= MAXFLD) | |
116 | error(FATAL, "record `%.20s...' has too many fields", record); | |
117 | if (!(fldtab[i].tval&FLD)) | |
118 | xfree(fldtab[i].sval); | |
119 | fldtab[i].sval = fr; | |
120 | fldtab[i].tval = FLD | STR; | |
121 | do | |
122 | *fr++ = *r++; | |
123 | while (*r != ' ' && *r != '\t' && *r != '\n' && *r != '\0'); | |
124 | *fr++ = 0; | |
125 | } | |
126 | else | |
127 | for (i = 0; ; ) { | |
128 | i++; | |
129 | if (i >= MAXFLD) | |
130 | error(FATAL, "record `%.20s...' has too many fields", record); | |
131 | if (!(fldtab[i].tval&FLD)) | |
132 | xfree(fldtab[i].sval); | |
133 | fldtab[i].sval = fr; | |
134 | fldtab[i].tval = FLD | STR; | |
135 | while (*r != sep && *r != '\n' && *r != '\0') /* \n always a separator */ | |
136 | *fr++ = *r++; | |
137 | *fr++ = '\0'; | |
138 | if (*r == 0) break; | |
139 | r++; | |
140 | } | |
141 | *fr = 0; | |
142 | for (j=maxfld; j>i; j--) { /* clean out junk from previous record */ | |
143 | if (!(fldtab[j].tval&FLD)) | |
144 | xfree(fldtab[j].sval); | |
145 | fldtab[j].tval = STR | FLD; | |
146 | fldtab[j].sval = NULL; | |
147 | } | |
148 | maxfld = i; | |
149 | donefld = 1; | |
150 | for(i=1; i<=maxfld; i++) | |
151 | if(isnumber(fldtab[i].sval)) | |
152 | { fldtab[i].fval = atof(fldtab[i].sval); | |
153 | fldtab[i].tval |= NUM; | |
154 | } | |
155 | setfval(lookup("NF", symtab), (awkfloat) maxfld); | |
156 | if (dbg) | |
157 | for (i = 0; i <= maxfld; i++) | |
158 | printf("field %d: |%s|\n", i, fldtab[i].sval); | |
159 | } | |
160 | ||
161 | recbld() | |
162 | { | |
163 | int i; | |
164 | register char *r, *p; | |
165 | ||
166 | if (donefld == 0 || donerec == 1) | |
167 | return; | |
168 | r = record; | |
169 | for (i = 1; i <= *NF; i++) { | |
170 | p = getsval(&fldtab[i]); | |
171 | while (*r++ = *p++) | |
172 | ; | |
173 | *(r-1) = **OFS; | |
174 | } | |
175 | *(r-1) = '\0'; | |
176 | dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL); | |
177 | recloc->tval = STR | FLD; | |
178 | dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL); | |
179 | if (r > record+RECSIZE) | |
180 | error(FATAL, "built giant record `%.20s...'", record); | |
181 | dprintf("recbld = |%s|\n", record, NULL, NULL); | |
182 | } | |
183 | ||
184 | cell *fieldadr(n) | |
185 | { | |
186 | if (n >= MAXFLD) | |
187 | error(FATAL, "trying to access field %d", n); | |
188 | return(&fldtab[n]); | |
189 | } | |
190 | ||
191 | int errorflag = 0; | |
192 | ||
193 | yyerror(s) char *s; { | |
194 | fprintf(stderr, "awk: %s near line %d\n", s, lineno); | |
195 | errorflag = 2; | |
196 | } | |
197 | ||
198 | error(f, s, a1, a2, a3, a4, a5, a6, a7) { | |
199 | fprintf(stderr, "awk: "); | |
200 | fprintf(stderr, s, a1, a2, a3, a4, a5, a6, a7); | |
201 | fprintf(stderr, "\n"); | |
202 | if (*NR > 0) | |
203 | fprintf(stderr, " record number %g\n", *NR); | |
204 | if (f) | |
205 | exit(2); | |
206 | } | |
207 | ||
208 | PUTS(s) char *s; { | |
209 | dprintf("%s\n", s, NULL, NULL); | |
210 | } | |
211 | ||
212 | /* | |
213 | isnumber(s) char *s; | |
214 | { | |
215 | for(;*s!=0; s++) | |
216 | { if(isdigit(*s) || *s=='-' || *s=='.' | |
217 | || *s=='+' || *s=='e' || *s=='E') | |
218 | continue; | |
219 | else return(0); | |
220 | } | |
221 | return(1); | |
222 | } | |
223 | */ | |
224 | isnumber(s) char *s; {return(0);} |