1 a0d146ed 2005-07-12 devnull #include "stdinc.h"
2 a0d146ed 2005-07-12 devnull #include "dat.h"
3 a0d146ed 2005-07-12 devnull #include "fns.h"
5 a0d146ed 2005-07-12 devnull static int verbose;
8 a0d146ed 2005-07-12 devnull usage(void)
10 a0d146ed 2005-07-12 devnull fprint(2, "usage: verifyarena [-v]\n");
11 a0d146ed 2005-07-12 devnull threadexitsall(0);
14 a0d146ed 2005-07-12 devnull static void
15 a0d146ed 2005-07-12 devnull readblock(uchar *buf, int n)
17 a0d146ed 2005-07-12 devnull int nr, m;
19 a0d146ed 2005-07-12 devnull for(nr = 0; nr < n; nr += m){
20 a0d146ed 2005-07-12 devnull m = n - nr;
21 a0d146ed 2005-07-12 devnull m = read(0, &buf[nr], m);
22 a0d146ed 2005-07-12 devnull if(m <= 0)
23 a0d146ed 2005-07-12 devnull sysfatal("can't read arena from standard input: %r");
27 a0d146ed 2005-07-12 devnull static void
28 a0d146ed 2005-07-12 devnull verifyarena(void)
30 a0d146ed 2005-07-12 devnull Arena arena;
31 a0d146ed 2005-07-12 devnull ArenaHead head;
32 a0d146ed 2005-07-12 devnull ZBlock *b;
33 a0d146ed 2005-07-12 devnull DigestState s;
34 a0d146ed 2005-07-12 devnull u64int n, e;
35 a0d146ed 2005-07-12 devnull u32int bs;
36 a0d146ed 2005-07-12 devnull u8int score[VtScoreSize];
38 a0d146ed 2005-07-12 devnull fprint(2, "verify arena from standard input\n");
40 a0d146ed 2005-07-12 devnull memset(&arena, 0, sizeof arena);
41 a0d146ed 2005-07-12 devnull memset(&s, 0, sizeof s);
44 a0d146ed 2005-07-12 devnull * read the little bit, which will included the header
46 a0d146ed 2005-07-12 devnull bs = MaxIoSize;
47 a0d146ed 2005-07-12 devnull b = alloczblock(bs, 0, 0);
48 a0d146ed 2005-07-12 devnull readblock(b->data, HeadSize);
49 a0d146ed 2005-07-12 devnull sha1(b->data, HeadSize, nil, &s);
50 a0d146ed 2005-07-12 devnull if(unpackarenahead(&head, b->data) < 0)
51 a0d146ed 2005-07-12 devnull sysfatal("corrupted arena header: %r");
52 a0d146ed 2005-07-12 devnull if(head.version != ArenaVersion4 && head.version != ArenaVersion5)
53 a0d146ed 2005-07-12 devnull fprint(2, "warning: unknown arena version %d\n", head.version);
56 a0d146ed 2005-07-12 devnull * now we know how much to read
57 a0d146ed 2005-07-12 devnull * read everything but the last block, which is special
59 a0d146ed 2005-07-12 devnull e = head.size - head.blocksize;
60 a0d146ed 2005-07-12 devnull for(n = HeadSize; n < e; n += bs){
61 a0d146ed 2005-07-12 devnull if(n + bs > e)
62 a0d146ed 2005-07-12 devnull bs = e - n;
63 a0d146ed 2005-07-12 devnull readblock(b->data, bs);
64 a0d146ed 2005-07-12 devnull sha1(b->data, bs, nil, &s);
68 a0d146ed 2005-07-12 devnull * read the last block update the sum.
69 a0d146ed 2005-07-12 devnull * the sum is calculated assuming the slot for the sum is zero.
71 a0d146ed 2005-07-12 devnull bs = head.blocksize;
72 a0d146ed 2005-07-12 devnull readblock(b->data, bs);
73 a0d146ed 2005-07-12 devnull sha1(b->data, bs-VtScoreSize, nil, &s);
74 a0d146ed 2005-07-12 devnull sha1(zeroscore, VtScoreSize, nil, &s);
75 a0d146ed 2005-07-12 devnull sha1(nil, 0, score, &s);
78 a0d146ed 2005-07-12 devnull * validity check on the trailer
80 a0d146ed 2005-07-12 devnull arena.blocksize = head.blocksize;
81 a0d146ed 2005-07-12 devnull if(unpackarena(&arena, b->data) < 0)
82 a0d146ed 2005-07-12 devnull sysfatal("corrupted arena trailer: %r");
83 a0d146ed 2005-07-12 devnull scorecp(arena.score, &b->data[arena.blocksize - VtScoreSize]);
85 a0d146ed 2005-07-12 devnull if(namecmp(arena.name, head.name) != 0)
86 a0d146ed 2005-07-12 devnull sysfatal("arena header and trailer names clash: %s vs. %s\n", head.name, arena.name);
87 a0d146ed 2005-07-12 devnull if(arena.version != head.version)
88 a0d146ed 2005-07-12 devnull sysfatal("arena header and trailer versions clash: %d vs. %d\n", head.version, arena.version);
89 a0d146ed 2005-07-12 devnull arena.size = head.size - 2 * head.blocksize;
92 a0d146ed 2005-07-12 devnull * check for no checksum or the same
94 a0d146ed 2005-07-12 devnull if(scorecmp(score, arena.score) != 0){
95 a0d146ed 2005-07-12 devnull if(scorecmp(zeroscore, arena.score) != 0)
96 a0d146ed 2005-07-12 devnull fprint(2, "warning: mismatched checksums for arena=%s, found=%V calculated=%V",
97 a0d146ed 2005-07-12 devnull arena.name, arena.score, score);
98 a0d146ed 2005-07-12 devnull scorecp(arena.score, score);
100 a0d146ed 2005-07-12 devnull fprint(2, "matched score\n");
102 a0d146ed 2005-07-12 devnull printarena(2, &arena);
106 a0d146ed 2005-07-12 devnull threadmain(int argc, char *argv[])
108 a0d146ed 2005-07-12 devnull ventifmtinstall();
109 a0d146ed 2005-07-12 devnull statsinit();
111 a0d146ed 2005-07-12 devnull ARGBEGIN{
112 a0d146ed 2005-07-12 devnull case 'v':
113 a0d146ed 2005-07-12 devnull verbose++;
115 a0d146ed 2005-07-12 devnull default:
116 a0d146ed 2005-07-12 devnull usage();
120 a0d146ed 2005-07-12 devnull readonly = 1;
122 a0d146ed 2005-07-12 devnull if(argc != 0)
123 a0d146ed 2005-07-12 devnull usage();
125 a0d146ed 2005-07-12 devnull verifyarena();
126 a0d146ed 2005-07-12 devnull threadexitsall(0);
