Subversion Repositories HelenOS

Rev

Rev 4726 | Rev 4738 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  1. /*
  2.  * Copyright (c) 2009 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 socket
  30.  *  @{
  31.  */
  32.  
  33. /** @file
  34.  *  \todo
  35.  */
  36.  
  37. #include "../err.h"
  38.  
  39. #include "../include/in.h"
  40. #include "../include/inet.h"
  41.  
  42. #include "../include/socket_codes.h"
  43. #include "../include/socket_errno.h"
  44.  
  45. #include "../structures/dynamic_fifo.h"
  46. #include "../structures/int_map.h"
  47. #include "../structures/packet/packet.h"
  48. #include "../structures/packet/packet_client.h"
  49.  
  50. #include "socket_core.h"
  51.  
  52. int socket_bind_insert( socket_ports_ref global_sockets, socket_core_ref socket, int port );
  53.  
  54. INT_MAP_IMPLEMENT( socket_cores, socket_core_t );
  55.  
  56. INT_MAP_IMPLEMENT( socket_ports, socket_core_ref );
  57.  
  58. int socket_bind( socket_cores_ref local_sockets, socket_ports_ref global_sockets, int socket_id, void * addr, size_t addrlen, int free_ports_start, int free_ports_end, int last_used_port ){
  59.     socket_core_ref         socket;
  60.     socket_core_ref *       socket_pointer;
  61.     struct sockaddr *       address;
  62.     struct sockaddr_in *    address_in;
  63.  
  64.     if( addrlen < sizeof( struct sockaddr )) return EINVAL;
  65.     address = ( struct sockaddr * ) addr;
  66.     switch( address->sa_family ){
  67.         case AF_INET:
  68.             if( addrlen != sizeof( struct sockaddr_in )) return EINVAL;
  69.             address_in = ( struct sockaddr_in * ) addr;
  70.             // find the socket
  71.             socket = socket_cores_find( local_sockets, socket_id );
  72.             if( ! socket ) return ENOTSOCK;
  73.             // bind a free port?
  74.             if( address_in->sin_port <= 0 ){
  75.                 return socket_bind_free_port( global_sockets, socket, free_ports_start, free_ports_end, last_used_port );
  76.             }
  77.             // try to find the port
  78.             socket_pointer = socket_ports_find( global_sockets, address_in->sin_port );
  79.             if( socket_pointer ){
  80.                 // already used
  81.                 return EADDRINUSE;
  82.             }
  83.             // unbind if bound
  84.             socket_ports_exclude( global_sockets, socket->port );
  85.             socket->port = -1;
  86.             return socket_bind_insert( global_sockets, socket, address_in->sin_port );
  87.             break;
  88.         // TODO IPv6
  89.     }
  90.     return EAFNOSUPPORT;
  91. }
  92.  
  93. int socket_bind_free_port( socket_ports_ref global_sockets, socket_core_ref socket, int free_ports_start, int free_ports_end, int last_used_port ){
  94.     int index;
  95.  
  96.     // from the last used one
  97.     index = last_used_port;
  98.     do{
  99.         ++ index;
  100.         // til the range end
  101.         if( index >= free_ports_end ){
  102.             // start from the range beginning
  103.             index = free_ports_start - 1;
  104.             do{
  105.                 ++ index;
  106.                 // til the last used one
  107.                 if( index >= last_used_port ){
  108.                     // none found
  109.                     return ENOTCONN;
  110.                 }
  111.             }while( socket_ports_find( global_sockets, index ) != NULL );
  112.             // found, break immediately
  113.             break;
  114.         }
  115.     }while( socket_ports_find( global_sockets, index ) != NULL );
  116.     return socket_bind_insert( global_sockets, socket, index );
  117. }
  118.  
  119. int socket_bind_insert( socket_ports_ref global_sockets, socket_core_ref socket, int port ){
  120.     ERROR_DECLARE;
  121.  
  122.     socket_core_ref *   socket_pointer;
  123.  
  124.     // create a wrapper
  125.     socket_pointer = ( socket_core_ref * ) malloc( sizeof( socket_core_ref ));
  126.     if( ! socket_pointer ) return ENOMEM;
  127.     * socket_pointer = socket;
  128.     // register the incomming port
  129.     ERROR_CODE = socket_ports_add( global_sockets, port, socket_pointer );
  130.     if( ERROR_CODE < 0 ){
  131.         free( socket_pointer );
  132.         return ERROR_CODE;
  133.     }
  134.     socket->port = port;
  135.     return EOK;
  136. }
  137.  
  138. int socket_create( socket_cores_ref local_sockets, int app_phone, void * specific_data, int * socket_id ){
  139.     ERROR_DECLARE;
  140.  
  141.     socket_core_ref socket;
  142.     int             res;
  143.  
  144.     if( ! socket_id ) return EBADMEM;
  145.     socket = ( socket_core_ref ) malloc( sizeof( * socket ));
  146.     if( ! socket ) return ENOMEM;
  147.     // initialize
  148.     socket->phone = app_phone;
  149.     socket->port = -1;
  150.     socket->specific_data = specific_data;
  151.     if( ERROR_OCCURRED( dyn_fifo_initialize( & socket->received, SOCKET_INITIAL_RECEIVED_SIZE ))){
  152.         free( socket );
  153.         return ERROR_CODE;
  154.     }
  155.     if( ERROR_OCCURRED( dyn_fifo_initialize( & socket->accepted, SOCKET_INITIAL_ACCEPTED_SIZE ))){
  156.         dyn_fifo_destroy( & socket->received );
  157.         free( socket );
  158.         return ERROR_CODE;
  159.     }
  160.     // get a next free socket number
  161.     socket->socket_id = socket_cores_count( local_sockets ) + 1;
  162.     // store the socket
  163.     res = socket_cores_add( local_sockets, socket->socket_id, socket );
  164.     if( res < 0 ){
  165.         dyn_fifo_destroy( & socket->received );
  166.         dyn_fifo_destroy( & socket->accepted );
  167.         free( socket );
  168.         return res;
  169.     }
  170.     // return the socket identifier
  171.     * socket_id = socket->socket_id;
  172.     return EOK;
  173. }
  174.  
  175. int socket_destroy( int packet_phone, int socket_id, socket_cores_ref local_sockets, socket_ports_ref global_sockets ){
  176.     socket_core_ref socket;
  177.     int             accepted_id;
  178.     int             packet_id;
  179.  
  180.     // find the socket
  181.     socket = socket_cores_find( local_sockets, socket_id );
  182.     if( ! socket ) return ENOTSOCK;
  183.     socket_ports_exclude( global_sockets, socket->port );
  184.     // destroy all accepted sockets
  185.     while(( accepted_id = dyn_fifo_pop( & socket->accepted )) >= 0 ){
  186.         socket_destroy( packet_phone, accepted_id, local_sockets, global_sockets );
  187.     }
  188.     // release all received packets
  189.     while(( packet_id = dyn_fifo_pop( & socket->received )) >= 0 ){
  190.         pq_release( packet_phone, packet_id );
  191.     }
  192.     dyn_fifo_destroy( & socket->received );
  193.     dyn_fifo_destroy( & socket->accepted );
  194.     socket_cores_exclude( local_sockets, socket_id );
  195.     return EOK;
  196. }
  197.  
  198. /** @}
  199.  */
  200.