source: trunk/src/sh_database.c@ 78

Last change on this file since 78 was 76, checked in by rainer, 18 years ago

Fix for ticket #38 (csv escaping) and #39 (building on cygwin). Also optimize a bit.
File size: 48.5 KB
Line 
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
46typedef 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
57typedef 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
127static my_attr * attr_tab_srch = NULL;
128static int attr_tab_srch_siz = 0;
129
130static 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)
220typedef unsigned int uint32;
221#elif defined(HAVE_LONG_32)
222typedef unsigned long uint32;
223#elif defined(HAVE_SHORT_32)
224typedef unsigned short uint32;
225#else
226#error No 32 byte type found !
227#endif
228
229typedef unsigned char uint8;
230
231typedef 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
244typedef unsigned char sh_byte;
245
246
247extern int md5Reset(register md5Param* p);
248extern int md5Update(md5Param* p, const sh_byte* data, int size);
249extern int md5Digest(md5Param* p, uint32* data);
250
251static char db_name[64] = "";
252static char db_table[64] = "";
253static char db_host[64] = "";
254static char db_user[64] = "";
255static char db_password[64] = "";
256
257static int sh_persistent_dbconn = S_TRUE;
258
259int sh_database_use_persistent (const char * str)
260{
261 return sh_util_flagval (str, &sh_persistent_dbconn);
262}
263
264static 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
274static void init_db_entry (dbins * ptr)
275{
276 memset (ptr, (int) '\0', sizeof(dbins));
277 ptr->next = NULL;
278 return;
279}
280
281
282int sh_database_set_database (const char * str)
283{
284 return insert_value (db_name, str);
285}
286int sh_database_set_table (const char * str)
287{
288 return insert_value (db_table, str);
289}
290int sh_database_set_host (const char * str)
291{
292 return insert_value (db_host, str);
293}
294int sh_database_set_user (const char * str)
295{
296 return insert_value (db_user, str);
297}
298int 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
335static SQLHENV u_handle;
336static SQLHDBC u_connection;
337static SQLHSTMT u_statement;
338static SQLINTEGER u_col;
339static SQLINTEGER u_rows;
340
341void sh_database_reset()
342{
343 return;
344}
345
346static
347int 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
498static OCIDefine * o_define;
499static OCIEnv * o_environment;
500static OCISvcCtx * o_servicecontext;
501static OCIError * o_error = NULL;
502static OCIStmt * o_statement;
503static OCIBind * o_bind = (OCIBind *) 0;
504static text o_errormsg[512];
505static sb4 o_errorcode;
506
507static int connected = 0;
508
509void 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
523static 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
534static
535int 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
878static int connection_status = S_FALSE;
879
880void sh_database_reset()
881{
882 connection_status = S_FALSE;
883 return;
884}
885
886static
887int 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
1032extern int flag_err_debug;
1033
1034static int connection_status = S_FALSE;
1035
1036void sh_database_reset()
1037{
1038 connection_status = S_FALSE;
1039 return;
1040}
1041
1042static
1043int 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
1194static
1195char * 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
1225static
1226long 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
1445static 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
1453static 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
1506int 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
1527static 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 */
1571static
1572char * 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
1736static int enter_wrapper = 1;
1737
1738int 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 */
1745long 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
1782int 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
Note: See TracBrowser for help on using the repository browser.