relay_mode.c

Go to the documentation of this file.

/*
 * Copyright 2026 Morse Micro
 *
 * This file is licensed under terms that can be found in the LICENSE.md file in the root
 * directory of the Morse Micro IoT SDK software package.
 */
 
#include <string.h>
#include "mmwlan.h"
#include "mmconfig.h"
#include "mmosal.h"
#include "mmutils.h"
#include "mm_app_common.h"
#include "mm_app_loadconfig.h"
#include "mmlog.h"
 
/*
 * --
 * Default SSID/Security configuration
 * --
 */
 
#ifndef AP_SSID
#define AP_SSID MorseMicroIoT
#endif
 
#ifndef SAE_PASSPHRASE
#define SAE_PASSPHRASE 12345678
#endif
 
/* Default security type  */
#ifndef SECURITY_TYPE
#define SECURITY_TYPE MMWLAN_SAE
#endif
 
/* Default PMF mode */
#ifndef PMF_MODE
#define PMF_MODE MMWLAN_PMF_REQUIRED
#endif
 
#ifndef MAX_STAS
#define MAX_STAS MMWLAN_DEFAULT_AP_MAX_STAS
#endif
 
#define _STRINGIFY(x) #x
#define STRINGIFY(x) _STRINGIFY(x)
 
uint32_t opaque_argument_value;
 
static char *mmwlan_ap_sta_state_to_str(enum mmwlan_ap_sta_state state)
{
    switch (state)
    {
        case MMWLAN_AP_STA_UNKNOWN:
            return "Unknown (disconnected)";
 
        case MMWLAN_AP_STA_AUTHENTICATED:
            return "Authenticated";
 
        case MMWLAN_AP_STA_ASSOCIATED:
            return "Associated";
 
        case MMWLAN_AP_STA_AUTHORIZED:
            return "Authorized";
    }
 
    return "Unrecognized";
}
 
static void handle_ap_sta_status(const struct mmwlan_ap_sta_status *sta_status, void *arg)
{
    /* Validate that the opaque argument received matches the value passed in. This is just for
     * testing purposes. */
    MMOSAL_ASSERT(arg == &opaque_argument_value);
 
    printf("AP STA " MM_MAC_ADDR_FMT " State: %s, AID: %u, Time: %lu ms\n",
           MM_MAC_ADDR_VAL(sta_status->mac_addr),
           mmwlan_ap_sta_state_to_str(sta_status->state),
           sta_status->aid,
           mmosal_get_time_ms());
}
 
void load_mmwlan_ap_args(struct mmwlan_ap_args *ap_args)
{
    char strval[32];
 
    /* Load SSID */
    (void)mmosal_safer_strcpy((char *)ap_args->ssid, STRINGIFY(AP_SSID), sizeof(ap_args->ssid));
    (void)mmconfig_read_string("wlan.ap_ssid", (char *)ap_args->ssid, sizeof(ap_args->ssid));
    ap_args->ssid_len = strlen((char *)ap_args->ssid);
 
    /* Load password */
    (void)mmosal_safer_strcpy(ap_args->passphrase,
                              STRINGIFY(SAE_PASSPHRASE),
                              sizeof(ap_args->passphrase));
    (void)mmconfig_read_string("wlan.ap_password",
                               ap_args->passphrase,
                               sizeof(ap_args->passphrase));
    ap_args->passphrase_len = strlen(ap_args->passphrase);
 
    /* Load security type */
    ap_args->security_type = SECURITY_TYPE;
    if (mmconfig_read_string("wlan.ap_security", strval, sizeof(strval)) > 0)
    {
        if (strncmp("sae", strval, sizeof(strval)) == 0)
        {
            ap_args->security_type = MMWLAN_SAE;
        }
        else if (strncmp("owe", strval, sizeof(strval)) == 0)
        {
            ap_args->security_type = MMWLAN_OWE;
        }
        else if (strncmp("open", strval, sizeof(strval)) == 0)
        {
            ap_args->security_type = MMWLAN_OPEN;
        }
        else
        {
            printf("Invalid value of %s read from config store: %s\n", "wlan.ap_security", strval);
        }
    }
 
    /* Load PMF mode */
    ap_args->pmf_mode = PMF_MODE;
    if (mmconfig_read_string("wlan.ap_pmf_mode", strval, sizeof(strval)) > 0)
    {
        if (strncmp("disabled", strval, sizeof(strval)) == 0)
        {
            printf("PMF disabled\n");
            ap_args->pmf_mode = MMWLAN_PMF_DISABLED;
        }
        else if (strncmp("required", strval, sizeof(strval)) == 0)
        {
            ap_args->pmf_mode = MMWLAN_PMF_REQUIRED;
        }
        else
        {
            printf("Invalid value of %s read from config store: %s\n", "wlan.ap_pmf_mode", strval);
        }
    }
 
    /* Load BSSID */
    if (mmconfig_read_string("wlan.ap_bssid", strval, sizeof(strval)) > 0)
    {
        int temp[6];
        int i;
 
        int ret = sscanf(strval,
                         "%x:%x:%x:%x:%x:%x",
                         &temp[0],
                         &temp[1],
                         &temp[2],
                         &temp[3],
                         &temp[4],
                         &temp[5]);
        if (ret == 6)
        {
            for (i = 0; i < 6; i++)
            {
                if (temp[i] > UINT8_MAX || temp[i] < 0)
                {
                    /* Invalid value, ignore and reset to default */
                    memset(ap_args->bssid, 0, sizeof(ap_args->bssid));
                    break;
                }
 
                ap_args->bssid[i] = (uint8_t)temp[i];
            }
        }
    }
}
 
void app_init(void)
{
    MMLOG_PRINTF("\n\nRelay Mode Example (Built " __DATE__ " " __TIME__ ")\n\n");
 
    /* Initialize and connect to Wi-Fi, blocks till connected */
    app_wlan_init();
    app_wlan_start();
 
    /* Query STA channel info to configure the AP interface. */
    struct mmwlan_vif_channel_info channel_info = {};
    enum mmwlan_status status = mmwlan_get_vif_channel_info(MMWLAN_VIF_STA, &channel_info);
    if (status != MMWLAN_SUCCESS)
    {
        MMLOG_ERR("Failed to retrieve STA channel info\n", status);
        MMOSAL_ASSERT(0);
    }
 
    /* Bring up the relay subsystem. */
    struct mmwlan_relay_args args = MMWLAN_RELAY_ARGS_INIT;
    status = mmwlan_relay_enable(&args);
    if (status != MMWLAN_SUCCESS)
    {
        MMLOG_ERR("Failed to start relay (status %d)\n", status);
        MMOSAL_ASSERT(0);
    }
 
    /* Configure the AP args. */
    struct mmwlan_ap_args ap_args = MMWLAN_AP_ARGS_INIT;
    load_mmwlan_ap_args(&ap_args);
    mmwlan_set_power_save_mode(MMWLAN_PS_DISABLED);
 
    ap_args.op_class = channel_info.op_class;
    ap_args.s1g_chan_num = channel_info.s1g_chan_num;
    ap_args.pri_bw_mhz = channel_info.pri_bw_mhz;
    ap_args.pri_1mhz_chan_idx = channel_info.pri_1mhz_chan_idx;
 
    ap_args.sta_status_cb = handle_ap_sta_status;
    ap_args.sta_status_cb_arg = &opaque_argument_value;
 
    ap_args.max_stas = MAX_STAS;
 
    status = mmwlan_ap_enable(&ap_args);
    if (status == MMWLAN_SUCCESS)
    {
        MMLOG_PRINTF("AP started with SSID: %s ", ap_args.ssid);
        if (ap_args.security_type == MMWLAN_SAE)
        {
            MMLOG_PRINTF("with passphrase %s", ap_args.passphrase);
        }
        MMLOG_PRINTF("\n");
    }
    else
    {
        MMLOG_PRINTF("Failed to start AP (status %d)\n", status);
    }
}