blob: eee65bc9bf38cb6f2ba7fc497fefa77f325209fb [file] [log] [blame]
Willy Tarreaue6b98942007-10-29 01:09:36 +01001/*
2 * AF_INET/AF_INET6 SOCK_STREAM protocol layer (tcp)
3 *
Willy Tarreau1a687942010-05-23 22:40:30 +02004 * Copyright 2000-2010 Willy Tarreau <w@1wt.eu>
Willy Tarreaue6b98942007-10-29 01:09:36 +01005 *
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
13#include <ctype.h>
14#include <errno.h>
15#include <fcntl.h>
16#include <stdio.h>
17#include <stdlib.h>
18#include <string.h>
19#include <time.h>
20
21#include <sys/param.h>
22#include <sys/socket.h>
23#include <sys/stat.h>
24#include <sys/types.h>
25#include <sys/un.h>
26
Dmitry Sivachenkocaf58982009-08-24 15:11:06 +040027#include <netinet/tcp.h>
28
Willy Tarreaub6866442008-07-14 23:54:42 +020029#include <common/cfgparse.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010030#include <common/compat.h>
31#include <common/config.h>
32#include <common/debug.h>
33#include <common/errors.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010034#include <common/mini-clist.h>
35#include <common/standard.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010036
Willy Tarreaue6b98942007-10-29 01:09:36 +010037#include <types/global.h>
Willy Tarreau9650f372009-08-16 14:02:45 +020038#include <types/server.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010039
40#include <proto/acl.h>
Willy Tarreau9fcb9842012-04-20 14:45:49 +020041#include <proto/arg.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010042#include <proto/buffers.h>
Willy Tarreau03fa5df2010-05-24 21:02:37 +020043#include <proto/frontend.h>
Willy Tarreau9650f372009-08-16 14:02:45 +020044#include <proto/log.h>
45#include <proto/port_range.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010046#include <proto/protocols.h>
47#include <proto/proto_tcp.h>
Willy Tarreaub6866442008-07-14 23:54:42 +020048#include <proto/proxy.h>
Willy Tarreaucd3b0942012-04-27 21:52:18 +020049#include <proto/sample.h>
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +020050#include <proto/session.h>
Willy Tarreauc63190d2012-05-11 14:23:52 +020051#include <proto/sock_raw.h>
Willy Tarreaua975b8f2010-06-05 19:13:27 +020052#include <proto/stick_table.h>
Willy Tarreau59b94792012-05-11 16:16:40 +020053#include <proto/stream_interface.h>
Willy Tarreaua975b8f2010-06-05 19:13:27 +020054#include <proto/task.h>
Emeric Brun97679e72010-09-23 17:56:44 +020055#include <proto/buffers.h>
Willy Tarreaue6b98942007-10-29 01:09:36 +010056
Willy Tarreaue8c66af2008-01-13 18:40:14 +010057#ifdef CONFIG_HAP_CTTPROXY
58#include <import/ip_tproxy.h>
59#endif
60
Emeric Bruncf20bf12010-10-22 16:06:11 +020061static int tcp_bind_listeners(struct protocol *proto, char *errmsg, int errlen);
62static int tcp_bind_listener(struct listener *listener, char *errmsg, int errlen);
Willy Tarreaueeda90e2012-05-11 19:53:32 +020063static int tcp_connect_write(int fd);
64static int tcp_connect_read(int fd);
Willy Tarreaue6b98942007-10-29 01:09:36 +010065
66/* Note: must not be declared <const> as its list will be overwritten */
67static struct protocol proto_tcpv4 = {
68 .name = "tcpv4",
69 .sock_domain = AF_INET,
70 .sock_type = SOCK_STREAM,
71 .sock_prot = IPPROTO_TCP,
72 .sock_family = AF_INET,
73 .sock_addrlen = sizeof(struct sockaddr_in),
74 .l3_addrlen = 32/8,
Willy Tarreaubbebbbf2012-05-07 21:22:09 +020075 .accept = &listener_accept,
Willy Tarreau26d8c592012-05-07 18:12:14 +020076 .connect = tcp_connect_server,
Emeric Bruncf20bf12010-10-22 16:06:11 +020077 .bind = tcp_bind_listener,
Willy Tarreaue6b98942007-10-29 01:09:36 +010078 .bind_all = tcp_bind_listeners,
79 .unbind_all = unbind_all_listeners,
80 .enable_all = enable_all_listeners,
Willy Tarreau59b94792012-05-11 16:16:40 +020081 .get_src = tcp_get_src,
82 .get_dst = tcp_get_dst,
Willy Tarreaue6b98942007-10-29 01:09:36 +010083 .listeners = LIST_HEAD_INIT(proto_tcpv4.listeners),
84 .nb_listeners = 0,
85};
86
87/* Note: must not be declared <const> as its list will be overwritten */
88static struct protocol proto_tcpv6 = {
89 .name = "tcpv6",
90 .sock_domain = AF_INET6,
91 .sock_type = SOCK_STREAM,
92 .sock_prot = IPPROTO_TCP,
93 .sock_family = AF_INET6,
94 .sock_addrlen = sizeof(struct sockaddr_in6),
95 .l3_addrlen = 128/8,
Willy Tarreaubbebbbf2012-05-07 21:22:09 +020096 .accept = &listener_accept,
Willy Tarreau26d8c592012-05-07 18:12:14 +020097 .connect = tcp_connect_server,
Emeric Bruncf20bf12010-10-22 16:06:11 +020098 .bind = tcp_bind_listener,
Willy Tarreaue6b98942007-10-29 01:09:36 +010099 .bind_all = tcp_bind_listeners,
100 .unbind_all = unbind_all_listeners,
101 .enable_all = enable_all_listeners,
Willy Tarreau59b94792012-05-11 16:16:40 +0200102 .get_src = tcp_get_src,
103 .get_dst = tcp_get_dst,
Willy Tarreaue6b98942007-10-29 01:09:36 +0100104 .listeners = LIST_HEAD_INIT(proto_tcpv6.listeners),
105 .nb_listeners = 0,
106};
107
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100108
David du Colombier6f5ccb12011-03-10 22:26:24 +0100109/* Binds ipv4/ipv6 address <local> to socket <fd>, unless <flags> is set, in which
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100110 * case we try to bind <remote>. <flags> is a 2-bit field consisting of :
111 * - 0 : ignore remote address (may even be a NULL pointer)
112 * - 1 : use provided address
113 * - 2 : use provided port
114 * - 3 : use both
115 *
116 * The function supports multiple foreign binding methods :
117 * - linux_tproxy: we directly bind to the foreign address
118 * - cttproxy: we bind to a local address then nat.
119 * The second one can be used as a fallback for the first one.
120 * This function returns 0 when everything's OK, 1 if it could not bind, to the
121 * local address, 2 if it could not bind to the foreign address.
122 */
David du Colombier6f5ccb12011-03-10 22:26:24 +0100123int tcp_bind_socket(int fd, int flags, struct sockaddr_storage *local, struct sockaddr_storage *remote)
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100124{
David du Colombier6f5ccb12011-03-10 22:26:24 +0100125 struct sockaddr_storage bind_addr;
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100126 int foreign_ok = 0;
127 int ret;
128
129#ifdef CONFIG_HAP_LINUX_TPROXY
130 static int ip_transp_working = 1;
131 if (flags && ip_transp_working) {
Simon Hormande072bd2011-06-24 15:11:37 +0900132 if (setsockopt(fd, SOL_IP, IP_TRANSPARENT, &one, sizeof(one)) == 0
133 || setsockopt(fd, SOL_IP, IP_FREEBIND, &one, sizeof(one)) == 0)
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100134 foreign_ok = 1;
135 else
136 ip_transp_working = 0;
137 }
138#endif
139 if (flags) {
140 memset(&bind_addr, 0, sizeof(bind_addr));
Willy Tarreau96dd0792011-04-19 07:20:57 +0200141 bind_addr.ss_family = remote->ss_family;
David du Colombier6f5ccb12011-03-10 22:26:24 +0100142 switch (remote->ss_family) {
143 case AF_INET:
144 if (flags & 1)
145 ((struct sockaddr_in *)&bind_addr)->sin_addr = ((struct sockaddr_in *)remote)->sin_addr;
146 if (flags & 2)
147 ((struct sockaddr_in *)&bind_addr)->sin_port = ((struct sockaddr_in *)remote)->sin_port;
148 break;
149 case AF_INET6:
150 if (flags & 1)
151 ((struct sockaddr_in6 *)&bind_addr)->sin6_addr = ((struct sockaddr_in6 *)remote)->sin6_addr;
152 if (flags & 2)
153 ((struct sockaddr_in6 *)&bind_addr)->sin6_port = ((struct sockaddr_in6 *)remote)->sin6_port;
154 break;
Willy Tarreau5dc1e982011-12-16 21:25:11 +0100155 default:
156 /* we don't want to try to bind to an unknown address family */
157 foreign_ok = 0;
David du Colombier6f5ccb12011-03-10 22:26:24 +0100158 }
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100159 }
160
Simon Hormande072bd2011-06-24 15:11:37 +0900161 setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100162 if (foreign_ok) {
Willy Tarreau1b4b7ce2011-04-05 16:56:50 +0200163 ret = bind(fd, (struct sockaddr *)&bind_addr, get_addr_len(&bind_addr));
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100164 if (ret < 0)
165 return 2;
166 }
167 else {
Willy Tarreau1b4b7ce2011-04-05 16:56:50 +0200168 ret = bind(fd, (struct sockaddr *)local, get_addr_len(local));
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100169 if (ret < 0)
170 return 1;
171 }
172
173 if (!flags)
174 return 0;
175
176#ifdef CONFIG_HAP_CTTPROXY
Willy Tarreau6f831b42011-03-20 14:03:54 +0100177 if (!foreign_ok && remote->ss_family == AF_INET) {
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100178 struct in_tproxy itp1, itp2;
179 memset(&itp1, 0, sizeof(itp1));
180
181 itp1.op = TPROXY_ASSIGN;
Willy Tarreau6f831b42011-03-20 14:03:54 +0100182 itp1.v.addr.faddr = ((struct sockaddr_in *)&bind_addr)->sin_addr;
183 itp1.v.addr.fport = ((struct sockaddr_in *)&bind_addr)->sin_port;
Willy Tarreaue8c66af2008-01-13 18:40:14 +0100184
185 /* set connect flag on socket */
186 itp2.op = TPROXY_FLAGS;
187 itp2.v.flags = ITP_CONNECT | ITP_ONCE;
188
189 if (setsockopt(fd, SOL_IP, IP_TPROXY, &itp1, sizeof(itp1)) != -1 &&
190 setsockopt(fd, SOL_IP, IP_TPROXY, &itp2, sizeof(itp2)) != -1) {
191 foreign_ok = 1;
192 }
193 }
194#endif
195 if (!foreign_ok)
196 /* we could not bind to a foreign address */
197 return 2;
198
199 return 0;
200}
Willy Tarreaue6b98942007-10-29 01:09:36 +0100201
Willy Tarreau9650f372009-08-16 14:02:45 +0200202
203/*
Willy Tarreauac825402011-03-04 22:04:29 +0100204 * This function initiates a connection to the target assigned to this session
Willy Tarreau6471afb2011-09-23 10:54:59 +0200205 * (si->{target,addr.to}). A source address may be pointed to by si->addr.from
Willy Tarreauac825402011-03-04 22:04:29 +0100206 * in case of transparent proxying. Normal source bind addresses are still
207 * determined locally (due to the possible need of a source port).
208 * si->target may point either to a valid server or to a backend, depending
209 * on si->target.type. Only TARG_TYPE_PROXY and TARG_TYPE_SERVER are supported.
Willy Tarreaub1d67742010-03-29 19:36:59 +0200210 *
Willy Tarreau9650f372009-08-16 14:02:45 +0200211 * It can return one of :
212 * - SN_ERR_NONE if everything's OK
213 * - SN_ERR_SRVTO if there are no more servers
214 * - SN_ERR_SRVCL if the connection was refused by the server
215 * - SN_ERR_PRXCOND if the connection has been limited by the proxy (maxconn)
216 * - SN_ERR_RESOURCE if a system resource is lacking (eg: fd limits, ports, ...)
217 * - SN_ERR_INTERNAL for any other purely internal errors
218 * Additionnally, in the case of SN_ERR_RESOURCE, an emergency log will be emitted.
219 */
Willy Tarreauf1536862011-03-03 18:27:32 +0100220
David du Colombier6f5ccb12011-03-10 22:26:24 +0100221int tcp_connect_server(struct stream_interface *si)
Willy Tarreau9650f372009-08-16 14:02:45 +0200222{
223 int fd;
Willy Tarreauac825402011-03-04 22:04:29 +0100224 struct server *srv;
225 struct proxy *be;
226
227 switch (si->target.type) {
228 case TARG_TYPE_PROXY:
229 be = si->target.ptr.p;
230 srv = NULL;
231 break;
232 case TARG_TYPE_SERVER:
233 srv = si->target.ptr.s;
234 be = srv->proxy;
235 break;
236 default:
237 return SN_ERR_INTERNAL;
238 }
Willy Tarreau9650f372009-08-16 14:02:45 +0200239
Willy Tarreau6471afb2011-09-23 10:54:59 +0200240 if ((fd = si->fd = socket(si->addr.to.ss_family, SOCK_STREAM, IPPROTO_TCP)) == -1) {
Willy Tarreau9650f372009-08-16 14:02:45 +0200241 qfprintf(stderr, "Cannot get a server socket.\n");
242
243 if (errno == ENFILE)
244 send_log(be, LOG_EMERG,
245 "Proxy %s reached system FD limit at %d. Please check system tunables.\n",
246 be->id, maxfd);
247 else if (errno == EMFILE)
248 send_log(be, LOG_EMERG,
249 "Proxy %s reached process FD limit at %d. Please check 'ulimit-n' and restart.\n",
250 be->id, maxfd);
251 else if (errno == ENOBUFS || errno == ENOMEM)
252 send_log(be, LOG_EMERG,
253 "Proxy %s reached system memory limit at %d sockets. Please check system tunables.\n",
254 be->id, maxfd);
255 /* this is a resource error */
256 return SN_ERR_RESOURCE;
257 }
258
259 if (fd >= global.maxsock) {
260 /* do not log anything there, it's a normal condition when this option
261 * is used to serialize connections to a server !
262 */
263 Alert("socket(): not enough free sockets. Raise -n argument. Giving up.\n");
264 close(fd);
265 return SN_ERR_PRXCOND; /* it is a configuration limit */
266 }
267
268 if ((fcntl(fd, F_SETFL, O_NONBLOCK)==-1) ||
Simon Hormande072bd2011-06-24 15:11:37 +0900269 (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &one, sizeof(one)) == -1)) {
Willy Tarreau9650f372009-08-16 14:02:45 +0200270 qfprintf(stderr,"Cannot set client socket to non blocking mode.\n");
271 close(fd);
272 return SN_ERR_INTERNAL;
273 }
274
275 if (be->options & PR_O_TCP_SRV_KA)
Simon Hormande072bd2011-06-24 15:11:37 +0900276 setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &one, sizeof(one));
Willy Tarreau9650f372009-08-16 14:02:45 +0200277
278 if (be->options & PR_O_TCP_NOLING)
Willy Tarreauf6f82252011-11-30 18:02:24 +0100279 si->flags |= SI_FL_NOLINGER;
Willy Tarreau9650f372009-08-16 14:02:45 +0200280
281 /* allow specific binding :
282 * - server-specific at first
283 * - proxy-specific next
284 */
285 if (srv != NULL && srv->state & SRV_BIND_SRC) {
Willy Tarreau9650f372009-08-16 14:02:45 +0200286 int ret, flags = 0;
287
Willy Tarreau9650f372009-08-16 14:02:45 +0200288 switch (srv->state & SRV_TPROXY_MASK) {
289 case SRV_TPROXY_ADDR:
Willy Tarreau9650f372009-08-16 14:02:45 +0200290 case SRV_TPROXY_CLI:
Willy Tarreaub1d67742010-03-29 19:36:59 +0200291 flags = 3;
292 break;
Willy Tarreau9650f372009-08-16 14:02:45 +0200293 case SRV_TPROXY_CIP:
Willy Tarreau090466c2009-09-07 11:51:47 +0200294 case SRV_TPROXY_DYN:
Willy Tarreaub1d67742010-03-29 19:36:59 +0200295 flags = 1;
Willy Tarreau9650f372009-08-16 14:02:45 +0200296 break;
297 }
Willy Tarreaub1d67742010-03-29 19:36:59 +0200298
Willy Tarreau9650f372009-08-16 14:02:45 +0200299#ifdef SO_BINDTODEVICE
300 /* Note: this might fail if not CAP_NET_RAW */
301 if (srv->iface_name)
302 setsockopt(fd, SOL_SOCKET, SO_BINDTODEVICE, srv->iface_name, srv->iface_len + 1);
303#endif
304
305 if (srv->sport_range) {
306 int attempts = 10; /* should be more than enough to find a spare port */
David du Colombier6f5ccb12011-03-10 22:26:24 +0100307 struct sockaddr_storage src;
Willy Tarreau9650f372009-08-16 14:02:45 +0200308
309 ret = 1;
310 src = srv->source_addr;
311
312 do {
313 /* note: in case of retry, we may have to release a previously
314 * allocated port, hence this loop's construct.
315 */
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200316 port_range_release_port(fdinfo[fd].port_range, fdinfo[fd].local_port);
317 fdinfo[fd].port_range = NULL;
Willy Tarreau9650f372009-08-16 14:02:45 +0200318
319 if (!attempts)
320 break;
321 attempts--;
322
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200323 fdinfo[fd].local_port = port_range_alloc_port(srv->sport_range);
324 if (!fdinfo[fd].local_port)
Willy Tarreau9650f372009-08-16 14:02:45 +0200325 break;
326
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200327 fdinfo[fd].port_range = srv->sport_range;
Willy Tarreau86ad42c2011-08-27 12:29:07 +0200328 set_host_port(&src, fdinfo[fd].local_port);
Willy Tarreau9650f372009-08-16 14:02:45 +0200329
Willy Tarreau6471afb2011-09-23 10:54:59 +0200330 ret = tcp_bind_socket(fd, flags, &src, &si->addr.from);
Willy Tarreau9650f372009-08-16 14:02:45 +0200331 } while (ret != 0); /* binding NOK */
332 }
333 else {
Willy Tarreau6471afb2011-09-23 10:54:59 +0200334 ret = tcp_bind_socket(fd, flags, &srv->source_addr, &si->addr.from);
Willy Tarreau9650f372009-08-16 14:02:45 +0200335 }
336
337 if (ret) {
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200338 port_range_release_port(fdinfo[fd].port_range, fdinfo[fd].local_port);
339 fdinfo[fd].port_range = NULL;
Willy Tarreau9650f372009-08-16 14:02:45 +0200340 close(fd);
341
342 if (ret == 1) {
343 Alert("Cannot bind to source address before connect() for server %s/%s. Aborting.\n",
344 be->id, srv->id);
345 send_log(be, LOG_EMERG,
346 "Cannot bind to source address before connect() for server %s/%s.\n",
347 be->id, srv->id);
348 } else {
349 Alert("Cannot bind to tproxy source address before connect() for server %s/%s. Aborting.\n",
350 be->id, srv->id);
351 send_log(be, LOG_EMERG,
352 "Cannot bind to tproxy source address before connect() for server %s/%s.\n",
353 be->id, srv->id);
354 }
355 return SN_ERR_RESOURCE;
356 }
357 }
358 else if (be->options & PR_O_BIND_SRC) {
Willy Tarreau9650f372009-08-16 14:02:45 +0200359 int ret, flags = 0;
360
Willy Tarreau9650f372009-08-16 14:02:45 +0200361 switch (be->options & PR_O_TPXY_MASK) {
362 case PR_O_TPXY_ADDR:
Willy Tarreau9650f372009-08-16 14:02:45 +0200363 case PR_O_TPXY_CLI:
Willy Tarreaub1d67742010-03-29 19:36:59 +0200364 flags = 3;
365 break;
Willy Tarreau9650f372009-08-16 14:02:45 +0200366 case PR_O_TPXY_CIP:
Willy Tarreau090466c2009-09-07 11:51:47 +0200367 case PR_O_TPXY_DYN:
Willy Tarreaub1d67742010-03-29 19:36:59 +0200368 flags = 1;
Willy Tarreau9650f372009-08-16 14:02:45 +0200369 break;
370 }
Willy Tarreaub1d67742010-03-29 19:36:59 +0200371
Willy Tarreau9650f372009-08-16 14:02:45 +0200372#ifdef SO_BINDTODEVICE
373 /* Note: this might fail if not CAP_NET_RAW */
374 if (be->iface_name)
375 setsockopt(fd, SOL_SOCKET, SO_BINDTODEVICE, be->iface_name, be->iface_len + 1);
376#endif
Willy Tarreau6471afb2011-09-23 10:54:59 +0200377 ret = tcp_bind_socket(fd, flags, &be->source_addr, &si->addr.from);
Willy Tarreau9650f372009-08-16 14:02:45 +0200378 if (ret) {
379 close(fd);
380 if (ret == 1) {
381 Alert("Cannot bind to source address before connect() for proxy %s. Aborting.\n",
382 be->id);
383 send_log(be, LOG_EMERG,
384 "Cannot bind to source address before connect() for proxy %s.\n",
385 be->id);
386 } else {
387 Alert("Cannot bind to tproxy source address before connect() for proxy %s. Aborting.\n",
388 be->id);
389 send_log(be, LOG_EMERG,
390 "Cannot bind to tproxy source address before connect() for proxy %s.\n",
391 be->id);
392 }
393 return SN_ERR_RESOURCE;
394 }
395 }
396
Dmitry Sivachenkocaf58982009-08-24 15:11:06 +0400397#if defined(TCP_QUICKACK)
Willy Tarreau9650f372009-08-16 14:02:45 +0200398 /* disabling tcp quick ack now allows the first request to leave the
399 * machine with the first ACK. We only do this if there are pending
400 * data in the buffer.
401 */
Willy Tarreau2e046c62012-03-01 16:08:30 +0100402 if ((be->options2 & PR_O2_SMARTCON) && si->ob->o)
Simon Hormande072bd2011-06-24 15:11:37 +0900403 setsockopt(fd, IPPROTO_TCP, TCP_QUICKACK, &zero, sizeof(zero));
Willy Tarreau9650f372009-08-16 14:02:45 +0200404#endif
405
Willy Tarreaue803de22010-01-21 17:43:04 +0100406 if (global.tune.server_sndbuf)
407 setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &global.tune.server_sndbuf, sizeof(global.tune.server_sndbuf));
408
409 if (global.tune.server_rcvbuf)
410 setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &global.tune.server_rcvbuf, sizeof(global.tune.server_rcvbuf));
411
Willy Tarreau9b061e32012-04-07 18:03:52 +0200412 si->flags &= ~SI_FL_FROM_SET;
Willy Tarreau6471afb2011-09-23 10:54:59 +0200413 if ((connect(fd, (struct sockaddr *)&si->addr.to, get_addr_len(&si->addr.to)) == -1) &&
Willy Tarreau9650f372009-08-16 14:02:45 +0200414 (errno != EINPROGRESS) && (errno != EALREADY) && (errno != EISCONN)) {
415
416 if (errno == EAGAIN || errno == EADDRINUSE) {
417 char *msg;
418 if (errno == EAGAIN) /* no free ports left, try again later */
419 msg = "no free ports";
420 else
421 msg = "local address already in use";
422
423 qfprintf(stderr,"Cannot connect: %s.\n",msg);
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200424 port_range_release_port(fdinfo[fd].port_range, fdinfo[fd].local_port);
425 fdinfo[fd].port_range = NULL;
Willy Tarreau9650f372009-08-16 14:02:45 +0200426 close(fd);
427 send_log(be, LOG_EMERG,
428 "Connect() failed for server %s/%s: %s.\n",
429 be->id, srv->id, msg);
430 return SN_ERR_RESOURCE;
431 } else if (errno == ETIMEDOUT) {
432 //qfprintf(stderr,"Connect(): ETIMEDOUT");
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200433 port_range_release_port(fdinfo[fd].port_range, fdinfo[fd].local_port);
434 fdinfo[fd].port_range = NULL;
Willy Tarreau9650f372009-08-16 14:02:45 +0200435 close(fd);
436 return SN_ERR_SRVTO;
437 } else {
438 // (errno == ECONNREFUSED || errno == ENETUNREACH || errno == EACCES || errno == EPERM)
439 //qfprintf(stderr,"Connect(): %d", errno);
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200440 port_range_release_port(fdinfo[fd].port_range, fdinfo[fd].local_port);
441 fdinfo[fd].port_range = NULL;
Willy Tarreau9650f372009-08-16 14:02:45 +0200442 close(fd);
443 return SN_ERR_SRVCL;
444 }
445 }
446
William Lallemandb7ff6a32012-03-02 14:35:21 +0100447 /* needs src ip/port for logging */
Willy Tarreau9b061e32012-04-07 18:03:52 +0200448 if (si->flags & SI_FL_SRC_ADDR)
Willy Tarreau59b94792012-05-11 16:16:40 +0200449 si_get_from_addr(si);
William Lallemandb7ff6a32012-03-02 14:35:21 +0100450
Willy Tarreau9650f372009-08-16 14:02:45 +0200451 fdtab[fd].owner = si;
452 fdtab[fd].state = FD_STCONN; /* connection in progress */
453 fdtab[fd].flags = FD_FL_TCP | FD_FL_TCP_NODELAY;
Willy Tarreau9650f372009-08-16 14:02:45 +0200454 fdtab[fd].cb[DIR_RD].b = si->ib;
Willy Tarreau9650f372009-08-16 14:02:45 +0200455 fdtab[fd].cb[DIR_WR].b = si->ob;
456
Willy Tarreaueeda90e2012-05-11 19:53:32 +0200457 /* If we have nothing to send or if we want to initialize the sock layer,
458 * we want to confirm that the TCP connection is established before doing
459 * so, so we use our own write callback then switch to the sock layer.
460 */
461 if (si->sock.init || ((si->ob->flags & BF_OUT_EMPTY) && !si->send_proxy_ofs)) {
462 fdtab[fd].cb[DIR_RD].f = tcp_connect_read;
463 fdtab[fd].cb[DIR_WR].f = tcp_connect_write;
464 }
465 else {
466 fdtab[fd].cb[DIR_RD].f = si->sock.read;
467 fdtab[fd].cb[DIR_WR].f = si->sock.write;
468 }
469
Willy Tarreau6471afb2011-09-23 10:54:59 +0200470 fdinfo[fd].peeraddr = (struct sockaddr *)&si->addr.to;
471 fdinfo[fd].peerlen = get_addr_len(&si->addr.to);
Willy Tarreau9650f372009-08-16 14:02:45 +0200472
473 fd_insert(fd);
474 EV_FD_SET(fd, DIR_WR); /* for connect status */
475
476 si->state = SI_ST_CON;
477 si->flags |= SI_FL_CAP_SPLTCP; /* TCP supports splicing */
478 si->exp = tick_add_ifset(now_ms, be->timeout.connect);
479
480 return SN_ERR_NONE; /* connection is OK */
481}
482
483
Willy Tarreau59b94792012-05-11 16:16:40 +0200484/*
485 * Retrieves the source address for the socket <fd>, with <dir> indicating
486 * if we're a listener (=0) or an initiator (!=0). It returns 0 in case of
487 * success, -1 in case of error. The socket's source address is stored in
488 * <sa> for <salen> bytes.
489 */
490int tcp_get_src(int fd, struct sockaddr *sa, socklen_t salen, int dir)
491{
492 if (dir)
493 return getsockname(fd, sa, &salen);
494 else
495 return getpeername(fd, sa, &salen);
496}
497
498
499/*
500 * Retrieves the original destination address for the socket <fd>, with <dir>
501 * indicating if we're a listener (=0) or an initiator (!=0). In the case of a
502 * listener, if the original destination address was translated, the original
503 * address is retrieved. It returns 0 in case of success, -1 in case of error.
504 * The socket's source address is stored in <sa> for <salen> bytes.
505 */
506int tcp_get_dst(int fd, struct sockaddr *sa, socklen_t salen, int dir)
507{
508 if (dir)
509 return getpeername(fd, sa, &salen);
510#if defined(TPROXY) && defined(SO_ORIGINAL_DST)
511 else if (getsockopt(fd, SOL_IP, SO_ORIGINAL_DST, sa, &salen) == 0)
512 return 0;
513#endif
514 else
515 return getsockname(fd, sa, &salen);
516}
517
Willy Tarreaueeda90e2012-05-11 19:53:32 +0200518/* This is the callback which is set when a connection establishment is pending
519 * and we have nothing to send, or if we have an init function we want to call
520 * once the connection is established.
521 */
522static int tcp_connect_write(int fd)
523{
524 struct stream_interface *si = fdtab[fd].owner;
525 struct buffer *b = si->ob;
526 int retval = 1;
527
528 if (fdtab[fd].state == FD_STERROR)
529 goto out_error;
530
531 if (fdtab[fd].state != FD_STCONN) {
532 retval = 0;
533 goto out_ignore; /* strange we were called while ready */
534 }
535
536 /* we might have been called just after an asynchronous shutw */
537 if (b->flags & BF_SHUTW)
538 goto out_wakeup;
539
540 /* We have no data to send to check the connection, and
541 * getsockopt() will not inform us whether the connection
542 * is still pending. So we'll reuse connect() to check the
543 * state of the socket. This has the advantage of giving us
544 * the following info :
545 * - error
546 * - connecting (EALREADY, EINPROGRESS)
547 * - connected (EISCONN, 0)
548 */
549 if ((connect(fd, fdinfo[fd].peeraddr, fdinfo[fd].peerlen) == 0))
550 errno = 0;
551
552 if (errno == EALREADY || errno == EINPROGRESS) {
553 retval = 0;
554 goto out_ignore;
555 }
556
557 if (errno && errno != EISCONN)
558 goto out_error;
559
560 /* OK we just need to indicate that we got a connection
561 * and that we wrote nothing.
562 */
563 b->flags |= BF_WRITE_NULL;
564
565 /* The FD is ready now, we can hand the handlers to the socket layer */
566 fdtab[fd].cb[DIR_RD].f = si->sock.read;
567 fdtab[fd].cb[DIR_WR].f = si->sock.write;
568 fdtab[fd].state = FD_STREADY;
569
570 out_wakeup:
571 task_wakeup(si->owner, TASK_WOKEN_IO);
572
573 out_ignore:
574 fdtab[fd].ev &= ~FD_POLL_OUT;
575 return retval;
576
577 out_error:
578 /* Write error on the file descriptor. We mark the FD as STERROR so
579 * that we don't use it anymore. The error is reported to the stream
580 * interface which will take proper action. We must not perturbate the
581 * buffer because the stream interface wants to ensure transparent
582 * connection retries.
583 */
584
585 fdtab[fd].state = FD_STERROR;
586 fdtab[fd].ev &= ~FD_POLL_STICKY;
587 EV_FD_REM(fd);
588 si->flags |= SI_FL_ERR;
589 goto out_wakeup;
590}
591
592
593/* might be used on connect error */
594static int tcp_connect_read(int fd)
595{
596 struct stream_interface *si = fdtab[fd].owner;
597 int retval;
598
599 retval = 1;
600
601 if (fdtab[fd].state == FD_STERROR)
602 goto out_error;
603
604 if (fdtab[fd].state != FD_STCONN) {
605 retval = 0;
606 goto out_ignore; /* strange we were called while ready */
607 }
608
609 /* stop here if we reached the end of data */
610 if ((fdtab[fd].ev & (FD_POLL_IN|FD_POLL_HUP)) == FD_POLL_HUP)
611 goto out_error;
612
613 out_wakeup:
614 task_wakeup(si->owner, TASK_WOKEN_IO);
615 out_ignore:
616 fdtab[fd].ev &= ~FD_POLL_IN;
617 return retval;
618
619 out_error:
620 /* Read error on the file descriptor. We mark the FD as STERROR so
621 * that we don't use it anymore. The error is reported to the stream
622 * interface which will take proper action. We must not perturbate the
623 * buffer because the stream interface wants to ensure transparent
624 * connection retries.
625 */
626
627 fdtab[fd].state = FD_STERROR;
628 fdtab[fd].ev &= ~FD_POLL_STICKY;
629 EV_FD_REM(fd);
630 si->flags |= SI_FL_ERR;
631 goto out_wakeup;
632}
633
Willy Tarreau59b94792012-05-11 16:16:40 +0200634
Willy Tarreaue6b98942007-10-29 01:09:36 +0100635/* This function tries to bind a TCPv4/v6 listener. It may return a warning or
636 * an error message in <err> if the message is at most <errlen> bytes long
637 * (including '\0'). The return value is composed from ERR_ABORT, ERR_WARN,
638 * ERR_ALERT, ERR_RETRYABLE and ERR_FATAL. ERR_NONE indicates that everything
639 * was alright and that no message was returned. ERR_RETRYABLE means that an
640 * error occurred but that it may vanish after a retry (eg: port in use), and
Aman Guptad94991d2012-04-06 17:39:26 -0700641 * ERR_FATAL indicates a non-fixable error. ERR_WARN and ERR_ALERT do not alter
Willy Tarreaue6b98942007-10-29 01:09:36 +0100642 * the meaning of the error, but just indicate that a message is present which
643 * should be displayed with the respective level. Last, ERR_ABORT indicates
644 * that it's pointless to try to start other listeners. No error message is
645 * returned if errlen is NULL.
646 */
647int tcp_bind_listener(struct listener *listener, char *errmsg, int errlen)
648{
649 __label__ tcp_return, tcp_close_return;
650 int fd, err;
651 const char *msg = NULL;
652
653 /* ensure we never return garbage */
654 if (errmsg && errlen)
655 *errmsg = 0;
656
657 if (listener->state != LI_ASSIGNED)
658 return ERR_NONE; /* already bound */
659
660 err = ERR_NONE;
661
662 if ((fd = socket(listener->addr.ss_family, SOCK_STREAM, IPPROTO_TCP)) == -1) {
663 err |= ERR_RETRYABLE | ERR_ALERT;
664 msg = "cannot create listening socket";
665 goto tcp_return;
666 }
Willy Tarreauedcf6682008-11-30 23:15:34 +0100667
Willy Tarreaue6b98942007-10-29 01:09:36 +0100668 if (fd >= global.maxsock) {
669 err |= ERR_FATAL | ERR_ABORT | ERR_ALERT;
670 msg = "not enough free sockets (raise '-n' parameter)";
671 goto tcp_close_return;
672 }
673
Willy Tarreaufb14edc2009-06-14 15:24:37 +0200674 if (fcntl(fd, F_SETFL, O_NONBLOCK) == -1) {
Willy Tarreaue6b98942007-10-29 01:09:36 +0100675 err |= ERR_FATAL | ERR_ALERT;
676 msg = "cannot make socket non-blocking";
677 goto tcp_close_return;
678 }
679
Simon Hormande072bd2011-06-24 15:11:37 +0900680 if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one)) == -1) {
Willy Tarreaue6b98942007-10-29 01:09:36 +0100681 /* not fatal but should be reported */
682 msg = "cannot do so_reuseaddr";
683 err |= ERR_ALERT;
684 }
685
686 if (listener->options & LI_O_NOLINGER)
Simon Hormande072bd2011-06-24 15:11:37 +0900687 setsockopt(fd, SOL_SOCKET, SO_LINGER, &nolinger, sizeof(struct linger));
Willy Tarreauedcf6682008-11-30 23:15:34 +0100688
Willy Tarreaue6b98942007-10-29 01:09:36 +0100689#ifdef SO_REUSEPORT
690 /* OpenBSD supports this. As it's present in old libc versions of Linux,
691 * it might return an error that we will silently ignore.
692 */
Simon Hormande072bd2011-06-24 15:11:37 +0900693 setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &one, sizeof(one));
Willy Tarreaue6b98942007-10-29 01:09:36 +0100694#endif
Willy Tarreaub1e52e82008-01-13 14:49:51 +0100695#ifdef CONFIG_HAP_LINUX_TPROXY
Willy Tarreauedcf6682008-11-30 23:15:34 +0100696 if ((listener->options & LI_O_FOREIGN)
Simon Hormande072bd2011-06-24 15:11:37 +0900697 && (setsockopt(fd, SOL_IP, IP_TRANSPARENT, &one, sizeof(one)) == -1)
698 && (setsockopt(fd, SOL_IP, IP_FREEBIND, &one, sizeof(one)) == -1)) {
Willy Tarreaub1e52e82008-01-13 14:49:51 +0100699 msg = "cannot make listening socket transparent";
700 err |= ERR_ALERT;
701 }
702#endif
Willy Tarreau5e6e2042009-02-04 17:19:29 +0100703#ifdef SO_BINDTODEVICE
704 /* Note: this might fail if not CAP_NET_RAW */
705 if (listener->interface) {
706 if (setsockopt(fd, SOL_SOCKET, SO_BINDTODEVICE,
Willy Tarreau604e8302009-03-06 00:48:23 +0100707 listener->interface, strlen(listener->interface) + 1) == -1) {
Willy Tarreau5e6e2042009-02-04 17:19:29 +0100708 msg = "cannot bind listener to device";
709 err |= ERR_WARN;
710 }
711 }
712#endif
Dmitry Sivachenkocaf58982009-08-24 15:11:06 +0400713#if defined(TCP_MAXSEG)
Willy Tarreau48a7e722010-12-24 15:26:39 +0100714 if (listener->maxseg > 0) {
Dmitry Sivachenkocaf58982009-08-24 15:11:06 +0400715 if (setsockopt(fd, IPPROTO_TCP, TCP_MAXSEG,
Willy Tarreaube1b9182009-06-14 18:48:19 +0200716 &listener->maxseg, sizeof(listener->maxseg)) == -1) {
717 msg = "cannot set MSS";
718 err |= ERR_WARN;
719 }
720 }
721#endif
Willy Tarreaucb6cd432009-10-13 07:34:14 +0200722#if defined(TCP_DEFER_ACCEPT)
723 if (listener->options & LI_O_DEF_ACCEPT) {
724 /* defer accept by up to one second */
725 int accept_delay = 1;
726 if (setsockopt(fd, IPPROTO_TCP, TCP_DEFER_ACCEPT, &accept_delay, sizeof(accept_delay)) == -1) {
727 msg = "cannot enable DEFER_ACCEPT";
728 err |= ERR_WARN;
729 }
730 }
731#endif
Willy Tarreaue6b98942007-10-29 01:09:36 +0100732 if (bind(fd, (struct sockaddr *)&listener->addr, listener->proto->sock_addrlen) == -1) {
733 err |= ERR_RETRYABLE | ERR_ALERT;
734 msg = "cannot bind socket";
735 goto tcp_close_return;
736 }
Willy Tarreauedcf6682008-11-30 23:15:34 +0100737
Willy Tarreauc73ce2b2008-01-06 10:55:10 +0100738 if (listen(fd, listener->backlog ? listener->backlog : listener->maxconn) == -1) {
Willy Tarreaue6b98942007-10-29 01:09:36 +0100739 err |= ERR_RETRYABLE | ERR_ALERT;
740 msg = "cannot listen to socket";
741 goto tcp_close_return;
742 }
Willy Tarreauedcf6682008-11-30 23:15:34 +0100743
Dmitry Sivachenkocaf58982009-08-24 15:11:06 +0400744#if defined(TCP_QUICKACK)
Willy Tarreau9ea05a72009-06-14 12:07:01 +0200745 if (listener->options & LI_O_NOQUICKACK)
Simon Hormande072bd2011-06-24 15:11:37 +0900746 setsockopt(fd, IPPROTO_TCP, TCP_QUICKACK, &zero, sizeof(zero));
Willy Tarreau9ea05a72009-06-14 12:07:01 +0200747#endif
748
Willy Tarreaue6b98942007-10-29 01:09:36 +0100749 /* the socket is ready */
750 listener->fd = fd;
751 listener->state = LI_LISTEN;
752
Willy Tarreaueabf3132008-08-29 23:36:51 +0200753 fdtab[fd].owner = listener; /* reference the listener instead of a task */
Willy Tarreaue6b98942007-10-29 01:09:36 +0100754 fdtab[fd].state = FD_STLISTEN;
Willy Tarreaueb472682010-05-28 18:46:57 +0200755 fdtab[fd].flags = FD_FL_TCP | ((listener->options & LI_O_NOLINGER) ? FD_FL_TCP_NOLING : 0);
756 fdtab[fd].cb[DIR_RD].f = listener->proto->accept;
757 fdtab[fd].cb[DIR_WR].f = NULL; /* never called */
758 fdtab[fd].cb[DIR_RD].b = fdtab[fd].cb[DIR_WR].b = NULL;
Willy Tarreau5d707e12009-06-28 11:09:07 +0200759
Willy Tarreau8d5d77e2009-10-18 07:25:52 +0200760 fdinfo[fd].peeraddr = NULL;
761 fdinfo[fd].peerlen = 0;
Willy Tarreaueb472682010-05-28 18:46:57 +0200762 fd_insert(fd);
763
Willy Tarreaue6b98942007-10-29 01:09:36 +0100764 tcp_return:
Cyril Bonté43ba1b32010-11-01 19:26:01 +0100765 if (msg && errlen) {
766 char pn[INET6_ADDRSTRLEN];
767
Willy Tarreau631f01c2011-09-05 00:36:48 +0200768 addr_to_str(&listener->addr, pn, sizeof(pn));
769 snprintf(errmsg, errlen, "%s [%s:%d]", msg, pn, get_host_port(&listener->addr));
Cyril Bonté43ba1b32010-11-01 19:26:01 +0100770 }
Willy Tarreaue6b98942007-10-29 01:09:36 +0100771 return err;
772
773 tcp_close_return:
774 close(fd);
775 goto tcp_return;
776}
777
778/* This function creates all TCP sockets bound to the protocol entry <proto>.
779 * It is intended to be used as the protocol's bind_all() function.
780 * The sockets will be registered but not added to any fd_set, in order not to
781 * loose them across the fork(). A call to enable_all_listeners() is needed
782 * to complete initialization. The return value is composed from ERR_*.
783 */
Emeric Bruncf20bf12010-10-22 16:06:11 +0200784static int tcp_bind_listeners(struct protocol *proto, char *errmsg, int errlen)
Willy Tarreaue6b98942007-10-29 01:09:36 +0100785{
786 struct listener *listener;
787 int err = ERR_NONE;
788
789 list_for_each_entry(listener, &proto->listeners, proto_list) {
Emeric Bruncf20bf12010-10-22 16:06:11 +0200790 err |= tcp_bind_listener(listener, errmsg, errlen);
791 if (err & ERR_ABORT)
Willy Tarreaue6b98942007-10-29 01:09:36 +0100792 break;
793 }
794
795 return err;
796}
797
798/* Add listener to the list of tcpv4 listeners. The listener's state
799 * is automatically updated from LI_INIT to LI_ASSIGNED. The number of
800 * listeners is updated. This is the function to use to add a new listener.
801 */
802void tcpv4_add_listener(struct listener *listener)
803{
804 if (listener->state != LI_INIT)
805 return;
806 listener->state = LI_ASSIGNED;
807 listener->proto = &proto_tcpv4;
808 LIST_ADDQ(&proto_tcpv4.listeners, &listener->proto_list);
809 proto_tcpv4.nb_listeners++;
810}
811
812/* Add listener to the list of tcpv4 listeners. The listener's state
813 * is automatically updated from LI_INIT to LI_ASSIGNED. The number of
814 * listeners is updated. This is the function to use to add a new listener.
815 */
816void tcpv6_add_listener(struct listener *listener)
817{
818 if (listener->state != LI_INIT)
819 return;
820 listener->state = LI_ASSIGNED;
821 listener->proto = &proto_tcpv6;
822 LIST_ADDQ(&proto_tcpv6.listeners, &listener->proto_list);
823 proto_tcpv6.nb_listeners++;
824}
825
Willy Tarreauedcf6682008-11-30 23:15:34 +0100826/* This function performs the TCP request analysis on the current request. It
827 * returns 1 if the processing can continue on next analysers, or zero if it
828 * needs more data, encounters an error, or wants to immediately abort the
Willy Tarreaufb356202010-08-03 14:02:05 +0200829 * request. It relies on buffers flags, and updates s->req->analysers. The
830 * function may be called for frontend rules and backend rules. It only relies
831 * on the backend pointer so this works for both cases.
Willy Tarreauedcf6682008-11-30 23:15:34 +0100832 */
Willy Tarreau3a816292009-07-07 10:55:49 +0200833int tcp_inspect_request(struct session *s, struct buffer *req, int an_bit)
Willy Tarreauedcf6682008-11-30 23:15:34 +0100834{
835 struct tcp_rule *rule;
Willy Tarreaud1f96522010-08-03 19:34:32 +0200836 struct stksess *ts;
837 struct stktable *t;
Willy Tarreauedcf6682008-11-30 23:15:34 +0100838 int partial;
839
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100840 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bh=%d analysers=%02x\n",
Willy Tarreauedcf6682008-11-30 23:15:34 +0100841 now_ms, __FUNCTION__,
842 s,
843 req,
844 req->rex, req->wex,
845 req->flags,
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100846 req->i,
Willy Tarreauedcf6682008-11-30 23:15:34 +0100847 req->analysers);
848
Willy Tarreauedcf6682008-11-30 23:15:34 +0100849 /* We don't know whether we have enough data, so must proceed
850 * this way :
851 * - iterate through all rules in their declaration order
852 * - if one rule returns MISS, it means the inspect delay is
853 * not over yet, then return immediately, otherwise consider
854 * it as a non-match.
855 * - if one rule returns OK, then return OK
856 * - if one rule returns KO, then return KO
857 */
858
Willy Tarreaub824b002010-09-29 16:36:16 +0200859 if (req->flags & (BF_SHUTR|BF_FULL) || !s->be->tcp_req.inspect_delay || tick_is_expired(req->analyse_exp, now_ms))
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200860 partial = SMP_OPT_FINAL;
Willy Tarreauedcf6682008-11-30 23:15:34 +0100861 else
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200862 partial = 0;
Willy Tarreauedcf6682008-11-30 23:15:34 +0100863
Willy Tarreaufb356202010-08-03 14:02:05 +0200864 list_for_each_entry(rule, &s->be->tcp_req.inspect_rules, list) {
Willy Tarreauedcf6682008-11-30 23:15:34 +0100865 int ret = ACL_PAT_PASS;
866
867 if (rule->cond) {
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200868 ret = acl_exec_cond(rule->cond, s->be, s, &s->txn, SMP_OPT_DIR_REQ | partial);
Willy Tarreauedcf6682008-11-30 23:15:34 +0100869 if (ret == ACL_PAT_MISS) {
Willy Tarreau520d95e2009-09-19 21:04:57 +0200870 buffer_dont_connect(req);
Willy Tarreauedcf6682008-11-30 23:15:34 +0100871 /* just set the request timeout once at the beginning of the request */
Willy Tarreaufb356202010-08-03 14:02:05 +0200872 if (!tick_isset(req->analyse_exp) && s->be->tcp_req.inspect_delay)
873 req->analyse_exp = tick_add_ifset(now_ms, s->be->tcp_req.inspect_delay);
Willy Tarreauedcf6682008-11-30 23:15:34 +0100874 return 0;
875 }
876
877 ret = acl_pass(ret);
878 if (rule->cond->pol == ACL_COND_UNLESS)
879 ret = !ret;
880 }
881
882 if (ret) {
883 /* we have a matching rule. */
884 if (rule->action == TCP_ACT_REJECT) {
885 buffer_abort(req);
886 buffer_abort(s->rep);
887 req->analysers = 0;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200888
Willy Tarreau7d0aaf32011-03-10 23:25:56 +0100889 s->be->be_counters.denied_req++;
890 s->fe->fe_counters.denied_req++;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200891 if (s->listener->counters)
Willy Tarreau23968d82010-05-23 23:50:44 +0200892 s->listener->counters->denied_req++;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200893
Willy Tarreauedcf6682008-11-30 23:15:34 +0100894 if (!(s->flags & SN_ERR_MASK))
895 s->flags |= SN_ERR_PRXCOND;
896 if (!(s->flags & SN_FINST_MASK))
897 s->flags |= SN_FINST_R;
898 return 0;
899 }
Willy Tarreau56123282010-08-06 19:06:56 +0200900 else if (rule->action == TCP_ACT_TRK_SC1) {
901 if (!s->stkctr1_entry) {
902 /* only the first valid track-sc1 directive applies.
Willy Tarreaud1f96522010-08-03 19:34:32 +0200903 * Also, note that right now we can only track SRC so we
904 * don't check how to get the key, but later we may need
905 * to consider rule->act_prm->trk_ctr.type.
906 */
907 t = rule->act_prm.trk_ctr.table.t;
David du Colombier4f92d322011-03-24 11:09:31 +0100908 ts = stktable_get_entry(t, tcp_src_to_stktable_key(s));
Willy Tarreau0a4838c2010-08-06 20:11:05 +0200909 if (ts) {
Willy Tarreau56123282010-08-06 19:06:56 +0200910 session_track_stkctr1(s, t, ts);
Willy Tarreau0a4838c2010-08-06 20:11:05 +0200911 if (s->fe != s->be)
912 s->flags |= SN_BE_TRACK_SC1;
913 }
Willy Tarreaud1f96522010-08-03 19:34:32 +0200914 }
915 }
Willy Tarreau56123282010-08-06 19:06:56 +0200916 else if (rule->action == TCP_ACT_TRK_SC2) {
917 if (!s->stkctr2_entry) {
918 /* only the first valid track-sc2 directive applies.
Willy Tarreaud1f96522010-08-03 19:34:32 +0200919 * Also, note that right now we can only track SRC so we
920 * don't check how to get the key, but later we may need
921 * to consider rule->act_prm->trk_ctr.type.
922 */
923 t = rule->act_prm.trk_ctr.table.t;
David du Colombier4f92d322011-03-24 11:09:31 +0100924 ts = stktable_get_entry(t, tcp_src_to_stktable_key(s));
Willy Tarreau0a4838c2010-08-06 20:11:05 +0200925 if (ts) {
Willy Tarreau56123282010-08-06 19:06:56 +0200926 session_track_stkctr2(s, t, ts);
Willy Tarreau0a4838c2010-08-06 20:11:05 +0200927 if (s->fe != s->be)
928 s->flags |= SN_BE_TRACK_SC2;
929 }
Willy Tarreaud1f96522010-08-03 19:34:32 +0200930 }
931 }
932 else {
Willy Tarreauedcf6682008-11-30 23:15:34 +0100933 /* otherwise accept */
Willy Tarreaud1f96522010-08-03 19:34:32 +0200934 break;
935 }
Willy Tarreauedcf6682008-11-30 23:15:34 +0100936 }
937 }
938
939 /* if we get there, it means we have no rule which matches, or
940 * we have an explicit accept, so we apply the default accept.
941 */
Willy Tarreau3a816292009-07-07 10:55:49 +0200942 req->analysers &= ~an_bit;
Willy Tarreauedcf6682008-11-30 23:15:34 +0100943 req->analyse_exp = TICK_ETERNITY;
944 return 1;
945}
946
Emeric Brun97679e72010-09-23 17:56:44 +0200947/* This function performs the TCP response analysis on the current response. It
948 * returns 1 if the processing can continue on next analysers, or zero if it
949 * needs more data, encounters an error, or wants to immediately abort the
950 * response. It relies on buffers flags, and updates s->rep->analysers. The
951 * function may be called for backend rules.
952 */
953int tcp_inspect_response(struct session *s, struct buffer *rep, int an_bit)
954{
955 struct tcp_rule *rule;
956 int partial;
957
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100958 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bh=%d analysers=%02x\n",
Emeric Brun97679e72010-09-23 17:56:44 +0200959 now_ms, __FUNCTION__,
960 s,
961 rep,
962 rep->rex, rep->wex,
963 rep->flags,
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100964 rep->i,
Emeric Brun97679e72010-09-23 17:56:44 +0200965 rep->analysers);
966
967 /* We don't know whether we have enough data, so must proceed
968 * this way :
969 * - iterate through all rules in their declaration order
970 * - if one rule returns MISS, it means the inspect delay is
971 * not over yet, then return immediately, otherwise consider
972 * it as a non-match.
973 * - if one rule returns OK, then return OK
974 * - if one rule returns KO, then return KO
975 */
976
977 if (rep->flags & BF_SHUTR || tick_is_expired(rep->analyse_exp, now_ms))
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200978 partial = SMP_OPT_FINAL;
Emeric Brun97679e72010-09-23 17:56:44 +0200979 else
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200980 partial = 0;
Emeric Brun97679e72010-09-23 17:56:44 +0200981
982 list_for_each_entry(rule, &s->be->tcp_rep.inspect_rules, list) {
983 int ret = ACL_PAT_PASS;
984
985 if (rule->cond) {
Willy Tarreau32a6f2e2012-04-25 10:13:36 +0200986 ret = acl_exec_cond(rule->cond, s->be, s, &s->txn, SMP_OPT_DIR_RES | partial);
Emeric Brun97679e72010-09-23 17:56:44 +0200987 if (ret == ACL_PAT_MISS) {
988 /* just set the analyser timeout once at the beginning of the response */
989 if (!tick_isset(rep->analyse_exp) && s->be->tcp_rep.inspect_delay)
990 rep->analyse_exp = tick_add_ifset(now_ms, s->be->tcp_rep.inspect_delay);
991 return 0;
992 }
993
994 ret = acl_pass(ret);
995 if (rule->cond->pol == ACL_COND_UNLESS)
996 ret = !ret;
997 }
998
999 if (ret) {
1000 /* we have a matching rule. */
1001 if (rule->action == TCP_ACT_REJECT) {
1002 buffer_abort(rep);
1003 buffer_abort(s->req);
1004 rep->analysers = 0;
1005
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001006 s->be->be_counters.denied_resp++;
1007 s->fe->fe_counters.denied_resp++;
Emeric Brun97679e72010-09-23 17:56:44 +02001008 if (s->listener->counters)
1009 s->listener->counters->denied_resp++;
1010
1011 if (!(s->flags & SN_ERR_MASK))
1012 s->flags |= SN_ERR_PRXCOND;
1013 if (!(s->flags & SN_FINST_MASK))
1014 s->flags |= SN_FINST_D;
1015 return 0;
1016 }
1017 else {
1018 /* otherwise accept */
1019 break;
1020 }
1021 }
1022 }
1023
1024 /* if we get there, it means we have no rule which matches, or
1025 * we have an explicit accept, so we apply the default accept.
1026 */
1027 rep->analysers &= ~an_bit;
1028 rep->analyse_exp = TICK_ETERNITY;
1029 return 1;
1030}
1031
1032
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001033/* This function performs the TCP layer4 analysis on the current request. It
1034 * returns 0 if a reject rule matches, otherwise 1 if either an accept rule
1035 * matches or if no more rule matches. It can only use rules which don't need
1036 * any data.
1037 */
1038int tcp_exec_req_rules(struct session *s)
1039{
1040 struct tcp_rule *rule;
Willy Tarreauf059a0f2010-08-03 16:29:52 +02001041 struct stksess *ts;
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001042 struct stktable *t = NULL;
1043 int result = 1;
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001044 int ret;
1045
1046 list_for_each_entry(rule, &s->fe->tcp_req.l4_rules, list) {
1047 ret = ACL_PAT_PASS;
1048
1049 if (rule->cond) {
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001050 ret = acl_exec_cond(rule->cond, s->fe, s, NULL, SMP_OPT_DIR_REQ|SMP_OPT_FINAL);
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001051 ret = acl_pass(ret);
1052 if (rule->cond->pol == ACL_COND_UNLESS)
1053 ret = !ret;
1054 }
1055
1056 if (ret) {
1057 /* we have a matching rule. */
1058 if (rule->action == TCP_ACT_REJECT) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001059 s->fe->fe_counters.denied_conn++;
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001060 if (s->listener->counters)
Willy Tarreau2799e982010-06-05 15:43:21 +02001061 s->listener->counters->denied_conn++;
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001062
1063 if (!(s->flags & SN_ERR_MASK))
1064 s->flags |= SN_ERR_PRXCOND;
1065 if (!(s->flags & SN_FINST_MASK))
1066 s->flags |= SN_FINST_R;
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001067 result = 0;
1068 break;
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001069 }
Willy Tarreau56123282010-08-06 19:06:56 +02001070 else if (rule->action == TCP_ACT_TRK_SC1) {
1071 if (!s->stkctr1_entry) {
1072 /* only the first valid track-sc1 directive applies.
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001073 * Also, note that right now we can only track SRC so we
1074 * don't check how to get the key, but later we may need
1075 * to consider rule->act_prm->trk_ctr.type.
1076 */
1077 t = rule->act_prm.trk_ctr.table.t;
David du Colombier4f92d322011-03-24 11:09:31 +01001078 ts = stktable_get_entry(t, tcp_src_to_stktable_key(s));
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001079 if (ts)
Willy Tarreau56123282010-08-06 19:06:56 +02001080 session_track_stkctr1(s, t, ts);
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001081 }
1082 }
Willy Tarreau56123282010-08-06 19:06:56 +02001083 else if (rule->action == TCP_ACT_TRK_SC2) {
1084 if (!s->stkctr2_entry) {
1085 /* only the first valid track-sc2 directive applies.
Willy Tarreauf059a0f2010-08-03 16:29:52 +02001086 * Also, note that right now we can only track SRC so we
1087 * don't check how to get the key, but later we may need
1088 * to consider rule->act_prm->trk_ctr.type.
1089 */
1090 t = rule->act_prm.trk_ctr.table.t;
David du Colombier4f92d322011-03-24 11:09:31 +01001091 ts = stktable_get_entry(t, tcp_src_to_stktable_key(s));
Willy Tarreauf059a0f2010-08-03 16:29:52 +02001092 if (ts)
Willy Tarreau56123282010-08-06 19:06:56 +02001093 session_track_stkctr2(s, t, ts);
Willy Tarreauf059a0f2010-08-03 16:29:52 +02001094 }
1095 }
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001096 else {
1097 /* otherwise it's an accept */
1098 break;
1099 }
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001100 }
1101 }
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001102 return result;
Willy Tarreaua5c0ab22010-05-31 10:30:33 +02001103}
1104
Emeric Brun97679e72010-09-23 17:56:44 +02001105/* Parse a tcp-response rule. Return a negative value in case of failure */
1106static int tcp_parse_response_rule(char **args, int arg, int section_type,
1107 struct proxy *curpx, struct proxy *defpx,
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001108 struct tcp_rule *rule, char **err)
Emeric Brun97679e72010-09-23 17:56:44 +02001109{
1110 if (curpx == defpx || !(curpx->cap & PR_CAP_BE)) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001111 memprintf(err, "%s %s is only allowed in 'backend' sections",
1112 args[0], args[1]);
Emeric Brun97679e72010-09-23 17:56:44 +02001113 return -1;
1114 }
1115
1116 if (strcmp(args[arg], "accept") == 0) {
1117 arg++;
1118 rule->action = TCP_ACT_ACCEPT;
1119 }
1120 else if (strcmp(args[arg], "reject") == 0) {
1121 arg++;
1122 rule->action = TCP_ACT_REJECT;
1123 }
1124 else {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001125 memprintf(err,
1126 "'%s %s' expects 'accept' or 'reject' in %s '%s' (got '%s')",
1127 args[0], args[1], proxy_type_str(curpx), curpx->id, args[arg]);
Emeric Brun97679e72010-09-23 17:56:44 +02001128 return -1;
1129 }
1130
1131 if (strcmp(args[arg], "if") == 0 || strcmp(args[arg], "unless") == 0) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001132 if ((rule->cond = build_acl_cond(NULL, 0, curpx, (const char **)args+arg, err)) == NULL) {
1133 memprintf(err,
1134 "'%s %s %s' : error detected in %s '%s' while parsing '%s' condition : %s",
1135 args[0], args[1], args[2], proxy_type_str(curpx), curpx->id, args[arg], *err);
Emeric Brun97679e72010-09-23 17:56:44 +02001136 return -1;
1137 }
1138 }
1139 else if (*args[arg]) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001140 memprintf(err,
1141 "'%s %s %s' only accepts 'if' or 'unless', in %s '%s' (got '%s')",
Emeric Brun97679e72010-09-23 17:56:44 +02001142 args[0], args[1], args[2], proxy_type_str(curpx), curpx->id, args[arg]);
1143 return -1;
1144 }
1145 return 0;
1146}
1147
1148
1149
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001150/* Parse a tcp-request rule. Return a negative value in case of failure */
1151static int tcp_parse_request_rule(char **args, int arg, int section_type,
1152 struct proxy *curpx, struct proxy *defpx,
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001153 struct tcp_rule *rule, char **err)
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001154{
1155 if (curpx == defpx) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001156 memprintf(err, "%s %s is not allowed in 'defaults' sections",
1157 args[0], args[1]);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001158 return -1;
1159 }
1160
1161 if (!strcmp(args[arg], "accept")) {
1162 arg++;
1163 rule->action = TCP_ACT_ACCEPT;
1164 }
1165 else if (!strcmp(args[arg], "reject")) {
1166 arg++;
1167 rule->action = TCP_ACT_REJECT;
1168 }
Willy Tarreau56123282010-08-06 19:06:56 +02001169 else if (strcmp(args[arg], "track-sc1") == 0) {
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001170 int ret;
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001171 int kw = arg;
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001172
1173 arg++;
1174 ret = parse_track_counters(args, &arg, section_type, curpx,
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001175 &rule->act_prm.trk_ctr, defpx, err);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001176
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001177 if (ret < 0) { /* nb: warnings are not handled yet */
1178 memprintf(err,
1179 "'%s %s %s' : %s in %s '%s'",
1180 args[0], args[1], args[kw], *err, proxy_type_str(curpx), curpx->id);
1181 return ret;
1182 }
Willy Tarreau56123282010-08-06 19:06:56 +02001183 rule->action = TCP_ACT_TRK_SC1;
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001184 }
Willy Tarreau56123282010-08-06 19:06:56 +02001185 else if (strcmp(args[arg], "track-sc2") == 0) {
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001186 int ret;
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001187 int kw = arg;
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001188
1189 arg++;
1190 ret = parse_track_counters(args, &arg, section_type, curpx,
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001191 &rule->act_prm.trk_ctr, defpx, err);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001192
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001193 if (ret < 0) { /* nb: warnings are not handled yet */
1194 memprintf(err,
1195 "'%s %s %s' : %s in %s '%s'",
1196 args[0], args[1], args[kw], *err, proxy_type_str(curpx), curpx->id);
1197 return ret;
1198 }
Willy Tarreau56123282010-08-06 19:06:56 +02001199 rule->action = TCP_ACT_TRK_SC2;
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001200 }
1201 else {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001202 memprintf(err,
1203 "'%s %s' expects 'accept', 'reject', 'track-sc1' "
1204 "or 'track-sc2' in %s '%s' (got '%s')",
1205 args[0], args[1], proxy_type_str(curpx), curpx->id, args[arg]);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001206 return -1;
1207 }
1208
1209 if (strcmp(args[arg], "if") == 0 || strcmp(args[arg], "unless") == 0) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001210 if ((rule->cond = build_acl_cond(NULL, 0, curpx, (const char **)args+arg, err)) == NULL) {
1211 memprintf(err,
1212 "'%s %s %s' : error detected in %s '%s' while parsing '%s' condition : %s",
1213 args[0], args[1], args[2], proxy_type_str(curpx), curpx->id, args[arg], *err);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001214 return -1;
1215 }
1216 }
1217 else if (*args[arg]) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001218 memprintf(err,
1219 "'%s %s %s' only accepts 'if' or 'unless', in %s '%s' (got '%s')",
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001220 args[0], args[1], args[2], proxy_type_str(curpx), curpx->id, args[arg]);
1221 return -1;
1222 }
1223 return 0;
1224}
1225
Emeric Brun97679e72010-09-23 17:56:44 +02001226/* This function should be called to parse a line starting with the "tcp-response"
1227 * keyword.
1228 */
1229static int tcp_parse_tcp_rep(char **args, int section_type, struct proxy *curpx,
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001230 struct proxy *defpx, char **err)
Emeric Brun97679e72010-09-23 17:56:44 +02001231{
1232 const char *ptr = NULL;
1233 unsigned int val;
Emeric Brun97679e72010-09-23 17:56:44 +02001234 int warn = 0;
1235 int arg;
1236 struct tcp_rule *rule;
1237
1238 if (!*args[1]) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001239 memprintf(err, "missing argument for '%s' in %s '%s'",
1240 args[0], proxy_type_str(curpx), curpx->id);
Emeric Brun97679e72010-09-23 17:56:44 +02001241 return -1;
1242 }
1243
1244 if (strcmp(args[1], "inspect-delay") == 0) {
1245 if (curpx == defpx || !(curpx->cap & PR_CAP_BE)) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001246 memprintf(err, "%s %s is only allowed in 'backend' sections",
1247 args[0], args[1]);
Emeric Brun97679e72010-09-23 17:56:44 +02001248 return -1;
1249 }
1250
1251 if (!*args[2] || (ptr = parse_time_err(args[2], &val, TIME_UNIT_MS))) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001252 memprintf(err,
1253 "'%s %s' expects a positive delay in milliseconds, in %s '%s'",
1254 args[0], args[1], proxy_type_str(curpx), curpx->id);
1255 if (ptr)
1256 memprintf(err, "%s (unexpected character '%c')", *err, *ptr);
Emeric Brun97679e72010-09-23 17:56:44 +02001257 return -1;
1258 }
1259
1260 if (curpx->tcp_rep.inspect_delay) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001261 memprintf(err, "ignoring %s %s (was already defined) in %s '%s'",
1262 args[0], args[1], proxy_type_str(curpx), curpx->id);
Emeric Brun97679e72010-09-23 17:56:44 +02001263 return 1;
1264 }
1265 curpx->tcp_rep.inspect_delay = val;
1266 return 0;
1267 }
1268
Simon Hormande072bd2011-06-24 15:11:37 +09001269 rule = calloc(1, sizeof(*rule));
Emeric Brun97679e72010-09-23 17:56:44 +02001270 LIST_INIT(&rule->list);
1271 arg = 1;
1272
1273 if (strcmp(args[1], "content") == 0) {
1274 arg++;
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001275 if (tcp_parse_response_rule(args, arg, section_type, curpx, defpx, rule, err) < 0)
Emeric Brun97679e72010-09-23 17:56:44 +02001276 goto error;
1277
1278 if (rule->cond && (rule->cond->requires & ACL_USE_L6REQ_VOLATILE)) {
1279 struct acl *acl;
1280 const char *name;
1281
1282 acl = cond_find_require(rule->cond, ACL_USE_L6REQ_VOLATILE);
1283 name = acl ? acl->name : "(unknown)";
1284
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001285 memprintf(err,
1286 "acl '%s' involves some request-only criteria which will be ignored in '%s %s'",
1287 name, args[0], args[1]);
Emeric Brun97679e72010-09-23 17:56:44 +02001288 warn++;
1289 }
1290
1291 LIST_ADDQ(&curpx->tcp_rep.inspect_rules, &rule->list);
1292 }
1293 else {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001294 memprintf(err,
1295 "'%s' expects 'inspect-delay' or 'content' in %s '%s' (got '%s')",
1296 args[0], proxy_type_str(curpx), curpx->id, args[1]);
Emeric Brun97679e72010-09-23 17:56:44 +02001297 goto error;
1298 }
1299
1300 return warn;
1301 error:
1302 free(rule);
1303 return -1;
1304}
1305
1306
Willy Tarreaub6866442008-07-14 23:54:42 +02001307/* This function should be called to parse a line starting with the "tcp-request"
1308 * keyword.
1309 */
1310static int tcp_parse_tcp_req(char **args, int section_type, struct proxy *curpx,
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001311 struct proxy *defpx, char **err)
Willy Tarreaub6866442008-07-14 23:54:42 +02001312{
1313 const char *ptr = NULL;
Willy Tarreauc7e961e2008-08-17 17:13:47 +02001314 unsigned int val;
Willy Tarreau1a687942010-05-23 22:40:30 +02001315 int warn = 0;
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001316 int arg;
Willy Tarreau1a687942010-05-23 22:40:30 +02001317 struct tcp_rule *rule;
Willy Tarreaub6866442008-07-14 23:54:42 +02001318
1319 if (!*args[1]) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001320 if (curpx == defpx)
1321 memprintf(err, "missing argument for '%s' in defaults section", args[0]);
1322 else
1323 memprintf(err, "missing argument for '%s' in %s '%s'",
1324 args[0], proxy_type_str(curpx), curpx->id);
Willy Tarreaub6866442008-07-14 23:54:42 +02001325 return -1;
1326 }
1327
1328 if (!strcmp(args[1], "inspect-delay")) {
1329 if (curpx == defpx) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001330 memprintf(err, "%s %s is not allowed in 'defaults' sections",
1331 args[0], args[1]);
Willy Tarreaub6866442008-07-14 23:54:42 +02001332 return -1;
1333 }
1334
Willy Tarreaub6866442008-07-14 23:54:42 +02001335 if (!*args[2] || (ptr = parse_time_err(args[2], &val, TIME_UNIT_MS))) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001336 memprintf(err,
1337 "'%s %s' expects a positive delay in milliseconds, in %s '%s'",
1338 args[0], args[1], proxy_type_str(curpx), curpx->id);
1339 if (ptr)
1340 memprintf(err, "%s (unexpected character '%c')", *err, *ptr);
Willy Tarreaub6866442008-07-14 23:54:42 +02001341 return -1;
1342 }
1343
1344 if (curpx->tcp_req.inspect_delay) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001345 memprintf(err, "ignoring %s %s (was already defined) in %s '%s'",
1346 args[0], args[1], proxy_type_str(curpx), curpx->id);
Willy Tarreaub6866442008-07-14 23:54:42 +02001347 return 1;
1348 }
1349 curpx->tcp_req.inspect_delay = val;
1350 return 0;
1351 }
1352
Simon Hormande072bd2011-06-24 15:11:37 +09001353 rule = calloc(1, sizeof(*rule));
Willy Tarreaufb024dc2010-08-20 13:35:41 +02001354 LIST_INIT(&rule->list);
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001355 arg = 1;
1356
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001357 if (strcmp(args[1], "content") == 0) {
Willy Tarreaud1f96522010-08-03 19:34:32 +02001358 arg++;
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001359 if (tcp_parse_request_rule(args, arg, section_type, curpx, defpx, rule, err) < 0)
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001360 goto error;
Willy Tarreaub6866442008-07-14 23:54:42 +02001361
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001362 if (rule->cond && (rule->cond->requires & ACL_USE_RTR_ANY)) {
Willy Tarreaudd64f8d2008-07-27 22:02:32 +02001363 struct acl *acl;
1364 const char *name;
1365
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001366 acl = cond_find_require(rule->cond, ACL_USE_RTR_ANY);
Willy Tarreaudd64f8d2008-07-27 22:02:32 +02001367 name = acl ? acl->name : "(unknown)";
1368
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001369 memprintf(err,
1370 "acl '%s' involves some response-only criteria which will be ignored in '%s %s'",
1371 name, args[0], args[1]);
Willy Tarreaudd64f8d2008-07-27 22:02:32 +02001372 warn++;
1373 }
Willy Tarreaufb024dc2010-08-20 13:35:41 +02001374 LIST_ADDQ(&curpx->tcp_req.inspect_rules, &rule->list);
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001375 }
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001376 else if (strcmp(args[1], "connection") == 0) {
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001377 arg++;
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001378
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001379 if (!(curpx->cap & PR_CAP_FE)) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001380 memprintf(err, "%s %s is not allowed because %s %s is not a frontend",
1381 args[0], args[1], proxy_type_str(curpx), curpx->id);
Simon Horman6c54d8b2011-07-15 13:14:06 +09001382 goto error;
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001383 }
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001384
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001385 if (tcp_parse_request_rule(args, arg, section_type, curpx, defpx, rule, err) < 0)
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001386 goto error;
1387
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001388 if (rule->cond && (rule->cond->requires & (ACL_USE_RTR_ANY|ACL_USE_L6_ANY|ACL_USE_L7_ANY))) {
1389 struct acl *acl;
1390 const char *name;
Willy Tarreauf059a0f2010-08-03 16:29:52 +02001391
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001392 acl = cond_find_require(rule->cond, ACL_USE_RTR_ANY|ACL_USE_L6_ANY|ACL_USE_L7_ANY);
1393 name = acl ? acl->name : "(unknown)";
Willy Tarreauf059a0f2010-08-03 16:29:52 +02001394
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001395 if (acl->requires & (ACL_USE_L6_ANY|ACL_USE_L7_ANY)) {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001396 memprintf(err,
1397 "'%s %s' may not reference acl '%s' which makes use of "
1398 "payload in %s '%s'. Please use '%s content' for this.",
1399 args[0], args[1], name, proxy_type_str(curpx), curpx->id, args[0]);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001400 goto error;
1401 }
1402 if (acl->requires & ACL_USE_RTR_ANY)
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001403 memprintf(err,
1404 "acl '%s' involves some response-only criteria which will be ignored in '%s %s'",
1405 name, args[0], args[1]);
Willy Tarreau68c03ab2010-08-06 15:08:45 +02001406 warn++;
1407 }
Willy Tarreaufb024dc2010-08-20 13:35:41 +02001408 LIST_ADDQ(&curpx->tcp_req.l4_rules, &rule->list);
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02001409 }
Willy Tarreau1a687942010-05-23 22:40:30 +02001410 else {
Willy Tarreau0a3dd742012-05-08 19:47:01 +02001411 if (curpx == defpx)
1412 memprintf(err,
1413 "'%s' expects 'inspect-delay', 'connection', or 'content' in defaults section (got '%s')",
1414 args[0], args[1]);
1415 else
1416 memprintf(err,
1417 "'%s' expects 'inspect-delay', 'connection', or 'content' in %s '%s' (got '%s')",
1418 args[0], args[1], proxy_type_str(curpx), curpx->id);
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001419 goto error;
Willy Tarreau1a687942010-05-23 22:40:30 +02001420 }
1421
Willy Tarreau1a687942010-05-23 22:40:30 +02001422 return warn;
Willy Tarreau6a984fa2010-06-14 16:44:27 +02001423 error:
1424 free(rule);
1425 return -1;
Willy Tarreaub6866442008-07-14 23:54:42 +02001426}
1427
Willy Tarreau645513a2010-05-24 20:55:15 +02001428
1429/************************************************************************/
Willy Tarreau32389b72012-04-23 23:13:20 +02001430/* All supported sample fetch functios must be declared here */
1431/************************************************************************/
1432
1433/* Fetch the request RDP cookie identified in the args, or any cookie if no arg
Willy Tarreau12785782012-04-27 21:37:17 +02001434 * is passed. It is usable both for ACL and for samples. Note: this decoder
Willy Tarreau32389b72012-04-23 23:13:20 +02001435 * only works with non-wrapping data. Accepts either 0 or 1 argument. Argument
1436 * is a string (cookie name), other types will lead to undefined behaviour.
1437 */
1438int
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001439smp_fetch_rdp_cookie(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreau32389b72012-04-23 23:13:20 +02001440 const struct arg *args, struct sample *smp)
1441{
1442 int bleft;
1443 const unsigned char *data;
1444
1445 if (!l4 || !l4->req)
1446 return 0;
1447
1448 smp->flags = 0;
1449 smp->type = SMP_T_CSTR;
1450
1451 bleft = l4->req->i;
1452 if (bleft <= 11)
1453 goto too_short;
1454
1455 data = (const unsigned char *)l4->req->p + 11;
1456 bleft -= 11;
1457
1458 if (bleft <= 7)
1459 goto too_short;
1460
1461 if (strncasecmp((const char *)data, "Cookie:", 7) != 0)
1462 goto not_cookie;
1463
1464 data += 7;
1465 bleft -= 7;
1466
1467 while (bleft > 0 && *data == ' ') {
1468 data++;
1469 bleft--;
1470 }
1471
1472 if (args) {
1473
1474 if (bleft <= args->data.str.len)
1475 goto too_short;
1476
1477 if ((data[args->data.str.len] != '=') ||
1478 strncasecmp(args->data.str.str, (const char *)data, args->data.str.len) != 0)
1479 goto not_cookie;
1480
1481 data += args->data.str.len + 1;
1482 bleft -= args->data.str.len + 1;
1483 } else {
1484 while (bleft > 0 && *data != '=') {
1485 if (*data == '\r' || *data == '\n')
1486 goto not_cookie;
1487 data++;
1488 bleft--;
1489 }
1490
1491 if (bleft < 1)
1492 goto too_short;
1493
1494 if (*data != '=')
1495 goto not_cookie;
1496
1497 data++;
1498 bleft--;
1499 }
1500
1501 /* data points to cookie value */
1502 smp->data.str.str = (char *)data;
1503 smp->data.str.len = 0;
1504
1505 while (bleft > 0 && *data != '\r') {
1506 data++;
1507 bleft--;
1508 }
1509
1510 if (bleft < 2)
1511 goto too_short;
1512
1513 if (data[0] != '\r' || data[1] != '\n')
1514 goto not_cookie;
1515
1516 smp->data.str.len = (char *)data - smp->data.str.str;
1517 smp->flags = SMP_F_VOLATILE;
1518 return 1;
1519
1520 too_short:
1521 smp->flags = SMP_F_MAY_CHANGE;
1522 not_cookie:
1523 return 0;
1524}
1525
Willy Tarreau32389b72012-04-23 23:13:20 +02001526/************************************************************************/
Willy Tarreau645513a2010-05-24 20:55:15 +02001527/* All supported ACL keywords must be declared here. */
1528/************************************************************************/
1529
Willy Tarreau32389b72012-04-23 23:13:20 +02001530/* returns either 1 or 0 depending on whether an RDP cookie is found or not */
1531static int
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001532acl_fetch_rdp_cookie_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreau24e32d82012-04-23 23:55:44 +02001533 const struct arg *args, struct sample *smp)
Willy Tarreau32389b72012-04-23 23:13:20 +02001534{
1535 int ret;
1536
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001537 ret = smp_fetch_rdp_cookie(px, l4, l7, opt, args, smp);
Willy Tarreau32389b72012-04-23 23:13:20 +02001538
1539 if (smp->flags & SMP_F_MAY_CHANGE)
1540 return 0;
1541
1542 smp->flags = SMP_F_VOLATILE;
1543 smp->type = SMP_T_UINT;
1544 smp->data.uint = ret;
1545 return 1;
1546}
1547
1548
Willy Tarreau4a129812012-04-25 17:31:42 +02001549/* fetch the connection's source IPv4/IPv6 address */
Willy Tarreau645513a2010-05-24 20:55:15 +02001550static int
Willy Tarreau4a129812012-04-25 17:31:42 +02001551smp_fetch_src(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreau24e32d82012-04-23 23:55:44 +02001552 const struct arg *args, struct sample *smp)
Willy Tarreau645513a2010-05-24 20:55:15 +02001553{
Willy Tarreauf4362b32011-12-16 17:49:52 +01001554 switch (l4->si[0].addr.from.ss_family) {
1555 case AF_INET:
Willy Tarreauf853c462012-04-23 18:53:56 +02001556 smp->data.ipv4 = ((struct sockaddr_in *)&l4->si[0].addr.from)->sin_addr;
1557 smp->type = SMP_T_IPV4;
Willy Tarreauf4362b32011-12-16 17:49:52 +01001558 break;
1559 case AF_INET6:
Willy Tarreauf853c462012-04-23 18:53:56 +02001560 smp->data.ipv6 = ((struct sockaddr_in6 *)(&l4->si[0].addr.from))->sin6_addr;
1561 smp->type = SMP_T_IPV6;
Willy Tarreauf4362b32011-12-16 17:49:52 +01001562 break;
1563 default:
Emeric Brunf769f512010-10-22 17:14:01 +02001564 return 0;
Willy Tarreauf4362b32011-12-16 17:49:52 +01001565 }
Emeric Brunf769f512010-10-22 17:14:01 +02001566
Willy Tarreau37406352012-04-23 16:16:37 +02001567 smp->flags = 0;
Willy Tarreau645513a2010-05-24 20:55:15 +02001568 return 1;
1569}
1570
Willy Tarreaua5e37562011-12-16 17:06:15 +01001571/* set temp integer to the connection's source port */
Willy Tarreau645513a2010-05-24 20:55:15 +02001572static int
Willy Tarreau25c1ebc2012-04-25 16:21:44 +02001573smp_fetch_sport(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreau24e32d82012-04-23 23:55:44 +02001574 const struct arg *args, struct sample *smp)
Willy Tarreau645513a2010-05-24 20:55:15 +02001575{
Willy Tarreauf853c462012-04-23 18:53:56 +02001576 smp->type = SMP_T_UINT;
1577 if (!(smp->data.uint = get_host_port(&l4->si[0].addr.from)))
Emeric Brunf769f512010-10-22 17:14:01 +02001578 return 0;
1579
Willy Tarreau37406352012-04-23 16:16:37 +02001580 smp->flags = 0;
Willy Tarreau645513a2010-05-24 20:55:15 +02001581 return 1;
1582}
1583
Willy Tarreau4a129812012-04-25 17:31:42 +02001584/* fetch the connection's destination IPv4/IPv6 address */
Willy Tarreau645513a2010-05-24 20:55:15 +02001585static int
Willy Tarreau4a129812012-04-25 17:31:42 +02001586smp_fetch_dst(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreau24e32d82012-04-23 23:55:44 +02001587 const struct arg *args, struct sample *smp)
Willy Tarreau645513a2010-05-24 20:55:15 +02001588{
Willy Tarreau59b94792012-05-11 16:16:40 +02001589 si_get_to_addr(&l4->si[0]);
Willy Tarreau645513a2010-05-24 20:55:15 +02001590
Willy Tarreauf4362b32011-12-16 17:49:52 +01001591 switch (l4->si[0].addr.to.ss_family) {
1592 case AF_INET:
Willy Tarreauf853c462012-04-23 18:53:56 +02001593 smp->data.ipv4 = ((struct sockaddr_in *)&l4->si[0].addr.to)->sin_addr;
1594 smp->type = SMP_T_IPV4;
Willy Tarreauf4362b32011-12-16 17:49:52 +01001595 break;
1596 case AF_INET6:
Willy Tarreauf853c462012-04-23 18:53:56 +02001597 smp->data.ipv6 = ((struct sockaddr_in6 *)(&l4->si[0].addr.to))->sin6_addr;
1598 smp->type = SMP_T_IPV6;
Willy Tarreauf4362b32011-12-16 17:49:52 +01001599 break;
1600 default:
Emeric Brunf769f512010-10-22 17:14:01 +02001601 return 0;
Willy Tarreauf4362b32011-12-16 17:49:52 +01001602 }
Emeric Brunf769f512010-10-22 17:14:01 +02001603
Willy Tarreau37406352012-04-23 16:16:37 +02001604 smp->flags = 0;
Willy Tarreau645513a2010-05-24 20:55:15 +02001605 return 1;
1606}
1607
Willy Tarreaua5e37562011-12-16 17:06:15 +01001608/* set temp integer to the frontend connexion's destination port */
Willy Tarreau645513a2010-05-24 20:55:15 +02001609static int
Willy Tarreau25c1ebc2012-04-25 16:21:44 +02001610smp_fetch_dport(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreau24e32d82012-04-23 23:55:44 +02001611 const struct arg *args, struct sample *smp)
Willy Tarreau645513a2010-05-24 20:55:15 +02001612{
Willy Tarreau59b94792012-05-11 16:16:40 +02001613 si_get_to_addr(&l4->si[0]);
Willy Tarreau645513a2010-05-24 20:55:15 +02001614
Willy Tarreauf853c462012-04-23 18:53:56 +02001615 smp->type = SMP_T_UINT;
1616 if (!(smp->data.uint = get_host_port(&l4->si[0].addr.to)))
Emeric Brunf769f512010-10-22 17:14:01 +02001617 return 0;
1618
Willy Tarreau37406352012-04-23 16:16:37 +02001619 smp->flags = 0;
Willy Tarreau645513a2010-05-24 20:55:15 +02001620 return 1;
1621}
1622
1623static int
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001624smp_fetch_payload_lv(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
1625 const struct arg *arg_p, struct sample *smp)
Emericf2d7cae2010-11-05 18:13:50 +01001626{
Willy Tarreau82ea8002012-04-25 19:19:43 +02001627 unsigned int len_offset = arg_p[0].data.uint;
1628 unsigned int len_size = arg_p[1].data.uint;
1629 unsigned int buf_offset;
1630 unsigned int buf_size = 0;
Emericf2d7cae2010-11-05 18:13:50 +01001631 struct buffer *b;
1632 int i;
1633
1634 /* Format is (len offset, len size, buf offset) or (len offset, len size) */
1635 /* by default buf offset == len offset + len size */
1636 /* buf offset could be absolute or relative to len offset + len size if prefixed by + or - */
1637
1638 if (!l4)
1639 return 0;
1640
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001641 b = ((opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? l4->rep : l4->req;
Emericf2d7cae2010-11-05 18:13:50 +01001642
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001643 if (!b)
Emericf2d7cae2010-11-05 18:13:50 +01001644 return 0;
1645
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001646 if (len_offset + len_size > b->i)
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001647 goto too_short;
Emericf2d7cae2010-11-05 18:13:50 +01001648
1649 for (i = 0; i < len_size; i++) {
Willy Tarreau89fa7062012-03-02 16:13:16 +01001650 buf_size = (buf_size << 8) + ((unsigned char *)b->p)[i + len_offset];
Emericf2d7cae2010-11-05 18:13:50 +01001651 }
1652
Willy Tarreau9fcb9842012-04-20 14:45:49 +02001653 /* buf offset may be implicit, absolute or relative */
1654 buf_offset = len_offset + len_size;
1655 if (arg_p[2].type == ARGT_UINT)
1656 buf_offset = arg_p[2].data.uint;
1657 else if (arg_p[2].type == ARGT_SINT)
1658 buf_offset += arg_p[2].data.sint;
1659
Willy Tarreau82ea8002012-04-25 19:19:43 +02001660 if (!buf_size || buf_size > b->size || buf_offset + buf_size > b->size) {
1661 /* will never match */
1662 smp->flags = 0;
1663 return 0;
1664 }
1665
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001666 if (buf_offset + buf_size > b->i)
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001667 goto too_short;
Emericf2d7cae2010-11-05 18:13:50 +01001668
1669 /* init chunk as read only */
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +02001670 smp->type = SMP_T_CBIN;
Willy Tarreau342acb42012-04-23 22:03:39 +02001671 chunk_initlen(&smp->data.str, b->p + buf_offset, 0, buf_size);
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001672 smp->flags = SMP_F_VOLATILE;
Emericf2d7cae2010-11-05 18:13:50 +01001673 return 1;
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001674
1675 too_short:
1676 smp->flags = SMP_F_MAY_CHANGE;
1677 return 0;
Emericf2d7cae2010-11-05 18:13:50 +01001678}
1679
1680static int
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001681smp_fetch_payload(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
1682 const struct arg *arg_p, struct sample *smp)
Emericf2d7cae2010-11-05 18:13:50 +01001683{
Willy Tarreau82ea8002012-04-25 19:19:43 +02001684 unsigned int buf_offset = arg_p[0].data.uint;
1685 unsigned int buf_size = arg_p[1].data.uint;
Emericf2d7cae2010-11-05 18:13:50 +01001686 struct buffer *b;
1687
1688 if (!l4)
1689 return 0;
1690
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001691 b = ((opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? l4->rep : l4->req;
Emericf2d7cae2010-11-05 18:13:50 +01001692
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001693 if (!b)
Emericf2d7cae2010-11-05 18:13:50 +01001694 return 0;
Willy Tarreau82ea8002012-04-25 19:19:43 +02001695
1696 if (!buf_size || buf_size > b->size || buf_offset + buf_size > b->size) {
1697 /* will never match */
1698 smp->flags = 0;
1699 return 0;
1700 }
Emericf2d7cae2010-11-05 18:13:50 +01001701
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001702 if (buf_offset + buf_size > b->i)
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001703 goto too_short;
Emericf2d7cae2010-11-05 18:13:50 +01001704
1705 /* init chunk as read only */
Willy Tarreaub8c8f1f2012-04-23 22:38:26 +02001706 smp->type = SMP_T_CBIN;
Willy Tarreau342acb42012-04-23 22:03:39 +02001707 chunk_initlen(&smp->data.str, b->p + buf_offset, 0, buf_size);
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001708 smp->flags = SMP_F_VOLATILE;
Simon Hormanab814e02011-06-24 14:50:20 +09001709 return 1;
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001710
1711 too_short:
1712 smp->flags = SMP_F_MAY_CHANGE;
1713 return 0;
Simon Hormanab814e02011-06-24 14:50:20 +09001714}
1715
Willy Tarreau21d68a62012-04-20 15:52:36 +02001716/* This function is used to validate the arguments passed to a "payload" fetch
1717 * keyword. This keyword expects two positive integers, with the second one
1718 * being strictly positive. It is assumed that the types are already the correct
1719 * ones. Returns 0 on error, non-zero if OK. If <err_msg> is not NULL, it will be
1720 * filled with a pointer to an error message in case of error, that the caller
1721 * is responsible for freeing. The initial location must either be freeable or
1722 * NULL.
1723 */
1724static int val_payload(struct arg *arg, char **err_msg)
1725{
1726 if (!arg[1].data.uint) {
1727 if (err_msg)
1728 memprintf(err_msg, "payload length must be > 0");
1729 return 0;
1730 }
1731 return 1;
1732}
1733
1734/* This function is used to validate the arguments passed to a "payload_lv" fetch
1735 * keyword. This keyword allows two positive integers and an optional signed one,
1736 * with the second one being strictly positive and the third one being greater than
1737 * the opposite of the two others if negative. It is assumed that the types are
1738 * already the correct ones. Returns 0 on error, non-zero if OK. If <err_msg> is
1739 * not NULL, it will be filled with a pointer to an error message in case of
1740 * error, that the caller is responsible for freeing. The initial location must
1741 * either be freeable or NULL.
1742 */
1743static int val_payload_lv(struct arg *arg, char **err_msg)
1744{
1745 if (!arg[1].data.uint) {
1746 if (err_msg)
1747 memprintf(err_msg, "payload length must be > 0");
1748 return 0;
1749 }
1750
1751 if (arg[2].type == ARGT_SINT &&
1752 (int)(arg[0].data.uint + arg[1].data.uint + arg[2].data.sint) < 0) {
1753 if (err_msg)
1754 memprintf(err_msg, "payload offset too negative");
1755 return 0;
1756 }
1757 return 1;
1758}
1759
Willy Tarreaub6866442008-07-14 23:54:42 +02001760static struct cfg_kw_list cfg_kws = {{ },{
1761 { CFG_LISTEN, "tcp-request", tcp_parse_tcp_req },
Emeric Brun97679e72010-09-23 17:56:44 +02001762 { CFG_LISTEN, "tcp-response", tcp_parse_tcp_rep },
Willy Tarreaub6866442008-07-14 23:54:42 +02001763 { 0, NULL, NULL },
1764}};
1765
Willy Tarreau61612d42012-04-19 18:42:05 +02001766/* Note: must not be declared <const> as its list will be overwritten.
1767 * Please take care of keeping this list alphabetically sorted.
1768 */
Willy Tarreaub6866442008-07-14 23:54:42 +02001769static struct acl_kw_list acl_kws = {{ },{
Willy Tarreau4a129812012-04-25 17:31:42 +02001770 { "dst", acl_parse_ip, smp_fetch_dst, acl_match_ip, ACL_USE_TCP4_PERMANENT|ACL_MAY_LOOKUP, 0 },
Willy Tarreau25c1ebc2012-04-25 16:21:44 +02001771 { "dst_port", acl_parse_int, smp_fetch_dport, acl_match_int, ACL_USE_TCP_PERMANENT, 0 },
Willy Tarreau0d5fe142012-04-26 12:24:45 +02001772 { "payload", acl_parse_str, smp_fetch_payload, acl_match_str, ACL_USE_L6REQ_VOLATILE|ACL_MAY_LOOKUP, ARG2(2,UINT,UINT), val_payload },
1773 { "payload_lv", acl_parse_str, smp_fetch_payload_lv, acl_match_str, ACL_USE_L6REQ_VOLATILE|ACL_MAY_LOOKUP, ARG3(2,UINT,UINT,SINT), val_payload_lv },
Willy Tarreau9fb4bc72012-04-24 00:09:26 +02001774 { "req_rdp_cookie", acl_parse_str, smp_fetch_rdp_cookie, acl_match_str, ACL_USE_L6REQ_VOLATILE|ACL_MAY_LOOKUP, ARG1(0,STR) },
Willy Tarreau32389b72012-04-23 23:13:20 +02001775 { "req_rdp_cookie_cnt", acl_parse_int, acl_fetch_rdp_cookie_cnt, acl_match_int, ACL_USE_L6REQ_VOLATILE, ARG1(0,STR) },
Willy Tarreau4a129812012-04-25 17:31:42 +02001776 { "src", acl_parse_ip, smp_fetch_src, acl_match_ip, ACL_USE_TCP4_PERMANENT|ACL_MAY_LOOKUP, 0 },
Willy Tarreau25c1ebc2012-04-25 16:21:44 +02001777 { "src_port", acl_parse_int, smp_fetch_sport, acl_match_int, ACL_USE_TCP_PERMANENT, 0 },
Willy Tarreaub6866442008-07-14 23:54:42 +02001778 { NULL, NULL, NULL, NULL },
1779}};
1780
Willy Tarreau4a129812012-04-25 17:31:42 +02001781/* Note: must not be declared <const> as its list will be overwritten.
1782 * Note: fetches that may return multiple types must be declared as the lowest
1783 * common denominator, the type that can be casted into all other ones. For
1784 * instance v4/v6 must be declared v4.
1785 */
Willy Tarreau12785782012-04-27 21:37:17 +02001786static struct sample_fetch_kw_list sample_fetch_keywords = {{ },{
Willy Tarreau4a129812012-04-25 17:31:42 +02001787 { "src", smp_fetch_src, 0, NULL, SMP_T_IPV4, SMP_CAP_REQ|SMP_CAP_RES },
Willy Tarreau4a129812012-04-25 17:31:42 +02001788 { "dst", smp_fetch_dst, 0, NULL, SMP_T_IPV4, SMP_CAP_REQ|SMP_CAP_RES },
Willy Tarreau25c1ebc2012-04-25 16:21:44 +02001789 { "dst_port", smp_fetch_dport, 0, NULL, SMP_T_UINT, SMP_CAP_REQ|SMP_CAP_RES },
Willy Tarreau0ce3aa02012-04-25 18:46:33 +02001790 { "payload", smp_fetch_payload, ARG2(2,UINT,UINT), val_payload, SMP_T_CBIN, SMP_CAP_REQ|SMP_CAP_RES },
1791 { "payload_lv", smp_fetch_payload_lv, ARG3(2,UINT,UINT,SINT), val_payload_lv, SMP_T_CBIN, SMP_CAP_REQ|SMP_CAP_RES },
Willy Tarreaud6281ae2012-04-26 11:23:39 +02001792 { "rdp_cookie", smp_fetch_rdp_cookie, ARG1(1,STR), NULL, SMP_T_CSTR, SMP_CAP_REQ|SMP_CAP_RES },
Willy Tarreau25c1ebc2012-04-25 16:21:44 +02001793 { "src_port", smp_fetch_sport, 0, NULL, SMP_T_UINT, SMP_CAP_REQ|SMP_CAP_RES },
Willy Tarreau9fcb9842012-04-20 14:45:49 +02001794 { NULL, NULL, 0, 0, 0 },
Willy Tarreau645513a2010-05-24 20:55:15 +02001795}};
1796
Willy Tarreaue6b98942007-10-29 01:09:36 +01001797__attribute__((constructor))
1798static void __tcp_protocol_init(void)
1799{
1800 protocol_register(&proto_tcpv4);
1801 protocol_register(&proto_tcpv6);
Willy Tarreau12785782012-04-27 21:37:17 +02001802 sample_register_fetches(&sample_fetch_keywords);
Willy Tarreaub6866442008-07-14 23:54:42 +02001803 cfg_register_keywords(&cfg_kws);
1804 acl_register_keywords(&acl_kws);
Willy Tarreaue6b98942007-10-29 01:09:36 +01001805}
1806
1807
1808/*
1809 * Local variables:
1810 * c-indent-level: 8
1811 * c-basic-offset: 8
1812 * End:
1813 */