#include <u.h>
#include <libc.h>
#include <fcall.h>
#include <thread.h>
#include <libsec.h>
#include <9p.h>
#include "cifs.h"

#define max(a,b)	(((a) > (b))? (a): (b))
#define min(a,b)	(((a) < (b))? (a): (b))

typedef struct Aux Aux;
struct Aux {
	Aux	*next;
	Aux	*prev;
	char	*path;		/* full path fo file */
	Share	*sp;		/* this share's info */
	long	expire;		/* expiration time of cache */
	long	off;		/* file pos of start of cache */
	long	end;		/* file pos of end of cache */
	long	mtime;		/* last modification time - windows updates it only on close */
	char	*cache;
	int	fh;		/* file handle */
	int	sh;		/* search handle */
	long	srch;		/* find first's internal state */
};

extern int chatty9p;

int Checkcase = 1;		/* enforce case significance on filenames */
int Dfstout = 100;		/* timeout (in ms) for ping of dfs servers (assume they are local)  */
int Billtrog = 1;		/* enable file owner/group resolution */
int Attachpid;			/* pid of proc that attaches (ugh !) */
char *Debug = nil;		/* messages */
Qid Root;			/* root of remote system */
Share Ipc;			/* Share info of IPC$ share */
Session *Sess;			/* current session */
int Active = IDLE_TIME;		/* secs until next keepalive is sent */
static int Keeppid;		/* process ID of keepalive thread */
Share Shares[MAX_SHARES]; 	/* table of connected shares */
int Nshares = 0;		/* number of Shares connected */
Aux *Openfiles = nil;		/* linked list of Aux structs */
char *Host = nil;		/* host we are connected to */

static char *Ipcname = "IPC$";

#define ptype(x)	(((x) & 0xf))
#define pindex(x)	(((x) & 0xff0) >> 4)

void
setup(void)
{
	/*
	 * This is revolting but I cannot see any other way to get
	 * the pid of the server.  We need this as Windows doesn't
	 * drop the TCP connection when it closes a connection.
	 * Thus we keepalive() to detect when/if we are thrown off.
	 */
	Attachpid = getpid();

	procsetname("%s network", Host);
}

int
filetableinfo(Fmt *f)
{
	Aux *ap;
	char *type;

	if((ap = Openfiles) != nil)
		do{
			type = "walked";
			if(ap->sh != -1)
				type = "opendir";
			if(ap->fh != -1)
				type = "openfile";
			fmtprint(f, "%-9s %s\n", type, ap->path);
			ap = ap->next;
		}while(ap != Openfiles);
	return 0;
}

Qid
mkqid(char *s, int is_dir, long vers, int subtype, long path)
{
	Qid q;
	union {				/* align digest suitably */
		uchar	digest[SHA1dlen];
		uvlong	uvl;
	} u;

	sha1((uchar *)s, strlen(s), u.digest, nil);
	q.type = (is_dir)? QTDIR: 0;
	q.vers = vers;
	if(subtype){
		q.path = *((uvlong *)u.digest) & ~0xfffL;
		q.path |= ((path & 0xff) << 4);
		q.path |= (subtype & 0xf);
	}
	else
		q.path = *((uvlong *)u.digest) & ~0xfL;
	return q;
}

/*
 * used only for root dir and shares
 */
static void
V2D(Dir *d, Qid qid, char *name)
{
	memset(d, 0, sizeof(Dir));
	d->type = 'C';
	d->dev = 1;
	d->name = estrdup9p(name);
	d->uid = estrdup9p("bill");
	d->muid = estrdup9p("boyd");
	d->gid = estrdup9p("trog");
	d->mode = 0755 | DMDIR;
	d->atime = time(nil);
	d->mtime = d->atime;
	d->length = 0;
	d->qid = qid;
}

static void
I2D(Dir *d, Share *sp, char *path, long mtime, FInfo *fi)
{
	char *name;

	if((name = strrchr(fi->name, '\\')) != nil)
		name++;
	else
		name = fi->name;
	d->name = estrdup9p(name);
	d->type = 'C';
	d->dev = sp->tid;
	d->uid = estrdup9p("bill");
	d->gid = estrdup9p("trog");
	d->muid = estrdup9p("boyd");
	d->atime = fi->accessed;
	if(mtime > fi->written)
		d->mtime = mtime;
	else
		d->mtime = fi->written;

	if(fi->attribs & ATTR_READONLY)
		d->mode = 0444;
	else
		d->mode = 0666;

	d->length = fi->size;
	d->qid = mkqid(path, fi->attribs & ATTR_DIRECTORY, fi->changed, 0, 0);

	if(fi->attribs & ATTR_DIRECTORY){
		d->length = 0;
		d->mode |= DMDIR|0111;
	}
}

static void
responderrstr(Req *r)
{
	char e[ERRMAX];

	*e = 0;
	rerrstr(e, sizeof e);
	respond(r, e);
}

static char *
newpath(char *path, char *name)
{
	char *p, *q;

	assert((p = strrchr(path, '/')) != nil);

	if(strcmp(name, "..") == 0){
		if(p == path)
			return estrdup9p("/");
		q = emalloc9p((p-path)+1);
		strecpy(q, q+(p-path)+1, path);
		return q;
	}
	if(strcmp(path, "/") == 0)
		return smprint("/%s", name);
	return smprint("%s/%s", path, name);
}

/*
 * get the last write time if the file is open -
 * windows only updates mtime when the file is closed
 * which is not good enough for acme.
 */
static long
realmtime(char *path)
{
	Aux *a;

	if((a = Openfiles) == nil)
		return 0;

	do{
		if(a->fh != -1 && cistrcmp(path, a->path) == 0)
			return a->mtime;
		a = a->next;
	}while(a != Openfiles);
	return 0;
}

/* remove "." and ".." from the cache */
static int
rmdots(Aux *a, int got)
{
	int i, num;
	FInfo *fi;

	num = 0;
	fi = (FInfo *)a->cache;
	for(i = 0; i < got; i++){
		if(strcmp(fi->name, ".") == 0 || strcmp(fi->name, "..") == 0){
			memmove(fi, fi+1, got * sizeof(FInfo));
			continue;
		}
		fi++;
		num++;
	}

	return num;
}

static int
dirgen(int slot, Dir *d, void *aux)
{
	long off;
	FInfo *fi;
	int rc, got;
	Aux *a = aux;
	char *npath;
	int numinf = numinfo();
	int slots;

	slots = 32;		/* number of dir entries to fetch at one time */

	if(strcmp(a->path, "/") == 0){
		if(slot < numinf){
			dirgeninfo(slot, d);
			return 0;
		} else
			slot -= numinf;

		if(slot >= Nshares)
			return -1;
		V2D(d, mkqid(Shares[slot].name, 1, 1, Pshare, slot),
			Shares[slot].name);
		return 0;
	}

	off = slot * sizeof(FInfo);
	if(off >= a->off && off < a->end && time(nil) < a->expire)
		goto from_cache;

	if(off == 0){
		npath = smprint("%s/*", mapfile(a->path));
		a->sh = T2findfirst(Sess, a->sp, slots, npath, &got, &a->srch,
			(FInfo *)a->cache);
		free(npath);
		if(a->sh == -1)
			return -1;

		got = rmdots(a, got);
		a->off = 0;
		a->end = got * sizeof(FInfo);
		goto from_cache;
	}

	while(off >= a->end && a->sh != -1){
		fi = (FInfo *)(a->cache + (a->end - a->off) - sizeof(FInfo));
		a->off = a->end;
		npath = smprint("%s/%s", mapfile(a->path), fi->name);
		rc = T2findnext(Sess, a->sp, slots, npath,
			&got, &a->srch, (FInfo *)a->cache, a->sh);
		free(npath);
		if(rc == -1 || got == 0)
			break;
		got = rmdots(a, got);
		a->end = a->off + got * sizeof(FInfo);
	}
	a->expire = time(nil) + CACHETIME;

	if(got < slots){
		if(a->sh != -1)
			CIFSfindclose2(Sess, a->sp, a->sh);
		a->sh = -1;
	}


from_cache:
	if(off >= a->end)
		return -1;

	fi = (FInfo *)(a->cache + (off - a->off));
	npath = smprint("%s/%s", mapfile(a->path), fi->name);
	I2D(d, a->sp, npath, realmtime(npath), fi);
	if(Billtrog == 0)
		upd_names(Sess, a->sp, npath, d);
	free(npath);
	return 0;
}

static void
fsattach(Req *r)
{
	Aux *a;
	static int first = 1;
	char *spec = r->ifcall.aname;

	if(first)
		setup();

	if(spec && *spec){
		respond(r, "invalid attach specifier");
		return;
	}

	r->ofcall.qid = mkqid("/", 1, 1, Proot, 0);
	r->fid->qid = r->ofcall.qid;

	a = r->fid->aux = emalloc9p(sizeof(Aux));
	a->path = estrdup9p("/");
	a->sp = nil;
	a->fh = -1;
	a->sh = -1;

	if(Openfiles){
		a->prev = Openfiles;
		a->next = Openfiles->next;
		Openfiles->next->prev = a;
		Openfiles->next = a;
	} else {
		Openfiles = a;
		a->next = a;
		a->prev = a;
	}
	respond(r, nil);
}

static char*
fsclone(Fid *ofid, Fid *fid)
{
	Aux *oa = ofid->aux;
	Aux *a = emalloc9p(sizeof(Aux));

	fid->aux = a;

	a->sh = -1;
	a->fh = -1;
	a->sp = oa->sp;
	a->path = estrdup9p(oa->path);

	if(Openfiles){
		a->prev = Openfiles;
		a->next = Openfiles->next;
		Openfiles->next->prev = a;
		Openfiles->next = a;
	} else {
		Openfiles = a;
		a->next = a;
		a->prev = a;
	}
	return nil;
}

/*
 * for some weird reason T2queryall() returns share names
 * in lower case so we have to do an extra test against
 * our share table to validate filename case.
 *
 * on top of this here (snell & Wilcox) most of our
 * redirections point to a share of the same name,
 * but some do not, thus the tail of the filename
 * returned by T2queryall() is not the same as
 * the name we wanted.
 *
 * We work around this by not validating the names
 * or files which resolve to share names as they must
 * be correct, having been enforced in the dfs layer.
 */
static int
validfile(char *found, char *want, char *winpath, Share *sp)
{
	char *share;

	if(strcmp(want, "..") == 0)
		return 1;
	if(strcmp(winpath, "/") == 0){
		share = trimshare(sp->name);
		if(cistrcmp(want, share) == 0)
			return strcmp(want, share) == 0;
		/*
		 * OK, a DFS redirection points us from a directory XXX
		 * to a share named YYY.  There is no case checking we can
		 * do so we allow either case - it's all we can do.
		 */
		return 1;
	}
	if(cistrcmp(found, want) != 0)
		return 0;
	if(!Checkcase)
		return 1;
	if(strcmp(found, want) == 0)
		return 1;
	return 0;
}


static char*
fswalk1(Fid *fid, char *name, Qid *qid)
{
	FInfo fi;
	int rc, n, i;
	Aux *a = fid->aux;
	static char e[ERRMAX];
	char *p, *npath, *winpath;

	*e = 0;
	npath = newpath(a->path, name);
	if(strcmp(npath, "/") == 0){			/* root dir */
		*qid = mkqid("/", 1, 1, Proot, 0);
		free(a->path);
		a->path = npath;
		fid->qid = *qid;
		return nil;
	}

	if(strrchr(npath, '/') == npath){		/* top level dir */
		if((n = walkinfo(name)) != -1){		/* info file */
			*qid = mkqid(npath, 0, 1, Pinfo, n);
		}
		else {					/* volume name */
			for(i = 0; i < Nshares; i++){
				n = strlen(Shares[i].name);
				if(cistrncmp(npath+1, Shares[i].name, n) != 0)
					continue;
				if(Checkcase && strncmp(npath+1, Shares[i].name, n) != 0)
					continue;
				if(npath[n+1] != 0 && npath[n+1] != '/')
					continue;
				break;
			}
			if(i >= Nshares){
				free(npath);
				return "not found";
			}
			a->sp = Shares+i;
			*qid = mkqid(npath, 1, 1, Pshare, i);
		}
		free(a->path);
		a->path = npath;
		fid->qid = *qid;
		return nil;
	}

	/* must be a vanilla file or directory */
again:
	if(mapshare(npath, &a->sp) == -1){
		rerrstr(e, sizeof(e));
		free(npath);
		return e;
	}

	winpath = mapfile(npath);
	memset(&fi, 0, sizeof fi);
	if(Sess->caps & CAP_NT_SMBS)
		rc = T2queryall(Sess, a->sp, winpath, &fi);
	else
		rc = T2querystandard(Sess, a->sp, winpath, &fi);

	if(rc == -1){
		rerrstr(e, sizeof(e));
		free(npath);
		return e;
	}

	if((a->sp->options & SMB_SHARE_IS_IN_DFS) != 0 &&
	    (fi.attribs & ATTR_REPARSE) != 0){
		if(redirect(Sess, a->sp, npath) != -1)
			goto again;
	}

	if((p = strrchr(fi.name, '/')) == nil && (p = strrchr(fi.name, '\\')) == nil)
		p = fi.name;
	else
		p++;

	if(! validfile(p, name, winpath, a->sp)){
		free(npath);
		return "not found";

	}
	*qid = mkqid(npath, fi.attribs & ATTR_DIRECTORY, fi.changed, 0, 0);

	a->mtime = realmtime(npath);

	free(a->path);
	a->path = npath;
	fid->qid = *qid;
	return nil;
}

static void
fsstat(Req *r)
{
	int rc;
	FInfo fi;
	Aux *a = r->fid->aux;

	if(ptype(r->fid->qid.path) == Proot)
		V2D(&r->d, r->fid->qid, "");
	else if(ptype(r->fid->qid.path) == Pinfo)
		dirgeninfo(pindex(r->fid->qid.path), &r->d);
	else if(ptype(r->fid->qid.path) == Pshare)
		V2D(&r->d, r->fid->qid, a->path +1);
	else{
		memset(&fi, 0, sizeof fi);
		if(Sess->caps & CAP_NT_SMBS)
			rc = T2queryall(Sess, a->sp, mapfile(a->path), &fi);
		else
			rc = T2querystandard(Sess, a->sp, mapfile(a->path), &fi);
		if(rc == -1){
			responderrstr(r);
			return;
		}
		I2D(&r->d, a->sp, a->path, a->mtime, &fi);
		if(Billtrog == 0)
			upd_names(Sess, a->sp, mapfile(a->path), &r->d);
	}
	respond(r, nil);
}

static int
smbcreateopen(Aux *a, char *path, int mode, int perm, int is_create,
	int is_dir, FInfo *fip)
{
	int rc, action, attrs, access, result;

	if(is_create && is_dir){
		if(CIFScreatedirectory(Sess, a->sp, path) == -1)
			return -1;
		return 0;
	}

	if(mode & DMAPPEND) {
		werrstr("filesystem does not support DMAPPEND");
		return -1;
	}

	if(is_create)
		action = 0x12;
	else if(mode & OTRUNC)
		action = 0x02;
	else
		action = 0x01;

	if(perm & 0222)
		attrs = ATTR_NORMAL;
	else
		attrs = ATTR_NORMAL|ATTR_READONLY;

	switch (mode & OMASK){
	case OREAD:
		access = 0;
		break;
	case OWRITE:
		access = 1;
		break;
	case ORDWR:
		access = 2;
		break;
	case OEXEC:
		access = 3;
		break;
	default:
		werrstr("%d bad open mode", mode & OMASK);
		return -1;
		break;
	}

	if((mode & DMEXCL) == 0)
		access |= 0x10;
	else
		access |= 0x40;

	if((a->fh = CIFS_SMB_opencreate(Sess, a->sp, path, access, attrs,
	    action, &result)) == -1)
		return -1;

	if(Sess->caps & CAP_NT_SMBS)
		rc = T2queryall(Sess, a->sp, mapfile(a->path), fip);
	else
		rc = T2querystandard(Sess, a->sp, mapfile(a->path), fip);
	if(rc == -1){
		fprint(2, "internal error: stat of newly open/created file failed\n");
		return -1;
	}

	if((mode & OEXCL) && (result & 0x8000) == 0){
		werrstr("%d bad open mode", mode & OMASK);
		return -1;
	}
	return 0;
}

/* Uncle Bill, you have a lot to answer for... */
static int
ntcreateopen(Aux *a, char *path, int mode, int perm, int is_create,
	int is_dir, FInfo *fip)
{
	int options, result, attrs, flags, access, action, share;

	if(mode & DMAPPEND){
		werrstr("CIFSopen, DMAPPEND not supported");
		return -1;
	}

	if(is_create){
		if(mode & OEXCL)
			action = FILE_OPEN;
		else if(mode & OTRUNC)
			action = FILE_CREATE;
		else
			action = FILE_OVERWRITE_IF;
	} else {
		if(mode & OTRUNC)
			action = FILE_OVERWRITE_IF;
		else
			action = FILE_OPEN_IF;
	}

	flags = 0;		/* FIXME: really not sure */

	if(mode & OEXCL)
		share = FILE_NO_SHARE;
	else
		share = FILE_SHARE_ALL;

	switch (mode & OMASK){
	case OREAD:
		access = GENERIC_READ;
		break;
	case OWRITE:
		access = GENERIC_WRITE;
		break;
	case ORDWR:
		access = GENERIC_ALL;
		break;
	case OEXEC:
		access = GENERIC_EXECUTE;
		break;
	default:
		werrstr("%d bad open mode", mode & OMASK);
		return -1;
		break;
	}

	if(is_dir){
		action = FILE_CREATE;
		options = FILE_DIRECTORY_FILE;
		if(perm & 0222)
			attrs = ATTR_DIRECTORY;
		else
			attrs = ATTR_DIRECTORY|ATTR_READONLY;
	} else {
		options = FILE_NON_DIRECTORY_FILE;
		if(perm & 0222)
			attrs = ATTR_NORMAL;
		else
			attrs = ATTR_NORMAL|ATTR_READONLY;
	}

	if(mode & ORCLOSE){
		options |= FILE_DELETE_ON_CLOSE;
		attrs |= ATTR_DELETE_ON_CLOSE;
	}

	if((a->fh = CIFS_NT_opencreate(Sess, a->sp, path, flags, options,
	    attrs, access, share, action, &result, fip)) == -1)
		return -1;

	if((mode & OEXCL) && (result & 0x8000) == 0){
		werrstr("%d bad open mode", mode & OMASK);
		return -1;
	}

	return 0;
}

static void
fscreate(Req *r)
{
	FInfo fi;
	int rc, is_dir;
	char *npath;
	Aux *a = r->fid->aux;

	a->end = a->off = 0;
	a->cache = emalloc9p(max(Sess->mtu, MTU));

	is_dir = (r->ifcall.perm & DMDIR) == DMDIR;
	npath = smprint("%s/%s", a->path, r->ifcall.name);

	if(Sess->caps & CAP_NT_SMBS)
		rc = ntcreateopen(a, mapfile(npath), r->ifcall.mode,
			r->ifcall.perm, 1, is_dir, &fi);
	else
		rc = smbcreateopen(a, mapfile(npath), r->ifcall.mode,
			r->ifcall.perm, 1, is_dir, &fi);
	if(rc == -1){
		free(npath);
		responderrstr(r);
		return;
	}

	r->fid->qid = mkqid(npath, fi.attribs & ATTR_DIRECTORY, fi.changed, 0, 0);

	r->ofcall.qid = r->fid->qid;
	free(a->path);
	a->path = npath;

	respond(r, nil);
}

static void
fsopen(Req *r)
{
	int rc;
	FInfo fi;
	Aux *a = r->fid->aux;

	a->end = a->off = 0;
	a->cache = emalloc9p(max(Sess->mtu, MTU));

	if(ptype(r->fid->qid.path) == Pinfo){
		if(makeinfo(pindex(r->fid->qid.path)) != -1)
			respond(r, nil);
		else
			respond(r, "cannot generate info");
		return;
	}

	if(r->fid->qid.type & QTDIR){
		respond(r, nil);
		return;
	}

	if(Sess->caps & CAP_NT_SMBS)
		rc = ntcreateopen(a, mapfile(a->path), r->ifcall.mode, 0777,
			0, 0, &fi);
	else
		rc = smbcreateopen(a, mapfile(a->path), r->ifcall.mode, 0777,
			0, 0, &fi);
	if(rc == -1){
		responderrstr(r);
		return;
	}
	respond(r, nil);
}

static void
fswrite(Req *r)
{
	vlong n, m, got;
	Aux *a = r->fid->aux;
	vlong len = r->ifcall.count;
	vlong off = r->ifcall.offset;
	char *buf = r->ifcall.data;

	got = 0;
	n = Sess->mtu -OVERHEAD;
	do{
		if(len - got < n)
			n = len - got;
		m = CIFSwrite(Sess, a->sp, a->fh, off + got, buf + got, n);
		if(m != -1)
			got += m;
	} while(got < len && m >= n);

	r->ofcall.count = got;
	a->mtime = time(nil);

	if(m == -1)
		responderrstr(r);
	else
		respond(r, nil);
}

static void
fsread(Req *r)
{
	vlong n, m, got;
	Aux *a = r->fid->aux;
	char *buf = r->ofcall.data;
	vlong len = r->ifcall.count;
	vlong off = r->ifcall.offset;

	if(ptype(r->fid->qid.path) == Pinfo){
		r->ofcall.count = readinfo(pindex(r->fid->qid.path), buf, len,
			off);
		respond(r, nil);
		return;
	}

	if(r->fid->qid.type & QTDIR){
		dirread9p(r, dirgen, a);
		respond(r, nil);
		return;
	}

	got = 0;
	n = Sess->mtu -OVERHEAD;
	do{
		if(len - got < n)
			n = len - got;
		m = CIFSread(Sess, a->sp, a->fh, off + got, buf + got, n, len);
		if(m != -1)
			got += m;
	} while(got < len && m >= n);

	r->ofcall.count = got;
	if(m == -1)
		responderrstr(r);
	else
		respond(r, nil);
}

static void
fsdestroyfid(Fid *f)
{
	Aux *a = f->aux;

	if(ptype(f->qid.path) == Pinfo)
		freeinfo(pindex(f->qid.path));
	f->omode = -1;
	if(! a)
		return;
	if(a->fh != -1)
		if(CIFSclose(Sess, a->sp, a->fh) == -1)
			fprint(2, "%s: close failed fh=%d %r\n", argv0, a->fh);
	if(a->sh != -1)
		if(CIFSfindclose2(Sess, a->sp, a->sh) == -1)
			fprint(2, "%s: findclose failed sh=%d %r\n",
				argv0, a->sh);
	if(a->path)
		free(a->path);
	if(a->cache)
		free(a->cache);

	if(a == Openfiles)
		Openfiles = a->next;
	a->prev->next = a->next;
	a->next->prev = a->prev;
	if(a->next == a->prev)
		Openfiles = nil;
	if(a)
		free(a);
}

int
rdonly(Session *s, Share *sp, char *path, int rdonly)
{
	int rc;
	FInfo fi;

	if(Sess->caps & CAP_NT_SMBS)
		rc = T2queryall(s, sp, path, &fi);
	else
		rc = T2querystandard(s, sp, path, &fi);
	if(rc == -1)
		return -1;

	if((rdonly && !(fi.attribs & ATTR_READONLY)) ||
	    (!rdonly && (fi.attribs & ATTR_READONLY))){
		fi.attribs &= ~ATTR_READONLY;
		fi.attribs |= rdonly? ATTR_READONLY: 0;
		rc = CIFSsetinfo(s, sp, path, &fi);
	}
	return rc;
}

static void
fsremove(Req *r)
{
	int try, rc;
	char e[ERRMAX];
	Aux *ap, *a = r->fid->aux;

	*e = 0;
	if(ptype(r->fid->qid.path) == Proot ||
	   ptype(r->fid->qid.path) == Pshare){
		respond(r, "illegal operation");
		return;
	}

	/* close all instences of this file/dir */
	if((ap = Openfiles) != nil)
		do{
			if(strcmp(ap->path, a->path) == 0){
				if(ap->sh != -1)
					CIFSfindclose2(Sess, ap->sp, ap->sh);
				ap->sh = -1;
				if(ap->fh != -1)
					CIFSclose(Sess, ap->sp, ap->fh);
				ap->fh = -1;
			}
			ap = ap->next;
		}while(ap != Openfiles);
	try = 0;
again:
	if(r->fid->qid.type & QTDIR)
		rc = CIFSdeletedirectory(Sess, a->sp, mapfile(a->path));
	else
		rc = CIFSdeletefile(Sess, a->sp, mapfile(a->path));

	rerrstr(e, sizeof(e));
	if(rc == -1 && try++ == 0 && strcmp(e, "permission denied") == 0 &&
	    rdonly(Sess, a->sp, mapfile(a->path), 0) == 0)
		goto again;
	if(rc == -1)
		responderrstr(r);
	else
		respond(r, nil);
}

static void
fswstat(Req *r)
{
	int fh, result, rc;
	FInfo fi, tmpfi;
	char *p, *from, *npath;
	Aux *a = r->fid->aux;

	if(ptype(r->fid->qid.path) == Proot ||
	   ptype(r->fid->qid.path) == Pshare){
		respond(r, "illegal operation");
		return;
	}

	if((r->d.uid && r->d.uid[0]) || (r->d.gid && r->d.gid[0])){
		respond(r, "cannot change ownership");
		return;
	}

	/*
	 * get current info
	 */
	if(Sess->caps & CAP_NT_SMBS)
		rc = T2queryall(Sess, a->sp, mapfile(a->path), &fi);
	else
		rc = T2querystandard(Sess, a->sp, mapfile(a->path), &fi);
	if(rc == -1){
		werrstr("(query) - %r");
		responderrstr(r);
		return;
	}

	/*
	 * always clear the readonly attribute if set,
	 * before trying to set any other fields.
	 * wstat() fails if the file/dir is readonly
	 * and this function is so full of races - who cares about one more?
	 */
	rdonly(Sess, a->sp, mapfile(a->path), 0);

	/*
	 * rename - one piece of joy, renaming open files
	 * is legal (sharing permitting).
	 */
	if(r->d.name && r->d.name[0]){
		if((p = strrchr(a->path, '/')) == nil){
			respond(r, "illegal path");
			return;
		}
		npath = emalloc9p((p-a->path)+strlen(r->d.name)+2);
		strecpy(npath, npath+(p- a->path)+2, a->path);
		strcat(npath, r->d.name);

		from = estrdup9p(mapfile(a->path));
		if(CIFSrename(Sess, a->sp, from, mapfile(npath)) == -1){
			werrstr("(rename) - %r");
			responderrstr(r);
			free(npath);
			free(from);
			return;
		}
		free(from);
		free(a->path);
		a->path = npath;
	}

	/*
	 * set the files length, do this before setting
	 * the file times as open() will alter them
	 */
	if(~r->d.length){
		fi.size = r->d.length;

		if(Sess->caps & CAP_NT_SMBS){
			if((fh = CIFS_NT_opencreate(Sess, a->sp, mapfile(a->path),
			    0, FILE_NON_DIRECTORY_FILE,
	    		    ATTR_NORMAL, GENERIC_WRITE, FILE_SHARE_ALL,
			    FILE_OPEN_IF, &result, &tmpfi)) == -1){
				werrstr("(set length, open) - %r");
				responderrstr(r);
				return;
			}
			rc = T2setfilelength(Sess, a->sp, fh, &fi);
			CIFSclose(Sess, a->sp, fh);
			if(rc == -1){
				werrstr("(set length), set) - %r");
				responderrstr(r);
				return;
			}
		} else {
			if((fh = CIFS_SMB_opencreate(Sess, a->sp, mapfile(a->path),
			    1, ATTR_NORMAL, 1, &result)) == -1){
				werrstr("(set length, open) failed - %r");
				responderrstr(r);
				return;
			}
			rc = CIFSwrite(Sess, a->sp, fh, fi.size, 0, 0);
			CIFSclose(Sess, a->sp, fh);
			if(rc == -1){
				werrstr("(set length, write) - %r");
				responderrstr(r);
				return;
			}
		}
	}

	/*
	 * This doesn't appear to set length or
	 * attributes, no idea why, so I do those seperately
	 */
	if(~r->d.mtime || ~r->d.atime){
		if(~r->d.mtime)
			fi.written = r->d.mtime;
		if(~r->d.atime)
			fi.accessed = r->d.atime;
		if(T2setpathinfo(Sess, a->sp, mapfile(a->path), &fi) == -1){
			werrstr("(set path info) - %r");
			responderrstr(r);
			return;
		}
	}

	/*
	 * always update the readonly flag as
	 * we may have cleared it above.
	 */
	if(~r->d.mode){
		if(r->d.mode & 0222)
			fi.attribs &= ~ATTR_READONLY;
		else
			fi.attribs |= ATTR_READONLY;
	}
	if(rdonly(Sess, a->sp, mapfile(a->path), fi.attribs & ATTR_READONLY) == -1){
		werrstr("(set info) - %r");
		responderrstr(r);
		return;
	}

	/*
	 * Win95 has a broken write-behind cache for metadata
	 * on open files (writes go to the cache, reads bypass
	 * the cache), so we must flush the file.
	 */
	if(r->fid->omode != -1 && CIFSflush(Sess, a->sp, a->fh) == -1){
		werrstr("(flush) %r");
		responderrstr(r);
		return;
	}
	respond(r, nil);
}

static void
fsend(Srv *srv)
{
	int i;
	USED(srv);

	for(i = 0; i < Nshares; i++)
		CIFStreedisconnect(Sess, Shares+i);
	CIFSlogoff(Sess);
	postnote(PNPROC, Keeppid, "die");
}

Srv fs = {
	.destroyfid =	fsdestroyfid,
	.attach=	fsattach,
	.open=		fsopen,
	.create=	fscreate,
	.read=		fsread,
	.write=		fswrite,
	.remove=	fsremove,
	.stat=		fsstat,
	.wstat=		fswstat,
	.clone= 	fsclone,
	.walk1= 	fswalk1,
	.end=		fsend,
};

void
usage(void)
{
	fprint(2, "usage: %s [-d name] [-Dvb] [-a auth-method] [-s srvname] "
		"[-n called-name] [-k factotum-params] [-m mntpnt] "
		"host [share...]\n", argv0);
	exits("usage");
}

/*
 * SMBecho looks like the function to use for keepalives,
 * sadly the echo packet does not seem to reload the
 * idle timer in Microsoft's servers.  Instead we use
 * "get file system size" on each share until we get one that succeeds.
 */
static void
keepalive(void)
{
	uvlong tot, fre;
	int i, slot, rc;

	procsetname("%s keepalive", Host);

	rc = 0;
	slot = 0;
	do{
		sleep(6000);
		if(Active-- != 0)
			continue;

		for(i = 0; i < Nshares; i++){
			if((rc = T2fssizeinfo(Sess, &Shares[slot], &tot, &fre)) == 0)
				break;
			if(++slot >= Nshares)
				slot = 0;
		}
	}while(rc != -1);
	postnote(PNPROC, Attachpid, "die");
}


static void
ding(void *u, char *msg)
{
	USED(u);
	if(strstr(msg, "alarm") != nil)
		noted(NCONT);
	noted(NDFLT);
}

void
dmpkey(char *s, void *v, int n)
{
	int i;
	unsigned char *p = (unsigned char *)v;

	print("%s", s);
	for(i = 0; i < n; i++)
		print("%02ux ", *p++);
	print("\n");
}

void
main(int argc, char **argv)
{
	int i, n, local;
	long svrtime;
	char windom[64], cname[64];
	char *p, *method, *sysname, *keyp, *mtpt, *svs;
	static char *sh[1024];

	local = 0;
	*cname = 0;
	keyp = "";
	method = nil;
	strcpy(windom, "unknown");
	mtpt = svs = nil;

	notify(ding);

	ARGBEGIN{
	case 'a':
		method = EARGF(autherr());
		break;
	case 'b':
		Billtrog ^= 1;
		break;
	case 'D':
		chatty9p++;
		break;
	case 'd':
		Debug = EARGF(usage());
		break;
	case 'i':
		Checkcase = 0;
		break;
	case 'k':
		keyp = EARGF(usage());
		break;
	case 'l':
		local++;
		break;
	case 'm':
		mtpt = EARGF(usage());
		break;
	case 'n':
		strncpy(cname, EARGF(usage()), sizeof(cname));
		cname[sizeof(cname) - 1] = 0;
		break;
	case 's':
		svs = EARGF(usage());
		break;
	case 't':
		Dfstout = atoi(EARGF(usage()));
		break;
	default:
		usage();
		break;
	}ARGEND

	if(argc < 1)
		usage();

	Host = argv[0];

	if(mtpt == nil && svs == nil){
		if((p = strchr(Host, '!')) != nil)
			mtpt = smprint("/n/%s", p+1);
		else
			mtpt = smprint("/n/%s", Host);
	}

	if((sysname = getenv("sysname")) == nil)
		sysname = "unknown";

	if(*cname && (Sess = cifsdial(Host, cname, sysname)) != nil)
		goto connected;

	if(calledname(Host, cname) == 0 &&
	    (Sess = cifsdial(Host, cname, sysname)) != nil)
		goto connected;

	strcpy(cname, Host);
	if((p = strchr(cname, '!')) != nil)
		strcpy(cname, p+1);

	if((Sess = cifsdial(Host, cname, sysname)) != nil ||
	   (Sess = cifsdial(Host, "*SMBSERVER", sysname)) != nil)
		goto connected;

	sysfatal("%s - cannot dial, %r", Host);
connected:
	if(CIFSnegotiate(Sess, &svrtime, windom, sizeof windom, cname, sizeof cname) == -1)
		sysfatal("%s - cannot negioate common protocol, %r", Host);

#ifndef DEBUG_MAC
	Sess->secmode &= ~SECMODE_SIGN_ENABLED;
#endif

	if(local)
		strcpy(windom, ".");

	Sess->auth = getauth(method, windom, keyp, Sess->secmode, Sess->chal,
		Sess->challen);

	if(CIFSsession(Sess) < 0)
		sysfatal("session authentication failed, %r");

	Sess->slip = svrtime - time(nil);
	Sess->cname = estrdup9p(cname);

	if(CIFStreeconnect(Sess, cname, Ipcname, &Ipc) == -1)
		fprint(2, "%s, %r - can't connect\n", Ipcname);

	Nshares = 0;
	if(argc == 1){
		Share *sip;

		if((n = RAPshareenum(Sess, &Ipc, &sip)) < 1)
			sysfatal("can't enumerate shares: %r - specify share "
				"names on command line\n");

		for(i = 0; i < n; i++){
#ifdef NO_HIDDEN_SHARES
			int l = strlen(sip[i].name);

			if(l > 1 && sip[i].name[l-1] == '$'){
				free(sip[i].name);
				continue;
			}
#endif
			memcpy(Shares+Nshares, sip+i, sizeof(Share));
			if(CIFStreeconnect(Sess, Sess->cname,
			    Shares[Nshares].name, Shares+Nshares) == -1){
				fprint(2, "%s: %s %q - can't connect to share"
					", %r\n", argv0, Host, Shares[Nshares].name);
				free(Shares[Nshares].name);
				continue;
			}
			Nshares++;
		}
		free(sip);
	} else
		for(i = 1; i < argc; i++){
			if(CIFStreeconnect(Sess, Sess->cname, argv[i],
			    Shares+Nshares) == -1){
				fprint(2, "%s: %s %q - can't connect to share"
					", %r\n", argv0, Host, argv[i]);
				continue;
			}
			Shares[Nshares].name = strlwr(estrdup9p(argv[i]));
			Nshares++;
		}

	if(Nshares == 0)
		fprint(2, "no available shares\n");

	if((i = rfork(RFPROC|RFMEM|RFNOTEG|RFFDG|RFNAMEG)) == 0){
		keepalive();
		exits(nil);
	}
	Keeppid = i;
	postmountsrv(&fs, svs, mtpt, MREPL|MCREATE);
	exits(nil);
}
