blob: dbf8dea3797498d7edbadd93b8539a06994f93e1 [file] [log] [blame]
Christopher Fauleta3d2a162018-10-22 08:59:39 +02001/*
2 * internal HTTP message
3 *
4 * Copyright 2018 HAProxy Technologies, Christopher Faulet <cfaulet@haproxy.com>
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
11 */
12
13#include <common/chunk.h>
Willy Tarreaub96b77e2018-12-11 10:22:41 +010014#include <common/htx.h>
Christopher Fauleta3d2a162018-10-22 08:59:39 +020015
16struct htx htx_empty = { .size = 0, .data = 0, .used = 0 };
17
18/* Defragments an HTTP message, removing unused blocks and unwrapping blocks and
19 * their contents. A temporary message is used to do so. This function never
20 * fails. if <blk> is not NULL, we replace it by the new block address, after
21 * the defragmentation. The new <blk> is returned.
22 */
23/* TODO: merge data blocks into one */
24struct htx_blk *htx_defrag(struct htx *htx, struct htx_blk *blk)
25{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010026 struct buffer *chunk = get_trash_chunk();
27 struct htx *tmp = htxbuf(chunk);
28 struct htx_blk *newblk, *oldblk;
Christopher Faulet200f8952019-01-02 11:23:44 +010029 uint32_t new, old, blkpos;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010030 uint32_t addr, blksz;
Christopher Faulet29f17582019-05-23 11:03:26 +020031 int32_t first = -1;
Christopher Fauleta3d2a162018-10-22 08:59:39 +020032
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010033 if (!htx->used)
34 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +020035
Christopher Faulet200f8952019-01-02 11:23:44 +010036 blkpos = -1;
37
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010038 new = 0;
39 addr = 0;
40 tmp->size = htx->size;
Christopher Fauleta3d2a162018-10-22 08:59:39 +020041
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010042 /* start from the head */
43 for (old = htx_get_head(htx); old != -1; old = htx_get_next(htx, old)) {
44 oldblk = htx_get_blk(htx, old);
Christopher Faulet28f29c72019-04-30 17:55:45 +020045 if (htx_get_blk_type(oldblk) == HTX_BLK_UNUSED)
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010046 continue;
Christopher Fauleta3d2a162018-10-22 08:59:39 +020047
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010048 newblk = htx_get_blk(tmp, new);
49 newblk->addr = addr;
50 newblk->info = oldblk->info;
51 blksz = htx_get_blksz(oldblk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +020052
Christopher Faulet9c66b982019-04-30 18:08:26 +020053 /* update the start-line position */
Christopher Faulet29f17582019-05-23 11:03:26 +020054 if (htx->first == old)
55 first = new;
Christopher Faulet174bfb12018-12-06 14:31:12 +010056
Christopher Faulet200f8952019-01-02 11:23:44 +010057 /* if <blk> is defined, set its new position */
58 if (blk != NULL && blk == oldblk)
59 blkpos = new;
60
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010061 memcpy((void *)tmp->blocks + addr, htx_get_blk_ptr(htx, oldblk), blksz);
62 new++;
63 addr += blksz;
Christopher Fauleta3d2a162018-10-22 08:59:39 +020064
Christopher Fauletb8fd4c02019-05-20 09:32:25 +020065 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +020066
Christopher Faulet28f29c72019-04-30 17:55:45 +020067 htx->used = new;
Christopher Faulet29f17582019-05-23 11:03:26 +020068 htx->first = first;
Christopher Faulet28f29c72019-04-30 17:55:45 +020069 htx->head = 0;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010070 htx->front = htx->tail = new - 1;
71 memcpy((void *)htx->blocks, (void *)tmp->blocks, htx->size);
Christopher Fauleta3d2a162018-10-22 08:59:39 +020072
Christopher Faulet200f8952019-01-02 11:23:44 +010073 return ((blkpos == -1) ? NULL : htx_get_blk(htx, blkpos));
Christopher Fauleta3d2a162018-10-22 08:59:39 +020074}
75
76/* Reserves a new block in the HTTP message <htx> with a content of <blksz>
77 * bytes. If there is not enough space, NULL is returned. Otherwise the reserved
78 * block is returned and the HTTP message is updated. Space for this new block
79 * is reserved in the HTTP message. But it is the caller responsibility to set
80 * right info in the block to reflect the stored data.
81 */
82static struct htx_blk *htx_reserve_nxblk(struct htx *htx, uint32_t blksz)
83{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010084 struct htx_blk *blk, *prevblk, *headblk, *frtblk;
85 uint32_t used;
86 uint32_t tail;
87 uint32_t prev;
88 uint32_t wrap;
89 uint32_t head;
90 int32_t headroom, tailroom;
Christopher Fauleta3d2a162018-10-22 08:59:39 +020091
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010092 if (blksz > htx_free_data_space(htx))
93 return NULL; /* full */
Christopher Fauleta3d2a162018-10-22 08:59:39 +020094
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010095 if (!htx->used) {
96 /* Empty message */
Christopher Faulet29f17582019-05-23 11:03:26 +020097 htx->front = htx->head = htx->tail = htx->first = 0;
Christopher Faulet28f29c72019-04-30 17:55:45 +020098 htx->used = 1;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +010099 blk = htx_get_blk(htx, htx->tail);
100 blk->addr = 0;
101 htx->data = blksz;
102 return blk;
103 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200104
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100105 used = htx->used + 1;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100106 prev = htx->tail;
Christopher Faulet28f29c72019-04-30 17:55:45 +0200107 head = htx->head;
108 tail = htx->tail + 1;
109 wrap = htx_get_wrap(htx);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200110
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100111 if (tail == wrap) {
112 frtblk = htx_get_blk(htx, htx->front);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200113
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100114 /* Blocks don't wrap for now. We either need to push the new one
115 * on top of others or to defragement the table. */
116 if (sizeof(htx->blocks[0]) * htx_pos_to_idx(htx, wrap+1) >= frtblk->addr + htx_get_blksz(frtblk))
117 wrap++;
118 else if (tail >= used) /* There is hole at the beginning */
119 tail = 0;
120 else {
121 /* No more room, tail hits data. We have to realign the
122 * whole message. */
123 goto defrag;
124 }
125 }
126 else if (used >= wrap) {
127 /* We have hit the tail, we need to reorganize the blocks. */
128 goto defrag;
129 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200130
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100131 /* Now we have updated tail, used and wrap, we know that there is some
132 * available room at least from the protocol's perspective. This space
133 * is split in two areas :
134 *
135 * 1: the space between the beginning of the blocks table and the
136 * front data's address. This space will only be used if data don't
137 * wrap yet.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200138
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100139 * 2: If the previous tail was the front block, the space between the
140 * beginning of the message and the head data's address.
141 * Otherwise, the space between the tail data's address and the
142 * tail's one.
143 */
144 prevblk = htx_get_blk(htx, prev);
145 headblk = htx_get_blk(htx, head);
146 if (prevblk->addr >= headblk->addr) {
147 /* the area was contiguous */
148 frtblk = htx_get_blk(htx, htx->front);
149 tailroom = sizeof(htx->blocks[0]) * htx_pos_to_idx(htx, wrap) - (frtblk->addr + htx_get_blksz(frtblk));
150 headroom = headblk->addr;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200151
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100152 if (tailroom >= (int32_t)blksz) {
153 /* install upfront and update ->front */
154 blk = htx_get_blk(htx, tail);
155 blk->addr = frtblk->addr + htx_get_blksz(frtblk);
156 htx->front = tail;
157 }
158 else if (headroom >= (int32_t)blksz) {
159 blk = htx_get_blk(htx, tail);
160 blk->addr = 0;
161 }
162 else {
163 /* need to defragment the table before inserting upfront */
164 goto defrag;
165 }
166 }
167 else {
168 /* it's already wrapped so we can't store anything in the tailroom */
169 headroom = headblk->addr - (prevblk->addr + htx_get_blksz(prevblk));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200170
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100171 if (headroom >= (int32_t)blksz) {
172 blk = htx_get_blk(htx, tail);
173 blk->addr = prevblk->addr + htx_get_blksz(prevblk);
174 }
175 else {
176 defrag:
177 /* need to defragment the table before inserting upfront */
178 htx_defrag(htx, NULL);
179 frtblk = htx_get_blk(htx, htx->front);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100180 tail = htx->tail + 1;
181 used = htx->used + 1;
182 blk = htx_get_blk(htx, tail);
183 blk->addr = frtblk->addr + htx_get_blksz(frtblk);
184 htx->front = tail;
185 }
186 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200187
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100188 htx->tail = tail;
189 htx->used = used;
190 htx->data += blksz;
Christopher Faulet29f17582019-05-23 11:03:26 +0200191 /* Set first position if not already set */
192 if (htx->first == -1)
193 htx->first = tail;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100194 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200195}
196
197/* Adds a new block of type <type> in the HTTP message <htx>. Its content size
198 * is passed but it is the caller responsibility to do the copy.
199 */
200struct htx_blk *htx_add_blk(struct htx *htx, enum htx_blk_type type, uint32_t blksz)
201{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100202 struct htx_blk *blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200203
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100204 blk = htx_reserve_nxblk(htx, blksz);
205 if (!blk)
206 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200207
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100208 blk->info = (type << 28);
209 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200210}
211
212/* Removes the block <blk> from the HTTP message <htx>. The function returns the
213 * block following <blk> or NULL if <blk> is the last block or the last
214 * inserted one.
215 */
216struct htx_blk *htx_remove_blk(struct htx *htx, struct htx_blk *blk)
217{
Christopher Faulet54483df2018-11-26 15:05:52 +0100218 enum htx_blk_type type = htx_get_blk_type(blk);
Christopher Faulet28f29c72019-04-30 17:55:45 +0200219 uint32_t next, pos, wrap;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200220
Christopher Faulet9c66b982019-04-30 18:08:26 +0200221 pos = htx_get_blk_pos(htx, blk);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100222 if (type != HTX_BLK_UNUSED) {
223 /* Mark the block as unused, decrement allocated size */
224 htx->data -= htx_get_blksz(blk);
225 blk->info = ((uint32_t)HTX_BLK_UNUSED << 28);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100226 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200227
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100228 /* This is the last block in use */
Christopher Faulet29f17582019-05-23 11:03:26 +0200229 if (htx->used == 1) {
230 htx_reset(htx);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100231 return NULL;
232 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200233
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100234 /* There is at least 2 blocks, so tail is always >= 0 */
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100235 blk = NULL;
236 next = pos + 1; /* By default retrun the next block */
Christopher Faulet28f29c72019-04-30 17:55:45 +0200237 wrap = htx_get_wrap(htx);
238 if (htx->tail + 1 == wrap) {
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100239 /* The HTTP message doesn't wrap */
Christopher Faulet28f29c72019-04-30 17:55:45 +0200240 if (pos == htx->head) {
241 /* move the head forward */
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100242 htx->used--;
Christopher Faulet28f29c72019-04-30 17:55:45 +0200243 htx->head++;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100244 }
245 else if (pos == htx->tail) {
246 /* remove the tail. this was the last inserted block so
247 * return NULL. */
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100248 htx->tail--;
249 htx->used--;
250 goto end;
251 }
252 }
253 else {
254 /* The HTTP message wraps */
255 if (pos == htx->tail) {
256 /* remove the tail. try to unwrap the message (pos == 0)
257 * and return NULL. */
Christopher Faulet28f29c72019-04-30 17:55:45 +0200258 htx->tail = ((pos == 0) ? wrap-1 : htx->tail-1);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100259 htx->used--;
260 goto end;
261 }
Christopher Faulet28f29c72019-04-30 17:55:45 +0200262 else if (pos == htx->head) {
263 /* move the head forward and try to unwrap the message
264 * (head+1 == wrap) */
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100265 htx->used--;
Christopher Faulet28f29c72019-04-30 17:55:45 +0200266 htx->head++;
267 if (htx->head == wrap)
268 htx->head = next = 0;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100269 }
270 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200271
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100272 blk = htx_get_blk(htx, next);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200273 end:
Christopher Faulet29f17582019-05-23 11:03:26 +0200274 if (pos == htx->first)
275 htx->first = (blk ? htx_get_blk_pos(htx, blk) : -1);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100276 if (pos == htx->front)
277 htx->front = htx_find_front(htx);
278 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200279}
280
Christopher Faulet00cf6972019-01-07 14:53:27 +0100281/* Truncate all blocks after the one containing the offset <offset>. This last
282 * one is truncated too.
283 */
284void htx_truncate(struct htx *htx, uint32_t offset)
285{
286 struct htx_blk *blk;
287 struct htx_ret htxret;
288
289 htxret = htx_find_blk(htx, offset);
290 blk = htxret.blk;
291 if (blk && htxret.ret) {
292 uint32_t sz = htx_get_blksz(blk);
293
294 htx_set_blk_value_len(blk, sz - htxret.ret);
295 blk = htx_get_next_blk(htx, blk);
296 }
297 while (blk)
298 blk = htx_remove_blk(htx, blk);
299}
300
Christopher Faulet549822f2019-02-25 10:23:19 +0100301/* Drain <count> bytes from the HTX message <htx>. DATA blocks will be cut if
302 * necessary. Others blocks will be removed at once if <count> is large
303 * enough. The function returns an htx_ret with the first block remaing in the
304 * messsage and the amount of data drained. If everything is removed,
305 * htx_ret.blk is set to NULL.
306 */
307struct htx_ret htx_drain(struct htx *htx, uint32_t count)
308{
309 struct htx_blk *blk;
310 struct htx_ret htxret = { .blk = NULL, .ret = 0 };
311
312 blk = htx_get_head_blk(htx);
313 while (count && blk) {
314 uint32_t sz = htx_get_blksz(blk);
315 enum htx_blk_type type = htx_get_blk_type(blk);
316
317 /* Ingore unused block */
318 if (type == HTX_BLK_UNUSED)
319 goto next;
320
321 if (sz > count) {
322 if (type == HTX_BLK_DATA) {
323 htx_cut_data_blk(htx, blk, count);
324 htxret.ret += count;
325 }
326 break;
327 }
328 count -= sz;
329 htxret.ret += sz;
330 next:
331 blk = htx_remove_blk(htx, blk);
332 }
333 htxret.blk = blk;
334
335 return htxret;
336}
337
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200338/* Tries to append data to the last inserted block, if the type matches and if
Christopher Faulet61775092019-05-07 21:42:27 +0200339 * there is enough non-wrapping space. Only DATA content can be appended. If the
340 * append fails, a new block is inserted. If an error occurred, NULL is
341 * returned. Otherwise, on success, the updated block (or the new one) is
342 * returned.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200343*/
344static struct htx_blk *htx_append_blk_value(struct htx *htx, enum htx_blk_type type,
345 const struct ist data)
346{
Christopher Fauletf1449b72019-04-10 14:54:46 +0200347 struct htx_blk *blk, *tailblk, *headblk, *frtblk;
348 struct ist v;
349 int32_t room;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200350
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100351 if (!htx->used)
352 goto add_new_block;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200353
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100354 /* Not enough space to store data */
355 if (data.len > htx_free_data_space(htx))
356 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200357
Christopher Faulet61775092019-05-07 21:42:27 +0200358 /* Append only DATA */
359 if (type != HTX_BLK_DATA)
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100360 goto add_new_block;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200361
Christopher Fauletf1449b72019-04-10 14:54:46 +0200362 /* get the tail and head block */
363 tailblk = htx_get_tail_blk(htx);
364 headblk = htx_get_head_blk(htx);
365 if (tailblk == NULL || headblk == NULL)
366 goto add_new_block;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200367
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100368 /* Don't try to append data if the last inserted block is not of the
369 * same type */
Christopher Fauletf1449b72019-04-10 14:54:46 +0200370 if (type != htx_get_blk_type(tailblk))
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100371 goto add_new_block;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200372
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100373 /*
374 * Same type and enough space: append data
375 */
Christopher Fauletf1449b72019-04-10 14:54:46 +0200376 frtblk = htx_get_blk(htx, htx->front);
377 if (tailblk->addr >= headblk->addr) {
378 if (htx->tail != htx->front)
379 goto add_new_block;
380 room = sizeof(htx->blocks[0]) * htx_pos_to_idx(htx, htx->tail) - (frtblk->addr + htx_get_blksz(frtblk));
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100381 }
Christopher Fauletf1449b72019-04-10 14:54:46 +0200382 else
383 room = headblk->addr - (tailblk->addr + htx_get_blksz(tailblk));
384
385 if (room < (int32_t)data.len)
386 tailblk = htx_defrag(htx, tailblk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200387
388 append_data:
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100389 /* get the value of the tail block */
390 /* FIXME: check v.len + data.len < 256MB */
Christopher Fauletf1449b72019-04-10 14:54:46 +0200391 v = htx_get_blk_value(htx, tailblk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200392
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100393 /* Append data and update the block itself */
394 memcpy(v.ptr + v.len, data.ptr, data.len);
Christopher Fauletf1449b72019-04-10 14:54:46 +0200395 htx_set_blk_value_len(tailblk, v.len + data.len);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200396
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100397 /* Update HTTP message */
398 htx->data += data.len;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200399
Christopher Fauletf1449b72019-04-10 14:54:46 +0200400 return tailblk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200401
402 add_new_block:
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100403 /* FIXME: check tlr.len (< 256MB) */
404 blk = htx_add_blk(htx, type, data.len);
405 if (!blk)
406 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200407
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100408 blk->info += data.len;
409 memcpy(htx_get_blk_ptr(htx, blk), data.ptr, data.len);
410 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200411}
412
413/* Replaces a value part of a block by a new one. The new part can be smaller or
414 * larger than the old one. This function works for any kind of block with
415 * attached data. It returns the new block on success, otherwise it returns
416 * NULL.
417 */
418struct htx_blk *htx_replace_blk_value(struct htx *htx, struct htx_blk *blk,
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100419 const struct ist old, const struct ist new)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200420{
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100421 struct buffer *tmp;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100422 struct ist n, v;
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100423 int32_t delta;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200424
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100425 n = htx_get_blk_name(htx, blk);
426 v = htx_get_blk_value(htx, blk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200427
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100428 delta = new.len - old.len;
429
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100430 /* easy case, new data are smaller, so replace it in-place */
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100431 if (delta <= 0) {
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100432 memcpy(old.ptr, new.ptr, new.len);
433 if (old.len != v.len)
434 memmove(old.ptr + new.len, old.ptr + old.len, (v.ptr + v.len) - (old.ptr + old.len));
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100435 htx_set_blk_value_len(blk, v.len + delta);
436 htx->data += delta;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100437 return blk;
438 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200439
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100440 /* we need to allocate more space to store the new header value */
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100441 if (delta > htx_free_space(htx))
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100442 return NULL; /* not enough space */
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200443
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100444 /*
445 * Copy the new header in a temp buffer
446 */
447 tmp = get_trash_chunk();
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200448
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100449 /* 1. copy the header name */
450 chunk_memcat(tmp, n.ptr, n.len);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200451
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100452 /* 2. copy value before old part, if any */
453 if (old.ptr != v.ptr)
454 chunk_memcat(tmp, v.ptr, old.ptr - v.ptr);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200455
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100456 /* 3. copy new value */
457 chunk_memcat(tmp, new.ptr, new.len);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200458
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100459 /* 4. copy value after old part if any */
460 if (old.len != v.len)
461 chunk_memcat(tmp, old.ptr + old.len, (v.ptr + v.len) - (old.ptr + old.len));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200462
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200463
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100464 /* Expand the block size. But data are not copied yet. Then defragment
465 * the HTX message.
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100466 */
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100467 htx_set_blk_value_len(blk, v.len + delta);
468 htx->data += delta;
469 blk = htx_defrag(htx, blk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200470
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100471 /* Finaly, copy data. */
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100472 memcpy(htx_get_blk_ptr(htx, blk), tmp->area, tmp->data);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200473
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100474 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200475}
476
477/* Transfer HTX blocks from <src> to <dst>, stopping on the first block of the
Christopher Faulet156852b2019-05-16 11:29:13 +0200478 * type <mark> (typically EOH, EOD or EOM) or when <count> bytes were moved
479 * (including payload and meta-data). It returns the number of bytes moved and
480 * the last HTX block inserted in <dst>.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200481 */
482struct htx_ret htx_xfer_blks(struct htx *dst, struct htx *src, uint32_t count,
483 enum htx_blk_type mark)
484{
485 struct htx_blk *blk, *dstblk;
486 enum htx_blk_type type;
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100487 uint32_t info, max, sz, ret;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200488
Christopher Faulet156852b2019-05-16 11:29:13 +0200489 ret = htx_used_space(dst);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200490 blk = htx_get_blk(src, htx_get_head(src));
491 dstblk = NULL;
Christopher Faulet156852b2019-05-16 11:29:13 +0200492
493 while (blk && count) {
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200494 type = htx_get_blk_type(blk);
495
496 /* Ingore unused block */
497 if (type == HTX_BLK_UNUSED)
498 goto next;
499
Christopher Fauleta61e97b2019-05-16 11:30:31 +0200500 /* Be sure to have enough space to xfer all headers in one
501 * time. If not while <dst> is empty, we report a parsing error
502 * on <src>.
503 */
504 if (mark >= HTX_BLK_EOH && (type == HTX_BLK_REQ_SL || type == HTX_BLK_RES_SL)) {
505 struct htx_sl *sl = htx_get_blk_ptr(src, blk);
506
507 if (sl->hdrs_bytes != -1 && sl->hdrs_bytes > count) {
508 if (htx_is_empty(dst))
509 src->flags |= HTX_FL_PARSING_ERROR;
510 break;
511 }
512 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200513
Christopher Faulet156852b2019-05-16 11:29:13 +0200514 sz = htx_get_blksz(blk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200515 info = blk->info;
Christopher Faulet156852b2019-05-16 11:29:13 +0200516 max = htx_get_max_blksz(dst, count);
517 if (!max)
518 break;
Willy Tarreau90caa072019-04-09 16:21:54 +0200519 if (sz > max) {
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200520 /* Headers and pseudo headers must be fully copied */
Christopher Faulet156852b2019-05-16 11:29:13 +0200521 if (type != HTX_BLK_DATA)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200522 break;
Christopher Faulet156852b2019-05-16 11:29:13 +0200523 sz = max;
524 info = (type << 28) + sz;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200525 }
526
527 dstblk = htx_reserve_nxblk(dst, sz);
528 if (!dstblk)
529 break;
530 dstblk->info = info;
531 memcpy(htx_get_blk_ptr(dst, dstblk), htx_get_blk_ptr(src, blk), sz);
532
Christopher Faulet156852b2019-05-16 11:29:13 +0200533 count -= sizeof(dstblk) + sz;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200534 if (blk->info != info) {
535 /* Partial move: don't remove <blk> from <src> but
536 * resize its content */
Christopher Faulet156852b2019-05-16 11:29:13 +0200537 htx_cut_data_blk(src, blk, sz);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200538 break;
539 }
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200540 next:
541 blk = htx_remove_blk(src, blk);
542 if (type == mark)
543 break;
544
545 }
546
Christopher Faulet156852b2019-05-16 11:29:13 +0200547 end:
548 ret = htx_used_space(dst) - ret;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200549 return (struct htx_ret){.ret = ret, .blk = dstblk};
550}
551
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200552/* Replaces an header by a new one. The new header can be smaller or larger than
553 * the old one. It returns the new block on success, otherwise it returns NULL.
Willy Tarreaued00e342018-12-07 08:47:45 +0100554 * The header name is always lower cased.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200555 */
556struct htx_blk *htx_replace_header(struct htx *htx, struct htx_blk *blk,
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100557 const struct ist name, const struct ist value)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200558{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100559 enum htx_blk_type type;
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100560 int32_t delta;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200561
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100562 type = htx_get_blk_type(blk);
563 if (type != HTX_BLK_HDR)
564 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200565
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100566 delta = name.len + value.len - htx_get_blksz(blk);
567
568 /* easy case, new value is smaller, so replace it in-place */
569 if (delta <= 0) {
570 blk->info = (type << 28) + (value.len << 8) + name.len;
571 htx->data += delta;
572 goto copy;
573 }
574
575 /* we need to allocate more space to store the new value */
576 if (delta > htx_free_space(htx))
577 return NULL; /* not enough space */
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200578
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100579 /* Expand the block size. But data are not copied yet. Then defragment
580 * the HTX message.
581 */
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200582 blk->info = (type << 28) + (value.len << 8) + name.len;
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100583 htx->data += delta;
584 blk = htx_defrag(htx, blk);
585
586 copy:
587 /* Finaly, copy data. */
Willy Tarreaued00e342018-12-07 08:47:45 +0100588 ist2bin_lc(htx_get_blk_ptr(htx, blk), name);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100589 memcpy(htx_get_blk_ptr(htx, blk) + name.len, value.ptr, value.len);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100590 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200591}
592
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100593/* Replaces the parts of the start-line. It returns the new start-line on
594 * success, otherwise it returns NULL. It is the caller responsibility to update
595 * sl->info, if necessary.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200596 */
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100597struct htx_sl *htx_replace_stline(struct htx *htx, struct htx_blk *blk, const struct ist p1,
598 const struct ist p2, const struct ist p3)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200599{
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100600 struct htx_sl *sl;
601 struct htx_sl tmp; /* used to save sl->info and sl->flags */
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100602 enum htx_blk_type type;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200603 uint32_t size;
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100604 int32_t delta;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200605
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100606 type = htx_get_blk_type(blk);
Willy Tarreauc706cd72018-12-07 17:12:22 +0100607 if (type != HTX_BLK_REQ_SL && type != HTX_BLK_RES_SL)
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100608 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200609
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100610 /* Save start-line info and flags */
611 sl = htx_get_blk_ptr(htx, blk);
612 tmp.info = sl->info;
613 tmp.flags = sl->flags;
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100614
615 size = sizeof(*sl) + p1.len + p2.len + p3.len;
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100616 delta = size - htx_get_blksz(blk);
617
618 /* easy case, new data are smaller, so replace it in-place */
619 if (delta <= 0) {
620 htx_set_blk_value_len(blk, size);
621 htx->data += delta;
622 goto copy;
623 }
624
625 /* we need to allocate more space to store the new value */
626 if (delta > htx_free_space(htx))
627 return NULL; /* not enough space */
628
629 /* Expand the block size. But data are not copied yet. Then defragment
630 * the HTX message.
631 */
632 htx_set_blk_value_len(blk, size);
633 htx->data += delta;
634 blk = htx_defrag(htx, blk);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200635
Christopher Faulete97f3ba2018-12-10 15:39:40 +0100636 copy:
637 /* Restore start-line info and flags and copy parts of the start-line */
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100638 sl = htx_get_blk_ptr(htx, blk);
639 sl->info = tmp.info;
640 sl->flags = tmp.flags;
Christopher Faulet05c083c2019-05-15 14:56:47 +0200641 if (sl->hdrs_bytes != -1)
642 sl->hdrs_bytes += delta;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200643
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100644 HTX_SL_P1_LEN(sl) = p1.len;
645 HTX_SL_P2_LEN(sl) = p2.len;
646 HTX_SL_P3_LEN(sl) = p3.len;
Christopher Faulet54483df2018-11-26 15:05:52 +0100647
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100648 memcpy(HTX_SL_P1_PTR(sl), p1.ptr, p1.len);
649 memcpy(HTX_SL_P2_PTR(sl), p2.ptr, p2.len);
650 memcpy(HTX_SL_P3_PTR(sl), p3.ptr, p3.len);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200651
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100652 return sl;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200653}
654
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100655/* Add a new start-line. It returns it on success, otherwise it returns NULL. It
656 * is the caller responsibility to set sl->info, if necessary.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200657 */
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100658struct htx_sl *htx_add_stline(struct htx *htx, enum htx_blk_type type, unsigned int flags,
659 const struct ist p1, const struct ist p2, const struct ist p3)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200660{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100661 struct htx_blk *blk;
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100662 struct htx_sl *sl;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200663 uint32_t size;
664
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100665 if (type != HTX_BLK_REQ_SL && type != HTX_BLK_RES_SL)
666 return NULL;
667
668 size = sizeof(*sl) + p1.len + p2.len + p3.len;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200669
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100670 /* FIXME: check size (< 256MB) */
671 blk = htx_add_blk(htx, type, size);
672 if (!blk)
673 return NULL;
674 blk->info += size;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200675
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100676 sl = htx_get_blk_ptr(htx, blk);
Christopher Faulet05c083c2019-05-15 14:56:47 +0200677 sl->hdrs_bytes = -1;
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100678 sl->flags = flags;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200679
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100680 HTX_SL_P1_LEN(sl) = p1.len;
681 HTX_SL_P2_LEN(sl) = p2.len;
682 HTX_SL_P3_LEN(sl) = p3.len;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200683
Christopher Fauletf1ba18d2018-11-26 21:37:08 +0100684 memcpy(HTX_SL_P1_PTR(sl), p1.ptr, p1.len);
685 memcpy(HTX_SL_P2_PTR(sl), p2.ptr, p2.len);
686 memcpy(HTX_SL_P3_PTR(sl), p3.ptr, p3.len);
687
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100688 return sl;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200689}
690
691/* Adds an HTX block of type HDR in <htx>. It returns the new block on
Willy Tarreaued00e342018-12-07 08:47:45 +0100692 * success. Otherwise, it returns NULL. The header name is always lower cased.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200693 */
694struct htx_blk *htx_add_header(struct htx *htx, const struct ist name,
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100695 const struct ist value)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200696{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100697 struct htx_blk *blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200698
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100699 /* FIXME: check name.len (< 256B) and value.len (< 1MB) */
700 blk = htx_add_blk(htx, HTX_BLK_HDR, name.len + value.len);
701 if (!blk)
702 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200703
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100704 blk->info += (value.len << 8) + name.len;
Willy Tarreaued00e342018-12-07 08:47:45 +0100705 ist2bin_lc(htx_get_blk_ptr(htx, blk), name);
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100706 memcpy(htx_get_blk_ptr(htx, blk) + name.len, value.ptr, value.len);
707 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200708}
709
Willy Tarreau52610e92019-01-03 18:26:17 +0100710/* Adds an HTX block of type <type> in <htx>, of size <blksz>. It returns the
711 * new block on success. Otherwise, it returns NULL. The caller is responsible
712 * for filling the block itself.
713 */
714struct htx_blk *htx_add_blk_type_size(struct htx *htx, enum htx_blk_type type, uint32_t blksz)
715{
716 struct htx_blk *blk;
717
718 blk = htx_add_blk(htx, type, blksz);
719 if (!blk)
720 return NULL;
721
722 blk->info += blksz;
723 return blk;
724}
725
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200726struct htx_blk *htx_add_all_headers(struct htx *htx, const struct http_hdr *hdrs)
727{
728 int i;
729
730 for (i = 0; hdrs[i].n.len; i++) {
731 if (!htx_add_header(htx, hdrs[i].n, hdrs[i].v))
732 return NULL;
733 }
734 return htx_add_endof(htx, HTX_BLK_EOH);
735}
736/* Adds an HTX block of type PHDR in <htx>. It returns the new block on
737 * success. Otherwise, it returns NULL.
738 */
739struct htx_blk *htx_add_pseudo_header(struct htx *htx, enum htx_phdr_type phdr,
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100740 const struct ist value)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200741{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100742 struct htx_blk *blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200743
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100744 /* FIXME: check value.len ( < 1MB) */
745 blk = htx_add_blk(htx, HTX_BLK_PHDR, value.len);
746 if (!blk)
747 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200748
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100749 blk->info += (value.len << 8) + phdr;
750 memcpy(htx_get_blk_ptr(htx, blk), value.ptr, value.len);
751 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200752}
753
754/* Adds an HTX block of type EOH,EOD or EOM in <htx>. It returns the new block
755 * on success. Otherwise, it returns NULL.
756 */
757struct htx_blk *htx_add_endof(struct htx *htx, enum htx_blk_type type)
758{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100759 struct htx_blk *blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200760
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100761 blk = htx_add_blk(htx, type, 1);
762 if (!blk)
763 return NULL;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200764
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100765 blk->info += 1;
766 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200767}
768
769
770/* Adds an HTX block of type DATA in <htx>. It first tries to append data if
771 * possible. It returns the new block on success. Otherwise, it returns NULL.
772 */
773struct htx_blk *htx_add_data(struct htx *htx, const struct ist data)
774{
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100775 return htx_append_blk_value(htx, HTX_BLK_DATA, data);
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200776}
777
Christopher Faulet61775092019-05-07 21:42:27 +0200778/* Adds an HTX block of type TLR in <htx>. It returns the new block on
779 * success. Otherwise, it returns NULL.
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200780 */
781struct htx_blk *htx_add_trailer(struct htx *htx, const struct ist tlr)
782{
Christopher Faulet61775092019-05-07 21:42:27 +0200783 struct htx_blk *blk;
784
785 /* FIXME: check tlr.len (< 256MB) */
786 blk = htx_add_blk(htx, HTX_BLK_TLR, tlr.len);
787 if (!blk)
788 return NULL;
789
790 blk->info += tlr.len;
791 memcpy(htx_get_blk_ptr(htx, blk), tlr.ptr, tlr.len);
792 return blk;
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200793}
794
Christopher Faulet24ed8352018-11-22 11:20:43 +0100795struct htx_blk *htx_add_data_before(struct htx *htx, const struct htx_blk *ref,
796 const struct ist data)
797{
798 struct htx_blk *blk;
799 int32_t prev;
800
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100801 /* FIXME: check data.len (< 256MB) */
802 blk = htx_add_blk(htx, HTX_BLK_DATA, data.len);
803 if (!blk)
804 return NULL;
Christopher Faulet24ed8352018-11-22 11:20:43 +0100805
Christopher Fauletaa75b3d2018-12-05 16:20:40 +0100806 blk->info += data.len;
807 memcpy(htx_get_blk_ptr(htx, blk), data.ptr, data.len);
Christopher Faulet24ed8352018-11-22 11:20:43 +0100808
809 for (prev = htx_get_prev(htx, htx->tail); prev != -1; prev = htx_get_prev(htx, prev)) {
810 struct htx_blk *pblk = htx_get_blk(htx, prev);
811
812 /* Swap .addr and .info fields */
813 blk->addr ^= pblk->addr; pblk->addr ^= blk->addr; blk->addr ^= pblk->addr;
814 blk->info ^= pblk->info; pblk->info ^= blk->info; blk->info ^= pblk->info;
815
816 if (blk->addr == pblk->addr)
817 blk->addr += htx_get_blksz(pblk);
818 htx->front = prev;
819
820 if (pblk == ref)
821 break;
822 blk = pblk;
823 }
Christopher Faulet05aab642019-04-11 13:43:57 +0200824
825 if (htx_get_blk_pos(htx, blk) != htx->front)
826 htx_defrag(htx, NULL);
827
Christopher Faulet24ed8352018-11-22 11:20:43 +0100828 return blk;
829}
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200830
Christopher Fauletc59ff232018-12-03 13:58:44 +0100831/* Appends the H1 representation of the request line block <blk> to the
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200832 * chunk <chk>. It returns 1 if data are successfully appended, otherwise it
833 * returns 0.
834 */
Christopher Fauletc59ff232018-12-03 13:58:44 +0100835int htx_reqline_to_h1(const struct htx_sl *sl, struct buffer *chk)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200836{
Christopher Faulet570d1612018-11-26 11:13:57 +0100837 if (HTX_SL_LEN(sl) + 4 > b_room(chk))
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200838 return 0;
839
Christopher Faulet570d1612018-11-26 11:13:57 +0100840 chunk_memcat(chk, HTX_SL_REQ_MPTR(sl), HTX_SL_REQ_MLEN(sl));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200841 chunk_memcat(chk, " ", 1);
Christopher Faulet570d1612018-11-26 11:13:57 +0100842 chunk_memcat(chk, HTX_SL_REQ_UPTR(sl), HTX_SL_REQ_ULEN(sl));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200843 chunk_memcat(chk, " ", 1);
Christopher Faulet1e7af462018-12-03 14:05:01 +0100844 if (sl->flags & HTX_SL_F_VER_11)
845 chunk_memcat(chk, "HTTP/1.1", 8);
846 else
847 chunk_memcat(chk, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl));
848
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200849 chunk_memcat(chk, "\r\n", 2);
850
851 return 1;
852}
853
Christopher Fauletc59ff232018-12-03 13:58:44 +0100854/* Appends the H1 representation of the status line block <blk> to the chunk
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200855 * <chk>. It returns 1 if data are successfully appended, otherwise it
856 * returns 0.
857 */
Christopher Fauletc59ff232018-12-03 13:58:44 +0100858int htx_stline_to_h1(const struct htx_sl *sl, struct buffer *chk)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200859{
Christopher Faulet570d1612018-11-26 11:13:57 +0100860 if (HTX_SL_LEN(sl) + 4 > b_size(chk))
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200861 return 0;
862
Christopher Faulet1e7af462018-12-03 14:05:01 +0100863 if (sl->flags & HTX_SL_F_VER_11)
864 chunk_memcat(chk, "HTTP/1.1", 8);
865 else
866 chunk_memcat(chk, HTX_SL_RES_VPTR(sl), HTX_SL_RES_VLEN(sl));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200867 chunk_memcat(chk, " ", 1);
Christopher Faulet570d1612018-11-26 11:13:57 +0100868 chunk_memcat(chk, HTX_SL_RES_CPTR(sl), HTX_SL_RES_CLEN(sl));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200869 chunk_memcat(chk, " ", 1);
Christopher Faulet570d1612018-11-26 11:13:57 +0100870 chunk_memcat(chk, HTX_SL_RES_RPTR(sl), HTX_SL_RES_RLEN(sl));
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200871 chunk_memcat(chk, "\r\n", 2);
872
873 return 1;
874}
875
Christopher Fauletc59ff232018-12-03 13:58:44 +0100876/* Appends the H1 representation of the header block <blk> to the chunk
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200877 * <chk>. It returns 1 if data are successfully appended, otherwise it returns
878 * 0.
879 */
Christopher Fauletc59ff232018-12-03 13:58:44 +0100880int htx_hdr_to_h1(const struct ist n, const struct ist v, struct buffer *chk)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200881{
882 if (n.len + v.len + 4 > b_room(chk))
883 return 0;
884
885 chunk_memcat(chk, n.ptr, n.len);
886 chunk_memcat(chk, ": ", 2);
887 chunk_memcat(chk, v.ptr, v.len);
888 chunk_memcat(chk, "\r\n", 2);
889
890 return 1;
891}
892
Christopher Fauletc59ff232018-12-03 13:58:44 +0100893/* Appends the H1 representation of the data block <blk> to the chunk
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200894 * <chk>. If <chunked> is non-zero, it emits HTTP/1 chunk-encoded data. It
895 * returns 1 if data are successfully appended, otherwise it returns 0.
896 */
Christopher Fauletc59ff232018-12-03 13:58:44 +0100897int htx_data_to_h1(const struct ist data, struct buffer *chk, int chunked)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200898{
899 if (chunked) {
900 uint32_t chksz;
901 char tmp[10];
902 char *beg, *end;
903
904 chksz = data.len;
905
906 beg = end = tmp+10;
907 *--beg = '\n';
908 *--beg = '\r';
909 do {
910 *--beg = hextab[chksz & 0xF];
911 } while (chksz >>= 4);
912
913 if (data.len + (end - beg) + 2 > b_room(chk))
914 return 0;
915 chunk_memcat(chk, beg, end - beg);
916 chunk_memcat(chk, data.ptr, data.len);
917 chunk_memcat(chk, "\r\n", 2);
918 }
919 else {
920 if (!chunk_memcat(chk, data.ptr, data.len))
921 return 0;
922 }
923
924 return 1;
925}
926
Christopher Fauletc59ff232018-12-03 13:58:44 +0100927/* Appends the h1 representation of the trailer block <blk> to the chunk
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200928 * <chk>. It returns 1 if data are successfully appended, otherwise it returns
929 * 0.
930 */
Christopher Fauletc59ff232018-12-03 13:58:44 +0100931int htx_trailer_to_h1(const struct ist tlr, struct buffer *chk)
Christopher Fauleta3d2a162018-10-22 08:59:39 +0200932{
933 /* FIXME: be sure the CRLF is here or remove it when inserted */
934 if (!chunk_memcat(chk, tlr.ptr, tlr.len))
935 return 0;
936 return 1;
937}