mirror of
http://git.haproxy.org/git/haproxy.git/
synced 2024-12-19 18:28:33 +00:00
68a71ed3f6
This avoids 6 build warnings when preprocessing happens before compiling with gcc >= 7.
1454 lines
43 KiB
C
1454 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.
|
|
*/
|
|
static 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.
|
|
*/
|
|
static 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 argumen 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 (!LIST_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)
|
|
{
|
|
struct logformat_node *lf, *lfb;
|
|
|
|
list_for_each_entry_safe(lf, lfb, &rule->arg.vars.fmt, list) {
|
|
LIST_DELETE(&lf->list);
|
|
release_sample_expr(lf->expr);
|
|
free(lf->arg);
|
|
free(lf);
|
|
}
|
|
|
|
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, ',');
|
|
}
|
|
|
|
LIST_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)++;
|
|
|
|
/* for late error reporting */
|
|
free(px->conf.lfs_file);
|
|
px->conf.lfs_file = strdup(px->conf.args.file);
|
|
px->conf.lfs_line = px->conf.args.line;
|
|
} 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, },
|
|
};
|
|
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)");
|
|
} 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, 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, },
|
|
};
|
|
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);
|