3 * - Stack Initialisation
6 #define VERSION VER2(0,10)
11 #include <tpl_drv_common.h>
12 #include <tpl_drv_network.h>
15 //! Default timeout value, 30 seconds
16 #define DEFAULT_TIMEOUT (30*1000)
19 extern int ARP_Initialise();
20 extern void UDP_Initialise();
21 extern void TCP_Initialise();
22 extern int IPv4_Initialise();
23 extern int IPv4_Ping(tInterface *Iface, tIPv4 Addr);
26 int IPStack_Install(char **Arguments);
27 int IPStack_IOCtlRoot(tVFS_Node *Node, int ID, void *Data);
28 char *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos);
29 tVFS_Node *IPStack_Root_FindDir(tVFS_Node *Node, char *Name);
30 int IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data);
31 int IPStack_AddInterface(char *Device);
32 tAdapter *IPStack_GetAdapter(char *Path);
33 char *IPStack_Iface_ReadDir(tVFS_Node *Node, int Pos);
34 tVFS_Node *IPStack_Iface_FindDir(tVFS_Node *Node, char *Name);
35 int IPStack_Iface_IOCtl(tVFS_Node *Node, int ID, void *Data);
38 MODULE_DEFINE(0, VERSION, IPStack, IPStack_Install, NULL, NULL);
39 tDevFS_Driver gIP_DriverInfo = {
42 .Size = 0, // Number of interfaces
44 .ACLs = &gVFS_ACL_EveryoneRX,
45 .Flags = VFS_FFLAG_DIRECTORY,
46 .ReadDir = IPStack_Root_ReadDir,
47 .FindDir = IPStack_Root_FindDir,
48 .IOCtl = IPStack_Root_IOCtl
51 tSpinlock glIP_Interfaces = 0;
52 tInterface *gIP_Interfaces = NULL;
53 tInterface *gIP_Interfaces_Last = NULL;
54 int giIP_NextIfaceId = 1;
55 tSpinlock glIP_Adapters = 0;
56 tAdapter *gIP_Adapters = NULL;
57 tSocketFile *gIP_FileTemplates;
61 * \fn int IPStack_Install(char **Arguments)
62 * \brief Intialise the relevant parts of the stack and register with DevFS
64 int IPStack_Install(char **Arguments)
76 // Parse module arguments
77 for( i = 0; Arguments[i]; i++ )
79 //if(strcmp(Arguments[i], "Device") == '=') {
85 DevFS_AddDevice( &gIP_DriverInfo );
91 * \brief Adds a file to the socket list
93 int IPStack_AddFile(tSocketFile *File)
95 File->Next = gIP_FileTemplates;
96 gIP_FileTemplates = File;
101 * \brief Read from the IP Stack's Device Directory
103 char *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos)
107 ENTER("pNode iPos", Node, Pos);
110 for( iface = gIP_Interfaces; iface && Pos--; iface = iface->Next ) ;
112 // Did we run off the end?
121 Pos = iface->Node.ImplInt;
127 name[0] = '0' + Pos/10;
128 name[1] = '0' + Pos%10;
132 name[0] = '0' + Pos/100;
133 name[1] = '0' + (Pos/10)%10;
134 name[2] = '0' + Pos%10;
139 // Return the pre-generated name
144 * \brief Get the node of an interface
146 tVFS_Node *IPStack_Root_FindDir(tVFS_Node *Node, char *Name)
151 ENTER("pNode sName", Node, Name);
154 while('0' <= Name[i] && Name[i] <= '9')
157 num += Name[i] - '0';
160 if(Name[i] != '\0') {
165 for( iface = gIP_Interfaces; iface; iface = iface->Next )
167 if( iface->Node.ImplInt == num )
169 LEAVE('p', &iface->Node);
177 static const char *casIOCtls_Root[] = { DRV_IOCTLNAMES, "add_interface", NULL };
179 * \brief Handles IOCtls for the IPStack root
181 int IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data)
184 ENTER("pNode iID pData", Node, ID, Data);
188 // --- Standard IOCtls (0-3) ---
190 LEAVE('i', DRV_TYPE_MISC);
191 return DRV_TYPE_MISC;
193 case DRV_IOCTL_IDENT:
194 tmp = ModUtil_SetIdent(Data, "IPStack");
198 case DRV_IOCTL_VERSION:
202 case DRV_IOCTL_LOOKUP:
203 tmp = ModUtil_LookupString( (char**)casIOCtls_Root, (char*)Data );
209 * - Adds a new IP interface and binds it to a device
212 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
213 if( !CheckString( Data ) ) LEAVE_RET('i', -1);
214 tmp = IPStack_AddInterface(Data);
222 * \brief Read from an interface's directory
224 char *IPStack_Iface_ReadDir(tVFS_Node *Node, int Pos)
226 tSocketFile *file = gIP_FileTemplates;
227 while(Pos-- && file) file = file->Next;
229 if(!file) return NULL;
231 return strdup(file->Name);
235 * \brief Gets a named node from an interface directory
237 tVFS_Node *IPStack_Iface_FindDir(tVFS_Node *Node, char *Name)
239 tSocketFile *file = gIP_FileTemplates;
241 // Get file definition
242 for(;file;file = file->Next)
244 if( strcmp(file->Name, Name) == 0 ) break;
246 if(!file) return NULL;
249 return file->Init(Node->ImplPtr);
253 * \brief Names for interface IOCtl Calls
255 static const char *casIOCtls_Iface[] = {
258 "get_address", "set_address",
260 "get_gateway", "set_gateway",
265 * \brief Handles IOCtls for the IPStack interfaces
267 int IPStack_Iface_IOCtl(tVFS_Node *Node, int ID, void *Data)
270 tInterface *iface = (tInterface*)Node->ImplPtr;
271 ENTER("pNode iID pData", Node, ID, Data);
275 // --- Standard IOCtls (0-3) ---
277 LEAVE('i', DRV_TYPE_MISC);
278 return DRV_TYPE_MISC;
280 case DRV_IOCTL_IDENT:
281 tmp = ModUtil_SetIdent(Data, STR(IDENT));
285 case DRV_IOCTL_VERSION:
289 case DRV_IOCTL_LOOKUP:
290 tmp = ModUtil_LookupString( (char**)casIOCtls_Iface, (char*)Data );
296 * - Get/Set the interface type
303 if( Threads_GetUID() != 0 ) {
304 LOG("Attempt by non-root to alter an interface (%i)", Threads_GetUID());
308 if( !CheckMem( Data, sizeof(int) ) ) {
309 LOG("Invalid pointer %p", Data);
313 switch( *(int*)Data )
317 memset(&iface->IP6, 0, sizeof(tIPv6)); // Clear address
321 memset(&iface->IP4, 0, sizeof(tIPv4));
325 memset(&iface->IP6, 0, sizeof(tIPv6));
332 LEAVE('i', iface->Type);
337 * - Get the interface's address
342 case 0: LEAVE_RET('i', 1);
344 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
345 memcpy( Data, &iface->IP4.Address, sizeof(tIPv4) );
348 if( !CheckMem( Data, sizeof(tIPv6) ) ) LEAVE_RET('i', -1);
349 memcpy( Data, &iface->IP6.Address, sizeof(tIPv6) );
356 * - Get the interface's address
359 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
362 case 0: LEAVE_RET('i', 1);
364 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
365 iface->Type = 0; // One very hacky mutex/trash protector
366 memcpy( &iface->IP4.Address, Data, sizeof(tIPv4) );
370 if( !CheckMem( Data, sizeof(tIPv6) ) ) LEAVE_RET('i', -1);
372 memcpy( &iface->IP6.Address, Data, sizeof(tIPv6) );
380 * - Get/Set the bits in the address subnet
386 switch( iface->Type )
388 case 4: LEAVE_RET('i', iface->IP4.SubnetBits);
389 case 6: LEAVE_RET('i', iface->IP6.SubnetBits);
390 default: LEAVE_RET('i', 0);
395 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
396 if( !CheckMem(Data, sizeof(int)) ) LEAVE_RET('i', -1);
398 // Check and set the subnet bits
399 switch( iface->Type )
402 if( *(int*)Data < 0 || *(int*)Data > 31 ) LEAVE_RET('i', -1);
403 iface->IP4.SubnetBits = *(int*)Data;
404 LEAVE_RET('i', iface->IP4.SubnetBits);
406 if( *(int*)Data < 0 || *(int*)Data > 127 ) LEAVE_RET('i', -1);
407 iface->IP6.SubnetBits = *(int*)Data;
408 LEAVE_RET('i', iface->IP6.SubnetBits);
418 * - Get the interface's IPv4 gateway
426 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
427 memcpy( Data, &iface->IP4.Gateway, sizeof(tIPv4) );
437 * - Get/Set the interface's IPv4 gateway
440 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
447 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
448 iface->Type = 0; // One very hacky mutex/trash protector
449 memcpy( &iface->IP4.Gateway, Data, sizeof(tIPv4) );
460 * - Send an ICMP Echo
469 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
470 tmp = IPv4_Ping(iface, *(tIPv4*)Data);
487 * \fn int IPStack_AddInterface(char *Device)
488 * \brief Adds an interface to the list
490 int IPStack_AddInterface(char *Device)
494 ENTER("sDevice", Device);
496 iface = malloc(sizeof(tInterface));
499 return -2; // Return ERR_MYBAD
503 iface->Type = 0; // Unset type
506 iface->Node.ImplPtr = iface;
507 iface->Node.ImplInt = giIP_NextIfaceId++;
508 iface->Node.Flags = VFS_FFLAG_DIRECTORY;
509 iface->Node.Size = 0;
510 iface->Node.NumACLs = 1;
511 iface->Node.ACLs = &gVFS_ACL_EveryoneRX;
512 iface->Node.ReadDir = IPStack_Iface_ReadDir;
513 iface->Node.FindDir = IPStack_Iface_FindDir;
514 iface->Node.IOCtl = IPStack_Iface_IOCtl;
517 iface->TimeoutDelay = DEFAULT_TIMEOUT;
519 // Get adapter handle
520 iface->Adapter = IPStack_GetAdapter(Device);
521 if( !iface->Adapter ) {
524 return -1; // Return ERR_YOUFAIL
528 LOCK( &glIP_Interfaces );
529 if( gIP_Interfaces ) {
530 gIP_Interfaces_Last->Next = iface;
531 gIP_Interfaces_Last = iface;
534 gIP_Interfaces = iface;
535 gIP_Interfaces_Last = iface;
537 RELEASE( &glIP_Interfaces );
539 gIP_DriverInfo.RootNode.Size ++;
542 LEAVE('i', iface->Node.ImplInt);
543 return iface->Node.ImplInt;
547 * \fn tAdapter *IPStack_GetAdapter(char *Path)
548 * \brief Gets/opens an adapter given the path
550 tAdapter *IPStack_GetAdapter(char *Path)
555 ENTER("sPath", Path);
557 LOCK( &glIP_Adapters );
559 // Check if this adapter is already open
560 for( dev = gIP_Adapters; dev; dev = dev->Next )
562 if( strcmp(dev->Device, Path) == 0 ) {
564 RELEASE( &glIP_Adapters );
570 // Ok, so let's open it
571 dev = malloc( sizeof(tAdapter) + strlen(Path) + 1 );
573 RELEASE( &glIP_Adapters );
579 strcpy( dev->Device, Path );
583 dev->DeviceFD = VFS_Open( dev->Device, VFS_OPENFLAG_READ|VFS_OPENFLAG_WRITE );
584 if( dev->DeviceFD == -1 ) {
586 RELEASE( &glIP_Adapters );
591 // Check that it is a network interface
592 tmp = VFS_IOCtl(dev->DeviceFD, 0, NULL);
593 LOG("Device type = %i", tmp);
594 if( tmp != DRV_TYPE_NETWORK ) {
595 Warning("IPStack_GetAdapter: '%s' is not a network interface", dev->Device);
596 VFS_Close( dev->DeviceFD );
598 RELEASE( &glIP_Adapters );
604 VFS_IOCtl(dev->DeviceFD, NET_IOCTL_GETMAC, &dev->MacAddr);
607 dev->Next = gIP_Adapters;
610 RELEASE( &glIP_Adapters );
613 Link_WatchDevice( dev );