Subversion Repositories HelenOS

Rev

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
 */