blob: f07fa158bb03f85a42dc3376659c8bca2a291615 [file] [log] [blame]
Willy Tarreau92fb9832007-10-16 17:34:28 +02001/*
2 * UNIX SOCK_STREAM protocol layer (uxst)
3 *
Willy Tarreau7c669d72008-06-20 15:04:11 +02004 * Copyright 2000-2008 Willy Tarreau <w@1wt.eu>
Willy Tarreau92fb9832007-10-16 17:34:28 +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 <ctype.h>
14#include <errno.h>
15#include <fcntl.h>
16#include <stdio.h>
17#include <stdlib.h>
18#include <string.h>
19#include <syslog.h>
20#include <time.h>
21
22#include <sys/param.h>
23#include <sys/socket.h>
24#include <sys/stat.h>
25#include <sys/types.h>
26#include <sys/un.h>
27
28#include <common/compat.h>
29#include <common/config.h>
30#include <common/debug.h>
Willy Tarreaud740bab2007-10-28 11:14:07 +010031#include <common/errors.h>
Willy Tarreau92fb9832007-10-16 17:34:28 +020032#include <common/memory.h>
33#include <common/mini-clist.h>
34#include <common/standard.h>
Willy Tarreau0c303ee2008-07-07 00:09:58 +020035#include <common/ticks.h>
Willy Tarreau92fb9832007-10-16 17:34:28 +020036#include <common/time.h>
37#include <common/version.h>
38
Willy Tarreau92fb9832007-10-16 17:34:28 +020039#include <types/global.h>
Willy Tarreau92fb9832007-10-16 17:34:28 +020040
41#include <proto/acl.h>
42#include <proto/backend.h>
43#include <proto/buffers.h>
Willy Tarreau3e76e722007-10-17 18:57:38 +020044#include <proto/dumpstats.h>
Willy Tarreau92fb9832007-10-16 17:34:28 +020045#include <proto/fd.h>
46#include <proto/log.h>
47#include <proto/protocols.h>
48#include <proto/proto_uxst.h>
49#include <proto/queue.h>
50#include <proto/session.h>
Willy Tarreaub1356cf2008-12-07 16:06:43 +010051#include <proto/stream_interface.h>
Willy Tarreau92fb9832007-10-16 17:34:28 +020052#include <proto/stream_sock.h>
53#include <proto/task.h>
54
55#ifndef MAXPATHLEN
56#define MAXPATHLEN 128
57#endif
58
Willy Tarreaudabf2e22007-10-28 21:59:24 +010059static int uxst_bind_listeners(struct protocol *proto);
60static int uxst_unbind_listeners(struct protocol *proto);
61
62/* Note: must not be declared <const> as its list will be overwritten */
63static struct protocol proto_unix = {
64 .name = "unix_stream",
65 .sock_domain = PF_UNIX,
66 .sock_type = SOCK_STREAM,
67 .sock_prot = 0,
68 .sock_family = AF_UNIX,
69 .sock_addrlen = sizeof(struct sockaddr_un),
70 .l3_addrlen = sizeof(((struct sockaddr_un*)0)->sun_path),/* path len */
71 .read = &stream_sock_read,
72 .write = &stream_sock_write,
73 .bind_all = uxst_bind_listeners,
74 .unbind_all = uxst_unbind_listeners,
75 .enable_all = enable_all_listeners,
76 .disable_all = disable_all_listeners,
77 .listeners = LIST_HEAD_INIT(proto_unix.listeners),
78 .nb_listeners = 0,
79};
80
81
82/********************************
83 * 1) low-level socket functions
84 ********************************/
85
86
Willy Tarreau92fb9832007-10-16 17:34:28 +020087/* This function creates a named PF_UNIX stream socket at address <path>. Note
Willy Tarreaue6ad2b12007-10-18 12:45:54 +020088 * that the path cannot be NULL nor empty. <uid> and <gid> different of -1 will
89 * be used to change the socket owner. If <mode> is not 0, it will be used to
90 * restrict access to the socket. While it is known not to be portable on every
91 * OS, it's still useful where it works.
Willy Tarreau92fb9832007-10-16 17:34:28 +020092 * It returns the assigned file descriptor, or -1 in the event of an error.
93 */
Willy Tarreaue6ad2b12007-10-18 12:45:54 +020094static int create_uxst_socket(const char *path, uid_t uid, gid_t gid, mode_t mode)
Willy Tarreau92fb9832007-10-16 17:34:28 +020095{
96 char tempname[MAXPATHLEN];
97 char backname[MAXPATHLEN];
98 struct sockaddr_un addr;
99
100 int ret, sock;
101
102 /* 1. create socket names */
103 if (!path[0]) {
104 Alert("Invalid name for a UNIX socket. Aborting.\n");
105 goto err_return;
106 }
107
108 ret = snprintf(tempname, MAXPATHLEN, "%s.%d.tmp", path, pid);
109 if (ret < 0 || ret >= MAXPATHLEN) {
110 Alert("name too long for UNIX socket. Aborting.\n");
111 goto err_return;
112 }
113
114 ret = snprintf(backname, MAXPATHLEN, "%s.%d.bak", path, pid);
115 if (ret < 0 || ret >= MAXPATHLEN) {
116 Alert("name too long for UNIX socket. Aborting.\n");
117 goto err_return;
118 }
119
120 /* 2. clean existing orphaned entries */
121 if (unlink(tempname) < 0 && errno != ENOENT) {
122 Alert("error when trying to unlink previous UNIX socket. Aborting.\n");
123 goto err_return;
124 }
125
126 if (unlink(backname) < 0 && errno != ENOENT) {
127 Alert("error when trying to unlink previous UNIX socket. Aborting.\n");
128 goto err_return;
129 }
130
131 /* 3. backup existing socket */
132 if (link(path, backname) < 0 && errno != ENOENT) {
133 Alert("error when trying to preserve previous UNIX socket. Aborting.\n");
134 goto err_return;
135 }
136
137 /* 4. prepare new socket */
138 addr.sun_family = AF_UNIX;
139 strncpy(addr.sun_path, tempname, sizeof(addr.sun_path));
140 addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
141
142 sock = socket(PF_UNIX, SOCK_STREAM, 0);
143 if (sock < 0) {
144 Alert("cannot create socket for UNIX listener. Aborting.\n");
145 goto err_unlink_back;
146 }
147
148 if (sock >= global.maxsock) {
149 Alert("socket(): not enough free sockets for UNIX listener. Raise -n argument. Aborting.\n");
150 goto err_unlink_temp;
151 }
152
153 if (fcntl(sock, F_SETFL, O_NONBLOCK) == -1) {
154 Alert("cannot make UNIX socket non-blocking. Aborting.\n");
155 goto err_unlink_temp;
156 }
157
158 if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
159 /* note that bind() creates the socket <tempname> on the file system */
160 Alert("cannot bind socket for UNIX listener. Aborting.\n");
161 goto err_unlink_temp;
162 }
163
Willy Tarreaue6ad2b12007-10-18 12:45:54 +0200164 if (((uid != -1 || gid != -1) && (chown(tempname, uid, gid) == -1)) ||
165 (mode != 0 && chmod(tempname, mode) == -1)) {
166 Alert("cannot change UNIX socket ownership. Aborting.\n");
167 goto err_unlink_temp;
168 }
169
Willy Tarreau92fb9832007-10-16 17:34:28 +0200170 if (listen(sock, 0) < 0) {
171 Alert("cannot listen to socket for UNIX listener. Aborting.\n");
172 goto err_unlink_temp;
173 }
174
175 /* 5. install.
176 * Point of no return: we are ready, we'll switch the sockets. We don't
177 * fear loosing the socket <path> because we have a copy of it in
178 * backname.
179 */
180 if (rename(tempname, path) < 0) {
181 Alert("cannot switch final and temporary sockets for UNIX listener. Aborting.\n");
182 goto err_rename;
183 }
184
185 /* 6. cleanup */
186 unlink(backname); /* no need to keep this one either */
187
188 return sock;
189
190 err_rename:
191 ret = rename(backname, path);
192 if (ret < 0 && errno == ENOENT)
193 unlink(path);
194 err_unlink_temp:
195 unlink(tempname);
196 close(sock);
197 err_unlink_back:
198 unlink(backname);
199 err_return:
200 return -1;
201}
202
203/* Tries to destroy the UNIX stream socket <path>. The socket must not be used
204 * anymore. It practises best effort, and no error is returned.
205 */
206static void destroy_uxst_socket(const char *path)
207{
208 struct sockaddr_un addr;
209 int sock, ret;
210
211 /* We might have been chrooted, so we may not be able to access the
212 * socket. In order to avoid bothering the other end, we connect with a
213 * wrong protocol, namely SOCK_DGRAM. The return code from connect()
214 * is enough to know if the socket is still live or not. If it's live
215 * in mode SOCK_STREAM, we get EPROTOTYPE or anything else but not
216 * ECONNREFUSED. In this case, we do not touch it because it's used
217 * by some other process.
218 */
219 sock = socket(PF_UNIX, SOCK_DGRAM, 0);
220 if (sock < 0)
221 return;
222
223 addr.sun_family = AF_UNIX;
224 strncpy(addr.sun_path, path, sizeof(addr.sun_path));
Willy Tarreau10ae5482007-10-18 16:15:52 +0200225 addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200226 ret = connect(sock, (struct sockaddr *)&addr, sizeof(addr));
227 if (ret < 0 && errno == ECONNREFUSED) {
228 /* Connect failed: the socket still exists but is not used
229 * anymore. Let's remove this socket now.
230 */
231 unlink(path);
232 }
233 close(sock);
234}
235
236
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100237/********************************
238 * 2) listener-oriented functions
239 ********************************/
240
241
242/* This function creates the UNIX socket associated to the listener. It changes
243 * the state from ASSIGNED to LISTEN. The socket is NOT enabled for polling.
244 * The return value is composed from ERR_NONE, ERR_RETRYABLE and ERR_FATAL.
245 */
246static int uxst_bind_listener(struct listener *listener)
247{
248 int fd;
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100249
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100250 if (listener->state != LI_ASSIGNED)
251 return ERR_NONE; /* already bound */
252
253 fd = create_uxst_socket(((struct sockaddr_un *)&listener->addr)->sun_path,
254 listener->perm.ux.uid,
255 listener->perm.ux.gid,
256 listener->perm.ux.mode);
257 if (fd == -1)
258 return ERR_FATAL;
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100259
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100260 /* the socket is now listening */
261 listener->fd = fd;
262 listener->state = LI_LISTEN;
263
264 /* the function for the accept() event */
265 fd_insert(fd);
266 fdtab[fd].cb[DIR_RD].f = listener->accept;
267 fdtab[fd].cb[DIR_WR].f = NULL; /* never called */
268 fdtab[fd].cb[DIR_RD].b = fdtab[fd].cb[DIR_WR].b = NULL;
Willy Tarreaueabf3132008-08-29 23:36:51 +0200269 fdtab[fd].owner = listener; /* reference the listener instead of a task */
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100270 fdtab[fd].state = FD_STLISTEN;
271 fdtab[fd].peeraddr = NULL;
272 fdtab[fd].peerlen = 0;
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100273 return ERR_NONE;
274}
275
276/* This function closes the UNIX sockets for the specified listener.
277 * The listener enters the LI_ASSIGNED state. It always returns ERR_NONE.
278 */
279static int uxst_unbind_listener(struct listener *listener)
280{
281 if (listener->state == LI_READY)
282 EV_FD_CLR(listener->fd, DIR_RD);
283
284 if (listener->state >= LI_LISTEN) {
Willy Tarreau8eebe5e2007-10-28 22:07:08 +0100285 fd_delete(listener->fd);
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100286 listener->state = LI_ASSIGNED;
287 destroy_uxst_socket(((struct sockaddr_un *)&listener->addr)->sun_path);
288 }
289 return ERR_NONE;
290}
291
292/* Add a listener to the list of unix stream listeners. The listener's state
293 * is automatically updated from LI_INIT to LI_ASSIGNED. The number of
294 * listeners is updated. This is the function to use to add a new listener.
295 */
296void uxst_add_listener(struct listener *listener)
297{
298 if (listener->state != LI_INIT)
299 return;
300 listener->state = LI_ASSIGNED;
301 listener->proto = &proto_unix;
302 LIST_ADDQ(&proto_unix.listeners, &listener->proto_list);
303 proto_unix.nb_listeners++;
304}
305
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100306/********************************
307 * 3) protocol-oriented functions
308 ********************************/
309
310
Willy Tarreau92fb9832007-10-16 17:34:28 +0200311/* This function creates all UNIX sockets bound to the protocol entry <proto>.
312 * It is intended to be used as the protocol's bind_all() function.
313 * The sockets will be registered but not added to any fd_set, in order not to
314 * loose them across the fork(). A call to uxst_enable_listeners() is needed
315 * to complete initialization.
316 *
317 * The return value is composed from ERR_NONE, ERR_RETRYABLE and ERR_FATAL.
318 */
319static int uxst_bind_listeners(struct protocol *proto)
320{
321 struct listener *listener;
322 int err = ERR_NONE;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200323
324 list_for_each_entry(listener, &proto->listeners, proto_list) {
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100325 err |= uxst_bind_listener(listener);
326 if (err != ERR_NONE)
Willy Tarreau92fb9832007-10-16 17:34:28 +0200327 continue;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200328 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200329 return err;
330}
331
Willy Tarreau92fb9832007-10-16 17:34:28 +0200332
333/* This function stops all listening UNIX sockets bound to the protocol
334 * <proto>. It does not detaches them from the protocol.
335 * It always returns ERR_NONE.
336 */
337static int uxst_unbind_listeners(struct protocol *proto)
338{
339 struct listener *listener;
340
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100341 list_for_each_entry(listener, &proto->listeners, proto_list)
342 uxst_unbind_listener(listener);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200343 return ERR_NONE;
344}
345
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100346
347/********************************
348 * 4) high-level functions
349 ********************************/
350
351
Willy Tarreau92fb9832007-10-16 17:34:28 +0200352/*
353 * This function is called on a read event from a listen socket, corresponding
354 * to an accept. It tries to accept as many connections as possible.
355 * It returns 0. Since we use UNIX sockets on the local system for monitoring
356 * purposes and other related things, we do not need to output as many messages
357 * as with TCP which can fall under attack.
358 */
359int uxst_event_accept(int fd) {
Willy Tarreaueabf3132008-08-29 23:36:51 +0200360 struct listener *l = fdtab[fd].owner;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200361 struct session *s;
362 struct task *t;
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100363 struct bref *bref, *back;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200364 int cfd;
365 int max_accept;
366
367 if (global.nbproc > 1)
368 max_accept = 8; /* let other processes catch some connections too */
369 else
370 max_accept = -1;
371
372 while (max_accept--) {
373 struct sockaddr_storage addr;
374 socklen_t laddr = sizeof(addr);
375
376 if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1) {
377 switch (errno) {
378 case EAGAIN:
379 case EINTR:
380 case ECONNABORTED:
381 return 0; /* nothing more to accept */
382 case ENFILE:
383 /* Process reached system FD limit. Check system tunables. */
384 return 0;
385 case EMFILE:
386 /* Process reached process FD limit. Check 'ulimit-n'. */
387 return 0;
388 case ENOBUFS:
389 case ENOMEM:
390 /* Process reached system memory limit. Check system tunables. */
391 return 0;
392 default:
393 return 0;
394 }
395 }
396
397 if (l->nbconn >= l->maxconn) {
398 /* too many connections, we shoot this one and return.
399 * FIXME: it would be better to simply switch the listener's
400 * state to LI_FULL and disable the FD. We could re-enable
401 * it upon fd_delete(), but this requires all protocols to
402 * be switched.
403 */
Willy Tarreaua11e9762008-12-01 01:44:25 +0100404 goto out_close;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200405 }
406
407 if ((s = pool_alloc2(pool2_session)) == NULL) {
408 Alert("out of memory in uxst_event_accept().\n");
Willy Tarreaua11e9762008-12-01 01:44:25 +0100409 goto out_close;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200410 }
411
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100412 LIST_ADDQ(&sessions, &s->list);
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100413 LIST_INIT(&s->back_refs);
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100414
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100415 s->flags = 0;
Willy Tarreauf8533202008-08-16 14:55:08 +0200416 s->term_trace = 0;
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100417
Willy Tarreau92fb9832007-10-16 17:34:28 +0200418 if ((t = pool_alloc2(pool2_task)) == NULL) {
419 Alert("out of memory in uxst_event_accept().\n");
Willy Tarreaua11e9762008-12-01 01:44:25 +0100420 goto out_free_session;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200421 }
422
423 s->cli_addr = addr;
424
425 /* FIXME: should be checked earlier */
426 if (cfd >= global.maxsock) {
427 Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
Willy Tarreaua11e9762008-12-01 01:44:25 +0100428 goto out_free_task;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200429 }
430
431 if (fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) {
432 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
Willy Tarreaua11e9762008-12-01 01:44:25 +0100433 goto out_free_task;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200434 }
435
Willy Tarreau9789f7b2008-06-24 08:17:16 +0200436 task_init(t);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200437 t->process = l->handler;
438 t->context = s;
Willy Tarreau91e99932008-06-30 07:51:00 +0200439 t->nice = -64; /* we want to boost priority for local stats */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200440
441 s->task = t;
Willy Tarreaub5654f62008-12-07 16:45:10 +0100442 s->listener = l;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200443 s->fe = NULL;
444 s->be = NULL;
445
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100446 s->ana_state = 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200447 s->req = s->rep = NULL; /* will be allocated later */
448
Willy Tarreaua11e9762008-12-01 01:44:25 +0100449 s->si[0].state = s->si[0].prev_state = SI_ST_EST;
450 s->si[0].err_type = SI_ET_NONE;
451 s->si[0].err_loc = NULL;
452 s->si[0].owner = t;
453 s->si[0].shutr = stream_sock_shutr;
454 s->si[0].shutw = stream_sock_shutw;
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100455 s->si[0].chk_rcv = stream_sock_chk_rcv;
456 s->si[0].chk_snd = stream_sock_chk_snd;
Willy Tarreaua11e9762008-12-01 01:44:25 +0100457 s->si[0].fd = cfd;
458 s->si[0].flags = SI_FL_NONE;
459 s->si[0].exp = TICK_ETERNITY;
Willy Tarreaua11e9762008-12-01 01:44:25 +0100460
461 s->si[1].state = s->si[1].prev_state = SI_ST_INI;
462 s->si[1].err_type = SI_ET_NONE;
463 s->si[1].err_loc = NULL;
464 s->si[1].owner = t;
465 s->si[1].shutr = stream_sock_shutr;
466 s->si[1].shutw = stream_sock_shutw;
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100467 s->si[1].chk_rcv = stream_sock_chk_rcv;
468 s->si[1].chk_snd = stream_sock_chk_snd;
Willy Tarreaua11e9762008-12-01 01:44:25 +0100469 s->si[1].exp = TICK_ETERNITY;
470 s->si[1].fd = -1; /* just to help with debugging */
471 s->si[1].flags = SI_FL_NONE;
472
473 s->srv = s->prev_srv = s->srv_conn = NULL;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200474 s->pend_pos = NULL;
475
476 memset(&s->logs, 0, sizeof(s->logs));
477 memset(&s->txn, 0, sizeof(s->txn));
478
Willy Tarreau3dfe6cd2008-12-07 22:29:48 +0100479 s->logs.tv_accept = now; /* corrected date for internal use */
480
Willy Tarreau3e76e722007-10-17 18:57:38 +0200481 s->data_state = DATA_ST_INIT;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200482 s->data_source = DATA_SRC_NONE;
483 s->uniq_id = totalconn;
484
Willy Tarreaua11e9762008-12-01 01:44:25 +0100485 if ((s->req = pool_alloc2(pool2_buffer)) == NULL)
486 goto out_free_task;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200487
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100488 buffer_init(s->req);
Willy Tarreaua11e9762008-12-01 01:44:25 +0100489 s->req->prod = &s->si[0];
490 s->req->cons = &s->si[1];
491 s->si[0].ib = s->si[1].ob = s->req;
492 s->req->flags |= BF_READ_ATTACHED; /* the producer is already connected */
493
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100494 s->req->analysers = l->analysers;
Willy Tarreaua11e9762008-12-01 01:44:25 +0100495
496 s->req->wto = TICK_ETERNITY;
497 s->req->cto = TICK_ETERNITY;
498 s->req->rto = TICK_ETERNITY;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200499
Willy Tarreaua11e9762008-12-01 01:44:25 +0100500 if ((s->rep = pool_alloc2(pool2_buffer)) == NULL)
501 goto out_free_req;
502
Willy Tarreau92fb9832007-10-16 17:34:28 +0200503 buffer_init(s->rep);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200504
Willy Tarreaua11e9762008-12-01 01:44:25 +0100505 s->rep->prod = &s->si[1];
506 s->rep->cons = &s->si[0];
507 s->si[0].ob = s->si[1].ib = s->rep;
508
509 s->rep->rto = TICK_ETERNITY;
510 s->rep->cto = TICK_ETERNITY;
511 s->rep->wto = TICK_ETERNITY;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200512
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200513 s->req->rex = TICK_ETERNITY;
514 s->req->wex = TICK_ETERNITY;
Willy Tarreauffab5b42008-08-17 18:03:28 +0200515 s->req->analyse_exp = TICK_ETERNITY;
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200516 s->rep->rex = TICK_ETERNITY;
517 s->rep->wex = TICK_ETERNITY;
Willy Tarreauffab5b42008-08-17 18:03:28 +0200518 s->rep->analyse_exp = TICK_ETERNITY;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200519
Willy Tarreaua11e9762008-12-01 01:44:25 +0100520 t->expire = TICK_ETERNITY;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200521
Willy Tarreaua11e9762008-12-01 01:44:25 +0100522 if (l->timeout) {
Willy Tarreau92fb9832007-10-16 17:34:28 +0200523 s->req->rto = *l->timeout;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200524 s->rep->wto = *l->timeout;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200525 }
526
Willy Tarreaua11e9762008-12-01 01:44:25 +0100527 fd_insert(cfd);
528 fdtab[cfd].owner = &s->si[0];
Willy Tarreaua11e9762008-12-01 01:44:25 +0100529 fdtab[cfd].state = FD_STREADY;
530 fdtab[cfd].cb[DIR_RD].f = l->proto->read;
531 fdtab[cfd].cb[DIR_RD].b = s->req;
532 fdtab[cfd].cb[DIR_WR].f = l->proto->write;
533 fdtab[cfd].cb[DIR_WR].b = s->rep;
534 fdtab[cfd].peeraddr = (struct sockaddr *)&s->cli_addr;
535 fdtab[cfd].peerlen = sizeof(s->cli_addr);
536
537 EV_FD_SET(cfd, DIR_RD);
538
Willy Tarreaufdccded2008-08-29 18:19:04 +0200539 task_wakeup(t, TASK_WOKEN_INIT);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200540
541 l->nbconn++; /* warning! right now, it's up to the handler to decrease this */
542 if (l->nbconn >= l->maxconn) {
543 EV_FD_CLR(l->fd, DIR_RD);
544 l->state = LI_FULL;
545 }
546 actconn++;
547 totalconn++;
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100548 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200549 return 0;
Willy Tarreaua11e9762008-12-01 01:44:25 +0100550
551 out_free_req:
552 pool_free2(pool2_buffer, s->req);
553 out_free_task:
554 pool_free2(pool2_task, t);
555 out_free_session:
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100556 list_for_each_entry_safe(bref, back, &s->back_refs, users) {
557 LIST_DEL(&bref->users);
558 LIST_ADDQ(&LIST_ELEM(s->list.n, struct session *, list)->back_refs, &bref->users);
559 bref->ref = s->list.n;
560 }
Willy Tarreaua11e9762008-12-01 01:44:25 +0100561 LIST_DEL(&s->list);
562 pool_free2(pool2_session, s);
563 out_close:
564 close(cfd);
565 return 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200566}
567
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100568/* Parses the request line in <cmd> and possibly starts dumping stats on
569 * s->rep with the hijack bit set. Returns 1 if OK, 0 in case of any error.
570 * The line is modified after parsing.
Willy Tarreau92fb9832007-10-16 17:34:28 +0200571 */
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100572int unix_sock_parse_request(struct session *s, char *line)
Willy Tarreau92fb9832007-10-16 17:34:28 +0200573{
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100574 char *args[MAX_UXST_ARGS + 1];
575 int arg;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200576
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100577 while (isspace((unsigned char)*line))
578 line++;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200579
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100580 arg = 0;
581 args[arg] = line;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200582
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100583 while (*line && arg < MAX_UXST_ARGS) {
584 if (isspace((unsigned char)*line)) {
585 *line++ = '\0';
Willy Tarreau92fb9832007-10-16 17:34:28 +0200586
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100587 while (isspace((unsigned char)*line))
588 line++;
589
590 args[++arg] = line;
591 continue;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200592 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100593
594 line++;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200595 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200596
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100597 while (++arg <= MAX_UXST_ARGS)
598 args[arg] = line;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200599
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100600 if (strcmp(args[0], "show") == 0) {
601 if (strcmp(args[1], "stat") == 0) {
602 if (*args[2] && *args[3] && *args[4]) {
603 s->data_ctx.stats.flags |= STAT_BOUND;
604 s->data_ctx.stats.iid = atoi(args[2]);
605 s->data_ctx.stats.type = atoi(args[3]);
606 s->data_ctx.stats.sid = atoi(args[4]);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200607 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100608
609 s->data_ctx.stats.flags |= STAT_SHOW_STAT;
610 s->data_ctx.stats.flags |= STAT_FMT_CSV;
611 s->ana_state = STATS_ST_REP;
Willy Tarreau01bf8672008-12-07 18:03:29 +0100612 buffer_install_hijacker(s, s->rep, stats_dump_raw_to_buffer);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200613 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100614 else if (strcmp(args[1], "info") == 0) {
615 s->data_ctx.stats.flags |= STAT_SHOW_INFO;
616 s->data_ctx.stats.flags |= STAT_FMT_CSV;
617 s->ana_state = STATS_ST_REP;
Willy Tarreau01bf8672008-12-07 18:03:29 +0100618 buffer_install_hijacker(s, s->rep, stats_dump_raw_to_buffer);
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100619 }
Willy Tarreau3dfe6cd2008-12-07 22:29:48 +0100620 else if (strcmp(args[1], "sess") == 0) {
621 s->ana_state = STATS_ST_REP;
622 buffer_install_hijacker(s, s->rep, stats_dump_sess_to_buffer);
623 }
624 else { /* neither "stat" nor "info" nor "sess" */
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100625 return 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200626 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100627 }
628 else { /* not "show" */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200629 return 0;
630 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100631 return 1;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200632}
633
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100634/* Processes the stats interpreter on the statistics socket.
635 * In order to ease the transition, we simply simulate the server status
636 * for now. It only knows states STATS_ST_INIT, STATS_ST_REQ, STATS_ST_REP, and
637 * STATS_ST_CLOSE. It removes the AN_REQ_UNIX_STATS bit from req->analysers
638 * once done. It always returns 0.
Willy Tarreau92fb9832007-10-16 17:34:28 +0200639 */
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100640int uxst_req_analyser_stats(struct session *s, struct buffer *req)
Willy Tarreau92fb9832007-10-16 17:34:28 +0200641{
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100642 char *line, *p;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200643
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100644 switch (s->ana_state) {
645 case STATS_ST_INIT:
646 /* Stats output not initialized yet */
647 memset(&s->data_ctx.stats, 0, sizeof(s->data_ctx.stats));
648 s->data_source = DATA_SRC_STATS;
649 s->ana_state = STATS_ST_REQ;
650 /* fall through */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200651
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100652 case STATS_ST_REQ:
653 /* Now, stats are initialized, hijack is not set, and
654 * we are waiting for a complete request line.
655 */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200656
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100657 line = s->req->data;
658 p = memchr(line, '\n', s->req->l);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200659
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100660 if (p) {
661 *p = '\0';
662 if (!unix_sock_parse_request(s, line)) {
663 /* invalid request */
664 buffer_shutw_now(s->req);
665 s->ana_state = 0;
666 req->analysers = 0;
667 return 0;
668 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200669 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200670
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100671 /* processing a valid or incomplete request */
672 if ((req->flags & BF_FULL) || /* invalid request */
673 (req->flags & BF_READ_ERROR) || /* input error */
674 (req->flags & BF_READ_TIMEOUT) || /* read timeout */
675 tick_is_expired(req->analyse_exp, now_ms) || /* request timeout */
676 (req->flags & BF_SHUTR)) { /* input closed */
677 buffer_shutw_now(s->req);
678 s->ana_state = 0;
679 req->analysers = 0;
680 return 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200681 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200682
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100683 /* don't forward nor abort */
684 buffer_write_dis(req);
685 return 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200686
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100687 case STATS_ST_REP:
688 /* do nothing while response is being processed */
689 buffer_write_dis(s->req);
690 return 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200691
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100692 case STATS_ST_CLOSE:
693 /* end of dump */
694 s->req->analysers &= ~AN_REQ_UNIX_STATS;
695 s->ana_state = 0;
696 break;
697 }
698 return 0;
699}
Willy Tarreau92fb9832007-10-16 17:34:28 +0200700
Willy Tarreau92fb9832007-10-16 17:34:28 +0200701
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100702/* This function is the unix-stream equivalent of the global process_session().
703 * It is currently limited to unix-stream processing on control sockets such as
704 * stats, and has no server-side. The two functions should be merged into one
705 * once client and server sides are better delimited. Note that the server-side
706 * still exists but remains in SI_ST_INI state forever, so that any call is a
707 * NOP.
708 */
709void uxst_process_session(struct task *t, int *next)
710{
711 struct session *s = t->context;
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100712 int resync;
713 unsigned int rqf_last, rpf_last;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200714
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100715 /* 1a: Check for low level timeouts if needed. We just set a flag on
716 * stream interfaces when their timeouts have expired.
717 */
718 if (unlikely(t->state & TASK_WOKEN_TIMER)) {
719 stream_int_check_timeouts(&s->si[0]);
720 buffer_check_timeouts(s->req);
721 buffer_check_timeouts(s->rep);
722 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200723
Willy Tarreau86491c32008-12-14 09:04:47 +0100724 s->req->flags &= ~BF_READ_NOEXP;
725
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100726 /* copy req/rep flags so that we can detect shutdowns */
727 rqf_last = s->req->flags;
728 rpf_last = s->rep->flags;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200729
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100730 /* 1b: check for low-level errors reported at the stream interface. */
731 if (unlikely(s->si[0].flags & SI_FL_ERR)) {
732 if (s->si[0].state == SI_ST_EST || s->si[0].state == SI_ST_DIS) {
733 s->si[0].shutr(&s->si[0]);
734 s->si[0].shutw(&s->si[0]);
735 stream_int_report_error(&s->si[0]);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200736 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100737 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200738
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100739 /* check buffer timeouts, and close the corresponding stream interfaces
740 * for future reads or writes. Note: this will also concern upper layers
741 * but we do not touch any other flag. We must be careful and correctly
742 * detect state changes when calling them.
743 */
744 if (unlikely(s->req->flags & (BF_READ_TIMEOUT|BF_WRITE_TIMEOUT))) {
745 if (s->req->flags & BF_READ_TIMEOUT)
746 s->req->prod->shutr(s->req->prod);
747 if (s->req->flags & BF_WRITE_TIMEOUT)
748 s->req->cons->shutw(s->req->cons);
749 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200750
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100751 if (unlikely(s->rep->flags & (BF_READ_TIMEOUT|BF_WRITE_TIMEOUT))) {
752 if (s->rep->flags & BF_READ_TIMEOUT)
753 s->rep->prod->shutr(s->rep->prod);
754 if (s->rep->flags & BF_WRITE_TIMEOUT)
755 s->rep->cons->shutw(s->rep->cons);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200756 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200757
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100758 /* Check for connection closure */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200759
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100760 resync_stream_interface:
Willy Tarreau92fb9832007-10-16 17:34:28 +0200761
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100762 /* nothing special to be done on client side */
763 if (unlikely(s->req->prod->state == SI_ST_DIS))
764 s->req->prod->state = SI_ST_CLO;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200765
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100766 /*
767 * Note: of the transient states (REQ, CER, DIS), only REQ may remain
768 * at this point.
769 */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200770
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100771 /**** Process layer 7 below ****/
Willy Tarreau92fb9832007-10-16 17:34:28 +0200772
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100773 resync = 0;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200774
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100775 /* Analyse request */
776 if ((s->req->flags & BF_MASK_ANALYSER) ||
777 (s->req->flags ^ rqf_last) & BF_MASK_STATIC) {
778 unsigned int flags = s->req->flags;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200779
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100780 if (s->req->prod->state >= SI_ST_EST) {
781 /* it's up to the analysers to reset write_ena */
782 buffer_write_ena(s->req);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200783
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100784 /* We will call all analysers for which a bit is set in
785 * s->req->analysers, following the bit order from LSB
786 * to MSB. The analysers must remove themselves from
787 * the list when not needed. This while() loop is in
788 * fact a cleaner if().
Willy Tarreau92fb9832007-10-16 17:34:28 +0200789 */
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100790 while (s->req->analysers) {
791 if (s->req->analysers & AN_REQ_UNIX_STATS)
792 if (!uxst_req_analyser_stats(s, s->req))
793 break;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200794
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100795 /* Just make sure that nobody set a wrong flag causing an endless loop */
796 s->req->analysers &= AN_REQ_UNIX_STATS;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200797
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100798 /* we don't want to loop anyway */
799 break;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200800 }
801 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100802 s->req->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
803 flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
804 if (s->req->flags != flags)
805 resync = 1;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200806 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200807
Willy Tarreauf890dc92008-12-13 21:12:26 +0100808 /* if noone is interested in analysing data, let's forward everything */
809 if (!s->req->analysers && !(s->req->flags & BF_HIJACK))
810 s->req->send_max = s->req->l;
811
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100812 /* if noone is interested in analysing data, let's forward everything
813 * and only wake up every 1-2 MB. We still wake up when send_max is
814 * reached though.
815 */
816 if (!s->req->send_max && s->req->prod->state >= SI_ST_EST &&
817 !s->req->analysers && !(s->req->flags & BF_HIJACK)) {
818 if (s->req->to_forward < FORWARD_DEFAULT_SIZE)
Willy Tarreau0abebcc2009-01-08 00:09:41 +0100819 buffer_forward(s->req, FORWARD_DEFAULT_SIZE);
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100820 }
821
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100822 /* reflect what the L7 analysers have seen last */
823 rqf_last = s->req->flags;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200824
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100825 /*
826 * Now forward all shutdown requests between both sides of the buffer
827 */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200828
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100829 /* first, let's check if the request buffer needs to shutdown(write) */
830 if (unlikely((s->req->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_EMPTY|BF_HIJACK|BF_WRITE_ENA|BF_SHUTR)) ==
831 (BF_EMPTY|BF_WRITE_ENA|BF_SHUTR)))
832 buffer_shutw_now(s->req);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200833
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100834 /* shutdown(write) pending */
835 if (unlikely((s->req->flags & (BF_SHUTW|BF_SHUTW_NOW)) == BF_SHUTW_NOW))
836 s->req->cons->shutw(s->req->cons);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200837
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100838 /* shutdown(write) done on server side, we must stop the client too */
839 if (unlikely((s->req->flags & (BF_SHUTW|BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTW &&
840 !s->req->analysers))
841 buffer_shutr_now(s->req);
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100842
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100843 /* shutdown(read) pending */
844 if (unlikely((s->req->flags & (BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTR_NOW))
845 s->req->prod->shutr(s->req->prod);
Krzysztof Piotr Oledzki583bc962007-11-24 22:12:47 +0100846
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100847 /*
848 * Here we want to check if we need to resync or not.
849 */
850 if ((s->req->flags ^ rqf_last) & BF_MASK_STATIC)
851 resync = 1;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200852
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100853 s->req->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200854
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100855 /* according to benchmarks, it makes sense to resync now */
856 if (resync)
857 goto resync_stream_interface;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200858
Willy Tarreau92fb9832007-10-16 17:34:28 +0200859
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100860 /* Analyse response */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200861
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100862 buffer_write_ena(s->rep);
863 if (unlikely(s->rep->flags & BF_HIJACK)) {
864 /* In inject mode, we wake up everytime something has
865 * happened on the write side of the buffer.
866 */
867 unsigned int flags = s->rep->flags;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200868
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100869 if ((s->rep->flags & (BF_WRITE_PARTIAL|BF_WRITE_ERROR|BF_SHUTW)) &&
870 !(s->rep->flags & BF_FULL)) {
Willy Tarreau01bf8672008-12-07 18:03:29 +0100871 s->rep->hijacker(s, s->rep);
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100872 }
873 s->rep->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
874 flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
875 if (s->rep->flags != flags)
876 resync = 1;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200877 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100878 else if ((s->rep->flags & BF_MASK_ANALYSER) ||
879 (s->rep->flags ^ rpf_last) & BF_MASK_STATIC) {
880 unsigned int flags = s->rep->flags;
Willy Tarreau3e76e722007-10-17 18:57:38 +0200881
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100882 if (s->rep->prod->state >= SI_ST_EST) {
883 /* it's up to the analysers to reset write_ena */
884 buffer_write_ena(s->rep);
Willy Tarreau3e76e722007-10-17 18:57:38 +0200885 }
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100886 s->rep->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
887 flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
888 if (s->rep->flags != flags)
889 resync = 1;
890 }
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100891
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100892 /* if noone is interested in analysing data, let's forward everything
893 * and only wake up every 1-2 MB. We still wake up when send_max is
894 * reached though.
895 */
896 if (!s->rep->send_max && s->rep->prod->state >= SI_ST_EST &&
897 !s->rep->analysers && !(s->rep->flags & BF_HIJACK)) {
Willy Tarreau0abebcc2009-01-08 00:09:41 +0100898 if (s->rep->to_forward < FORWARD_DEFAULT_SIZE)
899 buffer_forward(s->rep, FORWARD_DEFAULT_SIZE);
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100900 }
Willy Tarreauf890dc92008-12-13 21:12:26 +0100901
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100902 /* reflect what the L7 analysers have seen last */
903 rpf_last = s->rep->flags;
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100904
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100905 /*
906 * Now forward all shutdown requests between both sides of the buffer
907 */
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100908
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100909 /*
910 * FIXME: this is probably where we should produce error responses.
911 */
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100912
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100913 /* first, let's check if the request buffer needs to shutdown(write) */
914 if (unlikely((s->rep->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_EMPTY|BF_HIJACK|BF_WRITE_ENA|BF_SHUTR)) ==
915 (BF_EMPTY|BF_WRITE_ENA|BF_SHUTR)))
916 buffer_shutw_now(s->rep);
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100917
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100918 /* shutdown(write) pending */
919 if (unlikely((s->rep->flags & (BF_SHUTW|BF_SHUTW_NOW)) == BF_SHUTW_NOW))
920 s->rep->cons->shutw(s->rep->cons);
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100921
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100922 /* shutdown(write) done on the client side, we must stop the server too */
923 if (unlikely((s->rep->flags & (BF_SHUTW|BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTW))
924 buffer_shutr_now(s->rep);
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100925
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100926 /* shutdown(read) pending */
927 if (unlikely((s->rep->flags & (BF_SHUTR|BF_SHUTR_NOW)) == BF_SHUTR_NOW))
928 s->rep->prod->shutr(s->rep->prod);
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100929
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100930 /*
931 * Here we want to check if we need to resync or not.
932 */
933 if ((s->rep->flags ^ rpf_last) & BF_MASK_STATIC)
934 resync = 1;
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100935
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100936 s->rep->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100937
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100938 if (resync)
939 goto resync_stream_interface;
Krzysztof Piotr Oledzki2c6962c2008-03-02 02:42:14 +0100940
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100941 if (likely(s->rep->cons->state != SI_ST_CLO)) {
942 if (s->rep->cons->state == SI_ST_EST)
943 stream_sock_data_finish(s->rep->cons);
Willy Tarreau3e76e722007-10-17 18:57:38 +0200944
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100945 s->req->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
946 s->rep->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT;
947 s->si[0].prev_state = s->si[0].state;
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100948 s->si[0].flags &= ~(SI_FL_ERR|SI_FL_EXP);
Willy Tarreau3e76e722007-10-17 18:57:38 +0200949
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100950 /* Trick: if a request is being waiting for the server to respond,
951 * and if we know the server can timeout, we don't want the timeout
952 * to expire on the client side first, but we're still interested
953 * in passing data from the client to the server (eg: POST). Thus,
954 * we can cancel the client's request timeout if the server's
955 * request timeout is set and the server has not yet sent a response.
956 */
Willy Tarreau92fb9832007-10-16 17:34:28 +0200957
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100958 if ((s->rep->flags & (BF_WRITE_ENA|BF_SHUTR)) == 0 &&
Willy Tarreau86491c32008-12-14 09:04:47 +0100959 (tick_isset(s->req->wex) || tick_isset(s->rep->rex))) {
960 s->req->flags |= BF_READ_NOEXP;
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100961 s->req->rex = TICK_ETERNITY;
Willy Tarreau86491c32008-12-14 09:04:47 +0100962 }
Willy Tarreau92fb9832007-10-16 17:34:28 +0200963
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200964 t->expire = tick_first(tick_first(s->req->rex, s->req->wex),
965 tick_first(s->rep->rex, s->rep->wex));
Willy Tarreaub1356cf2008-12-07 16:06:43 +0100966 if (s->req->analysers)
967 t->expire = tick_first(t->expire, s->req->analyse_exp);
968
969 if (s->si[0].exp)
970 t->expire = tick_first(t->expire, s->si[0].exp);
Willy Tarreau92fb9832007-10-16 17:34:28 +0200971
972 /* restore t to its place in the task list */
973 task_queue(t);
974
975 *next = t->expire;
976 return; /* nothing more to do */
977 }
978
979 actconn--;
Willy Tarreaub5654f62008-12-07 16:45:10 +0100980 if (s->listener) {
981 s->listener->nbconn--;
982 if (s->listener->state == LI_FULL &&
983 s->listener->nbconn < s->listener->maxconn) {
Willy Tarreau92fb9832007-10-16 17:34:28 +0200984 /* we should reactivate the listener */
Willy Tarreaub5654f62008-12-07 16:45:10 +0100985 EV_FD_SET(s->listener->fd, DIR_RD);
986 s->listener->state = LI_READY;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200987 }
988 }
989
990 /* the task MUST not be in the run queue anymore */
991 task_delete(t);
992 session_free(s);
993 task_free(t);
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200994 *next = TICK_ETERNITY;
Willy Tarreau92fb9832007-10-16 17:34:28 +0200995}
996
Willy Tarreau92fb9832007-10-16 17:34:28 +0200997__attribute__((constructor))
998static void __uxst_protocol_init(void)
999{
1000 protocol_register(&proto_unix);
Willy Tarreau92fb9832007-10-16 17:34:28 +02001001}
1002
1003
1004/*
1005 * Local variables:
1006 * c-indent-level: 8
1007 * c-basic-offset: 8
1008 * End:
1009 */