1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License, Version 1.0 only 6 * (the "License"). You may not use this file except in compliance 7 * with the License. 8 * 9 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10 * or http://www.opensolaris.org/os/licensing. 11 * See the License for the specific language governing permissions 12 * and limitations under the License. 13 * 14 * When distributing Covered Code, include this CDDL HEADER in each 15 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16 * If applicable, add the following below this CDDL HEADER, with the 17 * fields enclosed by brackets "[]" replaced with your own identifying 18 * information: Portions Copyright [yyyy] [name of copyright owner] 19 * 20 * CDDL HEADER END 21 */ 22 /* 23 * Copyright 2006 Sun Microsystems, Inc. All rights reserved. 24 * Use is subject to license terms. 25 */ 26 27 /* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */ 28 /* All Rights Reserved */ 29 30 #pragma ident "%Z%%M% %I% %E% SMI" 31 32 #include <stdio.h> 33 #include "awk.def" 34 #include "awk.h" 35 #include <ctype.h> 36 #include <wctype.h> 37 #include "awktype.h" 38 #include <stdlib.h> 39 40 FILE *infile = NULL; 41 wchar_t *file; 42 #define RECSIZE (5 * 512) 43 wchar_t record[RECSIZE]; 44 wchar_t fields[RECSIZE]; 45 wchar_t L_NULL[] = L""; 46 47 48 #define MAXFLD 100 49 int donefld; /* 1 = implies rec broken into fields */ 50 int donerec; /* 1 = record is valid (no flds have changed) */ 51 int mustfld; /* 1 = NF seen, so always break */ 52 static wchar_t L_record[] = L"$record"; 53 54 55 #define FINIT { OCELL, CFLD, 0, L_NULL, 0.0, FLD|STR } 56 CELL fldtab[MAXFLD] = { /* room for fields */ 57 { OCELL, CFLD, L_record, record, 0.0, STR|FLD}, 58 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 59 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 60 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 61 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 62 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 63 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 64 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 65 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 66 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, 67 FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT, FINIT 68 }; 69 int maxfld = 0; /* last used field */ 70 /* pointer to CELL for maximum field assigned to */ 71 CELL *maxmfld = &fldtab[0]; 72 73 static int isclvar(wchar_t *); 74 static void setclvar(wchar_t *); 75 void fldbld(void); 76 77 int 78 getrec(void) 79 { 80 wchar_t *rr, *er; 81 int c, sep; 82 FILE *inf; 83 extern int svargc; 84 extern wchar_t **svargv; 85 86 87 dprintf("**RS=%o, **FS=%o\n", **RS, **FS, NULL); 88 donefld = 0; 89 donerec = 1; 90 record[0] = 0; 91 er = record + RECSIZE; 92 while (svargc > 0) { 93 dprintf("svargc=%d, *svargv=%ws\n", svargc, *svargv, NULL); 94 if (infile == NULL) { /* have to open a new file */ 95 /* 96 * If the argument contains a '=', determine if the 97 * argument needs to be treated as a variable assignment 98 * or as the pathname of a file. 99 */ 100 if (isclvar(*svargv)) { 101 /* it's a var=value argument */ 102 setclvar(*svargv); 103 if (svargc > 1) { 104 svargv++; 105 svargc--; 106 continue; 107 } 108 *svargv = L"-"; 109 } 110 *FILENAME = file = *svargv; 111 dprintf("opening file %ws\n", file, NULL, NULL); 112 if (*file == (wchar_t)L'-') 113 infile = stdin; 114 else if ((infile = fopen(toeuccode(file), "r")) == NULL) 115 error(FATAL, "can't open %ws", file); 116 } 117 if ((sep = **RS) == 0) 118 sep = '\n'; 119 inf = infile; 120 for (rr = record; /* dummy */; /* dummy */) { 121 for (; (c = getwc(inf)) != sep && c != EOF && rr < er; 122 *rr++ = c) 123 ; 124 if (rr >= er) 125 error(FATAL, "record `%.20ws...' too long", 126 record); 127 if (**RS == sep || c == EOF) 128 break; 129 if ((c = getwc(inf)) == '\n' || c == EOF) 130 /* 2 in a row */ 131 break; 132 *rr++ = '\n'; 133 *rr++ = c; 134 } 135 if (rr >= er) 136 error(FATAL, "record `%.20ws...' too long", record); 137 *rr = 0; 138 if (mustfld) 139 fldbld(); 140 if (c != EOF || rr > record) { /* normal record */ 141 recloc->tval &= ~NUM; 142 recloc->tval |= STR; 143 ++nrloc->fval; 144 nrloc->tval &= ~STR; 145 nrloc->tval |= NUM; 146 return (1); 147 } 148 /* EOF arrived on this file; set up next */ 149 if (infile != stdin) 150 fclose(infile); 151 infile = NULL; 152 svargc--; 153 svargv++; 154 } 155 return (0); /* true end of file */ 156 } 157 158 /* 159 * isclvar() 160 * 161 * Returns 1 if the input string, arg, is a variable assignment, 162 * otherwise returns 0. 163 * 164 * An argument to awk can be either a pathname of a file, or a variable 165 * assignment. An operand that begins with an undersore or alphabetic 166 * character from the portable character set, followed by a sequence of 167 * underscores, digits, and alphabetics from the portable character set, 168 * followed by the '=' character, shall specify a variable assignment 169 * rather than a pathname. 170 */ 171 static int 172 isclvar(wchar_t *arg) 173 { 174 wchar_t *tmpptr = arg; 175 176 if (tmpptr != NULL) { 177 178 /* Begins with an underscore or alphabetic character */ 179 if (iswalpha(*tmpptr) || *tmpptr == '_') { 180 181 /* 182 * followed by a sequence of underscores, digits, 183 * and alphabetics 184 */ 185 for (tmpptr++; *tmpptr; tmpptr++) { 186 if (!(iswalnum(*tmpptr) || (*tmpptr == '_'))) { 187 break; 188 } 189 } 190 return (*tmpptr == '='); 191 } 192 } 193 194 return (0); 195 } 196 197 static void 198 setclvar(wchar_t *s) /* set var=value from s */ 199 { 200 wchar_t *p; 201 CELL *q; 202 203 204 for (p = s; *p != '='; p++) 205 ; 206 *p++ = 0; 207 q = setsymtab(s, tostring(p), 0.0, STR, symtab); 208 setsval(q, p); 209 dprintf("command line set %ws to |%ws|\n", s, p, NULL); 210 } 211 212 213 void 214 fldbld(void) 215 { 216 wchar_t *r, *fr, sep, c; 217 static wchar_t L_NF[] = L"NF"; 218 CELL *p, *q; 219 int i, j; 220 221 222 r = record; 223 fr = fields; 224 i = 0; /* number of fields accumulated here */ 225 if ((sep = **FS) == ' ') 226 for (i = 0; /* dummy */; /* dummy */) { 227 c = *r; 228 while (iswblank(c) || c == '\t' || c == '\n') 229 c = *(++r); 230 if (*r == 0) 231 break; 232 i++; 233 if (i >= MAXFLD) 234 error(FATAL, 235 "record `%.20ws...' has too many fields", record); 236 if (!(fldtab[i].tval&FLD)) 237 xfree(fldtab[i].sval); 238 fldtab[i].sval = fr; 239 fldtab[i].tval = FLD | STR; 240 do { 241 *fr++ = *r++; 242 c = *r; 243 } while (! iswblank(c) && c != '\t' && 244 c != '\n' && c != '\0'); 245 246 247 *fr++ = 0; 248 249 } else if (*r != 0) /* if 0, it's a null field */ 250 for (;;) { 251 i++; 252 if (i >= MAXFLD) 253 error(FATAL, 254 "record `%.20ws...' has too many fields", record); 255 if (!(fldtab[i].tval&FLD)) 256 xfree(fldtab[i].sval); 257 fldtab[i].sval = fr; 258 fldtab[i].tval = FLD | STR; 259 while ((c = *r) != sep && c != '\n' && c != '\0') 260 /* \n always a separator */ 261 *fr++ = *r++; 262 *fr++ = 0; 263 if (*r++ == 0) 264 break; 265 } 266 *fr = 0; 267 /* clean out junk from previous record */ 268 for (p = maxmfld, q = &fldtab[i]; p > q; p--) { 269 if (!(p->tval&FLD)) 270 xfree(p->sval); 271 p->tval = STR | FLD; 272 p->sval = L_NULL; 273 } 274 maxfld = i; 275 maxmfld = &fldtab[i]; 276 donefld = 1; 277 for (i = 1; i <= maxfld; i++) 278 if (isanumber(fldtab[i].sval)) { 279 fldtab[i].fval = watof(fldtab[i].sval); 280 fldtab[i].tval |= NUM; 281 } 282 setfval(lookup(L_NF, symtab, 0), (awkfloat) maxfld); 283 if (dbg) 284 for (i = 0; i <= maxfld; i++) 285 printf("field %d: |%ws|\n", i, fldtab[i].sval); 286 } 287 288 289 void 290 recbld(void) 291 { 292 int i; 293 wchar_t *r, *p; 294 295 296 if (donefld == 0 || donerec == 1) 297 return; 298 r = record; 299 for (i = 1; i <= *NF; i++) { 300 p = getsval(&fldtab[i]); 301 while (*r++ = *p++) 302 ; 303 *(r-1) = **OFS; 304 } 305 *(r-1) = '\0'; 306 dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL); 307 recloc->tval = STR | FLD; 308 dprintf("in recbld FS=%o, recloc=%o\n", **FS, recloc, NULL); 309 if (r > record+RECSIZE) 310 error(FATAL, "built giant record `%.20ws...'", record); 311 dprintf("recbld = |%ws|\n", record, NULL, NULL); 312 } 313 314 315 CELL * 316 fieldadr(n) 317 { 318 if (n < 0 || n >= MAXFLD) 319 error(FATAL, "trying to access field %d", n); 320 return (&fldtab[n]); 321 } 322 323 324 int errorflag = 0; 325 326 327 int 328 yyerror(char *s) 329 { 330 fprintf(stderr, 331 gettext("awk: %s near line %lld\n"), gettext(s), lineno); 332 errorflag = 2; 333 return (0); 334 } 335 336 337 void 338 error(f, s, a1, a2, a3, a4, a5, a6, a7) 339 { 340 fprintf(stderr, "awk: "); 341 fprintf(stderr, gettext((char *)s), a1, a2, a3, a4, a5, a6, a7); 342 fprintf(stderr, "\n"); 343 if (NR && *NR > 0) 344 fprintf(stderr, gettext(" record number %g\n"), *NR); 345 if (f) 346 exit(2); 347 } 348 349 350 void 351 PUTS(char *s) 352 { 353 dprintf("%s\n", s, NULL, NULL); 354 } 355 356 357 #define MAXEXPON 38 /* maximum exponenet for fp number */ 358 359 360 int 361 isanumber(wchar_t *s) 362 { 363 int d1, d2; 364 int point; 365 wchar_t *es; 366 extern wchar_t radixpoint; 367 368 d1 = d2 = point = 0; 369 while (*s == ' ' || *s == '\t' || *s == '\n') 370 s++; 371 if (*s == '\0') 372 return (0); /* empty stuff isn't number */ 373 if (*s == '+' || *s == '-') 374 s++; 375 /* 376 * Since, iswdigit() will include digit from other than code set 0, 377 * we have to check it from code set 0 or not. 378 */ 379 if (!(iswdigit(*s) && iswascii(*s)) && *s != radixpoint) 380 return (0); 381 if (iswdigit(*s) && iswascii(*s)) { 382 do { 383 d1++; 384 s++; 385 } while (iswdigit(*s) && iswascii(*s)); 386 } 387 if (d1 >= MAXEXPON) 388 return (0); /* too many digits to convert */ 389 if (*s == radixpoint) { 390 point++; 391 s++; 392 } 393 if (iswdigit(*s) && iswascii(*s)) { 394 d2++; 395 do { 396 s++; 397 } while (iswdigit(*s) && iswascii(*s)); 398 } 399 400 401 if (!(d1 || point && d2)) 402 return (0); 403 if (*s == 'e' || *s == 'E') { 404 s++; 405 if (*s == '+' || *s == '-') 406 s++; 407 if (!(iswdigit(*s) && iswascii(*s))) 408 return (0); 409 es = s; 410 do { 411 s++; 412 } while (iswdigit(*s) && iswascii(*s)); 413 414 415 if (s - es > 2) 416 return (0); 417 else if (s - es == 2 && 418 10 * (*es-'0') + *(es+1)-'0' >= MAXEXPON) 419 return (0); 420 } 421 while (*s == ' ' || *s == '\t' || *s == '\n') 422 s++; 423 if (*s == '\0') 424 return (1); 425 else 426 return (0); 427 } 428 char * 429 toeuccode(str) 430 wchar_t *str; 431 { 432 static char euccode[RECSIZE]; 433 434 (void) wcstombs(euccode, str, RECSIZE); 435 return (euccode); 436 } 437