osi-audit-locking-fix-20060201
[openafs.git] / 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 }