blob: 9ed68b1c6080115abbe1dd706c4c079720342fe3 [file] [log] [blame]
Willy Tarreaubaaee002006-06-26 02:48:02 +02001/*
Willy Tarreau81f9aa32010-06-01 17:45:26 +02002 * Session management functions.
Willy Tarreaubaaee002006-06-26 02:48:02 +02003 *
Willy Tarreaud28c3532012-04-19 19:28:33 +02004 * Copyright 2000-2012 Willy Tarreau <w@1wt.eu>
Willy Tarreaubaaee002006-06-26 02:48:02 +02005 *
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 <stdlib.h>
Willy Tarreau81f9aa32010-06-01 17:45:26 +020014#include <unistd.h>
15#include <fcntl.h>
Willy Tarreaue3ba5f02006-06-29 18:54:54 +020016
17#include <common/config.h>
Willy Tarreau9b28e032012-10-12 23:49:43 +020018#include <common/buffer.h>
Willy Tarreau7c669d72008-06-20 15:04:11 +020019#include <common/debug.h>
Willy Tarreau2dd0d472006-06-29 17:53:05 +020020#include <common/memory.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020021
Willy Tarreaubaaee002006-06-26 02:48:02 +020022#include <types/capture.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010023#include <types/global.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020024
Willy Tarreau1d0dfb12009-07-07 15:10:31 +020025#include <proto/acl.h>
Willy Tarreau61612d42012-04-19 18:42:05 +020026#include <proto/arg.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010027#include <proto/backend.h>
Willy Tarreauc7e42382012-08-24 19:22:53 +020028#include <proto/channel.h>
Krzysztof Piotr Oledzki97f07b82009-12-15 22:31:24 +010029#include <proto/checks.h>
Willy Tarreaud2274c62012-07-06 14:29:45 +020030#include <proto/connection.h>
Willy Tarreau5ca791d2009-08-16 19:06:42 +020031#include <proto/dumpstats.h>
Willy Tarreaudd2f85e2012-09-02 22:34:23 +020032#include <proto/fd.h>
Willy Tarreau91c43d72010-06-20 11:19:22 +020033#include <proto/freq_ctr.h>
Willy Tarreau3041b9f2010-10-15 23:25:20 +020034#include <proto/frontend.h>
Willy Tarreau8d5d7f22007-01-21 19:16:41 +010035#include <proto/hdr_idx.h>
Willy Tarreaud1d54542012-09-12 22:58:11 +020036#include <proto/listener.h>
Willy Tarreau332f8bf2007-05-13 21:36:56 +020037#include <proto/log.h>
Willy Tarreaucbaaec42012-09-06 11:32:07 +020038#include <proto/raw_sock.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020039#include <proto/session.h>
Willy Tarreau3eba98a2009-01-25 13:56:13 +010040#include <proto/pipe.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010041#include <proto/proto_http.h>
42#include <proto/proto_tcp.h>
Willy Tarreau1d0dfb12009-07-07 15:10:31 +020043#include <proto/proxy.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020044#include <proto/queue.h>
Willy Tarreau7f062c42009-03-05 18:43:00 +010045#include <proto/server.h>
Willy Tarreaucd3b0942012-04-27 21:52:18 +020046#include <proto/sample.h>
Emeric Brun1d33b292010-01-04 15:47:17 +010047#include <proto/stick_table.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010048#include <proto/stream_interface.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010049#include <proto/task.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020050
Willy Tarreauc6ca1a02007-05-13 19:43:47 +020051struct pool_head *pool2_session;
Willy Tarreauf54f8bd2008-11-23 19:53:55 +010052struct list sessions;
Willy Tarreaubaaee002006-06-26 02:48:02 +020053
Willy Tarreau071e1372012-10-03 01:39:48 +020054static int conn_session_complete(struct connection *conn);
Willy Tarreau9683e9a2012-10-03 21:17:23 +020055static int conn_session_update(struct connection *conn);
Willy Tarreau2542b532012-08-31 16:01:23 +020056static struct task *expire_mini_session(struct task *t);
57int session_complete(struct session *s);
58
Willy Tarreau5e75e272012-10-02 21:21:20 +020059/* data layer callbacks for an embryonic session */
60struct data_cb sess_conn_cb = {
61 .recv = NULL,
62 .send = NULL,
Willy Tarreau9683e9a2012-10-03 21:17:23 +020063 .wake = conn_session_update,
Willy Tarreau071e1372012-10-03 01:39:48 +020064 .init = conn_session_complete,
Willy Tarreau5e75e272012-10-02 21:21:20 +020065};
66
Willy Tarreau2542b532012-08-31 16:01:23 +020067/* This function is called from the protocol layer accept() in order to
68 * instanciate a new embryonic session on behalf of a given listener and
69 * frontend. It returns a positive value upon success, 0 if the connection
70 * can be ignored, or a negative value upon critical failure. The accepted
71 * file descriptor is closed if we return <= 0.
Willy Tarreau81f9aa32010-06-01 17:45:26 +020072 */
73int session_accept(struct listener *l, int cfd, struct sockaddr_storage *addr)
74{
75 struct proxy *p = l->frontend;
76 struct session *s;
Willy Tarreau81f9aa32010-06-01 17:45:26 +020077 struct task *t;
Willy Tarreauabe8ea52010-11-11 10:56:04 +010078 int ret;
79
80
81 ret = -1; /* assume unrecoverable error by default */
Willy Tarreau81f9aa32010-06-01 17:45:26 +020082
Willy Tarreaufffe1322010-11-11 09:48:16 +010083 if (unlikely((s = pool_alloc2(pool2_session)) == NULL))
Willy Tarreau81f9aa32010-06-01 17:45:26 +020084 goto out_close;
Willy Tarreau81f9aa32010-06-01 17:45:26 +020085
Willy Tarreauf2943dc2012-10-26 20:10:28 +020086 if (unlikely((s->si[0].conn = pool_alloc2(pool2_connection)) == NULL))
87 goto out_fail_conn0;
88
89 if (unlikely((s->si[1].conn = pool_alloc2(pool2_connection)) == NULL))
90 goto out_fail_conn1;
91
Willy Tarreau2542b532012-08-31 16:01:23 +020092 /* minimum session initialization required for an embryonic session is
93 * fairly low. We need very little to execute L4 ACLs, then we need a
94 * task to make the client-side connection live on its own.
95 * - flags
96 * - stick-entry tracking
97 */
Willy Tarreau81f9aa32010-06-01 17:45:26 +020098 s->flags = 0;
99 s->logs.logwait = p->to_log;
Willy Tarreau9a355ec2013-06-11 17:45:46 +0200100 s->logs.level = 0;
Willy Tarreaud5ca9ab2013-05-28 17:40:25 +0200101
102 memset(s->stkctr, 0, sizeof(s->stkctr));
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200103
Willy Tarreau2542b532012-08-31 16:01:23 +0200104 s->listener = l;
105 s->fe = p;
Willy Tarreauabe8ea52010-11-11 10:56:04 +0100106
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200107 /* OK, we're keeping the session, so let's properly initialize the session */
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200108 s->si[0].conn->t.sock.fd = cfd;
109 s->si[0].conn->ctrl = l->proto;
Willy Tarreau38d58922013-11-16 00:15:40 +0100110 s->si[0].conn->flags = CO_FL_NONE | CO_FL_ADDR_FROM_SET;
Willy Tarreau14cba4b2012-11-30 17:33:05 +0100111 s->si[0].conn->err_code = CO_ER_NONE;
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200112 s->si[0].conn->addr.from = *addr;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100113 s->si[0].conn->target = &l->obj_type;
Willy Tarreau2542b532012-08-31 16:01:23 +0200114
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200115 s->logs.accept_date = date; /* user-visible date for logging */
116 s->logs.tv_accept = now; /* corrected date for internal use */
117 s->uniq_id = totalconn;
Willy Tarreau2542b532012-08-31 16:01:23 +0200118 p->feconn++;
119 /* This session was accepted, count it now */
120 if (p->feconn > p->fe_counters.conn_max)
121 p->fe_counters.conn_max = p->feconn;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200122
Willy Tarreau2542b532012-08-31 16:01:23 +0200123 proxy_inc_fe_conn_ctr(l, p);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200124
125 /* now evaluate the tcp-request layer4 rules. Since we expect to be able
126 * to abort right here as soon as possible, we check the rules before
127 * even initializing the stream interfaces.
128 */
129 if ((l->options & LI_O_TCP_RULES) && !tcp_exec_req_rules(s)) {
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200130 /* let's do a no-linger now to close with a single RST. */
131 setsockopt(cfd, SOL_SOCKET, SO_LINGER, (struct linger *) &nolinger, sizeof(struct linger));
Willy Tarreauabe8ea52010-11-11 10:56:04 +0100132 ret = 0; /* successful termination */
Willy Tarreau2542b532012-08-31 16:01:23 +0200133 goto out_free_session;
134 }
135
Willy Tarreau1bc4aab2012-10-08 20:11:03 +0200136#ifndef USE_ACCEPT4
137 /* Adjust some socket options if the connection was accepted by a plain
138 * accept() syscall.
139 */
Willy Tarreau82569f92012-09-27 23:48:56 +0200140 if (unlikely(fcntl(cfd, F_SETFL, O_NONBLOCK) == -1))
141 goto out_free_session;
Willy Tarreau1bc4aab2012-10-08 20:11:03 +0200142#endif
Willy Tarreau82569f92012-09-27 23:48:56 +0200143
144 /* monitor-net and health mode are processed immediately after TCP
145 * connection rules. This way it's possible to block them, but they
146 * never use the lower data layers, they send directly over the socket,
147 * as they were designed for. We first flush the socket receive buffer
148 * in order to avoid emission of an RST by the system. We ignore any
149 * error.
150 */
151 if (unlikely((p->mode == PR_MODE_HEALTH) ||
152 ((l->options & LI_O_CHK_MONNET) &&
153 addr->ss_family == AF_INET &&
154 (((struct sockaddr_in *)addr)->sin_addr.s_addr & p->mon_mask.s_addr) == p->mon_net.s_addr))) {
155 /* we have 4 possibilities here :
156 * - HTTP mode, from monitoring address => send "HTTP/1.0 200 OK"
157 * - HEALTH mode with HTTP check => send "HTTP/1.0 200 OK"
158 * - HEALTH mode without HTTP check => just send "OK"
159 * - TCP mode from monitoring address => just close
160 */
Willy Tarreau2b57cb82013-06-10 19:56:38 +0200161 if (l->proto->drain)
162 l->proto->drain(cfd);
Willy Tarreau82569f92012-09-27 23:48:56 +0200163 if (p->mode == PR_MODE_HTTP ||
164 (p->mode == PR_MODE_HEALTH && (p->options2 & PR_O2_CHK_ANY) == PR_O2_HTTP_CHK))
165 send(cfd, "HTTP/1.0 200 OK\r\n\r\n", 19, MSG_DONTWAIT|MSG_NOSIGNAL|MSG_MORE);
166 else if (p->mode == PR_MODE_HEALTH)
167 send(cfd, "OK\n", 3, MSG_DONTWAIT|MSG_NOSIGNAL|MSG_MORE);
168 ret = 0;
169 goto out_free_session;
170 }
171
172
Willy Tarreau22cda212012-08-31 17:43:29 +0200173 /* wait for a PROXY protocol header */
174 if (l->options & LI_O_ACC_PROXY) {
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200175 s->si[0].conn->flags |= CO_FL_ACCEPT_PROXY;
176 conn_sock_want_recv(s->si[0].conn);
Willy Tarreau22cda212012-08-31 17:43:29 +0200177 }
178
Willy Tarreau2542b532012-08-31 16:01:23 +0200179 if (unlikely((t = task_new()) == NULL))
180 goto out_free_session;
181
182 t->context = s;
183 t->nice = l->nice;
184 s->task = t;
185
Willy Tarreauf7bc57c2012-10-03 00:19:48 +0200186 /* Add the various callbacks. Right now the transport layer is present
187 * but not initialized. Also note we need to be careful as the stream
188 * int is not initialized yet.
Willy Tarreau2542b532012-08-31 16:01:23 +0200189 */
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200190 conn_prepare(s->si[0].conn, &sess_conn_cb, l->proto, l->xprt, s);
Willy Tarreau2542b532012-08-31 16:01:23 +0200191
192 /* finish initialization of the accepted file descriptor */
193 fd_insert(cfd);
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200194 fdtab[cfd].owner = s->si[0].conn;
Willy Tarreau2542b532012-08-31 16:01:23 +0200195 fdtab[cfd].iocb = conn_fd_handler;
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200196 conn_data_want_recv(s->si[0].conn);
197 if (conn_xprt_init(s->si[0].conn) < 0)
Willy Tarreauabe8ea52010-11-11 10:56:04 +0100198 goto out_free_task;
Willy Tarreau2542b532012-08-31 16:01:23 +0200199
200 /* OK, now either we have a pending handshake to execute with and
201 * then we must return to the I/O layer, or we can proceed with the
202 * end of the session initialization. In case of handshake, we also
203 * set the I/O timeout to the frontend's client timeout.
204 */
205
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200206 if (s->si[0].conn->flags & CO_FL_HANDSHAKE) {
Willy Tarreau2542b532012-08-31 16:01:23 +0200207 t->process = expire_mini_session;
208 t->expire = tick_add_ifset(now_ms, p->timeout.client);
209 task_queue(t);
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200210 s->si[0].conn->flags |= CO_FL_INIT_DATA | CO_FL_WAKE_DATA;
Willy Tarreau2542b532012-08-31 16:01:23 +0200211 return 1;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200212 }
213
Willy Tarreau815f5ec2012-11-06 00:14:25 +0100214 /* OK let's complete session initialization since there is no handshake */
215 s->si[0].conn->flags |= CO_FL_CONNECTED;
Willy Tarreau2542b532012-08-31 16:01:23 +0200216 ret = session_complete(s);
217 if (ret > 0)
218 return ret;
219
220 /* Error unrolling */
221 out_free_task:
222 task_free(t);
223 out_free_session:
224 p->feconn--;
Willy Tarreau20d46a52012-12-09 15:55:40 +0100225 if (s->stkctr[0].entry || s->stkctr[1].entry)
Willy Tarreau2542b532012-08-31 16:01:23 +0200226 session_store_counters(s);
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200227 pool_free2(pool2_connection, s->si[1].conn);
228 out_fail_conn1:
229 pool_free2(pool2_connection, s->si[0].conn);
230 out_fail_conn0:
Willy Tarreau2542b532012-08-31 16:01:23 +0200231 pool_free2(pool2_session, s);
232 out_close:
Willy Tarreauf7bc57c2012-10-03 00:19:48 +0200233 if (ret < 0 && l->xprt == &raw_sock && p->mode == PR_MODE_HTTP) {
Willy Tarreau2542b532012-08-31 16:01:23 +0200234 /* critical error, no more memory, try to emit a 500 response */
Willy Tarreau05bf5e12013-10-20 23:10:28 +0200235 struct chunk *err_msg = &p->errmsg[HTTP_ERR_500];
236 if (!err_msg->str)
237 err_msg = &http_err_chunks[HTTP_ERR_500];
Willy Tarreau2542b532012-08-31 16:01:23 +0200238 send(cfd, err_msg->str, err_msg->len, MSG_DONTWAIT|MSG_NOSIGNAL);
239 }
240
241 if (fdtab[cfd].owner)
242 fd_delete(cfd);
243 else
244 close(cfd);
245 return ret;
246}
Willy Tarreauabe8ea52010-11-11 10:56:04 +0100247
Willy Tarreau0af29122012-12-03 15:35:00 +0100248
249/* prepare the trash with a log prefix for session <s> */
250static void prepare_mini_sess_log_prefix(struct session *s)
251{
252 struct tm tm;
253 char pn[INET6_ADDRSTRLEN];
254 int ret;
255 char *end;
256
257 ret = addr_to_str(&s->si[0].conn->addr.from, pn, sizeof(pn));
258 if (ret <= 0)
259 chunk_printf(&trash, "unknown [");
260 else if (ret == AF_UNIX)
261 chunk_printf(&trash, "%s:%d [", pn, s->listener->luid);
262 else
263 chunk_printf(&trash, "%s:%d [", pn, get_host_port(&s->si[0].conn->addr.from));
264
265 get_localtime(s->logs.accept_date.tv_sec, &tm);
266 end = date2str_log(trash.str + trash.len, &tm, &(s->logs.accept_date), trash.size - trash.len);
267 trash.len = end - trash.str;
268 if (s->listener->name)
269 chunk_appendf(&trash, "] %s/%s", s->fe->id, s->listener->name);
270 else
271 chunk_appendf(&trash, "] %s/%d", s->fe->id, s->listener->luid);
272}
273
Willy Tarreau2542b532012-08-31 16:01:23 +0200274/* This function kills an existing embryonic session. It stops the connection's
Willy Tarreauf7bc57c2012-10-03 00:19:48 +0200275 * transport layer, releases assigned resources, resumes the listener if it was
Willy Tarreau2542b532012-08-31 16:01:23 +0200276 * disabled and finally kills the file descriptor.
277 */
278static void kill_mini_session(struct session *s)
279{
Willy Tarreau0af29122012-12-03 15:35:00 +0100280 int level = LOG_INFO;
281 struct connection *conn = s->si[0].conn;
282 unsigned int log = s->logs.logwait;
283 const char *err_msg;
284
285 if (s->fe->options2 & PR_O2_LOGERRORS)
286 level = LOG_ERR;
287
288 if (log && (s->fe->options & PR_O_NULLNOLOG)) {
289 /* with "option dontlognull", we don't log connections with no transfer */
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100290 if (!conn->err_code ||
Willy Tarreau20879a02012-12-03 16:32:10 +0100291 conn->err_code == CO_ER_PRX_EMPTY || conn->err_code == CO_ER_PRX_ABORT ||
292 conn->err_code == CO_ER_SSL_EMPTY || conn->err_code == CO_ER_SSL_ABORT)
Willy Tarreau0af29122012-12-03 15:35:00 +0100293 log = 0;
294 }
295
296 if (log) {
297 if (!conn->err_code && (s->task->state & TASK_WOKEN_TIMER)) {
298 if (conn->flags & CO_FL_ACCEPT_PROXY)
299 conn->err_code = CO_ER_PRX_TIMEOUT;
300 else if (conn->flags & CO_FL_SSL_WAIT_HS)
301 conn->err_code = CO_ER_SSL_TIMEOUT;
302 }
303
304 prepare_mini_sess_log_prefix(s);
305 err_msg = conn_err_code_str(conn);
306 if (err_msg)
307 send_log(s->fe, level, "%s: %s\n", trash.str, err_msg);
308 else
309 send_log(s->fe, level, "%s: unknown connection error (code=%d flags=%08x)\n",
310 trash.str, conn->err_code, conn->flags);
311 }
312
Willy Tarreau2542b532012-08-31 16:01:23 +0200313 /* kill the connection now */
Willy Tarreau2b199c92012-11-23 17:32:21 +0100314 conn_full_close(s->si[0].conn);
Willy Tarreau2542b532012-08-31 16:01:23 +0200315
316 s->fe->feconn--;
Willy Tarreau7af7d592013-07-01 18:07:03 +0200317 session_store_counters(s);
Willy Tarreau2542b532012-08-31 16:01:23 +0200318
319 if (!(s->listener->options & LI_O_UNLIMITED))
320 actconn--;
321 jobs--;
322 s->listener->nbconn--;
323 if (s->listener->state == LI_FULL)
324 resume_listener(s->listener);
325
326 /* Dequeues all of the listeners waiting for a resource */
327 if (!LIST_ISEMPTY(&global_listener_queue))
328 dequeue_all_listeners(&global_listener_queue);
329
330 if (!LIST_ISEMPTY(&s->fe->listener_queue) &&
331 (!s->fe->fe_sps_lim || freq_ctr_remain(&s->fe->fe_sess_per_sec, s->fe->fe_sps_lim, 0) > 0))
332 dequeue_all_listeners(&s->fe->listener_queue);
333
334 task_delete(s->task);
335 task_free(s->task);
336
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200337 pool_free2(pool2_connection, s->si[1].conn);
338 pool_free2(pool2_connection, s->si[0].conn);
Willy Tarreau2542b532012-08-31 16:01:23 +0200339 pool_free2(pool2_session, s);
340}
341
Willy Tarreau22cda212012-08-31 17:43:29 +0200342/* Finish initializing a session from a connection, or kills it if the
343 * connection shows and error. Returns <0 if the connection was killed.
Willy Tarreau2542b532012-08-31 16:01:23 +0200344 */
Willy Tarreau071e1372012-10-03 01:39:48 +0200345static int conn_session_complete(struct connection *conn)
Willy Tarreau2542b532012-08-31 16:01:23 +0200346{
Willy Tarreau5e75e272012-10-02 21:21:20 +0200347 struct session *s = conn->owner;
Willy Tarreau2542b532012-08-31 16:01:23 +0200348
Willy Tarreau22cda212012-08-31 17:43:29 +0200349 if (!(conn->flags & CO_FL_ERROR) && (session_complete(s) > 0)) {
Willy Tarreau071e1372012-10-03 01:39:48 +0200350 conn->flags &= ~CO_FL_INIT_DATA;
Willy Tarreau2542b532012-08-31 16:01:23 +0200351 return 0;
352 }
353
354 /* kill the connection now */
355 kill_mini_session(s);
356 return -1;
357}
358
Willy Tarreau9683e9a2012-10-03 21:17:23 +0200359/* Update an embryonic session status. The connection is killed in case of
360 * error, and <0 will be returned. Otherwise it does nothing.
361 */
362static int conn_session_update(struct connection *conn)
363{
364 if (conn->flags & CO_FL_ERROR) {
365 kill_mini_session(conn->owner);
366 return -1;
367 }
368 return 0;
369}
370
Willy Tarreau2542b532012-08-31 16:01:23 +0200371/* Manages embryonic sessions timeout. It is only called when the timeout
372 * strikes and performs the required cleanup.
373 */
374static struct task *expire_mini_session(struct task *t)
375{
376 struct session *s = t->context;
377
378 if (!(t->state & TASK_WOKEN_TIMER))
379 return t;
380
381 kill_mini_session(s);
382 return NULL;
383}
384
385/* This function is called from the I/O handler which detects the end of
386 * handshake, in order to complete initialization of a valid session. It must
387 * be called with an embryonic session. It returns a positive value upon
388 * success, 0 if the connection can be ignored, or a negative value upon
389 * critical failure. The accepted file descriptor is closed if we return <= 0.
390 */
391int session_complete(struct session *s)
392{
393 struct listener *l = s->listener;
394 struct proxy *p = s->fe;
395 struct http_txn *txn;
396 struct task *t = s->task;
397 int ret;
Willy Tarreau20d46a52012-12-09 15:55:40 +0100398 int i;
Willy Tarreau2542b532012-08-31 16:01:23 +0200399
400 ret = -1; /* assume unrecoverable error by default */
401
402 /* OK, we're keeping the session, so let's properly initialize the session */
403 LIST_ADDQ(&sessions, &s->list);
404 LIST_INIT(&s->back_refs);
Willy Tarreau5e75e272012-10-02 21:21:20 +0200405 si_takeover_conn(&s->si[0], l->proto, l->xprt);
Willy Tarreau2542b532012-08-31 16:01:23 +0200406 s->flags |= SN_INITIALIZED;
407
408 s->unique_id = NULL;
Willy Tarreau2542b532012-08-31 16:01:23 +0200409
410 t->process = l->handler;
411 t->context = s;
412 t->expire = TICK_ETERNITY;
413
414 /* Note: initially, the session's backend points to the frontend.
415 * This changes later when switching rules are executed or
416 * when the default backend is assigned.
417 */
418 s->be = s->fe;
419 s->req = s->rep = NULL; /* will be allocated later */
William Lallemand82fe75c2012-10-23 10:25:10 +0200420 s->comp_algo = NULL;
Willy Tarreau2542b532012-08-31 16:01:23 +0200421
422 /* Let's count a session now */
Willy Tarreaub36b4242010-06-04 20:59:39 +0200423 proxy_inc_fe_sess_ctr(l, p);
Willy Tarreau91c43d72010-06-20 11:19:22 +0200424
Willy Tarreaub4c84932013-07-23 19:15:30 +0200425 for (i = 0; i < MAX_SESS_STKCTR; i++) {
Willy Tarreau9e9879a2010-08-06 15:25:22 +0200426 void *ptr;
427
Willy Tarreau20d46a52012-12-09 15:55:40 +0100428 if (!s->stkctr[i].entry)
429 continue;
430
431 ptr = stktable_data_ptr(s->stkctr[i].table, s->stkctr[i].entry, STKTABLE_DT_SESS_CNT);
Willy Tarreau9e9879a2010-08-06 15:25:22 +0200432 if (ptr)
433 stktable_data_cast(ptr, sess_cnt)++;
434
Willy Tarreau20d46a52012-12-09 15:55:40 +0100435 ptr = stktable_data_ptr(s->stkctr[i].table, s->stkctr[i].entry, STKTABLE_DT_SESS_RATE);
Willy Tarreau9e9879a2010-08-06 15:25:22 +0200436 if (ptr)
437 update_freq_ctr_period(&stktable_data_cast(ptr, sess_rate),
Willy Tarreau20d46a52012-12-09 15:55:40 +0100438 s->stkctr[i].table->data_arg[STKTABLE_DT_SESS_RATE].u, 1);
Willy Tarreau9e9879a2010-08-06 15:25:22 +0200439 }
440
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200441 /* this part should be common with other protocols */
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200442 s->si[0].owner = t;
443 s->si[0].state = s->si[0].prev_state = SI_ST_EST;
444 s->si[0].err_type = SI_ET_NONE;
445 s->si[0].err_loc = NULL;
Willy Tarreau0bd05ea2010-07-02 11:18:03 +0200446 s->si[0].release = NULL;
Willy Tarreau63e7fe32012-05-08 15:20:43 +0200447 s->si[0].send_proxy_ofs = 0;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200448 s->si[0].exp = TICK_ETERNITY;
449 s->si[0].flags = SI_FL_NONE;
450
451 if (likely(s->fe->options2 & PR_O2_INDEPSTR))
452 s->si[0].flags |= SI_FL_INDEP_STR;
453
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200454 /* pre-initialize the other side's stream interface to an INIT state. The
455 * callbacks will be initialized before attempting to connect.
456 */
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200457 s->si[1].conn->t.sock.fd = -1; /* just to help with debugging */
458 s->si[1].conn->flags = CO_FL_NONE;
Willy Tarreau14cba4b2012-11-30 17:33:05 +0100459 s->si[1].conn->err_code = CO_ER_NONE;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200460 s->si[1].owner = t;
461 s->si[1].state = s->si[1].prev_state = SI_ST_INI;
462 s->si[1].err_type = SI_ET_NONE;
Willy Tarreau0b3a4112011-03-27 19:16:56 +0200463 s->si[1].conn_retries = 0; /* used for logging too */
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200464 s->si[1].err_loc = NULL;
Willy Tarreau0bd05ea2010-07-02 11:18:03 +0200465 s->si[1].release = NULL;
Willy Tarreau63e7fe32012-05-08 15:20:43 +0200466 s->si[1].send_proxy_ofs = 0;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100467 s->si[1].conn->target = NULL;
Willy Tarreauc5788912012-08-24 18:12:41 +0200468 si_prepare_embedded(&s->si[1]);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200469 s->si[1].exp = TICK_ETERNITY;
470 s->si[1].flags = SI_FL_NONE;
471
472 if (likely(s->fe->options2 & PR_O2_INDEPSTR))
473 s->si[1].flags |= SI_FL_INDEP_STR;
474
Willy Tarreau9bd0d742011-07-20 00:17:39 +0200475 session_init_srv_conn(s);
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100476 s->target = NULL;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200477 s->pend_pos = NULL;
478
479 /* init store persistence */
480 s->store_count = 0;
481
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200482 if (unlikely((s->req = pool_alloc2(pool2_channel)) == NULL))
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200483 goto out_free_task; /* no memory */
484
Willy Tarreau9b28e032012-10-12 23:49:43 +0200485 if (unlikely((s->req->buf = pool_alloc2(pool2_buffer)) == NULL))
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200486 goto out_free_req; /* no memory */
487
Willy Tarreau9b28e032012-10-12 23:49:43 +0200488 if (unlikely((s->rep = pool_alloc2(pool2_channel)) == NULL))
489 goto out_free_req_buf; /* no memory */
490
491 if (unlikely((s->rep->buf = pool_alloc2(pool2_buffer)) == NULL))
492 goto out_free_rep; /* no memory */
493
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200494 /* initialize the request buffer */
Willy Tarreau9b28e032012-10-12 23:49:43 +0200495 s->req->buf->size = global.tune.bufsize;
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200496 channel_init(s->req);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200497 s->req->prod = &s->si[0];
498 s->req->cons = &s->si[1];
499 s->si[0].ib = s->si[1].ob = s->req;
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200500 s->req->flags |= CF_READ_ATTACHED; /* the producer is already connected */
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200501
502 /* activate default analysers enabled for this listener */
503 s->req->analysers = l->analysers;
504
505 s->req->wto = TICK_ETERNITY;
506 s->req->rto = TICK_ETERNITY;
507 s->req->rex = TICK_ETERNITY;
508 s->req->wex = TICK_ETERNITY;
509 s->req->analyse_exp = TICK_ETERNITY;
510
511 /* initialize response buffer */
Willy Tarreau9b28e032012-10-12 23:49:43 +0200512 s->rep->buf->size = global.tune.bufsize;
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200513 channel_init(s->rep);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200514 s->rep->prod = &s->si[1];
515 s->rep->cons = &s->si[0];
516 s->si[0].ob = s->si[1].ib = s->rep;
517 s->rep->analysers = 0;
518
Willy Tarreau96e31212011-05-30 18:10:30 +0200519 if (s->fe->options2 & PR_O2_NODELAY) {
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200520 s->req->flags |= CF_NEVER_WAIT;
521 s->rep->flags |= CF_NEVER_WAIT;
Willy Tarreau96e31212011-05-30 18:10:30 +0200522 }
523
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200524 s->rep->rto = TICK_ETERNITY;
525 s->rep->wto = TICK_ETERNITY;
526 s->rep->rex = TICK_ETERNITY;
527 s->rep->wex = TICK_ETERNITY;
528 s->rep->analyse_exp = TICK_ETERNITY;
529
Willy Tarreau62f791e2012-03-09 11:32:30 +0100530 txn = &s->txn;
531 /* Those variables will be checked and freed if non-NULL in
532 * session.c:session_free(). It is important that they are
533 * properly initialized.
534 */
535 txn->sessid = NULL;
536 txn->srv_cookie = NULL;
537 txn->cli_cookie = NULL;
538 txn->uri = NULL;
539 txn->req.cap = NULL;
540 txn->rsp.cap = NULL;
541 txn->hdr_idx.v = NULL;
542 txn->hdr_idx.size = txn->hdr_idx.used = 0;
543 txn->req.flags = 0;
544 txn->rsp.flags = 0;
545 /* the HTTP messages need to know what buffer they're associated with */
Willy Tarreau394db372012-10-12 22:40:39 +0200546 txn->req.chn = s->req;
547 txn->rsp.chn = s->rep;
Willy Tarreau62f791e2012-03-09 11:32:30 +0100548
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200549 /* finish initialization of the accepted file descriptor */
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200550 conn_data_want_recv(s->si[0].conn);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200551
Willy Tarreauabe8ea52010-11-11 10:56:04 +0100552 if (p->accept && (ret = p->accept(s)) <= 0) {
553 /* Either we had an unrecoverable error (<0) or work is
554 * finished (=0, eg: monitoring), in both situations,
555 * we can release everything and close.
556 */
Willy Tarreau9b28e032012-10-12 23:49:43 +0200557 goto out_free_rep_buf;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200558 }
559
Willy Tarreau93dbc2b2012-10-12 18:01:49 +0200560 /* if logs require transport layer information, note it on the connection */
561 if (s->logs.logwait & LW_XPRT)
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200562 s->si[0].conn->flags |= CO_FL_XPRT_TRACKED;
Willy Tarreau93dbc2b2012-10-12 18:01:49 +0200563
Willy Tarreau2542b532012-08-31 16:01:23 +0200564 /* we want the connection handler to notify the stream interface about updates. */
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200565 s->si[0].conn->flags |= CO_FL_WAKE_DATA;
Willy Tarreau2542b532012-08-31 16:01:23 +0200566
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200567 /* it is important not to call the wakeup function directly but to
568 * pass through task_wakeup(), because this one knows how to apply
569 * priorities to tasks.
570 */
571 task_wakeup(t, TASK_WOKEN_INIT);
572 return 1;
573
574 /* Error unrolling */
Willy Tarreau9b28e032012-10-12 23:49:43 +0200575 out_free_rep_buf:
576 pool_free2(pool2_buffer, s->rep->buf);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200577 out_free_rep:
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200578 pool_free2(pool2_channel, s->rep);
Willy Tarreau9b28e032012-10-12 23:49:43 +0200579 out_free_req_buf:
580 pool_free2(pool2_buffer, s->req->buf);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200581 out_free_req:
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200582 pool_free2(pool2_channel, s->req);
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200583 out_free_task:
Willy Tarreauabe8ea52010-11-11 10:56:04 +0100584 return ret;
Willy Tarreau81f9aa32010-06-01 17:45:26 +0200585}
586
Willy Tarreaubaaee002006-06-26 02:48:02 +0200587/*
588 * frees the context associated to a session. It must have been removed first.
589 */
Simon Hormandec5be42011-06-08 09:19:07 +0900590static void session_free(struct session *s)
Willy Tarreaubaaee002006-06-26 02:48:02 +0200591{
Willy Tarreau4dbc4a22007-03-03 16:23:22 +0100592 struct http_txn *txn = &s->txn;
Willy Tarreau632f5a72007-07-11 10:42:35 +0200593 struct proxy *fe = s->fe;
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100594 struct bref *bref, *back;
Willy Tarreaua4cda672010-06-06 18:28:49 +0200595 int i;
Willy Tarreau0f7562b2007-01-07 15:46:13 +0100596
Willy Tarreaubaaee002006-06-26 02:48:02 +0200597 if (s->pend_pos)
598 pendconn_free(s->pend_pos);
Willy Tarreau922a8062008-12-04 09:33:58 +0100599
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100600 if (objt_server(s->target)) { /* there may be requests left pending in queue */
Willy Tarreau1e62de62008-11-11 20:20:02 +0100601 if (s->flags & SN_CURR_SESS) {
602 s->flags &= ~SN_CURR_SESS;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100603 objt_server(s->target)->cur_sess--;
Willy Tarreau1e62de62008-11-11 20:20:02 +0100604 }
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100605 if (may_dequeue_tasks(objt_server(s->target), s->be))
606 process_srv_queue(objt_server(s->target));
Willy Tarreau1e62de62008-11-11 20:20:02 +0100607 }
Willy Tarreau922a8062008-12-04 09:33:58 +0100608
Willy Tarreau7c669d72008-06-20 15:04:11 +0200609 if (unlikely(s->srv_conn)) {
610 /* the session still has a reserved slot on a server, but
611 * it should normally be only the same as the one above,
612 * so this should not happen in fact.
613 */
614 sess_change_server(s, NULL);
615 }
616
Willy Tarreau543db622012-11-15 16:41:22 +0100617 if (s->flags & SN_COMP_READY)
William Lallemand1c2d6222012-10-30 15:52:53 +0100618 s->comp_algo->end(&s->comp_ctx);
Willy Tarreau543db622012-11-15 16:41:22 +0100619 s->comp_algo = NULL;
620 s->flags &= ~SN_COMP_READY;
William Lallemand82fe75c2012-10-23 10:25:10 +0200621
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100622 if (s->req->pipe)
623 put_pipe(s->req->pipe);
Willy Tarreau259de1b2009-01-18 21:56:21 +0100624
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100625 if (s->rep->pipe)
626 put_pipe(s->rep->pipe);
Willy Tarreau259de1b2009-01-18 21:56:21 +0100627
Willy Tarreau9b28e032012-10-12 23:49:43 +0200628 pool_free2(pool2_buffer, s->req->buf);
629 pool_free2(pool2_buffer, s->rep->buf);
630
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200631 pool_free2(pool2_channel, s->req);
632 pool_free2(pool2_channel, s->rep);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200633
Willy Tarreau46023632010-01-07 22:51:47 +0100634 http_end_txn(s);
635
Willy Tarreau1e954912012-10-12 17:50:05 +0200636 /* ensure the client-side transport layer is destroyed */
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200637 s->si[0].conn->flags &= ~CO_FL_XPRT_TRACKED;
Willy Tarreau0ede5a32012-12-08 08:44:02 +0100638 conn_full_close(s->si[0].conn);
Willy Tarreau1e954912012-10-12 17:50:05 +0200639
Willy Tarreaua4cda672010-06-06 18:28:49 +0200640 for (i = 0; i < s->store_count; i++) {
641 if (!s->store[i].ts)
642 continue;
643 stksess_free(s->store[i].table, s->store[i].ts);
644 s->store[i].ts = NULL;
645 }
646
Willy Tarreau34eb6712011-10-24 18:15:04 +0200647 pool_free2(pool2_hdr_idx, txn->hdr_idx.v);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200648 if (fe) {
Willy Tarreau46023632010-01-07 22:51:47 +0100649 pool_free2(fe->rsp_cap_pool, txn->rsp.cap);
650 pool_free2(fe->req_cap_pool, txn->req.cap);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200651 }
Willy Tarreau0937bc42009-12-22 15:03:09 +0100652
Willy Tarreau7af7d592013-07-01 18:07:03 +0200653 session_store_counters(s);
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +0200654
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100655 list_for_each_entry_safe(bref, back, &s->back_refs, users) {
Willy Tarreaufd3828e2009-02-22 15:17:24 +0100656 /* we have to unlink all watchers. We must not relink them if
657 * this session was the last one in the list.
658 */
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100659 LIST_DEL(&bref->users);
Willy Tarreaufd3828e2009-02-22 15:17:24 +0100660 LIST_INIT(&bref->users);
661 if (s->list.n != &sessions)
662 LIST_ADDQ(&LIST_ELEM(s->list.n, struct session *, list)->back_refs, &bref->users);
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100663 bref->ref = s->list.n;
664 }
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100665 LIST_DEL(&s->list);
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200666 pool_free2(pool2_connection, s->si[1].conn);
667 pool_free2(pool2_connection, s->si[0].conn);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200668 pool_free2(pool2_session, s);
Willy Tarreau632f5a72007-07-11 10:42:35 +0200669
670 /* We may want to free the maximum amount of pools if the proxy is stopping */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200671 if (fe && unlikely(fe->state == PR_STSTOPPED)) {
Willy Tarreau9b28e032012-10-12 23:49:43 +0200672 pool_flush2(pool2_buffer);
Willy Tarreau8263d2b2012-08-28 00:06:31 +0200673 pool_flush2(pool2_channel);
Willy Tarreau34eb6712011-10-24 18:15:04 +0200674 pool_flush2(pool2_hdr_idx);
Willy Tarreau48d63db2008-08-03 17:41:33 +0200675 pool_flush2(pool2_requri);
676 pool_flush2(pool2_capture);
677 pool_flush2(pool2_session);
678 pool_flush2(fe->req_cap_pool);
679 pool_flush2(fe->rsp_cap_pool);
Willy Tarreau632f5a72007-07-11 10:42:35 +0200680 }
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200681}
682
683
684/* perform minimal intializations, report 0 in case of error, 1 if OK. */
685int init_session()
686{
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100687 LIST_INIT(&sessions);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200688 pool2_session = create_pool("session", sizeof(struct session), MEM_F_SHARED);
689 return pool2_session != NULL;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200690}
691
Willy Tarreau30e71012007-11-26 20:15:35 +0100692void session_process_counters(struct session *s)
693{
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100694 unsigned long long bytes;
Willy Tarreau20d46a52012-12-09 15:55:40 +0100695 void *ptr;
696 int i;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100697
Willy Tarreau30e71012007-11-26 20:15:35 +0100698 if (s->req) {
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100699 bytes = s->req->total - s->logs.bytes_in;
Willy Tarreau30e71012007-11-26 20:15:35 +0100700 s->logs.bytes_in = s->req->total;
701 if (bytes) {
Willy Tarreau20d46a52012-12-09 15:55:40 +0100702 s->fe->fe_counters.bytes_in += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100703
Willy Tarreau20d46a52012-12-09 15:55:40 +0100704 s->be->be_counters.bytes_in += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100705
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100706 if (objt_server(s->target))
Willy Tarreau20d46a52012-12-09 15:55:40 +0100707 objt_server(s->target)->counters.bytes_in += bytes;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200708
709 if (s->listener->counters)
Willy Tarreau20d46a52012-12-09 15:55:40 +0100710 s->listener->counters->bytes_in += bytes;
Willy Tarreau855e4bb2010-06-18 18:33:32 +0200711
Willy Tarreaub4c84932013-07-23 19:15:30 +0200712 for (i = 0; i < MAX_SESS_STKCTR; i++) {
Willy Tarreau20d46a52012-12-09 15:55:40 +0100713 if (!s->stkctr[i].entry)
714 continue;
Willy Tarreau6c59e0a2010-06-20 11:56:30 +0200715
Willy Tarreau20d46a52012-12-09 15:55:40 +0100716 ptr = stktable_data_ptr(s->stkctr[i].table,
717 s->stkctr[i].entry,
718 STKTABLE_DT_BYTES_IN_CNT);
Willy Tarreau855e4bb2010-06-18 18:33:32 +0200719 if (ptr)
720 stktable_data_cast(ptr, bytes_in_cnt) += bytes;
Willy Tarreau6c59e0a2010-06-20 11:56:30 +0200721
Willy Tarreau20d46a52012-12-09 15:55:40 +0100722 ptr = stktable_data_ptr(s->stkctr[i].table,
723 s->stkctr[i].entry,
724 STKTABLE_DT_BYTES_IN_RATE);
Willy Tarreau6c59e0a2010-06-20 11:56:30 +0200725 if (ptr)
726 update_freq_ctr_period(&stktable_data_cast(ptr, bytes_in_rate),
Willy Tarreau20d46a52012-12-09 15:55:40 +0100727 s->stkctr[i].table->data_arg[STKTABLE_DT_BYTES_IN_RATE].u, bytes);
Willy Tarreau855e4bb2010-06-18 18:33:32 +0200728 }
Willy Tarreau30e71012007-11-26 20:15:35 +0100729 }
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100730 }
731
Willy Tarreau30e71012007-11-26 20:15:35 +0100732 if (s->rep) {
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100733 bytes = s->rep->total - s->logs.bytes_out;
Willy Tarreau30e71012007-11-26 20:15:35 +0100734 s->logs.bytes_out = s->rep->total;
735 if (bytes) {
Willy Tarreau20d46a52012-12-09 15:55:40 +0100736 s->fe->fe_counters.bytes_out += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100737
Willy Tarreau20d46a52012-12-09 15:55:40 +0100738 s->be->be_counters.bytes_out += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100739
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100740 if (objt_server(s->target))
Willy Tarreau20d46a52012-12-09 15:55:40 +0100741 objt_server(s->target)->counters.bytes_out += bytes;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200742
743 if (s->listener->counters)
Willy Tarreau20d46a52012-12-09 15:55:40 +0100744 s->listener->counters->bytes_out += bytes;
Willy Tarreauf059a0f2010-08-03 16:29:52 +0200745
Willy Tarreaub4c84932013-07-23 19:15:30 +0200746 for (i = 0; i < MAX_SESS_STKCTR; i++) {
Willy Tarreau20d46a52012-12-09 15:55:40 +0100747 if (!s->stkctr[i].entry)
748 continue;
Willy Tarreau6c59e0a2010-06-20 11:56:30 +0200749
Willy Tarreau20d46a52012-12-09 15:55:40 +0100750 ptr = stktable_data_ptr(s->stkctr[i].table,
751 s->stkctr[i].entry,
752 STKTABLE_DT_BYTES_OUT_CNT);
Willy Tarreau855e4bb2010-06-18 18:33:32 +0200753 if (ptr)
754 stktable_data_cast(ptr, bytes_out_cnt) += bytes;
Willy Tarreau6c59e0a2010-06-20 11:56:30 +0200755
Willy Tarreau20d46a52012-12-09 15:55:40 +0100756 ptr = stktable_data_ptr(s->stkctr[i].table,
757 s->stkctr[i].entry,
758 STKTABLE_DT_BYTES_OUT_RATE);
Willy Tarreau6c59e0a2010-06-20 11:56:30 +0200759 if (ptr)
760 update_freq_ctr_period(&stktable_data_cast(ptr, bytes_out_rate),
Willy Tarreau20d46a52012-12-09 15:55:40 +0100761 s->stkctr[i].table->data_arg[STKTABLE_DT_BYTES_OUT_RATE].u, bytes);
Willy Tarreau855e4bb2010-06-18 18:33:32 +0200762 }
Willy Tarreau30e71012007-11-26 20:15:35 +0100763 }
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100764 }
765}
Willy Tarreaubaaee002006-06-26 02:48:02 +0200766
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100767/* This function is called with (si->state == SI_ST_CON) meaning that a
768 * connection was attempted and that the file descriptor is already allocated.
769 * We must check for establishment, error and abort. Possible output states
770 * are SI_ST_EST (established), SI_ST_CER (error), SI_ST_DIS (abort), and
771 * SI_ST_CON (no change). The function returns 0 if it switches to SI_ST_CER,
772 * otherwise 1.
773 */
Simon Hormandec5be42011-06-08 09:19:07 +0900774static int sess_update_st_con_tcp(struct session *s, struct stream_interface *si)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100775{
Willy Tarreau7421efb2012-07-02 15:11:27 +0200776 struct channel *req = si->ob;
777 struct channel *rep = si->ib;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100778
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100779 /* If we got an error, or if nothing happened and the connection timed
780 * out, we must give up. The CER state handler will take care of retry
781 * attempts and error reports.
782 */
783 if (unlikely(si->flags & (SI_FL_EXP|SI_FL_ERR))) {
Willy Tarreaue3224e82012-10-29 22:41:31 +0100784 if (unlikely(si->ob->flags & CF_WRITE_PARTIAL)) {
785 /* Some data were sent past the connection establishment,
786 * so we need to pretend we're established to log correctly
787 * and let later states handle the failure.
788 */
789 s->logs.t_connect = tv_ms_elapsed(&s->logs.tv_accept, &now);
790 si->exp = TICK_ETERNITY;
791 si->state = SI_ST_EST;
792 si->err_type = SI_ET_DATA_ERR;
793 si->ib->flags |= CF_READ_ERROR | CF_WRITE_ERROR;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100794 si->err_loc = objt_server(s->target);
Willy Tarreaue3224e82012-10-29 22:41:31 +0100795 return 1;
796 }
Willy Tarreau127334e2009-03-28 10:47:26 +0100797 si->exp = TICK_ETERNITY;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100798 si->state = SI_ST_CER;
Willy Tarreau0ede5a32012-12-08 08:44:02 +0100799
800 si->conn->flags &= ~CO_FL_XPRT_TRACKED;
801 conn_full_close(si->conn);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100802
Willy Tarreau0bd05ea2010-07-02 11:18:03 +0200803 if (si->release)
804 si->release(si);
805
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100806 if (si->err_type)
807 return 0;
808
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100809 si->err_loc = objt_server(s->target);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100810 if (si->flags & SI_FL_ERR)
811 si->err_type = SI_ET_CONN_ERR;
812 else
813 si->err_type = SI_ET_CONN_TO;
814 return 0;
815 }
816
817 /* OK, maybe we want to abort */
Willy Tarreaua7a7ebc2012-12-30 00:50:35 +0100818 if (!(req->flags & CF_WRITE_PARTIAL) &&
819 unlikely((rep->flags & CF_SHUTW) ||
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200820 ((req->flags & CF_SHUTW_NOW) && /* FIXME: this should not prevent a connection from establishing */
821 ((!(req->flags & CF_WRITE_ACTIVITY) && channel_is_empty(req)) ||
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100822 s->be->options & PR_O_ABRT_CLOSE)))) {
823 /* give up */
Willy Tarreau73b013b2012-05-21 16:31:45 +0200824 si_shutw(si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100825 si->err_type |= SI_ET_CONN_ABRT;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100826 si->err_loc = objt_server(s->target);
Willy Tarreau84455332009-03-15 22:34:05 +0100827 if (s->srv_error)
828 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100829 return 1;
830 }
831
832 /* we need to wait a bit more if there was no activity either */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200833 if (!(req->flags & CF_WRITE_ACTIVITY))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100834 return 1;
835
836 /* OK, this means that a connection succeeded. The caller will be
837 * responsible for handling the transition from CON to EST.
838 */
839 s->logs.t_connect = tv_ms_elapsed(&s->logs.tv_accept, &now);
Willy Tarreau127334e2009-03-28 10:47:26 +0100840 si->exp = TICK_ETERNITY;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100841 si->state = SI_ST_EST;
842 si->err_type = SI_ET_NONE;
843 si->err_loc = NULL;
844 return 1;
845}
846
847/* This function is called with (si->state == SI_ST_CER) meaning that a
848 * previous connection attempt has failed and that the file descriptor
849 * has already been released. Possible causes include asynchronous error
850 * notification and time out. Possible output states are SI_ST_CLO when
851 * retries are exhausted, SI_ST_TAR when a delay is wanted before a new
852 * connection attempt, SI_ST_ASS when it's wise to retry on the same server,
853 * and SI_ST_REQ when an immediate redispatch is wanted. The buffers are
854 * marked as in error state. It returns 0.
855 */
Simon Hormandec5be42011-06-08 09:19:07 +0900856static int sess_update_st_cer(struct session *s, struct stream_interface *si)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100857{
858 /* we probably have to release last session from the server */
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100859 if (objt_server(s->target)) {
860 health_adjust(objt_server(s->target), HANA_STATUS_L4_ERR);
Krzysztof Piotr Oledzki97f07b82009-12-15 22:31:24 +0100861
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100862 if (s->flags & SN_CURR_SESS) {
863 s->flags &= ~SN_CURR_SESS;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100864 objt_server(s->target)->cur_sess--;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100865 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100866 }
867
868 /* ensure that we have enough retries left */
Willy Tarreauee28de02010-06-01 09:51:00 +0200869 si->conn_retries--;
870 if (si->conn_retries < 0) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100871 if (!si->err_type) {
872 si->err_type = SI_ET_CONN_ERR;
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100873 si->err_loc = objt_server(s->target);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100874 }
875
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100876 if (objt_server(s->target))
877 objt_server(s->target)->counters.failed_conns++;
Willy Tarreau7d0aaf32011-03-10 23:25:56 +0100878 s->be->be_counters.failed_conns++;
Willy Tarreaub89cfca2010-12-29 14:32:28 +0100879 sess_change_server(s, NULL);
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100880 if (may_dequeue_tasks(objt_server(s->target), s->be))
881 process_srv_queue(objt_server(s->target));
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100882
883 /* shutw is enough so stop a connecting socket */
Willy Tarreau73b013b2012-05-21 16:31:45 +0200884 si_shutw(si);
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200885 si->ob->flags |= CF_WRITE_ERROR;
886 si->ib->flags |= CF_READ_ERROR;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100887
888 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100889 if (s->srv_error)
890 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100891 return 0;
892 }
893
894 /* If the "redispatch" option is set on the backend, we are allowed to
895 * retry on another server for the last retry. In order to achieve this,
896 * we must mark the session unassigned, and eventually clear the DIRECT
897 * bit to ignore any persistence cookie. We won't count a retry nor a
898 * redispatch yet, because this will depend on what server is selected.
899 */
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100900 if (objt_server(s->target) && si->conn_retries == 0 &&
Willy Tarreau4de91492010-01-22 19:10:05 +0100901 s->be->options & PR_O_REDISP && !(s->flags & SN_FORCE_PRST)) {
Willy Tarreaub89cfca2010-12-29 14:32:28 +0100902 sess_change_server(s, NULL);
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100903 if (may_dequeue_tasks(objt_server(s->target), s->be))
904 process_srv_queue(objt_server(s->target));
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100905
906 s->flags &= ~(SN_DIRECT | SN_ASSIGNED | SN_ADDR_SET);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100907 si->state = SI_ST_REQ;
908 } else {
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100909 if (objt_server(s->target))
910 objt_server(s->target)->counters.retries++;
Willy Tarreau7d0aaf32011-03-10 23:25:56 +0100911 s->be->be_counters.retries++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100912 si->state = SI_ST_ASS;
913 }
914
915 if (si->flags & SI_FL_ERR) {
916 /* The error was an asynchronous connection error, and we will
917 * likely have to retry connecting to the same server, most
918 * likely leading to the same result. To avoid this, we wait
919 * one second before retrying.
920 */
921
922 if (!si->err_type)
923 si->err_type = SI_ET_CONN_ERR;
924
925 si->state = SI_ST_TAR;
926 si->exp = tick_add(now_ms, MS_TO_TICKS(1000));
927 return 0;
928 }
929 return 0;
930}
931
932/*
933 * This function handles the transition between the SI_ST_CON state and the
Willy Tarreau85e7d002010-05-31 11:57:51 +0200934 * SI_ST_EST state. It must only be called after switching from SI_ST_CON (or
Willy Tarreau26d8c592012-05-07 18:12:14 +0200935 * SI_ST_INI) to SI_ST_EST, but only when a ->proto is defined.
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100936 */
Simon Hormandec5be42011-06-08 09:19:07 +0900937static void sess_establish(struct session *s, struct stream_interface *si)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100938{
Willy Tarreau7421efb2012-07-02 15:11:27 +0200939 struct channel *req = si->ob;
940 struct channel *rep = si->ib;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100941
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100942 if (objt_server(s->target))
943 health_adjust(objt_server(s->target), HANA_STATUS_L4_OK);
Krzysztof Piotr Oledzki97f07b82009-12-15 22:31:24 +0100944
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100945 if (s->be->mode == PR_MODE_TCP) { /* let's allow immediate data connection in this case */
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100946 /* if the user wants to log as soon as possible, without counting
947 * bytes from the server, then this is the right moment. */
Willy Tarreaud79a3b22012-12-28 09:40:16 +0100948 if (!LIST_ISEMPTY(&s->fe->logformat) && !(s->logs.logwait & LW_BYTES)) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100949 s->logs.t_close = s->logs.t_connect; /* to get a valid end date */
Willy Tarreaua5555ec2008-11-30 19:02:32 +0100950 s->do_log(s);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100951 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100952 }
953 else {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100954 s->txn.rsp.msg_state = HTTP_MSG_RPBEFORE;
955 /* reset hdr_idx which was already initialized by the request.
956 * right now, the http parser does it.
957 * hdr_idx_init(&s->txn.hdr_idx);
958 */
959 }
960
Willy Tarreau4e5b8282009-08-16 22:57:50 +0200961 rep->analysers |= s->fe->fe_rsp_ana | s->be->be_rsp_ana;
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200962 rep->flags |= CF_READ_ATTACHED; /* producer is now attached */
Willy Tarreau73b013b2012-05-21 16:31:45 +0200963 if (si_ctrl(si)) {
Willy Tarreaud04e8582010-05-31 12:31:35 +0200964 /* real connections have timeouts */
965 req->wto = s->be->timeout.server;
966 rep->rto = s->be->timeout.server;
967 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100968 req->wex = TICK_ETERNITY;
969}
970
971/* Update stream interface status for input states SI_ST_ASS, SI_ST_QUE, SI_ST_TAR.
972 * Other input states are simply ignored.
973 * Possible output states are SI_ST_CLO, SI_ST_TAR, SI_ST_ASS, SI_ST_REQ, SI_ST_CON.
974 * Flags must have previously been updated for timeouts and other conditions.
975 */
Simon Hormandec5be42011-06-08 09:19:07 +0900976static void sess_update_stream_int(struct session *s, struct stream_interface *si)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100977{
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100978 struct server *srv = objt_server(s->target);
Willy Tarreau827aee92011-03-10 16:55:02 +0100979
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100980 DPRINTF(stderr,"[%u] %s: sess=%p rq=%p, rp=%p, exp(r,w)=%u,%u rqf=%08x rpf=%08x rqh=%d rqt=%d rph=%d rpt=%d cs=%d ss=%d\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100981 now_ms, __FUNCTION__,
982 s,
983 s->req, s->rep,
984 s->req->rex, s->rep->wex,
985 s->req->flags, s->rep->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +0200986 s->req->buf->i, s->req->buf->o, s->rep->buf->i, s->rep->buf->o, s->rep->cons->state, s->req->cons->state);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100987
988 if (si->state == SI_ST_ASS) {
989 /* Server assigned to connection request, we have to try to connect now */
990 int conn_err;
991
992 conn_err = connect_server(s);
Willy Tarreau3fdb3662012-11-12 00:42:33 +0100993 srv = objt_server(s->target);
Willy Tarreau827aee92011-03-10 16:55:02 +0100994
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100995 if (conn_err == SN_ERR_NONE) {
996 /* state = SI_ST_CON now */
Willy Tarreau827aee92011-03-10 16:55:02 +0100997 if (srv)
998 srv_inc_sess_ctr(srv);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100999 return;
1000 }
1001
1002 /* We have received a synchronous error. We might have to
1003 * abort, retry immediately or redispatch.
1004 */
1005 if (conn_err == SN_ERR_INTERNAL) {
1006 if (!si->err_type) {
1007 si->err_type = SI_ET_CONN_OTHER;
Willy Tarreau827aee92011-03-10 16:55:02 +01001008 si->err_loc = srv;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001009 }
1010
Willy Tarreau827aee92011-03-10 16:55:02 +01001011 if (srv)
1012 srv_inc_sess_ctr(srv);
1013 if (srv)
1014 srv->counters.failed_conns++;
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001015 s->be->be_counters.failed_conns++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001016
1017 /* release other sessions waiting for this server */
Willy Tarreaub89cfca2010-12-29 14:32:28 +01001018 sess_change_server(s, NULL);
Willy Tarreau827aee92011-03-10 16:55:02 +01001019 if (may_dequeue_tasks(srv, s->be))
1020 process_srv_queue(srv);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001021
1022 /* Failed and not retryable. */
Willy Tarreau73b013b2012-05-21 16:31:45 +02001023 si_shutr(si);
1024 si_shutw(si);
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001025 si->ob->flags |= CF_WRITE_ERROR;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001026
1027 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
1028
1029 /* no session was ever accounted for this server */
1030 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +01001031 if (s->srv_error)
1032 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001033 return;
1034 }
1035
1036 /* We are facing a retryable error, but we don't want to run a
1037 * turn-around now, as the problem is likely a source port
1038 * allocation problem, so we want to retry now.
1039 */
1040 si->state = SI_ST_CER;
1041 si->flags &= ~SI_FL_ERR;
1042 sess_update_st_cer(s, si);
1043 /* now si->state is one of SI_ST_CLO, SI_ST_TAR, SI_ST_ASS, SI_ST_REQ */
1044 return;
1045 }
1046 else if (si->state == SI_ST_QUE) {
1047 /* connection request was queued, check for any update */
1048 if (!s->pend_pos) {
1049 /* The connection is not in the queue anymore. Either
1050 * we have a server connection slot available and we
1051 * go directly to the assigned state, or we need to
1052 * load-balance first and go to the INI state.
1053 */
1054 si->exp = TICK_ETERNITY;
1055 if (unlikely(!(s->flags & SN_ASSIGNED)))
1056 si->state = SI_ST_REQ;
1057 else {
1058 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
1059 si->state = SI_ST_ASS;
1060 }
1061 return;
1062 }
1063
1064 /* Connection request still in queue... */
1065 if (si->flags & SI_FL_EXP) {
1066 /* ... and timeout expired */
1067 si->exp = TICK_ETERNITY;
1068 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
Willy Tarreau827aee92011-03-10 16:55:02 +01001069 if (srv)
1070 srv->counters.failed_conns++;
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001071 s->be->be_counters.failed_conns++;
Willy Tarreau73b013b2012-05-21 16:31:45 +02001072 si_shutr(si);
1073 si_shutw(si);
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001074 si->ob->flags |= CF_WRITE_TIMEOUT;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001075 if (!si->err_type)
1076 si->err_type = SI_ET_QUEUE_TO;
1077 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +01001078 if (s->srv_error)
1079 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001080 return;
1081 }
1082
1083 /* Connection remains in queue, check if we have to abort it */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001084 if ((si->ob->flags & (CF_READ_ERROR)) ||
1085 ((si->ob->flags & CF_SHUTW_NOW) && /* empty and client aborted */
Willy Tarreau8e21bb92012-08-24 22:40:29 +02001086 (channel_is_empty(si->ob) || s->be->options & PR_O_ABRT_CLOSE))) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001087 /* give up */
1088 si->exp = TICK_ETERNITY;
1089 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
Willy Tarreau73b013b2012-05-21 16:31:45 +02001090 si_shutr(si);
1091 si_shutw(si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001092 si->err_type |= SI_ET_QUEUE_ABRT;
1093 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +01001094 if (s->srv_error)
1095 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001096 return;
1097 }
1098
1099 /* Nothing changed */
1100 return;
1101 }
1102 else if (si->state == SI_ST_TAR) {
1103 /* Connection request might be aborted */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001104 if ((si->ob->flags & (CF_READ_ERROR)) ||
1105 ((si->ob->flags & CF_SHUTW_NOW) && /* empty and client aborted */
Willy Tarreau8e21bb92012-08-24 22:40:29 +02001106 (channel_is_empty(si->ob) || s->be->options & PR_O_ABRT_CLOSE))) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001107 /* give up */
1108 si->exp = TICK_ETERNITY;
Willy Tarreau73b013b2012-05-21 16:31:45 +02001109 si_shutr(si);
1110 si_shutw(si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001111 si->err_type |= SI_ET_CONN_ABRT;
1112 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +01001113 if (s->srv_error)
1114 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001115 return;
1116 }
1117
1118 if (!(si->flags & SI_FL_EXP))
1119 return; /* still in turn-around */
1120
1121 si->exp = TICK_ETERNITY;
1122
1123 /* we keep trying on the same server as long as the session is
1124 * marked "assigned".
1125 * FIXME: Should we force a redispatch attempt when the server is down ?
1126 */
1127 if (s->flags & SN_ASSIGNED)
1128 si->state = SI_ST_ASS;
1129 else
1130 si->state = SI_ST_REQ;
1131 return;
1132 }
1133}
1134
Simon Hormandec5be42011-06-08 09:19:07 +09001135/* Set correct session termination flags in case no analyser has done it. It
1136 * also counts a failed request if the server state has not reached the request
1137 * stage.
1138 */
1139static void sess_set_term_flags(struct session *s)
1140{
1141 if (!(s->flags & SN_FINST_MASK)) {
1142 if (s->si[1].state < SI_ST_REQ) {
1143
1144 s->fe->fe_counters.failed_req++;
1145 if (s->listener->counters)
1146 s->listener->counters->failed_req++;
1147
1148 s->flags |= SN_FINST_R;
1149 }
1150 else if (s->si[1].state == SI_ST_QUE)
1151 s->flags |= SN_FINST_Q;
1152 else if (s->si[1].state < SI_ST_EST)
1153 s->flags |= SN_FINST_C;
1154 else if (s->si[1].state == SI_ST_EST || s->si[1].prev_state == SI_ST_EST)
1155 s->flags |= SN_FINST_D;
1156 else
1157 s->flags |= SN_FINST_L;
1158 }
1159}
1160
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001161/* This function initiates a server connection request on a stream interface
1162 * already in SI_ST_REQ state. Upon success, the state goes to SI_ST_ASS,
1163 * indicating that a server has been assigned. It may also return SI_ST_QUE,
1164 * or SI_ST_CLO upon error.
1165 */
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001166static void sess_prepare_conn_req(struct session *s, struct stream_interface *si)
1167{
1168 DPRINTF(stderr,"[%u] %s: sess=%p rq=%p, rp=%p, exp(r,w)=%u,%u rqf=%08x rpf=%08x rqh=%d rqt=%d rph=%d rpt=%d cs=%d ss=%d\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001169 now_ms, __FUNCTION__,
1170 s,
1171 s->req, s->rep,
1172 s->req->rex, s->rep->wex,
1173 s->req->flags, s->rep->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +02001174 s->req->buf->i, s->req->buf->o, s->rep->buf->i, s->rep->buf->o, s->rep->cons->state, s->req->cons->state);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001175
1176 if (si->state != SI_ST_REQ)
1177 return;
1178
1179 /* Try to assign a server */
1180 if (srv_redispatch_connect(s) != 0) {
1181 /* We did not get a server. Either we queued the
1182 * connection request, or we encountered an error.
1183 */
1184 if (si->state == SI_ST_QUE)
1185 return;
1186
1187 /* we did not get any server, let's check the cause */
Willy Tarreau73b013b2012-05-21 16:31:45 +02001188 si_shutr(si);
1189 si_shutw(si);
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001190 si->ob->flags |= CF_WRITE_ERROR;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001191 if (!si->err_type)
1192 si->err_type = SI_ET_CONN_OTHER;
1193 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +01001194 if (s->srv_error)
1195 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001196 return;
1197 }
1198
1199 /* The server is assigned */
1200 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
1201 si->state = SI_ST_ASS;
1202}
1203
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001204/* This stream analyser checks the switching rules and changes the backend
Willy Tarreau4de91492010-01-22 19:10:05 +01001205 * if appropriate. The default_backend rule is also considered, then the
1206 * target backend's forced persistence rules are also evaluated last if any.
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001207 * It returns 1 if the processing can continue on next analysers, or zero if it
1208 * either needs more data or wants to immediately abort the request.
1209 */
Willy Tarreau7421efb2012-07-02 15:11:27 +02001210static int process_switching_rules(struct session *s, struct channel *req, int an_bit)
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001211{
Cyril Bonté47fdd8e2010-04-25 00:00:51 +02001212 struct persist_rule *prst_rule;
Willy Tarreau4de91492010-01-22 19:10:05 +01001213
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001214 req->analysers &= ~an_bit;
1215 req->analyse_exp = TICK_ETERNITY;
1216
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001217 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bh=%d analysers=%02x\n",
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001218 now_ms, __FUNCTION__,
1219 s,
1220 req,
1221 req->rex, req->wex,
1222 req->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +02001223 req->buf->i,
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001224 req->analysers);
1225
1226 /* now check whether we have some switching rules for this request */
1227 if (!(s->flags & SN_BE_ASSIGNED)) {
1228 struct switching_rule *rule;
1229
1230 list_for_each_entry(rule, &s->fe->switching_rules, list) {
1231 int ret;
1232
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001233 ret = acl_exec_cond(rule->cond, s->fe, s, &s->txn, SMP_OPT_DIR_REQ|SMP_OPT_FINAL);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001234 ret = acl_pass(ret);
1235 if (rule->cond->pol == ACL_COND_UNLESS)
1236 ret = !ret;
1237
1238 if (ret) {
Willy Tarreaubedb9ba2009-07-12 08:27:39 +02001239 if (!session_set_backend(s, rule->be.backend))
1240 goto sw_failed;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001241 break;
1242 }
1243 }
1244
1245 /* To ensure correct connection accounting on the backend, we
1246 * have to assign one if it was not set (eg: a listen). This
1247 * measure also takes care of correctly setting the default
1248 * backend if any.
1249 */
1250 if (!(s->flags & SN_BE_ASSIGNED))
Willy Tarreaubedb9ba2009-07-12 08:27:39 +02001251 if (!session_set_backend(s, s->fe->defbe.be ? s->fe->defbe.be : s->be))
1252 goto sw_failed;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001253 }
1254
Willy Tarreaufb356202010-08-03 14:02:05 +02001255 /* we don't want to run the TCP or HTTP filters again if the backend has not changed */
1256 if (s->fe == s->be) {
1257 s->req->analysers &= ~AN_REQ_INSPECT_BE;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001258 s->req->analysers &= ~AN_REQ_HTTP_PROCESS_BE;
Willy Tarreaufb356202010-08-03 14:02:05 +02001259 }
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001260
Cyril Bonté47fdd8e2010-04-25 00:00:51 +02001261 /* as soon as we know the backend, we must check if we have a matching forced or ignored
Willy Tarreau4de91492010-01-22 19:10:05 +01001262 * persistence rule, and report that in the session.
1263 */
Cyril Bonté47fdd8e2010-04-25 00:00:51 +02001264 list_for_each_entry(prst_rule, &s->be->persist_rules, list) {
Willy Tarreau4de91492010-01-22 19:10:05 +01001265 int ret = 1;
1266
1267 if (prst_rule->cond) {
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001268 ret = acl_exec_cond(prst_rule->cond, s->be, s, &s->txn, SMP_OPT_DIR_REQ|SMP_OPT_FINAL);
Willy Tarreau4de91492010-01-22 19:10:05 +01001269 ret = acl_pass(ret);
1270 if (prst_rule->cond->pol == ACL_COND_UNLESS)
1271 ret = !ret;
1272 }
1273
1274 if (ret) {
1275 /* no rule, or the rule matches */
Cyril Bonté47fdd8e2010-04-25 00:00:51 +02001276 if (prst_rule->type == PERSIST_TYPE_FORCE) {
1277 s->flags |= SN_FORCE_PRST;
1278 } else {
1279 s->flags |= SN_IGNORE_PRST;
1280 }
Willy Tarreau4de91492010-01-22 19:10:05 +01001281 break;
1282 }
1283 }
1284
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001285 return 1;
Willy Tarreaubedb9ba2009-07-12 08:27:39 +02001286
1287 sw_failed:
1288 /* immediately abort this request in case of allocation failure */
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001289 channel_abort(s->req);
1290 channel_abort(s->rep);
Willy Tarreaubedb9ba2009-07-12 08:27:39 +02001291
1292 if (!(s->flags & SN_ERR_MASK))
1293 s->flags |= SN_ERR_RESOURCE;
1294 if (!(s->flags & SN_FINST_MASK))
1295 s->flags |= SN_FINST_R;
1296
1297 s->txn.status = 500;
1298 s->req->analysers = 0;
1299 s->req->analyse_exp = TICK_ETERNITY;
1300 return 0;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001301}
1302
Willy Tarreau4a5cade2012-04-05 21:09:48 +02001303/* This stream analyser works on a request. It applies all use-server rules on
1304 * it then returns 1. The data must already be present in the buffer otherwise
1305 * they won't match. It always returns 1.
1306 */
Willy Tarreau7421efb2012-07-02 15:11:27 +02001307static int process_server_rules(struct session *s, struct channel *req, int an_bit)
Willy Tarreau4a5cade2012-04-05 21:09:48 +02001308{
1309 struct proxy *px = s->be;
1310 struct server_rule *rule;
1311
1312 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bl=%d analysers=%02x\n",
1313 now_ms, __FUNCTION__,
1314 s,
1315 req,
1316 req->rex, req->wex,
1317 req->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +02001318 req->buf->i + req->buf->o,
Willy Tarreau4a5cade2012-04-05 21:09:48 +02001319 req->analysers);
1320
1321 if (!(s->flags & SN_ASSIGNED)) {
1322 list_for_each_entry(rule, &px->server_rules, list) {
1323 int ret;
1324
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001325 ret = acl_exec_cond(rule->cond, s->be, s, &s->txn, SMP_OPT_DIR_REQ|SMP_OPT_FINAL);
Willy Tarreau4a5cade2012-04-05 21:09:48 +02001326 ret = acl_pass(ret);
1327 if (rule->cond->pol == ACL_COND_UNLESS)
1328 ret = !ret;
1329
1330 if (ret) {
1331 struct server *srv = rule->srv.ptr;
1332
1333 if ((srv->state & SRV_RUNNING) ||
1334 (px->options & PR_O_PERSIST) ||
1335 (s->flags & SN_FORCE_PRST)) {
1336 s->flags |= SN_DIRECT | SN_ASSIGNED;
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001337 s->target = &srv->obj_type;
Willy Tarreau4a5cade2012-04-05 21:09:48 +02001338 break;
1339 }
1340 /* if the server is not UP, let's go on with next rules
1341 * just in case another one is suited.
1342 */
1343 }
1344 }
1345 }
1346
1347 req->analysers &= ~an_bit;
1348 req->analyse_exp = TICK_ETERNITY;
1349 return 1;
1350}
1351
Emeric Brun1d33b292010-01-04 15:47:17 +01001352/* This stream analyser works on a request. It applies all sticking rules on
1353 * it then returns 1. The data must already be present in the buffer otherwise
1354 * they won't match. It always returns 1.
1355 */
Willy Tarreau7421efb2012-07-02 15:11:27 +02001356static int process_sticking_rules(struct session *s, struct channel *req, int an_bit)
Emeric Brun1d33b292010-01-04 15:47:17 +01001357{
1358 struct proxy *px = s->be;
1359 struct sticking_rule *rule;
1360
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001361 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bh=%d analysers=%02x\n",
Emeric Brun1d33b292010-01-04 15:47:17 +01001362 now_ms, __FUNCTION__,
1363 s,
1364 req,
1365 req->rex, req->wex,
1366 req->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +02001367 req->buf->i,
Emeric Brun1d33b292010-01-04 15:47:17 +01001368 req->analysers);
1369
1370 list_for_each_entry(rule, &px->sticking_rules, list) {
1371 int ret = 1 ;
1372 int i;
1373
1374 for (i = 0; i < s->store_count; i++) {
1375 if (rule->table.t == s->store[i].table)
1376 break;
1377 }
1378
1379 if (i != s->store_count)
1380 continue;
1381
1382 if (rule->cond) {
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001383 ret = acl_exec_cond(rule->cond, px, s, &s->txn, SMP_OPT_DIR_REQ|SMP_OPT_FINAL);
Emeric Brun1d33b292010-01-04 15:47:17 +01001384 ret = acl_pass(ret);
1385 if (rule->cond->pol == ACL_COND_UNLESS)
1386 ret = !ret;
1387 }
1388
1389 if (ret) {
1390 struct stktable_key *key;
1391
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001392 key = stktable_fetch_key(rule->table.t, px, s, &s->txn, SMP_OPT_DIR_REQ|SMP_OPT_FINAL, rule->expr);
Emeric Brun1d33b292010-01-04 15:47:17 +01001393 if (!key)
1394 continue;
1395
1396 if (rule->flags & STK_IS_MATCH) {
1397 struct stksess *ts;
1398
Willy Tarreauf16d2b82010-06-06 15:38:59 +02001399 if ((ts = stktable_lookup_key(rule->table.t, key)) != NULL) {
Emeric Brun1d33b292010-01-04 15:47:17 +01001400 if (!(s->flags & SN_ASSIGNED)) {
1401 struct eb32_node *node;
Willy Tarreau13c29de2010-06-06 16:40:39 +02001402 void *ptr;
Emeric Brun1d33b292010-01-04 15:47:17 +01001403
1404 /* srv found in table */
Willy Tarreau13c29de2010-06-06 16:40:39 +02001405 ptr = stktable_data_ptr(rule->table.t, ts, STKTABLE_DT_SERVER_ID);
1406 node = eb32_lookup(&px->conf.used_server_id, stktable_data_cast(ptr, server_id));
Emeric Brun1d33b292010-01-04 15:47:17 +01001407 if (node) {
1408 struct server *srv;
1409
1410 srv = container_of(node, struct server, conf.id);
Willy Tarreau4de91492010-01-22 19:10:05 +01001411 if ((srv->state & SRV_RUNNING) ||
1412 (px->options & PR_O_PERSIST) ||
1413 (s->flags & SN_FORCE_PRST)) {
Emeric Brun1d33b292010-01-04 15:47:17 +01001414 s->flags |= SN_DIRECT | SN_ASSIGNED;
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001415 s->target = &srv->obj_type;
Emeric Brun1d33b292010-01-04 15:47:17 +01001416 }
1417 }
1418 }
Emeric Brun85e77c72010-09-23 18:16:52 +02001419 stktable_touch(rule->table.t, ts, 1);
Emeric Brun1d33b292010-01-04 15:47:17 +01001420 }
1421 }
1422 if (rule->flags & STK_IS_STORE) {
1423 if (s->store_count < (sizeof(s->store) / sizeof(s->store[0]))) {
1424 struct stksess *ts;
1425
1426 ts = stksess_new(rule->table.t, key);
1427 if (ts) {
1428 s->store[s->store_count].table = rule->table.t;
1429 s->store[s->store_count++].ts = ts;
1430 }
1431 }
1432 }
1433 }
1434 }
1435
1436 req->analysers &= ~an_bit;
1437 req->analyse_exp = TICK_ETERNITY;
1438 return 1;
1439}
1440
1441/* This stream analyser works on a response. It applies all store rules on it
1442 * then returns 1. The data must already be present in the buffer otherwise
1443 * they won't match. It always returns 1.
1444 */
Willy Tarreau7421efb2012-07-02 15:11:27 +02001445static int process_store_rules(struct session *s, struct channel *rep, int an_bit)
Emeric Brun1d33b292010-01-04 15:47:17 +01001446{
1447 struct proxy *px = s->be;
1448 struct sticking_rule *rule;
1449 int i;
1450
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001451 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bh=%d analysers=%02x\n",
Emeric Brun1d33b292010-01-04 15:47:17 +01001452 now_ms, __FUNCTION__,
1453 s,
Willy Tarreau2e2b3eb2010-02-09 20:55:44 +01001454 rep,
1455 rep->rex, rep->wex,
1456 rep->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +02001457 rep->buf->i,
Willy Tarreau2e2b3eb2010-02-09 20:55:44 +01001458 rep->analysers);
Emeric Brun1d33b292010-01-04 15:47:17 +01001459
1460 list_for_each_entry(rule, &px->storersp_rules, list) {
1461 int ret = 1 ;
1462 int storereqidx = -1;
1463
1464 for (i = 0; i < s->store_count; i++) {
1465 if (rule->table.t == s->store[i].table) {
1466 if (!(s->store[i].flags))
1467 storereqidx = i;
1468 break;
1469 }
1470 }
1471
1472 if ((i != s->store_count) && (storereqidx == -1))
1473 continue;
1474
1475 if (rule->cond) {
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001476 ret = acl_exec_cond(rule->cond, px, s, &s->txn, SMP_OPT_DIR_RES|SMP_OPT_FINAL);
Emeric Brun1d33b292010-01-04 15:47:17 +01001477 ret = acl_pass(ret);
1478 if (rule->cond->pol == ACL_COND_UNLESS)
1479 ret = !ret;
1480 }
1481
1482 if (ret) {
1483 struct stktable_key *key;
1484
Willy Tarreau32a6f2e2012-04-25 10:13:36 +02001485 key = stktable_fetch_key(rule->table.t, px, s, &s->txn, SMP_OPT_DIR_RES|SMP_OPT_FINAL, rule->expr);
Emeric Brun1d33b292010-01-04 15:47:17 +01001486 if (!key)
1487 continue;
1488
1489 if (storereqidx != -1) {
Willy Tarreau393379c2010-06-06 12:11:37 +02001490 stksess_setkey(s->store[storereqidx].table, s->store[storereqidx].ts, key);
Emeric Brun1d33b292010-01-04 15:47:17 +01001491 s->store[storereqidx].flags = 1;
1492 }
1493 else if (s->store_count < (sizeof(s->store) / sizeof(s->store[0]))) {
1494 struct stksess *ts;
1495
1496 ts = stksess_new(rule->table.t, key);
1497 if (ts) {
1498 s->store[s->store_count].table = rule->table.t;
1499 s->store[s->store_count].flags = 1;
1500 s->store[s->store_count++].ts = ts;
1501 }
1502 }
1503 }
1504 }
1505
1506 /* process store request and store response */
1507 for (i = 0; i < s->store_count; i++) {
Willy Tarreauf16d2b82010-06-06 15:38:59 +02001508 struct stksess *ts;
Willy Tarreau13c29de2010-06-06 16:40:39 +02001509 void *ptr;
Willy Tarreauf16d2b82010-06-06 15:38:59 +02001510
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001511 if (objt_server(s->target) && objt_server(s->target)->state & SRV_NON_STICK) {
Simon Hormanfa461682011-06-25 09:39:49 +09001512 stksess_free(s->store[i].table, s->store[i].ts);
1513 s->store[i].ts = NULL;
1514 continue;
1515 }
1516
Willy Tarreauf16d2b82010-06-06 15:38:59 +02001517 ts = stktable_lookup(s->store[i].table, s->store[i].ts);
1518 if (ts) {
1519 /* the entry already existed, we can free ours */
Emeric Brun85e77c72010-09-23 18:16:52 +02001520 stktable_touch(s->store[i].table, ts, 1);
Emeric Brun1d33b292010-01-04 15:47:17 +01001521 stksess_free(s->store[i].table, s->store[i].ts);
Emeric Brun1d33b292010-01-04 15:47:17 +01001522 }
Willy Tarreauf16d2b82010-06-06 15:38:59 +02001523 else
Emeric Brun85e77c72010-09-23 18:16:52 +02001524 ts = stktable_store(s->store[i].table, s->store[i].ts, 1);
Willy Tarreauf16d2b82010-06-06 15:38:59 +02001525
1526 s->store[i].ts = NULL;
Willy Tarreau13c29de2010-06-06 16:40:39 +02001527 ptr = stktable_data_ptr(s->store[i].table, ts, STKTABLE_DT_SERVER_ID);
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001528 stktable_data_cast(ptr, server_id) = objt_server(s->target)->puid;
Emeric Brun1d33b292010-01-04 15:47:17 +01001529 }
Willy Tarreau2a164ee2010-06-18 09:57:45 +02001530 s->store_count = 0; /* everything is stored */
Emeric Brun1d33b292010-01-04 15:47:17 +01001531
1532 rep->analysers &= ~an_bit;
1533 rep->analyse_exp = TICK_ETERNITY;
1534 return 1;
1535}
1536
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001537/* This macro is very specific to the function below. See the comments in
1538 * process_session() below to understand the logic and the tests.
1539 */
1540#define UPDATE_ANALYSERS(real, list, back, flag) { \
1541 list = (((list) & ~(flag)) | ~(back)) & (real); \
1542 back = real; \
1543 if (!(list)) \
1544 break; \
1545 if (((list) ^ ((list) & ((list) - 1))) < (flag)) \
1546 continue; \
1547}
1548
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001549/* Processes the client, server, request and response jobs of a session task,
1550 * then puts it back to the wait queue in a clean state, or cleans up its
1551 * resources if it must be deleted. Returns in <next> the date the task wants
1552 * to be woken up, or TICK_ETERNITY. In order not to call all functions for
1553 * nothing too many times, the request and response buffers flags are monitored
1554 * and each function is called only if at least another function has changed at
1555 * least one flag it is interested in.
1556 */
Willy Tarreau26c25062009-03-08 09:38:41 +01001557struct task *process_session(struct task *t)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001558{
Willy Tarreau827aee92011-03-10 16:55:02 +01001559 struct server *srv;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001560 struct session *s = t->context;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001561 unsigned int rqf_last, rpf_last;
Willy Tarreau815a9b22010-07-27 17:15:12 +02001562 unsigned int rq_prod_last, rq_cons_last;
1563 unsigned int rp_cons_last, rp_prod_last;
Willy Tarreau576507f2010-01-07 00:09:04 +01001564 unsigned int req_ana_back;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001565
1566 //DPRINTF(stderr, "%s:%d: cs=%d ss=%d(%d) rqf=0x%08x rpf=0x%08x\n", __FUNCTION__, __LINE__,
1567 // s->si[0].state, s->si[1].state, s->si[1].err_type, s->req->flags, s->rep->flags);
1568
Krzysztof Piotr Oledzkif9423ae2010-01-29 19:26:18 +01001569 /* this data may be no longer valid, clear it */
1570 memset(&s->txn.auth, 0, sizeof(s->txn.auth));
1571
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001572 /* This flag must explicitly be set every time */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001573 s->req->flags &= ~CF_READ_NOEXP;
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001574
1575 /* Keep a copy of req/rep flags so that we can detect shutdowns */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001576 rqf_last = s->req->flags & ~CF_MASK_ANALYSER;
1577 rpf_last = s->rep->flags & ~CF_MASK_ANALYSER;
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001578
Willy Tarreau89f7ef22009-09-05 20:57:35 +02001579 /* we don't want the stream interface functions to recursively wake us up */
1580 if (s->req->prod->owner == t)
1581 s->req->prod->flags |= SI_FL_DONT_WAKE;
1582 if (s->req->cons->owner == t)
1583 s->req->cons->flags |= SI_FL_DONT_WAKE;
1584
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001585 /* 1a: Check for low level timeouts if needed. We just set a flag on
1586 * stream interfaces when their timeouts have expired.
1587 */
1588 if (unlikely(t->state & TASK_WOKEN_TIMER)) {
1589 stream_int_check_timeouts(&s->si[0]);
1590 stream_int_check_timeouts(&s->si[1]);
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001591
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001592 /* check channel timeouts, and close the corresponding stream interfaces
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001593 * for future reads or writes. Note: this will also concern upper layers
1594 * but we do not touch any other flag. We must be careful and correctly
1595 * detect state changes when calling them.
1596 */
1597
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001598 channel_check_timeouts(s->req);
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001599
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001600 if (unlikely((s->req->flags & (CF_SHUTW|CF_WRITE_TIMEOUT)) == CF_WRITE_TIMEOUT)) {
Willy Tarreau14641402009-12-29 14:49:56 +01001601 s->req->cons->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02001602 si_shutw(s->req->cons);
Willy Tarreau14641402009-12-29 14:49:56 +01001603 }
1604
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001605 if (unlikely((s->req->flags & (CF_SHUTR|CF_READ_TIMEOUT)) == CF_READ_TIMEOUT)) {
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02001606 if (s->req->prod->flags & SI_FL_NOHALF)
1607 s->req->prod->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02001608 si_shutr(s->req->prod);
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02001609 }
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001610
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001611 channel_check_timeouts(s->rep);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001612
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001613 if (unlikely((s->rep->flags & (CF_SHUTW|CF_WRITE_TIMEOUT)) == CF_WRITE_TIMEOUT)) {
Willy Tarreau14641402009-12-29 14:49:56 +01001614 s->rep->cons->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02001615 si_shutw(s->rep->cons);
Willy Tarreau14641402009-12-29 14:49:56 +01001616 }
1617
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001618 if (unlikely((s->rep->flags & (CF_SHUTR|CF_READ_TIMEOUT)) == CF_READ_TIMEOUT)) {
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02001619 if (s->rep->prod->flags & SI_FL_NOHALF)
1620 s->rep->prod->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02001621 si_shutr(s->rep->prod);
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02001622 }
Willy Tarreau798f4322012-11-08 14:49:17 +01001623
1624 /* Once in a while we're woken up because the task expires. But
1625 * this does not necessarily mean that a timeout has been reached.
1626 * So let's not run a whole session processing if only an expiration
1627 * timeout needs to be refreshed.
1628 */
1629 if (!((s->req->flags | s->rep->flags) &
1630 (CF_SHUTR|CF_READ_ACTIVITY|CF_READ_TIMEOUT|CF_SHUTW|
1631 CF_WRITE_ACTIVITY|CF_WRITE_TIMEOUT|CF_ANA_TIMEOUT)) &&
1632 !((s->si[0].flags | s->si[1].flags) & (SI_FL_EXP|SI_FL_ERR)) &&
1633 ((t->state & TASK_WOKEN_ANY) == TASK_WOKEN_TIMER))
1634 goto update_exp_and_leave;
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001635 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001636
1637 /* 1b: check for low-level errors reported at the stream interface.
1638 * First we check if it's a retryable error (in which case we don't
1639 * want to tell the buffer). Otherwise we report the error one level
1640 * upper by setting flags into the buffers. Note that the side towards
1641 * the client cannot have connect (hence retryable) errors. Also, the
1642 * connection setup code must be able to deal with any type of abort.
1643 */
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001644 srv = objt_server(s->target);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001645 if (unlikely(s->si[0].flags & SI_FL_ERR)) {
1646 if (s->si[0].state == SI_ST_EST || s->si[0].state == SI_ST_DIS) {
Willy Tarreau73b013b2012-05-21 16:31:45 +02001647 si_shutr(&s->si[0]);
1648 si_shutw(&s->si[0]);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001649 stream_int_report_error(&s->si[0]);
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001650 if (!(s->req->analysers) && !(s->rep->analysers)) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001651 s->be->be_counters.cli_aborts++;
1652 s->fe->fe_counters.cli_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01001653 if (srv)
1654 srv->counters.cli_aborts++;
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001655 if (!(s->flags & SN_ERR_MASK))
1656 s->flags |= SN_ERR_CLICL;
1657 if (!(s->flags & SN_FINST_MASK))
1658 s->flags |= SN_FINST_D;
1659 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001660 }
1661 }
1662
1663 if (unlikely(s->si[1].flags & SI_FL_ERR)) {
1664 if (s->si[1].state == SI_ST_EST || s->si[1].state == SI_ST_DIS) {
Willy Tarreau73b013b2012-05-21 16:31:45 +02001665 si_shutr(&s->si[1]);
1666 si_shutw(&s->si[1]);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001667 stream_int_report_error(&s->si[1]);
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001668 s->be->be_counters.failed_resp++;
Willy Tarreau827aee92011-03-10 16:55:02 +01001669 if (srv)
1670 srv->counters.failed_resp++;
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001671 if (!(s->req->analysers) && !(s->rep->analysers)) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01001672 s->be->be_counters.srv_aborts++;
1673 s->fe->fe_counters.srv_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01001674 if (srv)
1675 srv->counters.srv_aborts++;
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001676 if (!(s->flags & SN_ERR_MASK))
1677 s->flags |= SN_ERR_SRVCL;
1678 if (!(s->flags & SN_FINST_MASK))
1679 s->flags |= SN_FINST_D;
1680 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001681 }
1682 /* note: maybe we should process connection errors here ? */
1683 }
1684
1685 if (s->si[1].state == SI_ST_CON) {
1686 /* we were trying to establish a connection on the server side,
1687 * maybe it succeeded, maybe it failed, maybe we timed out, ...
1688 */
1689 if (unlikely(!sess_update_st_con_tcp(s, &s->si[1])))
1690 sess_update_st_cer(s, &s->si[1]);
1691 else if (s->si[1].state == SI_ST_EST)
1692 sess_establish(s, &s->si[1]);
1693
1694 /* state is now one of SI_ST_CON (still in progress), SI_ST_EST
1695 * (established), SI_ST_DIS (abort), SI_ST_CLO (last error),
1696 * SI_ST_ASS/SI_ST_TAR/SI_ST_REQ for retryable errors.
1697 */
1698 }
1699
Willy Tarreau815a9b22010-07-27 17:15:12 +02001700 rq_prod_last = s->si[0].state;
1701 rq_cons_last = s->si[1].state;
1702 rp_cons_last = s->si[0].state;
1703 rp_prod_last = s->si[1].state;
1704
1705 resync_stream_interface:
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001706 /* Check for connection closure */
1707
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001708 DPRINTF(stderr,
Willy Tarreau02d6cfc2012-03-01 18:19:58 +01001709 "[%u] %s:%d: task=%p s=%p, sfl=0x%08x, rq=%p, rp=%p, exp(r,w)=%u,%u rqf=%08x rpf=%08x rqh=%d rqt=%d rph=%d rpt=%d cs=%d ss=%d, cet=0x%x set=0x%x retr=%d\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001710 now_ms, __FUNCTION__, __LINE__,
1711 t,
1712 s, s->flags,
1713 s->req, s->rep,
1714 s->req->rex, s->rep->wex,
1715 s->req->flags, s->rep->flags,
Willy Tarreau9b28e032012-10-12 23:49:43 +02001716 s->req->buf->i, s->req->buf->o, s->rep->buf->i, s->rep->buf->o, s->rep->cons->state, s->req->cons->state,
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001717 s->rep->cons->err_type, s->req->cons->err_type,
Willy Tarreauee28de02010-06-01 09:51:00 +02001718 s->req->cons->conn_retries);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001719
1720 /* nothing special to be done on client side */
1721 if (unlikely(s->req->prod->state == SI_ST_DIS))
1722 s->req->prod->state = SI_ST_CLO;
1723
1724 /* When a server-side connection is released, we have to count it and
1725 * check for pending connections on this server.
1726 */
1727 if (unlikely(s->req->cons->state == SI_ST_DIS)) {
1728 s->req->cons->state = SI_ST_CLO;
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001729 srv = objt_server(s->target);
Willy Tarreau827aee92011-03-10 16:55:02 +01001730 if (srv) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001731 if (s->flags & SN_CURR_SESS) {
1732 s->flags &= ~SN_CURR_SESS;
Willy Tarreau827aee92011-03-10 16:55:02 +01001733 srv->cur_sess--;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001734 }
1735 sess_change_server(s, NULL);
Willy Tarreau827aee92011-03-10 16:55:02 +01001736 if (may_dequeue_tasks(srv, s->be))
1737 process_srv_queue(srv);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001738 }
1739 }
1740
1741 /*
1742 * Note: of the transient states (REQ, CER, DIS), only REQ may remain
1743 * at this point.
1744 */
1745
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001746 resync_request:
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001747 /* Analyse request */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001748 if (((s->req->flags & ~rqf_last) & CF_MASK_ANALYSER) ||
1749 ((s->req->flags ^ rqf_last) & CF_MASK_STATIC) ||
Willy Tarreau815a9b22010-07-27 17:15:12 +02001750 s->si[0].state != rq_prod_last ||
1751 s->si[1].state != rq_cons_last) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001752 unsigned int flags = s->req->flags;
1753
1754 if (s->req->prod->state >= SI_ST_EST) {
Willy Tarreaue34070e2010-01-08 00:32:27 +01001755 int max_loops = global.tune.maxpollevents;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001756 unsigned int ana_list;
1757 unsigned int ana_back;
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001758
Willy Tarreau90deb182010-01-07 00:20:41 +01001759 /* it's up to the analysers to stop new connections,
1760 * disable reading or closing. Note: if an analyser
1761 * disables any of these bits, it is responsible for
1762 * enabling them again when it disables itself, so
1763 * that other analysers are called in similar conditions.
1764 */
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001765 channel_auto_read(s->req);
1766 channel_auto_connect(s->req);
1767 channel_auto_close(s->req);
Willy Tarreauedcf6682008-11-30 23:15:34 +01001768
1769 /* We will call all analysers for which a bit is set in
1770 * s->req->analysers, following the bit order from LSB
1771 * to MSB. The analysers must remove themselves from
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001772 * the list when not needed. Any analyser may return 0
1773 * to break out of the loop, either because of missing
1774 * data to take a decision, or because it decides to
1775 * kill the session. We loop at least once through each
1776 * analyser, and we may loop again if other analysers
1777 * are added in the middle.
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001778 *
1779 * We build a list of analysers to run. We evaluate all
1780 * of these analysers in the order of the lower bit to
1781 * the higher bit. This ordering is very important.
1782 * An analyser will often add/remove other analysers,
1783 * including itself. Any changes to itself have no effect
1784 * on the loop. If it removes any other analysers, we
1785 * want those analysers not to be called anymore during
1786 * this loop. If it adds an analyser that is located
1787 * after itself, we want it to be scheduled for being
1788 * processed during the loop. If it adds an analyser
1789 * which is located before it, we want it to switch to
1790 * it immediately, even if it has already been called
1791 * once but removed since.
1792 *
1793 * In order to achieve this, we compare the analyser
1794 * list after the call with a copy of it before the
1795 * call. The work list is fed with analyser bits that
1796 * appeared during the call. Then we compare previous
1797 * work list with the new one, and check the bits that
1798 * appeared. If the lowest of these bits is lower than
1799 * the current bit, it means we have enabled a previous
1800 * analyser and must immediately loop again.
Willy Tarreauedcf6682008-11-30 23:15:34 +01001801 */
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001802
1803 ana_list = ana_back = s->req->analysers;
Willy Tarreaue34070e2010-01-08 00:32:27 +01001804 while (ana_list && max_loops--) {
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001805 /* Warning! ensure that analysers are always placed in ascending order! */
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001806
Willy Tarreaufb356202010-08-03 14:02:05 +02001807 if (ana_list & AN_REQ_INSPECT_FE) {
1808 if (!tcp_inspect_request(s, s->req, AN_REQ_INSPECT_FE))
Willy Tarreauedcf6682008-11-30 23:15:34 +01001809 break;
Willy Tarreaufb356202010-08-03 14:02:05 +02001810 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_INSPECT_FE);
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001811 }
Willy Tarreauedcf6682008-11-30 23:15:34 +01001812
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001813 if (ana_list & AN_REQ_WAIT_HTTP) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001814 if (!http_wait_for_request(s, s->req, AN_REQ_WAIT_HTTP))
Willy Tarreaud787e662009-07-07 10:14:51 +02001815 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001816 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_WAIT_HTTP);
Willy Tarreaud787e662009-07-07 10:14:51 +02001817 }
1818
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001819 if (ana_list & AN_REQ_HTTP_PROCESS_FE) {
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001820 if (!http_process_req_common(s, s->req, AN_REQ_HTTP_PROCESS_FE, s->fe))
1821 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001822 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_PROCESS_FE);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001823 }
1824
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001825 if (ana_list & AN_REQ_SWITCHING_RULES) {
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001826 if (!process_switching_rules(s, s->req, AN_REQ_SWITCHING_RULES))
1827 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001828 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_SWITCHING_RULES);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001829 }
1830
Willy Tarreaufb356202010-08-03 14:02:05 +02001831 if (ana_list & AN_REQ_INSPECT_BE) {
1832 if (!tcp_inspect_request(s, s->req, AN_REQ_INSPECT_BE))
1833 break;
1834 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_INSPECT_BE);
1835 }
1836
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001837 if (ana_list & AN_REQ_HTTP_PROCESS_BE) {
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001838 if (!http_process_req_common(s, s->req, AN_REQ_HTTP_PROCESS_BE, s->be))
1839 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001840 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_PROCESS_BE);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001841 }
1842
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001843 if (ana_list & AN_REQ_HTTP_TARPIT) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001844 if (!http_process_tarpit(s, s->req, AN_REQ_HTTP_TARPIT))
Willy Tarreau60b85b02008-11-30 23:28:40 +01001845 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001846 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_TARPIT);
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001847 }
Willy Tarreau60b85b02008-11-30 23:28:40 +01001848
Willy Tarreau4a5cade2012-04-05 21:09:48 +02001849 if (ana_list & AN_REQ_SRV_RULES) {
1850 if (!process_server_rules(s, s->req, AN_REQ_SRV_RULES))
1851 break;
1852 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_SRV_RULES);
1853 }
1854
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001855 if (ana_list & AN_REQ_HTTP_INNER) {
Willy Tarreauc465fd72009-08-31 00:17:18 +02001856 if (!http_process_request(s, s->req, AN_REQ_HTTP_INNER))
1857 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001858 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_INNER);
Willy Tarreauc465fd72009-08-31 00:17:18 +02001859 }
1860
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001861 if (ana_list & AN_REQ_HTTP_BODY) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001862 if (!http_process_request_body(s, s->req, AN_REQ_HTTP_BODY))
Willy Tarreaud34af782008-11-30 23:36:37 +01001863 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001864 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_BODY);
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001865 }
Emeric Brun647caf12009-06-30 17:57:00 +02001866
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001867 if (ana_list & AN_REQ_PRST_RDP_COOKIE) {
Emeric Brun647caf12009-06-30 17:57:00 +02001868 if (!tcp_persist_rdp_cookie(s, s->req, AN_REQ_PRST_RDP_COOKIE))
1869 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001870 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_PRST_RDP_COOKIE);
Emeric Brun647caf12009-06-30 17:57:00 +02001871 }
Willy Tarreaud98cf932009-12-27 22:54:55 +01001872
Emeric Brun1d33b292010-01-04 15:47:17 +01001873 if (ana_list & AN_REQ_STICKING_RULES) {
1874 if (!process_sticking_rules(s, s->req, AN_REQ_STICKING_RULES))
1875 break;
1876 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_STICKING_RULES);
1877 }
1878
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001879 if (ana_list & AN_REQ_HTTP_XFER_BODY) {
Willy Tarreaud98cf932009-12-27 22:54:55 +01001880 if (!http_request_forward_body(s, s->req, AN_REQ_HTTP_XFER_BODY))
1881 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001882 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_XFER_BODY);
Willy Tarreaud98cf932009-12-27 22:54:55 +01001883 }
Willy Tarreaue34070e2010-01-08 00:32:27 +01001884 break;
1885 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001886 }
Willy Tarreau84455332009-03-15 22:34:05 +01001887
Willy Tarreau815a9b22010-07-27 17:15:12 +02001888 rq_prod_last = s->si[0].state;
1889 rq_cons_last = s->si[1].state;
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001890 s->req->flags &= ~CF_WAKE_ONCE;
Willy Tarreau815a9b22010-07-27 17:15:12 +02001891 rqf_last = s->req->flags;
1892
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001893 if ((s->req->flags ^ flags) & CF_MASK_STATIC)
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001894 goto resync_request;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001895 }
1896
Willy Tarreau576507f2010-01-07 00:09:04 +01001897 /* we'll monitor the request analysers while parsing the response,
1898 * because some response analysers may indirectly enable new request
1899 * analysers (eg: HTTP keep-alive).
1900 */
1901 req_ana_back = s->req->analysers;
1902
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001903 resync_response:
1904 /* Analyse response */
1905
Willy Tarreaub31c9712012-11-11 23:05:39 +01001906 if (((s->rep->flags & ~rpf_last) & CF_MASK_ANALYSER) ||
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001907 (s->rep->flags ^ rpf_last) & CF_MASK_STATIC ||
Willy Tarreau815a9b22010-07-27 17:15:12 +02001908 s->si[0].state != rp_cons_last ||
1909 s->si[1].state != rp_prod_last) {
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001910 unsigned int flags = s->rep->flags;
1911
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001912 if ((s->rep->flags & CF_MASK_ANALYSER) &&
Willy Tarreau0499e352010-12-17 07:13:42 +01001913 (s->rep->analysers & AN_REQ_WAIT_HTTP)) {
1914 /* Due to HTTP pipelining, the HTTP request analyser might be waiting
1915 * for some free space in the response buffer, so we might need to call
1916 * it when something changes in the response buffer, but still we pass
1917 * it the request buffer. Note that the SI state might very well still
1918 * be zero due to us returning a flow of redirects!
1919 */
1920 s->rep->analysers &= ~AN_REQ_WAIT_HTTP;
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001921 s->req->flags |= CF_WAKE_ONCE;
Willy Tarreau0499e352010-12-17 07:13:42 +01001922 }
1923
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001924 if (s->rep->prod->state >= SI_ST_EST) {
Willy Tarreaue34070e2010-01-08 00:32:27 +01001925 int max_loops = global.tune.maxpollevents;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001926 unsigned int ana_list;
1927 unsigned int ana_back;
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001928
Willy Tarreau90deb182010-01-07 00:20:41 +01001929 /* it's up to the analysers to stop disable reading or
1930 * closing. Note: if an analyser disables any of these
1931 * bits, it is responsible for enabling them again when
1932 * it disables itself, so that other analysers are called
1933 * in similar conditions.
1934 */
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001935 channel_auto_read(s->rep);
1936 channel_auto_close(s->rep);
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001937
1938 /* We will call all analysers for which a bit is set in
1939 * s->rep->analysers, following the bit order from LSB
1940 * to MSB. The analysers must remove themselves from
1941 * the list when not needed. Any analyser may return 0
1942 * to break out of the loop, either because of missing
1943 * data to take a decision, or because it decides to
1944 * kill the session. We loop at least once through each
1945 * analyser, and we may loop again if other analysers
1946 * are added in the middle.
1947 */
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001948
1949 ana_list = ana_back = s->rep->analysers;
Willy Tarreaue34070e2010-01-08 00:32:27 +01001950 while (ana_list && max_loops--) {
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001951 /* Warning! ensure that analysers are always placed in ascending order! */
1952
Emeric Brun97679e72010-09-23 17:56:44 +02001953 if (ana_list & AN_RES_INSPECT) {
1954 if (!tcp_inspect_response(s, s->rep, AN_RES_INSPECT))
1955 break;
1956 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_INSPECT);
1957 }
1958
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001959 if (ana_list & AN_RES_WAIT_HTTP) {
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001960 if (!http_wait_for_response(s, s->rep, AN_RES_WAIT_HTTP))
1961 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001962 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_WAIT_HTTP);
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001963 }
1964
Emeric Brun1d33b292010-01-04 15:47:17 +01001965 if (ana_list & AN_RES_STORE_RULES) {
1966 if (!process_store_rules(s, s->rep, AN_RES_STORE_RULES))
1967 break;
1968 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_STORE_RULES);
1969 }
1970
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001971 if (ana_list & AN_RES_HTTP_PROCESS_BE) {
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001972 if (!http_process_res_common(s, s->rep, AN_RES_HTTP_PROCESS_BE, s->be))
1973 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001974 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_HTTP_PROCESS_BE);
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001975 }
Willy Tarreaud98cf932009-12-27 22:54:55 +01001976
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001977 if (ana_list & AN_RES_HTTP_XFER_BODY) {
Willy Tarreaud98cf932009-12-27 22:54:55 +01001978 if (!http_response_forward_body(s, s->rep, AN_RES_HTTP_XFER_BODY))
1979 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001980 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_HTTP_XFER_BODY);
Willy Tarreaud98cf932009-12-27 22:54:55 +01001981 }
Willy Tarreaue34070e2010-01-08 00:32:27 +01001982 break;
1983 }
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001984 }
1985
Willy Tarreau815a9b22010-07-27 17:15:12 +02001986 rp_cons_last = s->si[0].state;
1987 rp_prod_last = s->si[1].state;
1988 rpf_last = s->rep->flags;
1989
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001990 if ((s->rep->flags ^ flags) & CF_MASK_STATIC)
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001991 goto resync_response;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001992 }
1993
Willy Tarreau576507f2010-01-07 00:09:04 +01001994 /* maybe someone has added some request analysers, so we must check and loop */
1995 if (s->req->analysers & ~req_ana_back)
1996 goto resync_request;
1997
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001998 if ((s->req->flags & ~rqf_last) & CF_MASK_ANALYSER)
Willy Tarreau0499e352010-12-17 07:13:42 +01001999 goto resync_request;
2000
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002001 /* FIXME: here we should call protocol handlers which rely on
2002 * both buffers.
2003 */
2004
2005
2006 /*
Willy Tarreauae526782010-03-04 20:34:23 +01002007 * Now we propagate unhandled errors to the session. Normally
2008 * we're just in a data phase here since it means we have not
2009 * seen any analyser who could set an error status.
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002010 */
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002011 srv = objt_server(s->target);
Willy Tarreau2f976e12010-11-11 14:28:47 +01002012 if (unlikely(!(s->flags & SN_ERR_MASK))) {
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002013 if (s->req->flags & (CF_READ_ERROR|CF_READ_TIMEOUT|CF_WRITE_ERROR|CF_WRITE_TIMEOUT)) {
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002014 /* Report it if the client got an error or a read timeout expired */
Willy Tarreau84455332009-03-15 22:34:05 +01002015 s->req->analysers = 0;
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002016 if (s->req->flags & CF_READ_ERROR) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002017 s->be->be_counters.cli_aborts++;
2018 s->fe->fe_counters.cli_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002019 if (srv)
2020 srv->counters.cli_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01002021 s->flags |= SN_ERR_CLICL;
Willy Tarreauae526782010-03-04 20:34:23 +01002022 }
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002023 else if (s->req->flags & CF_READ_TIMEOUT) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002024 s->be->be_counters.cli_aborts++;
2025 s->fe->fe_counters.cli_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002026 if (srv)
2027 srv->counters.cli_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01002028 s->flags |= SN_ERR_CLITO;
Willy Tarreauae526782010-03-04 20:34:23 +01002029 }
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002030 else if (s->req->flags & CF_WRITE_ERROR) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002031 s->be->be_counters.srv_aborts++;
2032 s->fe->fe_counters.srv_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002033 if (srv)
2034 srv->counters.srv_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01002035 s->flags |= SN_ERR_SRVCL;
Willy Tarreauae526782010-03-04 20:34:23 +01002036 }
2037 else {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002038 s->be->be_counters.srv_aborts++;
2039 s->fe->fe_counters.srv_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002040 if (srv)
2041 srv->counters.srv_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01002042 s->flags |= SN_ERR_SRVTO;
Willy Tarreauae526782010-03-04 20:34:23 +01002043 }
Willy Tarreau84455332009-03-15 22:34:05 +01002044 sess_set_term_flags(s);
2045 }
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002046 else if (s->rep->flags & (CF_READ_ERROR|CF_READ_TIMEOUT|CF_WRITE_ERROR|CF_WRITE_TIMEOUT)) {
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002047 /* Report it if the server got an error or a read timeout expired */
2048 s->rep->analysers = 0;
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002049 if (s->rep->flags & CF_READ_ERROR) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002050 s->be->be_counters.srv_aborts++;
2051 s->fe->fe_counters.srv_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002052 if (srv)
2053 srv->counters.srv_aborts++;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002054 s->flags |= SN_ERR_SRVCL;
Willy Tarreauae526782010-03-04 20:34:23 +01002055 }
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002056 else if (s->rep->flags & CF_READ_TIMEOUT) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002057 s->be->be_counters.srv_aborts++;
2058 s->fe->fe_counters.srv_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002059 if (srv)
2060 srv->counters.srv_aborts++;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002061 s->flags |= SN_ERR_SRVTO;
Willy Tarreauae526782010-03-04 20:34:23 +01002062 }
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002063 else if (s->rep->flags & CF_WRITE_ERROR) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002064 s->be->be_counters.cli_aborts++;
2065 s->fe->fe_counters.cli_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002066 if (srv)
2067 srv->counters.cli_aborts++;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002068 s->flags |= SN_ERR_CLICL;
Willy Tarreauae526782010-03-04 20:34:23 +01002069 }
2070 else {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002071 s->be->be_counters.cli_aborts++;
2072 s->fe->fe_counters.cli_aborts++;
Willy Tarreau827aee92011-03-10 16:55:02 +01002073 if (srv)
2074 srv->counters.cli_aborts++;
Willy Tarreauae526782010-03-04 20:34:23 +01002075 s->flags |= SN_ERR_CLITO;
2076 }
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002077 sess_set_term_flags(s);
2078 }
Willy Tarreau84455332009-03-15 22:34:05 +01002079 }
2080
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002081 /*
2082 * Here we take care of forwarding unhandled data. This also includes
2083 * connection establishments and shutdown requests.
2084 */
2085
2086
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002087 /* If noone is interested in analysing data, it's time to forward
Willy Tarreau31971e52009-09-20 12:07:52 +02002088 * everything. We configure the buffer to forward indefinitely.
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002089 * Note that we're checking CF_SHUTR_NOW as an indication of a possible
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002090 * recent call to channel_abort().
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01002091 */
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002092 if (!s->req->analysers &&
Willy Tarreaub31c9712012-11-11 23:05:39 +01002093 !(s->req->flags & (CF_SHUTW|CF_SHUTR_NOW)) &&
Willy Tarreau31971e52009-09-20 12:07:52 +02002094 (s->req->prod->state >= SI_ST_EST) &&
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002095 (s->req->to_forward != CHN_INFINITE_FORWARD)) {
Willy Tarreaub31c9712012-11-11 23:05:39 +01002096 /* This buffer is freewheeling, there's no analyser
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002097 * attached to it. If any data are left in, we'll permit them to
2098 * move.
2099 */
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002100 channel_auto_read(s->req);
2101 channel_auto_connect(s->req);
2102 channel_auto_close(s->req);
Willy Tarreau9b28e032012-10-12 23:49:43 +02002103 buffer_flush(s->req->buf);
Willy Tarreau5bd8c372009-01-19 00:32:22 +01002104
Willy Tarreauda4d9fe2010-11-07 20:26:56 +01002105 /* We'll let data flow between the producer (if still connected)
2106 * to the consumer (which might possibly not be connected yet).
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002107 */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002108 if (!(s->req->flags & (CF_SHUTR|CF_SHUTW_NOW)))
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002109 channel_forward(s->req, CHN_INFINITE_FORWARD);
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01002110 }
Willy Tarreauf890dc92008-12-13 21:12:26 +01002111
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002112 /* check if it is wise to enable kernel splicing to forward request data */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002113 if (!(s->req->flags & (CF_KERN_SPLICING|CF_SHUTR)) &&
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002114 s->req->to_forward &&
2115 (global.tune.options & GTUNE_USE_SPLICE) &&
Willy Tarreauf2943dc2012-10-26 20:10:28 +02002116 (s->si[0].conn->xprt && s->si[0].conn->xprt->rcv_pipe && s->si[0].conn->xprt->snd_pipe) &&
2117 (s->si[1].conn->xprt && s->si[1].conn->xprt->rcv_pipe && s->si[1].conn->xprt->snd_pipe) &&
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002118 (pipes_used < global.maxpipes) &&
2119 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_REQ) ||
2120 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_AUT) &&
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002121 (s->req->flags & CF_STREAMER_FAST)))) {
2122 s->req->flags |= CF_KERN_SPLICING;
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002123 }
2124
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002125 /* reflect what the L7 analysers have seen last */
2126 rqf_last = s->req->flags;
2127
2128 /*
2129 * Now forward all shutdown requests between both sides of the buffer
2130 */
2131
Willy Tarreau520d95e2009-09-19 21:04:57 +02002132 /* first, let's check if the request buffer needs to shutdown(write), which may
2133 * happen either because the input is closed or because we want to force a close
Willy Tarreaue4599762010-03-21 23:25:09 +01002134 * once the server has begun to respond.
Willy Tarreau520d95e2009-09-19 21:04:57 +02002135 */
Willy Tarreaub31c9712012-11-11 23:05:39 +01002136 if (unlikely((s->req->flags & (CF_SHUTW|CF_SHUTW_NOW|CF_AUTO_CLOSE|CF_SHUTR)) ==
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002137 (CF_AUTO_CLOSE|CF_SHUTR)))
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002138 channel_shutw_now(s->req);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002139
2140 /* shutdown(write) pending */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002141 if (unlikely((s->req->flags & (CF_SHUTW|CF_SHUTW_NOW)) == CF_SHUTW_NOW &&
Willy Tarreau8615c2a2013-06-21 08:20:19 +02002142 channel_is_empty(s->req))) {
2143 if (s->req->flags & CF_READ_ERROR)
2144 s->req->cons->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02002145 si_shutw(s->req->cons);
Willy Tarreau8615c2a2013-06-21 08:20:19 +02002146 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002147
2148 /* shutdown(write) done on server side, we must stop the client too */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002149 if (unlikely((s->req->flags & (CF_SHUTW|CF_SHUTR|CF_SHUTR_NOW)) == CF_SHUTW &&
Willy Tarreau3dbc6942008-12-07 13:05:04 +01002150 !s->req->analysers))
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002151 channel_shutr_now(s->req);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002152
2153 /* shutdown(read) pending */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002154 if (unlikely((s->req->flags & (CF_SHUTR|CF_SHUTR_NOW)) == CF_SHUTR_NOW)) {
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02002155 if (s->req->prod->flags & SI_FL_NOHALF)
2156 s->req->prod->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02002157 si_shutr(s->req->prod);
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02002158 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002159
Willy Tarreau520d95e2009-09-19 21:04:57 +02002160 /* it's possible that an upper layer has requested a connection setup or abort.
2161 * There are 2 situations where we decide to establish a new connection :
2162 * - there are data scheduled for emission in the buffer
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002163 * - the CF_AUTO_CONNECT flag is set (active connection)
Willy Tarreau520d95e2009-09-19 21:04:57 +02002164 */
2165 if (s->req->cons->state == SI_ST_INI) {
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002166 if (!(s->req->flags & CF_SHUTW)) {
2167 if ((s->req->flags & CF_AUTO_CONNECT) || !channel_is_empty(s->req)) {
Willy Tarreaub24281b2011-02-13 13:16:36 +01002168 /* If we have an applet without a connect method, we immediately
Willy Tarreau85e7d002010-05-31 11:57:51 +02002169 * switch to the connected state, otherwise we perform a connection
2170 * request.
Willy Tarreau520d95e2009-09-19 21:04:57 +02002171 */
Willy Tarreau85e7d002010-05-31 11:57:51 +02002172 s->req->cons->state = SI_ST_REQ; /* new connection requested */
Willy Tarreau070ceb62010-06-01 10:36:43 +02002173 s->req->cons->conn_retries = s->be->conn_retries;
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002174 if (unlikely(obj_type(s->req->cons->conn->target) == OBJ_TYPE_APPLET &&
Willy Tarreau73b013b2012-05-21 16:31:45 +02002175 !(si_ctrl(s->req->cons) && si_ctrl(s->req->cons)->connect))) {
Willy Tarreau520d95e2009-09-19 21:04:57 +02002176 s->req->cons->state = SI_ST_EST; /* connection established */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002177 s->rep->flags |= CF_READ_ATTACHED; /* producer is now attached */
Willy Tarreau85e7d002010-05-31 11:57:51 +02002178 s->req->wex = TICK_ETERNITY;
2179 }
Willy Tarreau520d95e2009-09-19 21:04:57 +02002180 }
Willy Tarreau73201222009-08-16 18:27:24 +02002181 }
Willy Tarreauf41ffdc2009-09-20 08:19:25 +02002182 else {
Willy Tarreau92795622009-03-06 12:51:23 +01002183 s->req->cons->state = SI_ST_CLO; /* shutw+ini = abort */
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002184 channel_shutw_now(s->req); /* fix buffer flags upon abort */
2185 channel_shutr_now(s->rep);
Willy Tarreauf41ffdc2009-09-20 08:19:25 +02002186 }
Willy Tarreau92795622009-03-06 12:51:23 +01002187 }
2188
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002189
2190 /* we may have a pending connection request, or a connection waiting
2191 * for completion.
2192 */
2193 if (s->si[1].state >= SI_ST_REQ && s->si[1].state < SI_ST_CON) {
2194 do {
2195 /* nb: step 1 might switch from QUE to ASS, but we first want
2196 * to give a chance to step 2 to perform a redirect if needed.
2197 */
2198 if (s->si[1].state != SI_ST_REQ)
2199 sess_update_stream_int(s, &s->si[1]);
Willy Tarreau1e5dfda2013-04-07 18:19:16 +02002200 if (s->si[1].state == SI_ST_REQ) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002201 sess_prepare_conn_req(s, &s->si[1]);
2202
Willy Tarreau1e5dfda2013-04-07 18:19:16 +02002203 /* Now we can add the server name to a header (if requested) */
2204 /* check for HTTP mode and proxy server_name_hdr_name != NULL */
2205 if ((s->flags & SN_BE_ASSIGNED) &&
2206 (s->be->mode == PR_MODE_HTTP) &&
Willy Tarreau0fc36e32013-07-04 11:44:27 +02002207 (s->be->server_id_hdr_name != NULL && s->target)) {
Willy Tarreau1e5dfda2013-04-07 18:19:16 +02002208 http_send_name_header(&s->txn, s->be, objt_server(s->target)->id);
2209 }
2210 }
2211
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002212 srv = objt_server(s->target);
Willy Tarreau827aee92011-03-10 16:55:02 +01002213 if (s->si[1].state == SI_ST_ASS && srv && srv->rdr_len && (s->flags & SN_REDIRECTABLE))
Willy Tarreau71241ab2012-12-27 11:30:54 +01002214 http_perform_server_redirect(s, &s->si[1]);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002215 } while (s->si[1].state == SI_ST_ASS);
2216 }
2217
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002218 /* Benchmarks have shown that it's optimal to do a full resync now */
Willy Tarreau0be0ef92009-03-08 19:20:25 +01002219 if (s->req->prod->state == SI_ST_DIS || s->req->cons->state == SI_ST_DIS)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002220 goto resync_stream_interface;
2221
Willy Tarreau815a9b22010-07-27 17:15:12 +02002222 /* otherwise we want to check if we need to resync the req buffer or not */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002223 if ((s->req->flags ^ rqf_last) & CF_MASK_STATIC)
Willy Tarreau0be0ef92009-03-08 19:20:25 +01002224 goto resync_request;
2225
Willy Tarreau3deb3d02009-06-21 22:43:05 +02002226 /* perform output updates to the response buffer */
Willy Tarreau84455332009-03-15 22:34:05 +01002227
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002228 /* If noone is interested in analysing data, it's time to forward
Willy Tarreau31971e52009-09-20 12:07:52 +02002229 * everything. We configure the buffer to forward indefinitely.
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002230 * Note that we're checking CF_SHUTR_NOW as an indication of a possible
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002231 * recent call to channel_abort().
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01002232 */
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002233 if (!s->rep->analysers &&
Willy Tarreaub31c9712012-11-11 23:05:39 +01002234 !(s->rep->flags & (CF_SHUTW|CF_SHUTR_NOW)) &&
Willy Tarreau31971e52009-09-20 12:07:52 +02002235 (s->rep->prod->state >= SI_ST_EST) &&
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002236 (s->rep->to_forward != CHN_INFINITE_FORWARD)) {
Willy Tarreaub31c9712012-11-11 23:05:39 +01002237 /* This buffer is freewheeling, there's no analyser
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002238 * attached to it. If any data are left in, we'll permit them to
2239 * move.
2240 */
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002241 channel_auto_read(s->rep);
2242 channel_auto_close(s->rep);
Willy Tarreau9b28e032012-10-12 23:49:43 +02002243 buffer_flush(s->rep->buf);
Willy Tarreauda4d9fe2010-11-07 20:26:56 +01002244
2245 /* We'll let data flow between the producer (if still connected)
2246 * to the consumer.
2247 */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002248 if (!(s->rep->flags & (CF_SHUTR|CF_SHUTW_NOW)))
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002249 channel_forward(s->rep, CHN_INFINITE_FORWARD);
Willy Tarreauce887fd2012-05-12 12:50:00 +02002250
2251 /* if we have no analyser anymore in any direction and have a
2252 * tunnel timeout set, use it now.
2253 */
2254 if (!s->req->analysers && s->be->timeout.tunnel) {
2255 s->req->rto = s->req->wto = s->rep->rto = s->rep->wto =
2256 s->be->timeout.tunnel;
2257 s->req->rex = s->req->wex = s->rep->rex = s->rep->wex =
2258 tick_add(now_ms, s->be->timeout.tunnel);
2259 }
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01002260 }
Willy Tarreauf890dc92008-12-13 21:12:26 +01002261
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002262 /* check if it is wise to enable kernel splicing to forward response data */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002263 if (!(s->rep->flags & (CF_KERN_SPLICING|CF_SHUTR)) &&
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002264 s->rep->to_forward &&
2265 (global.tune.options & GTUNE_USE_SPLICE) &&
Willy Tarreauf2943dc2012-10-26 20:10:28 +02002266 (s->si[0].conn->xprt && s->si[0].conn->xprt->rcv_pipe && s->si[0].conn->xprt->snd_pipe) &&
2267 (s->si[1].conn->xprt && s->si[1].conn->xprt->rcv_pipe && s->si[1].conn->xprt->snd_pipe) &&
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002268 (pipes_used < global.maxpipes) &&
2269 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_RTR) ||
2270 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_AUT) &&
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002271 (s->rep->flags & CF_STREAMER_FAST)))) {
2272 s->rep->flags |= CF_KERN_SPLICING;
Willy Tarreau7c84bab2009-03-08 21:38:23 +01002273 }
2274
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002275 /* reflect what the L7 analysers have seen last */
2276 rpf_last = s->rep->flags;
2277
2278 /*
2279 * Now forward all shutdown requests between both sides of the buffer
2280 */
2281
2282 /*
2283 * FIXME: this is probably where we should produce error responses.
2284 */
2285
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01002286 /* first, let's check if the response buffer needs to shutdown(write) */
Willy Tarreaub31c9712012-11-11 23:05:39 +01002287 if (unlikely((s->rep->flags & (CF_SHUTW|CF_SHUTW_NOW|CF_AUTO_CLOSE|CF_SHUTR)) ==
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002288 (CF_AUTO_CLOSE|CF_SHUTR)))
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002289 channel_shutw_now(s->rep);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002290
2291 /* shutdown(write) pending */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002292 if (unlikely((s->rep->flags & (CF_SHUTW|CF_SHUTW_NOW)) == CF_SHUTW_NOW &&
Willy Tarreau8e21bb92012-08-24 22:40:29 +02002293 channel_is_empty(s->rep)))
Willy Tarreau73b013b2012-05-21 16:31:45 +02002294 si_shutw(s->rep->cons);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002295
2296 /* shutdown(write) done on the client side, we must stop the server too */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002297 if (unlikely((s->rep->flags & (CF_SHUTW|CF_SHUTR|CF_SHUTR_NOW)) == CF_SHUTW) &&
Willy Tarreau3dbc6942008-12-07 13:05:04 +01002298 !s->rep->analysers)
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002299 channel_shutr_now(s->rep);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002300
2301 /* shutdown(read) pending */
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002302 if (unlikely((s->rep->flags & (CF_SHUTR|CF_SHUTR_NOW)) == CF_SHUTR_NOW)) {
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02002303 if (s->rep->prod->flags & SI_FL_NOHALF)
2304 s->rep->prod->flags |= SI_FL_NOLINGER;
Willy Tarreau73b013b2012-05-21 16:31:45 +02002305 si_shutr(s->rep->prod);
Willy Tarreau7bb68ab2012-05-13 14:48:59 +02002306 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002307
Willy Tarreau0be0ef92009-03-08 19:20:25 +01002308 if (s->req->prod->state == SI_ST_DIS || s->req->cons->state == SI_ST_DIS)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002309 goto resync_stream_interface;
2310
Willy Tarreau0be0ef92009-03-08 19:20:25 +01002311 if (s->req->flags != rqf_last)
2312 goto resync_request;
2313
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002314 if ((s->rep->flags ^ rpf_last) & CF_MASK_STATIC)
Willy Tarreau0be0ef92009-03-08 19:20:25 +01002315 goto resync_response;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002316
Willy Tarreau89f7ef22009-09-05 20:57:35 +02002317 /* we're interested in getting wakeups again */
2318 s->req->prod->flags &= ~SI_FL_DONT_WAKE;
2319 s->req->cons->flags &= ~SI_FL_DONT_WAKE;
2320
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002321 /* This is needed only when debugging is enabled, to indicate
2322 * client-side or server-side close. Please note that in the unlikely
2323 * event where both sides would close at once, the sequence is reported
2324 * on the server side first.
2325 */
2326 if (unlikely((global.mode & MODE_DEBUG) &&
2327 (!(global.mode & MODE_QUIET) ||
2328 (global.mode & MODE_VERBOSE)))) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002329 if (s->si[1].state == SI_ST_CLO &&
2330 s->si[1].prev_state == SI_ST_EST) {
Willy Tarreau19d14ef2012-10-29 16:51:55 +01002331 chunk_printf(&trash, "%08x:%s.srvcls[%04x:%04x]\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002332 s->uniq_id, s->be->id,
Willy Tarreau7f7ad912012-11-11 19:27:15 +01002333 (unsigned short)s->si[0].conn->t.sock.fd,
2334 (unsigned short)s->si[1].conn->t.sock.fd);
Willy Tarreau19d14ef2012-10-29 16:51:55 +01002335 if (write(1, trash.str, trash.len) < 0) /* shut gcc warning */;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002336 }
2337
2338 if (s->si[0].state == SI_ST_CLO &&
2339 s->si[0].prev_state == SI_ST_EST) {
Willy Tarreau19d14ef2012-10-29 16:51:55 +01002340 chunk_printf(&trash, "%08x:%s.clicls[%04x:%04x]\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002341 s->uniq_id, s->be->id,
Willy Tarreau7f7ad912012-11-11 19:27:15 +01002342 (unsigned short)s->si[0].conn->t.sock.fd,
2343 (unsigned short)s->si[1].conn->t.sock.fd);
Willy Tarreau19d14ef2012-10-29 16:51:55 +01002344 if (write(1, trash.str, trash.len) < 0) /* shut gcc warning */;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002345 }
2346 }
2347
2348 if (likely((s->rep->cons->state != SI_ST_CLO) ||
2349 (s->req->cons->state > SI_ST_INI && s->req->cons->state < SI_ST_CLO))) {
2350
2351 if ((s->fe->options & PR_O_CONTSTATS) && (s->flags & SN_BE_ASSIGNED))
2352 session_process_counters(s);
2353
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002354 if (s->rep->cons->state == SI_ST_EST && obj_type(s->rep->cons->conn->target) != OBJ_TYPE_APPLET)
Willy Tarreau73b013b2012-05-21 16:31:45 +02002355 si_update(s->rep->cons);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002356
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002357 if (s->req->cons->state == SI_ST_EST && obj_type(s->req->cons->conn->target) != OBJ_TYPE_APPLET)
Willy Tarreau73b013b2012-05-21 16:31:45 +02002358 si_update(s->req->cons);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002359
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002360 s->req->flags &= ~(CF_READ_NULL|CF_READ_PARTIAL|CF_WRITE_NULL|CF_WRITE_PARTIAL|CF_READ_ATTACHED);
2361 s->rep->flags &= ~(CF_READ_NULL|CF_READ_PARTIAL|CF_WRITE_NULL|CF_WRITE_PARTIAL|CF_READ_ATTACHED);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002362 s->si[0].prev_state = s->si[0].state;
2363 s->si[1].prev_state = s->si[1].state;
Willy Tarreaub0ef7352008-12-14 13:26:20 +01002364 s->si[0].flags &= ~(SI_FL_ERR|SI_FL_EXP);
2365 s->si[1].flags &= ~(SI_FL_ERR|SI_FL_EXP);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002366
2367 /* Trick: if a request is being waiting for the server to respond,
2368 * and if we know the server can timeout, we don't want the timeout
2369 * to expire on the client side first, but we're still interested
2370 * in passing data from the client to the server (eg: POST). Thus,
2371 * we can cancel the client's request timeout if the server's
2372 * request timeout is set and the server has not yet sent a response.
2373 */
2374
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002375 if ((s->rep->flags & (CF_AUTO_CLOSE|CF_SHUTR)) == 0 &&
Willy Tarreau86491c32008-12-14 09:04:47 +01002376 (tick_isset(s->req->wex) || tick_isset(s->rep->rex))) {
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002377 s->req->flags |= CF_READ_NOEXP;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002378 s->req->rex = TICK_ETERNITY;
Willy Tarreau86491c32008-12-14 09:04:47 +01002379 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002380
Willy Tarreau7a20aa62010-07-13 16:30:45 +02002381 /* Call the stream interfaces' I/O handlers when embedded.
Willy Tarreau1accfc02009-09-05 20:57:35 +02002382 * Note that this one may wake the task up again.
2383 */
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002384 if (obj_type(s->req->cons->conn->target) == OBJ_TYPE_APPLET ||
2385 obj_type(s->rep->cons->conn->target) == OBJ_TYPE_APPLET) {
2386 if (objt_applet(s->req->cons->conn->target))
2387 objt_applet(s->req->cons->conn->target)->fct(s->req->cons);
2388 if (objt_applet(s->rep->cons->conn->target))
2389 objt_applet(s->rep->cons->conn->target)->fct(s->rep->cons);
Willy Tarreau1accfc02009-09-05 20:57:35 +02002390 if (task_in_rq(t)) {
2391 /* If we woke up, we don't want to requeue the
2392 * task to the wait queue, but rather requeue
2393 * it into the runqueue ASAP.
2394 */
2395 t->expire = TICK_ETERNITY;
2396 return t;
2397 }
2398 }
2399
Willy Tarreau798f4322012-11-08 14:49:17 +01002400 update_exp_and_leave:
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002401 t->expire = tick_first(tick_first(s->req->rex, s->req->wex),
2402 tick_first(s->rep->rex, s->rep->wex));
2403 if (s->req->analysers)
2404 t->expire = tick_first(t->expire, s->req->analyse_exp);
2405
2406 if (s->si[0].exp)
2407 t->expire = tick_first(t->expire, s->si[0].exp);
2408
2409 if (s->si[1].exp)
2410 t->expire = tick_first(t->expire, s->si[1].exp);
2411
2412#ifdef DEBUG_FULL
Willy Tarreau127334e2009-03-28 10:47:26 +01002413 fprintf(stderr,
2414 "[%u] queuing with exp=%u req->rex=%u req->wex=%u req->ana_exp=%u"
2415 " rep->rex=%u rep->wex=%u, si[0].exp=%u, si[1].exp=%u, cs=%d, ss=%d\n",
2416 now_ms, t->expire, s->req->rex, s->req->wex, s->req->analyse_exp,
2417 s->rep->rex, s->rep->wex, s->si[0].exp, s->si[1].exp, s->si[0].state, s->si[1].state);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002418#endif
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002419
2420#ifdef DEBUG_DEV
2421 /* this may only happen when no timeout is set or in case of an FSM bug */
Willy Tarreaud0a201b2009-03-08 15:53:06 +01002422 if (!tick_isset(t->expire))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002423 ABORT_NOW();
2424#endif
Willy Tarreau26c25062009-03-08 09:38:41 +01002425 return t; /* nothing more to do */
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002426 }
2427
2428 s->fe->feconn--;
2429 if (s->flags & SN_BE_ASSIGNED)
2430 s->be->beconn--;
Willy Tarreau3c63fd82011-09-07 18:00:47 +02002431 if (!(s->listener->options & LI_O_UNLIMITED))
2432 actconn--;
Willy Tarreauaf7ad002010-08-31 15:39:26 +02002433 jobs--;
Willy Tarreau6e6fb2b2009-08-16 18:20:44 +02002434 s->listener->nbconn--;
Willy Tarreau62793712011-07-24 19:23:38 +02002435 if (s->listener->state == LI_FULL)
2436 resume_listener(s->listener);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002437
Willy Tarreau08ceb102011-07-24 22:58:00 +02002438 /* Dequeues all of the listeners waiting for a resource */
2439 if (!LIST_ISEMPTY(&global_listener_queue))
2440 dequeue_all_listeners(&global_listener_queue);
2441
Willy Tarreaub32907b2011-07-25 08:37:44 +02002442 if (!LIST_ISEMPTY(&s->fe->listener_queue) &&
2443 (!s->fe->fe_sps_lim || freq_ctr_remain(&s->fe->fe_sess_per_sec, s->fe->fe_sps_lim, 0) > 0))
Willy Tarreau07687c12011-07-24 23:55:06 +02002444 dequeue_all_listeners(&s->fe->listener_queue);
2445
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002446 if (unlikely((global.mode & MODE_DEBUG) &&
2447 (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE)))) {
Willy Tarreau19d14ef2012-10-29 16:51:55 +01002448 chunk_printf(&trash, "%08x:%s.closed[%04x:%04x]\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002449 s->uniq_id, s->be->id,
Willy Tarreau7f7ad912012-11-11 19:27:15 +01002450 (unsigned short)s->req->prod->conn->t.sock.fd,
2451 (unsigned short)s->req->cons->conn->t.sock.fd);
Willy Tarreau19d14ef2012-10-29 16:51:55 +01002452 if (write(1, trash.str, trash.len) < 0) /* shut gcc warning */;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002453 }
2454
2455 s->logs.t_close = tv_ms_elapsed(&s->logs.tv_accept, &now);
2456 session_process_counters(s);
2457
Krzysztof Piotr Oledzkide71d162009-10-24 15:36:15 +02002458 if (s->txn.status) {
2459 int n;
2460
2461 n = s->txn.status / 100;
2462 if (n < 1 || n > 5)
2463 n = 0;
2464
Willy Tarreau5e16cbc2012-11-24 14:54:13 +01002465 if (s->fe->mode == PR_MODE_HTTP) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002466 s->fe->fe_counters.p.http.rsp[n]++;
Willy Tarreau8139b992012-11-27 07:35:31 +01002467 if (s->comp_algo && (s->flags & SN_COMP_READY))
Willy Tarreau5e16cbc2012-11-24 14:54:13 +01002468 s->fe->fe_counters.p.http.comp_rsp++;
2469 }
Willy Tarreau24657792010-02-26 10:30:28 +01002470 if ((s->flags & SN_BE_ASSIGNED) &&
Willy Tarreau5e16cbc2012-11-24 14:54:13 +01002471 (s->be->mode == PR_MODE_HTTP)) {
Willy Tarreau7d0aaf32011-03-10 23:25:56 +01002472 s->be->be_counters.p.http.rsp[n]++;
Willy Tarreau5e16cbc2012-11-24 14:54:13 +01002473 s->be->be_counters.p.http.cum_req++;
Willy Tarreau8139b992012-11-27 07:35:31 +01002474 if (s->comp_algo && (s->flags & SN_COMP_READY))
Willy Tarreau5e16cbc2012-11-24 14:54:13 +01002475 s->be->be_counters.p.http.comp_rsp++;
2476 }
Krzysztof Piotr Oledzkide71d162009-10-24 15:36:15 +02002477 }
2478
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002479 /* let's do a final log if we need it */
Willy Tarreaud79a3b22012-12-28 09:40:16 +01002480 if (!LIST_ISEMPTY(&s->fe->logformat) && s->logs.logwait &&
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002481 !(s->flags & SN_MONITOR) &&
2482 (!(s->fe->options & PR_O_NULLNOLOG) || s->req->total)) {
Willy Tarreaua5555ec2008-11-30 19:02:32 +01002483 s->do_log(s);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002484 }
2485
2486 /* the task MUST not be in the run queue anymore */
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002487 session_free(s);
Willy Tarreau26c25062009-03-08 09:38:41 +01002488 task_delete(t);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002489 task_free(t);
Willy Tarreau26c25062009-03-08 09:38:41 +01002490 return NULL;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01002491}
2492
Willy Tarreau7c669d72008-06-20 15:04:11 +02002493/*
2494 * This function adjusts sess->srv_conn and maintains the previous and new
2495 * server's served session counts. Setting newsrv to NULL is enough to release
2496 * current connection slot. This function also notifies any LB algo which might
2497 * expect to be informed about any change in the number of active sessions on a
2498 * server.
2499 */
2500void sess_change_server(struct session *sess, struct server *newsrv)
2501{
2502 if (sess->srv_conn == newsrv)
2503 return;
2504
2505 if (sess->srv_conn) {
2506 sess->srv_conn->served--;
2507 if (sess->srv_conn->proxy->lbprm.server_drop_conn)
2508 sess->srv_conn->proxy->lbprm.server_drop_conn(sess->srv_conn);
Simon Hormanaf514952011-06-21 14:34:57 +09002509 session_del_srv_conn(sess);
Willy Tarreau7c669d72008-06-20 15:04:11 +02002510 }
2511
2512 if (newsrv) {
2513 newsrv->served++;
2514 if (newsrv->proxy->lbprm.server_take_conn)
2515 newsrv->proxy->lbprm.server_take_conn(newsrv);
Simon Hormanaf514952011-06-21 14:34:57 +09002516 session_add_srv_conn(sess, newsrv);
Willy Tarreau7c669d72008-06-20 15:04:11 +02002517 }
2518}
2519
Willy Tarreau84455332009-03-15 22:34:05 +01002520/* Handle server-side errors for default protocols. It is called whenever a a
2521 * connection setup is aborted or a request is aborted in queue. It sets the
2522 * session termination flags so that the caller does not have to worry about
2523 * them. It's installed as ->srv_error for the server-side stream_interface.
2524 */
2525void default_srv_error(struct session *s, struct stream_interface *si)
2526{
2527 int err_type = si->err_type;
2528 int err = 0, fin = 0;
2529
2530 if (err_type & SI_ET_QUEUE_ABRT) {
2531 err = SN_ERR_CLICL;
2532 fin = SN_FINST_Q;
2533 }
2534 else if (err_type & SI_ET_CONN_ABRT) {
2535 err = SN_ERR_CLICL;
2536 fin = SN_FINST_C;
2537 }
2538 else if (err_type & SI_ET_QUEUE_TO) {
2539 err = SN_ERR_SRVTO;
2540 fin = SN_FINST_Q;
2541 }
2542 else if (err_type & SI_ET_QUEUE_ERR) {
2543 err = SN_ERR_SRVCL;
2544 fin = SN_FINST_Q;
2545 }
2546 else if (err_type & SI_ET_CONN_TO) {
2547 err = SN_ERR_SRVTO;
2548 fin = SN_FINST_C;
2549 }
2550 else if (err_type & SI_ET_CONN_ERR) {
2551 err = SN_ERR_SRVCL;
2552 fin = SN_FINST_C;
2553 }
2554 else /* SI_ET_CONN_OTHER and others */ {
2555 err = SN_ERR_INTERNAL;
2556 fin = SN_FINST_C;
2557 }
2558
2559 if (!(s->flags & SN_ERR_MASK))
2560 s->flags |= err;
2561 if (!(s->flags & SN_FINST_MASK))
2562 s->flags |= fin;
2563}
Willy Tarreau7c669d72008-06-20 15:04:11 +02002564
Willy Tarreaua2a64e92011-09-07 23:01:56 +02002565/* kill a session and set the termination flags to <why> (one of SN_ERR_*) */
2566void session_shutdown(struct session *session, int why)
2567{
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02002568 if (session->req->flags & (CF_SHUTW|CF_SHUTW_NOW))
Willy Tarreaua2a64e92011-09-07 23:01:56 +02002569 return;
2570
Willy Tarreau8263d2b2012-08-28 00:06:31 +02002571 channel_shutw_now(session->req);
2572 channel_shutr_now(session->rep);
Willy Tarreaua2a64e92011-09-07 23:01:56 +02002573 session->task->nice = 1024;
2574 if (!(session->flags & SN_ERR_MASK))
2575 session->flags |= why;
2576 task_wakeup(session->task, TASK_WOKEN_OTHER);
2577}
Willy Tarreau9ba2dcc2010-06-14 21:04:55 +02002578
Willy Tarreau8b22a712010-06-18 17:46:06 +02002579/************************************************************************/
2580/* All supported ACL keywords must be declared here. */
2581/************************************************************************/
2582
Willy Tarreau4d4149c2013-07-23 19:33:46 +02002583/* Returns a pointer to a stkctr depending on the fetch keyword name.
2584 * It is designed to be called as sc[0-9]_* sc_* or src_* exclusively.
Willy Tarreaua65536c2013-07-22 22:40:11 +02002585 * sc[0-9]_* will return a pointer to the respective field in the
Willy Tarreau4d4149c2013-07-23 19:33:46 +02002586 * session <l4>. sc_* requires an UINT argument specifying the stick
2587 * counter number. src_* will fill a locally allocated structure with
Willy Tarreaua65536c2013-07-22 22:40:11 +02002588 * the table and entry corresponding to what is specified with src_*.
Willy Tarreau0f791d42013-07-23 19:56:43 +02002589 * NULL may be returned if the designated stkctr is not tracked. For
2590 * the sc_* and sc[0-9]_* forms, an optional table argument may be
2591 * passed. When present, the currently tracked key is then looked up
2592 * in the specified table instead of the current table. The purpose is
2593 * to be able to convery multiple values per key (eg: have gpc0 from
2594 * multiple tables).
Willy Tarreaua65536c2013-07-22 22:40:11 +02002595 */
2596static struct stkctr *
2597smp_fetch_sc_stkctr(struct session *l4, const struct arg *args, const char *kw)
2598{
2599 static struct stkctr stkctr;
Willy Tarreau4d4149c2013-07-23 19:33:46 +02002600 unsigned int num = kw[2] - '0';
Willy Tarreau0f791d42013-07-23 19:56:43 +02002601 int arg = 0;
Willy Tarreaua65536c2013-07-22 22:40:11 +02002602
Willy Tarreau0f791d42013-07-23 19:56:43 +02002603 if (num == '_' - '0') {
2604 /* sc_* variant, args[0] = ctr# (mandatory) */
2605 num = args[arg++].data.uint;
Willy Tarreau4d4149c2013-07-23 19:33:46 +02002606 if (num >= MAX_SESS_STKCTR)
2607 return NULL;
Willy Tarreaua65536c2013-07-22 22:40:11 +02002608 }
Willy Tarreau0f791d42013-07-23 19:56:43 +02002609 else if (num > 9) { /* src_* variant, args[0] = table */
Willy Tarreaua65536c2013-07-22 22:40:11 +02002610 struct stktable_key *key = addr_to_stktable_key(&l4->si[0].conn->addr.from);
2611
2612 if (!key)
2613 return NULL;
2614 stkctr.table = &args->data.prx->table;
2615 stkctr.entry = stktable_lookup_key(stkctr.table, key);
2616 return &stkctr;
2617 }
Willy Tarreau0f791d42013-07-23 19:56:43 +02002618
2619 /* Here, <num> contains the counter number from 0 to 9 for
2620 * the sc[0-9]_ form, or even higher using sc_(num) if needed.
2621 * args[arg] is the first optional argument.
2622 */
2623 if (unlikely(args[arg].type == ARGT_TAB)) {
2624 /* an alternate table was specified, let's look up the same key there */
2625 stkctr.table = &args[arg].data.prx->table;
2626 stkctr.entry = stktable_lookup(stkctr.table, l4->stkctr[num].entry);
2627 return &stkctr;
2628 }
2629 return l4->stkctr[num].entry ? &l4->stkctr[num] : NULL;
Willy Tarreaua65536c2013-07-22 22:40:11 +02002630}
2631
Willy Tarreau88821242013-07-22 18:29:29 +02002632/* set return a boolean indicating if the requested session counter is
2633 * currently being tracked or not.
2634 * Supports being called as "sc[0-9]_tracked" only.
2635 */
Willy Tarreau6f1615f2013-06-03 15:15:22 +02002636static int
Willy Tarreau88821242013-07-22 18:29:29 +02002637smp_fetch_sc_tracked(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02002638 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau6f1615f2013-06-03 15:15:22 +02002639{
2640 smp->flags = SMP_F_VOL_TEST;
2641 smp->type = SMP_T_BOOL;
Willy Tarreau88821242013-07-22 18:29:29 +02002642 smp->data.uint = !!l4->stkctr[kw[2] - '0'].entry;
Willy Tarreau6f1615f2013-06-03 15:15:22 +02002643 return 1;
2644}
2645
Willy Tarreau30b20462013-07-22 19:46:52 +02002646/* set <smp> to the General Purpose Counter 0 value from the session's tracked
2647 * frontend counters or from the src.
2648 * Supports being called as "sc[0-9]_get_gpc0" or "src_get_gpc0" only. Value
2649 * zero is returned if the key is new.
2650 */
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002651static int
Willy Tarreau30b20462013-07-22 19:46:52 +02002652smp_fetch_sc_get_gpc0(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2653 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002654{
Willy Tarreau30b20462013-07-22 19:46:52 +02002655 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2656
2657 if (!stkctr)
2658 return 0;
2659
Willy Tarreau37406352012-04-23 16:16:37 +02002660 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002661 smp->type = SMP_T_UINT;
2662 smp->data.uint = 0;
Willy Tarreau30b20462013-07-22 19:46:52 +02002663
2664 if (stkctr->entry != NULL) {
2665 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_GPC0);
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002666 if (!ptr)
2667 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002668 smp->data.uint = stktable_data_cast(ptr, gpc0);
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002669 }
2670 return 1;
2671}
2672
Willy Tarreaub5e0af02013-07-22 23:47:07 +02002673/* set <smp> to the General Purpose Counter 0's event rate from the session's
2674 * tracked frontend counters or from the src.
2675 * Supports being called as "sc[0-9]_gpc0_rate" or "src_gpc0_rate" only.
2676 * Value zero is returned if the key is new.
2677 */
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002678static int
Willy Tarreaub5e0af02013-07-22 23:47:07 +02002679smp_fetch_sc_gpc0_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2680 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002681{
Willy Tarreaub5e0af02013-07-22 23:47:07 +02002682 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2683
2684 if (!stkctr)
2685 return 0;
2686
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002687 smp->flags = SMP_F_VOL_TEST;
2688 smp->type = SMP_T_UINT;
2689 smp->data.uint = 0;
Willy Tarreaub5e0af02013-07-22 23:47:07 +02002690 if (stkctr->entry != NULL) {
2691 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_GPC0_RATE);
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002692 if (!ptr)
2693 return 0; /* parameter not stored */
2694 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, gpc0_rate),
Willy Tarreaub5e0af02013-07-22 23:47:07 +02002695 stkctr->table->data_arg[STKTABLE_DT_GPC0_RATE].u);
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002696 }
2697 return 1;
2698}
2699
Willy Tarreau710d38c2013-07-23 00:07:04 +02002700/* Increment the General Purpose Counter 0 value from the session's tracked
2701 * frontend counters and return it into temp integer.
2702 * Supports being called as "sc[0-9]_inc_gpc0" or "src_inc_gpc0" only.
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002703 */
2704static int
Willy Tarreau710d38c2013-07-23 00:07:04 +02002705smp_fetch_sc_inc_gpc0(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2706 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002707{
Willy Tarreau710d38c2013-07-23 00:07:04 +02002708 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2709
2710 if (!stkctr)
2711 return 0;
2712
Willy Tarreau37406352012-04-23 16:16:37 +02002713 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002714 smp->type = SMP_T_UINT;
2715 smp->data.uint = 0;
Willy Tarreau710d38c2013-07-23 00:07:04 +02002716 if (stkctr->entry != NULL) {
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002717 void *ptr;
2718
2719 /* First, update gpc0_rate if it's tracked. Second, update its
2720 * gpc0 if tracked. Returns gpc0's value otherwise the curr_ctr.
2721 */
Willy Tarreau710d38c2013-07-23 00:07:04 +02002722 ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_GPC0_RATE);
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002723 if (ptr) {
2724 update_freq_ctr_period(&stktable_data_cast(ptr, gpc0_rate),
Willy Tarreau710d38c2013-07-23 00:07:04 +02002725 stkctr->table->data_arg[STKTABLE_DT_GPC0_RATE].u, 1);
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002726 smp->data.uint = (&stktable_data_cast(ptr, gpc0_rate))->curr_ctr;
2727 }
2728
Willy Tarreau710d38c2013-07-23 00:07:04 +02002729 ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_GPC0);
Willy Tarreauba2ffd12013-05-29 15:54:14 +02002730 if (ptr)
2731 smp->data.uint = ++stktable_data_cast(ptr, gpc0);
2732
Willy Tarreauc3bd9722010-06-20 12:47:25 +02002733 }
2734 return 1;
2735}
2736
Willy Tarreaub9f441d2013-07-23 00:10:35 +02002737/* Clear the General Purpose Counter 0 value from the session's tracked
2738 * frontend counters and return its previous value into temp integer.
2739 * Supports being called as "sc[0-9]_clr_gpc0" or "src_clr_gpc0" only.
Willy Tarreauf73cd112011-08-13 01:45:16 +02002740 */
2741static int
Willy Tarreaub9f441d2013-07-23 00:10:35 +02002742smp_fetch_sc_clr_gpc0(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2743 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauf73cd112011-08-13 01:45:16 +02002744{
Willy Tarreaub9f441d2013-07-23 00:10:35 +02002745 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2746
2747 if (!stkctr)
2748 return 0;
2749
Willy Tarreau37406352012-04-23 16:16:37 +02002750 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002751 smp->type = SMP_T_UINT;
2752 smp->data.uint = 0;
Willy Tarreaub9f441d2013-07-23 00:10:35 +02002753 if (stkctr->entry != NULL) {
2754 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_GPC0);
Willy Tarreauf73cd112011-08-13 01:45:16 +02002755 if (!ptr)
2756 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002757 smp->data.uint = stktable_data_cast(ptr, gpc0);
Willy Tarreauf73cd112011-08-13 01:45:16 +02002758 stktable_data_cast(ptr, gpc0) = 0;
2759 }
2760 return 1;
2761}
2762
Willy Tarreau3b46c5c2013-07-23 00:22:50 +02002763/* set <smp> to the cumulated number of connections from the session's tracked
2764 * frontend counters. Supports being called as "sc[0-9]_conn_cnt" or
2765 * "src_conn_cnt" only.
2766 */
Willy Tarreau9a3f8492010-06-18 19:53:25 +02002767static int
Willy Tarreau3b46c5c2013-07-23 00:22:50 +02002768smp_fetch_sc_conn_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2769 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau9a3f8492010-06-18 19:53:25 +02002770{
Willy Tarreau3b46c5c2013-07-23 00:22:50 +02002771 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2772
2773 if (!stkctr)
2774 return 0;
2775
Willy Tarreau37406352012-04-23 16:16:37 +02002776 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002777 smp->type = SMP_T_UINT;
2778 smp->data.uint = 0;
Willy Tarreau3b46c5c2013-07-23 00:22:50 +02002779 if (stkctr->entry != NULL) {
2780 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_CONN_CNT);
Willy Tarreau9a3f8492010-06-18 19:53:25 +02002781 if (!ptr)
2782 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002783 smp->data.uint = stktable_data_cast(ptr, conn_cnt);
Willy Tarreau9a3f8492010-06-18 19:53:25 +02002784 }
Willy Tarreau9a3f8492010-06-18 19:53:25 +02002785 return 1;
2786}
2787
Willy Tarreauc8c65702013-07-23 15:09:35 +02002788/* set <smp> to the connection rate from the session's tracked frontend
2789 * counters. Supports being called as "sc[0-9]_conn_rate" or "src_conn_rate"
2790 * only.
2791 */
Willy Tarreau91c43d72010-06-20 11:19:22 +02002792static int
Willy Tarreauc8c65702013-07-23 15:09:35 +02002793smp_fetch_sc_conn_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2794 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau91c43d72010-06-20 11:19:22 +02002795{
Willy Tarreauc8c65702013-07-23 15:09:35 +02002796 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2797
2798 if (!stkctr)
2799 return 0;
2800
Willy Tarreau37406352012-04-23 16:16:37 +02002801 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002802 smp->type = SMP_T_UINT;
2803 smp->data.uint = 0;
Willy Tarreauc8c65702013-07-23 15:09:35 +02002804 if (stkctr->entry != NULL) {
2805 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_CONN_RATE);
Willy Tarreau91c43d72010-06-20 11:19:22 +02002806 if (!ptr)
2807 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002808 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, conn_rate),
Willy Tarreauc8c65702013-07-23 15:09:35 +02002809 stkctr->table->data_arg[STKTABLE_DT_CONN_RATE].u);
Willy Tarreau91c43d72010-06-20 11:19:22 +02002810 }
2811 return 1;
2812}
2813
Willy Tarreaua5e37562011-12-16 17:06:15 +01002814/* set temp integer to the number of connections from the session's source address
Willy Tarreau8b22a712010-06-18 17:46:06 +02002815 * in the table pointed to by expr, after updating it.
Willy Tarreau0146c2e2012-04-20 11:37:56 +02002816 * Accepts exactly 1 argument of type table.
Willy Tarreau8b22a712010-06-18 17:46:06 +02002817 */
2818static int
Willy Tarreau281c7992013-01-08 01:23:27 +01002819smp_fetch_src_updt_conn_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02002820 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau8b22a712010-06-18 17:46:06 +02002821{
2822 struct stksess *ts;
2823 struct stktable_key *key;
2824 void *ptr;
2825
Willy Tarreauf2943dc2012-10-26 20:10:28 +02002826 key = addr_to_stktable_key(&l4->si[0].conn->addr.from);
Willy Tarreau8b22a712010-06-18 17:46:06 +02002827 if (!key)
David du Colombier4f92d322011-03-24 11:09:31 +01002828 return 0;
Willy Tarreau8b22a712010-06-18 17:46:06 +02002829
Willy Tarreau24e32d82012-04-23 23:55:44 +02002830 px = args->data.prx;
Willy Tarreau8b22a712010-06-18 17:46:06 +02002831
Willy Tarreau1f7e9252010-06-20 12:27:21 +02002832 if ((ts = stktable_update_key(&px->table, key)) == NULL)
2833 /* entry does not exist and could not be created */
2834 return 0;
Willy Tarreau8b22a712010-06-18 17:46:06 +02002835
2836 ptr = stktable_data_ptr(&px->table, ts, STKTABLE_DT_CONN_CNT);
2837 if (!ptr)
2838 return 0; /* parameter not stored in this table */
2839
Willy Tarreauf853c462012-04-23 18:53:56 +02002840 smp->type = SMP_T_UINT;
2841 smp->data.uint = ++stktable_data_cast(ptr, conn_cnt);
Willy Tarreau37406352012-04-23 16:16:37 +02002842 smp->flags = SMP_F_VOL_TEST;
Willy Tarreau8b22a712010-06-18 17:46:06 +02002843 return 1;
2844}
2845
Willy Tarreauf44a5532013-07-23 15:17:53 +02002846/* set <smp> to the number of concurrent connections from the session's tracked
2847 * frontend counters. Supports being called as "sc[0-9]_conn_cur" or
2848 * "src_conn_cur" only.
2849 */
Willy Tarreau9b0ddcf2010-06-18 21:14:36 +02002850static int
Willy Tarreauf44a5532013-07-23 15:17:53 +02002851smp_fetch_sc_conn_cur(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2852 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau9b0ddcf2010-06-18 21:14:36 +02002853{
Willy Tarreauf44a5532013-07-23 15:17:53 +02002854 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2855
2856 if (!stkctr)
2857 return 0;
2858
Willy Tarreau37406352012-04-23 16:16:37 +02002859 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002860 smp->type = SMP_T_UINT;
2861 smp->data.uint = 0;
Willy Tarreauf44a5532013-07-23 15:17:53 +02002862 if (stkctr->entry != NULL) {
2863 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_CONN_CUR);
Willy Tarreau9b0ddcf2010-06-18 21:14:36 +02002864 if (!ptr)
2865 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002866 smp->data.uint = stktable_data_cast(ptr, conn_cur);
Willy Tarreau9b0ddcf2010-06-18 21:14:36 +02002867 }
2868 return 1;
2869}
2870
Willy Tarreau20843082013-07-23 15:35:33 +02002871/* set <smp> to the cumulated number of sessions from the session's tracked
2872 * frontend counters. Supports being called as "sc[0-9]_sess_cnt" or
2873 * "src_sess_cnt" only.
2874 */
Willy Tarreauf4d17d92010-06-18 22:10:12 +02002875static int
Willy Tarreau20843082013-07-23 15:35:33 +02002876smp_fetch_sc_sess_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2877 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauf4d17d92010-06-18 22:10:12 +02002878{
Willy Tarreau20843082013-07-23 15:35:33 +02002879 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2880
2881 if (!stkctr)
2882 return 0;
2883
Willy Tarreau37406352012-04-23 16:16:37 +02002884 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002885 smp->type = SMP_T_UINT;
2886 smp->data.uint = 0;
Willy Tarreau20843082013-07-23 15:35:33 +02002887 if (stkctr->entry != NULL) {
2888 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_SESS_CNT);
Willy Tarreauf4d17d92010-06-18 22:10:12 +02002889 if (!ptr)
2890 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002891 smp->data.uint = stktable_data_cast(ptr, sess_cnt);
Willy Tarreauf4d17d92010-06-18 22:10:12 +02002892 }
2893 return 1;
2894}
2895
Willy Tarreau3a96f3f2013-07-23 15:48:01 +02002896/* set <smp> to the session rate from the session's tracked frontend counters.
2897 * Supports being called as "sc[0-9]_sess_rate" or "src_sess_rate" only.
2898 */
Willy Tarreau91c43d72010-06-20 11:19:22 +02002899static int
Willy Tarreau3a96f3f2013-07-23 15:48:01 +02002900smp_fetch_sc_sess_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2901 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau91c43d72010-06-20 11:19:22 +02002902{
Willy Tarreau3a96f3f2013-07-23 15:48:01 +02002903 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2904
2905 if (!stkctr)
2906 return 0;
2907
Willy Tarreau37406352012-04-23 16:16:37 +02002908 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002909 smp->type = SMP_T_UINT;
2910 smp->data.uint = 0;
Willy Tarreau3a96f3f2013-07-23 15:48:01 +02002911 if (stkctr->entry != NULL) {
2912 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_SESS_RATE);
Willy Tarreau91c43d72010-06-20 11:19:22 +02002913 if (!ptr)
2914 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002915 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, sess_rate),
Willy Tarreau3a96f3f2013-07-23 15:48:01 +02002916 stkctr->table->data_arg[STKTABLE_DT_SESS_RATE].u);
Willy Tarreau91c43d72010-06-20 11:19:22 +02002917 }
2918 return 1;
2919}
2920
Willy Tarreau91200da2013-07-23 15:55:19 +02002921/* set <smp> to the cumulated number of HTTP requests from the session's tracked
2922 * frontend counters. Supports being called as "sc[0-9]_http_req_cnt" or
2923 * "src_http_req_cnt" only.
2924 */
Willy Tarreauda7ff642010-06-23 11:44:09 +02002925static int
Willy Tarreau91200da2013-07-23 15:55:19 +02002926smp_fetch_sc_http_req_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2927 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauda7ff642010-06-23 11:44:09 +02002928{
Willy Tarreau91200da2013-07-23 15:55:19 +02002929 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2930
2931 if (!stkctr)
2932 return 0;
2933
Willy Tarreau37406352012-04-23 16:16:37 +02002934 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002935 smp->type = SMP_T_UINT;
2936 smp->data.uint = 0;
Willy Tarreau91200da2013-07-23 15:55:19 +02002937 if (stkctr->entry != NULL) {
2938 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_HTTP_REQ_CNT);
Willy Tarreauda7ff642010-06-23 11:44:09 +02002939 if (!ptr)
2940 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002941 smp->data.uint = stktable_data_cast(ptr, http_req_cnt);
Willy Tarreauda7ff642010-06-23 11:44:09 +02002942 }
2943 return 1;
2944}
2945
Willy Tarreaucf477632013-07-23 16:04:37 +02002946/* set <smp> to the HTTP request rate from the session's tracked frontend
2947 * counters. Supports being called as "sc[0-9]_http_req_rate" or
2948 * "src_http_req_rate" only.
2949 */
Willy Tarreauda7ff642010-06-23 11:44:09 +02002950static int
Willy Tarreaucf477632013-07-23 16:04:37 +02002951smp_fetch_sc_http_req_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2952 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauda7ff642010-06-23 11:44:09 +02002953{
Willy Tarreaucf477632013-07-23 16:04:37 +02002954 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2955
2956 if (!stkctr)
2957 return 0;
2958
Willy Tarreau37406352012-04-23 16:16:37 +02002959 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002960 smp->type = SMP_T_UINT;
2961 smp->data.uint = 0;
Willy Tarreaucf477632013-07-23 16:04:37 +02002962 if (stkctr->entry != NULL) {
2963 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_HTTP_REQ_RATE);
Willy Tarreauda7ff642010-06-23 11:44:09 +02002964 if (!ptr)
2965 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002966 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, http_req_rate),
Willy Tarreaucf477632013-07-23 16:04:37 +02002967 stkctr->table->data_arg[STKTABLE_DT_HTTP_REQ_RATE].u);
Willy Tarreauda7ff642010-06-23 11:44:09 +02002968 }
2969 return 1;
2970}
2971
Willy Tarreau30d07c32013-07-23 16:45:38 +02002972/* set <smp> to the cumulated number of HTTP requests errors from the session's
2973 * tracked frontend counters. Supports being called as "sc[0-9]_http_err_cnt" or
2974 * "src_http_err_cnt" only.
2975 */
Willy Tarreauda7ff642010-06-23 11:44:09 +02002976static int
Willy Tarreau30d07c32013-07-23 16:45:38 +02002977smp_fetch_sc_http_err_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
2978 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauda7ff642010-06-23 11:44:09 +02002979{
Willy Tarreau30d07c32013-07-23 16:45:38 +02002980 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
2981
2982 if (!stkctr)
2983 return 0;
2984
Willy Tarreau37406352012-04-23 16:16:37 +02002985 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02002986 smp->type = SMP_T_UINT;
2987 smp->data.uint = 0;
Willy Tarreau30d07c32013-07-23 16:45:38 +02002988 if (stkctr->entry != NULL) {
2989 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_HTTP_ERR_CNT);
Willy Tarreauda7ff642010-06-23 11:44:09 +02002990 if (!ptr)
2991 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02002992 smp->data.uint = stktable_data_cast(ptr, http_err_cnt);
Willy Tarreauda7ff642010-06-23 11:44:09 +02002993 }
2994 return 1;
2995}
2996
Willy Tarreau9daf2622013-07-23 16:48:54 +02002997/* set <smp> to the HTTP request error rate from the session's tracked frontend
2998 * counters. Supports being called as "sc[0-9]_http_err_rate" or
2999 * "src_http_err_rate" only.
3000 */
Willy Tarreauda7ff642010-06-23 11:44:09 +02003001static int
Willy Tarreau9daf2622013-07-23 16:48:54 +02003002smp_fetch_sc_http_err_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
3003 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauda7ff642010-06-23 11:44:09 +02003004{
Willy Tarreau9daf2622013-07-23 16:48:54 +02003005 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
3006
3007 if (!stkctr)
3008 return 0;
3009
Willy Tarreau37406352012-04-23 16:16:37 +02003010 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003011 smp->type = SMP_T_UINT;
3012 smp->data.uint = 0;
Willy Tarreau9daf2622013-07-23 16:48:54 +02003013 if (stkctr->entry != NULL) {
3014 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_HTTP_ERR_RATE);
Willy Tarreauda7ff642010-06-23 11:44:09 +02003015 if (!ptr)
3016 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02003017 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, http_err_rate),
Willy Tarreau9daf2622013-07-23 16:48:54 +02003018 stkctr->table->data_arg[STKTABLE_DT_HTTP_ERR_RATE].u);
Willy Tarreauda7ff642010-06-23 11:44:09 +02003019 }
3020 return 1;
3021}
3022
Willy Tarreau5077d4b2013-07-23 17:17:10 +02003023/* set <smp> to the number of kbytes received from clients, as found in the
3024 * session's tracked frontend counters. Supports being called as
3025 * "sc[0-9]_kbytes_in" or "src_kbytes_in" only.
3026 */
Willy Tarreau1aa006f2010-06-18 21:52:52 +02003027static int
Willy Tarreau5077d4b2013-07-23 17:17:10 +02003028smp_fetch_sc_kbytes_in(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
3029 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau1aa006f2010-06-18 21:52:52 +02003030{
Willy Tarreau5077d4b2013-07-23 17:17:10 +02003031 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
3032
3033 if (!stkctr)
3034 return 0;
3035
Willy Tarreau37406352012-04-23 16:16:37 +02003036 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003037 smp->type = SMP_T_UINT;
3038 smp->data.uint = 0;
Willy Tarreau5077d4b2013-07-23 17:17:10 +02003039 if (stkctr->entry != NULL) {
3040 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_BYTES_IN_CNT);
Willy Tarreau1aa006f2010-06-18 21:52:52 +02003041 if (!ptr)
3042 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02003043 smp->data.uint = stktable_data_cast(ptr, bytes_in_cnt) >> 10;
Willy Tarreau1aa006f2010-06-18 21:52:52 +02003044 }
3045 return 1;
3046}
3047
Willy Tarreau613fe992013-07-23 17:39:19 +02003048/* set <smp> to the data rate received from clients in bytes/s, as found
3049 * in the session's tracked frontend counters. Supports being called as
3050 * "sc[0-9]_bytes_in_rate" or "src_bytes_in_rate" only.
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003051 */
3052static int
Willy Tarreau613fe992013-07-23 17:39:19 +02003053smp_fetch_sc_bytes_in_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
3054 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003055{
Willy Tarreau613fe992013-07-23 17:39:19 +02003056 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
3057
3058 if (!stkctr)
3059 return 0;
3060
Willy Tarreau37406352012-04-23 16:16:37 +02003061 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003062 smp->type = SMP_T_UINT;
3063 smp->data.uint = 0;
Willy Tarreau613fe992013-07-23 17:39:19 +02003064 if (stkctr->entry != NULL) {
3065 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_BYTES_IN_RATE);
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003066 if (!ptr)
3067 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02003068 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, bytes_in_rate),
Willy Tarreau613fe992013-07-23 17:39:19 +02003069 stkctr->table->data_arg[STKTABLE_DT_BYTES_IN_RATE].u);
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003070 }
3071 return 1;
3072}
3073
Willy Tarreau53aea102013-07-23 17:39:02 +02003074/* set <smp> to the number of kbytes sent to clients, as found in the
3075 * session's tracked frontend counters. Supports being called as
3076 * "sc[0-9]_kbytes_out" or "src_kbytes_out" only.
3077 */
Willy Tarreau1aa006f2010-06-18 21:52:52 +02003078static int
Willy Tarreau53aea102013-07-23 17:39:02 +02003079smp_fetch_sc_kbytes_out(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
3080 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau1aa006f2010-06-18 21:52:52 +02003081{
Willy Tarreau53aea102013-07-23 17:39:02 +02003082 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
3083
3084 if (!stkctr)
3085 return 0;
3086
Willy Tarreau37406352012-04-23 16:16:37 +02003087 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003088 smp->type = SMP_T_UINT;
3089 smp->data.uint = 0;
Willy Tarreau53aea102013-07-23 17:39:02 +02003090 if (stkctr->entry != NULL) {
3091 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_BYTES_OUT_CNT);
Willy Tarreau855e4bb2010-06-18 18:33:32 +02003092 if (!ptr)
3093 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02003094 smp->data.uint = stktable_data_cast(ptr, bytes_out_cnt) >> 10;
Willy Tarreau855e4bb2010-06-18 18:33:32 +02003095 }
Willy Tarreau855e4bb2010-06-18 18:33:32 +02003096 return 1;
3097}
3098
Willy Tarreaua0b68ed2013-07-23 18:26:32 +02003099/* set <smp> to the data rate sent to clients in bytes/s, as found in the
3100 * session's tracked frontend counters. Supports being called as
3101 * "sc[0-9]_bytes_out_rate" or "src_bytes_out_rate" only.
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003102 */
3103static int
Willy Tarreaua0b68ed2013-07-23 18:26:32 +02003104smp_fetch_sc_bytes_out_rate(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
3105 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003106{
Willy Tarreaua0b68ed2013-07-23 18:26:32 +02003107 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
3108
3109 if (!stkctr)
3110 return 0;
3111
Willy Tarreau37406352012-04-23 16:16:37 +02003112 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003113 smp->type = SMP_T_UINT;
3114 smp->data.uint = 0;
Willy Tarreaua0b68ed2013-07-23 18:26:32 +02003115 if (stkctr->entry != NULL) {
3116 void *ptr = stktable_data_ptr(stkctr->table, stkctr->entry, STKTABLE_DT_BYTES_OUT_RATE);
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003117 if (!ptr)
3118 return 0; /* parameter not stored */
Willy Tarreauf853c462012-04-23 18:53:56 +02003119 smp->data.uint = read_freq_ctr_period(&stktable_data_cast(ptr, bytes_out_rate),
Willy Tarreaua0b68ed2013-07-23 18:26:32 +02003120 stkctr->table->data_arg[STKTABLE_DT_BYTES_OUT_RATE].u);
Willy Tarreau6c59e0a2010-06-20 11:56:30 +02003121 }
3122 return 1;
3123}
3124
Willy Tarreau563eef42013-07-23 18:32:02 +02003125/* set <smp> to the number of active trackers on the SC entry in the session's
3126 * tracked frontend counters. Supports being called as "sc[0-9]_trackers" only.
3127 */
Willy Tarreau2406db42012-12-09 12:16:43 +01003128static int
Willy Tarreau563eef42013-07-23 18:32:02 +02003129smp_fetch_sc_trackers(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02003130 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreau2406db42012-12-09 12:16:43 +01003131{
Willy Tarreau563eef42013-07-23 18:32:02 +02003132 struct stkctr *stkctr = smp_fetch_sc_stkctr(l4, args, kw);
Willy Tarreau2406db42012-12-09 12:16:43 +01003133
Willy Tarreau563eef42013-07-23 18:32:02 +02003134 if (!stkctr)
Willy Tarreau2406db42012-12-09 12:16:43 +01003135 return 0;
3136
Willy Tarreau563eef42013-07-23 18:32:02 +02003137 smp->flags = SMP_F_VOL_TEST;
3138 smp->type = SMP_T_UINT;
3139 smp->data.uint = stkctr->entry->ref_cnt;
3140 return 1;
Willy Tarreaue25c9172013-05-28 18:32:20 +02003141}
3142
Willy Tarreau34db1082012-04-19 17:16:54 +02003143/* set temp integer to the number of used entries in the table pointed to by expr.
Willy Tarreau0146c2e2012-04-20 11:37:56 +02003144 * Accepts exactly 1 argument of type table.
Willy Tarreau34db1082012-04-19 17:16:54 +02003145 */
Willy Tarreauc735a072011-03-29 00:57:02 +02003146static int
Willy Tarreau281c7992013-01-08 01:23:27 +01003147smp_fetch_table_cnt(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02003148 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauc735a072011-03-29 00:57:02 +02003149{
Willy Tarreau37406352012-04-23 16:16:37 +02003150 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003151 smp->type = SMP_T_UINT;
Willy Tarreau24e32d82012-04-23 23:55:44 +02003152 smp->data.uint = args->data.prx->table.current;
Willy Tarreauc735a072011-03-29 00:57:02 +02003153 return 1;
3154}
3155
Willy Tarreau34db1082012-04-19 17:16:54 +02003156/* set temp integer to the number of free entries in the table pointed to by expr.
Willy Tarreau0146c2e2012-04-20 11:37:56 +02003157 * Accepts exactly 1 argument of type table.
Willy Tarreau34db1082012-04-19 17:16:54 +02003158 */
Willy Tarreauc735a072011-03-29 00:57:02 +02003159static int
Willy Tarreau281c7992013-01-08 01:23:27 +01003160smp_fetch_table_avl(struct proxy *px, struct session *l4, void *l7, unsigned int opt,
Willy Tarreauef38c392013-07-22 16:29:32 +02003161 const struct arg *args, struct sample *smp, const char *kw)
Willy Tarreauc735a072011-03-29 00:57:02 +02003162{
Willy Tarreau24e32d82012-04-23 23:55:44 +02003163 px = args->data.prx;
Willy Tarreau37406352012-04-23 16:16:37 +02003164 smp->flags = SMP_F_VOL_TEST;
Willy Tarreauf853c462012-04-23 18:53:56 +02003165 smp->type = SMP_T_UINT;
3166 smp->data.uint = px->table.size - px->table.current;
Willy Tarreauc735a072011-03-29 00:57:02 +02003167 return 1;
3168}
Willy Tarreau8b22a712010-06-18 17:46:06 +02003169
Willy Tarreau61612d42012-04-19 18:42:05 +02003170/* Note: must not be declared <const> as its list will be overwritten.
3171 * Please take care of keeping this list alphabetically sorted.
3172 */
Willy Tarreaudc13c112013-06-21 23:16:39 +02003173static struct acl_kw_list acl_kws = {ILH, {
Willy Tarreau281c7992013-01-08 01:23:27 +01003174 { /* END */ },
Willy Tarreau8b22a712010-06-18 17:46:06 +02003175}};
3176
Willy Tarreau281c7992013-01-08 01:23:27 +01003177/* Note: must not be declared <const> as its list will be overwritten.
3178 * Please take care of keeping this list alphabetically sorted.
3179 */
Willy Tarreaudc13c112013-06-21 23:16:39 +02003180static struct sample_fetch_kw_list smp_fetch_keywords = {ILH, {
Willy Tarreau0f791d42013-07-23 19:56:43 +02003181 { "sc_bytes_in_rate", smp_fetch_sc_bytes_in_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3182 { "sc_bytes_out_rate", smp_fetch_sc_bytes_out_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3183 { "sc_clr_gpc0", smp_fetch_sc_clr_gpc0, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3184 { "sc_conn_cnt", smp_fetch_sc_conn_cnt, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3185 { "sc_conn_cur", smp_fetch_sc_conn_cur, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3186 { "sc_conn_rate", smp_fetch_sc_conn_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3187 { "sc_get_gpc0", smp_fetch_sc_get_gpc0, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3188 { "sc_gpc0_rate", smp_fetch_sc_gpc0_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3189 { "sc_http_err_cnt", smp_fetch_sc_http_err_cnt, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3190 { "sc_http_err_rate", smp_fetch_sc_http_err_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3191 { "sc_http_req_cnt", smp_fetch_sc_http_req_cnt, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3192 { "sc_http_req_rate", smp_fetch_sc_http_req_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3193 { "sc_inc_gpc0", smp_fetch_sc_inc_gpc0, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3194 { "sc_kbytes_in", smp_fetch_sc_kbytes_in, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3195 { "sc_kbytes_out", smp_fetch_sc_kbytes_out, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3196 { "sc_sess_cnt", smp_fetch_sc_sess_cnt, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3197 { "sc_sess_rate", smp_fetch_sc_sess_rate, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3198 { "sc_tracked", smp_fetch_sc_tracked, ARG2(1,UINT,TAB), NULL, SMP_T_BOOL, SMP_USE_INTRN, },
3199 { "sc_trackers", smp_fetch_sc_trackers, ARG2(1,UINT,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3200 { "sc0_bytes_in_rate", smp_fetch_sc_bytes_in_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3201 { "sc0_bytes_out_rate", smp_fetch_sc_bytes_out_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3202 { "sc0_clr_gpc0", smp_fetch_sc_clr_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3203 { "sc0_conn_cnt", smp_fetch_sc_conn_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3204 { "sc0_conn_cur", smp_fetch_sc_conn_cur, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3205 { "sc0_conn_rate", smp_fetch_sc_conn_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3206 { "sc0_get_gpc0", smp_fetch_sc_get_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3207 { "sc0_gpc0_rate", smp_fetch_sc_gpc0_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3208 { "sc0_http_err_cnt", smp_fetch_sc_http_err_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3209 { "sc0_http_err_rate", smp_fetch_sc_http_err_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3210 { "sc0_http_req_cnt", smp_fetch_sc_http_req_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3211 { "sc0_http_req_rate", smp_fetch_sc_http_req_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3212 { "sc0_inc_gpc0", smp_fetch_sc_inc_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3213 { "sc0_kbytes_in", smp_fetch_sc_kbytes_in, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3214 { "sc0_kbytes_out", smp_fetch_sc_kbytes_out, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3215 { "sc0_sess_cnt", smp_fetch_sc_sess_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3216 { "sc0_sess_rate", smp_fetch_sc_sess_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3217 { "sc0_tracked", smp_fetch_sc_tracked, ARG1(0,TAB), NULL, SMP_T_BOOL, SMP_USE_INTRN, },
3218 { "sc0_trackers", smp_fetch_sc_trackers, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3219 { "sc1_bytes_in_rate", smp_fetch_sc_bytes_in_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3220 { "sc1_bytes_out_rate", smp_fetch_sc_bytes_out_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3221 { "sc1_clr_gpc0", smp_fetch_sc_clr_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3222 { "sc1_conn_cnt", smp_fetch_sc_conn_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3223 { "sc1_conn_cur", smp_fetch_sc_conn_cur, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3224 { "sc1_conn_rate", smp_fetch_sc_conn_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3225 { "sc1_get_gpc0", smp_fetch_sc_get_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3226 { "sc1_gpc0_rate", smp_fetch_sc_gpc0_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3227 { "sc1_http_err_cnt", smp_fetch_sc_http_err_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3228 { "sc1_http_err_rate", smp_fetch_sc_http_err_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3229 { "sc1_http_req_cnt", smp_fetch_sc_http_req_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3230 { "sc1_http_req_rate", smp_fetch_sc_http_req_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3231 { "sc1_inc_gpc0", smp_fetch_sc_inc_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3232 { "sc1_kbytes_in", smp_fetch_sc_kbytes_in, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3233 { "sc1_kbytes_out", smp_fetch_sc_kbytes_out, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3234 { "sc1_sess_cnt", smp_fetch_sc_sess_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3235 { "sc1_sess_rate", smp_fetch_sc_sess_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3236 { "sc1_tracked", smp_fetch_sc_tracked, ARG1(0,TAB), NULL, SMP_T_BOOL, SMP_USE_INTRN, },
3237 { "sc1_trackers", smp_fetch_sc_trackers, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3238 { "sc2_bytes_in_rate", smp_fetch_sc_bytes_in_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3239 { "sc2_bytes_out_rate", smp_fetch_sc_bytes_out_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3240 { "sc2_clr_gpc0", smp_fetch_sc_clr_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3241 { "sc2_conn_cnt", smp_fetch_sc_conn_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3242 { "sc2_conn_cur", smp_fetch_sc_conn_cur, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3243 { "sc2_conn_rate", smp_fetch_sc_conn_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3244 { "sc2_get_gpc0", smp_fetch_sc_get_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3245 { "sc2_gpc0_rate", smp_fetch_sc_gpc0_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3246 { "sc2_http_err_cnt", smp_fetch_sc_http_err_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3247 { "sc2_http_err_rate", smp_fetch_sc_http_err_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3248 { "sc2_http_req_cnt", smp_fetch_sc_http_req_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3249 { "sc2_http_req_rate", smp_fetch_sc_http_req_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3250 { "sc2_inc_gpc0", smp_fetch_sc_inc_gpc0, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3251 { "sc2_kbytes_in", smp_fetch_sc_kbytes_in, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3252 { "sc2_kbytes_out", smp_fetch_sc_kbytes_out, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3253 { "sc2_sess_cnt", smp_fetch_sc_sess_cnt, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3254 { "sc2_sess_rate", smp_fetch_sc_sess_rate, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3255 { "sc2_tracked", smp_fetch_sc_tracked, ARG1(0,TAB), NULL, SMP_T_BOOL, SMP_USE_INTRN, },
3256 { "sc2_trackers", smp_fetch_sc_trackers, ARG1(0,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3257 { "src_bytes_in_rate", smp_fetch_sc_bytes_in_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3258 { "src_bytes_out_rate", smp_fetch_sc_bytes_out_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3259 { "src_clr_gpc0", smp_fetch_sc_clr_gpc0, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3260 { "src_conn_cnt", smp_fetch_sc_conn_cnt, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3261 { "src_conn_cur", smp_fetch_sc_conn_cur, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3262 { "src_conn_rate", smp_fetch_sc_conn_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3263 { "src_get_gpc0", smp_fetch_sc_get_gpc0, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3264 { "src_gpc0_rate", smp_fetch_sc_gpc0_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3265 { "src_http_err_cnt", smp_fetch_sc_http_err_cnt, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3266 { "src_http_err_rate", smp_fetch_sc_http_err_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3267 { "src_http_req_cnt", smp_fetch_sc_http_req_cnt, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3268 { "src_http_req_rate", smp_fetch_sc_http_req_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3269 { "src_inc_gpc0", smp_fetch_sc_inc_gpc0, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3270 { "src_kbytes_in", smp_fetch_sc_kbytes_in, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3271 { "src_kbytes_out", smp_fetch_sc_kbytes_out, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3272 { "src_sess_cnt", smp_fetch_sc_sess_cnt, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3273 { "src_sess_rate", smp_fetch_sc_sess_rate, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3274 { "src_updt_conn_cnt", smp_fetch_src_updt_conn_cnt, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_L4CLI, },
3275 { "table_avl", smp_fetch_table_avl, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
3276 { "table_cnt", smp_fetch_table_cnt, ARG1(1,TAB), NULL, SMP_T_UINT, SMP_USE_INTRN, },
Willy Tarreau281c7992013-01-08 01:23:27 +01003277 { /* END */ },
3278}};
3279
Willy Tarreau8b22a712010-06-18 17:46:06 +02003280__attribute__((constructor))
3281static void __session_init(void)
3282{
Willy Tarreau281c7992013-01-08 01:23:27 +01003283 sample_register_fetches(&smp_fetch_keywords);
Willy Tarreau8b22a712010-06-18 17:46:06 +02003284 acl_register_keywords(&acl_kws);
3285}
3286
Willy Tarreaubaaee002006-06-26 02:48:02 +02003287/*
3288 * Local variables:
3289 * c-indent-level: 8
3290 * c-basic-offset: 8
3291 * End:
3292 */