Modules/RTL8139 - Fixed bug if BAR is invalid
[tpg/acess2.git] / Modules / Network / RTL8139 / rtl8139.c
1 /*
2  * Acess2 RTL8139 Driver
3  * - By John Hodge (thePowersGang)
4  */
5 #define DEBUG   0
6 #define VERSION ((0<<8)|20)
7 #include <acess.h>
8 #include <modules.h>
9 #include <fs_devfs.h>
10 #include <drv_pci.h>
11 #include <api_drv_network.h>
12 #include <semaphore.h>
13
14 // === CONSTANTS ===
15 #define VENDOR_ID       0x10EC
16 #define DEVICE_ID       0x8139
17
18 enum eRTL8139_Regs
19 {
20         // MAC Address
21         MAC0, MAC1, MAC2,
22         MAC3, MAC4, MAC5,
23         
24         // Multicast Registers
25         MAR0 = 0x08, MAR1, MAR2, MAR3,
26         MAR4, MAR5, MAR6, MAR7,
27         
28         // Transmit status of descriptors 0 - 3
29         TSD0 = 0x10,    TSD1 = 0x14,
30         TSD2 = 0x18,    TSD3 = 0x1C,
31         // Transmit start addresses
32         TSAD0 = 0x20,   TSAD1 = 0x24,
33         TSAD2 = 0x28,   TSAD3 = 0x2C,
34         
35         RBSTART = 0x30, //!< Recieve Buffer Start (DWord)
36         // Early Recieve Byte Count
37         ERBCR = 0x34,   // 16-bits
38         // Early RX Status Register
39         ERSR = 0x36,
40         
41         // ??, ??, ??, RST, RE, TE, ??, ??
42         CMD     = 0x37,
43         
44         CAPR    = 0x38, // Current address of packet read
45         CBA     = 0x3A, // Current Buffer Address - Total byte count in RX buffer
46         
47         IMR     = 0x3C, // Interrupt mask register
48         ISR     = 0x3E, // Interrupt status register
49         
50         TCR     = 0x40, // Transmit Configuration Register
51         RCR     = 0x44, // Recieve Configuration Register
52         TCTR    = 0x48, // 32-bit timer (count)
53         MPC     = 0x4C, // Missed packet count (due to RX overflow)
54         
55         CR_9346 = 0x50,
56         CONFIG0 = 0x51,
57         CONFIG1 = 0x52,
58         // 0x53 resvd
59         TIMERINT = 0x54,        // Fires a timeout when TCTR equals this value
60         
61 };
62
63 #define FLAG_ISR_TOK    0x04
64 #define FLAG_ISR_ROK    0x01
65
66 // === TYPES ===
67 typedef struct sCard
68 {
69         Uint16  IOBase;
70         Uint8   IRQ;
71         
72          int    NumWaitingPackets;
73         
74         char    *ReceiveBuffer;
75         tPAddr  PhysReceiveBuffer;
76          int    ReceiveBufferLength;
77          int    SeenOfs;        //!< End of the most recently seen packet (by IRQ)
78         tMutex  ReadMutex;
79         tSemaphore      ReadSemaphore;
80         
81         char    *TransmitBuffers[4];
82         tPAddr  PhysTransmitBuffers[4];
83         tMutex  TransmitInUse[4];
84         tMutex  CurTXProtector; //!< Protects \a .CurTXDescriptor
85          int    CurTXDescriptor;
86         
87         char    Name[2];
88         tVFS_Node       Node;
89         Uint8   MacAddr[6];
90 }       tCard;
91
92 // === PROTOTYPES ===
93  int    RTL8139_Install(char **Options);
94 char    *RTL8139_ReadDir(tVFS_Node *Node, int Pos);
95 tVFS_Node       *RTL8139_FindDir(tVFS_Node *Node, const char *Filename);
96  int    RTL8139_RootIOCtl(tVFS_Node *Node, int ID, void *Arg);
97 Uint64  RTL8139_Read(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Buffer);
98 Uint64  RTL8139_Write(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Buffer);
99  int    RTL8139_IOCtl(tVFS_Node *Node, int ID, void *Arg);
100 void    RTL8139_IRQHandler(int Num, void *Ptr);
101
102 // === GLOBALS ===
103 MODULE_DEFINE(0, VERSION, RTL8139, RTL8139_Install, NULL, NULL);
104 tDevFS_Driver   gRTL8139_DriverInfo = {
105         NULL, "RTL8139",
106         {
107         .NumACLs = 1,
108         .ACLs = &gVFS_ACL_EveryoneRX,
109         .Flags = VFS_FFLAG_DIRECTORY,
110         .ReadDir = RTL8139_ReadDir,
111         .FindDir = RTL8139_FindDir,
112         .IOCtl = RTL8139_RootIOCtl
113         }
114 };
115  int    giRTL8139_CardCount;
116 tCard   *gaRTL8139_Cards;
117
118 // === CODE ===
119 /**
120  * \brief Installs the RTL8139 Driver
121  */
122 int RTL8139_Install(char **Options)
123 {
124          int    id = -1;
125          int    i = 0;
126         Uint16  base;
127         tCard   *card;
128         
129         giRTL8139_CardCount = PCI_CountDevices(VENDOR_ID, DEVICE_ID);
130         Log_Debug("RTL8139", "%i cards", giRTL8139_CardCount);
131         
132         if( giRTL8139_CardCount == 0 )  return MODULE_ERR_NOTNEEDED;
133         
134         gaRTL8139_Cards = calloc( giRTL8139_CardCount, sizeof(tCard) );
135         
136         for( i = 0 ; (id = PCI_GetDevice(VENDOR_ID, DEVICE_ID, i)) != -1; i ++ )
137         {
138                 card = &gaRTL8139_Cards[i];
139                 base = PCI_GetBAR( id, 0 );
140                 if( !(base & 1) ) {
141                         Log_Warning("RTL8139", "Driver does not support MMIO, skipping card (addr %x)",
142                                 base);
143                         card->IOBase = 0;
144                         card->IRQ = 0;
145                         continue ;
146                 }
147                 base &= ~1;
148                 card->IOBase = base;
149                 card->IRQ = PCI_GetIRQ( id );
150                 
151                 // Install IRQ Handler
152                 IRQ_AddHandler(card->IRQ, RTL8139_IRQHandler, card);
153                 
154                 // Power on
155                 outb( base + CONFIG1, 0x00 );
156
157                 // Reset (0x10 to CMD)
158                 outb( base + CMD, 0x10 );       
159                 while( inb(base + CMD) & 0x10 ) ;
160                 
161                 // Set up recieve buffer
162                 // - Allocate 3 pages below 4GiB for the recieve buffer (Allows 8k+16+1500)
163                 card->ReceiveBuffer = (void*)MM_AllocDMA( 3, 32, &card->PhysReceiveBuffer );
164                 card->ReceiveBufferLength = 8*1024;
165                 outd(base + RBSTART, (Uint32)card->PhysReceiveBuffer);
166                 outd(base + CBA, 0);
167                 outd(base + CAPR, 0);
168                 // Set IMR to Transmit OK and Receive OK
169                 outw(base + IMR, 0x5);
170                 
171                 // Set up transmit buffers
172                 // - 2 non-contiguous pages (each page can fit 2 1500 byte packets)
173                 card->TransmitBuffers[0] = (void*)MM_AllocDMA( 1, 32, &card->PhysTransmitBuffers[0] );
174                 card->TransmitBuffers[1] = card->TransmitBuffers[0] + 0x800;
175                 card->PhysTransmitBuffers[1] = card->PhysTransmitBuffers[0] + 0x800;
176                 
177                 card->TransmitBuffers[2] = (void*)MM_AllocDMA( 1, 32, &card->PhysTransmitBuffers[2] );
178                 card->TransmitBuffers[3] = card->TransmitBuffers[2] + 0x800;
179                 card->PhysTransmitBuffers[3] = card->PhysTransmitBuffers[2] + 0x800;
180                 
181                 outd(base + TSAD0, card->PhysTransmitBuffers[0]);
182                 outd(base + TSAD1, card->PhysTransmitBuffers[1]);
183                 outd(base + TSAD2, card->PhysTransmitBuffers[2]);
184                 outd(base + TSAD3, card->PhysTransmitBuffers[3]);
185                 
186                 // Set recieve buffer size and recieve mask
187                 // - Bit 7 being set tells the card to overflow the recieve buffer if needed
188                 //   (i.e. when the packet starts at the end of the bufffer, it overflows up
189                 //    to 1500 bytes)
190                 outd(base + RCR, 0x8F);
191         
192                 // Recive Enable and Transmit Enable    
193                 outb(base + CMD, 0x0C);
194                 
195                 // Get the card's MAC address
196                 card->MacAddr[0] = inb(base+MAC0);
197                 card->MacAddr[1] = inb(base+MAC1);
198                 card->MacAddr[2] = inb(base+MAC2);
199                 card->MacAddr[3] = inb(base+MAC3);
200                 card->MacAddr[4] = inb(base+MAC4);
201                 card->MacAddr[5] = inb(base+MAC5);
202                 
203                 // Set VFS Node
204                 card->Name[0] = '0'+i;
205                 card->Name[1] = '\0';
206                 card->Node.ImplPtr = card;
207                 card->Node.NumACLs = 0;
208                 card->Node.CTime = now();
209                 card->Node.Write = RTL8139_Write;
210                 card->Node.Read = RTL8139_Read;
211                 card->Node.IOCtl = RTL8139_IOCtl;
212                 
213                 Log_Log("RTL8139", "Card %i 0x%04x, IRQ %i %02x:%02x:%02x:%02x:%02x:%02x",
214                         i, card->IOBase, card->IRQ,
215                         card->MacAddr[0], card->MacAddr[1], card->MacAddr[2],
216                         card->MacAddr[3], card->MacAddr[4], card->MacAddr[5]
217                         );
218         }
219         
220         gRTL8139_DriverInfo.RootNode.Size = giRTL8139_CardCount;
221         DevFS_AddDevice( &gRTL8139_DriverInfo );
222         
223         return MODULE_ERR_OK;
224 }
225
226 // --- Root Functions ---
227 char *RTL8139_ReadDir(tVFS_Node *Node, int Pos)
228 {
229         if( Pos < 0 || Pos >= giRTL8139_CardCount )     return NULL;
230         
231         return strdup( gaRTL8139_Cards[Pos].Name );
232 }
233
234 tVFS_Node *RTL8139_FindDir(tVFS_Node *Node, const char *Filename)
235 {
236         //TODO: It might be an idea to supprt >10 cards
237         if(Filename[0] == '\0' || Filename[1] != '\0')  return NULL;
238         if(Filename[0] < '0' || Filename[0] > '9')      return NULL;
239         return &gaRTL8139_Cards[ Filename[0]-'0' ].Node;
240 }
241
242 const char *csaRTL8139_RootIOCtls[] = {DRV_IOCTLNAMES, NULL};
243 int RTL8139_RootIOCtl(tVFS_Node *Node, int ID, void *Data)
244 {
245         ENTER("pNode iID pData", Node, ID, Data);
246         switch(ID)
247         {
248         BASE_IOCTLS(DRV_TYPE_NETWORK, "RTL8139", VERSION, csaRTL8139_RootIOCtls);
249         }
250         LEAVE('i', 0);
251         return 0;
252 }
253
254 // --- File Functions ---
255 Uint64 RTL8139_Read(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Buffer)
256 {
257         tCard   *card = Node->ImplPtr;
258         Uint16  read_ofs, pkt_length;
259          int    new_read_ofs;
260
261         ENTER("pNode XOffset XLength pBuffer", Node, Offset, Length, Buffer);
262
263 retry:
264         if( Semaphore_Wait( &card->ReadSemaphore, 1 ) != 1 )
265         {
266                 LEAVE_RET('i', 0);
267         }
268         
269         Mutex_Acquire( &card->ReadMutex );
270         
271         read_ofs = inw( card->IOBase + CAPR );
272         LOG("raw read_ofs = %i", read_ofs);
273         read_ofs = (read_ofs + 0x10) & 0xFFFF;
274         LOG("read_ofs = %i", read_ofs);
275         
276         pkt_length = *(Uint16*)&card->ReceiveBuffer[read_ofs+2];
277         
278         // Calculate new read offset
279         new_read_ofs = read_ofs + pkt_length + 4;
280         new_read_ofs = (new_read_ofs + 3) & ~3; // Align
281         if(new_read_ofs > card->ReceiveBufferLength) {
282                 LOG("wrapping read_ofs");
283                 new_read_ofs -= card->ReceiveBufferLength;
284         }
285         new_read_ofs -= 0x10;   // I dunno
286         LOG("new_read_ofs = %i", new_read_ofs);
287         
288         // Check for errors
289         if( *(Uint16*)&card->ReceiveBuffer[read_ofs] & 0x1E ) {
290                 // Update CAPR
291                 outw(card->IOBase + CAPR, new_read_ofs);
292                 Mutex_Release( &card->ReadMutex );
293                 goto retry;     // I feel evil
294         }
295         
296         // Get packet
297         if( Length > pkt_length )       Length = pkt_length;
298         memcpy(Buffer, &card->ReceiveBuffer[read_ofs+4], Length);
299         
300         // Update CAPR
301         outw(card->IOBase + CAPR, new_read_ofs);
302         
303         Mutex_Release( &card->ReadMutex );
304         
305         LEAVE('i', Length);
306         
307         return Length;
308 }
309
310 Uint64 RTL8139_Write(tVFS_Node *Node, Uint64 Offset, Uint64 Length, void *Buffer)
311 {
312          int    td;
313         Uint32  status;
314         tCard   *card = Node->ImplPtr;
315         
316         if( Length > 1500 )     return 0;       // MTU exceeded
317         
318         ENTER("pNode XLength pBuffer", Node, Length, Buffer);
319         
320         // TODO: Implement a semaphore for avaliable transmit buffers
321
322         // Find an avaliable descriptor
323         Mutex_Acquire(&card->CurTXProtector);
324         td = card->CurTXDescriptor;
325         card->CurTXDescriptor ++;
326         card->CurTXDescriptor %= 4;
327         Mutex_Release(&card->CurTXProtector);
328         // - Lock it
329         Mutex_Acquire( &card->TransmitInUse[td] );
330         
331         LOG("td = %i", td);
332         
333         // Transmit using descriptor `td`
334         LOG("card->PhysTransmitBuffers[td] = %P", card->PhysTransmitBuffers[td]);
335         outd(card->IOBase + TSAD0 + td*4, card->PhysTransmitBuffers[td]);
336         LOG("card->TransmitBuffers[td] = %p", card->TransmitBuffers[td]);
337         // Copy to buffer
338         memcpy(card->TransmitBuffers[td], Buffer, Length);
339         // Start
340         status = 0;
341         status |= Length & 0x1FFF;      // 0-12: Length
342         status |= 0 << 13;      // 13: OWN bit
343         status |= (0 & 0x3F) << 16;     // 16-21: Early TX threshold (zero atm, TODO: check)
344         LOG("status = 0x%08x", status);
345         outd(card->IOBase + TSD0 + td*4, status);
346         
347         LEAVE('i', (int)Length);
348         
349         return Length;
350 }
351
352 const char *csaRTL8139_NodeIOCtls[] = {DRV_IOCTLNAMES, NULL};
353 int RTL8139_IOCtl(tVFS_Node *Node, int ID, void *Data)
354 {
355         tCard   *card = Node->ImplPtr;
356         ENTER("pNode iID pData", Node, ID, Data);
357         switch(ID)
358         {
359         BASE_IOCTLS(DRV_TYPE_NETWORK, "RTL8139", VERSION, csaRTL8139_NodeIOCtls);
360         case NET_IOCTL_GETMAC:
361                 if( !CheckMem(Data, 6) ) {
362                         LEAVE('i', -1);
363                         return -1;
364                 }
365                 memcpy( Data, card->MacAddr, 6 );
366                 LEAVE('i', 1);
367                 return 1;
368         }
369         LEAVE('i', 0);
370         return 0;
371 }
372
373 void RTL8139_IRQHandler(int Num, void *Ptr)
374 {
375          int    j;
376         tCard   *card = Ptr;
377         Uint16  status;
378
379         LOG("Num = %i", Num);
380         
381         if( Num != card->IRQ )  return;
382                 
383         status = inw(card->IOBase + ISR);
384         LOG("status = 0x%02x", status);
385                 
386         // Transmit OK, a transmit descriptor is now free
387         if( status & FLAG_ISR_TOK )
388         {
389                 for( j = 0; j < 4; j ++ )
390                 {
391                         if( ind(card->IOBase + TSD0 + j*4) & 0x8000 ) { // TSD TOK
392                                 Mutex_Release( &card->TransmitInUse[j] );
393                                 // TODO: Update semaphore once implemented
394                         }
395                 }
396                 outw(card->IOBase + ISR, FLAG_ISR_TOK);
397         }
398         
399         // Recieve OK, inform read
400         if( status & FLAG_ISR_ROK )
401         {
402                  int    read_ofs, end_ofs;
403                  int    packet_count = 0;
404                  int    len;
405                 
406                 // Scan recieve buffer for packets
407                 end_ofs = inw(card->IOBase + CBA);
408                 read_ofs = card->SeenOfs;
409                 LOG("read_ofs = %i, end_ofs = %i", read_ofs, end_ofs);
410                 if( read_ofs > end_ofs )
411                 {
412                         while( read_ofs < card->ReceiveBufferLength )
413                         {
414                                 packet_count ++;
415                                 len = *(Uint16*)&card->ReceiveBuffer[read_ofs+2];
416                                 LOG("%i 0x%x Pkt Hdr: 0x%04x, len: 0x%04x",
417                                         packet_count, read_ofs,
418                                         *(Uint16*)&card->ReceiveBuffer[read_ofs],
419                                         len
420                                         );
421                                 if(len > 2000) {
422                                         Log_Warning("RTL8139", "IRQ: Packet in buffer exceeds sanity (%i>2000)", len);
423                                 }
424                                 read_ofs += len + 4;
425                                 read_ofs = (read_ofs + 3) & ~3; // Align
426                         }
427                         read_ofs -= card->ReceiveBufferLength;
428                         LOG("wrapped read_ofs");
429                 }
430                 while( read_ofs < end_ofs )
431                 {
432                         packet_count ++;
433                         LOG("%i 0x%x Pkt Hdr: 0x%04x, len: 0x%04x",
434                                 packet_count, read_ofs,
435                                 *(Uint16*)&card->ReceiveBuffer[read_ofs],
436                                 *(Uint16*)&card->ReceiveBuffer[read_ofs+2]
437                                 );
438                         read_ofs += *(Uint16*)&card->ReceiveBuffer[read_ofs+2] + 4;
439                         read_ofs = (read_ofs + 3) & ~3; // Align
440                 }
441                 if( read_ofs != end_ofs ) {
442                         Log_Warning("RTL8139", "IRQ: read_ofs (%i) != end_ofs(%i)", read_ofs, end_ofs);
443                         read_ofs = end_ofs;
444                 }
445                 card->SeenOfs = read_ofs;
446                 
447                 LOG("packet_count = %i, read_ofs = 0x%x", packet_count, read_ofs);
448                 
449                 if( packet_count )
450                 {
451                         if( Semaphore_Signal( &card->ReadSemaphore, packet_count ) != packet_count ) {
452                                 // Oops?
453                         }
454                         VFS_MarkAvaliable( &card->Node, 1 );
455                 }
456                 
457                 outw(card->IOBase + ISR, FLAG_ISR_ROK);
458         }       
459 }

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