X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;ds=sidebyside;f=KernelLand%2FModules%2FIPStack%2Farp.c;h=2957eeeac487c6efbf5fec35b8121ca1627459b0;hb=2a49f5a6be4fd478ae4249115ff2a3bf0e34d7e5;hp=e152e33d707facb99910caf98b44a1b61d7a62c6;hpb=33495f1efd207f0af4f804858f247db0983fcb8f;p=tpg%2Facess2.git

diff --git a/KernelLand/Modules/IPStack/arp.c b/KernelLand/Modules/IPStack/arp.c
index e152e33d..2957eeea 100644
--- a/KernelLand/Modules/IPStack/arp.c
+++ b/KernelLand/Modules/IPStack/arp.c
@@ -9,9 +9,11 @@
 #include "link.h"
 #include "ipv4.h"	// For IPv4_Netmask
 #include "include/adapters_int.h"	// for MAC addr
+#include <semaphore.h>
+#include <timers.h>
 
 #define ARPv6	0
-#define	ARP_CACHE_SIZE	64
+#define	ARP_CACHE_SIZE	128
 #define	ARP_MAX_AGE		(60*60*1000)	// 1Hr
 
 // === IMPORTS ===
@@ -34,6 +36,8 @@ struct sARP_Cache4 {
 }	*gaARP_Cache4;
  int	giARP_Cache4Space;
 tMutex	glARP_Cache4;
+tSemaphore	gARP_Cache4Semaphore;
+ int	giARP_WaitingThreads;
 #if ARPv6
 struct sARP_Cache6 {
 	tIPv6	IP;
@@ -44,7 +48,6 @@ struct sARP_Cache6 {
  int	giARP_Cache6Space;
 tMutex	glARP_Cache6;
 #endif
-volatile int	giARP_LastUpdateID = 0;
 
 // === CODE ===
 /**
@@ -64,6 +67,7 @@ int ARP_Initialise()
 	#endif
 	
 	Link_RegisterType(0x0806, ARP_int_GetPacket);
+	Semaphore_Init(&gARP_Cache4Semaphore, 0, 0, "ARP4", "Cache Changes");
 	return 1;
 }
 
@@ -72,10 +76,8 @@ int ARP_Initialise()
  */
 tMacAddr ARP_Resolve4(tInterface *Interface, tIPv4 Address)
 {
-	 int	lastID;
 	 int	i;
 	struct sArpRequest4	req;
-	Sint64	timeout;
 	
 	ENTER("pInterface xAddress", Interface, Address);
 	
@@ -133,10 +135,9 @@ tMacAddr ARP_Resolve4(tInterface *Interface, tIPv4 Address)
 		LEAVE('-');
 		return gaARP_Cache4[i].MAC;
 	}
+	giARP_WaitingThreads ++;
 	Mutex_Release( &glARP_Cache4 );
 	
-	lastID = giARP_LastUpdateID;
-	
 	// Create request
 	Log_Log("ARP4", "Asking for address %i.%i.%i.%i",
 		Address.B[0], Address.B[1], Address.B[2], Address.B[3]
@@ -151,35 +152,34 @@ tMacAddr ARP_Resolve4(tInterface *Interface, tIPv4 Address)
 	req.DestMac = cMAC_BROADCAST;
 	req.DestIP = Address;
 
-	tIPStackBuffer	*buffer = IPStack_Buffer_CreateBuffer(3);	// Assumes only a header and footer at link layer
+	// Assumes only a header and footer at link layer
+	tIPStackBuffer	*buffer = IPStack_Buffer_CreateBuffer(3);
 	IPStack_Buffer_AppendSubBuffer(buffer, sizeof(struct sArpRequest4), 0, &req, NULL, NULL);
 
 	// Send Request
 	Link_SendPacket(Interface->Adapter, 0x0806, req.DestMac, buffer);
 
+	// Clean up
 	IPStack_Buffer_DestroyBuffer(buffer);
 	
-	timeout = now() + Interface->TimeoutDelay;
-	
 	// Wait for a reply
+	Time_ScheduleTimer(NULL, Interface->TimeoutDelay);
 	for(;;)
 	{
-		while(lastID == giARP_LastUpdateID && now() < timeout) {
-			Threads_Yield();
-		}
-		
-		if( now() >= timeout ) {
+		if( Semaphore_Wait(&gARP_Cache4Semaphore, 1) != 1 )
+		{
+			giARP_WaitingThreads --;
 			Log_Log("ARP4", "Timeout");
-			break;	// Timeout
+			break;
 		}
-		
-		lastID = giARP_LastUpdateID;
+		Log_Debug("ARP4", "Cache change");
 		
 		Mutex_Acquire( &glARP_Cache4 );
 		for( i = 0; i < giARP_Cache4Space; i++ )
 		{
 			if(gaARP_Cache4[i].IP.L != Address.L)	continue;
 			
+			giARP_WaitingThreads --;
 			Mutex_Release( &glARP_Cache4 );
 			Log_Debug("ARP4", "Return %02x:%02x:%02x:%02x:%02x:%02x",
 				gaARP_Cache4[i].MAC.B[0], gaARP_Cache4[i].MAC.B[1], 
@@ -221,17 +221,20 @@ void ARP_UpdateCache4(tIPv4 SWAddr, tMacAddr HWAddr)
 		else
 			i = oldest;
 	}
-	
-	Log_Log("ARP4", "Caching %i.%i.%i.%i (%02x:%02x:%02x:%02x:%02x:%02x) in %i",
-		SWAddr.B[0], SWAddr.B[1], SWAddr.B[2], SWAddr.B[3],
-		HWAddr.B[0], HWAddr.B[1], HWAddr.B[2], HWAddr.B[3], HWAddr.B[4], HWAddr.B[5],
-		i
-		);
+
+	if( memcmp(&gaARP_Cache4[i].MAC, &HWAddr, sizeof(HWAddr)) != 0 )
+	{
+		Log_Log("ARP4", "Caching %i.%i.%i.%i (%02x:%02x:%02x:%02x:%02x:%02x) in %i",
+			SWAddr.B[0], SWAddr.B[1], SWAddr.B[2], SWAddr.B[3],
+			HWAddr.B[0], HWAddr.B[1], HWAddr.B[2], HWAddr.B[3], HWAddr.B[4], HWAddr.B[5],
+			i
+			);
 		
-	gaARP_Cache4[i].IP = SWAddr;
-	gaARP_Cache4[i].MAC = HWAddr;
-	gaARP_Cache4[i].LastUpdate = now();
-	giARP_LastUpdateID ++;
+		gaARP_Cache4[i].IP = SWAddr;
+		gaARP_Cache4[i].MAC = HWAddr;
+		gaARP_Cache4[i].LastUpdate = now();
+		Semaphore_Signal(&gARP_Cache4Semaphore, giARP_WaitingThreads);
+	}
 	Mutex_Release(&glARP_Cache4);
 }