blob: 8584dbe8e2a6d1ba0fe094f7671d7ff5edbdb6c8 [file] [log] [blame]
Emeric Brun107ca302010-01-04 16:16:05 +01001/*
Willy Tarreaucd3b0942012-04-27 21:52:18 +02002 * Sample management functions.
Emeric Brun107ca302010-01-04 16:16:05 +01003 *
4 * Copyright 2009-2010 EXCELIANCE, Emeric Brun <ebrun@exceliance.fr>
Willy Tarreaucd3b0942012-04-27 21:52:18 +02005 * Copyright (C) 2012 Willy Tarreau <w@1wt.eu>
Emeric Brun107ca302010-01-04 16:16:05 +01006 *
7 * This program is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version
10 * 2 of the License, or (at your option) any later version.
11 *
12 */
13
14#include <string.h>
15#include <arpa/inet.h>
Willy Tarreaudd2f85e2012-09-02 22:34:23 +020016#include <stdio.h>
Emeric Brun107ca302010-01-04 16:16:05 +010017
Willy Tarreau7e2c6472012-10-29 20:44:36 +010018#include <types/global.h>
19
Willy Tarreauc7e42382012-08-24 19:22:53 +020020#include <common/chunk.h>
Willy Tarreaucd3b0942012-04-27 21:52:18 +020021#include <common/standard.h>
Willy Tarreaua4312fa2013-04-02 16:34:32 +020022#include <common/uri_auth.h>
Willy Tarreaucd3b0942012-04-27 21:52:18 +020023
Willy Tarreau9fcb9842012-04-20 14:45:49 +020024#include <proto/arg.h>
Willy Tarreaua4312fa2013-04-02 16:34:32 +020025#include <proto/auth.h>
26#include <proto/log.h>
27#include <proto/proxy.h>
Willy Tarreaucd3b0942012-04-27 21:52:18 +020028#include <proto/sample.h>
Willy Tarreaua4312fa2013-04-02 16:34:32 +020029#include <proto/stick_table.h>
Emeric Brun107ca302010-01-04 16:16:05 +010030
Willy Tarreau12785782012-04-27 21:37:17 +020031/* static sample used in sample_process() when <p> is NULL */
Willy Tarreaub4a88f02012-04-23 21:35:11 +020032static struct sample temp_smp;
Emeric Brun107ca302010-01-04 16:16:05 +010033
Willy Tarreau12785782012-04-27 21:37:17 +020034/* list head of all known sample fetch keywords */
35static struct sample_fetch_kw_list sample_fetches = {
36 .list = LIST_HEAD_INIT(sample_fetches.list)
Emeric Brun107ca302010-01-04 16:16:05 +010037};
38
Willy Tarreau12785782012-04-27 21:37:17 +020039/* list head of all known sample format conversion keywords */
40static struct sample_conv_kw_list sample_convs = {
41 .list = LIST_HEAD_INIT(sample_convs.list)
Emeric Brun107ca302010-01-04 16:16:05 +010042};
43
Willy Tarreau80aca902013-01-07 15:42:20 +010044const unsigned int fetch_cap[SMP_SRC_ENTRIES] = {
45 [SMP_SRC_INTRN] = (SMP_VAL_FE_CON_ACC | SMP_VAL_FE_SES_ACC | SMP_VAL_FE_REQ_CNT |
46 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
47 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
48 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
49 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
50 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
51 SMP_VAL_FE_LOG_END),
52
53 [SMP_SRC_LISTN] = (SMP_VAL_FE_CON_ACC | SMP_VAL_FE_SES_ACC | SMP_VAL_FE_REQ_CNT |
54 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
55 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
56 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
57 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
58 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
59 SMP_VAL_FE_LOG_END),
60
61 [SMP_SRC_FTEND] = (SMP_VAL_FE_CON_ACC | SMP_VAL_FE_SES_ACC | SMP_VAL_FE_REQ_CNT |
62 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
63 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
64 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
65 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
66 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
67 SMP_VAL_FE_LOG_END),
68
69 [SMP_SRC_L4CLI] = (SMP_VAL_FE_CON_ACC | SMP_VAL_FE_SES_ACC | SMP_VAL_FE_REQ_CNT |
70 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
71 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
72 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
73 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
74 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
75 SMP_VAL_FE_LOG_END),
76
77 [SMP_SRC_L5CLI] = (SMP_VAL___________ | SMP_VAL_FE_SES_ACC | SMP_VAL_FE_REQ_CNT |
78 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
79 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
80 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
81 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
82 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
83 SMP_VAL_FE_LOG_END),
84
85 [SMP_SRC_TRACK] = (SMP_VAL_FE_CON_ACC | SMP_VAL_FE_SES_ACC | SMP_VAL_FE_REQ_CNT |
86 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
87 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
88 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
89 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
90 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
91 SMP_VAL_FE_LOG_END),
92
93 [SMP_SRC_L6REQ] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_FE_REQ_CNT |
94 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
95 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
96 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL___________ |
97 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
98 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
99 SMP_VAL___________),
100
101 [SMP_SRC_HRQHV] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_FE_REQ_CNT |
102 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
103 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
104 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL___________ |
105 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
106 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
107 SMP_VAL___________),
108
109 [SMP_SRC_HRQHP] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_FE_REQ_CNT |
110 SMP_VAL_FE_HRQ_HDR | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
111 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
112 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
113 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
114 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
115 SMP_VAL_FE_LOG_END),
116
117 [SMP_SRC_HRQBO] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
118 SMP_VAL___________ | SMP_VAL_FE_HRQ_BDY | SMP_VAL_FE_SET_BCK |
119 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
120 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL___________ |
121 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
122 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
123 SMP_VAL___________),
124
125 [SMP_SRC_BKEND] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
126 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
127 SMP_VAL_BE_REQ_CNT | SMP_VAL_BE_HRQ_HDR | SMP_VAL_BE_HRQ_BDY |
128 SMP_VAL_BE_SET_SRV | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
129 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
130 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
131 SMP_VAL_FE_LOG_END),
132
133 [SMP_SRC_SERVR] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
134 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
135 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
136 SMP_VAL___________ | SMP_VAL_BE_SRV_CON | SMP_VAL_BE_RES_CNT |
137 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
138 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
139 SMP_VAL_FE_LOG_END),
140
141 [SMP_SRC_L4SRV] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
142 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
143 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
144 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_BE_RES_CNT |
145 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
146 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
147 SMP_VAL_FE_LOG_END),
148
149 [SMP_SRC_L5SRV] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
150 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
151 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
152 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_BE_RES_CNT |
153 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
154 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
155 SMP_VAL_FE_LOG_END),
156
157 [SMP_SRC_L6RES] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
158 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
159 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
160 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_BE_RES_CNT |
161 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
162 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
163 SMP_VAL___________),
164
165 [SMP_SRC_HRSHV] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
166 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
167 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
168 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_BE_RES_CNT |
169 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
170 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
171 SMP_VAL___________),
172
173 [SMP_SRC_HRSHP] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
174 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
175 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
176 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL_BE_RES_CNT |
177 SMP_VAL_BE_HRS_HDR | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
178 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
179 SMP_VAL_FE_LOG_END),
180
181 [SMP_SRC_HRSBO] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
182 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
183 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
184 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
185 SMP_VAL___________ | SMP_VAL_BE_HRS_BDY | SMP_VAL_BE_STO_RUL |
186 SMP_VAL_FE_RES_CNT | SMP_VAL_FE_HRS_HDR | SMP_VAL_FE_HRS_BDY |
187 SMP_VAL___________),
188
189 [SMP_SRC_RQFIN] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
190 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
191 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
192 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
193 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
194 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
195 SMP_VAL_FE_LOG_END),
196
197 [SMP_SRC_RSFIN] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
198 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
199 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
200 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
201 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
202 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
203 SMP_VAL_FE_LOG_END),
204
205 [SMP_SRC_TXFIN] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
206 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
207 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
208 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
209 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
210 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
211 SMP_VAL_FE_LOG_END),
212
213 [SMP_SRC_SSFIN] = (SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
214 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
215 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
216 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
217 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
218 SMP_VAL___________ | SMP_VAL___________ | SMP_VAL___________ |
219 SMP_VAL_FE_LOG_END),
220};
221
222static const char *fetch_src_names[SMP_SRC_ENTRIES] = {
223 [SMP_SRC_INTRN] = "internal state",
224 [SMP_SRC_LISTN] = "listener",
225 [SMP_SRC_FTEND] = "frontend",
226 [SMP_SRC_L4CLI] = "client address",
227 [SMP_SRC_L5CLI] = "client-side connection",
228 [SMP_SRC_TRACK] = "track counters",
229 [SMP_SRC_L6REQ] = "request buffer",
230 [SMP_SRC_HRQHV] = "HTTP request headers",
231 [SMP_SRC_HRQHP] = "HTTP request",
232 [SMP_SRC_HRQBO] = "HTTP request body",
233 [SMP_SRC_BKEND] = "backend",
234 [SMP_SRC_SERVR] = "server",
235 [SMP_SRC_L4SRV] = "server address",
236 [SMP_SRC_L5SRV] = "server-side connection",
237 [SMP_SRC_L6RES] = "response buffer",
238 [SMP_SRC_HRSHV] = "HTTP response headers",
239 [SMP_SRC_HRSHP] = "HTTP response",
240 [SMP_SRC_HRSBO] = "HTTP response body",
241 [SMP_SRC_RQFIN] = "request buffer statistics",
242 [SMP_SRC_RSFIN] = "response buffer statistics",
243 [SMP_SRC_TXFIN] = "transaction statistics",
244 [SMP_SRC_SSFIN] = "session statistics",
245};
246
Willy Tarreaubf8e2512013-03-25 14:52:41 +0100247static const char *fetch_ckp_names[SMP_CKP_ENTRIES] = {
248 [SMP_CKP_FE_CON_ACC] = "frontend tcp-request connection rule",
249 [SMP_CKP_FE_SES_ACC] = "frontend tcp-request session rule",
250 [SMP_CKP_FE_REQ_CNT] = "frontend tcp-request content rule",
251 [SMP_CKP_FE_HRQ_HDR] = "frontend http-request header rule",
252 [SMP_CKP_FE_HRQ_BDY] = "frontend http-request body rule",
253 [SMP_CKP_FE_SET_BCK] = "frontend use-backend rule",
254 [SMP_CKP_BE_REQ_CNT] = "backend tcp-request content rule",
255 [SMP_CKP_BE_HRQ_HDR] = "backend http-request header rule",
256 [SMP_CKP_BE_HRQ_BDY] = "backend http-request body rule",
257 [SMP_CKP_BE_SET_SRV] = "backend use-server, balance or stick-match rule",
258 [SMP_CKP_BE_SRV_CON] = "server source selection",
259 [SMP_CKP_BE_RES_CNT] = "backend tcp-response content rule",
260 [SMP_CKP_BE_HRS_HDR] = "backend http-response header rule",
261 [SMP_CKP_BE_HRS_BDY] = "backend http-response body rule",
262 [SMP_CKP_BE_STO_RUL] = "backend stick-store rule",
263 [SMP_CKP_FE_RES_CNT] = "frontend tcp-response content rule",
264 [SMP_CKP_FE_HRS_HDR] = "frontend http-response header rule",
265 [SMP_CKP_FE_HRS_BDY] = "frontend http-response body rule",
266 [SMP_CKP_FE_LOG_END] = "logs",
267};
268
Thierry FOURNIER348971e2013-11-21 10:50:10 +0100269/* This function returns the type of the data returned by the sample_expr.
270 * It assumes that the <expr> and all of its converters are properly
271 * initialized.
272 */
273inline
274int smp_expr_output_type(struct sample_expr *expr)
275{
276 struct sample_conv_expr *smp_expr;
277
278 if (!LIST_ISEMPTY(&expr->conv_exprs)) {
279 smp_expr = LIST_PREV(&expr->conv_exprs, struct sample_conv_expr *, list);
280 return smp_expr->conv->out_type;
281 }
282 return expr->fetch->out_type;
283}
284
285
Willy Tarreau80aca902013-01-07 15:42:20 +0100286/* fill the trash with a comma-delimited list of source names for the <use> bit
287 * field which must be composed of a non-null set of SMP_USE_* flags. The return
288 * value is the pointer to the string in the trash buffer.
289 */
290const char *sample_src_names(unsigned int use)
291{
292 int bit;
293
294 trash.len = 0;
295 trash.str[0] = '\0';
296 for (bit = 0; bit < SMP_SRC_ENTRIES; bit++) {
297 if (!(use & ~((1 << bit) - 1)))
298 break; /* no more bits */
299
300 if (!(use & (1 << bit)))
301 continue; /* bit not set */
302
303 trash.len += snprintf(trash.str + trash.len, trash.size - trash.len, "%s%s",
304 (use & ((1 << bit) - 1)) ? "," : "",
305 fetch_src_names[bit]);
306 }
307 return trash.str;
308}
309
Willy Tarreaubf8e2512013-03-25 14:52:41 +0100310/* return a pointer to the correct sample checkpoint name, or "unknown" when
311 * the flags are invalid. Only the lowest bit is used, higher bits are ignored
312 * if set.
313 */
314const char *sample_ckp_names(unsigned int use)
315{
316 int bit;
317
318 for (bit = 0; bit < SMP_CKP_ENTRIES; bit++)
319 if (use & (1 << bit))
320 return fetch_ckp_names[bit];
321 return "unknown sample check place, please report this bug";
322}
323
Emeric Brun107ca302010-01-04 16:16:05 +0100324/*
Willy Tarreau80aca902013-01-07 15:42:20 +0100325 * Registers the sample fetch keyword list <kwl> as a list of valid keywords
326 * for next parsing sessions. The fetch keywords capabilities are also computed
327 * from their ->use field.
Emeric Brun107ca302010-01-04 16:16:05 +0100328 */
Willy Tarreau80aca902013-01-07 15:42:20 +0100329void sample_register_fetches(struct sample_fetch_kw_list *kwl)
Emeric Brun107ca302010-01-04 16:16:05 +0100330{
Willy Tarreau80aca902013-01-07 15:42:20 +0100331 struct sample_fetch *sf;
332 int bit;
333
334 for (sf = kwl->kw; sf->kw != NULL; sf++) {
335 for (bit = 0; bit < SMP_SRC_ENTRIES; bit++)
336 if (sf->use & (1 << bit))
337 sf->val |= fetch_cap[bit];
338 }
339 LIST_ADDQ(&sample_fetches.list, &kwl->list);
Emeric Brun107ca302010-01-04 16:16:05 +0100340}
341
342/*
Willy Tarreau12785782012-04-27 21:37:17 +0200343 * Registers the sample format coverstion keyword list <pckl> as a list of valid keywords for next
Emeric Brun107ca302010-01-04 16:16:05 +0100344 * parsing sessions.
345 */
Willy Tarreau12785782012-04-27 21:37:17 +0200346void sample_register_convs(struct sample_conv_kw_list *pckl)
Emeric Brun107ca302010-01-04 16:16:05 +0100347{
Willy Tarreau12785782012-04-27 21:37:17 +0200348 LIST_ADDQ(&sample_convs.list, &pckl->list);
Emeric Brun107ca302010-01-04 16:16:05 +0100349}
350
351/*
Willy Tarreau12785782012-04-27 21:37:17 +0200352 * Returns the pointer on sample fetch keyword structure identified by
Emeric Brun107ca302010-01-04 16:16:05 +0100353 * string of <len> in buffer <kw>.
354 *
355 */
Willy Tarreau12785782012-04-27 21:37:17 +0200356struct sample_fetch *find_sample_fetch(const char *kw, int len)
Emeric Brun107ca302010-01-04 16:16:05 +0100357{
358 int index;
Willy Tarreau12785782012-04-27 21:37:17 +0200359 struct sample_fetch_kw_list *kwl;
Emeric Brun107ca302010-01-04 16:16:05 +0100360
Willy Tarreau12785782012-04-27 21:37:17 +0200361 list_for_each_entry(kwl, &sample_fetches.list, list) {
Emeric Brun107ca302010-01-04 16:16:05 +0100362 for (index = 0; kwl->kw[index].kw != NULL; index++) {
363 if (strncmp(kwl->kw[index].kw, kw, len) == 0 &&
364 kwl->kw[index].kw[len] == '\0')
365 return &kwl->kw[index];
366 }
367 }
368 return NULL;
369}
370
371/*
Willy Tarreau12785782012-04-27 21:37:17 +0200372 * Returns the pointer on sample format conversion keyword structure identified by
Emeric Brun107ca302010-01-04 16:16:05 +0100373 * string of <len> in buffer <kw>.
374 *
375 */
Willy Tarreau12785782012-04-27 21:37:17 +0200376struct sample_conv *find_sample_conv(const char *kw, int len)
Emeric Brun107ca302010-01-04 16:16:05 +0100377{
378 int index;
Willy Tarreau12785782012-04-27 21:37:17 +0200379 struct sample_conv_kw_list *kwl;
Emeric Brun107ca302010-01-04 16:16:05 +0100380
Willy Tarreau12785782012-04-27 21:37:17 +0200381 list_for_each_entry(kwl, &sample_convs.list, list) {
Emeric Brun107ca302010-01-04 16:16:05 +0100382 for (index = 0; kwl->kw[index].kw != NULL; index++) {
383 if (strncmp(kwl->kw[index].kw, kw, len) == 0 &&
384 kwl->kw[index].kw[len] == '\0')
385 return &kwl->kw[index];
386 }
387 }
388 return NULL;
389}
390
Emeric Brun107ca302010-01-04 16:16:05 +0100391/******************************************************************/
Willy Tarreau12785782012-04-27 21:37:17 +0200392/* Sample casts functions */
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +0200393/* Note: these functions do *NOT* set the output type on the */
394/* sample, the caller is responsible for doing this on return. */
Emeric Brun107ca302010-01-04 16:16:05 +0100395/******************************************************************/
396
Willy Tarreau342acb42012-04-23 22:03:39 +0200397static int c_ip2int(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100398{
Willy Tarreau342acb42012-04-23 22:03:39 +0200399 smp->data.uint = ntohl(smp->data.ipv4.s_addr);
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100400 smp->type = SMP_T_UINT;
Emeric Brun107ca302010-01-04 16:16:05 +0100401 return 1;
402}
403
Willy Tarreau342acb42012-04-23 22:03:39 +0200404static int c_ip2str(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100405{
Willy Tarreau47ca5452012-12-23 20:22:19 +0100406 struct chunk *trash = get_trash_chunk();
Emeric Brun107ca302010-01-04 16:16:05 +0100407
Willy Tarreau342acb42012-04-23 22:03:39 +0200408 if (!inet_ntop(AF_INET, (void *)&smp->data.ipv4, trash->str, trash->size))
Emeric Brun107ca302010-01-04 16:16:05 +0100409 return 0;
410
411 trash->len = strlen(trash->str);
Willy Tarreau342acb42012-04-23 22:03:39 +0200412 smp->data.str = *trash;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100413 smp->type = SMP_T_STR;
Emeric Brun107ca302010-01-04 16:16:05 +0100414
415 return 1;
416}
417
Willy Tarreau342acb42012-04-23 22:03:39 +0200418static int c_ip2ipv6(struct sample *smp)
David du Colombier4f92d322011-03-24 11:09:31 +0100419{
Willy Tarreau342acb42012-04-23 22:03:39 +0200420 v4tov6(&smp->data.ipv6, &smp->data.ipv4);
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100421 smp->type = SMP_T_IPV6;
David du Colombier4f92d322011-03-24 11:09:31 +0100422 return 1;
423}
424
Willy Tarreau342acb42012-04-23 22:03:39 +0200425static int c_ipv62str(struct sample *smp)
David du Colombier4f92d322011-03-24 11:09:31 +0100426{
Willy Tarreau47ca5452012-12-23 20:22:19 +0100427 struct chunk *trash = get_trash_chunk();
David du Colombier4f92d322011-03-24 11:09:31 +0100428
Willy Tarreau342acb42012-04-23 22:03:39 +0200429 if (!inet_ntop(AF_INET6, (void *)&smp->data.ipv6, trash->str, trash->size))
David du Colombier4f92d322011-03-24 11:09:31 +0100430 return 0;
431
432 trash->len = strlen(trash->str);
Willy Tarreau342acb42012-04-23 22:03:39 +0200433 smp->data.str = *trash;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100434 smp->type = SMP_T_STR;
David du Colombier4f92d322011-03-24 11:09:31 +0100435 return 1;
436}
437
438/*
Willy Tarreau342acb42012-04-23 22:03:39 +0200439static int c_ipv62ip(struct sample *smp)
David du Colombier4f92d322011-03-24 11:09:31 +0100440{
Willy Tarreau342acb42012-04-23 22:03:39 +0200441 return v6tov4(&smp->data.ipv4, &smp->data.ipv6);
David du Colombier4f92d322011-03-24 11:09:31 +0100442}
443*/
444
Willy Tarreau342acb42012-04-23 22:03:39 +0200445static int c_int2ip(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100446{
Willy Tarreau342acb42012-04-23 22:03:39 +0200447 smp->data.ipv4.s_addr = htonl(smp->data.uint);
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100448 smp->type = SMP_T_IPV4;
Emeric Brun107ca302010-01-04 16:16:05 +0100449 return 1;
450}
451
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100452static int c_str2addr(struct sample *smp)
453{
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100454 if (!buf2ip(smp->data.str.str, smp->data.str.len, &smp->data.ipv4)) {
Thierry FOURNIERfd139902013-12-11 12:38:57 +0100455 if (!buf2ip6(smp->data.str.str, smp->data.str.len, &smp->data.ipv6))
456 return 0;
457 smp->type = SMP_T_IPV6;
458 return 1;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100459 }
460 smp->type = SMP_T_IPV4;
461 return 1;
462}
463
Willy Tarreau342acb42012-04-23 22:03:39 +0200464static int c_str2ip(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100465{
Willy Tarreau342acb42012-04-23 22:03:39 +0200466 if (!buf2ip(smp->data.str.str, smp->data.str.len, &smp->data.ipv4))
Emeric Brun107ca302010-01-04 16:16:05 +0100467 return 0;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100468 smp->type = SMP_T_IPV4;
Emeric Brun107ca302010-01-04 16:16:05 +0100469 return 1;
470}
471
Willy Tarreau342acb42012-04-23 22:03:39 +0200472static int c_str2ipv6(struct sample *smp)
David du Colombier4f92d322011-03-24 11:09:31 +0100473{
Thierry FOURNIERfd139902013-12-11 12:38:57 +0100474 if (!buf2ip6(smp->data.str.str, smp->data.str.len, &smp->data.ipv6))
475 return 0;
476 smp->type = SMP_T_IPV6;
477 return 1;
David du Colombier4f92d322011-03-24 11:09:31 +0100478}
479
Thierry FOURNIERe87cac12014-03-12 15:07:59 +0100480/* The sample is always copied into a new one so that smp->size is always
481 * valid. The NULL char always enforces the end of string if it is met.
482 */
Emeric Brun8ac33d92012-10-17 13:36:06 +0200483static int c_bin2str(struct sample *smp)
484{
Willy Tarreau47ca5452012-12-23 20:22:19 +0100485 struct chunk *trash = get_trash_chunk();
Emeric Brun8ac33d92012-10-17 13:36:06 +0200486 unsigned char c;
487 int ptr = 0;
488
Thierry FOURNIERe87cac12014-03-12 15:07:59 +0100489 while (ptr < smp->data.str.len) {
490 c = smp->data.str.str[ptr];
491 if (!c)
492 break;
493 trash->str[ptr] = c;
494 ptr++;
Emeric Brun8ac33d92012-10-17 13:36:06 +0200495 }
Thierry FOURNIERe87cac12014-03-12 15:07:59 +0100496 trash->len = ptr;
497 trash->str[ptr] = 0;
Emeric Brun8ac33d92012-10-17 13:36:06 +0200498 smp->data.str = *trash;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100499 smp->type = SMP_T_STR;
Emeric Brun8ac33d92012-10-17 13:36:06 +0200500 return 1;
501}
502
Willy Tarreau342acb42012-04-23 22:03:39 +0200503static int c_int2str(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100504{
Willy Tarreau47ca5452012-12-23 20:22:19 +0100505 struct chunk *trash = get_trash_chunk();
Emeric Brun107ca302010-01-04 16:16:05 +0100506 char *pos;
507
Willy Tarreau342acb42012-04-23 22:03:39 +0200508 pos = ultoa_r(smp->data.uint, trash->str, trash->size);
Emeric Brun107ca302010-01-04 16:16:05 +0100509
510 if (!pos)
511 return 0;
512
Emeric Brun485479d2010-09-23 18:02:19 +0200513 trash->size = trash->size - (pos - trash->str);
Emeric Brun107ca302010-01-04 16:16:05 +0100514 trash->str = pos;
515 trash->len = strlen(pos);
Willy Tarreau342acb42012-04-23 22:03:39 +0200516 smp->data.str = *trash;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100517 smp->type = SMP_T_STR;
Emeric Brun107ca302010-01-04 16:16:05 +0100518 return 1;
519}
520
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100521static inline void _c_datadup(struct sample *smp)
Emeric Brun485479d2010-09-23 18:02:19 +0200522{
Willy Tarreau47ca5452012-12-23 20:22:19 +0100523 struct chunk *trash = get_trash_chunk();
Emeric Brun485479d2010-09-23 18:02:19 +0200524
Willy Tarreau342acb42012-04-23 22:03:39 +0200525 trash->len = smp->data.str.len < trash->size ? smp->data.str.len : trash->size;
526 memcpy(trash->str, smp->data.str.str, trash->len);
527 smp->data.str = *trash;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100528}
529
530static int c_datadup(struct sample *smp)
531{
532 _c_datadup(smp);
533 if (smp->type == SMP_T_CSTR)
534 smp->type = SMP_T_STR;
535 else
536 smp->type = SMP_T_BIN;
537 return 1;
538}
539
540static int c_bindup(struct sample *smp)
541{
542 _c_datadup(smp);
543 smp->type = SMP_T_BIN;
Emeric Brun485479d2010-09-23 18:02:19 +0200544 return 1;
545}
546
547
Willy Tarreau342acb42012-04-23 22:03:39 +0200548static int c_none(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100549{
550 return 1;
551}
552
Willy Tarreau342acb42012-04-23 22:03:39 +0200553static int c_str2int(struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +0100554{
555 int i;
556 uint32_t ret = 0;
557
Thierry FOURNIER60bb0202014-01-27 18:20:48 +0100558 if (smp->data.str.len == 0)
559 return 0;
560
Willy Tarreau342acb42012-04-23 22:03:39 +0200561 for (i = 0; i < smp->data.str.len; i++) {
562 uint32_t val = smp->data.str.str[i] - '0';
Emeric Brun107ca302010-01-04 16:16:05 +0100563
Thierry FOURNIER60bb0202014-01-27 18:20:48 +0100564 if (val > 9) {
565 if (i == 0)
566 return 0;
Emeric Brun107ca302010-01-04 16:16:05 +0100567 break;
Thierry FOURNIER60bb0202014-01-27 18:20:48 +0100568 }
Emeric Brun107ca302010-01-04 16:16:05 +0100569
570 ret = ret * 10 + val;
571 }
572
Willy Tarreau342acb42012-04-23 22:03:39 +0200573 smp->data.uint = ret;
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100574 smp->type = SMP_T_UINT;
Emeric Brun107ca302010-01-04 16:16:05 +0100575 return 1;
576}
577
578/*****************************************************************/
Willy Tarreau12785782012-04-27 21:37:17 +0200579/* Sample casts matrix: */
580/* sample_casts[from type][to type] */
581/* NULL pointer used for impossible sample casts */
Emeric Brun107ca302010-01-04 16:16:05 +0100582/*****************************************************************/
Emeric Brun107ca302010-01-04 16:16:05 +0100583
Thierry FOURNIER8af6ff12013-11-21 10:53:12 +0100584sample_cast_fct sample_casts[SMP_TYPES][SMP_TYPES] = {
Thierry FOURNIERb805f712013-11-26 20:47:54 +0100585/* to: BOOL UINT SINT ADDR IPV4 IPV6 STR BIN CSTR CBIN */
586/* from: BOOL */ { c_none, c_none, c_none, NULL, NULL, NULL, c_int2str, NULL, c_int2str, NULL, },
587/* UINT */ { c_none, c_none, c_none, c_int2ip, c_int2ip, NULL, c_int2str, NULL, c_int2str, NULL, },
588/* SINT */ { c_none, c_none, c_none, c_int2ip, c_int2ip, NULL, c_int2str, NULL, c_int2str, NULL, },
589/* ADDR */ { NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, },
590/* IPV4 */ { NULL, c_ip2int, c_ip2int, c_none, c_none, c_ip2ipv6, c_ip2str, NULL, c_ip2str, NULL, },
591/* IPV6 */ { NULL, NULL, NULL, c_none, NULL, c_none, c_ipv62str, NULL, c_ipv62str, NULL, },
592/* STR */ { c_str2int, c_str2int, c_str2int, c_str2addr, c_str2ip, c_str2ipv6, c_none, c_none, c_none, c_none, },
593/* BIN */ { NULL, NULL, NULL, NULL, NULL, NULL, c_bin2str, c_none, c_bin2str, c_none, },
594/* CSTR */ { c_str2int, c_str2int, c_str2int, c_str2addr, c_str2ip, c_str2ipv6, c_datadup, c_bindup, c_none, c_none, },
595/* CBIN */ { NULL, NULL, NULL, NULL, NULL, NULL, c_bin2str, c_datadup, c_bin2str, c_none, },
Willy Tarreauf0b38bf2010-06-06 13:22:23 +0200596};
Emeric Brun107ca302010-01-04 16:16:05 +0100597
Emeric Brun107ca302010-01-04 16:16:05 +0100598/*
Willy Tarreau12785782012-04-27 21:37:17 +0200599 * Parse a sample expression configuration:
Emeric Brun107ca302010-01-04 16:16:05 +0100600 * fetch keyword followed by format conversion keywords.
Willy Tarreau12785782012-04-27 21:37:17 +0200601 * Returns a pointer on allocated sample expression structure.
Willy Tarreaua4312fa2013-04-02 16:34:32 +0200602 * The caller must have set al->ctx.
Emeric Brun107ca302010-01-04 16:16:05 +0100603 */
Willy Tarreau975c1782013-12-12 23:16:54 +0100604struct sample_expr *sample_parse_expr(char **str, int *idx, char **err_msg, struct arg_list *al)
Emeric Brun107ca302010-01-04 16:16:05 +0100605{
Willy Tarreau833cc792013-07-24 15:34:19 +0200606 const char *begw; /* beginning of word */
607 const char *endw; /* end of word */
608 const char *endt; /* end of term */
Willy Tarreau12785782012-04-27 21:37:17 +0200609 struct sample_expr *expr;
610 struct sample_fetch *fetch;
611 struct sample_conv *conv;
Emeric Brun107ca302010-01-04 16:16:05 +0100612 unsigned long prev_type;
Willy Tarreau833cc792013-07-24 15:34:19 +0200613 char *fkw = NULL;
614 char *ckw = NULL;
Willy Tarreau689a1df2013-12-13 00:40:11 +0100615 int err_arg;
Emeric Brun107ca302010-01-04 16:16:05 +0100616
Willy Tarreau833cc792013-07-24 15:34:19 +0200617 begw = str[*idx];
618 for (endw = begw; *endw && *endw != '(' && *endw != ','; endw++);
619
620 if (endw == begw) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100621 memprintf(err_msg, "missing fetch method");
Emeric Brun107ca302010-01-04 16:16:05 +0100622 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200623 }
Emeric Brun107ca302010-01-04 16:16:05 +0100624
Willy Tarreau833cc792013-07-24 15:34:19 +0200625 /* keep a copy of the current fetch keyword for error reporting */
626 fkw = my_strndup(begw, endw - begw);
Emeric Brun107ca302010-01-04 16:16:05 +0100627
Willy Tarreau833cc792013-07-24 15:34:19 +0200628 fetch = find_sample_fetch(begw, endw - begw);
629 if (!fetch) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100630 memprintf(err_msg, "unknown fetch method '%s'", fkw);
Emeric Brun107ca302010-01-04 16:16:05 +0100631 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200632 }
Emeric Brun107ca302010-01-04 16:16:05 +0100633
Willy Tarreau833cc792013-07-24 15:34:19 +0200634 endt = endw;
635 if (*endt == '(') {
Willy Tarreau689a1df2013-12-13 00:40:11 +0100636 /* look for the end of this term and skip the opening parenthesis */
637 endt = ++endw;
Willy Tarreau833cc792013-07-24 15:34:19 +0200638 while (*endt && *endt != ')')
639 endt++;
640 if (*endt != ')') {
Willy Tarreau975c1782013-12-12 23:16:54 +0100641 memprintf(err_msg, "missing closing ')' after arguments to fetch keyword '%s'", fkw);
Willy Tarreau833cc792013-07-24 15:34:19 +0200642 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200643 }
Emeric Brun485479d2010-09-23 18:02:19 +0200644 }
Emeric Brun107ca302010-01-04 16:16:05 +0100645
Willy Tarreau833cc792013-07-24 15:34:19 +0200646 /* At this point, we have :
647 * - begw : beginning of the keyword
Willy Tarreau689a1df2013-12-13 00:40:11 +0100648 * - endw : end of the keyword, first character not part of keyword
649 * nor the opening parenthesis (so first character of args
650 * if present).
Willy Tarreau833cc792013-07-24 15:34:19 +0200651 * - endt : end of the term (=endw or last parenthesis if args are present)
652 */
653
654 if (fetch->out_type >= SMP_TYPES) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100655 memprintf(err_msg, "returns type of fetch method '%s' is unknown", fkw);
Emeric Brun107ca302010-01-04 16:16:05 +0100656 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200657 }
Emeric Brun107ca302010-01-04 16:16:05 +0100658 prev_type = fetch->out_type;
Willy Tarreau833cc792013-07-24 15:34:19 +0200659
Willy Tarreau12785782012-04-27 21:37:17 +0200660 expr = calloc(1, sizeof(struct sample_expr));
Emeric Brun485479d2010-09-23 18:02:19 +0200661 if (!expr)
662 goto out_error;
Emeric Brun107ca302010-01-04 16:16:05 +0100663
664 LIST_INIT(&(expr->conv_exprs));
665 expr->fetch = fetch;
Willy Tarreau2e845be2012-10-19 19:49:09 +0200666 expr->arg_p = empty_arg_list;
Emeric Brun107ca302010-01-04 16:16:05 +0100667
Willy Tarreau689a1df2013-12-13 00:40:11 +0100668 /* Note that we call the argument parser even with an empty string,
669 * this allows it to automatically create entries for mandatory
670 * implicit arguments (eg: local proxy name).
671 */
672 al->kw = expr->fetch->kw;
673 al->conv = NULL;
674 if (make_arg_list(endw, endt - endw, fetch->arg_mask, &expr->arg_p, err_msg, NULL, &err_arg, al) < 0) {
675 memprintf(err_msg, "fetch method '%s' : %s", fkw, *err_msg);
676 goto out_error;
677 }
Willy Tarreau2e845be2012-10-19 19:49:09 +0200678
Willy Tarreau689a1df2013-12-13 00:40:11 +0100679 if (!expr->arg_p) {
680 expr->arg_p = empty_arg_list;
Emeric Brun485479d2010-09-23 18:02:19 +0200681 }
Willy Tarreau689a1df2013-12-13 00:40:11 +0100682 else if (fetch->val_args && !fetch->val_args(expr->arg_p, err_msg)) {
683 memprintf(err_msg, "invalid args in fetch method '%s' : %s", fkw, *err_msg);
Emeric Brun485479d2010-09-23 18:02:19 +0200684 goto out_error;
Emeric Brun107ca302010-01-04 16:16:05 +0100685 }
686
Willy Tarreau833cc792013-07-24 15:34:19 +0200687 /* Now process the converters if any. We have two supported syntaxes
688 * for the converters, which can be combined :
689 * - comma-delimited list of converters just after the keyword and args ;
690 * - one converter per keyword
691 * The combination allows to have each keyword being a comma-delimited
692 * series of converters.
693 *
694 * We want to process the former first, then the latter. For this we start
695 * from the beginning of the supposed place in the exiting conv chain, which
696 * starts at the last comma (endt).
697 */
698
699 while (1) {
Willy Tarreau12785782012-04-27 21:37:17 +0200700 struct sample_conv_expr *conv_expr;
Emeric Brun107ca302010-01-04 16:16:05 +0100701
Willy Tarreau833cc792013-07-24 15:34:19 +0200702 if (*endt == ')') /* skip last closing parenthesis */
703 endt++;
Emeric Brun107ca302010-01-04 16:16:05 +0100704
Willy Tarreau833cc792013-07-24 15:34:19 +0200705 if (*endt && *endt != ',') {
706 if (ckw)
Willy Tarreau975c1782013-12-12 23:16:54 +0100707 memprintf(err_msg, "missing comma after conv keyword '%s'", ckw);
Willy Tarreau833cc792013-07-24 15:34:19 +0200708 else
Willy Tarreau975c1782013-12-12 23:16:54 +0100709 memprintf(err_msg, "missing comma after fetch keyword '%s'", fkw);
Emeric Brun107ca302010-01-04 16:16:05 +0100710 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200711 }
Emeric Brun107ca302010-01-04 16:16:05 +0100712
Willy Tarreau833cc792013-07-24 15:34:19 +0200713 while (*endt == ',') /* then trailing commas */
714 endt++;
715
716 begw = endt; /* start of conv keyword */
717
718 if (!*begw) {
719 /* none ? skip to next string */
720 (*idx)++;
721 begw = str[*idx];
722 if (!begw || !*begw)
723 break;
724 }
725
726 for (endw = begw; *endw && *endw != '(' && *endw != ','; endw++);
727
728 free(ckw);
729 ckw = my_strndup(begw, endw - begw);
730
731 conv = find_sample_conv(begw, endw - begw);
732 if (!conv) {
733 /* we found an isolated keyword that we don't know, it's not ours */
734 if (begw == str[*idx])
735 break;
Willy Tarreau975c1782013-12-12 23:16:54 +0100736 memprintf(err_msg, "unknown conv method '%s'", ckw);
Willy Tarreau833cc792013-07-24 15:34:19 +0200737 goto out_error;
738 }
Emeric Brun107ca302010-01-04 16:16:05 +0100739
Willy Tarreau833cc792013-07-24 15:34:19 +0200740 endt = endw;
741 if (*endt == '(') {
742 /* look for the end of this term */
743 while (*endt && *endt != ')')
744 endt++;
745 if (*endt != ')') {
Willy Tarreau975c1782013-12-12 23:16:54 +0100746 memprintf(err_msg, "syntax error: missing ')' after conv keyword '%s'", ckw);
Willy Tarreau833cc792013-07-24 15:34:19 +0200747 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200748 }
Willy Tarreau833cc792013-07-24 15:34:19 +0200749 }
750
751 if (conv->in_type >= SMP_TYPES || conv->out_type >= SMP_TYPES) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100752 memprintf(err_msg, "returns type of conv method '%s' is unknown", ckw);
Emeric Brun107ca302010-01-04 16:16:05 +0100753 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200754 }
Emeric Brun107ca302010-01-04 16:16:05 +0100755
756 /* If impossible type conversion */
Willy Tarreau12785782012-04-27 21:37:17 +0200757 if (!sample_casts[prev_type][conv->in_type]) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100758 memprintf(err_msg, "conv method '%s' cannot be applied", ckw);
Emeric Brun107ca302010-01-04 16:16:05 +0100759 goto out_error;
Emeric Brun485479d2010-09-23 18:02:19 +0200760 }
Emeric Brun107ca302010-01-04 16:16:05 +0100761
762 prev_type = conv->out_type;
Willy Tarreau12785782012-04-27 21:37:17 +0200763 conv_expr = calloc(1, sizeof(struct sample_conv_expr));
Emeric Brun485479d2010-09-23 18:02:19 +0200764 if (!conv_expr)
765 goto out_error;
Emeric Brun107ca302010-01-04 16:16:05 +0100766
767 LIST_ADDQ(&(expr->conv_exprs), &(conv_expr->list));
768 conv_expr->conv = conv;
769
Willy Tarreau833cc792013-07-24 15:34:19 +0200770 if (endt != endw) {
Willy Tarreaub27c0d32012-04-20 16:04:47 +0200771 int err_arg;
Willy Tarreau21d68a62012-04-20 15:52:36 +0200772
Willy Tarreau9fcb9842012-04-20 14:45:49 +0200773 if (!conv->arg_mask) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100774 memprintf(err_msg, "conv method '%s' does not support any args", ckw);
Emeric Brun485479d2010-09-23 18:02:19 +0200775 goto out_error;
776 }
Willy Tarreau9e92d322010-01-26 17:58:06 +0100777
Willy Tarreaua4312fa2013-04-02 16:34:32 +0200778 al->kw = expr->fetch->kw;
779 al->conv = conv_expr->conv->kw;
Willy Tarreau975c1782013-12-12 23:16:54 +0100780 if (make_arg_list(endw + 1, endt - endw - 1, conv->arg_mask, &conv_expr->arg_p, err_msg, NULL, &err_arg, al) < 0) {
781 memprintf(err_msg, "invalid arg %d in conv method '%s' : %s", err_arg+1, ckw, *err_msg);
Willy Tarreau21d68a62012-04-20 15:52:36 +0200782 goto out_error;
783 }
784
Willy Tarreau2e845be2012-10-19 19:49:09 +0200785 if (!conv_expr->arg_p)
786 conv_expr->arg_p = empty_arg_list;
787
Willy Tarreau975c1782013-12-12 23:16:54 +0100788 if (conv->val_args && !conv->val_args(conv_expr->arg_p, conv, err_msg)) {
789 memprintf(err_msg, "invalid args in conv method '%s' : %s", ckw, *err_msg);
Emeric Brun485479d2010-09-23 18:02:19 +0200790 goto out_error;
791 }
792 }
Willy Tarreau833cc792013-07-24 15:34:19 +0200793 else if (ARGM(conv->arg_mask)) {
Willy Tarreau975c1782013-12-12 23:16:54 +0100794 memprintf(err_msg, "missing args for conv method '%s'", ckw);
Emeric Brun485479d2010-09-23 18:02:19 +0200795 goto out_error;
Emeric Brun107ca302010-01-04 16:16:05 +0100796 }
797 }
Emeric Brun485479d2010-09-23 18:02:19 +0200798
Willy Tarreau833cc792013-07-24 15:34:19 +0200799 out:
800 free(fkw);
801 free(ckw);
Emeric Brun107ca302010-01-04 16:16:05 +0100802 return expr;
803
804out_error:
Willy Tarreau12785782012-04-27 21:37:17 +0200805 /* TODO: prune_sample_expr(expr); */
Willy Tarreau833cc792013-07-24 15:34:19 +0200806 expr = NULL;
807 goto out;
Emeric Brun107ca302010-01-04 16:16:05 +0100808}
809
810/*
Willy Tarreau12785782012-04-27 21:37:17 +0200811 * Process a fetch + format conversion of defined by the sample expression <expr>
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200812 * on request or response considering the <opt> parameter.
Willy Tarreau12785782012-04-27 21:37:17 +0200813 * Returns a pointer on a typed sample structure containing the result or NULL if
814 * sample is not found or when format conversion failed.
Emeric Brun107ca302010-01-04 16:16:05 +0100815 * If <p> is not null, function returns results in structure pointed by <p>.
Willy Tarreau12785782012-04-27 21:37:17 +0200816 * If <p> is null, functions returns a pointer on a static sample structure.
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +0200817 *
818 * Note: the fetch functions are required to properly set the return type. The
819 * conversion functions must do so too. However the cast functions do not need
820 * to since they're made to cast mutiple types according to what is required.
Emeric Brun107ca302010-01-04 16:16:05 +0100821 */
Willy Tarreau12785782012-04-27 21:37:17 +0200822struct sample *sample_process(struct proxy *px, struct session *l4, void *l7,
823 unsigned int opt,
824 struct sample_expr *expr, struct sample *p)
Emeric Brun107ca302010-01-04 16:16:05 +0100825{
Willy Tarreau12785782012-04-27 21:37:17 +0200826 struct sample_conv_expr *conv_expr;
Emeric Brun107ca302010-01-04 16:16:05 +0100827
Willy Tarreau18387e22013-07-25 12:02:38 +0200828 if (p == NULL) {
Willy Tarreaub4a88f02012-04-23 21:35:11 +0200829 p = &temp_smp;
Willy Tarreau18387e22013-07-25 12:02:38 +0200830 p->flags = 0;
831 }
Emeric Brun107ca302010-01-04 16:16:05 +0100832
Willy Tarreauef38c392013-07-22 16:29:32 +0200833 if (!expr->fetch->process(px, l4, l7, opt, expr->arg_p, p, expr->fetch->kw))
Emeric Brun107ca302010-01-04 16:16:05 +0100834 return NULL;
835
Emeric Brun107ca302010-01-04 16:16:05 +0100836 list_for_each_entry(conv_expr, &expr->conv_exprs, list) {
Willy Tarreau12e50112012-04-25 17:21:49 +0200837 /* we want to ensure that p->type can be casted into
838 * conv_expr->conv->in_type. We have 3 possibilities :
839 * - NULL => not castable.
840 * - c_none => nothing to do (let's optimize it)
841 * - other => apply cast and prepare to fail
842 */
Willy Tarreau12785782012-04-27 21:37:17 +0200843 if (!sample_casts[p->type][conv_expr->conv->in_type])
Willy Tarreau12e50112012-04-25 17:21:49 +0200844 return NULL;
845
Willy Tarreau12785782012-04-27 21:37:17 +0200846 if (sample_casts[p->type][conv_expr->conv->in_type] != c_none &&
847 !sample_casts[p->type][conv_expr->conv->in_type](p))
Emeric Brun107ca302010-01-04 16:16:05 +0100848 return NULL;
849
Willy Tarreau12e50112012-04-25 17:21:49 +0200850 /* OK cast succeeded */
851
Willy Tarreau342acb42012-04-23 22:03:39 +0200852 if (!conv_expr->conv->process(conv_expr->arg_p, p))
Emeric Brun107ca302010-01-04 16:16:05 +0100853 return NULL;
Emeric Brun107ca302010-01-04 16:16:05 +0100854 }
855 return p;
856}
857
Willy Tarreaue7ad4bb2012-12-21 00:02:32 +0100858/*
Willy Tarreaua4312fa2013-04-02 16:34:32 +0200859 * Resolve all remaining arguments in proxy <p>. Returns the number of
860 * errors or 0 if everything is fine.
861 */
862int smp_resolve_args(struct proxy *p)
863{
864 struct arg_list *cur, *bak;
865 const char *ctx, *where;
866 const char *conv_ctx, *conv_pre, *conv_pos;
867 struct userlist *ul;
868 struct arg *arg;
869 int cfgerr = 0;
870
871 list_for_each_entry_safe(cur, bak, &p->conf.args.list, list) {
872 struct proxy *px;
873 struct server *srv;
874 char *pname, *sname;
875
876 arg = cur->arg;
877
878 /* prepare output messages */
879 conv_pre = conv_pos = conv_ctx = "";
880 if (cur->conv) {
881 conv_ctx = cur->conv;
882 conv_pre = "conversion keyword '";
883 conv_pos = "' for ";
884 }
885
886 where = "in";
887 ctx = "sample fetch keyword";
888 switch (cur->ctx) {
889 case ARGC_STK:where = "in stick rule in"; break;
890 case ARGC_TRK: where = "in tracking rule in"; break;
891 case ARGC_LOG: where = "in log-format string in"; break;
Thierry FOURNIER1c0054f2013-11-20 15:09:52 +0100892 case ARGC_HRQ: where = "in http-request header format string in"; break;
893 case ARGC_HRS: where = "in http-response header format string in"; break;
Willy Tarreaua4312fa2013-04-02 16:34:32 +0200894 case ARGC_UIF: where = "in unique-id-format string in"; break;
Thierry FOURNIERd18cd0f2013-11-29 12:15:45 +0100895 case ARGC_RDR: where = "in redirect format string in"; break;
Willy Tarreaua4312fa2013-04-02 16:34:32 +0200896 case ARGC_ACL: ctx = "ACL keyword"; break;
897 }
898
899 /* set a few default settings */
900 px = p;
901 pname = p->id;
902
903 switch (arg->type) {
904 case ARGT_SRV:
905 if (!arg->data.str.len) {
906 Alert("parsing [%s:%d] : missing server name in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
907 cur->file, cur->line,
908 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
909 cfgerr++;
910 continue;
911 }
912
913 /* we support two formats : "bck/srv" and "srv" */
914 sname = strrchr(arg->data.str.str, '/');
915
916 if (sname) {
917 *sname++ = '\0';
918 pname = arg->data.str.str;
919
920 px = findproxy(pname, PR_CAP_BE);
921 if (!px) {
922 Alert("parsing [%s:%d] : unable to find proxy '%s' referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
923 cur->file, cur->line, pname,
924 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
925 cfgerr++;
926 break;
927 }
928 }
929 else
930 sname = arg->data.str.str;
931
932 srv = findserver(px, sname);
933 if (!srv) {
934 Alert("parsing [%s:%d] : unable to find server '%s' in proxy '%s', referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
935 cur->file, cur->line, sname, pname,
936 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
937 cfgerr++;
938 break;
939 }
940
941 free(arg->data.str.str);
942 arg->data.str.str = NULL;
943 arg->unresolved = 0;
944 arg->data.srv = srv;
945 break;
946
947 case ARGT_FE:
948 if (arg->data.str.len) {
949 pname = arg->data.str.str;
950 px = findproxy(pname, PR_CAP_FE);
951 }
952
953 if (!px) {
954 Alert("parsing [%s:%d] : unable to find frontend '%s' referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
955 cur->file, cur->line, pname,
956 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
957 cfgerr++;
958 break;
959 }
960
961 if (!(px->cap & PR_CAP_FE)) {
962 Alert("parsing [%s:%d] : proxy '%s', referenced in arg %d of %s%s%s%s '%s' %s proxy '%s', has not frontend capability.\n",
963 cur->file, cur->line, pname,
964 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
965 cfgerr++;
966 break;
967 }
968
969 free(arg->data.str.str);
970 arg->data.str.str = NULL;
971 arg->unresolved = 0;
972 arg->data.prx = px;
973 break;
974
975 case ARGT_BE:
976 if (arg->data.str.len) {
977 pname = arg->data.str.str;
978 px = findproxy(pname, PR_CAP_BE);
979 }
980
981 if (!px) {
982 Alert("parsing [%s:%d] : unable to find backend '%s' referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
983 cur->file, cur->line, pname,
984 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
985 cfgerr++;
986 break;
987 }
988
989 if (!(px->cap & PR_CAP_BE)) {
990 Alert("parsing [%s:%d] : proxy '%s', referenced in arg %d of %s%s%s%s '%s' %s proxy '%s', has not backend capability.\n",
991 cur->file, cur->line, pname,
992 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
993 cfgerr++;
994 break;
995 }
996
997 free(arg->data.str.str);
998 arg->data.str.str = NULL;
999 arg->unresolved = 0;
1000 arg->data.prx = px;
1001 break;
1002
1003 case ARGT_TAB:
1004 if (arg->data.str.len) {
1005 pname = arg->data.str.str;
1006 px = find_stktable(pname);
1007 }
1008
1009 if (!px) {
1010 Alert("parsing [%s:%d] : unable to find table '%s' referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
1011 cur->file, cur->line, pname,
1012 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
1013 cfgerr++;
1014 break;
1015 }
1016
1017 if (!px->table.size) {
1018 Alert("parsing [%s:%d] : no table in proxy '%s' referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
1019 cur->file, cur->line, pname,
1020 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
1021 cfgerr++;
1022 break;
1023 }
1024
1025 free(arg->data.str.str);
1026 arg->data.str.str = NULL;
1027 arg->unresolved = 0;
1028 arg->data.prx = px;
1029 break;
1030
1031 case ARGT_USR:
1032 if (!arg->data.str.len) {
1033 Alert("parsing [%s:%d] : missing userlist name in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
1034 cur->file, cur->line,
1035 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
1036 cfgerr++;
1037 break;
1038 }
1039
1040 if (p->uri_auth && p->uri_auth->userlist &&
1041 !strcmp(p->uri_auth->userlist->name, arg->data.str.str))
1042 ul = p->uri_auth->userlist;
1043 else
1044 ul = auth_find_userlist(arg->data.str.str);
1045
1046 if (!ul) {
1047 Alert("parsing [%s:%d] : unable to find userlist '%s' referenced in arg %d of %s%s%s%s '%s' %s proxy '%s'.\n",
1048 cur->file, cur->line, arg->data.str.str,
1049 cur->arg_pos + 1, conv_pre, conv_ctx, conv_pos, ctx, cur->kw, where, p->id);
1050 cfgerr++;
1051 break;
1052 }
1053
1054 free(arg->data.str.str);
1055 arg->data.str.str = NULL;
1056 arg->unresolved = 0;
1057 arg->data.usr = ul;
1058 break;
1059 }
1060
1061 LIST_DEL(&cur->list);
1062 free(cur);
1063 } /* end of args processing */
1064
1065 return cfgerr;
1066}
1067
1068/*
Willy Tarreaue7ad4bb2012-12-21 00:02:32 +01001069 * Process a fetch + format conversion as defined by the sample expression <expr>
1070 * on request or response considering the <opt> parameter. The output is always of
1071 * type string. Returns either NULL if no sample could be extracted, or a pointer
1072 * to the converted result stored in static temp_smp in format string.
1073 */
1074struct sample *sample_fetch_string(struct proxy *px, struct session *l4, void *l7,
1075 unsigned int opt, struct sample_expr *expr)
1076{
1077 struct sample *smp;
1078
1079 smp = sample_process(px, l4, l7, opt, expr, NULL);
1080 if (!smp)
1081 return NULL;
1082
1083 if (!sample_casts[smp->type][SMP_T_CSTR])
1084 return NULL;
1085
1086 if (!sample_casts[smp->type][SMP_T_CSTR](smp))
1087 return NULL;
1088
1089 smp->type = SMP_T_CSTR;
1090 return smp;
1091}
1092
Emeric Brun107ca302010-01-04 16:16:05 +01001093/*****************************************************************/
Willy Tarreau12785782012-04-27 21:37:17 +02001094/* Sample format convert functions */
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +02001095/* These functions set the data type on return. */
Emeric Brun107ca302010-01-04 16:16:05 +01001096/*****************************************************************/
1097
Thierry FOURNIER2f49d6d2014-03-12 15:01:52 +01001098static int sample_conv_bin2hex(const struct arg *arg_p, struct sample *smp)
1099{
1100 struct chunk *trash = get_trash_chunk();
1101 unsigned char c;
1102 int ptr = 0;
1103
1104 trash->len = 0;
1105 while (ptr < smp->data.str.len && trash->len <= trash->size - 2) {
1106 c = smp->data.str.str[ptr++];
1107 trash->str[trash->len++] = hextab[(c >> 4) & 0xF];
1108 trash->str[trash->len++] = hextab[c & 0xF];
1109 }
1110 smp->data.str = *trash;
1111 smp->type = SMP_T_STR;
1112 return 1;
1113}
1114
Willy Tarreau12785782012-04-27 21:37:17 +02001115static int sample_conv_str2lower(const struct arg *arg_p, struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +01001116{
1117 int i;
1118
Willy Tarreau342acb42012-04-23 22:03:39 +02001119 if (!smp->data.str.size)
Emeric Brun485479d2010-09-23 18:02:19 +02001120 return 0;
1121
Willy Tarreau342acb42012-04-23 22:03:39 +02001122 for (i = 0; i < smp->data.str.len; i++) {
1123 if ((smp->data.str.str[i] >= 'A') && (smp->data.str.str[i] <= 'Z'))
1124 smp->data.str.str[i] += 'a' - 'A';
Emeric Brun107ca302010-01-04 16:16:05 +01001125 }
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +02001126 smp->type = SMP_T_STR;
Emeric Brun107ca302010-01-04 16:16:05 +01001127 return 1;
1128}
1129
Willy Tarreau12785782012-04-27 21:37:17 +02001130static int sample_conv_str2upper(const struct arg *arg_p, struct sample *smp)
Emeric Brun107ca302010-01-04 16:16:05 +01001131{
1132 int i;
1133
Willy Tarreau342acb42012-04-23 22:03:39 +02001134 if (!smp->data.str.size)
Emeric Brun485479d2010-09-23 18:02:19 +02001135 return 0;
1136
Willy Tarreau342acb42012-04-23 22:03:39 +02001137 for (i = 0; i < smp->data.str.len; i++) {
1138 if ((smp->data.str.str[i] >= 'a') && (smp->data.str.str[i] <= 'z'))
1139 smp->data.str.str[i] += 'A' - 'a';
Emeric Brun107ca302010-01-04 16:16:05 +01001140 }
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +02001141 smp->type = SMP_T_STR;
Emeric Brun107ca302010-01-04 16:16:05 +01001142 return 1;
1143}
1144
Willy Tarreauf9954102012-04-20 14:03:29 +02001145/* takes the netmask in arg_p */
Willy Tarreau12785782012-04-27 21:37:17 +02001146static int sample_conv_ipmask(const struct arg *arg_p, struct sample *smp)
Willy Tarreaud31d6eb2010-01-26 18:01:41 +01001147{
Willy Tarreau342acb42012-04-23 22:03:39 +02001148 smp->data.ipv4.s_addr &= arg_p->data.ipv4.s_addr;
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +02001149 smp->type = SMP_T_IPV4;
Willy Tarreaud31d6eb2010-01-26 18:01:41 +01001150 return 1;
1151}
1152
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001153/************************************************************************/
1154/* All supported sample fetch functions must be declared here */
1155/************************************************************************/
1156
1157/* force TRUE to be returned at the fetch level */
1158static int
1159smp_fetch_true(struct proxy *px, struct session *s, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02001160 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001161{
1162 smp->type = SMP_T_BOOL;
1163 smp->data.uint = 1;
1164 return 1;
1165}
1166
1167/* force FALSE to be returned at the fetch level */
1168static int
1169smp_fetch_false(struct proxy *px, struct session *s, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02001170 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001171{
1172 smp->type = SMP_T_BOOL;
1173 smp->data.uint = 0;
1174 return 1;
1175}
1176
1177/* retrieve environment variable $1 as a string */
1178static int
1179smp_fetch_env(struct proxy *px, struct session *s, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02001180 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001181{
1182 char *env;
1183
1184 if (!args || args[0].type != ARGT_STR)
1185 return 0;
1186
1187 env = getenv(args[0].data.str.str);
1188 if (!env)
1189 return 0;
1190
1191 smp->type = SMP_T_CSTR;
1192 smp->data.str.str = env;
1193 smp->data.str.len = strlen(env);
1194 return 1;
1195}
1196
Willy Tarreau6236d3a2013-07-25 14:28:25 +02001197/* retrieve the current local date in epoch time, and applies an optional offset
1198 * of args[0] seconds.
1199 */
1200static int
1201smp_fetch_date(struct proxy *px, struct session *s, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02001202 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau6236d3a2013-07-25 14:28:25 +02001203{
1204 smp->data.uint = date.tv_sec;
1205
1206 /* add offset */
1207 if (args && (args[0].type == ARGT_SINT || args[0].type == ARGT_UINT))
1208 smp->data.uint += args[0].data.sint;
1209
1210 smp->type = SMP_T_UINT;
1211 smp->flags |= SMP_F_VOL_TEST | SMP_F_MAY_CHANGE;
1212 return 1;
1213}
1214
Willy Tarreau84310e22014-02-14 11:59:04 +01001215/* generate a random 32-bit integer for whatever purpose, with an optional
1216 * range specified in argument.
1217 */
1218static int
1219smp_fetch_rand(struct proxy *px, struct session *s, void *l7, unsigned int opt,
1220 const struct arg *args, struct sample *smp, const char *kw)
1221{
1222 smp->data.uint = random();
1223
1224 /* reduce if needed. Don't do a modulo, use all bits! */
1225 if (args && args[0].type == ARGT_UINT)
1226 smp->data.uint = ((uint64_t)smp->data.uint * args[0].data.uint) >> 32;
1227
1228 smp->type = SMP_T_UINT;
1229 smp->flags |= SMP_F_VOL_TEST | SMP_F_MAY_CHANGE;
1230 return 1;
1231}
1232
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001233/* Note: must not be declared <const> as its list will be overwritten.
1234 * Note: fetches that may return multiple types must be declared as the lowest
1235 * common denominator, the type that can be casted into all other ones. For
1236 * instance IPv4/IPv6 must be declared IPv4.
1237 */
1238static struct sample_fetch_kw_list smp_kws = {ILH, {
1239 { "always_false", smp_fetch_false, 0, NULL, SMP_T_BOOL, SMP_USE_INTRN },
1240 { "always_true", smp_fetch_true, 0, NULL, SMP_T_BOOL, SMP_USE_INTRN },
1241 { "env", smp_fetch_env, ARG1(1,STR), NULL, SMP_T_CSTR, SMP_USE_INTRN },
Willy Tarreau6236d3a2013-07-25 14:28:25 +02001242 { "date", smp_fetch_date, ARG1(0,SINT), NULL, SMP_T_UINT, SMP_USE_INTRN },
Willy Tarreau84310e22014-02-14 11:59:04 +01001243 { "rand", smp_fetch_rand, ARG1(0,UINT), NULL, SMP_T_UINT, SMP_USE_INTRN },
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001244 { /* END */ },
1245}};
1246
Emeric Brun107ca302010-01-04 16:16:05 +01001247/* Note: must not be declared <const> as its list will be overwritten */
Willy Tarreaudc13c112013-06-21 23:16:39 +02001248static struct sample_conv_kw_list sample_conv_kws = {ILH, {
Willy Tarreau12785782012-04-27 21:37:17 +02001249 { "upper", sample_conv_str2upper, 0, NULL, SMP_T_STR, SMP_T_STR },
1250 { "lower", sample_conv_str2lower, 0, NULL, SMP_T_STR, SMP_T_STR },
Thierry FOURNIER2f49d6d2014-03-12 15:01:52 +01001251 { "hex", sample_conv_bin2hex, 0, NULL, SMP_T_BIN, SMP_T_STR },
Willy Tarreau12785782012-04-27 21:37:17 +02001252 { "ipmask", sample_conv_ipmask, ARG1(1,MSK4), NULL, SMP_T_IPV4, SMP_T_IPV4 },
Willy Tarreau9fcb9842012-04-20 14:45:49 +02001253 { NULL, NULL, 0, 0, 0 },
Emeric Brun107ca302010-01-04 16:16:05 +01001254}};
1255
1256__attribute__((constructor))
Willy Tarreau12785782012-04-27 21:37:17 +02001257static void __sample_init(void)
Emeric Brun107ca302010-01-04 16:16:05 +01001258{
Willy Tarreau5b8ad222013-07-25 12:17:57 +02001259 /* register sample fetch and format conversion keywords */
1260 sample_register_fetches(&smp_kws);
Willy Tarreau12785782012-04-27 21:37:17 +02001261 sample_register_convs(&sample_conv_kws);
Emeric Brun107ca302010-01-04 16:16:05 +01001262}