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