6 #include <netinet/in.h>
7 #include <sys/socket.h>
10 #include "psmb_priv.h"
12 static void dummy_log(void *log_priv, int priority, const char *format, ...) {}
14 psmb_ctx_t *psmb_new(void)
16 return psmb_new_mm(malloc, free, realloc);
19 psmb_ctx_t *psmb_new_mm(void *(*malloc)(size_t size),
20 void (*free)(void *ptr),
21 void *(*realloc)(void *ptr, size_t size))
23 psmb_ctx_t *ctx = (*malloc)(sizeof(psmb_ctx_t));
26 LOG(ctx, LOG_ERR, "failed to allocate psmb_ctx: %m");
32 .malloc = malloc, .free = free, .realloc = realloc,
34 .prefix = (struct in6_addr){{{ 0xff, 0xff, 0x01, 0x05,
35 0xb0, 0x55, 0xff, 0xe7,
36 0x00, 0x00, 0x00, 0x00,
37 0x00, 0x00, 0x00, 0x00 }}},
39 .pmtu = PSMB_DEFAULT_PMTU,
40 .port = PSMB_DEFAULT_PORT};
44 psmb_result_t psmb_set_logf(psmb_ctx_t *ctx,
45 void (*logf)(void *log_priv, int priority, const char *format, ...),
50 ctx->log_priv = log_priv;
51 return (psmb_result_t){PSMB_OK};
53 LOG(ctx, LOG_ERR, "psmb_set_...() used after psmb_open()");
55 return (psmb_result_t){PSMB_ERROR};
59 psmb_result_t psmb_set_pmtu(psmb_ctx_t *ctx, unsigned int pmtu)
63 return (psmb_result_t){PSMB_OK};
65 LOG(ctx, LOG_ERR, "psmb_set_...() used after psmb_open()");
67 return (psmb_result_t){PSMB_ERROR};
71 psmb_result_t psmb_set_port(psmb_ctx_t *ctx, unsigned short port)
75 return (psmb_result_t){PSMB_OK};
77 LOG(ctx, LOG_ERR, "psmb_set_...() used after psmb_open()");
79 return (psmb_result_t){PSMB_ERROR};
83 psmb_result_t psmb_set_mgrp(psmb_ctx_t *ctx, struct in6_addr prefix,
84 unsigned char prefixlen)
88 ctx->prefixlen = prefixlen;
89 return (psmb_result_t){PSMB_OK};
91 LOG(ctx, LOG_ERR, "psmb_set_...() used after psmb_open()");
93 return (psmb_result_t){PSMB_ERROR};
97 psmb_result_t psmb_open(psmb_ctx_t *ctx)
100 struct sockaddr_in6 addr = (struct sockaddr_in6){
101 .sin6_family = AF_INET6,
102 .sin6_addr = in6addr_any,
103 .sin6_port = htons(ctx->port)
107 LOG(ctx, LOG_ERR, "redundant call to psmb_open()");
109 return (psmb_result_t){PSMB_ERROR};
111 ctx->fd = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
114 LOG(ctx, LOG_ERR, "socket: %m");
116 return (psmb_result_t){PSMB_ERROR};
118 if (setsockopt(ctx->fd, SOL_SOCKET, SO_REUSEADDR,
119 &on, sizeof(on)) < 0) {
121 LOG(ctx, LOG_ERR, "setsockopt(..., SO_REUSEADDR, ...): %m");
125 return (psmb_result_t){PSMB_ERROR};
127 if (setsockopt(ctx->fd, IPPROTO_IPV6, IPV6_RECVPKTINFO,
128 &on, sizeof(on)) < 0) {
130 LOG(ctx, LOG_ERR, "setsockopt(..., IPV6_PKTINFO, ...): %m");
134 return (psmb_result_t){PSMB_ERROR};
136 if (bind(ctx->fd, (struct sockaddr *)&addr,
137 sizeof(struct sockaddr_in6)) == -1) {
139 LOG(ctx, LOG_ERR, "bind(): %m");
143 return (psmb_result_t){PSMB_ERROR};
145 return (psmb_result_t){PSMB_OK};
148 static psmb_result_t psmb_sub_unsub(psmb_ctx_t *ctx, char *channel, int option)
150 struct ipv6_mreq mreq = { 0 };
153 LOG(ctx, LOG_ERR, "subscribe: psmb is not open");
155 return (psmb_result_t){PSMB_ERROR};
157 mreq.ipv6mr_multiaddr = ctx->prefix; /* use hash of the channel */
158 mreq.ipv6mr_interface = 0; /* how to use this??? */
159 if (setsockopt(ctx->fd, IPPROTO_IPV6, option,
160 (void *)&mreq, sizeof(mreq)) == -1) {
162 LOG(ctx, LOG_ERR, "add_membership(): %m");
164 return (psmb_result_t){PSMB_ERROR};
166 return (psmb_result_t){PSMB_OK};
169 psmb_result_t psmb_subscribe(psmb_ctx_t *ctx, char *channel) {
170 return psmb_sub_unsub(ctx, channel, IPV6_ADD_MEMBERSHIP);
173 psmb_result_t psmb_unsubscribe(psmb_ctx_t *ctx, char *channel) {
174 return psmb_sub_unsub(ctx, channel, IPV6_DROP_MEMBERSHIP);
177 bool psmb_success(psmb_result_t result)
179 return !(result.code & PSMB_ERROR);
182 bool psmb_message_waiting(psmb_result_t result)
184 return !!(result.code & PSMB_MESSAGE);
187 bool psmb_need_write_wait(psmb_result_t result)
189 return !!(result.code & PSMB_NEED_WRITE);
192 void psmb_destroy(psmb_ctx_t *ctx)
195 LOG(ctx, LOG_ERR, "psmb_ctx is not open");
197 if (close(ctx->fd) == -1)
198 LOG(ctx, LOG_ERR, "close(): %m");
200 /* clean up the rest */