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