14cf49a43SJulian Elischer 24cf49a43SJulian Elischer /* 34cf49a43SJulian Elischer * sock.c 44cf49a43SJulian Elischer * 54cf49a43SJulian Elischer * Copyright (c) 1996-1999 Whistle Communications, Inc. 64cf49a43SJulian Elischer * All rights reserved. 74cf49a43SJulian Elischer * 84cf49a43SJulian Elischer * Subject to the following obligations and disclaimer of warranty, use and 94cf49a43SJulian Elischer * redistribution of this software, in source or object code forms, with or 104cf49a43SJulian Elischer * without modifications are expressly permitted by Whistle Communications; 114cf49a43SJulian Elischer * provided, however, that: 124cf49a43SJulian Elischer * 1. Any and all reproductions of the source or object code must include the 134cf49a43SJulian Elischer * copyright notice above and the following disclaimer of warranties; and 144cf49a43SJulian Elischer * 2. No rights are granted, in any manner or form, to use Whistle 154cf49a43SJulian Elischer * Communications, Inc. trademarks, including the mark "WHISTLE 164cf49a43SJulian Elischer * COMMUNICATIONS" on advertising, endorsements, or otherwise except as 174cf49a43SJulian Elischer * such appears in the above copyright notice or in the software. 184cf49a43SJulian Elischer * 194cf49a43SJulian Elischer * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND 204cf49a43SJulian Elischer * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO 214cf49a43SJulian Elischer * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE, 224cf49a43SJulian Elischer * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF 234cf49a43SJulian Elischer * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. 244cf49a43SJulian Elischer * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY 254cf49a43SJulian Elischer * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS 264cf49a43SJulian Elischer * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE. 274cf49a43SJulian Elischer * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES 284cf49a43SJulian Elischer * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING 294cf49a43SJulian Elischer * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, 304cf49a43SJulian Elischer * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR 314cf49a43SJulian Elischer * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY 324cf49a43SJulian Elischer * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 334cf49a43SJulian Elischer * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 344cf49a43SJulian Elischer * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY 354cf49a43SJulian Elischer * OF SUCH DAMAGE. 364cf49a43SJulian Elischer * 374cf49a43SJulian Elischer * Author: Archie Cobbs <archie@whistle.com> 384cf49a43SJulian Elischer * 394cf49a43SJulian Elischer * $FreeBSD$ 404cf49a43SJulian Elischer * $Whistle: sock.c,v 1.12 1999/01/20 00:57:23 archie Exp $ 414cf49a43SJulian Elischer */ 424cf49a43SJulian Elischer 434cf49a43SJulian Elischer #include <sys/types.h> 444cf49a43SJulian Elischer #include <stdarg.h> 454cf49a43SJulian Elischer #include <netgraph/ng_message.h> 464cf49a43SJulian Elischer #include <netgraph/ng_socket.h> 474cf49a43SJulian Elischer 484cf49a43SJulian Elischer #include "netgraph.h" 494cf49a43SJulian Elischer #include "internal.h" 504cf49a43SJulian Elischer 51fbffcf66SArchie Cobbs /* The socket node type KLD */ 52fbffcf66SArchie Cobbs #define NG_SOCKET_KLD "ng_socket.ko" 53fbffcf66SArchie Cobbs 544cf49a43SJulian Elischer /* 554cf49a43SJulian Elischer * Create a socket type node and give it the supplied name. 564cf49a43SJulian Elischer * Return data and control sockets corresponding to the node. 574cf49a43SJulian Elischer * Returns -1 if error and sets errno. 584cf49a43SJulian Elischer */ 594cf49a43SJulian Elischer int 604cf49a43SJulian Elischer NgMkSockNode(const char *name, int *csp, int *dsp) 614cf49a43SJulian Elischer { 624cf49a43SJulian Elischer char namebuf[NG_NODELEN + 1]; 634cf49a43SJulian Elischer int cs = -1; /* control socket */ 644cf49a43SJulian Elischer int ds = -1; /* data socket */ 654cf49a43SJulian Elischer int errnosv; 664cf49a43SJulian Elischer 674cf49a43SJulian Elischer /* Empty name means no name */ 684cf49a43SJulian Elischer if (name && *name == 0) 694cf49a43SJulian Elischer name = NULL; 704cf49a43SJulian Elischer 71fbffcf66SArchie Cobbs /* Create control socket; this also creates the netgraph node. 72fbffcf66SArchie Cobbs If we get a EPROTONOSUPPORT then the socket node type is 73fbffcf66SArchie Cobbs not loaded, so load it and try again. */ 744cf49a43SJulian Elischer if ((cs = socket(AF_NETGRAPH, SOCK_DGRAM, NG_CONTROL)) < 0) { 75fbffcf66SArchie Cobbs if (errno == EPROTONOSUPPORT) { 76fbffcf66SArchie Cobbs if (kldload(NG_SOCKET_KLD) < 0) { 77fbffcf66SArchie Cobbs errnosv = errno; 78fbffcf66SArchie Cobbs if (_gNgDebugLevel >= 1) 79fbffcf66SArchie Cobbs NGLOG("can't load %s", NG_SOCKET_KLD); 80fbffcf66SArchie Cobbs goto errout; 81fbffcf66SArchie Cobbs } 82fbffcf66SArchie Cobbs cs = socket(AF_NETGRAPH, SOCK_DGRAM, NG_CONTROL); 83fbffcf66SArchie Cobbs if (cs >= 0) 84fbffcf66SArchie Cobbs goto gotNode; 85fbffcf66SArchie Cobbs } 864cf49a43SJulian Elischer errnosv = errno; 874cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 884cf49a43SJulian Elischer NGLOG("socket"); 894cf49a43SJulian Elischer goto errout; 904cf49a43SJulian Elischer } 914cf49a43SJulian Elischer 92fbffcf66SArchie Cobbs gotNode: 934cf49a43SJulian Elischer /* Assign the node the desired name, if any */ 944cf49a43SJulian Elischer if (name != NULL) { 954cf49a43SJulian Elischer u_char sbuf[NG_NODELEN + 3]; 964cf49a43SJulian Elischer struct sockaddr_ng *const sg = (struct sockaddr_ng *) sbuf; 974cf49a43SJulian Elischer 984cf49a43SJulian Elischer /* Assign name */ 994cf49a43SJulian Elischer snprintf(sg->sg_data, NG_NODELEN + 1, "%s", name); 1004cf49a43SJulian Elischer sg->sg_family = AF_NETGRAPH; 1014cf49a43SJulian Elischer sg->sg_len = strlen(sg->sg_data) + 3; 1024cf49a43SJulian Elischer if (bind(cs, (struct sockaddr *) sg, sg->sg_len) < 0) { 1034cf49a43SJulian Elischer errnosv = errno; 1044cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 1054cf49a43SJulian Elischer NGLOG("bind(%s)", sg->sg_data); 1064cf49a43SJulian Elischer goto errout; 1074cf49a43SJulian Elischer } 1084cf49a43SJulian Elischer 1094cf49a43SJulian Elischer /* Save node name */ 1104cf49a43SJulian Elischer snprintf(namebuf, sizeof(namebuf), "%s", name); 1114cf49a43SJulian Elischer } else if (dsp != NULL) { 1124cf49a43SJulian Elischer u_char rbuf[sizeof(struct ng_mesg) + sizeof(struct nodeinfo)]; 1134cf49a43SJulian Elischer struct ng_mesg *const resp = (struct ng_mesg *) rbuf; 1144cf49a43SJulian Elischer struct nodeinfo *const ni = (struct nodeinfo *) resp->data; 1154cf49a43SJulian Elischer 1164cf49a43SJulian Elischer /* Find out the node ID */ 1174cf49a43SJulian Elischer if (NgSendMsg(cs, ".", NGM_GENERIC_COOKIE, 1184cf49a43SJulian Elischer NGM_NODEINFO, NULL, 0) < 0) { 1194cf49a43SJulian Elischer errnosv = errno; 1204cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 1214cf49a43SJulian Elischer NGLOG("send nodeinfo"); 1224cf49a43SJulian Elischer goto errout; 1234cf49a43SJulian Elischer } 1244cf49a43SJulian Elischer if (NgRecvMsg(cs, resp, sizeof(rbuf), NULL) < 0) { 1254cf49a43SJulian Elischer errnosv = errno; 1264cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 1274cf49a43SJulian Elischer NGLOG("recv nodeinfo"); 1284cf49a43SJulian Elischer goto errout; 1294cf49a43SJulian Elischer } 1304cf49a43SJulian Elischer 1314cf49a43SJulian Elischer /* Save node "name" */ 1324cf49a43SJulian Elischer snprintf(namebuf, sizeof(namebuf), "[%lx]", (u_long) ni->id); 1334cf49a43SJulian Elischer } 1344cf49a43SJulian Elischer 1354cf49a43SJulian Elischer /* Create data socket if desired */ 1364cf49a43SJulian Elischer if (dsp != NULL) { 1374cf49a43SJulian Elischer u_char sbuf[NG_NODELEN + 4]; 1384cf49a43SJulian Elischer struct sockaddr_ng *const sg = (struct sockaddr_ng *) sbuf; 1394cf49a43SJulian Elischer 1404cf49a43SJulian Elischer /* Create data socket, initially just "floating" */ 1414cf49a43SJulian Elischer if ((ds = socket(AF_NETGRAPH, SOCK_DGRAM, NG_DATA)) < 0) { 1424cf49a43SJulian Elischer errnosv = errno; 1434cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 1444cf49a43SJulian Elischer NGLOG("socket"); 1454cf49a43SJulian Elischer goto errout; 1464cf49a43SJulian Elischer } 1474cf49a43SJulian Elischer 1484cf49a43SJulian Elischer /* Associate the data socket with the node */ 1494cf49a43SJulian Elischer snprintf(sg->sg_data, NG_NODELEN + 2, "%s:", namebuf); 1504cf49a43SJulian Elischer sg->sg_family = AF_NETGRAPH; 1514cf49a43SJulian Elischer sg->sg_len = strlen(sg->sg_data) + 3; 1524cf49a43SJulian Elischer if (connect(ds, (struct sockaddr *) sg, sg->sg_len) < 0) { 1534cf49a43SJulian Elischer errnosv = errno; 1544cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 1554cf49a43SJulian Elischer NGLOG("connect(%s)", sg->sg_data); 1564cf49a43SJulian Elischer goto errout; 1574cf49a43SJulian Elischer } 1584cf49a43SJulian Elischer } 1594cf49a43SJulian Elischer 1604cf49a43SJulian Elischer /* Return the socket(s) */ 1614cf49a43SJulian Elischer if (csp) 1624cf49a43SJulian Elischer *csp = cs; 1634cf49a43SJulian Elischer else 1644cf49a43SJulian Elischer close(cs); 1654cf49a43SJulian Elischer if (dsp) 1664cf49a43SJulian Elischer *dsp = ds; 1674cf49a43SJulian Elischer return (0); 1684cf49a43SJulian Elischer 1694cf49a43SJulian Elischer errout: 1704cf49a43SJulian Elischer /* Failed */ 1714cf49a43SJulian Elischer if (cs >= 0) 1724cf49a43SJulian Elischer close(cs); 1734cf49a43SJulian Elischer if (ds >= 0) 1744cf49a43SJulian Elischer close(ds); 1754cf49a43SJulian Elischer errno = errnosv; 1764cf49a43SJulian Elischer return (-1); 1774cf49a43SJulian Elischer } 1784cf49a43SJulian Elischer 1794cf49a43SJulian Elischer /* 1804cf49a43SJulian Elischer * Assign a globally unique name to a node 1814cf49a43SJulian Elischer * Returns -1 if error and sets errno. 1824cf49a43SJulian Elischer */ 1834cf49a43SJulian Elischer int 1844cf49a43SJulian Elischer NgNameNode(int cs, const char *path, const char *fmt, ...) 1854cf49a43SJulian Elischer { 1864cf49a43SJulian Elischer struct ngm_name ngn; 1874cf49a43SJulian Elischer va_list args; 1884cf49a43SJulian Elischer 1894cf49a43SJulian Elischer /* Build message arg */ 1904cf49a43SJulian Elischer va_start(args, fmt); 1914cf49a43SJulian Elischer vsnprintf(ngn.name, sizeof(ngn.name), fmt, args); 1924cf49a43SJulian Elischer va_end(args); 1934cf49a43SJulian Elischer 1944cf49a43SJulian Elischer /* Send message */ 1954cf49a43SJulian Elischer if (NgSendMsg(cs, path, 1964cf49a43SJulian Elischer NGM_GENERIC_COOKIE, NGM_NAME, &ngn, sizeof(ngn)) < 0) { 1974cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 1984cf49a43SJulian Elischer NGLOGX("%s: failed", __FUNCTION__); 1994cf49a43SJulian Elischer return (-1); 2004cf49a43SJulian Elischer } 2014cf49a43SJulian Elischer 2024cf49a43SJulian Elischer /* Done */ 2034cf49a43SJulian Elischer return (0); 2044cf49a43SJulian Elischer } 2054cf49a43SJulian Elischer 2064cf49a43SJulian Elischer /* 2074cf49a43SJulian Elischer * Read a packet from a data socket 2084cf49a43SJulian Elischer * Returns -1 if error and sets errno. 2094cf49a43SJulian Elischer */ 2104cf49a43SJulian Elischer int 2114cf49a43SJulian Elischer NgRecvData(int ds, u_char * buf, size_t len, char *hook) 2124cf49a43SJulian Elischer { 2134cf49a43SJulian Elischer u_char frombuf[NG_HOOKLEN + sizeof(struct sockaddr_ng)]; 2144cf49a43SJulian Elischer struct sockaddr_ng *const from = (struct sockaddr_ng *) frombuf; 2154cf49a43SJulian Elischer int fromlen = sizeof(frombuf); 2164cf49a43SJulian Elischer int rtn, errnosv; 2174cf49a43SJulian Elischer 2184cf49a43SJulian Elischer /* Read packet */ 2194cf49a43SJulian Elischer rtn = recvfrom(ds, buf, len, 0, (struct sockaddr *) from, &fromlen); 2204cf49a43SJulian Elischer if (rtn < 0) { 2214cf49a43SJulian Elischer errnosv = errno; 2224cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 2234cf49a43SJulian Elischer NGLOG("recvfrom"); 2244cf49a43SJulian Elischer errno = errnosv; 2254cf49a43SJulian Elischer return (-1); 2264cf49a43SJulian Elischer } 2274cf49a43SJulian Elischer 2284cf49a43SJulian Elischer /* Copy hook name */ 2294cf49a43SJulian Elischer if (hook != NULL) 2304cf49a43SJulian Elischer snprintf(hook, NG_HOOKLEN + 1, "%s", from->sg_data); 2314cf49a43SJulian Elischer 2324cf49a43SJulian Elischer /* Debugging */ 2334cf49a43SJulian Elischer if (_gNgDebugLevel >= 2) { 2344cf49a43SJulian Elischer NGLOGX("READ %s from hook \"%s\" (%d bytes)", 2354cf49a43SJulian Elischer rtn ? "PACKET" : "EOF", from->sg_data, rtn); 2364cf49a43SJulian Elischer if (_gNgDebugLevel >= 3) 2374cf49a43SJulian Elischer _NgDebugBytes(buf, rtn); 2384cf49a43SJulian Elischer } 2394cf49a43SJulian Elischer 2404cf49a43SJulian Elischer /* Done */ 2414cf49a43SJulian Elischer return (rtn); 2424cf49a43SJulian Elischer } 2434cf49a43SJulian Elischer 2444cf49a43SJulian Elischer /* 2454cf49a43SJulian Elischer * Write a packet to a data socket. The packet will be sent 2464cf49a43SJulian Elischer * out the corresponding node on the specified hook. 2474cf49a43SJulian Elischer * Returns -1 if error and sets errno. 2484cf49a43SJulian Elischer */ 2494cf49a43SJulian Elischer int 2504cf49a43SJulian Elischer NgSendData(int ds, const char *hook, const u_char * buf, size_t len) 2514cf49a43SJulian Elischer { 2524cf49a43SJulian Elischer u_char sgbuf[NG_HOOKLEN + sizeof(struct sockaddr_ng)]; 2534cf49a43SJulian Elischer struct sockaddr_ng *const sg = (struct sockaddr_ng *) sgbuf; 2544cf49a43SJulian Elischer int errnosv; 2554cf49a43SJulian Elischer 2564cf49a43SJulian Elischer /* Set up destination hook */ 2574cf49a43SJulian Elischer sg->sg_family = AF_NETGRAPH; 2584cf49a43SJulian Elischer snprintf(sg->sg_data, NG_HOOKLEN + 1, "%s", hook); 2594cf49a43SJulian Elischer sg->sg_len = strlen(sg->sg_data) + 3; 2604cf49a43SJulian Elischer 2614cf49a43SJulian Elischer /* Debugging */ 2624cf49a43SJulian Elischer if (_gNgDebugLevel >= 2) { 2634cf49a43SJulian Elischer NGLOGX("WRITE PACKET to hook \"%s\" (%d bytes)", hook, len); 2644cf49a43SJulian Elischer _NgDebugSockaddr(sg); 2654cf49a43SJulian Elischer if (_gNgDebugLevel >= 3) 2664cf49a43SJulian Elischer _NgDebugBytes(buf, len); 2674cf49a43SJulian Elischer } 2684cf49a43SJulian Elischer 2694cf49a43SJulian Elischer /* Send packet */ 2704cf49a43SJulian Elischer if (sendto(ds, buf, len, 0, (struct sockaddr *) sg, sg->sg_len) < 0) { 2714cf49a43SJulian Elischer errnosv = errno; 2724cf49a43SJulian Elischer if (_gNgDebugLevel >= 1) 2734cf49a43SJulian Elischer NGLOG("sendto(%s)", sg->sg_data); 2744cf49a43SJulian Elischer errno = errnosv; 2754cf49a43SJulian Elischer return (-1); 2764cf49a43SJulian Elischer } 2774cf49a43SJulian Elischer 2784cf49a43SJulian Elischer /* Done */ 2794cf49a43SJulian Elischer return (0); 2804cf49a43SJulian Elischer } 2814cf49a43SJulian Elischer 282