]> git.tuebingen.mpg.de Git - paraslash.git/blobdiff - net.c
07_flow-opt-support.diff
[paraslash.git] / net.c
diff --git a/net.c b/net.c
index b83ab8c61d34c3a52f8e0f35dfa63d05c3584f21..d85321d931d51f2d21a0fa8e8ea160df432b1ece 100644 (file)
--- a/net.c
+++ b/net.c
@@ -34,6 +34,7 @@
 #include "crypt.h"
 #include "net.h"
 #include "string.h"
+#include "list.h"
 #include "fd.h"
 
 /**
@@ -242,6 +243,111 @@ static const char *layer4_name(const unsigned l4type)
        return "UNKNOWN PROTOCOL";
 }
 
+/**
+ * Flowopts: Transport-layer independent encapsulation of socket options.
+ *
+ * These collect individual socket options into a queue, which is disposed of
+ * directly after makesock(). The 'pre_conn_opt' structure is for internal use
+ * only and should not be visible elsewhere.
+ *
+ * \sa setsockopt(2), makesock()
+ */
+struct pre_conn_opt {
+       int             sock_level;     /**< Second argument to setsockopt() */
+       int             sock_option;    /**< Third argument to setsockopt()  */
+       char            *opt_name;      /**< Stringified \a sock_option      */
+       void            *opt_val;       /**< Fourth argument to setsockopt() */
+       socklen_t       opt_len;        /**< Fifth argument to setsockopt()  */
+
+       struct list_head node;          /**< FIFO, as sockopt order matters. */
+};
+
+/** FIFO list of pre-connection socket options to be set */
+struct flowopts {
+       struct list_head sockopts;
+};
+
+struct flowopts *flowopt_new(void)
+{
+       struct flowopts *new = para_malloc(sizeof(*new));
+
+       INIT_LIST_HEAD(&new->sockopts);
+       return new;
+}
+
+/**
+ * Append new socket option to flowopt queue.
+ *
+ * \param fo   The flowopt queue to append to.
+ * \param lev  Level at which \opt resides.
+ * \param opt  New option to add.
+ * \param name Stringified name of \a opt.
+ * \param val  The value to set \a opt to.
+ * \param len  Length of \a val.
+ *
+ *  \sa setsockopt(2)
+ */
+void flowopt_add(struct flowopts *fo, int lev, int opt,
+                char *name, const void *val, int len)
+{
+       struct pre_conn_opt *new = para_malloc(sizeof(*new));
+
+       new->sock_option = opt;
+       new->sock_level  = lev;
+       new->opt_name    = para_strdup(name);
+
+       if (val == NULL) {
+               new->opt_val = NULL;
+               new->opt_len = 0;
+       } else {
+               new->opt_val = para_malloc(len);
+               new->opt_len = len;
+               memcpy(new->opt_val, val, len);
+       }
+
+       list_add_tail(&new->node, &fo->sockopts);
+}
+
+void flowopt_add_bool(struct flowopts *fo, int lev, int opt,
+                     char *optname, bool on_or_off)
+{
+       int on = on_or_off;     /* kernel takes 'int' */
+
+       flowopt_add(fo, lev, opt, optname, &on, sizeof(on));
+}
+
+/** Set the entire bunch of pre-connection options at once. */
+static void flowopt_setopts(int sockfd, struct flowopts *fo)
+{
+       struct pre_conn_opt *pc;
+
+       if (fo == NULL)
+               return;
+
+       list_for_each_entry(pc, &fo->sockopts, node)
+               if (setsockopt(sockfd, pc->sock_level, pc->sock_option,
+                                      pc->opt_val, pc->opt_len) < 0) {
+                       PARA_EMERG_LOG("Can not set %s socket option: %s",
+                                      pc->opt_name, strerror(errno));
+                       exit(EXIT_FAILURE);
+               }
+}
+
+static void flowopt_cleanup(struct flowopts *fo)
+{
+       struct pre_conn_opt *cur, *next;
+
+       if (fo == NULL)
+               return;
+
+       list_for_each_entry_safe(cur, next, &fo->sockopts, node) {
+               free(cur->opt_name);
+               free(cur->opt_val);
+               free(cur);
+       }
+       free(fo);
+}
+
 /**
  * Resolve IPv4/IPv6 address and create a ready-to-use active or passive socket.
  *
@@ -250,6 +356,7 @@ static const char *layer4_name(const unsigned l4type)
  * \param passive Whether this is a passive (1) or active (0) socket.
  * \param host Remote or local hostname or IPv/6 address string.
  * \param port_number Decimal port number.
+ * \param fo Socket options to be set before making the connection.
  *
  * This creates a ready-made IPv4/v6 socket structure after looking up the
  * necessary parameters. The interpretation of \a host depends on the value of
@@ -263,7 +370,7 @@ static const char *layer4_name(const unsigned l4type)
  *
  * Furthermore, bind(2) is called on passive sockets, and connect(2) on active
  * sockets. The algorithm tries all possible address combinations until it
- * succeeds.
+ * succeeds. If \a fo is supplied, options are set and cleanup is performed.
  *
  * \return This function returns 1 on success and \a -E_ADDRESS_LOOKUP when no
  * matching connection could be set up (with details in the error log).
@@ -271,7 +378,8 @@ static const char *layer4_name(const unsigned l4type)
  *  \sa ipv6(7), getaddrinfo(3), bind(2), connect(2).
  */
 int makesock(unsigned l3type, unsigned l4type, int passive,
-               const char *host, unsigned short port_number)
+            const char *host, unsigned short port_number,
+            struct flowopts *fo)
 {
        struct addrinfo *local = NULL, *src,
                        *remote = NULL, *dst, hints;
@@ -331,6 +439,7 @@ int makesock(unsigned l3type, unsigned l4type, int passive,
                                strerror(errno));
                        return -ERRNO_TO_PARA_ERROR(errno);
                }
+               flowopt_setopts(sockfd, fo);
 
                if (src) {
                        if (bind(sockfd, src->ai_addr, src->ai_addrlen) < 0) {
@@ -355,6 +464,7 @@ get_next_src:
                freeaddrinfo(local);
        if (remote)
                freeaddrinfo(remote);
+       flowopt_cleanup(fo);
 
        if (src == NULL && dst == NULL) {
                PARA_ERROR_LOG("can not create %s socket %s#%s.\n",
@@ -379,7 +489,7 @@ get_next_src:
  */
 int para_listen(unsigned l3type, unsigned l4type, unsigned short port)
 {
-       int ret, fd = makesock(l3type, l4type, 1, NULL, port);
+       int ret, fd = makesock(l3type, l4type, 1, NULL, port, NULL);
 
        if (fd > 0) {
                ret = listen(fd, BACKLOG);
@@ -672,7 +782,7 @@ int para_accept(int fd, void *addr, socklen_t size)
  */
 const uint8_t *dccp_available_ccids(uint8_t *ccids, uint8_t *nccids)
 {
-       int fd = makesock(AF_UNSPEC, IPPROTO_DCCP, 0, NULL, 0);
+       int fd = makesock(AF_UNSPEC, IPPROTO_DCCP, 1, NULL, 0, NULL);
 
        if (fd < 0)
                return NULL;