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