Service Frame Class

thingset/packet.py

@@ -104,3 +104,101 @@ def data(self, data):
     @property
     def cbor(self):
         return self._cbor
+
+class ServiceFrame(TSPacket):
+    def __init__(self, priority=7, destination=0, source=0):
+        super().__init__()
+        self._messageType = False
+        self.priority = priority
+        self.source = source
+        self.destination = destination
+
+    @property
+    def messageType(self):
+        return "Service message"
+
+    @property
+    def priority(self):
+        return self._priority
+
+    @priority.setter
+    def priority(self, priority):
+        if priority not in range(0,8):
+            raise ValueError("Priority must be integer between 0 and 7 (got: {})".format(priority))
+        self._priority = priority
+
+    @property
+    def destination(self):
+        return self._destination
+
+    @destination.setter
+    def destination(self, destination):
+        if destination not in range(0,256):
+            raise ValueError("Destination ID must be integer between 0 and 255 (got: {})".format(destination))
+        self._destination = destination
+
+class RequestFrame(ServiceFrame):
+    SINGLE_ID_MASK = (0b10 << 24)
+
+    FRAME_TYPE_SINGLE = 0x0
+    FRAME_TYPE_FIRST = 0x1
+    FRAME_TYPE_CONSEC = 0x2
+    FRAME_TYPE_FLOWC = 0x3
+
+    def __init__(self, priority=7, src=0, dst=0, data=bytes()):
+        super().__init__()
+        self.priority = priority
+        self.source = src
+        self.destination = dst
+        self.type = (data[0] & 0xf0) >> 4
+        self.data = data
+
+        if self.type == self.FRAME_TYPE_SINGLE:
+            self.framesize = data[0] & 0xf
+        if self.type == self.FRAME_TYPE_FIRST:
+            self.framesize = ((data[0] & 0xF) << 8) | data[1]
+        if self.type == self.FRAME_TYPE_CONSEC:
+            self.index = data[0] & 0xf
+        if self.type == self.FRAME_TYPE_FLOWC:
+            self.fcflag = data[0] & 0xf
+            self.blocksize = data[1]
+            self.delay = data[2]
+
+
+    def parseIdentifier(self, identifier):
+        if not isinstance(identifier, int):
+            raise ValueError("Identifier must be integer, not {}.".format(identifier))
+        if identifier >= (1 << 30):
+            raise ValueError("Identifier too big. Cannot contain more than 29 bits")
+        if (identifier & TSPacket.TS_FRAME_FLAG):
+            raise ValueError("Not a request/response message.")
+        self.priority = identifier >> 26
+        self.destination = (identifier & 0xffff) >> 8
+        self.source = identifier & 0xff
+
+    @property
+    def identifier(self):
+        id_prio = self._priority << 26
+        id_fixed = 0xDA << 16
+        id_dst = self._destination << 8
+        return id_prio | self.SINGLE_ID_MASK | id_fixed | id_dst | self.source
+
+    @property
+    def type(self):
+        return self._type
+
+    @type.setter
+    def type(self, type):
+        self._type = type
+
+    @property
+    def data(self):
+        return self._data
+
+    @data.setter
+    def data(self, data):
+        if data is None:
+            self._data = bytes()
+        elif not isinstance(data, bytes):
+            raise TypeError("Wrong data type. Must be bytes, not {}".format(type(data)))
+        self._data = data