1 | /*
|
---|
2 | * Copyright (c) 2008 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 udp
|
---|
30 | * @{
|
---|
31 | */
|
---|
32 |
|
---|
33 | /** @file
|
---|
34 | * UDP module implementation.
|
---|
35 | * @see udp.h
|
---|
36 | */
|
---|
37 |
|
---|
38 | #include <async.h>
|
---|
39 | #include <fibril_synch.h>
|
---|
40 | #include <malloc.h>
|
---|
41 | #include <stdio.h>
|
---|
42 | #include <ipc/ipc.h>
|
---|
43 | #include <ipc/services.h>
|
---|
44 | #include <ipc/socket.h>
|
---|
45 | #include <errno.h>
|
---|
46 | #include <err.h>
|
---|
47 |
|
---|
48 | #include <net/socket_codes.h>
|
---|
49 | #include <net/ip_protocols.h>
|
---|
50 | #include <net/in.h>
|
---|
51 | #include <net/in6.h>
|
---|
52 | #include <net/inet.h>
|
---|
53 | #include <net/modules.h>
|
---|
54 |
|
---|
55 | #include <net_messages.h>
|
---|
56 | #include <adt/dynamic_fifo.h>
|
---|
57 | #include <packet_client.h>
|
---|
58 | #include <packet_remote.h>
|
---|
59 | #include <net_checksum.h>
|
---|
60 | #include <ip_client.h>
|
---|
61 | #include <ip_interface.h>
|
---|
62 | #include <icmp_client.h>
|
---|
63 | #include <icmp_interface.h>
|
---|
64 | #include <net_interface.h>
|
---|
65 | #include <socket_core.h>
|
---|
66 | #include <tl_common.h>
|
---|
67 | #include <tl_local.h>
|
---|
68 | #include <tl_interface.h>
|
---|
69 | #include <tl_messages.h>
|
---|
70 |
|
---|
71 | #include "udp.h"
|
---|
72 | #include "udp_header.h"
|
---|
73 | #include "udp_module.h"
|
---|
74 |
|
---|
75 | /** UDP module name. */
|
---|
76 | #define NAME "UDP protocol"
|
---|
77 |
|
---|
78 | /** Default UDP checksum computing. */
|
---|
79 | #define NET_DEFAULT_UDP_CHECKSUM_COMPUTING true
|
---|
80 |
|
---|
81 | /** Default UDP autobind when sending via unbound sockets. */
|
---|
82 | #define NET_DEFAULT_UDP_AUTOBINDING true
|
---|
83 |
|
---|
84 | /** Maximum UDP fragment size. */
|
---|
85 | #define MAX_UDP_FRAGMENT_SIZE 65535
|
---|
86 |
|
---|
87 | /** Free ports pool start. */
|
---|
88 | #define UDP_FREE_PORTS_START 1025
|
---|
89 |
|
---|
90 | /** Free ports pool end. */
|
---|
91 | #define UDP_FREE_PORTS_END 65535
|
---|
92 |
|
---|
93 | /** Processes the received UDP packet queue.
|
---|
94 | *
|
---|
95 | * Is used as an entry point from the underlying IP module.
|
---|
96 | * Locks the global lock and calls udp_process_packet() function.
|
---|
97 | *
|
---|
98 | * @param[in] device_id The receiving device identifier.
|
---|
99 | * @param[in,out] packet The received packet queue.
|
---|
100 | * @param receiver The target service. Ignored parameter.
|
---|
101 | * @param[in] error The packet error reporting service. Prefixes the
|
---|
102 | * received packet.
|
---|
103 | * @returns EOK on success.
|
---|
104 | * @returns Other error codes as defined for the
|
---|
105 | * udp_process_packet() function.
|
---|
106 | */
|
---|
107 | int
|
---|
108 | udp_received_msg(device_id_t device_id, packet_t packet, services_t receiver,
|
---|
109 | services_t error);
|
---|
110 |
|
---|
111 | /** Processes the received UDP packet queue.
|
---|
112 | *
|
---|
113 | * Notifies the destination socket application.
|
---|
114 | * Releases the packet on error or sends an ICMP error notification.
|
---|
115 | *
|
---|
116 | * @param[in] device_id The receiving device identifier.
|
---|
117 | * @param[in,out] packet The received packet queue.
|
---|
118 | * @param[in] error The packet error reporting service. Prefixes the
|
---|
119 | * received packet.
|
---|
120 | * @returns EOK on success.
|
---|
121 | * @returns EINVAL if the packet is not valid.
|
---|
122 | * @returns EINVAL if the stored packet address is not the
|
---|
123 | * an_addr_t.
|
---|
124 | * @returns EINVAL if the packet does not contain any data.
|
---|
125 | * @returns NO_DATA if the packet content is shorter than the user
|
---|
126 | * datagram header.
|
---|
127 | * @returns ENOMEM if there is not enough memory left.
|
---|
128 | * @returns EADDRNOTAVAIL if the destination socket does not exist.
|
---|
129 | * @returns Other error codes as defined for the
|
---|
130 | * ip_client_process_packet() function.
|
---|
131 | */
|
---|
132 | int
|
---|
133 | udp_process_packet(device_id_t device_id, packet_t packet, services_t error);
|
---|
134 |
|
---|
135 | /** Releases the packet and returns the result.
|
---|
136 | *
|
---|
137 | * @param[in] packet The packet queue to be released.
|
---|
138 | * @param[in] result The result to be returned.
|
---|
139 | * @return The result parameter.
|
---|
140 | */
|
---|
141 | int udp_release_and_return(packet_t packet, int result);
|
---|
142 |
|
---|
143 | /** @name Socket messages processing functions
|
---|
144 | */
|
---|
145 | /*@{*/
|
---|
146 |
|
---|
147 | /** Processes the socket client messages.
|
---|
148 | *
|
---|
149 | * Runs until the client module disconnects.
|
---|
150 | *
|
---|
151 | * @param[in] callid The message identifier.
|
---|
152 | * @param[in] call The message parameters.
|
---|
153 | * @returns EOK on success.
|
---|
154 | * @see socket.h
|
---|
155 | */
|
---|
156 | int udp_process_client_messages(ipc_callid_t callid, ipc_call_t call);
|
---|
157 |
|
---|
158 | /** Sends data from the socket to the remote address.
|
---|
159 | *
|
---|
160 | * Binds the socket to a free port if not already connected/bound.
|
---|
161 | * Handles the NET_SOCKET_SENDTO message.
|
---|
162 | * Supports AF_INET and AF_INET6 address families.
|
---|
163 | *
|
---|
164 | * @param[in,out] local_sockets The application local sockets.
|
---|
165 | * @param[in] socket_id Socket identifier.
|
---|
166 | * @param[in] addr The destination address.
|
---|
167 | * @param[in] addrlen The address length.
|
---|
168 | * @param[in] fragments The number of data fragments.
|
---|
169 | * @param[out] data_fragment_size The data fragment size in bytes.
|
---|
170 | * @param[in] flags Various send flags.
|
---|
171 | * @returns EOK on success.
|
---|
172 | * @returns EAFNOTSUPPORT if the address family is not supported.
|
---|
173 | * @returns ENOTSOCK if the socket is not found.
|
---|
174 | * @returns EINVAL if the address is invalid.
|
---|
175 | * @returns ENOTCONN if the sending socket is not and cannot be
|
---|
176 | * bound.
|
---|
177 | * @returns ENOMEM if there is not enough memory left.
|
---|
178 | * @returns Other error codes as defined for the
|
---|
179 | * socket_read_packet_data() function.
|
---|
180 | * @returns Other error codes as defined for the
|
---|
181 | * ip_client_prepare_packet() function.
|
---|
182 | * @returns Other error codes as defined for the ip_send_msg()
|
---|
183 | * function.
|
---|
184 | */
|
---|
185 | int
|
---|
186 | udp_sendto_message(socket_cores_ref local_sockets, int socket_id,
|
---|
187 | const struct sockaddr * addr, socklen_t addrlen, int fragments,
|
---|
188 | size_t * data_fragment_size, int flags);
|
---|
189 |
|
---|
190 | /** Receives data to the socket.
|
---|
191 | *
|
---|
192 | * Handles the NET_SOCKET_RECVFROM message.
|
---|
193 | * Replies the source address as well.
|
---|
194 | *
|
---|
195 | * @param[in] local_sockets The application local sockets.
|
---|
196 | * @param[in] socket_id Socket identifier.
|
---|
197 | * @param[in] flags Various receive flags.
|
---|
198 | * @param[out] addrlen The source address length.
|
---|
199 | * @returns The number of bytes received.
|
---|
200 | * @returns ENOTSOCK if the socket is not found.
|
---|
201 | * @returns NO_DATA if there are no received packets or data.
|
---|
202 | * @returns ENOMEM if there is not enough memory left.
|
---|
203 | * @returns EINVAL if the received address is not an IP address.
|
---|
204 | * @returns Other error codes as defined for the packet_translate()
|
---|
205 | * function.
|
---|
206 | * @returns Other error codes as defined for the data_reply()
|
---|
207 | * function.
|
---|
208 | */
|
---|
209 | int
|
---|
210 | udp_recvfrom_message(socket_cores_ref local_sockets, int socket_id, int flags,
|
---|
211 | size_t * addrlen);
|
---|
212 |
|
---|
213 | /*@}*/
|
---|
214 |
|
---|
215 | /** UDP global data.
|
---|
216 | */
|
---|
217 | udp_globals_t udp_globals;
|
---|
218 |
|
---|
219 | int udp_initialize(async_client_conn_t client_connection)
|
---|
220 | {
|
---|
221 | ERROR_DECLARE;
|
---|
222 |
|
---|
223 | measured_string_t names[] = {
|
---|
224 | {
|
---|
225 | str_dup("UDP_CHECKSUM_COMPUTING"),
|
---|
226 | 22
|
---|
227 | },
|
---|
228 | {
|
---|
229 | str_dup("UDP_AUTOBINDING"),
|
---|
230 | 15
|
---|
231 | }
|
---|
232 | };
|
---|
233 | measured_string_ref configuration;
|
---|
234 | size_t count = sizeof(names) / sizeof(measured_string_t);
|
---|
235 | char * data;
|
---|
236 |
|
---|
237 | fibril_rwlock_initialize(&udp_globals.lock);
|
---|
238 | fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
239 |
|
---|
240 | udp_globals.icmp_phone = icmp_connect_module(SERVICE_ICMP,
|
---|
241 | ICMP_CONNECT_TIMEOUT);
|
---|
242 | udp_globals.ip_phone = ip_bind_service(SERVICE_IP, IPPROTO_UDP,
|
---|
243 | SERVICE_UDP, client_connection);
|
---|
244 | if (udp_globals.ip_phone < 0)
|
---|
245 | return udp_globals.ip_phone;
|
---|
246 |
|
---|
247 | // read default packet dimensions
|
---|
248 | ERROR_PROPAGATE(ip_packet_size_req(udp_globals.ip_phone, -1,
|
---|
249 | &udp_globals.packet_dimension));
|
---|
250 | ERROR_PROPAGATE(socket_ports_initialize(&udp_globals.sockets));
|
---|
251 | if (ERROR_OCCURRED(packet_dimensions_initialize(
|
---|
252 | &udp_globals.dimensions))) {
|
---|
253 | socket_ports_destroy(&udp_globals.sockets);
|
---|
254 | return ERROR_CODE;
|
---|
255 | }
|
---|
256 | udp_globals.packet_dimension.prefix += sizeof(udp_header_t);
|
---|
257 | udp_globals.packet_dimension.content -= sizeof(udp_header_t);
|
---|
258 | udp_globals.last_used_port = UDP_FREE_PORTS_START - 1;
|
---|
259 |
|
---|
260 | // get configuration
|
---|
261 | udp_globals.checksum_computing = NET_DEFAULT_UDP_CHECKSUM_COMPUTING;
|
---|
262 | udp_globals.autobinding = NET_DEFAULT_UDP_AUTOBINDING;
|
---|
263 | configuration = &names[0];
|
---|
264 | ERROR_PROPAGATE(net_get_conf_req(udp_globals.net_phone, &configuration,
|
---|
265 | count, &data));
|
---|
266 | if (configuration) {
|
---|
267 | if (configuration[0].value)
|
---|
268 | udp_globals.checksum_computing =
|
---|
269 | (configuration[0].value[0] == 'y');
|
---|
270 |
|
---|
271 | if (configuration[1].value)
|
---|
272 | udp_globals.autobinding =
|
---|
273 | (configuration[1].value[0] == 'y');
|
---|
274 |
|
---|
275 | net_free_settings(configuration, data);
|
---|
276 | }
|
---|
277 |
|
---|
278 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
279 | return EOK;
|
---|
280 | }
|
---|
281 |
|
---|
282 | int
|
---|
283 | udp_received_msg(device_id_t device_id, packet_t packet, services_t receiver,
|
---|
284 | services_t error)
|
---|
285 | {
|
---|
286 | int result;
|
---|
287 |
|
---|
288 | fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
289 | result = udp_process_packet(device_id, packet, error);
|
---|
290 | if (result != EOK)
|
---|
291 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
292 |
|
---|
293 | return result;
|
---|
294 | }
|
---|
295 |
|
---|
296 | int udp_process_packet(device_id_t device_id, packet_t packet, services_t error)
|
---|
297 | {
|
---|
298 | ERROR_DECLARE;
|
---|
299 |
|
---|
300 | size_t length;
|
---|
301 | size_t offset;
|
---|
302 | int result;
|
---|
303 | udp_header_ref header;
|
---|
304 | socket_core_ref socket;
|
---|
305 | packet_t next_packet;
|
---|
306 | size_t total_length;
|
---|
307 | uint32_t checksum;
|
---|
308 | int fragments;
|
---|
309 | packet_t tmp_packet;
|
---|
310 | icmp_type_t type;
|
---|
311 | icmp_code_t code;
|
---|
312 | void *ip_header;
|
---|
313 | struct sockaddr *src;
|
---|
314 | struct sockaddr *dest;
|
---|
315 | packet_dimension_ref packet_dimension;
|
---|
316 |
|
---|
317 | if (error) {
|
---|
318 | switch (error) {
|
---|
319 | case SERVICE_ICMP:
|
---|
320 | // ignore error
|
---|
321 | // length = icmp_client_header_length(packet);
|
---|
322 | // process error
|
---|
323 | result = icmp_client_process_packet(packet, &type,
|
---|
324 | &code, NULL, NULL);
|
---|
325 | if (result < 0)
|
---|
326 | return udp_release_and_return(packet, result);
|
---|
327 | length = (size_t) result;
|
---|
328 | if (ERROR_OCCURRED(packet_trim(packet, length, 0)))
|
---|
329 | return udp_release_and_return(packet,
|
---|
330 | ERROR_CODE);
|
---|
331 | break;
|
---|
332 | default:
|
---|
333 | return udp_release_and_return(packet, ENOTSUP);
|
---|
334 | }
|
---|
335 | }
|
---|
336 |
|
---|
337 | // TODO process received ipopts?
|
---|
338 | result = ip_client_process_packet(packet, NULL, NULL, NULL, NULL, NULL);
|
---|
339 | if (result < 0)
|
---|
340 | return udp_release_and_return(packet, result);
|
---|
341 | offset = (size_t) result;
|
---|
342 |
|
---|
343 | length = packet_get_data_length(packet);
|
---|
344 | if (length <= 0)
|
---|
345 | return udp_release_and_return(packet, EINVAL);
|
---|
346 | if (length < UDP_HEADER_SIZE + offset)
|
---|
347 | return udp_release_and_return(packet, NO_DATA);
|
---|
348 |
|
---|
349 | // trim all but UDP header
|
---|
350 | if (ERROR_OCCURRED(packet_trim(packet, offset, 0)))
|
---|
351 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
352 |
|
---|
353 | // get udp header
|
---|
354 | header = (udp_header_ref) packet_get_data(packet);
|
---|
355 | if (!header)
|
---|
356 | return udp_release_and_return(packet, NO_DATA);
|
---|
357 |
|
---|
358 | // find the destination socket
|
---|
359 | socket = socket_port_find(&udp_globals.sockets,
|
---|
360 | ntohs(header->destination_port), SOCKET_MAP_KEY_LISTENING, 0);
|
---|
361 | if (!socket) {
|
---|
362 | if (tl_prepare_icmp_packet(udp_globals.net_phone,
|
---|
363 | udp_globals.icmp_phone, packet, error) == EOK) {
|
---|
364 | icmp_destination_unreachable_msg(udp_globals.icmp_phone,
|
---|
365 | ICMP_PORT_UNREACH, 0, packet);
|
---|
366 | }
|
---|
367 | return EADDRNOTAVAIL;
|
---|
368 | }
|
---|
369 |
|
---|
370 | // count the received packet fragments
|
---|
371 | next_packet = packet;
|
---|
372 | fragments = 0;
|
---|
373 | total_length = ntohs(header->total_length);
|
---|
374 |
|
---|
375 | // compute header checksum if set
|
---|
376 | if (header->checksum && (!error)) {
|
---|
377 | result = packet_get_addr(packet, (uint8_t **) &src,
|
---|
378 | (uint8_t **) &dest);
|
---|
379 | if( result <= 0)
|
---|
380 | return udp_release_and_return(packet, result);
|
---|
381 |
|
---|
382 | if (ERROR_OCCURRED(ip_client_get_pseudo_header(IPPROTO_UDP,
|
---|
383 | src, result, dest, result, total_length, &ip_header,
|
---|
384 | &length))) {
|
---|
385 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
386 | } else {
|
---|
387 | checksum = compute_checksum(0, ip_header, length);
|
---|
388 | // the udp header checksum will be added with the first
|
---|
389 | // fragment later
|
---|
390 | free(ip_header);
|
---|
391 | }
|
---|
392 | } else {
|
---|
393 | header->checksum = 0;
|
---|
394 | checksum = 0;
|
---|
395 | }
|
---|
396 |
|
---|
397 | do {
|
---|
398 | ++ fragments;
|
---|
399 | length = packet_get_data_length(next_packet);
|
---|
400 | if (length <= 0)
|
---|
401 | return udp_release_and_return(packet, NO_DATA);
|
---|
402 |
|
---|
403 | if (total_length < length) {
|
---|
404 | if (ERROR_OCCURRED(packet_trim(next_packet, 0,
|
---|
405 | length - total_length))) {
|
---|
406 | return udp_release_and_return(packet,
|
---|
407 | ERROR_CODE);
|
---|
408 | }
|
---|
409 |
|
---|
410 | // add partial checksum if set
|
---|
411 | if (header->checksum) {
|
---|
412 | checksum = compute_checksum(checksum,
|
---|
413 | packet_get_data(packet),
|
---|
414 | packet_get_data_length(packet));
|
---|
415 | }
|
---|
416 |
|
---|
417 | // relese the rest of the packet fragments
|
---|
418 | tmp_packet = pq_next(next_packet);
|
---|
419 | while (tmp_packet) {
|
---|
420 | next_packet = pq_detach(tmp_packet);
|
---|
421 | pq_release_remote(udp_globals.net_phone,
|
---|
422 | packet_get_id(tmp_packet));
|
---|
423 | tmp_packet = next_packet;
|
---|
424 | }
|
---|
425 |
|
---|
426 | // exit the loop
|
---|
427 | break;
|
---|
428 | }
|
---|
429 | total_length -= length;
|
---|
430 |
|
---|
431 | // add partial checksum if set
|
---|
432 | if (header->checksum) {
|
---|
433 | checksum = compute_checksum(checksum,
|
---|
434 | packet_get_data(packet),
|
---|
435 | packet_get_data_length(packet));
|
---|
436 | }
|
---|
437 |
|
---|
438 | } while ((next_packet = pq_next(next_packet)) && (total_length > 0));
|
---|
439 |
|
---|
440 | // check checksum
|
---|
441 | if (header->checksum) {
|
---|
442 | if (flip_checksum(compact_checksum(checksum)) !=
|
---|
443 | IP_CHECKSUM_ZERO) {
|
---|
444 | if (tl_prepare_icmp_packet(udp_globals.net_phone,
|
---|
445 | udp_globals.icmp_phone, packet, error) == EOK) {
|
---|
446 | // checksum error ICMP
|
---|
447 | icmp_parameter_problem_msg(
|
---|
448 | udp_globals.icmp_phone, ICMP_PARAM_POINTER,
|
---|
449 | ((size_t) ((void *) &header->checksum)) -
|
---|
450 | ((size_t) ((void *) header)), packet);
|
---|
451 | }
|
---|
452 | return EINVAL;
|
---|
453 | }
|
---|
454 | }
|
---|
455 |
|
---|
456 | // queue the received packet
|
---|
457 | if (ERROR_OCCURRED(dyn_fifo_push(&socket->received,
|
---|
458 | packet_get_id(packet), SOCKET_MAX_RECEIVED_SIZE)) ||
|
---|
459 | ERROR_OCCURRED(tl_get_ip_packet_dimension(udp_globals.ip_phone,
|
---|
460 | &udp_globals.dimensions, device_id, &packet_dimension))) {
|
---|
461 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
462 | }
|
---|
463 |
|
---|
464 | // notify the destination socket
|
---|
465 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
466 | async_msg_5(socket->phone, NET_SOCKET_RECEIVED,
|
---|
467 | (ipcarg_t) socket->socket_id, packet_dimension->content, 0, 0,
|
---|
468 | (ipcarg_t) fragments);
|
---|
469 |
|
---|
470 | return EOK;
|
---|
471 | }
|
---|
472 |
|
---|
473 | int
|
---|
474 | udp_message_standalone(ipc_callid_t callid, ipc_call_t * call,
|
---|
475 | ipc_call_t * answer, int * answer_count)
|
---|
476 | {
|
---|
477 | ERROR_DECLARE;
|
---|
478 |
|
---|
479 | packet_t packet;
|
---|
480 |
|
---|
481 | *answer_count = 0;
|
---|
482 |
|
---|
483 | switch (IPC_GET_METHOD(*call)) {
|
---|
484 | case NET_TL_RECEIVED:
|
---|
485 | if (!ERROR_OCCURRED(packet_translate_remote(
|
---|
486 | udp_globals.net_phone, &packet, IPC_GET_PACKET(call)))) {
|
---|
487 | ERROR_CODE = udp_received_msg(IPC_GET_DEVICE(call),
|
---|
488 | packet, SERVICE_UDP, IPC_GET_ERROR(call));
|
---|
489 | }
|
---|
490 | return ERROR_CODE;
|
---|
491 |
|
---|
492 | case IPC_M_CONNECT_TO_ME:
|
---|
493 | return udp_process_client_messages(callid, * call);
|
---|
494 | }
|
---|
495 |
|
---|
496 | return ENOTSUP;
|
---|
497 | }
|
---|
498 |
|
---|
499 | int udp_process_client_messages(ipc_callid_t callid, ipc_call_t call)
|
---|
500 | {
|
---|
501 | int res;
|
---|
502 | bool keep_on_going = true;
|
---|
503 | socket_cores_t local_sockets;
|
---|
504 | int app_phone = IPC_GET_PHONE(&call);
|
---|
505 | struct sockaddr *addr;
|
---|
506 | int socket_id;
|
---|
507 | size_t addrlen;
|
---|
508 | size_t size;
|
---|
509 | ipc_call_t answer;
|
---|
510 | int answer_count;
|
---|
511 | packet_dimension_ref packet_dimension;
|
---|
512 |
|
---|
513 | /*
|
---|
514 | * Accept the connection
|
---|
515 | * - Answer the first IPC_M_CONNECT_TO_ME call.
|
---|
516 | */
|
---|
517 | res = EOK;
|
---|
518 | answer_count = 0;
|
---|
519 |
|
---|
520 | // The client connection is only in one fibril and therefore no
|
---|
521 | // additional locks are needed.
|
---|
522 |
|
---|
523 | socket_cores_initialize(&local_sockets);
|
---|
524 |
|
---|
525 | while (keep_on_going) {
|
---|
526 |
|
---|
527 | // answer the call
|
---|
528 | answer_call(callid, res, &answer, answer_count);
|
---|
529 |
|
---|
530 | // refresh data
|
---|
531 | refresh_answer(&answer, &answer_count);
|
---|
532 |
|
---|
533 | // get the next call
|
---|
534 | callid = async_get_call(&call);
|
---|
535 |
|
---|
536 | // process the call
|
---|
537 | switch (IPC_GET_METHOD(call)) {
|
---|
538 | case IPC_M_PHONE_HUNGUP:
|
---|
539 | keep_on_going = false;
|
---|
540 | res = EHANGUP;
|
---|
541 | break;
|
---|
542 |
|
---|
543 | case NET_SOCKET:
|
---|
544 | socket_id = SOCKET_GET_SOCKET_ID(call);
|
---|
545 | res = socket_create(&local_sockets, app_phone, NULL,
|
---|
546 | &socket_id);
|
---|
547 | SOCKET_SET_SOCKET_ID(answer, socket_id);
|
---|
548 |
|
---|
549 | if (res != EOK)
|
---|
550 | break;
|
---|
551 |
|
---|
552 | if (tl_get_ip_packet_dimension(udp_globals.ip_phone,
|
---|
553 | &udp_globals.dimensions, DEVICE_INVALID_ID,
|
---|
554 | &packet_dimension) == EOK) {
|
---|
555 | SOCKET_SET_DATA_FRAGMENT_SIZE(answer,
|
---|
556 | packet_dimension->content);
|
---|
557 | }
|
---|
558 |
|
---|
559 | // SOCKET_SET_DATA_FRAGMENT_SIZE(answer,
|
---|
560 | // MAX_UDP_FRAGMENT_SIZE);
|
---|
561 | SOCKET_SET_HEADER_SIZE(answer, UDP_HEADER_SIZE);
|
---|
562 | answer_count = 3;
|
---|
563 | break;
|
---|
564 |
|
---|
565 | case NET_SOCKET_BIND:
|
---|
566 | res = data_receive((void **) &addr, &addrlen);
|
---|
567 | if (res != EOK)
|
---|
568 | break;
|
---|
569 | fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
570 | res = socket_bind(&local_sockets, &udp_globals.sockets,
|
---|
571 | SOCKET_GET_SOCKET_ID(call), addr, addrlen,
|
---|
572 | UDP_FREE_PORTS_START, UDP_FREE_PORTS_END,
|
---|
573 | udp_globals.last_used_port);
|
---|
574 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
575 | free(addr);
|
---|
576 | break;
|
---|
577 |
|
---|
578 | case NET_SOCKET_SENDTO:
|
---|
579 | res = data_receive((void **) &addr, &addrlen);
|
---|
580 | if (res != EOK)
|
---|
581 | break;
|
---|
582 |
|
---|
583 | fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
584 | res = udp_sendto_message(&local_sockets,
|
---|
585 | SOCKET_GET_SOCKET_ID(call), addr, addrlen,
|
---|
586 | SOCKET_GET_DATA_FRAGMENTS(call), &size,
|
---|
587 | SOCKET_GET_FLAGS(call));
|
---|
588 | SOCKET_SET_DATA_FRAGMENT_SIZE(answer, size);
|
---|
589 |
|
---|
590 | if (res != EOK)
|
---|
591 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
592 | else
|
---|
593 | answer_count = 2;
|
---|
594 |
|
---|
595 | free(addr);
|
---|
596 | break;
|
---|
597 |
|
---|
598 | case NET_SOCKET_RECVFROM:
|
---|
599 | fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
600 | res = udp_recvfrom_message(&local_sockets,
|
---|
601 | SOCKET_GET_SOCKET_ID(call), SOCKET_GET_FLAGS(call),
|
---|
602 | &addrlen);
|
---|
603 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
604 |
|
---|
605 | if (res <= 0)
|
---|
606 | break;
|
---|
607 |
|
---|
608 | SOCKET_SET_READ_DATA_LENGTH(answer, res);
|
---|
609 | SOCKET_SET_ADDRESS_LENGTH(answer, addrlen);
|
---|
610 | answer_count = 3;
|
---|
611 | res = EOK;
|
---|
612 | break;
|
---|
613 |
|
---|
614 | case NET_SOCKET_CLOSE:
|
---|
615 | fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
616 | res = socket_destroy(udp_globals.net_phone,
|
---|
617 | SOCKET_GET_SOCKET_ID(call), &local_sockets,
|
---|
618 | &udp_globals.sockets, NULL);
|
---|
619 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
620 | break;
|
---|
621 |
|
---|
622 | case NET_SOCKET_GETSOCKOPT:
|
---|
623 | case NET_SOCKET_SETSOCKOPT:
|
---|
624 | default:
|
---|
625 | res = ENOTSUP;
|
---|
626 | break;
|
---|
627 | }
|
---|
628 | }
|
---|
629 |
|
---|
630 | // release the application phone
|
---|
631 | ipc_hangup(app_phone);
|
---|
632 |
|
---|
633 | // release all local sockets
|
---|
634 | socket_cores_release(udp_globals.net_phone, &local_sockets,
|
---|
635 | &udp_globals.sockets, NULL);
|
---|
636 |
|
---|
637 | return res;
|
---|
638 | }
|
---|
639 |
|
---|
640 | int
|
---|
641 | udp_sendto_message(socket_cores_ref local_sockets, int socket_id,
|
---|
642 | const struct sockaddr *addr, socklen_t addrlen, int fragments,
|
---|
643 | size_t *data_fragment_size, int flags)
|
---|
644 | {
|
---|
645 | ERROR_DECLARE;
|
---|
646 |
|
---|
647 | socket_core_ref socket;
|
---|
648 | packet_t packet;
|
---|
649 | packet_t next_packet;
|
---|
650 | udp_header_ref header;
|
---|
651 | int index;
|
---|
652 | size_t total_length;
|
---|
653 | int result;
|
---|
654 | uint16_t dest_port;
|
---|
655 | uint32_t checksum;
|
---|
656 | void *ip_header;
|
---|
657 | size_t headerlen;
|
---|
658 | device_id_t device_id;
|
---|
659 | packet_dimension_ref packet_dimension;
|
---|
660 |
|
---|
661 | ERROR_PROPAGATE(tl_get_address_port(addr, addrlen, &dest_port));
|
---|
662 |
|
---|
663 | socket = socket_cores_find(local_sockets, socket_id);
|
---|
664 | if (!socket)
|
---|
665 | return ENOTSOCK;
|
---|
666 |
|
---|
667 | if ((socket->port <= 0) && udp_globals.autobinding) {
|
---|
668 | // bind the socket to a random free port if not bound
|
---|
669 | // do {
|
---|
670 | // try to find a free port
|
---|
671 | // fibril_rwlock_read_unlock(&udp_globals.lock);
|
---|
672 | // fibril_rwlock_write_lock(&udp_globals.lock);
|
---|
673 | // might be changed in the meantime
|
---|
674 | // if (socket->port <= 0) {
|
---|
675 | if (ERROR_OCCURRED(socket_bind_free_port(
|
---|
676 | &udp_globals.sockets, socket,
|
---|
677 | UDP_FREE_PORTS_START, UDP_FREE_PORTS_END,
|
---|
678 | udp_globals.last_used_port))) {
|
---|
679 | // fibril_rwlock_write_unlock(
|
---|
680 | // &udp_globals.lock);
|
---|
681 | // fibril_rwlock_read_lock(
|
---|
682 | // &udp_globals.lock);
|
---|
683 | return ERROR_CODE;
|
---|
684 | }
|
---|
685 | // set the next port as the search starting port
|
---|
686 | // number
|
---|
687 | udp_globals.last_used_port = socket->port;
|
---|
688 | // }
|
---|
689 | // fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
690 | // fibril_rwlock_read_lock(&udp_globals.lock);
|
---|
691 | // might be changed in the meantime
|
---|
692 | // } while (socket->port <= 0);
|
---|
693 | }
|
---|
694 |
|
---|
695 | if (udp_globals.checksum_computing) {
|
---|
696 | if (ERROR_OCCURRED(ip_get_route_req(udp_globals.ip_phone,
|
---|
697 | IPPROTO_UDP, addr, addrlen, &device_id, &ip_header,
|
---|
698 | &headerlen))) {
|
---|
699 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
700 | }
|
---|
701 | // get the device packet dimension
|
---|
702 | // ERROR_PROPAGATE(tl_get_ip_packet_dimension(udp_globals.ip_phone,
|
---|
703 | // &udp_globals.dimensions, device_id, &packet_dimension));
|
---|
704 | }
|
---|
705 | // } else {
|
---|
706 | // do not ask all the time
|
---|
707 | ERROR_PROPAGATE(ip_packet_size_req(udp_globals.ip_phone, -1,
|
---|
708 | &udp_globals.packet_dimension));
|
---|
709 | packet_dimension = &udp_globals.packet_dimension;
|
---|
710 | // }
|
---|
711 |
|
---|
712 | // read the first packet fragment
|
---|
713 | result = tl_socket_read_packet_data(udp_globals.net_phone, &packet,
|
---|
714 | UDP_HEADER_SIZE, packet_dimension, addr, addrlen);
|
---|
715 | if (result < 0)
|
---|
716 | return result;
|
---|
717 |
|
---|
718 | total_length = (size_t) result;
|
---|
719 | if (udp_globals.checksum_computing)
|
---|
720 | checksum = compute_checksum(0, packet_get_data(packet),
|
---|
721 | packet_get_data_length(packet));
|
---|
722 | else
|
---|
723 | checksum = 0;
|
---|
724 |
|
---|
725 | // prefix the udp header
|
---|
726 | header = PACKET_PREFIX(packet, udp_header_t);
|
---|
727 | if(! header)
|
---|
728 | return udp_release_and_return(packet, ENOMEM);
|
---|
729 |
|
---|
730 | bzero(header, sizeof(*header));
|
---|
731 | // read the rest of the packet fragments
|
---|
732 | for (index = 1; index < fragments; ++ index) {
|
---|
733 | result = tl_socket_read_packet_data(udp_globals.net_phone,
|
---|
734 | &next_packet, 0, packet_dimension, addr, addrlen);
|
---|
735 | if (result < 0)
|
---|
736 | return udp_release_and_return(packet, result);
|
---|
737 |
|
---|
738 | if (ERROR_OCCURRED(pq_add(&packet, next_packet, index, 0)))
|
---|
739 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
740 |
|
---|
741 | total_length += (size_t) result;
|
---|
742 | if (udp_globals.checksum_computing) {
|
---|
743 | checksum = compute_checksum(checksum,
|
---|
744 | packet_get_data(next_packet),
|
---|
745 | packet_get_data_length(next_packet));
|
---|
746 | }
|
---|
747 | }
|
---|
748 |
|
---|
749 | // set the udp header
|
---|
750 | header->source_port = htons((socket->port > 0) ? socket->port : 0);
|
---|
751 | header->destination_port = htons(dest_port);
|
---|
752 | header->total_length = htons(total_length + sizeof(*header));
|
---|
753 | header->checksum = 0;
|
---|
754 | if (udp_globals.checksum_computing) {
|
---|
755 | // update the pseudo header
|
---|
756 | if (ERROR_OCCURRED(ip_client_set_pseudo_header_data_length(
|
---|
757 | ip_header, headerlen, total_length + UDP_HEADER_SIZE))) {
|
---|
758 | free(ip_header);
|
---|
759 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
760 | }
|
---|
761 |
|
---|
762 | // finish the checksum computation
|
---|
763 | checksum = compute_checksum(checksum, ip_header, headerlen);
|
---|
764 | checksum = compute_checksum(checksum, (uint8_t *) header,
|
---|
765 | sizeof(*header));
|
---|
766 | header->checksum =
|
---|
767 | htons(flip_checksum(compact_checksum(checksum)));
|
---|
768 | free(ip_header);
|
---|
769 | } else {
|
---|
770 | device_id = DEVICE_INVALID_ID;
|
---|
771 | }
|
---|
772 |
|
---|
773 | // prepare the first packet fragment
|
---|
774 | if (ERROR_OCCURRED(ip_client_prepare_packet(packet, IPPROTO_UDP, 0, 0,
|
---|
775 | 0, 0))) {
|
---|
776 | return udp_release_and_return(packet, ERROR_CODE);
|
---|
777 | }
|
---|
778 |
|
---|
779 | // send the packet
|
---|
780 | fibril_rwlock_write_unlock(&udp_globals.lock);
|
---|
781 | ip_send_msg(udp_globals.ip_phone, device_id, packet, SERVICE_UDP, 0);
|
---|
782 |
|
---|
783 | return EOK;
|
---|
784 | }
|
---|
785 |
|
---|
786 | int
|
---|
787 | udp_recvfrom_message(socket_cores_ref local_sockets, int socket_id, int flags,
|
---|
788 | size_t *addrlen)
|
---|
789 | {
|
---|
790 | ERROR_DECLARE;
|
---|
791 |
|
---|
792 | socket_core_ref socket;
|
---|
793 | int packet_id;
|
---|
794 | packet_t packet;
|
---|
795 | udp_header_ref header;
|
---|
796 | struct sockaddr *addr;
|
---|
797 | size_t length;
|
---|
798 | uint8_t *data;
|
---|
799 | int result;
|
---|
800 |
|
---|
801 | // find the socket
|
---|
802 | socket = socket_cores_find(local_sockets, socket_id);
|
---|
803 | if (!socket)
|
---|
804 | return ENOTSOCK;
|
---|
805 |
|
---|
806 | // get the next received packet
|
---|
807 | packet_id = dyn_fifo_value(&socket->received);
|
---|
808 | if (packet_id < 0)
|
---|
809 | return NO_DATA;
|
---|
810 |
|
---|
811 | ERROR_PROPAGATE(packet_translate_remote(udp_globals.net_phone, &packet,
|
---|
812 | packet_id));
|
---|
813 |
|
---|
814 | // get udp header
|
---|
815 | data = packet_get_data(packet);
|
---|
816 | if (!data) {
|
---|
817 | pq_release_remote(udp_globals.net_phone, packet_id);
|
---|
818 | return NO_DATA;
|
---|
819 | }
|
---|
820 | header = (udp_header_ref) data;
|
---|
821 |
|
---|
822 | // set the source address port
|
---|
823 | result = packet_get_addr(packet, (uint8_t **) &addr, NULL);
|
---|
824 | if (ERROR_OCCURRED(tl_set_address_port(addr, result,
|
---|
825 | ntohs(header->source_port)))) {
|
---|
826 | pq_release_remote(udp_globals.net_phone, packet_id);
|
---|
827 | return ERROR_CODE;
|
---|
828 | }
|
---|
829 | *addrlen = (size_t) result;
|
---|
830 |
|
---|
831 | // send the source address
|
---|
832 | ERROR_PROPAGATE(data_reply(addr, * addrlen));
|
---|
833 |
|
---|
834 | // trim the header
|
---|
835 | ERROR_PROPAGATE(packet_trim(packet, UDP_HEADER_SIZE, 0));
|
---|
836 |
|
---|
837 | // reply the packets
|
---|
838 | ERROR_PROPAGATE(socket_reply_packets(packet, &length));
|
---|
839 |
|
---|
840 | // release the packet
|
---|
841 | dyn_fifo_pop(&socket->received);
|
---|
842 | pq_release_remote(udp_globals.net_phone, packet_get_id(packet));
|
---|
843 |
|
---|
844 | // return the total length
|
---|
845 | return (int) length;
|
---|
846 | }
|
---|
847 |
|
---|
848 | int udp_release_and_return(packet_t packet, int result)
|
---|
849 | {
|
---|
850 | pq_release_remote(udp_globals.net_phone, packet_get_id(packet));
|
---|
851 | return result;
|
---|
852 | }
|
---|
853 |
|
---|
854 | /** Default thread for new connections.
|
---|
855 | *
|
---|
856 | * @param[in] iid The initial message identifier.
|
---|
857 | * @param[in] icall The initial message call structure.
|
---|
858 | *
|
---|
859 | */
|
---|
860 | static void tl_client_connection(ipc_callid_t iid, ipc_call_t * icall)
|
---|
861 | {
|
---|
862 | /*
|
---|
863 | * Accept the connection
|
---|
864 | * - Answer the first IPC_M_CONNECT_ME_TO call.
|
---|
865 | */
|
---|
866 | ipc_answer_0(iid, EOK);
|
---|
867 |
|
---|
868 | while (true) {
|
---|
869 | ipc_call_t answer;
|
---|
870 | int answer_count;
|
---|
871 |
|
---|
872 | /* Clear the answer structure */
|
---|
873 | refresh_answer(&answer, &answer_count);
|
---|
874 |
|
---|
875 | /* Fetch the next message */
|
---|
876 | ipc_call_t call;
|
---|
877 | ipc_callid_t callid = async_get_call(&call);
|
---|
878 |
|
---|
879 | /* Process the message */
|
---|
880 | int res = tl_module_message_standalone(callid, &call, &answer,
|
---|
881 | &answer_count);
|
---|
882 |
|
---|
883 | /*
|
---|
884 | * End if said to either by the message or the processing result
|
---|
885 | */
|
---|
886 | if ((IPC_GET_METHOD(call) == IPC_M_PHONE_HUNGUP) ||
|
---|
887 | (res == EHANGUP))
|
---|
888 | return;
|
---|
889 |
|
---|
890 | /* Answer the message */
|
---|
891 | answer_call(callid, res, &answer, answer_count);
|
---|
892 | }
|
---|
893 | }
|
---|
894 |
|
---|
895 | /** Starts the module.
|
---|
896 | *
|
---|
897 | * @param argc The count of the command line arguments. Ignored
|
---|
898 | * parameter.
|
---|
899 | * @param argv The command line parameters. Ignored parameter.
|
---|
900 | *
|
---|
901 | * @returns EOK on success.
|
---|
902 | * @returns Other error codes as defined for each specific module
|
---|
903 | * start function.
|
---|
904 | */
|
---|
905 | int main(int argc, char *argv[])
|
---|
906 | {
|
---|
907 | ERROR_DECLARE;
|
---|
908 |
|
---|
909 | /* Start the module */
|
---|
910 | if (ERROR_OCCURRED(tl_module_start_standalone(tl_client_connection)))
|
---|
911 | return ERROR_CODE;
|
---|
912 |
|
---|
913 | return EOK;
|
---|
914 | }
|
---|
915 |
|
---|
916 | /** @}
|
---|
917 | */
|
---|