Starting to add suport for server and AP mode

src/main.c

@@ -5,10 +5,29 @@
 #include <lwip/udp.h>
 #include <time.h>
 
+#define CLIENT
+//#define SERVER
+#define STA_MODE
+//#define AP_MODE
+
+#ifdef STA_MODE
+
 #define SSID "thinkpad-T440p"
 #define PASSWORD "CDuKaka2000!"
 
-#define PORT 4243
+#endif // STA_MODE
+
+#ifdef AP_MODE
+
+#define AP_SSID "PicoAP"
+#define AP_PASSWORD "PicoPassword!"
+#define AP_CHANNEL 3
+
+#endif // AP_MODE
+
+#ifdef CLIENT
+
+#define CLIENT_PORT 4243
 
 static void udp_recv_callback(void *arg, struct udp_pcb *pcb, struct pbuf *p, const ip_addr_t *addr, u16_t port)
 {
@@ -21,6 +40,8 @@ static void udp_recv_callback(void *arg, struct udp_pcb *pcb, struct pbuf *p, co
     pbuf_free(p);
 }
 
+#endif // CLIENT
+
 int main()
 {
     stdio_init_all();
@@ -35,14 +56,21 @@ int main()
         return -1;
     }
 
-    cyw43_arch_enable_sta_mode();
-    cyw43_wifi_pm(&cyw43_state, CYW43_NO_POWERSAVE_MODE);
+    if(cyw43_wifi_pm(&cyw43_state, CYW43_NO_POWERSAVE_MODE))
+    {
+        puts("CYW43 no powersave mode failed");
+        return -1;
+    }
 
     puts("CYW43 arch initialized");
 
+#ifdef STA_MODE
+
+    cyw43_arch_enable_sta_mode();
+
     for(int error_code = 1; error_code; )
     {
-        puts("WiFi connection attempt");
+        printf("WiFi connection attempt to \"%s\"\n", SSID);
 
         error_code = cyw43_arch_wifi_connect_timeout_ms(SSID, PASSWORD, CYW43_AUTH_WPA2_AES_PSK, 10000);
 
@@ -71,6 +99,20 @@ int main()
 
     puts("WiFi connected");
 
+#endif // STA_MODE
+
+#ifdef AP_MODE
+
+    cyw43_arch_enable_ap_mode(AP_SSID, AP_PASSWORD, CYW43_AUTH_WPA2_AES_PSK);
+
+    cyw43_wifi_ap_set_channel(&cyw43_state, AP_CHANNEL);
+
+    puts("AP mode enabled");
+
+#endif //AP_MODE
+
+#ifdef CLIENT
+
     struct udp_pcb *pcb = udp_new();
 
     if(!pcb)
@@ -81,13 +123,20 @@ int main()
 
     udp_recv(pcb, udp_recv_callback, NULL);
 
-    udp_bind(pcb, IP4_ADDR_ANY, PORT);
+    if(udp_bind(pcb, IP4_ADDR_ANY, CLIENT_PORT))
+    {
+        puts("UDP pcb bind failed");
+        return -1;
+    }
+
+#endif // CLIENT
 
     // ========================== Main loop ==========================
     while(true)
     {
         cyw43_arch_poll();
 
+#ifdef STA_MODE
         int ret = cyw43_wifi_link_status(&cyw43_state, CYW43_ITF_STA);
         if(ret != CYW43_LINK_JOIN)
         {
@@ -116,6 +165,7 @@ int main()
             }
             puts(status_message);
         }
+#endif // STA_MODE
 
         tight_loop_contents();
     }