255 lines
9.1 KiB
C
255 lines
9.1 KiB
C
/*
|
|
* SPDX-FileCopyrightText: 2026 Espressif Systems (Shanghai) CO LTD
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
/* Get recv router csi
|
|
|
|
This example code is in the Public Domain (or CC0 licensed, at your option.)
|
|
|
|
Unless required by applicable law or agreed to in writing, this
|
|
software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
CONDITIONS OF ANY KIND, either express or implied.
|
|
*/
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include "freertos/FreeRTOS.h"
|
|
#include "freertos/task.h"
|
|
#include "freertos/event_groups.h"
|
|
|
|
#include "nvs_flash.h"
|
|
|
|
#include "esp_mac.h"
|
|
#include "rom/ets_sys.h"
|
|
#include "esp_log.h"
|
|
#include "esp_wifi.h"
|
|
#include "esp_netif.h"
|
|
#include "esp_now.h"
|
|
|
|
#include "lwip/inet.h"
|
|
#include "lwip/netdb.h"
|
|
#include "lwip/sockets.h"
|
|
#include "ping/ping_sock.h"
|
|
|
|
#include "protocol_examples_common.h"
|
|
#include "esp_csi_gain_ctrl.h"
|
|
|
|
#define CONFIG_SEND_FREQUENCY 100
|
|
#if CONFIG_IDF_TARGET_ESP32C5 || CONFIG_IDF_TARGET_ESP32C61
|
|
#define CSI_FORCE_LLTF 0
|
|
#endif
|
|
#define CONFIG_FORCE_GAIN 0
|
|
|
|
#if CONFIG_IDF_TARGET_ESP32S3 || CONFIG_IDF_TARGET_ESP32C3 || CONFIG_IDF_TARGET_ESP32C5 || CONFIG_IDF_TARGET_ESP32C6 || CONFIG_IDF_TARGET_ESP32C61
|
|
#define CONFIG_GAIN_CONTROL 1
|
|
#endif
|
|
|
|
#if ESP_IDF_VERSION >= ESP_IDF_VERSION_VAL(6, 0, 0)
|
|
#define ESP_IF_WIFI_STA ESP_MAC_WIFI_STA
|
|
#endif
|
|
|
|
static const char *TAG = "csi_recv_router";
|
|
|
|
/* UDP socket for CSI data transmission */
|
|
static int s_udp_socket = -1;
|
|
static struct sockaddr_in s_dest_addr;
|
|
static char s_udp_buffer[2048];
|
|
|
|
static void wifi_csi_rx_cb(void *ctx, wifi_csi_info_t *info)
|
|
{
|
|
if (!info || !info->buf) {
|
|
ESP_LOGW(TAG, "<%s> wifi_csi_cb", esp_err_to_name(ESP_ERR_INVALID_ARG));
|
|
return;
|
|
}
|
|
|
|
if (memcmp(info->mac, ctx, 6)) {
|
|
return;
|
|
}
|
|
|
|
const wifi_pkt_rx_ctrl_t *rx_ctrl = &info->rx_ctrl;
|
|
static int s_count = 0;
|
|
float compensate_gain = 1.0f;
|
|
static uint8_t agc_gain = 0;
|
|
static int8_t fft_gain = 0;
|
|
#if CONFIG_GAIN_CONTROL
|
|
static uint8_t agc_gain_baseline = 0;
|
|
static int8_t fft_gain_baseline = 0;
|
|
esp_csi_gain_ctrl_get_rx_gain(rx_ctrl, &agc_gain, &fft_gain);
|
|
if (s_count < 100) {
|
|
esp_csi_gain_ctrl_record_rx_gain(agc_gain, fft_gain);
|
|
} else if (s_count == 100) {
|
|
esp_csi_gain_ctrl_get_rx_gain_baseline(&agc_gain_baseline, &fft_gain_baseline);
|
|
#if CONFIG_FORCE_GAIN
|
|
esp_csi_gain_ctrl_set_rx_force_gain(agc_gain_baseline, fft_gain_baseline);
|
|
ESP_LOGI(TAG, "fft_force %d, agc_force %d", fft_gain_baseline, agc_gain_baseline);
|
|
#endif
|
|
}
|
|
esp_csi_gain_ctrl_get_gain_compensation(&compensate_gain, agc_gain, fft_gain);
|
|
ESP_LOGD(TAG, "compensate_gain %f, agc_gain %d, fft_gain %d", compensate_gain, agc_gain, fft_gain);
|
|
#endif
|
|
|
|
/* Build CSI data into buffer for UDP transmission */
|
|
int pos = 0;
|
|
|
|
#if CONFIG_IDF_TARGET_ESP32C5 || CONFIG_IDF_TARGET_ESP32C6 || CONFIG_IDF_TARGET_ESP32C61
|
|
if (!s_count) {
|
|
ESP_LOGI(TAG, "================ CSI RECV (UDP) ================");
|
|
}
|
|
pos = snprintf(s_udp_buffer, sizeof(s_udp_buffer),
|
|
"CSI_DATA,%d," MACSTR ",%d,%d,%d,%d,%d,%d,%d,%d,%d",
|
|
s_count, MAC2STR(info->mac), rx_ctrl->rssi, rx_ctrl->rate,
|
|
rx_ctrl->noise_floor, fft_gain, agc_gain, rx_ctrl->channel,
|
|
rx_ctrl->timestamp, rx_ctrl->sig_len, rx_ctrl->rx_state);
|
|
#else
|
|
if (!s_count) {
|
|
ESP_LOGI(TAG, "================ CSI RECV (UDP) ================");
|
|
}
|
|
pos = snprintf(s_udp_buffer, sizeof(s_udp_buffer),
|
|
"CSI_DATA,%d," MACSTR ",%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d",
|
|
s_count, MAC2STR(info->mac), rx_ctrl->rssi, rx_ctrl->rate, rx_ctrl->sig_mode,
|
|
rx_ctrl->mcs, rx_ctrl->cwb, rx_ctrl->smoothing, rx_ctrl->not_sounding,
|
|
rx_ctrl->aggregation, rx_ctrl->stbc, rx_ctrl->fec_coding, rx_ctrl->sgi,
|
|
rx_ctrl->noise_floor, rx_ctrl->ampdu_cnt, rx_ctrl->channel, rx_ctrl->secondary_channel,
|
|
rx_ctrl->timestamp, rx_ctrl->ant, rx_ctrl->sig_len, rx_ctrl->rx_state);
|
|
#endif
|
|
|
|
#if (CONFIG_IDF_TARGET_ESP32C5 || CONFIG_IDF_TARGET_ESP32C61) && CSI_FORCE_LLTF
|
|
int16_t csi = ((int16_t)(((((uint16_t)info->buf[1]) << 8) | info->buf[0]) << 4) >> 4);
|
|
pos += snprintf(s_udp_buffer + pos, sizeof(s_udp_buffer) - pos,
|
|
",%d,%d,\"[%d", (info->len - 2) / 2, info->first_word_invalid, (int16_t)(compensate_gain * csi));
|
|
for (int i = 2; i < (info->len - 2); i += 2) {
|
|
csi = ((int16_t)(((((uint16_t)info->buf[i + 1]) << 8) | info->buf[i]) << 4) >> 4);
|
|
pos += snprintf(s_udp_buffer + pos, sizeof(s_udp_buffer) - pos, ",%d", (int16_t)(compensate_gain * csi));
|
|
}
|
|
#else
|
|
pos += snprintf(s_udp_buffer + pos, sizeof(s_udp_buffer) - pos,
|
|
",%d,%d,\"[%d", info->len, info->first_word_invalid, (int16_t)(compensate_gain * info->buf[0]));
|
|
for (int i = 1; i < info->len; i++) {
|
|
pos += snprintf(s_udp_buffer + pos, sizeof(s_udp_buffer) - pos, ",%d", (int16_t)(compensate_gain * info->buf[i]));
|
|
}
|
|
#endif
|
|
pos += snprintf(s_udp_buffer + pos, sizeof(s_udp_buffer) - pos, "]\"\n");
|
|
|
|
/* Send via UDP */
|
|
if (s_udp_socket >= 0) {
|
|
sendto(s_udp_socket, s_udp_buffer, pos, 0, (struct sockaddr *)&s_dest_addr, sizeof(s_dest_addr));
|
|
}
|
|
|
|
s_count++;
|
|
}
|
|
|
|
static void wifi_csi_init()
|
|
{
|
|
/**
|
|
* @brief In order to ensure the compatibility of routers, only LLTF sub-carriers are selected.
|
|
*/
|
|
#if CONFIG_IDF_TARGET_ESP32C5 || CONFIG_IDF_TARGET_ESP32C61
|
|
wifi_csi_config_t csi_config = {
|
|
.enable = true,
|
|
.acquire_csi_legacy = true,
|
|
.acquire_csi_force_lltf = CSI_FORCE_LLTF,
|
|
.acquire_csi_ht20 = true,
|
|
.acquire_csi_ht40 = true,
|
|
.acquire_csi_vht = false,
|
|
.acquire_csi_su = false,
|
|
.acquire_csi_mu = false,
|
|
.acquire_csi_dcm = false,
|
|
.acquire_csi_beamformed = false,
|
|
.acquire_csi_he_stbc_mode = 2,
|
|
.val_scale_cfg = 0,
|
|
.dump_ack_en = false,
|
|
.reserved = false
|
|
};
|
|
#elif CONFIG_IDF_TARGET_ESP32C6
|
|
wifi_csi_config_t csi_config = {
|
|
.enable = true,
|
|
.acquire_csi_legacy = true,
|
|
.acquire_csi_ht20 = true,
|
|
.acquire_csi_ht40 = true,
|
|
.acquire_csi_su = false,
|
|
.acquire_csi_mu = false,
|
|
.acquire_csi_dcm = false,
|
|
.acquire_csi_beamformed = false,
|
|
.acquire_csi_he_stbc = 2,
|
|
.val_scale_cfg = false,
|
|
.dump_ack_en = false,
|
|
.reserved = false
|
|
};
|
|
#else
|
|
wifi_csi_config_t csi_config = {
|
|
.lltf_en = true,
|
|
.htltf_en = false,
|
|
.stbc_htltf2_en = false,
|
|
.ltf_merge_en = true,
|
|
.channel_filter_en = true,
|
|
.manu_scale = true,
|
|
.shift = true,
|
|
};
|
|
#endif
|
|
static wifi_ap_record_t s_ap_info = {0};
|
|
ESP_ERROR_CHECK(esp_wifi_sta_get_ap_info(&s_ap_info));
|
|
ESP_ERROR_CHECK(esp_wifi_set_csi_config(&csi_config));
|
|
ESP_ERROR_CHECK(esp_wifi_set_csi_rx_cb(wifi_csi_rx_cb, s_ap_info.bssid));
|
|
ESP_ERROR_CHECK(esp_wifi_set_csi(true));
|
|
}
|
|
|
|
static void udp_socket_init(void)
|
|
{
|
|
s_udp_socket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
if (s_udp_socket < 0) {
|
|
ESP_LOGE(TAG, "Failed to create UDP socket: errno %d", errno);
|
|
return;
|
|
}
|
|
|
|
memset(&s_dest_addr, 0, sizeof(s_dest_addr));
|
|
s_dest_addr.sin_family = AF_INET;
|
|
s_dest_addr.sin_port = htons(CONFIG_CSI_UDP_TARGET_PORT);
|
|
inet_pton(AF_INET, CONFIG_CSI_UDP_TARGET_IP, &s_dest_addr.sin_addr);
|
|
|
|
ESP_LOGI(TAG, "UDP socket initialized, sending to %s:%d",
|
|
CONFIG_CSI_UDP_TARGET_IP, CONFIG_CSI_UDP_TARGET_PORT);
|
|
}
|
|
|
|
static esp_err_t wifi_ping_router_start()
|
|
{
|
|
static esp_ping_handle_t ping_handle = NULL;
|
|
|
|
esp_ping_config_t ping_config = ESP_PING_DEFAULT_CONFIG();
|
|
ping_config.count = 0;
|
|
ping_config.interval_ms = 1000 / CONFIG_SEND_FREQUENCY;
|
|
ping_config.task_stack_size = 3072;
|
|
ping_config.data_size = 1;
|
|
|
|
esp_netif_ip_info_t local_ip;
|
|
esp_netif_get_ip_info(esp_netif_get_handle_from_ifkey("WIFI_STA_DEF"), &local_ip);
|
|
ESP_LOGI(TAG, "got ip:" IPSTR ", gw: " IPSTR, IP2STR(&local_ip.ip), IP2STR(&local_ip.gw));
|
|
ping_config.target_addr.u_addr.ip4.addr = ip4_addr_get_u32(&local_ip.gw);
|
|
ping_config.target_addr.type = ESP_IPADDR_TYPE_V4;
|
|
|
|
esp_ping_callbacks_t cbs = { 0 };
|
|
esp_ping_new_session(&ping_config, &cbs, &ping_handle);
|
|
esp_ping_start(ping_handle);
|
|
|
|
return ESP_OK;
|
|
}
|
|
|
|
void app_main()
|
|
{
|
|
ESP_ERROR_CHECK(nvs_flash_init());
|
|
ESP_ERROR_CHECK(esp_netif_init());
|
|
ESP_ERROR_CHECK(esp_event_loop_create_default());
|
|
|
|
/**
|
|
* @brief This helper function configures Wi-Fi, as selected in menuconfig.
|
|
* Read "Establishing Wi-Fi Connection" section in esp-idf/examples/protocols/README.md
|
|
* for more information about this function.
|
|
*/
|
|
ESP_ERROR_CHECK(example_connect());
|
|
|
|
udp_socket_init();
|
|
wifi_csi_init();
|
|
wifi_ping_router_start();
|
|
}
|