sys/netncp/ncp_subr.c

   1 /*-
   2  * Copyright (c) 1999, 2000, 2001 Boris Popov
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *    This product includes software developed by Boris Popov.
  16  * 4. Neither the name of the author nor the names of any co-contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  */
  32
  33 #include <sys/cdefs.h>
  34 __FBSDID("$FreeBSD$");
  35
  36 #include <sys/param.h>
  37 #include <sys/systm.h>
  38 #include <sys/errno.h>
  39 #include <sys/eventhandler.h>
  40 #include <sys/kernel.h>
  41 #include <sys/lock.h>
  42 #include <sys/malloc.h>
  43 #include <sys/proc.h>
  44 #include <sys/sysctl.h>
  45 #include <sys/time.h>
  46
  47 #include <netncp/ncp.h>
  48 #include <netncp/ncp_conn.h>
  49 #include <netncp/ncp_sock.h>
  50 #include <netncp/ncp_subr.h>
  51 #include <netncp/ncp_rq.h>
  52 #include <netncp/ncp_ncp.h>
  53 #include <netncp/nwerror.h>
  54
  55 int ncp_debuglevel = 0;
  56
  57 struct callout_handle ncp_timer_handle;
  58 static eventhandler_tag ncp_exit_tag;
  59
  60 static void ncp_at_exit(void *arg, struct proc *p);
  61 static void ncp_timer(void *arg);
  62
  63 /*
  64  * duplicate string from user space. It should be very-very slow.
  65  */
  66 char *
  67 ncp_str_dup(char *s) {
  68         char *p, bt;
  69         int len = 0;
  70
  71         for (p = s;;p++) {
  72                 if (copyin(p, &bt, 1)) return NULL;
  73                 len++;
  74                 if (bt == 0) break;
  75         }
  76         p = malloc(len, M_NCPDATA, M_WAITOK);
  77         copyin(s, p, len);
  78         return p;
  79 }
  80
  81
  82 void
  83 ncp_at_exit(void *arg, struct proc *p)
  84 {
  85         struct ncp_conn *ncp, *nncp;
  86         struct thread *td;
  87
  88         mtx_lock(&Giant);
  89         FOREACH_THREAD_IN_PROC(p, td) {
  90                 if (ncp_conn_putprochandles(td) == 0)
  91                         continue;
  92
  93                 ncp_conn_locklist(LK_EXCLUSIVE, td);
  94                 for (ncp = SLIST_FIRST(&conn_list); ncp; ncp = nncp) {
  95                         nncp = SLIST_NEXT(ncp, nc_next);
  96                         if (ncp_conn_lock(ncp, td, td->td_ucred,
  97                                           NCPM_READ | NCPM_EXECUTE | NCPM_WRITE))
  98                                 continue;
  99                         if (ncp_conn_free(ncp) != 0)
 100                                 ncp_conn_unlock(ncp, td);
 101                 }
 102                 ncp_conn_unlocklist(td);
 103         }
 104         mtx_unlock(&Giant);
 105 }
 106
 107 int
 108 ncp_init(void)
 109 {
 110         ncp_conn_init();
 111         ncp_exit_tag = EVENTHANDLER_REGISTER(process_exit, ncp_at_exit, NULL,
 112             EVENTHANDLER_PRI_ANY);
 113         ncp_timer_handle = timeout(ncp_timer, NULL, NCP_TIMER_TICK);
 114         return 0;
 115 }
 116
 117 int
 118 ncp_done(void)
 119 {
 120         int error;
 121
 122         error = ncp_conn_destroy();
 123         if (error)
 124                 return error;
 125         untimeout(ncp_timer, NULL, ncp_timer_handle);
 126         EVENTHANDLER_DEREGISTER(process_exit, ncp_exit_tag);
 127         return 0;
 128 }
 129
 130
 131 /* tick every second and check for watch dog packets and lost connections */
 132 static void
 133 ncp_timer(void *arg)
 134 {
 135         struct ncp_conn *conn;
 136
 137         if(ncp_conn_locklist(LK_SHARED | LK_NOWAIT, NULL) == 0) {
 138                 SLIST_FOREACH(conn, &conn_list, nc_next)
 139                         ncp_check_conn(conn);
 140                 ncp_conn_unlocklist(NULL);
 141         }
 142         ncp_timer_handle = timeout(ncp_timer, NULL, NCP_TIMER_TICK);
 143 }