rc.d: Introduce 'dhcp_client' to wrap over dhclient and dhcpcd
[dragonfly.git] / lib / libnetgraph / sock.c
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