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