| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <unistd.h> |
| #include <pthread.h> |
| #include <signal.h> // Include signal header |
| |
| #include "gsw_wifi_interface_sdk.h" |
| |
| // Hostapd 和 wpa_supplicant 进程 ID |
| pid_t hostapd_pid = 0; |
| pid_t wpa_pid = 0; |
| |
| // 辅助函数:将认证类型枚举转换为字符串 |
| const char *auth_type_to_string(gsw_wifi_auth_e auth) |
| { |
| switch (auth) |
| { |
| case GSW_WIFI_AUTH_OPEN: |
| return "OPEN"; |
| case GSW_WIFI_AUTH_WEP: |
| return "WEP"; |
| case GSW_WIFI_AUTH_WPA_PSK: |
| return "WPA_PSK"; |
| case GSW_WIFI_AUTH_WPA2_PSK: |
| return "WPA2_PSK"; |
| case GSW_WIFI_AUTH_WPA3_PSK: |
| return "WPA3_PSK"; |
| // 若有其他认证类型,可继续添加 case |
| default: |
| return "Unknown"; |
| } |
| } |
| |
| void handle_ctrl_c(int sig) |
| { |
| printf("\nExiting...\n"); |
| gsw_wifi_sta_stop(); |
| gsw_wifi_ap_stop(); |
| gsw_wifi_disable(); |
| exit(0); |
| } |
| |
| // 添加AP回调函数 |
| void ap_event_callback(void *arg, gsw_wifi_ap_status_e status) |
| { |
| if (status == GSW_WIFI_STATUS_CONNECT) |
| { |
| printf("AP mode is now running and ready for connections\n"); |
| } |
| else if (status == GSW_WIFI_STATUS_DISCONNECT) |
| { |
| printf("AP mode has been stopped\n"); |
| } |
| } |
| |
| void ap_setting(void) |
| { |
| int ret = 0; |
| char ssid[33] = "mtbk-test"; |
| char password[64] = "12345678"; |
| gsw_wifi_auth_e auth = GSW_WIFI_AUTH_WPA2_PSK; |
| int frequency = 2; |
| gsw_wifi_bandwidth_type_e bandwidth = GSW_WIFI_BANDWIDTH_HT80; |
| int channel = 0; |
| int temp_auth, temp_bandwidth; |
| |
| printf("AP mode configuration:\n"); |
| printf("Enter SSID (default: mtbk-test): "); |
| if (scanf("%32s", ssid) != 1) |
| { |
| strcpy(ssid, "mtbk-test"); |
| } |
| |
| ret = gsw_wifi_ap_ssid_set(ssid); |
| if (ret != 0) |
| { |
| printf("Failed to set SSID\n"); |
| gsw_wifi_ap_ssid_set("mtbk-test"); |
| } |
| |
| printf("Enter password (default: 12345678): "); |
| if (scanf("%63s", password) != 1) |
| { |
| strcpy(password, "12345678"); |
| } |
| |
| ret = gsw_wifi_ap_password_set(password); |
| if (ret != 0) |
| { |
| printf("Failed to set password\n"); |
| gsw_wifi_ap_password_set("12345678"); |
| } |
| |
| printf("Enter auth type (0: OPEN, 1: WEP(Only support open), 2: WPA_PSK(wifi4 and above are not supported), 3: WPA2_PSK, 4: WPA3_PSK, default: WPA2_PSK): "); |
| if (scanf("%d", &temp_auth) == 1) |
| { |
| auth = (gsw_wifi_auth_e)temp_auth; |
| } |
| |
| ret = gsw_wifi_ap_auth_set(auth); |
| if (ret != 0) |
| { |
| printf("Failed to set auth type\n"); |
| gsw_wifi_ap_auth_set(GSW_WIFI_AUTH_WPA2_PSK); |
| } |
| |
| printf("Enter frequency (1: 2.4G, 2: 5G, default: 2): "); |
| if (scanf("%d", &frequency) != 1) |
| { |
| frequency = 2; |
| } |
| |
| ret = gsw_wifi_ap_frequency_set(frequency); |
| if (ret != 0) |
| { |
| printf("Failed to set frequency\n"); |
| gsw_wifi_ap_frequency_set(2); |
| } |
| |
| printf("Enter bandwidth (0: HT10, 1: HT20, 2: HT40, 3: HT80, default: HT40): "); |
| if (scanf("%d", &temp_bandwidth) == 1) |
| { |
| bandwidth = (gsw_wifi_bandwidth_type_e)temp_bandwidth; |
| } |
| |
| ret = gsw_wifi_ap_bandwidth_set(bandwidth); |
| if (ret != 0) |
| { |
| printf("Failed to set bandwidth\n"); |
| gsw_wifi_ap_bandwidth_set(GSW_WIFI_BANDWIDTH_HT40); |
| } |
| |
| printf("Enter channel (0: auto, 2.4G: 1-13, 5G: 36-165, default: 0): "); |
| if (scanf("%d", &channel) != 1) |
| { |
| channel = 0; |
| } |
| |
| ret = gsw_wifi_ap_channel_set(channel); |
| if (ret != 0) |
| { |
| printf("Failed to set channel\n"); |
| gsw_wifi_ap_channel_set(0); |
| } |
| |
| printf("AP mode configuration set.\n"); |
| } |
| |
| int main() |
| { |
| signal(SIGINT, handle_ctrl_c); // Register signal handler |
| int ret = 0; |
| gsw_wifi_ap_run_status_e ap_status; |
| // 加载 Wi-Fi 驱动并启动 AP 模式 |
| gsw_wifi_sdk_interface_init(); |
| if (gsw_wifi_enable() != 0) |
| { |
| printf("Failed to enable Wi-Fi driver\n"); |
| return -1; |
| } |
| gsw_wifi_get_ap_status(&ap_status); |
| printf("ap status: %s\n", ap_status == GSW_WIFI_AP_STATUS_ENABLE ? "enable" : "disable"); |
| begin: |
| printf("Do you want to set default AP mode? (y/n): "); |
| char ap_seting; |
| if (scanf(" %c", &ap_seting) != 1) |
| { |
| printf("Error reading input\n"); |
| return -1; |
| } |
| |
| if (ap_seting == 'y' || ap_seting == 'Y') |
| { |
| gsw_wifi_ap_ssid_set("mtbk-测试"); |
| gsw_wifi_ap_password_set("12345678"); |
| gsw_wifi_ap_auth_set(GSW_WIFI_AUTH_WPA2_PSK); |
| gsw_wifi_ap_frequency_set(2); |
| gsw_wifi_ap_bandwidth_set(GSW_WIFI_BANDWIDTH_HT40); |
| gsw_wifi_ap_channel_set(0); |
| } |
| else if (ap_seting == 'n' || ap_seting == 'N') |
| { |
| ap_setting(); |
| } |
| else |
| { |
| printf("Invalid input\n"); |
| return -1; |
| } |
| |
| if (gsw_wifi_ap_start() != 0) |
| { |
| printf("Failed to start AP mode\n"); |
| goto begin; |
| } |
| // 注册AP回调 |
| if (gsw_wifi_reg_ap_event_callback(NULL, ap_event_callback) != 0) |
| { |
| printf("Failed to register AP event callback\n"); |
| return -1; |
| } |
| gsw_wifi_get_ap_status(&ap_status); |
| printf("ap status: %s\n", ap_status == GSW_WIFI_AP_STATUS_ENABLE ? "enable" : "disable"); |
| printf("Wi-Fi driver loaded and AP mode started.\n"); |
| |
| while (1) |
| { |
| char input[64]; |
| printf("\nEnter command (1: start STA; 2: scan hotspots; 3: connect to hotspot; 4: ap get; q: exit; ap-status: check ap status; restart-ap .) "); |
| if (scanf("%63s", input) != 1) |
| { // Check scanf return value |
| printf("Error reading input\n"); |
| break; |
| } |
| |
| if (strcmp(input, "q") == 0) |
| { |
| break; |
| } |
| else if (strcmp(input, "ap-status") == 0) |
| { |
| gsw_wifi_get_ap_status(&ap_status); |
| printf("ap status: %s\n", ap_status == GSW_WIFI_AP_STATUS_ENABLE ? "enable" : "disable"); |
| } |
| else if (strcmp(input, "ap-stop") == 0) |
| { |
| printf("stop AP...\n"); |
| gsw_wifi_ap_stop(); |
| } |
| else if (strcmp(input, "ap-mac") == 0) |
| { |
| char tmpmac[18U]; |
| unsigned char apmac[18U]; |
| gsw_wifi_get_interface_mac(GSW_WIFI_INTERFACE_1, tmpmac); |
| sscanf(tmpmac, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx", &apmac[0], &apmac[1], &apmac[2], &apmac[3], &apmac[4], &apmac[5]); |
| printf("connect ap mac: %02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx\n", apmac[0], apmac[1], apmac[2], apmac[3], apmac[4], apmac[5]); |
| } |
| else if (strcmp(input, "sta-mac") == 0) |
| { |
| char tmpmac[18U]; |
| unsigned char sta_mac[18U]; |
| gsw_wifi_get_interface_mac(GSW_WIFI_INTERFACE_0, tmpmac); |
| sscanf(tmpmac, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx", &sta_mac[0], &sta_mac[1], &sta_mac[2], &sta_mac[3], &sta_mac[4], &sta_mac[5]); |
| printf("connect sta mac: %02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx\n", sta_mac[0], sta_mac[1], sta_mac[2], sta_mac[3], sta_mac[4], sta_mac[5]); |
| } |
| else if (strcmp(input, "ap-start") == 0) |
| { |
| printf("y/Y: no set ; other: reset\n"); |
| if (scanf(" %c", &ap_seting) != 1) |
| { |
| printf("Error reading input\n"); |
| } |
| |
| if (ap_seting == 'y' || ap_seting == 'Y') |
| { |
| printf("ap is running, no need to set\n"); |
| } |
| else |
| { |
| ap_setting(); |
| } |
| gsw_wifi_ap_start(); |
| gsw_wifi_get_ap_status(&ap_status); |
| printf("ap status: %s\n", ap_status == GSW_WIFI_AP_STATUS_ENABLE ? "enable" : "disable"); |
| } |
| else if (strcmp(input, "restart-ap") == 0) |
| { |
| printf("Restarting AP...\n"); |
| ap_setting(); |
| gsw_wifi_ap_restart(); |
| gsw_wifi_get_ap_status(&ap_status); |
| printf("ap status: %s\n", ap_status == GSW_WIFI_AP_STATUS_ENABLE ? "enable" : "disable"); |
| } |
| |
| else if (strcmp(input, "1") == 0) |
| { |
| // 启动 STA 模式 |
| if (gsw_wifi_sta_start() != 0) |
| { |
| printf("Failed to start STA mode\n"); |
| } |
| else |
| { |
| printf("STA mode started.\n"); |
| } |
| } |
| else if (strcmp(input, "2") == 0) |
| { |
| // 搜索热点 |
| gsw_scan_info_s scan_list[100]; |
| int list_len; |
| if (gsw_wifi_get_scan_list(scan_list, 100, &list_len) == 0) |
| { |
| printf("Available hotspots:\n"); |
| for (int i = 0; i < list_len; i++) |
| { |
| const char *auth_str = auth_type_to_string(scan_list[i].auth); |
| printf("%d. SSID: %s, BSSID: %s, RSSI: %d, Auth: %s\n", i + 1, scan_list[i].ssid, scan_list[i].mac, scan_list[i].rssi, auth_str); |
| } |
| } |
| else |
| { |
| printf("Failed to scan for hotspots\n"); |
| } |
| } |
| else if (strcmp(input, "3") == 0) |
| { |
| // 输入对应的账户和密码进行连接,从扫描结果获取认证方式 |
| gsw_scan_info_s scan_list[100]; |
| int list_len; |
| if (gsw_wifi_get_scan_list(scan_list, 100, &list_len) == 0) |
| { |
| printf("Available hotspots:\n"); |
| for (int i = 0; i < list_len; i++) |
| { |
| const char *auth_str = auth_type_to_string(scan_list[i].auth); |
| printf("%d. SSID: %s, BSSID: %s, RSSI: %d, Auth: %s\n", i + 1, scan_list[i].ssid, scan_list[i].mac, scan_list[i].rssi, auth_str); |
| } |
| |
| int choice; |
| printf("Enter the number of the hotspot to connect: "); |
| if (scanf("%d", &choice) != 1) |
| { // Check scanf return value |
| printf("Error reading input\n"); |
| continue; |
| } |
| if (choice < 1 || choice > list_len) |
| { |
| printf("Invalid choice\n"); |
| continue; |
| } |
| |
| // 直接从扫描结果中获取认证方式 |
| gsw_wifi_auth_e auth = scan_list[choice - 1].auth; |
| const char *auth_str = auth_type_to_string(auth); |
| printf("Got authentication type from scan result: %s\n", auth_str); |
| |
| char password[64] = ""; // 默认空密码 |
| if (auth != GSW_WIFI_AUTH_OPEN) |
| { |
| printf("Enter password for SSID %s: ", scan_list[choice - 1].ssid); |
| if (scanf("%63s", password) != 1) |
| { |
| printf("Error reading input\n"); |
| continue; |
| } |
| } |
| else |
| { |
| printf("Open network, skipping password input.\n"); |
| } |
| |
| if (gsw_wifi_sta_connect(scan_list[choice - 1].ssid, auth, password) != 0) |
| { |
| printf("Failed to connect to SSID %s\n", scan_list[choice - 1].ssid); |
| } |
| else |
| { |
| printf("Attempting to connect to SSID %s\n", scan_list[choice - 1].ssid); |
| // 等待连接成功 |
| sleep(5); |
| |
| // 获取连接的 SSID |
| char connected_ssid[33]; |
| if (gsw_wifi_get_sta_ssid(connected_ssid) == 0) |
| { |
| printf("Connected to SSID: %s\n", connected_ssid); |
| } |
| else |
| { |
| printf("Failed to connect\n"); |
| } |
| } |
| } |
| else |
| { |
| printf("Failed to scan for hotspots\n"); |
| } |
| } |
| else if (strcmp(input, "4") == 0) |
| { |
| // 获取AP相关信息并格式化输出 |
| printf("\nAP Configuration:\n"); |
| printf("-----------------\n"); |
| |
| char ssid[33]; |
| if (gsw_wifi_ap_ssid_get(ssid) == 0) |
| { |
| printf("SSID: %-20s\n", ssid); |
| } |
| else |
| { |
| printf("SSID: Failed to get\n"); |
| } |
| |
| int frequency; |
| if (gsw_wifi_ap_frequency_get(&frequency) == 0) |
| { |
| printf("Frequency: %-15s\n", frequency == 1 ? "2.4GHz" : "5GHz"); |
| } |
| else |
| { |
| printf("Frequency: Failed to get\n"); |
| } |
| |
| gsw_wifi_bandwidth_type_e bandwidth; |
| if (gsw_wifi_ap_bandwidth_get(&bandwidth) == 0) |
| { |
| const char *bw_str[] = {"10MHz", "20MHz", "40MHz", "80MHz"}; |
| printf("Bandwidth: %-15s\n", bandwidth >= 0 && bandwidth <= 3 ? bw_str[bandwidth] : "Unknown"); |
| } |
| else |
| { |
| printf("Bandwidth: Failed to get\n"); |
| } |
| |
| int channel; |
| if (gsw_wifi_ap_channel_get(&channel) == 0) |
| { |
| printf("Channel: %-18d\n", channel); |
| } |
| else |
| { |
| printf("Channel: Failed to get\n"); |
| } |
| |
| gsw_wifi_auth_e auth; |
| if (gsw_wifi_ap_auth_get(&auth) == 0) |
| { |
| printf("Authentication: %-10s\n", auth_type_to_string(auth)); |
| } |
| else |
| { |
| printf("Authentication: Failed to get\n"); |
| } |
| |
| char password[65]; |
| if (gsw_wifi_ap_password_get(password) == 0) |
| { |
| printf("Password: %-18s\n", password); |
| } |
| else |
| { |
| printf("Password: Failed to get\n"); |
| } |
| |
| printf("-----------------\n"); |
| } |
| else |
| { |
| printf("Invalid command\n"); |
| } |
| } |
| |
| // 停止 AP 和 STA 模式,卸载 Wi-Fi 驱动 |
| gsw_wifi_ap_stop(); |
| gsw_wifi_sta_stop(); |
| gsw_wifi_disable(); |
| |
| printf("Wi-Fi driver unloaded and AP/STA modes stopped.\n"); |
| |
| return 0; |
| } |