From d1c4602f80e395d1d6ab0453b8f0a6cc10aefadf Mon Sep 17 00:00:00 2001
From: Laurent Bercot <ska-skaware@skarnet.org>
Date: Tue, 16 Jul 2024 01:51:04 +0000
Subject:  Refactor dcache, add prep for shibari-cache

Signed-off-by: Laurent Bercot <ska@appnovation.com>
---
 src/cache/udpqueue.c | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 src/cache/udpqueue.c

(limited to 'src/cache/udpqueue.c')

diff --git a/src/cache/udpqueue.c b/src/cache/udpqueue.c
new file mode 100644
index 0000000..022df3d
--- /dev/null
+++ b/src/cache/udpqueue.c
@@ -0,0 +1,80 @@
+/* ISC license. */
+
+#include <string.h>
+
+#include <skalibs/stralloc.h>
+#include <skalibs/genalloc.h>
+#include <skalibs/socket.h>
+#include <skalibs/ip46.h>
+
+#include "shibari-cache-internal.h"
+
+int udpqueue_add4 (udpqueue *q, char const *ip, uint16_t port, char const *s, uint16_t len)
+{
+  udp4msg msg = { .port = port, .len = len } ;
+  if (!stralloc_readyplus(&q->storage, len)) return 0 ;
+  memcpy(msg.ip, ip, 4) ;
+  if (!genalloc_append(udp4msg, &q->messages, &msg)) return 0 ;
+  stralloc_catb(&q->storage, s, len) ;
+  return 1 ;
+}
+
+int udpqueue_flush4 (udpqueue *q)
+{
+  size_t n = genalloc_n(udp4msg, &q->messages) ;
+  size_t shead = 0, head = 0 ;
+  while (head < n)
+  {
+    udp4msg const *msg = genalloc_s(udp4msg, &q->messages) + head ;
+    ssize_t r = socket_send4(q->fd, q->storage.s + shead, msg->len, msg->ip, msg->port) ;
+    if (r <= 0) goto adjust ;
+    shead += msg->len ;
+  }
+  q->storage.len = 0 ;
+  genalloc_setlen(udp4msg, &q->messages, 0) ;
+  return 1 ;
+
+ adjust:
+  memmove(q->storage.s, q->storage.s + shead, q->storage.len - shead) ;
+  q->storage.len -= shead ;
+  memmove(genalloc_s(&udp4msg, &q->messages), genalloc_s(&udp4msg, &q->messages) + head, (n - head) * sizeof(udp4msg)) ;
+  genalloc_setlen(&udp4msg, &q->messages, n - head) ;
+  return 0 ;
+}
+
+#ifdef SKALIBS_IPv6_ENABLED
+
+int udpqueue_add6 (udpqueue *q, char const *ip, uint16_t port, char const *s, uint16_t len)
+{
+  udp6msg msg = { .port = port, .len = len } ;
+  if (!stralloc_readyplus(&q->storage, len)) return 0 ;
+  memcpy(msg.ip, ip, 16) ;
+  if (!genalloc_append(udp6msg, &q->messages, &msg)) return 0 ;
+  stralloc_catb(&q->storage, s, len) ;
+  return 1 ;
+}
+
+int udpqueue_flush6 (udpqueue *q)
+{
+  size_t n = genalloc_n(udp6msg, &q->messages) ;
+  size_t shead = 0, head = 0 ;
+  while (head < n)
+  {
+    udp6msg const *msg = genalloc_s(udp4msg, &q->messages) + head ;
+    ssize_t r = socket_send6(q->fd, q->storage.s + shead, msg->len, msg->ip, msg->port) ;
+    if (r <= 0) goto adjust ;
+    shead += msg->len ;
+  }
+  q->storage.len = 0 ;
+  genalloc_setlen(udp6msg, &q->messages, 0) ;
+  return 1 ;
+
+ adjust:
+  memmove(q->storage.s, q->storage.s + shead, q->storage.len - shead) ;
+  q->storage.len -= shead ;
+  memmove(genalloc_s(&udp6msg, &q->messages), genalloc_s(&udp6msg, &q->messages) + head, (n - head) * sizeof(udp6msg)) ;
+  genalloc_setlen(&udp6msg, &q->messages, n - head) ;
+  return 0 ;
+}
+
+#endif
-- 
cgit v1.3.1