3 * - Address Resolution Protocol
6 #define VERSION ((0<<8)|10)
11 #include <tpl_drv_common.h>
12 #include <tpl_drv_network.h>
15 extern int ARP_Initialise();
16 extern int IPv4_Initialise();
19 int IPStack_Install(char **Arguments);
20 char *IPStack_ReadDir(tVFS_Node *Node, int Pos);
21 tVFS_Node *IPStack_FindDir(tVFS_Node *Node, char *Name);
22 int IPStack_IOCtl(tVFS_Node *Node, int ID, void *Data);
23 int IPStack_AddInterface(char *Device);
24 tAdapter *IPStack_GetAdapter(char *Path);
27 MODULE_DEFINE(0, VERSION, IPStack, IPStack_Install, NULL, NULL);
28 tDevFS_Driver gIP_DriverInfo = {
31 .Size = 0, // Number of interfaces
33 .ACLs = &gVFS_ACL_EveryoneRX,
34 .Flags = VFS_FFLAG_DIRECTORY,
35 .ReadDir = IPStack_ReadDir,
36 .FindDir = IPStack_FindDir,
37 .IOCtl = IPStack_IOCtl
40 tSpinlock glIP_Interfaces = 0;
41 tInterface *gIP_Interfaces = NULL;
42 tInterface *gIP_Interfaces_Last = NULL;
43 int giIP_NextIfaceId = 1;
44 tSpinlock glIP_Adapters = 0;
45 tAdapter *gIP_Adapters = NULL;
49 * \fn int IPStack_Install(char **Arguments)
50 * \brief Intialise the relevant parts of the stack and register with DevFS
52 int IPStack_Install(char **Arguments)
62 // Parse module arguments
63 for( i = 0; Arguments[i]; i++ )
65 //if(strcmp(Arguments[i], "Device") == '=') {
71 DevFS_AddDevice( &gIP_DriverInfo );
77 * \brief Read from the IP Stack's Device Directory
79 char *IPStack_ReadDir(tVFS_Node *Node, int Pos)
82 char name[] = "ip0\0\0\0";
83 ENTER("pNode iPos", Node, Pos);
86 for( iface = gIP_Interfaces; iface && Pos--; iface = iface->Next ) ;
88 // Did we run off the end?
95 Pos = iface->Node.ImplInt;
99 name[2] = '0' + Pos/10;
100 name[3] = '0' + Pos%10;
103 name[2] = '0' + Pos/100;
104 name[3] = '0' + (Pos/10)%10;
105 name[4] = '0' + Pos%10;
109 // Return the pre-generated name
114 * \brief Get the node of an interface
116 tVFS_Node *IPStack_FindDir(tVFS_Node *Node, char *Name)
121 ENTER("pNode sName", Node, Name);
123 if(Name[0] != 'i' || Name[1] != 'p') {
129 while('0' <= Name[i] && Name[i] <= '9')
132 num += Name[i] - '0';
135 if(Name[i] != '\0') {
140 for( iface = gIP_Interfaces; iface; iface = iface->Next )
142 if( iface->Node.ImplInt == num )
144 LEAVE('p', &iface->Node);
152 static const char *casIOCtls_Root[] = { DRV_IOCTLNAMES, "add_interface", NULL };
153 static const char *casIOCtls_Iface[] = {
156 "get_address", "set_address",
158 "get_gateway", "set_gateway",
162 * \brief Handles IOCtls for the IPStack root/interfaces
164 int IPStack_IOCtl(tVFS_Node *Node, int ID, void *Data)
167 ENTER("pNode iID pData", Node, ID, Data);
171 // --- Standard IOCtls (0-3) ---
173 LEAVE('i', DRV_TYPE_MISC);
174 return DRV_TYPE_MISC;
176 case DRV_IOCTL_IDENT:
177 if( !CheckMem( Data, 4 ) ) LEAVE_RET('i', -1);
178 memcpy(Data, "IP\0\0", 4);
182 case DRV_IOCTL_VERSION:
186 case DRV_IOCTL_LOOKUP:
187 if( !CheckString( Data ) ) LEAVE_RET('i', -1);
188 LOG("Lookup '%s'", Data);
189 if( Node == &gIP_DriverInfo.RootNode )
190 tmp = LookupString( (char**)casIOCtls_Root, (char*)Data );
192 tmp = LookupString( (char**)casIOCtls_Iface, (char*)Data );
197 if(Node == &gIP_DriverInfo.RootNode)
203 * - Adds a new IP interface and binds it to a device
206 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
207 if( !CheckString( Data ) ) LEAVE_RET('i', -1);
208 tmp = IPStack_AddInterface(Data);
216 tInterface *iface = (tInterface*)Node->ImplPtr;
221 * - Get/Set the interface type
228 if( Threads_GetUID() != 0 ) {
229 LOG("Attempt by non-root to alter an interface (%i)", Threads_GetUID());
233 if( !CheckMem( Data, sizeof(int) ) ) {
234 LOG("Invalid pointer %p", Data);
238 switch( *(int*)Data )
242 memset(&iface->IP6, 0, sizeof(tIPv6)); // Clear address
246 memset(&iface->IP4, 0, sizeof(tIPv4));
250 memset(&iface->IP6, 0, sizeof(tIPv6));
257 LEAVE('i', iface->Type);
262 * - Get the interface's address
267 case 0: LEAVE_RET('i', 1);
269 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
270 memcpy( Data, &iface->IP4.Address, sizeof(tIPv4) );
273 if( !CheckMem( Data, sizeof(tIPv6) ) ) LEAVE_RET('i', -1);
274 memcpy( Data, &iface->IP6.Address, sizeof(tIPv6) );
281 * - Get the interface's address
284 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
287 case 0: LEAVE_RET('i', 1);
289 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
290 iface->Type = 0; // One very hacky mutex/trash protector
291 memcpy( &iface->IP4.Address, Data, sizeof(tIPv4) );
295 if( !CheckMem( Data, sizeof(tIPv6) ) ) LEAVE_RET('i', -1);
297 memcpy( &iface->IP6.Address, Data, sizeof(tIPv6) );
305 * - Get/Set the bits in the address subnet
311 switch( iface->Type )
313 case 4: LEAVE_RET('i', iface->IP4.SubnetBits);
314 case 6: LEAVE_RET('i', iface->IP6.SubnetBits);
315 default: LEAVE_RET('i', 0);
320 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
321 if( !CheckMem(Data, sizeof(int)) ) LEAVE_RET('i', -1);
323 // Check and set the subnet bits
324 switch( iface->Type )
327 if( *(int*)Data < 0 || *(int*)Data > 31 ) LEAVE_RET('i', -1);
328 iface->IP4.SubnetBits = *(int*)Data;
329 LEAVE_RET('i', iface->IP4.SubnetBits);
331 if( *(int*)Data < 0 || *(int*)Data > 127 ) LEAVE_RET('i', -1);
332 iface->IP6.SubnetBits = *(int*)Data;
333 LEAVE_RET('i', iface->IP6.SubnetBits);
343 * - Get the interface's IPv4 gateway
351 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
352 memcpy( Data, &iface->IP4.Gateway, sizeof(tIPv4) );
362 * - Get the interface's IPv4 gateway
365 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
372 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
373 iface->Type = 0; // One very hacky mutex/trash protector
374 memcpy( &iface->IP4.Gateway, Data, sizeof(tIPv4) );
391 * \fn int IPStack_AddInterface(char *Device)
392 * \brief Adds an interface to the list
394 int IPStack_AddInterface(char *Device)
398 ENTER("sDevice", Device);
400 iface = malloc(sizeof(tInterface));
403 return -2; // Return ERR_MYBAD
407 iface->Type = 0; // Unset type
410 iface->Node.ImplPtr = iface;
411 iface->Node.ImplInt = giIP_NextIfaceId++;
412 iface->Node.Flags = VFS_FFLAG_DIRECTORY;
413 iface->Node.Size = 0;
414 iface->Node.NumACLs = 1;
415 iface->Node.ACLs = &gVFS_ACL_EveryoneRX;
416 iface->Node.ReadDir = NULL;
417 iface->Node.FindDir = NULL;
418 iface->Node.IOCtl = IPStack_IOCtl;
420 // Get adapter handle
421 iface->Adapter = IPStack_GetAdapter(Device);
422 if( !iface->Adapter ) {
425 return -1; // Return ERR_YOUFAIL
429 LOCK( &glIP_Interfaces );
430 if( gIP_Interfaces ) {
431 gIP_Interfaces_Last->Next = iface;
432 gIP_Interfaces_Last = iface;
435 gIP_Interfaces = iface;
436 gIP_Interfaces_Last = iface;
438 RELEASE( &glIP_Interfaces );
440 gIP_DriverInfo.RootNode.Size ++;
443 LEAVE('i', iface->Node.ImplInt);
444 return iface->Node.ImplInt;
448 * \fn tAdapter *IPStack_GetAdapter(char *Path)
449 * \brief Gets/opens an adapter given the path
451 tAdapter *IPStack_GetAdapter(char *Path)
456 ENTER("sPath", Path);
458 LOCK( &glIP_Adapters );
460 // Check if this adapter is already open
461 for( dev = gIP_Adapters; dev; dev = dev->Next )
463 if( strcmp(dev->Device, Path) == 0 ) {
465 RELEASE( &glIP_Adapters );
471 // Ok, so let's open it
472 dev = malloc( sizeof(tAdapter) + strlen(Path) + 1 );
474 RELEASE( &glIP_Adapters );
480 strcpy( dev->Device, Path );
484 dev->DeviceFD = VFS_Open( dev->Device, VFS_OPENFLAG_READ|VFS_OPENFLAG_WRITE );
485 if( dev->DeviceFD == -1 ) {
487 RELEASE( &glIP_Adapters );
492 // Check that it is a network interface
493 tmp = VFS_IOCtl(dev->DeviceFD, 0, NULL);
494 LOG("Device type = %i", tmp);
495 if( tmp != DRV_TYPE_NETWORK ) {
496 Warning("IPStack_GetAdapter: '%s' is not a network interface", dev->Device);
497 VFS_Close( dev->DeviceFD );
499 RELEASE( &glIP_Adapters );
505 VFS_IOCtl(dev->DeviceFD, NET_IOCTL_GETMAC, &dev->MacAddr);
508 dev->Next = gIP_Adapters;
511 RELEASE( &glIP_Adapters );
514 Link_WatchDevice( dev );