MINOR: buffer: replace buffer_replace2() with b_rep_blk()
This one is more generic and designed to work on a random block. It
may later get a b_rep_ist() variant since many strings are already
available as (ptr,len).
diff --git a/include/common/buf.h b/include/common/buf.h
index 1ad0052..b0b4e64 100644
--- a/include/common/buf.h
+++ b/include/common/buf.h
@@ -484,6 +484,42 @@
return len;
}
+/* b_rep_blk() : writes the block <blk> at position <pos> which must be in
+ * buffer <b>, and moves the part between <end> and the buffer's tail just
+ * after the end of the copy of <blk>. This effectively replaces the part
+ * located between <pos> and <end> with a copy of <blk> of length <len>. The
+ * buffer's length is automatically updated. This is used to replace a block
+ * with another one inside a buffer. The shift value (positive or negative) is
+ * returned. If there's no space left, the move is not done. If <len> is null,
+ * the <blk> pointer is allowed to be null, in order to erase a block.
+ */
+static inline int b_rep_blk(struct buffer *b, char *pos, char *end, const char *blk, size_t len)
+{
+ int delta;
+
+ delta = len - (end - pos);
+
+ if (b_tail(b) + delta > b_wrap(b))
+ return 0; /* no space left */
+
+ if (b_data(b) &&
+ b_tail(b) + delta > b_head(b) &&
+ b_head(b) >= b_tail(b))
+ return 0; /* no space left before wrapping data */
+
+ /* first, protect the end of the buffer */
+ memmove(end + delta, end, b_tail(b) - end);
+
+ /* now, copy blk over pos */
+ if (len)
+ memcpy(pos, blk, len);
+
+ b_add(b, delta);
+ b_realign_if_empty(b);
+
+ return delta;
+}
+
#endif /* _COMMON_BUF_H */
diff --git a/include/common/buffer.h b/include/common/buffer.h
index 3aaca53..1657275 100644
--- a/include/common/buffer.h
+++ b/include/common/buffer.h
@@ -50,7 +50,6 @@
int init_buffer();
void deinit_buffer();
-int buffer_replace2(struct buffer *b, char *pos, char *end, const char *str, int len);
void buffer_dump(FILE *o, struct buffer *b, int from, int to);
/*****************************************************************/
diff --git a/src/buffer.c b/src/buffer.c
index 9e77947..6f4c494 100644
--- a/src/buffer.c
+++ b/src/buffer.c
@@ -70,43 +70,6 @@
pool_destroy(pool_head_buffer);
}
-/* This function writes the string <str> at position <pos> which must be in
- * buffer <b>, and moves <end> just after the end of <str>. <b>'s parameters
- * <l> and <r> are updated to be valid after the shift. The shift value
- * (positive or negative) is returned. If there's no space left, the move is
- * not done. The function does not adjust ->o because it does not make sense to
- * use it on data scheduled to be sent. For the same reason, it does not make
- * sense to call this function on unparsed data, so <orig> is not updated. The
- * string length is taken from parameter <len>. If <len> is null, the <str>
- * pointer is allowed to be null.
- */
-int buffer_replace2(struct buffer *b, char *pos, char *end, const char *str, int len)
-{
- int delta;
-
- delta = len - (end - pos);
-
- if (b_tail(b) + delta > b_wrap(b))
- return 0; /* no space left */
-
- if (b_data(b) &&
- b_tail(b) + delta > b_head(b) &&
- b_head(b) >= b_tail(b))
- return 0; /* no space left before wrapping data */
-
- /* first, protect the end of the buffer */
- memmove(end + delta, end, b_tail(b) - end);
-
- /* now, copy str over pos */
- if (len)
- memcpy(pos, str, len);
-
- b_add(b, delta);
- b_realign_if_empty(b);
-
- return delta;
-}
-
/*
* Dumps part or all of a buffer.
*/
diff --git a/src/hlua.c b/src/hlua.c
index 2179c61..3fd4d15 100644
--- a/src/hlua.c
+++ b/src/hlua.c
@@ -2888,7 +2888,7 @@
len += len2;
}
luaL_pushresult(&b);
- buffer_replace2(chn->buf, ci_head(chn), ci_head(chn) + len, NULL, 0);
+ b_rep_blk(chn->buf, ci_head(chn), ci_head(chn) + len, NULL, 0);
hlua_resynchonize_proto(chn_strm(chn), !!(chn->flags & CF_ISRESP));
return 1;
}
@@ -3044,7 +3044,7 @@
channel_slow_realign(chn, trash.str);
/* Copy input data in the buffer. */
- max = buffer_replace2(chn->buf, ci_head(chn), ci_head(chn), str + l, max);
+ max = b_rep_blk(chn->buf, ci_head(chn), ci_head(chn), str + l, max);
/* buffer replace considers that the input part is filled.
* so, I must forward these new data in the output part.
diff --git a/src/proto_http.c b/src/proto_http.c
index acecd4e..9407542 100644
--- a/src/proto_http.c
+++ b/src/proto_http.c
@@ -880,7 +880,7 @@
hdr = &idx->v[cur_idx];
if (sol[ctx->del] == ':' && ctx->val + ctx->vlen + ctx->tws == hdr->len) {
/* This was the only value of the header, we must now remove it entirely. */
- delta = buffer_replace2(msg->chn->buf, sol, sol + hdr->len + hdr->cr + 1, NULL, 0);
+ delta = b_rep_blk(msg->chn->buf, sol, sol + hdr->len + hdr->cr + 1, NULL, 0);
http_msg_move_end(msg, delta);
idx->used--;
hdr->len = 0; /* unused entry */
@@ -900,7 +900,7 @@
*/
skip_comma = (ctx->val + ctx->vlen + ctx->tws == hdr->len) ? 0 : 1;
- delta = buffer_replace2(msg->chn->buf, sol + ctx->del + skip_comma,
+ delta = b_rep_blk(msg->chn->buf, sol + ctx->del + skip_comma,
sol + ctx->val + ctx->vlen + ctx->tws + skip_comma,
NULL, 0);
hdr->len += delta;
@@ -1354,7 +1354,7 @@
return 0;
}
/* add HTTP version */
- delta = buffer_replace2(msg->chn->buf, cur_end, cur_end, " HTTP/1.0\r\n", 11);
+ delta = b_rep_blk(msg->chn->buf, cur_end, cur_end, " HTTP/1.0\r\n", 11);
http_msg_move_end(msg, delta);
cur_end += delta;
cur_end = (char *)http_parse_reqline(msg,
@@ -2429,7 +2429,7 @@
if (output->len == -1)
return -1;
- delta = buffer_replace2(msg->chn->buf, val, val_end, output->str, output->len);
+ delta = b_rep_blk(msg->chn->buf, val, val_end, output->str, output->len);
hdr->len += delta;
http_msg_move_end(msg, delta);
@@ -3765,7 +3765,7 @@
char *cur_end = cur_ptr + txn->req.sl.rq.l;
int delta;
- delta = buffer_replace2(req->buf, cur_ptr + msg->sl.rq.u, path, NULL, 0);
+ delta = b_rep_blk(req->buf, cur_ptr + msg->sl.rq.u, path, NULL, 0);
http_msg_move_end(&txn->req, delta);
cur_end += delta;
if (http_parse_reqline(&txn->req, HTTP_MSG_RQMETH, cur_ptr, cur_end + 1, NULL, NULL) == NULL)
@@ -3776,7 +3776,7 @@
char *cur_end = cur_ptr + txn->req.sl.rq.l;
int delta;
- delta = buffer_replace2(req->buf, cur_ptr + msg->sl.rq.u,
+ delta = b_rep_blk(req->buf, cur_ptr + msg->sl.rq.u,
cur_ptr + msg->sl.rq.u + msg->sl.rq.u_l, "/", 1);
http_msg_move_end(&txn->req, delta);
cur_end += delta;
@@ -6516,7 +6516,7 @@
if (trash.len < 0)
return -1;
- delta = buffer_replace2(req->buf, cur_ptr, cur_end, trash.str, trash.len);
+ delta = b_rep_blk(req->buf, cur_ptr, cur_end, trash.str, trash.len);
/* FIXME: if the user adds a newline in the replacement, the
* index will not be recalculated for now, and the new line
* will not be counted as a new header.
@@ -6529,7 +6529,7 @@
break;
case ACT_REMOVE:
- delta = buffer_replace2(req->buf, cur_ptr, cur_next, NULL, 0);
+ delta = b_rep_blk(req->buf, cur_ptr, cur_next, NULL, 0);
cur_next += delta;
http_msg_move_end(&txn->req, delta);
@@ -6604,7 +6604,7 @@
if (trash.len < 0)
return -1;
- delta = buffer_replace2(req->buf, cur_ptr, cur_end, trash.str, trash.len);
+ delta = b_rep_blk(req->buf, cur_ptr, cur_end, trash.str, trash.len);
/* FIXME: if the user adds a newline in the replacement, the
* index will not be recalculated for now, and the new line
* will not be counted as a new header.
@@ -6764,7 +6764,7 @@
next++;
}
}
- return buffer_replace2(buf, prev, next, NULL, 0);
+ return b_rep_blk(buf, prev, next, NULL, 0);
}
/*
@@ -6935,13 +6935,13 @@
int stripped_after = 0;
if (att_end != equal) {
- stripped_before = buffer_replace2(req->buf, att_end, equal, NULL, 0);
+ stripped_before = b_rep_blk(req->buf, att_end, equal, NULL, 0);
equal += stripped_before;
val_beg += stripped_before;
}
if (val_beg > equal + 1) {
- stripped_after = buffer_replace2(req->buf, equal + 1, val_beg, NULL, 0);
+ stripped_after = b_rep_blk(req->buf, equal + 1, val_beg, NULL, 0);
val_beg += stripped_after;
stripped_before += stripped_after;
}
@@ -7122,7 +7122,7 @@
if ((s->be->ck_opts & PR_CK_PFX) && (delim != val_end)) {
int delta; /* negative */
- delta = buffer_replace2(req->buf, val_beg, delim + 1, NULL, 0);
+ delta = b_rep_blk(req->buf, val_beg, delim + 1, NULL, 0);
val_end += delta;
next += delta;
hdr_end += delta;
@@ -7180,7 +7180,7 @@
hdr_end = del_from;
cur_hdr->len += delta;
} else {
- delta = buffer_replace2(req->buf, hdr_beg, hdr_next, NULL, 0);
+ delta = b_rep_blk(req->buf, hdr_beg, hdr_next, NULL, 0);
/* FIXME: this should be a separate function */
txn->hdr_idx.v[old_idx].next = cur_hdr->next;
@@ -7252,7 +7252,7 @@
if (trash.len < 0)
return -1;
- delta = buffer_replace2(rtr->buf, cur_ptr, cur_end, trash.str, trash.len);
+ delta = b_rep_blk(rtr->buf, cur_ptr, cur_end, trash.str, trash.len);
/* FIXME: if the user adds a newline in the replacement, the
* index will not be recalculated for now, and the new line
* will not be counted as a new header.
@@ -7265,7 +7265,7 @@
break;
case ACT_REMOVE:
- delta = buffer_replace2(rtr->buf, cur_ptr, cur_next, NULL, 0);
+ delta = b_rep_blk(rtr->buf, cur_ptr, cur_next, NULL, 0);
cur_next += delta;
http_msg_move_end(&txn->rsp, delta);
@@ -7333,7 +7333,7 @@
if (trash.len < 0)
return -1;
- delta = buffer_replace2(rtr->buf, cur_ptr, cur_end, trash.str, trash.len);
+ delta = b_rep_blk(rtr->buf, cur_ptr, cur_end, trash.str, trash.len);
/* FIXME: if the user adds a newline in the replacement, the
* index will not be recalculated for now, and the new line
* will not be counted as a new header.
@@ -7588,13 +7588,13 @@
int stripped_after = 0;
if (att_end != equal) {
- stripped_before = buffer_replace2(res->buf, att_end, equal, NULL, 0);
+ stripped_before = b_rep_blk(res->buf, att_end, equal, NULL, 0);
equal += stripped_before;
val_beg += stripped_before;
}
if (val_beg > equal + 1) {
- stripped_after = buffer_replace2(res->buf, equal + 1, val_beg, NULL, 0);
+ stripped_after = b_rep_blk(res->buf, equal + 1, val_beg, NULL, 0);
val_beg += stripped_after;
stripped_before += stripped_after;
}
@@ -7652,7 +7652,7 @@
/* this cookie must be deleted */
if (*prev == ':' && next == hdr_end) {
/* whole header */
- delta = buffer_replace2(res->buf, hdr_beg, hdr_next, NULL, 0);
+ delta = b_rep_blk(res->buf, hdr_beg, hdr_next, NULL, 0);
txn->hdr_idx.v[old_idx].next = cur_hdr->next;
txn->hdr_idx.used--;
cur_hdr->len = 0;
@@ -7679,7 +7679,7 @@
/* replace bytes val_beg->val_end with the cookie name associated
* with this server since we know it.
*/
- delta = buffer_replace2(res->buf, val_beg, val_end, srv->cookie, srv->cklen);
+ delta = b_rep_blk(res->buf, val_beg, val_end, srv->cookie, srv->cklen);
next += delta;
hdr_end += delta;
hdr_next += delta;
@@ -7693,7 +7693,7 @@
/* insert the cookie name associated with this server
* before existing cookie, and insert a delimiter between them..
*/
- delta = buffer_replace2(res->buf, val_beg, val_beg, srv->cookie, srv->cklen + 1);
+ delta = b_rep_blk(res->buf, val_beg, val_beg, srv->cookie, srv->cklen + 1);
next += delta;
hdr_end += delta;
hdr_next += delta;
@@ -11927,7 +11927,7 @@
}
/* commit changes and adjust end of message */
- delta = buffer_replace2(s->req.buf, cur_ptr, cur_end, replace + offset, len - offset);
+ delta = b_rep_blk(s->req.buf, cur_ptr, cur_end, replace + offset, len - offset);
txn->req.sl.rq.l += delta;
txn->hdr_idx.v[0].len += delta;
http_msg_move_end(&txn->req, delta);
@@ -11966,7 +11966,7 @@
cur_end = ci_head(&s->res) + txn->rsp.sl.st.r + txn->rsp.sl.st.r_l;
/* commit changes and adjust message */
- delta = buffer_replace2(s->res.buf, cur_ptr, cur_end, trash.str, trash.len);
+ delta = b_rep_blk(s->res.buf, cur_ptr, cur_end, trash.str, trash.len);
/* adjust res line offsets and lengths */
txn->rsp.sl.st.r += c_l - txn->rsp.sl.st.c_l;