Convert buserver to libutil's logging
[openafs.git] / src / budb / server.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  *
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 #include <afsconfig.h>
11 #include <afs/param.h>
12 #include <afs/stds.h>
13
14 #include <roken.h>
15
16 #ifdef AFS_NT40_ENV
17 #include <WINNT/afsevent.h>
18 #endif
19 #include <afs/cmd.h>
20 #include <lwp.h>
21 #include <ubik.h>
22 #include <rx/xdr.h>
23 #include <rx/rx.h>
24 #include <rx/rxkad.h>
25 #include <rx/rx_globals.h>
26 #include <afs/cellconfig.h>
27 #include <afs/bubasics.h>
28 #include <afs/afsutil.h>
29 #include <afs/com_err.h>
30 #include <afs/audit.h>
31
32 #include "budb_errs.h"
33 #include "database.h"
34 #include "error_macros.h"
35 #include "budb_internal.h"
36 #include "globals.h"
37
38 struct ubik_dbase *BU_dbase;
39 struct afsconf_dir *BU_conf;    /* for getting cell info */
40
41 int argHandler(struct cmd_syndesc *, void *);
42 int truncateDatabase(void);
43 int parseServerList(struct cmd_item *);
44
45 char lcell[MAXKTCREALMLEN];
46 afs_uint32 myHost = 0;
47 int helpOption;
48
49 /* server's global configuration information. This is exported to other
50  * files/routines
51  */
52
53 buServerConfT globalConf;
54 buServerConfP globalConfPtr = &globalConf;
55 char dbDir[AFSDIR_PATH_MAX], cellConfDir[AFSDIR_PATH_MAX];
56 /* debugging control */
57 int debugging = 0;
58
59 int rxBind = 0;
60 int lwps   = 3;
61
62 #define MINLWP  3
63 #define MAXLWP 16
64
65 #define ADDRSPERSITE 16         /* Same global is in rx/rx_user.c */
66 afs_uint32 SHostAddrs[ADDRSPERSITE];
67
68 /* check whether caller is authorized to manage RX statistics */
69 int
70 BU_rxstat_userok(struct rx_call *call)
71 {
72     return afsconf_SuperUser(BU_conf, call, NULL);
73 }
74
75 /**
76  * Return true if this name is a member of the local realm.
77  */
78 int
79 BU_IsLocalRealmMatch(void *rock, char *name, char *inst, char *cell)
80 {
81     struct afsconf_dir *dir = (struct afsconf_dir *)rock;
82     afs_int32 islocal = 0;      /* default to no */
83     int code;
84
85     code = afsconf_IsLocalRealmMatch(dir, &islocal, name, inst, cell);
86     if (code) {
87         LogError(code, "Failed local realm check; name=%s, inst=%s, cell=%s\n",
88                  name, inst, cell);
89     }
90     return islocal;
91 }
92
93 int
94 convert_cell_to_ubik(struct afsconf_cell *cellinfo, afs_uint32 *myHost,
95                      afs_uint32 *serverList)
96 {
97     int i;
98     char hostname[64];
99     struct hostent *th;
100
101     /* get this host */
102     gethostname(hostname, sizeof(hostname));
103     th = gethostbyname(hostname);
104     if (!th) {
105         printf("prserver: couldn't get address of this host.\n");
106         BUDB_EXIT(1);
107     }
108     memcpy(myHost, th->h_addr, sizeof(afs_uint32));
109
110     for (i = 0; i < cellinfo->numServers; i++)
111         /* omit my host from serverList */
112         if (cellinfo->hostAddr[i].sin_addr.s_addr != *myHost)
113             *serverList++ = cellinfo->hostAddr[i].sin_addr.s_addr;
114
115     *serverList = 0;            /* terminate list */
116     return 0;
117 }
118
119 /* MyBeforeProc
120  *      The whole purpose of MyBeforeProc is to detect
121  *      if the -help option was not within the command line.
122  *      If it were, this routine would never have been called.
123  */
124 static int
125 MyBeforeProc(struct cmd_syndesc *as, void *arock)
126 {
127     helpOption = 0;
128     return 0;
129 }
130
131 /* initializeCommands
132  *      initialize all the supported commands and their arguments
133  */
134
135 void
136 initializeArgHandler(void)
137 {
138     struct cmd_syndesc *cptr;
139
140     cmd_SetBeforeProc(MyBeforeProc, NULL);
141
142     cptr = cmd_CreateSyntax(NULL, argHandler, NULL, "Backup database server");
143
144     cmd_AddParm(cptr, "-database", CMD_SINGLE, CMD_OPTIONAL,
145                 "database directory");
146
147     cmd_AddParm(cptr, "-cellservdb", CMD_SINGLE, CMD_OPTIONAL,
148                 "cell configuration directory");
149
150     cmd_AddParm(cptr, "-resetdb", CMD_FLAG, CMD_OPTIONAL,
151                 "truncate the database");
152
153     cmd_AddParm(cptr, "-noauth", CMD_FLAG, CMD_OPTIONAL,
154                 "run without authentication");
155
156     cmd_AddParm(cptr, "-smallht", CMD_FLAG, CMD_OPTIONAL,
157                 "use small hash tables");
158
159     cmd_AddParm(cptr, "-servers", CMD_LIST, CMD_OPTIONAL,
160                 "list of ubik database servers");
161
162     cmd_AddParm(cptr, "-ubikbuffers", CMD_SINGLE, CMD_OPTIONAL,
163                 "the number of ubik buffers");
164
165     cmd_AddParm(cptr, "-auditlog", CMD_SINGLE, CMD_OPTIONAL,
166                 "audit log path");
167
168     cmd_AddParm(cptr, "-p", CMD_SINGLE, CMD_OPTIONAL,
169                 "number of processes");
170
171     cmd_AddParm(cptr, "-rxbind", CMD_FLAG, CMD_OPTIONAL,
172                 "bind the Rx socket (primary interface only)");
173
174     cmd_AddParm(cptr, "-audit-interface", CMD_SINGLE, CMD_OPTIONAL,
175                 "audit interface (file or sysvmq)");
176 }
177
178 int
179 argHandler(struct cmd_syndesc *as, void *arock)
180 {
181
182     /* globalConfPtr provides the handle for the configuration information */
183
184     /* database directory */
185     if (as->parms[0].items != 0) {
186         globalConfPtr->databaseDirectory = strdup(as->parms[0].items->data);
187         if (globalConfPtr->databaseDirectory == 0)
188             BUDB_EXIT(-1);
189     }
190
191     /* -cellservdb, cell configuration directory */
192     if (as->parms[1].items != 0) {
193         globalConfPtr->cellConfigdir = strdup(as->parms[1].items->data);
194         if (globalConfPtr->cellConfigdir == 0)
195             BUDB_EXIT(-1);
196
197         globalConfPtr->debugFlags |= DF_RECHECKNOAUTH;
198     }
199
200     /* truncate the database */
201     if (as->parms[2].items != 0)
202         truncateDatabase();
203
204     /* run without authentication */
205     if (as->parms[3].items != 0)
206         globalConfPtr->debugFlags |= DF_NOAUTH;
207
208     /* use small hash tables */
209     if (as->parms[4].items != 0)
210         globalConfPtr->debugFlags |= DF_SMALLHT;
211
212     /* user provided list of ubik database servers */
213     if (as->parms[5].items != 0)
214         parseServerList(as->parms[5].items);
215
216     /* user provided the number of ubik buffers    */
217     if (as->parms[6].items != 0)
218         ubik_nBuffers = atoi(as->parms[6].items->data);
219     else
220         ubik_nBuffers = 0;
221
222     /* param 7 (-auditlog) handled below */
223
224     /* user provided the number of threads    */
225     if (as->parms[8].items != 0) {
226         lwps = atoi(as->parms[8].items->data);
227         if (lwps > MAXLWP) {
228             printf ("Warning: '-p %d' is too big; using %d instead\n",
229                 lwps, MAXLWP);
230             lwps = MAXLWP;
231         }
232         if (lwps < MINLWP) {
233             printf ("Warning: '-p %d' is too small; using %d instead\n",
234                 lwps, MINLWP);
235             lwps = MINLWP;
236         }
237     }
238
239     /* user provided rxbind option    */
240     if (as->parms[9].items != 0) {
241         rxBind = 1;
242     }
243
244     /* -audit-interface */
245     if (as->parms[10].items != 0) {
246         char *interface = as->parms[10].items->data;
247
248         if (osi_audit_interface(interface)) {
249             printf("Invalid audit interface '%s'\n", interface);
250             BUDB_EXIT(-1);
251         }
252     }
253
254     /* -auditlog */
255     /* needs to be after -audit-interface, so we osi_audit_interface
256      * before we osi_audit_file */
257     if (as->parms[7].items != 0) {
258         char *fileName = as->parms[7].items->data;
259
260         osi_audit_file(fileName);
261     }
262
263     return 0;
264 }
265
266 /* --- */
267
268 int
269 parseServerList(struct cmd_item *itemPtr)
270 {
271     struct cmd_item *save;
272     char **serverArgs;
273     char **ptr;
274     afs_int32 nservers = 0;
275     afs_int32 code = 0;
276
277     save = itemPtr;
278
279     /* compute number of servers in the list */
280     while (itemPtr) {
281         nservers++;
282         itemPtr = itemPtr->next;
283     }
284
285     LogDebug(3, "%d servers\n", nservers);
286
287     /* now can allocate the space for the server arguments */
288     serverArgs = malloc((nservers + 2) * sizeof(char *));
289     if (serverArgs == 0)
290         ERROR(-1);
291
292     ptr = serverArgs;
293     *ptr++ = "";
294     *ptr++ = "-servers";
295
296     /* now go through and construct the list of servers */
297     itemPtr = save;
298     while (itemPtr) {
299         *ptr++ = itemPtr->data;
300         itemPtr = itemPtr->next;
301     }
302
303     code =
304         ubik_ParseServerList(nservers + 2, serverArgs, &globalConfPtr->myHost,
305                              globalConfPtr->serverList);
306     if (code)
307         ERROR(code);
308
309     /* free space for the server args */
310     free(serverArgs);
311
312   error_exit:
313     return (code);
314 }
315
316 /* truncateDatabase
317  *      truncates just the database file.
318  */
319
320 int
321 truncateDatabase(void)
322 {
323     char *path;
324     afs_int32 code = 0;
325     int fd;
326
327     asprintf(&path, "%s%s%s",
328              globalConfPtr->databaseDirectory,
329              globalConfPtr->databaseName,
330              globalConfPtr->databaseExtension);
331     if (path == NULL)
332         ERROR(-1);
333
334     fd = open(path, O_RDWR, 0755);
335     if (!fd) {
336         code = errno;
337     } else {
338         if (ftruncate(fd, 0) != 0) {
339             code = errno;
340         } else
341             close(fd);
342     }
343
344     free(path);
345
346   error_exit:
347     return (code);
348 }
349
350
351 /* --- */
352
353 #include "AFS_component_version_number.c"
354
355 int
356 main(int argc, char **argv)
357 {
358     char *whoami = argv[0];
359     char *dbNamePtr = 0;
360     struct afsconf_cell cellinfo_s;
361     struct afsconf_cell *cellinfo = NULL;
362     time_t currentTime;
363     afs_int32 code = 0;
364     afs_uint32 host = ntohl(INADDR_ANY);
365
366     char  clones[MAXHOSTSPERCELL];
367
368     struct rx_service *tservice;
369     struct rx_securityClass **securityClasses;
370     afs_int32 numClasses;
371
372     extern int rx_stackSize;
373
374 #ifdef AFS_NT40_ENV
375     /* initialize winsock */
376     if (afs_winsockInit() < 0) {
377         ReportErrorEventAlt(AFSEVT_SVR_WINSOCK_INIT_FAILED, 0, argv[0], 0);
378         fprintf(stderr, "%s: Couldn't initialize winsock.\n", whoami);
379         exit(1);
380     }
381 #endif
382
383 #ifdef  AFS_AIX32_ENV
384     /*
385      * The following signal action for AIX is necessary so that in case of a
386      * crash (i.e. core is generated) we can include the user's data section
387      * in the core dump. Unfortunately, by default, only a partial core is
388      * generated which, in many cases, isn't too useful.
389      */
390     struct sigaction nsa;
391
392     sigemptyset(&nsa.sa_mask);
393     nsa.sa_handler = SIG_DFL;
394     nsa.sa_flags = SA_FULLDUMP;
395     sigaction(SIGSEGV, &nsa, NULL);
396     sigaction(SIGABRT, &nsa, NULL);
397 #endif
398
399     memset(&cellinfo_s, 0, sizeof(cellinfo_s));
400     memset(clones, 0, sizeof(clones));
401
402     osi_audit_init();
403     osi_audit(BUDB_StartEvent, 0, AUD_END);
404
405     initialize_BUDB_error_table();
406     initializeArgHandler();
407
408     /* Initialize dirpaths */
409     if (!(initAFSDirPath() & AFSDIR_SERVER_PATHS_OK)) {
410 #ifdef AFS_NT40_ENV
411         ReportErrorEventAlt(AFSEVT_SVR_NO_INSTALL_DIR, 0, argv[0], 0);
412 #endif
413         afs_com_err(whoami, errno, "; Unable to obtain AFS server directory.");
414         exit(2);
415     }
416
417     memset(globalConfPtr, 0, sizeof(*globalConfPtr));
418
419     /* set default configuration values */
420     strcpy(dbDir, AFSDIR_SERVER_DB_DIRPATH);
421     strcat(dbDir, "/");
422     globalConfPtr->databaseDirectory = dbDir;
423     globalConfPtr->databaseName = DEFAULT_DBPREFIX;
424     strcpy(cellConfDir, AFSDIR_SERVER_ETC_DIRPATH);
425     globalConfPtr->cellConfigdir = cellConfDir;
426
427     srandom(1);
428
429 #ifdef AFS_PTHREAD_ENV
430     SetLogThreadNumProgram( rx_GetThreadNum );
431 #endif
432
433     /* process the user supplied args */
434     helpOption = 1;
435     code = cmd_Dispatch(argc, argv);
436     if (code)
437         ERROR(code);
438
439     /* exit if there was a help option */
440     if (helpOption)
441         BUDB_EXIT(0);
442
443     /* open the log file */
444     OpenLog(AFSDIR_SERVER_BUDBLOG_FILEPATH);
445
446     /* open the cell's configuration directory */
447     LogDebug(4, "opening %s\n", globalConfPtr->cellConfigdir);
448
449     BU_conf = afsconf_Open(globalConfPtr->cellConfigdir);
450     if (BU_conf == 0) {
451         LogError(code, "Failed getting cell info\n");
452         afs_com_err(whoami, code, "Failed getting cell info");
453         ERROR(BUDB_NOCELLS);
454     }
455
456     code = afsconf_GetLocalCell(BU_conf, lcell, sizeof(lcell));
457     if (code) {
458         LogError(0, "** Can't determine local cell name!\n");
459         ERROR(code);
460     }
461
462     if (globalConfPtr->myHost == 0) {
463         /* if user hasn't supplied a list of servers, extract server
464          * list from the cell's database
465          */
466
467         cellinfo = &cellinfo_s;
468
469         LogDebug(1, "Using server list from %s cell database.\n", lcell);
470
471         code = afsconf_GetExtendedCellInfo (BU_conf, lcell, 0, cellinfo,
472                                             clones);
473         if (code) {
474             LogError(0, "Can't read cell information\n");
475             ERROR(code);
476         }
477
478         code =
479             convert_cell_to_ubik(cellinfo, &globalConfPtr->myHost,
480                                  globalConfPtr->serverList);
481         if (code)
482             ERROR(code);
483     }
484
485     /* initialize audit user check */
486     osi_audit_set_user_check(BU_conf, BU_IsLocalRealmMatch);
487
488     /* initialize ubik */
489     ubik_SetClientSecurityProcs(afsconf_ClientAuth, afsconf_UpToDate, BU_conf);
490     ubik_SetServerSecurityProcs(afsconf_BuildServerSecurityObjects,
491                                 afsconf_CheckAuth, BU_conf);
492
493     if (ubik_nBuffers == 0)
494         ubik_nBuffers = 400;
495
496     LogError(0, "Will allocate %d ubik buffers\n", ubik_nBuffers);
497
498     asprintf(&dbNamePtr, "%s%s", globalConfPtr->databaseDirectory,
499              globalConfPtr->databaseName);
500     if (dbNamePtr == 0)
501         ERROR(-1);
502
503     rx_SetRxDeadTime(60);       /* 60 seconds inactive before timeout */
504
505     if (rxBind) {
506         afs_int32 ccode;
507         if (AFSDIR_SERVER_NETRESTRICT_FILEPATH ||
508             AFSDIR_SERVER_NETINFO_FILEPATH) {
509             char reason[1024];
510             ccode = afsconf_ParseNetFiles(SHostAddrs, NULL, NULL,
511                                           ADDRSPERSITE, reason,
512                                           AFSDIR_SERVER_NETINFO_FILEPATH,
513                                           AFSDIR_SERVER_NETRESTRICT_FILEPATH);
514         } else
515         {
516             ccode = rx_getAllAddr(SHostAddrs, ADDRSPERSITE);
517         }
518         if (ccode == 1) {
519             host = SHostAddrs[0];
520             rx_InitHost(host, htons(AFSCONF_BUDBPORT));
521         }
522     }
523
524     /* Disable jumbograms */
525     rx_SetNoJumbo();
526
527     if (cellinfo) {
528         code = ubik_ServerInitByInfo(globalConfPtr->myHost,
529                                      htons(AFSCONF_BUDBPORT),
530                                      cellinfo,
531                                      clones,
532                                      dbNamePtr,           /* name prefix */
533                                      &BU_dbase);
534     } else {
535         code = ubik_ServerInit(globalConfPtr->myHost, htons(AFSCONF_BUDBPORT),
536                                globalConfPtr->serverList,
537                                dbNamePtr,  /* name prefix */
538                                &BU_dbase);
539     }
540
541     if (code) {
542         LogError(code, "Ubik init failed\n");
543         afs_com_err(whoami, code, "Ubik init failed");
544         ERROR(code);
545     }
546
547     afsconf_BuildServerSecurityObjects(BU_conf, &securityClasses, &numClasses);
548
549     tservice =
550         rx_NewServiceHost(host, 0, BUDB_SERVICE, "BackupDatabase",
551                           securityClasses, numClasses, BUDB_ExecuteRequest);
552
553     if (tservice == (struct rx_service *)0) {
554         LogError(0, "Could not create backup database rx service\n");
555         printf("Could not create backup database rx service\n");
556         BUDB_EXIT(3);
557     }
558     rx_SetMinProcs(tservice, 1);
559     rx_SetMaxProcs(tservice, lwps);
560     rx_SetStackSize(tservice, 10000);
561
562     /* allow super users to manage RX statistics */
563     rx_SetRxStatUserOk(BU_rxstat_userok);
564
565     /* misc. initialization */
566
567     /* database dump synchronization */
568     memset(dumpSyncPtr, 0, sizeof(*dumpSyncPtr));
569     Lock_Init(&dumpSyncPtr->ds_lock);
570
571     rx_StartServer(0);          /* start handling requests */
572
573     code = InitProcs();
574     if (code)
575         ERROR(code);
576
577
578     currentTime = time(0);
579     LogError(0, "Ready to process requests at %s\n", ctime(&currentTime));
580
581     rx_ServerProc(NULL);                /* donate this LWP */
582
583   error_exit:
584     osi_audit(BUDB_FinishEvent, code, AUD_END);
585     return (code);
586 }
587
588 void
589 consistencyCheckDb(void)
590 {
591     /* do consistency checks on structure sizes */
592     if ((sizeof(struct htBlock) > BLOCKSIZE)
593         || (sizeof(struct vfBlock) > BLOCKSIZE)
594         || (sizeof(struct viBlock) > BLOCKSIZE)
595         || (sizeof(struct dBlock) > BLOCKSIZE)
596         || (sizeof(struct tBlock) > BLOCKSIZE)
597         ) {
598         fprintf(stderr, "Block layout error!\n");
599         BUDB_EXIT(99);
600     }
601 }
602
603 void
604 LogDebug(int level, char *fmt, ... )
605 {
606     if (debugging >= level) {
607         va_list ap;
608         va_start(ap, fmt);
609         vFSLog(fmt, ap);
610         va_end(ap);
611     }
612 }
613
614 void
615 Log(char *fmt, ...)
616 {
617     va_list ap;
618
619     va_start(ap, fmt);
620     vFSLog(fmt, ap);
621     va_end(ap);
622 }
623
624 void
625 LogError(long code, char *fmt, ... )
626 {
627     va_list ap;
628     int len;
629     char buffer[1024];
630
631     va_start(ap, fmt);
632     len = vsnprintf(buffer, sizeof(buffer), fmt, ap);
633     va_end(ap);
634     if (len >= 1024) {
635         len = 1023;
636         buffer[1023] = '\0';
637     }
638     /* Be consistent with (unintentional?) historic behavior. */
639     if (code) {
640         FSLog("%s: %s\n", afs_error_table_name(code), afs_error_message(code));
641         WriteLogBuffer(buffer, len);
642     } else {
643         FSLog("%s", buffer);
644     }
645 }