Bugfixing and testing TCP stack
[tpg/acess2.git] / Modules / IPStack / main.c
index cf9adf6..7ec7689 100644 (file)
@@ -3,7 +3,7 @@
  * - Stack Initialisation
  */
 #define DEBUG  0
-#define VERSION        ((0<<8)|10)
+#define VERSION        VER2(0,10)
 #include "ipstack.h"
 #include "link.h"
 #include <modules.h>
 #include <tpl_drv_common.h>
 #include <tpl_drv_network.h>
 
+// === CONSTANTS ===
+//! Default timeout value, 30 seconds
+#define DEFAULT_TIMEOUT        (30*1000)
+
 // === IMPORTS ===
 extern int     ARP_Initialise();
+extern void    UDP_Initialise();
+extern void    TCP_Initialise();
 extern int     IPv4_Initialise();
+extern int     IPv4_Ping(tInterface *Iface, tIPv4 Addr);
 
 // === PROTOTYPES ===
  int   IPStack_Install(char **Arguments);
-char   *IPStack_ReadDir(tVFS_Node *Node, int Pos);
-tVFS_Node      *IPStack_FindDir(tVFS_Node *Node, char *Name);
- int   IPStack_IOCtl(tVFS_Node *Node, int ID, void *Data);
+ int   IPStack_IOCtlRoot(tVFS_Node *Node, int ID, void *Data);
+char   *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos);
+tVFS_Node      *IPStack_Root_FindDir(tVFS_Node *Node, char *Name);
+ int   IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data);
  int   IPStack_AddInterface(char *Device);
 tAdapter       *IPStack_GetAdapter(char *Path);
+char   *IPStack_Iface_ReadDir(tVFS_Node *Node, int Pos);
+tVFS_Node      *IPStack_Iface_FindDir(tVFS_Node *Node, char *Name);
+ int   IPStack_Iface_IOCtl(tVFS_Node *Node, int ID, void *Data);
 
 // === GLOBALS ===
 MODULE_DEFINE(0, VERSION, IPStack, IPStack_Install, NULL, NULL);
@@ -32,9 +43,9 @@ tDevFS_Driver gIP_DriverInfo = {
        .NumACLs = 1,
        .ACLs = &gVFS_ACL_EveryoneRX,
        .Flags = VFS_FFLAG_DIRECTORY,
-       .ReadDir = IPStack_ReadDir,
-       .FindDir = IPStack_FindDir,
-       .IOCtl = IPStack_IOCtl
+       .ReadDir = IPStack_Root_ReadDir,
+       .FindDir = IPStack_Root_FindDir,
+       .IOCtl = IPStack_Root_IOCtl
        }
 };
 tSpinlock      glIP_Interfaces = 0;
@@ -43,6 +54,7 @@ tInterface    *gIP_Interfaces_Last = NULL;
  int   giIP_NextIfaceId = 1;
 tSpinlock      glIP_Adapters = 0;
 tAdapter       *gIP_Adapters = NULL;
+tSocketFile    *gIP_FileTemplates;
 
 // === CODE ===
 /**
@@ -56,6 +68,8 @@ int IPStack_Install(char **Arguments)
        // Install Handlers
        ARP_Initialise();
        IPv4_Initialise();
+       TCP_Initialise();
+       UDP_Initialise();
        
        if(Arguments)
        {
@@ -73,13 +87,24 @@ int IPStack_Install(char **Arguments)
        return 1;
 }
 
+/**
+ * \brief Adds a file to the socket list
+ */
+int IPStack_AddFile(tSocketFile *File)
+{
+       Log("IPStack_AddFile: %s", File->Name);
+       File->Next = gIP_FileTemplates;
+       gIP_FileTemplates = File;
+       return 0;
+}
+
 /**
  * \brief Read from the IP Stack's Device Directory
  */
-char *IPStack_ReadDir(tVFS_Node *Node, int Pos)
+char *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos)
 {
        tInterface      *iface;
-       char    name[] = "ip0\0\0\0";
+       char    *name;
        ENTER("pNode iPos", Node, Pos);
        
        // Traverse the list
@@ -91,41 +116,42 @@ char *IPStack_ReadDir(tVFS_Node *Node, int Pos)
                return NULL;
        }
        
+       name = malloc(4);
+       
        // Create the name
        Pos = iface->Node.ImplInt;
-       if(Pos < 10)
-               name[2] = '0' + Pos;
+       if(Pos < 10) {
+               name[0] = '0' + Pos;
+               name[1] = '\0';
+       }
        else if(Pos < 100) {
-               name[2] = '0' + Pos/10;
-               name[3] = '0' + Pos%10;
+               name[0] = '0' + Pos/10;
+               name[1] = '0' + Pos%10;
+               name[2] = '\0';
        }
        else {
-               name[2] = '0' + Pos/100;
-               name[3] = '0' + (Pos/10)%10;
-               name[4] = '0' + Pos%10;
+               name[0] = '0' + Pos/100;
+               name[1] = '0' + (Pos/10)%10;
+               name[2] = '0' + Pos%10;
+               name[3] = '\0';
        }
        
        LEAVE('s', name);
        // Return the pre-generated name
-       return strdup(name);
+       return name;
 }
 
 /**
  * \brief Get the node of an interface
  */
-tVFS_Node *IPStack_FindDir(tVFS_Node *Node, char *Name)
+tVFS_Node *IPStack_Root_FindDir(tVFS_Node *Node, char *Name)
 {
         int    i, num;
        tInterface      *iface;
        
        ENTER("pNode sName", Node, Name);
        
-       if(Name[0] != 'i' || Name[1] != 'p') {
-               LEAVE('n');
-               return NULL;
-       }
-       
-       i = 2;  num = 0;
+       i = 0;  num = 0;
        while('0' <= Name[i] && Name[i] <= '9')
        {
                num *= 10;
@@ -150,20 +176,103 @@ tVFS_Node *IPStack_FindDir(tVFS_Node *Node, char *Name)
 }
 
 static const char *casIOCtls_Root[] = { DRV_IOCTLNAMES, "add_interface", NULL };
+/**
+ * \brief Handles IOCtls for the IPStack root
+ */
+int IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data)
+{
+        int    tmp;
+       ENTER("pNode iID pData", Node, ID, Data);
+       
+       switch(ID)
+       {
+       // --- Standard IOCtls (0-3) ---
+       case DRV_IOCTL_TYPE:
+               LEAVE('i', DRV_TYPE_MISC);
+               return DRV_TYPE_MISC;
+       
+       case DRV_IOCTL_IDENT:
+               tmp = ModUtil_SetIdent(Data, "IPStack");
+               LEAVE('i', 1);
+               return 1;
+       
+       case DRV_IOCTL_VERSION:
+               LEAVE('x', VERSION);
+               return VERSION;
+       
+       case DRV_IOCTL_LOOKUP:
+               tmp = ModUtil_LookupString( (char**)casIOCtls_Root, (char*)Data );
+               LEAVE('i', tmp);
+               return tmp;
+               
+               /*
+                * add_interface
+                * - Adds a new IP interface and binds it to a device
+                */
+       case 4:
+               if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
+               if( !CheckString( Data ) )      LEAVE_RET('i', -1);
+               tmp = IPStack_AddInterface(Data);
+               LEAVE_RET('i', tmp);
+       }
+       LEAVE('i', 0);
+       return 0;
+}
+
+/**
+ * \brief Read from an interface's directory
+ */
+char *IPStack_Iface_ReadDir(tVFS_Node *Node, int Pos)
+{
+       tSocketFile     *file = gIP_FileTemplates;
+       while(Pos-- && file) {
+               Log("IPStack_Iface_ReadDir: %s", file->Name);
+               file = file->Next;
+       }
+       
+       if(!file)       return NULL;
+       
+       return strdup(file->Name);
+}
+
+/**
+ * \brief Gets a named node from an interface directory
+ */
+tVFS_Node *IPStack_Iface_FindDir(tVFS_Node *Node, char *Name)
+{
+       tSocketFile     *file = gIP_FileTemplates;
+       
+       // Get file definition
+       for(;file;file = file->Next)
+       {
+               if( strcmp(file->Name, Name) == 0 )     break;
+               Log("IPStack_Iface_FindDir: strcmp('%s', '%s')", file->Name, Name);
+       }
+       if(!file)       return NULL;
+       
+       // Pass the buck!
+       return file->Init(Node->ImplPtr);
+}
+
+/**
+ * \brief Names for interface IOCtl Calls
+ */
 static const char *casIOCtls_Iface[] = {
        DRV_IOCTLNAMES,
        "getset_type",
        "get_address", "set_address",
        "getset_subnet",
        "get_gateway", "set_gateway",
+       "ping",
        NULL
        };
 /**
- * \brief Handles IOCtls for the IPStack root/interfaces
+ * \brief Handles IOCtls for the IPStack interfaces
  */
-int IPStack_IOCtl(tVFS_Node *Node, int ID, void *Data)
+int IPStack_Iface_IOCtl(tVFS_Node *Node, int ID, void *Data)
 {
         int    tmp;
+       tInterface      *iface = (tInterface*)Node->ImplPtr;
        ENTER("pNode iID pData", Node, ID, Data);
        
        switch(ID)
@@ -174,8 +283,7 @@ int IPStack_IOCtl(tVFS_Node *Node, int ID, void *Data)
                return DRV_TYPE_MISC;
        
        case DRV_IOCTL_IDENT:
-               if( !CheckMem( Data, 4 ) )      LEAVE_RET('i', -1);
-               memcpy(Data, "IP\0\0", 4);
+               tmp = ModUtil_SetIdent(Data, STR(IDENT));
                LEAVE('i', 1);
                return 1;
        
@@ -184,202 +292,195 @@ int IPStack_IOCtl(tVFS_Node *Node, int ID, void *Data)
                return VERSION;
        
        case DRV_IOCTL_LOOKUP:
-               if( !CheckString( Data ) )      LEAVE_RET('i', -1);
-               LOG("Lookup '%s'", Data);
-               if( Node == &gIP_DriverInfo.RootNode )
-                       tmp = LookupString( (char**)casIOCtls_Root, (char*)Data );
-               else
-                       tmp = LookupString( (char**)casIOCtls_Iface, (char*)Data );
+               tmp = ModUtil_LookupString( (char**)casIOCtls_Iface, (char*)Data );
                LEAVE('i', tmp);
                return tmp;
-       }
        
-       if(Node == &gIP_DriverInfo.RootNode)
-       {
-               switch(ID)
+       /*
+        * getset_type
+        * - Get/Set the interface type
+        */
+       case 4:
+               // Set Type?
+               if( Data )
                {
-               /*
-                * add_interface
-                * - Adds a new IP interface and binds it to a device
-                */
+                       // Ok, it's set type
+                       if( Threads_GetUID() != 0 ) {
+                               LOG("Attempt by non-root to alter an interface (%i)", Threads_GetUID());
+                               LEAVE('i', -1);
+                               return -1;
+                       }
+                       if( !CheckMem( Data, sizeof(int) ) ) {
+                               LOG("Invalid pointer %p", Data);
+                               LEAVE('i', -1);
+                               return -1;
+                       }
+                       switch( *(int*)Data )
+                       {
+                       case 0: // Disable
+                               iface->Type = 0;
+                               memset(&iface->IP6, 0, sizeof(tIPv6));  // Clear address
+                               break;
+                       case 4: // IPv4
+                               iface->Type = 4;
+                               memset(&iface->IP4, 0, sizeof(tIPv4));
+                               break;
+                       case 6: // IPv6
+                               iface->Type = 6;
+                               memset(&iface->IP6, 0, sizeof(tIPv6));
+                               break;
+                       default:
+                               LEAVE('i', -1);
+                               return -1;
+                       }
+               }
+               LEAVE('i', iface->Type);
+               return iface->Type;
+       
+       /*
+        * get_address
+        * - Get the interface's address
+        */
+       case 5:
+               switch(iface->Type)
+               {
+               case 0: LEAVE_RET('i', 1);
                case 4:
-                       if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
-                       if( !CheckString( Data ) )      LEAVE_RET('i', -1);
-                       tmp = IPStack_AddInterface(Data);
-                       LEAVE_RET('i', tmp);
+                       if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
+                       memcpy( Data, &iface->IP4.Address, sizeof(tIPv4) );
+                       LEAVE_RET('i', 1);
+               case 6:
+                       if( !CheckMem( Data, sizeof(tIPv6) ) )  LEAVE_RET('i', -1);
+                       memcpy( Data, &iface->IP6.Address, sizeof(tIPv6) );
+                       LEAVE_RET('i', 1);
                }
-               LEAVE('i', 0);
-               return 0;       
-       }
-       else
-       {
-               tInterface      *iface = (tInterface*)Node->ImplPtr;
-               switch(ID)
+               LEAVE_RET('i', 0);
+       
+       /*
+        * set_address
+        * - Get the interface's address
+        */
+       case 6:
+               if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
+               switch(iface->Type)
                {
-               /*
-                * getset_type
-                * - Get/Set the interface type
-                */
+               case 0: LEAVE_RET('i', 1);
                case 4:
-                       // Set Type?
-                       if( Data )
+                       if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
+                       iface->Type = 0;        // One very hacky mutex/trash protector
+                       memcpy( &iface->IP4.Address, Data, sizeof(tIPv4) );
+                       iface->Type = 4;
+                       LEAVE_RET('i', 1);
+               case 6:
+                       if( !CheckMem( Data, sizeof(tIPv6) ) )  LEAVE_RET('i', -1);
+                       iface->Type = 0;
+                       memcpy( &iface->IP6.Address, Data, sizeof(tIPv6) );
+                       iface->Type = 6;
+                       LEAVE_RET('i', 1);
+               }
+               LEAVE_RET('i', 0);
+       
+       /*
+        * getset_subnet
+        * - Get/Set the bits in the address subnet
+        */
+       case 7:
+               // Get?
+               if( Data == NULL )
+               {
+                       switch( iface->Type )
                        {
-                               // Ok, it's set type
-                               if( Threads_GetUID() != 0 ) {
-                                       LOG("Attempt by non-root to alter an interface (%i)", Threads_GetUID());
-                                       LEAVE('i', -1);
-                                       return -1;
-                               }
-                               if( !CheckMem( Data, sizeof(int) ) ) {
-                                       LOG("Invalid pointer %p", Data);
-                                       LEAVE('i', -1);
-                                       return -1;
-                               }
-                               switch( *(int*)Data )
-                               {
-                               case 0: // Disable
-                                       iface->Type = 0;
-                                       memset(&iface->IP6, 0, sizeof(tIPv6));  // Clear address
-                                       break;
-                               case 4: // IPv4
-                                       iface->Type = 4;
-                                       memset(&iface->IP4, 0, sizeof(tIPv4));
-                                       break;
-                               case 6: // IPv6
-                                       iface->Type = 6;
-                                       memset(&iface->IP6, 0, sizeof(tIPv6));
-                                       break;
-                               default:
-                                       LEAVE('i', -1);
-                                       return -1;
-                               }
+                       case 4:         LEAVE_RET('i', iface->IP4.SubnetBits);
+                       case 6:         LEAVE_RET('i', iface->IP6.SubnetBits);
+                       default:        LEAVE_RET('i', 0);
                        }
-                       LEAVE('i', iface->Type);
-                       return iface->Type;
+               }
                
-               /*
-                * get_address
-                * - Get the interface's address
-                */
-               case 5:
-                       switch(iface->Type)
-                       {
-                       case 0: LEAVE_RET('i', 1);
-                       case 4:
-                               if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
-                               memcpy( Data, &iface->IP4.Address, sizeof(tIPv4) );
-                               LEAVE_RET('i', 1);
-                       case 6:
-                               if( !CheckMem( Data, sizeof(tIPv6) ) )  LEAVE_RET('i', -1);
-                               memcpy( Data, &iface->IP6.Address, sizeof(tIPv6) );
-                               LEAVE_RET('i', 1);
-                       }
-                       LEAVE_RET('i', 0);
+               // Ok, set.
+               if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
+               if( !CheckMem(Data, sizeof(int)) )      LEAVE_RET('i', -1);
                
-               /*
-                * set_address
-                * - Get the interface's address
-                */
+               // Check and set the subnet bits
+               switch( iface->Type )
+               {
+               case 4:
+                       if( *(int*)Data < 0 || *(int*)Data > 31 )       LEAVE_RET('i', -1);
+                       iface->IP4.SubnetBits = *(int*)Data;
+                       LEAVE_RET('i', iface->IP4.SubnetBits);
                case 6:
-                       if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
-                       switch(iface->Type)
-                       {
-                       case 0: LEAVE_RET('i', 1);
-                       case 4:
-                               if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
-                               iface->Type = 0;        // One very hacky mutex/trash protector
-                               memcpy( &iface->IP4.Address, Data, sizeof(tIPv4) );
-                               iface->Type = 4;
-                               LEAVE_RET('i', 1);
-                       case 6:
-                               if( !CheckMem( Data, sizeof(tIPv6) ) )  LEAVE_RET('i', -1);
-                               iface->Type = 0;
-                               memcpy( &iface->IP6.Address, Data, sizeof(tIPv6) );
-                               iface->Type = 6;
-                               LEAVE_RET('i', 1);
-                       }
-                       LEAVE_RET('i', 0);
+                       if( *(int*)Data < 0 || *(int*)Data > 127 )      LEAVE_RET('i', -1);
+                       iface->IP6.SubnetBits = *(int*)Data;
+                       LEAVE_RET('i', iface->IP6.SubnetBits);
+               default:
+                       break;
+               }
                
-               /*
-                * getset_subnet
-                * - Get/Set the bits in the address subnet
-                */
-               case 7:
-                       // Get?
-                       if( Data == NULL )
-                       {
-                               switch( iface->Type )
-                               {
-                               case 4:         LEAVE_RET('i', iface->IP4.SubnetBits);
-                               case 6:         LEAVE_RET('i', iface->IP6.SubnetBits);
-                               default:        LEAVE_RET('i', 0);
-                               }
-                       }
-                       
-                       // Ok, set.
-                       if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
-                       if( !CheckMem(Data, sizeof(int)) )      LEAVE_RET('i', -1);
-                       
-                       // Check and set the subnet bits
-                       switch( iface->Type )
-                       {
-                       case 4:
-                               if( *(int*)Data < 0 || *(int*)Data > 31 )       LEAVE_RET('i', -1);
-                               iface->IP4.SubnetBits = *(int*)Data;
-                               LEAVE_RET('i', iface->IP4.SubnetBits);
-                       case 6:
-                               if( *(int*)Data < 0 || *(int*)Data > 127 )      LEAVE_RET('i', -1);
-                               iface->IP6.SubnetBits = *(int*)Data;
-                               LEAVE_RET('i', iface->IP6.SubnetBits);
-                       default:
-                               break;
-                       }
-                       
-                       LEAVE('i', 0);
-                       return 0;
+               LEAVE('i', 0);
+               return 0;
+               
+       /*
+        * get_gateway
+        * - Get the interface's IPv4 gateway
+        */
+       case 8:
+               switch(iface->Type)
+               {
+               case 0:
+                       LEAVE_RET('i', 1);
+               case 4:
+                       if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
+                       memcpy( Data, &iface->IP4.Gateway, sizeof(tIPv4) );
+                       LEAVE_RET('i', 1);
+               case 6:
+                       LEAVE_RET('i', 1);
+               }
+               LEAVE('i', 0);
+               return 0;
+       
+       /*
+        * set_gateway
+        * - Get/Set the interface's IPv4 gateway
+        */
+       case 9:
+               if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
+               switch(iface->Type)
+               {
+               case 0:
+                       LEAVE_RET('i', 1);
+               
+               case 4:
+                       if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
+                       iface->Type = 0;        // One very hacky mutex/trash protector
+                       memcpy( &iface->IP4.Gateway, Data, sizeof(tIPv4) );
+                       iface->Type = 4;
+                       LEAVE_RET('i', 1);
                        
-               /*
-                * get_gateway
-                * - Get the interface's IPv4 gateway
-                */
-               case 8:
-                       switch(iface->Type)
-                       {
-                       case 0:
-                               LEAVE_RET('i', 1);
-                       case 4:
-                               if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
-                               memcpy( Data, &iface->IP4.Gateway, sizeof(tIPv4) );
-                               LEAVE_RET('i', 1);
-                       case 6:
-                               LEAVE_RET('i', 1);
-                       }
-                       LEAVE('i', 0);
-                       return 0;
+               case 6:
+                       LEAVE_RET('i', 1);
+               }
+               break;
+       
+       /*
+        * ping
+        * - Send an ICMP Echo
+        */
+       case 10:
+               switch(iface->Type)
+               {
+               case 0:
+                       LEAVE_RET('i', 1);
                
-               /*
-                * set_gateway
-                * - Get the interface's IPv4 gateway
-                */
-               case 9:
-                       if( Threads_GetUID() != 0 )     LEAVE_RET('i', -1);
-                       switch(iface->Type)
-                       {
-                       case 0:
-                               LEAVE_RET('i', 1);
+               case 4:
+                       if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
+                       tmp = IPv4_Ping(iface, *(tIPv4*)Data);
+                       LEAVE('i', tmp);
+                       return tmp;
                        
-                       case 4:
-                               if( !CheckMem( Data, sizeof(tIPv4) ) )  LEAVE_RET('i', -1);
-                               iface->Type = 0;        // One very hacky mutex/trash protector
-                               memcpy( &iface->IP4.Gateway, Data, sizeof(tIPv4) );
-                               iface->Type = 4;
-                               LEAVE_RET('i', 1);
-                               
-                       case 6:
-                               LEAVE_RET('i', 1);
-                       }
-                       LEAVE_RET('i', 0);
+               case 6:
+                       LEAVE_RET('i', 1);
                }
+               break;
+       
        }
        
        LEAVE('i', 0);
@@ -410,12 +511,15 @@ int IPStack_AddInterface(char *Device)
        iface->Node.ImplPtr = iface;
        iface->Node.ImplInt = giIP_NextIfaceId++;
        iface->Node.Flags = VFS_FFLAG_DIRECTORY;
-       iface->Node.Size = 0;
+       iface->Node.Size = -1;
        iface->Node.NumACLs = 1;
        iface->Node.ACLs = &gVFS_ACL_EveryoneRX;
-       iface->Node.ReadDir = NULL;
-       iface->Node.FindDir = NULL;
-       iface->Node.IOCtl = IPStack_IOCtl;
+       iface->Node.ReadDir = IPStack_Iface_ReadDir;
+       iface->Node.FindDir = IPStack_Iface_FindDir;
+       iface->Node.IOCtl = IPStack_Iface_IOCtl;
+       
+       // Set Defaults
+       iface->TimeoutDelay = DEFAULT_TIMEOUT;
        
        // Get adapter handle
        iface->Adapter = IPStack_GetAdapter(Device);

UCC git Repository :: git.ucc.asn.au