Blame


1 7763a61a 2003-11-23 devnull #include "stdinc.h"
2 7763a61a 2003-11-23 devnull #include <bio.h>
3 7763a61a 2003-11-23 devnull
4 7763a61a 2003-11-23 devnull typedef struct Source Source;
5 7763a61a 2003-11-23 devnull
6 7763a61a 2003-11-23 devnull struct Source
7 7763a61a 2003-11-23 devnull {
8 7763a61a 2003-11-23 devnull ulong gen;
9 7763a61a 2003-11-23 devnull int psize;
10 7763a61a 2003-11-23 devnull int dsize;
11 7763a61a 2003-11-23 devnull int dir;
12 7763a61a 2003-11-23 devnull int active;
13 7763a61a 2003-11-23 devnull int depth;
14 7763a61a 2003-11-23 devnull uvlong size;
15 7763a61a 2003-11-23 devnull uchar score[VtScoreSize];
16 7763a61a 2003-11-23 devnull int reserved;
17 7763a61a 2003-11-23 devnull };
18 7763a61a 2003-11-23 devnull
19 7763a61a 2003-11-23 devnull int bsize;
20 7763a61a 2003-11-23 devnull Biobuf *bout;
21 7763a61a 2003-11-23 devnull VtRoot root;
22 7763a61a 2003-11-23 devnull int ver;
23 7763a61a 2003-11-23 devnull int cmp;
24 7763a61a 2003-11-23 devnull int all;
25 7763a61a 2003-11-23 devnull int find;
26 7763a61a 2003-11-23 devnull uchar fscore[VtScoreSize];
27 3d77c87e 2004-03-15 devnull VtConn *z;
28 7763a61a 2003-11-23 devnull
29 3d77c87e 2004-03-15 devnull int vtgetuint16(uchar *p);
30 3d77c87e 2004-03-15 devnull ulong vtgetuint32(uchar *p);
31 3d77c87e 2004-03-15 devnull uvlong vtgetuint48(uchar *p);
32 7763a61a 2003-11-23 devnull void usage(void);
33 3d77c87e 2004-03-15 devnull void readroot(VtRoot*, uchar *score, char *file);
34 3d77c87e 2004-03-15 devnull int dumpdir(Source*, int indent);
35 7763a61a 2003-11-23 devnull
36 7763a61a 2003-11-23 devnull void
37 3d77c87e 2004-03-15 devnull threadmain(int argc, char *argv[])
38 7763a61a 2003-11-23 devnull {
39 3d77c87e 2004-03-15 devnull char *host = nil, *pref;
40 7763a61a 2003-11-23 devnull uchar score[VtScoreSize];
41 7763a61a 2003-11-23 devnull Source source;
42 7763a61a 2003-11-23 devnull uchar buf[VtMaxLumpSize];
43 7763a61a 2003-11-23 devnull char *p;
44 7763a61a 2003-11-23 devnull int n;
45 7763a61a 2003-11-23 devnull
46 7763a61a 2003-11-23 devnull ARGBEGIN{
47 7763a61a 2003-11-23 devnull case 'h':
48 7763a61a 2003-11-23 devnull host = ARGF();
49 7763a61a 2003-11-23 devnull break;
50 7763a61a 2003-11-23 devnull case 'c':
51 7763a61a 2003-11-23 devnull cmp++;
52 7763a61a 2003-11-23 devnull break;
53 7763a61a 2003-11-23 devnull case 'f':
54 7763a61a 2003-11-23 devnull find++;
55 7763a61a 2003-11-23 devnull p = ARGF();
56 3d77c87e 2004-03-15 devnull if(p == nil || vtparsescore(p, &pref, fscore) < 0 || !pref || strcmp(pref, "vac") != 0)
57 7763a61a 2003-11-23 devnull usage();
58 7763a61a 2003-11-23 devnull break;
59 7763a61a 2003-11-23 devnull case 'a':
60 7763a61a 2003-11-23 devnull all = 1;
61 7763a61a 2003-11-23 devnull break;
62 7763a61a 2003-11-23 devnull }ARGEND
63 7763a61a 2003-11-23 devnull
64 3d77c87e 2004-03-15 devnull bout = vtmallocz(sizeof(Biobuf));
65 7763a61a 2003-11-23 devnull Binit(bout, 1, OWRITE);
66 7763a61a 2003-11-23 devnull
67 7763a61a 2003-11-23 devnull if(argc > 1)
68 7763a61a 2003-11-23 devnull usage();
69 7763a61a 2003-11-23 devnull
70 3d77c87e 2004-03-15 devnull fmtinstall('V', vtscorefmt);
71 3d77c87e 2004-03-15 devnull fmtinstall('H', encodefmt);
72 7763a61a 2003-11-23 devnull
73 3d77c87e 2004-03-15 devnull z = vtdial(host);
74 7763a61a 2003-11-23 devnull if(z == nil)
75 3d77c87e 2004-03-15 devnull sysfatal("could not connect to server: %r");
76 7763a61a 2003-11-23 devnull
77 3d77c87e 2004-03-15 devnull if(vtconnect(z) < 0)
78 3d77c87e 2004-03-15 devnull sysfatal("vtconnect: %r");
79 7763a61a 2003-11-23 devnull
80 3d77c87e 2004-03-15 devnull readroot(&root, score, argv[0]);
81 3d77c87e 2004-03-15 devnull bsize = root.blocksize;
82 7763a61a 2003-11-23 devnull if(!find) {
83 7763a61a 2003-11-23 devnull Bprint(bout, "score: %V\n", score);
84 7763a61a 2003-11-23 devnull Bprint(bout, "name: %s\n", root.name);
85 7763a61a 2003-11-23 devnull Bprint(bout, "type: %s\n", root.type);
86 7763a61a 2003-11-23 devnull Bprint(bout, "bsize: %d\n", bsize);
87 7763a61a 2003-11-23 devnull Bprint(bout, "prev: %V\n", root.prev);
88 7763a61a 2003-11-23 devnull }
89 7763a61a 2003-11-23 devnull
90 3d77c87e 2004-03-15 devnull n = vtread(z, root.score, VtDirType, buf, bsize);
91 7763a61a 2003-11-23 devnull if(n < 0)
92 7763a61a 2003-11-23 devnull sysfatal("could not read root dir");
93 7763a61a 2003-11-23 devnull
94 7763a61a 2003-11-23 devnull /* fake up top level source */
95 7763a61a 2003-11-23 devnull memset(&source, 0, sizeof(source));
96 7763a61a 2003-11-23 devnull memmove(source.score, root.score, VtScoreSize);
97 7763a61a 2003-11-23 devnull source.psize = bsize;
98 7763a61a 2003-11-23 devnull source.dsize = bsize;
99 7763a61a 2003-11-23 devnull source.dir = 1;
100 7763a61a 2003-11-23 devnull source.active = 1;
101 7763a61a 2003-11-23 devnull source.depth = 0;
102 7763a61a 2003-11-23 devnull source.size = n;
103 7763a61a 2003-11-23 devnull
104 3d77c87e 2004-03-15 devnull dumpdir(&source, 0);
105 7763a61a 2003-11-23 devnull
106 7763a61a 2003-11-23 devnull Bterm(bout);
107 7763a61a 2003-11-23 devnull
108 3d77c87e 2004-03-15 devnull vthangup(z);
109 3d77c87e 2004-03-15 devnull threadexitsall(0);
110 7763a61a 2003-11-23 devnull }
111 7763a61a 2003-11-23 devnull
112 7763a61a 2003-11-23 devnull void
113 3d77c87e 2004-03-15 devnull sourceprint(Source *s, int indent, int entry)
114 7763a61a 2003-11-23 devnull {
115 7763a61a 2003-11-23 devnull int i;
116 7763a61a 2003-11-23 devnull uvlong size;
117 7763a61a 2003-11-23 devnull int ne;
118 7763a61a 2003-11-23 devnull
119 7763a61a 2003-11-23 devnull for(i=0; i<indent; i++)
120 7763a61a 2003-11-23 devnull Bprint(bout, " ");
121 7763a61a 2003-11-23 devnull Bprint(bout, "%4d", entry);
122 7763a61a 2003-11-23 devnull if(s->active) {
123 7763a61a 2003-11-23 devnull /* dir size in directory entries */
124 7763a61a 2003-11-23 devnull if(s->dir) {
125 7763a61a 2003-11-23 devnull ne = s->dsize/VtEntrySize;
126 7763a61a 2003-11-23 devnull size = ne*(s->size/s->dsize) + (s->size%s->dsize)/VtEntrySize;
127 7763a61a 2003-11-23 devnull } else
128 7763a61a 2003-11-23 devnull size = s->size;
129 7763a61a 2003-11-23 devnull if(cmp) {
130 7763a61a 2003-11-23 devnull Bprint(bout, ": gen: %lud size: %llud",
131 7763a61a 2003-11-23 devnull s->gen, size);
132 7763a61a 2003-11-23 devnull if(!s->dir)
133 7763a61a 2003-11-23 devnull Bprint(bout, ": %V", s->score);
134 7763a61a 2003-11-23 devnull } else {
135 7763a61a 2003-11-23 devnull Bprint(bout, ": gen: %lud psize: %d dsize: %d",
136 7763a61a 2003-11-23 devnull s->gen, s->psize, s->dsize);
137 7763a61a 2003-11-23 devnull Bprint(bout, " depth: %d size: %llud: %V",
138 7763a61a 2003-11-23 devnull s->depth, size, s->score);
139 7763a61a 2003-11-23 devnull }
140 7763a61a 2003-11-23 devnull
141 7763a61a 2003-11-23 devnull if(s->reserved)
142 7763a61a 2003-11-23 devnull Bprint(bout, ": reserved not emtpy");
143 7763a61a 2003-11-23 devnull }
144 7763a61a 2003-11-23 devnull Bprint(bout, "\n");
145 7763a61a 2003-11-23 devnull }
146 7763a61a 2003-11-23 devnull
147 7763a61a 2003-11-23 devnull int
148 7763a61a 2003-11-23 devnull parse(Source *s, uchar *p)
149 7763a61a 2003-11-23 devnull {
150 7763a61a 2003-11-23 devnull VtEntry dir;
151 7763a61a 2003-11-23 devnull
152 7763a61a 2003-11-23 devnull memset(s, 0, sizeof(*s));
153 3d77c87e 2004-03-15 devnull if(vtentryunpack(&dir, p, 0) < 0)
154 3d77c87e 2004-03-15 devnull return -1;
155 7763a61a 2003-11-23 devnull
156 7763a61a 2003-11-23 devnull if(!(dir.flags & VtEntryActive))
157 3d77c87e 2004-03-15 devnull return 0;
158 7763a61a 2003-11-23 devnull
159 7763a61a 2003-11-23 devnull s->active = 1;
160 7763a61a 2003-11-23 devnull s->gen = dir.gen;
161 7763a61a 2003-11-23 devnull s->psize = dir.psize;
162 7763a61a 2003-11-23 devnull s->dsize = dir.size;
163 7763a61a 2003-11-23 devnull s->size = dir.size;
164 7763a61a 2003-11-23 devnull memmove(s->score, dir.score, VtScoreSize);
165 3d77c87e 2004-03-15 devnull if((dir.type&~VtTypeDepthMask) == VtDirType)
166 7763a61a 2003-11-23 devnull s->dir = 1;
167 3d77c87e 2004-03-15 devnull fprint(2, "sdir %d type %d %.*H\n", s->dir, dir.type, VtEntrySize, p);
168 3d77c87e 2004-03-15 devnull s->depth = dir.type&VtTypeDepthMask;
169 3d77c87e 2004-03-15 devnull return 0;
170 7763a61a 2003-11-23 devnull }
171 7763a61a 2003-11-23 devnull
172 7763a61a 2003-11-23 devnull int
173 3d77c87e 2004-03-15 devnull sourceread(Source *s, ulong block, uchar *p, int n)
174 7763a61a 2003-11-23 devnull {
175 3d77c87e 2004-03-15 devnull uchar *buf;
176 7763a61a 2003-11-23 devnull uchar score[VtScoreSize];
177 7763a61a 2003-11-23 devnull int i, nn, np, type;
178 7763a61a 2003-11-23 devnull int elem[VtPointerDepth];
179 7763a61a 2003-11-23 devnull
180 3d77c87e 2004-03-15 devnull buf = vtmalloc(VtMaxLumpSize);
181 3d77c87e 2004-03-15 devnull
182 7763a61a 2003-11-23 devnull memmove(score, s->score, VtScoreSize);
183 7763a61a 2003-11-23 devnull
184 7763a61a 2003-11-23 devnull np = s->psize/VtScoreSize;
185 7763a61a 2003-11-23 devnull for(i=0; i<s->depth; i++) {
186 7763a61a 2003-11-23 devnull elem[i] = block % np;
187 7763a61a 2003-11-23 devnull block /= np;
188 7763a61a 2003-11-23 devnull }
189 7763a61a 2003-11-23 devnull assert(block == 0);
190 7763a61a 2003-11-23 devnull
191 7763a61a 2003-11-23 devnull for(i=s->depth-1; i>=0; i--) {
192 3d77c87e 2004-03-15 devnull nn = vtread(z, score, (s->dir ? VtDirType : VtDataType)+1+i, buf, s->psize);
193 3d77c87e 2004-03-15 devnull if(nn < 0){
194 3d77c87e 2004-03-15 devnull fprint(2, "vtread %V %d: %r\n", score, (s->dir ? VtDirType : VtDataType)+1+i);
195 3d77c87e 2004-03-15 devnull free(buf);
196 7763a61a 2003-11-23 devnull return -1;
197 7763a61a 2003-11-23 devnull }
198 7763a61a 2003-11-23 devnull
199 3d77c87e 2004-03-15 devnull if((elem[i]+1)*VtScoreSize > nn){
200 3d77c87e 2004-03-15 devnull free(buf);
201 7763a61a 2003-11-23 devnull return 0;
202 3d77c87e 2004-03-15 devnull }
203 7763a61a 2003-11-23 devnull memmove(score, buf + elem[i]*VtScoreSize, VtScoreSize);
204 7763a61a 2003-11-23 devnull }
205 7763a61a 2003-11-23 devnull
206 7763a61a 2003-11-23 devnull if(s->dir)
207 7763a61a 2003-11-23 devnull type = VtDirType;
208 7763a61a 2003-11-23 devnull else
209 7763a61a 2003-11-23 devnull type = VtDataType;
210 7763a61a 2003-11-23 devnull
211 3d77c87e 2004-03-15 devnull nn = vtread(z, score, type, p, n);
212 3d77c87e 2004-03-15 devnull if(nn < 0){
213 3d77c87e 2004-03-15 devnull fprint(2, "vtread %V %d: %r\n", score, type);
214 3d77c87e 2004-03-15 devnull abort();
215 3d77c87e 2004-03-15 devnull free(buf);
216 7763a61a 2003-11-23 devnull return -1;
217 7763a61a 2003-11-23 devnull }
218 7763a61a 2003-11-23 devnull
219 3d77c87e 2004-03-15 devnull free(buf);
220 7763a61a 2003-11-23 devnull return nn;
221 7763a61a 2003-11-23 devnull }
222 7763a61a 2003-11-23 devnull
223 7763a61a 2003-11-23 devnull void
224 3d77c87e 2004-03-15 devnull dumpfilecontents(Source *s)
225 7763a61a 2003-11-23 devnull {
226 7763a61a 2003-11-23 devnull int nb, lb, i, n;
227 7763a61a 2003-11-23 devnull uchar buf[VtMaxLumpSize];
228 7763a61a 2003-11-23 devnull
229 7763a61a 2003-11-23 devnull nb = (s->size + s->dsize - 1)/s->dsize;
230 7763a61a 2003-11-23 devnull lb = s->size%s->dsize;
231 7763a61a 2003-11-23 devnull for(i=0; i<nb; i++) {
232 7763a61a 2003-11-23 devnull memset(buf, 0, s->dsize);
233 3d77c87e 2004-03-15 devnull n = sourceread(s, i, buf, s->dsize);
234 7763a61a 2003-11-23 devnull if(n < 0) {
235 3d77c87e 2004-03-15 devnull fprint(2, "could not read block: %d: %r\n", i);
236 7763a61a 2003-11-23 devnull continue;
237 7763a61a 2003-11-23 devnull }
238 7763a61a 2003-11-23 devnull if(i < nb-1)
239 7763a61a 2003-11-23 devnull Bwrite(bout, buf, s->dsize);
240 7763a61a 2003-11-23 devnull else
241 7763a61a 2003-11-23 devnull Bwrite(bout, buf, lb);
242 7763a61a 2003-11-23 devnull }
243 7763a61a 2003-11-23 devnull }
244 7763a61a 2003-11-23 devnull
245 7763a61a 2003-11-23 devnull void
246 3d77c87e 2004-03-15 devnull dumpfile(Source *s, int indent)
247 7763a61a 2003-11-23 devnull {
248 7763a61a 2003-11-23 devnull int nb, i, j, n;
249 3d77c87e 2004-03-15 devnull uchar *buf;
250 7763a61a 2003-11-23 devnull uchar score[VtScoreSize];
251 7763a61a 2003-11-23 devnull
252 3d77c87e 2004-03-15 devnull buf = vtmalloc(VtMaxLumpSize);
253 7763a61a 2003-11-23 devnull nb = (s->size + s->dsize - 1)/s->dsize;
254 7763a61a 2003-11-23 devnull for(i=0; i<nb; i++) {
255 7763a61a 2003-11-23 devnull memset(buf, 0, s->dsize);
256 3d77c87e 2004-03-15 devnull n = sourceread(s, i, buf, s->dsize);
257 7763a61a 2003-11-23 devnull if(n < 0) {
258 3d77c87e 2004-03-15 devnull fprint(2, "could not read block: %d: %r\n", i);
259 7763a61a 2003-11-23 devnull continue;
260 7763a61a 2003-11-23 devnull }
261 7763a61a 2003-11-23 devnull for(j=0; j<indent; j++)
262 7763a61a 2003-11-23 devnull Bprint(bout, " ");
263 3d77c87e 2004-03-15 devnull sha1(buf, n, score, nil);
264 7763a61a 2003-11-23 devnull Bprint(bout, "%4d: size: %ud: %V\n", i, n, score);
265 7763a61a 2003-11-23 devnull }
266 3d77c87e 2004-03-15 devnull vtfree(buf);
267 7763a61a 2003-11-23 devnull }
268 7763a61a 2003-11-23 devnull
269 7763a61a 2003-11-23 devnull int
270 3d77c87e 2004-03-15 devnull dumpdir(Source *s, int indent)
271 7763a61a 2003-11-23 devnull {
272 7763a61a 2003-11-23 devnull int pb, ne, nb, i, j, n, entry;
273 7763a61a 2003-11-23 devnull Source ss;
274 3d77c87e 2004-03-15 devnull uchar *buf;
275 7763a61a 2003-11-23 devnull
276 3d77c87e 2004-03-15 devnull buf = vtmalloc(VtMaxLumpSize);
277 7763a61a 2003-11-23 devnull pb = s->dsize/VtEntrySize;
278 7763a61a 2003-11-23 devnull ne = pb*(s->size/s->dsize) + (s->size%s->dsize)/VtEntrySize;
279 7763a61a 2003-11-23 devnull nb = (s->size + s->dsize - 1)/s->dsize;
280 7763a61a 2003-11-23 devnull for(i=0; i<nb; i++) {
281 7763a61a 2003-11-23 devnull memset(buf, 0, s->dsize);
282 3d77c87e 2004-03-15 devnull n = sourceread(s, i, buf, s->dsize);
283 7763a61a 2003-11-23 devnull if(n < 0) {
284 3d77c87e 2004-03-15 devnull fprint(2, "could not read block: %d: %r\n", i);
285 7763a61a 2003-11-23 devnull continue;
286 7763a61a 2003-11-23 devnull }
287 7763a61a 2003-11-23 devnull for(j=0; j<pb; j++) {
288 7763a61a 2003-11-23 devnull entry = i*pb + j;
289 7763a61a 2003-11-23 devnull if(entry >= ne)
290 7763a61a 2003-11-23 devnull break;
291 7763a61a 2003-11-23 devnull parse(&ss, buf + j * VtEntrySize);
292 7763a61a 2003-11-23 devnull
293 7763a61a 2003-11-23 devnull if(!find)
294 3d77c87e 2004-03-15 devnull sourceprint(&ss, indent, entry);
295 7763a61a 2003-11-23 devnull else if(memcmp(ss.score, fscore, VtScoreSize) == 0) {
296 3d77c87e 2004-03-15 devnull dumpfilecontents(&ss);
297 3d77c87e 2004-03-15 devnull free(buf);
298 3d77c87e 2004-03-15 devnull return -1;
299 7763a61a 2003-11-23 devnull }
300 7763a61a 2003-11-23 devnull
301 7763a61a 2003-11-23 devnull if(ss.dir) {
302 3d77c87e 2004-03-15 devnull if(dumpdir(&ss, indent+1) < 0){
303 3d77c87e 2004-03-15 devnull free(buf);
304 3d77c87e 2004-03-15 devnull return -1;
305 3d77c87e 2004-03-15 devnull }
306 7763a61a 2003-11-23 devnull } else if(all)
307 3d77c87e 2004-03-15 devnull dumpfile(&ss, indent+1);
308 7763a61a 2003-11-23 devnull }
309 7763a61a 2003-11-23 devnull }
310 3d77c87e 2004-03-15 devnull free(buf);
311 3d77c87e 2004-03-15 devnull return 0;
312 7763a61a 2003-11-23 devnull }
313 7763a61a 2003-11-23 devnull
314 7763a61a 2003-11-23 devnull void
315 7763a61a 2003-11-23 devnull usage(void)
316 7763a61a 2003-11-23 devnull {
317 7763a61a 2003-11-23 devnull fprint(2, "%s: [file]\n", argv0);
318 3d77c87e 2004-03-15 devnull threadexits("usage");
319 7763a61a 2003-11-23 devnull }
320 7763a61a 2003-11-23 devnull
321 7763a61a 2003-11-23 devnull void
322 3d77c87e 2004-03-15 devnull readroot(VtRoot *root, uchar *score, char *file)
323 7763a61a 2003-11-23 devnull {
324 7763a61a 2003-11-23 devnull int fd;
325 3d77c87e 2004-03-15 devnull char *pref;
326 3d77c87e 2004-03-15 devnull char buf[VtRootSize];
327 3d77c87e 2004-03-15 devnull int n, nn;
328 7763a61a 2003-11-23 devnull
329 7763a61a 2003-11-23 devnull if(file == 0)
330 7763a61a 2003-11-23 devnull fd = 0;
331 7763a61a 2003-11-23 devnull else {
332 7763a61a 2003-11-23 devnull fd = open(file, OREAD);
333 7763a61a 2003-11-23 devnull if(fd < 0)
334 7763a61a 2003-11-23 devnull sysfatal("could not open file: %s: %r\n", file);
335 7763a61a 2003-11-23 devnull }
336 7763a61a 2003-11-23 devnull n = readn(fd, buf, sizeof(buf)-1);
337 7763a61a 2003-11-23 devnull if(n < 0)
338 7763a61a 2003-11-23 devnull sysfatal("read failed: %r\n");
339 3d77c87e 2004-03-15 devnull if(n==0 || buf[n-1] != '\n')
340 3d77c87e 2004-03-15 devnull sysfatal("not a root file");
341 3d77c87e 2004-03-15 devnull buf[n-1] = 0;
342 7763a61a 2003-11-23 devnull close(fd);
343 7763a61a 2003-11-23 devnull
344 3d77c87e 2004-03-15 devnull if(vtparsescore(buf, &pref, score) < 0){
345 3d77c87e 2004-03-15 devnull sysfatal("not a root file");
346 7763a61a 2003-11-23 devnull }
347 3d77c87e 2004-03-15 devnull nn = vtread(z, score, VtRootType, buf, VtRootSize);
348 3d77c87e 2004-03-15 devnull if(nn < 0)
349 3d77c87e 2004-03-15 devnull sysfatal("cannot read root %V", score);
350 3d77c87e 2004-03-15 devnull if(vtrootunpack(root, buf) < 0)
351 3d77c87e 2004-03-15 devnull sysfatal("cannot parse root: %r");
352 7763a61a 2003-11-23 devnull }