Subversion Repositories HelenOS

/*

 * Copyright (c) 2009 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 socket

 *  @{

 */

/** @file

 */

#include "../err.h"

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

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

#include "../include/socket.h"

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

#include "../structures/dynamic_fifo.h"

#include "../structures/int_map.h"

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

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

#include "socket_core.h"

INT_MAP_IMPLEMENT( socket_cores, socket_core_t );

INT_MAP_IMPLEMENT( socket_ports, socket_core_ref );

int socket_bind( socket_cores_ref local_sockets, socket_ports_ref global_sockets, int socket_id, void * addr, size_t addrlen ){

    ERROR_DECLARE;

    socket_core_ref         socket;

    socket_core_ref *       socket_pointer;

    struct sockaddr *       address;

    struct sockaddr_in *    address_in;

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

    address = ( struct sockaddr * ) addr;

    switch( address->sa_family ){

        case AF_INET:

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

            address_in = ( struct sockaddr_in * ) addr;

            // find the socket

            socket = socket_cores_find( local_sockets, socket_id );

            if( ! socket ) return ENOTSOCK;

            // try to find the port

            socket_pointer = socket_ports_find( global_sockets, address_in->sin_port );

            if( socket_pointer ){

                // already used

                return EADDRINUSE;

            }

            // disbind if bound

            socket_ports_exclude( global_sockets, socket->port );

            // create a wrapper

            socket_pointer = ( socket_core_ref * ) malloc( sizeof( * socket_pointer ));

            if( ! socket_pointer ) return ENOMEM;

            * socket_pointer = socket;

            // register the port

            ERROR_CODE = socket_ports_add( global_sockets, address_in->sin_port, socket_pointer );

            if( ERROR_CODE < 0 ){

                free( socket_pointer );

            }

            socket->port = address_in->sin_port;

            break;

        // TODO IPv6

        default:

            return EAFNOSUPPORT;

    }

    return EOK;

}

int socket_create( socket_cores_ref local_sockets, int app_phone, int * socket_id ){

    ERROR_DECLARE;

    socket_core_ref socket;

    int             res;

    if( ! socket_id ) return EBADMEM;

    socket = ( socket_core_ref ) malloc( sizeof( * socket ));

    if( ! socket ) return ENOMEM;

    // initialize

    socket->phone = app_phone;

    socket->port = -1;

    socket->device_id = -1;

    socket->peer_addr = NULL;

    if( ERROR_OCCURRED( dyn_fifo_initialize( & socket->received, SOCKET_INITIAL_RECEIVED_SIZE ))){

        free( socket );

        return ERROR_CODE;

    }

    if( ERROR_OCCURRED( dyn_fifo_initialize( & socket->accepted, SOCKET_INITIAL_ACCEPTED_SIZE ))){

        dyn_fifo_destroy( & socket->received );

        free( socket );

        return ERROR_CODE;

    }

    // get a next free socket number

    socket->socket_id = socket_cores_count( local_sockets ) + 1;

    // store the socket

    res = socket_cores_add( local_sockets, socket->socket_id, socket );

    if( res < 0 ){

        dyn_fifo_destroy( & socket->received );

        dyn_fifo_destroy( & socket->accepted );

        free( socket );

        return res;

    }

    // return the socket identifier

    * socket_id = socket->socket_id;

    return EOK;

}

int socket_destroy( int packet_phone, int socket_id, socket_cores_ref local_sockets, socket_ports_ref global_sockets ){

    socket_core_ref socket;

    int             accepted_id;

    int             packet_id;

    // find the socket

    socket = socket_cores_find( local_sockets, socket_id );

    if( ! socket ) return ENOTSOCK;

    socket_ports_exclude( global_sockets, socket->port );

    // destroy all accepted sockets

    while(( accepted_id = dyn_fifo_pop( & socket->accepted ))){

        socket_destroy( packet_phone, accepted_id, local_sockets, global_sockets );

    }

    // release all received packets

    while(( packet_id = dyn_fifo_pop( & socket->received ))){

        pq_release( packet_phone, packet_id );

    }

    dyn_fifo_destroy( & socket->received );

    dyn_fifo_destroy( & socket->accepted );

    socket_cores_exclude( local_sockets, socket_id );

    return EOK;

}

/** @}

 */