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