IPStack - Heaps of network fixes
[tpg/acess2.git] / Modules / IPStack / link.c
1 /*
2  * Acess2 IP Stack
3  * - Link/Media Layer Interface
4  */
5 #include "ipstack.h"
6 #include "link.h"
7
8 // === CONSTANTS ===
9 #define MAX_PACKET_SIZE 2048
10
11 // === PROTOTYPES ===
12 void    Link_RegisterType(Uint16 Type, tPacketCallback Callback);
13 void    Link_InitCRC();
14 Uint32  Link_CalculateCRC(void *Data, int Length);
15 void    Link_SendPacket(tAdapter *Adapter, Uint16 Type, tMacAddr To, int Length, void *Buffer);
16 void    Link_WatchDevice(tAdapter *Adapter);
17
18 // === GLOBALS ===
19  int    giRegisteredTypes = 0;
20  int    giRegisteredTypeSpace = 0;
21 struct {
22         Uint16  Type;
23         tPacketCallback Callback;
24 }       *gaRegisteredTypes;
25  int    gbLink_CRCTableGenerated = 0;
26 Uint32  gaiLink_CRCTable[256];
27
28 // === CODE ===
29 /**
30  * \fn void Link_RegisterType(Uint16 Type, tPacketCallback Callback)
31  * \brief Registers a callback for a specific packet type
32  * 
33  * \todo Make thread safe (place a mutex on the list)
34  */
35 void Link_RegisterType(Uint16 Type, tPacketCallback Callback)
36 {
37          int    i;
38         void    *tmp;
39         
40         for( i = giRegisteredTypes; i -- ; )
41         {
42                 if(gaRegisteredTypes[i].Type == Type) {
43                         Log_Warning("Net Link", "Attempt to register 0x%x twice", Type);
44                         return ;
45                 }
46                 // Ooh! Free slot!
47                 if(gaRegisteredTypes[i].Callback == NULL)       break;
48         }
49         
50         if(i == -1)
51         {
52                 giRegisteredTypeSpace += 5;
53                 tmp = realloc(gaRegisteredTypes, giRegisteredTypeSpace*sizeof(*gaRegisteredTypes));
54                 if(!tmp) {
55                         Log_Warning("Net Link",
56                                 "Out of heap space! (Attempted to allocate %i)",
57                                 giRegisteredTypeSpace*sizeof(*gaRegisteredTypes)
58                                 );
59                         return ;
60                 }
61                 gaRegisteredTypes = tmp;
62                 i = giRegisteredTypes;
63                 giRegisteredTypes ++;
64         }
65         
66         gaRegisteredTypes[i].Callback = Callback;
67         gaRegisteredTypes[i].Type = Type;
68 }
69
70 /**
71  * \fn void Link_SendPacket(tAdapter *Adapter, Uint16 Type, tMacAddr To, int Length, void *Buffer)
72  * \brief Formats and sends a packet on the specified interface
73  */
74 void Link_SendPacket(tAdapter *Adapter, Uint16 Type, tMacAddr To, int Length, void *Buffer)
75 {
76          int    bufSize = sizeof(tEthernetHeader) + ((Length+3)&~3) + 4;
77         Uint8   buf[bufSize];   // dynamic stack arrays ftw!
78         tEthernetHeader *hdr = (void*)buf;
79         
80         Log_Log("Net Link", "Sending %i bytes to %02x:%02x:%02x:%02x:%02x:%02x (Type 0x%x)",
81                 Length, To.B[0], To.B[1], To.B[2], To.B[3], To.B[4], To.B[5], Type);
82         
83         hdr->Dest = To;
84         hdr->Src = Adapter->MacAddr;
85         hdr->Type = htons(Type);
86         
87         memcpy(hdr->Data, Buffer, Length);
88         
89         *(Uint32*) &hdr->Data[bufSize-4] = 0;
90         *(Uint32*) &hdr->Data[bufSize-4] = htonl( Link_CalculateCRC(buf, bufSize) );
91         
92         VFS_Write(Adapter->DeviceFD, bufSize, buf);
93 }
94
95 /**
96  * \fn void Link_WatchDevice(tAdapter *Adapter)
97  * \brief Spawns a worker thread to watch the specified adapter
98  */
99 void Link_WatchDevice(tAdapter *Adapter)
100 {
101          int    tid = Proc_SpawnWorker();       // Create a new worker thread
102         
103         if(tid < 0) {
104                 Log_Warning("NET", "Unable to create watcher thread for '%s'", Adapter->Device);
105                 return ;
106         }
107         
108         if(tid > 0) {
109                 Log_Log("Net Link", "Watching '%s' using tid %i", Adapter->Device, tid);
110                 return ;
111         }
112         
113         if( !gbLink_CRCTableGenerated )
114                 Link_InitCRC();
115         
116         Threads_SetName(Adapter->Device);
117         Log_Log("Net Link", "Thread %i watching '%s'", Threads_GetTID(), Adapter->Device);
118         
119         // Child Thread
120         while(Adapter->DeviceFD != -1)
121         {
122                 Uint8   buf[MAX_PACKET_SIZE];
123                 tEthernetHeader *hdr = (void*)buf;
124                  int    ret, i;
125                 Uint32  checksum;
126                 
127                 // Wait for a packet (Read on a network device is blocking)
128                 //Log_Debug("NET", "Waiting on adapter FD#0x%x", Adapter->DeviceFD);
129                 ret = VFS_Read(Adapter->DeviceFD, MAX_PACKET_SIZE, buf);
130                 if(ret == -1)   break;
131                 
132                 if(ret <= (int)sizeof(tEthernetHeader)) {
133                         Log_Log("NET", "Recieved an undersized packet");
134                         continue;
135                 }
136                 
137                 Log_Log("Net Link",
138                         "Packet from %02x:%02x:%02x:%02x:%02x:%02x"
139                         " to %02x:%02x:%02x:%02x:%02x:%02x",
140                         hdr->Src.B[0], hdr->Src.B[1], hdr->Src.B[2],
141                         hdr->Src.B[3], hdr->Src.B[4], hdr->Src.B[5],
142                         hdr->Dest.B[0], hdr->Dest.B[1], hdr->Dest.B[2],
143                         hdr->Dest.B[3], hdr->Dest.B[4], hdr->Dest.B[5]
144                         );
145                 checksum = *(Uint32*)&hdr->Data[ret-sizeof(tEthernetHeader)-4];
146                 //Log_Log("NET", "Checksum 0x%08x", checksum);
147                 // TODO: Check checksum
148                 
149                 // Check if there is a registered callback for this packet type
150                 for( i = giRegisteredTypes; i--; )
151                 {
152                         if(gaRegisteredTypes[i].Type == ntohs(hdr->Type))       break;
153                 }
154                 // No? Ignore it
155                 if( i == -1 ) {
156                         Log_Log("Net Link", "Unregistered type 0x%x", ntohs(hdr->Type));
157                         continue;
158                 }
159                 
160                 // Call the callback
161                 gaRegisteredTypes[i].Callback(
162                         Adapter,
163                         hdr->Src,
164                         ret - sizeof(tEthernetHeader),
165                         hdr->Data
166                         );
167         }
168         
169         Log_Log("NET", "Watcher terminated (file closed)");
170         
171         Threads_Exit(0, 0);
172 }
173
174 // From http://www.cl.cam.ac.uk/research/srg/bluebook/21/crc/node6.html
175 #define QUOTIENT        0x04c11db7
176 void Link_InitCRC(void)
177 {
178      int        i, j;
179     Uint32      crc;
180
181     for (i = 0; i < 256; i++)
182     {
183         crc = i << 24;
184         for (j = 0; j < 8; j++)
185         {
186             if (crc & 0x80000000)
187                 crc = (crc << 1) ^ QUOTIENT;
188             else
189                 crc = crc << 1;
190         }
191         gaiLink_CRCTable[i] = crc;
192     }
193         
194         gbLink_CRCTableGenerated = 1;
195 }
196
197 Uint32 Link_CalculateCRC(void *Data, int Length)
198 {
199         // x32 + x26 + x23 + x22 + x16 + x12 + x11 + x10 + x8 + x7 + x5 + x4 + x2 + x + 1
200         Uint32  result;
201      int        i;
202         Uint32  *data = Data;
203     
204     if(Length < 4)      return 0;
205
206     result = *data++ << 24;
207     result |= *data++ << 16;
208     result |= *data++ << 8;
209     result |= *data++;
210     result = ~ result;
211     Length -= 4;
212     
213     for( i = 0; i < Length; i++ )
214     {
215         result = (result << 8 | *data++) ^ gaiLink_CRCTable[result >> 24];
216     }
217     
218     return ~result;
219 }

UCC git Repository :: git.ucc.asn.au