From 70e79825b38827ebf1fb14619ce55c7cc1488c23 Mon Sep 17 00:00:00 2001 From: rlaphoenix Date: Thu, 21 Jul 2022 16:23:19 +0100 Subject: [PATCH] Device: Re-use magic reference across Structures --- pywidevine/device.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/pywidevine/device.py b/pywidevine/device.py index d158bc5..45ce92e 100644 --- a/pywidevine/device.py +++ b/pywidevine/device.py @@ -23,13 +23,15 @@ class _Types(Enum): class _Structures: + magic = Const(b"WVD") + header = Struct( - "signature" / Const(b"WVD"), + "signature" / magic, "version" / Int8ub ) v2 = Struct( - "signature" / Const(b"WVD"), + "signature" / magic, "version" / Const(Int8ub, 2), "type_" / CEnum( Int8ub, @@ -47,7 +49,7 @@ class _Structures: ) v1 = Struct( - "signature" / Const(b"WVD"), + "signature" / magic, "version" / Const(Int8ub, 1), "type_" / CEnum( Int8ub,