MINOR: buffer: split bi_contig_data() into ci_contig_data and b_config_data()
authorWilly Tarreau <w@1wt.eu>
Wed, 6 Jun 2018 14:55:45 +0000 (16:55 +0200)
committerWilly Tarreau <w@1wt.eu>
Thu, 19 Jul 2018 14:23:40 +0000 (16:23 +0200)
This function was sometimes used from a channel and sometimes from a buffer.
In both cases it requires knowledge of the size of the output data (to skip
them). Here the split ensures the channel can deal with this point, and that
other places not having output data can continue to work.

include/common/buf.h
include/common/buffer.h
include/proto/channel.h
src/cache.c
src/flt_http_comp.c
src/flt_trace.c
src/mux_h2.c

index fccad45..0ee9c54 100644 (file)
@@ -278,6 +278,21 @@ static inline int b_space_wraps(const struct buffer *b)
        return 1;
 }
 
+/* b_contig_data() : returns the amount of data that can contiguously be read
+ * at once starting from a relative offset <start> (which allows to easily
+ * pre-compute blocks for memcpy). The start point will typically contain the
+ * amount of past data already returned by a previous call to this function.
+ */
+static inline size_t b_contig_data(const struct buffer *b, size_t start)
+{
+       size_t data = b_wrap(b) - b_peek(b, start);
+       size_t limit = b_data(b) - start;
+
+       if (data > limit)
+               data = limit;
+       return data;
+}
+
 
 /*********************************************/
 /* Functions used to modify the buffer state */
index b34c389..da65e89 100644 (file)
@@ -123,16 +123,6 @@ static inline char *bi_end(const struct buffer *b)
        return ret;
 }
 
-/* Returns the amount of input data that can contiguously be read at once */
-static inline int bi_contig_data(const struct buffer *b)
-{
-       int data = b->data + b->size - b->p;
-
-       if (data > b->i)
-               data = b->i;
-       return data;
-}
-
 /* Returns the start of the output data in a buffer */
 static inline char *bo_ptr(const struct buffer *b)
 {
index 9429689..9568050 100644 (file)
@@ -309,6 +309,12 @@ static inline const char *ci_stop(const struct channel *c)
 }
 
 
+/* Returns the amount of input data that can contiguously be read at once */
+static inline size_t ci_contig_data(const struct channel *c)
+{
+       return b_contig_data(c->buf, co_data(c));
+}
+
 /* Initialize all fields in the channel. */
 static inline void channel_init(struct channel *chn)
 {
index aec9ba6..c1e6bcc 100644 (file)
@@ -225,7 +225,7 @@ cache_store_http_forward_data(struct stream *s, struct filter *filter,
                                ret = shctx_row_data_append(shctx,
                                                            st->first_block,
                                                            (unsigned char *)bi_ptr(msg->chn->buf),
-                                                           MIN(bi_contig_data(msg->chn->buf), len - st->hdrs_len));
+                                                           MIN(ci_contig_data(msg->chn), len - st->hdrs_len));
                                /* Rewind the buffer to forward all data */
                                c_rew(msg->chn, st->hdrs_len);
                                st->hdrs_len = 0;
index df59d7d..f26b28b 100644 (file)
@@ -206,7 +206,7 @@ comp_http_data(struct stream *s, struct filter *filter, struct http_msg *msg)
                len = MIN(tmpbuf->size - buffer_len(tmpbuf), len);
 
                c_adv(chn, *nxt);
-               block = bi_contig_data(buf);
+               block = ci_contig_data(chn);
                memcpy(bi_end(tmpbuf), bi_ptr(buf), block);
                if (len > block)
                        memcpy(bi_end(tmpbuf)+block, buf->data, len-block);
@@ -644,8 +644,8 @@ http_compression_buffer_add_data(struct comp_state *st, struct buffer *in,
        data_process_len = MIN(out->size - buffer_len(out), sz);
 
        block1 = data_process_len;
-       if (block1 > bi_contig_data(in))
-               block1 = bi_contig_data(in);
+       if (block1 > b_contig_data(in, in->o))
+               block1 = b_contig_data(in, in->o);
        block2 = data_process_len - block1;
 
        /* compressors return < 0 upon error or the amount of bytes read */
@@ -765,7 +765,7 @@ http_compression_buffer_end(struct comp_state *st, struct stream *s,
        /* copy the remaining data in the tmp buffer. */
        c_adv(chn, st->consumed);
        if (ib->i > 0) {
-               left = bi_contig_data(ib);
+               left = ci_contig_data(chn);
                memcpy(ob->p + ob->i, bi_ptr(ib), left);
                ob->i += left;
                if (ib->i - left) {
index f088340..b1d208f 100644 (file)
@@ -84,9 +84,9 @@ trace_hexdump(struct buffer *buf, int len)
        int block1, block2, i, j, padding;
 
        block1 = len;
-        if (block1 > bi_contig_data(buf))
-                block1 = bi_contig_data(buf);
-        block2 = len - block1;
+       if (block1 > b_contig_data(buf, buf->o))
+               block1 = b_contig_data(buf, buf->o);
+       block2 = len - block1;
 
        memcpy(p, buf->p, block1);
        memcpy(p+block1, buf->data, block2);
index 8aab234..2f41c6f 100644 (file)
@@ -2838,7 +2838,7 @@ static int h2_frt_transfer_data(struct h2s *h2s, struct buffer *buf, int count)
        /* Block1 is the length of the first block before the buffer wraps,
         * block2 is the optional second block to reach the end of the frame.
         */
-       block1 = bi_contig_data(h2c->dbuf);
+       block1 = b_contig_data(h2c->dbuf, 0);
        if (block1 > flen)
                block1 = flen;
        block2 = flen - block1;