Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 1 | /* |
Willy Tarreau | c7e4238 | 2012-08-24 19:22:53 +0200 | [diff] [blame] | 2 | * include/proto/channel.h |
| 3 | * Channel management definitions, macros and inline functions. |
Willy Tarreau | 7c3c541 | 2009-12-13 15:53:05 +0100 | [diff] [blame] | 4 | * |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 5 | * Copyright (C) 2000-2012 Willy Tarreau - w@1wt.eu |
Willy Tarreau | 7c3c541 | 2009-12-13 15:53:05 +0100 | [diff] [blame] | 6 | * |
| 7 | * This library is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU Lesser General Public |
| 9 | * License as published by the Free Software Foundation, version 2.1 |
| 10 | * exclusively. |
| 11 | * |
| 12 | * This library is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | * Lesser General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU Lesser General Public |
| 18 | * License along with this library; if not, write to the Free Software |
| 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | */ |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 21 | |
Willy Tarreau | c7e4238 | 2012-08-24 19:22:53 +0200 | [diff] [blame] | 22 | #ifndef _PROTO_CHANNEL_H |
| 23 | #define _PROTO_CHANNEL_H |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 24 | |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 25 | #include <stdio.h> |
Willy Tarreau | 0f77253 | 2006-12-23 20:51:41 +0100 | [diff] [blame] | 26 | #include <stdlib.h> |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 27 | #include <string.h> |
Willy Tarreau | 0f77253 | 2006-12-23 20:51:41 +0100 | [diff] [blame] | 28 | |
Willy Tarreau | e3ba5f0 | 2006-06-29 18:54:54 +0200 | [diff] [blame] | 29 | #include <common/config.h> |
Willy Tarreau | c7e4238 | 2012-08-24 19:22:53 +0200 | [diff] [blame] | 30 | #include <common/chunk.h> |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 31 | #include <common/memory.h> |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 32 | #include <common/ticks.h> |
Willy Tarreau | fa64558 | 2007-06-03 15:59:52 +0200 | [diff] [blame] | 33 | #include <common/time.h> |
| 34 | |
Thierry FOURNIER | ac836ba | 2014-12-16 15:41:18 +0100 | [diff] [blame] | 35 | #include <types/channel.h> |
Willy Tarreau | 7c3c541 | 2009-12-13 15:53:05 +0100 | [diff] [blame] | 36 | #include <types/global.h> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 37 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 38 | extern struct pool_head *pool2_channel; |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 39 | |
| 40 | /* perform minimal intializations, report 0 in case of error, 1 if OK. */ |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 41 | int init_channel(); |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 42 | |
Willy Tarreau | 55a6906 | 2012-10-26 00:21:52 +0200 | [diff] [blame] | 43 | unsigned long long __channel_forward(struct channel *chn, unsigned long long bytes); |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 44 | |
| 45 | /* SI-to-channel functions working with buffers */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 46 | int bi_putblk(struct channel *chn, const char *str, int len); |
Willy Tarreau | b034b25 | 2014-12-08 18:14:53 +0100 | [diff] [blame] | 47 | struct buffer *bi_swpbuf(struct channel *chn, struct buffer *buf); |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 48 | int bi_putchr(struct channel *chn, char c); |
| 49 | int bo_inject(struct channel *chn, const char *msg, int len); |
| 50 | int bo_getline(struct channel *chn, char *str, int len); |
| 51 | int bo_getblk(struct channel *chn, char *blk, int len, int offset); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 52 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 53 | /* Initialize all fields in the channel. */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 54 | static inline void channel_init(struct channel *chn) |
Willy Tarreau | 5446940 | 2006-07-29 16:59:06 +0200 | [diff] [blame] | 55 | { |
Willy Tarreau | 2a4b543 | 2014-11-24 11:39:34 +0100 | [diff] [blame] | 56 | chn->buf = &buf_empty; |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 57 | chn->to_forward = 0; |
Willy Tarreau | b145c78 | 2014-02-09 17:45:16 +0100 | [diff] [blame] | 58 | chn->last_read = now_ms; |
Willy Tarreau | 8f39dcd | 2014-02-09 08:31:49 +0100 | [diff] [blame] | 59 | chn->xfer_small = chn->xfer_large = 0; |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 60 | chn->total = 0; |
| 61 | chn->pipe = NULL; |
| 62 | chn->analysers = 0; |
| 63 | chn->cons = NULL; |
| 64 | chn->flags = 0; |
Willy Tarreau | 5446940 | 2006-07-29 16:59:06 +0200 | [diff] [blame] | 65 | } |
| 66 | |
Willy Tarreau | 55a6906 | 2012-10-26 00:21:52 +0200 | [diff] [blame] | 67 | /* Schedule up to <bytes> more bytes to be forwarded via the channel without |
| 68 | * notifying the owner task. Any data pending in the buffer are scheduled to be |
| 69 | * sent as well, in the limit of the number of bytes to forward. This must be |
| 70 | * the only method to use to schedule bytes to be forwarded. If the requested |
| 71 | * number is too large, it is automatically adjusted. The number of bytes taken |
| 72 | * into account is returned. Directly touching ->to_forward will cause lockups |
| 73 | * when buf->o goes down to zero if nobody is ready to push the remaining data. |
| 74 | */ |
| 75 | static inline unsigned long long channel_forward(struct channel *chn, unsigned long long bytes) |
| 76 | { |
| 77 | /* hint: avoid comparisons on long long for the fast case, since if the |
| 78 | * length does not fit in an unsigned it, it will never be forwarded at |
| 79 | * once anyway. |
| 80 | */ |
| 81 | if (bytes <= ~0U) { |
| 82 | unsigned int bytes32 = bytes; |
| 83 | |
| 84 | if (bytes32 <= chn->buf->i) { |
| 85 | /* OK this amount of bytes might be forwarded at once */ |
| 86 | b_adv(chn->buf, bytes32); |
| 87 | return bytes; |
| 88 | } |
| 89 | } |
| 90 | return __channel_forward(chn, bytes); |
| 91 | } |
| 92 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 93 | /*********************************************************************/ |
| 94 | /* These functions are used to compute various channel content sizes */ |
| 95 | /*********************************************************************/ |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 96 | |
Willy Tarreau | 8e21bb9 | 2012-08-24 22:40:29 +0200 | [diff] [blame] | 97 | /* Reports non-zero if the channel is empty, which means both its |
| 98 | * buffer and pipe are empty. The construct looks strange but is |
| 99 | * jump-less and much more efficient on both 32 and 64-bit than |
| 100 | * the boolean test. |
| 101 | */ |
| 102 | static inline unsigned int channel_is_empty(struct channel *c) |
| 103 | { |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 104 | return !(c->buf->o | (long)c->pipe); |
Willy Tarreau | 8e21bb9 | 2012-08-24 22:40:29 +0200 | [diff] [blame] | 105 | } |
| 106 | |
Willy Tarreau | ba0902e | 2015-01-13 14:39:16 +0100 | [diff] [blame] | 107 | /* Returns non-zero if the channel is rewritable, which means that the buffer |
| 108 | * it is attached to has at least <maxrewrite> bytes immediately available. |
| 109 | * This is used to decide when a request or response may be parsed when some |
| 110 | * data from a previous exchange might still be present. |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 111 | */ |
Willy Tarreau | ba0902e | 2015-01-13 14:39:16 +0100 | [diff] [blame] | 112 | static inline int channel_is_rewritable(const struct channel *chn) |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 113 | { |
| 114 | int rem = chn->buf->size; |
| 115 | |
| 116 | rem -= chn->buf->o; |
| 117 | rem -= chn->buf->i; |
| 118 | rem -= global.tune.maxrewrite; |
| 119 | return rem >= 0; |
| 120 | } |
| 121 | |
Willy Tarreau | 9c06ee4 | 2015-01-14 16:08:45 +0100 | [diff] [blame] | 122 | /* Tells whether data are likely to leave the buffer. This is used to know when |
| 123 | * we can safely ignore the reserve since we know we cannot retry a connection. |
| 124 | * It returns zero if data are blocked, non-zero otherwise. |
| 125 | */ |
| 126 | static inline int channel_may_send(const struct channel *chn) |
| 127 | { |
| 128 | return chn->cons->state == SI_ST_EST; |
| 129 | } |
| 130 | |
Willy Tarreau | 1a4484d | 2015-01-13 20:09:54 +0100 | [diff] [blame] | 131 | /* Returns the amount of bytes from the channel that are already scheduled for |
| 132 | * leaving (buf->o) or that are still part of the input and expected to be sent |
| 133 | * soon as covered by to_forward. This is useful to know by how much we can |
Willy Tarreau | 9c06ee4 | 2015-01-14 16:08:45 +0100 | [diff] [blame] | 134 | * shrink the rewrite reserve during forwards. Buffer data are not considered |
| 135 | * in transit until the channel is connected, so that the reserve remains |
| 136 | * protected. |
Willy Tarreau | 1a4484d | 2015-01-13 20:09:54 +0100 | [diff] [blame] | 137 | */ |
| 138 | static inline int channel_in_transit(const struct channel *chn) |
| 139 | { |
| 140 | int ret; |
| 141 | |
Willy Tarreau | 9c06ee4 | 2015-01-14 16:08:45 +0100 | [diff] [blame] | 142 | if (!channel_may_send(chn)) |
| 143 | return 0; |
| 144 | |
Willy Tarreau | 1a4484d | 2015-01-13 20:09:54 +0100 | [diff] [blame] | 145 | /* below, this is min(i, to_forward) optimized for the fast case */ |
| 146 | if (chn->to_forward >= chn->buf->i || |
| 147 | (CHN_INFINITE_FORWARD < MAX_RANGE(typeof(chn->buf->i)) && |
| 148 | chn->to_forward == CHN_INFINITE_FORWARD)) |
| 149 | ret = chn->buf->i; |
| 150 | else |
| 151 | ret = chn->to_forward; |
| 152 | |
| 153 | ret += chn->buf->o; |
| 154 | return ret; |
| 155 | } |
| 156 | |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 157 | /* Returns non-zero if the channel can still receive data. This is used to |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 158 | * decide when to stop reading into a buffer when we want to ensure that we |
| 159 | * leave the reserve untouched after all pending outgoing data are forwarded. |
| 160 | * The reserved space is taken into account if ->to_forward indicates that an |
| 161 | * end of transfer is close to happen. Note that both ->buf->o and ->to_forward |
| 162 | * are considered as available since they're supposed to leave the buffer. The |
| 163 | * test is optimized to avoid as many operations as possible for the fast case |
| 164 | * and to be used as an "if" condition. |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 165 | */ |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 166 | static inline int channel_may_recv(const struct channel *chn) |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 167 | { |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 168 | int rem = chn->buf->size; |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 169 | |
Willy Tarreau | 4428a29 | 2014-11-28 20:54:13 +0100 | [diff] [blame] | 170 | if (chn->buf == &buf_empty) |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 171 | return 1; |
Willy Tarreau | 4428a29 | 2014-11-28 20:54:13 +0100 | [diff] [blame] | 172 | |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 173 | rem -= chn->buf->o; |
| 174 | rem -= chn->buf->i; |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 175 | if (!rem) |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 176 | return 0; /* buffer already full */ |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 177 | |
Willy Tarreau | bb3f994 | 2015-01-13 19:07:23 +0100 | [diff] [blame] | 178 | /* now we know there's some room left, verify if we're touching |
| 179 | * the reserve with some permanent input data. |
| 180 | */ |
| 181 | if (chn->to_forward >= chn->buf->i || |
| 182 | (CHN_INFINITE_FORWARD < MAX_RANGE(typeof(chn->buf->i)) && // just there to ensure gcc |
| 183 | chn->to_forward == CHN_INFINITE_FORWARD)) // avoids the useless second |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 184 | return 1; // test whenever possible |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 185 | |
| 186 | rem -= global.tune.maxrewrite; |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 187 | rem += chn->buf->o; |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 188 | rem += chn->to_forward; |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 189 | return rem > 0; |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 190 | } |
| 191 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 192 | /* Returns true if the channel's input is already closed */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 193 | static inline int channel_input_closed(struct channel *chn) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 194 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 195 | return ((chn->flags & CF_SHUTR) != 0); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 196 | } |
| 197 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 198 | /* Returns true if the channel's output is already closed */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 199 | static inline int channel_output_closed(struct channel *chn) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 200 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 201 | return ((chn->flags & CF_SHUTW) != 0); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 202 | } |
| 203 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 204 | /* Check channel timeouts, and set the corresponding flags. The likely/unlikely |
| 205 | * have been optimized for fastest normal path. The read/write timeouts are not |
| 206 | * set if there was activity on the channel. That way, we don't have to update |
| 207 | * the timeout on every I/O. Note that the analyser timeout is always checked. |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 208 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 209 | static inline void channel_check_timeouts(struct channel *chn) |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 210 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 211 | if (likely(!(chn->flags & (CF_SHUTR|CF_READ_TIMEOUT|CF_READ_ACTIVITY|CF_READ_NOEXP))) && |
| 212 | unlikely(tick_is_expired(chn->rex, now_ms))) |
| 213 | chn->flags |= CF_READ_TIMEOUT; |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 214 | |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 215 | if (likely(!(chn->flags & (CF_SHUTW|CF_WRITE_TIMEOUT|CF_WRITE_ACTIVITY))) && |
| 216 | unlikely(tick_is_expired(chn->wex, now_ms))) |
| 217 | chn->flags |= CF_WRITE_TIMEOUT; |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 218 | |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 219 | if (likely(!(chn->flags & CF_ANA_TIMEOUT)) && |
| 220 | unlikely(tick_is_expired(chn->analyse_exp, now_ms))) |
| 221 | chn->flags |= CF_ANA_TIMEOUT; |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 222 | } |
| 223 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 224 | /* Erase any content from channel <buf> and adjusts flags accordingly. Note |
Willy Tarreau | 0abebcc | 2009-01-08 00:09:41 +0100 | [diff] [blame] | 225 | * that any spliced data is not affected since we may not have any access to |
| 226 | * it. |
Willy Tarreau | e393fe2 | 2008-08-16 22:18:07 +0200 | [diff] [blame] | 227 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 228 | static inline void channel_erase(struct channel *chn) |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 229 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 230 | chn->to_forward = 0; |
Willy Tarreau | 474cf54 | 2014-11-24 10:54:47 +0100 | [diff] [blame] | 231 | b_reset(chn->buf); |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 232 | } |
| 233 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 234 | /* marks the channel as "shutdown" ASAP for reads */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 235 | static inline void channel_shutr_now(struct channel *chn) |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 236 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 237 | chn->flags |= CF_SHUTR_NOW; |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 238 | } |
| 239 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 240 | /* marks the channel as "shutdown" ASAP for writes */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 241 | static inline void channel_shutw_now(struct channel *chn) |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 242 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 243 | chn->flags |= CF_SHUTW_NOW; |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 244 | } |
| 245 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 246 | /* marks the channel as "shutdown" ASAP in both directions */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 247 | static inline void channel_abort(struct channel *chn) |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 248 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 249 | chn->flags |= CF_SHUTR_NOW | CF_SHUTW_NOW; |
| 250 | chn->flags &= ~CF_AUTO_CONNECT; |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 251 | } |
| 252 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 253 | /* allow the consumer to try to establish a new connection. */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 254 | static inline void channel_auto_connect(struct channel *chn) |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 255 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 256 | chn->flags |= CF_AUTO_CONNECT; |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 257 | } |
| 258 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 259 | /* prevent the consumer from trying to establish a new connection, and also |
| 260 | * disable auto shutdown forwarding. |
| 261 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 262 | static inline void channel_dont_connect(struct channel *chn) |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 263 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 264 | chn->flags &= ~(CF_AUTO_CONNECT|CF_AUTO_CLOSE); |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 265 | } |
| 266 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 267 | /* allow the producer to forward shutdown requests */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 268 | static inline void channel_auto_close(struct channel *chn) |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 269 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 270 | chn->flags |= CF_AUTO_CLOSE; |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 271 | } |
| 272 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 273 | /* prevent the producer from forwarding shutdown requests */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 274 | static inline void channel_dont_close(struct channel *chn) |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 275 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 276 | chn->flags &= ~CF_AUTO_CLOSE; |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 277 | } |
| 278 | |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 279 | /* allow the producer to read / poll the input */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 280 | static inline void channel_auto_read(struct channel *chn) |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 281 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 282 | chn->flags &= ~CF_DONT_READ; |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 283 | } |
| 284 | |
| 285 | /* prevent the producer from read / poll the input */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 286 | static inline void channel_dont_read(struct channel *chn) |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 287 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 288 | chn->flags |= CF_DONT_READ; |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 289 | } |
| 290 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 291 | |
| 292 | /*************************************************/ |
| 293 | /* Buffer operations in the context of a channel */ |
| 294 | /*************************************************/ |
| 295 | |
| 296 | |
| 297 | /* Return the number of reserved bytes in the channel's visible |
| 298 | * buffer, which ensures that once all pending data are forwarded, the |
| 299 | * buffer still has global.tune.maxrewrite bytes free. The result is |
| 300 | * between 0 and global.tune.maxrewrite, which is itself smaller than |
Willy Tarreau | 0428a14 | 2015-01-08 11:34:55 +0100 | [diff] [blame] | 301 | * any chn->size. Special care is taken to avoid any possible integer |
| 302 | * overflow in the operations. |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 303 | */ |
Willy Tarreau | a417819 | 2015-01-14 20:16:52 +0100 | [diff] [blame] | 304 | static inline int channel_reserved(const struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 305 | { |
Willy Tarreau | fe57834 | 2015-01-14 14:07:13 +0100 | [diff] [blame] | 306 | int reserved; |
Willy Tarreau | 0428a14 | 2015-01-08 11:34:55 +0100 | [diff] [blame] | 307 | |
Willy Tarreau | fe57834 | 2015-01-14 14:07:13 +0100 | [diff] [blame] | 308 | reserved = global.tune.maxrewrite - channel_in_transit(chn); |
| 309 | if (reserved < 0) |
Willy Tarreau | 0428a14 | 2015-01-08 11:34:55 +0100 | [diff] [blame] | 310 | reserved = 0; |
Willy Tarreau | 0428a14 | 2015-01-08 11:34:55 +0100 | [diff] [blame] | 311 | return reserved; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 312 | } |
| 313 | |
| 314 | /* Return the max number of bytes the buffer can contain so that once all the |
Willy Tarreau | 3f5096d | 2015-01-14 20:21:43 +0100 | [diff] [blame] | 315 | * data in transit are forwarded, the buffer still has global.tune.maxrewrite |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 316 | * bytes free. The result sits between chn->size - maxrewrite and chn->size. |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 317 | */ |
Willy Tarreau | 3f5096d | 2015-01-14 20:21:43 +0100 | [diff] [blame] | 318 | static inline int channel_recv_limit(const struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 319 | { |
Willy Tarreau | a417819 | 2015-01-14 20:16:52 +0100 | [diff] [blame] | 320 | return chn->buf->size - channel_reserved(chn); |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 321 | } |
| 322 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 323 | /* Returns the amount of space available at the input of the buffer, taking the |
| 324 | * reserved space into account if ->to_forward indicates that an end of transfer |
| 325 | * is close to happen. The test is optimized to avoid as many operations as |
| 326 | * possible for the fast case. |
| 327 | */ |
Willy Tarreau | b5051f8 | 2015-01-14 20:25:34 +0100 | [diff] [blame] | 328 | static inline int channel_recv_max(const struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 329 | { |
Willy Tarreau | 27bb0e1 | 2015-01-14 15:56:50 +0100 | [diff] [blame] | 330 | int ret; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 331 | |
Willy Tarreau | 3f5096d | 2015-01-14 20:21:43 +0100 | [diff] [blame] | 332 | ret = channel_recv_limit(chn) - chn->buf->i - chn->buf->o; |
Willy Tarreau | 27bb0e1 | 2015-01-14 15:56:50 +0100 | [diff] [blame] | 333 | if (ret < 0) |
| 334 | ret = 0; |
| 335 | return ret; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 336 | } |
| 337 | |
Willy Tarreau | 319f745 | 2015-01-14 20:32:59 +0100 | [diff] [blame] | 338 | /* Truncate any unread data in the channel's buffer, and disable forwarding. |
| 339 | * Outgoing data are left intact. This is mainly to be used to send error |
| 340 | * messages after existing data. |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 341 | */ |
Willy Tarreau | 319f745 | 2015-01-14 20:32:59 +0100 | [diff] [blame] | 342 | static inline void channel_truncate(struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 343 | { |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 344 | if (!chn->buf->o) |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 345 | return channel_erase(chn); |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 346 | |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 347 | chn->to_forward = 0; |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 348 | if (!chn->buf->i) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 349 | return; |
| 350 | |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 351 | chn->buf->i = 0; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 352 | } |
| 353 | |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 354 | /* |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 355 | * Advance the channel buffer's read pointer by <len> bytes. This is useful |
| 356 | * when data have been read directly from the buffer. It is illegal to call |
| 357 | * this function with <len> causing a wrapping at the end of the buffer. It's |
| 358 | * the caller's responsibility to ensure that <len> is never larger than |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 359 | * chn->o. Channel flag WRITE_PARTIAL is set. |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 360 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 361 | static inline void bo_skip(struct channel *chn, int len) |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 362 | { |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 363 | chn->buf->o -= len; |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 364 | |
Willy Tarreau | 5fb3803 | 2012-12-16 19:39:09 +0100 | [diff] [blame] | 365 | if (buffer_empty(chn->buf)) |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 366 | chn->buf->p = chn->buf->data; |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 367 | |
Willy Tarreau | fb0e920 | 2009-09-23 23:47:55 +0200 | [diff] [blame] | 368 | /* notify that some data was written to the SI from the buffer */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 369 | chn->flags |= CF_WRITE_PARTIAL; |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 370 | } |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 371 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 372 | /* Tries to copy chunk <chunk> into the channel's buffer after length controls. |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 373 | * The chn->o and to_forward pointers are updated. If the channel's input is |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 374 | * closed, -2 is returned. If the block is too large for this buffer, -3 is |
| 375 | * returned. If there is not enough room left in the buffer, -1 is returned. |
| 376 | * Otherwise the number of bytes copied is returned (0 being a valid number). |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 377 | * Channel flag READ_PARTIAL is updated if some data can be transferred. The |
Willy Tarreau | f941cf2 | 2012-08-27 20:53:34 +0200 | [diff] [blame] | 378 | * chunk's length is updated with the number of bytes sent. |
Willy Tarreau | aeac319 | 2009-08-31 08:09:57 +0200 | [diff] [blame] | 379 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 380 | static inline int bi_putchk(struct channel *chn, struct chunk *chunk) |
Willy Tarreau | aeac319 | 2009-08-31 08:09:57 +0200 | [diff] [blame] | 381 | { |
| 382 | int ret; |
| 383 | |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 384 | ret = bi_putblk(chn, chunk->str, chunk->len); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 385 | if (ret > 0) |
| 386 | chunk->len -= ret; |
Willy Tarreau | aeac319 | 2009-08-31 08:09:57 +0200 | [diff] [blame] | 387 | return ret; |
| 388 | } |
| 389 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 390 | /* Tries to copy string <str> at once into the channel's buffer after length |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 391 | * controls. The chn->o and to_forward pointers are updated. If the channel's |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 392 | * input is closed, -2 is returned. If the block is too large for this buffer, |
| 393 | * -3 is returned. If there is not enough room left in the buffer, -1 is |
| 394 | * returned. Otherwise the number of bytes copied is returned (0 being a valid |
| 395 | * number). Channel flag READ_PARTIAL is updated if some data can be |
| 396 | * transferred. |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 397 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 398 | static inline int bi_putstr(struct channel *chn, const char *str) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 399 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 400 | return bi_putblk(chn, str, strlen(str)); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 401 | } |
| 402 | |
| 403 | /* |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 404 | * Return one char from the channel's buffer. If the buffer is empty and the |
| 405 | * channel is closed, return -2. If the buffer is just empty, return -1. The |
| 406 | * buffer's pointer is not advanced, it's up to the caller to call bo_skip(buf, |
| 407 | * 1) when it has consumed the char. Also note that this function respects the |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 408 | * chn->o limit. |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 409 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 410 | static inline int bo_getchr(struct channel *chn) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 411 | { |
| 412 | /* closed or empty + imminent close = -2; empty = -1 */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 413 | if (unlikely((chn->flags & CF_SHUTW) || channel_is_empty(chn))) { |
| 414 | if (chn->flags & (CF_SHUTW|CF_SHUTW_NOW)) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 415 | return -2; |
| 416 | return -1; |
| 417 | } |
Willy Tarreau | 9b28e03 | 2012-10-12 23:49:43 +0200 | [diff] [blame] | 418 | return *buffer_wrap_sub(chn->buf, chn->buf->p - chn->buf->o); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 419 | } |
| 420 | |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 421 | |
Willy Tarreau | c7e4238 | 2012-08-24 19:22:53 +0200 | [diff] [blame] | 422 | #endif /* _PROTO_CHANNEL_H */ |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 423 | |
| 424 | /* |
| 425 | * Local variables: |
| 426 | * c-indent-level: 8 |
| 427 | * c-basic-offset: 8 |
| 428 | * End: |
| 429 | */ |