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 | a27dc19 | 2014-11-27 22:10:04 +0100 | [diff] [blame] | 5 | * Copyright (C) 2000-2014 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 | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 25 | #include <stdint.h> |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 26 | #include <stdio.h> |
Willy Tarreau | 0f77253 | 2006-12-23 20:51:41 +0100 | [diff] [blame] | 27 | #include <stdlib.h> |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 28 | #include <string.h> |
Willy Tarreau | 0f77253 | 2006-12-23 20:51:41 +0100 | [diff] [blame] | 29 | |
Willy Tarreau | e3ba5f0 | 2006-06-29 18:54:54 +0200 | [diff] [blame] | 30 | #include <common/config.h> |
Willy Tarreau | c7e4238 | 2012-08-24 19:22:53 +0200 | [diff] [blame] | 31 | #include <common/chunk.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 | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 37 | #include <types/stream.h> |
Willy Tarreau | 7379653 | 2014-11-28 14:10:28 +0100 | [diff] [blame] | 38 | #include <types/stream_interface.h> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 39 | |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 40 | #include <proto/task.h> |
| 41 | |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 42 | /* perform minimal intializations, report 0 in case of error, 1 if OK. */ |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 43 | int init_channel(); |
Willy Tarreau | 7341d94 | 2007-05-13 19:56:02 +0200 | [diff] [blame] | 44 | |
Willy Tarreau | 55a6906 | 2012-10-26 00:21:52 +0200 | [diff] [blame] | 45 | unsigned long long __channel_forward(struct channel *chn, unsigned long long bytes); |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 46 | |
| 47 | /* SI-to-channel functions working with buffers */ |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 48 | int ci_putblk(struct channel *chn, const char *str, int len); |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 49 | int ci_putchr(struct channel *chn, char c); |
Willy Tarreau | 55f3ce1 | 2018-07-18 11:49:27 +0200 | [diff] [blame] | 50 | int ci_getline_nc(const struct channel *chn, char **blk1, size_t *len1, char **blk2, size_t *len2); |
| 51 | int ci_getblk_nc(const struct channel *chn, char **blk1, size_t *len1, char **blk2, size_t *len2); |
Willy Tarreau | 4d893d4 | 2018-07-12 15:43:32 +0200 | [diff] [blame] | 52 | int ci_insert_line2(struct channel *c, int pos, const char *str, int len); |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 53 | int co_inject(struct channel *chn, const char *msg, int len); |
Willy Tarreau | 41ab868 | 2017-10-19 14:58:40 +0200 | [diff] [blame] | 54 | int co_getline(const struct channel *chn, char *str, int len); |
| 55 | int co_getblk(const struct channel *chn, char *blk, int len, int offset); |
Willy Tarreau | 55f3ce1 | 2018-07-18 11:49:27 +0200 | [diff] [blame] | 56 | int co_getline_nc(const struct channel *chn, const char **blk1, size_t *len1, const char **blk2, size_t *len2); |
| 57 | int co_getblk_nc(const struct channel *chn, const char **blk1, size_t *len1, const char **blk2, size_t *len2); |
Thierry FOURNIER | ca16b03 | 2015-02-16 19:26:48 +0100 | [diff] [blame] | 58 | |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 59 | |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 60 | /* returns a pointer to the stream the channel belongs to */ |
Thierry FOURNIER | 27929fb | 2015-09-25 08:36:11 +0200 | [diff] [blame] | 61 | static inline struct stream *chn_strm(const struct channel *chn) |
Willy Tarreau | d5ccfa3 | 2014-12-28 13:03:53 +0100 | [diff] [blame] | 62 | { |
| 63 | if (chn->flags & CF_ISRESP) |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 64 | return LIST_ELEM(chn, struct stream *, res); |
Willy Tarreau | d5ccfa3 | 2014-12-28 13:03:53 +0100 | [diff] [blame] | 65 | else |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 66 | return LIST_ELEM(chn, struct stream *, req); |
Willy Tarreau | d5ccfa3 | 2014-12-28 13:03:53 +0100 | [diff] [blame] | 67 | } |
| 68 | |
Willy Tarreau | 7379653 | 2014-11-28 14:10:28 +0100 | [diff] [blame] | 69 | /* returns a pointer to the stream interface feeding the channel (producer) */ |
| 70 | static inline struct stream_interface *chn_prod(const struct channel *chn) |
| 71 | { |
Willy Tarreau | 5decc05 | 2014-11-28 14:22:12 +0100 | [diff] [blame] | 72 | if (chn->flags & CF_ISRESP) |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 73 | return &LIST_ELEM(chn, struct stream *, res)->si[1]; |
Willy Tarreau | 5decc05 | 2014-11-28 14:22:12 +0100 | [diff] [blame] | 74 | else |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 75 | return &LIST_ELEM(chn, struct stream *, req)->si[0]; |
Willy Tarreau | 7379653 | 2014-11-28 14:10:28 +0100 | [diff] [blame] | 76 | } |
| 77 | |
| 78 | /* returns a pointer to the stream interface consuming the channel (producer) */ |
| 79 | static inline struct stream_interface *chn_cons(const struct channel *chn) |
| 80 | { |
Willy Tarreau | 5decc05 | 2014-11-28 14:22:12 +0100 | [diff] [blame] | 81 | if (chn->flags & CF_ISRESP) |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 82 | return &LIST_ELEM(chn, struct stream *, res)->si[0]; |
Willy Tarreau | 5decc05 | 2014-11-28 14:22:12 +0100 | [diff] [blame] | 83 | else |
Willy Tarreau | 87b0966 | 2015-04-03 00:22:06 +0200 | [diff] [blame] | 84 | return &LIST_ELEM(chn, struct stream *, req)->si[1]; |
Willy Tarreau | 7379653 | 2014-11-28 14:10:28 +0100 | [diff] [blame] | 85 | } |
| 86 | |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 87 | /* c_orig() : returns the pointer to the channel buffer's origin */ |
| 88 | static inline char *c_orig(const struct channel *c) |
| 89 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 90 | return b_orig(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 91 | } |
| 92 | |
| 93 | /* c_size() : returns the size of the channel's buffer */ |
| 94 | static inline size_t c_size(const struct channel *c) |
| 95 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 96 | return b_size(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 97 | } |
| 98 | |
| 99 | /* c_wrap() : returns the pointer to the channel buffer's wrapping point */ |
| 100 | static inline char *c_wrap(const struct channel *c) |
| 101 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 102 | return b_wrap(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 103 | } |
| 104 | |
| 105 | /* c_data() : returns the amount of data in the channel's buffer */ |
| 106 | static inline size_t c_data(const struct channel *c) |
| 107 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 108 | return b_data(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 109 | } |
| 110 | |
| 111 | /* c_room() : returns the room left in the channel's buffer */ |
| 112 | static inline size_t c_room(const struct channel *c) |
| 113 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 114 | return b_size(&c->buf) - b_data(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 115 | } |
| 116 | |
| 117 | /* c_empty() : returns a boolean indicating if the channel's buffer is empty */ |
| 118 | static inline size_t c_empty(const struct channel *c) |
| 119 | { |
| 120 | return !c_data(c); |
| 121 | } |
| 122 | |
| 123 | /* c_full() : returns a boolean indicating if the channel's buffer is full */ |
| 124 | static inline size_t c_full(const struct channel *c) |
| 125 | { |
| 126 | return !c_room(c); |
| 127 | } |
| 128 | |
| 129 | /* co_data() : returns the amount of output data in the channel's buffer */ |
| 130 | static inline size_t co_data(const struct channel *c) |
| 131 | { |
Olivier Houchard | 08afac0 | 2018-06-22 19:26:39 +0200 | [diff] [blame] | 132 | return c->output; |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 133 | } |
| 134 | |
| 135 | /* ci_data() : returns the amount of input data in the channel's buffer */ |
| 136 | static inline size_t ci_data(const struct channel *c) |
| 137 | { |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 138 | return c_data(c) - co_data(c); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 139 | } |
| 140 | |
| 141 | /* ci_next() : for an absolute pointer <p> or a relative offset <o> pointing to |
| 142 | * a valid location within channel <c>'s buffer, returns either the absolute |
| 143 | * pointer or the relative offset pointing to the next byte, which usually is |
| 144 | * at (p + 1) unless p reaches the wrapping point and wrapping is needed. |
| 145 | */ |
| 146 | static inline size_t ci_next_ofs(const struct channel *c, size_t o) |
| 147 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 148 | return b_next_ofs(&c->buf, o); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 149 | } |
| 150 | static inline char *ci_next(const struct channel *c, const char *p) |
| 151 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 152 | return b_next(&c->buf, p); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 153 | } |
| 154 | |
| 155 | |
| 156 | /* c_ptr() : returns a pointer to an offset relative to the beginning of the |
| 157 | * input data in the buffer. If instead the offset is negative, a pointer to |
| 158 | * existing output data is returned. The function only takes care of wrapping, |
| 159 | * it's up to the caller to ensure the offset is always within byte count |
| 160 | * bounds. |
| 161 | */ |
| 162 | static inline char *c_ptr(const struct channel *c, ssize_t ofs) |
| 163 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 164 | return b_peek(&c->buf, co_data(c) + ofs); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 165 | } |
| 166 | |
| 167 | /* c_adv() : advances the channel's buffer by <adv> bytes, which means that the |
| 168 | * buffer's pointer advances, and that as many bytes from in are transferred |
| 169 | * from in to out. The caller is responsible for ensuring that adv is always |
| 170 | * smaller than or equal to b->i. |
| 171 | */ |
| 172 | static inline void c_adv(struct channel *c, size_t adv) |
| 173 | { |
Olivier Houchard | 08afac0 | 2018-06-22 19:26:39 +0200 | [diff] [blame] | 174 | c->output += adv; |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 175 | } |
| 176 | |
| 177 | /* c_rew() : rewinds the channel's buffer by <adv> bytes, which means that the |
| 178 | * buffer's pointer goes backwards, and that as many bytes from out are moved |
| 179 | * to in. The caller is responsible for ensuring that adv is always smaller |
| 180 | * than or equal to b->o. |
| 181 | */ |
| 182 | static inline void c_rew(struct channel *c, size_t adv) |
| 183 | { |
Olivier Houchard | 08afac0 | 2018-06-22 19:26:39 +0200 | [diff] [blame] | 184 | c->output -= adv; |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 185 | } |
| 186 | |
| 187 | /* c_realign_if_empty() : realign the channel's buffer if it's empty */ |
| 188 | static inline void c_realign_if_empty(struct channel *chn) |
| 189 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 190 | b_realign_if_empty(&chn->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 191 | } |
| 192 | |
Olivier Houchard | d4251a7 | 2018-06-29 16:17:34 +0200 | [diff] [blame] | 193 | /* Sets the amount of output for the channel */ |
| 194 | static inline void co_set_data(struct channel *c, size_t output) |
| 195 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 196 | c->buf.data += output - c->output; |
Olivier Houchard | 08afac0 | 2018-06-22 19:26:39 +0200 | [diff] [blame] | 197 | c->output = output; |
Olivier Houchard | d4251a7 | 2018-06-29 16:17:34 +0200 | [diff] [blame] | 198 | } |
| 199 | |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 200 | |
| 201 | /* co_head() : returns a pointer to the beginning of output data in the buffer. |
| 202 | * The "__" variants don't support wrapping, "ofs" are relative to |
| 203 | * the buffer's origin. |
| 204 | */ |
| 205 | static inline size_t __co_head_ofs(const struct channel *c) |
| 206 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 207 | return __b_peek_ofs(&c->buf, 0); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 208 | } |
| 209 | static inline char *__co_head(const struct channel *c) |
| 210 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 211 | return __b_peek(&c->buf, 0); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 212 | } |
| 213 | static inline size_t co_head_ofs(const struct channel *c) |
| 214 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 215 | return b_peek_ofs(&c->buf, 0); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 216 | } |
| 217 | static inline char *co_head(const struct channel *c) |
| 218 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 219 | return b_peek(&c->buf, 0); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 220 | } |
| 221 | |
| 222 | |
| 223 | /* co_tail() : returns a pointer to the end of output data in the buffer. |
| 224 | * The "__" variants don't support wrapping, "ofs" are relative to |
| 225 | * the buffer's origin. |
| 226 | */ |
| 227 | static inline size_t __co_tail_ofs(const struct channel *c) |
| 228 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 229 | return __b_peek_ofs(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 230 | } |
| 231 | static inline char *__co_tail(const struct channel *c) |
| 232 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 233 | return __b_peek(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 234 | } |
| 235 | static inline size_t co_tail_ofs(const struct channel *c) |
| 236 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 237 | return b_peek_ofs(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 238 | } |
| 239 | static inline char *co_tail(const struct channel *c) |
| 240 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 241 | return b_peek(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 242 | } |
| 243 | |
| 244 | |
| 245 | /* ci_head() : returns a pointer to the beginning of input data in the buffer. |
| 246 | * The "__" variants don't support wrapping, "ofs" are relative to |
| 247 | * the buffer's origin. |
| 248 | */ |
| 249 | static inline size_t __ci_head_ofs(const struct channel *c) |
| 250 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 251 | return __b_peek_ofs(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 252 | } |
| 253 | static inline char *__ci_head(const struct channel *c) |
| 254 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 255 | return __b_peek(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 256 | } |
| 257 | static inline size_t ci_head_ofs(const struct channel *c) |
| 258 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 259 | return b_peek_ofs(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 260 | } |
| 261 | static inline char *ci_head(const struct channel *c) |
| 262 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 263 | return b_peek(&c->buf, co_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 264 | } |
| 265 | |
| 266 | |
| 267 | /* ci_tail() : returns a pointer to the end of input data in the buffer. |
| 268 | * The "__" variants don't support wrapping, "ofs" are relative to |
| 269 | * the buffer's origin. |
| 270 | */ |
| 271 | static inline size_t __ci_tail_ofs(const struct channel *c) |
| 272 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 273 | return __b_peek_ofs(&c->buf, c_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 274 | } |
| 275 | static inline char *__ci_tail(const struct channel *c) |
| 276 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 277 | return __b_peek(&c->buf, c_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 278 | } |
| 279 | static inline size_t ci_tail_ofs(const struct channel *c) |
| 280 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 281 | return b_peek_ofs(&c->buf, c_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 282 | } |
| 283 | static inline char *ci_tail(const struct channel *c) |
| 284 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 285 | return b_peek(&c->buf, c_data(c)); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 286 | } |
| 287 | |
| 288 | |
| 289 | /* ci_stop() : returns the pointer to the byte following the end of input data |
| 290 | * in the channel buffer. It may be out of the buffer. It's used to |
| 291 | * compute lengths or stop pointers. |
| 292 | */ |
| 293 | static inline size_t __ci_stop_ofs(const struct channel *c) |
| 294 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 295 | return __b_stop_ofs(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 296 | } |
| 297 | static inline const char *__ci_stop(const struct channel *c) |
| 298 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 299 | return __b_stop(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 300 | } |
| 301 | static inline size_t ci_stop_ofs(const struct channel *c) |
| 302 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 303 | return b_stop_ofs(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 304 | } |
| 305 | static inline const char *ci_stop(const struct channel *c) |
| 306 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 307 | return b_stop(&c->buf); |
Willy Tarreau | 08d5ac8 | 2018-06-06 15:09:28 +0200 | [diff] [blame] | 308 | } |
| 309 | |
| 310 | |
Willy Tarreau | 7194d3c | 2018-06-06 16:55:45 +0200 | [diff] [blame] | 311 | /* Returns the amount of input data that can contiguously be read at once */ |
| 312 | static inline size_t ci_contig_data(const struct channel *c) |
| 313 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 314 | return b_contig_data(&c->buf, co_data(c)); |
Willy Tarreau | 7194d3c | 2018-06-06 16:55:45 +0200 | [diff] [blame] | 315 | } |
| 316 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 317 | /* Initialize all fields in the channel. */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 318 | static inline void channel_init(struct channel *chn) |
Willy Tarreau | 5446940 | 2006-07-29 16:59:06 +0200 | [diff] [blame] | 319 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 320 | chn->buf = BUF_NULL; |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 321 | chn->to_forward = 0; |
Willy Tarreau | b145c78 | 2014-02-09 17:45:16 +0100 | [diff] [blame] | 322 | chn->last_read = now_ms; |
Willy Tarreau | 8f39dcd | 2014-02-09 08:31:49 +0100 | [diff] [blame] | 323 | chn->xfer_small = chn->xfer_large = 0; |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 324 | chn->total = 0; |
| 325 | chn->pipe = NULL; |
| 326 | chn->analysers = 0; |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 327 | chn->flags = 0; |
Olivier Houchard | 08afac0 | 2018-06-22 19:26:39 +0200 | [diff] [blame] | 328 | chn->output = 0; |
Willy Tarreau | 5446940 | 2006-07-29 16:59:06 +0200 | [diff] [blame] | 329 | } |
| 330 | |
Willy Tarreau | 55a6906 | 2012-10-26 00:21:52 +0200 | [diff] [blame] | 331 | /* Schedule up to <bytes> more bytes to be forwarded via the channel without |
| 332 | * notifying the owner task. Any data pending in the buffer are scheduled to be |
| 333 | * sent as well, in the limit of the number of bytes to forward. This must be |
| 334 | * the only method to use to schedule bytes to be forwarded. If the requested |
| 335 | * number is too large, it is automatically adjusted. The number of bytes taken |
| 336 | * into account is returned. Directly touching ->to_forward will cause lockups |
| 337 | * when buf->o goes down to zero if nobody is ready to push the remaining data. |
| 338 | */ |
| 339 | static inline unsigned long long channel_forward(struct channel *chn, unsigned long long bytes) |
| 340 | { |
| 341 | /* hint: avoid comparisons on long long for the fast case, since if the |
| 342 | * length does not fit in an unsigned it, it will never be forwarded at |
| 343 | * once anyway. |
| 344 | */ |
| 345 | if (bytes <= ~0U) { |
| 346 | unsigned int bytes32 = bytes; |
| 347 | |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 348 | if (bytes32 <= ci_data(chn)) { |
Willy Tarreau | 55a6906 | 2012-10-26 00:21:52 +0200 | [diff] [blame] | 349 | /* OK this amount of bytes might be forwarded at once */ |
Willy Tarreau | bcbd393 | 2018-06-06 07:13:22 +0200 | [diff] [blame] | 350 | c_adv(chn, bytes32); |
Willy Tarreau | 55a6906 | 2012-10-26 00:21:52 +0200 | [diff] [blame] | 351 | return bytes; |
| 352 | } |
| 353 | } |
| 354 | return __channel_forward(chn, bytes); |
| 355 | } |
| 356 | |
Willy Tarreau | 8bf242b | 2016-05-04 14:05:58 +0200 | [diff] [blame] | 357 | /* Forwards any input data and marks the channel for permanent forwarding */ |
| 358 | static inline void channel_forward_forever(struct channel *chn) |
| 359 | { |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 360 | c_adv(chn, ci_data(chn)); |
Willy Tarreau | 8bf242b | 2016-05-04 14:05:58 +0200 | [diff] [blame] | 361 | chn->to_forward = CHN_INFINITE_FORWARD; |
| 362 | } |
| 363 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 364 | /*********************************************************************/ |
| 365 | /* These functions are used to compute various channel content sizes */ |
| 366 | /*********************************************************************/ |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 367 | |
Willy Tarreau | 8e21bb9 | 2012-08-24 22:40:29 +0200 | [diff] [blame] | 368 | /* Reports non-zero if the channel is empty, which means both its |
| 369 | * buffer and pipe are empty. The construct looks strange but is |
| 370 | * jump-less and much more efficient on both 32 and 64-bit than |
| 371 | * the boolean test. |
| 372 | */ |
Willy Tarreau | 41ab868 | 2017-10-19 14:58:40 +0200 | [diff] [blame] | 373 | static inline unsigned int channel_is_empty(const struct channel *c) |
Willy Tarreau | 8e21bb9 | 2012-08-24 22:40:29 +0200 | [diff] [blame] | 374 | { |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 375 | return !(co_data(c) | (long)c->pipe); |
Willy Tarreau | 8e21bb9 | 2012-08-24 22:40:29 +0200 | [diff] [blame] | 376 | } |
| 377 | |
Willy Tarreau | ba0902e | 2015-01-13 14:39:16 +0100 | [diff] [blame] | 378 | /* Returns non-zero if the channel is rewritable, which means that the buffer |
| 379 | * it is attached to has at least <maxrewrite> bytes immediately available. |
| 380 | * This is used to decide when a request or response may be parsed when some |
| 381 | * data from a previous exchange might still be present. |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 382 | */ |
Willy Tarreau | ba0902e | 2015-01-13 14:39:16 +0100 | [diff] [blame] | 383 | static inline int channel_is_rewritable(const struct channel *chn) |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 384 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 385 | int rem = chn->buf.size; |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 386 | |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 387 | rem -= b_data(&chn->buf); |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 388 | rem -= global.tune.maxrewrite; |
| 389 | return rem >= 0; |
| 390 | } |
| 391 | |
Willy Tarreau | 9c06ee4 | 2015-01-14 16:08:45 +0100 | [diff] [blame] | 392 | /* Tells whether data are likely to leave the buffer. This is used to know when |
| 393 | * we can safely ignore the reserve since we know we cannot retry a connection. |
| 394 | * It returns zero if data are blocked, non-zero otherwise. |
| 395 | */ |
| 396 | static inline int channel_may_send(const struct channel *chn) |
| 397 | { |
Willy Tarreau | 7379653 | 2014-11-28 14:10:28 +0100 | [diff] [blame] | 398 | return chn_cons(chn)->state == SI_ST_EST; |
Willy Tarreau | 9c06ee4 | 2015-01-14 16:08:45 +0100 | [diff] [blame] | 399 | } |
| 400 | |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 401 | /* 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] | 402 | * decide when to stop reading into a buffer when we want to ensure that we |
| 403 | * leave the reserve untouched after all pending outgoing data are forwarded. |
| 404 | * The reserved space is taken into account if ->to_forward indicates that an |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 405 | * end of transfer is close to happen. Note that both ->buf.o and ->to_forward |
Willy Tarreau | 379357a | 2013-06-08 12:55:46 +0200 | [diff] [blame] | 406 | * are considered as available since they're supposed to leave the buffer. The |
| 407 | * test is optimized to avoid as many operations as possible for the fast case |
Willy Tarreau | 4b46a3e | 2016-04-20 20:09:22 +0200 | [diff] [blame] | 408 | * and to be used as an "if" condition. Just like channel_recv_limit(), we |
| 409 | * never allow to overwrite the reserve until the output stream interface is |
| 410 | * connected, otherwise we could spin on a POST with http-send-name-header. |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 411 | */ |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 412 | static inline int channel_may_recv(const struct channel *chn) |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 413 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 414 | int rem = chn->buf.size; |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 415 | |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 416 | if (b_is_null(&chn->buf)) |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 417 | return 1; |
Willy Tarreau | 4428a29 | 2014-11-28 20:54:13 +0100 | [diff] [blame] | 418 | |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 419 | rem -= b_data(&chn->buf); |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 420 | if (!rem) |
Willy Tarreau | 3889fff | 2015-01-13 20:20:10 +0100 | [diff] [blame] | 421 | return 0; /* buffer already full */ |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 422 | |
Willy Tarreau | 93dc478 | 2016-04-21 12:12:45 +0200 | [diff] [blame] | 423 | if (rem > global.tune.maxrewrite) |
| 424 | return 1; /* reserve not yet reached */ |
Willy Tarreau | 4b46a3e | 2016-04-20 20:09:22 +0200 | [diff] [blame] | 425 | |
Willy Tarreau | 93dc478 | 2016-04-21 12:12:45 +0200 | [diff] [blame] | 426 | if (!channel_may_send(chn)) |
| 427 | return 0; /* don't touch reserve until we can send */ |
Willy Tarreau | 9dab5fc | 2012-05-07 11:56:55 +0200 | [diff] [blame] | 428 | |
Willy Tarreau | 93dc478 | 2016-04-21 12:12:45 +0200 | [diff] [blame] | 429 | /* Now we know there's some room left in the reserve and we may |
| 430 | * forward. As long as i-to_fwd < size-maxrw, we may still |
| 431 | * receive. This is equivalent to i+maxrw-size < to_fwd, |
| 432 | * which is logical since i+maxrw-size is what overlaps with |
| 433 | * the reserve, and we want to ensure they're covered by scheduled |
| 434 | * forwards. |
| 435 | */ |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 436 | rem = ci_data(chn) + global.tune.maxrewrite - chn->buf.size; |
Willy Tarreau | 93dc478 | 2016-04-21 12:12:45 +0200 | [diff] [blame] | 437 | return rem < 0 || (unsigned int)rem < chn->to_forward; |
Willy Tarreau | 4b517ca | 2011-11-25 20:33:58 +0100 | [diff] [blame] | 438 | } |
| 439 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 440 | /* Returns true if the channel's input is already closed */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 441 | static inline int channel_input_closed(struct channel *chn) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 442 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 443 | return ((chn->flags & CF_SHUTR) != 0); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 444 | } |
| 445 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 446 | /* Returns true if the channel's output is already closed */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 447 | static inline int channel_output_closed(struct channel *chn) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 448 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 449 | return ((chn->flags & CF_SHUTW) != 0); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 450 | } |
| 451 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 452 | /* Check channel timeouts, and set the corresponding flags. The likely/unlikely |
| 453 | * have been optimized for fastest normal path. The read/write timeouts are not |
| 454 | * set if there was activity on the channel. That way, we don't have to update |
| 455 | * 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] | 456 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 457 | static inline void channel_check_timeouts(struct channel *chn) |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 458 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 459 | if (likely(!(chn->flags & (CF_SHUTR|CF_READ_TIMEOUT|CF_READ_ACTIVITY|CF_READ_NOEXP))) && |
| 460 | unlikely(tick_is_expired(chn->rex, now_ms))) |
| 461 | chn->flags |= CF_READ_TIMEOUT; |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 462 | |
Christopher Faulet | c5a9d5b | 2017-11-09 09:36:43 +0100 | [diff] [blame] | 463 | if (likely(!(chn->flags & (CF_SHUTW|CF_WRITE_TIMEOUT|CF_WRITE_ACTIVITY|CF_WRITE_EVENT))) && |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 464 | unlikely(tick_is_expired(chn->wex, now_ms))) |
| 465 | chn->flags |= CF_WRITE_TIMEOUT; |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 466 | |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 467 | if (likely(!(chn->flags & CF_ANA_TIMEOUT)) && |
| 468 | unlikely(tick_is_expired(chn->analyse_exp, now_ms))) |
| 469 | chn->flags |= CF_ANA_TIMEOUT; |
Willy Tarreau | 2eb52f0 | 2008-09-04 09:14:08 +0200 | [diff] [blame] | 470 | } |
| 471 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 472 | /* Erase any content from channel <buf> and adjusts flags accordingly. Note |
Willy Tarreau | 0abebcc | 2009-01-08 00:09:41 +0100 | [diff] [blame] | 473 | * that any spliced data is not affected since we may not have any access to |
| 474 | * it. |
Willy Tarreau | e393fe2 | 2008-08-16 22:18:07 +0200 | [diff] [blame] | 475 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 476 | static inline void channel_erase(struct channel *chn) |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 477 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 478 | chn->to_forward = 0; |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 479 | b_reset(&chn->buf); |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 480 | } |
| 481 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 482 | /* marks the channel as "shutdown" ASAP for reads */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 483 | static inline void channel_shutr_now(struct channel *chn) |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 484 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 485 | chn->flags |= CF_SHUTR_NOW; |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 486 | } |
| 487 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 488 | /* marks the channel as "shutdown" ASAP for writes */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 489 | static inline void channel_shutw_now(struct channel *chn) |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 490 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 491 | chn->flags |= CF_SHUTW_NOW; |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 492 | } |
| 493 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 494 | /* marks the channel as "shutdown" ASAP in both directions */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 495 | static inline void channel_abort(struct channel *chn) |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 496 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 497 | chn->flags |= CF_SHUTR_NOW | CF_SHUTW_NOW; |
| 498 | chn->flags &= ~CF_AUTO_CONNECT; |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 499 | } |
| 500 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 501 | /* allow the consumer to try to establish a new connection. */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 502 | static inline void channel_auto_connect(struct channel *chn) |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 503 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 504 | chn->flags |= CF_AUTO_CONNECT; |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 505 | } |
| 506 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 507 | /* prevent the consumer from trying to establish a new connection, and also |
| 508 | * disable auto shutdown forwarding. |
| 509 | */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 510 | static inline void channel_dont_connect(struct channel *chn) |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 511 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 512 | chn->flags &= ~(CF_AUTO_CONNECT|CF_AUTO_CLOSE); |
Willy Tarreau | 3da77c5 | 2008-08-29 09:58:42 +0200 | [diff] [blame] | 513 | } |
| 514 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 515 | /* allow the producer to forward shutdown requests */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 516 | static inline void channel_auto_close(struct channel *chn) |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 517 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 518 | chn->flags |= CF_AUTO_CLOSE; |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 519 | } |
| 520 | |
Willy Tarreau | 520d95e | 2009-09-19 21:04:57 +0200 | [diff] [blame] | 521 | /* prevent the producer from forwarding shutdown requests */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 522 | static inline void channel_dont_close(struct channel *chn) |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 523 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 524 | chn->flags &= ~CF_AUTO_CLOSE; |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 525 | } |
| 526 | |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 527 | /* allow the producer to read / poll the input */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 528 | static inline void channel_auto_read(struct channel *chn) |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 529 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 530 | chn->flags &= ~CF_DONT_READ; |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 531 | } |
| 532 | |
| 533 | /* prevent the producer from read / poll the input */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 534 | static inline void channel_dont_read(struct channel *chn) |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 535 | { |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 536 | chn->flags |= CF_DONT_READ; |
Willy Tarreau | 90deb18 | 2010-01-07 00:20:41 +0100 | [diff] [blame] | 537 | } |
| 538 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 539 | |
| 540 | /*************************************************/ |
| 541 | /* Buffer operations in the context of a channel */ |
| 542 | /*************************************************/ |
| 543 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 544 | |
| 545 | /* Return the max number of bytes the buffer can contain so that once all the |
Willy Tarreau | 169c470 | 2016-04-20 18:05:17 +0200 | [diff] [blame] | 546 | * pending bytes are forwarded, the buffer still has global.tune.maxrewrite |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 547 | * bytes free. The result sits between chn->size - maxrewrite and chn->size. |
Willy Tarreau | 169c470 | 2016-04-20 18:05:17 +0200 | [diff] [blame] | 548 | * It is important to mention that if buf->i is already larger than size-maxrw |
| 549 | * the condition above cannot be satisfied and the lowest size will be returned |
| 550 | * anyway. The principles are the following : |
| 551 | * 0) the empty buffer has a limit of zero |
| 552 | * 1) a non-connected buffer cannot touch the reserve |
| 553 | * 2) infinite forward can always fill the buffer since all data will leave |
| 554 | * 3) all output bytes are considered in transit since they're leaving |
| 555 | * 4) all input bytes covered by to_forward are considered in transit since |
| 556 | * they'll be converted to output bytes. |
| 557 | * 5) all input bytes not covered by to_forward as considered remaining |
| 558 | * 6) all bytes scheduled to be forwarded minus what is already in the input |
| 559 | * buffer will be in transit during future rounds. |
| 560 | * 7) 4+5+6 imply that the amount of input bytes (i) is irrelevant to the max |
| 561 | * usable length, only to_forward and output count. The difference is |
| 562 | * visible when to_forward > i. |
| 563 | * 8) the reserve may be covered up to the amount of bytes in transit since |
| 564 | * these bytes will only take temporary space. |
Willy Tarreau | 999f643 | 2016-01-25 01:09:11 +0100 | [diff] [blame] | 565 | * |
Willy Tarreau | 169c470 | 2016-04-20 18:05:17 +0200 | [diff] [blame] | 566 | * A typical buffer looks like this : |
Willy Tarreau | 999f643 | 2016-01-25 01:09:11 +0100 | [diff] [blame] | 567 | * |
Willy Tarreau | 169c470 | 2016-04-20 18:05:17 +0200 | [diff] [blame] | 568 | * <-------------- max_len -----------> |
| 569 | * <---- o ----><----- i -----> <--- 0..maxrewrite ---> |
| 570 | * +------------+--------------+-------+----------------------+ |
| 571 | * |////////////|\\\\\\\\\\\\\\|xxxxxxx| reserve | |
| 572 | * +------------+--------+-----+-------+----------------------+ |
| 573 | * <- fwd -> <-avail-> |
| 574 | * |
| 575 | * Or when to_forward > i : |
| 576 | * |
| 577 | * <-------------- max_len -----------> |
| 578 | * <---- o ----><----- i -----> <--- 0..maxrewrite ---> |
| 579 | * +------------+--------------+-------+----------------------+ |
| 580 | * |////////////|\\\\\\\\\\\\\\|xxxxxxx| reserve | |
| 581 | * +------------+--------+-----+-------+----------------------+ |
| 582 | * <-avail-> |
| 583 | * <------------------ fwd ----------------> |
| 584 | * |
| 585 | * - the amount of buffer bytes in transit is : min(i, fwd) + o |
| 586 | * - some scheduled bytes may be in transit (up to fwd - i) |
| 587 | * - the reserve is max(0, maxrewrite - transit) |
| 588 | * - the maximum usable buffer length is size - reserve. |
| 589 | * - the available space is max_len - i - o |
| 590 | * |
| 591 | * So the formula to compute the buffer's maximum length to protect the reserve |
| 592 | * when reading new data is : |
| 593 | * |
| 594 | * max = size - maxrewrite + min(maxrewrite, transit) |
| 595 | * = size - max(maxrewrite - transit, 0) |
| 596 | * |
| 597 | * But WARNING! The conditions might change during the transfer and it could |
| 598 | * very well happen that a buffer would contain more bytes than max_len due to |
| 599 | * i+o already walking over the reserve (eg: after a header rewrite), including |
| 600 | * i or o alone hitting the limit. So it is critical to always consider that |
| 601 | * bounds may have already been crossed and that available space may be negative |
| 602 | * for example. Due to this it is perfectly possible for this function to return |
| 603 | * a value that is lower than current i+o. |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 604 | */ |
Willy Tarreau | 3f5096d | 2015-01-14 20:21:43 +0100 | [diff] [blame] | 605 | static inline int channel_recv_limit(const struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 606 | { |
Willy Tarreau | ef907fe | 2016-05-03 17:46:24 +0200 | [diff] [blame] | 607 | unsigned int transit; |
Willy Tarreau | 999f643 | 2016-01-25 01:09:11 +0100 | [diff] [blame] | 608 | int reserve; |
| 609 | |
| 610 | /* return zero if empty */ |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 611 | reserve = chn->buf.size; |
| 612 | if (b_is_null(&chn->buf)) |
Willy Tarreau | 999f643 | 2016-01-25 01:09:11 +0100 | [diff] [blame] | 613 | goto end; |
| 614 | |
| 615 | /* return size - maxrewrite if we can't send */ |
| 616 | reserve = global.tune.maxrewrite; |
| 617 | if (unlikely(!channel_may_send(chn))) |
| 618 | goto end; |
| 619 | |
Willy Tarreau | ef907fe | 2016-05-03 17:46:24 +0200 | [diff] [blame] | 620 | /* We need to check what remains of the reserve after o and to_forward |
| 621 | * have been transmitted, but they can overflow together and they can |
| 622 | * cause an integer underflow in the comparison since both are unsigned |
| 623 | * while maxrewrite is signed. |
| 624 | * The code below has been verified for being a valid check for this : |
| 625 | * - if (o + to_forward) overflow => return size [ large enough ] |
| 626 | * - if o + to_forward >= maxrw => return size [ large enough ] |
| 627 | * - otherwise return size - (maxrw - (o + to_forward)) |
Willy Tarreau | 999f643 | 2016-01-25 01:09:11 +0100 | [diff] [blame] | 628 | */ |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 629 | transit = co_data(chn) + chn->to_forward; |
Willy Tarreau | ef907fe | 2016-05-03 17:46:24 +0200 | [diff] [blame] | 630 | reserve -= transit; |
| 631 | if (transit < chn->to_forward || // addition overflow |
| 632 | transit >= (unsigned)global.tune.maxrewrite) // enough transit data |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 633 | return chn->buf.size; |
Willy Tarreau | 999f643 | 2016-01-25 01:09:11 +0100 | [diff] [blame] | 634 | end: |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 635 | return chn->buf.size - reserve; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 636 | } |
| 637 | |
Willy Tarreau | 2375233 | 2018-06-15 14:54:53 +0200 | [diff] [blame] | 638 | /* Returns non-zero if the channel's INPUT buffer's is considered full, which |
| 639 | * means that it holds at least as much INPUT data as (size - reserve). This |
| 640 | * also means that data that are scheduled for output are considered as potential |
| 641 | * free space, and that the reserved space is always considered as not usable. |
| 642 | * This information alone cannot be used as a general purpose free space indicator. |
| 643 | * However it accurately indicates that too many data were fed in the buffer |
| 644 | * for an analyzer for instance. See the channel_may_recv() function for a more |
| 645 | * generic function taking everything into account. |
| 646 | */ |
| 647 | static inline int channel_full(const struct channel *c, unsigned int reserve) |
| 648 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 649 | if (b_is_null(&c->buf)) |
Willy Tarreau | 2375233 | 2018-06-15 14:54:53 +0200 | [diff] [blame] | 650 | return 0; |
| 651 | |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 652 | return (ci_data(c) + reserve >= c_size(c)); |
Willy Tarreau | 2375233 | 2018-06-15 14:54:53 +0200 | [diff] [blame] | 653 | } |
| 654 | |
| 655 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 656 | /* Returns the amount of space available at the input of the buffer, taking the |
| 657 | * reserved space into account if ->to_forward indicates that an end of transfer |
| 658 | * is close to happen. The test is optimized to avoid as many operations as |
| 659 | * possible for the fast case. |
| 660 | */ |
Willy Tarreau | b5051f8 | 2015-01-14 20:25:34 +0100 | [diff] [blame] | 661 | static inline int channel_recv_max(const struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 662 | { |
Willy Tarreau | 27bb0e1 | 2015-01-14 15:56:50 +0100 | [diff] [blame] | 663 | int ret; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 664 | |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 665 | ret = channel_recv_limit(chn) - b_data(&chn->buf); |
Willy Tarreau | 27bb0e1 | 2015-01-14 15:56:50 +0100 | [diff] [blame] | 666 | if (ret < 0) |
| 667 | ret = 0; |
| 668 | return ret; |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 669 | } |
| 670 | |
Willy Tarreau | 3f67999 | 2018-06-15 15:06:42 +0200 | [diff] [blame] | 671 | /* Returns the amount of bytes that can be written over the input data at once, |
| 672 | * including reserved space which may be overwritten. This is used by Lua to |
| 673 | * insert data in the input side just before the other data using buffer_replace(). |
| 674 | * The goal is to transfer these new data in the output buffer. |
| 675 | */ |
| 676 | static inline int ci_space_for_replace(const struct channel *chn) |
| 677 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 678 | const struct buffer *buf = &chn->buf; |
Willy Tarreau | 3f67999 | 2018-06-15 15:06:42 +0200 | [diff] [blame] | 679 | const char *end; |
| 680 | |
| 681 | /* If the input side data overflows, we cannot insert data contiguously. */ |
| 682 | if (b_head(buf) + b_data(buf) >= b_wrap(buf)) |
| 683 | return 0; |
| 684 | |
| 685 | /* Check the last byte used in the buffer, it may be a byte of the output |
| 686 | * side if the buffer wraps, or its the end of the buffer. |
| 687 | */ |
| 688 | end = b_head(buf); |
| 689 | if (end <= ci_head(chn)) |
| 690 | end = b_wrap(buf); |
| 691 | |
| 692 | /* Compute the amount of bytes which can be written. */ |
| 693 | return end - ci_tail(chn); |
| 694 | } |
| 695 | |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 696 | /* Allocates a buffer for channel <chn>, but only if it's guaranteed that it's |
| 697 | * not the last available buffer or it's the response buffer. Unless the buffer |
| 698 | * is the response buffer, an extra control is made so that we always keep |
| 699 | * <tune.buffers.reserved> buffers available after this allocation. Returns 0 in |
| 700 | * case of failure, non-zero otherwise. |
| 701 | * |
| 702 | * If no buffer are available, the requester, represented by <wait> pointer, |
| 703 | * will be added in the list of objects waiting for an available buffer. |
| 704 | */ |
| 705 | static inline int channel_alloc_buffer(struct channel *chn, struct buffer_wait *wait) |
| 706 | { |
| 707 | int margin = 0; |
| 708 | |
| 709 | if (!(chn->flags & CF_ISRESP)) |
| 710 | margin = global.tune.reserved_bufs; |
| 711 | |
| 712 | if (b_alloc_margin(&chn->buf, margin) != NULL) |
| 713 | return 1; |
| 714 | |
Emeric Brun | a1dd243 | 2017-06-21 15:42:52 +0200 | [diff] [blame] | 715 | if (LIST_ISEMPTY(&wait->list)) { |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 716 | HA_SPIN_LOCK(BUF_WQ_LOCK, &buffer_wq_lock); |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 717 | LIST_ADDQ(&buffer_wq, &wait->list); |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 718 | HA_SPIN_UNLOCK(BUF_WQ_LOCK, &buffer_wq_lock); |
Emeric Brun | a1dd243 | 2017-06-21 15:42:52 +0200 | [diff] [blame] | 719 | } |
| 720 | |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 721 | return 0; |
| 722 | } |
| 723 | |
| 724 | /* Releases a possibly allocated buffer for channel <chn>. If it was not |
| 725 | * allocated, this function does nothing. Else the buffer is released and we try |
| 726 | * to wake up as many streams/applets as possible. */ |
| 727 | static inline void channel_release_buffer(struct channel *chn, struct buffer_wait *wait) |
| 728 | { |
Willy Tarreau | 0c7ed5d | 2018-07-10 09:53:31 +0200 | [diff] [blame] | 729 | if (c_size(chn) && c_empty(chn)) { |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 730 | b_free(&chn->buf); |
Olivier Houchard | 673867c | 2018-05-25 16:58:52 +0200 | [diff] [blame] | 731 | offer_buffers(wait->target, tasks_run_queue); |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 732 | } |
| 733 | } |
| 734 | |
Willy Tarreau | 319f745 | 2015-01-14 20:32:59 +0100 | [diff] [blame] | 735 | /* Truncate any unread data in the channel's buffer, and disable forwarding. |
| 736 | * Outgoing data are left intact. This is mainly to be used to send error |
| 737 | * messages after existing data. |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 738 | */ |
Willy Tarreau | 319f745 | 2015-01-14 20:32:59 +0100 | [diff] [blame] | 739 | static inline void channel_truncate(struct channel *chn) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 740 | { |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 741 | if (!co_data(chn)) |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 742 | return channel_erase(chn); |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 743 | |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 744 | chn->to_forward = 0; |
Willy Tarreau | 3ee8344 | 2018-06-15 16:42:02 +0200 | [diff] [blame] | 745 | if (!ci_data(chn)) |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 746 | return; |
| 747 | |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 748 | chn->buf.data = co_data(chn); |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 749 | } |
| 750 | |
Willy Tarreau | 4cf1300 | 2018-06-06 06:53:15 +0200 | [diff] [blame] | 751 | /* This function realigns a possibly wrapping channel buffer so that the input |
| 752 | * part is contiguous and starts at the beginning of the buffer and the output |
| 753 | * part ends at the end of the buffer. This provides the best conditions since |
| 754 | * it allows the largest inputs to be processed at once and ensures that once |
| 755 | * the output data leaves, the whole buffer is available at once. |
| 756 | */ |
Willy Tarreau | fd8d42f | 2018-07-12 10:57:15 +0200 | [diff] [blame] | 757 | static inline void channel_slow_realign(struct channel *chn, char *swap) |
Willy Tarreau | 4cf1300 | 2018-06-06 06:53:15 +0200 | [diff] [blame] | 758 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 759 | return b_slow_realign(&chn->buf, swap, co_data(chn)); |
Willy Tarreau | 4cf1300 | 2018-06-06 06:53:15 +0200 | [diff] [blame] | 760 | } |
| 761 | |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 762 | /* |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 763 | * Advance the channel buffer's read pointer by <len> bytes. This is useful |
| 764 | * when data have been read directly from the buffer. It is illegal to call |
| 765 | * this function with <len> causing a wrapping at the end of the buffer. It's |
| 766 | * the caller's responsibility to ensure that <len> is never larger than |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 767 | * chn->o. Channel flag WRITE_PARTIAL is set. |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 768 | */ |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 769 | static inline void co_skip(struct channel *chn, int len) |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 770 | { |
Willy Tarreau | c9fa048 | 2018-07-10 17:43:27 +0200 | [diff] [blame] | 771 | b_del(&chn->buf, len); |
Olivier Houchard | 08afac0 | 2018-06-22 19:26:39 +0200 | [diff] [blame] | 772 | chn->output -= len; |
Willy Tarreau | 0c7ed5d | 2018-07-10 09:53:31 +0200 | [diff] [blame] | 773 | c_realign_if_empty(chn); |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 774 | |
Willy Tarreau | fb0e920 | 2009-09-23 23:47:55 +0200 | [diff] [blame] | 775 | /* notify that some data was written to the SI from the buffer */ |
Christopher Faulet | c5a9d5b | 2017-11-09 09:36:43 +0100 | [diff] [blame] | 776 | chn->flags |= CF_WRITE_PARTIAL | CF_WRITE_EVENT; |
Willy Tarreau | 2b7addc | 2009-08-31 07:37:22 +0200 | [diff] [blame] | 777 | } |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 778 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 779 | /* 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] | 780 | * 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] | 781 | * closed, -2 is returned. If the block is too large for this buffer, -3 is |
| 782 | * returned. If there is not enough room left in the buffer, -1 is returned. |
| 783 | * 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] | 784 | * 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] | 785 | * chunk's length is updated with the number of bytes sent. |
Willy Tarreau | aeac319 | 2009-08-31 08:09:57 +0200 | [diff] [blame] | 786 | */ |
Willy Tarreau | 83061a8 | 2018-07-13 11:56:34 +0200 | [diff] [blame] | 787 | static inline int ci_putchk(struct channel *chn, struct buffer *chunk) |
Willy Tarreau | aeac319 | 2009-08-31 08:09:57 +0200 | [diff] [blame] | 788 | { |
| 789 | int ret; |
| 790 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 791 | ret = ci_putblk(chn, chunk->area, chunk->data); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 792 | if (ret > 0) |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 793 | chunk->data -= ret; |
Willy Tarreau | aeac319 | 2009-08-31 08:09:57 +0200 | [diff] [blame] | 794 | return ret; |
| 795 | } |
| 796 | |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 797 | /* 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] | 798 | * 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] | 799 | * input is closed, -2 is returned. If the block is too large for this buffer, |
| 800 | * -3 is returned. If there is not enough room left in the buffer, -1 is |
| 801 | * returned. Otherwise the number of bytes copied is returned (0 being a valid |
| 802 | * number). Channel flag READ_PARTIAL is updated if some data can be |
| 803 | * transferred. |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 804 | */ |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 805 | static inline int ci_putstr(struct channel *chn, const char *str) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 806 | { |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 807 | return ci_putblk(chn, str, strlen(str)); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 808 | } |
| 809 | |
| 810 | /* |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 811 | * Return one char from the channel's buffer. If the buffer is empty and the |
| 812 | * channel is closed, return -2. If the buffer is just empty, return -1. The |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 813 | * buffer's pointer is not advanced, it's up to the caller to call co_skip(buf, |
Willy Tarreau | 8263d2b | 2012-08-28 00:06:31 +0200 | [diff] [blame] | 814 | * 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] | 815 | * chn->o limit. |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 816 | */ |
Willy Tarreau | 06d80a9 | 2017-10-19 14:32:15 +0200 | [diff] [blame] | 817 | static inline int co_getchr(struct channel *chn) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 818 | { |
| 819 | /* closed or empty + imminent close = -2; empty = -1 */ |
Willy Tarreau | 974ced6 | 2012-10-12 23:11:02 +0200 | [diff] [blame] | 820 | if (unlikely((chn->flags & CF_SHUTW) || channel_is_empty(chn))) { |
| 821 | if (chn->flags & (CF_SHUTW|CF_SHUTW_NOW)) |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 822 | return -2; |
| 823 | return -1; |
| 824 | } |
Willy Tarreau | 50227f9 | 2018-06-15 15:18:17 +0200 | [diff] [blame] | 825 | return *co_head(chn); |
Willy Tarreau | 74b08c9 | 2010-09-08 17:04:31 +0200 | [diff] [blame] | 826 | } |
| 827 | |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 828 | |
Willy Tarreau | c7e4238 | 2012-08-24 19:22:53 +0200 | [diff] [blame] | 829 | #endif /* _PROTO_CHANNEL_H */ |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 830 | |
| 831 | /* |
| 832 | * Local variables: |
| 833 | * c-indent-level: 8 |
| 834 | * c-basic-offset: 8 |
| 835 | * End: |
| 836 | */ |