1*368c31abSDavid du Colombier #include <u.h>
2*368c31abSDavid du Colombier #include <libc.h>
3*368c31abSDavid du Colombier #include <thread.h>
4*368c31abSDavid du Colombier #include <venti.h>
5*368c31abSDavid du Colombier #include <bio.h>
6*368c31abSDavid du Colombier
7*368c31abSDavid du Colombier char *host;
8*368c31abSDavid du Colombier Biobuf b;
9*368c31abSDavid du Colombier VtConn *z;
10*368c31abSDavid du Colombier uchar *buf;
11*368c31abSDavid du Colombier void run(Biobuf*);
12*368c31abSDavid du Colombier int nn;
13*368c31abSDavid du Colombier
14*368c31abSDavid du Colombier void
usage(void)15*368c31abSDavid du Colombier usage(void)
16*368c31abSDavid du Colombier {
17*368c31abSDavid du Colombier fprint(2, "usage: readlist [-h host] list\n");
18*368c31abSDavid du Colombier threadexitsall("usage");
19*368c31abSDavid du Colombier }
20*368c31abSDavid du Colombier
21*368c31abSDavid du Colombier int
parsescore(uchar * score,char * buf,int n)22*368c31abSDavid du Colombier parsescore(uchar *score, char *buf, int n)
23*368c31abSDavid du Colombier {
24*368c31abSDavid du Colombier int i, c;
25*368c31abSDavid du Colombier
26*368c31abSDavid du Colombier memset(score, 0, VtScoreSize);
27*368c31abSDavid du Colombier
28*368c31abSDavid du Colombier if(n != VtScoreSize*2){
29*368c31abSDavid du Colombier werrstr("score wrong length %d", n);
30*368c31abSDavid du Colombier return -1;
31*368c31abSDavid du Colombier }
32*368c31abSDavid du Colombier for(i=0; i<VtScoreSize*2; i++) {
33*368c31abSDavid du Colombier if(buf[i] >= '0' && buf[i] <= '9')
34*368c31abSDavid du Colombier c = buf[i] - '0';
35*368c31abSDavid du Colombier else if(buf[i] >= 'a' && buf[i] <= 'f')
36*368c31abSDavid du Colombier c = buf[i] - 'a' + 10;
37*368c31abSDavid du Colombier else if(buf[i] >= 'A' && buf[i] <= 'F')
38*368c31abSDavid du Colombier c = buf[i] - 'A' + 10;
39*368c31abSDavid du Colombier else {
40*368c31abSDavid du Colombier c = buf[i];
41*368c31abSDavid du Colombier werrstr("bad score char %d '%c'", c, c);
42*368c31abSDavid du Colombier return -1;
43*368c31abSDavid du Colombier }
44*368c31abSDavid du Colombier
45*368c31abSDavid du Colombier if((i & 1) == 0)
46*368c31abSDavid du Colombier c <<= 4;
47*368c31abSDavid du Colombier
48*368c31abSDavid du Colombier score[i>>1] |= c;
49*368c31abSDavid du Colombier }
50*368c31abSDavid du Colombier return 0;
51*368c31abSDavid du Colombier }
52*368c31abSDavid du Colombier
53*368c31abSDavid du Colombier void
threadmain(int argc,char * argv[])54*368c31abSDavid du Colombier threadmain(int argc, char *argv[])
55*368c31abSDavid du Colombier {
56*368c31abSDavid du Colombier int fd, i;
57*368c31abSDavid du Colombier
58*368c31abSDavid du Colombier ARGBEGIN{
59*368c31abSDavid du Colombier case 'h':
60*368c31abSDavid du Colombier host = EARGF(usage());
61*368c31abSDavid du Colombier break;
62*368c31abSDavid du Colombier default:
63*368c31abSDavid du Colombier usage();
64*368c31abSDavid du Colombier break;
65*368c31abSDavid du Colombier }ARGEND
66*368c31abSDavid du Colombier
67*368c31abSDavid du Colombier fmtinstall('V', vtscorefmt);
68*368c31abSDavid du Colombier buf = vtmallocz(VtMaxLumpSize);
69*368c31abSDavid du Colombier z = vtdial(host);
70*368c31abSDavid du Colombier if(z == nil)
71*368c31abSDavid du Colombier sysfatal("could not connect to server: %r");
72*368c31abSDavid du Colombier if(vtconnect(z) < 0)
73*368c31abSDavid du Colombier sysfatal("vtconnect: %r");
74*368c31abSDavid du Colombier
75*368c31abSDavid du Colombier if(argc == 0){
76*368c31abSDavid du Colombier Binit(&b, 0, OREAD);
77*368c31abSDavid du Colombier run(&b);
78*368c31abSDavid du Colombier }else{
79*368c31abSDavid du Colombier for(i=0; i<argc; i++){
80*368c31abSDavid du Colombier if((fd = open(argv[i], OREAD)) < 0)
81*368c31abSDavid du Colombier sysfatal("open %s: %r", argv[i]);
82*368c31abSDavid du Colombier Binit(&b, fd, OREAD);
83*368c31abSDavid du Colombier run(&b);
84*368c31abSDavid du Colombier }
85*368c31abSDavid du Colombier }
86*368c31abSDavid du Colombier threadexitsall(nil);
87*368c31abSDavid du Colombier }
88*368c31abSDavid du Colombier
89*368c31abSDavid du Colombier void
run(Biobuf * b)90*368c31abSDavid du Colombier run(Biobuf *b)
91*368c31abSDavid du Colombier {
92*368c31abSDavid du Colombier char *p, *f[10];
93*368c31abSDavid du Colombier int nf;
94*368c31abSDavid du Colombier uchar score[20];
95*368c31abSDavid du Colombier int type, n;
96*368c31abSDavid du Colombier
97*368c31abSDavid du Colombier while((p = Brdline(b, '\n')) != nil){
98*368c31abSDavid du Colombier p[Blinelen(b)-1] = 0;
99*368c31abSDavid du Colombier nf = tokenize(p, f, nelem(f));
100*368c31abSDavid du Colombier if(nf != 2)
101*368c31abSDavid du Colombier sysfatal("syntax error in work list");
102*368c31abSDavid du Colombier if(parsescore(score, f[0], strlen(f[0])) < 0)
103*368c31abSDavid du Colombier sysfatal("bad score %s in work list", f[0]);
104*368c31abSDavid du Colombier type = atoi(f[1]);
105*368c31abSDavid du Colombier n = vtread(z, score, type, buf, VtMaxLumpSize);
106*368c31abSDavid du Colombier if(n < 0)
107*368c31abSDavid du Colombier sysfatal("could not read %s %s: %r", f[0], f[1]);
108*368c31abSDavid du Colombier /* write(1, buf, n); */
109*368c31abSDavid du Colombier if(++nn%1000 == 0)
110*368c31abSDavid du Colombier print("%d...", nn);
111*368c31abSDavid du Colombier }
112*368c31abSDavid du Colombier }
113