diff options
Diffstat (limited to 'src/server')
-rw-r--r-- | src/server/server.c | 119 |
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; +} |