1 #include <u.h>
2 #include <libc.h>
3 #include <venti.h>
4 #include <libsec.h>
5 #include <thread.h>
6 
7 void
8 usage(void)
9 {
10 	fprint(2, "usage: read [-h host] [-t type] score\n");
11 	threadexitsall("usage");
12 }
13 
14 void
15 threadmain(int argc, char *argv[])
16 {
17 	int type, n;
18 	uchar score[VtScoreSize];
19 	uchar *buf;
20 	VtConn *z;
21 	char *host;
22 
23 	fmtinstall('F', vtfcallfmt);
24 	fmtinstall('V', vtscorefmt);
25 
26 	host = nil;
27 	type = -1;
28 	ARGBEGIN{
29 	case 'h':
30 		host = EARGF(usage());
31 		break;
32 	case 't':
33 		type = atoi(EARGF(usage()));
34 		break;
35 	default:
36 		usage();
37 		break;
38 	}ARGEND
39 
40 	if(argc != 1)
41 		usage();
42 
43 	if(vtparsescore(argv[0], nil, score) < 0)
44 		sysfatal("could not parse score '%s': %r", argv[0]);
45 
46 	buf = vtmallocz(VtMaxLumpSize);
47 
48 	z = vtdial(host);
49 	if(z == nil)
50 		sysfatal("could not connect to server: %r");
51 
52 	if(vtconnect(z) < 0)
53 		sysfatal("vtconnect: %r");
54 
55 	if(type == -1){
56 		n = -1;
57 		for(type=0; type<VtMaxType; type++){
58 			n = vtread(z, score, type, buf, VtMaxLumpSize);
59 			if(n >= 0){
60 				fprint(2, "venti/read%s%s %V %d\n", host ? " -h" : "", host ? host : "",
61 					score, type);
62 				break;
63 			}
64 		}
65 	}else
66 		n = vtread(z, score, type, buf, VtMaxLumpSize);
67 
68 	vthangup(z);
69 	if(n < 0)
70 		sysfatal("could not read block: %r");
71 	if(write(1, buf, n) != n)
72 		sysfatal("write: %r");
73 	threadexitsall(0);
74 }