From b1bcc74deae12a4a376004aba9549eefb44f78b4 Mon Sep 17 00:00:00 2001
From: Aveen Ismail <aveen.ismail@yubico.com>
Date: Wed, 6 Dec 2023 15:34:31 +0100
Subject: [PATCH] Code refactoring

---
 lib/util.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/util.c b/lib/util.c
index 66a3337c..8fe0bca3 100644
--- a/lib/util.c
+++ b/lib/util.c
@@ -1494,10 +1494,11 @@ uint32_t ykpiv_util_slot_object(uint8_t slot) {
 
    // calculate the required length of the encoded object
    buf_len = 1 /* cert tag */ + 3 /* compression tag + data*/ + 2 /* lrc */;
-   buf_len += get_length_size(rawdata_len);
-   buf_len += rawdata_len;
+   buf_len += len_bytes + rawdata_len;
 
    if (buf_len > *certdata_len) {
+     DBG("Buffer too small");
+     *certdata_len = 0;
      return YKPIV_SIZE_ERROR;
    }