blob: f5aadcd18bed089e51716df469b0b74aec1b0067 [file] [log] [blame]
Willy Tarreaubaaee002006-06-26 02:48:02 +02001/*
Willy Tarreaub277d6e2012-05-11 16:59:14 +02002 * Functions used to send/receive data using SOCK_STREAM sockets.
Willy Tarreaubaaee002006-06-26 02:48:02 +02003 *
Willy Tarreaub277d6e2012-05-11 16:59:14 +02004 * Copyright 2000-2012 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
Willy Tarreau6b4aad42009-01-18 21:59:13 +010013#define _GNU_SOURCE
Willy Tarreaubaaee002006-06-26 02:48:02 +020014#include <errno.h>
15#include <fcntl.h>
16#include <stdio.h>
17#include <stdlib.h>
18
19#include <sys/socket.h>
20#include <sys/stat.h>
21#include <sys/types.h>
22
Dmitry Sivachenkocaf58982009-08-24 15:11:06 +040023#include <netinet/tcp.h>
24
Willy Tarreau2dd0d472006-06-29 17:53:05 +020025#include <common/compat.h>
Willy Tarreaue3ba5f02006-06-29 18:54:54 +020026#include <common/config.h>
Willy Tarreaud6f087e2008-01-18 17:20:13 +010027#include <common/debug.h>
Willy Tarreau83749182007-04-15 20:56:27 +020028#include <common/standard.h>
Willy Tarreau0c303ee2008-07-07 00:09:58 +020029#include <common/ticks.h>
Willy Tarreau2dd0d472006-06-29 17:53:05 +020030#include <common/time.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020031
Willy Tarreau2d212792008-08-27 21:41:35 +020032#include <proto/buffers.h>
Willy Tarreau8b117082012-08-06 15:06:49 +020033#include <proto/connection.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020034#include <proto/fd.h>
Willy Tarreaueb472682010-05-28 18:46:57 +020035#include <proto/freq_ctr.h>
36#include <proto/log.h>
Willy Tarreau3eba98a2009-01-25 13:56:13 +010037#include <proto/pipe.h>
Willy Tarreaufe598a72010-09-21 21:48:23 +020038#include <proto/protocols.h>
Willy Tarreauc63190d2012-05-11 14:23:52 +020039#include <proto/sock_raw.h>
Willy Tarreau73b013b2012-05-21 16:31:45 +020040#include <proto/stream_interface.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020041#include <proto/task.h>
42
Willy Tarreau5bd8c372009-01-19 00:32:22 +010043#include <types/global.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020044
Willy Tarreaub277d6e2012-05-11 16:59:14 +020045/* main event functions used to move data between sockets and buffers */
Willy Tarreauafad0e02012-08-09 14:45:22 +020046static void sock_raw_read(struct connection *conn);
47static void sock_raw_write(struct connection *conn);
Willy Tarreau3d8903f2012-08-06 17:00:18 +020048static void sock_raw_read0(struct stream_interface *si);
Willy Tarreaub277d6e2012-05-11 16:59:14 +020049static void sock_raw_chk_snd(struct stream_interface *si);
50
51
Willy Tarreau6b4aad42009-01-18 21:59:13 +010052#if defined(CONFIG_HAP_LINUX_SPLICE)
Willy Tarreau43d8fb22011-08-22 17:12:02 +020053#include <common/splice.h>
Willy Tarreau5bd8c372009-01-19 00:32:22 +010054
55/* A pipe contains 16 segments max, and it's common to see segments of 1448 bytes
56 * because of timestamps. Use this as a hint for not looping on splice().
57 */
58#define SPLICE_FULL_HINT 16*1448
59
Willy Tarreaua9de3332009-11-28 07:47:10 +010060/* how many data we attempt to splice at once when the buffer is configured for
61 * infinite forwarding */
62#define MAX_SPLICE_AT_ONCE (1<<30)
63
Willy Tarreau5bd8c372009-01-19 00:32:22 +010064/* Returns :
65 * -1 if splice is not possible or not possible anymore and we must switch to
66 * user-land copy (eg: to_forward reached)
Willy Tarreauafad0e02012-08-09 14:45:22 +020067 * 0 otherwise, including errors and close.
Willy Tarreau5bd8c372009-01-19 00:32:22 +010068 * Sets :
69 * BF_READ_NULL
70 * BF_READ_PARTIAL
71 * BF_WRITE_PARTIAL (during copy)
Willy Tarreauba0b63d2009-09-20 08:09:44 +020072 * BF_OUT_EMPTY (during copy)
Willy Tarreau5bd8c372009-01-19 00:32:22 +010073 * SI_FL_ERR
74 * SI_FL_WAIT_ROOM
75 * (SI_FL_WAIT_RECV)
Willy Tarreau3eba98a2009-01-25 13:56:13 +010076 *
77 * This function automatically allocates a pipe from the pipe pool. It also
78 * carefully ensures to clear b->pipe whenever it leaves the pipe empty.
Willy Tarreau5bd8c372009-01-19 00:32:22 +010079 */
Willy Tarreaub277d6e2012-05-11 16:59:14 +020080static int sock_raw_splice_in(struct buffer *b, struct stream_interface *si)
Willy Tarreau5bd8c372009-01-19 00:32:22 +010081{
Willy Tarreau82a04562011-12-11 22:37:06 +010082 static int splice_detects_close;
Willy Tarreaufb7508a2012-05-21 16:47:54 +020083 int fd = si_fd(si);
Willy Tarreau31971e52009-09-20 12:07:52 +020084 int ret;
85 unsigned long max;
Willy Tarreauafad0e02012-08-09 14:45:22 +020086 int retval = 0;
Willy Tarreau5bd8c372009-01-19 00:32:22 +010087
88 if (!b->to_forward)
89 return -1;
90
91 if (!(b->flags & BF_KERN_SPLICING))
92 return -1;
93
Willy Tarreau02d6cfc2012-03-01 18:19:58 +010094 if (buffer_not_empty(b)) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +010095 /* We're embarrassed, there are already data pending in
96 * the buffer and we don't want to have them at two
97 * locations at a time. Let's indicate we need some
98 * place and ask the consumer to hurry.
99 */
100 si->flags |= SI_FL_WAIT_ROOM;
Willy Tarreauf9dabec2012-08-17 17:33:53 +0200101 conn_data_stop_recv(&si->conn);
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100102 b->rex = TICK_ETERNITY;
Willy Tarreau73b013b2012-05-21 16:31:45 +0200103 si_chk_snd(b->cons);
Willy Tarreauafad0e02012-08-09 14:45:22 +0200104 return 0;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100105 }
106
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100107 if (unlikely(b->pipe == NULL)) {
108 if (pipes_used >= global.maxpipes || !(b->pipe = get_pipe())) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100109 b->flags &= ~BF_KERN_SPLICING;
110 return -1;
111 }
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100112 }
113
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100114 /* At this point, b->pipe is valid */
115
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100116 while (1) {
Willy Tarreaua9de3332009-11-28 07:47:10 +0100117 if (b->to_forward == BUF_INFINITE_FORWARD)
118 max = MAX_SPLICE_AT_ONCE;
119 else
120 max = b->to_forward;
121
Willy Tarreau31971e52009-09-20 12:07:52 +0200122 if (!max) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100123 /* It looks like the buffer + the pipe already contain
124 * the maximum amount of data to be transferred. Try to
125 * send those data immediately on the other side if it
126 * is currently waiting.
127 */
128 retval = -1; /* end of forwarding */
129 break;
130 }
131
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100132 ret = splice(fd, NULL, b->pipe->prod, NULL, max,
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100133 SPLICE_F_MOVE|SPLICE_F_NONBLOCK);
134
135 if (ret <= 0) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100136 if (ret == 0) {
Willy Tarreau98b306b2009-01-25 11:11:32 +0100137 /* connection closed. This is only detected by
Willy Tarreau82a04562011-12-11 22:37:06 +0100138 * recent kernels (>= 2.6.27.13). If we notice
139 * it works, we store the info for later use.
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100140 */
Willy Tarreau82a04562011-12-11 22:37:06 +0100141 splice_detects_close = 1;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100142 b->flags |= BF_READ_NULL;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100143 break;
144 }
145
146 if (errno == EAGAIN) {
147 /* there are two reasons for EAGAIN :
148 * - nothing in the socket buffer (standard)
149 * - pipe is full
Willy Tarreau98b306b2009-01-25 11:11:32 +0100150 * - the connection is closed (kernel < 2.6.27.13)
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100151 * Since we don't know if pipe is full, we'll
152 * stop if the pipe is not empty. Anyway, we
153 * will almost always fill/empty the pipe.
154 */
155
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100156 if (b->pipe->data) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100157 si->flags |= SI_FL_WAIT_ROOM;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100158 break;
159 }
160
Willy Tarreau82a04562011-12-11 22:37:06 +0100161 /* We don't know if the connection was closed,
162 * but if we know splice detects close, then we
163 * know it for sure.
Willy Tarreau98b306b2009-01-25 11:11:32 +0100164 * But if we're called upon POLLIN with an empty
Willy Tarreau82a04562011-12-11 22:37:06 +0100165 * pipe and get EAGAIN, it is suspect enough to
Willy Tarreau98b306b2009-01-25 11:11:32 +0100166 * try to fall back to the normal recv scheme
167 * which will be able to deal with the situation.
168 */
Willy Tarreau82a04562011-12-11 22:37:06 +0100169 if (splice_detects_close)
Willy Tarreauafad0e02012-08-09 14:45:22 +0200170 conn_data_poll_recv(&si->conn); /* we know for sure that it's EAGAIN */
Willy Tarreau82a04562011-12-11 22:37:06 +0100171 else
172 retval = -1;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100173 break;
174 }
Willy Tarreaudc340a92009-06-28 23:10:19 +0200175
Willy Tarreaua9de3332009-11-28 07:47:10 +0100176 if (errno == ENOSYS || errno == EINVAL) {
Willy Tarreaudc340a92009-06-28 23:10:19 +0200177 /* splice not supported on this end, disable it */
178 b->flags &= ~BF_KERN_SPLICING;
179 si->flags &= ~SI_FL_CAP_SPLICE;
180 put_pipe(b->pipe);
181 b->pipe = NULL;
182 return -1;
183 }
184
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100185 /* here we have another error */
186 si->flags |= SI_FL_ERR;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100187 break;
188 } /* ret <= 0 */
189
Willy Tarreau31971e52009-09-20 12:07:52 +0200190 if (b->to_forward != BUF_INFINITE_FORWARD)
191 b->to_forward -= ret;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100192 b->total += ret;
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100193 b->pipe->data += ret;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100194 b->flags |= BF_READ_PARTIAL;
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200195 b->flags &= ~BF_OUT_EMPTY;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100196
Willy Tarreau6f4a82c2009-03-21 20:43:57 +0100197 if (b->pipe->data >= SPLICE_FULL_HINT ||
198 ret >= global.tune.recv_enough) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100199 /* We've read enough of it for this time. */
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100200 break;
201 }
202 } /* while */
203
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100204 if (unlikely(!b->pipe->data)) {
205 put_pipe(b->pipe);
206 b->pipe = NULL;
207 }
208
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100209 return retval;
210}
211
Willy Tarreau6b4aad42009-01-18 21:59:13 +0100212#endif /* CONFIG_HAP_LINUX_SPLICE */
213
214
Willy Tarreaubaaee002006-06-26 02:48:02 +0200215/*
Willy Tarreaud7971282006-07-29 18:36:34 +0200216 * this function is called on a read event from a stream socket.
Willy Tarreaubaaee002006-06-26 02:48:02 +0200217 */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200218static void sock_raw_read(struct connection *conn)
Willy Tarreaub277d6e2012-05-11 16:59:14 +0200219{
Willy Tarreau239d7182012-07-23 18:53:03 +0200220 int fd = conn->t.sock.fd;
Willy Tarreau80184712012-07-06 14:54:49 +0200221 struct stream_interface *si = container_of(conn, struct stream_interface, conn);
Willy Tarreau48adac52008-08-30 04:58:38 +0200222 struct buffer *b = si->ib;
Willy Tarreauafad0e02012-08-09 14:45:22 +0200223 int ret, max, cur_read;
Willy Tarreaub8949f12007-03-23 22:39:59 +0100224 int read_poll = MAX_READ_POLL_LOOPS;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200225
226#ifdef DEBUG_FULL
Willy Tarreaub277d6e2012-05-11 16:59:14 +0200227 fprintf(stderr,"sock_raw_read : fd=%d, ev=0x%02x, owner=%p\n", fd, fdtab[fd].ev, fdtab[fd].owner);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200228#endif
Willy Tarreau71543652009-07-14 19:55:05 +0200229 /* stop immediately on errors. Note that we DON'T want to stop on
230 * POLL_ERR, as the poller might report a write error while there
231 * are still data available in the recv buffer. This typically
232 * happens when we send too large a request to a backend server
233 * which rejects it before reading it all.
234 */
Willy Tarreau80184712012-07-06 14:54:49 +0200235 if (conn->flags & CO_FL_ERROR)
Willy Tarreau6996e152007-04-30 14:37:43 +0200236 goto out_error;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100237
238 /* stop here if we reached the end of data */
239 if ((fdtab[fd].ev & (FD_POLL_IN|FD_POLL_HUP)) == FD_POLL_HUP)
240 goto out_shutdown_r;
Willy Tarreau83749182007-04-15 20:56:27 +0200241
Willy Tarreaud06e7112009-03-29 10:18:41 +0200242 /* maybe we were called immediately after an asynchronous shutr */
243 if (b->flags & BF_SHUTR)
Willy Tarreauafad0e02012-08-09 14:45:22 +0200244 return;
Willy Tarreaud06e7112009-03-29 10:18:41 +0200245
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100246#if defined(CONFIG_HAP_LINUX_SPLICE)
Willy Tarreau14acc702011-05-11 20:47:24 +0200247 if (b->to_forward >= MIN_SPLICE_FORWARD && b->flags & BF_KERN_SPLICING) {
Willy Tarreau98b306b2009-01-25 11:11:32 +0100248
249 /* Under Linux, if FD_POLL_HUP is set, we have reached the end.
250 * Since older splice() implementations were buggy and returned
251 * EAGAIN on end of read, let's bypass the call to splice() now.
252 */
253 if (fdtab[fd].ev & FD_POLL_HUP)
254 goto out_shutdown_r;
255
Willy Tarreauafad0e02012-08-09 14:45:22 +0200256 if (sock_raw_splice_in(b, si) >= 0) {
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100257 if (si->flags & SI_FL_ERR)
258 goto out_error;
259 if (b->flags & BF_READ_NULL)
260 goto out_shutdown_r;
Willy Tarreauafad0e02012-08-09 14:45:22 +0200261 return;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100262 }
263 /* splice not possible (anymore), let's go on on standard copy */
264 }
265#endif
Willy Tarreau8a7af602008-05-03 23:07:14 +0200266 cur_read = 0;
Willy Tarreau6996e152007-04-30 14:37:43 +0200267 while (1) {
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200268 max = bi_avail(b);
Willy Tarreau864e8252009-12-28 17:36:37 +0100269
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200270 if (!max) {
Willy Tarreau864e8252009-12-28 17:36:37 +0100271 b->flags |= BF_FULL;
272 si->flags |= SI_FL_WAIT_ROOM;
273 break;
274 }
275
Willy Tarreau6996e152007-04-30 14:37:43 +0200276 /*
277 * 1. compute the maximum block size we can read at once.
278 */
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100279 if (buffer_empty(b)) {
Willy Tarreau03d60bb2009-01-09 11:13:00 +0100280 /* let's realign the buffer to optimize I/O */
Willy Tarreaua458b672012-03-05 11:17:50 +0100281 b->p = b->data;
Willy Tarreau83749182007-04-15 20:56:27 +0200282 }
Willy Tarreau89fa7062012-03-02 16:13:16 +0100283 else if (b->data + b->o < b->p &&
284 b->p + b->i < b->data + b->size) {
Willy Tarreau864e8252009-12-28 17:36:37 +0100285 /* remaining space wraps at the end, with a moving limit */
Willy Tarreau363a5bb2012-03-02 20:14:45 +0100286 if (max > b->data + b->size - (b->p + b->i))
287 max = b->data + b->size - (b->p + b->i);
Willy Tarreau9c0fe592009-01-18 16:25:31 +0100288 }
Willy Tarreau864e8252009-12-28 17:36:37 +0100289 /* else max is already OK */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200290
Willy Tarreau6996e152007-04-30 14:37:43 +0200291 /*
292 * 2. read the largest possible block
293 */
Willy Tarreaucc5cfcb2012-05-04 21:35:27 +0200294 ret = recv(fd, bi_end(b), max, 0);
Willy Tarreaud6d06902009-08-19 11:22:33 +0200295
Willy Tarreau83749182007-04-15 20:56:27 +0200296 if (ret > 0) {
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100297 b->i += ret;
Willy Tarreau8a7af602008-05-03 23:07:14 +0200298 cur_read += ret;
Willy Tarreaub38903c2008-11-23 21:33:29 +0100299
Willy Tarreau2e046c62012-03-01 16:08:30 +0100300 /* if we're allowed to directly forward data, we must update ->o */
Willy Tarreau31971e52009-09-20 12:07:52 +0200301 if (b->to_forward && !(b->flags & (BF_SHUTW|BF_SHUTW_NOW))) {
302 unsigned long fwd = ret;
303 if (b->to_forward != BUF_INFINITE_FORWARD) {
304 if (fwd > b->to_forward)
305 fwd = b->to_forward;
306 b->to_forward -= fwd;
307 }
Willy Tarreau328582c2012-05-05 23:32:27 +0200308 b_adv(b, fwd);
Willy Tarreau0abebcc2009-01-08 00:09:41 +0100309 }
Willy Tarreauf890dc92008-12-13 21:12:26 +0100310
Willy Tarreau80184712012-07-06 14:54:49 +0200311 if (conn->flags & CO_FL_WAIT_L4_CONN) {
312 conn->flags &= ~CO_FL_WAIT_L4_CONN;
Willy Tarreau8ae52cb2012-05-20 10:38:46 +0200313 si->exp = TICK_ETERNITY;
314 }
Willy Tarreaub38903c2008-11-23 21:33:29 +0100315
Willy Tarreau3da77c52008-08-29 09:58:42 +0200316 b->flags |= BF_READ_PARTIAL;
Willy Tarreau83749182007-04-15 20:56:27 +0200317 b->total += ret;
Willy Tarreau9641e8f2007-03-23 23:02:09 +0100318
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200319 if (bi_full(b)) {
Willy Tarreau6996e152007-04-30 14:37:43 +0200320 /* The buffer is now full, there's no point in going through
321 * the loop again.
322 */
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100323 if (!(b->flags & BF_STREAMER_FAST) && (cur_read == buffer_len(b))) {
Willy Tarreau8a7af602008-05-03 23:07:14 +0200324 b->xfer_small = 0;
325 b->xfer_large++;
326 if (b->xfer_large >= 3) {
327 /* we call this buffer a fast streamer if it manages
328 * to be filled in one call 3 consecutive times.
329 */
330 b->flags |= (BF_STREAMER | BF_STREAMER_FAST);
331 //fputc('+', stderr);
332 }
333 }
334 else if ((b->flags & (BF_STREAMER | BF_STREAMER_FAST)) &&
Willy Tarreaua07a34e2009-08-16 23:27:46 +0200335 (cur_read <= b->size / 2)) {
Willy Tarreau8a7af602008-05-03 23:07:14 +0200336 b->xfer_large = 0;
337 b->xfer_small++;
338 if (b->xfer_small >= 2) {
339 /* if the buffer has been at least half full twice,
340 * we receive faster than we send, so at least it
341 * is not a "fast streamer".
342 */
343 b->flags &= ~BF_STREAMER_FAST;
344 //fputc('-', stderr);
345 }
346 }
347 else {
348 b->xfer_small = 0;
349 b->xfer_large = 0;
350 }
Willy Tarreau9c0fe592009-01-18 16:25:31 +0100351
352 b->flags |= BF_FULL;
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100353 si->flags |= SI_FL_WAIT_ROOM;
Willy Tarreau9c0fe592009-01-18 16:25:31 +0100354 break;
Willy Tarreau6996e152007-04-30 14:37:43 +0200355 }
356
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200357 /* if too many bytes were missing from last read, it means that
358 * it's pointless trying to read again because the system does
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100359 * not have them in buffers. BTW, if FD_POLL_HUP was present,
360 * it means that we have reached the end and that the connection
361 * is closed.
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200362 */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100363 if (ret < max) {
Willy Tarreau8a7af602008-05-03 23:07:14 +0200364 if ((b->flags & (BF_STREAMER | BF_STREAMER_FAST)) &&
Willy Tarreaua07a34e2009-08-16 23:27:46 +0200365 (cur_read <= b->size / 2)) {
Willy Tarreau8a7af602008-05-03 23:07:14 +0200366 b->xfer_large = 0;
367 b->xfer_small++;
368 if (b->xfer_small >= 3) {
369 /* we have read less than half of the buffer in
370 * one pass, and this happened at least 3 times.
371 * This is definitely not a streamer.
372 */
373 b->flags &= ~(BF_STREAMER | BF_STREAMER_FAST);
374 //fputc('!', stderr);
375 }
376 }
Willy Tarreau2bea3a12008-08-28 09:47:43 +0200377 /* unfortunately, on level-triggered events, POLL_HUP
378 * is generally delivered AFTER the system buffer is
379 * empty, so this one might never match.
380 */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100381 if (fdtab[fd].ev & FD_POLL_HUP)
382 goto out_shutdown_r;
Willy Tarreau2bea3a12008-08-28 09:47:43 +0200383
384 /* if a streamer has read few data, it may be because we
385 * have exhausted system buffers. It's not worth trying
386 * again.
387 */
388 if (b->flags & BF_STREAMER)
389 break;
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200390
Willy Tarreau6f4a82c2009-03-21 20:43:57 +0100391 /* generally if we read something smaller than 1 or 2 MSS,
392 * it means that either we have exhausted the system's
393 * buffers (streamer or question-response protocol) or
394 * that the connection will be closed. Streamers are
395 * easily detected so we return early. For other cases,
396 * it's still better to perform a last read to be sure,
397 * because it may save one complete poll/read/wakeup cycle
398 * in case of shutdown.
399 */
400 if (ret < MIN_RET_FOR_READ_LOOP && b->flags & BF_STREAMER)
401 break;
402
403 /* if we read a large block smaller than what we requested,
404 * it's almost certain we'll never get anything more.
405 */
406 if (ret >= global.tune.recv_enough)
407 break;
408 }
Willy Tarreau83749182007-04-15 20:56:27 +0200409
Willy Tarreau1b194fe2009-03-21 21:10:04 +0100410 if ((b->flags & BF_READ_DONTWAIT) || --read_poll <= 0)
Willy Tarreaubaaee002006-06-26 02:48:02 +0200411 break;
Willy Tarreau83749182007-04-15 20:56:27 +0200412 }
413 else if (ret == 0) {
Willy Tarreau6996e152007-04-30 14:37:43 +0200414 /* connection closed */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100415 goto out_shutdown_r;
Willy Tarreau83749182007-04-15 20:56:27 +0200416 }
Willy Tarreau9f195292007-04-15 21:26:58 +0200417 else if (errno == EAGAIN) {
418 /* Ignore EAGAIN but inform the poller that there is
Willy Tarreauaf78d0f2009-01-08 10:09:08 +0100419 * nothing to read left if we did not read much, ie
420 * less than what we were still expecting to read.
421 * But we may have done some work justifying to notify
422 * the task.
Willy Tarreau9f195292007-04-15 21:26:58 +0200423 */
Willy Tarreauaf78d0f2009-01-08 10:09:08 +0100424 if (cur_read < MIN_RET_FOR_READ_LOOP)
Willy Tarreauafad0e02012-08-09 14:45:22 +0200425 conn_data_poll_recv(conn);
Willy Tarreau83749182007-04-15 20:56:27 +0200426 break;
427 }
428 else {
Willy Tarreau6996e152007-04-30 14:37:43 +0200429 goto out_error;
Willy Tarreau83749182007-04-15 20:56:27 +0200430 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200431 } /* while (1) */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200432
Willy Tarreauafad0e02012-08-09 14:45:22 +0200433 return;
Willy Tarreau6996e152007-04-30 14:37:43 +0200434
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100435 out_shutdown_r:
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200436 /* we received a shutdown */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100437 fdtab[fd].ev &= ~FD_POLL_HUP;
438 b->flags |= BF_READ_NULL;
Willy Tarreau520d95e2009-09-19 21:04:57 +0200439 if (b->flags & BF_AUTO_CLOSE)
Willy Tarreau418fd472009-09-06 21:37:23 +0200440 buffer_shutw_now(b);
Willy Tarreau3d8903f2012-08-06 17:00:18 +0200441 sock_raw_read0(si);
Willy Tarreauafad0e02012-08-09 14:45:22 +0200442 return;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100443
Willy Tarreau6996e152007-04-30 14:37:43 +0200444 out_error:
Willy Tarreauafad0e02012-08-09 14:45:22 +0200445 /* Read error on the connection, report the error and stop I/O */
Willy Tarreau80184712012-07-06 14:54:49 +0200446 conn->flags |= CO_FL_ERROR;
Willy Tarreauf9dabec2012-08-17 17:33:53 +0200447 conn_data_stop_both(conn);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200448}
449
450
451/*
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100452 * This function is called to send buffer data to a stream socket.
Willy Tarreauafad0e02012-08-09 14:45:22 +0200453 * It returns -1 in case of unrecoverable error, otherwise zero.
Willy Tarreaubaaee002006-06-26 02:48:02 +0200454 */
Willy Tarreaufae44992012-08-20 14:02:10 +0200455static int sock_raw_write_loop(struct connection *conn)
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100456{
Willy Tarreaufae44992012-08-20 14:02:10 +0200457 struct stream_interface *si = container_of(conn, struct stream_interface, conn);
458 struct buffer *b = si->ob;
Willy Tarreau83749182007-04-15 20:56:27 +0200459 int write_poll = MAX_WRITE_POLL_LOOPS;
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100460 int ret, max;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200461
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100462#if defined(CONFIG_HAP_LINUX_SPLICE)
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100463 while (b->pipe) {
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200464 ret = splice(b->pipe->cons, NULL, si_fd(si), NULL, b->pipe->data,
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100465 SPLICE_F_MOVE|SPLICE_F_NONBLOCK);
466 if (ret <= 0) {
467 if (ret == 0 || errno == EAGAIN) {
Willy Tarreauafad0e02012-08-09 14:45:22 +0200468 conn_data_poll_send(&si->conn);
469 return 0;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100470 }
471 /* here we have another error */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200472 return -1;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100473 }
474
475 b->flags |= BF_WRITE_PARTIAL;
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100476 b->pipe->data -= ret;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100477
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100478 if (!b->pipe->data) {
479 put_pipe(b->pipe);
480 b->pipe = NULL;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100481 break;
Willy Tarreau3eba98a2009-01-25 13:56:13 +0100482 }
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100483
484 if (--write_poll <= 0)
Willy Tarreauafad0e02012-08-09 14:45:22 +0200485 return 0;
Willy Tarreaueb9fd512011-12-11 22:11:47 +0100486
487 /* The only reason we did not empty the pipe is that the output
488 * buffer is full.
489 */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200490 conn_data_poll_send(&si->conn);
Willy Tarreaueb9fd512011-12-11 22:11:47 +0100491 return 0;
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100492 }
493
494 /* At this point, the pipe is empty, but we may still have data pending
495 * in the normal buffer.
496 */
Willy Tarreau5bd8c372009-01-19 00:32:22 +0100497#endif
Willy Tarreau2e046c62012-03-01 16:08:30 +0100498 if (!b->o) {
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200499 b->flags |= BF_OUT_EMPTY;
Willy Tarreauafad0e02012-08-09 14:45:22 +0200500 return 0;
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200501 }
Willy Tarreau83749182007-04-15 20:56:27 +0200502
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100503 /* when we're in this loop, we already know that there is no spliced
504 * data left, and that there are sendable buffered data.
505 */
Willy Tarreau6996e152007-04-30 14:37:43 +0200506 while (1) {
Willy Tarreau89fa7062012-03-02 16:13:16 +0100507 max = b->o;
Willy Tarreau83749182007-04-15 20:56:27 +0200508
Willy Tarreau89fa7062012-03-02 16:13:16 +0100509 /* outgoing data may wrap at the end */
510 if (b->data + max > b->p)
511 max = b->data + max - b->p;
Willy Tarreauf890dc92008-12-13 21:12:26 +0100512
Willy Tarreau6db06d32009-08-19 11:14:11 +0200513 /* check if we want to inform the kernel that we're interested in
514 * sending more data after this call. We want this if :
515 * - we're about to close after this last send and want to merge
516 * the ongoing FIN with the last segment.
517 * - we know we can't send everything at once and must get back
518 * here because of unaligned data
Willy Tarreaud38b53b2010-01-03 11:18:34 +0100519 * - there is still a finite amount of data to forward
Willy Tarreau6db06d32009-08-19 11:14:11 +0200520 * The test is arranged so that the most common case does only 2
521 * tests.
Willy Tarreaufb14edc2009-06-14 15:24:37 +0200522 */
Willy Tarreaufb14edc2009-06-14 15:24:37 +0200523
Willy Tarreauface8392010-01-03 11:37:54 +0100524 if (MSG_NOSIGNAL && MSG_MORE) {
Willy Tarreau6db06d32009-08-19 11:14:11 +0200525 unsigned int send_flag = MSG_DONTWAIT | MSG_NOSIGNAL;
526
Willy Tarreau96e31212011-05-30 18:10:30 +0200527 if ((!(b->flags & BF_NEVER_WAIT) &&
528 ((b->to_forward && b->to_forward != BUF_INFINITE_FORWARD) ||
529 (b->flags & BF_EXPECT_MORE))) ||
Willy Tarreau2e046c62012-03-01 16:08:30 +0100530 ((b->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_HIJACK)) == BF_SHUTW_NOW && (max == b->o)) ||
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100531 (max != b->o)) {
Willy Tarreauface8392010-01-03 11:37:54 +0100532 send_flag |= MSG_MORE;
533 }
Willy Tarreau6db06d32009-08-19 11:14:11 +0200534
Willy Tarreau2be39392010-01-03 17:24:51 +0100535 /* this flag has precedence over the rest */
536 if (b->flags & BF_SEND_DONTWAIT)
537 send_flag &= ~MSG_MORE;
538
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200539 ret = send(si_fd(si), bo_ptr(b), max, send_flag);
Willy Tarreaud6d06902009-08-19 11:22:33 +0200540 } else {
Willy Tarreaubaaee002006-06-26 02:48:02 +0200541 int skerr;
542 socklen_t lskerr = sizeof(skerr);
543
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200544 ret = getsockopt(si_fd(si), SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
Willy Tarreauc6423482006-10-15 14:59:03 +0200545 if (ret == -1 || skerr)
Willy Tarreaubaaee002006-06-26 02:48:02 +0200546 ret = -1;
547 else
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200548 ret = send(si_fd(si), bo_ptr(b), max, MSG_DONTWAIT);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200549 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200550
551 if (ret > 0) {
Willy Tarreau505e34a2012-07-06 10:17:53 +0200552 if (si->conn.flags & CO_FL_WAIT_L4_CONN) {
553 si->conn.flags &= ~CO_FL_WAIT_L4_CONN;
Willy Tarreau8ae52cb2012-05-20 10:38:46 +0200554 si->exp = TICK_ETERNITY;
555 }
Willy Tarreaub38903c2008-11-23 21:33:29 +0100556
Willy Tarreau3da77c52008-08-29 09:58:42 +0200557 b->flags |= BF_WRITE_PARTIAL;
Willy Tarreaue393fe22008-08-16 22:18:07 +0200558
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100559 b->o -= ret;
560 if (likely(!buffer_len(b)))
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100561 /* optimize data alignment in the buffer */
Willy Tarreaua458b672012-03-05 11:17:50 +0100562 b->p = b->data;
Willy Tarreau83749182007-04-15 20:56:27 +0200563
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200564 if (likely(!bi_full(b)))
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100565 b->flags &= ~BF_FULL;
566
Willy Tarreau2e046c62012-03-01 16:08:30 +0100567 if (!b->o) {
Willy Tarreauf17810e2012-03-09 18:10:44 +0100568 /* Always clear both flags once everything has been sent, they're one-shot */
569 b->flags &= ~(BF_EXPECT_MORE | BF_SEND_DONTWAIT);
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200570 if (likely(!b->pipe))
571 b->flags |= BF_OUT_EMPTY;
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100572 break;
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200573 }
Willy Tarreau83749182007-04-15 20:56:27 +0200574
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200575 /* if the system buffer is full, don't insist */
576 if (ret < max)
577 break;
578
Willy Tarreau6996e152007-04-30 14:37:43 +0200579 if (--write_poll <= 0)
580 break;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200581 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200582 else if (ret == 0 || errno == EAGAIN) {
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100583 /* nothing written, we need to poll for write first */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200584 conn_data_poll_send(&si->conn);
585 return 0;
Willy Tarreau83749182007-04-15 20:56:27 +0200586 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200587 else {
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100588 /* bad, we got an error */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200589 return -1;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200590 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200591 } /* while (1) */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200592 return 0;
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100593}
Willy Tarreau6996e152007-04-30 14:37:43 +0200594
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100595
596/*
597 * This function is called on a write event from a stream socket.
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100598 */
Willy Tarreauafad0e02012-08-09 14:45:22 +0200599static void sock_raw_write(struct connection *conn)
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100600{
Willy Tarreau80184712012-07-06 14:54:49 +0200601 struct stream_interface *si = container_of(conn, struct stream_interface, conn);
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100602 struct buffer *b = si->ob;
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100603
604#ifdef DEBUG_FULL
Willy Tarreaub277d6e2012-05-11 16:59:14 +0200605 fprintf(stderr,"sock_raw_write : fd=%d, owner=%p\n", fd, fdtab[fd].owner);
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100606#endif
607
Willy Tarreau80184712012-07-06 14:54:49 +0200608 if (conn->flags & CO_FL_ERROR)
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100609 goto out_error;
610
Willy Tarreaud06e7112009-03-29 10:18:41 +0200611 /* we might have been called just after an asynchronous shutw */
612 if (b->flags & BF_SHUTW)
Willy Tarreauafad0e02012-08-09 14:45:22 +0200613 return;
Willy Tarreaud06e7112009-03-29 10:18:41 +0200614
Willy Tarreaufae44992012-08-20 14:02:10 +0200615 if (conn_data_snd_buf(conn) < 0)
Willy Tarreaueeda90e2012-05-11 19:53:32 +0200616 goto out_error;
Willy Tarreau0c2fc1f2009-01-18 15:30:37 +0100617
Willy Tarreauafad0e02012-08-09 14:45:22 +0200618 /* OK all done */
619 return;
Willy Tarreauac128fe2009-01-09 13:05:19 +0100620
Willy Tarreau6996e152007-04-30 14:37:43 +0200621 out_error:
Willy Tarreauafad0e02012-08-09 14:45:22 +0200622 /* Write error on the connection, report the error and stop I/O */
Willy Tarreaucff64112008-11-03 06:26:53 +0100623
Willy Tarreau80184712012-07-06 14:54:49 +0200624 conn->flags |= CO_FL_ERROR;
Willy Tarreauf9dabec2012-08-17 17:33:53 +0200625 conn_data_stop_both(conn);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200626}
627
Willy Tarreau48adac52008-08-30 04:58:38 +0200628/*
Willy Tarreau3d8903f2012-08-06 17:00:18 +0200629 * This function propagates a null read received on a connection. It updates
630 * the stream interface. If the stream interface has SI_FL_NOHALF, we also
631 * forward the close to the write side.
632 */
633static void sock_raw_read0(struct stream_interface *si)
634{
635 si->ib->flags &= ~BF_SHUTR_NOW;
636 if (si->ib->flags & BF_SHUTR)
637 return;
638 si->ib->flags |= BF_SHUTR;
639 si->ib->rex = TICK_ETERNITY;
640 si->flags &= ~SI_FL_WAIT_ROOM;
641
642 if (si->state != SI_ST_EST && si->state != SI_ST_CON)
643 return;
644
645 if (si->ob->flags & BF_SHUTW)
646 goto do_close;
647
648 if (si->flags & SI_FL_NOHALF) {
649 /* we have to shut before closing, otherwise some short messages
650 * may never leave the system, especially when there are remaining
651 * unread data in the socket input buffer, or when nolinger is set.
652 * However, if SI_FL_NOLINGER is explicitly set, we know there is
653 * no risk so we close both sides immediately.
654 */
655 if (si->flags & SI_FL_NOLINGER) {
656 si->flags &= ~SI_FL_NOLINGER;
657 setsockopt(si_fd(si), SOL_SOCKET, SO_LINGER,
658 (struct linger *) &nolinger, sizeof(struct linger));
659 }
660 goto do_close;
661 }
662
663 /* otherwise that's just a normal read shutdown */
Willy Tarreauf9dabec2012-08-17 17:33:53 +0200664 conn_data_stop_recv(&si->conn);
Willy Tarreau3d8903f2012-08-06 17:00:18 +0200665 return;
666
667 do_close:
668 conn_data_close(&si->conn);
669 fd_delete(si_fd(si));
670 si->state = SI_ST_DIS;
671 si->exp = TICK_ETERNITY;
672 if (si->release)
673 si->release(si);
674 return;
675}
676
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100677/* This function is used for inter-stream-interface calls. It is called by the
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100678 * producer to inform the consumer side that it may be interested in checking
679 * for data in the buffer. Note that it intentionally does not update timeouts,
680 * so that we can still check them later at wake-up.
681 */
Willy Tarreaub277d6e2012-05-11 16:59:14 +0200682static void sock_raw_chk_snd(struct stream_interface *si)
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100683{
684 struct buffer *ob = si->ob;
685
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100686 DPRINTF(stderr,"[%u] %s: fd=%d owner=%p ib=%p, ob=%p, exp(r,w)=%u,%u ibf=%08x obf=%08x ibh=%d ibt=%d obh=%d obd=%d si=%d\n",
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100687 now_ms, __FUNCTION__,
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200688 si_fd(si), fdtab[si_fd(si)].owner,
Vincenzo Farruggia9b97cff2009-01-30 16:49:10 +0000689 si->ib, ob,
690 si->ib->rex, ob->wex,
691 si->ib->flags, ob->flags,
Willy Tarreau02d6cfc2012-03-01 18:19:58 +0100692 si->ib->i, si->ib->o, ob->i, ob->o, si->state);
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100693
694 if (unlikely(si->state != SI_ST_EST || (ob->flags & BF_SHUTW)))
695 return;
696
Willy Tarreaua190d592012-05-20 18:35:19 +0200697 if (unlikely(ob->flags & BF_OUT_EMPTY)) /* called with nothing to send ! */
Willy Tarreaueb9fd512011-12-11 22:11:47 +0100698 return;
699
700 if (!ob->pipe && /* spliced data wants to be forwarded ASAP */
701 (!(si->flags & SI_FL_WAIT_DATA) || /* not waiting for data */
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200702 (fdtab[si_fd(si)].ev & FD_POLL_OUT))) /* we'll be called anyway */
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100703 return;
704
Willy Tarreaufae44992012-08-20 14:02:10 +0200705 if (conn_data_snd_buf(&si->conn) < 0) {
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100706 /* Write error on the file descriptor. We mark the FD as STERROR so
707 * that we don't use it anymore and we notify the task.
708 */
Willy Tarreau505e34a2012-07-06 10:17:53 +0200709 si->conn.flags |= CO_FL_ERROR;
Willy Tarreaufb7508a2012-05-21 16:47:54 +0200710 fdtab[si_fd(si)].ev &= ~FD_POLL_STICKY;
Willy Tarreauf9dabec2012-08-17 17:33:53 +0200711 conn_data_stop_both(&si->conn);
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100712 si->flags |= SI_FL_ERR;
713 goto out_wakeup;
714 }
715
Willy Tarreauafad0e02012-08-09 14:45:22 +0200716 /* OK, so now we know that some data might have been sent, and that we may
717 * have to poll first. We have to do that too if the buffer is not empty.
Willy Tarreauc54aef32009-07-27 20:08:06 +0200718 */
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200719 if (ob->flags & BF_OUT_EMPTY) {
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100720 /* the connection is established but we can't write. Either the
721 * buffer is empty, or we just refrain from sending because the
Willy Tarreau2e046c62012-03-01 16:08:30 +0100722 * ->o limit was reached. Maybe we just wrote the last
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100723 * chunk and need to close.
724 */
Willy Tarreau520d95e2009-09-19 21:04:57 +0200725 if (((ob->flags & (BF_SHUTW|BF_HIJACK|BF_AUTO_CLOSE|BF_SHUTW_NOW)) ==
726 (BF_AUTO_CLOSE|BF_SHUTW_NOW)) &&
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100727 (si->state == SI_ST_EST)) {
Willy Tarreau4a36b562012-08-06 19:31:45 +0200728 si_shutw(si);
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100729 goto out_wakeup;
730 }
Willy Tarreaud06e7112009-03-29 10:18:41 +0200731
Willy Tarreau59454bf2009-09-20 11:13:40 +0200732 if ((ob->flags & (BF_SHUTW|BF_SHUTW_NOW|BF_FULL|BF_HIJACK)) == 0)
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100733 si->flags |= SI_FL_WAIT_DATA;
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100734 ob->wex = TICK_ETERNITY;
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100735 }
736 else {
Willy Tarreauc54aef32009-07-27 20:08:06 +0200737 /* Otherwise there are remaining data to be sent in the buffer,
738 * which means we have to poll before doing so.
739 */
Willy Tarreauf9dabec2012-08-17 17:33:53 +0200740 conn_data_want_send(&si->conn);
Willy Tarreauc54aef32009-07-27 20:08:06 +0200741 si->flags &= ~SI_FL_WAIT_DATA;
742 if (!tick_isset(ob->wex))
743 ob->wex = tick_add_ifset(now_ms, ob->wto);
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100744 }
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100745
Willy Tarreauc9619462009-03-09 22:40:57 +0100746 if (likely(ob->flags & BF_WRITE_ACTIVITY)) {
747 /* update timeout if we have written something */
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200748 if ((ob->flags & (BF_OUT_EMPTY|BF_SHUTW|BF_WRITE_PARTIAL)) == BF_WRITE_PARTIAL)
Willy Tarreauc9619462009-03-09 22:40:57 +0100749 ob->wex = tick_add_ifset(now_ms, ob->wto);
750
Willy Tarreauf27b5ea2009-10-03 22:01:18 +0200751 if (tick_isset(si->ib->rex) && !(si->flags & SI_FL_INDEP_STR)) {
Willy Tarreauc9619462009-03-09 22:40:57 +0100752 /* Note: to prevent the client from expiring read timeouts
Willy Tarreauf27b5ea2009-10-03 22:01:18 +0200753 * during writes, we refresh it. We only do this if the
Jamie Gloudon801a0a32012-08-25 00:18:33 -0400754 * interface is not configured for "independent streams",
Willy Tarreauf27b5ea2009-10-03 22:01:18 +0200755 * because for some applications it's better not to do this,
756 * for instance when continuously exchanging small amounts
757 * of data which can full the socket buffers long before a
758 * write timeout is detected.
Willy Tarreauc9619462009-03-09 22:40:57 +0100759 */
760 si->ib->rex = tick_add_ifset(now_ms, si->ib->rto);
761 }
762 }
763
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100764 /* in case of special condition (error, shutdown, end of write...), we
765 * have to notify the task.
766 */
767 if (likely((ob->flags & (BF_WRITE_NULL|BF_WRITE_ERROR|BF_SHUTW)) ||
Willy Tarreauba0b63d2009-09-20 08:09:44 +0200768 ((ob->flags & BF_OUT_EMPTY) && !ob->to_forward) ||
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100769 si->state != SI_ST_EST)) {
770 out_wakeup:
Willy Tarreau89f7ef22009-09-05 20:57:35 +0200771 if (!(si->flags & SI_FL_DONT_WAKE) && si->owner)
772 task_wakeup(si->owner, TASK_WOKEN_IO);
Willy Tarreaua456f2a2009-01-18 17:38:44 +0100773 }
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100774}
775
Willy Tarreau5c979a92012-05-07 17:15:39 +0200776/* stream sock operations */
Willy Tarreaub277d6e2012-05-11 16:59:14 +0200777struct sock_ops sock_raw = {
Willy Tarreau100c4672012-08-20 12:06:26 +0200778 .update = stream_int_update_conn,
Willy Tarreau4a36b562012-08-06 19:31:45 +0200779 .shutr = NULL,
780 .shutw = NULL,
Willy Tarreau46a8d922012-08-20 12:38:36 +0200781 .chk_rcv = stream_int_chk_rcv_conn,
Willy Tarreaub277d6e2012-05-11 16:59:14 +0200782 .chk_snd = sock_raw_chk_snd,
783 .read = sock_raw_read,
784 .write = sock_raw_write,
Willy Tarreaufae44992012-08-20 14:02:10 +0200785 .snd_buf = sock_raw_write_loop,
Willy Tarreau24208272012-05-21 17:28:50 +0200786 .close = NULL,
Willy Tarreau5c979a92012-05-07 17:15:39 +0200787};
Willy Tarreaubaaee002006-06-26 02:48:02 +0200788
789/*
790 * Local variables:
791 * c-indent-level: 8
792 * c-basic-offset: 8
793 * End:
794 */