1 | /* SAMHAIN file system integrity testing */
|
---|
2 | /* Copyright (C) 2001 Rainer Wichmann */
|
---|
3 | /* */
|
---|
4 | /* This program is free software; you can redistribute it */
|
---|
5 | /* and/or modify */
|
---|
6 | /* it under the terms of the GNU General Public License as */
|
---|
7 | /* published by */
|
---|
8 | /* the Free Software Foundation; either version 2 of the License, or */
|
---|
9 | /* (at your option) any later version. */
|
---|
10 | /* */
|
---|
11 | /* This program is distributed in the hope that it will be useful, */
|
---|
12 | /* but WITHOUT ANY WARRANTY; without even the implied warranty of */
|
---|
13 | /* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the */
|
---|
14 | /* GNU General Public License for more details. */
|
---|
15 | /* */
|
---|
16 | /* You should have received a copy of the GNU General Public License */
|
---|
17 | /* along with this program; if not, write to the Free Software */
|
---|
18 | /* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */
|
---|
19 |
|
---|
20 | #include "config_xor.h"
|
---|
21 |
|
---|
22 | #include <stdio.h>
|
---|
23 | #include <stdlib.h>
|
---|
24 | #include <stdarg.h>
|
---|
25 | #include <stddef.h>
|
---|
26 | #include <string.h>
|
---|
27 | #include <ctype.h>
|
---|
28 | #include <sys/types.h>
|
---|
29 |
|
---|
30 | #ifdef WITH_DATABASE
|
---|
31 |
|
---|
32 | /* define this if you want to debug the Oracle database support */
|
---|
33 | /* #define DB_DEBUG */
|
---|
34 |
|
---|
35 | #define SH_REAL_SET
|
---|
36 |
|
---|
37 | #include "samhain.h"
|
---|
38 |
|
---|
39 | #include "sh_cat.h"
|
---|
40 | #include "sh_error.h"
|
---|
41 | #include "sh_utils.h"
|
---|
42 |
|
---|
43 | #undef FIL__
|
---|
44 | #define FIL__ _("sh_database.c")
|
---|
45 |
|
---|
46 | typedef struct my_attr_
|
---|
47 | {
|
---|
48 | char * attr;
|
---|
49 | char * attr_o;
|
---|
50 | int inHash;
|
---|
51 | int val;
|
---|
52 | int size;
|
---|
53 | int alen;
|
---|
54 | size_t off;
|
---|
55 | } my_attr;
|
---|
56 |
|
---|
57 | typedef struct dbins_ {
|
---|
58 | struct dbins_ * next;
|
---|
59 | char host[64];
|
---|
60 | char time[20];
|
---|
61 | char msg[1024];
|
---|
62 | char sev[8];
|
---|
63 | char path[12288];
|
---|
64 | char user[9];
|
---|
65 | char group[9];
|
---|
66 | char program[8];
|
---|
67 | char subroutine[16];
|
---|
68 | char status[12];
|
---|
69 | char hash[50];
|
---|
70 | char path_data[1024];
|
---|
71 | char hash_data[50];
|
---|
72 | char key_uid[64];
|
---|
73 | char key_uid_data[64];
|
---|
74 | char key_id[16];
|
---|
75 | char module[8];
|
---|
76 | char syscall[16];
|
---|
77 | char ip[16];
|
---|
78 | char tty[16];
|
---|
79 | char peer[64];
|
---|
80 | char fromhost[64];
|
---|
81 | char obj[1024];
|
---|
82 | char interface[64];
|
---|
83 | char ltime[64];
|
---|
84 | char dir[1024];
|
---|
85 | char linked_path[1024];
|
---|
86 | char service[64];
|
---|
87 | char facility[32];
|
---|
88 | char priority[32];
|
---|
89 | char syslog_msg[1024];
|
---|
90 |
|
---|
91 | char mode_old[16];
|
---|
92 | char mode_new[16];
|
---|
93 | char attr_old[16];
|
---|
94 | char attr_new[16];
|
---|
95 | char device_old[16];
|
---|
96 | char device_new[16];
|
---|
97 | char owner_old[9];
|
---|
98 | char owner_new[9];
|
---|
99 | char group_old[9];
|
---|
100 | char group_new[9];
|
---|
101 | char ctime_old[20];
|
---|
102 | char ctime_new[20];
|
---|
103 | char atime_old[20];
|
---|
104 | char atime_new[20];
|
---|
105 | char mtime_old[20];
|
---|
106 | char mtime_new[20];
|
---|
107 | char chksum_old[50];
|
---|
108 | char chksum_new[50];
|
---|
109 | char link_old[1024];
|
---|
110 | char link_new[1024];
|
---|
111 | char acl_old[1024];
|
---|
112 | char acl_new[1024];
|
---|
113 |
|
---|
114 | long long_data[20];
|
---|
115 |
|
---|
116 | /*
|
---|
117 | long size_old;
|
---|
118 | long size_new;
|
---|
119 | long hardlinks_old;
|
---|
120 | long hardlinks_new;
|
---|
121 | long inode_old;
|
---|
122 | long inode_new;
|
---|
123 | */
|
---|
124 |
|
---|
125 | } dbins;
|
---|
126 |
|
---|
127 | static my_attr * attr_tab_srch = NULL;
|
---|
128 | static int attr_tab_srch_siz = 0;
|
---|
129 |
|
---|
130 | static my_attr attr_tab[] = {
|
---|
131 | { NULL, N_("sev"), 0, 1, 8, 0, offsetof(struct dbins_, sev) },
|
---|
132 | { NULL, N_("tstamp"), 0, 2, 16, 0, offsetof(struct dbins_, time) },
|
---|
133 | { NULL, N_("remote_host"), 0, 3, 64, 0, offsetof(struct dbins_, host) },
|
---|
134 | { NULL, N_("msg"), 0, 4, 1024, 0, offsetof(struct dbins_, msg) },
|
---|
135 |
|
---|
136 | { NULL, N_("path"), 0, 5,12288, 0, offsetof(struct dbins_, path) },
|
---|
137 | /* username -> userid; replace (long) 'userid' - below - by 'dummy' */
|
---|
138 | { NULL, N_("userid"), 0, 6, 9, 0, offsetof(struct dbins_, user) },
|
---|
139 | { NULL, N_("group"), 0, 7, 9, 0, offsetof(struct dbins_, group) },
|
---|
140 | { NULL, N_("program"), 0, 8, 8, 0, offsetof(struct dbins_, program) },
|
---|
141 | { NULL, N_("subroutine"), 0, 9, 16, 0, offsetof(struct dbins_, subroutine)},
|
---|
142 | { NULL, N_("status"), 0, 10, 12, 0, offsetof(struct dbins_, status) },
|
---|
143 | { NULL, N_("hash"), 0, 11, 50, 0, offsetof(struct dbins_, hash) },
|
---|
144 | { NULL, N_("path_data"), 0, 12, 1024, 0, offsetof(struct dbins_, path_data) },
|
---|
145 | { NULL, N_("hash_data"), 0, 13, 50, 0, offsetof(struct dbins_, hash_data) },
|
---|
146 | { NULL, N_("key_uid"), 0, 14, 64, 0, offsetof(struct dbins_, key_uid) },
|
---|
147 | { NULL, N_("key_uid_data"),0, 15, 64, 0, offsetof(struct dbins_, key_uid_data)},
|
---|
148 | { NULL, N_("key_id"), 0, 16, 16, 0, offsetof(struct dbins_, key_id) },
|
---|
149 | { NULL, N_("module"), 0, 17, 8, 0, offsetof(struct dbins_, module) },
|
---|
150 | { NULL, N_("syscall"), 0, 19, 16, 0, offsetof(struct dbins_, syscall) },
|
---|
151 | { NULL, N_("ip"), 0, 20, 16, 0, offsetof(struct dbins_, ip) },
|
---|
152 | { NULL, N_("tty"), 0, 21, 16, 0, offsetof(struct dbins_, tty) },
|
---|
153 | { NULL, N_("peer"), 0, 22, 64, 0, offsetof(struct dbins_, peer) },
|
---|
154 | { NULL, N_("obj"), 0, 23, 1024, 0, offsetof(struct dbins_, obj) },
|
---|
155 | { NULL, N_("interface"), 0, 24, 64, 0, offsetof(struct dbins_, interface)},
|
---|
156 | { NULL, N_("time"), 0, 25, 64, 0, offsetof(struct dbins_, ltime) },
|
---|
157 | { NULL, N_("dir"), 0, 26, 1024, 0, offsetof(struct dbins_, dir) },
|
---|
158 | { NULL, N_("linked_path"), 0, 27, 1024, 0, offsetof(struct dbins_, linked_path)},
|
---|
159 | { NULL, N_("service"), 0, 29, 64, 0, offsetof(struct dbins_, service)},
|
---|
160 | { NULL, N_("facility"), 0, 30, 32, 0, offsetof(struct dbins_, facility) },
|
---|
161 | { NULL, N_("priority"), 0, 31, 32, 0, offsetof(struct dbins_, priority) },
|
---|
162 | { NULL, N_("syslog_msg"), 0, 32, 1024, 0, offsetof(struct dbins_, syslog_msg) },
|
---|
163 |
|
---|
164 | { NULL, N_("mode_old"), 0, 33, 16, 0, offsetof(struct dbins_, mode_old) },
|
---|
165 | { NULL, N_("mode_new"), 0, 34, 16, 0, offsetof(struct dbins_, mode_new) },
|
---|
166 | { NULL, N_("device_old"), 0, 35, 16, 0, offsetof(struct dbins_, device_old)},
|
---|
167 | { NULL, N_("device_new"), 0, 36, 16, 0, offsetof(struct dbins_, device_new)},
|
---|
168 | { NULL, N_("owner_old"), 0, 37, 9, 0, offsetof(struct dbins_, owner_old)},
|
---|
169 | { NULL, N_("owner_new"), 0, 38, 9, 0, offsetof(struct dbins_, owner_new)},
|
---|
170 | { NULL, N_("group_old"), 0, 39, 9, 0, offsetof(struct dbins_, group_old)},
|
---|
171 | { NULL, N_("group_new"), 0, 40, 9, 0, offsetof(struct dbins_, group_new)},
|
---|
172 | { NULL, N_("ctime_old"), 0, 41, 20, 0, offsetof(struct dbins_, ctime_old)},
|
---|
173 | { NULL, N_("ctime_new"), 0, 42, 20, 0, offsetof(struct dbins_, ctime_new)},
|
---|
174 | { NULL, N_("atime_old"), 0, 43, 20, 0, offsetof(struct dbins_, atime_old)},
|
---|
175 | { NULL, N_("atime_new"), 0, 44, 20, 0, offsetof(struct dbins_, atime_new)},
|
---|
176 | { NULL, N_("mtime_old"), 0, 45, 20, 0, offsetof(struct dbins_, mtime_old)},
|
---|
177 | { NULL, N_("mtime_new"), 0, 46, 20, 0, offsetof(struct dbins_, mtime_new)},
|
---|
178 | { NULL, N_("chksum_old"), 0, 47, 50, 0, offsetof(struct dbins_, chksum_old)},
|
---|
179 | { NULL, N_("chksum_new"), 0, 48, 50, 0, offsetof(struct dbins_, chksum_new)},
|
---|
180 | { NULL, N_("link_old"), 0, 49, 1024, 0, offsetof(struct dbins_, link_old)},
|
---|
181 | { NULL, N_("link_new"), 0, 50, 1024, 0, offsetof(struct dbins_, link_new)},
|
---|
182 |
|
---|
183 | { NULL, N_("size_old"), 0, 51, 0, 0, 0 },
|
---|
184 | { NULL, N_("size_new"), 0, 52, 0, 0, 0 },
|
---|
185 | { NULL, N_("hardlinks_old"),0, 53, 0, 0, 0 },
|
---|
186 | { NULL, N_("hardlinks_new"),0, 54, 0, 0, 0 },
|
---|
187 | { NULL, N_("inode_old"), 0, 55, 0, 0, 0 },
|
---|
188 | { NULL, N_("inode_new"), 0, 56, 0, 0, 0 },
|
---|
189 |
|
---|
190 | { NULL, N_("imode_old"), 0, 57, 0, 0, 0 },
|
---|
191 | { NULL, N_("imode_new"), 0, 58, 0, 0, 0 },
|
---|
192 | { NULL, N_("iattr_old"), 0, 59, 0, 0, 0 },
|
---|
193 | { NULL, N_("iattr_new"), 0, 60, 0, 0, 0 },
|
---|
194 | { NULL, N_("idevice_old"), 0, 61, 0, 0, 0 },
|
---|
195 | { NULL, N_("idevice_new"), 0, 62, 0, 0, 0 },
|
---|
196 | { NULL, N_("iowner_old"), 0, 63, 0, 0, 0 },
|
---|
197 | { NULL, N_("iowner_new"), 0, 64, 0, 0, 0 },
|
---|
198 | { NULL, N_("igroup_old"), 0, 65, 0, 0, 0 },
|
---|
199 | { NULL, N_("igroup_new"), 0, 66, 0, 0, 0 },
|
---|
200 |
|
---|
201 | { NULL, N_("port"), 0, 67, 0, 0, 0 },
|
---|
202 | { NULL, N_("return_code"), 0, 68, 0, 0, 0 },
|
---|
203 | /* { NULL, N_("userid"), 0, 69, 0, 0 }, old 'userid', 1.8.1 */
|
---|
204 |
|
---|
205 | { NULL, N_("host"), 0, 70, 64, 0, offsetof(struct dbins_, fromhost)},
|
---|
206 | { NULL, N_("attr_old"), 0, 71, 16, 0, offsetof(struct dbins_, attr_old)},
|
---|
207 | { NULL, N_("attr_new"), 0, 72, 16, 0, offsetof(struct dbins_, attr_new)},
|
---|
208 | { NULL, N_("acl_old"), 0, 73, 1024, 0, offsetof(struct dbins_, acl_old)},
|
---|
209 | { NULL, N_("acl_new"), 0, 74, 1024, 0, offsetof(struct dbins_, acl_new)},
|
---|
210 |
|
---|
211 | { NULL, NULL, 0, 0, 0, 0, 0 }
|
---|
212 | };
|
---|
213 |
|
---|
214 | #define SH_SLOT_HOST 70
|
---|
215 | #define SH_SLOT_GROUP 7
|
---|
216 | #define START_SEC_LONGS 51
|
---|
217 | #define END_SEC_LONGS 68
|
---|
218 |
|
---|
219 | #if defined(HAVE_INT_32)
|
---|
220 | typedef unsigned int uint32;
|
---|
221 | #elif defined(HAVE_LONG_32)
|
---|
222 | typedef unsigned long uint32;
|
---|
223 | #elif defined(HAVE_SHORT_32)
|
---|
224 | typedef unsigned short uint32;
|
---|
225 | #else
|
---|
226 | #error No 32 byte type found !
|
---|
227 | #endif
|
---|
228 |
|
---|
229 | typedef unsigned char uint8;
|
---|
230 |
|
---|
231 | typedef struct md5_ctx
|
---|
232 | {
|
---|
233 | uint32 A;
|
---|
234 | uint32 B;
|
---|
235 | uint32 C;
|
---|
236 | uint32 D;
|
---|
237 |
|
---|
238 | uint32 total[2];
|
---|
239 | uint32 buflen;
|
---|
240 | char buffer[128];
|
---|
241 | } md5Param;
|
---|
242 |
|
---|
243 |
|
---|
244 | typedef unsigned char sh_byte;
|
---|
245 |
|
---|
246 |
|
---|
247 | extern int md5Reset(register md5Param* p);
|
---|
248 | extern int md5Update(md5Param* p, const sh_byte* data, int size);
|
---|
249 | extern int md5Digest(md5Param* p, uint32* data);
|
---|
250 |
|
---|
251 | static char db_name[64] = "";
|
---|
252 | static char db_table[64] = "";
|
---|
253 | static char db_host[64] = "";
|
---|
254 | static char db_user[64] = "";
|
---|
255 | static char db_password[64] = "";
|
---|
256 |
|
---|
257 | static int sh_persistent_dbconn = S_TRUE;
|
---|
258 |
|
---|
259 | int sh_database_use_persistent (const char * str)
|
---|
260 | {
|
---|
261 | return sh_util_flagval (str, &sh_persistent_dbconn);
|
---|
262 | }
|
---|
263 |
|
---|
264 | static int insert_value (char * ptr, const char * str)
|
---|
265 | {
|
---|
266 | if (!ptr || !str)
|
---|
267 | return -1;
|
---|
268 | if (strlen(str) > 63)
|
---|
269 | return -1;
|
---|
270 | (void) sl_strlcpy(ptr, str, 64);
|
---|
271 | return 0;
|
---|
272 | }
|
---|
273 |
|
---|
274 | static void init_db_entry (dbins * ptr)
|
---|
275 | {
|
---|
276 | memset (ptr, (int) '\0', sizeof(dbins));
|
---|
277 | ptr->next = NULL;
|
---|
278 | return;
|
---|
279 | }
|
---|
280 |
|
---|
281 |
|
---|
282 | int sh_database_set_database (const char * str)
|
---|
283 | {
|
---|
284 | return insert_value (db_name, str);
|
---|
285 | }
|
---|
286 | int sh_database_set_table (const char * str)
|
---|
287 | {
|
---|
288 | return insert_value (db_table, str);
|
---|
289 | }
|
---|
290 | int sh_database_set_host (const char * str)
|
---|
291 | {
|
---|
292 | return insert_value (db_host, str);
|
---|
293 | }
|
---|
294 | int sh_database_set_user (const char * str)
|
---|
295 | {
|
---|
296 | return insert_value (db_user, str);
|
---|
297 | }
|
---|
298 | int sh_database_set_password (const char * str)
|
---|
299 | {
|
---|
300 | return insert_value (db_password, str);
|
---|
301 | }
|
---|
302 |
|
---|
303 | /******************************************************************
|
---|
304 | *
|
---|
305 | * Oracle and unixODBC stuff, only Oracle tested untested
|
---|
306 | *
|
---|
307 | * Based on the code in the snort output plugin (spo_database.c).
|
---|
308 | * Copyright/license statement in spo_database.c:
|
---|
309 | *
|
---|
310 | * Portions Copyright (C) 2000,2001,2002 Carnegie Mellon University
|
---|
311 | * Copyright (C) 2001 Jed Pickel <jed@pickel.net>
|
---|
312 | * Portions Copyright (C) 2001 Andrew R. Baker <andrewb@farm9.com>
|
---|
313 | *
|
---|
314 | * This program is free software; you can redistribute it and/or modify
|
---|
315 | * it under the terms of the GNU General Public License as published by
|
---|
316 | * the Free Software Foundation; either version 2 of the License, or
|
---|
317 | * (at your option) any later version.
|
---|
318 | *
|
---|
319 | * This program is distributed in the hope that it will be useful,
|
---|
320 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
321 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
322 | * GNU General Public License for more details.
|
---|
323 | *
|
---|
324 | * You should have received a copy of the GNU General Public License
|
---|
325 | * along with this program; if not, write to the Free Software
|
---|
326 | * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
---|
327 | *
|
---|
328 | ******************************************************************/
|
---|
329 | #ifdef WITH_ODBC
|
---|
330 |
|
---|
331 | #include <sql.h>
|
---|
332 | #include <sqlext.h>
|
---|
333 | #include <sqltypes.h>
|
---|
334 |
|
---|
335 | static SQLHENV u_handle;
|
---|
336 | static SQLHDBC u_connection;
|
---|
337 | static SQLHSTMT u_statement;
|
---|
338 | static SQLINTEGER u_col;
|
---|
339 | static SQLINTEGER u_rows;
|
---|
340 |
|
---|
341 | void sh_database_reset()
|
---|
342 | {
|
---|
343 | return;
|
---|
344 | }
|
---|
345 |
|
---|
346 | static
|
---|
347 | int sh_database_query (char * query, /*@out@*/ long * id)
|
---|
348 | {
|
---|
349 | static int fatal_error = 0;
|
---|
350 | int result = 0;
|
---|
351 | char row_query[128];
|
---|
352 | long result_call;
|
---|
353 |
|
---|
354 | SL_ENTER(_("sh_database_query"));
|
---|
355 |
|
---|
356 | *id = 0;
|
---|
357 |
|
---|
358 | if (fatal_error == 1)
|
---|
359 | {
|
---|
360 | SL_RETURN((-1), _("sh_database_query"));
|
---|
361 | }
|
---|
362 |
|
---|
363 | /* Connect
|
---|
364 | */
|
---|
365 | if (db_name[0] == '\0')
|
---|
366 | sl_strlcpy(db_name, _("samhain"), 64);
|
---|
367 |
|
---|
368 | if (db_user[0] == '\0')
|
---|
369 | sl_strlcpy(db_user, _("samhain"), 64);
|
---|
370 |
|
---|
371 | result_call = SQLAllocEnv(&u_handle);
|
---|
372 | if ((result_call != SQL_SUCCESS) && (result_call != SQL_SUCCESS_WITH_INFO))
|
---|
373 | {
|
---|
374 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, result_call,
|
---|
375 | MSG_E_SUBGEN,
|
---|
376 | _("Error in SQLAllocEnv when connecting to ODBC data source"),
|
---|
377 | _("sh_database_query"));
|
---|
378 | fatal_error = 1;
|
---|
379 | SL_RETURN((-1), _("sh_database_query"));
|
---|
380 | }
|
---|
381 | result_call = SQLAllocConnect(u_handle, &u_connection);
|
---|
382 | if ((result_call != SQL_SUCCESS) && (result_call != SQL_SUCCESS_WITH_INFO))
|
---|
383 | {
|
---|
384 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, result_call,
|
---|
385 | MSG_E_SUBGEN,
|
---|
386 | _("Error in SQLAllocEnv when connecting to ODBC data source"),
|
---|
387 | _("sh_database_query"));
|
---|
388 | fatal_error = 1;
|
---|
389 | SL_RETURN((-1), _("sh_database_query"));
|
---|
390 | }
|
---|
391 | result_call = SQLConnect(u_connection, db_name, SQL_NTS,
|
---|
392 | db_user, SQL_NTS, db_password, SQL_NTS);
|
---|
393 | if ((result_call != SQL_SUCCESS) && (result_call != SQL_SUCCESS_WITH_INFO))
|
---|
394 | {
|
---|
395 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, result_call,
|
---|
396 | MSG_E_SUBGEN,
|
---|
397 | _("Error in SQLAllocEnv when connecting to ODBC data source"),
|
---|
398 | _("sh_database_query"));
|
---|
399 | fatal_error = 1;
|
---|
400 | SL_RETURN((-1), _("sh_database_query"));
|
---|
401 | }
|
---|
402 |
|
---|
403 | /* Insert
|
---|
404 | */
|
---|
405 | result_call = SQLAllocStmt(u_connection, &u_statement);
|
---|
406 | if ((result_call == SQL_SUCCESS) || (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
407 | {
|
---|
408 | result_call = SQLPrepare(u_statement, query, SQL_NTS);
|
---|
409 | if ((result_call == SQL_SUCCESS) ||
|
---|
410 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
411 | {
|
---|
412 | result_call = SQLExecute(u_statement);
|
---|
413 | if((result_call == SQL_SUCCESS) ||
|
---|
414 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
415 | {
|
---|
416 | result = 1;
|
---|
417 | }
|
---|
418 | }
|
---|
419 | }
|
---|
420 |
|
---|
421 | if (result == 0)
|
---|
422 | {
|
---|
423 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
424 | _("Error inserting into ODBC data source"),
|
---|
425 | _("sh_database_query"));
|
---|
426 | goto odbc_disconnect;
|
---|
427 | }
|
---|
428 |
|
---|
429 | /* Select
|
---|
430 | */
|
---|
431 | result = 0;
|
---|
432 |
|
---|
433 | sl_strlcpy (row_query, _("SELECT MAX(log_index) FROM "), 128);
|
---|
434 | sl_strlcat (row_query, db_table, 128);
|
---|
435 |
|
---|
436 | result_call = SQLAllocStmt(u_connection, &u_statement);
|
---|
437 | if ((result_call == SQL_SUCCESS) ||
|
---|
438 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
439 | {
|
---|
440 | result_call = SQLPrepare(u_statement, row_query, SQL_NTS);
|
---|
441 | if ((result_call == SQL_SUCCESS) ||
|
---|
442 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
443 | {
|
---|
444 | result_call = SQLExecute(u_statement);
|
---|
445 | if ((result_call == SQL_SUCCESS) ||
|
---|
446 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
447 | {
|
---|
448 | result_call = SQLRowCount(u_statement, &u_rows);
|
---|
449 | if ((result_call == SQL_SUCCESS) ||
|
---|
450 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
451 | {
|
---|
452 | if((u_rows) && (u_rows == 1))
|
---|
453 | {
|
---|
454 | result_call = SQLFetch(u_statement);
|
---|
455 | if ((result_call == SQL_SUCCESS) ||
|
---|
456 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
457 | {
|
---|
458 | result_call = SQLGetData(u_statement, 1,
|
---|
459 | SQL_INTEGER, &u_col,
|
---|
460 | sizeof(u_col), NULL);
|
---|
461 | if ((result_call == SQL_SUCCESS) ||
|
---|
462 | (result_call == SQL_SUCCESS_WITH_INFO))
|
---|
463 | {
|
---|
464 | *id = (long int) u_col;
|
---|
465 | result = 1;
|
---|
466 | }
|
---|
467 | }
|
---|
468 | }
|
---|
469 | }
|
---|
470 | }
|
---|
471 | }
|
---|
472 | }
|
---|
473 |
|
---|
474 | if (result == 0)
|
---|
475 | {
|
---|
476 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
477 | _("Error selecting MAX(log_index) from ODBC data source"),
|
---|
478 | _("sh_database_query"));
|
---|
479 | }
|
---|
480 |
|
---|
481 | odbc_disconnect:
|
---|
482 | SQLFreeHandle(SQL_HANDLE_STMT, u_statement);
|
---|
483 | SQLDisconnect(u_connection);
|
---|
484 | SQLFreeHandle(SQL_HANDLE_DBC, u_connection);
|
---|
485 | SQLFreeHandle(SQL_HANDLE_ENV, u_handle);
|
---|
486 |
|
---|
487 | SL_RETURN(((result == 0) ? -1 : 0), _("sh_database_query"));
|
---|
488 |
|
---|
489 | }
|
---|
490 |
|
---|
491 | /* #ifdef WITH_ODBC */
|
---|
492 | #endif
|
---|
493 |
|
---|
494 | #ifdef WITH_ORACLE
|
---|
495 |
|
---|
496 | #include <oci.h>
|
---|
497 |
|
---|
498 | static OCIDefine * o_define;
|
---|
499 | static OCIEnv * o_environment;
|
---|
500 | static OCISvcCtx * o_servicecontext;
|
---|
501 | static OCIError * o_error = NULL;
|
---|
502 | static OCIStmt * o_statement;
|
---|
503 | static OCIBind * o_bind = (OCIBind *) 0;
|
---|
504 | static text o_errormsg[512];
|
---|
505 | static sb4 o_errorcode;
|
---|
506 |
|
---|
507 | static int connected = 0;
|
---|
508 |
|
---|
509 | void sh_database_reset()
|
---|
510 | {
|
---|
511 | if (connected == 1)
|
---|
512 | {
|
---|
513 | OCILogoff(o_servicecontext, o_error);
|
---|
514 | OCIHandleFree((dvoid *) o_statement, OCI_HTYPE_STMT);
|
---|
515 | OCIHandleFree((dvoid *) o_servicecontext, OCI_HTYPE_SVCCTX);
|
---|
516 | OCIHandleFree((dvoid *) o_error, OCI_HTYPE_ERROR);
|
---|
517 | o_error = NULL;
|
---|
518 | }
|
---|
519 | connected = 0;
|
---|
520 | return;
|
---|
521 | }
|
---|
522 |
|
---|
523 | static char * sh_stripnl (char * str)
|
---|
524 | {
|
---|
525 | size_t len = sl_strlen(str);
|
---|
526 | if (len > 0)
|
---|
527 | {
|
---|
528 | if (str[len-1] == '\n')
|
---|
529 | str[len-1] = '\0';
|
---|
530 | }
|
---|
531 | return str;
|
---|
532 | }
|
---|
533 |
|
---|
534 | static
|
---|
535 | int sh_database_query (char * query, /*@out@*/ long * id)
|
---|
536 | {
|
---|
537 | static int bad_init = 0;
|
---|
538 | int result = 0;
|
---|
539 | char row_query[128];
|
---|
540 | int retry = 0;
|
---|
541 |
|
---|
542 | SL_ENTER(_("sh_database_query"));
|
---|
543 |
|
---|
544 | *id = 0;
|
---|
545 |
|
---|
546 | if (bad_init == 1) {
|
---|
547 | SL_RETURN(-1, _("sh_database_query"));
|
---|
548 | }
|
---|
549 | else if (connected == 1) {
|
---|
550 | goto oracle_connected;
|
---|
551 | }
|
---|
552 |
|
---|
553 | /*
|
---|
554 | * Connect
|
---|
555 | */
|
---|
556 | #define PRINT_ORACLE_ERR(func_name) \
|
---|
557 | do { \
|
---|
558 | OCIErrorGet(o_error, 1, NULL, &o_errorcode, \
|
---|
559 | o_errormsg, sizeof(o_errormsg), \
|
---|
560 | OCI_HTYPE_ERROR); \
|
---|
561 | sh_stripnl (o_errormsg); \
|
---|
562 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN, \
|
---|
563 | o_errormsg, _("sh_database_query")); \
|
---|
564 | sl_snprintf(row_query, 127, \
|
---|
565 | _("%s: Connection to database '%s' failed"), \
|
---|
566 | func_name, db_name); \
|
---|
567 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN, \
|
---|
568 | row_query, _("sh_database_query")); \
|
---|
569 | bad_init = 1; \
|
---|
570 | SL_RETURN(-1, _("sh_database_query")); \
|
---|
571 | } while (1 == 0)
|
---|
572 |
|
---|
573 | oracle_doconnect:
|
---|
574 |
|
---|
575 | if (!getenv("ORACLE_HOME")) /* flawfinder: ignore */
|
---|
576 | {
|
---|
577 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
578 | _("ORACLE_HOME environment variable not set"),
|
---|
579 | _("sh_database_query"));
|
---|
580 | }
|
---|
581 | if (db_name[0] == '\0')
|
---|
582 | {
|
---|
583 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
584 | _("database name not set, using default 'samhain'"),
|
---|
585 | _("sh_database_query"));
|
---|
586 | sl_strlcpy(db_name, _("samhain"), 64);
|
---|
587 | }
|
---|
588 | if (db_user[0] == '\0')
|
---|
589 | {
|
---|
590 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
591 | _("database user not set, using default 'samhain'"),
|
---|
592 | _("sh_database_query"));
|
---|
593 | sl_strlcpy(db_user, _("samhain"), 64);
|
---|
594 | }
|
---|
595 | if (db_password[0] == '\0')
|
---|
596 | {
|
---|
597 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
598 | _("database password not set, cannot proceed"),
|
---|
599 | _("sh_database_query"));
|
---|
600 | bad_init = 1;
|
---|
601 | SL_RETURN(-1, _("sh_database_query"));
|
---|
602 | }
|
---|
603 |
|
---|
604 |
|
---|
605 | #ifdef DB_DEBUG
|
---|
606 | sl_snprintf(row_query, 127,
|
---|
607 | _("Conncting to oracle database '%s'"),
|
---|
608 | db_name);
|
---|
609 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
610 | row_query,
|
---|
611 | _("sh_database_query"));
|
---|
612 | #endif
|
---|
613 |
|
---|
614 | /* a) Oracle says use OCIEnvCreate instead of OCIInitialize/OCIEnvcreate
|
---|
615 | * b) why two times OCIEnvInit() ???
|
---|
616 | */
|
---|
617 | if (OCIInitialize(OCI_DEFAULT, NULL, NULL, NULL, NULL))
|
---|
618 | PRINT_ORACLE_ERR("OCIInitialize");
|
---|
619 |
|
---|
620 | if (OCIEnvInit(&o_environment, OCI_DEFAULT, 0, NULL))
|
---|
621 | PRINT_ORACLE_ERR("OCIEnvInit");
|
---|
622 |
|
---|
623 | if (OCIEnvInit(&o_environment, OCI_DEFAULT, 0, NULL))
|
---|
624 | PRINT_ORACLE_ERR("OCIEnvInit (2)");
|
---|
625 |
|
---|
626 | /* allocate and initialize the error handle
|
---|
627 | */
|
---|
628 | if (OCIHandleAlloc(o_environment, (dvoid **)&o_error,
|
---|
629 | OCI_HTYPE_ERROR, (size_t) 0, NULL))
|
---|
630 | PRINT_ORACLE_ERR("OCIHandleAlloc");
|
---|
631 |
|
---|
632 | /* logon and allocate the service context handle
|
---|
633 | */
|
---|
634 | if (OCILogon(o_environment, o_error, &o_servicecontext,
|
---|
635 | (OraText*) db_user, sl_strlen(db_user),
|
---|
636 | (OraText*) db_password, sl_strlen(db_password),
|
---|
637 | (OraText*) db_name, sl_strlen(db_name)))
|
---|
638 | {
|
---|
639 | OCIErrorGet(o_error, 1, NULL, &o_errorcode,
|
---|
640 | o_errormsg, sizeof(o_errormsg), OCI_HTYPE_ERROR);
|
---|
641 | sh_stripnl (o_errormsg);
|
---|
642 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
643 | o_errormsg,
|
---|
644 | _("sh_database_query"));
|
---|
645 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
646 | _("check database is listed in tnsnames.ora"),
|
---|
647 | _("sh_database_query"));
|
---|
648 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
649 | _("check tnsnames.ora readable"),
|
---|
650 | _("sh_database_query"));
|
---|
651 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
652 | _("check database accessible with sqlplus"),
|
---|
653 | _("sh_database_query"));
|
---|
654 | sl_snprintf(row_query, 127,
|
---|
655 | _("OCILogon: Connection to database '%s' failed"),
|
---|
656 | db_name);
|
---|
657 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
658 | row_query, _("sh_database_query"));
|
---|
659 | bad_init = 1;
|
---|
660 | SL_RETURN(-1, _("sh_database_query"));
|
---|
661 | }
|
---|
662 |
|
---|
663 | if (OCIHandleAlloc(o_environment, (dvoid **)&o_statement,
|
---|
664 | OCI_HTYPE_STMT, 0, NULL))
|
---|
665 | PRINT_ORACLE_ERR("OCIHandleAlloc (2)");
|
---|
666 |
|
---|
667 | /* Flag connection status
|
---|
668 | */
|
---|
669 | connected = 1;
|
---|
670 |
|
---|
671 | oracle_connected:
|
---|
672 |
|
---|
673 | /* Get row index
|
---|
674 | */
|
---|
675 | sl_strlcpy (row_query, _("SELECT log_log_index_seq.NEXTVAL FROM dual"), 128);
|
---|
676 |
|
---|
677 | #ifdef DB_DEBUG
|
---|
678 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
679 | row_query,
|
---|
680 | _("sh_database_query"));
|
---|
681 | #endif
|
---|
682 |
|
---|
683 | if (OCIStmtPrepare(o_statement, o_error,
|
---|
684 | (OraText*) row_query, sl_strlen(row_query),
|
---|
685 | OCI_NTV_SYNTAX, OCI_DEFAULT))
|
---|
686 | {
|
---|
687 | OCIErrorGet(o_error, 1, NULL,
|
---|
688 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
689 | OCI_HTYPE_ERROR);
|
---|
690 | sh_stripnl (o_errormsg);
|
---|
691 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
692 | o_errormsg,
|
---|
693 | _("sh_database_query"));
|
---|
694 | if (retry == 0 &&
|
---|
695 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
696 | {
|
---|
697 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
698 | }
|
---|
699 | goto err_out;
|
---|
700 | }
|
---|
701 |
|
---|
702 | if (OCIStmtExecute(o_servicecontext, o_statement, o_error,
|
---|
703 | 0, 0, NULL, NULL, OCI_DEFAULT))
|
---|
704 | {
|
---|
705 | OCIErrorGet(o_error, 1, NULL,
|
---|
706 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
707 | OCI_HTYPE_ERROR);
|
---|
708 | sh_stripnl (o_errormsg);
|
---|
709 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
710 | o_errormsg,
|
---|
711 | _("sh_database_query"));
|
---|
712 | if (retry == 0 &&
|
---|
713 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
714 | {
|
---|
715 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
716 | }
|
---|
717 | goto err_out;
|
---|
718 | }
|
---|
719 |
|
---|
720 | if (OCIDefineByPos (o_statement, &o_define, o_error, 1,
|
---|
721 | &result, sizeof(result),
|
---|
722 | SQLT_INT, 0, 0, 0, OCI_DEFAULT))
|
---|
723 | {
|
---|
724 | OCIErrorGet(o_error, 1, NULL,
|
---|
725 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
726 | OCI_HTYPE_ERROR);
|
---|
727 | sh_stripnl (o_errormsg);
|
---|
728 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
729 | o_errormsg,
|
---|
730 | _("sh_database_query"));
|
---|
731 | if (retry == 0 &&
|
---|
732 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
733 | {
|
---|
734 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
735 | }
|
---|
736 | goto err_out;
|
---|
737 | }
|
---|
738 | if (OCIStmtFetch (o_statement, o_error, 1, OCI_FETCH_NEXT, OCI_DEFAULT))
|
---|
739 | {
|
---|
740 | OCIErrorGet(o_error, 1, NULL,
|
---|
741 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
742 | OCI_HTYPE_ERROR);
|
---|
743 | sh_stripnl (o_errormsg);
|
---|
744 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
745 | o_errormsg,
|
---|
746 | _("sh_database_query"));
|
---|
747 | if (retry == 0 &&
|
---|
748 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
749 | {
|
---|
750 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
751 | }
|
---|
752 | goto err_out;
|
---|
753 | }
|
---|
754 |
|
---|
755 | #ifdef DB_DEBUG
|
---|
756 | sl_snprintf(row_query, 127, _("Returned value: %d"), result);
|
---|
757 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
758 | row_query,
|
---|
759 | _("sh_database_query"));
|
---|
760 | #endif
|
---|
761 |
|
---|
762 | *id = result;
|
---|
763 |
|
---|
764 | /* do the insert
|
---|
765 | */
|
---|
766 | #ifdef DB_DEBUG
|
---|
767 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
768 | query,
|
---|
769 | _("sh_database_query"));
|
---|
770 | #endif
|
---|
771 |
|
---|
772 | if (OCIStmtPrepare(o_statement, o_error,
|
---|
773 | (OraText*) query, sl_strlen(query),
|
---|
774 | OCI_NTV_SYNTAX, OCI_DEFAULT))
|
---|
775 | {
|
---|
776 | OCIErrorGet(o_error, 1, NULL,
|
---|
777 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
778 | OCI_HTYPE_ERROR);
|
---|
779 | sh_stripnl (o_errormsg);
|
---|
780 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
781 | o_errormsg,
|
---|
782 | _("sh_database_query"));
|
---|
783 | if (retry == 0 &&
|
---|
784 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
785 | {
|
---|
786 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
787 | }
|
---|
788 | goto err_out;
|
---|
789 | }
|
---|
790 |
|
---|
791 | if (OCIBindByPos(o_statement, &o_bind, o_error, 1,
|
---|
792 | (dvoid *) &result, (sword) sizeof(result), SQLT_INT,
|
---|
793 | (dvoid *) 0, (ub2 *) 0, (ub2 *) 0, (ub4) 0, (ub4 *) 0, OCI_DEFAULT))
|
---|
794 | {
|
---|
795 | OCIErrorGet(o_error, 1, NULL,
|
---|
796 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
797 | OCI_HTYPE_ERROR);
|
---|
798 | sh_stripnl (o_errormsg);
|
---|
799 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
800 | o_errormsg,
|
---|
801 | _("sh_database_query"));
|
---|
802 | if (retry == 0 &&
|
---|
803 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
804 | {
|
---|
805 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
806 | }
|
---|
807 | goto err_out;
|
---|
808 | }
|
---|
809 |
|
---|
810 | if (OCIStmtExecute(o_servicecontext,
|
---|
811 | o_statement, o_error, 1, 0,
|
---|
812 | NULL, NULL, OCI_COMMIT_ON_SUCCESS))
|
---|
813 | {
|
---|
814 | OCIErrorGet(o_error, 1, NULL,
|
---|
815 | &o_errorcode, o_errormsg, sizeof(o_errormsg),
|
---|
816 | OCI_HTYPE_ERROR);
|
---|
817 | sh_stripnl (o_errormsg);
|
---|
818 | sh_error_handle((-1), FIL__, __LINE__, (long) o_errorcode, MSG_E_SUBGEN,
|
---|
819 | o_errormsg,
|
---|
820 | _("sh_database_query"));
|
---|
821 | if (retry == 0 &&
|
---|
822 | (3114 == o_errorcode || 0 == strncmp(o_errormsg, _("ORA-03114"), 9)))
|
---|
823 | {
|
---|
824 | ++retry; sh_database_reset(); goto oracle_doconnect;
|
---|
825 | }
|
---|
826 | goto err_out;
|
---|
827 | }
|
---|
828 |
|
---|
829 | #ifdef DB_DEBUG
|
---|
830 | sh_error_handle(SH_ERR_NOTICE, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
831 | _("No error on insert"),
|
---|
832 | _("sh_database_query"));
|
---|
833 | #endif
|
---|
834 |
|
---|
835 | if (sh_persistent_dbconn == S_FALSE)
|
---|
836 | {
|
---|
837 | OCILogoff(o_servicecontext, o_error);
|
---|
838 | OCIHandleFree((dvoid *) o_statement, OCI_HTYPE_STMT);
|
---|
839 | OCIHandleFree((dvoid *) o_servicecontext, OCI_HTYPE_SVCCTX);
|
---|
840 | OCIHandleFree((dvoid *) o_error, OCI_HTYPE_ERROR);
|
---|
841 | o_error = NULL;
|
---|
842 | connected = 0;
|
---|
843 | }
|
---|
844 | SL_RETURN(0, _("sh_database_query"));
|
---|
845 |
|
---|
846 | err_out:
|
---|
847 | /*
|
---|
848 | * Error
|
---|
849 | */
|
---|
850 | if (sh_persistent_dbconn == S_FALSE)
|
---|
851 | {
|
---|
852 | OCILogoff(o_servicecontext, o_error);
|
---|
853 | OCIHandleFree((dvoid *) o_statement, OCI_HTYPE_STMT);
|
---|
854 | OCIHandleFree((dvoid *) o_servicecontext, OCI_HTYPE_SVCCTX);
|
---|
855 | OCIHandleFree((dvoid *) o_error, OCI_HTYPE_ERROR);
|
---|
856 | o_error = NULL;
|
---|
857 | connected = 0;
|
---|
858 | }
|
---|
859 | SL_RETURN(-1, _("sh_database_query"));
|
---|
860 | }
|
---|
861 |
|
---|
862 | /* #ifdef WITH_ORACLE */
|
---|
863 | #endif
|
---|
864 |
|
---|
865 | #ifdef WITH_POSTGRES
|
---|
866 | /******************************************************************
|
---|
867 | *
|
---|
868 | * Postgresql stuff, tested
|
---|
869 | *
|
---|
870 | ******************************************************************/
|
---|
871 |
|
---|
872 | #ifdef HAVE_PGSQL_LIBPQ_FE_H
|
---|
873 | #include <pgsql/libpq-fe.h>
|
---|
874 | #else
|
---|
875 | #include <libpq-fe.h>
|
---|
876 | #endif
|
---|
877 |
|
---|
878 | static int connection_status = S_FALSE;
|
---|
879 |
|
---|
880 | void sh_database_reset()
|
---|
881 | {
|
---|
882 | connection_status = S_FALSE;
|
---|
883 | return;
|
---|
884 | }
|
---|
885 |
|
---|
886 | static
|
---|
887 | int sh_database_query (char * query, /*@out@*/ long * id)
|
---|
888 | {
|
---|
889 | char conninfo[256];
|
---|
890 | char * p;
|
---|
891 | static PGconn * conn = NULL;
|
---|
892 | PGresult * res;
|
---|
893 | unsigned int i;
|
---|
894 | const char * params[1];
|
---|
895 | char id_param[32];
|
---|
896 | static SH_TIMEOUT sh_timer = { 0, 3600, S_TRUE };
|
---|
897 |
|
---|
898 | SL_ENTER(_("sh_database_query"));
|
---|
899 |
|
---|
900 | *id = 0;
|
---|
901 |
|
---|
902 | p = &conninfo[0];
|
---|
903 |
|
---|
904 | if (db_host[0] == '\0')
|
---|
905 | sl_strlcpy(db_host, _("localhost"), 64);
|
---|
906 | if (db_name[0] == '\0')
|
---|
907 | sl_strlcpy(db_name, _("samhain"), 64);
|
---|
908 | if (db_user[0] == '\0')
|
---|
909 | sl_strlcpy(db_user, _("samhain"), 64);
|
---|
910 |
|
---|
911 | if (db_host[0] != '\0' && NULL != strchr(db_host, '.'))
|
---|
912 | {
|
---|
913 | sl_snprintf(p, 255, "hostaddr=%s ", db_host);
|
---|
914 | p = &conninfo[strlen(conninfo)];
|
---|
915 | }
|
---|
916 | if (db_name[0] != '\0')
|
---|
917 | {
|
---|
918 | sl_snprintf(p, 255 - strlen(conninfo), "dbname=%s ", db_name);
|
---|
919 | p = &conninfo[strlen(conninfo)];
|
---|
920 | }
|
---|
921 |
|
---|
922 | if (db_user[0] != '\0')
|
---|
923 | {
|
---|
924 | sl_snprintf(p, 255 - strlen(conninfo), "user=%s ", db_user);
|
---|
925 | p = &conninfo[strlen(conninfo)];
|
---|
926 | }
|
---|
927 |
|
---|
928 | if (db_password[0] != '\0')
|
---|
929 | {
|
---|
930 | sl_snprintf(p, 255 - strlen(conninfo), "password=%s ", db_password);
|
---|
931 | }
|
---|
932 |
|
---|
933 | if (connection_status == S_FALSE)
|
---|
934 | {
|
---|
935 | if (conn)
|
---|
936 | PQfinish(conn);
|
---|
937 | conn = NULL;
|
---|
938 | conn = PQconnectdb(conninfo);
|
---|
939 | }
|
---|
940 | else
|
---|
941 | {
|
---|
942 | if (PQstatus(conn) == CONNECTION_BAD)
|
---|
943 | PQreset(conn);
|
---|
944 | }
|
---|
945 |
|
---|
946 | if ((conn == NULL) || (PQstatus(conn) == CONNECTION_BAD))
|
---|
947 | {
|
---|
948 | connection_status = S_FALSE;
|
---|
949 |
|
---|
950 | sh_timer.flag_ok = S_FALSE;
|
---|
951 | if (S_TRUE == sh_util_timeout_check(&sh_timer))
|
---|
952 | {
|
---|
953 | goto err_out;
|
---|
954 | }
|
---|
955 | else
|
---|
956 | {
|
---|
957 | if (conn)
|
---|
958 | PQfinish(conn);
|
---|
959 | conn = NULL;
|
---|
960 | SL_RETURN(0, _("sh_database_query"));
|
---|
961 | }
|
---|
962 | }
|
---|
963 | connection_status = S_TRUE;
|
---|
964 |
|
---|
965 |
|
---|
966 | /* get the unique row index
|
---|
967 | */
|
---|
968 | res = PQexec(conn, _("SELECT NEXTVAL('log_log_index_seq')"));
|
---|
969 | if (PQresultStatus(res) != PGRES_TUPLES_OK)
|
---|
970 | {
|
---|
971 | PQclear(res);
|
---|
972 | goto err_out;
|
---|
973 | }
|
---|
974 |
|
---|
975 | *id = atoi (PQgetvalue(res, 0, 0));
|
---|
976 | PQclear(res);
|
---|
977 |
|
---|
978 | sl_snprintf(id_param, 32, "%ld", *id);
|
---|
979 | params[0] = id_param;
|
---|
980 |
|
---|
981 | /* do the insert
|
---|
982 | */
|
---|
983 | res = PQexecParams(conn, query, 1, NULL, params, NULL, NULL, 1);
|
---|
984 | if (PQresultStatus(res) != PGRES_COMMAND_OK)
|
---|
985 | {
|
---|
986 | PQclear(res);
|
---|
987 | goto err_out;
|
---|
988 | }
|
---|
989 | PQclear(res);
|
---|
990 |
|
---|
991 | if (S_FALSE == sh_persistent_dbconn)
|
---|
992 | {
|
---|
993 | if (conn)
|
---|
994 | PQfinish(conn);
|
---|
995 | conn = NULL;
|
---|
996 | connection_status = S_FALSE;
|
---|
997 | }
|
---|
998 | SL_RETURN(0, _("sh_database_query"));
|
---|
999 |
|
---|
1000 |
|
---|
1001 | err_out:
|
---|
1002 | if (conn)
|
---|
1003 | {
|
---|
1004 | p = PQerrorMessage(conn);
|
---|
1005 | for (i = 0; i < sl_strlen(p); ++i)
|
---|
1006 | if (p[i] == '\n') p[i] = ' ';
|
---|
1007 | }
|
---|
1008 | else
|
---|
1009 | {
|
---|
1010 | p = NULL;
|
---|
1011 | }
|
---|
1012 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
1013 | (p == NULL ? _("(null)") : p),
|
---|
1014 | _("sh_database_query"));
|
---|
1015 | if (conn)
|
---|
1016 | PQfinish(conn);
|
---|
1017 | conn = NULL;
|
---|
1018 | connection_status = S_FALSE;
|
---|
1019 | SL_RETURN(-1, _("sh_database_query"));
|
---|
1020 | }
|
---|
1021 | #endif
|
---|
1022 |
|
---|
1023 |
|
---|
1024 | #ifdef WITH_MYSQL
|
---|
1025 |
|
---|
1026 | #ifdef HAVE_MYSQL_MYSQL_H
|
---|
1027 | #include <mysql/mysql.h>
|
---|
1028 | #else
|
---|
1029 | #include <mysql.h>
|
---|
1030 | #endif
|
---|
1031 |
|
---|
1032 | extern int flag_err_debug;
|
---|
1033 |
|
---|
1034 | static int connection_status = S_FALSE;
|
---|
1035 |
|
---|
1036 | void sh_database_reset()
|
---|
1037 | {
|
---|
1038 | connection_status = S_FALSE;
|
---|
1039 | return;
|
---|
1040 | }
|
---|
1041 |
|
---|
1042 | static
|
---|
1043 | int sh_database_query (char * query, /*@out@*/ long * id)
|
---|
1044 | {
|
---|
1045 | int status = 0;
|
---|
1046 | const char * p;
|
---|
1047 | static MYSQL * db_conn = NULL;
|
---|
1048 | static SH_TIMEOUT sh_timer = { 0, 3600, S_TRUE };
|
---|
1049 |
|
---|
1050 | SL_ENTER(_("sh_database_query"));
|
---|
1051 |
|
---|
1052 | *id = 0;
|
---|
1053 |
|
---|
1054 | if (query == NULL)
|
---|
1055 | {
|
---|
1056 | sh_error_handle(SH_ERR_ALL, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
1057 | _("NULL query"),
|
---|
1058 | _("sh_database_query"));
|
---|
1059 | SL_RETURN(0, _("sh_database_query"));
|
---|
1060 | }
|
---|
1061 |
|
---|
1062 | if (db_host[0] == '\0')
|
---|
1063 | (void) sl_strlcpy(db_host, _("localhost"), 64);
|
---|
1064 | if (db_name[0] == '\0')
|
---|
1065 | (void) sl_strlcpy(db_name, _("samhain"), 64);
|
---|
1066 | if (db_user[0] == '\0')
|
---|
1067 | (void) sl_strlcpy(db_user, _("samhain"), 64);
|
---|
1068 |
|
---|
1069 | if ((db_conn == NULL) || (connection_status == S_FALSE))
|
---|
1070 | {
|
---|
1071 | if (db_conn)
|
---|
1072 | {
|
---|
1073 | mysql_close(db_conn);
|
---|
1074 | db_conn = NULL;
|
---|
1075 | }
|
---|
1076 | connection_status = S_FALSE;
|
---|
1077 |
|
---|
1078 | db_conn = mysql_init(NULL);
|
---|
1079 | if (NULL == db_conn)
|
---|
1080 | {
|
---|
1081 | p = NULL; status = 0;
|
---|
1082 | sh_timer.flag_ok = S_FALSE;
|
---|
1083 | if (S_TRUE == sh_util_timeout_check(&sh_timer))
|
---|
1084 | {
|
---|
1085 | goto alt_out;
|
---|
1086 | }
|
---|
1087 | else
|
---|
1088 | {
|
---|
1089 | SL_RETURN(0, _("sh_database_query"));
|
---|
1090 | }
|
---|
1091 | }
|
---|
1092 |
|
---|
1093 | /* Read in defaults from /etc/my.cnf and associated files,
|
---|
1094 | * suggested by arjones at simultan dyndns org
|
---|
1095 | * see: - http://dev.mysql.com/doc/refman/5.0/en/option-files.html
|
---|
1096 | * for the my.cnf format,
|
---|
1097 | * - http://dev.mysql.com/doc/refman/5.0/en/mysql-options.html
|
---|
1098 | * for possible options
|
---|
1099 | * We don't check the return value because it's useless (failure due
|
---|
1100 | * to lack of access permission is not reported).
|
---|
1101 | */
|
---|
1102 | mysql_options(db_conn, MYSQL_READ_DEFAULT_GROUP, _("samhain"));
|
---|
1103 |
|
---|
1104 | status = 0;
|
---|
1105 |
|
---|
1106 | if (NULL == mysql_real_connect(db_conn,
|
---|
1107 | db_host[0] == '\0' ? NULL : db_host,
|
---|
1108 | db_user[0] == '\0' ? NULL : db_user,
|
---|
1109 | db_password[0] == '\0' ? NULL : db_password,
|
---|
1110 | db_name[0] == '\0' ? NULL : db_name,
|
---|
1111 | 0, NULL, 0))
|
---|
1112 | {
|
---|
1113 | sh_timer.flag_ok = S_FALSE;
|
---|
1114 | if (S_TRUE == sh_util_timeout_check(&sh_timer))
|
---|
1115 | {
|
---|
1116 | goto err_out;
|
---|
1117 | }
|
---|
1118 | else
|
---|
1119 | {
|
---|
1120 | SL_RETURN(0, _("sh_database_query"));
|
---|
1121 | }
|
---|
1122 | }
|
---|
1123 | connection_status = S_TRUE;
|
---|
1124 | }
|
---|
1125 | else
|
---|
1126 | {
|
---|
1127 | if (0 != mysql_ping(db_conn))
|
---|
1128 | {
|
---|
1129 | connection_status = S_FALSE;
|
---|
1130 | sh_timer.flag_ok = S_FALSE;
|
---|
1131 | if (S_TRUE == sh_util_timeout_check(&sh_timer))
|
---|
1132 | {
|
---|
1133 | goto err_out;
|
---|
1134 | }
|
---|
1135 | else
|
---|
1136 | {
|
---|
1137 | SL_RETURN(0, _("sh_database_query"));
|
---|
1138 | }
|
---|
1139 | }
|
---|
1140 | }
|
---|
1141 |
|
---|
1142 | if (0 != mysql_query(db_conn, query))
|
---|
1143 | {
|
---|
1144 | goto err_out;
|
---|
1145 | }
|
---|
1146 |
|
---|
1147 | if (flag_err_debug == SL_TRUE)
|
---|
1148 | {
|
---|
1149 | p = mysql_info (db_conn);
|
---|
1150 | if (p != NULL)
|
---|
1151 | {
|
---|
1152 | sh_error_handle(SH_ERR_ALL, FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
1153 | p,
|
---|
1154 | _("sh_database_query"));
|
---|
1155 | }
|
---|
1156 | }
|
---|
1157 |
|
---|
1158 | *id = (long) mysql_insert_id(db_conn);
|
---|
1159 | if (S_FALSE == sh_persistent_dbconn)
|
---|
1160 | {
|
---|
1161 | if (db_conn)
|
---|
1162 | mysql_close(db_conn);
|
---|
1163 | db_conn = NULL;
|
---|
1164 | connection_status = S_FALSE;
|
---|
1165 | }
|
---|
1166 | SL_RETURN(0, _("sh_database_query"));
|
---|
1167 |
|
---|
1168 | err_out:
|
---|
1169 |
|
---|
1170 | if (db_conn)
|
---|
1171 | {
|
---|
1172 | p = mysql_error (db_conn);
|
---|
1173 | status = (int) mysql_errno (db_conn);
|
---|
1174 | }
|
---|
1175 | else
|
---|
1176 | {
|
---|
1177 | p = NULL; p = 0;
|
---|
1178 | }
|
---|
1179 |
|
---|
1180 | alt_out:
|
---|
1181 |
|
---|
1182 | *id = 0;
|
---|
1183 | sh_error_handle((-1), FIL__, __LINE__, status, MSG_E_SUBGEN,
|
---|
1184 | (p == NULL ? _("(null)") : p),
|
---|
1185 | _("sh_database_query"));
|
---|
1186 | if (db_conn)
|
---|
1187 | mysql_close(db_conn);
|
---|
1188 | db_conn = NULL;
|
---|
1189 | connection_status = S_FALSE;
|
---|
1190 | SL_RETURN(status, _("sh_database_query"));
|
---|
1191 | }
|
---|
1192 | #endif
|
---|
1193 |
|
---|
1194 | static
|
---|
1195 | char * null_or_val (char * end, char * val, int * size, int flag)
|
---|
1196 | {
|
---|
1197 | long len;
|
---|
1198 |
|
---|
1199 | if (!((end == NULL) || (val == NULL) || (size == NULL)))
|
---|
1200 | {
|
---|
1201 | if (val[0] != '\0')
|
---|
1202 | {
|
---|
1203 | if (*size > 1)
|
---|
1204 | {
|
---|
1205 | *end = ','; ++end; (*size) -= 1;
|
---|
1206 | if (flag == 1) { *end = '\''; ++end; (*size) -= 1; }
|
---|
1207 | *end = '\0';
|
---|
1208 | }
|
---|
1209 | len = (long) strlen(val);
|
---|
1210 | if ((long) *size > (len+1))
|
---|
1211 | {
|
---|
1212 | (void) sl_strlcat(end, val, (size_t) *size);
|
---|
1213 | end += len; (*size) -= len;
|
---|
1214 | if (flag == 1) { *end = '\''; ++end; (*size) -= 1; }
|
---|
1215 | *end = '\0';
|
---|
1216 | }
|
---|
1217 | }
|
---|
1218 | }
|
---|
1219 |
|
---|
1220 | return end;
|
---|
1221 | }
|
---|
1222 |
|
---|
1223 | #define SH_QUERY_MAX 16383
|
---|
1224 |
|
---|
1225 | static
|
---|
1226 | long sh_database_entry (dbins * db_entry, long id)
|
---|
1227 | {
|
---|
1228 | /* This does not need to be re-entrant
|
---|
1229 | */
|
---|
1230 | char * query;
|
---|
1231 | static char columns[1024];
|
---|
1232 | char * values;
|
---|
1233 |
|
---|
1234 | int status;
|
---|
1235 | long the_id;
|
---|
1236 | int size;
|
---|
1237 | char * end;
|
---|
1238 | int c_size;
|
---|
1239 | char * c_end;
|
---|
1240 | char * p;
|
---|
1241 | int i;
|
---|
1242 | char num[64];
|
---|
1243 |
|
---|
1244 | md5Param crc;
|
---|
1245 | unsigned char md5buffer[16];
|
---|
1246 | char md5out[33];
|
---|
1247 | int cnt;
|
---|
1248 |
|
---|
1249 | size_t len_val;
|
---|
1250 | size_t len_col;
|
---|
1251 |
|
---|
1252 | SL_ENTER(_("sh_database_entry"));
|
---|
1253 |
|
---|
1254 | query = SH_ALLOC(SH_QUERY_MAX+1);
|
---|
1255 | values = SH_ALLOC(SH_QUERY_MAX+1);
|
---|
1256 |
|
---|
1257 | (void) md5Reset(&crc);
|
---|
1258 |
|
---|
1259 | if (db_entry->host[0] == '\0')
|
---|
1260 | {
|
---|
1261 | if (sh.host.name[0] == '\0')
|
---|
1262 | (void) strcpy (db_entry->host, _("localhost")); /* known to fit */
|
---|
1263 | else
|
---|
1264 | (void) sl_strlcpy (db_entry->host, sh.host.name, 64);
|
---|
1265 | }
|
---|
1266 |
|
---|
1267 | /*@-bufferoverflowhigh@*/
|
---|
1268 | if (id >= 0)
|
---|
1269 | sprintf(num, "%ld", id); /* known to fit */
|
---|
1270 | /*@+bufferoverflowhigh@*/
|
---|
1271 |
|
---|
1272 | #if defined(WITH_ORACLE)
|
---|
1273 | /* Oracle needs some help for the time format (fix by Michael Somers)
|
---|
1274 | */
|
---|
1275 | (void)
|
---|
1276 | sl_snprintf (values, SH_QUERY_MAX,
|
---|
1277 | _("(:1,%s,%c%s%c,to_date(%c%s%c,'YYYY-MM-DD HH24:MI:SS'),%c%s%c,%c%s%c"),
|
---|
1278 | id >= 0 ? num : _("NULL"),
|
---|
1279 | '\'', db_entry->host,'\'',
|
---|
1280 | '\'', db_entry->time,'\'',
|
---|
1281 | '\'', db_entry->sev, '\'',
|
---|
1282 | '\'',
|
---|
1283 | (db_entry->msg[0] == '\0' ? _("NULL") : db_entry->msg),
|
---|
1284 | '\'');
|
---|
1285 | (void) sl_snprintf (columns, 1023,
|
---|
1286 | _("(log_index,log_ref,log_host,log_time,log_sev,log_msg"));
|
---|
1287 | #elif defined(WITH_POSTGRES)
|
---|
1288 | /* Prepare query for PQexecParams
|
---|
1289 | */
|
---|
1290 | (void)
|
---|
1291 | sl_snprintf (values, SH_QUERY_MAX,
|
---|
1292 | _("($1,%s,%c%s%c,%c%s%c,%c%s%c,%c%s%c"),
|
---|
1293 | id >= 0 ? num : _("NULL"),
|
---|
1294 | '\'', db_entry->host,'\'',
|
---|
1295 | '\'', db_entry->time,'\'',
|
---|
1296 | '\'', db_entry->sev, '\'',
|
---|
1297 | '\'',
|
---|
1298 | (db_entry->msg[0] == '\0' ? _("NULL") : db_entry->msg),
|
---|
1299 | '\'');
|
---|
1300 | (void) sl_snprintf (columns, 1023,
|
---|
1301 | _("(log_index,log_ref,log_host,log_time,log_sev,log_msg"));
|
---|
1302 | #else
|
---|
1303 | (void)
|
---|
1304 | sl_snprintf (values, SH_QUERY_MAX, _("(%s,%c%s%c,%c%s%c,%c%s%c,%c%s%c"),
|
---|
1305 | id >= 0 ? num : _("NULL"),
|
---|
1306 | '\'', db_entry->host,'\'',
|
---|
1307 | '\'', db_entry->time,'\'',
|
---|
1308 | '\'', db_entry->sev, '\'',
|
---|
1309 | '\'',
|
---|
1310 | (db_entry->msg[0] == '\0' ? _("NULL") : db_entry->msg),
|
---|
1311 | '\'');
|
---|
1312 | (void) sl_snprintf (columns, 1023,
|
---|
1313 | _("(log_ref,log_host,log_time,log_sev,log_msg"));
|
---|
1314 | #endif
|
---|
1315 |
|
---|
1316 |
|
---|
1317 | /*@-type@*//* byte* versus char[..] */
|
---|
1318 | if (attr_tab[0].inHash == 1)
|
---|
1319 | (void) md5Update(&crc, (sh_byte*) db_entry->sev,
|
---|
1320 | (int) strlen(db_entry->sev));
|
---|
1321 | if (attr_tab[1].inHash == 1)
|
---|
1322 | (void) md5Update(&crc, (sh_byte*) db_entry->time,
|
---|
1323 | (int) strlen(db_entry->time));
|
---|
1324 | if (attr_tab[2].inHash == 1)
|
---|
1325 | (void) md5Update(&crc, (sh_byte*) db_entry->host,
|
---|
1326 | (int) strlen(db_entry->host));
|
---|
1327 | if (attr_tab[3].inHash == 1 && db_entry->msg[0] != '\0')
|
---|
1328 | (void) md5Update(&crc, (sh_byte*) db_entry->msg,
|
---|
1329 | (int) strlen(db_entry->sev));
|
---|
1330 | /*@+type@*/
|
---|
1331 |
|
---|
1332 | len_val = strlen(values);
|
---|
1333 | size = (int) (SH_QUERY_MAX - len_val);
|
---|
1334 | end = values + len_val;
|
---|
1335 |
|
---|
1336 | len_col = strlen(columns);
|
---|
1337 | c_size = 1023 - (int) len_col; /* sizeof(colums) == 1024 */
|
---|
1338 | c_end = columns + len_col;
|
---|
1339 |
|
---|
1340 | i = 4;
|
---|
1341 |
|
---|
1342 | while (attr_tab[i].attr != NULL)
|
---|
1343 | {
|
---|
1344 | if (attr_tab[i].size != 0)
|
---|
1345 | {
|
---|
1346 | if (attr_tab[i].val > 40 && attr_tab[i].val < 47)
|
---|
1347 | {
|
---|
1348 | /* remove the 'T' between date and time
|
---|
1349 | */
|
---|
1350 | p = (char *)(db_entry)+attr_tab[i].off;
|
---|
1351 | p = strchr(p, 'T');
|
---|
1352 | if (p) *p = ' ';
|
---|
1353 | }
|
---|
1354 | p = end;
|
---|
1355 | end = null_or_val(end,((char *)(db_entry)+attr_tab[i].off),&size,1);
|
---|
1356 | if (p != end)
|
---|
1357 | {
|
---|
1358 | if ((attr_tab[i].val != SH_SLOT_HOST) &&
|
---|
1359 | (attr_tab[i].val != SH_SLOT_GROUP))
|
---|
1360 | {
|
---|
1361 | c_end = null_or_val (c_end, attr_tab[i].attr, &c_size,0);
|
---|
1362 | }
|
---|
1363 | else
|
---|
1364 | {
|
---|
1365 | /*
|
---|
1366 | * 'host' is a reserved word in SQL
|
---|
1367 | */
|
---|
1368 | if (attr_tab[i].val == SH_SLOT_HOST)
|
---|
1369 | c_end = null_or_val (c_end, _("fromhost"), &c_size,0);
|
---|
1370 | /*
|
---|
1371 | * 'group' is a reserved word in SQL
|
---|
1372 | */
|
---|
1373 | else /* if (attr_tab[i].val == SH_SLOT_GROUP) */
|
---|
1374 | c_end = null_or_val (c_end, _("grp"), &c_size,0);
|
---|
1375 | }
|
---|
1376 | }
|
---|
1377 | /*@-type@*//* byte* versus char[..] */
|
---|
1378 | if (attr_tab[i].inHash == 1 &&
|
---|
1379 | ((char *)(db_entry)+attr_tab[i].off) != '\0')
|
---|
1380 | {
|
---|
1381 | (void)md5Update(&crc,
|
---|
1382 | (sh_byte*) ((char *)(db_entry)+attr_tab[i].off),
|
---|
1383 | (int)strlen((char *)(db_entry)+attr_tab[i].off));
|
---|
1384 | }
|
---|
1385 | /*@+type@*/
|
---|
1386 | }
|
---|
1387 | else if (attr_tab[i].val >= START_SEC_LONGS &&
|
---|
1388 | attr_tab[i].val <= END_SEC_LONGS)
|
---|
1389 | {
|
---|
1390 | (void)
|
---|
1391 | sl_snprintf(end, (size_t)(size-1), _(",\'%ld\'"),
|
---|
1392 | db_entry->long_data[attr_tab[i].val-START_SEC_LONGS]);
|
---|
1393 | while (*end != '\0') { ++end; --size; }
|
---|
1394 | (void) sl_snprintf(c_end, (size_t)(c_size-1),
|
---|
1395 | _(",%s"), attr_tab[i].attr);
|
---|
1396 | while (*c_end != '\0') { ++c_end; --c_size; }
|
---|
1397 | if (attr_tab[i].inHash == 1)
|
---|
1398 | {
|
---|
1399 | /*@-type@*//* byte* versus char[..] */
|
---|
1400 | (void)
|
---|
1401 | md5Update(&crc,
|
---|
1402 | (sh_byte *) db_entry->long_data[attr_tab[i].val-START_SEC_LONGS],
|
---|
1403 | sizeof(long));
|
---|
1404 | /*@+type@*/
|
---|
1405 | }
|
---|
1406 | }
|
---|
1407 |
|
---|
1408 | ++i;
|
---|
1409 | }
|
---|
1410 |
|
---|
1411 | (void) md5Digest(&crc, (uint32 *) md5buffer);
|
---|
1412 | /*@-bufferoverflowhigh -usedef@*/
|
---|
1413 | for (cnt = 0; cnt < 16; ++cnt)
|
---|
1414 | sprintf (&md5out[cnt*2], _("%02X"), /* known to fit */
|
---|
1415 | (unsigned int) md5buffer[cnt]);
|
---|
1416 | /*@+bufferoverflowhigh +usedef@*/
|
---|
1417 | md5out[32] = '\0';
|
---|
1418 |
|
---|
1419 | (void) sl_snprintf(end, (size_t) (size-1), _(",%c%s%c"), '\'', md5out, '\'');
|
---|
1420 | while (*end != '\0') { ++end; --size; }
|
---|
1421 | (void) sl_snprintf(c_end, (size_t) (c_size-1),_(",log_hash"));
|
---|
1422 | while (*c_end != '\0') { ++c_end; --c_size; }
|
---|
1423 |
|
---|
1424 |
|
---|
1425 | if (size > 1) { *end = ')'; ++end; *end = '\0'; }
|
---|
1426 | if (c_size > 1) { *c_end = ')'; ++c_end; *c_end = '\0'; }
|
---|
1427 |
|
---|
1428 | if (db_table[0] == '\0')
|
---|
1429 | (void) sl_strlcpy(db_table, _("log"), 64);
|
---|
1430 |
|
---|
1431 | (void) sl_snprintf (query, SH_QUERY_MAX,
|
---|
1432 | _("INSERT INTO %s %s VALUES %s"),
|
---|
1433 | db_table, columns, values);
|
---|
1434 |
|
---|
1435 | status = sh_database_query (query, &the_id);
|
---|
1436 |
|
---|
1437 | /*@-usedef@*//* no, 'values' is allocated here */
|
---|
1438 | SH_FREE(values);
|
---|
1439 | /*@+usedef@*/
|
---|
1440 | SH_FREE(query);
|
---|
1441 |
|
---|
1442 | SL_RETURN(the_id, _("sh_database_entry"));
|
---|
1443 | }
|
---|
1444 |
|
---|
1445 | static int sh_database_comp_attr (const void *m1, const void *m2)
|
---|
1446 | {
|
---|
1447 | my_attr *mi1 = (my_attr *) m1;
|
---|
1448 | my_attr *mi2 = (my_attr *) m2;
|
---|
1449 | return strcmp(mi1->attr, mi2->attr);
|
---|
1450 | }
|
---|
1451 |
|
---|
1452 |
|
---|
1453 | static void init_attr_table()
|
---|
1454 | {
|
---|
1455 | static int first = S_TRUE;
|
---|
1456 | int i, j;
|
---|
1457 |
|
---|
1458 | #ifdef SH_STEALTH
|
---|
1459 | int k;
|
---|
1460 |
|
---|
1461 | if (first == S_FALSE)
|
---|
1462 | return;
|
---|
1463 |
|
---|
1464 | i = 0;
|
---|
1465 | while (attr_tab[i].attr_o != NULL)
|
---|
1466 | {
|
---|
1467 | j = strlen(attr_tab[i].attr_o);
|
---|
1468 | attr_tab[i].attr = malloc (j+1); /* only once */
|
---|
1469 | if (NULL == attr_tab[i].attr)
|
---|
1470 | return;
|
---|
1471 | for (k = 0; k < j; ++k)
|
---|
1472 | attr_tab[i].attr[k] = attr_tab[i].attr_o[k] ^ XOR_CODE;
|
---|
1473 | attr_tab[i].attr[j] = '\0';
|
---|
1474 | attr_tab[i].alen = strlen(attr_tab[i].attr_o);
|
---|
1475 | ++i;
|
---|
1476 | }
|
---|
1477 | first = S_FALSE;
|
---|
1478 |
|
---|
1479 | #else
|
---|
1480 |
|
---|
1481 | if (first == S_FALSE)
|
---|
1482 | return;
|
---|
1483 |
|
---|
1484 | i = 0;
|
---|
1485 | while (attr_tab[i].attr_o != NULL)
|
---|
1486 | {
|
---|
1487 | attr_tab[i].attr = attr_tab[i].attr_o;
|
---|
1488 | attr_tab[i].alen = strlen(attr_tab[i].attr_o);
|
---|
1489 | ++i;
|
---|
1490 | }
|
---|
1491 | first = S_FALSE;
|
---|
1492 |
|
---|
1493 | #endif
|
---|
1494 |
|
---|
1495 | /* create a sorted table for binary search
|
---|
1496 | */
|
---|
1497 | attr_tab_srch = SH_ALLOC(i * sizeof(my_attr));
|
---|
1498 | for (j=0; j<i; ++j)
|
---|
1499 | memcpy(&attr_tab_srch[j], &attr_tab[j], sizeof(my_attr));
|
---|
1500 | qsort(attr_tab_srch, i, sizeof(my_attr), sh_database_comp_attr);
|
---|
1501 | attr_tab_srch_siz = i;
|
---|
1502 |
|
---|
1503 | return;
|
---|
1504 | }
|
---|
1505 |
|
---|
1506 | int sh_database_add_to_hash (const char * str)
|
---|
1507 | {
|
---|
1508 | int i;
|
---|
1509 |
|
---|
1510 | if (!str)
|
---|
1511 | return -1;
|
---|
1512 | init_attr_table();
|
---|
1513 | if (0 == strcmp(str, _("log_msg"))) { attr_tab[3].inHash = 1; return 0;}
|
---|
1514 | if (0 == strcmp(str, _("log_sev"))) { attr_tab[0].inHash = 1; return 0;}
|
---|
1515 | if (0 == strcmp(str, _("log_time"))) { attr_tab[1].inHash = 1; return 0;}
|
---|
1516 | if (0 == strcmp(str, _("log_host"))) { attr_tab[2].inHash = 1; return 0;}
|
---|
1517 | i = 4;
|
---|
1518 | while (attr_tab[i].attr != NULL)
|
---|
1519 | {
|
---|
1520 | if (0 == strcmp(str, attr_tab[i].attr))
|
---|
1521 | { attr_tab[i].inHash = 1; return 0; }
|
---|
1522 | ++i;
|
---|
1523 | }
|
---|
1524 | return -1;
|
---|
1525 | }
|
---|
1526 |
|
---|
1527 | static int is_escaped(char * p_in) {
|
---|
1528 |
|
---|
1529 | int escp = 0;
|
---|
1530 | int retv = S_TRUE;
|
---|
1531 | unsigned char * p = (unsigned char *) p_in;
|
---|
1532 |
|
---|
1533 | if (*p != '\0')
|
---|
1534 | {
|
---|
1535 | do
|
---|
1536 | {
|
---|
1537 | if (*p <= 0x7F)
|
---|
1538 | {
|
---|
1539 | if (escp == 0)
|
---|
1540 | {
|
---|
1541 | if (!((*p == '\'') || (*p == '\"') || (*p != '\\')))
|
---|
1542 | /* do nothing */;
|
---|
1543 | else if (*p == '\\') escp = 1;
|
---|
1544 | else retv = S_FALSE; /* (*p == '\'' || *p == '\"') */
|
---|
1545 | }
|
---|
1546 | else /* escp == 1 */
|
---|
1547 | {
|
---|
1548 | escp = 0;
|
---|
1549 | }
|
---|
1550 | }
|
---|
1551 | else /* *p > 0x7F */
|
---|
1552 | {
|
---|
1553 | retv = S_FALSE;
|
---|
1554 | }
|
---|
1555 |
|
---|
1556 | ++p;
|
---|
1557 |
|
---|
1558 | }
|
---|
1559 | while (*p != '\0');
|
---|
1560 | }
|
---|
1561 |
|
---|
1562 | if (escp == 0)
|
---|
1563 | return retv;
|
---|
1564 | else
|
---|
1565 | return S_FALSE;
|
---|
1566 | }
|
---|
1567 |
|
---|
1568 | /* this is not a real XML parser, but it copes with the XML format of
|
---|
1569 | * the log messages provided by sh_error_handle()
|
---|
1570 | */
|
---|
1571 | static
|
---|
1572 | char * sh_database_parse (char * message, dbins * db_entry)
|
---|
1573 | {
|
---|
1574 | static int first = S_TRUE;
|
---|
1575 | char * p;
|
---|
1576 | char * q;
|
---|
1577 | char * z;
|
---|
1578 | dbins * new;
|
---|
1579 | int i;
|
---|
1580 | size_t j;
|
---|
1581 | my_attr * res;
|
---|
1582 | my_attr key;
|
---|
1583 | char key_str[64];
|
---|
1584 |
|
---|
1585 | SL_ENTER(_("sh_database_parse"));
|
---|
1586 |
|
---|
1587 | if (!message || *message == '\0')
|
---|
1588 | SL_RETURN (NULL, _("sh_database_parse"));
|
---|
1589 |
|
---|
1590 | if (first == S_TRUE)
|
---|
1591 | {
|
---|
1592 | init_attr_table();
|
---|
1593 | first = S_FALSE;
|
---|
1594 | }
|
---|
1595 |
|
---|
1596 | p = strchr (message, '<');
|
---|
1597 | if (!p)
|
---|
1598 | SL_RETURN (NULL, _("sh_database_parse"));
|
---|
1599 |
|
---|
1600 | while ((*p != '\0') && (*p != '>'))
|
---|
1601 | {
|
---|
1602 | if (p[0] == 'l' && p[1] == 'o' && p[2] == 'g' &&
|
---|
1603 | (p[3] == ' ' || p[3] == '>'))
|
---|
1604 | {
|
---|
1605 | p = &p[4];
|
---|
1606 | goto parse;
|
---|
1607 | }
|
---|
1608 | else if (p[0] == '/' && p[1] == '>')
|
---|
1609 | SL_RETURN (&p[2], _("sh_database_parse"));
|
---|
1610 | else if (p[0] == '/' && p[1] == 'l' && p[2] == 'o' &&
|
---|
1611 | p[3] == 'g' && p[4] == '>')
|
---|
1612 | SL_RETURN (&p[5], _("sh_database_parse"));
|
---|
1613 | ++p;
|
---|
1614 | }
|
---|
1615 | SL_RETURN(NULL, _("sh_database_parse"));
|
---|
1616 |
|
---|
1617 | parse:
|
---|
1618 |
|
---|
1619 | while (*p == ' ' || *p == '>')
|
---|
1620 | ++p;
|
---|
1621 |
|
---|
1622 | if (*p == '\0')
|
---|
1623 | SL_RETURN(NULL, _("sh_database_parse"));
|
---|
1624 |
|
---|
1625 | if (*p != '<' && *p != '/')
|
---|
1626 | goto par2;
|
---|
1627 |
|
---|
1628 | if (p[0] == '<' && p[1] == 'l' &&
|
---|
1629 | p[2] == 'o' && p[3] == 'g')
|
---|
1630 | {
|
---|
1631 | /*
|
---|
1632 | * recursive call
|
---|
1633 | */
|
---|
1634 | new = SH_ALLOC(sizeof(dbins));
|
---|
1635 | init_db_entry(new);
|
---|
1636 | db_entry->next = new;
|
---|
1637 | p = sh_database_parse (p, new);
|
---|
1638 | }
|
---|
1639 |
|
---|
1640 | if (p[0] == '/' && p[1] == '>')
|
---|
1641 | SL_RETURN (&p[1], _("sh_database_parse"));
|
---|
1642 |
|
---|
1643 | if (p[0] == '<' && p[1] == '/' && p[2] == 'l' &&
|
---|
1644 | p[3] == 'o' && p[4] == 'g' && p[5] == '>')
|
---|
1645 | SL_RETURN (&p[5], _("sh_database_parse"));
|
---|
1646 |
|
---|
1647 | par2:
|
---|
1648 |
|
---|
1649 | /* non-whitespace
|
---|
1650 | */
|
---|
1651 | for (i=0; i < 64; ++i)
|
---|
1652 | {
|
---|
1653 | if (p[i] != '=')
|
---|
1654 | {
|
---|
1655 | key_str[i] = p[i];
|
---|
1656 | }
|
---|
1657 | else
|
---|
1658 | {
|
---|
1659 | key_str[i] = '\0';
|
---|
1660 | break;
|
---|
1661 | }
|
---|
1662 | }
|
---|
1663 | key_str[63] = '\0';
|
---|
1664 | key.attr = &key_str[0];
|
---|
1665 |
|
---|
1666 | res = bsearch(&key, attr_tab_srch, attr_tab_srch_siz,
|
---|
1667 | sizeof(my_attr), sh_database_comp_attr);
|
---|
1668 |
|
---|
1669 | if (res != NULL)
|
---|
1670 | {
|
---|
1671 | j = res->alen; /* strlen(attr_tab[i].attr); */
|
---|
1672 | if (p[j] == '=' && p[j+1] == '"')
|
---|
1673 | {
|
---|
1674 | q = strchr(&p[j+2], '"');
|
---|
1675 | if (q)
|
---|
1676 | {
|
---|
1677 | *q = '\0';
|
---|
1678 |
|
---|
1679 | if (S_TRUE == is_escaped(&p[j+2])) {
|
---|
1680 |
|
---|
1681 | if (res->val == 1)
|
---|
1682 | (void) sl_strlcpy(db_entry->sev, &p[j+2],
|
---|
1683 | (size_t)res->size);
|
---|
1684 | else if (res->val == 2)
|
---|
1685 | {
|
---|
1686 | z = strchr(&p[j+2], 'T');
|
---|
1687 | if (z) *z = ' ';
|
---|
1688 | (void) sl_strlcpy(db_entry->time, &p[j+2], 20);
|
---|
1689 | }
|
---|
1690 | else if (res->val == 3)
|
---|
1691 | (void) sl_strlcpy(db_entry->host, &p[j+2],
|
---|
1692 | (size_t) res->size);
|
---|
1693 | else if (res->val == 4)
|
---|
1694 | (void) sl_strlcpy(db_entry->msg, &p[j+2],
|
---|
1695 | (size_t) res->size);
|
---|
1696 | else if (res->size != 0)
|
---|
1697 | {
|
---|
1698 | (void) sl_strlcpy( (((char *)(db_entry))+ res->off),
|
---|
1699 | &p[j+2],
|
---|
1700 | (size_t) res->size);
|
---|
1701 | }
|
---|
1702 | else if (res->val >= START_SEC_LONGS)
|
---|
1703 | {
|
---|
1704 | db_entry->long_data[res->val-START_SEC_LONGS]
|
---|
1705 | = atol(&p[j+2]);
|
---|
1706 | }
|
---|
1707 |
|
---|
1708 | *q = '"';
|
---|
1709 | p = q;
|
---|
1710 | ++p;
|
---|
1711 |
|
---|
1712 | goto parse;
|
---|
1713 | }
|
---|
1714 | else { /* S_FALSE == is_escaped(&p[j+2]) */
|
---|
1715 | sh_error_handle((-1), FIL__, __LINE__, 0, MSG_E_SUBGEN,
|
---|
1716 | _("Message not properly escaped"),
|
---|
1717 | _("sh_database_parse"));
|
---|
1718 | SL_RETURN(NULL, _("sh_database_parse"));
|
---|
1719 | }
|
---|
1720 | }
|
---|
1721 | else /* q == NULL */
|
---|
1722 | {
|
---|
1723 | SL_RETURN(NULL, _("sh_database_parse"));
|
---|
1724 | }
|
---|
1725 | }
|
---|
1726 | }
|
---|
1727 |
|
---|
1728 | /* unknown attribute, skip
|
---|
1729 | */
|
---|
1730 | while ((p != NULL) && (*p != '\0') && (*p != ' '))
|
---|
1731 | ++p;
|
---|
1732 |
|
---|
1733 | goto parse;
|
---|
1734 | }
|
---|
1735 |
|
---|
1736 | static int enter_wrapper = 1;
|
---|
1737 |
|
---|
1738 | int set_enter_wrapper (const char * str)
|
---|
1739 | {
|
---|
1740 | return sh_util_flagval(str, &enter_wrapper);
|
---|
1741 | }
|
---|
1742 |
|
---|
1743 | /* recursively enter linked list of messages into database, last first
|
---|
1744 | */
|
---|
1745 | long sh_database_insert_rec (dbins * curr, int depth)
|
---|
1746 | {
|
---|
1747 | unsigned long id = 0;
|
---|
1748 | dbins * prev;
|
---|
1749 |
|
---|
1750 | SL_ENTER(_("sh_database_insert_rec"));
|
---|
1751 |
|
---|
1752 | if (curr->next)
|
---|
1753 | {
|
---|
1754 | prev = curr->next;
|
---|
1755 | sl_strlcpy(prev->host, curr->host, 64);
|
---|
1756 | id = sh_database_insert_rec (curr->next, (depth + 1));
|
---|
1757 | }
|
---|
1758 |
|
---|
1759 | if (id != 0) /* this is a server wrapper */
|
---|
1760 | {
|
---|
1761 | if (enter_wrapper != 0)
|
---|
1762 | {
|
---|
1763 | id = sh_database_entry (curr, id);
|
---|
1764 | }
|
---|
1765 | }
|
---|
1766 | else
|
---|
1767 | {
|
---|
1768 | /*
|
---|
1769 | * id = -1 is the client message; log_ref will be NULL
|
---|
1770 | */
|
---|
1771 | if (depth > 0) /* this is a client message */
|
---|
1772 | id = sh_database_entry (curr, -1);
|
---|
1773 | else /* this is a generic server message */
|
---|
1774 | id = sh_database_entry (curr, 0);
|
---|
1775 | }
|
---|
1776 |
|
---|
1777 | SH_FREE(curr);
|
---|
1778 |
|
---|
1779 | SL_RETURN(id, _("sh_database_insert_rec"));
|
---|
1780 | }
|
---|
1781 |
|
---|
1782 | int sh_database_insert (char * message)
|
---|
1783 | {
|
---|
1784 | dbins * db_entry;
|
---|
1785 |
|
---|
1786 | SL_ENTER(_("sh_database_insert"));
|
---|
1787 |
|
---|
1788 | db_entry = SH_ALLOC(sizeof(dbins));
|
---|
1789 | init_db_entry(db_entry);
|
---|
1790 |
|
---|
1791 | /* recursively parse the message into a linked list
|
---|
1792 | */
|
---|
1793 | (void) sh_database_parse (message, db_entry);
|
---|
1794 |
|
---|
1795 | /* recursively enter the linked list into the database
|
---|
1796 | */
|
---|
1797 | (void) sh_database_insert_rec (db_entry, 0);
|
---|
1798 |
|
---|
1799 | SL_RETURN(0, _("sh_database_insert"));
|
---|
1800 | }
|
---|
1801 |
|
---|
1802 | #endif
|
---|