blob: b06ba150b2031b521c3368e39932169b77252c3a [file] [log] [blame]
Christopher Faulete6c3b692015-09-02 17:15:16 +02001/*
2 * Stream filters related variables and functions.
3 *
4 * Copyright (C) 2015 Qualys Inc., Christopher Faulet <cfaulet@qualys.com>
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
11 */
12
Christopher Fauletfcd99f82016-10-31 11:27:21 +010013#include <ctype.h>
14
Willy Tarreau0108d902018-11-25 19:14:37 +010015#include <common/hathreads.h>
Willy Tarreaub96b77e2018-12-11 10:22:41 +010016#include <common/htx.h>
Willy Tarreau0108d902018-11-25 19:14:37 +010017#include <common/initcall.h>
Christopher Faulete6c3b692015-09-02 17:15:16 +020018#include <common/standard.h>
19#include <common/time.h>
20#include <common/tools.h>
21
22#include <types/channel.h>
23#include <types/filters.h>
24#include <types/global.h>
25#include <types/proxy.h>
26#include <types/stream.h>
27
28#include <proto/filters.h>
Christopher Faulete0aa6f72018-11-30 22:23:32 +010029#include <proto/http_htx.h>
Christopher Faulete6c3b692015-09-02 17:15:16 +020030#include <proto/log.h>
Christopher Fauletfc9cfe42019-07-16 14:54:53 +020031#include <proto/http_ana.h>
Christopher Faulete6c3b692015-09-02 17:15:16 +020032#include <proto/stream.h>
33
Christopher Fauletf4a4ef72018-12-07 17:39:53 +010034const char *trace_flt_id = "trace filter";
35
Christopher Faulete6c3b692015-09-02 17:15:16 +020036struct flt_ops trace_ops;
37
38struct trace_config {
39 struct proxy *proxy;
40 char *name;
41 int rand_parsing;
42 int rand_forwarding;
Christopher Fauletfcd99f82016-10-31 11:27:21 +010043 int hexdump;
Christopher Faulete6c3b692015-09-02 17:15:16 +020044};
45
Christopher Fauleta3ed2712019-11-04 11:35:42 +010046#define FLT_TRACE(conf, fmt, ...) \
Christopher Faulete6c3b692015-09-02 17:15:16 +020047 fprintf(stderr, "%d.%06d [%-20s] " fmt "\n", \
48 (int)now.tv_sec, (int)now.tv_usec, (conf)->name, \
49 ##__VA_ARGS__)
50
Christopher Fauleta3ed2712019-11-04 11:35:42 +010051#define FLT_STRM_TRACE(conf, strm, fmt, ...) \
Christopher Fauletfcd99f82016-10-31 11:27:21 +010052 fprintf(stderr, "%d.%06d [%-20s] [strm %p(%x) 0x%08x 0x%08x] " fmt "\n", \
53 (int)now.tv_sec, (int)now.tv_usec, (conf)->name, \
54 strm, (strm ? ((struct stream *)strm)->uniq_id : ~0U), \
55 (strm ? strm->req.analysers : 0), (strm ? strm->res.analysers : 0), \
Christopher Faulete6c3b692015-09-02 17:15:16 +020056 ##__VA_ARGS__)
57
58
59static const char *
60channel_label(const struct channel *chn)
61{
62 return (chn->flags & CF_ISRESP) ? "RESPONSE" : "REQUEST";
63}
64
65static const char *
66proxy_mode(const struct stream *s)
67{
68 struct proxy *px = (s->flags & SF_BE_ASSIGNED ? s->be : strm_fe(s));
69
Christopher Faulet386a0cd2019-07-15 21:22:44 +020070 return ((px->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
Christopher Faulete6c3b692015-09-02 17:15:16 +020071}
72
73static const char *
74stream_pos(const struct stream *s)
75{
76 return (s->flags & SF_BE_ASSIGNED) ? "backend" : "frontend";
77}
78
Christopher Faulet31ed32d2016-06-21 11:42:37 +020079static const char *
80filter_type(const struct filter *f)
81{
82 return (f->flags & FLT_FL_IS_BACKEND_FILTER) ? "backend" : "frontend";
83}
84
Christopher Fauletfcd99f82016-10-31 11:27:21 +010085static void
Christopher Faulete0aa6f72018-11-30 22:23:32 +010086trace_hexdump(struct ist ist)
Christopher Fauletfcd99f82016-10-31 11:27:21 +010087{
Christopher Faulete0aa6f72018-11-30 22:23:32 +010088 int i, j, padding;
Christopher Fauletfcd99f82016-10-31 11:27:21 +010089
Christopher Faulete0aa6f72018-11-30 22:23:32 +010090 padding = ((ist.len % 16) ? (16 - ist.len % 16) : 0);
91 for (i = 0; i < ist.len + padding; i++) {
Christopher Fauletfcd99f82016-10-31 11:27:21 +010092 if (!(i % 16))
93 fprintf(stderr, "\t0x%06x: ", i);
94 else if (!(i % 8))
95 fprintf(stderr, " ");
96
Christopher Faulete0aa6f72018-11-30 22:23:32 +010097 if (i < ist.len)
98 fprintf(stderr, "%02x ", (unsigned char)*(ist.ptr+i));
Christopher Fauletfcd99f82016-10-31 11:27:21 +010099 else
100 fprintf(stderr, " ");
101
102 /* print ASCII dump */
103 if (i % 16 == 15) {
104 fprintf(stderr, " |");
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100105 for(j = i - 15; j <= i && j < ist.len; j++)
Willy Tarreau90807112020-02-25 08:16:33 +0100106 fprintf(stderr, "%c", (isprint((unsigned char)*(ist.ptr+j)) ? *(ist.ptr+j) : '.'));
Christopher Fauletfcd99f82016-10-31 11:27:21 +0100107 fprintf(stderr, "|\n");
108 }
109 }
110}
111
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100112static void
Christopher Fauletb2e58492019-11-12 11:13:01 +0100113trace_raw_hexdump(struct buffer *buf, unsigned int offset, unsigned int len)
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100114{
115 unsigned char p[len];
116 int block1, block2;
117
118 block1 = len;
Christopher Fauletb2e58492019-11-12 11:13:01 +0100119 if (block1 > b_contig_data(buf, offset))
120 block1 = b_contig_data(buf, offset);
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100121 block2 = len - block1;
122
Christopher Fauletb2e58492019-11-12 11:13:01 +0100123 memcpy(p, b_peek(buf, offset), block1);
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100124 memcpy(p+block1, b_orig(buf), block2);
125 trace_hexdump(ist2(p, len));
126}
127
128static void
129trace_htx_hexdump(struct htx *htx, unsigned int offset, unsigned int len)
130{
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100131 struct htx_blk *blk;
132
Christopher Fauletee847d42019-05-23 11:55:33 +0200133 for (blk = htx_get_first_blk(htx); blk && len; blk = htx_get_next_blk(htx, blk)) {
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100134 enum htx_blk_type type = htx_get_blk_type(blk);
Christopher Fauletee847d42019-05-23 11:55:33 +0200135 uint32_t sz = htx_get_blksz(blk);
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100136 struct ist v;
137
Christopher Fauletee847d42019-05-23 11:55:33 +0200138 if (offset >= sz) {
139 offset -= sz;
140 continue;
141 }
142
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100143 v = htx_get_blk_value(htx, blk);
144 v.ptr += offset;
145 v.len -= offset;
146 offset = 0;
147
148 if (v.len > len)
149 v.len = len;
150 len -= v.len;
Christopher Faulet2d7c5392019-06-03 10:41:26 +0200151 if (type == HTX_BLK_DATA)
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100152 trace_hexdump(v);
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100153 }
154}
155
Christopher Fauletb2e58492019-11-12 11:13:01 +0100156static unsigned int
157trace_get_htx_datalen(struct htx *htx, unsigned int offset, unsigned int len)
158{
159 struct htx_blk *blk;
Christopher Faulet24598a42020-02-26 22:06:11 +0100160 struct htx_ret htxret = htx_find_offset(htx, offset);
161 uint32_t data = 0;
Christopher Fauletb2e58492019-11-12 11:13:01 +0100162
Christopher Faulet24598a42020-02-26 22:06:11 +0100163 blk = htxret.blk;
164 if (blk && htxret.ret && htx_get_blk_type(blk) == HTX_BLK_DATA) {
165 data += htxret.ret;
166 blk = htx_get_next_blk(htx, blk);
167 }
168 while (blk) {
169 if (htx_get_blk_type(blk) == HTX_BLK_UNUSED)
170 goto next;
171 else if (htx_get_blk_type(blk) != HTX_BLK_DATA)
Christopher Fauletb2e58492019-11-12 11:13:01 +0100172 break;
Christopher Faulet24598a42020-02-26 22:06:11 +0100173 data += htx_get_blksz(blk);
174 next:
175 blk = htx_get_next_blk(htx, blk);
Christopher Fauletb2e58492019-11-12 11:13:01 +0100176 }
177 return data;
178}
179
Christopher Faulete6c3b692015-09-02 17:15:16 +0200180/***************************************************************************
181 * Hooks that manage the filter lifecycle (init/check/deinit)
182 **************************************************************************/
183/* Initialize the filter. Returns -1 on error, else 0. */
184static int
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100185trace_init(struct proxy *px, struct flt_conf *fconf)
Christopher Faulete6c3b692015-09-02 17:15:16 +0200186{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100187 struct trace_config *conf = fconf->conf;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200188
189 if (conf->name)
190 memprintf(&conf->name, "%s/%s", conf->name, px->id);
191 else
192 memprintf(&conf->name, "TRACE/%s", px->id);
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100193
Christopher Faulet6e540952018-12-03 22:43:41 +0100194 fconf->flags |= FLT_CFG_FL_HTX;
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100195 fconf->conf = conf;
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100196
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100197 FLT_TRACE(conf, "filter initialized [read random=%s - fwd random=%s - hexdump=%s]",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200198 (conf->rand_parsing ? "true" : "false"),
Christopher Fauletfcd99f82016-10-31 11:27:21 +0100199 (conf->rand_forwarding ? "true" : "false"),
200 (conf->hexdump ? "true" : "false"));
Christopher Faulete6c3b692015-09-02 17:15:16 +0200201 return 0;
202}
203
204/* Free ressources allocated by the trace filter. */
205static void
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100206trace_deinit(struct proxy *px, struct flt_conf *fconf)
Christopher Faulete6c3b692015-09-02 17:15:16 +0200207{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100208 struct trace_config *conf = fconf->conf;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200209
210 if (conf) {
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100211 FLT_TRACE(conf, "filter deinitialized");
Christopher Faulete6c3b692015-09-02 17:15:16 +0200212 free(conf->name);
213 free(conf);
214 }
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100215 fconf->conf = NULL;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200216}
217
218/* Check configuration of a trace filter for a specified proxy.
219 * Return 1 on error, else 0. */
220static int
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100221trace_check(struct proxy *px, struct flt_conf *fconf)
Christopher Faulete6c3b692015-09-02 17:15:16 +0200222{
223 return 0;
224}
225
Christopher Fauletf2273722017-07-27 16:58:42 +0200226/* Initialize the filter for each thread. Return -1 on error, else 0. */
227static int
228trace_init_per_thread(struct proxy *px, struct flt_conf *fconf)
229{
230 struct trace_config *conf = fconf->conf;
231
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100232 FLT_TRACE(conf, "filter initialized for thread tid %u", tid);
Christopher Fauletf2273722017-07-27 16:58:42 +0200233 return 0;
234}
235
236/* Free ressources allocate by the trace filter for each thread. */
237static void
238trace_deinit_per_thread(struct proxy *px, struct flt_conf *fconf)
239{
240 struct trace_config *conf = fconf->conf;
241
242 if (conf)
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100243 FLT_TRACE(conf, "filter deinitialized for thread tid %u", tid);
Christopher Fauletf2273722017-07-27 16:58:42 +0200244}
245
Christopher Faulete6c3b692015-09-02 17:15:16 +0200246/**************************************************************************
247 * Hooks to handle start/stop of streams
248 *************************************************************************/
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200249/* Called when a filter instance is created and attach to a stream */
250static int
251trace_attach(struct stream *s, struct filter *filter)
252{
253 struct trace_config *conf = FLT_CONF(filter);
254
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100255 FLT_STRM_TRACE(conf, s, "%-25s: filter-type=%s",
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200256 __FUNCTION__, filter_type(filter));
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100257
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200258 return 1;
259}
260
261/* Called when a filter instance is detach from a stream, just before its
262 * destruction */
263static void
264trace_detach(struct stream *s, struct filter *filter)
265{
266 struct trace_config *conf = FLT_CONF(filter);
267
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100268 FLT_STRM_TRACE(conf, s, "%-25s: filter-type=%s",
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200269 __FUNCTION__, filter_type(filter));
270}
271
Christopher Faulete6c3b692015-09-02 17:15:16 +0200272/* Called when a stream is created */
273static int
274trace_stream_start(struct stream *s, struct filter *filter)
275{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100276 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200277
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100278 FLT_STRM_TRACE(conf, s, "%-25s",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200279 __FUNCTION__);
280 return 0;
281}
282
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200283
284/* Called when a backend is set for a stream */
285static int
286trace_stream_set_backend(struct stream *s, struct filter *filter,
287 struct proxy *be)
288{
289 struct trace_config *conf = FLT_CONF(filter);
290
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100291 FLT_STRM_TRACE(conf, s, "%-25s: backend=%s",
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200292 __FUNCTION__, be->id);
293 return 0;
294}
295
Christopher Faulete6c3b692015-09-02 17:15:16 +0200296/* Called when a stream is destroyed */
297static void
298trace_stream_stop(struct stream *s, struct filter *filter)
299{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100300 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200301
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100302 FLT_STRM_TRACE(conf, s, "%-25s",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200303 __FUNCTION__);
304}
305
Christopher Fauleta00d8172016-11-10 14:58:05 +0100306/* Called when the stream is woken up because of an expired timer */
307static void
308trace_check_timeouts(struct stream *s, struct filter *filter)
309{
310 struct trace_config *conf = FLT_CONF(filter);
311
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100312 FLT_STRM_TRACE(conf, s, "%-25s",
Christopher Fauleta00d8172016-11-10 14:58:05 +0100313 __FUNCTION__);
314}
315
Christopher Faulete6c3b692015-09-02 17:15:16 +0200316/**************************************************************************
317 * Hooks to handle channels activity
318 *************************************************************************/
319/* Called when analyze starts for a given channel */
320static int
321trace_chn_start_analyze(struct stream *s, struct filter *filter,
322 struct channel *chn)
323{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100324 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200325
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100326 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s)",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200327 __FUNCTION__,
328 channel_label(chn), proxy_mode(s), stream_pos(s));
Christopher Faulet3a394fa2016-05-11 17:13:39 +0200329 filter->pre_analyzers |= (AN_REQ_ALL | AN_RES_ALL);
330 filter->post_analyzers |= (AN_REQ_ALL | AN_RES_ALL);
Christopher Fauletfcd99f82016-10-31 11:27:21 +0100331 register_data_filter(s, chn, filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200332 return 1;
333}
334
335/* Called before a processing happens on a given channel */
336static int
337trace_chn_analyze(struct stream *s, struct filter *filter,
338 struct channel *chn, unsigned an_bit)
339{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100340 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200341 char *ana;
342
343 switch (an_bit) {
344 case AN_REQ_INSPECT_FE:
345 ana = "AN_REQ_INSPECT_FE";
346 break;
347 case AN_REQ_WAIT_HTTP:
348 ana = "AN_REQ_WAIT_HTTP";
349 break;
350 case AN_REQ_HTTP_BODY:
351 ana = "AN_REQ_HTTP_BODY";
352 break;
353 case AN_REQ_HTTP_PROCESS_FE:
354 ana = "AN_REQ_HTTP_PROCESS_FE";
355 break;
356 case AN_REQ_SWITCHING_RULES:
357 ana = "AN_REQ_SWITCHING_RULES";
358 break;
359 case AN_REQ_INSPECT_BE:
360 ana = "AN_REQ_INSPECT_BE";
361 break;
362 case AN_REQ_HTTP_PROCESS_BE:
363 ana = "AN_REQ_HTTP_PROCESS_BE";
364 break;
365 case AN_REQ_SRV_RULES:
366 ana = "AN_REQ_SRV_RULES";
367 break;
368 case AN_REQ_HTTP_INNER:
369 ana = "AN_REQ_HTTP_INNER";
370 break;
371 case AN_REQ_HTTP_TARPIT:
372 ana = "AN_REQ_HTTP_TARPIT";
373 break;
374 case AN_REQ_STICKING_RULES:
375 ana = "AN_REQ_STICKING_RULES";
376 break;
377 case AN_REQ_PRST_RDP_COOKIE:
378 ana = "AN_REQ_PRST_RDP_COOKIE";
379 break;
380 case AN_REQ_HTTP_XFER_BODY:
381 ana = "AN_REQ_HTTP_XFER_BODY";
382 break;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200383 case AN_RES_INSPECT:
384 ana = "AN_RES_INSPECT";
385 break;
386 case AN_RES_WAIT_HTTP:
387 ana = "AN_RES_WAIT_HTTP";
388 break;
389 case AN_RES_HTTP_PROCESS_FE: // AN_RES_HTTP_PROCESS_BE
390 ana = "AN_RES_HTTP_PROCESS_FE/BE";
391 break;
392 case AN_RES_STORE_RULES:
393 ana = "AN_RES_STORE_RULES";
394 break;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200395 case AN_RES_HTTP_XFER_BODY:
396 ana = "AN_RES_HTTP_XFER_BODY";
397 break;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200398 default:
399 ana = "unknown";
400 }
401
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100402 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s) - "
Christopher Faulet3a394fa2016-05-11 17:13:39 +0200403 "analyzer=%s - step=%s",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200404 __FUNCTION__,
405 channel_label(chn), proxy_mode(s), stream_pos(s),
Christopher Faulet3a394fa2016-05-11 17:13:39 +0200406 ana, ((chn->analysers & an_bit) ? "PRE" : "POST"));
Christopher Faulete6c3b692015-09-02 17:15:16 +0200407 return 1;
408}
409
410/* Called when analyze ends for a given channel */
411static int
412trace_chn_end_analyze(struct stream *s, struct filter *filter,
413 struct channel *chn)
414{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100415 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200416
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100417 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s)",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200418 __FUNCTION__,
419 channel_label(chn), proxy_mode(s), stream_pos(s));
420 return 1;
421}
422
423/**************************************************************************
424 * Hooks to filter HTTP messages
425 *************************************************************************/
426static int
Christopher Faulet1339d742016-05-11 16:48:33 +0200427trace_http_headers(struct stream *s, struct filter *filter,
428 struct http_msg *msg)
429{
430 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulet386a0cd2019-07-15 21:22:44 +0200431 struct htx *htx = htxbuf(&msg->chn->buf);
432 struct htx_sl *sl = http_get_stline(htx);
433 int32_t pos;
Christopher Faulet1339d742016-05-11 16:48:33 +0200434
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100435 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s)\t%.*s %.*s %.*s",
Christopher Faulet1339d742016-05-11 16:48:33 +0200436 __FUNCTION__,
Christopher Faulet386a0cd2019-07-15 21:22:44 +0200437 channel_label(msg->chn), proxy_mode(s), stream_pos(s),
438 HTX_SL_P1_LEN(sl), HTX_SL_P1_PTR(sl),
439 HTX_SL_P2_LEN(sl), HTX_SL_P2_PTR(sl),
440 HTX_SL_P3_LEN(sl), HTX_SL_P3_PTR(sl));
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100441
Christopher Faulet386a0cd2019-07-15 21:22:44 +0200442 for (pos = htx_get_first(htx); pos != -1; pos = htx_get_next(htx, pos)) {
443 struct htx_blk *blk = htx_get_blk(htx, pos);
444 enum htx_blk_type type = htx_get_blk_type(blk);
445 struct ist n, v;
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100446
Christopher Faulet386a0cd2019-07-15 21:22:44 +0200447 if (type == HTX_BLK_EOH)
448 break;
449 if (type != HTX_BLK_HDR)
450 continue;
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100451
Christopher Faulet386a0cd2019-07-15 21:22:44 +0200452 n = htx_get_blk_name(htx, blk);
453 v = htx_get_blk_value(htx, blk);
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100454 FLT_STRM_TRACE(conf, s, "\t%.*s: %.*s",
Christopher Faulet386a0cd2019-07-15 21:22:44 +0200455 (int)n.len, n.ptr, (int)v.len, v.ptr);
Christopher Faulet1339d742016-05-11 16:48:33 +0200456 }
Christopher Faulet1339d742016-05-11 16:48:33 +0200457 return 1;
458}
459
460static int
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100461trace_http_payload(struct stream *s, struct filter *filter, struct http_msg *msg,
462 unsigned int offset, unsigned int len)
463{
464 struct trace_config *conf = FLT_CONF(filter);
465 int ret = len;
466
Christopher Faulet0bdeeaa2019-06-04 22:09:53 +0200467 if (ret && conf->rand_forwarding) {
Christopher Fauletb2e58492019-11-12 11:13:01 +0100468 unsigned int data = trace_get_htx_datalen(htxbuf(&msg->chn->buf), offset, len);
Christopher Faulet0bdeeaa2019-06-04 22:09:53 +0200469
Christopher Fauletb2e58492019-11-12 11:13:01 +0100470 if (data) {
Willy Tarreau52bf8392020-03-08 00:42:37 +0100471 ret = ha_random() % (ret+1);
Christopher Faulet647fe1d2019-06-12 16:07:48 +0200472 if (!ret || ret >= data)
473 ret = len;
474 }
Christopher Faulet0bdeeaa2019-06-04 22:09:53 +0200475 }
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100476
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100477 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s) - "
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100478 "offset=%u - len=%u - forward=%d",
479 __FUNCTION__,
480 channel_label(msg->chn), proxy_mode(s), stream_pos(s),
481 offset, len, ret);
482
483 if (conf->hexdump)
Christopher Fauletb2e58492019-11-12 11:13:01 +0100484 trace_htx_hexdump(htxbuf(&msg->chn->buf), offset, ret);
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100485
486 if (ret != len)
487 task_wakeup(s->task, TASK_WOKEN_MSG);
488 return ret;
489}
490
491static int
Christopher Faulete6c3b692015-09-02 17:15:16 +0200492trace_http_end(struct stream *s, struct filter *filter,
493 struct http_msg *msg)
494{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100495 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200496
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100497 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s)",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200498 __FUNCTION__,
499 channel_label(msg->chn), proxy_mode(s), stream_pos(s));
500 return 1;
501}
502
503static void
504trace_http_reset(struct stream *s, struct filter *filter,
505 struct http_msg *msg)
506{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100507 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200508
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100509 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s)",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200510 __FUNCTION__,
511 channel_label(msg->chn), proxy_mode(s), stream_pos(s));
512}
513
514static void
515trace_http_reply(struct stream *s, struct filter *filter, short status,
Willy Tarreau83061a82018-07-13 11:56:34 +0200516 const struct buffer *msg)
Christopher Faulete6c3b692015-09-02 17:15:16 +0200517{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100518 struct trace_config *conf = FLT_CONF(filter);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200519
Christopher Fauleta3ed2712019-11-04 11:35:42 +0100520 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s)",
Christopher Faulete6c3b692015-09-02 17:15:16 +0200521 __FUNCTION__, "-", proxy_mode(s), stream_pos(s));
522}
523
Christopher Faulete6c3b692015-09-02 17:15:16 +0200524/**************************************************************************
525 * Hooks to filter TCP data
526 *************************************************************************/
527static int
Christopher Fauletb2e58492019-11-12 11:13:01 +0100528trace_tcp_payload(struct stream *s, struct filter *filter, struct channel *chn,
529 unsigned int offset, unsigned int len)
Christopher Faulete6c3b692015-09-02 17:15:16 +0200530{
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100531 struct trace_config *conf = FLT_CONF(filter);
Christopher Fauletb2e58492019-11-12 11:13:01 +0100532 int ret = len;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200533
Christopher Fauletb2e58492019-11-12 11:13:01 +0100534 if (s->flags & SF_HTX) {
535 if (ret && conf->rand_forwarding) {
536 unsigned int data = trace_get_htx_datalen(htxbuf(&chn->buf), offset, len);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200537
Christopher Fauletb2e58492019-11-12 11:13:01 +0100538 if (data) {
Willy Tarreau52bf8392020-03-08 00:42:37 +0100539 ret = ha_random() % (ret+1);
Christopher Fauletb2e58492019-11-12 11:13:01 +0100540 if (!ret || ret >= data)
541 ret = len;
542 }
543 }
Christopher Faulete6c3b692015-09-02 17:15:16 +0200544
Christopher Fauletb2e58492019-11-12 11:13:01 +0100545 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s) - "
546 "offset=%u - len=%u - forward=%d",
547 __FUNCTION__,
548 channel_label(chn), proxy_mode(s), stream_pos(s),
549 offset, len, ret);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200550
Christopher Fauletb2e58492019-11-12 11:13:01 +0100551 if (conf->hexdump)
552 trace_htx_hexdump(htxbuf(&chn->buf), offset, ret);
553 }
554 else {
Christopher Faulete6c3b692015-09-02 17:15:16 +0200555
Christopher Fauletb2e58492019-11-12 11:13:01 +0100556 if (ret && conf->rand_forwarding)
Willy Tarreau52bf8392020-03-08 00:42:37 +0100557 ret = ha_random() % (ret+1);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200558
Christopher Fauletb2e58492019-11-12 11:13:01 +0100559 FLT_STRM_TRACE(conf, s, "%-25s: channel=%-10s - mode=%-5s (%s) - "
560 "offset=%u - len=%u - forward=%d",
561 __FUNCTION__,
562 channel_label(chn), proxy_mode(s), stream_pos(s),
563 offset, len, ret);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200564
Christopher Fauletb2e58492019-11-12 11:13:01 +0100565 if (conf->hexdump)
566 trace_raw_hexdump(&chn->buf, offset, ret);
Christopher Fauletfcd99f82016-10-31 11:27:21 +0100567 }
568
Christopher Fauletb2e58492019-11-12 11:13:01 +0100569 if (ret != len)
570 task_wakeup(s->task, TASK_WOKEN_MSG);
Christopher Faulete6c3b692015-09-02 17:15:16 +0200571 return ret;
572}
Christopher Faulete6c3b692015-09-02 17:15:16 +0200573/********************************************************************
574 * Functions that manage the filter initialization
575 ********************************************************************/
576struct flt_ops trace_ops = {
577 /* Manage trace filter, called for each filter declaration */
Christopher Fauletf2273722017-07-27 16:58:42 +0200578 .init = trace_init,
579 .deinit = trace_deinit,
580 .check = trace_check,
581 .init_per_thread = trace_init_per_thread,
582 .deinit_per_thread = trace_deinit_per_thread,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200583
584 /* Handle start/stop of streams */
Christopher Faulet31ed32d2016-06-21 11:42:37 +0200585 .attach = trace_attach,
586 .detach = trace_detach,
587 .stream_start = trace_stream_start,
588 .stream_set_backend = trace_stream_set_backend,
589 .stream_stop = trace_stream_stop,
Christopher Fauleta00d8172016-11-10 14:58:05 +0100590 .check_timeouts = trace_check_timeouts,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200591
592 /* Handle channels activity */
593 .channel_start_analyze = trace_chn_start_analyze,
Christopher Faulet3a394fa2016-05-11 17:13:39 +0200594 .channel_pre_analyze = trace_chn_analyze,
595 .channel_post_analyze = trace_chn_analyze,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200596 .channel_end_analyze = trace_chn_end_analyze,
597
598 /* Filter HTTP requests and responses */
Christopher Faulet1339d742016-05-11 16:48:33 +0200599 .http_headers = trace_http_headers,
Christopher Faulete0aa6f72018-11-30 22:23:32 +0100600 .http_payload = trace_http_payload,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200601 .http_end = trace_http_end,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200602 .http_reset = trace_http_reset,
603 .http_reply = trace_http_reply,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200604
605 /* Filter TCP data */
Christopher Fauletb2e58492019-11-12 11:13:01 +0100606 .tcp_payload = trace_tcp_payload,
Christopher Faulete6c3b692015-09-02 17:15:16 +0200607};
608
609/* Return -1 on error, else 0 */
610static int
611parse_trace_flt(char **args, int *cur_arg, struct proxy *px,
Thierry Fournier3610c392016-04-13 18:27:51 +0200612 struct flt_conf *fconf, char **err, void *private)
Christopher Faulete6c3b692015-09-02 17:15:16 +0200613{
614 struct trace_config *conf;
615 int pos = *cur_arg;
616
617 conf = calloc(1, sizeof(*conf));
618 if (!conf) {
619 memprintf(err, "%s: out of memory", args[*cur_arg]);
620 return -1;
621 }
622 conf->proxy = px;
623
624 if (!strcmp(args[pos], "trace")) {
625 pos++;
626
627 while (*args[pos]) {
628 if (!strcmp(args[pos], "name")) {
629 if (!*args[pos + 1]) {
630 memprintf(err, "'%s' : '%s' option without value",
631 args[*cur_arg], args[pos]);
632 goto error;
633 }
634 conf->name = strdup(args[pos + 1]);
635 if (!conf->name) {
636 memprintf(err, "%s: out of memory", args[*cur_arg]);
637 goto error;
638 }
639 pos++;
640 }
641 else if (!strcmp(args[pos], "random-parsing"))
642 conf->rand_parsing = 1;
643 else if (!strcmp(args[pos], "random-forwarding"))
644 conf->rand_forwarding = 1;
Christopher Fauletfcd99f82016-10-31 11:27:21 +0100645 else if (!strcmp(args[pos], "hexdump"))
646 conf->hexdump = 1;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200647 else
648 break;
649 pos++;
650 }
651 *cur_arg = pos;
Christopher Fauletf4a4ef72018-12-07 17:39:53 +0100652 fconf->id = trace_flt_id;
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100653 fconf->ops = &trace_ops;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200654 }
655
Christopher Faulet443ea1a2016-02-04 13:40:26 +0100656 fconf->conf = conf;
Christopher Faulete6c3b692015-09-02 17:15:16 +0200657 return 0;
658
659 error:
660 if (conf->name)
661 free(conf->name);
662 free(conf);
663 return -1;
664}
665
666/* Declare the filter parser for "trace" keyword */
667static struct flt_kw_list flt_kws = { "TRACE", { }, {
Thierry Fournier3610c392016-04-13 18:27:51 +0200668 { "trace", parse_trace_flt, NULL },
669 { NULL, NULL, NULL },
Christopher Faulete6c3b692015-09-02 17:15:16 +0200670 }
671};
672
Willy Tarreau0108d902018-11-25 19:14:37 +0100673INITCALL1(STG_REGISTER, flt_register_keywords, &flt_kws);