16 fprint(2, "usage: reseal [-f] [-b blocksize] [-s ms] arenapart1 [name...]]\n");
21 pwriteblock(uchar *buf, int n, vlong off)
25 for(nr = 0; nr < n; nr += m){
27 m = pwrite(fd, &buf[nr], m, offset0+off+nr);
35 preadblock(uchar *buf, int n, vlong off)
39 for(nr = 0; nr < n; nr += m){
41 m = pread(fd, &buf[nr], m, offset0+off+nr);
52 loadheader(char *name, ArenaHead *head, Arena *arena, vlong off)
54 if(preadblock(data, head->blocksize, off + head->size - head->blocksize) < 0){
55 fprint(2, "%s: reading arena tail: %r\n", name);
59 memset(arena, 0, sizeof *arena);
60 if(unpackarena(arena, data) < 0){
61 fprint(2, "%s: unpack arena tail: %r\n", name);
64 arena->blocksize = head->blocksize;
65 arena->base = off + head->blocksize;
66 arena->clumpmax = arena->blocksize / ClumpInfoSize;
67 arena->size = head->size - 2*head->blocksize;
69 if(arena->diskstats.sealed)
70 scorecp(arena->score, data + head->blocksize - VtScoreSize);
74 uchar zero[VtScoreSize];
77 verify(Arena *arena, void *data, uchar *newscore)
81 uchar score[VtScoreSize];
84 * now we know how much to read
85 * read everything but the last block, which is special
87 e = arena->size + arena->blocksize;
88 o = arena->base - arena->blocksize;
89 bs = arena->blocksize;
90 memset(&ds, 0, sizeof ds);
91 for(n = 0; n < e; n += bs){
92 if(preadblock(data, bs, o + n) < 0){
98 sha1(data, bs, nil, &ds);
102 if(preadblock(data, arena->blocksize, o + e) < 0){
107 sha1(data, bs - VtScoreSize, nil, &ds);
108 sha1(zero, VtScoreSize, score, &ds);
109 if(scorecmp(score, arena->score) != 0){
111 werrstr("score mismatch: %V != %V", score, arena->score);
114 fprint(2, "warning: score mismatch %V != %V\n", score, arena->score);
117 /* prepare new last block */
118 memset(data, 0, arena->blocksize);
119 packarena(arena, data);
120 sha1(data, bs, newscore, &ds1);
121 scorecp((uchar*)data + arena->blocksize - VtScoreSize, newscore);
127 resealarena(char *name, vlong len)
133 uchar newscore[VtScoreSize];
135 fprint(2, "%s: begin reseal\n", name);
137 memset(&s, 0, sizeof s);
139 off = seek(fd, 0, 1);
142 * read a little bit, which will include the header
144 if(preadblock(data, HeadSize, off) < 0){
145 fprint(2, "%s: reading header: %r\n", name);
148 if(unpackarenahead(&head, data) < 0){
149 fprint(2, "%s: corrupt arena header: %r\n", name);
152 if(head.version != ArenaVersion4 && head.version != ArenaVersion5)
153 fprint(2, "%s: warning: unknown arena version %d\n", name, head.version);
154 if(len != 0 && len != head.size)
155 fprint(2, "%s: warning: unexpected length %lld != %lld\n", name, head.size, len);
156 if(strcmp(name, "<stdin>") != 0 && strcmp(head.name, name) != 0)
157 fprint(2, "%s: warning: unexpected name %s\n", name, head.name);
159 if(loadheader(name, &head, &arena, off) < 0)
162 if(!arena.diskstats.sealed){
163 fprint(2, "%s: not sealed\n", name);
167 if(verify(&arena, data, newscore) < 0){
168 fprint(2, "%s: failed to verify before reseal: %r\n", name);
172 if(pwriteblock(data, arena.blocksize, arena.base + arena.size) < 0){
173 fprint(2, "%s: writing new tail: %r\n", name);
176 scorecp(arena.score, newscore);
177 fprint(2, "%s: resealed: %V\n", name, newscore);
179 if(verify(&arena, data, newscore) < 0){
180 fprint(2, "%s: failed to verify after reseal!: %r\n", name);
184 fprint(2, "%s: verified: %V\n", name, newscore);
188 shouldcheck(char *name, char **s, int n)
196 if(s[i] && strcmp(name, s[i]) == 0){
205 readap(ArenaPart *ap)
209 if(preadblock(data, 8192, PartBlank) < 0)
210 sysfatal("read arena part header: %r");
211 if(unpackarenapart(ap, data) < 0)
212 sysfatal("corrupted arena part header: %r");
213 fprint(2, "# arena part version=%d blocksize=%d arenabase=%d\n",
214 ap->version, ap->blocksize, ap->arenabase);
215 ap->tabbase = (PartBlank+HeadSize+ap->blocksize-1)&~(ap->blocksize-1);
216 ap->tabsize = ap->arenabase - ap->tabbase;
217 table = malloc(ap->tabsize+1);
218 if(preadblock((uchar*)table, ap->tabsize, ap->tabbase) < 0)
219 sysfatal("reading arena part directory: %r");
220 table[ap->tabsize] = 0;
225 threadmain(int argc, char *argv[])
228 char *p, *q, *table, *f[10], line[256];
234 blocksize = MaxIoSize;
237 blocksize = unittoull(EARGF(usage()));
243 sleepms = atoi(EARGF(usage()));
253 data = vtmalloc(blocksize);
254 if((part = initpart(argv[0], ORDWR)) == nil)
255 sysfatal("open partition %s: %r", argv[0]);
257 offset0 = part->offset;
262 p = strchr(table, '\n');
265 for(i=0; i<nline; i++){
267 fprint(2, "warning: unexpected arena table end\n");
273 if(strlen(p) >= sizeof line){
274 fprint(2, "warning: long arena table line: %s\n", p);
279 memset(f, 0, sizeof f);
280 if(tokenize(line, f, nelem(f)) < 3){
281 fprint(2, "warning: bad arena table line: %s\n", p);
286 if(shouldcheck(f[0], argv+1, argc-1)){
287 start = strtoull(f[1], 0, 0);
288 stop = strtoull(f[2], 0, 0);
290 fprint(2, "%s: bad start,stop %lld,%lld\n", f[0], stop, start);
293 if(seek(fd, start, 0) < 0)
294 fprint(2, "%s: seek to start: %r\n", f[0]);
295 resealarena(f[0], stop - start);
298 for(i=2; i<argc; i++)
300 fprint(2, "%s: did not find arena\n", argv[i]);