Compile testing, what's that?
[tpg/acess2.git] / 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 }

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