diff --git a/open_esplibs/libwpa/wpa_main.c b/open_esplibs/libwpa/wpa_main.c index c1133e6..28acb45 100644 --- a/open_esplibs/libwpa/wpa_main.c +++ b/open_esplibs/libwpa/wpa_main.c @@ -7,5 +7,117 @@ #if OPEN_LIBWPA_WPA_MAIN // The contents of this file are only built if OPEN_LIBWPA_WPA_MAIN is set to true +#include "espressif/user_interface.h" +#include "etstimer.h" +#include "espressif/osapi.h" +#include "esplibs/libnet80211.h" +#include "esplibs/libmain.h" +#include "esplibs/libwpa.h" +#include "esplibs/libpp.h" +#include "lwip/dhcp.h" + +static void wpa_callback1(struct pbuf* pb) { + struct netif *netif = sdk_g_ic.v.station_netif_info->netif; + sdk_ieee80211_output_pbuf(netif, pb); +} + +static void wpa_callback2(int arg0) { + struct sdk_g_ic_netif_info *netif_info = sdk_g_ic.v.station_netif_info; + sdk_ieee80211_send_mgmt(netif_info, 192, arg0); + sdk_ieee80211_sta_new_state(&sdk_g_ic, 2, (arg0 << 8) | 192); +} + +void sdk_wpa_config_profile(struct sdk_g_ic_st *g_ic) { + uint8_t v = g_ic->s._unknown209; + + if (v == 2 || v == 3 || v == 6) { + sdk_wpa_set_profile(1); + return; + } + + if (v == 4 || v == 5 || v == 7) + sdk_wpa_set_profile(2); +} + +void sdk_wpa_config_bss(struct sdk_g_ic_st *g_ic, uint8_t (* hwaddr2)[6]) { + struct sdk_g_ic_netif_info *netif_info = g_ic->v.station_netif_info; + struct netif *netif = netif_info->netif; + sdk_wpa_set_bss(netif->hwaddr, hwaddr2, g_ic->s._unknown20a, g_ic->s._unknown20c, + g_ic->s.sta_password, g_ic->s._unknown1e4.sta_ssid, + (g_ic->s._unknown1e4._unknown1e6 << 16) | g_ic->s._unknown1e4._unknown1e4); +} + +void sdk_wpa_config_assoc_ie(int arg0, int16_t *arg1, int32_t arg2) { + if (arg0 != 1) + sdk_g_ic.v._unknown178 = arg1; + else + sdk_g_ic.v._unknown174 = arg1; + + *arg1 = arg2; +} + +void sdk_dhcp_bind_check() { + struct sdk_g_ic_netif_info *netif_info = sdk_g_ic.v.station_netif_info; + uint8_t connect_status = netif_info->connect_status; + uint8_t unknown20a = sdk_g_ic.s._unknown20a; + + if (connect_status != 5) { + if (unknown20a == 7 || unknown20a == 8) { + netif_info->connect_status = 2; + } + } +} + +void sdk_eagle_auth_done() { + struct sdk_g_ic_netif_info *netif_info = sdk_g_ic.v.station_netif_info; + struct netif *netif = netif_info->netif; + struct sdk_netif_conninfo *conninfo = netif_info->_unknown88; + + if (conninfo->_unknown08 & 1) + return; + + uint32_t channel = conninfo->_unknown78->channel; + char *ssid = (char *)sdk_g_ic.s._unknown1e4.sta_ssid; + printf("\nconnected with %s, channel %d\n", ssid, channel); + + RTCMEM_SYSTEM[61] = 0x00010000 | channel; + + ETSTimer *timer = &netif_info->timer; + sdk_os_timer_disarm(timer); + sdk_os_timer_setfn(timer, sdk_dhcp_bind_check, 0); + sdk_os_timer_arm(timer, 15000, 0); + + netif_info->statusb9 = 0; + conninfo->_unknown18 = 0; + conninfo->_unknown08 |= 1; + + // TODO lwip v2 removed the NETIF_FLAG_DHCP flag. + if (netif->flags & 0x08) // NETIF_FLAG_DHCP + return; + + // lwip v2: if (ip4_addr_isany_val(netif->ip_addr)) { + if (netif->ip_addr.addr == 0) { + if (sdk_dhcpc_flag != DHCP_STOPPED) { + printf("dhcp client start...\n"); + dhcp_start(netif); + } + return; + } + + system_station_got_ip_set(&netif->ip_addr, &netif->netmask, &netif->gw); + netif_set_up(netif); +} + +void sdk_wpa_neg_complete() { + sdk_eagle_auth_done(); +} + +void sdk_wpa_attach(struct sdk_g_ic_st *g_ic) { + g_ic->v._unknown180 = NULL; + g_ic->v._unknown184 = &(g_ic->v._unknown180); + sdk_wpa_register(0, wpa_callback1, sdk_wpa_config_assoc_ie, sdk_ppInstallKey, + wpa_callback2, sdk_wpa_neg_complete); + sdk_ppRegisterTxCallback(sdk_eapol_txcb, 3); +} #endif /* OPEN_LIBWPA_WPA_MAIN */