Import sources from 2011-03-30 iso image

1 files changed, 185 insertions, 0 deletions

diff --git a/sys/src/cmd/9nfs/mport.c b/sys/src/cmd/9nfs/mport.c
new file mode 100755
index 000000000..553c2bede
--- /dev/null
+++ b/sys/src/cmd/9nfs/mport.c
@@ -0,0 +1,185 @@
+#include "all.h"
+
+typedef struct Rpcconn	Rpcconn;
+
+struct Rpcconn
+{
+	int	data;
+	int	ctl;
+	Rpccall	cmd;
+	Rpccall	reply;
+	uchar	rpcbuf[8192];
+	uchar	argbuf[8192];
+};
+
+void	putauth(char*, Auth*);
+int	rpccall(Rpcconn*, int);
+
+int	rpcdebug;
+
+Rpcconn	r;
+char *	mach;
+
+void
+main(int argc, char **argv)
+{
+	char addr[64], dir[64], name[128];
+	char buf[33], *p;
+	uchar *dataptr, *argptr;
+	int i, fd, n, remport;
+
+	ARGBEGIN{
+	case 'm':
+		mach = ARGF();
+		break;
+	case 'D':
+		++rpcdebug;
+		break;
+	}ARGEND
+	if(argc != 1)
+		exits("usage");
+
+	snprint(addr, sizeof addr, "udp!%s!111", argv[0]);
+	r.data = dial(addr, 0, dir, &r.ctl);
+	if(r.data < 0){
+		fprint(2, "dial %s: %r\n", addr);
+		exits("dial");
+	}
+	if(rpcdebug)
+		fprint(2, "dial %s: dir=%s\n", addr, dir);
+	if(fprint(r.ctl, "headers") < 0){
+		fprint(2, "can't set header mode: %r\n");
+		exits("headers");
+	}
+	sprint(name, "%s/remote", dir);
+	fd = open(name, OREAD);
+	if(fd < 0){
+		fprint(2, "can't open %s: %r\n", name);
+		exits("remote");
+	}
+	n = read(fd, buf, sizeof buf-1);
+	if(n < 0){
+		fprint(2, "can't read %s: %r\n", name);
+		exits("remote");
+	}
+	close(fd);
+	buf[n] = 0;
+	p = buf;
+	r.cmd.host = 0;
+	for(i=0; i<4; i++, p++)
+		r.cmd.host = (r.cmd.host<<8)|strtol(p, &p, 10);
+	r.cmd.port = strtol(p, 0, 10);
+	fprint(2, "host=%ld.%ld.%ld.%ld, port=%lud\n",
+		(r.cmd.host>>24)&0xff, (r.cmd.host>>16)&0xff,
+		(r.cmd.host>>8)&0xff, r.cmd.host&0xff, r.cmd.port);
+	fprint(r.ctl, "disconnect");
+
+	r.cmd.xid = time(0);
+	r.cmd.mtype = CALL;
+	r.cmd.rpcvers = 2;
+	r.cmd.args = r.argbuf;
+	if(mach)
+		putauth(mach, &r.cmd.cred);
+
+	r.cmd.prog = 100000;	/* portmapper */
+	r.cmd.vers = 2;		/* vers */
+	r.cmd.proc = 3;		/* getport */
+	dataptr = r.cmd.args;
+
+	PLONG(100005);		/* mount */
+	PLONG(1);		/* vers */
+	PLONG(IPPROTO_UDP);
+	PLONG(0);
+
+	i = rpccall(&r, dataptr-(uchar*)r.cmd.args);
+	if(i != 4)
+		exits("trouble");
+	argptr = r.reply.results;
+	remport = GLONG();
+	fprint(2, "remote port = %d\n", remport);
+
+	r.cmd.port = remport;
+	r.cmd.prog = 100005;	/* mount */
+	r.cmd.vers = 1;		/* vers */
+	r.cmd.proc = 0;		/* null */
+	dataptr = r.cmd.args;
+
+	i = rpccall(&r, dataptr-(uchar*)r.cmd.args);
+	if(i != 0)
+		exits("trouble");
+	fprint(2, "OK ping mount\n");
+
+	r.cmd.prog = 100005;	/* mount */
+	r.cmd.vers = 1;		/* vers */
+	r.cmd.proc = 5;		/* export */
+	dataptr = r.cmd.args;
+
+	i = rpccall(&r, dataptr-(uchar*)r.cmd.args);
+	fprint(2, "export: %d bytes returned\n", i);
+	argptr = r.reply.results;
+	while (GLONG() != 0) {
+		n = GLONG();
+		p = GPTR(n);
+		print("%.*s\n", utfnlen(p, n), p);
+		while (GLONG() != 0) {
+			n = GLONG();
+			p = GPTR(n);
+			print("\t%.*s\n", utfnlen(p, n), p);
+		}
+	}
+
+	exits(0);
+}
+
+void
+putauth(char *mach, Auth *a)
+{
+	uchar *dataptr;
+	long stamp = time(0);
+	int n = strlen(mach);
+
+	dataptr = realloc(a->data, 2*4+ROUNDUP(n)+4*4);
+	a->data = dataptr;
+	a->flavor = AUTH_UNIX;
+	PLONG(stamp);
+	PLONG(n);
+	PPTR(mach, n);
+	PLONG(0);
+	PLONG(1);
+	PLONG(1);
+	PLONG(0);
+	a->count = dataptr - (uchar*)a->data;
+}
+
+int
+rpccall(Rpcconn *r, int narg)
+{
+	int n;
+
+	r->cmd.xid++;
+	n = rpcS2M(&r->cmd, narg, r->rpcbuf);
+	if(rpcdebug)
+		rpcprint(2, &r->cmd);
+	if(write(r->data, r->rpcbuf, n) < 0){
+		fprint(2, "rpc write: %r\n");
+		exits("rpc");
+	}
+	n = read(r->data, r->rpcbuf, sizeof r->rpcbuf);
+	if(n < 0){
+		fprint(2, "rpc read: %r\n");
+		exits("rpc");
+	}
+	if(rpcM2S(r->rpcbuf, &r->reply, n) != 0){
+		fprint(2, "rpc reply format error\n");
+		exits("rpc");
+	}
+	if(rpcdebug)
+		rpcprint(2, &r->reply);
+	if(r->reply.mtype != REPLY || r->reply.stat != MSG_ACCEPTED ||
+	   r->reply.astat != SUCCESS){
+		fprint(2, "rpc mtype, stat, astat = %ld, %ld, %ld\n",
+			r->reply.mtype, r->reply.stat, r->reply.astat);
+		exits("rpc");
+	}
+	return n - (((uchar *)r->reply.results) - r->rpcbuf);
+}