Rev 4713 | Rev 4722 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 4499 | mejdrech | 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 |
||
| 4701 | mejdrech | 34 | * UDP module implementation. |
| 35 | * @see udp.h |
||
| 4499 | mejdrech | 36 | */ |
| 37 | |||
| 38 | #include <async.h> |
||
| 4700 | mejdrech | 39 | #include <fibril_sync.h> |
| 4579 | mejdrech | 40 | #include <malloc.h> |
| 4499 | mejdrech | 41 | |
| 42 | #include <ipc/ipc.h> |
||
| 43 | #include <ipc/services.h> |
||
| 44 | |||
| 45 | #include "../../err.h" |
||
| 46 | #include "../../messages.h" |
||
| 47 | #include "../../modules.h" |
||
| 4589 | mejdrech | 48 | |
| 49 | #include "../../structures/dynamic_fifo.h" |
||
| 4499 | mejdrech | 50 | #include "../../structures/packet/packet_client.h" |
| 51 | |||
| 4579 | mejdrech | 52 | #include "../../include/in.h" |
| 4720 | mejdrech | 53 | #include "../../include/in6.h" |
| 4579 | mejdrech | 54 | #include "../../include/inet.h" |
| 4499 | mejdrech | 55 | #include "../../include/ip_client.h" |
| 56 | #include "../../include/ip_interface.h" |
||
| 57 | #include "../../include/ip_protocols.h" |
||
| 4707 | mejdrech | 58 | #include "../../include/icmp_client.h" |
| 59 | #include "../../include/icmp_interface.h" |
||
| 4713 | mejdrech | 60 | #include "../../include/socket_codes.h" |
| 4579 | mejdrech | 61 | #include "../../include/socket_errno.h" |
| 4499 | mejdrech | 62 | |
| 4579 | mejdrech | 63 | #include "../../socket/socket_core.h" |
| 64 | #include "../../socket/socket_messages.h" |
||
| 65 | |||
| 4499 | mejdrech | 66 | #include "../tl_messages.h" |
| 67 | |||
| 68 | #include "udp.h" |
||
| 4579 | mejdrech | 69 | #include "udp_header.h" |
| 4499 | mejdrech | 70 | #include "udp_module.h" |
| 71 | |||
| 4701 | mejdrech | 72 | /** Maximum UDP fragment size. |
| 73 | */ |
||
| 4589 | mejdrech | 74 | #define MAX_UDP_FRAGMENT_SIZE 65535 |
| 75 | |||
| 4701 | mejdrech | 76 | /** Free ports pool start. |
| 77 | */ |
||
| 4700 | mejdrech | 78 | #define UDP_FREE_PORTS_START 1025 |
| 4701 | mejdrech | 79 | |
| 80 | /** Free ports pool end. |
||
| 81 | */ |
||
| 4700 | mejdrech | 82 | #define UDP_FREE_PORTS_END 65535 |
| 83 | |||
| 4707 | mejdrech | 84 | /** Processes the received UDP packet queue. |
| 4701 | mejdrech | 85 | * Is used as an entry point from the underlying IP module. |
| 4707 | mejdrech | 86 | * Notifies the destination socket application. |
| 87 | * Releases the packet on error or send an ICMP error notification.. |
||
| 4701 | mejdrech | 88 | * @param device_id The device identifier. Ignored parameter. |
| 4707 | mejdrech | 89 | * @param packet The received packet queue. Input/output parameter. |
| 4701 | mejdrech | 90 | * @param receiver The target service. Ignored parameter. |
| 4707 | mejdrech | 91 | * @param error The packet error reporting service. Prefixes the received packet. Input parameter. |
| 4701 | mejdrech | 92 | * @returns EOK on success. |
| 93 | * @returns EINVAL if the packet is not valid. |
||
| 94 | * @returns EINVAL if the stored packet address is not the an_addr_t. |
||
| 95 | * @returns EINVAL if the packet does not contain any data. |
||
| 96 | * @returns NO_DATA if the packet content is shorter than the user datagram header. |
||
| 97 | * @returns ENOMEM if there is not enough memory left. |
||
| 98 | * @returns EADDRNOTAVAIL if the destination socket does not exist. |
||
| 99 | * @returns Other error codes as defined for the ip_client_process_packet() function. |
||
| 100 | */ |
||
| 4707 | mejdrech | 101 | int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error ); |
| 4701 | mejdrech | 102 | |
| 4707 | mejdrech | 103 | /** Releases the packet and returns the result. |
| 104 | * @param packet The packet queue to be released. Input parameter. |
||
| 105 | * @param result The result to be returned. Input parameter. |
||
| 106 | * @return The result parameter. |
||
| 107 | */ |
||
| 4713 | mejdrech | 108 | int udp_release_and_return( packet_t packet, int result ); |
| 4707 | mejdrech | 109 | |
| 110 | /** Sends the port unreachable ICMP notification. |
||
| 111 | * Sends the first packet and releases all the others. |
||
| 112 | * Releases the packet queu on error. |
||
| 113 | * @param packet The packet to be send. Input parameter. |
||
| 114 | * @param error The packet error reporting service. Prefixes the received packet. Input parameter. |
||
| 115 | */ |
||
| 116 | void udp_send_icmp_port_unreachable( packet_t packet, services_t error ); |
||
| 117 | |||
| 4701 | mejdrech | 118 | /** @name Socket messages processing functions |
| 119 | */ |
||
| 120 | /*@{*/ |
||
| 121 | |||
| 122 | /** Processes the socket client messages. |
||
| 123 | * Runs until the client module disconnects. |
||
| 124 | * @param callid The message identifier. Input parameter. |
||
| 125 | * @param call The message parameters. Input parameter. |
||
| 126 | * @returns EOK on success. |
||
| 127 | * @see socket.h |
||
| 128 | */ |
||
| 4713 | mejdrech | 129 | int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call ); |
| 4701 | mejdrech | 130 | |
| 131 | /** Sends data from the socket to the remote address. |
||
| 132 | * Binds the socket to a free port if not already connected/bound. |
||
| 133 | * Handles the NET_SOCKET_SENDTO message. |
||
| 4720 | mejdrech | 134 | * Supports AF_INET and AF_INET6 address families. |
| 4701 | mejdrech | 135 | * @param local_sockets The application local sockets. Input/output parameter. |
| 136 | * @param socket_id Socket identifier. Input parameter. |
||
| 137 | * @param addr The destination address. Input parameter. |
||
| 138 | * @param addrlen The address length. Input parameter. |
||
| 139 | * @param fragments The number of data fragments. Input parameter. |
||
| 140 | * @param flags Various send flags. Input parameter. |
||
| 141 | * @returns EOK on success. |
||
| 142 | * @returns EAFNOTSUPPORT if the address family is not supported. |
||
| 143 | * @returns ENOTSOCK if the socket is not found. |
||
| 144 | * @returns EINVAL if the address is invalid. |
||
| 145 | * @returns ENOTCONN if the sending socket is not and cannot be bound. |
||
| 146 | * @returns ENOMEM if there is not enough memory left. |
||
| 147 | * @returns Other error codes as defined for the socket_read_packet_data() function. |
||
| 148 | * @returns Other error codes as defined for the ip_client_prepare_packet() function. |
||
| 149 | * @returns Other error codes as defined for the ip_send_msg() function. |
||
| 150 | */ |
||
| 4720 | mejdrech | 151 | int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, int flags ); |
| 4701 | mejdrech | 152 | |
| 153 | /** Receives data to the socket. |
||
| 154 | * Handles the NET_SOCKET_RECVFROM message. |
||
| 4720 | mejdrech | 155 | * Replies the source address as well. |
| 4701 | mejdrech | 156 | * @param local_sockets The application local sockets. Input parameter. |
| 157 | * @param socket_id Socket identifier. Input parameter. |
||
| 158 | * @param flags Various receive flags. Input parameter. |
||
| 4720 | mejdrech | 159 | * @param addrlen The source address length. Output parameter. |
| 4701 | mejdrech | 160 | * @returns The number of bytes received. |
| 161 | * @returns ENOTSOCK if the socket is not found. |
||
| 162 | * @returns NO_DATA if there are no received packets or data. |
||
| 163 | * @returns ENOMEM if there is not enough memory left. |
||
| 164 | * @returns EINVAL if the received address is not an IP address. |
||
| 165 | * @returns Other error codes as defined for the packet_translate() function. |
||
| 166 | * @returns Other error codes as defined for the socket_write_data() function. |
||
| 167 | */ |
||
| 4720 | mejdrech | 168 | int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags, size_t * addrlen ); |
| 4701 | mejdrech | 169 | |
| 170 | /*@}*/ |
||
| 171 | |||
| 172 | /** Receives data from the socket. |
||
| 173 | * The received data buffer is allocated and returned. |
||
| 174 | * @param data The data buffer to be filled. Output parameter. |
||
| 175 | * @param length The buffer length. Output parameter. |
||
| 176 | * @returns EOK on success. |
||
| 177 | * @returns EBADMEM if the data or the length parameter is NULL. |
||
| 178 | * @returns EINVAL if the client does not send data. |
||
| 179 | * @returns ENOMEM if there is not enough memory left. |
||
| 180 | * @returns Other error codes as defined for the ipc_data_write_finalize() function. |
||
| 181 | */ |
||
| 4589 | mejdrech | 182 | int socket_read_data( void ** data, size_t * length ); |
| 4701 | mejdrech | 183 | |
| 184 | /** Receives data from the socket into a packet. |
||
| 185 | * @param packet The new created packet. Output parameter. |
||
| 186 | * @param prefix Reserved packet data prefix length. Input parameter. |
||
| 4720 | mejdrech | 187 | * @param addr The destination address. Input parameter. |
| 188 | * @param addrlen The address length. Input parameter. |
||
| 4701 | mejdrech | 189 | * @returns Number of bytes received. |
| 190 | * @returns EINVAL if the client does not send data. |
||
| 191 | * @returns ENOMEM if there is not enough memory left. |
||
| 192 | * @returns Other error codes as defined for the ipc_data_read_finalize() function. |
||
| 193 | */ |
||
| 4720 | mejdrech | 194 | int socket_read_packet_data( packet_ref packet, size_t prefix, const struct sockaddr * addr, socklen_t addrlen ); |
| 4701 | mejdrech | 195 | |
| 196 | /** Replies the data to the socket. |
||
| 197 | * @param data The data buffer to be sent. Input parameter. |
||
| 198 | * @param data_length The buffer length. Input parameter. |
||
| 199 | * @returns EOK on success. |
||
| 200 | * @returns EINVAL if the client does not expect all the data. |
||
| 201 | * @returns Other error codes as defined for the ipc_data_read_finalize() function. |
||
| 202 | */ |
||
| 4589 | mejdrech | 203 | int socket_write_data( void * data, size_t data_length ); |
| 4499 | mejdrech | 204 | |
| 4720 | mejdrech | 205 | /** Sets the address port. |
| 206 | * Supports AF_INET and AF_INET6 address families. |
||
| 207 | * @param addr The address to be updated. Input/output parameter. |
||
| 208 | * @param addrlen The address length. Input parameter. |
||
| 209 | * @param port The port to be set. Input parameter. |
||
| 210 | * @returns EOK on success. |
||
| 211 | * @returns EINVAL if the address length does not match the address family. |
||
| 212 | * @returns EAFNOSUPPORT if the address family is not supported. |
||
| 213 | */ |
||
| 214 | int udp_set_address_port( struct sockaddr * addr, int addrlen, uint16_t port ); |
||
| 215 | |||
| 4701 | mejdrech | 216 | /** UDP global data. |
| 217 | */ |
||
| 4499 | mejdrech | 218 | udp_globals_t udp_globals; |
| 219 | |||
| 220 | int udp_initialize( async_client_conn_t client_connection ){ |
||
| 4579 | mejdrech | 221 | ERROR_DECLARE; |
| 4499 | mejdrech | 222 | |
| 4700 | mejdrech | 223 | fibril_rwlock_initialize( & udp_globals.lock ); |
| 224 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 4707 | mejdrech | 225 | udp_globals.icmp_phone = icmp_connect_module( SERVICE_ICMP ); |
| 226 | if( udp_globals.icmp_phone < 0 ){ |
||
| 227 | return udp_globals.icmp_phone; |
||
| 228 | } |
||
| 4499 | mejdrech | 229 | udp_globals.ip_phone = ip_bind_service( SERVICE_IP, IPPROTO_UDP, SERVICE_UDP, client_connection, udp_received_msg ); |
| 4579 | mejdrech | 230 | if( udp_globals.ip_phone < 0 ){ |
| 231 | return udp_globals.ip_phone; |
||
| 232 | } |
||
| 233 | ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.addr_len, & udp_globals.prefix, & udp_globals.content, & udp_globals.suffix )); |
||
| 234 | ERROR_PROPAGATE( socket_ports_initialize( & udp_globals.sockets )); |
||
| 235 | udp_globals.prefix += sizeof( udp_header_t ); |
||
| 236 | udp_globals.content -= sizeof( udp_header_t ); |
||
| 4700 | mejdrech | 237 | udp_globals.last_used_port = UDP_FREE_PORTS_START - 1; |
| 238 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
||
| 4579 | mejdrech | 239 | return EOK; |
| 4499 | mejdrech | 240 | } |
| 241 | |||
| 4707 | mejdrech | 242 | int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error ){ |
| 4589 | mejdrech | 243 | ERROR_DECLARE; |
| 244 | |||
| 4708 | mejdrech | 245 | size_t length; |
| 246 | size_t offset; |
||
| 247 | int result; |
||
| 4707 | mejdrech | 248 | uint8_t * data; |
| 4589 | mejdrech | 249 | udp_header_ref header; |
| 250 | socket_core_ref * socket; |
||
| 251 | packet_t next_packet; |
||
| 4708 | mejdrech | 252 | size_t total_length; |
| 4589 | mejdrech | 253 | // uint16_t checksum; |
| 254 | int fragments; |
||
| 255 | packet_t tmp_packet; |
||
| 4707 | mejdrech | 256 | icmp_type_t type; |
| 257 | icmp_code_t code; |
||
| 4589 | mejdrech | 258 | |
| 4707 | mejdrech | 259 | if( error ){ |
| 260 | switch( error ){ |
||
| 261 | case SERVICE_ICMP: |
||
| 262 | // process error |
||
| 263 | // TODO remove debug dump |
||
| 264 | // length = icmp_client_header_length( packet ); |
||
| 4708 | mejdrech | 265 | result = icmp_client_process_packet( packet, & type, & code, NULL, NULL ); |
| 266 | if( result < 0 ){ |
||
| 4713 | mejdrech | 267 | return udp_release_and_return( packet, result ); |
| 4707 | mejdrech | 268 | } |
| 269 | printf( "ICMP error %d (%d) in packet %d\n", type, code, packet_get_id( packet ) ); |
||
| 4708 | mejdrech | 270 | length = ( size_t ) result; |
| 4707 | mejdrech | 271 | if( ERROR_OCCURRED( packet_trim( packet, length, 0 ))){ |
| 4713 | mejdrech | 272 | return udp_release_and_return( packet, ERROR_CODE ); |
| 4707 | mejdrech | 273 | } |
| 274 | break; |
||
| 275 | default: |
||
| 4713 | mejdrech | 276 | return udp_release_and_return( packet, ENOTSUP ); |
| 4707 | mejdrech | 277 | } |
| 278 | } |
||
| 4701 | mejdrech | 279 | // TODO process received ipopts? |
| 4708 | mejdrech | 280 | result = ip_client_process_packet( packet, NULL, NULL, NULL, NULL, NULL ); |
| 281 | if( result < 0 ){ |
||
| 4713 | mejdrech | 282 | return udp_release_and_return( packet, result ); |
| 4707 | mejdrech | 283 | } |
| 4708 | mejdrech | 284 | offset = ( size_t ) result; |
| 4589 | mejdrech | 285 | |
| 286 | length = packet_get_data_length( packet ); |
||
| 4707 | mejdrech | 287 | if( length <= 0 ){ |
| 4713 | mejdrech | 288 | return udp_release_and_return( packet, EINVAL ); |
| 4707 | mejdrech | 289 | } |
| 290 | if( length < sizeof( udp_header_t ) + offset ){ |
||
| 4713 | mejdrech | 291 | return udp_release_and_return( packet, NO_DATA ); |
| 4707 | mejdrech | 292 | } |
| 4499 | mejdrech | 293 | data = packet_get_data( packet ); |
| 4707 | mejdrech | 294 | if( ! data ){ |
| 4713 | mejdrech | 295 | return udp_release_and_return( packet, NO_DATA ); |
| 4707 | mejdrech | 296 | } |
| 4589 | mejdrech | 297 | // get udp header |
| 4707 | mejdrech | 298 | header = ( udp_header_ref )( data + offset ); |
| 4589 | mejdrech | 299 | // find the destination socket |
| 300 | socket = socket_ports_find( & udp_globals.sockets, ntohs( header->dest )); |
||
| 4707 | mejdrech | 301 | if( ! socket ){ |
| 302 | udp_send_icmp_port_unreachable( packet, error ); |
||
| 303 | return EADDRNOTAVAIL; |
||
| 304 | } |
||
| 305 | // trim after successful processing to be able to send an ICMP error message! |
||
| 306 | ERROR_PROPAGATE( packet_trim( packet, offset, 0 )); |
||
| 4589 | mejdrech | 307 | // count the received packet fragments |
| 308 | next_packet = packet; |
||
| 309 | fragments = 0; |
||
| 310 | total_length = ntohs( header->len ); |
||
| 311 | do{ |
||
| 312 | ++ fragments; |
||
| 313 | length = packet_get_data_length( packet ); |
||
| 4708 | mejdrech | 314 | if( length <= 0 ){ |
| 4713 | mejdrech | 315 | return udp_release_and_return( packet, NO_DATA ); |
| 4707 | mejdrech | 316 | } |
| 4589 | mejdrech | 317 | if( total_length < length ){ |
| 318 | // cut of the suffix if too long |
||
| 4707 | mejdrech | 319 | if( ERROR_OCCURRED( packet_trim( next_packet, 0, length - total_length ))){ |
| 4713 | mejdrech | 320 | return udp_release_and_return( packet, ERROR_CODE ); |
| 4707 | mejdrech | 321 | } |
| 4589 | mejdrech | 322 | // relese the rest of the packet fragments |
| 323 | tmp_packet = pq_next( next_packet ); |
||
| 324 | while( tmp_packet ){ |
||
| 325 | next_packet = pq_detach( tmp_packet ); |
||
| 326 | pq_release( udp_globals.net_phone, packet_get_id( tmp_packet )); |
||
| 327 | tmp_packet = next_packet; |
||
| 328 | } |
||
| 329 | break; |
||
| 330 | } |
||
| 331 | total_length -= length; |
||
| 332 | /* if( header->header_checksum ){ |
||
| 333 | } |
||
| 334 | */ |
||
| 335 | }while(( next_packet = pq_next( next_packet )) && ( total_length > 0 )); |
||
| 336 | // queue the received packet |
||
| 4707 | mejdrech | 337 | if( ERROR_OCCURRED( dyn_fifo_push( &( ** socket ).received, packet_get_id( packet ), SOCKET_MAX_RECEIVED_SIZE ))){ |
| 4713 | mejdrech | 338 | return udp_release_and_return( packet, ERROR_CODE ); |
| 4707 | mejdrech | 339 | } |
| 4499 | mejdrech | 340 | |
| 4589 | mejdrech | 341 | // notify the destination socket |
| 4708 | mejdrech | 342 | async_msg_2(( ** socket ).phone, NET_SOCKET_RECEIVED, ( ipcarg_t ) ( ** socket ).socket_id, ( ipcarg_t ) fragments ); |
| 4499 | mejdrech | 343 | return EOK; |
| 344 | } |
||
| 345 | |||
| 346 | int udp_message( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ){ |
||
| 4505 | mejdrech | 347 | ERROR_DECLARE; |
| 348 | |||
| 349 | packet_t packet; |
||
| 350 | |||
| 4499 | mejdrech | 351 | * answer_count = 0; |
| 352 | switch( IPC_GET_METHOD( * call )){ |
||
| 353 | case NET_TL_RECEIVED: |
||
| 4700 | mejdrech | 354 | fibril_rwlock_read_lock( & udp_globals.lock ); |
| 355 | if( ! ERROR_OCCURRED( packet_translate( udp_globals.net_phone, & packet, IPC_GET_PACKET( call )))){ |
||
| 4707 | mejdrech | 356 | ERROR_CODE = udp_received_msg( IPC_GET_DEVICE( call ), packet, SERVICE_UDP, IPC_GET_ERROR( call )); |
| 4700 | mejdrech | 357 | } |
| 358 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
||
| 359 | return ERROR_CODE; |
||
| 4579 | mejdrech | 360 | case IPC_M_CONNECT_TO_ME: |
| 4713 | mejdrech | 361 | return udp_process_client_messages( callid, * call ); |
| 4499 | mejdrech | 362 | } |
| 363 | return ENOTSUP; |
||
| 364 | } |
||
| 365 | |||
| 4713 | mejdrech | 366 | int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call ){ |
| 4579 | mejdrech | 367 | int res; |
| 368 | bool keep_on_going = true; |
||
| 369 | socket_cores_t local_sockets; |
||
| 4701 | mejdrech | 370 | int app_phone = IPC_GET_PHONE( & call ); |
| 4720 | mejdrech | 371 | struct sockaddr * addr; |
| 4579 | mejdrech | 372 | size_t addrlen; |
| 4700 | mejdrech | 373 | fibril_rwlock_t lock; |
| 4701 | mejdrech | 374 | ipc_call_t answer; |
| 375 | int answer_count; |
||
| 4579 | mejdrech | 376 | |
| 377 | /* |
||
| 378 | * Accept the connection |
||
| 379 | * - Answer the first IPC_M_CONNECT_ME_TO call. |
||
| 380 | */ |
||
| 381 | ipc_answer_0( callid, EOK ); |
||
| 382 | |||
| 383 | socket_cores_initialize( & local_sockets ); |
||
| 4700 | mejdrech | 384 | fibril_rwlock_initialize( & lock ); |
| 4579 | mejdrech | 385 | |
| 386 | while( keep_on_going ){ |
||
| 387 | // refresh data |
||
| 4713 | mejdrech | 388 | refresh_answer( & answer, & answer_count ); |
| 4579 | mejdrech | 389 | |
| 4701 | mejdrech | 390 | callid = async_get_call( & call ); |
| 4589 | mejdrech | 391 | // printf( "message %d\n", IPC_GET_METHOD( * call )); |
| 4579 | mejdrech | 392 | |
| 4701 | mejdrech | 393 | switch( IPC_GET_METHOD( call )){ |
| 4579 | mejdrech | 394 | case IPC_M_PHONE_HUNGUP: |
| 395 | keep_on_going = false; |
||
| 396 | res = EOK; |
||
| 397 | break; |
||
| 398 | case NET_SOCKET: |
||
| 4700 | mejdrech | 399 | fibril_rwlock_write_lock( & lock ); |
| 4589 | mejdrech | 400 | res = socket_create( & local_sockets, app_phone, SOCKET_SET_SOCKET_ID( answer )); |
| 4700 | mejdrech | 401 | fibril_rwlock_write_unlock( & lock ); |
| 4589 | mejdrech | 402 | * SOCKET_SET_HEADER_SIZE( answer ) = sizeof( udp_header_t ); |
| 403 | * SOCKET_SET_DATA_FRAGMENT_SIZE( answer ) = MAX_UDP_FRAGMENT_SIZE; |
||
| 4701 | mejdrech | 404 | answer_count = 3; |
| 4579 | mejdrech | 405 | break; |
| 406 | case NET_SOCKET_BIND: |
||
| 4720 | mejdrech | 407 | res = socket_read_data(( void ** ) & addr, & addrlen ); |
| 4713 | mejdrech | 408 | if( res == EOK ){ |
| 409 | fibril_rwlock_write_lock( & lock ); |
||
| 410 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 411 | res = socket_bind( & local_sockets, & udp_globals.sockets, SOCKET_GET_SOCKET_ID( call ), addr, addrlen, UDP_FREE_PORTS_START, UDP_FREE_PORTS_END, udp_globals.last_used_port ); |
||
| 412 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
||
| 413 | fibril_rwlock_write_unlock( & lock ); |
||
| 414 | free( addr ); |
||
| 4579 | mejdrech | 415 | } |
| 4713 | mejdrech | 416 | break; |
| 4579 | mejdrech | 417 | case NET_SOCKET_SENDTO: |
| 4720 | mejdrech | 418 | res = socket_read_data(( void ** ) & addr, & addrlen ); |
| 4713 | mejdrech | 419 | if( res == EOK ){ |
| 420 | fibril_rwlock_read_lock( & lock ); |
||
| 421 | fibril_rwlock_read_lock( & udp_globals.lock ); |
||
| 422 | res = udp_sendto_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), addr, addrlen, SOCKET_GET_DATA_FRAGMENTS( call ), SOCKET_GET_FLAGS( call )); |
||
| 423 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
||
| 424 | fibril_rwlock_read_unlock( & lock ); |
||
| 425 | free( addr ); |
||
| 4579 | mejdrech | 426 | } |
| 427 | break; |
||
| 428 | case NET_SOCKET_RECVFROM: |
||
| 4700 | mejdrech | 429 | fibril_rwlock_read_lock( & lock ); |
| 430 | fibril_rwlock_read_lock( & udp_globals.lock ); |
||
| 4720 | mejdrech | 431 | res = udp_recvfrom_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), SOCKET_GET_FLAGS( call ), & addrlen ); |
| 4700 | mejdrech | 432 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
| 433 | fibril_rwlock_read_unlock( & lock ); |
||
| 4589 | mejdrech | 434 | if( res > 0 ){ |
| 435 | * SOCKET_SET_READ_DATA_LENGTH( answer ) = res; |
||
| 4720 | mejdrech | 436 | * SOCKET_SET_ADDRESS_LENGTH( answer ) = addrlen; |
| 4701 | mejdrech | 437 | answer_count = 2; |
| 4589 | mejdrech | 438 | res = EOK; |
| 439 | } |
||
| 4579 | mejdrech | 440 | break; |
| 441 | case NET_SOCKET_CLOSE: |
||
| 4700 | mejdrech | 442 | fibril_rwlock_write_lock( & lock ); |
| 443 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 4579 | mejdrech | 444 | res = socket_destroy( udp_globals.net_phone, SOCKET_GET_SOCKET_ID( call ), & local_sockets, & udp_globals.sockets ); |
| 4700 | mejdrech | 445 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
| 446 | fibril_rwlock_write_unlock( & lock ); |
||
| 4579 | mejdrech | 447 | break; |
| 448 | case NET_SOCKET_GETSOCKOPT: |
||
| 449 | case NET_SOCKET_SETSOCKOPT: |
||
| 450 | default: |
||
| 451 | res = ENOTSUP; |
||
| 452 | break; |
||
| 453 | } |
||
| 454 | |||
| 4589 | mejdrech | 455 | // printf( "res = %d\n", res ); |
| 4579 | mejdrech | 456 | |
| 4713 | mejdrech | 457 | answer_call( callid, res, & answer, answer_count ); |
| 4579 | mejdrech | 458 | } |
| 459 | |||
| 460 | socket_cores_destroy( & local_sockets ); |
||
| 461 | |||
| 462 | return EOK; |
||
| 463 | } |
||
| 464 | |||
| 4720 | mejdrech | 465 | int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, int flags ){ |
| 4579 | mejdrech | 466 | ERROR_DECLARE; |
| 467 | |||
| 468 | socket_core_ref socket; |
||
| 469 | struct sockaddr_in * address_in; |
||
| 4720 | mejdrech | 470 | struct sockaddr_in6 * address_in6; |
| 4579 | mejdrech | 471 | packet_t packet; |
| 4589 | mejdrech | 472 | packet_t next_packet; |
| 4579 | mejdrech | 473 | udp_header_ref header; |
| 4589 | mejdrech | 474 | int index; |
| 4708 | mejdrech | 475 | size_t total_length; |
| 476 | int result; |
||
| 4720 | mejdrech | 477 | uint16_t dest_port; |
| 4579 | mejdrech | 478 | |
| 479 | if( addrlen < sizeof( struct sockaddr )) return EINVAL; |
||
| 4720 | mejdrech | 480 | switch( addr->sa_family ){ |
| 4579 | mejdrech | 481 | case AF_INET: |
| 482 | if( addrlen != sizeof( struct sockaddr_in )) return EINVAL; |
||
| 483 | address_in = ( struct sockaddr_in * ) addr; |
||
| 4720 | mejdrech | 484 | dest_port = address_in->sin_port; |
| 485 | break; |
||
| 486 | case AF_INET6: |
||
| 487 | if( addrlen != sizeof( struct sockaddr_in6 )) return EINVAL; |
||
| 488 | address_in6 = ( struct sockaddr_in6 * ) addr; |
||
| 489 | dest_port = address_in6->sin6_port; |
||
| 490 | break; |
||
| 491 | default: |
||
| 492 | return EAFNOSUPPORT; |
||
| 493 | } |
||
| 4589 | mejdrech | 494 | |
| 4720 | mejdrech | 495 | socket = socket_cores_find( local_sockets, socket_id ); |
| 496 | if( ! socket ) return ENOTSOCK; |
||
| 4589 | mejdrech | 497 | |
| 4720 | mejdrech | 498 | // bind the socket to a random free port if not bound |
| 499 | while( socket->port <= 0 ){ |
||
| 500 | // try to find a free port |
||
| 501 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
||
| 502 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 503 | if( socket->port <= 0 ){ |
||
| 504 | ERROR_PROPAGATE( socket_bind_free_port( & udp_globals.sockets, socket, UDP_FREE_PORTS_START, UDP_FREE_PORTS_END, udp_globals.last_used_port )); |
||
| 505 | // set the next port as the search starting port number |
||
| 506 | udp_globals.last_used_port = socket->port; |
||
| 507 | } |
||
| 508 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
||
| 509 | fibril_rwlock_read_lock( & udp_globals.lock ); |
||
| 4579 | mejdrech | 510 | } |
| 4720 | mejdrech | 511 | |
| 512 | // TODO do not ask all the time |
||
| 513 | ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.addr_len, & udp_globals.prefix, & udp_globals.content, & udp_globals.suffix )); |
||
| 514 | |||
| 515 | // read the first packet fragment |
||
| 516 | result = socket_read_packet_data( & packet, sizeof( udp_header_t ), addr, addrlen ); |
||
| 517 | if( result < 0 ) return result; |
||
| 518 | total_length = ( size_t ) result; |
||
| 519 | // prefix the udp header |
||
| 520 | header = PACKET_PREFIX( packet, udp_header_t ); |
||
| 521 | if( ! header ){ |
||
| 522 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 523 | return ENOMEM; |
||
| 524 | } |
||
| 525 | // read the rest of the packet fragments |
||
| 526 | for( index = 1; index < fragments; ++ index ){ |
||
| 527 | result = socket_read_packet_data( & next_packet, 0, addr, addrlen ); |
||
| 528 | if( result < 0 ){ |
||
| 529 | return udp_release_and_return( packet, result ); |
||
| 530 | } |
||
| 531 | packet = pq_add( packet, next_packet, index, 0 ); |
||
| 532 | total_length += ( size_t ) result; |
||
| 533 | } |
||
| 534 | // set the udp header |
||
| 535 | header->source = htons( socket->port ); |
||
| 536 | header->dest = htons( dest_port ); |
||
| 537 | header->len = htons( total_length + sizeof( udp_header_t )); |
||
| 538 | // TODO my ip address for the pseudo header checksum |
||
| 539 | header->check = 0; |
||
| 540 | // prepare the first packet fragment |
||
| 541 | if( ERROR_OCCURRED( ip_client_prepare_packet( packet, IPPROTO_UDP, 0, 0, 0, 0 ))){ |
||
| 542 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 543 | return ERROR_CODE; |
||
| 544 | } |
||
| 545 | // send the packet |
||
| 546 | return ip_send_msg( udp_globals.ip_phone, socket->device_id, packet, SERVICE_UDP, 0 ); |
||
| 4579 | mejdrech | 547 | } |
| 548 | |||
| 4720 | mejdrech | 549 | int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags, size_t * addrlen ){ |
| 4579 | mejdrech | 550 | ERROR_DECLARE; |
| 551 | |||
| 4589 | mejdrech | 552 | socket_core_ref socket; |
| 553 | int packet_id; |
||
| 554 | packet_t packet; |
||
| 555 | udp_header_ref header; |
||
| 4720 | mejdrech | 556 | struct sockaddr * addr; |
| 4708 | mejdrech | 557 | size_t length; |
| 4589 | mejdrech | 558 | packet_t next_packet; |
| 4708 | mejdrech | 559 | uint8_t * data; |
| 560 | size_t fragments; |
||
| 561 | size_t * lengths; |
||
| 4720 | mejdrech | 562 | size_t index; |
| 4708 | mejdrech | 563 | int result; |
| 4589 | mejdrech | 564 | |
| 565 | // find the socket |
||
| 566 | socket = socket_cores_find( local_sockets, socket_id ); |
||
| 567 | if( ! socket ) return ENOTSOCK; |
||
| 568 | // get the next received packet |
||
| 569 | packet_id = dyn_fifo_value( & socket->received ); |
||
| 4701 | mejdrech | 570 | if( packet_id < 0 ) return NO_DATA; |
| 4589 | mejdrech | 571 | ERROR_PROPAGATE( packet_translate( udp_globals.net_phone, & packet, packet_id )); |
| 572 | // get udp header |
||
| 573 | data = packet_get_data( packet ); |
||
| 574 | if( ! data ){ |
||
| 575 | pq_release( udp_globals.net_phone, packet_id ); |
||
| 576 | return NO_DATA; |
||
| 577 | } |
||
| 578 | header = ( udp_header_ref ) data; |
||
| 4720 | mejdrech | 579 | |
| 580 | // set the source address port |
||
| 581 | result = packet_get_addr( packet, ( uint8_t ** ) & addr, NULL ); |
||
| 582 | if( ERROR_OCCURRED( udp_set_address_port( addr, result, ntohs( header->source )))){ |
||
| 4589 | mejdrech | 583 | pq_release( udp_globals.net_phone, packet_id ); |
| 4720 | mejdrech | 584 | return ERROR_CODE; |
| 4589 | mejdrech | 585 | } |
| 4720 | mejdrech | 586 | * addrlen = ( size_t ) result; |
| 4589 | mejdrech | 587 | // send the source address |
| 4720 | mejdrech | 588 | ERROR_PROPAGATE( socket_write_data( addr, * addrlen )); |
| 589 | |||
| 4589 | mejdrech | 590 | next_packet = pq_next( packet ); |
| 591 | if( ! next_packet ){ |
||
| 592 | // write all if only one fragment |
||
| 593 | ERROR_PROPAGATE( socket_write_data( data + sizeof( udp_header_t ), packet_get_data_length( packet ) - sizeof( udp_header_t ))); |
||
| 594 | // store the total length |
||
| 595 | length = packet_get_data_length( packet ) - sizeof( udp_header_t ); |
||
| 596 | }else{ |
||
| 597 | // count the packet fragments |
||
| 598 | fragments = 1; |
||
| 599 | next_packet = pq_next( packet ); |
||
| 600 | while(( next_packet = pq_next( next_packet ))){ |
||
| 601 | ++ fragments; |
||
| 602 | } |
||
| 603 | // compute and store the fragment lengths |
||
| 4708 | mejdrech | 604 | lengths = ( size_t * ) malloc( sizeof( size_t ) * fragments + sizeof( size_t )); |
| 4589 | mejdrech | 605 | if( ! lengths ) return ENOMEM; |
| 606 | lengths[ 0 ] = packet_get_data_length( packet ) - sizeof( udp_header_t ); |
||
| 607 | lengths[ fragments ] = lengths[ 0 ]; |
||
| 608 | next_packet = pq_next( packet ); |
||
| 609 | for( index = 1; index < fragments; ++ index ){ |
||
| 610 | lengths[ index ] = packet_get_data_length( next_packet ); |
||
| 611 | lengths[ fragments ] += lengths[ index ]; |
||
| 612 | next_packet = pq_next( packet ); |
||
| 613 | }while( next_packet ); |
||
| 614 | // write the fragment lengths |
||
| 615 | ERROR_PROPAGATE( socket_write_data( lengths, sizeof( int ) * ( fragments + 1 ))); |
||
| 616 | // write the first fragment |
||
| 617 | ERROR_PROPAGATE( socket_write_data( data + sizeof( udp_header_t ), lengths[ 0 ] )); |
||
| 618 | next_packet = pq_next( packet ); |
||
| 619 | // write the rest of the fragments |
||
| 620 | for( index = 1; index < fragments; ++ index ){ |
||
| 621 | ERROR_PROPAGATE( socket_write_data( packet_get_data( next_packet ), lengths[ index ] )); |
||
| 622 | next_packet = pq_next( packet ); |
||
| 623 | }while( next_packet ); |
||
| 624 | // store the total length |
||
| 625 | length = lengths[ fragments ]; |
||
| 626 | free( lengths ); |
||
| 627 | } |
||
| 628 | // release the packet |
||
| 629 | dyn_fifo_pop( & socket->received ); |
||
| 630 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 631 | // return the total length |
||
| 4708 | mejdrech | 632 | return ( int ) length; |
| 4589 | mejdrech | 633 | } |
| 634 | |||
| 635 | int socket_write_data( void * data, size_t data_length ){ |
||
| 636 | size_t length; |
||
| 4579 | mejdrech | 637 | ipc_callid_t callid; |
| 638 | |||
| 4589 | mejdrech | 639 | if(( ! ipc_data_read_receive( & callid, & length )) |
| 640 | || ( length < data_length )){ |
||
| 641 | return EINVAL; |
||
| 642 | } |
||
| 643 | return ipc_data_read_finalize( callid, data, data_length ); |
||
| 644 | } |
||
| 645 | |||
| 646 | int socket_read_data( void ** data, size_t * length ){ |
||
| 647 | ERROR_DECLARE; |
||
| 648 | |||
| 649 | ipc_callid_t callid; |
||
| 650 | |||
| 4579 | mejdrech | 651 | if( !( data && length )) return EBADMEM; |
| 652 | if( ! ipc_data_write_receive( & callid, length )) return EINVAL; |
||
| 653 | * data = malloc( * length ); |
||
| 4701 | mejdrech | 654 | if( !( * data )) return ENOMEM; |
| 4579 | mejdrech | 655 | if( ERROR_OCCURRED( ipc_data_write_finalize( callid, * data, * length ))){ |
| 656 | free( data ); |
||
| 657 | return ERROR_CODE; |
||
| 658 | } |
||
| 659 | return EOK; |
||
| 660 | } |
||
| 661 | |||
| 4720 | mejdrech | 662 | int socket_read_packet_data( packet_ref packet, size_t prefix, const struct sockaddr * addr, socklen_t addrlen ){ |
| 4589 | mejdrech | 663 | ERROR_DECLARE; |
| 664 | |||
| 665 | ipc_callid_t callid; |
||
| 666 | size_t length; |
||
| 667 | void * data; |
||
| 668 | |||
| 669 | // get the data length |
||
| 670 | if( ! ipc_data_write_receive( & callid, & length )) return EINVAL; |
||
| 671 | // get a new packet |
||
| 672 | * packet = packet_get_4( udp_globals.net_phone, length, udp_globals.addr_len, prefix + udp_globals.prefix, udp_globals.suffix ); |
||
| 673 | if( ! packet ) return ENOMEM; |
||
| 674 | // allocate space in the packet |
||
| 675 | data = packet_suffix( * packet, length ); |
||
| 676 | if( ! data ){ |
||
| 677 | pq_release( udp_globals.net_phone, packet_get_id( * packet )); |
||
| 678 | return ENOMEM; |
||
| 679 | } |
||
| 680 | // read the data into the packet |
||
| 681 | if( ERROR_OCCURRED( ipc_data_write_finalize( callid, data, length )) |
||
| 682 | // set the packet destination address |
||
| 4720 | mejdrech | 683 | || ERROR_OCCURRED( packet_set_addr( * packet, NULL, ( uint8_t * ) addr, addrlen ))){ |
| 4589 | mejdrech | 684 | pq_release( udp_globals.net_phone, packet_get_id( * packet )); |
| 685 | return ERROR_CODE; |
||
| 686 | } |
||
| 4708 | mejdrech | 687 | return ( int ) length; |
| 4589 | mejdrech | 688 | } |
| 689 | |||
| 4713 | mejdrech | 690 | int udp_release_and_return( packet_t packet, int result ){ |
| 4707 | mejdrech | 691 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
| 692 | return result; |
||
| 693 | } |
||
| 694 | |||
| 695 | void udp_send_icmp_port_unreachable( packet_t packet, services_t error ){ |
||
| 696 | packet_t next; |
||
| 697 | uint8_t * src; |
||
| 698 | int length; |
||
| 699 | |||
| 700 | // detach the first packet and release the others |
||
| 701 | next = pq_detach( packet ); |
||
| 702 | if( next ){ |
||
| 703 | pq_release( udp_globals.net_phone, packet_get_id( next )); |
||
| 704 | } |
||
| 705 | length = packet_get_addr( packet, & src, NULL ); |
||
| 706 | if(( length > 0 ) |
||
| 707 | && ( ! error ) |
||
| 708 | && ( udp_globals.icmp_phone >= 0 ) |
||
| 709 | // set both addresses to the source one (avoids the source address deletion before setting the destination one) |
||
| 4708 | mejdrech | 710 | && ( packet_set_addr( packet, src, src, ( size_t ) length ) == EOK )){ |
| 4707 | mejdrech | 711 | icmp_destination_unreachable_msg( udp_globals.icmp_phone, ICMP_PORT_UNREACH, 0, packet ); |
| 712 | }else{ |
||
| 4713 | mejdrech | 713 | udp_release_and_return( packet, EINVAL ); |
| 4707 | mejdrech | 714 | } |
| 715 | } |
||
| 716 | |||
| 4720 | mejdrech | 717 | int udp_set_address_port( struct sockaddr * addr, int addrlen, uint16_t port ){ |
| 718 | struct sockaddr_in * address_in; |
||
| 719 | struct sockaddr_in6 * address_in6; |
||
| 720 | size_t length; |
||
| 721 | |||
| 722 | if( addrlen < 0 ) return EINVAL; |
||
| 723 | length = ( size_t ) addrlen; |
||
| 724 | if( length < sizeof( struct sockaddr )) return EINVAL; |
||
| 725 | switch( addr->sa_family ){ |
||
| 726 | case AF_INET: |
||
| 727 | if( length != sizeof( struct sockaddr_in )) return EINVAL; |
||
| 728 | address_in = ( struct sockaddr_in * ) addr; |
||
| 729 | address_in->sin_port = port; |
||
| 730 | return EOK; |
||
| 731 | case AF_INET6: |
||
| 732 | if( length != sizeof( struct sockaddr_in6 )) return EINVAL; |
||
| 733 | address_in6 = ( struct sockaddr_in6 * ) addr; |
||
| 734 | address_in6->sin6_port = port; |
||
| 735 | return EOK; |
||
| 736 | default: |
||
| 737 | return EAFNOSUPPORT; |
||
| 738 | } |
||
| 739 | } |
||
| 740 | |||
| 4499 | mejdrech | 741 | /** @} |
| 742 | */ |