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