581 lines · 9.1 KB · C
RawDownload
#include <u.h>#include <signal.h>#include <libc.h>#include <bio.h>#include <fcall.h>#include <9pclient.h>#include <auth.h>#include <thread.h> char *addr; voidusage(void){	fprint(2, "usage: 9p [-n] [-a address] [-A aname] cmd args...\n");	fprint(2, "possible cmds:\n");	fprint(2, "	read name\n");	fprint(2, "	readfd name\n");	fprint(2, "	write [-l] name\n");	fprint(2, "	writefd name\n");	fprint(2, "	stat name\n");	fprint(2, "	rdwr name\n");	fprint(2, "	ls [-ldn] name\n");	fprint(2, "without -a, name elem/path means /path on server unix!$ns/elem\n");	threadexitsall("usage");} CFsys *(*nsmnt)(char*, char*) = nsamount;CFsys *(*fsmnt)(int, char*) = fsamount; char *aname;void xread(int, char**);void xwrite(int, char**);void xreadfd(int, char**);void xwritefd(int, char**);void xstat(int, char**);void xls(int, char**);void xrdwr(int, char**);void xrm(int, char**);void xcreate(int, char**);void xcon(int, char**); struct {	char *s;	void (*f)(int, char**);} cmds[] = {	"con", xcon,	"read", xread,	"write", xwrite,	"readfd", xreadfd,	"writefd", xwritefd,	"stat", xstat,	"rdwr", xrdwr,	"ls", xls,	"rm", xrm,	"create", xcreate,}; voidthreadmain(int argc, char **argv){	char *cmd;	int i; 	ARGBEGIN{	case 'A':		aname = EARGF(usage());		break;	case 'a':		addr = EARGF(usage());		if(strchr(addr, '!') == nil)			addr = netmkaddr(addr, "tcp", "9fs");		break;	case 'n':		nsmnt = nsmount;		fsmnt = fsmount;		break;	case 'D':		chatty9pclient = 1;		break;	default:		usage();	}ARGEND 	signal(SIGINT, SIG_DFL); 	if(argc < 1)		usage(); 	cmd = argv[0];	for(i=0; i<nelem(cmds); i++){		if(strcmp(cmds[i].s, cmd) == 0){			cmds[i].f(argc, argv);			threadexitsall(0);		}	}	usage();	} CFsys*xparse(char *name, char **path){	int fd;	char *p;	CFsys *fs; 	if(addr == nil){		p = strchr(name, '/');		if(p == nil)			p = name+strlen(name);		else			*p++ = 0;		*path = p;		fs = nsmnt(name, aname);		if(fs == nil)			sysfatal("mount: %r");	}else{		*path = name;		if((fd = dial(addr, nil, nil, nil)) < 0)			sysfatal("dial: %r");		if((fs = fsmnt(fd, aname)) == nil)			sysfatal("mount: %r");	}	return fs;} CFid*xopen(char *name, int mode){	CFid *fid;	CFsys *fs; 	fs = xparse(name, &name);	fid = fsopen(fs, name, mode);	if(fid == nil)		sysfatal("fsopen %s: %r", name);	return fid;} intxopenfd(char *name, int mode){	CFsys *fs; 	fs = xparse(name, &name);	return fsopenfd(fs, name, mode);} voidxread(int argc, char **argv){	char buf[4096];	int n;	CFid *fid; 	ARGBEGIN{	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	fid = xopen(argv[0], OREAD);	while((n = fsread(fid, buf, sizeof buf)) > 0)		if(write(1, buf, n) < 0)			sysfatal("write error: %r");	fsclose(fid);	if(n < 0)		sysfatal("read error: %r");	threadexitsall(0);	} voidxreadfd(int argc, char **argv){	char buf[4096];	int n;	int fd; 	ARGBEGIN{	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	fd = xopenfd(argv[0], OREAD);	while((n = read(fd, buf, sizeof buf)) > 0)		if(write(1, buf, n) < 0)			sysfatal("write error: %r");	if(n < 0)		sysfatal("read error: %r");	threadexitsall(0);	} voidxwrite(int argc, char **argv){	char buf[4096];	int n, did;	CFid *fid;	Biobuf *b;	char *p;	int byline; 	byline = 0;	ARGBEGIN{	case 'l':		byline = 1;		break;	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	did = 0;	fid = xopen(argv[0], OWRITE|OTRUNC);	if(byline){		n = 0;		b = malloc(sizeof *b);		if(b == nil)			sysfatal("out of memory");		Binit(b, 0, OREAD);		while((p = Brdstr(b, '\n', 0)) != nil){			n = strlen(p);			did = 1;			if(fswrite(fid, p, n) != n)				fprint(2, "write: %r\n");		}		free(b);	}else{		while((n = read(0, buf, sizeof buf)) > 0){			did = 1;			if(fswrite(fid, buf, n) != n)				sysfatal("write error: %r");		}	}	if(n == 0 && !did){		if(fswrite(fid, buf, 0) != 0)			sysfatal("write error: %r");	}	if(n < 0)		sysfatal("read error: %r");	fsclose(fid);	threadexitsall(0);	} voidxwritefd(int argc, char **argv){	char buf[4096];	int n;	int fd; 	ARGBEGIN{	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	fd = xopenfd(argv[0], OWRITE|OTRUNC);	while((n = read(0, buf, sizeof buf)) > 0)		if(write(fd, buf, n) != n)			sysfatal("write error: %r");	if(n < 0)		sysfatal("read error: %r");	threadexitsall(0);	} voidxstat(int argc, char **argv){	Dir *d;	CFsys *fs;	char *name; 	ARGBEGIN{	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	name = argv[0];	fs = xparse(name, &name);	if((d = fsdirstat(fs, name)) == 0)		sysfatal("dirstat: %r");	fmtinstall('D', dirfmt);	fmtinstall('M', dirmodefmt);	print("%D\n", d);	threadexitsall(0);} voidxrdwr(int argc, char **argv){	char buf[4096];	int n;	CFid *fid; 	ARGBEGIN{	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	fid = xopen(argv[0], ORDWR);	for(;;){		fsseek(fid, 0, 0);		if((n = fsread(fid, buf, sizeof buf)) < 0)			fprint(2, "read: %r\n");		else{			if(write(1, buf, n) < 0 || write(1, "\n", 1) < 0)				sysfatal("write error: %r");		}		n = read(0, buf, sizeof buf);		if(n <= 0)			break;		if(buf[n-1] == '\n')			n--;		if(fswrite(fid, buf, n) != n)			fprint(2, "write: %r\n");	}	fsclose(fid);	threadexitsall(0);	} voidxcreate(int argc, char **argv){	int i;	CFsys *fs;	CFid *fid;	char *p; 	ARGBEGIN{	default:		usage();	}ARGEND		if(argc == 0)		usage();		for(i=0; i<argc; i++){		fs = xparse(argv[i], &p);		if((fid=fscreate(fs, p, OREAD, 0666)) == nil)			fprint(2, "create %s: %r\n", argv[i]);		else			fsclose(fid);		fsunmount(fs);	}} voidxrm(int argc, char **argv){	int i;	CFsys *fs;	char *p; 	ARGBEGIN{	default:		usage();	}ARGEND		if(argc == 0)		usage();		for(i=0; i<argc; i++){		fs = xparse(argv[i], &p);		if(fsremove(fs, p) < 0)			fprint(2, "remove %s: %r\n", argv[i]);		fsunmount(fs);	}} voidrdcon(void *v){	int n;	char buf[4096];	CFid *fid;		fid = v;	for(;;){		n = read(0, buf, sizeof buf);		if(n <= 0)			threadexitsall(0);		if(buf[0] == 'R'-'A'+1)			threadexitsall(0);		if(fswrite(fid, buf, n) != n)			fprint(2, "write: %r\n");	}} voidxcon(int argc, char **argv){	char buf[4096], *r, *w, *e;	int n, nocr;	CFid *fid;		nocr = 1; 	ARGBEGIN{	case 'r':		nocr = 0;		break;	default:		usage();	}ARGEND 	if(argc != 1)		usage(); 	fid = xopen(argv[0], ORDWR);	proccreate(rdcon, fid, 32768);	for(;;){		n = fsread(fid, buf, sizeof buf);		if(n <= 0)			threadexitsall(0);		if(nocr){			for(r=w=buf, e=buf+n; r<e; r++)				if(*r != '\r')					*w++ = *r;			n = w-buf;		}		if(write(1, buf, n) != n)			threadexitsall(0);	}	fsclose(fid);	threadexitsall(0);	} static char *mon[] = {	"Jan", "Feb", "Mar", "Apr", "May", "Jun", 	"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};  inttimefmt(Fmt *fmt){	ulong u;	static ulong time0;	Tm *tm;		if(time0 == 0)		time0 = time(0);	u = va_arg(fmt->args, ulong);	tm = localtime(u);	if((long)(time0-u) < 6*30*86400)		return fmtprint(fmt, "%s %2d %02d:%02d",			mon[tm->mon], tm->mday, tm->hour, tm->min);	return fmtprint(fmt, "%s %2d %5d",		mon[tm->mon], tm->mday, tm->year+1900);} static intdircmp(const void *va, const void *vb){	Dir *a, *b;		a = (Dir*)va;	b = (Dir*)vb;	return strcmp(a->name, b->name);} char *dot[] = { "." }; voidxls(int argc, char **argv){	char *err, *name, *xname, *f[4], buf[4096];	int nf, i, j, l, sort;	int lflag, dflag, n, len[4];	Dir *d;	CFid *fid;	CFsys *fs; 	err = nil;	sort = 0;	lflag = dflag = 0;	ARGBEGIN{	case 'n':		sort = 0;		break;	case 'l':		lflag = 1;		break;

Lines 1–500 of 581.

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t