Blame


1 a58f193d 2008-07-09 rsc #include <u.h>
2 a58f193d 2008-07-09 rsc #include <libc.h>
3 a58f193d 2008-07-09 rsc #include <bio.h>
4 a58f193d 2008-07-09 rsc #include <venti.h>
5 a58f193d 2008-07-09 rsc #include <libsec.h>
6 a58f193d 2008-07-09 rsc #include <thread.h>
7 a58f193d 2008-07-09 rsc
8 75d04888 2009-05-25 rsc enum
9 75d04888 2009-05-25 rsc {
10 75d04888 2009-05-25 rsc // XXX What to do here?
11 71ce470a 2012-05-29 rsc VtMaxLumpSize = 65535,
12 75d04888 2009-05-25 rsc };
13 75d04888 2009-05-25 rsc
14 a58f193d 2008-07-09 rsc VtConn *z;
15 a58f193d 2008-07-09 rsc char *host;
16 a58f193d 2008-07-09 rsc
17 a58f193d 2008-07-09 rsc void
18 a58f193d 2008-07-09 rsc usage(void)
19 a58f193d 2008-07-09 rsc {
20 a58f193d 2008-07-09 rsc fprint(2, "usage: venti/dump [-h host] score\n");
21 a58f193d 2008-07-09 rsc threadexitsall("usage");
22 a58f193d 2008-07-09 rsc }
23 a58f193d 2008-07-09 rsc
24 a58f193d 2008-07-09 rsc Biobuf bout;
25 a58f193d 2008-07-09 rsc char spaces[256];
26 a58f193d 2008-07-09 rsc
27 a58f193d 2008-07-09 rsc void
28 a58f193d 2008-07-09 rsc dump(int indent, uchar *score, int type)
29 a58f193d 2008-07-09 rsc {
30 a58f193d 2008-07-09 rsc int i, n;
31 a58f193d 2008-07-09 rsc uchar *buf;
32 a58f193d 2008-07-09 rsc VtEntry e;
33 a58f193d 2008-07-09 rsc VtRoot root;
34 fa325e9b 2020-01-10 cross
35 a58f193d 2008-07-09 rsc if(spaces[0] == 0)
36 a58f193d 2008-07-09 rsc memset(spaces, ' ', sizeof spaces-1);
37 a58f193d 2008-07-09 rsc
38 a58f193d 2008-07-09 rsc buf = vtmallocz(VtMaxLumpSize);
39 a58f193d 2008-07-09 rsc if(memcmp(score, vtzeroscore, VtScoreSize) == 0)
40 a58f193d 2008-07-09 rsc n = 0;
41 a58f193d 2008-07-09 rsc else
42 a58f193d 2008-07-09 rsc n = vtread(z, score, type, buf, VtMaxLumpSize);
43 a58f193d 2008-07-09 rsc if(n < 0){
44 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*serror reading %V: %r\n", indent*4, spaces, score);
45 a58f193d 2008-07-09 rsc goto out;
46 a58f193d 2008-07-09 rsc }
47 a58f193d 2008-07-09 rsc switch(type){
48 a58f193d 2008-07-09 rsc case VtRootType:
49 a58f193d 2008-07-09 rsc if(vtrootunpack(&root, buf) < 0){
50 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*serror unpacking root %V: %r\n", indent*4, spaces, score);
51 a58f193d 2008-07-09 rsc goto out;
52 a58f193d 2008-07-09 rsc }
53 36c242dd 2011-11-08 rsc Bprint(&bout, "%.*s%V root name=%s type=%s prev=%V bsize=%ld\n",
54 a58f193d 2008-07-09 rsc indent*4, spaces, score, root.name, root.type, root.prev, root.blocksize);
55 a58f193d 2008-07-09 rsc dump(indent+1, root.score, VtDirType);
56 a58f193d 2008-07-09 rsc break;
57 fa325e9b 2020-01-10 cross
58 a58f193d 2008-07-09 rsc case VtDirType:
59 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*s%V dir n=%d\n", indent*4, spaces, score, n);
60 a58f193d 2008-07-09 rsc for(i=0; i*VtEntrySize<n; i++){
61 a58f193d 2008-07-09 rsc if(vtentryunpack(&e, buf, i) < 0){
62 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*s%d: cannot unpack\n", indent+1, spaces, i);
63 a58f193d 2008-07-09 rsc continue;
64 a58f193d 2008-07-09 rsc }
65 36c242dd 2011-11-08 rsc Bprint(&bout, "%.*s%d: gen=%#lux psize=%ld dsize=%ld type=%d flags=%#x size=%llud score=%V\n",
66 a58f193d 2008-07-09 rsc (indent+1)*4, spaces, i, e.gen, e.psize, e.dsize, e.type, e.flags, e.size, e.score);
67 a58f193d 2008-07-09 rsc dump(indent+2, e.score, e.type);
68 a58f193d 2008-07-09 rsc }
69 a58f193d 2008-07-09 rsc break;
70 fa325e9b 2020-01-10 cross
71 a58f193d 2008-07-09 rsc case VtDataType:
72 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*s%V data n=%d", indent*4, spaces, score, n);
73 a58f193d 2008-07-09 rsc for(i=0; i<n; i++){
74 a58f193d 2008-07-09 rsc if(i%16 == 0)
75 a58f193d 2008-07-09 rsc Bprint(&bout, "\n%.*s", (indent+1)*4, spaces);
76 a58f193d 2008-07-09 rsc Bprint(&bout, " %02x", buf[i]);
77 a58f193d 2008-07-09 rsc }
78 a58f193d 2008-07-09 rsc Bprint(&bout, "\n");
79 a58f193d 2008-07-09 rsc break;
80 a58f193d 2008-07-09 rsc
81 a58f193d 2008-07-09 rsc default:
82 a58f193d 2008-07-09 rsc if(type >= VtDirType)
83 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*s%V dir+%d\n", indent*4, spaces, score, type-VtDirType);
84 a58f193d 2008-07-09 rsc else
85 a58f193d 2008-07-09 rsc Bprint(&bout, "%.*s%V data+%d\n", indent*4, spaces, score, type-VtDirType);
86 a58f193d 2008-07-09 rsc for(i=0; i<n; i+=VtScoreSize)
87 a58f193d 2008-07-09 rsc dump(indent+1, buf+i, type-1);
88 a58f193d 2008-07-09 rsc break;
89 a58f193d 2008-07-09 rsc }
90 a58f193d 2008-07-09 rsc out:
91 fa325e9b 2020-01-10 cross free(buf);
92 a58f193d 2008-07-09 rsc }
93 a58f193d 2008-07-09 rsc
94 a58f193d 2008-07-09 rsc
95 a58f193d 2008-07-09 rsc void
96 a58f193d 2008-07-09 rsc threadmain(int argc, char *argv[])
97 a58f193d 2008-07-09 rsc {
98 a58f193d 2008-07-09 rsc int type, n;
99 a58f193d 2008-07-09 rsc uchar score[VtScoreSize];
100 a58f193d 2008-07-09 rsc uchar *buf;
101 a58f193d 2008-07-09 rsc char *prefix;
102 a58f193d 2008-07-09 rsc
103 a58f193d 2008-07-09 rsc fmtinstall('F', vtfcallfmt);
104 a58f193d 2008-07-09 rsc fmtinstall('V', vtscorefmt);
105 a58f193d 2008-07-09 rsc
106 a58f193d 2008-07-09 rsc ARGBEGIN{
107 a58f193d 2008-07-09 rsc case 'h':
108 a58f193d 2008-07-09 rsc host = EARGF(usage());
109 a58f193d 2008-07-09 rsc break;
110 a58f193d 2008-07-09 rsc default:
111 a58f193d 2008-07-09 rsc usage();
112 a58f193d 2008-07-09 rsc }ARGEND
113 a58f193d 2008-07-09 rsc
114 a58f193d 2008-07-09 rsc if(argc != 1)
115 a58f193d 2008-07-09 rsc usage();
116 a58f193d 2008-07-09 rsc
117 a58f193d 2008-07-09 rsc if(vtparsescore(argv[0], &prefix, score) < 0)
118 a58f193d 2008-07-09 rsc sysfatal("could not parse score: %r");
119 a58f193d 2008-07-09 rsc
120 a58f193d 2008-07-09 rsc buf = vtmallocz(VtMaxLumpSize);
121 a58f193d 2008-07-09 rsc z = vtdial(host);
122 a58f193d 2008-07-09 rsc if(z == nil)
123 a58f193d 2008-07-09 rsc sysfatal("dialing venti: %r");
124 a58f193d 2008-07-09 rsc if(vtconnect(z) < 0)
125 a58f193d 2008-07-09 rsc sysfatal("vtconnect src: %r");
126 a58f193d 2008-07-09 rsc
127 a58f193d 2008-07-09 rsc for(type=0; type<VtMaxType; type++){
128 a58f193d 2008-07-09 rsc n = vtread(z, score, type, buf, VtMaxLumpSize);
129 a58f193d 2008-07-09 rsc if(n >= 0)
130 a58f193d 2008-07-09 rsc goto havetype;
131 a58f193d 2008-07-09 rsc }
132 a58f193d 2008-07-09 rsc sysfatal("cannot find block %V", score);
133 a58f193d 2008-07-09 rsc
134 a58f193d 2008-07-09 rsc havetype:
135 a58f193d 2008-07-09 rsc Binit(&bout, 1, OWRITE);
136 a58f193d 2008-07-09 rsc dump(0, score, type);
137 a58f193d 2008-07-09 rsc Bflush(&bout);
138 a58f193d 2008-07-09 rsc threadexitsall(nil);
139 a58f193d 2008-07-09 rsc }