Changed the interface parameters
[people/mcb30/basetools.git] / Source / C / PyEfiCompressor / EfiCompressor.c
index 6760191..38a81c3 100644 (file)
@@ -23,10 +23,9 @@ UefiDecompress(
 \r
   Status = PyArg_ParseTuple(\r
             Args,\r
-            "Oii",\r
+            "Oi",\r
             &SrcData,\r
-            &SrcDataSize,\r
-            &DstDataSize\r
+            &SrcDataSize\r
             );\r
   if (Status == 0) {\r
     return NULL;\r
@@ -41,7 +40,7 @@ UefiDecompress(
 \r
   // Because some Python objects which support "buffer" protocol have more than one\r
   // memory segment, we have to copy them into a contiguous memory.\r
-  SrcBuf = malloc(SrcDataSize);\r
+  SrcBuf = PyMem_Malloc(SrcDataSize);\r
   if (SrcBuf == NULL) {\r
     PyErr_SetString(PyExc_Exception, "Not enough memory\n");\r
     goto ERROR;\r
@@ -101,10 +100,9 @@ FrameworkDecompress(
 \r
   Status = PyArg_ParseTuple(\r
             Args,\r
-            "Oii",\r
+            "Oi",\r
             &SrcData,\r
-            &SrcDataSize,\r
-            &DstDataSize\r
+            &SrcDataSize\r
             );\r
   if (Status == 0) {\r
     return NULL;\r
@@ -119,7 +117,7 @@ FrameworkDecompress(
 \r
   // Because some Python objects which support "buffer" protocol have more than one\r
   // memory segment, we have to copy them into a contiguous memory.\r
-  SrcBuf = malloc(SrcDataSize);\r
+  SrcBuf = PyMem_Malloc(SrcDataSize);\r
   if (SrcBuf == NULL) {\r
     PyErr_SetString(PyExc_Exception, "Not enough memory\n");\r
     goto ERROR;\r