src/vlserver/vlserver.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
  13 RCSID
  14     ("$Header$");
  15
  16 #include <afs/stds.h>
  17 #include <sys/types.h>
  18 #include <signal.h>
  19 #include <sys/stat.h>
  20 #ifdef HAVE_FCNTL_H
  21 #include <fcntl.h>
  22 #endif
  23 #ifdef AFS_NT40_ENV
  24 #include <winsock2.h>
  25 #include <WINNT/afsevent.h>
  26 #endif
  27 #ifdef HAVE_SYS_FILE_H
  28 #include <sys/file.h>
  29 #endif
  30 #include <time.h>
  31 #ifdef HAVE_NETDB_H
  32 #include <netdb.h>
  33 #endif
  34 #ifdef HAVE_NETINET_IN_H
  35 #include <netinet/in.h>
  36 #endif
  37 #include <stdio.h>
  38
  39 #ifdef HAVE_STRING_H
  40 #include <string.h>
  41 #else
  42 #ifdef HAVE_STRINGS_H
  43 #include <strings.h>
  44 #endif
  45 #endif
  46
  47 #include <rx/xdr.h>
  48 #include <rx/rx.h>
  49 #include <rx/rx_globals.h>
  50 #include <afs/cellconfig.h>
  51 #include <afs/keys.h>
  52 #include <afs/auth.h>
  53 #include <lock.h>
  54 #include <ubik.h>
  55 #include <afs/afsutil.h>
  56 #include "vlserver.h"
  57
  58
  59 #define MAXLWP 16
  60 const char *vl_dbaseName;
  61 struct afsconf_dir *vldb_confdir = 0;   /* vldb configuration dir */
  62 int lwps = 9;
  63
  64 struct vldstats dynamic_statistics;
  65 struct ubik_dbase *VL_dbase;
  66 afs_uint32 HostAddress[MAXSERVERID + 1];
  67 extern int afsconf_CheckAuth();
  68 extern int afsconf_ServerAuth();
  69
  70 static CheckSignal();
  71 int LogLevel = 0;
  72 int smallMem = 0;
  73 int rxJumbograms = 1;           /* default is to send and receive jumbo grams */
  74
  75 static void
  76 CheckSignal_Signal()
  77 {
  78     IOMGR_SoftSig(CheckSignal, 0);
  79 }
  80
  81 static
  82 CheckSignal()
  83 {
  84     register int i, errorcode;
  85     struct ubik_trans *trans;
  86
  87     if (errorcode =
  88         Init_VLdbase(&trans, LOCKREAD, VLGETSTATS - VL_LOWEST_OPCODE))
  89         return errorcode;
  90     VLog(0, ("Dump name hash table out\n"));
  91     for (i = 0; i < HASHSIZE; i++) {
  92         HashNDump(trans, i);
  93     }
  94     VLog(0, ("Dump id hash table out\n"));
  95     for (i = 0; i < HASHSIZE; i++) {
  96         HashIdDump(trans, i);
  97     }
  98     return (ubik_EndTrans(trans));
  99 }                               /*CheckSignal */
 100
 101
 102 /* Initialize the stats for the opcodes */
 103 void
 104 initialize_dstats()
 105 {
 106     int i;
 107
 108     dynamic_statistics.start_time = (afs_uint32) time(0);
 109     for (i = 0; i < MAX_NUMBER_OPCODES; i++) {
 110         dynamic_statistics.requests[i] = 0;
 111         dynamic_statistics.aborts[i] = 0;
 112     }
 113 }
 114
 115 /* check whether caller is authorized to manage RX statistics */
 116 int
 117 vldb_rxstat_userok(call)
 118      struct rx_call *call;
 119 {
 120     return afsconf_SuperUser(vldb_confdir, call, NULL);
 121 }
 122
 123 /* Main server module */
 124
 125 #include "AFS_component_version_number.c"
 126
 127 main(argc, argv)
 128      int argc;
 129      char **argv;
 130 {
 131     register afs_int32 code;
 132     afs_int32 myHost;
 133     struct rx_service *tservice;
 134     struct rx_securityClass *sc[3];
 135     extern int VL_ExecuteRequest();
 136     extern int RXSTATS_ExecuteRequest();
 137     struct afsconf_dir *tdir;
 138     struct ktc_encryptionKey tkey;
 139     struct afsconf_cell info;
 140     struct hostent *th;
 141     char hostname[VL_MAXNAMELEN];
 142     int noAuth = 0, index, i;
 143     extern int rx_extraPackets;
 144     char commandLine[150];
 145     char clones[MAXHOSTSPERCELL];
 146
 147 #ifdef  AFS_AIX32_ENV
 148     /*
 149      * The following signal action for AIX is necessary so that in case of a
 150      * crash (i.e. core is generated) we can include the user's data section
 151      * in the core dump. Unfortunately, by default, only a partial core is
 152      * generated which, in many cases, isn't too useful.
 153      */
 154     struct sigaction nsa;
 155
 156     rx_extraPackets = 100;      /* should be a switch, I guess... */
 157     sigemptyset(&nsa.sa_mask);
 158     nsa.sa_handler = SIG_DFL;
 159     nsa.sa_flags = SA_FULLDUMP;
 160     sigaction(SIGABRT, &nsa, NULL);
 161     sigaction(SIGSEGV, &nsa, NULL);
 162 #endif
 163     osi_audit_init();
 164
 165     /* Parse command line */
 166     for (index = 1; index < argc; index++) {
 167         if (strcmp(argv[index], "-noauth") == 0) {
 168             noAuth = 1;
 169
 170         } else if (strcmp(argv[index], "-p") == 0) {
 171             lwps = atoi(argv[++index]);
 172             if (lwps > MAXLWP) {
 173                 printf("Warning: '-p %d' is too big; using %d instead\n",
 174                        lwps, MAXLWP);
 175                 lwps = MAXLWP;
 176             }
 177
 178         } else if (strcmp(argv[index], "-nojumbo") == 0) {
 179             rxJumbograms = 0;
 180
 181         } else if (strcmp(argv[index], "-smallmem") == 0) {
 182             smallMem = 1;
 183
 184         } else if (strcmp(argv[index], "-trace") == 0) {
 185             extern char rxi_tracename[80];
 186             strcpy(rxi_tracename, argv[++index]);
 187
 188        } else if (strcmp(argv[index], "-auditlog") == 0) {
 189            int tempfd, flags;
 190            FILE *auditout;
 191            char oldName[MAXPATHLEN];
 192            char *fileName = argv[++index];
 193
 194 #ifndef AFS_NT40_ENV
 195            struct stat statbuf;
 196
 197            if ((lstat(fileName, &statbuf) == 0)
 198                && (S_ISFIFO(statbuf.st_mode))) {
 199                flags = O_WRONLY | O_NONBLOCK;
 200            } else
 201 #endif
 202            {
 203                strcpy(oldName, fileName);
 204                strcat(oldName, ".old");
 205                renamefile(fileName, oldName);
 206                flags = O_WRONLY | O_TRUNC | O_CREAT;
 207            }
 208            tempfd = open(fileName, flags, 0666);
 209            if (tempfd > -1) {
 210                auditout = fdopen(tempfd, "a");
 211                if (auditout) {
 212                    osi_audit_file(auditout);
 213                } else
 214                    printf("Warning: auditlog %s not writable, ignored.\n", fileName);
 215            } else
 216                printf("Warning: auditlog %s not writable, ignored.\n", fileName);
 217         } else if (strcmp(argv[index], "-enable_peer_stats") == 0) {
 218             rx_enablePeerRPCStats();
 219         } else if (strcmp(argv[index], "-enable_process_stats") == 0) {
 220             rx_enableProcessRPCStats();
 221 #ifndef AFS_NT40_ENV
 222         } else if (strcmp(argv[index], "-syslog") == 0) {
 223             /* set syslog logging flag */
 224             serverLogSyslog = 1;
 225         } else if (strncmp(argv[index], "-syslog=", 8) == 0) {
 226             serverLogSyslog = 1;
 227             serverLogSyslogFacility = atoi(argv[index] + 8);
 228 #endif
 229         } else {
 230             /* support help flag */
 231 #ifndef AFS_NT40_ENV
 232             printf("Usage: vlserver [-p <number of processes>] [-nojumbo] "
 233                    "[-auditlog <log path>] "
 234                    "[-syslog[=FACILITY]] "
 235                    "[-enable_peer_stats] [-enable_process_stats] "
 236                    "[-help]\n");
 237 #else
 238             printf("Usage: vlserver [-p <number of processes>] [-nojumbo] "
 239                    "[-auditlog <log path>] "
 240                    "[-enable_peer_stats] [-enable_process_stats] "
 241                    "[-help]\n");
 242 #endif
 243             fflush(stdout);
 244             exit(0);
 245         }
 246     }
 247
 248     /* Initialize dirpaths */
 249     if (!(initAFSDirPath() & AFSDIR_SERVER_PATHS_OK)) {
 250 #ifdef AFS_NT40_ENV
 251         ReportErrorEventAlt(AFSEVT_SVR_NO_INSTALL_DIR, 0, argv[0], 0);
 252 #endif
 253         fprintf(stderr, "%s: Unable to obtain AFS server directory.\n",
 254                 argv[0]);
 255         exit(2);
 256     }
 257     vl_dbaseName = AFSDIR_SERVER_VLDB_FILEPATH;
 258
 259 #ifndef AFS_NT40_ENV
 260     serverLogSyslogTag = "vlserver";
 261 #endif
 262     OpenLog(AFSDIR_SERVER_VLOG_FILEPATH);       /* set up logging */
 263     SetupLogSignals();
 264
 265     tdir = afsconf_Open(AFSDIR_SERVER_ETC_DIRPATH);
 266     if (!tdir) {
 267         printf
 268             ("vlserver: can't open configuration files in dir %s, giving up.\n",
 269              AFSDIR_SERVER_ETC_DIRPATH);
 270         exit(1);
 271     }
 272 #ifdef AFS_NT40_ENV
 273     /* initialize winsock */
 274     if (afs_winsockInit() < 0) {
 275         ReportErrorEventAlt(AFSEVT_SVR_WINSOCK_INIT_FAILED, 0, argv[0], 0);
 276         fprintf(stderr, "vlserver: couldn't initialize winsock. \n");
 277         exit(1);
 278     }
 279 #endif
 280     /* get this host */
 281     gethostname(hostname, sizeof(hostname));
 282     th = gethostbyname(hostname);
 283     if (!th) {
 284         printf("vlserver: couldn't get address of this host (%s).\n",
 285                hostname);
 286         exit(1);
 287     }
 288     memcpy(&myHost, th->h_addr, sizeof(afs_int32));
 289
 290 #if !defined(AFS_HPUX_ENV) && !defined(AFS_NT40_ENV) && !defined(AFS_DJGPP_ENV)
 291     signal(SIGXCPU, CheckSignal_Signal);
 292 #endif
 293     /* get list of servers */
 294     code =
 295         afsconf_GetExtendedCellInfo(tdir, NULL, AFSCONF_VLDBSERVICE, &info,
 296                                     &clones);
 297     if (code) {
 298         printf("vlserver: Couldn't get cell server list for 'afsvldb'.\n");
 299         exit(2);
 300     }
 301
 302     vldb_confdir = tdir;        /* Preserve our configuration dir */
 303     /* rxvab no longer supported */
 304     memset(&tkey, 0, sizeof(tkey));
 305
 306     if (noAuth)
 307         afsconf_SetNoAuthFlag(tdir, 1);
 308
 309     ubik_nBuffers = 512;
 310     ubik_CRXSecurityProc = afsconf_ClientAuth;
 311     ubik_CRXSecurityRock = (char *)tdir;
 312     ubik_SRXSecurityProc = afsconf_ServerAuth;
 313     ubik_SRXSecurityRock = (char *)tdir;
 314     ubik_CheckRXSecurityProc = afsconf_CheckAuth;
 315     ubik_CheckRXSecurityRock = (char *)tdir;
 316     code =
 317         ubik_ServerInitByInfo(myHost, htons(AFSCONF_VLDBPORT), &info, &clones,
 318                               vl_dbaseName, &VL_dbase);
 319     if (code) {
 320         printf("vlserver: Ubik init failed with code %d\n", code);
 321         exit(2);
 322     }
 323     if (!rxJumbograms) {
 324         rx_SetNoJumbo();
 325     }
 326     rx_SetRxDeadTime(50);
 327
 328     memset(HostAddress, 0, sizeof(HostAddress));
 329     initialize_dstats();
 330
 331     sc[0] = rxnull_NewServerSecurityObject();
 332     sc[1] = (struct rx_securityClass *)0;
 333     sc[2] = rxkad_NewServerSecurityObject(0, tdir, afsconf_GetKey, NULL);
 334     tservice =
 335         rx_NewService(0, USER_SERVICE_ID, "Vldb server", sc, 3,
 336                       VL_ExecuteRequest);
 337     if (tservice == (struct rx_service *)0) {
 338         printf("vlserver: Could not create VLDB_SERVICE rx service\n");
 339         exit(3);
 340     }
 341     rx_SetMinProcs(tservice, 2);
 342     if (lwps < 4)
 343         lwps = 4;
 344     rx_SetMaxProcs(tservice, lwps);
 345
 346     tservice =
 347         rx_NewService(0, RX_STATS_SERVICE_ID, "rpcstats", sc, 3,
 348                       RXSTATS_ExecuteRequest);
 349     if (tservice == (struct rx_service *)0) {
 350         printf("vlserver: Could not create rpc stats rx service\n");
 351         exit(3);
 352     }
 353     rx_SetMinProcs(tservice, 2);
 354     rx_SetMaxProcs(tservice, 4);
 355
 356     for (commandLine[0] = '\0', i = 0; i < argc; i++) {
 357         if (i > 0)
 358             strcat(commandLine, " ");
 359         strcat(commandLine, argv[i]);
 360     }
 361     ViceLog(0,
 362             ("Starting AFS vlserver %d (%s)\n", VLDBVERSION_4, commandLine));
 363     printf("%s\n", cml_version_number); /* Goes to the log */
 364
 365     /* allow super users to manage RX statistics */
 366     rx_SetRxStatUserOk(vldb_rxstat_userok);
 367
 368     rx_StartServer(1);          /* Why waste this idle process?? */
 369 }