X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Modules%2FIPStack%2Fmain.c;h=5f650b3321464162115034886620773c6125c96c;hb=7c5a60c8be053f65ecceb47be773ea21a050b186;hp=aaeeb9af5e3c6afebf64fa329d52e229dabe4f7c;hpb=6c5a509b5e14e097ca537c539bc9babe3b8f0c4c;p=tpg%2Facess2.git diff --git a/Modules/IPStack/main.c b/Modules/IPStack/main.c index aaeeb9af..5f650b33 100644 --- a/Modules/IPStack/main.c +++ b/Modules/IPStack/main.c @@ -1,33 +1,47 @@ /* * Acess2 IP Stack - * - Address Resolution Protocol + * - Stack Initialisation */ +#define DEBUG 0 +#define VERSION VER2(0,10) #include "ipstack.h" +#include "link.h" #include #include // === IMPORTS === - int ARP_Initialise(); +extern int ARP_Initialise(); +extern void UDP_Initialise(); +extern void TCP_Initialise(); +extern int IPv4_Initialise(); +extern int IPv6_Initialise(); + +extern tAdapter *IPStack_GetAdapter(const char *Path); +extern char *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos); +extern tVFS_Node *IPStack_Root_FindDir(tVFS_Node *Node, const char *Name); +extern int IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data); +extern tInterface gIP_LoopInterface; +extern tInterface *IPStack_AddInterface(const char *Device, const char *Name); +extern tRoute *IPStack_AddRoute(const char *Interface, void *Network, int SubnetBits, void *NextHop, int Metric); // === 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_CompareAddress(int AddressType, const void *Address1, const void *Address2, int CheckBits); // === GLOBALS === -MODULE_DEFINE(0, 0x000A, IPStack, IPStack_Install, NULL); +MODULE_DEFINE(0, VERSION, IPStack, IPStack_Install, NULL, NULL); tDevFS_Driver gIP_DriverInfo = { NULL, "ip", { + .Size = -1, // Number of interfaces .NumACLs = 1, .ACLs = &gVFS_ACL_EveryoneRX, .Flags = VFS_FFLAG_DIRECTORY, - .ReadDir = IPStack_ReadDir, - .FindDir = IPStack_FindDir + .ReadDir = IPStack_Root_ReadDir, + .FindDir = IPStack_Root_FindDir, + .IOCtl = IPStack_Root_IOCtl } }; - int giIP_NumInterfaces; -tInterface *gIP_Interfaces = NULL; // === CODE === /** @@ -38,50 +52,224 @@ int IPStack_Install(char **Arguments) { int i = 0; - // Install Handlers + // Layer 3 - Network Layer Protocols ARP_Initialise(); + IPv4_Initialise(); + IPv6_Initialise(); + // Layer 4 - Transport Layer Protocols + TCP_Initialise(); + UDP_Initialise(); - // Parse module arguments - for( i = 0; Arguments[i]; i++ ) + if(Arguments) { - //if(strcmp(Arguments[i], "Device") == '=') { - // - //} + // Parse module arguments + for( i = 0; Arguments[i]; i++ ) + { + // TODO: + // Define interfaces by ::: + // Where: + // - is the device path (E.g. /Devices/ne2k/0) + // - is a number (e.g. 4) or symbol (e.g. AF_INET4) + // - is a condensed hexadecimal stream (in big endian) + // (E.g. 0A000201 for 10.0.2.1 IPv4) + // - is the number of subnet bits (E.g. 24 for an IPv4 Class C) + // Example: /Devices/ne2k/0:4:0A00020A:24 + // would define an interface with the address 10.0.2.10/24 + if( Arguments[i][0] == '/' ) { + // Define Interface + char *dev, *type, *addr, *bits; + + // Read definition + dev = Arguments[i]; + type = strchr(dev, ':'); + if( !type ) { + Log_Warning("IPStack", ":::"); + continue; + } + *type = '\0'; type ++; + + addr = strchr(type, ':'); + if( !addr ) { + Log_Warning("IPStack", ":::"); + continue; + } + *addr = '\0'; addr ++; + + bits = strchr(addr, ':'); + if( !bits ) { + Log_Warning("IPStack", ":::"); + continue; + } + *bits = '\0'; bits ++; + + // Define interface + { + int iType = atoi(type); + int size = IPStack_GetAddressSize(iType); + Uint8 addrData[size]; + int iBits = atoi(bits); + + UnHex(addrData, size, addr); + + tInterface *iface = IPStack_AddInterface(dev, ""); + if( !iface ) { + Log_Warning("IPStack", "Unable to add interface on '%s'", dev); + continue ; + } + iface->Type = iType; + memcpy(iface->Address, addrData, size); + iface->SubnetBits = iBits; + + // Route for addrData/iBits, no next hop, default metric + IPStack_AddRoute(iface->Name, iface->Address, iBits, NULL, 0); + + Log_Notice("IPStack", "Boot interface %s/%i on %s", + IPStack_PrintAddress(iType, addrData), iBits, + dev); + } + + continue; + } + + // I could also define routes using ::[:] + // Example: 1:00000000:0:0A000201 + if( '0' <= Arguments[i][0] && Arguments[i][0] <= '9' ) + { + // Define Interface + char *ifaceName, *network, *bits, *gateway; + + // Read definition + ifaceName = Arguments[i]; + + network = strchr(ifaceName, ':'); + if( !network ) { + Log_Warning("IPStack", ":::"); + continue; + } + *network = '\0'; network ++; + + bits = strchr(network, ':'); + if( !bits ) { + Log_Warning("IPStack", ":::"); + continue; + } + *bits = '\0'; bits ++; + + gateway = strchr(bits, ':'); + if( gateway ) { + *gateway = '\0'; gateway ++; + } + + // Define route + { + tVFS_Node *node = IPStack_Root_FindDir(NULL, ifaceName); + if( !node ) { + Log_Warning("IPStack", "Unknown interface '%s' in arg %i", ifaceName, i); + continue ; + } + tInterface *iface = node->ImplPtr; + + int size = IPStack_GetAddressSize(iface->Type); + Uint8 netData[size]; + Uint8 gwData[size]; + int iBits = atoi(bits); + + UnHex(netData, size, network); + if( gateway ) + UnHex(gwData, size, gateway); + else + memset(gwData, 0, size); + + IPStack_AddRoute(ifaceName, netData, iBits, gwData, 30); + } + + continue; + } + } } - return 1; + // Initialise loopback interface + gIP_LoopInterface.Adapter = IPStack_GetAdapter("LOOPBACK"); + + DevFS_AddDevice( &gIP_DriverInfo ); + + return MODULE_ERR_OK; } /** - * \brief Read from the IP Stack's Device Directory + * \brief Gets the size (in bytes) of a specified form of address */ -char *IPStack_ReadDir(tVFS_Node *Node, int Pos) +int IPStack_GetAddressSize(int AddressType) { - tInterface *iface; - char name[5] = "ip0\0\0"; + switch(AddressType) + { + case -1: // -1 = maximum + return sizeof(tIPv6); - // Create the name - if(Pos < 10) - name[2] = '0' + Pos; - else { - name[2] = '0' + Pos/10; - name[3] = '0' + Pos%10; + case AF_NULL: + return 0; + + case AF_INET4: + return sizeof(tIPv4); + case AF_INET6: + return sizeof(tIPv6); + + default: + return 0; } +} + +/** + * \brief Compare two IP Addresses masked by CheckBits + */ +int IPStack_CompareAddress(int AddressType, const void *Address1, const void *Address2, int CheckBits) +{ + int size = IPStack_GetAddressSize(AddressType); + Uint8 mask; + const Uint8 *addr1 = Address1, *addr2 = Address2; + + // Sanity check size + if( CheckBits < 0 ) CheckBits = size*8; + if( CheckBits > size*8 ) CheckBits = size*8; + + if( CheckBits == 0 ) return 1; // /0 matches anything - // Traverse the list - for( iface = gIP_Interfaces; iface && Pos--; iface = iface->Next ) ; + // Check first bits/8 bytes + if( memcmp(Address1, Address2, CheckBits/8) != 0 ) return 0; - // Did we run off the end? - if(!iface) return NULL; + // Check if the mask is a multiple of 8 + if( CheckBits % 8 == 0 ) return 1; - // Return the pre-generated name - return strdup(name); + // Check last bits + mask = 0xFF << (8 - (CheckBits % 8)); + if( (addr1[CheckBits/8] & mask) == (addr2[CheckBits/8] & mask) ) + return 1; + + return 0; } -/** - * \brief Get the node of an interface - */ -tVFS_Node *IPStack_FindDir(tVFS_Node *Node, char *Name) +const char *IPStack_PrintAddress(int AddressType, const void *Address) { - return NULL; + switch( AddressType ) + { + case 4: { + static char ret[4*3+3+1]; + const Uint8 *addr = Address; + sprintf(ret, "%i.%i.%i.%i", addr[0], addr[1], addr[2], addr[3]); + return ret; + } + + case 6: { // TODO: address compression + static char ret[8*4+7+1]; + const Uint16 *addr = Address; + sprintf(ret, "%x:%x:%x:%x:%x:%x:%x:%x", + addr[0], addr[1], addr[2], addr[3], + addr[4], addr[5], addr[6], addr[7] + ); + return ret; + } + + default: + return ""; + } }