blob: 0ccd9959dfdeedf0a77f7a8adf99c885c01f4302 [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 Tarreau81f9aa32010-06-01 17:45:26 +02004 * Copyright 2000-2010 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 Tarreau7c669d72008-06-20 15:04:11 +020018#include <common/debug.h>
Willy Tarreau2dd0d472006-06-29 17:53:05 +020019#include <common/memory.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020020
Willy Tarreaubaaee002006-06-26 02:48:02 +020021#include <types/capture.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010022#include <types/global.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020023
Willy Tarreau1d0dfb12009-07-07 15:10:31 +020024#include <proto/acl.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010025#include <proto/backend.h>
Willy Tarreau7341d942007-05-13 19:56:02 +020026#include <proto/buffers.h>
Krzysztof Piotr Oledzki97f07b82009-12-15 22:31:24 +010027#include <proto/checks.h>
Willy Tarreau5ca791d2009-08-16 19:06:42 +020028#include <proto/dumpstats.h>
Willy Tarreau8d5d7f22007-01-21 19:16:41 +010029#include <proto/hdr_idx.h>
Willy Tarreau332f8bf2007-05-13 21:36:56 +020030#include <proto/log.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020031#include <proto/session.h>
Emeric Brun1d33b292010-01-04 15:47:17 +010032#include <proto/pattern.h>
Willy Tarreau3eba98a2009-01-25 13:56:13 +010033#include <proto/pipe.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010034#include <proto/proto_http.h>
35#include <proto/proto_tcp.h>
Willy Tarreau1d0dfb12009-07-07 15:10:31 +020036#include <proto/proxy.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020037#include <proto/queue.h>
Willy Tarreau7f062c42009-03-05 18:43:00 +010038#include <proto/server.h>
Emeric Brun1d33b292010-01-04 15:47:17 +010039#include <proto/stick_table.h>
Willy Tarreau55a8d0e2008-11-30 18:47:21 +010040#include <proto/stream_interface.h>
41#include <proto/stream_sock.h>
42#include <proto/task.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020043
Willy Tarreauc6ca1a02007-05-13 19:43:47 +020044struct pool_head *pool2_session;
Willy Tarreauf54f8bd2008-11-23 19:53:55 +010045struct list sessions;
Willy Tarreaubaaee002006-06-26 02:48:02 +020046
Willy Tarreau81f9aa32010-06-01 17:45:26 +020047/* This function is called from the protocol layer accept() in order to instanciate
48 * a new session on behalf of a given listener and frontend. It returns a positive
49 * value upon success, 0 if the connection needs to be closed and ignored, or a
50 * negative value upon critical failure.
51 */
52int session_accept(struct listener *l, int cfd, struct sockaddr_storage *addr)
53{
54 struct proxy *p = l->frontend;
55 struct session *s;
56 struct http_txn *txn;
57 struct task *t;
58
59 if (unlikely((s = pool_alloc2(pool2_session)) == NULL)) {
60 Alert("out of memory in event_accept().\n");
61 goto out_close;
62 }
63
64 /* minimum session initialization required for monitor mode below */
65 s->flags = 0;
66 s->logs.logwait = p->to_log;
67
68 /* if this session comes from a known monitoring system, we want to ignore
69 * it as soon as possible, which means closing it immediately for TCP, but
70 * cleanly.
71 */
72 if (unlikely((l->options & LI_O_CHK_MONNET) &&
73 addr->ss_family == AF_INET &&
74 (((struct sockaddr_in *)addr)->sin_addr.s_addr & p->mon_mask.s_addr) == p->mon_net.s_addr)) {
75 if (p->mode == PR_MODE_TCP) {
76 pool_free2(pool2_session, s);
77 return 0;
78 }
79 s->flags |= SN_MONITOR;
80 s->logs.logwait = 0;
81 }
82
83 /* OK, we're keeping the session, so let's properly initialize the session */
84 LIST_ADDQ(&sessions, &s->list);
85 LIST_INIT(&s->back_refs);
86
87 if (unlikely((t = task_new()) == NULL)) { /* disable this proxy for a while */
88 Alert("out of memory in event_accept().\n");
89 goto out_free_session;
90 }
91
92 s->term_trace = 0;
93 s->cli_addr = *addr;
94 s->logs.accept_date = date; /* user-visible date for logging */
95 s->logs.tv_accept = now; /* corrected date for internal use */
96 s->uniq_id = totalconn;
97 proxy_inc_fe_ctr(l, p); /* note: cum_beconn will be increased once assigned */
98
99 t->process = l->handler;
100 t->context = s;
101 t->nice = l->nice;
102 t->expire = TICK_ETERNITY;
103
104 s->task = t;
105 s->listener = l;
106
107 /* Note: initially, the session's backend points to the frontend.
108 * This changes later when switching rules are executed or
109 * when the default backend is assigned.
110 */
111 s->be = s->fe = p;
112 s->req = s->rep = NULL; /* will be allocated later */
113
114 /* now evaluate the tcp-request layer4 rules. Since we expect to be able
115 * to abort right here as soon as possible, we check the rules before
116 * even initializing the stream interfaces.
117 */
118 if ((l->options & LI_O_TCP_RULES) && !tcp_exec_req_rules(s)) {
119 task_free(t);
120 LIST_DEL(&s->list);
121 pool_free2(pool2_session, s);
122 /* let's do a no-linger now to close with a single RST. */
123 setsockopt(cfd, SOL_SOCKET, SO_LINGER, (struct linger *) &nolinger, sizeof(struct linger));
124 return 0;
125 }
126
127 /* this part should be common with other protocols */
128 s->si[0].fd = cfd;
129 s->si[0].owner = t;
130 s->si[0].state = s->si[0].prev_state = SI_ST_EST;
131 s->si[0].err_type = SI_ET_NONE;
132 s->si[0].err_loc = NULL;
133 s->si[0].connect = NULL;
134 s->si[0].iohandler = NULL;
135 s->si[0].exp = TICK_ETERNITY;
136 s->si[0].flags = SI_FL_NONE;
137
138 if (likely(s->fe->options2 & PR_O2_INDEPSTR))
139 s->si[0].flags |= SI_FL_INDEP_STR;
140
141 if (addr->ss_family == AF_INET || addr->ss_family == AF_INET6)
142 s->si[0].flags = SI_FL_CAP_SPLTCP; /* TCP/TCPv6 splicing possible */
143
144 /* add the various callbacks */
145 stream_sock_prepare_interface(&s->si[0]);
146
147 /* pre-initialize the other side's stream interface to an INIT state. The
148 * callbacks will be initialized before attempting to connect.
149 */
150 s->si[1].fd = -1; /* just to help with debugging */
151 s->si[1].owner = t;
152 s->si[1].state = s->si[1].prev_state = SI_ST_INI;
153 s->si[1].err_type = SI_ET_NONE;
154 s->si[1].err_loc = NULL;
155 s->si[1].connect = NULL;
156 s->si[1].iohandler = NULL;
157 s->si[1].shutr = stream_int_shutr;
158 s->si[1].shutw = stream_int_shutw;
159 s->si[1].exp = TICK_ETERNITY;
160 s->si[1].flags = SI_FL_NONE;
161
162 if (likely(s->fe->options2 & PR_O2_INDEPSTR))
163 s->si[1].flags |= SI_FL_INDEP_STR;
164
165 s->srv = s->prev_srv = s->srv_conn = NULL;
166 s->pend_pos = NULL;
167
168 /* init store persistence */
169 s->store_count = 0;
170
171 /* Adjust some socket options */
172 if (unlikely(fcntl(cfd, F_SETFL, O_NONBLOCK) == -1)) {
173 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
174 goto out_free_task;
175 }
176
177 txn = &s->txn;
178 /* Those variables will be checked and freed if non-NULL in
179 * session.c:session_free(). It is important that they are
180 * properly initialized.
181 */
182 txn->sessid = NULL;
183 txn->srv_cookie = NULL;
184 txn->cli_cookie = NULL;
185 txn->uri = NULL;
186 txn->req.cap = NULL;
187 txn->rsp.cap = NULL;
188 txn->hdr_idx.v = NULL;
189 txn->hdr_idx.size = txn->hdr_idx.used = 0;
190
191 if (unlikely((s->req = pool_alloc2(pool2_buffer)) == NULL))
192 goto out_free_task; /* no memory */
193
194 if (unlikely((s->rep = pool_alloc2(pool2_buffer)) == NULL))
195 goto out_free_req; /* no memory */
196
197 /* initialize the request buffer */
198 s->req->size = global.tune.bufsize;
199 buffer_init(s->req);
200 s->req->prod = &s->si[0];
201 s->req->cons = &s->si[1];
202 s->si[0].ib = s->si[1].ob = s->req;
203 s->req->flags |= BF_READ_ATTACHED; /* the producer is already connected */
204
205 /* activate default analysers enabled for this listener */
206 s->req->analysers = l->analysers;
207
208 s->req->wto = TICK_ETERNITY;
209 s->req->rto = TICK_ETERNITY;
210 s->req->rex = TICK_ETERNITY;
211 s->req->wex = TICK_ETERNITY;
212 s->req->analyse_exp = TICK_ETERNITY;
213
214 /* initialize response buffer */
215 s->rep->size = global.tune.bufsize;
216 buffer_init(s->rep);
217 s->rep->prod = &s->si[1];
218 s->rep->cons = &s->si[0];
219 s->si[0].ob = s->si[1].ib = s->rep;
220 s->rep->analysers = 0;
221
222 s->rep->rto = TICK_ETERNITY;
223 s->rep->wto = TICK_ETERNITY;
224 s->rep->rex = TICK_ETERNITY;
225 s->rep->wex = TICK_ETERNITY;
226 s->rep->analyse_exp = TICK_ETERNITY;
227
228 /* finish initialization of the accepted file descriptor */
229 fd_insert(cfd);
230 fdtab[cfd].owner = &s->si[0];
231 fdtab[cfd].state = FD_STREADY;
232 fdtab[cfd].flags = 0;
233 fdtab[cfd].cb[DIR_RD].f = l->proto->read;
234 fdtab[cfd].cb[DIR_RD].b = s->req;
235 fdtab[cfd].cb[DIR_WR].f = l->proto->write;
236 fdtab[cfd].cb[DIR_WR].b = s->rep;
237 fdinfo[cfd].peeraddr = (struct sockaddr *)&s->cli_addr;
238 fdinfo[cfd].peerlen = sizeof(s->cli_addr);
239 EV_FD_SET(cfd, DIR_RD);
240
241 if (p->accept) {
242 int ret = p->accept(s);
243 if (unlikely(ret < 0))
244 goto out_free_rep;
245
246 if (unlikely(ret == 0)) {
247 /* work is finished, we can release everything (eg: monitoring) */
248 pool_free2(pool2_buffer, s->rep);
249 pool_free2(pool2_buffer, s->req);
250 task_free(t);
251 LIST_DEL(&s->list);
252 pool_free2(pool2_session, s);
253 return 0;
254 }
255 }
256
257 /* it is important not to call the wakeup function directly but to
258 * pass through task_wakeup(), because this one knows how to apply
259 * priorities to tasks.
260 */
261 task_wakeup(t, TASK_WOKEN_INIT);
262 return 1;
263
264 /* Error unrolling */
265 out_free_rep:
266 pool_free2(pool2_buffer, s->rep);
267 out_free_req:
268 pool_free2(pool2_buffer, s->req);
269 out_free_task:
270 task_free(t);
271 out_free_session:
272 LIST_DEL(&s->list);
273 pool_free2(pool2_session, s);
274 out_close:
275 return -1;
276}
277
Willy Tarreaubaaee002006-06-26 02:48:02 +0200278/*
279 * frees the context associated to a session. It must have been removed first.
280 */
281void session_free(struct session *s)
282{
Willy Tarreau4dbc4a22007-03-03 16:23:22 +0100283 struct http_txn *txn = &s->txn;
Willy Tarreau632f5a72007-07-11 10:42:35 +0200284 struct proxy *fe = s->fe;
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100285 struct bref *bref, *back;
Willy Tarreaua4cda672010-06-06 18:28:49 +0200286 int i;
Willy Tarreau0f7562b2007-01-07 15:46:13 +0100287
Willy Tarreaubaaee002006-06-26 02:48:02 +0200288 if (s->pend_pos)
289 pendconn_free(s->pend_pos);
Willy Tarreau922a8062008-12-04 09:33:58 +0100290
Willy Tarreau1e62de62008-11-11 20:20:02 +0100291 if (s->srv) { /* there may be requests left pending in queue */
292 if (s->flags & SN_CURR_SESS) {
293 s->flags &= ~SN_CURR_SESS;
294 s->srv->cur_sess--;
295 }
Willy Tarreau922a8062008-12-04 09:33:58 +0100296 if (may_dequeue_tasks(s->srv, s->be))
297 process_srv_queue(s->srv);
Willy Tarreau1e62de62008-11-11 20:20:02 +0100298 }
Willy Tarreau922a8062008-12-04 09:33:58 +0100299
Willy Tarreau7c669d72008-06-20 15:04:11 +0200300 if (unlikely(s->srv_conn)) {
301 /* the session still has a reserved slot on a server, but
302 * it should normally be only the same as the one above,
303 * so this should not happen in fact.
304 */
305 sess_change_server(s, NULL);
306 }
307
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100308 if (s->req->pipe)
309 put_pipe(s->req->pipe);
Willy Tarreau259de1b2009-01-18 21:56:21 +0100310
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100311 if (s->rep->pipe)
312 put_pipe(s->rep->pipe);
Willy Tarreau259de1b2009-01-18 21:56:21 +0100313
Willy Tarreau48d63db2008-08-03 17:41:33 +0200314 pool_free2(pool2_buffer, s->req);
315 pool_free2(pool2_buffer, s->rep);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200316
Willy Tarreau46023632010-01-07 22:51:47 +0100317 http_end_txn(s);
318
Willy Tarreaua4cda672010-06-06 18:28:49 +0200319 for (i = 0; i < s->store_count; i++) {
320 if (!s->store[i].ts)
321 continue;
322 stksess_free(s->store[i].table, s->store[i].ts);
323 s->store[i].ts = NULL;
324 }
325
Willy Tarreau92fb9832007-10-16 17:34:28 +0200326 if (fe) {
Willy Tarreau48d63db2008-08-03 17:41:33 +0200327 pool_free2(fe->hdr_idx_pool, txn->hdr_idx.v);
Willy Tarreau46023632010-01-07 22:51:47 +0100328 pool_free2(fe->rsp_cap_pool, txn->rsp.cap);
329 pool_free2(fe->req_cap_pool, txn->req.cap);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200330 }
Willy Tarreau0937bc42009-12-22 15:03:09 +0100331
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100332 list_for_each_entry_safe(bref, back, &s->back_refs, users) {
Willy Tarreaufd3828e2009-02-22 15:17:24 +0100333 /* we have to unlink all watchers. We must not relink them if
334 * this session was the last one in the list.
335 */
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100336 LIST_DEL(&bref->users);
Willy Tarreaufd3828e2009-02-22 15:17:24 +0100337 LIST_INIT(&bref->users);
338 if (s->list.n != &sessions)
339 LIST_ADDQ(&LIST_ELEM(s->list.n, struct session *, list)->back_refs, &bref->users);
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100340 bref->ref = s->list.n;
341 }
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100342 LIST_DEL(&s->list);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200343 pool_free2(pool2_session, s);
Willy Tarreau632f5a72007-07-11 10:42:35 +0200344
345 /* We may want to free the maximum amount of pools if the proxy is stopping */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200346 if (fe && unlikely(fe->state == PR_STSTOPPED)) {
Willy Tarreau48d63db2008-08-03 17:41:33 +0200347 pool_flush2(pool2_buffer);
348 pool_flush2(fe->hdr_idx_pool);
349 pool_flush2(pool2_requri);
350 pool_flush2(pool2_capture);
351 pool_flush2(pool2_session);
352 pool_flush2(fe->req_cap_pool);
353 pool_flush2(fe->rsp_cap_pool);
Willy Tarreau632f5a72007-07-11 10:42:35 +0200354 }
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200355}
356
357
358/* perform minimal intializations, report 0 in case of error, 1 if OK. */
359int init_session()
360{
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100361 LIST_INIT(&sessions);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200362 pool2_session = create_pool("session", sizeof(struct session), MEM_F_SHARED);
363 return pool2_session != NULL;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200364}
365
Willy Tarreau30e71012007-11-26 20:15:35 +0100366void session_process_counters(struct session *s)
367{
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100368 unsigned long long bytes;
369
Willy Tarreau30e71012007-11-26 20:15:35 +0100370 if (s->req) {
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100371 bytes = s->req->total - s->logs.bytes_in;
Willy Tarreau30e71012007-11-26 20:15:35 +0100372 s->logs.bytes_in = s->req->total;
373 if (bytes) {
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200374 s->fe->counters.bytes_in += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100375
Willy Tarreau30e71012007-11-26 20:15:35 +0100376 if (s->be != s->fe)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200377 s->be->counters.bytes_in += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100378
Willy Tarreau30e71012007-11-26 20:15:35 +0100379 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200380 s->srv->counters.bytes_in += bytes;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200381
382 if (s->listener->counters)
383 s->listener->counters->bytes_in += bytes;
Willy Tarreau30e71012007-11-26 20:15:35 +0100384 }
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100385 }
386
Willy Tarreau30e71012007-11-26 20:15:35 +0100387 if (s->rep) {
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100388 bytes = s->rep->total - s->logs.bytes_out;
Willy Tarreau30e71012007-11-26 20:15:35 +0100389 s->logs.bytes_out = s->rep->total;
390 if (bytes) {
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200391 s->fe->counters.bytes_out += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100392
Willy Tarreau30e71012007-11-26 20:15:35 +0100393 if (s->be != s->fe)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200394 s->be->counters.bytes_out += bytes;
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100395
Willy Tarreau30e71012007-11-26 20:15:35 +0100396 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200397 s->srv->counters.bytes_out += bytes;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +0200398
399 if (s->listener->counters)
400 s->listener->counters->bytes_out += bytes;
Willy Tarreau30e71012007-11-26 20:15:35 +0100401 }
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100402 }
403}
Willy Tarreaubaaee002006-06-26 02:48:02 +0200404
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100405/* This function is called with (si->state == SI_ST_CON) meaning that a
406 * connection was attempted and that the file descriptor is already allocated.
407 * We must check for establishment, error and abort. Possible output states
408 * are SI_ST_EST (established), SI_ST_CER (error), SI_ST_DIS (abort), and
409 * SI_ST_CON (no change). The function returns 0 if it switches to SI_ST_CER,
410 * otherwise 1.
411 */
412int sess_update_st_con_tcp(struct session *s, struct stream_interface *si)
413{
414 struct buffer *req = si->ob;
415 struct buffer *rep = si->ib;
416
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100417 /* If we got an error, or if nothing happened and the connection timed
418 * out, we must give up. The CER state handler will take care of retry
419 * attempts and error reports.
420 */
421 if (unlikely(si->flags & (SI_FL_EXP|SI_FL_ERR))) {
Willy Tarreau127334e2009-03-28 10:47:26 +0100422 si->exp = TICK_ETERNITY;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100423 si->state = SI_ST_CER;
Willy Tarreaudc340a92009-06-28 23:10:19 +0200424 si->flags &= ~SI_FL_CAP_SPLICE;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100425 fd_delete(si->fd);
426
427 if (si->err_type)
428 return 0;
429
430 si->err_loc = s->srv;
431 if (si->flags & SI_FL_ERR)
432 si->err_type = SI_ET_CONN_ERR;
433 else
434 si->err_type = SI_ET_CONN_TO;
435 return 0;
436 }
437
438 /* OK, maybe we want to abort */
Willy Tarreau418fd472009-09-06 21:37:23 +0200439 if (unlikely((rep->flags & BF_SHUTW) ||
440 ((req->flags & BF_SHUTW_NOW) && /* FIXME: this should not prevent a connection from establishing */
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200441 (((req->flags & (BF_OUT_EMPTY|BF_WRITE_ACTIVITY)) == BF_OUT_EMPTY) ||
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100442 s->be->options & PR_O_ABRT_CLOSE)))) {
443 /* give up */
444 si->shutw(si);
445 si->err_type |= SI_ET_CONN_ABRT;
446 si->err_loc = s->srv;
Willy Tarreaudc340a92009-06-28 23:10:19 +0200447 si->flags &= ~SI_FL_CAP_SPLICE;
Willy Tarreau84455332009-03-15 22:34:05 +0100448 if (s->srv_error)
449 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100450 return 1;
451 }
452
453 /* we need to wait a bit more if there was no activity either */
454 if (!(req->flags & BF_WRITE_ACTIVITY))
455 return 1;
456
457 /* OK, this means that a connection succeeded. The caller will be
458 * responsible for handling the transition from CON to EST.
459 */
460 s->logs.t_connect = tv_ms_elapsed(&s->logs.tv_accept, &now);
Willy Tarreau127334e2009-03-28 10:47:26 +0100461 si->exp = TICK_ETERNITY;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100462 si->state = SI_ST_EST;
463 si->err_type = SI_ET_NONE;
464 si->err_loc = NULL;
465 return 1;
466}
467
468/* This function is called with (si->state == SI_ST_CER) meaning that a
469 * previous connection attempt has failed and that the file descriptor
470 * has already been released. Possible causes include asynchronous error
471 * notification and time out. Possible output states are SI_ST_CLO when
472 * retries are exhausted, SI_ST_TAR when a delay is wanted before a new
473 * connection attempt, SI_ST_ASS when it's wise to retry on the same server,
474 * and SI_ST_REQ when an immediate redispatch is wanted. The buffers are
475 * marked as in error state. It returns 0.
476 */
477int sess_update_st_cer(struct session *s, struct stream_interface *si)
478{
479 /* we probably have to release last session from the server */
480 if (s->srv) {
Krzysztof Piotr Oledzki97f07b82009-12-15 22:31:24 +0100481 health_adjust(s->srv, HANA_STATUS_L4_ERR);
482
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100483 if (s->flags & SN_CURR_SESS) {
484 s->flags &= ~SN_CURR_SESS;
485 s->srv->cur_sess--;
486 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100487 }
488
489 /* ensure that we have enough retries left */
Willy Tarreauee28de02010-06-01 09:51:00 +0200490 si->conn_retries--;
491 if (si->conn_retries < 0) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100492 if (!si->err_type) {
493 si->err_type = SI_ET_CONN_ERR;
494 si->err_loc = s->srv;
495 }
496
497 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200498 s->srv->counters.failed_conns++;
499 s->be->counters.failed_conns++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100500 if (may_dequeue_tasks(s->srv, s->be))
501 process_srv_queue(s->srv);
502
503 /* shutw is enough so stop a connecting socket */
504 si->shutw(si);
505 si->ob->flags |= BF_WRITE_ERROR;
506 si->ib->flags |= BF_READ_ERROR;
507
508 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100509 if (s->srv_error)
510 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100511 return 0;
512 }
513
514 /* If the "redispatch" option is set on the backend, we are allowed to
515 * retry on another server for the last retry. In order to achieve this,
516 * we must mark the session unassigned, and eventually clear the DIRECT
517 * bit to ignore any persistence cookie. We won't count a retry nor a
518 * redispatch yet, because this will depend on what server is selected.
519 */
Willy Tarreauee28de02010-06-01 09:51:00 +0200520 if (s->srv && si->conn_retries == 0 &&
Willy Tarreau4de91492010-01-22 19:10:05 +0100521 s->be->options & PR_O_REDISP && !(s->flags & SN_FORCE_PRST)) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100522 if (may_dequeue_tasks(s->srv, s->be))
523 process_srv_queue(s->srv);
524
525 s->flags &= ~(SN_DIRECT | SN_ASSIGNED | SN_ADDR_SET);
526 s->prev_srv = s->srv;
527 si->state = SI_ST_REQ;
528 } else {
529 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200530 s->srv->counters.retries++;
531 s->be->counters.retries++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100532 si->state = SI_ST_ASS;
533 }
534
535 if (si->flags & SI_FL_ERR) {
536 /* The error was an asynchronous connection error, and we will
537 * likely have to retry connecting to the same server, most
538 * likely leading to the same result. To avoid this, we wait
539 * one second before retrying.
540 */
541
542 if (!si->err_type)
543 si->err_type = SI_ET_CONN_ERR;
544
545 si->state = SI_ST_TAR;
546 si->exp = tick_add(now_ms, MS_TO_TICKS(1000));
547 return 0;
548 }
549 return 0;
550}
551
552/*
553 * This function handles the transition between the SI_ST_CON state and the
Willy Tarreau85e7d002010-05-31 11:57:51 +0200554 * SI_ST_EST state. It must only be called after switching from SI_ST_CON (or
555 * SI_ST_INI) to SI_ST_EST, but only when a ->connect function is defined.
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100556 */
557void sess_establish(struct session *s, struct stream_interface *si)
558{
559 struct buffer *req = si->ob;
560 struct buffer *rep = si->ib;
561
Krzysztof Piotr Oledzki97f07b82009-12-15 22:31:24 +0100562 if (s->srv)
563 health_adjust(s->srv, HANA_STATUS_L4_OK);
564
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100565 if (s->be->mode == PR_MODE_TCP) { /* let's allow immediate data connection in this case */
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100566 /* if the user wants to log as soon as possible, without counting
567 * bytes from the server, then this is the right moment. */
568 if (s->fe->to_log && !(s->logs.logwait & LW_BYTES)) {
569 s->logs.t_close = s->logs.t_connect; /* to get a valid end date */
Willy Tarreaua5555ec2008-11-30 19:02:32 +0100570 s->do_log(s);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100571 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100572 }
573 else {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100574 s->txn.rsp.msg_state = HTTP_MSG_RPBEFORE;
575 /* reset hdr_idx which was already initialized by the request.
576 * right now, the http parser does it.
577 * hdr_idx_init(&s->txn.hdr_idx);
578 */
579 }
580
Willy Tarreau4e5b8282009-08-16 22:57:50 +0200581 rep->analysers |= s->fe->fe_rsp_ana | s->be->be_rsp_ana;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100582 rep->flags |= BF_READ_ATTACHED; /* producer is now attached */
Willy Tarreaud04e8582010-05-31 12:31:35 +0200583 if (si->connect) {
584 /* real connections have timeouts */
585 req->wto = s->be->timeout.server;
586 rep->rto = s->be->timeout.server;
587 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100588 req->wex = TICK_ETERNITY;
589}
590
591/* Update stream interface status for input states SI_ST_ASS, SI_ST_QUE, SI_ST_TAR.
592 * Other input states are simply ignored.
593 * Possible output states are SI_ST_CLO, SI_ST_TAR, SI_ST_ASS, SI_ST_REQ, SI_ST_CON.
594 * Flags must have previously been updated for timeouts and other conditions.
595 */
596void sess_update_stream_int(struct session *s, struct stream_interface *si)
597{
598 DPRINTF(stderr,"[%u] %s: sess=%p rq=%p, rp=%p, exp(r,w)=%u,%u rqf=%08x rpf=%08x rql=%d rpl=%d cs=%d ss=%d\n",
599 now_ms, __FUNCTION__,
600 s,
601 s->req, s->rep,
602 s->req->rex, s->rep->wex,
603 s->req->flags, s->rep->flags,
604 s->req->l, s->rep->l, s->rep->cons->state, s->req->cons->state);
605
606 if (si->state == SI_ST_ASS) {
607 /* Server assigned to connection request, we have to try to connect now */
608 int conn_err;
609
610 conn_err = connect_server(s);
611 if (conn_err == SN_ERR_NONE) {
612 /* state = SI_ST_CON now */
Willy Tarreau8f6457c2008-12-01 00:08:28 +0100613 if (s->srv)
Willy Tarreau7f062c42009-03-05 18:43:00 +0100614 srv_inc_sess_ctr(s->srv);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100615 return;
616 }
617
618 /* We have received a synchronous error. We might have to
619 * abort, retry immediately or redispatch.
620 */
621 if (conn_err == SN_ERR_INTERNAL) {
622 if (!si->err_type) {
623 si->err_type = SI_ET_CONN_OTHER;
624 si->err_loc = s->srv;
625 }
626
627 if (s->srv)
Willy Tarreau7f062c42009-03-05 18:43:00 +0100628 srv_inc_sess_ctr(s->srv);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100629 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200630 s->srv->counters.failed_conns++;
631 s->be->counters.failed_conns++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100632
633 /* release other sessions waiting for this server */
634 if (may_dequeue_tasks(s->srv, s->be))
635 process_srv_queue(s->srv);
636
637 /* Failed and not retryable. */
638 si->shutr(si);
639 si->shutw(si);
640 si->ob->flags |= BF_WRITE_ERROR;
641
642 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
643
644 /* no session was ever accounted for this server */
645 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100646 if (s->srv_error)
647 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100648 return;
649 }
650
651 /* We are facing a retryable error, but we don't want to run a
652 * turn-around now, as the problem is likely a source port
653 * allocation problem, so we want to retry now.
654 */
655 si->state = SI_ST_CER;
656 si->flags &= ~SI_FL_ERR;
657 sess_update_st_cer(s, si);
658 /* now si->state is one of SI_ST_CLO, SI_ST_TAR, SI_ST_ASS, SI_ST_REQ */
659 return;
660 }
661 else if (si->state == SI_ST_QUE) {
662 /* connection request was queued, check for any update */
663 if (!s->pend_pos) {
664 /* The connection is not in the queue anymore. Either
665 * we have a server connection slot available and we
666 * go directly to the assigned state, or we need to
667 * load-balance first and go to the INI state.
668 */
669 si->exp = TICK_ETERNITY;
670 if (unlikely(!(s->flags & SN_ASSIGNED)))
671 si->state = SI_ST_REQ;
672 else {
673 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
674 si->state = SI_ST_ASS;
675 }
676 return;
677 }
678
679 /* Connection request still in queue... */
680 if (si->flags & SI_FL_EXP) {
681 /* ... and timeout expired */
682 si->exp = TICK_ETERNITY;
683 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
684 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +0200685 s->srv->counters.failed_conns++;
686 s->be->counters.failed_conns++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100687 si->shutr(si);
688 si->shutw(si);
689 si->ob->flags |= BF_WRITE_TIMEOUT;
690 if (!si->err_type)
691 si->err_type = SI_ET_QUEUE_TO;
692 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100693 if (s->srv_error)
694 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100695 return;
696 }
697
698 /* Connection remains in queue, check if we have to abort it */
Willy Tarreau418fd472009-09-06 21:37:23 +0200699 if ((si->ob->flags & (BF_READ_ERROR)) ||
700 ((si->ob->flags & BF_SHUTW_NOW) && /* empty and client aborted */
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200701 (si->ob->flags & BF_OUT_EMPTY || s->be->options & PR_O_ABRT_CLOSE))) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100702 /* give up */
703 si->exp = TICK_ETERNITY;
704 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
705 si->shutr(si);
706 si->shutw(si);
707 si->err_type |= SI_ET_QUEUE_ABRT;
708 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100709 if (s->srv_error)
710 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100711 return;
712 }
713
714 /* Nothing changed */
715 return;
716 }
717 else if (si->state == SI_ST_TAR) {
718 /* Connection request might be aborted */
Willy Tarreau418fd472009-09-06 21:37:23 +0200719 if ((si->ob->flags & (BF_READ_ERROR)) ||
720 ((si->ob->flags & BF_SHUTW_NOW) && /* empty and client aborted */
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200721 (si->ob->flags & BF_OUT_EMPTY || s->be->options & PR_O_ABRT_CLOSE))) {
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100722 /* give up */
723 si->exp = TICK_ETERNITY;
724 si->shutr(si);
725 si->shutw(si);
726 si->err_type |= SI_ET_CONN_ABRT;
727 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100728 if (s->srv_error)
729 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100730 return;
731 }
732
733 if (!(si->flags & SI_FL_EXP))
734 return; /* still in turn-around */
735
736 si->exp = TICK_ETERNITY;
737
738 /* we keep trying on the same server as long as the session is
739 * marked "assigned".
740 * FIXME: Should we force a redispatch attempt when the server is down ?
741 */
742 if (s->flags & SN_ASSIGNED)
743 si->state = SI_ST_ASS;
744 else
745 si->state = SI_ST_REQ;
746 return;
747 }
748}
749
750/* This function initiates a server connection request on a stream interface
751 * already in SI_ST_REQ state. Upon success, the state goes to SI_ST_ASS,
752 * indicating that a server has been assigned. It may also return SI_ST_QUE,
753 * or SI_ST_CLO upon error.
754 */
755static void sess_prepare_conn_req(struct session *s, struct stream_interface *si) {
756 DPRINTF(stderr,"[%u] %s: sess=%p rq=%p, rp=%p, exp(r,w)=%u,%u rqf=%08x rpf=%08x rql=%d rpl=%d cs=%d ss=%d\n",
757 now_ms, __FUNCTION__,
758 s,
759 s->req, s->rep,
760 s->req->rex, s->rep->wex,
761 s->req->flags, s->rep->flags,
762 s->req->l, s->rep->l, s->rep->cons->state, s->req->cons->state);
763
764 if (si->state != SI_ST_REQ)
765 return;
766
767 /* Try to assign a server */
768 if (srv_redispatch_connect(s) != 0) {
769 /* We did not get a server. Either we queued the
770 * connection request, or we encountered an error.
771 */
772 if (si->state == SI_ST_QUE)
773 return;
774
775 /* we did not get any server, let's check the cause */
776 si->shutr(si);
777 si->shutw(si);
778 si->ob->flags |= BF_WRITE_ERROR;
779 if (!si->err_type)
780 si->err_type = SI_ET_CONN_OTHER;
781 si->state = SI_ST_CLO;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100782 if (s->srv_error)
783 s->srv_error(s, si);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +0100784 return;
785 }
786
787 /* The server is assigned */
788 s->logs.t_queue = tv_ms_elapsed(&s->logs.tv_accept, &now);
789 si->state = SI_ST_ASS;
790}
791
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200792/* This stream analyser checks the switching rules and changes the backend
Willy Tarreau4de91492010-01-22 19:10:05 +0100793 * if appropriate. The default_backend rule is also considered, then the
794 * target backend's forced persistence rules are also evaluated last if any.
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200795 * It returns 1 if the processing can continue on next analysers, or zero if it
796 * either needs more data or wants to immediately abort the request.
797 */
798int process_switching_rules(struct session *s, struct buffer *req, int an_bit)
799{
Cyril Bonté47fdd8e2010-04-25 00:00:51 +0200800 struct persist_rule *prst_rule;
Willy Tarreau4de91492010-01-22 19:10:05 +0100801
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200802 req->analysers &= ~an_bit;
803 req->analyse_exp = TICK_ETERNITY;
804
805 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bl=%d analysers=%02x\n",
806 now_ms, __FUNCTION__,
807 s,
808 req,
809 req->rex, req->wex,
810 req->flags,
811 req->l,
812 req->analysers);
813
814 /* now check whether we have some switching rules for this request */
815 if (!(s->flags & SN_BE_ASSIGNED)) {
816 struct switching_rule *rule;
817
818 list_for_each_entry(rule, &s->fe->switching_rules, list) {
819 int ret;
820
821 ret = acl_exec_cond(rule->cond, s->fe, s, &s->txn, ACL_DIR_REQ);
822 ret = acl_pass(ret);
823 if (rule->cond->pol == ACL_COND_UNLESS)
824 ret = !ret;
825
826 if (ret) {
Willy Tarreaubedb9ba2009-07-12 08:27:39 +0200827 if (!session_set_backend(s, rule->be.backend))
828 goto sw_failed;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200829 break;
830 }
831 }
832
833 /* To ensure correct connection accounting on the backend, we
834 * have to assign one if it was not set (eg: a listen). This
835 * measure also takes care of correctly setting the default
836 * backend if any.
837 */
838 if (!(s->flags & SN_BE_ASSIGNED))
Willy Tarreaubedb9ba2009-07-12 08:27:39 +0200839 if (!session_set_backend(s, s->fe->defbe.be ? s->fe->defbe.be : s->be))
840 goto sw_failed;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200841 }
842
843 /* we don't want to run the HTTP filters again if the backend has not changed */
844 if (s->fe == s->be)
845 s->req->analysers &= ~AN_REQ_HTTP_PROCESS_BE;
846
Cyril Bonté47fdd8e2010-04-25 00:00:51 +0200847 /* 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 +0100848 * persistence rule, and report that in the session.
849 */
Cyril Bonté47fdd8e2010-04-25 00:00:51 +0200850 list_for_each_entry(prst_rule, &s->be->persist_rules, list) {
Willy Tarreau4de91492010-01-22 19:10:05 +0100851 int ret = 1;
852
853 if (prst_rule->cond) {
854 ret = acl_exec_cond(prst_rule->cond, s->be, s, &s->txn, ACL_DIR_REQ);
855 ret = acl_pass(ret);
856 if (prst_rule->cond->pol == ACL_COND_UNLESS)
857 ret = !ret;
858 }
859
860 if (ret) {
861 /* no rule, or the rule matches */
Cyril Bonté47fdd8e2010-04-25 00:00:51 +0200862 if (prst_rule->type == PERSIST_TYPE_FORCE) {
863 s->flags |= SN_FORCE_PRST;
864 } else {
865 s->flags |= SN_IGNORE_PRST;
866 }
Willy Tarreau4de91492010-01-22 19:10:05 +0100867 break;
868 }
869 }
870
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200871 return 1;
Willy Tarreaubedb9ba2009-07-12 08:27:39 +0200872
873 sw_failed:
874 /* immediately abort this request in case of allocation failure */
875 buffer_abort(s->req);
876 buffer_abort(s->rep);
877
878 if (!(s->flags & SN_ERR_MASK))
879 s->flags |= SN_ERR_RESOURCE;
880 if (!(s->flags & SN_FINST_MASK))
881 s->flags |= SN_FINST_R;
882
883 s->txn.status = 500;
884 s->req->analysers = 0;
885 s->req->analyse_exp = TICK_ETERNITY;
886 return 0;
Willy Tarreau1d0dfb12009-07-07 15:10:31 +0200887}
888
Emeric Brun1d33b292010-01-04 15:47:17 +0100889/* This stream analyser works on a request. It applies all sticking rules on
890 * it then returns 1. The data must already be present in the buffer otherwise
891 * they won't match. It always returns 1.
892 */
893int process_sticking_rules(struct session *s, struct buffer *req, int an_bit)
894{
895 struct proxy *px = s->be;
896 struct sticking_rule *rule;
897
898 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bl=%d analysers=%02x\n",
899 now_ms, __FUNCTION__,
900 s,
901 req,
902 req->rex, req->wex,
903 req->flags,
904 req->l,
905 req->analysers);
906
907 list_for_each_entry(rule, &px->sticking_rules, list) {
908 int ret = 1 ;
909 int i;
910
911 for (i = 0; i < s->store_count; i++) {
912 if (rule->table.t == s->store[i].table)
913 break;
914 }
915
916 if (i != s->store_count)
917 continue;
918
919 if (rule->cond) {
920 ret = acl_exec_cond(rule->cond, px, s, &s->txn, ACL_DIR_REQ);
921 ret = acl_pass(ret);
922 if (rule->cond->pol == ACL_COND_UNLESS)
923 ret = !ret;
924 }
925
926 if (ret) {
927 struct stktable_key *key;
928
929 key = pattern_process_key(px, s, &s->txn, PATTERN_FETCH_REQ, rule->expr, rule->table.t->type);
930 if (!key)
931 continue;
932
933 if (rule->flags & STK_IS_MATCH) {
934 struct stksess *ts;
935
936 if ((ts = stktable_lookup(rule->table.t, key)) != NULL) {
937 if (!(s->flags & SN_ASSIGNED)) {
938 struct eb32_node *node;
939
940 /* srv found in table */
941 node = eb32_lookup(&px->conf.used_server_id, ts->sid);
942 if (node) {
943 struct server *srv;
944
945 srv = container_of(node, struct server, conf.id);
Willy Tarreau4de91492010-01-22 19:10:05 +0100946 if ((srv->state & SRV_RUNNING) ||
947 (px->options & PR_O_PERSIST) ||
948 (s->flags & SN_FORCE_PRST)) {
Emeric Brun1d33b292010-01-04 15:47:17 +0100949 s->flags |= SN_DIRECT | SN_ASSIGNED;
950 s->srv = srv;
951 }
952 }
953 }
954 ts->expire = tick_add(now_ms, MS_TO_TICKS(rule->table.t->expire));
955 }
956 }
957 if (rule->flags & STK_IS_STORE) {
958 if (s->store_count < (sizeof(s->store) / sizeof(s->store[0]))) {
959 struct stksess *ts;
960
961 ts = stksess_new(rule->table.t, key);
962 if (ts) {
963 s->store[s->store_count].table = rule->table.t;
964 s->store[s->store_count++].ts = ts;
965 }
966 }
967 }
968 }
969 }
970
971 req->analysers &= ~an_bit;
972 req->analyse_exp = TICK_ETERNITY;
973 return 1;
974}
975
976/* This stream analyser works on a response. It applies all store rules on it
977 * then returns 1. The data must already be present in the buffer otherwise
978 * they won't match. It always returns 1.
979 */
980int process_store_rules(struct session *s, struct buffer *rep, int an_bit)
981{
982 struct proxy *px = s->be;
983 struct sticking_rule *rule;
984 int i;
985
986 DPRINTF(stderr,"[%u] %s: session=%p b=%p, exp(r,w)=%u,%u bf=%08x bl=%d analysers=%02x\n",
987 now_ms, __FUNCTION__,
988 s,
Willy Tarreau2e2b3eb2010-02-09 20:55:44 +0100989 rep,
990 rep->rex, rep->wex,
991 rep->flags,
992 rep->l,
993 rep->analysers);
Emeric Brun1d33b292010-01-04 15:47:17 +0100994
995 list_for_each_entry(rule, &px->storersp_rules, list) {
996 int ret = 1 ;
997 int storereqidx = -1;
998
999 for (i = 0; i < s->store_count; i++) {
1000 if (rule->table.t == s->store[i].table) {
1001 if (!(s->store[i].flags))
1002 storereqidx = i;
1003 break;
1004 }
1005 }
1006
1007 if ((i != s->store_count) && (storereqidx == -1))
1008 continue;
1009
1010 if (rule->cond) {
1011 ret = acl_exec_cond(rule->cond, px, s, &s->txn, ACL_DIR_RTR);
1012 ret = acl_pass(ret);
1013 if (rule->cond->pol == ACL_COND_UNLESS)
1014 ret = !ret;
1015 }
1016
1017 if (ret) {
1018 struct stktable_key *key;
1019
1020 key = pattern_process_key(px, s, &s->txn, PATTERN_FETCH_RTR, rule->expr, rule->table.t->type);
1021 if (!key)
1022 continue;
1023
1024 if (storereqidx != -1) {
1025 stksess_key(s->store[storereqidx].table, s->store[storereqidx].ts, key);
1026 s->store[storereqidx].flags = 1;
1027 }
1028 else if (s->store_count < (sizeof(s->store) / sizeof(s->store[0]))) {
1029 struct stksess *ts;
1030
1031 ts = stksess_new(rule->table.t, key);
1032 if (ts) {
1033 s->store[s->store_count].table = rule->table.t;
1034 s->store[s->store_count].flags = 1;
1035 s->store[s->store_count++].ts = ts;
1036 }
1037 }
1038 }
1039 }
1040
1041 /* process store request and store response */
1042 for (i = 0; i < s->store_count; i++) {
1043 if (stktable_store(s->store[i].table, s->store[i].ts, s->srv->puid) > 0) {
1044 stksess_free(s->store[i].table, s->store[i].ts);
1045 s->store[i].ts = NULL;
1046 }
1047 }
1048
1049 rep->analysers &= ~an_bit;
1050 rep->analyse_exp = TICK_ETERNITY;
1051 return 1;
1052}
1053
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001054/* This macro is very specific to the function below. See the comments in
1055 * process_session() below to understand the logic and the tests.
1056 */
1057#define UPDATE_ANALYSERS(real, list, back, flag) { \
1058 list = (((list) & ~(flag)) | ~(back)) & (real); \
1059 back = real; \
1060 if (!(list)) \
1061 break; \
1062 if (((list) ^ ((list) & ((list) - 1))) < (flag)) \
1063 continue; \
1064}
1065
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001066/* Processes the client, server, request and response jobs of a session task,
1067 * then puts it back to the wait queue in a clean state, or cleans up its
1068 * resources if it must be deleted. Returns in <next> the date the task wants
1069 * to be woken up, or TICK_ETERNITY. In order not to call all functions for
1070 * nothing too many times, the request and response buffers flags are monitored
1071 * and each function is called only if at least another function has changed at
1072 * least one flag it is interested in.
1073 */
Willy Tarreau26c25062009-03-08 09:38:41 +01001074struct task *process_session(struct task *t)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001075{
1076 struct session *s = t->context;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001077 unsigned int rqf_last, rpf_last;
Willy Tarreau576507f2010-01-07 00:09:04 +01001078 unsigned int req_ana_back;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001079
1080 //DPRINTF(stderr, "%s:%d: cs=%d ss=%d(%d) rqf=0x%08x rpf=0x%08x\n", __FUNCTION__, __LINE__,
1081 // s->si[0].state, s->si[1].state, s->si[1].err_type, s->req->flags, s->rep->flags);
1082
Krzysztof Piotr Oledzkif9423ae2010-01-29 19:26:18 +01001083 /* this data may be no longer valid, clear it */
1084 memset(&s->txn.auth, 0, sizeof(s->txn.auth));
1085
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001086 /* This flag must explicitly be set every time */
1087 s->req->flags &= ~BF_READ_NOEXP;
1088
1089 /* Keep a copy of req/rep flags so that we can detect shutdowns */
1090 rqf_last = s->req->flags;
1091 rpf_last = s->rep->flags;
1092
Willy Tarreau89f7ef22009-09-05 20:57:35 +02001093 /* we don't want the stream interface functions to recursively wake us up */
1094 if (s->req->prod->owner == t)
1095 s->req->prod->flags |= SI_FL_DONT_WAKE;
1096 if (s->req->cons->owner == t)
1097 s->req->cons->flags |= SI_FL_DONT_WAKE;
1098
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001099 /* 1a: Check for low level timeouts if needed. We just set a flag on
1100 * stream interfaces when their timeouts have expired.
1101 */
1102 if (unlikely(t->state & TASK_WOKEN_TIMER)) {
1103 stream_int_check_timeouts(&s->si[0]);
1104 stream_int_check_timeouts(&s->si[1]);
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001105
1106 /* check buffer timeouts, and close the corresponding stream interfaces
1107 * for future reads or writes. Note: this will also concern upper layers
1108 * but we do not touch any other flag. We must be careful and correctly
1109 * detect state changes when calling them.
1110 */
1111
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001112 buffer_check_timeouts(s->req);
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001113
Willy Tarreau14641402009-12-29 14:49:56 +01001114 if (unlikely((s->req->flags & (BF_SHUTW|BF_WRITE_TIMEOUT)) == BF_WRITE_TIMEOUT)) {
1115 s->req->cons->flags |= SI_FL_NOLINGER;
1116 s->req->cons->shutw(s->req->cons);
1117 }
1118
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001119 if (unlikely((s->req->flags & (BF_SHUTR|BF_READ_TIMEOUT)) == BF_READ_TIMEOUT))
1120 s->req->prod->shutr(s->req->prod);
1121
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001122 buffer_check_timeouts(s->rep);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001123
Willy Tarreau14641402009-12-29 14:49:56 +01001124 if (unlikely((s->rep->flags & (BF_SHUTW|BF_WRITE_TIMEOUT)) == BF_WRITE_TIMEOUT)) {
1125 s->rep->cons->flags |= SI_FL_NOLINGER;
1126 s->rep->cons->shutw(s->rep->cons);
1127 }
1128
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001129 if (unlikely((s->rep->flags & (BF_SHUTR|BF_READ_TIMEOUT)) == BF_READ_TIMEOUT))
1130 s->rep->prod->shutr(s->rep->prod);
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001131 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001132
1133 /* 1b: check for low-level errors reported at the stream interface.
1134 * First we check if it's a retryable error (in which case we don't
1135 * want to tell the buffer). Otherwise we report the error one level
1136 * upper by setting flags into the buffers. Note that the side towards
1137 * the client cannot have connect (hence retryable) errors. Also, the
1138 * connection setup code must be able to deal with any type of abort.
1139 */
1140 if (unlikely(s->si[0].flags & SI_FL_ERR)) {
1141 if (s->si[0].state == SI_ST_EST || s->si[0].state == SI_ST_DIS) {
1142 s->si[0].shutr(&s->si[0]);
1143 s->si[0].shutw(&s->si[0]);
1144 stream_int_report_error(&s->si[0]);
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001145 if (!(s->req->analysers) && !(s->rep->analysers)) {
Willy Tarreauae526782010-03-04 20:34:23 +01001146 s->be->counters.cli_aborts++;
1147 if (s->srv)
1148 s->srv->counters.cli_aborts++;
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001149 if (!(s->flags & SN_ERR_MASK))
1150 s->flags |= SN_ERR_CLICL;
1151 if (!(s->flags & SN_FINST_MASK))
1152 s->flags |= SN_FINST_D;
1153 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001154 }
1155 }
1156
1157 if (unlikely(s->si[1].flags & SI_FL_ERR)) {
1158 if (s->si[1].state == SI_ST_EST || s->si[1].state == SI_ST_DIS) {
1159 s->si[1].shutr(&s->si[1]);
1160 s->si[1].shutw(&s->si[1]);
1161 stream_int_report_error(&s->si[1]);
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +02001162 s->be->counters.failed_resp++;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001163 if (s->srv)
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +02001164 s->srv->counters.failed_resp++;
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001165 if (!(s->req->analysers) && !(s->rep->analysers)) {
Willy Tarreauae526782010-03-04 20:34:23 +01001166 s->be->counters.srv_aborts++;
1167 if (s->srv)
1168 s->srv->counters.srv_aborts++;
Willy Tarreau05cb29b2008-12-14 11:44:04 +01001169 if (!(s->flags & SN_ERR_MASK))
1170 s->flags |= SN_ERR_SRVCL;
1171 if (!(s->flags & SN_FINST_MASK))
1172 s->flags |= SN_FINST_D;
1173 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001174 }
1175 /* note: maybe we should process connection errors here ? */
1176 }
1177
1178 if (s->si[1].state == SI_ST_CON) {
1179 /* we were trying to establish a connection on the server side,
1180 * maybe it succeeded, maybe it failed, maybe we timed out, ...
1181 */
1182 if (unlikely(!sess_update_st_con_tcp(s, &s->si[1])))
1183 sess_update_st_cer(s, &s->si[1]);
1184 else if (s->si[1].state == SI_ST_EST)
1185 sess_establish(s, &s->si[1]);
1186
1187 /* state is now one of SI_ST_CON (still in progress), SI_ST_EST
1188 * (established), SI_ST_DIS (abort), SI_ST_CLO (last error),
1189 * SI_ST_ASS/SI_ST_TAR/SI_ST_REQ for retryable errors.
1190 */
1191 }
1192
Willy Tarreaub67a9b82009-06-21 22:03:51 +02001193resync_stream_interface:
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001194 /* Check for connection closure */
1195
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001196 DPRINTF(stderr,
1197 "[%u] %s:%d: task=%p s=%p, sfl=0x%08x, rq=%p, rp=%p, exp(r,w)=%u,%u rqf=%08x rpf=%08x rql=%d rpl=%d cs=%d ss=%d, cet=0x%x set=0x%x retr=%d\n",
1198 now_ms, __FUNCTION__, __LINE__,
1199 t,
1200 s, s->flags,
1201 s->req, s->rep,
1202 s->req->rex, s->rep->wex,
1203 s->req->flags, s->rep->flags,
1204 s->req->l, s->rep->l, s->rep->cons->state, s->req->cons->state,
1205 s->rep->cons->err_type, s->req->cons->err_type,
Willy Tarreauee28de02010-06-01 09:51:00 +02001206 s->req->cons->conn_retries);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001207
1208 /* nothing special to be done on client side */
1209 if (unlikely(s->req->prod->state == SI_ST_DIS))
1210 s->req->prod->state = SI_ST_CLO;
1211
1212 /* When a server-side connection is released, we have to count it and
1213 * check for pending connections on this server.
1214 */
1215 if (unlikely(s->req->cons->state == SI_ST_DIS)) {
1216 s->req->cons->state = SI_ST_CLO;
1217 if (s->srv) {
1218 if (s->flags & SN_CURR_SESS) {
1219 s->flags &= ~SN_CURR_SESS;
1220 s->srv->cur_sess--;
1221 }
1222 sess_change_server(s, NULL);
1223 if (may_dequeue_tasks(s->srv, s->be))
1224 process_srv_queue(s->srv);
1225 }
1226 }
1227
1228 /*
1229 * Note: of the transient states (REQ, CER, DIS), only REQ may remain
1230 * at this point.
1231 */
1232
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001233 resync_request:
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001234 /* Analyse request */
1235 if ((s->req->flags & BF_MASK_ANALYSER) ||
1236 (s->req->flags ^ rqf_last) & BF_MASK_STATIC) {
1237 unsigned int flags = s->req->flags;
1238
1239 if (s->req->prod->state >= SI_ST_EST) {
Willy Tarreaue34070e2010-01-08 00:32:27 +01001240 int max_loops = global.tune.maxpollevents;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001241 unsigned int ana_list;
1242 unsigned int ana_back;
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001243
Willy Tarreau90deb182010-01-07 00:20:41 +01001244 /* it's up to the analysers to stop new connections,
1245 * disable reading or closing. Note: if an analyser
1246 * disables any of these bits, it is responsible for
1247 * enabling them again when it disables itself, so
1248 * that other analysers are called in similar conditions.
1249 */
1250 buffer_auto_read(s->req);
Willy Tarreau520d95e2009-09-19 21:04:57 +02001251 buffer_auto_connect(s->req);
1252 buffer_auto_close(s->req);
Willy Tarreauedcf6682008-11-30 23:15:34 +01001253
1254 /* We will call all analysers for which a bit is set in
1255 * s->req->analysers, following the bit order from LSB
1256 * to MSB. The analysers must remove themselves from
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001257 * the list when not needed. Any analyser may return 0
1258 * to break out of the loop, either because of missing
1259 * data to take a decision, or because it decides to
1260 * kill the session. We loop at least once through each
1261 * analyser, and we may loop again if other analysers
1262 * are added in the middle.
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001263 *
1264 * We build a list of analysers to run. We evaluate all
1265 * of these analysers in the order of the lower bit to
1266 * the higher bit. This ordering is very important.
1267 * An analyser will often add/remove other analysers,
1268 * including itself. Any changes to itself have no effect
1269 * on the loop. If it removes any other analysers, we
1270 * want those analysers not to be called anymore during
1271 * this loop. If it adds an analyser that is located
1272 * after itself, we want it to be scheduled for being
1273 * processed during the loop. If it adds an analyser
1274 * which is located before it, we want it to switch to
1275 * it immediately, even if it has already been called
1276 * once but removed since.
1277 *
1278 * In order to achieve this, we compare the analyser
1279 * list after the call with a copy of it before the
1280 * call. The work list is fed with analyser bits that
1281 * appeared during the call. Then we compare previous
1282 * work list with the new one, and check the bits that
1283 * appeared. If the lowest of these bits is lower than
1284 * the current bit, it means we have enabled a previous
1285 * analyser and must immediately loop again.
Willy Tarreauedcf6682008-11-30 23:15:34 +01001286 */
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001287
1288 ana_list = ana_back = s->req->analysers;
Willy Tarreaue34070e2010-01-08 00:32:27 +01001289 while (ana_list && max_loops--) {
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001290 /* Warning! ensure that analysers are always placed in ascending order! */
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001291
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001292 if (ana_list & AN_REQ_INSPECT) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001293 if (!tcp_inspect_request(s, s->req, AN_REQ_INSPECT))
Willy Tarreauedcf6682008-11-30 23:15:34 +01001294 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001295 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_INSPECT);
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001296 }
Willy Tarreauedcf6682008-11-30 23:15:34 +01001297
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001298 if (ana_list & AN_REQ_WAIT_HTTP) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001299 if (!http_wait_for_request(s, s->req, AN_REQ_WAIT_HTTP))
Willy Tarreaud787e662009-07-07 10:14:51 +02001300 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001301 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_WAIT_HTTP);
Willy Tarreaud787e662009-07-07 10:14:51 +02001302 }
1303
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001304 if (ana_list & AN_REQ_HTTP_PROCESS_FE) {
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001305 if (!http_process_req_common(s, s->req, AN_REQ_HTTP_PROCESS_FE, s->fe))
1306 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001307 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_PROCESS_FE);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001308 }
1309
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001310 if (ana_list & AN_REQ_SWITCHING_RULES) {
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001311 if (!process_switching_rules(s, s->req, AN_REQ_SWITCHING_RULES))
1312 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001313 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_SWITCHING_RULES);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001314 }
1315
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001316 if (ana_list & AN_REQ_HTTP_PROCESS_BE) {
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001317 if (!http_process_req_common(s, s->req, AN_REQ_HTTP_PROCESS_BE, s->be))
1318 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001319 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_PROCESS_BE);
Willy Tarreau1d0dfb12009-07-07 15:10:31 +02001320 }
1321
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001322 if (ana_list & AN_REQ_HTTP_TARPIT) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001323 if (!http_process_tarpit(s, s->req, AN_REQ_HTTP_TARPIT))
Willy Tarreau60b85b02008-11-30 23:28:40 +01001324 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001325 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_TARPIT);
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001326 }
Willy Tarreau60b85b02008-11-30 23:28:40 +01001327
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001328 if (ana_list & AN_REQ_HTTP_INNER) {
Willy Tarreauc465fd72009-08-31 00:17:18 +02001329 if (!http_process_request(s, s->req, AN_REQ_HTTP_INNER))
1330 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001331 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_INNER);
Willy Tarreauc465fd72009-08-31 00:17:18 +02001332 }
1333
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001334 if (ana_list & AN_REQ_HTTP_BODY) {
Willy Tarreau3a816292009-07-07 10:55:49 +02001335 if (!http_process_request_body(s, s->req, AN_REQ_HTTP_BODY))
Willy Tarreaud34af782008-11-30 23:36:37 +01001336 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001337 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_BODY);
Willy Tarreau1a52dbd2009-06-28 19:37:53 +02001338 }
Emeric Brun647caf12009-06-30 17:57:00 +02001339
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001340 if (ana_list & AN_REQ_PRST_RDP_COOKIE) {
Emeric Brun647caf12009-06-30 17:57:00 +02001341 if (!tcp_persist_rdp_cookie(s, s->req, AN_REQ_PRST_RDP_COOKIE))
1342 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001343 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_PRST_RDP_COOKIE);
Emeric Brun647caf12009-06-30 17:57:00 +02001344 }
Willy Tarreaud98cf932009-12-27 22:54:55 +01001345
Emeric Brun1d33b292010-01-04 15:47:17 +01001346 if (ana_list & AN_REQ_STICKING_RULES) {
1347 if (!process_sticking_rules(s, s->req, AN_REQ_STICKING_RULES))
1348 break;
1349 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_STICKING_RULES);
1350 }
1351
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001352 if (ana_list & AN_REQ_HTTP_XFER_BODY) {
Willy Tarreaud98cf932009-12-27 22:54:55 +01001353 if (!http_request_forward_body(s, s->req, AN_REQ_HTTP_XFER_BODY))
1354 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001355 UPDATE_ANALYSERS(s->req->analysers, ana_list, ana_back, AN_REQ_HTTP_XFER_BODY);
Willy Tarreaud98cf932009-12-27 22:54:55 +01001356 }
Willy Tarreaue34070e2010-01-08 00:32:27 +01001357 break;
1358 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001359 }
Willy Tarreau84455332009-03-15 22:34:05 +01001360
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001361 if ((s->req->flags ^ flags) & BF_MASK_STATIC) {
1362 rqf_last = s->req->flags;
1363 goto resync_request;
1364 }
1365 }
1366
Willy Tarreau576507f2010-01-07 00:09:04 +01001367 /* we'll monitor the request analysers while parsing the response,
1368 * because some response analysers may indirectly enable new request
1369 * analysers (eg: HTTP keep-alive).
1370 */
1371 req_ana_back = s->req->analysers;
1372
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001373 resync_response:
1374 /* Analyse response */
1375
1376 if (unlikely(s->rep->flags & BF_HIJACK)) {
1377 /* In inject mode, we wake up everytime something has
1378 * happened on the write side of the buffer.
1379 */
1380 unsigned int flags = s->rep->flags;
1381
1382 if ((s->rep->flags & (BF_WRITE_PARTIAL|BF_WRITE_ERROR|BF_SHUTW)) &&
1383 !(s->rep->flags & BF_FULL)) {
1384 s->rep->hijacker(s, s->rep);
1385 }
1386
1387 if ((s->rep->flags ^ flags) & BF_MASK_STATIC) {
1388 rpf_last = s->rep->flags;
1389 goto resync_response;
1390 }
1391 }
1392 else if ((s->rep->flags & BF_MASK_ANALYSER) ||
1393 (s->rep->flags ^ rpf_last) & BF_MASK_STATIC) {
1394 unsigned int flags = s->rep->flags;
1395
1396 if (s->rep->prod->state >= SI_ST_EST) {
Willy Tarreaue34070e2010-01-08 00:32:27 +01001397 int max_loops = global.tune.maxpollevents;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001398 unsigned int ana_list;
1399 unsigned int ana_back;
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001400
Willy Tarreau90deb182010-01-07 00:20:41 +01001401 /* it's up to the analysers to stop disable reading or
1402 * closing. Note: if an analyser disables any of these
1403 * bits, it is responsible for enabling them again when
1404 * it disables itself, so that other analysers are called
1405 * in similar conditions.
1406 */
1407 buffer_auto_read(s->rep);
Willy Tarreau520d95e2009-09-19 21:04:57 +02001408 buffer_auto_close(s->rep);
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001409
1410 /* We will call all analysers for which a bit is set in
1411 * s->rep->analysers, following the bit order from LSB
1412 * to MSB. The analysers must remove themselves from
1413 * the list when not needed. Any analyser may return 0
1414 * to break out of the loop, either because of missing
1415 * data to take a decision, or because it decides to
1416 * kill the session. We loop at least once through each
1417 * analyser, and we may loop again if other analysers
1418 * are added in the middle.
1419 */
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001420
1421 ana_list = ana_back = s->rep->analysers;
Willy Tarreaue34070e2010-01-08 00:32:27 +01001422 while (ana_list && max_loops--) {
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001423 /* Warning! ensure that analysers are always placed in ascending order! */
1424
1425 if (ana_list & AN_RES_WAIT_HTTP) {
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001426 if (!http_wait_for_response(s, s->rep, AN_RES_WAIT_HTTP))
1427 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001428 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_WAIT_HTTP);
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001429 }
1430
Emeric Brun1d33b292010-01-04 15:47:17 +01001431 if (ana_list & AN_RES_STORE_RULES) {
1432 if (!process_store_rules(s, s->rep, AN_RES_STORE_RULES))
1433 break;
1434 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_STORE_RULES);
1435 }
1436
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001437 if (ana_list & AN_RES_HTTP_PROCESS_BE) {
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001438 if (!http_process_res_common(s, s->rep, AN_RES_HTTP_PROCESS_BE, s->be))
1439 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001440 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_HTTP_PROCESS_BE);
Willy Tarreaub37c27e2009-10-18 22:53:08 +02001441 }
Willy Tarreaud98cf932009-12-27 22:54:55 +01001442
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001443 if (ana_list & AN_RES_HTTP_XFER_BODY) {
Willy Tarreaud98cf932009-12-27 22:54:55 +01001444 if (!http_response_forward_body(s, s->rep, AN_RES_HTTP_XFER_BODY))
1445 break;
Willy Tarreau1e0bbaf2010-01-06 23:53:24 +01001446 UPDATE_ANALYSERS(s->rep->analysers, ana_list, ana_back, AN_RES_HTTP_XFER_BODY);
Willy Tarreaud98cf932009-12-27 22:54:55 +01001447 }
Willy Tarreaue34070e2010-01-08 00:32:27 +01001448 break;
1449 }
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001450 }
1451
1452 if ((s->rep->flags ^ flags) & BF_MASK_STATIC) {
1453 rpf_last = s->rep->flags;
1454 goto resync_response;
1455 }
1456 }
1457
Willy Tarreau576507f2010-01-07 00:09:04 +01001458 /* maybe someone has added some request analysers, so we must check and loop */
1459 if (s->req->analysers & ~req_ana_back)
1460 goto resync_request;
1461
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001462 /* FIXME: here we should call protocol handlers which rely on
1463 * both buffers.
1464 */
1465
1466
1467 /*
Willy Tarreauae526782010-03-04 20:34:23 +01001468 * Now we propagate unhandled errors to the session. Normally
1469 * we're just in a data phase here since it means we have not
1470 * seen any analyser who could set an error status.
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001471 */
1472 if (!(s->flags & SN_ERR_MASK)) {
1473 if (s->req->flags & (BF_READ_ERROR|BF_READ_TIMEOUT|BF_WRITE_ERROR|BF_WRITE_TIMEOUT)) {
1474 /* Report it if the client got an error or a read timeout expired */
Willy Tarreau84455332009-03-15 22:34:05 +01001475 s->req->analysers = 0;
Willy Tarreauae526782010-03-04 20:34:23 +01001476 if (s->req->flags & BF_READ_ERROR) {
1477 s->be->counters.cli_aborts++;
1478 if (s->srv)
1479 s->srv->counters.cli_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01001480 s->flags |= SN_ERR_CLICL;
Willy Tarreauae526782010-03-04 20:34:23 +01001481 }
1482 else if (s->req->flags & BF_READ_TIMEOUT) {
1483 s->be->counters.cli_aborts++;
1484 if (s->srv)
1485 s->srv->counters.cli_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01001486 s->flags |= SN_ERR_CLITO;
Willy Tarreauae526782010-03-04 20:34:23 +01001487 }
1488 else if (s->req->flags & BF_WRITE_ERROR) {
1489 s->be->counters.srv_aborts++;
1490 if (s->srv)
1491 s->srv->counters.srv_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01001492 s->flags |= SN_ERR_SRVCL;
Willy Tarreauae526782010-03-04 20:34:23 +01001493 }
1494 else {
1495 s->be->counters.srv_aborts++;
1496 if (s->srv)
1497 s->srv->counters.srv_aborts++;
Willy Tarreau84455332009-03-15 22:34:05 +01001498 s->flags |= SN_ERR_SRVTO;
Willy Tarreauae526782010-03-04 20:34:23 +01001499 }
Willy Tarreau84455332009-03-15 22:34:05 +01001500 sess_set_term_flags(s);
1501 }
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001502 else if (s->rep->flags & (BF_READ_ERROR|BF_READ_TIMEOUT|BF_WRITE_ERROR|BF_WRITE_TIMEOUT)) {
1503 /* Report it if the server got an error or a read timeout expired */
1504 s->rep->analysers = 0;
Willy Tarreauae526782010-03-04 20:34:23 +01001505 if (s->rep->flags & BF_READ_ERROR) {
1506 s->be->counters.srv_aborts++;
1507 if (s->srv)
1508 s->srv->counters.srv_aborts++;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001509 s->flags |= SN_ERR_SRVCL;
Willy Tarreauae526782010-03-04 20:34:23 +01001510 }
1511 else if (s->rep->flags & BF_READ_TIMEOUT) {
1512 s->be->counters.srv_aborts++;
1513 if (s->srv)
1514 s->srv->counters.srv_aborts++;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001515 s->flags |= SN_ERR_SRVTO;
Willy Tarreauae526782010-03-04 20:34:23 +01001516 }
1517 else if (s->rep->flags & BF_WRITE_ERROR) {
1518 s->be->counters.cli_aborts++;
1519 if (s->srv)
1520 s->srv->counters.cli_aborts++;
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001521 s->flags |= SN_ERR_CLICL;
Willy Tarreauae526782010-03-04 20:34:23 +01001522 }
1523 else {
1524 s->be->counters.cli_aborts++;
1525 if (s->srv)
1526 s->srv->counters.cli_aborts++;
1527 s->flags |= SN_ERR_CLITO;
1528 }
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001529 sess_set_term_flags(s);
1530 }
Willy Tarreau84455332009-03-15 22:34:05 +01001531 }
1532
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001533 /*
1534 * Here we take care of forwarding unhandled data. This also includes
1535 * connection establishments and shutdown requests.
1536 */
1537
1538
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001539 /* If noone is interested in analysing data, it's time to forward
Willy Tarreau31971e52009-09-20 12:07:52 +02001540 * everything. We configure the buffer to forward indefinitely.
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01001541 */
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001542 if (!s->req->analysers &&
Willy Tarreau82eeaf22009-12-29 12:09:05 +01001543 !(s->req->flags & (BF_HIJACK|BF_SHUTW|BF_SHUTW_NOW)) &&
Willy Tarreau31971e52009-09-20 12:07:52 +02001544 (s->req->prod->state >= SI_ST_EST) &&
1545 (s->req->to_forward != BUF_INFINITE_FORWARD)) {
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001546 /* This buffer is freewheeling, there's no analyser nor hijacker
1547 * attached to it. If any data are left in, we'll permit them to
1548 * move.
1549 */
Willy Tarreau90deb182010-01-07 00:20:41 +01001550 buffer_auto_read(s->req);
Willy Tarreau520d95e2009-09-19 21:04:57 +02001551 buffer_auto_connect(s->req);
1552 buffer_auto_close(s->req);
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001553 buffer_flush(s->req);
Willy Tarreau5bd8c372009-01-19 00:32:22 +01001554
Willy Tarreau31971e52009-09-20 12:07:52 +02001555 /* If the producer is still connected, we'll enable data to flow
1556 * from the producer to the consumer (which might possibly not be
1557 * connected yet).
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001558 */
Willy Tarreau31971e52009-09-20 12:07:52 +02001559 if (!(s->req->flags & (BF_SHUTR|BF_SHUTW|BF_SHUTW_NOW)))
1560 buffer_forward(s->req, BUF_INFINITE_FORWARD);
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01001561 }
Willy Tarreauf890dc92008-12-13 21:12:26 +01001562
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001563 /* check if it is wise to enable kernel splicing to forward request data */
1564 if (!(s->req->flags & (BF_KERN_SPLICING|BF_SHUTR)) &&
1565 s->req->to_forward &&
1566 (global.tune.options & GTUNE_USE_SPLICE) &&
Willy Tarreaudc340a92009-06-28 23:10:19 +02001567 (s->si[0].flags & s->si[1].flags & SI_FL_CAP_SPLICE) &&
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001568 (pipes_used < global.maxpipes) &&
1569 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_REQ) ||
1570 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_AUT) &&
1571 (s->req->flags & BF_STREAMER_FAST)))) {
1572 s->req->flags |= BF_KERN_SPLICING;
1573 }
1574
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001575 /* reflect what the L7 analysers have seen last */
1576 rqf_last = s->req->flags;
1577
1578 /*
1579 * Now forward all shutdown requests between both sides of the buffer
1580 */
1581
Willy Tarreau520d95e2009-09-19 21:04:57 +02001582 /* first, let's check if the request buffer needs to shutdown(write), which may
1583 * happen either because the input is closed or because we want to force a close
Willy Tarreaue4599762010-03-21 23:25:09 +01001584 * once the server has begun to respond.
Willy Tarreau520d95e2009-09-19 21:04:57 +02001585 */
Willy Tarreau82eeaf22009-12-29 12:09:05 +01001586 if (unlikely((s->req->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_HIJACK|BF_AUTO_CLOSE|BF_SHUTR)) ==
Willy Tarreaue4599762010-03-21 23:25:09 +01001587 (BF_AUTO_CLOSE|BF_SHUTR)))
Willy Tarreauba0b63d2009-09-20 08:09:44 +02001588 buffer_shutw_now(s->req);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001589
1590 /* shutdown(write) pending */
Willy Tarreauba0b63d2009-09-20 08:09:44 +02001591 if (unlikely((s->req->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_OUT_EMPTY)) == (BF_SHUTW_NOW|BF_OUT_EMPTY)))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001592 s->req->cons->shutw(s->req->cons);
1593
1594 /* shutdown(write) done on server side, we must stop the client too */
Willy Tarreau3dbc6942008-12-07 13:05:04 +01001595 if (unlikely((s->req->flags & (BF_SHUTW|BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTW &&
1596 !s->req->analysers))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001597 buffer_shutr_now(s->req);
1598
1599 /* shutdown(read) pending */
1600 if (unlikely((s->req->flags & (BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTR_NOW))
1601 s->req->prod->shutr(s->req->prod);
1602
Willy Tarreau520d95e2009-09-19 21:04:57 +02001603 /* it's possible that an upper layer has requested a connection setup or abort.
1604 * There are 2 situations where we decide to establish a new connection :
1605 * - there are data scheduled for emission in the buffer
1606 * - the BF_AUTO_CONNECT flag is set (active connection)
1607 */
1608 if (s->req->cons->state == SI_ST_INI) {
Willy Tarreaue4599762010-03-21 23:25:09 +01001609 if (!(s->req->flags & BF_SHUTW)) {
Willy Tarreauba0b63d2009-09-20 08:09:44 +02001610 if ((s->req->flags & (BF_AUTO_CONNECT|BF_OUT_EMPTY)) != BF_OUT_EMPTY) {
Willy Tarreau85e7d002010-05-31 11:57:51 +02001611 /* If we have an iohandler without a connect method, we immediately
1612 * switch to the connected state, otherwise we perform a connection
1613 * request.
Willy Tarreau520d95e2009-09-19 21:04:57 +02001614 */
Willy Tarreau85e7d002010-05-31 11:57:51 +02001615 s->req->cons->state = SI_ST_REQ; /* new connection requested */
Willy Tarreau070ceb62010-06-01 10:36:43 +02001616 s->req->cons->conn_retries = s->be->conn_retries;
Willy Tarreau85e7d002010-05-31 11:57:51 +02001617 if (unlikely(s->req->cons->iohandler && !s->req->cons->connect)) {
Willy Tarreau520d95e2009-09-19 21:04:57 +02001618 s->req->cons->state = SI_ST_EST; /* connection established */
Willy Tarreau85e7d002010-05-31 11:57:51 +02001619 s->rep->flags |= BF_READ_ATTACHED; /* producer is now attached */
1620 s->req->wex = TICK_ETERNITY;
1621 }
Willy Tarreau520d95e2009-09-19 21:04:57 +02001622 }
Willy Tarreau73201222009-08-16 18:27:24 +02001623 }
Willy Tarreauf41ffdc2009-09-20 08:19:25 +02001624 else {
Willy Tarreau92795622009-03-06 12:51:23 +01001625 s->req->cons->state = SI_ST_CLO; /* shutw+ini = abort */
Willy Tarreauf41ffdc2009-09-20 08:19:25 +02001626 buffer_shutw_now(s->req); /* fix buffer flags upon abort */
1627 buffer_shutr_now(s->rep);
1628 }
Willy Tarreau92795622009-03-06 12:51:23 +01001629 }
1630
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001631
1632 /* we may have a pending connection request, or a connection waiting
1633 * for completion.
1634 */
1635 if (s->si[1].state >= SI_ST_REQ && s->si[1].state < SI_ST_CON) {
1636 do {
1637 /* nb: step 1 might switch from QUE to ASS, but we first want
1638 * to give a chance to step 2 to perform a redirect if needed.
1639 */
1640 if (s->si[1].state != SI_ST_REQ)
1641 sess_update_stream_int(s, &s->si[1]);
1642 if (s->si[1].state == SI_ST_REQ)
1643 sess_prepare_conn_req(s, &s->si[1]);
1644
1645 if (s->si[1].state == SI_ST_ASS && s->srv &&
1646 s->srv->rdr_len && (s->flags & SN_REDIRECTABLE))
1647 perform_http_redirect(s, &s->si[1]);
1648 } while (s->si[1].state == SI_ST_ASS);
1649 }
1650
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001651 /* Benchmarks have shown that it's optimal to do a full resync now */
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001652 if (s->req->prod->state == SI_ST_DIS || s->req->cons->state == SI_ST_DIS)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001653 goto resync_stream_interface;
1654
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001655 /* otherwise wewant to check if we need to resync the req buffer or not */
1656 if ((s->req->flags ^ rqf_last) & BF_MASK_STATIC)
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001657 goto resync_request;
1658
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001659 /* perform output updates to the response buffer */
Willy Tarreau84455332009-03-15 22:34:05 +01001660
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001661 /* If noone is interested in analysing data, it's time to forward
Willy Tarreau31971e52009-09-20 12:07:52 +02001662 * everything. We configure the buffer to forward indefinitely.
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01001663 */
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001664 if (!s->rep->analysers &&
Willy Tarreau82eeaf22009-12-29 12:09:05 +01001665 !(s->rep->flags & (BF_HIJACK|BF_SHUTW|BF_SHUTW_NOW)) &&
Willy Tarreau31971e52009-09-20 12:07:52 +02001666 (s->rep->prod->state >= SI_ST_EST) &&
1667 (s->rep->to_forward != BUF_INFINITE_FORWARD)) {
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001668 /* This buffer is freewheeling, there's no analyser nor hijacker
1669 * attached to it. If any data are left in, we'll permit them to
1670 * move.
1671 */
Willy Tarreau90deb182010-01-07 00:20:41 +01001672 buffer_auto_read(s->rep);
Willy Tarreau520d95e2009-09-19 21:04:57 +02001673 buffer_auto_close(s->rep);
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001674 buffer_flush(s->rep);
Willy Tarreau31971e52009-09-20 12:07:52 +02001675 if (!(s->rep->flags & (BF_SHUTR|BF_SHUTW|BF_SHUTW_NOW)))
1676 buffer_forward(s->rep, BUF_INFINITE_FORWARD);
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01001677 }
Willy Tarreauf890dc92008-12-13 21:12:26 +01001678
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001679 /* check if it is wise to enable kernel splicing to forward response data */
1680 if (!(s->rep->flags & (BF_KERN_SPLICING|BF_SHUTR)) &&
1681 s->rep->to_forward &&
1682 (global.tune.options & GTUNE_USE_SPLICE) &&
Willy Tarreaudc340a92009-06-28 23:10:19 +02001683 (s->si[0].flags & s->si[1].flags & SI_FL_CAP_SPLICE) &&
Willy Tarreau7c84bab2009-03-08 21:38:23 +01001684 (pipes_used < global.maxpipes) &&
1685 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_RTR) ||
1686 (((s->fe->options2|s->be->options2) & PR_O2_SPLIC_AUT) &&
1687 (s->rep->flags & BF_STREAMER_FAST)))) {
1688 s->rep->flags |= BF_KERN_SPLICING;
1689 }
1690
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001691 /* reflect what the L7 analysers have seen last */
1692 rpf_last = s->rep->flags;
1693
1694 /*
1695 * Now forward all shutdown requests between both sides of the buffer
1696 */
1697
1698 /*
1699 * FIXME: this is probably where we should produce error responses.
1700 */
1701
Willy Tarreau6b66f3e2008-12-14 17:31:54 +01001702 /* first, let's check if the response buffer needs to shutdown(write) */
Willy Tarreau520d95e2009-09-19 21:04:57 +02001703 if (unlikely((s->rep->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_HIJACK|BF_AUTO_CLOSE|BF_SHUTR)) ==
1704 (BF_AUTO_CLOSE|BF_SHUTR)))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001705 buffer_shutw_now(s->rep);
1706
1707 /* shutdown(write) pending */
Willy Tarreauba0b63d2009-09-20 08:09:44 +02001708 if (unlikely((s->rep->flags & (BF_SHUTW|BF_OUT_EMPTY|BF_SHUTW_NOW)) == (BF_OUT_EMPTY|BF_SHUTW_NOW)))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001709 s->rep->cons->shutw(s->rep->cons);
1710
1711 /* shutdown(write) done on the client side, we must stop the server too */
Willy Tarreau3dbc6942008-12-07 13:05:04 +01001712 if (unlikely((s->rep->flags & (BF_SHUTW|BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTW) &&
1713 !s->rep->analysers)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001714 buffer_shutr_now(s->rep);
1715
1716 /* shutdown(read) pending */
1717 if (unlikely((s->rep->flags & (BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTR_NOW))
1718 s->rep->prod->shutr(s->rep->prod);
1719
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001720 if (s->req->prod->state == SI_ST_DIS || s->req->cons->state == SI_ST_DIS)
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001721 goto resync_stream_interface;
1722
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001723 if (s->req->flags != rqf_last)
1724 goto resync_request;
1725
Willy Tarreau3deb3d02009-06-21 22:43:05 +02001726 if ((s->rep->flags ^ rpf_last) & BF_MASK_STATIC)
Willy Tarreau0be0ef92009-03-08 19:20:25 +01001727 goto resync_response;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001728
Willy Tarreau89f7ef22009-09-05 20:57:35 +02001729 /* we're interested in getting wakeups again */
1730 s->req->prod->flags &= ~SI_FL_DONT_WAKE;
1731 s->req->cons->flags &= ~SI_FL_DONT_WAKE;
1732
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001733 /* This is needed only when debugging is enabled, to indicate
1734 * client-side or server-side close. Please note that in the unlikely
1735 * event where both sides would close at once, the sequence is reported
1736 * on the server side first.
1737 */
1738 if (unlikely((global.mode & MODE_DEBUG) &&
1739 (!(global.mode & MODE_QUIET) ||
1740 (global.mode & MODE_VERBOSE)))) {
1741 int len;
1742
1743 if (s->si[1].state == SI_ST_CLO &&
1744 s->si[1].prev_state == SI_ST_EST) {
1745 len = sprintf(trash, "%08x:%s.srvcls[%04x:%04x]\n",
1746 s->uniq_id, s->be->id,
1747 (unsigned short)s->si[0].fd,
1748 (unsigned short)s->si[1].fd);
1749 write(1, trash, len);
1750 }
1751
1752 if (s->si[0].state == SI_ST_CLO &&
1753 s->si[0].prev_state == SI_ST_EST) {
1754 len = sprintf(trash, "%08x:%s.clicls[%04x:%04x]\n",
1755 s->uniq_id, s->be->id,
1756 (unsigned short)s->si[0].fd,
1757 (unsigned short)s->si[1].fd);
1758 write(1, trash, len);
1759 }
1760 }
1761
1762 if (likely((s->rep->cons->state != SI_ST_CLO) ||
1763 (s->req->cons->state > SI_ST_INI && s->req->cons->state < SI_ST_CLO))) {
1764
1765 if ((s->fe->options & PR_O_CONTSTATS) && (s->flags & SN_BE_ASSIGNED))
1766 session_process_counters(s);
1767
Willy Tarreau1accfc02009-09-05 20:57:35 +02001768 if (s->rep->cons->state == SI_ST_EST && !s->rep->cons->iohandler)
Willy Tarreaudc85b392009-08-18 07:38:19 +02001769 s->rep->cons->update(s->rep->cons);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001770
Willy Tarreau1accfc02009-09-05 20:57:35 +02001771 if (s->req->cons->state == SI_ST_EST && !s->req->cons->iohandler)
Willy Tarreaudc85b392009-08-18 07:38:19 +02001772 s->req->cons->update(s->req->cons);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001773
Willy Tarreaua6eebb32010-06-04 11:40:20 +02001774 s->req->flags &= ~(BF_READ_NULL|BF_READ_PARTIAL|BF_WRITE_NULL|BF_WRITE_PARTIAL|BF_READ_ATTACHED);
1775 s->rep->flags &= ~(BF_READ_NULL|BF_READ_PARTIAL|BF_WRITE_NULL|BF_WRITE_PARTIAL|BF_READ_ATTACHED);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001776 s->si[0].prev_state = s->si[0].state;
1777 s->si[1].prev_state = s->si[1].state;
Willy Tarreaub0ef7352008-12-14 13:26:20 +01001778 s->si[0].flags &= ~(SI_FL_ERR|SI_FL_EXP);
1779 s->si[1].flags &= ~(SI_FL_ERR|SI_FL_EXP);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001780
1781 /* Trick: if a request is being waiting for the server to respond,
1782 * and if we know the server can timeout, we don't want the timeout
1783 * to expire on the client side first, but we're still interested
1784 * in passing data from the client to the server (eg: POST). Thus,
1785 * we can cancel the client's request timeout if the server's
1786 * request timeout is set and the server has not yet sent a response.
1787 */
1788
Willy Tarreau520d95e2009-09-19 21:04:57 +02001789 if ((s->rep->flags & (BF_AUTO_CLOSE|BF_SHUTR)) == 0 &&
Willy Tarreau86491c32008-12-14 09:04:47 +01001790 (tick_isset(s->req->wex) || tick_isset(s->rep->rex))) {
1791 s->req->flags |= BF_READ_NOEXP;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001792 s->req->rex = TICK_ETERNITY;
Willy Tarreau86491c32008-12-14 09:04:47 +01001793 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001794
Willy Tarreau1accfc02009-09-05 20:57:35 +02001795 /* Call the second stream interface's I/O handler if it's embedded.
1796 * Note that this one may wake the task up again.
1797 */
1798 if (s->req->cons->iohandler) {
1799 s->req->cons->iohandler(s->req->cons);
1800 if (task_in_rq(t)) {
1801 /* If we woke up, we don't want to requeue the
1802 * task to the wait queue, but rather requeue
1803 * it into the runqueue ASAP.
1804 */
1805 t->expire = TICK_ETERNITY;
1806 return t;
1807 }
1808 }
1809
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001810 t->expire = tick_first(tick_first(s->req->rex, s->req->wex),
1811 tick_first(s->rep->rex, s->rep->wex));
1812 if (s->req->analysers)
1813 t->expire = tick_first(t->expire, s->req->analyse_exp);
1814
1815 if (s->si[0].exp)
1816 t->expire = tick_first(t->expire, s->si[0].exp);
1817
1818 if (s->si[1].exp)
1819 t->expire = tick_first(t->expire, s->si[1].exp);
1820
1821#ifdef DEBUG_FULL
Willy Tarreau127334e2009-03-28 10:47:26 +01001822 fprintf(stderr,
1823 "[%u] queuing with exp=%u req->rex=%u req->wex=%u req->ana_exp=%u"
1824 " rep->rex=%u rep->wex=%u, si[0].exp=%u, si[1].exp=%u, cs=%d, ss=%d\n",
1825 now_ms, t->expire, s->req->rex, s->req->wex, s->req->analyse_exp,
1826 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 +01001827#endif
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001828
1829#ifdef DEBUG_DEV
1830 /* this may only happen when no timeout is set or in case of an FSM bug */
Willy Tarreaud0a201b2009-03-08 15:53:06 +01001831 if (!tick_isset(t->expire))
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001832 ABORT_NOW();
1833#endif
Willy Tarreau26c25062009-03-08 09:38:41 +01001834 return t; /* nothing more to do */
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001835 }
1836
1837 s->fe->feconn--;
1838 if (s->flags & SN_BE_ASSIGNED)
1839 s->be->beconn--;
1840 actconn--;
Willy Tarreau6e6fb2b2009-08-16 18:20:44 +02001841 s->listener->nbconn--;
1842 if (s->listener->state == LI_FULL &&
1843 s->listener->nbconn < s->listener->maxconn) {
1844 /* we should reactivate the listener */
1845 EV_FD_SET(s->listener->fd, DIR_RD);
1846 s->listener->state = LI_READY;
1847 }
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001848
1849 if (unlikely((global.mode & MODE_DEBUG) &&
1850 (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE)))) {
1851 int len;
Willy Tarreauec22b2c2009-03-06 13:07:40 +01001852 len = sprintf(trash, "%08x:%s.closed[%04x:%04x]\n",
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001853 s->uniq_id, s->be->id,
Willy Tarreauec22b2c2009-03-06 13:07:40 +01001854 (unsigned short)s->req->prod->fd, (unsigned short)s->req->cons->fd);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001855 write(1, trash, len);
1856 }
1857
1858 s->logs.t_close = tv_ms_elapsed(&s->logs.tv_accept, &now);
1859 session_process_counters(s);
1860
Krzysztof Piotr Oledzkide71d162009-10-24 15:36:15 +02001861 if (s->txn.status) {
1862 int n;
1863
1864 n = s->txn.status / 100;
1865 if (n < 1 || n > 5)
1866 n = 0;
1867
1868 if (s->fe->mode == PR_MODE_HTTP)
Willy Tarreau24657792010-02-26 10:30:28 +01001869 s->fe->counters.fe.http.rsp[n]++;
Krzysztof Piotr Oledzkide71d162009-10-24 15:36:15 +02001870
Willy Tarreau24657792010-02-26 10:30:28 +01001871 if ((s->flags & SN_BE_ASSIGNED) &&
Krzysztof Piotr Oledzkide71d162009-10-24 15:36:15 +02001872 (s->be->mode == PR_MODE_HTTP))
Willy Tarreau24657792010-02-26 10:30:28 +01001873 s->be->counters.be.http.rsp[n]++;
Krzysztof Piotr Oledzkide71d162009-10-24 15:36:15 +02001874 }
1875
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001876 /* let's do a final log if we need it */
1877 if (s->logs.logwait &&
1878 !(s->flags & SN_MONITOR) &&
1879 (!(s->fe->options & PR_O_NULLNOLOG) || s->req->total)) {
Willy Tarreaua5555ec2008-11-30 19:02:32 +01001880 s->do_log(s);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001881 }
1882
1883 /* the task MUST not be in the run queue anymore */
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001884 session_free(s);
Willy Tarreau26c25062009-03-08 09:38:41 +01001885 task_delete(t);
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001886 task_free(t);
Willy Tarreau26c25062009-03-08 09:38:41 +01001887 return NULL;
Willy Tarreau55a8d0e2008-11-30 18:47:21 +01001888}
1889
Willy Tarreau7c669d72008-06-20 15:04:11 +02001890/*
1891 * This function adjusts sess->srv_conn and maintains the previous and new
1892 * server's served session counts. Setting newsrv to NULL is enough to release
1893 * current connection slot. This function also notifies any LB algo which might
1894 * expect to be informed about any change in the number of active sessions on a
1895 * server.
1896 */
1897void sess_change_server(struct session *sess, struct server *newsrv)
1898{
1899 if (sess->srv_conn == newsrv)
1900 return;
1901
1902 if (sess->srv_conn) {
1903 sess->srv_conn->served--;
1904 if (sess->srv_conn->proxy->lbprm.server_drop_conn)
1905 sess->srv_conn->proxy->lbprm.server_drop_conn(sess->srv_conn);
1906 sess->srv_conn = NULL;
1907 }
1908
1909 if (newsrv) {
1910 newsrv->served++;
1911 if (newsrv->proxy->lbprm.server_take_conn)
1912 newsrv->proxy->lbprm.server_take_conn(newsrv);
1913 sess->srv_conn = newsrv;
1914 }
1915}
1916
Willy Tarreau84455332009-03-15 22:34:05 +01001917/* Set correct session termination flags in case no analyser has done it. It
1918 * also counts a failed request if the server state has not reached the request
1919 * stage.
1920 */
1921void sess_set_term_flags(struct session *s)
1922{
1923 if (!(s->flags & SN_FINST_MASK)) {
1924 if (s->si[1].state < SI_ST_REQ) {
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +02001925
Krzysztof Piotr Oledzki052d4fd2009-10-04 14:52:57 +02001926 s->fe->counters.failed_req++;
Krzysztof Piotr Oledzkiaeebf9b2009-10-04 15:43:17 +02001927 if (s->listener->counters)
1928 s->listener->counters->failed_req++;
1929
Willy Tarreau84455332009-03-15 22:34:05 +01001930 s->flags |= SN_FINST_R;
1931 }
1932 else if (s->si[1].state == SI_ST_QUE)
1933 s->flags |= SN_FINST_Q;
1934 else if (s->si[1].state < SI_ST_EST)
1935 s->flags |= SN_FINST_C;
Willy Tarreau033b2db2010-03-04 17:54:21 +01001936 else if (s->si[1].state == SI_ST_EST || s->si[1].prev_state == SI_ST_EST)
Willy Tarreau84455332009-03-15 22:34:05 +01001937 s->flags |= SN_FINST_D;
1938 else
1939 s->flags |= SN_FINST_L;
1940 }
1941}
1942
1943/* Handle server-side errors for default protocols. It is called whenever a a
1944 * connection setup is aborted or a request is aborted in queue. It sets the
1945 * session termination flags so that the caller does not have to worry about
1946 * them. It's installed as ->srv_error for the server-side stream_interface.
1947 */
1948void default_srv_error(struct session *s, struct stream_interface *si)
1949{
1950 int err_type = si->err_type;
1951 int err = 0, fin = 0;
1952
1953 if (err_type & SI_ET_QUEUE_ABRT) {
1954 err = SN_ERR_CLICL;
1955 fin = SN_FINST_Q;
1956 }
1957 else if (err_type & SI_ET_CONN_ABRT) {
1958 err = SN_ERR_CLICL;
1959 fin = SN_FINST_C;
1960 }
1961 else if (err_type & SI_ET_QUEUE_TO) {
1962 err = SN_ERR_SRVTO;
1963 fin = SN_FINST_Q;
1964 }
1965 else if (err_type & SI_ET_QUEUE_ERR) {
1966 err = SN_ERR_SRVCL;
1967 fin = SN_FINST_Q;
1968 }
1969 else if (err_type & SI_ET_CONN_TO) {
1970 err = SN_ERR_SRVTO;
1971 fin = SN_FINST_C;
1972 }
1973 else if (err_type & SI_ET_CONN_ERR) {
1974 err = SN_ERR_SRVCL;
1975 fin = SN_FINST_C;
1976 }
1977 else /* SI_ET_CONN_OTHER and others */ {
1978 err = SN_ERR_INTERNAL;
1979 fin = SN_FINST_C;
1980 }
1981
1982 if (!(s->flags & SN_ERR_MASK))
1983 s->flags |= err;
1984 if (!(s->flags & SN_FINST_MASK))
1985 s->flags |= fin;
1986}
Willy Tarreau7c669d72008-06-20 15:04:11 +02001987
Willy Tarreaubaaee002006-06-26 02:48:02 +02001988/*
1989 * Local variables:
1990 * c-indent-level: 8
1991 * c-basic-offset: 8
1992 * End:
1993 */