| #include <stdio.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <ctype.h> |
| #include <uci.h> |
| #include "wifi-test-tool.h" |
| |
| static int mac_addr_aton(unsigned char *mac_addr, char *arg) |
| { |
| sscanf(arg, "%02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx", &mac_addr[0], &mac_addr[1], &mac_addr[2], &mac_addr[3], &mac_addr[4], &mac_addr[5]); |
| return 0; |
| } |
| |
| static void mac_addr_ntoa(char *mac_addr, unsigned char *arg) |
| { |
| snprintf(mac_addr, 20, "%02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx", arg[0], arg[1],arg[2],arg[3],arg[4],arg[5]); |
| return; |
| } |
| |
| static int _syscmd(char *cmd, char *retBuf, int retBufSize) |
| { |
| FILE *f; |
| char *ptr = retBuf; |
| int bufSize=retBufSize, bufbytes=0, readbytes=0, cmd_ret=0; |
| |
| |
| if((f = popen(cmd, "r")) == NULL) { |
| fprintf(stderr,"\npopen %s error\n", cmd); |
| return RETURN_ERR; |
| } |
| |
| while(!feof(f)) |
| { |
| *ptr = 0; |
| if(bufSize>=128) { |
| bufbytes=128; |
| } else { |
| bufbytes=bufSize-1; |
| } |
| |
| fgets(ptr,bufbytes,f); |
| readbytes=strlen(ptr); |
| |
| if(!readbytes) |
| break; |
| |
| bufSize-=readbytes; |
| ptr += readbytes; |
| } |
| cmd_ret = pclose(f); |
| retBuf[retBufSize-1]=0; |
| |
| return cmd_ret >> 8; |
| } |
| |
| int phy_index_to_radio(int phyIndex) |
| { |
| char cmd[128] = {0}; |
| char buf[64] = {0}; |
| int radioIndex = 0; |
| snprintf(cmd, sizeof(cmd), "ls /tmp | grep phy%d | cut -d '-' -f2 | tr -d '\n'", phyIndex); |
| _syscmd(cmd, buf, sizeof(buf)); |
| |
| if (strlen(buf) == 0 || strstr(buf, "wifi") == NULL) { |
| fprintf(stderr, "%s: failed to get wifi index\n", __func__); |
| return RETURN_ERR; |
| } |
| sscanf(buf, "wifi%d", &radioIndex); |
| return radioIndex; |
| } |
| |
| void set_channel(wifi_radio_param *radio_param, char *channel) |
| { |
| if (strcmp(channel, "auto") == 0) { |
| radio_param->auto_channel = TRUE; |
| radio_param->channel = 0; |
| } else { |
| radio_param->auto_channel = FALSE; |
| radio_param->channel = strtol(channel, NULL, 10); |
| } |
| return; |
| } |
| |
| void set_country(wifi_radio_param *radio_param, char *country) |
| { |
| strcpy(radio_param->country, country); |
| } |
| |
| void set_band(wifi_radio_param *radio_param, char *band) |
| { |
| strcpy(radio_param->band, band); |
| } |
| |
| void set_noscan(wifi_radio_param *radio_param, char *noscan) |
| { |
| snprintf(radio_param->noscan, 2, "%s", noscan); |
| radio_param->noscan[1] = '\0'; |
| } |
| |
| void set_hwmode(wifi_radio_param *radio_param, char *hwmode) |
| { |
| if (strncmp(hwmode, "11a", 3) == 0) |
| strcpy(radio_param->hwmode, "a"); |
| if (strncmp(hwmode, "11b", 3) == 0) |
| strcpy(radio_param->hwmode, "b"); |
| if (strncmp(hwmode, "11g", 3) == 0) |
| strcpy(radio_param->hwmode, "g"); |
| } |
| |
| void set_htmode(wifi_radio_param *radio_param, char *htmode) |
| { |
| char tmp[16] = {0}; |
| char *ptr = htmode; |
| ULONG bandwidth = 0; |
| radio_param->bandwidth = 20; |
| while (*ptr) { |
| if (isdigit(*ptr)) { |
| bandwidth = strtoul(ptr, NULL, 10); |
| radio_param->bandwidth = bandwidth; |
| break; |
| } |
| ptr++; |
| } |
| |
| // HT40 -> 11NGHT40PLUS |
| // VHT40+ -> 11ACVHT40PLUS |
| // HE80 -> 11AXHE80 |
| // EHT320-1 -> 11BEEHT320-1 |
| if (strstr(htmode, "40+") != NULL) { |
| strncpy(tmp, htmode, strlen(htmode) - 1); |
| strcat(tmp, "PLUS"); |
| } else if (strstr(htmode, "40-") != NULL) { |
| strncpy(tmp, htmode, strlen(htmode) - 1); |
| strcat(tmp, "MINUS"); |
| } else |
| strcpy(tmp, htmode); |
| |
| |
| if (strstr(htmode, "VHT") != NULL) { |
| snprintf(radio_param->htmode, sizeof(radio_param->htmode), "11AC%s", tmp); |
| } else if (strstr(htmode, "EHT") != NULL) { |
| snprintf(radio_param->htmode, sizeof(radio_param->htmode), "11BE%s", tmp); |
| if (strstr(htmode, "320-1") != NULL) |
| radio_param->eht_320_conf = 1; |
| else if (strstr(htmode, "320") != NULL) // EHT320 or EHT320-2 |
| radio_param->eht_320_conf = 2; |
| else |
| radio_param->eht_320_conf = 0; |
| } else if (strstr(htmode, "HT") != NULL && strstr(htmode, "NO") == NULL) { |
| snprintf(radio_param->htmode, sizeof(radio_param->htmode), "11NG%s", tmp); |
| } else if (strstr(htmode, "HE") != NULL) { |
| snprintf(radio_param->htmode, sizeof(radio_param->htmode), "11AX%s", tmp); |
| } else { // NOHT or NONE should be parsed with the band, so just fill the original string. |
| strcpy(radio_param->htmode, tmp); |
| } |
| |
| } |
| |
| void set_disable(wifi_radio_param *radio_param, char *disable) |
| { |
| if (strcmp(disable, "1") == 0) |
| radio_param->disabled = TRUE; |
| else |
| radio_param->disabled = FALSE; |
| } |
| |
| void set_rxant(wifi_radio_param *radio_param, char *mask) |
| { |
| radio_param->rxantenna = strtol(mask, NULL, 16); |
| } |
| |
| void set_txant(wifi_radio_param *radio_param, char *mask) |
| { |
| radio_param->txantenna = strtol(mask, NULL, 16); |
| } |
| |
| void set_igmpsn_enable(wifi_intf_param *intf_param, char *enable) |
| { |
| if (strcmp(enable, "1") == 0) |
| intf_param->igmpsn_enable = TRUE; |
| else |
| intf_param->igmpsn_enable = FALSE; |
| } |
| |
| void set_wps_state(wifi_intf_param *intf_param, char *wps_state) |
| { |
| intf_param->wps_state = strtol(wps_state, NULL, 10); |
| } |
| |
| void set_wps_cancel(wifi_intf_param *intf_param, char *enable) |
| { |
| if (strcmp(enable, "1") == 0) |
| intf_param->wps_cancel = TRUE; |
| else |
| intf_param->wps_cancel = FALSE; |
| } |
| |
| void set_wps_pushbutton(wifi_intf_param *intf_param, char *enable) |
| { |
| if (strcmp(enable, "1") == 0) |
| intf_param->wps_pushbutton = TRUE; |
| else |
| intf_param->wps_pushbutton = FALSE; |
| } |
| |
| |
| void set_macfilter(wifi_intf_param *intf_param, char *macfilter) |
| { |
| strncpy(intf_param->macfilter, macfilter, 10); |
| } |
| |
| void set_maclist(wifi_intf_param *intf_param, char *maclist) |
| { |
| strncpy(intf_param->maclist, maclist, 512); |
| } |
| |
| void set_htcoex(wifi_radio_param *radio_param, char *ht_coex) |
| { |
| radio_param->ht_coex = strtol(ht_coex, NULL, 10); |
| } |
| |
| void set_rts(wifi_radio_param *radio_param, char *rts) |
| { |
| radio_param->rtsThreshold = strtol(rts, NULL, 10); |
| } |
| |
| void set_radionum(wifi_intf_param *intf_param, char *phy_name) |
| { |
| int radio_num = 0; |
| char *ptr = phy_name; |
| int phyId = 0; |
| |
| while (*ptr) { |
| if (isdigit(*ptr)) { |
| phyId = strtoul(ptr, NULL, 10); |
| radio_num = phy_index_to_radio(phyId); |
| intf_param->radio_index = radio_num; |
| break; |
| } |
| ptr++; |
| } |
| } |
| |
| void set_ssid(wifi_intf_param *intf_param, char *ssid) |
| { |
| strncpy(intf_param->ssid, ssid, 32); |
| } |
| |
| void set_encryption(wifi_intf_param *intf_param, char *encryption_mode) |
| { |
| if (strcmp(encryption_mode, "none") == 0) { |
| intf_param->security.mode = wifi_security_mode_none; |
| intf_param->security.encr = wifi_encryption_none; |
| }else if(strncmp(encryption_mode, "psk2", 4) == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa2_personal; |
| }else if(strncmp(encryption_mode, "psk-",4) == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa_wpa2_personal; |
| }else if(strncmp(encryption_mode, "psk",3) == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa_personal; |
| }else if(strncmp(encryption_mode, "wpa2",4) == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa2_enterprise; |
| }else if(strncmp(encryption_mode, "wpa-",4) == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa_wpa2_enterprise; |
| }else if(strcmp(encryption_mode, "sae") == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa3_personal; |
| }else if(strcmp(encryption_mode, "wpa3") == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa3_enterprise; |
| }else if(strcmp(encryption_mode, "sae-mixed") == 0){ |
| intf_param->security.mode = wifi_security_mode_wpa3_transition; |
| }else if(strcmp(encryption_mode, "owe") == 0){ |
| intf_param->security.mode = wifi_security_mode_enhanced_open; |
| } |
| |
| if(strstr(encryption_mode, "tkip") && (strstr(encryption_mode, "ccmp") || strstr(encryption_mode, "aes") )){ |
| intf_param->security.encr = wifi_encryption_aes_tkip; |
| }else if (strstr(encryption_mode, "tkip")){ |
| intf_param->security.encr = wifi_encryption_tkip; |
| }else{ |
| intf_param->security.encr = wifi_encryption_aes; |
| } |
| |
| if(!strcmp(encryption_mode, "wpa3") || !strcmp(encryption_mode, "sae") || !strcmp(encryption_mode, "owe")){ |
| intf_param->security.mfp = wifi_mfp_cfg_required; |
| }else if (!strcmp(encryption_mode, "sae-mixed")){ |
| intf_param->security.mfp = wifi_mfp_cfg_optional; |
| }else{ |
| intf_param->security.mfp = wifi_mfp_cfg_disabled; |
| } |
| } |
| |
| void set_key(wifi_intf_param *intf_param, char *key) |
| { |
| strncpy(intf_param->security.u.key.key, key, 64); |
| } |
| |
| void set_ifname(wifi_intf_param *intf_param, char *ifname) |
| { |
| if (strlen(ifname) > 15) |
| return; |
| strncpy(intf_param->ifname, ifname, strlen(ifname) + 1); |
| } |
| |
| void set_wds(wifi_intf_param *intf_param, char *wds_enable) |
| { |
| intf_param->wds_mode = FALSE; |
| if (strncmp(wds_enable, "1", 1) == 0) |
| intf_param->wds_mode = TRUE; |
| } |
| |
| void set_hidden(wifi_intf_param *intf_param, char *hidden) |
| { |
| intf_param->hidden = strtol(hidden, NULL, 10); |
| } |
| |
| int set_interface_bssid(int phy_index, int offset, mac_address_t *bssid) |
| { |
| FILE *f; |
| char mac_file[64] = {0}; |
| char mac_address[20] = {0}; |
| |
| sprintf(mac_file, "/sys/class/ieee80211/phy%d/macaddress", phy_index); |
| f = fopen(mac_file, "r"); |
| if (f == NULL) |
| return -1; |
| fgets(mac_address, 20, f); |
| fclose(f); |
| |
| mac_addr_aton(&(*bssid)[0], mac_address); |
| (*bssid)[0] += offset*2; |
| return 0; |
| } |
| |
| void set_radio_param(wifi_radio_param radio_parameter) |
| { |
| int ret = 0; |
| wifi_radio_operationParam_t operationParam = {0}; |
| |
| if(radio_parameter.radio_index == -1) |
| return; |
| |
| if (radio_parameter.disabled == TRUE) { |
| wifi_setRadioEnable(radio_parameter.radio_index, FALSE); |
| return; |
| } |
| |
| fprintf(stderr, "Start setting radio\n"); |
| |
| if (radio_parameter.txantenna != 0 && radio_parameter.txantenna == radio_parameter.rxantenna) { |
| ret = wifi_setRadioTxChainMask(radio_parameter.radio_index, radio_parameter.txantenna); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set Tx Chain mask failed!!!]\n"); |
| } |
| |
| // Get current radio setting |
| ret = wifi_getRadioOperatingParameters(radio_parameter.radio_index, &operationParam); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Get OperatingParameters failed!!!]\n"); |
| operationParam.enable = TRUE; |
| |
| // Channel |
| operationParam.autoChannelEnabled = radio_parameter.auto_channel; |
| operationParam.channel = radio_parameter.channel; |
| |
| // bandwidth |
| if (radio_parameter.bandwidth == 20){ |
| operationParam.channelWidth = WIFI_CHANNELBANDWIDTH_20MHZ; |
| }else if (radio_parameter.bandwidth == 40){ |
| operationParam.channelWidth = WIFI_CHANNELBANDWIDTH_40MHZ; |
| }else if (radio_parameter.bandwidth == 80){ |
| operationParam.channelWidth = WIFI_CHANNELBANDWIDTH_80MHZ; |
| }else if (radio_parameter.bandwidth == 160){ |
| operationParam.channelWidth = WIFI_CHANNELBANDWIDTH_160MHZ; |
| }else if (radio_parameter.bandwidth == 320){ |
| if ((radio_parameter.eht_320_conf == 1 || radio_parameter.channel <= 29) && radio_parameter.channel < 193) |
| operationParam.channelWidth = WIFI_CHANNELBANDWIDTH_320_1MHZ; |
| else if (radio_parameter.eht_320_conf == 2 || radio_parameter.channel >= 193) |
| operationParam.channelWidth = WIFI_CHANNELBANDWIDTH_320_2MHZ; |
| else |
| fprintf(stderr, "[Set EHT 320 bandwidth error with conf %d!!!]\n", radio_parameter.eht_320_conf); |
| } |
| |
| // htmode |
| unsigned int mode = 0; // enum wifi_ieee80211Variant_t |
| if (strcmp(radio_parameter.band, "2g") == 0) { |
| mode |= WIFI_80211_VARIANT_B | WIFI_80211_VARIANT_G; |
| if (strcmp(radio_parameter.htmode, "NOHT") == 0 || strcmp(radio_parameter.htmode, "NONE") == 0) |
| strcpy(radio_parameter.htmode, "11G"); |
| else if (strstr(radio_parameter.htmode, "HE") != NULL) |
| mode |= WIFI_80211_VARIANT_N | WIFI_80211_VARIANT_AX; |
| else if (strstr(radio_parameter.htmode, "EHT") != NULL) |
| mode |= WIFI_80211_VARIANT_N | WIFI_80211_VARIANT_AX | WIFI_80211_VARIANT_BE; |
| else if (strstr(radio_parameter.htmode, "HT") != NULL) |
| mode |= WIFI_80211_VARIANT_N; |
| |
| } else if (strcmp(radio_parameter.band, "5g") == 0) { |
| mode |= WIFI_80211_VARIANT_A; |
| if (strcmp(radio_parameter.htmode, "NOHT") == 0 || strcmp(radio_parameter.htmode, "NONE") == 0) |
| strcpy(radio_parameter.htmode, "11A"); |
| else if (strstr(radio_parameter.htmode, "VHT") != NULL) |
| mode |= WIFI_80211_VARIANT_N | WIFI_80211_VARIANT_AC; |
| else if (strstr(radio_parameter.htmode, "HE") != NULL) |
| mode |= WIFI_80211_VARIANT_N | WIFI_80211_VARIANT_AC | WIFI_80211_VARIANT_AX; |
| else if (strstr(radio_parameter.htmode, "EHT") != NULL) |
| mode |= WIFI_80211_VARIANT_N | WIFI_80211_VARIANT_AC | WIFI_80211_VARIANT_AX | WIFI_80211_VARIANT_BE; |
| else if (strstr(radio_parameter.htmode, "HT") != NULL) |
| mode |= WIFI_80211_VARIANT_N; |
| } else if (strcmp(radio_parameter.band, "6g") == 0) { |
| mode |= WIFI_80211_VARIANT_AX; |
| if (strstr(radio_parameter.htmode, "EHT") != NULL) |
| mode |= WIFI_80211_VARIANT_BE; |
| } |
| |
| operationParam.variant = mode; |
| |
| // rtsThreshold, zero means not set |
| if ((radio_parameter.rtsThreshold < 65535) && radio_parameter.rtsThreshold) |
| operationParam.rtsThreshold = radio_parameter.rtsThreshold; |
| |
| //ht_coex |
| operationParam.obssCoex = radio_parameter.ht_coex; |
| |
| // apply setting |
| ret = wifi_setRadioOperatingParameters(radio_parameter.radio_index, &operationParam); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Apply setting failed!!!]\n"); |
| |
| // Country |
| fprintf(stderr, "Set Country: %s\n", radio_parameter.country); |
| ret = wifi_setRadioCountryCode(radio_parameter.radio_index, radio_parameter.country); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set Country failed!!!]\n"); |
| ret = 0; |
| |
| // hwmode |
| fprintf(stderr, "Set hwmode: %s\n", radio_parameter.hwmode); |
| ret = wifi_setRadioHwMode(radio_parameter.radio_index, radio_parameter.hwmode); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set hwmode failed!!!]\n"); |
| ret = 0; |
| |
| // noscan |
| fprintf(stderr, "Set noscan: %s \n", radio_parameter.noscan); |
| if(strlen(radio_parameter.noscan)){ |
| ret = wifi_setNoscan(radio_parameter.radio_index, radio_parameter.noscan); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set noscan failed!!!]\n"); |
| } |
| ret = 0; |
| |
| } |
| |
| void set_ap_param(wifi_intf_param ap_param , wifi_vap_info_map_t *map) |
| { |
| int ret = 0; |
| int vap_index_in_map = 0; |
| int phy_index = 0; |
| int key_len = 0; |
| wifi_vap_info_t vap_info = {0}; |
| BOOL radio_enable = FALSE; |
| char *maclist; |
| |
| if(ap_param.radio_index == -1) |
| return; |
| |
| wifi_getRadioEnable(ap_param.radio_index, &radio_enable); |
| if (radio_enable == FALSE) |
| return; |
| |
| |
| // get the index of the map |
| for (int i = 0; i < MAX_NUM_VAP_PER_RADIO; i++) { |
| if (map->vap_array[i].vap_index == ap_param.ap_index) { |
| vap_index_in_map = i; |
| break; |
| } |
| } |
| |
| |
| fprintf(stderr, "Start setting ap vap_index_in_map=%d\n", vap_index_in_map); |
| |
| vap_info = map->vap_array[vap_index_in_map]; |
| vap_info.u.bss_info.enabled = TRUE; |
| phy_index = radio_index_to_phy(vap_info.radio_index); |
| if (set_interface_bssid(phy_index, ap_param.mac_offset, &vap_info.u.bss_info.bssid) == -1) { |
| fprintf(stderr, "Get mac address failed.\n"); |
| return; |
| } |
| |
| // SSID |
| strncpy(vap_info.u.bss_info.ssid, ap_param.ssid, 33); |
| vap_info.u.bss_info.ssid[32] = '\0'; |
| |
| // interface |
| if (strlen(ap_param.ifname) != 0) { |
| strncpy(vap_info.vap_name, ap_param.ifname, 16); |
| vap_info.vap_name[15] = '\0'; |
| } |
| |
| // Security |
| if (ap_param.security.mode == wifi_security_mode_wpa3_personal || ap_param.security.mode == wifi_security_mode_wpa3_transition){ |
| // OpenWrt script only set psk, here we choose to set both psk and sae. |
| ap_param.security.u.key.type = wifi_security_key_type_psk_sae; |
| } else { |
| key_len = strlen(ap_param.security.u.key.key); |
| if (key_len == 64) |
| ap_param.security.u.key.type = wifi_security_key_type_psk; |
| else if (key_len >= 8 && key_len < 64) |
| ap_param.security.u.key.type = wifi_security_key_type_pass; |
| } |
| |
| vap_info.u.bss_info.security.mode = ap_param.security.mode; |
| vap_info.u.bss_info.security.encr = ap_param.security.encr; |
| vap_info.u.bss_info.security.mfp = ap_param.security.mfp; |
| vap_info.u.bss_info.security.u.key.type = ap_param.security.u.key.type; |
| strncpy(vap_info.u.bss_info.security.u.key.key, ap_param.security.u.key.key, 64); |
| |
| // hidden |
| vap_info.u.bss_info.showSsid = (ap_param.hidden ? 0 : 1); |
| |
| // igmpsn_enable |
| vap_info.u.bss_info.mcast2ucast = ap_param.igmpsn_enable; |
| fprintf(stderr, "Set igmpsn_enable: %d \n", ap_param.igmpsn_enable); |
| |
| // wps_state |
| fprintf(stderr, "Set wps_state: %d \n", ap_param.wps_state); |
| if (ap_param.wps_state == 2) |
| ret = wifi_setApWpsEnable(ap_param.ap_index, 1); |
| else |
| ret = wifi_setApWpsEnable(ap_param.ap_index, 0); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set wps_state failed!!!]\n"); |
| ret = 0; |
| |
| // macfilter |
| fprintf(stderr, "Set macfilter: %s \n", ap_param.macfilter); |
| if ((strcmp(ap_param.macfilter, "disable") == 0) || (strcmp(ap_param.macfilter, "\0") == 0) ) |
| vap_info.u.bss_info.mac_filter_enable = false; |
| else if (strcmp(ap_param.macfilter, "deny") == 0){ |
| vap_info.u.bss_info.mac_filter_enable = true; |
| vap_info.u.bss_info.mac_filter_mode = wifi_mac_filter_mode_black_list; |
| } |
| else if (strcmp(ap_param.macfilter, "allow") == 0){ |
| vap_info.u.bss_info.mac_filter_enable = true; |
| vap_info.u.bss_info.mac_filter_mode = wifi_mac_filter_mode_white_list; |
| } |
| else |
| fprintf(stderr, "The macfilter tpye: %s is invalid!!!\n", ap_param.macfilter); |
| |
| // maclist |
| fprintf(stderr, "Set maclist: %s \n", ap_param.maclist); |
| if ((strlen(ap_param.maclist) == 0)){ |
| ret = wifi_delApAclDevices(ap_param.ap_index); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Del all maclist failed!!!]\n"); |
| ret = 0; |
| } |
| else if (strcmp(ap_param.macfilter, "allow") == 0) { |
| maclist = strtok(ap_param.maclist, ";"); |
| while (maclist != NULL) |
| { |
| ret = wifi_addApAclDevice(ap_param.ap_index, maclist); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Add maclist failed!!!]\n"); |
| ret = 0; |
| maclist = strtok(NULL, ";"); |
| } |
| } |
| else if (strcmp(ap_param.macfilter, "deny") == 0) { |
| maclist = strtok(ap_param.maclist, ";"); |
| while (maclist != NULL) |
| { |
| ret = wifi_addApDenyAclDevice(ap_param.ap_index, maclist); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Add maclist failed!!!]\n"); |
| ret = 0; |
| maclist = strtok(NULL, ";"); |
| } |
| } |
| |
| ret = 0; |
| // Replace the setting with uci config |
| map->vap_array[vap_index_in_map] = vap_info; |
| } |
| |
| void set_sta_param(wifi_intf_param sta_param) |
| { |
| wifi_sta_network_t *sta = NULL; |
| mac_address_t sta_mac = {0}; |
| char sta_mac_str[20] = {0}; |
| char key_mgmt[16] = {0}; |
| char pairwise[16] = {0}; |
| int phy_index = 0; |
| |
| sta = calloc(1, sizeof(wifi_sta_network_t)); |
| |
| phy_index = radio_index_to_phy(sta_param.radio_index); |
| set_interface_bssid(phy_index, sta_param.mac_offset, &sta_mac); |
| mac_addr_ntoa(sta_mac_str, sta_mac); |
| snprintf(sta->ssid, 31, "%s", sta_param.ssid); |
| sta->ssid[31] = '\0'; |
| snprintf(sta->psk, 64, "%s", sta_param.password); |
| |
| if (sta_param.security.mode == wifi_security_mode_none) |
| strcpy(key_mgmt, "NONE"); |
| else if (sta_param.security.mode == wifi_security_mode_wpa3_personal) |
| strcpy(key_mgmt, "SAE"); |
| else if (sta_param.security.mode == wifi_security_mode_enhanced_open) |
| strcpy(key_mgmt, "OWE"); |
| else |
| strcpy(key_mgmt, "WPA-PSK"); |
| snprintf(sta->key_mgmt, 64, "%s", key_mgmt); |
| |
| if (sta_param.security.encr == wifi_encryption_aes) |
| strcpy(pairwise, "CCMP"); |
| else if (sta_param.security.encr == wifi_encryption_tkip) |
| strcpy(pairwise, "TKIP"); |
| else |
| strcpy(pairwise, "CCMP TKIP"); |
| snprintf(sta->pairwise, 64, "%s", pairwise); |
| |
| if (strlen(sta_param.security.u.key.key) > 0) |
| strncpy(sta->psk, sta_param.security.u.key.key, 127); |
| sta->psk[127] = '\0'; |
| sta->psk_len = strlen(sta->psk); |
| |
| if (sta_param.wds_mode == TRUE) |
| sta->flags |= WIFI_STA_NET_F_4ADDR_MULTI_AP; |
| |
| wifi_createSTAInterface(sta_param.sta_index, sta_mac_str, sta_param.wds_mode); |
| |
| if (wifi_setSTANetworks(sta_param.sta_index, &sta, 1, FALSE) == RETURN_ERR) { |
| fprintf(stderr, "Write to sta %d config file failed\n", sta_param.sta_index); |
| free(sta); |
| return; |
| } |
| free(sta); |
| |
| if (wifi_setSTAEnabled(sta_param.sta_index, TRUE) == RETURN_ERR) { |
| fprintf(stderr, "Enable station failed\n"); |
| return; |
| } |
| } |
| |
| void show_ap_param(wifi_intf_param ap_param , wifi_vap_info_map_t *map) |
| { |
| int ret = 0; |
| int filter_mode = 0; |
| int vap_index_in_map = 0; |
| int phy_index = 0; |
| wifi_vap_info_t vap_info = {0}; |
| BOOL radio_enable = FALSE; |
| BOOL wps_state = FALSE; |
| BOOL igmpsn_state = FALSE; |
| UINT buf_size = 1024; |
| char macArray[1024] = ""; |
| |
| if(ap_param.radio_index == -1) |
| return; |
| |
| wifi_getRadioEnable(ap_param.radio_index, &radio_enable); |
| if (radio_enable == FALSE) |
| return; |
| |
| |
| // get the index of the map |
| for (int i = 0; i < map->num_vaps; i++) { |
| if (map->vap_array[i].vap_index == ap_param.ap_index) { |
| vap_index_in_map = i; |
| break; |
| } |
| } |
| |
| vap_info = map->vap_array[vap_index_in_map]; |
| vap_info.u.bss_info.enabled = TRUE; |
| phy_index = radio_index_to_phy(vap_info.radio_index); |
| |
| // SSID |
| printf("wifi%d ssid: %s\n", ap_param.ap_index, vap_info.u.bss_info.ssid); |
| |
| // igmpsn_enable |
| ret = wifi_getRadioIGMPSnoopingEnable(ap_param.radio_index, &igmpsn_state); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Get igmpsn_state failed!!!]\n"); |
| else |
| printf("wifi%d igmpsn_state: %d\n", ap_param.ap_index, igmpsn_state); |
| ret = 0; |
| |
| // wps_state |
| ret = wifi_getApWpsEnable(ap_param.ap_index, &wps_state); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Get wps_state failed!!!]\n"); |
| else |
| printf("wifi%d wps_state: %d\n", ap_param.ap_index, wps_state); |
| ret = 0; |
| |
| // macfilter |
| ret = wifi_getApMacAddressControlMode(ap_param.ap_index, &filter_mode); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Get macfilter failed!!!]\n"); |
| else{ |
| if (filter_mode == 0) |
| printf("wifi%d macfilter: disable\n", ap_param.ap_index); |
| else if (filter_mode == 1) |
| printf("wifi%d macfilter: allow\n", ap_param.ap_index); |
| else if (filter_mode == 2) |
| printf("wifi%d macfilter: deny\n", ap_param.ap_index); |
| } |
| ret = 0; |
| |
| // maclist |
| printf("wifi%d maclist: \n", ap_param.ap_index); |
| if (filter_mode == 0) |
| printf("wifi%d macfilter is disable\n", ap_param.ap_index); |
| else if (filter_mode == 1){ |
| ret = wifi_getApAclDevices(ap_param.ap_index, macArray, buf_size); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Get maclist failed!!!]\n"); |
| else |
| printf("%s \n", macArray); |
| ret = 0; |
| } |
| else if (filter_mode == 2){ |
| ret = wifi_getApDenyAclDevices(ap_param.ap_index, macArray, buf_size); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Get maclist failed!!!]\n"); |
| else |
| printf("%s \n", macArray); |
| ret = 0; |
| } |
| |
| } |
| |
| int apply_uci_config () |
| { |
| struct uci_context *uci_ctx = uci_alloc_context(); |
| struct uci_package *uci_pkg = NULL; |
| struct uci_element *e; |
| // struct uci_section *s; |
| const char cfg_name[] = "wireless"; |
| int max_radio_num = 0; |
| BOOL parsing_radio = FALSE; |
| int apCount[3] = {0}; |
| int staCount[3] = {0}; |
| wifi_vap_info_map_t vap_map[3] = {0}; |
| int ret = 0; |
| int i = 0, j = 0; |
| |
| wifi_getMaxRadioNumber(&max_radio_num); |
| fprintf(stderr, "max radio number: %d\n", max_radio_num); |
| for (i = 0; i < max_radio_num ;i++ ){ |
| ret = wifi_getRadioVapInfoMap(i, &vap_map[i]); |
| if (ret != RETURN_OK) { // if failed, we set assume this vap as the first vap. |
| fprintf(stderr, "[Get vap map failed!!!]\n"); |
| } |
| |
| /*reset radio's ap number, ap number ++ by uci config */ |
| vap_map[i].num_vaps = 0; |
| } |
| if (uci_load(uci_ctx, cfg_name, &uci_pkg) != UCI_OK) { |
| uci_free_context(uci_ctx); |
| fprintf(stderr, "%s: load uci failed.\n", __func__); |
| return RETURN_ERR; |
| } |
| |
| uci_foreach_element(&uci_pkg->sections, e) { |
| |
| struct uci_section *s = uci_to_section(e); |
| struct uci_element *option = NULL; |
| wifi_radio_param radio_param = {0}; |
| wifi_intf_param intf_param = {0}; |
| int phyId = 0; |
| radio_param.radio_index = -1; |
| intf_param.ap_index = -1; |
| |
| if (strcmp(s->type, "wifi-device") == 0) { |
| sscanf(s->e.name, "radio%d", &phyId); |
| radio_param.radio_index = phy_index_to_radio(phyId); |
| parsing_radio = TRUE; |
| fprintf(stderr, "\n----- Start parsing radio %d config. -----\n", radio_param.radio_index); |
| } else if (strcmp(s->type, "wifi-iface") == 0) { |
| parsing_radio = FALSE; |
| } |
| |
| uci_foreach_element(&s->options, option) { |
| |
| struct uci_option *op = uci_to_option(option); |
| if (parsing_radio == TRUE) { |
| // transform the type from input string and store the value in radio_param. |
| if (strcmp(op->e.name, "channel") == 0) |
| set_channel(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "hwmode") == 0) |
| set_hwmode(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "htmode") == 0) |
| set_htmode(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "disabled") == 0) |
| set_disable(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "band") == 0) |
| set_band(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "country") == 0) |
| set_country(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "noscan") == 0) |
| set_noscan(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "rxantenna") == 0) |
| set_rxant(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "txantenna") == 0) |
| set_txant(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "ht_coex") == 0) |
| set_htcoex(&radio_param, op->v.string); |
| else if (strcmp(op->e.name, "rts") == 0) |
| set_rts(&radio_param, op->v.string); |
| else |
| fprintf(stderr, "[%s %s not set!]\n", op->e.name, op->v.string); |
| } else { |
| // parsing iface |
| if (strcmp(op->e.name, "device") == 0){ |
| set_radionum(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "mode") == 0){ |
| intf_param.mac_offset = staCount[intf_param.radio_index] + apCount[intf_param.radio_index]; |
| if (strncmp(op->v.string, "sta", 3) == 0) { |
| intf_param.sta_mode = TRUE; |
| intf_param.sta_index = intf_param.radio_index + staCount[intf_param.radio_index]*max_radio_num; |
| staCount[intf_param.radio_index] ++ ; |
| fprintf(stderr, "\n----- Start parsing sta %d config. -----\n", intf_param.sta_index); |
| } else if (strncmp(op->v.string, "ap", 2) == 0) { |
| intf_param.sta_mode = FALSE; |
| intf_param.ap_index = intf_param.radio_index + apCount[intf_param.radio_index]*max_radio_num; |
| apCount[intf_param.radio_index] ++ ; |
| fprintf(stderr, "\n----- Start parsing ap %d config. -----\n", intf_param.ap_index); |
| } |
| }else if (strcmp(op->e.name, "ssid") == 0){ |
| set_ssid(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "encryption") == 0){ |
| set_encryption(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "key") == 0){ |
| set_key(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "ifname") == 0){ |
| set_ifname(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "wds") == 0){ |
| set_wds(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "hidden") == 0){ |
| set_hidden(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "igmpsn_enable") == 0){ |
| set_igmpsn_enable(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "wps_state") == 0){ |
| set_wps_state(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "macfilter") == 0){ |
| set_macfilter(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "maclist") == 0){ |
| set_maclist(&intf_param, op->v.string); |
| }else{ |
| fprintf(stderr, "[%s %s not set!]\n", op->e.name, op->v.string); |
| } |
| } |
| } |
| if (parsing_radio == TRUE) { |
| set_radio_param(radio_param); |
| } |
| else if (intf_param.sta_mode == TRUE) |
| set_sta_param(intf_param); |
| else { |
| set_ap_param(intf_param, &vap_map[intf_param.radio_index]); |
| vap_map[intf_param.radio_index].num_vaps++; |
| } |
| } |
| |
| /* disable all interface, re-enable by UCI configuration */ |
| |
| fprintf(stderr, "\n----- Disable all interfaces. -----\n"); |
| |
| for (i = 0; i < max_radio_num; i++ ){ |
| for(j = max_radio_num*(MAX_NUM_VAP_PER_RADIO-1) + i; j >= 0; j -= max_radio_num) { |
| ret = wifi_setApEnable(j, FALSE); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[disable ap %d failed!!!]\n", j); |
| else |
| fprintf(stderr, "[disable ap %d success.]\n", j); |
| } |
| } |
| fprintf(stderr, "\n----- Start setting Vaps. -----\n"); |
| |
| for (i = 0; i < max_radio_num ;i++ ){ |
| |
| fprintf(stderr, "\n create Vap radio:%d num_vaps=%d.\n", i, vap_map[i].num_vaps); |
| ret = wifi_createVAP(i, &vap_map[i]); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Apply vap setting failed!!!]\n"); |
| } |
| |
| for (i = 0; i < 3 ;i++ ){ |
| apCount[i] = 0; |
| staCount[i] = 0; |
| } |
| |
| uci_foreach_element(&uci_pkg->sections, e) { |
| |
| struct uci_section *s = uci_to_section(e); |
| struct uci_element *option = NULL; |
| wifi_radio_param radio_param = {0}; |
| wifi_intf_param intf_param = {0}; |
| int phyId = 0; |
| radio_param.radio_index = -1; |
| intf_param.ap_index = -1; |
| |
| if (strcmp(s->type, "wifi-iface") == 0) { |
| uci_foreach_element(&s->options, option) { |
| |
| struct uci_option *op = uci_to_option(option); |
| { |
| // parsing iface |
| if (strcmp(op->e.name, "device") == 0){ |
| set_radionum(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "mode") == 0){ |
| intf_param.mac_offset = staCount[intf_param.radio_index] + apCount[intf_param.radio_index]; |
| if (strncmp(op->v.string, "sta", 3) == 0) { |
| intf_param.sta_mode = TRUE; |
| intf_param.sta_index = intf_param.radio_index + staCount[intf_param.radio_index]*max_radio_num; |
| staCount[intf_param.radio_index] ++ ; |
| fprintf(stderr, "\n----- Start parsing sta %d config. -----\n", intf_param.sta_index); |
| } else if (strncmp(op->v.string, "ap", 2) == 0) { |
| intf_param.sta_mode = FALSE; |
| intf_param.ap_index = intf_param.radio_index + apCount[intf_param.radio_index]*max_radio_num; |
| apCount[intf_param.radio_index] ++ ; |
| fprintf(stderr, "\n----- Start parsing ap %d config. -----\n", intf_param.ap_index); |
| } |
| }else if (strcmp(op->e.name, "wps_cancel") == 0){ |
| set_wps_cancel(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "wps_pushbutton") == 0){ |
| set_wps_pushbutton(&intf_param, op->v.string); |
| } |
| } |
| } |
| |
| // wps_cancel |
| fprintf(stderr, "Set wps_cancel: %d \n", intf_param.wps_cancel); |
| if (intf_param.wps_cancel){ |
| ret = wifi_cancelApWPS(intf_param.ap_index); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set wps_cancel failed!!!]\n"); |
| ret = 0; |
| } |
| |
| // wps_pushbutton |
| fprintf(stderr, "Set wps_pushbutton: %d \n", intf_param.wps_pushbutton); |
| if (intf_param.wps_pushbutton){ |
| ret = wifi_setApWpsButtonPush(intf_param.ap_index); |
| if (ret != RETURN_OK) |
| fprintf(stderr, "[Set wps_pushbutton failed!!!]\n"); |
| ret = 0; |
| } |
| } |
| } |
| |
| uci_unload(uci_ctx, uci_pkg); |
| uci_free_context(uci_ctx); |
| return RETURN_OK; |
| } |
| |
| int dump_wifi_status () |
| { |
| |
| struct uci_context *uci_ctx = uci_alloc_context(); |
| struct uci_package *uci_pkg = NULL; |
| struct uci_element *e; |
| // struct uci_section *s; |
| const char cfg_name[] = "wireless"; |
| int max_radio_num = 0; |
| BOOL parsing_radio = FALSE; |
| int apCount[3] = {0}; |
| int staCount[3] = {0}; |
| wifi_vap_info_map_t vap_map[3] = {0}; |
| int ret = 0; |
| int i = 0; |
| |
| wifi_getMaxRadioNumber(&max_radio_num); |
| fprintf(stderr, "max radio number: %d\n", max_radio_num); |
| for (i = 0; i < max_radio_num ;i++ ){ |
| ret = wifi_getRadioVapInfoMap(i, &vap_map[i]); |
| if (ret != RETURN_OK) { // if failed, we set assume this vap as the first vap. |
| fprintf(stderr, "[Get vap map failed!!!]\n"); |
| vap_map[i].num_vaps = MAX_NUM_VAP_PER_RADIO; |
| } |
| } |
| if (uci_load(uci_ctx, cfg_name, &uci_pkg) != UCI_OK) { |
| uci_free_context(uci_ctx); |
| fprintf(stderr, "%s: load uci failed.\n", __func__); |
| return RETURN_ERR; |
| } |
| |
| uci_foreach_element(&uci_pkg->sections, e) { |
| |
| struct uci_section *s = uci_to_section(e); |
| struct uci_element *option = NULL; |
| wifi_radio_param radio_param = {0}; |
| wifi_intf_param intf_param = {0}; |
| int phyId = 0; |
| radio_param.radio_index = -1; |
| intf_param.ap_index = -1; |
| |
| if (strcmp(s->type, "wifi-device") == 0) { |
| sscanf(s->e.name, "radio%d", &phyId); |
| radio_param.radio_index = phy_index_to_radio(phyId); |
| parsing_radio = TRUE; |
| fprintf(stderr, "\n----- Start show radio %d config. -----\n", radio_param.radio_index); |
| } else if (strcmp(s->type, "wifi-iface") == 0) { |
| parsing_radio = FALSE; |
| } |
| |
| uci_foreach_element(&s->options, option) { |
| |
| struct uci_option *op = uci_to_option(option); |
| if (parsing_radio == TRUE) { |
| // transform the type from input string and store the value in radio_param. |
| } else { |
| // parsing iface |
| if (strcmp(op->e.name, "device") == 0){ |
| set_radionum(&intf_param, op->v.string); |
| }else if (strcmp(op->e.name, "mode") == 0){ |
| intf_param.mac_offset = staCount[intf_param.radio_index] + apCount[intf_param.radio_index]; |
| if (strncmp(op->v.string, "sta", 3) == 0) { |
| intf_param.sta_mode = TRUE; |
| intf_param.sta_index = intf_param.radio_index + staCount[intf_param.radio_index]*max_radio_num; |
| staCount[intf_param.radio_index] ++ ; |
| fprintf(stderr, "\n----- Start show sta %d config. -----\n", intf_param.sta_index); |
| } else if (strncmp(op->v.string, "ap", 2) == 0) { |
| intf_param.sta_mode = FALSE; |
| intf_param.ap_index = intf_param.radio_index + apCount[intf_param.radio_index]*max_radio_num; |
| apCount[intf_param.radio_index] ++ ; |
| fprintf(stderr, "\n----- Start show ap %d config. -----\n", intf_param.ap_index); |
| } |
| } |
| } |
| } |
| if (parsing_radio == TRUE) |
| printf("show radio params: \n"); |
| // show_radio_param(radio_param); |
| else if (intf_param.sta_mode == TRUE) |
| printf("show sta params: \n"); |
| // show_sta_param(intf_param); |
| else{ |
| printf("show ap params: \n"); |
| show_ap_param(intf_param, &vap_map[intf_param.radio_index]); |
| } |
| } |
| |
| uci_unload(uci_ctx, uci_pkg); |
| uci_free_context(uci_ctx); |
| return RETURN_OK; |
| |
| } |
| |
| int main(int argc, char **argv) |
| { |
| if (argc != 2 || (strcmp(argv[1], "reload") != 0 && strcmp(argv[1], "dump") != 0) ){ |
| fprintf(stderr, "Usage: wifi reload/wifi dump.\nThis tool is only for RDKB MSP/SQC test.\n"); |
| return -1; |
| } |
| if (strcmp(argv[1], "reload") == 0) |
| apply_uci_config(); |
| else if (strcmp(argv[1], "dump") == 0) |
| dump_wifi_status(); |
| return 0; |
| } |