Rev 4603 | Rev 4701 | 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 |
||
| 34 | */ |
||
| 35 | |||
| 36 | #include <async.h> |
||
| 4700 | mejdrech | 37 | #include <fibril_sync.h> |
| 4579 | mejdrech | 38 | #include <malloc.h> |
| 4499 | mejdrech | 39 | #include <stdio.h> |
| 40 | |||
| 41 | #include <ipc/ipc.h> |
||
| 42 | #include <ipc/services.h> |
||
| 43 | |||
| 44 | #include "../../err.h" |
||
| 45 | #include "../../messages.h" |
||
| 46 | #include "../../modules.h" |
||
| 4589 | mejdrech | 47 | |
| 48 | #include "../../structures/dynamic_fifo.h" |
||
| 4499 | mejdrech | 49 | #include "../../structures/packet/packet_client.h" |
| 50 | |||
| 4579 | mejdrech | 51 | #include "../../include/in.h" |
| 52 | #include "../../include/inet.h" |
||
| 4499 | mejdrech | 53 | #include "../../include/ip_client.h" |
| 54 | #include "../../include/ip_interface.h" |
||
| 55 | #include "../../include/ip_protocols.h" |
||
| 4579 | mejdrech | 56 | #include "../../include/socket.h" |
| 57 | #include "../../include/socket_errno.h" |
||
| 4499 | mejdrech | 58 | |
| 4579 | mejdrech | 59 | #include "../../socket/socket_core.h" |
| 60 | #include "../../socket/socket_messages.h" |
||
| 61 | |||
| 4499 | mejdrech | 62 | #include "../tl_messages.h" |
| 63 | |||
| 64 | #include "udp.h" |
||
| 4579 | mejdrech | 65 | #include "udp_header.h" |
| 4499 | mejdrech | 66 | #include "udp_module.h" |
| 67 | |||
| 4589 | mejdrech | 68 | #define MAX_UDP_FRAGMENT_SIZE 65535 |
| 69 | |||
| 4700 | mejdrech | 70 | #define UDP_FREE_PORTS_START 1025 |
| 71 | #define UDP_FREE_PORTS_END 65535 |
||
| 72 | |||
| 4499 | mejdrech | 73 | int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver ); |
| 4589 | mejdrech | 74 | int udp_process_packet( packet_t packet ); |
| 4579 | mejdrech | 75 | int process_client_messages( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ); |
| 4589 | mejdrech | 76 | int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, void * addr, size_t addrlen, int fragments, int flags ); |
| 77 | int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags ); |
||
| 78 | int socket_read_data( void ** data, size_t * length ); |
||
| 79 | int socket_read_packet_data( packet_ref packet, size_t prefix, struct sockaddr_in * address_in ); |
||
| 80 | int socket_write_data( void * data, size_t data_length ); |
||
| 4499 | mejdrech | 81 | |
| 82 | udp_globals_t udp_globals; |
||
| 83 | |||
| 84 | /** Initializes the module. |
||
| 85 | */ |
||
| 86 | int udp_initialize( async_client_conn_t client_connection ){ |
||
| 4579 | mejdrech | 87 | ERROR_DECLARE; |
| 4499 | mejdrech | 88 | |
| 4700 | mejdrech | 89 | fibril_rwlock_initialize( & udp_globals.lock ); |
| 90 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 4499 | mejdrech | 91 | udp_globals.ip_phone = ip_bind_service( SERVICE_IP, IPPROTO_UDP, SERVICE_UDP, client_connection, udp_received_msg ); |
| 4579 | mejdrech | 92 | if( udp_globals.ip_phone < 0 ){ |
| 93 | return udp_globals.ip_phone; |
||
| 94 | } |
||
| 95 | ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.addr_len, & udp_globals.prefix, & udp_globals.content, & udp_globals.suffix )); |
||
| 96 | ERROR_PROPAGATE( socket_ports_initialize( & udp_globals.sockets )); |
||
| 97 | udp_globals.prefix += sizeof( udp_header_t ); |
||
| 98 | udp_globals.content -= sizeof( udp_header_t ); |
||
| 4700 | mejdrech | 99 | udp_globals.last_used_port = UDP_FREE_PORTS_START - 1; |
| 100 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
||
| 4579 | mejdrech | 101 | return EOK; |
| 4499 | mejdrech | 102 | } |
| 103 | |||
| 104 | int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver ){ |
||
| 4589 | mejdrech | 105 | ERROR_DECLARE; |
| 106 | |||
| 107 | if( ERROR_OCCURRED( udp_process_packet( packet ))){ |
||
| 108 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 109 | return ERROR_CODE; |
||
| 110 | } |
||
| 111 | |||
| 112 | return EOK; |
||
| 113 | } |
||
| 114 | |||
| 115 | int udp_process_packet( packet_t packet ){ |
||
| 116 | ERROR_DECLARE; |
||
| 117 | |||
| 118 | uint8_t * src; |
||
| 119 | uint8_t * dest; |
||
| 120 | int length; |
||
| 121 | void * data; |
||
| 122 | udp_header_ref header; |
||
| 123 | socket_core_ref * socket; |
||
| 124 | packet_t next_packet; |
||
| 125 | int total_length; |
||
| 126 | // uint16_t checksum; |
||
| 127 | int fragments; |
||
| 128 | packet_t tmp_packet; |
||
| 129 | |||
| 130 | // get packet data |
||
| 131 | length = packet_get_addr( packet, & src, & dest ); |
||
| 132 | if( length < 0 ) return length; |
||
| 133 | if( length != sizeof( in_addr_t )) return EINVAL; |
||
| 134 | // TODO received ipopts? |
||
| 135 | ERROR_PROPAGATE( ip_client_process_packet( packet, NULL, NULL, NULL, NULL, NULL )); |
||
| 136 | |||
| 4499 | mejdrech | 137 | // TODO remove debug dump: |
| 4589 | mejdrech | 138 | /* uint8_t * rdata; |
| 139 | rdata = packet_get_data( packet ); |
||
| 140 | printf( "Receiving udp packet:\n\tid\t= %d\n\tlength\t= %d\n\tdata\t= %.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX\n\t\t%.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX\n", packet_get_id( packet ), packet_get_data_length( packet ), rdata[ 0 ], rdata[ 1 ], rdata[ 2 ], rdata[ 3 ], rdata[ 4 ], rdata[ 5 ], rdata[ 6 ], rdata[ 7 ], rdata[ 8 ], rdata[ 9 ], rdata[ 10 ], rdata[ 11 ], rdata[ 12 ], rdata[ 13 ], rdata[ 14 ], rdata[ 15 ], rdata[ 16 ], rdata[ 17 ], rdata[ 18 ], rdata[ 19 ], rdata[ 20 ], rdata[ 21 ], rdata[ 22 ], rdata[ 23 ], rdata[ 24 ], rdata[ 25 ], rdata[ 26 ], rdata[ 27 ], rdata[ 28 ], rdata[ 29 ], rdata[ 30 ], rdata[ 31 ], rdata[ 32 ], rdata[ 33 ], rdata[ 34 ], rdata[ 35 ], rdata[ 36 ], rdata[ 37 ], rdata[ 38 ], rdata[ 39 ], rdata[ 40 ], rdata[ 41 ], rdata[ 42 ], rdata[ 43 ], rdata[ 44 ], rdata[ 45 ], rdata[ 46 ], rdata[ 47 ], rdata[ 48 ], rdata[ 49 ], rdata[ 50 ], rdata[ 51 ], rdata[ 52 ], rdata[ 53 ], rdata[ 54 ], rdata[ 55 ], rdata[ 56 ], rdata[ 57 ], rdata[ 58 ], rdata[ 59 ] ); |
||
| 141 | */ |
||
| 142 | length = packet_get_data_length( packet ); |
||
| 143 | if( length < 0 ) return length; |
||
| 144 | if( length < sizeof( udp_header_t )) return NO_DATA; |
||
| 4499 | mejdrech | 145 | data = packet_get_data( packet ); |
| 4589 | mejdrech | 146 | if( ! data ) return NO_DATA; |
| 147 | // get udp header |
||
| 148 | header = ( udp_header_ref ) data; |
||
| 149 | // find the destination socket |
||
| 150 | socket = socket_ports_find( & udp_globals.sockets, ntohs( header->dest )); |
||
| 151 | if( ! socket ) return EADDRNOTAVAIL; |
||
| 152 | // count the received packet fragments |
||
| 153 | next_packet = packet; |
||
| 154 | fragments = 0; |
||
| 155 | total_length = ntohs( header->len ); |
||
| 156 | do{ |
||
| 157 | ++ fragments; |
||
| 158 | length = packet_get_data_length( packet ); |
||
| 159 | if( length < 0 ) return length; |
||
| 160 | if( ! length ) return NO_DATA; |
||
| 161 | if( total_length < length ){ |
||
| 162 | // cut of the suffix if too long |
||
| 163 | ERROR_PROPAGATE( packet_trim( next_packet, 0, length - total_length )); |
||
| 164 | // relese the rest of the packet fragments |
||
| 165 | tmp_packet = pq_next( next_packet ); |
||
| 166 | while( tmp_packet ){ |
||
| 167 | next_packet = pq_detach( tmp_packet ); |
||
| 168 | pq_release( udp_globals.net_phone, packet_get_id( tmp_packet )); |
||
| 169 | tmp_packet = next_packet; |
||
| 170 | } |
||
| 171 | break; |
||
| 172 | } |
||
| 173 | total_length -= length; |
||
| 174 | /* if( header->header_checksum ){ |
||
| 175 | } |
||
| 176 | */ |
||
| 177 | }while(( next_packet = pq_next( next_packet )) && ( total_length > 0 )); |
||
| 178 | // queue the received packet |
||
| 179 | ERROR_PROPAGATE( dyn_fifo_push( &( ** socket ).received, packet_get_id( packet ), SOCKET_MAX_RECEIVED_SIZE )); |
||
| 4499 | mejdrech | 180 | |
| 4589 | mejdrech | 181 | // notify the destination socket |
| 182 | async_msg_2(( ** socket ).phone, NET_SOCKET_RECEIVED, ( ** socket ).socket_id, fragments ); |
||
| 4499 | mejdrech | 183 | return EOK; |
| 184 | } |
||
| 185 | |||
| 186 | int udp_message( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ){ |
||
| 4505 | mejdrech | 187 | ERROR_DECLARE; |
| 188 | |||
| 189 | packet_t packet; |
||
| 190 | |||
| 4499 | mejdrech | 191 | * answer_count = 0; |
| 192 | switch( IPC_GET_METHOD( * call )){ |
||
| 193 | case NET_TL_RECEIVED: |
||
| 4700 | mejdrech | 194 | fibril_rwlock_read_lock( & udp_globals.lock ); |
| 195 | if( ! ERROR_OCCURRED( packet_translate( udp_globals.net_phone, & packet, IPC_GET_PACKET( call )))){ |
||
| 196 | ERROR_CODE = udp_received_msg( IPC_GET_DEVICE( call ), packet, 0 ); |
||
| 197 | } |
||
| 198 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
||
| 199 | return ERROR_CODE; |
||
| 4579 | mejdrech | 200 | case IPC_M_CONNECT_TO_ME: |
| 201 | return process_client_messages( callid, call, answer, answer_count ); |
||
| 4499 | mejdrech | 202 | } |
| 203 | return ENOTSUP; |
||
| 204 | } |
||
| 205 | |||
| 4579 | mejdrech | 206 | int process_client_messages( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ){ |
| 207 | ERROR_DECLARE; |
||
| 208 | |||
| 209 | int res; |
||
| 210 | bool keep_on_going = true; |
||
| 211 | socket_cores_t local_sockets; |
||
| 212 | int app_phone = IPC_GET_PHONE( call ); |
||
| 213 | void * addr; |
||
| 214 | size_t addrlen; |
||
| 4700 | mejdrech | 215 | fibril_rwlock_t lock; |
| 4579 | mejdrech | 216 | |
| 217 | /* |
||
| 218 | * Accept the connection |
||
| 219 | * - Answer the first IPC_M_CONNECT_ME_TO call. |
||
| 220 | */ |
||
| 221 | ipc_answer_0( callid, EOK ); |
||
| 222 | |||
| 223 | socket_cores_initialize( & local_sockets ); |
||
| 4700 | mejdrech | 224 | fibril_rwlock_initialize( & lock ); |
| 4579 | mejdrech | 225 | |
| 226 | while( keep_on_going ){ |
||
| 227 | // refresh data |
||
| 228 | * answer_count = 0; |
||
| 229 | IPC_SET_RETVAL( * answer, 0 ); |
||
| 230 | // just to be precize |
||
| 231 | IPC_SET_METHOD( * answer, 0 ); |
||
| 232 | IPC_SET_ARG1( * answer, 0 ); |
||
| 233 | IPC_SET_ARG2( * answer, 0 ); |
||
| 234 | IPC_SET_ARG3( * answer, 0 ); |
||
| 235 | IPC_SET_ARG4( * answer, 0 ); |
||
| 236 | IPC_SET_ARG5( * answer, 0 ); |
||
| 237 | |||
| 238 | callid = async_get_call( call ); |
||
| 4589 | mejdrech | 239 | // printf( "message %d\n", IPC_GET_METHOD( * call )); |
| 4579 | mejdrech | 240 | |
| 241 | switch( IPC_GET_METHOD( * call )){ |
||
| 242 | case IPC_M_PHONE_HUNGUP: |
||
| 243 | keep_on_going = false; |
||
| 244 | res = EOK; |
||
| 245 | break; |
||
| 246 | case NET_SOCKET: |
||
| 4700 | mejdrech | 247 | fibril_rwlock_write_lock( & lock ); |
| 4589 | mejdrech | 248 | res = socket_create( & local_sockets, app_phone, SOCKET_SET_SOCKET_ID( answer )); |
| 4700 | mejdrech | 249 | fibril_rwlock_write_unlock( & lock ); |
| 4589 | mejdrech | 250 | * SOCKET_SET_HEADER_SIZE( answer ) = sizeof( udp_header_t ); |
| 251 | * SOCKET_SET_DATA_FRAGMENT_SIZE( answer ) = MAX_UDP_FRAGMENT_SIZE; |
||
| 252 | * answer_count = 3; |
||
| 4579 | mejdrech | 253 | break; |
| 254 | case NET_SOCKET_BIND: |
||
| 4589 | mejdrech | 255 | if( ERROR_OCCURRED( socket_read_data( & addr, & addrlen ))){ |
| 4579 | mejdrech | 256 | res = ERROR_CODE; |
| 257 | break; |
||
| 258 | } |
||
| 4700 | mejdrech | 259 | fibril_rwlock_write_lock( & lock ); |
| 260 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 261 | 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 ); |
||
| 262 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
||
| 263 | fibril_rwlock_write_unlock( & lock ); |
||
| 4579 | mejdrech | 264 | free( addr ); |
| 265 | break; |
||
| 266 | case NET_SOCKET_SENDTO: |
||
| 4589 | mejdrech | 267 | if( ERROR_OCCURRED( socket_read_data( & addr, & addrlen ))){ |
| 4579 | mejdrech | 268 | res = ERROR_CODE; |
| 269 | break; |
||
| 270 | } |
||
| 4700 | mejdrech | 271 | fibril_rwlock_read_lock( & lock ); |
| 272 | fibril_rwlock_read_lock( & udp_globals.lock ); |
||
| 4589 | mejdrech | 273 | res = udp_sendto_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), addr, addrlen, SOCKET_GET_DATA_FRAGMENTS( call ), SOCKET_GET_FLAGS( call )); |
| 4700 | mejdrech | 274 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
| 275 | fibril_rwlock_read_unlock( & lock ); |
||
| 4579 | mejdrech | 276 | free( addr ); |
| 277 | break; |
||
| 278 | case NET_SOCKET_RECVFROM: |
||
| 4700 | mejdrech | 279 | fibril_rwlock_read_lock( & lock ); |
| 280 | fibril_rwlock_read_lock( & udp_globals.lock ); |
||
| 4589 | mejdrech | 281 | res = udp_recvfrom_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), SOCKET_GET_FLAGS( call )); |
| 4700 | mejdrech | 282 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
| 283 | fibril_rwlock_read_unlock( & lock ); |
||
| 4589 | mejdrech | 284 | if( res > 0 ){ |
| 285 | * SOCKET_SET_READ_DATA_LENGTH( answer ) = res; |
||
| 4603 | mejdrech | 286 | * SOCKET_SET_ADDRESS_LENGTH( answer ) = sizeof( struct sockaddr_in ); |
| 287 | * answer_count = 2; |
||
| 4589 | mejdrech | 288 | res = EOK; |
| 289 | } |
||
| 4579 | mejdrech | 290 | break; |
| 291 | case NET_SOCKET_CLOSE: |
||
| 4700 | mejdrech | 292 | fibril_rwlock_write_lock( & lock ); |
| 293 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 4579 | mejdrech | 294 | res = socket_destroy( udp_globals.net_phone, SOCKET_GET_SOCKET_ID( call ), & local_sockets, & udp_globals.sockets ); |
| 4700 | mejdrech | 295 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
| 296 | fibril_rwlock_write_unlock( & lock ); |
||
| 4579 | mejdrech | 297 | break; |
| 298 | case NET_SOCKET_GETSOCKOPT: |
||
| 299 | case NET_SOCKET_SETSOCKOPT: |
||
| 300 | default: |
||
| 301 | res = ENOTSUP; |
||
| 302 | break; |
||
| 303 | } |
||
| 304 | |||
| 4589 | mejdrech | 305 | // printf( "res = %d\n", res ); |
| 4579 | mejdrech | 306 | |
| 307 | switch( * answer_count ){ |
||
| 308 | case 0: ipc_answer_0( callid, res ); |
||
| 309 | continue; |
||
| 310 | case 1: ipc_answer_1( callid, res, IPC_GET_ARG1( * answer )); |
||
| 311 | continue; |
||
| 312 | case 2: ipc_answer_2( callid, res, IPC_GET_ARG1( * answer ), IPC_GET_ARG2( * answer )); |
||
| 313 | continue; |
||
| 314 | case 3: ipc_answer_3( callid, res, IPC_GET_ARG1( * answer ), IPC_GET_ARG2( * answer ), IPC_GET_ARG3( * answer )); |
||
| 315 | continue; |
||
| 316 | case 4: ipc_answer_4( callid, res, IPC_GET_ARG1( * answer ), IPC_GET_ARG2( * answer ), IPC_GET_ARG3( * answer ), IPC_GET_ARG4( * answer )); |
||
| 317 | continue; |
||
| 318 | default: ipc_answer_5( callid, res, IPC_GET_ARG1( * answer ), IPC_GET_ARG2( * answer ), IPC_GET_ARG3( * answer ), IPC_GET_ARG4( * answer ), IPC_GET_ARG5( * answer )); |
||
| 319 | continue; |
||
| 320 | } |
||
| 321 | } |
||
| 322 | |||
| 323 | socket_cores_destroy( & local_sockets ); |
||
| 324 | |||
| 325 | return EOK; |
||
| 326 | } |
||
| 327 | |||
| 4589 | mejdrech | 328 | int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, void * addr, size_t addrlen, int fragments, int flags ){ |
| 4579 | mejdrech | 329 | ERROR_DECLARE; |
| 330 | |||
| 331 | socket_core_ref socket; |
||
| 332 | struct sockaddr * address; |
||
| 333 | struct sockaddr_in * address_in; |
||
| 334 | packet_t packet; |
||
| 4589 | mejdrech | 335 | packet_t next_packet; |
| 4579 | mejdrech | 336 | udp_header_ref header; |
| 4589 | mejdrech | 337 | int index; |
| 338 | int total_length; |
||
| 339 | int length; |
||
| 4579 | mejdrech | 340 | |
| 341 | if( addrlen < sizeof( struct sockaddr )) return EINVAL; |
||
| 342 | address = ( struct sockaddr * ) addr; |
||
| 343 | switch( address->sa_family ){ |
||
| 344 | case AF_INET: |
||
| 345 | if( addrlen != sizeof( struct sockaddr_in )) return EINVAL; |
||
| 346 | address_in = ( struct sockaddr_in * ) addr; |
||
| 347 | socket = socket_cores_find( local_sockets, socket_id ); |
||
| 348 | if( ! socket ) return ENOTSOCK; |
||
| 4589 | mejdrech | 349 | |
| 4700 | mejdrech | 350 | // bind the socket to a random free port if not bound |
| 351 | if( socket->port <= 0 ){ |
||
| 352 | // try to find a free port |
||
| 353 | fibril_rwlock_read_unlock( & udp_globals.lock ); |
||
| 354 | fibril_rwlock_write_lock( & udp_globals.lock ); |
||
| 355 | ERROR_PROPAGATE( socket_bind_free_port( & udp_globals.sockets, socket, UDP_FREE_PORTS_START, UDP_FREE_PORTS_END, udp_globals.last_used_port )); |
||
| 356 | fibril_rwlock_write_unlock( & udp_globals.lock ); |
||
| 357 | fibril_rwlock_read_lock( & udp_globals.lock ); |
||
| 358 | // set the next port as the search starting port number |
||
| 359 | udp_globals.last_used_port = socket->port; |
||
| 360 | } |
||
| 4579 | mejdrech | 361 | // TODO do not ask all the time |
| 362 | ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.addr_len, & udp_globals.prefix, & udp_globals.content, & udp_globals.suffix )); |
||
| 4589 | mejdrech | 363 | |
| 364 | // read the first packet fragment |
||
| 365 | total_length = socket_read_packet_data( & packet, sizeof( udp_header_t ), address_in ); |
||
| 366 | if( total_length < 0 ) return total_length; |
||
| 367 | // prefix the udp header |
||
| 4579 | mejdrech | 368 | header = PACKET_PREFIX( packet, udp_header_t ); |
| 369 | if( ! header ){ |
||
| 370 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 371 | return ENOMEM; |
||
| 372 | } |
||
| 4589 | mejdrech | 373 | // read the rest of the packet fragments |
| 374 | for( index = 1; index < fragments; ++ index ){ |
||
| 375 | length = socket_read_packet_data( & next_packet, 0, address_in ); |
||
| 376 | if( length < 0 ){ |
||
| 377 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 378 | return length; |
||
| 379 | } |
||
| 380 | packet = pq_add( packet, next_packet, index, 0 ); |
||
| 381 | total_length += length; |
||
| 382 | } |
||
| 383 | // set the udp header |
||
| 4579 | mejdrech | 384 | header->source = ( socket->port < 0 ) ? 0 : htons( socket->port ); |
| 385 | header->dest = htons( address_in->sin_port ); |
||
| 4589 | mejdrech | 386 | header->len = htons( total_length + sizeof( udp_header_t )); |
| 4579 | mejdrech | 387 | // TODO my ip address for the pseudo header checksum |
| 388 | header->check = 0; |
||
| 4589 | mejdrech | 389 | // prepare the first packet fragment |
| 4579 | mejdrech | 390 | if( ERROR_OCCURRED( ip_client_prepare_packet( packet, IPPROTO_UDP, 0, 0, 0, 0 ))){ |
| 391 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 392 | return ERROR_CODE; |
||
| 393 | } |
||
| 4589 | mejdrech | 394 | // send the packet |
| 4579 | mejdrech | 395 | return ip_send_msg( udp_globals.ip_phone, socket->device_id, packet, SERVICE_UDP ); |
| 396 | // TODO IPv6 |
||
| 397 | default: |
||
| 398 | return EAFNOSUPPORT; |
||
| 399 | } |
||
| 400 | return EOK; |
||
| 401 | } |
||
| 402 | |||
| 4589 | mejdrech | 403 | int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags ){ |
| 4579 | mejdrech | 404 | ERROR_DECLARE; |
| 405 | |||
| 4589 | mejdrech | 406 | socket_core_ref socket; |
| 407 | int packet_id; |
||
| 408 | packet_t packet; |
||
| 409 | udp_header_ref header; |
||
| 410 | struct sockaddr_in address; |
||
| 411 | int length; |
||
| 412 | packet_t next_packet; |
||
| 413 | void * data; |
||
| 414 | int fragments; |
||
| 415 | int * lengths; |
||
| 416 | int index; |
||
| 417 | uint8_t * addr; |
||
| 418 | |||
| 419 | // find the socket |
||
| 420 | socket = socket_cores_find( local_sockets, socket_id ); |
||
| 421 | if( ! socket ) return ENOTSOCK; |
||
| 422 | // get the next received packet |
||
| 423 | packet_id = dyn_fifo_value( & socket->received ); |
||
| 424 | if( packet_id < 0 ) return packet_id; |
||
| 425 | ERROR_PROPAGATE( packet_translate( udp_globals.net_phone, & packet, packet_id )); |
||
| 426 | // get udp header |
||
| 427 | data = packet_get_data( packet ); |
||
| 428 | if( ! data ){ |
||
| 429 | pq_release( udp_globals.net_phone, packet_id ); |
||
| 430 | return NO_DATA; |
||
| 431 | } |
||
| 432 | header = ( udp_header_ref ) data; |
||
| 433 | // set the source address |
||
| 434 | address.sin_family = PF_INET; |
||
| 4603 | mejdrech | 435 | address.sin_port = ntohs( header->source ); |
| 4589 | mejdrech | 436 | length = packet_get_addr( packet, & addr, NULL ); |
| 437 | if( length != sizeof( address.sin_addr.s_addr )){ |
||
| 438 | pq_release( udp_globals.net_phone, packet_id ); |
||
| 439 | return EINVAL; |
||
| 440 | } |
||
| 441 | address.sin_addr.s_addr = *(( uint32_t * ) addr ); |
||
| 442 | bzero( & address.sin_zero, sizeof( address.sin_zero )); |
||
| 443 | // send the source address |
||
| 444 | ERROR_PROPAGATE( socket_write_data( & address, sizeof( address ))); |
||
| 445 | next_packet = pq_next( packet ); |
||
| 446 | if( ! next_packet ){ |
||
| 447 | // write all if only one fragment |
||
| 448 | ERROR_PROPAGATE( socket_write_data( data + sizeof( udp_header_t ), packet_get_data_length( packet ) - sizeof( udp_header_t ))); |
||
| 449 | // store the total length |
||
| 450 | length = packet_get_data_length( packet ) - sizeof( udp_header_t ); |
||
| 451 | }else{ |
||
| 452 | // count the packet fragments |
||
| 453 | fragments = 1; |
||
| 454 | next_packet = pq_next( packet ); |
||
| 455 | while(( next_packet = pq_next( next_packet ))){ |
||
| 456 | ++ fragments; |
||
| 457 | } |
||
| 458 | // compute and store the fragment lengths |
||
| 459 | lengths = ( int * ) malloc( sizeof( int ) * ( fragments + 1 )); |
||
| 460 | if( ! lengths ) return ENOMEM; |
||
| 461 | lengths[ 0 ] = packet_get_data_length( packet ) - sizeof( udp_header_t ); |
||
| 462 | lengths[ fragments ] = lengths[ 0 ]; |
||
| 463 | next_packet = pq_next( packet ); |
||
| 464 | for( index = 1; index < fragments; ++ index ){ |
||
| 465 | lengths[ index ] = packet_get_data_length( next_packet ); |
||
| 466 | lengths[ fragments ] += lengths[ index ]; |
||
| 467 | next_packet = pq_next( packet ); |
||
| 468 | }while( next_packet ); |
||
| 469 | // write the fragment lengths |
||
| 470 | ERROR_PROPAGATE( socket_write_data( lengths, sizeof( int ) * ( fragments + 1 ))); |
||
| 471 | // write the first fragment |
||
| 472 | ERROR_PROPAGATE( socket_write_data( data + sizeof( udp_header_t ), lengths[ 0 ] )); |
||
| 473 | next_packet = pq_next( packet ); |
||
| 474 | // write the rest of the fragments |
||
| 475 | for( index = 1; index < fragments; ++ index ){ |
||
| 476 | ERROR_PROPAGATE( socket_write_data( packet_get_data( next_packet ), lengths[ index ] )); |
||
| 477 | next_packet = pq_next( packet ); |
||
| 478 | }while( next_packet ); |
||
| 479 | // store the total length |
||
| 480 | length = lengths[ fragments ]; |
||
| 481 | free( lengths ); |
||
| 482 | } |
||
| 483 | // release the packet |
||
| 484 | dyn_fifo_pop( & socket->received ); |
||
| 485 | pq_release( udp_globals.net_phone, packet_get_id( packet )); |
||
| 486 | // return the total length |
||
| 487 | return length; |
||
| 488 | } |
||
| 489 | |||
| 490 | int socket_write_data( void * data, size_t data_length ){ |
||
| 491 | size_t length; |
||
| 4579 | mejdrech | 492 | ipc_callid_t callid; |
| 493 | |||
| 4589 | mejdrech | 494 | if(( ! ipc_data_read_receive( & callid, & length )) |
| 495 | || ( length < data_length )){ |
||
| 496 | return EINVAL; |
||
| 497 | } |
||
| 498 | return ipc_data_read_finalize( callid, data, data_length ); |
||
| 499 | } |
||
| 500 | |||
| 501 | int socket_read_data( void ** data, size_t * length ){ |
||
| 502 | ERROR_DECLARE; |
||
| 503 | |||
| 504 | ipc_callid_t callid; |
||
| 505 | |||
| 4579 | mejdrech | 506 | if( !( data && length )) return EBADMEM; |
| 507 | if( ! ipc_data_write_receive( & callid, length )) return EINVAL; |
||
| 508 | * data = malloc( * length ); |
||
| 509 | if( ! data ) return ENOMEM; |
||
| 510 | if( ERROR_OCCURRED( ipc_data_write_finalize( callid, * data, * length ))){ |
||
| 511 | free( data ); |
||
| 512 | return ERROR_CODE; |
||
| 513 | } |
||
| 514 | return EOK; |
||
| 515 | } |
||
| 516 | |||
| 4589 | mejdrech | 517 | int socket_read_packet_data( packet_ref packet, size_t prefix, struct sockaddr_in * address_in ){ |
| 518 | ERROR_DECLARE; |
||
| 519 | |||
| 520 | ipc_callid_t callid; |
||
| 521 | size_t length; |
||
| 522 | void * data; |
||
| 523 | |||
| 524 | // get the data length |
||
| 525 | if( ! ipc_data_write_receive( & callid, & length )) return EINVAL; |
||
| 526 | // get a new packet |
||
| 527 | * packet = packet_get_4( udp_globals.net_phone, length, udp_globals.addr_len, prefix + udp_globals.prefix, udp_globals.suffix ); |
||
| 528 | if( ! packet ) return ENOMEM; |
||
| 529 | // allocate space in the packet |
||
| 530 | data = packet_suffix( * packet, length ); |
||
| 531 | if( ! data ){ |
||
| 532 | pq_release( udp_globals.net_phone, packet_get_id( * packet )); |
||
| 533 | return ENOMEM; |
||
| 534 | } |
||
| 535 | // read the data into the packet |
||
| 536 | if( ERROR_OCCURRED( ipc_data_write_finalize( callid, data, length )) |
||
| 537 | // set the packet destination address |
||
| 538 | || ERROR_OCCURRED( packet_set_addr( * packet, NULL, ( uint8_t * ) & address_in->sin_addr.s_addr, sizeof( address_in->sin_addr.s_addr )))){ |
||
| 539 | pq_release( udp_globals.net_phone, packet_get_id( * packet )); |
||
| 540 | return ERROR_CODE; |
||
| 541 | } |
||
| 542 | return length; |
||
| 543 | } |
||
| 544 | |||
| 4499 | mejdrech | 545 | /** @} |
| 546 | */ |