From ea053d96f7e89e053d4af8d39b04c5428760345f Mon Sep 17 00:00:00 2001
From: David Härdeman <david@hardeman.nu>
Date: Tue, 23 Jun 2020 20:56:22 +0200
Subject: Big renaming, move some more functionality to shared lib

---
 minecproxy/announce.c | 116 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 116 insertions(+)
 create mode 100644 minecproxy/announce.c

(limited to 'minecproxy/announce.c')

diff --git a/minecproxy/announce.c b/minecproxy/announce.c
new file mode 100644
index 0000000..13ef423
--- /dev/null
+++ b/minecproxy/announce.c
@@ -0,0 +1,116 @@
+#include <inttypes.h>
+#include <sys/socket.h>
+#include <netinet/ip.h>
+#include <arpa/inet.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "main.h"
+#include "uring.h"
+#include "announce.h"
+#include "server.h"
+#include "ptimer.h"
+
+struct announce {
+	uint64_t value;
+	struct uring_task task;
+	struct ptimer_task ptask;
+	int mcast_fd;
+};
+
+static void
+announce_cb(struct ptimer_task *ptask)
+{
+	struct announce *announce = container_of(ptask, struct announce, ptask);
+	struct server *server;
+
+	assert_return(ptask);
+	assert_task_alive(DBG_ANN, &announce->task);
+
+	debug(DBG_ANN, "announcing servers");
+	list_for_each_entry(server, &cfg->servers, list)
+		server_announce(server, announce->mcast_fd);
+}
+
+static void
+announce_free(struct uring_task *task)
+{
+	struct announce *announce = container_of(task, struct announce, task);
+
+	assert_return(task);
+	debug(DBG_ANN, "task %p, announce 0x%p, mcast_fd: %i",
+	      task, announce, announce->mcast_fd);
+	close(announce->mcast_fd);
+	xfree(announce);
+}
+
+void
+announce_refdump()
+{
+	assert_return_silent(cfg->announce);
+
+	uring_task_refdump(&cfg->announce->task);
+}
+
+void
+announce_delete()
+{
+	assert_return_silent(cfg->announce);
+
+	debug(DBG_ANN, "called");
+	announce_stop();
+	uring_task_destroy(&cfg->announce->task);
+	cfg->announce = NULL;
+}
+
+void
+announce_stop()
+{
+	struct announce *announce = cfg->announce;
+
+	assert_return_silent(announce);
+
+	ptimer_del_task(&announce->ptask);
+}
+
+void
+announce_start(unsigned duration)
+{
+	struct announce *announce = cfg->announce;
+	unsigned times;
+
+	assert_return_silent(announce);
+
+	if (duration == 0)
+		times = 0;
+	else
+		times = MAX(announce->ptask.times,
+			    DIV_ROUND_UP(duration, cfg->announce_interval));
+
+	announce->ptask.times = times;
+	ptimer_add_task(&announce->ptask);
+}
+
+void
+announce_init()
+{
+	struct announce *announce;
+	int sfd;
+ 
+	assert_return(!cfg->announce);
+	assert_return_silent(cfg->announce_interval > 0);
+
+	announce = zmalloc(sizeof(*announce));
+	if (!announce)
+		die("malloc: %m");
+
+	sfd = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0);
+	if (sfd < 0)
+		die("socket: %m");
+
+	uring_task_init(&announce->task, "announce", uring_parent(), announce_free);
+	ptask_init(&announce->ptask, cfg->announce_interval, 0, announce_cb);
+	announce->mcast_fd = sfd;
+	cfg->announce = announce;
+}
+
-- 
cgit v1.2.3