varnish-cache/bin/varnishtest/vtc.c
0
/*-
1
 * Copyright (c) 2008-2011 Varnish Software AS
2
 * All rights reserved.
3
 *
4
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
5
 *
6
 * SPDX-License-Identifier: BSD-2-Clause
7
 *
8
 * Redistribution and use in source and binary forms, with or without
9
 * modification, are permitted provided that the following conditions
10
 * are met:
11
 * 1. Redistributions of source code must retain the above copyright
12
 *    notice, this list of conditions and the following disclaimer.
13
 * 2. Redistributions in binary form must reproduce the above copyright
14
 *    notice, this list of conditions and the following disclaimer in the
15
 *    documentation and/or other materials provided with the distribution.
16
 *
17
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
21
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27
 * SUCH DAMAGE.
28
 */
29
30
#include "config.h"
31
32
#include <sys/wait.h>
33
34
#include <ctype.h>
35
#include <stdarg.h>
36
#include <stdio.h>
37
#include <stdlib.h>
38
#include <string.h>
39
#include <unistd.h>
40
41
#include "vtc.h"
42
#include "vtc_log.h"
43
44
#include "vav.h"
45
#include "vrnd.h"
46
47
#define         MAX_TOKENS              200
48
49
volatile sig_atomic_t   vtc_error;      /* Error encountered */
50
int                     vtc_stop;       /* Stops current test without error */
51
pthread_t               vtc_thread;
52
int                     ign_unknown_macro = 0;
53
static struct vtclog    *vltop;
54
55
static pthread_mutex_t  vtc_vrnd_mtx;
56
57
static void
58 165120
vtc_vrnd_lock(void)
59
{
60 165120
        PTOK(pthread_mutex_lock(&vtc_vrnd_mtx));
61 165120
}
62
63
static void
64 165120
vtc_vrnd_unlock(void)
65
{
66 165120
        PTOK(pthread_mutex_unlock(&vtc_vrnd_mtx));
67 165120
}
68
69
static const char *tfn;
70
71
/**********************************************************************
72
 * Macro facility
73
 */
74
75
struct macro {
76
        unsigned                magic;
77
#define MACRO_MAGIC             0x803423e3
78
        VTAILQ_ENTRY(macro)     list;
79
        char                    *name;
80
        char                    *val;
81
        macro_f                 *func;
82
};
83
84
static VTAILQ_HEAD(,macro) macro_list = VTAILQ_HEAD_INITIALIZER(macro_list);
85
86
static const struct cmds global_cmds[] = {
87
#define CMD_GLOBAL(n) { #n, cmd_##n },
88
#include "cmds.h"
89
        { NULL, NULL }
90
};
91
92
static const struct cmds top_cmds[] = {
93
#define CMD_TOP(n) { #n, cmd_##n },
94
#include "cmds.h"
95
        { NULL, NULL }
96
};
97
98
/**********************************************************************/
99
100
static struct macro *
101 982920
macro_def_int(const char *name, macro_f *func, const char *fmt, va_list ap)
102
{
103
        struct macro *m;
104
        char buf[2048];
105
106 13157800
        VTAILQ_FOREACH(m, &macro_list, list)
107 12194440
                if (!strcmp(name, m->name))
108 19560
                        break;
109 982920
        if (m == NULL) {
110 963360
                ALLOC_OBJ(m, MACRO_MAGIC);
111 963360
                AN(m);
112 963360
                REPLACE(m->name, name);
113 963360
                AN(m->name);
114 963360
                VTAILQ_INSERT_TAIL(&macro_list, m, list);
115 963360
        }
116 982920
        AN(m);
117 982920
        if (func != NULL) {
118 77520
                AZ(fmt);
119 77520
                m->func = func;
120 77520
        } else {
121 905400
                AN(fmt);
122 905400
                vbprintf(buf, fmt, ap);
123 905400
                REPLACE(m->val, buf);
124 905400
                AN(m->val);
125
        }
126 982920
        return (m);
127
}
128
129
130
/**********************************************************************
131
 * This is for defining macros before we fork the child process which
132
 * runs the test-case.
133
 */
134
135
void
136 423640
extmacro_def(const char *name, macro_f *func, const char *fmt, ...)
137
{
138
        va_list ap;
139
140 423640
        va_start(ap, fmt);
141 423640
        (void)macro_def_int(name, func, fmt, ap);
142 423640
        va_end(ap);
143 423640
}
144
145
/**********************************************************************
146
 * Below this point is run inside the testing child-process.
147
 */
148
149
static pthread_mutex_t          macro_mtx;
150
151
static void
152 39640
init_macro(void)
153
{
154
        struct macro *m;
155
156
        /* Dump the extmacros for completeness */
157 474320
        VTAILQ_FOREACH(m, &macro_list, list) {
158 434680
                if (m->val != NULL)
159 710800
                        vtc_log(vltop, 4,
160 355400
                            "extmacro def %s=%s", m->name, m->val);
161
                else
162 79280
                        vtc_log(vltop, 4, "extmacro def %s(...)", m->name);
163 434680
        }
164
165 39640
        PTOK(pthread_mutex_init(&macro_mtx, NULL));
166 39640
}
167
168
void
169 559280
macro_def(struct vtclog *vl, const char *instance, const char *name,
170
    const char *fmt, ...)
171
{
172
        char buf1[256];
173
        struct macro *m;
174
        va_list ap;
175
176 559280
        AN(fmt);
177
178 559280
        if (instance != NULL) {
179 440360
                bprintf(buf1, "%s_%s", instance, name);
180 440360
                name = buf1;
181 440360
        }
182
183 559280
        PTOK(pthread_mutex_lock(&macro_mtx));
184 559280
        va_start(ap, fmt);
185 559280
        m = macro_def_int(name, NULL, fmt, ap);
186 559280
        va_end(ap);
187 559280
        vtc_log(vl, 4, "macro def %s=%s", name, m->val);
188 559280
        PTOK(pthread_mutex_unlock(&macro_mtx));
189 559280
}
190
191
void
192 139703
macro_undef(struct vtclog *vl, const char *instance, const char *name)
193
{
194
        char buf1[256];
195
        struct macro *m;
196
197 139703
        if (instance != NULL) {
198 139703
                bprintf(buf1, "%s_%s", instance, name);
199 139703
                name = buf1;
200 139703
        }
201
202 139703
        PTOK(pthread_mutex_lock(&macro_mtx));
203 2260874
        VTAILQ_FOREACH(m, &macro_list, list)
204 2253011
                if (!strcmp(name, m->name))
205 131840
                        break;
206 139703
        if (m != NULL) {
207 131840
                if (!vtc_stop)
208 6960
                        vtc_log(vl, 4, "macro undef %s", name);
209 131840
                CHECK_OBJ(m, MACRO_MAGIC);
210 131840
                VTAILQ_REMOVE(&macro_list, m, list);
211 131840
                free(m->name);
212 131840
                free(m->val);
213 131840
                FREE_OBJ(m);
214 131840
        }
215 139703
        PTOK(pthread_mutex_unlock(&macro_mtx));
216 139703
}
217
218
unsigned
219 73600
macro_isdef(const char *instance, const char *name)
220
{
221
        char buf1[256];
222
        struct macro *m;
223
224 73600
        if (instance != NULL) {
225 0
                bprintf(buf1, "%s_%s", instance, name);
226 0
                name = buf1;
227 0
        }
228
229 73600
        PTOK(pthread_mutex_lock(&macro_mtx));
230 1349600
        VTAILQ_FOREACH(m, &macro_list, list)
231 1276000
                if (!strcmp(name, m->name))
232 0
                        break;
233 73600
        PTOK(pthread_mutex_unlock(&macro_mtx));
234
235 73600
        return (m != NULL);
236
}
237
238
void
239 458616
macro_cat(struct vtclog *vl, struct vsb *vsb, const char *b, const char *e)
240
{
241
        struct macro *m;
242 458616
        char **argv, *retval = NULL;
243 458616
        const char *err = NULL;
244
        int argc;
245
246 458616
        AN(b);
247 458616
        if (e == NULL)
248 241850
                e = strchr(b, '\0');
249 458616
        AN(e);
250
251 458616
        argv = VAV_ParseTxt(b, e, &argc, ARGV_COMMA);
252 458616
        AN(argv);
253
254 458616
        if (*argv != NULL)
255 0
                vtc_fatal(vl, "Macro ${%.*s} parsing failed: %s",
256 0
                    (int)(e - b), b, *argv);
257
258 458616
        assert(argc >= 2);
259
260 458616
        PTOK(pthread_mutex_lock(&macro_mtx));
261 4265807
        VTAILQ_FOREACH(m, &macro_list, list) {
262 4263807
                CHECK_OBJ_NOTNULL(m, MACRO_MAGIC);
263 4263807
                if (!strcmp(argv[1], m->name))
264 456616
                        break;
265 3807191
        }
266 458616
        if (m != NULL) {
267 456616
                if (m->func != NULL) {
268 85976
                        AZ(m->val);
269 85976
                        retval = m->func(argc, argv, &err);
270 85976
                        if (err == NULL)
271 85976
                                AN(retval);
272 85976
                } else {
273 370640
                        AN(m->val);
274 370640
                        if (argc == 2)
275 370640
                                REPLACE(retval, m->val);
276
                        else
277 0
                                err = "macro does not take arguments";
278
                }
279 456616
        }
280 458616
        PTOK(pthread_mutex_unlock(&macro_mtx));
281
282 458616
        VAV_Free(argv);
283
284 458616
        if (err != NULL)
285 0
                vtc_fatal(vl, "Macro ${%.*s} failed: %s",
286 0
                    (int)(e - b), b, err);
287
288 458616
        if (retval == NULL) {
289 2000
                if (!ign_unknown_macro)
290 0
                        vtc_fatal(vl, "Macro ${%.*s} not found",
291 0
                            (int)(e - b), b);
292 2000
                VSB_printf(vsb, "${%.*s}", (int)(e - b), b);
293 2000
                return;
294
        }
295
296 456616
        VSB_cat(vsb, retval);
297 456616
        free(retval);
298 458616
}
299
300
struct vsb *
301 54878
macro_expandf(struct vtclog *vl, const char *fmt, ...)
302
{
303
        va_list ap;
304
        struct vsb *vsb1, *vsb2;
305
306 54878
        vsb1 = VSB_new_auto();
307 54878
        AN(vsb1);
308 54878
        va_start(ap, fmt);
309 54878
        VSB_vprintf(vsb1, fmt, ap);
310 54878
        va_end(ap);
311 54878
        AZ(VSB_finish(vsb1));
312 54878
        vsb2 = macro_expand(vl, VSB_data(vsb1));
313 54878
        VSB_destroy(&vsb1);
314 54878
        return (vsb2);
315
}
316
317
struct vsb *
318 206280
macro_expand(struct vtclog *vl, const char *text)
319
{
320
        struct vsb *vsb;
321
        const char *p, *q;
322
323 206280
        vsb = VSB_new_auto();
324 206280
        AN(vsb);
325 423039
        while (*text != '\0') {
326 350679
                p = strstr(text, "${");
327 350679
                if (p == NULL) {
328 133920
                        VSB_cat(vsb, text);
329 133920
                        break;
330
                }
331 216759
                VSB_bcat(vsb, text, p - text);
332 216759
                q = strchr(p, '}');
333 216759
                if (q == NULL) {
334 0
                        VSB_cat(vsb, text);
335 0
                        break;
336
                }
337 216759
                assert(p[0] == '$');
338 216759
                assert(p[1] == '{');
339 216759
                assert(q[0] == '}');
340 216759
                p += 2;
341 216759
                macro_cat(vl, vsb, p, q);
342 216759
                text = q + 1;
343
        }
344 206280
        AZ(VSB_finish(vsb));
345 206280
        return (vsb);
346
}
347
348
/**********************************************************************
349
 * Parse a string
350
 *
351
 * We make a copy of the string and deliberately leak it, so that all
352
 * the cmd functions we call don't have to strdup(3) all over the place.
353
 *
354
 * Static checkers like Coverity may bitch about this, but we don't care.
355
 */
356
357
358
void
359 306617
parse_string(struct vtclog *vl, void *priv, const char *spec)
360
{
361
        char *token_s[MAX_TOKENS], *token_e[MAX_TOKENS];
362
        struct vsb *token_exp;
363
        char *e, *p, *q, *f, *buf;
364
        int nest_brace;
365
        int tn;
366
        unsigned n, m;
367
        const struct cmds *cp;
368
369 306617
        AN(spec);
370 306617
        buf = strdup(spec);
371 306617
        AN(buf);
372 306617
        e = strchr(buf, '\0');
373 306617
        AN(e);
374 3996394
        for (p = buf; p < e; p++) {
375 3690926
                if (vtc_error || vtc_stop)
376 1135
                        break;
377
                /* Start of line */
378 3689797
                if (isspace(*p))
379 2163867
                        continue;
380 1525930
                if (*p == '\n')
381 0
                        continue;
382
383 1525930
                if (*p == '#') {
384 3470728
                        for (; *p != '\0' && *p != '\n'; p++)
385
                                ;
386 84000
                        if (*p == '\0')
387 0
                                break;
388 84000
                        continue;
389
                }
390
391 1441908
                q = strchr(p, '\n');
392 1441908
                if (q == NULL)
393 14912
                        q = strchr(p, '\0');
394 1441908
                if (q - p > 60)
395 115456
                        vtc_log(vl, 2, "=== %.60s...", p);
396
                else
397 1326452
                        vtc_log(vl, 2, "=== %.*s", (int)(q - p), p);
398
399
                /* First content on line, collect tokens */
400 1441908
                memset(token_s, 0, sizeof token_s);
401 1441908
                memset(token_e, 0, sizeof token_e);
402 1441908
                tn = 0;
403 1441908
                f = p;
404 9752340
                while (p < e) {
405 9737773
                        assert(tn < MAX_TOKENS);
406 9737773
                        assert(p < e);
407 9737773
                        if (*p == '\n') { /* End on NL */
408 1427341
                                break;
409
                        }
410 8310432
                        if (isspace(*p)) { /* Inter-token whitespace */
411 3528352
                                p++;
412 3528352
                                continue;
413
                        }
414 4782080
                        if (*p == '\\' && p[1] == '\n') { /* line-cont */
415 37920
                                p += 2;
416 37920
                                continue;
417
                        }
418 4744160
                        if (*p == '"') { /* quotes */
419 456027
                                token_s[tn] = ++p;
420 456027
                                q = p;
421 11073543
                                for (; *p != '\0'; p++) {
422 11072903
                                        assert(p < e);
423 11072903
                                        if (*p == '"')
424 455387
                                                break;
425 10617516
                                        if (*p == '\\') {
426 110536
                                                p += VAV_BackSlash(p, q) - 1;
427 110536
                                                q++;
428 110536
                                        } else {
429 10506980
                                                if (*p == '\n')
430 0
                                                        vtc_fatal(vl,
431
                                "Unterminated quoted string in line: %*.*s",
432 0
                                (int)(p - f), (int)(p - f), f);
433 10506980
                                                assert(*p != '\n');
434 10506980
                                                *q++ = *p;
435
                                        }
436 10617516
                                }
437 456027
                                token_e[tn++] = q;
438 456027
                                p++;
439 4744160
                        } else if (*p == '{') { /* Braces */
440 298486
                                nest_brace = 0;
441 298486
                                token_s[tn] = p + 1;
442 48302078
                                for (; p < e; p++) {
443 48302078
                                        if (*p == '{')
444 546327
                                                nest_brace++;
445 47755751
                                        else if (*p == '}') {
446 546326
                                                if (--nest_brace == 0)
447 298486
                                                        break;
448 247840
                                        }
449 48003592
                                }
450 298486
                                assert(*p == '}');
451 298486
                                token_e[tn++] = p++;
452 298486
                        } else { /* other tokens */
453 3989647
                                token_s[tn] = p;
454 189922529
                                for (; p < e && !isspace(*p); p++)
455 185932882
                                        continue;
456 3989647
                                token_e[tn++] = p;
457
                        }
458
                }
459
460 1441908
                assert(p <= e);
461 1441908
                assert(tn < MAX_TOKENS);
462 1441908
                token_s[tn] = NULL;
463 6186232
                for (tn = 0; token_s[tn] != NULL; tn++) {
464 4744324
                        AN(token_e[tn]);        /*lint !e771 */
465 4744324
                        *token_e[tn] = '\0';    /*lint !e771 */
466 4744324
                        if (NULL != strstr(token_s[tn], "${")) {
467 41960
                                token_exp = macro_expand(vl, token_s[tn]);
468 41960
                                if (vtc_error)
469 0
                                        return;
470 41960
                                token_s[tn] = VSB_data(token_exp);
471 41960
                                token_e[tn] = strchr(token_s[tn], '\0');
472 41960
                        }
473 4744324
                }
474
475
476
/* SECTION: loop loop
477
 *
478
 * loop NUMBER STRING
479
 *         Process STRING as a specification, NUMBER times.
480
 *
481
 * This works inside all specification strings
482
 */
483
484 1441908
                if (!strcmp(token_s[0], "loop")) {
485 1691
                        n = strtoul(token_s[1], NULL, 0);
486 71965
                        for (m = 0; m < n; m++) {
487 70274
                                vtc_log(vl, 4, "Loop #%u", m);
488 70274
                                parse_string(vl, priv, token_s[2]);
489 70274
                        }
490 1691
                        continue;
491
                }
492
493 1440217
                AN(vl->cmds);
494 24650143
                for (cp = vl->cmds; cp->name != NULL; cp++)
495 24582016
                        if (!strcmp(token_s[0], cp->name))
496 1372090
                                break;
497
498 1440217
                if (cp->name == NULL) {
499 128898
                        for (cp = global_cmds; cp->name != NULL; cp++)
500 128895
                                if (!strcmp(token_s[0], cp->name))
501 68144
                                        break;
502 68149
                }
503
504 1440219
                if (cp->name == NULL)
505 0
                        vtc_fatal(vl, "Unknown command: \"%s\"", token_s[0]);
506
507 1440219
                assert(cp->cmd != NULL);
508 1440219
                cp->cmd(token_s, priv, vl);
509 1440219
        }
510 305901
}
511
512
/**********************************************************************
513
 * Reset commands (between tests)
514
 */
515
516
static void
517 79280
reset_cmds(const struct cmds *cmd)
518
{
519
520 792800
        for (; cmd->name != NULL; cmd++)
521 713520
                cmd->cmd(NULL, NULL, NULL);
522 79280
}
523
524
/**********************************************************************
525
 * Execute a file
526
 */
527
528
int
529 39640
fail_out(void)
530
{
531
        unsigned old_err;
532
        static int once = 0;
533
534 39640
        if (once++) {
535 0
                vtc_log(vltop, 1, "failure during reset");
536 0
                return (vtc_error);
537
        }
538 39640
        old_err = vtc_error;
539 39640
        if (!vtc_stop)
540 39160
                vtc_stop = 1;
541 39640
        vtc_log(vltop, 1, "RESETTING after %s", tfn);
542 39640
        reset_cmds(global_cmds);
543 39640
        reset_cmds(top_cmds);
544 39640
        vtc_error |= old_err;
545
546 39640
        if (vtc_error)
547 80
                vtc_log(vltop, 1, "TEST %s FAILED", tfn);
548
        else
549 39560
                vtc_log(vltop, 1, "TEST %s completed", tfn);
550
551 39640
        if (vtc_stop > 1)
552 480
                return (1);
553 39160
        return (vtc_error);
554 39640
}
555
556
int
557 39640
exec_file(const char *fn, const char *script, const char *tmpdir,
558
    char *logbuf, unsigned loglen)
559
{
560
        FILE *f;
561
        struct vsb *vsb;
562
        const char *p;
563
564 39640
        (void)signal(SIGPIPE, SIG_IGN);
565
566 39640
        PTOK(pthread_mutex_init(&vtc_vrnd_mtx, NULL));
567 39640
        VRND_Lock = vtc_vrnd_lock;
568 39640
        VRND_Unlock = vtc_vrnd_unlock;
569 39640
        VRND_SeedAll();
570
571 39640
        tfn = fn;
572 39640
        vtc_loginit(logbuf, loglen);
573 39640
        vltop = vtc_logopen("top");
574 39640
        AN(vltop);
575 39640
        vtc_log_set_cmd(vltop, top_cmds);
576
577 39640
        vtc_log(vltop, 1, "TEST %s starting", fn);
578
579 39640
        init_macro();
580 39640
        init_server();
581 39640
        init_syslog();
582 39640
        init_tunnel();
583
584 39640
        vsb = VSB_new_auto();
585 39640
        AN(vsb);
586 39640
        if (*fn != '/')
587 39240
                macro_cat(vltop, vsb, "pwd", NULL);
588 39640
        p = strrchr(fn, '/');
589 39640
        if (p != NULL) {
590 39320
                VSB_putc(vsb, '/');
591 39320
                VSB_bcat(vsb, fn, p - fn);
592 39320
        }
593 39640
        if (VSB_len(vsb) == 0)
594 0
                VSB_putc(vsb, '/');
595 39640
        AZ(VSB_finish(vsb));
596 39640
        macro_def(vltop, NULL, "testdir", "%s", VSB_data(vsb));
597 39640
        VSB_destroy(&vsb);
598
599
        /* Move into our tmpdir */
600 39640
        AZ(chdir(tmpdir));
601 39640
        macro_def(vltop, NULL, "tmpdir", "%s", tmpdir);
602 39640
        p = strrchr(tmpdir, '/');
603 39640
        AN(p);
604 39640
        p++;
605 39640
        AN(*p);
606 39640
        macro_def(vltop, NULL, "vtcid", "%s", p);
607
608
        /* Drop file to tell what was going on here */
609 39640
        f = fopen("INFO", "w");
610 39640
        AN(f);
611 39640
        fprintf(f, "Test case: %s\n", fn);
612 39640
        AZ(fclose(f));
613
614 39640
        vtc_stop = 0;
615
616 39640
        vtc_thread = pthread_self();
617 39640
        parse_string(vltop, NULL, script);
618 39640
        return (fail_out());
619
}