vinyl-cache/lib/libvinylapi/vsl.c
0
/*-
1
 * Copyright (c) 2006 Verdens Gang AS
2
 * Copyright (c) 2006-2015 Varnish Software AS
3
 * All rights reserved.
4
 *
5
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
6
 * Author: Martin Blix Grydeland <martin@varnish-software.com>
7
 *
8
 * SPDX-License-Identifier: BSD-2-Clause
9
 *
10
 * Redistribution and use in source and binary forms, with or without
11
 * modification, are permitted provided that the following conditions
12
 * are met:
13
 * 1. Redistributions of source code must retain the above copyright
14
 *    notice, this list of conditions and the following disclaimer.
15
 * 2. Redistributions in binary form must reproduce the above copyright
16
 *    notice, this list of conditions and the following disclaimer in the
17
 *    documentation and/or other materials provided with the distribution.
18
 *
19
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
23
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29
 * SUCH DAMAGE.
30
 */
31
32
#include "config.h"
33
34
#include <stdarg.h>
35
#include <stdint.h>
36
#include <stdio.h>
37
38
#include "vdef.h"
39
#include "vas.h"
40
#include "miniobj.h"
41
42
#include "vbm.h"
43
#include "vqueue.h"
44
#include "vre.h"
45
#include "vsb.h"
46
47
#include "vapi/vsl.h"
48
49
#include "vsl_api.h"
50
51
/*--------------------------------------------------------------------*/
52
53
const char                      vsl_file_id[] = {'V', 'S', 'L', '2'};
54
55
const char * const VSL_tags[SLT__MAX] = {
56
#  define SLTM(foo,flags,sdesc,ldesc)       [SLT_##foo] = #foo,
57
#  include "tbl/vsl_tags.h"
58
};
59
60
const unsigned VSL_tagflags[SLT__MAX] = {
61
#  define SLTM(foo, flags, sdesc, ldesc)        [SLT_##foo] = flags,
62
#  include "tbl/vsl_tags.h"
63
#  undef SLTM
64
};
65
66
int
67 142781
vsl_diag(struct VSL_data *vsl, const char *fmt, ...)
68
{
69
        va_list ap;
70
71 142781
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
72 142781
        AN(fmt);
73
74 142781
        if (vsl->diag == NULL)
75 142780
                vsl->diag = VSB_new_auto();
76 142781
        AN(vsl->diag);
77 142781
        VSB_clear(vsl->diag);
78 142781
        va_start(ap, fmt);
79 142781
        VSB_vprintf(vsl->diag, fmt, ap);
80 142781
        va_end(ap);
81 142781
        AZ(VSB_finish(vsl->diag));
82 142781
        return (-1);
83
}
84
85
struct VSL_data *
86 32444
VSL_New(void)
87
{
88
        struct VSL_data *vsl;
89
90 32444
        ALLOC_OBJ(vsl, VSL_MAGIC);
91 32444
        if (vsl == NULL)
92 0
                return (NULL);
93
94 32444
        vsl->L_opt = 1000;
95 32444
        vsl->T_opt = 120.;
96 32444
        vsl->vbm_select = vbit_new(SLT__MAX);
97 32444
        vsl->vbm_suppress = vbit_new(SLT__MAX);
98 32444
        VTAILQ_INIT(&vsl->vslf_select);
99 32444
        VTAILQ_INIT(&vsl->vslf_suppress);
100
101 32444
        return (vsl);
102 32444
}
103
104
static void
105 58714
vsl_IX_free(vslf_list *filters)
106
{
107
        struct vslf *vslf;
108
109 58798
        while (!VTAILQ_EMPTY(filters)) {
110 84
                vslf = VTAILQ_FIRST(filters);
111 84
                CHECK_OBJ_NOTNULL(vslf, VSLF_MAGIC);
112 84
                VTAILQ_REMOVE(filters, vslf, list);
113 84
                if (vslf->tags)
114 0
                        vbit_destroy(vslf->tags);
115 84
                AN(vslf->vre);
116 84
                VRE_free(&vslf->vre);
117 84
                AZ(vslf->vre);
118 84
                FREE_OBJ(vslf);
119
        }
120 58714
}
121
122
void
123 29357
VSL_Delete(struct VSL_data *vsl)
124
{
125
126 29357
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
127
128 29357
        vbit_destroy(vsl->vbm_select);
129 29357
        vbit_destroy(vsl->vbm_suppress);
130 29357
        vsl_IX_free(&vsl->vslf_select);
131 29357
        vsl_IX_free(&vsl->vslf_suppress);
132 29357
        VSL_ResetError(vsl);
133 29357
        FREE_OBJ(vsl);
134 29357
}
135
136
const char *
137 142780
VSL_Error(const struct VSL_data *vsl)
138
{
139
140 142780
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
141
142 142780
        if (vsl->diag == NULL)
143 0
                return (NULL);
144
        else
145 142780
                return (VSB_data(vsl->diag));
146 142780
}
147
148
void
149 171025
VSL_ResetError(struct VSL_data *vsl)
150
{
151
152 171025
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
153
154 171025
        if (vsl->diag == NULL)
155 29357
                return;
156 141668
        VSB_destroy(&vsl->diag);
157 171025
}
158
159
static int
160 1092
vsl_match_IX(struct VSL_data *vsl, const vslf_list *list,
161
    const struct VSL_cursor *c)
162
{
163
        enum VSL_tag_e tag;
164
        const char *cdata;
165
        int len;
166
        const struct vslf *vslf;
167
168 1092
        (void)vsl;
169 1092
        tag = VSL_TAG(c->rec.ptr);
170 1092
        cdata = VSL_CDATA(c->rec.ptr);
171 1092
        len = VSL_LEN(c->rec.ptr);
172
173 1890
        VTAILQ_FOREACH(vslf, list, list) {
174 1092
                CHECK_OBJ_NOTNULL(vslf, VSLF_MAGIC);
175 1092
                if (vslf->tags != NULL && !vbit_test(vslf->tags, tag))
176 0
                        continue;
177 1092
                if (VRE_match(vslf->vre, cdata, len, 0, NULL) >= 0)
178 294
                        return (1);
179 798
        }
180 798
        return (0);
181 1092
}
182
183
int
184 679955
VSL_Match(struct VSL_data *vsl, const struct VSL_cursor *c)
185
{
186
        enum VSL_tag_e tag;
187
188 679955
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
189 679955
        if (c == NULL || c->rec.ptr == NULL)
190 14
                return (0);
191 679955
        tag = VSL_TAG(c->rec.ptr);
192 679955
        if (tag <= SLT__Bogus || tag >= SLT__Reserved)
193 12
                return (0);
194 679947
        if (!vsl->c_opt || !vsl->b_opt) {
195 679235
                if (vsl->c_opt && !VSL_CLIENT(c->rec.ptr))
196 2058
                        return (0);
197 677177
                if (vsl->b_opt && !VSL_BACKEND(c->rec.ptr))
198 0
                        return (0);
199 677177
        }
200 677889
        if (!VTAILQ_EMPTY(&vsl->vslf_select) &&
201 126
            vsl_match_IX(vsl, &vsl->vslf_select, c))
202 42
                return (1);
203 677847
        else if (vbit_test(vsl->vbm_select, tag))
204 0
                return (1);
205 677847
        else if (!VTAILQ_EMPTY(&vsl->vslf_suppress) &&
206 966
            vsl_match_IX(vsl, &vsl->vslf_suppress, c))
207 252
                return (0);
208 677595
        else if (vbit_test(vsl->vbm_suppress, tag))
209 35679
                return (0);
210
211
        /* Default show */
212 641916
        return (1);
213 679947
}
214
215
static const char * const VSL_transactions[VSL_t__MAX] = {
216
        /*                 12345678901234 */
217
        [VSL_t_unknown] = "<< Unknown  >>",
218
        [VSL_t_sess]    = "<< Session  >>",
219
        [VSL_t_req]     = "<< Request  >>",
220
        [VSL_t_bereq]   = "<< BeReq    >>",
221
        [VSL_t_raw]     = "<< Record   >>",
222
};
223
224
#define VSL_PRINT(...)                                  \
225
        do {                                            \
226
                if (0 > fprintf(__VA_ARGS__))           \
227
                        return (-5);                    \
228
        } while (0)
229
230
static int
231 1686
vsl_print_unsafe(FILE *fo, unsigned len, const char *data)
232
{
233
234 1686
        VSL_PRINT(fo, "\"");
235 97294
        while (len-- > 0) {
236 95608
                if (*data >= ' ' && *data <= '~')
237 93922
                        VSL_PRINT(fo, "%c", *data);
238
                else
239 1686
                        VSL_PRINT(fo, "%%%02x", (unsigned char)*data);
240 95608
                data++;
241
        }
242 1686
        VSL_PRINT(fo, "\"\n");
243 1686
        return (0);
244 1686
}
245
246
247
static int
248 470
vsl_print_binary(FILE *fo, unsigned len, const char *data)
249
{
250
251 470
        VSL_PRINT(fo, "[");
252 10883
        while (len-- > 0) {
253 10413
                VSL_PRINT(fo, "%02x", (unsigned char)*data);
254 10413
                data++;
255
        }
256 470
        VSL_PRINT(fo, "]\n");
257 470
        return (0);
258 470
}
259
260
static int
261 11889
vsl_print(const struct VSL_data *vsl, const struct VSL_cursor *c, void *fo,
262
    int terse)
263
{
264
        enum VSL_tag_e tag;
265
        uint64_t vxid;
266
        unsigned len;
267
        const char *data;
268
        int type;
269
270 11889
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
271 11889
        if (c == NULL || c->rec.ptr == NULL)
272 0
                return (0);
273 11889
        if (fo == NULL)
274 0
                fo = stdout;
275 11889
        tag = VSL_TAG(c->rec.ptr);
276 11889
        vxid = VSL_ID(c->rec.ptr);
277 11889
        len = VSL_LEN(c->rec.ptr);
278 11889
        type = VSL_CLIENT(c->rec.ptr) ? 'c' : VSL_BACKEND(c->rec.ptr) ?
279
            'b' : '-';
280 11889
        data = VSL_CDATA(c->rec.ptr);
281
282 11889
        if (!terse)
283 9516
                VSL_PRINT(fo, "%10ju ", (uintmax_t)vxid);
284 11889
        VSL_PRINT(fo, "%-14s ", VSL_tags[tag]);
285 11889
        if (!terse)
286 9516
                VSL_PRINT(fo, "%c ", type);
287
288 11889
        if (VSL_tagflags[tag] & SLT_F_UNSAFE)
289 1686
                (void)vsl_print_unsafe(fo, len, data);
290 10203
        else if (VSL_tagflags[tag] & SLT_F_BINARY)
291 470
                (void)vsl_print_binary(fo, len, data);
292
        else
293 9733
                VSL_PRINT(fo, "%.*s\n", (int)len, data);
294
295 11889
        return (0);
296 11889
}
297
298
int
299 9516
VSL_Print(const struct VSL_data *vsl, const struct VSL_cursor *c, void *fo)
300
{
301
302 9516
        return (vsl_print(vsl, c, fo, 0));
303
}
304
305
int
306 2373
VSL_PrintTerse(const struct VSL_data *vsl, const struct VSL_cursor *c, void *fo)
307
{
308
309 2373
        return (vsl_print(vsl, c, fo, 1));
310
}
311
312
int
313 0
VSL_PrintAll(struct VSL_data *vsl, const struct VSL_cursor *c, void *fo)
314
{
315
        int i;
316
317 0
        if (c == NULL)
318 0
                return (0);
319 0
        while (1) {
320 0
                i = VSL_Next(c);
321 0
                if (i <= 0)
322 0
                        return (i);
323 0
                if (!VSL_Match(vsl, c))
324 0
                        continue;
325 0
                i = VSL_Print(vsl, c, fo);
326 0
                if (i != 0)
327 0
                        return (i);
328
        }
329 0
}
330
331
int v_matchproto_(VSLQ_dispatch_f)
332 17706
VSL_PrintTransactions(struct VSL_data *vsl, struct VSL_transaction * const pt[],
333
    void *fo)
334
{
335
        struct VSL_transaction *t;
336
        int i;
337 17706
        int delim = 0;
338
        int verbose;
339
340 17706
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
341 17706
        if (fo == NULL)
342 14679
                fo = stdout;
343 17706
        if (pt[0] == NULL)
344 0
                return (0);
345
346 35538
        for (t = pt[0]; t != NULL; t = *++pt) {
347 17832
                if (vsl->c_opt || vsl->b_opt) {
348 8085
                        switch (t->type) {
349
                        case VSL_t_req:
350 420
                                if (!vsl->c_opt)
351 0
                                        continue;
352 420
                                if (t->reason == VSL_r_esi && !vsl->E_opt)
353 84
                                        continue;
354 336
                                break;
355
                        case VSL_t_bereq:
356 84
                                if (!vsl->b_opt)
357 42
                                        continue;
358 42
                                break;
359
                        case VSL_t_raw:
360 7518
                                break;
361
                        default:
362 63
                                continue;
363
                        }
364 7896
                }
365
366 17643
                verbose = 0;
367 17643
                if (t->level == 0 || vsl->v_opt)
368 17139
                        verbose = 1;
369
370 17643
                if (t->level) {
371
                        /* Print header */
372 525
                        if (t->level > 3)
373 0
                                VSL_PRINT(fo, "*%1.1d* ", t->level);
374
                        else
375 525
                                VSL_PRINT(fo, "%-3.*s ",
376
                                    (int)(t->level), "***");
377 525
                        VSL_PRINT(fo, "%*.s%-14s %*.s%-10ju\n",
378
                            verbose ? 10 + 1 : 0, " ",
379
                            VSL_transactions[t->type],
380
                            verbose ? 1 + 1 : 0, " ",
381
                            (uintmax_t)t->vxid);
382 525
                        delim = 1;
383 525
                }
384
385 29532
                while (1) {
386
                        /* Print records */
387 51330
                        i = VSL_Next(t->c);
388 51330
                        if (i < 0)
389 0
                                return (i);
390 51330
                        if (i == 0)
391 17643
                                break;
392 33687
                        if (!VSL_Match(vsl, t->c))
393 21798
                                continue;
394 11889
                        if (t->level > 3)
395 0
                                VSL_PRINT(fo, "-%1.1d- ", t->level);
396 11889
                        else if (t->level)
397 2394
                                VSL_PRINT(fo, "%-3.*s ",
398
                                    (int)(t->level), "---");
399 11889
                        if (verbose)
400 9516
                                i = VSL_Print(vsl, t->c, fo);
401
                        else
402 2373
                                i = VSL_PrintTerse(vsl, t->c, fo);
403 11889
                        if (i != 0)
404 0
                                return (i);
405
                }
406 17643
        }
407
408 17706
        if (delim)
409 525
                VSL_PRINT(fo, "\n");;
410
411 17706
        return (0);
412 17706
}
413
414
FILE*
415 84
VSL_WriteOpen(struct VSL_data *vsl, const char *name, int append, int unbuf)
416
{
417
        FILE* f;
418
419 84
        if (!vstrcmp(name, "-"))
420 21
                f = stdout;
421
        else
422 63
                f = fopen(name, append ? "a" : "w");
423 84
        if (f == NULL) {
424 21
                vsl_diag(vsl, "%s", strerror(errno));
425 21
                return (NULL);
426
        }
427 63
        if (unbuf)
428 0
                setbuf(f, NULL);
429 63
        if (ftell(f) == 0 || f == stdout) {
430 63
                if (fwrite(VSL_FILE_ID, 1, sizeof VSL_FILE_ID, f) !=
431
                    sizeof VSL_FILE_ID) {
432 0
                        vsl_diag(vsl, "%s", strerror(errno));
433 0
                        (void)fclose(f);
434 0
                        return (NULL);
435
                }
436 63
        }
437 63
        return (f);
438 84
}
439
440
int
441 2247
VSL_Write(const struct VSL_data *vsl, const struct VSL_cursor *c, void *fo)
442
{
443
        size_t r;
444
445 2247
        CHECK_OBJ_NOTNULL(vsl, VSL_MAGIC);
446 2247
        if (c == NULL || c->rec.ptr == NULL)
447 0
                return (0);
448 2247
        if (fo == NULL)
449 0
                fo = stdout;
450 4494
        r = fwrite(c->rec.ptr, sizeof *c->rec.ptr,
451 2247
            VSL_NEXT(c->rec.ptr) - c->rec.ptr, fo);
452 2247
        if (r == 0)
453 0
                return (-5);
454 2247
        return (0);
455 2247
}
456
457
int
458 189
VSL_WriteAll(struct VSL_data *vsl, const struct VSL_cursor *c, void *fo)
459
{
460
        int i;
461
462 189
        if (c == NULL)
463 0
                return (0);
464 2436
        while (1) {
465 5754
                i = VSL_Next(c);
466 5754
                if (i <= 0)
467 189
                        return (i);
468 5565
                if (!VSL_Match(vsl, c))
469 3318
                        continue;
470 2247
                i = VSL_Write(vsl, c, fo);
471 2247
                if (i != 0)
472 0
                        return (i);
473
        }
474 189
}
475
476
int v_matchproto_(VSLQ_dispatch_f)
477 189
VSL_WriteTransactions(struct VSL_data *vsl, struct VSL_transaction * const pt[],
478
    void *fo)
479
{
480
        struct VSL_transaction *t;
481
        int i;
482
483 189
        if (pt == NULL)
484 0
                return (0);
485 378
        for (i = 0, t = pt[0]; i == 0 && t != NULL; t = *++pt)
486 189
                i = VSL_WriteAll(vsl, t->c, fo);
487 189
        return (i);
488 189
}