Rev 2 | Blame | Compare with Previous | Last modification | View Log | RSS feed
#include "u.h"
#include "lib.h"
#include "dat.h"
#include "fns.h"
#include "error.h"
Chan*
fdtochan(int fd, int mode, int chkmnt, int iref)
{
Fgrp *f;
Chan *c;
c = 0;
f = up->fgrp;
lock(&f->ref.lk);
if(fd<0 || NFD<=fd || (c = f->fd[fd])==0) {
unlock(&f->ref.lk);
error(Ebadfd);
}
if(iref)
refinc(&c->ref);
unlock(&f->ref.lk);
if(chkmnt && (c->flag&CMSG))
goto bad;
if(mode<0 || c->mode==ORDWR)
return c;
if((mode&OTRUNC) && c->mode==OREAD)
goto bad;
if((mode&~OTRUNC) != c->mode)
goto bad;
return c;
bad:
if(iref)
cclose(c);
error(Ebadusefd);
return nil; /* shut up compiler */
}
static void
fdclose(int fd, int flag)
{
int i;
Chan *c;
Fgrp *f;
f = up->fgrp;
lock(&f->ref.lk);
c = f->fd[fd];
if(c == 0) {
unlock(&f->ref.lk);
return;
}
if(flag) {
if(c==0 || !(c->flag&flag)) {
unlock(&f->ref.lk);
return;
}
}
f->fd[fd] = 0;
if(fd == f->maxfd)
for(i=fd; --i>=0 && f->fd[i]==0; )
f->maxfd = i;
unlock(&f->ref.lk);
cclose(c);
}
static int
newfd(Chan *c)
{
int i;
Fgrp *f;
f = up->fgrp;
lock(&f->ref.lk);
for(i=0; i<NFD; i++)
if(f->fd[i] == 0){
if(i > f->maxfd)
f->maxfd = i;
f->fd[i] = c;
unlock(&f->ref.lk);
return i;
}
unlock(&f->ref.lk);
error("no file descriptors");
return 0;
}
int
sysclose(int fd)
{
if(waserror())
return -1;
fdtochan(fd, -1, 0, 0);
fdclose(fd, 0);
poperror();
return 0;
}
int
syscreate(char *path, int mode, ulong perm)
{
int fd;
Chan *c = 0;
if(waserror()) {
cclose(c);
return -1;
}
openmode(mode); /* error check only */
c = namec(path, Acreate, mode, perm);
fd = newfd((Chan*)c);
poperror();
return fd;
}
int
sysdup(int old, int new)
{
Chan *oc;
Fgrp *f = up->fgrp;
Chan *c = 0;
if(waserror())
return -1;
c = fdtochan(old, -1, 0, 1);
if(new != -1) {
if(new < 0 || NFD <= new) {
cclose(c);
error(Ebadfd);
}
lock(&f->ref.lk);
if(new > f->maxfd)
f->maxfd = new;
oc = f->fd[new];
f->fd[new] = (Chan*)c;
unlock(&f->ref.lk);
if(oc != 0)
cclose(oc);
}
else {
if(waserror()) {
cclose(c);
nexterror();
}
new = newfd((Chan*)c);
poperror();
}
poperror();
return new;
}
int
sysfstat(int fd, char *buf)
{
Chan *c = 0;
if(waserror()) {
cclose(c);
return -1;
}
c = fdtochan(fd, -1, 0, 1);
devtab[c->type]->stat((Chan*)c, buf);
poperror();
cclose(c);
return 0;
}
int
sysfwstat(int fd, char *buf)
{
Chan *c = 0;
if(waserror()) {
cclose(c);
return -1;
}
nameok(buf);
c = fdtochan(fd, -1, 1, 1);
devtab[c->type]->wstat((Chan*)c, buf);
poperror();
cclose(c);
return 0;
}
int
syschdir(char *dir)
{
return 0;
}
long
bindmount(Chan *c0, char *old, int flag, char *spec)
{
int ret;
Chan *c1 = 0;
if(flag>MMASK || (flag&MORDER) == (MBEFORE|MAFTER))
error(Ebadarg);
c1 = namec(old, Amount, 0, 0);
if(waserror()){
cclose(c1);
nexterror();
}
ret = cmount(c0, c1, flag, spec);
poperror();
cclose(c1);
return ret;
}
int
sysbind(char *new, char *old, int flags)
{
long r;
Chan *c0 = 0;
if(waserror()) {
cclose(c0);
return -1;
}
c0 = namec(new, Aaccess, 0, 0);
r = bindmount(c0, old, flags, "");
poperror();
cclose(c0);
return 0;
}
int
sysmount(int fd, char *old, int flags, char *spec)
{
long r;
Chan *c0 = 0, *bc = 0;
struct {
Chan* chan;
char* spec;
int flags;
} mntparam;
if(waserror()) {
cclose(bc);
cclose(c0);
return -1;
}
bc = fdtochan(fd, ORDWR, 0, 1);
mntparam.chan = (Chan*)bc;
mntparam.spec = spec;
mntparam.flags = flags;
c0 = (*devtab[devno('M', 0)].attach)(&mntparam);
cclose(bc);
r = bindmount(c0, old, flags, spec);
poperror();
cclose(c0);
return r;
}
int
sysunmount(char *old, char *new)
{
Chan *cmount = 0, *cmounted = 0;
if(waserror()) {
cclose(cmount);
cclose(cmounted);
return -1;
}
cmount = namec(new, Amount, OREAD, 0);
if(old != 0)
cmounted = namec(old, Aopen, OREAD, 0);
cunmount(cmount, cmounted);
poperror();
cclose(cmount);
cclose(cmounted);
return 0;
}
int
sysopen(char *path, int mode)
{
int fd;
Chan *c = 0;
if(waserror()){
cclose(c);
return -1;
}
openmode(mode); /* error check only */
c = namec(path, Aopen, mode, 0);
fd = newfd((Chan*)c);
poperror();
return fd;
}
long
unionread(Chan *c, void *va, long n)
{
long nr;
Chan *nc = 0;
Pgrp *pg = 0;
pg = up->pgrp;
rlock(&pg->ns);
for(;;) {
if(waserror()) {
runlock(&pg->ns);
nexterror();
}
nc = clone(c->mnt->to, 0);
poperror();
if(c->mountid != c->mnt->mountid) {
runlock(&pg->ns);
cclose(nc);
return 0;
}
/* Error causes component of union to be skipped */
if(waserror()) {
cclose(nc);
goto next;
}
nc = (*devtab[nc->type].open)((Chan*)nc, OREAD);
nc->offset = c->offset;
nr = (*devtab[nc->type].read)((Chan*)nc, va, n, nc->offset);
/* devdirread e.g. changes it */
c->offset = nc->offset;
poperror();
cclose(nc);
if(nr > 0) {
runlock(&pg->ns);
return nr;
}
/* Advance to next element */
next:
c->mnt = c->mnt->next;
if(c->mnt == 0)
break;
c->mountid = c->mnt->mountid;
c->offset = 0;
}
runlock(&pg->ns);
return 0;
}
long
sysread(int fd, void *va, long n)
{
int dir;
Lock *cl;
Chan *c = 0;
if(waserror()) {
cclose(c);
return -1;
}
c = fdtochan(fd, OREAD, 1, 1);
dir = c->qid.path&CHDIR;
if(dir) {
n -= n%DIRLEN;
if(c->offset%DIRLEN || n==0)
error(Etoosmall);
}
if(dir && c->mnt)
n = unionread((Chan*)c, va, n);
else
n = (*devtab[c->type].read)((Chan*)c, va, n, c->offset);
cl = (Lock*)&c->r.l;
lock(cl);
c->offset += n;
unlock(cl);
poperror();
cclose(c);
return n;
}
int
sysremove(char *path)
{
Chan *c = 0;
if(waserror()) {
if(c != 0)
c->type = 0; /* see below */
cclose(c);
return -1;
}
c = namec(path, Aaccess, 0, 0);
(*devtab[c->type].remove)((Chan*)c);
/*
* Remove clunks the fid, but we need to recover the Chan
* so fake it up. rootclose() is known to be a nop.
*/
c->type = 0;
poperror();
cclose(c);
return 0;
}
long
sysseek(int fd, long off, int whence)
{
Dir dir;
Chan *c;
char buf[DIRLEN];
if(waserror())
return -1;
c = fdtochan(fd, -1, 1, 0);
if(c->qid.path & CHDIR)
error(Eisdir);
switch(whence) {
case 0:
c->offset = off;
break;
case 1:
lock(&c->r.l); /* lock for read/write update */
c->offset += off;
off = c->offset;
unlock(&c->r.l);
break;
case 2:
(*devtab[c->type].stat)(c, buf);
convM2D(buf, &dir);
c->offset = dir.length + off;
off = c->offset;
break;
}
poperror();
return off;
}
int
sysstat(char *path, char *buf)
{
Chan *c = 0;
if(waserror()){
cclose(c);
return -1;
}
c = namec(path, Aaccess, 0, 0);
(*devtab[c->type].stat)((Chan*)c, buf);
poperror();
cclose(c);
return 0;
}
long
syswrite(int fd, void *va, long n)
{
Lock *cl;
Chan *c = 0;
if(waserror()) {
cclose(c);
return -1;
}
c = fdtochan(fd, OWRITE, 1, 1);
if(c->qid.path & CHDIR)
error(Eisdir);
n = (*devtab[c->type].write)((Chan*)c, va, n, c->offset);
cl = (Lock*)&c->r.l;
lock(cl);
c->offset += n;
unlock(cl);
poperror();
cclose(c);
return n;
}
int
syswstat(char *path, char *buf)
{
Chan *c = 0;
if(waserror()) {
cclose(c);
return -1;
}
nameok(buf);
c = namec(path, Aaccess, 0, 0);
(*devtab[c->type].wstat)((Chan*)c, buf);
poperror();
cclose(c);
return 0;
}
int
sysdirstat(char *name, Dir *dir)
{
char buf[DIRLEN];
if(sysstat(name, buf) == -1)
return -1;
convM2D(buf, dir);
return 0;
}
int
sysdirfstat(int fd, Dir *dir)
{
char buf[DIRLEN];
if(sysfstat(fd, buf) == -1)
return -1;
convM2D(buf, dir);
return 0;
}
int
sysdirwstat(char *name, Dir *dir)
{
char buf[DIRLEN];
convD2M(dir, buf);
return syswstat(name, buf);
}
int
sysdirfwstat(int fd, Dir *dir)
{
char buf[DIRLEN];
convD2M(dir, buf);
return sysfwstat(fd, buf);
}
long
sysdirread(int fd, Dir *dbuf, long count)
{
int c, n, i, r;
char buf[DIRLEN*50];
n = 0;
count = (count/sizeof(Dir)) * DIRLEN;
while(n < count) {
c = count - n;
if(c > sizeof(buf))
c = sizeof(buf);
r = sysread(fd, buf, c);
if(r == 0)
break;
if(r < 0 || r % DIRLEN)
return -1;
for(i=0; i<r; i+=DIRLEN) {
convM2D(buf+i, dbuf);
dbuf++;
}
n += r;
if(r != c)
break;
}
return (n/DIRLEN) * sizeof(Dir);
}
static int
call(char *clone, char *dest, int *cfdp, char *dir, char *local)
{
int fd, cfd, n;
char *p, name[3*NAMELEN+5], data[3*NAMELEN+10];
cfd = sysopen(clone, ORDWR);
if(cfd < 0){
werrstr("%s: %r", clone);
return -1;
}
/* get directory name */
n = sysread(cfd, name, sizeof(name)-1);
if(n < 0) {
sysclose(cfd);
return -1;
}
name[n] = 0;
sprint(name, "%d", strtoul(name, 0, 0));
p = strrchr(clone, '/');
*p = 0;
if(dir)
snprint(dir, 2*NAMELEN, "%s/%s", clone, name);
snprint(data, sizeof(data), "%s/%s/data", clone, name);
/* connect */
/* set local side (port number, for example) if we need to */
if(local)
snprint(name, sizeof(name), "connect %s %s", dest, local);
else
snprint(name, sizeof(name), "connect %s", dest);
if(syswrite(cfd, name, strlen(name)) < 0){
werrstr("%s failed: %r", name);
sysclose(cfd);
return -1;
}
/* open data connection */
fd = sysopen(data, ORDWR);
if(fd < 0){
werrstr("can't open %s: %r", data);
sysclose(cfd);
return -1;
}
if(cfdp)
*cfdp = cfd;
else
sysclose(cfd);
return fd;
}
int
sysdial(char *dest, char *local, char *dir, int *cfdp)
{
int n, fd, rv;
char *p, net[128], clone[NAMELEN+12];
/* go for a standard form net!... */
p = strchr(dest, '!');
if(p == 0){
snprint(net, sizeof(net), "net!%s", dest);
} else {
strncpy(net, dest, sizeof(net)-1);
net[sizeof(net)-1] = 0;
}
/* call the connection server */
fd = sysopen("/net/cs", ORDWR);
if(fd < 0){
/* no connection server, don't translate */
p = strchr(net, '!');
*p++ = 0;
snprint(clone, sizeof(clone), "/net/%s/clone", net);
return call(clone, p, cfdp, dir, local);
}
/*
* send dest to connection to translate
*/
if(syswrite(fd, net, strlen(net)) < 0){
werrstr("%s: %r", net);
sysclose(fd);
return -1;
}
/*
* loop through each address from the connection server till
* we get one that works.
*/
rv = -1;
sysseek(fd, 0, 0);
while((n = sysread(fd, net, sizeof(net) - 1)) > 0){
net[n] = 0;
p = strchr(net, ' ');
if(p == 0)
continue;
*p++ = 0;
rv = call(net, p, cfdp, dir, local);
if(rv >= 0)
break;
}
sysclose(fd);
return rv;
}
static int
identtrans(char *addr, char *naddr, int na, char *file, int nf)
{
char *p;
char reply[4*NAMELEN];
/* parse the network */
strncpy(reply, addr, sizeof(reply));
reply[sizeof(reply)-1] = 0;
p = strchr(reply, '!');
if(p)
*p++ = 0;
sprint(file, "/net/%.*s/clone", na - sizeof("/net//clone"), reply);
strncpy(naddr, p, na);
naddr[na-1] = 0;
return 1;
}
static int
nettrans(char *addr, char *naddr, int na, char *file, int nf)
{
long n;
int fd;
char *cp;
char reply[4*NAMELEN];
/*
* ask the connection server
*/
fd = sysopen("/net/cs", ORDWR);
if(fd < 0)
return identtrans(addr, naddr, na, file, nf);
if(syswrite(fd, addr, strlen(addr)) < 0){
sysclose(fd);
return -1;
}
sysseek(fd, 0, 0);
n = sysread(fd, reply, sizeof(reply)-1);
sysclose(fd);
if(n <= 0)
return -1;
reply[n] = '\0';
/*
* parse the reply
*/
cp = strchr(reply, ' ');
if(cp == 0)
return -1;
*cp++ = 0;
strncpy(naddr, cp, na);
naddr[na-1] = 0;
strncpy(file, reply, nf);
file[nf-1] = 0;
return 0;
}
int
sysannounce(char *addr, char *dir)
{
char *cp;
int ctl, n, m;
char buf[3*NAMELEN];
char buf2[3*NAMELEN];
char netdir[2*NAMELEN];
char naddr[3*NAMELEN];
/*
* translate the address
*/
if(nettrans(addr, naddr, sizeof(naddr), netdir, sizeof(netdir)) < 0){
werrstr("can't translate address");
return -1;
}
/*
* get a control channel
*/
ctl = sysopen(netdir, ORDWR);
if(ctl<0){
werrstr("can't open control channel");
return -1;
}
cp = strrchr(netdir, '/');
*cp = 0;
/*
* find out which line we have
*/
n = sprint(buf, "%.*s/", 2*NAMELEN+1, netdir);
m = sysread(ctl, &buf[n], sizeof(buf)-n-1);
if(m <= 0) {
sysclose(ctl);
werrstr("can't read control file");
return -1;
}
buf[n+m] = 0;
/*
* make the call
*/
n = sprint(buf2, "announce %.*s", 2*NAMELEN, naddr);
if(syswrite(ctl, buf2, n) != n) {
sysclose(ctl);
werrstr("announcement fails");
return -1;
}
strcpy(dir, buf);
return ctl;
}
int
syslisten(char *dir, char *newdir)
{
char *cp;
int ctl, n, m;
char buf[3*NAMELEN];
/*
* open listen, wait for a call
*/
sprint(buf, "%.*s/listen", 2*NAMELEN+1, dir);
ctl = sysopen(buf, ORDWR);
if(ctl < 0)
return -1;
/*
* find out which line we have
*/
strcpy(buf, dir);
cp = strrchr(buf, '/');
*++cp = 0;
n = cp-buf;
m = sysread(ctl, cp, sizeof(buf) - n - 1);
if(n<=0){
sysclose(ctl);
return -1;
}
buf[n+m] = 0;
strcpy(newdir, buf);
return ctl;
}