Rev 4743 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
4740 | mejdrech | 1 | /* |
2 | * Copyright (c) 2009 Lukas Mejdrech |
||
3 | * All rights reserved. |
||
4 | * |
||
5 | * Redistribution and use in source and binary forms, with or without |
||
6 | * modification, are permitted provided that the following conditions |
||
7 | * are met: |
||
8 | * |
||
9 | * - Redistributions of source code must retain the above copyright |
||
10 | * notice, this list of conditions and the following disclaimer. |
||
11 | * - Redistributions in binary form must reproduce the above copyright |
||
12 | * notice, this list of conditions and the following disclaimer in the |
||
13 | * documentation and/or other materials provided with the distribution. |
||
14 | * - The name of the author may not be used to endorse or promote products |
||
15 | * derived from this software without specific prior written permission. |
||
16 | * |
||
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
||
18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
||
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
||
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
||
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
||
22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||
23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||
24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||
25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
||
26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||
27 | */ |
||
28 | |||
4749 | mejdrech | 29 | /** @addtogroup nettest |
4740 | mejdrech | 30 | * @{ |
31 | */ |
||
32 | |||
33 | /** @file |
||
4749 | mejdrech | 34 | * Networking test 1 application - sockets. |
4740 | mejdrech | 35 | */ |
36 | |||
37 | #include <malloc.h> |
||
38 | #include <stdio.h> |
||
39 | #include <string.h> |
||
40 | #include <task.h> |
||
4743 | mejdrech | 41 | #include <time.h> |
4740 | mejdrech | 42 | |
43 | #include "../../include/in.h" |
||
44 | #include "../../include/in6.h" |
||
45 | #include "../../include/inet.h" |
||
46 | #include "../../include/socket.h" |
||
47 | |||
48 | #include "../../err.h" |
||
49 | |||
50 | #include "../parse.h" |
||
51 | #include "../print_error.h" |
||
52 | |||
53 | /** Echo module name. |
||
54 | */ |
||
55 | #define NAME "Nettest1" |
||
56 | |||
57 | /** Packet data pattern. |
||
58 | */ |
||
4749 | mejdrech | 59 | #define NETTEST1_TEXT "Networking test 1 - sockets" |
4740 | mejdrech | 60 | |
61 | /** Module entry point. |
||
62 | * Starts testing. |
||
63 | * @param argc The number of command line parameters. Input parameter. |
||
64 | * @param argv The command line parameters. Input parameter. |
||
65 | * @returns EOK on success. |
||
66 | */ |
||
67 | int main( int argc, char * argv[] ); |
||
68 | |||
69 | /** Prints the application help. |
||
70 | */ |
||
71 | void print_help( void ); |
||
72 | |||
73 | /** Translates the character string to the protocol family number. |
||
74 | * @param name The protocol family name. Input parameter. |
||
75 | * @returns The corresponding protocol family number. |
||
76 | * @returns EPFNOSUPPORTED if the protocol family is not supported. |
||
77 | */ |
||
78 | int parse_protocol_family( const char * name ); |
||
79 | |||
80 | /** Translates the character string to the socket type number. |
||
81 | * @param name The socket type name. Input parameter. |
||
82 | * @returns The corresponding socket type number. |
||
83 | * @returns ESOCKNOSUPPORTED if the socket type is not supported. |
||
84 | */ |
||
85 | int parse_socket_type( const char * name ); |
||
86 | |||
4749 | mejdrech | 87 | /** Refreshes the data. |
88 | * Fills the data block with the NETTEST1_TEXT pattern. |
||
89 | * @param data The data block. Output parameter. |
||
90 | * @param size The data block size in bytes. Input parameter |
||
91 | */ |
||
4743 | mejdrech | 92 | void refresh_data( char * data, size_t size ); |
4749 | mejdrech | 93 | |
94 | /** Creates new sockets. |
||
95 | * @param verbose A value indicating whether to print out verbose information. Input parameter. |
||
96 | * @param socket_ids A field to store the socket identifiers. Output parameter. |
||
97 | * @param sockets The number of sockets to create. Should be at most the size of the field. Input parameter. |
||
98 | * @param family The socket address family. Input parameter. |
||
99 | * @param type The socket type. Input parameter. |
||
100 | * @returns EOK on success. |
||
101 | * @returns Other error codes as defined for the socket() function. |
||
102 | */ |
||
4740 | mejdrech | 103 | int sockets_create( int verbose, int * socket_ids, int sockets, int family, sock_type_t type ); |
4749 | mejdrech | 104 | |
105 | /** Closes sockets. |
||
106 | * @param verbose A value indicating whether to print out verbose information. Input parameter. |
||
107 | * @param socket_ids A field of stored socket identifiers. Input parameter. |
||
108 | * @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
||
109 | * @returns EOK on success. |
||
110 | * @returns Other error codes as defined for the closesocket() function. |
||
111 | */ |
||
4740 | mejdrech | 112 | int sockets_close( int verbose, int * socket_ids, int sockets ); |
4749 | mejdrech | 113 | |
114 | /** Connects sockets. |
||
115 | * @param verbose A value indicating whether to print out verbose information. Input parameter. |
||
116 | * @param socket_ids A field of stored socket identifiers. Input parameter. |
||
117 | * @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
||
118 | * @param address The destination host address to connect to. Input parameter. |
||
119 | * @param addrlen The length of the destination address in bytes. Input parameter. |
||
120 | * @returns EOK on success. |
||
121 | * @returns Other error codes as defined for the connect() function. |
||
122 | */ |
||
4743 | mejdrech | 123 | int sockets_connect( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen ); |
4749 | mejdrech | 124 | |
125 | /** Sends data via sockets. |
||
126 | * @param verbose A value indicating whether to print out verbose information. Input parameter. |
||
127 | * @param socket_ids A field of stored socket identifiers. Input parameter. |
||
128 | * @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
||
129 | * @param address The destination host address to send data to. Input parameter. |
||
130 | * @param addrlen The length of the destination address in bytes. Input parameter. |
||
131 | * @param data The data to be sent. Input parameter. |
||
132 | * @param size The data size in bytes. Input parameter |
||
133 | * @param messages The number of datagrams per socket to be sent. Input parameter. |
||
134 | * @returns EOK on success. |
||
135 | * @returns Other error codes as defined for the sendto() function. |
||
136 | */ |
||
4740 | mejdrech | 137 | int sockets_sendto( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen, char * data, int size, int messages ); |
4749 | mejdrech | 138 | |
139 | /** Receives data via sockets. |
||
140 | * @param verbose A value indicating whether to print out verbose information. Input parameter. |
||
141 | * @param socket_ids A field of stored socket identifiers. Input parameter. |
||
142 | * @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
||
143 | * @param address The source host address of received datagrams. Input parameter. |
||
144 | * @param addrlen The maximum length of the source address in bytes. The actual size of the source address is set instead. Input/output parameter. |
||
145 | * @param data The received data. Output parameter. |
||
146 | * @param size The maximum data size in bytes. Input parameter |
||
147 | * @param messages The number of datagrams per socket to be received. Input parameter. |
||
148 | * @returns EOK on success. |
||
149 | * @returns Other error codes as defined for the recvfrom() function. |
||
150 | */ |
||
4740 | mejdrech | 151 | int sockets_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ); |
4749 | mejdrech | 152 | |
153 | /** Sends and receives data via sockets. |
||
154 | * Each datagram is sent and a reply read consequently. |
||
155 | * The next datagram is sent after the reply is received. |
||
156 | * @param verbose A value indicating whether to print out verbose information. Input parameter. |
||
157 | * @param socket_ids A field of stored socket identifiers. Input parameter. |
||
158 | * @param sockets The number of sockets in the field. Should be at most the size of the field. Input parameter. |
||
159 | * @param address The destination host address to send data to. The source host address of received datagrams is set instead. Input/output parameter. |
||
160 | * @param addrlen The length of the destination address in bytes. Input parameter. |
||
161 | * @param data The data to be sent. The received data are set instead. Input/output parameter. |
||
162 | * @param size The data size in bytes. Input parameter |
||
163 | * @param messages The number of datagrams per socket to be received. Input parameter. |
||
164 | * @returns EOK on success. |
||
165 | * @returns Other error codes as defined for the recvfrom() function. |
||
166 | */ |
||
4740 | mejdrech | 167 | int sockets_sendto_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ); |
4749 | mejdrech | 168 | |
169 | /** Prints a mark. |
||
170 | * If the index is a multiple of ten, a different mark is printed. |
||
171 | * @param index The index of the mark to be printed. Input parameter. |
||
172 | */ |
||
4740 | mejdrech | 173 | void print_mark( int index ); |
174 | |||
175 | void print_help( void ){ |
||
176 | printf( |
||
4749 | mejdrech | 177 | "Network Networking test 1 aplication - sockets\n" \ |
4740 | mejdrech | 178 | "Usage: echo [options] numeric_address\n" \ |
179 | "Where options are:\n" \ |
||
180 | "-f protocol_family | --family=protocol_family\n" \ |
||
181 | "\tThe listenning socket protocol family. Only the PF_INET and PF_INET6 are supported.\n" |
||
182 | "\n" \ |
||
183 | "-h | --help\n" \ |
||
184 | "\tShow this application help.\n" |
||
185 | "\n" \ |
||
186 | "-m count | --messages=count\n" \ |
||
4749 | mejdrech | 187 | "\tThe number of messages to send and receive per socket. The default is 10.\n" \ |
4740 | mejdrech | 188 | "\n" \ |
189 | "-n sockets | --sockets=count\n" \ |
||
4749 | mejdrech | 190 | "\tThe number of sockets to use. The default is 10.\n" \ |
4740 | mejdrech | 191 | "\n" \ |
192 | "-p port_number | --port=port_number\n" \ |
||
193 | "\tThe port number the application should send messages to. The default is 7.\n" \ |
||
194 | "\n" \ |
||
195 | "-s packet_size | --size=packet_size\n" \ |
||
4749 | mejdrech | 196 | "\tThe packet data size the application sends. The default is 28 bytes.\n" \ |
4740 | mejdrech | 197 | "\n" \ |
198 | "-v | --verbose\n" \ |
||
199 | "\tShow all output messages.\n" |
||
200 | ); |
||
201 | } |
||
202 | |||
203 | int parse_protocol_family( const char * name ){ |
||
204 | if( str_lcmp( name, "PF_INET", 7 ) == 0 ){ |
||
205 | return PF_INET; |
||
206 | }else if( str_lcmp( name, "PF_INET6", 8 ) == 0 ){ |
||
207 | return PF_INET6; |
||
208 | } |
||
209 | return EPFNOSUPPORT; |
||
210 | } |
||
211 | |||
212 | int parse_socket_type( const char * name ){ |
||
213 | if( str_lcmp( name, "SOCK_DGRAM", 11 ) == 0 ){ |
||
214 | return SOCK_DGRAM; |
||
215 | }else if( str_lcmp( name, "SOCK_STREAM", 12 ) == 0 ){ |
||
216 | return SOCK_STREAM; |
||
217 | } |
||
218 | return ESOCKTNOSUPPORT; |
||
219 | } |
||
220 | |||
4743 | mejdrech | 221 | void refresh_data( char * data, size_t size ){ |
222 | size_t length; |
||
4740 | mejdrech | 223 | |
224 | // fill the data |
||
225 | length = 0; |
||
4749 | mejdrech | 226 | while( size > length + sizeof( NETTEST1_TEXT ) - 1 ){ |
227 | memcpy( data + length, NETTEST1_TEXT, sizeof( NETTEST1_TEXT ) - 1 ); |
||
228 | length += sizeof( NETTEST1_TEXT ) - 1; |
||
4740 | mejdrech | 229 | } |
230 | memcpy( data + length, NETTEST1_TEXT, size - length ); |
||
4749 | mejdrech | 231 | data[ size ] = '\0'; |
4740 | mejdrech | 232 | } |
233 | |||
234 | int sockets_create( int verbose, int * socket_ids, int sockets, int family, sock_type_t type ){ |
||
235 | int index; |
||
236 | |||
237 | if( verbose ) printf( "Create\t" ); |
||
238 | fflush( stdout ); |
||
239 | for( index = 0; index < sockets; ++ index ){ |
||
240 | socket_ids[ index ] = socket( family, type, 0 ); |
||
241 | if( socket_ids[ index ] < 0 ){ |
||
242 | printf( "Socket %d (%d) error:\n", index, socket_ids[ index ] ); |
||
243 | socket_print_error( stderr, socket_ids[ index ], "Socket create: ", "\n" ); |
||
244 | return socket_ids[ index ]; |
||
245 | } |
||
246 | if( verbose ) print_mark( index ); |
||
247 | } |
||
248 | return EOK; |
||
249 | } |
||
250 | |||
251 | int sockets_close( int verbose, int * socket_ids, int sockets ){ |
||
252 | ERROR_DECLARE; |
||
253 | |||
254 | int index; |
||
255 | |||
256 | if( verbose ) printf( "\tClose\t" ); |
||
257 | fflush( stdout ); |
||
258 | for( index = 0; index < sockets; ++ index ){ |
||
259 | if( ERROR_OCCURRED( closesocket( socket_ids[ index ] ))){ |
||
260 | printf( "Socket %d (%d) error:\n", index, socket_ids[ index ] ); |
||
261 | socket_print_error( stderr, ERROR_CODE, "Socket close: ", "\n" ); |
||
262 | return ERROR_CODE; |
||
263 | } |
||
264 | if( verbose ) print_mark( index ); |
||
265 | } |
||
266 | return EOK; |
||
267 | } |
||
268 | |||
4743 | mejdrech | 269 | int sockets_connect( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen ){ |
4740 | mejdrech | 270 | ERROR_DECLARE; |
271 | |||
272 | int index; |
||
273 | |||
4743 | mejdrech | 274 | if( verbose ) printf( "\tConnect\t" ); |
4740 | mejdrech | 275 | fflush( stdout ); |
276 | for( index = 0; index < sockets; ++ index ){ |
||
4743 | mejdrech | 277 | if( ERROR_OCCURRED( connect( socket_ids[ index ], address, addrlen ))){ |
278 | socket_print_error( stderr, ERROR_CODE, "Socket connect: ", "\n" ); |
||
4740 | mejdrech | 279 | return ERROR_CODE; |
280 | } |
||
281 | if( verbose ) print_mark( index ); |
||
282 | } |
||
283 | return EOK; |
||
284 | } |
||
285 | |||
286 | int sockets_sendto( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t addrlen, char * data, int size, int messages ){ |
||
287 | ERROR_DECLARE; |
||
288 | |||
289 | int index; |
||
290 | int message; |
||
291 | |||
292 | if( verbose ) printf( "\tSendto\t" ); |
||
293 | fflush( stdout ); |
||
294 | for( index = 0; index < sockets; ++ index ){ |
||
295 | for( message = 0; message < messages; ++ message ){ |
||
296 | if( ERROR_OCCURRED( sendto( socket_ids[ index ], data, size, 0, address, addrlen ))){ |
||
297 | printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
||
298 | socket_print_error( stderr, ERROR_CODE, "Socket send: ", "\n" ); |
||
299 | return ERROR_CODE; |
||
300 | } |
||
301 | } |
||
302 | if( verbose ) print_mark( index ); |
||
303 | } |
||
304 | return EOK; |
||
305 | } |
||
306 | |||
307 | int sockets_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ){ |
||
308 | int value; |
||
309 | int index; |
||
310 | int message; |
||
311 | |||
312 | if( verbose ) printf( "\tRecvfrom\t" ); |
||
313 | fflush( stdout ); |
||
314 | for( index = 0; index < sockets; ++ index ){ |
||
315 | for( message = 0; message < messages; ++ message ){ |
||
316 | value = recvfrom( socket_ids[ index ], data, size, 0, address, addrlen ); |
||
317 | if( value < 0 ){ |
||
318 | printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
||
319 | socket_print_error( stderr, value, "Socket receive: ", "\n" ); |
||
320 | return value; |
||
321 | } |
||
322 | } |
||
323 | if( verbose ) print_mark( index ); |
||
324 | } |
||
325 | return EOK; |
||
326 | } |
||
327 | |||
328 | int sockets_sendto_recvfrom( int verbose, int * socket_ids, int sockets, struct sockaddr * address, socklen_t * addrlen, char * data, int size, int messages ){ |
||
329 | ERROR_DECLARE; |
||
330 | |||
331 | int value; |
||
332 | int index; |
||
333 | int message; |
||
334 | |||
335 | if( verbose ) printf( "\tSendto and recvfrom\t" ); |
||
336 | fflush( stdout ); |
||
337 | for( index = 0; index < sockets; ++ index ){ |
||
338 | for( message = 0; message < messages; ++ message ){ |
||
339 | if( ERROR_OCCURRED( sendto( socket_ids[ index ], data, size, 0, address, * addrlen ))){ |
||
340 | printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
||
341 | socket_print_error( stderr, ERROR_CODE, "Socket send: ", "\n" ); |
||
342 | return ERROR_CODE; |
||
343 | } |
||
344 | value = recvfrom( socket_ids[ index ], data, size, 0, address, addrlen ); |
||
345 | if( value < 0 ){ |
||
346 | printf( "Socket %d (%d), message %d error:\n", index, socket_ids[ index ], message ); |
||
347 | socket_print_error( stderr, value, "Socket receive: ", "\n" ); |
||
348 | return value; |
||
349 | } |
||
350 | } |
||
351 | if( verbose ) print_mark( index ); |
||
352 | } |
||
353 | return EOK; |
||
354 | } |
||
355 | |||
356 | void print_mark( int index ){ |
||
357 | if(( index + 1 ) % 10 ){ |
||
358 | printf( "*" ); |
||
359 | }else{ |
||
360 | printf( "|" ); |
||
361 | } |
||
362 | fflush( stdout ); |
||
363 | } |
||
364 | |||
365 | int main( int argc, char * argv[] ){ |
||
366 | ERROR_DECLARE; |
||
367 | |||
4749 | mejdrech | 368 | size_t size = 28; |
4740 | mejdrech | 369 | int verbose = 0; |
370 | sock_type_t type = SOCK_DGRAM; |
||
371 | int sockets = 10; |
||
372 | int messages = 10; |
||
373 | int family = PF_INET; |
||
374 | uint16_t port = 7; |
||
375 | |||
376 | socklen_t max_length = sizeof( struct sockaddr_in6 ); |
||
377 | uint8_t address_data[ max_length ]; |
||
378 | struct sockaddr * address = ( struct sockaddr * ) address_data; |
||
379 | struct sockaddr_in * address_in = ( struct sockaddr_in * ) address; |
||
380 | struct sockaddr_in6 * address_in6 = ( struct sockaddr_in6 * ) address; |
||
381 | socklen_t addrlen; |
||
382 | // char address_string[ INET6_ADDRSTRLEN ]; |
||
383 | uint8_t * address_start; |
||
384 | |||
385 | int * socket_ids; |
||
386 | char * data; |
||
387 | int value; |
||
388 | int index; |
||
4743 | mejdrech | 389 | struct timeval time_before; |
390 | struct timeval time_after; |
||
4740 | mejdrech | 391 | |
392 | printf( "Task %d - ", task_get_id()); |
||
393 | printf( "%s\n", NAME ); |
||
394 | |||
395 | if( argc <= 1 ){ |
||
396 | print_help(); |
||
397 | return EINVAL; |
||
398 | } |
||
399 | |||
400 | for( index = 1; ( index < argc - 1 ) || (( index == argc ) && ( argv[ index ][ 0 ] == '-' )); ++ index ){ |
||
401 | if( argv[ index ][ 0 ] == '-' ){ |
||
402 | switch( argv[ index ][ 1 ] ){ |
||
403 | case 'f': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & family, "protocol family", 0, parse_protocol_family )); |
||
404 | break; |
||
405 | case 'h': print_help(); |
||
406 | return EOK; |
||
407 | break; |
||
408 | case 'm': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & messages, "message count", 0 )); |
||
409 | break; |
||
410 | case 'n': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & sockets, "socket count", 0 )); |
||
411 | break; |
||
412 | case 'p': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 0 )); |
||
413 | port = ( uint16_t ) value; |
||
414 | break; |
||
415 | case 's': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "packet size", 0 )); |
||
416 | size = (value >= 0 ) ? ( size_t ) value : 0; |
||
417 | break; |
||
418 | case 't': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "socket type", 0, parse_socket_type )); |
||
419 | type = ( sock_type_t ) value; |
||
420 | break; |
||
421 | case 'v': verbose = 1; |
||
422 | break; |
||
423 | case '-': if( str_lcmp( argv[ index ] + 2, "family=", 7 ) == 0 ){ |
||
424 | ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & family, "protocol family", 9, parse_protocol_family )); |
||
425 | }else if( str_lcmp( argv[ index ] + 2, "help", 5 ) == 0 ){ |
||
426 | print_help(); |
||
427 | return EOK; |
||
428 | }else if( str_lcmp( argv[ index ] + 2, "messages=", 6 ) == 0 ){ |
||
429 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & messages, "message count", 8 )); |
||
430 | }else if( str_lcmp( argv[ index ] + 2, "sockets=", 6 ) == 0 ){ |
||
431 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & sockets, "socket count", 8 )); |
||
432 | }else if( str_lcmp( argv[ index ] + 2, "port=", 5 ) == 0 ){ |
||
433 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 7 )); |
||
434 | port = ( uint16_t ) value; |
||
435 | }else if( str_lcmp( argv[ index ] + 2, "type=", 5 ) == 0 ){ |
||
436 | ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "socket type", 7, parse_socket_type )); |
||
437 | type = ( sock_type_t ) value; |
||
438 | }else if( str_lcmp( argv[ index ] + 2, "verbose", 8 ) == 0 ){ |
||
439 | verbose = 1; |
||
440 | }else{ |
||
441 | print_unrecognized( index, argv[ index ] + 2 ); |
||
442 | print_help(); |
||
443 | return EINVAL; |
||
444 | } |
||
445 | break; |
||
446 | default: |
||
447 | print_unrecognized( index, argv[ index ] + 1 ); |
||
448 | print_help(); |
||
449 | return EINVAL; |
||
450 | } |
||
451 | }else{ |
||
452 | print_unrecognized( index, argv[ index ] ); |
||
453 | print_help(); |
||
454 | return EINVAL; |
||
455 | } |
||
456 | } |
||
457 | |||
458 | bzero( address_data, max_length ); |
||
459 | switch( family ){ |
||
460 | case PF_INET: |
||
461 | address_in->sin_family = AF_INET; |
||
462 | address_in->sin_port = htons( port ); |
||
463 | address_start = ( uint8_t * ) & address_in->sin_addr.s_addr; |
||
464 | addrlen = sizeof( struct sockaddr_in ); |
||
465 | break; |
||
466 | case PF_INET6: |
||
467 | address_in6->sin6_family = AF_INET6; |
||
468 | address_in6->sin6_port = htons( port ); |
||
469 | address_start = ( uint8_t * ) & address_in6->sin6_addr.s6_addr; |
||
470 | addrlen = sizeof( struct sockaddr_in6 ); |
||
471 | break; |
||
472 | default: |
||
473 | fprintf( stderr, "Address family is not supported\n" ); |
||
474 | return EAFNOSUPPORT; |
||
475 | } |
||
476 | |||
477 | if( ERROR_OCCURRED( inet_pton( family, argv[ argc - 1 ], address_start ))){ |
||
478 | fprintf( stderr, "Address parse error %d\n", ERROR_CODE ); |
||
479 | return ERROR_CODE; |
||
480 | } |
||
481 | |||
482 | if( size <= 0 ){ |
||
483 | fprintf( stderr, "Data buffer size too small (%d). Using 1024 bytes instead.\n", size ); |
||
484 | size = 1024; |
||
485 | } |
||
486 | // size plus terminating null (\0) |
||
487 | data = ( char * ) malloc( size + 1 ); |
||
488 | if( ! data ){ |
||
489 | fprintf( stderr, "Failed to allocate data buffer.\n" ); |
||
490 | return ENOMEM; |
||
491 | } |
||
492 | refresh_data( data, size ); |
||
493 | |||
494 | if( sockets <= 0 ){ |
||
495 | fprintf( stderr, "Socket count too small (%d). Using 2 instead.\n", sockets ); |
||
496 | sockets = 2; |
||
497 | } |
||
498 | // count plus terminating null (\0) |
||
499 | socket_ids = ( int * ) malloc( sizeof( int ) * ( sockets + 1 )); |
||
500 | if( ! socket_ids ){ |
||
501 | fprintf( stderr, "Failed to allocate receive buffer.\n" ); |
||
502 | return ENOMEM; |
||
503 | } |
||
504 | socket_ids[ sockets ] = NULL; |
||
505 | |||
506 | if( verbose ) printf( "Starting tests\n" ); |
||
507 | |||
508 | if( verbose ) printf( "1 socket, 1 message\n" ); |
||
509 | |||
4743 | mejdrech | 510 | if( ERROR_OCCURRED( gettimeofday( & time_before, NULL ))){ |
511 | fprintf( stderr, "Get time of day error %d\n", ERROR_CODE ); |
||
512 | return ERROR_CODE; |
||
513 | } |
||
514 | |||
4740 | mejdrech | 515 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, 1, family, type )); |
516 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, 1 )); |
||
517 | if( verbose ) printf( "\tOK\n" ); |
||
518 | |||
519 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, 1, family, type )); |
||
4743 | mejdrech | 520 | if( type == SOCK_STREAM ){ |
521 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, 1, address, addrlen )); |
||
4740 | mejdrech | 522 | } |
4743 | mejdrech | 523 | ERROR_PROPAGATE( sockets_sendto_recvfrom( verbose, socket_ids, 1, address, & addrlen, data, size, 1 )); |
4740 | mejdrech | 524 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, 1 )); |
525 | if( verbose ) printf( "\tOK\n" ); |
||
526 | |||
527 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, 1, family, type )); |
||
4743 | mejdrech | 528 | if( type == SOCK_STREAM ){ |
529 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, 1, address, addrlen )); |
||
4740 | mejdrech | 530 | } |
4743 | mejdrech | 531 | ERROR_PROPAGATE( sockets_sendto( verbose, socket_ids, 1, address, addrlen, data, size, 1 )); |
4740 | mejdrech | 532 | ERROR_PROPAGATE( sockets_recvfrom( verbose, socket_ids, 1, address, & addrlen, data, size, 1 )); |
533 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, 1 )); |
||
534 | if( verbose ) printf( "\tOK\n" ); |
||
535 | |||
536 | if( verbose ) printf( "1 socket, %d messages\n", messages ); |
||
537 | |||
538 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, 1, family, type )); |
||
4743 | mejdrech | 539 | if( type == SOCK_STREAM ){ |
540 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, 1, address, addrlen )); |
||
4740 | mejdrech | 541 | } |
4743 | mejdrech | 542 | ERROR_PROPAGATE( sockets_sendto_recvfrom( verbose, socket_ids, 1, address, & addrlen, data, size, messages )); |
4740 | mejdrech | 543 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, 1 )); |
544 | if( verbose ) printf( "\tOK\n" ); |
||
545 | |||
546 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, 1, family, type )); |
||
4743 | mejdrech | 547 | if( type == SOCK_STREAM ){ |
548 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, 1, address, addrlen )); |
||
4740 | mejdrech | 549 | } |
4743 | mejdrech | 550 | ERROR_PROPAGATE( sockets_sendto( verbose, socket_ids, 1, address, addrlen, data, size, messages )); |
4740 | mejdrech | 551 | ERROR_PROPAGATE( sockets_recvfrom( verbose, socket_ids, 1, address, & addrlen, data, size, messages )); |
552 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, 1 )); |
||
553 | if( verbose ) printf( "\tOK\n" ); |
||
554 | |||
555 | if( verbose ) printf( "%d sockets, 1 message\n", sockets ); |
||
556 | |||
557 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, sockets, family, type )); |
||
558 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, sockets )); |
||
559 | if( verbose ) printf( "\tOK\n" ); |
||
560 | |||
561 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, sockets, family, type )); |
||
4743 | mejdrech | 562 | if( type == SOCK_STREAM ){ |
563 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, sockets, address, addrlen )); |
||
4740 | mejdrech | 564 | } |
4743 | mejdrech | 565 | ERROR_PROPAGATE( sockets_sendto_recvfrom( verbose, socket_ids, sockets, address, & addrlen, data, size, 1 )); |
4740 | mejdrech | 566 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, sockets )); |
567 | if( verbose ) printf( "\tOK\n" ); |
||
568 | |||
569 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, sockets, family, type )); |
||
4743 | mejdrech | 570 | if( type == SOCK_STREAM ){ |
571 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, sockets, address, addrlen )); |
||
4740 | mejdrech | 572 | } |
4743 | mejdrech | 573 | ERROR_PROPAGATE( sockets_sendto( verbose, socket_ids, sockets, address, addrlen, data, size, 1 )); |
4740 | mejdrech | 574 | ERROR_PROPAGATE( sockets_recvfrom( verbose, socket_ids, sockets, address, & addrlen, data, size, 1 )); |
575 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, sockets )); |
||
576 | if( verbose ) printf( "\tOK\n" ); |
||
577 | |||
578 | if( verbose ) printf( "%d sockets, %d messages\n", sockets, messages ); |
||
579 | |||
580 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, sockets, family, type )); |
||
4743 | mejdrech | 581 | if( type == SOCK_STREAM ){ |
582 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, sockets, address, addrlen )); |
||
4740 | mejdrech | 583 | } |
4743 | mejdrech | 584 | ERROR_PROPAGATE( sockets_sendto_recvfrom( verbose, socket_ids, sockets, address, & addrlen, data, size, messages )); |
4740 | mejdrech | 585 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, sockets )); |
586 | if( verbose ) printf( "\tOK\n" ); |
||
587 | |||
588 | ERROR_PROPAGATE( sockets_create( verbose, socket_ids, sockets, family, type )); |
||
4743 | mejdrech | 589 | if( type == SOCK_STREAM ){ |
590 | ERROR_PROPAGATE( sockets_connect( verbose, socket_ids, sockets, address, addrlen )); |
||
4740 | mejdrech | 591 | } |
4743 | mejdrech | 592 | ERROR_PROPAGATE( sockets_sendto( verbose, socket_ids, sockets, address, addrlen, data, size, messages )); |
4740 | mejdrech | 593 | ERROR_PROPAGATE( sockets_recvfrom( verbose, socket_ids, sockets, address, & addrlen, data, size, messages )); |
594 | ERROR_PROPAGATE( sockets_close( verbose, socket_ids, sockets )); |
||
4743 | mejdrech | 595 | |
596 | if( ERROR_OCCURRED( gettimeofday( & time_after, NULL ))){ |
||
597 | fprintf( stderr, "Get time of day error %d\n", ERROR_CODE ); |
||
598 | return ERROR_CODE; |
||
4740 | mejdrech | 599 | } |
600 | |||
4743 | mejdrech | 601 | if( verbose ) printf( "\tOK\n" ); |
4740 | mejdrech | 602 | |
4743 | mejdrech | 603 | printf( "Tested in %d microseconds\n", tv_sub( & time_after, & time_before )); |
604 | |||
4740 | mejdrech | 605 | if( verbose ) printf( "Exiting\n" ); |
606 | |||
607 | return EOK; |
||
608 | } |
||
609 | |||
610 | /** @} |
||
611 | */ |