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