251 lines
6.7 KiB
C
251 lines
6.7 KiB
C
/*-
|
|
* Copyright (c) 2015 Nuxi, https://nuxi.nl/
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <sys/cdefs.h>
|
|
__FBSDID("$FreeBSD$");
|
|
|
|
#include <sys/param.h>
|
|
#include <sys/capsicum.h>
|
|
#include <sys/lock.h>
|
|
#include <sys/malloc.h>
|
|
#include <sys/mutex.h>
|
|
#include <sys/protosw.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/socketvar.h>
|
|
#include <sys/syscallsubr.h>
|
|
#include <sys/sysproto.h>
|
|
#include <sys/systm.h>
|
|
#include <sys/un.h>
|
|
|
|
#include <net/vnet.h>
|
|
|
|
#include <netinet/in.h>
|
|
|
|
#include <compat/cloudabi/cloudabi_proto.h>
|
|
#include <compat/cloudabi/cloudabi_syscalldefs.h>
|
|
#include <compat/cloudabi/cloudabi_util.h>
|
|
|
|
/* Converts FreeBSD's struct sockaddr to CloudABI's cloudabi_sockaddr_t. */
|
|
void
|
|
cloudabi_convert_sockaddr(const struct sockaddr *sa, socklen_t sal,
|
|
cloudabi_sockaddr_t *rsa)
|
|
{
|
|
const struct sockaddr_in *sin;
|
|
const struct sockaddr_in6 *sin6;
|
|
|
|
/* Zero-sized socket address. */
|
|
if (sal < offsetof(struct sockaddr, sa_family) + sizeof(sa->sa_family))
|
|
return;
|
|
|
|
switch (sa->sa_family) {
|
|
case AF_INET:
|
|
if (sal < sizeof(struct sockaddr_in))
|
|
return;
|
|
sin = (const struct sockaddr_in *)sa;
|
|
rsa->sa_family = CLOUDABI_AF_INET;
|
|
memcpy(&rsa->sa_inet.addr, &sin->sin_addr,
|
|
sizeof(rsa->sa_inet.addr));
|
|
rsa->sa_inet.port = ntohs(sin->sin_port);
|
|
return;
|
|
case AF_INET6:
|
|
if (sal < sizeof(struct sockaddr_in6))
|
|
return;
|
|
sin6 = (const struct sockaddr_in6 *)sa;
|
|
rsa->sa_family = CLOUDABI_AF_INET6;
|
|
memcpy(&rsa->sa_inet6.addr, &sin6->sin6_addr,
|
|
sizeof(rsa->sa_inet6.addr));
|
|
rsa->sa_inet6.port = ntohs(sin6->sin6_port);
|
|
return;
|
|
case AF_UNIX:
|
|
rsa->sa_family = CLOUDABI_AF_UNIX;
|
|
return;
|
|
}
|
|
}
|
|
|
|
/* Copies a pathname into a UNIX socket address structure. */
|
|
static int
|
|
copyin_sockaddr_un(const char *path, size_t pathlen, struct sockaddr_un *sun)
|
|
{
|
|
int error;
|
|
|
|
/* Copy in pathname string if there's enough space. */
|
|
if (pathlen >= sizeof(sun->sun_path))
|
|
return (ENAMETOOLONG);
|
|
error = copyin(path, &sun->sun_path, pathlen);
|
|
if (error != 0)
|
|
return (error);
|
|
if (memchr(sun->sun_path, '\0', pathlen) != NULL)
|
|
return (EINVAL);
|
|
|
|
/* Initialize the rest of the socket address. */
|
|
sun->sun_path[pathlen] = '\0';
|
|
sun->sun_family = AF_UNIX;
|
|
sun->sun_len = sizeof(*sun);
|
|
return (0);
|
|
}
|
|
|
|
int
|
|
cloudabi_sys_sock_accept(struct thread *td,
|
|
struct cloudabi_sys_sock_accept_args *uap)
|
|
{
|
|
struct sockaddr *sa;
|
|
cloudabi_sockstat_t ss = {};
|
|
socklen_t sal;
|
|
int error;
|
|
|
|
if (uap->buf == NULL) {
|
|
/* Only return the new file descriptor number. */
|
|
return (kern_accept(td, uap->s, NULL, NULL, NULL));
|
|
} else {
|
|
/* Also return properties of the new socket descriptor. */
|
|
sal = MAX(sizeof(struct sockaddr_in),
|
|
sizeof(struct sockaddr_in6));
|
|
error = kern_accept(td, uap->s, (void *)&sa, &sal, NULL);
|
|
if (error != 0)
|
|
return (error);
|
|
|
|
/* TODO(ed): Fill the other members of cloudabi_sockstat_t. */
|
|
cloudabi_convert_sockaddr(sa, sal, &ss.ss_peername);
|
|
free(sa, M_SONAME);
|
|
return (copyout(&ss, uap->buf, sizeof(ss)));
|
|
}
|
|
}
|
|
|
|
int
|
|
cloudabi_sys_sock_bind(struct thread *td,
|
|
struct cloudabi_sys_sock_bind_args *uap)
|
|
{
|
|
struct sockaddr_un sun;
|
|
int error;
|
|
|
|
error = copyin_sockaddr_un(uap->path, uap->pathlen, &sun);
|
|
if (error != 0)
|
|
return (error);
|
|
return (kern_bindat(td, uap->fd, uap->s, (struct sockaddr *)&sun));
|
|
}
|
|
|
|
int
|
|
cloudabi_sys_sock_connect(struct thread *td,
|
|
struct cloudabi_sys_sock_connect_args *uap)
|
|
{
|
|
struct sockaddr_un sun;
|
|
int error;
|
|
|
|
error = copyin_sockaddr_un(uap->path, uap->pathlen, &sun);
|
|
if (error != 0)
|
|
return (error);
|
|
return (kern_connectat(td, uap->fd, uap->s, (struct sockaddr *)&sun));
|
|
}
|
|
|
|
int
|
|
cloudabi_sys_sock_listen(struct thread *td,
|
|
struct cloudabi_sys_sock_listen_args *uap)
|
|
{
|
|
struct listen_args listen_args = {
|
|
.s = uap->s,
|
|
.backlog = uap->backlog,
|
|
};
|
|
|
|
return (sys_listen(td, &listen_args));
|
|
}
|
|
|
|
int
|
|
cloudabi_sys_sock_shutdown(struct thread *td,
|
|
struct cloudabi_sys_sock_shutdown_args *uap)
|
|
{
|
|
struct shutdown_args shutdown_args = {
|
|
.s = uap->fd,
|
|
};
|
|
|
|
switch (uap->how) {
|
|
case CLOUDABI_SHUT_RD:
|
|
shutdown_args.how = SHUT_RD;
|
|
break;
|
|
case CLOUDABI_SHUT_WR:
|
|
shutdown_args.how = SHUT_WR;
|
|
break;
|
|
case CLOUDABI_SHUT_RD | CLOUDABI_SHUT_WR:
|
|
shutdown_args.how = SHUT_RDWR;
|
|
break;
|
|
default:
|
|
return (EINVAL);
|
|
}
|
|
|
|
return (sys_shutdown(td, &shutdown_args));
|
|
}
|
|
|
|
int
|
|
cloudabi_sys_sock_stat_get(struct thread *td,
|
|
struct cloudabi_sys_sock_stat_get_args *uap)
|
|
{
|
|
cloudabi_sockstat_t ss = {};
|
|
cap_rights_t rights;
|
|
struct file *fp;
|
|
struct sockaddr *sa;
|
|
struct socket *so;
|
|
int error;
|
|
|
|
error = getsock_cap(td, uap->fd, cap_rights_init(&rights,
|
|
CAP_GETSOCKOPT | CAP_GETPEERNAME | CAP_GETSOCKNAME), &fp, NULL);
|
|
if (error != 0)
|
|
return (error);
|
|
so = fp->f_data;
|
|
|
|
CURVNET_SET(so->so_vnet);
|
|
|
|
/* Set ss_sockname. */
|
|
error = so->so_proto->pr_usrreqs->pru_sockaddr(so, &sa);
|
|
if (error == 0) {
|
|
cloudabi_convert_sockaddr(sa, sa->sa_len, &ss.ss_sockname);
|
|
free(sa, M_SONAME);
|
|
}
|
|
|
|
/* Set ss_peername. */
|
|
if ((so->so_state & (SS_ISCONNECTED | SS_ISCONFIRMING)) != 0) {
|
|
error = so->so_proto->pr_usrreqs->pru_peeraddr(so, &sa);
|
|
if (error == 0) {
|
|
cloudabi_convert_sockaddr(sa, sa->sa_len,
|
|
&ss.ss_peername);
|
|
free(sa, M_SONAME);
|
|
}
|
|
}
|
|
|
|
CURVNET_RESTORE();
|
|
|
|
/* Set ss_error. */
|
|
SOCK_LOCK(so);
|
|
ss.ss_error = cloudabi_convert_errno(so->so_error);
|
|
if ((uap->flags & CLOUDABI_SOCKSTAT_CLEAR_ERROR) != 0)
|
|
so->so_error = 0;
|
|
SOCK_UNLOCK(so);
|
|
|
|
/* Set ss_state. */
|
|
if ((so->so_options & SO_ACCEPTCONN) != 0)
|
|
ss.ss_state |= CLOUDABI_SOCKSTAT_ACCEPTCONN;
|
|
|
|
fdrop(fp, td);
|
|
return (copyout(&ss, uap->buf, sizeof(ss)));
|
|
}
|