173 lines
3.4 KiB
C
173 lines
3.4 KiB
C
// Necessary includes from the Pico SDK
|
|
#include <pico/stdio.h>
|
|
#include <stdio.h>
|
|
#include <pico/cyw43_arch.h>
|
|
#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!"
|
|
|
|
#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)
|
|
{
|
|
puts("Callback");
|
|
|
|
if(p->len == 1)
|
|
{
|
|
printf("a:%d\n", ((uint8_t *)p->payload)[0]);
|
|
}
|
|
pbuf_free(p);
|
|
}
|
|
|
|
#endif // CLIENT
|
|
|
|
int main()
|
|
{
|
|
stdio_init_all();
|
|
|
|
sleep_ms(2000);
|
|
|
|
puts("Pi Starting...");
|
|
|
|
if(cyw43_arch_init_with_country(CYW43_COUNTRY_FRANCE))
|
|
{
|
|
puts("CYW43 arch init failed");
|
|
return -1;
|
|
}
|
|
|
|
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; )
|
|
{
|
|
printf("WiFi connection attempt to \"%s\"\n", SSID);
|
|
|
|
error_code = cyw43_arch_wifi_connect_timeout_ms(SSID, PASSWORD, CYW43_AUTH_WPA2_AES_PSK, 10000);
|
|
|
|
if(error_code)
|
|
{
|
|
const char *error_message;
|
|
|
|
switch(error_code)
|
|
{
|
|
case PICO_ERROR_TIMEOUT:
|
|
error_message = "Timeout reached";
|
|
break;
|
|
|
|
case PICO_ERROR_BADAUTH:
|
|
error_message = "Bad password";
|
|
break;
|
|
|
|
case PICO_ERROR_CONNECT_FAILED:
|
|
error_message = "Some reason";
|
|
break;
|
|
}
|
|
|
|
printf("Connection failed : %s\n", error_message);
|
|
}
|
|
}
|
|
|
|
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)
|
|
{
|
|
puts("UDP pcb init failed");
|
|
return -1;
|
|
}
|
|
|
|
udp_recv(pcb, udp_recv_callback, NULL);
|
|
|
|
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)
|
|
{
|
|
const char *status_message;
|
|
switch(ret)
|
|
{
|
|
case CYW43_LINK_DOWN:
|
|
status_message = "WiFi down";
|
|
break;
|
|
|
|
case CYW43_LINK_JOIN:
|
|
status_message = "WiFi up";
|
|
break;
|
|
|
|
case CYW43_LINK_FAIL:
|
|
status_message = "WiFi connection failed";
|
|
break;
|
|
|
|
case CYW43_LINK_NONET:
|
|
status_message = "Network not found";
|
|
break;
|
|
|
|
case CYW43_LINK_BADAUTH:
|
|
status_message = "Bad auth";
|
|
break;
|
|
}
|
|
puts(status_message);
|
|
}
|
|
#endif // STA_MODE
|
|
|
|
tight_loop_contents();
|
|
}
|
|
}
|