Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 1 | /* |
| 2 | * UNIX SOCK_STREAM protocol layer (uxst) |
| 3 | * |
Willy Tarreau | 7c669d7 | 2008-06-20 15:04:11 +0200 | [diff] [blame] | 4 | * Copyright 2000-2008 Willy Tarreau <w@1wt.eu> |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 5 | * |
| 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 Tarreau | d740bab | 2007-10-28 11:14:07 +0100 | [diff] [blame] | 31 | #include <common/errors.h> |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 32 | #include <common/memory.h> |
| 33 | #include <common/mini-clist.h> |
| 34 | #include <common/standard.h> |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 35 | #include <common/ticks.h> |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 36 | #include <common/time.h> |
| 37 | #include <common/version.h> |
| 38 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 39 | #include <types/global.h> |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 40 | |
| 41 | #include <proto/acl.h> |
| 42 | #include <proto/backend.h> |
| 43 | #include <proto/buffers.h> |
Willy Tarreau | 3e76e72 | 2007-10-17 18:57:38 +0200 | [diff] [blame] | 44 | #include <proto/dumpstats.h> |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 45 | #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 Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 51 | #include <proto/stream_interface.h> |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 52 | #include <proto/stream_sock.h> |
| 53 | #include <proto/task.h> |
| 54 | |
| 55 | #ifndef MAXPATHLEN |
| 56 | #define MAXPATHLEN 128 |
| 57 | #endif |
| 58 | |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 59 | static int uxst_bind_listeners(struct protocol *proto); |
| 60 | static int uxst_unbind_listeners(struct protocol *proto); |
| 61 | |
| 62 | /* Note: must not be declared <const> as its list will be overwritten */ |
| 63 | static 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 87 | /* This function creates a named PF_UNIX stream socket at address <path>. Note |
Willy Tarreau | e6ad2b1 | 2007-10-18 12:45:54 +0200 | [diff] [blame] | 88 | * 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 92 | * It returns the assigned file descriptor, or -1 in the event of an error. |
| 93 | */ |
Willy Tarreau | e6ad2b1 | 2007-10-18 12:45:54 +0200 | [diff] [blame] | 94 | static int create_uxst_socket(const char *path, uid_t uid, gid_t gid, mode_t mode) |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 95 | { |
| 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 Tarreau | e6ad2b1 | 2007-10-18 12:45:54 +0200 | [diff] [blame] | 164 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 170 | 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 | */ |
| 206 | static 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 Tarreau | 10ae548 | 2007-10-18 16:15:52 +0200 | [diff] [blame] | 225 | addr.sun_path[sizeof(addr.sun_path) - 1] = 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 226 | 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 Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 237 | /******************************** |
| 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 | */ |
| 246 | static int uxst_bind_listener(struct listener *listener) |
| 247 | { |
| 248 | int fd; |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 249 | |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 250 | 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 Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 259 | |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 260 | /* 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 Tarreau | eabf313 | 2008-08-29 23:36:51 +0200 | [diff] [blame] | 269 | fdtab[fd].owner = listener; /* reference the listener instead of a task */ |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 270 | fdtab[fd].state = FD_STLISTEN; |
| 271 | fdtab[fd].peeraddr = NULL; |
| 272 | fdtab[fd].peerlen = 0; |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 273 | 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 | */ |
| 279 | static 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 Tarreau | 8eebe5e | 2007-10-28 22:07:08 +0100 | [diff] [blame] | 285 | fd_delete(listener->fd); |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 286 | 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 | */ |
| 296 | void 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 Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 306 | /******************************** |
| 307 | * 3) protocol-oriented functions |
| 308 | ********************************/ |
| 309 | |
| 310 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 311 | /* 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 | */ |
| 319 | static int uxst_bind_listeners(struct protocol *proto) |
| 320 | { |
| 321 | struct listener *listener; |
| 322 | int err = ERR_NONE; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 323 | |
| 324 | list_for_each_entry(listener, &proto->listeners, proto_list) { |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 325 | err |= uxst_bind_listener(listener); |
| 326 | if (err != ERR_NONE) |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 327 | continue; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 328 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 329 | return err; |
| 330 | } |
| 331 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 332 | |
| 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 | */ |
| 337 | static int uxst_unbind_listeners(struct protocol *proto) |
| 338 | { |
| 339 | struct listener *listener; |
| 340 | |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 341 | list_for_each_entry(listener, &proto->listeners, proto_list) |
| 342 | uxst_unbind_listener(listener); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 343 | return ERR_NONE; |
| 344 | } |
| 345 | |
Willy Tarreau | dabf2e2 | 2007-10-28 21:59:24 +0100 | [diff] [blame] | 346 | |
| 347 | /******************************** |
| 348 | * 4) high-level functions |
| 349 | ********************************/ |
| 350 | |
| 351 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 352 | /* |
| 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 | */ |
| 359 | int uxst_event_accept(int fd) { |
Willy Tarreau | eabf313 | 2008-08-29 23:36:51 +0200 | [diff] [blame] | 360 | struct listener *l = fdtab[fd].owner; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 361 | struct session *s; |
| 362 | struct task *t; |
Willy Tarreau | 62e4f1d | 2008-12-07 20:16:23 +0100 | [diff] [blame] | 363 | struct bref *bref, *back; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 364 | 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 Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 404 | goto out_close; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 405 | } |
| 406 | |
| 407 | if ((s = pool_alloc2(pool2_session)) == NULL) { |
| 408 | Alert("out of memory in uxst_event_accept().\n"); |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 409 | goto out_close; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 410 | } |
| 411 | |
Willy Tarreau | f54f8bd | 2008-11-23 19:53:55 +0100 | [diff] [blame] | 412 | LIST_ADDQ(&sessions, &s->list); |
Willy Tarreau | 62e4f1d | 2008-12-07 20:16:23 +0100 | [diff] [blame] | 413 | LIST_INIT(&s->back_refs); |
Willy Tarreau | f54f8bd | 2008-11-23 19:53:55 +0100 | [diff] [blame] | 414 | |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 415 | s->flags = 0; |
Willy Tarreau | f853320 | 2008-08-16 14:55:08 +0200 | [diff] [blame] | 416 | s->term_trace = 0; |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 417 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 418 | if ((t = pool_alloc2(pool2_task)) == NULL) { |
| 419 | Alert("out of memory in uxst_event_accept().\n"); |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 420 | goto out_free_session; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 421 | } |
| 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 Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 428 | goto out_free_task; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 429 | } |
| 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 Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 433 | goto out_free_task; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 434 | } |
| 435 | |
Willy Tarreau | 9789f7b | 2008-06-24 08:17:16 +0200 | [diff] [blame] | 436 | task_init(t); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 437 | t->process = l->handler; |
| 438 | t->context = s; |
Willy Tarreau | 91e9993 | 2008-06-30 07:51:00 +0200 | [diff] [blame] | 439 | t->nice = -64; /* we want to boost priority for local stats */ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 440 | |
| 441 | s->task = t; |
Willy Tarreau | b5654f6 | 2008-12-07 16:45:10 +0100 | [diff] [blame] | 442 | s->listener = l; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 443 | s->fe = NULL; |
| 444 | s->be = NULL; |
| 445 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 446 | s->ana_state = 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 447 | s->req = s->rep = NULL; /* will be allocated later */ |
| 448 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 449 | 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 Tarreau | 3ffeba1 | 2008-12-14 14:42:35 +0100 | [diff] [blame] | 455 | s->si[0].chk_rcv = stream_sock_chk_rcv; |
| 456 | s->si[0].chk_snd = stream_sock_chk_snd; |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 457 | s->si[0].fd = cfd; |
| 458 | s->si[0].flags = SI_FL_NONE; |
| 459 | s->si[0].exp = TICK_ETERNITY; |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 460 | |
| 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 Tarreau | 3ffeba1 | 2008-12-14 14:42:35 +0100 | [diff] [blame] | 467 | s->si[1].chk_rcv = stream_sock_chk_rcv; |
| 468 | s->si[1].chk_snd = stream_sock_chk_snd; |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 469 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 474 | s->pend_pos = NULL; |
| 475 | |
| 476 | memset(&s->logs, 0, sizeof(s->logs)); |
| 477 | memset(&s->txn, 0, sizeof(s->txn)); |
| 478 | |
Willy Tarreau | 3dfe6cd | 2008-12-07 22:29:48 +0100 | [diff] [blame] | 479 | s->logs.tv_accept = now; /* corrected date for internal use */ |
| 480 | |
Willy Tarreau | 3e76e72 | 2007-10-17 18:57:38 +0200 | [diff] [blame] | 481 | s->data_state = DATA_ST_INIT; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 482 | s->data_source = DATA_SRC_NONE; |
| 483 | s->uniq_id = totalconn; |
| 484 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 485 | if ((s->req = pool_alloc2(pool2_buffer)) == NULL) |
| 486 | goto out_free_task; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 487 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 488 | buffer_init(s->req); |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 489 | 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 Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 494 | s->req->analysers = l->analysers; |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 495 | |
| 496 | s->req->wto = TICK_ETERNITY; |
| 497 | s->req->cto = TICK_ETERNITY; |
| 498 | s->req->rto = TICK_ETERNITY; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 499 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 500 | if ((s->rep = pool_alloc2(pool2_buffer)) == NULL) |
| 501 | goto out_free_req; |
| 502 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 503 | buffer_init(s->rep); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 504 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 505 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 512 | |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 513 | s->req->rex = TICK_ETERNITY; |
| 514 | s->req->wex = TICK_ETERNITY; |
Willy Tarreau | ffab5b4 | 2008-08-17 18:03:28 +0200 | [diff] [blame] | 515 | s->req->analyse_exp = TICK_ETERNITY; |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 516 | s->rep->rex = TICK_ETERNITY; |
| 517 | s->rep->wex = TICK_ETERNITY; |
Willy Tarreau | ffab5b4 | 2008-08-17 18:03:28 +0200 | [diff] [blame] | 518 | s->rep->analyse_exp = TICK_ETERNITY; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 519 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 520 | t->expire = TICK_ETERNITY; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 521 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 522 | if (l->timeout) { |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 523 | s->req->rto = *l->timeout; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 524 | s->rep->wto = *l->timeout; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 525 | } |
| 526 | |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 527 | fd_insert(cfd); |
| 528 | fdtab[cfd].owner = &s->si[0]; |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 529 | 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 Tarreau | fdccded | 2008-08-29 18:19:04 +0200 | [diff] [blame] | 539 | task_wakeup(t, TASK_WOKEN_INIT); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 540 | |
| 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 Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 548 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 549 | return 0; |
Willy Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 550 | |
| 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 Tarreau | 62e4f1d | 2008-12-07 20:16:23 +0100 | [diff] [blame] | 556 | 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 Tarreau | a11e976 | 2008-12-01 01:44:25 +0100 | [diff] [blame] | 561 | LIST_DEL(&s->list); |
| 562 | pool_free2(pool2_session, s); |
| 563 | out_close: |
| 564 | close(cfd); |
| 565 | return 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 566 | } |
| 567 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 568 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 571 | */ |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 572 | int unix_sock_parse_request(struct session *s, char *line) |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 573 | { |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 574 | char *args[MAX_UXST_ARGS + 1]; |
| 575 | int arg; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 576 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 577 | while (isspace((unsigned char)*line)) |
| 578 | line++; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 579 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 580 | arg = 0; |
| 581 | args[arg] = line; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 582 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 583 | while (*line && arg < MAX_UXST_ARGS) { |
| 584 | if (isspace((unsigned char)*line)) { |
| 585 | *line++ = '\0'; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 586 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 587 | while (isspace((unsigned char)*line)) |
| 588 | line++; |
| 589 | |
| 590 | args[++arg] = line; |
| 591 | continue; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 592 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 593 | |
| 594 | line++; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 595 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 596 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 597 | while (++arg <= MAX_UXST_ARGS) |
| 598 | args[arg] = line; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 599 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 600 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 607 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 608 | |
| 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 Tarreau | 01bf867 | 2008-12-07 18:03:29 +0100 | [diff] [blame] | 612 | buffer_install_hijacker(s, s->rep, stats_dump_raw_to_buffer); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 613 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 614 | 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 Tarreau | 01bf867 | 2008-12-07 18:03:29 +0100 | [diff] [blame] | 618 | buffer_install_hijacker(s, s->rep, stats_dump_raw_to_buffer); |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 619 | } |
Willy Tarreau | 3dfe6cd | 2008-12-07 22:29:48 +0100 | [diff] [blame] | 620 | 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 Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 625 | return 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 626 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 627 | } |
| 628 | else { /* not "show" */ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 629 | return 0; |
| 630 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 631 | return 1; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 632 | } |
| 633 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 634 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 639 | */ |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 640 | int uxst_req_analyser_stats(struct session *s, struct buffer *req) |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 641 | { |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 642 | char *line, *p; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 643 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 644 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 651 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 652 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 656 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 657 | line = s->req->data; |
| 658 | p = memchr(line, '\n', s->req->l); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 659 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 660 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 669 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 670 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 671 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 681 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 682 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 683 | /* don't forward nor abort */ |
| 684 | buffer_write_dis(req); |
| 685 | return 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 686 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 687 | case STATS_ST_REP: |
| 688 | /* do nothing while response is being processed */ |
| 689 | buffer_write_dis(s->req); |
| 690 | return 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 691 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 692 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 700 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 701 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 702 | /* 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 | */ |
| 709 | void uxst_process_session(struct task *t, int *next) |
| 710 | { |
| 711 | struct session *s = t->context; |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 712 | int resync; |
| 713 | unsigned int rqf_last, rpf_last; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 714 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 715 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 723 | |
Willy Tarreau | 86491c3 | 2008-12-14 09:04:47 +0100 | [diff] [blame] | 724 | s->req->flags &= ~BF_READ_NOEXP; |
| 725 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 726 | /* copy req/rep flags so that we can detect shutdowns */ |
| 727 | rqf_last = s->req->flags; |
| 728 | rpf_last = s->rep->flags; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 729 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 730 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 736 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 737 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 738 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 739 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 750 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 751 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 756 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 757 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 758 | /* Check for connection closure */ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 759 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 760 | resync_stream_interface: |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 761 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 762 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 765 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 766 | /* |
| 767 | * Note: of the transient states (REQ, CER, DIS), only REQ may remain |
| 768 | * at this point. |
| 769 | */ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 770 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 771 | /**** Process layer 7 below ****/ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 772 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 773 | resync = 0; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 774 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 775 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 779 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 780 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 783 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 784 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 789 | */ |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 790 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 794 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 795 | /* Just make sure that nobody set a wrong flag causing an endless loop */ |
| 796 | s->req->analysers &= AN_REQ_UNIX_STATS; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 797 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 798 | /* we don't want to loop anyway */ |
| 799 | break; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 800 | } |
| 801 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 802 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 806 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 807 | |
Willy Tarreau | f890dc9 | 2008-12-13 21:12:26 +0100 | [diff] [blame] | 808 | /* 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 Tarreau | 6b66f3e | 2008-12-14 17:31:54 +0100 | [diff] [blame] | 812 | /* 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 Tarreau | 0abebcc | 2009-01-08 00:09:41 +0100 | [diff] [blame] | 819 | buffer_forward(s->req, FORWARD_DEFAULT_SIZE); |
Willy Tarreau | 6b66f3e | 2008-12-14 17:31:54 +0100 | [diff] [blame] | 820 | } |
| 821 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 822 | /* reflect what the L7 analysers have seen last */ |
| 823 | rqf_last = s->req->flags; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 824 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 825 | /* |
| 826 | * Now forward all shutdown requests between both sides of the buffer |
| 827 | */ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 828 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 829 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 833 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 834 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 837 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 838 | /* 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 Oledzki | 583bc96 | 2007-11-24 22:12:47 +0100 | [diff] [blame] | 842 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 843 | /* 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 Oledzki | 583bc96 | 2007-11-24 22:12:47 +0100 | [diff] [blame] | 846 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 847 | /* |
| 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 852 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 853 | s->req->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 854 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 855 | /* according to benchmarks, it makes sense to resync now */ |
| 856 | if (resync) |
| 857 | goto resync_stream_interface; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 858 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 859 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 860 | /* Analyse response */ |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 861 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 862 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 868 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 869 | if ((s->rep->flags & (BF_WRITE_PARTIAL|BF_WRITE_ERROR|BF_SHUTW)) && |
| 870 | !(s->rep->flags & BF_FULL)) { |
Willy Tarreau | 01bf867 | 2008-12-07 18:03:29 +0100 | [diff] [blame] | 871 | s->rep->hijacker(s, s->rep); |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 872 | } |
| 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 877 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 878 | 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 Tarreau | 3e76e72 | 2007-10-17 18:57:38 +0200 | [diff] [blame] | 881 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 882 | 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 Tarreau | 3e76e72 | 2007-10-17 18:57:38 +0200 | [diff] [blame] | 885 | } |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 886 | 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 Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 891 | |
Willy Tarreau | 6b66f3e | 2008-12-14 17:31:54 +0100 | [diff] [blame] | 892 | /* 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 Tarreau | 0abebcc | 2009-01-08 00:09:41 +0100 | [diff] [blame] | 898 | if (s->rep->to_forward < FORWARD_DEFAULT_SIZE) |
| 899 | buffer_forward(s->rep, FORWARD_DEFAULT_SIZE); |
Willy Tarreau | 6b66f3e | 2008-12-14 17:31:54 +0100 | [diff] [blame] | 900 | } |
Willy Tarreau | f890dc9 | 2008-12-13 21:12:26 +0100 | [diff] [blame] | 901 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 902 | /* reflect what the L7 analysers have seen last */ |
| 903 | rpf_last = s->rep->flags; |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 904 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 905 | /* |
| 906 | * Now forward all shutdown requests between both sides of the buffer |
| 907 | */ |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 908 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 909 | /* |
| 910 | * FIXME: this is probably where we should produce error responses. |
| 911 | */ |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 912 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 913 | /* 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 Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 917 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 918 | /* 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 Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 921 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 922 | /* 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 Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 925 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 926 | /* 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 Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 929 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 930 | /* |
| 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 Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 935 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 936 | s->rep->flags &= BF_CLEAR_READ & BF_CLEAR_WRITE & BF_CLEAR_TIMEOUT; |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 937 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 938 | if (resync) |
| 939 | goto resync_stream_interface; |
Krzysztof Piotr Oledzki | 2c6962c | 2008-03-02 02:42:14 +0100 | [diff] [blame] | 940 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 941 | 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 Tarreau | 3e76e72 | 2007-10-17 18:57:38 +0200 | [diff] [blame] | 944 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 945 | 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 Tarreau | b0ef735 | 2008-12-14 13:26:20 +0100 | [diff] [blame] | 948 | s->si[0].flags &= ~(SI_FL_ERR|SI_FL_EXP); |
Willy Tarreau | 3e76e72 | 2007-10-17 18:57:38 +0200 | [diff] [blame] | 949 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 950 | /* 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 957 | |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 958 | if ((s->rep->flags & (BF_WRITE_ENA|BF_SHUTR)) == 0 && |
Willy Tarreau | 86491c3 | 2008-12-14 09:04:47 +0100 | [diff] [blame] | 959 | (tick_isset(s->req->wex) || tick_isset(s->rep->rex))) { |
| 960 | s->req->flags |= BF_READ_NOEXP; |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 961 | s->req->rex = TICK_ETERNITY; |
Willy Tarreau | 86491c3 | 2008-12-14 09:04:47 +0100 | [diff] [blame] | 962 | } |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 963 | |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 964 | t->expire = tick_first(tick_first(s->req->rex, s->req->wex), |
| 965 | tick_first(s->rep->rex, s->rep->wex)); |
Willy Tarreau | b1356cf | 2008-12-07 16:06:43 +0100 | [diff] [blame] | 966 | 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 Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 971 | |
| 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 Tarreau | b5654f6 | 2008-12-07 16:45:10 +0100 | [diff] [blame] | 980 | if (s->listener) { |
| 981 | s->listener->nbconn--; |
| 982 | if (s->listener->state == LI_FULL && |
| 983 | s->listener->nbconn < s->listener->maxconn) { |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 984 | /* we should reactivate the listener */ |
Willy Tarreau | b5654f6 | 2008-12-07 16:45:10 +0100 | [diff] [blame] | 985 | EV_FD_SET(s->listener->fd, DIR_RD); |
| 986 | s->listener->state = LI_READY; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 987 | } |
| 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 Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 994 | *next = TICK_ETERNITY; |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 995 | } |
| 996 | |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 997 | __attribute__((constructor)) |
| 998 | static void __uxst_protocol_init(void) |
| 999 | { |
| 1000 | protocol_register(&proto_unix); |
Willy Tarreau | 92fb983 | 2007-10-16 17:34:28 +0200 | [diff] [blame] | 1001 | } |
| 1002 | |
| 1003 | |
| 1004 | /* |
| 1005 | * Local variables: |
| 1006 | * c-indent-level: 8 |
| 1007 | * c-basic-offset: 8 |
| 1008 | * End: |
| 1009 | */ |