blob: b3e19804d5264a213c00dc275a29cd8fe3668c9a [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 Tarreau7f062c42009-03-05 18:43:00 +010036#include <proto/proxy.h>
Willy Tarreauc6ca1a02007-05-13 19:43:47 +020037#include <proto/session.h>
Willy Tarreaudded32d2008-11-30 19:48:07 +010038#include <proto/stream_interface.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020039#include <proto/stream_sock.h>
40#include <proto/task.h>
41
42
Willy Tarreau14c8aac2007-05-08 19:46:30 +020043/* Retrieves the original destination address used by the client, and sets the
44 * SN_FRT_ADDR_SET flag.
45 */
46void get_frt_addr(struct session *s)
47{
48 socklen_t namelen = sizeof(s->frt_addr);
49
Willy Tarreau7e5067d2008-12-07 16:27:56 +010050 if (get_original_dst(s->si[0].fd, (struct sockaddr_in *)&s->frt_addr, &namelen) == -1)
51 getsockname(s->si[0].fd, (struct sockaddr *)&s->frt_addr, &namelen);
Willy Tarreau14c8aac2007-05-08 19:46:30 +020052 s->flags |= SN_FRT_ADDR_SET;
53}
Willy Tarreaubaaee002006-06-26 02:48:02 +020054
55/*
56 * FIXME: This should move to the STREAM_SOCK code then split into TCP and HTTP.
57 */
Willy Tarreauf54f8bd2008-11-23 19:53:55 +010058
Willy Tarreaubaaee002006-06-26 02:48:02 +020059/*
60 * this function is called on a read event from a listen socket, corresponding
61 * to an accept. It tries to accept as many connections as possible.
62 * It returns 0.
63 */
64int event_accept(int fd) {
Willy Tarreaueabf3132008-08-29 23:36:51 +020065 struct listener *l = fdtab[fd].owner;
Willy Tarreaue6b98942007-10-29 01:09:36 +010066 struct proxy *p = (struct proxy *)l->private; /* attached frontend */
Willy Tarreaubaaee002006-06-26 02:48:02 +020067 struct session *s;
Willy Tarreauc2168d32007-03-03 20:51:44 +010068 struct http_txn *txn;
Willy Tarreaubaaee002006-06-26 02:48:02 +020069 struct task *t;
70 int cfd;
Willy Tarreaua0250ba2008-01-06 11:22:57 +010071 int max_accept = global.tune.maxaccept;
Willy Tarreaubaaee002006-06-26 02:48:02 +020072
Willy Tarreau79584222009-03-06 09:18:27 +010073 if (p->fe_maxsps) {
74 int max = freq_ctr_remain(&p->fe_sess_per_sec, p->fe_maxsps, 0);
75 if (max_accept > max)
76 max_accept = max;
77 }
78
Willy Tarreaub00f9c42009-03-21 22:43:12 +010079 while (p->feconn < p->maxconn && actconn < global.maxconn && max_accept--) {
Willy Tarreaubaaee002006-06-26 02:48:02 +020080 struct sockaddr_storage addr;
81 socklen_t laddr = sizeof(addr);
82
83 if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1) {
84 switch (errno) {
85 case EAGAIN:
86 case EINTR:
87 case ECONNABORTED:
88 return 0; /* nothing more to accept */
89 case ENFILE:
90 send_log(p, LOG_EMERG,
91 "Proxy %s reached system FD limit at %d. Please check system tunables.\n",
92 p->id, maxfd);
93 return 0;
94 case EMFILE:
95 send_log(p, LOG_EMERG,
96 "Proxy %s reached process FD limit at %d. Please check 'ulimit-n' and restart.\n",
97 p->id, maxfd);
98 return 0;
99 case ENOBUFS:
100 case ENOMEM:
101 send_log(p, LOG_EMERG,
102 "Proxy %s reached system memory limit at %d sockets. Please check system tunables.\n",
103 p->id, maxfd);
104 return 0;
105 default:
106 return 0;
107 }
108 }
109
Willy Tarreauc6ca1a02007-05-13 19:43:47 +0200110 if ((s = pool_alloc2(pool2_session)) == NULL) { /* disable this proxy for a while */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200111 Alert("out of memory in event_accept().\n");
Willy Tarreauf161a342007-04-08 16:59:42 +0200112 EV_FD_CLR(fd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200113 p->state = PR_STIDLE;
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100114 goto out_close;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200115 }
116
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100117 LIST_ADDQ(&sessions, &s->list);
Willy Tarreau62e4f1d2008-12-07 20:16:23 +0100118 LIST_INIT(&s->back_refs);
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100119
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200120 s->flags = 0;
Willy Tarreauf8533202008-08-16 14:55:08 +0200121 s->term_trace = 0;
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200122
Willy Tarreaubaaee002006-06-26 02:48:02 +0200123 /* if this session comes from a known monitoring system, we want to ignore
124 * it as soon as possible, which means closing it immediately for TCP.
125 */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200126 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 Tarreaua4613182009-03-21 18:13:21 +0100137 if ((t = task_new()) == 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;
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100141 goto out_free_session;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200142 }
143
144 s->cli_addr = addr;
145 if (cfd >= global.maxsock) {
146 Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100147 goto out_free_task;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200148 }
149
150 if ((fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) ||
151 (setsockopt(cfd, IPPROTO_TCP, TCP_NODELAY,
152 (char *) &one, sizeof(one)) == -1)) {
153 Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100154 goto out_free_task;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200155 }
156
157 if (p->options & PR_O_TCP_CLI_KA)
158 setsockopt(cfd, SOL_SOCKET, SO_KEEPALIVE, (char *) &one, sizeof(one));
159
Alexandre Cassen87ea5482007-10-11 20:48:58 +0200160 if (p->options & PR_O_TCP_NOLING)
161 setsockopt(cfd, SOL_SOCKET, SO_LINGER, (struct linger *) &nolinger, sizeof(struct linger));
162
Willy Tarreau3bc13772008-12-07 11:50:35 +0100163 t->process = l->handler;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200164 t->context = s;
165
166 s->task = t;
Willy Tarreaub5654f62008-12-07 16:45:10 +0100167 s->listener = l;
Willy Tarreau830ff452006-12-17 19:31:23 +0100168 s->be = s->fe = p;
Willy Tarreau73de9892006-11-30 11:40:23 +0100169
Willy Tarreaua7e76142007-11-03 14:28:39 +0100170 /* in HTTP mode, content switching requires that the backend
171 * first points to the same proxy as the frontend. However, in
172 * TCP mode there will be no header processing so any default
173 * backend must be assigned if set.
174 */
Willy Tarreau67f0eea2008-08-10 22:55:22 +0200175 if (p->mode == PR_MODE_TCP) {
176 if (p->defbe.be)
177 s->be = p->defbe.be;
178 s->flags |= SN_BE_ASSIGNED;
Willy Tarreaua7e76142007-11-03 14:28:39 +0100179 }
180
Willy Tarreauff8d42e2008-12-07 14:37:09 +0100181 s->ana_state = 0; /* analysers may change it but must reset it upon exit */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200182 s->req = s->rep = NULL; /* will be allocated later */
183
Willy Tarreau35374672008-09-03 18:11:02 +0200184 s->si[0].state = s->si[0].prev_state = SI_ST_EST;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200185 s->si[0].err_type = SI_ET_NONE;
186 s->si[0].err_loc = NULL;
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200187 s->si[0].owner = t;
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200188 s->si[0].shutr = stream_sock_shutr;
Willy Tarreau48adac52008-08-30 04:58:38 +0200189 s->si[0].shutw = stream_sock_shutw;
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100190 s->si[0].chk_rcv = stream_sock_chk_rcv;
191 s->si[0].chk_snd = stream_sock_chk_snd;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200192 s->si[0].fd = cfd;
Willy Tarreaud7704b52008-09-04 11:51:16 +0200193 s->si[0].flags = SI_FL_NONE;
Willy Tarreau35374672008-09-03 18:11:02 +0200194 s->si[0].exp = TICK_ETERNITY;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200195
Willy Tarreau35374672008-09-03 18:11:02 +0200196 s->si[1].state = s->si[1].prev_state = SI_ST_INI;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200197 s->si[1].err_type = SI_ET_NONE;
198 s->si[1].err_loc = NULL;
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200199 s->si[1].owner = t;
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200200 s->si[1].shutr = stream_sock_shutr;
Willy Tarreau48adac52008-08-30 04:58:38 +0200201 s->si[1].shutw = stream_sock_shutw;
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100202 s->si[1].chk_rcv = stream_sock_chk_rcv;
203 s->si[1].chk_snd = stream_sock_chk_snd;
Willy Tarreau35374672008-09-03 18:11:02 +0200204 s->si[1].exp = TICK_ETERNITY;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200205 s->si[1].fd = -1; /* just to help with debugging */
Willy Tarreaud7704b52008-09-04 11:51:16 +0200206 s->si[1].flags = SI_FL_NONE;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200207
Willy Tarreau7c669d72008-06-20 15:04:11 +0200208 s->srv = s->prev_srv = s->srv_conn = NULL;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200209 s->pend_pos = NULL;
Willy Tarreaua7e76142007-11-03 14:28:39 +0100210 s->conn_retries = s->be->conn_retries;
Willy Tarreauddb358d2006-12-17 22:55:52 +0100211
212 /* FIXME: the logs are horribly complicated now, because they are
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200213 * defined in <p>, <p>, and later <be> and <be>.
Willy Tarreauddb358d2006-12-17 22:55:52 +0100214 */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200215
216 if (s->flags & SN_MONITOR)
217 s->logs.logwait = 0;
218 else
219 s->logs.logwait = p->to_log;
220
Willy Tarreaua5555ec2008-11-30 19:02:32 +0100221 if (s->logs.logwait & LW_REQ)
222 s->do_log = http_sess_log;
223 else
224 s->do_log = tcp_sess_log;
225
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100226 if (p->mode == PR_MODE_HTTP)
227 s->srv_error = http_return_srv_error;
228 else
Willy Tarreau84455332009-03-15 22:34:05 +0100229 s->srv_error = default_srv_error;
Willy Tarreau0cac36f2008-11-30 20:44:17 +0100230
Willy Tarreaub7f694f2008-06-22 17:18:02 +0200231 s->logs.accept_date = date; /* user-visible date for logging */
232 s->logs.tv_accept = now; /* corrected date for internal use */
Willy Tarreau70089872008-06-13 21:12:51 +0200233 tv_zero(&s->logs.tv_request);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200234 s->logs.t_queue = -1;
235 s->logs.t_connect = -1;
236 s->logs.t_data = -1;
237 s->logs.t_close = 0;
Willy Tarreau35d66b02007-01-02 00:28:21 +0100238 s->logs.bytes_in = s->logs.bytes_out = 0;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200239 s->logs.prx_queue_size = 0; /* we get the number of pending conns before us */
240 s->logs.srv_queue_size = 0; /* we will get this number soon */
241
242 s->data_source = DATA_SRC_NONE;
243
244 s->uniq_id = totalconn;
Willy Tarreau7f062c42009-03-05 18:43:00 +0100245 proxy_inc_fe_ctr(p); /* note: cum_beconn will be increased once assigned */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200246
Willy Tarreauc2168d32007-03-03 20:51:44 +0100247 txn = &s->txn;
Willy Tarreau3d300592007-03-18 18:34:41 +0100248 txn->flags = 0;
Willy Tarreau042cc792007-03-19 16:20:06 +0100249 /* Those variables will be checked and freed if non-NULL in
250 * session.c:session_free(). It is important that they are
251 * properly initialized.
252 */
253 txn->srv_cookie = NULL;
254 txn->cli_cookie = NULL;
255 txn->uri = NULL;
Willy Tarreauc2168d32007-03-03 20:51:44 +0100256 txn->req.cap = NULL;
257 txn->rsp.cap = NULL;
258 txn->hdr_idx.v = NULL;
259 txn->hdr_idx.size = txn->hdr_idx.used = 0;
Willy Tarreau45e73e32006-12-17 00:05:15 +0100260
261 if (p->mode == PR_MODE_HTTP) {
Willy Tarreau3bac9ff2007-03-18 17:31:28 +0100262 txn->status = -1;
matt.farnsworth@nokia.com1c2ab962008-04-14 20:47:37 +0200263 txn->req.hdr_content_len = 0LL;
264 txn->rsp.hdr_content_len = 0LL;
Willy Tarreauc2168d32007-03-03 20:51:44 +0100265 txn->req.msg_state = HTTP_MSG_RQBEFORE; /* at the very beginning of the request */
Willy Tarreauc11416f2007-06-17 16:58:38 +0200266 txn->rsp.msg_state = HTTP_MSG_RPBEFORE; /* at the very beginning of the response */
Willy Tarreauc2168d32007-03-03 20:51:44 +0100267 txn->req.sol = txn->req.eol = NULL;
268 txn->req.som = txn->req.eoh = 0; /* relative to the buffer */
Willy Tarreau32a4ec02009-04-02 11:35:18 +0200269 txn->req.err_pos = txn->rsp.err_pos = -2; /* block buggy requests/responses */
270 if (p->options2 & PR_O2_REQBUG_OK)
271 txn->req.err_pos = -1; /* let buggy requests pass */
Willy Tarreauc2168d32007-03-03 20:51:44 +0100272 txn->auth_hdr.len = -1;
Willy Tarreau45e73e32006-12-17 00:05:15 +0100273
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200274 if (p->nb_req_cap > 0) {
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100275 if ((txn->req.cap = pool_alloc2(p->req_cap_pool)) == NULL)
276 goto out_fail_reqcap; /* no memory */
277
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200278 memset(txn->req.cap, 0, p->nb_req_cap*sizeof(char *));
Willy Tarreaubaaee002006-06-26 02:48:02 +0200279 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200280
Willy Tarreau45e73e32006-12-17 00:05:15 +0100281
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200282 if (p->nb_rsp_cap > 0) {
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100283 if ((txn->rsp.cap = pool_alloc2(p->rsp_cap_pool)) == NULL)
284 goto out_fail_rspcap; /* no memory */
285
Willy Tarreaue2e27a52007-04-01 00:01:37 +0200286 memset(txn->rsp.cap, 0, p->nb_rsp_cap*sizeof(char *));
Willy Tarreaubaaee002006-06-26 02:48:02 +0200287 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200288
Willy Tarreau45e73e32006-12-17 00:05:15 +0100289
Willy Tarreau1d4154a2007-05-13 22:57:02 +0200290 txn->hdr_idx.size = MAX_HTTP_HDR;
291
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100292 if ((txn->hdr_idx.v = pool_alloc2(p->hdr_idx_pool)) == NULL)
293 goto out_fail_idx; /* no memory */
294
Willy Tarreauc2168d32007-03-03 20:51:44 +0100295 hdr_idx_init(&txn->hdr_idx);
Willy Tarreaue5f20dc2006-12-03 15:21:35 +0100296 }
297
Willy Tarreaubaaee002006-06-26 02:48:02 +0200298 if ((p->mode == PR_MODE_TCP || p->mode == PR_MODE_HTTP)
299 && (p->logfac1 >= 0 || p->logfac2 >= 0)) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200300 if (p->to_log) {
301 /* we have the client ip */
302 if (s->logs.logwait & LW_CLIP)
303 if (!(s->logs.logwait &= ~LW_CLIP))
Willy Tarreaua5555ec2008-11-30 19:02:32 +0100304 s->do_log(s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200305 }
306 else if (s->cli_addr.ss_family == AF_INET) {
307 char pn[INET_ADDRSTRLEN], sn[INET_ADDRSTRLEN];
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200308
309 if (!(s->flags & SN_FRT_ADDR_SET))
310 get_frt_addr(s);
311
312 if (inet_ntop(AF_INET, (const void *)&((struct sockaddr_in *)&s->frt_addr)->sin_addr,
Willy Tarreaubaaee002006-06-26 02:48:02 +0200313 sn, sizeof(sn)) &&
314 inet_ntop(AF_INET, (const void *)&((struct sockaddr_in *)&s->cli_addr)->sin_addr,
315 pn, sizeof(pn))) {
316 send_log(p, LOG_INFO, "Connect from %s:%d to %s:%d (%s/%s)\n",
317 pn, ntohs(((struct sockaddr_in *)&s->cli_addr)->sin_port),
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200318 sn, ntohs(((struct sockaddr_in *)&s->frt_addr)->sin_port),
Willy Tarreaubaaee002006-06-26 02:48:02 +0200319 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
320 }
321 }
322 else {
323 char pn[INET6_ADDRSTRLEN], sn[INET6_ADDRSTRLEN];
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200324
325 if (!(s->flags & SN_FRT_ADDR_SET))
326 get_frt_addr(s);
327
328 if (inet_ntop(AF_INET6, (const void *)&((struct sockaddr_in6 *)&s->frt_addr)->sin6_addr,
Willy Tarreaubaaee002006-06-26 02:48:02 +0200329 sn, sizeof(sn)) &&
330 inet_ntop(AF_INET6, (const void *)&((struct sockaddr_in6 *)&s->cli_addr)->sin6_addr,
331 pn, sizeof(pn))) {
332 send_log(p, LOG_INFO, "Connect from %s:%d to %s:%d (%s/%s)\n",
333 pn, ntohs(((struct sockaddr_in6 *)&s->cli_addr)->sin6_port),
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200334 sn, ntohs(((struct sockaddr_in6 *)&s->frt_addr)->sin6_port),
Willy Tarreaubaaee002006-06-26 02:48:02 +0200335 p->id, (p->mode == PR_MODE_HTTP) ? "HTTP" : "TCP");
336 }
337 }
338 }
339
340 if ((global.mode & MODE_DEBUG) && (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200341 int len;
Willy Tarreau14c8aac2007-05-08 19:46:30 +0200342
343 if (!(s->flags & SN_FRT_ADDR_SET))
344 get_frt_addr(s);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200345
346 if (s->cli_addr.ss_family == AF_INET) {
347 char pn[INET_ADDRSTRLEN];
348 inet_ntop(AF_INET,
349 (const void *)&((struct sockaddr_in *)&s->cli_addr)->sin_addr,
350 pn, sizeof(pn));
351
352 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%s:%d]\n",
353 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
354 pn, ntohs(((struct sockaddr_in *)&s->cli_addr)->sin_port));
355 }
356 else {
357 char pn[INET6_ADDRSTRLEN];
358 inet_ntop(AF_INET6,
359 (const void *)&((struct sockaddr_in6 *)(&s->cli_addr))->sin6_addr,
360 pn, sizeof(pn));
361
362 len = sprintf(trash, "%08x:%s.accept(%04x)=%04x from [%s:%d]\n",
363 s->uniq_id, p->id, (unsigned short)fd, (unsigned short)cfd,
364 pn, ntohs(((struct sockaddr_in6 *)(&s->cli_addr))->sin6_port));
365 }
366
367 write(1, trash, len);
368 }
369
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100370 if ((s->req = pool_alloc2(pool2_buffer)) == NULL)
371 goto out_fail_req; /* no memory */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200372
Willy Tarreau54469402006-07-29 16:59:06 +0200373 buffer_init(s->req);
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200374 s->req->prod = &s->si[0];
375 s->req->cons = &s->si[1];
Willy Tarreau48adac52008-08-30 04:58:38 +0200376 s->si[0].ib = s->si[1].ob = s->req;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200377
Willy Tarreau9a2d1542008-08-30 12:31:07 +0200378 s->req->flags |= BF_READ_ATTACHED; /* the producer is already connected */
379
Willy Tarreau1b194fe2009-03-21 21:10:04 +0100380 if (p->mode == PR_MODE_HTTP) { /* reserve some space for header rewriting */
Willy Tarreau03d60bb2009-01-09 11:13:00 +0100381 s->req->max_len -= MAXREWRITE;
Willy Tarreau1b194fe2009-03-21 21:10:04 +0100382 s->req->flags |= BF_READ_DONTWAIT; /* one read is usually enough */
383 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200384
Willy Tarreau3bc13772008-12-07 11:50:35 +0100385 /* activate default analysers enabled for this listener */
386 s->req->analysers = l->analysers;
Willy Tarreau2df28e82008-08-17 15:20:19 +0200387
388 if (!s->req->analysers)
Willy Tarreau3da77c52008-08-29 09:58:42 +0200389 buffer_write_ena(s->req); /* don't wait to establish connection */
Willy Tarreaudc0a6a02008-08-03 20:38:13 +0200390
Willy Tarreaud7c30f92007-12-03 01:38:36 +0100391 s->req->rto = s->fe->timeout.client;
392 s->req->wto = s->be->timeout.server;
393 s->req->cto = s->be->timeout.connect;
Willy Tarreaud7971282006-07-29 18:36:34 +0200394
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100395 if ((s->rep = pool_alloc2(pool2_buffer)) == NULL)
396 goto out_fail_rep; /* no memory */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200397
Willy Tarreau54469402006-07-29 16:59:06 +0200398 buffer_init(s->rep);
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200399 s->rep->prod = &s->si[1];
400 s->rep->cons = &s->si[0];
Willy Tarreau48adac52008-08-30 04:58:38 +0200401 s->si[0].ob = s->si[1].ib = s->rep;
Willy Tarreau54469402006-07-29 16:59:06 +0200402
Willy Tarreaud7c30f92007-12-03 01:38:36 +0100403 s->rep->rto = s->be->timeout.server;
404 s->rep->wto = s->fe->timeout.client;
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200405 s->rep->cto = TICK_ETERNITY;
Willy Tarreaud7971282006-07-29 18:36:34 +0200406
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200407 s->req->rex = TICK_ETERNITY;
408 s->req->wex = TICK_ETERNITY;
Willy Tarreauffab5b42008-08-17 18:03:28 +0200409 s->req->analyse_exp = TICK_ETERNITY;
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200410 s->rep->rex = TICK_ETERNITY;
411 s->rep->wex = TICK_ETERNITY;
Willy Tarreauffab5b42008-08-17 18:03:28 +0200412 s->rep->analyse_exp = TICK_ETERNITY;
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200413 t->expire = TICK_ETERNITY;
414
Willy Tarreau7a966482007-04-15 10:58:02 +0200415 fd_insert(cfd);
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200416 fdtab[cfd].owner = &s->si[0];
Willy Tarreaubaaee002006-06-26 02:48:02 +0200417 fdtab[cfd].state = FD_STREADY;
Willy Tarreaue6b98942007-10-29 01:09:36 +0100418 fdtab[cfd].cb[DIR_RD].f = l->proto->read;
Willy Tarreau54469402006-07-29 16:59:06 +0200419 fdtab[cfd].cb[DIR_RD].b = s->req;
Willy Tarreaue6b98942007-10-29 01:09:36 +0100420 fdtab[cfd].cb[DIR_WR].f = l->proto->write;
Willy Tarreau54469402006-07-29 16:59:06 +0200421 fdtab[cfd].cb[DIR_WR].b = s->rep;
Willy Tarreaue94ebd02007-10-09 17:14:37 +0200422 fdtab[cfd].peeraddr = (struct sockaddr *)&s->cli_addr;
423 fdtab[cfd].peerlen = sizeof(s->cli_addr);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200424
425 if ((p->mode == PR_MODE_HTTP && (s->flags & SN_MONITOR)) ||
Willy Tarreau0f772532006-12-23 20:51:41 +0100426 (p->mode == PR_MODE_HEALTH && (p->options & PR_O_HTTP_CHK))) {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200427 /* Either we got a request from a monitoring system on an HTTP instance,
428 * or we're in health check mode with the 'httpchk' option enabled. In
429 * both cases, we return a fake "HTTP/1.0 200 OK" response and we exit.
430 */
Willy Tarreau0f772532006-12-23 20:51:41 +0100431 struct chunk msg = { .str = "HTTP/1.0 200 OK\r\n\r\n", .len = 19 };
Willy Tarreaudded32d2008-11-30 19:48:07 +0100432 stream_int_retnclose(&s->si[0], &msg); /* forge a 200 response */
Willy Tarreau2ade3012009-03-06 19:16:39 +0100433 s->req->analysers = 0;
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200434 t->expire = s->rep->wex;
Willy Tarreau0f772532006-12-23 20:51:41 +0100435 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200436 else if (p->mode == PR_MODE_HEALTH) { /* health check mode, no client reading */
Willy Tarreau0f772532006-12-23 20:51:41 +0100437 struct chunk msg = { .str = "OK\n", .len = 3 };
Willy Tarreaudded32d2008-11-30 19:48:07 +0100438 stream_int_retnclose(&s->si[0], &msg); /* forge an "OK" response */
Willy Tarreau2ade3012009-03-06 19:16:39 +0100439 s->req->analysers = 0;
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200440 t->expire = s->rep->wex;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200441 }
442 else {
Willy Tarreauf161a342007-04-08 16:59:42 +0200443 EV_FD_SET(cfd, DIR_RD);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200444 }
445
Willy Tarreauc65a3ba2008-08-11 23:42:50 +0200446 /* it is important not to call the wakeup function directly but to
447 * pass through task_wakeup(), because this one knows how to apply
448 * priorities to tasks.
449 */
Willy Tarreau721fdbc2009-03-08 12:25:07 +0100450 task_wakeup(t, TASK_WOKEN_INIT);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200451
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100452 p->feconn++; /* beconn will be increased later */
453 if (p->feconn > p->feconn_max)
454 p->feconn_max = p->feconn;
Willy Tarreaua7e76142007-11-03 14:28:39 +0100455
456 if (s->flags & SN_BE_ASSIGNED) {
Willy Tarreau7f062c42009-03-05 18:43:00 +0100457 proxy_inc_be_ctr(s->be);
Willy Tarreaua7e76142007-11-03 14:28:39 +0100458 s->be->beconn++;
459 if (s->be->beconn > s->be->beconn_max)
460 s->be->beconn_max = s->be->beconn;
461 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200462 actconn++;
463 totalconn++;
464
465 // fprintf(stderr, "accepting from %p => %d conn, %d total, task=%p\n", p, actconn, totalconn, t);
Willy Tarreauf1221aa2006-12-17 22:14:12 +0100466 } /* end of while (p->feconn < p->maxconn) */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200467 return 0;
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100468
469 /* Error unrolling */
470 out_fail_rep:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200471 pool_free2(pool2_buffer, s->req);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100472 out_fail_req:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200473 pool_free2(p->hdr_idx_pool, txn->hdr_idx.v);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100474 out_fail_idx:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200475 pool_free2(p->rsp_cap_pool, txn->rsp.cap);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100476 out_fail_rspcap:
Willy Tarreau48d63db2008-08-03 17:41:33 +0200477 pool_free2(p->req_cap_pool, txn->req.cap);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100478 out_fail_reqcap:
479 out_free_task:
Willy Tarreaua4613182009-03-21 18:13:21 +0100480 task_free(t);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100481 out_free_session:
Willy Tarreauf54f8bd2008-11-23 19:53:55 +0100482 LIST_DEL(&s->list);
Willy Tarreau8ced9a42007-11-04 17:51:50 +0100483 pool_free2(pool2_session, s);
484 out_close:
485 close(cfd);
486 return 0;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200487}
488
489
490
Willy Tarreau8797c062007-05-07 00:55:35 +0200491/************************************************************************/
492/* All supported keywords must be declared here. */
493/************************************************************************/
494
495/* set test->ptr to point to the source IPv4/IPv6 address and test->i to the family */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200496static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200497acl_fetch_src(struct proxy *px, struct session *l4, void *l7, int dir,
498 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200499{
500 test->i = l4->cli_addr.ss_family;
501 if (test->i == AF_INET)
502 test->ptr = (void *)&((struct sockaddr_in *)&l4->cli_addr)->sin_addr;
503 else
504 test->ptr = (void *)&((struct sockaddr_in6 *)(&l4->cli_addr))->sin6_addr;
505 test->flags = ACL_TEST_F_READ_ONLY;
506 return 1;
507}
508
509
510/* set test->i to the connexion's source port */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200511static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200512acl_fetch_sport(struct proxy *px, struct session *l4, void *l7, int dir,
513 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200514{
515 if (l4->cli_addr.ss_family == AF_INET)
516 test->i = ntohs(((struct sockaddr_in *)&l4->cli_addr)->sin_port);
517 else
518 test->i = ntohs(((struct sockaddr_in6 *)(&l4->cli_addr))->sin6_port);
519 test->flags = 0;
520 return 1;
521}
522
Willy Tarreau662b2d82007-05-08 19:56:15 +0200523
524/* 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 +0200525static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200526acl_fetch_dst(struct proxy *px, struct session *l4, void *l7, int dir,
527 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau662b2d82007-05-08 19:56:15 +0200528{
529 if (!(l4->flags & SN_FRT_ADDR_SET))
530 get_frt_addr(l4);
531
532 test->i = l4->frt_addr.ss_family;
533 if (test->i == AF_INET)
534 test->ptr = (void *)&((struct sockaddr_in *)&l4->frt_addr)->sin_addr;
535 else
536 test->ptr = (void *)&((struct sockaddr_in6 *)(&l4->frt_addr))->sin6_addr;
537 test->flags = ACL_TEST_F_READ_ONLY;
538 return 1;
539}
540
541
542/* set test->i to the frontend connexion's destination port */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200543static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200544acl_fetch_dport(struct proxy *px, struct session *l4, void *l7, int dir,
545 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau662b2d82007-05-08 19:56:15 +0200546{
547 if (!(l4->flags & SN_FRT_ADDR_SET))
548 get_frt_addr(l4);
549
550 if (l4->frt_addr.ss_family == AF_INET)
551 test->i = ntohs(((struct sockaddr_in *)&l4->frt_addr)->sin_port);
552 else
553 test->i = ntohs(((struct sockaddr_in6 *)(&l4->frt_addr))->sin6_port);
554 test->flags = 0;
555 return 1;
556}
557
Alexandre Cassen5eb1a902007-11-29 15:43:32 +0100558
Willy Tarreau8797c062007-05-07 00:55:35 +0200559/* set test->i to the number of connexions to the proxy */
Willy Tarreaud41f8d82007-06-10 10:06:18 +0200560static int
Willy Tarreau97be1452007-06-10 11:47:14 +0200561acl_fetch_dconn(struct proxy *px, struct session *l4, void *l7, int dir,
562 struct acl_expr *expr, struct acl_test *test)
Willy Tarreau8797c062007-05-07 00:55:35 +0200563{
564 test->i = px->feconn;
565 return 1;
566}
567
568
569/* Note: must not be declared <const> as its list will be overwritten */
570static struct acl_kw_list acl_kws = {{ },{
Willy Tarreau0ceba5a2008-07-25 19:31:03 +0200571 { "src_port", acl_parse_int, acl_fetch_sport, acl_match_int, ACL_USE_TCP_PERMANENT },
572 { "src", acl_parse_ip, acl_fetch_src, acl_match_ip, ACL_USE_TCP4_PERMANENT },
573 { "dst", acl_parse_ip, acl_fetch_dst, acl_match_ip, ACL_USE_TCP4_PERMANENT },
574 { "dst_port", acl_parse_int, acl_fetch_dport, acl_match_int, ACL_USE_TCP_PERMANENT },
Willy Tarreau662b2d82007-05-08 19:56:15 +0200575#if 0
Alexandre Cassen5eb1a902007-11-29 15:43:32 +0100576 { "src_limit", acl_parse_int, acl_fetch_sconn, acl_match_int },
Willy Tarreau8797c062007-05-07 00:55:35 +0200577#endif
Willy Tarreau0ceba5a2008-07-25 19:31:03 +0200578 { "dst_conn", acl_parse_int, acl_fetch_dconn, acl_match_int, ACL_USE_NOTHING },
Willy Tarreau8797c062007-05-07 00:55:35 +0200579 { NULL, NULL, NULL, NULL },
580}};
581
582
583__attribute__((constructor))
584static void __client_init(void)
585{
586 acl_register_keywords(&acl_kws);
587}
588
589
Willy Tarreaubaaee002006-06-26 02:48:02 +0200590/*
591 * Local variables:
592 * c-indent-level: 8
593 * c-basic-offset: 8
594 * End:
595 */