nfsiod.tproj/nfsiod.c

   1 /*
   2  * Copyright (c) 1999-2003 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * "Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
   7  * Reserved.  This file contains Original Code and/or Modifications of
   8  * Original Code as defined in and that are subject to the Apple Public
   9  * Source License Version 1.0 (the 'License').  You may not use this file
  10  * except in compliance with the License.  Please obtain a copy of the
  11  * License at http://www.apple.com/publicsource and read it before using
  12  * this file.
  13  *
  14  * The Original Code and all software distributed under the License are
  15  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  16  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  17  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
  19  * License for the specific language governing rights and limitations
  20  * under the License."
  21  *
  22  * @APPLE_LICENSE_HEADER_END@
  23  */
  24 /*
  25  * Copyright (c) 1989, 1993
  26  *      The Regents of the University of California.  All rights reserved.
  27  *
  28  * This code is derived from software contributed to Berkeley by
  29  * Rick Macklem at The University of Guelph.
  30  *
  31  * Redistribution and use in source and binary forms, with or without
  32  * modification, are permitted provided that the following conditions
  33  * are met:
  34  * 1. Redistributions of source code must retain the above copyright
  35  *    notice, this list of conditions and the following disclaimer.
  36  * 2. Redistributions in binary form must reproduce the above copyright
  37  *    notice, this list of conditions and the following disclaimer in the
  38  *    documentation and/or other materials provided with the distribution.
  39  * 3. All advertising materials mentioning features or use of this software
  40  *    must display the following acknowledgement:
  41  *      This product includes software developed by the University of
  42  *      California, Berkeley and its contributors.
  43  * 4. Neither the name of the University nor the names of its contributors
  44  *    may be used to endorse or promote products derived from this software
  45  *    without specific prior written permission.
  46  *
  47  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  48  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  49  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  50  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  51  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  52  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  53  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  54  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  55  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  56  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  57  * SUCH DAMAGE.
  58  */
  59
  60
  61 #include <sys/param.h>
  62 #include <sys/ioctl.h>
  63 #include <sys/syslog.h>
  64 #include <sys/ucred.h>
  65 #include <sys/wait.h>
  66
  67 #include <sys/mount.h>
  68 #include <sys/time.h>
  69 #include <nfs/rpcv2.h>
  70 #include <nfs/nfsproto.h>
  71 #include <nfs/nfs.h>
  72
  73 #include <err.h>
  74 #include <errno.h>
  75 #include <fcntl.h>
  76 #include <signal.h>
  77 #include <stdio.h>
  78 #include <stdlib.h>
  79 #include <unistd.h>
  80 #include <string.h>
  81 #include <pthread.h>
  82
  83 /* Global defs */
  84 #ifdef DEBUG
  85 int debug = 1;
  86 #else
  87 int debug = 0;
  88 #endif
  89 int *thread_status = NULL;
  90 pthread_cond_t cond;
  91 pthread_mutex_t mutex;
  92
  93 void nonfs __P((int));
  94 void usage __P((void));
  95 void *nfsiod_thread __P((void *));
  96
  97 /*
  98  * Nfsiod does asynchronous buffered I/O on behalf of the NFS client.
  99  * It does not have to be running for correct operation, but will
 100  * improve throughput.
 101  */
 102 int
 103 main(argc, argv)
 104         int argc;
 105         char *argv[];
 106 {
 107         int ch, num_servers;
 108         int i, rv, threadcnt;
 109
 110 #define MAXNFSIODCNT      32
 111 #define DEFNFSIODCNT       1
 112         num_servers = DEFNFSIODCNT;
 113         while ((ch = getopt(argc, argv, "n:")) != EOF)
 114                 switch (ch) {
 115                 case 'n':
 116                         num_servers = atoi(optarg);
 117                         if (num_servers < 1 || num_servers > MAXNFSIODCNT) {
 118                                 warnx("nfsiod count %d; reset to %d",
 119                                     num_servers, DEFNFSIODCNT);
 120                                 num_servers = DEFNFSIODCNT;
 121                         }
 122                         break;
 123                 case '?':
 124                 default:
 125                         usage();
 126                 }
 127         argc -= optind;
 128         argv += optind;
 129
 130         /*
 131          * XXX
 132          * Backward compatibility, trailing number is the count of daemons.
 133          */
 134         if (argc > 1)
 135                 usage();
 136         if (argc == 1) {
 137                 num_servers = atoi(argv[0]);
 138                 if (num_servers < 1 || num_servers > MAXNFSIODCNT) {
 139                         warnx("nfsiod count %d; reset to %d",
 140                                 num_servers, DEFNFSIODCNT);
 141                         num_servers = DEFNFSIODCNT;
 142                 }
 143         }
 144
 145         thread_status = malloc(sizeof(int) * num_servers);
 146         if (thread_status == NULL)
 147                 errx(1, "unable to allocate memory");
 148         rv = pthread_cond_init(&cond, NULL);
 149         if (rv)
 150                 errc(1, rv, "condition variable init failed");
 151         rv = pthread_mutex_init(&mutex, NULL);
 152         if (rv)
 153                 errc(1, rv, "mutex init failed");
 154
 155         if (debug == 0) {
 156                 daemon(0, 0);
 157                 (void)signal(SIGHUP, SIG_IGN);
 158                 (void)signal(SIGINT, SIG_IGN);
 159                 (void)signal(SIGQUIT, SIG_IGN);
 160                 (void)signal(SIGSYS, nonfs);
 161         }
 162
 163         openlog("nfsiod:", LOG_PID, LOG_DAEMON);
 164
 165         threadcnt = 0;
 166         for (i=0; i < num_servers; i++) {
 167                 pthread_t thd;
 168                 thread_status[i] = 1;
 169                 rv = pthread_create(&thd, NULL, nfsiod_thread, (void*)i);
 170                 if (rv) {
 171                         syslog(LOG_ERR, "thread_create: %s", strerror(rv));
 172                         thread_status[i] = 0;
 173                         continue;
 174                 }
 175                 threadcnt++;
 176         }
 177         /* if no threads started exit */
 178         if (!threadcnt)
 179                 errx(1, "unable to start any threads");
 180         if (threadcnt != num_servers)
 181                 syslog(LOG_ERR, "only able to create %d of %d threads",
 182                         threadcnt, num_servers);
 183
 184         /* wait for threads to complete */
 185         rv = pthread_mutex_lock(&mutex);
 186         if (rv)
 187                 errc(1, rv, "mutex lock failed");
 188         while (threadcnt > 0) {
 189                 rv = pthread_cond_wait(&cond, &mutex);
 190                 if (rv)
 191                         errc(1, rv, "nfsiod: cond wait failed");
 192                 for (i=0; i < num_servers; i++) {
 193                         if (!thread_status[i])
 194                                 continue;
 195                         if (thread_status[i] == 1)
 196                                 continue;
 197                         threadcnt--;
 198                         thread_status[i] = 0;
 199                         syslog(LOG_ERR, "lost nfsiod thread %d - "
 200                                 "%d of %d threads remain",
 201                                 i, threadcnt, num_servers);
 202                 }
 203                 rv = pthread_mutex_lock(&mutex);
 204                 if (rv)
 205                         errc(1, rv, "mutex lock failed");
 206         }
 207
 208         exit (0);
 209 }
 210
 211 void *
 212 nfsiod_thread(void *arg)
 213 {
 214         int rv, thread = (int)arg;
 215         if ((rv = nfssvc(NFSSVC_BIOD, NULL)) < 0) {
 216                 thread_status[thread] = rv;
 217                 syslog(LOG_ERR, "nfssvc: %s", strerror(rv));
 218                 pthread_cond_signal(&cond);
 219                 return NULL;
 220         }
 221         thread_status[thread] = 0;
 222         pthread_cond_signal(&cond);
 223         return NULL;
 224 }
 225
 226 void
 227 nonfs(signo)
 228         int signo;
 229 {
 230         syslog(LOG_ERR, "missing system call: NFS not available.");
 231 }
 232
 233 void
 234 usage()
 235 {
 236         (void)fprintf(stderr, "usage: nfsiod [-n num_servers]\n");
 237         exit(1);
 238 }