policy: fix hardcoded size in description

Fixes: tpm2-software#546 Signed-off-by: William Roberts <[email protected]>
williamcroberts · Nov 6, 2023 · 7497038 · 7497038
1 parent 96a94c0
commit 7497038
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/tpm2_pytss/policy.py b/src/tpm2_pytss/policy.py
@@ -570,13 +570,16 @@ def get_calculated_json(self) -> bytes:
 
         cjson = ffi.new("uint8_t[]", size[0])
         _chkrc(lib.Tss2_PolicyGetCalculatedJSON(self._ctx, cjson, size))
+
         return ffi.string(cjson, size[0])
 
     @property
     def description(self) -> bytes:
         """bytes: The policy description."""
-        size = ffi.new("size_t *", 8096)
-        desc = ffi.new("uint8_t[]", 8096)
+        size = ffi.new("size_t *")
+        _chkrc(lib.Tss2_PolicyGetDescription(self._ctx, ffi.NULL, size))
+
+        desc = ffi.new("uint8_t[]", size[0])
         _chkrc(lib.Tss2_PolicyGetDescription(self._ctx, desc, size))
         return ffi.string(desc, size[0])