From 7369505397cdfddf0883e2c24e1652df8bd488fe Mon Sep 17 00:00:00 2001
From: Guangxiong Lin <hi@gxlin.org>
Date: Fri, 16 Dec 2022 13:53:35 +0800
Subject: Refactor file structure

---
 src/tsocket.c | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 src/tsocket.c

(limited to 'src/tsocket.c')

diff --git a/src/tsocket.c b/src/tsocket.c
new file mode 100644
index 0000000..8d8038a
--- /dev/null
+++ b/src/tsocket.c
@@ -0,0 +1,59 @@
+#include <arpa/inet.h>
+#include <stdlib.h>
+#include <sys/socket.h>
+#include <unistd.h>
+
+#include "tsocket.h"
+
+struct tsocket *tsocket_create()
+{
+    int fd = socket(AF_INET, SOCK_STREAM, 0);
+    if (fd == -1)
+        return NULL;
+
+    struct tsocket *sock = malloc(sizeof(*sock));
+    sock->fd = fd;
+
+    return sock;
+}
+
+int tsocket_bind(struct tsocket *sock, const char *addr, int hostport)
+{
+    struct sockaddr_in sock_addr;
+    sock_addr.sin_family = AF_INET;
+    sock_addr.sin_addr.s_addr = inet_addr(addr);
+    sock_addr.sin_port = htons(hostport);
+
+    sock->addr = addr;
+    sock->port = hostport;
+
+    return bind(sock->fd, (struct sockaddr *)&sock_addr, sizeof(sock_addr));
+}
+
+int tsocket_listen(struct tsocket *sock)
+{
+    return listen(sock->fd, SOMAXCONN);
+}
+
+struct tsocket *tsocket_accept(struct tsocket *sock)
+{
+    struct sockaddr_in addr;
+    socklen_t addr_len = sizeof(addr);
+
+    int fd = accept(sock->fd, (struct sockaddr *)&addr, &addr_len);
+    if (fd == -1)
+        return NULL;
+
+    struct tsocket *conn_sock = malloc(sizeof(*conn_sock));
+    conn_sock->fd = fd;
+    conn_sock->addr = inet_ntoa(addr.sin_addr);
+    conn_sock->port = ntohs(addr.sin_port);
+
+    return conn_sock;
+}
+
+void tsocket_destroy(struct tsocket *sock)
+{
+    close(sock->fd);
+    free(sock);
+}
-- 
cgit v1.2.3