tpm: free measureddata when SHIM_UNIT_TEST is set

Keep from cluttering up valgrind with allocations that aren't part of
the tested info (yet).

Signed-off-by: Peter Jones <pjones@redhat.com>

1 files changed, 22 insertions, 0 deletions

diff --git a/tpm.c b/tpm.c
index 808e0444..5af5f173 100644
--- a/tpm.c
+++ b/tpm.c
@@ -353,3 +353,25 @@ fallback_should_prefer_reset(void)
 		return EFI_NOT_FOUND;
 	return EFI_SUCCESS;
 }
+
+#ifdef SHIM_UNIT_TEST
+static void DESTRUCTOR
+tpm_clean_up_measurements(void)
+{
+	for (UINTN i = 0; i < measuredcount; i++) {
+		VARIABLE_RECORD *vr = &measureddata[i];
+
+		if (vr->VariableName)
+			FreePool(vr->VariableName);
+		if (vr->VendorGuid)
+			FreePool(vr->VendorGuid);
+		if (vr->Data)
+			FreePool(vr->Data);
+	}
+	if (measureddata)
+		FreePool(measureddata);
+
+	measuredcount = 0;
+	measureddata = NULL;
+}
+#endif