WebSVN – HelenOS – Diff – /branches/network/uspace/srv/net/tl/udp/udp.c


/*
 * Copyright (c) 2008 Lukas Mejdrech
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
 
/** @addtogroup udp
 *  @{
 */
 
/** @file
 *  UDP module implementation.
 *  @see udp.h
 */
 
#include <async.h>
#include <fibril_sync.h>
#include <malloc.h>
 
#include <ipc/ipc.h>
#include <ipc/services.h>
 
#include "../../err.h"
#include "../../messages.h"
#include "../../modules.h"
 
#include "../../structures/dynamic_fifo.h"
#include "../../structures/packet/packet_client.h"
 
#include "../../include/crc.h"
#include "../../include/in.h"
#include "../../include/in6.h"
#include "../../include/inet.h"
#include "../../include/ip_client.h"
#include "../../include/ip_interface.h"
#include "../../include/ip_protocols.h"
#include "../../include/icmp_client.h"
#include "../../include/icmp_interface.h"
#include "../../include/net_interface.h"
#include "../../include/socket_codes.h"
#include "../../include/socket_errno.h"
 
#include "../../socket/socket_core.h"
#include "../../socket/socket_messages.h"
 
#include "../tl_common.h"
#include "../tl_messages.h"
 
#include "udp.h"
#include "udp_header.h"
#include "udp_module.h"
 
/** Default UDP checksum computing.
 */
#define NET_DEFAULT_UDP_CHECKSUM_COMPUTING  true
 
/** Maximum UDP fragment size.
 */
#define MAX_UDP_FRAGMENT_SIZE   65535
 
/** Free ports pool start.
 */
#define UDP_FREE_PORTS_START    1025
 
/** Free ports pool end.
 */
#define UDP_FREE_PORTS_END      65535
 
/** Processes the received UDP packet queue.
 *  Is used as an entry point from the underlying IP module.
 *  Notifies the destination socket application.
 *  Releases the packet on error or send an ICMP error notification..
 *  @param device_id The device identifier. Ignored parameter.
 *  @param packet The received packet queue. Input/output parameter.
 *  @param receiver The target service. Ignored parameter.
 *  @param error The packet error reporting service. Prefixes the received packet. Input parameter.
 *  @returns EOK on success.
 *  @returns EINVAL if the packet is not valid.
 *  @returns EINVAL if the stored packet address is not the an_addr_t.
 *  @returns EINVAL if the packet does not contain any data.
 *  @returns NO_DATA if the packet content is shorter than the user datagram header.
 *  @returns ENOMEM if there is not enough memory left.
 *  @returns EADDRNOTAVAIL if the destination socket does not exist.
 *  @returns Other error codes as defined for the ip_client_process_packet() function.
 */
int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error );
 
/** Releases the packet and returns the result.
 *  @param packet The packet queue to be released. Input parameter.
 *  @param result The result to be returned. Input parameter.
 *  @return The result parameter.
 */
int udp_release_and_return( packet_t packet, int result );
 
/** @name Socket messages processing functions
 */
/*@{*/
 
/** Processes the socket client messages.
 *  Runs until the client module disconnects.
 *  @param callid The message identifier. Input parameter.
 *  @param call The message parameters. Input parameter.
 *  @returns EOK on success.
 *  @see socket.h
 */
int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call );
 
/** Sends data from the socket to the remote address.
 *  Binds the socket to a free port if not already connected/bound.
 *  Handles the NET_SOCKET_SENDTO message.
 *  Supports AF_INET and AF_INET6 address families.
 *  @param local_sockets The application local sockets. Input/output parameter.
 *  @param socket_id Socket identifier. Input parameter.
 *  @param addr The destination address. Input parameter.
 *  @param addrlen The address length. Input parameter.
 *  @param fragments The number of data fragments. Input parameter.
 *  @param data_fragment_size The data fragment size in bytes. Input parameter.
 *  @param flags Various send flags. Input parameter.
 *  @returns EOK on success.
 *  @returns EAFNOTSUPPORT if the address family is not supported.
 *  @returns ENOTSOCK if the socket is not found.
 *  @returns EINVAL if the address is invalid.
 *  @returns ENOTCONN if the sending socket is not and cannot be bound.
 *  @returns ENOMEM if there is not enough memory left.
 *  @returns Other error codes as defined for the socket_read_packet_data() function.
 *  @returns Other error codes as defined for the ip_client_prepare_packet() function.
 *  @returns Other error codes as defined for the ip_send_msg() function.
 */
int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, size_t data_fragment_size, int flags );
 
/** Receives data to the socket.
 *  Handles the NET_SOCKET_RECVFROM message.
 *  Replies the source address as well.
 *  @param local_sockets The application local sockets. Input parameter.
 *  @param socket_id Socket identifier. Input parameter.
 *  @param flags Various receive flags. Input parameter.
 *  @param addrlen The source address length. Output parameter.
 *  @returns The number of bytes received.
 *  @returns ENOTSOCK if the socket is not found.
 *  @returns NO_DATA if there are no received packets or data.
 *  @returns ENOMEM if there is not enough memory left.
 *  @returns EINVAL if the received address is not an IP address.
 *  @returns Other error codes as defined for the packet_translate() function.
 *  @returns Other error codes as defined for the data_reply() function.
 */
int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags, size_t * addrlen );
 
/*@}*/
 
/** UDP global data.
 */
udp_globals_t   udp_globals;
 
int udp_initialize( async_client_conn_t client_connection ){
    ERROR_DECLARE;
 
    measured_string_t   names[] = {{ "UDP_CHECKSUM_COMPUTING", 22 }};
    measured_string_ref configuration;
    size_t              count = sizeof( names ) / sizeof( measured_string_t );
    char *              data;
 
    fibril_rwlock_initialize( & udp_globals.lock );
    fibril_rwlock_write_lock( & udp_globals.lock );
    udp_globals.icmp_phone = icmp_connect_module( SERVICE_ICMP );
    if( udp_globals.icmp_phone < 0 ){
        return udp_globals.icmp_phone;
    }
    udp_globals.ip_phone = ip_bind_service( SERVICE_IP, IPPROTO_UDP, SERVICE_UDP, client_connection, udp_received_msg );
    if( udp_globals.ip_phone < 0 ){
        return udp_globals.ip_phone;
    }
    ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.packet_dimension.addr_len, & udp_globals.packet_dimension.prefix, & udp_globals.packet_dimension.content, & udp_globals.packet_dimension.suffix ));
    ERROR_PROPAGATE( socket_ports_initialize( & udp_globals.sockets ));
    udp_globals.packet_dimension.prefix += sizeof( udp_header_t );
    udp_globals.packet_dimension.content -= sizeof( udp_header_t );
    udp_globals.last_used_port = UDP_FREE_PORTS_START - 1;
    // get configuration
    udp_globals.checksum_computing = NET_DEFAULT_UDP_CHECKSUM_COMPUTING;
    configuration = & names[ 0 ];
    ERROR_PROPAGATE( net_get_conf_req( udp_globals.net_phone, & configuration, count, & data ));
    if( configuration ){
        if( configuration[ 0 ].value ){
            udp_globals.checksum_computing = ( configuration[ 0 ].value[ 0 ] == 'y' );
        }
        net_free_settings( configuration, data );
    }
    fibril_rwlock_write_unlock( & udp_globals.lock );
    return EOK;
}
 
int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error ){
    ERROR_DECLARE;
 
    size_t          length;
    size_t          offset;
    int             result;
    uint8_t *       data;
    udp_header_ref  header;
    socket_core_ref *   socket;
    packet_t        next_packet;
    size_t          total_length;
    uint32_t        checksum;
    int             fragments;
    packet_t        tmp_packet;
    icmp_type_t     type;
    icmp_code_t     code;
    ip_pseudo_header_ref    ip_header;
    struct sockaddr *       src;
    struct sockaddr *       dest;
 
    if( error ){
        switch( error ){
            case SERVICE_ICMP:
                // process error
                // TODO remove debug dump
                // length = icmp_client_header_length( packet );
                result = icmp_client_process_packet( packet, & type, & code, NULL, NULL );
                if( result < 0 ){
                    return udp_release_and_return( packet, result );
                }
                printf( "ICMP error %d (%d) in packet %d\n", type, code, packet_get_id( packet ) );
                length = ( size_t ) result;
                if( ERROR_OCCURRED( packet_trim( packet, length, 0 ))){
                    return udp_release_and_return( packet, ERROR_CODE );
                }
                break;
            default:
                return udp_release_and_return( packet, ENOTSUP );
        }
    }
    // TODO process received ipopts?
    result = ip_client_process_packet( packet, NULL, NULL, NULL, NULL, NULL );
    if( result < 0 ){
        return udp_release_and_return( packet, result );
    }
    offset = ( size_t ) result;
 
    length = packet_get_data_length( packet );
    if( length <= 0 ){
        return udp_release_and_return( packet, EINVAL );
    }
    if( length < sizeof( udp_header_t ) + offset ){
        return udp_release_and_return( packet, NO_DATA );
    }
    data = packet_get_data( packet );
    if( ! data ){
        return udp_release_and_return( packet, NO_DATA );
    }
    // get udp header
    header = ( udp_header_ref )( data + offset );
    // find the destination socket
    socket = socket_ports_find( & udp_globals.sockets, ntohs( header->destination_port ));
    if( ! socket ){
        tl_send_icmp_port_unreachable( udp_globals.net_phone, udp_globals.icmp_phone, packet, error );
        return EADDRNOTAVAIL;
    }
    // trim after successful processing to be able to send an ICMP error message!
    ERROR_PROPAGATE( packet_trim( packet, offset, 0 ));
    // count the received packet fragments
    next_packet = packet;
    fragments = 0;
    total_length = ntohs( header->total_length );
    // compute header checksum if set
    if( header->checksum && ( ! error )){
        result = packet_get_addr( packet, ( uint8_t ** ) & src, ( uint8_t ** ) & dest );
        if( result <= 0 ){
            return udp_release_and_return( packet, result );
        }
        if( ERROR_OCCURRED( ip_client_get_pseudo_header( IPPROTO_UDP, src, result, dest, result, total_length, & ip_header, & length ))){
            return udp_release_and_return( packet, ERROR_CODE );
        }else{
            checksum = compute_checksum( 0, ip_header, length );
            // the udp header checksum will be added with the first fragment later
            free( ip_header );
        }
    }else{
        header->checksum = 0;
        checksum = 0;
    }
    do{
        ++ fragments;
        length = packet_get_data_length( next_packet );
        if( length <= 0 ){
            return udp_release_and_return( packet, NO_DATA );
        }
        if( total_length < length ){
            if( ERROR_OCCURRED( packet_trim( next_packet, 0, length - total_length ))){
                return udp_release_and_return( packet, ERROR_CODE );
            }
            // add partial checksum if set
            if( header->checksum ){
                checksum = compute_checksum( checksum, packet_get_data( packet ), packet_get_data_length( packet ));
            }
            // relese the rest of the packet fragments
            tmp_packet = pq_next( next_packet );
            while( tmp_packet ){
                next_packet = pq_detach( tmp_packet );
                pq_release( udp_globals.net_phone, packet_get_id( tmp_packet ));
                tmp_packet = next_packet;
            }
            // exit the loop
            break;
        }
        total_length -= length;
        // add partial checksum if set
        if( header->checksum ){
            checksum = compute_checksum( checksum, packet_get_data( packet ), packet_get_data_length( packet ));
        }
    }while(( next_packet = pq_next( next_packet )) && ( total_length > 0 ));
    // check checksum
    if( header->checksum ){
        if( flip_checksum( compact_checksum( checksum ))){
            // TODO checksum error ICMP?
            // TODO remove debug dump
            printf("udp check failed %x => %x\n", header->checksum, flip_checksum( compact_checksum( checksum )));
            return udp_release_and_return( packet, EINVAL );
        }
    }
    // queue the received packet
    if( ERROR_OCCURRED( dyn_fifo_push( &( ** socket ).received, packet_get_id( packet ), SOCKET_MAX_RECEIVED_SIZE ))){
        return udp_release_and_return( packet, ERROR_CODE );
    }
 
    // notify the destination socket
    async_msg_5(( ** socket ).phone, NET_SOCKET_RECEIVED, ( ipcarg_t ) ( ** socket ).socket_id, 0, 0, 0, ( ipcarg_t ) fragments );
    return EOK;
}
 
int udp_message( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ){
    ERROR_DECLARE;
 
    packet_t    packet;
 
    * answer_count = 0;
    switch( IPC_GET_METHOD( * call )){
        case NET_TL_RECEIVED:
            fibril_rwlock_read_lock( & udp_globals.lock );
            if( ! ERROR_OCCURRED( packet_translate( udp_globals.net_phone, & packet, IPC_GET_PACKET( call )))){
                ERROR_CODE = udp_received_msg( IPC_GET_DEVICE( call ), packet, SERVICE_UDP, IPC_GET_ERROR( call ));
            }
            fibril_rwlock_read_unlock( & udp_globals.lock );
            return ERROR_CODE;
        case IPC_M_CONNECT_TO_ME:
            return udp_process_client_messages( callid, * call );
    }
    return ENOTSUP;
}
 
int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call ){
    int                     res;
    bool                    keep_on_going = true;
    socket_cores_t          local_sockets;
    int                     app_phone = IPC_GET_PHONE( & call );
    struct sockaddr *       addr;
    size_t                  addrlen;
//  fibril_rwlock_t         lock;
    ipc_call_t              answer;
    int                     answer_count;
 
    /*
     * Accept the connection
     *  - Answer the first IPC_M_CONNECT_ME_TO call.
     */
    ipc_answer_0( callid, EOK );
 
    // The client connection is only in one fibril and therefore no additional locks are needed.
 
    socket_cores_initialize( & local_sockets );
//  fibril_rwlock_initialize( & lock );
 
    while( keep_on_going ){
        // refresh data
        refresh_answer( & answer, & answer_count );
 
        callid = async_get_call( & call );
//      printf( "message %d\n", IPC_GET_METHOD( * call ));
 
        switch( IPC_GET_METHOD( call )){
            case IPC_M_PHONE_HUNGUP:
                keep_on_going = false;
                res = EOK;
                break;
            case NET_SOCKET:
//              fibril_rwlock_write_lock( & lock );
                res = socket_create( & local_sockets, app_phone, NULL, SOCKET_SET_SOCKET_ID( answer ));
//              fibril_rwlock_write_unlock( & lock );
                * SOCKET_SET_DATA_FRAGMENT_SIZE( answer ) = MAX_UDP_FRAGMENT_SIZE;
                * SOCKET_SET_HEADER_SIZE( answer ) = sizeof( udp_header_t );
                answer_count = 3;
                break;
            case NET_SOCKET_BIND:
                res = data_receive(( void ** ) & addr, & addrlen );
                if( res == EOK ){
//                  fibril_rwlock_write_lock( & lock );
                    fibril_rwlock_write_lock( & udp_globals.lock );
                    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 );
                    fibril_rwlock_write_unlock( & udp_globals.lock );
//                  fibril_rwlock_write_unlock( & lock );
                    free( addr );
                }
                break;
            case NET_SOCKET_SENDTO:
                res = data_receive(( void ** ) & addr, & addrlen );
                if( res == EOK ){
//                  fibril_rwlock_read_lock( & lock );
                    fibril_rwlock_read_lock( & udp_globals.lock );
                    res = udp_sendto_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), addr, addrlen, SOCKET_GET_DATA_FRAGMENTS( call ), SOCKET_GET_DATA_FRAGMENT_SIZE( call ), SOCKET_GET_FLAGS( call ));
                    fibril_rwlock_read_unlock( & udp_globals.lock );
//                  fibril_rwlock_read_unlock( & lock );
                    free( addr );
                }
                break;
            case NET_SOCKET_RECVFROM:
//              fibril_rwlock_read_lock( & lock );
                fibril_rwlock_read_lock( & udp_globals.lock );
                res = udp_recvfrom_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), SOCKET_GET_FLAGS( call ), & addrlen );
                fibril_rwlock_read_unlock( & udp_globals.lock );
//              fibril_rwlock_read_unlock( & lock );
                if( res > 0 ){
                    * SOCKET_SET_READ_DATA_LENGTH( answer ) = res;
                    * SOCKET_SET_ADDRESS_LENGTH( answer ) = addrlen;
                    answer_count = 2;
                    res = EOK;
                }
                break;
            case NET_SOCKET_CLOSE:
//              fibril_rwlock_write_lock( & lock );
                fibril_rwlock_write_lock( & udp_globals.lock );
                res = socket_destroy( udp_globals.net_phone, SOCKET_GET_SOCKET_ID( call ), & local_sockets, & udp_globals.sockets );
                fibril_rwlock_write_unlock( & udp_globals.lock );
//              fibril_rwlock_write_unlock( & lock );
                break;
            case NET_SOCKET_GETSOCKOPT:
            case NET_SOCKET_SETSOCKOPT:
            default:
                res = ENOTSUP;
                break;
        }
 
//      printf( "res = %d\n", res );
 
        answer_call( callid, res, & answer, answer_count );
    }
 
    // TODO call socket_destroy() on all bound!
    socket_cores_destroy( & local_sockets );
 
    return EOK;
}
 
int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, size_t data_fragment_size, int flags ){
    ERROR_DECLARE;
 
    socket_core_ref         socket;
    packet_t                packet;
    packet_t                next_packet;
    udp_header_ref          header;
    int                     index;
    size_t                  total_length;
    int                     result;
    uint16_t                dest_port;
    uint32_t                checksum;
    ip_pseudo_header_ref    ip_header;
    size_t                  headerlen;
    device_id_t             device_id;
 
    ERROR_PROPAGATE( tl_get_address_port( addr, addrlen, & dest_port ));
 
    socket = socket_cores_find( local_sockets, socket_id );
    if( ! socket ) return ENOTSOCK;
 
    // bind the socket to a random free port if not bound
    while( socket->port <= 0 ){
        // try to find a free port
        fibril_rwlock_read_unlock( & udp_globals.lock );
        fibril_rwlock_write_lock( & udp_globals.lock );
        if( socket->port <= 0 ){
            ERROR_PROPAGATE( socket_bind_free_port( & udp_globals.sockets, socket, UDP_FREE_PORTS_START, UDP_FREE_PORTS_END, udp_globals.last_used_port ));
            // set the next port as the search starting port number
            udp_globals.last_used_port = socket->port;
        }
        fibril_rwlock_write_unlock( & udp_globals.lock );
        fibril_rwlock_read_lock( & udp_globals.lock );
    }
 
    // TODO do not ask all the time
    ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.packet_dimension.addr_len, & udp_globals.packet_dimension.prefix, & udp_globals.packet_dimension.content, & udp_globals.packet_dimension.suffix ));
 
    // read the first packet fragment
    result = tl_socket_read_packet_data( udp_globals.net_phone, & packet, sizeof( udp_header_t ), & udp_globals.packet_dimension, addr, addrlen );
    if( result < 0 ) return result;
    total_length = ( size_t ) result;
    if( udp_globals.checksum_computing ){
        checksum = compute_checksum( 0, packet_get_data( packet ), packet_get_data_length( packet ));
    }else{
        checksum = 0;
    }
    // prefix the udp header
    header = PACKET_PREFIX( packet, udp_header_t );
    if( ! header ){
        return udp_release_and_return( packet, ENOMEM );
    }
    bzero( header, sizeof( * header ));
    // read the rest of the packet fragments
    for( index = 1; index < fragments; ++ index ){
        result = tl_socket_read_packet_data( udp_globals.net_phone, & next_packet, 0, & udp_globals.packet_dimension, addr, addrlen );
        if( result < 0 ){
            return udp_release_and_return( packet, result );
        }
        packet = pq_add( packet, next_packet, index, 0 );
        total_length += ( size_t ) result;
        if( udp_globals.checksum_computing ){
            checksum = compute_checksum( checksum, packet_get_data( next_packet ), packet_get_data_length( next_packet ));
        }
    }
    // set the udp header
    header->source_port = htons( socket->port );
    header->destination_port = htons( dest_port );
    header->total_length = htons( total_length + sizeof( udp_header_t ));
    header->checksum = 0;
    if( udp_globals.checksum_computing ){
        if( ERROR_OCCURRED( ip_get_route_req( udp_globals.ip_phone, IPPROTO_UDP, addr, addrlen, & device_id, & ip_header, & headerlen ))){
            return udp_release_and_return( packet, ERROR_CODE );
        }
        if( ERROR_OCCURRED( ip_client_set_pseudo_header_data_length( ip_header, headerlen, total_length + sizeof( udp_header_t )))){
            free( ip_header );
            return udp_release_and_return( packet, ERROR_CODE );
        }
/*//        TODO remove debug dump:
        uint8_t *   data;
        data = ip_header;
        printf( "ip_header:\tlength\t= %d\n\tdata\t= %.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX:%.2hhX %.2hhX %.2hhX %.2hhX\n", headerlen, data[ 0 ], data[ 1 ], data[ 2 ], data[ 3 ], data[ 4 ], data[ 5 ], data[ 6 ], data[ 7 ], data[ 8 ], data[ 9 ], data[ 10 ], data[ 11 ] );
*/      checksum = compute_checksum( checksum, ip_header, headerlen );
        checksum = compute_checksum( checksum, ( uint8_t * ) header, sizeof( * header ));
        header->checksum = htons( flip_checksum( compact_checksum( checksum )));
        free( ip_header );
    }else{
        device_id = -1;
    }
    // prepare the first packet fragment
    if( ERROR_OCCURRED( ip_client_prepare_packet( packet, IPPROTO_UDP, 0, 0, 0, 0 ))){
        return udp_release_and_return( packet, ERROR_CODE );
    }
    // send the packet
    return ip_send_msg( udp_globals.ip_phone, device_id, packet, SERVICE_UDP, 0 );
}
 
int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags, size_t * addrlen ){
    ERROR_DECLARE;
 
    socket_core_ref socket;
    int             packet_id;
    packet_t        packet;
    udp_header_ref  header;
    struct sockaddr *   addr;
    size_t          length;
    packet_t        next_packet;
    uint8_t *       data;
    size_t          fragments;
    size_t *        lengths;
    size_t          index;
    int             result;
 
    // find the socket
    socket = socket_cores_find( local_sockets, socket_id );
    if( ! socket ) return ENOTSOCK;
    // get the next received packet
    packet_id = dyn_fifo_value( & socket->received );
    if( packet_id < 0 ) return NO_DATA;
    ERROR_PROPAGATE( packet_translate( udp_globals.net_phone, & packet, packet_id ));
    // get udp header
    data = packet_get_data( packet );
    if( ! data ){
        pq_release( udp_globals.net_phone, packet_id );
        return NO_DATA;
    }
    header = ( udp_header_ref ) data;
 
    // set the source address port
    result = packet_get_addr( packet, ( uint8_t ** ) & addr, NULL );
    if( ERROR_OCCURRED( tl_set_address_port( addr, result, ntohs( header->source_port )))){
        pq_release( udp_globals.net_phone, packet_id );
        return ERROR_CODE;
    }
    * addrlen = ( size_t ) result;
    // send the source address
    ERROR_PROPAGATE( data_reply( addr, * addrlen ));
 
    next_packet = pq_next( packet );
    if( ! next_packet ){
        // write all if only one fragment
        ERROR_PROPAGATE( data_reply( data + sizeof( udp_header_t ), packet_get_data_length( packet ) - sizeof( udp_header_t )));
        // store the total length
        length = packet_get_data_length( packet ) - sizeof( udp_header_t );
    }else{
        // count the packet fragments
        fragments = 1;
        next_packet = pq_next( packet );
        while(( next_packet = pq_next( next_packet ))){
            ++ fragments;
        }
        // compute and store the fragment lengths
        lengths = ( size_t * ) malloc( sizeof( size_t ) * fragments + sizeof( size_t ));
        if( ! lengths ) return ENOMEM;
        lengths[ 0 ] = packet_get_data_length( packet ) - sizeof( udp_header_t );
        lengths[ fragments ] = lengths[ 0 ];
        next_packet = pq_next( packet );
        for( index = 1; index < fragments; ++ index ){
            lengths[ index ] = packet_get_data_length( next_packet );
            lengths[ fragments ] += lengths[ index ];
            next_packet = pq_next( packet );
        }while( next_packet );
        // write the fragment lengths
        ERROR_PROPAGATE( data_reply( lengths, sizeof( int ) * ( fragments + 1 )));
        // write the first fragment
        ERROR_PROPAGATE( data_reply( data + sizeof( udp_header_t ), lengths[ 0 ] ));
        next_packet = pq_next( packet );
        // write the rest of the fragments
        for( index = 1; index < fragments; ++ index ){
            ERROR_PROPAGATE( data_reply( packet_get_data( next_packet ), lengths[ index ] ));
            next_packet = pq_next( packet );
        }while( next_packet );
        // store the total length
        length = lengths[ fragments ];
        free( lengths );
    }
    // release the packet
    dyn_fifo_pop( & socket->received );
    pq_release( udp_globals.net_phone, packet_get_id( packet ));
    // return the total length
    return ( int ) length;
}
 
int udp_release_and_return( packet_t packet, int result ){
    pq_release( udp_globals.net_phone, packet_get_id( packet ));
    return result;
}
 
/** @}
 */
 

Rev 4726	Rev 4728
1	/*	1	/*
2	* Copyright (c) 2008 Lukas Mejdrech	2	* Copyright (c) 2008 Lukas Mejdrech
3	* All rights reserved.	3	* All rights reserved.
4	*	4	*
5	* Redistribution and use in source and binary forms, with or without	5	* Redistribution and use in source and binary forms, with or without
6	* modification, are permitted provided that the following conditions	6	* modification, are permitted provided that the following conditions
7	* are met:	7	* are met:
8	*	8	*
9	* - Redistributions of source code must retain the above copyright	9	* - Redistributions of source code must retain the above copyright
10	* notice, this list of conditions and the following disclaimer.	10	* notice, this list of conditions and the following disclaimer.
11	* - Redistributions in binary form must reproduce the above copyright	11	* - Redistributions in binary form must reproduce the above copyright
12	* notice, this list of conditions and the following disclaimer in the	12	* notice, this list of conditions and the following disclaimer in the
13	* documentation and/or other materials provided with the distribution.	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	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.	15	* derived from this software without specific prior written permission.
16	*	16	*
17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	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	18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	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	23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	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	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.	26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	*/	27	*/
28		28
29	/** @addtogroup udp	29	/** @addtogroup udp
30	* @{	30	* @{
31	*/	31	*/
32		32
33	/** @file	33	/** @file
34	* UDP module implementation.	34	* UDP module implementation.
35	* @see udp.h	35	* @see udp.h
36	*/	36	*/
37		37
38	#include <async.h>	38	#include <async.h>
39	#include <fibril_sync.h>	39	#include <fibril_sync.h>
40	#include <malloc.h>	40	#include <malloc.h>
41		41
42	#include <ipc/ipc.h>	42	#include <ipc/ipc.h>
43	#include <ipc/services.h>	43	#include <ipc/services.h>
44		44
45	#include "../../err.h"	45	#include "../../err.h"
46	#include "../../messages.h"	46	#include "../../messages.h"
47	#include "../../modules.h"	47	#include "../../modules.h"
48		48
49	#include "../../structures/dynamic_fifo.h"	49	#include "../../structures/dynamic_fifo.h"
50	#include "../../structures/packet/packet_client.h"	50	#include "../../structures/packet/packet_client.h"
51		51
52	#include "../../include/crc.h"	52	#include "../../include/crc.h"
53	#include "../../include/in.h"	53	#include "../../include/in.h"
54	#include "../../include/in6.h"	54	#include "../../include/in6.h"
55	#include "../../include/inet.h"	55	#include "../../include/inet.h"
56	#include "../../include/ip_client.h"	56	#include "../../include/ip_client.h"
57	#include "../../include/ip_interface.h"	57	#include "../../include/ip_interface.h"
58	#include "../../include/ip_protocols.h"	58	#include "../../include/ip_protocols.h"
59	#include "../../include/icmp_client.h"	59	#include "../../include/icmp_client.h"
60	#include "../../include/icmp_interface.h"	60	#include "../../include/icmp_interface.h"
61	#include "../../include/net_interface.h"	61	#include "../../include/net_interface.h"
62	#include "../../include/socket_codes.h"	62	#include "../../include/socket_codes.h"
63	#include "../../include/socket_errno.h"	63	#include "../../include/socket_errno.h"
64		64
65	#include "../../socket/socket_core.h"	65	#include "../../socket/socket_core.h"
66	#include "../../socket/socket_messages.h"	66	#include "../../socket/socket_messages.h"
67		67
68	#include "../tl_common.h"	68	#include "../tl_common.h"
69	#include "../tl_messages.h"	69	#include "../tl_messages.h"
70		70
71	#include "udp.h"	71	#include "udp.h"
72	#include "udp_header.h"	72	#include "udp_header.h"
73	#include "udp_module.h"	73	#include "udp_module.h"
74		74
75	/** Default UDP checksum computing.	75	/** Default UDP checksum computing.
76	*/	76	*/
77	#define NET_DEFAULT_UDP_CHECKSUM_COMPUTING true	77	#define NET_DEFAULT_UDP_CHECKSUM_COMPUTING true
78		78
79	/** Maximum UDP fragment size.	79	/** Maximum UDP fragment size.
80	*/	80	*/
81	#define MAX_UDP_FRAGMENT_SIZE 65535	81	#define MAX_UDP_FRAGMENT_SIZE 65535
82		82
83	/** Free ports pool start.	83	/** Free ports pool start.
84	*/	84	*/
85	#define UDP_FREE_PORTS_START 1025	85	#define UDP_FREE_PORTS_START 1025
86		86
87	/** Free ports pool end.	87	/** Free ports pool end.
88	*/	88	*/
89	#define UDP_FREE_PORTS_END 65535	89	#define UDP_FREE_PORTS_END 65535
90		90
91	/** Processes the received UDP packet queue.	91	/** Processes the received UDP packet queue.
92	* Is used as an entry point from the underlying IP module.	92	* Is used as an entry point from the underlying IP module.
93	* Notifies the destination socket application.	93	* Notifies the destination socket application.
94	* Releases the packet on error or send an ICMP error notification..	94	* Releases the packet on error or send an ICMP error notification..
95	* @param device_id The device identifier. Ignored parameter.	95	* @param device_id The device identifier. Ignored parameter.
96	* @param packet The received packet queue. Input/output parameter.	96	* @param packet The received packet queue. Input/output parameter.
97	* @param receiver The target service. Ignored parameter.	97	* @param receiver The target service. Ignored parameter.
98	* @param error The packet error reporting service. Prefixes the received packet. Input parameter.	98	* @param error The packet error reporting service. Prefixes the received packet. Input parameter.
99	* @returns EOK on success.	99	* @returns EOK on success.
100	* @returns EINVAL if the packet is not valid.	100	* @returns EINVAL if the packet is not valid.
101	* @returns EINVAL if the stored packet address is not the an_addr_t.	101	* @returns EINVAL if the stored packet address is not the an_addr_t.
102	* @returns EINVAL if the packet does not contain any data.	102	* @returns EINVAL if the packet does not contain any data.
103	* @returns NO_DATA if the packet content is shorter than the user datagram header.	103	* @returns NO_DATA if the packet content is shorter than the user datagram header.
104	* @returns ENOMEM if there is not enough memory left.	104	* @returns ENOMEM if there is not enough memory left.
105	* @returns EADDRNOTAVAIL if the destination socket does not exist.	105	* @returns EADDRNOTAVAIL if the destination socket does not exist.
106	* @returns Other error codes as defined for the ip_client_process_packet() function.	106	* @returns Other error codes as defined for the ip_client_process_packet() function.
107	*/	107	*/
108	int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error );	108	int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error );
109		109
110	/** Releases the packet and returns the result.	110	/** Releases the packet and returns the result.
111	* @param packet The packet queue to be released. Input parameter.	111	* @param packet The packet queue to be released. Input parameter.
112	* @param result The result to be returned. Input parameter.	112	* @param result The result to be returned. Input parameter.
113	* @return The result parameter.	113	* @return The result parameter.
114	*/	114	*/
115	int udp_release_and_return( packet_t packet, int result );	115	int udp_release_and_return( packet_t packet, int result );
116		116
117	/** @name Socket messages processing functions	117	/** @name Socket messages processing functions
118	*/	118	*/
119	/@{/	119	/@{/
120		120
121	/** Processes the socket client messages.	121	/** Processes the socket client messages.
122	* Runs until the client module disconnects.	122	* Runs until the client module disconnects.
123	* @param callid The message identifier. Input parameter.	123	* @param callid The message identifier. Input parameter.
124	* @param call The message parameters. Input parameter.	124	* @param call The message parameters. Input parameter.
125	* @returns EOK on success.	125	* @returns EOK on success.
126	* @see socket.h	126	* @see socket.h
127	*/	127	*/
128	int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call );	128	int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call );
129		129
130	/** Sends data from the socket to the remote address.	130	/** Sends data from the socket to the remote address.
131	* Binds the socket to a free port if not already connected/bound.	131	* Binds the socket to a free port if not already connected/bound.
132	* Handles the NET_SOCKET_SENDTO message.	132	* Handles the NET_SOCKET_SENDTO message.
133	* Supports AF_INET and AF_INET6 address families.	133	* Supports AF_INET and AF_INET6 address families.
134	* @param local_sockets The application local sockets. Input/output parameter.	134	* @param local_sockets The application local sockets. Input/output parameter.
135	* @param socket_id Socket identifier. Input parameter.	135	* @param socket_id Socket identifier. Input parameter.
136	* @param addr The destination address. Input parameter.	136	* @param addr The destination address. Input parameter.
137	* @param addrlen The address length. Input parameter.	137	* @param addrlen The address length. Input parameter.
138	* @param fragments The number of data fragments. Input parameter.	138	* @param fragments The number of data fragments. Input parameter.
139	* @param data_fragment_size The data fragment size in bytes. Input parameter.	139	* @param data_fragment_size The data fragment size in bytes. Input parameter.
140	* @param flags Various send flags. Input parameter.	140	* @param flags Various send flags. Input parameter.
141	* @returns EOK on success.	141	* @returns EOK on success.
142	* @returns EAFNOTSUPPORT if the address family is not supported.	142	* @returns EAFNOTSUPPORT if the address family is not supported.
143	* @returns ENOTSOCK if the socket is not found.	143	* @returns ENOTSOCK if the socket is not found.
144	* @returns EINVAL if the address is invalid.	144	* @returns EINVAL if the address is invalid.
145	* @returns ENOTCONN if the sending socket is not and cannot be bound.	145	* @returns ENOTCONN if the sending socket is not and cannot be bound.
146	* @returns ENOMEM if there is not enough memory left.	146	* @returns ENOMEM if there is not enough memory left.
147	* @returns Other error codes as defined for the socket_read_packet_data() function.	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.	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.	149	* @returns Other error codes as defined for the ip_send_msg() function.
150	*/	150	*/
151	int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, size_t data_fragment_size, int flags );	151	int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, size_t data_fragment_size, int flags );
152		152
153	/** Receives data to the socket.	153	/** Receives data to the socket.
154	* Handles the NET_SOCKET_RECVFROM message.	154	* Handles the NET_SOCKET_RECVFROM message.
155	* Replies the source address as well.	155	* Replies the source address as well.
156	* @param local_sockets The application local sockets. Input parameter.	156	* @param local_sockets The application local sockets. Input parameter.
157	* @param socket_id Socket identifier. Input parameter.	157	* @param socket_id Socket identifier. Input parameter.
158	* @param flags Various receive flags. Input parameter.	158	* @param flags Various receive flags. Input parameter.
159	* @param addrlen The source address length. Output parameter.	159	* @param addrlen The source address length. Output parameter.
160	* @returns The number of bytes received.	160	* @returns The number of bytes received.
161	* @returns ENOTSOCK if the socket is not found.	161	* @returns ENOTSOCK if the socket is not found.
162	* @returns NO_DATA if there are no received packets or data.	162	* @returns NO_DATA if there are no received packets or data.
163	* @returns ENOMEM if there is not enough memory left.	163	* @returns ENOMEM if there is not enough memory left.
164	* @returns EINVAL if the received address is not an IP address.	164	* @returns EINVAL if the received address is not an IP address.
165	* @returns Other error codes as defined for the packet_translate() function.	165	* @returns Other error codes as defined for the packet_translate() function.
166	* @returns Other error codes as defined for the data_reply() function.	166	* @returns Other error codes as defined for the data_reply() function.
167	*/	167	*/
168	int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags, size_t * addrlen );	168	int udp_recvfrom_message( socket_cores_ref local_sockets, int socket_id, int flags, size_t * addrlen );
169		169
170	/@}/	170	/@}/
171		171
172	/** UDP global data.	172	/** UDP global data.
173	*/	173	*/
174	udp_globals_t udp_globals;	174	udp_globals_t udp_globals;
175		175
176	int udp_initialize( async_client_conn_t client_connection ){	176	int udp_initialize( async_client_conn_t client_connection ){
177	ERROR_DECLARE;	177	ERROR_DECLARE;
178		178
179	measured_string_t names[] = {{ "UDP_CHECKSUM_COMPUTING", 22 }};	179	measured_string_t names[] = {{ "UDP_CHECKSUM_COMPUTING", 22 }};
180	measured_string_ref configuration;	180	measured_string_ref configuration;
181	size_t count = sizeof( names ) / sizeof( measured_string_t );	181	size_t count = sizeof( names ) / sizeof( measured_string_t );
182	char * data;	182	char * data;
183		183
184	fibril_rwlock_initialize( & udp_globals.lock );	184	fibril_rwlock_initialize( & udp_globals.lock );
185	fibril_rwlock_write_lock( & udp_globals.lock );	185	fibril_rwlock_write_lock( & udp_globals.lock );
186	udp_globals.icmp_phone = icmp_connect_module( SERVICE_ICMP );	186	udp_globals.icmp_phone = icmp_connect_module( SERVICE_ICMP );
187	if( udp_globals.icmp_phone < 0 ){	187	if( udp_globals.icmp_phone < 0 ){
188	return udp_globals.icmp_phone;	188	return udp_globals.icmp_phone;
189	}	189	}
190	udp_globals.ip_phone = ip_bind_service( SERVICE_IP, IPPROTO_UDP, SERVICE_UDP, client_connection, udp_received_msg );	190	udp_globals.ip_phone = ip_bind_service( SERVICE_IP, IPPROTO_UDP, SERVICE_UDP, client_connection, udp_received_msg );
191	if( udp_globals.ip_phone < 0 ){	191	if( udp_globals.ip_phone < 0 ){
192	return udp_globals.ip_phone;	192	return udp_globals.ip_phone;
193	}	193	}
194	ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.packet_dimension.addr_len, & udp_globals.packet_dimension.prefix, & udp_globals.packet_dimension.content, & udp_globals.packet_dimension.suffix ));	194	ERROR_PROPAGATE( ip_packet_size_req( udp_globals.ip_phone, -1, & udp_globals.packet_dimension.addr_len, & udp_globals.packet_dimension.prefix, & udp_globals.packet_dimension.content, & udp_globals.packet_dimension.suffix ));
195	ERROR_PROPAGATE( socket_ports_initialize( & udp_globals.sockets ));	195	ERROR_PROPAGATE( socket_ports_initialize( & udp_globals.sockets ));
196	udp_globals.packet_dimension.prefix += sizeof( udp_header_t );	196	udp_globals.packet_dimension.prefix += sizeof( udp_header_t );
197	udp_globals.packet_dimension.content -= sizeof( udp_header_t );	197	udp_globals.packet_dimension.content -= sizeof( udp_header_t );
198	udp_globals.last_used_port = UDP_FREE_PORTS_START - 1;	198	udp_globals.last_used_port = UDP_FREE_PORTS_START - 1;
199	// get configuration	199	// get configuration
200	udp_globals.checksum_computing = NET_DEFAULT_UDP_CHECKSUM_COMPUTING;	200	udp_globals.checksum_computing = NET_DEFAULT_UDP_CHECKSUM_COMPUTING;
201	configuration = & names[ 0 ];	201	configuration = & names[ 0 ];
202	ERROR_PROPAGATE( net_get_conf_req( udp_globals.net_phone, & configuration, count, & data ));	202	ERROR_PROPAGATE( net_get_conf_req( udp_globals.net_phone, & configuration, count, & data ));
203	if( configuration ){	203	if( configuration ){
204	if( configuration[ 0 ].value ){	204	if( configuration[ 0 ].value ){
205	udp_globals.checksum_computing = ( configuration[ 0 ].value[ 0 ] == 'y' );	205	udp_globals.checksum_computing = ( configuration[ 0 ].value[ 0 ] == 'y' );
206	}	206	}
207	net_free_settings( configuration, data );	207	net_free_settings( configuration, data );
208	}	208	}
209	fibril_rwlock_write_unlock( & udp_globals.lock );	209	fibril_rwlock_write_unlock( & udp_globals.lock );
210	return EOK;	210	return EOK;
211	}	211	}
212		212
213	int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error ){	213	int udp_received_msg( device_id_t device_id, packet_t packet, services_t receiver, services_t error ){
214	ERROR_DECLARE;	214	ERROR_DECLARE;
215		215
216	size_t length;	216	size_t length;
217	size_t offset;	217	size_t offset;
218	int result;	218	int result;
219	uint8_t * data;	219	uint8_t * data;
220	udp_header_ref header;	220	udp_header_ref header;
221	socket_core_ref * socket;	221	socket_core_ref * socket;
222	packet_t next_packet;	222	packet_t next_packet;
223	size_t total_length;	223	size_t total_length;
224	uint32_t checksum;	224	uint32_t checksum;
225	int fragments;	225	int fragments;
226	packet_t tmp_packet;	226	packet_t tmp_packet;
227	icmp_type_t type;	227	icmp_type_t type;
228	icmp_code_t code;	228	icmp_code_t code;
229	ip_pseudo_header_ref ip_header;	229	ip_pseudo_header_ref ip_header;
230	struct sockaddr * src;	230	struct sockaddr * src;
231	struct sockaddr * dest;	231	struct sockaddr * dest;
232		232
233	if( error ){	233	if( error ){
234	switch( error ){	234	switch( error ){
235	case SERVICE_ICMP:	235	case SERVICE_ICMP:
236	// process error	236	// process error
237	// TODO remove debug dump	237	// TODO remove debug dump
238	// length = icmp_client_header_length( packet );	238	// length = icmp_client_header_length( packet );
239	result = icmp_client_process_packet( packet, & type, & code, NULL, NULL );	239	result = icmp_client_process_packet( packet, & type, & code, NULL, NULL );
240	if( result < 0 ){	240	if( result < 0 ){
241	return udp_release_and_return( packet, result );	241	return udp_release_and_return( packet, result );
242	}	242	}
243	printf( "ICMP error %d (%d) in packet %d\n", type, code, packet_get_id( packet ) );	243	printf( "ICMP error %d (%d) in packet %d\n", type, code, packet_get_id( packet ) );
244	length = ( size_t ) result;	244	length = ( size_t ) result;
245	if( ERROR_OCCURRED( packet_trim( packet, length, 0 ))){	245	if( ERROR_OCCURRED( packet_trim( packet, length, 0 ))){
246	return udp_release_and_return( packet, ERROR_CODE );	246	return udp_release_and_return( packet, ERROR_CODE );
247	}	247	}
248	break;	248	break;
249	default:	249	default:
250	return udp_release_and_return( packet, ENOTSUP );	250	return udp_release_and_return( packet, ENOTSUP );
251	}	251	}
252	}	252	}
253	// TODO process received ipopts?	253	// TODO process received ipopts?
254	result = ip_client_process_packet( packet, NULL, NULL, NULL, NULL, NULL );	254	result = ip_client_process_packet( packet, NULL, NULL, NULL, NULL, NULL );
255	if( result < 0 ){	255	if( result < 0 ){
256	return udp_release_and_return( packet, result );	256	return udp_release_and_return( packet, result );
257	}	257	}
258	offset = ( size_t ) result;	258	offset = ( size_t ) result;
259		259
260	length = packet_get_data_length( packet );	260	length = packet_get_data_length( packet );
261	if( length <= 0 ){	261	if( length <= 0 ){
262	return udp_release_and_return( packet, EINVAL );	262	return udp_release_and_return( packet, EINVAL );
263	}	263	}
264	if( length < sizeof( udp_header_t ) + offset ){	264	if( length < sizeof( udp_header_t ) + offset ){
265	return udp_release_and_return( packet, NO_DATA );	265	return udp_release_and_return( packet, NO_DATA );
266	}	266	}
267	data = packet_get_data( packet );	267	data = packet_get_data( packet );
268	if( ! data ){	268	if( ! data ){
269	return udp_release_and_return( packet, NO_DATA );	269	return udp_release_and_return( packet, NO_DATA );
270	}	270	}
271	// get udp header	271	// get udp header
272	header = ( udp_header_ref )( data + offset );	272	header = ( udp_header_ref )( data + offset );
273	// find the destination socket	273	// find the destination socket
274	socket = socket_ports_find( & udp_globals.sockets, ntohs( header->dest ));	274	socket = socket_ports_find( & udp_globals.sockets, ntohs( header->destination_port ));
275	if( ! socket ){	275	if( ! socket ){
276	tl_send_icmp_port_unreachable( udp_globals.net_phone, udp_globals.icmp_phone, packet, error );	276	tl_send_icmp_port_unreachable( udp_globals.net_phone, udp_globals.icmp_phone, packet, error );
277	return EADDRNOTAVAIL;	277	return EADDRNOTAVAIL;
278	}	278	}
279	// trim after successful processing to be able to send an ICMP error message!	279	// trim after successful processing to be able to send an ICMP error message!
280	ERROR_PROPAGATE( packet_trim( packet, offset, 0 ));	280	ERROR_PROPAGATE( packet_trim( packet, offset, 0 ));
281	// count the received packet fragments	281	// count the received packet fragments
282	next_packet = packet;	282	next_packet = packet;
283	fragments = 0;	283	fragments = 0;
284	total_length = ntohs( header->len );	284	total_length = ntohs( header->total_length );
285	// compute header checksum if set	285	// compute header checksum if set
286	if( header->check && ( ! error )){	286	if( header->checksum && ( ! error )){
287	result = packet_get_addr( packet, ( uint8_t ) & src, ( uint8_t ) & dest );	287	result = packet_get_addr( packet, ( uint8_t ) & src, ( uint8_t ) & dest );
288	if( result <= 0 ){	288	if( result <= 0 ){
289	return udp_release_and_return( packet, result );	289	return udp_release_and_return( packet, result );
290	}	290	}
291	if( ERROR_OCCURRED( ip_client_get_pseudo_header( IPPROTO_UDP, src, result, dest, result, total_length, & ip_header, & length ))){	291	if( ERROR_OCCURRED( ip_client_get_pseudo_header( IPPROTO_UDP, src, result, dest, result, total_length, & ip_header, & length ))){
292	return udp_release_and_return( packet, ERROR_CODE );	292	return udp_release_and_return( packet, ERROR_CODE );
293	}else{	293	}else{
294	checksum = compute_checksum( 0, ip_header, length );	294	checksum = compute_checksum( 0, ip_header, length );
295	// the udp header checksum will be added with the first fragment later	295	// the udp header checksum will be added with the first fragment later
296	free( ip_header );	296	free( ip_header );
297	}	297	}
298	}else{	298	}else{
299	header->check = 0;	299	header->checksum = 0;
300	checksum = 0;	300	checksum = 0;
301	}	301	}
302	do{	302	do{
303	++ fragments;	303	++ fragments;
304	length = packet_get_data_length( next_packet );	304	length = packet_get_data_length( next_packet );
305	if( length <= 0 ){	305	if( length <= 0 ){
306	return udp_release_and_return( packet, NO_DATA );	306	return udp_release_and_return( packet, NO_DATA );
307	}	307	}
308	if( total_length < length ){	308	if( total_length < length ){
309	if( ERROR_OCCURRED( packet_trim( next_packet, 0, length - total_length ))){	309	if( ERROR_OCCURRED( packet_trim( next_packet, 0, length - total_length ))){
310	return udp_release_and_return( packet, ERROR_CODE );	310	return udp_release_and_return( packet, ERROR_CODE );
311	}	311	}
312	// add partial checksum if set	312	// add partial checksum if set
313	if( header->check ){	313	if( header->checksum ){
314	checksum = compute_checksum( checksum, packet_get_data( packet ), packet_get_data_length( packet ));	314	checksum = compute_checksum( checksum, packet_get_data( packet ), packet_get_data_length( packet ));
315	}	315	}
316	// relese the rest of the packet fragments	316	// relese the rest of the packet fragments
317	tmp_packet = pq_next( next_packet );	317	tmp_packet = pq_next( next_packet );
318	while( tmp_packet ){	318	while( tmp_packet ){
319	next_packet = pq_detach( tmp_packet );	319	next_packet = pq_detach( tmp_packet );
320	pq_release( udp_globals.net_phone, packet_get_id( tmp_packet ));	320	pq_release( udp_globals.net_phone, packet_get_id( tmp_packet ));
321	tmp_packet = next_packet;	321	tmp_packet = next_packet;
322	}	322	}
323	// exit the loop	323	// exit the loop
324	break;	324	break;
325	}	325	}
326	total_length -= length;	326	total_length -= length;
327	// add partial checksum if set	327	// add partial checksum if set
328	if( header->check ){	328	if( header->checksum ){
329	checksum = compute_checksum( checksum, packet_get_data( packet ), packet_get_data_length( packet ));	329	checksum = compute_checksum( checksum, packet_get_data( packet ), packet_get_data_length( packet ));
330	}	330	}
331	}while(( next_packet = pq_next( next_packet )) && ( total_length > 0 ));	331	}while(( next_packet = pq_next( next_packet )) && ( total_length > 0 ));
332	// check checksum	332	// check checksum
333	if( header->check ){	333	if( header->checksum ){
334	if( flip_checksum( compact_checksum( checksum ))){	334	if( flip_checksum( compact_checksum( checksum ))){
335	// TODO checksum error ICMP?	335	// TODO checksum error ICMP?
336	// TODO remove debug dump	336	// TODO remove debug dump
337	printf("udp check failed %x => %x\n", header->check, flip_checksum( compact_checksum( checksum )));	337	printf("udp check failed %x => %x\n", header->checksum, flip_checksum( compact_checksum( checksum )));
338	return udp_release_and_return( packet, EINVAL );	338	return udp_release_and_return( packet, EINVAL );
339	}	339	}
340	}	340	}
341	// queue the received packet	341	// queue the received packet
342	if( ERROR_OCCURRED( dyn_fifo_push( &( ** socket ).received, packet_get_id( packet ), SOCKET_MAX_RECEIVED_SIZE ))){	342	if( ERROR_OCCURRED( dyn_fifo_push( &( ** socket ).received, packet_get_id( packet ), SOCKET_MAX_RECEIVED_SIZE ))){
343	return udp_release_and_return( packet, ERROR_CODE );	343	return udp_release_and_return( packet, ERROR_CODE );
344	}	344	}
345		345
346	// notify the destination socket	346	// notify the destination socket
347	async_msg_5(( socket ).phone, NET_SOCKET_RECEIVED, ( ipcarg_t ) ( socket ).socket_id, 0, 0, 0, ( ipcarg_t ) fragments );	347	async_msg_5(( socket ).phone, NET_SOCKET_RECEIVED, ( ipcarg_t ) ( socket ).socket_id, 0, 0, 0, ( ipcarg_t ) fragments );
348	return EOK;	348	return EOK;
349	}	349	}
350		350
351	int udp_message( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ){	351	int udp_message( ipc_callid_t callid, ipc_call_t * call, ipc_call_t * answer, int * answer_count ){
352	ERROR_DECLARE;	352	ERROR_DECLARE;
353		353
354	packet_t packet;	354	packet_t packet;
355		355
356	* answer_count = 0;	356	* answer_count = 0;
357	switch( IPC_GET_METHOD( * call )){	357	switch( IPC_GET_METHOD( * call )){
358	case NET_TL_RECEIVED:	358	case NET_TL_RECEIVED:
359	fibril_rwlock_read_lock( & udp_globals.lock );	359	fibril_rwlock_read_lock( & udp_globals.lock );
360	if( ! ERROR_OCCURRED( packet_translate( udp_globals.net_phone, & packet, IPC_GET_PACKET( call )))){	360	if( ! ERROR_OCCURRED( packet_translate( udp_globals.net_phone, & packet, IPC_GET_PACKET( call )))){
361	ERROR_CODE = udp_received_msg( IPC_GET_DEVICE( call ), packet, SERVICE_UDP, IPC_GET_ERROR( call ));	361	ERROR_CODE = udp_received_msg( IPC_GET_DEVICE( call ), packet, SERVICE_UDP, IPC_GET_ERROR( call ));
362	}	362	}
363	fibril_rwlock_read_unlock( & udp_globals.lock );	363	fibril_rwlock_read_unlock( & udp_globals.lock );
364	return ERROR_CODE;	364	return ERROR_CODE;
365	case IPC_M_CONNECT_TO_ME:	365	case IPC_M_CONNECT_TO_ME:
366	return udp_process_client_messages( callid, * call );	366	return udp_process_client_messages( callid, * call );
367	}	367	}
368	return ENOTSUP;	368	return ENOTSUP;
369	}	369	}
370		370
371	int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call ){	371	int udp_process_client_messages( ipc_callid_t callid, ipc_call_t call ){
372	int res;	372	int res;
373	bool keep_on_going = true;	373	bool keep_on_going = true;
374	socket_cores_t local_sockets;	374	socket_cores_t local_sockets;
375	int app_phone = IPC_GET_PHONE( & call );	375	int app_phone = IPC_GET_PHONE( & call );
376	struct sockaddr * addr;	376	struct sockaddr * addr;
377	size_t addrlen;	377	size_t addrlen;
378	// fibril_rwlock_t lock;	378	// fibril_rwlock_t lock;
379	ipc_call_t answer;	379	ipc_call_t answer;
380	int answer_count;	380	int answer_count;
381		381
382	/*	382	/*
383	* Accept the connection	383	* Accept the connection
384	* - Answer the first IPC_M_CONNECT_ME_TO call.	384	* - Answer the first IPC_M_CONNECT_ME_TO call.
385	*/	385	*/
386	ipc_answer_0( callid, EOK );	386	ipc_answer_0( callid, EOK );
387		387
388	// The client connection is only in one fibril and therefore no additional locks are needed.	388	// The client connection is only in one fibril and therefore no additional locks are needed.
389		389
390	socket_cores_initialize( & local_sockets );	390	socket_cores_initialize( & local_sockets );
391	// fibril_rwlock_initialize( & lock );	391	// fibril_rwlock_initialize( & lock );
392		392
393	while( keep_on_going ){	393	while( keep_on_going ){
394	// refresh data	394	// refresh data
395	refresh_answer( & answer, & answer_count );	395	refresh_answer( & answer, & answer_count );
396		396
397	callid = async_get_call( & call );	397	callid = async_get_call( & call );
398	// printf( "message %d\n", IPC_GET_METHOD( * call ));	398	// printf( "message %d\n", IPC_GET_METHOD( * call ));
399		399
400	switch( IPC_GET_METHOD( call )){	400	switch( IPC_GET_METHOD( call )){
401	case IPC_M_PHONE_HUNGUP:	401	case IPC_M_PHONE_HUNGUP:
402	keep_on_going = false;	402	keep_on_going = false;
403	res = EOK;	403	res = EOK;
404	break;	404	break;
405	case NET_SOCKET:	405	case NET_SOCKET:
406	// fibril_rwlock_write_lock( & lock );	406	// fibril_rwlock_write_lock( & lock );
407	res = socket_create( & local_sockets, app_phone, NULL, SOCKET_SET_SOCKET_ID( answer ));	407	res = socket_create( & local_sockets, app_phone, NULL, SOCKET_SET_SOCKET_ID( answer ));
408	// fibril_rwlock_write_unlock( & lock );	408	// fibril_rwlock_write_unlock( & lock );
409	* SOCKET_SET_DATA_FRAGMENT_SIZE( answer ) = MAX_UDP_FRAGMENT_SIZE;	409	* SOCKET_SET_DATA_FRAGMENT_SIZE( answer ) = MAX_UDP_FRAGMENT_SIZE;
410	* SOCKET_SET_HEADER_SIZE( answer ) = sizeof( udp_header_t );	410	* SOCKET_SET_HEADER_SIZE( answer ) = sizeof( udp_header_t );
411	answer_count = 3;	411	answer_count = 3;
412	break;	412	break;
413	case NET_SOCKET_BIND:	413	case NET_SOCKET_BIND:
414	res = data_receive(( void ** ) & addr, & addrlen );	414	res = data_receive(( void ** ) & addr, & addrlen );
415	if( res == EOK ){	415	if( res == EOK ){
416	// fibril_rwlock_write_lock( & lock );	416	// fibril_rwlock_write_lock( & lock );
417	fibril_rwlock_write_lock( & udp_globals.lock );	417	fibril_rwlock_write_lock( & udp_globals.lock );
418	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 );	418	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 );
419	fibril_rwlock_write_unlock( & udp_globals.lock );	419	fibril_rwlock_write_unlock( & udp_globals.lock );
420	// fibril_rwlock_write_unlock( & lock );	420	// fibril_rwlock_write_unlock( & lock );
421	free( addr );	421	free( addr );
422	}	422	}
423	break;	423	break;
424	case NET_SOCKET_SENDTO:	424	case NET_SOCKET_SENDTO:
425	res = data_receive(( void ** ) & addr, & addrlen );	425	res = data_receive(( void ** ) & addr, & addrlen );
426	if( res == EOK ){	426	if( res == EOK ){
427	// fibril_rwlock_read_lock( & lock );	427	// fibril_rwlock_read_lock( & lock );
428	fibril_rwlock_read_lock( & udp_globals.lock );	428	fibril_rwlock_read_lock( & udp_globals.lock );
429	res = udp_sendto_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), addr, addrlen, SOCKET_GET_DATA_FRAGMENTS( call ), SOCKET_GET_DATA_FRAGMENT_SIZE( call ), SOCKET_GET_FLAGS( call ));	429	res = udp_sendto_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), addr, addrlen, SOCKET_GET_DATA_FRAGMENTS( call ), SOCKET_GET_DATA_FRAGMENT_SIZE( call ), SOCKET_GET_FLAGS( call ));
430	fibril_rwlock_read_unlock( & udp_globals.lock );	430	fibril_rwlock_read_unlock( & udp_globals.lock );
431	// fibril_rwlock_read_unlock( & lock );	431	// fibril_rwlock_read_unlock( & lock );
432	free( addr );	432	free( addr );
433	}	433	}
434	break;	434	break;
435	case NET_SOCKET_RECVFROM:	435	case NET_SOCKET_RECVFROM:
436	// fibril_rwlock_read_lock( & lock );	436	// fibril_rwlock_read_lock( & lock );
437	fibril_rwlock_read_lock( & udp_globals.lock );	437	fibril_rwlock_read_lock( & udp_globals.lock );
438	res = udp_recvfrom_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), SOCKET_GET_FLAGS( call ), & addrlen );	438	res = udp_recvfrom_message( & local_sockets, SOCKET_GET_SOCKET_ID( call ), SOCKET_GET_FLAGS( call ), & addrlen );
439	fibril_rwlock_read_unlock( & udp_globals.lock );	439	fibril_rwlock_read_unlock( & udp_globals.lock );
440	// fibril_rwlock_read_unlock( & lock );	440	// fibril_rwlock_read_unlock( & lock );
441	if( res > 0 ){	441	if( res > 0 ){
442	* SOCKET_SET_READ_DATA_LENGTH( answer ) = res;	442	* SOCKET_SET_READ_DATA_LENGTH( answer ) = res;
443	* SOCKET_SET_ADDRESS_LENGTH( answer ) = addrlen;	443	* SOCKET_SET_ADDRESS_LENGTH( answer ) = addrlen;
444	answer_count = 2;	444	answer_count = 2;
445	res = EOK;	445	res = EOK;
446	}	446	}
447	break;	447	break;
448	case NET_SOCKET_CLOSE:	448	case NET_SOCKET_CLOSE:
449	// fibril_rwlock_write_lock( & lock );	449	// fibril_rwlock_write_lock( & lock );
450	fibril_rwlock_write_lock( & udp_globals.lock );	450	fibril_rwlock_write_lock( & udp_globals.lock );
451	res = socket_destroy( udp_globals.net_phone, SOCKET_GET_SOCKET_ID( call ), & local_sockets, & udp_globals.sockets );	451	res = socket_destroy( udp_globals.net_phone, SOCKET_GET_SOCKET_ID( call ), & local_sockets, & udp_globals.sockets );
452	fibril_rwlock_write_unlock( & udp_globals.lock );	452	fibril_rwlock_write_unlock( & udp_globals.lock );
453	// fibril_rwlock_write_unlock( & lock );	453	// fibril_rwlock_write_unlock( & lock );
454	break;	454	break;
455	case NET_SOCKET_GETSOCKOPT:	455	case NET_SOCKET_GETSOCKOPT:
456	case NET_SOCKET_SETSOCKOPT:	456	case NET_SOCKET_SETSOCKOPT:
457	default:	457	default:
458	res = ENOTSUP;	458	res = ENOTSUP;
459	break;	459	break;
460	}	460	}
461		461
462	// printf( "res = %d\n", res );	462	// printf( "res = %d\n", res );
463		463
464	answer_call( callid, res, & answer, answer_count );	464	answer_call( callid, res, & answer, answer_count );
465	}	465	}
466		466
467	// TODO call socket_destroy() on all bound!	467	// TODO call socket_destroy() on all bound!
468	socket_cores_destroy( & local_sockets );	468	socket_cores_destroy( & local_sockets );
469		469
470	return EOK;	470	return EOK;
471	}	471	}
472		472
473	int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, size_t data_fragment_size, int flags ){	473	int udp_sendto_message( socket_cores_ref local_sockets, int socket_id, const struct sockaddr * addr, socklen_t addrlen, int fragments, size_t data_fragment_size, int flags ){
474	ERROR_DECLARE;	474	ERROR_DECLARE;
475		475
476	socket_core_ref socket;	476	socket_core_ref socket;
477	packet_t packet;	477	packet_t packet;
478	packet_t next_packet;	478	packet_t next_packet;
479	udp_header_ref header;	479	udp_header_ref header;
480	int index;	480	int index;
481	size_t total_length;	481	size_t total_length;
482	int result;	482	int result;
483	uint16_t dest_port;	483	uint16_t dest_port;
484	uint32_t checksum;	484	uint32_t checksum;
485	ip_pseudo_header_ref ip_header;	485	ip_pseudo_header_ref ip_header;
486	size_t headerlen;	486	size_t headerlen;
487	device_id_t device_id;	487	device_id_t device_id;
488		488
489	ERROR_PROPAGATE( tl_get_address_port( addr, addrlen, & dest_port ));	489	ERROR_PROPAGATE( tl_get_address_port( addr, addrlen, & dest_port ));
490		490
491	socket = socket_cores_find( local_sockets, socket_id );	491	socket = socket_cores_find( local_sockets, socket_id );
492	if( ! socket ) return ENOTSOCK;	492	if( ! socket ) return ENOTSOCK;
493		493
494	// bind the socket to a random free port if not bound	494	// bind the socket to a random free port if not bound
495	while( socket->port <= 0 ){	495	while( socket->port <= 0 ){
496	// try to find a free port	496	// try to find a free port
497	fibril_rwlock_read_unlock( & udp_globals.lock );	497	fibril_rwlock_read_unlock( & udp_globals.lock );
498	fibril_rwlock_write_lock( & udp_globals.lock );	498	fibril_rwlock_write_lock( & udp_globals.lock );
499	if( socket->port <= 0 ){	499	if( socket->port <= 0 ){
500	ERROR_PROPAGATE( socket_bind_free_port( & udp_globals.sockets, socket, UDP_FREE_PORTS_START, UDP_FREE_PORTS_END, udp_globals.last_used_port ));	500	ERROR_PROPAGATE( socket_bind_free_port( & udp_globals.sockets, socket, UDP_FREE_PORTS_START, UDP_FREE_PORTS_END, udp_globals.last_used_port ));

Subversion Repositories HelenOS

(root)/branches/network/uspace/srv/net/tl/udp/udp.c – Rev 4726 → 4728