esp-idf/examples/openthread/ot_cli/main/esp_ot_iperf.c

/*
 * SPDX-FileCopyrightText: 2021 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: CC0
 *
 * OpenThread Command Line Example
 *
 * 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 "esp_check.h"
#include "esp_log.h"
#include "esp_ot_iperf.h"
#include "iperf.h"
#include "openthread/cli.h"
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#define TAG "ot-iperf"
static char s_dest_ip6_addr[50];

void esp_ot_process_iperf(void *aContext, uint8_t aArgsLength, char *aArgs[])
{
    (void)(aContext);
    (void)(aArgsLength);
    iperf_cfg_t cfg;
    memset(&cfg, 0, sizeof(cfg));
    // set iperf default flag: tcp server
    IPERF_FLAG_SET(cfg.flag, IPERF_FLAG_TCP);
    IPERF_FLAG_SET(cfg.flag, IPERF_FLAG_SERVER);
    cfg.time = IPERF_DEFAULT_TIME;
    cfg.type = IPERF_IP_TYPE_IPV4;
    if (aArgsLength == 0) {
        otCliOutputFormat("---iperf parameter---\n");
        otCliOutputFormat("-V                  :     use IPV6 address\n");
        otCliOutputFormat("-s                  :     server mode, only receive\n");
        otCliOutputFormat("-u                  :     upd mode\n");
        otCliOutputFormat("-c <addr>           :     client mode, only transmit\n");
        otCliOutputFormat("-i <interval>       :     seconds between periodic bandwidth reports\n");
        otCliOutputFormat("-t <time>           :     time in seconds to transmit for (default 30 secs)\n");
        otCliOutputFormat("-p <port>           :     server port to listen on/connect to\n");
        otCliOutputFormat("-l <len_send_buf>   :     the lenth of send buffer\n");
        otCliOutputFormat("---example---\n");
        otCliOutputFormat("create a tcp server :     iperf -s -i 3 -p 5001 -t 60\n");
        otCliOutputFormat("create a udp client :     iperf -c <addr> -u -i 3 -t 60 -p 5001 -l 512\n");
    } else {
        for (int i = 0; i < aArgsLength; i++) {
            if (strcmp(aArgs[i], "-c") == 0) {
                IPERF_FLAG_SET(cfg.flag, IPERF_FLAG_CLIENT);
                IPERF_FLAG_CLR(cfg.flag, IPERF_FLAG_SERVER);
                i++;
                strcpy(s_dest_ip6_addr, aArgs[i]);
                cfg.destination_ip6 = s_dest_ip6_addr;
                otCliOutputFormat("ip:%s\n", cfg.destination_ip6);
            } else if (strcmp(aArgs[i], "-s") == 0) {
                IPERF_FLAG_SET(cfg.flag, IPERF_FLAG_SERVER);
                IPERF_FLAG_CLR(cfg.flag, IPERF_FLAG_CLIENT);
            } else if (strcmp(aArgs[i], "-V") == 0) {
                cfg.type = IPERF_IP_TYPE_IPV6;
            } else if (strcmp(aArgs[i], "-u") == 0) {
                IPERF_FLAG_SET(cfg.flag, IPERF_FLAG_UDP);
                IPERF_FLAG_CLR(cfg.flag, IPERF_FLAG_TCP);
            } else if (strcmp(aArgs[i], "-p") == 0) {
                i++;
                if (cfg.flag & IPERF_FLAG_SERVER) {
                    cfg.sport = atoi(aArgs[i]);
                    cfg.dport = IPERF_DEFAULT_PORT;
                } else {
                    cfg.sport = IPERF_DEFAULT_PORT;
                    cfg.dport = atoi(aArgs[i]);
                }
                otCliOutputFormat("dp:%d\n", cfg.dport);
                otCliOutputFormat("sp:%d\n", cfg.sport);
            } else if (strcmp(aArgs[i], "-i") == 0) {
                i++;
                if (atoi(aArgs[i]) < 0) {
                    cfg.interval = IPERF_DEFAULT_INTERVAL;
                } else {
                    cfg.interval = atoi(aArgs[i]);
                }
                otCliOutputFormat("i:%d\n", cfg.interval);
            } else if (strcmp(aArgs[i], "-t") == 0) {
                i++;
                cfg.time = atoi(aArgs[i]);
                if (cfg.time <= cfg.interval) {
                    cfg.time = cfg.interval;
                }
                otCliOutputFormat("t:%d\n", cfg.time);
            } else if (strcmp(aArgs[i], "-l") == 0) {
                i++;
                if (atoi(aArgs[i]) <= 0) {
                    ESP_LOGE(TAG, "Invalid arguments.");
                } else {
                    cfg.len_send_buf = atoi(aArgs[i]);
                }
            } else if (strcmp(aArgs[i], "-a") == 0) {
                iperf_stop();
                return;
            }
        }
        iperf_start(&cfg);
    }
}