#include <stdlib.h>
+#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <sys/types.h>
#include <psmb.h>
#include "psmb_priv.h"
+#include "hash64.h"
static void dummy_log(void *log_priv, int priority, const char *format, ...) {}
.fd = -1,
.malloc = malloc, .free = free, .realloc = realloc,
.logf = dummy_log,
+ .prefix = (struct in6_addr){{{ 0xff, 0x15, 'P', 'S',
+ 'M', 'B', '0', '1',
+ 0x00, 0x00, 0x00, 0x00,
+ 0x00, 0x00, 0x00, 0x00 }}},
+ .prefixlen = 64,
.pmtu = PSMB_DEFAULT_PMTU,
.port = PSMB_DEFAULT_PORT};
return ctx;
psmb_result_t psmb_set_mgrp(psmb_ctx_t *ctx, struct in6_addr prefix,
unsigned char prefixlen)
{
+ if (prefixlen > 128) {
+ LOG(ctx, LOG_ERR, "psmb_set_mgrp() prefixlen %d is too big",
+ prefixlen);
+ errno = EINVAL;
+ return (psmb_result_t){PSMB_ERROR};
+ }
if (ctx->fd == -1) {
ctx->prefix = prefix;
ctx->prefixlen = prefixlen;
return (psmb_result_t){PSMB_OK};
}
+static struct in6_addr multiaddr(struct in6_addr prefix,
+ unsigned char prefixlen, uint64_t suffix)
+{
+ struct in6_addr result = prefix;
+ unsigned char len = prefixlen > 64 ? prefixlen : 64;
+ uint64_t mask = len == 64 ? ~(uint64_t)0 : ((uint64_t)1 << len) - 1;
+
+ *(uint64_t *)(&result.__in6_u.__u6_addr32[2]) &= ~mask;
+ *(uint64_t *)(&result.__in6_u.__u6_addr32[2]) |= (suffix & mask);
+ return result;
+}
+
+static psmb_result_t psmb_sub_unsub(psmb_ctx_t *ctx, char *channel, int option)
+{
+ struct ipv6_mreq mreq = { 0 };
+ char mgrp_str[INET6_ADDRSTRLEN+1];
+
+ if (ctx->fd == -1) {
+ LOG(ctx, LOG_ERR, "subscribe: psmb is not open");
+ errno = EINVAL;
+ return (psmb_result_t){PSMB_ERROR};
+ }
+ mreq.ipv6mr_multiaddr = multiaddr(ctx->prefix, ctx->prefixlen,
+ hash64(channel, strlen(channel)));
+ (void)inet_ntop(AF_INET6, &mreq.ipv6mr_multiaddr,
+ mgrp_str, sizeof(mgrp_str));
+ LOG(ctx, LOG_DEBUG, "using multiaddr %s for channel \"%s\"",
+ mgrp_str, channel);
+ mreq.ipv6mr_interface = 0; /* how to use this??? */
+ if (setsockopt(ctx->fd, IPPROTO_IPV6, option,
+ (void *)&mreq, sizeof(mreq)) == -1) {
+ int sverr = errno;
+ LOG(ctx, LOG_ERR, "add_membership(): %m");
+ errno = sverr;
+ return (psmb_result_t){PSMB_ERROR};
+ }
+ return (psmb_result_t){PSMB_OK};
+}
+
+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) {
+ return psmb_sub_unsub(ctx, channel, IPV6_DROP_MEMBERSHIP);
+}
+
+bool psmb_success(psmb_result_t result)
+{
+ return !(result.code & PSMB_ERROR);
+}
+
+bool psmb_message_waiting(psmb_result_t result)
+{
+ return !!(result.code & PSMB_MESSAGE);
+}
+
+bool psmb_need_write_wait(psmb_result_t result)
+{
+ return !!(result.code & PSMB_NEED_WRITE);
+}
+
void psmb_destroy(psmb_ctx_t *ctx)
{
if (ctx->fd == -1) {