Subversion Repositories HelenOS

/*

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

 *  @{

 */

/** @file

 *  Transport layer common functions implementation.

 *  @see tl_common.h

 */

#include <async.h>

#include <ipc/services.h>

#include "../err.h"

#include "../structures/packet/packet.h"

#include "../structures/packet/packet_client.h"

#include "../include/icmp_interface.h"

#include "../include/in.h"

#include "../include/in6.h"

#include "../include/inet.h"

#include "../include/ip_interface.h"

#include "../include/socket_codes.h"

#include "../include/socket_errno.h"

#include "tl_common.h"

DEVICE_MAP_IMPLEMENT( packet_dimensions, packet_dimension_t );

int tl_get_address_port( const struct sockaddr * addr, int addrlen, uint16_t * port ){

    const struct sockaddr_in *  address_in;

    const struct sockaddr_in6 * address_in6;

    if(( addrlen <= 0 ) || (( size_t ) addrlen < sizeof( struct sockaddr ))) return EINVAL;

    switch( addr->sa_family ){

        case AF_INET:

            if( addrlen != sizeof( struct sockaddr_in )) return EINVAL;

            address_in = ( struct sockaddr_in * ) addr;

            * port = ntohs( address_in->sin_port );

            break;

        case AF_INET6:

            if( addrlen != sizeof( struct sockaddr_in6 )) return EINVAL;

            address_in6 = ( struct sockaddr_in6 * ) addr;

            * port = ntohs( address_in6->sin6_port );

            break;

        default:

            return EAFNOSUPPORT;

    }

    return EOK;

}

int tl_get_ip_packet_dimension( int ip_phone, packet_dimensions_ref packet_dimensions, device_id_t device_id, packet_dimension_ref * packet_dimension ){

    ERROR_DECLARE;

    if( ! packet_dimension ) return EBADMEM;

    * packet_dimension = packet_dimensions_find( packet_dimensions, device_id );

    if( ! * packet_dimension ){

        // ask for and remember them if not found

        * packet_dimension = malloc( sizeof( ** packet_dimension ));

        if( ! * packet_dimension ) return ENOMEM;

        if( ERROR_OCCURRED( ip_packet_size_req( ip_phone, device_id, & ( ** packet_dimension ).addr_len, & ( ** packet_dimension ).prefix, & ( ** packet_dimension ).content, & ( ** packet_dimension ).suffix ))){

            free( * packet_dimension );

            return ERROR_CODE;

        }

        ERROR_CODE = packet_dimensions_add( packet_dimensions, device_id, * packet_dimension );

        if( ERROR_CODE < 0 ){

            free( * packet_dimension );

            return ERROR_CODE;

        }

    }

    return EOK;

}

int tl_set_address_port( struct sockaddr * addr, int addrlen, uint16_t port ){

    struct sockaddr_in *    address_in;

    struct sockaddr_in6 *   address_in6;

    size_t                  length;

    if( addrlen < 0 ) return EINVAL;

    length = ( size_t ) addrlen;

    if( length < sizeof( struct sockaddr )) return EINVAL;

    switch( addr->sa_family ){

        case AF_INET:

            if( length != sizeof( struct sockaddr_in )) return EINVAL;

            address_in = ( struct sockaddr_in * ) addr;

            address_in->sin_port = htons( port );

            return EOK;

        case AF_INET6:

            if( length != sizeof( struct sockaddr_in6 )) return EINVAL;

            address_in6 = ( struct sockaddr_in6 * ) addr;

            address_in6->sin6_port = htons( port );

            return EOK;

        default:

            return EAFNOSUPPORT;

    }

}

int tl_prepare_icmp_packet( int packet_phone, int icmp_phone, packet_t packet, services_t error ){

    packet_t    next;

    uint8_t *   src;

    int         length;

    // detach the first packet and release the others

    next = pq_detach( packet );

    if( next ){

        pq_release( packet_phone, packet_get_id( next ));

    }

    length = packet_get_addr( packet, & src, NULL );

    if(( length > 0 )

    && ( ! error )

    && ( icmp_phone >= 0 )

    // set both addresses to the source one (avoids the source address deletion before setting the destination one)

    && ( packet_set_addr( packet, src, src, ( size_t ) length ) == EOK )){

        return EOK;

    }else{

        pq_release( packet_phone, packet_get_id( packet ));

    }

    return ENOENT;

}

int tl_socket_read_packet_data( int packet_phone, packet_ref packet, size_t prefix, const packet_dimension_ref dimension, const struct sockaddr * addr, socklen_t addrlen ){

    ERROR_DECLARE;

    ipc_callid_t    callid;

    size_t          length;

    void *          data;

    if( ! dimension ) return EINVAL;

    // get the data length

    if( ! ipc_data_write_receive( & callid, & length )) return EINVAL;

    // get a new packet

    * packet = packet_get_4( packet_phone, length, dimension->addr_len, prefix + dimension->prefix, dimension->suffix );

    if( ! packet ) return ENOMEM;

    // allocate space in the packet

    data = packet_suffix( * packet, length );

    if( ! data ){

        pq_release( packet_phone, packet_get_id( * packet ));

        return ENOMEM;

    }

    // read the data into the packet

    if( ERROR_OCCURRED( ipc_data_write_finalize( callid, data, length ))

    // set the packet destination address

    || ERROR_OCCURRED( packet_set_addr( * packet, NULL, ( uint8_t * ) addr, addrlen ))){

        pq_release( packet_phone, packet_get_id( * packet ));

        return ERROR_CODE;

    }

    return ( int ) length;

}

/** @}

 */