sockin.c revision 1.1.2.2
1 1.1.2.2 mjf /* $NetBSD: sockin.c,v 1.1.2.2 2008/10/05 20:11:34 mjf Exp $ */ 2 1.1.2.2 mjf 3 1.1.2.2 mjf /* 4 1.1.2.2 mjf * Copyright (c) 2008 Antti Kantee. All Rights Reserved. 5 1.1.2.2 mjf * 6 1.1.2.2 mjf * Redistribution and use in source and binary forms, with or without 7 1.1.2.2 mjf * modification, are permitted provided that the following conditions 8 1.1.2.2 mjf * are met: 9 1.1.2.2 mjf * 1. Redistributions of source code must retain the above copyright 10 1.1.2.2 mjf * notice, this list of conditions and the following disclaimer. 11 1.1.2.2 mjf * 2. Redistributions in binary form must reproduce the above copyright 12 1.1.2.2 mjf * notice, this list of conditions and the following disclaimer in the 13 1.1.2.2 mjf * documentation and/or other materials provided with the distribution. 14 1.1.2.2 mjf * 15 1.1.2.2 mjf * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS 16 1.1.2.2 mjf * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 17 1.1.2.2 mjf * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 18 1.1.2.2 mjf * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 1.1.2.2 mjf * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 1.1.2.2 mjf * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 21 1.1.2.2 mjf * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 1.1.2.2 mjf * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 1.1.2.2 mjf * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 1.1.2.2 mjf * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 1.1.2.2 mjf * SUCH DAMAGE. 26 1.1.2.2 mjf */ 27 1.1.2.2 mjf 28 1.1.2.2 mjf #include <sys/param.h> 29 1.1.2.2 mjf #include <sys/condvar.h> 30 1.1.2.2 mjf #include <sys/domain.h> 31 1.1.2.2 mjf #include <sys/kmem.h> 32 1.1.2.2 mjf #include <sys/kthread.h> 33 1.1.2.2 mjf #include <sys/mbuf.h> 34 1.1.2.2 mjf #include <sys/mutex.h> 35 1.1.2.2 mjf #include <sys/poll.h> 36 1.1.2.2 mjf #include <sys/protosw.h> 37 1.1.2.2 mjf #include <sys/queue.h> 38 1.1.2.2 mjf #include <sys/socket.h> 39 1.1.2.2 mjf #include <sys/socketvar.h> 40 1.1.2.2 mjf #include <sys/time.h> 41 1.1.2.2 mjf 42 1.1.2.2 mjf #include <netinet/in.h> 43 1.1.2.2 mjf #include <netinet/in_systm.h> 44 1.1.2.2 mjf #include <netinet/ip.h> 45 1.1.2.2 mjf 46 1.1.2.2 mjf #include <rump/rumpuser.h> 47 1.1.2.2 mjf 48 1.1.2.2 mjf /* 49 1.1.2.2 mjf * An inet communication domain which uses the socket interface. 50 1.1.2.2 mjf * Currently supports only IPv4 UDP, but could easily be extended to 51 1.1.2.2 mjf * support IPv6 and TCP by adding more stuff to the protosw. 52 1.1.2.2 mjf */ 53 1.1.2.2 mjf 54 1.1.2.2 mjf DOMAIN_DEFINE(sockindomain); 55 1.1.2.2 mjf 56 1.1.2.2 mjf static void sockin_init(void); 57 1.1.2.2 mjf static int sockin_usrreq(struct socket *, int, struct mbuf *, 58 1.1.2.2 mjf struct mbuf *, struct mbuf *, struct lwp *); 59 1.1.2.2 mjf 60 1.1.2.2 mjf const struct protosw sockinsw[] = { 61 1.1.2.2 mjf { .pr_type = SOCK_DGRAM, 62 1.1.2.2 mjf .pr_domain = &sockindomain, 63 1.1.2.2 mjf .pr_protocol = IPPROTO_UDP, 64 1.1.2.2 mjf .pr_flags = PR_ATOMIC|PR_ADDR, 65 1.1.2.2 mjf .pr_usrreq = sockin_usrreq, 66 1.1.2.2 mjf .pr_init = sockin_init, 67 1.1.2.2 mjf }}; 68 1.1.2.2 mjf 69 1.1.2.2 mjf struct domain sockindomain = { 70 1.1.2.2 mjf .dom_family = PF_INET, 71 1.1.2.2 mjf .dom_name = "socket_inet", 72 1.1.2.2 mjf .dom_init = NULL, 73 1.1.2.2 mjf .dom_externalize = NULL, 74 1.1.2.2 mjf .dom_dispose = NULL, 75 1.1.2.2 mjf .dom_protosw = sockinsw, 76 1.1.2.2 mjf .dom_protoswNPROTOSW = &sockinsw[__arraycount(&sockinsw)], 77 1.1.2.2 mjf .dom_rtattach = NULL, 78 1.1.2.2 mjf .dom_rtoffset = 0, 79 1.1.2.2 mjf .dom_maxrtkey = 0, 80 1.1.2.2 mjf .dom_ifattach = NULL, 81 1.1.2.2 mjf .dom_ifdetach = NULL, 82 1.1.2.2 mjf .dom_ifqueues = { NULL }, 83 1.1.2.2 mjf .dom_link = { NULL }, 84 1.1.2.2 mjf .dom_mowner = MOWNER_INIT("",""), 85 1.1.2.2 mjf .dom_rtcache = { NULL }, 86 1.1.2.2 mjf .dom_sockaddr_cmp = NULL 87 1.1.2.2 mjf }; 88 1.1.2.2 mjf 89 1.1.2.2 mjf /* only for testing */ 90 1.1.2.2 mjf #if 0 91 1.1.2.2 mjf #define SOCKIN_NOTHREAD 92 1.1.2.2 mjf #endif 93 1.1.2.2 mjf 94 1.1.2.2 mjf #define SO2S(so) ((int)(so->so_internal)) 95 1.1.2.2 mjf 96 1.1.2.2 mjf static void 97 1.1.2.2 mjf sockin_process(struct socket *so) 98 1.1.2.2 mjf { 99 1.1.2.2 mjf struct sockaddr_in from; 100 1.1.2.2 mjf struct iovec io; 101 1.1.2.2 mjf struct msghdr rmsg; 102 1.1.2.2 mjf struct mbuf *m; 103 1.1.2.2 mjf ssize_t n; 104 1.1.2.2 mjf size_t plen; 105 1.1.2.2 mjf int error; 106 1.1.2.2 mjf 107 1.1.2.2 mjf plen = IP_MAXPACKET; 108 1.1.2.2 mjf m = m_gethdr(M_WAIT, MT_DATA); 109 1.1.2.2 mjf MEXTMALLOC(m, plen, M_WAIT); 110 1.1.2.2 mjf 111 1.1.2.2 mjf memset(&rmsg, 0, sizeof(rmsg)); 112 1.1.2.2 mjf io.iov_base = mtod(m, void *); 113 1.1.2.2 mjf io.iov_len = plen; 114 1.1.2.2 mjf rmsg.msg_iov = &io; 115 1.1.2.2 mjf rmsg.msg_iovlen = 1; 116 1.1.2.2 mjf rmsg.msg_name = (struct sockaddr *)&from; 117 1.1.2.2 mjf rmsg.msg_namelen = sizeof(from); 118 1.1.2.2 mjf 119 1.1.2.2 mjf n = rumpuser_net_recvmsg(SO2S(so), &rmsg, 0, &error); 120 1.1.2.2 mjf if (n <= 0) { 121 1.1.2.2 mjf m_freem(m); 122 1.1.2.2 mjf return; 123 1.1.2.2 mjf } 124 1.1.2.2 mjf m->m_len = m->m_pkthdr.len = n; 125 1.1.2.2 mjf 126 1.1.2.2 mjf if (!sbappendaddr(&so->so_rcv, rmsg.msg_name, m, NULL)) { 127 1.1.2.2 mjf m_freem(m); 128 1.1.2.2 mjf } 129 1.1.2.2 mjf sorwakeup(so); 130 1.1.2.2 mjf } 131 1.1.2.2 mjf 132 1.1.2.2 mjf struct sockin_unit { 133 1.1.2.2 mjf struct socket *su_so; 134 1.1.2.2 mjf 135 1.1.2.2 mjf LIST_ENTRY(sockin_unit) su_entries; 136 1.1.2.2 mjf }; 137 1.1.2.2 mjf static LIST_HEAD(, sockin_unit) su_ent = LIST_HEAD_INITIALIZER(su_ent); 138 1.1.2.2 mjf static kmutex_t su_mtx; 139 1.1.2.2 mjf static bool rebuild; 140 1.1.2.2 mjf static int nsock; 141 1.1.2.2 mjf 142 1.1.2.2 mjf #ifndef SOCKIN_NOTHREAD 143 1.1.2.2 mjf #define POLLTIMEOUT 100 /* check for new entries every 100ms */ 144 1.1.2.2 mjf 145 1.1.2.2 mjf /* XXX: doesn't handle socket (kernel) locking properly? */ 146 1.1.2.2 mjf static void 147 1.1.2.2 mjf sockinworker(void *arg) 148 1.1.2.2 mjf { 149 1.1.2.2 mjf struct pollfd *pfds = NULL, *npfds; 150 1.1.2.2 mjf struct sockin_unit *su_iter; 151 1.1.2.2 mjf int cursock = 0, i, rv, error; 152 1.1.2.2 mjf 153 1.1.2.2 mjf /* 154 1.1.2.2 mjf * Loop reading requests. Check for new sockets periodically 155 1.1.2.2 mjf * (could be smarter, but I'm lazy). 156 1.1.2.2 mjf */ 157 1.1.2.2 mjf for (;;) { 158 1.1.2.2 mjf if (rebuild) { 159 1.1.2.2 mjf npfds = NULL; 160 1.1.2.2 mjf mutex_enter(&su_mtx); 161 1.1.2.2 mjf if (nsock) 162 1.1.2.2 mjf npfds = kmem_alloc(nsock * sizeof(*npfds), 163 1.1.2.2 mjf KM_NOSLEEP); 164 1.1.2.2 mjf if (npfds || nsock == 0) { 165 1.1.2.2 mjf if (pfds) 166 1.1.2.2 mjf kmem_free(pfds, cursock*sizeof(*pfds)); 167 1.1.2.2 mjf pfds = npfds; 168 1.1.2.2 mjf cursock = nsock; 169 1.1.2.2 mjf rebuild = false; 170 1.1.2.2 mjf 171 1.1.2.2 mjf i = 0; 172 1.1.2.2 mjf LIST_FOREACH(su_iter, &su_ent, su_entries) { 173 1.1.2.2 mjf pfds[i].fd = SO2S(su_iter->su_so); 174 1.1.2.2 mjf pfds[i].events = POLLIN; 175 1.1.2.2 mjf pfds[i].revents = 0; 176 1.1.2.2 mjf i++; 177 1.1.2.2 mjf } 178 1.1.2.2 mjf KASSERT(i == nsock); 179 1.1.2.2 mjf } 180 1.1.2.2 mjf mutex_exit(&su_mtx); 181 1.1.2.2 mjf } 182 1.1.2.2 mjf 183 1.1.2.2 mjf /* find affected sockets & process */ 184 1.1.2.2 mjf rv = rumpuser_poll(pfds, cursock, POLLTIMEOUT, &error); 185 1.1.2.2 mjf for (i = 0; i < cursock && rv > 0; i++) { 186 1.1.2.2 mjf if (pfds[i].revents & POLLIN) { 187 1.1.2.2 mjf mutex_enter(&su_mtx); 188 1.1.2.2 mjf LIST_FOREACH(su_iter, &su_ent, su_entries) { 189 1.1.2.2 mjf if (SO2S(su_iter->su_so)==pfds[i].fd) { 190 1.1.2.2 mjf mutex_enter(softnet_lock); 191 1.1.2.2 mjf sockin_process(su_iter->su_so); 192 1.1.2.2 mjf mutex_exit(softnet_lock); 193 1.1.2.2 mjf break; 194 1.1.2.2 mjf } 195 1.1.2.2 mjf } 196 1.1.2.2 mjf /* if we can't find it, just wing it */ 197 1.1.2.2 mjf KASSERT(rebuild || su_iter); 198 1.1.2.2 mjf mutex_exit(&su_mtx); 199 1.1.2.2 mjf pfds[i].revents = 0; 200 1.1.2.2 mjf rv--; 201 1.1.2.2 mjf i = -1; 202 1.1.2.2 mjf continue; 203 1.1.2.2 mjf } 204 1.1.2.2 mjf 205 1.1.2.2 mjf /* something else? ignore */ 206 1.1.2.2 mjf if (pfds[i].revents) { 207 1.1.2.2 mjf pfds[i].revents = 0; 208 1.1.2.2 mjf rv--; 209 1.1.2.2 mjf } 210 1.1.2.2 mjf } 211 1.1.2.2 mjf KASSERT(rv <= 0); 212 1.1.2.2 mjf } 213 1.1.2.2 mjf 214 1.1.2.2 mjf } 215 1.1.2.2 mjf #endif /* SOCKIN_NOTHREAD */ 216 1.1.2.2 mjf 217 1.1.2.2 mjf static void 218 1.1.2.2 mjf sockin_init() 219 1.1.2.2 mjf { 220 1.1.2.2 mjf #ifndef SOCKIN_NOTHREAD 221 1.1.2.2 mjf int rv; 222 1.1.2.2 mjf 223 1.1.2.2 mjf if ((rv = kthread_create(PRI_NONE, 0, NULL, sockinworker, 224 1.1.2.2 mjf NULL, NULL, "sockwork")) != 0) 225 1.1.2.2 mjf panic("sockin_init: could not create worker thread\n"); 226 1.1.2.2 mjf #endif 227 1.1.2.2 mjf mutex_init(&su_mtx, MUTEX_DEFAULT, IPL_NONE); 228 1.1.2.2 mjf } 229 1.1.2.2 mjf 230 1.1.2.2 mjf static int 231 1.1.2.2 mjf sockin_usrreq(struct socket *so, int req, struct mbuf *m, struct mbuf *nam, 232 1.1.2.2 mjf struct mbuf *control, struct lwp *l) 233 1.1.2.2 mjf { 234 1.1.2.2 mjf int error = 0; 235 1.1.2.2 mjf 236 1.1.2.2 mjf switch (req) { 237 1.1.2.2 mjf case PRU_ATTACH: 238 1.1.2.2 mjf { 239 1.1.2.2 mjf struct sockin_unit *su; 240 1.1.2.2 mjf int news; 241 1.1.2.2 mjf 242 1.1.2.2 mjf sosetlock(so); 243 1.1.2.2 mjf 244 1.1.2.2 mjf su = kmem_alloc(sizeof(*su), KM_NOSLEEP); 245 1.1.2.2 mjf if (!su) { 246 1.1.2.2 mjf error = ENOMEM; 247 1.1.2.2 mjf break; 248 1.1.2.2 mjf } 249 1.1.2.2 mjf 250 1.1.2.2 mjf news = rumpuser_net_socket(PF_INET, SOCK_DGRAM, 0, &error); 251 1.1.2.2 mjf if (news == -1) { 252 1.1.2.2 mjf kmem_free(su, sizeof(*su)); 253 1.1.2.2 mjf break; 254 1.1.2.2 mjf } 255 1.1.2.2 mjf so->so_internal = (void *)news; 256 1.1.2.2 mjf su->su_so = so; 257 1.1.2.2 mjf 258 1.1.2.2 mjf mutex_enter(&su_mtx); 259 1.1.2.2 mjf LIST_INSERT_HEAD(&su_ent, su, su_entries); 260 1.1.2.2 mjf nsock++; 261 1.1.2.2 mjf rebuild = true; 262 1.1.2.2 mjf mutex_exit(&su_mtx); 263 1.1.2.2 mjf break; 264 1.1.2.2 mjf } 265 1.1.2.2 mjf 266 1.1.2.2 mjf case PRU_CONNECT: 267 1.1.2.2 mjf /* sorry dear, not today */ 268 1.1.2.2 mjf break; 269 1.1.2.2 mjf 270 1.1.2.2 mjf case PRU_SEND: 271 1.1.2.2 mjf { 272 1.1.2.2 mjf struct sockaddr *saddr; 273 1.1.2.2 mjf struct msghdr mhdr; 274 1.1.2.2 mjf struct iovec iov[16]; 275 1.1.2.2 mjf struct mbuf *m2; 276 1.1.2.2 mjf size_t tot; 277 1.1.2.2 mjf int i, s; 278 1.1.2.2 mjf 279 1.1.2.2 mjf memset(&mhdr, 0, sizeof(mhdr)); 280 1.1.2.2 mjf 281 1.1.2.2 mjf tot = 0; 282 1.1.2.2 mjf for (i = 0, m2 = m; m2; m2 = m2->m_next, i++) { 283 1.1.2.2 mjf if (i > 16) 284 1.1.2.2 mjf panic("lazy bum"); 285 1.1.2.2 mjf iov[i].iov_base = m2->m_data; 286 1.1.2.2 mjf iov[i].iov_len = m2->m_len; 287 1.1.2.2 mjf tot += m2->m_len; 288 1.1.2.2 mjf 289 1.1.2.2 mjf } 290 1.1.2.2 mjf saddr = mtod(nam, struct sockaddr *); 291 1.1.2.2 mjf mhdr.msg_name = saddr; 292 1.1.2.2 mjf mhdr.msg_namelen = saddr->sa_len; 293 1.1.2.2 mjf mhdr.msg_iov = iov; 294 1.1.2.2 mjf mhdr.msg_iovlen = i; 295 1.1.2.2 mjf 296 1.1.2.2 mjf s = (int)so->so_internal; 297 1.1.2.2 mjf rumpuser_net_sendmsg(s, &mhdr, 0, &error); 298 1.1.2.2 mjf m_freem(m); 299 1.1.2.2 mjf m_freem(control); 300 1.1.2.2 mjf #ifdef SOCKIN_NOTHREAD 301 1.1.2.2 mjf /* this assumes too many things to list.. buthey, testing */ 302 1.1.2.2 mjf sockin_process(so); 303 1.1.2.2 mjf #endif 304 1.1.2.2 mjf } 305 1.1.2.2 mjf break; 306 1.1.2.2 mjf 307 1.1.2.2 mjf case PRU_SHUTDOWN: 308 1.1.2.2 mjf { 309 1.1.2.2 mjf struct sockin_unit *su_iter; 310 1.1.2.2 mjf 311 1.1.2.2 mjf mutex_enter(&su_mtx); 312 1.1.2.2 mjf LIST_FOREACH(su_iter, &su_ent, su_entries) { 313 1.1.2.2 mjf if (su_iter->su_so == so) 314 1.1.2.2 mjf break; 315 1.1.2.2 mjf } 316 1.1.2.2 mjf if (!su_iter) 317 1.1.2.2 mjf panic("no such socket"); 318 1.1.2.2 mjf 319 1.1.2.2 mjf LIST_REMOVE(su_iter, su_entries); 320 1.1.2.2 mjf nsock--; 321 1.1.2.2 mjf rebuild = true; 322 1.1.2.2 mjf mutex_exit(&su_mtx); 323 1.1.2.2 mjf 324 1.1.2.2 mjf rumpuser_close(SO2S(su_iter->su_so), &error); 325 1.1.2.2 mjf kmem_free(su_iter, sizeof(*su_iter)); 326 1.1.2.2 mjf } 327 1.1.2.2 mjf break; 328 1.1.2.2 mjf 329 1.1.2.2 mjf default: 330 1.1.2.2 mjf panic("sockin_usrreq: IMPLEMENT ME, req %d not supported", req); 331 1.1.2.2 mjf } 332 1.1.2.2 mjf 333 1.1.2.2 mjf return error; 334 1.1.2.2 mjf } 335
Indexes created Wed Oct 01 19:09:53 GMT 2025