shithub: riscv

ref: 717237303d368378283ad719ee4f2c580b99fd2b
dir: /sys/src/lib9p/listen.c/

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

static void listenproc(void*);
static void srvproc(void*);
static void srvfree(Srv *);
static char *getremotesys(char*);

void
listensrv(Srv *os, char *addr)
{
	Srv *s;

	s = emalloc9p(sizeof *s);
	*s = *os;

	s->addr = estrdup9p(addr);
	s->infd = s->outfd = -1;
	s->fpool = nil;
	s->rpool = nil;
	s->msize = 0;
	s->rbuf = nil;
	s->wbuf = nil;
	memset(&s->rlock, 0, sizeof(s->rlock));
	memset(&s->wlock, 0, sizeof(s->wlock));
	memset(&s->slock, 0, sizeof(s->slock));
	memset(&s->sref, 0, sizeof(s->sref));
	memset(&s->rref, 0, sizeof(s->rref));
	s->spid = 0;
	s->free = nil;

	if(s->forker == nil)
		s->forker = srvforker;
	(*s->forker)(listenproc, s, 0);
}

static void
listenproc(void *v)
{
	char ndir[NETPATHLEN], dir[NETPATHLEN];
	int ctl, data, nctl;
	Srv *os, *s;
	
	os = v;
	ctl = announce(os->addr, dir);
	if(ctl < 0){
		fprint(2, "%s: announce %s: %r\n", argv0, os->addr);
		return;
	}

	for(;;){
		nctl = listen(dir, ndir);
		if(nctl < 0){
			fprint(2, "%s: listen %s: %r\n", argv0, os->addr);
			break;
		}
		
		data = accept(nctl, ndir);
		if(data < 0){
			fprint(2, "%s: accept %s: %r\n", argv0, ndir);
			close(nctl);
			continue;
		}
		close(nctl);

		s = emalloc9p(sizeof *s);
		*s = *os;
		s->addr = getremotesys(ndir);
		s->infd = s->outfd = data;
		s->free = srvfree;
		(*s->forker)(srvproc, s, 0);
	}
	free(os->addr);
	free(os);
}

static void
srvproc(void *v)
{
	srv((Srv*)v);
}

static void
srvfree(Srv *s)
{
	close(s->infd);
	free(s->addr);
	free(s);
}

static char*
getremotesys(char *ndir)
{
	char buf[128], *serv, *sys;
	int fd, n;

	snprint(buf, sizeof buf, "%s/remote", ndir);
	sys = nil;
	fd = open(buf, OREAD|OCEXEC);
	if(fd >= 0){
		n = read(fd, buf, sizeof(buf)-1);
		if(n>0){
			buf[n-1] = 0;
			serv = strchr(buf, '!');
			if(serv)
				*serv = 0;
			sys = estrdup9p(buf);
		}
		close(fd);
	}
	if(sys == nil)
		sys = estrdup9p("unknown");
	return sys;
}