blob: 43c4c65dae10fb5ab42d015eec1d1935fc3bc20b [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 tarreaueedaa9f2005-12-17 14:08:03 +010056#define HAPROXY_VERSION "1.1.23"
57#define HAPROXY_DATE "2003/09/20"
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 tarreau5cbea6f2005-12-17 12:48:26 +0100667 printf("Copyright 2000-2002 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;
1950
1951 /* This is a first attempt at a better logging system.
1952 * For now, we rely on send_log() to provide the date, although it obviously
1953 * is the date of the log and not of the request, and most fields are not
1954 * computed.
1955 */
1956
willy tarreaua1598082005-12-17 13:08:06 +01001957 log = p->to_log & ~s->logs.logwait;
willy tarreau9fe663a2005-12-17 13:02:59 +01001958
1959 pn = (log & LW_CLIP) ?
1960 (unsigned char *)&s->cli_addr.sin_addr :
1961 (unsigned char *)"\0\0\0\0";
1962
willy tarreaua1598082005-12-17 13:08:06 +01001963 uri = (log & LW_REQ) ? s->logs.uri : "<BADREQ>";
willy tarreau9fe663a2005-12-17 13:02:59 +01001964 pxid = p->id;
1965 //srv = (log & LW_SVID) ? s->srv->id : "<svid>";
willy tarreaua1598082005-12-17 13:08:06 +01001966 srv = ((p->to_log & LW_SVID) && s->srv != NULL) ? s->srv->id : "<NOSRV>";
1967
1968 if (p->to_log & LW_DATE) {
1969 struct tm *tm = localtime(&s->logs.tv_accept.tv_sec);
1970
willy tarreau036e1ce2005-12-17 13:46:33 +01001971 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 +01001972 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port),
1973 tm->tm_mday, monthname[tm->tm_mon], tm->tm_year+1900,
1974 tm->tm_hour, tm->tm_min, tm->tm_sec,
1975 pxid, srv,
1976 s->logs.t_request,
1977 (s->logs.t_connect >= 0) ? s->logs.t_connect - s->logs.t_request : -1,
1978 (s->logs.t_data >= 0) ? s->logs.t_data - s->logs.t_connect : -1,
1979 s->logs.t_close,
1980 s->logs.status, s->logs.bytes,
willy tarreau8337c6b2005-12-17 13:41:01 +01001981 s->logs.cli_cookie ? s->logs.cli_cookie : "-",
1982 s->logs.srv_cookie ? s->logs.srv_cookie : "-",
willy tarreau036e1ce2005-12-17 13:46:33 +01001983 sess_term_cond[(s->flags & SN_ERR_MASK) >> SN_ERR_SHIFT],
1984 sess_fin_state[(s->flags & SN_FINST_MASK) >> SN_FINST_SHIFT],
1985 (p->options & PR_O_COOK_ANY) ? sess_cookie[(s->flags & SN_CK_MASK) >> SN_CK_SHIFT] : '-',
1986 (p->options & PR_O_COOK_ANY) ? sess_set_cookie[(s->flags & SN_SCK_MASK) >> SN_SCK_SHIFT] : '-',
willy tarreaua1598082005-12-17 13:08:06 +01001987 uri);
1988 }
1989 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01001990 send_log(p, LOG_INFO, "%d.%d.%d.%d:%d %s %s %d/%d/%d/%d %d %lld %s %s %c%c%c%c \"%s\"\n",
willy tarreaua1598082005-12-17 13:08:06 +01001991 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port),
1992 pxid, srv,
1993 s->logs.t_request,
1994 (s->logs.t_connect >= 0) ? s->logs.t_connect - s->logs.t_request : -1,
1995 (s->logs.t_data >= 0) ? s->logs.t_data - s->logs.t_connect : -1,
1996 s->logs.t_close,
1997 s->logs.status, s->logs.bytes,
willy tarreau8337c6b2005-12-17 13:41:01 +01001998 s->logs.cli_cookie ? s->logs.cli_cookie : "-",
1999 s->logs.srv_cookie ? s->logs.srv_cookie : "-",
willy tarreau036e1ce2005-12-17 13:46:33 +01002000 sess_term_cond[(s->flags & SN_ERR_MASK) >> SN_ERR_SHIFT],
2001 sess_fin_state[(s->flags & SN_FINST_MASK) >> SN_FINST_SHIFT],
2002 (p->options & PR_O_COOK_ANY) ? sess_cookie[(s->flags & SN_CK_MASK) >> SN_CK_SHIFT] : '-',
2003 (p->options & PR_O_COOK_ANY) ? sess_set_cookie[(s->flags & SN_SCK_MASK) >> SN_SCK_SHIFT] : '-',
willy tarreaua1598082005-12-17 13:08:06 +01002004 uri);
2005 }
2006
2007 s->logs.logwait = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01002008}
2009
willy tarreaue39cd132005-12-17 13:00:18 +01002010
2011/*
willy tarreau0f7af912005-12-17 12:21:26 +01002012 * this function is called on a read event from a listen socket, corresponding
willy tarreau5cbea6f2005-12-17 12:48:26 +01002013 * to an accept. It tries to accept as many connections as possible.
2014 * It returns 0.
willy tarreau0f7af912005-12-17 12:21:26 +01002015 */
2016int event_accept(int fd) {
2017 struct proxy *p = (struct proxy *)fdtab[fd].owner;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002018 struct session *s;
2019 struct task *t;
willy tarreau0f7af912005-12-17 12:21:26 +01002020 int cfd;
2021 int one = 1;
2022
willy tarreau5cbea6f2005-12-17 12:48:26 +01002023 while (p->nbconn < p->maxconn) {
2024 struct sockaddr_in addr;
2025 int laddr = sizeof(addr);
2026 if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1)
2027 return 0; /* nothing more to accept */
willy tarreau0f7af912005-12-17 12:21:26 +01002028
willy tarreau5cbea6f2005-12-17 12:48:26 +01002029 if ((s = pool_alloc(session)) == NULL) { /* disable this proxy for a while */
2030 Alert("out of memory in event_accept().\n");
2031 FD_CLR(fd, StaticReadEvent);
2032 p->state = PR_STIDLE;
2033 close(cfd);
2034 return 0;
2035 }
willy tarreau0f7af912005-12-17 12:21:26 +01002036
willy tarreau5cbea6f2005-12-17 12:48:26 +01002037 if ((t = pool_alloc(task)) == NULL) { /* disable this proxy for a while */
2038 Alert("out of memory in event_accept().\n");
2039 FD_CLR(fd, StaticReadEvent);
2040 p->state = PR_STIDLE;
2041 close(cfd);
2042 pool_free(session, s);
2043 return 0;
2044 }
willy tarreau0f7af912005-12-17 12:21:26 +01002045
willy tarreau5cbea6f2005-12-17 12:48:26 +01002046 s->cli_addr = addr;
willy tarreau9fe663a2005-12-17 13:02:59 +01002047 if (cfd >= global.maxsock) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002048 Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
2049 close(cfd);
2050 pool_free(task, t);
2051 pool_free(session, s);
2052 return 0;
2053 }
willy tarreau0f7af912005-12-17 12:21:26 +01002054
willy tarreau5cbea6f2005-12-17 12:48:26 +01002055 if ((fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) ||
2056 (setsockopt(cfd, IPPROTO_TCP, TCP_NODELAY,
2057 (char *) &one, sizeof(one)) == -1)) {
2058 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
2059 close(cfd);
2060 pool_free(task, t);
2061 pool_free(session, s);
2062 return 0;
2063 }
willy tarreau0f7af912005-12-17 12:21:26 +01002064
willy tarreau9fe663a2005-12-17 13:02:59 +01002065 t->next = t->prev = t->rqnext = NULL; /* task not in run queue yet */
2066 t->wq = LIST_HEAD(wait_queue); /* but already has a wait queue assigned */
2067 t->state = TASK_IDLE;
2068 t->process = process_session;
2069 t->context = s;
2070
2071 s->task = t;
2072 s->proxy = p;
2073 s->cli_state = (p->mode == PR_MODE_HTTP) ? CL_STHEADERS : CL_STDATA; /* no HTTP headers for non-HTTP proxies */
2074 s->srv_state = SV_STIDLE;
2075 s->req = s->rep = NULL; /* will be allocated later */
2076 s->flags = 0;
2077 s->res_cr = s->res_cw = s->res_sr = s->res_sw = RES_SILENT;
2078 s->cli_fd = cfd;
2079 s->srv_fd = -1;
willy tarreaua1598082005-12-17 13:08:06 +01002080 s->srv = NULL;
willy tarreau9fe663a2005-12-17 13:02:59 +01002081 s->conn_retries = p->conn_retries;
willy tarreaua1598082005-12-17 13:08:06 +01002082
2083 s->logs.logwait = p->to_log;
2084 s->logs.tv_accept = now;
2085 s->logs.t_request = -1;
2086 s->logs.t_connect = -1;
2087 s->logs.t_data = -1;
2088 s->logs.t_close = 0;
2089 s->logs.uri = NULL;
willy tarreau8337c6b2005-12-17 13:41:01 +01002090 s->logs.cli_cookie = NULL;
2091 s->logs.srv_cookie = NULL;
willy tarreaua1598082005-12-17 13:08:06 +01002092 s->logs.status = -1;
2093 s->logs.bytes = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01002094
willy tarreau2f6ba652005-12-17 13:57:42 +01002095 s->uniq_id = totalconn;
2096
willy tarreau5cbea6f2005-12-17 12:48:26 +01002097 if ((p->mode == PR_MODE_TCP || p->mode == PR_MODE_HTTP)
2098 && (p->logfac1 >= 0 || p->logfac2 >= 0)) {
willy tarreau535ae7a2005-12-17 12:58:00 +01002099 struct sockaddr_in sockname;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002100 unsigned char *pn, *sn;
2101 int namelen;
willy tarreau0f7af912005-12-17 12:21:26 +01002102
willy tarreau5cbea6f2005-12-17 12:48:26 +01002103 namelen = sizeof(sockname);
2104 if (get_original_dst(cfd, (struct sockaddr_in *)&sockname, &namelen) == -1)
2105 getsockname(cfd, (struct sockaddr *)&sockname, &namelen);
2106 sn = (unsigned char *)&sockname.sin_addr;
willy tarreau535ae7a2005-12-17 12:58:00 +01002107 pn = (unsigned char *)&s->cli_addr.sin_addr;
willy tarreau0f7af912005-12-17 12:21:26 +01002108
willy tarreau9fe663a2005-12-17 13:02:59 +01002109 if (p->to_log) {
2110 /* we have the client ip */
willy tarreaua1598082005-12-17 13:08:06 +01002111 if (s->logs.logwait & LW_CLIP)
2112 if (!(s->logs.logwait &= ~LW_CLIP))
willy tarreau9fe663a2005-12-17 13:02:59 +01002113 sess_log(s);
2114 }
2115 else
2116 send_log(p, LOG_INFO, "Connect from %d.%d.%d.%d:%d to %d.%d.%d.%d:%d (%s/%s)\n",
2117 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port),
2118 sn[0], sn[1], sn[2], sn[3], ntohs(sockname.sin_port),
2119 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
willy tarreau5cbea6f2005-12-17 12:48:26 +01002120 }
willy tarreau0f7af912005-12-17 12:21:26 +01002121
willy tarreau9fe663a2005-12-17 13:02:59 +01002122 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau2f6ba652005-12-17 13:57:42 +01002123 struct sockaddr_in sockname;
2124 unsigned char *pn, *sn;
2125 int namelen;
willy tarreauef900ab2005-12-17 12:52:52 +01002126 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01002127
2128 namelen = sizeof(sockname);
2129 if (get_original_dst(cfd, (struct sockaddr_in *)&sockname, &namelen) == -1)
2130 getsockname(cfd, (struct sockaddr *)&sockname, &namelen);
2131 sn = (unsigned char *)&sockname.sin_addr;
2132 pn = (unsigned char *)&s->cli_addr.sin_addr;
2133
2134 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%d.%d.%d.%d:%d]\n",
2135 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
2136 pn[0], pn[1], pn[2], pn[3], ntohs(s->cli_addr.sin_port));
willy tarreauef900ab2005-12-17 12:52:52 +01002137 write(1, trash, len);
2138 }
willy tarreau0f7af912005-12-17 12:21:26 +01002139
willy tarreau5cbea6f2005-12-17 12:48:26 +01002140 if ((s->req = pool_alloc(buffer)) == NULL) { /* no memory */
2141 close(cfd); /* nothing can be done for this fd without memory */
2142 pool_free(task, t);
2143 pool_free(session, s);
2144 return 0;
2145 }
2146 s->req->l = 0;
willy tarreaua1598082005-12-17 13:08:06 +01002147 s->req->total = 0;
willy tarreauef900ab2005-12-17 12:52:52 +01002148 s->req->h = s->req->r = s->req->lr = s->req->w = s->req->data; /* r and w will be reset further */
2149 s->req->rlim = s->req->data + BUFSIZE;
willy tarreaub1ff9db2005-12-17 13:51:03 +01002150 if (s->cli_state == CL_STHEADERS) /* reserve some space for header rewriting */
willy tarreauef900ab2005-12-17 12:52:52 +01002151 s->req->rlim -= MAXREWRITE;
willy tarreau0f7af912005-12-17 12:21:26 +01002152
willy tarreau5cbea6f2005-12-17 12:48:26 +01002153 if ((s->rep = pool_alloc(buffer)) == NULL) { /* no memory */
2154 pool_free(buffer, s->req);
2155 close(cfd); /* nothing can be done for this fd without memory */
2156 pool_free(task, t);
2157 pool_free(session, s);
2158 return 0;
2159 }
2160 s->rep->l = 0;
willy tarreaua1598082005-12-17 13:08:06 +01002161 s->rep->total = 0;
willy tarreauef900ab2005-12-17 12:52:52 +01002162 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 +01002163
willy tarreau5cbea6f2005-12-17 12:48:26 +01002164 fdtab[cfd].read = &event_cli_read;
2165 fdtab[cfd].write = &event_cli_write;
2166 fdtab[cfd].owner = t;
2167 fdtab[cfd].state = FD_STREADY;
willy tarreau0f7af912005-12-17 12:21:26 +01002168
willy tarreau5cbea6f2005-12-17 12:48:26 +01002169 if (p->mode == PR_MODE_HEALTH) { /* health check mode, no client reading */
willy tarreaue39cd132005-12-17 13:00:18 +01002170 client_retnclose(s, 3, "OK\n"); /* forge an "OK" response */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002171 }
2172 else {
2173 FD_SET(cfd, StaticReadEvent);
2174 }
2175
2176 fd_insert(cfd);
2177
2178 tv_eternity(&s->cnexpire);
2179 tv_eternity(&s->srexpire);
2180 tv_eternity(&s->swexpire);
2181 tv_eternity(&s->cwexpire);
2182
2183 if (s->proxy->clitimeout)
2184 tv_delayfrom(&s->crexpire, &now, s->proxy->clitimeout);
2185 else
2186 tv_eternity(&s->crexpire);
2187
2188 t->expire = s->crexpire;
2189
2190 task_queue(t);
willy tarreauef900ab2005-12-17 12:52:52 +01002191
2192 if (p->mode != PR_MODE_HEALTH)
2193 task_wakeup(&rq, t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002194
2195 p->nbconn++;
2196 actconn++;
2197 totalconn++;
2198
2199 // fprintf(stderr, "accepting from %p => %d conn, %d total\n", p, actconn, totalconn);
2200 } /* end of while (p->nbconn < p->maxconn) */
2201 return 0;
2202}
willy tarreau0f7af912005-12-17 12:21:26 +01002203
willy tarreau0f7af912005-12-17 12:21:26 +01002204
willy tarreau5cbea6f2005-12-17 12:48:26 +01002205/*
2206 * This function is used only for server health-checks. It handles
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002207 * the connection acknowledgement. If the proxy requires HTTP health-checks,
2208 * it sends the request. In other cases, it returns 1 if the socket is OK,
willy tarreau5cbea6f2005-12-17 12:48:26 +01002209 * or -1 if an error occured.
2210 */
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002211int event_srv_chk_w(int fd) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002212 struct task *t = fdtab[fd].owner;
2213 struct server *s = t->context;
willy tarreau0f7af912005-12-17 12:21:26 +01002214
willy tarreau5cbea6f2005-12-17 12:48:26 +01002215 int skerr, lskerr;
willy tarreauef900ab2005-12-17 12:52:52 +01002216 lskerr = sizeof(skerr);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002217 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
2218 if (skerr)
2219 s->result = -1;
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002220 else {
2221 if (s->proxy->options & PR_O_HTTP_CHK) {
2222 int ret;
willy tarreau2f6ba652005-12-17 13:57:42 +01002223 /* we want to check if this host replies to "OPTIONS / HTTP/1.0"
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002224 * so we'll send the request, and won't wake the checker up now.
2225 */
2226#ifndef MSG_NOSIGNAL
willy tarreau2f6ba652005-12-17 13:57:42 +01002227 ret = send(fd, s->proxy->check_req, s->proxy->check_len, MSG_DONTWAIT);
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002228#else
willy tarreau2f6ba652005-12-17 13:57:42 +01002229 ret = send(fd, s->proxy->check_req, s->proxy->check_len, MSG_DONTWAIT | MSG_NOSIGNAL);
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002230#endif
2231 if (ret == 22) {
2232 FD_SET(fd, StaticReadEvent); /* prepare for reading reply */
2233 FD_CLR(fd, StaticWriteEvent); /* nothing more to write */
2234 return 0;
2235 }
2236 else
2237 s->result = -1;
2238 }
2239 else {
2240 /* good TCP connection is enough */
2241 s->result = 1;
2242 }
2243 }
2244
2245 task_wakeup(&rq, t);
2246 return 0;
2247}
2248
willy tarreau0f7af912005-12-17 12:21:26 +01002249
willy tarreaubc4e1fb2005-12-17 13:32:07 +01002250/*
2251 * This function is used only for server health-checks. It handles
2252 * the server's reply to an HTTP request. It returns 1 if the server replies
2253 * 2xx or 3xx (valid responses), or -1 in other cases.
2254 */
2255int event_srv_chk_r(int fd) {
2256 char reply[64];
2257 int len;
2258 struct task *t = fdtab[fd].owner;
2259 struct server *s = t->context;
2260
2261 int skerr, lskerr;
2262 lskerr = sizeof(skerr);
2263 getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
2264 s->result = -1;
2265 if (!skerr) {
2266#ifndef MSG_NOSIGNAL
2267 len = recv(fd, reply, sizeof(reply), 0);
2268#else
2269 len = recv(fd, reply, sizeof(reply), MSG_NOSIGNAL);
2270#endif
2271 if ((len >= sizeof("HTTP/1.0 000")) &&
2272 !memcmp(reply, "HTTP/1.", 7) &&
2273 (reply[9] == '2' || reply[9] == '3')) /* 2xx or 3xx */
2274 s->result = 1;
2275 }
2276
2277 FD_CLR(fd, StaticReadEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002278 task_wakeup(&rq, t);
willy tarreau0f7af912005-12-17 12:21:26 +01002279 return 0;
2280}
2281
2282
2283/*
2284 * this function writes the string <str> at position <pos> which must be in buffer <b>,
2285 * and moves <end> just after the end of <str>.
2286 * <b>'s parameters (l, r, w, h, lr) are recomputed to be valid after the shift.
2287 * the shift value (positive or negative) is returned.
2288 * If there's no space left, the move is not done.
2289 *
2290 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002291int buffer_replace(struct buffer *b, char *pos, char *end, char *str) {
willy tarreau0f7af912005-12-17 12:21:26 +01002292 int delta;
2293 int len;
2294
2295 len = strlen(str);
2296 delta = len - (end - pos);
2297
2298 if (delta + b->r >= b->data + BUFSIZE)
2299 return 0; /* no space left */
2300
2301 /* first, protect the end of the buffer */
2302 memmove(end + delta, end, b->data + b->l - end);
2303
2304 /* now, copy str over pos */
2305 memcpy(pos, str,len);
2306
willy tarreau5cbea6f2005-12-17 12:48:26 +01002307 /* we only move data after the displaced zone */
2308 if (b->r > pos) b->r += delta;
2309 if (b->w > pos) b->w += delta;
2310 if (b->h > pos) b->h += delta;
2311 if (b->lr > pos) b->lr += delta;
willy tarreau0f7af912005-12-17 12:21:26 +01002312 b->l += delta;
2313
2314 return delta;
2315}
2316
willy tarreau8337c6b2005-12-17 13:41:01 +01002317/* same except that the string length is given, which allows str to be NULL if
willy tarreau240afa62005-12-17 13:14:35 +01002318 * len is 0.
2319 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002320int buffer_replace2(struct buffer *b, char *pos, char *end, char *str, int len) {
willy tarreau0f7af912005-12-17 12:21:26 +01002321 int delta;
2322
2323 delta = len - (end - pos);
2324
2325 if (delta + b->r >= b->data + BUFSIZE)
2326 return 0; /* no space left */
2327
2328 /* first, protect the end of the buffer */
2329 memmove(end + delta, end, b->data + b->l - end);
2330
2331 /* now, copy str over pos */
willy tarreau240afa62005-12-17 13:14:35 +01002332 if (len)
2333 memcpy(pos, str, len);
willy tarreau0f7af912005-12-17 12:21:26 +01002334
willy tarreau5cbea6f2005-12-17 12:48:26 +01002335 /* we only move data after the displaced zone */
2336 if (b->r > pos) b->r += delta;
2337 if (b->w > pos) b->w += delta;
2338 if (b->h > pos) b->h += delta;
2339 if (b->lr > pos) b->lr += delta;
willy tarreau0f7af912005-12-17 12:21:26 +01002340 b->l += delta;
2341
2342 return delta;
2343}
2344
2345
2346int exp_replace(char *dst, char *src, char *str, regmatch_t *matches) {
2347 char *old_dst = dst;
2348
2349 while (*str) {
2350 if (*str == '\\') {
2351 str++;
willy tarreauc29948c2005-12-17 13:10:27 +01002352 if (isdigit((int)*str)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002353 int len, num;
2354
2355 num = *str - '0';
2356 str++;
2357
2358 if (matches[num].rm_so > -1) {
2359 len = matches[num].rm_eo - matches[num].rm_so;
2360 memcpy(dst, src + matches[num].rm_so, len);
2361 dst += len;
2362 }
2363
2364 }
2365 else if (*str == 'x') {
2366 unsigned char hex1, hex2;
2367 str++;
2368
2369 hex1=toupper(*str++) - '0'; hex2=toupper(*str++) - '0';
2370
2371 if (hex1 > 9) hex1 -= 'A' - '9' - 1;
2372 if (hex2 > 9) hex2 -= 'A' - '9' - 1;
2373 *dst++ = (hex1<<4) + hex2;
2374 }
2375 else
2376 *dst++ = *str++;
2377 }
2378 else
2379 *dst++ = *str++;
2380 }
2381 *dst = 0;
2382 return dst - old_dst;
2383}
2384
willy tarreau9fe663a2005-12-17 13:02:59 +01002385
willy tarreau0f7af912005-12-17 12:21:26 +01002386/*
2387 * manages the client FSM and its socket. BTW, it also tries to handle the
2388 * cookie. It returns 1 if a state has changed (and a resync may be needed),
2389 * 0 else.
2390 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002391int process_cli(struct session *t) {
willy tarreau0f7af912005-12-17 12:21:26 +01002392 int s = t->srv_state;
2393 int c = t->cli_state;
2394 struct buffer *req = t->req;
2395 struct buffer *rep = t->rep;
2396
willy tarreau750a4722005-12-17 13:21:24 +01002397#ifdef DEBUG_FULL
2398 fprintf(stderr,"process_cli: c=%s, s=%s\n", cli_stnames[c], srv_stnames[s]);
2399#endif
willy tarreau0f7af912005-12-17 12:21:26 +01002400 //fprintf(stderr,"process_cli: c=%d, s=%d, cr=%d, cw=%d, sr=%d, sw=%d\n", c, s,
2401 //FD_ISSET(t->cli_fd, StaticReadEvent), FD_ISSET(t->cli_fd, StaticWriteEvent),
2402 //FD_ISSET(t->srv_fd, StaticReadEvent), FD_ISSET(t->srv_fd, StaticWriteEvent)
2403 //);
2404 if (c == CL_STHEADERS) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002405 /* now parse the partial (or complete) headers */
2406 while (req->lr < req->r) { /* this loop only sees one header at each iteration */
2407 char *ptr;
2408 int delete_header;
willy tarreau0f7af912005-12-17 12:21:26 +01002409
willy tarreau5cbea6f2005-12-17 12:48:26 +01002410 ptr = req->lr;
willy tarreau0f7af912005-12-17 12:21:26 +01002411
willy tarreau0f7af912005-12-17 12:21:26 +01002412 /* look for the end of the current header */
2413 while (ptr < req->r && *ptr != '\n' && *ptr != '\r')
2414 ptr++;
2415
willy tarreau5cbea6f2005-12-17 12:48:26 +01002416 if (ptr == req->h) { /* empty line, end of headers */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002417 int line, len;
2418 /* we can only get here after an end of headers */
2419 /* we'll have something else to do here : add new headers ... */
willy tarreau0f7af912005-12-17 12:21:26 +01002420
willy tarreaue39cd132005-12-17 13:00:18 +01002421 if (t->flags & SN_CLDENY) {
2422 /* no need to go further */
willy tarreaua1598082005-12-17 13:08:06 +01002423 t->logs.status = 403;
willy tarreau8337c6b2005-12-17 13:41:01 +01002424 client_retnclose(t, t->proxy->errmsg.len403, t->proxy->errmsg.msg403);
willy tarreau036e1ce2005-12-17 13:46:33 +01002425 if (!(t->flags & SN_ERR_MASK))
2426 t->flags |= SN_ERR_PRXCOND;
2427 if (!(t->flags & SN_FINST_MASK))
2428 t->flags |= SN_FINST_R;
willy tarreaue39cd132005-12-17 13:00:18 +01002429 return 1;
2430 }
2431
willy tarreau5cbea6f2005-12-17 12:48:26 +01002432 for (line = 0; line < t->proxy->nb_reqadd; line++) {
willy tarreau750a4722005-12-17 13:21:24 +01002433 len = sprintf(trash, "%s\r\n", t->proxy->req_add[line]);
2434 buffer_replace2(req, req->h, req->h, trash, len);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002435 }
willy tarreau0f7af912005-12-17 12:21:26 +01002436
willy tarreau9fe663a2005-12-17 13:02:59 +01002437 if (t->proxy->options & PR_O_FWDFOR) {
2438 /* insert an X-Forwarded-For header */
2439 unsigned char *pn;
2440 pn = (unsigned char *)&t->cli_addr.sin_addr;
willy tarreau750a4722005-12-17 13:21:24 +01002441 len = sprintf(trash, "X-Forwarded-For: %d.%d.%d.%d\r\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01002442 pn[0], pn[1], pn[2], pn[3]);
willy tarreau750a4722005-12-17 13:21:24 +01002443 buffer_replace2(req, req->h, req->h, trash, len);
willy tarreau9fe663a2005-12-17 13:02:59 +01002444 }
2445
willy tarreaucd878942005-12-17 13:27:43 +01002446 if (!memcmp(req->data, "POST ", 5))
2447 t->flags |= SN_POST; /* this is a POST request */
2448
willy tarreau5cbea6f2005-12-17 12:48:26 +01002449 t->cli_state = CL_STDATA;
willy tarreauef900ab2005-12-17 12:52:52 +01002450 req->rlim = req->data + BUFSIZE; /* no more rewrite needed */
willy tarreau0f7af912005-12-17 12:21:26 +01002451
willy tarreau750a4722005-12-17 13:21:24 +01002452 t->logs.t_request = tv_diff(&t->logs.tv_accept, &now);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002453 /* FIXME: we'll set the client in a wait state while we try to
2454 * connect to the server. Is this really needed ? wouldn't it be
2455 * better to release the maximum of system buffers instead ? */
willy tarreauef900ab2005-12-17 12:52:52 +01002456 //FD_CLR(t->cli_fd, StaticReadEvent);
2457 //tv_eternity(&t->crexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002458 break;
2459 }
willy tarreau0f7af912005-12-17 12:21:26 +01002460
willy tarreau5cbea6f2005-12-17 12:48:26 +01002461 /* to get a complete header line, we need the ending \r\n, \n\r, \r or \n too */
2462 if (ptr > req->r - 2) {
2463 /* this is a partial header, let's wait for more to come */
2464 req->lr = ptr;
2465 break;
2466 }
willy tarreau0f7af912005-12-17 12:21:26 +01002467
willy tarreau5cbea6f2005-12-17 12:48:26 +01002468 /* now we know that *ptr is either \r or \n,
2469 * and that there are at least 1 char after it.
2470 */
2471 if ((ptr[0] == ptr[1]) || (ptr[1] != '\r' && ptr[1] != '\n'))
2472 req->lr = ptr + 1; /* \r\r, \n\n, \r[^\n], \n[^\r] */
2473 else
2474 req->lr = ptr + 2; /* \r\n or \n\r */
willy tarreau0f7af912005-12-17 12:21:26 +01002475
willy tarreau5cbea6f2005-12-17 12:48:26 +01002476 /*
2477 * now we know that we have a full header ; we can do whatever
2478 * we want with these pointers :
2479 * req->h = beginning of header
2480 * ptr = end of header (first \r or \n)
2481 * req->lr = beginning of next line (next rep->h)
2482 * req->r = end of data (not used at this stage)
2483 */
willy tarreau0f7af912005-12-17 12:21:26 +01002484
willy tarreau8337c6b2005-12-17 13:41:01 +01002485 if (t->logs.logwait & LW_REQ) {
willy tarreau9fe663a2005-12-17 13:02:59 +01002486 /* we have a complete HTTP request that we must log */
2487 int urilen;
2488
willy tarreaua1598082005-12-17 13:08:06 +01002489 if ((t->logs.uri = pool_alloc(requri)) == NULL) {
willy tarreau9fe663a2005-12-17 13:02:59 +01002490 Alert("HTTP logging : out of memory.\n");
willy tarreau750a4722005-12-17 13:21:24 +01002491 t->logs.status = 500;
willy tarreau8337c6b2005-12-17 13:41:01 +01002492 client_retnclose(t, t->proxy->errmsg.len500, t->proxy->errmsg.msg500);
willy tarreau036e1ce2005-12-17 13:46:33 +01002493 if (!(t->flags & SN_ERR_MASK))
2494 t->flags |= SN_ERR_PRXCOND;
2495 if (!(t->flags & SN_FINST_MASK))
2496 t->flags |= SN_FINST_R;
willy tarreau9fe663a2005-12-17 13:02:59 +01002497 return 1;
2498 }
2499
2500 urilen = ptr - req->h;
2501 if (urilen >= REQURI_LEN)
2502 urilen = REQURI_LEN - 1;
willy tarreaua1598082005-12-17 13:08:06 +01002503 memcpy(t->logs.uri, req->h, urilen);
2504 t->logs.uri[urilen] = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01002505
willy tarreaua1598082005-12-17 13:08:06 +01002506 if (!(t->logs.logwait &= ~LW_REQ))
willy tarreau9fe663a2005-12-17 13:02:59 +01002507 sess_log(t);
2508 }
2509
willy tarreau5cbea6f2005-12-17 12:48:26 +01002510 delete_header = 0;
willy tarreau0f7af912005-12-17 12:21:26 +01002511
willy tarreau9fe663a2005-12-17 13:02:59 +01002512 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002513 int len, max;
willy tarreau2f6ba652005-12-17 13:57:42 +01002514 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 +01002515 max = ptr - req->h;
2516 UBOUND(max, sizeof(trash) - len - 1);
willy tarreau750a4722005-12-17 13:21:24 +01002517 len += strlcpy2(trash + len, req->h, max + 1);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002518 trash[len++] = '\n';
2519 write(1, trash, len);
2520 }
willy tarreau0f7af912005-12-17 12:21:26 +01002521
willy tarreau5cbea6f2005-12-17 12:48:26 +01002522 /* try headers regexps */
willy tarreaue39cd132005-12-17 13:00:18 +01002523 if (t->proxy->req_exp != NULL && !(t->flags & SN_CLDENY)) {
2524 struct hdr_exp *exp;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002525 char term;
2526
2527 term = *ptr;
2528 *ptr = '\0';
willy tarreaue39cd132005-12-17 13:00:18 +01002529 exp = t->proxy->req_exp;
2530 do {
2531 if (regexec(exp->preg, req->h, MAX_MATCH, pmatch, 0) == 0) {
2532 switch (exp->action) {
2533 case ACT_ALLOW:
2534 if (!(t->flags & SN_CLDENY))
2535 t->flags |= SN_CLALLOW;
2536 break;
2537 case ACT_REPLACE:
2538 if (!(t->flags & SN_CLDENY)) {
2539 int len = exp_replace(trash, req->h, exp->replace, pmatch);
2540 ptr += buffer_replace2(req, req->h, ptr, trash, len);
2541 }
2542 break;
2543 case ACT_REMOVE:
2544 if (!(t->flags & SN_CLDENY))
2545 delete_header = 1;
2546 break;
2547 case ACT_DENY:
2548 if (!(t->flags & SN_CLALLOW))
2549 t->flags |= SN_CLDENY;
2550 break;
willy tarreau036e1ce2005-12-17 13:46:33 +01002551 case ACT_PASS: /* we simply don't deny this one */
2552 break;
willy tarreau0f7af912005-12-17 12:21:26 +01002553 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002554 break;
willy tarreau0f7af912005-12-17 12:21:26 +01002555 }
willy tarreaue39cd132005-12-17 13:00:18 +01002556 } while ((exp = exp->next) != NULL);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002557 *ptr = term; /* restore the string terminator */
willy tarreau0f7af912005-12-17 12:21:26 +01002558 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002559
willy tarreau240afa62005-12-17 13:14:35 +01002560 /* Now look for cookies. Conforming to RFC2109, we have to support
2561 * attributes whose name begin with a '$', and associate them with
2562 * the right cookie, if we want to delete this cookie.
2563 * So there are 3 cases for each cookie read :
2564 * 1) it's a special attribute, beginning with a '$' : ignore it.
2565 * 2) it's a server id cookie that we *MAY* want to delete : save
2566 * some pointers on it (last semi-colon, beginning of cookie...)
2567 * 3) it's an application cookie : we *MAY* have to delete a previous
2568 * "special" cookie.
2569 * At the end of loop, if a "special" cookie remains, we may have to
2570 * remove it. If no application cookie persists in the header, we
2571 * *MUST* delete it
2572 */
willy tarreau8337c6b2005-12-17 13:41:01 +01002573 if (!delete_header && (t->proxy->cookie_name != NULL || t->proxy->capture_name != NULL)
willy tarreau240afa62005-12-17 13:14:35 +01002574 && !(t->flags & SN_CLDENY) && (ptr >= req->h + 8)
willy tarreau906b2682005-12-17 13:49:52 +01002575 && (strncasecmp(req->h, "Cookie: ", 8) == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01002576 char *p1, *p2, *p3, *p4;
willy tarreau240afa62005-12-17 13:14:35 +01002577 char *del_colon, *del_cookie, *colon;
2578 int app_cookies;
2579
willy tarreau5cbea6f2005-12-17 12:48:26 +01002580 p1 = req->h + 8; /* first char after 'Cookie: ' */
willy tarreau240afa62005-12-17 13:14:35 +01002581 colon = p1;
2582 /* del_cookie == NULL => nothing to be deleted */
2583 del_colon = del_cookie = NULL;
2584 app_cookies = 0;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002585
2586 while (p1 < ptr) {
willy tarreau240afa62005-12-17 13:14:35 +01002587 /* skip spaces and colons, but keep an eye on these ones */
2588 while (p1 < ptr) {
2589 if (*p1 == ';' || *p1 == ',')
2590 colon = p1;
2591 else if (!isspace((int)*p1))
2592 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002593 p1++;
willy tarreau240afa62005-12-17 13:14:35 +01002594 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002595
2596 if (p1 == ptr)
2597 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002598
2599 /* p1 is at the beginning of the cookie name */
2600 p2 = p1;
willy tarreau240afa62005-12-17 13:14:35 +01002601 while (p2 < ptr && *p2 != '=')
willy tarreau5cbea6f2005-12-17 12:48:26 +01002602 p2++;
2603
2604 if (p2 == ptr)
2605 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002606
2607 p3 = p2 + 1; /* skips the '=' sign */
2608 if (p3 == ptr)
2609 break;
2610
willy tarreau240afa62005-12-17 13:14:35 +01002611 p4 = p3;
2612 while (p4 < ptr && !isspace((int)*p4) && *p4 != ';' && *p4 != ',')
willy tarreau5cbea6f2005-12-17 12:48:26 +01002613 p4++;
2614
2615 /* here, we have the cookie name between p1 and p2,
2616 * and its value between p3 and p4.
2617 * we can process it.
2618 */
2619
willy tarreau240afa62005-12-17 13:14:35 +01002620 if (*p1 == '$') {
2621 /* skip this one */
2622 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002623 else {
2624 /* first, let's see if we want to capture it */
2625 if (t->proxy->capture_name != NULL &&
2626 t->logs.cli_cookie == NULL &&
2627 (p4 - p1 >= t->proxy->capture_namelen) &&
2628 memcmp(p1, t->proxy->capture_name, t->proxy->capture_namelen) == 0) {
2629 int log_len = p4 - p1;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002630
willy tarreau8337c6b2005-12-17 13:41:01 +01002631 if ((t->logs.cli_cookie = pool_alloc(capture)) == NULL) {
2632 Alert("HTTP logging : out of memory.\n");
2633 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002634
willy tarreau8337c6b2005-12-17 13:41:01 +01002635 if (log_len > t->proxy->capture_len)
2636 log_len = t->proxy->capture_len;
2637 memcpy(t->logs.cli_cookie, p1, log_len);
2638 t->logs.cli_cookie[log_len] = 0;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002639 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002640
2641 if ((p2 - p1 == t->proxy->cookie_len) && (t->proxy->cookie_name != NULL) &&
2642 (memcmp(p1, t->proxy->cookie_name, p2 - p1) == 0)) {
2643 /* Cool... it's the right one */
2644 struct server *srv = t->proxy->srv;
2645
2646 while (srv &&
2647 ((srv->cklen != p4 - p3) || memcmp(p3, srv->cookie, p4 - p3))) {
2648 srv = srv->next;
2649 }
2650
willy tarreau036e1ce2005-12-17 13:46:33 +01002651 if (!srv) {
2652 t->flags &= ~SN_CK_MASK;
2653 t->flags |= SN_CK_INVALID;
2654 }
2655 else if (srv->state & SRV_RUNNING || t->proxy->options & PR_O_PERSIST) {
willy tarreau8337c6b2005-12-17 13:41:01 +01002656 /* we found the server and it's usable */
willy tarreau036e1ce2005-12-17 13:46:33 +01002657 t->flags &= ~SN_CK_MASK;
2658 t->flags |= SN_CK_VALID | SN_DIRECT;
willy tarreau8337c6b2005-12-17 13:41:01 +01002659 t->srv = srv;
2660 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002661 else {
2662 t->flags &= ~SN_CK_MASK;
2663 t->flags |= SN_CK_DOWN;
2664 }
2665
willy tarreau8337c6b2005-12-17 13:41:01 +01002666 /* if this cookie was set in insert+indirect mode, then it's better that the
2667 * server never sees it.
2668 */
2669 if (del_cookie == NULL &&
2670 (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 +01002671 del_cookie = p1;
2672 del_colon = colon;
willy tarreau8337c6b2005-12-17 13:41:01 +01002673 }
willy tarreau240afa62005-12-17 13:14:35 +01002674 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002675 else {
2676 /* now we know that we must keep this cookie since it's
2677 * not ours. But if we wanted to delete our cookie
2678 * earlier, we cannot remove the complete header, but we
2679 * can remove the previous block itself.
2680 */
2681 app_cookies++;
2682
2683 if (del_cookie != NULL) {
2684 buffer_replace2(req, del_cookie, p1, NULL, 0);
2685 p4 -= (p1 - del_cookie);
2686 ptr -= (p1 - del_cookie);
2687 del_cookie = del_colon = NULL;
2688 }
willy tarreau240afa62005-12-17 13:14:35 +01002689 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002690 }
willy tarreau240afa62005-12-17 13:14:35 +01002691
willy tarreau5cbea6f2005-12-17 12:48:26 +01002692 /* we'll have to look for another cookie ... */
2693 p1 = p4;
2694 } /* while (p1 < ptr) */
willy tarreau240afa62005-12-17 13:14:35 +01002695
2696 /* There's no more cookie on this line.
2697 * We may have marked the last one(s) for deletion.
2698 * We must do this now in two ways :
2699 * - if there is no app cookie, we simply delete the header ;
2700 * - if there are app cookies, we must delete the end of the
2701 * string properly, including the colon/semi-colon before
2702 * the cookie name.
2703 */
2704 if (del_cookie != NULL) {
2705 if (app_cookies) {
2706 buffer_replace2(req, del_colon, ptr, NULL, 0);
2707 /* WARNING! <ptr> becomes invalid for now. If some code
2708 * below needs to rely on it before the end of the global
2709 * header loop, we need to correct it with this code :
2710 * ptr = del_colon;
2711 */
2712 }
2713 else
2714 delete_header = 1;
2715 }
2716 } /* end of cookie processing on this header */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002717
2718 /* let's look if we have to delete this header */
willy tarreaue39cd132005-12-17 13:00:18 +01002719 if (delete_header && !(t->flags & SN_CLDENY)) {
willy tarreau240afa62005-12-17 13:14:35 +01002720 buffer_replace2(req, req->h, req->lr, NULL, 0);
willy tarreau0f7af912005-12-17 12:21:26 +01002721 }
willy tarreau240afa62005-12-17 13:14:35 +01002722 /* WARNING: ptr is not valid anymore, since the header may have been deleted or truncated ! */
2723
willy tarreau5cbea6f2005-12-17 12:48:26 +01002724 req->h = req->lr;
2725 } /* while (req->lr < req->r) */
2726
2727 /* end of header processing (even if incomplete) */
2728
willy tarreauef900ab2005-12-17 12:52:52 +01002729 if ((req->l < req->rlim - req->data) && ! FD_ISSET(t->cli_fd, StaticReadEvent)) {
2730 /* fd in StaticReadEvent was disabled, perhaps because of a previous buffer
2731 * full. We cannot loop here since event_cli_read will disable it only if
2732 * req->l == rlim-data
2733 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002734 FD_SET(t->cli_fd, StaticReadEvent);
2735 if (t->proxy->clitimeout)
2736 tv_delayfrom(&t->crexpire, &now, t->proxy->clitimeout);
2737 else
2738 tv_eternity(&t->crexpire);
2739 }
2740
willy tarreaue39cd132005-12-17 13:00:18 +01002741 /* Since we are in header mode, if there's no space left for headers, we
willy tarreauef900ab2005-12-17 12:52:52 +01002742 * won't be able to free more later, so the session will never terminate.
2743 */
willy tarreaue39cd132005-12-17 13:00:18 +01002744 if (req->l >= req->rlim - req->data) {
willy tarreaua1598082005-12-17 13:08:06 +01002745 t->logs.status = 400;
willy tarreau8337c6b2005-12-17 13:41:01 +01002746 client_retnclose(t, t->proxy->errmsg.len400, t->proxy->errmsg.msg400);
willy tarreau036e1ce2005-12-17 13:46:33 +01002747 if (!(t->flags & SN_ERR_MASK))
2748 t->flags |= SN_ERR_PRXCOND;
2749 if (!(t->flags & SN_FINST_MASK))
2750 t->flags |= SN_FINST_R;
willy tarreaue39cd132005-12-17 13:00:18 +01002751 return 1;
2752 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002753 else if (t->res_cr == RES_ERROR || t->res_cr == RES_NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01002754 /* read error, or last read : give up. */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002755 tv_eternity(&t->crexpire);
2756 fd_delete(t->cli_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002757 t->cli_state = CL_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01002758 if (!(t->flags & SN_ERR_MASK))
2759 t->flags |= SN_ERR_CLICL;
2760 if (!(t->flags & SN_FINST_MASK))
2761 t->flags |= SN_FINST_R;
willy tarreau5cbea6f2005-12-17 12:48:26 +01002762 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01002763 }
willy tarreau8337c6b2005-12-17 13:41:01 +01002764 else if (tv_cmp2_ms(&t->crexpire, &now) <= 0) {
2765
2766 /* read timeout : give up with an error message.
2767 */
2768 t->logs.status = 408;
2769 client_retnclose(t, t->proxy->errmsg.len408, t->proxy->errmsg.msg408);
willy tarreau036e1ce2005-12-17 13:46:33 +01002770 if (!(t->flags & SN_ERR_MASK))
2771 t->flags |= SN_ERR_CLITO;
2772 if (!(t->flags & SN_FINST_MASK))
2773 t->flags |= SN_FINST_R;
willy tarreau8337c6b2005-12-17 13:41:01 +01002774 return 1;
2775 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01002776
2777 return t->cli_state != CL_STHEADERS;
willy tarreau0f7af912005-12-17 12:21:26 +01002778 }
2779 else if (c == CL_STDATA) {
2780 /* read or write error */
2781 if (t->res_cw == RES_ERROR || t->res_cr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01002782 tv_eternity(&t->crexpire);
2783 tv_eternity(&t->cwexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01002784 fd_delete(t->cli_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002785 t->cli_state = CL_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01002786 if (!(t->flags & SN_ERR_MASK))
2787 t->flags |= SN_ERR_CLICL;
2788 if (!(t->flags & SN_FINST_MASK))
2789 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01002790 return 1;
2791 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002792 /* last read, or end of server write */
2793 else if (t->res_cr == RES_NULL || s == SV_STSHUTW || s == SV_STCLOSE) {
willy tarreau0f7af912005-12-17 12:21:26 +01002794 FD_CLR(t->cli_fd, StaticReadEvent);
2795 // if (req->l == 0) /* nothing to write on the server side */
2796 // FD_CLR(t->srv_fd, StaticWriteEvent);
2797 tv_eternity(&t->crexpire);
2798 shutdown(t->cli_fd, SHUT_RD);
2799 t->cli_state = CL_STSHUTR;
2800 return 1;
2801 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002802 /* last server read and buffer empty */
2803 else if ((s == SV_STSHUTR || s == SV_STCLOSE) && (rep->l == 0)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002804 FD_CLR(t->cli_fd, StaticWriteEvent);
2805 tv_eternity(&t->cwexpire);
2806 shutdown(t->cli_fd, SHUT_WR);
2807 t->cli_state = CL_STSHUTW;
2808 return 1;
2809 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002810 /* read timeout */
2811 else if (tv_cmp2_ms(&t->crexpire, &now) <= 0) {
2812 FD_CLR(t->cli_fd, StaticReadEvent);
2813 // if (req->l == 0) /* nothing to write on the server side */
2814 // FD_CLR(t->srv_fd, StaticWriteEvent);
2815 tv_eternity(&t->crexpire);
2816 shutdown(t->cli_fd, SHUT_RD);
2817 t->cli_state = CL_STSHUTR;
2818 if (!(t->flags & SN_ERR_MASK))
2819 t->flags |= SN_ERR_CLITO;
2820 if (!(t->flags & SN_FINST_MASK))
2821 t->flags |= SN_FINST_D;
2822 return 1;
2823 }
2824 /* write timeout */
2825 else if (tv_cmp2_ms(&t->cwexpire, &now) <= 0) {
2826 FD_CLR(t->cli_fd, StaticWriteEvent);
2827 tv_eternity(&t->cwexpire);
2828 shutdown(t->cli_fd, SHUT_WR);
2829 t->cli_state = CL_STSHUTW;
2830 if (!(t->flags & SN_ERR_MASK))
willy tarreaub1ff9db2005-12-17 13:51:03 +01002831 t->flags |= SN_ERR_CLITO;
willy tarreau036e1ce2005-12-17 13:46:33 +01002832 if (!(t->flags & SN_FINST_MASK))
2833 t->flags |= SN_FINST_D;
2834 return 1;
2835 }
willy tarreau0f7af912005-12-17 12:21:26 +01002836
willy tarreauef900ab2005-12-17 12:52:52 +01002837 if (req->l >= req->rlim - req->data) {
2838 /* no room to read more data */
willy tarreau0f7af912005-12-17 12:21:26 +01002839 if (FD_ISSET(t->cli_fd, StaticReadEvent)) {
willy tarreauef900ab2005-12-17 12:52:52 +01002840 /* stop reading until we get some space */
willy tarreau0f7af912005-12-17 12:21:26 +01002841 FD_CLR(t->cli_fd, StaticReadEvent);
2842 tv_eternity(&t->crexpire);
2843 }
2844 }
2845 else {
willy tarreauef900ab2005-12-17 12:52:52 +01002846 /* there's still some space in the buffer */
willy tarreau0f7af912005-12-17 12:21:26 +01002847 if (! FD_ISSET(t->cli_fd, StaticReadEvent)) {
2848 FD_SET(t->cli_fd, StaticReadEvent);
2849 if (t->proxy->clitimeout)
2850 tv_delayfrom(&t->crexpire, &now, t->proxy->clitimeout);
2851 else
2852 tv_eternity(&t->crexpire);
2853 }
2854 }
2855
2856 if ((rep->l == 0) ||
willy tarreau5cbea6f2005-12-17 12:48:26 +01002857 ((s == SV_STHEADERS) /* FIXME: this may be optimized && (rep->w == rep->h)*/)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002858 if (FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2859 FD_CLR(t->cli_fd, StaticWriteEvent); /* stop writing */
2860 tv_eternity(&t->cwexpire);
2861 }
2862 }
2863 else { /* buffer not empty */
2864 if (! FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2865 FD_SET(t->cli_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01002866 if (t->proxy->clitimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01002867 tv_delayfrom(&t->cwexpire, &now, t->proxy->clitimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01002868 /* FIXME: to avoid the client to read-time-out during writes, we refresh it */
2869 t->crexpire = t->cwexpire;
2870 }
willy tarreau0f7af912005-12-17 12:21:26 +01002871 else
2872 tv_eternity(&t->cwexpire);
2873 }
2874 }
2875 return 0; /* other cases change nothing */
2876 }
2877 else if (c == CL_STSHUTR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01002878 if (t->res_cw == RES_ERROR) {
2879 tv_eternity(&t->cwexpire);
2880 fd_delete(t->cli_fd);
2881 t->cli_state = CL_STCLOSE;
2882 if (!(t->flags & SN_ERR_MASK))
2883 t->flags |= SN_ERR_CLICL;
2884 if (!(t->flags & SN_FINST_MASK))
2885 t->flags |= SN_FINST_D;
2886 return 1;
2887 }
2888 else if ((s == SV_STSHUTR || s == SV_STCLOSE) && (rep->l == 0)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002889 tv_eternity(&t->cwexpire);
2890 fd_delete(t->cli_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002891 t->cli_state = CL_STCLOSE;
2892 return 1;
2893 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002894 else if (tv_cmp2_ms(&t->cwexpire, &now) <= 0) {
2895 tv_eternity(&t->cwexpire);
2896 fd_delete(t->cli_fd);
2897 t->cli_state = CL_STCLOSE;
2898 if (!(t->flags & SN_ERR_MASK))
2899 t->flags |= SN_ERR_CLITO;
2900 if (!(t->flags & SN_FINST_MASK))
2901 t->flags |= SN_FINST_D;
2902 return 1;
2903 }
willy tarreau0f7af912005-12-17 12:21:26 +01002904 else if ((rep->l == 0) ||
willy tarreau5cbea6f2005-12-17 12:48:26 +01002905 ((s == SV_STHEADERS) /* FIXME: this may be optimized && (rep->w == rep->h)*/)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002906 if (FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2907 FD_CLR(t->cli_fd, StaticWriteEvent); /* stop writing */
2908 tv_eternity(&t->cwexpire);
2909 }
2910 }
2911 else { /* buffer not empty */
2912 if (! FD_ISSET(t->cli_fd, StaticWriteEvent)) {
2913 FD_SET(t->cli_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01002914 if (t->proxy->clitimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01002915 tv_delayfrom(&t->cwexpire, &now, t->proxy->clitimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01002916 /* FIXME: to avoid the client to read-time-out during writes, we refresh it */
2917 t->crexpire = t->cwexpire;
2918 }
willy tarreau0f7af912005-12-17 12:21:26 +01002919 else
2920 tv_eternity(&t->cwexpire);
2921 }
2922 }
2923 return 0;
2924 }
2925 else if (c == CL_STSHUTW) {
willy tarreau036e1ce2005-12-17 13:46:33 +01002926 if (t->res_cr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01002927 tv_eternity(&t->crexpire);
2928 fd_delete(t->cli_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01002929 t->cli_state = CL_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01002930 if (!(t->flags & SN_ERR_MASK))
2931 t->flags |= SN_ERR_CLICL;
2932 if (!(t->flags & SN_FINST_MASK))
2933 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01002934 return 1;
2935 }
willy tarreau036e1ce2005-12-17 13:46:33 +01002936 else if (t->res_cr == RES_NULL || s == SV_STSHUTW || s == SV_STCLOSE) {
2937 tv_eternity(&t->crexpire);
2938 fd_delete(t->cli_fd);
2939 t->cli_state = CL_STCLOSE;
2940 return 1;
2941 }
2942 else if (tv_cmp2_ms(&t->crexpire, &now) <= 0) {
2943 tv_eternity(&t->crexpire);
2944 fd_delete(t->cli_fd);
2945 t->cli_state = CL_STCLOSE;
2946 if (!(t->flags & SN_ERR_MASK))
2947 t->flags |= SN_ERR_CLITO;
2948 if (!(t->flags & SN_FINST_MASK))
2949 t->flags |= SN_FINST_D;
2950 return 1;
2951 }
willy tarreauef900ab2005-12-17 12:52:52 +01002952 else if (req->l >= req->rlim - req->data) {
2953 /* no room to read more data */
willy tarreau0f7af912005-12-17 12:21:26 +01002954 if (FD_ISSET(t->cli_fd, StaticReadEvent)) {
willy tarreauef900ab2005-12-17 12:52:52 +01002955 /* stop reading until we get some space */
willy tarreau0f7af912005-12-17 12:21:26 +01002956 FD_CLR(t->cli_fd, StaticReadEvent);
2957 tv_eternity(&t->crexpire);
2958 }
2959 }
2960 else {
willy tarreauef900ab2005-12-17 12:52:52 +01002961 /* there's still some space in the buffer */
willy tarreau0f7af912005-12-17 12:21:26 +01002962 if (! FD_ISSET(t->cli_fd, StaticReadEvent)) {
2963 FD_SET(t->cli_fd, StaticReadEvent);
2964 if (t->proxy->clitimeout)
2965 tv_delayfrom(&t->crexpire, &now, t->proxy->clitimeout);
2966 else
2967 tv_eternity(&t->crexpire);
2968 }
2969 }
2970 return 0;
2971 }
2972 else { /* CL_STCLOSE: nothing to do */
willy tarreau9fe663a2005-12-17 13:02:59 +01002973 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau0f7af912005-12-17 12:21:26 +01002974 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01002975 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 +01002976 write(1, trash, len);
2977 }
2978 return 0;
2979 }
2980 return 0;
2981}
2982
2983
2984/*
2985 * manages the server FSM and its socket. It returns 1 if a state has changed
2986 * (and a resync may be needed), 0 else.
2987 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01002988int process_srv(struct session *t) {
willy tarreau0f7af912005-12-17 12:21:26 +01002989 int s = t->srv_state;
2990 int c = t->cli_state;
2991 struct buffer *req = t->req;
2992 struct buffer *rep = t->rep;
2993
willy tarreau750a4722005-12-17 13:21:24 +01002994#ifdef DEBUG_FULL
2995 fprintf(stderr,"process_srv: c=%s, s=%s\n", cli_stnames[c], srv_stnames[s]);
2996#endif
willy tarreau5cbea6f2005-12-17 12:48:26 +01002997 //fprintf(stderr,"process_srv: c=%d, s=%d, cr=%d, cw=%d, sr=%d, sw=%d\n", c, s,
2998 //FD_ISSET(t->cli_fd, StaticReadEvent), FD_ISSET(t->cli_fd, StaticWriteEvent),
2999 //FD_ISSET(t->srv_fd, StaticReadEvent), FD_ISSET(t->srv_fd, StaticWriteEvent)
3000 //);
willy tarreau0f7af912005-12-17 12:21:26 +01003001 if (s == SV_STIDLE) {
3002 if (c == CL_STHEADERS)
3003 return 0; /* stay in idle, waiting for data to reach the client side */
3004 else if (c == CL_STCLOSE ||
3005 c == CL_STSHUTW ||
3006 (c == CL_STSHUTR && t->req->l == 0)) { /* give up */
3007 tv_eternity(&t->cnexpire);
3008 t->srv_state = SV_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01003009 if (!(t->flags & SN_ERR_MASK))
3010 t->flags |= SN_ERR_CLICL;
3011 if (!(t->flags & SN_FINST_MASK))
3012 t->flags |= SN_FINST_C;
willy tarreau0f7af912005-12-17 12:21:26 +01003013 return 1;
3014 }
3015 else { /* go to SV_STCONN */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003016 if (connect_server(t) == 0) { /* initiate a connection to the server */
willy tarreau0f7af912005-12-17 12:21:26 +01003017 //fprintf(stderr,"0: c=%d, s=%d\n", c, s);
3018 t->srv_state = SV_STCONN;
3019 }
3020 else { /* try again */
3021 while (t->conn_retries-- > 0) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003022 if ((t->proxy->options & PR_O_REDISP) && (t->conn_retries == 0)) {
willy tarreaue39cd132005-12-17 13:00:18 +01003023 t->flags &= ~SN_DIRECT; /* ignore cookie and force to use the dispatcher */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003024 t->srv = NULL; /* it's left to the dispatcher to choose a server */
willy tarreau036e1ce2005-12-17 13:46:33 +01003025 if ((t->flags & SN_CK_MASK) == SN_CK_VALID) {
3026 t->flags &= ~SN_CK_MASK;
3027 t->flags |= SN_CK_DOWN;
3028 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01003029 }
3030
3031 if (connect_server(t) == 0) {
willy tarreau0f7af912005-12-17 12:21:26 +01003032 t->srv_state = SV_STCONN;
3033 break;
3034 }
3035 }
3036 if (t->conn_retries < 0) {
3037 /* if conn_retries < 0 or other error, let's abort */
3038 tv_eternity(&t->cnexpire);
3039 t->srv_state = SV_STCLOSE;
willy tarreau8337c6b2005-12-17 13:41:01 +01003040 t->logs.status = 503;
willy tarreau750a4722005-12-17 13:21:24 +01003041 if (t->proxy->mode == PR_MODE_HTTP)
willy tarreau8337c6b2005-12-17 13:41:01 +01003042 client_return(t, t->proxy->errmsg.len503, t->proxy->errmsg.msg503);
willy tarreau036e1ce2005-12-17 13:46:33 +01003043 if (!(t->flags & SN_ERR_MASK))
3044 t->flags |= SN_ERR_SRVCL;
3045 if (!(t->flags & SN_FINST_MASK))
3046 t->flags |= SN_FINST_C;
willy tarreau0f7af912005-12-17 12:21:26 +01003047 }
3048 }
3049 return 1;
3050 }
3051 }
3052 else if (s == SV_STCONN) { /* connection in progress */
3053 if (t->res_sw == RES_SILENT && tv_cmp2_ms(&t->cnexpire, &now) > 0) {
3054 //fprintf(stderr,"1: c=%d, s=%d\n", c, s);
3055 return 0; /* nothing changed */
3056 }
3057 else if (t->res_sw == RES_SILENT || t->res_sw == RES_ERROR) {
3058 //fprintf(stderr,"2: c=%d, s=%d\n", c, s);
3059 /* timeout, connect error or first write error */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003060 //FD_CLR(t->srv_fd, StaticWriteEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01003061 fd_delete(t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003062 //close(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003063 t->conn_retries--;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003064 if (t->conn_retries >= 0) {
3065 if ((t->proxy->options & PR_O_REDISP) && (t->conn_retries == 0)) {
willy tarreaue39cd132005-12-17 13:00:18 +01003066 t->flags &= ~SN_DIRECT; /* ignore cookie and force to use the dispatcher */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003067 t->srv = NULL; /* it's left to the dispatcher to choose a server */
willy tarreau036e1ce2005-12-17 13:46:33 +01003068 if ((t->flags & SN_CK_MASK) == SN_CK_VALID) {
3069 t->flags &= ~SN_CK_MASK;
3070 t->flags |= SN_CK_DOWN;
3071 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01003072 }
3073 if (connect_server(t) == 0)
3074 return 0; /* no state changed */
willy tarreau0f7af912005-12-17 12:21:26 +01003075 }
3076 /* if conn_retries < 0 or other error, let's abort */
3077 tv_eternity(&t->cnexpire);
3078 t->srv_state = SV_STCLOSE;
willy tarreau8337c6b2005-12-17 13:41:01 +01003079 t->logs.status = 503;
willy tarreau750a4722005-12-17 13:21:24 +01003080 if (t->proxy->mode == PR_MODE_HTTP)
willy tarreau8337c6b2005-12-17 13:41:01 +01003081 client_return(t, t->proxy->errmsg.len503, t->proxy->errmsg.msg503);
willy tarreau036e1ce2005-12-17 13:46:33 +01003082 if (!(t->flags & SN_ERR_MASK))
3083 t->flags |= SN_ERR_SRVCL;
3084 if (!(t->flags & SN_FINST_MASK))
3085 t->flags |= SN_FINST_C;
willy tarreau0f7af912005-12-17 12:21:26 +01003086 return 1;
3087 }
3088 else { /* no error or write 0 */
willy tarreau750a4722005-12-17 13:21:24 +01003089 t->logs.t_connect = tv_diff(&t->logs.tv_accept, &now);
willy tarreaua1598082005-12-17 13:08:06 +01003090
willy tarreau0f7af912005-12-17 12:21:26 +01003091 //fprintf(stderr,"3: c=%d, s=%d\n", c, s);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003092 if (req->l == 0) /* nothing to write */ {
willy tarreau0f7af912005-12-17 12:21:26 +01003093 FD_CLR(t->srv_fd, StaticWriteEvent);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003094 tv_eternity(&t->swexpire);
3095 } else /* need the right to write */ {
willy tarreau0f7af912005-12-17 12:21:26 +01003096 FD_SET(t->srv_fd, StaticWriteEvent);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003097 if (t->proxy->srvtimeout) {
3098 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
3099 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3100 t->srexpire = t->swexpire;
3101 }
3102 else
3103 tv_eternity(&t->swexpire);
3104 }
willy tarreau0f7af912005-12-17 12:21:26 +01003105
3106 if (t->proxy->mode == PR_MODE_TCP) { /* let's allow immediate data connection in this case */
3107 FD_SET(t->srv_fd, StaticReadEvent);
3108 if (t->proxy->srvtimeout)
3109 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3110 else
3111 tv_eternity(&t->srexpire);
3112
3113 t->srv_state = SV_STDATA;
willy tarreauef900ab2005-12-17 12:52:52 +01003114 rep->rlim = rep->data + BUFSIZE; /* no rewrite needed */
willy tarreau0f7af912005-12-17 12:21:26 +01003115 }
willy tarreauef900ab2005-12-17 12:52:52 +01003116 else {
willy tarreau0f7af912005-12-17 12:21:26 +01003117 t->srv_state = SV_STHEADERS;
willy tarreauef900ab2005-12-17 12:52:52 +01003118 rep->rlim = rep->data + BUFSIZE - MAXREWRITE; /* rewrite needed */
3119 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01003120 tv_eternity(&t->cnexpire);
willy tarreau0f7af912005-12-17 12:21:26 +01003121 return 1;
3122 }
3123 }
3124 else if (s == SV_STHEADERS) { /* receiving server headers */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003125 /* now parse the partial (or complete) headers */
3126 while (rep->lr < rep->r) { /* this loop only sees one header at each iteration */
3127 char *ptr;
3128 int delete_header;
3129
3130 ptr = rep->lr;
3131
3132 /* look for the end of the current header */
3133 while (ptr < rep->r && *ptr != '\n' && *ptr != '\r')
3134 ptr++;
3135
3136 if (ptr == rep->h) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003137 int line, len;
3138
3139 /* we can only get here after an end of headers */
3140 /* we'll have something else to do here : add new headers ... */
3141
willy tarreaucd878942005-12-17 13:27:43 +01003142 if ((t->srv) && !(t->flags & SN_DIRECT) && (t->proxy->options & PR_O_COOK_INS) &&
3143 (!(t->proxy->options & PR_O_COOK_POST) || (t->flags & SN_POST))) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003144 /* the server is known, it's not the one the client requested, we have to
willy tarreaucd878942005-12-17 13:27:43 +01003145 * insert a set-cookie here, except if we want to insert only on POST
3146 * requests and this one isn't.
willy tarreau5cbea6f2005-12-17 12:48:26 +01003147 */
willy tarreau750a4722005-12-17 13:21:24 +01003148 len = sprintf(trash, "Set-Cookie: %s=%s; path=/\r\n",
willy tarreau8337c6b2005-12-17 13:41:01 +01003149 t->proxy->cookie_name,
3150 t->srv->cookie ? t->srv->cookie : "");
willy tarreau750a4722005-12-17 13:21:24 +01003151
willy tarreau036e1ce2005-12-17 13:46:33 +01003152 t->flags |= SN_SCK_INSERTED;
3153
willy tarreau750a4722005-12-17 13:21:24 +01003154 /* Here, we will tell an eventual cache on the client side that we don't
3155 * want it to cache this reply because HTTP/1.0 caches also cache cookies !
3156 * Some caches understand the correct form: 'no-cache="set-cookie"', but
3157 * others don't (eg: apache <= 1.3.26). So we use 'private' instead.
3158 */
willy tarreau240afa62005-12-17 13:14:35 +01003159 if (t->proxy->options & PR_O_COOK_NOC)
willy tarreau750a4722005-12-17 13:21:24 +01003160 //len += sprintf(newhdr + len, "Cache-control: no-cache=\"set-cookie\"\r\n");
3161 len += sprintf(trash + len, "Cache-control: private\r\n");
willy tarreaucd878942005-12-17 13:27:43 +01003162
willy tarreau750a4722005-12-17 13:21:24 +01003163 buffer_replace2(rep, rep->h, rep->h, trash, len);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003164 }
3165
3166 /* headers to be added */
3167 for (line = 0; line < t->proxy->nb_rspadd; line++) {
willy tarreau750a4722005-12-17 13:21:24 +01003168 len = sprintf(trash, "%s\r\n", t->proxy->rsp_add[line]);
3169 buffer_replace2(rep, rep->h, rep->h, trash, len);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003170 }
3171
3172 t->srv_state = SV_STDATA;
willy tarreauef900ab2005-12-17 12:52:52 +01003173 rep->rlim = rep->data + BUFSIZE; /* no more rewrite needed */
willy tarreau750a4722005-12-17 13:21:24 +01003174 t->logs.t_data = tv_diff(&t->logs.tv_accept, &now);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003175 break;
3176 }
3177
3178 /* to get a complete header line, we need the ending \r\n, \n\r, \r or \n too */
3179 if (ptr > rep->r - 2) {
3180 /* this is a partial header, let's wait for more to come */
3181 rep->lr = ptr;
3182 break;
3183 }
3184
3185 // fprintf(stderr,"h=%p, ptr=%p, lr=%p, r=%p, *h=", rep->h, ptr, rep->lr, rep->r);
3186 // write(2, rep->h, ptr - rep->h); fprintf(stderr,"\n");
3187
3188 /* now we know that *ptr is either \r or \n,
3189 * and that there are at least 1 char after it.
3190 */
3191 if ((ptr[0] == ptr[1]) || (ptr[1] != '\r' && ptr[1] != '\n'))
3192 rep->lr = ptr + 1; /* \r\r, \n\n, \r[^\n], \n[^\r] */
3193 else
3194 rep->lr = ptr + 2; /* \r\n or \n\r */
3195
3196 /*
3197 * now we know that we have a full header ; we can do whatever
3198 * we want with these pointers :
3199 * rep->h = beginning of header
3200 * ptr = end of header (first \r or \n)
3201 * rep->lr = beginning of next line (next rep->h)
3202 * rep->r = end of data (not used at this stage)
3203 */
3204
willy tarreaua1598082005-12-17 13:08:06 +01003205
3206 if (t->logs.logwait & LW_RESP) {
3207 t->logs.logwait &= ~LW_RESP;
3208 t->logs.status = atoi(rep->h + 9);
3209 }
3210
willy tarreau5cbea6f2005-12-17 12:48:26 +01003211 delete_header = 0;
3212
willy tarreau9fe663a2005-12-17 13:02:59 +01003213 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003214 int len, max;
willy tarreau2f6ba652005-12-17 13:57:42 +01003215 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 +01003216 max = ptr - rep->h;
3217 UBOUND(max, sizeof(trash) - len - 1);
willy tarreau750a4722005-12-17 13:21:24 +01003218 len += strlcpy2(trash + len, rep->h, max + 1);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003219 trash[len++] = '\n';
3220 write(1, trash, len);
3221 }
3222
3223 /* try headers regexps */
willy tarreaue39cd132005-12-17 13:00:18 +01003224 if (t->proxy->rsp_exp != NULL && !(t->flags & SN_SVDENY)) {
3225 struct hdr_exp *exp;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003226 char term;
3227
3228 term = *ptr;
3229 *ptr = '\0';
willy tarreaue39cd132005-12-17 13:00:18 +01003230 exp = t->proxy->rsp_exp;
3231 do {
3232 if (regexec(exp->preg, rep->h, MAX_MATCH, pmatch, 0) == 0) {
3233 switch (exp->action) {
3234 case ACT_ALLOW:
3235 if (!(t->flags & SN_SVDENY))
3236 t->flags |= SN_SVALLOW;
3237 break;
3238 case ACT_REPLACE:
3239 if (!(t->flags & SN_SVDENY)) {
3240 int len = exp_replace(trash, rep->h, exp->replace, pmatch);
3241 ptr += buffer_replace2(rep, rep->h, ptr, trash, len);
3242 }
3243 break;
3244 case ACT_REMOVE:
3245 if (!(t->flags & SN_SVDENY))
3246 delete_header = 1;
3247 break;
3248 case ACT_DENY:
3249 if (!(t->flags & SN_SVALLOW))
3250 t->flags |= SN_SVDENY;
3251 break;
willy tarreau036e1ce2005-12-17 13:46:33 +01003252 case ACT_PASS: /* we simply don't deny this one */
3253 break;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003254 }
3255 break;
3256 }
willy tarreaue39cd132005-12-17 13:00:18 +01003257 } while ((exp = exp->next) != NULL);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003258 *ptr = term; /* restore the string terminator */
3259 }
3260
3261 /* check for server cookies */
willy tarreau8337c6b2005-12-17 13:41:01 +01003262 if (!delete_header /*&& (t->proxy->options & PR_O_COOK_ANY)*/
3263 && (t->proxy->cookie_name != NULL || t->proxy->capture_name != NULL)
3264 && (ptr >= rep->h + 12)
willy tarreau906b2682005-12-17 13:49:52 +01003265 && (strncasecmp(rep->h, "Set-Cookie: ", 12) == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003266 char *p1, *p2, *p3, *p4;
3267
3268 p1 = rep->h + 12; /* first char after 'Set-Cookie: ' */
3269
3270 while (p1 < ptr) { /* in fact, we'll break after the first cookie */
willy tarreauc29948c2005-12-17 13:10:27 +01003271 while (p1 < ptr && (isspace((int)*p1)))
willy tarreau5cbea6f2005-12-17 12:48:26 +01003272 p1++;
3273
3274 if (p1 == ptr || *p1 == ';') /* end of cookie */
3275 break;
3276
3277 /* p1 is at the beginning of the cookie name */
3278 p2 = p1;
3279
3280 while (p2 < ptr && *p2 != '=' && *p2 != ';')
3281 p2++;
3282
3283 if (p2 == ptr || *p2 == ';') /* next cookie */
3284 break;
3285
3286 p3 = p2 + 1; /* skips the '=' sign */
3287 if (p3 == ptr)
3288 break;
3289
3290 p4 = p3;
willy tarreauc29948c2005-12-17 13:10:27 +01003291 while (p4 < ptr && !isspace((int)*p4) && *p4 != ';')
willy tarreau5cbea6f2005-12-17 12:48:26 +01003292 p4++;
3293
3294 /* here, we have the cookie name between p1 and p2,
3295 * and its value between p3 and p4.
3296 * we can process it.
3297 */
willy tarreau8337c6b2005-12-17 13:41:01 +01003298
3299 /* first, let's see if we want to capture it */
3300 if (t->proxy->capture_name != NULL &&
3301 t->logs.srv_cookie == NULL &&
3302 (p4 - p1 >= t->proxy->capture_namelen) &&
3303 memcmp(p1, t->proxy->capture_name, t->proxy->capture_namelen) == 0) {
3304 int log_len = p4 - p1;
3305
3306 if ((t->logs.srv_cookie = pool_alloc(capture)) == NULL) {
3307 Alert("HTTP logging : out of memory.\n");
3308 }
3309
3310 if (log_len > t->proxy->capture_len)
3311 log_len = t->proxy->capture_len;
3312 memcpy(t->logs.srv_cookie, p1, log_len);
3313 t->logs.srv_cookie[log_len] = 0;
3314 }
3315
3316 if ((p2 - p1 == t->proxy->cookie_len) && (t->proxy->cookie_name != NULL) &&
3317 (memcmp(p1, t->proxy->cookie_name, p2 - p1) == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003318 /* Cool... it's the right one */
willy tarreau036e1ce2005-12-17 13:46:33 +01003319 t->flags |= SN_SCK_SEEN;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003320
3321 /* If the cookie is in insert mode on a known server, we'll delete
3322 * this occurrence because we'll insert another one later.
3323 * We'll delete it too if the "indirect" option is set and we're in
3324 * a direct access. */
3325 if (((t->srv) && (t->proxy->options & PR_O_COOK_INS)) ||
willy tarreaue39cd132005-12-17 13:00:18 +01003326 ((t->flags & SN_DIRECT) && (t->proxy->options & PR_O_COOK_IND))) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003327 /* this header must be deleted */
3328 delete_header = 1;
willy tarreau036e1ce2005-12-17 13:46:33 +01003329 t->flags |= SN_SCK_DELETED;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003330 }
3331 else if ((t->srv) && (t->proxy->options & PR_O_COOK_RW)) {
3332 /* replace bytes p3->p4 with the cookie name associated
3333 * with this server since we know it.
3334 */
3335 buffer_replace2(rep, p3, p4, t->srv->cookie, t->srv->cklen);
willy tarreau036e1ce2005-12-17 13:46:33 +01003336 t->flags |= SN_SCK_INSERTED | SN_SCK_DELETED;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003337 }
3338 break;
3339 }
3340 else {
3341 // fprintf(stderr,"Ignoring unknown cookie : ");
3342 // write(2, p1, p2-p1);
3343 // fprintf(stderr," = ");
3344 // write(2, p3, p4-p3);
3345 // fprintf(stderr,"\n");
3346 }
3347 break; /* we don't want to loop again since there cannot be another cookie on the same line */
3348 } /* we're now at the end of the cookie value */
3349 } /* end of cookie processing */
3350
3351 /* let's look if we have to delete this header */
willy tarreaue39cd132005-12-17 13:00:18 +01003352 if (delete_header && !(t->flags & SN_SVDENY))
willy tarreau5cbea6f2005-12-17 12:48:26 +01003353 buffer_replace2(rep, rep->h, rep->lr, "", 0);
willy tarreaue39cd132005-12-17 13:00:18 +01003354
willy tarreau5cbea6f2005-12-17 12:48:26 +01003355 rep->h = rep->lr;
3356 } /* while (rep->lr < rep->r) */
3357
3358 /* end of header processing (even if incomplete) */
3359
willy tarreauef900ab2005-12-17 12:52:52 +01003360 if ((rep->l < rep->rlim - rep->data) && ! FD_ISSET(t->srv_fd, StaticReadEvent)) {
3361 /* fd in StaticReadEvent was disabled, perhaps because of a previous buffer
3362 * full. We cannot loop here since event_srv_read will disable it only if
3363 * rep->l == rlim-data
3364 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003365 FD_SET(t->srv_fd, StaticReadEvent);
3366 if (t->proxy->srvtimeout)
3367 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3368 else
3369 tv_eternity(&t->srexpire);
3370 }
willy tarreau0f7af912005-12-17 12:21:26 +01003371
willy tarreau8337c6b2005-12-17 13:41:01 +01003372 /* read error, write error */
willy tarreau0f7af912005-12-17 12:21:26 +01003373 if (t->res_sw == RES_ERROR || t->res_sr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01003374 tv_eternity(&t->srexpire);
3375 tv_eternity(&t->swexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003376 fd_delete(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003377 t->srv_state = SV_STCLOSE;
willy tarreaucd878942005-12-17 13:27:43 +01003378 t->logs.status = 502;
willy tarreau8337c6b2005-12-17 13:41:01 +01003379 client_return(t, t->proxy->errmsg.len502, t->proxy->errmsg.msg502);
willy tarreau036e1ce2005-12-17 13:46:33 +01003380 if (!(t->flags & SN_ERR_MASK))
3381 t->flags |= SN_ERR_SRVCL;
3382 if (!(t->flags & SN_FINST_MASK))
3383 t->flags |= SN_FINST_H;
willy tarreau0f7af912005-12-17 12:21:26 +01003384 return 1;
3385 }
willy tarreau8337c6b2005-12-17 13:41:01 +01003386 /* end of client write or end of server read.
willy tarreauef900ab2005-12-17 12:52:52 +01003387 * since we are in header mode, if there's no space left for headers, we
3388 * won't be able to free more later, so the session will never terminate.
3389 */
willy tarreau8337c6b2005-12-17 13:41:01 +01003390 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 +01003391 FD_CLR(t->srv_fd, StaticReadEvent);
3392 tv_eternity(&t->srexpire);
3393 shutdown(t->srv_fd, SHUT_RD);
3394 t->srv_state = SV_STSHUTR;
3395 return 1;
willy tarreau8337c6b2005-12-17 13:41:01 +01003396 }
3397 /* read timeout : return a 504 to the client.
3398 */
3399 else if (FD_ISSET(t->srv_fd, StaticReadEvent) && tv_cmp2_ms(&t->srexpire, &now) <= 0) {
3400 tv_eternity(&t->srexpire);
3401 tv_eternity(&t->swexpire);
3402 fd_delete(t->srv_fd);
3403 t->srv_state = SV_STCLOSE;
3404 t->logs.status = 504;
3405 client_return(t, t->proxy->errmsg.len504, t->proxy->errmsg.msg504);
willy tarreau036e1ce2005-12-17 13:46:33 +01003406 if (!(t->flags & SN_ERR_MASK))
3407 t->flags |= SN_ERR_SRVTO;
3408 if (!(t->flags & SN_FINST_MASK))
3409 t->flags |= SN_FINST_H;
willy tarreau8337c6b2005-12-17 13:41:01 +01003410 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01003411
3412 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003413 /* last client read and buffer empty */
willy tarreau750a4722005-12-17 13:21:24 +01003414 /* FIXME!!! here, we don't want to switch to SHUTW if the
3415 * client shuts read too early, because we may still have
3416 * some work to do on the headers.
willy tarreau036e1ce2005-12-17 13:46:33 +01003417 * The side-effect is that if the client completely closes its
3418 * connection during SV_STHEADER, the connection to the server
3419 * is kept until a response comes back or the timeout is reached.
willy tarreau750a4722005-12-17 13:21:24 +01003420 */
willy tarreau036e1ce2005-12-17 13:46:33 +01003421 else if ((/*c == CL_STSHUTR ||*/ c == CL_STCLOSE) && (req->l == 0)) {
willy tarreau0f7af912005-12-17 12:21:26 +01003422 FD_CLR(t->srv_fd, StaticWriteEvent);
3423 tv_eternity(&t->swexpire);
3424 shutdown(t->srv_fd, SHUT_WR);
3425 t->srv_state = SV_STSHUTW;
3426 return 1;
3427 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003428 /* write timeout */
3429 /* FIXME!!! here, we don't want to switch to SHUTW if the
3430 * client shuts read too early, because we may still have
3431 * some work to do on the headers.
3432 */
3433 else if (FD_ISSET(t->srv_fd, StaticWriteEvent) && tv_cmp2_ms(&t->swexpire, &now) <= 0) {
3434 FD_CLR(t->srv_fd, StaticWriteEvent);
3435 tv_eternity(&t->swexpire);
3436 shutdown(t->srv_fd, SHUT_WR);
3437 t->srv_state = SV_STSHUTW;
3438 if (!(t->flags & SN_ERR_MASK))
3439 t->flags |= SN_ERR_SRVTO;
3440 if (!(t->flags & SN_FINST_MASK))
3441 t->flags |= SN_FINST_H;
3442 return 1;
3443 }
willy tarreau0f7af912005-12-17 12:21:26 +01003444
3445 if (req->l == 0) {
3446 if (FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3447 FD_CLR(t->srv_fd, StaticWriteEvent); /* stop writing */
3448 tv_eternity(&t->swexpire);
3449 }
3450 }
3451 else { /* client buffer not empty */
3452 if (! FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3453 FD_SET(t->srv_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01003454 if (t->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01003455 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003456 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3457 t->srexpire = t->swexpire;
3458 }
willy tarreau0f7af912005-12-17 12:21:26 +01003459 else
3460 tv_eternity(&t->swexpire);
3461 }
3462 }
3463
willy tarreau5cbea6f2005-12-17 12:48:26 +01003464 /* be nice with the client side which would like to send a complete header
3465 * FIXME: COMPLETELY BUGGY !!! not all headers may be processed because the client
3466 * would read all remaining data at once ! The client should not write past rep->lr
3467 * when the server is in header state.
3468 */
3469 //return header_processed;
3470 return t->srv_state != SV_STHEADERS;
willy tarreau0f7af912005-12-17 12:21:26 +01003471 }
3472 else if (s == SV_STDATA) {
3473 /* read or write error */
3474 if (t->res_sw == RES_ERROR || t->res_sr == RES_ERROR) {
willy tarreau0f7af912005-12-17 12:21:26 +01003475 tv_eternity(&t->srexpire);
3476 tv_eternity(&t->swexpire);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003477 fd_delete(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003478 t->srv_state = SV_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01003479 if (!(t->flags & SN_ERR_MASK))
3480 t->flags |= SN_ERR_SRVCL;
3481 if (!(t->flags & SN_FINST_MASK))
3482 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01003483 return 1;
3484 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003485 /* last read, or end of client write */
3486 else if (t->res_sr == RES_NULL || c == CL_STSHUTW || c == CL_STCLOSE) {
willy tarreau0f7af912005-12-17 12:21:26 +01003487 FD_CLR(t->srv_fd, StaticReadEvent);
3488 tv_eternity(&t->srexpire);
3489 shutdown(t->srv_fd, SHUT_RD);
3490 t->srv_state = SV_STSHUTR;
3491 return 1;
willy tarreaua41a8b42005-12-17 14:02:24 +01003492 }
3493 /* end of client read and no more data to send */
3494 else if ((c == CL_STSHUTR || c == CL_STCLOSE) && (req->l == 0)) {
3495 FD_CLR(t->srv_fd, StaticWriteEvent);
3496 tv_eternity(&t->swexpire);
3497 shutdown(t->srv_fd, SHUT_WR);
3498 t->srv_state = SV_STSHUTW;
3499 return 1;
3500 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003501 /* read timeout */
3502 else if (tv_cmp2_ms(&t->srexpire, &now) <= 0) {
3503 FD_CLR(t->srv_fd, StaticReadEvent);
3504 tv_eternity(&t->srexpire);
3505 shutdown(t->srv_fd, SHUT_RD);
3506 t->srv_state = SV_STSHUTR;
3507 if (!(t->flags & SN_ERR_MASK))
3508 t->flags |= SN_ERR_SRVTO;
3509 if (!(t->flags & SN_FINST_MASK))
3510 t->flags |= SN_FINST_D;
3511 return 1;
willy tarreau0f7af912005-12-17 12:21:26 +01003512 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003513 /* write timeout */
3514 else if (tv_cmp2_ms(&t->swexpire, &now) <= 0) {
willy tarreau0f7af912005-12-17 12:21:26 +01003515 FD_CLR(t->srv_fd, StaticWriteEvent);
3516 tv_eternity(&t->swexpire);
3517 shutdown(t->srv_fd, SHUT_WR);
3518 t->srv_state = SV_STSHUTW;
willy tarreau036e1ce2005-12-17 13:46:33 +01003519 if (!(t->flags & SN_ERR_MASK))
3520 t->flags |= SN_ERR_SRVTO;
3521 if (!(t->flags & SN_FINST_MASK))
3522 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01003523 return 1;
3524 }
willy tarreaub1ff9db2005-12-17 13:51:03 +01003525
3526 /* recompute request time-outs */
3527 if (req->l == 0) {
willy tarreau0f7af912005-12-17 12:21:26 +01003528 if (FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3529 FD_CLR(t->srv_fd, StaticWriteEvent); /* stop writing */
3530 tv_eternity(&t->swexpire);
3531 }
3532 }
willy tarreaub1ff9db2005-12-17 13:51:03 +01003533 else { /* buffer not empty, there are still data to be transferred */
willy tarreau0f7af912005-12-17 12:21:26 +01003534 if (! FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3535 FD_SET(t->srv_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01003536 if (t->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01003537 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003538 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3539 t->srexpire = t->swexpire;
3540 }
willy tarreau0f7af912005-12-17 12:21:26 +01003541 else
3542 tv_eternity(&t->swexpire);
3543 }
3544 }
3545
willy tarreaub1ff9db2005-12-17 13:51:03 +01003546 /* recompute response time-outs */
willy tarreau0f7af912005-12-17 12:21:26 +01003547 if (rep->l == BUFSIZE) { /* no room to read more data */
3548 if (FD_ISSET(t->srv_fd, StaticReadEvent)) {
3549 FD_CLR(t->srv_fd, StaticReadEvent);
3550 tv_eternity(&t->srexpire);
3551 }
3552 }
3553 else {
3554 if (! FD_ISSET(t->srv_fd, StaticReadEvent)) {
3555 FD_SET(t->srv_fd, StaticReadEvent);
3556 if (t->proxy->srvtimeout)
3557 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3558 else
3559 tv_eternity(&t->srexpire);
3560 }
3561 }
3562
3563 return 0; /* other cases change nothing */
3564 }
3565 else if (s == SV_STSHUTR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01003566 if (t->res_sw == RES_ERROR) {
3567 //FD_CLR(t->srv_fd, StaticWriteEvent);
3568 tv_eternity(&t->swexpire);
3569 fd_delete(t->srv_fd);
3570 //close(t->srv_fd);
3571 t->srv_state = SV_STCLOSE;
3572 if (!(t->flags & SN_ERR_MASK))
3573 t->flags |= SN_ERR_SRVCL;
3574 if (!(t->flags & SN_FINST_MASK))
3575 t->flags |= SN_FINST_D;
3576 return 1;
3577 }
3578 else if ((c == CL_STSHUTR || c == CL_STCLOSE) && (req->l == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003579 //FD_CLR(t->srv_fd, StaticWriteEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01003580 tv_eternity(&t->swexpire);
3581 fd_delete(t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003582 //close(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003583 t->srv_state = SV_STCLOSE;
3584 return 1;
3585 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003586 else if (tv_cmp2_ms(&t->swexpire, &now) <= 0) {
3587 //FD_CLR(t->srv_fd, StaticWriteEvent);
3588 tv_eternity(&t->swexpire);
3589 fd_delete(t->srv_fd);
3590 //close(t->srv_fd);
3591 t->srv_state = SV_STCLOSE;
3592 if (!(t->flags & SN_ERR_MASK))
3593 t->flags |= SN_ERR_SRVTO;
3594 if (!(t->flags & SN_FINST_MASK))
3595 t->flags |= SN_FINST_D;
3596 return 1;
3597 }
willy tarreau0f7af912005-12-17 12:21:26 +01003598 else if (req->l == 0) {
3599 if (FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3600 FD_CLR(t->srv_fd, StaticWriteEvent); /* stop writing */
3601 tv_eternity(&t->swexpire);
3602 }
3603 }
3604 else { /* buffer not empty */
3605 if (! FD_ISSET(t->srv_fd, StaticWriteEvent)) {
3606 FD_SET(t->srv_fd, StaticWriteEvent); /* restart writing */
willy tarreaub1ff9db2005-12-17 13:51:03 +01003607 if (t->proxy->srvtimeout) {
willy tarreau0f7af912005-12-17 12:21:26 +01003608 tv_delayfrom(&t->swexpire, &now, t->proxy->srvtimeout);
willy tarreaub1ff9db2005-12-17 13:51:03 +01003609 /* FIXME: to avoid the server to read-time-out during writes, we refresh it */
3610 t->srexpire = t->swexpire;
3611 }
willy tarreau0f7af912005-12-17 12:21:26 +01003612 else
3613 tv_eternity(&t->swexpire);
3614 }
3615 }
3616 return 0;
3617 }
3618 else if (s == SV_STSHUTW) {
willy tarreau036e1ce2005-12-17 13:46:33 +01003619 if (t->res_sr == RES_ERROR) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003620 //FD_CLR(t->srv_fd, StaticReadEvent);
willy tarreau0f7af912005-12-17 12:21:26 +01003621 tv_eternity(&t->srexpire);
3622 fd_delete(t->srv_fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003623 //close(t->srv_fd);
willy tarreau0f7af912005-12-17 12:21:26 +01003624 t->srv_state = SV_STCLOSE;
willy tarreau036e1ce2005-12-17 13:46:33 +01003625 if (!(t->flags & SN_ERR_MASK))
3626 t->flags |= SN_ERR_SRVCL;
3627 if (!(t->flags & SN_FINST_MASK))
3628 t->flags |= SN_FINST_D;
willy tarreau0f7af912005-12-17 12:21:26 +01003629 return 1;
3630 }
willy tarreau036e1ce2005-12-17 13:46:33 +01003631 else if (t->res_sr == RES_NULL || c == CL_STSHUTW || c == CL_STCLOSE) {
3632 //FD_CLR(t->srv_fd, StaticReadEvent);
3633 tv_eternity(&t->srexpire);
3634 fd_delete(t->srv_fd);
3635 //close(t->srv_fd);
3636 t->srv_state = SV_STCLOSE;
3637 return 1;
3638 }
3639 else if (tv_cmp2_ms(&t->srexpire, &now) <= 0) {
3640 //FD_CLR(t->srv_fd, StaticReadEvent);
3641 tv_eternity(&t->srexpire);
3642 fd_delete(t->srv_fd);
3643 //close(t->srv_fd);
3644 t->srv_state = SV_STCLOSE;
3645 if (!(t->flags & SN_ERR_MASK))
3646 t->flags |= SN_ERR_SRVTO;
3647 if (!(t->flags & SN_FINST_MASK))
3648 t->flags |= SN_FINST_D;
3649 return 1;
3650 }
willy tarreau0f7af912005-12-17 12:21:26 +01003651 else if (rep->l == BUFSIZE) { /* no room to read more data */
3652 if (FD_ISSET(t->srv_fd, StaticReadEvent)) {
3653 FD_CLR(t->srv_fd, StaticReadEvent);
3654 tv_eternity(&t->srexpire);
3655 }
3656 }
3657 else {
3658 if (! FD_ISSET(t->srv_fd, StaticReadEvent)) {
3659 FD_SET(t->srv_fd, StaticReadEvent);
3660 if (t->proxy->srvtimeout)
3661 tv_delayfrom(&t->srexpire, &now, t->proxy->srvtimeout);
3662 else
3663 tv_eternity(&t->srexpire);
3664 }
3665 }
3666 return 0;
3667 }
3668 else { /* SV_STCLOSE : nothing to do */
willy tarreau9fe663a2005-12-17 13:02:59 +01003669 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau0f7af912005-12-17 12:21:26 +01003670 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01003671 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 +01003672 write(1, trash, len);
3673 }
3674 return 0;
3675 }
3676 return 0;
3677}
3678
3679
willy tarreau5cbea6f2005-12-17 12:48:26 +01003680/* Processes the client and server jobs of a session task, then
3681 * puts it back to the wait queue in a clean state, or
3682 * cleans up its resources if it must be deleted. Returns
3683 * the time the task accepts to wait, or -1 for infinity
willy tarreau0f7af912005-12-17 12:21:26 +01003684 */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003685int process_session(struct task *t) {
3686 struct session *s = t->context;
3687 int fsm_resync = 0;
willy tarreau0f7af912005-12-17 12:21:26 +01003688
willy tarreau5cbea6f2005-12-17 12:48:26 +01003689 do {
3690 fsm_resync = 0;
3691 //fprintf(stderr,"before_cli:cli=%d, srv=%d\n", t->cli_state, t->srv_state);
3692 fsm_resync |= process_cli(s);
3693 //fprintf(stderr,"cli/srv:cli=%d, srv=%d\n", t->cli_state, t->srv_state);
3694 fsm_resync |= process_srv(s);
3695 //fprintf(stderr,"after_srv:cli=%d, srv=%d\n", t->cli_state, t->srv_state);
3696 } while (fsm_resync);
3697
3698 if (s->cli_state != CL_STCLOSE || s->srv_state != SV_STCLOSE) {
willy tarreau0f7af912005-12-17 12:21:26 +01003699 struct timeval min1, min2;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003700 s->res_cw = s->res_cr = s->res_sw = s->res_sr = RES_SILENT;
willy tarreau0f7af912005-12-17 12:21:26 +01003701
willy tarreau5cbea6f2005-12-17 12:48:26 +01003702 tv_min(&min1, &s->crexpire, &s->cwexpire);
3703 tv_min(&min2, &s->srexpire, &s->swexpire);
3704 tv_min(&min1, &min1, &s->cnexpire);
willy tarreau0f7af912005-12-17 12:21:26 +01003705 tv_min(&t->expire, &min1, &min2);
3706
3707 /* restore t to its place in the task list */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003708 task_queue(t);
willy tarreau0f7af912005-12-17 12:21:26 +01003709
willy tarreau5cbea6f2005-12-17 12:48:26 +01003710 return tv_remain(&now, &t->expire); /* nothing more to do */
willy tarreau0f7af912005-12-17 12:21:26 +01003711 }
3712
willy tarreau5cbea6f2005-12-17 12:48:26 +01003713 s->proxy->nbconn--;
willy tarreau0f7af912005-12-17 12:21:26 +01003714 actconn--;
3715
willy tarreau9fe663a2005-12-17 13:02:59 +01003716 if ((global.mode & MODE_DEBUG) && !(global.mode & MODE_QUIET)) {
willy tarreau0f7af912005-12-17 12:21:26 +01003717 int len;
willy tarreau2f6ba652005-12-17 13:57:42 +01003718 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 +01003719 write(1, trash, len);
3720 }
3721
willy tarreau750a4722005-12-17 13:21:24 +01003722 s->logs.t_close = tv_diff(&s->logs.tv_accept, &now);
willy tarreaua1598082005-12-17 13:08:06 +01003723 if (s->rep != NULL)
3724 s->logs.bytes = s->rep->total;
3725
willy tarreau9fe663a2005-12-17 13:02:59 +01003726 /* let's do a final log if we need it */
willy tarreaua1598082005-12-17 13:08:06 +01003727 if (s->logs.logwait && (!(s->proxy->options & PR_O_NULLNOLOG) || s->req->total))
willy tarreau9fe663a2005-12-17 13:02:59 +01003728 sess_log(s);
3729
willy tarreau0f7af912005-12-17 12:21:26 +01003730 /* the task MUST not be in the run queue anymore */
3731 task_delete(t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003732 session_free(s);
willy tarreau0f7af912005-12-17 12:21:26 +01003733 task_free(t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003734 return -1; /* rest in peace for eternity */
3735}
3736
3737
3738
3739/*
3740 * manages a server health-check. Returns
3741 * the time the task accepts to wait, or -1 for infinity.
3742 */
3743int process_chk(struct task *t) {
3744 struct server *s = t->context;
willy tarreaua41a8b42005-12-17 14:02:24 +01003745 struct sockaddr_in sa;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003746 int fd = s->curfd;
3747 int one = 1;
3748
willy tarreauef900ab2005-12-17 12:52:52 +01003749 //fprintf(stderr, "process_chk: task=%p\n", t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003750
3751 if (fd < 0) { /* no check currently running */
3752 //fprintf(stderr, "process_chk: 2\n");
3753 if (tv_cmp2_ms(&t->expire, &now) > 0) { /* not good time yet */
3754 task_queue(t); /* restore t to its place in the task list */
3755 return tv_remain(&now, &t->expire);
3756 }
3757
3758 /* we'll initiate a new check */
3759 s->result = 0; /* no result yet */
3760 if ((fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) != -1) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003761 if ((fd < global.maxsock) &&
willy tarreau5cbea6f2005-12-17 12:48:26 +01003762 (fcntl(fd, F_SETFL, O_NONBLOCK) != -1) &&
3763 (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char *) &one, sizeof(one)) != -1)) {
3764 //fprintf(stderr, "process_chk: 3\n");
3765
willy tarreaua41a8b42005-12-17 14:02:24 +01003766 /* we'll connect to the check port on the server */
3767 sa = s->addr;
3768 sa.sin_port = htons(s->check_port);
3769
willy tarreau036e1ce2005-12-17 13:46:33 +01003770 /* allow specific binding */
3771 if (s->proxy->options & PR_O_BIND_SRC &&
3772 bind(fd, (struct sockaddr *)&s->proxy->source_addr, sizeof(s->proxy->source_addr)) == -1) {
3773 Alert("Cannot bind to source address before connect() for proxy %s. Aborting.\n", s->proxy->id);
3774 close(fd);
3775 s->result = -1;
3776 }
willy tarreaua41a8b42005-12-17 14:02:24 +01003777 else if ((connect(fd, (struct sockaddr *)&sa, sizeof(sa)) != -1) || (errno == EINPROGRESS)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01003778 /* OK, connection in progress or established */
3779
3780 //fprintf(stderr, "process_chk: 4\n");
3781
3782 s->curfd = fd; /* that's how we know a test is in progress ;-) */
3783 fdtab[fd].owner = t;
willy tarreaubc4e1fb2005-12-17 13:32:07 +01003784 fdtab[fd].read = &event_srv_chk_r;
3785 fdtab[fd].write = &event_srv_chk_w;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003786 fdtab[fd].state = FD_STCONN; /* connection in progress */
3787 FD_SET(fd, StaticWriteEvent); /* for connect status */
3788 fd_insert(fd);
willy tarreaue47c8d72005-12-17 12:55:52 +01003789 /* FIXME: we allow up to <inter> for a connection to establish, but we should use another parameter */
3790 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003791 task_queue(t); /* restore t to its place in the task list */
3792 return tv_remain(&now, &t->expire);
3793 }
3794 else if (errno != EALREADY && errno != EISCONN && errno != EAGAIN) {
3795 s->result = -1; /* a real error */
3796 }
3797 }
3798 //fprintf(stderr, "process_chk: 5\n");
3799 close(fd);
3800 }
3801
3802 if (!s->result) { /* nothing done */
3803 //fprintf(stderr, "process_chk: 6\n");
willy tarreaue47c8d72005-12-17 12:55:52 +01003804 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003805 task_queue(t); /* restore t to its place in the task list */
3806 return tv_remain(&now, &t->expire);
3807 }
3808
3809 /* here, we have seen a failure */
willy tarreaue47c8d72005-12-17 12:55:52 +01003810 if (s->health > s->rise)
willy tarreau5cbea6f2005-12-17 12:48:26 +01003811 s->health--; /* still good */
3812 else {
willy tarreau535ae7a2005-12-17 12:58:00 +01003813 if (s->health == s->rise) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003814 if (!(global.mode & MODE_QUIET))
willy tarreau8337c6b2005-12-17 13:41:01 +01003815 Warning("server %s/%s DOWN.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003816
willy tarreau9fe663a2005-12-17 13:02:59 +01003817 send_log(s->proxy, LOG_ALERT, "Server %s/%s is DOWN.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003818 }
willy tarreauef900ab2005-12-17 12:52:52 +01003819
willy tarreau5cbea6f2005-12-17 12:48:26 +01003820 s->health = 0; /* failure */
3821 s->state &= ~SRV_RUNNING;
3822 }
3823
3824 //fprintf(stderr, "process_chk: 7\n");
willy tarreaue47c8d72005-12-17 12:55:52 +01003825 /* FIXME: we allow up to <inter> for a connection to establish, but we should use another parameter */
3826 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003827 }
3828 else {
3829 //fprintf(stderr, "process_chk: 8\n");
3830 /* there was a test running */
3831 if (s->result > 0) { /* good server detected */
3832 //fprintf(stderr, "process_chk: 9\n");
3833 s->health++; /* was bad, stays for a while */
willy tarreaue47c8d72005-12-17 12:55:52 +01003834 if (s->health >= s->rise) {
willy tarreau535ae7a2005-12-17 12:58:00 +01003835 if (s->health == s->rise) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003836 if (!(global.mode & MODE_QUIET))
willy tarreau8337c6b2005-12-17 13:41:01 +01003837 Warning("server %s/%s UP.\n", s->proxy->id, s->id);
willy tarreau9fe663a2005-12-17 13:02:59 +01003838 send_log(s->proxy, LOG_NOTICE, "Server %s/%s is UP.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003839 }
willy tarreauef900ab2005-12-17 12:52:52 +01003840
willy tarreaue47c8d72005-12-17 12:55:52 +01003841 s->health = s->rise + s->fall - 1; /* OK now */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003842 s->state |= SRV_RUNNING;
3843 }
willy tarreauef900ab2005-12-17 12:52:52 +01003844 s->curfd = -1; /* no check running anymore */
3845 //FD_CLR(fd, StaticWriteEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003846 fd_delete(fd);
willy tarreaue47c8d72005-12-17 12:55:52 +01003847 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003848 }
3849 else if (s->result < 0 || tv_cmp2_ms(&t->expire, &now) <= 0) {
3850 //fprintf(stderr, "process_chk: 10\n");
3851 /* failure or timeout detected */
willy tarreaue47c8d72005-12-17 12:55:52 +01003852 if (s->health > s->rise)
willy tarreau5cbea6f2005-12-17 12:48:26 +01003853 s->health--; /* still good */
3854 else {
willy tarreau535ae7a2005-12-17 12:58:00 +01003855 if (s->health == s->rise) {
willy tarreau9fe663a2005-12-17 13:02:59 +01003856 if (!(global.mode & MODE_QUIET))
willy tarreau8337c6b2005-12-17 13:41:01 +01003857 Warning("server %s/%s DOWN.\n", s->proxy->id, s->id);
willy tarreau9fe663a2005-12-17 13:02:59 +01003858
3859 send_log(s->proxy, LOG_ALERT, "Server %s/%s is DOWN.\n", s->proxy->id, s->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01003860 }
willy tarreauef900ab2005-12-17 12:52:52 +01003861
willy tarreau5cbea6f2005-12-17 12:48:26 +01003862 s->health = 0; /* failure */
3863 s->state &= ~SRV_RUNNING;
3864 }
3865 s->curfd = -1;
willy tarreauef900ab2005-12-17 12:52:52 +01003866 //FD_CLR(fd, StaticWriteEvent);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003867 fd_delete(fd);
willy tarreaue47c8d72005-12-17 12:55:52 +01003868 tv_delayfrom(&t->expire, &now, s->inter);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003869 }
3870 /* if result is 0 and there's no timeout, we have to wait again */
3871 }
3872 //fprintf(stderr, "process_chk: 11\n");
3873 s->result = 0;
3874 task_queue(t); /* restore t to its place in the task list */
3875 return tv_remain(&now, &t->expire);
willy tarreau0f7af912005-12-17 12:21:26 +01003876}
3877
3878
willy tarreau5cbea6f2005-12-17 12:48:26 +01003879
willy tarreau0f7af912005-12-17 12:21:26 +01003880#if STATTIME > 0
3881int stats(void);
3882#endif
3883
3884/*
3885 * Main select() loop.
3886 */
3887
3888void select_loop() {
3889 int next_time;
willy tarreau0f7af912005-12-17 12:21:26 +01003890 int time2;
willy tarreau0f7af912005-12-17 12:21:26 +01003891 int status;
3892 int fd,i;
3893 struct timeval delta;
3894 int readnotnull, writenotnull;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003895 struct task *t, *tnext;
willy tarreau0f7af912005-12-17 12:21:26 +01003896
willy tarreau5cbea6f2005-12-17 12:48:26 +01003897 tv_now(&now);
3898
3899 while (1) {
3900 next_time = -1; /* set the timer to wait eternally first */
willy tarreau0f7af912005-12-17 12:21:26 +01003901
willy tarreau5cbea6f2005-12-17 12:48:26 +01003902 /* look for expired tasks and add them to the run queue.
3903 */
3904 tnext = ((struct task *)LIST_HEAD(wait_queue))->next;
3905 while ((t = tnext) != LIST_HEAD(wait_queue)) { /* we haven't looped ? */
3906 tnext = t->next;
willy tarreauef900ab2005-12-17 12:52:52 +01003907 if (t->state & TASK_RUNNING)
3908 continue;
willy tarreau5cbea6f2005-12-17 12:48:26 +01003909
3910 /* wakeup expired entries. It doesn't matter if they are
3911 * already running because of a previous event
3912 */
3913 if (tv_cmp2_ms(&t->expire, &now) <= 0) {
willy tarreauef900ab2005-12-17 12:52:52 +01003914 //fprintf(stderr,"task_wakeup(%p, %p)\n", &rq, t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003915 task_wakeup(&rq, t);
3916 }
3917 else {
willy tarreauef900ab2005-12-17 12:52:52 +01003918 /* first non-runnable task. Use its expiration date as an upper bound */
3919 int temp_time = tv_remain(&now, &t->expire);
3920 if (temp_time)
3921 next_time = temp_time;
3922 //fprintf(stderr,"no_task_wakeup(%p, %p) : expire in %d ms\n", &rq, t, temp_time);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003923 break;
3924 }
3925 }
3926
3927 /* process each task in the run queue now. Each task may be deleted
3928 * since we only use tnext.
3929 */
3930 tnext = rq;
3931 while ((t = tnext) != NULL) {
3932 int temp_time;
3933
3934 tnext = t->rqnext;
3935 task_sleep(&rq, t);
willy tarreauef900ab2005-12-17 12:52:52 +01003936 //fprintf(stderr,"task %p\n",t);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003937 temp_time = t->process(t);
3938 next_time = MINTIME(temp_time, next_time);
willy tarreauef900ab2005-12-17 12:52:52 +01003939 //fprintf(stderr,"process(%p)=%d -> next_time=%d)\n", t, temp_time, next_time);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003940 }
3941
willy tarreauef900ab2005-12-17 12:52:52 +01003942 //fprintf(stderr,"---end of run---\n");
willy tarreau5cbea6f2005-12-17 12:48:26 +01003943
3944 /* maintain all proxies in a consistent state. This should quickly become a task */
3945 time2 = maintain_proxies();
3946 next_time = MINTIME(time2, next_time);
3947
3948 /* stop when there's no connection left and we don't allow them anymore */
3949 if (!actconn && listeners == 0)
3950 break;
3951
willy tarreau0f7af912005-12-17 12:21:26 +01003952
3953#if STATTIME > 0
3954 time2 = stats();
3955 // fprintf(stderr," stats = %d\n", time2);
3956 next_time = MINTIME(time2, next_time);
3957#endif
3958
willy tarreau5cbea6f2005-12-17 12:48:26 +01003959 if (next_time > 0) { /* FIXME */
willy tarreau0f7af912005-12-17 12:21:26 +01003960 /* Convert to timeval */
willy tarreau5cbea6f2005-12-17 12:48:26 +01003961 /* to avoid eventual select loops due to timer precision */
3962 next_time += SCHEDULER_RESOLUTION;
3963 delta.tv_sec = next_time / 1000;
3964 delta.tv_usec = (next_time % 1000) * 1000;
3965 }
3966 else if (next_time == 0) { /* allow select to return immediately when needed */
3967 delta.tv_sec = delta.tv_usec = 0;
willy tarreau0f7af912005-12-17 12:21:26 +01003968 }
3969
3970
3971 /* let's restore fdset state */
3972
3973 readnotnull = 0; writenotnull = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01003974 for (i = 0; i < (global.maxsock + FD_SETSIZE - 1)/(8*sizeof(int)); i++) {
willy tarreau0f7af912005-12-17 12:21:26 +01003975 readnotnull |= (*(((int*)ReadEvent)+i) = *(((int*)StaticReadEvent)+i)) != 0;
3976 writenotnull |= (*(((int*)WriteEvent)+i) = *(((int*)StaticWriteEvent)+i)) != 0;
3977 }
3978
3979// /* just a verification code, needs to be removed for performance */
3980// for (i=0; i<maxfd; i++) {
3981// if (FD_ISSET(i, ReadEvent) != FD_ISSET(i, StaticReadEvent))
3982// abort();
3983// if (FD_ISSET(i, WriteEvent) != FD_ISSET(i, StaticWriteEvent))
3984// abort();
3985//
3986// }
3987
3988 status=select(maxfd,
3989 readnotnull ? ReadEvent : NULL,
3990 writenotnull ? WriteEvent : NULL,
3991 NULL,
3992 (next_time >= 0) ? &delta : NULL);
3993
willy tarreau5cbea6f2005-12-17 12:48:26 +01003994 /* this is an experiment on the separation of the select work */
3995 // status = (readnotnull ? select(maxfd, ReadEvent, NULL, NULL, (next_time >= 0) ? &delta : NULL) : 0);
3996 // status |= (writenotnull ? select(maxfd, NULL, WriteEvent, NULL, (next_time >= 0) ? &delta : NULL) : 0);
3997
willy tarreau0f7af912005-12-17 12:21:26 +01003998 tv_now(&now);
willy tarreau5cbea6f2005-12-17 12:48:26 +01003999
willy tarreau0f7af912005-12-17 12:21:26 +01004000 if (status > 0) { /* must proceed with events */
4001
4002 int fds;
4003 char count;
4004
4005 for (fds = 0; (fds << INTBITS) < maxfd; fds++)
4006 if ((((int *)(ReadEvent))[fds] | ((int *)(WriteEvent))[fds]) != 0)
4007 for (count = 1<<INTBITS, fd = fds << INTBITS; count && fd < maxfd; count--, fd++) {
4008
willy tarreau5cbea6f2005-12-17 12:48:26 +01004009 /* if we specify read first, the accepts and zero reads will be
4010 * seen first. Moreover, system buffers will be flushed faster.
4011 */
willy tarreau0f7af912005-12-17 12:21:26 +01004012 if (fdtab[fd].state == FD_STCLOSE)
4013 continue;
willy tarreau0f7af912005-12-17 12:21:26 +01004014
4015 if (FD_ISSET(fd, ReadEvent))
4016 fdtab[fd].read(fd);
willy tarreau5cbea6f2005-12-17 12:48:26 +01004017
willy tarreau5cbea6f2005-12-17 12:48:26 +01004018 if (FD_ISSET(fd, WriteEvent))
4019 fdtab[fd].write(fd);
willy tarreau0f7af912005-12-17 12:21:26 +01004020 }
4021 }
4022 else {
4023 // fprintf(stderr,"select returned %d, maxfd=%d\n", status, maxfd);
4024 }
willy tarreau0f7af912005-12-17 12:21:26 +01004025 }
4026}
4027
4028
4029#if STATTIME > 0
4030/*
4031 * Display proxy statistics regularly. It is designed to be called from the
4032 * select_loop().
4033 */
4034int stats(void) {
4035 static int lines;
4036 static struct timeval nextevt;
4037 static struct timeval lastevt;
4038 static struct timeval starttime = {0,0};
4039 unsigned long totaltime, deltatime;
4040 int ret;
4041
willy tarreau750a4722005-12-17 13:21:24 +01004042 if (tv_cmp(&now, &nextevt) > 0) {
willy tarreau6e682ce2005-12-17 13:26:49 +01004043 deltatime = (tv_diff(&lastevt, &now)?:1);
4044 totaltime = (tv_diff(&starttime, &now)?:1);
willy tarreau0f7af912005-12-17 12:21:26 +01004045
willy tarreau9fe663a2005-12-17 13:02:59 +01004046 if (global.mode & MODE_STATS) {
4047 if ((lines++ % 16 == 0) && !(global.mode & MODE_LOG))
willy tarreau5cbea6f2005-12-17 12:48:26 +01004048 qfprintf(stderr,
willy tarreau0f7af912005-12-17 12:21:26 +01004049 "\n active total tsknew tskgood tskleft tskrght tsknsch tsklsch tskrsch\n");
4050 if (lines>1) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01004051 qfprintf(stderr,"%07d %07d %07d %07d %07d %07d %07d %07d %07d\n",
willy tarreau0f7af912005-12-17 12:21:26 +01004052 actconn, totalconn,
4053 stats_tsk_new, stats_tsk_good,
4054 stats_tsk_left, stats_tsk_right,
4055 stats_tsk_nsrch, stats_tsk_lsrch, stats_tsk_rsrch);
4056 }
4057 }
4058
4059 tv_delayfrom(&nextevt, &now, STATTIME);
4060
4061 lastevt=now;
4062 }
4063 ret = tv_remain(&now, &nextevt);
4064 return ret;
4065}
4066#endif
4067
4068
4069/*
4070 * this function enables proxies when there are enough free sessions,
4071 * or stops them when the table is full. It is designed to be called from the
willy tarreau5cbea6f2005-12-17 12:48:26 +01004072 * select_loop(). It returns the time left before next expiration event
4073 * during stop time, -1 otherwise.
willy tarreau0f7af912005-12-17 12:21:26 +01004074 */
4075static int maintain_proxies(void) {
4076 struct proxy *p;
willy tarreaua41a8b42005-12-17 14:02:24 +01004077 struct listener *l;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004078 int tleft; /* time left */
willy tarreau0f7af912005-12-17 12:21:26 +01004079
4080 p = proxy;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004081 tleft = -1; /* infinite time */
willy tarreau0f7af912005-12-17 12:21:26 +01004082
4083 /* if there are enough free sessions, we'll activate proxies */
willy tarreau9fe663a2005-12-17 13:02:59 +01004084 if (actconn < global.maxconn) {
willy tarreau0f7af912005-12-17 12:21:26 +01004085 while (p) {
4086 if (p->nbconn < p->maxconn) {
4087 if (p->state == PR_STIDLE) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004088 for (l = p->listen; l != NULL; l = l->next) {
4089 FD_SET(l->fd, StaticReadEvent);
4090 }
willy tarreau0f7af912005-12-17 12:21:26 +01004091 p->state = PR_STRUN;
4092 }
4093 }
4094 else {
4095 if (p->state == PR_STRUN) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004096 for (l = p->listen; l != NULL; l = l->next) {
4097 FD_CLR(l->fd, StaticReadEvent);
4098 }
willy tarreau0f7af912005-12-17 12:21:26 +01004099 p->state = PR_STIDLE;
4100 }
4101 }
4102 p = p->next;
4103 }
4104 }
4105 else { /* block all proxies */
4106 while (p) {
4107 if (p->state == PR_STRUN) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004108 for (l = p->listen; l != NULL; l = l->next) {
4109 FD_CLR(l->fd, StaticReadEvent);
4110 }
willy tarreau0f7af912005-12-17 12:21:26 +01004111 p->state = PR_STIDLE;
4112 }
4113 p = p->next;
4114 }
4115 }
4116
willy tarreau5cbea6f2005-12-17 12:48:26 +01004117 if (stopping) {
4118 p = proxy;
4119 while (p) {
4120 if (p->state != PR_STDISABLED) {
4121 int t;
4122 t = tv_remain(&now, &p->stop_time);
4123 if (t == 0) {
willy tarreau535ae7a2005-12-17 12:58:00 +01004124 Warning("Proxy %s stopped.\n", p->id);
willy tarreau9fe663a2005-12-17 13:02:59 +01004125 send_log(p, LOG_WARNING, "Proxy %s stopped.\n", p->id);
willy tarreau535ae7a2005-12-17 12:58:00 +01004126
willy tarreaua41a8b42005-12-17 14:02:24 +01004127 for (l = p->listen; l != NULL; l = l->next) {
4128 fd_delete(l->fd);
4129 listeners--;
4130 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004131 p->state = PR_STDISABLED;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004132 }
4133 else {
4134 tleft = MINTIME(t, tleft);
4135 }
4136 }
4137 p = p->next;
4138 }
4139 }
4140 return tleft;
willy tarreau0f7af912005-12-17 12:21:26 +01004141}
4142
4143/*
4144 * this function disables health-check servers so that the process will quickly be ignored
4145 * by load balancers.
4146 */
4147static void soft_stop(void) {
4148 struct proxy *p;
4149
4150 stopping = 1;
4151 p = proxy;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004152 tv_now(&now); /* else, the old time before select will be used */
willy tarreau0f7af912005-12-17 12:21:26 +01004153 while (p) {
willy tarreau535ae7a2005-12-17 12:58:00 +01004154 if (p->state != PR_STDISABLED) {
4155 Warning("Stopping proxy %s in %d ms.\n", p->id, p->grace);
willy tarreau9fe663a2005-12-17 13:02:59 +01004156 send_log(p, LOG_WARNING, "Stopping proxy %s in %d ms.\n", p->id, p->grace);
willy tarreau0f7af912005-12-17 12:21:26 +01004157 tv_delayfrom(&p->stop_time, &now, p->grace);
willy tarreau535ae7a2005-12-17 12:58:00 +01004158 }
willy tarreau0f7af912005-12-17 12:21:26 +01004159 p = p->next;
4160 }
4161}
4162
4163/*
4164 * upon SIGUSR1, let's have a soft stop.
4165 */
4166void sig_soft_stop(int sig) {
4167 soft_stop();
4168 signal(sig, SIG_IGN);
4169}
4170
4171
willy tarreau8337c6b2005-12-17 13:41:01 +01004172/*
4173 * this function dumps every server's state when the process receives SIGHUP.
4174 */
4175void sig_dump_state(int sig) {
4176 struct proxy *p = proxy;
4177
4178 Warning("SIGHUP received, dumping servers states.\n");
4179 while (p) {
4180 struct server *s = p->srv;
4181
4182 send_log(p, LOG_NOTICE, "SIGUP received, dumping servers states.\n");
4183 while (s) {
4184 if (s->state & SRV_RUNNING) {
4185 Warning("SIGHUP: server %s/%s is UP.\n", p->id, s->id);
4186 send_log(p, LOG_NOTICE, "SIGUP: server %s/%s is UP.\n", p->id, s->id);
4187 }
4188 else {
4189 Warning("SIGHUP: server %s/%s is DOWN.\n", p->id, s->id);
4190 send_log(p, LOG_NOTICE, "SIGHUP: server %s/%s is DOWN.\n", p->id, s->id);
4191 }
4192 s = s->next;
4193 }
4194 p = p->next;
4195 }
4196 signal(sig, sig_dump_state);
4197}
4198
willy tarreau0f7af912005-12-17 12:21:26 +01004199void dump(int sig) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01004200 struct task *t, *tnext;
4201 struct session *s;
willy tarreau0f7af912005-12-17 12:21:26 +01004202
willy tarreau5cbea6f2005-12-17 12:48:26 +01004203 tnext = ((struct task *)LIST_HEAD(wait_queue))->next;
4204 while ((t = tnext) != LIST_HEAD(wait_queue)) { /* we haven't looped ? */
4205 tnext = t->next;
4206 s = t->context;
4207 qfprintf(stderr,"[dump] wq: task %p, still %ld ms, "
4208 "cli=%d, srv=%d, cr=%d, cw=%d, sr=%d, sw=%d, "
4209 "req=%d, rep=%d, clifd=%d\n",
4210 s, tv_remain(&now, &t->expire),
4211 s->cli_state,
4212 s->srv_state,
4213 FD_ISSET(s->cli_fd, StaticReadEvent),
4214 FD_ISSET(s->cli_fd, StaticWriteEvent),
4215 FD_ISSET(s->srv_fd, StaticReadEvent),
4216 FD_ISSET(s->srv_fd, StaticWriteEvent),
4217 s->req->l, s->rep?s->rep->l:0, s->cli_fd
4218 );
willy tarreau0f7af912005-12-17 12:21:26 +01004219 }
4220}
4221
willy tarreaue39cd132005-12-17 13:00:18 +01004222void chain_regex(struct hdr_exp **head, regex_t *preg, int action, char *replace) {
4223 struct hdr_exp *exp;
4224
4225 while (*head != NULL)
4226 head = &(*head)->next;
4227
4228 exp = calloc(1, sizeof(struct hdr_exp));
4229
4230 exp->preg = preg;
4231 exp->replace = replace;
4232 exp->action = action;
4233 *head = exp;
4234}
4235
willy tarreau9fe663a2005-12-17 13:02:59 +01004236
willy tarreau0f7af912005-12-17 12:21:26 +01004237/*
willy tarreau9fe663a2005-12-17 13:02:59 +01004238 * parse a line in a <global> section. Returns 0 if OK, -1 if error.
willy tarreau0f7af912005-12-17 12:21:26 +01004239 */
willy tarreau9fe663a2005-12-17 13:02:59 +01004240int cfg_parse_global(char *file, int linenum, char **args) {
willy tarreau0f7af912005-12-17 12:21:26 +01004241
willy tarreau9fe663a2005-12-17 13:02:59 +01004242 if (!strcmp(args[0], "global")) { /* new section */
4243 /* no option, nothing special to do */
4244 return 0;
4245 }
4246 else if (!strcmp(args[0], "daemon")) {
4247 global.mode |= MODE_DAEMON;
4248 }
4249 else if (!strcmp(args[0], "debug")) {
4250 global.mode |= MODE_DEBUG;
4251 }
4252 else if (!strcmp(args[0], "quiet")) {
4253 global.mode |= MODE_QUIET;
4254 }
4255 else if (!strcmp(args[0], "stats")) {
4256 global.mode |= MODE_STATS;
4257 }
4258 else if (!strcmp(args[0], "uid")) {
4259 if (global.uid != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004260 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004261 return 0;
willy tarreau0f7af912005-12-17 12:21:26 +01004262 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004263 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004264 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004265 return -1;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004266 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004267 global.uid = atol(args[1]);
4268 }
4269 else if (!strcmp(args[0], "gid")) {
4270 if (global.gid != 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 tarreau0f7af912005-12-17 12:21:26 +01004276 return -1;
4277 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004278 global.gid = atol(args[1]);
4279 }
4280 else if (!strcmp(args[0], "nbproc")) {
4281 if (global.nbproc != 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 tarreau9fe663a2005-12-17 13:02:59 +01004287 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004288 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004289 global.nbproc = atol(args[1]);
4290 }
4291 else if (!strcmp(args[0], "maxconn")) {
4292 if (global.maxconn != 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.maxconn = atol(args[1]);
4301 }
4302 else if (!strcmp(args[0], "chroot")) {
4303 if (global.chroot != NULL) {
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;
4306 }
4307 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004308 Alert("parsing [%s:%d] : '%s' expects a directory as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004309 return -1;
4310 }
4311 global.chroot = strdup(args[1]);
4312 }
4313 else if (!strcmp(args[0], "log")) { /* syslog server address */
4314 struct sockaddr_in *sa;
willy tarreau8337c6b2005-12-17 13:41:01 +01004315 int facility, level;
willy tarreau9fe663a2005-12-17 13:02:59 +01004316
4317 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004318 Alert("parsing [%s:%d] : '%s' expects <address> and <facility> as arguments.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004319 return -1;
4320 }
4321
4322 for (facility = 0; facility < NB_LOG_FACILITIES; facility++)
4323 if (!strcmp(log_facilities[facility], args[2]))
4324 break;
4325
4326 if (facility >= NB_LOG_FACILITIES) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004327 Alert("parsing [%s:%d] : unknown log facility '%s'\n", file, linenum, args[2]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004328 exit(1);
4329 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004330
4331 level = 7; /* max syslog level = debug */
4332 if (*(args[3])) {
4333 while (level >= 0 && strcmp(log_levels[level], args[3]))
4334 level--;
4335 if (level < 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004336 Alert("parsing [%s:%d] : unknown optional log level '%s'\n", file, linenum, args[3]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004337 exit(1);
4338 }
4339 }
4340
willy tarreau9fe663a2005-12-17 13:02:59 +01004341 sa = str2sa(args[1]);
4342 if (!sa->sin_port)
4343 sa->sin_port = htons(SYSLOG_PORT);
4344
4345 if (global.logfac1 == -1) {
4346 global.logsrv1 = *sa;
4347 global.logfac1 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004348 global.loglev1 = level;
willy tarreau9fe663a2005-12-17 13:02:59 +01004349 }
4350 else if (global.logfac2 == -1) {
4351 global.logsrv2 = *sa;
4352 global.logfac2 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004353 global.loglev2 = level;
willy tarreau9fe663a2005-12-17 13:02:59 +01004354 }
4355 else {
4356 Alert("parsing [%s:%d] : too many syslog servers\n", file, linenum);
4357 return -1;
4358 }
4359
4360 }
4361 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004362 Alert("parsing [%s:%d] : unknown keyword '%s' in '%s' section\n", file, linenum, args[0], "global");
willy tarreau9fe663a2005-12-17 13:02:59 +01004363 return -1;
4364 }
4365 return 0;
4366}
4367
4368
willy tarreaua41a8b42005-12-17 14:02:24 +01004369void init_default_instance() {
4370 memset(&defproxy, 0, sizeof(defproxy));
4371 defproxy.mode = PR_MODE_TCP;
4372 defproxy.state = PR_STNEW;
4373 defproxy.maxconn = cfg_maxpconn;
4374 defproxy.conn_retries = CONN_RETRIES;
4375 defproxy.logfac1 = defproxy.logfac2 = -1; /* log disabled */
4376}
4377
willy tarreau9fe663a2005-12-17 13:02:59 +01004378/*
4379 * parse a line in a <listen> section. Returns 0 if OK, -1 if error.
4380 */
4381int cfg_parse_listen(char *file, int linenum, char **args) {
4382 static struct proxy *curproxy = NULL;
4383 struct server *newsrv = NULL;
4384
4385 if (!strcmp(args[0], "listen")) { /* new proxy */
willy tarreaua41a8b42005-12-17 14:02:24 +01004386 if (!*args[1]) {
4387 Alert("parsing [%s:%d] : '%s' expects an <id> argument and\n"
4388 " optionnally supports [addr1]:port1[-end1]{,[addr]:port[-end]}...\n",
willy tarreau036e1ce2005-12-17 13:46:33 +01004389 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004390 return -1;
4391 }
4392
4393 if ((curproxy = (struct proxy *)calloc(1, sizeof(struct proxy))) == NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004394 Alert("parsing [%s:%d] : out of memory.\n", file, linenum);
willy tarreau9fe663a2005-12-17 13:02:59 +01004395 return -1;
4396 }
4397 curproxy->next = proxy;
4398 proxy = curproxy;
4399 curproxy->id = strdup(args[1]);
willy tarreaua41a8b42005-12-17 14:02:24 +01004400 if (strchr(args[2], ':') != NULL)
4401 curproxy->listen = str2listener(args[2], curproxy->listen);
4402
willy tarreau9fe663a2005-12-17 13:02:59 +01004403 /* set default values */
willy tarreaua41a8b42005-12-17 14:02:24 +01004404 curproxy->state = defproxy.state;
4405 curproxy->maxconn = defproxy.maxconn;
4406 curproxy->conn_retries = defproxy.conn_retries;
4407 curproxy->options = defproxy.options;
willy tarreaueedaa9f2005-12-17 14:08:03 +01004408
4409 if (defproxy.check_req)
4410 curproxy->check_req = strdup(defproxy.check_req);
4411 curproxy->check_len = defproxy.check_len;
4412
4413 if (defproxy.cookie_name)
4414 curproxy->cookie_name = strdup(defproxy.cookie_name);
4415 curproxy->cookie_len = defproxy.cookie_len;
4416
4417 if (defproxy.capture_name)
4418 curproxy->capture_name = strdup(defproxy.capture_name);
4419 curproxy->capture_namelen = defproxy.capture_namelen;
4420 curproxy->capture_len = defproxy.capture_len;
4421
4422 if (defproxy.errmsg.msg400)
4423 curproxy->errmsg.msg400 = strdup(defproxy.errmsg.msg400);
4424 curproxy->errmsg.len400 = defproxy.errmsg.len400;
4425
4426 if (defproxy.errmsg.msg403)
4427 curproxy->errmsg.msg403 = strdup(defproxy.errmsg.msg403);
4428 curproxy->errmsg.len403 = defproxy.errmsg.len403;
4429
4430 if (defproxy.errmsg.msg408)
4431 curproxy->errmsg.msg408 = strdup(defproxy.errmsg.msg408);
4432 curproxy->errmsg.len408 = defproxy.errmsg.len408;
4433
4434 if (defproxy.errmsg.msg500)
4435 curproxy->errmsg.msg500 = strdup(defproxy.errmsg.msg500);
4436 curproxy->errmsg.len500 = defproxy.errmsg.len500;
4437
4438 if (defproxy.errmsg.msg502)
4439 curproxy->errmsg.msg502 = strdup(defproxy.errmsg.msg502);
4440 curproxy->errmsg.len502 = defproxy.errmsg.len502;
4441
4442 if (defproxy.errmsg.msg503)
4443 curproxy->errmsg.msg503 = strdup(defproxy.errmsg.msg503);
4444 curproxy->errmsg.len503 = defproxy.errmsg.len503;
4445
4446 if (defproxy.errmsg.msg504)
4447 curproxy->errmsg.msg504 = strdup(defproxy.errmsg.msg504);
4448 curproxy->errmsg.len504 = defproxy.errmsg.len504;
4449
willy tarreaua41a8b42005-12-17 14:02:24 +01004450 curproxy->clitimeout = defproxy.clitimeout;
4451 curproxy->contimeout = defproxy.contimeout;
4452 curproxy->srvtimeout = defproxy.srvtimeout;
4453 curproxy->mode = defproxy.mode;
4454 curproxy->logfac1 = defproxy.logfac1;
4455 curproxy->logsrv1 = defproxy.logsrv1;
4456 curproxy->loglev1 = defproxy.loglev1;
4457 curproxy->logfac2 = defproxy.logfac2;
4458 curproxy->logsrv2 = defproxy.logsrv2;
4459 curproxy->loglev2 = defproxy.loglev2;
4460 curproxy->to_log = defproxy.to_log;
4461 curproxy->grace = defproxy.grace;
4462 curproxy->source_addr = defproxy.source_addr;
4463 return 0;
4464 }
4465 else if (!strcmp(args[0], "defaults")) { /* use this one to assign default values */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004466 /* some variables may have already been initialized earlier */
4467 if (defproxy.check_req) free(defproxy.check_req);
4468 if (defproxy.cookie_name) free(defproxy.cookie_name);
4469 if (defproxy.capture_name) free(defproxy.capture_name);
4470 if (defproxy.errmsg.msg400) free(defproxy.errmsg.msg400);
4471 if (defproxy.errmsg.msg403) free(defproxy.errmsg.msg403);
4472 if (defproxy.errmsg.msg408) free(defproxy.errmsg.msg408);
4473 if (defproxy.errmsg.msg500) free(defproxy.errmsg.msg500);
4474 if (defproxy.errmsg.msg502) free(defproxy.errmsg.msg502);
4475 if (defproxy.errmsg.msg503) free(defproxy.errmsg.msg503);
4476 if (defproxy.errmsg.msg504) free(defproxy.errmsg.msg504);
4477
4478 init_default_instance();
willy tarreaua41a8b42005-12-17 14:02:24 +01004479 curproxy = &defproxy;
willy tarreau9fe663a2005-12-17 13:02:59 +01004480 return 0;
4481 }
4482 else if (curproxy == NULL) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004483 Alert("parsing [%s:%d] : 'listen' or 'defaults' expected.\n", file, linenum);
willy tarreau9fe663a2005-12-17 13:02:59 +01004484 return -1;
4485 }
4486
willy tarreaua41a8b42005-12-17 14:02:24 +01004487 if (!strcmp(args[0], "bind")) { /* new listen addresses */
4488 if (curproxy == &defproxy) {
4489 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4490 return -1;
4491 }
4492
4493 if (strchr(args[1], ':') == NULL) {
4494 Alert("parsing [%s:%d] : '%s' expects [addr1]:port1[-end1]{,[addr]:port[-end]}... as arguments.\n",
4495 file, linenum, args[0]);
4496 return -1;
4497 }
4498 curproxy->listen = str2listener(args[1], curproxy->listen);
4499 return 0;
4500 }
4501 else if (!strcmp(args[0], "mode")) { /* sets the proxy mode */
willy tarreau9fe663a2005-12-17 13:02:59 +01004502 if (!strcmp(args[1], "http")) curproxy->mode = PR_MODE_HTTP;
4503 else if (!strcmp(args[1], "tcp")) curproxy->mode = PR_MODE_TCP;
4504 else if (!strcmp(args[1], "health")) curproxy->mode = PR_MODE_HEALTH;
4505 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004506 Alert("parsing [%s:%d] : unknown proxy mode '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004507 return -1;
4508 }
4509 }
4510 else if (!strcmp(args[0], "disabled")) { /* disables this proxy */
4511 curproxy->state = PR_STDISABLED;
4512 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004513 else if (!strcmp(args[0], "enabled")) { /* enables this proxy (used to revert a disabled default) */
4514 curproxy->state = PR_STNEW;
4515 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004516 else if (!strcmp(args[0], "cookie")) { /* cookie name */
4517 int cur_arg;
willy tarreaueedaa9f2005-12-17 14:08:03 +01004518// if (curproxy == &defproxy) {
4519// Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4520// return -1;
4521// }
willy tarreaua41a8b42005-12-17 14:02:24 +01004522
willy tarreau9fe663a2005-12-17 13:02:59 +01004523 if (curproxy->cookie_name != NULL) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01004524// Alert("parsing [%s:%d] : cookie name already specified. Continuing.\n",
4525// file, linenum);
4526// return 0;
4527 free(curproxy->cookie_name);
willy tarreau9fe663a2005-12-17 13:02:59 +01004528 }
4529
4530 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004531 Alert("parsing [%s:%d] : '%s' expects <cookie_name> as argument.\n",
4532 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004533 return -1;
4534 }
4535 curproxy->cookie_name = strdup(args[1]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004536 curproxy->cookie_len = strlen(curproxy->cookie_name);
willy tarreau9fe663a2005-12-17 13:02:59 +01004537
4538 cur_arg = 2;
4539 while (*(args[cur_arg])) {
4540 if (!strcmp(args[cur_arg], "rewrite")) {
4541 curproxy->options |= PR_O_COOK_RW;
willy tarreau0f7af912005-12-17 12:21:26 +01004542 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004543 else if (!strcmp(args[cur_arg], "indirect")) {
4544 curproxy->options |= PR_O_COOK_IND;
willy tarreau0f7af912005-12-17 12:21:26 +01004545 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004546 else if (!strcmp(args[cur_arg], "insert")) {
4547 curproxy->options |= PR_O_COOK_INS;
willy tarreau0f7af912005-12-17 12:21:26 +01004548 }
willy tarreau240afa62005-12-17 13:14:35 +01004549 else if (!strcmp(args[cur_arg], "nocache")) {
4550 curproxy->options |= PR_O_COOK_NOC;
4551 }
willy tarreaucd878942005-12-17 13:27:43 +01004552 else if (!strcmp(args[cur_arg], "postonly")) {
4553 curproxy->options |= PR_O_COOK_POST;
4554 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004555 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004556 Alert("parsing [%s:%d] : '%s' supports 'rewrite', 'insert', 'indirect', 'nocache' and 'postonly' options.\n",
4557 file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01004558 return -1;
4559 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004560 cur_arg++;
4561 }
4562 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 +01004563 Alert("parsing [%s:%d] : cookie 'rewrite' and 'indirect' mode are incompatible.\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01004564 file, linenum);
4565 return -1;
4566 }
4567 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004568 else if (!strcmp(args[0], "capture")) { /* name of a cookie to capture */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004569// if (curproxy == &defproxy) {
4570// Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4571// return -1;
4572// }
willy tarreaua41a8b42005-12-17 14:02:24 +01004573
willy tarreau8337c6b2005-12-17 13:41:01 +01004574 if (curproxy->capture_name != NULL) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01004575// Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n",
4576// file, linenum, args[0]);
4577// return 0;
4578 free(curproxy->capture_name);
willy tarreau8337c6b2005-12-17 13:41:01 +01004579 }
4580
4581 if (*(args[4]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004582 Alert("parsing [%s:%d] : '%s' expects 'cookie' <cookie_name> 'len' <len>.\n",
4583 file, linenum, args[0]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004584 return -1;
4585 }
4586 curproxy->capture_name = strdup(args[2]);
4587 curproxy->capture_namelen = strlen(curproxy->capture_name);
4588 curproxy->capture_len = atol(args[4]);
4589 if (curproxy->capture_len >= CAPTURE_LEN) {
4590 Warning("parsing [%s:%d] : truncating capture length to %d bytes.\n",
4591 file, linenum, CAPTURE_LEN - 1);
4592 curproxy->capture_len = CAPTURE_LEN - 1;
4593 }
4594 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004595 else if (!strcmp(args[0], "contimeout")) { /* connect timeout */
willy tarreaua41a8b42005-12-17 14:02:24 +01004596 if (curproxy->contimeout != defproxy.contimeout) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004597 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004598 return 0;
4599 }
4600 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004601 Alert("parsing [%s:%d] : '%s' expects an integer <time_in_ms> as argument.\n",
4602 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004603 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004604 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004605 curproxy->contimeout = atol(args[1]);
4606 }
4607 else if (!strcmp(args[0], "clitimeout")) { /* client timeout */
willy tarreaua41a8b42005-12-17 14:02:24 +01004608 if (curproxy->clitimeout != defproxy.clitimeout) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004609 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n",
4610 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004611 return 0;
4612 }
4613 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004614 Alert("parsing [%s:%d] : '%s' expects an integer <time_in_ms> as argument.\n",
4615 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004616 return -1;
4617 }
4618 curproxy->clitimeout = atol(args[1]);
4619 }
4620 else if (!strcmp(args[0], "srvtimeout")) { /* server timeout */
willy tarreaua41a8b42005-12-17 14:02:24 +01004621 if (curproxy->srvtimeout != defproxy.srvtimeout) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004622 Alert("parsing [%s:%d] : '%s' already specified. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004623 return 0;
4624 }
4625 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004626 Alert("parsing [%s:%d] : '%s' expects an integer <time_in_ms> as argument.\n",
4627 file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01004628 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004629 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004630 curproxy->srvtimeout = atol(args[1]);
4631 }
4632 else if (!strcmp(args[0], "retries")) { /* connection retries */
4633 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004634 Alert("parsing [%s:%d] : '%s' expects an integer argument (dispatch counts for one).\n",
4635 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004636 return -1;
4637 }
4638 curproxy->conn_retries = atol(args[1]);
4639 }
4640 else if (!strcmp(args[0], "option")) {
4641 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004642 Alert("parsing [%s:%d] : '%s' expects an option name.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004643 return -1;
4644 }
4645 if (!strcmp(args[1], "redispatch"))
willy tarreau5cbea6f2005-12-17 12:48:26 +01004646 /* enable reconnections to dispatch */
4647 curproxy->options |= PR_O_REDISP;
willy tarreaua1598082005-12-17 13:08:06 +01004648#ifdef TPROXY
willy tarreau9fe663a2005-12-17 13:02:59 +01004649 else if (!strcmp(args[1], "transparent"))
willy tarreau5cbea6f2005-12-17 12:48:26 +01004650 /* enable transparent proxy connections */
4651 curproxy->options |= PR_O_TRANSP;
willy tarreau9fe663a2005-12-17 13:02:59 +01004652#endif
4653 else if (!strcmp(args[1], "keepalive"))
4654 /* enable keep-alive */
4655 curproxy->options |= PR_O_KEEPALIVE;
4656 else if (!strcmp(args[1], "forwardfor"))
4657 /* insert x-forwarded-for field */
4658 curproxy->options |= PR_O_FWDFOR;
4659 else if (!strcmp(args[1], "httplog")) {
4660 /* generate a complete HTTP log */
willy tarreaua1598082005-12-17 13:08:06 +01004661 curproxy->to_log |= LW_DATE | LW_CLIP | LW_SVID | LW_REQ | LW_PXID | LW_RESP;
4662 }
4663 else if (!strcmp(args[1], "dontlognull")) {
4664 /* don't log empty requests */
4665 curproxy->options |= PR_O_NULLNOLOG;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004666 }
willy tarreaubc4e1fb2005-12-17 13:32:07 +01004667 else if (!strcmp(args[1], "httpchk")) {
4668 /* use HTTP request to check servers' health */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004669 if (curproxy->check_req != NULL) {
4670 free(curproxy->check_req);
4671 }
willy tarreaubc4e1fb2005-12-17 13:32:07 +01004672 curproxy->options |= PR_O_HTTP_CHK;
willy tarreaueedaa9f2005-12-17 14:08:03 +01004673 if (!*args[2]) { /* no argument */
4674 curproxy->check_req = strdup(DEF_CHECK_REQ); /* default request */
4675 curproxy->check_len = strlen(DEF_CHECK_REQ);
4676 } else if (!*args[3]) { /* one argument : URI */
willy tarreau2f6ba652005-12-17 13:57:42 +01004677 int reqlen = strlen(args[2]) + strlen("OPTIONS / HTTP/1.0\r\n\r\n");
4678 curproxy->check_req = (char *)malloc(reqlen);
4679 curproxy->check_len = snprintf(curproxy->check_req, reqlen,
4680 "OPTIONS %s HTTP/1.0\r\n\r\n", args[2]); /* URI to use */
willy tarreaueedaa9f2005-12-17 14:08:03 +01004681 } else { /* more arguments : METHOD URI [HTTP_VER] */
4682 int reqlen = strlen(args[2]) + strlen(args[3]) + 3 + strlen("\r\n\r\n");
4683 if (*args[4])
4684 reqlen += strlen(args[4]);
4685 else
4686 reqlen += strlen("HTTP/1.0");
4687
4688 curproxy->check_req = (char *)malloc(reqlen);
4689 curproxy->check_len = snprintf(curproxy->check_req, reqlen,
4690 "%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 +01004691 }
willy tarreaubc4e1fb2005-12-17 13:32:07 +01004692 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004693 else if (!strcmp(args[1], "persist")) {
4694 /* persist on using the server specified by the cookie, even when it's down */
4695 curproxy->options |= PR_O_PERSIST;
4696 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004697 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004698 Alert("parsing [%s:%d] : unknown option '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004699 return -1;
4700 }
4701 return 0;
4702 }
4703 else if (!strcmp(args[0], "redispatch") || !strcmp(args[0], "redisp")) {
4704 /* enable reconnections to dispatch */
4705 curproxy->options |= PR_O_REDISP;
4706 }
willy tarreaua1598082005-12-17 13:08:06 +01004707#ifdef TPROXY
willy tarreau9fe663a2005-12-17 13:02:59 +01004708 else if (!strcmp(args[0], "transparent")) {
4709 /* enable transparent proxy connections */
4710 curproxy->options |= PR_O_TRANSP;
4711 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004712#endif
willy tarreau9fe663a2005-12-17 13:02:59 +01004713 else if (!strcmp(args[0], "maxconn")) { /* maxconn */
4714 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004715 Alert("parsing [%s:%d] : '%s' expects an integer argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004716 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004717 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004718 curproxy->maxconn = atol(args[1]);
4719 }
4720 else if (!strcmp(args[0], "grace")) { /* grace time (ms) */
4721 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004722 Alert("parsing [%s:%d] : '%s' expects a time in milliseconds.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004723 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004724 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004725 curproxy->grace = atol(args[1]);
4726 }
4727 else if (!strcmp(args[0], "dispatch")) { /* dispatch address */
willy tarreaua41a8b42005-12-17 14:02:24 +01004728 if (curproxy == &defproxy) {
4729 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4730 return -1;
4731 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004732 if (strchr(args[1], ':') == NULL) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004733 Alert("parsing [%s:%d] : '%s' expects <addr:port> as argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004734 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004735 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004736 curproxy->dispatch_addr = *str2sa(args[1]);
4737 }
willy tarreau036e1ce2005-12-17 13:46:33 +01004738 else if (!strcmp(args[0], "balance")) { /* set balancing with optional algorithm */
willy tarreau9fe663a2005-12-17 13:02:59 +01004739 if (*(args[1])) {
4740 if (!strcmp(args[1], "roundrobin")) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01004741 curproxy->options |= PR_O_BALANCE_RR;
willy tarreau9fe663a2005-12-17 13:02:59 +01004742 }
4743 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004744 Alert("parsing [%s:%d] : '%s' only supports 'roundrobin' option.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004745 return -1;
4746 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004747 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004748 else /* if no option is set, use round-robin by default */
4749 curproxy->options |= PR_O_BALANCE_RR;
4750 }
4751 else if (!strcmp(args[0], "server")) { /* server address */
4752 int cur_arg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004753 char *rport;
4754 char *raddr;
4755 short realport;
4756 int do_check;
4757
4758 if (curproxy == &defproxy) {
4759 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4760 return -1;
4761 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01004762
willy tarreaua41a8b42005-12-17 14:02:24 +01004763 if (!*args[2]) {
4764 Alert("parsing [%s:%d] : '%s' expects <name> and <addr>[:<port>] as arguments.\n",
willy tarreau036e1ce2005-12-17 13:46:33 +01004765 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004766 return -1;
4767 }
4768 if ((newsrv = (struct server *)calloc(1, sizeof(struct server))) == NULL) {
4769 Alert("parsing [%s:%d] : out of memory.\n", file, linenum);
4770 return -1;
4771 }
4772 newsrv->next = curproxy->srv;
4773 curproxy->srv = newsrv;
4774 newsrv->proxy = curproxy;
willy tarreaua41a8b42005-12-17 14:02:24 +01004775
4776 do_check = 0;
willy tarreau9fe663a2005-12-17 13:02:59 +01004777 newsrv->state = SRV_RUNNING; /* early server setup */
willy tarreaua41a8b42005-12-17 14:02:24 +01004778 newsrv->id = strdup(args[1]);
4779
4780 /* several ways to check the port component :
4781 * - IP => port=+0, relative
4782 * - IP: => port=+0, relative
4783 * - IP:N => port=N, absolute
4784 * - IP:+N => port=+N, relative
4785 * - IP:-N => port=-N, relative
4786 */
4787 raddr = strdup(args[2]);
4788 rport = strchr(raddr, ':');
4789 if (rport) {
4790 *rport++ = 0;
4791 realport = atol(rport);
4792 if (!isdigit((int)*rport))
4793 newsrv->state |= SRV_MAPPORTS;
4794 } else {
4795 realport = 0;
4796 newsrv->state |= SRV_MAPPORTS;
4797 }
4798
4799 newsrv->addr = *str2sa(raddr);
4800 newsrv->addr.sin_port = htons(realport);
4801 free(raddr);
4802
willy tarreau9fe663a2005-12-17 13:02:59 +01004803 newsrv->curfd = -1; /* no health-check in progress */
4804 newsrv->inter = DEF_CHKINTR;
4805 newsrv->rise = DEF_RISETIME;
4806 newsrv->fall = DEF_FALLTIME;
4807 newsrv->health = newsrv->rise; /* up, but will fall down at first failure */
4808 cur_arg = 3;
4809 while (*args[cur_arg]) {
4810 if (!strcmp(args[cur_arg], "cookie")) {
4811 newsrv->cookie = strdup(args[cur_arg + 1]);
4812 newsrv->cklen = strlen(args[cur_arg + 1]);
4813 cur_arg += 2;
willy tarreau0f7af912005-12-17 12:21:26 +01004814 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004815 else if (!strcmp(args[cur_arg], "rise")) {
4816 newsrv->rise = atol(args[cur_arg + 1]);
4817 newsrv->health = newsrv->rise;
4818 cur_arg += 2;
willy tarreau0f7af912005-12-17 12:21:26 +01004819 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004820 else if (!strcmp(args[cur_arg], "fall")) {
4821 newsrv->fall = atol(args[cur_arg + 1]);
4822 cur_arg += 2;
4823 }
4824 else if (!strcmp(args[cur_arg], "inter")) {
4825 newsrv->inter = atol(args[cur_arg + 1]);
4826 cur_arg += 2;
4827 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004828 else if (!strcmp(args[cur_arg], "port")) {
4829 newsrv->check_port = atol(args[cur_arg + 1]);
4830 cur_arg += 2;
4831 }
willy tarreau8337c6b2005-12-17 13:41:01 +01004832 else if (!strcmp(args[cur_arg], "backup")) {
4833 newsrv->state |= SRV_BACKUP;
4834 cur_arg ++;
4835 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004836 else if (!strcmp(args[cur_arg], "check")) {
willy tarreaua41a8b42005-12-17 14:02:24 +01004837 do_check = 1;
willy tarreau9fe663a2005-12-17 13:02:59 +01004838 cur_arg += 1;
willy tarreau5cbea6f2005-12-17 12:48:26 +01004839 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004840 else {
willy tarreaua41a8b42005-12-17 14:02:24 +01004841 Alert("parsing [%s:%d] : server %s only supports options 'backup', 'cookie', 'check', 'inter', 'rise' and 'fall'.\n",
4842 file, linenum, newsrv->id);
4843 return -1;
4844 }
4845 }
4846
4847 if (do_check) {
4848 struct task *t;
4849
4850 if (!newsrv->check_port && !(newsrv->state & SRV_MAPPORTS))
4851 newsrv->check_port = realport; /* by default */
4852 if (!newsrv->check_port) {
4853 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 +01004854 file, linenum, newsrv->id);
willy tarreau0f7af912005-12-17 12:21:26 +01004855 return -1;
4856 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004857
4858 if ((t = pool_alloc(task)) == NULL) {
4859 Alert("parsing [%s:%d] : out of memory.\n", file, linenum);
4860 return -1;
4861 }
4862
4863 t->next = t->prev = t->rqnext = NULL; /* task not in run queue yet */
4864 t->wq = LIST_HEAD(wait_queue); /* but already has a wait queue assigned */
4865 t->state = TASK_IDLE;
4866 t->process = process_chk;
4867 t->context = newsrv;
4868
4869 if (curproxy->state != PR_STDISABLED) {
4870 tv_delayfrom(&t->expire, &now, newsrv->inter); /* check this every ms */
4871 task_queue(t);
4872 task_wakeup(&rq, t);
4873 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004874 }
willy tarreaua41a8b42005-12-17 14:02:24 +01004875
willy tarreau9fe663a2005-12-17 13:02:59 +01004876 curproxy->nbservers++;
4877 }
4878 else if (!strcmp(args[0], "log")) { /* syslog server address */
4879 struct sockaddr_in *sa;
4880 int facility;
4881
4882 if (*(args[1]) && *(args[2]) == 0 && !strcmp(args[1], "global")) {
4883 curproxy->logfac1 = global.logfac1;
4884 curproxy->logsrv1 = global.logsrv1;
willy tarreau8337c6b2005-12-17 13:41:01 +01004885 curproxy->loglev1 = global.loglev1;
willy tarreau9fe663a2005-12-17 13:02:59 +01004886 curproxy->logfac2 = global.logfac2;
4887 curproxy->logsrv2 = global.logsrv2;
willy tarreau8337c6b2005-12-17 13:41:01 +01004888 curproxy->loglev2 = global.loglev2;
willy tarreau9fe663a2005-12-17 13:02:59 +01004889 }
4890 else if (*(args[1]) && *(args[2])) {
willy tarreau8337c6b2005-12-17 13:41:01 +01004891 int level;
4892
willy tarreau0f7af912005-12-17 12:21:26 +01004893 for (facility = 0; facility < NB_LOG_FACILITIES; facility++)
4894 if (!strcmp(log_facilities[facility], args[2]))
4895 break;
willy tarreau9fe663a2005-12-17 13:02:59 +01004896
willy tarreau0f7af912005-12-17 12:21:26 +01004897 if (facility >= NB_LOG_FACILITIES) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004898 Alert("parsing [%s:%d] : unknown log facility '%s'\n", file, linenum, args[2]);
willy tarreau0f7af912005-12-17 12:21:26 +01004899 exit(1);
4900 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004901
willy tarreau8337c6b2005-12-17 13:41:01 +01004902 level = 7; /* max syslog level = debug */
4903 if (*(args[3])) {
4904 while (level >= 0 && strcmp(log_levels[level], args[3]))
4905 level--;
4906 if (level < 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004907 Alert("parsing [%s:%d] : unknown optional log level '%s'\n", file, linenum, args[3]);
willy tarreau8337c6b2005-12-17 13:41:01 +01004908 exit(1);
4909 }
4910 }
4911
willy tarreau0f7af912005-12-17 12:21:26 +01004912 sa = str2sa(args[1]);
4913 if (!sa->sin_port)
4914 sa->sin_port = htons(SYSLOG_PORT);
willy tarreau9fe663a2005-12-17 13:02:59 +01004915
willy tarreau0f7af912005-12-17 12:21:26 +01004916 if (curproxy->logfac1 == -1) {
4917 curproxy->logsrv1 = *sa;
4918 curproxy->logfac1 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004919 curproxy->loglev1 = level;
willy tarreau0f7af912005-12-17 12:21:26 +01004920 }
4921 else if (curproxy->logfac2 == -1) {
4922 curproxy->logsrv2 = *sa;
4923 curproxy->logfac2 = facility;
willy tarreau8337c6b2005-12-17 13:41:01 +01004924 curproxy->loglev2 = level;
willy tarreau0f7af912005-12-17 12:21:26 +01004925 }
4926 else {
4927 Alert("parsing [%s:%d] : too many syslog servers\n", file, linenum);
willy tarreau9fe663a2005-12-17 13:02:59 +01004928 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01004929 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004930 }
4931 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01004932 Alert("parsing [%s:%d] : 'log' expects either <address[:port]> and <facility> or 'global' as arguments.\n",
willy tarreau9fe663a2005-12-17 13:02:59 +01004933 file, linenum);
4934 return -1;
4935 }
4936 }
willy tarreaua1598082005-12-17 13:08:06 +01004937 else if (!strcmp(args[0], "source")) { /* address to which we bind when connecting */
willy tarreaua41a8b42005-12-17 14:02:24 +01004938 if (!*args[1]) {
4939 Alert("parsing [%s:%d] : '%s' expects <addr>[:<port>] as argument.\n",
willy tarreau036e1ce2005-12-17 13:46:33 +01004940 file, linenum, "source");
willy tarreaua1598082005-12-17 13:08:06 +01004941 return -1;
4942 }
4943
4944 curproxy->source_addr = *str2sa(args[1]);
4945 curproxy->options |= PR_O_BIND_SRC;
4946 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004947 else if (!strcmp(args[0], "cliexp") || !strcmp(args[0], "reqrep")) { /* replace request header from a regex */
4948 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004949 if (curproxy == &defproxy) {
4950 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4951 return -1;
4952 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004953
4954 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004955 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
4956 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004957 return -1;
4958 }
4959
4960 preg = calloc(1, sizeof(regex_t));
4961 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004962 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004963 return -1;
4964 }
4965
4966 chain_regex(&curproxy->req_exp, preg, ACT_REPLACE, strdup(args[2]));
4967 }
4968 else if (!strcmp(args[0], "reqdel")) { /* delete request header from a regex */
4969 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004970 if (curproxy == &defproxy) {
4971 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4972 return -1;
4973 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004974
4975 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004976 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004977 return -1;
4978 }
4979
4980 preg = calloc(1, sizeof(regex_t));
4981 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004982 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004983 return -1;
4984 }
4985
4986 chain_regex(&curproxy->req_exp, preg, ACT_REMOVE, NULL);
4987 }
4988 else if (!strcmp(args[0], "reqdeny")) { /* deny a request if a header matches this regex */
4989 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01004990 if (curproxy == &defproxy) {
4991 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
4992 return -1;
4993 }
willy tarreau9fe663a2005-12-17 13:02:59 +01004994
4995 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01004996 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01004997 return -1;
4998 }
4999
5000 preg = calloc(1, sizeof(regex_t));
5001 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005002 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005003 return -1;
5004 }
5005
5006 chain_regex(&curproxy->req_exp, preg, ACT_DENY, NULL);
5007 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005008 else if (!strcmp(args[0], "reqpass")) { /* pass this header without allowing or denying the request */
5009 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005010 if (curproxy == &defproxy) {
5011 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5012 return -1;
5013 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005014
5015 if (*(args[1]) == 0) {
5016 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
5017 return -1;
5018 }
5019
5020 preg = calloc(1, sizeof(regex_t));
5021 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
5022 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
5023 return -1;
5024 }
5025
5026 chain_regex(&curproxy->req_exp, preg, ACT_PASS, NULL);
5027 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005028 else if (!strcmp(args[0], "reqallow")) { /* allow a request if a header matches this regex */
5029 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005030 if (curproxy == &defproxy) {
5031 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5032 return -1;
5033 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005034
5035 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005036 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005037 return -1;
5038 }
5039
5040 preg = calloc(1, sizeof(regex_t));
5041 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005042 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005043 return -1;
5044 }
5045
5046 chain_regex(&curproxy->req_exp, preg, ACT_ALLOW, NULL);
5047 }
5048 else if (!strcmp(args[0], "reqirep")) { /* replace request header from a regex, ignoring case */
5049 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005050 if (curproxy == &defproxy) {
5051 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5052 return -1;
5053 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005054
5055 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005056 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
5057 file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005058 return -1;
5059 }
5060
5061 preg = calloc(1, sizeof(regex_t));
5062 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005063 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005064 return -1;
5065 }
5066
5067 chain_regex(&curproxy->req_exp, preg, ACT_REPLACE, strdup(args[2]));
5068 }
5069 else if (!strcmp(args[0], "reqidel")) { /* delete request header from a regex ignoring case */
5070 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005071 if (curproxy == &defproxy) {
5072 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5073 return -1;
5074 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005075
5076 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005077 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005078 return -1;
5079 }
5080
5081 preg = calloc(1, sizeof(regex_t));
5082 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005083 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005084 return -1;
5085 }
5086
5087 chain_regex(&curproxy->req_exp, preg, ACT_REMOVE, NULL);
5088 }
5089 else if (!strcmp(args[0], "reqideny")) { /* deny a request if a header matches this regex ignoring case */
5090 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005091 if (curproxy == &defproxy) {
5092 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5093 return -1;
5094 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005095
5096 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005097 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005098 return -1;
5099 }
5100
5101 preg = calloc(1, sizeof(regex_t));
5102 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005103 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005104 return -1;
5105 }
5106
5107 chain_regex(&curproxy->req_exp, preg, ACT_DENY, NULL);
5108 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005109 else if (!strcmp(args[0], "reqipass")) { /* pass this header without allowing or denying the request */
5110 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005111 if (curproxy == &defproxy) {
5112 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5113 return -1;
5114 }
willy tarreau036e1ce2005-12-17 13:46:33 +01005115
5116 if (*(args[1]) == 0) {
5117 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
5118 return -1;
5119 }
5120
5121 preg = calloc(1, sizeof(regex_t));
5122 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
5123 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
5124 return -1;
5125 }
5126
5127 chain_regex(&curproxy->req_exp, preg, ACT_PASS, NULL);
5128 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005129 else if (!strcmp(args[0], "reqiallow")) { /* allow a request if a header matches this regex ignoring case */
5130 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005131 if (curproxy == &defproxy) {
5132 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5133 return -1;
5134 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005135
5136 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005137 Alert("parsing [%s:%d] : '%s' expects <regex> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005138 return -1;
5139 }
5140
5141 preg = calloc(1, sizeof(regex_t));
5142 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005143 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005144 return -1;
5145 }
5146
5147 chain_regex(&curproxy->req_exp, preg, ACT_ALLOW, NULL);
5148 }
5149 else if (!strcmp(args[0], "reqadd")) { /* add request header */
willy tarreaua41a8b42005-12-17 14:02:24 +01005150 if (curproxy == &defproxy) {
5151 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5152 return -1;
5153 }
5154
willy tarreau9fe663a2005-12-17 13:02:59 +01005155 if (curproxy->nb_reqadd >= MAX_NEWHDR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005156 Alert("parsing [%s:%d] : too many '%s'. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005157 return 0;
5158 }
5159
5160 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005161 Alert("parsing [%s:%d] : '%s' expects <header> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005162 return -1;
5163 }
5164
5165 curproxy->req_add[curproxy->nb_reqadd++] = strdup(args[1]);
willy tarreau0f7af912005-12-17 12:21:26 +01005166 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005167 else if (!strcmp(args[0], "srvexp") || !strcmp(args[0], "rsprep")) { /* replace response header from a regex */
willy tarreau0f7af912005-12-17 12:21:26 +01005168 regex_t *preg;
willy tarreau0f7af912005-12-17 12:21:26 +01005169
5170 if (*(args[1]) == 0 || *(args[2]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005171 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
5172 file, linenum, args[0]);
willy tarreau0f7af912005-12-17 12:21:26 +01005173 return -1;
5174 }
5175
5176 preg = calloc(1, sizeof(regex_t));
5177 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005178 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau0f7af912005-12-17 12:21:26 +01005179 return -1;
5180 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005181
5182 chain_regex(&curproxy->rsp_exp, preg, ACT_REPLACE, strdup(args[2]));
5183 }
5184 else if (!strcmp(args[0], "rspdel")) { /* delete response header from a regex */
5185 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005186 if (curproxy == &defproxy) {
5187 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5188 return -1;
5189 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005190
5191 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005192 Alert("parsing [%s:%d] : '%s' expects <search> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005193 return -1;
5194 }
willy tarreaue39cd132005-12-17 13:00:18 +01005195
willy tarreau9fe663a2005-12-17 13:02:59 +01005196 preg = calloc(1, sizeof(regex_t));
5197 if (regcomp(preg, args[1], REG_EXTENDED) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005198 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005199 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01005200 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005201
5202 chain_regex(&curproxy->rsp_exp, preg, ACT_REMOVE, strdup(args[2]));
5203 }
5204 else if (!strcmp(args[0], "rspirep")) { /* replace response header from a regex ignoring case */
willy tarreaua41a8b42005-12-17 14:02:24 +01005205 regex_t *preg;
5206 if (curproxy == &defproxy) {
5207 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5208 return -1;
5209 }
willy tarreaue39cd132005-12-17 13:00:18 +01005210
willy tarreaua41a8b42005-12-17 14:02:24 +01005211 if (*(args[1]) == 0 || *(args[2]) == 0) {
5212 Alert("parsing [%s:%d] : '%s' expects <search> and <replace> as arguments.\n",
5213 file, linenum, args[0]);
5214 return -1;
5215 }
willy tarreaue39cd132005-12-17 13:00:18 +01005216
willy tarreaua41a8b42005-12-17 14:02:24 +01005217 preg = calloc(1, sizeof(regex_t));
5218 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
5219 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
5220 return -1;
willy tarreau9fe663a2005-12-17 13:02:59 +01005221 }
willy tarreaua41a8b42005-12-17 14:02:24 +01005222
5223 chain_regex(&curproxy->rsp_exp, preg, ACT_REPLACE, strdup(args[2]));
5224 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005225 else if (!strcmp(args[0], "rspidel")) { /* delete response header from a regex ignoring case */
5226 regex_t *preg;
willy tarreaua41a8b42005-12-17 14:02:24 +01005227 if (curproxy == &defproxy) {
5228 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5229 return -1;
5230 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005231
5232 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005233 Alert("parsing [%s:%d] : '%s' expects <search> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005234 return -1;
5235 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005236
willy tarreau9fe663a2005-12-17 13:02:59 +01005237 preg = calloc(1, sizeof(regex_t));
5238 if (regcomp(preg, args[1], REG_EXTENDED | REG_ICASE) != 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005239 Alert("parsing [%s:%d] : bad regular expression '%s'.\n", file, linenum, args[1]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005240 return -1;
willy tarreaue39cd132005-12-17 13:00:18 +01005241 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005242
5243 chain_regex(&curproxy->rsp_exp, preg, ACT_REMOVE, strdup(args[2]));
5244 }
5245 else if (!strcmp(args[0], "rspadd")) { /* add response header */
willy tarreaua41a8b42005-12-17 14:02:24 +01005246 if (curproxy == &defproxy) {
5247 Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5248 return -1;
5249 }
5250
willy tarreau9fe663a2005-12-17 13:02:59 +01005251 if (curproxy->nb_rspadd >= MAX_NEWHDR) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005252 Alert("parsing [%s:%d] : too many '%s'. Continuing.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005253 return 0;
5254 }
5255
5256 if (*(args[1]) == 0) {
willy tarreau036e1ce2005-12-17 13:46:33 +01005257 Alert("parsing [%s:%d] : '%s' expects <header> as an argument.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005258 return -1;
5259 }
5260
5261 curproxy->rsp_add[curproxy->nb_rspadd++] = strdup(args[1]);
5262 }
willy tarreau8337c6b2005-12-17 13:41:01 +01005263 else if (!strcmp(args[0], "errorloc")) { /* error location */
5264 int errnum;
5265 char *err;
5266
willy tarreaueedaa9f2005-12-17 14:08:03 +01005267 // if (curproxy == &defproxy) {
5268 // Alert("parsing [%s:%d] : '%s' not allowed in 'defaults' section.\n", file, linenum, args[0]);
5269 // return -1;
5270 // }
willy tarreaua41a8b42005-12-17 14:02:24 +01005271
willy tarreau8337c6b2005-12-17 13:41:01 +01005272 if (*(args[2]) == 0) {
5273 Alert("parsing [%s:%d] : <errorloc> expects <error> and <url> as arguments.\n", file, linenum);
5274 return -1;
5275 }
5276
5277 errnum = atol(args[1]);
5278 err = malloc(strlen(HTTP_302) + strlen(args[2]) + 5);
5279 sprintf(err, "%s%s\r\n\r\n", HTTP_302, args[2]);
5280
5281 if (errnum == 400) {
5282 if (curproxy->errmsg.msg400) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005283 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005284 free(curproxy->errmsg.msg400);
5285 }
5286 curproxy->errmsg.msg400 = err;
5287 curproxy->errmsg.len400 = strlen(err);
5288 }
5289 else if (errnum == 403) {
5290 if (curproxy->errmsg.msg403) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005291 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005292 free(curproxy->errmsg.msg403);
5293 }
5294 curproxy->errmsg.msg403 = err;
5295 curproxy->errmsg.len403 = strlen(err);
5296 }
5297 else if (errnum == 408) {
5298 if (curproxy->errmsg.msg408) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005299 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005300 free(curproxy->errmsg.msg408);
5301 }
5302 curproxy->errmsg.msg408 = err;
5303 curproxy->errmsg.len408 = strlen(err);
5304 }
5305 else if (errnum == 500) {
5306 if (curproxy->errmsg.msg500) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005307 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005308 free(curproxy->errmsg.msg500);
5309 }
5310 curproxy->errmsg.msg500 = err;
5311 curproxy->errmsg.len500 = strlen(err);
5312 }
5313 else if (errnum == 502) {
5314 if (curproxy->errmsg.msg502) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005315 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005316 free(curproxy->errmsg.msg502);
5317 }
5318 curproxy->errmsg.msg502 = err;
5319 curproxy->errmsg.len502 = strlen(err);
5320 }
5321 else if (errnum == 503) {
5322 if (curproxy->errmsg.msg503) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005323 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005324 free(curproxy->errmsg.msg503);
5325 }
5326 curproxy->errmsg.msg503 = err;
5327 curproxy->errmsg.len503 = strlen(err);
5328 }
5329 else if (errnum == 504) {
5330 if (curproxy->errmsg.msg504) {
willy tarreaueedaa9f2005-12-17 14:08:03 +01005331 //Warning("parsing [%s:%d] : error %d already defined.\n", file, linenum, errnum);
willy tarreau8337c6b2005-12-17 13:41:01 +01005332 free(curproxy->errmsg.msg504);
5333 }
5334 curproxy->errmsg.msg504 = err;
5335 curproxy->errmsg.len504 = strlen(err);
5336 }
5337 else {
5338 Warning("parsing [%s:%d] : error %d relocation will be ignored.\n", file, linenum, errnum);
5339 free(err);
5340 }
5341 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005342 else {
willy tarreau036e1ce2005-12-17 13:46:33 +01005343 Alert("parsing [%s:%d] : unknown keyword '%s' in '%s' section\n", file, linenum, args[0], "listen");
willy tarreau9fe663a2005-12-17 13:02:59 +01005344 return -1;
5345 }
5346 return 0;
5347}
willy tarreaue39cd132005-12-17 13:00:18 +01005348
willy tarreau5cbea6f2005-12-17 12:48:26 +01005349
willy tarreau9fe663a2005-12-17 13:02:59 +01005350/*
5351 * This function reads and parses the configuration file given in the argument.
5352 * returns 0 if OK, -1 if error.
5353 */
5354int readcfgfile(char *file) {
5355 char thisline[256];
5356 char *line;
5357 FILE *f;
5358 int linenum = 0;
5359 char *end;
5360 char *args[MAX_LINE_ARGS];
5361 int arg;
5362 int cfgerr = 0;
5363 int confsect = CFG_NONE;
willy tarreaue39cd132005-12-17 13:00:18 +01005364
willy tarreau9fe663a2005-12-17 13:02:59 +01005365 struct proxy *curproxy = NULL;
5366 struct server *newsrv = NULL;
willy tarreaue39cd132005-12-17 13:00:18 +01005367
willy tarreau9fe663a2005-12-17 13:02:59 +01005368 if ((f=fopen(file,"r")) == NULL)
5369 return -1;
willy tarreaue39cd132005-12-17 13:00:18 +01005370
willy tarreaueedaa9f2005-12-17 14:08:03 +01005371 init_default_instance();
5372
willy tarreau9fe663a2005-12-17 13:02:59 +01005373 while (fgets(line = thisline, sizeof(thisline), f) != NULL) {
5374 linenum++;
willy tarreau5cbea6f2005-12-17 12:48:26 +01005375
willy tarreau9fe663a2005-12-17 13:02:59 +01005376 end = line + strlen(line);
willy tarreau5cbea6f2005-12-17 12:48:26 +01005377
willy tarreau9fe663a2005-12-17 13:02:59 +01005378 /* skip leading spaces */
willy tarreauc29948c2005-12-17 13:10:27 +01005379 while (isspace((int)*line))
willy tarreau9fe663a2005-12-17 13:02:59 +01005380 line++;
5381
5382 arg = 0;
5383 args[arg] = line;
willy tarreau0f7af912005-12-17 12:21:26 +01005384
willy tarreau9fe663a2005-12-17 13:02:59 +01005385 while (*line && arg < MAX_LINE_ARGS) {
5386 /* first, we'll replace \\, \<space>, \#, \r, \n, \t, \xXX with their
5387 * C equivalent value. Other combinations left unchanged (eg: \1).
5388 */
5389 if (*line == '\\') {
5390 int skip = 0;
5391 if (line[1] == ' ' || line[1] == '\\' || line[1] == '#') {
5392 *line = line[1];
5393 skip = 1;
5394 }
5395 else if (line[1] == 'r') {
5396 *line = '\r';
5397 skip = 1;
5398 }
5399 else if (line[1] == 'n') {
5400 *line = '\n';
5401 skip = 1;
5402 }
5403 else if (line[1] == 't') {
5404 *line = '\t';
5405 skip = 1;
5406 }
5407 else if (line[1] == 'x' && (line + 3 < end )) {
5408 unsigned char hex1, hex2;
5409 hex1 = toupper(line[2]) - '0'; hex2 = toupper(line[3]) - '0';
5410 if (hex1 > 9) hex1 -= 'A' - '9' - 1;
5411 if (hex2 > 9) hex2 -= 'A' - '9' - 1;
5412 *line = (hex1<<4) + hex2;
5413 skip = 3;
5414 }
5415 if (skip) {
5416 memmove(line + 1, line + 1 + skip, end - (line + skip + 1));
5417 end -= skip;
5418 }
5419 line++;
willy tarreau0f7af912005-12-17 12:21:26 +01005420 }
willy tarreaua1598082005-12-17 13:08:06 +01005421 else if (*line == '#' || *line == '\n' || *line == '\r') {
5422 /* end of string, end of loop */
5423 *line = 0;
5424 break;
5425 }
willy tarreauc29948c2005-12-17 13:10:27 +01005426 else if (isspace((int)*line)) {
willy tarreau9fe663a2005-12-17 13:02:59 +01005427 /* a non-escaped space is an argument separator */
willy tarreaua1598082005-12-17 13:08:06 +01005428 *line++ = 0;
willy tarreauc29948c2005-12-17 13:10:27 +01005429 while (isspace((int)*line))
willy tarreaua1598082005-12-17 13:08:06 +01005430 line++;
5431 args[++arg] = line;
5432 }
5433 else {
5434 line++;
willy tarreau0f7af912005-12-17 12:21:26 +01005435 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005436 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005437
willy tarreau9fe663a2005-12-17 13:02:59 +01005438 /* empty line */
5439 if (!**args)
5440 continue;
willy tarreaue39cd132005-12-17 13:00:18 +01005441
willy tarreau9fe663a2005-12-17 13:02:59 +01005442 /* zero out remaining args */
5443 while (++arg < MAX_LINE_ARGS) {
5444 args[arg] = line;
willy tarreau5cbea6f2005-12-17 12:48:26 +01005445 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005446
willy tarreaua41a8b42005-12-17 14:02:24 +01005447 if (!strcmp(args[0], "listen") || !strcmp(args[0], "defaults")) /* new proxy */
willy tarreau9fe663a2005-12-17 13:02:59 +01005448 confsect = CFG_LISTEN;
5449 else if (!strcmp(args[0], "global")) /* global config */
5450 confsect = CFG_GLOBAL;
5451 /* else it's a section keyword */
willy tarreau5cbea6f2005-12-17 12:48:26 +01005452
willy tarreau9fe663a2005-12-17 13:02:59 +01005453 switch (confsect) {
5454 case CFG_LISTEN:
5455 if (cfg_parse_listen(file, linenum, args) < 0)
5456 return -1;
5457 break;
5458 case CFG_GLOBAL:
5459 if (cfg_parse_global(file, linenum, args) < 0)
5460 return -1;
5461 break;
5462 default:
willy tarreau036e1ce2005-12-17 13:46:33 +01005463 Alert("parsing [%s:%d] : unknown keyword '%s' out of section.\n", file, linenum, args[0]);
willy tarreau9fe663a2005-12-17 13:02:59 +01005464 return -1;
willy tarreau0f7af912005-12-17 12:21:26 +01005465 }
willy tarreau9fe663a2005-12-17 13:02:59 +01005466
5467
willy tarreau0f7af912005-12-17 12:21:26 +01005468 }
5469 fclose(f);
5470
5471 /*
5472 * Now, check for the integrity of all that we have collected.
5473 */
5474
5475 if ((curproxy = proxy) == NULL) {
5476 Alert("parsing %s : no <listen> line. Nothing to do !\n",
5477 file);
5478 return -1;
5479 }
5480
5481 while (curproxy != NULL) {
willy tarreauef900ab2005-12-17 12:52:52 +01005482 if (curproxy->state == PR_STDISABLED) {
5483 curproxy = curproxy->next;
5484 continue;
5485 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005486 if ((curproxy->mode != PR_MODE_HEALTH) &&
5487 !(curproxy->options & (PR_O_TRANSP | PR_O_BALANCE)) &&
willy tarreaua1598082005-12-17 13:08:06 +01005488 (*(int *)&curproxy->dispatch_addr.sin_addr == 0)) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01005489 Alert("parsing %s : listener %s has no dispatch address and is not in transparent or balance mode.\n",
5490 file, curproxy->id);
5491 cfgerr++;
5492 }
5493 else if ((curproxy->mode != PR_MODE_HEALTH) && (curproxy->options & PR_O_BALANCE)) {
5494 if (curproxy->options & PR_O_TRANSP) {
5495 Alert("parsing %s : listener %s cannot use both transparent and balance mode.\n",
5496 file, curproxy->id);
5497 cfgerr++;
5498 }
5499 else if (curproxy->srv == NULL) {
5500 Alert("parsing %s : listener %s needs at least 1 server in balance mode.\n",
5501 file, curproxy->id);
5502 cfgerr++;
5503 }
willy tarreaua1598082005-12-17 13:08:06 +01005504 else if (*(int *)&curproxy->dispatch_addr.sin_addr != 0) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01005505 Warning("parsing %s : dispatch address of listener %s will be ignored in balance mode.\n",
5506 file, curproxy->id);
5507 }
5508 }
5509 else if (curproxy->mode == PR_MODE_TCP || curproxy->mode == PR_MODE_HEALTH) { /* TCP PROXY or HEALTH CHECK */
willy tarreau0f7af912005-12-17 12:21:26 +01005510 if (curproxy->cookie_name != NULL) {
5511 Warning("parsing %s : cookie will be ignored for listener %s.\n",
5512 file, curproxy->id);
5513 }
5514 if ((newsrv = curproxy->srv) != NULL) {
5515 Warning("parsing %s : servers will be ignored for listener %s.\n",
5516 file, curproxy->id);
5517 }
willy tarreaue39cd132005-12-17 13:00:18 +01005518 if (curproxy->rsp_exp != NULL) {
willy tarreau0f7af912005-12-17 12:21:26 +01005519 Warning("parsing %s : server regular expressions will be ignored for listener %s.\n",
5520 file, curproxy->id);
5521 }
willy tarreaue39cd132005-12-17 13:00:18 +01005522 if (curproxy->req_exp != NULL) {
willy tarreau0f7af912005-12-17 12:21:26 +01005523 Warning("parsing %s : client regular expressions will be ignored for listener %s.\n",
5524 file, curproxy->id);
5525 }
5526 }
5527 else if (curproxy->mode == PR_MODE_HTTP) { /* HTTP PROXY */
5528 if ((curproxy->cookie_name != NULL) && ((newsrv = curproxy->srv) == NULL)) {
5529 Alert("parsing %s : HTTP proxy %s has a cookie but no server list !\n",
5530 file, curproxy->id);
5531 cfgerr++;
5532 }
5533 else {
5534 while (newsrv != NULL) {
5535 /* nothing to check for now */
5536 newsrv = newsrv->next;
5537 }
5538 }
5539 }
willy tarreau8337c6b2005-12-17 13:41:01 +01005540 if (curproxy->errmsg.msg400 == NULL) {
5541 curproxy->errmsg.msg400 = (char *)HTTP_400;
5542 curproxy->errmsg.len400 = strlen(HTTP_400);
5543 }
5544 if (curproxy->errmsg.msg403 == NULL) {
5545 curproxy->errmsg.msg403 = (char *)HTTP_403;
5546 curproxy->errmsg.len403 = strlen(HTTP_403);
5547 }
5548 if (curproxy->errmsg.msg408 == NULL) {
5549 curproxy->errmsg.msg408 = (char *)HTTP_408;
5550 curproxy->errmsg.len408 = strlen(HTTP_408);
5551 }
5552 if (curproxy->errmsg.msg500 == NULL) {
5553 curproxy->errmsg.msg500 = (char *)HTTP_500;
5554 curproxy->errmsg.len500 = strlen(HTTP_500);
5555 }
5556 if (curproxy->errmsg.msg502 == NULL) {
5557 curproxy->errmsg.msg502 = (char *)HTTP_502;
5558 curproxy->errmsg.len502 = strlen(HTTP_502);
5559 }
5560 if (curproxy->errmsg.msg503 == NULL) {
5561 curproxy->errmsg.msg503 = (char *)HTTP_503;
5562 curproxy->errmsg.len503 = strlen(HTTP_503);
5563 }
5564 if (curproxy->errmsg.msg504 == NULL) {
5565 curproxy->errmsg.msg504 = (char *)HTTP_504;
5566 curproxy->errmsg.len504 = strlen(HTTP_504);
5567 }
willy tarreau0f7af912005-12-17 12:21:26 +01005568 curproxy = curproxy->next;
5569 }
5570 if (cfgerr > 0) {
5571 Alert("Errors found in configuration file, aborting.\n");
5572 return -1;
5573 }
5574 else
5575 return 0;
5576}
5577
5578
5579/*
5580 * This function initializes all the necessary variables. It only returns
5581 * if everything is OK. If something fails, it exits.
5582 */
5583void init(int argc, char **argv) {
5584 int i;
willy tarreau9fe663a2005-12-17 13:02:59 +01005585 int arg_mode = 0; /* MODE_DEBUG, ... */
willy tarreau0f7af912005-12-17 12:21:26 +01005586 char *old_argv = *argv;
5587 char *tmp;
willy tarreau9fe663a2005-12-17 13:02:59 +01005588 int cfg_maxconn = 0; /* # of simultaneous connections, (-n) */
willy tarreau0f7af912005-12-17 12:21:26 +01005589
5590 if (1<<INTBITS != sizeof(int)*8) {
willy tarreau5cbea6f2005-12-17 12:48:26 +01005591 qfprintf(stderr,
willy tarreau0f7af912005-12-17 12:21:26 +01005592 "Error: wrong architecture. Recompile so that sizeof(int)=%d\n",
5593 sizeof(int)*8);
5594 exit(1);
5595 }
5596
5597 pid = getpid();
5598 progname = *argv;
5599 while ((tmp = strchr(progname, '/')) != NULL)
5600 progname = tmp + 1;
5601
5602 argc--; argv++;
5603 while (argc > 0) {
5604 char *flag;
5605
5606 if (**argv == '-') {
5607 flag = *argv+1;
5608
5609 /* 1 arg */
5610 if (*flag == 'v') {
5611 display_version();
5612 exit(0);
5613 }
5614 else if (*flag == 'd')
willy tarreau9fe663a2005-12-17 13:02:59 +01005615 arg_mode |= MODE_DEBUG;
willy tarreau0f7af912005-12-17 12:21:26 +01005616 else if (*flag == 'D')
willy tarreau9fe663a2005-12-17 13:02:59 +01005617 arg_mode |= MODE_DAEMON | MODE_QUIET;
willy tarreau5cbea6f2005-12-17 12:48:26 +01005618 else if (*flag == 'q')
willy tarreau9fe663a2005-12-17 13:02:59 +01005619 arg_mode |= MODE_QUIET;
willy tarreau0f7af912005-12-17 12:21:26 +01005620#if STATTIME > 0
5621 else if (*flag == 's')
willy tarreau9fe663a2005-12-17 13:02:59 +01005622 arg_mode |= MODE_STATS;
willy tarreau0f7af912005-12-17 12:21:26 +01005623 else if (*flag == 'l')
willy tarreau9fe663a2005-12-17 13:02:59 +01005624 arg_mode |= MODE_LOG;
willy tarreau0f7af912005-12-17 12:21:26 +01005625#endif
5626 else { /* >=2 args */
5627 argv++; argc--;
5628 if (argc == 0)
5629 usage(old_argv);
5630
5631 switch (*flag) {
5632 case 'n' : cfg_maxconn = atol(*argv); break;
5633 case 'N' : cfg_maxpconn = atol(*argv); break;
5634 case 'f' : cfg_cfgfile = *argv; break;
5635 default: usage(old_argv);
5636 }
5637 }
5638 }
5639 else
5640 usage(old_argv);
5641 argv++; argc--;
5642 }
5643
willy tarreau0f7af912005-12-17 12:21:26 +01005644 if (!cfg_cfgfile)
5645 usage(old_argv);
5646
5647 gethostname(hostname, MAX_HOSTNAME_LEN);
5648
5649 if (readcfgfile(cfg_cfgfile) < 0) {
5650 Alert("Error reading configuration file : %s\n", cfg_cfgfile);
5651 exit(1);
5652 }
5653
willy tarreau9fe663a2005-12-17 13:02:59 +01005654 if (cfg_maxconn > 0)
5655 global.maxconn = cfg_maxconn;
5656
5657 if (global.maxconn == 0)
5658 global.maxconn = DEFAULT_MAXCONN;
5659
5660 global.maxsock = global.maxconn * 2; /* each connection needs two sockets */
5661
5662 if (arg_mode & MODE_DEBUG) {
5663 /* command line debug mode inhibits configuration mode */
5664 global.mode &= ~(MODE_DAEMON | MODE_QUIET);
5665 }
willy tarreau750a4722005-12-17 13:21:24 +01005666 global.mode |= (arg_mode & (MODE_DAEMON | MODE_QUIET | MODE_DEBUG | MODE_STATS | MODE_LOG));
willy tarreau9fe663a2005-12-17 13:02:59 +01005667
5668 if ((global.mode & MODE_DEBUG) && (global.mode & (MODE_DAEMON | MODE_QUIET))) {
5669 Warning("<debug> mode incompatible with <quiet> and <daemon>. Keeping <debug> only.\n");
5670 global.mode &= ~(MODE_DAEMON | MODE_QUIET);
5671 }
5672
5673 if ((global.nbproc > 1) && !(global.mode & MODE_DAEMON)) {
5674 Warning("<nbproc> is only meaningful in daemon mode. Setting limit to 1 process.\n");
5675 global.nbproc = 1;
5676 }
5677
5678 if (global.nbproc < 1)
5679 global.nbproc = 1;
5680
willy tarreau0f7af912005-12-17 12:21:26 +01005681 ReadEvent = (fd_set *)calloc(1,
5682 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005683 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005684 WriteEvent = (fd_set *)calloc(1,
5685 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005686 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005687 StaticReadEvent = (fd_set *)calloc(1,
5688 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005689 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005690 StaticWriteEvent = (fd_set *)calloc(1,
5691 sizeof(fd_set) *
willy tarreau9fe663a2005-12-17 13:02:59 +01005692 (global.maxsock + FD_SETSIZE - 1) / FD_SETSIZE);
willy tarreau0f7af912005-12-17 12:21:26 +01005693
5694 fdtab = (struct fdtab *)calloc(1,
willy tarreau9fe663a2005-12-17 13:02:59 +01005695 sizeof(struct fdtab) * (global.maxsock));
5696 for (i = 0; i < global.maxsock; i++) {
willy tarreau0f7af912005-12-17 12:21:26 +01005697 fdtab[i].state = FD_STCLOSE;
5698 }
5699}
5700
5701/*
5702 * this function starts all the proxies. It returns 0 if OK, -1 if not.
5703 */
5704int start_proxies() {
5705 struct proxy *curproxy;
willy tarreaua41a8b42005-12-17 14:02:24 +01005706 struct listener *listener;
willy tarreau0f7af912005-12-17 12:21:26 +01005707 int one = 1;
5708 int fd;
5709
5710 for (curproxy = proxy; curproxy != NULL; curproxy = curproxy->next) {
willy tarreau0f7af912005-12-17 12:21:26 +01005711 if (curproxy->state == PR_STDISABLED)
5712 continue;
5713
willy tarreaua41a8b42005-12-17 14:02:24 +01005714 for (listener = curproxy->listen; listener != NULL; listener = listener->next) {
5715 if ((fd = listener->fd =
5716 socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1) {
5717 Alert("cannot create listening socket for proxy %s. Aborting.\n",
5718 curproxy->id);
5719 return -1;
5720 }
willy tarreau0f7af912005-12-17 12:21:26 +01005721
willy tarreaua41a8b42005-12-17 14:02:24 +01005722 if (fd >= global.maxsock) {
5723 Alert("socket(): not enough free sockets for proxy %s. Raise -n argument. Aborting.\n",
5724 curproxy->id);
5725 close(fd);
5726 return -1;
5727 }
willy tarreau5cbea6f2005-12-17 12:48:26 +01005728
willy tarreaua41a8b42005-12-17 14:02:24 +01005729 if ((fcntl(fd, F_SETFL, O_NONBLOCK) == -1) ||
5730 (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY,
5731 (char *) &one, sizeof(one)) == -1)) {
5732 Alert("cannot make socket non-blocking for proxy %s. Aborting.\n",
5733 curproxy->id);
5734 close(fd);
5735 return -1;
5736 }
willy tarreau0f7af912005-12-17 12:21:26 +01005737
willy tarreaua41a8b42005-12-17 14:02:24 +01005738 if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *) &one, sizeof(one)) == -1) {
5739 Alert("cannot do so_reuseaddr for proxy %s. Continuing.\n",
5740 curproxy->id);
5741 }
willy tarreau0f7af912005-12-17 12:21:26 +01005742
willy tarreaua41a8b42005-12-17 14:02:24 +01005743 if (bind(fd,
5744 (struct sockaddr *)&listener->addr,
5745 sizeof(listener->addr)) == -1) {
5746 Alert("cannot bind socket for proxy %s. Aborting.\n",
5747 curproxy->id);
5748 close(fd);
5749 return -1;
5750 }
willy tarreau0f7af912005-12-17 12:21:26 +01005751
willy tarreaua41a8b42005-12-17 14:02:24 +01005752 if (listen(fd, curproxy->maxconn) == -1) {
5753 Alert("cannot listen to socket for proxy %s. Aborting.\n",
5754 curproxy->id);
5755 close(fd);
5756 return -1;
5757 }
willy tarreau0f7af912005-12-17 12:21:26 +01005758
willy tarreaua41a8b42005-12-17 14:02:24 +01005759 /* the function for the accept() event */
5760 fdtab[fd].read = &event_accept;
5761 fdtab[fd].write = NULL; /* never called */
5762 fdtab[fd].owner = (struct task *)curproxy; /* reference the proxy instead of a task */
5763 curproxy->state = PR_STRUN;
5764 fdtab[fd].state = FD_STLISTEN;
5765 FD_SET(fd, StaticReadEvent);
5766 fd_insert(fd);
5767 listeners++;
5768 }
willy tarreaua1598082005-12-17 13:08:06 +01005769 send_log(curproxy, LOG_NOTICE, "Proxy %s started.\n", curproxy->id);
willy tarreau0f7af912005-12-17 12:21:26 +01005770 }
5771 return 0;
5772}
5773
5774
5775int main(int argc, char **argv) {
5776 init(argc, argv);
5777
willy tarreau9fe663a2005-12-17 13:02:59 +01005778 if (global.mode & MODE_QUIET) {
willy tarreau0f7af912005-12-17 12:21:26 +01005779 /* detach from the tty */
willy tarreau5cbea6f2005-12-17 12:48:26 +01005780 fclose(stdin); fclose(stdout); fclose(stderr);
willy tarreau0f7af912005-12-17 12:21:26 +01005781 close(0); close(1); close(2);
willy tarreau0f7af912005-12-17 12:21:26 +01005782 }
5783
5784 signal(SIGQUIT, dump);
5785 signal(SIGUSR1, sig_soft_stop);
willy tarreau8337c6b2005-12-17 13:41:01 +01005786 signal(SIGHUP, sig_dump_state);
willy tarreau0f7af912005-12-17 12:21:26 +01005787
5788 /* on very high loads, a sigpipe sometimes happen just between the
5789 * getsockopt() which tells "it's OK to write", and the following write :-(
5790 */
willy tarreau3242e862005-12-17 12:27:53 +01005791#ifndef MSG_NOSIGNAL
5792 signal(SIGPIPE, SIG_IGN);
5793#endif
willy tarreau0f7af912005-12-17 12:21:26 +01005794
5795 if (start_proxies() < 0)
5796 exit(1);
5797
willy tarreau9fe663a2005-12-17 13:02:59 +01005798 /* open log files */
5799
5800 /* chroot if needed */
5801 if (global.chroot != NULL) {
5802 if (chroot(global.chroot) == -1) {
5803 Alert("[%s.main()] Cannot chroot(%s).\n", argv[0], global.chroot);
5804 exit(1);
5805 }
5806 chdir("/");
5807 }
5808
5809 /* setgid / setuid */
willy tarreau036e1ce2005-12-17 13:46:33 +01005810 if (global.gid && setgid(global.gid) == -1) {
willy tarreau9fe663a2005-12-17 13:02:59 +01005811 Alert("[%s.main()] Cannot set gid %d.\n", argv[0], global.gid);
5812 exit(1);
5813 }
5814
willy tarreau036e1ce2005-12-17 13:46:33 +01005815 if (global.uid && setuid(global.uid) == -1) {
willy tarreau9fe663a2005-12-17 13:02:59 +01005816 Alert("[%s.main()] Cannot set uid %d.\n", argv[0], global.uid);
5817 exit(1);
5818 }
5819
5820 if (global.mode & MODE_DAEMON) {
5821 int ret = 0;
5822 int proc;
5823
5824 /* the father launches the required number of processes */
5825 for (proc = 0; proc < global.nbproc; proc++) {
5826 ret = fork();
5827 if (ret < 0) {
5828 Alert("[%s.main()] Cannot fork.\n", argv[0]);
5829 exit(1); /* there has been an error */
5830 }
5831 else if (ret == 0) /* child breaks here */
5832 break;
5833 }
5834 if (proc == global.nbproc)
5835 exit(0); /* parent must leave */
5836
willy tarreau750a4722005-12-17 13:21:24 +01005837 /* if we're NOT in QUIET mode, we should now close the 3 first FDs to ensure
5838 * that we can detach from the TTY. We MUST NOT do it in other cases since
5839 * it would have already be done, and 0-2 would have been affected to listening
5840 * sockets
5841 */
5842 if (!(global.mode & MODE_QUIET)) {
5843 /* detach from the tty */
5844 fclose(stdin); fclose(stdout); fclose(stderr);
5845 close(0); close(1); close(2); /* close all fd's */
5846 global.mode |= MODE_QUIET; /* ensure that we won't say anything from now */
5847 }
willy tarreaua1598082005-12-17 13:08:06 +01005848 pid = getpid(); /* update child's pid */
willy tarreaue867b482005-12-17 13:28:43 +01005849 setsid();
willy tarreau9fe663a2005-12-17 13:02:59 +01005850 }
5851
willy tarreau0f7af912005-12-17 12:21:26 +01005852 select_loop();
5853
5854 exit(0);
5855}