mirror of
https://git.haproxy.org/git/haproxy.git/
synced 2025-08-06 07:07:04 +02:00
This patch tries to address a design flaw with how logformat expressions are parsed from config. Indeed, some parse_logformat_string() calls are performed during config parsing when the proxy mode is not yet known. Here's a config example that illustrates the issue: defaults mode tcp listen test bind :8888 http-response set-header custom-hdr "%trl" # needs http mode http The above config should work, because the effective proxy mode is http, yet haproxy fails with this error: [ALERT] (99051) : config : parsing [repro.conf:6] : error detected in proxy 'test' while parsing 'http-response set-header' rule : format tag 'trl' is reserved for HTTP mode. To fix the issue once and for all, let's implement smart postparsing for logformat expressions encountered during config parsing: - split parse_logformat_string() (and subfonctions) in order to create a new lf_expr_postcheck() function that must be called to finish preparing and checking the logformat expression once the proxy type is known. - save some config hints info during parse_logformat_string() to generate more precise error messages during lf_expr_postcheck(), if needed, we rely on curpx->conf.args.{file,line} hints for that because parse_logformat_string() doesn't know about current file and line number. - lf_expr_postcheck() uses PR_FL_CHECKED proxy flag to know if the function may try to make the proxy compatible with the expression, or if it should simply fail as soon as an incompatibility is detected. - if parse_logformat_string() is called from an unchecked proxy, then schedule the expression for postparsing, else (ie: during runtime), run the postcheck right away. This change will also allow for some logformat expression error handling simplifications in the future.
1445 lines
43 KiB
C
1445 lines
43 KiB
C
#include <ctype.h>
|
|
|
|
#include <haproxy/api.h>
|
|
#include <haproxy/arg.h>
|
|
#include <haproxy/buf.h>
|
|
#include <haproxy/cfgparse.h>
|
|
#include <haproxy/check.h>
|
|
#include <haproxy/cli.h>
|
|
#include <haproxy/global.h>
|
|
#include <haproxy/http.h>
|
|
#include <haproxy/http_rules.h>
|
|
#include <haproxy/list.h>
|
|
#include <haproxy/log.h>
|
|
#include <haproxy/sample.h>
|
|
#include <haproxy/session.h>
|
|
#include <haproxy/stream-t.h>
|
|
#include <haproxy/tcp_rules.h>
|
|
#include <haproxy/tcpcheck.h>
|
|
#include <haproxy/tools.h>
|
|
#include <haproxy/vars.h>
|
|
#include <haproxy/xxhash.h>
|
|
|
|
|
|
/* This contains a pool of struct vars */
|
|
DECLARE_STATIC_POOL(var_pool, "vars", sizeof(struct var));
|
|
|
|
/* list of variables for the process scope. */
|
|
struct vars proc_vars THREAD_ALIGNED(64);
|
|
|
|
/* This array of int contains the system limits per context. */
|
|
static unsigned int var_global_limit = 0;
|
|
static unsigned int var_proc_limit = 0;
|
|
static unsigned int var_sess_limit = 0;
|
|
static unsigned int var_txn_limit = 0;
|
|
static unsigned int var_reqres_limit = 0;
|
|
static unsigned int var_check_limit = 0;
|
|
static uint64_t var_name_hash_seed = 0;
|
|
|
|
/* Structure and array matching set-var conditions to their respective flag
|
|
* value.
|
|
*/
|
|
struct var_set_condition {
|
|
const char *cond_str;
|
|
uint flag;
|
|
};
|
|
|
|
static struct var_set_condition conditions_array[] = {
|
|
{ "ifexists", VF_COND_IFEXISTS },
|
|
{ "ifnotexists", VF_COND_IFNOTEXISTS },
|
|
{ "ifempty", VF_COND_IFEMPTY },
|
|
{ "ifnotempty", VF_COND_IFNOTEMPTY },
|
|
{ "ifset", VF_COND_IFSET },
|
|
{ "ifnotset", VF_COND_IFNOTSET },
|
|
{ "ifgt", VF_COND_IFGT },
|
|
{ "iflt", VF_COND_IFLT },
|
|
{ NULL, 0 }
|
|
};
|
|
|
|
/* returns the struct vars pointer for a session, stream and scope, or NULL if
|
|
* it does not exist.
|
|
*/
|
|
static inline struct vars *get_vars(struct session *sess, struct stream *strm, enum vars_scope scope)
|
|
{
|
|
switch (scope) {
|
|
case SCOPE_PROC:
|
|
return &proc_vars;
|
|
case SCOPE_SESS:
|
|
return sess ? &sess->vars : NULL;
|
|
case SCOPE_CHECK: {
|
|
struct check *check = sess ? objt_check(sess->origin) : NULL;
|
|
|
|
return check ? &check->vars : NULL;
|
|
}
|
|
case SCOPE_TXN:
|
|
return strm ? &strm->vars_txn : NULL;
|
|
case SCOPE_REQ:
|
|
case SCOPE_RES:
|
|
default:
|
|
return strm ? &strm->vars_reqres : NULL;
|
|
}
|
|
}
|
|
|
|
/* This function adds or remove memory size from the accounting. The inner
|
|
* pointers may be null when setting the outer ones only.
|
|
*/
|
|
void var_accounting_diff(struct vars *vars, struct session *sess, struct stream *strm, int size)
|
|
{
|
|
switch (vars->scope) {
|
|
case SCOPE_REQ:
|
|
case SCOPE_RES:
|
|
if (var_reqres_limit && strm)
|
|
_HA_ATOMIC_ADD(&strm->vars_reqres.size, size);
|
|
__fallthrough;
|
|
case SCOPE_TXN:
|
|
if (var_txn_limit && strm)
|
|
_HA_ATOMIC_ADD(&strm->vars_txn.size, size);
|
|
goto scope_sess;
|
|
case SCOPE_CHECK:
|
|
if (var_check_limit) {
|
|
struct check *check = objt_check(sess->origin);
|
|
|
|
if (check)
|
|
_HA_ATOMIC_ADD(&check->vars.size, size);
|
|
}
|
|
scope_sess:
|
|
__fallthrough;
|
|
case SCOPE_SESS:
|
|
if (var_sess_limit)
|
|
_HA_ATOMIC_ADD(&sess->vars.size, size);
|
|
__fallthrough;
|
|
case SCOPE_PROC:
|
|
if (var_proc_limit || var_global_limit)
|
|
_HA_ATOMIC_ADD(&proc_vars.size, size);
|
|
}
|
|
}
|
|
|
|
/* This function returns 1 if the <size> is available in the var
|
|
* pool <vars>, otherwise returns 0. If the space is available,
|
|
* the size is reserved. The inner pointers may be null when setting
|
|
* the outer ones only. The accounting uses either <sess> or <strm>
|
|
* depending on the scope. <strm> may be NULL when no stream is known
|
|
* and only the session exists (eg: tcp-request connection).
|
|
*/
|
|
static int var_accounting_add(struct vars *vars, struct session *sess, struct stream *strm, int size)
|
|
{
|
|
switch (vars->scope) {
|
|
case SCOPE_REQ:
|
|
case SCOPE_RES:
|
|
if (var_reqres_limit && strm && strm->vars_reqres.size + size > var_reqres_limit)
|
|
return 0;
|
|
__fallthrough;
|
|
case SCOPE_TXN:
|
|
if (var_txn_limit && strm && strm->vars_txn.size + size > var_txn_limit)
|
|
return 0;
|
|
goto scope_sess;
|
|
case SCOPE_CHECK: {
|
|
struct check *check = objt_check(sess->origin);
|
|
|
|
if (var_check_limit && check && check->vars.size + size > var_check_limit)
|
|
return 0;
|
|
}
|
|
scope_sess:
|
|
__fallthrough;
|
|
case SCOPE_SESS:
|
|
if (var_sess_limit && sess->vars.size + size > var_sess_limit)
|
|
return 0;
|
|
__fallthrough;
|
|
case SCOPE_PROC:
|
|
/* note: scope proc collects all others and is currently identical to the
|
|
* global limit.
|
|
*/
|
|
if (var_proc_limit && proc_vars.size + size > var_proc_limit)
|
|
return 0;
|
|
if (var_global_limit && proc_vars.size + size > var_global_limit)
|
|
return 0;
|
|
}
|
|
var_accounting_diff(vars, sess, strm, size);
|
|
return 1;
|
|
}
|
|
|
|
/* This function removes a variable from the list and frees the memory it was
|
|
* using. If the variable is marked "VF_PERMANENT", the sample_data is only
|
|
* reset to SMP_T_ANY unless <force> is non nul. Returns the freed size.
|
|
*/
|
|
unsigned int var_clear(struct var *var, int force)
|
|
{
|
|
unsigned int size = 0;
|
|
|
|
if (var->data.type == SMP_T_STR || var->data.type == SMP_T_BIN) {
|
|
ha_free(&var->data.u.str.area);
|
|
size += var->data.u.str.data;
|
|
}
|
|
else if (var->data.type == SMP_T_METH && var->data.u.meth.meth == HTTP_METH_OTHER) {
|
|
ha_free(&var->data.u.meth.str.area);
|
|
size += var->data.u.meth.str.data;
|
|
}
|
|
/* wipe the sample */
|
|
var->data.type = SMP_T_ANY;
|
|
|
|
if (!(var->flags & VF_PERMANENT) || force) {
|
|
LIST_DELETE(&var->l);
|
|
pool_free(var_pool, var);
|
|
size += sizeof(struct var);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
/* This function free all the memory used by all the variables
|
|
* in the list.
|
|
*/
|
|
void vars_prune(struct vars *vars, struct session *sess, struct stream *strm)
|
|
{
|
|
struct var *var, *tmp;
|
|
unsigned int size = 0;
|
|
|
|
vars_wrlock(vars);
|
|
list_for_each_entry_safe(var, tmp, &vars->head, l) {
|
|
size += var_clear(var, 1);
|
|
}
|
|
vars_wrunlock(vars);
|
|
var_accounting_diff(vars, sess, strm, -size);
|
|
}
|
|
|
|
/* This function frees all the memory used by all the session variables in the
|
|
* list starting at <vars>.
|
|
*/
|
|
void vars_prune_per_sess(struct vars *vars)
|
|
{
|
|
struct var *var, *tmp;
|
|
unsigned int size = 0;
|
|
|
|
vars_wrlock(vars);
|
|
list_for_each_entry_safe(var, tmp, &vars->head, l) {
|
|
size += var_clear(var, 1);
|
|
}
|
|
vars_wrunlock(vars);
|
|
|
|
if (var_sess_limit)
|
|
_HA_ATOMIC_SUB(&vars->size, size);
|
|
if (var_proc_limit || var_global_limit)
|
|
_HA_ATOMIC_SUB(&proc_vars.size, size);
|
|
}
|
|
|
|
/* This function initializes a variables list head */
|
|
void vars_init_head(struct vars *vars, enum vars_scope scope)
|
|
{
|
|
LIST_INIT(&vars->head);
|
|
vars->scope = scope;
|
|
vars->size = 0;
|
|
HA_RWLOCK_INIT(&vars->rwlock);
|
|
}
|
|
|
|
/* This function returns a hash value and a scope for a variable name of a
|
|
* specified length. It makes sure that the scope is valid. It returns non-zero
|
|
* on success, 0 on failure. Neither hash nor scope may be NULL.
|
|
*/
|
|
static int vars_hash_name(const char *name, int len, enum vars_scope *scope,
|
|
uint64_t *hash, char **err)
|
|
{
|
|
const char *tmp;
|
|
|
|
/* Check length. */
|
|
if (len == 0) {
|
|
memprintf(err, "Empty variable name cannot be accepted");
|
|
return 0;
|
|
}
|
|
|
|
/* Check scope. */
|
|
if (len > 5 && strncmp(name, "proc.", 5) == 0) {
|
|
name += 5;
|
|
len -= 5;
|
|
*scope = SCOPE_PROC;
|
|
}
|
|
else if (len > 5 && strncmp(name, "sess.", 5) == 0) {
|
|
name += 5;
|
|
len -= 5;
|
|
*scope = SCOPE_SESS;
|
|
}
|
|
else if (len > 4 && strncmp(name, "txn.", 4) == 0) {
|
|
name += 4;
|
|
len -= 4;
|
|
*scope = SCOPE_TXN;
|
|
}
|
|
else if (len > 4 && strncmp(name, "req.", 4) == 0) {
|
|
name += 4;
|
|
len -= 4;
|
|
*scope = SCOPE_REQ;
|
|
}
|
|
else if (len > 4 && strncmp(name, "res.", 4) == 0) {
|
|
name += 4;
|
|
len -= 4;
|
|
*scope = SCOPE_RES;
|
|
}
|
|
else if (len > 6 && strncmp(name, "check.", 6) == 0) {
|
|
name += 6;
|
|
len -= 6;
|
|
*scope = SCOPE_CHECK;
|
|
}
|
|
else {
|
|
memprintf(err, "invalid variable name '%.*s'. A variable name must be start by its scope. "
|
|
"The scope can be 'proc', 'sess', 'txn', 'req', 'res' or 'check'", len, name);
|
|
return 0;
|
|
}
|
|
|
|
/* Check variable name syntax. */
|
|
for (tmp = name; tmp < name + len; tmp++) {
|
|
if (!isalnum((unsigned char)*tmp) && *tmp != '_' && *tmp != '.') {
|
|
memprintf(err, "invalid syntax at char '%s'", tmp);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
*hash = XXH3(name, len, var_name_hash_seed);
|
|
return 1;
|
|
}
|
|
|
|
/* This function returns the variable from the given list that matches
|
|
* <name_hash> or returns NULL if not found. It's only a linked list since it
|
|
* is not expected to have many variables per scope (a few tens at best).
|
|
* The caller is responsible for ensuring that <vars> is properly locked.
|
|
*/
|
|
static struct var *var_get(struct vars *vars, uint64_t name_hash)
|
|
{
|
|
struct var *var;
|
|
|
|
list_for_each_entry(var, &vars->head, l)
|
|
if (var->name_hash == name_hash)
|
|
return var;
|
|
return NULL;
|
|
}
|
|
|
|
/* Returns 0 if fails, else returns 1. */
|
|
static int smp_fetch_var(const struct arg *args, struct sample *smp, const char *kw, void *private)
|
|
{
|
|
const struct var_desc *var_desc = &args[0].data.var;
|
|
const struct buffer *def = NULL;
|
|
|
|
if (args[1].type == ARGT_STR)
|
|
def = &args[1].data.str;
|
|
|
|
return vars_get_by_desc(var_desc, smp, def);
|
|
}
|
|
|
|
/*
|
|
* Clear the contents of a variable so that it can be reset directly.
|
|
* This function is used just before a variable is filled out of a sample's
|
|
* content.
|
|
*/
|
|
static inline void var_clear_buffer(struct sample *smp, struct vars *vars, struct var *var, int var_type)
|
|
{
|
|
if (var_type == SMP_T_STR || var_type == SMP_T_BIN) {
|
|
ha_free(&var->data.u.str.area);
|
|
var_accounting_diff(vars, smp->sess, smp->strm,
|
|
-var->data.u.str.data);
|
|
}
|
|
else if (var_type == SMP_T_METH && var->data.u.meth.meth == HTTP_METH_OTHER) {
|
|
ha_free(&var->data.u.meth.str.area);
|
|
var_accounting_diff(vars, smp->sess, smp->strm,
|
|
-var->data.u.meth.str.data);
|
|
}
|
|
}
|
|
|
|
/* This function tries to create a variable whose name hash is <name_hash> in
|
|
* scope <scope> and store sample <smp> as its value.
|
|
*
|
|
* The stream and session are extracted from <smp>, whose stream may be NULL
|
|
* when scope is SCOPE_SESS. In case there wouldn't be enough memory to store
|
|
* the sample while the variable was already created, it would be changed to
|
|
* a bool (which is memory-less).
|
|
*
|
|
* Flags is a bitfield that may contain one of the following flags:
|
|
* - VF_CREATEONLY: do nothing if the variable already exists (success).
|
|
* - VF_PERMANENT: this flag will be passed to the variable upon creation
|
|
*
|
|
* - VF_COND_IFEXISTS: only set variable if it already exists
|
|
* - VF_COND_IFNOTEXISTS: only set variable if it did not exist yet
|
|
* - VF_COND_IFEMPTY: only set variable if sample is empty
|
|
* - VF_COND_IFNOTEMPTY: only set variable if sample is not empty
|
|
* - VF_COND_IFSET: only set variable if its type is not SMP_TYPE_ANY
|
|
* - VF_COND_IFNOTSET: only set variable if its type is ANY
|
|
* - VF_COND_IFGT: only set variable if its value is greater than the sample's
|
|
* - VF_COND_IFLT: only set variable if its value is less than the sample's
|
|
*
|
|
* It returns 0 on failure, non-zero on success.
|
|
*/
|
|
int var_set(uint64_t name_hash, enum vars_scope scope, struct sample *smp, uint flags)
|
|
{
|
|
struct vars *vars;
|
|
struct var *var;
|
|
int ret = 0;
|
|
int previous_type = SMP_T_ANY;
|
|
|
|
vars = get_vars(smp->sess, smp->strm, scope);
|
|
if (!vars || vars->scope != scope)
|
|
return 0;
|
|
|
|
vars_wrlock(vars);
|
|
|
|
/* Look for existing variable name. */
|
|
var = var_get(vars, name_hash);
|
|
|
|
if (var) {
|
|
if (flags & VF_CREATEONLY) {
|
|
ret = 1;
|
|
goto unlock;
|
|
}
|
|
|
|
if (flags & VF_COND_IFNOTEXISTS)
|
|
goto unlock;
|
|
} else {
|
|
if (flags & VF_COND_IFEXISTS)
|
|
goto unlock;
|
|
|
|
/* Check memory available. */
|
|
if (!var_accounting_add(vars, smp->sess, smp->strm, sizeof(struct var)))
|
|
goto unlock;
|
|
|
|
/* Create new entry. */
|
|
var = pool_alloc(var_pool);
|
|
if (!var)
|
|
goto unlock;
|
|
LIST_APPEND(&vars->head, &var->l);
|
|
var->name_hash = name_hash;
|
|
var->flags = flags & VF_PERMANENT;
|
|
var->data.type = SMP_T_ANY;
|
|
}
|
|
|
|
/* A variable of type SMP_T_ANY is considered as unset (either created
|
|
* and never set or unset-var was called on it).
|
|
*/
|
|
if ((flags & VF_COND_IFSET && var->data.type == SMP_T_ANY) ||
|
|
(flags & VF_COND_IFNOTSET && var->data.type != SMP_T_ANY))
|
|
goto unlock;
|
|
|
|
/* Set type. */
|
|
previous_type = var->data.type;
|
|
var->data.type = smp->data.type;
|
|
|
|
if (flags & VF_COND_IFEMPTY) {
|
|
switch(smp->data.type) {
|
|
case SMP_T_ANY:
|
|
case SMP_T_STR:
|
|
case SMP_T_BIN:
|
|
/* The actual test on the contents of the sample will be
|
|
* performed later.
|
|
*/
|
|
break;
|
|
default:
|
|
/* The sample cannot be empty since it has a scalar type. */
|
|
var->data.type = previous_type;
|
|
goto unlock;
|
|
}
|
|
}
|
|
|
|
/* Copy data. If the data needs memory, the function can fail. */
|
|
switch (var->data.type) {
|
|
case SMP_T_BOOL:
|
|
var_clear_buffer(smp, vars, var, previous_type);
|
|
var->data.u.sint = smp->data.u.sint;
|
|
break;
|
|
case SMP_T_SINT:
|
|
if (previous_type == var->data.type) {
|
|
if (((flags & VF_COND_IFGT) && !(var->data.u.sint > smp->data.u.sint)) ||
|
|
((flags & VF_COND_IFLT) && !(var->data.u.sint < smp->data.u.sint)))
|
|
goto unlock;
|
|
}
|
|
var_clear_buffer(smp, vars, var, previous_type);
|
|
var->data.u.sint = smp->data.u.sint;
|
|
break;
|
|
case SMP_T_IPV4:
|
|
var_clear_buffer(smp, vars, var, previous_type);
|
|
var->data.u.ipv4 = smp->data.u.ipv4;
|
|
break;
|
|
case SMP_T_IPV6:
|
|
var_clear_buffer(smp, vars, var, previous_type);
|
|
var->data.u.ipv6 = smp->data.u.ipv6;
|
|
break;
|
|
case SMP_T_STR:
|
|
case SMP_T_BIN:
|
|
if ((flags & VF_COND_IFNOTEMPTY && !smp->data.u.str.data) ||
|
|
(flags & VF_COND_IFEMPTY && smp->data.u.str.data)) {
|
|
var->data.type = previous_type;
|
|
goto unlock;
|
|
}
|
|
var_clear_buffer(smp, vars, var, previous_type);
|
|
if (!var_accounting_add(vars, smp->sess, smp->strm, smp->data.u.str.data)) {
|
|
var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */
|
|
goto unlock;
|
|
}
|
|
|
|
var->data.u.str.area = malloc(smp->data.u.str.data);
|
|
if (!var->data.u.str.area) {
|
|
var_accounting_diff(vars, smp->sess, smp->strm,
|
|
-smp->data.u.str.data);
|
|
var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */
|
|
goto unlock;
|
|
}
|
|
var->data.u.str.data = smp->data.u.str.data;
|
|
memcpy(var->data.u.str.area, smp->data.u.str.area,
|
|
var->data.u.str.data);
|
|
break;
|
|
case SMP_T_METH:
|
|
var_clear_buffer(smp, vars, var, previous_type);
|
|
var->data.u.meth.meth = smp->data.u.meth.meth;
|
|
if (smp->data.u.meth.meth != HTTP_METH_OTHER)
|
|
break;
|
|
|
|
if (!var_accounting_add(vars, smp->sess, smp->strm, smp->data.u.meth.str.data)) {
|
|
var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */
|
|
goto unlock;
|
|
}
|
|
|
|
var->data.u.meth.str.area = malloc(smp->data.u.meth.str.data);
|
|
if (!var->data.u.meth.str.area) {
|
|
var_accounting_diff(vars, smp->sess, smp->strm,
|
|
-smp->data.u.meth.str.data);
|
|
var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */
|
|
goto unlock;
|
|
}
|
|
var->data.u.meth.str.data = smp->data.u.meth.str.data;
|
|
var->data.u.meth.str.size = smp->data.u.meth.str.data;
|
|
memcpy(var->data.u.meth.str.area, smp->data.u.meth.str.area,
|
|
var->data.u.meth.str.data);
|
|
break;
|
|
}
|
|
|
|
/* OK, now done */
|
|
ret = 1;
|
|
unlock:
|
|
vars_wrunlock(vars);
|
|
return ret;
|
|
}
|
|
|
|
/* Deletes a variable matching name hash <name_hash> and scope <scope> for the
|
|
* session and stream found in <smp>. Note that stream may be null for
|
|
* SCOPE_SESS. Returns 0 if the scope was not found otherwise 1.
|
|
*/
|
|
int var_unset(uint64_t name_hash, enum vars_scope scope, struct sample *smp)
|
|
{
|
|
struct vars *vars;
|
|
struct var *var;
|
|
unsigned int size = 0;
|
|
|
|
vars = get_vars(smp->sess, smp->strm, scope);
|
|
if (!vars || vars->scope != scope)
|
|
return 0;
|
|
|
|
/* Look for existing variable name. */
|
|
vars_wrlock(vars);
|
|
var = var_get(vars, name_hash);
|
|
if (var) {
|
|
size = var_clear(var, 0);
|
|
var_accounting_diff(vars, smp->sess, smp->strm, -size);
|
|
}
|
|
vars_wrunlock(vars);
|
|
return 1;
|
|
}
|
|
|
|
|
|
/*
|
|
* Convert a string set-var condition into its numerical value.
|
|
* The corresponding bit is set in the <cond_bitmap> parameter if the
|
|
* <cond> is known.
|
|
* Returns 1 in case of success.
|
|
*/
|
|
static int vars_parse_cond_param(const struct buffer *cond, uint *cond_bitmap, char **err)
|
|
{
|
|
struct var_set_condition *cond_elt = &conditions_array[0];
|
|
|
|
/* The conditions array is NULL terminated. */
|
|
while (cond_elt->cond_str) {
|
|
if (chunk_strcmp(cond, cond_elt->cond_str) == 0) {
|
|
*cond_bitmap |= cond_elt->flag;
|
|
break;
|
|
}
|
|
++cond_elt;
|
|
}
|
|
|
|
if (cond_elt->cond_str == NULL && err)
|
|
memprintf(err, "unknown condition \"%.*s\"", (int)cond->data, cond->area);
|
|
|
|
return cond_elt->cond_str != NULL;
|
|
}
|
|
|
|
/* Returns 0 if fails, else returns 1. */
|
|
static int smp_conv_store(const struct arg *args, struct sample *smp, void *private)
|
|
{
|
|
uint conditions = 0;
|
|
int cond_idx = 1;
|
|
|
|
while (args[cond_idx].type == ARGT_STR) {
|
|
if (vars_parse_cond_param(&args[cond_idx++].data.str, &conditions, NULL) == 0)
|
|
break;
|
|
}
|
|
|
|
return var_set(args[0].data.var.name_hash, args[0].data.var.scope, smp, conditions);
|
|
}
|
|
|
|
/* Returns 0 if fails, else returns 1. */
|
|
static int smp_conv_clear(const struct arg *args, struct sample *smp, void *private)
|
|
{
|
|
return var_unset(args[0].data.var.name_hash, args[0].data.var.scope, smp);
|
|
}
|
|
|
|
/* This functions check an argument entry and fill it with a variable
|
|
* type. The argument must be a string. If the variable lookup fails,
|
|
* the function returns 0 and fill <err>, otherwise it returns 1.
|
|
*/
|
|
int vars_check_arg(struct arg *arg, char **err)
|
|
{
|
|
enum vars_scope scope;
|
|
struct sample empty_smp = { };
|
|
uint64_t hash;
|
|
|
|
/* Check arg type. */
|
|
if (arg->type != ARGT_STR) {
|
|
memprintf(err, "unexpected argument type");
|
|
return 0;
|
|
}
|
|
|
|
/* Register new variable name. */
|
|
if (!vars_hash_name(arg->data.str.area, arg->data.str.data, &scope, &hash, err))
|
|
return 0;
|
|
|
|
if (scope == SCOPE_PROC && !var_set(hash, scope, &empty_smp, VF_CREATEONLY|VF_PERMANENT))
|
|
return 0;
|
|
|
|
/* properly destroy the chunk */
|
|
chunk_destroy(&arg->data.str);
|
|
|
|
/* Use the global variable name pointer. */
|
|
arg->type = ARGT_VAR;
|
|
arg->data.var.name_hash = hash;
|
|
arg->data.var.scope = scope;
|
|
return 1;
|
|
}
|
|
|
|
/* This function stores a sample in a variable unless it is of type "proc" and
|
|
* not defined yet.
|
|
* Returns zero on failure and non-zero otherwise. The variable not being
|
|
* defined is treated as a failure.
|
|
*/
|
|
int vars_set_by_name_ifexist(const char *name, size_t len, struct sample *smp)
|
|
{
|
|
enum vars_scope scope;
|
|
uint64_t hash;
|
|
|
|
/* Resolve name and scope. */
|
|
if (!vars_hash_name(name, len, &scope, &hash, NULL))
|
|
return 0;
|
|
|
|
/* Variable creation is allowed for all scopes apart from the PROC one. */
|
|
return var_set(hash, scope, smp, (scope == SCOPE_PROC) ? VF_COND_IFEXISTS : 0);
|
|
}
|
|
|
|
|
|
/* This function stores a sample in a variable.
|
|
* Returns zero on failure and non-zero otherwise.
|
|
*/
|
|
int vars_set_by_name(const char *name, size_t len, struct sample *smp)
|
|
{
|
|
enum vars_scope scope;
|
|
uint64_t hash;
|
|
|
|
/* Resolve name and scope. */
|
|
if (!vars_hash_name(name, len, &scope, &hash, NULL))
|
|
return 0;
|
|
|
|
return var_set(hash, scope, smp, 0);
|
|
}
|
|
|
|
/* This function unsets a variable if it was already defined.
|
|
* Returns zero on failure and non-zero otherwise.
|
|
*/
|
|
int vars_unset_by_name_ifexist(const char *name, size_t len, struct sample *smp)
|
|
{
|
|
enum vars_scope scope;
|
|
uint64_t hash;
|
|
|
|
/* Resolve name and scope. */
|
|
if (!vars_hash_name(name, len, &scope, &hash, NULL))
|
|
return 0;
|
|
|
|
return var_unset(hash, scope, smp);
|
|
}
|
|
|
|
|
|
/* This retrieves variable whose hash matches <name_hash> from variables <vars>,
|
|
* and if found and not empty, duplicates the result into sample <smp>.
|
|
* smp_dup() is used in order to release the variables lock ASAP (so a pre-
|
|
* allocated chunk is obtained via get_trash_shunk()). The variables' lock is
|
|
* used for reads.
|
|
*
|
|
* The function returns 0 if the variable was not found and no default
|
|
* value was provided in <def>, otherwise 1 with the sample filled.
|
|
* Default values are always returned as strings.
|
|
*/
|
|
static int var_to_smp(struct vars *vars, uint64_t name_hash, struct sample *smp, const struct buffer *def)
|
|
{
|
|
struct var *var;
|
|
|
|
/* Get the variable entry. */
|
|
vars_rdlock(vars);
|
|
var = var_get(vars, name_hash);
|
|
if (!var || !var->data.type) {
|
|
if (!def) {
|
|
vars_rdunlock(vars);
|
|
return 0;
|
|
}
|
|
|
|
/* not found but we have a default value */
|
|
smp->data.type = SMP_T_STR;
|
|
smp->data.u.str = *def;
|
|
}
|
|
else
|
|
smp->data = var->data;
|
|
|
|
/* Copy sample. */
|
|
smp_dup(smp);
|
|
|
|
vars_rdunlock(vars);
|
|
return 1;
|
|
}
|
|
|
|
/* This function fills a sample with the variable content.
|
|
*
|
|
* Keep in mind that a sample content is duplicated by using smp_dup()
|
|
* and it therefore uses a pre-allocated trash chunk as returned by
|
|
* get_trash_chunk().
|
|
*
|
|
* If the variable is not valid in this scope, 0 is always returned.
|
|
* If the variable is valid but not found, either the default value
|
|
* <def> is returned if not NULL, or zero is returned.
|
|
*
|
|
* Returns 1 if the sample is filled, otherwise it returns 0.
|
|
*/
|
|
int vars_get_by_name(const char *name, size_t len, struct sample *smp, const struct buffer *def)
|
|
{
|
|
struct vars *vars;
|
|
enum vars_scope scope;
|
|
uint64_t hash;
|
|
|
|
/* Resolve name and scope. */
|
|
if (!vars_hash_name(name, len, &scope, &hash, NULL))
|
|
return 0;
|
|
|
|
/* Select "vars" pool according with the scope. */
|
|
vars = get_vars(smp->sess, smp->strm, scope);
|
|
if (!vars || vars->scope != scope)
|
|
return 0;
|
|
|
|
return var_to_smp(vars, hash, smp, def);
|
|
}
|
|
|
|
/* This function fills a sample with the content of the variable described
|
|
* by <var_desc>.
|
|
*
|
|
* Keep in mind that a sample content is duplicated by using smp_dup()
|
|
* and it therefore uses a pre-allocated trash chunk as returned by
|
|
* get_trash_chunk().
|
|
*
|
|
* If the variable is not valid in this scope, 0 is always returned.
|
|
* If the variable is valid but not found, either the default value
|
|
* <def> is returned if not NULL, or zero is returned.
|
|
*
|
|
* Returns 1 if the sample is filled, otherwise it returns 0.
|
|
*/
|
|
int vars_get_by_desc(const struct var_desc *var_desc, struct sample *smp, const struct buffer *def)
|
|
{
|
|
struct vars *vars;
|
|
|
|
/* Select "vars" pool according with the scope. */
|
|
vars = get_vars(smp->sess, smp->strm, var_desc->scope);
|
|
|
|
/* Check if the scope is available a this point of processing. */
|
|
if (!vars || vars->scope != var_desc->scope)
|
|
return 0;
|
|
|
|
return var_to_smp(vars, var_desc->name_hash, smp, def);
|
|
}
|
|
|
|
/* Always returns ACT_RET_CONT even if an error occurs. */
|
|
static enum act_return action_store(struct act_rule *rule, struct proxy *px,
|
|
struct session *sess, struct stream *s, int flags)
|
|
{
|
|
struct buffer *fmtstr = NULL;
|
|
struct sample smp;
|
|
int dir;
|
|
|
|
switch (rule->from) {
|
|
case ACT_F_TCP_REQ_CON: dir = SMP_OPT_DIR_REQ; break;
|
|
case ACT_F_TCP_REQ_SES: dir = SMP_OPT_DIR_REQ; break;
|
|
case ACT_F_TCP_REQ_CNT: dir = SMP_OPT_DIR_REQ; break;
|
|
case ACT_F_TCP_RES_CNT: dir = SMP_OPT_DIR_RES; break;
|
|
case ACT_F_HTTP_REQ: dir = SMP_OPT_DIR_REQ; break;
|
|
case ACT_F_HTTP_RES: dir = SMP_OPT_DIR_RES; break;
|
|
case ACT_F_TCP_CHK: dir = SMP_OPT_DIR_REQ; break;
|
|
case ACT_F_CFG_PARSER: dir = SMP_OPT_DIR_REQ; break; /* not used anyway */
|
|
case ACT_F_CLI_PARSER: dir = SMP_OPT_DIR_REQ; break; /* not used anyway */
|
|
default:
|
|
send_log(px, LOG_ERR, "Vars: internal error while execute action store.");
|
|
if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
|
|
ha_alert("Vars: internal error while execute action store.\n");
|
|
return ACT_RET_CONT;
|
|
}
|
|
|
|
/* Process the expression. */
|
|
memset(&smp, 0, sizeof(smp));
|
|
|
|
if (!lf_expr_isempty(&rule->arg.vars.fmt)) {
|
|
/* a format-string is used */
|
|
|
|
fmtstr = alloc_trash_chunk();
|
|
if (!fmtstr) {
|
|
send_log(px, LOG_ERR, "Vars: memory allocation failure while processing store rule.");
|
|
if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE))
|
|
ha_alert("Vars: memory allocation failure while processing store rule.\n");
|
|
return ACT_RET_CONT;
|
|
}
|
|
|
|
/* execute the log-format expression */
|
|
fmtstr->data = sess_build_logline(sess, s, fmtstr->area, fmtstr->size, &rule->arg.vars.fmt);
|
|
|
|
/* convert it to a sample of type string as it's what the vars
|
|
* API consumes, and store it.
|
|
*/
|
|
smp_set_owner(&smp, px, sess, s, 0);
|
|
smp.data.type = SMP_T_STR;
|
|
smp.data.u.str = *fmtstr;
|
|
var_set(rule->arg.vars.name_hash, rule->arg.vars.scope, &smp, rule->arg.vars.conditions);
|
|
}
|
|
else {
|
|
/* an expression is used */
|
|
if (!sample_process(px, sess, s, dir|SMP_OPT_FINAL,
|
|
rule->arg.vars.expr, &smp))
|
|
return ACT_RET_CONT;
|
|
}
|
|
|
|
/* Store the sample, and ignore errors. */
|
|
var_set(rule->arg.vars.name_hash, rule->arg.vars.scope, &smp, rule->arg.vars.conditions);
|
|
free_trash_chunk(fmtstr);
|
|
return ACT_RET_CONT;
|
|
}
|
|
|
|
/* Always returns ACT_RET_CONT even if an error occurs. */
|
|
static enum act_return action_clear(struct act_rule *rule, struct proxy *px,
|
|
struct session *sess, struct stream *s, int flags)
|
|
{
|
|
struct sample smp;
|
|
|
|
memset(&smp, 0, sizeof(smp));
|
|
smp_set_owner(&smp, px, sess, s, SMP_OPT_FINAL);
|
|
|
|
/* Clear the variable using the sample context, and ignore errors. */
|
|
var_unset(rule->arg.vars.name_hash, rule->arg.vars.scope, &smp);
|
|
return ACT_RET_CONT;
|
|
}
|
|
|
|
static void release_store_rule(struct act_rule *rule)
|
|
{
|
|
lf_expr_deinit(&rule->arg.vars.fmt);
|
|
|
|
release_sample_expr(rule->arg.vars.expr);
|
|
}
|
|
|
|
/* This two function checks the variable name and replace the
|
|
* configuration string name by the global string name. its
|
|
* the same string, but the global pointer can be easy to
|
|
* compare. They return non-zero on success, zero on failure.
|
|
*
|
|
* The first function checks a sample-fetch and the second
|
|
* checks a converter.
|
|
*/
|
|
static int smp_check_var(struct arg *args, char **err)
|
|
{
|
|
return vars_check_arg(&args[0], err);
|
|
}
|
|
|
|
static int conv_check_var(struct arg *args, struct sample_conv *conv,
|
|
const char *file, int line, char **err_msg)
|
|
{
|
|
int cond_idx = 1;
|
|
uint conditions = 0;
|
|
int retval = vars_check_arg(&args[0], err_msg);
|
|
|
|
while (retval && args[cond_idx].type == ARGT_STR)
|
|
retval = vars_parse_cond_param(&args[cond_idx++].data.str, &conditions, err_msg);
|
|
|
|
return retval;
|
|
}
|
|
|
|
/* This function is a common parser for using variables. It understands
|
|
* the format:
|
|
*
|
|
* set-var-fmt(<variable-name>[,<cond> ...]) <format-string>
|
|
* set-var(<variable-name>[,<cond> ...]) <expression>
|
|
* unset-var(<variable-name>)
|
|
*
|
|
* It returns ACT_RET_PRS_ERR if fails and <err> is filled with an error
|
|
* message. Otherwise, it returns ACT_RET_PRS_OK and the variable <expr>
|
|
* is filled with the pointer to the expression to execute. The proxy is
|
|
* only used to retrieve the ->conf entries.
|
|
*/
|
|
static enum act_parse_ret parse_store(const char **args, int *arg, struct proxy *px,
|
|
struct act_rule *rule, char **err)
|
|
{
|
|
const char *var_name = args[*arg-1];
|
|
int var_len;
|
|
const char *kw_name;
|
|
int flags = 0, set_var = 0; /* 0=unset-var, 1=set-var, 2=set-var-fmt */
|
|
struct sample empty_smp = { };
|
|
struct ist condition = IST_NULL;
|
|
struct ist var = IST_NULL;
|
|
struct ist varname_ist = IST_NULL;
|
|
|
|
if (strncmp(var_name, "set-var-fmt", 11) == 0) {
|
|
var_name += 11;
|
|
set_var = 2;
|
|
}
|
|
else if (strncmp(var_name, "set-var", 7) == 0) {
|
|
var_name += 7;
|
|
set_var = 1;
|
|
}
|
|
else if (strncmp(var_name, "unset-var", 9) == 0) {
|
|
var_name += 9;
|
|
set_var = 0;
|
|
}
|
|
|
|
if (*var_name != '(') {
|
|
memprintf(err, "invalid or incomplete action '%s'. Expects 'set-var(<var-name>)', 'set-var-fmt(<var-name>)' or 'unset-var(<var-name>)'",
|
|
args[*arg-1]);
|
|
return ACT_RET_PRS_ERR;
|
|
}
|
|
var_name++; /* jump the '(' */
|
|
var_len = strlen(var_name);
|
|
var_len--; /* remove the ')' */
|
|
if (var_name[var_len] != ')') {
|
|
memprintf(err, "incomplete argument after action '%s'. Expects 'set-var(<var-name>)', 'set-var-fmt(<var-name>)' or 'unset-var(<var-name>)'",
|
|
args[*arg-1]);
|
|
return ACT_RET_PRS_ERR;
|
|
}
|
|
|
|
/* Parse the optional conditions. */
|
|
var = ist2(var_name, var_len);
|
|
varname_ist = istsplit(&var, ',');
|
|
var_len = istlen(varname_ist);
|
|
|
|
condition = istsplit(&var, ',');
|
|
|
|
if (istlen(condition) && set_var == 0) {
|
|
memprintf(err, "unset-var does not expect parameters after the variable name. Only \"set-var\" and \"set-var-fmt\" manage conditions");
|
|
return ACT_RET_PRS_ERR;
|
|
}
|
|
|
|
while (istlen(condition)) {
|
|
struct buffer cond = {};
|
|
|
|
chunk_initlen(&cond, istptr(condition), 0, istlen(condition));
|
|
if (vars_parse_cond_param(&cond, &rule->arg.vars.conditions, err) == 0)
|
|
return ACT_RET_PRS_ERR;
|
|
|
|
condition = istsplit(&var, ',');
|
|
}
|
|
|
|
lf_expr_init(&rule->arg.vars.fmt);
|
|
if (!vars_hash_name(var_name, var_len, &rule->arg.vars.scope, &rule->arg.vars.name_hash, err))
|
|
return ACT_RET_PRS_ERR;
|
|
|
|
if (rule->arg.vars.scope == SCOPE_PROC &&
|
|
!var_set(rule->arg.vars.name_hash, rule->arg.vars.scope, &empty_smp, VF_CREATEONLY|VF_PERMANENT))
|
|
return 0;
|
|
|
|
/* There is no fetch method when variable is unset. Just set the right
|
|
* action and return. */
|
|
if (!set_var) {
|
|
rule->action = ACT_CUSTOM;
|
|
rule->action_ptr = action_clear;
|
|
rule->release_ptr = release_store_rule;
|
|
return ACT_RET_PRS_OK;
|
|
}
|
|
|
|
kw_name = args[*arg-1];
|
|
|
|
switch (rule->from) {
|
|
case ACT_F_TCP_REQ_CON:
|
|
flags = SMP_VAL_FE_CON_ACC;
|
|
px->conf.args.ctx = ARGC_TCO;
|
|
break;
|
|
case ACT_F_TCP_REQ_SES:
|
|
flags = SMP_VAL_FE_SES_ACC;
|
|
px->conf.args.ctx = ARGC_TSE;
|
|
break;
|
|
case ACT_F_TCP_REQ_CNT:
|
|
if (px->cap & PR_CAP_FE)
|
|
flags |= SMP_VAL_FE_REQ_CNT;
|
|
if (px->cap & PR_CAP_BE)
|
|
flags |= SMP_VAL_BE_REQ_CNT;
|
|
px->conf.args.ctx = ARGC_TRQ;
|
|
break;
|
|
case ACT_F_TCP_RES_CNT:
|
|
if (px->cap & PR_CAP_FE)
|
|
flags |= SMP_VAL_FE_RES_CNT;
|
|
if (px->cap & PR_CAP_BE)
|
|
flags |= SMP_VAL_BE_RES_CNT;
|
|
px->conf.args.ctx = ARGC_TRS;
|
|
break;
|
|
case ACT_F_HTTP_REQ:
|
|
if (px->cap & PR_CAP_FE)
|
|
flags |= SMP_VAL_FE_HRQ_HDR;
|
|
if (px->cap & PR_CAP_BE)
|
|
flags |= SMP_VAL_BE_HRQ_HDR;
|
|
px->conf.args.ctx = ARGC_HRQ;
|
|
break;
|
|
case ACT_F_HTTP_RES:
|
|
if (px->cap & PR_CAP_FE)
|
|
flags |= SMP_VAL_FE_HRS_HDR;
|
|
if (px->cap & PR_CAP_BE)
|
|
flags |= SMP_VAL_BE_HRS_HDR;
|
|
px->conf.args.ctx = ARGC_HRS;
|
|
break;
|
|
case ACT_F_TCP_CHK:
|
|
flags = SMP_VAL_BE_CHK_RUL;
|
|
px->conf.args.ctx = ARGC_TCK;
|
|
break;
|
|
case ACT_F_CFG_PARSER:
|
|
flags = SMP_VAL_CFG_PARSER;
|
|
px->conf.args.ctx = ARGC_CFG;
|
|
break;
|
|
case ACT_F_CLI_PARSER:
|
|
flags = SMP_VAL_CLI_PARSER;
|
|
px->conf.args.ctx = ARGC_CLI;
|
|
break;
|
|
default:
|
|
memprintf(err,
|
|
"internal error, unexpected rule->from=%d, please report this bug!",
|
|
rule->from);
|
|
return ACT_RET_PRS_ERR;
|
|
}
|
|
|
|
if (set_var == 2) { /* set-var-fmt */
|
|
if (!parse_logformat_string(args[*arg], px, &rule->arg.vars.fmt, 0, flags, err))
|
|
return ACT_RET_PRS_ERR;
|
|
|
|
(*arg)++;
|
|
} else {
|
|
/* set-var */
|
|
rule->arg.vars.expr = sample_parse_expr((char **)args, arg, px->conf.args.file,
|
|
px->conf.args.line, err, &px->conf.args, NULL);
|
|
if (!rule->arg.vars.expr)
|
|
return ACT_RET_PRS_ERR;
|
|
|
|
if (!(rule->arg.vars.expr->fetch->val & flags)) {
|
|
memprintf(err,
|
|
"fetch method '%s' extracts information from '%s', none of which is available here",
|
|
kw_name, sample_src_names(rule->arg.vars.expr->fetch->use));
|
|
free(rule->arg.vars.expr);
|
|
return ACT_RET_PRS_ERR;
|
|
}
|
|
}
|
|
|
|
rule->action = ACT_CUSTOM;
|
|
rule->action_ptr = action_store;
|
|
rule->release_ptr = release_store_rule;
|
|
return ACT_RET_PRS_OK;
|
|
}
|
|
|
|
|
|
/* parses a global "set-var" directive. It will create a temporary rule and
|
|
* expression that are parsed, processed, and released on the fly so that we
|
|
* respect the real set-var syntax. These directives take the following format:
|
|
* set-var <name> <expression>
|
|
* set-var-fmt <name> <fmt>
|
|
* Note that parse_store() expects "set-var(name) <expression>" so we have to
|
|
* temporarily replace the keyword here.
|
|
*/
|
|
static int vars_parse_global_set_var(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
struct proxy px = {
|
|
.id = "CFG",
|
|
.conf.args = { .file = file, .line = line, },
|
|
.flags = PR_FL_CHECKED,
|
|
};
|
|
struct act_rule rule = {
|
|
.arg.vars.scope = SCOPE_PROC,
|
|
.from = ACT_F_CFG_PARSER,
|
|
.conf = { .file = (char *)file, .line = line, },
|
|
};
|
|
enum obj_type objt = OBJ_TYPE_NONE;
|
|
struct session *sess = NULL;
|
|
enum act_parse_ret p_ret;
|
|
char *old_arg1;
|
|
char *tmp_arg1;
|
|
int arg = 2; // variable name
|
|
int ret = -1;
|
|
int use_fmt = 0;
|
|
|
|
LIST_INIT(&px.conf.args.list);
|
|
|
|
use_fmt = strcmp(args[0], "set-var-fmt") == 0;
|
|
|
|
if (!*args[1] || !*args[2]) {
|
|
if (use_fmt)
|
|
memprintf(err, "'%s' requires a process-wide variable name ('proc.<name>') and a format string.", args[0]);
|
|
else
|
|
memprintf(err, "'%s' requires a process-wide variable name ('proc.<name>') and a sample expression.", args[0]);
|
|
goto end;
|
|
}
|
|
|
|
tmp_arg1 = NULL;
|
|
if (!memprintf(&tmp_arg1, "set-var%s(%s)", use_fmt ? "-fmt" : "", args[1]))
|
|
goto end;
|
|
|
|
/* parse_store() will always return a message in <err> on error */
|
|
old_arg1 = args[1]; args[1] = tmp_arg1;
|
|
p_ret = parse_store((const char **)args, &arg, &px, &rule, err);
|
|
free(args[1]); args[1] = old_arg1;
|
|
|
|
if (p_ret != ACT_RET_PRS_OK)
|
|
goto end;
|
|
|
|
if (rule.arg.vars.scope != SCOPE_PROC) {
|
|
memprintf(err, "'%s': cannot set variable '%s', only scope 'proc' is permitted in the global section.", args[0], args[1]);
|
|
goto end;
|
|
}
|
|
|
|
if (smp_resolve_args(&px, err) != 0) {
|
|
release_sample_expr(rule.arg.vars.expr);
|
|
indent_msg(err, 2);
|
|
goto end;
|
|
}
|
|
|
|
if (use_fmt && !(sess = session_new(&px, NULL, &objt))) {
|
|
release_sample_expr(rule.arg.vars.expr);
|
|
memprintf(err, "'%s': out of memory when trying to set variable '%s' in the global section.", args[0], args[1]);
|
|
goto end;
|
|
}
|
|
|
|
action_store(&rule, &px, sess, NULL, 0);
|
|
release_sample_expr(rule.arg.vars.expr);
|
|
if (sess)
|
|
session_free(sess);
|
|
|
|
ret = 0;
|
|
end:
|
|
return ret;
|
|
}
|
|
|
|
/* parse CLI's "get var <name>" */
|
|
static int vars_parse_cli_get_var(char **args, char *payload, struct appctx *appctx, void *private)
|
|
{
|
|
struct vars *vars;
|
|
struct sample smp = { };
|
|
int i;
|
|
|
|
if (!cli_has_level(appctx, ACCESS_LVL_OPER))
|
|
return 1;
|
|
|
|
if (!*args[2])
|
|
return cli_err(appctx, "Missing process-wide variable identifier.\n");
|
|
|
|
vars = get_vars(NULL, NULL, SCOPE_PROC);
|
|
if (!vars || vars->scope != SCOPE_PROC)
|
|
return 0;
|
|
|
|
if (!vars_get_by_name(args[2], strlen(args[2]), &smp, NULL))
|
|
return cli_err(appctx, "Variable not found.\n");
|
|
|
|
/* the sample returned by vars_get_by_name() is allocated into a trash
|
|
* chunk so we have no constraint to manipulate it.
|
|
*/
|
|
chunk_printf(&trash, "%s: type=%s value=", args[2], smp_to_type[smp.data.type]);
|
|
|
|
if (!sample_casts[smp.data.type][SMP_T_STR] ||
|
|
!sample_casts[smp.data.type][SMP_T_STR](&smp)) {
|
|
chunk_appendf(&trash, "(undisplayable)\n");
|
|
} else {
|
|
/* Display the displayable chars*. */
|
|
b_putchr(&trash, '<');
|
|
for (i = 0; i < smp.data.u.str.data; i++) {
|
|
if (isprint((unsigned char)smp.data.u.str.area[i]))
|
|
b_putchr(&trash, smp.data.u.str.area[i]);
|
|
else
|
|
b_putchr(&trash, '.');
|
|
}
|
|
b_putchr(&trash, '>');
|
|
b_putchr(&trash, '\n');
|
|
b_putchr(&trash, 0);
|
|
}
|
|
return cli_msg(appctx, LOG_INFO, trash.area);
|
|
}
|
|
|
|
/* parse CLI's "set var <name>". It accepts:
|
|
* - set var <name> <expression>
|
|
* - set var <name> expr <expression>
|
|
* - set var <name> fmt <format>
|
|
*/
|
|
static int vars_parse_cli_set_var(char **args, char *payload, struct appctx *appctx, void *private)
|
|
{
|
|
struct proxy px = {
|
|
.id = "CLI",
|
|
.conf.args = { .file = "CLI", .line = 0, },
|
|
.flags = PR_FL_CHECKED,
|
|
};
|
|
struct act_rule rule = {
|
|
.arg.vars.scope = SCOPE_PROC,
|
|
.from = ACT_F_CLI_PARSER,
|
|
.conf = { .file = "CLI", .line = 0, },
|
|
};
|
|
enum obj_type objt = OBJ_TYPE_NONE;
|
|
struct session *sess = NULL;
|
|
enum act_parse_ret p_ret;
|
|
const char *tmp_args[3];
|
|
int tmp_arg;
|
|
char *tmp_act;
|
|
char *err = NULL;
|
|
int nberr;
|
|
int use_fmt = 0;
|
|
|
|
LIST_INIT(&px.conf.args.list);
|
|
|
|
if (!cli_has_level(appctx, ACCESS_LVL_OPER))
|
|
return 1;
|
|
|
|
if (!*args[2])
|
|
return cli_err(appctx, "Missing process-wide variable identifier.\n");
|
|
|
|
if (!*args[3])
|
|
return cli_err(appctx, "Missing either 'expr', 'fmt' or expression.\n");
|
|
|
|
if (*args[4]) {
|
|
/* this is the long format */
|
|
if (strcmp(args[3], "fmt") == 0)
|
|
use_fmt = 1;
|
|
else if (strcmp(args[3], "expr") != 0) {
|
|
memprintf(&err, "'%s %s': arg type must be either 'expr' or 'fmt' but got '%s'.", args[0], args[1], args[3]);
|
|
goto fail;
|
|
}
|
|
}
|
|
|
|
tmp_act = NULL;
|
|
if (!memprintf(&tmp_act, "set-var%s(%s)", use_fmt ? "-fmt" : "", args[2])) {
|
|
memprintf(&err, "memory allocation error.");
|
|
goto fail;
|
|
}
|
|
|
|
/* parse_store() will always return a message in <err> on error */
|
|
tmp_args[0] = tmp_act;
|
|
tmp_args[1] = (*args[4]) ? args[4] : args[3];
|
|
tmp_args[2] = "";
|
|
tmp_arg = 1; // must point to the first arg after the action
|
|
p_ret = parse_store(tmp_args, &tmp_arg, &px, &rule, &err);
|
|
free(tmp_act);
|
|
|
|
if (p_ret != ACT_RET_PRS_OK)
|
|
goto fail;
|
|
|
|
if (rule.arg.vars.scope != SCOPE_PROC) {
|
|
memprintf(&err, "'%s %s': cannot set variable '%s', only scope 'proc' is permitted here.", args[0], args[1], args[2]);
|
|
goto fail;
|
|
}
|
|
|
|
err = NULL;
|
|
nberr = smp_resolve_args(&px, &err);
|
|
if (nberr) {
|
|
release_sample_expr(rule.arg.vars.expr);
|
|
indent_msg(&err, 2);
|
|
goto fail;
|
|
}
|
|
|
|
if (use_fmt && !(sess = session_new(&px, NULL, &objt))) {
|
|
release_sample_expr(rule.arg.vars.expr);
|
|
memprintf(&err, "memory allocation error.");
|
|
goto fail;
|
|
}
|
|
|
|
action_store(&rule, &px, sess, NULL, 0);
|
|
release_sample_expr(rule.arg.vars.expr);
|
|
if (sess)
|
|
session_free(sess);
|
|
|
|
appctx->st0 = CLI_ST_PROMPT;
|
|
return 0;
|
|
fail:
|
|
return cli_dynerr(appctx, err);
|
|
}
|
|
|
|
static int vars_max_size(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err, unsigned int *limit)
|
|
{
|
|
char *error;
|
|
|
|
*limit = strtol(args[1], &error, 10);
|
|
if (*error != 0) {
|
|
memprintf(err, "%s: '%s' is an invalid size", args[0], args[1]);
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int vars_max_size_global(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_global_limit);
|
|
}
|
|
|
|
static int vars_max_size_proc(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_proc_limit);
|
|
}
|
|
|
|
static int vars_max_size_sess(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_sess_limit);
|
|
}
|
|
|
|
static int vars_max_size_txn(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_txn_limit);
|
|
}
|
|
|
|
static int vars_max_size_reqres(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_reqres_limit);
|
|
}
|
|
|
|
static int vars_max_size_check(char **args, int section_type, struct proxy *curpx,
|
|
const struct proxy *defpx, const char *file, int line,
|
|
char **err)
|
|
{
|
|
return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_check_limit);
|
|
}
|
|
|
|
/* early boot initialization */
|
|
static void vars_init()
|
|
{
|
|
var_name_hash_seed = ha_random64();
|
|
/* Initialize process vars */
|
|
vars_init_head(&proc_vars, SCOPE_PROC);
|
|
}
|
|
|
|
INITCALL0(STG_PREPARE, vars_init);
|
|
|
|
static struct sample_fetch_kw_list sample_fetch_keywords = {ILH, {
|
|
|
|
{ "var", smp_fetch_var, ARG2(1,STR,STR), smp_check_var, SMP_T_ANY, SMP_USE_CONST },
|
|
{ /* END */ },
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, sample_register_fetches, &sample_fetch_keywords);
|
|
|
|
static struct sample_conv_kw_list sample_conv_kws = {ILH, {
|
|
{ "set-var", smp_conv_store, ARG5(1,STR,STR,STR,STR,STR), conv_check_var, SMP_T_ANY, SMP_T_ANY },
|
|
{ "unset-var", smp_conv_clear, ARG1(1,STR), conv_check_var, SMP_T_ANY, SMP_T_ANY },
|
|
{ /* END */ },
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, sample_register_convs, &sample_conv_kws);
|
|
|
|
static struct action_kw_list tcp_req_conn_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, tcp_req_conn_keywords_register, &tcp_req_conn_kws);
|
|
|
|
static struct action_kw_list tcp_req_sess_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, tcp_req_sess_keywords_register, &tcp_req_sess_kws);
|
|
|
|
static struct action_kw_list tcp_req_cont_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, tcp_req_cont_keywords_register, &tcp_req_cont_kws);
|
|
|
|
static struct action_kw_list tcp_res_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, tcp_res_cont_keywords_register, &tcp_res_kws);
|
|
|
|
static struct action_kw_list tcp_check_kws = {ILH, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, tcp_check_keywords_register, &tcp_check_kws);
|
|
|
|
static struct action_kw_list http_req_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, http_req_keywords_register, &http_req_kws);
|
|
|
|
static struct action_kw_list http_res_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, http_res_keywords_register, &http_res_kws);
|
|
|
|
static struct action_kw_list http_after_res_kws = { { }, {
|
|
{ "set-var-fmt", parse_store, KWF_MATCH_PREFIX },
|
|
{ "set-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ "unset-var", parse_store, KWF_MATCH_PREFIX },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, http_after_res_keywords_register, &http_after_res_kws);
|
|
|
|
static struct cfg_kw_list cfg_kws = {{ },{
|
|
{ CFG_GLOBAL, "set-var", vars_parse_global_set_var },
|
|
{ CFG_GLOBAL, "set-var-fmt", vars_parse_global_set_var },
|
|
{ CFG_GLOBAL, "tune.vars.global-max-size", vars_max_size_global },
|
|
{ CFG_GLOBAL, "tune.vars.proc-max-size", vars_max_size_proc },
|
|
{ CFG_GLOBAL, "tune.vars.sess-max-size", vars_max_size_sess },
|
|
{ CFG_GLOBAL, "tune.vars.txn-max-size", vars_max_size_txn },
|
|
{ CFG_GLOBAL, "tune.vars.reqres-max-size", vars_max_size_reqres },
|
|
{ CFG_GLOBAL, "tune.vars.check-max-size", vars_max_size_check },
|
|
{ /* END */ }
|
|
}};
|
|
|
|
INITCALL1(STG_REGISTER, cfg_register_keywords, &cfg_kws);
|
|
|
|
|
|
/* register cli keywords */
|
|
static struct cli_kw_list cli_kws = {{ },{
|
|
{ { "get", "var", NULL }, "get var <name> : retrieve contents of a process-wide variable", vars_parse_cli_get_var, NULL },
|
|
{ { "set", "var", NULL }, "set var <name> [fmt|expr] {<fmt>|<expr>}: set variable from an expression or a format", vars_parse_cli_set_var, NULL, NULL, NULL, ACCESS_EXPERIMENTAL },
|
|
{ { NULL }, NULL, NULL, NULL }
|
|
}};
|
|
INITCALL1(STG_REGISTER, cli_register_kw, &cli_kws);
|