/************************************************************************
* NAME: fapp_bench_udp_tx
*
* DESCRIPTION: Start TX TCP Benchmark.
************************************************************************/
static void fapp_bench_udp_tx (struct fapp_bench_tx_params *params)
{
fnet_int32_t send_result;
fnet_char_t ip_str[FNET_IP_ADDR_STR_SIZE];
fnet_index_t i;
fnet_int32_t received;
const struct linger linger_option = {FNET_TRUE, /*l_onoff*/
4 /*l_linger*/
};
const fnet_size_t bufsize_option = FAPP_BENCH_SOCKET_BUF_SIZE;
struct sockaddr foreign_addr;
fnet_bool_t exit_flag = FNET_FALSE;
fnet_int32_t sock_err;
fnet_size_t option_len;
fnet_shell_desc_t desc = params->desc;
fnet_size_t packet_size = params->packet_size;
fnet_index_t cur_packet_number;
fnet_index_t iterations = params->iteration_number;
fnet_address_family_t family = params->foreign_addr.sa_family;
if(packet_size > FAPP_BENCH_BUFFER_SIZE) /* Check max size.*/
packet_size = FAPP_BENCH_BUFFER_SIZE;
fapp_bench.socket_listen = FNET_ERR;
/* ------ Start test.----------- */
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fnet_shell_println(desc, " UDP TX Test" );
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_S, "Remote IP addr", fnet_inet_ntop(family, params->foreign_addr.sa_data, ip_str, sizeof(ip_str)));
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Remote Port", fnet_ntohs(params->foreign_addr.sa_port));
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Message Size", params->packet_size);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Num. of messages", params->packet_number);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Num. of iterations", params->iteration_number);
fnet_shell_println(desc, FAPP_TOCANCEL_STR);
fnet_shell_println(desc, FAPP_DELIMITER_STR);
while(iterations--)
{
/* Create socket */
if((fapp_bench.socket_foreign = fnet_socket(family, SOCK_DGRAM, 0)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket creation error.\n");
iterations = 0;
goto ERROR_1;
}
/* Set Socket options. */
if( /* Setup linger option. */
(fnet_socket_setopt (fapp_bench.socket_foreign, SOL_SOCKET, SO_LINGER, (fnet_uint8_t *)&linger_option, sizeof(linger_option)) == FNET_ERR) ||
/* Set socket buffer size. */
(fnet_socket_setopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_RCVBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR) ||
(fnet_socket_setopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_SNDBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR)
)
{
FNET_DEBUG("BENCH: Socket setsockopt error.\n");
iterations = 0;
goto ERROR_2;
}
/* Bind to the server.*/
fnet_shell_println(desc,"Connecting.");
fnet_memcpy(&foreign_addr, ¶ms->foreign_addr, sizeof(foreign_addr));
if(fnet_socket_connect(fapp_bench.socket_foreign, &foreign_addr, sizeof(foreign_addr))== FNET_ERR)
{
fnet_shell_println(desc, "Connection failed.");
iterations = 0;
goto ERROR_2;
}
/* Sending.*/
fnet_shell_println(desc,"Sending.");
fapp_bench.bytes = 0;
fapp_bench.remote_bytes = 0;
cur_packet_number = 0;
fapp_bench.first_time = fnet_timer_ticks();
while(1)
{
send_result = fnet_socket_send( fapp_bench.socket_foreign, (fnet_uint8_t*)(&fapp_bench.buffer[0]), packet_size, 0);
fapp_bench.last_time = fnet_timer_ticks();
if ( send_result == FNET_ERR )
{
option_len = sizeof(sock_err);
fnet_socket_getopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_ERROR, (fnet_uint8_t*)&sock_err, &option_len);
fnet_shell_println(desc, "socket_error = %d", sock_err);
iterations = 0;
goto ERROR_2;
}
else
{
fapp_bench.bytes += send_result;
cur_packet_number ++;
exit_flag = fnet_shell_ctrlc (desc); /* Check [Ctrl+c]*/
if((cur_packet_number >= params->packet_number)|| exit_flag)
{
if(exit_flag)
{
fnet_shell_println(desc, FAPP_SHELL_CANCELED_CTRL_C);
iterations = 0;
}
break;/* => TX END. */
}
}
}
/* Send End mark.*/
for(i=0; i < FAPP_BENCH_TX_UDP_END_ITERATIONS; i++)
{
fnet_uint32_t ack_bytes;
/* Send END mark.*/
fnet_socket_send( fapp_bench.socket_foreign, (fnet_uint8_t*)(&fapp_bench.buffer[0]), 1, 0);
fnet_timer_delay(1);
/* Check ACK. It should contain recieved amount of data.*/
received = fnet_socket_recv(fapp_bench.socket_foreign, (fnet_uint8_t*)(&ack_bytes), sizeof(ack_bytes), 0);
if(received == sizeof(ack_bytes)) /* ACK received.*/
{
fapp_bench.remote_bytes = fnet_ntohl(ack_bytes);
break;
}
else if(received == FNET_ERR)
{
break;
}
else
{}
}
/* Print benchmark results.*/
fapp_bench_print_results (desc);
ERROR_2:
fnet_socket_close(fapp_bench.socket_foreign);
}
ERROR_1:
fnet_shell_println(desc, FAPP_BENCH_COMPLETED_STR);
}
/************************************************************************
* NAME: fapp_bench_tcp_tx
*
* DESCRIPTION: Start TX TCP Benchmark.
************************************************************************/
static void fapp_bench_tcp_tx (struct fapp_bench_tx_params *params)
{
fnet_int32_t send_result;
fnet_char_t ip_str[FNET_IP_ADDR_STR_SIZE];
const struct linger linger_option = {FNET_TRUE, /*l_onoff*/
4 /*l_linger*/
};
const fnet_size_t bufsize_option = FAPP_BENCH_SOCKET_BUF_SIZE;
const fnet_int32_t keepalive_option = 1;
const fnet_int32_t keepcnt_option = FAPP_BENCH_TCP_KEEPCNT;
const fnet_int32_t keepintvl_option = FAPP_BENCH_TCP_KEEPINTVL;
const fnet_int32_t keepidle_option = FAPP_BENCH_TCP_KEEPIDLE;
struct sockaddr foreign_addr;
fnet_bool_t exit_flag = FNET_FALSE;
fnet_int32_t sock_err ;
fnet_size_t option_len;
fnet_shell_desc_t desc = params->desc;
fnet_socket_state_t connection_state;
fnet_size_t packet_size = params->packet_size;
fnet_index_t cur_packet_number;
fnet_size_t buffer_offset;
fnet_index_t iterations = params->iteration_number;
fnet_address_family_t family = params->foreign_addr.sa_family;
if(packet_size > FAPP_BENCH_BUFFER_SIZE) /* Check max size.*/
packet_size = FAPP_BENCH_BUFFER_SIZE;
fapp_bench.socket_listen = FNET_ERR;
/* ------ Start test.----------- */
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fnet_shell_println(desc, " TCP TX Test" );
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_S, "Remote IP Addr", fnet_inet_ntop(family, params->foreign_addr.sa_data, ip_str, sizeof(ip_str)));
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Remote Port", fnet_ntohs(params->foreign_addr.sa_port));
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Message Size", params->packet_size);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Num. of messages", params->packet_number);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Num. of iterations", params->iteration_number);
fnet_shell_println(desc, FAPP_TOCANCEL_STR);
fnet_shell_println(desc, FAPP_DELIMITER_STR);
while(iterations--)
{
/* Create socket */
if((fapp_bench.socket_foreign = fnet_socket(family, SOCK_STREAM, 0)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket creation error.\n");
iterations = 0;
goto ERROR_1;
}
/* Set Socket options. */
if( /* Setup linger option. */
(fnet_socket_setopt (fapp_bench.socket_foreign, SOL_SOCKET, SO_LINGER, (fnet_uint8_t *)&linger_option, sizeof(linger_option)) == FNET_ERR) ||
/* Set socket buffer size. */
(fnet_socket_setopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_RCVBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR) ||
(fnet_socket_setopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_SNDBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR) ||
/* Enable keepalive_option option. */
(fnet_socket_setopt (fapp_bench.socket_foreign, SOL_SOCKET, SO_KEEPALIVE, (fnet_uint8_t *)&keepalive_option, sizeof(keepalive_option)) == FNET_ERR) ||
/* Keepalive probe retransmit limit. */
(fnet_socket_setopt (fapp_bench.socket_foreign, IPPROTO_TCP, TCP_KEEPCNT, (fnet_uint8_t *)&keepcnt_option, sizeof(keepcnt_option)) == FNET_ERR) ||
/* Keepalive retransmit interval.*/
(fnet_socket_setopt (fapp_bench.socket_foreign, IPPROTO_TCP, TCP_KEEPINTVL, (fnet_uint8_t *)&keepintvl_option, sizeof(keepintvl_option)) == FNET_ERR) ||
/* Time between keepalive probes.*/
(fnet_socket_setopt (fapp_bench.socket_foreign, IPPROTO_TCP, TCP_KEEPIDLE, (fnet_uint8_t *)&keepidle_option, sizeof(keepidle_option)) == FNET_ERR)
)
{
FNET_DEBUG("BENCH: Socket setsockopt error.\n");
iterations = 0;
goto ERROR_2;
}
/* Connect to the server.*/
fnet_shell_println(desc,"Connecting.");
fnet_memcpy(&foreign_addr, ¶ms->foreign_addr, sizeof(foreign_addr));
fnet_socket_connect(fapp_bench.socket_foreign, (struct sockaddr *)(&foreign_addr), sizeof(foreign_addr));
do
{
option_len = sizeof(connection_state);
fnet_socket_getopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_STATE, (fnet_uint8_t*)&connection_state, &option_len);
}
while (connection_state == SS_CONNECTING);
if(connection_state != SS_CONNECTED)
{
fnet_shell_println(desc, "Connection failed.");
iterations = 0;
goto ERROR_2;
}
/* Sending.*/
fnet_shell_println(desc,"Sending.");
fapp_bench.bytes = 0;
fapp_bench.remote_bytes = 0;
cur_packet_number = 0;
buffer_offset = 0;
fapp_bench.first_time = fnet_timer_ticks();
while(1)
{
send_result = fnet_socket_send( fapp_bench.socket_foreign, (fnet_uint8_t*)(&fapp_bench.buffer[buffer_offset]), (packet_size-buffer_offset), 0);
fapp_bench.last_time = fnet_timer_ticks();
if ( send_result == FNET_ERR )
{
option_len = sizeof(sock_err);
fnet_socket_getopt(fapp_bench.socket_foreign, SOL_SOCKET, SO_ERROR, &sock_err, &option_len);
fnet_shell_println(desc, "Socket error = %d", sock_err);
iterations = 0;
#if 0
/* Print benchmark results.*/
fapp_bench_print_results (desc);
#endif
break;
}
else if(send_result)
{
fapp_bench.bytes += send_result;
buffer_offset += send_result;
if(buffer_offset == packet_size)
{
cur_packet_number ++;
buffer_offset = 0;
}
exit_flag = fnet_shell_ctrlc(desc); /* Check [Ctrl+c]*/
if((cur_packet_number >= params->packet_number)|| exit_flag)
{
if(exit_flag)
{
fnet_shell_println(desc, FAPP_SHELL_CANCELED_CTRL_C);
iterations = 0;
}
/* Print benchmark results.*/
fapp_bench_print_results (desc);
break;
}
}
else
{}
}
ERROR_2:
fnet_socket_close(fapp_bench.socket_foreign);
}
ERROR_1:
fnet_shell_println(desc, FAPP_BENCH_COMPLETED_STR);
}
/************************************************************************
* NAME: fapp_bench_udp_rx
*
* DESCRIPTION: Start Benchmark UDP server.
************************************************************************/
static void fapp_bench_udp_rx (fnet_shell_desc_t desc, fnet_address_family_t family, struct sockaddr *multicast_address /* optional, set to 0*/)
{
struct sockaddr local_addr;
const fnet_size_t bufsize_option = FAPP_BENCH_SOCKET_BUF_SIZE;
fnet_int32_t received;
fnet_char_t ip_str[FNET_IP_ADDR_STR_SIZE];
struct sockaddr addr;
fnet_size_t addr_len;
fnet_bool_t is_first = FNET_TRUE;
fnet_bool_t exit_flag = FNET_FALSE;
/* Create listen socket */
if((fapp_bench.socket_listen = fnet_socket(family, SOCK_DGRAM, 0)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket creation error.\n");
goto ERROR_1;
}
/*Bind.*/
fnet_memset_zero(&local_addr, sizeof(local_addr));
local_addr.sa_port = FAPP_BENCH_PORT;
local_addr.sa_family = family;
if(fnet_socket_bind(fapp_bench.socket_listen, &local_addr, sizeof(local_addr)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket bind error.\n");
goto ERROR_2;
}
/* Set socket options. */
if(
/* Set socket buffer size. */
(fnet_socket_setopt(fapp_bench.socket_listen, SOL_SOCKET, SO_RCVBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR) ||
(fnet_socket_setopt(fapp_bench.socket_listen, SOL_SOCKET, SO_SNDBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR)
)
{
FNET_DEBUG("BENCH: Socket setsockopt error.\n");
goto ERROR_2;
}
/* Join multicast group, if set. */
if(multicast_address)
{
#if FNET_CFG_IP4
if(multicast_address->sa_family == AF_INET)
{
struct ip_mreq mreq; /* Multicast group information.*/
mreq.imr_multiaddr.s_addr = ((struct sockaddr_in*)multicast_address)->sin_addr.s_addr;
mreq.imr_interface = 0; /* Default Interface.*/
/* Join multicast group. */
if(fnet_socket_setopt(fapp_bench.socket_listen, IPPROTO_IP, IP_ADD_MEMBERSHIP, (fnet_uint8_t *)&mreq, sizeof(mreq)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Joining to multicast group is failed.\n");
goto ERROR_2;
}
}
#endif
#if FNET_CFG_IP6
if(multicast_address->sa_family == AF_INET6)
{
struct ipv6_mreq mreq6; /* Multicast group information.*/
FNET_IP6_ADDR_COPY(&((struct sockaddr_in6*)multicast_address)->sin6_addr.s6_addr, &mreq6.ipv6imr_multiaddr.s6_addr);
mreq6.ipv6imr_interface = ((struct sockaddr_in6*)multicast_address)->sin6_scope_id;
/* Join multicast group. */
if(fnet_socket_setopt(fapp_bench.socket_listen, IPPROTO_IPV6, IPV6_JOIN_GROUP, (fnet_uint8_t *)&mreq6, sizeof(mreq6)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Joining to multicast group is failed.\n");
goto ERROR_2;
}
}
#endif
}
/* ------ Start test.----------- */
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fnet_shell_println(desc, " UDP RX Test" );
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fapp_netif_addr_print(desc, family, fnet_netif_get_default(), FNET_FALSE);
if(multicast_address)
{
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_S, "Multicast Group", fnet_inet_ntop(multicast_address->sa_family, (fnet_uint8_t*)(multicast_address->sa_data), ip_str, sizeof(ip_str)) );
}
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Local Port", FNET_NTOHS(FAPP_BENCH_PORT));
fnet_shell_println(desc, FAPP_TOCANCEL_STR);
fnet_shell_println(desc, FAPP_DELIMITER_STR);
while(exit_flag == FNET_FALSE) /* Main loop */
{
fnet_shell_println(desc, "Waiting.");
fapp_bench.bytes = 0;
fapp_bench.remote_bytes = 0;
addr_len = sizeof(addr);
is_first = FNET_TRUE;
while(exit_flag == FNET_FALSE) /* Test loop. */
{
/* Receive data */
received = fnet_socket_recvfrom (fapp_bench.socket_listen, (fnet_uint8_t*)(&fapp_bench.buffer[0]), FAPP_BENCH_BUFFER_SIZE, 0,
&addr, &addr_len );
if(received >= FAPP_BENCH_UDP_END_BUFFER_LENGTH)
{
/* Reset timeout. */
fapp_bench.last_time = fnet_timer_ticks();
if(is_first)
{
if( received > FAPP_BENCH_UDP_END_BUFFER_LENGTH )
{
fnet_shell_println(desc,"Receiving from %s:%d", fnet_inet_ntop(addr.sa_family, (fnet_uint8_t*)(addr.sa_data), ip_str, sizeof(ip_str)), fnet_ntohs(addr.sa_port));
fapp_bench.first_time = fnet_timer_ticks();
is_first = FNET_FALSE;
}
}
else
{
if(received == FAPP_BENCH_UDP_END_BUFFER_LENGTH ) /* End of transfer. */
{
/* Send ACK containing amount of received data.*/
fnet_uint32_t ack_bytes = fnet_htonl(fapp_bench.bytes);
/* Send several times, just to be sure that it is received/not lost.*/
fnet_socket_sendto(fapp_bench.socket_listen, (fnet_uint8_t*)(&ack_bytes), sizeof(ack_bytes), 0, (struct sockaddr*)&addr, sizeof(addr));
fnet_socket_sendto(fapp_bench.socket_listen, (fnet_uint8_t*)(&ack_bytes), sizeof(ack_bytes), 0, (struct sockaddr*)&addr, sizeof(addr));
fnet_socket_sendto(fapp_bench.socket_listen, (fnet_uint8_t*)(&ack_bytes), sizeof(ack_bytes), 0, (struct sockaddr*)&addr, sizeof(addr));
/* Print benchmark results.*/
fapp_bench_print_results (desc);
break;
}
else
fapp_bench.bytes += received;
}
}
else
{
/* Check error. Check timeout */
if(received == FNET_ERR)
{
fnet_shell_println(desc, "BENCH: Error (%d).", fnet_error_get());
break;
}
/* Check timeout. */
if((is_first == FNET_FALSE) &&
(fnet_timer_get_interval(fapp_bench.last_time, fnet_timer_ticks()) > (FAPP_UDP_TIMEOUT_MS/FNET_TIMER_PERIOD_MS)))
{
fnet_shell_println(desc, "BENCH: Exit on timeout.");
fapp_bench_print_results (desc);
break;
}
}
exit_flag = fnet_shell_ctrlc (desc);
}
}
ERROR_2:
fnet_socket_close(fapp_bench.socket_listen);
ERROR_1:
fnet_shell_println(desc, FAPP_BENCH_COMPLETED_STR);
}
/************************************************************************
* NAME: fapp_bench_tcp_rx
*
* DESCRIPTION: Start Benchmark TCP server.
************************************************************************/
static void fapp_bench_tcp_rx (fnet_shell_desc_t desc, fnet_address_family_t family)
{
struct sockaddr local_addr;
fnet_int32_t received;
fnet_char_t ip_str[FNET_IP_ADDR_STR_SIZE];
struct linger linger_option = {FNET_TRUE, /*l_onoff*/
4 /*l_linger*/
};
fnet_size_t bufsize_option = FAPP_BENCH_SOCKET_BUF_SIZE;
fnet_int32_t keepalive_option = 1;
fnet_int32_t keepcnt_option = FAPP_BENCH_TCP_KEEPCNT;
fnet_int32_t keepintvl_option = FAPP_BENCH_TCP_KEEPINTVL;
fnet_int32_t keepidle_option = FAPP_BENCH_TCP_KEEPIDLE;
struct sockaddr foreign_addr;
fnet_size_t addr_len;
fnet_bool_t exit_flag = FNET_FALSE;
fapp_bench.socket_foreign = FNET_ERR;
/* Create listen socket */
if((fapp_bench.socket_listen = fnet_socket(family, SOCK_STREAM, 0)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket creation error.");
goto ERROR_1;
}
/* Bind socket.*/
fnet_memset_zero(&local_addr, sizeof(local_addr));
local_addr.sa_port = FAPP_BENCH_PORT;
local_addr.sa_family = family;
if(fnet_socket_bind(fapp_bench.socket_listen, &local_addr, sizeof(local_addr)) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket bind error.");
goto ERROR_2;
}
/* Set Socket options. */
if( /* Setup linger option. */
(fnet_socket_setopt (fapp_bench.socket_listen, SOL_SOCKET, SO_LINGER, (fnet_uint8_t *)&linger_option, sizeof(linger_option)) == FNET_ERR) ||
/* Set socket buffer size. */
(fnet_socket_setopt(fapp_bench.socket_listen, SOL_SOCKET, SO_RCVBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR) ||
(fnet_socket_setopt(fapp_bench.socket_listen, SOL_SOCKET, SO_SNDBUF, (fnet_uint8_t *) &bufsize_option, sizeof(bufsize_option))== FNET_ERR) ||
/* Enable keepalive_option option. */
(fnet_socket_setopt (fapp_bench.socket_listen, SOL_SOCKET, SO_KEEPALIVE, (fnet_uint8_t *)&keepalive_option, sizeof(keepalive_option)) == FNET_ERR) ||
/* Keepalive probe retransmit limit. */
(fnet_socket_setopt (fapp_bench.socket_listen, IPPROTO_TCP, TCP_KEEPCNT, (fnet_uint8_t *)&keepcnt_option, sizeof(keepcnt_option)) == FNET_ERR) ||
/* Keepalive retransmit interval.*/
(fnet_socket_setopt (fapp_bench.socket_listen, IPPROTO_TCP, TCP_KEEPINTVL, (fnet_uint8_t *)&keepintvl_option, sizeof(keepintvl_option)) == FNET_ERR) ||
/* Time between keepalive probes.*/
(fnet_socket_setopt (fapp_bench.socket_listen, IPPROTO_TCP, TCP_KEEPIDLE, (fnet_uint8_t *)&keepidle_option, sizeof(keepidle_option)) == FNET_ERR)
)
{
FNET_DEBUG("BENCH: Socket setsockopt error.\n");
goto ERROR_2;
}
/* Listen. */
if(fnet_socket_listen(fapp_bench.socket_listen, 1) == FNET_ERR)
{
FNET_DEBUG("BENCH: Socket listen error.\n");
goto ERROR_2;
}
/* ------ Start test.----------- */
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fnet_shell_println(desc, " TCP RX Test");
fnet_shell_println(desc, FAPP_DELIMITER_STR);
fapp_netif_addr_print(desc, family, fnet_netif_get_default(), FNET_FALSE);
fnet_shell_println(desc, FAPP_SHELL_INFO_FORMAT_D, "Local Port", FNET_NTOHS(FAPP_BENCH_PORT));
fnet_shell_println(desc, FAPP_TOCANCEL_STR);
fnet_shell_println(desc, FAPP_DELIMITER_STR);
while(exit_flag == FNET_FALSE)
{
fnet_shell_println(desc, "Waiting.");
fapp_bench.bytes = 0;
fapp_bench.remote_bytes = 0;
if(fapp_bench.socket_foreign != FNET_ERR)
{
fnet_socket_close(fapp_bench.socket_foreign);
fapp_bench.socket_foreign = FNET_ERR;
}
while((fapp_bench.socket_foreign == FNET_ERR) && (exit_flag == FNET_FALSE))
{
/*Accept*/
addr_len = sizeof(foreign_addr);
fapp_bench.socket_foreign = fnet_socket_accept(fapp_bench.socket_listen, &foreign_addr, &addr_len);
exit_flag = fnet_shell_ctrlc (desc);
if(fapp_bench.socket_foreign != FNET_ERR)
{
fnet_shell_println(desc,"Receiving from %s:%d", fnet_inet_ntop(foreign_addr.sa_family, (fnet_uint8_t*)(foreign_addr.sa_data), ip_str, sizeof(ip_str)), fnet_ntohs(foreign_addr.sa_port));
fapp_bench.first_time = fnet_timer_ticks();
while(1) /* Receiving data.*/
{
received = fnet_socket_recv(fapp_bench.socket_foreign, (fnet_uint8_t*)(&fapp_bench.buffer[0]), FAPP_BENCH_BUFFER_SIZE, 0);
if ((received == FNET_ERR) || exit_flag)
{
fapp_bench.last_time = fnet_timer_ticks();
/* Print benchmark results.*/
fapp_bench_print_results (desc);
break;
}
else
{
fapp_bench.bytes += received;
}
exit_flag = fnet_shell_ctrlc (desc); /* Check [Ctrl+c]*/
}
}
}
}
fnet_socket_close(fapp_bench.socket_foreign);
ERROR_2:
fnet_socket_close(fapp_bench.socket_listen);
ERROR_1:
fnet_shell_println(desc, FAPP_BENCH_COMPLETED_STR);
}
/************************************************************************
* NAME: fnet_ping_request
*
* DESCRIPTION: Initializes PING service.
************************************************************************/
fnet_return_t fnet_ping_request( struct fnet_ping_params *params )
{
const fnet_size_t bufsize_option = FNET_PING_BUFFER_SIZE;
/* Check input parameters. */
if((params == 0) || (params->packet_count == 0u) || (fnet_socket_addr_is_unspecified(¶ms->target_addr)))
{
FNET_DEBUG_PING(FNET_PING_ERR_PARAMS);
goto ERROR;
}
/* Check if PING service is free.*/
if(fnet_ping_if.state != FNET_PING_STATE_DISABLED)
{
FNET_DEBUG_PING(FNET_PING_ERR_IS_INITIALIZED);
goto ERROR;
}
/* Save input parmeters.*/
fnet_ping_if.handler = params->handler;
fnet_ping_if.handler_cookie = params->cookie;
fnet_ping_if.timeout_clk = params->timeout/FNET_TIMER_PERIOD_MS;
if(fnet_ping_if.timeout_clk == 0u)
{
fnet_ping_if.timeout_clk = 1u;
}
fnet_ping_if.family = params->target_addr.sa_family;
fnet_ping_if.packet_count = params->packet_count;
fnet_ping_if.pattern = params->pattern;
fnet_ping_if.packet_size = params->packet_size;
if(fnet_ping_if.packet_size > FNET_CFG_PING_PACKET_MAX)
{
fnet_ping_if.packet_size = FNET_CFG_PING_PACKET_MAX;
}
fnet_ping_if.target_addr = params->target_addr;
/* Create socket */
if((fnet_ping_if.socket_foreign = fnet_socket(fnet_ping_if.family, SOCK_RAW, (fnet_uint32_t)((params->target_addr.sa_family == AF_INET) ? IPPROTO_ICMP : IPPROTO_ICMPV6))) == FNET_ERR)
{
FNET_DEBUG_PING(FNET_PING_ERR_SOCKET_CREATION);
goto ERROR;
}
/* Set Socket options. */
#if FNET_CFG_IP4
if(fnet_ping_if.family == AF_INET)
{
fnet_socket_setopt(fnet_ping_if.socket_foreign, IPPROTO_IP, IP_TTL, (fnet_uint8_t *) ¶ms->ttl, sizeof(params->ttl));
}
#endif
#if FNET_CFG_IP6
if(fnet_ping_if.family == AF_INET6)
{
fnet_socket_setopt(fnet_ping_if.socket_foreign, IPPROTO_IPV6, IPV6_UNICAST_HOPS, (fnet_uint8_t *) ¶ms->ttl, sizeof(params->ttl));
}
#endif
fnet_socket_setopt(fnet_ping_if.socket_foreign, SOL_SOCKET, SO_RCVBUF, &bufsize_option, sizeof(bufsize_option));
fnet_socket_setopt(fnet_ping_if.socket_foreign, SOL_SOCKET, SO_SNDBUF, &bufsize_option, sizeof(bufsize_option));
/* Register PING service. */
fnet_ping_if.service_descriptor = fnet_poll_service_register(fnet_ping_state_machine, (void *) &fnet_ping_if);
if(fnet_ping_if.service_descriptor == (fnet_poll_desc_t)FNET_ERR)
{
FNET_DEBUG_PING(FNET_PING_ERR_SERVICE);
goto ERROR_1;
}
fnet_ping_if.state = FNET_PING_STATE_SENDING_REQUEST;
return FNET_OK;
ERROR_1:
fnet_socket_close(fnet_ping_if.socket_foreign);
ERROR:
return FNET_ERR;
}
/************************************************************************
* DESCRIPTION: Initialization of the Telnet server.
*************************************************************************/
fnet_telnet_desc_t fnet_telnet_init( struct fnet_telnet_params *params )
{
struct sockaddr local_addr;
struct fnet_telnet_if *telnet_if = 0;
/* Socket options. */
const struct linger linger_option = {FNET_TRUE, /*l_onoff*/
4 /*l_linger*/
};
const fnet_size_t bufsize_option = FNET_CFG_TELNET_SOCKET_BUF_SIZE;
const fnet_int32_t keepalive_option = 1;
const fnet_int32_t keepcnt_option = FNET_TELNET_TCP_KEEPCNT;
const fnet_int32_t keepintvl_option = FNET_TELNET_TCP_KEEPINTVL;
const fnet_int32_t keepidle_option = FNET_TELNET_TCP_KEEPIDLE;
fnet_index_t i;
if(params == 0 )
{
FNET_DEBUG_TELNET("TELNET: Wrong init parameters.");
goto ERROR_1;
}
/* Try to find free Telnet server descriptor. */
for(i = 0u; i < FNET_CFG_TELNET_MAX; i++)
{
if(telnet_if_list[i].enabled == FNET_FALSE)
{
telnet_if = &telnet_if_list[i];
break;
}
}
/* No free Telnet server descriptor. */
if(telnet_if == 0)
{
FNET_DEBUG_TELNET("TELNET: No free Telnet Server.");
goto ERROR_1;
}
fnet_memcpy(&local_addr, ¶ms->address, sizeof(local_addr));
if(local_addr.sa_port == 0u)
{
local_addr.sa_port = FNET_CFG_TELNET_PORT; /* Aply the default port. */
}
if(local_addr.sa_family == AF_UNSPEC)
{
local_addr.sa_family = AF_SUPPORTED; /* Assign supported families.*/
}
/* Create listen socket */
if((telnet_if->socket_listen = fnet_socket(local_addr.sa_family, SOCK_STREAM, 0u)) == FNET_ERR)
{
FNET_DEBUG_TELNET("TELNET: Socket creation error.");
goto ERROR_1;
}
if(fnet_socket_bind(telnet_if->socket_listen, (struct sockaddr *)(&local_addr), sizeof(local_addr)) == FNET_ERR)
{
FNET_DEBUG_TELNET("TELNET: Socket bind error.");
goto ERROR_2;
}
/* Set socket options. */
if( /* Setup linger option. */
(fnet_socket_setopt (telnet_if->socket_listen, SOL_SOCKET, SO_LINGER, &linger_option, sizeof(linger_option)) == FNET_ERR) ||
/* Set socket buffer size. */
(fnet_socket_setopt(telnet_if->socket_listen, SOL_SOCKET, SO_RCVBUF, &bufsize_option, sizeof(bufsize_option)) == FNET_ERR) ||
(fnet_socket_setopt(telnet_if->socket_listen, SOL_SOCKET, SO_SNDBUF, &bufsize_option, sizeof(bufsize_option)) == FNET_ERR) ||
/* Enable keepalive_option option. */
(fnet_socket_setopt (telnet_if->socket_listen, SOL_SOCKET, SO_KEEPALIVE, &keepalive_option, sizeof(keepalive_option)) == FNET_ERR) ||
/* Keepalive probe retransmit limit. */
(fnet_socket_setopt (telnet_if->socket_listen, IPPROTO_TCP, TCP_KEEPCNT, &keepcnt_option, sizeof(keepcnt_option)) == FNET_ERR) ||
/* Keepalive retransmit interval.*/
(fnet_socket_setopt (telnet_if->socket_listen, IPPROTO_TCP, TCP_KEEPINTVL, &keepintvl_option, sizeof(keepintvl_option)) == FNET_ERR) ||
/* Time between keepalive probes.*/
(fnet_socket_setopt (telnet_if->socket_listen, IPPROTO_TCP, TCP_KEEPIDLE, &keepidle_option, sizeof(keepidle_option)) == FNET_ERR)
)
{
FNET_DEBUG_TELNET("TELNET: Socket setsockopt() error.");
goto ERROR_2;
}
telnet_if->backlog = FNET_CFG_TELNET_SESSION_MAX;
if(fnet_socket_listen(telnet_if->socket_listen, telnet_if->backlog) == FNET_ERR)
{
FNET_DEBUG_TELNET("TELNET: Socket listen error.");
goto ERROR_2;
}
/* Register service. */
telnet_if->service_descriptor = fnet_poll_service_register(fnet_telnet_state_machine, (void *) telnet_if);
if(telnet_if->service_descriptor == 0)
{
FNET_DEBUG_TELNET("TELNET: Service registration error.");
goto ERROR_2;
}
for(i = 0u; i < FNET_CFG_TELNET_SESSION_MAX; i++)
{
struct fnet_telnet_session_if *session = &telnet_if->session[i];
/* Reset buffer pointers. Move it to init state. */
session->tx_buffer_head_index = 0u;
session->rx_buffer_head = session->rx_buffer;
session->rx_buffer_tail = session->rx_buffer;
session->rx_buffer_end = &session->rx_buffer[FNET_TELNET_RX_BUFFER_SIZE];
/* Setup stream. */
session->stream.id = (fnet_index_t)(session);
session->stream.putchar = fnet_telnet_putchar;
session->stream.getchar = fnet_telnet_getchar;
session->stream.flush = fnet_telnet_flush;
/* Init shell. */
session->shell_params.shell = params->shell;
session->shell_params.cmd_line_buffer = session->cmd_line_buffer;
session->shell_params.cmd_line_buffer_size = sizeof(session->cmd_line_buffer);
session->shell_params.stream = &session->stream;
session->shell_params.echo = FNET_CFG_TELNET_SHELL_ECHO ? FNET_TRUE : FNET_FALSE;
session->socket_foreign = FNET_ERR;
session->state = FNET_TELNET_STATE_LISTENING;
}
telnet_if->session_active = FNET_NULL;
telnet_if->enabled = FNET_TRUE;
return (fnet_telnet_desc_t)telnet_if;
ERROR_2:
fnet_socket_close(telnet_if->socket_listen);
ERROR_1:
return 0;
}
