Initial commit

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

4 files changed, 246 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..e660fd9
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+bin/
diff --git a/Makefile b/Makefile
new file mode 100644
index 0000000..dc235bb
--- /dev/null
+++ b/Makefile
@@ -0,0 +1,16 @@
+CFILES_SERVER = $(shell find src/server -name "*.c")
+CFILES_CLIENT = $(shell find src/client -name "*.c")
+CFLAGS_OMOC = -pedantic -lostp -lssl -lcrypto -lpthread -Iinclude/
+
+.PHONY: all
+all: bin/server bin/client
+
+.PHONY: bin/client
+bin/server:
+	mkdir -p $(@D)
+	$(CC) $(CFILES_SERVER) $(CFLAGS_OMOC) -o $@
+
+.PHONY: bin/client
+bin/client:
+	mkdir -p $(@D)
+	$(CC) $(CFILES_CLIENT) $(CFLAGS_OMOC) -o $@
diff --git a/src/client/client.c b/src/client/client.c
new file mode 100644
index 0000000..f8d3b9a
--- /dev/null
+++ b/src/client/client.c
@@ -0,0 +1,110 @@
+/*
+ * 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 <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <signal.h>
+#include <pthread.h>
+
+// #define CENTRAL_SERVER "127.0.0.1"
+#define CENTRAL_SERVER "149.248.6.149"
+
+static struct ostp_session s;
+static pthread_t td;
+static volatile int got_thread = 0;
+
+static void *
+recv_td(void *args)
+{
+    char buf[4096];
+    got_thread = 1;
+
+    while (1) {
+        if (session_recv(buf, sizeof(buf) - 1, &s) < 0) {
+            printf("session_recv() failure\n");
+            return NULL;
+        }
+        printf("%s\n>> ", buf);
+        fflush(stdout);
+    }
+}
+
+static void
+sighandle(int dmmy)
+{
+    printf("Cleaning up... goodbye!\n");
+    if (got_thread) {
+        pthread_cancel(td);
+    }
+    session_close(&s);
+    exit(0);
+}
+
+int
+main(int argc, char **argv)
+{
+    int c, error;
+    char input[2048];
+    char send_buf[2048];
+    size_t len;
+
+    setbuf(stdin, NULL);
+    setbuf(stdout, NULL);
+    signal(SIGINT, sighandle);
+
+    if ((error = session_new(CENTRAL_SERVER, &s)) < 0) {
+        printf("Failed to create new session\n");
+        return error;
+    }
+
+    error = pthread_create(&td, NULL, recv_td, NULL);
+    if (error != 0) {
+        session_close(&s);
+        return error;
+    }
+
+    while (1) {
+        printf(">> ");
+        fflush(stdout);
+
+        if (fgets(input, sizeof(input), stdin) == NULL) {
+            printf("\n");
+            continue;
+        }
+
+        input[strcspn(input, "\n")] = '\0';
+        snprintf(send_buf, sizeof(send_buf), "%s: %s", s.username, input);
+        session_send(send_buf, strlen(send_buf) + 1, &s);
+    }
+
+    __builtin_unreachable();
+}
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;
+}