1 files changed, 22 insertions, 2 deletions

diff --git a/core/network.c b/core/network.c
index 8c6fa7b6..7e3b344a 100644
--- a/core/network.c
+++ b/core/network.c
@@ -71,7 +71,7 @@ int sendpacket(IP_Port ip_port, uint8_t * data, uint32_t length)
    the packet data into data
    the packet length into length.
    dump all empty packets. */
-int receivepacket(IP_Port * ip_port, uint8_t * data, uint32_t * length)
+static int receivepacket(IP_Port * ip_port, uint8_t * data, uint32_t * length)
 {
     ADDR addr;
 #ifdef WIN32
@@ -88,6 +88,27 @@ int receivepacket(IP_Port * ip_port, uint8_t * data, uint32_t * length)
     return 0;
 }
 
+static packet_handler_callback packethandlers[256] = {0};
+
+void networking_registerhandler(uint8_t byte, packet_handler_callback cb)
+{
+    packethandlers[byte] = cb;
+}
+
+void networking_poll()
+{
+    IP_Port ip_port;
+    uint8_t data[MAX_UDP_PACKET_SIZE];
+    uint32_t length;
+    
+    while (receivepacket(&ip_port, data, &length) != -1)
+    {
+        if (length < 1) continue;
+        if (!packethandlers[data[0]]) continue;
+        packethandlers[data[0]](ip_port, data, length);
+    }
+}
+
 /* initialize networking
    bind to ip and port
    ip must be in network order EX: 127.0.0.1 = (7F000001)
@@ -149,7 +170,6 @@ int init_networking(IP ip, uint16_t port)
     bind(sock, (struct sockaddr*)&addr, sizeof(addr));
 
     return 0;
-
 }
 
 /* function to cleanup networking stuff */