int _sock_srv(char *path, int fd)
{
int sfd;
char msg[8 + 256 + 1];
/* change the path to something in srv */
_sock_srvname(msg, path);
/* remove any previous instance */
unlink(msg);
/* put the fd in /srv and then close it */
sfd = creat(msg, 0666);
if (sfd < 0) {
close(fd);
return -1;
}
snprintf(msg, sizeof msg, "%d", fd);
if (write(sfd, msg, strlen(msg)) < 0) {
close(sfd);
close(fd);
return -1;
}
close(sfd);
close(fd);
return 0;
}
int __libc_accept4(int fd, __SOCKADDR_ARG addr, socklen_t *alen, int a4_flags)
{
int nfd, lcfd;
socklen_t n;
Rock *r, *nr;
struct sockaddr_in *ip;
char name[Ctlsize];
char file[8 + Ctlsize + 1];
const char *net = 0;
char listen[Ctlsize];
int open_flags;
r = _sock_findrock(fd, 0);
if (r == 0) {
errno = ENOTSOCK;
return -1;
}
switch (r->domain) {
case PF_INET:
switch (r->stype) {
case SOCK_DGRAM:
net = "udp";
break;
case SOCK_STREAM:
net = "tcp";
break;
}
/* at this point, our FD is for the data file. we need to open
* the listen file. */
_sock_get_conv_filename(r, "listen", listen);
open_flags = O_RDWR;
/* This is for the listen - maybe don't block on open */
open_flags |= (r->sopts & SOCK_NONBLOCK ? O_NONBLOCK : 0);
/* This is for the ctl we get back - maybe CLOEXEC, based on
* what accept4 wants for the child */
open_flags |= (a4_flags & SOCK_CLOEXEC ? O_CLOEXEC : 0);
lcfd = open(listen, open_flags);
if (lcfd < 0)
return -1;
/* at this point, we have a new conversation, and lcfd is its
* ctl fd. nfd will be the FD for that conv's data file.
* sock_data will store our lcfd in the rock and return the data
* file fd.
*
* Note, we pass the listen socket's stype, but not it's sopts.
* The sopts (e.g. SOCK_NONBLOCK) apply to the original socket,
* not to the new one. Instead, we pass the accept4 flags,
* which are the sopts for the new socket. Note that this is
* just the sopts. Both the listen socket and the new socket
* have the same stype. */
nfd = _sock_data(lcfd, net, r->domain, a4_flags | r->stype,
r->protocol, &nr);
if (nfd < 0)
return -1;
/* get remote address */
ip = (struct sockaddr_in *)&nr->raddr;
_sock_ingetaddr(nr, ip, &n, "remote");
if (addr.__sockaddr__) {
memmove(addr.__sockaddr_in__, ip,
sizeof(struct sockaddr_in));
*alen = sizeof(struct sockaddr_in);
}
return nfd;
case PF_UNIX:
if (r->other >= 0) {
errno = EINVAL; // was EGREG
return -1;
}
for (;;) {
/* read path to new connection */
n = read(fd, name, sizeof(name) - 1);
if (n < 0)
return -1;
if (n == 0)
continue;
name[n] = 0;
/* open new connection */
_sock_srvname(file, name);
open_flags = O_RDWR;
/* This is for the listen - maybe don't block on open */
open_flags |= (r->sopts &
SOCK_NONBLOCK ? O_NONBLOCK : 0);
/* This is for the ctl we get back - maybe CLOEXEC,
* based on what accept4 wants for the child */
open_flags |= (a4_flags & SOCK_CLOEXEC ? O_CLOEXEC : 0);
nfd = open(file, open_flags);
if (nfd < 0)
continue;
/* confirm opening on new connection */
if (write(nfd, name, strlen(name)) > 0)
break;
close(nfd);
}
nr = _sock_newrock(nfd);
if (nr == 0) {
close(nfd);
return -1;
}
nr->domain = r->domain;
nr->stype = r->stype;
nr->sopts = a4_flags;
nr->protocol = r->protocol;
return nfd;
default:
errno = EOPNOTSUPP;
return -1;
}
}
int
accept(int fd, void *a, int *alen)
{
int n, nfd, cfd;
Rock *r, *nr;
struct sockaddr_in *ip;
char name[Ctlsize];
char file[8+Ctlsize+1];
char *net;
r = _sock_findrock(fd, 0);
if(r == 0){
errno = ENOTSOCK;
return -1;
}
switch(r->domain){
case PF_INET:
switch(r->stype){
case SOCK_DGRAM:
net = "udp";
break;
case SOCK_STREAM:
net = "tcp";
break;
default:
net = "gok";
break;
}
/* get control file name from listener process */
n = read(fd, name, sizeof(name)-1);
if(n <= 0){
_syserrno();
return -1;
}
name[n] = 0;
cfd = open(name, O_RDWR);
if(cfd < 0){
_syserrno();
return -1;
}
nfd = _sock_data(cfd, net, r->domain, r->stype, r->protocol, &nr);
if(nfd < 0){
_syserrno();
return -1;
}
if(write(fd, "OK", 2) < 0){
close(nfd);
_syserrno();
return -1;
}
/* get remote address */
ip = (struct sockaddr_in*)&nr->raddr;
_sock_ingetaddr(nr, ip, &n, "remote");
if(a){
memmove(a, ip, sizeof(struct sockaddr_in));
*alen = sizeof(struct sockaddr_in);
}
return nfd;
case PF_UNIX:
if(r->other >= 0){
errno = EGREG;
return -1;
}
for(;;){
/* read path to new connection */
n = read(fd, name, sizeof(name) - 1);
if(n < 0)
return -1;
if(n == 0)
continue;
name[n] = 0;
/* open new connection */
_sock_srvname(file, name);
nfd = open(file, O_RDWR);
if(nfd < 0)
continue;
/* confirm opening on new connection */
if(write(nfd, name, strlen(name)) > 0)
break;
close(nfd);
}
nr = _sock_newrock(nfd);
if(nr == 0){
close(nfd);
return -1;
}
nr->domain = r->domain;
nr->stype = r->stype;
nr->protocol = r->protocol;
return nfd;
default:
errno = EOPNOTSUPP;
return -1;
}
}
