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