diff options
Diffstat (limited to 'source3/registry')
-rw-r--r-- | source3/registry/reg_eventlog.c | 192 | ||||
-rw-r--r-- | source3/registry/reg_util.c | 16 |
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 ) } - - - - |