1 files changed, 20 insertions, 0 deletions
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);