summaryrefslogtreecommitdiffstats
path: root/dhcp6_ctl.c
diff options
context:
space:
mode:
Diffstat (limited to 'dhcp6_ctl.c')
-rw-r--r--dhcp6_ctl.c368
1 files changed, 368 insertions, 0 deletions
diff --git a/dhcp6_ctl.c b/dhcp6_ctl.c
new file mode 100644
index 0000000..ab1f18d
--- /dev/null
+++ b/dhcp6_ctl.c
@@ -0,0 +1,368 @@
+/* $KAME: dhcp6_ctl.c,v 1.4 2004/09/07 05:03:03 jinmei Exp $ */
+
+/*
+ * Copyright (C) 2004 WIDE Project.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * 3. Neither the name of the project nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+#include <sys/param.h>
+#include <sys/types.h>
+#include <sys/queue.h>
+#include <sys/socket.h>
+#if TIME_WITH_SYS_TIME
+# include <sys/time.h>
+# include <time.h>
+#else
+# if HAVE_SYS_TIME_H
+# include <sys/time.h>
+# else
+# include <time.h>
+# endif
+#endif
+
+#include <netinet/in.h>
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <string.h>
+#include <syslog.h>
+#include <netdb.h>
+#include <errno.h>
+
+#include <dhcp6.h>
+#include <config.h>
+#include <common.h>
+#include <auth.h>
+#include <base64.h>
+#include <control.h>
+#include <dhcp6_ctl.h>
+
+TAILQ_HEAD(dhcp6_commandqueue, dhcp6_commandctx);
+
+static struct dhcp6_commandqueue commandqueue_head;
+static int max_commands;
+static int commands = 0;
+
+struct dhcp6_commandctx {
+ TAILQ_ENTRY(dhcp6_commandctx) link;
+
+ int s; /* communication socket */
+ char inputbuf[1024]; /* input buffer */
+ ssize_t input_len;
+ ssize_t input_filled;
+ int (*callback) __P((char *, ssize_t));
+};
+
+int
+dhcp6_ctl_init(addr, port, max, sockp)
+ char *addr, *port;
+ int max, *sockp;
+{
+ struct addrinfo hints, *res = NULL;
+ int on;
+ int error;
+ int ctlsock = -1;
+
+ memset(&hints, 0, sizeof(hints));
+ hints.ai_family = AF_INET6;
+ hints.ai_socktype = SOCK_STREAM;
+ hints.ai_protocol = IPPROTO_TCP;
+ error = getaddrinfo(addr, port, &hints, &res);
+ if (error) {
+ dprintf(LOG_ERR, FNAME, "getaddrinfo: %s",
+ gai_strerror(error));
+ return (-1);
+ }
+ ctlsock = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
+ if (ctlsock < 0) {
+ dprintf(LOG_ERR, FNAME, "socket(control sock): %s",
+ strerror(errno));
+ goto fail;
+ }
+ on = 1;
+ if (setsockopt(ctlsock, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))
+ < 0) {
+ dprintf(LOG_ERR, FNAME,
+ "setsockopt(control sock, SO_REUSEADDR: %s",
+ strerror(errno));
+ goto fail;
+ }
+ if (bind(ctlsock, res->ai_addr, res->ai_addrlen) < 0) {
+ dprintf(LOG_ERR, FNAME, "bind(control sock): %s",
+ strerror(errno));
+ goto fail;
+ }
+ freeaddrinfo(res);
+ if (listen(ctlsock, 1)) {
+ dprintf(LOG_ERR, FNAME, "listen(control sock): %s",
+ strerror(errno));
+ goto fail;
+ }
+
+ TAILQ_INIT(&commandqueue_head);
+
+ if (max <= 0) {
+ dprintf(LOG_ERR, FNAME,
+ "invalid maximum number of commands (%d)", max_commands);
+ goto fail;
+ }
+ max_commands = max;
+
+ *sockp = ctlsock;
+ return (0);
+
+ fail:
+ if (res != NULL)
+ freeaddrinfo(res);
+ if (ctlsock >= 0)
+ close(ctlsock);
+
+ return (-1);
+}
+
+int
+dhcp6_ctl_authinit(keyfile, keyinfop, digestlenp)
+ char *keyfile;
+ struct keyinfo **keyinfop;
+ int *digestlenp;
+{
+ FILE *fp = NULL;
+ struct keyinfo *ctlkey = NULL;
+ char line[1024], secret[1024];
+ int secretlen;
+
+ /* Currently, we only support HMAC-MD5 for authentication. */
+ *digestlenp = MD5_DIGESTLENGTH;
+
+ if ((fp = fopen(keyfile, "r")) == NULL) {
+ dprintf(LOG_ERR, FNAME, "failed to open %s: %s", keyfile,
+ strerror(errno));
+ return (-1);
+ }
+ if (fgets(line, sizeof(line), fp) == NULL && ferror(fp)) {
+ dprintf(LOG_ERR, FNAME, "failed to read key file: %s",
+ strerror(errno));
+ goto fail;
+ }
+ if ((secretlen = base64_decodestring(line, secret, sizeof(secret)))
+ < 0) {
+ dprintf(LOG_ERR, FNAME, "failed to decode base64 string");
+ goto fail;
+ }
+ if ((ctlkey = malloc(sizeof(*ctlkey))) == NULL) {
+ dprintf(LOG_WARNING, FNAME, "failed to allocate control key");
+ goto fail;
+ }
+ memset(ctlkey, 0, sizeof(*ctlkey));
+ if ((ctlkey->secret = malloc(secretlen)) == NULL) {
+ dprintf(LOG_WARNING, FNAME, "failed to allocate secret key");
+ goto fail;
+ }
+ ctlkey->secretlen = (size_t)secretlen;
+ memcpy(ctlkey->secret, secret, secretlen);
+
+ fclose(fp);
+
+ *keyinfop = ctlkey;
+ return (0);
+
+ fail:
+ if (fp != NULL)
+ fclose(fp);
+ if (ctlkey != NULL && ctlkey->secret != NULL)
+ free(ctlkey->secret);
+ if (ctlkey != NULL)
+ free(ctlkey);
+
+ return (-1);
+}
+
+int
+dhcp6_ctl_acceptcommand(sl, callback)
+ int sl;
+ int (*callback) __P((char *, ssize_t));
+{
+ int s;
+ struct sockaddr_storage from_ss;
+ struct sockaddr *from = (struct sockaddr *)&from_ss;
+ socklen_t fromlen;
+ struct dhcp6_commandctx *ctx, *new;
+
+ fromlen = sizeof(from_ss);
+ if ((s = accept(sl, from, &fromlen)) < 0) {
+ dprintf(LOG_WARNING, FNAME,
+ "failed to accept control connection: %s",
+ strerror(errno));
+ return (-1);
+ }
+
+ dprintf(LOG_DEBUG, FNAME, "accept control connection from %s",
+ addr2str(from));
+
+ if (max_commands <= 0) {
+ dprintf(LOG_ERR, FNAME, "command queue is not initialized");
+ close(s);
+ return (-1);
+ }
+
+ new = malloc(sizeof(*new));
+ if (new == NULL) {
+ dprintf(LOG_WARNING, FNAME,
+ "failed to allocate new command context");
+ goto fail;
+ }
+
+ /* if the command queue is full, purge the oldest one */
+ if (commands == max_commands) {
+ ctx = TAILQ_FIRST(&commandqueue_head);
+
+ dprintf(LOG_INFO, FNAME, "command queue is full. "
+ "drop the oldest one (fd=%d)", ctx->s);
+
+ TAILQ_REMOVE(&commandqueue_head, ctx, link);
+ dhcp6_ctl_closecommand(ctx);
+ }
+
+ /* insert the next context to the queue */
+ memset(new, 0, sizeof(*new));
+ new->s = s;
+ new->callback = callback;
+ new->input_len = sizeof(struct dhcp6ctl);
+ TAILQ_INSERT_TAIL(&commandqueue_head, new, link);
+ commands++;
+
+ return (0);
+
+ fail:
+ close(s);
+
+ return (-1);
+}
+
+void
+dhcp6_ctl_closecommand(ctx)
+ struct dhcp6_commandctx *ctx;
+{
+ close(ctx->s);
+ free(ctx);
+
+ if (commands == 0) {
+ dprintf(LOG_ERR, FNAME, "assumption error: "
+ "command queue is empty?");
+ exit(1); /* XXX */
+ }
+ commands--;
+
+ return;
+}
+
+int
+dhcp6_ctl_readcommand(read_fds)
+ fd_set *read_fds;
+{
+ struct dhcp6_commandctx *ctx, *ctx_next;
+ char *cp;
+ int cc, resid, result;
+ struct dhcp6ctl *ctlhead;
+
+ for (ctx = TAILQ_FIRST(&commandqueue_head); ctx != NULL;
+ ctx = ctx_next) {
+ ctx_next = TAILQ_NEXT(ctx, link);
+
+ if (FD_ISSET(ctx->s, read_fds)) {
+ cp = ctx->inputbuf + ctx->input_filled;
+ resid = ctx->input_len - ctx->input_filled;
+
+ cc = read(ctx->s, cp, resid);
+ if (cc < 0) {
+ dprintf(LOG_WARNING, FNAME, "read failed: %s",
+ strerror(errno));
+ goto closecommand;
+ }
+ if (cc == 0) {
+ dprintf(LOG_INFO, FNAME,
+ "control channel was reset by peer");
+ goto closecommand;
+ }
+
+ ctx->input_filled += cc;
+ if (ctx->input_filled < ctx->input_len)
+ continue; /* we need more data */
+ else if (ctx->input_filled == sizeof(*ctlhead)) {
+ ctlhead = (struct dhcp6ctl *)ctx->inputbuf;
+ ctx->input_len += ntohs(ctlhead->len);
+ }
+
+ if (ctx->input_filled == ctx->input_len) {
+ /* we're done. execute the command. */
+ result = (ctx->callback)(ctx->inputbuf,
+ ctx->input_len);
+
+ switch (result) {
+ case DHCP6CTL_R_DONE:
+ case DHCP6CTL_R_FAILURE:
+ goto closecommand;
+ default:
+ break;
+ }
+ } else if (ctx->input_len > sizeof(ctx->inputbuf)) {
+ dprintf(LOG_INFO, FNAME,
+ "too large command (%d bytes)",
+ ctx->input_len);
+ goto closecommand;
+ }
+
+ continue;
+
+ closecommand:
+ TAILQ_REMOVE(&commandqueue_head, ctx, link);
+ dhcp6_ctl_closecommand(ctx);
+ }
+ }
+
+ return (0);
+}
+
+int
+dhcp6_ctl_setreadfds(read_fds, maxfdp)
+ fd_set *read_fds;
+ int *maxfdp;
+{
+ int maxfd = *maxfdp;
+ struct dhcp6_commandctx *ctx;
+
+ for (ctx = TAILQ_FIRST(&commandqueue_head); ctx != NULL;
+ ctx = TAILQ_NEXT(ctx, link)) {
+ FD_SET(ctx->s, read_fds);
+ if (ctx->s > maxfd)
+ maxfd = ctx->s;
+ }
+
+ *maxfdp = maxfd;
+
+ return (0);
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-21 16:24:14 +0000