pt_util: fix group line check for input files
[openafs.git] / src / ptserver / pt_util.c
1 /* $Id$ */
2
3 /*
4  *
5  * pt_util: Program to dump the AFS protection server database
6  *         into an ascii file.
7  *
8  *      Assumptions: We *cheat* here and read the datafile directly, ie.
9  *                   not going through the ubik distributed data manager.
10  *                   therefore the database must be quiescent for the
11  *                   output of this program to be valid.
12  */
13
14 #include <afsconfig.h>
15 #include <afs/param.h>
16
17 #include <roken.h>
18
19 #ifndef _WIN32
20 #include <sys/file.h>
21 #else
22 #define L_SET SEEK_SET
23 #endif
24 #include <ctype.h>
25
26 #include <afs/com_err.h>
27 #include <afs/cmd.h>            /*Command line parsing */
28 #include <afs/afsutil.h>
29 #include <lock.h>
30 #define UBIK_INTERNALS
31 #include <ubik.h>
32 #include <rx/xdr.h>
33 #include <rx/rx.h>
34
35 #include "ptint.h"
36 #include "ptserver.h"
37 #include "pterror.h"
38 #include "ptprototypes.h"
39
40 #define IDHash(x) (abs(x) % HASHSIZE)
41 #define print_id(x) ( ((flags&DO_SYS)==0 && (x<-32767 || x>97536)) || \
42                       ((flags&DO_OTR)==0 && (x>-32768 && x<97537)))
43
44 extern char *optarg;
45 extern int optind;
46
47 int restricted = 0;
48
49 static int display_entry(int);
50 static void add_group(long);
51 static void display_groups(void);
52 static void display_group(int);
53 static void fix_pre(struct prentry *);
54 static char *id_to_name(int);
55 static char *checkin(struct prentry *);
56 static char *check_core(int);
57 static int CommandProc(struct cmd_syndesc *, void *);
58
59 struct hash_entry {
60     char h_name[PR_MAXNAMELEN];
61     int h_id;
62     struct hash_entry *next;
63 };
64 struct hash_entry *hat[HASHSIZE];
65
66 static struct contentry prco;
67 static struct prentry pre;
68 static struct prheader prh;
69 static struct ubik_version uv;
70
71 struct grp_list {
72     struct grp_list *next;
73     long groups[1024];
74 };
75 static struct grp_list *grp_head = 0;
76 static long grp_count = 0;
77
78 struct usr_list {
79     struct usr_list *next;
80     char name[PR_MAXNAMELEN];
81     long uid;
82 };
83 static struct usr_list *usr_head = 0;
84
85 char buffer[1024];
86 int dbase_fd;
87 FILE *dfp;
88
89 #define FMT_BASE "%-10s %d/%d %d %d %d\n"
90 #define FMT_MEM  "   %-8s %d\n"
91
92 #define DO_USR 1
93 #define DO_GRP 2
94 #define DO_MEM 4
95 #define DO_SYS 8
96 #define DO_OTR 16
97
98 int nflag = 0;
99 int wflag = 0;
100 int flags = 0;
101
102 int
103 main(int argc, char **argv)
104 {
105
106     struct cmd_syndesc *cs;     /*Command line syntax descriptor */
107     afs_int32 code;     /*Return code */
108
109     cs = cmd_CreateSyntax(NULL, CommandProc, NULL,
110                           "access protection database");
111     cmd_AddParm(cs, "-w", CMD_FLAG, CMD_OPTIONAL,
112                 "update prdb with contents of data file");
113     cmd_AddParm(cs, "-user", CMD_FLAG, CMD_OPTIONAL, "display users");
114     cmd_AddParm(cs, "-group", CMD_FLAG, CMD_OPTIONAL, "display groups");
115     cmd_AddParm(cs, "-members", CMD_FLAG, CMD_OPTIONAL,
116                 "display group members");
117     cmd_AddParm(cs, "-name", CMD_FLAG, CMD_OPTIONAL,
118                 "follow name hash chains (not id hashes)");
119     cmd_AddParm(cs, "-system", CMD_FLAG, CMD_OPTIONAL,
120                 "display only system data");
121     cmd_AddParm(cs, "-xtra", CMD_FLAG, CMD_OPTIONAL,
122                 "display extra users/groups");
123     cmd_AddParm(cs, "-prdb", CMD_SINGLE, CMD_OPTIONAL, "prdb file");
124     cmd_AddParm(cs, "-datafile", CMD_SINGLE, CMD_OPTIONAL, "data file");
125     code = cmd_Dispatch(argc, argv);
126
127     exit(code);
128
129 }
130
131 static int
132 CommandProc(struct cmd_syndesc *a_as, void *arock)
133 {
134     int i;
135     long code = 0;
136     long upos;
137     long gpos = 0;
138     struct prentry uentry, gentry;
139     struct ubik_hdr *uh;
140     char *dfile = 0;
141     const char *pbase = AFSDIR_SERVER_PRDB_FILEPATH;
142     char *pfile = NULL;
143     char pbuffer[1028];
144     struct cmd_parmdesc *tparm;
145
146     tparm = a_as->parms;
147
148     if (tparm[0].items) {
149         wflag++;
150     }
151     if (tparm[1].items) {
152         flags |= DO_USR;
153     }
154     if (tparm[2].items) {
155         flags |= DO_GRP;
156     }
157     if (tparm[3].items) {
158         flags |= (DO_GRP | DO_MEM);
159     }
160     if (tparm[4].items) {
161         nflag++;
162     }
163     if (tparm[5].items) {
164         flags |= DO_SYS;
165     }
166     if (tparm[6].items) {
167         flags |= DO_OTR;
168     }
169     if (tparm[7].items) {
170         pfile = tparm[7].items->data;
171     }
172     if (tparm[8].items) {
173         dfile = tparm[8].items->data;
174     }
175
176     if (pfile == NULL) {
177         snprintf(pbuffer, sizeof(pbuffer), "%s.DB0", pbase);
178         pfile = pbuffer;
179     }
180     if ((dbase_fd = open(pfile, (wflag ? O_RDWR : O_RDONLY) | O_CREAT, 0600))
181         < 0) {
182         fprintf(stderr, "pt_util: cannot open %s: %s\n", pfile,
183                 strerror(errno));
184         exit(1);
185     }
186     if (read(dbase_fd, buffer, HDRSIZE) < 0) {
187         fprintf(stderr, "pt_util: error reading %s: %s\n", pfile,
188                 strerror(errno));
189         exit(1);
190     }
191
192     if (dfile) {
193         if ((dfp = fopen(dfile, wflag ? "r" : "w")) == 0) {
194             fprintf(stderr, "pt_util: error opening %s: %s\n", dfile,
195                     strerror(errno));
196             exit(1);
197         }
198     } else
199         dfp = (wflag ? stdin : stdout);
200
201     uh = (struct ubik_hdr *)buffer;
202     if (ntohl(uh->magic) != UBIK_MAGIC)
203         fprintf(stderr, "pt_util: %s: Bad UBIK_MAGIC. Is %x should be %x\n",
204                 pfile, ntohl(uh->magic), UBIK_MAGIC);
205     memcpy(&uv, &uh->version, sizeof(struct ubik_version));
206
207     if (wflag && ntohl(uv.epoch) == 0 && ntohl(uv.counter) == 0) {
208         uv.epoch = htonl(2); /* a ubik version of 0 or 1 has special meaning */
209         memcpy(&uh->version, &uv, sizeof(struct ubik_version));
210         lseek(dbase_fd, 0, SEEK_SET);
211         if (write(dbase_fd, buffer, HDRSIZE) < 0) {
212             fprintf(stderr, "pt_util: error writing ubik version to %s: %s\n",
213                     pfile, strerror(errno));
214             exit(1);
215         }
216     }
217
218     /* Now that any writeback is done, swap these */
219     uv.epoch = ntohl(uv.epoch);
220     uv.counter = ntohl(uv.counter);
221
222     fprintf(stderr, "Ubik Version is: %d.%d\n", uv.epoch, uv.counter);
223     if (read(dbase_fd, &prh, sizeof(struct prheader)) < 0) {
224         fprintf(stderr, "pt_util: error reading %s: %s\n", pfile,
225                 strerror(errno));
226         exit(1);
227     }
228
229     Initdb();
230     initialize_PT_error_table();
231
232     if (wflag) {
233         struct usr_list *u;
234         int seenGroup = 0;
235
236         while (fgets(buffer, sizeof(buffer), dfp)) {
237             int id, oid, cid, flags, quota, uid;
238             char name[PR_MAXNAMELEN], mem[PR_MAXNAMELEN];
239
240             if (isspace(*buffer)) {
241                 code = sscanf(buffer, "%s %d", mem, &uid);
242                 if (code != 2) {
243                     fprintf(stderr,
244                             "Insuffient data provided for group membership\n");
245                     exit(1);
246                 }
247
248                 if (!seenGroup) {
249                     fprintf(stderr,
250                             "Group member %s listed outside of group\n",
251                             mem);
252                     exit(1);
253                 }
254
255                 for (u = usr_head; u; u = u->next)
256                     if (u->uid && u->uid == uid)
257                         break;
258                 if (u) {
259                     /* Add user - deferred because it is probably foreign */
260                     u->uid = 0;
261                     if (FindByID(0, uid))
262                         code = PRIDEXIST;
263                     else {
264                         if (!code
265                             && (flags & (PRGRP | PRQUOTA)) ==
266                             (PRGRP | PRQUOTA)) {
267                             gentry.ngroups++;
268                             code = pr_WriteEntry(0, 0, gpos, &gentry);
269                             if (code)
270                                 fprintf(stderr,
271                                         "Error setting group count on %s: %s\n",
272                                         name, afs_error_message(code));
273                         }
274                         code = CreateEntry(0, u->name, &uid, 1 /*idflag */ ,
275                                            1 /*gflag */ ,
276                                            SYSADMINID /*oid */ ,
277                                            SYSADMINID /*cid */ );
278                     }
279                     if (code)
280                         fprintf(stderr, "Error while creating %s: %s\n",
281                                 u->name, afs_error_message(code));
282                     continue;
283                 }
284                 /* Add user to group */
285                 if (id == ANYUSERID || id == AUTHUSERID || uid == ANONYMOUSID) {
286                     code = PRPERM;
287                 } else if ((upos = FindByID(0, uid))
288                            && (gpos = FindByID(0, id))) {
289                     code = pr_ReadEntry(0, 0, upos, &uentry);
290                     if (!code)
291                         code = pr_ReadEntry(0, 0, gpos, &gentry);
292                     if (!code)
293                         code = AddToEntry(0, &gentry, gpos, uid);
294                     if (!code)
295                         code = AddToEntry(0, &uentry, upos, id);
296                 } else
297                     code = PRNOENT;
298
299                 if (code)
300                     fprintf(stderr, "Error while adding %s to %s: %s\n", mem,
301                             name, afs_error_message(code));
302             } else {
303                 code = sscanf(buffer, "%s %d/%d %d %d %d", name, &flags, &quota, &id,
304                               &oid, &cid);
305                 if (code != 6) {
306                     fprintf(stderr,
307                             "Insufficient data provided for user/group\n");
308                     exit(1);
309                 }
310
311                 seenGroup = 1;
312
313                 if (FindByID(0, id))
314                     code = PRIDEXIST;
315                 else
316                     code = CreateEntry(0, name, &id, 1 /*idflag */ ,
317                                        flags & PRGRP, oid, cid);
318                 if (code == PRBADNAM) {
319                     u = malloc(sizeof(struct usr_list));
320                     u->next = usr_head;
321                     u->uid = id;
322                     strcpy(u->name, name);
323                     usr_head = u;
324                 } else if (code) {
325                     fprintf(stderr, "Error while creating %s: %s\n", name,
326                             afs_error_message(code));
327                 } else if ((flags & PRACCESS)
328                            || (flags & (PRGRP | PRQUOTA)) ==
329                            (PRGRP | PRQUOTA)) {
330                     gpos = FindByID(0, id);
331                     code = pr_ReadEntry(0, 0, gpos, &gentry);
332                     if (!code) {
333                         gentry.flags = flags;
334                         gentry.ngroups = quota;
335                         code = pr_WriteEntry(0, 0, gpos, &gentry);
336                     }
337                     if (code)
338                         fprintf(stderr,
339                                 "Error while setting flags on %s: %s\n", name,
340                                 afs_error_message(code));
341                 }
342             }
343         }
344         for (u = usr_head; u; u = u->next)
345             if (u->uid)
346                 fprintf(stderr, "Error while creating %s: %s\n", u->name,
347                         afs_error_message(PRBADNAM));
348     } else {
349         for (i = 0; i < HASHSIZE; i++) {
350             upos = nflag ? ntohl(prh.nameHash[i]) : ntohl(prh.idHash[i]);
351             while (upos) {
352                 long newpos;
353                 newpos = display_entry(upos);
354                 if (newpos == upos) {
355                     fprintf(stderr, "pt_util: hash error in %s chain %d\n",
356                             nflag ? "name":"id", i);
357                     exit(1);
358                 } else
359                     upos = newpos;
360             }
361         }
362         if (flags & DO_GRP)
363             display_groups();
364     }
365
366     lseek(dbase_fd, 0, L_SET);  /* rewind to beginning of file */
367     if (read(dbase_fd, buffer, HDRSIZE) < 0) {
368         fprintf(stderr, "pt_util: error reading %s: %s\n", pfile,
369                 strerror(errno));
370         exit(1);
371     }
372     uh = (struct ubik_hdr *)buffer;
373
374     uh->version.epoch = ntohl(uh->version.epoch);
375     uh->version.counter = ntohl(uh->version.counter);
376
377     if ((uh->version.epoch != uv.epoch)
378         || (uh->version.counter != uv.counter)) {
379         fprintf(stderr,
380                 "pt_util: Ubik Version number changed during execution.\n");
381         fprintf(stderr, "Old Version = %d.%d, new version = %d.%d\n",
382                 uv.epoch, uv.counter, uh->version.epoch, uh->version.counter);
383     }
384     close(dbase_fd);
385     exit(0);
386 }
387
388 static int
389 display_entry(int offset)
390 {
391     lseek(dbase_fd, offset + HDRSIZE, L_SET);
392     read(dbase_fd, &pre, sizeof(struct prentry));
393
394     fix_pre(&pre);
395
396     if ((pre.flags & PRFREE) == 0) {
397         if (pre.flags & PRGRP) {
398             if (flags & DO_GRP)
399                 add_group(pre.id);
400         } else {
401             if (print_id(pre.id) && (flags & DO_USR))
402                 fprintf(dfp, FMT_BASE, pre.name, pre.flags, pre.ngroups,
403                         pre.id, pre.owner, pre.creator);
404             checkin(&pre);
405         }
406     }
407     return (nflag ? pre.nextName : pre.nextID);
408 }
409
410 static void
411 add_group(long id)
412 {
413     struct grp_list *g;
414     long i;
415
416     i = grp_count++ % 1024;
417     if (i == 0) {
418         g = malloc(sizeof(struct grp_list));
419         g->next = grp_head;
420         grp_head = g;
421     }
422     g = grp_head;
423     g->groups[i] = id;
424 }
425
426 static void
427 display_groups(void)
428 {
429     int i, id;
430     struct grp_list *g;
431
432     g = grp_head;
433     while (grp_count--) {
434         i = grp_count % 1024;
435         id = g->groups[i];
436         display_group(id);
437         if (i == 0) {
438             grp_head = g->next;
439             free(g);
440             g = grp_head;
441         }
442     }
443 }
444
445 static void
446 display_group(int id)
447 {
448     int i, offset;
449     int print_grp = 0;
450
451     offset = ntohl(prh.idHash[IDHash(id)]);
452     while (offset) {
453         lseek(dbase_fd, offset + HDRSIZE, L_SET);
454         if (read(dbase_fd, &pre, sizeof(struct prentry)) < 0) {
455             fprintf(stderr, "pt_util: read i/o error: %s\n", strerror(errno));
456             exit(1);
457         }
458         fix_pre(&pre);
459         if (pre.id == id)
460             break;
461         offset = pre.nextID;
462     }
463
464     if (print_id(id)) {
465         fprintf(dfp, FMT_BASE, pre.name, pre.flags, pre.ngroups, pre.id,
466                 pre.owner, pre.creator);
467         print_grp = 1;
468     }
469
470     if ((flags & DO_MEM) == 0)
471         return;
472
473     for (i = 0; i < PRSIZE; i++) {
474         if ((id = pre.entries[i]) == 0)
475             break;
476         if (id == PRBADID)
477             continue;
478         if (print_id(id) || print_grp == 1) {
479             if (print_grp == 0) {
480                 fprintf(dfp, FMT_BASE, pre.name, pre.flags, pre.ngroups,
481                         pre.id, pre.owner, pre.creator);
482                 print_grp = 2;
483             }
484             fprintf(dfp, FMT_MEM, id_to_name(id), id);
485         }
486     }
487     if (i == PRSIZE) {
488         offset = pre.next;
489         while (offset) {
490             lseek(dbase_fd, offset + HDRSIZE, L_SET);
491             read(dbase_fd, &prco, sizeof(struct contentry));
492             prco.next = ntohl(prco.next);
493             for (i = 0; i < COSIZE; i++) {
494                 prco.entries[i] = ntohl(prco.entries[i]);
495                 if ((id = prco.entries[i]) == 0)
496                     break;
497                 if (id == PRBADID)
498                     continue;
499                 if (print_id(id) || print_grp == 1) {
500                     if (print_grp == 0) {
501                         fprintf(dfp, FMT_BASE, pre.name, pre.flags,
502                                 pre.ngroups, pre.id, pre.owner, pre.creator);
503                         print_grp = 2;
504                     }
505                     fprintf(dfp, FMT_MEM, id_to_name(id), id);
506                 }
507             }
508             if ((i == COSIZE) && prco.next)
509                 offset = prco.next;
510             else
511                 offset = 0;
512         }
513     }
514 }
515
516 static void
517 fix_pre(struct prentry *pre)
518 {
519     int i;
520
521     pre->flags = ntohl(pre->flags);
522     pre->id = ntohl(pre->id);
523     pre->cellid = ntohl(pre->cellid);
524     pre->next = ntohl(pre->next);
525     pre->nextID = ntohl(pre->nextID);
526     pre->nextName = ntohl(pre->nextName);
527     pre->owner = ntohl(pre->owner);
528     pre->creator = ntohl(pre->creator);
529     pre->ngroups = ntohl(pre->ngroups);
530     pre->nusers = ntohl(pre->nusers);
531     pre->count = ntohl(pre->count);
532     pre->instance = ntohl(pre->instance);
533     pre->owned = ntohl(pre->owned);
534     pre->nextOwned = ntohl(pre->nextOwned);
535     pre->parent = ntohl(pre->parent);
536     pre->sibling = ntohl(pre->sibling);
537     pre->child = ntohl(pre->child);
538     for (i = 0; i < PRSIZE; i++) {
539         pre->entries[i] = ntohl(pre->entries[i]);
540     }
541 }
542
543 static char *
544 id_to_name(int id)
545 {
546     int offset;
547     static struct prentry pre;
548     char *name;
549
550     name = check_core(id);
551     if (name)
552         return (name);
553     offset = ntohl(prh.idHash[IDHash(id)]);
554     while (offset) {
555         lseek(dbase_fd, offset + HDRSIZE, L_SET);
556         if (read(dbase_fd, &pre, sizeof(struct prentry)) < 0) {
557             fprintf(stderr, "pt_util: read i/o error: %s\n", strerror(errno));
558             exit(1);
559         }
560         pre.id = ntohl(pre.id);
561         if (pre.id == id) {
562             name = checkin(&pre);
563             return (name);
564         }
565         offset = ntohl(pre.nextID);
566     }
567     return 0;
568 }
569
570 static char *
571 checkin(struct prentry *pre)
572 {
573     struct hash_entry *he, *last;
574     int id;
575
576     id = pre->id;
577     last = (struct hash_entry *)0;
578     he = hat[IDHash(id)];
579     while (he) {
580         if (id == he->h_id)
581             return (he->h_name);
582         last = he;
583         he = he->next;
584     }
585     he = malloc(sizeof(struct hash_entry));
586     if (he == 0) {
587         fprintf(stderr, "pt_util: No Memory for internal hash table.\n");
588         exit(1);
589     }
590     he->h_id = id;
591     he->next = (struct hash_entry *)0;
592     strncpy(he->h_name, pre->name, PR_MAXNAMELEN);
593     if (last == (struct hash_entry *)0)
594         hat[IDHash(id)] = he;
595     else
596         last->next = he;
597     return (he->h_name);
598 }
599
600 static char *
601 check_core(int id)
602 {
603     struct hash_entry *he;
604     he = hat[IDHash(id)];
605     while (he) {
606         if (id == he->h_id)
607             return (he->h_name);
608         he = he->next;
609     }
610     return 0;
611 }