blob: 3995b53d671ef46e38c2d0c5b5082adb80fe37ab [file] [log] [blame]
Willy Tarreaubaaee002006-06-26 02:48:02 +02001/*
2 * Client-side variables and functions.
3 *
Willy Tarreaud825eef2007-05-12 22:35:00 +02004 * Copyright 2000-2007 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 Tarreau8797c062007-05-07 00:55:35 +020027#include <types/acl.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020028#include <types/backend.h>
29#include <types/buffers.h>
30#include <types/global.h>
31#include <types/httperr.h>
32#include <types/polling.h>
33#include <types/proxy.h>
34#include <types/server.h>
35#include <types/session.h>
36
Willy Tarreau8797c062007-05-07 00:55:35 +020037#include <proto/acl.h>
Willy Tarreau54469402006-07-29 16:59:06 +020038#include <proto/buffers.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020039#include <proto/client.h>
40#include <proto/fd.h>
41#include <proto/log.h>
Willy Tarreaue5f20dc2006-12-03 15:21:35 +010042#include <proto/hdr_idx.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020043#include <proto/proto_http.h>
Willy Tarreauc6ca1a02007-05-13 19:43:47 +020044#include <proto/session.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020045#include <proto/stream_sock.h>
46#include <proto/task.h>
47
48
Willy Tarreau14c8aac2007-05-08 19:46:30 +020049/* Retrieves the original destination address used by the client, and sets the
50 * SN_FRT_ADDR_SET flag.
51 */
52void get_frt_addr(struct session *s)
53{
54 socklen_t namelen = sizeof(s->frt_addr);
55
56 if (get_original_dst(s->cli_fd, (struct sockaddr_in *)&s->frt_addr, &namelen) == -1)
57 getsockname(s->cli_fd, (struct sockaddr *)&s->frt_addr, &namelen);
58 s->flags |= SN_FRT_ADDR_SET;
59}
Willy Tarreaubaaee002006-06-26 02:48:02 +020060
61/*
62 * FIXME: This should move to the STREAM_SOCK code then split into TCP and HTTP.
63 */
64
65/*
66 * this function is called on a read event from a listen socket, corresponding
67 * to an accept. It tries to accept as many connections as possible.
68 * It returns 0.
69 */
70int event_accept(int fd) {
71 struct proxy *p = (struct proxy *)fdtab[fd].owner;
72 struct session *s;
Willy Tarreauc2168d32007-03-03 20:51:44 +010073 struct http_txn *txn;
Willy Tarreaubaaee002006-06-26 02:48:02 +020074 struct task *t;
75 int cfd;
76 int max_accept;
77
78 if (global.nbproc > 1)
79 max_accept = 8; /* let other processes catch some connections too */
80 else
81 max_accept = -1;
82
Willy Tarreauf1221aa2006-12-17 22:14:12 +010083 while (p->feconn < p->maxconn && max_accept--) {
Willy Tarreaubaaee002006-06-26 02:48:02 +020084 struct sockaddr_storage addr;
85 socklen_t laddr = sizeof(addr);
86
87 if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1) {
88 switch (errno) {
89 case EAGAIN:
90 case EINTR:
91 case ECONNABORTED:
92 return 0; /* nothing more to accept */
93 case ENFILE:
94 send_log(p, LOG_EMERG,
95 "Proxy %s reached system FD limit at %d. Please check system tunables.\n",
96 p->id, maxfd);
97 return 0;
98 case EMFILE:
99 send_log(p, LOG_EMERG,
100 "Proxy %s reached process FD limit at %d. Please check 'ulimit-n' and restart.\n",
101 p->id, maxfd);
102 return 0;
103 case ENOBUFS:
104 case ENOMEM:
105 send_log(p, LOG_EMERG,
106 "Proxy %s reached system memory limit at %d sockets. Please check system tunables.\n",
107 p->id, maxfd);
108 return 0;
109 default:
110 return 0;
111 }
112 }
113
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200114 if ((s = pool_alloc2(pool2_session)) == NULL) { /* disable this proxy for a while */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200115 Alert("out of memory in event_accept().\n");
Willy Tarreauf161a342007-04-08 16:59:42 +0200116 EV_FD_CLR(fd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200117 p->state = PR_STIDLE;
118 close(cfd);
119 return 0;
120 }
121
122 /* if this session comes from a known monitoring system, we want to ignore
123 * it as soon as possible, which means closing it immediately for TCP.
124 */
125 s->flags = 0;
126 if (addr.ss_family == AF_INET &&
127 p->mon_mask.s_addr &&
128 (((struct sockaddr_in *)&addr)->sin_addr.s_addr & p->mon_mask.s_addr) == p->mon_net.s_addr) {
129 if (p->mode == PR_MODE_TCP) {
130 close(cfd);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200131 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200132 continue;
133 }
134 s->flags |= SN_MONITOR;
135 }
136
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200137 if ((t = pool_alloc2(pool2_task)) == NULL) { /* disable this proxy for a while */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200138 Alert("out of memory in event_accept().\n");
Willy Tarreauf161a342007-04-08 16:59:42 +0200139 EV_FD_CLR(fd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200140 p->state = PR_STIDLE;
141 close(cfd);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200142 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200143 return 0;
144 }
145
146 s->cli_addr = addr;
147 if (cfd >= global.maxsock) {
148 Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
149 close(cfd);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200150 pool_free2(pool2_task, t);
151 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200152 return 0;
153 }
154
155 if ((fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) ||
156 (setsockopt(cfd, IPPROTO_TCP, TCP_NODELAY,
157 (char *) &one, sizeof(one)) == -1)) {
158 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
159 close(cfd);
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200160 pool_free2(pool2_task, t);
161 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200162 return 0;
163 }
164
165 if (p->options & PR_O_TCP_CLI_KA)
166 setsockopt(cfd, SOL_SOCKET, SO_KEEPALIVE, (char *) &one, sizeof(one));
167
Willy Tarreau964c9362007-01-07 00:38:00 +0100168 t->wq = NULL;
Willy Tarreau96bcfd72007-04-29 10:41:56 +0200169 t->qlist.p = NULL;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200170 t->state = TASK_IDLE;
171 t->process = process_session;
172 t->context = s;
173
174 s->task = t;
Willy Tarreau830ff452006-12-17 19:31:23 +0100175 s->be = s->fe = p;
Willy Tarreau73de9892006-11-30 11:40:23 +0100176
Willy Tarreaubaaee002006-06-26 02:48:02 +0200177 s->cli_state = (p->mode == PR_MODE_HTTP) ? CL_STHEADERS : CL_STDATA; /* no HTTP headers for non-HTTP proxies */
178 s->srv_state = SV_STIDLE;
179 s->req = s->rep = NULL; /* will be allocated later */
180
Willy Tarreaubaaee002006-06-26 02:48:02 +0200181 s->cli_fd = cfd;
182 s->srv_fd = -1;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200183 s->srv = NULL;
184 s->pend_pos = NULL;
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200185 s->conn_retries = p->conn_retries;
Willy Tarreauddb358d2006-12-17 22:55:52 +0100186
187 /* FIXME: the logs are horribly complicated now, because they are
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200188 * defined in <p>, <p>, and later <be> and <be>.
Willy Tarreauddb358d2006-12-17 22:55:52 +0100189 */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200190
191 if (s->flags & SN_MONITOR)
192 s->logs.logwait = 0;
193 else
194 s->logs.logwait = p->to_log;
195
196 s->logs.tv_accept = now;
197 s->logs.t_request = -1;
198 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;
227
Willy Tarreauc2168d32007-03-03 20:51:44 +0100228 txn->req.msg_state = HTTP_MSG_RQBEFORE; /* at the very beginning of the request */
229 txn->req.sol = txn->req.eol = NULL;
230 txn->req.som = txn->req.eoh = 0; /* relative to the buffer */
231 txn->auth_hdr.len = -1;
Willy Tarreau45e73e32006-12-17 00:05:15 +0100232
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200233 if (p->nb_req_cap > 0) {
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200234 if ((txn->req.cap = pool_alloc2(p->req_cap_pool)) == NULL) {
235 /* no memory */
Willy Tarreau45e73e32006-12-17 00:05:15 +0100236 close(cfd); /* nothing can be done for this fd without memory */
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200237 pool_free2(pool2_task, t);
238 pool_free2(pool2_session, s);
Willy Tarreau45e73e32006-12-17 00:05:15 +0100239 return 0;
240 }
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200241 memset(txn->req.cap, 0, p->nb_req_cap*sizeof(char *));
Willy Tarreaubaaee002006-06-26 02:48:02 +0200242 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200243
Willy Tarreau45e73e32006-12-17 00:05:15 +0100244
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200245 if (p->nb_rsp_cap > 0) {
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200246 if ((txn->rsp.cap = pool_alloc2(p->rsp_cap_pool)) == NULL) {
247 /* no memory */
Willy Tarreauc2168d32007-03-03 20:51:44 +0100248 if (txn->req.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200249 pool_free2(p->req_cap_pool, txn->req.cap);
Willy Tarreau45e73e32006-12-17 00:05:15 +0100250 close(cfd); /* nothing can be done for this fd without memory */
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200251 pool_free2(pool2_task, t);
252 pool_free2(pool2_session, s);
Willy Tarreau45e73e32006-12-17 00:05:15 +0100253 return 0;
254 }
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200255 memset(txn->rsp.cap, 0, p->nb_rsp_cap*sizeof(char *));
Willy Tarreaubaaee002006-06-26 02:48:02 +0200256 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200257
Willy Tarreau45e73e32006-12-17 00:05:15 +0100258
Willy Tarreau1d4154a2007-05-13 22:57:02 +0200259 txn->hdr_idx.size = MAX_HTTP_HDR;
260
261 if ((txn->hdr_idx.v = pool_alloc2(p->hdr_idx_pool)) == NULL) {
262 /* no memory */
Willy Tarreauc2168d32007-03-03 20:51:44 +0100263 if (txn->rsp.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200264 pool_free2(p->rsp_cap_pool, txn->rsp.cap);
Willy Tarreauc2168d32007-03-03 20:51:44 +0100265 if (txn->req.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200266 pool_free2(p->req_cap_pool, txn->req.cap);
Willy Tarreaue5f20dc2006-12-03 15:21:35 +0100267 close(cfd); /* nothing can be done for this fd without memory */
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200268 pool_free2(pool2_task, t);
269 pool_free2(pool2_session, s);
Willy Tarreaue5f20dc2006-12-03 15:21:35 +0100270 return 0;
271 }
Willy Tarreauc2168d32007-03-03 20:51:44 +0100272 hdr_idx_init(&txn->hdr_idx);
Willy Tarreaue5f20dc2006-12-03 15:21:35 +0100273 }
274
Willy Tarreaubaaee002006-06-26 02:48:02 +0200275 if ((p->mode == PR_MODE_TCP || p->mode == PR_MODE_HTTP)
276 && (p->logfac1 >= 0 || p->logfac2 >= 0)) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200277 if (p->to_log) {
278 /* we have the client ip */
279 if (s->logs.logwait & LW_CLIP)
280 if (!(s->logs.logwait &= ~LW_CLIP))
Willy Tarreau42250582007-04-01 01:30:43 +0200281 tcp_sess_log(s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200282 }
283 else if (s->cli_addr.ss_family == AF_INET) {
284 char pn[INET_ADDRSTRLEN], sn[INET_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_INET, (const void *)&((struct sockaddr_in *)&s->frt_addr)->sin_addr,
Willy Tarreaubaaee002006-06-26 02:48:02 +0200290 sn, sizeof(sn)) &&
291 inet_ntop(AF_INET, (const void *)&((struct sockaddr_in *)&s->cli_addr)->sin_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_in *)&s->cli_addr)->sin_port),
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200295 sn, ntohs(((struct sockaddr_in *)&s->frt_addr)->sin_port),
Willy Tarreaubaaee002006-06-26 02:48:02 +0200296 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
297 }
298 }
299 else {
300 char pn[INET6_ADDRSTRLEN], sn[INET6_ADDRSTRLEN];
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200301
302 if (!(s->flags & SN_FRT_ADDR_SET))
303 get_frt_addr(s);
304
305 if (inet_ntop(AF_INET6, (const void *)&((struct sockaddr_in6 *)&s->frt_addr)->sin6_addr,
Willy Tarreaubaaee002006-06-26 02:48:02 +0200306 sn, sizeof(sn)) &&
307 inet_ntop(AF_INET6, (const void *)&((struct sockaddr_in6 *)&s->cli_addr)->sin6_addr,
308 pn, sizeof(pn))) {
309 send_log(p, LOG_INFO, "Connect from %s:%d to %s:%d (%s/%s)\n",
310 pn, ntohs(((struct sockaddr_in6 *)&s->cli_addr)->sin6_port),
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200311 sn, ntohs(((struct sockaddr_in6 *)&s->frt_addr)->sin6_port),
Willy Tarreaubaaee002006-06-26 02:48:02 +0200312 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
313 }
314 }
315 }
316
317 if ((global.mode & MODE_DEBUG) && (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200318 int len;
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200319
320 if (!(s->flags & SN_FRT_ADDR_SET))
321 get_frt_addr(s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200322
323 if (s->cli_addr.ss_family == AF_INET) {
324 char pn[INET_ADDRSTRLEN];
325 inet_ntop(AF_INET,
326 (const void *)&((struct sockaddr_in *)&s->cli_addr)->sin_addr,
327 pn, sizeof(pn));
328
329 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%s:%d]\n",
330 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
331 pn, ntohs(((struct sockaddr_in *)&s->cli_addr)->sin_port));
332 }
333 else {
334 char pn[INET6_ADDRSTRLEN];
335 inet_ntop(AF_INET6,
336 (const void *)&((struct sockaddr_in6 *)(&s->cli_addr))->sin6_addr,
337 pn, sizeof(pn));
338
339 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%s:%d]\n",
340 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
341 pn, ntohs(((struct sockaddr_in6 *)(&s->cli_addr))->sin6_port));
342 }
343
344 write(1, trash, len);
345 }
346
Willy Tarreau7341d942007-05-13 19:56:02 +0200347 if ((s->req = pool_alloc2(pool2_buffer)) == NULL) { /* no memory */
Willy Tarreauc2168d32007-03-03 20:51:44 +0100348 if (txn->hdr_idx.v != NULL)
Willy Tarreau1d4154a2007-05-13 22:57:02 +0200349 pool_free2(p->hdr_idx_pool, txn->hdr_idx.v);
Willy Tarreauc2168d32007-03-03 20:51:44 +0100350 if (txn->rsp.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200351 pool_free2(p->rsp_cap_pool, txn->rsp.cap);
Willy Tarreauc2168d32007-03-03 20:51:44 +0100352 if (txn->req.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200353 pool_free2(p->req_cap_pool, txn->req.cap);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200354 close(cfd); /* nothing can be done for this fd without memory */
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200355 pool_free2(pool2_task, t);
356 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200357 return 0;
358 }
359
Willy Tarreau54469402006-07-29 16:59:06 +0200360 buffer_init(s->req);
361 s->req->rlim += BUFSIZE;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200362 if (s->cli_state == CL_STHEADERS) /* reserve some space for header rewriting */
363 s->req->rlim -= MAXREWRITE;
364
Willy Tarreau73de9892006-11-30 11:40:23 +0100365 s->req->rto = s->fe->clitimeout;
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200366 s->req->wto = s->be->srvtimeout;
367 s->req->cto = s->be->srvtimeout;
Willy Tarreaud7971282006-07-29 18:36:34 +0200368
Willy Tarreau7341d942007-05-13 19:56:02 +0200369 if ((s->rep = pool_alloc2(pool2_buffer)) == NULL) { /* no memory */
370 pool_free2(pool2_buffer, s->req);
Willy Tarreauc2168d32007-03-03 20:51:44 +0100371 if (txn->hdr_idx.v != NULL)
Willy Tarreau1d4154a2007-05-13 22:57:02 +0200372 pool_free2(p->hdr_idx_pool, txn->hdr_idx.v);
Willy Tarreauc2168d32007-03-03 20:51:44 +0100373 if (txn->rsp.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200374 pool_free2(p->rsp_cap_pool, txn->rsp.cap);
Willy Tarreauc2168d32007-03-03 20:51:44 +0100375 if (txn->req.cap != NULL)
Willy Tarreaucf7f3202007-05-13 22:46:04 +0200376 pool_free2(p->req_cap_pool, txn->req.cap);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200377 close(cfd); /* nothing can be done for this fd without memory */
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200378 pool_free2(pool2_task, t);
379 pool_free2(pool2_session, s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200380 return 0;
381 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200382
Willy Tarreau54469402006-07-29 16:59:06 +0200383 buffer_init(s->rep);
384
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200385 s->rep->rto = s->be->srvtimeout;
Willy Tarreau830ff452006-12-17 19:31:23 +0100386 s->rep->wto = s->fe->clitimeout;
Willy Tarreauee991362007-05-14 14:37:50 +0200387 tv_eternity(&s->rep->cto);
Willy Tarreaud7971282006-07-29 18:36:34 +0200388
Willy Tarreau7a966482007-04-15 10:58:02 +0200389 fd_insert(cfd);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200390 fdtab[cfd].owner = t;
391 fdtab[cfd].state = FD_STREADY;
Willy Tarreaud7971282006-07-29 18:36:34 +0200392 fdtab[cfd].cb[DIR_RD].f = &stream_sock_read;
Willy Tarreau54469402006-07-29 16:59:06 +0200393 fdtab[cfd].cb[DIR_RD].b = s->req;
Willy Tarreauf8306d52006-07-29 19:01:31 +0200394 fdtab[cfd].cb[DIR_WR].f = &stream_sock_write;
Willy Tarreau54469402006-07-29 16:59:06 +0200395 fdtab[cfd].cb[DIR_WR].b = s->rep;
Willy Tarreau3d32d3a2007-04-15 11:31:05 +0200396 fdtab[cfd].ev = 0;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200397
398 if ((p->mode == PR_MODE_HTTP && (s->flags & SN_MONITOR)) ||
Willy Tarreau0f772532006-12-23 20:51:41 +0100399 (p->mode == PR_MODE_HEALTH && (p->options & PR_O_HTTP_CHK))) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200400 /* Either we got a request from a monitoring system on an HTTP instance,
401 * or we're in health check mode with the 'httpchk' option enabled. In
402 * both cases, we return a fake "HTTP/1.0 200 OK" response and we exit.
403 */
Willy Tarreau0f772532006-12-23 20:51:41 +0100404 struct chunk msg = { .str = "HTTP/1.0 200 OK\r\n\r\n", .len = 19 };
405 client_retnclose(s, &msg); /* forge a 200 response */
406 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200407 else if (p->mode == PR_MODE_HEALTH) { /* health check mode, no client reading */
Willy Tarreau0f772532006-12-23 20:51:41 +0100408 struct chunk msg = { .str = "OK\n", .len = 3 };
409 client_retnclose(s, &msg); /* forge an "OK" response */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200410 }
411 else {
Willy Tarreauf161a342007-04-08 16:59:42 +0200412 EV_FD_SET(cfd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200413 }
414
Willy Tarreaud7971282006-07-29 18:36:34 +0200415 tv_eternity(&s->req->rex);
416 tv_eternity(&s->req->wex);
417 tv_eternity(&s->req->cex);
418 tv_eternity(&s->rep->rex);
419 tv_eternity(&s->rep->wex);
Willy Tarreau5465e112007-04-29 19:09:47 +0200420 tv_eternity(&t->expire);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200421
Willy Tarreaud825eef2007-05-12 22:35:00 +0200422 if (tv_isset(&s->fe->clitimeout)) {
Willy Tarreau5465e112007-04-29 19:09:47 +0200423 if (EV_FD_ISSET(cfd, DIR_RD)) {
Willy Tarreaud825eef2007-05-12 22:35:00 +0200424 tv_add(&s->req->rex, &now, &s->fe->clitimeout);
Willy Tarreau5465e112007-04-29 19:09:47 +0200425 t->expire = s->req->rex;
426 }
427 if (EV_FD_ISSET(cfd, DIR_WR)) {
Willy Tarreaud825eef2007-05-12 22:35:00 +0200428 tv_add(&s->rep->wex, &now, &s->fe->clitimeout);
Willy Tarreau5465e112007-04-29 19:09:47 +0200429 t->expire = s->req->rex;
430 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200431 }
432
Willy Tarreaubaaee002006-06-26 02:48:02 +0200433 task_queue(t);
434
435 if (p->mode != PR_MODE_HEALTH)
Willy Tarreau96bcfd72007-04-29 10:41:56 +0200436 task_wakeup(t);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200437
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100438 p->feconn++; /* beconn will be increased later */
439 if (p->feconn > p->feconn_max)
440 p->feconn_max = p->feconn;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200441 actconn++;
442 totalconn++;
443
444 // fprintf(stderr, "accepting from %p => %d conn, %d total, task=%p\n", p, actconn, totalconn, t);
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100445 } /* end of while (p->feconn < p->maxconn) */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200446 return 0;
447}
448
449
450
Willy Tarreau8797c062007-05-07 00:55:35 +0200451/************************************************************************/
452/* All supported keywords must be declared here. */
453/************************************************************************/
454
455/* set test->ptr to point to the source IPv4/IPv6 address and test->i to the family */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200456static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200457acl_fetch_src(struct proxy *px, struct session *l4, void *l7, int dir,
458 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200459{
460 test->i = l4->cli_addr.ss_family;
461 if (test->i == AF_INET)
462 test->ptr = (void *)&((struct sockaddr_in *)&l4->cli_addr)->sin_addr;
463 else
464 test->ptr = (void *)&((struct sockaddr_in6 *)(&l4->cli_addr))->sin6_addr;
465 test->flags = ACL_TEST_F_READ_ONLY;
466 return 1;
467}
468
469
470/* set test->i to the connexion's source port */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200471static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200472acl_fetch_sport(struct proxy *px, struct session *l4, void *l7, int dir,
473 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200474{
475 if (l4->cli_addr.ss_family == AF_INET)
476 test->i = ntohs(((struct sockaddr_in *)&l4->cli_addr)->sin_port);
477 else
478 test->i = ntohs(((struct sockaddr_in6 *)(&l4->cli_addr))->sin6_port);
479 test->flags = 0;
480 return 1;
481}
482
Willy Tarreau662b2d82007-05-08 19:56:15 +0200483
484/* 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 +0200485static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200486acl_fetch_dst(struct proxy *px, struct session *l4, void *l7, int dir,
487 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau662b2d82007-05-08 19:56:15 +0200488{
489 if (!(l4->flags & SN_FRT_ADDR_SET))
490 get_frt_addr(l4);
491
492 test->i = l4->frt_addr.ss_family;
493 if (test->i == AF_INET)
494 test->ptr = (void *)&((struct sockaddr_in *)&l4->frt_addr)->sin_addr;
495 else
496 test->ptr = (void *)&((struct sockaddr_in6 *)(&l4->frt_addr))->sin6_addr;
497 test->flags = ACL_TEST_F_READ_ONLY;
498 return 1;
499}
500
501
502/* set test->i to the frontend connexion's destination port */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200503static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200504acl_fetch_dport(struct proxy *px, struct session *l4, void *l7, int dir,
505 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau662b2d82007-05-08 19:56:15 +0200506{
507 if (!(l4->flags & SN_FRT_ADDR_SET))
508 get_frt_addr(l4);
509
510 if (l4->frt_addr.ss_family == AF_INET)
511 test->i = ntohs(((struct sockaddr_in *)&l4->frt_addr)->sin_port);
512 else
513 test->i = ntohs(((struct sockaddr_in6 *)(&l4->frt_addr))->sin6_port);
514 test->flags = 0;
515 return 1;
516}
517
Willy Tarreau8797c062007-05-07 00:55:35 +0200518/* set test->i to the number of connexions to the proxy */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200519static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200520acl_fetch_dconn(struct proxy *px, struct session *l4, void *l7, int dir,
521 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200522{
523 test->i = px->feconn;
524 return 1;
525}
526
527
528/* Note: must not be declared <const> as its list will be overwritten */
529static struct acl_kw_list acl_kws = {{ },{
Willy Tarreauae8b7962007-06-09 23:10:04 +0200530 { "src_port", acl_parse_int, acl_fetch_sport, acl_match_int },
531 { "src", acl_parse_ip, acl_fetch_src, acl_match_ip },
532 { "dst", acl_parse_ip, acl_fetch_dst, acl_match_ip },
533 { "dst_port", acl_parse_int, acl_fetch_dport, acl_match_int },
Willy Tarreau662b2d82007-05-08 19:56:15 +0200534#if 0
Willy Tarreauae8b7962007-06-09 23:10:04 +0200535 { "src_limit", acl_parse_int, acl_fetch_sconn, acl_match_int },
Willy Tarreau8797c062007-05-07 00:55:35 +0200536#endif
Willy Tarreauae8b7962007-06-09 23:10:04 +0200537 { "dst_conn", acl_parse_int, acl_fetch_dconn, acl_match_int },
Willy Tarreau8797c062007-05-07 00:55:35 +0200538 { NULL, NULL, NULL, NULL },
539}};
540
541
542__attribute__((constructor))
543static void __client_init(void)
544{
545 acl_register_keywords(&acl_kws);
546}
547
548
Willy Tarreaubaaee002006-06-26 02:48:02 +0200549/*
550 * Local variables:
551 * c-indent-level: 8
552 * c-basic-offset: 8
553 * End:
554 */