liteeth/test/etherbone_tb.py

   1 from migen.fhdl.std import *
   2 from migen.bus import wishbone
   3 from migen.bus.transactions import *
   4 from migen.sim.generic import run_simulation
   5
   6 from liteeth.common import *
   7 from liteeth.core import LiteEthUDPIPCore
   8 from liteeth.core.etherbone import LiteEthEtherbone
   9
  10 from liteeth.test.common import *
  11 from liteeth.test.model import phy, mac, arp, ip, udp, etherbone
  12
  13 ip_address = 0x12345678
  14 mac_address = 0x12345678abcd
  15
  16 class TB(Module):
  17         def __init__(self):
  18                 self.submodules.phy_model = phy.PHY(8, debug=True)
  19                 self.submodules.mac_model = mac.MAC(self.phy_model, debug=True, loopback=False)
  20                 self.submodules.arp_model = arp.ARP(self.mac_model, mac_address, ip_address, debug=False)
  21                 self.submodules.ip_model = ip.IP(self.mac_model, mac_address, ip_address, debug=True, loopback=False)
  22                 self.submodules.udp_model = udp.UDP(self.ip_model, ip_address, debug=True, loopback=False)
  23                 self.submodules.etherbone_model = etherbone.Etherbone(self.udp_model, debug=True)
  24
  25                 self.submodules.core = LiteEthUDPIPCore(self.phy_model, mac_address, ip_address, 100000)
  26                 self.submodules.etherbone = LiteEthEtherbone(self.core.udp, 20000)
  27
  28                 self.submodules.sram = wishbone.SRAM(1024)
  29                 self.submodules.interconnect = wishbone.InterconnectPointToPoint(self.etherbone.master.bus, self.sram.bus)
  30
  31
  32
  33                 # use sys_clk for each clock_domain
  34                 self.clock_domains.cd_eth_rx = ClockDomain()
  35                 self.clock_domains.cd_eth_tx = ClockDomain()
  36                 self.comb += [
  37                         self.cd_eth_rx.clk.eq(ClockSignal()),
  38                         self.cd_eth_rx.rst.eq(ResetSignal()),
  39                         self.cd_eth_tx.clk.eq(ClockSignal()),
  40                         self.cd_eth_tx.rst.eq(ResetSignal()),
  41                 ]
  42
  43         def gen_simulation(self, selfp):
  44                 selfp.cd_eth_rx.rst = 1
  45                 selfp.cd_eth_tx.rst = 1
  46                 yield
  47                 selfp.cd_eth_rx.rst = 0
  48                 selfp.cd_eth_tx.rst = 0
  49
  50                 for i in range(100):
  51                         yield
  52
  53                 test_probe = False
  54                 test_writes = True
  55                 test_reads = True
  56
  57                 # test probe
  58                 if test_probe:
  59                         packet = etherbone.EtherbonePacket()
  60                         packet.pf = 1
  61                         self.etherbone_model.send(packet)
  62
  63                         for i in range(1024):
  64                                 yield
  65
  66                 # test writes
  67                 if test_writes:
  68                         writes = etherbone.EtherboneWrites(base_addr=0x1000)
  69                         for i in range(16):
  70                                 writes.add(etherbone.EtherboneWrite(i))
  71                         record = etherbone.EtherboneRecord()
  72                         record.writes = writes
  73                         record.reads = None
  74                         record.bca = 0
  75                         record.rca = 0
  76                         record.rff = 0
  77                         record.cyc = 0
  78                         record.wca = 0
  79                         record.wff = 0
  80                         record.byte_enable = 0xf
  81                         record.wcount = 16
  82                         record.rcount = 0
  83
  84                         packet = etherbone.EtherbonePacket()
  85                         packet.records = [record]
  86                         self.etherbone_model.send(packet)
  87
  88                         for i in range(1024):
  89                                 yield
  90
  91                 # test reads
  92                 if test_reads:
  93                         reads = etherbone.EtherboneReads(base_ret_addr=0x1000)
  94                         for i in range(16):
  95                                 reads.add(etherbone.EtherboneRead(i))
  96                         record = etherbone.EtherboneRecord()
  97                         record.writes = None
  98                         record.reads = reads
  99                         record.bca = 0
 100                         record.rca = 0
 101                         record.rff = 0
 102                         record.cyc = 0
 103                         record.wca = 0
 104                         record.wff = 0
 105                         record.byte_enable = 0xf
 106                         record.wcount = 0
 107                         record.rcount = 16
 108
 109                         packet = etherbone.EtherbonePacket()
 110                         packet.records = [record]
 111                         self.etherbone_model.send(packet)
 112
 113                         for i in range(1024):
 114                                 yield
 115
 116 if __name__ == "__main__":
 117         run_simulation(TB(), ncycles=4096, vcd_name="my.vcd", keep_files=True)