KernelLand/Modules/IPStack/main.c

   1 /*
   2  * Acess2 IP Stack
   3  * - Stack Initialisation
   4  */
   5 #define DEBUG   0
   6 #define VERSION VER2(0,10)
   7 #include "ipstack.h"
   8 #include "link.h"
   9 #include <modules.h>
  10 #include <fs_devfs.h>
  11 #include "include/adapters.h"
  12
  13 // === IMPORTS ===
  14 extern int      ARP_Initialise();
  15 extern void     UDP_Initialise();
  16 extern void     TCP_Initialise();
  17 extern int      IPv4_Initialise();
  18 extern int      IPv6_Initialise();
  19
  20 extern char     *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos);
  21 extern tVFS_Node        *IPStack_Root_FindDir(tVFS_Node *Node, const char *Name);
  22 extern int      IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data);
  23 extern tInterface       gIP_LoopInterface;
  24 extern tInterface       *IPStack_AddInterface(const char *Device, const char *Name);
  25 extern tRoute   *IPStack_AddRoute(const char *Interface, void *Network, int SubnetBits, void *NextHop, int Metric);
  26
  27 // === PROTOTYPES ===
  28  int    IPStack_Install(char **Arguments);
  29  int    IPStack_CompareAddress(int AddressType, const void *Address1, const void *Address2, int CheckBits);
  30
  31 // === GLOBALS ===
  32 MODULE_DEFINE(0, VERSION, IPStack, IPStack_Install, NULL, NULL);
  33 tVFS_NodeType   gIP_RootNodeType = {
  34         .ReadDir = IPStack_Root_ReadDir,
  35         .FindDir = IPStack_Root_FindDir,
  36         .IOCtl = IPStack_Root_IOCtl
  37 };
  38 tDevFS_Driver   gIP_DriverInfo = {
  39         NULL, "ip",
  40         {
  41         .Size = -1,     // Number of interfaces
  42         .NumACLs = 1,
  43         .ACLs = &gVFS_ACL_EveryoneRX,
  44         .Flags = VFS_FFLAG_DIRECTORY,
  45         .Type = &gIP_RootNodeType
  46         }
  47 };
  48
  49 // === CODE ===
  50 /**
  51  * \fn int IPStack_Install(char **Arguments)
  52  * \brief Intialise the relevant parts of the stack and register with DevFS
  53  */
  54 int IPStack_Install(char **Arguments)
  55 {
  56         // TODO: different Layer 2 protocols
  57         // Layer 3 - Network Layer Protocols
  58         ARP_Initialise();
  59         IPv4_Initialise();
  60         IPv6_Initialise();
  61         // Layer 4 - Transport Layer Protocols
  62         TCP_Initialise();
  63         UDP_Initialise();
  64
  65         // Initialise loopback interface
  66         gIP_LoopInterface.Adapter = Adapter_GetByName("lo");
  67
  68         DevFS_AddDevice( &gIP_DriverInfo );
  69
  70         return MODULE_ERR_OK;
  71 }
  72
  73 /**
  74  * \brief Gets the size (in bytes) of a specified form of address
  75  */
  76 int IPStack_GetAddressSize(int AddressType)
  77 {
  78         switch(AddressType)
  79         {
  80         case -1:        // -1 = maximum
  81                 return sizeof(tIPv6);
  82
  83         case AF_NULL:
  84                 return 0;
  85
  86         case AF_INET4:
  87                 return sizeof(tIPv4);
  88         case AF_INET6:
  89                 return sizeof(tIPv6);
  90
  91         default:
  92                 return 0;
  93         }
  94 }
  95
  96 /**
  97  * \brief Compare two IP Addresses masked by CheckBits
  98  */
  99 int IPStack_CompareAddress(int AddressType, const void *Address1, const void *Address2, int CheckBits)
 100 {
 101          int    size = IPStack_GetAddressSize(AddressType);
 102         Uint8   mask;
 103         const Uint8     *addr1 = Address1, *addr2 = Address2;
 104
 105         // Sanity check size
 106         if( CheckBits < 0 )     CheckBits = size*8;
 107         if( CheckBits > size*8 )        CheckBits = size*8;
 108
 109         if( CheckBits == 0 )    return 1;       // /0 matches anything
 110
 111         // Check first bits/8 bytes
 112         if( memcmp(Address1, Address2, CheckBits/8) != 0 )      return 0;
 113
 114         // Check if the mask is a multiple of 8
 115         if( CheckBits % 8 == 0 )        return 1;
 116
 117         // Check last bits
 118         mask = 0xFF << (8 - (CheckBits % 8));
 119         if( (addr1[CheckBits/8] & mask) == (addr2[CheckBits/8] & mask) )
 120                 return 1;
 121
 122         return 0;
 123 }
 124
 125 const char *IPStack_PrintAddress(int AddressType, const void *Address)
 126 {
 127         switch( AddressType )
 128         {
 129         case 4: {
 130                 static char     ret[4*3+3+1];
 131                 const Uint8     *addr = Address;
 132                 sprintf(ret, "%i.%i.%i.%i", addr[0], addr[1], addr[2], addr[3]);
 133                 return ret;
 134                 }
 135
 136         case 6: {       // TODO: address compression
 137                 static char     ret[8*4+7+1];
 138                 const Uint16    *addr = Address;
 139                 sprintf(ret, "%x:%x:%x:%x:%x:%x:%x:%x",
 140                         ntohs(addr[0]), ntohs(addr[1]), ntohs(addr[2]), ntohs(addr[3]),
 141                         ntohs(addr[4]), ntohs(addr[5]), ntohs(addr[6]), ntohs(addr[7])
 142                         );
 143                 return ret;
 144                 }
 145
 146         default:
 147                 return "";
 148         }
 149 }