Initial commit

Signed-off-by: Ian Moffett <ian@osmora.org>

1 files changed, 119 insertions, 0 deletions

diff --git a/src/server/server.c b/src/server/server.c
new file mode 100644
index 0000000..d66bd77
--- /dev/null
+++ b/src/server/server.c
@@ -0,0 +1,119 @@
+/*
+ * Copyright (c) 2023-2024 Ian Marco Moffett and the Osmora Team.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ *    this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. Neither the name of Hyra nor the names of its
+ *    contributors may be used to endorse or promote products derived from
+ *    this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <ostp/session.h>
+#include <ostp/server.h>
+#include <stdio.h>
+#include <string.h>
+
+#define OMOC_PORT 5352
+
+static struct ostp_listener l;
+
+static int
+send_to(void *buf, size_t len, struct ostp_client *c)
+{
+    struct ostp_session *s;
+
+    s = &c->session;
+    return send_frame(s->sockfd, buf, len, s->session_key);
+}
+
+static int
+relay_others(struct ostp_client *sender, const void *buf, size_t len)
+{
+    struct ostp_client *clients[MAX_CLIENTS];
+    char send_buf[4096];
+    int client_count;
+
+    client_count = listener_clients(&l, clients);
+    if (client_count < 0) {
+        return client_count;
+    }
+
+    memcpy(send_buf, buf, len);
+    printf("MESSAGE FROM SOCKFD: %d\n", sender->sockfd);
+    printf("SERVER SOCKFD: %d\n", l.serv_sockfd);
+
+    /* Relay to all *other* clients */
+    for (int i = 0; i < client_count; ++i) {
+        /* Do not relay to server or self */
+        if (clients[i]->sockfd == l.serv_sockfd) {
+            printf("SKIPPED SERVER SOCKET (%d)\n", i);
+            continue;
+        } else if (clients[i]->sockfd == sender->sockfd) {
+            printf("SKIPPED SENDER SOCKET (%d)\n", i);
+            continue;
+        }
+
+        printf("SEND TO CLIENT: %d\n", i);
+        session_send(send_buf, len, &clients[i]->session);
+        printf("SEND TO CLIENT DONE\n");
+    }
+
+    return 0;
+}
+
+static int
+recv_msg(struct ostp_client *c, const void *buf, size_t len)
+{
+    return relay_others(c, buf, len);
+}
+
+static int
+on_conn(struct ostp_client *c)
+{
+    struct ostp_session *s;
+    char buf[4096];
+
+    s = &c->session;
+    snprintf(buf, sizeof(buf), "[system]: %s has joined", s->username);
+    return relay_others(c, buf, sizeof(buf));
+}
+
+int
+main(void)
+{
+    int error;
+
+    listener_init(&l);
+    l.port = OMOC_PORT;
+    l.on_recv = recv_msg;
+    l.on_connect = on_conn;
+
+    /* Bind and poll the listener */
+    if ((error = listener_bind(&l)) < 0) {
+        return error;
+    }
+    if ((error = listener_poll(&l)) < 0) {
+        return error;
+    }
+
+    return 0;
+}