1 files changed, 71 insertions, 0 deletions
diff --git a/pyecsca/sca/target/simpleserial.py b/pyecsca/sca/target/simpleserial.py
new file mode 100644
index 0000000..3d4b7e0
--- /dev/null
+++ b/pyecsca/sca/target/simpleserial.py
@@ -0,0 +1,71 @@
+from time import time_ns, sleep
+from typing import Mapping, Union
+
+from public import public
+
+from .serial import SerialTarget
+
+
+@public
+class SimpleSerialMessage(object):
+    char: str
+    data: str
+
+    def __init__(self, char: str, data: str):
+        self.char = char
+        self.data = data
+
+    @staticmethod
+    def from_raw(raw: Union[str, bytes]) -> "SimpleSerialMessage":
+        if isinstance(raw, bytes):
+            raw = raw.decode()
+        return SimpleSerialMessage(raw[0], raw[1:])
+
+    def __bytes__(self):
+        return str(self).encode()
+
+    def __str__(self):
+        return self.char + self.data
+
+    def __repr__(self):
+        return str(self)
+
+
+@public
+class SimpleSerialTarget(SerialTarget):
+
+    def recv_msgs(self, timeout: int) -> Mapping[str, SimpleSerialMessage]:
+        start = time_ns() // 1000000
+        buffer = bytes()
+        while not buffer.endswith(b"z00\n"):
+            wait = timeout - ((time_ns() // 1000000) - start)
+            if wait <= 0:
+                break
+            buffer += self.read(1 if not buffer else 0, wait)
+        if not buffer:
+            return {}
+        msgs = buffer.split(b"\n")
+        if buffer.endswith(b"\n"):
+            msgs.pop()
+
+        result = {}
+        for raw in msgs:
+            msg = SimpleSerialMessage.from_raw(raw)
+            result[msg.char] = msg
+        return result
+
+    def send_cmd(self, cmd: SimpleSerialMessage, timeout: int) -> Mapping[str, SimpleSerialMessage]:
+        """
+
+        :param cmd:
+        :param timeout:
+        :return:
+        """
+        data = bytes(cmd)
+        for i in range(0, len(data), 64):
+            chunk = data[i:i+64]
+            sleep(0.010)
+            self.write(chunk)
+        self.write(b"\n")
+        return self.recv_msgs(timeout)
+