source/utils/eventlogadm.c

   1
   2 /*
   3  * Samba Unix/Linux SMB client utility
   4  * Write Eventlog records to a tdb, perform other eventlog related functions
   5  *
   6  *
   7  * Copyright (C) Brian Moran                2005.
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23
  24 #include "includes.h"
  25
  26 #undef  DBGC_CLASS
  27 #define DBGC_CLASS DBGC_UTIL_EVENTLOG
  28
  29
  30 extern int optind;
  31 extern char *optarg;
  32
  33 int opt_debug = 0;
  34
  35 static void usage( char *s )
  36 {
  37         printf( "\nUsage: %s [OPTION]\n\n", s );
  38         printf( " -o write <Eventlog Name> \t\t\t\t\tWrites records to eventlog from STDIN\n" );
  39         printf( " -o addsource <EventlogName> <sourcename> <msgfileDLLname> \tAdds the specified source & DLL eventlog registry entry\n" );
  40         printf( "\nMiscellaneous options:\n" );
  41         printf( " -d\t\t\t\t\t\t\t\tturn debug on\n" );
  42         printf( " -h\t\t\t\t\t\t\t\tdisplay help\n\n" );
  43 }
  44
  45 static void display_eventlog_names( void )
  46 {
  47         const char **elogs;
  48         int i;
  49
  50         elogs = lp_eventlog_list(  );
  51         printf( "Active eventlog names (from smb.conf):\n" );
  52         printf( "--------------------------------------\n" );
  53         if ( elogs ) {
  54                 for ( i = 0; elogs[i]; i++ ) {
  55                         printf( "\t%s\n", elogs[i] );
  56                 }
  57         }
  58         else
  59                 printf( "\t<None specified>\n");
  60 }
  61
  62 static int DoAddSourceCommand( int argc, char **argv, bool debugflag, char *exename )
  63 {
  64
  65         if ( argc < 3 ) {
  66                 printf( "need more arguments:\n" );
  67                 printf( "-o addsource EventlogName SourceName /path/to/EventMessageFile.dll\n" );
  68                 return -1;
  69         }
  70         /* must open the registry before we access it */
  71         if (!W_ERROR_IS_OK(regdb_init())) {
  72                 printf( "Can't open the registry.\n" );
  73                 return -1;
  74         }
  75
  76         if ( !eventlog_add_source( argv[0], argv[1], argv[2] ) )
  77                 return -2;
  78         return 0;
  79 }
  80
  81 static int DoWriteCommand( int argc, char **argv, bool debugflag, char *exename )
  82 {
  83         FILE *f1;
  84         char *argfname;
  85         ELOG_TDB *etdb;
  86
  87         /* fixed constants are bad bad bad  */
  88         char linein[1024];
  89         bool is_eor;
  90         Eventlog_entry ee;
  91         int rcnum;
  92
  93         f1 = stdin;
  94         if ( !f1 ) {
  95                 printf( "Can't open STDIN\n" );
  96                 return -1;
  97         }
  98
  99         if ( debugflag ) {
 100                 printf( "Starting write for eventlog [%s]\n", argv[0] );
 101                 display_eventlog_names(  );
 102         }
 103
 104         argfname = argv[0];
 105
 106         if ( !( etdb = elog_open_tdb( argfname, False ) ) ) {
 107                 printf( "can't open the eventlog TDB (%s)\n", argfname );
 108                 return -1;
 109         }
 110
 111         ZERO_STRUCT( ee );      /* MUST initialize between records */
 112
 113         while ( !feof( f1 ) ) {
 114                 if (fgets( linein, sizeof( linein ) - 1, f1 ) == NULL) {
 115                         break;
 116                 }
 117                 linein[strlen( linein ) - 1] = 0;       /* whack the line delimiter */
 118
 119                 if ( debugflag )
 120                         printf( "Read line [%s]\n", linein );
 121
 122                 is_eor = False;
 123
 124
 125                 parse_logentry( ( char * ) &linein, &ee, &is_eor );
 126                 /* should we do something with the return code? */
 127
 128                 if ( is_eor ) {
 129                         fixup_eventlog_entry( &ee );
 130
 131                         if ( opt_debug )
 132                                 printf( "record number [%d], tg [%d] , tw [%d]\n", ee.record.record_number, ee.record.time_generated, ee.record.time_written );
 133
 134                         if ( ee.record.time_generated != 0 ) {
 135
 136                                 /* printf("Writing to the event log\n"); */
 137
 138                                 rcnum = write_eventlog_tdb( ELOG_TDB_CTX(etdb), &ee );
 139                                 if ( !rcnum ) {
 140                                         printf( "Can't write to the event log\n" );
 141                                 } else {
 142                                         if ( opt_debug )
 143                                                 printf( "Wrote record %d\n",
 144                                                         rcnum );
 145                                 }
 146                         } else {
 147                                 if ( opt_debug )
 148                                         printf( "<null record>\n" );
 149                         }
 150                         ZERO_STRUCT( ee );      /* MUST initialize between records */
 151                 }
 152         }
 153
 154         elog_close_tdb( etdb , False );
 155
 156         return 0;
 157 }
 158
 159 /* would be nice to use the popT stuff here, however doing so forces us to drag in a lot of other infrastructure */
 160
 161 int main( int argc, char *argv[] )
 162 {
 163         int opt, rc;
 164         char *exename;
 165         TALLOC_CTX *frame = talloc_stackframe();
 166
 167
 168         fstring opname;
 169
 170         load_case_tables();
 171
 172         opt_debug = 0;          /* todo set this from getopts */
 173
 174         lp_load(get_dyn_CONFIGFILE(), True, False, False, True);
 175
 176         exename = argv[0];
 177
 178         /* default */
 179
 180         fstrcpy( opname, "write" );     /* the default */
 181
 182 #if 0                           /* TESTING CODE */
 183         eventlog_add_source( "System", "TestSourceX", "SomeTestPathX" );
 184 #endif
 185         while ( ( opt = getopt( argc, argv, "dho:" ) ) != EOF ) {
 186                 switch ( opt ) {
 187
 188                 case 'o':
 189                         fstrcpy( opname, optarg );
 190                         break;
 191
 192                 case 'h':
 193                         usage( exename );
 194                         display_eventlog_names(  );
 195                         exit( 0 );
 196                         break;
 197
 198                 case 'd':
 199                         opt_debug = 1;
 200                         break;
 201                 }
 202         }
 203
 204         argc -= optind;
 205         argv += optind;
 206
 207         if ( argc < 1 ) {
 208                 printf( "\nNot enough arguments!\n" );
 209                 usage( exename );
 210                 exit( 1 );
 211         }
 212
 213         /*  note that the separate command types should call usage if they need to... */
 214         while ( 1 ) {
 215                 if ( !StrCaseCmp( opname, "addsource" ) ) {
 216                         rc = DoAddSourceCommand( argc, argv, opt_debug,
 217                                                  exename );
 218                         break;
 219                 }
 220                 if ( !StrCaseCmp( opname, "write" ) ) {
 221                         rc = DoWriteCommand( argc, argv, opt_debug, exename );
 222                         break;
 223                 }
 224                 printf( "unknown command [%s]\n", opname );
 225                 usage( exename );
 226                 exit( 1 );
 227                 break;
 228         }
 229         TALLOC_FREE(frame);
 230         return rc;
 231 }