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