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