M2-PT-DRP/source/packets/KeyPacket.py

23 lines
450 B
Python

import dataclasses
import msgpack
from source.packets import base
@dataclasses.dataclass
class KeyPacket(base.BasePacket):
"""
Represent a packet containing a secret symmetric key
"""
secret_key: bytes = dataclasses.field(repr=False)
def pack(self) -> bytes:
return msgpack.packb((
self.secret_key
))
@classmethod
def unpack(cls, data: bytes):
return cls(*msgpack.unpackb(data))