use 'struct in6_pktinfo' from linux/ipv6.h

[psmb.git] / src / psmb_socket.c

diff --git a/src/psmb_socket.c b/src/psmb_socket.c
index d18ad57c506327fbfb03078fdc6fd1b1d44ae07c..a68b1ebd995144e23f2cf7cff432cba50b8eecd7 100644 (file)
--- a/src/psmb_socket.c
+++ b/src/psmb_socket.c
@@ -1,3 +1,4 @@
+#include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
@@ -11,6 +12,15 @@
 #include "psmb_priv.h"
 #include "hash64.h"
 
+#include <linux/ipv6.h> /* contains definition of `struct in6_pktinfo`, */
+                       /* but only if _GNU_SOURCE is defined. Arrgh!   */
+/* The structure itself *should* be like this:
+struct in6_pktinfo {
+       struct in6_addr ipi6_addr;
+       int             ipi6_ifindex;
+};
+*/
+
 static void dummy_log(void *log_priv, int priority, const char *format, ...) {}
 
 psmb_ctx_t *psmb_new(void)
@@ -135,7 +145,7 @@ psmb_result_t psmb_open(psmb_ctx_t *ctx)
        if (setsockopt(ctx->fd, IPPROTO_IPV6, IPV6_RECVPKTINFO,
                        &on, sizeof(on)) < 0) {
                int sverr = errno;
-               LOG(ctx, LOG_ERR, "setsockopt(..., IPV6_PKTINFO, ...): %m");
+               LOG(ctx, LOG_ERR, "setsockopt(..., IPV6_RECVPKTINFO, ...): %m");
                close(ctx->fd);
                ctx->fd = -1;
                errno = sverr;
@@ -150,6 +160,7 @@ psmb_result_t psmb_open(psmb_ctx_t *ctx)
                errno = sverr;
                return (psmb_result_t){PSMB_ERROR};
        }
+       /* TODO: set non-blocking */
        return (psmb_result_t){PSMB_OK};
 }
 
@@ -192,14 +203,99 @@ static psmb_result_t psmb_sub_unsub(psmb_ctx_t *ctx, char *channel, int option)
        return (psmb_result_t){PSMB_OK};
 }
 
-psmb_result_t psmb_subscribe(psmb_ctx_t *ctx, char *channel) {
+psmb_result_t psmb_subscribe(psmb_ctx_t *ctx, char *channel)
+{
        return psmb_sub_unsub(ctx, channel, IPV6_ADD_MEMBERSHIP);
 }
 
-psmb_result_t psmb_unsubscribe(psmb_ctx_t *ctx, char *channel) {
+psmb_result_t psmb_unsubscribe(psmb_ctx_t *ctx, char *channel)
+{
        return psmb_sub_unsub(ctx, channel, IPV6_DROP_MEMBERSHIP);
 }
 
+psmb_result_t psmb_ev_rd(psmb_ctx_t *ctx)
+{
+       ssize_t readsize;
+       struct sockaddr_in6 peer_addr;
+       struct in6_addr self_addr = {{{0}}};
+       unsigned char msgbuf[BUFSIZ];
+       unsigned char cmsgbuf[BUFSIZ];
+       struct iovec iov[1] = {{
+               .iov_base=msgbuf,
+               .iov_len=sizeof(msgbuf),
+       }};
+       struct msghdr message = {
+               .msg_name=&peer_addr,
+               .msg_namelen=sizeof(peer_addr),
+               .msg_iov=iov,
+               .msg_iovlen=1,
+               .msg_control=cmsgbuf,
+               .msg_controllen=sizeof(cmsgbuf),
+       };
+       struct cmsghdr *cmsg;
+       char peer_str[INET6_ADDRSTRLEN+1];
+       char self_str[INET6_ADDRSTRLEN+1];
+
+       if ((readsize = recvmsg(ctx->fd, &message, 0)) == -1) {
+               if (errno == EWOULDBLOCK)
+                       return (psmb_result_t){PSMB_OK};
+               else {
+                       int sverr = errno;
+                       LOG(ctx, LOG_ERR, "recvmsg(..., 0): %m");
+                       errno = sverr;
+                       return (psmb_result_t){PSMB_ERROR};
+               }
+       }
+       for (cmsg = CMSG_FIRSTHDR(&message);
+            cmsg != NULL;
+            cmsg = CMSG_NXTHDR(&message, cmsg)) {
+               LOG(ctx, LOG_DEBUG, "CMSG: level %d, type %d - skip",
+                                       cmsg->cmsg_level, cmsg->cmsg_type);
+               if (cmsg->cmsg_level == IPPROTO_IPV6 &&
+                               cmsg->cmsg_type == IPV6_PKTINFO) {
+                       struct in6_pktinfo *pi =
+                               (struct in6_pktinfo *)CMSG_DATA(cmsg);
+                       self_addr = pi->ipi6_addr;
+               }
+       }
+       (void)inet_ntop(AF_INET6, &peer_addr.sin6_addr,
+                       peer_str, sizeof(peer_str));
+       (void)inet_ntop(AF_INET6, &self_addr,
+                       self_str, sizeof(self_str));
+       LOG(ctx, LOG_DEBUG, "CMSG: %d bytes from %s to %s",
+                       readsize, peer_str, self_str);
+
+       return (psmb_result_t){PSMB_OK};
+}
+
+psmb_result_t psmb_ev_wr(psmb_ctx_t *ctx)
+{
+       return (psmb_result_t){PSMB_OK};
+}
+
+psmb_result_t psmb_ev_ex(psmb_ctx_t *ctx)
+{
+       return (psmb_result_t){PSMB_OK};
+}
+
+psmb_result_t psmb_publish(psmb_ctx_t *ctx, char *channel,
+       void *data, size_t size)
+{
+       /*
+       unsigned long ifindex = 0;
+       setsockopt(ctx->fd, IPPROTO_IPV6, IPV6_MULTICAST_IF,
+                       &ifindex, sizeof(ifindex));
+       */
+
+       return (psmb_result_t){PSMB_OK};
+}
+
+psmb_result_t psmb_get_message(psmb_ctx_t *ctx, char **channel,
+       void **data, size_t *size)
+{
+       return (psmb_result_t){PSMB_OK};
+}
+
 bool psmb_success(psmb_result_t result)
 {
        return !(result.code & PSMB_ERROR);