source: mainline/uspace/app/nettest2/nettest2.c@ 3090715

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 3090715 was 69e0d6d, checked in by Jiri Svoboda <jiri@…>, 15 years ago

Slightly clean up nettest1.
  • Property mode set to 100644
File size: 10.5 KB
Line 
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
29/** @addtogroup nettest
30 * @{
31 */
32
33/** @file
34 * Networking test 2 application - transfer.
35 */
36
37#include "nettest.h"
38#include "print_error.h"
39
40#include <malloc.h>
41#include <stdio.h>
42#include <str.h>
43#include <task.h>
44#include <time.h>
45#include <arg_parse.h>
46#include <bool.h>
47
48#include <net/in.h>
49#include <net/in6.h>
50#include <net/inet.h>
51#include <net/socket.h>
52#include <net/socket_parse.h>
53
54/** Echo module name. */
55#define NAME "Nettest2"
56
57/** Packet data pattern. */
58#define NETTEST2_TEXT "Networking test 2 - transfer"
59
60static size_t size;
61static bool verbose;
62static sock_type_t type;
63static int sockets;
64static int messages;
65static int family;
66static uint16_t port;
67
68static void nettest2_print_help(void)
69{
70 printf(
71 "Network Networking test 2 aplication - UDP transfer\n"
72 "Usage: echo [options] address\n"
73 "Where options are:\n"
74 "-f protocol_family | --family=protocol_family\n"
75 "\tThe listenning socket protocol family. Only the PF_INET and "
76 "PF_INET6 are supported.\n"
77 "\n"
78 "-h | --help\n"
79 "\tShow this application help.\n"
80 "\n"
81 "-m count | --messages=count\n"
82 "\tThe number of messages to send and receive per socket. The "
83 "default is 10.\n"
84 "\n"
85 "-n sockets | --sockets=count\n"
86 "\tThe number of sockets to use. The default is 10.\n"
87 "\n"
88 "-p port_number | --port=port_number\n"
89 "\tThe port number the application should send messages to. The "
90 "default is 7.\n"
91 "\n"
92 "-s packet_size | --size=packet_size\n"
93 "\tThe packet data size the application sends. The default is 29 "
94 "bytes.\n"
95 "\n"
96 "-v | --verbose\n"
97 "\tShow all output messages.\n");
98}
99
100/** Fill buffer with the NETTEST2_TEXT pattern.
101 *
102 * @param buffer Data buffer.
103 * @param size Buffer size in bytes.
104 */
105static void nettest2_fill_buffer(char *buffer, size_t size)
106{
107 size_t length;
108
109 length = 0;
110 while (size > length + sizeof(NETTEST2_TEXT) - 1) {
111 memcpy(buffer + length, NETTEST2_TEXT,
112 sizeof(NETTEST2_TEXT) - 1);
113 length += sizeof(NETTEST2_TEXT) - 1;
114 }
115
116 memcpy(buffer + length, NETTEST2_TEXT, size - length);
117 buffer[size] = '\0';
118}
119
120/** Parse one command-line option.
121 *
122 * @param argc Number of all command-line arguments.
123 * @param argv All command-line arguments.
124 * @param index Current argument index (in, out).
125 */
126static int nettest2_parse_opt(int argc, char *argv[], int *index)
127{
128 int value;
129 int rc;
130
131 switch (argv[*index][1]) {
132 /*
133 * Short options with only one letter
134 */
135 case 'f':
136 rc = arg_parse_name_int(argc, argv, index, &family, 0,
137 socket_parse_protocol_family);
138 if (rc != EOK)
139 return rc;
140 break;
141 case 'h':
142 nettest2_print_help();
143 return EOK;
144 break;
145 case 'm':
146 rc = arg_parse_int(argc, argv, index, &messages, 0);
147 if (rc != EOK)
148 return rc;
149 break;
150 case 'n':
151 rc = arg_parse_int(argc, argv, index, &sockets, 0);
152 if (rc != EOK)
153 return rc;
154 break;
155 case 'p':
156 rc = arg_parse_int(argc, argv, index, &value, 0);
157 if (rc != EOK)
158 return rc;
159 port = (uint16_t) value;
160 break;
161 case 's':
162 rc = arg_parse_int(argc, argv, index, &value, 0);
163 if (rc != EOK)
164 return rc;
165 size = (value >= 0) ? (size_t) value : 0;
166 break;
167 case 't':
168 rc = arg_parse_name_int(argc, argv, index, &value, 0,
169 socket_parse_socket_type);
170 if (rc != EOK)
171 return rc;
172 type = (sock_type_t) value;
173 break;
174 case 'v':
175 verbose = true;
176 break;
177 /*
178 * Long options with double dash ('-')
179 */
180 case '-':
181 if (str_lcmp(argv[*index] + 2, "family=", 7) == 0) {
182 rc = arg_parse_name_int(argc, argv, index, &family, 9,
183 socket_parse_protocol_family);
184 if (rc != EOK)
185 return rc;
186 } else if (str_lcmp(argv[*index] + 2, "help", 5) == 0) {
187 nettest2_print_help();
188 return EOK;
189 } else if (str_lcmp(argv[*index] + 2, "messages=", 6) == 0) {
190 rc = arg_parse_int(argc, argv, index, &messages, 8);
191 if (rc != EOK)
192 return rc;
193 } else if (str_lcmp(argv[*index] + 2, "sockets=", 6) == 0) {
194 rc = arg_parse_int(argc, argv, index, &sockets, 8);
195 if (rc != EOK)
196 return rc;
197 } else if (str_lcmp(argv[*index] + 2, "port=", 5) == 0) {
198 rc = arg_parse_int(argc, argv, index, &value, 7);
199 if (rc != EOK)
200 return rc;
201 port = (uint16_t) value;
202 } else if (str_lcmp(argv[*index] + 2, "type=", 5) == 0) {
203 rc = arg_parse_name_int(argc, argv, index, &value, 7,
204 socket_parse_socket_type);
205 if (rc != EOK)
206 return rc;
207 type = (sock_type_t) value;
208 } else if (str_lcmp(argv[*index] + 2, "verbose", 8) == 0) {
209 verbose = 1;
210 } else {
211 nettest2_print_help();
212 return EINVAL;
213 }
214 break;
215 default:
216 nettest2_print_help();
217 return EINVAL;
218 }
219
220 return EOK;
221}
222
223int main(int argc, char *argv[])
224{
225 socklen_t max_length;
226 uint8_t address_data[sizeof(struct sockaddr_in6)];
227 struct sockaddr *address;
228 struct sockaddr_in *address_in;
229 struct sockaddr_in6 *address_in6;
230 socklen_t addrlen;
231 uint8_t *address_start;
232
233 int *socket_ids;
234 char *data;
235 int index;
236 struct timeval time_before;
237 struct timeval time_after;
238
239 int rc;
240
241 size = 28;
242 verbose = false;
243 type = SOCK_DGRAM;
244 sockets = 10;
245 messages = 10;
246 family = PF_INET;
247 port = 7;
248
249 max_length = sizeof(address_data);
250 address = (struct sockaddr *) address_data;
251 address_in = (struct sockaddr_in *) address;
252 address_in6 = (struct sockaddr_in6 *) address;
253
254 /*
255 * Parse the command line arguments.
256 *
257 * Stop before the last argument if it does not start with dash ('-')
258 */
259 for (index = 1; (index < argc - 1) || ((index == argc - 1) &&
260 (argv[index][0] == '-')); ++index) {
261
262 /* Options should start with dash ('-') */
263 if (argv[index][0] == '-') {
264 rc = nettest2_parse_opt(argc, argv, &index);
265 if (rc != EOK)
266 return rc;
267 } else {
268 nettest2_print_help();
269 return EINVAL;
270 }
271 }
272
273 /* If not before the last argument containing the address */
274 if (index >= argc) {
275 printf("Command line error: missing address\n");
276 nettest2_print_help();
277 return EINVAL;
278 }
279
280 /* Prepare the address buffer */
281 bzero(address_data, max_length);
282
283 switch (family) {
284 case PF_INET:
285 address_in->sin_family = AF_INET;
286 address_in->sin_port = htons(port);
287 address_start = (uint8_t *) &address_in->sin_addr.s_addr;
288 addrlen = sizeof(struct sockaddr_in);
289 break;
290 case PF_INET6:
291 address_in6->sin6_family = AF_INET6;
292 address_in6->sin6_port = htons(port);
293 address_start = (uint8_t *) &address_in6->sin6_addr.s6_addr;
294 addrlen = sizeof(struct sockaddr_in6);
295 break;
296 default:
297 fprintf(stderr, "Address family is not supported\n");
298 return EAFNOSUPPORT;
299 }
300
301 /* Parse the last argument which should contain the address. */
302 rc = inet_pton(family, argv[argc - 1], address_start);
303 if (rc != EOK) {
304 fprintf(stderr, "Address parse error %d\n", rc);
305 return rc;
306 }
307
308 /* Check data buffer size. */
309 if (size <= 0) {
310 fprintf(stderr, "Data buffer size too small (%d). Using 1024 "
311 "bytes instead.\n", size);
312 size = 1024;
313 }
314
315 /*
316 * Prepare the buffer. Allocate size bytes plus one for terminating
317 * null character.
318 */
319 data = (char *) malloc(size + 1);
320 if (!data) {
321 fprintf(stderr, "Failed to allocate data buffer.\n");
322 return ENOMEM;
323 }
324
325 /* Fill buffer with a pattern. */
326 nettest2_fill_buffer(data, size);
327
328 /* Check socket count. */
329 if (sockets <= 0) {
330 fprintf(stderr, "Socket count too small (%d). Using "
331 "2 instead.\n", sockets);
332 sockets = 2;
333 }
334
335 /*
336 * Prepare the socket buffer.
337 * Allocate count entries plus the terminating null (\0)
338 */
339 socket_ids = (int *) malloc(sizeof(int) * (sockets + 1));
340 if (!socket_ids) {
341 fprintf(stderr, "Failed to allocate receive buffer.\n");
342 return ENOMEM;
343 }
344 socket_ids[sockets] = NULL;
345
346 if (verbose)
347 printf("Starting tests\n");
348
349 rc = sockets_create(verbose, socket_ids, sockets, family, type);
350 if (rc != EOK)
351 return rc;
352
353 if (type == SOCK_STREAM) {
354 rc = sockets_connect(verbose, socket_ids, sockets,
355 address, addrlen);
356 if (rc != EOK)
357 return rc;
358 }
359
360 if (verbose)
361 printf("\n");
362
363 rc = gettimeofday(&time_before, NULL);
364 if (rc != EOK) {
365 fprintf(stderr, "Get time of day error %d\n", rc);
366 return rc;
367 }
368
369 rc = sockets_sendto_recvfrom(verbose, socket_ids, sockets, address,
370 &addrlen, data, size, messages);
371 if (rc != EOK)
372 return rc;
373
374 rc = gettimeofday(&time_after, NULL);
375 if (rc != EOK) {
376 fprintf(stderr, "Get time of day error %d\n", rc);
377 return rc;
378 }
379
380 if (verbose)
381 printf("\tOK\n");
382
383 printf("sendto + recvfrom tested in %d microseconds\n",
384 tv_sub(&time_after, &time_before));
385
386 rc = gettimeofday(&time_before, NULL);
387 if (rc != EOK) {
388 fprintf(stderr, "Get time of day error %d\n", rc);
389 return rc;
390 }
391
392 rc = sockets_sendto(verbose, socket_ids, sockets, address, addrlen,
393 data, size, messages);
394 if (rc != EOK)
395 return rc;
396
397 rc = sockets_recvfrom(verbose, socket_ids, sockets, address, &addrlen,
398 data, size, messages);
399 if (rc != EOK)
400 return rc;
401
402 rc = gettimeofday(&time_after, NULL);
403 if (rc != EOK) {
404 fprintf(stderr, "Get time of day error %d\n", rc);
405 return rc;
406 }
407
408 if (verbose)
409 printf("\tOK\n");
410
411 printf("sendto, recvfrom tested in %d microseconds\n",
412 tv_sub(&time_after, &time_before));
413
414 rc = sockets_close(verbose, socket_ids, sockets);
415 if (rc != EOK)
416 return rc;
417
418 if (verbose)
419 printf("\nExiting\n");
420
421 return EOK;
422}
423
424/** @}
425 */
Note: See TracBrowser for help on using the repository browser.