David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 1 | #include <stdio.h> |
| 2 | |
| 3 | #include <common/cfgparse.h> |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 4 | #include <common/errors.h> |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 5 | #include <proto/arg.h> |
Willy Tarreau | 79e5733 | 2018-10-02 16:01:16 +0200 | [diff] [blame^] | 6 | #include <proto/http_fetch.h> |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 7 | #include <proto/log.h> |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 8 | #include <proto/proto_http.h> |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 9 | #include <proto/sample.h> |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 10 | #include <dac.h> |
| 11 | |
| 12 | static struct { |
| 13 | void *atlasimgptr; |
| 14 | char *jsonpath; |
| 15 | char *cookiename; |
| 16 | size_t cookienamelen; |
| 17 | da_atlas_t atlas; |
| 18 | da_evidence_id_t useragentid; |
| 19 | da_severity_t loglevel; |
| 20 | char separator; |
| 21 | unsigned char daset:1; |
| 22 | } global_deviceatlas = { |
| 23 | .loglevel = 0, |
| 24 | .jsonpath = 0, |
| 25 | .cookiename = 0, |
| 26 | .cookienamelen = 0, |
| 27 | .useragentid = 0, |
| 28 | .daset = 0, |
| 29 | .separator = '|', |
| 30 | }; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 31 | |
| 32 | static int da_json_file(char **args, int section_type, struct proxy *curpx, |
| 33 | struct proxy *defpx, const char *file, int line, |
| 34 | char **err) |
| 35 | { |
| 36 | if (*(args[1]) == 0) { |
| 37 | memprintf(err, "deviceatlas json file : expects a json path.\n"); |
| 38 | return -1; |
| 39 | } |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 40 | global_deviceatlas.jsonpath = strdup(args[1]); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 41 | return 0; |
| 42 | } |
| 43 | |
| 44 | static int da_log_level(char **args, int section_type, struct proxy *curpx, |
| 45 | struct proxy *defpx, const char *file, int line, |
| 46 | char **err) |
| 47 | { |
| 48 | int loglevel; |
| 49 | if (*(args[1]) == 0) { |
| 50 | memprintf(err, "deviceatlas log level : expects an integer argument.\n"); |
| 51 | return -1; |
| 52 | } |
| 53 | |
| 54 | loglevel = atol(args[1]); |
| 55 | if (loglevel < 0 || loglevel > 3) { |
| 56 | memprintf(err, "deviceatlas log level : expects a log level between 0 and 3, %s given.\n", args[1]); |
| 57 | } else { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 58 | global_deviceatlas.loglevel = (da_severity_t)loglevel; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 59 | } |
| 60 | |
| 61 | return 0; |
| 62 | } |
| 63 | |
| 64 | static int da_property_separator(char **args, int section_type, struct proxy *curpx, |
| 65 | struct proxy *defpx, const char *file, int line, |
| 66 | char **err) |
| 67 | { |
| 68 | if (*(args[1]) == 0) { |
| 69 | memprintf(err, "deviceatlas property separator : expects a character argument.\n"); |
| 70 | return -1; |
| 71 | } |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 72 | global_deviceatlas.separator = *args[1]; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 73 | return 0; |
| 74 | } |
| 75 | |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 76 | static int da_properties_cookie(char **args, int section_type, struct proxy *curpx, |
| 77 | struct proxy *defpx, const char *file, int line, |
| 78 | char **err) |
| 79 | { |
| 80 | if (*(args[1]) == 0) { |
| 81 | memprintf(err, "deviceatlas cookie name : expects a string argument.\n"); |
| 82 | return -1; |
| 83 | } else { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 84 | global_deviceatlas.cookiename = strdup(args[1]); |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 85 | } |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 86 | global_deviceatlas.cookienamelen = strlen(global_deviceatlas.cookiename); |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 87 | return 0; |
| 88 | } |
| 89 | |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 90 | static size_t da_haproxy_read(void *ctx, size_t len, char *buf) |
| 91 | { |
| 92 | return fread(buf, 1, len, ctx); |
| 93 | } |
| 94 | |
| 95 | static da_status_t da_haproxy_seek(void *ctx, off_t off) |
| 96 | { |
| 97 | return fseek(ctx, off, SEEK_SET) != -1 ? DA_OK : DA_SYS; |
| 98 | } |
| 99 | |
| 100 | static void da_haproxy_log(da_severity_t severity, da_status_t status, |
| 101 | const char *fmt, va_list args) |
| 102 | { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 103 | if (global_deviceatlas.loglevel && severity <= global_deviceatlas.loglevel) { |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 104 | char logbuf[256]; |
| 105 | vsnprintf(logbuf, sizeof(logbuf), fmt, args); |
Christopher Faulet | 767a84b | 2017-11-24 16:50:31 +0100 | [diff] [blame] | 106 | ha_warning("deviceatlas : %s.\n", logbuf); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 107 | } |
| 108 | } |
| 109 | |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 110 | #define DA_COOKIENAME_DEFAULT "DAPROPS" |
| 111 | |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 112 | /* |
| 113 | * module init / deinit functions. Returns 0 if OK, or a combination of ERR_*. |
| 114 | */ |
| 115 | static int init_deviceatlas(void) |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 116 | { |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 117 | int err_code = 0; |
| 118 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 119 | if (global_deviceatlas.jsonpath != 0) { |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 120 | FILE *jsonp; |
| 121 | da_property_decl_t extraprops[] = {{0, 0}}; |
| 122 | size_t atlasimglen; |
| 123 | da_status_t status; |
| 124 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 125 | jsonp = fopen(global_deviceatlas.jsonpath, "r"); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 126 | if (jsonp == 0) { |
Christopher Faulet | 767a84b | 2017-11-24 16:50:31 +0100 | [diff] [blame] | 127 | ha_alert("deviceatlas : '%s' json file has invalid path or is not readable.\n", |
| 128 | global_deviceatlas.jsonpath); |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 129 | err_code |= ERR_ALERT | ERR_FATAL; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 130 | goto out; |
| 131 | } |
| 132 | |
| 133 | da_init(); |
| 134 | da_seterrorfunc(da_haproxy_log); |
| 135 | status = da_atlas_compile(jsonp, da_haproxy_read, da_haproxy_seek, |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 136 | &global_deviceatlas.atlasimgptr, &atlasimglen); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 137 | fclose(jsonp); |
| 138 | if (status != DA_OK) { |
Christopher Faulet | 767a84b | 2017-11-24 16:50:31 +0100 | [diff] [blame] | 139 | ha_alert("deviceatlas : '%s' json file is invalid.\n", |
| 140 | global_deviceatlas.jsonpath); |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 141 | err_code |= ERR_ALERT | ERR_FATAL; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 142 | goto out; |
| 143 | } |
| 144 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 145 | status = da_atlas_open(&global_deviceatlas.atlas, extraprops, |
| 146 | global_deviceatlas.atlasimgptr, atlasimglen); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 147 | |
| 148 | if (status != DA_OK) { |
Christopher Faulet | 767a84b | 2017-11-24 16:50:31 +0100 | [diff] [blame] | 149 | ha_alert("deviceatlas : data could not be compiled.\n"); |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 150 | err_code |= ERR_ALERT | ERR_FATAL; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 151 | goto out; |
| 152 | } |
| 153 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 154 | if (global_deviceatlas.cookiename == 0) { |
| 155 | global_deviceatlas.cookiename = strdup(DA_COOKIENAME_DEFAULT); |
| 156 | global_deviceatlas.cookienamelen = strlen(global_deviceatlas.cookiename); |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 157 | } |
| 158 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 159 | global_deviceatlas.useragentid = da_atlas_header_evidence_id(&global_deviceatlas.atlas, |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 160 | "user-agent"); |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 161 | global_deviceatlas.daset = 1; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 162 | |
| 163 | fprintf(stdout, "Deviceatlas module loaded.\n"); |
| 164 | } |
| 165 | |
| 166 | out: |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 167 | return err_code; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 168 | } |
| 169 | |
Willy Tarreau | b149eed | 2016-12-21 21:03:49 +0100 | [diff] [blame] | 170 | static void deinit_deviceatlas(void) |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 171 | { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 172 | if (global_deviceatlas.jsonpath != 0) { |
| 173 | free(global_deviceatlas.jsonpath); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 174 | } |
| 175 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 176 | if (global_deviceatlas.daset == 1) { |
| 177 | free(global_deviceatlas.cookiename); |
| 178 | da_atlas_close(&global_deviceatlas.atlas); |
| 179 | free(global_deviceatlas.atlasimgptr); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 180 | } |
| 181 | |
| 182 | da_fini(); |
| 183 | } |
| 184 | |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 185 | static int da_haproxy(const struct arg *args, struct sample *smp, da_deviceinfo_t *devinfo) |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 186 | { |
Willy Tarreau | 83061a8 | 2018-07-13 11:56:34 +0200 | [diff] [blame] | 187 | struct buffer *tmp; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 188 | da_propid_t prop, *pprop; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 189 | da_status_t status; |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 190 | da_type_t proptype; |
| 191 | const char *propname; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 192 | int i; |
| 193 | |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 194 | tmp = get_trash_chunk(); |
| 195 | chunk_reset(tmp); |
| 196 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 197 | propname = (const char *) args[0].data.str.area; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 198 | i = 0; |
| 199 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 200 | for (; propname != 0; i ++, |
| 201 | propname = (const char *) args[i].data.str.area) { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 202 | status = da_atlas_getpropid(&global_deviceatlas.atlas, |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 203 | propname, &prop); |
| 204 | if (status != DA_OK) { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 205 | chunk_appendf(tmp, "%c", global_deviceatlas.separator); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 206 | continue; |
| 207 | } |
| 208 | pprop = ∝ |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 209 | da_atlas_getproptype(&global_deviceatlas.atlas, *pprop, &proptype); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 210 | |
| 211 | switch (proptype) { |
| 212 | case DA_TYPE_BOOLEAN: { |
| 213 | bool val; |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 214 | status = da_getpropboolean(devinfo, *pprop, &val); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 215 | if (status == DA_OK) { |
| 216 | chunk_appendf(tmp, "%d", val); |
| 217 | } |
| 218 | break; |
| 219 | } |
| 220 | case DA_TYPE_INTEGER: |
| 221 | case DA_TYPE_NUMBER: { |
| 222 | long val; |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 223 | status = da_getpropinteger(devinfo, *pprop, &val); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 224 | if (status == DA_OK) { |
| 225 | chunk_appendf(tmp, "%ld", val); |
| 226 | } |
| 227 | break; |
| 228 | } |
| 229 | case DA_TYPE_STRING: { |
| 230 | const char *val; |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 231 | status = da_getpropstring(devinfo, *pprop, &val); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 232 | if (status == DA_OK) { |
| 233 | chunk_appendf(tmp, "%s", val); |
| 234 | } |
| 235 | break; |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 236 | } |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 237 | default: |
| 238 | break; |
| 239 | } |
| 240 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 241 | chunk_appendf(tmp, "%c", global_deviceatlas.separator); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 242 | } |
| 243 | |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 244 | da_close(devinfo); |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 245 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 246 | if (tmp->data) { |
| 247 | --tmp->data; |
| 248 | tmp->area[tmp->data] = 0; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 249 | } |
| 250 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 251 | smp->data.u.str.area = tmp->area; |
| 252 | smp->data.u.str.data = tmp->data; |
David Carlier | 8167f30 | 2015-06-01 13:50:06 +0200 | [diff] [blame] | 253 | |
| 254 | return 1; |
| 255 | } |
| 256 | |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 257 | static int da_haproxy_conv(const struct arg *args, struct sample *smp, void *private) |
| 258 | { |
| 259 | da_deviceinfo_t devinfo; |
| 260 | da_status_t status; |
| 261 | const char *useragent; |
| 262 | char useragentbuf[1024] = { 0 }; |
| 263 | int i; |
| 264 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 265 | if (global_deviceatlas.daset == 0 || smp->data.u.str.data == 0) { |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 266 | return 1; |
| 267 | } |
| 268 | |
Willy Tarreau | 843b7cb | 2018-07-13 10:54:26 +0200 | [diff] [blame] | 269 | i = smp->data.u.str.data > sizeof(useragentbuf) ? sizeof(useragentbuf) : smp->data.u.str.data; |
| 270 | memcpy(useragentbuf, smp->data.u.str.area, i - 1); |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 271 | useragentbuf[i - 1] = 0; |
| 272 | |
| 273 | useragent = (const char *)useragentbuf; |
| 274 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 275 | status = da_search(&global_deviceatlas.atlas, &devinfo, |
| 276 | global_deviceatlas.useragentid, useragent, 0); |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 277 | |
| 278 | return status != DA_OK ? 0 : da_haproxy(args, smp, &devinfo); |
| 279 | } |
| 280 | |
| 281 | #define DA_MAX_HEADERS 24 |
| 282 | |
| 283 | static int da_haproxy_fetch(const struct arg *args, struct sample *smp, const char *kw, void *private) |
| 284 | { |
| 285 | struct hdr_idx *hidx; |
| 286 | struct hdr_ctx hctx; |
| 287 | const struct http_msg *hmsg; |
| 288 | da_evidence_t ev[DA_MAX_HEADERS]; |
| 289 | da_deviceinfo_t devinfo; |
| 290 | da_status_t status; |
| 291 | char vbuf[DA_MAX_HEADERS][1024] = {{ 0 }}; |
| 292 | int i, nbh = 0; |
| 293 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 294 | if (global_deviceatlas.daset == 0) { |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 295 | return 1; |
| 296 | } |
| 297 | |
| 298 | CHECK_HTTP_MESSAGE_FIRST(); |
| 299 | smp->data.type = SMP_T_STR; |
| 300 | |
| 301 | /** |
| 302 | * Here we go through the whole list of headers from start |
| 303 | * they will be filtered via the DeviceAtlas API itself |
| 304 | */ |
| 305 | hctx.idx = 0; |
| 306 | hidx = &smp->strm->txn->hdr_idx; |
| 307 | hmsg = &smp->strm->txn->req; |
| 308 | |
| 309 | while (http_find_next_header(hmsg->chn->buf->p, hidx, &hctx) == 1 && |
| 310 | nbh < DA_MAX_HEADERS) { |
| 311 | char *pval; |
| 312 | size_t vlen; |
| 313 | da_evidence_id_t evid = -1; |
| 314 | char hbuf[24] = { 0 }; |
| 315 | |
| 316 | /* The HTTP headers used by the DeviceAtlas API are not longer */ |
David Carlier | 91a88b0 | 2017-11-17 08:47:25 +0000 | [diff] [blame] | 317 | if (hctx.del >= sizeof(hbuf) || hctx.del <= 0 || hctx.vlen <= 0) { |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 318 | continue; |
| 319 | } |
| 320 | |
| 321 | vlen = hctx.vlen; |
| 322 | memcpy(hbuf, hctx.line, hctx.del); |
| 323 | hbuf[hctx.del] = 0; |
| 324 | pval = (hctx.line + hctx.val); |
| 325 | |
| 326 | if (strcmp(hbuf, "Accept-Language") == 0) { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 327 | evid = da_atlas_accept_language_evidence_id(&global_deviceatlas. |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 328 | atlas); |
| 329 | } else if (strcmp(hbuf, "Cookie") == 0) { |
| 330 | char *p, *eval; |
| 331 | int pl; |
| 332 | |
| 333 | eval = pval + hctx.vlen; |
| 334 | /** |
| 335 | * The cookie value, if it exists, is located between the current header's |
| 336 | * value position and the next one |
| 337 | */ |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 338 | if (extract_cookie_value(pval, eval, global_deviceatlas.cookiename, |
| 339 | global_deviceatlas.cookienamelen, 1, &p, &pl) == NULL) { |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 340 | continue; |
| 341 | } |
| 342 | |
| 343 | vlen = (size_t)pl; |
| 344 | pval = p; |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 345 | evid = da_atlas_clientprop_evidence_id(&global_deviceatlas.atlas); |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 346 | } else { |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 347 | evid = da_atlas_header_evidence_id(&global_deviceatlas.atlas, |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 348 | hbuf); |
| 349 | } |
| 350 | |
| 351 | if (evid == -1) { |
| 352 | continue; |
| 353 | } |
| 354 | |
| 355 | i = vlen > sizeof(vbuf[nbh]) ? sizeof(vbuf[nbh]) : vlen; |
| 356 | memcpy(vbuf[nbh], pval, i - 1); |
| 357 | vbuf[nbh][i - 1] = 0; |
| 358 | ev[nbh].key = evid; |
| 359 | ev[nbh].value = vbuf[nbh]; |
| 360 | ev[nbh].value[vlen] = 0; |
| 361 | ++ nbh; |
| 362 | } |
| 363 | |
Willy Tarreau | bee9dde | 2016-12-21 21:25:06 +0100 | [diff] [blame] | 364 | status = da_searchv(&global_deviceatlas.atlas, &devinfo, |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 365 | ev, nbh); |
| 366 | |
| 367 | return status != DA_OK ? 0 : da_haproxy(args, smp, &devinfo); |
| 368 | } |
| 369 | |
Willy Tarreau | 0d74f77 | 2015-06-01 15:42:29 +0200 | [diff] [blame] | 370 | static struct cfg_kw_list dacfg_kws = {{ }, { |
| 371 | { CFG_GLOBAL, "deviceatlas-json-file", da_json_file }, |
| 372 | { CFG_GLOBAL, "deviceatlas-log-level", da_log_level }, |
| 373 | { CFG_GLOBAL, "deviceatlas-property-separator", da_property_separator }, |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 374 | { CFG_GLOBAL, "deviceatlas-properties-cookie", da_properties_cookie }, |
Willy Tarreau | 0d74f77 | 2015-06-01 15:42:29 +0200 | [diff] [blame] | 375 | { 0, NULL, NULL }, |
| 376 | }}; |
| 377 | |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 378 | /* Note: must not be declared <const> as its list will be overwritten */ |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 379 | static struct sample_fetch_kw_list fetch_kws = {ILH, { |
David Carlier | 840b024 | 2016-03-16 10:09:55 +0000 | [diff] [blame] | 380 | { "da-csv-fetch", da_haproxy_fetch, ARG12(1,STR,STR,STR,STR,STR,STR,STR,STR,STR,STR,STR,STR), NULL, SMP_T_STR, SMP_USE_HRQHV }, |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 381 | { NULL, NULL, 0, 0, 0 }, |
| 382 | }}; |
| 383 | |
| 384 | /* Note: must not be declared <const> as its list will be overwritten */ |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 385 | static struct sample_conv_kw_list conv_kws = {ILH, { |
David Carlier | 840b024 | 2016-03-16 10:09:55 +0000 | [diff] [blame] | 386 | { "da-csv-conv", da_haproxy_conv, ARG12(1,STR,STR,STR,STR,STR,STR,STR,STR,STR,STR,STR,STR), NULL, SMP_T_STR, SMP_T_STR }, |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 387 | { NULL, NULL, 0, 0, 0 }, |
| 388 | }}; |
| 389 | |
| 390 | __attribute__((constructor)) |
| 391 | static void __da_init(void) |
| 392 | { |
| 393 | /* register sample fetch and format conversion keywords */ |
David Carlier | 608c65a | 2015-09-25 14:16:30 +0100 | [diff] [blame] | 394 | sample_register_fetches(&fetch_kws); |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 395 | sample_register_convs(&conv_kws); |
Willy Tarreau | 0d74f77 | 2015-06-01 15:42:29 +0200 | [diff] [blame] | 396 | cfg_register_keywords(&dacfg_kws); |
Willy Tarreau | 3dd483e | 2016-12-21 18:50:22 +0100 | [diff] [blame] | 397 | hap_register_build_opts("Built with DeviceAtlas support.", 0); |
Willy Tarreau | 876054d | 2016-12-21 20:39:16 +0100 | [diff] [blame] | 398 | hap_register_post_check(init_deviceatlas); |
Willy Tarreau | b149eed | 2016-12-21 21:03:49 +0100 | [diff] [blame] | 399 | hap_register_post_deinit(deinit_deviceatlas); |
Willy Tarreau | f63386a | 2015-06-01 15:39:50 +0200 | [diff] [blame] | 400 | } |