WebSVN – HelenOS – Diff – /branches/network/uspace/srv/net/socket/socket_core.c

 /*
  * 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 socket_bind_insert( socket_ports_ref global_sockets, socket_core_ref socket, int port );
 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 ){
+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 ){
-    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;
+            // bind a free port?
+            if( address_in->sin_port <= 0 ){
+                return socket_bind_free_port( global_sockets, socket, free_ports_start, free_ports_end, last_used_port );
+            }
             // 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 );
+            socket->port = -1;
+            return socket_bind_insert( global_sockets, socket, address_in->sin_port );
+            break;
+        // TODO IPv6
+        default:
+            return EAFNOSUPPORT;
+    }
+    return EOK;
+}
+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 ){
+    int index;
+    // from the last used one
+    index = last_used_port;
+    do{
+        ++ index;
+        // til the range end
+        if( index >= free_ports_end ){
+            // start from the range beginning
+            index = free_ports_start - 1;
+            do{
+                ++ index;
+                // til the last used one
+                if( index >= last_used_port ){
+                    // none found
+                    return ENOTCONN;
+                }
+            }while( socket_ports_find( global_sockets, index ) != NULL );
+            // found, break immediately
+            break;
+        }
+    }while( socket_ports_find( global_sockets, index ) != NULL );
+    return socket_bind_insert( global_sockets, socket, index );
+}
+int socket_bind_insert( socket_ports_ref global_sockets, socket_core_ref socket, int port ){
+    ERROR_DECLARE;
+    socket_core_ref *   socket_pointer;
-            // create a wrapper
+    // create a wrapper
-            socket_pointer = ( socket_core_ref * ) malloc( sizeof( * socket_pointer ));
+    socket_pointer = ( socket_core_ref * ) malloc( sizeof( * socket_pointer ));
-            if( ! socket_pointer ) return ENOMEM;
+    if( ! socket_pointer ) return ENOMEM;
-            * socket_pointer = socket;
+    * socket_pointer = socket;
-            // register the port
+    // register the incomming port
-            ERROR_CODE = socket_ports_add( global_sockets, address_in->sin_port, socket_pointer );
+    ERROR_CODE = socket_ports_add( global_sockets, port, socket_pointer );
-            if( ERROR_CODE < 0 ){
+    if( ERROR_CODE < 0 ){
-                free( socket_pointer );
+        free( socket_pointer );
+        return ERROR_CODE;
+    }
-            socket->port = address_in->sin_port;
+    socket->port = 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 )) >= 0 ){
         socket_destroy( packet_phone, accepted_id, local_sockets, global_sockets );
+    }
     // release all received packets
     while(( packet_id = dyn_fifo_pop( & socket->received )) >= 0 ){
         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;
+}
 /** @}
  */

Subversion Repositories HelenOS

(root)/branches/network/uspace/srv/net/socket/socket_core.c @ 4705 – Rev 4603 → 4700