MEDIUM: htx: Insert headers and trailers before corresponding end-of block

htx_add_header() and htx_add_trailer() were updated to automatically insert
the block at the right place, just before the corresponding end-of block.
Indeed, it is invalid, unexepected and probably unsupported to insert a
header after the EOH block without moving it. So it is easier to handle it
in HTX function directly.

It means that http_add_header() was simplified.

diff --git a/include/haproxy/htx.h b/include/haproxy/htx.h
index fac0dc72d984e466e632b1ca546116fe47368ab5..8e237c03238eba085b6e1b3763ee189d9402878e 100644 (file)
--- a/include/haproxy/htx.h
+++ b/include/haproxy/htx.h
@@ -519,7 +519,7 @@ static inline struct htx_sl *htx_add_stline(struct htx *htx, enum htx_blk_type t
 static inline struct htx_blk *htx_add_header(struct htx *htx, const struct ist name,
                                             const struct ist value)
 {
-       struct htx_blk *blk, *tailblk;
+       struct htx_blk *blk, *tailblk, *prevblk;
 
        if (name.len > 255 || value.len > 1048575)
                return NULL;
@@ -532,8 +532,20 @@ static inline struct htx_blk *htx_add_header(struct htx *htx, const struct ist n
        blk->info += (value.len << 8) + name.len;
        ist2bin_lc(htx_get_blk_ptr(htx, blk), name);
        memcpy(htx_get_blk_ptr(htx, blk)  + name.len, value.ptr, value.len);
-       if (tailblk && htx_get_blk_type(tailblk) >= HTX_BLK_EOH)
-               htx->flags |= HTX_FL_UNORDERED;
+       if (!tailblk)
+               goto end;
+
+       for (prevblk = htx_get_prev_blk(htx, blk);
+            prevblk && htx_get_blk_type(prevblk) > HTX_BLK_HDR;
+            prevblk = htx_get_prev_blk(htx, blk)) {
+
+               /* Swap .addr and .info fields */
+               blk->addr ^= prevblk->addr; prevblk->addr ^= blk->addr; blk->addr ^= prevblk->addr;
+               blk->info ^= prevblk->info; prevblk->info ^= blk->info; blk->info ^= prevblk->info;
+               blk = prevblk;
+       }
+
+  end:
        return blk;
 }
 
@@ -543,7 +555,7 @@ static inline struct htx_blk *htx_add_header(struct htx *htx, const struct ist n
 static inline struct htx_blk *htx_add_trailer(struct htx *htx, const struct ist name,
                                              const struct ist value)
 {
-       struct htx_blk *blk, *tailblk;
+       struct htx_blk *blk, *tailblk, *prevblk;
 
        if (name.len > 255 || value.len > 1048575)
                return NULL;
@@ -556,8 +568,20 @@ static inline struct htx_blk *htx_add_trailer(struct htx *htx, const struct ist
        blk->info += (value.len << 8) + name.len;
        ist2bin_lc(htx_get_blk_ptr(htx, blk), name);
        memcpy(htx_get_blk_ptr(htx, blk)  + name.len, value.ptr, value.len);
-       if (tailblk && htx_get_blk_type(tailblk) >= HTX_BLK_EOT)
-               htx->flags |= HTX_FL_UNORDERED;
+       if (!tailblk)
+               goto end;
+
+       for (prevblk = htx_get_prev_blk(htx, blk);
+            prevblk && htx_get_blk_type(prevblk) > HTX_BLK_TLR;
+            prevblk = htx_get_prev_blk(htx, blk)) {
+
+               /* Swap .addr and .info fields */
+               blk->addr ^= prevblk->addr; prevblk->addr ^= blk->addr; blk->addr ^= prevblk->addr;
+               blk->info ^= prevblk->info; prevblk->info ^= blk->info; blk->info ^= prevblk->info;
+               blk = prevblk;
+       }
+
+  end:
        return blk;
 }
 

diff --git a/src/http_htx.c b/src/http_htx.c
index 3d77e3aa817df73901b11c97a97a102054890ba6..a0993f654965f31f65258f899ce12c204e983d11 100644 (file)
--- a/src/http_htx.c
+++ b/src/http_htx.c
@@ -285,41 +285,12 @@ int http_match_header(const struct htx *htx, const struct my_regex *re, struct h
  */
 int http_add_header(struct htx *htx, const struct ist n, const struct ist v, int update_authority)
 {
-       struct htx_blk *blk;
-       struct htx_sl *sl;
-       enum htx_blk_type type = htx_get_tail_type(htx);
-       int32_t prev;
-
-       blk = htx_add_header(htx, n, v);
-       if (!blk)
+       if (!htx_add_header(htx, n, v))
                goto fail;
 
-       if (unlikely(type < HTX_BLK_EOH))
-               goto end;
-
-       /* <blk> is the head, swap it iteratively with its predecessor to place
-        * it just before the end-of-header block. So blocks remains ordered. */
-       for (prev = htx_get_prev(htx, htx->tail); prev != htx->first; prev = htx_get_prev(htx, prev)) {
-               struct htx_blk   *pblk = htx_get_blk(htx, prev);
-               enum htx_blk_type ptype = htx_get_blk_type(pblk);
-
-               /* Swap .addr and .info fields */
-               blk->addr ^= pblk->addr; pblk->addr ^= blk->addr; blk->addr ^= pblk->addr;
-               blk->info ^= pblk->info; pblk->info ^= blk->info; blk->info ^= pblk->info;
-
-               if (blk->addr == pblk->addr)
-                       blk->addr += htx_get_blksz(pblk);
-
-               /* Stop when end-of-header is reached */
-               if (ptype == HTX_BLK_EOH)
-                       break;
-
-               blk = pblk;
-       }
-
-  end:
        if (update_authority) {
-               sl = http_get_stline(htx);
+               struct htx_sl *sl = http_get_stline(htx);
+
                if (sl && (sl->flags & HTX_SL_F_HAS_AUTHORITY) && isteqi(n, ist("host"))) {
                        if (!http_update_authority(htx, sl, v))
                                goto fail;