code fix

Author: ph4r05

pytss/__init__.py

@@ -97,20 +97,6 @@ def get_policy_object(self, poltype):
         policy_obj = TspiPolicy(self.context, None, handle=policy)
         return policy_obj
 
-    def generate_random_data(self, length):
-        """
-        Generates random data in TPM of the given length in bytes.
-
-        :param length: Length of data to generate in bytes
-
-        :return: bytearray containing the random data
-        """
-        blob = ffi.new('BYTE **')
-        tss_lib.Tspi_TPM_GetRandom(self.get_handle(), length, blob)
-        ret = bytearray(blob[0][0:length])
-        tss_lib.Tspi_Context_FreeMemory(self.context, blob[0])
-        return ret
-
 
 class TspiNV(TspiObject):
     def __init__(self, context, flags):
@@ -488,6 +474,21 @@ def extend_pcr(self, pcr, data, event):
         tss_lib.Tspi_Context_FreeMemory(self.context, blob[0])
         return ret
 
+    def generate_random_data(self, length):
+        """
+        Generates random data in TPM of the given length in bytes.
+
+        :param length: Length of data to generate in bytes
+
+        :return: bytearray containing the random data
+        """
+        blob = ffi.new('BYTE **')
+        tss_lib.Tspi_TPM_GetRandom(self.get_handle(), length, blob)
+        ret = bytearray(blob[0][0:length])
+        tss_lib.Tspi_Context_FreeMemory(self.context, blob[0])
+        return ret
+
+
 class TspiContext():
     def __init__(self):
         self.context = ffi.new('TSS_HCONTEXT *')