1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#ifndef _ULOGD_DB_H
#define _ULOGD_DB_H
#include <ulogd/ulogd.h>
#include <stdbool.h>
/**
* default buffer size used for batching. If this limit is reached
* a commit is done (or after timer elapsed).
*/
#define ULOGD_DB_BUFFER_DEFAULT 100
struct db_driver {
/* prepare SQL statement (optional) */
int (* prepare)(struct ulogd_pluginstance *);
int (* interp)(struct ulogd_pluginstance *);
int (* commit)(struct ulogd_pluginstance *, int);
int (* open_db)(struct ulogd_pluginstance *upi);
int (* close_db)(struct ulogd_pluginstance *upi);
/* set input keys depending on database shema (optional) */
int (* get_columns)(struct ulogd_pluginstance *upi);
int (*execute)(struct ulogd_pluginstance *upi,
const char *stmt, unsigned int len);
};
/**
* A generic database row.
*
* The number and contents of the values has to be in sync with
* %db_instace.cols.
*/
struct db_row {
struct llist_head link;
struct ulogd_value value[];
};
/*
* Map column to a particular ulogd key.
*
* This mapping is determined during each propagation because the mapping
* of ulogd keys to database columns is N:1.
*/
struct db_map {
struct ulogd_key *key;
};
#define DB_F_OPEN 0x0001
struct db_instance {
unsigned flags;
const struct db_driver *driver;
/*
* Map key to a database column
*/
struct db_map *map;
int num_cols;
/* batching */
struct llist_head rows;
struct llist_head rows_committed;
int num_rows;
int buffer_size;
int max_backlog;
unsigned overlimit_msg : 1;
struct ulogd_timer timer;
char *stmt; /* buffer for our insert statement */
char *stmt_val; /* pointer to the beginning of the "VALUES" part */
char *schema;
};
static inline bool
db_has_prepare(const struct db_instance *di)
{
return di->driver->prepare != NULL;
}
#define TIME_ERR ((time_t)-1) /* Be paranoid */
/*
* GENERIC DATABASE CONFIGURATION SWITCHES
*
* buffer=N Buffer size used for batching. If this limit is reached
* a commit is forced (or after timer elapsed).
*/
#define DB_CES \
CONFIG_KEY("table", STRING, CONFIG_OPT_MANDATORY), \
CONFIG_KEY("reconnect", INT, 0), \
CONFIG_KEY("ip_as_string", INT, 0), \
CONFIG_KEY("connect_timeout", INT, 0), \
CONFIG_KEY_INT("buffer", ULOGD_DB_BUFFER_DEFAULT), \
CONFIG_KEY_INT("disable", 0), \
CONFIG_KEY_INT("blackhole", 0), \
CONFIG_KEY_STR("keymap", NULL), \
CONFIG_KEY_STR("insert", NULL)
#define __DB_CE_BASE 0
#define DB_CE_NUM 9
#define table_ce(pi) ulogd_config_str(pi, __DB_CE_BASE)
#define reconnect_ce(pi) ulogd_config_int(pi, __DB_CE_BASE + 1)
#define asstring_ce(pi) ulogd_config_int(pi, __DB_CE_BASE + 2)
#define timeout_ce(pi) ulogd_config_int(pi, __DB_CE_BASE + 3)
#define db_buffer_ce(pi) ulogd_config_int(pi, __DB_CE_BASE + 4)
#define disable_ce(pi) ulogd_config_int(pi, __DB_CE_BASE + 5)
#define blackhole_ce(pi) ulogd_config_int(pi, __DB_CE_BASE + 6)
#define keymap_ce(pi) ulogd_config_str(pi, __DB_CE_BASE + 7)
#define insert_ce(pi) ulogd_config_str(pi, __DB_CE_BASE + 8)
int ulogd_db_start(struct ulogd_pluginstance *upi);
int ulogd_db_stop(struct ulogd_pluginstance *upi);
int ulogd_db_interp(struct ulogd_pluginstance *upi, unsigned *flags);
int ulogd_db_configure(struct ulogd_pluginstance *upi);
int ulogd_db_signal(struct ulogd_pluginstance *upi, int signal);
/* generic row handling */
struct db_row *db_row_new(struct ulogd_pluginstance *pi);
int db_row_add(struct ulogd_pluginstance *pi, struct db_row *);
void db_row_del(struct ulogd_pluginstance *pi, struct db_row *);
#endif
|
