blob: ef46dfefd2c10fce67a69bb50d648c2cccf1c618 [file] [log] [blame]
willy tarreau0f7af912005-12-17 12:21:26 +01001/*
willy tarreau5cbea6f2005-12-17 12:48:26 +01002 * HA-Proxy : High Availability-enabled HTTP/TCP proxy
willy tarreau036e1ce2005-12-17 13:46:33 +01003 * 2000-2003 - Willy Tarreau - willy AT meta-x DOT org.
willy tarreau0f7af912005-12-17 12:21:26 +01004 *
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU General Public License
7 * as published by the Free Software Foundation; either version
8 * 2 of the License, or (at your option) any later version.
9 *
willy tarreau906b2682005-12-17 13:49:52 +010010 * Please refer to RFC2068 or RFC2616 for informations about HTTP protocol, and
11 * RFC2965 for informations about cookies usage.
12 *
13 * Pending bugs (may be not fixed because never reproduced) :
willy tarreaua1598082005-12-17 13:08:06 +010014 * - solaris only : sometimes, an HTTP proxy with only a dispatch address causes
15 * the proxy to terminate (no core) if the client breaks the connection during
willy tarreauc29948c2005-12-17 13:10:27 +010016 * the response. Seen on 1.1.8pre4, but never reproduced. May not be related to
willy tarreau8337c6b2005-12-17 13:41:01 +010017 * the snprintf() bug since requests were simple (GET / HTTP/1.0), but may be
18 * related to missing setsid() (fixed in 1.1.15)
willy tarreauef900ab2005-12-17 12:52:52 +010019 * - a proxy with an invalid config will prevent the startup even if disabled.
20 *
willy tarreau036e1ce2005-12-17 13:46:33 +010021 * ChangeLog has moved to the CHANGELOG file.
willy tarreau0f7af912005-12-17 12:21:26 +010022 *
willy tarreau5cbea6f2005-12-17 12:48:26 +010023 * TODO:
24 * - handle properly intermediate incomplete server headers. Done ?
willy tarreau5cbea6f2005-12-17 12:48:26 +010025 * - handle hot-reconfiguration
willy tarreau906b2682005-12-17 13:49:52 +010026 * - fix client/server state transition when server is in connect or headers state
27 * and client suddenly disconnects. The server *should* switch to SHUT_WR, but
28 * still handle HTTP headers.
willy tarreau0f7af912005-12-17 12:21:26 +010029 *
30 */
31
32#include <stdio.h>
33#include <stdlib.h>
34#include <unistd.h>
35#include <string.h>
36#include <ctype.h>
37#include <sys/time.h>
38#include <sys/types.h>
39#include <sys/socket.h>
40#include <netinet/tcp.h>
41#include <netinet/in.h>
42#include <arpa/inet.h>
43#include <netdb.h>
44#include <fcntl.h>
45#include <errno.h>
46#include <signal.h>
47#include <stdarg.h>
48#include <sys/resource.h>
49#include <time.h>
50#include <regex.h>
51#include <syslog.h>
willy tarreaua1598082005-12-17 13:08:06 +010052#if defined(TPROXY) && defined(NETFILTER)
willy tarreau5cbea6f2005-12-17 12:48:26 +010053#include <linux/netfilter_ipv4.h>
54#endif
willy tarreau0f7af912005-12-17 12:21:26 +010055
willy tarreauc1cae632005-12-17 14:12:23 +010056#define HAPROXY_VERSION "1.1.25"
57#define HAPROXY_DATE "2003/10/15"
willy tarreau0f7af912005-12-17 12:21:26 +010058
59/* this is for libc5 for example */
60#ifndef TCP_NODELAY
61#define TCP_NODELAY 1
62#endif
63
64#ifndef SHUT_RD
65#define SHUT_RD 0
66#endif
67
68#ifndef SHUT_WR
69#define SHUT_WR 1
70#endif
71
willy tarreau535ae7a2005-12-17 12:58:00 +010072#define BUFSIZE 8192
willy tarreau0f7af912005-12-17 12:21:26 +010073
74// reserved buffer space for header rewriting
willy tarreau535ae7a2005-12-17 12:58:00 +010075#define MAXREWRITE 4096
willy tarreau9fe663a2005-12-17 13:02:59 +010076#define REQURI_LEN 1024
willy tarreau8337c6b2005-12-17 13:41:01 +010077#define CAPTURE_LEN 64
willy tarreau0f7af912005-12-17 12:21:26 +010078
willy tarreau5cbea6f2005-12-17 12:48:26 +010079// max # args on a configuration line
willy tarreaue39cd132005-12-17 13:00:18 +010080#define MAX_LINE_ARGS 40
willy tarreau5cbea6f2005-12-17 12:48:26 +010081
willy tarreaue39cd132005-12-17 13:00:18 +010082// max # of added headers per request
83#define MAX_NEWHDR 10
willy tarreau0f7af912005-12-17 12:21:26 +010084
85// max # of matches per regexp
86#define MAX_MATCH 10
87
willy tarreau5cbea6f2005-12-17 12:48:26 +010088/* FIXME: serverid_len and cookiename_len are no longer checked in configuration file */
willy tarreau0f7af912005-12-17 12:21:26 +010089#define COOKIENAME_LEN 16
90#define SERVERID_LEN 16
91#define CONN_RETRIES 3
92
willy tarreau5cbea6f2005-12-17 12:48:26 +010093#define CHK_CONNTIME 2000
willy tarreaue47c8d72005-12-17 12:55:52 +010094#define DEF_CHKINTR 2000
95#define DEF_FALLTIME 3
96#define DEF_RISETIME 2
willy tarreau2f6ba652005-12-17 13:57:42 +010097#define DEF_CHECK_REQ "OPTIONS / HTTP/1.0\r\n\r\n"
willy tarreau5cbea6f2005-12-17 12:48:26 +010098
willy tarreau9fe663a2005-12-17 13:02:59 +010099/* default connections limit */
100#define DEFAULT_MAXCONN 2000
101
willy tarreau0f7af912005-12-17 12:21:26 +0100102/* how many bits are needed to code the size of an int (eg: 32bits -> 5) */
103#define INTBITS 5
104
105/* show stats this every millisecond, 0 to disable */
106#ifndef STATTIME
107#define STATTIME 2000
108#endif
109
willy tarreau5cbea6f2005-12-17 12:48:26 +0100110/* this reduces the number of calls to select() by choosing appropriate
111 * sheduler precision in milliseconds. It should be near the minimum
112 * time that is needed by select() to collect all events. All timeouts
113 * are rounded up by adding this value prior to pass it to select().
114 */
115#define SCHEDULER_RESOLUTION 9
116
willy tarreau0f7af912005-12-17 12:21:26 +0100117#define MINTIME(old, new) (((new)<0)?(old):(((old)<0||(new)<(old))?(new):(old)))
118#define SETNOW(a) (*a=now)
119
willy tarreau9da061b2005-12-17 12:29:56 +0100120/****** string-specific macros and functions ******/
121/* if a > max, then bound <a> to <max>. The macro returns the new <a> */
122#define UBOUND(a, max) ({ typeof(a) b = (max); if ((a) > b) (a) = b; (a); })
123
124/* if a < min, then bound <a> to <min>. The macro returns the new <a> */
125#define LBOUND(a, min) ({ typeof(a) b = (min); if ((a) < b) (a) = b; (a); })
126
willy tarreau9da061b2005-12-17 12:29:56 +0100127/*
128 * copies at most <size-1> chars from <src> to <dst>. Last char is always
129 * set to 0, unless <size> is 0. The number of chars copied is returned
130 * (excluding the terminating zero).
131 * This code has been optimized for size and speed : on x86, it's 45 bytes
132 * long, uses only registers, and consumes only 4 cycles per char.
133 */
willy tarreau750a4722005-12-17 13:21:24 +0100134int strlcpy2(char *dst, const char *src, int size) {
willy tarreau9da061b2005-12-17 12:29:56 +0100135 char *orig = dst;
136 if (size) {
137 while (--size && (*dst = *src)) {
138 src++; dst++;
139 }
140 *dst = 0;
141 }
142 return dst - orig;
143}
willy tarreau9da061b2005-12-17 12:29:56 +0100144
willy tarreau0f7af912005-12-17 12:21:26 +0100145#define MEM_OPTIM
146#ifdef MEM_OPTIM
147/*
148 * Returns a pointer to type <type> taken from the
149 * pool <pool_type> or dynamically allocated. In the
150 * first case, <pool_type> is updated to point to the
151 * next element in the list.
152 */
153#define pool_alloc(type) ({ \
154 void *p; \
155 if ((p = pool_##type) == NULL) \
156 p = malloc(sizeof_##type); \
157 else { \
158 pool_##type = *(void **)pool_##type; \
159 } \
160 p; \
161})
162
163/*
164 * Puts a memory area back to the corresponding pool.
165 * Items are chained directly through a pointer that
166 * is written in the beginning of the memory area, so
willy tarreau9da061b2005-12-17 12:29:56 +0100167 * there's no need for any carrier cell. This implies
willy tarreau0f7af912005-12-17 12:21:26 +0100168 * that each memory area is at least as big as one
169 * pointer.
170 */
171#define pool_free(type, ptr) ({ \
172 *(void **)ptr = (void *)pool_##type; \
173 pool_##type = (void *)ptr; \
174})
175
176#else
177#define pool_alloc(type) (calloc(1,sizeof_##type));
178#define pool_free(type, ptr) (free(ptr));
179#endif /* MEM_OPTIM */
180
willy tarreau5cbea6f2005-12-17 12:48:26 +0100181#define sizeof_task sizeof(struct task)
182#define sizeof_session sizeof(struct session)
willy tarreau0f7af912005-12-17 12:21:26 +0100183#define sizeof_buffer sizeof(struct buffer)
184#define sizeof_fdtab sizeof(struct fdtab)
willy tarreau9fe663a2005-12-17 13:02:59 +0100185#define sizeof_requri REQURI_LEN
willy tarreau8337c6b2005-12-17 13:41:01 +0100186#define sizeof_capture CAPTURE_LEN
willy tarreau0f7af912005-12-17 12:21:26 +0100187
willy tarreau5cbea6f2005-12-17 12:48:26 +0100188/* different possible states for the sockets */
willy tarreau0f7af912005-12-17 12:21:26 +0100189#define FD_STCLOSE 0
190#define FD_STLISTEN 1
191#define FD_STCONN 2
192#define FD_STREADY 3
193#define FD_STERROR 4
194
willy tarreau5cbea6f2005-12-17 12:48:26 +0100195/* values for task->state */
willy tarreau0f7af912005-12-17 12:21:26 +0100196#define TASK_IDLE 0
197#define TASK_RUNNING 1
198
willy tarreau5cbea6f2005-12-17 12:48:26 +0100199/* values for proxy->state */
willy tarreau0f7af912005-12-17 12:21:26 +0100200#define PR_STNEW 0
201#define PR_STIDLE 1
202#define PR_STRUN 2
203#define PR_STDISABLED 3
204
willy tarreau5cbea6f2005-12-17 12:48:26 +0100205/* values for proxy->mode */
willy tarreau0f7af912005-12-17 12:21:26 +0100206#define PR_MODE_TCP 0
207#define PR_MODE_HTTP 1
208#define PR_MODE_HEALTH 2
209
willy tarreau5cbea6f2005-12-17 12:48:26 +0100210/* bits for proxy->options */
211#define PR_O_REDISP 1 /* allow reconnection to dispatch in case of errors */
212#define PR_O_TRANSP 2 /* transparent mode : use original DEST as dispatch */
213#define PR_O_COOK_RW 4 /* rewrite all direct cookies with the right serverid */
214#define PR_O_COOK_IND 8 /* keep only indirect cookies */
215#define PR_O_COOK_INS 16 /* insert cookies when not accessing a server directly */
216#define PR_O_COOK_ANY (PR_O_COOK_RW | PR_O_COOK_IND | PR_O_COOK_INS)
217#define PR_O_BALANCE_RR 32 /* balance in round-robin mode */
218#define PR_O_BALANCE (PR_O_BALANCE_RR)
willy tarreau9fe663a2005-12-17 13:02:59 +0100219#define PR_O_KEEPALIVE 64 /* follow keep-alive sessions */
220#define PR_O_FWDFOR 128 /* insert x-forwarded-for with client address */
willy tarreaua1598082005-12-17 13:08:06 +0100221#define PR_O_BIND_SRC 256 /* bind to a specific source address when connect()ing */
222#define PR_O_NULLNOLOG 512 /* a connect without request will not be logged */
willy tarreau240afa62005-12-17 13:14:35 +0100223#define PR_O_COOK_NOC 1024 /* add a 'Cache-control' header with the cookie */
willy tarreaucd878942005-12-17 13:27:43 +0100224#define PR_O_COOK_POST 2048 /* don't insert cookies for requests other than a POST */
willy tarreaubc4e1fb2005-12-17 13:32:07 +0100225#define PR_O_HTTP_CHK 4096 /* use HTTP 'OPTIONS' method to check server health */
willy tarreau8337c6b2005-12-17 13:41:01 +0100226#define PR_O_PERSIST 8192 /* server persistence stays effective even when server is down */
willy tarreau9fe663a2005-12-17 13:02:59 +0100227
willy tarreau5cbea6f2005-12-17 12:48:26 +0100228
willy tarreaue39cd132005-12-17 13:00:18 +0100229/* various session flags */
willy tarreau036e1ce2005-12-17 13:46:33 +0100230#define SN_DIRECT 0x00000001 /* connection made on the server matching the client cookie */
231#define SN_CLDENY 0x00000002 /* a client header matches a deny regex */
232#define SN_CLALLOW 0x00000004 /* a client header matches an allow regex */
233#define SN_SVDENY 0x00000008 /* a server header matches a deny regex */
234#define SN_SVALLOW 0x00000010 /* a server header matches an allow regex */
235#define SN_POST 0x00000020 /* the request was an HTTP POST */
236
237#define SN_CK_NONE 0x00000000 /* this session had no cookie */
238#define SN_CK_INVALID 0x00000040 /* this session had a cookie which matches no server */
239#define SN_CK_DOWN 0x00000080 /* this session had cookie matching a down server */
240#define SN_CK_VALID 0x000000C0 /* this session had cookie matching a valid server */
241#define SN_CK_MASK 0x000000C0 /* mask to get this session's cookie flags */
242#define SN_CK_SHIFT 6 /* bit shift */
243
244#define SN_ERR_CLITO 0x00000100 /* client time-out */
245#define SN_ERR_CLICL 0x00000200 /* client closed (read/write error) */
246#define SN_ERR_SRVTO 0x00000300 /* server time-out, connect time-out */
247#define SN_ERR_SRVCL 0x00000400 /* server closed (connect/read/write error) */
248#define SN_ERR_PRXCOND 0x00000500 /* the proxy decided to close (deny...) */
249#define SN_ERR_MASK 0x00000700 /* mask to get only session error flags */
250#define SN_ERR_SHIFT 8 /* bit shift */
251
252#define SN_FINST_R 0x00001000 /* session ended during client request */
253#define SN_FINST_C 0x00002000 /* session ended during server connect */
254#define SN_FINST_H 0x00003000 /* session ended during server headers */
255#define SN_FINST_D 0x00004000 /* session ended during data phase */
256#define SN_FINST_L 0x00005000 /* session ended while pushing last data to client */
257#define SN_FINST_MASK 0x00007000 /* mask to get only final session state flags */
258#define SN_FINST_SHIFT 12 /* bit shift */
259
260#define SN_SCK_NONE 0x00000000 /* no set-cookie seen for the server cookie */
261#define SN_SCK_DELETED 0x00010000 /* existing set-cookie deleted or changed */
262#define SN_SCK_INSERTED 0x00020000 /* new set-cookie inserted or changed existing one */
263#define SN_SCK_SEEN 0x00040000 /* set-cookie seen for the server cookie */
264#define SN_SCK_MASK 0x00070000 /* mask to get the set-cookie field */
265#define SN_SCK_SHIFT 16 /* bit shift */
266
willy tarreau5cbea6f2005-12-17 12:48:26 +0100267
268/* different possible states for the client side */
willy tarreau0f7af912005-12-17 12:21:26 +0100269#define CL_STHEADERS 0
270#define CL_STDATA 1
271#define CL_STSHUTR 2
272#define CL_STSHUTW 3
273#define CL_STCLOSE 4
274
willy tarreau5cbea6f2005-12-17 12:48:26 +0100275/* different possible states for the server side */
willy tarreau0f7af912005-12-17 12:21:26 +0100276#define SV_STIDLE 0
277#define SV_STCONN 1
278#define SV_STHEADERS 2
279#define SV_STDATA 3
280#define SV_STSHUTR 4
281#define SV_STSHUTW 5
282#define SV_STCLOSE 6
283
284/* result of an I/O event */
285#define RES_SILENT 0 /* didn't happen */
286#define RES_DATA 1 /* data were sent or received */
287#define RES_NULL 2 /* result is 0 (read == 0), or connect without need for writing */
288#define RES_ERROR 3 /* result -1 or error on the socket (eg: connect()) */
289
willy tarreau9fe663a2005-12-17 13:02:59 +0100290/* modes of operation (global.mode) */
willy tarreau0f7af912005-12-17 12:21:26 +0100291#define MODE_DEBUG 1
292#define MODE_STATS 2
293#define MODE_LOG 4
294#define MODE_DAEMON 8
willy tarreau5cbea6f2005-12-17 12:48:26 +0100295#define MODE_QUIET 16
296
297/* server flags */
willy tarreaua41a8b42005-12-17 14:02:24 +0100298#define SRV_RUNNING 1 /* the server is UP */
299#define SRV_BACKUP 2 /* this server is a backup server */
300#define SRV_MAPPORTS 4 /* this server uses mapped ports */
willy tarreau0f7af912005-12-17 12:21:26 +0100301
willy tarreaue39cd132005-12-17 13:00:18 +0100302/* what to do when a header matches a regex */
303#define ACT_ALLOW 0 /* allow the request */
304#define ACT_REPLACE 1 /* replace the matching header */
305#define ACT_REMOVE 2 /* remove the matching header */
306#define ACT_DENY 3 /* deny the request */
willy tarreau036e1ce2005-12-17 13:46:33 +0100307#define ACT_PASS 4 /* pass this header without allowing or denying the request */
willy tarreaue39cd132005-12-17 13:00:18 +0100308
willy tarreau9fe663a2005-12-17 13:02:59 +0100309/* configuration sections */
310#define CFG_NONE 0
311#define CFG_GLOBAL 1
312#define CFG_LISTEN 2
313
willy tarreaua1598082005-12-17 13:08:06 +0100314/* fields that need to be logged. They appear as flags in session->logs.logwait */
willy tarreau9fe663a2005-12-17 13:02:59 +0100315#define LW_DATE 1 /* date */
316#define LW_CLIP 2 /* CLient IP */
317#define LW_SVIP 4 /* SerVer IP */
318#define LW_SVID 8 /* server ID */
319#define LW_REQ 16 /* http REQuest */
320#define LW_RESP 32 /* http RESPonse */
321#define LW_PXIP 64 /* proxy IP */
322#define LW_PXID 128 /* proxy ID */
willy tarreaua1598082005-12-17 13:08:06 +0100323#define LW_BYTES 256 /* bytes read from server */
willy tarreau9fe663a2005-12-17 13:02:59 +0100324
willy tarreau0f7af912005-12-17 12:21:26 +0100325/*********************************************************************/
326
327#define LIST_HEAD(a) ((void *)(&(a)))
328
329/*********************************************************************/
330
331struct hdr_exp {
willy tarreaue39cd132005-12-17 13:00:18 +0100332 struct hdr_exp *next;
333 regex_t *preg; /* expression to look for */
334 int action; /* ACT_ALLOW, ACT_REPLACE, ACT_REMOVE, ACT_DENY */
335 char *replace; /* expression to set instead */
willy tarreau0f7af912005-12-17 12:21:26 +0100336};
337
338struct buffer {
339 unsigned int l; /* data length */
340 char *r, *w, *h, *lr; /* read ptr, write ptr, last header ptr, last read */
willy tarreauef900ab2005-12-17 12:52:52 +0100341 char *rlim; /* read limit, used for header rewriting */
willy tarreaua1598082005-12-17 13:08:06 +0100342 unsigned long long total; /* total data read */
willy tarreau0f7af912005-12-17 12:21:26 +0100343 char data[BUFSIZE];
344};
345
346struct server {
347 struct server *next;
willy tarreau5cbea6f2005-12-17 12:48:26 +0100348 int state; /* server state (SRV_*) */
349 int cklen; /* the len of the cookie, to speed up checks */
350 char *cookie; /* the id set in the cookie */
351 char *id; /* just for identification */
willy tarreau0f7af912005-12-17 12:21:26 +0100352 struct sockaddr_in addr; /* the address to connect to */
willy tarreaua41a8b42005-12-17 14:02:24 +0100353 short check_port; /* the port to use for the health checks */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100354 int health; /* 0->rise-1 = bad; rise->rise+fall-1 = good */
willy tarreaue47c8d72005-12-17 12:55:52 +0100355 int rise, fall; /* time in iterations */
356 int inter; /* time in milliseconds */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100357 int result; /* 0 = connect OK, -1 = connect KO */
358 int curfd; /* file desc used for current test, or -1 if not in test */
willy tarreau535ae7a2005-12-17 12:58:00 +0100359 struct proxy *proxy; /* the proxy this server belongs to */
willy tarreau0f7af912005-12-17 12:21:26 +0100360};
361
willy tarreau5cbea6f2005-12-17 12:48:26 +0100362/* The base for all tasks */
willy tarreau0f7af912005-12-17 12:21:26 +0100363struct task {
364 struct task *next, *prev; /* chaining ... */
365 struct task *rqnext; /* chaining in run queue ... */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100366 struct task *wq; /* the wait queue this task is in */
willy tarreau0f7af912005-12-17 12:21:26 +0100367 int state; /* task state : IDLE or RUNNING */
368 struct timeval expire; /* next expiration time for this task, use only for fast sorting */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100369 int (*process)(struct task *t); /* the function which processes the task */
370 void *context; /* the task's context */
371};
372
373/* WARNING: if new fields are added, they must be initialized in event_accept() */
374struct session {
375 struct task *task; /* the task associated with this session */
willy tarreau0f7af912005-12-17 12:21:26 +0100376 /* application specific below */
377 struct timeval crexpire; /* expiration date for a client read */
378 struct timeval cwexpire; /* expiration date for a client write */
379 struct timeval srexpire; /* expiration date for a server read */
380 struct timeval swexpire; /* expiration date for a server write */
381 struct timeval cnexpire; /* expiration date for a connect */
382 char res_cr, res_cw, res_sr, res_sw;/* results of some events */
383 struct proxy *proxy; /* the proxy this socket belongs to */
384 int cli_fd; /* the client side fd */
385 int srv_fd; /* the server side fd */
386 int cli_state; /* state of the client side */
387 int srv_state; /* state of the server side */
388 int conn_retries; /* number of connect retries left */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100389 int flags; /* some flags describing the session */
willy tarreau0f7af912005-12-17 12:21:26 +0100390 struct buffer *req; /* request buffer */
391 struct buffer *rep; /* response buffer */
392 struct sockaddr_in cli_addr; /* the client address */
393 struct sockaddr_in srv_addr; /* the address to connect to */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100394 struct server *srv; /* the server being used */
willy tarreaua1598082005-12-17 13:08:06 +0100395 struct {
396 int logwait; /* log fields waiting to be collected : LW_* */
397 struct timeval tv_accept; /* date of the accept() (beginning of the session) */
398 long t_request; /* delay before the end of the request arrives, -1 if never occurs */
399 long t_connect; /* delay before the connect() to the server succeeds, -1 if never occurs */
400 long t_data; /* delay before the first data byte from the server ... */
401 unsigned long t_close; /* total session duration */
402 char *uri; /* first line if log needed, NULL otherwise */
willy tarreau8337c6b2005-12-17 13:41:01 +0100403 char *cli_cookie; /* cookie presented by the client, in capture mode */
404 char *srv_cookie; /* cookie presented by the server, in capture mode */
willy tarreaua1598082005-12-17 13:08:06 +0100405 int status; /* HTTP status from the server, negative if from proxy */
406 long long bytes; /* number of bytes transferred from the server */
407 } logs;
willy tarreau2f6ba652005-12-17 13:57:42 +0100408 unsigned int uniq_id; /* unique ID used for the traces */
willy tarreau0f7af912005-12-17 12:21:26 +0100409};
410
willy tarreaua41a8b42005-12-17 14:02:24 +0100411struct listener {
412 int fd; /* the listen socket */
413 struct sockaddr_in addr; /* the address we listen to */
414 struct listener *next; /* next address or NULL */
415};
416
417
willy tarreau0f7af912005-12-17 12:21:26 +0100418struct proxy {
willy tarreaua41a8b42005-12-17 14:02:24 +0100419 struct listener *listen; /* the listen addresses and sockets */
willy tarreau0f7af912005-12-17 12:21:26 +0100420 int state; /* proxy state */
willy tarreau0f7af912005-12-17 12:21:26 +0100421 struct sockaddr_in dispatch_addr; /* the default address to connect to */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100422 struct server *srv, *cursrv; /* known servers, current server */
423 int nbservers; /* # of servers */
willy tarreau0f7af912005-12-17 12:21:26 +0100424 char *cookie_name; /* name of the cookie to look for */
willy tarreau8337c6b2005-12-17 13:41:01 +0100425 int cookie_len; /* strlen(cookie_len), computed only once */
426 char *capture_name; /* beginning of the name of the cookie to capture */
427 int capture_namelen; /* length of the cookie name to match */
428 int capture_len; /* length of the string to be captured */
willy tarreau0f7af912005-12-17 12:21:26 +0100429 int clitimeout; /* client I/O timeout (in milliseconds) */
430 int srvtimeout; /* server I/O timeout (in milliseconds) */
431 int contimeout; /* connect timeout (in milliseconds) */
432 char *id; /* proxy id */
433 int nbconn; /* # of active sessions */
434 int maxconn; /* max # of active sessions */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100435 int conn_retries; /* maximum number of connect retries */
436 int options; /* PR_O_REDISP, PR_O_TRANSP */
437 int mode; /* mode = PR_MODE_TCP, PR_MODE_HTTP or PR_MODE_HEALTH */
willy tarreaua1598082005-12-17 13:08:06 +0100438 struct sockaddr_in source_addr; /* the address to which we want to bind for connect() */
willy tarreau0f7af912005-12-17 12:21:26 +0100439 struct proxy *next;
440 struct sockaddr_in logsrv1, logsrv2; /* 2 syslog servers */
441 char logfac1, logfac2; /* log facility for both servers. -1 = disabled */
willy tarreau8337c6b2005-12-17 13:41:01 +0100442 int loglev1, loglev2; /* log level for each server, 7 by default */
willy tarreau9fe663a2005-12-17 13:02:59 +0100443 int to_log; /* things to be logged (LW_*) */
willy tarreau0f7af912005-12-17 12:21:26 +0100444 struct timeval stop_time; /* date to stop listening, when stopping != 0 */
willy tarreaue39cd132005-12-17 13:00:18 +0100445 int nb_reqadd, nb_rspadd;
446 struct hdr_exp *req_exp; /* regular expressions for request headers */
447 struct hdr_exp *rsp_exp; /* regular expressions for response headers */
448 char *req_add[MAX_NEWHDR], *rsp_add[MAX_NEWHDR]; /* headers to be added */
willy tarreau0f7af912005-12-17 12:21:26 +0100449 int grace; /* grace time after stop request */
willy tarreau2f6ba652005-12-17 13:57:42 +0100450 char *check_req; /* HTTP request to use if PR_O_HTTP_CHK is set, else NULL */
451 int check_len; /* Length of the HTTP request */
willy tarreau8337c6b2005-12-17 13:41:01 +0100452 struct {
453 char *msg400; /* message for error 400 */
454 int len400; /* message length for error 400 */
455 char *msg403; /* message for error 403 */
456 int len403; /* message length for error 403 */
457 char *msg408; /* message for error 408 */
458 int len408; /* message length for error 408 */
459 char *msg500; /* message for error 500 */
460 int len500; /* message length for error 500 */
461 char *msg502; /* message for error 502 */
462 int len502; /* message length for error 502 */
463 char *msg503; /* message for error 503 */
464 int len503; /* message length for error 503 */
465 char *msg504; /* message for error 504 */
466 int len504; /* message length for error 504 */
467 } errmsg;
willy tarreau0f7af912005-12-17 12:21:26 +0100468};
469
470/* info about one given fd */
471struct fdtab {
472 int (*read)(int fd); /* read function */
473 int (*write)(int fd); /* write function */
474 struct task *owner; /* the session (or proxy) associated with this fd */
475 int state; /* the state of this fd */
476};
477
478/*********************************************************************/
479
willy tarreau0f7af912005-12-17 12:21:26 +0100480int cfg_maxpconn = 2000; /* # of simultaneous connections per proxy (-N) */
willy tarreau0f7af912005-12-17 12:21:26 +0100481char *cfg_cfgfile = NULL; /* configuration file */
482char *progname = NULL; /* program name */
483int pid; /* current process id */
willy tarreau9fe663a2005-12-17 13:02:59 +0100484
485/* global options */
486static struct {
487 int uid;
488 int gid;
489 int nbproc;
490 int maxconn;
491 int maxsock; /* max # of sockets */
492 int mode;
493 char *chroot;
494 int logfac1, logfac2;
willy tarreau8337c6b2005-12-17 13:41:01 +0100495 int loglev1, loglev2;
willy tarreau9fe663a2005-12-17 13:02:59 +0100496 struct sockaddr_in logsrv1, logsrv2;
497} global = {
498 logfac1 : -1,
499 logfac2 : -1,
willy tarreau8337c6b2005-12-17 13:41:01 +0100500 loglev1 : 7, /* max syslog level : debug */
501 loglev2 : 7,
willy tarreau9fe663a2005-12-17 13:02:59 +0100502 /* others NULL OK */
503};
504
willy tarreau0f7af912005-12-17 12:21:26 +0100505/*********************************************************************/
506
507fd_set *ReadEvent,
508 *WriteEvent,
509 *StaticReadEvent,
510 *StaticWriteEvent;
511
512void **pool_session = NULL,
513 **pool_buffer = NULL,
514 **pool_fdtab = NULL,
willy tarreau9fe663a2005-12-17 13:02:59 +0100515 **pool_requri = NULL,
willy tarreau8337c6b2005-12-17 13:41:01 +0100516 **pool_task = NULL,
517 **pool_capture = NULL;
willy tarreau0f7af912005-12-17 12:21:26 +0100518
519struct proxy *proxy = NULL; /* list of all existing proxies */
520struct fdtab *fdtab = NULL; /* array of all the file descriptors */
willy tarreau5cbea6f2005-12-17 12:48:26 +0100521struct task *rq = NULL; /* global run queue */
522struct task wait_queue = { /* global wait queue */
523 prev:LIST_HEAD(wait_queue),
524 next:LIST_HEAD(wait_queue)
525};
willy tarreau0f7af912005-12-17 12:21:26 +0100526
willy tarreau0f7af912005-12-17 12:21:26 +0100527static int totalconn = 0; /* total # of terminated sessions */
528static int actconn = 0; /* # of active sessions */
529static int maxfd = 0; /* # of the highest fd + 1 */
530static int listeners = 0; /* # of listeners */
531static int stopping = 0; /* non zero means stopping in progress */
532static struct timeval now = {0,0}; /* the current date at any moment */
willy tarreaua41a8b42005-12-17 14:02:24 +0100533static struct proxy defproxy; /* fake proxy used to assign default values on all instances */
willy tarreau0f7af912005-12-17 12:21:26 +0100534
535static regmatch_t pmatch[MAX_MATCH]; /* rm_so, rm_eo for regular expressions */
willy tarreau750a4722005-12-17 13:21:24 +0100536/* this is used to drain data, and as a temporary buffer for sprintf()... */
willy tarreau0f7af912005-12-17 12:21:26 +0100537static char trash[BUFSIZE];
538
539/*
willy tarreau036e1ce2005-12-17 13:46:33 +0100540 * Syslog facilities and levels. Conforming to RFC3164.
willy tarreau0f7af912005-12-17 12:21:26 +0100541 */
542
543#define MAX_SYSLOG_LEN 1024
544#define NB_LOG_FACILITIES 24
545const char *log_facilities[NB_LOG_FACILITIES] = {
546 "kern", "user", "mail", "daemon",
547 "auth", "syslog", "lpr", "news",
548 "uucp", "cron", "auth2", "ftp",
549 "ntp", "audit", "alert", "cron2",
550 "local0", "local1", "local2", "local3",
551 "local4", "local5", "local6", "local7"
552};
553
554
555#define NB_LOG_LEVELS 8
556const char *log_levels[NB_LOG_LEVELS] = {
557 "emerg", "alert", "crit", "err",
558 "warning", "notice", "info", "debug"
559};
560
561#define SYSLOG_PORT 514
562
563const char *monthname[12] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun",
564 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec" };
willy tarreau036e1ce2005-12-17 13:46:33 +0100565
566const char sess_term_cond[8] = "-cCsSP67"; /* normal, CliTo, CliErr, SrvTo, SrvErr, PxErr, unknown */
567const char sess_fin_state[8] = "-RCHDL67"; /* cliRequest, srvConnect, srvHeader, Data, Last, unknown */
568const char sess_cookie[4] = "NIDV"; /* No cookie, Invalid cookie, cookie for a Down server, Valid cookie */
569const char sess_set_cookie[8] = "N1I3PD5R"; /* No set-cookie, unknown, Set-Cookie Inserted, unknown,
570 Set-cookie seen and left unchanged (passive), Set-cookie Deleted,
571 unknown, Set-cookie Rewritten */
572
willy tarreau0f7af912005-12-17 12:21:26 +0100573#define MAX_HOSTNAME_LEN 32
574static char hostname[MAX_HOSTNAME_LEN] = "";
575
willy tarreau8337c6b2005-12-17 13:41:01 +0100576const char *HTTP_302 =
577 "HTTP/1.0 302 Found\r\n"
578 "Cache-Control: no-cache\r\n"
579 "Connection: close\r\n"
580 "Location: "; /* not terminated since it will be concatenated with the URL */
581
willy tarreaua1598082005-12-17 13:08:06 +0100582const char *HTTP_400 =
583 "HTTP/1.0 400 Bad request\r\n"
willy tarreaue39cd132005-12-17 13:00:18 +0100584 "Cache-Control: no-cache\r\n"
585 "Connection: close\r\n"
586 "\r\n"
willy tarreau750a4722005-12-17 13:21:24 +0100587 "<html><body><h1>400 Bad request</h1>\nYour browser sent an invalid request.\n</body></html>\n";
willy tarreaue39cd132005-12-17 13:00:18 +0100588
willy tarreaua1598082005-12-17 13:08:06 +0100589const char *HTTP_403 =
590 "HTTP/1.0 403 Forbidden\r\n"
willy tarreaue39cd132005-12-17 13:00:18 +0100591 "Cache-Control: no-cache\r\n"
592 "Connection: close\r\n"
593 "\r\n"
willy tarreau750a4722005-12-17 13:21:24 +0100594 "<html><body><h1>403 Forbidden</h1>\nRequest forbidden by administrative rules.\n</body></html>\n";
595
willy tarreau8337c6b2005-12-17 13:41:01 +0100596const char *HTTP_408 =
597 "HTTP/1.0 408 Request Time-out\r\n"
598 "Cache-Control: no-cache\r\n"
599 "Connection: close\r\n"
600 "\r\n"
601 "<html><body><h1>408 Request Time-out</h1>\nYour browser didn't send a complete request in time.\n</body></html>\n";
602
willy tarreau750a4722005-12-17 13:21:24 +0100603const char *HTTP_500 =
604 "HTTP/1.0 500 Server Error\r\n"
605 "Cache-Control: no-cache\r\n"
606 "Connection: close\r\n"
607 "\r\n"
608 "<html><body><h1>500 Server Error</h1>\nAn internal server error occured.\n</body></html>\n";
willy tarreaue39cd132005-12-17 13:00:18 +0100609
610const char *HTTP_502 =
willy tarreau8337c6b2005-12-17 13:41:01 +0100611 "HTTP/1.0 502 Bad Gateway\r\n"
willy tarreaue39cd132005-12-17 13:00:18 +0100612 "Cache-Control: no-cache\r\n"
613 "Connection: close\r\n"
614 "\r\n"
willy tarreau8337c6b2005-12-17 13:41:01 +0100615 "<html><body><h1>502 Bad Gateway</h1>\nThe server returned an invalid or incomplete response.\n</body></html>\n";
616
617const char *HTTP_503 =
618 "HTTP/1.0 503 Service Unavailable\r\n"
619 "Cache-Control: no-cache\r\n"
620 "Connection: close\r\n"
621 "\r\n"
622 "<html><body><h1>503 Service Unavailable</h1>\nNo server is available to handle this request.\n</body></html>\n";
623
624const char *HTTP_504 =
625 "HTTP/1.0 504 Gateway Time-out\r\n"
626 "Cache-Control: no-cache\r\n"
627 "Connection: close\r\n"
628 "\r\n"
629 "<html><body><h1>504 Gateway Time-out</h1>\nThe server didn't respond in time.\n</body></html>\n";
willy tarreaue39cd132005-12-17 13:00:18 +0100630
willy tarreau0f7af912005-12-17 12:21:26 +0100631/*********************************************************************/
632/* statistics ******************************************************/
633/*********************************************************************/
634
willy tarreau750a4722005-12-17 13:21:24 +0100635#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +0100636static int stats_tsk_lsrch, stats_tsk_rsrch,
637 stats_tsk_good, stats_tsk_right, stats_tsk_left,
638 stats_tsk_new, stats_tsk_nsrch;
willy tarreau750a4722005-12-17 13:21:24 +0100639#endif
willy tarreau0f7af912005-12-17 12:21:26 +0100640
641
642/*********************************************************************/
willy tarreau750a4722005-12-17 13:21:24 +0100643/* debugging *******************************************************/
644/*********************************************************************/
645#ifdef DEBUG_FULL
646static char *cli_stnames[5] = {"HDR", "DAT", "SHR", "SHW", "CLS" };
647static char *srv_stnames[7] = {"IDL", "CON", "HDR", "DAT", "SHR", "SHW", "CLS" };
648#endif
649
650/*********************************************************************/
willy tarreau0f7af912005-12-17 12:21:26 +0100651/* function prototypes *********************************************/
652/*********************************************************************/
653
654int event_accept(int fd);
655int event_cli_read(int fd);
656int event_cli_write(int fd);
657int event_srv_read(int fd);
658int event_srv_write(int fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +0100659int process_session(struct task *t);
willy tarreau0f7af912005-12-17 12:21:26 +0100660
661/*********************************************************************/
662/* general purpose functions ***************************************/
663/*********************************************************************/
664
665void display_version() {
666 printf("HA-Proxy version " HAPROXY_VERSION " " HAPROXY_DATE"\n");
willy tarreauc1cae632005-12-17 14:12:23 +0100667 printf("Copyright 2000-2003 Willy Tarreau <willy AT meta-x DOT org>\n\n");
willy tarreau0f7af912005-12-17 12:21:26 +0100668}
669
670/*
671 * This function prints the command line usage and exits
672 */
673void usage(char *name) {
674 display_version();
675 fprintf(stderr,
676 "Usage : %s -f <cfgfile> [ -vd"
677#if STATTIME > 0
678 "sl"
679#endif
680 "D ] [ -n <maxconn> ] [ -N <maxpconn> ]\n"
681 " -v displays version\n"
682 " -d enters debug mode\n"
683#if STATTIME > 0
684 " -s enables statistics output\n"
685 " -l enables long statistics format\n"
686#endif
willy tarreau5cbea6f2005-12-17 12:48:26 +0100687 " -D goes daemon ; implies -q\n"
688 " -q quiet mode : don't display messages\n"
willy tarreau0f7af912005-12-17 12:21:26 +0100689 " -n sets the maximum total # of connections (%d)\n"
690 " -N sets the default, per-proxy maximum # of connections (%d)\n\n",
willy tarreau9fe663a2005-12-17 13:02:59 +0100691 name, DEFAULT_MAXCONN, cfg_maxpconn);
willy tarreau0f7af912005-12-17 12:21:26 +0100692 exit(1);
693}
694
695
696/*
697 * Displays the message on stderr with the date and pid.
698 */
699void Alert(char *fmt, ...) {
700 va_list argp;
701 struct timeval tv;
702 struct tm *tm;
703
willy tarreau9fe663a2005-12-17 13:02:59 +0100704 if (!(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +0100705 va_start(argp, fmt);
willy tarreau0f7af912005-12-17 12:21:26 +0100706
willy tarreau5cbea6f2005-12-17 12:48:26 +0100707 gettimeofday(&tv, NULL);
708 tm=localtime(&tv.tv_sec);
709 fprintf(stderr, "[ALERT] %03d/%02d%02d%02d (%d) : ",
willy tarreaua1598082005-12-17 13:08:06 +0100710 tm->tm_yday, tm->tm_hour, tm->tm_min, tm->tm_sec, (int)getpid());
willy tarreau5cbea6f2005-12-17 12:48:26 +0100711 vfprintf(stderr, fmt, argp);
712 fflush(stderr);
713 va_end(argp);
714 }
willy tarreau0f7af912005-12-17 12:21:26 +0100715}
716
717
718/*
719 * Displays the message on stderr with the date and pid.
720 */
721void Warning(char *fmt, ...) {
722 va_list argp;
723 struct timeval tv;
724 struct tm *tm;
725
willy tarreau9fe663a2005-12-17 13:02:59 +0100726 if (!(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +0100727 va_start(argp, fmt);
willy tarreau0f7af912005-12-17 12:21:26 +0100728
willy tarreau5cbea6f2005-12-17 12:48:26 +0100729 gettimeofday(&tv, NULL);
730 tm=localtime(&tv.tv_sec);
731 fprintf(stderr, "[WARNING] %03d/%02d%02d%02d (%d) : ",
willy tarreaua1598082005-12-17 13:08:06 +0100732 tm->tm_yday, tm->tm_hour, tm->tm_min, tm->tm_sec, (int)getpid());
willy tarreau5cbea6f2005-12-17 12:48:26 +0100733 vfprintf(stderr, fmt, argp);
734 fflush(stderr);
735 va_end(argp);
736 }
737}
738
739/*
740 * Displays the message on <out> only if quiet mode is not set.
741 */
742void qfprintf(FILE *out, char *fmt, ...) {
743 va_list argp;
744
willy tarreau9fe663a2005-12-17 13:02:59 +0100745 if (!(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +0100746 va_start(argp, fmt);
747 vfprintf(out, fmt, argp);
748 fflush(out);
749 va_end(argp);
750 }
willy tarreau0f7af912005-12-17 12:21:26 +0100751}
752
753
754/*
755 * converts <str> to a struct sockaddr_in* which is locally allocated.
756 * The format is "addr:port", where "addr" can be empty or "*" to indicate
757 * INADDR_ANY.
758 */
759struct sockaddr_in *str2sa(char *str) {
760 static struct sockaddr_in sa;
761 char *c;
762 int port;
763
willy tarreaua1598082005-12-17 13:08:06 +0100764 memset(&sa, 0, sizeof(sa));
willy tarreau0f7af912005-12-17 12:21:26 +0100765 str=strdup(str);
766
767 if ((c=strrchr(str,':')) != NULL) {
768 *c++=0;
769 port=atol(c);
770 }
771 else
772 port=0;
773
774 if (*str == '*' || *str == '\0') { /* INADDR_ANY */
775 sa.sin_addr.s_addr = INADDR_ANY;
776 }
777 else if (
778#ifndef SOLARIS
779 !inet_aton(str, &sa.sin_addr)
780#else
781 !inet_pton(AF_INET, str, &sa.sin_addr)
782#endif
783 ) {
784 struct hostent *he;
785
786 if ((he = gethostbyname(str)) == NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +0100787 Alert("Invalid server name: '%s'\n", str);
willy tarreau0f7af912005-12-17 12:21:26 +0100788 }
789 else
790 sa.sin_addr = *(struct in_addr *) *(he->h_addr_list);
791 }
792 sa.sin_port=htons(port);
793 sa.sin_family=AF_INET;
794
795 free(str);
796 return &sa;
797}
798
willy tarreau9fe663a2005-12-17 13:02:59 +0100799
800/*
willy tarreaua41a8b42005-12-17 14:02:24 +0100801 * converts <str> to a list of listeners which are dynamically allocated.
802 * The format is "{addr|'*'}:port[-end][,{addr|'*'}:port[-end]]*", where :
803 * - <addr> can be empty or "*" to indicate INADDR_ANY ;
804 * - <port> is a numerical port from 1 to 65535 ;
805 * - <end> indicates to use the range from <port> to <end> instead (inclusive).
806 * This can be repeated as many times as necessary, separated by a coma.
807 * The <tail> argument is a pointer to a current list which should be appended
808 * to the tail of the new list. The pointer to the new list is returned.
809 */
810struct listener *str2listener(char *str, struct listener *tail) {
811 struct listener *l;
812 char *c, *next, *range, *dupstr;
813 int port, end;
814
815 next = dupstr = strdup(str);
816
817 while (next && *next) {
818 str = next;
819 /* 1) look for the end of the first address */
820 if ((next = strrchr(str, ',')) != NULL) {
821 *next++ = 0;
822 }
823
824 /* 2) look for the addr/port delimiter */
825 if ((range = strrchr(str, ':')) != NULL) {
826 *range++ = 0;
827 }
828 else {
829 Alert("Missing port number: '%s'\n", str);
830 }
831
832 /* 3) look for the port-end delimiter */
833 if ((c = strchr(range, '-')) != NULL) {
834 *c++ = 0;
835 end = atol(c);
836 }
837 else {
838 end = atol(range);
839 }
840
841 for (port = atol(range); port <= end; port++) {
842 l = (struct listener *)calloc(1, sizeof(struct listener));
843 l->next = tail;
844 tail = l;
845
846 if (*str == '*' || *str == '\0') { /* INADDR_ANY */
847 l->addr.sin_addr.s_addr = INADDR_ANY;
848 }
849 else if (
850#ifndef SOLARIS
851 !inet_aton(str, &l->addr.sin_addr)
852#else
853 !inet_pton(AF_INET, str, &l->addr.sin_addr)
854#endif
855 ) {
856 struct hostent *he;
857
858 if ((he = gethostbyname(str)) == NULL) {
859 Alert("Invalid server name: '%s'\n", str);
860 }
861 else
862 l->addr.sin_addr = *(struct in_addr *) *(he->h_addr_list);
863 }
864 l->addr.sin_port=htons(port);
865 l->addr.sin_family=AF_INET;
866 } /* end for(port) */
867 } /* end while(next) */
868 free(dupstr);
869 return tail;
870}
871
872
873/*
willy tarreau9fe663a2005-12-17 13:02:59 +0100874 * This function sends a syslog message to both log servers of a proxy,
875 * or to global log servers if the proxy is NULL.
876 * It also tries not to waste too much time computing the message header.
877 * It doesn't care about errors nor does it report them.
willy tarreau9fe663a2005-12-17 13:02:59 +0100878 */
879void send_log(struct proxy *p, int level, char *message, ...) {
880 static int logfd = -1; /* syslog UDP socket */
881 static long tvsec = -1; /* to force the string to be initialized */
882 struct timeval tv;
883 va_list argp;
884 static char logmsg[MAX_SYSLOG_LEN];
885 static char *dataptr = NULL;
886 int fac_level;
887 int hdr_len, data_len;
888 struct sockaddr_in *sa[2];
willy tarreau8337c6b2005-12-17 13:41:01 +0100889 int facilities[2], loglevel[2];
willy tarreau9fe663a2005-12-17 13:02:59 +0100890 int nbloggers = 0;
891 char *log_ptr;
892
893 if (logfd < 0) {
894 if ((logfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
895 return;
896 }
897
898 if (level < 0 || progname == NULL || message == NULL)
899 return;
900
901 gettimeofday(&tv, NULL);
willy tarreauc29948c2005-12-17 13:10:27 +0100902 if (tv.tv_sec != tvsec || dataptr == NULL) {
willy tarreau9fe663a2005-12-17 13:02:59 +0100903 /* this string is rebuild only once a second */
904 struct tm *tm = localtime(&tv.tv_sec);
905 tvsec = tv.tv_sec;
906
willy tarreauc29948c2005-12-17 13:10:27 +0100907 hdr_len = snprintf(logmsg, sizeof(logmsg),
908 "<<<<>%s %2d %02d:%02d:%02d %s[%d]: ",
909 monthname[tm->tm_mon],
910 tm->tm_mday, tm->tm_hour, tm->tm_min, tm->tm_sec,
911 progname, pid);
912 /* WARNING: depending upon implementations, snprintf may return
913 * either -1 or the number of bytes that would be needed to store
914 * the total message. In both cases, we must adjust it.
willy tarreau9fe663a2005-12-17 13:02:59 +0100915 */
willy tarreauc29948c2005-12-17 13:10:27 +0100916 if (hdr_len < 0 || hdr_len > sizeof(logmsg))
917 hdr_len = sizeof(logmsg);
918
919 dataptr = logmsg + hdr_len;
willy tarreau9fe663a2005-12-17 13:02:59 +0100920 }
921
922 va_start(argp, message);
923 data_len = vsnprintf(dataptr, logmsg + sizeof(logmsg) - dataptr, message, argp);
willy tarreauc29948c2005-12-17 13:10:27 +0100924 if (data_len < 0 || data_len > (logmsg + sizeof(logmsg) - dataptr))
925 data_len = logmsg + sizeof(logmsg) - dataptr;
willy tarreau9fe663a2005-12-17 13:02:59 +0100926 va_end(argp);
willy tarreauc29948c2005-12-17 13:10:27 +0100927 dataptr[data_len - 1] = '\n'; /* force a break on ultra-long lines */
willy tarreau9fe663a2005-12-17 13:02:59 +0100928
929 if (p == NULL) {
930 if (global.logfac1 >= 0) {
931 sa[nbloggers] = &global.logsrv1;
932 facilities[nbloggers] = global.logfac1;
willy tarreau8337c6b2005-12-17 13:41:01 +0100933 loglevel[nbloggers] = global.loglev1;
willy tarreau9fe663a2005-12-17 13:02:59 +0100934 nbloggers++;
935 }
936 if (global.logfac2 >= 0) {
937 sa[nbloggers] = &global.logsrv2;
938 facilities[nbloggers] = global.logfac2;
willy tarreau8337c6b2005-12-17 13:41:01 +0100939 loglevel[nbloggers] = global.loglev2;
willy tarreau9fe663a2005-12-17 13:02:59 +0100940 nbloggers++;
941 }
942 } else {
943 if (p->logfac1 >= 0) {
944 sa[nbloggers] = &p->logsrv1;
945 facilities[nbloggers] = p->logfac1;
willy tarreau8337c6b2005-12-17 13:41:01 +0100946 loglevel[nbloggers] = p->loglev1;
willy tarreau9fe663a2005-12-17 13:02:59 +0100947 nbloggers++;
948 }
949 if (p->logfac2 >= 0) {
950 sa[nbloggers] = &p->logsrv2;
951 facilities[nbloggers] = p->logfac2;
willy tarreau8337c6b2005-12-17 13:41:01 +0100952 loglevel[nbloggers] = p->loglev2;
willy tarreau9fe663a2005-12-17 13:02:59 +0100953 nbloggers++;
954 }
955 }
956
957 while (nbloggers-- > 0) {
willy tarreau8337c6b2005-12-17 13:41:01 +0100958 /* we can filter the level of the messages that are sent to each logger */
959 if (level > loglevel[nbloggers])
960 continue;
961
willy tarreauc29948c2005-12-17 13:10:27 +0100962 /* For each target, we may have a different facility.
963 * We can also have a different log level for each message.
964 * This induces variations in the message header length.
965 * Since we don't want to recompute it each time, nor copy it every
966 * time, we only change the facility in the pre-computed header,
967 * and we change the pointer to the header accordingly.
968 */
willy tarreau9fe663a2005-12-17 13:02:59 +0100969 fac_level = (facilities[nbloggers] << 3) + level;
970 log_ptr = logmsg + 3; /* last digit of the log level */
971 do {
972 *log_ptr = '0' + fac_level % 10;
973 fac_level /= 10;
974 log_ptr--;
975 } while (fac_level && log_ptr > logmsg);
976 *log_ptr = '<';
willy tarreau9fe663a2005-12-17 13:02:59 +0100977
willy tarreauc29948c2005-12-17 13:10:27 +0100978 /* the total syslog message now starts at logptr, for dataptr+data_len-logptr */
willy tarreau9fe663a2005-12-17 13:02:59 +0100979
980#ifndef MSG_NOSIGNAL
willy tarreauc29948c2005-12-17 13:10:27 +0100981 sendto(logfd, log_ptr, dataptr + data_len - log_ptr, MSG_DONTWAIT,
willy tarreau9fe663a2005-12-17 13:02:59 +0100982 (struct sockaddr *)sa[nbloggers], sizeof(**sa));
983#else
willy tarreauc29948c2005-12-17 13:10:27 +0100984 sendto(logfd, log_ptr, dataptr + data_len - log_ptr, MSG_DONTWAIT | MSG_NOSIGNAL,
willy tarreau9fe663a2005-12-17 13:02:59 +0100985 (struct sockaddr *)sa[nbloggers], sizeof(**sa));
986#endif
987 }
willy tarreau0f7af912005-12-17 12:21:26 +0100988}
989
990
991/* sets <tv> to the current time */
992static inline struct timeval *tv_now(struct timeval *tv) {
993 if (tv)
994 gettimeofday(tv, NULL);
995 return tv;
996}
997
998/*
999 * adds <ms> ms to <from>, set the result to <tv> and returns a pointer <tv>
1000 */
1001static inline struct timeval *tv_delayfrom(struct timeval *tv, struct timeval *from, int ms) {
1002 if (!tv || !from)
1003 return NULL;
1004 tv->tv_usec = from->tv_usec + (ms%1000)*1000;
1005 tv->tv_sec = from->tv_sec + (ms/1000);
1006 while (tv->tv_usec >= 1000000) {
1007 tv->tv_usec -= 1000000;
1008 tv->tv_sec++;
1009 }
1010 return tv;
1011}
1012
1013/*
1014 * compares <tv1> and <tv2> : returns 0 if equal, -1 if tv1 < tv2, 1 if tv1 > tv2
1015 */
1016static inline int tv_cmp(struct timeval *tv1, struct timeval *tv2) {
willy tarreau750a4722005-12-17 13:21:24 +01001017 if (tv1->tv_sec < tv2->tv_sec)
willy tarreau0f7af912005-12-17 12:21:26 +01001018 return -1;
willy tarreau750a4722005-12-17 13:21:24 +01001019 else if (tv1->tv_sec > tv2->tv_sec)
willy tarreau0f7af912005-12-17 12:21:26 +01001020 return 1;
1021 else if (tv1->tv_usec < tv2->tv_usec)
1022 return -1;
willy tarreau750a4722005-12-17 13:21:24 +01001023 else if (tv1->tv_usec > tv2->tv_usec)
1024 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01001025 else
1026 return 0;
1027}
1028
1029/*
1030 * returns the absolute difference, in ms, between tv1 and tv2
1031 */
1032unsigned long tv_delta(struct timeval *tv1, struct timeval *tv2) {
1033 int cmp;
1034 unsigned long ret;
1035
1036
willy tarreauef900ab2005-12-17 12:52:52 +01001037 cmp = tv_cmp(tv1, tv2);
willy tarreau0f7af912005-12-17 12:21:26 +01001038 if (!cmp)
1039 return 0; /* same dates, null diff */
willy tarreau750a4722005-12-17 13:21:24 +01001040 else if (cmp < 0) {
willy tarreauef900ab2005-12-17 12:52:52 +01001041 struct timeval *tmp = tv1;
1042 tv1 = tv2;
1043 tv2 = tmp;
willy tarreau0f7af912005-12-17 12:21:26 +01001044 }
willy tarreauef900ab2005-12-17 12:52:52 +01001045 ret = (tv1->tv_sec - tv2->tv_sec) * 1000;
willy tarreau0f7af912005-12-17 12:21:26 +01001046 if (tv1->tv_usec > tv2->tv_usec)
willy tarreauef900ab2005-12-17 12:52:52 +01001047 ret += (tv1->tv_usec - tv2->tv_usec) / 1000;
willy tarreau0f7af912005-12-17 12:21:26 +01001048 else
willy tarreauef900ab2005-12-17 12:52:52 +01001049 ret -= (tv2->tv_usec - tv1->tv_usec) / 1000;
willy tarreau0f7af912005-12-17 12:21:26 +01001050 return (unsigned long) ret;
1051}
1052
1053/*
willy tarreau750a4722005-12-17 13:21:24 +01001054 * returns the difference, in ms, between tv1 and tv2
1055 */
1056static inline unsigned long tv_diff(struct timeval *tv1, struct timeval *tv2) {
1057 unsigned long ret;
1058
willy tarreau6e682ce2005-12-17 13:26:49 +01001059 ret = (tv2->tv_sec - tv1->tv_sec) * 1000;
1060 if (tv2->tv_usec > tv1->tv_usec)
1061 ret += (tv2->tv_usec - tv1->tv_usec) / 1000;
willy tarreau750a4722005-12-17 13:21:24 +01001062 else
willy tarreau6e682ce2005-12-17 13:26:49 +01001063 ret -= (tv1->tv_usec - tv2->tv_usec) / 1000;
willy tarreau750a4722005-12-17 13:21:24 +01001064 return (unsigned long) ret;
1065}
1066
1067/*
willy tarreau0f7af912005-12-17 12:21:26 +01001068 * compares <tv1> and <tv2> modulo 1ms: returns 0 if equal, -1 if tv1 < tv2, 1 if tv1 > tv2
1069 */
1070static inline int tv_cmp_ms(struct timeval *tv1, struct timeval *tv2) {
willy tarreauefae1842005-12-17 12:51:03 +01001071 if (tv1->tv_sec == tv2->tv_sec) {
willy tarreau750a4722005-12-17 13:21:24 +01001072 if (tv2->tv_usec > tv1->tv_usec + 1000)
willy tarreauefae1842005-12-17 12:51:03 +01001073 return -1;
willy tarreau750a4722005-12-17 13:21:24 +01001074 else if (tv1->tv_usec > tv2->tv_usec + 1000)
1075 return 1;
willy tarreauefae1842005-12-17 12:51:03 +01001076 else
1077 return 0;
1078 }
willy tarreau0f7af912005-12-17 12:21:26 +01001079 else if ((tv2->tv_sec > tv1->tv_sec + 1) ||
willy tarreauef900ab2005-12-17 12:52:52 +01001080 ((tv2->tv_sec == tv1->tv_sec + 1) && (tv2->tv_usec + 1000000 > tv1->tv_usec + 1000)))
willy tarreau0f7af912005-12-17 12:21:26 +01001081 return -1;
willy tarreau750a4722005-12-17 13:21:24 +01001082 else if ((tv1->tv_sec > tv2->tv_sec + 1) ||
1083 ((tv1->tv_sec == tv2->tv_sec + 1) && (tv1->tv_usec + 1000000 > tv2->tv_usec + 1000)))
1084 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01001085 else
1086 return 0;
1087}
1088
1089/*
1090 * returns the remaining time between tv1=now and event=tv2
1091 * if tv2 is passed, 0 is returned.
1092 */
1093static inline unsigned long tv_remain(struct timeval *tv1, struct timeval *tv2) {
1094 unsigned long ret;
1095
willy tarreau0f7af912005-12-17 12:21:26 +01001096 if (tv_cmp_ms(tv1, tv2) >= 0)
1097 return 0; /* event elapsed */
1098
willy tarreauef900ab2005-12-17 12:52:52 +01001099 ret = (tv2->tv_sec - tv1->tv_sec) * 1000;
willy tarreau0f7af912005-12-17 12:21:26 +01001100 if (tv2->tv_usec > tv1->tv_usec)
willy tarreauef900ab2005-12-17 12:52:52 +01001101 ret += (tv2->tv_usec - tv1->tv_usec) / 1000;
willy tarreau0f7af912005-12-17 12:21:26 +01001102 else
willy tarreauef900ab2005-12-17 12:52:52 +01001103 ret -= (tv1->tv_usec - tv2->tv_usec) / 1000;
willy tarreau0f7af912005-12-17 12:21:26 +01001104 return (unsigned long) ret;
1105}
1106
1107
1108/*
1109 * zeroes a struct timeval
1110 */
1111
1112static inline struct timeval *tv_eternity(struct timeval *tv) {
1113 tv->tv_sec = tv->tv_usec = 0;
1114 return tv;
1115}
1116
1117/*
1118 * returns 1 if tv is null, else 0
1119 */
1120static inline int tv_iseternity(struct timeval *tv) {
1121 if (tv->tv_sec == 0 && tv->tv_usec == 0)
1122 return 1;
1123 else
1124 return 0;
1125}
1126
1127/*
1128 * compares <tv1> and <tv2> : returns 0 if equal, -1 if tv1 < tv2, 1 if tv1 > tv2,
1129 * considering that 0 is the eternity.
1130 */
1131static inline int tv_cmp2(struct timeval *tv1, struct timeval *tv2) {
1132 if (tv_iseternity(tv1))
1133 if (tv_iseternity(tv2))
1134 return 0; /* same */
1135 else
1136 return 1; /* tv1 later than tv2 */
1137 else if (tv_iseternity(tv2))
1138 return -1; /* tv2 later than tv1 */
1139
1140 if (tv1->tv_sec > tv2->tv_sec)
1141 return 1;
1142 else if (tv1->tv_sec < tv2->tv_sec)
1143 return -1;
1144 else if (tv1->tv_usec > tv2->tv_usec)
1145 return 1;
1146 else if (tv1->tv_usec < tv2->tv_usec)
1147 return -1;
1148 else
1149 return 0;
1150}
1151
1152/*
1153 * compares <tv1> and <tv2> modulo 1 ms: returns 0 if equal, -1 if tv1 < tv2, 1 if tv1 > tv2,
1154 * considering that 0 is the eternity.
1155 */
1156static inline int tv_cmp2_ms(struct timeval *tv1, struct timeval *tv2) {
1157 if (tv_iseternity(tv1))
1158 if (tv_iseternity(tv2))
1159 return 0; /* same */
1160 else
1161 return 1; /* tv1 later than tv2 */
1162 else if (tv_iseternity(tv2))
1163 return -1; /* tv2 later than tv1 */
1164
willy tarreauefae1842005-12-17 12:51:03 +01001165 if (tv1->tv_sec == tv2->tv_sec) {
willy tarreauef900ab2005-12-17 12:52:52 +01001166 if (tv1->tv_usec > tv2->tv_usec + 1000)
willy tarreauefae1842005-12-17 12:51:03 +01001167 return 1;
willy tarreauef900ab2005-12-17 12:52:52 +01001168 else if (tv2->tv_usec > tv1->tv_usec + 1000)
willy tarreauefae1842005-12-17 12:51:03 +01001169 return -1;
1170 else
1171 return 0;
1172 }
1173 else if ((tv1->tv_sec > tv2->tv_sec + 1) ||
willy tarreauef900ab2005-12-17 12:52:52 +01001174 ((tv1->tv_sec == tv2->tv_sec + 1) && (tv1->tv_usec + 1000000 > tv2->tv_usec + 1000)))
willy tarreau0f7af912005-12-17 12:21:26 +01001175 return 1;
1176 else if ((tv2->tv_sec > tv1->tv_sec + 1) ||
willy tarreauef900ab2005-12-17 12:52:52 +01001177 ((tv2->tv_sec == tv1->tv_sec + 1) && (tv2->tv_usec + 1000000 > tv1->tv_usec + 1000)))
willy tarreau0f7af912005-12-17 12:21:26 +01001178 return -1;
1179 else
1180 return 0;
1181}
1182
1183/*
1184 * returns the first event between tv1 and tv2 into tvmin.
1185 * a zero tv is ignored. tvmin is returned.
1186 */
1187static inline struct timeval *tv_min(struct timeval *tvmin,
1188 struct timeval *tv1, struct timeval *tv2) {
1189
1190 if (tv_cmp2(tv1, tv2) <= 0)
1191 *tvmin = *tv1;
1192 else
1193 *tvmin = *tv2;
1194
1195 return tvmin;
1196}
1197
1198
1199
1200/***********************************************************/
1201/* fd management ***************************************/
1202/***********************************************************/
1203
1204
1205
willy tarreau5cbea6f2005-12-17 12:48:26 +01001206/* Deletes an FD from the fdsets, and recomputes the maxfd limit.
1207 * The file descriptor is also closed.
1208 */
willy tarreau0f7af912005-12-17 12:21:26 +01001209static inline void fd_delete(int fd) {
willy tarreau0f7af912005-12-17 12:21:26 +01001210 FD_CLR(fd, StaticReadEvent);
1211 FD_CLR(fd, StaticWriteEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01001212 close(fd);
1213 fdtab[fd].state = FD_STCLOSE;
willy tarreau0f7af912005-12-17 12:21:26 +01001214
1215 while ((maxfd-1 >= 0) && (fdtab[maxfd-1].state == FD_STCLOSE))
1216 maxfd--;
1217}
1218
1219/* recomputes the maxfd limit from the fd */
1220static inline void fd_insert(int fd) {
1221 if (fd+1 > maxfd)
1222 maxfd = fd+1;
1223}
1224
1225/*************************************************************/
1226/* task management ***************************************/
1227/*************************************************************/
1228
willy tarreau5cbea6f2005-12-17 12:48:26 +01001229/* puts the task <t> in run queue <q>, and returns <t> */
1230static inline struct task *task_wakeup(struct task **q, struct task *t) {
1231 if (t->state == TASK_RUNNING)
1232 return t;
willy tarreau0f7af912005-12-17 12:21:26 +01001233 else {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001234 t->rqnext = *q;
1235 t->state = TASK_RUNNING;
1236 return *q = t;
willy tarreau0f7af912005-12-17 12:21:26 +01001237 }
1238}
1239
willy tarreau5cbea6f2005-12-17 12:48:26 +01001240/* removes the task <t> from the queue <q>
1241 * <s> MUST be <q>'s first task.
willy tarreau0f7af912005-12-17 12:21:26 +01001242 * set the run queue to point to the next one, and return it
1243 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001244static inline struct task *task_sleep(struct task **q, struct task *t) {
1245 if (t->state == TASK_RUNNING) {
1246 *q = t->rqnext;
1247 t->state = TASK_IDLE; /* tell that s has left the run queue */
willy tarreau0f7af912005-12-17 12:21:26 +01001248 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01001249 return *q; /* return next running task */
willy tarreau0f7af912005-12-17 12:21:26 +01001250}
1251
1252/*
willy tarreau5cbea6f2005-12-17 12:48:26 +01001253 * removes the task <t> from its wait queue. It must have already been removed
willy tarreau0f7af912005-12-17 12:21:26 +01001254 * from the run queue. A pointer to the task itself is returned.
1255 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001256static inline struct task *task_delete(struct task *t) {
1257 t->prev->next = t->next;
1258 t->next->prev = t->prev;
1259 return t;
willy tarreau0f7af912005-12-17 12:21:26 +01001260}
1261
1262/*
willy tarreau5cbea6f2005-12-17 12:48:26 +01001263 * frees a task. Its context must have been freed since it will be lost.
willy tarreau0f7af912005-12-17 12:21:26 +01001264 */
1265static inline void task_free(struct task *t) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001266 pool_free(task, t);
willy tarreau0f7af912005-12-17 12:21:26 +01001267}
1268
willy tarreau5cbea6f2005-12-17 12:48:26 +01001269/* inserts <task> into its assigned wait queue, where it may already be. In this case, it
willy tarreau0f7af912005-12-17 12:21:26 +01001270 * may be only moved or left where it was, depending on its timing requirements.
1271 * <task> is returned.
1272 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001273struct task *task_queue(struct task *task) {
1274 struct task *list = task->wq;
willy tarreau0f7af912005-12-17 12:21:26 +01001275 struct task *start_from;
1276
1277 /* first, test if the task was already in a list */
1278 if (task->prev == NULL) {
1279 // start_from = list;
1280 start_from = list->prev;
willy tarreau750a4722005-12-17 13:21:24 +01001281#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001282 stats_tsk_new++;
willy tarreau750a4722005-12-17 13:21:24 +01001283#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001284 /* insert the unlinked <task> into the list, searching back from the last entry */
1285 while (start_from != list && tv_cmp2(&task->expire, &start_from->expire) < 0) {
1286 start_from = start_from->prev;
willy tarreau750a4722005-12-17 13:21:24 +01001287#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001288 stats_tsk_nsrch++;
willy tarreau750a4722005-12-17 13:21:24 +01001289#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001290 }
1291
1292 // while (start_from->next != list && tv_cmp2(&task->expire, &start_from->next->expire) > 0) {
1293 // start_from = start_from->next;
1294 // stats_tsk_nsrch++;
1295 // }
1296 }
1297 else if (task->prev == list ||
1298 tv_cmp2(&task->expire, &task->prev->expire) >= 0) { /* walk right */
1299 start_from = task->next;
1300 if (start_from == list || tv_cmp2(&task->expire, &start_from->expire) <= 0) {
willy tarreau750a4722005-12-17 13:21:24 +01001301#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001302 stats_tsk_good++;
willy tarreau750a4722005-12-17 13:21:24 +01001303#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001304 return task; /* it's already in the right place */
1305 }
1306
willy tarreau750a4722005-12-17 13:21:24 +01001307#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001308 stats_tsk_right++;
willy tarreau750a4722005-12-17 13:21:24 +01001309#endif
1310
1311 /* if the task is not at the right place, there's little chance that
1312 * it has only shifted a bit, and it will nearly always be queued
1313 * at the end of the list because of constant timeouts
1314 * (observed in real case).
1315 */
1316#ifndef WE_REALLY_THINK_THAT_THIS_TASK_MAY_HAVE_SHIFTED
1317 start_from = list->prev; /* assume we'll queue to the end of the list */
1318 while (start_from != list && tv_cmp2(&task->expire, &start_from->expire) < 0) {
1319 start_from = start_from->prev;
1320#if STATTIME > 0
1321 stats_tsk_lsrch++;
1322#endif
1323 }
1324#else /* WE_REALLY_... */
willy tarreau0f7af912005-12-17 12:21:26 +01001325 /* insert the unlinked <task> into the list, searching after position <start_from> */
1326 while (start_from->next != list && tv_cmp2(&task->expire, &start_from->next->expire) > 0) {
1327 start_from = start_from->next;
willy tarreau750a4722005-12-17 13:21:24 +01001328#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001329 stats_tsk_rsrch++;
willy tarreau750a4722005-12-17 13:21:24 +01001330#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001331 }
willy tarreau750a4722005-12-17 13:21:24 +01001332#endif /* WE_REALLY_... */
1333
willy tarreau0f7af912005-12-17 12:21:26 +01001334 /* we need to unlink it now */
1335 task_delete(task);
1336 }
1337 else { /* walk left. */
willy tarreau750a4722005-12-17 13:21:24 +01001338#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001339 stats_tsk_left++;
willy tarreau750a4722005-12-17 13:21:24 +01001340#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001341#ifdef LEFT_TO_TOP /* not very good */
1342 start_from = list;
1343 while (start_from->next != list && tv_cmp2(&task->expire, &start_from->next->expire) > 0) {
1344 start_from = start_from->next;
willy tarreau750a4722005-12-17 13:21:24 +01001345#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001346 stats_tsk_lsrch++;
willy tarreau750a4722005-12-17 13:21:24 +01001347#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001348 }
1349#else
1350 start_from = task->prev->prev; /* valid because of the previous test above */
1351 while (start_from != list && tv_cmp2(&task->expire, &start_from->expire) < 0) {
1352 start_from = start_from->prev;
willy tarreau750a4722005-12-17 13:21:24 +01001353#if STATTIME > 0
willy tarreau0f7af912005-12-17 12:21:26 +01001354 stats_tsk_lsrch++;
willy tarreau750a4722005-12-17 13:21:24 +01001355#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001356 }
1357#endif
1358 /* we need to unlink it now */
1359 task_delete(task);
1360 }
1361 task->prev = start_from;
1362 task->next = start_from->next;
1363 task->next->prev = task;
1364 start_from->next = task;
1365 return task;
1366}
1367
1368
1369/*********************************************************************/
1370/* more specific functions ***************************************/
1371/*********************************************************************/
1372
1373/* some prototypes */
1374static int maintain_proxies(void);
1375
willy tarreau5cbea6f2005-12-17 12:48:26 +01001376/* this either returns the sockname or the original destination address. Code
1377 * inspired from Patrick Schaaf's example of nf_getsockname() implementation.
1378 */
1379static int get_original_dst(int fd, struct sockaddr_in *sa, int *salen) {
willy tarreaua1598082005-12-17 13:08:06 +01001380#if defined(TPROXY) && defined(SO_ORIGINAL_DST)
willy tarreau5cbea6f2005-12-17 12:48:26 +01001381 return getsockopt(fd, SOL_IP, SO_ORIGINAL_DST, (void *)sa, salen);
1382#else
willy tarreaua1598082005-12-17 13:08:06 +01001383#if defined(TPROXY) && defined(USE_GETSOCKNAME)
willy tarreau5cbea6f2005-12-17 12:48:26 +01001384 return getsockname(fd, (struct sockaddr *)sa, salen);
1385#else
1386 return -1;
1387#endif
1388#endif
1389}
1390
1391/*
1392 * frees the context associated to a session. It must have been removed first.
1393 */
1394static inline void session_free(struct session *s) {
1395 if (s->req)
1396 pool_free(buffer, s->req);
1397 if (s->rep)
1398 pool_free(buffer, s->rep);
willy tarreaua1598082005-12-17 13:08:06 +01001399 if (s->logs.uri)
1400 pool_free(requri, s->logs.uri);
willy tarreau8337c6b2005-12-17 13:41:01 +01001401 if (s->logs.cli_cookie)
1402 pool_free(capture, s->logs.cli_cookie);
1403 if (s->logs.srv_cookie)
1404 pool_free(capture, s->logs.srv_cookie);
willy tarreau9fe663a2005-12-17 13:02:59 +01001405
willy tarreau5cbea6f2005-12-17 12:48:26 +01001406 pool_free(session, s);
1407}
1408
willy tarreau0f7af912005-12-17 12:21:26 +01001409
1410/*
willy tarreau8337c6b2005-12-17 13:41:01 +01001411 * This function tries to find a running server for the proxy <px>. A first
1412 * pass looks for active servers, and if none is found, a second pass also
1413 * looks for backup servers.
1414 * If no valid server is found, NULL is returned and px->cursrv is left undefined.
1415 */
1416static inline struct server *find_server(struct proxy *px) {
1417 struct server *srv = px->cursrv;
1418 int ignore_backup = 1;
1419
1420 do {
1421 do {
1422 if (srv == NULL)
1423 srv = px->srv;
1424 if (srv->state & SRV_RUNNING
1425 && !((srv->state & SRV_BACKUP) && ignore_backup))
1426 return srv;
1427 srv = srv->next;
1428 } while (srv != px->cursrv);
1429 } while (ignore_backup--);
1430 return NULL;
1431}
1432
1433/*
willy tarreau5cbea6f2005-12-17 12:48:26 +01001434 * This function initiates a connection to the current server (s->srv) if (s->direct)
1435 * is set, or to the dispatch server if (s->direct) is 0. It returns 0 if
willy tarreau0f7af912005-12-17 12:21:26 +01001436 * it's OK, -1 if it's impossible.
1437 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001438int connect_server(struct session *s) {
willy tarreau0f7af912005-12-17 12:21:26 +01001439 int one = 1;
1440 int fd;
1441
1442 // fprintf(stderr,"connect_server : s=%p\n",s);
1443
willy tarreaue39cd132005-12-17 13:00:18 +01001444 if (s->flags & SN_DIRECT) { /* srv cannot be null */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001445 s->srv_addr = s->srv->addr;
1446 }
1447 else if (s->proxy->options & PR_O_BALANCE) {
1448 if (s->proxy->options & PR_O_BALANCE_RR) {
willy tarreau8337c6b2005-12-17 13:41:01 +01001449 struct server *srv;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001450
willy tarreau8337c6b2005-12-17 13:41:01 +01001451 srv = find_server(s->proxy);
1452
1453 if (srv == NULL) /* no server left */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001454 return -1;
1455
willy tarreau8337c6b2005-12-17 13:41:01 +01001456 s->srv_addr = srv->addr;
1457 s->srv = srv;
1458 s->proxy->cursrv = srv->next;
willy tarreau0f7af912005-12-17 12:21:26 +01001459 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01001460 else /* unknown balancing algorithm */
1461 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01001462 }
willy tarreaua1598082005-12-17 13:08:06 +01001463 else if (*(int *)&s->proxy->dispatch_addr.sin_addr) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001464 /* connect to the defined dispatch addr */
willy tarreau0f7af912005-12-17 12:21:26 +01001465 s->srv_addr = s->proxy->dispatch_addr;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001466 }
1467 else if (s->proxy->options & PR_O_TRANSP) {
1468 /* in transparent mode, use the original dest addr if no dispatch specified */
1469 int salen = sizeof(struct sockaddr_in);
1470 if (get_original_dst(s->cli_fd, &s->srv_addr, &salen) == -1) {
1471 qfprintf(stderr, "Cannot get original server address.\n");
1472 return -1;
1473 }
1474 }
willy tarreau0f7af912005-12-17 12:21:26 +01001475
willy tarreaua41a8b42005-12-17 14:02:24 +01001476 /* if this server remaps proxied ports, we'll use
1477 * the port the client connected to with an offset. */
willy tarreaueedaa9f2005-12-17 14:08:03 +01001478 if (s->srv != NULL && s->srv->state & SRV_MAPPORTS) {
willy tarreaua41a8b42005-12-17 14:02:24 +01001479 struct sockaddr_in sockname;
1480 int namelen;
1481
1482 namelen = sizeof(sockname);
1483 if (get_original_dst(s->cli_fd, (struct sockaddr_in *)&sockname, &namelen) == -1)
1484 getsockname(s->cli_fd, (struct sockaddr *)&sockname, &namelen);
1485 s->srv_addr.sin_port = htons(ntohs(s->srv_addr.sin_port) + ntohs(sockname.sin_port));
1486 }
1487
willy tarreau0f7af912005-12-17 12:21:26 +01001488 if ((fd = s->srv_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001489 qfprintf(stderr, "Cannot get a server socket.\n");
willy tarreau0f7af912005-12-17 12:21:26 +01001490 return -1;
1491 }
1492
willy tarreau9fe663a2005-12-17 13:02:59 +01001493 if (fd >= global.maxsock) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001494 Alert("socket(): not enough free sockets. Raise -n argument. Giving up.\n");
1495 close(fd);
1496 return -1;
1497 }
1498
willy tarreau0f7af912005-12-17 12:21:26 +01001499 if ((fcntl(fd, F_SETFL, O_NONBLOCK)==-1) ||
1500 (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char *) &one, sizeof(one)) == -1)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001501 qfprintf(stderr,"Cannot set client socket to non blocking mode.\n");
willy tarreau0f7af912005-12-17 12:21:26 +01001502 close(fd);
1503 return -1;
1504 }
1505
willy tarreaua1598082005-12-17 13:08:06 +01001506 /* allow specific binding */
1507 if (s->proxy->options & PR_O_BIND_SRC &&
1508 bind(fd, (struct sockaddr *)&s->proxy->source_addr, sizeof(s->proxy->source_addr)) == -1) {
1509 Alert("Cannot bind to source address before connect() for proxy %s. Aborting.\n", s->proxy->id);
1510 close(fd);
1511 return -1;
1512 }
1513
willy tarreau0f7af912005-12-17 12:21:26 +01001514 if ((connect(fd, (struct sockaddr *)&s->srv_addr, sizeof(s->srv_addr)) == -1) && (errno != EINPROGRESS)) {
1515 if (errno == EAGAIN) { /* no free ports left, try again later */
willy tarreau5cbea6f2005-12-17 12:48:26 +01001516 qfprintf(stderr,"Cannot connect, no free ports.\n");
willy tarreau0f7af912005-12-17 12:21:26 +01001517 close(fd);
1518 return -1;
1519 }
1520 else if (errno != EALREADY && errno != EISCONN) {
1521 close(fd);
1522 return -1;
1523 }
1524 }
1525
willy tarreau5cbea6f2005-12-17 12:48:26 +01001526 fdtab[fd].owner = s->task;
willy tarreau0f7af912005-12-17 12:21:26 +01001527 fdtab[fd].read = &event_srv_read;
1528 fdtab[fd].write = &event_srv_write;
1529 fdtab[fd].state = FD_STCONN; /* connection in progress */
1530
1531 FD_SET(fd, StaticWriteEvent); /* for connect status */
1532
1533 fd_insert(fd);
1534
1535 if (s->proxy->contimeout)
1536 tv_delayfrom(&s->cnexpire, &now, s->proxy->contimeout);
1537 else
1538 tv_eternity(&s->cnexpire);
1539 return 0;
1540}
1541
1542/*
1543 * this function is called on a read event from a client socket.
1544 * It returns 0.
1545 */
1546int event_cli_read(int fd) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001547 struct task *t = fdtab[fd].owner;
1548 struct session *s = t->context;
willy tarreau0f7af912005-12-17 12:21:26 +01001549 struct buffer *b = s->req;
1550 int ret, max;
willy tarreau0f7af912005-12-17 12:21:26 +01001551
1552 // fprintf(stderr,"event_cli_read : fd=%d, s=%p\n", fd, s);
1553
willy tarreau0f7af912005-12-17 12:21:26 +01001554 if (fdtab[fd].state != FD_STERROR) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001555 while (1) {
1556 if (b->l == 0) { /* let's realign the buffer to optimize I/O */
1557 b->r = b->w = b->h = b->lr = b->data;
willy tarreauef900ab2005-12-17 12:52:52 +01001558 max = b->rlim - b->data;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001559 }
1560 else if (b->r > b->w) {
willy tarreauef900ab2005-12-17 12:52:52 +01001561 max = b->rlim - b->r;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001562 }
1563 else {
1564 max = b->w - b->r;
willy tarreauef900ab2005-12-17 12:52:52 +01001565 /* FIXME: theorically, if w>0, we shouldn't have rlim < data+size anymore
1566 * since it means that the rewrite protection has been removed. This
1567 * implies that the if statement can be removed.
1568 */
1569 if (max > b->rlim - b->data)
1570 max = b->rlim - b->data;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001571 }
1572
1573 if (max == 0) { /* not anymore room to store data */
1574 FD_CLR(fd, StaticReadEvent);
willy tarreauef900ab2005-12-17 12:52:52 +01001575 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001576 }
1577
willy tarreau3242e862005-12-17 12:27:53 +01001578#ifndef MSG_NOSIGNAL
willy tarreau5cbea6f2005-12-17 12:48:26 +01001579 {
1580 int skerr, lskerr;
1581
1582 lskerr = sizeof(skerr);
1583 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
1584 if (skerr)
1585 ret = -1;
1586 else
1587 ret = recv(fd, b->r, max, 0);
1588 }
willy tarreau3242e862005-12-17 12:27:53 +01001589#else
willy tarreau5cbea6f2005-12-17 12:48:26 +01001590 ret = recv(fd, b->r, max, MSG_NOSIGNAL);
willy tarreau3242e862005-12-17 12:27:53 +01001591#endif
willy tarreau5cbea6f2005-12-17 12:48:26 +01001592 if (ret > 0) {
1593 b->r += ret;
1594 b->l += ret;
1595 s->res_cr = RES_DATA;
1596
1597 if (b->r == b->data + BUFSIZE) {
1598 b->r = b->data; /* wrap around the buffer */
1599 }
willy tarreaua1598082005-12-17 13:08:06 +01001600
1601 b->total += ret;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001602 /* we hope to read more data or to get a close on next round */
1603 continue;
willy tarreau0f7af912005-12-17 12:21:26 +01001604 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01001605 else if (ret == 0) {
1606 s->res_cr = RES_NULL;
1607 break;
1608 }
1609 else if (errno == EAGAIN) {/* ignore EAGAIN */
1610 break;
1611 }
1612 else {
1613 s->res_cr = RES_ERROR;
1614 fdtab[fd].state = FD_STERROR;
1615 break;
1616 }
1617 } /* while(1) */
willy tarreau0f7af912005-12-17 12:21:26 +01001618 }
1619 else {
1620 s->res_cr = RES_ERROR;
1621 fdtab[fd].state = FD_STERROR;
1622 }
1623
willy tarreau5cbea6f2005-12-17 12:48:26 +01001624 if (s->res_cr != RES_SILENT) {
willy tarreaub1ff9db2005-12-17 13:51:03 +01001625 if (s->proxy->clitimeout && FD_ISSET(fd, StaticReadEvent))
willy tarreau5cbea6f2005-12-17 12:48:26 +01001626 tv_delayfrom(&s->crexpire, &now, s->proxy->clitimeout);
1627 else
1628 tv_eternity(&s->crexpire);
1629
1630 task_wakeup(&rq, t);
1631 }
willy tarreau0f7af912005-12-17 12:21:26 +01001632
willy tarreau0f7af912005-12-17 12:21:26 +01001633 return 0;
1634}
1635
1636
1637/*
1638 * this function is called on a read event from a server socket.
1639 * It returns 0.
1640 */
1641int event_srv_read(int fd) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001642 struct task *t = fdtab[fd].owner;
1643 struct session *s = t->context;
willy tarreau0f7af912005-12-17 12:21:26 +01001644 struct buffer *b = s->rep;
1645 int ret, max;
willy tarreau0f7af912005-12-17 12:21:26 +01001646
1647 // fprintf(stderr,"event_srv_read : fd=%d, s=%p\n", fd, s);
1648
willy tarreau0f7af912005-12-17 12:21:26 +01001649 if (fdtab[fd].state != FD_STERROR) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001650 while (1) {
1651 if (b->l == 0) { /* let's realign the buffer to optimize I/O */
1652 b->r = b->w = b->h = b->lr = b->data;
willy tarreauef900ab2005-12-17 12:52:52 +01001653 max = b->rlim - b->data;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001654 }
1655 else if (b->r > b->w) {
willy tarreauef900ab2005-12-17 12:52:52 +01001656 max = b->rlim - b->r;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001657 }
1658 else {
1659 max = b->w - b->r;
willy tarreauef900ab2005-12-17 12:52:52 +01001660 /* FIXME: theorically, if w>0, we shouldn't have rlim < data+size anymore
1661 * since it means that the rewrite protection has been removed. This
1662 * implies that the if statement can be removed.
1663 */
1664 if (max > b->rlim - b->data)
1665 max = b->rlim - b->data;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001666 }
1667
1668 if (max == 0) { /* not anymore room to store data */
1669 FD_CLR(fd, StaticReadEvent);
1670 break;
1671 }
1672
willy tarreau3242e862005-12-17 12:27:53 +01001673#ifndef MSG_NOSIGNAL
willy tarreau5cbea6f2005-12-17 12:48:26 +01001674 {
1675 int skerr, lskerr;
1676
1677 lskerr = sizeof(skerr);
1678 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
1679 if (skerr)
1680 ret = -1;
1681 else
1682 ret = recv(fd, b->r, max, 0);
1683 }
willy tarreau3242e862005-12-17 12:27:53 +01001684#else
willy tarreau5cbea6f2005-12-17 12:48:26 +01001685 ret = recv(fd, b->r, max, MSG_NOSIGNAL);
willy tarreau3242e862005-12-17 12:27:53 +01001686#endif
willy tarreau5cbea6f2005-12-17 12:48:26 +01001687 if (ret > 0) {
1688 b->r += ret;
1689 b->l += ret;
1690 s->res_sr = RES_DATA;
willy tarreau0f7af912005-12-17 12:21:26 +01001691
willy tarreau5cbea6f2005-12-17 12:48:26 +01001692 if (b->r == b->data + BUFSIZE) {
1693 b->r = b->data; /* wrap around the buffer */
1694 }
willy tarreaua1598082005-12-17 13:08:06 +01001695
1696 b->total += ret;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001697 /* we hope to read more data or to get a close on next round */
1698 continue;
willy tarreau0f7af912005-12-17 12:21:26 +01001699 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01001700 else if (ret == 0) {
1701 s->res_sr = RES_NULL;
1702 break;
1703 }
1704 else if (errno == EAGAIN) {/* ignore EAGAIN */
1705 break;
1706 }
1707 else {
1708 s->res_sr = RES_ERROR;
1709 fdtab[fd].state = FD_STERROR;
1710 break;
1711 }
1712 } /* while(1) */
willy tarreau0f7af912005-12-17 12:21:26 +01001713 }
1714 else {
1715 s->res_sr = RES_ERROR;
1716 fdtab[fd].state = FD_STERROR;
1717 }
1718
willy tarreau5cbea6f2005-12-17 12:48:26 +01001719 if (s->res_sr != RES_SILENT) {
willy tarreaub1ff9db2005-12-17 13:51:03 +01001720 if (s->proxy->srvtimeout && FD_ISSET(fd, StaticReadEvent))
willy tarreau5cbea6f2005-12-17 12:48:26 +01001721 tv_delayfrom(&s->srexpire, &now, s->proxy->srvtimeout);
1722 else
1723 tv_eternity(&s->srexpire);
1724
1725 task_wakeup(&rq, t);
1726 }
willy tarreau0f7af912005-12-17 12:21:26 +01001727
willy tarreau0f7af912005-12-17 12:21:26 +01001728 return 0;
1729}
1730
1731/*
1732 * this function is called on a write event from a client socket.
1733 * It returns 0.
1734 */
1735int event_cli_write(int fd) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001736 struct task *t = fdtab[fd].owner;
1737 struct session *s = t->context;
willy tarreau0f7af912005-12-17 12:21:26 +01001738 struct buffer *b = s->rep;
1739 int ret, max;
willy tarreau0f7af912005-12-17 12:21:26 +01001740
1741 // fprintf(stderr,"event_cli_write : fd=%d, s=%p\n", fd, s);
1742
1743 if (b->l == 0) { /* let's realign the buffer to optimize I/O */
willy tarreau9da061b2005-12-17 12:29:56 +01001744 b->r = b->w = b->h = b->lr = b->data;
willy tarreau0f7af912005-12-17 12:21:26 +01001745 // max = BUFSIZE; BUG !!!!
1746 max = 0;
1747 }
1748 else if (b->r > b->w) {
1749 max = b->r - b->w;
1750 }
1751 else
1752 max = b->data + BUFSIZE - b->w;
1753
willy tarreau0f7af912005-12-17 12:21:26 +01001754 if (fdtab[fd].state != FD_STERROR) {
willy tarreau3242e862005-12-17 12:27:53 +01001755#ifndef MSG_NOSIGNAL
1756 int skerr, lskerr;
1757#endif
willy tarreauef900ab2005-12-17 12:52:52 +01001758
1759 if (max == 0) {
1760 s->res_cw = RES_NULL;
1761 task_wakeup(&rq, t);
willy tarreaub1ff9db2005-12-17 13:51:03 +01001762 tv_eternity(&s->cwexpire);
1763 FD_CLR(fd, StaticWriteEvent);
willy tarreauef900ab2005-12-17 12:52:52 +01001764 return 0;
willy tarreau0f7af912005-12-17 12:21:26 +01001765 }
1766
willy tarreau3242e862005-12-17 12:27:53 +01001767#ifndef MSG_NOSIGNAL
1768 lskerr=sizeof(skerr);
1769 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
1770 if (skerr)
1771 ret = -1;
1772 else
1773 ret = send(fd, b->w, max, MSG_DONTWAIT);
1774#else
willy tarreau0f7af912005-12-17 12:21:26 +01001775 ret = send(fd, b->w, max, MSG_DONTWAIT | MSG_NOSIGNAL);
willy tarreau3242e862005-12-17 12:27:53 +01001776#endif
willy tarreau0f7af912005-12-17 12:21:26 +01001777
1778 if (ret > 0) {
1779 b->l -= ret;
1780 b->w += ret;
1781
1782 s->res_cw = RES_DATA;
1783
1784 if (b->w == b->data + BUFSIZE) {
1785 b->w = b->data; /* wrap around the buffer */
1786 }
1787 }
1788 else if (ret == 0) {
1789 /* nothing written, just make as if we were never called */
1790// s->res_cw = RES_NULL;
1791 return 0;
1792 }
1793 else if (errno == EAGAIN) /* ignore EAGAIN */
1794 return 0;
1795 else {
1796 s->res_cw = RES_ERROR;
1797 fdtab[fd].state = FD_STERROR;
1798 }
1799 }
1800 else {
1801 s->res_cw = RES_ERROR;
1802 fdtab[fd].state = FD_STERROR;
1803 }
1804
willy tarreaub1ff9db2005-12-17 13:51:03 +01001805 if (s->proxy->clitimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01001806 tv_delayfrom(&s->cwexpire, &now, s->proxy->clitimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01001807 /* FIXME: to avoid the client to read-time-out during writes, we refresh it */
1808 s->crexpire = s->cwexpire;
1809 }
willy tarreau0f7af912005-12-17 12:21:26 +01001810 else
1811 tv_eternity(&s->cwexpire);
1812
willy tarreau5cbea6f2005-12-17 12:48:26 +01001813 task_wakeup(&rq, t);
willy tarreau0f7af912005-12-17 12:21:26 +01001814 return 0;
1815}
1816
1817
1818/*
1819 * this function is called on a write event from a server socket.
1820 * It returns 0.
1821 */
1822int event_srv_write(int fd) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01001823 struct task *t = fdtab[fd].owner;
1824 struct session *s = t->context;
willy tarreau0f7af912005-12-17 12:21:26 +01001825 struct buffer *b = s->req;
1826 int ret, max;
willy tarreau0f7af912005-12-17 12:21:26 +01001827
1828 //fprintf(stderr,"event_srv_write : fd=%d, s=%p\n", fd, s);
1829
1830 if (b->l == 0) { /* let's realign the buffer to optimize I/O */
willy tarreau9da061b2005-12-17 12:29:56 +01001831 b->r = b->w = b->h = b->lr = b->data;
willy tarreau0f7af912005-12-17 12:21:26 +01001832 // max = BUFSIZE; BUG !!!!
1833 max = 0;
1834 }
1835 else if (b->r > b->w) {
1836 max = b->r - b->w;
1837 }
1838 else
1839 max = b->data + BUFSIZE - b->w;
1840
willy tarreau0f7af912005-12-17 12:21:26 +01001841 if (fdtab[fd].state != FD_STERROR) {
willy tarreau3242e862005-12-17 12:27:53 +01001842#ifndef MSG_NOSIGNAL
1843 int skerr, lskerr;
1844#endif
willy tarreauef900ab2005-12-17 12:52:52 +01001845 if (max == 0) {
1846 /* may be we have received a connection acknowledgement in TCP mode without data */
willy tarreau0f7af912005-12-17 12:21:26 +01001847 s->res_sw = RES_NULL;
willy tarreau5cbea6f2005-12-17 12:48:26 +01001848 task_wakeup(&rq, t);
willy tarreauef900ab2005-12-17 12:52:52 +01001849 fdtab[fd].state = FD_STREADY;
willy tarreaub1ff9db2005-12-17 13:51:03 +01001850 tv_eternity(&s->swexpire);
1851 FD_CLR(fd, StaticWriteEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01001852 return 0;
1853 }
1854
willy tarreauef900ab2005-12-17 12:52:52 +01001855
willy tarreau3242e862005-12-17 12:27:53 +01001856#ifndef MSG_NOSIGNAL
1857 lskerr=sizeof(skerr);
1858 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
1859 if (skerr)
1860 ret = -1;
1861 else
1862 ret = send(fd, b->w, max, MSG_DONTWAIT);
1863#else
willy tarreau0f7af912005-12-17 12:21:26 +01001864 ret = send(fd, b->w, max, MSG_DONTWAIT | MSG_NOSIGNAL);
willy tarreau3242e862005-12-17 12:27:53 +01001865#endif
willy tarreauef900ab2005-12-17 12:52:52 +01001866 fdtab[fd].state = FD_STREADY;
willy tarreau0f7af912005-12-17 12:21:26 +01001867 if (ret > 0) {
1868 b->l -= ret;
1869 b->w += ret;
1870
1871 s->res_sw = RES_DATA;
1872
1873 if (b->w == b->data + BUFSIZE) {
1874 b->w = b->data; /* wrap around the buffer */
1875 }
1876 }
1877 else if (ret == 0) {
1878 /* nothing written, just make as if we were never called */
1879 // s->res_sw = RES_NULL;
1880 return 0;
1881 }
1882 else if (errno == EAGAIN) /* ignore EAGAIN */
1883 return 0;
1884 else {
1885 s->res_sw = RES_ERROR;
1886 fdtab[fd].state = FD_STERROR;
1887 }
1888 }
1889 else {
1890 s->res_sw = RES_ERROR;
1891 fdtab[fd].state = FD_STERROR;
1892 }
1893
willy tarreaub1ff9db2005-12-17 13:51:03 +01001894 if (s->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01001895 tv_delayfrom(&s->swexpire, &now, s->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01001896 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
1897 s->srexpire = s->swexpire;
1898 }
willy tarreau0f7af912005-12-17 12:21:26 +01001899 else
1900 tv_eternity(&s->swexpire);
1901
willy tarreau5cbea6f2005-12-17 12:48:26 +01001902 task_wakeup(&rq, t);
willy tarreau0f7af912005-12-17 12:21:26 +01001903 return 0;
1904}
1905
1906
1907/*
willy tarreaue39cd132005-12-17 13:00:18 +01001908 * returns a message to the client ; the connection is shut down for read,
1909 * and the request is cleared so that no server connection can be initiated.
1910 * The client must be in a valid state for this (HEADER, DATA ...).
1911 * Nothing is performed on the server side.
willy tarreau8337c6b2005-12-17 13:41:01 +01001912 * The reply buffer doesn't need to be empty before this.
willy tarreaue39cd132005-12-17 13:00:18 +01001913 */
1914void client_retnclose(struct session *s, int len, const char *msg) {
1915 FD_CLR(s->cli_fd, StaticReadEvent);
1916 FD_SET(s->cli_fd, StaticWriteEvent);
1917 tv_eternity(&s->crexpire);
1918 shutdown(s->cli_fd, SHUT_RD);
1919 s->cli_state = CL_STSHUTR;
1920 strcpy(s->rep->data, msg);
1921 s->rep->l = len;
willy tarreau8337c6b2005-12-17 13:41:01 +01001922 s->rep->r = s->rep->h = s->rep->lr = s->rep->w = s->rep->data;
willy tarreaue39cd132005-12-17 13:00:18 +01001923 s->rep->r += len;
1924 s->req->l = 0;
1925}
1926
1927
1928/*
1929 * returns a message into the rep buffer, and flushes the req buffer.
willy tarreau8337c6b2005-12-17 13:41:01 +01001930 * The reply buffer doesn't need to be empty before this.
willy tarreaue39cd132005-12-17 13:00:18 +01001931 */
1932void client_return(struct session *s, int len, const char *msg) {
1933 strcpy(s->rep->data, msg);
1934 s->rep->l = len;
willy tarreau8337c6b2005-12-17 13:41:01 +01001935 s->rep->r = s->rep->h = s->rep->lr = s->rep->w = s->rep->data;
willy tarreaue39cd132005-12-17 13:00:18 +01001936 s->rep->r += len;
1937 s->req->l = 0;
1938}
1939
willy tarreau9fe663a2005-12-17 13:02:59 +01001940/*
1941 * send a log for the session when we have enough info about it
1942 */
1943void sess_log(struct session *s) {
1944 unsigned char *pn;
1945 struct proxy *p = s->proxy;
1946 int log;
1947 char *uri;
1948 char *pxid;
1949 char *srv;
willy tarreauc1cae632005-12-17 14:12:23 +01001950 struct tm *tm;
willy tarreau9fe663a2005-12-17 13:02:59 +01001951
1952 /* This is a first attempt at a better logging system.
1953 * For now, we rely on send_log() to provide the date, although it obviously
1954 * is the date of the log and not of the request, and most fields are not
1955 * computed.
1956 */
1957
willy tarreaua1598082005-12-17 13:08:06 +01001958 log = p->to_log & ~s->logs.logwait;
willy tarreau9fe663a2005-12-17 13:02:59 +01001959
1960 pn = (log & LW_CLIP) ?
1961 (unsigned char *)&s->cli_addr.sin_addr :
1962 (unsigned char *)"\0\0\0\0";
1963
willy tarreauc1cae632005-12-17 14:12:23 +01001964 uri = (log & LW_REQ) ? s->logs.uri ? s->logs.uri : "<BADREQ>" : "";
willy tarreau9fe663a2005-12-17 13:02:59 +01001965 pxid = p->id;
willy tarreauc1cae632005-12-17 14:12:23 +01001966 srv = (p->to_log & LW_SVID) ? (s->srv != NULL) ? s->srv->id : "<NOSRV>" : "-";
willy tarreaua1598082005-12-17 13:08:06 +01001967
willy tarreauc1cae632005-12-17 14:12:23 +01001968 tm = localtime(&s->logs.tv_accept.tv_sec);
1969 if (p->to_log & LW_REQ) {
willy tarreau036e1ce2005-12-17 13:46:33 +01001970 send_log(p, LOG_INFO, "%d.%d.%d.%d:%d [%02d/%s/%04d:%02d:%02d:%02d] %s %s %d/%d/%d/%d %d %lld %s %s %c%c%c%c \"%s\"\n",
willy tarreaua1598082005-12-17 13:08:06 +01001971 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port),
1972 tm->tm_mday, monthname[tm->tm_mon], tm->tm_year+1900,
1973 tm->tm_hour, tm->tm_min, tm->tm_sec,
1974 pxid, srv,
1975 s->logs.t_request,
1976 (s->logs.t_connect >= 0) ? s->logs.t_connect - s->logs.t_request : -1,
1977 (s->logs.t_data >= 0) ? s->logs.t_data - s->logs.t_connect : -1,
1978 s->logs.t_close,
1979 s->logs.status, s->logs.bytes,
willy tarreau8337c6b2005-12-17 13:41:01 +01001980 s->logs.cli_cookie ? s->logs.cli_cookie : "-",
1981 s->logs.srv_cookie ? s->logs.srv_cookie : "-",
willy tarreau036e1ce2005-12-17 13:46:33 +01001982 sess_term_cond[(s->flags & SN_ERR_MASK) >> SN_ERR_SHIFT],
1983 sess_fin_state[(s->flags & SN_FINST_MASK) >> SN_FINST_SHIFT],
1984 (p->options & PR_O_COOK_ANY) ? sess_cookie[(s->flags & SN_CK_MASK) >> SN_CK_SHIFT] : '-',
1985 (p->options & PR_O_COOK_ANY) ? sess_set_cookie[(s->flags & SN_SCK_MASK) >> SN_SCK_SHIFT] : '-',
willy tarreaua1598082005-12-17 13:08:06 +01001986 uri);
1987 }
1988 else {
willy tarreauc1cae632005-12-17 14:12:23 +01001989 send_log(p, LOG_INFO, "%d.%d.%d.%d:%d [%02d/%s/%04d:%02d:%02d:%02d] %s %s %d/%d %lld %c%c\n",
willy tarreaua1598082005-12-17 13:08:06 +01001990 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port),
willy tarreauc1cae632005-12-17 14:12:23 +01001991 tm->tm_mday, monthname[tm->tm_mon], tm->tm_year+1900,
1992 tm->tm_hour, tm->tm_min, tm->tm_sec,
willy tarreaua1598082005-12-17 13:08:06 +01001993 pxid, srv,
willy tarreauc1cae632005-12-17 14:12:23 +01001994 (s->logs.t_connect >= 0) ? s->logs.t_connect : -1,
willy tarreaua1598082005-12-17 13:08:06 +01001995 s->logs.t_close,
willy tarreauc1cae632005-12-17 14:12:23 +01001996 s->logs.bytes,
willy tarreau036e1ce2005-12-17 13:46:33 +01001997 sess_term_cond[(s->flags & SN_ERR_MASK) >> SN_ERR_SHIFT],
willy tarreauc1cae632005-12-17 14:12:23 +01001998 sess_fin_state[(s->flags & SN_FINST_MASK) >> SN_FINST_SHIFT]);
willy tarreaua1598082005-12-17 13:08:06 +01001999 }
2000
2001 s->logs.logwait = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01002002}
2003
willy tarreaue39cd132005-12-17 13:00:18 +01002004
2005/*
willy tarreau0f7af912005-12-17 12:21:26 +01002006 * this function is called on a read event from a listen socket, corresponding
willy tarreau5cbea6f2005-12-17 12:48:26 +01002007 * to an accept. It tries to accept as many connections as possible.
2008 * It returns 0.
willy tarreau0f7af912005-12-17 12:21:26 +01002009 */
2010int event_accept(int fd) {
2011 struct proxy *p = (struct proxy *)fdtab[fd].owner;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002012 struct session *s;
2013 struct task *t;
willy tarreau0f7af912005-12-17 12:21:26 +01002014 int cfd;
2015 int one = 1;
2016
willy tarreau5cbea6f2005-12-17 12:48:26 +01002017 while (p->nbconn < p->maxconn) {
2018 struct sockaddr_in addr;
2019 int laddr = sizeof(addr);
2020 if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1)
2021 return 0; /* nothing more to accept */
willy tarreau0f7af912005-12-17 12:21:26 +01002022
willy tarreau5cbea6f2005-12-17 12:48:26 +01002023 if ((s = pool_alloc(session)) == NULL) { /* disable this proxy for a while */
2024 Alert("out of memory in event_accept().\n");
2025 FD_CLR(fd, StaticReadEvent);
2026 p->state = PR_STIDLE;
2027 close(cfd);
2028 return 0;
2029 }
willy tarreau0f7af912005-12-17 12:21:26 +01002030
willy tarreau5cbea6f2005-12-17 12:48:26 +01002031 if ((t = pool_alloc(task)) == NULL) { /* disable this proxy for a while */
2032 Alert("out of memory in event_accept().\n");
2033 FD_CLR(fd, StaticReadEvent);
2034 p->state = PR_STIDLE;
2035 close(cfd);
2036 pool_free(session, s);
2037 return 0;
2038 }
willy tarreau0f7af912005-12-17 12:21:26 +01002039
willy tarreau5cbea6f2005-12-17 12:48:26 +01002040 s->cli_addr = addr;
willy tarreau9fe663a2005-12-17 13:02:59 +01002041 if (cfd >= global.maxsock) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002042 Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
2043 close(cfd);
2044 pool_free(task, t);
2045 pool_free(session, s);
2046 return 0;
2047 }
willy tarreau0f7af912005-12-17 12:21:26 +01002048
willy tarreau5cbea6f2005-12-17 12:48:26 +01002049 if ((fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) ||
2050 (setsockopt(cfd, IPPROTO_TCP, TCP_NODELAY,
2051 (char *) &one, sizeof(one)) == -1)) {
2052 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
2053 close(cfd);
2054 pool_free(task, t);
2055 pool_free(session, s);
2056 return 0;
2057 }
willy tarreau0f7af912005-12-17 12:21:26 +01002058
willy tarreau9fe663a2005-12-17 13:02:59 +01002059 t->next = t->prev = t->rqnext = NULL; /* task not in run queue yet */
2060 t->wq = LIST_HEAD(wait_queue); /* but already has a wait queue assigned */
2061 t->state = TASK_IDLE;
2062 t->process = process_session;
2063 t->context = s;
2064
2065 s->task = t;
2066 s->proxy = p;
2067 s->cli_state = (p->mode == PR_MODE_HTTP) ? CL_STHEADERS : CL_STDATA; /* no HTTP headers for non-HTTP proxies */
2068 s->srv_state = SV_STIDLE;
2069 s->req = s->rep = NULL; /* will be allocated later */
2070 s->flags = 0;
2071 s->res_cr = s->res_cw = s->res_sr = s->res_sw = RES_SILENT;
2072 s->cli_fd = cfd;
2073 s->srv_fd = -1;
willy tarreaua1598082005-12-17 13:08:06 +01002074 s->srv = NULL;
willy tarreau9fe663a2005-12-17 13:02:59 +01002075 s->conn_retries = p->conn_retries;
willy tarreaua1598082005-12-17 13:08:06 +01002076
2077 s->logs.logwait = p->to_log;
2078 s->logs.tv_accept = now;
2079 s->logs.t_request = -1;
2080 s->logs.t_connect = -1;
2081 s->logs.t_data = -1;
2082 s->logs.t_close = 0;
2083 s->logs.uri = NULL;
willy tarreau8337c6b2005-12-17 13:41:01 +01002084 s->logs.cli_cookie = NULL;
2085 s->logs.srv_cookie = NULL;
willy tarreaua1598082005-12-17 13:08:06 +01002086 s->logs.status = -1;
2087 s->logs.bytes = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01002088
willy tarreau2f6ba652005-12-17 13:57:42 +01002089 s->uniq_id = totalconn;
2090
willy tarreau5cbea6f2005-12-17 12:48:26 +01002091 if ((p->mode == PR_MODE_TCP || p->mode == PR_MODE_HTTP)
2092 && (p->logfac1 >= 0 || p->logfac2 >= 0)) {
willy tarreau535ae7a2005-12-17 12:58:00 +01002093 struct sockaddr_in sockname;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002094 unsigned char *pn, *sn;
2095 int namelen;
willy tarreau0f7af912005-12-17 12:21:26 +01002096
willy tarreau5cbea6f2005-12-17 12:48:26 +01002097 namelen = sizeof(sockname);
2098 if (get_original_dst(cfd, (struct sockaddr_in *)&sockname, &namelen) == -1)
2099 getsockname(cfd, (struct sockaddr *)&sockname, &namelen);
2100 sn = (unsigned char *)&sockname.sin_addr;
willy tarreau535ae7a2005-12-17 12:58:00 +01002101 pn = (unsigned char *)&s->cli_addr.sin_addr;
willy tarreau0f7af912005-12-17 12:21:26 +01002102
willy tarreau9fe663a2005-12-17 13:02:59 +01002103 if (p->to_log) {
2104 /* we have the client ip */
willy tarreaua1598082005-12-17 13:08:06 +01002105 if (s->logs.logwait & LW_CLIP)
2106 if (!(s->logs.logwait &= ~LW_CLIP))
willy tarreau9fe663a2005-12-17 13:02:59 +01002107 sess_log(s);
2108 }
2109 else
2110 send_log(p, LOG_INFO, "Connect from %d.%d.%d.%d:%d to %d.%d.%d.%d:%d (%s/%s)\n",
2111 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port),
2112 sn[0], sn[1], sn[2], sn[3], ntohs(sockname.sin_port),
2113 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
willy tarreau5cbea6f2005-12-17 12:48:26 +01002114 }
willy tarreau0f7af912005-12-17 12:21:26 +01002115
willy tarreau9fe663a2005-12-17 13:02:59 +01002116 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau2f6ba652005-12-17 13:57:42 +01002117 struct sockaddr_in sockname;
2118 unsigned char *pn, *sn;
2119 int namelen;
willy tarreauef900ab2005-12-17 12:52:52 +01002120 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01002121
2122 namelen = sizeof(sockname);
2123 if (get_original_dst(cfd, (struct sockaddr_in *)&sockname, &namelen) == -1)
2124 getsockname(cfd, (struct sockaddr *)&sockname, &namelen);
2125 sn = (unsigned char *)&sockname.sin_addr;
2126 pn = (unsigned char *)&s->cli_addr.sin_addr;
2127
2128 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%d.%d.%d.%d:%d]\n",
2129 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
2130 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port));
willy tarreauef900ab2005-12-17 12:52:52 +01002131 write(1, trash, len);
2132 }
willy tarreau0f7af912005-12-17 12:21:26 +01002133
willy tarreau5cbea6f2005-12-17 12:48:26 +01002134 if ((s->req = pool_alloc(buffer)) == NULL) { /* no memory */
2135 close(cfd); /* nothing can be done for this fd without memory */
2136 pool_free(task, t);
2137 pool_free(session, s);
2138 return 0;
2139 }
2140 s->req->l = 0;
willy tarreaua1598082005-12-17 13:08:06 +01002141 s->req->total = 0;
willy tarreauef900ab2005-12-17 12:52:52 +01002142 s->req->h = s->req->r = s->req->lr = s->req->w = s->req->data; /* r and w will be reset further */
2143 s->req->rlim = s->req->data + BUFSIZE;
willy tarreaub1ff9db2005-12-17 13:51:03 +01002144 if (s->cli_state == CL_STHEADERS) /* reserve some space for header rewriting */
willy tarreauef900ab2005-12-17 12:52:52 +01002145 s->req->rlim -= MAXREWRITE;
willy tarreau0f7af912005-12-17 12:21:26 +01002146
willy tarreau5cbea6f2005-12-17 12:48:26 +01002147 if ((s->rep = pool_alloc(buffer)) == NULL) { /* no memory */
2148 pool_free(buffer, s->req);
2149 close(cfd); /* nothing can be done for this fd without memory */
2150 pool_free(task, t);
2151 pool_free(session, s);
2152 return 0;
2153 }
2154 s->rep->l = 0;
willy tarreaua1598082005-12-17 13:08:06 +01002155 s->rep->total = 0;
willy tarreauef900ab2005-12-17 12:52:52 +01002156 s->rep->h = s->rep->r = s->rep->lr = s->rep->w = s->rep->rlim = s->rep->data;
willy tarreau0f7af912005-12-17 12:21:26 +01002157
willy tarreau5cbea6f2005-12-17 12:48:26 +01002158 fdtab[cfd].read = &event_cli_read;
2159 fdtab[cfd].write = &event_cli_write;
2160 fdtab[cfd].owner = t;
2161 fdtab[cfd].state = FD_STREADY;
willy tarreau0f7af912005-12-17 12:21:26 +01002162
willy tarreau5cbea6f2005-12-17 12:48:26 +01002163 if (p->mode == PR_MODE_HEALTH) { /* health check mode, no client reading */
willy tarreau197e8ec2005-12-17 14:10:59 +01002164 if (p->options & PR_O_HTTP_CHK) /* "option httpchk" will make it speak HTTP */
2165 client_retnclose(s, 19, "HTTP/1.0 200 OK\r\n\r\n"); /* forge a 200 response */
2166 else
2167 client_retnclose(s, 3, "OK\n"); /* forge an "OK" response */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002168 }
2169 else {
2170 FD_SET(cfd, StaticReadEvent);
2171 }
2172
2173 fd_insert(cfd);
2174
2175 tv_eternity(&s->cnexpire);
2176 tv_eternity(&s->srexpire);
2177 tv_eternity(&s->swexpire);
2178 tv_eternity(&s->cwexpire);
2179
2180 if (s->proxy->clitimeout)
2181 tv_delayfrom(&s->crexpire, &now, s->proxy->clitimeout);
2182 else
2183 tv_eternity(&s->crexpire);
2184
2185 t->expire = s->crexpire;
2186
2187 task_queue(t);
willy tarreauef900ab2005-12-17 12:52:52 +01002188
2189 if (p->mode != PR_MODE_HEALTH)
2190 task_wakeup(&rq, t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002191
2192 p->nbconn++;
2193 actconn++;
2194 totalconn++;
2195
2196 // fprintf(stderr, "accepting from %p => %d conn, %d total\n", p, actconn, totalconn);
2197 } /* end of while (p->nbconn < p->maxconn) */
2198 return 0;
2199}
willy tarreau0f7af912005-12-17 12:21:26 +01002200
willy tarreau0f7af912005-12-17 12:21:26 +01002201
willy tarreau5cbea6f2005-12-17 12:48:26 +01002202/*
2203 * This function is used only for server health-checks. It handles
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002204 * the connection acknowledgement. If the proxy requires HTTP health-checks,
2205 * it sends the request. In other cases, it returns 1 if the socket is OK,
willy tarreau5cbea6f2005-12-17 12:48:26 +01002206 * or -1 if an error occured.
2207 */
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002208int event_srv_chk_w(int fd) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002209 struct task *t = fdtab[fd].owner;
2210 struct server *s = t->context;
willy tarreau0f7af912005-12-17 12:21:26 +01002211
willy tarreau5cbea6f2005-12-17 12:48:26 +01002212 int skerr, lskerr;
willy tarreauef900ab2005-12-17 12:52:52 +01002213 lskerr = sizeof(skerr);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002214 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
willy tarreau197e8ec2005-12-17 14:10:59 +01002215 /* in case of TCP only, this tells us if the connection succeeded */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002216 if (skerr)
2217 s->result = -1;
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002218 else {
2219 if (s->proxy->options & PR_O_HTTP_CHK) {
2220 int ret;
willy tarreau2f6ba652005-12-17 13:57:42 +01002221 /* we want to check if this host replies to "OPTIONS / HTTP/1.0"
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002222 * so we'll send the request, and won't wake the checker up now.
2223 */
2224#ifndef MSG_NOSIGNAL
willy tarreau2f6ba652005-12-17 13:57:42 +01002225 ret = send(fd, s->proxy->check_req, s->proxy->check_len, MSG_DONTWAIT);
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002226#else
willy tarreau2f6ba652005-12-17 13:57:42 +01002227 ret = send(fd, s->proxy->check_req, s->proxy->check_len, MSG_DONTWAIT | MSG_NOSIGNAL);
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002228#endif
2229 if (ret == 22) {
2230 FD_SET(fd, StaticReadEvent); /* prepare for reading reply */
2231 FD_CLR(fd, StaticWriteEvent); /* nothing more to write */
2232 return 0;
2233 }
2234 else
2235 s->result = -1;
2236 }
2237 else {
2238 /* good TCP connection is enough */
2239 s->result = 1;
2240 }
2241 }
2242
2243 task_wakeup(&rq, t);
2244 return 0;
2245}
2246
willy tarreau0f7af912005-12-17 12:21:26 +01002247
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002248/*
2249 * This function is used only for server health-checks. It handles
2250 * the server's reply to an HTTP request. It returns 1 if the server replies
2251 * 2xx or 3xx (valid responses), or -1 in other cases.
2252 */
2253int event_srv_chk_r(int fd) {
2254 char reply[64];
2255 int len;
2256 struct task *t = fdtab[fd].owner;
2257 struct server *s = t->context;
2258
2259 int skerr, lskerr;
2260 lskerr = sizeof(skerr);
willy tarreau197e8ec2005-12-17 14:10:59 +01002261
2262 s->result = len = -1;
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002263#ifndef MSG_NOSIGNAL
willy tarreau197e8ec2005-12-17 14:10:59 +01002264 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
2265 if (!skerr)
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002266 len = recv(fd, reply, sizeof(reply), 0);
2267#else
willy tarreau197e8ec2005-12-17 14:10:59 +01002268 /* Warning! Linux returns EAGAIN on SO_ERROR if data are still available
2269 * but the connection was closed on the remote end. Fortunately, recv still
2270 * works correctly and we don't need to do the getsockopt() on linux.
2271 */
2272 len = recv(fd, reply, sizeof(reply), MSG_NOSIGNAL);
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002273#endif
willy tarreau197e8ec2005-12-17 14:10:59 +01002274 if ((len >= sizeof("HTTP/1.0 000")) &&
2275 !memcmp(reply, "HTTP/1.", 7) &&
2276 (reply[9] == '2' || reply[9] == '3')) /* 2xx or 3xx */
2277 s->result = 1;
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002278
2279 FD_CLR(fd, StaticReadEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002280 task_wakeup(&rq, t);
willy tarreau0f7af912005-12-17 12:21:26 +01002281 return 0;
2282}
2283
2284
2285/*
2286 * this function writes the string <str> at position <pos> which must be in buffer <b>,
2287 * and moves <end> just after the end of <str>.
2288 * <b>'s parameters (l, r, w, h, lr) are recomputed to be valid after the shift.
2289 * the shift value (positive or negative) is returned.
2290 * If there's no space left, the move is not done.
2291 *
2292 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002293int buffer_replace(struct buffer *b, char *pos, char *end, char *str) {
willy tarreau0f7af912005-12-17 12:21:26 +01002294 int delta;
2295 int len;
2296
2297 len = strlen(str);
2298 delta = len - (end - pos);
2299
2300 if (delta + b->r >= b->data + BUFSIZE)
2301 return 0; /* no space left */
2302
2303 /* first, protect the end of the buffer */
2304 memmove(end + delta, end, b->data + b->l - end);
2305
2306 /* now, copy str over pos */
2307 memcpy(pos, str,len);
2308
willy tarreau5cbea6f2005-12-17 12:48:26 +01002309 /* we only move data after the displaced zone */
2310 if (b->r > pos) b->r += delta;
2311 if (b->w > pos) b->w += delta;
2312 if (b->h > pos) b->h += delta;
2313 if (b->lr > pos) b->lr += delta;
willy tarreau0f7af912005-12-17 12:21:26 +01002314 b->l += delta;
2315
2316 return delta;
2317}
2318
willy tarreau8337c6b2005-12-17 13:41:01 +01002319/* same except that the string length is given, which allows str to be NULL if
willy tarreau240afa62005-12-17 13:14:35 +01002320 * len is 0.
2321 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002322int buffer_replace2(struct buffer *b, char *pos, char *end, char *str, int len) {
willy tarreau0f7af912005-12-17 12:21:26 +01002323 int delta;
2324
2325 delta = len - (end - pos);
2326
2327 if (delta + b->r >= b->data + BUFSIZE)
2328 return 0; /* no space left */
2329
2330 /* first, protect the end of the buffer */
2331 memmove(end + delta, end, b->data + b->l - end);
2332
2333 /* now, copy str over pos */
willy tarreau240afa62005-12-17 13:14:35 +01002334 if (len)
2335 memcpy(pos, str, len);
willy tarreau0f7af912005-12-17 12:21:26 +01002336
willy tarreau5cbea6f2005-12-17 12:48:26 +01002337 /* we only move data after the displaced zone */
2338 if (b->r > pos) b->r += delta;
2339 if (b->w > pos) b->w += delta;
2340 if (b->h > pos) b->h += delta;
2341 if (b->lr > pos) b->lr += delta;
willy tarreau0f7af912005-12-17 12:21:26 +01002342 b->l += delta;
2343
2344 return delta;
2345}
2346
2347
2348int exp_replace(char *dst, char *src, char *str, regmatch_t *matches) {
2349 char *old_dst = dst;
2350
2351 while (*str) {
2352 if (*str == '\\') {
2353 str++;
willy tarreauc29948c2005-12-17 13:10:27 +01002354 if (isdigit((int)*str)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002355 int len, num;
2356
2357 num = *str - '0';
2358 str++;
2359
2360 if (matches[num].rm_so > -1) {
2361 len = matches[num].rm_eo - matches[num].rm_so;
2362 memcpy(dst, src + matches[num].rm_so, len);
2363 dst += len;
2364 }
2365
2366 }
2367 else if (*str == 'x') {
2368 unsigned char hex1, hex2;
2369 str++;
2370
2371 hex1=toupper(*str++) - '0'; hex2=toupper(*str++) - '0';
2372
2373 if (hex1 > 9) hex1 -= 'A' - '9' - 1;
2374 if (hex2 > 9) hex2 -= 'A' - '9' - 1;
2375 *dst++ = (hex1<<4) + hex2;
2376 }
2377 else
2378 *dst++ = *str++;
2379 }
2380 else
2381 *dst++ = *str++;
2382 }
2383 *dst = 0;
2384 return dst - old_dst;
2385}
2386
willy tarreau9fe663a2005-12-17 13:02:59 +01002387
willy tarreau0f7af912005-12-17 12:21:26 +01002388/*
2389 * manages the client FSM and its socket. BTW, it also tries to handle the
2390 * cookie. It returns 1 if a state has changed (and a resync may be needed),
2391 * 0 else.
2392 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002393int process_cli(struct session *t) {
willy tarreau0f7af912005-12-17 12:21:26 +01002394 int s = t->srv_state;
2395 int c = t->cli_state;
2396 struct buffer *req = t->req;
2397 struct buffer *rep = t->rep;
2398
willy tarreau750a4722005-12-17 13:21:24 +01002399#ifdef DEBUG_FULL
2400 fprintf(stderr,"process_cli: c=%s, s=%s\n", cli_stnames[c], srv_stnames[s]);
2401#endif
willy tarreau0f7af912005-12-17 12:21:26 +01002402 //fprintf(stderr,"process_cli: c=%d, s=%d, cr=%d, cw=%d, sr=%d, sw=%d\n", c, s,
2403 //FD_ISSET(t->cli_fd, StaticReadEvent), FD_ISSET(t->cli_fd, StaticWriteEvent),
2404 //FD_ISSET(t->srv_fd, StaticReadEvent), FD_ISSET(t->srv_fd, StaticWriteEvent)
2405 //);
2406 if (c == CL_STHEADERS) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002407 /* now parse the partial (or complete) headers */
2408 while (req->lr < req->r) { /* this loop only sees one header at each iteration */
2409 char *ptr;
2410 int delete_header;
willy tarreau0f7af912005-12-17 12:21:26 +01002411
willy tarreau5cbea6f2005-12-17 12:48:26 +01002412 ptr = req->lr;
willy tarreau0f7af912005-12-17 12:21:26 +01002413
willy tarreau0f7af912005-12-17 12:21:26 +01002414 /* look for the end of the current header */
2415 while (ptr < req->r && *ptr != '\n' && *ptr != '\r')
2416 ptr++;
2417
willy tarreau5cbea6f2005-12-17 12:48:26 +01002418 if (ptr == req->h) { /* empty line, end of headers */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002419 int line, len;
2420 /* we can only get here after an end of headers */
2421 /* we'll have something else to do here : add new headers ... */
willy tarreau0f7af912005-12-17 12:21:26 +01002422
willy tarreaue39cd132005-12-17 13:00:18 +01002423 if (t->flags & SN_CLDENY) {
2424 /* no need to go further */
willy tarreaua1598082005-12-17 13:08:06 +01002425 t->logs.status = 403;
willy tarreau8337c6b2005-12-17 13:41:01 +01002426 client_retnclose(t, t->proxy->errmsg.len403, t->proxy->errmsg.msg403);
willy tarreau036e1ce2005-12-17 13:46:33 +01002427 if (!(t->flags & SN_ERR_MASK))
2428 t->flags |= SN_ERR_PRXCOND;
2429 if (!(t->flags & SN_FINST_MASK))
2430 t->flags |= SN_FINST_R;
willy tarreaue39cd132005-12-17 13:00:18 +01002431 return 1;
2432 }
2433
willy tarreau5cbea6f2005-12-17 12:48:26 +01002434 for (line = 0; line < t->proxy->nb_reqadd; line++) {
willy tarreau750a4722005-12-17 13:21:24 +01002435 len = sprintf(trash, "%s\r\n", t->proxy->req_add[line]);
2436 buffer_replace2(req, req->h, req->h, trash, len);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002437 }
willy tarreau0f7af912005-12-17 12:21:26 +01002438
willy tarreau9fe663a2005-12-17 13:02:59 +01002439 if (t->proxy->options & PR_O_FWDFOR) {
2440 /* insert an X-Forwarded-For header */
2441 unsigned char *pn;
2442 pn = (unsigned char *)&t->cli_addr.sin_addr;
willy tarreau750a4722005-12-17 13:21:24 +01002443 len = sprintf(trash, "X-Forwarded-For: %d.%d.%d.%d\r\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01002444 pn[0], pn[1], pn[2], pn[3]);
willy tarreau750a4722005-12-17 13:21:24 +01002445 buffer_replace2(req, req->h, req->h, trash, len);
willy tarreau9fe663a2005-12-17 13:02:59 +01002446 }
2447
willy tarreaucd878942005-12-17 13:27:43 +01002448 if (!memcmp(req->data, "POST ", 5))
2449 t->flags |= SN_POST; /* this is a POST request */
2450
willy tarreau5cbea6f2005-12-17 12:48:26 +01002451 t->cli_state = CL_STDATA;
willy tarreauef900ab2005-12-17 12:52:52 +01002452 req->rlim = req->data + BUFSIZE; /* no more rewrite needed */
willy tarreau0f7af912005-12-17 12:21:26 +01002453
willy tarreau750a4722005-12-17 13:21:24 +01002454 t->logs.t_request = tv_diff(&t->logs.tv_accept, &now);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002455 /* FIXME: we'll set the client in a wait state while we try to
2456 * connect to the server. Is this really needed ? wouldn't it be
2457 * better to release the maximum of system buffers instead ? */
willy tarreauef900ab2005-12-17 12:52:52 +01002458 //FD_CLR(t->cli_fd, StaticReadEvent);
2459 //tv_eternity(&t->crexpire);
willy tarreau197e8ec2005-12-17 14:10:59 +01002460
2461 /* FIXME: if we break here (as up to 1.1.23), having the client
2462 * shutdown its connection can lead to an abort further.
2463 * it's better to either return 1 or even jump directly to the
2464 * data state which will save one schedule.
2465 */
2466 //break;
2467 goto process_data;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002468 }
willy tarreau0f7af912005-12-17 12:21:26 +01002469
willy tarreau5cbea6f2005-12-17 12:48:26 +01002470 /* to get a complete header line, we need the ending \r\n, \n\r, \r or \n too */
2471 if (ptr > req->r - 2) {
2472 /* this is a partial header, let's wait for more to come */
2473 req->lr = ptr;
2474 break;
2475 }
willy tarreau0f7af912005-12-17 12:21:26 +01002476
willy tarreau5cbea6f2005-12-17 12:48:26 +01002477 /* now we know that *ptr is either \r or \n,
2478 * and that there are at least 1 char after it.
2479 */
2480 if ((ptr[0] == ptr[1]) || (ptr[1] != '\r' && ptr[1] != '\n'))
2481 req->lr = ptr + 1; /* \r\r, \n\n, \r[^\n], \n[^\r] */
2482 else
2483 req->lr = ptr + 2; /* \r\n or \n\r */
willy tarreau0f7af912005-12-17 12:21:26 +01002484
willy tarreau5cbea6f2005-12-17 12:48:26 +01002485 /*
2486 * now we know that we have a full header ; we can do whatever
2487 * we want with these pointers :
2488 * req->h = beginning of header
2489 * ptr = end of header (first \r or \n)
2490 * req->lr = beginning of next line (next rep->h)
2491 * req->r = end of data (not used at this stage)
2492 */
willy tarreau0f7af912005-12-17 12:21:26 +01002493
willy tarreau8337c6b2005-12-17 13:41:01 +01002494 if (t->logs.logwait & LW_REQ) {
willy tarreau9fe663a2005-12-17 13:02:59 +01002495 /* we have a complete HTTP request that we must log */
2496 int urilen;
2497
willy tarreaua1598082005-12-17 13:08:06 +01002498 if ((t->logs.uri = pool_alloc(requri)) == NULL) {
willy tarreau9fe663a2005-12-17 13:02:59 +01002499 Alert("HTTP logging : out of memory.\n");
willy tarreau750a4722005-12-17 13:21:24 +01002500 t->logs.status = 500;
willy tarreau8337c6b2005-12-17 13:41:01 +01002501 client_retnclose(t, t->proxy->errmsg.len500, t->proxy->errmsg.msg500);
willy tarreau036e1ce2005-12-17 13:46:33 +01002502 if (!(t->flags & SN_ERR_MASK))
2503 t->flags |= SN_ERR_PRXCOND;
2504 if (!(t->flags & SN_FINST_MASK))
2505 t->flags |= SN_FINST_R;
willy tarreau9fe663a2005-12-17 13:02:59 +01002506 return 1;
2507 }
2508
2509 urilen = ptr - req->h;
2510 if (urilen >= REQURI_LEN)
2511 urilen = REQURI_LEN - 1;
willy tarreaua1598082005-12-17 13:08:06 +01002512 memcpy(t->logs.uri, req->h, urilen);
2513 t->logs.uri[urilen] = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01002514
willy tarreaua1598082005-12-17 13:08:06 +01002515 if (!(t->logs.logwait &= ~LW_REQ))
willy tarreau9fe663a2005-12-17 13:02:59 +01002516 sess_log(t);
2517 }
2518
willy tarreau5cbea6f2005-12-17 12:48:26 +01002519 delete_header = 0;
willy tarreau0f7af912005-12-17 12:21:26 +01002520
willy tarreau9fe663a2005-12-17 13:02:59 +01002521 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002522 int len, max;
willy tarreau2f6ba652005-12-17 13:57:42 +01002523 len = sprintf(trash, "%08x:%s.clihdr[%04x:%04x]: ", t->uniq_id, t->proxy->id, (unsigned short)t->cli_fd, (unsigned short)t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002524 max = ptr - req->h;
2525 UBOUND(max, sizeof(trash) - len - 1);
willy tarreau750a4722005-12-17 13:21:24 +01002526 len += strlcpy2(trash + len, req->h, max + 1);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002527 trash[len++] = '\n';
2528 write(1, trash, len);
2529 }
willy tarreau0f7af912005-12-17 12:21:26 +01002530
willy tarreau5cbea6f2005-12-17 12:48:26 +01002531 /* try headers regexps */
willy tarreaue39cd132005-12-17 13:00:18 +01002532 if (t->proxy->req_exp != NULL && !(t->flags & SN_CLDENY)) {
2533 struct hdr_exp *exp;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002534 char term;
2535
2536 term = *ptr;
2537 *ptr = '\0';
willy tarreaue39cd132005-12-17 13:00:18 +01002538 exp = t->proxy->req_exp;
2539 do {
2540 if (regexec(exp->preg, req->h, MAX_MATCH, pmatch, 0) == 0) {
2541 switch (exp->action) {
2542 case ACT_ALLOW:
2543 if (!(t->flags & SN_CLDENY))
2544 t->flags |= SN_CLALLOW;
2545 break;
2546 case ACT_REPLACE:
2547 if (!(t->flags & SN_CLDENY)) {
2548 int len = exp_replace(trash, req->h, exp->replace, pmatch);
2549 ptr += buffer_replace2(req, req->h, ptr, trash, len);
2550 }
2551 break;
2552 case ACT_REMOVE:
2553 if (!(t->flags & SN_CLDENY))
2554 delete_header = 1;
2555 break;
2556 case ACT_DENY:
2557 if (!(t->flags & SN_CLALLOW))
2558 t->flags |= SN_CLDENY;
2559 break;
willy tarreau036e1ce2005-12-17 13:46:33 +01002560 case ACT_PASS: /* we simply don't deny this one */
2561 break;
willy tarreau0f7af912005-12-17 12:21:26 +01002562 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002563 break;
willy tarreau0f7af912005-12-17 12:21:26 +01002564 }
willy tarreaue39cd132005-12-17 13:00:18 +01002565 } while ((exp = exp->next) != NULL);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002566 *ptr = term; /* restore the string terminator */
willy tarreau0f7af912005-12-17 12:21:26 +01002567 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002568
willy tarreau240afa62005-12-17 13:14:35 +01002569 /* Now look for cookies. Conforming to RFC2109, we have to support
2570 * attributes whose name begin with a '$', and associate them with
2571 * the right cookie, if we want to delete this cookie.
2572 * So there are 3 cases for each cookie read :
2573 * 1) it's a special attribute, beginning with a '$' : ignore it.
2574 * 2) it's a server id cookie that we *MAY* want to delete : save
2575 * some pointers on it (last semi-colon, beginning of cookie...)
2576 * 3) it's an application cookie : we *MAY* have to delete a previous
2577 * "special" cookie.
2578 * At the end of loop, if a "special" cookie remains, we may have to
2579 * remove it. If no application cookie persists in the header, we
2580 * *MUST* delete it
2581 */
willy tarreau8337c6b2005-12-17 13:41:01 +01002582 if (!delete_header && (t->proxy->cookie_name != NULL || t->proxy->capture_name != NULL)
willy tarreau240afa62005-12-17 13:14:35 +01002583 && !(t->flags & SN_CLDENY) && (ptr >= req->h + 8)
willy tarreau906b2682005-12-17 13:49:52 +01002584 && (strncasecmp(req->h, "Cookie: ", 8) == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002585 char *p1, *p2, *p3, *p4;
willy tarreau240afa62005-12-17 13:14:35 +01002586 char *del_colon, *del_cookie, *colon;
2587 int app_cookies;
2588
willy tarreau5cbea6f2005-12-17 12:48:26 +01002589 p1 = req->h + 8; /* first char after 'Cookie: ' */
willy tarreau240afa62005-12-17 13:14:35 +01002590 colon = p1;
2591 /* del_cookie == NULL => nothing to be deleted */
2592 del_colon = del_cookie = NULL;
2593 app_cookies = 0;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002594
2595 while (p1 < ptr) {
willy tarreau240afa62005-12-17 13:14:35 +01002596 /* skip spaces and colons, but keep an eye on these ones */
2597 while (p1 < ptr) {
2598 if (*p1 == ';' || *p1 == ',')
2599 colon = p1;
2600 else if (!isspace((int)*p1))
2601 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002602 p1++;
willy tarreau240afa62005-12-17 13:14:35 +01002603 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002604
2605 if (p1 == ptr)
2606 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002607
2608 /* p1 is at the beginning of the cookie name */
2609 p2 = p1;
willy tarreau240afa62005-12-17 13:14:35 +01002610 while (p2 < ptr && *p2 != '=')
willy tarreau5cbea6f2005-12-17 12:48:26 +01002611 p2++;
2612
2613 if (p2 == ptr)
2614 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002615
2616 p3 = p2 + 1; /* skips the '=' sign */
2617 if (p3 == ptr)
2618 break;
2619
willy tarreau240afa62005-12-17 13:14:35 +01002620 p4 = p3;
2621 while (p4 < ptr && !isspace((int)*p4) && *p4 != ';' && *p4 != ',')
willy tarreau5cbea6f2005-12-17 12:48:26 +01002622 p4++;
2623
2624 /* here, we have the cookie name between p1 and p2,
2625 * and its value between p3 and p4.
2626 * we can process it.
2627 */
2628
willy tarreau240afa62005-12-17 13:14:35 +01002629 if (*p1 == '$') {
2630 /* skip this one */
2631 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002632 else {
2633 /* first, let's see if we want to capture it */
2634 if (t->proxy->capture_name != NULL &&
2635 t->logs.cli_cookie == NULL &&
2636 (p4 - p1 >= t->proxy->capture_namelen) &&
2637 memcmp(p1, t->proxy->capture_name, t->proxy->capture_namelen) == 0) {
2638 int log_len = p4 - p1;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002639
willy tarreau8337c6b2005-12-17 13:41:01 +01002640 if ((t->logs.cli_cookie = pool_alloc(capture)) == NULL) {
2641 Alert("HTTP logging : out of memory.\n");
2642 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002643
willy tarreau8337c6b2005-12-17 13:41:01 +01002644 if (log_len > t->proxy->capture_len)
2645 log_len = t->proxy->capture_len;
2646 memcpy(t->logs.cli_cookie, p1, log_len);
2647 t->logs.cli_cookie[log_len] = 0;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002648 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002649
2650 if ((p2 - p1 == t->proxy->cookie_len) && (t->proxy->cookie_name != NULL) &&
2651 (memcmp(p1, t->proxy->cookie_name, p2 - p1) == 0)) {
2652 /* Cool... it's the right one */
2653 struct server *srv = t->proxy->srv;
2654
2655 while (srv &&
2656 ((srv->cklen != p4 - p3) || memcmp(p3, srv->cookie, p4 - p3))) {
2657 srv = srv->next;
2658 }
2659
willy tarreau036e1ce2005-12-17 13:46:33 +01002660 if (!srv) {
2661 t->flags &= ~SN_CK_MASK;
2662 t->flags |= SN_CK_INVALID;
2663 }
2664 else if (srv->state & SRV_RUNNING || t->proxy->options & PR_O_PERSIST) {
willy tarreau8337c6b2005-12-17 13:41:01 +01002665 /* we found the server and it's usable */
willy tarreau036e1ce2005-12-17 13:46:33 +01002666 t->flags &= ~SN_CK_MASK;
2667 t->flags |= SN_CK_VALID | SN_DIRECT;
willy tarreau8337c6b2005-12-17 13:41:01 +01002668 t->srv = srv;
2669 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002670 else {
2671 t->flags &= ~SN_CK_MASK;
2672 t->flags |= SN_CK_DOWN;
2673 }
2674
willy tarreau8337c6b2005-12-17 13:41:01 +01002675 /* if this cookie was set in insert+indirect mode, then it's better that the
2676 * server never sees it.
2677 */
2678 if (del_cookie == NULL &&
2679 (t->proxy->options & (PR_O_COOK_INS | PR_O_COOK_IND)) == (PR_O_COOK_INS | PR_O_COOK_IND)) {
willy tarreau240afa62005-12-17 13:14:35 +01002680 del_cookie = p1;
2681 del_colon = colon;
willy tarreau8337c6b2005-12-17 13:41:01 +01002682 }
willy tarreau240afa62005-12-17 13:14:35 +01002683 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002684 else {
2685 /* now we know that we must keep this cookie since it's
2686 * not ours. But if we wanted to delete our cookie
2687 * earlier, we cannot remove the complete header, but we
2688 * can remove the previous block itself.
2689 */
2690 app_cookies++;
2691
2692 if (del_cookie != NULL) {
2693 buffer_replace2(req, del_cookie, p1, NULL, 0);
2694 p4 -= (p1 - del_cookie);
2695 ptr -= (p1 - del_cookie);
2696 del_cookie = del_colon = NULL;
2697 }
willy tarreau240afa62005-12-17 13:14:35 +01002698 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002699 }
willy tarreau240afa62005-12-17 13:14:35 +01002700
willy tarreau5cbea6f2005-12-17 12:48:26 +01002701 /* we'll have to look for another cookie ... */
2702 p1 = p4;
2703 } /* while (p1 < ptr) */
willy tarreau240afa62005-12-17 13:14:35 +01002704
2705 /* There's no more cookie on this line.
2706 * We may have marked the last one(s) for deletion.
2707 * We must do this now in two ways :
2708 * - if there is no app cookie, we simply delete the header ;
2709 * - if there are app cookies, we must delete the end of the
2710 * string properly, including the colon/semi-colon before
2711 * the cookie name.
2712 */
2713 if (del_cookie != NULL) {
2714 if (app_cookies) {
2715 buffer_replace2(req, del_colon, ptr, NULL, 0);
2716 /* WARNING! <ptr> becomes invalid for now. If some code
2717 * below needs to rely on it before the end of the global
2718 * header loop, we need to correct it with this code :
2719 * ptr = del_colon;
2720 */
2721 }
2722 else
2723 delete_header = 1;
2724 }
2725 } /* end of cookie processing on this header */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002726
2727 /* let's look if we have to delete this header */
willy tarreaue39cd132005-12-17 13:00:18 +01002728 if (delete_header && !(t->flags & SN_CLDENY)) {
willy tarreau240afa62005-12-17 13:14:35 +01002729 buffer_replace2(req, req->h, req->lr, NULL, 0);
willy tarreau0f7af912005-12-17 12:21:26 +01002730 }
willy tarreau240afa62005-12-17 13:14:35 +01002731 /* WARNING: ptr is not valid anymore, since the header may have been deleted or truncated ! */
2732
willy tarreau5cbea6f2005-12-17 12:48:26 +01002733 req->h = req->lr;
2734 } /* while (req->lr < req->r) */
2735
2736 /* end of header processing (even if incomplete) */
2737
willy tarreauef900ab2005-12-17 12:52:52 +01002738 if ((req->l < req->rlim - req->data) && ! FD_ISSET(t->cli_fd, StaticReadEvent)) {
2739 /* fd in StaticReadEvent was disabled, perhaps because of a previous buffer
2740 * full. We cannot loop here since event_cli_read will disable it only if
2741 * req->l == rlim-data
2742 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002743 FD_SET(t->cli_fd, StaticReadEvent);
2744 if (t->proxy->clitimeout)
2745 tv_delayfrom(&t->crexpire, &now, t->proxy->clitimeout);
2746 else
2747 tv_eternity(&t->crexpire);
2748 }
2749
willy tarreaue39cd132005-12-17 13:00:18 +01002750 /* Since we are in header mode, if there's no space left for headers, we
willy tarreauef900ab2005-12-17 12:52:52 +01002751 * won't be able to free more later, so the session will never terminate.
2752 */
willy tarreaue39cd132005-12-17 13:00:18 +01002753 if (req->l >= req->rlim - req->data) {
willy tarreaua1598082005-12-17 13:08:06 +01002754 t->logs.status = 400;
willy tarreau8337c6b2005-12-17 13:41:01 +01002755 client_retnclose(t, t->proxy->errmsg.len400, t->proxy->errmsg.msg400);
willy tarreau036e1ce2005-12-17 13:46:33 +01002756 if (!(t->flags & SN_ERR_MASK))
2757 t->flags |= SN_ERR_PRXCOND;
2758 if (!(t->flags & SN_FINST_MASK))
2759 t->flags |= SN_FINST_R;
willy tarreaue39cd132005-12-17 13:00:18 +01002760 return 1;
2761 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002762 else if (t->res_cr == RES_ERROR || t->res_cr == RES_NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01002763 /* read error, or last read : give up. */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002764 tv_eternity(&t->crexpire);
2765 fd_delete(t->cli_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002766 t->cli_state = CL_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01002767 if (!(t->flags & SN_ERR_MASK))
2768 t->flags |= SN_ERR_CLICL;
2769 if (!(t->flags & SN_FINST_MASK))
2770 t->flags |= SN_FINST_R;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002771 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01002772 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002773 else if (tv_cmp2_ms(&t->crexpire, &now) <= 0) {
2774
2775 /* read timeout : give up with an error message.
2776 */
2777 t->logs.status = 408;
2778 client_retnclose(t, t->proxy->errmsg.len408, t->proxy->errmsg.msg408);
willy tarreau036e1ce2005-12-17 13:46:33 +01002779 if (!(t->flags & SN_ERR_MASK))
2780 t->flags |= SN_ERR_CLITO;
2781 if (!(t->flags & SN_FINST_MASK))
2782 t->flags |= SN_FINST_R;
willy tarreau8337c6b2005-12-17 13:41:01 +01002783 return 1;
2784 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002785
2786 return t->cli_state != CL_STHEADERS;
willy tarreau0f7af912005-12-17 12:21:26 +01002787 }
2788 else if (c == CL_STDATA) {
willy tarreau197e8ec2005-12-17 14:10:59 +01002789 process_data:
willy tarreauc1cae632005-12-17 14:12:23 +01002790 /* FIXME: this error handling is partly buggy because we always report
2791 * a 'DATA' phase while we don't know if the server was in IDLE, CONN
2792 * or HEADER phase. BTW, it's not logical to expire the client while
2793 * we're waiting for the server to connect.
2794 */
willy tarreau0f7af912005-12-17 12:21:26 +01002795 /* read or write error */
2796 if (t->res_cw == RES_ERROR || t->res_cr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01002797 tv_eternity(&t->crexpire);
2798 tv_eternity(&t->cwexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002799 fd_delete(t->cli_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002800 t->cli_state = CL_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01002801 if (!(t->flags & SN_ERR_MASK))
2802 t->flags |= SN_ERR_CLICL;
2803 if (!(t->flags & SN_FINST_MASK))
2804 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01002805 return 1;
2806 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002807 /* last read, or end of server write */
2808 else if (t->res_cr == RES_NULL || s == SV_STSHUTW || s == SV_STCLOSE) {
willy tarreau0f7af912005-12-17 12:21:26 +01002809 FD_CLR(t->cli_fd, StaticReadEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01002810 tv_eternity(&t->crexpire);
2811 shutdown(t->cli_fd, SHUT_RD);
2812 t->cli_state = CL_STSHUTR;
2813 return 1;
2814 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002815 /* last server read and buffer empty */
2816 else if ((s == SV_STSHUTR || s == SV_STCLOSE) && (rep->l == 0)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002817 FD_CLR(t->cli_fd, StaticWriteEvent);
2818 tv_eternity(&t->cwexpire);
2819 shutdown(t->cli_fd, SHUT_WR);
2820 t->cli_state = CL_STSHUTW;
2821 return 1;
2822 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002823 /* read timeout */
2824 else if (tv_cmp2_ms(&t->crexpire, &now) <= 0) {
2825 FD_CLR(t->cli_fd, StaticReadEvent);
willy tarreau036e1ce2005-12-17 13:46:33 +01002826 tv_eternity(&t->crexpire);
2827 shutdown(t->cli_fd, SHUT_RD);
2828 t->cli_state = CL_STSHUTR;
2829 if (!(t->flags & SN_ERR_MASK))
2830 t->flags |= SN_ERR_CLITO;
2831 if (!(t->flags & SN_FINST_MASK))
2832 t->flags |= SN_FINST_D;
2833 return 1;
2834 }
2835 /* write timeout */
2836 else if (tv_cmp2_ms(&t->cwexpire, &now) <= 0) {
2837 FD_CLR(t->cli_fd, StaticWriteEvent);
2838 tv_eternity(&t->cwexpire);
2839 shutdown(t->cli_fd, SHUT_WR);
2840 t->cli_state = CL_STSHUTW;
2841 if (!(t->flags & SN_ERR_MASK))
willy tarreaub1ff9db2005-12-17 13:51:03 +01002842 t->flags |= SN_ERR_CLITO;
willy tarreau036e1ce2005-12-17 13:46:33 +01002843 if (!(t->flags & SN_FINST_MASK))
2844 t->flags |= SN_FINST_D;
2845 return 1;
2846 }
willy tarreau0f7af912005-12-17 12:21:26 +01002847
willy tarreauc1cae632005-12-17 14:12:23 +01002848 if (req->l >= req->rlim - req->data || t->srv_state < SV_STDATA) {
2849 /* no room to read more data, or server not ready yet */
willy tarreau0f7af912005-12-17 12:21:26 +01002850 if (FD_ISSET(t->cli_fd, StaticReadEvent)) {
willy tarreauef900ab2005-12-17 12:52:52 +01002851 /* stop reading until we get some space */
willy tarreau0f7af912005-12-17 12:21:26 +01002852 FD_CLR(t->cli_fd, StaticReadEvent);
2853 tv_eternity(&t->crexpire);
2854 }
2855 }
2856 else {
willy tarreauef900ab2005-12-17 12:52:52 +01002857 /* there's still some space in the buffer */
willy tarreau0f7af912005-12-17 12:21:26 +01002858 if (! FD_ISSET(t->cli_fd, StaticReadEvent)) {
2859 FD_SET(t->cli_fd, StaticReadEvent);
2860 if (t->proxy->clitimeout)
2861 tv_delayfrom(&t->crexpire, &now, t->proxy->clitimeout);
2862 else
2863 tv_eternity(&t->crexpire);
2864 }
2865 }
2866
2867 if ((rep->l == 0) ||
willy tarreauc1cae632005-12-17 14:12:23 +01002868 ((s < SV_STDATA) /* FIXME: this may be optimized && (rep->w == rep->h)*/)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002869 if (FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2870 FD_CLR(t->cli_fd, StaticWriteEvent); /* stop writing */
2871 tv_eternity(&t->cwexpire);
2872 }
2873 }
2874 else { /* buffer not empty */
2875 if (! FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2876 FD_SET(t->cli_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01002877 if (t->proxy->clitimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01002878 tv_delayfrom(&t->cwexpire, &now, t->proxy->clitimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01002879 /* FIXME: to avoid the client to read-time-out during writes, we refresh it */
2880 t->crexpire = t->cwexpire;
2881 }
willy tarreau0f7af912005-12-17 12:21:26 +01002882 else
2883 tv_eternity(&t->cwexpire);
2884 }
2885 }
2886 return 0; /* other cases change nothing */
2887 }
2888 else if (c == CL_STSHUTR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01002889 if (t->res_cw == RES_ERROR) {
2890 tv_eternity(&t->cwexpire);
2891 fd_delete(t->cli_fd);
2892 t->cli_state = CL_STCLOSE;
2893 if (!(t->flags & SN_ERR_MASK))
2894 t->flags |= SN_ERR_CLICL;
2895 if (!(t->flags & SN_FINST_MASK))
2896 t->flags |= SN_FINST_D;
2897 return 1;
2898 }
2899 else if ((s == SV_STSHUTR || s == SV_STCLOSE) && (rep->l == 0)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002900 tv_eternity(&t->cwexpire);
2901 fd_delete(t->cli_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002902 t->cli_state = CL_STCLOSE;
2903 return 1;
2904 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002905 else if (tv_cmp2_ms(&t->cwexpire, &now) <= 0) {
2906 tv_eternity(&t->cwexpire);
2907 fd_delete(t->cli_fd);
2908 t->cli_state = CL_STCLOSE;
2909 if (!(t->flags & SN_ERR_MASK))
2910 t->flags |= SN_ERR_CLITO;
2911 if (!(t->flags & SN_FINST_MASK))
2912 t->flags |= SN_FINST_D;
2913 return 1;
2914 }
willy tarreau0f7af912005-12-17 12:21:26 +01002915 else if ((rep->l == 0) ||
willy tarreau5cbea6f2005-12-17 12:48:26 +01002916 ((s == SV_STHEADERS) /* FIXME: this may be optimized && (rep->w == rep->h)*/)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002917 if (FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2918 FD_CLR(t->cli_fd, StaticWriteEvent); /* stop writing */
2919 tv_eternity(&t->cwexpire);
2920 }
2921 }
2922 else { /* buffer not empty */
2923 if (! FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2924 FD_SET(t->cli_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01002925 if (t->proxy->clitimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01002926 tv_delayfrom(&t->cwexpire, &now, t->proxy->clitimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01002927 /* FIXME: to avoid the client to read-time-out during writes, we refresh it */
2928 t->crexpire = t->cwexpire;
2929 }
willy tarreau0f7af912005-12-17 12:21:26 +01002930 else
2931 tv_eternity(&t->cwexpire);
2932 }
2933 }
2934 return 0;
2935 }
2936 else if (c == CL_STSHUTW) {
willy tarreau036e1ce2005-12-17 13:46:33 +01002937 if (t->res_cr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01002938 tv_eternity(&t->crexpire);
2939 fd_delete(t->cli_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002940 t->cli_state = CL_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01002941 if (!(t->flags & SN_ERR_MASK))
2942 t->flags |= SN_ERR_CLICL;
2943 if (!(t->flags & SN_FINST_MASK))
2944 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01002945 return 1;
2946 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002947 else if (t->res_cr == RES_NULL || s == SV_STSHUTW || s == SV_STCLOSE) {
2948 tv_eternity(&t->crexpire);
2949 fd_delete(t->cli_fd);
2950 t->cli_state = CL_STCLOSE;
2951 return 1;
2952 }
2953 else if (tv_cmp2_ms(&t->crexpire, &now) <= 0) {
2954 tv_eternity(&t->crexpire);
2955 fd_delete(t->cli_fd);
2956 t->cli_state = CL_STCLOSE;
2957 if (!(t->flags & SN_ERR_MASK))
2958 t->flags |= SN_ERR_CLITO;
2959 if (!(t->flags & SN_FINST_MASK))
2960 t->flags |= SN_FINST_D;
2961 return 1;
2962 }
willy tarreauef900ab2005-12-17 12:52:52 +01002963 else if (req->l >= req->rlim - req->data) {
2964 /* no room to read more data */
willy tarreau0f7af912005-12-17 12:21:26 +01002965 if (FD_ISSET(t->cli_fd, StaticReadEvent)) {
willy tarreauef900ab2005-12-17 12:52:52 +01002966 /* stop reading until we get some space */
willy tarreau0f7af912005-12-17 12:21:26 +01002967 FD_CLR(t->cli_fd, StaticReadEvent);
2968 tv_eternity(&t->crexpire);
2969 }
2970 }
2971 else {
willy tarreauef900ab2005-12-17 12:52:52 +01002972 /* there's still some space in the buffer */
willy tarreau0f7af912005-12-17 12:21:26 +01002973 if (! FD_ISSET(t->cli_fd, StaticReadEvent)) {
2974 FD_SET(t->cli_fd, StaticReadEvent);
2975 if (t->proxy->clitimeout)
2976 tv_delayfrom(&t->crexpire, &now, t->proxy->clitimeout);
2977 else
2978 tv_eternity(&t->crexpire);
2979 }
2980 }
2981 return 0;
2982 }
2983 else { /* CL_STCLOSE: nothing to do */
willy tarreau9fe663a2005-12-17 13:02:59 +01002984 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002985 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01002986 len = sprintf(trash, "%08x:%s.clicls[%04x:%04x]\n", t->uniq_id, t->proxy->id, (unsigned short)t->cli_fd, (unsigned short)t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002987 write(1, trash, len);
2988 }
2989 return 0;
2990 }
2991 return 0;
2992}
2993
2994
2995/*
2996 * manages the server FSM and its socket. It returns 1 if a state has changed
2997 * (and a resync may be needed), 0 else.
2998 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002999int process_srv(struct session *t) {
willy tarreau0f7af912005-12-17 12:21:26 +01003000 int s = t->srv_state;
3001 int c = t->cli_state;
3002 struct buffer *req = t->req;
3003 struct buffer *rep = t->rep;
3004
willy tarreau750a4722005-12-17 13:21:24 +01003005#ifdef DEBUG_FULL
3006 fprintf(stderr,"process_srv: c=%s, s=%s\n", cli_stnames[c], srv_stnames[s]);
3007#endif
willy tarreau5cbea6f2005-12-17 12:48:26 +01003008 //fprintf(stderr,"process_srv: c=%d, s=%d, cr=%d, cw=%d, sr=%d, sw=%d\n", c, s,
3009 //FD_ISSET(t->cli_fd, StaticReadEvent), FD_ISSET(t->cli_fd, StaticWriteEvent),
3010 //FD_ISSET(t->srv_fd, StaticReadEvent), FD_ISSET(t->srv_fd, StaticWriteEvent)
3011 //);
willy tarreau0f7af912005-12-17 12:21:26 +01003012 if (s == SV_STIDLE) {
3013 if (c == CL_STHEADERS)
3014 return 0; /* stay in idle, waiting for data to reach the client side */
3015 else if (c == CL_STCLOSE ||
3016 c == CL_STSHUTW ||
3017 (c == CL_STSHUTR && t->req->l == 0)) { /* give up */
3018 tv_eternity(&t->cnexpire);
3019 t->srv_state = SV_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01003020 if (!(t->flags & SN_ERR_MASK))
3021 t->flags |= SN_ERR_CLICL;
3022 if (!(t->flags & SN_FINST_MASK))
3023 t->flags |= SN_FINST_C;
willy tarreau0f7af912005-12-17 12:21:26 +01003024 return 1;
3025 }
3026 else { /* go to SV_STCONN */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003027 if (connect_server(t) == 0) { /* initiate a connection to the server */
willy tarreau0f7af912005-12-17 12:21:26 +01003028 //fprintf(stderr,"0: c=%d, s=%d\n", c, s);
3029 t->srv_state = SV_STCONN;
3030 }
3031 else { /* try again */
3032 while (t->conn_retries-- > 0) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003033 if ((t->proxy->options & PR_O_REDISP) && (t->conn_retries == 0)) {
willy tarreaue39cd132005-12-17 13:00:18 +01003034 t->flags &= ~SN_DIRECT; /* ignore cookie and force to use the dispatcher */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003035 t->srv = NULL; /* it's left to the dispatcher to choose a server */
willy tarreau036e1ce2005-12-17 13:46:33 +01003036 if ((t->flags & SN_CK_MASK) == SN_CK_VALID) {
3037 t->flags &= ~SN_CK_MASK;
3038 t->flags |= SN_CK_DOWN;
3039 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01003040 }
3041
3042 if (connect_server(t) == 0) {
willy tarreau0f7af912005-12-17 12:21:26 +01003043 t->srv_state = SV_STCONN;
3044 break;
3045 }
3046 }
3047 if (t->conn_retries < 0) {
3048 /* if conn_retries < 0 or other error, let's abort */
3049 tv_eternity(&t->cnexpire);
3050 t->srv_state = SV_STCLOSE;
willy tarreau8337c6b2005-12-17 13:41:01 +01003051 t->logs.status = 503;
willy tarreau750a4722005-12-17 13:21:24 +01003052 if (t->proxy->mode == PR_MODE_HTTP)
willy tarreau8337c6b2005-12-17 13:41:01 +01003053 client_return(t, t->proxy->errmsg.len503, t->proxy->errmsg.msg503);
willy tarreau036e1ce2005-12-17 13:46:33 +01003054 if (!(t->flags & SN_ERR_MASK))
3055 t->flags |= SN_ERR_SRVCL;
3056 if (!(t->flags & SN_FINST_MASK))
3057 t->flags |= SN_FINST_C;
willy tarreau0f7af912005-12-17 12:21:26 +01003058 }
3059 }
3060 return 1;
3061 }
3062 }
3063 else if (s == SV_STCONN) { /* connection in progress */
3064 if (t->res_sw == RES_SILENT && tv_cmp2_ms(&t->cnexpire, &now) > 0) {
3065 //fprintf(stderr,"1: c=%d, s=%d\n", c, s);
3066 return 0; /* nothing changed */
3067 }
3068 else if (t->res_sw == RES_SILENT || t->res_sw == RES_ERROR) {
3069 //fprintf(stderr,"2: c=%d, s=%d\n", c, s);
3070 /* timeout, connect error or first write error */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003071 //FD_CLR(t->srv_fd, StaticWriteEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01003072 fd_delete(t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003073 //close(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003074 t->conn_retries--;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003075 if (t->conn_retries >= 0) {
3076 if ((t->proxy->options & PR_O_REDISP) && (t->conn_retries == 0)) {
willy tarreaue39cd132005-12-17 13:00:18 +01003077 t->flags &= ~SN_DIRECT; /* ignore cookie and force to use the dispatcher */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003078 t->srv = NULL; /* it's left to the dispatcher to choose a server */
willy tarreau036e1ce2005-12-17 13:46:33 +01003079 if ((t->flags & SN_CK_MASK) == SN_CK_VALID) {
3080 t->flags &= ~SN_CK_MASK;
3081 t->flags |= SN_CK_DOWN;
3082 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01003083 }
3084 if (connect_server(t) == 0)
3085 return 0; /* no state changed */
willy tarreau0f7af912005-12-17 12:21:26 +01003086 }
3087 /* if conn_retries < 0 or other error, let's abort */
3088 tv_eternity(&t->cnexpire);
3089 t->srv_state = SV_STCLOSE;
willy tarreau8337c6b2005-12-17 13:41:01 +01003090 t->logs.status = 503;
willy tarreau750a4722005-12-17 13:21:24 +01003091 if (t->proxy->mode == PR_MODE_HTTP)
willy tarreau8337c6b2005-12-17 13:41:01 +01003092 client_return(t, t->proxy->errmsg.len503, t->proxy->errmsg.msg503);
willy tarreau036e1ce2005-12-17 13:46:33 +01003093 if (!(t->flags & SN_ERR_MASK))
3094 t->flags |= SN_ERR_SRVCL;
3095 if (!(t->flags & SN_FINST_MASK))
3096 t->flags |= SN_FINST_C;
willy tarreau0f7af912005-12-17 12:21:26 +01003097 return 1;
3098 }
3099 else { /* no error or write 0 */
willy tarreau750a4722005-12-17 13:21:24 +01003100 t->logs.t_connect = tv_diff(&t->logs.tv_accept, &now);
willy tarreaua1598082005-12-17 13:08:06 +01003101
willy tarreau0f7af912005-12-17 12:21:26 +01003102 //fprintf(stderr,"3: c=%d, s=%d\n", c, s);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003103 if (req->l == 0) /* nothing to write */ {
willy tarreau0f7af912005-12-17 12:21:26 +01003104 FD_CLR(t->srv_fd, StaticWriteEvent);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003105 tv_eternity(&t->swexpire);
3106 } else /* need the right to write */ {
willy tarreau0f7af912005-12-17 12:21:26 +01003107 FD_SET(t->srv_fd, StaticWriteEvent);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003108 if (t->proxy->srvtimeout) {
3109 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
3110 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3111 t->srexpire = t->swexpire;
3112 }
3113 else
3114 tv_eternity(&t->swexpire);
3115 }
willy tarreau0f7af912005-12-17 12:21:26 +01003116
3117 if (t->proxy->mode == PR_MODE_TCP) { /* let's allow immediate data connection in this case */
3118 FD_SET(t->srv_fd, StaticReadEvent);
3119 if (t->proxy->srvtimeout)
3120 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3121 else
3122 tv_eternity(&t->srexpire);
3123
3124 t->srv_state = SV_STDATA;
willy tarreauef900ab2005-12-17 12:52:52 +01003125 rep->rlim = rep->data + BUFSIZE; /* no rewrite needed */
willy tarreau0f7af912005-12-17 12:21:26 +01003126 }
willy tarreauef900ab2005-12-17 12:52:52 +01003127 else {
willy tarreau0f7af912005-12-17 12:21:26 +01003128 t->srv_state = SV_STHEADERS;
willy tarreauef900ab2005-12-17 12:52:52 +01003129 rep->rlim = rep->data + BUFSIZE - MAXREWRITE; /* rewrite needed */
3130 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01003131 tv_eternity(&t->cnexpire);
willy tarreau0f7af912005-12-17 12:21:26 +01003132 return 1;
3133 }
3134 }
3135 else if (s == SV_STHEADERS) { /* receiving server headers */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003136 /* now parse the partial (or complete) headers */
3137 while (rep->lr < rep->r) { /* this loop only sees one header at each iteration */
3138 char *ptr;
3139 int delete_header;
3140
3141 ptr = rep->lr;
3142
3143 /* look for the end of the current header */
3144 while (ptr < rep->r && *ptr != '\n' && *ptr != '\r')
3145 ptr++;
3146
3147 if (ptr == rep->h) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003148 int line, len;
3149
3150 /* we can only get here after an end of headers */
3151 /* we'll have something else to do here : add new headers ... */
3152
willy tarreaucd878942005-12-17 13:27:43 +01003153 if ((t->srv) && !(t->flags & SN_DIRECT) && (t->proxy->options & PR_O_COOK_INS) &&
3154 (!(t->proxy->options & PR_O_COOK_POST) || (t->flags & SN_POST))) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003155 /* the server is known, it's not the one the client requested, we have to
willy tarreaucd878942005-12-17 13:27:43 +01003156 * insert a set-cookie here, except if we want to insert only on POST
3157 * requests and this one isn't.
willy tarreau5cbea6f2005-12-17 12:48:26 +01003158 */
willy tarreau750a4722005-12-17 13:21:24 +01003159 len = sprintf(trash, "Set-Cookie: %s=%s; path=/\r\n",
willy tarreau8337c6b2005-12-17 13:41:01 +01003160 t->proxy->cookie_name,
3161 t->srv->cookie ? t->srv->cookie : "");
willy tarreau750a4722005-12-17 13:21:24 +01003162
willy tarreau036e1ce2005-12-17 13:46:33 +01003163 t->flags |= SN_SCK_INSERTED;
3164
willy tarreau750a4722005-12-17 13:21:24 +01003165 /* Here, we will tell an eventual cache on the client side that we don't
3166 * want it to cache this reply because HTTP/1.0 caches also cache cookies !
3167 * Some caches understand the correct form: 'no-cache="set-cookie"', but
3168 * others don't (eg: apache <= 1.3.26). So we use 'private' instead.
3169 */
willy tarreau240afa62005-12-17 13:14:35 +01003170 if (t->proxy->options & PR_O_COOK_NOC)
willy tarreau750a4722005-12-17 13:21:24 +01003171 //len += sprintf(newhdr + len, "Cache-control: no-cache=\"set-cookie\"\r\n");
3172 len += sprintf(trash + len, "Cache-control: private\r\n");
willy tarreaucd878942005-12-17 13:27:43 +01003173
willy tarreau750a4722005-12-17 13:21:24 +01003174 buffer_replace2(rep, rep->h, rep->h, trash, len);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003175 }
3176
3177 /* headers to be added */
3178 for (line = 0; line < t->proxy->nb_rspadd; line++) {
willy tarreau750a4722005-12-17 13:21:24 +01003179 len = sprintf(trash, "%s\r\n", t->proxy->rsp_add[line]);
3180 buffer_replace2(rep, rep->h, rep->h, trash, len);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003181 }
3182
3183 t->srv_state = SV_STDATA;
willy tarreauef900ab2005-12-17 12:52:52 +01003184 rep->rlim = rep->data + BUFSIZE; /* no more rewrite needed */
willy tarreau750a4722005-12-17 13:21:24 +01003185 t->logs.t_data = tv_diff(&t->logs.tv_accept, &now);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003186 break;
3187 }
3188
3189 /* to get a complete header line, we need the ending \r\n, \n\r, \r or \n too */
3190 if (ptr > rep->r - 2) {
3191 /* this is a partial header, let's wait for more to come */
3192 rep->lr = ptr;
3193 break;
3194 }
3195
3196 // fprintf(stderr,"h=%p, ptr=%p, lr=%p, r=%p, *h=", rep->h, ptr, rep->lr, rep->r);
3197 // write(2, rep->h, ptr - rep->h); fprintf(stderr,"\n");
3198
3199 /* now we know that *ptr is either \r or \n,
3200 * and that there are at least 1 char after it.
3201 */
3202 if ((ptr[0] == ptr[1]) || (ptr[1] != '\r' && ptr[1] != '\n'))
3203 rep->lr = ptr + 1; /* \r\r, \n\n, \r[^\n], \n[^\r] */
3204 else
3205 rep->lr = ptr + 2; /* \r\n or \n\r */
3206
3207 /*
3208 * now we know that we have a full header ; we can do whatever
3209 * we want with these pointers :
3210 * rep->h = beginning of header
3211 * ptr = end of header (first \r or \n)
3212 * rep->lr = beginning of next line (next rep->h)
3213 * rep->r = end of data (not used at this stage)
3214 */
3215
willy tarreaua1598082005-12-17 13:08:06 +01003216
3217 if (t->logs.logwait & LW_RESP) {
3218 t->logs.logwait &= ~LW_RESP;
3219 t->logs.status = atoi(rep->h + 9);
3220 }
3221
willy tarreau5cbea6f2005-12-17 12:48:26 +01003222 delete_header = 0;
3223
willy tarreau9fe663a2005-12-17 13:02:59 +01003224 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003225 int len, max;
willy tarreau2f6ba652005-12-17 13:57:42 +01003226 len = sprintf(trash, "%08x:%s.srvhdr[%04x:%04x]: ", t->uniq_id, t->proxy->id, (unsigned short)t->cli_fd, (unsigned short)t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003227 max = ptr - rep->h;
3228 UBOUND(max, sizeof(trash) - len - 1);
willy tarreau750a4722005-12-17 13:21:24 +01003229 len += strlcpy2(trash + len, rep->h, max + 1);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003230 trash[len++] = '\n';
3231 write(1, trash, len);
3232 }
3233
3234 /* try headers regexps */
willy tarreaue39cd132005-12-17 13:00:18 +01003235 if (t->proxy->rsp_exp != NULL && !(t->flags & SN_SVDENY)) {
3236 struct hdr_exp *exp;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003237 char term;
3238
3239 term = *ptr;
3240 *ptr = '\0';
willy tarreaue39cd132005-12-17 13:00:18 +01003241 exp = t->proxy->rsp_exp;
3242 do {
3243 if (regexec(exp->preg, rep->h, MAX_MATCH, pmatch, 0) == 0) {
3244 switch (exp->action) {
3245 case ACT_ALLOW:
3246 if (!(t->flags & SN_SVDENY))
3247 t->flags |= SN_SVALLOW;
3248 break;
3249 case ACT_REPLACE:
3250 if (!(t->flags & SN_SVDENY)) {
3251 int len = exp_replace(trash, rep->h, exp->replace, pmatch);
3252 ptr += buffer_replace2(rep, rep->h, ptr, trash, len);
3253 }
3254 break;
3255 case ACT_REMOVE:
3256 if (!(t->flags & SN_SVDENY))
3257 delete_header = 1;
3258 break;
3259 case ACT_DENY:
3260 if (!(t->flags & SN_SVALLOW))
3261 t->flags |= SN_SVDENY;
3262 break;
willy tarreau036e1ce2005-12-17 13:46:33 +01003263 case ACT_PASS: /* we simply don't deny this one */
3264 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003265 }
3266 break;
3267 }
willy tarreaue39cd132005-12-17 13:00:18 +01003268 } while ((exp = exp->next) != NULL);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003269 *ptr = term; /* restore the string terminator */
3270 }
3271
3272 /* check for server cookies */
willy tarreau8337c6b2005-12-17 13:41:01 +01003273 if (!delete_header /*&& (t->proxy->options & PR_O_COOK_ANY)*/
3274 && (t->proxy->cookie_name != NULL || t->proxy->capture_name != NULL)
3275 && (ptr >= rep->h + 12)
willy tarreau906b2682005-12-17 13:49:52 +01003276 && (strncasecmp(rep->h, "Set-Cookie: ", 12) == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003277 char *p1, *p2, *p3, *p4;
3278
3279 p1 = rep->h + 12; /* first char after 'Set-Cookie: ' */
3280
3281 while (p1 < ptr) { /* in fact, we'll break after the first cookie */
willy tarreauc29948c2005-12-17 13:10:27 +01003282 while (p1 < ptr && (isspace((int)*p1)))
willy tarreau5cbea6f2005-12-17 12:48:26 +01003283 p1++;
3284
3285 if (p1 == ptr || *p1 == ';') /* end of cookie */
3286 break;
3287
3288 /* p1 is at the beginning of the cookie name */
3289 p2 = p1;
3290
3291 while (p2 < ptr && *p2 != '=' && *p2 != ';')
3292 p2++;
3293
3294 if (p2 == ptr || *p2 == ';') /* next cookie */
3295 break;
3296
3297 p3 = p2 + 1; /* skips the '=' sign */
3298 if (p3 == ptr)
3299 break;
3300
3301 p4 = p3;
willy tarreauc29948c2005-12-17 13:10:27 +01003302 while (p4 < ptr && !isspace((int)*p4) && *p4 != ';')
willy tarreau5cbea6f2005-12-17 12:48:26 +01003303 p4++;
3304
3305 /* here, we have the cookie name between p1 and p2,
3306 * and its value between p3 and p4.
3307 * we can process it.
3308 */
willy tarreau8337c6b2005-12-17 13:41:01 +01003309
3310 /* first, let's see if we want to capture it */
3311 if (t->proxy->capture_name != NULL &&
3312 t->logs.srv_cookie == NULL &&
3313 (p4 - p1 >= t->proxy->capture_namelen) &&
3314 memcmp(p1, t->proxy->capture_name, t->proxy->capture_namelen) == 0) {
3315 int log_len = p4 - p1;
3316
3317 if ((t->logs.srv_cookie = pool_alloc(capture)) == NULL) {
3318 Alert("HTTP logging : out of memory.\n");
3319 }
3320
3321 if (log_len > t->proxy->capture_len)
3322 log_len = t->proxy->capture_len;
3323 memcpy(t->logs.srv_cookie, p1, log_len);
3324 t->logs.srv_cookie[log_len] = 0;
3325 }
3326
3327 if ((p2 - p1 == t->proxy->cookie_len) && (t->proxy->cookie_name != NULL) &&
3328 (memcmp(p1, t->proxy->cookie_name, p2 - p1) == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003329 /* Cool... it's the right one */
willy tarreau036e1ce2005-12-17 13:46:33 +01003330 t->flags |= SN_SCK_SEEN;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003331
3332 /* If the cookie is in insert mode on a known server, we'll delete
3333 * this occurrence because we'll insert another one later.
3334 * We'll delete it too if the "indirect" option is set and we're in
3335 * a direct access. */
3336 if (((t->srv) && (t->proxy->options & PR_O_COOK_INS)) ||
willy tarreaue39cd132005-12-17 13:00:18 +01003337 ((t->flags & SN_DIRECT) && (t->proxy->options & PR_O_COOK_IND))) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003338 /* this header must be deleted */
3339 delete_header = 1;
willy tarreau036e1ce2005-12-17 13:46:33 +01003340 t->flags |= SN_SCK_DELETED;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003341 }
3342 else if ((t->srv) && (t->proxy->options & PR_O_COOK_RW)) {
3343 /* replace bytes p3->p4 with the cookie name associated
3344 * with this server since we know it.
3345 */
3346 buffer_replace2(rep, p3, p4, t->srv->cookie, t->srv->cklen);
willy tarreau036e1ce2005-12-17 13:46:33 +01003347 t->flags |= SN_SCK_INSERTED | SN_SCK_DELETED;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003348 }
3349 break;
3350 }
3351 else {
3352 // fprintf(stderr,"Ignoring unknown cookie : ");
3353 // write(2, p1, p2-p1);
3354 // fprintf(stderr," = ");
3355 // write(2, p3, p4-p3);
3356 // fprintf(stderr,"\n");
3357 }
3358 break; /* we don't want to loop again since there cannot be another cookie on the same line */
3359 } /* we're now at the end of the cookie value */
3360 } /* end of cookie processing */
3361
3362 /* let's look if we have to delete this header */
willy tarreaue39cd132005-12-17 13:00:18 +01003363 if (delete_header && !(t->flags & SN_SVDENY))
willy tarreau5cbea6f2005-12-17 12:48:26 +01003364 buffer_replace2(rep, rep->h, rep->lr, "", 0);
willy tarreaue39cd132005-12-17 13:00:18 +01003365
willy tarreau5cbea6f2005-12-17 12:48:26 +01003366 rep->h = rep->lr;
3367 } /* while (rep->lr < rep->r) */
3368
3369 /* end of header processing (even if incomplete) */
3370
willy tarreauef900ab2005-12-17 12:52:52 +01003371 if ((rep->l < rep->rlim - rep->data) && ! FD_ISSET(t->srv_fd, StaticReadEvent)) {
3372 /* fd in StaticReadEvent was disabled, perhaps because of a previous buffer
3373 * full. We cannot loop here since event_srv_read will disable it only if
3374 * rep->l == rlim-data
3375 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003376 FD_SET(t->srv_fd, StaticReadEvent);
3377 if (t->proxy->srvtimeout)
3378 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3379 else
3380 tv_eternity(&t->srexpire);
3381 }
willy tarreau0f7af912005-12-17 12:21:26 +01003382
willy tarreau8337c6b2005-12-17 13:41:01 +01003383 /* read error, write error */
willy tarreau0f7af912005-12-17 12:21:26 +01003384 if (t->res_sw == RES_ERROR || t->res_sr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01003385 tv_eternity(&t->srexpire);
3386 tv_eternity(&t->swexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003387 fd_delete(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003388 t->srv_state = SV_STCLOSE;
willy tarreaucd878942005-12-17 13:27:43 +01003389 t->logs.status = 502;
willy tarreau8337c6b2005-12-17 13:41:01 +01003390 client_return(t, t->proxy->errmsg.len502, t->proxy->errmsg.msg502);
willy tarreau036e1ce2005-12-17 13:46:33 +01003391 if (!(t->flags & SN_ERR_MASK))
3392 t->flags |= SN_ERR_SRVCL;
3393 if (!(t->flags & SN_FINST_MASK))
3394 t->flags |= SN_FINST_H;
willy tarreau0f7af912005-12-17 12:21:26 +01003395 return 1;
3396 }
willy tarreau8337c6b2005-12-17 13:41:01 +01003397 /* end of client write or end of server read.
willy tarreauef900ab2005-12-17 12:52:52 +01003398 * since we are in header mode, if there's no space left for headers, we
3399 * won't be able to free more later, so the session will never terminate.
3400 */
willy tarreau8337c6b2005-12-17 13:41:01 +01003401 else if (t->res_sr == RES_NULL || c == CL_STSHUTW || c == CL_STCLOSE || rep->l >= rep->rlim - rep->data) {
willy tarreau0f7af912005-12-17 12:21:26 +01003402 FD_CLR(t->srv_fd, StaticReadEvent);
3403 tv_eternity(&t->srexpire);
3404 shutdown(t->srv_fd, SHUT_RD);
3405 t->srv_state = SV_STSHUTR;
3406 return 1;
willy tarreau8337c6b2005-12-17 13:41:01 +01003407 }
3408 /* read timeout : return a 504 to the client.
3409 */
3410 else if (FD_ISSET(t->srv_fd, StaticReadEvent) && tv_cmp2_ms(&t->srexpire, &now) <= 0) {
3411 tv_eternity(&t->srexpire);
3412 tv_eternity(&t->swexpire);
3413 fd_delete(t->srv_fd);
3414 t->srv_state = SV_STCLOSE;
3415 t->logs.status = 504;
3416 client_return(t, t->proxy->errmsg.len504, t->proxy->errmsg.msg504);
willy tarreau036e1ce2005-12-17 13:46:33 +01003417 if (!(t->flags & SN_ERR_MASK))
3418 t->flags |= SN_ERR_SRVTO;
3419 if (!(t->flags & SN_FINST_MASK))
3420 t->flags |= SN_FINST_H;
willy tarreau8337c6b2005-12-17 13:41:01 +01003421 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01003422
3423 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003424 /* last client read and buffer empty */
willy tarreau750a4722005-12-17 13:21:24 +01003425 /* FIXME!!! here, we don't want to switch to SHUTW if the
3426 * client shuts read too early, because we may still have
3427 * some work to do on the headers.
willy tarreau036e1ce2005-12-17 13:46:33 +01003428 * The side-effect is that if the client completely closes its
3429 * connection during SV_STHEADER, the connection to the server
3430 * is kept until a response comes back or the timeout is reached.
willy tarreau750a4722005-12-17 13:21:24 +01003431 */
willy tarreau036e1ce2005-12-17 13:46:33 +01003432 else if ((/*c == CL_STSHUTR ||*/ c == CL_STCLOSE) && (req->l == 0)) {
willy tarreau0f7af912005-12-17 12:21:26 +01003433 FD_CLR(t->srv_fd, StaticWriteEvent);
3434 tv_eternity(&t->swexpire);
3435 shutdown(t->srv_fd, SHUT_WR);
3436 t->srv_state = SV_STSHUTW;
3437 return 1;
3438 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003439 /* write timeout */
3440 /* FIXME!!! here, we don't want to switch to SHUTW if the
3441 * client shuts read too early, because we may still have
3442 * some work to do on the headers.
3443 */
3444 else if (FD_ISSET(t->srv_fd, StaticWriteEvent) && tv_cmp2_ms(&t->swexpire, &now) <= 0) {
3445 FD_CLR(t->srv_fd, StaticWriteEvent);
3446 tv_eternity(&t->swexpire);
3447 shutdown(t->srv_fd, SHUT_WR);
3448 t->srv_state = SV_STSHUTW;
3449 if (!(t->flags & SN_ERR_MASK))
3450 t->flags |= SN_ERR_SRVTO;
3451 if (!(t->flags & SN_FINST_MASK))
3452 t->flags |= SN_FINST_H;
3453 return 1;
3454 }
willy tarreau0f7af912005-12-17 12:21:26 +01003455
3456 if (req->l == 0) {
3457 if (FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3458 FD_CLR(t->srv_fd, StaticWriteEvent); /* stop writing */
3459 tv_eternity(&t->swexpire);
3460 }
3461 }
3462 else { /* client buffer not empty */
3463 if (! FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3464 FD_SET(t->srv_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01003465 if (t->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01003466 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003467 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3468 t->srexpire = t->swexpire;
3469 }
willy tarreau0f7af912005-12-17 12:21:26 +01003470 else
3471 tv_eternity(&t->swexpire);
3472 }
3473 }
3474
willy tarreau5cbea6f2005-12-17 12:48:26 +01003475 /* be nice with the client side which would like to send a complete header
3476 * FIXME: COMPLETELY BUGGY !!! not all headers may be processed because the client
3477 * would read all remaining data at once ! The client should not write past rep->lr
3478 * when the server is in header state.
3479 */
3480 //return header_processed;
3481 return t->srv_state != SV_STHEADERS;
willy tarreau0f7af912005-12-17 12:21:26 +01003482 }
3483 else if (s == SV_STDATA) {
3484 /* read or write error */
3485 if (t->res_sw == RES_ERROR || t->res_sr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01003486 tv_eternity(&t->srexpire);
3487 tv_eternity(&t->swexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003488 fd_delete(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003489 t->srv_state = SV_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01003490 if (!(t->flags & SN_ERR_MASK))
3491 t->flags |= SN_ERR_SRVCL;
3492 if (!(t->flags & SN_FINST_MASK))
3493 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01003494 return 1;
3495 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003496 /* last read, or end of client write */
3497 else if (t->res_sr == RES_NULL || c == CL_STSHUTW || c == CL_STCLOSE) {
willy tarreau0f7af912005-12-17 12:21:26 +01003498 FD_CLR(t->srv_fd, StaticReadEvent);
3499 tv_eternity(&t->srexpire);
3500 shutdown(t->srv_fd, SHUT_RD);
3501 t->srv_state = SV_STSHUTR;
3502 return 1;
willy tarreaua41a8b42005-12-17 14:02:24 +01003503 }
3504 /* end of client read and no more data to send */
3505 else if ((c == CL_STSHUTR || c == CL_STCLOSE) && (req->l == 0)) {
3506 FD_CLR(t->srv_fd, StaticWriteEvent);
3507 tv_eternity(&t->swexpire);
3508 shutdown(t->srv_fd, SHUT_WR);
3509 t->srv_state = SV_STSHUTW;
3510 return 1;
3511 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003512 /* read timeout */
3513 else if (tv_cmp2_ms(&t->srexpire, &now) <= 0) {
3514 FD_CLR(t->srv_fd, StaticReadEvent);
3515 tv_eternity(&t->srexpire);
3516 shutdown(t->srv_fd, SHUT_RD);
3517 t->srv_state = SV_STSHUTR;
3518 if (!(t->flags & SN_ERR_MASK))
3519 t->flags |= SN_ERR_SRVTO;
3520 if (!(t->flags & SN_FINST_MASK))
3521 t->flags |= SN_FINST_D;
3522 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01003523 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003524 /* write timeout */
3525 else if (tv_cmp2_ms(&t->swexpire, &now) <= 0) {
willy tarreau0f7af912005-12-17 12:21:26 +01003526 FD_CLR(t->srv_fd, StaticWriteEvent);
3527 tv_eternity(&t->swexpire);
3528 shutdown(t->srv_fd, SHUT_WR);
3529 t->srv_state = SV_STSHUTW;
willy tarreau036e1ce2005-12-17 13:46:33 +01003530 if (!(t->flags & SN_ERR_MASK))
3531 t->flags |= SN_ERR_SRVTO;
3532 if (!(t->flags & SN_FINST_MASK))
3533 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01003534 return 1;
3535 }
willy tarreaub1ff9db2005-12-17 13:51:03 +01003536
3537 /* recompute request time-outs */
3538 if (req->l == 0) {
willy tarreau0f7af912005-12-17 12:21:26 +01003539 if (FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3540 FD_CLR(t->srv_fd, StaticWriteEvent); /* stop writing */
3541 tv_eternity(&t->swexpire);
3542 }
3543 }
willy tarreaub1ff9db2005-12-17 13:51:03 +01003544 else { /* buffer not empty, there are still data to be transferred */
willy tarreau0f7af912005-12-17 12:21:26 +01003545 if (! FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3546 FD_SET(t->srv_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01003547 if (t->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01003548 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003549 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3550 t->srexpire = t->swexpire;
3551 }
willy tarreau0f7af912005-12-17 12:21:26 +01003552 else
3553 tv_eternity(&t->swexpire);
3554 }
3555 }
3556
willy tarreaub1ff9db2005-12-17 13:51:03 +01003557 /* recompute response time-outs */
willy tarreau0f7af912005-12-17 12:21:26 +01003558 if (rep->l == BUFSIZE) { /* no room to read more data */
3559 if (FD_ISSET(t->srv_fd, StaticReadEvent)) {
3560 FD_CLR(t->srv_fd, StaticReadEvent);
3561 tv_eternity(&t->srexpire);
3562 }
3563 }
3564 else {
3565 if (! FD_ISSET(t->srv_fd, StaticReadEvent)) {
3566 FD_SET(t->srv_fd, StaticReadEvent);
3567 if (t->proxy->srvtimeout)
3568 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3569 else
3570 tv_eternity(&t->srexpire);
3571 }
3572 }
3573
3574 return 0; /* other cases change nothing */
3575 }
3576 else if (s == SV_STSHUTR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01003577 if (t->res_sw == RES_ERROR) {
3578 //FD_CLR(t->srv_fd, StaticWriteEvent);
3579 tv_eternity(&t->swexpire);
3580 fd_delete(t->srv_fd);
3581 //close(t->srv_fd);
3582 t->srv_state = SV_STCLOSE;
3583 if (!(t->flags & SN_ERR_MASK))
3584 t->flags |= SN_ERR_SRVCL;
3585 if (!(t->flags & SN_FINST_MASK))
3586 t->flags |= SN_FINST_D;
3587 return 1;
3588 }
3589 else if ((c == CL_STSHUTR || c == CL_STCLOSE) && (req->l == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003590 //FD_CLR(t->srv_fd, StaticWriteEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01003591 tv_eternity(&t->swexpire);
3592 fd_delete(t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003593 //close(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003594 t->srv_state = SV_STCLOSE;
3595 return 1;
3596 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003597 else if (tv_cmp2_ms(&t->swexpire, &now) <= 0) {
3598 //FD_CLR(t->srv_fd, StaticWriteEvent);
3599 tv_eternity(&t->swexpire);
3600 fd_delete(t->srv_fd);
3601 //close(t->srv_fd);
3602 t->srv_state = SV_STCLOSE;
3603 if (!(t->flags & SN_ERR_MASK))
3604 t->flags |= SN_ERR_SRVTO;
3605 if (!(t->flags & SN_FINST_MASK))
3606 t->flags |= SN_FINST_D;
3607 return 1;
3608 }
willy tarreau0f7af912005-12-17 12:21:26 +01003609 else if (req->l == 0) {
3610 if (FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3611 FD_CLR(t->srv_fd, StaticWriteEvent); /* stop writing */
3612 tv_eternity(&t->swexpire);
3613 }
3614 }
3615 else { /* buffer not empty */
3616 if (! FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3617 FD_SET(t->srv_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01003618 if (t->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01003619 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003620 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3621 t->srexpire = t->swexpire;
3622 }
willy tarreau0f7af912005-12-17 12:21:26 +01003623 else
3624 tv_eternity(&t->swexpire);
3625 }
3626 }
3627 return 0;
3628 }
3629 else if (s == SV_STSHUTW) {
willy tarreau036e1ce2005-12-17 13:46:33 +01003630 if (t->res_sr == RES_ERROR) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003631 //FD_CLR(t->srv_fd, StaticReadEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01003632 tv_eternity(&t->srexpire);
3633 fd_delete(t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003634 //close(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003635 t->srv_state = SV_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01003636 if (!(t->flags & SN_ERR_MASK))
3637 t->flags |= SN_ERR_SRVCL;
3638 if (!(t->flags & SN_FINST_MASK))
3639 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01003640 return 1;
3641 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003642 else if (t->res_sr == RES_NULL || c == CL_STSHUTW || c == CL_STCLOSE) {
3643 //FD_CLR(t->srv_fd, StaticReadEvent);
3644 tv_eternity(&t->srexpire);
3645 fd_delete(t->srv_fd);
3646 //close(t->srv_fd);
3647 t->srv_state = SV_STCLOSE;
3648 return 1;
3649 }
3650 else if (tv_cmp2_ms(&t->srexpire, &now) <= 0) {
3651 //FD_CLR(t->srv_fd, StaticReadEvent);
3652 tv_eternity(&t->srexpire);
3653 fd_delete(t->srv_fd);
3654 //close(t->srv_fd);
3655 t->srv_state = SV_STCLOSE;
3656 if (!(t->flags & SN_ERR_MASK))
3657 t->flags |= SN_ERR_SRVTO;
3658 if (!(t->flags & SN_FINST_MASK))
3659 t->flags |= SN_FINST_D;
3660 return 1;
3661 }
willy tarreau0f7af912005-12-17 12:21:26 +01003662 else if (rep->l == BUFSIZE) { /* no room to read more data */
3663 if (FD_ISSET(t->srv_fd, StaticReadEvent)) {
3664 FD_CLR(t->srv_fd, StaticReadEvent);
3665 tv_eternity(&t->srexpire);
3666 }
3667 }
3668 else {
3669 if (! FD_ISSET(t->srv_fd, StaticReadEvent)) {
3670 FD_SET(t->srv_fd, StaticReadEvent);
3671 if (t->proxy->srvtimeout)
3672 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3673 else
3674 tv_eternity(&t->srexpire);
3675 }
3676 }
3677 return 0;
3678 }
3679 else { /* SV_STCLOSE : nothing to do */
willy tarreau9fe663a2005-12-17 13:02:59 +01003680 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau0f7af912005-12-17 12:21:26 +01003681 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01003682 len = sprintf(trash, "%08x:%s.srvcls[%04x:%04x]\n", t->uniq_id, t->proxy->id, (unsigned short)t->cli_fd, (unsigned short)t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003683 write(1, trash, len);
3684 }
3685 return 0;
3686 }
3687 return 0;
3688}
3689
3690
willy tarreau5cbea6f2005-12-17 12:48:26 +01003691/* Processes the client and server jobs of a session task, then
3692 * puts it back to the wait queue in a clean state, or
3693 * cleans up its resources if it must be deleted. Returns
3694 * the time the task accepts to wait, or -1 for infinity
willy tarreau0f7af912005-12-17 12:21:26 +01003695 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003696int process_session(struct task *t) {
3697 struct session *s = t->context;
3698 int fsm_resync = 0;
willy tarreau0f7af912005-12-17 12:21:26 +01003699
willy tarreau5cbea6f2005-12-17 12:48:26 +01003700 do {
3701 fsm_resync = 0;
3702 //fprintf(stderr,"before_cli:cli=%d, srv=%d\n", t->cli_state, t->srv_state);
3703 fsm_resync |= process_cli(s);
3704 //fprintf(stderr,"cli/srv:cli=%d, srv=%d\n", t->cli_state, t->srv_state);
3705 fsm_resync |= process_srv(s);
3706 //fprintf(stderr,"after_srv:cli=%d, srv=%d\n", t->cli_state, t->srv_state);
3707 } while (fsm_resync);
3708
3709 if (s->cli_state != CL_STCLOSE || s->srv_state != SV_STCLOSE) {
willy tarreau0f7af912005-12-17 12:21:26 +01003710 struct timeval min1, min2;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003711 s->res_cw = s->res_cr = s->res_sw = s->res_sr = RES_SILENT;
willy tarreau0f7af912005-12-17 12:21:26 +01003712
willy tarreau5cbea6f2005-12-17 12:48:26 +01003713 tv_min(&min1, &s->crexpire, &s->cwexpire);
3714 tv_min(&min2, &s->srexpire, &s->swexpire);
3715 tv_min(&min1, &min1, &s->cnexpire);
willy tarreau0f7af912005-12-17 12:21:26 +01003716 tv_min(&t->expire, &min1, &min2);
3717
3718 /* restore t to its place in the task list */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003719 task_queue(t);
willy tarreau0f7af912005-12-17 12:21:26 +01003720
willy tarreau5cbea6f2005-12-17 12:48:26 +01003721 return tv_remain(&now, &t->expire); /* nothing more to do */
willy tarreau0f7af912005-12-17 12:21:26 +01003722 }
3723
willy tarreau5cbea6f2005-12-17 12:48:26 +01003724 s->proxy->nbconn--;
willy tarreau0f7af912005-12-17 12:21:26 +01003725 actconn--;
3726
willy tarreau9fe663a2005-12-17 13:02:59 +01003727 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau0f7af912005-12-17 12:21:26 +01003728 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01003729 len = sprintf(trash, "%08x:%s.closed[%04x:%04x]\n", s->uniq_id, s->proxy->id, (unsigned short)s->cli_fd, (unsigned short)s->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003730 write(1, trash, len);
3731 }
3732
willy tarreau750a4722005-12-17 13:21:24 +01003733 s->logs.t_close = tv_diff(&s->logs.tv_accept, &now);
willy tarreaua1598082005-12-17 13:08:06 +01003734 if (s->rep != NULL)
3735 s->logs.bytes = s->rep->total;
3736
willy tarreau9fe663a2005-12-17 13:02:59 +01003737 /* let's do a final log if we need it */
willy tarreaua1598082005-12-17 13:08:06 +01003738 if (s->logs.logwait && (!(s->proxy->options & PR_O_NULLNOLOG) || s->req->total))
willy tarreau9fe663a2005-12-17 13:02:59 +01003739 sess_log(s);
3740
willy tarreau0f7af912005-12-17 12:21:26 +01003741 /* the task MUST not be in the run queue anymore */
3742 task_delete(t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003743 session_free(s);
willy tarreau0f7af912005-12-17 12:21:26 +01003744 task_free(t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003745 return -1; /* rest in peace for eternity */
3746}
3747
3748
3749
3750/*
3751 * manages a server health-check. Returns
3752 * the time the task accepts to wait, or -1 for infinity.
3753 */
3754int process_chk(struct task *t) {
3755 struct server *s = t->context;
willy tarreaua41a8b42005-12-17 14:02:24 +01003756 struct sockaddr_in sa;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003757 int fd = s->curfd;
3758 int one = 1;
3759
willy tarreauef900ab2005-12-17 12:52:52 +01003760 //fprintf(stderr, "process_chk: task=%p\n", t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003761
3762 if (fd < 0) { /* no check currently running */
3763 //fprintf(stderr, "process_chk: 2\n");
3764 if (tv_cmp2_ms(&t->expire, &now) > 0) { /* not good time yet */
3765 task_queue(t); /* restore t to its place in the task list */
3766 return tv_remain(&now, &t->expire);
3767 }
3768
3769 /* we'll initiate a new check */
3770 s->result = 0; /* no result yet */
3771 if ((fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) != -1) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003772 if ((fd < global.maxsock) &&
willy tarreau5cbea6f2005-12-17 12:48:26 +01003773 (fcntl(fd, F_SETFL, O_NONBLOCK) != -1) &&
3774 (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char *) &one, sizeof(one)) != -1)) {
3775 //fprintf(stderr, "process_chk: 3\n");
3776
willy tarreaua41a8b42005-12-17 14:02:24 +01003777 /* we'll connect to the check port on the server */
3778 sa = s->addr;
3779 sa.sin_port = htons(s->check_port);
3780
willy tarreau036e1ce2005-12-17 13:46:33 +01003781 /* allow specific binding */
3782 if (s->proxy->options & PR_O_BIND_SRC &&
3783 bind(fd, (struct sockaddr *)&s->proxy->source_addr, sizeof(s->proxy->source_addr)) == -1) {
3784 Alert("Cannot bind to source address before connect() for proxy %s. Aborting.\n", s->proxy->id);
3785 close(fd);
3786 s->result = -1;
3787 }
willy tarreaua41a8b42005-12-17 14:02:24 +01003788 else if ((connect(fd, (struct sockaddr *)&sa, sizeof(sa)) != -1) || (errno == EINPROGRESS)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003789 /* OK, connection in progress or established */
3790
3791 //fprintf(stderr, "process_chk: 4\n");
3792
3793 s->curfd = fd; /* that's how we know a test is in progress ;-) */
3794 fdtab[fd].owner = t;
willy tarreaubc4e1fb2005-12-17 13:32:07 +01003795 fdtab[fd].read = &event_srv_chk_r;
3796 fdtab[fd].write = &event_srv_chk_w;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003797 fdtab[fd].state = FD_STCONN; /* connection in progress */
3798 FD_SET(fd, StaticWriteEvent); /* for connect status */
3799 fd_insert(fd);
willy tarreaue47c8d72005-12-17 12:55:52 +01003800 /* FIXME: we allow up to <inter> for a connection to establish, but we should use another parameter */
3801 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003802 task_queue(t); /* restore t to its place in the task list */
3803 return tv_remain(&now, &t->expire);
3804 }
3805 else if (errno != EALREADY && errno != EISCONN && errno != EAGAIN) {
3806 s->result = -1; /* a real error */
3807 }
3808 }
3809 //fprintf(stderr, "process_chk: 5\n");
3810 close(fd);
3811 }
3812
3813 if (!s->result) { /* nothing done */
3814 //fprintf(stderr, "process_chk: 6\n");
willy tarreaue47c8d72005-12-17 12:55:52 +01003815 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003816 task_queue(t); /* restore t to its place in the task list */
3817 return tv_remain(&now, &t->expire);
3818 }
3819
3820 /* here, we have seen a failure */
willy tarreaue47c8d72005-12-17 12:55:52 +01003821 if (s->health > s->rise)
willy tarreau5cbea6f2005-12-17 12:48:26 +01003822 s->health--; /* still good */
3823 else {
willy tarreau535ae7a2005-12-17 12:58:00 +01003824 if (s->health == s->rise) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003825 if (!(global.mode & MODE_QUIET))
willy tarreau8337c6b2005-12-17 13:41:01 +01003826 Warning("server %s/%s DOWN.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003827
willy tarreau9fe663a2005-12-17 13:02:59 +01003828 send_log(s->proxy, LOG_ALERT, "Server %s/%s is DOWN.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003829 }
willy tarreauef900ab2005-12-17 12:52:52 +01003830
willy tarreau5cbea6f2005-12-17 12:48:26 +01003831 s->health = 0; /* failure */
3832 s->state &= ~SRV_RUNNING;
3833 }
3834
3835 //fprintf(stderr, "process_chk: 7\n");
willy tarreaue47c8d72005-12-17 12:55:52 +01003836 /* FIXME: we allow up to <inter> for a connection to establish, but we should use another parameter */
3837 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003838 }
3839 else {
3840 //fprintf(stderr, "process_chk: 8\n");
3841 /* there was a test running */
3842 if (s->result > 0) { /* good server detected */
3843 //fprintf(stderr, "process_chk: 9\n");
3844 s->health++; /* was bad, stays for a while */
willy tarreaue47c8d72005-12-17 12:55:52 +01003845 if (s->health >= s->rise) {
willy tarreau535ae7a2005-12-17 12:58:00 +01003846 if (s->health == s->rise) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003847 if (!(global.mode & MODE_QUIET))
willy tarreau8337c6b2005-12-17 13:41:01 +01003848 Warning("server %s/%s UP.\n", s->proxy->id, s->id);
willy tarreau9fe663a2005-12-17 13:02:59 +01003849 send_log(s->proxy, LOG_NOTICE, "Server %s/%s is UP.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003850 }
willy tarreauef900ab2005-12-17 12:52:52 +01003851
willy tarreaue47c8d72005-12-17 12:55:52 +01003852 s->health = s->rise + s->fall - 1; /* OK now */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003853 s->state |= SRV_RUNNING;
3854 }
willy tarreauef900ab2005-12-17 12:52:52 +01003855 s->curfd = -1; /* no check running anymore */
3856 //FD_CLR(fd, StaticWriteEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003857 fd_delete(fd);
willy tarreaue47c8d72005-12-17 12:55:52 +01003858 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003859 }
3860 else if (s->result < 0 || tv_cmp2_ms(&t->expire, &now) <= 0) {
3861 //fprintf(stderr, "process_chk: 10\n");
3862 /* failure or timeout detected */
willy tarreaue47c8d72005-12-17 12:55:52 +01003863 if (s->health > s->rise)
willy tarreau5cbea6f2005-12-17 12:48:26 +01003864 s->health--; /* still good */
3865 else {
willy tarreau535ae7a2005-12-17 12:58:00 +01003866 if (s->health == s->rise) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003867 if (!(global.mode & MODE_QUIET))
willy tarreau8337c6b2005-12-17 13:41:01 +01003868 Warning("server %s/%s DOWN.\n", s->proxy->id, s->id);
willy tarreau9fe663a2005-12-17 13:02:59 +01003869
3870 send_log(s->proxy, LOG_ALERT, "Server %s/%s is DOWN.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003871 }
willy tarreauef900ab2005-12-17 12:52:52 +01003872
willy tarreau5cbea6f2005-12-17 12:48:26 +01003873 s->health = 0; /* failure */
3874 s->state &= ~SRV_RUNNING;
3875 }
3876 s->curfd = -1;
willy tarreauef900ab2005-12-17 12:52:52 +01003877 //FD_CLR(fd, StaticWriteEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003878 fd_delete(fd);
willy tarreaue47c8d72005-12-17 12:55:52 +01003879 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003880 }
3881 /* if result is 0 and there's no timeout, we have to wait again */
3882 }
3883 //fprintf(stderr, "process_chk: 11\n");
3884 s->result = 0;
3885 task_queue(t); /* restore t to its place in the task list */
3886 return tv_remain(&now, &t->expire);
willy tarreau0f7af912005-12-17 12:21:26 +01003887}
3888
3889
willy tarreau5cbea6f2005-12-17 12:48:26 +01003890
willy tarreau0f7af912005-12-17 12:21:26 +01003891#if STATTIME > 0
3892int stats(void);
3893#endif
3894
3895/*
3896 * Main select() loop.
3897 */
3898
3899void select_loop() {
3900 int next_time;
willy tarreau0f7af912005-12-17 12:21:26 +01003901 int time2;
willy tarreau0f7af912005-12-17 12:21:26 +01003902 int status;
3903 int fd,i;
3904 struct timeval delta;
3905 int readnotnull, writenotnull;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003906 struct task *t, *tnext;
willy tarreau0f7af912005-12-17 12:21:26 +01003907
willy tarreau5cbea6f2005-12-17 12:48:26 +01003908 tv_now(&now);
3909
3910 while (1) {
3911 next_time = -1; /* set the timer to wait eternally first */
willy tarreau0f7af912005-12-17 12:21:26 +01003912
willy tarreau5cbea6f2005-12-17 12:48:26 +01003913 /* look for expired tasks and add them to the run queue.
3914 */
3915 tnext = ((struct task *)LIST_HEAD(wait_queue))->next;
3916 while ((t = tnext) != LIST_HEAD(wait_queue)) { /* we haven't looped ? */
3917 tnext = t->next;
willy tarreauef900ab2005-12-17 12:52:52 +01003918 if (t->state & TASK_RUNNING)
3919 continue;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003920
3921 /* wakeup expired entries. It doesn't matter if they are
3922 * already running because of a previous event
3923 */
3924 if (tv_cmp2_ms(&t->expire, &now) <= 0) {
willy tarreauef900ab2005-12-17 12:52:52 +01003925 //fprintf(stderr,"task_wakeup(%p, %p)\n", &rq, t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003926 task_wakeup(&rq, t);
3927 }
3928 else {
willy tarreauef900ab2005-12-17 12:52:52 +01003929 /* first non-runnable task. Use its expiration date as an upper bound */
3930 int temp_time = tv_remain(&now, &t->expire);
3931 if (temp_time)
3932 next_time = temp_time;
3933 //fprintf(stderr,"no_task_wakeup(%p, %p) : expire in %d ms\n", &rq, t, temp_time);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003934 break;
3935 }
3936 }
3937
3938 /* process each task in the run queue now. Each task may be deleted
3939 * since we only use tnext.
3940 */
3941 tnext = rq;
3942 while ((t = tnext) != NULL) {
3943 int temp_time;
3944
3945 tnext = t->rqnext;
3946 task_sleep(&rq, t);
willy tarreauef900ab2005-12-17 12:52:52 +01003947 //fprintf(stderr,"task %p\n",t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003948 temp_time = t->process(t);
3949 next_time = MINTIME(temp_time, next_time);
willy tarreauef900ab2005-12-17 12:52:52 +01003950 //fprintf(stderr,"process(%p)=%d -> next_time=%d)\n", t, temp_time, next_time);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003951 }
3952
willy tarreauef900ab2005-12-17 12:52:52 +01003953 //fprintf(stderr,"---end of run---\n");
willy tarreau5cbea6f2005-12-17 12:48:26 +01003954
3955 /* maintain all proxies in a consistent state. This should quickly become a task */
3956 time2 = maintain_proxies();
3957 next_time = MINTIME(time2, next_time);
3958
3959 /* stop when there's no connection left and we don't allow them anymore */
3960 if (!actconn && listeners == 0)
3961 break;
3962
willy tarreau0f7af912005-12-17 12:21:26 +01003963
3964#if STATTIME > 0
3965 time2 = stats();
3966 // fprintf(stderr," stats = %d\n", time2);
3967 next_time = MINTIME(time2, next_time);
3968#endif
3969
willy tarreau5cbea6f2005-12-17 12:48:26 +01003970 if (next_time > 0) { /* FIXME */
willy tarreau0f7af912005-12-17 12:21:26 +01003971 /* Convert to timeval */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003972 /* to avoid eventual select loops due to timer precision */
3973 next_time += SCHEDULER_RESOLUTION;
3974 delta.tv_sec = next_time / 1000;
3975 delta.tv_usec = (next_time % 1000) * 1000;
3976 }
3977 else if (next_time == 0) { /* allow select to return immediately when needed */
3978 delta.tv_sec = delta.tv_usec = 0;
willy tarreau0f7af912005-12-17 12:21:26 +01003979 }
3980
3981
3982 /* let's restore fdset state */
3983
3984 readnotnull = 0; writenotnull = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01003985 for (i = 0; i < (global.maxsock + FD_SETSIZE - 1)/(8*sizeof(int)); i++) {
willy tarreau0f7af912005-12-17 12:21:26 +01003986 readnotnull |= (*(((int*)ReadEvent)+i) = *(((int*)StaticReadEvent)+i)) != 0;
3987 writenotnull |= (*(((int*)WriteEvent)+i) = *(((int*)StaticWriteEvent)+i)) != 0;
3988 }
3989
3990// /* just a verification code, needs to be removed for performance */
3991// for (i=0; i<maxfd; i++) {
3992// if (FD_ISSET(i, ReadEvent) != FD_ISSET(i, StaticReadEvent))
3993// abort();
3994// if (FD_ISSET(i, WriteEvent) != FD_ISSET(i, StaticWriteEvent))
3995// abort();
3996//
3997// }
3998
willy tarreau197e8ec2005-12-17 14:10:59 +01003999 status = select(maxfd,
4000 readnotnull ? ReadEvent : NULL,
4001 writenotnull ? WriteEvent : NULL,
4002 NULL,
4003 (next_time >= 0) ? &delta : NULL);
willy tarreau0f7af912005-12-17 12:21:26 +01004004
willy tarreau5cbea6f2005-12-17 12:48:26 +01004005 /* this is an experiment on the separation of the select work */
4006 // status = (readnotnull ? select(maxfd, ReadEvent, NULL, NULL, (next_time >= 0) ? &delta : NULL) : 0);
4007 // status |= (writenotnull ? select(maxfd, NULL, WriteEvent, NULL, (next_time >= 0) ? &delta : NULL) : 0);
4008
willy tarreau0f7af912005-12-17 12:21:26 +01004009 tv_now(&now);
willy tarreau5cbea6f2005-12-17 12:48:26 +01004010
willy tarreau0f7af912005-12-17 12:21:26 +01004011 if (status > 0) { /* must proceed with events */
4012
4013 int fds;
4014 char count;
4015
4016 for (fds = 0; (fds << INTBITS) < maxfd; fds++)
4017 if ((((int *)(ReadEvent))[fds] | ((int *)(WriteEvent))[fds]) != 0)
4018 for (count = 1<<INTBITS, fd = fds << INTBITS; count && fd < maxfd; count--, fd++) {
4019
willy tarreau5cbea6f2005-12-17 12:48:26 +01004020 /* if we specify read first, the accepts and zero reads will be
4021 * seen first. Moreover, system buffers will be flushed faster.
4022 */
willy tarreau0f7af912005-12-17 12:21:26 +01004023 if (fdtab[fd].state == FD_STCLOSE)
4024 continue;
willy tarreau0f7af912005-12-17 12:21:26 +01004025
4026 if (FD_ISSET(fd, ReadEvent))
4027 fdtab[fd].read(fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01004028
willy tarreau5cbea6f2005-12-17 12:48:26 +01004029 if (FD_ISSET(fd, WriteEvent))
4030 fdtab[fd].write(fd);
willy tarreau0f7af912005-12-17 12:21:26 +01004031 }
4032 }
4033 else {
4034 // fprintf(stderr,"select returned %d, maxfd=%d\n", status, maxfd);
4035 }
willy tarreau0f7af912005-12-17 12:21:26 +01004036 }
4037}
4038
4039
4040#if STATTIME > 0
4041/*
4042 * Display proxy statistics regularly. It is designed to be called from the
4043 * select_loop().
4044 */
4045int stats(void) {
4046 static int lines;
4047 static struct timeval nextevt;
4048 static struct timeval lastevt;
4049 static struct timeval starttime = {0,0};
4050 unsigned long totaltime, deltatime;
4051 int ret;
4052
willy tarreau750a4722005-12-17 13:21:24 +01004053 if (tv_cmp(&now, &nextevt) > 0) {
willy tarreau6e682ce2005-12-17 13:26:49 +01004054 deltatime = (tv_diff(&lastevt, &now)?:1);
4055 totaltime = (tv_diff(&starttime, &now)?:1);
willy tarreau0f7af912005-12-17 12:21:26 +01004056
willy tarreau9fe663a2005-12-17 13:02:59 +01004057 if (global.mode & MODE_STATS) {
4058 if ((lines++ % 16 == 0) && !(global.mode & MODE_LOG))
willy tarreau5cbea6f2005-12-17 12:48:26 +01004059 qfprintf(stderr,
willy tarreau0f7af912005-12-17 12:21:26 +01004060 "\n active total tsknew tskgood tskleft tskrght tsknsch tsklsch tskrsch\n");
4061 if (lines>1) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01004062 qfprintf(stderr,"%07d %07d %07d %07d %07d %07d %07d %07d %07d\n",
willy tarreau0f7af912005-12-17 12:21:26 +01004063 actconn, totalconn,
4064 stats_tsk_new, stats_tsk_good,
4065 stats_tsk_left, stats_tsk_right,
4066 stats_tsk_nsrch, stats_tsk_lsrch, stats_tsk_rsrch);
4067 }
4068 }
4069
4070 tv_delayfrom(&nextevt, &now, STATTIME);
4071
4072 lastevt=now;
4073 }
4074 ret = tv_remain(&now, &nextevt);
4075 return ret;
4076}
4077#endif
4078
4079
4080/*
4081 * this function enables proxies when there are enough free sessions,
4082 * or stops them when the table is full. It is designed to be called from the
willy tarreau5cbea6f2005-12-17 12:48:26 +01004083 * select_loop(). It returns the time left before next expiration event
4084 * during stop time, -1 otherwise.
willy tarreau0f7af912005-12-17 12:21:26 +01004085 */
4086static int maintain_proxies(void) {
4087 struct proxy *p;
willy tarreaua41a8b42005-12-17 14:02:24 +01004088 struct listener *l;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004089 int tleft; /* time left */
willy tarreau0f7af912005-12-17 12:21:26 +01004090
4091 p = proxy;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004092 tleft = -1; /* infinite time */
willy tarreau0f7af912005-12-17 12:21:26 +01004093
4094 /* if there are enough free sessions, we'll activate proxies */
willy tarreau9fe663a2005-12-17 13:02:59 +01004095 if (actconn < global.maxconn) {
willy tarreau0f7af912005-12-17 12:21:26 +01004096 while (p) {
4097 if (p->nbconn < p->maxconn) {
4098 if (p->state == PR_STIDLE) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004099 for (l = p->listen; l != NULL; l = l->next) {
4100 FD_SET(l->fd, StaticReadEvent);
4101 }
willy tarreau0f7af912005-12-17 12:21:26 +01004102 p->state = PR_STRUN;
4103 }
4104 }
4105 else {
4106 if (p->state == PR_STRUN) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004107 for (l = p->listen; l != NULL; l = l->next) {
4108 FD_CLR(l->fd, StaticReadEvent);
4109 }
willy tarreau0f7af912005-12-17 12:21:26 +01004110 p->state = PR_STIDLE;
4111 }
4112 }
4113 p = p->next;
4114 }
4115 }
4116 else { /* block all proxies */
4117 while (p) {
4118 if (p->state == PR_STRUN) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004119 for (l = p->listen; l != NULL; l = l->next) {
4120 FD_CLR(l->fd, StaticReadEvent);
4121 }
willy tarreau0f7af912005-12-17 12:21:26 +01004122 p->state = PR_STIDLE;
4123 }
4124 p = p->next;
4125 }
4126 }
4127
willy tarreau5cbea6f2005-12-17 12:48:26 +01004128 if (stopping) {
4129 p = proxy;
4130 while (p) {
4131 if (p->state != PR_STDISABLED) {
4132 int t;
4133 t = tv_remain(&now, &p->stop_time);
4134 if (t == 0) {
willy tarreau535ae7a2005-12-17 12:58:00 +01004135 Warning("Proxy %s stopped.\n", p->id);
willy tarreau9fe663a2005-12-17 13:02:59 +01004136 send_log(p, LOG_WARNING, "Proxy %s stopped.\n", p->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01004137
willy tarreaua41a8b42005-12-17 14:02:24 +01004138 for (l = p->listen; l != NULL; l = l->next) {
4139 fd_delete(l->fd);
4140 listeners--;
4141 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004142 p->state = PR_STDISABLED;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004143 }
4144 else {
4145 tleft = MINTIME(t, tleft);
4146 }
4147 }
4148 p = p->next;
4149 }
4150 }
4151 return tleft;
willy tarreau0f7af912005-12-17 12:21:26 +01004152}
4153
4154/*
4155 * this function disables health-check servers so that the process will quickly be ignored
4156 * by load balancers.
4157 */
4158static void soft_stop(void) {
4159 struct proxy *p;
4160
4161 stopping = 1;
4162 p = proxy;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004163 tv_now(&now); /* else, the old time before select will be used */
willy tarreau0f7af912005-12-17 12:21:26 +01004164 while (p) {
willy tarreau535ae7a2005-12-17 12:58:00 +01004165 if (p->state != PR_STDISABLED) {
4166 Warning("Stopping proxy %s in %d ms.\n", p->id, p->grace);
willy tarreau9fe663a2005-12-17 13:02:59 +01004167 send_log(p, LOG_WARNING, "Stopping proxy %s in %d ms.\n", p->id, p->grace);
willy tarreau0f7af912005-12-17 12:21:26 +01004168 tv_delayfrom(&p->stop_time, &now, p->grace);
willy tarreau535ae7a2005-12-17 12:58:00 +01004169 }
willy tarreau0f7af912005-12-17 12:21:26 +01004170 p = p->next;
4171 }
4172}
4173
4174/*
4175 * upon SIGUSR1, let's have a soft stop.
4176 */
4177void sig_soft_stop(int sig) {
4178 soft_stop();
4179 signal(sig, SIG_IGN);
4180}
4181
4182
willy tarreau8337c6b2005-12-17 13:41:01 +01004183/*
4184 * this function dumps every server's state when the process receives SIGHUP.
4185 */
4186void sig_dump_state(int sig) {
4187 struct proxy *p = proxy;
4188
4189 Warning("SIGHUP received, dumping servers states.\n");
4190 while (p) {
4191 struct server *s = p->srv;
4192
4193 send_log(p, LOG_NOTICE, "SIGUP received, dumping servers states.\n");
4194 while (s) {
4195 if (s->state & SRV_RUNNING) {
4196 Warning("SIGHUP: server %s/%s is UP.\n", p->id, s->id);
4197 send_log(p, LOG_NOTICE, "SIGUP: server %s/%s is UP.\n", p->id, s->id);
4198 }
4199 else {
4200 Warning("SIGHUP: server %s/%s is DOWN.\n", p->id, s->id);
4201 send_log(p, LOG_NOTICE, "SIGHUP: server %s/%s is DOWN.\n", p->id, s->id);
4202 }
4203 s = s->next;
4204 }
4205 p = p->next;
4206 }
4207 signal(sig, sig_dump_state);
4208}
4209
willy tarreau0f7af912005-12-17 12:21:26 +01004210void dump(int sig) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01004211 struct task *t, *tnext;
4212 struct session *s;
willy tarreau0f7af912005-12-17 12:21:26 +01004213
willy tarreau5cbea6f2005-12-17 12:48:26 +01004214 tnext = ((struct task *)LIST_HEAD(wait_queue))->next;
4215 while ((t = tnext) != LIST_HEAD(wait_queue)) { /* we haven't looped ? */
4216 tnext = t->next;
4217 s = t->context;
4218 qfprintf(stderr,"[dump] wq: task %p, still %ld ms, "
4219 "cli=%d, srv=%d, cr=%d, cw=%d, sr=%d, sw=%d, "
4220 "req=%d, rep=%d, clifd=%d\n",
4221 s, tv_remain(&now, &t->expire),
4222 s->cli_state,
4223 s->srv_state,
4224 FD_ISSET(s->cli_fd, StaticReadEvent),
4225 FD_ISSET(s->cli_fd, StaticWriteEvent),
4226 FD_ISSET(s->srv_fd, StaticReadEvent),
4227 FD_ISSET(s->srv_fd, StaticWriteEvent),
4228 s->req->l, s->rep?s->rep->l:0, s->cli_fd
4229 );
willy tarreau0f7af912005-12-17 12:21:26 +01004230 }
4231}
4232
willy tarreaue39cd132005-12-17 13:00:18 +01004233void chain_regex(struct hdr_exp **head, regex_t *preg, int action, char *replace) {
4234 struct hdr_exp *exp;
4235
4236 while (*head != NULL)
4237 head = &(*head)->next;
4238
4239 exp = calloc(1, sizeof(struct hdr_exp));
4240
4241 exp->preg = preg;
4242 exp->replace = replace;
4243 exp->action = action;
4244 *head = exp;
4245}
4246
willy tarreau9fe663a2005-12-17 13:02:59 +01004247
willy tarreau0f7af912005-12-17 12:21:26 +01004248/*
willy tarreau9fe663a2005-12-17 13:02:59 +01004249 * parse a line in a <global> section. Returns 0 if OK, -1 if error.
willy tarreau0f7af912005-12-17 12:21:26 +01004250 */
willy tarreau9fe663a2005-12-17 13:02:59 +01004251int cfg_parse_global(char *file, int linenum, char **args) {
willy tarreau0f7af912005-12-17 12:21:26 +01004252
willy tarreau9fe663a2005-12-17 13:02:59 +01004253 if (!strcmp(args[0], "global")) { /* new section */
4254 /* no option, nothing special to do */
4255 return 0;
4256 }
4257 else if (!strcmp(args[0], "daemon")) {
4258 global.mode |= MODE_DAEMON;
4259 }
4260 else if (!strcmp(args[0], "debug")) {
4261 global.mode |= MODE_DEBUG;
4262 }
4263 else if (!strcmp(args[0], "quiet")) {
4264 global.mode |= MODE_QUIET;
4265 }
4266 else if (!strcmp(args[0], "stats")) {
4267 global.mode |= MODE_STATS;
4268 }
4269 else if (!strcmp(args[0], "uid")) {
4270 if (global.uid != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004271 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004272 return 0;
willy tarreau0f7af912005-12-17 12:21:26 +01004273 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004274 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004275 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004276 return -1;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004277 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004278 global.uid = atol(args[1]);
4279 }
4280 else if (!strcmp(args[0], "gid")) {
4281 if (global.gid != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004282 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004283 return 0;
willy tarreau0f7af912005-12-17 12:21:26 +01004284 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004285 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004286 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01004287 return -1;
4288 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004289 global.gid = atol(args[1]);
4290 }
4291 else if (!strcmp(args[0], "nbproc")) {
4292 if (global.nbproc != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004293 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004294 return 0;
willy tarreau0f7af912005-12-17 12:21:26 +01004295 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004296 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004297 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004298 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004299 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004300 global.nbproc = atol(args[1]);
4301 }
4302 else if (!strcmp(args[0], "maxconn")) {
4303 if (global.maxconn != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004304 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004305 return 0;
willy tarreau0f7af912005-12-17 12:21:26 +01004306 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004307 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004308 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004309 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004310 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004311 global.maxconn = atol(args[1]);
4312 }
4313 else if (!strcmp(args[0], "chroot")) {
4314 if (global.chroot != NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004315 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004316 return 0;
4317 }
4318 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004319 Alert("parsing [%s:%d] : '%s' expects a directory as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004320 return -1;
4321 }
4322 global.chroot = strdup(args[1]);
4323 }
4324 else if (!strcmp(args[0], "log")) { /* syslog server address */
4325 struct sockaddr_in *sa;
willy tarreau8337c6b2005-12-17 13:41:01 +01004326 int facility, level;
willy tarreau9fe663a2005-12-17 13:02:59 +01004327
4328 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004329 Alert("parsing [%s:%d] : '%s' expects <address> and <facility> as arguments.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004330 return -1;
4331 }
4332
4333 for (facility = 0; facility < NB_LOG_FACILITIES; facility++)
4334 if (!strcmp(log_facilities[facility], args[2]))
4335 break;
4336
4337 if (facility >= NB_LOG_FACILITIES) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004338 Alert("parsing [%s:%d] : unknown log facility '%s'\n", file, linenum, args[2]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004339 exit(1);
4340 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004341
4342 level = 7; /* max syslog level = debug */
4343 if (*(args[3])) {
4344 while (level >= 0 && strcmp(log_levels[level], args[3]))
4345 level--;
4346 if (level < 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004347 Alert("parsing [%s:%d] : unknown optional log level '%s'\n", file, linenum, args[3]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004348 exit(1);
4349 }
4350 }
4351
willy tarreau9fe663a2005-12-17 13:02:59 +01004352 sa = str2sa(args[1]);
4353 if (!sa->sin_port)
4354 sa->sin_port = htons(SYSLOG_PORT);
4355
4356 if (global.logfac1 == -1) {
4357 global.logsrv1 = *sa;
4358 global.logfac1 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004359 global.loglev1 = level;
willy tarreau9fe663a2005-12-17 13:02:59 +01004360 }
4361 else if (global.logfac2 == -1) {
4362 global.logsrv2 = *sa;
4363 global.logfac2 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004364 global.loglev2 = level;
willy tarreau9fe663a2005-12-17 13:02:59 +01004365 }
4366 else {
4367 Alert("parsing [%s:%d] : too many syslog servers\n", file, linenum);
4368 return -1;
4369 }
4370
4371 }
4372 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004373 Alert("parsing [%s:%d] : unknown keyword '%s' in '%s' section\n", file, linenum, args[0], "global");
willy tarreau9fe663a2005-12-17 13:02:59 +01004374 return -1;
4375 }
4376 return 0;
4377}
4378
4379
willy tarreaua41a8b42005-12-17 14:02:24 +01004380void init_default_instance() {
4381 memset(&defproxy, 0, sizeof(defproxy));
4382 defproxy.mode = PR_MODE_TCP;
4383 defproxy.state = PR_STNEW;
4384 defproxy.maxconn = cfg_maxpconn;
4385 defproxy.conn_retries = CONN_RETRIES;
4386 defproxy.logfac1 = defproxy.logfac2 = -1; /* log disabled */
4387}
4388
willy tarreau9fe663a2005-12-17 13:02:59 +01004389/*
4390 * parse a line in a <listen> section. Returns 0 if OK, -1 if error.
4391 */
4392int cfg_parse_listen(char *file, int linenum, char **args) {
4393 static struct proxy *curproxy = NULL;
4394 struct server *newsrv = NULL;
4395
4396 if (!strcmp(args[0], "listen")) { /* new proxy */
willy tarreaua41a8b42005-12-17 14:02:24 +01004397 if (!*args[1]) {
4398 Alert("parsing [%s:%d] : '%s' expects an <id> argument and\n"
4399 " optionnally supports [addr1]:port1[-end1]{,[addr]:port[-end]}...\n",
willy tarreau036e1ce2005-12-17 13:46:33 +01004400 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004401 return -1;
4402 }
4403
4404 if ((curproxy = (struct proxy *)calloc(1, sizeof(struct proxy))) == NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004405 Alert("parsing [%s:%d] : out of memory.\n", file, linenum);
willy tarreau9fe663a2005-12-17 13:02:59 +01004406 return -1;
4407 }
4408 curproxy->next = proxy;
4409 proxy = curproxy;
4410 curproxy->id = strdup(args[1]);
willy tarreaua41a8b42005-12-17 14:02:24 +01004411 if (strchr(args[2], ':') != NULL)
4412 curproxy->listen = str2listener(args[2], curproxy->listen);
4413
willy tarreau9fe663a2005-12-17 13:02:59 +01004414 /* set default values */
willy tarreaua41a8b42005-12-17 14:02:24 +01004415 curproxy->state = defproxy.state;
4416 curproxy->maxconn = defproxy.maxconn;
4417 curproxy->conn_retries = defproxy.conn_retries;
4418 curproxy->options = defproxy.options;
willy tarreaueedaa9f2005-12-17 14:08:03 +01004419
4420 if (defproxy.check_req)
4421 curproxy->check_req = strdup(defproxy.check_req);
4422 curproxy->check_len = defproxy.check_len;
4423
4424 if (defproxy.cookie_name)
4425 curproxy->cookie_name = strdup(defproxy.cookie_name);
4426 curproxy->cookie_len = defproxy.cookie_len;
4427
4428 if (defproxy.capture_name)
4429 curproxy->capture_name = strdup(defproxy.capture_name);
4430 curproxy->capture_namelen = defproxy.capture_namelen;
4431 curproxy->capture_len = defproxy.capture_len;
4432
4433 if (defproxy.errmsg.msg400)
4434 curproxy->errmsg.msg400 = strdup(defproxy.errmsg.msg400);
4435 curproxy->errmsg.len400 = defproxy.errmsg.len400;
4436
4437 if (defproxy.errmsg.msg403)
4438 curproxy->errmsg.msg403 = strdup(defproxy.errmsg.msg403);
4439 curproxy->errmsg.len403 = defproxy.errmsg.len403;
4440
4441 if (defproxy.errmsg.msg408)
4442 curproxy->errmsg.msg408 = strdup(defproxy.errmsg.msg408);
4443 curproxy->errmsg.len408 = defproxy.errmsg.len408;
4444
4445 if (defproxy.errmsg.msg500)
4446 curproxy->errmsg.msg500 = strdup(defproxy.errmsg.msg500);
4447 curproxy->errmsg.len500 = defproxy.errmsg.len500;
4448
4449 if (defproxy.errmsg.msg502)
4450 curproxy->errmsg.msg502 = strdup(defproxy.errmsg.msg502);
4451 curproxy->errmsg.len502 = defproxy.errmsg.len502;
4452
4453 if (defproxy.errmsg.msg503)
4454 curproxy->errmsg.msg503 = strdup(defproxy.errmsg.msg503);
4455 curproxy->errmsg.len503 = defproxy.errmsg.len503;
4456
4457 if (defproxy.errmsg.msg504)
4458 curproxy->errmsg.msg504 = strdup(defproxy.errmsg.msg504);
4459 curproxy->errmsg.len504 = defproxy.errmsg.len504;
4460
willy tarreaua41a8b42005-12-17 14:02:24 +01004461 curproxy->clitimeout = defproxy.clitimeout;
4462 curproxy->contimeout = defproxy.contimeout;
4463 curproxy->srvtimeout = defproxy.srvtimeout;
4464 curproxy->mode = defproxy.mode;
4465 curproxy->logfac1 = defproxy.logfac1;
4466 curproxy->logsrv1 = defproxy.logsrv1;
4467 curproxy->loglev1 = defproxy.loglev1;
4468 curproxy->logfac2 = defproxy.logfac2;
4469 curproxy->logsrv2 = defproxy.logsrv2;
4470 curproxy->loglev2 = defproxy.loglev2;
4471 curproxy->to_log = defproxy.to_log;
4472 curproxy->grace = defproxy.grace;
4473 curproxy->source_addr = defproxy.source_addr;
4474 return 0;
4475 }
4476 else if (!strcmp(args[0], "defaults")) { /* use this one to assign default values */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004477 /* some variables may have already been initialized earlier */
4478 if (defproxy.check_req) free(defproxy.check_req);
4479 if (defproxy.cookie_name) free(defproxy.cookie_name);
4480 if (defproxy.capture_name) free(defproxy.capture_name);
4481 if (defproxy.errmsg.msg400) free(defproxy.errmsg.msg400);
4482 if (defproxy.errmsg.msg403) free(defproxy.errmsg.msg403);
4483 if (defproxy.errmsg.msg408) free(defproxy.errmsg.msg408);
4484 if (defproxy.errmsg.msg500) free(defproxy.errmsg.msg500);
4485 if (defproxy.errmsg.msg502) free(defproxy.errmsg.msg502);
4486 if (defproxy.errmsg.msg503) free(defproxy.errmsg.msg503);
4487 if (defproxy.errmsg.msg504) free(defproxy.errmsg.msg504);
4488
4489 init_default_instance();
willy tarreaua41a8b42005-12-17 14:02:24 +01004490 curproxy = &defproxy;
willy tarreau9fe663a2005-12-17 13:02:59 +01004491 return 0;
4492 }
4493 else if (curproxy == NULL) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004494 Alert("parsing [%s:%d] : 'listen' or 'defaults' expected.\n", file, linenum);
willy tarreau9fe663a2005-12-17 13:02:59 +01004495 return -1;
4496 }
4497
willy tarreaua41a8b42005-12-17 14:02:24 +01004498 if (!strcmp(args[0], "bind")) { /* new listen addresses */
4499 if (curproxy == &defproxy) {
4500 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4501 return -1;
4502 }
4503
4504 if (strchr(args[1], ':') == NULL) {
4505 Alert("parsing [%s:%d] : '%s' expects [addr1]:port1[-end1]{,[addr]:port[-end]}... as arguments.\n",
4506 file, linenum, args[0]);
4507 return -1;
4508 }
4509 curproxy->listen = str2listener(args[1], curproxy->listen);
4510 return 0;
4511 }
4512 else if (!strcmp(args[0], "mode")) { /* sets the proxy mode */
willy tarreau9fe663a2005-12-17 13:02:59 +01004513 if (!strcmp(args[1], "http")) curproxy->mode = PR_MODE_HTTP;
4514 else if (!strcmp(args[1], "tcp")) curproxy->mode = PR_MODE_TCP;
4515 else if (!strcmp(args[1], "health")) curproxy->mode = PR_MODE_HEALTH;
4516 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004517 Alert("parsing [%s:%d] : unknown proxy mode '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004518 return -1;
4519 }
4520 }
4521 else if (!strcmp(args[0], "disabled")) { /* disables this proxy */
4522 curproxy->state = PR_STDISABLED;
4523 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004524 else if (!strcmp(args[0], "enabled")) { /* enables this proxy (used to revert a disabled default) */
4525 curproxy->state = PR_STNEW;
4526 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004527 else if (!strcmp(args[0], "cookie")) { /* cookie name */
4528 int cur_arg;
willy tarreaueedaa9f2005-12-17 14:08:03 +01004529// if (curproxy == &defproxy) {
4530// Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4531// return -1;
4532// }
willy tarreaua41a8b42005-12-17 14:02:24 +01004533
willy tarreau9fe663a2005-12-17 13:02:59 +01004534 if (curproxy->cookie_name != NULL) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01004535// Alert("parsing [%s:%d] : cookie name already specified. Continuing.\n",
4536// file, linenum);
4537// return 0;
4538 free(curproxy->cookie_name);
willy tarreau9fe663a2005-12-17 13:02:59 +01004539 }
4540
4541 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004542 Alert("parsing [%s:%d] : '%s' expects <cookie_name> as argument.\n",
4543 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004544 return -1;
4545 }
4546 curproxy->cookie_name = strdup(args[1]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004547 curproxy->cookie_len = strlen(curproxy->cookie_name);
willy tarreau9fe663a2005-12-17 13:02:59 +01004548
4549 cur_arg = 2;
4550 while (*(args[cur_arg])) {
4551 if (!strcmp(args[cur_arg], "rewrite")) {
4552 curproxy->options |= PR_O_COOK_RW;
willy tarreau0f7af912005-12-17 12:21:26 +01004553 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004554 else if (!strcmp(args[cur_arg], "indirect")) {
4555 curproxy->options |= PR_O_COOK_IND;
willy tarreau0f7af912005-12-17 12:21:26 +01004556 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004557 else if (!strcmp(args[cur_arg], "insert")) {
4558 curproxy->options |= PR_O_COOK_INS;
willy tarreau0f7af912005-12-17 12:21:26 +01004559 }
willy tarreau240afa62005-12-17 13:14:35 +01004560 else if (!strcmp(args[cur_arg], "nocache")) {
4561 curproxy->options |= PR_O_COOK_NOC;
4562 }
willy tarreaucd878942005-12-17 13:27:43 +01004563 else if (!strcmp(args[cur_arg], "postonly")) {
4564 curproxy->options |= PR_O_COOK_POST;
4565 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004566 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004567 Alert("parsing [%s:%d] : '%s' supports 'rewrite', 'insert', 'indirect', 'nocache' and 'postonly' options.\n",
4568 file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01004569 return -1;
4570 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004571 cur_arg++;
4572 }
4573 if ((curproxy->options & (PR_O_COOK_RW|PR_O_COOK_IND)) == (PR_O_COOK_RW|PR_O_COOK_IND)) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004574 Alert("parsing [%s:%d] : cookie 'rewrite' and 'indirect' mode are incompatible.\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01004575 file, linenum);
4576 return -1;
4577 }
4578 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004579 else if (!strcmp(args[0], "capture")) { /* name of a cookie to capture */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004580// if (curproxy == &defproxy) {
4581// Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4582// return -1;
4583// }
willy tarreaua41a8b42005-12-17 14:02:24 +01004584
willy tarreau8337c6b2005-12-17 13:41:01 +01004585 if (curproxy->capture_name != NULL) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01004586// Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n",
4587// file, linenum, args[0]);
4588// return 0;
4589 free(curproxy->capture_name);
willy tarreau8337c6b2005-12-17 13:41:01 +01004590 }
4591
4592 if (*(args[4]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004593 Alert("parsing [%s:%d] : '%s' expects 'cookie' <cookie_name> 'len' <len>.\n",
4594 file, linenum, args[0]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004595 return -1;
4596 }
4597 curproxy->capture_name = strdup(args[2]);
4598 curproxy->capture_namelen = strlen(curproxy->capture_name);
4599 curproxy->capture_len = atol(args[4]);
4600 if (curproxy->capture_len >= CAPTURE_LEN) {
4601 Warning("parsing [%s:%d] : truncating capture length to %d bytes.\n",
4602 file, linenum, CAPTURE_LEN - 1);
4603 curproxy->capture_len = CAPTURE_LEN - 1;
4604 }
4605 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004606 else if (!strcmp(args[0], "contimeout")) { /* connect timeout */
willy tarreaua41a8b42005-12-17 14:02:24 +01004607 if (curproxy->contimeout != defproxy.contimeout) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004608 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004609 return 0;
4610 }
4611 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004612 Alert("parsing [%s:%d] : '%s' expects an integer <time_in_ms> as argument.\n",
4613 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004614 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004615 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004616 curproxy->contimeout = atol(args[1]);
4617 }
4618 else if (!strcmp(args[0], "clitimeout")) { /* client timeout */
willy tarreaua41a8b42005-12-17 14:02:24 +01004619 if (curproxy->clitimeout != defproxy.clitimeout) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004620 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n",
4621 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004622 return 0;
4623 }
4624 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004625 Alert("parsing [%s:%d] : '%s' expects an integer <time_in_ms> as argument.\n",
4626 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004627 return -1;
4628 }
4629 curproxy->clitimeout = atol(args[1]);
4630 }
4631 else if (!strcmp(args[0], "srvtimeout")) { /* server timeout */
willy tarreaua41a8b42005-12-17 14:02:24 +01004632 if (curproxy->srvtimeout != defproxy.srvtimeout) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004633 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004634 return 0;
4635 }
4636 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004637 Alert("parsing [%s:%d] : '%s' expects an integer <time_in_ms> as argument.\n",
4638 file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01004639 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004640 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004641 curproxy->srvtimeout = atol(args[1]);
4642 }
4643 else if (!strcmp(args[0], "retries")) { /* connection retries */
4644 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004645 Alert("parsing [%s:%d] : '%s' expects an integer argument (dispatch counts for one).\n",
4646 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004647 return -1;
4648 }
4649 curproxy->conn_retries = atol(args[1]);
4650 }
4651 else if (!strcmp(args[0], "option")) {
4652 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004653 Alert("parsing [%s:%d] : '%s' expects an option name.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004654 return -1;
4655 }
4656 if (!strcmp(args[1], "redispatch"))
willy tarreau5cbea6f2005-12-17 12:48:26 +01004657 /* enable reconnections to dispatch */
4658 curproxy->options |= PR_O_REDISP;
willy tarreaua1598082005-12-17 13:08:06 +01004659#ifdef TPROXY
willy tarreau9fe663a2005-12-17 13:02:59 +01004660 else if (!strcmp(args[1], "transparent"))
willy tarreau5cbea6f2005-12-17 12:48:26 +01004661 /* enable transparent proxy connections */
4662 curproxy->options |= PR_O_TRANSP;
willy tarreau9fe663a2005-12-17 13:02:59 +01004663#endif
4664 else if (!strcmp(args[1], "keepalive"))
4665 /* enable keep-alive */
4666 curproxy->options |= PR_O_KEEPALIVE;
4667 else if (!strcmp(args[1], "forwardfor"))
4668 /* insert x-forwarded-for field */
4669 curproxy->options |= PR_O_FWDFOR;
willy tarreauc1cae632005-12-17 14:12:23 +01004670 else if (!strcmp(args[1], "httplog"))
willy tarreau9fe663a2005-12-17 13:02:59 +01004671 /* generate a complete HTTP log */
willy tarreaua1598082005-12-17 13:08:06 +01004672 curproxy->to_log |= LW_DATE | LW_CLIP | LW_SVID | LW_REQ | LW_PXID | LW_RESP;
willy tarreauc1cae632005-12-17 14:12:23 +01004673 else if (!strcmp(args[1], "tcplog"))
4674 /* generate a detailed TCP log */
4675 curproxy->to_log |= LW_DATE | LW_CLIP | LW_SVID | LW_PXID;
willy tarreaua1598082005-12-17 13:08:06 +01004676 else if (!strcmp(args[1], "dontlognull")) {
4677 /* don't log empty requests */
4678 curproxy->options |= PR_O_NULLNOLOG;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004679 }
willy tarreaubc4e1fb2005-12-17 13:32:07 +01004680 else if (!strcmp(args[1], "httpchk")) {
4681 /* use HTTP request to check servers' health */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004682 if (curproxy->check_req != NULL) {
4683 free(curproxy->check_req);
4684 }
willy tarreaubc4e1fb2005-12-17 13:32:07 +01004685 curproxy->options |= PR_O_HTTP_CHK;
willy tarreaueedaa9f2005-12-17 14:08:03 +01004686 if (!*args[2]) { /* no argument */
4687 curproxy->check_req = strdup(DEF_CHECK_REQ); /* default request */
4688 curproxy->check_len = strlen(DEF_CHECK_REQ);
4689 } else if (!*args[3]) { /* one argument : URI */
willy tarreau2f6ba652005-12-17 13:57:42 +01004690 int reqlen = strlen(args[2]) + strlen("OPTIONS / HTTP/1.0\r\n\r\n");
4691 curproxy->check_req = (char *)malloc(reqlen);
4692 curproxy->check_len = snprintf(curproxy->check_req, reqlen,
4693 "OPTIONS %s HTTP/1.0\r\n\r\n", args[2]); /* URI to use */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004694 } else { /* more arguments : METHOD URI [HTTP_VER] */
4695 int reqlen = strlen(args[2]) + strlen(args[3]) + 3 + strlen("\r\n\r\n");
4696 if (*args[4])
4697 reqlen += strlen(args[4]);
4698 else
4699 reqlen += strlen("HTTP/1.0");
4700
4701 curproxy->check_req = (char *)malloc(reqlen);
4702 curproxy->check_len = snprintf(curproxy->check_req, reqlen,
4703 "%s %s %s\r\n\r\n", args[2], args[3], *args[4]?args[4]:"HTTP/1.0");
willy tarreau2f6ba652005-12-17 13:57:42 +01004704 }
willy tarreaubc4e1fb2005-12-17 13:32:07 +01004705 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004706 else if (!strcmp(args[1], "persist")) {
4707 /* persist on using the server specified by the cookie, even when it's down */
4708 curproxy->options |= PR_O_PERSIST;
4709 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004710 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004711 Alert("parsing [%s:%d] : unknown option '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004712 return -1;
4713 }
4714 return 0;
4715 }
4716 else if (!strcmp(args[0], "redispatch") || !strcmp(args[0], "redisp")) {
4717 /* enable reconnections to dispatch */
4718 curproxy->options |= PR_O_REDISP;
4719 }
willy tarreaua1598082005-12-17 13:08:06 +01004720#ifdef TPROXY
willy tarreau9fe663a2005-12-17 13:02:59 +01004721 else if (!strcmp(args[0], "transparent")) {
4722 /* enable transparent proxy connections */
4723 curproxy->options |= PR_O_TRANSP;
4724 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004725#endif
willy tarreau9fe663a2005-12-17 13:02:59 +01004726 else if (!strcmp(args[0], "maxconn")) { /* maxconn */
4727 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004728 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004729 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004730 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004731 curproxy->maxconn = atol(args[1]);
4732 }
4733 else if (!strcmp(args[0], "grace")) { /* grace time (ms) */
4734 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004735 Alert("parsing [%s:%d] : '%s' expects a time in milliseconds.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004736 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004737 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004738 curproxy->grace = atol(args[1]);
4739 }
4740 else if (!strcmp(args[0], "dispatch")) { /* dispatch address */
willy tarreaua41a8b42005-12-17 14:02:24 +01004741 if (curproxy == &defproxy) {
4742 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4743 return -1;
4744 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004745 if (strchr(args[1], ':') == NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004746 Alert("parsing [%s:%d] : '%s' expects <addr:port> as argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004747 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004748 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004749 curproxy->dispatch_addr = *str2sa(args[1]);
4750 }
willy tarreau036e1ce2005-12-17 13:46:33 +01004751 else if (!strcmp(args[0], "balance")) { /* set balancing with optional algorithm */
willy tarreau9fe663a2005-12-17 13:02:59 +01004752 if (*(args[1])) {
4753 if (!strcmp(args[1], "roundrobin")) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01004754 curproxy->options |= PR_O_BALANCE_RR;
willy tarreau9fe663a2005-12-17 13:02:59 +01004755 }
4756 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004757 Alert("parsing [%s:%d] : '%s' only supports 'roundrobin' option.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004758 return -1;
4759 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004760 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004761 else /* if no option is set, use round-robin by default */
4762 curproxy->options |= PR_O_BALANCE_RR;
4763 }
4764 else if (!strcmp(args[0], "server")) { /* server address */
4765 int cur_arg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004766 char *rport;
4767 char *raddr;
4768 short realport;
4769 int do_check;
4770
4771 if (curproxy == &defproxy) {
4772 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4773 return -1;
4774 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004775
willy tarreaua41a8b42005-12-17 14:02:24 +01004776 if (!*args[2]) {
4777 Alert("parsing [%s:%d] : '%s' expects <name> and <addr>[:<port>] as arguments.\n",
willy tarreau036e1ce2005-12-17 13:46:33 +01004778 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004779 return -1;
4780 }
4781 if ((newsrv = (struct server *)calloc(1, sizeof(struct server))) == NULL) {
4782 Alert("parsing [%s:%d] : out of memory.\n", file, linenum);
4783 return -1;
4784 }
4785 newsrv->next = curproxy->srv;
4786 curproxy->srv = newsrv;
4787 newsrv->proxy = curproxy;
willy tarreaua41a8b42005-12-17 14:02:24 +01004788
4789 do_check = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01004790 newsrv->state = SRV_RUNNING; /* early server setup */
willy tarreaua41a8b42005-12-17 14:02:24 +01004791 newsrv->id = strdup(args[1]);
4792
4793 /* several ways to check the port component :
4794 * - IP => port=+0, relative
4795 * - IP: => port=+0, relative
4796 * - IP:N => port=N, absolute
4797 * - IP:+N => port=+N, relative
4798 * - IP:-N => port=-N, relative
4799 */
4800 raddr = strdup(args[2]);
4801 rport = strchr(raddr, ':');
4802 if (rport) {
4803 *rport++ = 0;
4804 realport = atol(rport);
4805 if (!isdigit((int)*rport))
4806 newsrv->state |= SRV_MAPPORTS;
4807 } else {
4808 realport = 0;
4809 newsrv->state |= SRV_MAPPORTS;
4810 }
4811
4812 newsrv->addr = *str2sa(raddr);
4813 newsrv->addr.sin_port = htons(realport);
4814 free(raddr);
4815
willy tarreau9fe663a2005-12-17 13:02:59 +01004816 newsrv->curfd = -1; /* no health-check in progress */
4817 newsrv->inter = DEF_CHKINTR;
4818 newsrv->rise = DEF_RISETIME;
4819 newsrv->fall = DEF_FALLTIME;
4820 newsrv->health = newsrv->rise; /* up, but will fall down at first failure */
4821 cur_arg = 3;
4822 while (*args[cur_arg]) {
4823 if (!strcmp(args[cur_arg], "cookie")) {
4824 newsrv->cookie = strdup(args[cur_arg + 1]);
4825 newsrv->cklen = strlen(args[cur_arg + 1]);
4826 cur_arg += 2;
willy tarreau0f7af912005-12-17 12:21:26 +01004827 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004828 else if (!strcmp(args[cur_arg], "rise")) {
4829 newsrv->rise = atol(args[cur_arg + 1]);
4830 newsrv->health = newsrv->rise;
4831 cur_arg += 2;
willy tarreau0f7af912005-12-17 12:21:26 +01004832 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004833 else if (!strcmp(args[cur_arg], "fall")) {
4834 newsrv->fall = atol(args[cur_arg + 1]);
4835 cur_arg += 2;
4836 }
4837 else if (!strcmp(args[cur_arg], "inter")) {
4838 newsrv->inter = atol(args[cur_arg + 1]);
4839 cur_arg += 2;
4840 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004841 else if (!strcmp(args[cur_arg], "port")) {
4842 newsrv->check_port = atol(args[cur_arg + 1]);
4843 cur_arg += 2;
4844 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004845 else if (!strcmp(args[cur_arg], "backup")) {
4846 newsrv->state |= SRV_BACKUP;
4847 cur_arg ++;
4848 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004849 else if (!strcmp(args[cur_arg], "check")) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004850 do_check = 1;
willy tarreau9fe663a2005-12-17 13:02:59 +01004851 cur_arg += 1;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004852 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004853 else {
willy tarreaua41a8b42005-12-17 14:02:24 +01004854 Alert("parsing [%s:%d] : server %s only supports options 'backup', 'cookie', 'check', 'inter', 'rise' and 'fall'.\n",
4855 file, linenum, newsrv->id);
4856 return -1;
4857 }
4858 }
4859
4860 if (do_check) {
4861 struct task *t;
4862
4863 if (!newsrv->check_port && !(newsrv->state & SRV_MAPPORTS))
4864 newsrv->check_port = realport; /* by default */
4865 if (!newsrv->check_port) {
4866 Alert("parsing [%s:%d] : server %s has neither service port nor check port. Check has been disabled.\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01004867 file, linenum, newsrv->id);
willy tarreau0f7af912005-12-17 12:21:26 +01004868 return -1;
4869 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004870
4871 if ((t = pool_alloc(task)) == NULL) {
4872 Alert("parsing [%s:%d] : out of memory.\n", file, linenum);
4873 return -1;
4874 }
4875
4876 t->next = t->prev = t->rqnext = NULL; /* task not in run queue yet */
4877 t->wq = LIST_HEAD(wait_queue); /* but already has a wait queue assigned */
4878 t->state = TASK_IDLE;
4879 t->process = process_chk;
4880 t->context = newsrv;
4881
4882 if (curproxy->state != PR_STDISABLED) {
4883 tv_delayfrom(&t->expire, &now, newsrv->inter); /* check this every ms */
4884 task_queue(t);
4885 task_wakeup(&rq, t);
4886 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004887 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004888
willy tarreau9fe663a2005-12-17 13:02:59 +01004889 curproxy->nbservers++;
4890 }
4891 else if (!strcmp(args[0], "log")) { /* syslog server address */
4892 struct sockaddr_in *sa;
4893 int facility;
4894
4895 if (*(args[1]) && *(args[2]) == 0 && !strcmp(args[1], "global")) {
4896 curproxy->logfac1 = global.logfac1;
4897 curproxy->logsrv1 = global.logsrv1;
willy tarreau8337c6b2005-12-17 13:41:01 +01004898 curproxy->loglev1 = global.loglev1;
willy tarreau9fe663a2005-12-17 13:02:59 +01004899 curproxy->logfac2 = global.logfac2;
4900 curproxy->logsrv2 = global.logsrv2;
willy tarreau8337c6b2005-12-17 13:41:01 +01004901 curproxy->loglev2 = global.loglev2;
willy tarreau9fe663a2005-12-17 13:02:59 +01004902 }
4903 else if (*(args[1]) && *(args[2])) {
willy tarreau8337c6b2005-12-17 13:41:01 +01004904 int level;
4905
willy tarreau0f7af912005-12-17 12:21:26 +01004906 for (facility = 0; facility < NB_LOG_FACILITIES; facility++)
4907 if (!strcmp(log_facilities[facility], args[2]))
4908 break;
willy tarreau9fe663a2005-12-17 13:02:59 +01004909
willy tarreau0f7af912005-12-17 12:21:26 +01004910 if (facility >= NB_LOG_FACILITIES) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004911 Alert("parsing [%s:%d] : unknown log facility '%s'\n", file, linenum, args[2]);
willy tarreau0f7af912005-12-17 12:21:26 +01004912 exit(1);
4913 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004914
willy tarreau8337c6b2005-12-17 13:41:01 +01004915 level = 7; /* max syslog level = debug */
4916 if (*(args[3])) {
4917 while (level >= 0 && strcmp(log_levels[level], args[3]))
4918 level--;
4919 if (level < 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004920 Alert("parsing [%s:%d] : unknown optional log level '%s'\n", file, linenum, args[3]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004921 exit(1);
4922 }
4923 }
4924
willy tarreau0f7af912005-12-17 12:21:26 +01004925 sa = str2sa(args[1]);
4926 if (!sa->sin_port)
4927 sa->sin_port = htons(SYSLOG_PORT);
willy tarreau9fe663a2005-12-17 13:02:59 +01004928
willy tarreau0f7af912005-12-17 12:21:26 +01004929 if (curproxy->logfac1 == -1) {
4930 curproxy->logsrv1 = *sa;
4931 curproxy->logfac1 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004932 curproxy->loglev1 = level;
willy tarreau0f7af912005-12-17 12:21:26 +01004933 }
4934 else if (curproxy->logfac2 == -1) {
4935 curproxy->logsrv2 = *sa;
4936 curproxy->logfac2 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004937 curproxy->loglev2 = level;
willy tarreau0f7af912005-12-17 12:21:26 +01004938 }
4939 else {
4940 Alert("parsing [%s:%d] : too many syslog servers\n", file, linenum);
willy tarreau9fe663a2005-12-17 13:02:59 +01004941 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004942 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004943 }
4944 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004945 Alert("parsing [%s:%d] : 'log' expects either <address[:port]> and <facility> or 'global' as arguments.\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01004946 file, linenum);
4947 return -1;
4948 }
4949 }
willy tarreaua1598082005-12-17 13:08:06 +01004950 else if (!strcmp(args[0], "source")) { /* address to which we bind when connecting */
willy tarreaua41a8b42005-12-17 14:02:24 +01004951 if (!*args[1]) {
4952 Alert("parsing [%s:%d] : '%s' expects <addr>[:<port>] as argument.\n",
willy tarreau036e1ce2005-12-17 13:46:33 +01004953 file, linenum, "source");
willy tarreaua1598082005-12-17 13:08:06 +01004954 return -1;
4955 }
4956
4957 curproxy->source_addr = *str2sa(args[1]);
4958 curproxy->options |= PR_O_BIND_SRC;
4959 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004960 else if (!strcmp(args[0], "cliexp") || !strcmp(args[0], "reqrep")) { /* replace request header from a regex */
4961 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004962 if (curproxy == &defproxy) {
4963 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4964 return -1;
4965 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004966
4967 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004968 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
4969 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004970 return -1;
4971 }
4972
4973 preg = calloc(1, sizeof(regex_t));
4974 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004975 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004976 return -1;
4977 }
4978
4979 chain_regex(&curproxy->req_exp, preg, ACT_REPLACE, strdup(args[2]));
4980 }
4981 else if (!strcmp(args[0], "reqdel")) { /* delete request header from a regex */
4982 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004983 if (curproxy == &defproxy) {
4984 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4985 return -1;
4986 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004987
4988 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004989 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004990 return -1;
4991 }
4992
4993 preg = calloc(1, sizeof(regex_t));
4994 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004995 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004996 return -1;
4997 }
4998
4999 chain_regex(&curproxy->req_exp, preg, ACT_REMOVE, NULL);
5000 }
5001 else if (!strcmp(args[0], "reqdeny")) { /* deny a request if a header matches this regex */
5002 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005003 if (curproxy == &defproxy) {
5004 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5005 return -1;
5006 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005007
5008 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005009 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005010 return -1;
5011 }
5012
5013 preg = calloc(1, sizeof(regex_t));
5014 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005015 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005016 return -1;
5017 }
5018
5019 chain_regex(&curproxy->req_exp, preg, ACT_DENY, NULL);
5020 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005021 else if (!strcmp(args[0], "reqpass")) { /* pass this header without allowing or denying the request */
5022 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005023 if (curproxy == &defproxy) {
5024 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5025 return -1;
5026 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005027
5028 if (*(args[1]) == 0) {
5029 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
5030 return -1;
5031 }
5032
5033 preg = calloc(1, sizeof(regex_t));
5034 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
5035 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
5036 return -1;
5037 }
5038
5039 chain_regex(&curproxy->req_exp, preg, ACT_PASS, NULL);
5040 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005041 else if (!strcmp(args[0], "reqallow")) { /* allow a request if a header matches this regex */
5042 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005043 if (curproxy == &defproxy) {
5044 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5045 return -1;
5046 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005047
5048 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005049 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005050 return -1;
5051 }
5052
5053 preg = calloc(1, sizeof(regex_t));
5054 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005055 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005056 return -1;
5057 }
5058
5059 chain_regex(&curproxy->req_exp, preg, ACT_ALLOW, NULL);
5060 }
5061 else if (!strcmp(args[0], "reqirep")) { /* replace request header from a regex, ignoring case */
5062 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005063 if (curproxy == &defproxy) {
5064 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5065 return -1;
5066 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005067
5068 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005069 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
5070 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005071 return -1;
5072 }
5073
5074 preg = calloc(1, sizeof(regex_t));
5075 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005076 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005077 return -1;
5078 }
5079
5080 chain_regex(&curproxy->req_exp, preg, ACT_REPLACE, strdup(args[2]));
5081 }
5082 else if (!strcmp(args[0], "reqidel")) { /* delete request header from a regex ignoring case */
5083 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005084 if (curproxy == &defproxy) {
5085 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5086 return -1;
5087 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005088
5089 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005090 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005091 return -1;
5092 }
5093
5094 preg = calloc(1, sizeof(regex_t));
5095 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005096 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005097 return -1;
5098 }
5099
5100 chain_regex(&curproxy->req_exp, preg, ACT_REMOVE, NULL);
5101 }
5102 else if (!strcmp(args[0], "reqideny")) { /* deny a request if a header matches this regex ignoring case */
5103 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005104 if (curproxy == &defproxy) {
5105 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5106 return -1;
5107 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005108
5109 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005110 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005111 return -1;
5112 }
5113
5114 preg = calloc(1, sizeof(regex_t));
5115 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005116 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005117 return -1;
5118 }
5119
5120 chain_regex(&curproxy->req_exp, preg, ACT_DENY, NULL);
5121 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005122 else if (!strcmp(args[0], "reqipass")) { /* pass this header without allowing or denying the request */
5123 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005124 if (curproxy == &defproxy) {
5125 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5126 return -1;
5127 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005128
5129 if (*(args[1]) == 0) {
5130 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
5131 return -1;
5132 }
5133
5134 preg = calloc(1, sizeof(regex_t));
5135 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
5136 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
5137 return -1;
5138 }
5139
5140 chain_regex(&curproxy->req_exp, preg, ACT_PASS, NULL);
5141 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005142 else if (!strcmp(args[0], "reqiallow")) { /* allow a request if a header matches this regex ignoring case */
5143 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005144 if (curproxy == &defproxy) {
5145 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5146 return -1;
5147 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005148
5149 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005150 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005151 return -1;
5152 }
5153
5154 preg = calloc(1, sizeof(regex_t));
5155 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005156 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005157 return -1;
5158 }
5159
5160 chain_regex(&curproxy->req_exp, preg, ACT_ALLOW, NULL);
5161 }
5162 else if (!strcmp(args[0], "reqadd")) { /* add request header */
willy tarreaua41a8b42005-12-17 14:02:24 +01005163 if (curproxy == &defproxy) {
5164 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5165 return -1;
5166 }
5167
willy tarreau9fe663a2005-12-17 13:02:59 +01005168 if (curproxy->nb_reqadd >= MAX_NEWHDR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005169 Alert("parsing [%s:%d] : too many '%s'. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005170 return 0;
5171 }
5172
5173 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005174 Alert("parsing [%s:%d] : '%s' expects <header> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005175 return -1;
5176 }
5177
5178 curproxy->req_add[curproxy->nb_reqadd++] = strdup(args[1]);
willy tarreau0f7af912005-12-17 12:21:26 +01005179 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005180 else if (!strcmp(args[0], "srvexp") || !strcmp(args[0], "rsprep")) { /* replace response header from a regex */
willy tarreau0f7af912005-12-17 12:21:26 +01005181 regex_t *preg;
willy tarreau0f7af912005-12-17 12:21:26 +01005182
5183 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005184 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
5185 file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01005186 return -1;
5187 }
5188
5189 preg = calloc(1, sizeof(regex_t));
5190 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005191 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau0f7af912005-12-17 12:21:26 +01005192 return -1;
5193 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005194
5195 chain_regex(&curproxy->rsp_exp, preg, ACT_REPLACE, strdup(args[2]));
5196 }
5197 else if (!strcmp(args[0], "rspdel")) { /* delete response header from a regex */
5198 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005199 if (curproxy == &defproxy) {
5200 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5201 return -1;
5202 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005203
5204 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005205 Alert("parsing [%s:%d] : '%s' expects <search> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005206 return -1;
5207 }
willy tarreaue39cd132005-12-17 13:00:18 +01005208
willy tarreau9fe663a2005-12-17 13:02:59 +01005209 preg = calloc(1, sizeof(regex_t));
5210 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005211 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005212 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01005213 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005214
5215 chain_regex(&curproxy->rsp_exp, preg, ACT_REMOVE, strdup(args[2]));
5216 }
5217 else if (!strcmp(args[0], "rspirep")) { /* replace response header from a regex ignoring case */
willy tarreaua41a8b42005-12-17 14:02:24 +01005218 regex_t *preg;
5219 if (curproxy == &defproxy) {
5220 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5221 return -1;
5222 }
willy tarreaue39cd132005-12-17 13:00:18 +01005223
willy tarreaua41a8b42005-12-17 14:02:24 +01005224 if (*(args[1]) == 0 || *(args[2]) == 0) {
5225 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
5226 file, linenum, args[0]);
5227 return -1;
5228 }
willy tarreaue39cd132005-12-17 13:00:18 +01005229
willy tarreaua41a8b42005-12-17 14:02:24 +01005230 preg = calloc(1, sizeof(regex_t));
5231 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
5232 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
5233 return -1;
willy tarreau9fe663a2005-12-17 13:02:59 +01005234 }
willy tarreaua41a8b42005-12-17 14:02:24 +01005235
5236 chain_regex(&curproxy->rsp_exp, preg, ACT_REPLACE, strdup(args[2]));
5237 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005238 else if (!strcmp(args[0], "rspidel")) { /* delete response header from a regex ignoring case */
5239 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005240 if (curproxy == &defproxy) {
5241 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5242 return -1;
5243 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005244
5245 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005246 Alert("parsing [%s:%d] : '%s' expects <search> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005247 return -1;
5248 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005249
willy tarreau9fe663a2005-12-17 13:02:59 +01005250 preg = calloc(1, sizeof(regex_t));
5251 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005252 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005253 return -1;
willy tarreaue39cd132005-12-17 13:00:18 +01005254 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005255
5256 chain_regex(&curproxy->rsp_exp, preg, ACT_REMOVE, strdup(args[2]));
5257 }
5258 else if (!strcmp(args[0], "rspadd")) { /* add response header */
willy tarreaua41a8b42005-12-17 14:02:24 +01005259 if (curproxy == &defproxy) {
5260 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5261 return -1;
5262 }
5263
willy tarreau9fe663a2005-12-17 13:02:59 +01005264 if (curproxy->nb_rspadd >= MAX_NEWHDR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005265 Alert("parsing [%s:%d] : too many '%s'. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005266 return 0;
5267 }
5268
5269 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005270 Alert("parsing [%s:%d] : '%s' expects <header> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005271 return -1;
5272 }
5273
5274 curproxy->rsp_add[curproxy->nb_rspadd++] = strdup(args[1]);
5275 }
willy tarreau8337c6b2005-12-17 13:41:01 +01005276 else if (!strcmp(args[0], "errorloc")) { /* error location */
5277 int errnum;
5278 char *err;
5279
willy tarreaueedaa9f2005-12-17 14:08:03 +01005280 // if (curproxy == &defproxy) {
5281 // Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5282 // return -1;
5283 // }
willy tarreaua41a8b42005-12-17 14:02:24 +01005284
willy tarreau8337c6b2005-12-17 13:41:01 +01005285 if (*(args[2]) == 0) {
5286 Alert("parsing [%s:%d] : <errorloc> expects <error> and <url> as arguments.\n", file, linenum);
5287 return -1;
5288 }
5289
5290 errnum = atol(args[1]);
5291 err = malloc(strlen(HTTP_302) + strlen(args[2]) + 5);
5292 sprintf(err, "%s%s\r\n\r\n", HTTP_302, args[2]);
5293
5294 if (errnum == 400) {
5295 if (curproxy->errmsg.msg400) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005296 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005297 free(curproxy->errmsg.msg400);
5298 }
5299 curproxy->errmsg.msg400 = err;
5300 curproxy->errmsg.len400 = strlen(err);
5301 }
5302 else if (errnum == 403) {
5303 if (curproxy->errmsg.msg403) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005304 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005305 free(curproxy->errmsg.msg403);
5306 }
5307 curproxy->errmsg.msg403 = err;
5308 curproxy->errmsg.len403 = strlen(err);
5309 }
5310 else if (errnum == 408) {
5311 if (curproxy->errmsg.msg408) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005312 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005313 free(curproxy->errmsg.msg408);
5314 }
5315 curproxy->errmsg.msg408 = err;
5316 curproxy->errmsg.len408 = strlen(err);
5317 }
5318 else if (errnum == 500) {
5319 if (curproxy->errmsg.msg500) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005320 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005321 free(curproxy->errmsg.msg500);
5322 }
5323 curproxy->errmsg.msg500 = err;
5324 curproxy->errmsg.len500 = strlen(err);
5325 }
5326 else if (errnum == 502) {
5327 if (curproxy->errmsg.msg502) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005328 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005329 free(curproxy->errmsg.msg502);
5330 }
5331 curproxy->errmsg.msg502 = err;
5332 curproxy->errmsg.len502 = strlen(err);
5333 }
5334 else if (errnum == 503) {
5335 if (curproxy->errmsg.msg503) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005336 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005337 free(curproxy->errmsg.msg503);
5338 }
5339 curproxy->errmsg.msg503 = err;
5340 curproxy->errmsg.len503 = strlen(err);
5341 }
5342 else if (errnum == 504) {
5343 if (curproxy->errmsg.msg504) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005344 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005345 free(curproxy->errmsg.msg504);
5346 }
5347 curproxy->errmsg.msg504 = err;
5348 curproxy->errmsg.len504 = strlen(err);
5349 }
5350 else {
5351 Warning("parsing [%s:%d] : error %d relocation will be ignored.\n", file, linenum, errnum);
5352 free(err);
5353 }
5354 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005355 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01005356 Alert("parsing [%s:%d] : unknown keyword '%s' in '%s' section\n", file, linenum, args[0], "listen");
willy tarreau9fe663a2005-12-17 13:02:59 +01005357 return -1;
5358 }
5359 return 0;
5360}
willy tarreaue39cd132005-12-17 13:00:18 +01005361
willy tarreau5cbea6f2005-12-17 12:48:26 +01005362
willy tarreau9fe663a2005-12-17 13:02:59 +01005363/*
5364 * This function reads and parses the configuration file given in the argument.
5365 * returns 0 if OK, -1 if error.
5366 */
5367int readcfgfile(char *file) {
5368 char thisline[256];
5369 char *line;
5370 FILE *f;
5371 int linenum = 0;
5372 char *end;
5373 char *args[MAX_LINE_ARGS];
5374 int arg;
5375 int cfgerr = 0;
5376 int confsect = CFG_NONE;
willy tarreaue39cd132005-12-17 13:00:18 +01005377
willy tarreau9fe663a2005-12-17 13:02:59 +01005378 struct proxy *curproxy = NULL;
5379 struct server *newsrv = NULL;
willy tarreaue39cd132005-12-17 13:00:18 +01005380
willy tarreau9fe663a2005-12-17 13:02:59 +01005381 if ((f=fopen(file,"r")) == NULL)
5382 return -1;
willy tarreaue39cd132005-12-17 13:00:18 +01005383
willy tarreaueedaa9f2005-12-17 14:08:03 +01005384 init_default_instance();
5385
willy tarreau9fe663a2005-12-17 13:02:59 +01005386 while (fgets(line = thisline, sizeof(thisline), f) != NULL) {
5387 linenum++;
willy tarreau5cbea6f2005-12-17 12:48:26 +01005388
willy tarreau9fe663a2005-12-17 13:02:59 +01005389 end = line + strlen(line);
willy tarreau5cbea6f2005-12-17 12:48:26 +01005390
willy tarreau9fe663a2005-12-17 13:02:59 +01005391 /* skip leading spaces */
willy tarreauc29948c2005-12-17 13:10:27 +01005392 while (isspace((int)*line))
willy tarreau9fe663a2005-12-17 13:02:59 +01005393 line++;
5394
5395 arg = 0;
5396 args[arg] = line;
willy tarreau0f7af912005-12-17 12:21:26 +01005397
willy tarreau9fe663a2005-12-17 13:02:59 +01005398 while (*line && arg < MAX_LINE_ARGS) {
5399 /* first, we'll replace \\, \<space>, \#, \r, \n, \t, \xXX with their
5400 * C equivalent value. Other combinations left unchanged (eg: \1).
5401 */
5402 if (*line == '\\') {
5403 int skip = 0;
5404 if (line[1] == ' ' || line[1] == '\\' || line[1] == '#') {
5405 *line = line[1];
5406 skip = 1;
5407 }
5408 else if (line[1] == 'r') {
5409 *line = '\r';
5410 skip = 1;
5411 }
5412 else if (line[1] == 'n') {
5413 *line = '\n';
5414 skip = 1;
5415 }
5416 else if (line[1] == 't') {
5417 *line = '\t';
5418 skip = 1;
5419 }
5420 else if (line[1] == 'x' && (line + 3 < end )) {
5421 unsigned char hex1, hex2;
5422 hex1 = toupper(line[2]) - '0'; hex2 = toupper(line[3]) - '0';
5423 if (hex1 > 9) hex1 -= 'A' - '9' - 1;
5424 if (hex2 > 9) hex2 -= 'A' - '9' - 1;
5425 *line = (hex1<<4) + hex2;
5426 skip = 3;
5427 }
5428 if (skip) {
5429 memmove(line + 1, line + 1 + skip, end - (line + skip + 1));
5430 end -= skip;
5431 }
5432 line++;
willy tarreau0f7af912005-12-17 12:21:26 +01005433 }
willy tarreaua1598082005-12-17 13:08:06 +01005434 else if (*line == '#' || *line == '\n' || *line == '\r') {
5435 /* end of string, end of loop */
5436 *line = 0;
5437 break;
5438 }
willy tarreauc29948c2005-12-17 13:10:27 +01005439 else if (isspace((int)*line)) {
willy tarreau9fe663a2005-12-17 13:02:59 +01005440 /* a non-escaped space is an argument separator */
willy tarreaua1598082005-12-17 13:08:06 +01005441 *line++ = 0;
willy tarreauc29948c2005-12-17 13:10:27 +01005442 while (isspace((int)*line))
willy tarreaua1598082005-12-17 13:08:06 +01005443 line++;
5444 args[++arg] = line;
5445 }
5446 else {
5447 line++;
willy tarreau0f7af912005-12-17 12:21:26 +01005448 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005449 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005450
willy tarreau9fe663a2005-12-17 13:02:59 +01005451 /* empty line */
5452 if (!**args)
5453 continue;
willy tarreaue39cd132005-12-17 13:00:18 +01005454
willy tarreau9fe663a2005-12-17 13:02:59 +01005455 /* zero out remaining args */
5456 while (++arg < MAX_LINE_ARGS) {
5457 args[arg] = line;
willy tarreau5cbea6f2005-12-17 12:48:26 +01005458 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005459
willy tarreaua41a8b42005-12-17 14:02:24 +01005460 if (!strcmp(args[0], "listen") || !strcmp(args[0], "defaults")) /* new proxy */
willy tarreau9fe663a2005-12-17 13:02:59 +01005461 confsect = CFG_LISTEN;
5462 else if (!strcmp(args[0], "global")) /* global config */
5463 confsect = CFG_GLOBAL;
5464 /* else it's a section keyword */
willy tarreau5cbea6f2005-12-17 12:48:26 +01005465
willy tarreau9fe663a2005-12-17 13:02:59 +01005466 switch (confsect) {
5467 case CFG_LISTEN:
5468 if (cfg_parse_listen(file, linenum, args) < 0)
5469 return -1;
5470 break;
5471 case CFG_GLOBAL:
5472 if (cfg_parse_global(file, linenum, args) < 0)
5473 return -1;
5474 break;
5475 default:
willy tarreau036e1ce2005-12-17 13:46:33 +01005476 Alert("parsing [%s:%d] : unknown keyword '%s' out of section.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005477 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01005478 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005479
5480
willy tarreau0f7af912005-12-17 12:21:26 +01005481 }
5482 fclose(f);
5483
5484 /*
5485 * Now, check for the integrity of all that we have collected.
5486 */
5487
5488 if ((curproxy = proxy) == NULL) {
5489 Alert("parsing %s : no <listen> line. Nothing to do !\n",
5490 file);
5491 return -1;
5492 }
5493
5494 while (curproxy != NULL) {
willy tarreauef900ab2005-12-17 12:52:52 +01005495 if (curproxy->state == PR_STDISABLED) {
5496 curproxy = curproxy->next;
5497 continue;
5498 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005499 if ((curproxy->mode != PR_MODE_HEALTH) &&
5500 !(curproxy->options & (PR_O_TRANSP | PR_O_BALANCE)) &&
willy tarreaua1598082005-12-17 13:08:06 +01005501 (*(int *)&curproxy->dispatch_addr.sin_addr == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01005502 Alert("parsing %s : listener %s has no dispatch address and is not in transparent or balance mode.\n",
5503 file, curproxy->id);
5504 cfgerr++;
5505 }
5506 else if ((curproxy->mode != PR_MODE_HEALTH) && (curproxy->options & PR_O_BALANCE)) {
5507 if (curproxy->options & PR_O_TRANSP) {
5508 Alert("parsing %s : listener %s cannot use both transparent and balance mode.\n",
5509 file, curproxy->id);
5510 cfgerr++;
5511 }
5512 else if (curproxy->srv == NULL) {
5513 Alert("parsing %s : listener %s needs at least 1 server in balance mode.\n",
5514 file, curproxy->id);
5515 cfgerr++;
5516 }
willy tarreaua1598082005-12-17 13:08:06 +01005517 else if (*(int *)&curproxy->dispatch_addr.sin_addr != 0) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01005518 Warning("parsing %s : dispatch address of listener %s will be ignored in balance mode.\n",
5519 file, curproxy->id);
5520 }
5521 }
5522 else if (curproxy->mode == PR_MODE_TCP || curproxy->mode == PR_MODE_HEALTH) { /* TCP PROXY or HEALTH CHECK */
willy tarreau0f7af912005-12-17 12:21:26 +01005523 if (curproxy->cookie_name != NULL) {
5524 Warning("parsing %s : cookie will be ignored for listener %s.\n",
5525 file, curproxy->id);
5526 }
5527 if ((newsrv = curproxy->srv) != NULL) {
5528 Warning("parsing %s : servers will be ignored for listener %s.\n",
5529 file, curproxy->id);
5530 }
willy tarreaue39cd132005-12-17 13:00:18 +01005531 if (curproxy->rsp_exp != NULL) {
willy tarreau0f7af912005-12-17 12:21:26 +01005532 Warning("parsing %s : server regular expressions will be ignored for listener %s.\n",
5533 file, curproxy->id);
5534 }
willy tarreaue39cd132005-12-17 13:00:18 +01005535 if (curproxy->req_exp != NULL) {
willy tarreau0f7af912005-12-17 12:21:26 +01005536 Warning("parsing %s : client regular expressions will be ignored for listener %s.\n",
5537 file, curproxy->id);
5538 }
5539 }
5540 else if (curproxy->mode == PR_MODE_HTTP) { /* HTTP PROXY */
5541 if ((curproxy->cookie_name != NULL) && ((newsrv = curproxy->srv) == NULL)) {
5542 Alert("parsing %s : HTTP proxy %s has a cookie but no server list !\n",
5543 file, curproxy->id);
5544 cfgerr++;
5545 }
5546 else {
5547 while (newsrv != NULL) {
5548 /* nothing to check for now */
5549 newsrv = newsrv->next;
5550 }
5551 }
5552 }
willy tarreau8337c6b2005-12-17 13:41:01 +01005553 if (curproxy->errmsg.msg400 == NULL) {
5554 curproxy->errmsg.msg400 = (char *)HTTP_400;
5555 curproxy->errmsg.len400 = strlen(HTTP_400);
5556 }
5557 if (curproxy->errmsg.msg403 == NULL) {
5558 curproxy->errmsg.msg403 = (char *)HTTP_403;
5559 curproxy->errmsg.len403 = strlen(HTTP_403);
5560 }
5561 if (curproxy->errmsg.msg408 == NULL) {
5562 curproxy->errmsg.msg408 = (char *)HTTP_408;
5563 curproxy->errmsg.len408 = strlen(HTTP_408);
5564 }
5565 if (curproxy->errmsg.msg500 == NULL) {
5566 curproxy->errmsg.msg500 = (char *)HTTP_500;
5567 curproxy->errmsg.len500 = strlen(HTTP_500);
5568 }
5569 if (curproxy->errmsg.msg502 == NULL) {
5570 curproxy->errmsg.msg502 = (char *)HTTP_502;
5571 curproxy->errmsg.len502 = strlen(HTTP_502);
5572 }
5573 if (curproxy->errmsg.msg503 == NULL) {
5574 curproxy->errmsg.msg503 = (char *)HTTP_503;
5575 curproxy->errmsg.len503 = strlen(HTTP_503);
5576 }
5577 if (curproxy->errmsg.msg504 == NULL) {
5578 curproxy->errmsg.msg504 = (char *)HTTP_504;
5579 curproxy->errmsg.len504 = strlen(HTTP_504);
5580 }
willy tarreau0f7af912005-12-17 12:21:26 +01005581 curproxy = curproxy->next;
5582 }
5583 if (cfgerr > 0) {
5584 Alert("Errors found in configuration file, aborting.\n");
5585 return -1;
5586 }
5587 else
5588 return 0;
5589}
5590
5591
5592/*
5593 * This function initializes all the necessary variables. It only returns
5594 * if everything is OK. If something fails, it exits.
5595 */
5596void init(int argc, char **argv) {
5597 int i;
willy tarreau9fe663a2005-12-17 13:02:59 +01005598 int arg_mode = 0; /* MODE_DEBUG, ... */
willy tarreau0f7af912005-12-17 12:21:26 +01005599 char *old_argv = *argv;
5600 char *tmp;
willy tarreau9fe663a2005-12-17 13:02:59 +01005601 int cfg_maxconn = 0; /* # of simultaneous connections, (-n) */
willy tarreau0f7af912005-12-17 12:21:26 +01005602
5603 if (1<<INTBITS != sizeof(int)*8) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01005604 qfprintf(stderr,
willy tarreau0f7af912005-12-17 12:21:26 +01005605 "Error: wrong architecture. Recompile so that sizeof(int)=%d\n",
5606 sizeof(int)*8);
5607 exit(1);
5608 }
5609
5610 pid = getpid();
5611 progname = *argv;
5612 while ((tmp = strchr(progname, '/')) != NULL)
5613 progname = tmp + 1;
5614
5615 argc--; argv++;
5616 while (argc > 0) {
5617 char *flag;
5618
5619 if (**argv == '-') {
5620 flag = *argv+1;
5621
5622 /* 1 arg */
5623 if (*flag == 'v') {
5624 display_version();
5625 exit(0);
5626 }
5627 else if (*flag == 'd')
willy tarreau9fe663a2005-12-17 13:02:59 +01005628 arg_mode |= MODE_DEBUG;
willy tarreau0f7af912005-12-17 12:21:26 +01005629 else if (*flag == 'D')
willy tarreau9fe663a2005-12-17 13:02:59 +01005630 arg_mode |= MODE_DAEMON | MODE_QUIET;
willy tarreau5cbea6f2005-12-17 12:48:26 +01005631 else if (*flag == 'q')
willy tarreau9fe663a2005-12-17 13:02:59 +01005632 arg_mode |= MODE_QUIET;
willy tarreau0f7af912005-12-17 12:21:26 +01005633#if STATTIME > 0
5634 else if (*flag == 's')
willy tarreau9fe663a2005-12-17 13:02:59 +01005635 arg_mode |= MODE_STATS;
willy tarreau0f7af912005-12-17 12:21:26 +01005636 else if (*flag == 'l')
willy tarreau9fe663a2005-12-17 13:02:59 +01005637 arg_mode |= MODE_LOG;
willy tarreau0f7af912005-12-17 12:21:26 +01005638#endif
5639 else { /* >=2 args */
5640 argv++; argc--;
5641 if (argc == 0)
5642 usage(old_argv);
5643
5644 switch (*flag) {
5645 case 'n' : cfg_maxconn = atol(*argv); break;
5646 case 'N' : cfg_maxpconn = atol(*argv); break;
5647 case 'f' : cfg_cfgfile = *argv; break;
5648 default: usage(old_argv);
5649 }
5650 }
5651 }
5652 else
5653 usage(old_argv);
5654 argv++; argc--;
5655 }
5656
willy tarreau0f7af912005-12-17 12:21:26 +01005657 if (!cfg_cfgfile)
5658 usage(old_argv);
5659
5660 gethostname(hostname, MAX_HOSTNAME_LEN);
5661
5662 if (readcfgfile(cfg_cfgfile) < 0) {
5663 Alert("Error reading configuration file : %s\n", cfg_cfgfile);
5664 exit(1);
5665 }
5666
willy tarreau9fe663a2005-12-17 13:02:59 +01005667 if (cfg_maxconn > 0)
5668 global.maxconn = cfg_maxconn;
5669
5670 if (global.maxconn == 0)
5671 global.maxconn = DEFAULT_MAXCONN;
5672
5673 global.maxsock = global.maxconn * 2; /* each connection needs two sockets */
5674
5675 if (arg_mode & MODE_DEBUG) {
5676 /* command line debug mode inhibits configuration mode */
5677 global.mode &= ~(MODE_DAEMON | MODE_QUIET);
5678 }
willy tarreau750a4722005-12-17 13:21:24 +01005679 global.mode |= (arg_mode & (MODE_DAEMON | MODE_QUIET | MODE_DEBUG | MODE_STATS | MODE_LOG));
willy tarreau9fe663a2005-12-17 13:02:59 +01005680
5681 if ((global.mode & MODE_DEBUG) && (global.mode & (MODE_DAEMON | MODE_QUIET))) {
5682 Warning("<debug> mode incompatible with <quiet> and <daemon>. Keeping <debug> only.\n");
5683 global.mode &= ~(MODE_DAEMON | MODE_QUIET);
5684 }
5685
5686 if ((global.nbproc > 1) && !(global.mode & MODE_DAEMON)) {
5687 Warning("<nbproc> is only meaningful in daemon mode. Setting limit to 1 process.\n");
5688 global.nbproc = 1;
5689 }
5690
5691 if (global.nbproc < 1)
5692 global.nbproc = 1;
5693
willy tarreau0f7af912005-12-17 12:21:26 +01005694 ReadEvent = (fd_set *)calloc(1,
5695 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005696 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005697 WriteEvent = (fd_set *)calloc(1,
5698 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005699 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005700 StaticReadEvent = (fd_set *)calloc(1,
5701 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005702 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005703 StaticWriteEvent = (fd_set *)calloc(1,
5704 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005705 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005706
5707 fdtab = (struct fdtab *)calloc(1,
willy tarreau9fe663a2005-12-17 13:02:59 +01005708 sizeof(struct fdtab) * (global.maxsock));
5709 for (i = 0; i < global.maxsock; i++) {
willy tarreau0f7af912005-12-17 12:21:26 +01005710 fdtab[i].state = FD_STCLOSE;
5711 }
5712}
5713
5714/*
5715 * this function starts all the proxies. It returns 0 if OK, -1 if not.
5716 */
5717int start_proxies() {
5718 struct proxy *curproxy;
willy tarreaua41a8b42005-12-17 14:02:24 +01005719 struct listener *listener;
willy tarreau0f7af912005-12-17 12:21:26 +01005720 int one = 1;
5721 int fd;
5722
5723 for (curproxy = proxy; curproxy != NULL; curproxy = curproxy->next) {
willy tarreau0f7af912005-12-17 12:21:26 +01005724 if (curproxy->state == PR_STDISABLED)
5725 continue;
5726
willy tarreaua41a8b42005-12-17 14:02:24 +01005727 for (listener = curproxy->listen; listener != NULL; listener = listener->next) {
5728 if ((fd = listener->fd =
5729 socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1) {
5730 Alert("cannot create listening socket for proxy %s. Aborting.\n",
5731 curproxy->id);
5732 return -1;
5733 }
willy tarreau0f7af912005-12-17 12:21:26 +01005734
willy tarreaua41a8b42005-12-17 14:02:24 +01005735 if (fd >= global.maxsock) {
5736 Alert("socket(): not enough free sockets for proxy %s. Raise -n argument. Aborting.\n",
5737 curproxy->id);
5738 close(fd);
5739 return -1;
5740 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005741
willy tarreaua41a8b42005-12-17 14:02:24 +01005742 if ((fcntl(fd, F_SETFL, O_NONBLOCK) == -1) ||
5743 (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY,
5744 (char *) &one, sizeof(one)) == -1)) {
5745 Alert("cannot make socket non-blocking for proxy %s. Aborting.\n",
5746 curproxy->id);
5747 close(fd);
5748 return -1;
5749 }
willy tarreau0f7af912005-12-17 12:21:26 +01005750
willy tarreaua41a8b42005-12-17 14:02:24 +01005751 if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *) &one, sizeof(one)) == -1) {
5752 Alert("cannot do so_reuseaddr for proxy %s. Continuing.\n",
5753 curproxy->id);
5754 }
willy tarreau0f7af912005-12-17 12:21:26 +01005755
willy tarreaua41a8b42005-12-17 14:02:24 +01005756 if (bind(fd,
5757 (struct sockaddr *)&listener->addr,
5758 sizeof(listener->addr)) == -1) {
5759 Alert("cannot bind socket for proxy %s. Aborting.\n",
5760 curproxy->id);
5761 close(fd);
5762 return -1;
5763 }
willy tarreau0f7af912005-12-17 12:21:26 +01005764
willy tarreaua41a8b42005-12-17 14:02:24 +01005765 if (listen(fd, curproxy->maxconn) == -1) {
5766 Alert("cannot listen to socket for proxy %s. Aborting.\n",
5767 curproxy->id);
5768 close(fd);
5769 return -1;
5770 }
willy tarreau0f7af912005-12-17 12:21:26 +01005771
willy tarreaua41a8b42005-12-17 14:02:24 +01005772 /* the function for the accept() event */
5773 fdtab[fd].read = &event_accept;
5774 fdtab[fd].write = NULL; /* never called */
5775 fdtab[fd].owner = (struct task *)curproxy; /* reference the proxy instead of a task */
5776 curproxy->state = PR_STRUN;
5777 fdtab[fd].state = FD_STLISTEN;
5778 FD_SET(fd, StaticReadEvent);
5779 fd_insert(fd);
5780 listeners++;
5781 }
willy tarreaua1598082005-12-17 13:08:06 +01005782 send_log(curproxy, LOG_NOTICE, "Proxy %s started.\n", curproxy->id);
willy tarreau0f7af912005-12-17 12:21:26 +01005783 }
5784 return 0;
5785}
5786
5787
5788int main(int argc, char **argv) {
5789 init(argc, argv);
5790
willy tarreau9fe663a2005-12-17 13:02:59 +01005791 if (global.mode & MODE_QUIET) {
willy tarreau0f7af912005-12-17 12:21:26 +01005792 /* detach from the tty */
willy tarreau5cbea6f2005-12-17 12:48:26 +01005793 fclose(stdin); fclose(stdout); fclose(stderr);
willy tarreau0f7af912005-12-17 12:21:26 +01005794 close(0); close(1); close(2);
willy tarreau0f7af912005-12-17 12:21:26 +01005795 }
5796
5797 signal(SIGQUIT, dump);
5798 signal(SIGUSR1, sig_soft_stop);
willy tarreau8337c6b2005-12-17 13:41:01 +01005799 signal(SIGHUP, sig_dump_state);
willy tarreau0f7af912005-12-17 12:21:26 +01005800
5801 /* on very high loads, a sigpipe sometimes happen just between the
5802 * getsockopt() which tells "it's OK to write", and the following write :-(
5803 */
willy tarreau3242e862005-12-17 12:27:53 +01005804#ifndef MSG_NOSIGNAL
5805 signal(SIGPIPE, SIG_IGN);
5806#endif
willy tarreau0f7af912005-12-17 12:21:26 +01005807
5808 if (start_proxies() < 0)
5809 exit(1);
5810
willy tarreau9fe663a2005-12-17 13:02:59 +01005811 /* open log files */
5812
5813 /* chroot if needed */
5814 if (global.chroot != NULL) {
5815 if (chroot(global.chroot) == -1) {
5816 Alert("[%s.main()] Cannot chroot(%s).\n", argv[0], global.chroot);
5817 exit(1);
5818 }
5819 chdir("/");
5820 }
5821
5822 /* setgid / setuid */
willy tarreau036e1ce2005-12-17 13:46:33 +01005823 if (global.gid && setgid(global.gid) == -1) {
willy tarreau9fe663a2005-12-17 13:02:59 +01005824 Alert("[%s.main()] Cannot set gid %d.\n", argv[0], global.gid);
5825 exit(1);
5826 }
5827
willy tarreau036e1ce2005-12-17 13:46:33 +01005828 if (global.uid && setuid(global.uid) == -1) {
willy tarreau9fe663a2005-12-17 13:02:59 +01005829 Alert("[%s.main()] Cannot set uid %d.\n", argv[0], global.uid);
5830 exit(1);
5831 }
5832
5833 if (global.mode & MODE_DAEMON) {
5834 int ret = 0;
5835 int proc;
5836
5837 /* the father launches the required number of processes */
5838 for (proc = 0; proc < global.nbproc; proc++) {
5839 ret = fork();
5840 if (ret < 0) {
5841 Alert("[%s.main()] Cannot fork.\n", argv[0]);
5842 exit(1); /* there has been an error */
5843 }
5844 else if (ret == 0) /* child breaks here */
5845 break;
5846 }
5847 if (proc == global.nbproc)
5848 exit(0); /* parent must leave */
5849
willy tarreau750a4722005-12-17 13:21:24 +01005850 /* if we're NOT in QUIET mode, we should now close the 3 first FDs to ensure
5851 * that we can detach from the TTY. We MUST NOT do it in other cases since
5852 * it would have already be done, and 0-2 would have been affected to listening
5853 * sockets
5854 */
5855 if (!(global.mode & MODE_QUIET)) {
5856 /* detach from the tty */
5857 fclose(stdin); fclose(stdout); fclose(stderr);
5858 close(0); close(1); close(2); /* close all fd's */
5859 global.mode |= MODE_QUIET; /* ensure that we won't say anything from now */
5860 }
willy tarreaua1598082005-12-17 13:08:06 +01005861 pid = getpid(); /* update child's pid */
willy tarreaue867b482005-12-17 13:28:43 +01005862 setsid();
willy tarreau9fe663a2005-12-17 13:02:59 +01005863 }
5864
willy tarreau0f7af912005-12-17 12:21:26 +01005865 select_loop();
5866
5867 exit(0);
5868}