1 a0d146ed 2005-07-12 devnull /* Copyright (c) 2004 Russ Cox */
  2 a0d146ed 2005-07-12 devnull #include <u.h>
  3 a0d146ed 2005-07-12 devnull #include <libc.h>
  4 a0d146ed 2005-07-12 devnull #include <venti.h>
  5 a0d146ed 2005-07-12 devnull #include <thread.h>
  6 a0d146ed 2005-07-12 devnull #include <libsec.h>
  7 a0d146ed 2005-07-12 devnull 
  8 a0d146ed 2005-07-12 devnull #ifndef _UNISTD_H_
  9 a0d146ed 2005-07-12 devnull #pragma varargck type "F" VtFcall*
 10 a0d146ed 2005-07-12 devnull #pragma varargck type "T" void
 11 a0d146ed 2005-07-12 devnull #endif
 12 a0d146ed 2005-07-12 devnull 
 13 a0d146ed 2005-07-12 devnull VtConn *z;
 14 a0d146ed 2005-07-12 devnull int verbose;
 15 a0d146ed 2005-07-12 devnull 
 16 a0d146ed 2005-07-12 devnull enum
 17 a0d146ed 2005-07-12 devnull {
 18 a0d146ed 2005-07-12 devnull 	STACK = 8192,
 19 a0d146ed 2005-07-12 devnull };
 20 a0d146ed 2005-07-12 devnull 
 21 a0d146ed 2005-07-12 devnull void
 22 a0d146ed 2005-07-12 devnull usage(void)
 23 a0d146ed 2005-07-12 devnull {
 24 a0d146ed 2005-07-12 devnull 	fprint(2, "usage: venti/ro [-v] [-a address] [-h address]\n");
 25 a0d146ed 2005-07-12 devnull 	threadexitsall("usage");
 26 a0d146ed 2005-07-12 devnull }
 27 a0d146ed 2005-07-12 devnull 
 28 a0d146ed 2005-07-12 devnull void
 29 a0d146ed 2005-07-12 devnull readthread(void *v)
 30 a0d146ed 2005-07-12 devnull {
 31 a0d146ed 2005-07-12 devnull 	char err[ERRMAX];
 32 a0d146ed 2005-07-12 devnull 	VtReq *r;
 33 a0d146ed 2005-07-12 devnull 	uchar *buf;
 34 a0d146ed 2005-07-12 devnull 	int n;
 35 a0d146ed 2005-07-12 devnull 	
 36 a0d146ed 2005-07-12 devnull 	r = v;
 37 a0d146ed 2005-07-12 devnull 	buf = vtmalloc(r->tx.count);
 38 a0d146ed 2005-07-12 devnull 	if((n=vtread(z, r->tx.score, r->tx.blocktype, buf, r->tx.count)) < 0){
 39 a0d146ed 2005-07-12 devnull 		r->rx.msgtype = VtRerror;
 40 a0d146ed 2005-07-12 devnull 		rerrstr(err, sizeof err);
 41 a0d146ed 2005-07-12 devnull 		r->rx.error = vtstrdup(err);
 42 a0d146ed 2005-07-12 devnull 		free(buf);
 43 a0d146ed 2005-07-12 devnull 	}else{
 44 a0d146ed 2005-07-12 devnull 		r->rx.data = packetforeign(buf, n, free, buf);
 45 a0d146ed 2005-07-12 devnull 	}
 46 a0d146ed 2005-07-12 devnull 	if(verbose)
 47 a0d146ed 2005-07-12 devnull 		fprint(2, "-> %F\n", &r->rx);
 48 a0d146ed 2005-07-12 devnull 	vtrespond(r);
 49 a0d146ed 2005-07-12 devnull }
 50 a0d146ed 2005-07-12 devnull 
 51 a0d146ed 2005-07-12 devnull void
 52 a0d146ed 2005-07-12 devnull threadmain(int argc, char **argv)
 53 a0d146ed 2005-07-12 devnull {
 54 a0d146ed 2005-07-12 devnull 	VtReq *r;
 55 a0d146ed 2005-07-12 devnull 	VtSrv *srv;
 56 a0d146ed 2005-07-12 devnull 	char *address, *ventiaddress;
 57 a0d146ed 2005-07-12 devnull 
 58 a0d146ed 2005-07-12 devnull 	fmtinstall('F', vtfcallfmt);
 59 a0d146ed 2005-07-12 devnull 	fmtinstall('V', vtscorefmt);
 60 a0d146ed 2005-07-12 devnull 	
 61 a0d146ed 2005-07-12 devnull 	address = "tcp!*!venti";
 62 a0d146ed 2005-07-12 devnull 	ventiaddress = nil;
 63 a0d146ed 2005-07-12 devnull 	
 64 a0d146ed 2005-07-12 devnull 	ARGBEGIN{
 65 a0d146ed 2005-07-12 devnull 	case 'v':
 66 a0d146ed 2005-07-12 devnull 		verbose++;
 67 a0d146ed 2005-07-12 devnull 		break;
 68 a0d146ed 2005-07-12 devnull 	case 'a':
 69 a0d146ed 2005-07-12 devnull 		address = EARGF(usage());
 70 a0d146ed 2005-07-12 devnull 		break;
 71 a0d146ed 2005-07-12 devnull 	case 'h':
 72 a0d146ed 2005-07-12 devnull 		ventiaddress = EARGF(usage());
 73 a0d146ed 2005-07-12 devnull 		break;
 74 a0d146ed 2005-07-12 devnull 	default:
 75 a0d146ed 2005-07-12 devnull 		usage();
 76 a0d146ed 2005-07-12 devnull 	}ARGEND
 77 a0d146ed 2005-07-12 devnull 
 78 a0d146ed 2005-07-12 devnull 	if((z = vtdial(ventiaddress)) == nil)
 79 a0d146ed 2005-07-12 devnull 		sysfatal("vtdial %s: %r", ventiaddress);
 80 a0d146ed 2005-07-12 devnull 	if(vtconnect(z) < 0)
 81 a0d146ed 2005-07-12 devnull 		sysfatal("vtconnect: %r");
 82 a0d146ed 2005-07-12 devnull 
 83 a0d146ed 2005-07-12 devnull 	srv = vtlisten(address);
 84 a0d146ed 2005-07-12 devnull 	if(srv == nil)
 85 a0d146ed 2005-07-12 devnull 		sysfatal("vtlisten %s: %r", address);
 86 a0d146ed 2005-07-12 devnull 
 87 a0d146ed 2005-07-12 devnull 	while((r = vtgetreq(srv)) != nil){
 88 a0d146ed 2005-07-12 devnull 		r->rx.msgtype = r->tx.msgtype+1;
 89 a0d146ed 2005-07-12 devnull 		if(verbose)
 90 a0d146ed 2005-07-12 devnull 			fprint(2, "<- %F\n", &r->tx);
 91 a0d146ed 2005-07-12 devnull 		switch(r->tx.msgtype){
 92 a0d146ed 2005-07-12 devnull 		case VtTping:
 93 a0d146ed 2005-07-12 devnull 			break;
 94 a0d146ed 2005-07-12 devnull 		case VtTgoodbye:
 95 a0d146ed 2005-07-12 devnull 			break;
 96 a0d146ed 2005-07-12 devnull 		case VtTread:
 97 a0d146ed 2005-07-12 devnull 			threadcreate(readthread, r, 16384);
 98 a0d146ed 2005-07-12 devnull 			continue;
 99 a0d146ed 2005-07-12 devnull 		case VtTwrite:
100 a0d146ed 2005-07-12 devnull 			r->rx.error = vtstrdup("read-only server");
101 a0d146ed 2005-07-12 devnull 			r->rx.msgtype = VtRerror;
102 a0d146ed 2005-07-12 devnull 			break;
103 a0d146ed 2005-07-12 devnull 		case VtTsync:
104 a0d146ed 2005-07-12 devnull 			break;
105 a0d146ed 2005-07-12 devnull 		}
106 a0d146ed 2005-07-12 devnull 		if(verbose)
107 a0d146ed 2005-07-12 devnull 			fprint(2, "-> %F\n", &r->rx);
108 a0d146ed 2005-07-12 devnull 		vtrespond(r);
109 a0d146ed 2005-07-12 devnull 	}
110 a0d146ed 2005-07-12 devnull 	threadexitsall(nil);
111 a0d146ed 2005-07-12 devnull }
112 a0d146ed 2005-07-12 devnull