60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
# This file is Copyright (c) 2015-2019 Florent Kermarrec <florent@enjoy-digital.fr>
|
|
# License: BSD
|
|
|
|
import math
|
|
|
|
from litex.soc.interconnect.stream_sim import *
|
|
from litex.tools.remote.etherbone import *
|
|
|
|
from liteeth.common import *
|
|
|
|
from test.model import udp
|
|
|
|
# Helpers ------------------------------------------------------------------------------------------
|
|
|
|
def print_etherbone(s):
|
|
print_with_prefix(s, "[ETHERBONE]")
|
|
|
|
|
|
# Etherbone ----------------------------------------------------------------------------------------
|
|
|
|
class Etherbone(Module):
|
|
def __init__(self, udp, debug=False):
|
|
self.udp = udp
|
|
self.debug = debug
|
|
self.tx_packets = []
|
|
self.tx_packet = EtherbonePacket()
|
|
self.rx_packet = EtherbonePacket()
|
|
|
|
udp.set_etherbone_callback(self.callback)
|
|
|
|
def send(self, packet):
|
|
packet.encode()
|
|
if self.debug:
|
|
print_etherbone(">>>>>>>>")
|
|
print_etherbone(packet)
|
|
udp_packet = udp.UDPPacket(packet)
|
|
udp_packet.src_port = 0x1234 # FIXME
|
|
udp_packet.dst_port = 0x1234 # FIXME
|
|
udp_packet.length = len(packet)
|
|
udp_packet.checksum = 0
|
|
self.udp.send(udp_packet)
|
|
|
|
def receive(self):
|
|
self.rx_packet = EtherbonePacket()
|
|
while not self.rx_packet.done:
|
|
yield
|
|
|
|
def callback(self, packet):
|
|
packet = EtherbonePacket(packet)
|
|
packet.decode()
|
|
if self.debug:
|
|
print_etherbone("<<<<<<<<")
|
|
print_etherbone(packet)
|
|
self.rx_packet = packet
|
|
self.rx_packet.done = True
|
|
self.process(packet)
|
|
|
|
def process(self, packet):
|
|
pass
|