Proxmox-Port/libtpms
1
0
Fork 0
mirror of https://github.com/stefanberger/libtpms synced 2026-08-11 12:43:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests: fuzz: Have state suspended and resumed after every fuzz step

Browse Source 
Have the TPM 2's state suspended and resumed at every step to
ensure that we can marshal and unmarshal it.

Signed-off-by: Stefan Berger <stefanb@linux.ibm.com>
This commit is contained in:
Stefan Berger 2021-07-22 11:44:46 -04:00 committed by Stefan Berger
parent 70ebecdf6a
commit a6c9836af5
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tests/fuzz.cc
View File

@ -59,6 +59,10 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
uint32_t rlength;
uint32_t rtotal = 0;
TPM_RESULT res;
unsigned char *vol_buffer = NULL;
uint32_t vol_buffer_len;
unsigned char *perm_buffer = NULL;
uint32_t perm_buffer_len;
unsigned char startup[] = {
0x80, 0x01, 0x00, 0x00, 0x00, 0x0c, 0x00, 0x00, 0x01, 0x44, 0x00, 0x00
};
@ -92,8 +96,34 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
if (res != TPM_SUCCESS)
die("Error: TPMLIB_Process(fuzz-command) failed\n");
/* state suspend */
res = TPMLIB_GetState(TPMLIB_STATE_VOLATILE, &vol_buffer, &vol_buffer_len);
if (res != TPM_SUCCESS)
die("Error: TPMLIB_GetState(TPMLIB_STATE_VOLATILE) failed\n");
res = TPMLIB_GetState(TPMLIB_STATE_PERMANENT, &perm_buffer, &perm_buffer_len);
if (res != TPM_SUCCESS)
die("Error: TPMLIB_GetState(TPMLIB_STATE_PERMANENT) failed\n");
TPMLIB_Terminate();
/* state resume */
res = TPMLIB_SetState(TPMLIB_STATE_PERMANENT, perm_buffer, perm_buffer_len);
if (res != TPM_SUCCESS)
die("Error: TPMLIB_SetState(TPMLIB_STATE_PERMANENT) failed\n");
res = TPMLIB_SetState(TPMLIB_STATE_VOLATILE, vol_buffer, vol_buffer_len);
if (res != TPM_SUCCESS)
die("Error: TPMLIB_SetState(TPMLIB_STATE_VOLATILE) failed\n");
res = TPMLIB_MainInit();
if (res != TPM_SUCCESS)
die("Error: TPMLIB_MainInit() to resume with the state failed\n");
TPMLIB_Terminate();
TPM_Free(rbuffer);
TPM_Free(vol_buffer);
TPM_Free(perm_buffer);
return 0;
}