| 1 | |
| 2 | /* |
| 3 | * sock.c |
| 4 | * |
| 5 | * Copyright (c) 1996-1999 Whistle Communications, Inc. |
| 6 | * All rights reserved. |
| 7 | * |
| 8 | * Subject to the following obligations and disclaimer of warranty, use and |
| 9 | * redistribution of this software, in source or object code forms, with or |
| 10 | * without modifications are expressly permitted by Whistle Communications; |
| 11 | * provided, however, that: |
| 12 | * 1. Any and all reproductions of the source or object code must include the |
| 13 | * copyright notice above and the following disclaimer of warranties; and |
| 14 | * 2. No rights are granted, in any manner or form, to use Whistle |
| 15 | * Communications, Inc. trademarks, including the mark "WHISTLE |
| 16 | * COMMUNICATIONS" on advertising, endorsements, or otherwise except as |
| 17 | * such appears in the above copyright notice or in the software. |
| 18 | * |
| 19 | * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND |
| 20 | * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO |
| 21 | * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE, |
| 22 | * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF |
| 23 | * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. |
| 24 | * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY |
| 25 | * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS |
| 26 | * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE. |
| 27 | * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES |
| 28 | * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING |
| 29 | * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, |
| 30 | * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR |
| 31 | * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY |
| 32 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 33 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| 34 | * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY |
| 35 | * OF SUCH DAMAGE. |
| 36 | * |
| 37 | * Author: Archie Cobbs <archie@whistle.com> |
| 38 | * |
| 39 | * $FreeBSD: src/lib/libnetgraph/sock.c,v 1.2 2000/01/28 00:48:27 archie Exp $ |
| 40 | * $DragonFly: src/lib/libnetgraph/sock.c,v 1.5 2007/06/03 23:41:25 swildner Exp $ |
| 41 | * $Whistle: sock.c,v 1.12 1999/01/20 00:57:23 archie Exp $ |
| 42 | */ |
| 43 | |
| 44 | #include <sys/types.h> |
| 45 | #include <stdarg.h> |
| 46 | #include <netgraph/ng_message.h> |
| 47 | #include <netgraph/socket/ng_socket.h> |
| 48 | |
| 49 | #include "netgraph.h" |
| 50 | #include "internal.h" |
| 51 | |
| 52 | /* The socket node type KLD */ |
| 53 | #define NG_SOCKET_KLD "ng_socket.ko" |
| 54 | |
| 55 | /* |
| 56 | * Create a socket type node and give it the supplied name. |
| 57 | * Return data and control sockets corresponding to the node. |
| 58 | * Returns -1 if error and sets errno. |
| 59 | */ |
| 60 | int |
| 61 | NgMkSockNode(const char *name, int *csp, int *dsp) |
| 62 | { |
| 63 | char namebuf[NG_NODESIZ]; |
| 64 | int cs = -1; /* control socket */ |
| 65 | int ds = -1; /* data socket */ |
| 66 | int errnosv; |
| 67 | |
| 68 | /* Empty name means no name */ |
| 69 | if (name && *name == 0) |
| 70 | name = NULL; |
| 71 | |
| 72 | /* Create control socket; this also creates the netgraph node. |
| 73 | If we get a EPROTONOSUPPORT then the socket node type is |
| 74 | not loaded, so load it and try again. */ |
| 75 | if ((cs = socket(AF_NETGRAPH, SOCK_DGRAM, NG_CONTROL)) < 0) { |
| 76 | if (errno == EPROTONOSUPPORT) { |
| 77 | if (kldload(NG_SOCKET_KLD) < 0) { |
| 78 | errnosv = errno; |
| 79 | if (_gNgDebugLevel >= 1) |
| 80 | NGLOG("can't load %s", NG_SOCKET_KLD); |
| 81 | goto errout; |
| 82 | } |
| 83 | cs = socket(AF_NETGRAPH, SOCK_DGRAM, NG_CONTROL); |
| 84 | if (cs >= 0) |
| 85 | goto gotNode; |
| 86 | } |
| 87 | errnosv = errno; |
| 88 | if (_gNgDebugLevel >= 1) |
| 89 | NGLOG("socket"); |
| 90 | goto errout; |
| 91 | } |
| 92 | |
| 93 | gotNode: |
| 94 | /* Assign the node the desired name, if any */ |
| 95 | if (name != NULL) { |
| 96 | u_char sbuf[NG_NODESIZ + 2]; |
| 97 | struct sockaddr_ng *const sg = (struct sockaddr_ng *) sbuf; |
| 98 | |
| 99 | /* Assign name */ |
| 100 | snprintf(sg->sg_data, NG_NODESIZ, "%s", name); |
| 101 | sg->sg_family = AF_NETGRAPH; |
| 102 | sg->sg_len = strlen(sg->sg_data) + 3; |
| 103 | if (bind(cs, (struct sockaddr *) sg, sg->sg_len) < 0) { |
| 104 | errnosv = errno; |
| 105 | if (_gNgDebugLevel >= 1) |
| 106 | NGLOG("bind(%s)", sg->sg_data); |
| 107 | goto errout; |
| 108 | } |
| 109 | |
| 110 | /* Save node name */ |
| 111 | snprintf(namebuf, sizeof(namebuf), "%s", name); |
| 112 | } else if (dsp != NULL) { |
| 113 | u_char rbuf[sizeof(struct ng_mesg) + sizeof(struct nodeinfo)]; |
| 114 | struct ng_mesg *const resp = (struct ng_mesg *) rbuf; |
| 115 | struct nodeinfo *const ni = (struct nodeinfo *) resp->data; |
| 116 | |
| 117 | /* Find out the node ID */ |
| 118 | if (NgSendMsg(cs, ".", NGM_GENERIC_COOKIE, |
| 119 | NGM_NODEINFO, NULL, 0) < 0) { |
| 120 | errnosv = errno; |
| 121 | if (_gNgDebugLevel >= 1) |
| 122 | NGLOG("send nodeinfo"); |
| 123 | goto errout; |
| 124 | } |
| 125 | if (NgRecvMsg(cs, resp, sizeof(rbuf), NULL) < 0) { |
| 126 | errnosv = errno; |
| 127 | if (_gNgDebugLevel >= 1) |
| 128 | NGLOG("recv nodeinfo"); |
| 129 | goto errout; |
| 130 | } |
| 131 | |
| 132 | /* Save node "name" */ |
| 133 | snprintf(namebuf, sizeof(namebuf), "[%lx]", (u_long) ni->id); |
| 134 | } |
| 135 | |
| 136 | /* Create data socket if desired */ |
| 137 | if (dsp != NULL) { |
| 138 | u_char sbuf[NG_NODESIZ + 3]; |
| 139 | struct sockaddr_ng *const sg = (struct sockaddr_ng *) sbuf; |
| 140 | |
| 141 | /* Create data socket, initially just "floating" */ |
| 142 | if ((ds = socket(AF_NETGRAPH, SOCK_DGRAM, NG_DATA)) < 0) { |
| 143 | errnosv = errno; |
| 144 | if (_gNgDebugLevel >= 1) |
| 145 | NGLOG("socket"); |
| 146 | goto errout; |
| 147 | } |
| 148 | |
| 149 | /* Associate the data socket with the node */ |
| 150 | snprintf(sg->sg_data, NG_NODESIZ + 1, "%s:", namebuf); |
| 151 | sg->sg_family = AF_NETGRAPH; |
| 152 | sg->sg_len = strlen(sg->sg_data) + 3; |
| 153 | if (connect(ds, (struct sockaddr *) sg, sg->sg_len) < 0) { |
| 154 | errnosv = errno; |
| 155 | if (_gNgDebugLevel >= 1) |
| 156 | NGLOG("connect(%s)", sg->sg_data); |
| 157 | goto errout; |
| 158 | } |
| 159 | } |
| 160 | |
| 161 | /* Return the socket(s) */ |
| 162 | if (csp) |
| 163 | *csp = cs; |
| 164 | else |
| 165 | close(cs); |
| 166 | if (dsp) |
| 167 | *dsp = ds; |
| 168 | return (0); |
| 169 | |
| 170 | errout: |
| 171 | /* Failed */ |
| 172 | if (cs >= 0) |
| 173 | close(cs); |
| 174 | if (ds >= 0) |
| 175 | close(ds); |
| 176 | errno = errnosv; |
| 177 | return (-1); |
| 178 | } |
| 179 | |
| 180 | /* |
| 181 | * Assign a globally unique name to a node |
| 182 | * Returns -1 if error and sets errno. |
| 183 | */ |
| 184 | int |
| 185 | NgNameNode(int cs, const char *path, const char *fmt, ...) |
| 186 | { |
| 187 | struct ngm_name ngn; |
| 188 | va_list args; |
| 189 | |
| 190 | /* Build message arg */ |
| 191 | va_start(args, fmt); |
| 192 | vsnprintf(ngn.name, sizeof(ngn.name), fmt, args); |
| 193 | va_end(args); |
| 194 | |
| 195 | /* Send message */ |
| 196 | if (NgSendMsg(cs, path, |
| 197 | NGM_GENERIC_COOKIE, NGM_NAME, &ngn, sizeof(ngn)) < 0) { |
| 198 | if (_gNgDebugLevel >= 1) |
| 199 | NGLOGX("%s: failed", __func__); |
| 200 | return (-1); |
| 201 | } |
| 202 | |
| 203 | /* Done */ |
| 204 | return (0); |
| 205 | } |
| 206 | |
| 207 | /* |
| 208 | * Read a packet from a data socket |
| 209 | * Returns -1 if error and sets errno. |
| 210 | */ |
| 211 | int |
| 212 | NgRecvData(int ds, u_char * buf, size_t len, char *hook) |
| 213 | { |
| 214 | u_char frombuf[NG_HOOKSIZ - 1 + sizeof(struct sockaddr_ng)]; |
| 215 | struct sockaddr_ng *const from = (struct sockaddr_ng *) frombuf; |
| 216 | int fromlen = sizeof(frombuf); |
| 217 | int rtn, errnosv; |
| 218 | |
| 219 | /* Read packet */ |
| 220 | rtn = recvfrom(ds, buf, len, 0, (struct sockaddr *) from, &fromlen); |
| 221 | if (rtn < 0) { |
| 222 | errnosv = errno; |
| 223 | if (_gNgDebugLevel >= 1) |
| 224 | NGLOG("recvfrom"); |
| 225 | errno = errnosv; |
| 226 | return (-1); |
| 227 | } |
| 228 | |
| 229 | /* Copy hook name */ |
| 230 | if (hook != NULL) |
| 231 | snprintf(hook, NG_HOOKSIZ, "%s", from->sg_data); |
| 232 | |
| 233 | /* Debugging */ |
| 234 | if (_gNgDebugLevel >= 2) { |
| 235 | NGLOGX("READ %s from hook \"%s\" (%d bytes)", |
| 236 | rtn ? "PACKET" : "EOF", from->sg_data, rtn); |
| 237 | if (_gNgDebugLevel >= 3) |
| 238 | _NgDebugBytes(buf, rtn); |
| 239 | } |
| 240 | |
| 241 | /* Done */ |
| 242 | return (rtn); |
| 243 | } |
| 244 | |
| 245 | /* |
| 246 | * Write a packet to a data socket. The packet will be sent |
| 247 | * out the corresponding node on the specified hook. |
| 248 | * Returns -1 if error and sets errno. |
| 249 | */ |
| 250 | int |
| 251 | NgSendData(int ds, const char *hook, const u_char * buf, size_t len) |
| 252 | { |
| 253 | u_char sgbuf[NG_HOOKSIZ - 1 + sizeof(struct sockaddr_ng)]; |
| 254 | struct sockaddr_ng *const sg = (struct sockaddr_ng *) sgbuf; |
| 255 | int errnosv; |
| 256 | |
| 257 | /* Set up destination hook */ |
| 258 | sg->sg_family = AF_NETGRAPH; |
| 259 | snprintf(sg->sg_data, NG_HOOKSIZ, "%s", hook); |
| 260 | sg->sg_len = strlen(sg->sg_data) + 3; |
| 261 | |
| 262 | /* Debugging */ |
| 263 | if (_gNgDebugLevel >= 2) { |
| 264 | NGLOGX("WRITE PACKET to hook \"%s\" (%d bytes)", hook, len); |
| 265 | _NgDebugSockaddr(sg); |
| 266 | if (_gNgDebugLevel >= 3) |
| 267 | _NgDebugBytes(buf, len); |
| 268 | } |
| 269 | |
| 270 | /* Send packet */ |
| 271 | if (sendto(ds, buf, len, 0, (struct sockaddr *) sg, sg->sg_len) < 0) { |
| 272 | errnosv = errno; |
| 273 | if (_gNgDebugLevel >= 1) |
| 274 | NGLOG("sendto(%s)", sg->sg_data); |
| 275 | errno = errnosv; |
| 276 | return (-1); |
| 277 | } |
| 278 | |
| 279 | /* Done */ |
| 280 | return (0); |
| 281 | } |
| 282 | |