Rev 4704 | Rev 4709 | Go to most recent revision | Only display areas with differences | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed
Rev 4704 | Rev 4708 | ||
---|---|---|---|
1 | /* |
1 | /* |
2 | * Copyright (c) 2009 Lukas Mejdrech |
2 | * Copyright (c) 2009 Lukas Mejdrech |
3 | * All rights reserved. |
3 | * All rights reserved. |
4 | * |
4 | * |
5 | * Redistribution and use in source and binary forms, with or without |
5 | * Redistribution and use in source and binary forms, with or without |
6 | * modification, are permitted provided that the following conditions |
6 | * modification, are permitted provided that the following conditions |
7 | * are met: |
7 | * are met: |
8 | * |
8 | * |
9 | * - Redistributions of source code must retain the above copyright |
9 | * - Redistributions of source code must retain the above copyright |
10 | * notice, this list of conditions and the following disclaimer. |
10 | * notice, this list of conditions and the following disclaimer. |
11 | * - Redistributions in binary form must reproduce the above copyright |
11 | * - Redistributions in binary form must reproduce the above copyright |
12 | * notice, this list of conditions and the following disclaimer in the |
12 | * notice, this list of conditions and the following disclaimer in the |
13 | * documentation and/or other materials provided with the distribution. |
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 |
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. |
15 | * derived from this software without specific prior written permission. |
16 | * |
16 | * |
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
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 |
18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
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 |
23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
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 |
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. |
26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
27 | */ |
27 | */ |
28 | 28 | ||
29 | /** @addtogroup echo |
29 | /** @addtogroup echo |
30 | * @{ |
30 | * @{ |
31 | */ |
31 | */ |
32 | 32 | ||
33 | /** @file |
33 | /** @file |
34 | * Echo application. |
34 | * Echo application. |
35 | * Answers received packets. |
35 | * Answers received packets. |
36 | */ |
36 | */ |
37 | 37 | ||
38 | #include <malloc.h> |
38 | #include <malloc.h> |
39 | #include <stdio.h> |
39 | #include <stdio.h> |
40 | #include <string.h> |
40 | #include <string.h> |
41 | #include <task.h> |
41 | #include <task.h> |
42 | 42 | ||
43 | #include "../../include/in.h" |
43 | #include "../../include/in.h" |
44 | #include "../../include/inet.h" |
44 | #include "../../include/inet.h" |
45 | #include "../../include/socket.h" |
45 | #include "../../include/socket.h" |
46 | 46 | ||
47 | #include "../../err.h" |
47 | #include "../../err.h" |
48 | 48 | ||
49 | /** Echo module name. |
49 | /** Echo module name. |
50 | */ |
50 | */ |
51 | #define NAME "Echo" |
51 | #define NAME "Echo" |
52 | 52 | ||
53 | /** Module entry point. |
53 | /** Module entry point. |
54 | * Reads command line parameters and starts listenning. |
54 | * Reads command line parameters and starts listenning. |
55 | * @param argc The number of command line parameters. Input parameter. |
55 | * @param argc The number of command line parameters. Input parameter. |
56 | * @param argv The command line parameters. Input parameter. |
56 | * @param argv The command line parameters. Input parameter. |
57 | * @returns EOK on success. |
57 | * @returns EOK on success. |
58 | */ |
58 | */ |
59 | int main( int argc, char * argv[] ); |
59 | int main( int argc, char * argv[] ); |
60 | 60 | ||
61 | /** @name Output printing functions |
61 | /** @name Output printing functions |
62 | */ |
62 | */ |
63 | /*@{*/ |
63 | /*@{*/ |
64 | 64 | ||
65 | /** Prints the application help. |
65 | /** Prints the application help. |
66 | */ |
66 | */ |
67 | void print_help( void ); |
67 | void print_help( void ); |
68 | 68 | ||
69 | /** Prints the parameter unrecognized message and the application help. |
69 | /** Prints the parameter unrecognized message and the application help. |
70 | * @param index The index of the parameter. Input parameter. |
70 | * @param index The index of the parameter. Input parameter. |
71 | * @param parameter The parameter name. Input parameter. |
71 | * @param parameter The parameter name. Input parameter. |
72 | */ |
72 | */ |
73 | void print_unrecognized( int index, const char * parameter ); |
73 | void print_unrecognized( int index, const char * parameter ); |
74 | 74 | ||
75 | /*@}*/ |
75 | /*@}*/ |
76 | 76 | ||
77 | /** @name Command line argumets parsing functions |
77 | /** @name Command line argumets parsing functions |
78 | */ |
78 | */ |
79 | /*@{*/ |
79 | /*@{*/ |
80 | 80 | ||
81 | /** Parses the next parameter as an integral number. |
81 | /** Parses the next parameter as an integral number. |
82 | * Uses the offseted actual parameter if the offset is set or the next one if not. |
82 | * Uses the offseted actual parameter if the offset is set or the next one if not. |
83 | * @param argc The total number of the parameters. Input parameter. |
83 | * @param argc The total number of the parameters. Input parameter. |
84 | * @param argv The parameters. Input parameter. |
84 | * @param argv The parameters. Input parameter. |
85 | * @param index The actual parameter index. Input/output parameter. |
85 | * @param index The actual parameter index. Input/output parameter. |
86 | * @param value The parsed parameter value. Output parameter. |
86 | * @param value The parsed parameter value. Output parameter. |
87 | * @param name The parameter name to be printed on errors. Input parameter. |
87 | * @param name The parameter name to be printed on errors. Input parameter. |
88 | * @param offset The value offset in the actual parameter. If not set, the next parameter is parsed instead. Input parameter. |
88 | * @param offset The value offset in the actual parameter. If not set, the next parameter is parsed instead. Input parameter. |
89 | * @returns EOK on success. |
89 | * @returns EOK on success. |
90 | * @returns EINVAL if the parameter is missing. |
90 | * @returns EINVAL if the parameter is missing. |
91 | * @returns EINVAL if the parameter is in wrong format. |
91 | * @returns EINVAL if the parameter is in wrong format. |
92 | */ |
92 | */ |
93 | int parse_parameter_int( int argc, char ** argv, int * index, int * value, const char * name, int offset ); |
93 | int parse_parameter_int( int argc, char ** argv, int * index, int * value, const char * name, int offset ); |
94 | 94 | ||
95 | /** Parses the next parameter as a character string. |
95 | /** Parses the next parameter as a character string. |
96 | * Uses the offseted actual parameter if the offset is set or the next one if not. |
96 | * Uses the offseted actual parameter if the offset is set or the next one if not. |
97 | * @param argc The total number of the parameters. Input parameter. |
97 | * @param argc The total number of the parameters. Input parameter. |
98 | * @param argv The parameters. Input parameter. |
98 | * @param argv The parameters. Input parameter. |
99 | * @param index The actual parameter index. Input/output parameter. |
99 | * @param index The actual parameter index. Input/output parameter. |
100 | * @param value The parsed parameter value. Output parameter. |
100 | * @param value The parsed parameter value. Output parameter. |
101 | * @param name The parameter name to be printed on errors. Input parameter. |
101 | * @param name The parameter name to be printed on errors. Input parameter. |
102 | * @param offset The value offset in the actual parameter. If not set, the next parameter is parsed instead. Input parameter. |
102 | * @param offset The value offset in the actual parameter. If not set, the next parameter is parsed instead. Input parameter. |
103 | * @returns EOK on success. |
103 | * @returns EOK on success. |
104 | * @returns EINVAL if the parameter is missing. |
104 | * @returns EINVAL if the parameter is missing. |
105 | */ |
105 | */ |
106 | int parse_parameter_string( int argc, char ** argv, int * index, char ** value, const char * name, int offset ); |
106 | int parse_parameter_string( int argc, char ** argv, int * index, char ** value, const char * name, int offset ); |
107 | 107 | ||
108 | /** Parses the next named parameter as an integral number. |
108 | /** Parses the next named parameter as an integral number. |
109 | * Uses the offseted actual parameter if the offset is set or the next one if not. |
109 | * Uses the offseted actual parameter if the offset is set or the next one if not. |
110 | * Translates the parameter using the parse_value function. |
110 | * Translates the parameter using the parse_value function. |
111 | * @param argc The total number of the parameters. Input parameter. |
111 | * @param argc The total number of the parameters. Input parameter. |
112 | * @param argv The parameters. Input parameter. |
112 | * @param argv The parameters. Input parameter. |
113 | * @param index The actual parameter index. Input/output parameter. |
113 | * @param index The actual parameter index. Input/output parameter. |
114 | * @param value The parsed parameter value. Output parameter. |
114 | * @param value The parsed parameter value. Output parameter. |
115 | * @param name The parameter name to be printed on errors. Input parameter. |
115 | * @param name The parameter name to be printed on errors. Input parameter. |
116 | * @param offset The value offset in the actual parameter. If not set, the next parameter is parsed instead. Input parameter. |
116 | * @param offset The value offset in the actual parameter. If not set, the next parameter is parsed instead. Input parameter. |
117 | * @param parse_value The translation function to parse the named value. |
117 | * @param parse_value The translation function to parse the named value. |
118 | * @returns EOK on success. |
118 | * @returns EOK on success. |
119 | * @returns EINVAL if the parameter is missing. |
119 | * @returns EINVAL if the parameter is missing. |
120 | * @returns ENOENT if the parameter name has not been found. |
120 | * @returns ENOENT if the parameter name has not been found. |
121 | */ |
121 | */ |
122 | int parse_parameter_name_int( int argc, char ** argv, int * index, int * value, const char * name, int offset, int ( * parse_value )( const char * value )); |
122 | int parse_parameter_name_int( int argc, char ** argv, int * index, int * value, const char * name, int offset, int ( * parse_value )( const char * value )); |
123 | 123 | ||
124 | /** Translates the character string to the protocol family number. |
124 | /** Translates the character string to the protocol family number. |
125 | * @param name The protocol family name. Input parameter. |
125 | * @param name The protocol family name. Input parameter. |
126 | * @returns The corresponding protocol family number. |
126 | * @returns The corresponding protocol family number. |
127 | */ |
127 | */ |
128 | int parse_protocol_family( const char * name ); |
128 | int parse_protocol_family( const char * name ); |
129 | 129 | ||
130 | /** Translates the character string to the socket type number. |
130 | /** Translates the character string to the socket type number. |
131 | * @param name The socket type name. Input parameter. |
131 | * @param name The socket type name. Input parameter. |
132 | * @returns The corresponding socket type number. |
132 | * @returns The corresponding socket type number. |
133 | */ |
133 | */ |
134 | int parse_socket_type( const char * name ); |
134 | int parse_socket_type( const char * name ); |
135 | 135 | ||
136 | /*@}*/ |
136 | /*@}*/ |
137 | 137 | ||
138 | void print_help( void ){ |
138 | void print_help( void ){ |
139 | printf( |
139 | printf( |
140 | "Network Echo aplication\n" \ |
140 | "Network Echo aplication\n" \ |
141 | "Usage: echo [options]\n" \ |
141 | "Usage: echo [options]\n" \ |
142 | "Where options are:\n" \ |
142 | "Where options are:\n" \ |
143 | "-p port_number | --port=port_number\n" \ |
143 | "-p port_number | --port=port_number\n" \ |
144 | "\tThe port number the application should listen at. The default is 7.\n" \ |
144 | "\tThe port number the application should listen at. The default is 7.\n" \ |
145 | "\n" \ |
145 | "\n" \ |
146 | "-s receive_size | --size=receive_size\n" \ |
146 | "-s receive_size | --size=receive_size\n" \ |
147 | "\tThe maximum receive data size the application should accept. The default is 1024 bytes.\n" \ |
147 | "\tThe maximum receive data size the application should accept. The default is 1024 bytes.\n" \ |
148 | "\n" \ |
148 | "\n" \ |
149 | "-c count | --count\n" \ |
149 | "-c count | --count\n" \ |
150 | "\tThe number of received messages to handle. A negative number means infinity. The default is infinity.\n" \ |
150 | "\tThe number of received messages to handle. A negative number means infinity. The default is infinity.\n" \ |
151 | "\n" \ |
151 | "\n" \ |
152 | "-r reply_string | --reply=reply_string\n" \ |
152 | "-r reply_string | --reply=reply_string\n" \ |
153 | "\tThe constant reply string. The default is the original data received.\n" \ |
153 | "\tThe constant reply string. The default is the original data received.\n" \ |
154 | "\n" \ |
154 | "\n" \ |
155 | "-f protocol_family | --family=protocol_family\n" \ |
155 | "-f protocol_family | --family=protocol_family\n" \ |
156 | "\tThe listenning socket protocol family. Only the PF_INET is supported.\n" |
156 | "\tThe listenning socket protocol family. Only the PF_INET is supported.\n" |
157 | "\n" \ |
157 | "\n" \ |
158 | "-h | --help\n" \ |
158 | "-h | --help\n" \ |
159 | "\tShow this application help.\n" |
159 | "\tShow this application help.\n" |
160 | "\n" \ |
160 | "\n" \ |
161 | "-t socket_type | --type=socket_type\n" \ |
161 | "-t socket_type | --type=socket_type\n" \ |
162 | "\tThe listenning socket type. Only the SOCK_DGRAM is supported.\n" \ |
162 | "\tThe listenning socket type. Only the SOCK_DGRAM is supported.\n" \ |
163 | "\n" \ |
163 | "\n" \ |
164 | "-v | --verbose\n" \ |
164 | "-v | --verbose\n" \ |
165 | "\tShow all output messages.\n" |
165 | "\tShow all output messages.\n" |
166 | ); |
166 | ); |
167 | } |
167 | } |
168 | 168 | ||
169 | int parse_parameter_int( int argc, char ** argv, int * index, int * value, const char * name, int offset ){ |
169 | int parse_parameter_int( int argc, char ** argv, int * index, int * value, const char * name, int offset ){ |
170 | char * rest; |
170 | char * rest; |
171 | 171 | ||
172 | if( offset ){ |
172 | if( offset ){ |
173 | * value = strtol( argv[ * index ] + offset, & rest, 10 ); |
173 | * value = strtol( argv[ * index ] + offset, & rest, 10 ); |
174 | }else if(( * index ) + 1 < argc ){ |
174 | }else if(( * index ) + 1 < argc ){ |
175 | ++ ( * index ); |
175 | ++ ( * index ); |
176 | * value = strtol( argv[ * index ], & rest, 10 ); |
176 | * value = strtol( argv[ * index ], & rest, 10 ); |
177 | }else{ |
177 | }else{ |
178 | fprintf( stderr, "Command line error: missing %s\n", name ); |
178 | fprintf( stderr, "Command line error: missing %s\n", name ); |
179 | return EINVAL; |
179 | return EINVAL; |
180 | } |
180 | } |
181 | if( rest && ( * rest )){ |
181 | if( rest && ( * rest )){ |
182 | fprintf( stderr, "Command line error: %s unrecognized (%d: %s)\n", name, * index, argv[ * index ] ); |
182 | fprintf( stderr, "Command line error: %s unrecognized (%d: %s)\n", name, * index, argv[ * index ] ); |
183 | return EINVAL; |
183 | return EINVAL; |
184 | } |
184 | } |
185 | return EOK; |
185 | return EOK; |
186 | } |
186 | } |
187 | 187 | ||
188 | int parse_parameter_string( int argc, char ** argv, int * index, char ** value, const char * name, int offset ){ |
188 | int parse_parameter_string( int argc, char ** argv, int * index, char ** value, const char * name, int offset ){ |
189 | if( offset ){ |
189 | if( offset ){ |
190 | * value = argv[ * index ] + offset; |
190 | * value = argv[ * index ] + offset; |
191 | }else if(( * index ) + 1 < argc ){ |
191 | }else if(( * index ) + 1 < argc ){ |
192 | ++ ( * index ); |
192 | ++ ( * index ); |
193 | * value = argv[ * index ]; |
193 | * value = argv[ * index ]; |
194 | }else{ |
194 | }else{ |
195 | fprintf( stderr, "Command line error: missing %s\n", name ); |
195 | fprintf( stderr, "Command line error: missing %s\n", name ); |
196 | return EINVAL; |
196 | return EINVAL; |
197 | } |
197 | } |
198 | return EOK; |
198 | return EOK; |
199 | } |
199 | } |
200 | 200 | ||
201 | int parse_parameter_name_int( int argc, char ** argv, int * index, int * value, const char * name, int offset, int ( * parse_value )( const char * value )){ |
201 | int parse_parameter_name_int( int argc, char ** argv, int * index, int * value, const char * name, int offset, int ( * parse_value )( const char * value )){ |
202 | ERROR_DECLARE; |
202 | ERROR_DECLARE; |
203 | 203 | ||
204 | char * parameter; |
204 | char * parameter; |
205 | 205 | ||
206 | ERROR_PROPAGATE( parse_parameter_string( argc, argv, index, & parameter, name, offset )); |
206 | ERROR_PROPAGATE( parse_parameter_string( argc, argv, index, & parameter, name, offset )); |
207 | * value = ( * parse_value )( parameter ); |
207 | * value = ( * parse_value )( parameter ); |
208 | if(( * value ) == ENOENT ){ |
208 | if(( * value ) == ENOENT ){ |
209 | fprintf( stderr, "Command line error: unrecognized %s value (%d: %s)\n", name, * index, parameter ); |
209 | fprintf( stderr, "Command line error: unrecognized %s value (%d: %s)\n", name, * index, parameter ); |
210 | return ENOENT; |
210 | return ENOENT; |
211 | } |
211 | } |
212 | return EOK; |
212 | return EOK; |
213 | } |
213 | } |
214 | 214 | ||
215 | int parse_protocol_family( const char * name ){ |
215 | int parse_protocol_family( const char * name ){ |
216 | if( str_lcmp( name, "PF_INET", 7 ) == 0 ){ |
216 | if( str_lcmp( name, "PF_INET", 7 ) == 0 ){ |
217 | return PF_INET; |
217 | return PF_INET; |
218 | } |
218 | } |
219 | return ENOENT; |
219 | return ENOENT; |
220 | } |
220 | } |
221 | 221 | ||
222 | int parse_socket_type( const char * name ){ |
222 | int parse_socket_type( const char * name ){ |
223 | if( str_lcmp( name, "SOCK_DGRAM", 11 ) == 0 ){ |
223 | if( str_lcmp( name, "SOCK_DGRAM", 11 ) == 0 ){ |
224 | return SOCK_DGRAM; |
224 | return SOCK_DGRAM; |
225 | } |
225 | } |
226 | return ENOENT; |
226 | return ENOENT; |
227 | } |
227 | } |
228 | 228 | ||
229 | void print_unrecognized( int index, const char * parameter ){ |
229 | void print_unrecognized( int index, const char * parameter ){ |
230 | fprintf( stderr, "Command line error - unrecognized parameter (%d: %s)\n", index, parameter ); |
230 | fprintf( stderr, "Command line error - unrecognized parameter (%d: %s)\n", index, parameter ); |
231 | print_help(); |
231 | print_help(); |
232 | } |
232 | } |
233 | 233 | ||
234 | int main( int argc, char * argv[] ){ |
234 | int main( int argc, char * argv[] ){ |
235 | ERROR_DECLARE; |
235 | ERROR_DECLARE; |
236 | 236 | ||
237 | int size = 1024; |
237 | size_t size = 1024; |
238 | int verbose = 0; |
238 | int verbose = 0; |
239 | char * reply = NULL; |
239 | char * reply = NULL; |
240 | sock_type_t type = SOCK_DGRAM; |
240 | sock_type_t type = SOCK_DGRAM; |
241 | int count = -1; |
241 | int count = -1; |
242 | struct sockaddr_in address = { .sin_family = PF_INET, .sin_port = 7 }; |
242 | struct sockaddr_in address = { .sin_family = PF_INET, .sin_port = 7 }; |
243 | 243 | ||
244 | int socket_id; |
244 | int socket_id; |
245 | int address_length; |
245 | int address_length; |
246 | char address_string[ INET_ADDRSTRLEN ]; |
246 | char address_string[ INET_ADDRSTRLEN ]; |
247 | char * data; |
247 | char * data; |
248 | int length; |
248 | size_t length; |
249 | int index; |
249 | int index; |
250 | size_t reply_length; |
250 | size_t reply_length; |
251 | int value; |
251 | int value; |
252 | 252 | ||
253 | printf( "Task %d - ", task_get_id()); |
253 | printf( "Task %d - ", task_get_id()); |
254 | printf( "%s\n", NAME ); |
254 | printf( "%s\n", NAME ); |
255 | 255 | ||
256 | for( index = 1; index < argc; ++ index ){ |
256 | for( index = 1; index < argc; ++ index ){ |
257 | if( argv[ index ][ 0 ] == '-' ){ |
257 | if( argv[ index ][ 0 ] == '-' ){ |
258 | switch( argv[ index ][ 1 ] ){ |
258 | switch( argv[ index ][ 1 ] ){ |
259 | case 'c': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & count, "count", 0 )); |
259 | case 'c': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & count, "count", 0 )); |
260 | break; |
260 | break; |
261 | case 'f': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "protocol family", 0, parse_protocol_family )); |
261 | case 'f': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "protocol family", 0, parse_protocol_family )); |
262 | address.sin_family = ( uint16_t ) value; |
262 | address.sin_family = ( uint16_t ) value; |
263 | break; |
263 | break; |
264 | case 'h': print_help(); |
264 | case 'h': print_help(); |
265 | return EOK; |
265 | return EOK; |
266 | break; |
266 | break; |
267 | case 'p': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 0 )); |
267 | case 'p': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 0 )); |
268 | address.sin_port = value; |
268 | address.sin_port = ( uint16_t ) value; |
269 | break; |
269 | break; |
270 | case 'r': ERROR_PROPAGATE( parse_parameter_string( argc, argv, & index, & reply, "reply string", 0 )); |
270 | case 'r': ERROR_PROPAGATE( parse_parameter_string( argc, argv, & index, & reply, "reply string", 0 )); |
271 | break; |
271 | break; |
272 | case 's': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & size, "receive size", 0 )); |
272 | case 's': ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "receive size", 0 )); |
- | 273 | size = (value >= 0 ) ? ( size_t ) value : 0; |
|
273 | break; |
274 | break; |
274 | case 't': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, ( int * ) & type, "socket_type", 0, parse_socket_type )); |
275 | case 't': ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "socket_type", 0, parse_socket_type )); |
- | 276 | type = ( sock_type_t ) value; |
|
275 | break; |
277 | break; |
276 | case 'v': verbose = 1; |
278 | case 'v': verbose = 1; |
277 | break; |
279 | break; |
278 | case '-': if( str_lcmp( argv[ index ] + 2, "count=", 6 ) == 0 ){ |
280 | case '-': if( str_lcmp( argv[ index ] + 2, "count=", 6 ) == 0 ){ |
279 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & count, "received count", 8 )) |
281 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & count, "received count", 8 )) |
280 | }else if( str_lcmp( argv[ index ] + 2, "family=", 7 ) == 0 ){ |
282 | }else if( str_lcmp( argv[ index ] + 2, "family=", 7 ) == 0 ){ |
281 | ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "protocol family", 9, parse_protocol_family )); |
283 | ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "protocol family", 9, parse_protocol_family )); |
282 | address.sin_family = value; |
284 | address.sin_family = ( uint16_t ) value; |
283 | }else if( str_lcmp( argv[ index ] + 2, "help", 5 ) == 0 ){ |
285 | }else if( str_lcmp( argv[ index ] + 2, "help", 5 ) == 0 ){ |
284 | print_help(); |
286 | print_help(); |
285 | return EOK; |
287 | return EOK; |
286 | }else if( str_lcmp( argv[ index ] + 2, "port=", 5 ) == 0 ){ |
288 | }else if( str_lcmp( argv[ index ] + 2, "port=", 5 ) == 0 ){ |
287 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 7 )); |
289 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "port number", 7 )); |
288 | address.sin_port = value; |
290 | address.sin_port = ( uint16_t ) value; |
289 | }else if( str_lcmp( argv[ index ] + 2, "reply=", 6 ) == 0 ){ |
291 | }else if( str_lcmp( argv[ index ] + 2, "reply=", 6 ) == 0 ){ |
290 | ERROR_PROPAGATE( parse_parameter_string( argc, argv, & index, & reply, "reply string", 8 )); |
292 | ERROR_PROPAGATE( parse_parameter_string( argc, argv, & index, & reply, "reply string", 8 )); |
291 | }else if( str_lcmp( argv[ index ] + 2, "size=", 5 ) == 0 ){ |
293 | }else if( str_lcmp( argv[ index ] + 2, "size=", 5 ) == 0 ){ |
292 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & size, "receive size", 7 )); |
294 | ERROR_PROPAGATE( parse_parameter_int( argc, argv, & index, & value, "receive size", 7 )); |
- | 295 | size = (value >= 0 ) ? ( size_t ) value : 0; |
|
293 | }else if( str_lcmp( argv[ index ] + 2, "type=", 5 ) == 0 ){ |
296 | }else if( str_lcmp( argv[ index ] + 2, "type=", 5 ) == 0 ){ |
294 | ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, ( int * ) & type, "socket_type", 7, parse_socket_type )); |
297 | ERROR_PROPAGATE( parse_parameter_name_int( argc, argv, & index, & value, "socket_type", 7, parse_socket_type )); |
- | 298 | type = ( sock_type_t ) value; |
|
295 | }else if( str_lcmp( argv[ index ] + 2, "verbose", 8 ) == 0 ){ |
299 | }else if( str_lcmp( argv[ index ] + 2, "verbose", 8 ) == 0 ){ |
296 | verbose = 1; |
300 | verbose = 1; |
297 | }else{ |
301 | }else{ |
298 | print_unrecognized( index, argv[ index ] + 2 ); |
302 | print_unrecognized( index, argv[ index ] + 2 ); |
299 | return EINVAL; |
303 | return EINVAL; |
300 | } |
304 | } |
301 | break; |
305 | break; |
302 | default: |
306 | default: |
303 | print_unrecognized( index, argv[ index ] + 1 ); |
307 | print_unrecognized( index, argv[ index ] + 1 ); |
304 | return EINVAL; |
308 | return EINVAL; |
305 | } |
309 | } |
306 | }else{ |
310 | }else{ |
307 | print_unrecognized( index, argv[ index ] ); |
311 | print_unrecognized( index, argv[ index ] ); |
308 | return EINVAL; |
312 | return EINVAL; |
309 | } |
313 | } |
310 | } |
314 | } |
311 | 315 | ||
312 | if( size <= 0 ){ |
316 | if( size <= 0 ){ |
313 | fprintf( stderr, "Receive size too small (%d). Using 1024 bytes instead.\n", size ); |
317 | fprintf( stderr, "Receive size too small (%d). Using 1024 bytes instead.\n", size ); |
314 | size = 1024; |
318 | size = 1024; |
315 | } |
319 | } |
316 | data = ( char * ) malloc( size + 1 ); |
320 | data = ( char * ) malloc( size + 1 ); |
317 | if( ! data ){ |
321 | if( ! data ){ |
318 | fprintf( stderr, "Failed to allocate receive buffer.\n" ); |
322 | fprintf( stderr, "Failed to allocate receive buffer.\n" ); |
319 | return ENOMEM; |
323 | return ENOMEM; |
320 | } |
324 | } |
321 | 325 | ||
322 | reply_length = reply ? str_length( reply ) : 0; |
326 | reply_length = reply ? str_length( reply ) : 0; |
323 | 327 | ||
324 | socket_id = socket( address.sin_family, type, 0 ); |
328 | socket_id = socket( address.sin_family, type, 0 ); |
325 | if( socket_id < 0 ){ |
329 | if( socket_id < 0 ){ |
326 | fprintf( stderr, "Socket create error %d\n", socket_id ); |
330 | fprintf( stderr, "Socket create error %d\n", socket_id ); |
327 | return socket_id; |
331 | return socket_id; |
328 | } |
332 | } |
329 | if( ERROR_OCCURRED( bind( socket_id, ( struct sockaddr * ) & address, sizeof( address )))){ |
333 | if( ERROR_OCCURRED( bind( socket_id, ( struct sockaddr * ) & address, sizeof( address )))){ |
330 | fprintf( stderr, "Socket bind error %d\n", ERROR_CODE ); |
334 | fprintf( stderr, "Socket bind error %d\n", ERROR_CODE ); |
331 | return ERROR_CODE; |
335 | return ERROR_CODE; |
332 | } |
336 | } |
333 | 337 | ||
334 | if( verbose ) printf( "Listenning at %d\n", address.sin_port ); |
338 | if( verbose ) printf( "Listenning at %d\n", address.sin_port ); |
335 | 339 | ||
336 | while( count ){ |
340 | while( count ){ |
337 | address_length = sizeof( address ); |
341 | address_length = sizeof( address ); |
338 | length = recvfrom( socket_id, data, size, 0, ( struct sockaddr * ) & address, & address_length ); |
342 | value = recvfrom( socket_id, data, size, 0, ( struct sockaddr * ) & address, & address_length ); |
339 | if( length < 0 ){ |
343 | if( value < 0 ){ |
340 | fprintf( stderr, "Socket receive error %d\n", length ); |
344 | fprintf( stderr, "Socket receive error %d\n", value ); |
341 | }else{ |
345 | }else{ |
- | 346 | length = ( size_t ) value; |
|
342 | if( verbose ){ |
347 | if( verbose ){ |
343 | if( ERROR_OCCURRED( inet_ntop( address.sin_family, ( uint8_t * ) & address.sin_addr.s_addr, address_string, sizeof( address_string )))){ |
348 | if( ERROR_OCCURRED( inet_ntop( address.sin_family, ( uint8_t * ) & address.sin_addr.s_addr, address_string, sizeof( address_string )))){ |
344 | fprintf( stderr, "Received address error %d\n", ERROR_CODE ); |
349 | fprintf( stderr, "Received address error %d\n", ERROR_CODE ); |
345 | continue; |
350 | continue; |
346 | }else{ |
351 | }else{ |
347 | data[ length ] = '\0'; |
352 | data[ length ] = '\0'; |
348 | printf( "Received from %s:%d\n%s\n", address_string, address.sin_port, data ); |
353 | printf( "Received from %s:%d\n%s\n", address_string, address.sin_port, data ); |
349 | } |
354 | } |
350 | } |
355 | } |
351 | if( ERROR_OCCURRED( sendto( socket_id, reply ? reply : data, reply ? reply_length : ( size_t ) length, 0, ( struct sockaddr * ) & address, sizeof( address )))){ |
356 | if( ERROR_OCCURRED( sendto( socket_id, reply ? reply : data, reply ? reply_length : length, 0, ( struct sockaddr * ) & address, sizeof( address )))){ |
352 | fprintf( stderr, "Socket send error %d\n", ERROR_CODE ); |
357 | fprintf( stderr, "Socket send error %d\n", ERROR_CODE ); |
353 | } |
358 | } |
354 | } |
359 | } |
355 | if( count > 0 ){ |
360 | if( count > 0 ){ |
356 | -- count; |
361 | -- count; |
357 | if( verbose ) printf( "Waiting for next %d packet(s)\n", count ); |
362 | if( verbose ) printf( "Waiting for next %d packet(s)\n", count ); |
358 | } |
363 | } |
359 | } |
364 | } |
360 | 365 | ||
361 | if( verbose ) printf( "Closing the socket\n" ); |
366 | if( verbose ) printf( "Closing the socket\n" ); |
362 | 367 | ||
363 | if( ERROR_OCCURRED( closesocket( socket_id ))){ |
368 | if( ERROR_OCCURRED( closesocket( socket_id ))){ |
364 | fprintf( stderr, "Close socket error %d\n", ERROR_CODE ); |
369 | fprintf( stderr, "Close socket error %d\n", ERROR_CODE ); |
365 | return ERROR_CODE; |
370 | return ERROR_CODE; |
366 | } |
371 | } |
367 | 372 | ||
368 | if( verbose ) printf( "Exiting\n" ); |
373 | if( verbose ) printf( "Exiting\n" ); |
369 | 374 | ||
370 | return EOK; |
375 | return EOK; |
371 | } |
376 | } |
372 | 377 | ||
373 | /** @} |
378 | /** @} |
374 | */ |
379 | */ |
375 | 380 |