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