#include "luat_base.h"
#include "luat_malloc.h"
#include "luat_msgbus.h"
#include "luat_wlan.h"


#include <string.h>
#include "wm_irq.h"
#include "tls_sys.h"
#include "wm_ram_config.h"
#include "wm_efuse.h"
#include "wm_regs.h"
#include "wm_wifi.h"
#include "wm_netif.h"
#include "wm_debug.h"
#include "wm_internal_flash.h"

#define LUAT_LOG_TAG "wlan"
#include "luat_log.h"

#include "lwip/netif.h"
#include "luat_network_adapter.h"
#include "luat_timer.h"
#include "net_lwip2.h"
#include "lwip/tcp.h"

void net_lwip2_set_link_state(uint8_t adapter_index, uint8_t updown);
int luat_wlan_raw_in(const u8 *bssid, u8 *buf, u32 buf_len);
extern u8 *hostapd_get_mac(void);

#define SCAN_DONE       (0x73)
#define ONESHOT_RESULT  (0x74)

static int wlan_init;
static int wlan_state;

char luat_sta_hostname[32];

extern const char WiFiVer[];
extern u8 tx_gain_group[];
extern void *tls_wl_init(u8 *tx_gain, u8 *mac_addr, u8 *hwver);
extern int wpa_supplicant_init(u8 *mac_addr);
extern void tls_pmu_chipsleep_callback(int sleeptime);

static int l_wlan_cb(lua_State*L, void* ptr) {
    (void)ptr;
    u8 ssid[33]= {0};
    u8 pwd[65] = {0};
    char sta_ip[16] = {0};
    rtos_msg_t* msg = (rtos_msg_t*)lua_topointer(L, -1);
    // 然后再发出IP事件, 如果启用wlan的话
    lua_getglobal(L, "sys_pub");
    switch (msg->arg1)
    {
    case NETIF_IP_NET_UP:
        #ifdef LUAT_USE_NETWORK
        net_lwip2_set_link_state(NW_ADAPTER_INDEX_LWIP_WIFI_STA, 1);
        luat_wlan_get_ip(0, sta_ip);
        LLOGD("sta ip %s", sta_ip);
        lua_pushstring(L, "IP_READY");
        lua_pushstring(L, sta_ip);
        lua_pushinteger(L, NW_ADAPTER_INDEX_LWIP_WIFI_STA);
        lua_call(L, 3, 0);
        #endif
        break;
    case NETIF_WIFI_DISCONNECTED:
        #ifdef LUAT_USE_NETWORK
        net_lwip2_set_link_state(NW_ADAPTER_INDEX_LWIP_WIFI_STA, 0);
        lua_pushstring(L, "IP_LOSE");
        lua_call(L, 1, 0); // 暂时只发个IP_LOSE
        #endif
        break;
    case SCAN_DONE :
        lua_pushstring(L, "WLAN_SCAN_DONE");
        lua_call(L, 1, 0);
        break;
    case ONESHOT_RESULT:
        #ifdef LUAT_USE_NETWORK
        lua_pushstring(L, "SC_RESULT");
        tls_wifi_get_oneshot_ssidpwd(ssid, pwd);
        LLOGD("oneshot %s %s", ssid, pwd);
        lua_pushstring(L, (const char*)ssid);
        lua_pushstring(L, (const char*)pwd);
        lua_call(L, 3, 0);
        #endif
        break;
    case NETIF_WIFI_JOIN_FAILED:
    case NETIF_WIFI_JOIN_SUCCESS:
        lua_pushstring(L, "WLAN_STATUS");
        lua_pushinteger(L, wlan_state);
        lua_call(L, 2, 0);
        break;
    }
    return 0;
}

static void netif_event_cb(u8 status) {
    rtos_msg_t msg = {0};
    struct tls_param_ip ip_param;
    // LLOGD("netif_event %d", status);
    msg.handler = l_wlan_cb;
    msg.arg1 = status;
	switch (status)
    {
    case NETIF_WIFI_JOIN_FAILED:
        LLOGI("join failed");
        #ifndef LUAT_USE_NETWORK
        tls_auto_reconnect(0);
        #endif
        wlan_state = 0;
        msg.arg1 = status;
        luat_msgbus_put(&msg, 0);
        break;
    case NETIF_WIFI_DISCONNECTED:
        wlan_state = 0;
        LLOGI("disconnected");
        msg.arg1 = status;
        luat_msgbus_put(&msg, 0);
        #ifndef LUAT_USE_NETWORK
        tls_auto_reconnect(0);
        #endif
        break;
    case NETIF_WIFI_JOIN_SUCCESS :
        wlan_state = 1;
        LLOGI("join success");
        #ifdef LUAT_USE_NETWORK
        tls_param_get(TLS_PARAM_ID_IP, (void *)&ip_param, false);
        if (!ip_param.dhcp_enable) {
            LLOGI("dhcp is disable, so 'join success' as 'IP_READY'");
        }
        #endif
        luat_msgbus_put(&msg, 0);
        break;
    case NETIF_IP_NET_UP :
        #ifdef LUAT_USE_NETWORK
        LLOGI("IP READY");
        msg.arg1 = status;
        luat_msgbus_put(&msg, 0);
        #endif
        break;
#if TLS_CONFIG_AP
    case NETIF_WIFI_SOFTAP_SUCCESS :
        LLOGI("softap create success");
        #ifdef LUAT_USE_NETWORK
        net_lwip2_set_link_state(NW_ADAPTER_INDEX_LWIP_WIFI_AP, 1);
        #endif
        break;
    case NETIF_WIFI_SOFTAP_FAILED:
        LLOGI("softap create failed");
        #ifdef LUAT_USE_NETWORK
        net_lwip2_set_link_state(NW_ADAPTER_INDEX_LWIP_WIFI_AP, 0);
        #endif
        break;
    case NETIF_WIFI_SOFTAP_CLOSED:
        LLOGI("softap create closed");
        #ifdef LUAT_USE_NETWORK
        net_lwip2_set_link_state(NW_ADAPTER_INDEX_LWIP_WIFI_AP, 0);
        #endif
        break;
    case NETIF_IP_NET2_UP :
        LLOGI("softap netif up");
        break;
#endif
    default:
        break;
    }
}


int luat_wlan_init(luat_wlan_config_t *conf) {
    (void)conf;
    if (wlan_init == 0) {
        //------------------------------------
        //u8 enable = 0;
        u8 mac_addr[6];
        tls_get_tx_gain(&tx_gain_group[0]);
        TLS_DBGPRT_INFO("tx gain ");
        TLS_DBGPRT_DUMP((char *)(&tx_gain_group[0]), 27);
        // 计算wifi内存大小的公式 (TX+RX)*2*1638+4096
        // 例如 TX=6, RX=4, (6+4)*2*1638+4096=36856, 向上取整, 36k
        // 然后sram的最后位置是 0x20047FFF
        if (tls_wifi_mem_cfg((0x20048000u - (36*1024)), 6, 4)) /*wifi tx&rx mem customized interface*/
        {
            TLS_DBGPRT_INFO("wl mem initial failured\n");
        }

        luat_wlan_get_mac(0, mac_addr);
        TLS_DBGPRT_INFO("mac addr ");
        TLS_DBGPRT_DUMP((char *)(&mac_addr[0]), 6);
        if(tls_wl_init(NULL, &mac_addr[0], NULL) == NULL)
        {
            TLS_DBGPRT_INFO("wl driver initial failured\n");
        }
        if (wpa_supplicant_init(mac_addr))
        {
            TLS_DBGPRT_INFO("supplicant initial failured\n");
        }
        // 设置到AP的mac地址位置去
        luat_wlan_get_mac(1, mac_addr);
        memcpy(hostapd_get_mac(), mac_addr, 6);
        
    	/*wifi-temperature compensation,default:open*/
        if (tls_wifi_get_tempcomp_flag() != 0)
    	    tls_wifi_set_tempcomp_flag(0);
        if (tls_wifi_get_psm_chipsleep_flag() != 0)
    	    tls_wifi_set_psm_chipsleep_flag(0);
    	tls_wifi_psm_chipsleep_cb_register((tls_wifi_psm_chipsleep_callback)tls_pmu_chipsleep_callback, NULL, NULL);
    
        u8 wireless_protocol = 0;
        tls_param_get(TLS_PARAM_ID_WPROTOCOL, (void *) &wireless_protocol, TRUE);
        // LLOGD("wireless_protocol %d", wireless_protocol);
        if (TLS_PARAM_IEEE80211_INFRA != wireless_protocol)
        {
            wireless_protocol = TLS_PARAM_IEEE80211_INFRA;
            tls_param_set(TLS_PARAM_ID_WPROTOCOL, (void *) &wireless_protocol, FALSE);
        }
        // tls_wifi_enable_log(1);
        luat_wlan_get_hostname(0); // 调用一下就行
        wlan_init = 1;

        #ifdef LUAT_USE_ZLINK_WLAN
        extern void luat_zlink_wlan_init(void);
        luat_zlink_wlan_init();
        #else
        #ifdef LUAT_USE_NETWORK
        tls_ethernet_init();
        tls_sys_init();
        tls_netif_add_status_event(netif_event_cb);
        #else
        tls_wifi_status_change_cb_register(netif_event_cb);
        #ifdef LUAT_USE_WLAN_RAW
        tls_ethernet_data_rx_callback(luat_wlan_raw_in);
        #endif
        #endif
        tls_wifi_scan_result_cb_register(NULL);
        #endif

        //-----------------------------------
        #ifdef LUAT_USE_NETWORK
        struct netif *et0 = tls_get_netif();
        net_lwip2_set_netif(NW_ADAPTER_INDEX_LWIP_WIFI_STA, et0);
        #if TLS_CONFIG_AP
        extern struct netif *nif4apsta;
        if (nif4apsta) {
            net_lwip2_set_netif(NW_ADAPTER_INDEX_LWIP_WIFI_AP, nif4apsta);
            net_lwip2_register_adapter(NW_ADAPTER_INDEX_LWIP_WIFI_AP);
        }
        #endif
        net_lwip2_register_adapter(NW_ADAPTER_INDEX_LWIP_WIFI_STA);

        // 确保DHCP是默认开启
        struct tls_param_ip ip_param;
        tls_param_get(TLS_PARAM_ID_IP, (void *)&ip_param, false);
        if (!ip_param.dhcp_enable) {
            ip_param.dhcp_enable = 1;
            tls_param_set(TLS_PARAM_ID_IP, (void *)&ip_param, false);
        }
        #endif
    }
    tls_wifi_set_psflag(FALSE, FALSE);
	return 0;
}

int luat_wlan_mode(luat_wlan_config_t *conf) {
    (void)conf;
    #if TLS_CONFIG_AP
    // 不需要设置, 反正都能用
    if (conf->mode == LUAT_WLAN_MODE_STA) {
        tls_wifi_softap_destroy();
    }
    // else if (conf->mode == LUAT_WLAN_MODE_AP) {
    //     tls_wifi_disconnect();
    // }
    #endif
    return 0;
}

int luat_wlan_ready(void) {
    return wlan_state;
}

int luat_wlan_connect(luat_wlan_conninfo_t* info) {
    int ret = 0;
    if (info->bssid[5]) {
        if (strlen(info->ssid) > 0) {
            ret = tls_wifi_connect_by_ssid_bssid((u8*)info->ssid, strlen(info->ssid), (u8*)info->bssid, (u8*)info->password, strlen(info->password));
        }
        else {
            ret = tls_wifi_connect_by_bssid((u8*)info->bssid, (u8*)info->password, strlen(info->password));
        }
    }
    else {
        ret = tls_wifi_connect((u8*)info->ssid, strlen(info->ssid), (u8*)info->password, strlen(info->password));
    }
    u8 opt = WIFI_AUTO_CNT_FLAG_SET;
    u8 mode = WIFI_AUTO_CNT_ON;
    tls_wifi_auto_connect_flag(opt, &mode);
    return ret;
}

int luat_wlan_disconnect(void) {
    u8 opt = WIFI_AUTO_CNT_FLAG_SET;
    u8 mode = WIFI_AUTO_CNT_OFF;
    tls_wifi_auto_connect_flag(opt, &mode);
    tls_wifi_disconnect();
    return 0;
}

static void scan_event_cb(void *ptmr, void *parg) {
    (void)ptmr;
    (void)parg;
    rtos_msg_t msg = {0};
    msg.handler = l_wlan_cb;
    msg.arg1 = SCAN_DONE;
    luat_msgbus_put(&msg, 0);
}

int luat_wlan_scan(void) {
    int ret = tls_wifi_scan();
    if (ret)
        LLOGD("tls_wifi_scan %d", ret);
    static tls_os_timer_t *scan_timer = NULL;
	tls_os_timer_create(&scan_timer, scan_event_cb, NULL, 3000, 0, NULL);
	tls_os_timer_start(scan_timer);
    return ret;
}

int luat_wlan_scan_get_result(luat_wlan_scan_result_t *results, size_t ap_limit) {
    size_t buffsize = ap_limit * 48 + 8;
    u8* buff = luat_heap_malloc(buffsize);
    if (buff == NULL)
        return 0;
    memset(buff, 0, buffsize);
    tls_wifi_get_scan_rslt((u8*)buff, buffsize);
    struct tls_scan_bss_t *wsr = (struct tls_scan_bss_t*)buff;
    struct tls_bss_info_t *bss_info;
    if (ap_limit > wsr->count)
        ap_limit = wsr->count;
    bss_info = wsr->bss;
    for (size_t i = 0; i < ap_limit; i++)
    {
        memset(results[i].ssid, 0, 33);
        memcpy(results[i].ssid, bss_info[i].ssid, bss_info[i].ssid_len);
        memcpy(results[i].bssid, bss_info[i].bssid, ETH_ALEN);
        results[i].rssi = bss_info[i].rssi;
        results[i].ch = bss_info[i].channel;
        // break;
    }
    luat_heap_free(buff);
    return ap_limit;
}

void luat_sc_callback(enum tls_wifi_oneshot_result_type type) {
    if (type == WM_WIFI_ONESHOT_TYPE_SSIDPWD) {
        LLOGD("oneshot Got!!");
        rtos_msg_t msg = {.handler = l_wlan_cb, .arg1=ONESHOT_RESULT};
        luat_msgbus_put(&msg, 0);
    }
}

extern u8 gucssidData[];
extern u8 gucpwdData[];
int luat_wlan_smartconfig_start(int tp) {
    (void)tp;
    #ifdef LUAT_USE_NETWORK
    gucssidData[0] = 0;
    gucpwdData[0] = 0;
    tls_wifi_oneshot_result_cb_register(luat_sc_callback);
    return tls_wifi_set_oneshot_flag(1);
    #else
    return -1;
    #endif
}

int luat_wlan_smartconfig_stop(void) {
    #ifdef LUAT_USE_NETWORK
    return tls_wifi_set_oneshot_flag(0);
    #else
    return -1;
    #endif
}

// 数据类
int luat_wlan_get_mac(int id, char* mac) {
    if (id == 0)
        tls_ft_param_get(CMD_WIFI_MAC, mac, 6);
    else
        tls_ft_param_get(CMD_WIFI_MACAP, mac, 6);
    return 0;
}

int luat_wlan_set_mac(int id, const char* mac_addr) {
    u8 mac[8] = {0};
    int ret = 0;
    if (id == 0) {
        luat_wlan_get_mac(0, mac);
    }
    else {
        luat_wlan_get_mac(1, mac);
    }
    if (memcmp(mac_addr, mac, 6) == 0) {
        // 完全相同, 不需要设置
        return 0;
    }
    if (id == 0) {
        ret = tls_ft_param_set(CMD_WIFI_MAC, (void*)mac_addr, 6);
    }
    else {
        ret = tls_ft_param_set(CMD_WIFI_MACAP, (void*)mac_addr, 6);
    }
    return ret;
}

int luat_wlan_get_ip(int type, char* data) {
    (void)type;
    #ifdef LUAT_USE_NETWORK
    struct netif *et0 = tls_get_netif();
    if (et0 == NULL || ip_addr_isany(&et0->ip_addr))
        return -1;
    ipaddr_ntoa_r(&et0->ip_addr, data, 16);
    return 0;
    #else
    return -1;
    #endif
}

// 设置和获取省电模式
int luat_wlan_set_ps(int mode) {
    tls_wifi_set_psm_chipsleep_flag(mode == 0 ? 0 : 1);
    return 0;
}

int luat_wlan_get_ps(void) {
    return tls_wifi_get_psm_chipsleep_flag();
}

int luat_wlan_get_ap_bssid(char* buff) {
    struct tls_curr_bss_t bss = {0};
    tls_wifi_get_current_bss(&bss);
    memcpy(buff, bss.bssid, ETH_ALEN);
    return 0;
}

int luat_wlan_get_ap_rssi(void) {
    struct tls_curr_bss_t bss = {0};
    tls_wifi_get_current_bss(&bss);
    return bss.rssi;
}

int luat_wlan_get_ap_gateway(char* buff) {
    #ifdef LUAT_USE_NETWORK
    struct netif *et0 = tls_get_netif();
    if (et0 == NULL || ip_addr_isany(&et0->gw))
        return -1;
    ipaddr_ntoa_r(&et0->gw, buff, 16);
    #else
    buff[0] = 0;
    #endif
    return 0;
}

// AP类
int luat_wlan_ap_start(luat_wlan_apinfo_t *apinfo2) {
    #if TLS_CONFIG_AP
    int ret = 0;
    struct tls_softap_info_t apinfo = {0};
    struct tls_ip_info_t ipinfo = {
        .ip_addr = {192, 168, 4, 1},
        .netmask = {255, 255, 255, 0}
    };
    if (apinfo2->gateway[0]) {
        memcpy(ipinfo.ip_addr, apinfo2->gateway, 4);
    }
    if (apinfo2->netmask[0]) {
        memcpy(ipinfo.netmask, apinfo2->netmask, 4);
    }
    memcpy(apinfo.ssid, apinfo2->ssid, strlen(apinfo2->ssid) + 1);
    if (strlen(apinfo2->password)) {
        apinfo.keyinfo.format = 1;
        apinfo.keyinfo.key_len = strlen(apinfo2->password);
        apinfo.keyinfo.index = 1;
        memcpy(apinfo.keyinfo.key, apinfo2->password, strlen(apinfo2->password)+1);
        apinfo.encrypt = IEEE80211_ENCRYT_CCMP_WPA2;
    }
    else {
        apinfo.encrypt = IEEE80211_ENCRYT_NONE;
    }
    if (apinfo2->channel > 0)
        apinfo.channel = apinfo2->channel;
    else {
        apinfo.channel = 6;
    }
    LLOGD("AP GW %d.%d.%d.%d MASK %d.%d.%d.%d", 
                ipinfo.ip_addr[0],ipinfo.ip_addr[1],ipinfo.ip_addr[2],ipinfo.ip_addr[3],
                ipinfo.netmask[0],ipinfo.netmask[1],ipinfo.netmask[2],ipinfo.netmask[3]
    );
    // ----------------------------
    u8 mac[6] = {0};
    luat_wlan_get_mac(1, mac);
    sprintf_((char*)ipinfo.dnsname, "LUATOS_%02X%02X%02X%02X%02X%02X", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
    //------------------------------

    u8 value = 0;
    tls_param_get(TLS_PARAM_ID_WPROTOCOL, (void *) &value, TRUE);
    // LLOGD("wireless_protocol %d", wireless_protocol);
    if (TLS_PARAM_IEEE80211_SOFTAP != value)
    {
        value = TLS_PARAM_IEEE80211_SOFTAP;
        tls_param_set(TLS_PARAM_ID_WPROTOCOL, (void *) &value, FALSE);
    }
    if (apinfo2->hidden) {
        value = 0;
        tls_param_set(TLS_PARAM_ID_BRDSSID, (void *) &value, FALSE);
    }
    else {
        value = 1;
        tls_param_set(TLS_PARAM_ID_BRDSSID, (void *) &value, FALSE);
    }
    
    ret = tls_wifi_softap_create(&apinfo, &ipinfo);
    if (ret)
        LLOGD("tls_wifi_softap_create %d", ret);
    return ret;
    #else
    return -1;
    #endif
}

extern u8 *wpa_supplicant_get_mac(void);
const char* luat_wlan_get_hostname(int id) {
    (void)id;
    if (luat_sta_hostname[0] == 0) {
        u8* mac_addr = wpa_supplicant_get_mac();
        sprintf_(luat_sta_hostname, "LUATOS_%02X%02X%02X%02X%02X%02X", mac_addr[0], mac_addr[1], mac_addr[2], mac_addr[3], mac_addr[4], mac_addr[5]);
    }
    return (const char*)luat_sta_hostname;
}

int luat_wlan_set_hostname(int id, const char* hostname) {
    (void)id;
    if (hostname == NULL || hostname[0] == 0) {
        return 0;
    }
    memcpy(luat_sta_hostname, hostname, strlen(hostname) + 1);
    return 0;
}

int luat_wlan_ap_stop(void) {
    #if TLS_CONFIG_AP
    tls_wifi_softap_destroy();
    #endif
    return 0;
}

int luat_wlan_set_station_ip(luat_wlan_station_info_t *info) {
    #ifdef LUAT_USE_NETWORK
    struct tls_ethif *ethif;
    struct tls_param_ip param_ip;
    if (info == NULL) {
        return -1;
    }
    ethif=tls_netif_get_ethif();
	if (ethif == NULL)
	{
        LLOGE("call wlan.init() first!!");
		return -1;
	}

    if (info->dhcp_enable) {
        LLOGD("dhcp enable");
    }
    else {
        LLOGD("dhcp disable");
        LLOGD("Sta IP %d.%d.%d.%d MASK %d.%d.%d.%d GW %d.%d.%d.%d", 
                info->ipv4_addr[0],   info->ipv4_addr[1],   info->ipv4_addr[2],   info->ipv4_addr[3],
                info->ipv4_netmask[0],info->ipv4_netmask[1],info->ipv4_netmask[2],info->ipv4_netmask[3],
                info->ipv4_gateway[0],info->ipv4_gateway[1],info->ipv4_gateway[2],info->ipv4_gateway[3]
        );
    }

	if (WM_WIFI_JOINED == tls_wifi_get_state())
	{
	    if (info->dhcp_enable) {
	        /* enable dhcp */
	        tls_dhcp_start();
	    } else {
	        tls_dhcp_stop();

			MEMCPY((char *)ip_2_ip4(&ethif->ip_addr) , info->ipv4_addr, 4);
            // MEMCPY((char *)ip_2_ip4(&ethif->dns1), &params->dns1, 4);
	        MEMCPY((char *)ip_2_ip4(&ethif->netmask), info->ipv4_netmask, 4);
	        MEMCPY((char *)ip_2_ip4(&ethif->gw), info->ipv4_gateway, 4);
	        tls_netif_set_addr(&ethif->ip_addr, &ethif->netmask, &ethif->gw);
	    }
	}

    /* update flash params */
    param_ip.dhcp_enable = info->dhcp_enable;
    MEMCPY((char *)param_ip.gateway, info->ipv4_gateway, 4);
    MEMCPY((char *)param_ip.ip, info->ipv4_addr, 4);
    MEMCPY((char *)param_ip.netmask, info->ipv4_netmask, 4);
    tls_param_set(TLS_PARAM_ID_IP, (void *)&param_ip, false);
    #endif
    return 0;
}

// MAC地址修正逻辑


extern const u8 default_mac[];

static int is_mac_ok(u8 mac_addr[6]) {

	// 如果是默认值, 那就是不合法
	if (!memcmp(mac_addr, default_mac, 6)) {
        return 0;
    }
	// 如果任意一位是0x00或者0xFF,那就是不合法
    for (size_t i = 0; i < 6; i++)
    {
        if (mac_addr[i] == 0 || mac_addr[i] == 0xFF) {
            return 0;
        }
    }
	return 1;
}

void sys_mac_init() {
	u8 tmp_mac[6] = {0};
    u8 mac_addr[6] = {0};
	u8 ap_mac[6] = {0};
    unsigned  char unique_id [20] = {0};
    tls_fls_read_unique_id(unique_id);
	int ret = 0;

#ifdef LUAT_USE_NIMBLE
	// 读蓝牙mac, 如果是默认值,就根据unique_id读取
	uint8_t bt_mac[6];
	// 缺省mac C0:25:08:09:01:10
	tls_get_bt_mac_addr(bt_mac);
	if (!memcmp(bt_mac, default_mac, 6)) { // 看来是默认MAC, 那就改一下吧
		if (unique_id[1] == 0x10){
			memcpy(bt_mac, unique_id + 10, 6);
		}
		else {
			memcpy(bt_mac, unique_id + 2, 6);
		}
		tls_set_bt_mac_addr(bt_mac);
	}
	LLOGD("BLE_4.2 %02X:%02X:%02X:%02X:%02X:%02X", bt_mac[0], bt_mac[1], bt_mac[2], bt_mac[3], bt_mac[4], bt_mac[5]);
#endif

#ifdef LUAT_USE_WLAN
    luat_wlan_get_mac(0, mac_addr);
	luat_wlan_get_mac(1, ap_mac);
	//printf("CMD_WIFI_MAC ret %d %02X%02X%02X%02X%02X%02X\n", ret, mac_addr[0], mac_addr[1], mac_addr[2], mac_addr[3], mac_addr[4], mac_addr[5]);
	int sta_mac_ok = is_mac_ok(mac_addr);
	int ap_mac_ok = is_mac_ok(ap_mac);

	// 其中一个合法呢?
	if (sta_mac_ok && !ap_mac_ok) {
		printf("sta地址合法, ap地址不合法\n");
		memcpy(ap_mac, mac_addr, 6);
	}
	else if (!sta_mac_ok && ap_mac_ok) {
		printf("sta地址不合法, ap地址合法\n");
		memcpy(mac_addr, ap_mac, 6);
	}
	else if (!sta_mac_ok && !ap_mac_ok){ // 均不合法, 那就用uniqid
		printf("sta地址不合法, ap地址不合法\n");
		if (unique_id[1] == 0x10){
			memcpy(mac_addr, unique_id + 12, 6);
		}
		else {
			memcpy(mac_addr, unique_id + 4, 6);
		}
        mac_addr[0] = 0x0C;
        for (size_t i = 1; i < 6; i++)
        {
            if (mac_addr[i] == 0 || mac_addr[i] == 0xFF) {
                mac_addr[i] = 0x01;
            }
        }
		memcpy(ap_mac, mac_addr, 6);
	}

	if (!memcmp(mac_addr, ap_mac, 6)) {
		printf("sta与ap地址相同, 调整ap地址\n");
		// 完全相同, 那就改一下ap的地址
		if (ap_mac[5] == 0x01 || ap_mac[5] == 0xFE) {
			ap_mac[5] = 0x02;
		}
		else {
			ap_mac[5] += 1;
		}
	}

	// 全部mac都得到了, 然后重新读一下mac, 不相同的就重新写入
	
	// 先判断sta的mac
    luat_wlan_get_mac(0, tmp_mac);
	if (memcmp(tmp_mac, mac_addr, 6)) {
		// 不一致, 写入
		tls_ft_param_set(CMD_WIFI_MAC, mac_addr, 6);
		printf("更新STA mac %02X%02X%02X%02X%02X%02X\n", mac_addr[0], mac_addr[1], mac_addr[2], mac_addr[3], mac_addr[4], mac_addr[5]);
	}

	// 再判断ap的mac
    luat_wlan_get_mac(1, tmp_mac);
	if (memcmp(tmp_mac, ap_mac, 6)) {
		// 不一致, 写入
		tls_ft_param_set(CMD_WIFI_MACAP, ap_mac, 6);
		printf("更新AP mac %02X%02X%02X%02X%02X%02X\n", ap_mac[0], ap_mac[1], ap_mac[2], ap_mac[3], ap_mac[4], ap_mac[5]);
	}
#endif
}