source: mainline/uspace/app/nettest1/nettest1.c@ 88b127b

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

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