Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 1 | /**************************************************************** |
| 2 | Copyright (C) Lucent Technologies 1997 |
| 3 | All Rights Reserved |
| 4 | |
| 5 | Permission to use, copy, modify, and distribute this software and |
| 6 | its documentation for any purpose and without fee is hereby |
| 7 | granted, provided that the above copyright notice appear in all |
| 8 | copies and that both that the copyright notice and this |
| 9 | permission notice and warranty disclaimer appear in supporting |
| 10 | documentation, and that the name Lucent Technologies or any of |
| 11 | its entities not be used in advertising or publicity pertaining |
| 12 | to distribution of the software without specific, written prior |
| 13 | permission. |
| 14 | |
| 15 | LUCENT DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, |
| 16 | INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. |
| 17 | IN NO EVENT SHALL LUCENT OR ANY OF ITS ENTITIES BE LIABLE FOR ANY |
| 18 | SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| 19 | WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER |
| 20 | IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, |
| 21 | ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF |
| 22 | THIS SOFTWARE. |
| 23 | ****************************************************************/ |
| 24 | |
| 25 | #define DEBUG |
| 26 | #include <stdio.h> |
| 27 | #include <string.h> |
Michael Forney | 38e525f | 2020-12-15 04:46:30 -0800 | [diff] [blame] | 28 | #include <strings.h> |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 29 | #include <ctype.h> |
| 30 | #include <errno.h> |
| 31 | #include <stdlib.h> |
| 32 | #include <stdarg.h> |
Michael Forney | 6932571 | 2020-01-30 22:23:34 -0800 | [diff] [blame] | 33 | #include <limits.h> |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 34 | #include <math.h> |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 35 | #include "awk.h" |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 36 | |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 37 | char EMPTY[] = { '\0' }; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 38 | FILE *infile = NULL; |
zoulasc | ffee778 | 2020-02-28 06:23:54 -0500 | [diff] [blame] | 39 | bool innew; /* true = infile has not been read by readrec */ |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 40 | char *file = EMPTY; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 41 | char *record; |
| 42 | int recsize = RECSIZE; |
| 43 | char *fields; |
| 44 | int fieldssize = RECSIZE; |
| 45 | |
| 46 | Cell **fldtab; /* pointers to Cells */ |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 47 | static size_t len_inputFS = 0; |
| 48 | static char *inputFS = NULL; /* FS at time of input, for field splitting */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 49 | |
| 50 | #define MAXFLD 2 |
| 51 | int nfields = MAXFLD; /* last allocated slot for $i */ |
| 52 | |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 53 | bool donefld; /* true = implies rec broken into fields */ |
| 54 | bool donerec; /* true = record is valid (no flds have changed) */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 55 | |
| 56 | int lastfld = 0; /* last used field */ |
| 57 | int argno = 1; /* current input argument number */ |
| 58 | extern Awkfloat *ARGC; |
| 59 | |
zoulasc | c2c8ecb | 2020-02-19 13:44:49 -0500 | [diff] [blame] | 60 | static Cell dollar0 = { OCELL, CFLD, NULL, EMPTY, 0.0, REC|STR|DONTFREE, NULL, NULL }; |
| 61 | static Cell dollar1 = { OCELL, CFLD, NULL, EMPTY, 0.0, FLD|STR|DONTFREE, NULL, NULL }; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 62 | |
| 63 | void recinit(unsigned int n) |
| 64 | { |
Arnold D. Robbins | 3b42cfa | 2020-10-13 20:52:43 +0300 | [diff] [blame] | 65 | if ( (record = (char *) malloc(n)) == NULL |
| 66 | || (fields = (char *) malloc(n+1)) == NULL |
| 67 | || (fldtab = (Cell **) calloc(nfields+2, sizeof(*fldtab))) == NULL |
| 68 | || (fldtab[0] = (Cell *) malloc(sizeof(**fldtab))) == NULL) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 69 | FATAL("out of space for $0 and fields"); |
M. Warner Losh | 0939e33 | 2019-06-02 15:17:29 -0600 | [diff] [blame] | 70 | *record = '\0'; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 71 | *fldtab[0] = dollar0; |
| 72 | fldtab[0]->sval = record; |
| 73 | fldtab[0]->nval = tostring("0"); |
| 74 | makefields(1, nfields); |
| 75 | } |
| 76 | |
| 77 | void makefields(int n1, int n2) /* create $n1..$n2 inclusive */ |
| 78 | { |
| 79 | char temp[50]; |
| 80 | int i; |
| 81 | |
| 82 | for (i = n1; i <= n2; i++) { |
Arnold D. Robbins | 3b42cfa | 2020-10-13 20:52:43 +0300 | [diff] [blame] | 83 | fldtab[i] = (Cell *) malloc(sizeof(**fldtab)); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 84 | if (fldtab[i] == NULL) |
| 85 | FATAL("out of space in makefields %d", i); |
| 86 | *fldtab[i] = dollar1; |
zoulasc | 0d8778b | 2019-10-25 10:59:10 -0400 | [diff] [blame] | 87 | snprintf(temp, sizeof(temp), "%d", i); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 88 | fldtab[i]->nval = tostring(temp); |
| 89 | } |
| 90 | } |
| 91 | |
| 92 | void initgetrec(void) |
| 93 | { |
| 94 | int i; |
| 95 | char *p; |
| 96 | |
| 97 | for (i = 1; i < *ARGC; i++) { |
| 98 | p = getargv(i); /* find 1st real filename */ |
| 99 | if (p == NULL || *p == '\0') { /* deleted or zapped */ |
| 100 | argno++; |
| 101 | continue; |
| 102 | } |
| 103 | if (!isclvar(p)) { |
| 104 | setsval(lookup("FILENAME", symtab), p); |
| 105 | return; |
| 106 | } |
| 107 | setclvar(p); /* a commandline assignment before filename */ |
| 108 | argno++; |
| 109 | } |
| 110 | infile = stdin; /* no filenames, so use stdin */ |
zoulasc | ffee778 | 2020-02-28 06:23:54 -0500 | [diff] [blame] | 111 | innew = true; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 112 | } |
| 113 | |
Cody Peter Mello | b463680 | 2018-10-19 15:07:53 -0700 | [diff] [blame] | 114 | /* |
| 115 | * POSIX specifies that fields are supposed to be evaluated as if they were |
| 116 | * split using the value of FS at the time that the record's value ($0) was |
| 117 | * read. |
| 118 | * |
| 119 | * Since field-splitting is done lazily, we save the current value of FS |
| 120 | * whenever a new record is read in (implicitly or via getline), or when |
| 121 | * a new value is assigned to $0. |
| 122 | */ |
| 123 | void savefs(void) |
| 124 | { |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 125 | size_t len; |
| 126 | if ((len = strlen(getsval(fsloc))) < len_inputFS) { |
| 127 | strcpy(inputFS, *FS); /* for subsequent field splitting */ |
| 128 | return; |
| 129 | } |
| 130 | |
| 131 | len_inputFS = len + 1; |
Arnold D. Robbins | 3b42cfa | 2020-10-13 20:52:43 +0300 | [diff] [blame] | 132 | inputFS = (char *) realloc(inputFS, len_inputFS); |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 133 | if (inputFS == NULL) |
Cody Peter Mello | b463680 | 2018-10-19 15:07:53 -0700 | [diff] [blame] | 134 | FATAL("field separator %.10s... is too long", *FS); |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 135 | memcpy(inputFS, *FS, len_inputFS); |
Cody Peter Mello | b463680 | 2018-10-19 15:07:53 -0700 | [diff] [blame] | 136 | } |
| 137 | |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 138 | static bool firsttime = true; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 139 | |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 140 | int getrec(char **pbuf, int *pbufsize, bool isrecord) /* get next input record */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 141 | { /* note: cares whether buf == record */ |
| 142 | int c; |
| 143 | char *buf = *pbuf; |
| 144 | uschar saveb0; |
| 145 | int bufsize = *pbufsize, savebufsize = bufsize; |
| 146 | |
| 147 | if (firsttime) { |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 148 | firsttime = false; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 149 | initgetrec(); |
| 150 | } |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 151 | DPRINTF("RS=<%s>, FS=<%s>, ARGC=%g, FILENAME=%s\n", |
| 152 | *RS, *FS, *ARGC, *FILENAME); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 153 | if (isrecord) { |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 154 | donefld = false; |
| 155 | donerec = true; |
Cody Peter Mello | b463680 | 2018-10-19 15:07:53 -0700 | [diff] [blame] | 156 | savefs(); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 157 | } |
| 158 | saveb0 = buf[0]; |
| 159 | buf[0] = 0; |
| 160 | while (argno < *ARGC || infile == stdin) { |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 161 | DPRINTF("argno=%d, file=|%s|\n", argno, file); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 162 | if (infile == NULL) { /* have to open a new file */ |
| 163 | file = getargv(argno); |
| 164 | if (file == NULL || *file == '\0') { /* deleted or zapped */ |
| 165 | argno++; |
| 166 | continue; |
| 167 | } |
| 168 | if (isclvar(file)) { /* a var=value arg */ |
| 169 | setclvar(file); |
| 170 | argno++; |
| 171 | continue; |
| 172 | } |
| 173 | *FILENAME = file; |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 174 | DPRINTF("opening file %s\n", file); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 175 | if (*file == '-' && *(file+1) == '\0') |
| 176 | infile = stdin; |
| 177 | else if ((infile = fopen(file, "r")) == NULL) |
| 178 | FATAL("can't open file %s", file); |
Miguel Pineiro Jr | feaf62d | 2021-04-16 20:31:36 -0400 | [diff] [blame] | 179 | innew = true; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 180 | setfval(fnrloc, 0.0); |
| 181 | } |
zoulasc | ffee778 | 2020-02-28 06:23:54 -0500 | [diff] [blame] | 182 | c = readrec(&buf, &bufsize, infile, innew); |
| 183 | if (innew) |
| 184 | innew = false; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 185 | if (c != 0 || buf[0] != '\0') { /* normal record */ |
| 186 | if (isrecord) { |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 187 | double result; |
| 188 | |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 189 | if (freeable(fldtab[0])) |
| 190 | xfree(fldtab[0]->sval); |
| 191 | fldtab[0]->sval = buf; /* buf == record */ |
| 192 | fldtab[0]->tval = REC | STR | DONTFREE; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 193 | if (is_number(fldtab[0]->sval, & result)) { |
| 194 | fldtab[0]->fval = result; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 195 | fldtab[0]->tval |= NUM; |
| 196 | } |
| 197 | } |
| 198 | setfval(nrloc, nrloc->fval+1); |
| 199 | setfval(fnrloc, fnrloc->fval+1); |
| 200 | *pbuf = buf; |
| 201 | *pbufsize = bufsize; |
| 202 | return 1; |
| 203 | } |
| 204 | /* EOF arrived on this file; set up next */ |
| 205 | if (infile != stdin) |
| 206 | fclose(infile); |
| 207 | infile = NULL; |
| 208 | argno++; |
| 209 | } |
| 210 | buf[0] = saveb0; |
| 211 | *pbuf = buf; |
| 212 | *pbufsize = savebufsize; |
| 213 | return 0; /* true end of file */ |
| 214 | } |
| 215 | |
| 216 | void nextfile(void) |
| 217 | { |
| 218 | if (infile != NULL && infile != stdin) |
| 219 | fclose(infile); |
| 220 | infile = NULL; |
| 221 | argno++; |
| 222 | } |
| 223 | |
zoulasc | ffee778 | 2020-02-28 06:23:54 -0500 | [diff] [blame] | 224 | int readrec(char **pbuf, int *pbufsize, FILE *inf, bool newflag) /* read one record into buf */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 225 | { |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 226 | int sep, c, isrec; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 227 | char *rr, *buf = *pbuf; |
| 228 | int bufsize = *pbufsize; |
Cody Peter Mello | 52566c0 | 2018-09-18 15:45:55 -0700 | [diff] [blame] | 229 | char *rs = getsval(rsloc); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 230 | |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 231 | if (*rs && rs[1]) { |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 232 | bool found; |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 233 | |
| 234 | fa *pfa = makedfa(rs, 1); |
zoulasc | ffee778 | 2020-02-28 06:23:54 -0500 | [diff] [blame] | 235 | if (newflag) |
| 236 | found = fnematch(pfa, inf, &buf, &bufsize, recsize); |
| 237 | else { |
| 238 | int tempstat = pfa->initstat; |
| 239 | pfa->initstat = 2; |
| 240 | found = fnematch(pfa, inf, &buf, &bufsize, recsize); |
| 241 | pfa->initstat = tempstat; |
| 242 | } |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 243 | if (found) |
zoulasc | 6589208 | 2019-10-24 09:40:15 -0400 | [diff] [blame] | 244 | setptr(patbeg, '\0'); |
ozan yigit | aa8731e | 2021-07-25 14:37:03 -0400 | [diff] [blame] | 245 | isrec = (found == 0 && *buf == '\0') ? false : true; |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 246 | } else { |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 247 | if ((sep = *rs) == 0) { |
| 248 | sep = '\n'; |
| 249 | while ((c=getc(inf)) == '\n' && c != EOF) /* skip leading \n's */ |
| 250 | ; |
| 251 | if (c != EOF) |
| 252 | ungetc(c, inf); |
| 253 | } |
| 254 | for (rr = buf; ; ) { |
| 255 | for (; (c=getc(inf)) != sep && c != EOF; ) { |
| 256 | if (rr-buf+1 > bufsize) |
| 257 | if (!adjbuf(&buf, &bufsize, 1+rr-buf, |
| 258 | recsize, &rr, "readrec 1")) |
| 259 | FATAL("input record `%.30s...' too long", buf); |
| 260 | *rr++ = c; |
| 261 | } |
| 262 | if (*rs == sep || c == EOF) |
| 263 | break; |
| 264 | if ((c = getc(inf)) == '\n' || c == EOF) /* 2 in a row */ |
| 265 | break; |
| 266 | if (!adjbuf(&buf, &bufsize, 2+rr-buf, recsize, &rr, |
| 267 | "readrec 2")) |
| 268 | FATAL("input record `%.30s...' too long", buf); |
| 269 | *rr++ = '\n'; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 270 | *rr++ = c; |
| 271 | } |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 272 | if (!adjbuf(&buf, &bufsize, 1+rr-buf, recsize, &rr, "readrec 3")) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 273 | FATAL("input record `%.30s...' too long", buf); |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 274 | *rr = 0; |
ozan yigit | aa8731e | 2021-07-25 14:37:03 -0400 | [diff] [blame] | 275 | isrec = (c == EOF && rr == buf) ? false : true; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 276 | } |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 277 | *pbuf = buf; |
| 278 | *pbufsize = bufsize; |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 279 | DPRINTF("readrec saw <%s>, returns %d\n", buf, isrec); |
Arnold D. Robbins | 643a5a3 | 2019-07-28 20:12:05 +0300 | [diff] [blame] | 280 | return isrec; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 281 | } |
| 282 | |
| 283 | char *getargv(int n) /* get ARGV[n] */ |
| 284 | { |
| 285 | Cell *x; |
| 286 | char *s, temp[50]; |
| 287 | extern Array *ARGVtab; |
| 288 | |
zoulasc | 0d8778b | 2019-10-25 10:59:10 -0400 | [diff] [blame] | 289 | snprintf(temp, sizeof(temp), "%d", n); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 290 | if (lookup(temp, ARGVtab) == NULL) |
| 291 | return NULL; |
| 292 | x = setsymtab(temp, "", 0.0, STR, ARGVtab); |
| 293 | s = getsval(x); |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 294 | DPRINTF("getargv(%d) returns |%s|\n", n, s); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 295 | return s; |
| 296 | } |
| 297 | |
| 298 | void setclvar(char *s) /* set var=value from s */ |
| 299 | { |
| 300 | char *p; |
| 301 | Cell *q; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 302 | double result; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 303 | |
| 304 | for (p=s; *p != '='; p++) |
| 305 | ; |
| 306 | *p++ = 0; |
| 307 | p = qstring(p, '\0'); |
| 308 | q = setsymtab(s, p, 0.0, STR, symtab); |
| 309 | setsval(q, p); |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 310 | if (is_number(q->sval, & result)) { |
| 311 | q->fval = result; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 312 | q->tval |= NUM; |
| 313 | } |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 314 | DPRINTF("command line set %s to |%s|\n", s, p); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 315 | } |
| 316 | |
| 317 | |
| 318 | void fldbld(void) /* create fields from current record */ |
| 319 | { |
| 320 | /* this relies on having fields[] the same length as $0 */ |
| 321 | /* the fields are all stored in this one array with \0's */ |
| 322 | /* possibly with a final trailing \0 not associated with any field */ |
| 323 | char *r, *fr, sep; |
| 324 | Cell *p; |
| 325 | int i, j, n; |
| 326 | |
| 327 | if (donefld) |
| 328 | return; |
| 329 | if (!isstr(fldtab[0])) |
| 330 | getsval(fldtab[0]); |
| 331 | r = fldtab[0]->sval; |
| 332 | n = strlen(r); |
| 333 | if (n > fieldssize) { |
| 334 | xfree(fields); |
Arnold D. Robbins | 3b42cfa | 2020-10-13 20:52:43 +0300 | [diff] [blame] | 335 | if ((fields = (char *) malloc(n+2)) == NULL) /* possibly 2 final \0s */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 336 | FATAL("out of space for fields in fldbld %d", n); |
| 337 | fieldssize = n; |
| 338 | } |
| 339 | fr = fields; |
| 340 | i = 0; /* number of fields accumulated here */ |
Arnold D. Robbins | 754cf93 | 2020-06-05 12:25:15 +0300 | [diff] [blame] | 341 | if (inputFS == NULL) /* make sure we have a copy of FS */ |
| 342 | savefs(); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 343 | if (strlen(inputFS) > 1) { /* it's a regular expression */ |
| 344 | i = refldbld(r, inputFS); |
| 345 | } else if ((sep = *inputFS) == ' ') { /* default whitespace */ |
| 346 | for (i = 0; ; ) { |
| 347 | while (*r == ' ' || *r == '\t' || *r == '\n') |
| 348 | r++; |
| 349 | if (*r == 0) |
| 350 | break; |
| 351 | i++; |
| 352 | if (i > nfields) |
| 353 | growfldtab(i); |
| 354 | if (freeable(fldtab[i])) |
| 355 | xfree(fldtab[i]->sval); |
| 356 | fldtab[i]->sval = fr; |
| 357 | fldtab[i]->tval = FLD | STR | DONTFREE; |
| 358 | do |
| 359 | *fr++ = *r++; |
| 360 | while (*r != ' ' && *r != '\t' && *r != '\n' && *r != '\0'); |
| 361 | *fr++ = 0; |
| 362 | } |
| 363 | *fr = 0; |
| 364 | } else if ((sep = *inputFS) == 0) { /* new: FS="" => 1 char/field */ |
Arnold D. Robbins | c879fbf | 2019-11-08 14:40:18 +0200 | [diff] [blame] | 365 | for (i = 0; *r != '\0'; r += n) { |
Michael Forney | 6932571 | 2020-01-30 22:23:34 -0800 | [diff] [blame] | 366 | char buf[MB_LEN_MAX + 1]; |
Arnold D. Robbins | c879fbf | 2019-11-08 14:40:18 +0200 | [diff] [blame] | 367 | |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 368 | i++; |
| 369 | if (i > nfields) |
| 370 | growfldtab(i); |
| 371 | if (freeable(fldtab[i])) |
| 372 | xfree(fldtab[i]->sval); |
Michael Forney | 6932571 | 2020-01-30 22:23:34 -0800 | [diff] [blame] | 373 | n = mblen(r, MB_LEN_MAX); |
Arnold D. Robbins | c879fbf | 2019-11-08 14:40:18 +0200 | [diff] [blame] | 374 | if (n < 0) |
| 375 | n = 1; |
| 376 | memcpy(buf, r, n); |
| 377 | buf[n] = '\0'; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 378 | fldtab[i]->sval = tostring(buf); |
| 379 | fldtab[i]->tval = FLD | STR; |
| 380 | } |
| 381 | *fr = 0; |
| 382 | } else if (*r != 0) { /* if 0, it's a null field */ |
| 383 | /* subtlecase : if length(FS) == 1 && length(RS > 0) |
| 384 | * \n is NOT a field separator (cf awk book 61,84). |
| 385 | * this variable is tested in the inner while loop. |
| 386 | */ |
| 387 | int rtest = '\n'; /* normal case */ |
| 388 | if (strlen(*RS) > 0) |
| 389 | rtest = '\0'; |
| 390 | for (;;) { |
| 391 | i++; |
| 392 | if (i > nfields) |
| 393 | growfldtab(i); |
| 394 | if (freeable(fldtab[i])) |
| 395 | xfree(fldtab[i]->sval); |
| 396 | fldtab[i]->sval = fr; |
| 397 | fldtab[i]->tval = FLD | STR | DONTFREE; |
| 398 | while (*r != sep && *r != rtest && *r != '\0') /* \n is always a separator */ |
| 399 | *fr++ = *r++; |
| 400 | *fr++ = 0; |
| 401 | if (*r++ == 0) |
| 402 | break; |
| 403 | } |
| 404 | *fr = 0; |
| 405 | } |
| 406 | if (i > nfields) |
| 407 | FATAL("record `%.30s...' has too many fields; can't happen", r); |
| 408 | cleanfld(i+1, lastfld); /* clean out junk from previous record */ |
| 409 | lastfld = i; |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 410 | donefld = true; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 411 | for (j = 1; j <= lastfld; j++) { |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 412 | double result; |
| 413 | |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 414 | p = fldtab[j]; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 415 | if(is_number(p->sval, & result)) { |
| 416 | p->fval = result; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 417 | p->tval |= NUM; |
| 418 | } |
| 419 | } |
| 420 | setfval(nfloc, (Awkfloat) lastfld); |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 421 | donerec = true; /* restore */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 422 | if (dbg) { |
| 423 | for (j = 0; j <= lastfld; j++) { |
| 424 | p = fldtab[j]; |
| 425 | printf("field %d (%s): |%s|\n", j, p->nval, p->sval); |
| 426 | } |
| 427 | } |
| 428 | } |
| 429 | |
| 430 | void cleanfld(int n1, int n2) /* clean out fields n1 .. n2 inclusive */ |
| 431 | { /* nvals remain intact */ |
| 432 | Cell *p; |
| 433 | int i; |
| 434 | |
| 435 | for (i = n1; i <= n2; i++) { |
| 436 | p = fldtab[i]; |
| 437 | if (freeable(p)) |
| 438 | xfree(p->sval); |
zoulasc | 6a87709 | 2020-01-24 04:11:59 -0500 | [diff] [blame] | 439 | p->sval = EMPTY, |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 440 | p->tval = FLD | STR | DONTFREE; |
| 441 | } |
| 442 | } |
| 443 | |
| 444 | void newfld(int n) /* add field n after end of existing lastfld */ |
| 445 | { |
| 446 | if (n > nfields) |
| 447 | growfldtab(n); |
| 448 | cleanfld(lastfld+1, n); |
| 449 | lastfld = n; |
| 450 | setfval(nfloc, (Awkfloat) n); |
| 451 | } |
| 452 | |
Arnold D. Robbins | 32093f5 | 2018-08-22 20:40:26 +0300 | [diff] [blame] | 453 | void setlastfld(int n) /* set lastfld cleaning fldtab cells if necessary */ |
| 454 | { |
Cody Peter Mello | 179536a | 2018-09-25 21:19:49 -0700 | [diff] [blame] | 455 | if (n < 0) |
| 456 | FATAL("cannot set NF to a negative value"); |
Arnold D. Robbins | 32093f5 | 2018-08-22 20:40:26 +0300 | [diff] [blame] | 457 | if (n > nfields) |
| 458 | growfldtab(n); |
| 459 | |
| 460 | if (lastfld < n) |
| 461 | cleanfld(lastfld+1, n); |
| 462 | else |
| 463 | cleanfld(n+1, lastfld); |
| 464 | |
| 465 | lastfld = n; |
| 466 | } |
| 467 | |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 468 | Cell *fieldadr(int n) /* get nth field */ |
| 469 | { |
| 470 | if (n < 0) |
| 471 | FATAL("trying to access out of range field %d", n); |
| 472 | if (n > nfields) /* fields after NF are empty */ |
| 473 | growfldtab(n); /* but does not increase NF */ |
| 474 | return(fldtab[n]); |
| 475 | } |
| 476 | |
| 477 | void growfldtab(int n) /* make new fields up to at least $n */ |
| 478 | { |
| 479 | int nf = 2 * nfields; |
| 480 | size_t s; |
| 481 | |
| 482 | if (n > nf) |
| 483 | nf = n; |
| 484 | s = (nf+1) * (sizeof (struct Cell *)); /* freebsd: how much do we need? */ |
zoulasc | c2c8ecb | 2020-02-19 13:44:49 -0500 | [diff] [blame] | 485 | if (s / sizeof(struct Cell *) - 1 == (size_t)nf) /* didn't overflow */ |
Arnold D. Robbins | 3b42cfa | 2020-10-13 20:52:43 +0300 | [diff] [blame] | 486 | fldtab = (Cell **) realloc(fldtab, s); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 487 | else /* overflow sizeof int */ |
| 488 | xfree(fldtab); /* make it null */ |
| 489 | if (fldtab == NULL) |
| 490 | FATAL("out of space creating %d fields", nf); |
| 491 | makefields(nfields+1, nf); |
| 492 | nfields = nf; |
| 493 | } |
| 494 | |
| 495 | int refldbld(const char *rec, const char *fs) /* build fields from reg expr in FS */ |
| 496 | { |
| 497 | /* this relies on having fields[] the same length as $0 */ |
| 498 | /* the fields are all stored in this one array with \0's */ |
| 499 | char *fr; |
| 500 | int i, tempstat, n; |
| 501 | fa *pfa; |
| 502 | |
| 503 | n = strlen(rec); |
| 504 | if (n > fieldssize) { |
| 505 | xfree(fields); |
Arnold D. Robbins | 3b42cfa | 2020-10-13 20:52:43 +0300 | [diff] [blame] | 506 | if ((fields = (char *) malloc(n+1)) == NULL) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 507 | FATAL("out of space for fields in refldbld %d", n); |
| 508 | fieldssize = n; |
| 509 | } |
| 510 | fr = fields; |
| 511 | *fr = '\0'; |
| 512 | if (*rec == '\0') |
| 513 | return 0; |
| 514 | pfa = makedfa(fs, 1); |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 515 | DPRINTF("into refldbld, rec = <%s>, pat = <%s>\n", rec, fs); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 516 | tempstat = pfa->initstat; |
| 517 | for (i = 1; ; i++) { |
| 518 | if (i > nfields) |
| 519 | growfldtab(i); |
| 520 | if (freeable(fldtab[i])) |
| 521 | xfree(fldtab[i]->sval); |
| 522 | fldtab[i]->tval = FLD | STR | DONTFREE; |
| 523 | fldtab[i]->sval = fr; |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 524 | DPRINTF("refldbld: i=%d\n", i); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 525 | if (nematch(pfa, rec)) { |
| 526 | pfa->initstat = 2; /* horrible coupling to b.c */ |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 527 | DPRINTF("match %s (%d chars)\n", patbeg, patlen); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 528 | strncpy(fr, rec, patbeg-rec); |
| 529 | fr += patbeg - rec + 1; |
| 530 | *(fr-1) = '\0'; |
| 531 | rec = patbeg + patlen; |
| 532 | } else { |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 533 | DPRINTF("no match %s\n", rec); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 534 | strcpy(fr, rec); |
| 535 | pfa->initstat = tempstat; |
| 536 | break; |
| 537 | } |
| 538 | } |
Arnold D. Robbins | 795a06b | 2019-07-28 05:51:52 -0600 | [diff] [blame] | 539 | return i; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 540 | } |
| 541 | |
| 542 | void recbld(void) /* create $0 from $1..$NF if necessary */ |
| 543 | { |
| 544 | int i; |
| 545 | char *r, *p; |
Cody Peter Mello | 52566c0 | 2018-09-18 15:45:55 -0700 | [diff] [blame] | 546 | char *sep = getsval(ofsloc); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 547 | |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 548 | if (donerec) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 549 | return; |
| 550 | r = record; |
| 551 | for (i = 1; i <= *NF; i++) { |
| 552 | p = getsval(fldtab[i]); |
| 553 | if (!adjbuf(&record, &recsize, 1+strlen(p)+r-record, recsize, &r, "recbld 1")) |
| 554 | FATAL("created $0 `%.30s...' too long", record); |
| 555 | while ((*r = *p++) != 0) |
| 556 | r++; |
| 557 | if (i < *NF) { |
Cody Peter Mello | 52566c0 | 2018-09-18 15:45:55 -0700 | [diff] [blame] | 558 | if (!adjbuf(&record, &recsize, 2+strlen(sep)+r-record, recsize, &r, "recbld 2")) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 559 | FATAL("created $0 `%.30s...' too long", record); |
Cody Peter Mello | 52566c0 | 2018-09-18 15:45:55 -0700 | [diff] [blame] | 560 | for (p = sep; (*r = *p++) != 0; ) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 561 | r++; |
| 562 | } |
| 563 | } |
| 564 | if (!adjbuf(&record, &recsize, 2+r-record, recsize, &r, "recbld 3")) |
| 565 | FATAL("built giant record `%.30s...'", record); |
| 566 | *r = '\0'; |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 567 | DPRINTF("in recbld inputFS=%s, fldtab[0]=%p\n", inputFS, (void*)fldtab[0]); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 568 | |
| 569 | if (freeable(fldtab[0])) |
| 570 | xfree(fldtab[0]->sval); |
| 571 | fldtab[0]->tval = REC | STR | DONTFREE; |
| 572 | fldtab[0]->sval = record; |
| 573 | |
Todd C. Miller | 292d39f | 2020-06-25 12:32:34 -0600 | [diff] [blame] | 574 | DPRINTF("in recbld inputFS=%s, fldtab[0]=%p\n", inputFS, (void*)fldtab[0]); |
| 575 | DPRINTF("recbld = |%s|\n", record); |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 576 | donerec = true; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 577 | } |
| 578 | |
| 579 | int errorflag = 0; |
| 580 | |
| 581 | void yyerror(const char *s) |
| 582 | { |
| 583 | SYNTAX("%s", s); |
| 584 | } |
| 585 | |
| 586 | void SYNTAX(const char *fmt, ...) |
| 587 | { |
| 588 | extern char *cmdname, *curfname; |
| 589 | static int been_here = 0; |
| 590 | va_list varg; |
| 591 | |
| 592 | if (been_here++ > 2) |
| 593 | return; |
| 594 | fprintf(stderr, "%s: ", cmdname); |
| 595 | va_start(varg, fmt); |
| 596 | vfprintf(stderr, fmt, varg); |
| 597 | va_end(varg); |
| 598 | fprintf(stderr, " at source line %d", lineno); |
| 599 | if (curfname != NULL) |
| 600 | fprintf(stderr, " in function %s", curfname); |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 601 | if (compile_time == COMPILING && cursource() != NULL) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 602 | fprintf(stderr, " source file %s", cursource()); |
| 603 | fprintf(stderr, "\n"); |
| 604 | errorflag = 2; |
| 605 | eprint(); |
| 606 | } |
| 607 | |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 608 | extern int bracecnt, brackcnt, parencnt; |
| 609 | |
| 610 | void bracecheck(void) |
| 611 | { |
| 612 | int c; |
| 613 | static int beenhere = 0; |
| 614 | |
| 615 | if (beenhere++) |
| 616 | return; |
| 617 | while ((c = input()) != EOF && c != '\0') |
| 618 | bclass(c); |
| 619 | bcheck2(bracecnt, '{', '}'); |
| 620 | bcheck2(brackcnt, '[', ']'); |
| 621 | bcheck2(parencnt, '(', ')'); |
| 622 | } |
| 623 | |
| 624 | void bcheck2(int n, int c1, int c2) |
| 625 | { |
| 626 | if (n == 1) |
| 627 | fprintf(stderr, "\tmissing %c\n", c2); |
| 628 | else if (n > 1) |
| 629 | fprintf(stderr, "\t%d missing %c's\n", n, c2); |
| 630 | else if (n == -1) |
| 631 | fprintf(stderr, "\textra %c\n", c2); |
| 632 | else if (n < -1) |
| 633 | fprintf(stderr, "\t%d extra %c's\n", -n, c2); |
| 634 | } |
| 635 | |
| 636 | void FATAL(const char *fmt, ...) |
| 637 | { |
| 638 | extern char *cmdname; |
| 639 | va_list varg; |
| 640 | |
| 641 | fflush(stdout); |
| 642 | fprintf(stderr, "%s: ", cmdname); |
| 643 | va_start(varg, fmt); |
| 644 | vfprintf(stderr, fmt, varg); |
| 645 | va_end(varg); |
| 646 | error(); |
| 647 | if (dbg > 1) /* core dump if serious debugging on */ |
| 648 | abort(); |
| 649 | exit(2); |
| 650 | } |
| 651 | |
| 652 | void WARNING(const char *fmt, ...) |
| 653 | { |
| 654 | extern char *cmdname; |
| 655 | va_list varg; |
| 656 | |
| 657 | fflush(stdout); |
| 658 | fprintf(stderr, "%s: ", cmdname); |
| 659 | va_start(varg, fmt); |
| 660 | vfprintf(stderr, fmt, varg); |
| 661 | va_end(varg); |
| 662 | error(); |
| 663 | } |
| 664 | |
| 665 | void error() |
| 666 | { |
| 667 | extern Node *curnode; |
| 668 | |
| 669 | fprintf(stderr, "\n"); |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 670 | if (compile_time != ERROR_PRINTING) { |
| 671 | if (NR && *NR > 0) { |
| 672 | fprintf(stderr, " input record number %d", (int) (*FNR)); |
| 673 | if (strcmp(*FILENAME, "-") != 0) |
| 674 | fprintf(stderr, ", file %s", *FILENAME); |
| 675 | fprintf(stderr, "\n"); |
| 676 | } |
| 677 | if (curnode) |
| 678 | fprintf(stderr, " source line number %d", curnode->lineno); |
| 679 | else if (lineno) |
| 680 | fprintf(stderr, " source line number %d", lineno); |
Todd C. Miller | feb247a | 2020-12-03 10:30:36 -0700 | [diff] [blame] | 681 | if (compile_time == COMPILING && cursource() != NULL) |
| 682 | fprintf(stderr, " source file %s", cursource()); |
| 683 | fprintf(stderr, "\n"); |
| 684 | eprint(); |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 685 | } |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 686 | } |
| 687 | |
| 688 | void eprint(void) /* try to print context around error */ |
| 689 | { |
| 690 | char *p, *q; |
| 691 | int c; |
| 692 | static int been_here = 0; |
| 693 | extern char ebuf[], *ep; |
| 694 | |
Arnold D. Robbins | 108224b | 2019-11-10 21:19:18 +0200 | [diff] [blame] | 695 | if (compile_time != COMPILING || been_here++ > 0 || ebuf == ep) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 696 | return; |
Brian Kernighan | 3ed9e24 | 2018-08-15 10:45:03 -0400 | [diff] [blame] | 697 | if (ebuf == ep) |
| 698 | return; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 699 | p = ep - 1; |
| 700 | if (p > ebuf && *p == '\n') |
| 701 | p--; |
| 702 | for ( ; p > ebuf && *p != '\n' && *p != '\0'; p--) |
| 703 | ; |
| 704 | while (*p == '\n') |
| 705 | p++; |
| 706 | fprintf(stderr, " context is\n\t"); |
| 707 | for (q=ep-1; q>=p && *q!=' ' && *q!='\t' && *q!='\n'; q--) |
| 708 | ; |
| 709 | for ( ; p < q; p++) |
| 710 | if (*p) |
| 711 | putc(*p, stderr); |
| 712 | fprintf(stderr, " >>> "); |
| 713 | for ( ; p < ep; p++) |
| 714 | if (*p) |
| 715 | putc(*p, stderr); |
| 716 | fprintf(stderr, " <<< "); |
| 717 | if (*ep) |
| 718 | while ((c = input()) != '\n' && c != '\0' && c != EOF) { |
| 719 | putc(c, stderr); |
| 720 | bclass(c); |
| 721 | } |
| 722 | putc('\n', stderr); |
| 723 | ep = ebuf; |
| 724 | } |
| 725 | |
| 726 | void bclass(int c) |
| 727 | { |
| 728 | switch (c) { |
| 729 | case '{': bracecnt++; break; |
| 730 | case '}': bracecnt--; break; |
| 731 | case '[': brackcnt++; break; |
| 732 | case ']': brackcnt--; break; |
| 733 | case '(': parencnt++; break; |
| 734 | case ')': parencnt--; break; |
| 735 | } |
| 736 | } |
| 737 | |
| 738 | double errcheck(double x, const char *s) |
| 739 | { |
| 740 | |
| 741 | if (errno == EDOM) { |
| 742 | errno = 0; |
| 743 | WARNING("%s argument out of domain", s); |
| 744 | x = 1; |
| 745 | } else if (errno == ERANGE) { |
| 746 | errno = 0; |
| 747 | WARNING("%s result out of range", s); |
| 748 | x = 1; |
| 749 | } |
| 750 | return x; |
| 751 | } |
| 752 | |
| 753 | int isclvar(const char *s) /* is s of form var=something ? */ |
| 754 | { |
| 755 | const char *os = s; |
| 756 | |
| 757 | if (!isalpha((uschar) *s) && *s != '_') |
| 758 | return 0; |
| 759 | for ( ; *s; s++) |
| 760 | if (!(isalnum((uschar) *s) || *s == '_')) |
| 761 | break; |
Arnold D. Robbins | 4189ef5 | 2019-05-29 21:04:18 +0300 | [diff] [blame] | 762 | return *s == '=' && s > os; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 763 | } |
| 764 | |
| 765 | /* strtod is supposed to be a proper test of what's a valid number */ |
| 766 | /* appears to be broken in gcc on linux: thinks 0x123 is a valid FP number */ |
| 767 | /* wrong: violates 4.10.1.4 of ansi C standard */ |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 768 | |
Arnold D. Robbins | cef5180 | 2020-06-12 14:30:03 +0300 | [diff] [blame] | 769 | /* well, not quite. As of C99, hex floating point is allowed. so this is |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 770 | * a bit of a mess. We work around the mess by checking for a hexadecimal |
| 771 | * value and disallowing it. Similarly, we now follow gawk and allow only |
| 772 | * +nan, -nan, +inf, and -inf for NaN and infinity values. |
Arnold D. Robbins | cef5180 | 2020-06-12 14:30:03 +0300 | [diff] [blame] | 773 | */ |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 774 | |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 775 | /* |
| 776 | * This routine now has a more complicated interface, the main point |
| 777 | * being to avoid the double conversion of a string to double, and |
| 778 | * also to convey out, if requested, the information that the numeric |
| 779 | * value was a leading string or is all of the string. The latter bit |
| 780 | * is used in getfval(). |
| 781 | */ |
| 782 | |
| 783 | bool is_valid_number(const char *s, bool trailing_stuff_ok, |
| 784 | bool *no_trailing, double *result) |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 785 | { |
| 786 | double r; |
| 787 | char *ep; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 788 | bool retval = false; |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 789 | bool is_nan = false; |
| 790 | bool is_inf = false; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 791 | |
| 792 | if (no_trailing) |
| 793 | *no_trailing = false; |
| 794 | |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 795 | while (isspace(*s)) |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 796 | s++; |
| 797 | |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 798 | // no hex floating point, sorry |
| 799 | if (s[0] == '0' && tolower(s[1]) == 'x') |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 800 | return false; |
| 801 | |
| 802 | // allow +nan, -nan, +inf, -inf, any other letter, no |
| 803 | if (s[0] == '+' || s[0] == '-') { |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 804 | is_nan = (strncasecmp(s+1, "nan", 3) == 0); |
| 805 | is_inf = (strncasecmp(s+1, "inf", 3) == 0); |
| 806 | if ((is_nan || is_inf) |
| 807 | && (isspace(s[4]) || s[4] == '\0')) |
| 808 | goto convert; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 809 | else if (! isdigit(s[1]) && s[1] != '.') |
| 810 | return false; |
| 811 | } |
| 812 | else if (! isdigit(s[0]) && s[0] != '.') |
| 813 | return false; |
| 814 | |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 815 | convert: |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 816 | errno = 0; |
| 817 | r = strtod(s, &ep); |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 818 | if (ep == s || errno == ERANGE) |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 819 | return false; |
| 820 | |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 821 | if (isnan(r) && s[0] == '-' && signbit(r) == 0) |
| 822 | r = -r; |
| 823 | |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 824 | if (result != NULL) |
| 825 | *result = r; |
| 826 | |
ozan s. yigit | 1fd5fa3 | 2021-01-06 18:37:48 -0500 | [diff] [blame] | 827 | /* |
| 828 | * check for trailing stuff |
| 829 | */ |
| 830 | while (isspace(*ep)) |
| 831 | ep++; |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 832 | |
Arnold D. Robbins | 8909e00 | 2020-12-18 11:57:48 +0200 | [diff] [blame] | 833 | if (no_trailing != NULL) |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 834 | *no_trailing = (*ep == '\0'); |
| 835 | |
ozan s. yigit | 1fd5fa3 | 2021-01-06 18:37:48 -0500 | [diff] [blame] | 836 | // return true if found the end, or trailing stuff is allowed |
| 837 | retval = *ep == '\0' || trailing_stuff_ok; |
| 838 | |
Arnold Robbins | cc9e9b6 | 2020-12-08 08:05:22 +0200 | [diff] [blame] | 839 | return retval; |
Brian Kernighan | 87b9493 | 2012-12-22 10:35:39 -0500 | [diff] [blame] | 840 | } |