blob: f5d6bb9d95016a66c78c12c30677d07a0a4485a5 [file] [log] [blame]
Willy Tarreaubaaee002006-06-26 02:48:02 +02001/*
2 * Client-side variables and functions.
3 *
Willy Tarreaub7f694f2008-06-22 17:18:02 +02004 * Copyright 2000-2008 Willy Tarreau <w@1wt.eu>
Willy Tarreaubaaee002006-06-26 02:48:02 +02005 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
11 */
12
13#include <errno.h>
14#include <fcntl.h>
15#include <stdio.h>
16#include <stdlib.h>
Willy Tarreau2dd0d472006-06-29 17:53:05 +020017#include <string.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020018
19#include <sys/socket.h>
20#include <sys/stat.h>
21#include <sys/types.h>
22
Willy Tarreau2dd0d472006-06-29 17:53:05 +020023#include <common/compat.h>
Willy Tarreaue3ba5f02006-06-29 18:54:54 +020024#include <common/config.h>
Willy Tarreau2dd0d472006-06-29 17:53:05 +020025#include <common/time.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020026
Willy Tarreaubaaee002006-06-26 02:48:02 +020027#include <types/global.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020028
Willy Tarreau8797c062007-05-07 00:55:35 +020029#include <proto/acl.h>
Willy Tarreau54469402006-07-29 16:59:06 +020030#include <proto/buffers.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020031#include <proto/client.h>
32#include <proto/fd.h>
33#include <proto/log.h>
Willy Tarreaue5f20dc2006-12-03 15:21:35 +010034#include <proto/hdr_idx.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020035#include <proto/proto_http.h>
Willy Tarreauc6ca1a02007-05-13 19:43:47 +020036#include <proto/session.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020037#include <proto/stream_sock.h>
38#include <proto/task.h>
39
40
Willy Tarreau14c8aac2007-05-08 19:46:30 +020041/* Retrieves the original destination address used by the client, and sets the
42 * SN_FRT_ADDR_SET flag.
43 */
44void get_frt_addr(struct session *s)
45{
46 socklen_t namelen = sizeof(s->frt_addr);
47
48 if (get_original_dst(s->cli_fd, (struct sockaddr_in *)&s->frt_addr, &namelen) == -1)
49 getsockname(s->cli_fd, (struct sockaddr *)&s->frt_addr, &namelen);
50 s->flags |= SN_FRT_ADDR_SET;
51}
Willy Tarreaubaaee002006-06-26 02:48:02 +020052
53/*
54 * FIXME: This should move to the STREAM_SOCK code then split into TCP and HTTP.
55 */
56
57/*
58 * this function is called on a read event from a listen socket, corresponding
59 * to an accept. It tries to accept as many connections as possible.
60 * It returns 0.
61 */
62int event_accept(int fd) {
Willy Tarreaue6b98942007-10-29 01:09:36 +010063 struct listener *l = (struct listener *)fdtab[fd].owner;
64 struct proxy *p = (struct proxy *)l->private; /* attached frontend */
Willy Tarreaubaaee002006-06-26 02:48:02 +020065 struct session *s;
Willy Tarreauc2168d32007-03-03 20:51:44 +010066 struct http_txn *txn;
Willy Tarreaubaaee002006-06-26 02:48:02 +020067 struct task *t;
68 int cfd;
Willy Tarreaua0250ba2008-01-06 11:22:57 +010069 int max_accept = global.tune.maxaccept;
Willy Tarreaubaaee002006-06-26 02:48:02 +020070
Willy Tarreauf1221aa2006-12-17 22:14:12 +010071 while (p->feconn < p->maxconn && max_accept--) {
Willy Tarreaubaaee002006-06-26 02:48:02 +020072 struct sockaddr_storage addr;
73 socklen_t laddr = sizeof(addr);
74
75 if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1) {
76 switch (errno) {
77 case EAGAIN:
78 case EINTR:
79 case ECONNABORTED:
80 return 0; /* nothing more to accept */
81 case ENFILE:
82 send_log(p, LOG_EMERG,
83 "Proxy %s reached system FD limit at %d. Please check system tunables.\n",
84 p->id, maxfd);
85 return 0;
86 case EMFILE:
87 send_log(p, LOG_EMERG,
88 "Proxy %s reached process FD limit at %d. Please check 'ulimit-n' and restart.\n",
89 p->id, maxfd);
90 return 0;
91 case ENOBUFS:
92 case ENOMEM:
93 send_log(p, LOG_EMERG,
94 "Proxy %s reached system memory limit at %d sockets. Please check system tunables.\n",
95 p->id, maxfd);
96 return 0;
97 default:
98 return 0;
99 }
100 }
101
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200102 if ((s = pool_alloc2(pool2_session)) == NULL) { /* disable this proxy for a while */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200103 Alert("out of memory in event_accept().\n");
Willy Tarreauf161a342007-04-08 16:59:42 +0200104 EV_FD_CLR(fd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200105 p->state = PR_STIDLE;
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100106 goto out_close;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200107 }
108
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200109 s->flags = 0;
110 s->analysis = 0;
111
Willy Tarreaubaaee002006-06-26 02:48:02 +0200112 /* if this session comes from a known monitoring system, we want to ignore
113 * it as soon as possible, which means closing it immediately for TCP.
114 */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200115 if (addr.ss_family == AF_INET &&
116 p->mon_mask.s_addr &&
117 (((struct sockaddr_in *)&addr)->sin_addr.s_addr & p->mon_mask.s_addr) == p->mon_net.s_addr) {
118 if (p->mode == PR_MODE_TCP) {
119 close(cfd);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200120 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200121 continue;
122 }
123 s->flags |= SN_MONITOR;
124 }
125
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200126 if ((t = pool_alloc2(pool2_task)) == NULL) { /* disable this proxy for a while */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200127 Alert("out of memory in event_accept().\n");
Willy Tarreauf161a342007-04-08 16:59:42 +0200128 EV_FD_CLR(fd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200129 p->state = PR_STIDLE;
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100130 goto out_free_session;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200131 }
132
133 s->cli_addr = addr;
134 if (cfd >= global.maxsock) {
135 Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100136 goto out_free_task;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200137 }
138
139 if ((fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) ||
140 (setsockopt(cfd, IPPROTO_TCP, TCP_NODELAY,
141 (char *) &one, sizeof(one)) == -1)) {
142 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100143 goto out_free_task;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200144 }
145
146 if (p->options & PR_O_TCP_CLI_KA)
147 setsockopt(cfd, SOL_SOCKET, SO_KEEPALIVE, (char *) &one, sizeof(one));
148
Alexandre Cassen87ea5482007-10-11 20:48:58 +0200149 if (p->options & PR_O_TCP_NOLING)
150 setsockopt(cfd, SOL_SOCKET, SO_LINGER, (struct linger *) &nolinger, sizeof(struct linger));
151
Willy Tarreau9789f7b2008-06-24 08:17:16 +0200152 task_init(t);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200153 t->process = process_session;
154 t->context = s;
155
156 s->task = t;
Willy Tarreau830ff452006-12-17 19:31:23 +0100157 s->be = s->fe = p;
Willy Tarreau73de9892006-11-30 11:40:23 +0100158
Willy Tarreaua7e76142007-11-03 14:28:39 +0100159 /* in HTTP mode, content switching requires that the backend
160 * first points to the same proxy as the frontend. However, in
161 * TCP mode there will be no header processing so any default
162 * backend must be assigned if set.
163 */
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200164 if (p->mode == PR_MODE_TCP) {
165 if (p->defbe.be)
166 s->be = p->defbe.be;
167 s->flags |= SN_BE_ASSIGNED;
Willy Tarreaua7e76142007-11-03 14:28:39 +0100168 }
169
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200170 if (p->mode == PR_MODE_HTTP)
171 s->analysis |= AN_REQ_HTTP_HDR;
172
173 if (s->fe->tcp_req.inspect_delay)
174 s->analysis |= AN_REQ_INSPECT;
175
176 s->cli_state = CL_STDATA;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200177 s->srv_state = SV_STIDLE;
178 s->req = s->rep = NULL; /* will be allocated later */
179
Willy Tarreaubaaee002006-06-26 02:48:02 +0200180 s->cli_fd = cfd;
181 s->srv_fd = -1;
Willy Tarreau7c669d72008-06-20 15:04:11 +0200182 s->srv = s->prev_srv = s->srv_conn = NULL;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200183 s->pend_pos = NULL;
Willy Tarreaua7e76142007-11-03 14:28:39 +0100184 s->conn_retries = s->be->conn_retries;
Willy Tarreauddb358d2006-12-17 22:55:52 +0100185
186 /* FIXME: the logs are horribly complicated now, because they are
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200187 * defined in <p>, <p>, and later <be> and <be>.
Willy Tarreauddb358d2006-12-17 22:55:52 +0100188 */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200189
190 if (s->flags & SN_MONITOR)
191 s->logs.logwait = 0;
192 else
193 s->logs.logwait = p->to_log;
194
Willy Tarreaub7f694f2008-06-22 17:18:02 +0200195 s->logs.accept_date = date; /* user-visible date for logging */
196 s->logs.tv_accept = now; /* corrected date for internal use */
Willy Tarreau70089872008-06-13 21:12:51 +0200197 tv_zero(&s->logs.tv_request);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200198 s->logs.t_queue = -1;
199 s->logs.t_connect = -1;
200 s->logs.t_data = -1;
201 s->logs.t_close = 0;
Willy Tarreau35d66b02007-01-02 00:28:21 +0100202 s->logs.bytes_in = s->logs.bytes_out = 0;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200203 s->logs.prx_queue_size = 0; /* we get the number of pending conns before us */
204 s->logs.srv_queue_size = 0; /* we will get this number soon */
205
206 s->data_source = DATA_SRC_NONE;
207
208 s->uniq_id = totalconn;
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100209 p->cum_feconn++; /* cum_beconn will be increased once assigned */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200210
Willy Tarreauc2168d32007-03-03 20:51:44 +0100211 txn = &s->txn;
Willy Tarreau3d300592007-03-18 18:34:41 +0100212 txn->flags = 0;
Willy Tarreau042cc792007-03-19 16:20:06 +0100213 /* Those variables will be checked and freed if non-NULL in
214 * session.c:session_free(). It is important that they are
215 * properly initialized.
216 */
217 txn->srv_cookie = NULL;
218 txn->cli_cookie = NULL;
219 txn->uri = NULL;
Willy Tarreauc2168d32007-03-03 20:51:44 +0100220 txn->req.cap = NULL;
221 txn->rsp.cap = NULL;
222 txn->hdr_idx.v = NULL;
223 txn->hdr_idx.size = txn->hdr_idx.used = 0;
Willy Tarreau45e73e32006-12-17 00:05:15 +0100224
225 if (p->mode == PR_MODE_HTTP) {
Willy Tarreau3bac9ff2007-03-18 17:31:28 +0100226 txn->status = -1;
matt.farnsworth@nokia.com1c2ab962008-04-14 20:47:37 +0200227 txn->req.hdr_content_len = 0LL;
228 txn->rsp.hdr_content_len = 0LL;
Willy Tarreauc2168d32007-03-03 20:51:44 +0100229 txn->req.msg_state = HTTP_MSG_RQBEFORE; /* at the very beginning of the request */
Willy Tarreauc11416f2007-06-17 16:58:38 +0200230 txn->rsp.msg_state = HTTP_MSG_RPBEFORE; /* at the very beginning of the response */
Willy Tarreauc2168d32007-03-03 20:51:44 +0100231 txn->req.sol = txn->req.eol = NULL;
232 txn->req.som = txn->req.eoh = 0; /* relative to the buffer */
233 txn->auth_hdr.len = -1;
Willy Tarreau45e73e32006-12-17 00:05:15 +0100234
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200235 if (p->nb_req_cap > 0) {
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100236 if ((txn->req.cap = pool_alloc2(p->req_cap_pool)) == NULL)
237 goto out_fail_reqcap; /* no memory */
238
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200239 memset(txn->req.cap, 0, p->nb_req_cap*sizeof(char *));
Willy Tarreaubaaee002006-06-26 02:48:02 +0200240 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200241
Willy Tarreau45e73e32006-12-17 00:05:15 +0100242
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200243 if (p->nb_rsp_cap > 0) {
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100244 if ((txn->rsp.cap = pool_alloc2(p->rsp_cap_pool)) == NULL)
245 goto out_fail_rspcap; /* no memory */
246
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200247 memset(txn->rsp.cap, 0, p->nb_rsp_cap*sizeof(char *));
Willy Tarreaubaaee002006-06-26 02:48:02 +0200248 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200249
Willy Tarreau45e73e32006-12-17 00:05:15 +0100250
Willy Tarreau1d4154a2007-05-13 22:57:02 +0200251 txn->hdr_idx.size = MAX_HTTP_HDR;
252
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100253 if ((txn->hdr_idx.v = pool_alloc2(p->hdr_idx_pool)) == NULL)
254 goto out_fail_idx; /* no memory */
255
Willy Tarreauc2168d32007-03-03 20:51:44 +0100256 hdr_idx_init(&txn->hdr_idx);
Willy Tarreaue5f20dc2006-12-03 15:21:35 +0100257 }
258
Willy Tarreaubaaee002006-06-26 02:48:02 +0200259 if ((p->mode == PR_MODE_TCP || p->mode == PR_MODE_HTTP)
260 && (p->logfac1 >= 0 || p->logfac2 >= 0)) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200261 if (p->to_log) {
262 /* we have the client ip */
263 if (s->logs.logwait & LW_CLIP)
264 if (!(s->logs.logwait &= ~LW_CLIP))
Willy Tarreau42250582007-04-01 01:30:43 +0200265 tcp_sess_log(s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200266 }
267 else if (s->cli_addr.ss_family == AF_INET) {
268 char pn[INET_ADDRSTRLEN], sn[INET_ADDRSTRLEN];
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200269
270 if (!(s->flags & SN_FRT_ADDR_SET))
271 get_frt_addr(s);
272
273 if (inet_ntop(AF_INET, (const void *)&((struct sockaddr_in *)&s->frt_addr)->sin_addr,
Willy Tarreaubaaee002006-06-26 02:48:02 +0200274 sn, sizeof(sn)) &&
275 inet_ntop(AF_INET, (const void *)&((struct sockaddr_in *)&s->cli_addr)->sin_addr,
276 pn, sizeof(pn))) {
277 send_log(p, LOG_INFO, "Connect from %s:%d to %s:%d (%s/%s)\n",
278 pn, ntohs(((struct sockaddr_in *)&s->cli_addr)->sin_port),
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200279 sn, ntohs(((struct sockaddr_in *)&s->frt_addr)->sin_port),
Willy Tarreaubaaee002006-06-26 02:48:02 +0200280 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
281 }
282 }
283 else {
284 char pn[INET6_ADDRSTRLEN], sn[INET6_ADDRSTRLEN];
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200285
286 if (!(s->flags & SN_FRT_ADDR_SET))
287 get_frt_addr(s);
288
289 if (inet_ntop(AF_INET6, (const void *)&((struct sockaddr_in6 *)&s->frt_addr)->sin6_addr,
Willy Tarreaubaaee002006-06-26 02:48:02 +0200290 sn, sizeof(sn)) &&
291 inet_ntop(AF_INET6, (const void *)&((struct sockaddr_in6 *)&s->cli_addr)->sin6_addr,
292 pn, sizeof(pn))) {
293 send_log(p, LOG_INFO, "Connect from %s:%d to %s:%d (%s/%s)\n",
294 pn, ntohs(((struct sockaddr_in6 *)&s->cli_addr)->sin6_port),
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200295 sn, ntohs(((struct sockaddr_in6 *)&s->frt_addr)->sin6_port),
Willy Tarreaubaaee002006-06-26 02:48:02 +0200296 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
297 }
298 }
299 }
300
301 if ((global.mode & MODE_DEBUG) && (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200302 int len;
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200303
304 if (!(s->flags & SN_FRT_ADDR_SET))
305 get_frt_addr(s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200306
307 if (s->cli_addr.ss_family == AF_INET) {
308 char pn[INET_ADDRSTRLEN];
309 inet_ntop(AF_INET,
310 (const void *)&((struct sockaddr_in *)&s->cli_addr)->sin_addr,
311 pn, sizeof(pn));
312
313 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%s:%d]\n",
314 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
315 pn, ntohs(((struct sockaddr_in *)&s->cli_addr)->sin_port));
316 }
317 else {
318 char pn[INET6_ADDRSTRLEN];
319 inet_ntop(AF_INET6,
320 (const void *)&((struct sockaddr_in6 *)(&s->cli_addr))->sin6_addr,
321 pn, sizeof(pn));
322
323 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%s:%d]\n",
324 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
325 pn, ntohs(((struct sockaddr_in6 *)(&s->cli_addr))->sin6_port));
326 }
327
328 write(1, trash, len);
329 }
330
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100331 if ((s->req = pool_alloc2(pool2_buffer)) == NULL)
332 goto out_fail_req; /* no memory */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200333
Willy Tarreau54469402006-07-29 16:59:06 +0200334 buffer_init(s->req);
335 s->req->rlim += BUFSIZE;
Willy Tarreaub6866442008-07-14 23:54:42 +0200336 if (p->mode == PR_MODE_HTTP) /* reserve some space for header rewriting */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200337 s->req->rlim -= MAXREWRITE;
338
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200339 if (!(s->analysis & AN_REQ_ANY))
Willy Tarreaudc0a6a02008-08-03 20:38:13 +0200340 s->req->flags |= BF_MAY_CONNECT; /* don't wait to establish connection */
341
Willy Tarreaud7c30f92007-12-03 01:38:36 +0100342 s->req->rto = s->fe->timeout.client;
343 s->req->wto = s->be->timeout.server;
344 s->req->cto = s->be->timeout.connect;
Willy Tarreaud7971282006-07-29 18:36:34 +0200345
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100346 if ((s->rep = pool_alloc2(pool2_buffer)) == NULL)
347 goto out_fail_rep; /* no memory */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200348
Willy Tarreau54469402006-07-29 16:59:06 +0200349 buffer_init(s->rep);
350
Willy Tarreaud7c30f92007-12-03 01:38:36 +0100351 s->rep->rto = s->be->timeout.server;
352 s->rep->wto = s->fe->timeout.client;
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200353 s->rep->cto = TICK_ETERNITY;
Willy Tarreaud7971282006-07-29 18:36:34 +0200354
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200355 s->req->rex = TICK_ETERNITY;
356 s->req->wex = TICK_ETERNITY;
357 s->req->cex = TICK_ETERNITY;
358 s->rep->rex = TICK_ETERNITY;
359 s->rep->wex = TICK_ETERNITY;
360 s->txn.exp = TICK_ETERNITY;
361 s->inspect_exp = TICK_ETERNITY;
362 t->expire = TICK_ETERNITY;
363
Willy Tarreau7a966482007-04-15 10:58:02 +0200364 fd_insert(cfd);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200365 fdtab[cfd].owner = t;
Willy Tarreaue6b98942007-10-29 01:09:36 +0100366 fdtab[cfd].listener = l;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200367 fdtab[cfd].state = FD_STREADY;
Willy Tarreaue6b98942007-10-29 01:09:36 +0100368 fdtab[cfd].cb[DIR_RD].f = l->proto->read;
Willy Tarreau54469402006-07-29 16:59:06 +0200369 fdtab[cfd].cb[DIR_RD].b = s->req;
Willy Tarreaue6b98942007-10-29 01:09:36 +0100370 fdtab[cfd].cb[DIR_WR].f = l->proto->write;
Willy Tarreau54469402006-07-29 16:59:06 +0200371 fdtab[cfd].cb[DIR_WR].b = s->rep;
Willy Tarreaue94ebd02007-10-09 17:14:37 +0200372 fdtab[cfd].peeraddr = (struct sockaddr *)&s->cli_addr;
373 fdtab[cfd].peerlen = sizeof(s->cli_addr);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200374
375 if ((p->mode == PR_MODE_HTTP && (s->flags & SN_MONITOR)) ||
Willy Tarreau0f772532006-12-23 20:51:41 +0100376 (p->mode == PR_MODE_HEALTH && (p->options & PR_O_HTTP_CHK))) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200377 /* Either we got a request from a monitoring system on an HTTP instance,
378 * or we're in health check mode with the 'httpchk' option enabled. In
379 * both cases, we return a fake "HTTP/1.0 200 OK" response and we exit.
380 */
Willy Tarreau0f772532006-12-23 20:51:41 +0100381 struct chunk msg = { .str = "HTTP/1.0 200 OK\r\n\r\n", .len = 19 };
382 client_retnclose(s, &msg); /* forge a 200 response */
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200383 t->expire = s->rep->wex;
Willy Tarreau0f772532006-12-23 20:51:41 +0100384 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200385 else if (p->mode == PR_MODE_HEALTH) { /* health check mode, no client reading */
Willy Tarreau0f772532006-12-23 20:51:41 +0100386 struct chunk msg = { .str = "OK\n", .len = 3 };
387 client_retnclose(s, &msg); /* forge an "OK" response */
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200388 t->expire = s->rep->wex;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200389 }
390 else {
Willy Tarreauf161a342007-04-08 16:59:42 +0200391 EV_FD_SET(cfd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200392 }
393
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200394 /* it is important not to call the wakeup function directly but to
395 * pass through task_wakeup(), because this one knows how to apply
396 * priorities to tasks.
397 */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200398 if (p->mode != PR_MODE_HEALTH)
Willy Tarreau96bcfd72007-04-29 10:41:56 +0200399 task_wakeup(t);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200400
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100401 p->feconn++; /* beconn will be increased later */
402 if (p->feconn > p->feconn_max)
403 p->feconn_max = p->feconn;
Willy Tarreaua7e76142007-11-03 14:28:39 +0100404
405 if (s->flags & SN_BE_ASSIGNED) {
406 s->be->cum_beconn++;
407 s->be->beconn++;
408 if (s->be->beconn > s->be->beconn_max)
409 s->be->beconn_max = s->be->beconn;
410 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200411 actconn++;
412 totalconn++;
413
414 // fprintf(stderr, "accepting from %p => %d conn, %d total, task=%p\n", p, actconn, totalconn, t);
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100415 } /* end of while (p->feconn < p->maxconn) */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200416 return 0;
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100417
418 /* Error unrolling */
419 out_fail_rep:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200420 pool_free2(pool2_buffer, s->req);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100421 out_fail_req:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200422 pool_free2(p->hdr_idx_pool, txn->hdr_idx.v);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100423 out_fail_idx:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200424 pool_free2(p->rsp_cap_pool, txn->rsp.cap);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100425 out_fail_rspcap:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200426 pool_free2(p->req_cap_pool, txn->req.cap);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100427 out_fail_reqcap:
428 out_free_task:
429 pool_free2(pool2_task, t);
430 out_free_session:
431 pool_free2(pool2_session, s);
432 out_close:
433 close(cfd);
434 return 0;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200435}
436
437
438
Willy Tarreau8797c062007-05-07 00:55:35 +0200439/************************************************************************/
440/* All supported keywords must be declared here. */
441/************************************************************************/
442
443/* set test->ptr to point to the source IPv4/IPv6 address and test->i to the family */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200444static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200445acl_fetch_src(struct proxy *px, struct session *l4, void *l7, int dir,
446 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200447{
448 test->i = l4->cli_addr.ss_family;
449 if (test->i == AF_INET)
450 test->ptr = (void *)&((struct sockaddr_in *)&l4->cli_addr)->sin_addr;
451 else
452 test->ptr = (void *)&((struct sockaddr_in6 *)(&l4->cli_addr))->sin6_addr;
453 test->flags = ACL_TEST_F_READ_ONLY;
454 return 1;
455}
456
457
458/* set test->i to the connexion's source port */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200459static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200460acl_fetch_sport(struct proxy *px, struct session *l4, void *l7, int dir,
461 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200462{
463 if (l4->cli_addr.ss_family == AF_INET)
464 test->i = ntohs(((struct sockaddr_in *)&l4->cli_addr)->sin_port);
465 else
466 test->i = ntohs(((struct sockaddr_in6 *)(&l4->cli_addr))->sin6_port);
467 test->flags = 0;
468 return 1;
469}
470
Willy Tarreau662b2d82007-05-08 19:56:15 +0200471
472/* set test->ptr to point to the frontend's IPv4/IPv6 address and test->i to the family */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200473static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200474acl_fetch_dst(struct proxy *px, struct session *l4, void *l7, int dir,
475 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau662b2d82007-05-08 19:56:15 +0200476{
477 if (!(l4->flags & SN_FRT_ADDR_SET))
478 get_frt_addr(l4);
479
480 test->i = l4->frt_addr.ss_family;
481 if (test->i == AF_INET)
482 test->ptr = (void *)&((struct sockaddr_in *)&l4->frt_addr)->sin_addr;
483 else
484 test->ptr = (void *)&((struct sockaddr_in6 *)(&l4->frt_addr))->sin6_addr;
485 test->flags = ACL_TEST_F_READ_ONLY;
486 return 1;
487}
488
489
490/* set test->i to the frontend connexion's destination port */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200491static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200492acl_fetch_dport(struct proxy *px, struct session *l4, void *l7, int dir,
493 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau662b2d82007-05-08 19:56:15 +0200494{
495 if (!(l4->flags & SN_FRT_ADDR_SET))
496 get_frt_addr(l4);
497
498 if (l4->frt_addr.ss_family == AF_INET)
499 test->i = ntohs(((struct sockaddr_in *)&l4->frt_addr)->sin_port);
500 else
501 test->i = ntohs(((struct sockaddr_in6 *)(&l4->frt_addr))->sin6_port);
502 test->flags = 0;
503 return 1;
504}
505
Alexandre Cassen5eb1a902007-11-29 15:43:32 +0100506
Willy Tarreau8797c062007-05-07 00:55:35 +0200507/* set test->i to the number of connexions to the proxy */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200508static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200509acl_fetch_dconn(struct proxy *px, struct session *l4, void *l7, int dir,
510 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200511{
512 test->i = px->feconn;
513 return 1;
514}
515
516
517/* Note: must not be declared <const> as its list will be overwritten */
518static struct acl_kw_list acl_kws = {{ },{
Willy Tarreau0ceba5a2008-07-25 19:31:03 +0200519 { "src_port", acl_parse_int, acl_fetch_sport, acl_match_int, ACL_USE_TCP_PERMANENT },
520 { "src", acl_parse_ip, acl_fetch_src, acl_match_ip, ACL_USE_TCP4_PERMANENT },
521 { "dst", acl_parse_ip, acl_fetch_dst, acl_match_ip, ACL_USE_TCP4_PERMANENT },
522 { "dst_port", acl_parse_int, acl_fetch_dport, acl_match_int, ACL_USE_TCP_PERMANENT },
Willy Tarreau662b2d82007-05-08 19:56:15 +0200523#if 0
Alexandre Cassen5eb1a902007-11-29 15:43:32 +0100524 { "src_limit", acl_parse_int, acl_fetch_sconn, acl_match_int },
Willy Tarreau8797c062007-05-07 00:55:35 +0200525#endif
Willy Tarreau0ceba5a2008-07-25 19:31:03 +0200526 { "dst_conn", acl_parse_int, acl_fetch_dconn, acl_match_int, ACL_USE_NOTHING },
Willy Tarreau8797c062007-05-07 00:55:35 +0200527 { NULL, NULL, NULL, NULL },
528}};
529
530
531__attribute__((constructor))
532static void __client_init(void)
533{
534 acl_register_keywords(&acl_kws);
535}
536
537
Willy Tarreaubaaee002006-06-26 02:48:02 +0200538/*
539 * Local variables:
540 * c-indent-level: 8
541 * c-basic-offset: 8
542 * End:
543 */