esp-protocols/components/esp_wifi_remote/wifi_remote_net.c

/*
 * SPDX-FileCopyrightText: 2023-2024 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Apache-2.0
 */
#include <esp_private/wifi.h>
#include "esp_err.h"
#include "esp_wifi_remote.h"
#include "esp_log.h"

#define CHANNELS 2

static esp_remote_channel_tx_fn_t s_tx_cb[CHANNELS];
static esp_remote_channel_t s_channel[CHANNELS];
static wifi_rxcb_t s_rx_fn[CHANNELS];

esp_err_t esp_wifi_remote_channel_rx(void *h, void *buffer, void *buff_to_free, size_t len)
{
    assert(h);
    if (h == s_channel[0]) {
        assert(s_rx_fn[0]);
        return s_rx_fn[0](buffer, len, buff_to_free);
    }
    if (h == s_channel[1]) {
        assert(s_rx_fn[1]);
        return s_rx_fn[1](buffer, len, buff_to_free);
    }
    return ESP_FAIL;
}

esp_err_t esp_wifi_remote_channel_set(wifi_interface_t ifx, void *h, esp_remote_channel_tx_fn_t tx_cb)
{
    if (ifx == WIFI_IF_STA) {
        s_channel[0] = h;
        s_tx_cb[0] = tx_cb;
        return ESP_OK;
    }
    if (ifx == WIFI_IF_AP) {
        s_channel[1] = h;
        s_tx_cb[1] = tx_cb;
        return ESP_OK;
    }
    return ESP_FAIL;
}


esp_err_t esp_wifi_internal_set_sta_ip(void)
{
    return ESP_OK;
}

esp_err_t esp_wifi_internal_reg_netstack_buf_cb(wifi_netstack_buf_ref_cb_t ref, wifi_netstack_buf_free_cb_t free)
{
    return ESP_OK;
}

void esp_wifi_internal_free_rx_buffer(void *buffer)
{
    if (buffer) {
        free(buffer);
    }
}

int esp_wifi_internal_tx(wifi_interface_t ifx, void *buffer, uint16_t len)
{
    if (ifx == WIFI_IF_STA) {
        /* TODO: If not needed, remove arg3 */
        return s_tx_cb[0](s_channel[0], buffer, len);
    }
    if (ifx == WIFI_IF_AP) {
        return s_tx_cb[1](s_channel[1], buffer, len);
    }

    return -1;
}

esp_err_t esp_wifi_internal_reg_rxcb(wifi_interface_t ifx, wifi_rxcb_t fn)
{
    if (ifx == WIFI_IF_STA) {
        ESP_LOGI("esp_wifi_remote", "%s: sta: %p", __func__, fn);
        s_rx_fn[0] = fn;
        return ESP_OK;
    }
    if (ifx == WIFI_IF_AP) {
        s_rx_fn[1] = fn;
        return ESP_OK;
    }

    return ESP_FAIL;
}