mirror of
https://github.com/enjoy-digital/litex.git
synced 2025-01-04 09:52:26 -05:00
bfm: rewrite it and split Physical / Link layers
This commit is contained in:
Florent Kermarrec
parent
6e6243e983
commit
9da1f7fcbb
1 changed files with 98 additions and 81 deletions
|
|
@ -5,17 +5,17 @@ from migen.fhdl.std import *
|
|||
from lib.sata.std import *
|
||||
from lib.sata.link.test.common import *
|
||||
|
||||
class BFMDword():
|
||||
class PHYDword:
|
||||
def __init__(self, dat=0):
|
||||
self.dat = dat
|
||||
self.start = 1
|
||||
self.done = 0
|
||||
|
||||
class BFMSource(Module):
|
||||
def __init__(self, dw):
|
||||
self.source = Source(phy_layout(dw))
|
||||
class PHYSource(Module):
|
||||
def __init__(self):
|
||||
self.source = Source(phy_layout(32))
|
||||
###
|
||||
self.dword = BFMDword()
|
||||
self.dword = PHYDword()
|
||||
|
||||
def send(self, dword):
|
||||
self.dword = dword
|
||||
|
|
@ -28,11 +28,11 @@ class BFMSource(Module):
|
|||
selfp.source.charisk = 0b0001
|
||||
selfp.source.data = self.dword.dat
|
||||
|
||||
class BFMSink(Module):
|
||||
def __init__(self, dw):
|
||||
self.sink = Sink(phy_layout(dw))
|
||||
class PHYSink(Module):
|
||||
def __init__(self):
|
||||
self.sink = Sink(phy_layout(32))
|
||||
###
|
||||
self.dword = BFMDword()
|
||||
self.dword = PHYDword()
|
||||
|
||||
def receive(self):
|
||||
self.dword.done = 0
|
||||
|
|
@ -46,115 +46,125 @@ class BFMSink(Module):
|
|||
self.dword.done = 1
|
||||
self.dword.dat = selfp.sink.data
|
||||
|
||||
class BFMPHY(Module):
|
||||
def __init__(self, dw):
|
||||
self.dw = dw
|
||||
class PHYLayer(Module):
|
||||
def __init__(self, debug):
|
||||
self.debug = debug
|
||||
|
||||
self.submodules.bfm_sink = BFMSink(dw)
|
||||
self.submodules.bfm_source = BFMSource(dw)
|
||||
self.submodules.rx = PHYSink()
|
||||
self.submodules.tx = PHYSource()
|
||||
|
||||
self.source = self.bfm_source.source
|
||||
self.sink = self.bfm_sink.sink
|
||||
|
||||
self.dword = 0
|
||||
self.source = self.tx.source
|
||||
self.sink = self.rx.sink
|
||||
|
||||
def send(self, dword):
|
||||
packet = BFMDword(dword)
|
||||
self.bfm_source.send(packet)
|
||||
packet = PHYDword(dword)
|
||||
self.tx.send(packet)
|
||||
|
||||
def receive(self):
|
||||
yield from self.bfm_sink.receive()
|
||||
self.rx_dword = self.bfm_sink.dword.dat
|
||||
yield from self.rx.receive()
|
||||
if self.debug:
|
||||
print(self)
|
||||
|
||||
def __repr__(self):
|
||||
# receive
|
||||
receiving = "%08x " %self.rx_dword
|
||||
receiving += decode_primitive(self.rx_dword)
|
||||
receiving = "%08x " %self.rx.dword.dat
|
||||
receiving += decode_primitive(self.rx.dword.dat)
|
||||
receiving += " "*(16-len(receiving))
|
||||
|
||||
# send
|
||||
sending = "%08x " %self.bfm_source.dword.dat
|
||||
sending += decode_primitive(self.bfm_source.dword.dat)
|
||||
sending = "%08x " %self.tx.dword.dat
|
||||
sending += decode_primitive(self.tx.dword.dat)
|
||||
sending += " "*(16-len(sending))
|
||||
|
||||
return receiving + sending
|
||||
|
||||
class LinkPacket(list):
|
||||
def __init__(self):
|
||||
self.ongoing = False
|
||||
self.scrambled_datas = self.import_scrambler_datas()
|
||||
|
||||
class BFM(Module):
|
||||
def __init__(self, dw, debug=False, hold_random_level=0):
|
||||
self.debug = debug
|
||||
self.hold_random_level = hold_random_level
|
||||
|
||||
###
|
||||
|
||||
self.submodules.phy = BFMPHY(dw)
|
||||
self.get_scrambler_ref()
|
||||
|
||||
self.rx_cont_ongoing = False
|
||||
self.rx_packet_ongoing = False
|
||||
self.rx_packet = []
|
||||
|
||||
self.run = True
|
||||
|
||||
def get_scrambler_ref(self):
|
||||
def import_scrambler_datas(self):
|
||||
with subprocess.Popen(["./scrambler"], stdin=subprocess.PIPE, stdout=subprocess.PIPE) as process:
|
||||
process.stdin.write("0x10000".encode("ASCII"))
|
||||
out, err = process.communicate()
|
||||
self.scrambler_ref = [int(e, 16) for e in out.decode("utf-8").split("\n")[:-1]]
|
||||
return [int(e, 16) for e in out.decode("utf-8").split("\n")[:-1]]
|
||||
|
||||
def descramble(self, packet):
|
||||
p = []
|
||||
for i in range(len(packet)):
|
||||
v = packet[i] ^ self.scrambler_ref[i]
|
||||
p.append(v)
|
||||
return p
|
||||
class LinkRXPacket(LinkPacket):
|
||||
def decode(self):
|
||||
self.descramble()
|
||||
return self.check_crc()
|
||||
|
||||
def check_crc(self, packet):
|
||||
def descramble(self):
|
||||
for i in range(len(self)):
|
||||
self[i] = self[i] ^ self.scrambled_datas[i]
|
||||
|
||||
def check_crc(self):
|
||||
stdin = ""
|
||||
for v in packet[:-1]:
|
||||
for v in self[:-1]:
|
||||
stdin += "0x%08x " %v
|
||||
stdin += "exit"
|
||||
with subprocess.Popen("./crc", stdin=subprocess.PIPE, stdout=subprocess.PIPE) as process:
|
||||
process.stdin.write(stdin.encode("ASCII"))
|
||||
out, err = process.communicate()
|
||||
crc = int(out.decode("ASCII"), 16)
|
||||
if packet[-1] != crc:
|
||||
return []
|
||||
else:
|
||||
return packet[:-1]
|
||||
r = (self[-1] == crc)
|
||||
self.pop()
|
||||
return r
|
||||
|
||||
def packet_callback(self, packet):
|
||||
packet = self.descramble(packet)
|
||||
packet = self.check_crc(packet)
|
||||
class LinkTXPacket(LinkPacket):
|
||||
def encode(self):
|
||||
self.scramble()
|
||||
self.insert_crc()
|
||||
|
||||
def scramble(self):
|
||||
for i in range(len(self)):
|
||||
self[i] = self[i] ^ self.scrambled_datas[i]
|
||||
|
||||
def insert_crc(self):
|
||||
stdin = ""
|
||||
for v in self[:-1]:
|
||||
stdin += "0x%08x " %v
|
||||
stdin += "exit"
|
||||
with subprocess.Popen("./crc", stdin=subprocess.PIPE, stdout=subprocess.PIPE) as process:
|
||||
process.stdin.write(stdin.encode("ASCII"))
|
||||
out, err = process.communicate()
|
||||
crc = int(out.decode("ASCII"), 16)
|
||||
self.append(crc)
|
||||
|
||||
def transport_callback(packet):
|
||||
print("----")
|
||||
for v in packet:
|
||||
print("%08x" %v)
|
||||
print("----")
|
||||
|
||||
def dword_callback(self, dword):
|
||||
# CONT detection
|
||||
if dword == primitives["CONT"]:
|
||||
self.rx_cont_ongoing = True
|
||||
elif is_primitive(dword):
|
||||
self.rx_cont_ongoing = False
|
||||
class LinkLayer(Module):
|
||||
def __init__(self, phy, debug, hold_random_level=0):
|
||||
self.phy = phy
|
||||
self.debug = debug
|
||||
self.hold_random_level = hold_random_level
|
||||
self.tx_packet = LinkTXPacket()
|
||||
self.rx_packet = LinkRXPacket()
|
||||
self.rx_cont = False
|
||||
|
||||
def callback(self, dword):
|
||||
if dword == primitives["CONT"]:
|
||||
self.rx_cont = True
|
||||
elif is_primitive(dword):
|
||||
self.rx_cont = False
|
||||
|
||||
# X_RDY / WTRM response
|
||||
if dword == primitives["X_RDY"]:
|
||||
self.phy.send(primitives["R_RDY"])
|
||||
|
||||
elif dword == primitives["WTRM"]:
|
||||
self.phy.send(primitives["R_OK"])
|
||||
|
||||
# HOLD response
|
||||
elif dword == primitives["HOLD"]:
|
||||
self.phy.send(primitives["HOLDA"])
|
||||
|
||||
# Packet capture
|
||||
elif dword == primitives["EOF"]:
|
||||
self.rx_packet_ongoing = False
|
||||
self.packet_callback(self.rx_packet)
|
||||
self.rx_packet.decode()
|
||||
transport_callback(self.rx_packet)
|
||||
self.rx_packet.ongoing = False
|
||||
|
||||
elif self.rx_packet_ongoing:
|
||||
elif self.rx_packet.ongoing:
|
||||
if dword != primitives["HOLD"]:
|
||||
n = randn(100)
|
||||
if n < self.hold_random_level:
|
||||
|
|
@ -162,17 +172,24 @@ class BFM(Module):
|
|||
else:
|
||||
self.phy.send(primitives["R_RDY"])
|
||||
if not is_primitive(dword):
|
||||
if not self.rx_cont_ongoing:
|
||||
if not self.rx_cont:
|
||||
self.rx_packet.append(dword)
|
||||
|
||||
elif dword == primitives["SOF"]:
|
||||
self.rx_packet_ongoing = True
|
||||
self.rx_packet = []
|
||||
self.rx_packet = LinkRXPacket()
|
||||
self.rx_packet.ongoing = True
|
||||
|
||||
def send(self, packet):
|
||||
pass
|
||||
|
||||
def gen_simulation(self, selfp):
|
||||
self.phy.send(primitives["SYNC"])
|
||||
while True:
|
||||
yield from self.phy.receive()
|
||||
if self.debug:
|
||||
print(self.phy)
|
||||
self.dword_callback(self.phy.rx_dword)
|
||||
self.callback(self.phy.rx.dword.dat)
|
||||
|
||||
class BFM(Module):
|
||||
def __init__(self, dw, debug=False, hold_random_level=0):
|
||||
###
|
||||
self.submodules.phy = PHYLayer(debug)
|
||||
self.submodules.link = LinkLayer(self.phy, debug, hold_random_level)
|
||||
|
|
|
|||
Loading…
Reference in a new issue