0,0 → 1,554 |
/* |
* Copyright (c) 2009 Lukas Mejdrech |
* All rights reserved. |
* |
* Redistribution and use in source and binary forms, with or without |
* modification, are permitted provided that the following conditions |
* are met: |
* |
* - Redistributions of source code must retain the above copyright |
* notice, this list of conditions and the following disclaimer. |
* - 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. |
* - 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. |
*/ |
|
/** @addtogroup nettest |
* @{ |
*/ |
|
/** @file |
* Networking test 2 application - transfer. |
*/ |
|
#include <malloc.h> |
#include <stdio.h> |
#include <string.h> |
#include <task.h> |
#include <time.h> |
|
#include "../../include/in.h" |
#include "../../include/in6.h" |
#include "../../include/inet.h" |
#include "../../include/socket.h" |
|
#include "../../err.h" |
|
#include "../parse.h" |
#include "../print_error.h" |
|
/** Echo module name. |
*/ |
#define NAME "Nettest2" |
|
/** Packet data pattern. |
*/ |
#define NETTEST2_TEXT "Networking test 2 - transfer" |
|
/** Module entry point. |
* Starts testing. |
* @param argc The number of command line parameters. Input parameter. |
* @param argv The command line parameters. Input parameter. |
* @returns EOK on success. |
*/ |
int main( int argc, char * argv[] ); |
|
/** Prints the application help. |
*/ |
void print_help( void ); |
|
/** Translates the character string to the protocol family number. |
* @param name The protocol family name. Input parameter. |
* @returns The corresponding protocol family number. |
* @returns EPFNOSUPPORTED if the protocol family is not supported. |
*/ |
int parse_protocol_family( const char * name ); |
|
/** Translates the character string to the socket type number. |
* @param name The socket type name. Input parameter. |
* @returns The corresponding socket type number. |
* @returns ESOCKNOSUPPORTED if the socket type is not supported. |
*/ |
int parse_socket_type( const char * name ); |
|
/** Refreshes the data. |
* Fills the data block with the NETTEST1_TEXT pattern. |
* @param data The data block. Output parameter. |
* @param size The data block size in bytes. Input parameter |
*/ |
void refresh_data( char * data, size_t size ); |
|
/** Creates new sockets. |
* @param verbose A value indicating whether to print out verbose information. Input parameter. |
* @param socket_ids A field to store the socket identifiers. Output parameter. |
* @param sockets The number of sockets to create. Should be at most the size of the field. Input parameter. |
* @param family The socket address family. Input parameter. |
* @param type The socket type. Input parameter. |
* @returns EOK on success. |
* @returns Other error codes as defined for the socket() function. |
*/ |
int sockets_create( int verbose, int * socket_ids, int sockets, int family, sock_type_t type ); |
|
/** Closes sockets. |
* @param verbose A value indicating whether to print out verbose information. Input parameter. |
* @param socket_ids A field of stored socket identifiers. Input parameter. |
* @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
* @returns EOK on success. |
* @returns Other error codes as defined for the closesocket() function. |
*/ |
int sockets_close( int verbose, int * socket_ids, int sockets ); |
|
/** Connects sockets. |
* @param verbose A value indicating whether to print out verbose information. Input parameter. |
* @param socket_ids A field of stored socket identifiers. Input parameter. |
* @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
* @param address The destination host address to connect to. Input parameter. |
* @param addrlen The length of the destination address in bytes. Input parameter. |
* @returns EOK on success. |
* @returns Other error codes as defined for the connect() function. |
*/ |
int sockets_connect( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen ); |
|
/** Sends data via sockets. |
* @param verbose A value indicating whether to print out verbose information. Input parameter. |
* @param socket_ids A field of stored socket identifiers. Input parameter. |
* @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
* @param address The destination host address to send data to. Input parameter. |
* @param addrlen The length of the destination address in bytes. Input parameter. |
* @param data The data to be sent. Input parameter. |
* @param size The data size in bytes. Input parameter |
* @param messages The number of datagrams per socket to be sent. Input parameter. |
* @returns EOK on success. |
* @returns Other error codes as defined for the sendto() function. |
*/ |
int sockets_sendto( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen, char * data, int size, int messages ); |
|
/** Receives data via sockets. |
* @param verbose A value indicating whether to print out verbose information. Input parameter. |
* @param socket_ids A field of stored socket identifiers. Input parameter. |
* @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
* @param address The source host address of received datagrams. Input parameter. |
* @param addrlen The maximum length of the source address in bytes. The actual size of the source address is set instead. Input/output parameter. |
* @param data The received data. Output parameter. |
* @param size The maximum data size in bytes. Input parameter |
* @param messages The number of datagrams per socket to be received. Input parameter. |
* @returns EOK on success. |
* @returns Other error codes as defined for the recvfrom() function. |
*/ |
int sockets_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ); |
|
/** Sends and receives data via sockets. |
* Each datagram is sent and a reply read consequently. |
* The next datagram is sent after the reply is received. |
* @param verbose A value indicating whether to print out verbose information. Input parameter. |
* @param socket_ids A field of stored socket identifiers. Input parameter. |
* @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
* @param address The destination host address to send data to. The source host address of received datagrams is set instead. Input/output parameter. |
* @param addrlen The length of the destination address in bytes. Input parameter. |
* @param data The data to be sent. The received data are set instead. Input/output parameter. |
* @param size The data size in bytes. Input parameter |
* @param messages The number of datagrams per socket to be received. Input parameter. |
* @returns EOK on success. |
* @returns Other error codes as defined for the recvfrom() function. |
*/ |
int sockets_sendto_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ); |
|
/** Prints a mark. |
* If the index is a multiple of ten, a different mark is printed. |
* @param index The index of the mark to be printed. Input parameter. |
*/ |
void print_mark( int index ); |
|
void print_help( void ){ |
printf( |
"Network Networking test 2 aplication - UDP transfer\n" \ |
"Usage: echo [options] numeric_address\n" \ |
"Where options are:\n" \ |
"-f protocol_family | --family=protocol_family\n" \ |
"\tThe listenning socket protocol family. Only the PF_INET and PF_INET6 are supported.\n" |
"\n" \ |
"-h | --help\n" \ |
"\tShow this application help.\n" |
"\n" \ |
"-m count | --messages=count\n" \ |
"\tThe number of messages to send and receive per socket. The default is 10.\n" \ |
"\n" \ |
"-n sockets | --sockets=count\n" \ |
"\tThe number of sockets to use. The default is 10.\n" \ |
"\n" \ |
"-p port_number | --port=port_number\n" \ |
"\tThe port number the application should send messages to. The default is 7.\n" \ |
"\n" \ |
"-s packet_size | --size=packet_size\n" \ |
"\tThe packet data size the application sends. The default is 29 bytes.\n" \ |
"\n" \ |
"-v | --verbose\n" \ |
"\tShow all output messages.\n" |
); |
} |
|
int parse_protocol_family( const char * name ){ |
if( str_lcmp( name, "PF_INET", 7 ) == 0 ){ |
return PF_INET; |
}else if( str_lcmp( name, "PF_INET6", 8 ) == 0 ){ |
return PF_INET6; |
} |
return EPFNOSUPPORT; |
} |
|
int parse_socket_type( const char * name ){ |
if( str_lcmp( name, "SOCK_DGRAM", 11 ) == 0 ){ |
return SOCK_DGRAM; |
}else if( str_lcmp( name, "SOCK_STREAM", 12 ) == 0 ){ |
return SOCK_STREAM; |
} |
return ESOCKTNOSUPPORT; |
} |
|
void refresh_data( char * data, size_t size ){ |
size_t length; |
|
// fill the data |
length = 0; |
while( size > length + sizeof( NETTEST2_TEXT ) - 1 ){ |
memcpy( data + length, NETTEST2_TEXT, sizeof( NETTEST2_TEXT ) - 1 ); |
length += sizeof( NETTEST2_TEXT ) - 1; |
} |
memcpy( data + length, NETTEST2_TEXT, size - length ); |
data[ size ] = '\0'; |
} |
|
int sockets_create( int verbose, int * socket_ids, int sockets, int family, sock_type_t type ){ |
int index; |
|
if( verbose ) printf( "Create\t" ); |
fflush( stdout ); |
for( index = 0; index < sockets; ++ index ){ |
socket_ids[ index ] = socket( family, type, 0 ); |
if( socket_ids[ index ] < 0 ){ |
printf( "Socket %d (%d) error:\n", index, socket_ids[ index ] ); |
socket_print_error( stderr, socket_ids[ index ], "Socket create: ", "\n" ); |
return socket_ids[ index ]; |
} |
if( verbose ) print_mark( index ); |
} |
return EOK; |
} |
|
int sockets_close( int verbose, int * socket_ids, int sockets ){ |
ERROR_DECLARE; |
|
int index; |
|
if( verbose ) printf( "\tClose\t" ); |
fflush( stdout ); |
for( index = 0; index < sockets; ++ index ){ |
if( ERROR_OCCURRED( closesocket( socket_ids[ index ] ))){ |
printf( "Socket %d (%d) error:\n", index, socket_ids[ index ] ); |
socket_print_error( stderr, ERROR_CODE, "Socket close: ", "\n" ); |
return ERROR_CODE; |
} |
if( verbose ) print_mark( index ); |
} |
return EOK; |
} |
|
int sockets_connect( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen ){ |
ERROR_DECLARE; |
|
int index; |
|
if( verbose ) printf( "\tConnect\t" ); |
fflush( stdout ); |
for( index = 0; index < sockets; ++ index ){ |
if( ERROR_OCCURRED( connect( socket_ids[ index ], address, addrlen ))){ |
socket_print_error( stderr, ERROR_CODE, "Socket connect: ", "\n" ); |
return ERROR_CODE; |
} |
if( verbose ) print_mark( index ); |
} |
return EOK; |
} |
|
int sockets_sendto( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen, char * data, int size, int messages ){ |
ERROR_DECLARE; |
|
int index; |
int message; |
|
if( verbose ) printf( "\tSendto\t" ); |
fflush( stdout ); |
for( index = 0; index < sockets; ++ index ){ |
for( message = 0; message < messages; ++ message ){ |
if( ERROR_OCCURRED( sendto( socket_ids[ index ], data, size, 0, address, addrlen ))){ |
printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
socket_print_error( stderr, ERROR_CODE, "Socket send: ", "\n" ); |
return ERROR_CODE; |
} |
} |
if( verbose ) print_mark( index ); |
} |
return EOK; |
} |
|
int sockets_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ){ |
int value; |
int index; |
int message; |
|
if( verbose ) printf( "\tRecvfrom\t" ); |
fflush( stdout ); |
for( index = 0; index < sockets; ++ index ){ |
for( message = 0; message < messages; ++ message ){ |
value = recvfrom( socket_ids[ index ], data, size, 0, address, addrlen ); |
if( value < 0 ){ |
printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
socket_print_error( stderr, value, "Socket receive: ", "\n" ); |
return value; |
} |
} |
if( verbose ) print_mark( index ); |
} |
return EOK; |
} |
|
int sockets_sendto_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ){ |
ERROR_DECLARE; |
|
int value; |
int index; |
int message; |
|
if( verbose ) printf( "\tSendto and recvfrom\t" ); |
fflush( stdout ); |
for( index = 0; index < sockets; ++ index ){ |
for( message = 0; message < messages; ++ message ){ |
if( ERROR_OCCURRED( sendto( socket_ids[ index ], data, size, 0, address, * addrlen ))){ |
printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
socket_print_error( stderr, ERROR_CODE, "Socket send: ", "\n" ); |
return ERROR_CODE; |
} |
value = recvfrom( socket_ids[ index ], data, size, 0, address, addrlen ); |
if( value < 0 ){ |
printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
socket_print_error( stderr, value, "Socket receive: ", "\n" ); |
return value; |
} |
} |
if( verbose ) print_mark( index ); |
} |
return EOK; |
} |
|
void print_mark( int index ){ |
if(( index + 1 ) % 10 ){ |
printf( "*" ); |
}else{ |
printf( "|" ); |
} |
fflush( stdout ); |
} |
|
int main( int argc, char * argv[] ){ |
ERROR_DECLARE; |
|
size_t size = 29; |
int verbose = 0; |
sock_type_t type = SOCK_DGRAM; |
int sockets = 10; |
int messages = 10; |
int family = PF_INET; |
uint16_t port = 7; |
|
socklen_t max_length = sizeof( struct sockaddr_in6 ); |
uint8_t address_data[ max_length ]; |
struct sockaddr * address = ( struct sockaddr * ) address_data; |
struct sockaddr_in * address_in = ( struct sockaddr_in * ) address; |
struct sockaddr_in6 * address_in6 = ( struct sockaddr_in6 * ) address; |
socklen_t addrlen; |
// char address_string[ INET6_ADDRSTRLEN ]; |
uint8_t * address_start; |
|
int * socket_ids; |
char * data; |
int value; |
int index; |
struct timeval time_before; |
struct timeval time_after; |
|
printf( "Task %d - ", task_get_id()); |
printf( "%s\n", NAME ); |
|
if( argc <= 1 ){ |
print_help(); |
return EINVAL; |
} |
|
for( index = 1; ( index < argc - 1 ) || (( index == argc ) && ( argv[ index ][ 0 ] == '-' )); ++ index ){ |
if( argv[ index ][ 0 ] == '-' ){ |
switch( argv[ index ][ 1 ] ){ |
case 'f': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & family, "protocol family", 0, parse_protocol_family )); |
break; |
case 'h': print_help(); |
return EOK; |
break; |
case 'm': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & messages, "message count", 0 )); |
break; |
case 'n': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & sockets, "socket count", 0 )); |
break; |
case 'p': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 0 )); |
port = ( uint16_t ) value; |
break; |
case 's': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "packet size", 0 )); |
size = (value >= 0 ) ? ( size_t ) value : 0; |
break; |
case 't': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "socket type", 0, parse_socket_type )); |
type = ( sock_type_t ) value; |
break; |
case 'v': verbose = 1; |
break; |
case '-': if( str_lcmp( argv[ index ] + 2, "family=", 7 ) == 0 ){ |
ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & family, "protocol family", 9, parse_protocol_family )); |
}else if( str_lcmp( argv[ index ] + 2, "help", 5 ) == 0 ){ |
print_help(); |
return EOK; |
}else if( str_lcmp( argv[ index ] + 2, "messages=", 6 ) == 0 ){ |
ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & messages, "message count", 8 )); |
}else if( str_lcmp( argv[ index ] + 2, "sockets=", 6 ) == 0 ){ |
ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & sockets, "socket count", 8 )); |
}else if( str_lcmp( argv[ index ] + 2, "port=", 5 ) == 0 ){ |
ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 7 )); |
port = ( uint16_t ) value; |
}else if( str_lcmp( argv[ index ] + 2, "type=", 5 ) == 0 ){ |
ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "socket type", 7, parse_socket_type )); |
type = ( sock_type_t ) value; |
}else if( str_lcmp( argv[ index ] + 2, "verbose", 8 ) == 0 ){ |
verbose = 1; |
}else{ |
print_unrecognized( index, argv[ index ] + 2 ); |
print_help(); |
return EINVAL; |
} |
break; |
default: |
print_unrecognized( index, argv[ index ] + 1 ); |
print_help(); |
return EINVAL; |
} |
}else{ |
print_unrecognized( index, argv[ index ] ); |
print_help(); |
return EINVAL; |
} |
} |
|
bzero( address_data, max_length ); |
switch( family ){ |
case PF_INET: |
address_in->sin_family = AF_INET; |
address_in->sin_port = htons( port ); |
address_start = ( uint8_t * ) & address_in->sin_addr.s_addr; |
addrlen = sizeof( struct sockaddr_in ); |
break; |
case PF_INET6: |
address_in6->sin6_family = AF_INET6; |
address_in6->sin6_port = htons( port ); |
address_start = ( uint8_t * ) & address_in6->sin6_addr.s6_addr; |
addrlen = sizeof( struct sockaddr_in6 ); |
break; |
default: |
fprintf( stderr, "Address family is not supported\n" ); |
return EAFNOSUPPORT; |
} |
|
if( ERROR_OCCURRED( inet_pton( family, argv[ argc - 1 ], address_start ))){ |
fprintf( stderr, "Address parse error %d\n", ERROR_CODE ); |
return ERROR_CODE; |
} |
|
if( size <= 0 ){ |
fprintf( stderr, "Data buffer size too small (%d). Using 1024 bytes instead.\n", size ); |
size = 1024; |
} |
// size plus terminating null (\0) |
data = ( char * ) malloc( size + 1 ); |
if( ! data ){ |
fprintf( stderr, "Failed to allocate data buffer.\n" ); |
return ENOMEM; |
} |
refresh_data( data, size ); |
|
if( sockets <= 0 ){ |
fprintf( stderr, "Socket count too small (%d). Using 2 instead.\n", sockets ); |
sockets = 2; |
} |
// count plus terminating null (\0) |
socket_ids = ( int * ) malloc( sizeof( int ) * ( sockets + 1 )); |
if( ! socket_ids ){ |
fprintf( stderr, "Failed to allocate receive buffer.\n" ); |
return ENOMEM; |
} |
socket_ids[ sockets ] = NULL; |
|
if( verbose ) printf( "Starting tests\n" ); |
|
ERROR_PROPAGATE( sockets_create( verbose, socket_ids, sockets, family, type )); |
if( type == SOCK_STREAM ){ |
ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, sockets, address, addrlen )); |
} |
|
if( ERROR_OCCURRED( gettimeofday( & time_before, NULL ))){ |
fprintf( stderr, "Get time of day error %d\n", ERROR_CODE ); |
return ERROR_CODE; |
} |
|
ERROR_PROPAGATE( sockets_sendto_recvfrom( verbose, socket_ids, sockets, address, & addrlen, data, size, messages )); |
|
if( ERROR_OCCURRED( gettimeofday( & time_after, NULL ))){ |
fprintf( stderr, "Get time of day error %d\n", ERROR_CODE ); |
return ERROR_CODE; |
} |
|
if( verbose ) printf( "\tOK\n" ); |
|
printf( "sendto + recvfrom tested in %d microseconds\n", tv_sub( & time_after, & time_before )); |
|
if( ERROR_OCCURRED( gettimeofday( & time_before, NULL ))){ |
fprintf( stderr, "Get time of day error %d\n", ERROR_CODE ); |
return ERROR_CODE; |
} |
|
ERROR_PROPAGATE( sockets_sendto( verbose, socket_ids, sockets, address, addrlen, data, size, messages )); |
ERROR_PROPAGATE( sockets_recvfrom( verbose, socket_ids, sockets, address, & addrlen, data, size, messages )); |
|
if( ERROR_OCCURRED( gettimeofday( & time_after, NULL ))){ |
fprintf( stderr, "Get time of day error %d\n", ERROR_CODE ); |
return ERROR_CODE; |
} |
|
if( verbose ) printf( "\tOK\n" ); |
|
printf( "sendto, recvfrom tested in %d microseconds\n", tv_sub( & time_after, & time_before )); |
|
ERROR_PROPAGATE( sockets_close( verbose, socket_ids, sockets )); |
|
if( verbose ) printf( "Exiting\n" ); |
|
return EOK; |
} |
|
/** @} |
*/ |
Property changes: |
Added: svn:eol-style |
+native |
\ No newline at end of property |