From f74e6747fbddcae866333d8dcb207cb8a2e4f280 Mon Sep 17 00:00:00 2001
From: Stef Walter <stef@memberwebs.com>
Date: Wed, 24 Dec 2008 21:31:52 +0000
Subject: A few last checks.

---
 src/key.c | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'src/key.c')

diff --git a/src/key.c b/src/key.c
index 6f182e6..2405b36 100644
--- a/src/key.c
+++ b/src/key.c
@@ -113,6 +113,26 @@ test_rsa_public(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE object)
 
 	P11T_SECTION("CKK_RSA Public");
 
+	/*
+	 * These attribute buffer codes have already been tested here and there, but
+	 * do it again since sometimes the big num code is different from general attrs.
+	 */
+	if(p11t_test_unexpected) {
+
+		/* Too small */
+		attr.type = CKA_MODULUS;
+		attr.pValue = modulus;
+		attr.ulValueLen = 4;
+		rv = (p11t_module_funcs->C_GetAttributeValue)(session, object, &attr, 1);
+		P11T_CHECK_RV("CKA_MODULUS", rv, CKR_BUFFER_TOO_SMALL);
+
+		/* No buffer */
+		attr.type = CKA_MODULUS;
+		attr.pValue = NULL;
+		rv = (p11t_module_funcs->C_GetAttributeValue)(session, object, &attr, 1);
+		P11T_CHECK_RV("CKA_MODULUS", rv, CKR_OK);
+	}
+
 	attr.type = CKA_MODULUS;
 	attr.pValue = modulus;
 	attr.ulValueLen = sizeof(modulus);
-- 
cgit v1.2.3