Modules/IPStack/arp.c

   1 /*
   2  * Acess2 IP Stack
   3  * - Address Resolution Protocol
   4  */
   5 #define DEBUG   0
   6 #include "ipstack.h"
   7 #include "arp.h"
   8 #include "link.h"
   9
  10 #define ARP_CACHE_SIZE  64
  11 #define ARP_MAX_AGE             (60*60*1000)    // 1Hr
  12
  13 // === IMPORTS ===
  14 extern tInterface       *IPv4_GetInterface(tAdapter *Adapter, tIPv4 Address, int Broadcast);
  15 extern tInterface       *IPv6_GetInterface(tAdapter *Adapter, tIPv6 Address, int Broadcast);
  16
  17 // === PROTOTYPES ===
  18  int    ARP_Initialise();
  19 tMacAddr        ARP_Resolve4(tInterface *Interface, tIPv4 Address);
  20 void    ARP_int_GetPacket(tAdapter *Adapter, tMacAddr From, int Length, void *Buffer);
  21
  22 // === GLOBALS ===
  23 struct sARP_Cache4 {
  24         tIPv4   IP;
  25         tMacAddr        MAC;
  26         Sint64  LastUpdate;
  27         Sint64  LastUsed;
  28 }       *gaARP_Cache4;
  29  int    giARP_Cache4Space;
  30 tMutex  glARP_Cache4;
  31 struct sARP_Cache6 {
  32         tIPv6   IP;
  33         tMacAddr        MAC;
  34         Sint64  LastUpdate;
  35         Sint64  LastUsed;
  36 }       *gaARP_Cache6;
  37  int    giARP_Cache6Space;
  38 tMutex  glARP_Cache6;
  39  int    giARP_LastUpdateID = 0;
  40
  41 // === CODE ===
  42 /**
  43  * \fn int ARP_Initialise()
  44  * \brief Initalise the ARP section
  45  */
  46 int ARP_Initialise()
  47 {
  48         gaARP_Cache4 = malloc( ARP_CACHE_SIZE * sizeof(struct sARP_Cache4) );
  49         memset( gaARP_Cache4, 0, ARP_CACHE_SIZE * sizeof(struct sARP_Cache4) );
  50         giARP_Cache4Space = ARP_CACHE_SIZE;
  51
  52         gaARP_Cache6 = malloc( ARP_CACHE_SIZE * sizeof(struct sARP_Cache6) );
  53         memset( gaARP_Cache6, 0, ARP_CACHE_SIZE * sizeof(struct sARP_Cache6) );
  54         giARP_Cache6Space = ARP_CACHE_SIZE;
  55
  56         Link_RegisterType(0x0806, ARP_int_GetPacket);
  57         return 1;
  58 }
  59
  60 /**
  61  * \brief Resolves a MAC address from an IPv4 address
  62  */
  63 tMacAddr ARP_Resolve4(tInterface *Interface, tIPv4 Address)
  64 {
  65          int    lastID;
  66          int    i;
  67         struct sArpRequest4     req;
  68
  69         ENTER("pInterface xAddress", Interface, Address);
  70
  71         Mutex_Acquire( &glARP_Cache4 );
  72         for( i = 0; i < giARP_Cache4Space; i++ )
  73         {
  74                 if(gaARP_Cache4[i].IP.L != Address.L)   continue;
  75
  76                 // Check if the entry needs to be refreshed
  77                 if( now() - gaARP_Cache4[i].LastUpdate > ARP_MAX_AGE )  break;
  78
  79                 Mutex_Release( &glARP_Cache4 );
  80                 LOG("Return %x:%x:%x:%x:%x:%x",
  81                         gaARP_Cache4[i].MAC.B[0], gaARP_Cache4[i].MAC.B[1],
  82                         gaARP_Cache4[i].MAC.B[2], gaARP_Cache4[i].MAC.B[3],
  83                         gaARP_Cache4[i].MAC.B[4], gaARP_Cache4[i].MAC.B[5]
  84                         );
  85                 LEAVE('-');
  86                 return gaARP_Cache4[i].MAC;
  87         }
  88         Mutex_Release( &glARP_Cache4 );
  89
  90         lastID = giARP_LastUpdateID;
  91
  92         // Create request
  93         Log_Log("ARP4", "Asking for address %i.%i.%i.%i",
  94                 Address.B[0], Address.B[1], Address.B[2], Address.B[3]
  95                 );
  96         req.HWType = htons(0x0001);     // Ethernet
  97         req.Type   = htons(0x0800);
  98         req.HWSize = 6;
  99         req.SWSize = 4;
 100         req.Request = htons(1);
 101         req.SourceMac = Interface->Adapter->MacAddr;
 102         req.SourceIP = Interface->IP4.Address;
 103         req.DestMac = cMAC_BROADCAST;
 104         req.DestIP = Address;
 105
 106         // Send Request
 107         Link_SendPacket(Interface->Adapter, 0x0806, req.DestMac, sizeof(struct sArpRequest4), &req);
 108
 109         // Wait for a reply
 110         for(;;)
 111         {
 112                 while(lastID == giARP_LastUpdateID)     Threads_Yield();
 113                 lastID = giARP_LastUpdateID;
 114
 115                 Mutex_Acquire( &glARP_Cache4 );
 116                 for( i = 0; i < giARP_Cache4Space; i++ )
 117                 {
 118                         if(gaARP_Cache4[i].IP.L != Address.L)   continue;
 119
 120                         Mutex_Release( &glARP_Cache4 );
 121                         return gaARP_Cache4[i].MAC;
 122                 }
 123                 Mutex_Release( &glARP_Cache4 );
 124         }
 125 }
 126
 127 /**
 128  * \brief Updates the ARP Cache entry for an IPv4 Address
 129  */
 130 void ARP_UpdateCache4(tIPv4 SWAddr, tMacAddr HWAddr)
 131 {
 132          int    i;
 133          int    free = -1;
 134          int    oldest = 0;
 135
 136         // Find an entry for the IP address in the cache
 137         Mutex_Acquire(&glARP_Cache4);
 138         for( i = giARP_Cache4Space; i--; )
 139         {
 140                 if(gaARP_Cache4[oldest].LastUpdate > gaARP_Cache4[i].LastUpdate) {
 141                         oldest = i;
 142                 }
 143                 if( gaARP_Cache4[i].IP.L == SWAddr.L )  break;
 144                 if( gaARP_Cache4[i].LastUpdate == 0 && free == -1 )     free = i;
 145         }
 146         // If there was no match, we need to make one
 147         if(i == -1) {
 148                 if(free != -1)
 149                         i = free;
 150                 else
 151                         i = oldest;
 152                 gaARP_Cache4[i].IP = SWAddr;
 153         }
 154
 155         Log_Log("ARP4", "Caching %i.%i.%i.%i (%02x:%02x:%02x:%02x:%02x:%02x) in %i",
 156                 SWAddr.B[0], SWAddr.B[1], SWAddr.B[2], SWAddr.B[3],
 157                 HWAddr.B[0], HWAddr.B[1], HWAddr.B[2], HWAddr.B[3], HWAddr.B[4], HWAddr.B[5],
 158                 i
 159                 );
 160
 161         gaARP_Cache4[i].MAC = HWAddr;
 162         gaARP_Cache4[i].LastUpdate = now();
 163         giARP_LastUpdateID ++;
 164         Mutex_Release(&glARP_Cache4);
 165 }
 166
 167 /**
 168  * \brief Updates the ARP Cache entry for an IPv6 Address
 169  */
 170 void ARP_UpdateCache6(tIPv6 SWAddr, tMacAddr HWAddr)
 171 {
 172          int    i;
 173          int    free = -1;
 174          int    oldest = 0;
 175
 176         // Find an entry for the MAC address in the cache
 177         Mutex_Acquire(&glARP_Cache6);
 178         for( i = giARP_Cache6Space; i--; )
 179         {
 180                 if(gaARP_Cache6[oldest].LastUpdate > gaARP_Cache6[i].LastUpdate) {
 181                         oldest = i;
 182                 }
 183                 if( MAC_EQU(gaARP_Cache6[i].MAC, HWAddr) )      break;
 184                 if( gaARP_Cache6[i].LastUpdate == 0 && free == -1 )     free = i;
 185         }
 186         // If there was no match, we need to make one
 187         if(i == -1) {
 188                 if(free != -1)
 189                         i = free;
 190                 else
 191                         i = oldest;
 192                 gaARP_Cache6[i].MAC = HWAddr;
 193         }
 194
 195         gaARP_Cache6[i].IP = SWAddr;
 196         gaARP_Cache6[i].LastUpdate = now();
 197         giARP_LastUpdateID ++;
 198         Mutex_Release(&glARP_Cache6);
 199 }
 200
 201 /**
 202  * \fn void ARP_int_GetPacket(tAdapter *Adapter, tMacAddr From, int Length, void *Buffer)
 203  * \brief Called when an ARP packet is recieved
 204  */
 205 void ARP_int_GetPacket(tAdapter *Adapter, tMacAddr From, int Length, void *Buffer)
 206 {
 207         tArpRequest4    *req4 = Buffer;
 208         tArpRequest6    *req6 = Buffer;
 209         tInterface      *iface;
 210
 211         // Sanity Check Packet
 212         if( Length < (int)sizeof(tArpRequest4) ) {
 213                 Log_Log("ARP", "Recieved undersized packet");
 214                 return ;
 215         }
 216         if( ntohs(req4->Type) != 0x0800 ) {
 217                 Log_Log("ARP", "Recieved a packet with a bad type (0x%x)", ntohs(req4->Type));
 218                 return ;
 219         }
 220         if( req4->HWSize != 6 ) {
 221                 Log_Log("ARP", "Recieved a packet with HWSize != 6 (%i)", req4->HWSize);
 222                 return;
 223         }
 224         #if ARP_DETECT_SPOOFS
 225         if( !MAC_EQU(req4->SourceMac, From) ) {
 226                 Log_Log("ARP", "ARP spoofing detected "
 227                         "(%02x%02x:%02x%02x:%02x%02x != %02x%02x:%02x%02x:%02x%02x)",
 228                         req4->SourceMac.B[0], req4->SourceMac.B[1], req4->SourceMac.B[2],
 229                         req4->SourceMac.B[3], req4->SourceMac.B[4], req4->SourceMac.B[5],
 230                         From.B[0], From.B[1], From.B[2],
 231                         From.B[3], From.B[4], From.B[5]
 232                         );
 233                 return;
 234         }
 235         #endif
 236
 237         Log_Debug("ARP", "Request ID %i", ntohs(req4->Request));
 238
 239         switch( ntohs(req4->Request) )
 240         {
 241         case 1: // You want my IP?
 242                 Log_Debug("ARP", "ARP Request Address class %i", req4->SWSize);
 243                 // Check what type of IP it is
 244                 switch( req4->SWSize )
 245                 {
 246                 case 4:
 247                         Log_Debug("ARP", "From MAC %02x:%02x:%02x:%02x:%02x:%02x",
 248                                 req4->SourceMac.B[0], req4->SourceMac.B[1],
 249                                 req4->SourceMac.B[2], req4->SourceMac.B[3],
 250                                 req4->SourceMac.B[4], req4->SourceMac.B[5]);
 251                         //Log_Debug("ARP", "to MAC %02x:%02x:%02x:%02x:%02x:%02x",
 252                         //      req4->DestMac.B[0], req4->DestMac.B[1],
 253                         //      req4->DestMac.B[2], req4->DestMac.B[3],
 254                         //      req4->DestMac.B[4], req4->DestMac.B[5]);
 255                         Log_Debug("ARP", "ARP Request IPv4 Address %i.%i.%i.%i from %i.%i.%i.%i",
 256                                 req4->DestIP.B[0], req4->DestIP.B[1], req4->DestIP.B[2],
 257                                 req4->DestIP.B[3],
 258                                 req4->SourceIP.B[0], req4->SourceIP.B[1],
 259                                 req4->SourceIP.B[2], req4->SourceIP.B[3]);
 260                         iface = IPv4_GetInterface(Adapter, req4->DestIP, 0);
 261                         if( iface )
 262                         {
 263                                 Log_Debug("ARP", "Caching sender's IP Address");
 264                                 ARP_UpdateCache4(req4->SourceIP, req4->SourceMac);
 265
 266                                 req4->DestIP = req4->SourceIP;
 267                                 req4->DestMac = req4->SourceMac;
 268                                 req4->SourceIP = iface->IP4.Address;
 269                                 req4->SourceMac = Adapter->MacAddr;
 270                                 req4->Request = htons(2);
 271                                 Log_Debug("ARP", "Sending back us (%02x:%02x:%02x:%02x:%02x:%02x)",
 272                                         req4->SourceMac.B[0], req4->SourceMac.B[1],
 273                                         req4->SourceMac.B[2], req4->SourceMac.B[3],
 274                                         req4->SourceMac.B[4], req4->SourceMac.B[5]);
 275                                 Link_SendPacket(Adapter, 0x0806, req4->DestMac, sizeof(tArpRequest4), req4);
 276                         }
 277                         break;
 278                 case 6:
 279                         if( Length < (int)sizeof(tArpRequest6) ) {
 280                                 Log_Log("ARP", "Recieved undersized packet (IPv6)");
 281                                 return ;
 282                         }
 283                         Log_Debug("ARP", "ARP Request IPv6 Address %08x:%08x:%08x:%08x",
 284                                 ntohl(req6->DestIP.L[0]), ntohl(req6->DestIP.L[1]),
 285                                 ntohl(req6->DestIP.L[2]), ntohl(req6->DestIP.L[3])
 286                                 );
 287                         iface = IPv6_GetInterface(Adapter, req6->DestIP, 0);
 288                         if( iface )
 289                         {
 290                                 req6->DestIP = req6->SourceIP;
 291                                 req6->DestMac = req6->SourceMac;
 292                                 req6->SourceIP = iface->IP6.Address;
 293                                 req6->SourceMac = Adapter->MacAddr;
 294                                 req6->Request = htons(2);
 295                                 Log_Debug("ARP", "Sending back us (%02x:%02x:%02x:%02x:%02x:%02x)",
 296                                         req4->SourceMac.B[0], req4->SourceMac.B[1],
 297                                         req4->SourceMac.B[2], req4->SourceMac.B[3],
 298                                         req4->SourceMac.B[4], req4->SourceMac.B[5]);
 299                                 Link_SendPacket(Adapter, 0x0806, req6->DestMac, sizeof(tArpRequest6), req6);
 300                         }
 301                         break;
 302                 default:
 303                         Log_Debug("ARP", "Unknown Protocol Address size (%i)", req4->SWSize);
 304                         return ;
 305                 }
 306
 307                 break;
 308
 309         case 2: // Ooh! A response!
 310                 // Check what type of IP it is
 311                 switch( req4->SWSize )
 312                 {
 313                 case 4:
 314                         ARP_UpdateCache4( req4->SourceIP, From );
 315                         break;
 316                 case 6:
 317                         if( Length < (int)sizeof(tArpRequest6) ) {
 318                                 Log_Debug("ARP", "Recieved undersized packet (IPv6)");
 319                                 return ;
 320                         }
 321                         ARP_UpdateCache6( req6->SourceIP, From );
 322                         break;
 323                 default:
 324                         Log_Debug("ARP", "Unknown Protocol Address size (%i)", req4->SWSize);
 325                         return ;
 326                 }
 327
 328                 break;
 329
 330         default:
 331                 Log_Warning("ARP", "Unknown Request ID %i", ntohs(req4->Request));
 332                 break;
 333         }
 334 }