Use strdup to copy strings
[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 = (char **)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((char *)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     path = malloc(strlen(globalConfPtr->databaseDirectory) +
328                   strlen(globalConfPtr->databaseName) +
329                   strlen(globalConfPtr->databaseExtension) + 1);
330     if (path == NULL)
331         ERROR(-1);
332
333     /* construct the database name */
334     strcpy(path, globalConfPtr->databaseDirectory);
335     strcat(path, globalConfPtr->databaseName);
336     strcat(path, globalConfPtr->databaseExtension);
337
338     fd = open(path, O_RDWR, 0755);
339     if (!fd) {
340         code = errno;
341     } else {
342         if (ftruncate(fd, 0) != 0) {
343             code = errno;
344         } else
345             close(fd);
346     }
347
348     free(path);
349
350   error_exit:
351     return (code);
352 }
353
354
355 /* --- */
356
357 #include "AFS_component_version_number.c"
358
359 int
360 main(int argc, char **argv)
361 {
362     char *whoami = argv[0];
363     char *dbNamePtr = 0;
364     struct afsconf_cell cellinfo;
365     time_t currentTime;
366     afs_int32 code = 0;
367     afs_uint32 host = ntohl(INADDR_ANY);
368
369     char  clones[MAXHOSTSPERCELL];
370
371     struct rx_service *tservice;
372     struct rx_securityClass **securityClasses;
373     afs_int32 numClasses;
374
375     extern int rx_stackSize;
376
377 #ifdef AFS_NT40_ENV
378     /* initialize winsock */
379     if (afs_winsockInit() < 0) {
380         ReportErrorEventAlt(AFSEVT_SVR_WINSOCK_INIT_FAILED, 0, argv[0], 0);
381         fprintf(stderr, "%s: Couldn't initialize winsock.\n", whoami);
382         exit(1);
383     }
384 #endif
385
386 #ifdef  AFS_AIX32_ENV
387     /*
388      * The following signal action for AIX is necessary so that in case of a
389      * crash (i.e. core is generated) we can include the user's data section
390      * in the core dump. Unfortunately, by default, only a partial core is
391      * generated which, in many cases, isn't too useful.
392      */
393     struct sigaction nsa;
394
395     sigemptyset(&nsa.sa_mask);
396     nsa.sa_handler = SIG_DFL;
397     nsa.sa_flags = SA_FULLDUMP;
398     sigaction(SIGSEGV, &nsa, NULL);
399     sigaction(SIGABRT, &nsa, NULL);
400 #endif
401     osi_audit_init();
402     osi_audit(BUDB_StartEvent, 0, AUD_END);
403
404     initialize_BUDB_error_table();
405     initializeArgHandler();
406
407     /* Initialize dirpaths */
408     if (!(initAFSDirPath() & AFSDIR_SERVER_PATHS_OK)) {
409 #ifdef AFS_NT40_ENV
410         ReportErrorEventAlt(AFSEVT_SVR_NO_INSTALL_DIR, 0, argv[0], 0);
411 #endif
412         afs_com_err(whoami, errno, "; Unable to obtain AFS server directory.");
413         exit(2);
414     }
415
416     memset(globalConfPtr, 0, sizeof(*globalConfPtr));
417
418     /* set default configuration values */
419     strcpy(dbDir, AFSDIR_SERVER_DB_DIRPATH);
420     strcat(dbDir, "/");
421     globalConfPtr->databaseDirectory = dbDir;
422     globalConfPtr->databaseName = DEFAULT_DBPREFIX;
423     strcpy(cellConfDir, AFSDIR_SERVER_ETC_DIRPATH);
424     globalConfPtr->cellConfigdir = cellConfDir;
425
426     /* open the log file */
427 /*
428     globalConfPtr->log = fopen(DEFAULT_LOGNAME,"a");
429     if ( globalConfPtr->log == NULL )
430     {
431         printf("Can't open log file %s - aborting\n", DEFAULT_LOGNAME);
432         BUDB_EXIT(-1);
433     }
434 */
435
436     srandom(1);
437
438 #ifdef AFS_PTHREAD_ENV
439     SetLogThreadNumProgram( rx_GetThreadNum );
440 #endif
441
442     /* process the user supplied args */
443     helpOption = 1;
444     code = cmd_Dispatch(argc, argv);
445     if (code)
446         ERROR(code);
447
448     /* exit if there was a help option */
449     if (helpOption)
450         BUDB_EXIT(0);
451
452     /* open the log file */
453     globalConfPtr->log = fopen(AFSDIR_SERVER_BUDBLOG_FILEPATH, "a");
454     if (globalConfPtr->log == NULL) {
455         printf("Can't open log file %s - aborting\n",
456                AFSDIR_SERVER_BUDBLOG_FILEPATH);
457         BUDB_EXIT(-1);
458     }
459
460     /* keep log closed so can remove it */
461
462     fclose(globalConfPtr->log);
463
464     /* open the cell's configuration directory */
465     LogDebug(4, "opening %s\n", globalConfPtr->cellConfigdir);
466
467     BU_conf = afsconf_Open(globalConfPtr->cellConfigdir);
468     if (BU_conf == 0) {
469         LogError(code, "Failed getting cell info\n");
470         afs_com_err(whoami, code, "Failed getting cell info");
471         ERROR(BUDB_NOCELLS);
472     }
473
474     code = afsconf_GetLocalCell(BU_conf, lcell, sizeof(lcell));
475     if (code) {
476         LogError(0, "** Can't determine local cell name!\n");
477         ERROR(code);
478     }
479
480     if (globalConfPtr->myHost == 0) {
481         /* if user hasn't supplied a list of servers, extract server
482          * list from the cell's database
483          */
484
485         LogDebug(1, "Using server list from %s cell database.\n", lcell);
486
487         code = afsconf_GetExtendedCellInfo (BU_conf, lcell, 0, &cellinfo,
488                                             clones);
489         code =
490             convert_cell_to_ubik(&cellinfo, &globalConfPtr->myHost,
491                                  globalConfPtr->serverList);
492         if (code)
493             ERROR(code);
494     }
495
496     /* initialize audit user check */
497     osi_audit_set_user_check(BU_conf, BU_IsLocalRealmMatch);
498
499     /* initialize ubik */
500     ubik_SetClientSecurityProcs(afsconf_ClientAuth, afsconf_UpToDate, BU_conf);
501     ubik_SetServerSecurityProcs(afsconf_BuildServerSecurityObjects,
502                                 afsconf_CheckAuth, BU_conf);
503
504     if (ubik_nBuffers == 0)
505         ubik_nBuffers = 400;
506
507     LogError(0, "Will allocate %d ubik buffers\n", ubik_nBuffers);
508
509     dbNamePtr =
510         (char *)malloc(strlen(globalConfPtr->databaseDirectory) +
511                        strlen(globalConfPtr->databaseName) + 1);
512     if (dbNamePtr == 0)
513         ERROR(-1);
514
515     /* construct the database name */
516     strcpy(dbNamePtr, globalConfPtr->databaseDirectory);
517     strcat(dbNamePtr, globalConfPtr->databaseName);     /* name prefix */
518
519     rx_SetRxDeadTime(60);       /* 60 seconds inactive before timeout */
520
521     if (rxBind) {
522         afs_int32 ccode;
523         if (AFSDIR_SERVER_NETRESTRICT_FILEPATH ||
524             AFSDIR_SERVER_NETINFO_FILEPATH) {
525             char reason[1024];
526             ccode = parseNetFiles(SHostAddrs, NULL, NULL,
527                                            ADDRSPERSITE, reason,
528                                            AFSDIR_SERVER_NETINFO_FILEPATH,
529                                            AFSDIR_SERVER_NETRESTRICT_FILEPATH);
530         } else
531         {
532             ccode = rx_getAllAddr(SHostAddrs, ADDRSPERSITE);
533         }
534         if (ccode == 1) {
535             host = SHostAddrs[0];
536             rx_InitHost(host, htons(AFSCONF_BUDBPORT));
537         }
538     }
539
540     /* Disable jumbograms */
541     rx_SetNoJumbo();
542
543     code = ubik_ServerInitByInfo (globalConfPtr->myHost,
544                                   htons(AFSCONF_BUDBPORT),
545                                   &cellinfo,
546                                   clones,
547                                   dbNamePtr,           /* name prefix */
548                                   &BU_dbase);
549
550     if (code) {
551         LogError(code, "Ubik init failed\n");
552         afs_com_err(whoami, code, "Ubik init failed");
553         ERROR(code);
554     }
555
556     afsconf_BuildServerSecurityObjects(BU_conf, &securityClasses, &numClasses);
557
558     tservice =
559         rx_NewServiceHost(host, 0, BUDB_SERVICE, "BackupDatabase",
560                           securityClasses, numClasses, BUDB_ExecuteRequest);
561
562     if (tservice == (struct rx_service *)0) {
563         LogError(0, "Could not create backup database rx service\n");
564         printf("Could not create backup database rx service\n");
565         BUDB_EXIT(3);
566     }
567     rx_SetMinProcs(tservice, 1);
568     rx_SetMaxProcs(tservice, lwps);
569     rx_SetStackSize(tservice, 10000);
570
571     /* allow super users to manage RX statistics */
572     rx_SetRxStatUserOk(BU_rxstat_userok);
573
574     /* misc. initialization */
575
576     /* database dump synchronization */
577     memset(dumpSyncPtr, 0, sizeof(*dumpSyncPtr));
578     Lock_Init(&dumpSyncPtr->ds_lock);
579
580     rx_StartServer(0);          /* start handling requests */
581
582     code = InitProcs();
583     if (code)
584         ERROR(code);
585
586
587     currentTime = time(0);
588     LogError(0, "Ready to process requests at %s\n", ctime(&currentTime));
589
590     rx_ServerProc(NULL);                /* donate this LWP */
591
592   error_exit:
593     osi_audit(BUDB_FinishEvent, code, AUD_END);
594     return (code);
595 }
596
597 void
598 consistencyCheckDb(void)
599 {
600     /* do consistency checks on structure sizes */
601     if ((sizeof(struct htBlock) > BLOCKSIZE)
602         || (sizeof(struct vfBlock) > BLOCKSIZE)
603         || (sizeof(struct viBlock) > BLOCKSIZE)
604         || (sizeof(struct dBlock) > BLOCKSIZE)
605         || (sizeof(struct tBlock) > BLOCKSIZE)
606         ) {
607         fprintf(stderr, "Block layout error!\n");
608         BUDB_EXIT(99);
609     }
610 }
611
612 void
613 LogDebug(int level, char *fmt, ... )
614 {
615     va_list ap;
616
617     va_start(ap, fmt);
618
619     if (debugging >= level) {
620         /* log normally closed so can remove it */
621         globalConfPtr->log = fopen(AFSDIR_SERVER_BUDBLOG_FILEPATH, "a");
622         if (globalConfPtr->log != NULL) {
623             vfprintf(globalConfPtr->log, fmt, ap);
624             fflush(globalConfPtr->log);
625             fclose(globalConfPtr->log);
626         }
627     }
628     va_end(ap);
629 }
630
631 static char *
632 TimeStamp(time_t t)
633 {
634     struct tm *lt;
635     static char timestamp[20];
636
637     lt = localtime(&t);
638     strftime(timestamp, 20, "%m/%d/%Y %H:%M:%S", lt);
639     return timestamp;
640 }
641
642 void
643 Log(char *fmt, ...)
644 {
645     va_list ap;
646     time_t now;
647
648     va_start(ap, fmt);
649     globalConfPtr->log = fopen(AFSDIR_SERVER_BUDBLOG_FILEPATH, "a");
650     if (globalConfPtr->log != NULL) {
651         now = time(0);
652         fprintf(globalConfPtr->log, "%s ", TimeStamp(now));
653
654         vfprintf(globalConfPtr->log, fmt, ap);
655         fflush(globalConfPtr->log);
656         fclose(globalConfPtr->log);
657     }
658     va_end(ap);
659 }
660
661 void
662 LogError(long code, char *fmt, ... )
663 {
664     va_list ap;
665     time_t now;
666
667     va_start(ap, fmt);
668     globalConfPtr->log = fopen(AFSDIR_SERVER_BUDBLOG_FILEPATH, "a");
669
670     if (globalConfPtr->log != NULL) {
671         now = time(0);
672         fprintf(globalConfPtr->log, "%s ", TimeStamp(now));
673
674         if (code)
675             fprintf(globalConfPtr->log, "%s: %s\n", afs_error_table_name(code),
676                     afs_error_message(code));
677         vfprintf(globalConfPtr->log, fmt, ap );
678         fflush(globalConfPtr->log);
679         fclose(globalConfPtr->log);
680     }
681 }
682
683
684 /*  ----------------
685  * debug
686  * ----------------
687  */
688
689 void
690 LogNetDump(struct dump *dumpPtr)
691 {
692     struct dump hostDump;
693     extern buServerConfP globalConfPtr;
694
695     dump_ntoh(dumpPtr, &hostDump);
696
697     globalConfPtr->log = fopen(AFSDIR_SERVER_BUDBLOG_FILEPATH, "a");
698     if (globalConfPtr->log != NULL) {
699         printDump(globalConfPtr->log, &hostDump);
700         fclose(globalConfPtr->log);
701     }
702 }
703