1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
// MIT License, Copyright (c) 2020 Marvin Borner
// Uses parts of the ToAruOS Project, released under the terms of the NCSA
// Copyright (C) 2011-2018 K. Lange
#include <drivers/cpu.h>
#include <def.h>
#include <drivers/interrupts.h>
#include <mem.h>
#include <net.h>
#include <drivers/pci.h>
#include <print.h>
#include <drivers/rtl8139.h>
static int rtl_irq = 0;
static u8 mac[6] = { 0 };
static u8 *last_packet = NULL;
static u8 *rtl_rx_buffer = NULL;
static u32 rtl_device_pci = 0;
static u32 rtl_iobase = 0;
static u32 cur_rx = 0;
u8 *rtl8139_get_mac(void)
{
if (!rtl_device_pci)
return NULL;
return mac;
}
static void rtl8139_receive_packet(void)
{
while ((inb(rtl_iobase + RTL_PORT_CMD) & 0x01) == 0) {
int offset = cur_rx % 0x2000;
u32 *buf_start = (u32 *)((u32)rtl_rx_buffer + offset);
u32 rx_status = buf_start[0];
u32 rx_size = rx_status >> 16;
if (rx_status & (0x0020 | 0x0010 | 0x0004 | 0x0002)) {
print("RX Error\n");
} else {
u8 *buf_8 = (u8 *)&(buf_start[1]);
last_packet = malloc(rx_size);
u32 packet_end = (u32)buf_8 + rx_size;
if (packet_end > (u32)rtl_rx_buffer + 0x2000) {
u32 s = ((u32)rtl_rx_buffer + 0x2000) - (u32)buf_8;
memcpy(last_packet, buf_8, s);
memcpy((void *)((u32)last_packet + s), rtl_rx_buffer, rx_size - s);
} else {
memcpy(last_packet, buf_8, rx_size);
}
u16 *t = (u16 *)(rtl_rx_buffer + cur_rx);
ethernet_handle_packet((struct ethernet_packet *)last_packet, *(t + 1));
}
cur_rx = (cur_rx + rx_size + 4 + 3) & ~3;
outw(rtl_iobase + RTL_PORT_RXPTR, cur_rx - 16);
}
}
static u8 tsad_array[4] = { 0x20, 0x24, 0x28, 0x2C };
static u8 tsd_array[4] = { 0x10, 0x14, 0x18, 0x1C };
static u8 tx_current = 0;
void rtl8139_send_packet(void *data, u32 len)
{
if (!rtl_device_pci)
return;
outl(rtl_iobase + tsad_array[tx_current], (u32)data);
outl(rtl_iobase + tsd_array[tx_current++], len);
if (tx_current > 3)
tx_current = 0;
}
static void rtl8139_find(u32 device, u16 vendor_id, u16 device_id, void *extra)
{
if ((vendor_id == 0x10ec) && (device_id == 0x8139)) {
*((u32 *)extra) = device;
}
}
static void rtl8139_irq_handler(struct regs *r)
{
UNUSED(r);
u16 status = inw(rtl_iobase + RTL_PORT_ISR);
if (!status)
return;
outw(rtl_iobase + RTL_PORT_ISR, status);
if (status & 0x01 || status & 0x02)
rtl8139_receive_packet();
}
static void rtl8139_init(void)
{
if (!rtl_device_pci)
return;
u16 command_reg = (u16)pci_read_field(rtl_device_pci, PCI_COMMAND, 4);
if ((command_reg & (1 << 2)) == 0) {
command_reg |= (1 << 2);
pci_write_field(rtl_device_pci, PCI_COMMAND, command_reg);
}
rtl_irq = pci_get_interrupt(rtl_device_pci);
irq_install_handler(rtl_irq, rtl8139_irq_handler);
u32 rtl_bar0 = pci_read_field(rtl_device_pci, PCI_BAR0, 4);
/* u32 rtl_bar1 = pci_read_field(rtl_device_pci, PCI_BAR1, 4); */
rtl_iobase = 0;
if (rtl_bar0 & 0x00000001)
rtl_iobase = rtl_bar0 & 0xFFFFFFFC;
// Get mac address
for (int i = 0; i < 6; ++i)
mac[i] = inb(rtl_iobase + RTL_PORT_MAC + i);
printf("Mac address: %x:%x:%x:%x:%x:%x\n", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
// Activate
outb(rtl_iobase + RTL_PORT_CONFIG, 0x0);
// Reset
outb((u16)(rtl_iobase + RTL_PORT_CMD), 0x10);
while ((inb(rtl_iobase + RTL_PORT_CMD) & 0x10) != 0)
;
// Set receive buffer
rtl_rx_buffer = (u8 *)malloc(0x3000);
memset(rtl_rx_buffer, 0, 0x3000);
outl(rtl_iobase + RTL_PORT_RBSTART, (u32)rtl_rx_buffer);
// Enable IRQs
outw(rtl_iobase + RTL_PORT_IMR, 0x8000 | /* PCI error */
0x4000 | /* PCS timeout */
0x40 | /* Rx FIFO over */
0x20 | /* Rx underrun */
0x10 | /* Rx overflow */
0x08 | /* Tx error */
0x04 | /* Tx okay */
0x02 | /* Rx error */
0x01 /* Rx okay */
);
// Configure transmit
outl(rtl_iobase + RTL_PORT_TCR, 0);
// Configure receive
outl(rtl_iobase + RTL_PORT_RCR, (0) | /* 8K receive */
0x08 | /* broadcast */
0x01 /* all physical */
);
// Enable receive and transmit
outb(rtl_iobase + RTL_PORT_CMD, 0x08 | 0x04);
// Reset rx statistics
outl(rtl_iobase + RTL_PORT_RXMISS, 0);
}
int rtl8139_installed(void)
{
return rtl_device_pci != 0;
}
int rtl8139_install(void)
{
pci_scan(&rtl8139_find, -1, &rtl_device_pci);
if (rtl_device_pci) {
print("Found rtl8139 card\n");
rtl8139_init();
}
return rtl_device_pci;
}
|
