Fixed warning, freeing tpm context memory in extend_pcr function

diff --git a/src/libpts/pts/pts.c b/src/libpts/pts/pts.c
index 2f9e0fa..496dc88 100644 (file)
--- a/src/libpts/pts/pts.c
+++ b/src/libpts/pts/pts.c
@@ -739,7 +739,7 @@ METHOD(pts_t, read_pcr, bool,
        {
                goto err;
        }
        {
                goto err;
        }

-       result = Tspi_TPM_PcrRead(hTPM, pcr_num, &rgbPcrValue.len, &rgbPcrValue.ptr);
+       result = Tspi_TPM_PcrRead(hTPM, pcr_num, (UINT32*)&rgbPcrValue.len, &rgbPcrValue.ptr);

        if (result != TSS_SUCCESS)
        {
                goto err;
        if (result != TSS_SUCCESS)
        {
                goto err;


@@ -797,16 +797,22 @@ METHOD(pts_t, extend_pcr, bool,
        *output = pcr_value;
        *output = chunk_clone(*output);
 
        *output = pcr_value;
        *output = chunk_clone(*output);
 

-       chunk_clear(&pcr_value);
-       Tspi_Context_Close(hContext);

        DBG3(DBG_PTS, "PCR %d extended with:      %B", pcr_num, &input);
        DBG3(DBG_PTS, "PCR %d value after extend: %B", pcr_num, output);
        DBG3(DBG_PTS, "PCR %d extended with:      %B", pcr_num, &input);
        DBG3(DBG_PTS, "PCR %d value after extend: %B", pcr_num, output);

+       
+       chunk_clear(&pcr_value);
+       Tspi_Context_FreeMemory(hContext, NULL);
+       Tspi_Context_Close(hContext);
+

        return TRUE;
 
 err:
        return TRUE;
 
 err:

-       chunk_clear(&pcr_value);

        DBG1(DBG_PTS, "TPM not available: tss error 0x%x", result);
        DBG1(DBG_PTS, "TPM not available: tss error 0x%x", result);

+       
+       chunk_clear(&pcr_value);
+       Tspi_Context_FreeMemory(hContext, NULL);

        Tspi_Context_Close(hContext);
        Tspi_Context_Close(hContext);

+       

        return FALSE;
 }
 
        return FALSE;
 }