return of venti

src/cmd/venti/readlist.c

@@ -0,0 +1,112 @@
+#include <u.h>
+#include <libc.h>
+#include <thread.h>
+#include <venti.h>
+#include <bio.h>
+
+char *host;
+Biobuf b;
+VtConn *z;
+uchar *buf;
+void run(Biobuf*);
+int nn;
+
+void
+usage(void)
+{
+	fprint(2, "usage: readlist [-h host] list\n");
+	threadexitsall("usage");
+}
+
+int
+parsescore(uchar *score, char *buf, int n)
+{
+	int i, c;
+
+	memset(score, 0, VtScoreSize);
+
+	if(n != VtScoreSize*2){
+		werrstr("score wrong length %d", n);
+		return -1;
+	}
+	for(i=0; i<VtScoreSize*2; i++) {
+		if(buf[i] >= '0' && buf[i] <= '9')
+			c = buf[i] - '0';
+		else if(buf[i] >= 'a' && buf[i] <= 'f')
+			c = buf[i] - 'a' + 10;
+		else if(buf[i] >= 'A' && buf[i] <= 'F')
+			c = buf[i] - 'A' + 10;
+		else {
+			c = buf[i];
+			werrstr("bad score char %d '%c'", c, c);
+			return -1;
+		}
+
+		if((i & 1) == 0)
+			c <<= 4;
+	
+		score[i>>1] |= c;
+	}
+	return 0;
+}
+
+void
+threadmain(int argc, char *argv[])
+{
+	int fd, i;
+
+	ARGBEGIN{
+	case 'h':
+		host = EARGF(usage());
+		break;
+	default:
+		usage();
+		break;
+	}ARGEND
+
+	fmtinstall('V', vtscorefmt);
+	buf = vtmallocz(VtMaxLumpSize);
+	z = vtdial(host);
+	if(z == nil)
+		sysfatal("could not connect to server: %r");
+	if(vtconnect(z) < 0)
+		sysfatal("vtconnect: %r");
+
+	if(argc == 0){
+		Binit(&b, 0, OREAD);
+		run(&b);
+	}else{
+		for(i=0; i<argc; i++){
+			if((fd = open(argv[i], OREAD)) < 0)
+				sysfatal("open %s: %r", argv[i]);
+			Binit(&b, fd, OREAD);
+			run(&b);
+		}
+	}
+	threadexitsall(nil);
+}
+
+void
+run(Biobuf *b)
+{
+	char *p, *f[10];
+	int nf;
+	uchar score[20];
+	int type, n;
+
+	while((p = Brdline(b, '\n')) != nil){
+		p[Blinelen(b)-1] = 0;
+		nf = tokenize(p, f, nelem(f));
+		if(nf != 2)
+			sysfatal("syntax error in work list");
+		if(parsescore(score, f[0], strlen(f[0])) < 0)
+			sysfatal("bad score %s in work list", f[0]);
+		type = atoi(f[1]);
+		n = vtread(z, score, type, buf, VtMaxLumpSize);
+		if(n < 0)
+			sysfatal("could not read %s %s: %r", f[0], f[1]);
+		// write(1, buf, n);
+		if(++nn%1000 == 0)
+			print("%d...", nn);
+	}
+}