/* * Samba Unix/Linux SMB client utility * Write Eventlog records to a tdb, perform other eventlog related functions * * * Copyright (C) Brian Moran 2005. * Copyright (C) Guenther Deschner 2009. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, see <http://www.gnu.org/licenses/>. */ #include "includes.h" #undef DBGC_CLASS #define DBGC_CLASS DBGC_UTIL_EVENTLOG extern int optind; extern char *optarg; int opt_debug = 0; static void usage( char *s ) { printf( "\nUsage: %s [OPTION]\n\n", s ); printf( " -o write <Eventlog Name> \t\t\t\t\tWrites records to eventlog from STDIN\n" ); printf( " -o addsource <EventlogName> <sourcename> <msgfileDLLname> \tAdds the specified source & DLL eventlog registry entry\n" ); printf( " -o dump <Eventlog Name> <starting_record>\t\t\t\t\tDump stored eventlog entries on STDOUT\n" ); printf( "\nMiscellaneous options:\n" ); printf( " -d\t\t\t\t\t\t\t\tturn debug on\n" ); printf( " -h\t\t\t\t\t\t\t\tdisplay help\n\n" ); } static void display_eventlog_names( void ) { const char **elogs; int i; elogs = lp_eventlog_list( ); printf( "Active eventlog names (from smb.conf):\n" ); printf( "--------------------------------------\n" ); if ( elogs ) { for ( i = 0; elogs[i]; i++ ) { printf( "\t%s\n", elogs[i] ); } } else printf( "\t<None specified>\n"); } static int DoAddSourceCommand( int argc, char **argv, bool debugflag, char *exename ) { if ( argc < 3 ) { printf( "need more arguments:\n" ); printf( "-o addsource EventlogName SourceName /path/to/EventMessageFile.dll\n" ); return -1; } /* must open the registry before we access it */ if (!W_ERROR_IS_OK(regdb_init())) { printf( "Can't open the registry.\n" ); return -1; } if ( !eventlog_add_source( argv[0], argv[1], argv[2] ) ) return -2; return 0; } static int DoWriteCommand( int argc, char **argv, bool debugflag, char *exename ) { FILE *f1; char *argfname; ELOG_TDB *etdb; NTSTATUS status; /* fixed constants are bad bad bad */ char linein[1024]; bool is_eor; struct eventlog_Record_tdb ee; uint32_t record_number = 0; TALLOC_CTX *mem_ctx = talloc_tos(); f1 = stdin; if ( !f1 ) { printf( "Can't open STDIN\n" ); return -1; } if ( debugflag ) { printf( "Starting write for eventlog [%s]\n", argv[0] ); display_eventlog_names( ); } argfname = argv[0]; if ( !( etdb = elog_open_tdb( argfname, False, False ) ) ) { printf( "can't open the eventlog TDB (%s)\n", argfname ); return -1; } ZERO_STRUCT( ee ); /* MUST initialize between records */ while ( !feof( f1 ) ) { if (fgets( linein, sizeof( linein ) - 1, f1 ) == NULL) { break; } linein[strlen( linein ) - 1] = 0; /* whack the line delimiter */ if ( debugflag ) printf( "Read line [%s]\n", linein ); is_eor = False; parse_logentry( mem_ctx, ( char * ) &linein, &ee, &is_eor ); /* should we do something with the return code? */ if ( is_eor ) { fixup_eventlog_record_tdb( &ee ); if ( opt_debug ) printf( "record number [%d], tg [%d] , tw [%d]\n", ee.record_number, (int)ee.time_generated, (int)ee.time_written ); if ( ee.time_generated != 0 ) { /* printf("Writing to the event log\n"); */ status = evlog_push_record_tdb( mem_ctx, ELOG_TDB_CTX(etdb), &ee, &record_number ); if ( !NT_STATUS_IS_OK(status) ) { printf( "Can't write to the event log: %s\n", nt_errstr(status) ); } else { if ( opt_debug ) printf( "Wrote record %d\n", record_number ); } } else { if ( opt_debug ) printf( "<null record>\n" ); } ZERO_STRUCT( ee ); /* MUST initialize between records */ } } elog_close_tdb( etdb , False ); return 0; } static int DoDumpCommand(int argc, char **argv, bool debugflag, char *exename) { ELOG_TDB *etdb; TALLOC_CTX *mem_ctx = talloc_tos(); const char *tdb_filename; uint32_t count = 1; if (argc > 2) { return -1; } tdb_filename = argv[0]; if (argc > 1) { count = atoi(argv[1]); } etdb = elog_open_tdb(argv[0], false, true); if (!etdb) { printf("can't open the eventlog TDB (%s)\n", argv[0]); return -1; } while (1) { struct eventlog_Record_tdb *r; char *s; r = evlog_pull_record_tdb(mem_ctx, etdb->tdb, count); if (!r) { break; } printf("displaying record: %d\n", count); s = NDR_PRINT_STRUCT_STRING(mem_ctx, eventlog_Record_tdb, r); if (s) { printf("%s\n", s); talloc_free(s); } count++; } elog_close_tdb(etdb, false); return 0; } /* would be nice to use the popT stuff here, however doing so forces us to drag in a lot of other infrastructure */ int main( int argc, char *argv[] ) { int opt, rc; char *exename; TALLOC_CTX *frame = talloc_stackframe(); fstring opname; load_case_tables(); opt_debug = 0; /* todo set this from getopts */ lp_load(get_dyn_CONFIGFILE(), True, False, False, True); exename = argv[0]; /* default */ fstrcpy( opname, "write" ); /* the default */ #if 0 /* TESTING CODE */ eventlog_add_source( "System", "TestSourceX", "SomeTestPathX" ); #endif while ( ( opt = getopt( argc, argv, "dho:" ) ) != EOF ) { switch ( opt ) { case 'o': fstrcpy( opname, optarg ); break; case 'h': usage( exename ); display_eventlog_names( ); exit( 0 ); break; case 'd': opt_debug = 1; break; } } argc -= optind; argv += optind; if ( argc < 1 ) { printf( "\nNot enough arguments!\n" ); usage( exename ); exit( 1 ); } /* note that the separate command types should call usage if they need to... */ while ( 1 ) { if ( !StrCaseCmp( opname, "addsource" ) ) { rc = DoAddSourceCommand( argc, argv, opt_debug, exename ); break; } if ( !StrCaseCmp( opname, "write" ) ) { rc = DoWriteCommand( argc, argv, opt_debug, exename ); break; } if ( !StrCaseCmp( opname, "dump" ) ) { rc = DoDumpCommand( argc, argv, opt_debug, exename ); break; } printf( "unknown command [%s]\n", opname ); usage( exename ); exit( 1 ); break; } TALLOC_FREE(frame); return rc; }