blob: dbb60e7d11cd7a462fa46dd4a375661838f3f90c [file] [log] [blame]
Willy Tarreaubaaee002006-06-26 02:48:02 +02001/*
2 * Functions operating on SOCK_STREAM and buffers.
3 *
Willy Tarreau0c303ee2008-07-07 00:09:58 +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>
17
18#include <sys/socket.h>
19#include <sys/stat.h>
20#include <sys/types.h>
21
Willy Tarreau2dd0d472006-06-29 17:53:05 +020022#include <common/compat.h>
Willy Tarreaue3ba5f02006-06-29 18:54:54 +020023#include <common/config.h>
Willy Tarreaud6f087e2008-01-18 17:20:13 +010024#include <common/debug.h>
Willy Tarreau83749182007-04-15 20:56:27 +020025#include <common/standard.h>
Willy Tarreau0c303ee2008-07-07 00:09:58 +020026#include <common/ticks.h>
Willy Tarreau2dd0d472006-06-29 17:53:05 +020027#include <common/time.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020028
Willy Tarreau2d212792008-08-27 21:41:35 +020029#include <proto/buffers.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020030#include <proto/client.h>
31#include <proto/fd.h>
Willy Tarreaubaaee002006-06-26 02:48:02 +020032#include <proto/stream_sock.h>
33#include <proto/task.h>
34
35
36/*
Willy Tarreaud7971282006-07-29 18:36:34 +020037 * this function is called on a read event from a stream socket.
Willy Tarreau83749182007-04-15 20:56:27 +020038 * It returns 0 if we have a high confidence that we will not be
39 * able to read more data without polling first. Returns non-zero
40 * otherwise.
Willy Tarreaubaaee002006-06-26 02:48:02 +020041 */
Willy Tarreaud7971282006-07-29 18:36:34 +020042int stream_sock_read(int fd) {
Willy Tarreau0c303ee2008-07-07 00:09:58 +020043 __label__ out_wakeup, out_shutdown_r, out_error;
Willy Tarreaue5ed4062008-08-30 03:17:31 +020044 struct stream_interface *si = fdtab[fd].owner;
Willy Tarreau48adac52008-08-30 04:58:38 +020045 struct buffer *b = si->ib;
Willy Tarreau8a7af602008-05-03 23:07:14 +020046 int ret, max, retval, cur_read;
Willy Tarreaub8949f12007-03-23 22:39:59 +010047 int read_poll = MAX_READ_POLL_LOOPS;
Willy Tarreaubaaee002006-06-26 02:48:02 +020048
49#ifdef DEBUG_FULL
Willy Tarreaud6f087e2008-01-18 17:20:13 +010050 fprintf(stderr,"stream_sock_read : fd=%d, ev=0x%02x, owner=%p\n", fd, fdtab[fd].ev, fdtab[fd].owner);
Willy Tarreaubaaee002006-06-26 02:48:02 +020051#endif
52
Willy Tarreau83749182007-04-15 20:56:27 +020053 retval = 1;
54
Willy Tarreaud6f087e2008-01-18 17:20:13 +010055 /* stop immediately on errors */
56 if (fdtab[fd].state == FD_STERROR || (fdtab[fd].ev & FD_POLL_ERR))
Willy Tarreau6996e152007-04-30 14:37:43 +020057 goto out_error;
Willy Tarreaud6f087e2008-01-18 17:20:13 +010058
59 /* stop here if we reached the end of data */
60 if ((fdtab[fd].ev & (FD_POLL_IN|FD_POLL_HUP)) == FD_POLL_HUP)
61 goto out_shutdown_r;
Willy Tarreau83749182007-04-15 20:56:27 +020062
Willy Tarreau8a7af602008-05-03 23:07:14 +020063 cur_read = 0;
Willy Tarreau6996e152007-04-30 14:37:43 +020064 while (1) {
65 /*
66 * 1. compute the maximum block size we can read at once.
67 */
Willy Tarreau83749182007-04-15 20:56:27 +020068 if (b->l == 0) { /* let's realign the buffer to optimize I/O */
69 b->r = b->w = b->lr = b->data;
70 max = b->rlim - b->data;
71 }
72 else if (b->r > b->w) {
73 max = b->rlim - b->r;
74 }
75 else {
76 max = b->w - b->r;
77 /* FIXME: theorically, if w>0, we shouldn't have rlim < data+size anymore
78 * since it means that the rewrite protection has been removed. This
79 * implies that the if statement can be removed.
80 */
81 if (max > b->rlim - b->data)
Willy Tarreaubaaee002006-06-26 02:48:02 +020082 max = b->rlim - b->data;
Willy Tarreau83749182007-04-15 20:56:27 +020083 }
Willy Tarreau74ab2ac2008-11-23 17:23:07 +010084
Willy Tarreau6996e152007-04-30 14:37:43 +020085 if (unlikely(max == 0)) {
86 /* Not anymore room to store data. This should theorically
87 * never happen, but better safe than sorry !
88 */
Willy Tarreaub0ef7352008-12-14 13:26:20 +010089 si->flags |= SI_FL_WAIT_ROOM;
Willy Tarreaue393fe22008-08-16 22:18:07 +020090 b->flags |= BF_FULL;
Willy Tarreau83749182007-04-15 20:56:27 +020091 EV_FD_CLR(fd, DIR_RD);
Willy Tarreau0c303ee2008-07-07 00:09:58 +020092 b->rex = TICK_ETERNITY;
93 goto out_wakeup;
Willy Tarreau83749182007-04-15 20:56:27 +020094 }
Willy Tarreaubaaee002006-06-26 02:48:02 +020095
Willy Tarreau6996e152007-04-30 14:37:43 +020096 /*
97 * 2. read the largest possible block
98 */
Willy Tarreaubaaee002006-06-26 02:48:02 +020099#ifndef MSG_NOSIGNAL
Willy Tarreau83749182007-04-15 20:56:27 +0200100 {
101 int skerr;
102 socklen_t lskerr = sizeof(skerr);
103
104 ret = getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
105 if (ret == -1 || skerr)
106 ret = -1;
107 else
108 ret = recv(fd, b->r, max, 0);
109 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200110#else
Willy Tarreau83749182007-04-15 20:56:27 +0200111 ret = recv(fd, b->r, max, MSG_NOSIGNAL);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200112#endif
Willy Tarreau83749182007-04-15 20:56:27 +0200113 if (ret > 0) {
114 b->r += ret;
115 b->l += ret;
Willy Tarreau8a7af602008-05-03 23:07:14 +0200116 cur_read += ret;
Willy Tarreaub38903c2008-11-23 21:33:29 +0100117
Willy Tarreau0abebcc2009-01-08 00:09:41 +0100118 /* if we're allowed to directly forward data, we must update send_max */
119 if (b->to_forward > 0) {
120 int fwd = MIN(b->to_forward, ret);
121 b->send_max += fwd;
122 b->to_forward -= fwd;
123 }
Willy Tarreauf890dc92008-12-13 21:12:26 +0100124
Willy Tarreaub38903c2008-11-23 21:33:29 +0100125 if (fdtab[fd].state == FD_STCONN)
126 fdtab[fd].state = FD_STREADY;
127
Willy Tarreau3da77c52008-08-29 09:58:42 +0200128 b->flags |= BF_READ_PARTIAL;
Willy Tarreaue393fe22008-08-16 22:18:07 +0200129 b->flags &= ~BF_EMPTY;
Willy Tarreau74ab2ac2008-11-23 17:23:07 +0100130
Willy Tarreau83749182007-04-15 20:56:27 +0200131 if (b->r == b->data + BUFSIZE) {
132 b->r = b->data; /* wrap around the buffer */
133 }
Willy Tarreau9641e8f2007-03-23 23:02:09 +0100134
Willy Tarreau83749182007-04-15 20:56:27 +0200135 b->total += ret;
Willy Tarreau9641e8f2007-03-23 23:02:09 +0100136
Willy Tarreaue393fe22008-08-16 22:18:07 +0200137 if (b->l >= b->rlim - b->data) {
Willy Tarreau6996e152007-04-30 14:37:43 +0200138 /* The buffer is now full, there's no point in going through
139 * the loop again.
140 */
Willy Tarreau8a7af602008-05-03 23:07:14 +0200141 if (!(b->flags & BF_STREAMER_FAST) && (cur_read == b->l)) {
142 b->xfer_small = 0;
143 b->xfer_large++;
144 if (b->xfer_large >= 3) {
145 /* we call this buffer a fast streamer if it manages
146 * to be filled in one call 3 consecutive times.
147 */
148 b->flags |= (BF_STREAMER | BF_STREAMER_FAST);
149 //fputc('+', stderr);
150 }
151 }
152 else if ((b->flags & (BF_STREAMER | BF_STREAMER_FAST)) &&
153 (cur_read <= BUFSIZE / 2)) {
154 b->xfer_large = 0;
155 b->xfer_small++;
156 if (b->xfer_small >= 2) {
157 /* if the buffer has been at least half full twice,
158 * we receive faster than we send, so at least it
159 * is not a "fast streamer".
160 */
161 b->flags &= ~BF_STREAMER_FAST;
162 //fputc('-', stderr);
163 }
164 }
165 else {
166 b->xfer_small = 0;
167 b->xfer_large = 0;
168 }
169
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100170 si->flags |= SI_FL_WAIT_ROOM;
Willy Tarreaue393fe22008-08-16 22:18:07 +0200171 b->flags |= BF_FULL;
Willy Tarreau6996e152007-04-30 14:37:43 +0200172 EV_FD_CLR(fd, DIR_RD);
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200173 b->rex = TICK_ETERNITY;
174 goto out_wakeup;
Willy Tarreau6996e152007-04-30 14:37:43 +0200175 }
176
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200177 /* if too many bytes were missing from last read, it means that
178 * it's pointless trying to read again because the system does
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100179 * not have them in buffers. BTW, if FD_POLL_HUP was present,
180 * it means that we have reached the end and that the connection
181 * is closed.
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200182 */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100183 if (ret < max) {
Willy Tarreau8a7af602008-05-03 23:07:14 +0200184 if ((b->flags & (BF_STREAMER | BF_STREAMER_FAST)) &&
185 (cur_read <= BUFSIZE / 2)) {
186 b->xfer_large = 0;
187 b->xfer_small++;
188 if (b->xfer_small >= 3) {
189 /* we have read less than half of the buffer in
190 * one pass, and this happened at least 3 times.
191 * This is definitely not a streamer.
192 */
193 b->flags &= ~(BF_STREAMER | BF_STREAMER_FAST);
194 //fputc('!', stderr);
195 }
196 }
Willy Tarreau2bea3a12008-08-28 09:47:43 +0200197 /* unfortunately, on level-triggered events, POLL_HUP
198 * is generally delivered AFTER the system buffer is
199 * empty, so this one might never match.
200 */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100201 if (fdtab[fd].ev & FD_POLL_HUP)
202 goto out_shutdown_r;
Willy Tarreau2bea3a12008-08-28 09:47:43 +0200203
204 /* if a streamer has read few data, it may be because we
205 * have exhausted system buffers. It's not worth trying
206 * again.
207 */
208 if (b->flags & BF_STREAMER)
209 break;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100210 }
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200211
212 /* generally if we read something smaller than 1 or 2 MSS,
Willy Tarreau2bea3a12008-08-28 09:47:43 +0200213 * it means that either we have exhausted the system's
214 * buffers (streamer or question-response protocol) or that
215 * the connection will be closed. Streamers are easily
216 * detected so we return early. For other cases, it's still
217 * better to perform a last read to be sure, because it may
218 * save one complete poll/read/wakeup cycle in case of shutdown.
Willy Tarreau83749182007-04-15 20:56:27 +0200219 */
Willy Tarreau2bea3a12008-08-28 09:47:43 +0200220 if (ret < MIN_RET_FOR_READ_LOOP && b->flags & BF_STREAMER)
Willy Tarreaubaaee002006-06-26 02:48:02 +0200221 break;
Willy Tarreau83749182007-04-15 20:56:27 +0200222
Willy Tarreau6996e152007-04-30 14:37:43 +0200223 if (--read_poll <= 0)
Willy Tarreaubaaee002006-06-26 02:48:02 +0200224 break;
Willy Tarreau83749182007-04-15 20:56:27 +0200225 }
226 else if (ret == 0) {
Willy Tarreau6996e152007-04-30 14:37:43 +0200227 /* connection closed */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100228 goto out_shutdown_r;
Willy Tarreau83749182007-04-15 20:56:27 +0200229 }
Willy Tarreau9f195292007-04-15 21:26:58 +0200230 else if (errno == EAGAIN) {
231 /* Ignore EAGAIN but inform the poller that there is
Willy Tarreauaf78d0f2009-01-08 10:09:08 +0100232 * nothing to read left if we did not read much, ie
233 * less than what we were still expecting to read.
234 * But we may have done some work justifying to notify
235 * the task.
Willy Tarreau9f195292007-04-15 21:26:58 +0200236 */
Willy Tarreauaf78d0f2009-01-08 10:09:08 +0100237 if (cur_read < MIN_RET_FOR_READ_LOOP)
238 retval = 0;
Willy Tarreau83749182007-04-15 20:56:27 +0200239 break;
240 }
241 else {
Willy Tarreau6996e152007-04-30 14:37:43 +0200242 goto out_error;
Willy Tarreau83749182007-04-15 20:56:27 +0200243 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200244 } /* while (1) */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200245
Willy Tarreau6996e152007-04-30 14:37:43 +0200246 /*
247 * The only way to get out of this loop is to have stopped reading
248 * without any error nor close, either by limiting the number of
249 * loops, or because of an EAGAIN. We only rearm the timer if we
250 * have at least read something.
251 */
252
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100253 if ((b->flags & (BF_READ_PARTIAL|BF_FULL|BF_READ_NOEXP)) == BF_READ_PARTIAL)
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200254 b->rex = tick_add_ifset(now_ms, b->rto);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200255
Willy Tarreau3da77c52008-08-29 09:58:42 +0200256 if (!(b->flags & BF_READ_ACTIVITY))
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200257 goto out_skip_wakeup;
Willy Tarreau6996e152007-04-30 14:37:43 +0200258 out_wakeup:
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100259 /* the consumer might be waiting for data */
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100260 if (b->cons->flags & SI_FL_WAIT_DATA && (b->flags & BF_READ_PARTIAL) && !(b->flags & BF_EMPTY))
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100261 b->cons->chk_snd(b->cons);
262
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100263 /* we have to wake up if there is a special event or if we don't have
264 * any more data to forward.
265 */
266 if ((b->flags & (BF_READ_NULL|BF_READ_ERROR|BF_SHUTR)) ||
267 !b->to_forward ||
268 si->state != SI_ST_EST ||
269 b->cons->state != SI_ST_EST)
270 task_wakeup(si->owner, TASK_WOKEN_IO);
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200271
272 out_skip_wakeup:
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100273 fdtab[fd].ev &= ~FD_POLL_IN;
Willy Tarreau83749182007-04-15 20:56:27 +0200274 return retval;
Willy Tarreau6996e152007-04-30 14:37:43 +0200275
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100276 out_shutdown_r:
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200277 /* we received a shutdown */
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100278 fdtab[fd].ev &= ~FD_POLL_HUP;
279 b->flags |= BF_READ_NULL;
Willy Tarreau99126c32008-11-27 10:30:51 +0100280 stream_sock_shutr(si);
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200281 goto out_wakeup;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100282
Willy Tarreau6996e152007-04-30 14:37:43 +0200283 out_error:
Willy Tarreaucff64112008-11-03 06:26:53 +0100284 /* Read error on the file descriptor. We mark the FD as STERROR so
285 * that we don't use it anymore. The error is reported to the stream
286 * interface which will take proper action. We must not perturbate the
287 * buffer because the stream interface wants to ensure transparent
288 * connection retries.
Willy Tarreau6996e152007-04-30 14:37:43 +0200289 */
Willy Tarreaucff64112008-11-03 06:26:53 +0100290
Willy Tarreau6996e152007-04-30 14:37:43 +0200291 fdtab[fd].state = FD_STERROR;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100292 fdtab[fd].ev &= ~FD_POLL_STICKY;
Willy Tarreaucff64112008-11-03 06:26:53 +0100293 si->flags |= SI_FL_ERR;
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200294 task_wakeup(si->owner, TASK_WOKEN_IO);
295 return 1;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200296}
297
298
299/*
Willy Tarreauf8306d52006-07-29 19:01:31 +0200300 * this function is called on a write event from a stream socket.
Willy Tarreau83749182007-04-15 20:56:27 +0200301 * It returns 0 if we have a high confidence that we will not be
302 * able to write more data without polling first. Returns non-zero
303 * otherwise.
Willy Tarreaubaaee002006-06-26 02:48:02 +0200304 */
Willy Tarreauf8306d52006-07-29 19:01:31 +0200305int stream_sock_write(int fd) {
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200306 __label__ out_wakeup, out_error;
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200307 struct stream_interface *si = fdtab[fd].owner;
Willy Tarreau48adac52008-08-30 04:58:38 +0200308 struct buffer *b = si->ob;
Willy Tarreau83749182007-04-15 20:56:27 +0200309 int ret, max, retval;
310 int write_poll = MAX_WRITE_POLL_LOOPS;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200311
312#ifdef DEBUG_FULL
Willy Tarreauf8306d52006-07-29 19:01:31 +0200313 fprintf(stderr,"stream_sock_write : fd=%d, owner=%p\n", fd, fdtab[fd].owner);
Willy Tarreaubaaee002006-06-26 02:48:02 +0200314#endif
315
Willy Tarreau83749182007-04-15 20:56:27 +0200316 retval = 1;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100317 if (fdtab[fd].state == FD_STERROR || (fdtab[fd].ev & FD_POLL_ERR))
Willy Tarreau6996e152007-04-30 14:37:43 +0200318 goto out_error;
Willy Tarreau83749182007-04-15 20:56:27 +0200319
Willy Tarreau6996e152007-04-30 14:37:43 +0200320 while (1) {
Willy Tarreau83749182007-04-15 20:56:27 +0200321 if (b->l == 0) { /* let's realign the buffer to optimize I/O */
322 b->r = b->w = b->lr = b->data;
323 max = 0;
324 }
325 else if (b->r > b->w) {
326 max = b->r - b->w;
327 }
328 else {
329 max = b->data + BUFSIZE - b->w;
330 }
331
Willy Tarreauf890dc92008-12-13 21:12:26 +0100332 /* limit the amount of outgoing data if required */
333 if (max > b->send_max)
334 max = b->send_max;
335
Willy Tarreaubaaee002006-06-26 02:48:02 +0200336 if (max == 0) {
Willy Tarreauf8306d52006-07-29 19:01:31 +0200337 /* may be we have received a connection acknowledgement in TCP mode without data */
Willy Tarreau6996e152007-04-30 14:37:43 +0200338 if (likely(fdtab[fd].state == FD_STCONN)) {
Willy Tarreau6996e152007-04-30 14:37:43 +0200339 /* We have no data to send to check the connection, and
340 * getsockopt() will not inform us whether the connection
341 * is still pending. So we'll reuse connect() to check the
342 * state of the socket. This has the advantage of givig us
343 * the following info :
344 * - error
345 * - connecting (EALREADY, EINPROGRESS)
346 * - connected (EISCONN, 0)
347 */
Willy Tarreaue94ebd02007-10-09 17:14:37 +0200348 if ((connect(fd, fdtab[fd].peeraddr, fdtab[fd].peerlen) == 0))
Willy Tarreau6996e152007-04-30 14:37:43 +0200349 errno = 0;
350
351 if (errno == EALREADY || errno == EINPROGRESS) {
352 retval = 0;
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200353 goto out_may_wakeup;
Willy Tarreauf8306d52006-07-29 19:01:31 +0200354 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200355
356 if (errno && errno != EISCONN)
357 goto out_error;
358
359 /* OK we just need to indicate that we got a connection
360 * and that we wrote nothing.
361 */
362 b->flags |= BF_WRITE_NULL;
363 fdtab[fd].state = FD_STREADY;
Willy Tarreauf8306d52006-07-29 19:01:31 +0200364 }
365
Willy Tarreau6996e152007-04-30 14:37:43 +0200366 /* Funny, we were called to write something but there wasn't
Willy Tarreau4d9b1de2009-01-07 20:10:39 +0100367 * anything. We can get there, for example if we were woken up
368 * on a write event to finish the splice, but the send_max is 0
369 * so we cannot write anything from the buffer. Let's disable
370 * the write event and pretend we never came there.
Willy Tarreau6996e152007-04-30 14:37:43 +0200371 */
Willy Tarreau4d9b1de2009-01-07 20:10:39 +0100372 goto write_nothing;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200373 }
374
375#ifndef MSG_NOSIGNAL
376 {
377 int skerr;
378 socklen_t lskerr = sizeof(skerr);
379
Willy Tarreauc6423482006-10-15 14:59:03 +0200380 ret = getsockopt(fd, SOL_SOCKET, SO_ERROR, &skerr, &lskerr);
381 if (ret == -1 || skerr)
Willy Tarreaubaaee002006-06-26 02:48:02 +0200382 ret = -1;
383 else
384 ret = send(fd, b->w, max, MSG_DONTWAIT);
385 }
386#else
387 ret = send(fd, b->w, max, MSG_DONTWAIT | MSG_NOSIGNAL);
388#endif
389
390 if (ret > 0) {
391 b->l -= ret;
392 b->w += ret;
Willy Tarreauf890dc92008-12-13 21:12:26 +0100393 b->send_max -= ret;
Willy Tarreau74ab2ac2008-11-23 17:23:07 +0100394
Willy Tarreaub38903c2008-11-23 21:33:29 +0100395 if (fdtab[fd].state == FD_STCONN)
396 fdtab[fd].state = FD_STREADY;
397
Willy Tarreau3da77c52008-08-29 09:58:42 +0200398 b->flags |= BF_WRITE_PARTIAL;
Willy Tarreaue393fe22008-08-16 22:18:07 +0200399
400 if (b->l < b->rlim - b->data)
401 b->flags &= ~BF_FULL;
Willy Tarreau74ab2ac2008-11-23 17:23:07 +0100402
Willy Tarreaubaaee002006-06-26 02:48:02 +0200403 if (b->w == b->data + BUFSIZE) {
404 b->w = b->data; /* wrap around the buffer */
405 }
Willy Tarreau83749182007-04-15 20:56:27 +0200406
Willy Tarreaudcef33f2009-01-07 19:33:39 +0100407 if (!b->l && !b->splice_len) {
Willy Tarreaue393fe22008-08-16 22:18:07 +0200408 b->flags |= BF_EMPTY;
Willy Tarreau48adac52008-08-30 04:58:38 +0200409
Willy Tarreau4d9b1de2009-01-07 20:10:39 +0100410 write_nothing:
Willy Tarreau48adac52008-08-30 04:58:38 +0200411 /* Maybe we just wrote the last chunk and need to close ? */
412 if ((b->flags & (BF_SHUTW|BF_EMPTY|BF_HIJACK|BF_WRITE_ENA|BF_SHUTR)) == (BF_EMPTY|BF_WRITE_ENA|BF_SHUTR)) {
413 if (si->state == SI_ST_EST) {
Willy Tarreau99126c32008-11-27 10:30:51 +0100414 stream_sock_shutw(si);
415 b->wex = TICK_ETERNITY;
416 goto out_wakeup;
Willy Tarreau48adac52008-08-30 04:58:38 +0200417 }
418 }
419
Willy Tarreau4d9b1de2009-01-07 20:10:39 +0100420 /* we may either get there when the buffer is empty or when
421 * we refrain from sending due to send_max reached.
422 */
423 if (!b->l && !b->splice_len)
424 si->flags |= SI_FL_WAIT_DATA;
Willy Tarreau6996e152007-04-30 14:37:43 +0200425 EV_FD_CLR(fd, DIR_WR);
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200426 b->wex = TICK_ETERNITY;
427 goto out_wakeup;
Willy Tarreau6996e152007-04-30 14:37:43 +0200428 }
Willy Tarreau83749182007-04-15 20:56:27 +0200429
Willy Tarreauab3e1d32007-06-03 14:10:36 +0200430 /* if the system buffer is full, don't insist */
431 if (ret < max)
432 break;
433
Willy Tarreau6996e152007-04-30 14:37:43 +0200434 if (--write_poll <= 0)
435 break;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200436 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200437 else if (ret == 0 || errno == EAGAIN) {
438 /* nothing written, just pretend we were never called
439 * and wait for the socket to be ready. But we may have
440 * done some work justifying to notify the task.
441 */
Willy Tarreau83749182007-04-15 20:56:27 +0200442 retval = 0;
443 break;
444 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200445 else {
Willy Tarreau6996e152007-04-30 14:37:43 +0200446 goto out_error;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200447 }
Willy Tarreau6996e152007-04-30 14:37:43 +0200448 } /* while (1) */
Willy Tarreaubaaee002006-06-26 02:48:02 +0200449
Willy Tarreau6996e152007-04-30 14:37:43 +0200450 /*
451 * The only way to get out of this loop is to have stopped writing
452 * without any error, either by limiting the number of loops, or
453 * because of an EAGAIN. We only rearm the timer if we have at least
454 * written something.
455 */
456
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100457 if ((b->flags & (BF_WRITE_PARTIAL|BF_EMPTY|BF_SHUTW)) == BF_WRITE_PARTIAL) {
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200458 b->wex = tick_add_ifset(now_ms, b->wto);
Willy Tarreau86491c32008-12-14 09:04:47 +0100459 if (tick_isset(b->wex) & tick_isset(si->ib->rex)) {
Willy Tarreau83749182007-04-15 20:56:27 +0200460 /* FIXME: to prevent the client from expiring read timeouts during writes,
461 * we refresh it. A solution would be to merge read+write timeouts into a
462 * unique one, although that needs some study particularly on full-duplex
463 * TCP connections. */
Willy Tarreau86491c32008-12-14 09:04:47 +0100464 si->ib->rex = b->wex;
Willy Tarreau83749182007-04-15 20:56:27 +0200465 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200466 }
Willy Tarreaubaaee002006-06-26 02:48:02 +0200467
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200468 out_may_wakeup:
Willy Tarreau3da77c52008-08-29 09:58:42 +0200469 if (!(b->flags & BF_WRITE_ACTIVITY))
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200470 goto out_skip_wakeup;
Willy Tarreau6996e152007-04-30 14:37:43 +0200471 out_wakeup:
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100472 /* the producer might be waiting for more room to store data */
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100473 if ((b->prod->flags & SI_FL_WAIT_ROOM) && (b->flags & BF_WRITE_PARTIAL) && !(b->flags & BF_FULL))
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100474 b->prod->chk_rcv(b->prod);
475
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100476 /* we have to wake up if there is a special event or if we don't have
Willy Tarreau0abebcc2009-01-08 00:09:41 +0100477 * any more data to forward and it's not planned to send any more.
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100478 */
479 if ((b->flags & (BF_WRITE_NULL|BF_WRITE_ERROR|BF_SHUTW)) ||
Willy Tarreau0abebcc2009-01-08 00:09:41 +0100480 (!b->to_forward && !b->send_max && !b->splice_len) ||
Willy Tarreau6b66f3e2008-12-14 17:31:54 +0100481 si->state != SI_ST_EST ||
482 b->prod->state != SI_ST_EST)
483 task_wakeup(si->owner, TASK_WOKEN_IO);
Willy Tarreaufa7e1022008-10-19 07:30:41 +0200484
485 out_skip_wakeup:
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100486 fdtab[fd].ev &= ~FD_POLL_OUT;
Willy Tarreau83749182007-04-15 20:56:27 +0200487 return retval;
Willy Tarreau6996e152007-04-30 14:37:43 +0200488
489 out_error:
Willy Tarreaucff64112008-11-03 06:26:53 +0100490 /* Write error on the file descriptor. We mark the FD as STERROR so
491 * that we don't use it anymore. The error is reported to the stream
492 * interface which will take proper action. We must not perturbate the
493 * buffer because the stream interface wants to ensure transparent
494 * connection retries.
Willy Tarreau6996e152007-04-30 14:37:43 +0200495 */
Willy Tarreaucff64112008-11-03 06:26:53 +0100496
Willy Tarreau6996e152007-04-30 14:37:43 +0200497 fdtab[fd].state = FD_STERROR;
Willy Tarreaud6f087e2008-01-18 17:20:13 +0100498 fdtab[fd].ev &= ~FD_POLL_STICKY;
Willy Tarreaucff64112008-11-03 06:26:53 +0100499 si->flags |= SI_FL_ERR;
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200500 task_wakeup(si->owner, TASK_WOKEN_IO);
501 return 1;
Willy Tarreaubaaee002006-06-26 02:48:02 +0200502}
503
Willy Tarreau48adac52008-08-30 04:58:38 +0200504/*
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200505 * This function performs a shutdown-write on a stream interface in a connected or
506 * init state (it does nothing for other states). It either shuts the write side
Willy Tarreau99126c32008-11-27 10:30:51 +0100507 * or closes the file descriptor and marks itself as closed. The buffer flags are
508 * updated to reflect the new state.
Willy Tarreau48adac52008-08-30 04:58:38 +0200509 */
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100510void stream_sock_shutw(struct stream_interface *si)
Willy Tarreau48adac52008-08-30 04:58:38 +0200511{
Willy Tarreau99126c32008-11-27 10:30:51 +0100512 if (si->ob->flags & BF_SHUTW)
513 return;
514 si->ob->flags |= BF_SHUTW;
515 si->ob->wex = TICK_ETERNITY;
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100516 si->flags &= ~SI_FL_WAIT_DATA;
Willy Tarreau99126c32008-11-27 10:30:51 +0100517
Willy Tarreaub38903c2008-11-23 21:33:29 +0100518 switch (si->state) {
Willy Tarreaub38903c2008-11-23 21:33:29 +0100519 case SI_ST_EST:
520 if (!(si->ib->flags & BF_SHUTR)) {
521 EV_FD_CLR(si->fd, DIR_WR);
522 shutdown(si->fd, SHUT_WR);
523 return;
524 }
525 /* fall through */
526 case SI_ST_CON:
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100527 si->flags &= ~SI_FL_WAIT_ROOM;
Willy Tarreau8bfa4262008-11-27 09:25:45 +0100528 /* we may have to close a pending connection, and mark the
529 * response buffer as shutr
530 */
Willy Tarreau48adac52008-08-30 04:58:38 +0200531 fd_delete(si->fd);
Willy Tarreaufe3718a2008-11-30 18:14:12 +0100532 /* fall through */
533 case SI_ST_CER:
Willy Tarreau7f006512008-12-07 14:04:04 +0100534 si->state = SI_ST_DIS;
535 default:
Willy Tarreau99126c32008-11-27 10:30:51 +0100536 si->ib->flags |= BF_SHUTR;
Willy Tarreaufe3718a2008-11-30 18:14:12 +0100537 si->ib->rex = TICK_ETERNITY;
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100538 return;
Willy Tarreau48adac52008-08-30 04:58:38 +0200539 }
Willy Tarreau48adac52008-08-30 04:58:38 +0200540}
Willy Tarreaubaaee002006-06-26 02:48:02 +0200541
Willy Tarreau2d212792008-08-27 21:41:35 +0200542/*
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200543 * This function performs a shutdown-read on a stream interface in a connected or
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100544 * init state (it does nothing for other states). It either shuts the read side
Willy Tarreau99126c32008-11-27 10:30:51 +0100545 * or closes the file descriptor and marks itself as closed. The buffer flags are
546 * updated to reflect the new state.
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200547 */
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100548void stream_sock_shutr(struct stream_interface *si)
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200549{
Willy Tarreau99126c32008-11-27 10:30:51 +0100550 if (si->ib->flags & BF_SHUTR)
551 return;
552 si->ib->flags |= BF_SHUTR;
553 si->ib->rex = TICK_ETERNITY;
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100554 si->flags &= ~SI_FL_WAIT_ROOM;
Willy Tarreau99126c32008-11-27 10:30:51 +0100555
Willy Tarreau8bfa4262008-11-27 09:25:45 +0100556 if (si->state != SI_ST_EST && si->state != SI_ST_CON)
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100557 return;
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200558
Willy Tarreaucff64112008-11-03 06:26:53 +0100559 if (si->ob->flags & BF_SHUTW) {
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200560 fd_delete(si->fd);
Willy Tarreau74ab2ac2008-11-23 17:23:07 +0100561 si->state = SI_ST_DIS;
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100562 return;
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200563 }
564 EV_FD_CLR(si->fd, DIR_RD);
Willy Tarreau0a5d5dd2008-11-23 19:31:35 +0100565 return;
Willy Tarreau3c6ab2e2008-09-04 11:19:41 +0200566}
567
568/*
Willy Tarreau3a16b2c2008-08-28 08:54:27 +0200569 * Updates a connected stream_sock file descriptor status and timeouts
570 * according to the buffers' flags. It should only be called once after the
571 * buffer flags have settled down, and before they are cleared. It doesn't
572 * harm to call it as often as desired (it just slightly hurts performance).
573 */
Willy Tarreaub0253252008-11-30 21:37:12 +0100574void stream_sock_data_finish(struct stream_interface *si)
Willy Tarreau3a16b2c2008-08-28 08:54:27 +0200575{
Willy Tarreaub0253252008-11-30 21:37:12 +0100576 struct buffer *ib = si->ib;
577 struct buffer *ob = si->ob;
578 int fd = si->fd;
Willy Tarreau3a16b2c2008-08-28 08:54:27 +0200579
Willy Tarreaue5ed4062008-08-30 03:17:31 +0200580 DPRINTF(stderr,"[%u] %s: fd=%d owner=%p ib=%p, ob=%p, exp(r,w)=%u,%u ibf=%08x obf=%08x ibl=%d obl=%d si=%d\n",
Willy Tarreau3a16b2c2008-08-28 08:54:27 +0200581 now_ms, __FUNCTION__,
582 fd, fdtab[fd].owner,
583 ib, ob,
584 ib->rex, ob->wex,
585 ib->flags, ob->flags,
Willy Tarreaub0253252008-11-30 21:37:12 +0100586 ib->l, ob->l, si->state);
Willy Tarreau3a16b2c2008-08-28 08:54:27 +0200587
588 /* Check if we need to close the read side */
589 if (!(ib->flags & BF_SHUTR)) {
Willy Tarreau2d212792008-08-27 21:41:35 +0200590 /* Read not closed, update FD status and timeout for reads */
Willy Tarreau3a16b2c2008-08-28 08:54:27 +0200591 if (ib->flags & (BF_FULL|BF_HIJACK)) {
Willy Tarreau2d212792008-08-27 21:41:35 +0200592 /* stop reading */
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100593 if ((ib->flags & (BF_FULL|BF_HIJACK)) == BF_FULL)
594 si->flags |= SI_FL_WAIT_ROOM;
Willy Tarreau2d212792008-08-27 21:41:35 +0200595 EV_FD_COND_C(fd, DIR_RD);
596 ib->rex = TICK_ETERNITY;
597 }
598 else {
599 /* (re)start reading and update timeout. Note: we don't recompute the timeout
600 * everytime we get here, otherwise it would risk never to expire. We only
601 * update it if is was not yet set, or if we already got some read status.
602 */
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100603 si->flags &= ~SI_FL_WAIT_ROOM;
Willy Tarreau2d212792008-08-27 21:41:35 +0200604 EV_FD_COND_S(fd, DIR_RD);
Willy Tarreau86491c32008-12-14 09:04:47 +0100605 if (!(ib->flags & BF_READ_NOEXP) &&
606 (!tick_isset(ib->rex) || ib->flags & BF_READ_ACTIVITY))
Willy Tarreau2d212792008-08-27 21:41:35 +0200607 ib->rex = tick_add_ifset(now_ms, ib->rto);
608 }
609 }
610
611 /* Check if we need to close the write side */
612 if (!(ob->flags & BF_SHUTW)) {
Willy Tarreau2d212792008-08-27 21:41:35 +0200613 /* Write not closed, update FD status and timeout for writes */
Willy Tarreaudcef33f2009-01-07 19:33:39 +0100614 if ((ob->send_max == 0 && ob->splice_len == 0) ||
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100615 (ob->flags & BF_EMPTY) ||
Willy Tarreau3da77c52008-08-29 09:58:42 +0200616 (ob->flags & (BF_HIJACK|BF_WRITE_ENA)) == 0) {
Willy Tarreau2d212792008-08-27 21:41:35 +0200617 /* stop writing */
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100618 if ((ob->flags & (BF_EMPTY|BF_HIJACK|BF_WRITE_ENA)) == (BF_EMPTY|BF_WRITE_ENA))
619 si->flags |= SI_FL_WAIT_DATA;
Willy Tarreau2d212792008-08-27 21:41:35 +0200620 EV_FD_COND_C(fd, DIR_WR);
621 ob->wex = TICK_ETERNITY;
622 }
623 else {
624 /* (re)start writing and update timeout. Note: we don't recompute the timeout
625 * everytime we get here, otherwise it would risk never to expire. We only
626 * update it if is was not yet set, or if we already got some write status.
627 */
Willy Tarreaub0ef7352008-12-14 13:26:20 +0100628 si->flags &= ~SI_FL_WAIT_DATA;
Willy Tarreau2d212792008-08-27 21:41:35 +0200629 EV_FD_COND_S(fd, DIR_WR);
Willy Tarreau3da77c52008-08-29 09:58:42 +0200630 if (!tick_isset(ob->wex) || ob->flags & BF_WRITE_ACTIVITY) {
Willy Tarreau2d212792008-08-27 21:41:35 +0200631 ob->wex = tick_add_ifset(now_ms, ob->wto);
Willy Tarreau21e1be82008-08-29 11:30:14 +0200632 if (tick_isset(ob->wex) && tick_isset(ib->rex)) {
Willy Tarreau2d212792008-08-27 21:41:35 +0200633 /* Note: depending on the protocol, we don't know if we're waiting
634 * for incoming data or not. So in order to prevent the socket from
635 * expiring read timeouts during writes, we refresh the read timeout,
636 * except if it was already infinite.
637 */
638 ib->rex = ob->wex;
639 }
640 }
641 }
642 }
Willy Tarreau2d212792008-08-27 21:41:35 +0200643}
644
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100645/* This function is used for inter-stream-interface calls. It is called by the
646 * consumer to inform the producer side that it may be interested in checking
647 * for free space in the buffer. Note that it intentionally does not update
648 * timeouts, so that we can still check them later at wake-up.
649 */
650void stream_sock_chk_rcv(struct stream_interface *si)
651{
652 struct buffer *ib = si->ib;
653
654 DPRINTF(stderr,"[%u] %s: fd=%d owner=%p ib=%p, ob=%p, exp(r,w)=%u,%u ibf=%08x obf=%08x ibl=%d obl=%d si=%d\n",
655 now_ms, __FUNCTION__,
656 fd, fdtab[fd].owner,
657 ib, ob,
658 ib->rex, ob->wex,
659 ib->flags, ob->flags,
660 ib->l, ob->l, si->state);
661
662 if (unlikely(si->state != SI_ST_EST || (ib->flags & BF_SHUTR)))
663 return;
664
665 if (ib->flags & (BF_FULL|BF_HIJACK)) {
666 /* stop reading */
667 if ((ib->flags & (BF_FULL|BF_HIJACK)) == BF_FULL)
668 si->flags |= SI_FL_WAIT_ROOM;
669 EV_FD_COND_C(si->fd, DIR_RD);
670 }
671 else {
672 /* (re)start reading */
673 si->flags &= ~SI_FL_WAIT_ROOM;
674 EV_FD_COND_S(si->fd, DIR_RD);
675 }
676}
677
678
679/* This function is used for inter-stream-interface calls. It is called by the
680 * producer to inform the consumer side that it may be interested in checking
681 * for data in the buffer. Note that it intentionally does not update timeouts,
682 * so that we can still check them later at wake-up.
683 */
684void stream_sock_chk_snd(struct stream_interface *si)
685{
686 struct buffer *ob = si->ob;
687
688 DPRINTF(stderr,"[%u] %s: fd=%d owner=%p ib=%p, ob=%p, exp(r,w)=%u,%u ibf=%08x obf=%08x ibl=%d obl=%d si=%d\n",
689 now_ms, __FUNCTION__,
690 fd, fdtab[fd].owner,
691 ib, ob,
692 ib->rex, ob->wex,
693 ib->flags, ob->flags,
694 ib->l, ob->l, si->state);
695
696 if (unlikely(si->state != SI_ST_EST || (ob->flags & BF_SHUTW)))
697 return;
698
Willy Tarreaudcef33f2009-01-07 19:33:39 +0100699 if ((ob->send_max == 0 && ob->splice_len == 0) ||
Willy Tarreau3ffeba12008-12-14 14:42:35 +0100700 (ob->flags & BF_EMPTY) ||
701 (ob->flags & (BF_HIJACK|BF_WRITE_ENA)) == 0) {
702 /* stop writing */
703 if ((ob->flags & (BF_EMPTY|BF_HIJACK|BF_WRITE_ENA)) == (BF_EMPTY|BF_WRITE_ENA))
704 si->flags |= SI_FL_WAIT_DATA;
705 EV_FD_COND_C(si->fd, DIR_WR);
706 }
707 else {
708 /* (re)start writing. */
709 si->flags &= ~SI_FL_WAIT_DATA;
710 EV_FD_COND_S(si->fd, DIR_WR);
711 }
712}
713
Willy Tarreaubaaee002006-06-26 02:48:02 +0200714
715/*
716 * Local variables:
717 * c-indent-level: 8
718 * c-basic-offset: 8
719 * End:
720 */