ng_ksocket.c revision 184205
1/* 2 * ng_ksocket.c 3 */ 4 5/*- 6 * Copyright (c) 1996-1999 Whistle Communications, Inc. 7 * All rights reserved. 8 * 9 * Subject to the following obligations and disclaimer of warranty, use and 10 * redistribution of this software, in source or object code forms, with or 11 * without modifications are expressly permitted by Whistle Communications; 12 * provided, however, that: 13 * 1. Any and all reproductions of the source or object code must include the 14 * copyright notice above and the following disclaimer of warranties; and 15 * 2. No rights are granted, in any manner or form, to use Whistle 16 * Communications, Inc. trademarks, including the mark "WHISTLE 17 * COMMUNICATIONS" on advertising, endorsements, or otherwise except as 18 * such appears in the above copyright notice or in the software. 19 * 20 * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND 21 * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO 22 * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE, 23 * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF 24 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. 25 * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY 26 * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS 27 * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE. 28 * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES 29 * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING 30 * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, 31 * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR 32 * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY 33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 35 * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY 36 * OF SUCH DAMAGE. 37 * 38 * Author: Archie Cobbs <archie@freebsd.org> 39 * 40 * $FreeBSD: head/sys/netgraph/ng_ksocket.c 184205 2008-10-23 15:53:51Z des $ 41 * $Whistle: ng_ksocket.c,v 1.1 1999/11/16 20:04:40 archie Exp $ 42 */ 43 44/* 45 * Kernel socket node type. This node type is basically a kernel-mode 46 * version of a socket... kindof like the reverse of the socket node type. 47 */ 48 49#include <sys/param.h> 50#include <sys/systm.h> 51#include <sys/kernel.h> 52#include <sys/mbuf.h> 53#include <sys/proc.h> 54#include <sys/malloc.h> 55#include <sys/ctype.h> 56#include <sys/protosw.h> 57#include <sys/errno.h> 58#include <sys/socket.h> 59#include <sys/socketvar.h> 60#include <sys/uio.h> 61#include <sys/un.h> 62 63#include <netgraph/ng_message.h> 64#include <netgraph/netgraph.h> 65#include <netgraph/ng_parse.h> 66#include <netgraph/ng_ksocket.h> 67 68#include <netinet/in.h> 69#include <netatalk/at.h> 70 71#ifdef NG_SEPARATE_MALLOC 72MALLOC_DEFINE(M_NETGRAPH_KSOCKET, "netgraph_ksock", "netgraph ksock node "); 73#else 74#define M_NETGRAPH_KSOCKET M_NETGRAPH 75#endif 76 77#define OFFSETOF(s, e) ((char *)&((s *)0)->e - (char *)((s *)0)) 78#define SADATA_OFFSET (OFFSETOF(struct sockaddr, sa_data)) 79 80/* Node private data */ 81struct ng_ksocket_private { 82 node_p node; 83 hook_p hook; 84 struct socket *so; 85 int fn_sent; /* FN call on incoming event was sent */ 86 LIST_HEAD(, ng_ksocket_private) embryos; 87 LIST_ENTRY(ng_ksocket_private) siblings; 88 u_int32_t flags; 89 u_int32_t response_token; 90 ng_ID_t response_addr; 91}; 92typedef struct ng_ksocket_private *priv_p; 93 94/* Flags for priv_p */ 95#define KSF_CONNECTING 0x00000001 /* Waiting for connection complete */ 96#define KSF_ACCEPTING 0x00000002 /* Waiting for accept complete */ 97#define KSF_EOFSEEN 0x00000004 /* Have sent 0-length EOF mbuf */ 98#define KSF_CLONED 0x00000008 /* Cloned from an accepting socket */ 99#define KSF_EMBRYONIC 0x00000010 /* Cloned node with no hooks yet */ 100 101/* Netgraph node methods */ 102static ng_constructor_t ng_ksocket_constructor; 103static ng_rcvmsg_t ng_ksocket_rcvmsg; 104static ng_shutdown_t ng_ksocket_shutdown; 105static ng_newhook_t ng_ksocket_newhook; 106static ng_rcvdata_t ng_ksocket_rcvdata; 107static ng_connect_t ng_ksocket_connect; 108static ng_disconnect_t ng_ksocket_disconnect; 109 110/* Alias structure */ 111struct ng_ksocket_alias { 112 const char *name; 113 const int value; 114 const int family; 115}; 116 117/* Protocol family aliases */ 118static const struct ng_ksocket_alias ng_ksocket_families[] = { 119 { "local", PF_LOCAL }, 120 { "inet", PF_INET }, 121 { "inet6", PF_INET6 }, 122 { "atalk", PF_APPLETALK }, 123 { "ipx", PF_IPX }, 124 { "atm", PF_ATM }, 125 { NULL, -1 }, 126}; 127 128/* Socket type aliases */ 129static const struct ng_ksocket_alias ng_ksocket_types[] = { 130 { "stream", SOCK_STREAM }, 131 { "dgram", SOCK_DGRAM }, 132 { "raw", SOCK_RAW }, 133 { "rdm", SOCK_RDM }, 134 { "seqpacket", SOCK_SEQPACKET }, 135 { NULL, -1 }, 136}; 137 138/* Protocol aliases */ 139static const struct ng_ksocket_alias ng_ksocket_protos[] = { 140 { "ip", IPPROTO_IP, PF_INET }, 141 { "raw", IPPROTO_RAW, PF_INET }, 142 { "icmp", IPPROTO_ICMP, PF_INET }, 143 { "igmp", IPPROTO_IGMP, PF_INET }, 144 { "tcp", IPPROTO_TCP, PF_INET }, 145 { "udp", IPPROTO_UDP, PF_INET }, 146 { "gre", IPPROTO_GRE, PF_INET }, 147 { "esp", IPPROTO_ESP, PF_INET }, 148 { "ah", IPPROTO_AH, PF_INET }, 149 { "swipe", IPPROTO_SWIPE, PF_INET }, 150 { "encap", IPPROTO_ENCAP, PF_INET }, 151 { "divert", IPPROTO_DIVERT, PF_INET }, 152 { "pim", IPPROTO_PIM, PF_INET }, 153 { "ddp", ATPROTO_DDP, PF_APPLETALK }, 154 { "aarp", ATPROTO_AARP, PF_APPLETALK }, 155 { NULL, -1 }, 156}; 157 158/* Helper functions */ 159static int ng_ksocket_check_accept(priv_p); 160static void ng_ksocket_finish_accept(priv_p); 161static void ng_ksocket_incoming(struct socket *so, void *arg, int waitflag); 162static int ng_ksocket_parse(const struct ng_ksocket_alias *aliases, 163 const char *s, int family); 164static void ng_ksocket_incoming2(node_p node, hook_p hook, 165 void *arg1, int arg2); 166 167/************************************************************************ 168 STRUCT SOCKADDR PARSE TYPE 169 ************************************************************************/ 170 171/* Get the length of the data portion of a generic struct sockaddr */ 172static int 173ng_parse_generic_sockdata_getLength(const struct ng_parse_type *type, 174 const u_char *start, const u_char *buf) 175{ 176 const struct sockaddr *sa; 177 178 sa = (const struct sockaddr *)(buf - SADATA_OFFSET); 179 return (sa->sa_len < SADATA_OFFSET) ? 0 : sa->sa_len - SADATA_OFFSET; 180} 181 182/* Type for the variable length data portion of a generic struct sockaddr */ 183static const struct ng_parse_type ng_ksocket_generic_sockdata_type = { 184 &ng_parse_bytearray_type, 185 &ng_parse_generic_sockdata_getLength 186}; 187 188/* Type for a generic struct sockaddr */ 189static const struct ng_parse_struct_field 190 ng_parse_generic_sockaddr_type_fields[] = { 191 { "len", &ng_parse_uint8_type }, 192 { "family", &ng_parse_uint8_type }, 193 { "data", &ng_ksocket_generic_sockdata_type }, 194 { NULL } 195}; 196static const struct ng_parse_type ng_ksocket_generic_sockaddr_type = { 197 &ng_parse_struct_type, 198 &ng_parse_generic_sockaddr_type_fields 199}; 200 201/* Convert a struct sockaddr from ASCII to binary. If its a protocol 202 family that we specially handle, do that, otherwise defer to the 203 generic parse type ng_ksocket_generic_sockaddr_type. */ 204static int 205ng_ksocket_sockaddr_parse(const struct ng_parse_type *type, 206 const char *s, int *off, const u_char *const start, 207 u_char *const buf, int *buflen) 208{ 209 struct sockaddr *const sa = (struct sockaddr *)buf; 210 enum ng_parse_token tok; 211 char fambuf[32]; 212 int family, len; 213 char *t; 214 215 /* If next token is a left curly brace, use generic parse type */ 216 if ((tok = ng_parse_get_token(s, off, &len)) == T_LBRACE) { 217 return (*ng_ksocket_generic_sockaddr_type.supertype->parse) 218 (&ng_ksocket_generic_sockaddr_type, 219 s, off, start, buf, buflen); 220 } 221 222 /* Get socket address family followed by a slash */ 223 while (isspace(s[*off])) 224 (*off)++; 225 if ((t = index(s + *off, '/')) == NULL) 226 return (EINVAL); 227 if ((len = t - (s + *off)) > sizeof(fambuf) - 1) 228 return (EINVAL); 229 strncpy(fambuf, s + *off, len); 230 fambuf[len] = '\0'; 231 *off += len + 1; 232 if ((family = ng_ksocket_parse(ng_ksocket_families, fambuf, 0)) == -1) 233 return (EINVAL); 234 235 /* Set family */ 236 if (*buflen < SADATA_OFFSET) 237 return (ERANGE); 238 sa->sa_family = family; 239 240 /* Set family-specific data and length */ 241 switch (sa->sa_family) { 242 case PF_LOCAL: /* Get pathname */ 243 { 244 const int pathoff = OFFSETOF(struct sockaddr_un, sun_path); 245 struct sockaddr_un *const sun = (struct sockaddr_un *)sa; 246 int toklen, pathlen; 247 char *path; 248 249 if ((path = ng_get_string_token(s, off, &toklen, NULL)) == NULL) 250 return (EINVAL); 251 pathlen = strlen(path); 252 if (pathlen > SOCK_MAXADDRLEN) { 253 free(path, M_NETGRAPH_KSOCKET); 254 return (E2BIG); 255 } 256 if (*buflen < pathoff + pathlen) { 257 free(path, M_NETGRAPH_KSOCKET); 258 return (ERANGE); 259 } 260 *off += toklen; 261 bcopy(path, sun->sun_path, pathlen); 262 sun->sun_len = pathoff + pathlen; 263 free(path, M_NETGRAPH_KSOCKET); 264 break; 265 } 266 267 case PF_INET: /* Get an IP address with optional port */ 268 { 269 struct sockaddr_in *const sin = (struct sockaddr_in *)sa; 270 int i; 271 272 /* Parse this: <ipaddress>[:port] */ 273 for (i = 0; i < 4; i++) { 274 u_long val; 275 char *eptr; 276 277 val = strtoul(s + *off, &eptr, 10); 278 if (val > 0xff || eptr == s + *off) 279 return (EINVAL); 280 *off += (eptr - (s + *off)); 281 ((u_char *)&sin->sin_addr)[i] = (u_char)val; 282 if (i < 3) { 283 if (s[*off] != '.') 284 return (EINVAL); 285 (*off)++; 286 } else if (s[*off] == ':') { 287 (*off)++; 288 val = strtoul(s + *off, &eptr, 10); 289 if (val > 0xffff || eptr == s + *off) 290 return (EINVAL); 291 *off += (eptr - (s + *off)); 292 sin->sin_port = htons(val); 293 } else 294 sin->sin_port = 0; 295 } 296 bzero(&sin->sin_zero, sizeof(sin->sin_zero)); 297 sin->sin_len = sizeof(*sin); 298 break; 299 } 300 301#if 0 302 case PF_APPLETALK: /* XXX implement these someday */ 303 case PF_INET6: 304 case PF_IPX: 305#endif 306 307 default: 308 return (EINVAL); 309 } 310 311 /* Done */ 312 *buflen = sa->sa_len; 313 return (0); 314} 315 316/* Convert a struct sockaddr from binary to ASCII */ 317static int 318ng_ksocket_sockaddr_unparse(const struct ng_parse_type *type, 319 const u_char *data, int *off, char *cbuf, int cbuflen) 320{ 321 const struct sockaddr *sa = (const struct sockaddr *)(data + *off); 322 int slen = 0; 323 324 /* Output socket address, either in special or generic format */ 325 switch (sa->sa_family) { 326 case PF_LOCAL: 327 { 328 const int pathoff = OFFSETOF(struct sockaddr_un, sun_path); 329 const struct sockaddr_un *sun = (const struct sockaddr_un *)sa; 330 const int pathlen = sun->sun_len - pathoff; 331 char pathbuf[SOCK_MAXADDRLEN + 1]; 332 char *pathtoken; 333 334 bcopy(sun->sun_path, pathbuf, pathlen); 335 if ((pathtoken = ng_encode_string(pathbuf, pathlen)) == NULL) 336 return (ENOMEM); 337 slen += snprintf(cbuf, cbuflen, "local/%s", pathtoken); 338 free(pathtoken, M_NETGRAPH_KSOCKET); 339 if (slen >= cbuflen) 340 return (ERANGE); 341 *off += sun->sun_len; 342 return (0); 343 } 344 345 case PF_INET: 346 { 347 const struct sockaddr_in *sin = (const struct sockaddr_in *)sa; 348 349 slen += snprintf(cbuf, cbuflen, "inet/%d.%d.%d.%d", 350 ((const u_char *)&sin->sin_addr)[0], 351 ((const u_char *)&sin->sin_addr)[1], 352 ((const u_char *)&sin->sin_addr)[2], 353 ((const u_char *)&sin->sin_addr)[3]); 354 if (sin->sin_port != 0) { 355 slen += snprintf(cbuf + strlen(cbuf), 356 cbuflen - strlen(cbuf), ":%d", 357 (u_int)ntohs(sin->sin_port)); 358 } 359 if (slen >= cbuflen) 360 return (ERANGE); 361 *off += sizeof(*sin); 362 return(0); 363 } 364 365#if 0 366 case PF_APPLETALK: /* XXX implement these someday */ 367 case PF_INET6: 368 case PF_IPX: 369#endif 370 371 default: 372 return (*ng_ksocket_generic_sockaddr_type.supertype->unparse) 373 (&ng_ksocket_generic_sockaddr_type, 374 data, off, cbuf, cbuflen); 375 } 376} 377 378/* Parse type for struct sockaddr */ 379static const struct ng_parse_type ng_ksocket_sockaddr_type = { 380 NULL, 381 NULL, 382 NULL, 383 &ng_ksocket_sockaddr_parse, 384 &ng_ksocket_sockaddr_unparse, 385 NULL /* no such thing as a default struct sockaddr */ 386}; 387 388/************************************************************************ 389 STRUCT NG_KSOCKET_SOCKOPT PARSE TYPE 390 ************************************************************************/ 391 392/* Get length of the struct ng_ksocket_sockopt value field, which is the 393 just the excess of the message argument portion over the length of 394 the struct ng_ksocket_sockopt. */ 395static int 396ng_parse_sockoptval_getLength(const struct ng_parse_type *type, 397 const u_char *start, const u_char *buf) 398{ 399 static const int offset = OFFSETOF(struct ng_ksocket_sockopt, value); 400 const struct ng_ksocket_sockopt *sopt; 401 const struct ng_mesg *msg; 402 403 sopt = (const struct ng_ksocket_sockopt *)(buf - offset); 404 msg = (const struct ng_mesg *)((const u_char *)sopt - sizeof(*msg)); 405 return msg->header.arglen - sizeof(*sopt); 406} 407 408/* Parse type for the option value part of a struct ng_ksocket_sockopt 409 XXX Eventually, we should handle the different socket options specially. 410 XXX This would avoid byte order problems, eg an integer value of 1 is 411 XXX going to be "[1]" for little endian or "[3=1]" for big endian. */ 412static const struct ng_parse_type ng_ksocket_sockoptval_type = { 413 &ng_parse_bytearray_type, 414 &ng_parse_sockoptval_getLength 415}; 416 417/* Parse type for struct ng_ksocket_sockopt */ 418static const struct ng_parse_struct_field ng_ksocket_sockopt_type_fields[] 419 = NG_KSOCKET_SOCKOPT_INFO(&ng_ksocket_sockoptval_type); 420static const struct ng_parse_type ng_ksocket_sockopt_type = { 421 &ng_parse_struct_type, 422 &ng_ksocket_sockopt_type_fields 423}; 424 425/* Parse type for struct ng_ksocket_accept */ 426static const struct ng_parse_struct_field ng_ksocket_accept_type_fields[] 427 = NGM_KSOCKET_ACCEPT_INFO; 428static const struct ng_parse_type ng_ksocket_accept_type = { 429 &ng_parse_struct_type, 430 &ng_ksocket_accept_type_fields 431}; 432 433/* List of commands and how to convert arguments to/from ASCII */ 434static const struct ng_cmdlist ng_ksocket_cmds[] = { 435 { 436 NGM_KSOCKET_COOKIE, 437 NGM_KSOCKET_BIND, 438 "bind", 439 &ng_ksocket_sockaddr_type, 440 NULL 441 }, 442 { 443 NGM_KSOCKET_COOKIE, 444 NGM_KSOCKET_LISTEN, 445 "listen", 446 &ng_parse_int32_type, 447 NULL 448 }, 449 { 450 NGM_KSOCKET_COOKIE, 451 NGM_KSOCKET_ACCEPT, 452 "accept", 453 NULL, 454 &ng_ksocket_accept_type 455 }, 456 { 457 NGM_KSOCKET_COOKIE, 458 NGM_KSOCKET_CONNECT, 459 "connect", 460 &ng_ksocket_sockaddr_type, 461 &ng_parse_int32_type 462 }, 463 { 464 NGM_KSOCKET_COOKIE, 465 NGM_KSOCKET_GETNAME, 466 "getname", 467 NULL, 468 &ng_ksocket_sockaddr_type 469 }, 470 { 471 NGM_KSOCKET_COOKIE, 472 NGM_KSOCKET_GETPEERNAME, 473 "getpeername", 474 NULL, 475 &ng_ksocket_sockaddr_type 476 }, 477 { 478 NGM_KSOCKET_COOKIE, 479 NGM_KSOCKET_SETOPT, 480 "setopt", 481 &ng_ksocket_sockopt_type, 482 NULL 483 }, 484 { 485 NGM_KSOCKET_COOKIE, 486 NGM_KSOCKET_GETOPT, 487 "getopt", 488 &ng_ksocket_sockopt_type, 489 &ng_ksocket_sockopt_type 490 }, 491 { 0 } 492}; 493 494/* Node type descriptor */ 495static struct ng_type ng_ksocket_typestruct = { 496 .version = NG_ABI_VERSION, 497 .name = NG_KSOCKET_NODE_TYPE, 498 .constructor = ng_ksocket_constructor, 499 .rcvmsg = ng_ksocket_rcvmsg, 500 .shutdown = ng_ksocket_shutdown, 501 .newhook = ng_ksocket_newhook, 502 .connect = ng_ksocket_connect, 503 .rcvdata = ng_ksocket_rcvdata, 504 .disconnect = ng_ksocket_disconnect, 505 .cmdlist = ng_ksocket_cmds, 506}; 507NETGRAPH_INIT(ksocket, &ng_ksocket_typestruct); 508 509#define ERROUT(x) do { error = (x); goto done; } while (0) 510 511/************************************************************************ 512 NETGRAPH NODE STUFF 513 ************************************************************************/ 514 515/* 516 * Node type constructor 517 * The NODE part is assumed to be all set up. 518 * There is already a reference to the node for us. 519 */ 520static int 521ng_ksocket_constructor(node_p node) 522{ 523 priv_p priv; 524 525 /* Allocate private structure */ 526 priv = malloc(sizeof(*priv), 527 M_NETGRAPH_KSOCKET, M_NOWAIT | M_ZERO); 528 if (priv == NULL) 529 return (ENOMEM); 530 531 LIST_INIT(&priv->embryos); 532 /* cross link them */ 533 priv->node = node; 534 NG_NODE_SET_PRIVATE(node, priv); 535 536 /* Done */ 537 return (0); 538} 539 540/* 541 * Give our OK for a hook to be added. The hook name is of the 542 * form "<family>/<type>/<proto>" where the three components may 543 * be decimal numbers or else aliases from the above lists. 544 * 545 * Connecting a hook amounts to opening the socket. Disconnecting 546 * the hook closes the socket and destroys the node as well. 547 */ 548static int 549ng_ksocket_newhook(node_p node, hook_p hook, const char *name0) 550{ 551 struct thread *td = curthread; /* XXX broken */ 552 const priv_p priv = NG_NODE_PRIVATE(node); 553 char *s1, *s2, name[NG_HOOKSIZ]; 554 int family, type, protocol, error; 555 556 /* Check if we're already connected */ 557 if (priv->hook != NULL) 558 return (EISCONN); 559 560 if (priv->flags & KSF_CLONED) { 561 if (priv->flags & KSF_EMBRYONIC) { 562 /* Remove ourselves from our parent's embryo list */ 563 LIST_REMOVE(priv, siblings); 564 priv->flags &= ~KSF_EMBRYONIC; 565 } 566 } else { 567 /* Extract family, type, and protocol from hook name */ 568 snprintf(name, sizeof(name), "%s", name0); 569 s1 = name; 570 if ((s2 = index(s1, '/')) == NULL) 571 return (EINVAL); 572 *s2++ = '\0'; 573 family = ng_ksocket_parse(ng_ksocket_families, s1, 0); 574 if (family == -1) 575 return (EINVAL); 576 s1 = s2; 577 if ((s2 = index(s1, '/')) == NULL) 578 return (EINVAL); 579 *s2++ = '\0'; 580 type = ng_ksocket_parse(ng_ksocket_types, s1, 0); 581 if (type == -1) 582 return (EINVAL); 583 s1 = s2; 584 protocol = ng_ksocket_parse(ng_ksocket_protos, s1, family); 585 if (protocol == -1) 586 return (EINVAL); 587 588 /* Create the socket */ 589 error = socreate(family, &priv->so, type, protocol, 590 td->td_ucred, td); 591 if (error != 0) 592 return (error); 593 594 /* XXX call soreserve() ? */ 595 596 } 597 598 /* OK */ 599 priv->hook = hook; 600 601 /* 602 * In case of misconfigured routing a packet may reenter 603 * ksocket node recursively. Decouple stack to avoid possible 604 * panics about sleeping with locks held. 605 */ 606 NG_HOOK_FORCE_QUEUE(hook); 607 608 return(0); 609} 610 611static int 612ng_ksocket_connect(hook_p hook) 613{ 614 node_p node = NG_HOOK_NODE(hook); 615 const priv_p priv = NG_NODE_PRIVATE(node); 616 struct socket *const so = priv->so; 617 618 /* Add our hook for incoming data and other events */ 619 priv->so->so_upcallarg = (caddr_t)node; 620 priv->so->so_upcall = ng_ksocket_incoming; 621 SOCKBUF_LOCK(&priv->so->so_rcv); 622 priv->so->so_rcv.sb_flags |= SB_UPCALL; 623 SOCKBUF_UNLOCK(&priv->so->so_rcv); 624 SOCKBUF_LOCK(&priv->so->so_snd); 625 priv->so->so_snd.sb_flags |= SB_UPCALL; 626 SOCKBUF_UNLOCK(&priv->so->so_snd); 627 SOCK_LOCK(priv->so); 628 priv->so->so_state |= SS_NBIO; 629 SOCK_UNLOCK(priv->so); 630 /* 631 * --Original comment-- 632 * On a cloned socket we may have already received one or more 633 * upcalls which we couldn't handle without a hook. Handle 634 * those now. 635 * We cannot call the upcall function directly 636 * from here, because until this function has returned our 637 * hook isn't connected. 638 * 639 * ---meta comment for -current --- 640 * XXX This is dubius. 641 * Upcalls between the time that the hook was 642 * first created and now (on another processesor) will 643 * be earlier on the queue than the request to finalise the hook. 644 * By the time the hook is finalised, 645 * The queued upcalls will have happenned and the code 646 * will have discarded them because of a lack of a hook. 647 * (socket not open). 648 * 649 * This is a bad byproduct of the complicated way in which hooks 650 * are now created (3 daisy chained async events). 651 * 652 * Since we are a netgraph operation 653 * We know that we hold a lock on this node. This forces the 654 * request we make below to be queued rather than implemented 655 * immediatly which will cause the upcall function to be called a bit 656 * later. 657 * However, as we will run any waiting queued operations immediatly 658 * after doing this one, if we have not finalised the other end 659 * of the hook, those queued operations will fail. 660 */ 661 if (priv->flags & KSF_CLONED) { 662 ng_send_fn(node, NULL, &ng_ksocket_incoming2, so, M_NOWAIT); 663 } 664 665 return (0); 666} 667 668/* 669 * Receive a control message 670 */ 671static int 672ng_ksocket_rcvmsg(node_p node, item_p item, hook_p lasthook) 673{ 674 struct thread *td = curthread; /* XXX broken */ 675 const priv_p priv = NG_NODE_PRIVATE(node); 676 struct socket *const so = priv->so; 677 struct ng_mesg *resp = NULL; 678 int error = 0; 679 struct ng_mesg *msg; 680 ng_ID_t raddr; 681 682 NGI_GET_MSG(item, msg); 683 switch (msg->header.typecookie) { 684 case NGM_KSOCKET_COOKIE: 685 switch (msg->header.cmd) { 686 case NGM_KSOCKET_BIND: 687 { 688 struct sockaddr *const sa 689 = (struct sockaddr *)msg->data; 690 691 /* Sanity check */ 692 if (msg->header.arglen < SADATA_OFFSET 693 || msg->header.arglen < sa->sa_len) 694 ERROUT(EINVAL); 695 if (so == NULL) 696 ERROUT(ENXIO); 697 698 /* Bind */ 699 error = sobind(so, sa, td); 700 break; 701 } 702 case NGM_KSOCKET_LISTEN: 703 { 704 /* Sanity check */ 705 if (msg->header.arglen != sizeof(int32_t)) 706 ERROUT(EINVAL); 707 if (so == NULL) 708 ERROUT(ENXIO); 709 710 /* Listen */ 711 error = solisten(so, *((int32_t *)msg->data), td); 712 break; 713 } 714 715 case NGM_KSOCKET_ACCEPT: 716 { 717 /* Sanity check */ 718 if (msg->header.arglen != 0) 719 ERROUT(EINVAL); 720 if (so == NULL) 721 ERROUT(ENXIO); 722 723 /* Make sure the socket is capable of accepting */ 724 if (!(so->so_options & SO_ACCEPTCONN)) 725 ERROUT(EINVAL); 726 if (priv->flags & KSF_ACCEPTING) 727 ERROUT(EALREADY); 728 729 error = ng_ksocket_check_accept(priv); 730 if (error != 0 && error != EWOULDBLOCK) 731 ERROUT(error); 732 733 /* 734 * If a connection is already complete, take it. 735 * Otherwise let the upcall function deal with 736 * the connection when it comes in. 737 */ 738 priv->response_token = msg->header.token; 739 raddr = priv->response_addr = NGI_RETADDR(item); 740 if (error == 0) { 741 ng_ksocket_finish_accept(priv); 742 } else 743 priv->flags |= KSF_ACCEPTING; 744 break; 745 } 746 747 case NGM_KSOCKET_CONNECT: 748 { 749 struct sockaddr *const sa 750 = (struct sockaddr *)msg->data; 751 752 /* Sanity check */ 753 if (msg->header.arglen < SADATA_OFFSET 754 || msg->header.arglen < sa->sa_len) 755 ERROUT(EINVAL); 756 if (so == NULL) 757 ERROUT(ENXIO); 758 759 /* Do connect */ 760 if ((so->so_state & SS_ISCONNECTING) != 0) 761 ERROUT(EALREADY); 762 if ((error = soconnect(so, sa, td)) != 0) { 763 so->so_state &= ~SS_ISCONNECTING; 764 ERROUT(error); 765 } 766 if ((so->so_state & SS_ISCONNECTING) != 0) { 767 /* We will notify the sender when we connect */ 768 priv->response_token = msg->header.token; 769 raddr = priv->response_addr = NGI_RETADDR(item); 770 priv->flags |= KSF_CONNECTING; 771 ERROUT(EINPROGRESS); 772 } 773 break; 774 } 775 776 case NGM_KSOCKET_GETNAME: 777 case NGM_KSOCKET_GETPEERNAME: 778 { 779 int (*func)(struct socket *so, struct sockaddr **nam); 780 struct sockaddr *sa = NULL; 781 int len; 782 783 /* Sanity check */ 784 if (msg->header.arglen != 0) 785 ERROUT(EINVAL); 786 if (so == NULL) 787 ERROUT(ENXIO); 788 789 /* Get function */ 790 if (msg->header.cmd == NGM_KSOCKET_GETPEERNAME) { 791 if ((so->so_state 792 & (SS_ISCONNECTED|SS_ISCONFIRMING)) == 0) 793 ERROUT(ENOTCONN); 794 func = so->so_proto->pr_usrreqs->pru_peeraddr; 795 } else 796 func = so->so_proto->pr_usrreqs->pru_sockaddr; 797 798 /* Get local or peer address */ 799 if ((error = (*func)(so, &sa)) != 0) 800 goto bail; 801 len = (sa == NULL) ? 0 : sa->sa_len; 802 803 /* Send it back in a response */ 804 NG_MKRESPONSE(resp, msg, len, M_NOWAIT); 805 if (resp == NULL) { 806 error = ENOMEM; 807 goto bail; 808 } 809 bcopy(sa, resp->data, len); 810 811 bail: 812 /* Cleanup */ 813 if (sa != NULL) 814 free(sa, M_SONAME); 815 break; 816 } 817 818 case NGM_KSOCKET_GETOPT: 819 { 820 struct ng_ksocket_sockopt *ksopt = 821 (struct ng_ksocket_sockopt *)msg->data; 822 struct sockopt sopt; 823 824 /* Sanity check */ 825 if (msg->header.arglen != sizeof(*ksopt)) 826 ERROUT(EINVAL); 827 if (so == NULL) 828 ERROUT(ENXIO); 829 830 /* Get response with room for option value */ 831 NG_MKRESPONSE(resp, msg, sizeof(*ksopt) 832 + NG_KSOCKET_MAX_OPTLEN, M_NOWAIT); 833 if (resp == NULL) 834 ERROUT(ENOMEM); 835 836 /* Get socket option, and put value in the response */ 837 sopt.sopt_dir = SOPT_GET; 838 sopt.sopt_level = ksopt->level; 839 sopt.sopt_name = ksopt->name; 840 sopt.sopt_td = NULL; 841 sopt.sopt_valsize = NG_KSOCKET_MAX_OPTLEN; 842 ksopt = (struct ng_ksocket_sockopt *)resp->data; 843 sopt.sopt_val = ksopt->value; 844 if ((error = sogetopt(so, &sopt)) != 0) { 845 NG_FREE_MSG(resp); 846 break; 847 } 848 849 /* Set actual value length */ 850 resp->header.arglen = sizeof(*ksopt) 851 + sopt.sopt_valsize; 852 break; 853 } 854 855 case NGM_KSOCKET_SETOPT: 856 { 857 struct ng_ksocket_sockopt *const ksopt = 858 (struct ng_ksocket_sockopt *)msg->data; 859 const int valsize = msg->header.arglen - sizeof(*ksopt); 860 struct sockopt sopt; 861 862 /* Sanity check */ 863 if (valsize < 0) 864 ERROUT(EINVAL); 865 if (so == NULL) 866 ERROUT(ENXIO); 867 868 /* Set socket option */ 869 sopt.sopt_dir = SOPT_SET; 870 sopt.sopt_level = ksopt->level; 871 sopt.sopt_name = ksopt->name; 872 sopt.sopt_val = ksopt->value; 873 sopt.sopt_valsize = valsize; 874 sopt.sopt_td = NULL; 875 error = sosetopt(so, &sopt); 876 break; 877 } 878 879 default: 880 error = EINVAL; 881 break; 882 } 883 break; 884 default: 885 error = EINVAL; 886 break; 887 } 888done: 889 NG_RESPOND_MSG(error, node, item, resp); 890 NG_FREE_MSG(msg); 891 return (error); 892} 893 894/* 895 * Receive incoming data on our hook. Send it out the socket. 896 */ 897static int 898ng_ksocket_rcvdata(hook_p hook, item_p item) 899{ 900 struct thread *td = curthread; /* XXX broken */ 901 const node_p node = NG_HOOK_NODE(hook); 902 const priv_p priv = NG_NODE_PRIVATE(node); 903 struct socket *const so = priv->so; 904 struct sockaddr *sa = NULL; 905 int error; 906 struct mbuf *m; 907 struct sa_tag *stag; 908 909 /* Extract data */ 910 NGI_GET_M(item, m); 911 NG_FREE_ITEM(item); 912 913 /* 914 * Look if socket address is stored in packet tags. 915 * If sockaddr is ours, or provided by a third party (zero id), 916 * then we accept it. 917 */ 918 if (((stag = (struct sa_tag *)m_tag_locate(m, NGM_KSOCKET_COOKIE, 919 NG_KSOCKET_TAG_SOCKADDR, NULL)) != NULL) && 920 (stag->id == NG_NODE_ID(node) || stag->id == 0)) 921 sa = &stag->sa; 922 923 /* Reset specific mbuf flags to prevent addressing problems. */ 924 m->m_flags &= ~(M_BCAST|M_MCAST); 925 926 /* Send packet */ 927 error = sosend(so, sa, 0, m, 0, 0, td); 928 929 return (error); 930} 931 932/* 933 * Destroy node 934 */ 935static int 936ng_ksocket_shutdown(node_p node) 937{ 938 const priv_p priv = NG_NODE_PRIVATE(node); 939 priv_p embryo; 940 941 /* Close our socket (if any) */ 942 if (priv->so != NULL) { 943 SOCKBUF_LOCK(&priv->so->so_rcv); 944 priv->so->so_rcv.sb_flags &= ~SB_UPCALL; 945 SOCKBUF_UNLOCK(&priv->so->so_rcv); 946 SOCKBUF_LOCK(&priv->so->so_snd); 947 priv->so->so_snd.sb_flags &= ~SB_UPCALL; 948 SOCKBUF_UNLOCK(&priv->so->so_snd); 949 priv->so->so_upcall = NULL; 950 soclose(priv->so); 951 priv->so = NULL; 952 } 953 954 /* If we are an embryo, take ourselves out of the parent's list */ 955 if (priv->flags & KSF_EMBRYONIC) { 956 LIST_REMOVE(priv, siblings); 957 priv->flags &= ~KSF_EMBRYONIC; 958 } 959 960 /* Remove any embryonic children we have */ 961 while (!LIST_EMPTY(&priv->embryos)) { 962 embryo = LIST_FIRST(&priv->embryos); 963 ng_rmnode_self(embryo->node); 964 } 965 966 /* Take down netgraph node */ 967 bzero(priv, sizeof(*priv)); 968 free(priv, M_NETGRAPH_KSOCKET); 969 NG_NODE_SET_PRIVATE(node, NULL); 970 NG_NODE_UNREF(node); /* let the node escape */ 971 return (0); 972} 973 974/* 975 * Hook disconnection 976 */ 977static int 978ng_ksocket_disconnect(hook_p hook) 979{ 980 KASSERT(NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0, 981 ("%s: numhooks=%d?", __func__, 982 NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)))); 983 if (NG_NODE_IS_VALID(NG_HOOK_NODE(hook))) 984 ng_rmnode_self(NG_HOOK_NODE(hook)); 985 return (0); 986} 987 988/************************************************************************ 989 HELPER STUFF 990 ************************************************************************/ 991/* 992 * You should not "just call" a netgraph node function from an external 993 * asynchronous event. This is because in doing so you are ignoring the 994 * locking on the netgraph nodes. Instead call your function via ng_send_fn(). 995 * This will call the function you chose, but will first do all the 996 * locking rigmarole. Your function MAY only be called at some distant future 997 * time (several millisecs away) so don't give it any arguments 998 * that may be revoked soon (e.g. on your stack). 999 * 1000 * To decouple stack, we use queue version of ng_send_fn(). 1001 */ 1002 1003static void 1004ng_ksocket_incoming(struct socket *so, void *arg, int waitflag) 1005{ 1006 const node_p node = arg; 1007 const priv_p priv = NG_NODE_PRIVATE(node); 1008 int wait = ((waitflag & M_WAITOK) ? NG_WAITOK : 0) | NG_QUEUE; 1009 1010 /* 1011 * Even if node is not locked, as soon as we are called, we assume 1012 * it exist and it's private area is valid. With some care we can 1013 * access it. Mark node that incoming event for it was sent to 1014 * avoid unneded queue trashing. 1015 */ 1016 if (atomic_cmpset_int(&priv->fn_sent, 0, 1) && 1017 ng_send_fn1(node, NULL, &ng_ksocket_incoming2, so, 0, wait)) { 1018 atomic_store_rel_int(&priv->fn_sent, 0); 1019 } 1020} 1021 1022 1023/* 1024 * When incoming data is appended to the socket, we get notified here. 1025 * This is also called whenever a significant event occurs for the socket. 1026 * Our original caller may have queued this even some time ago and 1027 * we cannot trust that he even still exists. The node however is being 1028 * held with a reference by the queueing code and guarantied to be valid. 1029 */ 1030static void 1031ng_ksocket_incoming2(node_p node, hook_p hook, void *arg1, int arg2) 1032{ 1033 struct socket *so = arg1; 1034 const priv_p priv = NG_NODE_PRIVATE(node); 1035 struct mbuf *m; 1036 struct ng_mesg *response; 1037 struct uio auio; 1038 int s, flags, error; 1039 1040 s = splnet(); 1041 1042 /* so = priv->so; *//* XXX could have derived this like so */ 1043 KASSERT(so == priv->so, ("%s: wrong socket", __func__)); 1044 1045 /* Allow next incoming event to be queued. */ 1046 atomic_store_rel_int(&priv->fn_sent, 0); 1047 1048 /* Check whether a pending connect operation has completed */ 1049 if (priv->flags & KSF_CONNECTING) { 1050 if ((error = so->so_error) != 0) { 1051 so->so_error = 0; 1052 so->so_state &= ~SS_ISCONNECTING; 1053 } 1054 if (!(so->so_state & SS_ISCONNECTING)) { 1055 NG_MKMESSAGE(response, NGM_KSOCKET_COOKIE, 1056 NGM_KSOCKET_CONNECT, sizeof(int32_t), M_NOWAIT); 1057 if (response != NULL) { 1058 response->header.flags |= NGF_RESP; 1059 response->header.token = priv->response_token; 1060 *(int32_t *)response->data = error; 1061 /* 1062 * send an async "response" message 1063 * to the node that set us up 1064 * (if it still exists) 1065 */ 1066 NG_SEND_MSG_ID(error, node, 1067 response, priv->response_addr, 0); 1068 } 1069 priv->flags &= ~KSF_CONNECTING; 1070 } 1071 } 1072 1073 /* Check whether a pending accept operation has completed */ 1074 if (priv->flags & KSF_ACCEPTING) { 1075 error = ng_ksocket_check_accept(priv); 1076 if (error != EWOULDBLOCK) 1077 priv->flags &= ~KSF_ACCEPTING; 1078 if (error == 0) 1079 ng_ksocket_finish_accept(priv); 1080 } 1081 1082 /* 1083 * If we don't have a hook, we must handle data events later. When 1084 * the hook gets created and is connected, this upcall function 1085 * will be called again. 1086 */ 1087 if (priv->hook == NULL) { 1088 splx(s); 1089 return; 1090 } 1091 1092 /* Read and forward available mbuf's */ 1093 auio.uio_td = NULL; 1094 auio.uio_resid = 1000000000; 1095 flags = MSG_DONTWAIT; 1096 while (1) { 1097 struct sockaddr *sa = NULL; 1098 struct mbuf *n; 1099 1100 /* Try to get next packet from socket */ 1101 if ((error = soreceive(so, (so->so_state & SS_ISCONNECTED) ? 1102 NULL : &sa, &auio, &m, (struct mbuf **)0, &flags)) != 0) 1103 break; 1104 1105 /* See if we got anything */ 1106 if (m == NULL) { 1107 if (sa != NULL) 1108 free(sa, M_SONAME); 1109 break; 1110 } 1111 1112 /* 1113 * Don't trust the various socket layers to get the 1114 * packet header and length correct (e.g. kern/15175). 1115 * 1116 * Also, do not trust that soreceive() will clear m_nextpkt 1117 * for us (e.g. kern/84952, kern/82413). 1118 */ 1119 m->m_pkthdr.csum_flags = 0; 1120 for (n = m, m->m_pkthdr.len = 0; n != NULL; n = n->m_next) { 1121 m->m_pkthdr.len += n->m_len; 1122 n->m_nextpkt = NULL; 1123 } 1124 1125 /* Put peer's socket address (if any) into a tag */ 1126 if (sa != NULL) { 1127 struct sa_tag *stag; 1128 1129 stag = (struct sa_tag *)m_tag_alloc(NGM_KSOCKET_COOKIE, 1130 NG_KSOCKET_TAG_SOCKADDR, sizeof(ng_ID_t) + 1131 sa->sa_len, M_NOWAIT); 1132 if (stag == NULL) { 1133 free(sa, M_SONAME); 1134 goto sendit; 1135 } 1136 bcopy(sa, &stag->sa, sa->sa_len); 1137 free(sa, M_SONAME); 1138 stag->id = NG_NODE_ID(node); 1139 m_tag_prepend(m, &stag->tag); 1140 } 1141 1142sendit: /* Forward data with optional peer sockaddr as packet tag */ 1143 NG_SEND_DATA_ONLY(error, priv->hook, m); 1144 } 1145 1146 /* 1147 * If the peer has closed the connection, forward a 0-length mbuf 1148 * to indicate end-of-file. 1149 */ 1150 if (so->so_rcv.sb_state & SBS_CANTRCVMORE && !(priv->flags & KSF_EOFSEEN)) { 1151 MGETHDR(m, M_NOWAIT, MT_DATA); 1152 if (m != NULL) { 1153 m->m_len = m->m_pkthdr.len = 0; 1154 NG_SEND_DATA_ONLY(error, priv->hook, m); 1155 } 1156 priv->flags |= KSF_EOFSEEN; 1157 } 1158 splx(s); 1159} 1160 1161/* 1162 * Check for a completed incoming connection and return 0 if one is found. 1163 * Otherwise return the appropriate error code. 1164 */ 1165static int 1166ng_ksocket_check_accept(priv_p priv) 1167{ 1168 struct socket *const head = priv->so; 1169 int error; 1170 1171 if ((error = head->so_error) != 0) { 1172 head->so_error = 0; 1173 return error; 1174 } 1175 /* Unlocked read. */ 1176 if (TAILQ_EMPTY(&head->so_comp)) { 1177 if (head->so_rcv.sb_state & SBS_CANTRCVMORE) 1178 return ECONNABORTED; 1179 return EWOULDBLOCK; 1180 } 1181 return 0; 1182} 1183 1184/* 1185 * Handle the first completed incoming connection, assumed to be already 1186 * on the socket's so_comp queue. 1187 */ 1188static void 1189ng_ksocket_finish_accept(priv_p priv) 1190{ 1191 struct socket *const head = priv->so; 1192 struct socket *so; 1193 struct sockaddr *sa = NULL; 1194 struct ng_mesg *resp; 1195 struct ng_ksocket_accept *resp_data; 1196 node_p node; 1197 priv_p priv2; 1198 int len; 1199 int error; 1200 1201 ACCEPT_LOCK(); 1202 so = TAILQ_FIRST(&head->so_comp); 1203 if (so == NULL) { /* Should never happen */ 1204 ACCEPT_UNLOCK(); 1205 return; 1206 } 1207 TAILQ_REMOVE(&head->so_comp, so, so_list); 1208 head->so_qlen--; 1209 so->so_qstate &= ~SQ_COMP; 1210 so->so_head = NULL; 1211 SOCK_LOCK(so); 1212 soref(so); 1213 so->so_state |= SS_NBIO; 1214 SOCK_UNLOCK(so); 1215 ACCEPT_UNLOCK(); 1216 1217 /* XXX KNOTE(&head->so_rcv.sb_sel.si_note, 0); */ 1218 1219 soaccept(so, &sa); 1220 1221 len = OFFSETOF(struct ng_ksocket_accept, addr); 1222 if (sa != NULL) 1223 len += sa->sa_len; 1224 1225 NG_MKMESSAGE(resp, NGM_KSOCKET_COOKIE, NGM_KSOCKET_ACCEPT, len, 1226 M_NOWAIT); 1227 if (resp == NULL) { 1228 soclose(so); 1229 goto out; 1230 } 1231 resp->header.flags |= NGF_RESP; 1232 resp->header.token = priv->response_token; 1233 1234 /* Clone a ksocket node to wrap the new socket */ 1235 error = ng_make_node_common(&ng_ksocket_typestruct, &node); 1236 if (error) { 1237 free(resp, M_NETGRAPH); 1238 soclose(so); 1239 goto out; 1240 } 1241 1242 if (ng_ksocket_constructor(node) != 0) { 1243 NG_NODE_UNREF(node); 1244 free(resp, M_NETGRAPH); 1245 soclose(so); 1246 goto out; 1247 } 1248 1249 priv2 = NG_NODE_PRIVATE(node); 1250 priv2->so = so; 1251 priv2->flags |= KSF_CLONED | KSF_EMBRYONIC; 1252 1253 /* 1254 * Insert the cloned node into a list of embryonic children 1255 * on the parent node. When a hook is created on the cloned 1256 * node it will be removed from this list. When the parent 1257 * is destroyed it will destroy any embryonic children it has. 1258 */ 1259 LIST_INSERT_HEAD(&priv->embryos, priv2, siblings); 1260 1261 so->so_upcallarg = (caddr_t)node; 1262 so->so_upcall = ng_ksocket_incoming; 1263 SOCKBUF_LOCK(&so->so_rcv); 1264 so->so_rcv.sb_flags |= SB_UPCALL; 1265 SOCKBUF_UNLOCK(&so->so_rcv); 1266 SOCKBUF_LOCK(&so->so_snd); 1267 so->so_snd.sb_flags |= SB_UPCALL; 1268 SOCKBUF_UNLOCK(&so->so_snd); 1269 1270 /* Fill in the response data and send it or return it to the caller */ 1271 resp_data = (struct ng_ksocket_accept *)resp->data; 1272 resp_data->nodeid = NG_NODE_ID(node); 1273 if (sa != NULL) 1274 bcopy(sa, &resp_data->addr, sa->sa_len); 1275 NG_SEND_MSG_ID(error, node, resp, priv->response_addr, 0); 1276 1277out: 1278 if (sa != NULL) 1279 free(sa, M_SONAME); 1280} 1281 1282/* 1283 * Parse out either an integer value or an alias. 1284 */ 1285static int 1286ng_ksocket_parse(const struct ng_ksocket_alias *aliases, 1287 const char *s, int family) 1288{ 1289 int k, val; 1290 char *eptr; 1291 1292 /* Try aliases */ 1293 for (k = 0; aliases[k].name != NULL; k++) { 1294 if (strcmp(s, aliases[k].name) == 0 1295 && aliases[k].family == family) 1296 return aliases[k].value; 1297 } 1298 1299 /* Try parsing as a number */ 1300 val = (int)strtoul(s, &eptr, 10); 1301 if (val < 0 || *eptr != '\0') 1302 return (-1); 1303 return (val); 1304} 1305 1306