sys/dev/iscsi/icl_soft_proxy.c

   1 /*-
   2  * Copyright (c) 2012 The FreeBSD Foundation
   3  * All rights reserved.
   4  *
   5  * This software was developed by Edward Tomasz Napierala under sponsorship
   6  * from the FreeBSD Foundation.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  *
  29  */
  30 /*-
  31  * Copyright (c) 1982, 1986, 1989, 1990, 1993
  32  *      The Regents of the University of California.  All rights reserved.
  33  *
  34  * sendfile(2) and related extensions:
  35  * Copyright (c) 1998, David Greenman. All rights reserved.
  36  *
  37  * Redistribution and use in source and binary forms, with or without
  38  * modification, are permitted provided that the following conditions
  39  * are met:
  40  * 1. Redistributions of source code must retain the above copyright
  41  *    notice, this list of conditions and the following disclaimer.
  42  * 2. Redistributions in binary form must reproduce the above copyright
  43  *    notice, this list of conditions and the following disclaimer in the
  44  *    documentation and/or other materials provided with the distribution.
  45  * 3. Neither the name of the University nor the names of its contributors
  46  *    may be used to endorse or promote products derived from this software
  47  *    without specific prior written permission.
  48  *
  49  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  50  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  51  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  52  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  53  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  54  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  55  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  56  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  57  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  58  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  59  * SUCH DAMAGE.
  60  *
  61  *      @(#)uipc_syscalls.c     8.4 (Berkeley) 2/21/94
  62  */
  63
  64 /*
  65  * iSCSI Common Layer, kernel proxy part.
  66  */
  67
  68 #ifdef ICL_KERNEL_PROXY
  69
  70 #include <sys/cdefs.h>
  71 __FBSDID("$FreeBSD$");
  72
  73 #include <sys/param.h>
  74 #include <sys/capsicum.h>
  75 #include <sys/condvar.h>
  76 #include <sys/conf.h>
  77 #include <sys/kernel.h>
  78 #include <sys/kthread.h>
  79 #include <sys/malloc.h>
  80 #include <sys/proc.h>
  81 #include <sys/socket.h>
  82 #include <sys/socketvar.h>
  83 #include <sys/sx.h>
  84 #include <sys/systm.h>
  85 #include <netinet/in.h>
  86 #include <netinet/tcp.h>
  87
  88 #include <dev/iscsi/icl.h>
  89
  90 struct icl_listen_sock {
  91         TAILQ_ENTRY(icl_listen_sock)    ils_next;
  92         struct icl_listen               *ils_listen;
  93         struct socket                   *ils_socket;
  94         bool                            ils_running;
  95         bool                            ils_disconnecting;
  96         int                             ils_id;
  97 };
  98
  99 struct icl_listen       {
 100         TAILQ_HEAD(, icl_listen_sock)   il_sockets;
 101         struct sx                       il_lock;
 102         void                            (*il_accept)(struct socket *,
 103                                             struct sockaddr *, int);
 104 };
 105
 106 static MALLOC_DEFINE(M_ICL_PROXY, "ICL_PROXY", "iSCSI common layer proxy");
 107
 108 int
 109 icl_soft_proxy_connect(struct icl_conn *ic, int domain, int socktype,
 110     int protocol, struct sockaddr *from_sa, struct sockaddr *to_sa)
 111 {
 112         struct socket *so;
 113         int error;
 114         int interrupted = 0;
 115
 116         error = socreate(domain, &so, socktype, protocol,
 117             curthread->td_ucred, curthread);
 118         if (error != 0)
 119                 return (error);
 120
 121         if (from_sa != NULL) {
 122                 error = sobind(so, from_sa, curthread);
 123                 if (error != 0) {
 124                         soclose(so);
 125                         return (error);
 126                 }
 127         }
 128
 129         error = soconnect(so, to_sa, curthread);
 130         if (error != 0) {
 131                 soclose(so);
 132                 return (error);
 133         }
 134
 135         SOCK_LOCK(so);
 136         while ((so->so_state & SS_ISCONNECTING) && so->so_error == 0) {
 137                 error = msleep(&so->so_timeo, SOCK_MTX(so), PSOCK | PCATCH,
 138                     "icl_connect", 0);
 139                 if (error) {
 140                         if (error == EINTR || error == ERESTART)
 141                                 interrupted = 1;
 142                         break;
 143                 }
 144         }
 145         if (error == 0) {
 146                 error = so->so_error;
 147                 so->so_error = 0;
 148         }
 149         SOCK_UNLOCK(so);
 150
 151         if (error != 0) {
 152                 soclose(so);
 153                 return (error);
 154         }
 155
 156         error = icl_soft_handoff_sock(ic, so);
 157         if (error != 0)
 158                 soclose(so);
 159
 160         return (error);
 161 }
 162
 163 struct icl_listen *
 164 icl_listen_new(void (*accept_cb)(struct socket *, struct sockaddr *, int))
 165 {
 166         struct icl_listen *il;
 167
 168         il = malloc(sizeof(*il), M_ICL_PROXY, M_ZERO | M_WAITOK);
 169         TAILQ_INIT(&il->il_sockets);
 170         sx_init(&il->il_lock, "icl_listen");
 171         il->il_accept = accept_cb;
 172
 173         return (il);
 174 }
 175
 176 void
 177 icl_listen_free(struct icl_listen *il)
 178 {
 179         struct icl_listen_sock *ils;
 180
 181         sx_xlock(&il->il_lock);
 182         while (!TAILQ_EMPTY(&il->il_sockets)) {
 183                 ils = TAILQ_FIRST(&il->il_sockets);
 184                 while (ils->ils_running) {
 185                         ICL_DEBUG("waiting for accept thread to terminate");
 186                         sx_xunlock(&il->il_lock);
 187                         ils->ils_disconnecting = true;
 188                         wakeup(&ils->ils_socket->so_timeo);
 189                         pause("icl_unlisten", 1 * hz);
 190                         sx_xlock(&il->il_lock);
 191                 }
 192
 193                 TAILQ_REMOVE(&il->il_sockets, ils, ils_next);
 194                 soclose(ils->ils_socket);
 195                 free(ils, M_ICL_PROXY);
 196         }
 197         sx_xunlock(&il->il_lock);
 198
 199         free(il, M_ICL_PROXY);
 200 }
 201
 202 /*
 203  * XXX: Doing accept in a separate thread in each socket might not be the best way
 204  *      to do stuff, but it's pretty clean and debuggable - and you probably won't
 205  *      have hundreds of listening sockets anyway.
 206  */
 207 static void
 208 icl_accept_thread(void *arg)
 209 {
 210         struct icl_listen_sock *ils;
 211         struct socket *head, *so;
 212         struct sockaddr *sa;
 213         int error;
 214
 215         ils = arg;
 216         head = ils->ils_socket;
 217
 218         ils->ils_running = true;
 219
 220         for (;;) {
 221                 ACCEPT_LOCK();
 222                 while (TAILQ_EMPTY(&head->so_comp) && head->so_error == 0 && ils->ils_disconnecting == false) {
 223                         if (head->so_rcv.sb_state & SBS_CANTRCVMORE) {
 224                                 head->so_error = ECONNABORTED;
 225                                 break;
 226                         }
 227                         error = msleep(&head->so_timeo, &accept_mtx, PSOCK | PCATCH,
 228                             "accept", 0);
 229                         if (error) {
 230                                 ACCEPT_UNLOCK();
 231                                 ICL_WARN("msleep failed with error %d", error);
 232                                 continue;
 233                         }
 234                         if (ils->ils_disconnecting) {
 235                                 ACCEPT_UNLOCK();
 236                                 ICL_DEBUG("terminating");
 237                                 ils->ils_running = false;
 238                                 kthread_exit();
 239                                 return;
 240                         }
 241                 }
 242                 if (head->so_error) {
 243                         error = head->so_error;
 244                         head->so_error = 0;
 245                         ACCEPT_UNLOCK();
 246                         ICL_WARN("socket error %d", error);
 247                         continue;
 248                 }
 249                 so = TAILQ_FIRST(&head->so_comp);
 250                 KASSERT(so != NULL, ("NULL so"));
 251                 KASSERT(!(so->so_qstate & SQ_INCOMP), ("accept1: so SQ_INCOMP"));
 252                 KASSERT(so->so_qstate & SQ_COMP, ("accept1: so not SQ_COMP"));
 253
 254                 /*
 255                  * Before changing the flags on the socket, we have to bump the
 256                  * reference count.  Otherwise, if the protocol calls sofree(),
 257                  * the socket will be released due to a zero refcount.
 258                  */
 259                 SOCK_LOCK(so);                  /* soref() and so_state update */
 260                 soref(so);                      /* file descriptor reference */
 261
 262                 TAILQ_REMOVE(&head->so_comp, so, so_list);
 263                 head->so_qlen--;
 264                 so->so_state |= (head->so_state & SS_NBIO);
 265                 so->so_qstate &= ~SQ_COMP;
 266                 so->so_head = NULL;
 267
 268                 SOCK_UNLOCK(so);
 269                 ACCEPT_UNLOCK();
 270
 271                 sa = NULL;
 272                 error = soaccept(so, &sa);
 273                 if (error != 0) {
 274                         ICL_WARN("soaccept error %d", error);
 275                         if (sa != NULL)
 276                                 free(sa, M_SONAME);
 277                         soclose(so);
 278                         continue;
 279                 }
 280
 281                 (ils->ils_listen->il_accept)(so, sa, ils->ils_id);
 282         }
 283 }
 284
 285 static int
 286 icl_listen_add_tcp(struct icl_listen *il, int domain, int socktype,
 287     int protocol, struct sockaddr *sa, int portal_id)
 288 {
 289         struct icl_listen_sock *ils;
 290         struct socket *so;
 291         struct sockopt sopt;
 292         int error, one = 1;
 293
 294         error = socreate(domain, &so, socktype, protocol,
 295             curthread->td_ucred, curthread);
 296         if (error != 0) {
 297                 ICL_WARN("socreate failed with error %d", error);
 298                 return (error);
 299         }
 300
 301         sopt.sopt_dir = SOPT_SET;
 302         sopt.sopt_level = SOL_SOCKET;
 303         sopt.sopt_name = SO_REUSEADDR;
 304         sopt.sopt_val = &one;
 305         sopt.sopt_valsize = sizeof(one);
 306         sopt.sopt_td = NULL;
 307         error = sosetopt(so, &sopt);
 308         if (error != 0) {
 309                 ICL_WARN("failed to set SO_REUSEADDR with error %d", error);
 310                 soclose(so);
 311                 return (error);
 312         }
 313
 314         error = sobind(so, sa, curthread);
 315         if (error != 0) {
 316                 ICL_WARN("sobind failed with error %d", error);
 317                 soclose(so);
 318                 return (error);
 319         }
 320
 321         error = solisten(so, -1, curthread);
 322         if (error != 0) {
 323                 ICL_WARN("solisten failed with error %d", error);
 324                 soclose(so);
 325                 return (error);
 326         }
 327
 328         ils = malloc(sizeof(*ils), M_ICL_PROXY, M_ZERO | M_WAITOK);
 329         ils->ils_listen = il;
 330         ils->ils_socket = so;
 331         ils->ils_id = portal_id;
 332
 333         error = kthread_add(icl_accept_thread, ils, NULL, NULL, 0, 0, "iclacc");
 334         if (error != 0) {
 335                 ICL_WARN("kthread_add failed with error %d", error);
 336                 soclose(so);
 337                 free(ils, M_ICL_PROXY);
 338
 339                 return (error);
 340         }
 341
 342         sx_xlock(&il->il_lock);
 343         TAILQ_INSERT_TAIL(&il->il_sockets, ils, ils_next);
 344         sx_xunlock(&il->il_lock);
 345
 346         return (0);
 347 }
 348
 349 int
 350 icl_listen_add(struct icl_listen *il, bool rdma, int domain, int socktype,
 351     int protocol, struct sockaddr *sa, int portal_id)
 352 {
 353
 354         if (rdma) {
 355                 ICL_DEBUG("RDMA not supported");
 356                 return (EOPNOTSUPP);
 357         }
 358
 359
 360         return (icl_listen_add_tcp(il, domain, socktype, protocol, sa,
 361             portal_id));
 362 }
 363
 364 int
 365 icl_listen_remove(struct icl_listen *il, struct sockaddr *sa)
 366 {
 367
 368         /*
 369          * XXX
 370          */
 371
 372         return (EOPNOTSUPP);
 373 }
 374
 375 #endif /* ICL_KERNEL_PROXY */