114 lines
4.0 KiB
Python
114 lines
4.0 KiB
Python
import unittest
|
|
from litex.gen import *
|
|
|
|
from litex.soc.interconnect import wishbone
|
|
from litex.soc.interconnect.stream_sim import *
|
|
|
|
from liteeth.common import *
|
|
from liteeth.core import LiteEthUDPIPCore
|
|
from liteeth.frontend.etherbone import LiteEthEtherbone
|
|
|
|
from test.model import phy, mac, arp, ip, udp, etherbone
|
|
|
|
ip_address = 0x12345678
|
|
mac_address = 0x12345678abcd
|
|
|
|
|
|
class DUT(Module):
|
|
def __init__(self):
|
|
self.submodules.phy_model = phy.PHY(8, debug=False)
|
|
self.submodules.mac_model = mac.MAC(self.phy_model, debug=False, loopback=False)
|
|
self.submodules.arp_model = arp.ARP(self.mac_model, mac_address, ip_address, debug=False)
|
|
self.submodules.ip_model = ip.IP(self.mac_model, mac_address, ip_address, debug=False, loopback=False)
|
|
self.submodules.udp_model = udp.UDP(self.ip_model, ip_address, debug=False, loopback=False)
|
|
self.submodules.etherbone_model = etherbone.Etherbone(self.udp_model, debug=False)
|
|
|
|
self.submodules.core = LiteEthUDPIPCore(self.phy_model, mac_address, ip_address, 100000)
|
|
self.submodules.etherbone = LiteEthEtherbone(self.core.udp, 1234)
|
|
|
|
self.submodules.sram = wishbone.SRAM(1024)
|
|
self.submodules.interconnect = wishbone.InterconnectPointToPoint(self.etherbone.master.bus, self.sram.bus)
|
|
|
|
|
|
def main_generator(dut):
|
|
test_probe = True
|
|
test_writes = True
|
|
test_reads = True
|
|
|
|
# test probe
|
|
if test_probe:
|
|
packet = etherbone.EtherbonePacket()
|
|
packet.pf = 1
|
|
dut.etherbone_model.send(packet)
|
|
yield from dut.etherbone_model.receive()
|
|
print("probe: " + str(bool(dut.etherbone_model.rx_packet.pr)))
|
|
|
|
for i in range(8):
|
|
# test writes
|
|
if test_writes:
|
|
writes_datas = [j for j in range(16)]
|
|
writes = etherbone.EtherboneWrites(base_addr=0x1000,
|
|
datas=writes_datas)
|
|
record = etherbone.EtherboneRecord()
|
|
record.writes = writes
|
|
record.reads = None
|
|
record.bca = 0
|
|
record.rca = 0
|
|
record.rff = 0
|
|
record.cyc = 0
|
|
record.wca = 0
|
|
record.wff = 0
|
|
record.byte_enable = 0xf
|
|
record.wcount = len(writes_datas)
|
|
record.rcount = 0
|
|
|
|
packet = etherbone.EtherbonePacket()
|
|
packet.records = [record]
|
|
dut.etherbone_model.send(packet)
|
|
for i in range(256):
|
|
yield
|
|
|
|
# test reads
|
|
if test_reads:
|
|
reads_addrs = [0x1000 + 4*j for j in range(16)]
|
|
reads = etherbone.EtherboneReads(base_ret_addr=0x1000,
|
|
addrs=reads_addrs)
|
|
record = etherbone.EtherboneRecord()
|
|
record.writes = None
|
|
record.reads = reads
|
|
record.bca = 0
|
|
record.rca = 0
|
|
record.rff = 0
|
|
record.cyc = 0
|
|
record.wca = 0
|
|
record.wff = 0
|
|
record.byte_enable = 0xf
|
|
record.wcount = 0
|
|
record.rcount = len(reads_addrs)
|
|
|
|
packet = etherbone.EtherbonePacket()
|
|
packet.records = [record]
|
|
dut.etherbone_model.send(packet)
|
|
yield from dut.etherbone_model.receive()
|
|
loopback_writes_datas = []
|
|
loopback_writes_datas = dut.etherbone_model.rx_packet.records.pop().writes.get_datas()
|
|
|
|
# check resultss
|
|
s, l, e = check(writes_datas, loopback_writes_datas)
|
|
print("shift " + str(s) + " / length " + str(l) + " / errors " + str(e))
|
|
|
|
|
|
class TestEtherbone(unittest.TestCase):
|
|
def test(self):
|
|
dut = DUT()
|
|
generators = {
|
|
"sys" : [main_generator(dut)],
|
|
"eth_tx": [dut.phy_model.phy_sink.generator(),
|
|
dut.phy_model.generator()],
|
|
"eth_rx": dut.phy_model.phy_source.generator()
|
|
}
|
|
clocks = {"sys": 10,
|
|
"eth_rx": 10,
|
|
"eth_tx": 10}
|
|
run_simulation(dut, generators, clocks, vcd_name="sim.vcd")
|