基于STM32F407MAC与DP83848实现以太网通讯六（IPerf网络速度测试）
一、IPerf2网络测试工具
Iperf2是一个用于测试网络带宽的工具。它是Iperf的旧版本，专注于提供基本的带宽测量功能。通过在客户端和服务器之间发送测试数据流并测量其性能，用户可以评估网络连接的速度和稳定性。Iperf2提供了一种简单而有效的方式来评估网络性能。
IPerf3已经发布了，但是我们测试网络的代码是基于IPerf2实现的，因此还是使用IPerf2进行网络测试。
- 测量带宽：IPerf2可以测试网络的带宽，帮助用户了解网络的传输速率。
- 测量延迟：除了带宽，IPerf2还可以测量网络的延迟，即数据从源到目的地所需的时间。
- 测试网络稳定性：IPerf2可以用于测试网络的稳定性和可靠性，以评估网络连接的质量。
- 多种测试模式：IPerf2支持多种测试模式，包括TCP和UDP，用户可以根据需要选择适合的测试模式。
二、lwiperf.c测试代码的获取
在lwip官网下载的文件中lwip-2.1.2\src\apps\lwiperf中存放了iperf的测试文件，但是当前文件只实现了TCP的速率测试，本文使用的是先楫HPM6750EVK开发板提供的SDK中提供的lwiperf.c文件，该文件在官方的lwiperf.c文件上实现了对于UDP性能的测试，其测试例程也是参考了先楫HPM6750EVK开发板的测试例程。
SDK的使用参考上述链接中的野火文档，这里就不赘述了，我们只需要将SDK生成的IPerf例程的Demo和lwiperf.c源文件拿出来参考。
先楫文档的测试结果如下
三、IPerf测试
lwiperf.c源文件如下：
lwiperf.c
/**
* @file
* lwIP iPerf server implementation
*/
/**
* @defgroup iperf Iperf server
* @ingroup apps
*
* This is a simple performance measuring client/server to check your bandwith using
* iPerf2 on a PC as server/client.
* It is currently a minimal implementation providing a TCP client/server only.
*
* @todo:
* - implement UDP mode
* - protect combined sessions handling (via 'related_master_state') against reallocation
* (this is a pointer address, currently, so if the same memory is allocated again,
* session pairs (tx/rx) can be confused on reallocation)
*/
/*
* Copyright (c) 2014 Simon Goldschmidt
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* 3. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
* SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
* OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
* OF SUCH DAMAGE.
*
* This file is part of the lwIP TCP/IP stack.
*
* Author: Simon Goldschmidt
*/
#include "lwiperf.h"
#include "lwip/udp.h"
#include "lwip/tcp.h"
#include "lwip/sys.h"
#include "lwip/inet.h"
#include "lwip/timeouts.h"
#include "lwip/igmp.h"
#include "lwip/mld6.h"
#include <string.h>
long udp_last_lost_time = 0;
long udp_start_lost_time = 0;
/* Currently, only TCP is implemented */
#if LWIP_TCP && LWIP_CALLBACK_API
/** Specify the idle timeout (in seconds) after that the test fails */
#ifndef LWIPERF_TCP_MAX_IDLE_SEC
#define LWIPERF_TCP_MAX_IDLE_SEC 10U
#endif
#if LWIPERF_TCP_MAX_IDLE_SEC > 255
#error LWIPERF_TCP_MAX_IDLE_SEC must fit into an u8_t
#endif
/** Change this if you don't want to lwiperf to listen to any IP version */
#ifndef LWIPERF_SERVER_IP_TYPE
#define LWIPERF_SERVER_IP_TYPE IPADDR_TYPE_ANY
#endif
/* File internal memory allocation (struct lwiperf_*): this defaults to
the heap */
#ifndef LWIPERF_ALLOC
#define LWIPERF_ALLOC(type) mem_malloc(sizeof(type))
#define LWIPERF_FREE(type, item) mem_free(item)
#endif
/** If this is 1, check that received data has the correct format */
#ifndef LWIPERF_CHECK_RX_DATA
#define LWIPERF_CHECK_RX_DATA 0
#endif
/** The IDs of clocks used by clock_gettime() */
#define CLOCK_ID 1
/** The resolution of the clock in microseconds */
#define CLOCK_RESOLUTION_US 1000u
/** Added for compatibility */
struct lwip_timespec {
long tv_sec;
long tv_nsec;
};
/** This is the Iperf settings struct sent from the client */
typedef struct _lwiperf_settings {
#define LWIPERF_FLAGS_ANSWER_TEST 0x80000000
#define LWIPERF_FLAGS_EXTEND 0x40000000
#define LWIPERF_FLAGS_ANSWER_NOW 0x00000001
u32_t flags;
u32_t num_threads; /* unused for now */
u32_t remote_port;
u32_t buffer_len; /* unused for now */
u32_t win_band; /* TCP window / UDP rate: unused for now */
u32_t amount; /* pos. value: bytes?; neg. values: time (unit is 10ms: 1/100 second) */
} lwiperf_settings_t;
typedef struct _lwiperf_settings_ext {
#define LWIPERF_EFLAGS_UNITS_PPS 0x00000001
lwiperf_settings_t base;
s32_t type;
s32_t len; /* length from flags to real_time */
s32_t eflags;
s32_t version_u;
s32_t version_l;
s32_t reserved;
s32_t rate;
s32_t UDP_rate_units;
s32_t real_time;
} lwiperf_settings_ext_t;
/** This is the header structure for all UDP datagram */
struct UDP_datagram {
int32_t id;
uint32_t tv_sec;
uint32_t tv_usec;
};
/** This is the Iperf reporting struct sent to the client */
typedef struct _lwiperf_udp_report {
int32_t flags; /* LWIPERF_FLAGS_* */
int32_t total_len1;
int32_t total_len2;
int32_t stop_sec;
int32_t stop_usec;
int32_t error_cnt;
int32_t outorder_cnt;
int32_t datagrams;
int32_t jitter1;
int32_t jitter2;
} lwiperf_udp_report_t;
/** Basic connection handle */
struct _lwiperf_state_base;
typedef struct _lwiperf_state_base lwiperf_state_base_t;
struct _lwiperf_state_base {
/* linked list */
lwiperf_state_base_t *next;
/* 1=tcp, 0=udp */
u8_t tcp;
/* 1=server, 0=client */
u8_t server;
/* master state used to abort sessions (e.g. listener, main client) */
lwiperf_state_base_t *related_master_state;
};
/** Connection handle for a UDP iperf session */
typedef struct _lwiperf_state_udp {
lwiperf_state_base_t base;
struct udp_pcb *pcb;
struct pbuf *reported;
ip_addr_t remote_addr;
u16_t remote_port;
u8_t report_count;
u8_t have_settings_buf;
lwiperf_settings_ext_t settings;
u32_t delay_target;
u32_t frames_per_delay;
u32_t time_started;
u64_t bytes_transferred;
lwiperf_report_fn report_fn;
void *report_arg;
struct lwip_timespec udp_lastpkt;
u32_t udp_seq;
u32_t udp_rx_lost;
u32_t udp_rx_outorder;
u32_t udp_rx_total_pkt;
u32_t udp_rx_total_size;
u32_t udp_last_transit;
long jitter;
} lwiperf_state_udp_t;
/** Connection handle for a TCP iperf session */
typedef struct _lwiperf_state_tcp {
lwiperf_state_base_t base;
struct tcp_pcb *server_pcb;
struct tcp_pcb *conn_pcb;
u32_t time_started;
lwiperf_report_fn report_fn;
void *report_arg;
u8_t poll_count;
u8_t next_num;
/* 1=start server when client is closed */
u8_t client_tradeoff_mode;
u32_t bytes_transferred;
lwiperf_settings_t settings;
u8_t have_settings_buf;
u8_t specific_remote;
ip_addr_t remote_addr;
} lwiperf_state_tcp_t;
/** List of active iperf sessions */
static lwiperf_state_base_t *lwiperf_all_connections;
/** A const buffer to send from: we want to measure sending, not copying! */
static const u8_t lwiperf_txbuf_const[1600] = {
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
};
static err_t lwiperf_tcp_poll(void *arg, struct tcp_pcb *tpcb);
static void lwiperf_tcp_err(void *arg, err_t err);
static err_t lwiperf_start_tcp_server_impl(const ip_addr_t *local_addr, u16_t local_port,
lwiperf_report_fn report_fn, void *report_arg,
lwiperf_state_base_t *related_master_state, lwiperf_state_tcp_t **state);
static void lwiperf_udp_recv(void *arg, struct udp_pcb *pcb, struct pbuf *p,
const ip_addr_t *addr, u16_t port);
static err_t lwiperf_udp_tx_start(lwiperf_state_udp_t *conn);
static int
clock_gettime(int clk_id, struct lwip_timespec *tp)
{
u32_t now = sys_now();
tp->tv_sec = now / 1000;
tp->tv_nsec = (now % 1000) * 1000000;
return 0;
}
static inline void
diff_ts(const struct lwip_timespec *start, const struct lwip_timespec *stop, struct lwip_timespec *result)
{
if ((stop->tv_nsec - start->tv_nsec) < 0) {
result->tv_sec = stop->tv_sec - start->tv_sec - 1;
result->tv_nsec = stop->tv_nsec - start->tv_nsec + 1000000000;
} else {
result->tv_sec = stop->tv_sec - start->tv_sec;
result->tv_nsec = stop->tv_nsec - start->tv_nsec;
}
}
/** Add an iperf session to the 'active' list */
static void
lwiperf_list_add(lwiperf_state_base_t *item)
{
item->next = lwiperf_all_connections;
lwiperf_all_connections = item;
}
/** Remove an iperf session from the 'active' list */
static void
lwiperf_list_remove(lwiperf_state_base_t *item)
{
lwiperf_state_base_t *prev = NULL;
lwiperf_state_base_t *iter;
for (iter = lwiperf_all_connections; iter != NULL; prev = iter, iter = iter->next) {
if (iter == item) {
if (prev == NULL) {
lwiperf_all_connections = iter->next;
} else {
prev->next = iter->next;
}
/* @debug: ensure this item is listed only once */
for (iter = iter->next; iter != NULL; iter = iter->next) {
LWIP_ASSERT("duplicate entry", iter != item);
}
break;
}
}
}
static lwiperf_state_base_t *
lwiperf_list_find(lwiperf_state_base_t *item)
{
lwiperf_state_base_t *iter;
for (iter = lwiperf_all_connections; iter != NULL; iter = iter->next) {
if (iter == item) {
return item;
}
}
return NULL;
}
/** Call the report function of an iperf tcp session */
static void
lwip_tcp_conn_report(lwiperf_state_tcp_t *conn, enum lwiperf_report_type report_type)
{
if ((conn != NULL) && (conn->report_fn != NULL)) {
u32_t now, duration_ms, bandwidth_kbitpsec;
now = sys_now();
duration_ms = now - conn->time_started;
if (duration_ms == 0) {
bandwidth_kbitpsec = 0;
} else {
bandwidth_kbitpsec = (conn->bytes_transferred / duration_ms) * 8U;
}
conn->report_fn(conn->report_arg, report_type,
&conn->conn_pcb->local_ip, conn->conn_pcb->local_port,
&conn->conn_pcb->remote_ip, conn->conn_pcb->remote_port,
conn->bytes_transferred, duration_ms, bandwidth_kbitpsec);
}
}
/** Close an iperf tcp session */
static void
lwiperf_tcp_close(lwiperf_state_tcp_t *conn, enum lwiperf_report_type report_type)
{
err_t err;
lwiperf_list_remove(&conn->base);
lwip_tcp_conn_report(conn, report_type);
if (conn->conn_pcb != NULL) {
tcp_arg(conn->conn_pcb, NULL);
tcp_poll(conn->conn_pcb, NULL, 0);
tcp_sent(conn->conn_pcb, NULL);
tcp_recv(conn->conn_pcb, NULL);
tcp_err(conn->conn_pcb, NULL);
err = tcp_close(conn->conn_pcb);
if (err != ERR_OK) {
/* don't want to wait for free memory here... */
tcp_abort(conn->conn_pcb);
}
} else {
/* no conn pcb, this is the listener pcb */
err = tcp_close(conn->server_pcb);
LWIP_ASSERT("error", err == ERR_OK);
}
LWIPERF_FREE(lwiperf_state_tcp_t, conn);
}
/** Try to send more data on an iperf tcp session */
static err_t
lwiperf_tcp_client_send_more(lwiperf_state_tcp_t *conn)
{
int send_more;
err_t err;
u16_t txlen;
u16_t txlen_max;
void *txptr;
u8_t apiflags;
LWIP_ASSERT("conn invalid", (conn != NULL) && conn->base.tcp && (conn->base.server == 0));
do {
send_more = 0;
if (conn->settings.amount & PP_HTONL(0x80000000)) {
/* this session is time-limited */
u32_t now = sys_now();
u32_t diff_ms = now - conn->time_started;
u32_t time = (u32_t) - (s32_t)lwip_htonl(conn->settings.amount);
u32_t time_ms = time * 10;
if (diff_ms >= time_ms) {
/* time specified by the client is over -> close the connection */
lwiperf_tcp_close(conn, LWIPERF_TCP_DONE_CLIENT);
return ERR_OK;
}
} else {
/* this session is byte-limited */
u32_t amount_bytes = lwip_htonl(conn->settings.amount);
/* @todo: this can send up to 1*MSS more than requested... */
if (amount_bytes >= conn->bytes_transferred) {
/* all requested bytes transferred -> close the connection */
lwiperf_tcp_close(conn, LWIPERF_TCP_DONE_CLIENT);
return ERR_OK;
}
}
if (conn->bytes_transferred < 24) {
/* transmit the settings a first time */
txptr = &((u8_t *)&conn->settings)[conn->bytes_transferred];
txlen_max = (u16_t)(24 - conn->bytes_transferred);
apiflags = TCP_WRITE_FLAG_COPY;
} else if (conn->bytes_transferred < 48) {
/* transmit the settings a second time */
txptr = &((u8_t *)&conn->settings)[conn->bytes_transferred - 24];
txlen_max = (u16_t)(48 - conn->bytes_transferred);
apiflags = TCP_WRITE_FLAG_COPY | TCP_WRITE_FLAG_MORE;
send_more = 1;
} else {
/* transmit data */
/* @todo: every x bytes, transmit the settings again */
txptr = LWIP_CONST_CAST(void *, &lwiperf_txbuf_const[conn->bytes_transferred % 10]);
txlen_max = TCP_MSS;
if (conn->bytes_transferred == 48) { /* @todo: fix this for intermediate settings, too */
txlen_max = TCP_MSS - 24;
}
apiflags = 0; /* no copying needed */
send_more = 1;
}
txlen = txlen_max;
do {
err = tcp_write(conn->conn_pcb, txptr, txlen, apiflags);
if (err == ERR_MEM) {
txlen /= 2;
}
} while ((err == ERR_MEM) && (txlen >= (TCP_MSS / 2)));
if (err == ERR_OK) {
conn->bytes_transferred += txlen;
} else {
send_more = 0;
}
} while (send_more);
tcp_output(conn->conn_pcb);
return ERR_OK;
}
/** TCP sent callback, try to send more data */
static err_t
lwiperf_tcp_client_sent(void *arg, struct tcp_pcb *tpcb, u16_t len)
{
lwiperf_state_tcp_t *conn = (lwiperf_state_tcp_t *)arg;
/* @todo: check 'len' (e.g. to time ACK of all data)? for now, we just send more... */
LWIP_ASSERT("invalid conn", conn->conn_pcb == tpcb);
LWIP_UNUSED_ARG(tpcb);
LWIP_UNUSED_ARG(len);
conn->poll_count = 0;
return lwiperf_tcp_client_send_more(conn);
}
/** TCP connected callback (active connection), send data now */
static err_t
lwiperf_tcp_client_connected(void *arg, struct tcp_pcb *tpcb, err_t err)
{
lwiperf_state_tcp_t *conn = (lwiperf_state_tcp_t *)arg;
LWIP_ASSERT("invalid conn", conn->conn_pcb == tpcb);
LWIP_UNUSED_ARG(tpcb);
if (err != ERR_OK) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_REMOTE);
return ERR_OK;
}
conn->poll_count = 0;
conn->time_started = sys_now();
return lwiperf_tcp_client_send_more(conn);
}
/** Start TCP connection back to the client (either parallel or after the
* receive test has finished.
*/
static err_t
lwiperf_tx_start_impl(const ip_addr_t *remote_ip, u16_t remote_port, lwiperf_settings_t *settings, lwiperf_report_fn report_fn,
void *report_arg, lwiperf_state_base_t *related_master_state, lwiperf_state_tcp_t **new_conn)
{
err_t err;
lwiperf_state_tcp_t *client_conn;
struct tcp_pcb *newpcb;
ip_addr_t remote_addr;
LWIP_ASSERT("remote_ip != NULL", remote_ip != NULL);
LWIP_ASSERT("remote_ip != NULL", settings != NULL);
LWIP_ASSERT("new_conn != NULL", new_conn != NULL);
*new_conn = NULL;
client_conn = (lwiperf_state_tcp_t *)LWIPERF_ALLOC(lwiperf_state_tcp_t);
if (client_conn == NULL) {
return ERR_MEM;
}
newpcb = tcp_new_ip_type(IP_GET_TYPE(remote_ip));
if (newpcb == NULL) {
LWIPERF_FREE(lwiperf_state_tcp_t, client_conn);
return ERR_MEM;
}
memset(client_conn, 0, sizeof(lwiperf_state_tcp_t));
client_conn->base.tcp = 1;
client_conn->base.related_master_state = related_master_state;
client_conn->conn_pcb = newpcb;
client_conn->time_started = sys_now(); /* @todo: set this again on 'connected' */
client_conn->report_fn = report_fn;
client_conn->report_arg = report_arg;
client_conn->next_num = 4; /* initial nr is '4' since the header has 24 byte */
client_conn->bytes_transferred = 0;
memcpy(&client_conn->settings, settings, sizeof(*settings));
client_conn->have_settings_buf = 1;
tcp_arg(newpcb, client_conn);
tcp_sent(newpcb, lwiperf_tcp_client_sent);
tcp_poll(newpcb, lwiperf_tcp_poll, 2U);
tcp_err(newpcb, lwiperf_tcp_err);
ip_addr_copy(remote_addr, *remote_ip);
err = tcp_connect(newpcb, &remote_addr, remote_port, lwiperf_tcp_client_connected);
if (err != ERR_OK) {
lwiperf_tcp_close(client_conn, LWIPERF_TCP_ABORTED_LOCAL);
return err;
}
lwiperf_list_add(&client_conn->base);
*new_conn = client_conn;
return ERR_OK;
}
static err_t
lwiperf_tx_start_passive(lwiperf_state_tcp_t *conn)
{
err_t ret;
lwiperf_state_tcp_t *new_conn = NULL;
u16_t remote_port = (u16_t)lwip_htonl(conn->settings.remote_port);
ret = lwiperf_tx_start_impl(&conn->conn_pcb->remote_ip, remote_port, &conn->settings, conn->report_fn, conn->report_arg,
conn->base.related_master_state, &new_conn);
if (ret == ERR_OK) {
LWIP_ASSERT("new_conn != NULL", new_conn != NULL);
new_conn->settings.flags = 0; /* prevent the remote side starting back as client again */
}
return ret;
}
/** Receive data on an iperf tcp session */
static err_t
lwiperf_tcp_recv(void *arg, struct tcp_pcb *tpcb, struct pbuf *p, err_t err)
{
u8_t tmp;
u16_t tot_len;
u32_t packet_idx;
struct pbuf *q;
lwiperf_state_tcp_t *conn = (lwiperf_state_tcp_t *)arg;
LWIP_ASSERT("pcb mismatch", conn->conn_pcb == tpcb);
LWIP_UNUSED_ARG(tpcb);
if (err != ERR_OK) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_REMOTE);
return ERR_OK;
}
if (p == NULL) {
/* connection closed -> test done */
if (conn->settings.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST)) {
if ((conn->settings.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_NOW)) == 0) {
/* client requested transmission after end of test */
lwiperf_tx_start_passive(conn);
}
}
lwiperf_tcp_close(conn, LWIPERF_TCP_DONE_SERVER);
return ERR_OK;
}
tot_len = p->tot_len;
conn->poll_count = 0;
if ((!conn->have_settings_buf) || ((conn->bytes_transferred - 24) % (1024 * 128) == 0)) {
/* wait for 24-byte header */
if (p->tot_len < sizeof(lwiperf_settings_t)) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_LOCAL_DATAERROR);
pbuf_free(p);
return ERR_OK;
}
if (!conn->have_settings_buf) {
if (pbuf_copy_partial(p, &conn->settings, sizeof(lwiperf_settings_t), 0) != sizeof(lwiperf_settings_t)) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_LOCAL);
pbuf_free(p);
return ERR_OK;
}
conn->have_settings_buf = 1;
if (conn->settings.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST)) {
if (conn->settings.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_NOW)) {
/* client requested parallel transmission test */
err_t err2 = lwiperf_tx_start_passive(conn);
if (err2 != ERR_OK) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_LOCAL_TXERROR);
pbuf_free(p);
return ERR_OK;
}
}
}
} else {
if (conn->settings.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST)) {
if (pbuf_memcmp(p, 0, &conn->settings, sizeof(lwiperf_settings_t)) != 0) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_LOCAL_DATAERROR);
pbuf_free(p);
return ERR_OK;
}
}
}
conn->bytes_transferred += sizeof(lwiperf_settings_t);
if (conn->bytes_transferred <= 24) {
conn->time_started = sys_now();
tcp_recved(tpcb, p->tot_len);
pbuf_free(p);
return ERR_OK;
}
conn->next_num = 4; /* 24 bytes received... */
tmp = pbuf_remove_header(p, 24);
LWIP_ASSERT("pbuf_remove_header failed", tmp == 0);
LWIP_UNUSED_ARG(tmp); /* for LWIP_NOASSERT */
}
packet_idx = 0;
for (q = p; q != NULL; q = q->next) {
#if LWIPERF_CHECK_RX_DATA
const u8_t *payload = (const u8_t *)q->payload;
u16_t i;
for (i = 0; i < q->len; i++) {
u8_t val = payload[i];
u8_t num = val - '0';
if (num == conn->next_num) {
conn->next_num++;
if (conn->next_num == 10) {
conn->next_num = 0;
}
} else {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_LOCAL_DATAERROR);
pbuf_free(p);
return ERR_OK;
}
}
#endif
packet_idx += q->len;
}
LWIP_ASSERT("count mismatch", packet_idx == p->tot_len);
conn->bytes_transferred += packet_idx;
tcp_recved(tpcb, tot_len);
pbuf_free(p);
return ERR_OK;
}
/** Error callback, iperf tcp session aborted */
static void
lwiperf_tcp_err(void *arg, err_t err)
{
lwiperf_state_tcp_t *conn = (lwiperf_state_tcp_t *)arg;
LWIP_UNUSED_ARG(err);
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_REMOTE);
}
/** TCP poll callback, try to send more data */
static err_t
lwiperf_tcp_poll(void *arg, struct tcp_pcb *tpcb)
{
lwiperf_state_tcp_t *conn = (lwiperf_state_tcp_t *)arg;
LWIP_ASSERT("pcb mismatch", conn->conn_pcb == tpcb);
LWIP_UNUSED_ARG(tpcb);
if (++conn->poll_count >= LWIPERF_TCP_MAX_IDLE_SEC) {
lwiperf_tcp_close(conn, LWIPERF_TCP_ABORTED_LOCAL);
return ERR_OK; /* lwiperf_tcp_close frees conn */
}
if (!conn->base.server) {
lwiperf_tcp_client_send_more(conn);
}
return ERR_OK;
}
/** This is called when a new client connects for an iperf tcp session */
static err_t
lwiperf_tcp_accept(void *arg, struct tcp_pcb *newpcb, err_t err)
{
lwiperf_state_tcp_t *s, *conn;
if ((err != ERR_OK) || (newpcb == NULL) || (arg == NULL)) {
return ERR_VAL;
}
s = (lwiperf_state_tcp_t *)arg;
LWIP_ASSERT("invalid session", s->base.server);
LWIP_ASSERT("invalid listen pcb", s->server_pcb != NULL);
LWIP_ASSERT("invalid conn pcb", s->conn_pcb == NULL);
if (s->specific_remote) {
LWIP_ASSERT("s->base.related_master_state != NULL", s->base.related_master_state != NULL);
if (!ip_addr_cmp(&newpcb->remote_ip, &s->remote_addr)) {
/* this listener belongs to a client session, and this is not the correct remote */
return ERR_VAL;
}
} else {
LWIP_ASSERT("s->base.related_master_state == NULL", s->base.related_master_state == NULL);
}
conn = (lwiperf_state_tcp_t *)LWIPERF_ALLOC(lwiperf_state_tcp_t);
if (conn == NULL) {
return ERR_MEM;
}
memset(conn, 0, sizeof(lwiperf_state_tcp_t));
conn->base.tcp = 1;
conn->base.server = 1;
conn->base.related_master_state = &s->base;
conn->conn_pcb = newpcb;
conn->time_started = sys_now();
conn->report_fn = s->report_fn;
conn->report_arg = s->report_arg;
/* setup the tcp rx connection */
tcp_arg(newpcb, conn);
tcp_recv(newpcb, lwiperf_tcp_recv);
tcp_poll(newpcb, lwiperf_tcp_poll, 2U);
tcp_err(conn->conn_pcb, lwiperf_tcp_err);
if (s->specific_remote) {
/* this listener belongs to a client, so make the client the master of the newly created connection */
conn->base.related_master_state = s->base.related_master_state;
/* if dual mode or (tradeoff mode AND client is done): close the listener */
if (!s->client_tradeoff_mode || !lwiperf_list_find(s->base.related_master_state)) {
/* prevent report when closing: this is expected */
s->report_fn = NULL;
lwiperf_tcp_close(s, LWIPERF_TCP_ABORTED_LOCAL);
}
}
lwiperf_list_add(&conn->base);
return ERR_OK;
}
/**
* @ingroup iperf
* Start a TCP iperf server on the default TCP port (5001) and listen for
* incoming connections from iperf clients.
*
* @returns a connection handle that can be used to abort the server
* by calling @ref lwiperf_abort()
*/
void *
lwiperf_start_tcp_server_default(lwiperf_report_fn report_fn, void *report_arg)
{
return lwiperf_start_tcp_server(IP_ADDR_ANY, LWIPERF_TCP_PORT_DEFAULT,
report_fn, report_arg);
}
/**
* @ingroup iperf
* Start a TCP iperf server on a specific IP address and port and listen for
* incoming connections from iperf clients.
*
* @returns a connection handle that can be used to abort the server
* by calling @ref lwiperf_abort()
*/
void *
lwiperf_start_tcp_server(const ip_addr_t *local_addr, u16_t local_port,
lwiperf_report_fn report_fn, void *report_arg)
{
err_t err;
lwiperf_state_tcp_t *state = NULL;
err = lwiperf_start_tcp_server_impl(local_addr, local_port, report_fn, report_arg,
NULL, &state);
if (err == ERR_OK) {
return state;
}
return NULL;
}
static err_t lwiperf_start_tcp_server_impl(const ip_addr_t *local_addr, u16_t local_port,
lwiperf_report_fn report_fn, void *report_arg,
lwiperf_state_base_t *related_master_state, lwiperf_state_tcp_t **state)
{
err_t err;
struct tcp_pcb *pcb;
lwiperf_state_tcp_t *s;
LWIP_ASSERT_CORE_LOCKED();
LWIP_ASSERT("state != NULL", state != NULL);
if (local_addr == NULL) {
return ERR_ARG;
}
s = (lwiperf_state_tcp_t *)LWIPERF_ALLOC(lwiperf_state_tcp_t);
if (s == NULL) {
return ERR_MEM;
}
memset(s, 0, sizeof(lwiperf_state_tcp_t));
s->base.tcp = 1;
s->base.server = 1;
s->base.related_master_state = related_master_state;
s->report_fn = report_fn;
s->report_arg = report_arg;
pcb = tcp_new_ip_type(LWIPERF_SERVER_IP_TYPE);
if (pcb == NULL) {
return ERR_MEM;
}
err = tcp_bind(pcb, local_addr, local_port);
if (err != ERR_OK) {
return err;
}
s->server_pcb = tcp_listen_with_backlog(pcb, 1);
if (s->server_pcb == NULL) {
if (pcb != NULL) {
tcp_close(pcb);
}
LWIPERF_FREE(lwiperf_state_tcp_t, s);
return ERR_MEM;
}
pcb = NULL;
tcp_arg(s->server_pcb, s);
tcp_accept(s->server_pcb, lwiperf_tcp_accept);
lwiperf_list_add(&s->base);
*state = s;
return ERR_OK;
}
/**
* @ingroup iperf
* Start a TCP iperf client to the default TCP port (5001).
*
* @returns a connection handle that can be used to abort the client
* by calling @ref lwiperf_abort()
*/
void* lwiperf_start_tcp_client_default(const ip_addr_t* remote_addr,
lwiperf_report_fn report_fn, void* report_arg)
{
return lwiperf_start_tcp_client(remote_addr, LWIPERF_TCP_PORT_DEFAULT, LWIPERF_CLIENT,
report_fn, report_arg);
}
/**
* @ingroup iperf
* Start a TCP iperf client to a specific IP address and port.
*
* @returns a connection handle that can be used to abort the client
* by calling @ref lwiperf_abort()
*/
void* lwiperf_start_tcp_client(const ip_addr_t* remote_addr, u16_t remote_port,
enum lwiperf_client_type type, lwiperf_report_fn report_fn, void* report_arg)
{
err_t ret;
lwiperf_settings_t settings;
lwiperf_state_tcp_t *state = NULL;
memset(&settings, 0, sizeof(settings));
switch (type) {
case LWIPERF_CLIENT:
/* Unidirectional tx only test */
settings.flags = 0;
break;
case LWIPERF_DUAL:
/* Do a bidirectional test simultaneously */
settings.flags = htonl(LWIPERF_FLAGS_ANSWER_TEST | LWIPERF_FLAGS_ANSWER_NOW);
break;
case LWIPERF_TRADEOFF:
/* Do a bidirectional test individually */
settings.flags = htonl(LWIPERF_FLAGS_ANSWER_TEST);
break;
default:
/* invalid argument */
return NULL;
}
settings.num_threads = htonl(1);
settings.remote_port = htonl(LWIPERF_TCP_PORT_DEFAULT);
/* TODO: implement passing duration/amount of bytes to transfer */
settings.amount = htonl((u32_t)-1000);
ret = lwiperf_tx_start_impl(remote_addr, remote_port, &settings, report_fn, report_arg, NULL, &state);
if (ret == ERR_OK) {
LWIP_ASSERT("state != NULL", state != NULL);
if (type != LWIPERF_CLIENT) {
/* start corresponding server now */
lwiperf_state_tcp_t *server = NULL;
ret = lwiperf_start_tcp_server_impl(&state->conn_pcb->local_ip, LWIPERF_TCP_PORT_DEFAULT,
report_fn, report_arg, (lwiperf_state_base_t *)state, &server);
if (ret != ERR_OK) {
/* starting server failed, abort client */
lwiperf_abort(state);
return NULL;
}
/* make this server accept one connection only */
server->specific_remote = 1;
server->remote_addr = state->conn_pcb->remote_ip;
if (type == LWIPERF_TRADEOFF) {
/* tradeoff means that the remote host connects only after the client is done,
so keep the listen pcb open until the client is done */
server->client_tradeoff_mode = 1;
}
}
return state;
}
return NULL;
}
/** This is called when a new client connects for an iperf udp session */
static lwiperf_state_udp_t *
lwiperf_udp_new_client(lwiperf_state_udp_t *s)
{
lwiperf_state_udp_t *conn;
if (s == NULL)
return NULL;
conn = (lwiperf_state_udp_t *)LWIPERF_ALLOC(lwiperf_state_udp_t);
if (conn == NULL)
return NULL;
memset(conn, 0, sizeof(lwiperf_state_udp_t));
conn->base.tcp = 0;
conn->base.server = 1;
conn->base.related_master_state = &s->base;
conn->pcb = NULL;
conn->time_started = sys_now();
conn->report_fn = s->report_fn;
conn->report_arg = s->report_arg;
lwiperf_list_add(&conn->base);
return conn;
}
/** This is called when a new client connects for an iperf udp session */
static lwiperf_state_udp_t *
lwiperf_udp_search_client(lwiperf_state_udp_t *s,
const ip_addr_t *addr, u16_t port)
{
lwiperf_state_udp_t *iter;
for (iter = (lwiperf_state_udp_t *)lwiperf_all_connections;
iter != NULL;
iter = (lwiperf_state_udp_t *)iter->base.next) {
if (iter->base.tcp || !iter->base.related_master_state)
continue;
if (iter->base.related_master_state != (lwiperf_state_base_t *)s)
continue;
if (ip_addr_cmp(addr, &iter->remote_addr) && (port == iter->remote_port))
break;
}
return iter;
}
/** Call the report function of an iperf tcp session */
static void
lwip_udp_conn_report(lwiperf_state_udp_t *conn, enum lwiperf_report_type report_type)
{
if ((conn != NULL) && (conn->report_fn != NULL)) {
u32_t now, duration_ms, bandwidth_kbitpsec;
now = sys_now();
duration_ms = now - conn->time_started;
if (duration_ms == 0) {
bandwidth_kbitpsec = 0;
} else {
bandwidth_kbitpsec = (u32_t)((8ull * conn->bytes_transferred) / duration_ms);
}
ip_addr_t *local_ip = NULL;
u16_t local_port = 0u;
if (conn->pcb != NULL) {
local_ip = &(conn->pcb->local_ip);
local_port = conn->pcb->local_port;
} else if (conn->base.related_master_state != NULL) {
/* conn->pcb is NULL for incoming UDP clients, try to take local IP
* and port number from server pcb stored in related master state
*/
lwiperf_state_udp_t *s = (lwiperf_state_udp_t *)conn->base.related_master_state;
if (s->pcb != NULL) {
local_ip = &(s->pcb->local_ip);
local_port = s->pcb->local_port;
}
}
conn->report_fn(conn->report_arg, report_type,
local_ip, local_port,
&conn->remote_addr, conn->remote_port,
conn->bytes_transferred, duration_ms, bandwidth_kbitpsec);
}
}
/** Close an iperf udp session */
static void
lwiperf_udp_close(lwiperf_state_udp_t *conn, enum lwiperf_report_type report_type)
{
lwip_udp_conn_report(conn, report_type);
lwiperf_list_remove(&conn->base);
if (conn->pcb != NULL) {
ip_addr_t *local_addr = &conn->pcb->local_ip;
if (ip_addr_ismulticast(local_addr)) {
if (IP_IS_V6 (local_addr)) {
#if LWIP_IPV6_MLD
mld6_leavegroup (IP6_ADDR_ANY6, ip_2_ip6 (local_addr));
#endif
}
else {
#if LWIP_IGMP
igmp_leavegroup (IP4_ADDR_ANY4, ip_2_ip4 (local_addr));
#endif
}
}
udp_remove(conn->pcb);
conn->pcb = NULL;
}
LWIPERF_FREE(lwiperf_state_udp_t, conn);
}
/** Receive data on an iperf udp session */
static void
lwiperf_udp_send_report(lwiperf_state_udp_t *conn)
{
lwiperf_state_udp_t *s = (lwiperf_state_udp_t *)conn->base.related_master_state;
struct pbuf *q = conn->reported;
LWIP_ASSERT("no report buffer!", q != NULL);
udp_sendto(s->pcb, q, &conn->remote_addr, conn->remote_port);
conn->report_count++;
if (conn->report_count < 2) {
/* Send twice after a little delay! */
sys_timeout(10, (sys_timeout_handler)lwiperf_udp_send_report, conn);
} else {
pbuf_free(q);
conn->reported = NULL;
/* reported twice -> test done */
if (conn->settings.base.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST)) {
if ((conn->settings.base.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_NOW)) == 0) {
/* client requested transmission after end of test */
LWIP_PLATFORM_DIAG(("client requested transmission after end of test\n"));
lwiperf_udp_tx_start(conn);
}
}
lwiperf_udp_close(conn, LWIPERF_UDP_DONE_SERVER);
if (s->base.server & 0x80) {
/* this is a temporary server for tradeoff or dualtest, ensure no report for this temporary server */
s->report_fn = NULL;
lwiperf_udp_close(s, LWIPERF_UDP_DONE_SERVER);
}
}
}
static void
lwiperf_udp_set_client_rate(lwiperf_state_udp_t *c, s32_t rate, u32_t buf_len)
{
/* compute delay for bandwidth restriction, constrained to [0,1]s in microseconds */
c->delay_target = (uint32_t)((buf_len * 8 * 1000000ull) / rate);
LWIP_PLATFORM_DIAG(("Ideal frame delay: %lu us\n", c->delay_target));
/* truncate the delay according to clock resolution, may result in a higher bitrate */
c->delay_target = (c->delay_target / CLOCK_RESOLUTION_US) * CLOCK_RESOLUTION_US;
if (c->delay_target == 0u) {
/* bitrate is high - have to send more than 1 frame per CLOCK_RESOLUTION_US
* period, may result in a lower bitrate and/or a higher jitter */
c->delay_target = CLOCK_RESOLUTION_US;
c->frames_per_delay = CLOCK_RESOLUTION_US / (uint32_t)((buf_len * 8 * 1000000ull) / rate);
} else {
c->frames_per_delay = 1u;
}
LWIP_PLATFORM_DIAG(("Send %u frame(s) once per %lu us\n", c->frames_per_delay, c->delay_target));
}
/** Try to send more data on an iperf udp session */
/* Must be called in main loop */
static void
lwiperf_udp_client_send_more(lwiperf_state_udp_t *conn)
{
struct pbuf *p;
struct lwip_timespec ts, dt;
err_t err;
int ending = 0;
int i;
LWIP_ASSERT("conn invalid", (conn != NULL) && !conn->base.tcp && (conn->base.server == 0));
if (conn->settings.base.amount & PP_HTONL(0x80000000)) {
/* this session is time-limited */
u32_t now = sys_now();
u32_t diff_ms = now - conn->time_started;
u32_t time = (u32_t) - (s32_t)lwip_ntohl(conn->settings.base.amount);
u32_t time_ms = time * 10;
if (diff_ms >= time_ms) {
ending = 1;
if (diff_ms > (time_ms + 500))
ending ++;
}
} else {
/* this session is byte-limited */
u32_t amount_bytes = lwip_ntohl(conn->settings.base.amount);
if (conn->bytes_transferred >= amount_bytes) {
ending = 1;
if (conn->bytes_transferred >= (amount_bytes + 4096))
ending++;
}
}
if (ending && (ending > 1 || conn->report_count > 0)) {
lwiperf_udp_close(conn, LWIPERF_UDP_DONE_CLIENT);
return;
}
/* check time/bw */
clock_gettime(CLOCK_ID, &ts);
diff_ts(&conn->udp_lastpkt, &ts, &dt);
if ((uint32_t)((dt.tv_sec * 1000000) + (dt.tv_nsec / 1000)) < conn->delay_target)
return;
for (i = 0; i < conn->frames_per_delay; i++) {
/* make pbuf to transmit */
p = pbuf_alloc(PBUF_TRANSPORT, lwip_ntohl(conn->settings.base.buffer_len), PBUF_POOL);
if (p) {
struct UDP_datagram *pkt = (struct UDP_datagram *)p->payload;
int hsz = (conn->settings.base.flags & PP_HTONL(LWIPERF_FLAGS_EXTEND) ?
sizeof(conn->settings) : sizeof(conn->settings.base));
int offset = sizeof(struct UDP_datagram);
/* set UDP datagram header */
if (ending)
pkt->id = htonl(-conn->udp_seq);
else
pkt->id = htonl(conn->udp_seq);
pkt->tv_sec = htonl(ts.tv_sec);
pkt->tv_usec = htonl(ts.tv_nsec / 1000);
/* save last packet time */
conn->udp_lastpkt = ts;
/* add settings in all buffer sent */
pbuf_take_at(p, &conn->settings, hsz, offset);
offset += hsz;
/* add data */
pbuf_take_at(p, &lwiperf_txbuf_const[0], p->tot_len - offset, offset);
/* send it */
err = udp_send(conn->pcb, p);
if (err == ERR_OK) {
conn->udp_seq++;
conn->bytes_transferred += p->tot_len;
} else {
/* Do not close connection when udp_send() fails - tx may be overloaded
* momentarily, the datagram will be just lost: */
/* lwiperf_udp_close(conn, LWIPERF_UDP_ABORTED_LOCAL_TXERROR); */
/* release pbuf */
pbuf_free(p);
return;
}
/* release pbuf */
pbuf_free(p);
/* adjust delay for ending retries */
if (ending) {
conn->delay_target = 250000; /* ending retry delay : 250ms */
conn->frames_per_delay = 1U;
}
} else {
/* Do not close connection when pbuf_alloc() fails - it may recover later */
/* lwiperf_udp_close(conn, LWIPERF_UDP_ABORTED_LOCAL); */
return;
}
}
}
/** Create a new UDP connection back to the client.
*/
static lwiperf_state_udp_t *
lwiperf_udp_tx_new(lwiperf_state_udp_t *conn)
{
lwiperf_state_udp_t *client_conn;
struct udp_pcb *newpcb;
client_conn = (lwiperf_state_udp_t *)LWIPERF_ALLOC(lwiperf_state_udp_t);
if (client_conn == NULL) {
return NULL;
}
memset(client_conn, 0, sizeof(lwiperf_state_udp_t));
newpcb = udp_new();
if (newpcb == NULL) {
LWIPERF_FREE(lwiperf_state_udp_t, client_conn);
return NULL;
}
if (conn) {
MEMCPY(client_conn, conn, sizeof(lwiperf_state_udp_t));
}
client_conn->base.tcp = 0;
client_conn->base.server = 0;
client_conn->pcb = newpcb;
client_conn->time_started = sys_now();
client_conn->bytes_transferred = 0;
client_conn->settings.base.flags = 0; /* prevent the remote side starting back as client again */
udp_recv(newpcb, lwiperf_udp_recv, client_conn);
return client_conn;
}
/** Start UDP connection back to the client (either parallel or after the
* receive test has finished).
*/
static err_t
lwiperf_udp_tx_start(lwiperf_state_udp_t *conn)
{
lwiperf_state_udp_t *cc;
u32_t buf_len;
err_t err;
cc = lwiperf_udp_tx_new(conn);
if (cc == NULL) {
return ERR_MEM;
}
/* take remote port to use from received settings */
cc->remote_port = (u16_t)lwip_htonl(cc->settings.base.remote_port);
/* buffer_len depend on address type */
buf_len = (u32_t)(IP_IS_V6(&cc->remote_addr) ? 1450 : 1470);
cc->settings.base.buffer_len = lwip_htonl(buf_len);
/* compute delay for bandwidth restriction, constrained to [0,1000] milliseconds */
/* 10Mbit/s by default if not extended settings */
if (cc->settings.base.flags & PP_HTONL(LWIPERF_FLAGS_EXTEND)) {
lwiperf_udp_set_client_rate(cc, lwip_ntohl(cc->settings.rate), buf_len);
}
else if (cc->settings.base.win_band != 0) {
lwiperf_udp_set_client_rate(cc, lwip_ntohl(cc->settings.base.win_band), buf_len);
}
else {
lwiperf_udp_set_client_rate(cc, (1024 * 1024), buf_len);
}
err = udp_connect(cc->pcb, &cc->remote_addr, cc->remote_port);
if (err != ERR_OK) {
lwiperf_udp_close(cc, LWIPERF_TCP_ABORTED_LOCAL);
return err;
}
lwiperf_list_add(&cc->base);
/* start sending immediately */
lwiperf_udp_client_send_more(cc);
return ERR_OK;
}
/** Receive data on an iperf udp session
* If client session, will receive final FIN from server,
* else, will receive all server traffic.
*/
static void
lwiperf_udp_recv(void *arg, struct udp_pcb *pcb, struct pbuf *p,
const ip_addr_t *addr, u16_t port)
{
lwiperf_state_udp_t *server = (lwiperf_state_udp_t *)arg;
lwiperf_state_udp_t *conn = NULL;
struct UDP_datagram *pkt;
int32_t datagramID;
u16_t tot_len = p->tot_len;
LWIP_ASSERT("pcb mismatch", server->pcb == pcb);
LWIP_UNUSED_ARG(pcb);
/* lookup client using remote addr */
if (server->base.server)
conn = lwiperf_udp_search_client(server, addr, port);
else
conn = server;
/* Read packet iperf data. */
pkt = (struct UDP_datagram *)p->payload;
datagramID = ntohl(pkt->id);
if (conn && !conn->base.server) {
/* received server reports for client instances -> close it. */
if (!conn->report_count) {
lwiperf_udp_report_t *hdr;
pkt = (struct UDP_datagram *)p->payload;
pkt->id = htonl(datagramID);
hdr = (lwiperf_udp_report_t *)(pkt + 1);
if (hdr->flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST)) {
/* Adjust bytes transferred with the one from server report */
LWIP_PLATFORM_DIAG(("Received report from server (0x%x).\n", lwip_ntohl(hdr->flags)));
/*LWIP_PLATFORM_DIAG(("Stop %ld.%03ld sec, ", ntohl(hdr->stop_sec), ntohl(hdr->stop_usec)/1000));
LWIP_PLATFORM_DIAG(("Total %ldKB, ", ntohl(hdr->total_len2)/1024));*/
LWIP_PLATFORM_DIAG(("Jitter %ld.%03ld, ", ntohl(hdr->jitter1), ntohl(hdr->jitter2)));
LWIP_PLATFORM_DIAG(("Lost %ld/%ld datagrams, OoO %ld\n",
ntohl(hdr->error_cnt), ntohl(hdr->datagrams), ntohl(hdr->outorder_cnt)));
conn->bytes_transferred = (((u64_t)ntohl(hdr->total_len1)) << 32) + ntohl(hdr->total_len2);
}
if (hdr->flags & PP_HTONL(LWIPERF_FLAGS_EXTEND)) {
LWIP_PLATFORM_DIAG(("Extended report unsupported yet.\n"));
}
}
conn->report_count++;
}
else if (datagramID >= 0) {
struct lwip_timespec ts, dt;
uint32_t transit;
clock_gettime(1, &ts);
if (!conn || !conn->have_settings_buf) {
/* allocate struct for a new client */
if (!conn) {
conn = lwiperf_udp_new_client(server);
if (!conn) {
pbuf_free(p);
return;
}
}
/* wait for 24-byte header */
if (p->tot_len < (sizeof(struct UDP_datagram) + sizeof(lwiperf_settings_t))) {
lwiperf_udp_close(conn, LWIPERF_UDP_ABORTED_LOCAL_DATAERROR);
pbuf_free(p);
return;
}
/* copy settings */
if (pbuf_copy_partial(p, &conn->settings, sizeof(lwiperf_settings_t), sizeof(struct UDP_datagram))
!= sizeof(lwiperf_settings_t)) {
lwiperf_udp_close(conn, LWIPERF_UDP_ABORTED_LOCAL);
pbuf_free(p);
return;
}
conn->have_settings_buf = 1;
LWIP_PLATFORM_DIAG(("New UDP client (settings flags 0x%x)\n", lwip_ntohl(conn->settings.base.flags)));
udp_start_lost_time = ntohl(pkt->tv_sec);
/* Save client data. */
ip_addr_copy(conn->remote_addr, *addr);
conn->remote_port = port;
conn->time_started = sys_now();
/* Reset tradeoff & dualtest setting if temporary server */
if (server->base.server & 0x80) {
conn->settings.base.flags &= ~PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST|LWIPERF_FLAGS_ANSWER_NOW);
}
/* check if dualtest requested */
if (conn->settings.base.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST)) {
if (conn->settings.base.flags & PP_HTONL(LWIPERF_FLAGS_ANSWER_NOW)) {
/* client requested parallel transmission test */
LWIP_PLATFORM_DIAG(("client requested parallel transmission test\n"));
err_t err2 = lwiperf_udp_tx_start(conn);
if (err2 != ERR_OK) {
lwiperf_udp_close(conn, LWIPERF_UDP_ABORTED_LOCAL_TXERROR);
pbuf_free(p);
return;
}
}
}
}
/* Update stats. */
conn->udp_lastpkt.tv_sec = ntohl(pkt->tv_sec);
conn->udp_lastpkt.tv_nsec = ntohl(pkt->tv_usec) * 1000;
if (conn->udp_seq != (uint32_t) datagramID) {
conn->udp_rx_lost += (uint32_t) datagramID - conn->udp_seq;
conn->udp_seq = datagramID + 1;
conn->udp_rx_outorder += 1;
udp_last_lost_time = conn->udp_lastpkt.tv_sec;
if(udp_last_lost_time != udp_start_lost_time)
{
LWIP_PLATFORM_DIAG(("udp_last_lost_time:%ld\n",udp_last_lost_time - udp_start_lost_time));
lwiperf_udp_close(conn, LWIPERF_UDP_DONE_SERVER);
}
} else {
conn->bytes_transferred += tot_len;
conn->udp_rx_total_pkt += 1;
conn->udp_seq += 1;
}
/* Jitter calculation
* from RFC 1889, Real Time Protocol (RTP)
* J = J + ( | D(i-1,i) | - J ) / Compute jitter
*/
diff_ts (&conn->udp_lastpkt, &ts, &dt);
transit = (uint32_t)((dt.tv_sec * 1000000) + (dt.tv_nsec / 1000));
if (conn->udp_last_transit) {
long deltaTransit = (long)(transit - conn->udp_last_transit);
if ( deltaTransit < 0.0 ) {
deltaTransit = -deltaTransit;
}
conn->jitter += (deltaTransit - conn->jitter) >> 4;
}
else {
conn->udp_last_transit = transit;
}
} else {
if (conn && conn->have_settings_buf && !conn->report_count) {
lwiperf_udp_report_t *hdr;
u32_t now, duration_ms;
now = sys_now();
duration_ms = now - conn->time_started;
/* Copy packet and send report back. */
struct pbuf *q = pbuf_clone(PBUF_TRANSPORT, PBUF_POOL, p);
LWIP_ASSERT("can't clone buffer", q != NULL);
pkt = (struct UDP_datagram *)q->payload;
pkt->id = htonl(datagramID);
hdr = (lwiperf_udp_report_t *)(pkt + 1);
hdr->flags = PP_HTONL(LWIPERF_FLAGS_ANSWER_TEST);
hdr->total_len1 = htonl((u32_t)(conn->bytes_transferred >> 32));
hdr->total_len2 = htonl((u32_t)(conn->bytes_transferred & 0xFFFFFFFF));
hdr->stop_sec = htonl(duration_ms / 1000);
hdr->stop_usec = htonl((duration_ms % 1000) * 1000);
hdr->error_cnt = htonl(conn->udp_rx_lost);
hdr->outorder_cnt = htonl(conn->udp_rx_outorder);
hdr->datagrams = htonl(conn->udp_rx_total_pkt);
hdr->jitter1 = htonl(conn->jitter / 1000000);
hdr->jitter2 = htonl((conn->jitter % 1000000) / 1000);
/* Adjust bytes transferred with the one from server report */
LWIP_PLATFORM_DIAG(("Sending report back to client (0x%x).\n", hdr->flags));
/*LWIP_PLATFORM_DIAG(("Stop %ld.%03ld sec, ", ntohl(hdr->stop_sec), ntohl(hdr->stop_usec)/1000));
LWIP_PLATFORM_DIAG(("Total %ldKB, ", ntohl(hdr->total_len2)/1024));*/
LWIP_PLATFORM_DIAG(("Jitter %ld.%03ld, ", ntohl(hdr->jitter1), ntohl(hdr->jitter2)));
LWIP_PLATFORM_DIAG(("Lost %ld/%ld datagrams, OoO %ld\n",
ntohl(hdr->error_cnt), ntohl(hdr->datagrams), ntohl(hdr->outorder_cnt)));
/* Store report buffer in conn structure */
conn->reported = q;
/* Send report to client. */
lwiperf_udp_send_report(conn);
}
}
pbuf_free(p);
}
/**
* @ingroup iperf
* Start an UDP iperf server on a specific IP address and port and listen for
* incoming datagrams from iperf clients.
*
* @returns a connection handle that can be used to abort the server
* by calling @ref lwiperf_abort()
*/
void *
lwiperf_start_udp_server(const ip_addr_t *local_addr, u16_t local_port,
lwiperf_report_fn report_fn, void *report_arg)
{
lwiperf_state_udp_t *s;
err_t err;
LWIP_ASSERT_CORE_LOCKED();
if (local_addr == NULL)
return NULL;
s = (lwiperf_state_udp_t *)LWIPERF_ALLOC(lwiperf_state_udp_t);
if (s == NULL)
return NULL;
do {
memset(s, 0, sizeof(lwiperf_state_udp_t));
s->base.tcp = 0;
s->base.server = 1;
s->report_fn = report_fn;
s->report_arg = report_arg;
/* allocate udp pcb */
s->pcb = udp_new();
if (s->pcb == NULL)
break;
/* bind to locat port/address */
err = udp_bind(s->pcb, local_addr, local_port);
if (err != ERR_OK)
break;
/* join group if multicast address */
if (ip_addr_ismulticast(local_addr)) {
if (IP_IS_V6 (local_addr)) {
#if LWIP_IPV6_MLD
err = mld6_joingroup (IP6_ADDR_ANY6, ip_2_ip6 (local_addr));
#endif
}
else {
#if LWIP_IGMP
err = igmp_joingroup (IP4_ADDR_ANY4, ip_2_ip4 (local_addr));
#endif
}
if (err != ERR_OK)
break;
}
/* start receiving datagrams */
udp_recv(s->pcb, lwiperf_udp_recv, s);
/* finally, add to list */
lwiperf_list_add(&s->base);
return s;
} while (0);
/* error occurs, cleanup */
if (s->pcb)
udp_remove(s->pcb);
LWIPERF_FREE(lwiperf_state_udp_t, s);
return NULL;
}
/**
* @ingroup iperf
* Start a UDP iperf client connected to a specific IP address and port.
*
* @returns a connection handle that can be used to abort the client
* by calling @ref lwiperf_abort()
*/
void *
lwiperf_start_udp_client(const ip_addr_t *local_addr, u16_t local_port,
const ip_addr_t *remote_addr, u16_t remote_port,
enum lwiperf_client_type type, int amount, s32_t rate, u8_t tos,
lwiperf_report_fn report_fn, void *report_arg)
{
err_t err;
lwiperf_state_udp_t *c;
lwiperf_state_udp_t *s = NULL;
u32_t buf_len, flags;
u16_t sport = 0;
switch (type) {
case LWIPERF_CLIENT:
/* Unidirectional tx only test */
flags = 0;
break;
case LWIPERF_DUAL:
/* Do a bidirectional test simultaneously */
flags = htonl(LWIPERF_FLAGS_ANSWER_TEST | LWIPERF_FLAGS_ANSWER_NOW);
break;
case LWIPERF_TRADEOFF:
/* Do a bidirectional test individually */
flags = htonl(LWIPERF_FLAGS_ANSWER_TEST);
break;
default:
/* invalid argument */
return NULL;
}
if (type != LWIPERF_CLIENT) {
/* tradeoff or dualtest requested. need to start a new server on another port */
s = lwiperf_start_udp_server(
local_addr ? local_addr : IP4_ADDR_ANY, local_port,
report_fn, report_arg);
if (s) {
s->base.server |= 0x80; /* put a temporary server mark */
sport = s->pcb->local_port; /* retrieve this server port */
LWIP_PLATFORM_DIAG(("Dualtest port: %u\n", sport));
}
else {
LWIP_PLATFORM_DIAG(("Dualtest disabled!\n"));
}
}
/* Create client */
c = lwiperf_udp_tx_new(NULL);
if (c == NULL)
return NULL;
c->report_fn = report_fn;
c->report_arg = report_arg;
/* save remote */
ip_addr_copy(c->remote_addr, *remote_addr);
c->remote_port = remote_port;
/* save client settings, to be copied in each packets */
c->have_settings_buf = 1;
c->settings.base.amount = lwip_htonl((uint32_t)amount);
buf_len = (u32_t)(IP_IS_V6(remote_addr) ? 1180 : 1200); // buf_len = (u32_t)(IP_IS_V6(remote_addr) ? 1450 : 1470);
c->settings.base.buffer_len = lwip_htonl(buf_len);
if (rate != (1024 * 1024)) { /* 1Mb/s is the default if not specified. */
c->settings.rate = lwip_htonl(rate);
c->settings.base.flags |= PP_HTONL(LWIPERF_FLAGS_EXTEND);
c->settings.base.win_band = lwip_htonl(rate);
}
if (sport) {
/* tradeoff or dualtest requested need to put server port in settings */
c->settings.base.flags |= flags;
c->settings.base.remote_port = lwip_htonl(sport);
}
lwiperf_udp_set_client_rate(c, rate, buf_len);
/* set tos if specified */
if (tos)
c->pcb->tos = tos;
do {
if (local_addr) {
/* bind to local address if specified */
err = udp_bind(c->pcb, local_addr, 0);
if (err != ERR_OK)
break;
/* join multicast group? */
if (ip_addr_ismulticast(local_addr)) {
if (IP_IS_V6 (local_addr)) {
#if LWIP_IPV6_MLD
err = mld6_joingroup (IP6_ADDR_ANY6, ip_2_ip6 (local_addr));
#endif
}
else {
#if LWIP_IGMP
err = igmp_joingroup (IP4_ADDR_ANY4, ip_2_ip4 (local_addr));
#endif
}
if (err != ERR_OK)
break;
}
}
/* set pseudo-connect parameters */
err = udp_connect(c->pcb, remote_addr, remote_port);
if (err != ERR_OK)
break;
/* and add to instance list */
lwiperf_list_add(&c->base);
if (s != NULL)
s->base.related_master_state = &c->base;
/* start sending immediately */
lwiperf_udp_client_send_more(c);
return c;
} while (0);
/* error occurs, cleanup */
if (c->pcb)
udp_remove(c->pcb);
LWIPERF_FREE(lwiperf_state_udp_t, c);
return NULL;
}
/**
* @ingroup iperf
* Poll all running UDP client to send more according to specified BW.
*/
void lwiperf_poll_udp_client(void)
{
lwiperf_state_base_t *c;
for (c = lwiperf_all_connections; c != NULL; c = c->next) {
if (!c->server && !c->tcp)
lwiperf_udp_client_send_more((lwiperf_state_udp_t *)c);
}
}
/**
* @ingroup iperf
* Abort an iperf session (handle returned by lwiperf_start_tcp_server*())
*/
void
lwiperf_abort(void *lwiperf_session)
{
lwiperf_state_base_t *i, *dealloc, *last = NULL;
LWIP_ASSERT_CORE_LOCKED();
for (i = lwiperf_all_connections; i != NULL; ) {
if ((i == lwiperf_session) || (i->related_master_state == lwiperf_session)) {
dealloc = i;
i = i->next;
if (last != NULL) {
last->next = i;
}
LWIPERF_FREE(lwiperf_state_tcp_t, dealloc); /* @todo: type? */
} else {
last = i;
i = i->next;
}
}
}
#endif /* LWIP_TCP && LWIP_CALLBACK_API */
lwiperf.h
/**
* @file
* lwIP iPerf server implementation
*/
/*
* Copyright (c) 2014 Simon Goldschmidt
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* 3. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
* SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
* OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
* OF SUCH DAMAGE.
*
* This file is part of the lwIP TCP/IP stack.
*
* Author: Simon Goldschmidt
*
*/
#ifndef LWIP_HDR_APPS_LWIPERF_H
#define LWIP_HDR_APPS_LWIPERF_H
#include "lwip/opt.h"
#include "lwip/ip_addr.h"
#ifdef __cplusplus
extern "C" {
#endif
#define LWIPERF_TCP_PORT_DEFAULT 5001
#define LWIPERF_UDP_PORT_DEFAULT 5001
/** lwIPerf test results */
enum lwiperf_report_type
{
/** The server side test is done */
LWIPERF_TCP_DONE_SERVER,
/** The client side test is done */
LWIPERF_TCP_DONE_CLIENT,
/** Local error lead to test abort */
LWIPERF_TCP_ABORTED_LOCAL,
/** Data check error lead to test abort */
LWIPERF_TCP_ABORTED_LOCAL_DATAERROR,
/** Transmit error lead to test abort */
LWIPERF_TCP_ABORTED_LOCAL_TXERROR,
/** Remote side aborted the test */
LWIPERF_TCP_ABORTED_REMOTE,
/** The server side test is done */
LWIPERF_UDP_DONE_SERVER,
/** The client side test is done */
LWIPERF_UDP_DONE_CLIENT,
/** Local error lead to test abort */
LWIPERF_UDP_ABORTED_LOCAL,
/** Data check error lead to test abort */
LWIPERF_UDP_ABORTED_LOCAL_DATAERROR,
/** Transmit error lead to test abort */
LWIPERF_UDP_ABORTED_LOCAL_TXERROR,
/** Remote side aborted the test */
LWIPERF_UDP_ABORTED_REMOTE
};
/** Control */
enum lwiperf_client_type
{
/** Unidirectional tx only test */
LWIPERF_CLIENT,
/** Do a bidirectional test simultaneously */
LWIPERF_DUAL,
/** Do a bidirectional test individually */
LWIPERF_TRADEOFF
};
/** Prototype of a report function that is called when a session is finished.
This report function can show the test results.
@param report_type contains the test result */
typedef void (*lwiperf_report_fn)(void *arg, enum lwiperf_report_type report_type,
const ip_addr_t* local_addr, u16_t local_port, const ip_addr_t* remote_addr, u16_t remote_port,
u32_t bytes_transferred, u32_t ms_duration, u32_t bandwidth_kbitpsec);
void* lwiperf_start_tcp_server(const ip_addr_t* local_addr, u16_t local_port,
lwiperf_report_fn report_fn, void* report_arg);
void* lwiperf_start_tcp_server_default(lwiperf_report_fn report_fn, void* report_arg);
void* lwiperf_start_tcp_client(const ip_addr_t* remote_addr, u16_t remote_port,
enum lwiperf_client_type type,
lwiperf_report_fn report_fn, void* report_arg);
void* lwiperf_start_tcp_client_default(const ip_addr_t* remote_addr,
lwiperf_report_fn report_fn, void* report_arg);
void* lwiperf_start_udp_server(const ip_addr_t *local_addr, u16_t local_port,
lwiperf_report_fn report_fn, void *report_arg);
void *lwiperf_start_udp_client(const ip_addr_t *local_addr, u16_t local_port,
const ip_addr_t *remote_addr, u16_t remote_port,
enum lwiperf_client_type type, int amount, s32_t rate, u8_t tos,
lwiperf_report_fn report_fn, void *report_arg);
void lwiperf_poll_udp_client(void);
void lwiperf_abort(void* lwiperf_session);
#ifdef __cplusplus
}
#endif
#endif /* LWIP_HDR_APPS_LWIPERF_H */
lwiperf_start.c代码如下:
lwiperf_start.c
#include "lwiperf_start.h"
#include "sys.h"
//#include "sys_arch.h"
//#include "ethernetif.h"
//#include "lwip.h"
#include "netif/ethernetif.h"
#include "lwip/init.h"
#include "lwip/timeouts.h"
#include "lwiperf.h"
/* lwiperf.c 修改
* 减小了buf的大小，每一个周期发包的数量增多了，但是发送速率计算进度提高
* 1539: buf_len = (u32_t)(IP_IS_V6(remote_addr) ? 1180 : 1200);
* lwiperf_udp_recv中添加了计算lost和outorder时在第一个循环发包之后出现丢包或者乱序会直接关闭测试的代码
*
*
*
*
*
*
*
*
*
*
*/
/*设置UDP发送速率，分辨率与buf_len大小有相关*/
#ifndef IPERF_UDP_CLIENT_RATE
#if RGMII
#define IPERF_UDP_CLIENT_RATE (85 * 1024 * 1024)
#else
#define IPERF_UDP_CLIENT_RATE (100 * 1024 * 1024)
#endif
#endif
/*测试时间 单位10m/s*/
#ifndef IPERF_CLIENT_AMOUNT
#define IPERF_CLIENT_AMOUNT (-1000) /* 10seconds */
#endif
static bool select_mode(bool *server_mode, bool *tcp, enum lwiperf_client_type *client_type)
{
char code;
//if (!enet_get_link_status()) {
// return false;
//}
printf("\n");
printf("1: TCP Server Mode\n");
printf("2: TCP Client Mode\n");
printf("3: UDP Server Mode\n");
printf("4: UDP Client Mode\n");
printf("Please enter one of modes above (e.g. 1 or 2 ...)\n");
code = getchar();
printf("Select Mode: %c\n", code);
switch (code)
{
case '1':
*server_mode = true;
*tcp = true;
*client_type = LWIPERF_CLIENT;
break;
case '2':
*server_mode = false;
*tcp = true;
*client_type = LWIPERF_CLIENT;
break;
case '3':
*server_mode = true;
*tcp = false;
*client_type = LWIPERF_CLIENT;
break;
case '4':
*server_mode = false;
*tcp = false;
*client_type = LWIPERF_CLIENT;
break;
default:
break;
}
return true;
}
static void
lwiperf_report(void *arg, enum lwiperf_report_type report_type,
const ip_addr_t* local_addr, u16_t local_port, const ip_addr_t* remote_addr, u16_t remote_port,
u32_t bytes_transferred, u32_t ms_duration, u32_t bandwidth_kbitpsec)
{
LWIP_UNUSED_ARG(arg);
LWIP_UNUSED_ARG(local_addr);
LWIP_UNUSED_ARG(local_port);
LWIP_PLATFORM_DIAG(("iperf report: type=%d, remote: %s:%d, total bytes: %"U32_F", duration in ms: %"U32_F", kbits/s: %"U32_F"\n",
(int)report_type, ipaddr_ntoa(remote_addr), (int)remote_port, bytes_transferred, ms_duration, bandwidth_kbitpsec));
}
static void *start_iperf(void)
{
bool server = false;
bool tcp = false;
enum lwiperf_client_type client_type;
void *session;
ip_addr_t remote_addr;
if (!select_mode(&server, &tcp, &client_type)) {
return NULL;
}
if (server) {
if (tcp) {
session = lwiperf_start_tcp_server_default(lwiperf_report, NULL);
} else {
session = lwiperf_start_udp_server(netif_ip_addr4(netif_default), LWIPERF_UDP_PORT_DEFAULT,
lwiperf_report, NULL);
}
} else {
IP_ADDR4(&remote_addr, REMOTE_IP_ADDR0, REMOTE_IP_ADDR1, REMOTE_IP_ADDR2, REMOTE_IP_ADDR3);
if (tcp) {
session = lwiperf_start_tcp_client_default(&remote_addr, lwiperf_report, NULL);
} else {
session = lwiperf_start_udp_client(netif_ip_addr4(netif_default), LWIPERF_UDP_PORT_DEFAULT,
&remote_addr, LWIPERF_UDP_PORT_DEFAULT, client_type,
IPERF_CLIENT_AMOUNT, IPERF_UDP_CLIENT_RATE, 0,
lwiperf_report, NULL);
}
}
return session;
}
void iperf(void)
{
static void *session = NULL;
if (session == NULL) {
session = start_iperf();
}
/*此函数判断是否为udp client 是则发送数据*/
lwiperf_poll_udp_client();
}
lwiperf_start.h
#ifndef __LWIPERF__START__H
#define __LWIPERF__START__H
/* Includes ------------------------------------------------------------------*/
#include <stdbool.h>
/* Exported Macros------------------------------------------------------------*/
#if RGMII
#define ENET_INF_TYPE enet_inf_rgmii
#define ENET BOARD_ENET_RGMII
#else
#define ENET_INF_TYPE enet_inf_rmii
#define ENET BOARD_ENET_RMII
#endif
#define ENET_TX_BUFF_COUNT (50U)
#define ENET_RX_BUFF_COUNT (60U)
#define ENET_RX_BUFF_SIZE ENET_MAX_FRAME_SIZE
#define ENET_TX_BUFF_SIZE ENET_MAX_FRAME_SIZE
/* Remote IP Address */
#define REMOTE_IP_ADDR0 192
#define REMOTE_IP_ADDR1 168
#define REMOTE_IP_ADDR2 2
#define REMOTE_IP_ADDR3 2
/*iPerf client/server默认使用5001端口*/
//#define DEST_PORT 1000
//#define UDP_SERVER_PORT 2300 /* define the UDP local connection port */
//#define UDP_CLIENT_PORT 1000 /* define the UDP remote connection port */
//#define LOCAL_PORT 2300
/*Static IP ADDRESS: IP_ADDR0.IP_ADDR1.IP_ADDR2.IP_ADDR3 */
#define IP_ADDR0 192
#define IP_ADDR1 168
#define IP_ADDR2 2
#define IP_ADDR3 120
/*NETMASK*/
#define NETMASK_ADDR0 255
#define NETMASK_ADDR1 255
#define NETMASK_ADDR2 255
#define NETMASK_ADDR3 0
/*Gateway Address*/
#define GW_ADDR0 192
#define GW_ADDR1 168
#define GW_ADDR2 2
#define GW_ADDR3 1
void iperf(void);
/* Exported Variables ------------------------------------------------------*/
#endif /* _LWIPERF__START__H */
上述两个文件 lwiperf.c为测试源码，lwiperf_start.c为用编写的调用lwiperf.c中测试函数的代码。
start_iperf.c:IPERF_UDP_CLIENT_RATE配置UDP的发送速率（100 * 1024 * 1024 为100Mbit/s），IPERF_CLIENT_AMOUNT为开发板作为Client端的测试时间（-1000 为10s）。select_mode(): 通过getchar实现选择4种测试模式（需要实现fgetc, USART需要关接收中断）lwiperf_report(): lwiperf发送测试报告函数，lwiperf会在测试发送或者接收到最后一帧时有Server端发送测试报告，例如使用开发板为Client则会接收到上位机Server端的report，开发板为Server则会向上位机发送report。（需要实现fputc）start_iperf(): iperf测试启动，会根据用户的输入启动相应的测试代码(UDP/TCP client 或 UDP/TCP Server)，测试统一为Server端接收Client端发送。iperf(): 外部调用的lwiperf启动函数。start_iperf.h文件配置了网络的相关参数如本地IP、子网掩码、网关、远端IP
UPD/TCP网络的端口为 5001 这个时IPerf的默认配置，在lwiperf.h定义。
测试代码main.c如下：
#include "sys.h"
#include "delay.h"
#include "usart.h"
#include "led.h"
#include "timer.h"
#include "lwip/timeouts.h"
#include "stm32f4x7_eth.h"
#include "DP83848.h"
#include "lwip_comm.h"
/*使用JPerf进行速度测试*/
#include "lwiperf_start.h"
#include "lwiperf.h"
int main(void)
{
NVIC_PriorityGroupConfig(NVIC_PriorityGroup_2); // 中断分组配置
delay_init(168); // 初始化延时函数
uart_init(); // 初始化串口
LED_Init(); // 初始化LED端口
TIM3_Int_Init(999, 839);
lwip_comm_init();
printf("PHY_SR:0x%04x\n", ETH_ReadPHYRegister(0x01, PHY_SR));
while (1)
{
iperf();
sys_check_timeouts();
}
}
只需要初始lwip，配置UDP/TCP已经通过lwiperf_start_xxx_xxx()函数完成。
四、测试结果
以下为收发100Mbit/s测试
4.1、UDP client
先启动iperf UDP Server
iperf.exe -s -c 192.168.2.120 -u -P 0 -i 1 -p 5001 -f k
再通过串口打开开发板UDP Client 输入字符 4
4.2、UDP server
先通过串口打开开发板UDP Server 输入字符 3
再启动IPerf UDP client
iperf.exe -c 192.168.2.120 -u -P 1 -i 1 -p 5001 -f k -b 100.0M -t 10 -T 1
4.3、TCP client
先启动iperf TCP Server
iperf.exe -s -c 192.168.2.120 -P 0 -i 1 -p 5001 -f k
再通过串口打开开发板TCP Client 输入字符 2
4.4、TCP server
先通过串口打开开发板TCP Server 输入字符 1
再启动IPerf TCP client
五、小结
本节通过使用IPerf2.0测试了STM32 DP83848移植lwip的速率，最大速率接近100Mbit/s，说明移植比较成功。关于UDP/TCP RAW的编程可以直接参考lwiperf.c的源文件。
Jperf是IPerf的图形化软件，上手更方便一点，但是更新很慢而且需要安装java环境，IPerf的linux版本更新较快，需要使用新版的测试就需要在linux上测试了。
