tchanges from erik quanstrom - plan9port - [fork] Plan 9 from user space
(HTM) git clone git://src.adamsgaard.dk/plan9port
(DIR) Log
(DIR) Files
(DIR) Refs
(DIR) README
(DIR) LICENSE
---
(DIR) commit 19256e0dd9c3c383ada7cc28b67487400c9162e3
(DIR) parent 05f5d46b6368978fd22cfbea7155e983b0faa9a7
(HTM) Author: rsc <devnull@localhost>
Date: Fri, 17 Feb 2006 19:21:49 +0000
changes from erik quanstrom
Diffstat:
M src/cmd/ndb/dnarea.c | 32 +++++++++----------------------
M src/cmd/ndb/dns.c | 16 +++++++++-------
M src/cmd/ndb/dns.h | 4 ++++
M src/cmd/ndb/dnsdebug.c | 8 ++++----
M src/cmd/ndb/dntcpserver.c | 23 ++++++++++-------------
M src/cmd/ndb/dnudpserver.c | 8 +++-----
M src/cmd/ndb/mkfile | 1 +
A src/cmd/ndb/runproc.c | 37 +++++++++++++++++++++++++++++++
8 files changed, 77 insertions(+), 52 deletions(-)
---
(DIR) diff --git a/src/cmd/ndb/dnarea.c b/src/cmd/ndb/dnarea.c
t@@ -91,12 +91,15 @@ freearea(Area **l)
* this entails running a command 'zonerefreshprogram'. This could
* copy over databases from elsewhere or just do a zone transfer.
*/
-/* XXX WRONG - can't use fork and exec */
void
refresh_areas(Area *s)
{
- int pid;
Waitmsg *w;
+ char *argv[3];
+
+ argv[0] = zonerefreshprogram;
+ argv[1] = "XXX";
+ argv[2] = nil;
for(; s != nil; s = s->next){
if(!s->needrefresh)
t@@ -107,26 +110,9 @@ refresh_areas(Area *s)
continue;
}
- switch(pid = fork()){
- case -1:
- break;
- case 0:
- execl(zonerefreshprogram, "zonerefresh", s->soarr->owner->name, 0);
- threadexitsall(0);
- break;
- default:
- for(;;){
- w = wait();
- if(w == nil)
- break;
- if(w->pid == pid){
- if(w->msg == nil || *w->msg == 0)
- s->needrefresh = 0;
- free(w);
- break;
- }
- free(w);
- }
- }
+ argv[1] = s->soarr->owner->name;
+ w = runproc(argv[0], argv, 0);
+ free(w);
}
}
+
(DIR) diff --git a/src/cmd/ndb/dns.c b/src/cmd/ndb/dns.c
t@@ -61,7 +61,6 @@ struct {
Mfile *inuse; /* active mfile's */
} mfalloc;
-int haveip;
int mfd[2];
int debug;
int traceactivity;
t@@ -97,7 +96,7 @@ Job* newjob(void);
void freejob(Job*);
void setext(char*, int, char*);
-char *portname = "domain";
+char *portname = "domain";
char *logfile = "dns";
char *dbfile;
char mntpt[Maxpath];
t@@ -106,7 +105,7 @@ char *LOG;
void
usage(void)
{
- fprint(2, "usage: %s [-dnrstT] [-a maxage] [-f ndb-file] [-p port] [-x service] [-z zoneprog]\n", argv0);
+ fprint(2, "usage: dns [-dnrstT] [-a maxage] [-f ndb-file] [-p port] [-x service] [-z zoneprog]\n");
threadexitsall("usage");
}
t@@ -134,7 +133,7 @@ threadmain(int argc, char *argv[])
resolver = 1;
break;
case 's':
- serveudp = 1; /* serve network */
+ serveudp = 1;
cachedb = 1;
break;
case 'T':
t@@ -165,7 +164,7 @@ threadmain(int argc, char *argv[])
/* start syslog before we fork */
fmtinstall('F', fcallfmt);
dninit();
- if(!haveip && myipaddr(ipaddr, mntpt) < 0)
+ if(myipaddr(ipaddr, mntpt) < 0)
sysfatal("can't read my ip address");
syslog(0, logfile, "starting dns on %I", ipaddr);
t@@ -350,7 +349,7 @@ ioproc0(void *v)
for(;;){
n = read9pmsg(mfd[0], mdata, sizeof mdata);
- if(n<=0){
+ if(n <= 0){
syslog(0, logfile, "error reading mntpt: %r");
break;
}
t@@ -665,7 +664,10 @@ rwrite(Job *job, Mfile *mf, Request *req)
dndump(dumpfile);
goto send;
} else if(strncmp(p, "dump ", 5) == 0){
- dndump(p+5);
+ if(*(p+5))
+ dndump(p+5);
+ else
+ err = "bad filename";
goto send;
} else if(strcmp(p, "refresh")==0){
needrefresh = 1;
(DIR) diff --git a/src/cmd/ndb/dns.h b/src/cmd/ndb/dns.h
t@@ -381,6 +381,10 @@ extern char* convM2DNS(uchar*, int, DNSmsg*);
extern void mallocsanity(void*);
extern void lasthist(void*, int, ulong);
+/* runproc.c */
+extern Waitmsg* runproc(char*, char**, int);
+extern Waitmsg* runprocfd(char*, char**, int[3]);
+
extern int debug;
extern int traceactivity;
extern char *trace;
(DIR) diff --git a/src/cmd/ndb/dnsdebug.c b/src/cmd/ndb/dnsdebug.c
t@@ -33,6 +33,7 @@ char *logfile = "dns";
char *dbfile;
char mntpt[Maxpath];
char *zonerefreshprogram;
+char *portname = "domain";
int prettyrrfmt(Fmt*);
void preloadserveraddrs(void);
t@@ -47,7 +48,6 @@ usage(void)
fprint(2, "usage: dnsdebug -rxf [-p port] [query ...]\n");
threadexitsall("usage");
}
-char *portname = "domain";
void
threadmain(int argc, char *argv[])
t@@ -60,9 +60,6 @@ threadmain(int argc, char *argv[])
strcpy(mntpt, "/net");
ARGBEGIN{
- case 'p': /* XXX */
- portname = EARGF(usage());
- break;
case 'r':
resolver = 1;
break;
t@@ -73,6 +70,9 @@ threadmain(int argc, char *argv[])
case 'f':
dbfile = EARGF(usage());
break;
+ case 'p':
+ portname = EARGF(usage());
+ break;
default:
usage();
}ARGEND
(DIR) diff --git a/src/cmd/ndb/dntcpserver.c b/src/cmd/ndb/dntcpserver.c
t@@ -35,10 +35,10 @@ connreadmsg(int tfd, int *fd, uchar *buf, int max)
if (lfd < 0)
return -1;
*fd = accept(lfd, ldir);
- n = -1;
- if (*fd < 0)
+ if (*fd >= 0)
n = readmsg(*fd, buf, max);
- //close(fd);
+ else
+ n = -1;
close(lfd);
return n;
}
t@@ -181,7 +181,7 @@ out:
void
tcpproc(void *v)
{
- int len;
+ int len, rv;
Request req;
DNSmsg reqmsg, repmsg;
char *err;
t@@ -191,7 +191,7 @@ tcpproc(void *v)
NetConnInfo *caller;
rfd = -1;
- fd = (int)v;
+ fd = (uintptr)v;
caller = 0;
/* loop on requests */
for(;; putactivity()){
t@@ -244,12 +244,13 @@ tcpproc(void *v)
break;
} else {
dnserver(&reqmsg, &repmsg, &req);
- if(reply(rfd, &repmsg, &req, caller) < 0)
- break;
+ rv = reply(rfd, &repmsg, &req, caller);
rrfreelist(repmsg.qd);
rrfreelist(repmsg.an);
rrfreelist(repmsg.ns);
rrfreelist(repmsg.ar);
+ if(rv < 0)
+ break;
}
}
t@@ -264,8 +265,6 @@ enum {
Maxactivetcp = 4,
};
-extern char *portname;
-
static int
tcpannounce(char *mntpt)
{
t@@ -283,12 +282,10 @@ void
dntcpserver(void *v)
{
int i, fd;
- char *mntpt;
- mntpt = v;
- while((fd = tcpannounce(mntpt)) < 0)
+ while((fd = tcpannounce(v)) < 0)
sleep(5*1000);
for(i=0; i<Maxactivetcp; i++)
- proccreate(tcpproc, (void*)fd, STACK);
+ proccreate(tcpproc, (void*)(uintptr)fd, STACK);
}
(DIR) diff --git a/src/cmd/ndb/dnudpserver.c b/src/cmd/ndb/dnudpserver.c
t@@ -72,15 +72,13 @@ udpproc(void *v)
char tname[32];
Udphdr *uh;
- fd = (int)v;
+ fd = (uintptr)v;
/* loop on requests */
for(;; putactivity()){
memset(&repmsg, 0, sizeof(repmsg));
memset(&reqmsg, 0, sizeof(reqmsg));
- alarm(60*1000);
len = udpread(fd, (Udphdr*)buf, buf+Udphdrsize, sizeof(buf)-Udphdrsize);
- alarm(0);
if(len <= 0)
continue;
uh = (Udphdr*)buf;
t@@ -161,7 +159,7 @@ udpannounce(char *mntpt)
snprint(buf, sizeof buf, "udp!*!%s", portname);
if((fd=announce(buf, buf)) < 0)
- warning("can't announce on dns udp port");
+ warning("announce %s: %r", buf);
return fd;
}
t@@ -202,6 +200,6 @@ dnudpserver(void *v)
while((fd = udpannounce(v)) < 0)
sleep(5*1000);
for(i=0; i<Maxactive; i++)
- proccreate(udpproc, (void*)fd, STACK);
+ proccreate(udpproc, (void*)(uintptr)fd, STACK);
}
(DIR) diff --git a/src/cmd/ndb/mkfile b/src/cmd/ndb/mkfile
t@@ -24,6 +24,7 @@ DNSOFILES=\
dnresolve.$O\
dnserver.$O\
dntcpserver.$O\
+ runproc.$O\
$DNSOFILES dns.$O dnstcp.$O dnsdebug.$O: dns.h
(DIR) diff --git a/src/cmd/ndb/runproc.c b/src/cmd/ndb/runproc.c
t@@ -0,0 +1,37 @@
+#include <u.h>
+#include <libc.h>
+#include <thread.h>
+#include <bio.h>
+#include <ndb.h>
+#include "dns.h"
+
+Waitmsg*
+runprocfd(char *file, char **v, int fd[3])
+{
+ int pid, i;
+
+ threadwaitchan();
+ pid = threadspawn(fd, file, v);
+ for(i=0; i<3; i++)
+ close(fd[i]);
+ if(pid < 0)
+ return nil;
+ return procwait(pid);
+}
+
+Waitmsg*
+runproc(char *file, char **v, int devnull)
+{
+ int fd[3], i;
+
+ if(devnull){
+ fd[0] = open("/dev/null", ORDWR);
+ fd[1] = dup(1, fd[0]);
+ fd[2] = dup(2, fd[0]);
+ }else{
+ for(i=0; i<3; i++)
+ fd[i] = dup(i, -1);
+ }
+ return runprocfd(file, v, fd);
+}
+