viced: fix get-statistics64 buffer overflow
[openafs.git] / src / ptserver / ptserver.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 /*
11  *                      A general description of the supergroup changes
12  *                      to ptserver:
13  *
14  *                      In AFS users can be members of groups. When you add
15  *                      a user, u1, to a group, g1, you add the user id for u1
16  *                      to an entries list in g1. This is a list of all the
17  *                      members of g1.
18  *                      You also add the id for g1 to an entries list in u1.
19  *                      This is a list of all the groups this user is a
20  *                      member of.
21  *                      The list has room for 10 entries. If more are required,
22  *                      a continuation record is created.
23  *
24  *                      With UMICH mods, u1 can be a group. When u1 is a group
25  *                      a new field is required to list the groups this group
26  *                      is a member of (since the entries field is used to
27  *                      list it's members). This new field is supergroups and
28  *                      has two entries. If more are required, a continuation
29  *                      record is formed.
30  *                      There are two additional fields required, nextsg is
31  *                      an address of the next continuation record for this
32  *                      group, and countsg is the count for the number of
33  *                      groups this group is a member of.
34  *
35  *
36  *
37  *      09/18/95 jjm    Add mdw's changes to afs-3.3a Changes:
38  *                      (1) Add the parameter -groupdepth or -depth to
39  *                          define the maximum search depth for supergroups.
40  *                          Define the variable depthsg to be the value of
41  *                          the parameter. The default value is set to 5.
42  *
43  *                      (3) Make sure the sizes of the struct prentry and
44  *                          struct prentryg are equal. If they aren't equal
45  *                          the pt database will be corrupted.
46  *                          The error is reported with an fprintf statement,
47  *                          but this doesn't print when ptserver is started by
48  *                          bos, so all you see is an error message in the
49  *                          /usr/afs/logs/BosLog file. If you start the
50  *                          ptserver without bos the fprintf will print
51  *                          on stdout.
52  *                          The program will terminate with a PT_EXIT(1).
53  *
54  *
55  *                      Transarc does not currently use opcodes past 520, but
56  *                      they *could* decide at any time to use more opcodes.
57  *                      If they did, then one part of our local mods,
58  *                      ListSupergroups, would break.  I've therefore
59  *                      renumbered it to 530, and put logic in to enable the
60  *                      old opcode to work (for now).
61  *
62  *      2/1/98 jjm      Add mdw's changes for bit mapping for supergroups
63  *                      Overview:
64  *                      Before fetching a supergroup, this version of ptserver
65  *                      checks to see if it was marked "found" and "not a
66  *                      member".  If and only if so, it doesn't fetch the group.
67  *                      Since this should be the case with most groups, this
68  *                      should save a significant amount of CPU in redundant
69  *                      fetches of the same group.  After fetching the group,
70  *                      it sets "found", and either sets or clears "not a
71  *                      member", depending on if the group was a member of
72  *                      other groups.  When it writes group entries to the
73  *                      database, it clears the "found" flag.
74  */
75
76 #if defined(SUPERGROUPS)
77 /*
78  *  A general description of the supergroup changes
79  *  to ptserver:
80  *
81  *  In AFS users can be members of groups. When you add a user, u1,
82  *  to a group, g1, you add the user id for u1 to an entries list
83  *  in g1. This is a list of all the members of g1.  You also add
84  *  the id for g1 to an entries list in u1.  This is a list of all
85  *  the groups this user is a member of.  The list has room for 10
86  *  entries. If more are required, a continuation record is created.
87  *
88  *  With UMICH mods, u1 can be a group. When u1 is a group a new
89  *  field is required to list the groups this group is a member of
90  *  (since the entries field is used to list it's members). This
91  *  new field is supergroups and has two entries. If more are
92  *  required, a continuation record is formed.
93  *
94  *  There are two additional fields required, nextsg is an address
95  *  of the next continuation record for this group, and countsg is
96  *  the count for the number of groups this group is a member of.
97  *
98  *  Bit mapping support for supergroups:
99  *
100  *  Before fetching a supergroup, this version of ptserver checks to
101  *  see if it was marked "found" and "not a member".  If and only if
102  *  so, it doesn't fetch the group.  Since this should be the case
103  *  with most groups, this should save a significant amount of CPU in
104  *  redundant fetches of the same group.  After fetching the group, it
105  *  sets "found", and either sets or clears "not a member", depending
106  *  on if the group was a member of other groups.  When it writes
107  *  group entries to the database, it clears the "found" flag.
108  */
109 #endif
110
111 #include <afsconfig.h>
112 #include <afs/param.h>
113 #include <afs/stds.h>
114
115 #include <roken.h>
116 #include <afs/opr.h>
117
118 #ifdef AFS_NT40_ENV
119 #include <WINNT/afsevent.h>
120 #endif
121
122 #include <rx/xdr.h>
123 #include <rx/rx.h>
124 #include <rx/rx_globals.h>
125 #include <rx/rxstat.h>
126 #include <lock.h>
127 #include <ubik.h>
128 #include <afs/cmd.h>
129 #include <afs/cellconfig.h>
130 #include <afs/auth.h>
131 #include <afs/keys.h>
132 #include <afs/afsutil.h>
133 #include <afs/audit.h>
134 #include <afs/com_err.h>
135
136 #include "ptserver.h"
137 #include "ptprototypes.h"
138 #include "error_macros.h"
139
140 /* make all of these into a structure if you want */
141 struct prheader cheader;
142 struct ubik_dbase *dbase;
143 struct afsconf_dir *prdir;
144
145 #if defined(SUPERGROUPS)
146 extern afs_int32 depthsg;
147 #endif
148
149 int restricted = 0;
150 int rxMaxMTU = -1;
151 int rxBind = 0;
152 int rxkadDisableDotCheck = 0;
153
154 #define ADDRSPERSITE 16         /* Same global is in rx/rx_user.c */
155 afs_uint32 SHostAddrs[ADDRSPERSITE];
156
157 static struct afsconf_cell info;
158
159 extern int prp_group_default;
160 extern int prp_user_default;
161
162 #include "AFS_component_version_number.c"
163
164 int
165 prp_access_mask(char *s)
166 {
167     int r;
168     if (*s >= '0' && *s <= '9') {
169         return strtol(s, NULL, 0);
170     }
171     r = 0;
172     while (*s) switch(*s++)
173     {
174     case 'S':   r |= PRP_STATUS_ANY; break;
175     case 's':   r |= PRP_STATUS_MEM; break;
176     case 'O':   r |= PRP_OWNED_ANY; break;
177     case 'M':   r |= PRP_MEMBER_ANY; break;
178     case 'm':   r |= PRP_MEMBER_MEM; break;
179     case 'A':   r |= PRP_ADD_ANY; break;
180     case 'a':   r |= PRP_ADD_MEM; break;
181     case 'r':   r |= PRP_REMOVE_MEM; break;
182     }
183     return r;
184 }
185
186 /* check whether caller is authorized to manage RX statistics */
187 int
188 pr_rxstat_userok(struct rx_call *call)
189 {
190     return afsconf_SuperUser(prdir, call, NULL);
191 }
192
193 /**
194  * Return true if this name is a member of the local realm.
195  */
196 int
197 pr_IsLocalRealmMatch(void *rock, char *name, char *inst, char *cell)
198 {
199     struct afsconf_dir *dir = (struct afsconf_dir *)rock;
200     afs_int32 islocal = 0;      /* default to no */
201     int code;
202
203     code = afsconf_IsLocalRealmMatch(dir, &islocal, name, inst, cell);
204     if (code) {
205         ViceLog(0, ("Failed local realm check; code=%d, name=%s, inst=%s, cell=%s\n",
206                  code, name, inst, cell));
207     }
208     return islocal;
209 }
210
211
212 enum optionsList {
213     OPT_database,
214     OPT_access,
215     OPT_groupdepth,
216     OPT_restricted,
217     OPT_auditlog,
218     OPT_auditiface,
219     OPT_config,
220     OPT_debug,
221     OPT_logfile,
222     OPT_threads,
223     OPT_syslog,
224     OPT_peer,
225     OPT_process,
226     OPT_rxbind,
227     OPT_rxmaxmtu,
228     OPT_dotted
229 };
230
231 int
232 main(int argc, char **argv)
233 {
234     afs_int32 code;
235     afs_uint32 myHost;
236     struct hostent *th;
237     char hostname[64];
238     struct rx_service *tservice;
239     struct rx_securityClass **securityClasses;
240     afs_int32 numClasses;
241     int lwps = 3;
242     char clones[MAXHOSTSPERCELL];
243     afs_uint32 host = htonl(INADDR_ANY);
244     struct cmd_syndesc *opts;
245     struct cmd_item *list;
246
247     char *pr_dbaseName;
248     char *configDir;
249     char *logFile;
250     char *whoami = "ptserver";
251
252     char *auditFileName = NULL;
253     char *interface = NULL;
254
255 #ifdef  AFS_AIX32_ENV
256     /*
257      * The following signal action for AIX is necessary so that in case of a
258      * crash (i.e. core is generated) we can include the user's data section
259      * in the core dump. Unfortunately, by default, only a partial core is
260      * generated which, in many cases, isn't too useful.
261      */
262     struct sigaction nsa;
263
264     sigemptyset(&nsa.sa_mask);
265     nsa.sa_handler = SIG_DFL;
266     nsa.sa_flags = SA_FULLDUMP;
267     sigaction(SIGABRT, &nsa, NULL);
268     sigaction(SIGSEGV, &nsa, NULL);
269 #endif
270     osi_audit_init();
271     osi_audit(PTS_StartEvent, 0, AUD_END);
272
273     /* Initialize dirpaths */
274     if (!(initAFSDirPath() & AFSDIR_SERVER_PATHS_OK)) {
275 #ifdef AFS_NT40_ENV
276         ReportErrorEventAlt(AFSEVT_SVR_NO_INSTALL_DIR, 0, argv[0], 0);
277 #endif
278         fprintf(stderr, "%s: Unable to obtain AFS server directory.\n",
279                 argv[0]);
280         exit(2);
281     }
282
283     pr_dbaseName = strdup(AFSDIR_SERVER_PRDB_FILEPATH);
284     configDir = strdup(AFSDIR_SERVER_ETC_DIRPATH);
285     logFile = strdup(AFSDIR_SERVER_PTLOG_FILEPATH);
286
287 #if defined(SUPERGROUPS)
288     /* make sure the structures for database records are the same size */
289     if ((sizeof(struct prentry) != ENTRYSIZE)
290         || (sizeof(struct prentryg) != ENTRYSIZE)) {
291         fprintf(stderr,
292                 "The structures for the database records are different"
293                 " sizes\n" "struct prentry = %" AFS_SIZET_FMT "\n"
294                 "struct prentryg = %" AFS_SIZET_FMT "\n"
295                 "ENTRYSIZE = %d\n", sizeof(struct prentry),
296                 sizeof(struct prentryg), ENTRYSIZE);
297         PT_EXIT(1);
298     }
299 #endif
300
301     cmd_DisableAbbreviations();
302     cmd_DisablePositionalCommands();
303     opts = cmd_CreateSyntax(NULL, NULL, NULL, NULL);
304
305 /* ptserver specific options */
306     cmd_AddParmAtOffset(opts, OPT_database, "-database", CMD_SINGLE,
307                         CMD_OPTIONAL, "database file");
308     cmd_AddParmAlias(opts, OPT_database, "-db");
309
310     cmd_AddParmAtOffset(opts, OPT_access, "-default_access", CMD_SINGLE,
311                         CMD_OPTIONAL, "default access flags for new entries");
312 #if defined(SUPERGROUPS)
313     cmd_AddParmAtOffset(opts, OPT_groupdepth, "-groupdepth", CMD_SINGLE,
314                         CMD_OPTIONAL, "max search depth for supergroups");
315     cmd_AddParmAlias(opts, OPT_groupdepth, "-depth");
316 #endif
317     cmd_AddParmAtOffset(opts, OPT_restricted, "-restricted", CMD_FLAG,
318                         CMD_OPTIONAL, "enable restricted mode");
319
320     /* general server options */
321     cmd_AddParmAtOffset(opts, OPT_auditlog, "-auditlog", CMD_SINGLE,
322                         CMD_OPTIONAL, "location of audit log");
323     cmd_AddParmAtOffset(opts, OPT_auditiface, "-audit-interface", CMD_SINGLE,
324                         CMD_OPTIONAL, "interface to use for audit logging");
325     cmd_AddParmAtOffset(opts, OPT_config, "-config", CMD_SINGLE,
326                         CMD_OPTIONAL, "configuration location");
327     cmd_AddParmAtOffset(opts, OPT_debug, "-d", CMD_SINGLE,
328                         CMD_OPTIONAL, "debug level");
329     cmd_AddParmAtOffset(opts, OPT_logfile, "-logfile", CMD_SINGLE,
330                         CMD_OPTIONAL, "location of logfile");
331     cmd_AddParmAtOffset(opts, OPT_threads, "-p", CMD_SINGLE,
332                         CMD_OPTIONAL, "number of threads");
333 #if !defined(AFS_NT40_ENV)
334     cmd_AddParmAtOffset(opts, OPT_syslog, "-syslog", CMD_SINGLE_OR_FLAG, 
335                         CMD_OPTIONAL, "log to syslog");
336 #endif
337
338     /* rx options */
339     cmd_AddParmAtOffset(opts, OPT_peer, "-enable_peer_stats", CMD_FLAG,
340                         CMD_OPTIONAL, "enable RX transport statistics");
341     cmd_AddParmAtOffset(opts, OPT_process, "-enable_process_stats", CMD_FLAG,
342                         CMD_OPTIONAL, "enable RX RPC statistics");
343     cmd_AddParmAtOffset(opts, OPT_rxbind, "-rxbind", CMD_FLAG,
344                         CMD_OPTIONAL, "bind only to the primary interface");
345     cmd_AddParmAtOffset(opts, OPT_rxmaxmtu, "-rxmaxmtu", CMD_SINGLE,
346                         CMD_OPTIONAL, "maximum MTU for RX");
347
348     /* rxkad options */
349     cmd_AddParmAtOffset(opts, OPT_dotted, "-allow-dotted-principals",
350                         CMD_FLAG, CMD_OPTIONAL,
351                         "permit Kerberos 5 principals with dots");
352
353     code = cmd_Parse(argc, argv, &opts);
354     if (code == CMD_HELP) {
355         PT_EXIT(0);
356     }
357     if (code)
358         PT_EXIT(1);
359
360     cmd_OptionAsString(opts, OPT_config, &configDir);
361
362     cmd_OpenConfigFile(AFSDIR_SERVER_CONFIG_FILE_FILEPATH);
363     cmd_SetCommandName("ptserver");
364
365     if (cmd_OptionAsList(opts, OPT_access, &list) == 0) {
366         prp_user_default = prp_access_mask(list->data);
367         if (list->next == NULL || list->next->data == NULL) {
368             fprintf(stderr, "Missing second argument for -default_access\n");
369             PT_EXIT(1);
370         }
371         prp_group_default = prp_access_mask(list->next->data);
372     }
373
374 #if defined(SUPERGROUPS)
375     cmd_OptionAsInt(opts, OPT_groupdepth, &depthsg);
376 #endif
377
378     cmd_OptionAsFlag(opts, OPT_restricted, &restricted);
379
380     /* general server options */
381     cmd_OptionAsString(opts, OPT_auditlog, &auditFileName);
382
383     if (cmd_OptionAsString(opts, OPT_auditiface, &interface) == 0) {
384         if (osi_audit_interface(interface)) {
385             printf("Invalid audit interface '%s'\n", interface);
386             PT_EXIT(1);
387         }
388         free(interface);
389     }
390
391     cmd_OptionAsInt(opts, OPT_debug, &LogLevel);
392     cmd_OptionAsString(opts, OPT_database, &pr_dbaseName);
393     cmd_OptionAsString(opts, OPT_logfile, &logFile);
394
395     if (cmd_OptionAsInt(opts, OPT_threads, &lwps) == 0) {
396         if (lwps > 64) {        /* maximum of 64 */
397             printf("Warning: '-p %d' is too big; using %d instead\n",
398                    lwps, 64);
399             lwps = 64;
400         } else if (lwps < 3) {  /* minimum of 3 */
401             printf("Warning: '-p %d' is too small; using %d instead\n",
402                    lwps, 3);
403             lwps = 3;
404         }
405     }
406
407 #ifndef AFS_NT40_ENV
408     if (cmd_OptionPresent(opts, OPT_syslog)) {
409         serverLogSyslog = 1;
410         cmd_OptionAsInt(opts, OPT_syslog, &serverLogSyslogFacility);
411     }
412 #endif
413
414     /* rx options */
415     if (cmd_OptionPresent(opts, OPT_peer))
416         rx_enablePeerRPCStats();
417
418     if (cmd_OptionPresent(opts, OPT_process))
419         rx_enableProcessRPCStats();
420
421     cmd_OptionAsFlag(opts, OPT_rxbind, &rxBind);
422
423     cmd_OptionAsInt(opts, OPT_rxmaxmtu, &rxMaxMTU);
424
425     /* rxkad options */
426     cmd_OptionAsFlag(opts, OPT_dotted, &rxkadDisableDotCheck);
427
428     cmd_FreeOptions(&opts);
429
430     if (auditFileName) {
431         osi_audit_file(auditFileName);
432         osi_audit(PTS_StartEvent, 0, AUD_END);
433     }
434
435 #ifndef AFS_NT40_ENV
436     serverLogSyslogTag = "ptserver";
437 #endif
438     OpenLog(logFile);   /* set up logging */
439     SetupLogSignals();
440
441     prdir = afsconf_Open(configDir);
442     if (!prdir) {
443         fprintf(stderr, "ptserver: can't open configuration directory.\n");
444         PT_EXIT(1);
445     }
446     if (afsconf_GetNoAuthFlag(prdir))
447         printf("ptserver: running unauthenticated\n");
448
449 #ifdef AFS_NT40_ENV
450     /* initialize winsock */
451     if (afs_winsockInit() < 0) {
452         ReportErrorEventAlt(AFSEVT_SVR_WINSOCK_INIT_FAILED, 0, argv[0], 0);
453
454         fprintf(stderr, "ptserver: couldn't initialize winsock. \n");
455         PT_EXIT(1);
456     }
457 #endif
458     /* get this host */
459     gethostname(hostname, sizeof(hostname));
460     th = gethostbyname(hostname);
461     if (!th) {
462         fprintf(stderr, "ptserver: couldn't get address of this host.\n");
463         PT_EXIT(1);
464     }
465     memcpy(&myHost, th->h_addr, sizeof(afs_uint32));
466
467     /* get list of servers */
468     code =
469         afsconf_GetExtendedCellInfo(prdir, NULL, "afsprot", &info, clones);
470     if (code) {
471         afs_com_err(whoami, code, "Couldn't get server list");
472         PT_EXIT(2);
473     }
474
475     /* initialize audit user check */
476     osi_audit_set_user_check(prdir, pr_IsLocalRealmMatch);
477
478     /* initialize ubik */
479     ubik_SetClientSecurityProcs(afsconf_ClientAuth, afsconf_UpToDate, prdir);
480     ubik_SetServerSecurityProcs(afsconf_BuildServerSecurityObjects,
481                                 afsconf_CheckAuth, prdir);
482
483     /* The max needed is when deleting an entry.  A full CoEntry deletion
484      * required removal from 39 entries.  Each of which may refers to the entry
485      * being deleted in one of its CoEntries.  If a CoEntry is freed its
486      * predecessor CoEntry will be modified as well.  Any freed blocks also
487      * modifies the database header.  Counting the entry being deleted and its
488      * CoEntry this adds up to as much as 1+1+39*3 = 119.  If all these entries
489      * and the header are in separate Ubik buffers then 120 buffers may be
490      * required. */
491     ubik_nBuffers = 120 + /*fudge */ 40;
492
493     if (rxBind) {
494         afs_int32 ccode;
495         if (AFSDIR_SERVER_NETRESTRICT_FILEPATH ||
496             AFSDIR_SERVER_NETINFO_FILEPATH) {
497             char reason[1024];
498             ccode = afsconf_ParseNetFiles(SHostAddrs, NULL, NULL,
499                                           ADDRSPERSITE, reason,
500                                           AFSDIR_SERVER_NETINFO_FILEPATH,
501                                           AFSDIR_SERVER_NETRESTRICT_FILEPATH);
502         } else
503         {
504             ccode = rx_getAllAddr(SHostAddrs, ADDRSPERSITE);
505         }
506         if (ccode == 1) {
507             host = SHostAddrs[0];
508             /* the following call is idempotent so if/when it gets called
509              * again by the ubik init stuff, it doesn't really matter
510              * -- klm
511              */
512             rx_InitHost(host, htons(AFSCONF_PROTPORT));
513         }
514     }
515
516     /* Disable jumbograms */
517     rx_SetNoJumbo();
518
519     if (rxMaxMTU != -1) {
520         if (rx_SetMaxMTU(rxMaxMTU) != 0) {
521             printf("rxMaxMTU %d is invalid\n", rxMaxMTU);
522             PT_EXIT(1);
523         }
524     }
525
526     code =
527         ubik_ServerInitByInfo(myHost, htons(AFSCONF_PROTPORT), &info, clones,
528                               pr_dbaseName, &dbase);
529     if (code) {
530         afs_com_err(whoami, code, "Ubik init failed");
531         PT_EXIT(2);
532     }
533
534 #if defined(SUPERGROUPS)
535     pt_hook_write();
536 #endif
537
538     afsconf_BuildServerSecurityObjects(prdir, &securityClasses, &numClasses);
539
540     tservice =
541         rx_NewServiceHost(host, 0, PRSRV, "Protection Server", securityClasses,
542                           numClasses, PR_ExecuteRequest);
543     if (tservice == (struct rx_service *)0) {
544         fprintf(stderr, "ptserver: Could not create new rx service.\n");
545         PT_EXIT(3);
546     }
547     rx_SetMinProcs(tservice, 2);
548     rx_SetMaxProcs(tservice, lwps);
549     if (rxkadDisableDotCheck) {
550         rx_SetSecurityConfiguration(tservice, RXS_CONFIG_FLAGS,
551                                     (void *)RXS_CONFIG_FLAGS_DISABLE_DOTCHECK);
552     }
553
554     tservice =
555         rx_NewServiceHost(host, 0, RX_STATS_SERVICE_ID, "rpcstats",
556                           securityClasses, numClasses, RXSTATS_ExecuteRequest);
557     if (tservice == (struct rx_service *)0) {
558         fprintf(stderr, "ptserver: Could not create new rx service.\n");
559         PT_EXIT(3);
560     }
561     rx_SetMinProcs(tservice, 2);
562     rx_SetMaxProcs(tservice, 4);
563
564     /* allow super users to manage RX statistics */
565     rx_SetRxStatUserOk(pr_rxstat_userok);
566
567     LogCommandLine(argc, argv, "ptserver",
568 #if defined(SUPERGROUPS)
569                    "1.1",
570 #else
571                    "1.0",
572 #endif
573                    "Starting AFS", FSLog);
574     if (afsconf_GetLatestKey(prdir, NULL, NULL) == 0) {
575         LogDesWarning();
576     }
577
578     rx_StartServer(1);
579     osi_audit(PTS_FinishEvent, -1, AUD_END);
580     exit(0);
581 }