sys/compat/cloudabi32/cloudabi32_sock.c

   1 /*-
   2  * Copyright (c) 2015-2017 Nuxi, https://nuxi.nl/
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  16  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  17  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  18  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  19  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  20  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  21  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  22  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  23  * SUCH DAMAGE.
  24  */
  25
  26 #include <sys/cdefs.h>
  27 __FBSDID("$FreeBSD$");
  28
  29 #include <sys/param.h>
  30 #include <sys/kernel.h>
  31 #include <sys/malloc.h>
  32 #include <sys/systm.h>
  33 #include <sys/uio.h>
  34
  35 #include <contrib/cloudabi/cloudabi32_types.h>
  36
  37 #include <compat/cloudabi/cloudabi_util.h>
  38
  39 #include <compat/cloudabi32/cloudabi32_proto.h>
  40 #include <compat/cloudabi32/cloudabi32_util.h>
  41
  42 static MALLOC_DEFINE(M_SOCKET, "socket", "CloudABI socket");
  43
  44 int
  45 cloudabi32_sys_sock_recv(struct thread *td,
  46     struct cloudabi32_sys_sock_recv_args *uap)
  47 {
  48         cloudabi32_recv_in_t ri;
  49         cloudabi32_recv_out_t ro = {};
  50         cloudabi32_iovec_t iovobj;
  51         struct iovec *iov;
  52         const cloudabi32_iovec_t *user_iov;
  53         size_t i, rdatalen, rfdslen;
  54         int error;
  55
  56         error = copyin(uap->in, &ri, sizeof(ri));
  57         if (error != 0)
  58                 return (error);
  59
  60         /* Convert iovecs to native format. */
  61         if (ri.ri_data_len > UIO_MAXIOV)
  62                 return (EINVAL);
  63         iov = mallocarray(ri.ri_data_len, sizeof(struct iovec),
  64             M_SOCKET, M_WAITOK);
  65         user_iov = TO_PTR(ri.ri_data);
  66         for (i = 0; i < ri.ri_data_len; i++) {
  67                 error = copyin(&user_iov[i], &iovobj, sizeof(iovobj));
  68                 if (error != 0) {
  69                         free(iov, M_SOCKET);
  70                         return (error);
  71                 }
  72                 iov[i].iov_base = TO_PTR(iovobj.buf);
  73                 iov[i].iov_len = iovobj.buf_len;
  74         }
  75
  76         error = cloudabi_sock_recv(td, uap->sock, iov, ri.ri_data_len,
  77             TO_PTR(ri.ri_fds), ri.ri_fds_len, ri.ri_flags, &rdatalen,
  78             &rfdslen, &ro.ro_flags);
  79         free(iov, M_SOCKET);
  80         if (error != 0)
  81                 return (error);
  82
  83         ro.ro_datalen = rdatalen;
  84         ro.ro_fdslen = rfdslen;
  85         return (copyout(&ro, uap->out, sizeof(ro)));
  86 }
  87
  88 int
  89 cloudabi32_sys_sock_send(struct thread *td,
  90     struct cloudabi32_sys_sock_send_args *uap)
  91 {
  92         cloudabi32_send_in_t si;
  93         cloudabi32_send_out_t so = {};
  94         cloudabi32_ciovec_t iovobj;
  95         struct iovec *iov;
  96         const cloudabi32_ciovec_t *user_iov;
  97         size_t datalen, i;
  98         int error;
  99
 100         error = copyin(uap->in, &si, sizeof(si));
 101         if (error != 0)
 102                 return (error);
 103
 104         /* Convert iovecs to native format. */
 105         if (si.si_data_len > UIO_MAXIOV)
 106                 return (EINVAL);
 107         iov = mallocarray(si.si_data_len, sizeof(struct iovec),
 108             M_SOCKET, M_WAITOK);
 109         user_iov = TO_PTR(si.si_data);
 110         for (i = 0; i < si.si_data_len; i++) {
 111                 error = copyin(&user_iov[i], &iovobj, sizeof(iovobj));
 112                 if (error != 0) {
 113                         free(iov, M_SOCKET);
 114                         return (error);
 115                 }
 116                 iov[i].iov_base = TO_PTR(iovobj.buf);
 117                 iov[i].iov_len = iovobj.buf_len;
 118         }
 119
 120         error = cloudabi_sock_send(td, uap->sock, iov, si.si_data_len,
 121             TO_PTR(si.si_fds), si.si_fds_len, &datalen);
 122         free(iov, M_SOCKET);
 123         if (error != 0)
 124                 return (error);
 125
 126         so.so_datalen = datalen;
 127         return (copyout(&so, uap->out, sizeof(so)));
 128 }