1 a0d146ed 2005-07-12 devnull #include <u.h>
 2 a0d146ed 2005-07-12 devnull #include <libc.h>
 3 a0d146ed 2005-07-12 devnull #include <venti.h>
 4 a0d146ed 2005-07-12 devnull #include <libsec.h>
 5 a0d146ed 2005-07-12 devnull #include <thread.h>
 6 a0d146ed 2005-07-12 devnull 
 7 a0d146ed 2005-07-12 devnull void
 8 a0d146ed 2005-07-12 devnull usage(void)
 9 a0d146ed 2005-07-12 devnull {
10 a0d146ed 2005-07-12 devnull 	fprint(2, "usage: write [-z] [-h host] [-t type] <datablock\n");
11 a0d146ed 2005-07-12 devnull 	threadexitsall("usage");
12 a0d146ed 2005-07-12 devnull }
13 a0d146ed 2005-07-12 devnull 
14 a0d146ed 2005-07-12 devnull void
15 a0d146ed 2005-07-12 devnull threadmain(int argc, char *argv[])
16 a0d146ed 2005-07-12 devnull {
17 a0d146ed 2005-07-12 devnull 	char *host;
18 a0d146ed 2005-07-12 devnull 	int dotrunc, n, type;
19 a0d146ed 2005-07-12 devnull 	uchar *p, score[VtScoreSize];
20 a0d146ed 2005-07-12 devnull 	VtConn *z;
21 a0d146ed 2005-07-12 devnull 
22 a0d146ed 2005-07-12 devnull 	fmtinstall('F', vtfcallfmt);
23 a0d146ed 2005-07-12 devnull 	fmtinstall('V', vtscorefmt);
24 a0d146ed 2005-07-12 devnull 
25 a0d146ed 2005-07-12 devnull 	host = nil;
26 a0d146ed 2005-07-12 devnull 	dotrunc = 0;
27 a0d146ed 2005-07-12 devnull 	type = VtDataType;
28 a0d146ed 2005-07-12 devnull 	ARGBEGIN{
29 a0d146ed 2005-07-12 devnull 	case 'z':
30 a0d146ed 2005-07-12 devnull 		dotrunc = 1;
31 a0d146ed 2005-07-12 devnull 		break;
32 a0d146ed 2005-07-12 devnull 	case 'h':
33 a0d146ed 2005-07-12 devnull 		host = EARGF(usage());
34 a0d146ed 2005-07-12 devnull 		break;
35 a0d146ed 2005-07-12 devnull 	case 't':
36 a0d146ed 2005-07-12 devnull 		type = atoi(EARGF(usage()));
37 a0d146ed 2005-07-12 devnull 		break;
38 a0d146ed 2005-07-12 devnull 	default:
39 a0d146ed 2005-07-12 devnull 		usage();
40 a0d146ed 2005-07-12 devnull 		break;
41 a0d146ed 2005-07-12 devnull 	}ARGEND
42 a0d146ed 2005-07-12 devnull 
43 a0d146ed 2005-07-12 devnull 	if(argc != 0)
44 a0d146ed 2005-07-12 devnull 		usage();
45 a0d146ed 2005-07-12 devnull 
46 a0d146ed 2005-07-12 devnull 	p = vtmallocz(VtMaxLumpSize+1);
47 a0d146ed 2005-07-12 devnull 	n = readn(0, p, VtMaxLumpSize+1);
48 a0d146ed 2005-07-12 devnull 	if(n > VtMaxLumpSize)
49 a0d146ed 2005-07-12 devnull 		sysfatal("input too big: max block size is %d", VtMaxLumpSize);
50 a0d146ed 2005-07-12 devnull 	z = vtdial(host);
51 a0d146ed 2005-07-12 devnull 	if(z == nil)
52 a0d146ed 2005-07-12 devnull 		sysfatal("could not connect to server: %r");
53 a0d146ed 2005-07-12 devnull 	if(vtconnect(z) < 0)
54 a0d146ed 2005-07-12 devnull 		sysfatal("vtconnect: %r");
55 a0d146ed 2005-07-12 devnull 	if(dotrunc)
56 a0d146ed 2005-07-12 devnull 		n = vtzerotruncate(type, p, n);
57 a0d146ed 2005-07-12 devnull 	if(vtwrite(z, score, type, p, n) < 0)
58 a0d146ed 2005-07-12 devnull 		sysfatal("vtwrite: %r");
59 a0d146ed 2005-07-12 devnull 	vthangup(z);
60 a0d146ed 2005-07-12 devnull 	print("%V\n", score);
61 a0d146ed 2005-07-12 devnull 	threadexitsall(0);
62 a0d146ed 2005-07-12 devnull }