harvey/sys/src/cmd/aquarela/cifscmd.c

#include "headers.h"
#include <bio.h>

SmbClient *c;
Biobuf bin, bout;
static int verbose = 1;

#define	SEP(c)	(((c)==' ')||((c)=='\t')||((c)=='\n'))

typedef struct Slut {
	char *name;
	int val;
} Slut;

static char *
tokenise(char *s, char **start, char **end)
{
	char *to;
	Rune r;
	int n;

	while(*s && SEP(*s))				/* skip leading white space */
		s++;
	to = *start = s;
	while(*s){
		n = chartorune(&r, s);
		if(SEP(r)){
			if(to != *start)		/* we have data */
				break;
			s += n;				/* null string - keep looking */
			while(*s && SEP(*s))	
				s++;
			to = *start = s;
		} 
		else if(r == '\''){
			s += n;				/* skip leading quote */
			while(*s){
				n = chartorune(&r, s);
				if(r == '\''){
					if(s[1] != '\'')
						break;
					s++;		/* embedded quote */
				}
				while (n--)
					*to++ = *s++;
			}
			if(!*s)				/* no trailing quote */
				break;
			s++;				/* skip trailing quote */ 
		}
		else  {
			while(n--)
				*to++ = *s++;
		}
	}
	*end = to;
	return s;
}

static int
parse(char *s, char *fields[], int nfields)
{
	int c, argc;
	char *start, *end;

	argc = 0;
	c = *s;
	while(c){
		s = tokenise(s, &start, &end);
		c = *s++;
		if(*start == 0)
			break;
		if(argc >= nfields-1)
			return -1;
		*end = 0;
		fields[argc++] = start;
	}
	fields[argc] = 0;
Bprint(&bout, "parse returns %d\n", argc);
	return argc;
}

typedef struct {
	char *name;
	long (*f)(SmbClient *, int, char *[]);
	int connected;
	char *help;
} Cmd;
static Cmd cmd[];

static long
cmdhelp(SmbClient *, int argc, char *argv[])
{
	Cmd *cp;
	char *p;

	if(argc)
		p = argv[0];
	else
		p = 0;
	for (cp = cmd; cp->name; cp++) {
		if (p == 0 || strcmp(p, cp->name) == 0)
			Bprint(&bout, "%s\n", cp->help);
	}
	return 0;
}

static Slut sharemodeslut[] = {
	{ "compatibility", SMB_OPEN_MODE_SHARE_COMPATIBILITY },
	{ "exclusive", SMB_OPEN_MODE_SHARE_EXCLUSIVE },
	{ "denywrite", SMB_OPEN_MODE_SHARE_DENY_WRITE },
	{ "denyread", SMB_OPEN_MODE_SHARE_DENY_READOREXEC },
	{ "denynone", SMB_OPEN_MODE_SHARE_DENY_NONE },
	{ 0 }
};

static Slut openmodeslut[] = {
	{ "oread", OREAD },
	{ "owrite", OWRITE },
	{ "ordwr", ORDWR },
	{ "oexec", OEXEC },
	{ 0 }
};

static int
slut(Slut *s, char *pat)
{
	while (s->name) {
		if (cistrcmp(s->name, pat) == 0)
			return s->val;
		s++;
	}
	Bprint(&bout, "%s unrecognised\n", pat);
	return -1;	
}

static long
cmdopen(SmbClient *c, int argc, char *argv[])
{
	char *errmsg;
	int sm, om;
	int rv;
	uchar errclass;
	ushort error;
	ushort fid, attr;
	ulong mtime, size;
	ushort accessallowed;

	if (argc != 3) {
		Bprint(&bout, "wrong number of arguments\n");
		return -1;
	}
	sm = slut(sharemodeslut, argv[1]);
	if (sm < 0) 
		return -1;
	om = slut(openmodeslut, argv[2]);
	if (om < 0)
		return -1;
	errmsg = nil;
	rv = smbclientopen(c, (sm << 3) | om, argv[0], &errclass, &error, &fid, &attr, &mtime, &size, &accessallowed, &errmsg);
	if (rv == 0) {
		if (errmsg) {
			print("local error %s\n", errmsg);
			free(errmsg);
			return -1;
		}
		return (errclass << 16) | error;
	}
	print("fid 0x%.4ux attr 0x%.4ux time %ld size %lud accessallowed %ud\n",
		fid, attr, mtime, size, accessallowed);
	return 0;
}

static Cmd cmd[] = {
	{ "help",	cmdhelp,	0, "help" },
	{ "open", cmdopen, 1, "open name sharemode openmode" },
	{ 0, 0 },
};

void
threadmain(int argc, char *argv[])
{
	char *errmsg;
	int ac;
	char *ap, *av[256];
	Cmd *cp;
	long status;

	if (argc > 3) {
		print("usage: cifscmd [to [share]]\n");
		exits("args");
	}
	smbglobalsguess(1);
	errmsg = nil;

	if (Binit(&bin, 0, OREAD) == Beof || Binit(&bout, 1, OWRITE) == Beof) {
		fprint(2, "%s: can't init bio: %r\n", argv0);
		threadexits("Binit");
	}

	if (argc > 1) {
		c = smbconnect(argv[1], argc == 3 ? argv[2] : nil, &errmsg);
		if (c == nil)
			fprint(2, "failed to connect: %s\n", errmsg);
	}
	while (ap = Brdline(&bin, '\n')) {
		ap[Blinelen(&bin) - 1] = 0;
		switch (ac = parse(ap, av, nelem(av))) {
		default:
			for (cp = cmd; cp->name; cp++) {
				if (strcmp(cp->name, av[0]) == 0)
					break;
			}
			if (cp->name == 0) {
				Bprint(&bout, "eh?\n");
				break;
			}
			if (c == 0 && cp->connected) {
				Bprint(&bout, "not currently connected\n");
				break;
			}
			if ((status = (*cp->f)(c, ac - 1, &av[1])) != -1) {
				if(verbose)
					Bprint(&bout, "ok %ld/%ld\n", status >> 16, status & 0xffff);
				break;
			}
			break;

		case -1:
			Bprint(&bout, "eh?\n");
			break;

		case 0:
			break;
		}
		Bflush(&bout);
	}
	threadexits(0);
}