Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1 | #include <ctype.h> |
| 2 | |
Willy Tarreau | 4c7e4b7 | 2020-05-27 12:58:42 +0200 | [diff] [blame] | 3 | #include <haproxy/api.h> |
Willy Tarreau | b255105 | 2020-06-09 09:07:15 +0200 | [diff] [blame] | 4 | #include <haproxy/arg.h> |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 5 | #include <haproxy/buf.h> |
Willy Tarreau | 6be7849 | 2020-06-05 00:00:29 +0200 | [diff] [blame] | 6 | #include <haproxy/cfgparse.h> |
Willy Tarreau | 4aa573d | 2020-06-04 18:21:56 +0200 | [diff] [blame] | 7 | #include <haproxy/check.h> |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 8 | #include <haproxy/cli.h> |
Willy Tarreau | dfd3de8 | 2020-06-04 23:46:14 +0200 | [diff] [blame] | 9 | #include <haproxy/global.h> |
Willy Tarreau | cd72d8c | 2020-06-02 19:11:26 +0200 | [diff] [blame] | 10 | #include <haproxy/http.h> |
Willy Tarreau | c761f84 | 2020-06-04 11:40:28 +0200 | [diff] [blame] | 11 | #include <haproxy/http_rules.h> |
Willy Tarreau | 853b297 | 2020-05-27 18:01:47 +0200 | [diff] [blame] | 12 | #include <haproxy/list.h> |
Willy Tarreau | 36979d9 | 2020-06-05 17:27:29 +0200 | [diff] [blame] | 13 | #include <haproxy/log.h> |
Willy Tarreau | e6ce10b | 2020-06-04 15:33:47 +0200 | [diff] [blame] | 14 | #include <haproxy/sample.h> |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 15 | #include <haproxy/session.h> |
Willy Tarreau | dfd3de8 | 2020-06-04 23:46:14 +0200 | [diff] [blame] | 16 | #include <haproxy/stream-t.h> |
Willy Tarreau | 8b550af | 2020-06-04 17:42:48 +0200 | [diff] [blame] | 17 | #include <haproxy/tcp_rules.h> |
Willy Tarreau | b255105 | 2020-06-09 09:07:15 +0200 | [diff] [blame] | 18 | #include <haproxy/tcpcheck.h> |
Willy Tarreau | 67046bf | 2021-05-08 13:56:31 +0200 | [diff] [blame] | 19 | #include <haproxy/tools.h> |
Willy Tarreau | a171892 | 2020-06-04 16:25:31 +0200 | [diff] [blame] | 20 | #include <haproxy/vars.h> |
Willy Tarreau | b555eb1 | 2021-10-06 17:11:51 +0200 | [diff] [blame] | 21 | #include <haproxy/xxhash.h> |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 22 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 23 | |
| 24 | /* This contains a pool of struct vars */ |
Willy Tarreau | 8ceae72 | 2018-11-26 11:58:30 +0100 | [diff] [blame] | 25 | DECLARE_STATIC_POOL(var_pool, "vars", sizeof(struct var)); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 26 | |
Willy Tarreau | cfc4f24 | 2021-05-08 11:41:28 +0200 | [diff] [blame] | 27 | /* list of variables for the process scope. */ |
| 28 | struct vars proc_vars THREAD_ALIGNED(64); |
| 29 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 30 | /* This array of int contains the system limits per context. */ |
| 31 | static unsigned int var_global_limit = 0; |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 32 | static unsigned int var_proc_limit = 0; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 33 | static unsigned int var_sess_limit = 0; |
| 34 | static unsigned int var_txn_limit = 0; |
| 35 | static unsigned int var_reqres_limit = 0; |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 36 | static unsigned int var_check_limit = 0; |
Willy Tarreau | 2c897d9 | 2021-08-31 08:48:55 +0200 | [diff] [blame] | 37 | static uint64_t var_name_hash_seed = 0; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 38 | |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 39 | /* Structure and array matching set-var conditions to their respective flag |
| 40 | * value. |
| 41 | */ |
| 42 | struct var_set_condition { |
| 43 | const char *cond_str; |
| 44 | uint flag; |
| 45 | }; |
| 46 | |
| 47 | static struct var_set_condition conditions_array[] = { |
| 48 | { "ifexists", VF_COND_IFEXISTS }, |
| 49 | { "ifnotexists", VF_COND_IFNOTEXISTS }, |
| 50 | { "ifempty", VF_COND_IFEMPTY }, |
| 51 | { "ifnotempty", VF_COND_IFNOTEMPTY }, |
| 52 | { "ifset", VF_COND_IFSET }, |
| 53 | { "ifnotset", VF_COND_IFNOTSET }, |
| 54 | { "ifgt", VF_COND_IFGT }, |
| 55 | { "iflt", VF_COND_IFLT }, |
| 56 | { NULL, 0 } |
| 57 | }; |
| 58 | |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 59 | /* returns the struct vars pointer for a session, stream and scope, or NULL if |
| 60 | * it does not exist. |
| 61 | */ |
| 62 | static inline struct vars *get_vars(struct session *sess, struct stream *strm, enum vars_scope scope) |
| 63 | { |
| 64 | switch (scope) { |
| 65 | case SCOPE_PROC: |
Willy Tarreau | cfc4f24 | 2021-05-08 11:41:28 +0200 | [diff] [blame] | 66 | return &proc_vars; |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 67 | case SCOPE_SESS: |
Willy Tarreau | a07d61b | 2021-03-26 11:27:59 +0100 | [diff] [blame] | 68 | return sess ? &sess->vars : NULL; |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 69 | case SCOPE_CHECK: { |
Christopher Faulet | c4439f7 | 2021-06-02 11:48:42 +0200 | [diff] [blame] | 70 | struct check *check = sess ? objt_check(sess->origin) : NULL; |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 71 | |
Christopher Faulet | 0fca7ed | 2020-04-21 11:53:32 +0200 | [diff] [blame] | 72 | return check ? &check->vars : NULL; |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 73 | } |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 74 | case SCOPE_TXN: |
| 75 | return strm ? &strm->vars_txn : NULL; |
| 76 | case SCOPE_REQ: |
| 77 | case SCOPE_RES: |
| 78 | default: |
| 79 | return strm ? &strm->vars_reqres : NULL; |
| 80 | } |
| 81 | } |
| 82 | |
Willy Tarreau | 7233098 | 2015-06-19 11:21:56 +0200 | [diff] [blame] | 83 | /* This function adds or remove memory size from the accounting. The inner |
| 84 | * pointers may be null when setting the outer ones only. |
| 85 | */ |
Miroslav Zagorac | 6deab79 | 2020-12-09 16:34:29 +0100 | [diff] [blame] | 86 | void var_accounting_diff(struct vars *vars, struct session *sess, struct stream *strm, int size) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 87 | { |
| 88 | switch (vars->scope) { |
| 89 | case SCOPE_REQ: |
| 90 | case SCOPE_RES: |
Willy Tarreau | 55f8a83 | 2021-09-08 15:51:06 +0200 | [diff] [blame] | 91 | if (var_reqres_limit && strm) |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 92 | _HA_ATOMIC_ADD(&strm->vars_reqres.size, size); |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 93 | /* fall through */ |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 94 | case SCOPE_TXN: |
Willy Tarreau | 55f8a83 | 2021-09-08 15:51:06 +0200 | [diff] [blame] | 95 | if (var_txn_limit && strm) |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 96 | _HA_ATOMIC_ADD(&strm->vars_txn.size, size); |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 97 | goto scope_sess; |
Willy Tarreau | 55f8a83 | 2021-09-08 15:51:06 +0200 | [diff] [blame] | 98 | case SCOPE_CHECK: |
| 99 | if (var_check_limit) { |
Christopher Faulet | 0fca7ed | 2020-04-21 11:53:32 +0200 | [diff] [blame] | 100 | struct check *check = objt_check(sess->origin); |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 101 | |
Christopher Faulet | 0fca7ed | 2020-04-21 11:53:32 +0200 | [diff] [blame] | 102 | if (check) |
| 103 | _HA_ATOMIC_ADD(&check->vars.size, size); |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 104 | } |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 105 | /* fall through */ |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 106 | scope_sess: |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 107 | case SCOPE_SESS: |
Willy Tarreau | 55f8a83 | 2021-09-08 15:51:06 +0200 | [diff] [blame] | 108 | if (var_sess_limit) |
| 109 | _HA_ATOMIC_ADD(&sess->vars.size, size); |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 110 | /* fall through */ |
| 111 | case SCOPE_PROC: |
Willy Tarreau | 55f8a83 | 2021-09-08 15:51:06 +0200 | [diff] [blame] | 112 | if (var_proc_limit || var_global_limit) |
| 113 | _HA_ATOMIC_ADD(&proc_vars.size, size); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 114 | } |
| 115 | } |
| 116 | |
| 117 | /* This function returns 1 if the <size> is available in the var |
Joseph Herlant | 0767689 | 2018-11-15 09:19:50 -0800 | [diff] [blame] | 118 | * pool <vars>, otherwise returns 0. If the space is available, |
Willy Tarreau | 7233098 | 2015-06-19 11:21:56 +0200 | [diff] [blame] | 119 | * the size is reserved. The inner pointers may be null when setting |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 120 | * the outer ones only. The accounting uses either <sess> or <strm> |
| 121 | * depending on the scope. <strm> may be NULL when no stream is known |
| 122 | * and only the session exists (eg: tcp-request connection). |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 123 | */ |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 124 | static int var_accounting_add(struct vars *vars, struct session *sess, struct stream *strm, int size) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 125 | { |
| 126 | switch (vars->scope) { |
| 127 | case SCOPE_REQ: |
| 128 | case SCOPE_RES: |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 129 | if (var_reqres_limit && strm && strm->vars_reqres.size + size > var_reqres_limit) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 130 | return 0; |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 131 | /* fall through */ |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 132 | case SCOPE_TXN: |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 133 | if (var_txn_limit && strm && strm->vars_txn.size + size > var_txn_limit) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 134 | return 0; |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 135 | goto scope_sess; |
| 136 | case SCOPE_CHECK: { |
Christopher Faulet | 0fca7ed | 2020-04-21 11:53:32 +0200 | [diff] [blame] | 137 | struct check *check = objt_check(sess->origin); |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 138 | |
Christopher Faulet | 0fca7ed | 2020-04-21 11:53:32 +0200 | [diff] [blame] | 139 | if (var_check_limit && check && check->vars.size + size > var_check_limit) |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 140 | return 0; |
| 141 | } |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 142 | /* fall through */ |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 143 | scope_sess: |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 144 | case SCOPE_SESS: |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 145 | if (var_sess_limit && sess->vars.size + size > var_sess_limit) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 146 | return 0; |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 147 | /* fall through */ |
| 148 | case SCOPE_PROC: |
Willy Tarreau | 3b78f2a | 2021-09-08 15:40:58 +0200 | [diff] [blame] | 149 | /* note: scope proc collects all others and is currently identical to the |
| 150 | * global limit. |
| 151 | */ |
Willy Tarreau | cfc4f24 | 2021-05-08 11:41:28 +0200 | [diff] [blame] | 152 | if (var_proc_limit && proc_vars.size + size > var_proc_limit) |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 153 | return 0; |
Willy Tarreau | 3b78f2a | 2021-09-08 15:40:58 +0200 | [diff] [blame] | 154 | if (var_global_limit && proc_vars.size + size > var_global_limit) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 155 | return 0; |
| 156 | } |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 157 | var_accounting_diff(vars, sess, strm, size); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 158 | return 1; |
| 159 | } |
| 160 | |
Willy Tarreau | c1c88f4 | 2021-09-08 15:03:58 +0200 | [diff] [blame] | 161 | /* This function removes a variable from the list and frees the memory it was |
| 162 | * using. If the variable is marked "VF_PERMANENT", the sample_data is only |
| 163 | * reset to SMP_T_ANY unless <force> is non nul. Returns the freed size. |
| 164 | */ |
| 165 | unsigned int var_clear(struct var *var, int force) |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 166 | { |
| 167 | unsigned int size = 0; |
| 168 | |
| 169 | if (var->data.type == SMP_T_STR || var->data.type == SMP_T_BIN) { |
Willy Tarreau | 5b52b00 | 2021-02-26 21:19:53 +0100 | [diff] [blame] | 170 | ha_free(&var->data.u.str.area); |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 171 | size += var->data.u.str.data; |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 172 | } |
Christopher Faulet | d02210c | 2017-07-24 16:24:39 +0200 | [diff] [blame] | 173 | else if (var->data.type == SMP_T_METH && var->data.u.meth.meth == HTTP_METH_OTHER) { |
Willy Tarreau | 5b52b00 | 2021-02-26 21:19:53 +0100 | [diff] [blame] | 174 | ha_free(&var->data.u.meth.str.area); |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 175 | size += var->data.u.meth.str.data; |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 176 | } |
Willy Tarreau | c1c88f4 | 2021-09-08 15:03:58 +0200 | [diff] [blame] | 177 | /* wipe the sample */ |
| 178 | var->data.type = SMP_T_ANY; |
| 179 | |
| 180 | if (!(var->flags & VF_PERMANENT) || force) { |
| 181 | LIST_DELETE(&var->l); |
| 182 | pool_free(var_pool, var); |
| 183 | size += sizeof(struct var); |
| 184 | } |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 185 | return size; |
| 186 | } |
| 187 | |
Joseph Herlant | 0767689 | 2018-11-15 09:19:50 -0800 | [diff] [blame] | 188 | /* This function free all the memory used by all the variables |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 189 | * in the list. |
| 190 | */ |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 191 | void vars_prune(struct vars *vars, struct session *sess, struct stream *strm) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 192 | { |
| 193 | struct var *var, *tmp; |
Willy Tarreau | 7233098 | 2015-06-19 11:21:56 +0200 | [diff] [blame] | 194 | unsigned int size = 0; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 195 | |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 196 | vars_wrlock(vars); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 197 | list_for_each_entry_safe(var, tmp, &vars->head, l) { |
Willy Tarreau | c1c88f4 | 2021-09-08 15:03:58 +0200 | [diff] [blame] | 198 | size += var_clear(var, 1); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 199 | } |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 200 | vars_wrunlock(vars); |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 201 | var_accounting_diff(vars, sess, strm, -size); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 202 | } |
| 203 | |
Willy Tarreau | ebcd484 | 2015-06-19 11:59:02 +0200 | [diff] [blame] | 204 | /* This function frees all the memory used by all the session variables in the |
| 205 | * list starting at <vars>. |
| 206 | */ |
| 207 | void vars_prune_per_sess(struct vars *vars) |
| 208 | { |
| 209 | struct var *var, *tmp; |
| 210 | unsigned int size = 0; |
| 211 | |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 212 | vars_wrlock(vars); |
Willy Tarreau | ebcd484 | 2015-06-19 11:59:02 +0200 | [diff] [blame] | 213 | list_for_each_entry_safe(var, tmp, &vars->head, l) { |
Willy Tarreau | c1c88f4 | 2021-09-08 15:03:58 +0200 | [diff] [blame] | 214 | size += var_clear(var, 1); |
Willy Tarreau | ebcd484 | 2015-06-19 11:59:02 +0200 | [diff] [blame] | 215 | } |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 216 | vars_wrunlock(vars); |
Christopher Faulet | e95f2c3 | 2017-07-24 16:30:34 +0200 | [diff] [blame] | 217 | |
Willy Tarreau | 55f8a83 | 2021-09-08 15:51:06 +0200 | [diff] [blame] | 218 | if (var_sess_limit) |
| 219 | _HA_ATOMIC_SUB(&vars->size, size); |
| 220 | if (var_proc_limit || var_global_limit) |
| 221 | _HA_ATOMIC_SUB(&proc_vars.size, size); |
Willy Tarreau | ebcd484 | 2015-06-19 11:59:02 +0200 | [diff] [blame] | 222 | } |
| 223 | |
Willy Tarreau | b7bfcb3 | 2021-08-31 08:13:25 +0200 | [diff] [blame] | 224 | /* This function initializes a variables list head */ |
| 225 | void vars_init_head(struct vars *vars, enum vars_scope scope) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 226 | { |
| 227 | LIST_INIT(&vars->head); |
| 228 | vars->scope = scope; |
| 229 | vars->size = 0; |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 230 | HA_RWLOCK_INIT(&vars->rwlock); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 231 | } |
| 232 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 233 | /* This function returns a hash value and a scope for a variable name of a |
| 234 | * specified length. It makes sure that the scope is valid. It returns non-zero |
| 235 | * on success, 0 on failure. Neither hash nor scope may be NULL. |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 236 | */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 237 | static int vars_hash_name(const char *name, int len, enum vars_scope *scope, |
| 238 | uint64_t *hash, char **err) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 239 | { |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 240 | const char *tmp; |
| 241 | |
| 242 | /* Check length. */ |
| 243 | if (len == 0) { |
| 244 | memprintf(err, "Empty variable name cannot be accepted"); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 245 | return 0; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 246 | } |
| 247 | |
| 248 | /* Check scope. */ |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 249 | if (len > 5 && strncmp(name, "proc.", 5) == 0) { |
| 250 | name += 5; |
| 251 | len -= 5; |
| 252 | *scope = SCOPE_PROC; |
| 253 | } |
| 254 | else if (len > 5 && strncmp(name, "sess.", 5) == 0) { |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 255 | name += 5; |
| 256 | len -= 5; |
| 257 | *scope = SCOPE_SESS; |
| 258 | } |
| 259 | else if (len > 4 && strncmp(name, "txn.", 4) == 0) { |
| 260 | name += 4; |
| 261 | len -= 4; |
| 262 | *scope = SCOPE_TXN; |
| 263 | } |
| 264 | else if (len > 4 && strncmp(name, "req.", 4) == 0) { |
| 265 | name += 4; |
| 266 | len -= 4; |
| 267 | *scope = SCOPE_REQ; |
| 268 | } |
| 269 | else if (len > 4 && strncmp(name, "res.", 4) == 0) { |
| 270 | name += 4; |
| 271 | len -= 4; |
| 272 | *scope = SCOPE_RES; |
| 273 | } |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 274 | else if (len > 6 && strncmp(name, "check.", 6) == 0) { |
| 275 | name += 6; |
| 276 | len -= 6; |
| 277 | *scope = SCOPE_CHECK; |
| 278 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 279 | else { |
Willy Tarreau | 1402fef | 2021-09-03 10:12:55 +0200 | [diff] [blame] | 280 | memprintf(err, "invalid variable name '%.*s'. A variable name must be start by its scope. " |
| 281 | "The scope can be 'proc', 'sess', 'txn', 'req', 'res' or 'check'", len, name); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 282 | return 0; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 283 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 284 | |
| 285 | /* Check variable name syntax. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 286 | for (tmp = name; tmp < name + len; tmp++) { |
Willy Tarreau | 9080711 | 2020-02-25 08:16:33 +0100 | [diff] [blame] | 287 | if (!isalnum((unsigned char)*tmp) && *tmp != '_' && *tmp != '.') { |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 288 | memprintf(err, "invalid syntax at char '%s'", tmp); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 289 | return 0; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 290 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 291 | } |
Christopher Faulet | e95f2c3 | 2017-07-24 16:30:34 +0200 | [diff] [blame] | 292 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 293 | *hash = XXH3(name, len, var_name_hash_seed); |
| 294 | return 1; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 295 | } |
| 296 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 297 | /* This function returns the variable from the given list that matches |
| 298 | * <name_hash> or returns NULL if not found. It's only a linked list since it |
| 299 | * is not expected to have many variables per scope (a few tens at best). |
| 300 | * The caller is responsible for ensuring that <vars> is properly locked. |
| 301 | */ |
| 302 | static struct var *var_get(struct vars *vars, uint64_t name_hash) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 303 | { |
| 304 | struct var *var; |
| 305 | |
| 306 | list_for_each_entry(var, &vars->head, l) |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 307 | if (var->name_hash == name_hash) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 308 | return var; |
| 309 | return NULL; |
| 310 | } |
| 311 | |
| 312 | /* Returns 0 if fails, else returns 1. */ |
| 313 | static int smp_fetch_var(const struct arg *args, struct sample *smp, const char *kw, void *private) |
| 314 | { |
| 315 | const struct var_desc *var_desc = &args[0].data.var; |
Willy Tarreau | 54496a6 | 2021-09-03 12:00:13 +0200 | [diff] [blame] | 316 | const struct buffer *def = NULL; |
Christopher Faulet | e95f2c3 | 2017-07-24 16:30:34 +0200 | [diff] [blame] | 317 | |
Willy Tarreau | 54496a6 | 2021-09-03 12:00:13 +0200 | [diff] [blame] | 318 | if (args[1].type == ARGT_STR) |
| 319 | def = &args[1].data.str; |
| 320 | |
| 321 | return vars_get_by_desc(var_desc, smp, def); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 322 | } |
| 323 | |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 324 | /* |
| 325 | * Clear the contents of a variable so that it can be reset directly. |
| 326 | * This function is used just before a variable is filled out of a sample's |
| 327 | * content. |
| 328 | */ |
| 329 | static inline void var_clear_buffer(struct sample *smp, struct vars *vars, struct var *var, int var_type) |
| 330 | { |
| 331 | if (var_type == SMP_T_STR || var_type == SMP_T_BIN) { |
| 332 | ha_free(&var->data.u.str.area); |
| 333 | var_accounting_diff(vars, smp->sess, smp->strm, |
| 334 | -var->data.u.str.data); |
| 335 | } |
| 336 | else if (var_type == SMP_T_METH && var->data.u.meth.meth == HTTP_METH_OTHER) { |
| 337 | ha_free(&var->data.u.meth.str.area); |
| 338 | var_accounting_diff(vars, smp->sess, smp->strm, |
| 339 | -var->data.u.meth.str.data); |
| 340 | } |
| 341 | } |
| 342 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 343 | /* This function tries to create a variable whose name hash is <name_hash> in |
| 344 | * scope <scope> and store sample <smp> as its value. |
| 345 | * |
| 346 | * The stream and session are extracted from <smp>, whose stream may be NULL |
| 347 | * when scope is SCOPE_SESS. In case there wouldn't be enough memory to store |
| 348 | * the sample while the variable was already created, it would be changed to |
| 349 | * a bool (which is memory-less). |
Willy Tarreau | 7978c5c | 2021-09-07 14:24:07 +0200 | [diff] [blame] | 350 | * |
| 351 | * Flags is a bitfield that may contain one of the following flags: |
Willy Tarreau | 4994b57 | 2021-09-08 11:38:25 +0200 | [diff] [blame] | 352 | * - VF_CREATEONLY: do nothing if the variable already exists (success). |
Willy Tarreau | 3dc6dc3 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 353 | * - VF_PERMANENT: this flag will be passed to the variable upon creation |
Willy Tarreau | 7978c5c | 2021-09-07 14:24:07 +0200 | [diff] [blame] | 354 | * |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 355 | * - VF_COND_IFEXISTS: only set variable if it already exists |
| 356 | * - VF_COND_IFNOTEXISTS: only set variable if it did not exist yet |
| 357 | * - VF_COND_IFEMPTY: only set variable if sample is empty |
| 358 | * - VF_COND_IFNOTEMPTY: only set variable if sample is not empty |
| 359 | * - VF_COND_IFSET: only set variable if its type is not SMP_TYPE_ANY |
| 360 | * - VF_COND_IFNOTSET: only set variable if its type is ANY |
| 361 | * - VF_COND_IFGT: only set variable if its value is greater than the sample's |
| 362 | * - VF_COND_IFLT: ony set variable if its value is less than the sample's |
| 363 | * |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 364 | * It returns 0 on failure, non-zero on success. |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 365 | */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 366 | static int var_set(uint64_t name_hash, enum vars_scope scope, struct sample *smp, uint flags) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 367 | { |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 368 | struct vars *vars; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 369 | struct var *var; |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 370 | int ret = 0; |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 371 | int previous_type = SMP_T_ANY; |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 372 | |
| 373 | vars = get_vars(smp->sess, smp->strm, scope); |
| 374 | if (!vars || vars->scope != scope) |
| 375 | return 0; |
| 376 | |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 377 | vars_wrlock(vars); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 378 | |
| 379 | /* Look for existing variable name. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 380 | var = var_get(vars, name_hash); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 381 | |
| 382 | if (var) { |
Willy Tarreau | 4994b57 | 2021-09-08 11:38:25 +0200 | [diff] [blame] | 383 | if (flags & VF_CREATEONLY) { |
| 384 | ret = 1; |
| 385 | goto unlock; |
| 386 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 387 | } else { |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 388 | if (flags & VF_COND_IFEXISTS) |
Willy Tarreau | 7978c5c | 2021-09-07 14:24:07 +0200 | [diff] [blame] | 389 | goto unlock; |
| 390 | |
Joseph Herlant | 0767689 | 2018-11-15 09:19:50 -0800 | [diff] [blame] | 391 | /* Check memory available. */ |
Willy Tarreau | 6204cd9 | 2016-03-10 16:33:04 +0100 | [diff] [blame] | 392 | if (!var_accounting_add(vars, smp->sess, smp->strm, sizeof(struct var))) |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 393 | goto unlock; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 394 | |
| 395 | /* Create new entry. */ |
Willy Tarreau | bafbe01 | 2017-11-24 17:34:44 +0100 | [diff] [blame] | 396 | var = pool_alloc(var_pool); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 397 | if (!var) |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 398 | goto unlock; |
Willy Tarreau | 2b71810 | 2021-04-21 07:32:39 +0200 | [diff] [blame] | 399 | LIST_APPEND(&vars->head, &var->l); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 400 | var->name_hash = name_hash; |
Willy Tarreau | 3dc6dc3 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 401 | var->flags = flags & VF_PERMANENT; |
Remi Tricot-Le Breton | 1bd9805 | 2021-12-16 17:14:35 +0100 | [diff] [blame] | 402 | var->data.type = SMP_T_ANY; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 403 | } |
| 404 | |
| 405 | /* Set type. */ |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 406 | previous_type = var->data.type; |
Thierry FOURNIER | 8c542ca | 2015-08-19 09:00:18 +0200 | [diff] [blame] | 407 | var->data.type = smp->data.type; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 408 | |
| 409 | /* Copy data. If the data needs memory, the function can fail. */ |
| 410 | switch (var->data.type) { |
| 411 | case SMP_T_BOOL: |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 412 | case SMP_T_SINT: |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 413 | var_clear_buffer(smp, vars, var, previous_type); |
Thierry FOURNIER | 136f9d3 | 2015-08-19 09:07:19 +0200 | [diff] [blame] | 414 | var->data.u.sint = smp->data.u.sint; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 415 | break; |
| 416 | case SMP_T_IPV4: |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 417 | var_clear_buffer(smp, vars, var, previous_type); |
Thierry FOURNIER | 136f9d3 | 2015-08-19 09:07:19 +0200 | [diff] [blame] | 418 | var->data.u.ipv4 = smp->data.u.ipv4; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 419 | break; |
| 420 | case SMP_T_IPV6: |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 421 | var_clear_buffer(smp, vars, var, previous_type); |
Thierry FOURNIER | 136f9d3 | 2015-08-19 09:07:19 +0200 | [diff] [blame] | 422 | var->data.u.ipv6 = smp->data.u.ipv6; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 423 | break; |
| 424 | case SMP_T_STR: |
| 425 | case SMP_T_BIN: |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 426 | var_clear_buffer(smp, vars, var, previous_type); |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 427 | if (!var_accounting_add(vars, smp->sess, smp->strm, smp->data.u.str.data)) { |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 428 | var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */ |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 429 | goto unlock; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 430 | } |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 431 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 432 | var->data.u.str.area = malloc(smp->data.u.str.data); |
| 433 | if (!var->data.u.str.area) { |
| 434 | var_accounting_diff(vars, smp->sess, smp->strm, |
| 435 | -smp->data.u.str.data); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 436 | var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */ |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 437 | goto unlock; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 438 | } |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 439 | var->data.u.str.data = smp->data.u.str.data; |
| 440 | memcpy(var->data.u.str.area, smp->data.u.str.area, |
| 441 | var->data.u.str.data); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 442 | break; |
| 443 | case SMP_T_METH: |
Remi Tricot-Le Breton | 25fccd5 | 2021-12-16 17:14:36 +0100 | [diff] [blame] | 444 | var_clear_buffer(smp, vars, var, previous_type); |
Christopher Faulet | d02210c | 2017-07-24 16:24:39 +0200 | [diff] [blame] | 445 | var->data.u.meth.meth = smp->data.u.meth.meth; |
| 446 | if (smp->data.u.meth.meth != HTTP_METH_OTHER) |
| 447 | break; |
| 448 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 449 | if (!var_accounting_add(vars, smp->sess, smp->strm, smp->data.u.meth.str.data)) { |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 450 | var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */ |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 451 | goto unlock; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 452 | } |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 453 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 454 | var->data.u.meth.str.area = malloc(smp->data.u.meth.str.data); |
| 455 | if (!var->data.u.meth.str.area) { |
| 456 | var_accounting_diff(vars, smp->sess, smp->strm, |
| 457 | -smp->data.u.meth.str.data); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 458 | var->data.type = SMP_T_BOOL; /* This type doesn't use additional memory. */ |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 459 | goto unlock; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 460 | } |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 461 | var->data.u.meth.str.data = smp->data.u.meth.str.data; |
| 462 | var->data.u.meth.str.size = smp->data.u.meth.str.data; |
| 463 | memcpy(var->data.u.meth.str.area, smp->data.u.meth.str.area, |
| 464 | var->data.u.meth.str.data); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 465 | break; |
| 466 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 467 | |
Willy Tarreau | f1cb0eb | 2021-09-07 11:37:37 +0200 | [diff] [blame] | 468 | /* OK, now done */ |
| 469 | ret = 1; |
| 470 | unlock: |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 471 | vars_wrunlock(vars); |
Christopher Faulet | e95f2c3 | 2017-07-24 16:30:34 +0200 | [diff] [blame] | 472 | return ret; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 473 | } |
| 474 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 475 | /* Deletes a variable matching name hash <name_hash> and scope <scope> for the |
| 476 | * session and stream found in <smp>. Note that stream may be null for |
| 477 | * SCOPE_SESS. Returns 0 if the scope was not found otherwise 1. |
Willy Tarreau | d378eb8 | 2021-09-07 11:44:41 +0200 | [diff] [blame] | 478 | */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 479 | static int var_unset(uint64_t name_hash, enum vars_scope scope, struct sample *smp) |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 480 | { |
| 481 | struct vars *vars; |
| 482 | struct var *var; |
| 483 | unsigned int size = 0; |
| 484 | |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 485 | vars = get_vars(smp->sess, smp->strm, scope); |
| 486 | if (!vars || vars->scope != scope) |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 487 | return 0; |
| 488 | |
| 489 | /* Look for existing variable name. */ |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 490 | vars_wrlock(vars); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 491 | var = var_get(vars, name_hash); |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 492 | if (var) { |
Willy Tarreau | c1c88f4 | 2021-09-08 15:03:58 +0200 | [diff] [blame] | 493 | size = var_clear(var, 0); |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 494 | var_accounting_diff(vars, smp->sess, smp->strm, -size); |
| 495 | } |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 496 | vars_wrunlock(vars); |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 497 | return 1; |
| 498 | } |
| 499 | |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 500 | |
| 501 | /* |
| 502 | * Convert a string set-var condition into its numerical value. |
| 503 | * The corresponding bit is set in the <cond_bitmap> parameter if the |
| 504 | * <cond> is known. |
| 505 | * Returns 1 in case of success. |
| 506 | */ |
| 507 | static int vars_parse_cond_param(const struct buffer *cond, uint *cond_bitmap, char **err) |
| 508 | { |
| 509 | struct var_set_condition *cond_elt = &conditions_array[0]; |
| 510 | |
| 511 | /* The conditions array is NULL terminated. */ |
| 512 | while (cond_elt->cond_str) { |
| 513 | if (chunk_strcmp(cond, cond_elt->cond_str) == 0) { |
| 514 | *cond_bitmap |= cond_elt->flag; |
| 515 | break; |
| 516 | } |
| 517 | ++cond_elt; |
| 518 | } |
| 519 | |
| 520 | if (cond_elt->cond_str == NULL && err) |
| 521 | memprintf(err, "unknown condition \"%.*s\"", (int)cond->data, cond->area); |
| 522 | |
| 523 | return cond_elt->cond_str != NULL; |
| 524 | } |
| 525 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 526 | /* Returns 0 if fails, else returns 1. */ |
| 527 | static int smp_conv_store(const struct arg *args, struct sample *smp, void *private) |
| 528 | { |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 529 | uint conditions = 0; |
| 530 | int cond_idx = 1; |
| 531 | |
| 532 | while (args[cond_idx].type == ARGT_STR) { |
| 533 | if (vars_parse_cond_param(&args[cond_idx++].data.str, &conditions, NULL) == 0) |
| 534 | break; |
| 535 | } |
| 536 | |
| 537 | return var_set(args[0].data.var.name_hash, args[0].data.var.scope, smp, conditions); |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 538 | } |
| 539 | |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 540 | /* Returns 0 if fails, else returns 1. */ |
| 541 | static int smp_conv_clear(const struct arg *args, struct sample *smp, void *private) |
| 542 | { |
Remi Tricot-Le Breton | bb3e80e | 2021-11-26 18:08:39 +0100 | [diff] [blame] | 543 | return var_unset(args[0].data.var.name_hash, args[0].data.var.scope, smp); |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 544 | } |
| 545 | |
Joseph Herlant | 0767689 | 2018-11-15 09:19:50 -0800 | [diff] [blame] | 546 | /* This functions check an argument entry and fill it with a variable |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 547 | * type. The argumen must be a string. If the variable lookup fails, |
Joseph Herlant | 0767689 | 2018-11-15 09:19:50 -0800 | [diff] [blame] | 548 | * the function returns 0 and fill <err>, otherwise it returns 1. |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 549 | */ |
| 550 | int vars_check_arg(struct arg *arg, char **err) |
| 551 | { |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 552 | enum vars_scope scope; |
Willy Tarreau | df8eeb1 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 553 | struct sample empty_smp = { }; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 554 | uint64_t hash; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 555 | |
| 556 | /* Check arg type. */ |
| 557 | if (arg->type != ARGT_STR) { |
| 558 | memprintf(err, "unexpected argument type"); |
| 559 | return 0; |
| 560 | } |
| 561 | |
| 562 | /* Register new variable name. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 563 | if (!vars_hash_name(arg->data.str.area, arg->data.str.data, &scope, &hash, err)) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 564 | return 0; |
| 565 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 566 | if (scope == SCOPE_PROC && !var_set(hash, scope, &empty_smp, VF_CREATEONLY|VF_PERMANENT)) |
Willy Tarreau | df8eeb1 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 567 | return 0; |
| 568 | |
Tim Duesterhus | a6cc7e8 | 2019-05-13 10:53:29 +0200 | [diff] [blame] | 569 | /* properly destroy the chunk */ |
| 570 | chunk_destroy(&arg->data.str); |
| 571 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 572 | /* Use the global variable name pointer. */ |
| 573 | arg->type = ARGT_VAR; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 574 | arg->data.var.name_hash = hash; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 575 | arg->data.var.scope = scope; |
| 576 | return 1; |
| 577 | } |
| 578 | |
Remi Tricot-Le Breton | 7055301 | 2021-12-16 17:14:34 +0100 | [diff] [blame] | 579 | /* This function stores a sample in a variable unless it is of type "proc" and |
| 580 | * not defined yet. |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 581 | * Returns zero on failure and non-zero otherwise. The variable not being |
| 582 | * defined is treated as a failure. |
Christopher Faulet | 09c9df2 | 2016-10-31 11:05:37 +0100 | [diff] [blame] | 583 | */ |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 584 | int vars_set_by_name_ifexist(const char *name, size_t len, struct sample *smp) |
Christopher Faulet | 09c9df2 | 2016-10-31 11:05:37 +0100 | [diff] [blame] | 585 | { |
| 586 | enum vars_scope scope; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 587 | uint64_t hash; |
Christopher Faulet | 09c9df2 | 2016-10-31 11:05:37 +0100 | [diff] [blame] | 588 | |
| 589 | /* Resolve name and scope. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 590 | if (!vars_hash_name(name, len, &scope, &hash, NULL)) |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 591 | return 0; |
Christopher Faulet | 09c9df2 | 2016-10-31 11:05:37 +0100 | [diff] [blame] | 592 | |
Remi Tricot-Le Breton | 7055301 | 2021-12-16 17:14:34 +0100 | [diff] [blame] | 593 | /* Variable creation is allowed for all scopes apart from the PROC one. */ |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 594 | return var_set(hash, scope, smp, (scope == SCOPE_PROC) ? VF_COND_IFEXISTS : 0); |
Christopher Faulet | 09c9df2 | 2016-10-31 11:05:37 +0100 | [diff] [blame] | 595 | } |
| 596 | |
| 597 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 598 | /* This function stores a sample in a variable. |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 599 | * Returns zero on failure and non-zero otherwise. |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 600 | */ |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 601 | int vars_set_by_name(const char *name, size_t len, struct sample *smp) |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 602 | { |
| 603 | enum vars_scope scope; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 604 | uint64_t hash; |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 605 | |
| 606 | /* Resolve name and scope. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 607 | if (!vars_hash_name(name, len, &scope, &hash, NULL)) |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 608 | return 0; |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 609 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 610 | return var_set(hash, scope, smp, 0); |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 611 | } |
| 612 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 613 | /* This function unsets a variable if it was already defined. |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 614 | * Returns zero on failure and non-zero otherwise. |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 615 | */ |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 616 | int vars_unset_by_name_ifexist(const char *name, size_t len, struct sample *smp) |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 617 | { |
| 618 | enum vars_scope scope; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 619 | uint64_t hash; |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 620 | |
| 621 | /* Resolve name and scope. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 622 | if (!vars_hash_name(name, len, &scope, &hash, NULL)) |
Tim Duesterhus | b4fac1e | 2020-05-19 13:49:40 +0200 | [diff] [blame] | 623 | return 0; |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 624 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 625 | return var_unset(hash, scope, smp); |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 626 | } |
| 627 | |
| 628 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 629 | /* This retrieves variable whose hash matches <name_hash> from variables <vars>, |
| 630 | * and if found and not empty, duplicates the result into sample <smp>. |
| 631 | * smp_dup() is used in order to release the variables lock ASAP (so a pre- |
| 632 | * allocated chunk is obtained via get_trash_shunk()). The variables' lock is |
| 633 | * used for reads. |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 634 | * |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 635 | * The function returns 0 if the variable was not found and no default |
| 636 | * value was provided in <def>, otherwise 1 with the sample filled. |
| 637 | * Default values are always returned as strings. |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 638 | */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 639 | static int var_to_smp(struct vars *vars, uint64_t name_hash, struct sample *smp, const struct buffer *def) |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 640 | { |
| 641 | struct var *var; |
| 642 | |
| 643 | /* Get the variable entry. */ |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 644 | vars_rdlock(vars); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 645 | var = var_get(vars, name_hash); |
Willy Tarreau | 63c3066 | 2021-09-08 13:58:19 +0200 | [diff] [blame] | 646 | if (!var || !var->data.type) { |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 647 | if (!def) { |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 648 | vars_rdunlock(vars); |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 649 | return 0; |
| 650 | } |
| 651 | |
| 652 | /* not found but we have a default value */ |
| 653 | smp->data.type = SMP_T_STR; |
| 654 | smp->data.u.str = *def; |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 655 | } |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 656 | else |
| 657 | smp->data = var->data; |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 658 | |
| 659 | /* Copy sample. */ |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 660 | smp_dup(smp); |
| 661 | |
Willy Tarreau | dc72fbb | 2021-09-08 15:19:57 +0200 | [diff] [blame] | 662 | vars_rdunlock(vars); |
Willy Tarreau | be7e00d | 2021-09-03 11:40:58 +0200 | [diff] [blame] | 663 | return 1; |
| 664 | } |
| 665 | |
Dragan Dosen | 14518f2 | 2021-02-22 17:20:01 +0100 | [diff] [blame] | 666 | /* This function fills a sample with the variable content. |
| 667 | * |
| 668 | * Keep in mind that a sample content is duplicated by using smp_dup() |
| 669 | * and it therefore uses a pre-allocated trash chunk as returned by |
| 670 | * get_trash_chunk(). |
| 671 | * |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 672 | * If the variable is not valid in this scope, 0 is always returned. |
| 673 | * If the variable is valid but not found, either the default value |
| 674 | * <def> is returned if not NULL, or zero is returned. |
| 675 | * |
Dragan Dosen | 14518f2 | 2021-02-22 17:20:01 +0100 | [diff] [blame] | 676 | * Returns 1 if the sample is filled, otherwise it returns 0. |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 677 | */ |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 678 | int vars_get_by_name(const char *name, size_t len, struct sample *smp, const struct buffer *def) |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 679 | { |
| 680 | struct vars *vars; |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 681 | enum vars_scope scope; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 682 | uint64_t hash; |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 683 | |
| 684 | /* Resolve name and scope. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 685 | if (!vars_hash_name(name, len, &scope, &hash, NULL)) |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 686 | return 0; |
| 687 | |
| 688 | /* Select "vars" pool according with the scope. */ |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 689 | vars = get_vars(smp->sess, smp->strm, scope); |
| 690 | if (!vars || vars->scope != scope) |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 691 | return 0; |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 692 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 693 | return var_to_smp(vars, hash, smp, def); |
Thierry FOURNIER | c365d99 | 2015-06-09 12:27:17 +0200 | [diff] [blame] | 694 | } |
| 695 | |
Dragan Dosen | 14518f2 | 2021-02-22 17:20:01 +0100 | [diff] [blame] | 696 | /* This function fills a sample with the content of the variable described |
| 697 | * by <var_desc>. |
| 698 | * |
| 699 | * Keep in mind that a sample content is duplicated by using smp_dup() |
| 700 | * and it therefore uses a pre-allocated trash chunk as returned by |
| 701 | * get_trash_chunk(). |
| 702 | * |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 703 | * If the variable is not valid in this scope, 0 is always returned. |
| 704 | * If the variable is valid but not found, either the default value |
| 705 | * <def> is returned if not NULL, or zero is returned. |
| 706 | * |
Dragan Dosen | 14518f2 | 2021-02-22 17:20:01 +0100 | [diff] [blame] | 707 | * Returns 1 if the sample is filled, otherwise it returns 0. |
Thierry FOURNIER | fd77e05 | 2015-07-07 21:20:42 +0200 | [diff] [blame] | 708 | */ |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 709 | int vars_get_by_desc(const struct var_desc *var_desc, struct sample *smp, const struct buffer *def) |
Thierry FOURNIER | fd77e05 | 2015-07-07 21:20:42 +0200 | [diff] [blame] | 710 | { |
| 711 | struct vars *vars; |
Thierry FOURNIER | fd77e05 | 2015-07-07 21:20:42 +0200 | [diff] [blame] | 712 | |
| 713 | /* Select "vars" pool according with the scope. */ |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 714 | vars = get_vars(smp->sess, smp->strm, var_desc->scope); |
Thierry FOURNIER | fd77e05 | 2015-07-07 21:20:42 +0200 | [diff] [blame] | 715 | |
Joseph Herlant | 0767689 | 2018-11-15 09:19:50 -0800 | [diff] [blame] | 716 | /* Check if the scope is available a this point of processing. */ |
Willy Tarreau | f37b140 | 2019-06-04 16:27:36 +0200 | [diff] [blame] | 717 | if (!vars || vars->scope != var_desc->scope) |
Thierry FOURNIER | fd77e05 | 2015-07-07 21:20:42 +0200 | [diff] [blame] | 718 | return 0; |
| 719 | |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 720 | return var_to_smp(vars, var_desc->name_hash, smp, def); |
Thierry FOURNIER | fd77e05 | 2015-07-07 21:20:42 +0200 | [diff] [blame] | 721 | } |
| 722 | |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 723 | /* Always returns ACT_RET_CONT even if an error occurs. */ |
| 724 | static enum act_return action_store(struct act_rule *rule, struct proxy *px, |
Willy Tarreau | 658b85b | 2015-09-27 10:00:49 +0200 | [diff] [blame] | 725 | struct session *sess, struct stream *s, int flags) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 726 | { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 727 | struct buffer *fmtstr = NULL; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 728 | struct sample smp; |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 729 | int dir; |
| 730 | |
| 731 | switch (rule->from) { |
Jaroslaw Rzeszótko | c863703 | 2021-11-02 16:56:05 +0100 | [diff] [blame] | 732 | case ACT_F_TCP_REQ_CON: dir = SMP_OPT_DIR_REQ; break; |
Willy Tarreau | 620408f | 2016-10-21 16:37:51 +0200 | [diff] [blame] | 733 | case ACT_F_TCP_REQ_SES: dir = SMP_OPT_DIR_REQ; break; |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 734 | case ACT_F_TCP_REQ_CNT: dir = SMP_OPT_DIR_REQ; break; |
| 735 | case ACT_F_TCP_RES_CNT: dir = SMP_OPT_DIR_RES; break; |
| 736 | case ACT_F_HTTP_REQ: dir = SMP_OPT_DIR_REQ; break; |
| 737 | case ACT_F_HTTP_RES: dir = SMP_OPT_DIR_RES; break; |
Gaetan Rivet | 0c39ecc | 2020-02-24 17:34:11 +0100 | [diff] [blame] | 738 | case ACT_F_TCP_CHK: dir = SMP_OPT_DIR_REQ; break; |
Willy Tarreau | 01d580a | 2021-03-26 11:11:34 +0100 | [diff] [blame] | 739 | case ACT_F_CFG_PARSER: dir = SMP_OPT_DIR_REQ; break; /* not used anyway */ |
Willy Tarreau | 2f836de | 2021-03-26 15:36:44 +0100 | [diff] [blame] | 740 | case ACT_F_CLI_PARSER: dir = SMP_OPT_DIR_REQ; break; /* not used anyway */ |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 741 | default: |
| 742 | send_log(px, LOG_ERR, "Vars: internal error while execute action store."); |
| 743 | if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE)) |
Christopher Faulet | 767a84b | 2017-11-24 16:50:31 +0100 | [diff] [blame] | 744 | ha_alert("Vars: internal error while execute action store.\n"); |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 745 | return ACT_RET_CONT; |
| 746 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 747 | |
| 748 | /* Process the expression. */ |
| 749 | memset(&smp, 0, sizeof(smp)); |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 750 | |
| 751 | if (!LIST_ISEMPTY(&rule->arg.vars.fmt)) { |
| 752 | /* a format-string is used */ |
| 753 | |
| 754 | fmtstr = alloc_trash_chunk(); |
| 755 | if (!fmtstr) { |
| 756 | send_log(px, LOG_ERR, "Vars: memory allocation failure while processing store rule."); |
| 757 | if (!(global.mode & MODE_QUIET) || (global.mode & MODE_VERBOSE)) |
| 758 | ha_alert("Vars: memory allocation failure while processing store rule.\n"); |
| 759 | return ACT_RET_CONT; |
| 760 | } |
| 761 | |
| 762 | /* execute the log-format expression */ |
| 763 | fmtstr->data = sess_build_logline(sess, s, fmtstr->area, fmtstr->size, &rule->arg.vars.fmt); |
| 764 | |
| 765 | /* convert it to a sample of type string as it's what the vars |
| 766 | * API consumes, and store it. |
| 767 | */ |
| 768 | smp_set_owner(&smp, px, sess, s, 0); |
| 769 | smp.data.type = SMP_T_STR; |
| 770 | smp.data.u.str = *fmtstr; |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 771 | var_set(rule->arg.vars.name_hash, rule->arg.vars.scope, &smp, 0); |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 772 | } |
| 773 | else { |
| 774 | /* an expression is used */ |
| 775 | if (!sample_process(px, sess, s, dir|SMP_OPT_FINAL, |
| 776 | rule->arg.vars.expr, &smp)) |
| 777 | return ACT_RET_CONT; |
| 778 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 779 | |
| 780 | /* Store the sample, and ignore errors. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 781 | var_set(rule->arg.vars.name_hash, rule->arg.vars.scope, &smp, 0); |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 782 | free_trash_chunk(fmtstr); |
Thierry FOURNIER | 24ff6c6 | 2015-08-06 08:52:53 +0200 | [diff] [blame] | 783 | return ACT_RET_CONT; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 784 | } |
| 785 | |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 786 | /* Always returns ACT_RET_CONT even if an error occurs. */ |
| 787 | static enum act_return action_clear(struct act_rule *rule, struct proxy *px, |
| 788 | struct session *sess, struct stream *s, int flags) |
| 789 | { |
| 790 | struct sample smp; |
| 791 | |
| 792 | memset(&smp, 0, sizeof(smp)); |
| 793 | smp_set_owner(&smp, px, sess, s, SMP_OPT_FINAL); |
| 794 | |
| 795 | /* Clear the variable using the sample context, and ignore errors. */ |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 796 | var_unset(rule->arg.vars.name_hash, rule->arg.vars.scope, &smp); |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 797 | return ACT_RET_CONT; |
| 798 | } |
| 799 | |
Tim Duesterhus | 01a0ce3 | 2020-06-14 17:27:36 +0200 | [diff] [blame] | 800 | static void release_store_rule(struct act_rule *rule) |
| 801 | { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 802 | struct logformat_node *lf, *lfb; |
Willy Tarreau | c77bad2 | 2021-09-03 10:58:07 +0200 | [diff] [blame] | 803 | |
| 804 | list_for_each_entry_safe(lf, lfb, &rule->arg.vars.fmt, list) { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 805 | LIST_DELETE(&lf->list); |
| 806 | release_sample_expr(lf->expr); |
| 807 | free(lf->arg); |
| 808 | free(lf); |
| 809 | } |
| 810 | |
Tim Duesterhus | 01a0ce3 | 2020-06-14 17:27:36 +0200 | [diff] [blame] | 811 | release_sample_expr(rule->arg.vars.expr); |
| 812 | } |
| 813 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 814 | /* This two function checks the variable name and replace the |
| 815 | * configuration string name by the global string name. its |
| 816 | * the same string, but the global pointer can be easy to |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 817 | * compare. They return non-zero on success, zero on failure. |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 818 | * |
| 819 | * The first function checks a sample-fetch and the second |
| 820 | * checks a converter. |
| 821 | */ |
| 822 | static int smp_check_var(struct arg *args, char **err) |
| 823 | { |
| 824 | return vars_check_arg(&args[0], err); |
| 825 | } |
| 826 | |
| 827 | static int conv_check_var(struct arg *args, struct sample_conv *conv, |
| 828 | const char *file, int line, char **err_msg) |
| 829 | { |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 830 | int cond_idx = 1; |
| 831 | uint conditions = 0; |
| 832 | int retval = vars_check_arg(&args[0], err_msg); |
| 833 | |
| 834 | while (retval && args[cond_idx].type == ARGT_STR) |
| 835 | retval = vars_parse_cond_param(&args[cond_idx++].data.str, &conditions, err_msg); |
| 836 | |
| 837 | return retval; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 838 | } |
| 839 | |
| 840 | /* This function is a common parser for using variables. It understands |
| 841 | * the format: |
| 842 | * |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 843 | * set-var-fmt(<variable-name>) <format-string> |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 844 | * set-var(<variable-name>) <expression> |
Willy Tarreau | 4b7531f | 2019-06-04 16:43:29 +0200 | [diff] [blame] | 845 | * unset-var(<variable-name>) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 846 | * |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 847 | * It returns ACT_RET_PRS_ERR if fails and <err> is filled with an error |
| 848 | * message. Otherwise, it returns ACT_RET_PRS_OK and the variable <expr> |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 849 | * is filled with the pointer to the expression to execute. The proxy is |
| 850 | * only used to retrieve the ->conf entries. |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 851 | */ |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 852 | static enum act_parse_ret parse_store(const char **args, int *arg, struct proxy *px, |
| 853 | struct act_rule *rule, char **err) |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 854 | { |
| 855 | const char *var_name = args[*arg-1]; |
| 856 | int var_len; |
Thierry FOURNIER | 48a9cd1 | 2015-07-28 19:00:28 +0200 | [diff] [blame] | 857 | const char *kw_name; |
Christopher Faulet | 7a06ffb | 2021-10-13 17:22:17 +0200 | [diff] [blame] | 858 | int flags = 0, set_var = 0; /* 0=unset-var, 1=set-var, 2=set-var-fmt */ |
Willy Tarreau | df8eeb1 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 859 | struct sample empty_smp = { }; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 860 | |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 861 | if (strncmp(var_name, "set-var-fmt", 11) == 0) { |
| 862 | var_name += 11; |
| 863 | set_var = 2; |
| 864 | } |
| 865 | else if (strncmp(var_name, "set-var", 7) == 0) { |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 866 | var_name += 7; |
| 867 | set_var = 1; |
| 868 | } |
Willy Tarreau | 2819210 | 2021-09-02 18:46:22 +0200 | [diff] [blame] | 869 | else if (strncmp(var_name, "unset-var", 9) == 0) { |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 870 | var_name += 9; |
| 871 | set_var = 0; |
| 872 | } |
| 873 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 874 | if (*var_name != '(') { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 875 | memprintf(err, "invalid or incomplete action '%s'. Expects 'set-var(<var-name>)', 'set-var-fmt(<var-name>)' or 'unset-var(<var-name>)'", |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 876 | args[*arg-1]); |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 877 | return ACT_RET_PRS_ERR; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 878 | } |
| 879 | var_name++; /* jump the '(' */ |
| 880 | var_len = strlen(var_name); |
| 881 | var_len--; /* remove the ')' */ |
| 882 | if (var_name[var_len] != ')') { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 883 | memprintf(err, "incomplete argument after action '%s'. Expects 'set-var(<var-name>)', 'set-var-fmt(<var-name>)' or 'unset-var(<var-name>)'", |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 884 | args[*arg-1]); |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 885 | return ACT_RET_PRS_ERR; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 886 | } |
| 887 | |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 888 | LIST_INIT(&rule->arg.vars.fmt); |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 889 | if (!vars_hash_name(var_name, var_len, &rule->arg.vars.scope, &rule->arg.vars.name_hash, err)) |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 890 | return ACT_RET_PRS_ERR; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 891 | |
Willy Tarreau | df8eeb1 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 892 | if (rule->arg.vars.scope == SCOPE_PROC && |
Willy Tarreau | 3a4bedc | 2021-08-31 08:51:02 +0200 | [diff] [blame] | 893 | !var_set(rule->arg.vars.name_hash, rule->arg.vars.scope, &empty_smp, VF_CREATEONLY|VF_PERMANENT)) |
Willy Tarreau | df8eeb1 | 2021-09-08 11:07:32 +0200 | [diff] [blame] | 894 | return 0; |
| 895 | |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 896 | /* There is no fetch method when variable is unset. Just set the right |
| 897 | * action and return. */ |
| 898 | if (!set_var) { |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 899 | rule->action = ACT_CUSTOM; |
| 900 | rule->action_ptr = action_clear; |
Tim Duesterhus | 01a0ce3 | 2020-06-14 17:27:36 +0200 | [diff] [blame] | 901 | rule->release_ptr = release_store_rule; |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 902 | return ACT_RET_PRS_OK; |
| 903 | } |
| 904 | |
Thierry FOURNIER | 48a9cd1 | 2015-07-28 19:00:28 +0200 | [diff] [blame] | 905 | kw_name = args[*arg-1]; |
| 906 | |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 907 | switch (rule->from) { |
Jaroslaw Rzeszótko | c863703 | 2021-11-02 16:56:05 +0100 | [diff] [blame] | 908 | case ACT_F_TCP_REQ_CON: |
| 909 | flags = SMP_VAL_FE_CON_ACC; |
| 910 | px->conf.args.ctx = ARGC_TCO; |
| 911 | break; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 912 | case ACT_F_TCP_REQ_SES: |
| 913 | flags = SMP_VAL_FE_SES_ACC; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 914 | px->conf.args.ctx = ARGC_TSE; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 915 | break; |
| 916 | case ACT_F_TCP_REQ_CNT: |
Christopher Faulet | 7a06ffb | 2021-10-13 17:22:17 +0200 | [diff] [blame] | 917 | if (px->cap & PR_CAP_FE) |
| 918 | flags |= SMP_VAL_FE_REQ_CNT; |
| 919 | if (px->cap & PR_CAP_BE) |
| 920 | flags |= SMP_VAL_BE_REQ_CNT; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 921 | px->conf.args.ctx = ARGC_TRQ; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 922 | break; |
| 923 | case ACT_F_TCP_RES_CNT: |
Christopher Faulet | 7a06ffb | 2021-10-13 17:22:17 +0200 | [diff] [blame] | 924 | if (px->cap & PR_CAP_FE) |
| 925 | flags |= SMP_VAL_FE_RES_CNT; |
| 926 | if (px->cap & PR_CAP_BE) |
| 927 | flags |= SMP_VAL_BE_RES_CNT; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 928 | px->conf.args.ctx = ARGC_TRS; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 929 | break; |
| 930 | case ACT_F_HTTP_REQ: |
Christopher Faulet | 7a06ffb | 2021-10-13 17:22:17 +0200 | [diff] [blame] | 931 | if (px->cap & PR_CAP_FE) |
| 932 | flags |= SMP_VAL_FE_HRQ_HDR; |
| 933 | if (px->cap & PR_CAP_BE) |
| 934 | flags |= SMP_VAL_BE_HRQ_HDR; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 935 | px->conf.args.ctx = ARGC_HRQ; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 936 | break; |
| 937 | case ACT_F_HTTP_RES: |
Christopher Faulet | 7a06ffb | 2021-10-13 17:22:17 +0200 | [diff] [blame] | 938 | if (px->cap & PR_CAP_FE) |
| 939 | flags |= SMP_VAL_FE_HRS_HDR; |
| 940 | if (px->cap & PR_CAP_BE) |
| 941 | flags |= SMP_VAL_BE_HRS_HDR; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 942 | px->conf.args.ctx = ARGC_HRS; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 943 | break; |
| 944 | case ACT_F_TCP_CHK: |
| 945 | flags = SMP_VAL_BE_CHK_RUL; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 946 | px->conf.args.ctx = ARGC_TCK; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 947 | break; |
| 948 | case ACT_F_CFG_PARSER: |
| 949 | flags = SMP_VAL_CFG_PARSER; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 950 | px->conf.args.ctx = ARGC_CFG; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 951 | break; |
| 952 | case ACT_F_CLI_PARSER: |
| 953 | flags = SMP_VAL_CLI_PARSER; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 954 | px->conf.args.ctx = ARGC_CLI; |
Willy Tarreau | 843096d | 2021-09-02 19:03:07 +0200 | [diff] [blame] | 955 | break; |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 956 | default: |
| 957 | memprintf(err, |
| 958 | "internal error, unexpected rule->from=%d, please report this bug!", |
| 959 | rule->from); |
Thierry FOURNIER | afa8049 | 2015-08-19 09:04:15 +0200 | [diff] [blame] | 960 | return ACT_RET_PRS_ERR; |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 961 | } |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 962 | |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 963 | if (set_var == 2) { /* set-var-fmt */ |
| 964 | if (!parse_logformat_string(args[*arg], px, &rule->arg.vars.fmt, 0, flags, err)) |
| 965 | return ACT_RET_PRS_ERR; |
Willy Tarreau | 54b96d9 | 2021-09-02 19:46:08 +0200 | [diff] [blame] | 966 | |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 967 | (*arg)++; |
| 968 | |
| 969 | /* for late error reporting */ |
| 970 | free(px->conf.lfs_file); |
| 971 | px->conf.lfs_file = strdup(px->conf.args.file); |
| 972 | px->conf.lfs_line = px->conf.args.line; |
| 973 | } else { |
| 974 | /* set-var */ |
| 975 | rule->arg.vars.expr = sample_parse_expr((char **)args, arg, px->conf.args.file, |
Christopher Faulet | 6ff7de5 | 2021-10-13 15:18:36 +0200 | [diff] [blame] | 976 | px->conf.args.line, err, &px->conf.args, NULL); |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 977 | if (!rule->arg.vars.expr) |
| 978 | return ACT_RET_PRS_ERR; |
| 979 | |
| 980 | if (!(rule->arg.vars.expr->fetch->val & flags)) { |
| 981 | memprintf(err, |
| 982 | "fetch method '%s' extracts information from '%s', none of which is available here", |
| 983 | kw_name, sample_src_names(rule->arg.vars.expr->fetch->use)); |
| 984 | free(rule->arg.vars.expr); |
| 985 | return ACT_RET_PRS_ERR; |
| 986 | } |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 987 | } |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 988 | |
Thierry FOURNIER | 4214873 | 2015-09-02 17:17:33 +0200 | [diff] [blame] | 989 | rule->action = ACT_CUSTOM; |
Thierry FOURNIER | 4dc15d1 | 2015-08-06 18:25:56 +0200 | [diff] [blame] | 990 | rule->action_ptr = action_store; |
Tim Duesterhus | 01a0ce3 | 2020-06-14 17:27:36 +0200 | [diff] [blame] | 991 | rule->release_ptr = release_store_rule; |
Thierry FOURNIER | afa8049 | 2015-08-19 09:04:15 +0200 | [diff] [blame] | 992 | return ACT_RET_PRS_OK; |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 993 | } |
| 994 | |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 995 | |
| 996 | /* parses a global "set-var" directive. It will create a temporary rule and |
| 997 | * expression that are parsed, processed, and released on the fly so that we |
| 998 | * respect the real set-var syntax. These directives take the following format: |
| 999 | * set-var <name> <expression> |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1000 | * set-var-fmt <name> <fmt> |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1001 | * Note that parse_store() expects "set-var(name) <expression>" so we have to |
| 1002 | * temporarily replace the keyword here. |
| 1003 | */ |
| 1004 | static int vars_parse_global_set_var(char **args, int section_type, struct proxy *curpx, |
| 1005 | const struct proxy *defpx, const char *file, int line, |
| 1006 | char **err) |
| 1007 | { |
| 1008 | struct proxy px = { |
Willy Tarreau | 9c20433 | 2021-09-03 08:19:43 +0200 | [diff] [blame] | 1009 | .id = "CFG", |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1010 | .conf.args.file = file, |
| 1011 | .conf.args.line = line, |
| 1012 | }; |
| 1013 | struct act_rule rule = { |
| 1014 | .arg.vars.scope = SCOPE_PROC, |
| 1015 | .from = ACT_F_CFG_PARSER, |
Willy Tarreau | c9e4868 | 2021-10-11 09:13:07 +0200 | [diff] [blame] | 1016 | .conf.file = (char *)file, |
| 1017 | .conf.line = line, |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1018 | }; |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1019 | enum obj_type objt = OBJ_TYPE_NONE; |
| 1020 | struct session *sess = NULL; |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1021 | enum act_parse_ret p_ret; |
| 1022 | char *old_arg1; |
| 1023 | char *tmp_arg1; |
| 1024 | int arg = 2; // variable name |
| 1025 | int ret = -1; |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1026 | int use_fmt = 0; |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1027 | |
| 1028 | LIST_INIT(&px.conf.args.list); |
| 1029 | |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1030 | use_fmt = strcmp(args[0], "set-var-fmt") == 0; |
| 1031 | |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1032 | if (!*args[1] || !*args[2]) { |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1033 | if (use_fmt) |
| 1034 | memprintf(err, "'%s' requires a process-wide variable name ('proc.<name>') and a format string.", args[0]); |
| 1035 | else |
| 1036 | memprintf(err, "'%s' requires a process-wide variable name ('proc.<name>') and a sample expression.", args[0]); |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1037 | goto end; |
| 1038 | } |
| 1039 | |
| 1040 | tmp_arg1 = NULL; |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1041 | if (!memprintf(&tmp_arg1, "set-var%s(%s)", use_fmt ? "-fmt" : "", args[1])) |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1042 | goto end; |
| 1043 | |
| 1044 | /* parse_store() will always return a message in <err> on error */ |
| 1045 | old_arg1 = args[1]; args[1] = tmp_arg1; |
| 1046 | p_ret = parse_store((const char **)args, &arg, &px, &rule, err); |
| 1047 | free(args[1]); args[1] = old_arg1; |
| 1048 | |
| 1049 | if (p_ret != ACT_RET_PRS_OK) |
| 1050 | goto end; |
| 1051 | |
| 1052 | if (rule.arg.vars.scope != SCOPE_PROC) { |
| 1053 | memprintf(err, "'%s': cannot set variable '%s', only scope 'proc' is permitted in the global section.", args[0], args[1]); |
| 1054 | goto end; |
| 1055 | } |
| 1056 | |
| 1057 | if (smp_resolve_args(&px, err) != 0) { |
| 1058 | release_sample_expr(rule.arg.vars.expr); |
| 1059 | indent_msg(err, 2); |
| 1060 | goto end; |
| 1061 | } |
| 1062 | |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1063 | if (use_fmt && !(sess = session_new(&px, NULL, &objt))) { |
| 1064 | release_sample_expr(rule.arg.vars.expr); |
| 1065 | memprintf(err, "'%s': out of memory when trying to set variable '%s' in the global section.", args[0], args[1]); |
| 1066 | goto end; |
| 1067 | } |
| 1068 | |
| 1069 | action_store(&rule, &px, sess, NULL, 0); |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1070 | release_sample_expr(rule.arg.vars.expr); |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1071 | if (sess) |
| 1072 | session_free(sess); |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1073 | |
| 1074 | ret = 0; |
| 1075 | end: |
| 1076 | return ret; |
| 1077 | } |
| 1078 | |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 1079 | /* parse CLI's "get var <name>" */ |
| 1080 | static int vars_parse_cli_get_var(char **args, char *payload, struct appctx *appctx, void *private) |
| 1081 | { |
| 1082 | struct vars *vars; |
Willy Tarreau | 374edc7 | 2021-04-01 17:01:43 +0200 | [diff] [blame] | 1083 | struct sample smp = { }; |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 1084 | int i; |
| 1085 | |
| 1086 | if (!cli_has_level(appctx, ACCESS_LVL_OPER)) |
| 1087 | return 1; |
| 1088 | |
| 1089 | if (!*args[2]) |
| 1090 | return cli_err(appctx, "Missing process-wide variable identifier.\n"); |
| 1091 | |
| 1092 | vars = get_vars(NULL, NULL, SCOPE_PROC); |
| 1093 | if (!vars || vars->scope != SCOPE_PROC) |
| 1094 | return 0; |
| 1095 | |
Willy Tarreau | e352b9d | 2021-09-03 11:52:38 +0200 | [diff] [blame] | 1096 | if (!vars_get_by_name(args[2], strlen(args[2]), &smp, NULL)) |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 1097 | return cli_err(appctx, "Variable not found.\n"); |
| 1098 | |
| 1099 | /* the sample returned by vars_get_by_name() is allocated into a trash |
| 1100 | * chunk so we have no constraint to manipulate it. |
| 1101 | */ |
| 1102 | chunk_printf(&trash, "%s: type=%s value=", args[2], smp_to_type[smp.data.type]); |
| 1103 | |
| 1104 | if (!sample_casts[smp.data.type][SMP_T_STR] || |
| 1105 | !sample_casts[smp.data.type][SMP_T_STR](&smp)) { |
| 1106 | chunk_appendf(&trash, "(undisplayable)"); |
| 1107 | } else { |
| 1108 | /* Display the displayable chars*. */ |
| 1109 | b_putchr(&trash, '<'); |
| 1110 | for (i = 0; i < smp.data.u.str.data; i++) { |
| 1111 | if (isprint((unsigned char)smp.data.u.str.area[i])) |
| 1112 | b_putchr(&trash, smp.data.u.str.area[i]); |
| 1113 | else |
| 1114 | b_putchr(&trash, '.'); |
| 1115 | } |
| 1116 | b_putchr(&trash, '>'); |
| 1117 | b_putchr(&trash, 0); |
| 1118 | } |
| 1119 | return cli_msg(appctx, LOG_INFO, trash.area); |
| 1120 | } |
| 1121 | |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1122 | /* parse CLI's "set var <name>". It accepts: |
| 1123 | * - set var <name> <expression> |
| 1124 | * - set var <name> expr <expression> |
| 1125 | * - set var <name> fmt <format> |
| 1126 | */ |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1127 | static int vars_parse_cli_set_var(char **args, char *payload, struct appctx *appctx, void *private) |
| 1128 | { |
| 1129 | struct proxy px = { |
| 1130 | .id = "CLI", |
| 1131 | .conf.args.file = "CLI", |
| 1132 | .conf.args.line = 0, |
| 1133 | }; |
| 1134 | struct act_rule rule = { |
| 1135 | .arg.vars.scope = SCOPE_PROC, |
| 1136 | .from = ACT_F_CLI_PARSER, |
Willy Tarreau | c9e4868 | 2021-10-11 09:13:07 +0200 | [diff] [blame] | 1137 | .conf.file = "CLI", |
| 1138 | .conf.line = 0, |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1139 | }; |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1140 | enum obj_type objt = OBJ_TYPE_NONE; |
| 1141 | struct session *sess = NULL; |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1142 | enum act_parse_ret p_ret; |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1143 | const char *tmp_args[3]; |
| 1144 | int tmp_arg; |
| 1145 | char *tmp_act; |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1146 | char *err = NULL; |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1147 | int nberr; |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1148 | int use_fmt = 0; |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1149 | |
| 1150 | LIST_INIT(&px.conf.args.list); |
| 1151 | |
| 1152 | if (!cli_has_level(appctx, ACCESS_LVL_OPER)) |
| 1153 | return 1; |
| 1154 | |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1155 | if (!*args[2]) |
| 1156 | return cli_err(appctx, "Missing process-wide variable identifier.\n"); |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1157 | |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1158 | if (!*args[3]) |
| 1159 | return cli_err(appctx, "Missing either 'expr', 'fmt' or expression.\n"); |
| 1160 | |
| 1161 | if (*args[4]) { |
| 1162 | /* this is the long format */ |
| 1163 | if (strcmp(args[3], "fmt") == 0) |
| 1164 | use_fmt = 1; |
| 1165 | else if (strcmp(args[3], "expr") != 0) { |
| 1166 | memprintf(&err, "'%s %s': arg type must be either 'expr' or 'fmt' but got '%s'.", args[0], args[1], args[3]); |
| 1167 | goto fail; |
| 1168 | } |
| 1169 | } |
| 1170 | |
| 1171 | tmp_act = NULL; |
| 1172 | if (!memprintf(&tmp_act, "set-var%s(%s)", use_fmt ? "-fmt" : "", args[2])) { |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1173 | memprintf(&err, "memory allocation error."); |
| 1174 | goto fail; |
| 1175 | } |
| 1176 | |
| 1177 | /* parse_store() will always return a message in <err> on error */ |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1178 | tmp_args[0] = tmp_act; |
| 1179 | tmp_args[1] = (*args[4]) ? args[4] : args[3]; |
| 1180 | tmp_args[2] = ""; |
| 1181 | tmp_arg = 1; // must point to the first arg after the action |
| 1182 | p_ret = parse_store(tmp_args, &tmp_arg, &px, &rule, &err); |
| 1183 | free(tmp_act); |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1184 | |
| 1185 | if (p_ret != ACT_RET_PRS_OK) |
| 1186 | goto fail; |
| 1187 | |
| 1188 | if (rule.arg.vars.scope != SCOPE_PROC) { |
Willy Tarreau | c767eeb | 2021-09-03 10:23:26 +0200 | [diff] [blame] | 1189 | memprintf(&err, "'%s %s': cannot set variable '%s', only scope 'proc' is permitted here.", args[0], args[1], args[2]); |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1190 | goto fail; |
| 1191 | } |
| 1192 | |
| 1193 | err = NULL; |
| 1194 | nberr = smp_resolve_args(&px, &err); |
| 1195 | if (nberr) { |
| 1196 | release_sample_expr(rule.arg.vars.expr); |
| 1197 | indent_msg(&err, 2); |
| 1198 | goto fail; |
| 1199 | } |
| 1200 | |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1201 | if (use_fmt && !(sess = session_new(&px, NULL, &objt))) { |
| 1202 | release_sample_expr(rule.arg.vars.expr); |
| 1203 | memprintf(&err, "memory allocation error."); |
| 1204 | goto fail; |
| 1205 | } |
| 1206 | |
| 1207 | action_store(&rule, &px, sess, NULL, 0); |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1208 | release_sample_expr(rule.arg.vars.expr); |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1209 | if (sess) |
| 1210 | session_free(sess); |
| 1211 | |
Willy Tarreau | b8bd1ee | 2021-03-26 15:19:50 +0100 | [diff] [blame] | 1212 | appctx->st0 = CLI_ST_PROMPT; |
| 1213 | return 0; |
| 1214 | fail: |
| 1215 | return cli_dynerr(appctx, err); |
| 1216 | } |
| 1217 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1218 | static int vars_max_size(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1219 | const struct proxy *defpx, const char *file, int line, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1220 | char **err, unsigned int *limit) |
| 1221 | { |
| 1222 | char *error; |
| 1223 | |
| 1224 | *limit = strtol(args[1], &error, 10); |
| 1225 | if (*error != 0) { |
| 1226 | memprintf(err, "%s: '%s' is an invalid size", args[0], args[1]); |
| 1227 | return -1; |
| 1228 | } |
| 1229 | return 0; |
| 1230 | } |
| 1231 | |
| 1232 | static int vars_max_size_global(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1233 | const struct proxy *defpx, const char *file, int line, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1234 | char **err) |
| 1235 | { |
| 1236 | return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_global_limit); |
| 1237 | } |
| 1238 | |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 1239 | static int vars_max_size_proc(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1240 | const struct proxy *defpx, const char *file, int line, |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 1241 | char **err) |
| 1242 | { |
| 1243 | return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_proc_limit); |
| 1244 | } |
| 1245 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1246 | static int vars_max_size_sess(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1247 | const struct proxy *defpx, const char *file, int line, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1248 | char **err) |
| 1249 | { |
| 1250 | return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_sess_limit); |
| 1251 | } |
| 1252 | |
| 1253 | static int vars_max_size_txn(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1254 | const struct proxy *defpx, const char *file, int line, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1255 | char **err) |
| 1256 | { |
| 1257 | return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_txn_limit); |
| 1258 | } |
| 1259 | |
| 1260 | static int vars_max_size_reqres(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1261 | const struct proxy *defpx, const char *file, int line, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1262 | char **err) |
| 1263 | { |
| 1264 | return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_reqres_limit); |
| 1265 | } |
| 1266 | |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 1267 | static int vars_max_size_check(char **args, int section_type, struct proxy *curpx, |
Willy Tarreau | 0182516 | 2021-03-09 09:53:46 +0100 | [diff] [blame] | 1268 | const struct proxy *defpx, const char *file, int line, |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 1269 | char **err) |
| 1270 | { |
| 1271 | return vars_max_size(args, section_type, curpx, defpx, file, line, err, &var_check_limit); |
| 1272 | } |
| 1273 | |
Willy Tarreau | 2c897d9 | 2021-08-31 08:48:55 +0200 | [diff] [blame] | 1274 | /* early boot initialization */ |
| 1275 | static void vars_init() |
| 1276 | { |
| 1277 | var_name_hash_seed = ha_random64(); |
| 1278 | } |
| 1279 | |
| 1280 | INITCALL0(STG_PREPARE, vars_init); |
| 1281 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1282 | static struct sample_fetch_kw_list sample_fetch_keywords = {ILH, { |
| 1283 | |
Willy Tarreau | 44c5ff6 | 2021-11-02 17:08:15 +0100 | [diff] [blame] | 1284 | { "var", smp_fetch_var, ARG2(1,STR,STR), smp_check_var, SMP_T_ANY, SMP_USE_CONST }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1285 | { /* END */ }, |
| 1286 | }}; |
| 1287 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1288 | INITCALL1(STG_REGISTER, sample_register_fetches, &sample_fetch_keywords); |
| 1289 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1290 | static struct sample_conv_kw_list sample_conv_kws = {ILH, { |
Remi Tricot-Le Breton | 51899d2 | 2021-12-16 17:14:37 +0100 | [diff] [blame^] | 1291 | { "set-var", smp_conv_store, ARG5(1,STR,STR,STR,STR,STR), conv_check_var, SMP_T_ANY, SMP_T_ANY }, |
Christopher Faulet | 85d79c9 | 2016-11-09 16:54:56 +0100 | [diff] [blame] | 1292 | { "unset-var", smp_conv_clear, ARG1(1,STR), conv_check_var, SMP_T_ANY, SMP_T_ANY }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1293 | { /* END */ }, |
| 1294 | }}; |
| 1295 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1296 | INITCALL1(STG_REGISTER, sample_register_convs, &sample_conv_kws); |
| 1297 | |
Jaroslaw Rzeszótko | c863703 | 2021-11-02 16:56:05 +0100 | [diff] [blame] | 1298 | static struct action_kw_list tcp_req_conn_kws = { { }, { |
| 1299 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
| 1300 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1301 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
| 1302 | { /* END */ } |
| 1303 | }}; |
| 1304 | |
| 1305 | INITCALL1(STG_REGISTER, tcp_req_conn_keywords_register, &tcp_req_conn_kws); |
| 1306 | |
Willy Tarreau | 620408f | 2016-10-21 16:37:51 +0200 | [diff] [blame] | 1307 | static struct action_kw_list tcp_req_sess_kws = { { }, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1308 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1309 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1310 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Willy Tarreau | 620408f | 2016-10-21 16:37:51 +0200 | [diff] [blame] | 1311 | { /* END */ } |
| 1312 | }}; |
| 1313 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1314 | INITCALL1(STG_REGISTER, tcp_req_sess_keywords_register, &tcp_req_sess_kws); |
| 1315 | |
Willy Tarreau | 620408f | 2016-10-21 16:37:51 +0200 | [diff] [blame] | 1316 | static struct action_kw_list tcp_req_cont_kws = { { }, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1317 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1318 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1319 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1320 | { /* END */ } |
| 1321 | }}; |
| 1322 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1323 | INITCALL1(STG_REGISTER, tcp_req_cont_keywords_register, &tcp_req_cont_kws); |
| 1324 | |
Thierry FOURNIER | 36481b8 | 2015-08-19 09:01:53 +0200 | [diff] [blame] | 1325 | static struct action_kw_list tcp_res_kws = { { }, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1326 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1327 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1328 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1329 | { /* END */ } |
| 1330 | }}; |
| 1331 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1332 | INITCALL1(STG_REGISTER, tcp_res_cont_keywords_register, &tcp_res_kws); |
| 1333 | |
Gaetan Rivet | 707b52f | 2020-02-21 18:14:59 +0100 | [diff] [blame] | 1334 | static struct action_kw_list tcp_check_kws = {ILH, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1335 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1336 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1337 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Gaetan Rivet | 707b52f | 2020-02-21 18:14:59 +0100 | [diff] [blame] | 1338 | { /* END */ } |
| 1339 | }}; |
| 1340 | |
| 1341 | INITCALL1(STG_REGISTER, tcp_check_keywords_register, &tcp_check_kws); |
| 1342 | |
Thierry FOURNIER | 36481b8 | 2015-08-19 09:01:53 +0200 | [diff] [blame] | 1343 | static struct action_kw_list http_req_kws = { { }, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1344 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1345 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1346 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1347 | { /* END */ } |
| 1348 | }}; |
| 1349 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1350 | INITCALL1(STG_REGISTER, http_req_keywords_register, &http_req_kws); |
| 1351 | |
Thierry FOURNIER | 36481b8 | 2015-08-19 09:01:53 +0200 | [diff] [blame] | 1352 | static struct action_kw_list http_res_kws = { { }, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1353 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1354 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1355 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1356 | { /* END */ } |
| 1357 | }}; |
| 1358 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1359 | INITCALL1(STG_REGISTER, http_res_keywords_register, &http_res_kws); |
| 1360 | |
Christopher Faulet | 6d0c3df | 2020-01-22 09:26:35 +0100 | [diff] [blame] | 1361 | static struct action_kw_list http_after_res_kws = { { }, { |
Willy Tarreau | 9a621ae | 2021-09-02 21:00:38 +0200 | [diff] [blame] | 1362 | { "set-var-fmt", parse_store, KWF_MATCH_PREFIX }, |
Amaury Denoyelle | e4a617c | 2021-05-06 15:33:09 +0200 | [diff] [blame] | 1363 | { "set-var", parse_store, KWF_MATCH_PREFIX }, |
| 1364 | { "unset-var", parse_store, KWF_MATCH_PREFIX }, |
Christopher Faulet | 6d0c3df | 2020-01-22 09:26:35 +0100 | [diff] [blame] | 1365 | { /* END */ } |
| 1366 | }}; |
| 1367 | |
| 1368 | INITCALL1(STG_REGISTER, http_after_res_keywords_register, &http_after_res_kws); |
| 1369 | |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1370 | static struct cfg_kw_list cfg_kws = {{ },{ |
Willy Tarreau | 13d2ba2 | 2021-03-26 11:38:08 +0100 | [diff] [blame] | 1371 | { CFG_GLOBAL, "set-var", vars_parse_global_set_var }, |
Willy Tarreau | 753d4db | 2021-09-03 09:02:47 +0200 | [diff] [blame] | 1372 | { CFG_GLOBAL, "set-var-fmt", vars_parse_global_set_var }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1373 | { CFG_GLOBAL, "tune.vars.global-max-size", vars_max_size_global }, |
Christopher Faulet | ff2613e | 2016-11-09 11:36:17 +0100 | [diff] [blame] | 1374 | { CFG_GLOBAL, "tune.vars.proc-max-size", vars_max_size_proc }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1375 | { CFG_GLOBAL, "tune.vars.sess-max-size", vars_max_size_sess }, |
| 1376 | { CFG_GLOBAL, "tune.vars.txn-max-size", vars_max_size_txn }, |
| 1377 | { CFG_GLOBAL, "tune.vars.reqres-max-size", vars_max_size_reqres }, |
Gaetan Rivet | 13a5043 | 2020-02-21 18:13:44 +0100 | [diff] [blame] | 1378 | { CFG_GLOBAL, "tune.vars.check-max-size", vars_max_size_check }, |
Thierry FOURNIER | 4834bc7 | 2015-06-06 19:29:07 +0200 | [diff] [blame] | 1379 | { /* END */ } |
| 1380 | }}; |
| 1381 | |
Willy Tarreau | 0108d90 | 2018-11-25 19:14:37 +0100 | [diff] [blame] | 1382 | INITCALL1(STG_REGISTER, cfg_register_keywords, &cfg_kws); |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 1383 | |
| 1384 | |
| 1385 | /* register cli keywords */ |
| 1386 | static struct cli_kw_list cli_kws = {{ },{ |
Willy Tarreau | b205bfd | 2021-05-07 11:38:37 +0200 | [diff] [blame] | 1387 | { { "get", "var", NULL }, "get var <name> : retrieve contents of a process-wide variable", vars_parse_cli_get_var, NULL }, |
Willy Tarreau | e93bff4 | 2021-09-03 09:47:37 +0200 | [diff] [blame] | 1388 | { { "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 }, |
Willy Tarreau | c35eb38 | 2021-03-26 14:51:31 +0100 | [diff] [blame] | 1389 | { { NULL }, NULL, NULL, NULL } |
| 1390 | }}; |
| 1391 | INITCALL1(STG_REGISTER, cli_register_kw, &cli_kws); |