diff options
Diffstat (limited to '')
| -rw-r--r-- | src/http_htx.c | 3028 |
1 files changed, 3028 insertions, 0 deletions
diff --git a/src/http_htx.c b/src/http_htx.c new file mode 100644 index 0000000..004d343 --- /dev/null +++ b/src/http_htx.c @@ -0,0 +1,3028 @@ +/* + * Functions to manipulate HTTP messages using the internal representation. + * + * Copyright (C) 2018 HAProxy Technologies, Christopher Faulet <cfaulet@haproxy.com> + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version + * 2 of the License, or (at your option) any later version. + * + */ +#include <sys/types.h> +#include <sys/stat.h> +#include <ctype.h> +#include <fcntl.h> +#include <unistd.h> + +#include <haproxy/api.h> +#include <haproxy/arg.h> +#include <haproxy/cfgparse.h> +#include <haproxy/global.h> +#include <haproxy/h1.h> +#include <haproxy/http.h> +#include <haproxy/http-hdr.h> +#include <haproxy/http_fetch.h> +#include <haproxy/http_htx.h> +#include <haproxy/htx.h> +#include <haproxy/log.h> +#include <haproxy/regex.h> +#include <haproxy/sample.h> +#include <haproxy/tools.h> + + +struct buffer http_err_chunks[HTTP_ERR_SIZE]; +struct http_reply http_err_replies[HTTP_ERR_SIZE]; + +struct eb_root http_error_messages = EB_ROOT; +struct list http_errors_list = LIST_HEAD_INIT(http_errors_list); +struct list http_replies_list = LIST_HEAD_INIT(http_replies_list); + +/* The declaration of an errorfiles/errorfile directives. Used during config + * parsing only. */ +struct conf_errors { + char type; /* directive type (0: errorfiles, 1: errorfile) */ + union { + struct { + int status; /* the status code associated to this error */ + struct http_reply *reply; /* the http reply for the errorfile */ + } errorfile; /* describe an "errorfile" directive */ + struct { + char *name; /* the http-errors section name */ + char status[HTTP_ERR_SIZE]; /* list of status to import (0: ignore, 1: implicit import, 2: explicit import) */ + } errorfiles; /* describe an "errorfiles" directive */ + } info; + + char *file; /* file where the directive appears */ + int line; /* line where the directive appears */ + + struct list list; /* next conf_errors */ +}; + +/* Returns the next unporocessed start line in the HTX message. It returns NULL + * if the start-line is undefined (first == -1). Otherwise, it returns the + * pointer on the htx_sl structure. + */ +struct htx_sl *http_get_stline(const struct htx *htx) +{ + struct htx_blk *blk; + + blk = htx_get_first_blk(htx); + if (!blk || (htx_get_blk_type(blk) != HTX_BLK_REQ_SL && htx_get_blk_type(blk) != HTX_BLK_RES_SL)) + return NULL; + return htx_get_blk_ptr(htx, blk); +} + +/* Returns the headers size in the HTX message */ +size_t http_get_hdrs_size(struct htx *htx) +{ + struct htx_blk *blk; + size_t sz = 0; + + blk = htx_get_first_blk(htx); + if (!blk || htx_get_blk_type(blk) > HTX_BLK_EOH) + return sz; + + for (; blk; blk = htx_get_next_blk(htx, blk)) { + sz += htx_get_blksz(blk); + if (htx_get_blk_type(blk) == HTX_BLK_EOH) + break; + } + return sz; +} + +/* Finds the first or next occurrence of header matching <pattern> in the HTX + * message <htx> using the context <ctx>. This structure holds everything + * necessary to use the header and find next occurrence. If its <blk> member is + * NULL, the header is searched from the beginning. Otherwise, the next + * occurrence is returned. The function returns 1 when it finds a value, and 0 + * when there is no more. It is designed to work with headers defined as + * comma-separated lists. If HTTP_FIND_FL_FULL flag is set, it works on + * full-line headers in whose comma is not a delimiter but is part of the + * syntax. A special case, if ctx->value is NULL when searching for a new values + * of a header, the current header is rescanned. This allows rescanning after a + * header deletion. + * + * The matching method is chosen by checking the flags : + * + * * HTTP_FIND_FL_MATCH_REG : <pattern> is a regex. header names matching + * the regex are evaluated. + * * HTTP_FIND_FL_MATCH_STR : <pattern> is a string. The header names equal + * to the string are evaluated. + * * HTTP_FIND_FL_MATCH_PFX : <pattern> is a string. The header names + * starting by the string are evaluated. + * * HTTP_FIND_FL_MATCH_SFX : <pattern> is a string. The header names + * ending by the string are evaluated. + * * HTTP_FIND_FL_MATCH_SUB : <pattern> is a string. The header names + * containing the string are evaluated. + */ + +#define HTTP_FIND_FL_MATCH_STR 0x0001 +#define HTTP_FIND_FL_MATCH_PFX 0x0002 +#define HTTP_FIND_FL_MATCH_SFX 0x0003 +#define HTTP_FIND_FL_MATCH_SUB 0x0004 +#define HTTP_FIND_FL_MATCH_REG 0x0005 +/* 0x0006..0x000f: for other matching methods */ +#define HTTP_FIND_FL_MATCH_TYPE 0x000F +#define HTTP_FIND_FL_FULL 0x0010 + +static int __http_find_header(const struct htx *htx, const void *pattern, struct http_hdr_ctx *ctx, int flags) +{ + struct htx_blk *blk = ctx->blk; + struct ist n, v; + enum htx_blk_type type; + + if (blk) { + char *p; + + if (!isttest(ctx->value)) + goto rescan_hdr; + if (flags & HTTP_FIND_FL_FULL) + goto next_blk; + v = htx_get_blk_value(htx, blk); + p = istend(ctx->value) + ctx->lws_after; + v.len -= (p - v.ptr); + v.ptr = p; + if (!v.len) + goto next_blk; + /* Skip comma */ + if (*(v.ptr) == ',') { + v = istnext(v); + } + + goto return_hdr; + } + + if (htx_is_empty(htx)) + return 0; + + for (blk = htx_get_first_blk(htx); blk; blk = htx_get_next_blk(htx, blk)) { + rescan_hdr: + type = htx_get_blk_type(blk); + if (type == HTX_BLK_EOH) + break; + if (type != HTX_BLK_HDR) + continue; + + if ((flags & HTTP_FIND_FL_MATCH_TYPE) == HTTP_FIND_FL_MATCH_REG) { + const struct my_regex *re = pattern; + + n = htx_get_blk_name(htx, blk); + if (!regex_exec2(re, n.ptr, n.len)) + goto next_blk; + } + else { + const struct ist name = *(const struct ist *)(pattern); + + /* If no name was passed, we want any header. So skip the comparison */ + if (!istlen(name)) + goto match; + + n = htx_get_blk_name(htx, blk); + switch (flags & HTTP_FIND_FL_MATCH_TYPE) { + case HTTP_FIND_FL_MATCH_STR: + if (!isteqi(n, name)) + goto next_blk; + break; + case HTTP_FIND_FL_MATCH_PFX: + if (istlen(n) < istlen(name)) + goto next_blk; + + n = ist2(istptr(n), istlen(name)); + if (!isteqi(n, name)) + goto next_blk; + break; + case HTTP_FIND_FL_MATCH_SFX: + if (istlen(n) < istlen(name)) + goto next_blk; + + n = ist2(istend(n) - istlen(name), + istlen(name)); + if (!isteqi(n, name)) + goto next_blk; + break; + case HTTP_FIND_FL_MATCH_SUB: + if (!strnistr(n.ptr, n.len, name.ptr, name.len)) + goto next_blk; + break; + default: + goto next_blk; + break; + } + } + match: + v = htx_get_blk_value(htx, blk); + + return_hdr: + ctx->lws_before = 0; + ctx->lws_after = 0; + while (v.len && HTTP_IS_LWS(*v.ptr)) { + v = istnext(v); + ctx->lws_before++; + } + if (!(flags & HTTP_FIND_FL_FULL)) + v.len = http_find_hdr_value_end(v.ptr, istend(v)) - v.ptr; + + while (v.len && HTTP_IS_LWS(*(istend(v) - 1))) { + v.len--; + ctx->lws_after++; + } + ctx->blk = blk; + ctx->value = v; + return 1; + + next_blk: + ; + } + + ctx->blk = NULL; + ctx->value = ist(""); + ctx->lws_before = ctx->lws_after = 0; + return 0; +} + + +/* Header names must match <name> */ +int http_find_header(const struct htx *htx, const struct ist name, struct http_hdr_ctx *ctx, int full) +{ + return __http_find_header(htx, &name, ctx, HTTP_FIND_FL_MATCH_STR | (full ? HTTP_FIND_FL_FULL : 0)); +} + +/* Header names must match <name>. Same than http_find_header */ +int http_find_str_header(const struct htx *htx, const struct ist name, struct http_hdr_ctx *ctx, int full) +{ + return __http_find_header(htx, &name, ctx, HTTP_FIND_FL_MATCH_STR | (full ? HTTP_FIND_FL_FULL : 0)); +} + + +/* Header names must start with <prefix> */ +int http_find_pfx_header(const struct htx *htx, const struct ist prefix, struct http_hdr_ctx *ctx, int full) +{ + return __http_find_header(htx, &prefix, ctx, HTTP_FIND_FL_MATCH_PFX | (full ? HTTP_FIND_FL_FULL : 0)); +} + +/* Header names must end with <suffix> */ +int http_find_sfx_header(const struct htx *htx, const struct ist suffix, struct http_hdr_ctx *ctx, int full) +{ + return __http_find_header(htx, &suffix, ctx, HTTP_FIND_FL_MATCH_SFX | (full ? HTTP_FIND_FL_FULL : 0)); +} +/* Header names must contain <sub> */ +int http_find_sub_header(const struct htx *htx, const struct ist sub, struct http_hdr_ctx *ctx, int full) +{ + return __http_find_header(htx, &sub, ctx, HTTP_FIND_FL_MATCH_SUB | (full ? HTTP_FIND_FL_FULL : 0)); +} + +/* Header names must match <re> regex*/ +int http_match_header(const struct htx *htx, const struct my_regex *re, struct http_hdr_ctx *ctx, int full) +{ + return __http_find_header(htx, re, ctx, HTTP_FIND_FL_MATCH_REG | (full ? HTTP_FIND_FL_FULL : 0)); +} + + +/* Adds a header block int the HTX message <htx>, just before the EOH block. It + * returns 1 on success, otherwise it returns 0. + */ +int http_add_header(struct htx *htx, const struct ist n, const struct ist v) +{ + 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) + 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 type = 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 (type == HTX_BLK_EOH) + break; + + blk = pblk; + } + + end: + 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; + } + return 1; + + fail: + return 0; +} + +/* Replaces parts of the start-line of the HTX message <htx>. It returns 1 on + * success, otherwise it returns 0. + */ +int http_replace_stline(struct htx *htx, const struct ist p1, const struct ist p2, const struct ist p3) +{ + struct htx_blk *blk; + + blk = htx_get_first_blk(htx); + if (!blk || !htx_replace_stline(htx, blk, p1, p2, p3)) + return 0; + return 1; +} + +/* Replace the request method in the HTX message <htx> by <meth>. It returns 1 + * on success, otherwise 0. + */ +int http_replace_req_meth(struct htx *htx, const struct ist meth) +{ + struct buffer *temp = get_trash_chunk(); + struct htx_sl *sl = http_get_stline(htx); + struct ist uri, vsn; + + if (!sl) + return 0; + + /* Start by copying old uri and version */ + chunk_memcat(temp, HTX_SL_REQ_UPTR(sl), HTX_SL_REQ_ULEN(sl)); /* uri */ + uri = ist2(temp->area, HTX_SL_REQ_ULEN(sl)); + + chunk_memcat(temp, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area + uri.len, HTX_SL_REQ_VLEN(sl)); + + /* create the new start line */ + sl->info.req.meth = find_http_meth(meth.ptr, meth.len); + return http_replace_stline(htx, meth, uri, vsn); +} + +/* Replace the request uri in the HTX message <htx> by <uri>. It returns 1 on + * success, otherwise 0. + */ +int http_replace_req_uri(struct htx *htx, const struct ist uri) +{ + struct buffer *temp = get_trash_chunk(); + struct htx_sl *sl = http_get_stline(htx); + struct ist meth, vsn; + + if (!sl) + goto fail; + + /* Start by copying old method and version */ + chunk_memcat(temp, HTX_SL_REQ_MPTR(sl), HTX_SL_REQ_MLEN(sl)); /* meth */ + meth = ist2(temp->area, HTX_SL_REQ_MLEN(sl)); + + chunk_memcat(temp, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area + meth.len, HTX_SL_REQ_VLEN(sl)); + + /* create the new start line */ + if (!http_replace_stline(htx, meth, uri, vsn)) + goto fail; + + sl = http_get_stline(htx); + ALREADY_CHECKED(sl); /* the stline exists because http_replace_stline() succeeded */ + + sl->flags &= ~HTX_SL_F_NORMALIZED_URI; + if (!http_update_host(htx, sl, uri)) + goto fail; + + return 1; + fail: + return 0; +} + +/* Replace the request path in the HTX message <htx> by <path>. The host part is + * preserverd. if <with_qs> is set, the query string is evaluated as part of the + * path and replaced. Otherwise, it is preserved too. It returns 1 on success, + * otherwise 0. + */ +int http_replace_req_path(struct htx *htx, const struct ist path, int with_qs) +{ + struct buffer *temp = get_trash_chunk(); + struct htx_sl *sl = http_get_stline(htx); + struct ist meth, uri, vsn, p; + size_t plen = 0; + struct http_uri_parser parser; + + if (!sl) + return 0; + + uri = htx_sl_req_uri(sl); + parser = http_uri_parser_init(uri); + p = http_parse_path(&parser); + if (!isttest(p)) + p = uri; + if (with_qs) + plen = p.len; + else { + while (plen < p.len && *(p.ptr + plen) != '?') + plen++; + } + + /* Start by copying old method and version and create the new uri */ + chunk_memcat(temp, HTX_SL_REQ_MPTR(sl), HTX_SL_REQ_MLEN(sl)); /* meth */ + meth = ist2(temp->area, HTX_SL_REQ_MLEN(sl)); + + chunk_memcat(temp, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area + meth.len, HTX_SL_REQ_VLEN(sl)); + + chunk_memcat(temp, uri.ptr, p.ptr - uri.ptr); /* uri: host part */ + chunk_istcat(temp, path); /* uri: new path */ + chunk_memcat(temp, p.ptr + plen, p.len - plen); /* uri: QS part */ + uri = ist2(temp->area + meth.len + vsn.len, uri.len - plen + path.len); + + /* create the new start line */ + return http_replace_stline(htx, meth, uri, vsn); +} + +/* Replace the request query-string in the HTX message <htx> by <query>. The + * host part and the path are preserved. It returns 1 on success, otherwise + * 0. + */ +int http_replace_req_query(struct htx *htx, const struct ist query) +{ + struct buffer *temp = get_trash_chunk(); + struct htx_sl *sl = http_get_stline(htx); + struct ist meth, uri, vsn, q; + int offset = 1; + + if (!sl) + return 0; + + uri = htx_sl_req_uri(sl); + q = uri; + while (q.len > 0 && *(q.ptr) != '?') { + q = istnext(q); + } + + /* skip the question mark or indicate that we must insert it + * (but only if the format string is not empty then). + */ + if (q.len) { + q = istnext(q); + } + else if (query.len > 1) + offset = 0; + + /* Start by copying old method and version and create the new uri */ + chunk_memcat(temp, HTX_SL_REQ_MPTR(sl), HTX_SL_REQ_MLEN(sl)); /* meth */ + meth = ist2(temp->area, HTX_SL_REQ_MLEN(sl)); + + chunk_memcat(temp, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area + meth.len, HTX_SL_REQ_VLEN(sl)); + + chunk_memcat(temp, uri.ptr, q.ptr - uri.ptr); /* uri: host + path part */ + chunk_memcat(temp, query.ptr + offset, query.len - offset); /* uri: new QS */ + uri = ist2(temp->area + meth.len + vsn.len, uri.len - q.len + query.len - offset); + + /* create the new start line */ + return http_replace_stline(htx, meth, uri, vsn); +} + +/* Replace the response status in the HTX message <htx> by <status>. It returns + * 1 on success, otherwise 0. +*/ +int http_replace_res_status(struct htx *htx, const struct ist status, const struct ist reason) +{ + struct buffer *temp = get_trash_chunk(); + struct htx_sl *sl = http_get_stline(htx); + struct ist vsn, r; + + if (!sl) + return 0; + + /* Start by copying old uri and version */ + chunk_memcat(temp, HTX_SL_RES_VPTR(sl), HTX_SL_RES_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area, HTX_SL_RES_VLEN(sl)); + r = reason; + if (!isttest(r)) { + chunk_memcat(temp, HTX_SL_RES_RPTR(sl), HTX_SL_RES_RLEN(sl)); /* reason */ + r = ist2(temp->area + vsn.len, HTX_SL_RES_RLEN(sl)); + } + + /* create the new start line */ + sl->info.res.status = strl2ui(status.ptr, status.len); + return http_replace_stline(htx, vsn, status, r); +} + +/* Replace the response reason in the HTX message <htx> by <reason>. It returns + * 1 on success, otherwise 0. +*/ +int http_replace_res_reason(struct htx *htx, const struct ist reason) +{ + struct buffer *temp = get_trash_chunk(); + struct htx_sl *sl = http_get_stline(htx); + struct ist vsn, status; + + if (!sl) + return 0; + + /* Start by copying old uri and version */ + chunk_memcat(temp, HTX_SL_RES_VPTR(sl), HTX_SL_RES_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area, HTX_SL_RES_VLEN(sl)); + + chunk_memcat(temp, HTX_SL_RES_CPTR(sl), HTX_SL_RES_CLEN(sl)); /* code */ + status = ist2(temp->area + vsn.len, HTX_SL_RES_CLEN(sl)); + + /* create the new start line */ + return http_replace_stline(htx, vsn, status, reason); +} + +/* Append new value <data> after <ctx> value in header + * if header is not empty (at least one value exists): + * - ',' delimiter is added before <data> is appended + * - <ctx> must be valid and must point to an existing value, + * else it is an error and prepend_value should be used instead. + * + * ctx is updated to point to new value + * + * Returns 1 on success and 0 on failure. + */ +int http_append_header_value(struct htx *htx, struct http_hdr_ctx *ctx, const struct ist data) +{ + char *start; + struct htx_blk *blk = ctx->blk; + struct ist v; + uint32_t off = 0; + + if (!blk) + goto fail; + + v = htx_get_blk_value(htx, blk); + + if (!istlen(v)) { + start = v.ptr; + goto empty; /* header is empty, append without ',' */ + } + if (unlikely(!istlen(ctx->value))) + goto fail; /* invalid: value is empty, not supported */ + + start = istend(ctx->value) + ctx->lws_after; + off = start - v.ptr; + + blk = htx_replace_blk_value(htx, blk, ist2(start, 0), ist(",")); + if (!blk) + goto fail; + off += 1; /* add 1 for ',' */ + v = htx_get_blk_value(htx, blk); + start = v.ptr + off; + + empty: + blk = htx_replace_blk_value(htx, blk, ist2(start, 0), data); + if (!blk) + goto fail; + v = htx_get_blk_value(htx, blk); + + ctx->blk = blk; + ctx->value = ist2(v.ptr + off, data.len); + ctx->lws_before = ctx->lws_after = 0; + + return 1; + fail: + return 0; +} + +/* Prepend new value <data> before <ctx> value in header + * if <ctx> is not first value (at least one value exists): + * - ',' delimiter is added after <data> is prepended + * + * ctx is updated to point to new value + * + * Returns 1 on success and 0 on failure. + */ +int http_prepend_header_value(struct htx *htx, struct http_hdr_ctx *ctx, const struct ist data) +{ + char *start; + struct htx_blk *blk = ctx->blk; + struct ist v; + uint32_t off = 0; + uint8_t first; + + if (!blk) + goto fail; + + v = htx_get_blk_value(htx, blk); + + first = !istlen(v); + start = first ? v.ptr : istptr(ctx->value) - ctx->lws_before; + + if (unlikely(!istlen(ctx->value))) + goto fail; /* invalid: value is empty, not supported */ + + off = start - v.ptr; + + blk = htx_replace_blk_value(htx, blk, ist2(start, 0), data); + if (!blk) + goto fail; + v = htx_get_blk_value(htx, blk); + + if (first) + goto end; /* header is empty, don't append ',' */ + + start = v.ptr + off + data.len; + + blk = htx_replace_blk_value(htx, blk, ist2(start, 0), ist(",")); + if (!blk) + goto fail; + v = htx_get_blk_value(htx, blk); + + end: + ctx->blk = blk; + ctx->value = ist2(v.ptr + off, data.len); + ctx->lws_before = ctx->lws_after = 0; + + return 1; + fail: + return 0; +} + +/* Replaces a part of a header value referenced in the context <ctx> by + * <data>. It returns 1 on success, otherwise it returns 0. The context is + * updated if necessary. + */ +int http_replace_header_value(struct htx *htx, struct http_hdr_ctx *ctx, const struct ist data) +{ + struct htx_blk *blk = ctx->blk; + struct htx_sl *sl; + char *start; + struct ist v; + uint32_t len, off; + + if (!blk) + goto fail; + + v = htx_get_blk_value(htx, blk); + start = ctx->value.ptr - ctx->lws_before; + len = ctx->lws_before + ctx->value.len + ctx->lws_after; + off = start - v.ptr; + + blk = htx_replace_blk_value(htx, blk, ist2(start, len), data); + if (!blk) + goto fail; + + v = htx_get_blk_value(htx, blk); + + sl = http_get_stline(htx); + if (sl && (sl->flags & HTX_SL_F_HAS_AUTHORITY)) { + struct ist n = htx_get_blk_name(htx, blk); + + if (isteq(n, ist("host"))) { + if (!http_update_authority(htx, sl, v)) + goto fail; + ctx->blk = NULL; + http_find_header(htx, ist("host"), ctx, 1); + blk = ctx->blk; + v = htx_get_blk_value(htx, blk); + } + } + + ctx->blk = blk; + ctx->value = ist2(v.ptr + off, data.len); + ctx->lws_before = ctx->lws_after = 0; + + return 1; + fail: + return 0; +} + +/* Fully replaces a header referenced in the context <ctx> by the name <name> + * with the value <value>. It returns 1 on success, otherwise it returns 0. The + * context is updated if necessary. + */ +int http_replace_header(struct htx *htx, struct http_hdr_ctx *ctx, + const struct ist name, const struct ist value) +{ + struct htx_blk *blk = ctx->blk; + struct htx_sl *sl; + + if (!blk) + goto fail; + + blk = htx_replace_header(htx, blk, name, value); + if (!blk) + goto fail; + + sl = http_get_stline(htx); + if (sl && (sl->flags & HTX_SL_F_HAS_AUTHORITY) && isteqi(name, ist("host"))) { + if (!http_update_authority(htx, sl, value)) + goto fail; + ctx->blk = NULL; + http_find_header(htx, ist("host"), ctx, 1); + blk = ctx->blk; + } + + ctx->blk = blk; + ctx->value = ist(NULL); + ctx->lws_before = ctx->lws_after = 0; + + return 1; + fail: + return 0; +} + +/* Remove one value of a header. This only works on a <ctx> returned by + * http_find_header function. The value is removed, as well as surrounding commas + * if any. If the removed value was alone, the whole header is removed. The + * <ctx> is always updated accordingly, as well as the HTX message <htx>. It + * returns 1 on success. Otherwise, it returns 0. The <ctx> is always left in a + * form that can be handled by http_find_header() to find next occurrence. + */ +int http_remove_header(struct htx *htx, struct http_hdr_ctx *ctx) +{ + struct htx_blk *blk = ctx->blk; + char *start; + struct ist v; + uint32_t len; + + if (!blk) + return 0; + + start = ctx->value.ptr - ctx->lws_before; + len = ctx->lws_before + ctx->value.len + ctx->lws_after; + + v = htx_get_blk_value(htx, blk); + if (len == v.len) { + blk = htx_remove_blk(htx, blk); + if (blk || htx_is_empty(htx)) { + ctx->blk = blk; + ctx->value = IST_NULL; + ctx->lws_before = ctx->lws_after = 0; + } + else { + ctx->blk = htx_get_blk(htx, htx->tail); + ctx->value = htx_get_blk_value(htx, ctx->blk); + ctx->lws_before = ctx->lws_after = 0; + } + return 1; + } + + /* This was not the only value of this header. We have to remove the + * part pointed by ctx->value. If it is the last entry of the list, we + * remove the last separator. + */ + if (start == v.ptr) { + /* It's the first header part but not the only one. So remove + * the comma after it. */ + len++; + } + else { + /* There is at least one header part before the removed one. So + * remove the comma between them. */ + start--; + len++; + } + /* Update the block content and its len */ + memmove(start, start+len, v.len-len); + htx_change_blk_value_len(htx, blk, v.len-len); + + /* Finally update the ctx */ + ctx->value = ist2(start, 0); + ctx->lws_before = ctx->lws_after = 0; + + return 1; +} + +/* Updates the authority part of the uri with the value <host>. It happens when + * the header host is modified. It returns 0 on failure and 1 on success. It is + * the caller responsibility to provide the start-line and to be sure the uri + * contains an authority. Thus, if no authority is found in the uri, an error is + * returned. + */ +int http_update_authority(struct htx *htx, struct htx_sl *sl, const struct ist host) +{ + struct buffer *temp = get_trash_chunk(); + struct ist meth, vsn, uri, authority; + struct http_uri_parser parser; + + uri = htx_sl_req_uri(sl); + parser = http_uri_parser_init(uri); + authority = http_parse_authority(&parser, 1); + if (!authority.len) + return 0; + + /* Don't update the uri if there is no change */ + if (isteq(host, authority)) + return 1; + + /* Start by copying old method and version */ + chunk_memcat(temp, HTX_SL_REQ_MPTR(sl), HTX_SL_REQ_MLEN(sl)); /* meth */ + meth = ist2(temp->area, HTX_SL_REQ_MLEN(sl)); + + chunk_memcat(temp, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl)); /* vsn */ + vsn = ist2(temp->area + meth.len, HTX_SL_REQ_VLEN(sl)); + + chunk_memcat(temp, uri.ptr, authority.ptr - uri.ptr); + chunk_istcat(temp, host); + chunk_memcat(temp, istend(authority), istend(uri) - istend(authority)); + uri = ist2(temp->area + meth.len + vsn.len, host.len + uri.len - authority.len); /* uri */ + + return http_replace_stline(htx, meth, uri, vsn); + +} + +/* Update the header host by extracting the authority of the uri <uri>. flags of + * the start-line are also updated accordingly. For orgin-form and asterisk-form + * uri, the header host is not changed and the flag HTX_SL_F_HAS_AUTHORITY is + * removed from the flags of the start-line. Otherwise, this flag is set and the + * authority is used to set the value of the header host. This function returns + * 0 on failure and 1 on success. +*/ +int http_update_host(struct htx *htx, struct htx_sl *sl, const struct ist uri) +{ + struct ist authority; + struct http_hdr_ctx ctx; + struct http_uri_parser parser = http_uri_parser_init(uri); + + if (parser.format == URI_PARSER_FORMAT_EMPTY || + parser.format == URI_PARSER_FORMAT_ASTERISK || + parser.format == URI_PARSER_FORMAT_ABSPATH) { + sl->flags &= ~HTX_SL_F_HAS_AUTHORITY; + } + else { + sl->flags |= HTX_SL_F_HAS_AUTHORITY; + if (sl->info.req.meth != HTTP_METH_CONNECT) { + // absolute-form (RFC7320 #5.3.2) + sl->flags |= HTX_SL_F_HAS_SCHM; + if (uri.len > 4 && (uri.ptr[0] | 0x20) == 'h') + sl->flags |= ((uri.ptr[4] == ':') ? HTX_SL_F_SCHM_HTTP : HTX_SL_F_SCHM_HTTPS); + + authority = http_parse_authority(&parser, 1); + if (!authority.len) + goto fail; + } + else { + // authority-form (RFC7320 #5.3.3) + authority = uri; + } + + /* Replace header host value */ + ctx.blk = NULL; + while (http_find_header(htx, ist("host"), &ctx, 1)) { + if (!http_replace_header_value(htx, &ctx, authority)) + goto fail; + } + + } + return 1; + fail: + return 0; +} + +/* Return in <vptr> and <vlen> the pointer and length of occurrence <occ> of + * header whose name is <hname> of length <hlen>. If <ctx> is null, lookup is + * performed over the whole headers. Otherwise it must contain a valid header + * context, initialised with ctx->blk=NULL for the first lookup in a series. If + * <occ> is positive or null, occurrence #occ from the beginning (or last ctx) + * is returned. Occ #0 and #1 are equivalent. If <occ> is negative (and no less + * than -MAX_HDR_HISTORY), the occurrence is counted from the last one which is + * -1. The value fetch stops at commas, so this function is suited for use with + * list headers. + * The return value is 0 if nothing was found, or non-zero otherwise. + */ +unsigned int http_get_htx_hdr(const struct htx *htx, const struct ist hdr, + int occ, struct http_hdr_ctx *ctx, char **vptr, size_t *vlen) +{ + struct http_hdr_ctx local_ctx; + struct ist val_hist[MAX_HDR_HISTORY]; + unsigned int hist_idx; + int found; + + if (!ctx) { + local_ctx.blk = NULL; + ctx = &local_ctx; + } + + if (occ >= 0) { + /* search from the beginning */ + while (http_find_header(htx, hdr, ctx, 0)) { + occ--; + if (occ <= 0) { + *vptr = ctx->value.ptr; + *vlen = ctx->value.len; + return 1; + } + } + return 0; + } + + /* negative occurrence, we scan all the list then walk back */ + if (-occ > MAX_HDR_HISTORY) + return 0; + + found = hist_idx = 0; + while (http_find_header(htx, hdr, ctx, 0)) { + val_hist[hist_idx] = ctx->value; + if (++hist_idx >= MAX_HDR_HISTORY) + hist_idx = 0; + found++; + } + if (-occ > found) + return 0; + + /* OK now we have the last occurrence in [hist_idx-1], and we need to + * find occurrence -occ. 0 <= hist_idx < MAX_HDR_HISTORY, and we have + * -10 <= occ <= -1. So we have to check [hist_idx%MAX_HDR_HISTORY+occ] + * to remain in the 0..9 range. + */ + hist_idx += occ + MAX_HDR_HISTORY; + if (hist_idx >= MAX_HDR_HISTORY) + hist_idx -= MAX_HDR_HISTORY; + *vptr = val_hist[hist_idx].ptr; + *vlen = val_hist[hist_idx].len; + return 1; +} + +/* Return in <vptr> and <vlen> the pointer and length of occurrence <occ> of + * header whose name is <hname> of length <hlen>. If <ctx> is null, lookup is + * performed over the whole headers. Otherwise it must contain a valid header + * context, initialised with ctx->blk=NULL for the first lookup in a series. If + * <occ> is positive or null, occurrence #occ from the beginning (or last ctx) + * is returned. Occ #0 and #1 are equivalent. If <occ> is negative (and no less + * than -MAX_HDR_HISTORY), the occurrence is counted from the last one which is + * -1. This function differs from http_get_hdr() in that it only returns full + * line header values and does not stop at commas. + * The return value is 0 if nothing was found, or non-zero otherwise. + */ +unsigned int http_get_htx_fhdr(const struct htx *htx, const struct ist hdr, + int occ, struct http_hdr_ctx *ctx, char **vptr, size_t *vlen) +{ + struct http_hdr_ctx local_ctx; + struct ist val_hist[MAX_HDR_HISTORY]; + unsigned int hist_idx; + int found; + + if (!ctx) { + local_ctx.blk = NULL; + ctx = &local_ctx; + } + + if (occ >= 0) { + /* search from the beginning */ + while (http_find_header(htx, hdr, ctx, 1)) { + occ--; + if (occ <= 0) { + *vptr = ctx->value.ptr; + *vlen = ctx->value.len; + return 1; + } + } + return 0; + } + + /* negative occurrence, we scan all the list then walk back */ + if (-occ > MAX_HDR_HISTORY) + return 0; + + found = hist_idx = 0; + while (http_find_header(htx, hdr, ctx, 1)) { + val_hist[hist_idx] = ctx->value; + if (++hist_idx >= MAX_HDR_HISTORY) + hist_idx = 0; + found++; + } + if (-occ > found) + return 0; + + /* OK now we have the last occurrence in [hist_idx-1], and we need to + * find occurrence -occ. 0 <= hist_idx < MAX_HDR_HISTORY, and we have + * -10 <= occ <= -1. So we have to check [hist_idx%MAX_HDR_HISTORY+occ] + * to remain in the 0..9 range. + */ + hist_idx += occ + MAX_HDR_HISTORY; + if (hist_idx >= MAX_HDR_HISTORY) + hist_idx -= MAX_HDR_HISTORY; + *vptr = val_hist[hist_idx].ptr; + *vlen = val_hist[hist_idx].len; + return 1; +} + +int http_str_to_htx(struct buffer *buf, struct ist raw, char **errmsg) +{ + struct htx *htx; + struct htx_sl *sl; + struct h1m h1m; + struct http_hdr hdrs[global.tune.max_http_hdr]; + union h1_sl h1sl; + unsigned int flags = HTX_SL_F_IS_RESP; + int ret = 0; + + b_reset(buf); + if (!raw.len) { + buf->size = 0; + buf->area = NULL; + return 1; + } + + buf->size = global.tune.bufsize; + buf->area = malloc(buf->size); + if (!buf->area) + goto error; + + h1m_init_res(&h1m); + h1m.flags |= H1_MF_NO_PHDR; + ret = h1_headers_to_hdr_list(raw.ptr, istend(raw), + hdrs, sizeof(hdrs)/sizeof(hdrs[0]), &h1m, &h1sl); + if (ret <= 0) { + memprintf(errmsg, "unable to parse headers (error offset: %d)", h1m.err_pos); + goto error; + } + + if (unlikely(h1sl.st.v.len != 8)) { + memprintf(errmsg, "invalid http version (%.*s)", (int)h1sl.st.v.len, h1sl.st.v.ptr); + goto error; + } + if ((*(h1sl.st.v.ptr + 5) > '1') || + ((*(h1sl.st.v.ptr + 5) == '1') && (*(h1sl.st.v.ptr + 7) >= '1'))) + h1m.flags |= H1_MF_VER_11; + + if (h1sl.st.status < 200 && (h1sl.st.status == 100 || h1sl.st.status >= 102)) { + memprintf(errmsg, "invalid http status code for an error message (%u)", + h1sl.st.status); + goto error; + } + + if (h1sl.st.status == 204 || h1sl.st.status == 304) { + /* Responses known to have no body. */ + h1m.flags &= ~(H1_MF_CLEN|H1_MF_CHNK); + h1m.flags |= H1_MF_XFER_LEN; + h1m.curr_len = h1m.body_len = 0; + } + else if (h1m.flags & (H1_MF_CLEN|H1_MF_CHNK)) + h1m.flags |= H1_MF_XFER_LEN; + + if (h1m.flags & H1_MF_VER_11) + flags |= HTX_SL_F_VER_11; + if (h1m.flags & H1_MF_XFER_ENC) + flags |= HTX_SL_F_XFER_ENC; + if (h1m.flags & H1_MF_XFER_LEN) { + flags |= HTX_SL_F_XFER_LEN; + if (h1m.flags & H1_MF_CHNK) { + memprintf(errmsg, "chunk-encoded payload not supported"); + goto error; + } + else if (h1m.flags & H1_MF_CLEN) { + flags |= HTX_SL_F_CLEN; + if (h1m.body_len == 0) + flags |= HTX_SL_F_BODYLESS; + } + else + flags |= HTX_SL_F_BODYLESS; + } + + if ((flags & HTX_SL_F_BODYLESS) && raw.len > ret) { + memprintf(errmsg, "message payload not expected"); + goto error; + } + if ((flags & HTX_SL_F_CLEN) && h1m.body_len != (raw.len - ret)) { + memprintf(errmsg, "payload size does not match the announced content-length (%lu != %lu)", + (unsigned long)(raw.len - ret), (unsigned long)h1m.body_len); + goto error; + } + + htx = htx_from_buf(buf); + sl = htx_add_stline(htx, HTX_BLK_RES_SL, flags, h1sl.st.v, h1sl.st.c, h1sl.st.r); + if (!sl || !htx_add_all_headers(htx, hdrs)) { + memprintf(errmsg, "unable to add headers into the HTX message"); + goto error; + } + sl->info.res.status = h1sl.st.status; + + while (raw.len > ret) { + int sent = htx_add_data(htx, ist2(raw.ptr + ret, raw.len - ret)); + if (!sent) { + memprintf(errmsg, "unable to add payload into the HTX message"); + goto error; + } + ret += sent; + } + + htx->flags |= HTX_FL_EOM; + + return 1; + +error: + if (buf->size) + free(buf->area); + return 0; +} + +void release_http_reply(struct http_reply *http_reply) +{ + struct logformat_node *lf, *lfb; + struct http_reply_hdr *hdr, *hdrb; + + if (!http_reply) + return; + + ha_free(&http_reply->ctype); + list_for_each_entry_safe(hdr, hdrb, &http_reply->hdrs, list) { + LIST_DELETE(&hdr->list); + list_for_each_entry_safe(lf, lfb, &hdr->value, list) { + LIST_DELETE(&lf->list); + release_sample_expr(lf->expr); + free(lf->arg); + free(lf); + } + istfree(&hdr->name); + free(hdr); + } + + if (http_reply->type == HTTP_REPLY_ERRFILES) { + ha_free(&http_reply->body.http_errors); + } + else if (http_reply->type == HTTP_REPLY_RAW) + chunk_destroy(&http_reply->body.obj); + else if (http_reply->type == HTTP_REPLY_LOGFMT) { + list_for_each_entry_safe(lf, lfb, &http_reply->body.fmt, list) { + LIST_DELETE(&lf->list); + release_sample_expr(lf->expr); + free(lf->arg); + free(lf); + } + } + free(http_reply); +} + +static int http_htx_init(void) +{ + struct buffer chk; + struct ist raw; + char *errmsg = NULL; + int rc; + int err_code = 0; + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (!http_err_msgs[rc]) { + ha_alert("Internal error: no default message defined for HTTP return code %d", rc); + err_code |= ERR_ALERT | ERR_FATAL; + continue; + } + + raw = ist(http_err_msgs[rc]); + if (!http_str_to_htx(&chk, raw, &errmsg)) { + ha_alert("Internal error: invalid default message for HTTP return code %d: %s.\n", + http_err_codes[rc], errmsg); + err_code |= ERR_ALERT | ERR_FATAL; + } + else if (errmsg) { + ha_warning("invalid default message for HTTP return code %d: %s.\n", http_err_codes[rc], errmsg); + err_code |= ERR_WARN; + } + + /* Reset errmsg */ + ha_free(&errmsg); + + http_err_chunks[rc] = chk; + http_err_replies[rc].type = HTTP_REPLY_ERRMSG; + http_err_replies[rc].status = http_err_codes[rc]; + http_err_replies[rc].ctype = NULL; + LIST_INIT(&http_err_replies[rc].hdrs); + http_err_replies[rc].body.errmsg = &http_err_chunks[rc]; + } +end: + return err_code; +} + +static void http_htx_deinit(void) +{ + struct http_errors *http_errs, *http_errsb; + struct http_reply *http_rep, *http_repb; + struct ebpt_node *node, *next; + struct http_error_msg *http_errmsg; + int rc; + + node = ebpt_first(&http_error_messages); + while (node) { + next = ebpt_next(node); + ebpt_delete(node); + http_errmsg = container_of(node, typeof(*http_errmsg), node); + chunk_destroy(&http_errmsg->msg); + free(node->key); + free(http_errmsg); + node = next; + } + + list_for_each_entry_safe(http_errs, http_errsb, &http_errors_list, list) { + free(http_errs->conf.file); + free(http_errs->id); + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) + release_http_reply(http_errs->replies[rc]); + LIST_DELETE(&http_errs->list); + free(http_errs); + } + + list_for_each_entry_safe(http_rep, http_repb, &http_replies_list, list) { + LIST_DELETE(&http_rep->list); + release_http_reply(http_rep); + } + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) + chunk_destroy(&http_err_chunks[rc]); +} + +REGISTER_CONFIG_POSTPARSER("http_htx", http_htx_init); +REGISTER_POST_DEINIT(http_htx_deinit); + +/* Reads content of the error file <file> and convert it into an HTX message. On + * success, the HTX message is returned. On error, NULL is returned and an error + * message is written into the <errmsg> buffer. + */ +struct buffer *http_load_errorfile(const char *file, char **errmsg) +{ + struct buffer *buf = NULL; + struct buffer chk; + struct ebpt_node *node; + struct http_error_msg *http_errmsg; + struct stat stat; + char *err = NULL; + int errnum, errlen; + int fd = -1; + + /* already loaded */ + node = ebis_lookup_len(&http_error_messages, file, strlen(file)); + if (node) { + http_errmsg = container_of(node, typeof(*http_errmsg), node); + buf = &http_errmsg->msg; + goto out; + } + + /* Read the error file content */ + fd = open(file, O_RDONLY); + if ((fd < 0) || (fstat(fd, &stat) < 0)) { + memprintf(errmsg, "error opening file '%s'.", file); + goto out; + } + + if (stat.st_size <= global.tune.bufsize) + errlen = stat.st_size; + else { + ha_warning("custom error message file '%s' larger than %d bytes. Truncating.\n", + file, global.tune.bufsize); + errlen = global.tune.bufsize; + } + + err = malloc(errlen); + if (!err) { + memprintf(errmsg, "out of memory."); + goto out; + } + + errnum = read(fd, err, errlen); + if (errnum != errlen) { + memprintf(errmsg, "error reading file '%s'.", file); + goto out; + } + + /* Create the node corresponding to the error file */ + http_errmsg = calloc(1, sizeof(*http_errmsg)); + if (!http_errmsg) { + memprintf(errmsg, "out of memory."); + goto out; + } + http_errmsg->node.key = strdup(file); + if (!http_errmsg->node.key) { + memprintf(errmsg, "out of memory."); + free(http_errmsg); + goto out; + } + + /* Convert the error file into an HTX message */ + if (!http_str_to_htx(&chk, ist2(err, errlen), errmsg)) { + memprintf(errmsg, "'%s': %s", file, *errmsg); + free(http_errmsg->node.key); + free(http_errmsg); + goto out; + } + + /* Insert the node in the tree and return the HTX message */ + http_errmsg->msg = chk; + ebis_insert(&http_error_messages, &http_errmsg->node); + buf = &http_errmsg->msg; + + out: + if (fd >= 0) + close(fd); + free(err); + return buf; +} + +/* Convert the raw http message <msg> into an HTX message. On success, the HTX + * message is returned. On error, NULL is returned and an error message is + * written into the <errmsg> buffer. + */ +struct buffer *http_load_errormsg(const char *key, const struct ist msg, char **errmsg) +{ + struct buffer *buf = NULL; + struct buffer chk; + struct ebpt_node *node; + struct http_error_msg *http_errmsg; + + /* already loaded */ + node = ebis_lookup_len(&http_error_messages, key, strlen(key)); + if (node) { + http_errmsg = container_of(node, typeof(*http_errmsg), node); + buf = &http_errmsg->msg; + goto out; + } + /* Create the node corresponding to the error file */ + http_errmsg = calloc(1, sizeof(*http_errmsg)); + if (!http_errmsg) { + memprintf(errmsg, "out of memory."); + goto out; + } + http_errmsg->node.key = strdup(key); + if (!http_errmsg->node.key) { + memprintf(errmsg, "out of memory."); + free(http_errmsg); + goto out; + } + + /* Convert the error file into an HTX message */ + if (!http_str_to_htx(&chk, msg, errmsg)) { + memprintf(errmsg, "invalid error message: %s", *errmsg); + free(http_errmsg->node.key); + free(http_errmsg); + goto out; + } + + /* Insert the node in the tree and return the HTX message */ + http_errmsg->msg = chk; + ebis_insert(&http_error_messages, &http_errmsg->node); + buf = &http_errmsg->msg; + out: + return buf; +} + +/* This function parses the raw HTTP error file <file> for the status code + * <status>. It returns NULL if there is any error, otherwise it return the + * corresponding HTX message. + */ +struct buffer *http_parse_errorfile(int status, const char *file, char **errmsg) +{ + struct buffer *buf = NULL; + int rc; + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (http_err_codes[rc] == status) { + buf = http_load_errorfile(file, errmsg); + break; + } + } + + if (rc >= HTTP_ERR_SIZE) + memprintf(errmsg, "status code '%d' not handled.", status); + return buf; +} + +/* This function creates HTX error message corresponding to a redirect message + * for the status code <status>. <url> is used as location url for the + * redirect. <errloc> is used to know if it is a 302 or a 303 redirect. It + * returns NULL if there is any error, otherwise it return the corresponding HTX + * message. + */ +struct buffer *http_parse_errorloc(int errloc, int status, const char *url, char **errmsg) +{ + static const char *HTTP_302 = + "HTTP/1.1 302 Found\r\n" + "Cache-Control: no-cache\r\n" + "Content-length: 0\r\n" + "Location: "; /* not terminated since it will be concatenated with the URL */ + static const char *HTTP_303 = + "HTTP/1.1 303 See Other\r\n" + "Cache-Control: no-cache\r\n" + "Content-length: 0\r\n" + "Location: "; /* not terminated since it will be concatenated with the URL */ + + struct buffer *buf = NULL; + const char *msg; + char *key = NULL, *err = NULL; + int rc, errlen; + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (http_err_codes[rc] == status) { + /* Create the error key */ + if (!memprintf(&key, "errorloc%d %s", errloc, url)) { + memprintf(errmsg, "out of memory."); + goto out; + } + /* Create the error message */ + msg = (errloc == 302 ? HTTP_302 : HTTP_303); + errlen = strlen(msg) + strlen(url) + 5; + err = malloc(errlen); + if (!err) { + memprintf(errmsg, "out of memory."); + goto out; + } + errlen = snprintf(err, errlen, "%s%s\r\n\r\n", msg, url); + + /* Load it */ + buf = http_load_errormsg(key, ist2(err, errlen), errmsg); + break; + } + } + + if (rc >= HTTP_ERR_SIZE) + memprintf(errmsg, "status code '%d' not handled.", status); +out: + free(key); + free(err); + return buf; +} + +/* Check an "http reply" and, for replies referencing an http-errors section, + * try to find the right section and the right error message in this section. If + * found, the reply is updated. If the http-errors section exists but the error + * message is not found, no error message is set to fallback on the default + * ones. Otherwise (unknown section) an error is returned. + * + * The function returns 1 in success case, otherwise, it returns 0 and errmsg is + * filled. + */ +int http_check_http_reply(struct http_reply *reply, struct proxy *px, char **errmsg) +{ + struct http_errors *http_errs; + int ret = 1; + + if (reply->type != HTTP_REPLY_ERRFILES) + goto end; + + list_for_each_entry(http_errs, &http_errors_list, list) { + if (strcmp(http_errs->id, reply->body.http_errors) == 0) { + reply->type = HTTP_REPLY_INDIRECT; + free(reply->body.http_errors); + reply->body.reply = http_errs->replies[http_get_status_idx(reply->status)]; + if (!reply->body.reply) + ha_warning("Proxy '%s': status '%d' referenced by an http reply " + "not declared in http-errors section '%s'.\n", + px->id, reply->status, http_errs->id); + break; + } + } + + if (&http_errs->list == &http_errors_list) { + memprintf(errmsg, "unknown http-errors section '%s' referenced by an http reply ", + reply->body.http_errors); + ret = 0; + } + + end: + return ret; +} + +/* Parse an "http reply". It returns the reply on success or NULL on error. This + * function creates one of the following http replies : + * + * - HTTP_REPLY_EMPTY : dummy response, no payload + * - HTTP_REPLY_ERRMSG : implicit error message depending on the status code or explicit one + * - HTTP_REPLY_ERRFILES : points on an http-errors section (resolved during post-parsing) + * - HTTP_REPLY_RAW : explicit file object ('file' argument) + * - HTTP_REPLY_LOGFMT : explicit log-format string ('content' argument) + * + * The content-type must be defined for non-empty payload. It is ignored for + * error messages (implicit or explicit). When an http-errors section is + * referenced (HTTP_REPLY_ERRFILES), the real error message should be resolved + * during the configuration validity check or dynamically. It is the caller + * responsibility to choose. If no status code is configured, <default_status> + * is set. + */ +struct http_reply *http_parse_http_reply(const char **args, int *orig_arg, struct proxy *px, + int default_status, char **errmsg) +{ + struct logformat_node *lf, *lfb; + struct http_reply *reply = NULL; + struct http_reply_hdr *hdr, *hdrb; + struct stat stat; + const char *act_arg = NULL; + char *obj = NULL; + int cur_arg, cap = 0, objlen = 0, fd = -1; + + + reply = calloc(1, sizeof(*reply)); + if (!reply) { + memprintf(errmsg, "out of memory"); + goto error; + } + LIST_INIT(&reply->hdrs); + reply->type = HTTP_REPLY_EMPTY; + reply->status = default_status; + + if (px->conf.args.ctx == ARGC_HERR) + cap = (SMP_VAL_REQUEST | SMP_VAL_RESPONSE); + else { + if (px->cap & PR_CAP_FE) + cap |= ((px->conf.args.ctx == ARGC_HRQ) ? SMP_VAL_FE_HRQ_HDR : SMP_VAL_FE_HRS_HDR); + if (px->cap & PR_CAP_BE) + cap |= ((px->conf.args.ctx == ARGC_HRQ) ? SMP_VAL_BE_HRQ_HDR : SMP_VAL_BE_HRS_HDR); + } + + cur_arg = *orig_arg; + while (*args[cur_arg]) { + if (strcmp(args[cur_arg], "status") == 0) { + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <status_code> as argument", args[cur_arg-1]); + goto error; + } + reply->status = atol(args[cur_arg]); + if (reply->status < 200 || reply->status > 599) { + memprintf(errmsg, "Unexpected status code '%d'", reply->status); + goto error; + } + cur_arg++; + } + else if (strcmp(args[cur_arg], "content-type") == 0) { + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <ctype> as argument", args[cur_arg-1]); + goto error; + } + free(reply->ctype); + reply->ctype = strdup(args[cur_arg]); + cur_arg++; + } + else if (strcmp(args[cur_arg], "errorfiles") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <name> as argument", args[cur_arg-1]); + goto error; + } + reply->body.http_errors = strdup(args[cur_arg]); + if (!reply->body.http_errors) { + memprintf(errmsg, "out of memory"); + goto error; + } + reply->type = HTTP_REPLY_ERRFILES; + cur_arg++; + } + else if (strcmp(args[cur_arg], "default-errorfiles") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + reply->type = HTTP_REPLY_ERRMSG; + cur_arg++; + } + else if (strcmp(args[cur_arg], "errorfile") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <fmt> as argument", args[cur_arg-1]); + goto error; + } + reply->body.errmsg = http_load_errorfile(args[cur_arg], errmsg); + if (!reply->body.errmsg) { + goto error; + } + reply->type = HTTP_REPLY_ERRMSG; + cur_arg++; + } + else if (strcmp(args[cur_arg], "file") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <file> as argument", args[cur_arg-1]); + goto error; + } + fd = open(args[cur_arg], O_RDONLY); + if ((fd < 0) || (fstat(fd, &stat) < 0)) { + memprintf(errmsg, "error opening file '%s'", args[cur_arg]); + goto error; + } + if (stat.st_size > global.tune.bufsize) { + memprintf(errmsg, "file '%s' exceeds the buffer size (%lld > %d)", + args[cur_arg], (long long)stat.st_size, global.tune.bufsize); + goto error; + } + objlen = stat.st_size; + obj = malloc(objlen); + if (!obj || read(fd, obj, objlen) != objlen) { + memprintf(errmsg, "error reading file '%s'", args[cur_arg]); + goto error; + } + close(fd); + fd = -1; + reply->type = HTTP_REPLY_RAW; + chunk_initlen(&reply->body.obj, obj, global.tune.bufsize, objlen); + obj = NULL; + cur_arg++; + } + else if (strcmp(args[cur_arg], "string") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <str> as argument", args[cur_arg-1]); + goto error; + } + obj = strdup(args[cur_arg]); + objlen = strlen(args[cur_arg]); + if (!obj) { + memprintf(errmsg, "out of memory"); + goto error; + } + reply->type = HTTP_REPLY_RAW; + chunk_initlen(&reply->body.obj, obj, global.tune.bufsize, objlen); + obj = NULL; + cur_arg++; + } + else if (strcmp(args[cur_arg], "lf-file") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <file> as argument", args[cur_arg-1]); + goto error; + } + fd = open(args[cur_arg], O_RDONLY); + if ((fd < 0) || (fstat(fd, &stat) < 0)) { + memprintf(errmsg, "error opening file '%s'", args[cur_arg]); + goto error; + } + if (stat.st_size > global.tune.bufsize) { + memprintf(errmsg, "file '%s' exceeds the buffer size (%lld > %d)", + args[cur_arg], (long long)stat.st_size, global.tune.bufsize); + goto error; + } + objlen = stat.st_size; + obj = malloc(objlen + 1); + if (!obj || read(fd, obj, objlen) != objlen) { + memprintf(errmsg, "error reading file '%s'", args[cur_arg]); + goto error; + } + close(fd); + fd = -1; + obj[objlen] = '\0'; + reply->type = HTTP_REPLY_LOGFMT; + LIST_INIT(&reply->body.fmt); + cur_arg++; + } + else if (strcmp(args[cur_arg], "lf-string") == 0) { + if (reply->type != HTTP_REPLY_EMPTY) { + memprintf(errmsg, "unexpected '%s' argument, '%s' already defined", args[cur_arg], act_arg); + goto error; + } + act_arg = args[cur_arg]; + cur_arg++; + if (!*args[cur_arg]) { + memprintf(errmsg, "'%s' expects <fmt> as argument", args[cur_arg-1]); + goto error; + } + obj = strdup(args[cur_arg]); + objlen = strlen(args[cur_arg]); + reply->type = HTTP_REPLY_LOGFMT; + LIST_INIT(&reply->body.fmt); + cur_arg++; + } + else if (strcmp(args[cur_arg], "hdr") == 0) { + cur_arg++; + if (!*args[cur_arg] || !*args[cur_arg+1]) { + memprintf(errmsg, "'%s' expects <name> and <value> as arguments", args[cur_arg-1]); + goto error; + } + if (strcasecmp(args[cur_arg], "content-length") == 0 || + strcasecmp(args[cur_arg], "transfer-encoding") == 0 || + strcasecmp(args[cur_arg], "content-type") == 0) { + ha_warning("parsing [%s:%d] : header '%s' always ignored by the http reply.\n", + px->conf.args.file, px->conf.args.line, args[cur_arg]); + cur_arg += 2; + continue; + } + hdr = calloc(1, sizeof(*hdr)); + if (!hdr) { + memprintf(errmsg, "'%s' : out of memory", args[cur_arg-1]); + goto error; + } + LIST_APPEND(&reply->hdrs, &hdr->list); + LIST_INIT(&hdr->value); + hdr->name = ist(strdup(args[cur_arg])); + if (!isttest(hdr->name)) { + memprintf(errmsg, "out of memory"); + goto error; + } + if (!parse_logformat_string(args[cur_arg+1], px, &hdr->value, LOG_OPT_HTTP, cap, errmsg)) + goto error; + + free(px->conf.lfs_file); + px->conf.lfs_file = strdup(px->conf.args.file); + px->conf.lfs_line = px->conf.args.line; + cur_arg += 2; + } + else + break; + } + + if (reply->type == HTTP_REPLY_EMPTY) { /* no payload */ + if (reply->ctype) { + ha_warning("parsing [%s:%d] : content-type '%s' ignored by the http reply because" + " neither errorfile nor payload defined.\n", + px->conf.args.file, px->conf.args.line, reply->ctype); + ha_free(&reply->ctype); + } + } + else if (reply->type == HTTP_REPLY_ERRFILES || reply->type == HTTP_REPLY_ERRMSG) { /* errorfiles or errorfile */ + + if (reply->type != HTTP_REPLY_ERRMSG || !reply->body.errmsg) { + /* default errorfile or errorfiles: check the status */ + int rc; + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (http_err_codes[rc] == reply->status) + break; + } + + if (rc >= HTTP_ERR_SIZE) { + memprintf(errmsg, "status code '%d' not handled by default with '%s' argument.", + reply->status, act_arg); + goto error; + } + } + + if (reply->ctype) { + ha_warning("parsing [%s:%d] : content-type '%s' ignored by the http reply when used " + "with an erorrfile.\n", + px->conf.args.file, px->conf.args.line, reply->ctype); + ha_free(&reply->ctype); + } + if (!LIST_ISEMPTY(&reply->hdrs)) { + ha_warning("parsing [%s:%d] : hdr parameters ignored by the http reply when used " + "with an erorrfile.\n", + px->conf.args.file, px->conf.args.line); + list_for_each_entry_safe(hdr, hdrb, &reply->hdrs, list) { + LIST_DELETE(&hdr->list); + list_for_each_entry_safe(lf, lfb, &hdr->value, list) { + LIST_DELETE(&lf->list); + release_sample_expr(lf->expr); + free(lf->arg); + free(lf); + } + istfree(&hdr->name); + free(hdr); + } + } + } + else if (reply->type == HTTP_REPLY_RAW) { /* explicit parameter using 'file' parameter*/ + if ((reply->status == 204 || reply->status == 304) && objlen) { + memprintf(errmsg, "No body expected for %d responses", reply->status); + goto error; + } + if (!reply->ctype && objlen) { + memprintf(errmsg, "a content type must be defined when non-empty payload is configured"); + goto error; + } + if (reply->ctype && !b_data(&reply->body.obj)) { + ha_warning("parsing [%s:%d] : content-type '%s' ignored by the http reply when used " + "with an empty payload.\n", + px->conf.args.file, px->conf.args.line, reply->ctype); + ha_free(&reply->ctype); + } + if (b_room(&reply->body.obj) < global.tune.maxrewrite) { + ha_warning("parsing [%s:%d] : http reply payload runs over the buffer space reserved to headers rewriting." + " It may lead to internal errors if strict rewriting mode is enabled.\n", + px->conf.args.file, px->conf.args.line); + } + } + else if (reply->type == HTTP_REPLY_LOGFMT) { /* log-format payload using 'lf-file' of 'lf-string' parameter */ + LIST_INIT(&reply->body.fmt); + if ((reply->status == 204 || reply->status == 304)) { + memprintf(errmsg, "No body expected for %d responses", reply->status); + goto error; + } + if (!reply->ctype) { + memprintf(errmsg, "a content type must be defined with a log-format payload"); + goto error; + } + if (!parse_logformat_string(obj, px, &reply->body.fmt, LOG_OPT_HTTP, cap, errmsg)) + goto error; + + free(px->conf.lfs_file); + px->conf.lfs_file = strdup(px->conf.args.file); + px->conf.lfs_line = px->conf.args.line; + } + + free(obj); + *orig_arg = cur_arg; + return reply; + + error: + free(obj); + if (fd >= 0) + close(fd); + release_http_reply(reply); + return NULL; +} + +/* Apply schemed-based normalization as described on rfc3986 on section 6.3.2. + * Returns 0 if no error has been found else non-zero. + * + * The normalization is processed on the target-uri at the condition that it is + * in absolute-form. In the case where the target-uri was normalized, every + * host headers values found are also replaced by the normalized hostname. This + * assumes that the target-uri and host headers were properly identify as + * similar before calling this function. + */ +int http_scheme_based_normalize(struct htx *htx) +{ + struct http_hdr_ctx ctx; + struct htx_sl *sl; + struct ist uri, scheme, authority, host, port; + struct http_uri_parser parser; + + sl = http_get_stline(htx); + + if (!sl || !(sl->flags & (HTX_SL_F_HAS_SCHM|HTX_SL_F_HAS_AUTHORITY))) + return 0; + + uri = htx_sl_req_uri(sl); + + parser = http_uri_parser_init(uri); + scheme = http_parse_scheme(&parser); + /* if no scheme found, no normalization to proceed */ + if (!isttest(scheme)) + return 0; + + /* Extract the port if present in authority */ + authority = http_parse_authority(&parser, 1); + port = http_get_host_port(authority); + if (!isttest(port)) { + /* if no port found, no normalization to proceed */ + return 0; + } + host = isttrim(authority, istlen(authority) - istlen(port) - 1); + + if (http_is_default_port(scheme, port)) { + /* reconstruct the uri with removal of the port */ + struct buffer *temp = get_trash_chunk(); + struct ist meth, vsn; + + /* meth */ + chunk_memcat(temp, HTX_SL_REQ_MPTR(sl), HTX_SL_REQ_MLEN(sl)); + meth = ist2(temp->area, HTX_SL_REQ_MLEN(sl)); + + /* vsn */ + chunk_memcat(temp, HTX_SL_REQ_VPTR(sl), HTX_SL_REQ_VLEN(sl)); + vsn = ist2(temp->area + meth.len, HTX_SL_REQ_VLEN(sl)); + + /* reconstruct uri without port */ + chunk_memcat(temp, uri.ptr, authority.ptr - uri.ptr); + chunk_istcat(temp, host); + chunk_memcat(temp, istend(authority), istend(uri) - istend(authority)); + uri = ist2(temp->area + meth.len + vsn.len, host.len + uri.len - authority.len); /* uri */ + + http_replace_stline(htx, meth, uri, vsn); + + /* replace every host headers values by the normalized host */ + ctx.blk = NULL; + while (http_find_header(htx, ist("host"), &ctx, 0)) { + if (!http_replace_header_value(htx, &ctx, host)) + goto fail; + } + } + + return 0; + + fail: + return 1; +} + +/* First step function to merge multiple cookie headers in a single entry. + * + * Use it for each cookie header at <idx> index over HTTP headers in <list>. + * <first> and <last> are state variables used internally and must be + * initialized to -1 before the first invocation. + */ +void http_cookie_register(struct http_hdr *list, int idx, int *first, int *last) +{ + /* Build a linked list of cookie headers. Use header length to point to + * the next one. The last entry will contains -1. + */ + + /* Caller is responsible to initialize *first and *last to -1 on first + * invocation. Both will thus be set to a valid index after it. + */ + BUG_ON(*first > 0 && *last < 0); + + /* Mark the current end of cookie linked list. */ + list[idx].n.len = -1; + if (*first < 0) { + /* Save first found cookie for http_cookie_merge call. */ + *first = idx; + } + else { + /* Update linked list of cookies. */ + list[*last].n.len = idx; + } + + *last = idx; +} + +/* Second step to merge multiple cookie headers in a single entry. + * + * Use it when looping over HTTP headers is done and <htx> message is built. + * This will concatenate each cookie headers present from <list> directly into + * <htx> message. <first> is reused from previous http_cookie_register + * invocation. + * + * Returns 0 on success else non-zero. + */ +int http_cookie_merge(struct htx *htx, struct http_hdr *list, int first) +{ + uint32_t fs; /* free space */ + uint32_t bs; /* block size */ + uint32_t vl; /* value len */ + uint32_t tl; /* total length */ + struct htx_blk *blk; + + if (first < 0) + return 0; + + blk = htx_add_header(htx, ist("cookie"), list[first].v); + if (!blk) + return 1; + + tl = list[first].v.len; + fs = htx_free_data_space(htx); + bs = htx_get_blksz(blk); + + /* for each extra cookie, we'll extend the cookie's value and insert + * ";" before the new value. + */ + fs += tl; /* first one is already counted */ + + /* Loop over cookies linked list built from http_cookie_register. */ + while ((first = list[first].n.len) >= 0) { + vl = list[first].v.len; + tl += vl + 2; + if (tl > fs) + return 1; + + htx_change_blk_value_len(htx, blk, tl); + *(char *)(htx_get_blk_ptr(htx, blk) + bs + 0) = ';'; + *(char *)(htx_get_blk_ptr(htx, blk) + bs + 1) = ' '; + memcpy(htx_get_blk_ptr(htx, blk) + bs + 2, + list[first].v.ptr, vl); + bs += vl + 2; + } + + return 0; +} + +/* Parses the "errorloc[302|303]" proxy keyword */ +static int proxy_parse_errorloc(char **args, int section, struct proxy *curpx, + const struct proxy *defpx, const char *file, int line, + char **errmsg) +{ + struct conf_errors *conf_err; + struct http_reply *reply; + struct buffer *msg; + int errloc, status; + int ret = 0; + + if (warnifnotcap(curpx, PR_CAP_FE | PR_CAP_BE, file, line, args[0], NULL)) { + ret = 1; + goto out; + } + + if (*(args[1]) == 0 || *(args[2]) == 0) { + memprintf(errmsg, "%s : expects <status_code> and <url> as arguments.\n", args[0]); + ret = -1; + goto out; + } + + status = atol(args[1]); + errloc = (strcmp(args[0], "errorloc303") == 0 ? 303 : 302); + msg = http_parse_errorloc(errloc, status, args[2], errmsg); + if (!msg) { + memprintf(errmsg, "%s : %s", args[0], *errmsg); + ret = -1; + goto out; + } + + reply = calloc(1, sizeof(*reply)); + if (!reply) { + memprintf(errmsg, "%s : out of memory.", args[0]); + ret = -1; + goto out; + } + reply->type = HTTP_REPLY_ERRMSG; + reply->status = status; + reply->ctype = NULL; + LIST_INIT(&reply->hdrs); + reply->body.errmsg = msg; + LIST_APPEND(&http_replies_list, &reply->list); + + conf_err = calloc(1, sizeof(*conf_err)); + if (!conf_err) { + memprintf(errmsg, "%s : out of memory.", args[0]); + free(reply); + ret = -1; + goto out; + } + conf_err->type = 1; + conf_err->info.errorfile.status = status; + conf_err->info.errorfile.reply = reply; + + conf_err->file = strdup(file); + conf_err->line = line; + LIST_APPEND(&curpx->conf.errors, &conf_err->list); + + /* handle warning message */ + if (*errmsg) + ret = 1; + out: + return ret; + +} + +/* Parses the "errorfile" proxy keyword */ +static int proxy_parse_errorfile(char **args, int section, struct proxy *curpx, + const struct proxy *defpx, const char *file, int line, + char **errmsg) +{ + struct conf_errors *conf_err; + struct http_reply *reply; + struct buffer *msg; + int status; + int ret = 0; + + if (warnifnotcap(curpx, PR_CAP_FE | PR_CAP_BE, file, line, args[0], NULL)) { + ret = 1; + goto out; + } + + if (*(args[1]) == 0 || *(args[2]) == 0) { + memprintf(errmsg, "%s : expects <status_code> and <file> as arguments.\n", args[0]); + ret = -1; + goto out; + } + + status = atol(args[1]); + msg = http_parse_errorfile(status, args[2], errmsg); + if (!msg) { + memprintf(errmsg, "%s : %s", args[0], *errmsg); + ret = -1; + goto out; + } + + reply = calloc(1, sizeof(*reply)); + if (!reply) { + memprintf(errmsg, "%s : out of memory.", args[0]); + ret = -1; + goto out; + } + reply->type = HTTP_REPLY_ERRMSG; + reply->status = status; + reply->ctype = NULL; + LIST_INIT(&reply->hdrs); + reply->body.errmsg = msg; + LIST_APPEND(&http_replies_list, &reply->list); + + conf_err = calloc(1, sizeof(*conf_err)); + if (!conf_err) { + memprintf(errmsg, "%s : out of memory.", args[0]); + free(reply); + ret = -1; + goto out; + } + conf_err->type = 1; + conf_err->info.errorfile.status = status; + conf_err->info.errorfile.reply = reply; + conf_err->file = strdup(file); + conf_err->line = line; + LIST_APPEND(&curpx->conf.errors, &conf_err->list); + + /* handle warning message */ + if (*errmsg) + ret = 1; + out: + return ret; + +} + +/* Parses the "errorfiles" proxy keyword */ +static int proxy_parse_errorfiles(char **args, int section, struct proxy *curpx, + const struct proxy *defpx, const char *file, int line, + char **err) +{ + struct conf_errors *conf_err = NULL; + char *name = NULL; + int rc, ret = 0; + + if (warnifnotcap(curpx, PR_CAP_FE | PR_CAP_BE, file, line, args[0], NULL)) { + ret = 1; + goto out; + } + + if (!*(args[1])) { + memprintf(err, "%s : expects <name> as argument.", args[0]); + ret = -1; + goto out; + } + + name = strdup(args[1]); + conf_err = calloc(1, sizeof(*conf_err)); + if (!name || !conf_err) { + memprintf(err, "%s : out of memory.", args[0]); + goto error; + } + conf_err->type = 0; + + conf_err->info.errorfiles.name = name; + if (!*(args[2])) { + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) + conf_err->info.errorfiles.status[rc] = 1; + } + else { + int cur_arg, status; + for (cur_arg = 2; *(args[cur_arg]); cur_arg++) { + status = atol(args[cur_arg]); + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (http_err_codes[rc] == status) { + conf_err->info.errorfiles.status[rc] = 2; + break; + } + } + if (rc >= HTTP_ERR_SIZE) { + memprintf(err, "%s : status code '%d' not handled.", args[0], status); + goto error; + } + } + } + conf_err->file = strdup(file); + conf_err->line = line; + LIST_APPEND(&curpx->conf.errors, &conf_err->list); + out: + return ret; + + error: + free(name); + free(conf_err); + ret = -1; + goto out; +} + +/* Parses the "http-error" proxy keyword */ +static int proxy_parse_http_error(char **args, int section, struct proxy *curpx, + const struct proxy *defpx, const char *file, int line, + char **errmsg) +{ + struct conf_errors *conf_err; + struct http_reply *reply = NULL; + int rc, cur_arg, ret = 0; + + if (warnifnotcap(curpx, PR_CAP_FE | PR_CAP_BE, file, line, args[0], NULL)) { + ret = 1; + goto out; + } + + cur_arg = 1; + curpx->conf.args.ctx = ARGC_HERR; + reply = http_parse_http_reply((const char **)args, &cur_arg, curpx, 0, errmsg); + if (!reply) { + memprintf(errmsg, "%s : %s", args[0], *errmsg); + goto error; + } + else if (!reply->status) { + memprintf(errmsg, "%s : expects at least a <status> as arguments.\n", args[0]); + goto error; + } + + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (http_err_codes[rc] == reply->status) + break; + } + + if (rc >= HTTP_ERR_SIZE) { + memprintf(errmsg, "%s: status code '%d' not handled.", args[0], reply->status); + goto error; + } + if (*args[cur_arg]) { + memprintf(errmsg, "%s : unknown keyword '%s'.", args[0], args[cur_arg]); + goto error; + } + + conf_err = calloc(1, sizeof(*conf_err)); + if (!conf_err) { + memprintf(errmsg, "%s : out of memory.", args[0]); + goto error; + } + if (reply->type == HTTP_REPLY_ERRFILES) { + int rc = http_get_status_idx(reply->status); + + conf_err->type = 2; + conf_err->info.errorfiles.name = reply->body.http_errors; + conf_err->info.errorfiles.status[rc] = 2; + reply->body.http_errors = NULL; + release_http_reply(reply); + } + else { + conf_err->type = 1; + conf_err->info.errorfile.status = reply->status; + conf_err->info.errorfile.reply = reply; + LIST_APPEND(&http_replies_list, &reply->list); + } + conf_err->file = strdup(file); + conf_err->line = line; + LIST_APPEND(&curpx->conf.errors, &conf_err->list); + + /* handle warning message */ + if (*errmsg) + ret = 1; + out: + return ret; + + error: + release_http_reply(reply); + ret = -1; + goto out; + +} + +/* Check "errorfiles" proxy keyword */ +static int proxy_check_errors(struct proxy *px) +{ + struct conf_errors *conf_err, *conf_err_back; + struct http_errors *http_errs; + int rc, err = ERR_NONE; + + list_for_each_entry_safe(conf_err, conf_err_back, &px->conf.errors, list) { + if (conf_err->type == 1) { + /* errorfile */ + rc = http_get_status_idx(conf_err->info.errorfile.status); + px->replies[rc] = conf_err->info.errorfile.reply; + + /* For proxy, to rely on default replies, just don't reference a reply */ + if (px->replies[rc]->type == HTTP_REPLY_ERRMSG && !px->replies[rc]->body.errmsg) + px->replies[rc] = NULL; + } + else { + /* errorfiles */ + list_for_each_entry(http_errs, &http_errors_list, list) { + if (strcmp(http_errs->id, conf_err->info.errorfiles.name) == 0) + break; + } + + /* unknown http-errors section */ + if (&http_errs->list == &http_errors_list) { + ha_alert("proxy '%s': unknown http-errors section '%s' (at %s:%d).\n", + px->id, conf_err->info.errorfiles.name, conf_err->file, conf_err->line); + err |= ERR_ALERT | ERR_FATAL; + free(conf_err->info.errorfiles.name); + goto next; + } + + free(conf_err->info.errorfiles.name); + for (rc = 0; rc < HTTP_ERR_SIZE; rc++) { + if (conf_err->info.errorfiles.status[rc] > 0) { + if (http_errs->replies[rc]) + px->replies[rc] = http_errs->replies[rc]; + else if (conf_err->info.errorfiles.status[rc] == 2) + ha_warning("config: proxy '%s' : status '%d' not declared in" + " http-errors section '%s' (at %s:%d).\n", + px->id, http_err_codes[rc], http_errs->id, + conf_err->file, conf_err->line); + } + } + } + next: + LIST_DELETE(&conf_err->list); + free(conf_err->file); + free(conf_err); + } + + out: + return err; +} + +static int post_check_errors() +{ + struct ebpt_node *node; + struct http_error_msg *http_errmsg; + struct htx *htx; + int err_code = ERR_NONE; + + node = ebpt_first(&http_error_messages); + while (node) { + http_errmsg = container_of(node, typeof(*http_errmsg), node); + if (b_is_null(&http_errmsg->msg)) + goto next; + htx = htxbuf(&http_errmsg->msg); + if (htx_free_data_space(htx) < global.tune.maxrewrite) { + ha_warning("config: errorfile '%s' runs over the buffer space" + " reserved to headers rewriting. It may lead to internal errors if " + " http-after-response rules are evaluated on this message.\n", + (char *)node->key); + err_code |= ERR_WARN; + } + next: + node = ebpt_next(node); + } + + return err_code; +} + +int proxy_dup_default_conf_errors(struct proxy *curpx, const struct proxy *defpx, char **errmsg) +{ + struct conf_errors *conf_err, *new_conf_err = NULL; + int ret = 0; + + list_for_each_entry(conf_err, &defpx->conf.errors, list) { + new_conf_err = calloc(1, sizeof(*new_conf_err)); + if (!new_conf_err) { + memprintf(errmsg, "unable to duplicate default errors (out of memory)."); + goto out; + } + new_conf_err->type = conf_err->type; + if (conf_err->type == 1) { + new_conf_err->info.errorfile.status = conf_err->info.errorfile.status; + new_conf_err->info.errorfile.reply = conf_err->info.errorfile.reply; + } + else { + new_conf_err->info.errorfiles.name = strdup(conf_err->info.errorfiles.name); + if (!new_conf_err->info.errorfiles.name) { + memprintf(errmsg, "unable to duplicate default errors (out of memory)."); + goto out; + } + memcpy(&new_conf_err->info.errorfiles.status, &conf_err->info.errorfiles.status, + sizeof(conf_err->info.errorfiles.status)); + } + new_conf_err->file = strdup(conf_err->file); + new_conf_err->line = conf_err->line; + LIST_APPEND(&curpx->conf.errors, &new_conf_err->list); + new_conf_err = NULL; + } + ret = 1; + + out: + free(new_conf_err); + return ret; +} + +void proxy_release_conf_errors(struct proxy *px) +{ + struct conf_errors *conf_err, *conf_err_back; + + list_for_each_entry_safe(conf_err, conf_err_back, &px->conf.errors, list) { + if (conf_err->type == 0) + free(conf_err->info.errorfiles.name); + LIST_DELETE(&conf_err->list); + free(conf_err->file); + free(conf_err); + } +} + +/* + * Parse an <http-errors> section. + * Returns the error code, 0 if OK, or any combination of : + * - ERR_ABORT: must abort ASAP + * - ERR_FATAL: we can continue parsing but not start the service + * - ERR_WARN: a warning has been emitted + * - ERR_ALERT: an alert has been emitted + * Only the two first ones can stop processing, the two others are just + * indicators. + */ +static int cfg_parse_http_errors(const char *file, int linenum, char **args, int kwm) +{ + static struct http_errors *curr_errs = NULL; + int err_code = 0; + const char *err; + char *errmsg = NULL; + + if (strcmp(args[0], "http-errors") == 0) { /* new errors section */ + if (!*args[1]) { + ha_alert("parsing [%s:%d] : missing name for http-errors section.\n", file, linenum); + err_code |= ERR_ALERT | ERR_ABORT; + goto out; + } + + err = invalid_char(args[1]); + if (err) { + ha_alert("parsing [%s:%d] : character '%c' is not permitted in '%s' name '%s'.\n", + file, linenum, *err, args[0], args[1]); + err_code |= ERR_ALERT | ERR_FATAL; + } + + list_for_each_entry(curr_errs, &http_errors_list, list) { + /* Error if two errors section owns the same name */ + if (strcmp(curr_errs->id, args[1]) == 0) { + ha_alert("parsing [%s:%d]: http-errors section '%s' already exists (declared at %s:%d).\n", + file, linenum, args[1], curr_errs->conf.file, curr_errs->conf.line); + err_code |= ERR_ALERT | ERR_FATAL; + } + } + + if ((curr_errs = calloc(1, sizeof(*curr_errs))) == NULL) { + ha_alert("parsing [%s:%d] : out of memory.\n", file, linenum); + err_code |= ERR_ALERT | ERR_ABORT; + goto out; + } + + LIST_APPEND(&http_errors_list, &curr_errs->list); + curr_errs->id = strdup(args[1]); + curr_errs->conf.file = strdup(file); + curr_errs->conf.line = linenum; + } + else if (strcmp(args[0], "errorfile") == 0) { /* error message from a file */ + struct http_reply *reply; + struct buffer *msg; + int status, rc; + + if (*(args[1]) == 0 || *(args[2]) == 0) { + ha_alert("parsing [%s:%d] : %s: expects <status_code> and <file> as arguments.\n", + file, linenum, args[0]); + err_code |= ERR_ALERT | ERR_FATAL; + goto out; + } + + status = atol(args[1]); + msg = http_parse_errorfile(status, args[2], &errmsg); + if (!msg) { + ha_alert("parsing [%s:%d] : %s : %s\n", file, linenum, args[0], errmsg); + err_code |= ERR_ALERT | ERR_FATAL; + goto out; + } + if (errmsg) { + ha_warning("parsing [%s:%d] : %s: %s\n", file, linenum, args[0], errmsg); + err_code |= ERR_WARN; + } + + reply = calloc(1, sizeof(*reply)); + if (!reply) { + ha_alert("parsing [%s:%d] : %s : out of memory.\n", file, linenum, args[0]); + err_code |= ERR_ALERT | ERR_FATAL; + goto out; + } + reply->type = HTTP_REPLY_ERRMSG; + reply->status = status; + reply->ctype = NULL; + LIST_INIT(&reply->hdrs); + reply->body.errmsg = msg; + + rc = http_get_status_idx(status); + curr_errs->replies[rc] = reply; + } + else if (*args[0] != 0) { + ha_alert("parsing [%s:%d] : unknown keyword '%s' in '%s' section\n", file, linenum, args[0], cursection); + err_code |= ERR_ALERT | ERR_FATAL; + goto out; + } + +out: + free(errmsg); + return err_code; +} + +static struct cfg_kw_list cfg_kws = {ILH, { + { CFG_LISTEN, "errorloc", proxy_parse_errorloc }, + { CFG_LISTEN, "errorloc302", proxy_parse_errorloc }, + { CFG_LISTEN, "errorloc303", proxy_parse_errorloc }, + { CFG_LISTEN, "errorfile", proxy_parse_errorfile }, + { CFG_LISTEN, "errorfiles", proxy_parse_errorfiles }, + { CFG_LISTEN, "http-error", proxy_parse_http_error }, + { 0, NULL, NULL }, +}}; + +INITCALL1(STG_REGISTER, cfg_register_keywords, &cfg_kws); +REGISTER_POST_PROXY_CHECK(proxy_check_errors); +REGISTER_POST_CHECK(post_check_errors); + +REGISTER_CONFIG_SECTION("http-errors", cfg_parse_http_errors, NULL); + +/************************************************************************/ +/* HTX sample fetches */ +/************************************************************************/ + +/* Returns 1 if a stream is an HTX stream. Otherwise, it returns 0. */ +static int +smp_fetch_is_htx(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + if (!smp->strm) + return 0; + + smp->data.u.sint = !!IS_HTX_STRM(smp->strm); + smp->data.type = SMP_T_BOOL; + return 1; +} + +/* Returns the number of blocks in an HTX message. The channel is chosen + * depending on the sample direction. */ +static int +smp_fetch_htx_nbblks(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = htx_nbblks(htx); + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the size of an HTX message. The channel is chosen depending on the + * sample direction. */ +static int +smp_fetch_htx_size(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = htx->size; + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the data size of an HTX message. The channel is chosen depending on the + * sample direction. */ +static int +smp_fetch_htx_data(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = htx->data; + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the used space (data+meta) of an HTX message. The channel is chosen + * depending on the sample direction. */ +static int +smp_fetch_htx_used(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = htx_used_space(htx); + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the free space (size-used) of an HTX message. The channel is chosen + * depending on the sample direction. */ +static int +smp_fetch_htx_free(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = htx_free_space(htx); + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the free space for data (free-sizeof(blk)) of an HTX message. The + * channel is chosen depending on the sample direction. */ +static int +smp_fetch_htx_free_data(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = htx_free_data_space(htx); + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns 1 if the HTX message contains EOM flag. Otherwise it returns 0. The + * channel is chosen depending on the sample direction. + */ +static int +smp_fetch_htx_has_eom(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + + if (!smp->strm) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + smp->data.u.sint = !!(htx->flags & HTX_FL_EOM); + smp->data.type = SMP_T_BOOL; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the type of a specific HTX block, if found in the message. Otherwise + * HTX_BLK_UNUSED is returned. Any positive integer (>= 0) is supported or + * "head", "tail" or "first". The channel is chosen depending on the sample + * direction. */ +static int +smp_fetch_htx_blk_type(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + enum htx_blk_type type; + int32_t pos; + + if (!smp->strm || !arg_p) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + pos = arg_p[0].data.sint; + if (pos == -1) + type = htx_get_head_type(htx); + else if (pos == -2) + type = htx_get_tail_type(htx); + else if (pos == -3) + type = htx_get_first_type(htx); + else + type = ((pos >= htx->head && pos <= htx->tail) + ? htx_get_blk_type(htx_get_blk(htx, pos)) + : HTX_BLK_UNUSED); + + chunk_initstr(&smp->data.u.str, htx_blk_type_str(type)); + smp->data.type = SMP_T_STR; + smp->flags = SMP_F_CONST | SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the size of a specific HTX block, if found in the message. Otherwise + * 0 is returned. Any positive integer (>= 0) is supported or "head", "tail" or + * "first". The channel is chosen depending on the sample direction. */ +static int +smp_fetch_htx_blk_size(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + struct htx_blk *blk; + int32_t pos; + + if (!smp->strm || !arg_p) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + pos = arg_p[0].data.sint; + if (pos == -1) + blk = htx_get_head_blk(htx); + else if (pos == -2) + blk = htx_get_tail_blk(htx); + else if (pos == -3) + blk = htx_get_first_blk(htx); + else + blk = ((pos >= htx->head && pos <= htx->tail) ? htx_get_blk(htx, pos) : NULL); + + smp->data.u.sint = (blk ? htx_get_blksz(blk) : 0); + smp->data.type = SMP_T_SINT; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the start-line if the selected HTX block exists and is a + * start-line. Otherwise 0 an empty string. Any positive integer (>= 0) is + * supported or "head", "tail" or "first". The channel is chosen depending on + * the sample direction. */ +static int +smp_fetch_htx_blk_stline(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct buffer *temp; + struct channel *chn; + struct htx *htx; + struct htx_blk *blk; + struct htx_sl *sl; + int32_t pos; + + if (!smp->strm || !arg_p) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + pos = arg_p[0].data.sint; + if (pos == -1) + blk = htx_get_head_blk(htx); + else if (pos == -2) + blk = htx_get_tail_blk(htx); + else if (pos == -3) + blk = htx_get_first_blk(htx); + else + blk = ((pos >= htx->head && pos <= htx->tail) ? htx_get_blk(htx, pos) : NULL); + + if (!blk || (htx_get_blk_type(blk) != HTX_BLK_REQ_SL && htx_get_blk_type(blk) != HTX_BLK_RES_SL)) { + smp->data.u.str.size = 0; + smp->data.u.str.area = ""; + smp->data.u.str.data = 0; + } + else { + sl = htx_get_blk_ptr(htx, blk); + + temp = get_trash_chunk(); + chunk_istcat(temp, htx_sl_p1(sl)); + temp->area[temp->data++] = ' '; + chunk_istcat(temp, htx_sl_p2(sl)); + temp->area[temp->data++] = ' '; + chunk_istcat(temp, htx_sl_p3(sl)); + + smp->data.u.str = *temp; + } + + smp->data.type = SMP_T_STR; + smp->flags = SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the header name if the selected HTX block exists and is a header or a + * trailer. Otherwise 0 an empty string. Any positive integer (>= 0) is + * supported or "head", "tail" or "first". The channel is chosen depending on + * the sample direction. */ +static int +smp_fetch_htx_blk_hdrname(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + struct htx_blk *blk; + int32_t pos; + + if (!smp->strm || !arg_p) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + pos = arg_p[0].data.sint; + if (pos == -1) + blk = htx_get_head_blk(htx); + else if (pos == -2) + blk = htx_get_tail_blk(htx); + else if (pos == -3) + blk = htx_get_first_blk(htx); + else + blk = ((pos >= htx->head && pos <= htx->tail) ? htx_get_blk(htx, pos) : NULL); + + if (!blk || (htx_get_blk_type(blk) != HTX_BLK_HDR && htx_get_blk_type(blk) != HTX_BLK_TLR)) { + smp->data.u.str.size = 0; + smp->data.u.str.area = ""; + smp->data.u.str.data = 0; + } + else { + struct ist name = htx_get_blk_name(htx, blk); + + chunk_initlen(&smp->data.u.str, name.ptr, name.len, name.len); + } + smp->data.type = SMP_T_STR; + smp->flags = SMP_F_CONST | SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the header value if the selected HTX block exists and is a header or + * a trailer. Otherwise 0 an empty string. Any positive integer (>= 0) is + * supported or "head", "tail" or "first". The channel is chosen depending on + * the sample direction. */ +static int +smp_fetch_htx_blk_hdrval(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + struct htx_blk *blk; + int32_t pos; + + if (!smp->strm || !arg_p) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + pos = arg_p[0].data.sint; + if (pos == -1) + blk = htx_get_head_blk(htx); + else if (pos == -2) + blk = htx_get_tail_blk(htx); + else if (pos == -3) + blk = htx_get_first_blk(htx); + else + blk = ((pos >= htx->head && pos <= htx->tail) ? htx_get_blk(htx, pos) : NULL); + + if (!blk || (htx_get_blk_type(blk) != HTX_BLK_HDR && htx_get_blk_type(blk) != HTX_BLK_TLR)) { + smp->data.u.str.size = 0; + smp->data.u.str.area = ""; + smp->data.u.str.data = 0; + } + else { + struct ist val = htx_get_blk_value(htx, blk); + + chunk_initlen(&smp->data.u.str, val.ptr, val.len, val.len); + } + smp->data.type = SMP_T_STR; + smp->flags = SMP_F_CONST | SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* Returns the value if the selected HTX block exists and is a data + * block. Otherwise 0 an empty string. Any positive integer (>= 0) is supported + * or "head", "tail" or "first". The channel is chosen depending on the sample + * direction. */ +static int +smp_fetch_htx_blk_data(const struct arg *arg_p, struct sample *smp, const char *kw, void *private) +{ + struct channel *chn; + struct htx *htx; + struct htx_blk *blk; + int32_t pos; + + if (!smp->strm || !arg_p) + return 0; + + chn = ((smp->opt & SMP_OPT_DIR) == SMP_OPT_DIR_RES) ? &smp->strm->res : &smp->strm->req; + htx = smp_prefetch_htx(smp, chn, NULL, 0); + if (!htx) + return 0; + + pos = arg_p[0].data.sint; + if (pos == -1) + blk = htx_get_head_blk(htx); + else if (pos == -2) + blk = htx_get_tail_blk(htx); + else if (pos == -3) + blk = htx_get_first_blk(htx); + else + blk = ((pos >= htx->head && pos <= htx->tail) ? htx_get_blk(htx, pos) : NULL); + + if (!blk || htx_get_blk_type(blk) != HTX_BLK_DATA) { + smp->data.u.str.size = 0; + smp->data.u.str.area = ""; + smp->data.u.str.data = 0; + } + else { + struct ist val = htx_get_blk_value(htx, blk); + + chunk_initlen(&smp->data.u.str, val.ptr, val.len, val.len); + } + smp->data.type = SMP_T_BIN; + smp->flags = SMP_F_CONST | SMP_F_VOLATILE | SMP_F_MAY_CHANGE; + return 1; +} + +/* This function is used to validate the arguments passed to any "htx_blk" fetch + * keywords. An argument is expected by these keywords. It must be a positive + * integer or on of the following strings: "head", "tail" or "first". It returns + * 0 on error, and a non-zero value if OK. + */ +int val_blk_arg(struct arg *arg, char **err_msg) +{ + if (arg[0].type != ARGT_STR || !arg[0].data.str.data) { + memprintf(err_msg, "a block position is expected (> 0) or a special block name (head, tail, first)"); + return 0; + } + if (arg[0].data.str.data == 4 && !strncmp(arg[0].data.str.area, "head", 4)) { + chunk_destroy(&arg[0].data.str); + arg[0].type = ARGT_SINT; + arg[0].data.sint = -1; + } + else if (arg[0].data.str.data == 4 && !strncmp(arg[0].data.str.area, "tail", 4)) { + chunk_destroy(&arg[0].data.str); + arg[0].type = ARGT_SINT; + arg[0].data.sint = -2; + } + else if (arg[0].data.str.data == 5 && !strncmp(arg[0].data.str.area, "first", 5)) { + chunk_destroy(&arg[0].data.str); + arg[0].type = ARGT_SINT; + arg[0].data.sint = -3; + } + else { + int pos; + + for (pos = 0; pos < arg[0].data.str.data; pos++) { + if (!isdigit((unsigned char)arg[0].data.str.area[pos])) { + memprintf(err_msg, "invalid block position"); + return 0; + } + } + + pos = strl2uic(arg[0].data.str.area, arg[0].data.str.data); + if (pos < 0) { + memprintf(err_msg, "block position must not be negative"); + return 0; + } + chunk_destroy(&arg[0].data.str); + arg[0].type = ARGT_SINT; + arg[0].data.sint = pos; + } + + return 1; +} + + +/* Note: must not be declared <const> as its list will be overwritten. + * Note: htx sample fetches should only used for development purpose. + */ +static struct sample_fetch_kw_list sample_fetch_keywords = {ILH, { + { "internal.strm.is_htx", smp_fetch_is_htx, 0, NULL, SMP_T_BOOL, SMP_USE_INTRN }, + + { "internal.htx.nbblks", smp_fetch_htx_nbblks, 0, NULL, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx.size", smp_fetch_htx_size, 0, NULL, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx.data", smp_fetch_htx_data, 0, NULL, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx.used", smp_fetch_htx_used, 0, NULL, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx.free", smp_fetch_htx_free, 0, NULL, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx.free_data", smp_fetch_htx_free_data, 0, NULL, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx.has_eom", smp_fetch_htx_has_eom, 0, NULL, SMP_T_BOOL, SMP_USE_HRQHV|SMP_USE_HRSHV}, + + { "internal.htx_blk.type", smp_fetch_htx_blk_type, ARG1(1,STR), val_blk_arg, SMP_T_STR, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx_blk.size", smp_fetch_htx_blk_size, ARG1(1,STR), val_blk_arg, SMP_T_SINT, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx_blk.start_line", smp_fetch_htx_blk_stline, ARG1(1,STR), val_blk_arg, SMP_T_STR, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx_blk.hdrname", smp_fetch_htx_blk_hdrname, ARG1(1,STR), val_blk_arg, SMP_T_STR, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx_blk.hdrval", smp_fetch_htx_blk_hdrval, ARG1(1,STR), val_blk_arg, SMP_T_STR, SMP_USE_HRQHV|SMP_USE_HRSHV}, + { "internal.htx_blk.data", smp_fetch_htx_blk_data, ARG1(1,STR), val_blk_arg, SMP_T_BIN, SMP_USE_HRQHV|SMP_USE_HRSHV}, + + { /* END */ }, +}}; + +INITCALL1(STG_REGISTER, sample_register_fetches, &sample_fetch_keywords); |