Thierry FOURNIER | 4aec0a4 | 2018-02-23 11:42:57 +0100 | [diff] [blame] | 1 | /* Main SPOA server includes |
| 2 | * |
| 3 | * Copyright 2016 HAProxy Technologies, Christopher Faulet <cfaulet@haproxy.com> |
| 4 | * Copyright 2018 OZON / Thierry Fournier <thierry.fournier@ozon.io> |
| 5 | * |
| 6 | * This program is free software; you can redistribute it and/or |
| 7 | * modify it under the terms of the GNU General Public License |
| 8 | * as published by the Free Software Foundation; either version |
| 9 | * 2 of the License, or (at your option) any later version. |
| 10 | */ |
| 11 | #ifndef __SPOA_H__ |
| 12 | #define __SPOA_H__ |
| 13 | |
Thierry FOURNIER | 880d7e1 | 2018-02-25 10:54:56 +0100 | [diff] [blame] | 14 | #include <pthread.h> |
Thierry FOURNIER | 4aec0a4 | 2018-02-23 11:42:57 +0100 | [diff] [blame] | 15 | #include <stdbool.h> |
| 16 | #include <stdint.h> |
| 17 | #include <netinet/in.h> |
Thierry FOURNIER | 880d7e1 | 2018-02-25 10:54:56 +0100 | [diff] [blame] | 18 | #include <sys/time.h> |
Thierry FOURNIER | 4aec0a4 | 2018-02-23 11:42:57 +0100 | [diff] [blame] | 19 | |
| 20 | #define MAX_FRAME_SIZE 16384 |
| 21 | #define SPOP_VERSION "1.0" |
| 22 | #define SPOA_CAPABILITIES "" |
| 23 | |
| 24 | /* All supported data types */ |
| 25 | enum spoe_data_type { |
| 26 | SPOE_DATA_T_NULL = 0, |
| 27 | SPOE_DATA_T_BOOL, |
| 28 | SPOE_DATA_T_INT32, |
| 29 | SPOE_DATA_T_UINT32, |
| 30 | SPOE_DATA_T_INT64, |
| 31 | SPOE_DATA_T_UINT64, |
| 32 | SPOE_DATA_T_IPV4, |
| 33 | SPOE_DATA_T_IPV6, |
| 34 | SPOE_DATA_T_STR, |
| 35 | SPOE_DATA_T_BIN, |
| 36 | SPOE_DATA_TYPES |
| 37 | }; |
| 38 | |
| 39 | /* Scopes used for variables set by agents. It is a way to be agnotic to vars |
| 40 | * scope. */ |
| 41 | enum spoe_vars_scope { |
| 42 | SPOE_SCOPE_PROC = 0, /* <=> SCOPE_PROC */ |
| 43 | SPOE_SCOPE_SESS, /* <=> SCOPE_SESS */ |
| 44 | SPOE_SCOPE_TXN, /* <=> SCOPE_TXN */ |
| 45 | SPOE_SCOPE_REQ, /* <=> SCOPE_REQ */ |
| 46 | SPOE_SCOPE_RES, /* <=> SCOPE_RES */ |
| 47 | }; |
| 48 | |
| 49 | struct worker { |
| 50 | unsigned int id; |
| 51 | char buf[MAX_FRAME_SIZE]; |
| 52 | unsigned int len; |
| 53 | unsigned int size; |
| 54 | int status_code; |
| 55 | unsigned int stream_id; |
| 56 | unsigned int frame_id; |
| 57 | bool healthcheck; |
| 58 | int ip_score; /* -1 if unset, else between 0 and 100 */ |
| 59 | }; |
| 60 | |
| 61 | struct chunk { |
| 62 | char *str; /* beginning of the string itself. Might not be 0-terminated */ |
| 63 | int len; /* current size of the string from first to last char */ |
| 64 | }; |
| 65 | |
| 66 | union spoe_value { |
| 67 | bool boolean; /* use for boolean */ |
| 68 | int32_t sint32; /* used for signed 32bits integers */ |
| 69 | uint32_t uint32; /* used for signed 32bits integers */ |
| 70 | int32_t sint64; /* used for signed 64bits integers */ |
| 71 | uint32_t uint64; /* used for signed 64bits integers */ |
| 72 | struct in_addr ipv4; /* used for ipv4 addresses */ |
| 73 | struct in6_addr ipv6; /* used for ipv6 addresses */ |
| 74 | struct chunk buffer; /* used for char strings or buffers */ |
| 75 | }; |
| 76 | |
| 77 | /* Used to store sample constant */ |
| 78 | struct spoe_data { |
| 79 | enum spoe_data_type type; /* SPOE_DATA_T_* */ |
| 80 | union spoe_value u; /* spoe data value */ |
| 81 | }; |
| 82 | |
Thierry FOURNIER | 64eaa33 | 2018-02-23 14:58:40 +0100 | [diff] [blame] | 83 | struct spoe_kv { |
| 84 | struct chunk name; |
| 85 | struct spoe_data value; |
| 86 | }; |
| 87 | |
| 88 | struct ps { |
| 89 | struct ps *next; |
| 90 | char *ext; |
| 91 | int (*init_worker)(struct worker *w); |
Thierry FOURNIER | 892f664 | 2018-02-23 14:27:05 +0100 | [diff] [blame^] | 92 | int (*exec_message)(struct worker *w, void *ref, int nargs, struct spoe_kv *args); |
Thierry FOURNIER | 64eaa33 | 2018-02-23 14:58:40 +0100 | [diff] [blame] | 93 | }; |
| 94 | |
Thierry FOURNIER | 892f664 | 2018-02-23 14:27:05 +0100 | [diff] [blame^] | 95 | struct ps_message { |
| 96 | struct ps_message *next; |
| 97 | const char *name; |
| 98 | struct ps *ps; |
| 99 | void *ref; |
| 100 | }; |
| 101 | |
Thierry FOURNIER | 880d7e1 | 2018-02-25 10:54:56 +0100 | [diff] [blame] | 102 | extern bool debug; |
| 103 | extern pthread_key_t worker_id; |
| 104 | |
Thierry FOURNIER | 64eaa33 | 2018-02-23 14:58:40 +0100 | [diff] [blame] | 105 | void ps_register(struct ps *ps); |
Thierry FOURNIER | 892f664 | 2018-02-23 14:27:05 +0100 | [diff] [blame^] | 106 | void ps_register_message(struct ps *ps, const char *name, void *ref); |
Thierry FOURNIER | 64eaa33 | 2018-02-23 14:58:40 +0100 | [diff] [blame] | 107 | |
Thierry FOURNIER | 880d7e1 | 2018-02-25 10:54:56 +0100 | [diff] [blame] | 108 | #define LOG(fmt, args...) \ |
| 109 | do { \ |
| 110 | struct timeval now; \ |
| 111 | int wid = *((int*)pthread_getspecific(worker_id)); \ |
| 112 | \ |
| 113 | gettimeofday(&now, NULL); \ |
| 114 | fprintf(stderr, "%ld.%06ld [%02d] " fmt "\n", \ |
| 115 | now.tv_sec, now.tv_usec, wid, ##args); \ |
| 116 | } while (0) |
| 117 | |
| 118 | #define DEBUG(x...) \ |
| 119 | do { \ |
| 120 | if (debug) \ |
| 121 | LOG(x); \ |
| 122 | } while (0) |
| 123 | |
Thierry FOURNIER | 4aec0a4 | 2018-02-23 11:42:57 +0100 | [diff] [blame] | 124 | #endif /* __SPOA_H__ */ |