main

2 files changed, 203 insertions, 0 deletions

diff --git a/net/net.c b/net/net.c
new file mode 100644
index 0000000..7e1dbf4
--- /dev/null
+++ b/net/net.c
@@ -0,0 +1,168 @@
+#include <net.h>
+
+void itob (char *buf, int n) {
+    buf[0] = (n >> 24) & 0xFF;
+    buf[1] = (n >> 16) & 0xFF;
+    buf[2] = (n >> 8) & 0xFF;
+    buf[3] = n & 0xFF;
+}
+
+int btoi (char *buf) {
+    int n = 0;
+    n += buf[0]*256*256*256;
+    n += buf[1]*256*256;
+    n += buf[2]*256;
+    n += buf[3];
+    return n;
+}
+
+
+void net_init () {
+    p_libsys_init ();
+}
+
+int net_open_socket(PSocket **socket, int port) {
+    *socket = p_socket_new (P_SOCKET_FAMILY_INET,
+        P_SOCKET_TYPE_STREAM,
+        P_SOCKET_PROTOCOL_TCP,
+        NULL);
+    p_socket_set_blocking (*socket, FALSE);
+    p_socket_set_timeout (*socket, 10);
+    PSocketAddress *sock_addr = p_socket_address_new("127.0.0.1", port);
+    int sender_port = 0;
+    if (p_socket_bind (*socket, sock_addr, FALSE, NULL) == FALSE) {
+        p_socket_free (*socket);
+        p_socket_address_free (sock_addr);
+        return 1;
+    } else {
+        p_socket_address_free (sock_addr);
+        PSocketAddress *local_addr = p_socket_get_local_address(*socket, NULL);
+        if (local_addr == NULL) {
+            p_socket_free (*socket);
+            p_uthread_exit (-1);
+            return 2;
+        }
+        sender_port = p_socket_address_get_port (local_addr);
+        p_socket_address_free (local_addr);
+    }
+    return 0;
+}
+
+int net_recv (PSocket *socket, char buffer[]) {
+    int len = 0;
+    for (int i=0; ; i++) {
+        psize recv_now = p_socket_receive (socket, 
+            buffer+i*1024, 1024, NULL);
+        if (recv_now == -1) {
+            if (len > 0) { 
+                printf("received (%d)\n", len);
+                return len; 
+            }
+            printf("error server recv\n");
+            return -1;
+        } else {
+            len += recv_now;
+            if (recv_now < 1024) { 
+                printf("received (%d)\n", len);
+                return len; 
+            }
+        }
+    }
+    printf("received (%d)\n", len);
+    return len;
+}
+
+
+void net_client_open(net_client *c) {
+    int err = net_open_socket(&c->socket, 0);
+    if (err > 0) printf("error opening client socket: %d\n", err);
+}
+
+int net_client_connect (net_client *c, char ip[], int port) {
+    c->addr_server = p_socket_address_new(ip, port);
+    if (c->addr_server == NULL) printf("error client making server addr\n");
+    int is_connected = p_socket_connect(c->socket, c->addr_server, NULL);
+    if (is_connected == FALSE) {
+        if (p_socket_io_condition_wait 
+            (c->socket, P_SOCKET_IO_CONDITION_POLLOUT, NULL) == TRUE &&
+            p_socket_check_connect_result (c->socket, NULL) == FALSE)
+        {
+            p_socket_address_free (c->addr_server);
+            p_socket_free (c->socket);
+            p_uthread_exit (-1);
+            printf("error client socket freed\n");
+        }
+    }
+
+    is_connected = p_socket_is_connected (c->socket);
+    if (is_connected == FALSE) {
+        printf("error client can't connect to %s:%d\n", ip, port);
+        return 1;
+    }
+    return 0;
+}
+
+void net_client_send (net_client *c, char data[], int sizeofdata) {
+    if(p_socket_is_connected (c->socket) == FALSE) { 
+        printf("error client send lost connection\n");
+    }
+    PError *err;
+    int send_now = p_socket_send (c->socket, data, sizeofdata, &err);
+    if (send_now == -1) printf("error: %d\n", p_error_get_code(err));
+}
+
+int net_client_recv (net_client *c, char buffer[]) {
+    net_recv(c->socket, buffer);
+}
+
+void net_client_close (net_client *c) {
+    p_socket_close (c->socket, NULL);
+    p_socket_address_free (c->addr_server);
+	p_socket_free (c->socket);
+}
+
+
+
+void net_server_open(net_server *s, int port) {
+    int err = net_open_socket(&s->socket, port);
+    if (err > 0) printf("error opening server socket: %d\n", err);
+    if (p_socket_listen (s->socket, NULL) == FALSE) {
+        printf("error server socket listen, nobody in.\n");
+        return;
+    }
+}
+
+void net_server_accept (net_server *s) {
+    pboolean poll = p_socket_io_condition_wait (s->socket, 
+        P_SOCKET_IO_CONDITION_POLLIN, NULL);
+    if (poll == FALSE) {
+        printf("server socket, no one in queue\n");
+        return;
+    }
+    s->sock_client = p_socket_accept (s->socket, NULL);
+    if (s->sock_client == NULL) {
+        printf("server socket, no accept\n");
+    }
+    p_socket_set_blocking (s->sock_client, FALSE);
+    printf("connected\n");
+}
+
+int net_server_recv (net_server *s, char buffer[]) {
+    net_recv(s->sock_client, buffer);
+}
+
+void net_server_send (net_server *s, char data[], int sizeofdata) {
+    if(p_socket_is_connected (s->sock_client) == FALSE) { 
+        printf("error client send lost connection\n");
+        return;
+    }
+    PError *err;
+    int send_now = p_socket_send (s->sock_client, data, sizeofdata, &err);
+    if (send_now == -1) printf("error: %d\n", p_error_get_code(err));
+}
+
+void net_server_close (net_server *s) {
+    p_socket_close (s->socket, NULL);
+    p_socket_free (s->sock_client);
+	p_socket_free(s->socket);
+}
diff --git a/net/net.h b/net/net.h
new file mode 100644
index 0000000..056114b
--- /dev/null
+++ b/net/net.h
@@ -0,0 +1,35 @@
+#ifndef NET_H
+#define NET_H
+
+#include <plibsys.h>
+
+#define SERVER_PORT 9999
+
+void itob (char *buf, int n);
+int btoi (char *buf);
+
+void net_init ();
+
+typedef struct {
+    PSocket *socket;
+    PSocketAddress * addr_server;
+} net_client;
+
+void net_client_open(net_client *c);
+int net_client_connect (net_client *c, char ip[], int port);
+int net_client_recv(net_client *c, char buffer[]);
+void net_client_send (net_client *c, char data[], int sizeofdata);
+void net_client_close (net_client *c);
+
+typedef struct {
+    PSocket *socket;
+    PSocket *sock_client;
+} net_server;
+
+void net_server_open(net_server *s, int port);
+void net_server_accept(net_server *s);
+int net_server_recv(net_server *s, char buffer[]);
+void net_server_send (net_server *s, char data[], int sizeofdata);
+void net_server_close (net_server *s);
+
+#endif
\ No newline at end of file