1 files changed, 76 insertions, 0 deletions
diff --git a/code/fe310/eos/wifi.c b/code/fe310/eos/wifi.c
new file mode 100644
index 0000000..95fec7d
--- /dev/null
+++ b/code/fe310/eos/wifi.c
@@ -0,0 +1,76 @@
+#include <stdlib.h>
+#include <stdint.h>
+#include <string.h>
+
+#include "eos.h"
+#include "event.h"
+#include "net.h"
+
+#include "wifi.h"
+
+static eos_wifi_fptr_t wifi_handler[EOS_WIFI_MAX_MTYPE];
+static uint16_t wifi_handler_flags_buf_free = 0;
+static uint16_t wifi_handler_flags_buf_acq = 0;
+
+static void wifi_handler_evt(unsigned char type, unsigned char *buffer, uint16_t len) {
+    if ((buffer == NULL) || (len < 1)) {
+        eos_evtq_bad_handler(type, buffer, len);
+        eos_net_free(buffer, 0);
+        return;
+    }
+
+    uint8_t mtype = buffer[0];
+    if (mtype >= EOS_WIFI_MAX_MTYPE) {
+        eos_evtq_bad_handler(type, buffer, len);
+        eos_net_free(buffer, 0);
+        return;
+    }
+
+    uint16_t buf_free = ((uint16_t)1 << mtype) & wifi_handler_flags_buf_free;
+    uint16_t buf_acq = ((uint16_t)1 << mtype) & wifi_handler_flags_buf_acq;
+    if (buf_free) {
+        eos_net_free(buffer, buf_acq);
+        buffer = NULL;
+        len = 0;
+    }
+
+    wifi_handler[mtype](buffer, len);
+
+    if (buf_free && buf_acq) eos_net_release();
+}
+
+void eos_wifi_init(void) {
+    eos_net_set_handler(EOS_NET_MTYPE_WIFI, wifi_handler_evt, 0);
+}
+
+void eos_wifi_connect(const char *ssid, const char *pass) {
+    int ssid_len = strlen(ssid);
+    int pass_len = strlen(pass);
+    unsigned char *buffer = eos_net_alloc(0);
+
+    buffer[0] = EOS_WIFI_MTYPE_CONNECT;
+    strcpy(buffer+1, ssid);
+    buffer[ssid_len+1] = 0;
+    strcpy(buffer+ssid_len+2, pass);
+    buffer[ssid_len+pass_len+2] = 0;
+    eos_net_send(EOS_NET_MTYPE_WIFI, buffer, ssid_len+pass_len+3, 0);
+}
+
+void eos_wifi_disconnect(void) {
+    unsigned char *buffer = eos_net_alloc(0);
+    buffer[0] = EOS_WIFI_MTYPE_DISCONNECT;
+    eos_net_send(EOS_NET_MTYPE_WIFI, buffer, 1, 0);
+}
+
+void eos_wifi_set_handler(int mtype, eos_wifi_fptr_t handler, uint8_t flags) {
+    if (mtype >= EOS_WIFI_MAX_MTYPE) {
+        return;
+    }
+
+    if (flags) {
+        uint16_t flag = (uint16_t)1 << mtype;
+        if (flags & EOS_NET_FLAG_BUF_FREE) wifi_handler_flags_buf_free |= flag;
+        if (flags & EOS_NET_FLAG_BUF_ACQ) wifi_handler_flags_buf_acq |= flag;
+    }
+    wifi_handler[mtype] = handler;
+}