From c42450a294c4267998aa16a477e9218ee5953aa9 Mon Sep 17 00:00:00 2001 From: Jeffrey Ferreira Date: Thu, 19 Sep 2019 13:32:00 -0700 Subject: [PATCH] tcti-device: getPollHandles should allow num_handles query Signed-off-by: Jeffrey Ferreira --- src/tss2-tcti/tcti-device.c | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/tss2-tcti/tcti-device.c b/src/tss2-tcti/tcti-device.c index 44c9fe2083d5..53a698cad061 100644 --- a/src/tss2-tcti/tcti-device.c +++ b/src/tss2-tcti/tcti-device.c @@ -368,12 +368,19 @@ tcti_device_get_poll_handles ( return TSS2_TCTI_RC_BAD_CONTEXT; } - if (handles == NULL || num_handles == NULL) { + if (num_handles == NULL) { return TSS2_TCTI_RC_BAD_REFERENCE; } + if (handles != NULL && *num_handles < 1) { + return TSS2_TCTI_RC_INSUFFICIENT_BUFFER; + } + *num_handles = 1; - handles->fd = tcti_dev->fd; + if (handles != NULL) { + handles->fd = tcti_dev->fd; + } + return TSS2_RC_SUCCESS; #else (void)(tctiContext); -- 2.27.0