Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 1 | /* |
| 2 | * include/types/stick_table.h |
| 3 | * Macros, variables and structures for stick tables management. |
| 4 | * |
| 5 | * Copyright (C) 2009-2010 EXCELIANCE, Emeric Brun <ebrun@exceliance.fr> |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 6 | * Copyright (C) 2010 Willy Tarreau <w@1wt.eu> |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 7 | * |
| 8 | * This library is free software; you can redistribute it and/or |
| 9 | * modify it under the terms of the GNU Lesser General Public |
| 10 | * License as published by the Free Software Foundation, version 2.1 |
| 11 | * exclusively. |
| 12 | * |
| 13 | * This library is distributed in the hope that it will be useful, |
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 16 | * Lesser General Public License for more details. |
| 17 | * |
| 18 | * You should have received a copy of the GNU Lesser General Public |
| 19 | * License along with this library; if not, write to the Free Software |
| 20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 21 | */ |
| 22 | |
| 23 | #ifndef _TYPES_STICK_TABLE_H |
| 24 | #define _TYPES_STICK_TABLE_H |
| 25 | |
| 26 | #include <sys/socket.h> |
| 27 | #include <netinet/in.h> |
| 28 | |
| 29 | #include <ebtree.h> |
| 30 | #include <ebmbtree.h> |
| 31 | #include <eb32tree.h> |
| 32 | #include <common/memory.h> |
Willy Tarreau | 91c43d7 | 2010-06-20 11:19:22 +0200 | [diff] [blame] | 33 | #include <types/freq_ctr.h> |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 34 | |
| 35 | /* stick table key types */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 36 | enum { |
| 37 | STKTABLE_TYPE_IP = 0, /* table key is ipv4 */ |
David du Colombier | 4f92d32 | 2011-03-24 11:09:31 +0100 | [diff] [blame] | 38 | STKTABLE_TYPE_IPV6, /* table key is ipv6 */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 39 | STKTABLE_TYPE_INTEGER, /* table key is unsigned 32bit integer */ |
| 40 | STKTABLE_TYPE_STRING, /* table key is a null terminated string */ |
Emeric Brun | 485479d | 2010-09-23 18:02:19 +0200 | [diff] [blame] | 41 | STKTABLE_TYPE_BINARY, /* table key is a buffer of data */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 42 | STKTABLE_TYPES /* Number of types, must always be last */ |
| 43 | }; |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 44 | |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 45 | /* The types of extra data we can store in a stick table */ |
| 46 | enum { |
Willy Tarreau | 13c29de | 2010-06-06 16:40:39 +0200 | [diff] [blame] | 47 | STKTABLE_DT_SERVER_ID, /* the server ID to use with this session if > 0 */ |
Willy Tarreau | c3bd972 | 2010-06-20 12:47:25 +0200 | [diff] [blame] | 48 | STKTABLE_DT_GPC0, /* General Purpose Counter 0 (unsigned 32-bit integer) */ |
Willy Tarreau | 8fb12c4 | 2010-06-18 20:16:39 +0200 | [diff] [blame] | 49 | STKTABLE_DT_CONN_CNT, /* cumulated number of connections */ |
Willy Tarreau | 91c43d7 | 2010-06-20 11:19:22 +0200 | [diff] [blame] | 50 | STKTABLE_DT_CONN_RATE, /* incoming connection rate */ |
Willy Tarreau | 38285c1 | 2010-06-18 16:35:43 +0200 | [diff] [blame] | 51 | STKTABLE_DT_CONN_CUR, /* concurrent number of connections */ |
Willy Tarreau | f4d17d9 | 2010-06-18 22:10:12 +0200 | [diff] [blame] | 52 | STKTABLE_DT_SESS_CNT, /* cumulated number of sessions (accepted connections) */ |
Willy Tarreau | 91c43d7 | 2010-06-20 11:19:22 +0200 | [diff] [blame] | 53 | STKTABLE_DT_SESS_RATE, /* accepted sessions rate */ |
Willy Tarreau | da7ff64 | 2010-06-23 11:44:09 +0200 | [diff] [blame] | 54 | STKTABLE_DT_HTTP_REQ_CNT, /* cumulated number of incoming HTTP requests */ |
| 55 | STKTABLE_DT_HTTP_REQ_RATE,/* incoming HTTP request rate */ |
| 56 | STKTABLE_DT_HTTP_ERR_CNT, /* cumulated number of HTTP requests errors (4xx) */ |
| 57 | STKTABLE_DT_HTTP_ERR_RATE,/* HTTP request error rate */ |
Willy Tarreau | 855e4bb | 2010-06-18 18:33:32 +0200 | [diff] [blame] | 58 | STKTABLE_DT_BYTES_IN_CNT, /* cumulated bytes count from client to servers */ |
Willy Tarreau | 6c59e0a | 2010-06-20 11:56:30 +0200 | [diff] [blame] | 59 | STKTABLE_DT_BYTES_IN_RATE,/* bytes rate from client to servers */ |
Willy Tarreau | 855e4bb | 2010-06-18 18:33:32 +0200 | [diff] [blame] | 60 | STKTABLE_DT_BYTES_OUT_CNT,/* cumulated bytes count from servers to client */ |
Willy Tarreau | 6c59e0a | 2010-06-20 11:56:30 +0200 | [diff] [blame] | 61 | STKTABLE_DT_BYTES_OUT_RATE,/* bytes rate from servers to client */ |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 62 | STKTABLE_DATA_TYPES /* Number of data types, must always be last */ |
| 63 | }; |
| 64 | |
Willy Tarreau | 3b9c6e0 | 2010-07-18 08:04:30 +0200 | [diff] [blame] | 65 | /* The equivalent standard types of the stored data */ |
| 66 | enum { |
| 67 | STD_T_SINT = 0, /* data is of type signed int */ |
| 68 | STD_T_UINT, /* data is of type unsigned int */ |
| 69 | STD_T_ULL, /* data is of type unsigned long long */ |
| 70 | STD_T_FRQP, /* data is of type freq_ctr_period */ |
| 71 | }; |
| 72 | |
Willy Tarreau | 888617d | 2010-06-20 09:11:39 +0200 | [diff] [blame] | 73 | /* The types of optional arguments to stored data */ |
| 74 | enum { |
| 75 | ARG_T_NONE = 0, /* data type takes no argument (default) */ |
| 76 | ARG_T_INT, /* signed integer */ |
| 77 | ARG_T_DELAY, /* a delay which supports time units */ |
| 78 | }; |
| 79 | |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 80 | /* stick_table extra data. This is mainly used for casting or size computation */ |
| 81 | union stktable_data { |
Willy Tarreau | 3b9c6e0 | 2010-07-18 08:04:30 +0200 | [diff] [blame] | 82 | /* standard types for easy casting */ |
| 83 | int std_t_sint; |
| 84 | unsigned int std_t_uint; |
| 85 | unsigned long long std_t_ull; |
| 86 | struct freq_ctr_period std_t_frqp; |
| 87 | |
| 88 | /* types of each storable data */ |
Willy Tarreau | 13c29de | 2010-06-06 16:40:39 +0200 | [diff] [blame] | 89 | int server_id; |
Willy Tarreau | c3bd972 | 2010-06-20 12:47:25 +0200 | [diff] [blame] | 90 | unsigned int gpc0; |
Willy Tarreau | 38285c1 | 2010-06-18 16:35:43 +0200 | [diff] [blame] | 91 | unsigned int conn_cnt; |
Willy Tarreau | 91c43d7 | 2010-06-20 11:19:22 +0200 | [diff] [blame] | 92 | struct freq_ctr_period conn_rate; |
Willy Tarreau | 38285c1 | 2010-06-18 16:35:43 +0200 | [diff] [blame] | 93 | unsigned int conn_cur; |
Willy Tarreau | f4d17d9 | 2010-06-18 22:10:12 +0200 | [diff] [blame] | 94 | unsigned int sess_cnt; |
Willy Tarreau | 91c43d7 | 2010-06-20 11:19:22 +0200 | [diff] [blame] | 95 | struct freq_ctr_period sess_rate; |
Willy Tarreau | da7ff64 | 2010-06-23 11:44:09 +0200 | [diff] [blame] | 96 | unsigned int http_req_cnt; |
| 97 | struct freq_ctr_period http_req_rate; |
| 98 | unsigned int http_err_cnt; |
| 99 | struct freq_ctr_period http_err_rate; |
Willy Tarreau | 855e4bb | 2010-06-18 18:33:32 +0200 | [diff] [blame] | 100 | unsigned long long bytes_in_cnt; |
Willy Tarreau | 6c59e0a | 2010-06-20 11:56:30 +0200 | [diff] [blame] | 101 | struct freq_ctr_period bytes_in_rate; |
Willy Tarreau | 855e4bb | 2010-06-18 18:33:32 +0200 | [diff] [blame] | 102 | unsigned long long bytes_out_cnt; |
Willy Tarreau | 6c59e0a | 2010-06-20 11:56:30 +0200 | [diff] [blame] | 103 | struct freq_ctr_period bytes_out_rate; |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 104 | }; |
| 105 | |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 106 | /* known data types */ |
| 107 | struct stktable_data_type { |
| 108 | const char *name; /* name of the data type */ |
Willy Tarreau | 3b9c6e0 | 2010-07-18 08:04:30 +0200 | [diff] [blame] | 109 | int std_type; /* standard type we can use for this data, STD_T_* */ |
Willy Tarreau | 888617d | 2010-06-20 09:11:39 +0200 | [diff] [blame] | 110 | int arg_type; /* type of optional argument, ARG_T_* */ |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 111 | }; |
| 112 | |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 113 | /* stick table key type flags */ |
| 114 | #define STK_F_CUSTOM_KEYSIZE 0x00000001 /* this table's key size is configurable */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 115 | |
| 116 | /* stick table keyword type */ |
| 117 | struct stktable_type { |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 118 | const char *kw; /* keyword string */ |
| 119 | int flags; /* type flags */ |
| 120 | size_t default_size; /* default key size */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 121 | }; |
| 122 | |
Emeric Brun | 485479d | 2010-09-23 18:02:19 +0200 | [diff] [blame] | 123 | extern struct stktable_type stktable_types[]; |
| 124 | |
Willy Tarreau | 393379c | 2010-06-06 12:11:37 +0200 | [diff] [blame] | 125 | /* Sticky session. |
| 126 | * Any additional data related to the stuck session is installed *before* |
| 127 | * stksess (with negative offsets). This allows us to run variable-sized |
| 128 | * keys and variable-sized data without making use of intermediate pointers. |
| 129 | */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 130 | struct stksess { |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 131 | unsigned int expire; /* session expiration date */ |
Willy Tarreau | e7f3d7a | 2010-06-14 14:53:07 +0200 | [diff] [blame] | 132 | unsigned int ref_cnt; /* reference count, can only purge when zero */ |
Willy Tarreau | 86257dc | 2010-06-06 12:57:10 +0200 | [diff] [blame] | 133 | struct eb32_node exp; /* ebtree node used to hold the session in expiration tree */ |
Emeric Brun | 85e77c7 | 2010-09-23 18:16:52 +0200 | [diff] [blame] | 134 | struct eb32_node upd; /* ebtree node used to hold the update sequence tree */ |
Willy Tarreau | 86257dc | 2010-06-06 12:57:10 +0200 | [diff] [blame] | 135 | struct ebmb_node key; /* ebtree node used to hold the session in table */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 136 | /* WARNING! do not put anything after <keys>, it's used by the key */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 137 | }; |
| 138 | |
Emeric Brun | 85e77c7 | 2010-09-23 18:16:52 +0200 | [diff] [blame] | 139 | |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 140 | /* stick table */ |
| 141 | struct stktable { |
Emeric Brun | 85e77c7 | 2010-09-23 18:16:52 +0200 | [diff] [blame] | 142 | char *id; /* table id name */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 143 | struct eb_root keys; /* head of sticky session tree */ |
| 144 | struct eb_root exps; /* head of sticky session expiration tree */ |
Emeric Brun | 85e77c7 | 2010-09-23 18:16:52 +0200 | [diff] [blame] | 145 | struct eb_root updates; /* head of sticky updates sequence tree */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 146 | struct pool_head *pool; /* pool used to allocate sticky sessions */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 147 | struct task *exp_task; /* expiration task */ |
Emeric Brun | 85e77c7 | 2010-09-23 18:16:52 +0200 | [diff] [blame] | 148 | struct task *sync_task; /* sync task */ |
| 149 | unsigned int update; |
| 150 | unsigned int localupdate; |
Emeric Brun | 32da3c4 | 2010-09-23 18:39:19 +0200 | [diff] [blame] | 151 | union { |
| 152 | struct peers *p; /* sync peers */ |
| 153 | char *name; |
| 154 | } peers; |
| 155 | |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 156 | unsigned long type; /* type of table (determines key format) */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 157 | size_t key_size; /* size of a key, maximum size in case of string */ |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 158 | unsigned int size; /* maximum number of sticky sessions in table */ |
| 159 | unsigned int current; /* number of sticky sessions currently in table */ |
| 160 | int nopurge; /* if non-zero, don't purge sticky sessions when full */ |
| 161 | int exp_next; /* next expiration date (ticks) */ |
| 162 | int expire; /* time to live for sticky sessions (milliseconds) */ |
Willy Tarreau | 393379c | 2010-06-06 12:11:37 +0200 | [diff] [blame] | 163 | int data_size; /* the size of the data that is prepended *before* stksess */ |
Willy Tarreau | 08d5f98 | 2010-06-06 13:34:54 +0200 | [diff] [blame] | 164 | int data_ofs[STKTABLE_DATA_TYPES]; /* negative offsets of present data types, or 0 if absent */ |
Willy Tarreau | 888617d | 2010-06-20 09:11:39 +0200 | [diff] [blame] | 165 | union { |
| 166 | int i; |
| 167 | unsigned int u; |
| 168 | void *p; |
| 169 | } data_arg[STKTABLE_DATA_TYPES]; /* optional argument of each data type */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 170 | }; |
| 171 | |
Willy Tarreau | 795e4a9 | 2010-07-18 07:38:17 +0200 | [diff] [blame] | 172 | extern struct stktable_data_type stktable_data_types[STKTABLE_DATA_TYPES]; |
Willy Tarreau | aea940e | 2010-06-06 11:56:36 +0200 | [diff] [blame] | 173 | |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 174 | /* stick table key data */ |
| 175 | union stktable_key_data { |
David du Colombier | 4f92d32 | 2011-03-24 11:09:31 +0100 | [diff] [blame] | 176 | struct in_addr ip; /* used to store an ipv4 key */ |
| 177 | struct in6_addr ipv6; /* used to store an ipv6 key */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 178 | uint32_t integer; /* used to store an integer key */ |
Emeric Brun | 485479d | 2010-09-23 18:02:19 +0200 | [diff] [blame] | 179 | char buf[BUFSIZE]; /* used to store a null terminated string key or a buffer of data */ |
Emeric Brun | 3bd697e | 2010-01-04 15:23:48 +0100 | [diff] [blame] | 180 | }; |
| 181 | |
| 182 | /* stick table key */ |
| 183 | struct stktable_key { |
| 184 | void *key; /* pointer on key buffer */ |
| 185 | size_t key_len; /* data len to read in buff in case of null terminated string */ |
| 186 | union stktable_key_data data; /* data */ |
| 187 | }; |
| 188 | |
| 189 | #endif /* _TYPES_STICK_TABLE_H */ |
| 190 | |