summaryrefslogtreecommitdiff
path: root/source3/registry
diff options
context:
space:
mode:
Diffstat (limited to 'source3/registry')
-rw-r--r--source3/registry/reg_eventlog.c192
-rw-r--r--source3/registry/reg_util.c16
2 files changed, 198 insertions, 10 deletions
diff --git a/source3/registry/reg_eventlog.c b/source3/registry/reg_eventlog.c
index 204c4c6e1c..bf843e5518 100644
--- a/source3/registry/reg_eventlog.c
+++ b/source3/registry/reg_eventlog.c
@@ -1,3 +1,4 @@
+
/*
* Unix SMB/CIFS implementation.
* Virtual Windows Registry Layer
@@ -84,6 +85,7 @@ BOOL eventlog_init_keys( void )
regdb_fetch_values( evtlogpath, values );
if ( !regval_ctr_key_exists( values, "MaxSize" ) ) {
+
/* assume we have none, add them all */
/* hard code some initial values */
@@ -149,5 +151,195 @@ BOOL eventlog_init_keys( void )
TALLOC_FREE( values );
elogs++;
}
+
+ return True;
+}
+
+/*********************************************************************
+ for an eventlog, add in a source name. If the eventlog doesn't
+ exist (not in the list) do nothing. If a source for the log
+ already exists, change the information (remove, replace)
+*********************************************************************/
+
+BOOL eventlog_add_source( const char *eventlog, const char *sourcename,
+ const char *messagefile )
+{
+ /* Find all of the eventlogs, add keys for each of them */
+ /* need to add to the value KEY_EVENTLOG/<eventlog>/Sources string (Creating if necessary)
+ need to add KEY of source to KEY_EVENTLOG/<eventlog>/<source> */
+
+ const char **elogs = lp_eventlog_list( );
+ char **wrklist, **wp;
+ pstring evtlogpath;
+ REGSUBKEY_CTR *subkeys;
+ REGVAL_CTR *values;
+ REGISTRY_VALUE *rval;
+ UNISTR2 data;
+ uint16 *msz_wp;
+ int mbytes, ii;
+ BOOL already_in;
+ int i;
+ int numsources;
+
+ for ( i=0; elogs[i]; i++ ) {
+ if ( strequal( elogs[i], eventlog ) )
+ break;
+ }
+
+ if ( !elogs[i] ) {
+ DEBUG( 0,
+ ( "Eventlog [%s] not found in list of valid event logs\n",
+ eventlog ) );
+ return False; /* invalid named passed in */
+ }
+
+ /* have to assume that the evenlog key itself exists at this point */
+ /* add in a key of [sourcename] under the eventlog key */
+
+ /* todo add to Sources */
+
+ if ( !( values = TALLOC_ZERO_P( NULL, REGVAL_CTR ) ) ) {
+ DEBUG( 0, ( "talloc() failure!\n" ) );
+ return False;
+ }
+
+ pstr_sprintf( evtlogpath, "%s\\%s", KEY_EVENTLOG, eventlog );
+
+ regdb_fetch_values( evtlogpath, values );
+
+ if ( !( rval = regval_ctr_getvalue( values, "Sources" ) ) ) {
+ DEBUG( 0, ( "No Sources value for [%s]!\n", eventlog ) );
+ return False;
+ }
+
+ /* perhaps this adding a new string to a multi_sz should be a fn? */
+ /* check to see if it's there already */
+
+ if ( rval->type != REG_MULTI_SZ ) {
+ DEBUG( 0,
+ ( "Wrong type for Sources, should be REG_MULTI_SZ\n" ) );
+ return False;
+ }
+
+ /* convert to a 'regulah' chars to do some comparisons */
+
+ DEBUG( 0, ( "Rval size is %d\n", rval->size ) );
+
+ already_in = False;
+ wrklist = NULL;
+ dump_data( 1, rval->data_p, rval->size );
+ if ( ( numsources =
+ regval_convert_multi_sz( ( uint16 * ) rval->data_p, rval->size,
+ &wrklist ) ) > 0 ) {
+
+ DEBUG( 10, ( "numsources is %d\n", numsources ) );
+ ii = numsources;
+ /* see if it's in there already */
+ wp = wrklist;
+ while ( ii && wp && *wp ) {
+ DEBUG( 5,
+ ( "Comparing [%s] to [%s]\n", sourcename,
+ *wp ) );
+ if ( strequal( *wp, sourcename ) ) {
+ DEBUG( 5,
+ ( "Source name %s already exists, \n",
+ sourcename ) );
+ already_in = True;
+ break;
+ }
+ wp++;
+ ii--;
+ }
+ } else {
+ if ( numsources < 0 ) {
+ DEBUG( 3, ( "problem in getting the sources\n" ) );
+ return False;
+ }
+ DEBUG( 3,
+ ( "Nothing in the sources list, this might be a problem\n" ) );
+ }
+
+ wp = wrklist;
+
+ if ( !already_in ) {
+ /* make a new list with an additional entry; copy values, add another */
+ wp = TALLOC_ARRAY( NULL, char *, numsources + 2 );
+
+ if ( !wp ) {
+ DEBUG( 0, ( "talloc() failed \n" ) );
+ return False;
+ }
+ DEBUG( 0, ( "Number of sources [%d]\n", numsources ) );
+ memcpy( wp, wrklist, sizeof( char * ) * numsources );
+ *( wp + numsources ) = ( char * ) sourcename;
+ *( wp + numsources + 1 ) = NULL;
+ mbytes = regval_build_multi_sz( wp, &msz_wp );
+ DEBUG( 0, ( "Number of mbytes [%d]\n", mbytes ) );
+ dump_data( 1, (char*)msz_wp, mbytes );
+ regval_ctr_addvalue( values, "Sources", REG_MULTI_SZ,
+ ( char * ) msz_wp, mbytes );
+ regdb_store_values( evtlogpath, values );
+ TALLOC_FREE( msz_wp );
+ } else {
+ DEBUG( 0,
+ ( "Source name [%s] found in existing list of sources\n",
+ sourcename ) );
+ }
+ TALLOC_FREE( values );
+
+ DEBUG( 5,
+ ( "Added source to sources string, now adding subkeys\n" ) );
+
+ if ( !( subkeys = TALLOC_ZERO_P( NULL, REGSUBKEY_CTR ) ) ) {
+ DEBUG( 0, ( "talloc() failure!\n" ) );
+ return False;
+ }
+ pstr_sprintf( evtlogpath, "%s\\%s", KEY_EVENTLOG, eventlog );
+
+ regdb_fetch_keys( evtlogpath, subkeys );
+
+ if ( !regsubkey_ctr_key_exists( subkeys, sourcename ) ) {
+ DEBUG( 5,
+ ( " Source name [%s] for eventlog [%s] didn't exist, adding \n",
+ sourcename, eventlog ) );
+ regsubkey_ctr_addkey( subkeys, sourcename );
+ if ( !regdb_store_keys( evtlogpath, subkeys ) )
+ return False;
+ }
+ TALLOC_FREE( subkeys );
+
+ /* at this point KEY_EVENTLOG/<eventlog>/<sourcename> key is in there. Now need to add EventLogMessageFile */
+
+ /* now allocate room for the source's subkeys */
+
+ if ( !( subkeys = TALLOC_ZERO_P( NULL, REGSUBKEY_CTR ) ) ) {
+ DEBUG( 0, ( "talloc() failure!\n" ) );
+ return False;
+ }
+ slprintf( evtlogpath, sizeof( evtlogpath ) - 1, "%s\\%s\\%s",
+ KEY_EVENTLOG, eventlog, sourcename );
+
+ regdb_fetch_keys( evtlogpath, subkeys );
+
+ /* now add the values to the KEY_EVENTLOG/Application form key */
+ if ( !( values = TALLOC_ZERO_P( NULL, REGVAL_CTR ) ) ) {
+ DEBUG( 0, ( "talloc() failure!\n" ) );
+ return False;
+ }
+ DEBUG( 5,
+ ( "Storing EventLogMessageFile [%s] to eventlog path of [%s]\n",
+ messagefile, evtlogpath ) );
+
+ regdb_fetch_values( evtlogpath, values );
+
+ init_unistr2( &data, messagefile, UNI_STR_TERMINATE );
+
+ regval_ctr_addvalue( values, "EventLogMessageFile", REG_EXPAND_SZ,
+ ( char * ) data.buffer,
+ data.uni_str_len * sizeof( uint16 ) );
+ regdb_store_values( evtlogpath, values );
+
+ TALLOC_FREE( values );
+
return True;
}
diff --git a/source3/registry/reg_util.c b/source3/registry/reg_util.c
index 511e9b4335..9fd9dba27d 100644
--- a/source3/registry/reg_util.c
+++ b/source3/registry/reg_util.c
@@ -138,12 +138,12 @@ int regval_convert_multi_sz( uint16 *multi_string, size_t multi_len, char ***val
*values = NULL;
- if ( !multi_string || !*values )
+ if ( !multi_string || !values )
return 0;
/* just count the NULLs */
- for ( i=0; (i<multi_len-1) && !(multi_string[i]==0x0 && multi_string[i+1]==0x0); i++ ) {
+ for ( i=0; (i<multi_len-1) && !(multi_string[i]==0x0 && multi_string[i+1]==0x0); i+=2 ) {
if ( multi_string[i] == 0x0 )
num_strings++;
}
@@ -187,7 +187,7 @@ size_t regval_build_multi_sz( char **values, uint16 **buffer )
uint16 *buf, *b;
UNISTR2 sz;
- if ( !values || !*buffer )
+ if ( !values || !buffer )
return 0;
/* go ahead and alloc some space */
@@ -199,7 +199,7 @@ size_t regval_build_multi_sz( char **values, uint16 **buffer )
for ( i=0; values[i]; i++ ) {
ZERO_STRUCT( sz );
-
+ /* DEBUG(0,("regval_build_multi_sz: building [%s]\n",values[i])); */
init_unistr2( &sz, values[i], UNI_STR_TERMINATE );
/* Alloc some more memory. Always add one one to account for the
@@ -214,8 +214,8 @@ size_t regval_build_multi_sz( char **values, uint16 **buffer )
buf = b;
/* copy the unistring2 buffer and increment the size */
-
- memcpy( buf+buf_size, sz.buffer, sz.uni_str_len );
+ /* dump_data(1,sz.buffer,sz.uni_str_len*2); */
+ memcpy( buf+buf_size, sz.buffer, sz.uni_str_len*2 );
buf_size += sz.uni_str_len;
/* cleanup rather than leaving memory hanging around */
@@ -231,7 +231,3 @@ size_t regval_build_multi_sz( char **values, uint16 **buffer )
}
-
-
-
-