Changeset 20 for trunk/src/sh_prelude.c

Timestamp:

Feb 13, 2006, 11:54:42 PM (19 years ago)

Author:

rainer

Message:

Enable command-line parsing for prelude, and make prelude regression test safer.

File:

• trunk/src/sh_prelude.c (modified) (4 diffs)

trunk/src/sh_prelude.c

@@ -55 +55 @@
 #include <pwd.h>
 #include <grp.h>
+
+int     sh_argc_store;
+char ** sh_argv_store;
 
 #if defined(HAVE_LIBPRELUDE) && defined(HAVE_LIBPRELUDE_9)
@@ -933 +936 @@
         idmef_analyzer_t *analyzer;
         prelude_client_flags_t flags;
+#ifdef SH_NOFAILOVER
+        prelude_connection_pool_t *pool;
+        prelude_connection_pool_flags_t conn_flags;
+#endif
 
         if (ready_for_init == 0)
@@ -941 +948 @@
 
         prelude_thread_init(NULL);
-        prelude_init(NULL, NULL);
+        prelude_init(&sh_argc_store, sh_argv_store);
 
         ret = prelude_client_new(&client, profile ? profile : _("samhain"));
@@ -968 +975 @@
         prelude_string_set_dup(str, IDMEF_ANALYZER_VERSION);
         
+#ifdef SH_NOFAILOVER
+        pool = prelude_client_get_connection_pool(client);
+        conn_flags = prelude_connection_pool_get_flags(pool);
+
+        conn_flags &= ~PRELUDE_CONNECTION_POOL_FLAGS_FAILOVER;
+        prelude_connection_pool_set_flags(pool, conn_flags);
+#endif
+
         ret = prelude_client_start(client);
         if ( ret < 0 ) {