WiFi 协议栈与配网

static u8 buf[1 * 1024] __attribute__((aligned(32))); //用于串口接收缓存数据的循环buf
static void uart_test_main(void *priv)
{
    char recv_buf[256];
    int len;
    void *hdl = dev_open("uart1", NULL);
    ...
    /* 1 . 设置串口接收缓存数据的循环buf地址 */
    dev_ioctl(hdl, IOCTL_UART_SET_CIRCULAR_BUFF_ADDR, (int)buf);
    /* 2 . 设置串口接收缓存数据的循环buf长度 */
    dev_ioctl(hdl, IOCTL_UART_SET_CIRCULAR_BUFF_LENTH, sizeof(buf));
    /* 3 . 设置接收数据为阻塞方式,需要非阻塞可以去掉,建议加上超时设置 */
    dev_ioctl(hdl, IOCTL_UART_SET_RECV_BLOCK, 1);
    ...
    while (1) {
        len = dev_read(hdl, recv_buf, sizeof(recv_buf));
        if (len <= 0) {
            if (len == UART_CIRCULAR_BUFFER_WRITE_OVERLAY) {
                dev_ioctl(hdl, IOCTL_UART_FLUSH, 0); //回卷覆盖则冲掉循环buf重新接收
            } else if (len == UART_RECV_TIMEOUT) {
                puts("UART_RECV_TIMEOUT...\r\n");
            }
            continue;
        }
        recv_buf[len] = '\0';
        cmd_do(recv_buf);
    }
}

void cmd_do(char *buf)
{
    char *cmd = buf;
    ...
    if (0 == memcmp(buf, "sta", strlen("sta"))) {  //串口输入 sta <ssid> <pwd> 进入STA模式。uart1 Rx为PA1
        if (sscanf(buf, "sta %s %s", ssid, pwd) == 2) {
            printf("Enter sta mode, ssid : %s, pwd: %s\n", ssid, pwd);
            wifi_sta_connect(ssid, pwd, 1);
        } else if (sscanf(buf, "sta %s", ssid) == 1) {
            wifi_sta_connect(ssid, "", 1);
        }
    } else if (0 == memcmp(buf, "ap", strlen("ap"))) {	//ap <ssid> <pwd> 进入AP模式
        if (sscanf(buf, "ap  %s %s", ssid, pwd) == 2) {
            wifi_enter_ap_mode(ssid, pwd);
        } else if (sscanf(buf, "ap %s", ssid) == 1) {
            wifi_enter_ap_mode(ssid, "");
        }
    } else if (0 == memcmp(buf, "go", strlen("go"))) {	//go <name> 进入GO模式
        ...
        wifi_enter_p2p_mode(P2P_GO_MODE, P2P_Device_Name);
    } else if (0 == memcmp(buf, "gc", strlen("gc"))) {	//gc <name> 进入GC模式
        ...
        wifi_enter_p2p_mode(P2P_GC_MODE, P2P_Device_Name);
    } else if (0 == memcmp(buf, "scan", strlen("scan"))) {	//scan
        void wifi_get_list_example(void);
        wifi_get_list_example();
    } else if (0 == memcmp(buf, "get_mac", strlen("get_mac"))) {	//get_mac 获取mac地址
        wifi_get_mac(mac_addr);
        printf("Mac addr: %02X:%02X:%02X:%02X:%02X:%02X", ...);
    } else if (0 == memcmp(buf, "smp_cfg", strlen("smp_cfg"))) {	//smp_cfg 进入smp_cfg模式
        wifi_enter_smp_cfg_mode();
    } else if (0 == memcmp(buf, "ssid_store_test", strlen("ssid_store_test"))) {
        ... // 见"SSID 记忆与自动重连"
    }
}

static void wifi_set_lan_setting_info(void)
{
    struct lan_setting lan_setting_info = {
        .WIRELESS_IP_ADDR0  = 192, .WIRELESS_IP_ADDR1  = 168,
        .WIRELESS_IP_ADDR2  = 4,   .WIRELESS_IP_ADDR3  = 1,
        .WIRELESS_NETMASK0  = 255, .WIRELESS_NETMASK1  = 255,
        .WIRELESS_NETMASK2  = 255, .WIRELESS_NETMASK3  = 0,
        .WIRELESS_GATEWAY0  = 192, .WIRELESS_GATEWAY1  = 168,
        .WIRELESS_GATEWAY2  = 4,   .WIRELESS_GATEWAY3  = 1,
        .SERVER_IPADDR1  = 192, .SERVER_IPADDR2  = 168,
        .SERVER_IPADDR3  = 4,   .SERVER_IPADDR4  = 1,
        .CLIENT_IPADDR1  = 192, .CLIENT_IPADDR2  = 168,
        .CLIENT_IPADDR3  = 4,   .CLIENT_IPADDR4  = 2,
        .SUB_NET_MASK1   = 255, .SUB_NET_MASK2  = 255,
        .SUB_NET_MASK3   = 255, .SUB_NET_MASK4  = 0,
    };
    net_set_lan_info(&lan_setting_info);
}

static char save_ssid_flag, request_connect_flag;

void wifi_sta_connect(char *ssid, char *pwd, char save)
{
    save_ssid_flag = save;
    request_connect_flag = 1;
    wifi_set_sta_connect_best_ssid(0);
    wifi_enter_sta_mode(ssid, pwd);
}

static void wifi_sta_save_ssid(void)
{
    if (save_ssid_flag) {
        save_ssid_flag = 0;
        struct wifi_mode_info info;
        info.mode = STA_MODE;
        wifi_get_mode_cur_info(&info);
        wifi_store_mode_info(STA_MODE, info.ssid, info.pwd);
    }
}

void wifi_return_sta_mode(void)
{
    if (!wifi_is_on()) {
        return;
    }
    struct wifi_mode_info info;
    info.mode = STA_MODE;
    int ret = wifi_get_mode_stored_info(&info);
    if (ret) {//如果没保存过SSID
        info.ssid = STA_SSID;
        info.pwd = STA_PWD;
    }
    wifi_clear_scan_result(); //根据扫描结果连接信号最优ssid之前先清除之前结果,防止之前最优信号的ssid已下线
    wifi_set_sta_connect_best_ssid(1); //自动连接保存过的最佳WIFI
    save_ssid_flag = 0;
    wifi_enter_sta_mode(info.ssid, info.pwd);
}

u8 ssid_stored_cnt;
struct wifi_stored_sta_info wifi_stored_sta_info[32];
os_time_dly(5 * 100); //假设等待已经连接上路由器
wifi_store_mode_info(STA_MODE, "GJ12", "123456789");
wifi_store_mode_info(STA_MODE, "WL83", "12345678");
wifi_store_mode_info(STA_MODE, "WIFI_PS_TEST", "12345678");
ssid_stored_cnt = wifi_get_stored_sta_info(wifi_stored_sta_info);
for (int i = 0; i < ssid_stored_cnt; i++) {
    printf("1:wifi_get_stored_sta_info[%d]= %s \r\n", i, wifi_stored_sta_info[i].ssid);
}
wifi_del_stored_sta_info("WIFI_PS_TEST"); //中途删除掉其中一个存储过的SSID

struct wifi_scan_ssid_info *wifi_scan_list_get(u32 *total_scan_num)
{
    u32 sta_ssid_num;
    struct wifi_scan_ssid_info *sta_ssid_info;

    if (!os_sem_valid(&pscan_list->scan_sem)) {
        os_sem_create(&pscan_list->scan_sem, 0);
    }
    if (!os_mutex_valid(&pscan_list->scan_mutex)) {
        os_mutex_create(&pscan_list->scan_mutex);
    }

    scan_list_mutex_pend();
    wifi_scan_req();
    //等待WIFI_EVENT_STA_SCAN_COMPLETED事件
    if (scan_list_sem_pend(500)) { //单位10ms
        printf("get_scan_list timeout");
        pscan_list->sem_flag = 0;
        scan_list_mutex_post();
        return NULL;
    }

    sta_ssid_num = 0;
    sta_ssid_info = wifi_get_scan_result(&sta_ssid_num);
    if (total_scan_num) {
        *total_scan_num = sta_ssid_num;
    }
    wifi_clear_scan_result();
    scan_list_mutex_post();
    return sta_ssid_info;
}

void wifi_scan_list_release(struct wifi_scan_ssid_info *list)
{
    if (list) {
        free(list);
    }
}

void wifi_get_list_example(void)
{
    u32 scan_num = 0;
    struct wifi_scan_ssid_info *list = wifi_scan_list_get(&scan_num);
    if (!list) {
        printf("wifi_scan_list_get fail\n");
        return;
    }
    printf("sta_ssid_num: %d\n", scan_num);
    for (int i = 0; i < scan_num; i++) {
        log_info("wifi_sta_scan_channel_test ssid = [%s],rssi = %d,snr = %d, auth_mode = %s",
                 list[i].ssid, list[i].rssi, list[i].snr, get_wifi_auth_mode(list[i].auth_mode));
    }
    wifi_scan_list_release(list);
}

static int compare_dhcp_ipaddr(void)
{
    use_static_ipaddr_flag = 0;
    u8 local_mac[6];
    u8 gw_mac[6];
    struct sta_ip_info  sta_ip_info;
    struct netif_info netif_info;
    int ret = syscfg_read(VM_STA_IPADDR_INDEX, (char *)&sta_ip_info, sizeof(struct sta_ip_info));
    if (ret < 0) {
        log_info("compare_dhcp_ipaddr NO VM_STA_IPADDR_INDEX");
        return -1;
    }
    lwip_get_netif_info(WIFI_NETIF, &netif_info);
    struct wifi_mode_info info;
    info.mode = STA_MODE;
    wifi_get_mode_cur_info(&info);
    wifi_get_bssid(gw_mac);
    wifi_get_mac(local_mac);
    if (!strcmp(info.ssid, (const char *)sta_ip_info.ssid)
        && !memcmp(local_mac, sta_ip_info.local_mac, 6)
        && !memcmp(gw_mac, sta_ip_info.gw_mac, 6)) {
        use_static_ipaddr_flag = 1;
        log_info("compare_dhcp_ipaddr Match");
        return 0;
    }
    return -1;
}

static void store_dhcp_ipaddr(void)
{
    struct sta_ip_info  sta_ip_info = {0};
    u8 sta_channel, local_mac[6], gw_mac[6];
    if (use_static_ipaddr_flag) { //记忆IP匹配成功,不需要重新保存
        return;
    }
    struct netif_info netif_info;
    lwip_get_netif_info(WIFI_NETIF, &netif_info);
    ...
    strcpy((char *)sta_ip_info.ssid, info.ssid);
    memcpy(sta_ip_info.gw_mac, gw_mac, 6);
    memcpy(sta_ip_info.local_mac, local_mac, 6);
    sta_ip_info.ip = netif_info.ip;
    sta_ip_info.netmask = netif_info.netmask;
    sta_ip_info.gw = netif_info.gw;
    sta_ip_info.chanel = sta_channel;
    sta_ip_info.dns = *(u32 *)dns_getserver(0);
    syscfg_write(VM_STA_IPADDR_INDEX, (char *) &sta_ip_info, sizeof(struct sta_ip_info));
    log_info("store_dhcp_ipaddr");
}

#define DROP_CRC_ERR   BIT(0) //drop CRC error frame
#define DROP_PHY_ERR   BIT(1) //drop physical error frame
#define DROP_UC_NOME   BIT(2) //drop not to me unicast frame
#define DROP_NOT_MYBSS BIT(3) //drop frame that is not my BSSID
#define DROP_VER_ERR   BIT(4) //drop 802.11 version error frame
#define DROP_DROP_MC   BIT(5) //drop multicast frame
#define DROP_BC        BIT(6) //drop broadcast frame
#define DROP_DUPL      BIT(7) //drop duplicated frame
#define DROP_CFACK     BIT(8) //drop CF-END + CF-ACK
#define DROP_CFEND     BIT(9) //drop CF-END
#define DROP_ACK       BIT(10) //drop ACK
#define DROP_CTS       BIT(11) //drop CTS
#define DROP_RTS       BIT(12) //drop RTS
#define DROP_PSPOLL    BIT(13) //drop PS-POLL
#define DROP_BA        BIT(14) //drop BA
#define DROP_BAR       BIT(15) //drop BAR
#define DROP_CTRL_RSV  BIT(16) //drop reserve control subtype

//复位对应的接收过滤包类型
void wf_rx_filt_set(unsigned int reg);
//设置对应的接收过滤包类型
void wf_rx_filt_reset(unsigned int reg);
//设置硬件MAC_BSSID寄存器 ,需要在wifi_raw_on后调用
void wf_asic_set_bssid(unsigned char *bssid);
//设置本机硬件MAC, 需要在wifi_raw_on后调用
void wf_asic_set_mac(unsigned char *mac);
//使能channel power检测,需要在wifi_raw_on后调用
void wf_channel_power_detect_enable(void);
//获取当前channel power
char get_current_channel_power(void);