aux/listen: remove Service struct
An announcing process already knows what port it announced on, we don't need to derive it from every connection.
This commit is contained in:
parent
1da69597bd
commit
8fbfba1ecf
1 changed files with 29 additions and 92 deletions
|
@ -2,14 +2,9 @@
|
||||||
#include <libc.h>
|
#include <libc.h>
|
||||||
#include <auth.h>
|
#include <auth.h>
|
||||||
|
|
||||||
#define NAMELEN 64 /* reasonable upper limit for name elements */
|
enum{
|
||||||
|
Maxpath = 1024,
|
||||||
typedef struct Service Service;
|
Maxserv = 64,
|
||||||
struct Service
|
|
||||||
{
|
|
||||||
char serv[NAMELEN]; /* name of the service */
|
|
||||||
char remote[3*NAMELEN]; /* address of remote system */
|
|
||||||
char prog[5*NAMELEN+1]; /* program to execute */
|
|
||||||
};
|
};
|
||||||
|
|
||||||
typedef struct Announce Announce;
|
typedef struct Announce Announce;
|
||||||
|
@ -24,9 +19,7 @@ struct Announce
|
||||||
|
|
||||||
int readstr(char*, char*, char*, int);
|
int readstr(char*, char*, char*, int);
|
||||||
void dolisten(char*, int, char*, char*, long*);
|
void dolisten(char*, int, char*, char*, long*);
|
||||||
void newcall(int, char*, Service*);
|
void newcall(int, char*, char*, char*);
|
||||||
int findserv(char*, Service*, char*);
|
|
||||||
int getserv(char*, Service*);
|
|
||||||
void error(char*);
|
void error(char*);
|
||||||
void scandir(char*);
|
void scandir(char*);
|
||||||
void becomenone(void);
|
void becomenone(void);
|
||||||
|
@ -42,7 +35,6 @@ char *proto;
|
||||||
char *protodir;
|
char *protodir;
|
||||||
char *addr;
|
char *addr;
|
||||||
Announce *announcements;
|
Announce *announcements;
|
||||||
#define SEC 1000
|
|
||||||
|
|
||||||
char *namespace;
|
char *namespace;
|
||||||
|
|
||||||
|
@ -56,7 +48,6 @@ usage(void)
|
||||||
void
|
void
|
||||||
main(int argc, char *argv[])
|
main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
Service *s;
|
|
||||||
char *trustdir;
|
char *trustdir;
|
||||||
char *servdir;
|
char *servdir;
|
||||||
|
|
||||||
|
@ -102,9 +93,9 @@ main(int argc, char *argv[])
|
||||||
if(!servdir && !trustdir)
|
if(!servdir && !trustdir)
|
||||||
servdir = "/bin/service";
|
servdir = "/bin/service";
|
||||||
|
|
||||||
if(servdir && strlen(servdir) + NAMELEN >= sizeof(s->prog))
|
if(servdir && strlen(servdir) + Maxserv >= Maxpath)
|
||||||
error("service directory too long");
|
error("service directory too long");
|
||||||
if(trustdir && strlen(trustdir) + NAMELEN >= sizeof(s->prog))
|
if(trustdir && strlen(trustdir) + Maxserv >= Maxpath)
|
||||||
error("trusted service directory too long");
|
error("trusted service directory too long");
|
||||||
|
|
||||||
switch(argc){
|
switch(argc){
|
||||||
|
@ -186,8 +177,7 @@ listendir(char *srvdir, int trusted)
|
||||||
|
|
||||||
sleep((pid*10)%200);
|
sleep((pid*10)%200);
|
||||||
|
|
||||||
/* copy to stack */
|
snprint(ds, sizeof ds, "%s!%s!%s", protodir, addr, a->a);
|
||||||
strncpy(ds, a->a, sizeof(ds));
|
|
||||||
whined = a->whined;
|
whined = a->whined;
|
||||||
|
|
||||||
/* a process per service */
|
/* a process per service */
|
||||||
|
@ -211,7 +201,7 @@ listendir(char *srvdir, int trusted)
|
||||||
else
|
else
|
||||||
exits("ctl");
|
exits("ctl");
|
||||||
}
|
}
|
||||||
dolisten(dir, ctl, srvdir, ds, &childs);
|
dolisten(dir, ctl, srvdir, a->a, &childs);
|
||||||
close(ctl);
|
close(ctl);
|
||||||
}
|
}
|
||||||
default:
|
default:
|
||||||
|
@ -290,7 +280,6 @@ scandir(char *dname)
|
||||||
Announce *a, **l;
|
Announce *a, **l;
|
||||||
int fd, i, n, nlen;
|
int fd, i, n, nlen;
|
||||||
char *nm;
|
char *nm;
|
||||||
char ds[128];
|
|
||||||
Dir *db;
|
Dir *db;
|
||||||
|
|
||||||
for(a = announcements; a != nil; a = a->next)
|
for(a = announcements; a != nil; a = a->next)
|
||||||
|
@ -310,8 +299,7 @@ scandir(char *dname)
|
||||||
continue;
|
continue;
|
||||||
if(strncmp(nm, proto, nlen) != 0)
|
if(strncmp(nm, proto, nlen) != 0)
|
||||||
continue;
|
continue;
|
||||||
snprint(ds, sizeof ds, "%s!%s!%s", protodir, addr, nm + nlen);
|
addannounce(nm + nlen);
|
||||||
addannounce(ds);
|
|
||||||
}
|
}
|
||||||
free(db);
|
free(db);
|
||||||
}
|
}
|
||||||
|
@ -341,14 +329,16 @@ becomenone(void)
|
||||||
}
|
}
|
||||||
|
|
||||||
void
|
void
|
||||||
dolisten(char *dir, int ctl, char *srvdir, char *dialstr, long *pchilds)
|
dolisten(char *dir, int ctl, char *srvdir, char *port, long *pchilds)
|
||||||
{
|
{
|
||||||
Service s;
|
|
||||||
char ndir[40], wbuf[64];
|
char ndir[40], wbuf[64];
|
||||||
|
char prog[Maxpath], serv[Maxserv];
|
||||||
int nctl, data, wfd, nowait;
|
int nctl, data, wfd, nowait;
|
||||||
|
|
||||||
procsetname("%s %s", dir, dialstr);
|
procsetname("%s %s!%s!%s", dir, proto, addr, port);
|
||||||
|
snprint(serv, sizeof serv, "%s%s", proto, port);
|
||||||
|
snprint(prog, sizeof prog, "%s/%s", srvdir, serv);
|
||||||
|
|
||||||
wfd = -1;
|
wfd = -1;
|
||||||
nowait = RFNOWAIT;
|
nowait = RFNOWAIT;
|
||||||
if(pchilds && maxprocs > 0){
|
if(pchilds && maxprocs > 0){
|
||||||
|
@ -413,17 +403,6 @@ dolisten(char *dir, int ctl, char *srvdir, char *dialstr, long *pchilds)
|
||||||
close(nctl);
|
close(nctl);
|
||||||
continue;
|
continue;
|
||||||
case 0:
|
case 0:
|
||||||
/*
|
|
||||||
* see if we know the service requested
|
|
||||||
*/
|
|
||||||
memset(&s, 0, sizeof s);
|
|
||||||
if(!findserv(ndir, &s, srvdir)){
|
|
||||||
if(!quiet)
|
|
||||||
syslog(1, listenlog, "%s: unknown service '%s' from '%s': %r",
|
|
||||||
proto, s.serv, s.remote);
|
|
||||||
reject(nctl, ndir, "connection refused");
|
|
||||||
exits(0);
|
|
||||||
}
|
|
||||||
data = accept(nctl, ndir);
|
data = accept(nctl, ndir);
|
||||||
if(data < 0){
|
if(data < 0){
|
||||||
syslog(1, listenlog, "can't open %s/data: %r", ndir);
|
syslog(1, listenlog, "can't open %s/data: %r", ndir);
|
||||||
|
@ -434,7 +413,7 @@ dolisten(char *dir, int ctl, char *srvdir, char *dialstr, long *pchilds)
|
||||||
close(nctl);
|
close(nctl);
|
||||||
if(wfd >= 0)
|
if(wfd >= 0)
|
||||||
close(wfd);
|
close(wfd);
|
||||||
newcall(data, ndir, &s);
|
newcall(data, ndir, prog, serv);
|
||||||
exits(0);
|
exits(0);
|
||||||
default:
|
default:
|
||||||
close(nctl);
|
close(nctl);
|
||||||
|
@ -447,61 +426,19 @@ dolisten(char *dir, int ctl, char *srvdir, char *dialstr, long *pchilds)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
* look in the service directory for the service.
|
|
||||||
* if the shell script or program is zero-length, ignore it,
|
|
||||||
* thus providing a way to disable a service with a bind.
|
|
||||||
*/
|
|
||||||
int
|
|
||||||
findserv(char *dir, Service *s, char *srvdir)
|
|
||||||
{
|
|
||||||
int rv;
|
|
||||||
Dir *d;
|
|
||||||
|
|
||||||
if(!getserv(dir, s))
|
|
||||||
return 0;
|
|
||||||
snprint(s->prog, sizeof s->prog, "%s/%s", srvdir, s->serv);
|
|
||||||
d = dirstat(s->prog);
|
|
||||||
rv = d && d->length > 0; /* ignore unless it's non-empty */
|
|
||||||
free(d);
|
|
||||||
return rv;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* get the service name out of the local address
|
|
||||||
*/
|
|
||||||
int
|
|
||||||
getserv(char *dir, Service *s)
|
|
||||||
{
|
|
||||||
char addr[128], *serv, *p;
|
|
||||||
long n;
|
|
||||||
|
|
||||||
readstr(dir, "remote", s->remote, sizeof(s->remote)-1);
|
|
||||||
if(p = utfrune(s->remote, '\n'))
|
|
||||||
*p = '\0';
|
|
||||||
if(p = utfrune(s->remote, '!'))
|
|
||||||
*p = '\0';
|
|
||||||
|
|
||||||
n = readstr(dir, "local", addr, sizeof(addr)-1);
|
|
||||||
if(n <= 0)
|
|
||||||
return 0;
|
|
||||||
if(p = utfrune(addr, '\n'))
|
|
||||||
*p = '\0';
|
|
||||||
serv = utfrune(addr, '!');
|
|
||||||
if(!serv)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
snprint(s->serv, sizeof s->serv, "%s%s", proto, serv+1);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
void
|
||||||
newcall(int fd, char *dir, Service *s)
|
newcall(int fd, char *dir, char *prog, char *serv)
|
||||||
{
|
{
|
||||||
char data[4*NAMELEN];
|
char data[Maxpath];
|
||||||
|
char remote[128];
|
||||||
|
char *p;
|
||||||
|
|
||||||
if(!quiet)
|
if(!quiet){
|
||||||
syslog(0, listenlog, "%s call for %s on chan %s (%s)", proto, s->serv, dir, s->remote);
|
readstr(dir, "remote", remote, sizeof remote);
|
||||||
|
if(p = utfrune(remote, '!'))
|
||||||
|
*p = '\0';
|
||||||
|
syslog(0, listenlog, "%s call for %s on chan %s (%s)", proto, serv, dir, remote);
|
||||||
|
}
|
||||||
|
|
||||||
snprint(data, sizeof data, "%s/data", dir);
|
snprint(data, sizeof data, "%s/data", dir);
|
||||||
bind(data, "/dev/cons", MREPL);
|
bind(data, "/dev/cons", MREPL);
|
||||||
|
@ -515,8 +452,8 @@ newcall(int fd, char *dir, Service *s)
|
||||||
*/
|
*/
|
||||||
for(fd=3; fd<20; fd++)
|
for(fd=3; fd<20; fd++)
|
||||||
close(fd);
|
close(fd);
|
||||||
execl(s->prog, s->prog, s->serv, proto, dir, nil);
|
execl(prog, prog, serv, proto, dir, nil);
|
||||||
error(s->prog);
|
error(prog);
|
||||||
}
|
}
|
||||||
|
|
||||||
void
|
void
|
||||||
|
@ -533,7 +470,7 @@ int
|
||||||
readstr(char *dir, char *info, char *s, int len)
|
readstr(char *dir, char *info, char *s, int len)
|
||||||
{
|
{
|
||||||
int n, fd;
|
int n, fd;
|
||||||
char buf[3*NAMELEN+4];
|
char buf[Maxpath];
|
||||||
|
|
||||||
snprint(buf, sizeof buf, "%s/%s", dir, info);
|
snprint(buf, sizeof buf, "%s/%s", dir, info);
|
||||||
fd = open(buf, OREAD);
|
fd = open(buf, OREAD);
|
||||||
|
|
Loading…
Reference in a new issue