src/volser/volmain.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 <sys/types.h>
  17 #ifdef AFS_NT40_ENV
  18 #include <time.h>
  19 #include <fcntl.h>
  20 #include <windows.h>
  21 #include <WINNT/afsevent.h>
  22 #else
  23 #include <sys/time.h>
  24 #include <sys/file.h>
  25 #include <netinet/in.h>
  26 #endif
  27 #ifdef HAVE_STRING_H
  28 #include <string.h>
  29 #else
  30 #ifdef HAVE_STRINGS_H
  31 #include <strings.h>
  32 #endif
  33 #endif
  34 #include <rx/xdr.h>
  35 #include <afs/afsint.h>
  36 #include <stdio.h>
  37 #include <signal.h>
  38 #ifdef AFS_PTHREAD_ENV
  39 #include <assert.h>
  40 #else /* AFS_PTHREAD_ENV */
  41 #include <afs/assert.h>
  42 #endif /* AFS_PTHREAD_ENV */
  43 #include <afs/prs_fs.h>
  44 #include <afs/nfs.h>
  45 #include <lwp.h>
  46 #include <lock.h>
  47 #include <afs/afssyscalls.h>
  48 #include <afs/ihandle.h>
  49 #ifdef AFS_NT40_ENV
  50 #include <afs/ntops.h>
  51 #endif
  52 #include <afs/vnode.h>
  53 #include <afs/volume.h>
  54 #include <afs/partition.h>
  55 #include <rx/rx.h>
  56 #include <rx/rx_globals.h>
  57 #include <afs/auth.h>
  58 #include <rx/rxkad.h>
  59 #include <afs/cellconfig.h>
  60 #include <afs/keys.h>
  61 #include <ubik.h>
  62
  63 #include "volser.h"
  64 #include <errno.h>
  65 #include <afs/audit.h>
  66 #include <afs/afsutil.h>
  67
  68 /*@printflike@*/ extern void Log(const char *format, ...);
  69 /*@printflike@*/ extern void Abort(const char *format, ...);
  70
  71 #define VolserVersion "2.0"
  72 #define N_SECURITY_OBJECTS 3
  73
  74 extern struct volser_trans *TransList();
  75 #ifndef AFS_PTHREAD_ENV
  76 extern int (*vol_PollProc) ();
  77 extern int IOMGR_Poll();
  78 #endif
  79 char *GlobalNameHack = NULL;
  80 int hackIsIn = 0;
  81 afs_int32 GlobalVolCloneId, GlobalVolParentId;
  82 int GlobalVolType;
  83 int VolumeChanged;              /* XXXX */
  84 static char busyFlags[MAXHELPERS];
  85 struct volser_trans *QI_GlobalWriteTrans = 0;
  86 extern void AFSVolExecuteRequest();
  87 extern void RXSTATS_ExecuteRequest();
  88 struct afsconf_dir *tdir;
  89 static afs_int32 runningCalls = 0;
  90 int DoLogging = 0;
  91 #define MAXLWP 16
  92 int lwps = 9;
  93 int udpBufSize = 0;             /* UDP buffer size for receive */
  94
  95 int Testing = 0;                /* for ListViceInodes */
  96
  97 #define VS_EXIT(code)  {                                          \
  98                           osi_audit(VS_ExitEvent, code, AUD_END); \
  99                           exit(code);                             \
 100                        }
 101
 102
 103 static afs_int32
 104 MyBeforeProc(struct rx_call *acall)
 105 {
 106     VTRANS_LOCK;
 107     runningCalls++;
 108     VTRANS_UNLOCK;
 109     return 0;
 110 }
 111
 112 static afs_int32
 113 MyAfterProc(struct rx_call *acall, afs_int32 code)
 114 {
 115     VTRANS_LOCK;
 116     runningCalls--;
 117     VTRANS_UNLOCK;
 118     return 0;
 119 }
 120
 121 /* Called every GCWAKEUP seconds to try to unlock all our partitions,
 122  * if we're idle and there are no active transactions
 123  */
 124 static void
 125 TryUnlock()
 126 {
 127     /* if there are no running calls, and there are no active transactions, then
 128      * it should be safe to release any partition locks we've accumulated */
 129     VTRANS_LOCK;
 130     if (runningCalls == 0 && TransList() == (struct volser_trans *)0) {
 131         VTRANS_UNLOCK;
 132         VPFullUnlock();         /* in volprocs.c */
 133     } else
 134         VTRANS_UNLOCK;
 135 }
 136
 137 /* background daemon for timing out transactions */
 138 static void
 139 BKGLoop()
 140 {
 141     struct timeval tv;
 142     int loop = 0;
 143
 144     while (1) {
 145         tv.tv_sec = GCWAKEUP;
 146         tv.tv_usec = 0;
 147 #ifdef AFS_PTHREAD_ENV
 148         select(0, 0, 0, 0, &tv);
 149 #else
 150         (void)IOMGR_Select(0, 0, 0, 0, &tv);
 151 #endif
 152         GCTrans();
 153         TryUnlock();
 154         loop++;
 155         if (loop == 10) {       /* reopen log every 5 minutes */
 156             loop = 0;
 157             ReOpenLog(AFSDIR_SERVER_VOLSERLOG_FILEPATH);
 158         }
 159     }
 160 }
 161
 162 /* Background daemon for sleeping so the volserver does not become I/O bound */
 163 afs_int32 TTsleep, TTrun;
 164 static void
 165 BKGSleep()
 166 {
 167     struct volser_trans *tt;
 168
 169     if (TTsleep) {
 170         while (1) {
 171 #ifdef AFS_PTHREAD_ENV
 172             sleep(TTrun);
 173 #else /* AFS_PTHREAD_ENV */
 174             IOMGR_Sleep(TTrun);
 175 #endif
 176             VTRANS_LOCK;
 177             for (tt = TransList(); tt; tt = tt->next) {
 178                 if ((strcmp(tt->lastProcName, "DeleteVolume") == 0)
 179                     || (strcmp(tt->lastProcName, "Clone") == 0)
 180                     || (strcmp(tt->lastProcName, "ReClone") == 0)
 181                     || (strcmp(tt->lastProcName, "Forward") == 0)
 182                     || (strcmp(tt->lastProcName, "Restore") == 0)
 183                     || (strcmp(tt->lastProcName, "ForwardMulti") == 0))
 184                     break;
 185             }
 186             if (tt) {
 187                 VTRANS_UNLOCK;
 188                 sleep(TTsleep);
 189             } else
 190                 VTRANS_UNLOCK;
 191         }
 192     }
 193 }
 194
 195 #ifndef AFS_NT40_ENV
 196 int
 197 volser_syscall(afs_uint32 a3, afs_uint32 a4, void *a5)
 198 {
 199     afs_uint32 rcode;
 200     void (*old) ();
 201
 202 #ifndef AFS_LINUX20_ENV
 203     old = signal(SIGSYS, SIG_IGN);
 204 #endif
 205     rcode =
 206         syscall(AFS_SYSCALL /* AFS_SYSCALL */ , 28 /* AFSCALL_CALL */ , a3,
 207                 a4, a5);
 208 #ifndef AFS_LINUX20_ENV
 209     signal(SIGSYS, old);
 210 #endif
 211
 212     return rcode;
 213 }
 214 #endif
 215
 216
 217 /* check whether caller is authorized to manage RX statistics */
 218 int
 219 vol_rxstat_userok(struct rx_call *call)
 220 {
 221     return afsconf_SuperUser(tdir, call, NULL);
 222 }
 223
 224 #include "AFS_component_version_number.c"
 225 int
 226 main(int argc, char **argv)
 227 {
 228     register afs_int32 code;
 229     struct rx_securityClass *(securityObjects[3]);
 230     struct rx_service *service;
 231     struct ktc_encryptionKey tkey;
 232     int rxpackets = 100;
 233     char commandLine[150];
 234     int i;
 235     int rxJumbograms = 1;       /* default is to send and receive jumbograms. */
 236     int bufSize = 0;            /* temp variable to read in udp socket buf size */
 237
 238 #ifdef  AFS_AIX32_ENV
 239     /*
 240      * The following signal action for AIX is necessary so that in case of a
 241      * crash (i.e. core is generated) we can include the user's data section
 242      * in the core dump. Unfortunately, by default, only a partial core is
 243      * generated which, in many cases, isn't too useful.
 244      */
 245     struct sigaction nsa;
 246
 247     sigemptyset(&nsa.sa_mask);
 248     nsa.sa_handler = SIG_DFL;
 249     nsa.sa_flags = SA_FULLDUMP;
 250     sigaction(SIGABRT, &nsa, NULL);
 251     sigaction(SIGSEGV, &nsa, NULL);
 252 #endif
 253     osi_audit(VS_StartEvent, 0, AUD_END);
 254
 255     /* Initialize dirpaths */
 256     if (!(initAFSDirPath() & AFSDIR_SERVER_PATHS_OK)) {
 257 #ifdef AFS_NT40_ENV
 258         ReportErrorEventAlt(AFSEVT_SVR_NO_INSTALL_DIR, 0, argv[0], 0);
 259 #endif
 260         fprintf(stderr, "%s: Unable to obtain AFS server directory.\n",
 261                 argv[0]);
 262         exit(2);
 263     }
 264
 265     for (commandLine[0] = '\0', i = 0; i < argc; i++) {
 266         if (i > 0)
 267             strcat(commandLine, " ");
 268         strcat(commandLine, argv[i]);
 269     }
 270
 271     TTsleep = TTrun = 0;
 272
 273     /* parse cmd line */
 274     for (code = 1; code < argc; code++) {
 275         if (strcmp(argv[code], "-log") == 0) {
 276             /* set extra logging flag */
 277             DoLogging = 1;
 278         } else if (strcmp(argv[code], "-help") == 0) {
 279             goto usage;
 280         } else if (strcmp(argv[code], "-p") == 0) {
 281             lwps = atoi(argv[++code]);
 282             if (lwps > MAXLWP) {
 283                 printf("Warning: '-p %d' is too big; using %d instead\n",
 284                        lwps, MAXLWP);
 285                 lwps = MAXLWP;
 286             }
 287         } else if (strcmp(argv[code], "-nojumbo") == 0) {
 288             rxJumbograms = 0;
 289         } else if (strcmp(argv[code], "-sleep") == 0) {
 290             sscanf(argv[++code], "%d/%d", &TTsleep, &TTrun);
 291             if ((TTsleep < 0) || (TTrun <= 0)) {
 292                 printf("Warning: '-sleep %d/%d' is incorrect; ignoring\n",
 293                        TTsleep, TTrun);
 294                 TTsleep = TTrun = 0;
 295             }
 296         } else if (strcmp(argv[code], "-udpsize") == 0) {
 297             if ((code + 1) >= argc) {
 298                 printf("You have to specify -udpsize <integer value>\n");
 299                 exit(1);
 300             }
 301             sscanf(argv[++code], "%d", &bufSize);
 302             if (bufSize < rx_GetMinUdpBufSize())
 303                 printf
 304                     ("Warning:udpsize %d is less than minimum %d; ignoring\n",
 305                      bufSize, rx_GetMinUdpBufSize());
 306             else
 307                 udpBufSize = bufSize;
 308         } else if (strcmp(argv[code], "-enable_peer_stats") == 0) {
 309             rx_enablePeerRPCStats();
 310         } else if (strcmp(argv[code], "-enable_process_stats") == 0) {
 311             rx_enableProcessRPCStats();
 312         }
 313 #ifndef AFS_NT40_ENV
 314         else if (strcmp(argv[code], "-syslog") == 0) {
 315             /* set syslog logging flag */
 316             serverLogSyslog = 1;
 317         } else if (strncmp(argv[code], "-syslog=", 8) == 0) {
 318             serverLogSyslog = 1;
 319             serverLogSyslogFacility = atoi(argv[code] + 8);
 320         }
 321 #endif
 322         else {
 323             printf("volserver: unrecognized flag '%s'\n", argv[code]);
 324           usage:
 325 #ifndef AFS_NT40_ENV
 326             printf("Usage: volserver [-log] [-p <number of processes>] "
 327                    "[-udpsize <size of socket buffer in bytes>] "
 328                    "[-syslog[=FACILITY]] "
 329                    "[-enable_peer_stats] [-enable_process_stats] "
 330                    "[-help]\n");
 331 #else
 332             printf("Usage: volserver [-log] [-p <number of processes>] "
 333                    "[-udpsize <size of socket buffer in bytes>] "
 334                    "[-enable_peer_stats] [-enable_process_stats] "
 335                    "[-help]\n");
 336 #endif
 337             VS_EXIT(1);
 338         }
 339     }
 340 #ifdef AFS_SGI_VNODE_GLUE
 341     if (afs_init_kernel_config(-1) < 0) {
 342         printf
 343             ("Can't determine NUMA configuration, not starting volserver.\n");
 344         exit(1);
 345     }
 346 #endif
 347     InitErrTabs();
 348
 349 #ifdef AFS_NT40_ENV
 350     if (afs_winsockInit() < 0) {
 351         ReportErrorEventAlt(AFSEVT_SVR_WINSOCK_INIT_FAILED, 0, argv[0], 0);
 352         printf("Volume server unable to start winsock, exiting.\n");
 353         exit(1);
 354     }
 355 #endif
 356     VInitVolumePackage(volumeUtility, 0, 0, CONNECT_FS, 0);
 357     DInit(40);
 358 #ifndef AFS_PTHREAD_ENV
 359     vol_PollProc = IOMGR_Poll;  /* tell vol pkg to poll io system periodically */
 360 #endif
 361 #ifndef AFS_NT40_ENV
 362     rxi_syscallp = volser_syscall;
 363 #endif
 364     rx_nPackets = rxpackets;    /* set the max number of packets */
 365     if (udpBufSize)
 366         rx_SetUdpBufSize(udpBufSize);   /* set the UDP buffer size for receive */
 367     code = rx_Init((int)htons(AFSCONF_VOLUMEPORT));
 368     if (code) {
 369         fprintf(stderr, "rx init failed on socket AFSCONF_VOLUMEPORT %u\n",
 370                 AFSCONF_VOLUMEPORT);
 371         VS_EXIT(1);
 372     }
 373     if (!rxJumbograms) {
 374         /* Don't allow 3.4 vos clients to send jumbograms and we don't send. */
 375         rx_SetNoJumbo();
 376     }
 377     rx_GetIFInfo();
 378     rx_SetRxDeadTime(420);
 379     memset(busyFlags, 0, sizeof(busyFlags));
 380
 381     /* Open FileLog and map stdout, stderr into it */
 382     OpenLog(AFSDIR_SERVER_VOLSERLOG_FILEPATH);
 383     SetupLogSignals();
 384
 385     {
 386 #ifdef AFS_PTHREAD_ENV
 387         pthread_t tid;
 388         pthread_attr_t tattr;
 389         assert(pthread_attr_init(&tattr) == 0);
 390         assert(pthread_attr_setdetachstate(&tattr, PTHREAD_CREATE_DETACHED) == 0);
 391
 392         assert(pthread_create(&tid, &tattr, (void *)BKGLoop, NULL) == 0);
 393 #else
 394         PROCESS pid;
 395         LWP_CreateProcess(BKGLoop, 16*1024, 3, 0, "vol bkg daemon", &pid);
 396         LWP_CreateProcess(BKGSleep,16*1024, 3, 0, "vol slp daemon", &pid);
 397 #endif
 398     }
 399
 400     /* Create a single security object, in this case the null security object, for unauthenticated connections, which will be used to control security on connections made to this server */
 401
 402     tdir = afsconf_Open(AFSDIR_SERVER_ETC_DIRPATH);
 403     if (!tdir) {
 404         Abort("volser: could not open conf files in %s\n",
 405               AFSDIR_SERVER_ETC_DIRPATH);
 406         VS_EXIT(1);
 407     }
 408     afsconf_GetKey(tdir, 999, &tkey);
 409     securityObjects[0] = rxnull_NewServerSecurityObject();
 410     securityObjects[1] = (struct rx_securityClass *)0;  /* don't bother with rxvab */
 411     securityObjects[2] =
 412         rxkad_NewServerSecurityObject(0, tdir, afsconf_GetKey, NULL);
 413     if (securityObjects[0] == (struct rx_securityClass *)0)
 414         Abort("rxnull_NewServerSecurityObject");
 415     service =
 416         rx_NewService(0, VOLSERVICE_ID, "VOLSER", securityObjects, 3,
 417                       AFSVolExecuteRequest);
 418     if (service == (struct rx_service *)0)
 419         Abort("rx_NewService");
 420     rx_SetBeforeProc(service, MyBeforeProc);
 421     rx_SetAfterProc(service, MyAfterProc);
 422     rx_SetIdleDeadTime(service, 0);     /* never timeout */
 423     if (lwps < 4)
 424         lwps = 4;
 425     rx_SetMaxProcs(service, lwps);
 426 #ifdef AFS_SGI_ENV
 427     rx_SetStackSize(service, 49152);
 428 #else
 429     rx_SetStackSize(service, 32768);
 430 #endif
 431
 432     service =
 433         rx_NewService(0, RX_STATS_SERVICE_ID, "rpcstats", securityObjects, 3,
 434                       RXSTATS_ExecuteRequest);
 435     if (service == (struct rx_service *)0)
 436         Abort("rx_NewService");
 437     rx_SetMinProcs(service, 2);
 438     rx_SetMaxProcs(service, 4);
 439
 440     Log("Starting AFS Volserver %s (%s)\n", VolserVersion, commandLine);
 441     if (TTsleep) {
 442         Log("Will sleep %d second%s every %d second%s\n", TTsleep,
 443             (TTsleep > 1) ? "s" : "", TTrun + TTsleep,
 444             (TTrun + TTsleep > 1) ? "s" : "");
 445     }
 446
 447     /* allow super users to manage RX statistics */
 448     rx_SetRxStatUserOk(vol_rxstat_userok);
 449
 450     rx_StartServer(1);          /* Donate this process to the server process pool */
 451
 452     osi_audit(VS_FinishEvent, (-1), AUD_END);
 453     Abort("StartServer returned?");
 454 }