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);
24 extern int IPv6_Initialise();
25 //extern int IPv6_Ping(tInterface *Iface, tIPv6 Addr);
28 int IPStack_Install(char **Arguments);
29 int IPStack_IOCtlRoot(tVFS_Node *Node, int ID, void *Data);
30 char *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos);
31 tVFS_Node *IPStack_Root_FindDir(tVFS_Node *Node, char *Name);
32 int IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data);
33 int IPStack_AddInterface(char *Device);
34 tAdapter *IPStack_GetAdapter(char *Path);
35 char *IPStack_Iface_ReadDir(tVFS_Node *Node, int Pos);
36 tVFS_Node *IPStack_Iface_FindDir(tVFS_Node *Node, char *Name);
37 int IPStack_Iface_IOCtl(tVFS_Node *Node, int ID, void *Data);
40 MODULE_DEFINE(0, VERSION, IPStack, IPStack_Install, NULL, NULL);
41 tDevFS_Driver gIP_DriverInfo = {
44 .Size = 0, // Number of interfaces
46 .ACLs = &gVFS_ACL_EveryoneRX,
47 .Flags = VFS_FFLAG_DIRECTORY,
48 .ReadDir = IPStack_Root_ReadDir,
49 .FindDir = IPStack_Root_FindDir,
50 .IOCtl = IPStack_Root_IOCtl
53 tSpinlock glIP_Interfaces = 0;
54 tInterface *gIP_Interfaces = NULL;
55 tInterface *gIP_Interfaces_Last = NULL;
56 int giIP_NextIfaceId = 1;
57 tSpinlock glIP_Adapters = 0;
58 tAdapter *gIP_Adapters = NULL;
59 tSocketFile *gIP_FileTemplates;
63 * \fn int IPStack_Install(char **Arguments)
64 * \brief Intialise the relevant parts of the stack and register with DevFS
66 int IPStack_Install(char **Arguments)
70 // Layer 2 - Data Link Layer
72 // Layer 3 - Network Layer
75 // Layer 4 - Transport Layer
81 // Parse module arguments
82 for( i = 0; Arguments[i]; i++ )
84 //if(strcmp(Arguments[i], "Device") == '=') {
90 DevFS_AddDevice( &gIP_DriverInfo );
96 * \brief Adds a file to the socket list
98 int IPStack_AddFile(tSocketFile *File)
100 Log_Log("IPStack", "Added file '%s'", File->Name);
101 File->Next = gIP_FileTemplates;
102 gIP_FileTemplates = File;
107 * \brief Read from the IP Stack's Device Directory
109 char *IPStack_Root_ReadDir(tVFS_Node *Node, int Pos)
113 ENTER("pNode iPos", Node, Pos);
116 for( iface = gIP_Interfaces; iface && Pos--; iface = iface->Next ) ;
118 // Did we run off the end?
127 Pos = iface->Node.ImplInt;
133 name[0] = '0' + Pos/10;
134 name[1] = '0' + Pos%10;
138 name[0] = '0' + Pos/100;
139 name[1] = '0' + (Pos/10)%10;
140 name[2] = '0' + Pos%10;
145 // Return the pre-generated name
150 * \brief Get the node of an interface
152 tVFS_Node *IPStack_Root_FindDir(tVFS_Node *Node, char *Name)
157 ENTER("pNode sName", Node, Name);
160 while('0' <= Name[i] && Name[i] <= '9')
163 num += Name[i] - '0';
166 if(Name[i] != '\0') {
171 for( iface = gIP_Interfaces; iface; iface = iface->Next )
173 if( iface->Node.ImplInt == num )
175 LEAVE('p', &iface->Node);
183 static const char *casIOCtls_Root[] = { DRV_IOCTLNAMES, "add_interface", NULL };
185 * \brief Handles IOCtls for the IPStack root
187 int IPStack_Root_IOCtl(tVFS_Node *Node, int ID, void *Data)
190 ENTER("pNode iID pData", Node, ID, Data);
194 // --- Standard IOCtls (0-3) ---
196 LEAVE('i', DRV_TYPE_MISC);
197 return DRV_TYPE_MISC;
199 case DRV_IOCTL_IDENT:
200 tmp = ModUtil_SetIdent(Data, "IPStack");
204 case DRV_IOCTL_VERSION:
208 case DRV_IOCTL_LOOKUP:
209 tmp = ModUtil_LookupString( (char**)casIOCtls_Root, (char*)Data );
215 * - Adds a new IP interface and binds it to a device
218 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
219 if( !CheckString( Data ) ) LEAVE_RET('i', -1);
220 tmp = IPStack_AddInterface(Data);
228 * \brief Read from an interface's directory
230 char *IPStack_Iface_ReadDir(tVFS_Node *Node, int Pos)
232 tSocketFile *file = gIP_FileTemplates;
233 while(Pos-- && file) {
234 Log("IPStack_Iface_ReadDir: %s", file->Name);
238 if(!file) return NULL;
240 return strdup(file->Name);
244 * \brief Gets a named node from an interface directory
246 tVFS_Node *IPStack_Iface_FindDir(tVFS_Node *Node, char *Name)
248 tSocketFile *file = gIP_FileTemplates;
250 // Get file definition
251 for(;file;file = file->Next)
253 if( strcmp(file->Name, Name) == 0 ) break;
254 Log("IPStack_Iface_FindDir: strcmp('%s', '%s')", file->Name, Name);
256 if(!file) return NULL;
259 return file->Init(Node->ImplPtr);
263 * \brief Names for interface IOCtl Calls
265 static const char *casIOCtls_Iface[] = {
268 "get_address", "set_address",
270 "get_gateway", "set_gateway",
276 * \brief Handles IOCtls for the IPStack interfaces
278 int IPStack_Iface_IOCtl(tVFS_Node *Node, int ID, void *Data)
281 tInterface *iface = (tInterface*)Node->ImplPtr;
282 ENTER("pNode iID pData", Node, ID, Data);
286 // --- Standard IOCtls (0-3) ---
288 LEAVE('i', DRV_TYPE_MISC);
289 return DRV_TYPE_MISC;
291 case DRV_IOCTL_IDENT:
292 tmp = ModUtil_SetIdent(Data, STR(IDENT));
296 case DRV_IOCTL_VERSION:
300 case DRV_IOCTL_LOOKUP:
301 tmp = ModUtil_LookupString( (char**)casIOCtls_Iface, (char*)Data );
307 * - Get/Set the interface type
314 if( Threads_GetUID() != 0 ) {
315 LOG("Attempt by non-root to alter an interface (%i)", Threads_GetUID());
319 if( !CheckMem( Data, sizeof(int) ) ) {
320 LOG("Invalid pointer %p", Data);
324 switch( *(int*)Data )
328 memset(&iface->IP6, 0, sizeof(tIPv6)); // Clear address
332 memset(&iface->IP4, 0, sizeof(tIPv4));
336 memset(&iface->IP6, 0, sizeof(tIPv6));
343 LEAVE('i', iface->Type);
348 * - Get the interface's address
353 case 0: LEAVE_RET('i', 1);
355 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
356 memcpy( Data, &iface->IP4.Address, sizeof(tIPv4) );
359 if( !CheckMem( Data, sizeof(tIPv6) ) ) LEAVE_RET('i', -1);
360 memcpy( Data, &iface->IP6.Address, sizeof(tIPv6) );
367 * - Get the interface's address
370 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
373 case 0: LEAVE_RET('i', 1);
375 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
376 iface->Type = 0; // One very hacky mutex/trash protector
377 memcpy( &iface->IP4.Address, Data, sizeof(tIPv4) );
381 if( !CheckMem( Data, sizeof(tIPv6) ) ) LEAVE_RET('i', -1);
383 memcpy( &iface->IP6.Address, Data, sizeof(tIPv6) );
391 * - Get/Set the bits in the address subnet
397 switch( iface->Type )
399 case 4: LEAVE_RET('i', iface->IP4.SubnetBits);
400 case 6: LEAVE_RET('i', iface->IP6.SubnetBits);
401 default: LEAVE_RET('i', 0);
406 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
407 if( !CheckMem(Data, sizeof(int)) ) LEAVE_RET('i', -1);
409 // Check and set the subnet bits
410 switch( iface->Type )
413 if( *(int*)Data < 0 || *(int*)Data > 31 ) LEAVE_RET('i', -1);
414 iface->IP4.SubnetBits = *(int*)Data;
415 LEAVE_RET('i', iface->IP4.SubnetBits);
417 if( *(int*)Data < 0 || *(int*)Data > 127 ) LEAVE_RET('i', -1);
418 iface->IP6.SubnetBits = *(int*)Data;
419 LEAVE_RET('i', iface->IP6.SubnetBits);
429 * - Get the interface's IPv4 gateway
437 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
438 memcpy( Data, &iface->IP4.Gateway, sizeof(tIPv4) );
448 * - Get/Set the interface's IPv4 gateway
451 if( Threads_GetUID() != 0 ) LEAVE_RET('i', -1);
458 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
459 iface->Type = 0; // One very hacky mutex/trash protector
460 memcpy( &iface->IP4.Gateway, Data, sizeof(tIPv4) );
471 * - Gets the name of the attached device
475 LEAVE_RET('i', iface->Adapter->DeviceLen);
476 if( !CheckMem( Data, iface->Adapter->DeviceLen+1 ) )
478 strcpy( Data, iface->Adapter->Device );
479 return iface->Adapter->DeviceLen;
483 * - Send an ICMP Echo
492 if( !CheckMem( Data, sizeof(tIPv4) ) ) LEAVE_RET('i', -1);
493 tmp = IPv4_Ping(iface, *(tIPv4*)Data);
510 * \fn int IPStack_AddInterface(char *Device)
511 * \brief Adds an interface to the list
513 int IPStack_AddInterface(char *Device)
517 ENTER("sDevice", Device);
519 iface = malloc(sizeof(tInterface));
522 return -2; // Return ERR_MYBAD
526 iface->Type = 0; // Unset type
529 iface->Node.ImplPtr = iface;
530 iface->Node.ImplInt = giIP_NextIfaceId++;
531 iface->Node.Flags = VFS_FFLAG_DIRECTORY;
532 iface->Node.Size = -1;
533 iface->Node.NumACLs = 1;
534 iface->Node.ACLs = &gVFS_ACL_EveryoneRX;
535 iface->Node.ReadDir = IPStack_Iface_ReadDir;
536 iface->Node.FindDir = IPStack_Iface_FindDir;
537 iface->Node.IOCtl = IPStack_Iface_IOCtl;
540 iface->TimeoutDelay = DEFAULT_TIMEOUT;
542 // Get adapter handle
543 iface->Adapter = IPStack_GetAdapter(Device);
544 if( !iface->Adapter ) {
547 return -1; // Return ERR_YOUFAIL
551 LOCK( &glIP_Interfaces );
552 if( gIP_Interfaces ) {
553 gIP_Interfaces_Last->Next = iface;
554 gIP_Interfaces_Last = iface;
557 gIP_Interfaces = iface;
558 gIP_Interfaces_Last = iface;
560 RELEASE( &glIP_Interfaces );
562 gIP_DriverInfo.RootNode.Size ++;
565 LEAVE('i', iface->Node.ImplInt);
566 return iface->Node.ImplInt;
570 * \fn tAdapter *IPStack_GetAdapter(char *Path)
571 * \brief Gets/opens an adapter given the path
573 tAdapter *IPStack_GetAdapter(char *Path)
578 ENTER("sPath", Path);
580 LOCK( &glIP_Adapters );
582 // Check if this adapter is already open
583 for( dev = gIP_Adapters; dev; dev = dev->Next )
585 if( strcmp(dev->Device, Path) == 0 ) {
587 RELEASE( &glIP_Adapters );
593 // Ok, so let's open it
594 dev = malloc( sizeof(tAdapter) + strlen(Path) + 1 );
596 RELEASE( &glIP_Adapters );
602 strcpy( dev->Device, Path );
604 dev->DeviceLen = strlen(Path);
607 dev->DeviceFD = VFS_Open( dev->Device, VFS_OPENFLAG_READ|VFS_OPENFLAG_WRITE );
608 if( dev->DeviceFD == -1 ) {
610 RELEASE( &glIP_Adapters );
615 // Check that it is a network interface
616 tmp = VFS_IOCtl(dev->DeviceFD, 0, NULL);
617 LOG("Device type = %i", tmp);
618 if( tmp != DRV_TYPE_NETWORK ) {
619 Warning("IPStack_GetAdapter: '%s' is not a network interface", dev->Device);
620 VFS_Close( dev->DeviceFD );
622 RELEASE( &glIP_Adapters );
628 VFS_IOCtl(dev->DeviceFD, NET_IOCTL_GETMAC, &dev->MacAddr);
631 dev->Next = gIP_Adapters;
634 RELEASE( &glIP_Adapters );
637 Link_WatchDevice( dev );