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