diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml index bdafcab8..ef84a163 100644 --- a/.github/workflows/unit-test.yml +++ b/.github/workflows/unit-test.yml @@ -16,11 +16,21 @@ jobs: with: config: --enable-singlethreaded + aesecb: + uses: ./.github/workflows/build-workflow.yml + with: + config: --enable-aesecb + tpm: uses: ./.github/workflows/build-workflow.yml with: config: --enable-singlethreaded --enable-wolftpm --disable-dh C_EXTRA_FLAGS="-DWOLFPKCS11_TPM_STORE" check: ./tests/pkcs11str && ./tests/pkcs11test && ./tests/rsa_session_persistence_test + tpm_debug: + uses: ./.github/workflows/build-workflow.yml + with: + config: --enable-singlethreaded --enable-wolftpm --disable-dh --enable-debug C_EXTRA_FLAGS="-DWOLFPKCS11_TPM_STORE" + check: ./tests/token_store_test tpm_no_store: uses: ./.github/workflows/build-workflow.yml @@ -56,6 +66,11 @@ jobs: uses: ./.github/workflows/build-workflow.yml with: config: CFLAGS="-DWOLFPKCS11_NO_STORE" + no_env: + uses: ./.github/workflows/build-workflow.yml + with: + config: C_EXTRA_FLAGS="-DWOLFPKCS11_NO_ENV" --with-default-token-path=./store/default + check: mkdir -p store && ./tests/pkcs11test no_aesgcm: uses: ./.github/workflows/build-workflow.yml with: diff --git a/CMakeLists.txt b/CMakeLists.txt index 2d6a09a6..3989eaad 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -158,8 +158,7 @@ if(NOT WOLFPKCS11_SINGLE_THREADED) "-D_POSIX_THREADS") endif() else() - set(SINGLE_THREADED 1) - list(APPEND WOLFPKCS11_DEFINITIONS "-DSINGLE_THREADED") + list(APPEND WOLFPKCS11_DEFINITIONS "-DWOLFPKCS11_SINGLE_THREADED") endif() diff --git a/cmake/options.h.in b/cmake/options.h.in index 594bcdb3..14e5d04f 100644 --- a/cmake/options.h.in +++ b/cmake/options.h.in @@ -38,8 +38,8 @@ extern "C" { #endif #undef DEBUG_WOLFPKCS11 #cmakedefine DEBUG_WOLFPKCS11 -#undef SINGLE_THREADED -#cmakedefine SINGLE_THREADED +#undef WOLFPKCS11_SINGLE_THREADED +#cmakedefine WOLFPKCS11_SINGLE_THREADED #undef NO_RSA #cmakedefine NO_RSA #undef WC_NO_RSA_OAEP diff --git a/configure.ac b/configure.ac index 2ef69ffa..24ea49bc 100644 --- a/configure.ac +++ b/configure.ac @@ -127,7 +127,7 @@ AS_IF([ test "x$ENABLED_SINGLETHREADED" = "xno" ],[ ]) ]) -AS_IF([ test "x$ENABLED_SINGLETHREADED" = "xyes" ],[ AM_CFLAGS="-DSINGLE_THREADED $AM_CFLAGS" ]) +AS_IF([ test "x$ENABLED_SINGLETHREADED" = "xyes" ],[ AM_CFLAGS="-DWOLFPKCS11_SINGLE_THREADED $AM_CFLAGS" ]) HAVE_PTHREAD=0 AS_IF([test "x$ax_pthread_ok" = "xyes"],[HAVE_PTHREAD=1]) diff --git a/src/crypto.c b/src/crypto.c index 92de2f52..52fb919e 100644 --- a/src/crypto.c +++ b/src/crypto.c @@ -54,6 +54,12 @@ #define PRF_KEY_SIZE 48 +/* Derive mechanisms that build their output into a secret key object. */ +#if defined(HAVE_ECC) || !defined(NO_DH) || defined(WOLFPKCS11_HKDF) || \ + (!defined(NO_AES) && defined(HAVE_AES_CBC)) || defined(WOLFSSL_HAVE_PRF) + #define WP11_DERIVE_SECRET_KEY +#endif + /* Check that a CK_ULONG value fits in word32 with room for overhead such as * authentication tags, key wrap blocks, or padding. On LP64 platforms CK_ULONG * is 64-bit but wolfCrypt functions use word32/int for lengths. */ @@ -263,7 +269,7 @@ typedef struct AttributeType { /* List of recognized attributes and their data type. */ static AttributeType attrType[] = { { CKA_CLASS, ATTR_TYPE_ULONG }, - { CKA_TOKEN, ATTR_TYPE_DATA }, + { CKA_TOKEN, ATTR_TYPE_BOOL }, { CKA_PRIVATE, ATTR_TYPE_BOOL }, { CKA_LABEL, ATTR_TYPE_DATA }, { CKA_UNIQUE_ID, ATTR_TYPE_DATA }, @@ -532,14 +538,22 @@ static CK_RV CheckAttributes(CK_ATTRIBUTE* pTemplate, CK_ULONG ulCount, int set) if (attr->pValue == NULL && set) return CKR_ATTRIBUTE_VALUE_INVALID; if ((attr->pValue != NULL) && - (attr->ulValueLen != sizeof(CK_BBOOL))) - return CKR_BUFFER_TOO_SMALL; + (attr->ulValueLen != sizeof(CK_BBOOL))) { + /* Object creation reports a malformed CKA_TOKEN as invalid. */ + if (attr->type != CKA_TOKEN) + return CKR_BUFFER_TOO_SMALL; + if (set) + return CKR_ATTRIBUTE_VALUE_INVALID; + } if (set && *(CK_BBOOL*)attr->pValue != CK_TRUE && *(CK_BBOOL*)attr->pValue != CK_FALSE) { return CKR_ATTRIBUTE_VALUE_INVALID; } } else if (attrType[j].type == ATTR_TYPE_DATE) { + /* An empty date is valid and leaves the date unspecified. */ + if (set && attr->ulValueLen == 0) + continue; if (attr->pValue == NULL && set) return CKR_ATTRIBUTE_VALUE_INVALID; if ((attr->pValue != NULL) && @@ -549,6 +563,11 @@ static CK_RV CheckAttributes(CK_ATTRIBUTE* pTemplate, CK_ULONG ulCount, int set) else if (attrType[j].type == ATTR_TYPE_DATA) { if (set && attr->ulValueLen == CK_UNAVAILABLE_INFORMATION) return CKR_ATTRIBUTE_VALUE_INVALID; + if (set && attr->pValue == NULL && attr->ulValueLen != 0) + return CKR_ATTRIBUTE_VALUE_INVALID; + /* Object setters take int lengths. */ + if (set && attr->ulValueLen > (CK_ULONG)INT_MAX) + return CKR_ATTRIBUTE_VALUE_INVALID; } } @@ -599,6 +618,37 @@ static CK_RV SetInitialStates(WP11_Object* key) return rv; } +#ifdef WP11_DERIVE_SECRET_KEY +/* PKCS#11 derive rules: the historical flags also need the base key's. */ +static CK_RV SetDerivedStates(WP11_Object* key, CK_BBOOL baseAlwaysSensitive, + CK_BBOOL baseNeverExtractable) +{ + CK_RV rv; + CK_BBOOL getVar = CK_FALSE; + CK_BBOOL state; + CK_ULONG getVarLen = sizeof(CK_BBOOL); + + rv = WP11_Object_GetAttr(key, CKA_SENSITIVE, &getVar, &getVarLen); + if (rv == CKR_OK) { + state = (baseAlwaysSensitive == CK_TRUE && getVar == CK_TRUE) ? + CK_TRUE : CK_FALSE; + rv = WP11_Object_SetAttr(key, CKA_ALWAYS_SENSITIVE, &state, + sizeof(CK_BBOOL)); + } + if (rv == CKR_OK) { + getVarLen = sizeof(CK_BBOOL); + rv = WP11_Object_GetAttr(key, CKA_EXTRACTABLE, &getVar, &getVarLen); + } + if (rv == CKR_OK) { + state = (baseNeverExtractable == CK_TRUE && getVar == CK_FALSE) ? + CK_TRUE : CK_FALSE; + rv = WP11_Object_SetAttr(key, CKA_NEVER_EXTRACTABLE, &state, + sizeof(CK_BBOOL)); + } + return rv; +} +#endif + static CK_RV TemplateHasAttribute(CK_ATTRIBUTE_TYPE type, CK_ATTRIBUTE_PTR pTemplate, CK_ULONG ulCount) { @@ -860,6 +910,123 @@ static CK_RV SetAttributeDefaults(WP11_Object* obj, CK_OBJECT_CLASS keyType, return ret; } +/* Policy checks for one template attribute against the object's current + * state. Run over the whole template before any change, and again just before + * each write so a concurrent update is not undone using a stale result. */ +static CK_RV CheckAttributeUpdate(WP11_Session* session, WP11_Object* obj, + CK_ATTRIBUTE* attr, CK_OBJECT_CLASS objClass, + CK_BBOOL newObject) +{ + CK_RV rv; + CK_BBOOL getVar; + CK_ULONG getVarLen; + byte roCur[sizeof(CK_ULONG)]; + CK_ULONG roCurLen; + int ret; + + /* Cannot change sensitive from true to false */ + if (attr->type == CKA_SENSITIVE) { + getVarLen = sizeof(getVar); + rv = WP11_Object_GetAttr(obj, CKA_SENSITIVE, &getVar, &getVarLen); + if (rv != CKR_OK) + return rv; + + if ((getVar == CK_TRUE) && (*(CK_BBOOL*)attr->pValue == CK_FALSE)) + return CKR_ATTRIBUTE_READ_ONLY; + } + /* Cannot change extractable from false to true */ + if (!newObject && attr->type == CKA_EXTRACTABLE) { + getVarLen = sizeof(getVar); + rv = WP11_Object_GetAttr(obj, CKA_EXTRACTABLE, &getVar, + &getVarLen); + if (rv != CKR_OK) + return rv; + + if ((getVar == CK_FALSE) && (*(CK_BBOOL*)attr->pValue == CK_TRUE)) + return CKR_ATTRIBUTE_READ_ONLY; + } + /* PKCS#11 v2.40 sec 4.4.1: once CKA_COPYABLE/CKA_DESTROYABLE has been + * set to CK_FALSE it cannot be set back to CK_TRUE. Read the stored + * flag bit directly so the check is independent of the GetAttr view + * (which the legacy macro may override). */ + if (!newObject && attr->type == CKA_COPYABLE) { + if (attr->pValue == NULL || + attr->ulValueLen != sizeof(CK_BBOOL)) + return CKR_ATTRIBUTE_VALUE_INVALID; + if (!WP11_Object_IsCopyable(obj) && + *(CK_BBOOL*)attr->pValue == CK_TRUE) + return CKR_ATTRIBUTE_READ_ONLY; + } + if (!newObject && attr->type == CKA_DESTROYABLE) { + if (attr->pValue == NULL || + attr->ulValueLen != sizeof(CK_BBOOL)) + return CKR_ATTRIBUTE_VALUE_INVALID; + if (!WP11_Object_IsDestroyable(obj) && + *(CK_BBOOL*)attr->pValue == CK_TRUE) + return CKR_ATTRIBUTE_READ_ONLY; + } + /* PKCS#11 v2.40 sec 4.5: only an SO session may set CKA_TRUSTED to + * CK_TRUE. A regular-user session must not forge trust and bypass the + * CKA_WRAP_WITH_TRUSTED export gate enforced by C_WrapKey. Not + * qualified with !newObject so it also stops C_CreateObject / + * C_GenerateKey from minting a trusted key. CheckAttributes above has + * already validated CKA_TRUSTED as a well-formed CK_BBOOL. */ + if (attr->type == CKA_TRUSTED && + *(CK_BBOOL*)attr->pValue == CK_TRUE && + WP11_Session_GetState(session) != WP11_APP_STATE_RW_SO) { + return CKR_ATTRIBUTE_READ_ONLY; + } + /* Context-specific login is not supported, so per-use authentication + * could not be enforced. */ + if (attr->type == CKA_ALWAYS_AUTHENTICATE && + *(CK_BBOOL*)attr->pValue == CK_TRUE) { + return CKR_ATTRIBUTE_VALUE_INVALID; + } + /* Certificate attributes; NSS trust objects carry them too. */ + if ((attr->type == CKA_ISSUER || attr->type == CKA_SERIAL_NUMBER) && + objClass != CKO_CERTIFICATE +#ifdef WOLFPKCS11_NSS + && objClass != CKO_NSS_TRUST +#endif + ) { + return CKR_ATTRIBUTE_TYPE_INVALID; + } + /* The type-specific storage was allocated for the creation class. */ + if (newObject && attr->type == CKA_CLASS && + *(CK_OBJECT_CLASS*)attr->pValue != objClass) { + return CKR_TEMPLATE_INCONSISTENT; + } + /* Derived by the token; never accepted in a creation template. */ + if (newObject && (attr->type == CKA_ALWAYS_SENSITIVE || + attr->type == CKA_NEVER_EXTRACTABLE)) { + return CKR_ATTRIBUTE_READ_ONLY; + } + /* These class/identity and generated-state attributes are read-only + * once the object exists; reject a change. Setting the current value + * is a no-op. */ + if (!newObject && (attr->type == CKA_CLASS || + attr->type == CKA_KEY_TYPE || + attr->type == CKA_CERTIFICATE_TYPE || + attr->type == CKA_LOCAL || + attr->type == CKA_KEY_GEN_MECHANISM || + attr->type == CKA_ALWAYS_SENSITIVE || + attr->type == CKA_NEVER_EXTRACTABLE)) { + roCurLen = sizeof(roCur); + ret = WP11_Object_GetAttr(obj, attr->type, roCur, &roCurLen); + /* Not an attribute of this object, e.g. CKA_KEY_TYPE on a + * certificate. */ + if (ret == CKR_ATTRIBUTE_TYPE_INVALID) + return CKR_ATTRIBUTE_TYPE_INVALID; + if (ret == 0 && + (attr->pValue == NULL || attr->ulValueLen != roCurLen || + XMEMCMP(attr->pValue, roCur, roCurLen) != 0)) { + return CKR_ATTRIBUTE_READ_ONLY; + } + } + + return CKR_OK; +} + /** * Set the values of the attributes into the object. * @@ -867,7 +1034,7 @@ static CK_RV SetAttributeDefaults(WP11_Object* obj, CK_OBJECT_CLASS keyType, * @param obj [in] Object to set value against. * @param pTemplate [in] Template of attributes set against object. * @param ulCount [in] Number of attribute triplets in template. - * @return CKR_ARGUMENTS_BAD when pTemplate is NULL. + * @return CKR_ARGUMENTS_BAD when pTemplate is NULL or ulCount is too large. * CKR_SESSION_READ_ONLY when the session cannot modify objects. * CKR_ATTRIBUTE_TYPE_INVALID if the attribute type is not supported. * CKR_ATTRIBUTE_VALUE_INVALID if value is not valid for data type. @@ -894,12 +1061,8 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, CK_BBOOL attrsFound = 0; CK_KEY_TYPE type; CK_OBJECT_CLASS objClass; - CK_BBOOL getVar; - CK_ULONG getVarLen = 1; - byte roCur[sizeof(CK_ULONG)]; - CK_ULONG roCurLen; - if (pTemplate == NULL) + if (pTemplate == NULL || ulCount > (CK_ULONG)INT_MAX) return CKR_ARGUMENTS_BAD; /* Only require R/W session for token objects */ if (!WP11_Session_IsRW(session) && WP11_Object_OnToken(obj)) @@ -1002,6 +1165,32 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, } } + /* Key components and a certificate's value are fixed once the object + * exists (PKCS#11 v2.40 sec 4.6.3, 4.8). */ + if (!newObject && objClass != CKO_DATA +#ifdef WOLFPKCS11_NSS + && objClass != CKO_NSS_TRUST +#endif + ) { + for (i = 0; i < cnt; i++) { + if (objClass == CKO_CERTIFICATE && attrs[i] != CKA_VALUE) + continue; + for (j = 0; j < (int)ulCount; j++) { + if (attrs[i] == pTemplate[j].type) + return CKR_ATTRIBUTE_READ_ONLY; + } + } + } + + /* Validate the whole template before any setter runs so a rejected + * update leaves the object unchanged. */ + for (i = 0; i < (int)ulCount; i++) { + rv = CheckAttributeUpdate(session, obj, &pTemplate[i], objClass, + newObject); + if (rv != CKR_OK) + return rv; + } + for (i = 0; i < cnt; i++) { for (j = 0; j < (int)ulCount; j++) { if (attrs[i] == pTemplate[j].type) { @@ -1013,7 +1202,7 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, if (objClass != CKO_DATA) return CKR_ATTRIBUTE_VALUE_INVALID; } - len[i] = (int)pTemplate[j].ulValueLen; + len[i] = pTemplate[j].ulValueLen; break; } } @@ -1039,11 +1228,19 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, #ifndef NO_RSA case CKK_RSA: ret = WP11_Object_SetRsaKey(obj, data, len); + /* A modulus size or component the key cannot take. */ + if (ret == BAD_FUNC_ARG) + return CKR_ATTRIBUTE_VALUE_INVALID; break; #endif #ifdef HAVE_ECC case CKK_EC: ret = WP11_Object_SetEcKey(obj, data, len); + /* Key material without CKA_EC_PARAMS, or a bad value. */ + if (ret == BAD_FUNC_ARG && data[0] == NULL) + return CKR_TEMPLATE_INCOMPLETE; + if (ret == BAD_FUNC_ARG) + return CKR_ATTRIBUTE_VALUE_INVALID; break; #endif #ifdef WOLFPKCS11_MLDSA @@ -1054,6 +1251,9 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, #ifndef NO_DH case CKK_DH: ret = WP11_Object_SetDhKey(obj, data, len); + /* Parameters or a value larger than the key can hold. */ + if (ret == BAD_FUNC_ARG) + return CKR_ATTRIBUTE_VALUE_INVALID; break; #endif #ifdef WOLFPKCS11_MLKEM @@ -1094,6 +1294,9 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, #endif case CKK_GENERIC_SECRET: ret = WP11_Object_SetSecretKey(obj, data, len); + /* A length the key type cannot take. */ + if (ret == BAD_FUNC_ARG) + return CKR_ATTRIBUTE_VALUE_INVALID; break; default: break; @@ -1108,79 +1311,9 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, /* Set remaining attributes - key specific attributes ignored. */ for (i = 0; i < (int)ulCount; i++) { attr = &pTemplate[i]; - - /* Cannot change sensitive from true to false */ - if (attr->type == CKA_SENSITIVE) { - rv = WP11_Object_GetAttr(obj, CKA_SENSITIVE, &getVar, &getVarLen); - if (rv != CKR_OK) - return rv; - - if ((getVar == CK_TRUE) && (*(CK_BBOOL*)attr->pValue == CK_FALSE)) - return CKR_ATTRIBUTE_READ_ONLY; - } - /* Cannot change extractable from false to true */ - if (!newObject && attr->type == CKA_EXTRACTABLE) { - getVarLen = sizeof(getVar); - rv = WP11_Object_GetAttr(obj, CKA_EXTRACTABLE, &getVar, - &getVarLen); - if (rv != CKR_OK) - return rv; - - if ((getVar == CK_FALSE) && (*(CK_BBOOL*)attr->pValue == CK_TRUE)) - return CKR_ATTRIBUTE_READ_ONLY; - } - /* PKCS#11 v2.40 sec 4.4.1: once CKA_COPYABLE/CKA_DESTROYABLE has been - * set to CK_FALSE it cannot be set back to CK_TRUE. Read the stored - * flag bit directly so the check is independent of the GetAttr view - * (which the legacy macro may override). */ - if (!newObject && attr->type == CKA_COPYABLE) { - if (attr->pValue == NULL || - attr->ulValueLen != sizeof(CK_BBOOL)) - return CKR_ATTRIBUTE_VALUE_INVALID; - if (!WP11_Object_IsCopyable(obj) && - *(CK_BBOOL*)attr->pValue == CK_TRUE) - return CKR_ATTRIBUTE_READ_ONLY; - } - if (!newObject && attr->type == CKA_DESTROYABLE) { - if (attr->pValue == NULL || - attr->ulValueLen != sizeof(CK_BBOOL)) - return CKR_ATTRIBUTE_VALUE_INVALID; - if (!WP11_Object_IsDestroyable(obj) && - *(CK_BBOOL*)attr->pValue == CK_TRUE) - return CKR_ATTRIBUTE_READ_ONLY; - } - /* PKCS#11 v2.40 sec 4.5: only an SO session may set CKA_TRUSTED to - * CK_TRUE. A regular-user session must not forge trust and bypass the - * CKA_WRAP_WITH_TRUSTED export gate enforced by C_WrapKey. Not - * qualified with !newObject so it also stops C_CreateObject / - * C_GenerateKey from minting a trusted key. CheckAttributes above has - * already validated CKA_TRUSTED as a well-formed CK_BBOOL. */ - if (attr->type == CKA_TRUSTED && - *(CK_BBOOL*)attr->pValue == CK_TRUE && - WP11_Session_GetState(session) != WP11_APP_STATE_RW_SO) { - return CKR_ATTRIBUTE_READ_ONLY; - } - /* These class/identity and generated-state attributes are read-only - * once the object exists; reject a change. Setting the current value - * is a no-op. */ - if (!newObject && (attr->type == CKA_CLASS || - attr->type == CKA_KEY_TYPE || - attr->type == CKA_LOCAL || - attr->type == CKA_KEY_GEN_MECHANISM || - attr->type == CKA_ALWAYS_SENSITIVE || - attr->type == CKA_NEVER_EXTRACTABLE)) { - roCurLen = sizeof(roCur); - ret = WP11_Object_GetAttr(obj, attr->type, roCur, &roCurLen); - /* Not an attribute of this object, e.g. CKA_KEY_TYPE on a - * certificate. */ - if (ret == CKR_ATTRIBUTE_TYPE_INVALID) - return CKR_ATTRIBUTE_TYPE_INVALID; - if (ret == 0 && - (attr->pValue == NULL || attr->ulValueLen != roCurLen || - XMEMCMP(attr->pValue, roCur, roCurLen) != 0)) { - return CKR_ATTRIBUTE_READ_ONLY; - } - } + rv = CheckAttributeUpdate(session, obj, attr, objClass, newObject); + if (rv != CKR_OK) + return rv; ret = WP11_Object_SetAttr(obj, attr->type, (byte*)attr->pValue, attr->ulValueLen); if (ret == MEMORY_E) @@ -1189,6 +1322,12 @@ static CK_RV SetAttributeValue(WP11_Session* session, WP11_Object* obj, return CKR_ATTRIBUTE_VALUE_INVALID; else if (ret == BUFFER_E) return CKR_BUFFER_TOO_SMALL; + else if (ret == BAD_STATE_E) { + /* Logged in yet undecoded: the stored key failed to verify. */ + if (WP11_Slot_IsUserLoggedIn(WP11_Session_GetSlot(session))) + return CKR_FUNCTION_FAILED; + return CKR_USER_NOT_LOGGED_IN; + } else if (ret != 0) return CKR_FUNCTION_FAILED; } @@ -1368,7 +1507,7 @@ static CK_RV AddRSAPrivateKeyObject(WP11_Session* session, CK_OBJECT_HANDLE hPub; CK_ATTRIBUTE pubt[] = { - {CKA_TOKEN, NULL, sizeof(CK_BBOOL)}, + {CKA_TOKEN, &falseVal, sizeof(CK_BBOOL)}, {CKA_LABEL, NULL, 0}, {CKA_WRAP, &falseVal, sizeof(falseVal)}, {CKA_VERIFY, &trueVal, sizeof(trueVal)}, @@ -1479,9 +1618,7 @@ static CK_RV CreateObject(WP11_Session* session, CK_ATTRIBUTE_PTR pTemplate, } } else if (objectClass == CKO_DATA) { - FindAttributeType(pTemplate, ulCount, CKA_VALUE, &attr); - if (attr == NULL) - return CKR_TEMPLATE_INCOMPLETE; + /* Every data object attribute is optional. */ } #ifdef WOLFPKCS11_NSS else if (objectClass == CKO_NSS_TRUST) { @@ -1556,6 +1693,143 @@ static CK_RV CreateObject(WP11_Session* session, CK_ATTRIBUTE_PTR pTemplate, return rv; } +/* Find an attribute that must be supplied with a non-empty value. */ +static CK_RV RequireAttribute(CK_ATTRIBUTE_PTR pTemplate, CK_ULONG ulCount, + CK_ATTRIBUTE_TYPE type) +{ + CK_ATTRIBUTE* attr; + + FindAttributeType(pTemplate, ulCount, type, &attr); + if (attr == NULL) + return CKR_TEMPLATE_INCOMPLETE; + if (attr->pValue == NULL || attr->ulValueLen == 0) + return CKR_ATTRIBUTE_VALUE_INVALID; + return CKR_OK; +} + +/* Attributes required at C_CreateObject, and a secret key class needs a secret + * key type. Unwrap, derive and decapsulate also use CreateObject but supply the + * key value afterwards. A malformed or unsupported class or key type is left + * for CreateObject to report. */ +static CK_RV CheckCreateRequiredAttributes(CK_ATTRIBUTE_PTR pTemplate, + CK_ULONG ulCount) +{ + CK_RV rv = CKR_OK; + CK_ATTRIBUTE* attr; + CK_OBJECT_CLASS objectClass; + CK_KEY_TYPE keyType; + int secretType; + + FindAttributeType(pTemplate, ulCount, CKA_CLASS, &attr); + if (attr == NULL || attr->pValue == NULL || + attr->ulValueLen != sizeof(CK_OBJECT_CLASS)) { + return CKR_OK; + } + objectClass = *(CK_OBJECT_CLASS*)attr->pValue; + + if (objectClass == CKO_CERTIFICATE) { + /* CKA_URL is not stored, so a certificate needs its value. */ + rv = RequireAttribute(pTemplate, ulCount, CKA_VALUE); + return rv; + } + + FindAttributeType(pTemplate, ulCount, CKA_KEY_TYPE, &attr); + if (attr == NULL || attr->pValue == NULL || + attr->ulValueLen != sizeof(CK_KEY_TYPE)) { + return CKR_OK; + } + keyType = *(CK_KEY_TYPE*)attr->pValue; + + switch (keyType) { + case CKK_GENERIC_SECRET: + #ifndef NO_AES + case CKK_AES: + #endif + #ifdef WOLFPKCS11_HKDF + case CKK_HKDF: + #endif + secretType = 1; + break; + #ifndef NO_RSA + case CKK_RSA: + #endif + #ifdef HAVE_ECC + case CKK_EC: + #endif + #ifndef NO_DH + case CKK_DH: + #endif + #ifdef WOLFPKCS11_MLDSA + case CKK_ML_DSA: + #endif + #ifdef WOLFPKCS11_MLKEM + case CKK_ML_KEM: + #endif + #ifdef WOLFPKCS11_LMS + case CKK_HSS: + #endif + #ifdef WOLFPKCS11_XMSS + case CKK_XMSS: + case CKK_XMSSMT: + #endif + secretType = 0; + break; + default: + return CKR_OK; + } + + if (objectClass == CKO_SECRET_KEY) { + if (!secretType) + return CKR_TEMPLATE_INCONSISTENT; + rv = RequireAttribute(pTemplate, ulCount, CKA_VALUE); + } + else if (objectClass == CKO_PUBLIC_KEY || objectClass == CKO_PRIVATE_KEY) { + switch (keyType) { + #ifndef NO_RSA + case CKK_RSA: + FindAttributeType(pTemplate, ulCount, CKA_MODULUS, &attr); + if (objectClass == CKO_PUBLIC_KEY) { + rv = RequireAttribute(pTemplate, ulCount, CKA_MODULUS); + if (rv == CKR_OK) { + rv = RequireAttribute(pTemplate, ulCount, + CKA_PUBLIC_EXPONENT); + } + } + else if (attr == NULL) { + /* The modulus is computed from the primes. */ + rv = RequireAttribute(pTemplate, ulCount, CKA_PRIME_1); + if (rv == CKR_OK) { + rv = RequireAttribute(pTemplate, ulCount, + CKA_PRIME_2); + } + } + else { + rv = RequireAttribute(pTemplate, ulCount, CKA_MODULUS); + } + if (rv == CKR_OK && objectClass == CKO_PRIVATE_KEY) { + rv = RequireAttribute(pTemplate, ulCount, + CKA_PRIVATE_EXPONENT); + } + break; + #endif + #ifdef HAVE_ECC + case CKK_EC: + rv = RequireAttribute(pTemplate, ulCount, CKA_EC_PARAMS); + if (rv == CKR_OK) { + rv = RequireAttribute(pTemplate, ulCount, + (objectClass == CKO_PUBLIC_KEY) ? + CKA_EC_POINT : CKA_VALUE); + } + break; + #endif + default: + break; + } + } + + return rv; +} + /** * Create an object in the session or on the token associated with the session. * @@ -1565,9 +1839,11 @@ static CK_RV CreateObject(WP11_Session* session, CK_ATTRIBUTE_PTR pTemplate, * @param phObject [out] Handle of object created. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pTemplate or phObject is NULL. + * CKR_ARGUMENTS_BAD when pTemplate or phObject is NULL, or ulCount + * is too large. * CKR_SESSION_READ_ONLY when the session cannot create objects. - * CKR_TEMPLATE_INCOMPLETE when CKA_KEY_TYPE is missing. + * CKR_TEMPLATE_INCOMPLETE when CKA_KEY_TYPE or an attribute required + * for the object is missing. * CKR_ATTRIBUTE_VALUE_INVALID when an attribute has invalid value or * length. * CKR_DEVICE_MEMORY when dynamic memory allocation fails. @@ -1600,7 +1876,7 @@ static CK_RV wp11_C_CreateObject(CK_SESSION_HANDLE hSession, WOLFPKCS11_LEAVE("C_CreateObject", rv); return rv; } - if (pTemplate == NULL || phObject == NULL) { + if (pTemplate == NULL || phObject == NULL || ulCount > (CK_ULONG)INT_MAX) { rv = CKR_ARGUMENTS_BAD; WOLFPKCS11_LEAVE("C_CreateObject", rv); return rv; @@ -1640,6 +1916,12 @@ static CK_RV wp11_C_CreateObject(CK_SESSION_HANDLE hSession, return rv; } + rv = CheckCreateRequiredAttributes(pTemplate, ulCount); + if (rv != CKR_OK) { + WOLFPKCS11_LEAVE("C_CreateObject", rv); + return rv; + } + rv = CreateObject(session, pTemplate, ulCount, &object); if (rv != CKR_OK) { WOLFPKCS11_LEAVE("C_CreateObject", rv); @@ -2045,7 +2327,7 @@ CK_RV C_GetObjectSize(CK_SESSION_HANDLE hSession, * @param ulCount [in] Number of attribute triplets in template. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pTemplate is NULL. + * CKR_ARGUMENTS_BAD when pTemplate is NULL or ulCount is too large. * CKR_OBJECT_HANDLE_INVALID when handle is not to a valid object. * CKR_ATTRIBUTE_TYPE_INVALID if the attribute type is not supported * or not valid for the object. @@ -2085,7 +2367,7 @@ static CK_RV wp11_C_GetAttributeValue(CK_SESSION_HANDLE hSession, WOLFPKCS11_LEAVE("C_GetAttributeValue", rv); return rv; } - if (pTemplate == NULL) { + if (pTemplate == NULL || ulCount > (CK_ULONG)INT_MAX) { rv = CKR_ARGUMENTS_BAD; WOLFPKCS11_LEAVE("C_GetAttributeValue", rv); return rv; @@ -2152,6 +2434,40 @@ CK_RV C_GetAttributeValue(CK_SESSION_HANDLE hSession, return rv; } +/* PKCS#11 v2.40 sec 4.4: CKA_TOKEN, CKA_PRIVATE and CKA_MODIFIABLE can only be + * changed by C_CopyObject. Restating the current value is allowed. Malformed + * values are left for CheckAttributes to report. */ +static CK_RV CheckCopyOnlyAttributes(WP11_Object* obj, + CK_ATTRIBUTE_PTR pTemplate, + CK_ULONG ulCount) +{ + CK_RV rv = CKR_OK; + CK_ATTRIBUTE* attr; + CK_ULONG i; + CK_BBOOL cur; + CK_ULONG curLen; + + for (i = 0; rv == CKR_OK && i < ulCount; i++) { + attr = &pTemplate[i]; + if (attr->type != CKA_TOKEN && attr->type != CKA_PRIVATE && + attr->type != CKA_MODIFIABLE) { + continue; + } + if (attr->pValue == NULL || attr->ulValueLen != sizeof(CK_BBOOL) || + (*(CK_BBOOL*)attr->pValue != CK_TRUE && + *(CK_BBOOL*)attr->pValue != CK_FALSE)) { + continue; + } + curLen = sizeof(cur); + if (WP11_Object_GetAttr(obj, attr->type, &cur, &curLen) != 0 || + cur != *(CK_BBOOL*)attr->pValue) { + rv = CKR_ATTRIBUTE_READ_ONLY; + } + } + + return rv; +} + /** * Set the values of the attributes into the object. * @@ -2161,7 +2477,7 @@ CK_RV C_GetAttributeValue(CK_SESSION_HANDLE hSession, * @param ulCount [in] Number of attribute triplets in template. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pTemplate is NULL. + * CKR_ARGUMENTS_BAD when pTemplate is NULL or ulCount is too large. * CKR_SESSION_READ_ONLY when the session cannot modify objects. * CKR_OBJECT_HANDLE_INVALID when handle is not to a valid object. * CKR_ATTRIBUTE_TYPE_INVALID if the attribute type is not supported. @@ -2202,7 +2518,7 @@ static CK_RV wp11_C_SetAttributeValue(CK_SESSION_HANDLE hSession, WOLFPKCS11_LEAVE("C_SetAttributeValue", rv); return rv; } - if (pTemplate == NULL) { + if (pTemplate == NULL || ulCount > (CK_ULONG)INT_MAX) { rv = CKR_ARGUMENTS_BAD; WOLFPKCS11_LEAVE("C_SetAttributeValue", rv); return rv; @@ -2232,7 +2548,9 @@ static CK_RV wp11_C_SetAttributeValue(CK_SESSION_HANDLE hSession, return rv; } - rv = SetAttributeValue(session, obj, pTemplate, ulCount, CK_FALSE); + rv = CheckCopyOnlyAttributes(obj, pTemplate, ulCount); + if (rv == CKR_OK) + rv = SetAttributeValue(session, obj, pTemplate, ulCount, CK_FALSE); WOLFPKCS11_LEAVE("C_SetAttributeValue", rv); return rv; } @@ -2355,7 +2673,7 @@ CK_RV C_FindObjects(CK_SESSION_HANDLE hSession, CK_ULONG_PTR pulObjectCount) { CK_RV rv; - int i; + CK_ULONG i; CK_OBJECT_HANDLE handle; WP11_Session* session; @@ -2388,7 +2706,7 @@ CK_RV C_FindObjects(CK_SESSION_HANDLE hSession, return rv; } - for (i = 0; i < (int)ulMaxObjectCount; i++) { + for (i = 0; i < ulMaxObjectCount; i++) { if (WP11_Session_FindGet(session, &handle) == FIND_NO_MORE_E) break; phObject[i] = handle; @@ -2446,6 +2764,35 @@ CK_RV C_FindObjectsFinal(CK_SESSION_HANDLE hSession) } +#ifdef WOLFPKCS11_PKCS11_V3_0 +/** + * Terminate the active operation of a kind, as requested by calling an Init + * function with a NULL mechanism. + * + * @param session [in] Session object. + * @param opCategory [in] Operation category (WP11_OP_*). + * @param recover [in] Whether the kind is verification with recovery. + * @return CKR_OK as releasing an operation cannot fail. + */ +static CK_RV CancelOperation(WP11_Session* session, int opCategory, + int recover) +{ + int recoverActive; + + recoverActive = WP11_Session_IsOpInitialized(session, + WP11_INIT_RSA_PKCS_VERIFY_RECOVER) || + WP11_Session_IsOpInitialized(session, + WP11_INIT_RSA_X_509_VERIFY_RECOVER); + /* Not IsOpCategoryActive: an op whose key was destroyed still holds + * state that must be released. */ + if (WP11_Session_IsOpCategoryInit(session, opCategory) && + recoverActive == recover) { + WP11_Session_AbortOp(session); + } + return CKR_OK; +} +#endif + static CK_RV EncryptInit(CK_SESSION_HANDLE hSession, CK_MECHANISM_PTR pMechanism, CK_OBJECT_HANDLE hKey, byte skipOpCheck) @@ -2475,7 +2822,11 @@ static CK_RV EncryptInit(CK_SESSION_HANDLE hSession, return rv; } if (pMechanism == NULL) { +#ifdef WOLFPKCS11_PKCS11_V3_0 + rv = CancelOperation(session, WP11_OP_ENCRYPT, 0); +#else rv = CKR_ARGUMENTS_BAD; +#endif WOLFPKCS11_LEAVE("C_EncryptInit", rv); return rv; } @@ -2536,6 +2887,8 @@ static CK_RV EncryptInit(CK_SESSION_HANDLE hSession, params = (CK_RSA_PKCS_OAEP_PARAMS*)pMechanism->pParameter; if (params->source != CKZ_DATA_SPECIFIED) return CKR_MECHANISM_PARAM_INVALID; + if (params->ulSourceDataLen > (CK_ULONG)INT_MAX) + return CKR_MECHANISM_PARAM_INVALID; ret = WP11_Session_SetOaepParams(session, params->hashAlg, params->mgf, (byte*)params->pSourceData, @@ -2642,6 +2995,11 @@ static CK_RV EncryptInit(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; params = (CK_GCM_PARAMS*)pMechanism->pParameter; + if (params->ulIvLen > (CK_ULONG)INT_MAX || + params->ulAADLen > (CK_ULONG)INT_MAX || + params->ulTagBits > 128) { + return CKR_MECHANISM_PARAM_INVALID; + } ret = WP11_Session_SetGcmParams(session, params->pIv, (int)params->ulIvLen, params->pAAD, (int)params->ulAADLen, @@ -2665,6 +3023,12 @@ static CK_RV EncryptInit(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; params = (CK_CCM_PARAMS*)pMechanism->pParameter; + if (params->ulDataLen > (CK_ULONG)INT_MAX || + params->ulIvLen > (CK_ULONG)INT_MAX || + params->ulAADLen > (CK_ULONG)INT_MAX || + params->ulMacLen > (CK_ULONG)INT_MAX) { + return CKR_MECHANISM_PARAM_INVALID; + } ret = WP11_Session_SetCcmParams(session, (int)params->ulDataLen, params->pIv, (int)params->ulIvLen, @@ -2730,7 +3094,8 @@ static CK_RV EncryptInit(CK_SESSION_HANDLE hSession, * @param hKey [in] Handle to key object. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pMechanism is NULL. + * CKR_ARGUMENTS_BAD when pMechanism is NULL; PKCS#11 v3.0 builds + * instead cancel the active operation and return CKR_OK. * CKR_OBJECT_HANDLE_INVALID when key object handle is not valid. * CKR_KEY_TYPE_INCONSISTENT when the key type is not valid for the * mechanism (operation). @@ -2884,6 +3249,11 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CBC_ENC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulDataLen)) { + WP11_Session_AbortOp(session); + return CKR_DATA_LEN_RANGE; + } + if ((ulDataLen % AES_BLOCK_SIZE) != 0) { + WP11_AesCbc_EncryptFinal(session); WP11_Session_SetOpInitialized(session, 0); return CKR_DATA_LEN_RANGE; } @@ -2908,12 +3278,12 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, return CKR_OPERATION_NOT_INITIALIZED; } if (!CK_ULONG_FITS_WORD32(ulDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } /* Ensure padded result fits in word32 */ if (ulDataLen > (CK_ULONG)(0xFFFFFFFF - AES_BLOCK_SIZE)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -2938,6 +3308,10 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, case CKM_AES_CTR: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTR_ENC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulDataLen)) { + WP11_AesCtr_Final(session); + return CKR_DATA_LEN_RANGE; + } if (pEncryptedData == NULL) { *pulEncryptedDataLen = ulDataLen; @@ -2961,7 +3335,7 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_GCM_ENC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -2986,7 +3360,7 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CCM_ENC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -3011,7 +3385,11 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_ECB_ENC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); + return CKR_DATA_LEN_RANGE; + } + if ((ulDataLen % AES_BLOCK_SIZE) != 0) { + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -3034,6 +3412,10 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, case CKM_AES_CTS: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTS_ENC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulDataLen)) { + WP11_Session_AbortOp(session); + return CKR_DATA_LEN_RANGE; + } encDataLen = (word32)*pulEncryptedDataLen; if (pEncryptedData == NULL) { @@ -3055,7 +3437,7 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_KEYWRAP_ENC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -3080,7 +3462,7 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_KEYWRAP_ENC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulDataLen) || ulDataLen == 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -3110,7 +3492,7 @@ static CK_RV wp11_C_Encrypt(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, (void)encDataLen; (void)ulDataLen; (void)pEncryptedData; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } @@ -3205,7 +3587,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, return rv; } if (!CK_ULONG_FITS_WORD32(ulPartLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -3227,7 +3609,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, pEncryptedPart, &encPartLen, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); rv = CKR_FUNCTION_FAILED; WOLFPKCS11_LEAVE("C_EncryptUpdate", rv); return rv; @@ -3240,7 +3622,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, return CKR_OPERATION_NOT_INITIALIZED; } if (!CK_ULONG_FITS_WORD32(ulPartLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -3256,7 +3638,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, ret = WP11_AesCbcPad_EncryptUpdate(pPart, (int)ulPartLen, pEncryptedPart, &encPartLen, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulEncryptedPartLen = encPartLen; @@ -3266,6 +3648,10 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, case CKM_AES_CTR: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTR_ENC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulPartLen)) { + WP11_AesCtr_Final(session); + return CKR_DATA_LEN_RANGE; + } if (pEncryptedPart == NULL) { *pulEncryptedPartLen = ulPartLen; @@ -3282,7 +3668,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, return CKR_DATA_LEN_RANGE; } if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulEncryptedPartLen = encPartLen; @@ -3292,6 +3678,12 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, case CKM_AES_GCM: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_GCM_ENC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulPartLen) || + ulPartLen > (CK_ULONG)(INT_MAX - + WP11_AesGcm_EncDataLen(session))) { + WP11_Session_AbortOp(session); + return CKR_DATA_LEN_RANGE; + } encPartLen = (word32)ulPartLen; if (pEncryptedPart == NULL) { @@ -3305,7 +3697,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, pEncryptedPart, &encPartLen, obj, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulEncryptedPartLen = encPartLen; @@ -3315,6 +3707,10 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, case CKM_AES_CTS: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTS_ENC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulPartLen)) { + WP11_Session_AbortOp(session); + return CKR_DATA_LEN_RANGE; + } if (pEncryptedPart == NULL) { *pulEncryptedPartLen = ulPartLen + AES_BLOCK_SIZE * 2; @@ -3327,7 +3723,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, if (ret == BUFFER_E) return CKR_BUFFER_TOO_SMALL; if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulEncryptedPartLen = encPartLen; @@ -3339,7 +3735,7 @@ static CK_RV wp11_C_EncryptUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, (void)ret; (void)ulPartLen; (void)pEncryptedPart; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); rv = CKR_MECHANISM_INVALID; WOLFPKCS11_LEAVE("C_EncryptUpdate", rv); return rv; @@ -3435,7 +3831,7 @@ static CK_RV wp11_C_EncryptFinal(CK_SESSION_HANDLE hSession, ret = WP11_AesCbc_EncryptFinal(session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } break; @@ -3450,15 +3846,16 @@ static CK_RV wp11_C_EncryptFinal(CK_SESSION_HANDLE hSession, *pulLastEncryptedPartLen = encPartLen; return CKR_OK; } - if (encPartLen > (word32)*pulLastEncryptedPartLen) + if ((CK_ULONG)encPartLen > *pulLastEncryptedPartLen) return CKR_BUFFER_TOO_SMALL; ret = WP11_AesCbcPad_EncryptFinal(pLastEncryptedPart, &encPartLen, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } + *pulLastEncryptedPartLen = encPartLen; break; #endif #ifdef HAVE_AESCTR @@ -3473,7 +3870,7 @@ static CK_RV wp11_C_EncryptFinal(CK_SESSION_HANDLE hSession, ret = WP11_AesCtr_Final(session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastEncryptedPartLen = 0; @@ -3489,13 +3886,13 @@ static CK_RV wp11_C_EncryptFinal(CK_SESSION_HANDLE hSession, *pulLastEncryptedPartLen = encPartLen; return CKR_OK; } - if (encPartLen > (word32)*pulLastEncryptedPartLen) + if ((CK_ULONG)encPartLen > *pulLastEncryptedPartLen) return CKR_BUFFER_TOO_SMALL; ret = WP11_AesGcm_EncryptFinal(pLastEncryptedPart, &encPartLen, obj, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastEncryptedPartLen = encPartLen; @@ -3511,13 +3908,18 @@ static CK_RV wp11_C_EncryptFinal(CK_SESSION_HANDLE hSession, return CKR_OK; } - encPartLen = (word32)*pulLastEncryptedPartLen; + /* The final output is at most two blocks; cap the capacity to + * avoid truncating a large CK_ULONG. */ + encPartLen = (*pulLastEncryptedPartLen < + (CK_ULONG)(AES_BLOCK_SIZE * 2)) ? + (word32)*pulLastEncryptedPartLen : + (word32)(AES_BLOCK_SIZE * 2); ret = WP11_AesCts_EncryptFinal(pLastEncryptedPart, &encPartLen, session); if (ret == BUFFER_E) return CKR_BUFFER_TOO_SMALL; if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastEncryptedPartLen = encPartLen; @@ -3528,7 +3930,7 @@ static CK_RV wp11_C_EncryptFinal(CK_SESSION_HANDLE hSession, (void)encPartLen; (void)ret; (void)pLastEncryptedPart; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } @@ -3575,8 +3977,13 @@ static CK_RV DecryptInit(CK_SESSION_HANDLE hSession, } if (WP11_Session_Get(hSession, &session) != 0) return CKR_SESSION_HANDLE_INVALID; - if (pMechanism == NULL) + if (pMechanism == NULL) { +#ifdef WOLFPKCS11_PKCS11_V3_0 + return CancelOperation(session, WP11_OP_DECRYPT, 0); +#else return CKR_ARGUMENTS_BAD; +#endif + } ret = WP11_Object_Find(session, hKey, &obj); if (ret != 0) @@ -3629,6 +4036,8 @@ static CK_RV DecryptInit(CK_SESSION_HANDLE hSession, params = (CK_RSA_PKCS_OAEP_PARAMS*)pMechanism->pParameter; if (params->source != CKZ_DATA_SPECIFIED) return CKR_MECHANISM_PARAM_INVALID; + if (params->ulSourceDataLen > (CK_ULONG)INT_MAX) + return CKR_MECHANISM_PARAM_INVALID; ret = WP11_Session_SetOaepParams(session, params->hashAlg, params->mgf, (byte*)params->pSourceData, @@ -3705,6 +4114,11 @@ static CK_RV DecryptInit(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; params = (CK_GCM_PARAMS*)pMechanism->pParameter; + if (params->ulIvLen > (CK_ULONG)INT_MAX || + params->ulAADLen > (CK_ULONG)INT_MAX || + params->ulTagBits > 128) { + return CKR_MECHANISM_PARAM_INVALID; + } ret = WP11_Session_SetGcmParams(session, params->pIv, (int)params->ulIvLen, params->pAAD, (int)params->ulAADLen, @@ -3727,6 +4141,12 @@ static CK_RV DecryptInit(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; params = (CK_CCM_PARAMS*)pMechanism->pParameter; + if (params->ulDataLen > (CK_ULONG)INT_MAX || + params->ulIvLen > (CK_ULONG)INT_MAX || + params->ulAADLen > (CK_ULONG)INT_MAX || + params->ulMacLen > (CK_ULONG)INT_MAX) { + return CKR_MECHANISM_PARAM_INVALID; + } ret = WP11_Session_SetCcmParams(session, (int)params->ulDataLen, params->pIv, (int)params->ulIvLen, @@ -3817,7 +4237,8 @@ static CK_RV DecryptInit(CK_SESSION_HANDLE hSession, * @param hKey [in] Handle to key object. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pMechanism is NULL. + * CKR_ARGUMENTS_BAD when pMechanism is NULL; PKCS#11 v3.0 builds + * instead cancel the active operation and return CKR_OK. * CKR_OBJECT_HANDLE_INVALID when key object handle is not valid. * CKR_KEY_TYPE_INCONSISTENT when the key type is not valid for the * mechanism (operation). @@ -3968,9 +4389,14 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CBC_DEC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } + if ((ulEncryptedDataLen % AES_BLOCK_SIZE) != 0) { + WP11_AesCbc_DecryptFinal(session); + WP11_Session_SetOpInitialized(session, 0); + return CKR_ENCRYPTED_DATA_LEN_RANGE; + } decDataLen = (word32)ulEncryptedDataLen; if (pData == NULL) { @@ -3992,7 +4418,7 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, return CKR_OPERATION_NOT_INITIALIZED; } if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -4033,8 +4459,12 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, case CKM_AES_CTR: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTR_DEC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen)) { + WP11_AesCtr_Final(session); + return CKR_ENCRYPTED_DATA_LEN_RANGE; + } - if (pEncryptedData == NULL) { + if (pData == NULL) { *pulDataLen = ulEncryptedDataLen; return CKR_OK; } @@ -4057,7 +4487,7 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, return CKR_OPERATION_NOT_INITIALIZED; if (ulEncryptedDataLen < (CK_ULONG)WP11_AesGcm_GetTagBits(session) / 8) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_ENCRYPTED_DATA_LEN_RANGE; } decDataLen = (word32)ulEncryptedDataLen - @@ -4082,7 +4512,7 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, return CKR_OPERATION_NOT_INITIALIZED; if (ulEncryptedDataLen < (CK_ULONG)WP11_AesCcm_GetMacLen(session)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_ENCRYPTED_DATA_LEN_RANGE; } decDataLen = (word32)ulEncryptedDataLen - @@ -4106,9 +4536,13 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_ECB_DEC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } + if ((ulEncryptedDataLen % AES_BLOCK_SIZE) != 0) { + WP11_Session_AbortOp(session); + return CKR_ENCRYPTED_DATA_LEN_RANGE; + } decDataLen = (word32)ulEncryptedDataLen; if (pData == NULL) { @@ -4129,6 +4563,10 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, case CKM_AES_CTS: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTS_DEC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen)) { + WP11_Session_AbortOp(session); + return CKR_ENCRYPTED_DATA_LEN_RANGE; + } decDataLen = (word32)*pulDataLen; if (pData == NULL) { @@ -4154,8 +4592,9 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, /* AES Key Wrap ciphertext is at least two semiblocks: one data * semiblock plus the 8-byte integrity check value. */ - if (ulEncryptedDataLen < 2 * KEYWRAP_BLOCK_SIZE) { - WP11_Session_SetOpInitialized(session, 0); + if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen) || + ulEncryptedDataLen < 2 * KEYWRAP_BLOCK_SIZE) { + WP11_Session_AbortOp(session); return CKR_ENCRYPTED_DATA_LEN_RANGE; } decDataLen = (word32)(ulEncryptedDataLen - KEYWRAP_BLOCK_SIZE); @@ -4179,9 +4618,10 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, /* RFC 5649 ciphertext is at least two semiblocks and a multiple of * 8. The recovered length is encoded in the AIV and is known only * after unwrapping; the upper bound is ciphertext - 8. */ - if (ulEncryptedDataLen < 2 * KEYWRAP_BLOCK_SIZE || + if (!CK_ULONG_FITS_WORD32(ulEncryptedDataLen) || + ulEncryptedDataLen < 2 * KEYWRAP_BLOCK_SIZE || (ulEncryptedDataLen % KEYWRAP_BLOCK_SIZE) != 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_ENCRYPTED_DATA_LEN_RANGE; } decDataLen = (word32)(ulEncryptedDataLen - KEYWRAP_BLOCK_SIZE); @@ -4214,7 +4654,7 @@ static CK_RV wp11_C_Decrypt(CK_SESSION_HANDLE hSession, (void)ret; (void)ulEncryptedDataLen; (void)pData; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } @@ -4301,7 +4741,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CBC_DEC)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulEncryptedPartLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -4319,7 +4759,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, (int)ulEncryptedPartLen, pPart, &decPartLen, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulPartLen = decPartLen; @@ -4330,7 +4770,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, return CKR_OPERATION_NOT_INITIALIZED; } if (!CK_ULONG_FITS_WORD32(ulEncryptedPartLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -4357,7 +4797,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, return CKR_BUFFER_TOO_SMALL; } if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulPartLen = decPartLen; @@ -4367,6 +4807,10 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, case CKM_AES_CTR: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_CTR_DEC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulEncryptedPartLen)) { + WP11_AesCtr_Final(session); + return CKR_ENCRYPTED_DATA_LEN_RANGE; + } if (pPart == NULL) { *pulPartLen = ulEncryptedPartLen; @@ -4375,7 +4819,8 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, if (ulEncryptedPartLen > *pulPartLen) return CKR_BUFFER_TOO_SMALL; - decPartLen = (word32)*pulPartLen; + /* Output is the same length as the input, which fits the buffer. */ + decPartLen = (word32)ulEncryptedPartLen; ret = WP11_AesCtr_Update(pEncryptedPart, (word32)ulEncryptedPartLen, pPart, &decPartLen, session); if (ret == WP11_CTR_OVERFLOW_E) { @@ -4383,7 +4828,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, return CKR_DATA_LEN_RANGE; } if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulPartLen = decPartLen; @@ -4393,6 +4838,12 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, case CKM_AES_GCM: if (!WP11_Session_IsOpInitialized(session, WP11_INIT_AES_GCM_DEC)) return CKR_OPERATION_NOT_INITIALIZED; + if (!CK_ULONG_FITS_WORD32(ulEncryptedPartLen) || + ulEncryptedPartLen > (CK_ULONG)(INT_MAX - + WP11_AesGcm_EncDataLen(session))) { + WP11_Session_AbortOp(session); + return CKR_ENCRYPTED_DATA_LEN_RANGE; + } *pulPartLen = 0; if (pPart == NULL) @@ -4401,7 +4852,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, ret = WP11_AesGcm_DecryptUpdate(pEncryptedPart, (int)ulEncryptedPartLen, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } break; @@ -4422,7 +4873,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, if (ret == BUFFER_E) return CKR_BUFFER_TOO_SMALL; if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulPartLen = decPartLen; @@ -4434,7 +4885,7 @@ static CK_RV wp11_C_DecryptUpdate(CK_SESSION_HANDLE hSession, (void)ret; (void)ulEncryptedPartLen; (void)pPart; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } @@ -4527,7 +4978,7 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, ret = WP11_AesCbc_DecryptFinal(session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } break; @@ -4562,7 +5013,7 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, return CKR_BUFFER_TOO_SMALL; } if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastPartLen = decPartLen; @@ -4580,7 +5031,7 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, ret = WP11_AesCtr_Final(session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastPartLen = 0; @@ -4593,7 +5044,7 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, if (WP11_AesGcm_EncDataLen(session) < WP11_AesGcm_GetTagBits(session) / 8) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_ENCRYPTED_DATA_LEN_RANGE; } decPartLen = WP11_AesGcm_EncDataLen(session) - @@ -4602,13 +5053,13 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, *pulLastPartLen = decPartLen; return CKR_OK; } - if (decPartLen > (word32)*pulLastPartLen) + if ((CK_ULONG)decPartLen > *pulLastPartLen) return CKR_BUFFER_TOO_SMALL; ret = WP11_AesGcm_DecryptFinal(pLastPart, &decPartLen, obj, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastPartLen = decPartLen; @@ -4629,7 +5080,7 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, if (ret == BUFFER_E) return CKR_BUFFER_TOO_SMALL; if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } *pulLastPartLen = decPartLen; @@ -4640,7 +5091,7 @@ static CK_RV wp11_C_DecryptFinal(CK_SESSION_HANDLE hSession, (void)decPartLen; (void)ret; (void)pLastPart; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } @@ -4668,7 +5119,8 @@ CK_RV C_DecryptFinal(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pLastPart, * @param pMechanism [in] Type of operation to perform with parameters. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pMechanism is NULL. + * CKR_ARGUMENTS_BAD when pMechanism is NULL; PKCS#11 v3.0 builds + * instead cancel the active operation and return CKR_OK. * CKR_MECHANISM_INVALID when the mechanism is not supported with this * type of operation. */ @@ -4698,7 +5150,11 @@ CK_RV C_DigestInit(CK_SESSION_HANDLE hSession, return rv; } if (pMechanism == NULL) { +#ifdef WOLFPKCS11_PKCS11_V3_0 + rv = CancelOperation(session, WP11_OP_DIGEST, 0); +#else rv = CKR_ARGUMENTS_BAD; +#endif WOLFPKCS11_LEAVE("C_DigestInit", rv); return rv; } @@ -4829,7 +5285,7 @@ CK_RV C_DigestUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, ret = WP11_Digest_Update(pPart, (word32)ulPartLen, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } return CKR_OK; @@ -4873,21 +5329,21 @@ static CK_RV wp11_C_DigestKey(CK_SESSION_HANDLE hSession, CK_OBJECT_HANDLE hKey) ret = WP11_Object_Find(session, hKey, &obj); if (ret != 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_OBJECT_HANDLE_INVALID; } ret = WP11_Digest_Key(obj, session); if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } if (ret > 0) { /* Positive return is a CK_RV (e.g. CKR_FUNCTION_NOT_SUPPORTED on * WOLFPKCS11_NO_STORE builds). Pass it through but still terminate * the digest operation so the session is not left active. */ - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return (CK_RV)ret; } return CKR_OK; @@ -5097,7 +5553,8 @@ static int GetInitValue(CK_MECHANISM_TYPE mechanism) { * @param hKey [in] Handle to key object. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pMechanism is NULL. + * CKR_ARGUMENTS_BAD when pMechanism is NULL; PKCS#11 v3.0 builds + * instead cancel the active operation and return CKR_OK. * CKR_OBJECT_HANDLE_INVALID when key object handle is not valid. * CKR_KEY_TYPE_INCONSISTENT when the key type is not valid for the * mechanism (operation). @@ -5132,8 +5589,13 @@ static CK_RV wp11_C_SignInit(CK_SESSION_HANDLE hSession, } if (WP11_Session_Get(hSession, &session) != 0) return CKR_SESSION_HANDLE_INVALID; - if (pMechanism == NULL) + if (pMechanism == NULL) { +#ifdef WOLFPKCS11_PKCS11_V3_0 + return CancelOperation(session, WP11_OP_SIGN, 0); +#else return CKR_ARGUMENTS_BAD; +#endif + } ret = WP11_Object_Find(session, hKey, &obj); #ifdef WOLFSSL_MAXQ10XX_CRYPTO @@ -5804,7 +6266,7 @@ static CK_RV wp11_C_Sign(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, (void)sigLen; (void)ulDataLen; (void)pSignature; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } WP11_Session_SetOpInitialized(session, 0); @@ -5935,7 +6397,7 @@ static CK_RV wp11_C_SignUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, if (!WP11_Session_IsOpInitialized(session, WP11_INIT_TLS_MAC_SIGN)) return CKR_OPERATION_NOT_INITIALIZED; if (!CK_ULONG_FITS_WORD32(ulPartLen)) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_DATA_LEN_RANGE; } @@ -5944,11 +6406,11 @@ static CK_RV wp11_C_SignUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, #endif default: (void)ulPartLen; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } @@ -6121,7 +6583,7 @@ static CK_RV wp11_C_SignFinal(CK_SESSION_HANDLE hSession, default: (void)sigLen; (void)pSignature; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } WP11_Session_SetOpInitialized(session, 0); @@ -6276,7 +6738,8 @@ CK_RV C_SignRecover(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, * @param hKey [in] Handle to key object. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pMechanism is NULL. + * CKR_ARGUMENTS_BAD when pMechanism is NULL; PKCS#11 v3.0 builds + * instead cancel the active operation and return CKR_OK. * CKR_OBJECT_HANDLE_INVALID when key object handle is not valid. * CKR_KEY_TYPE_INCONSISTENT when the key type is not valid for the * mechanism (operation). @@ -6316,7 +6779,11 @@ static CK_RV wp11_C_VerifyInit(CK_SESSION_HANDLE hSession, return rv; } if (pMechanism == NULL) { +#ifdef WOLFPKCS11_PKCS11_V3_0 + rv = CancelOperation(session, WP11_OP_VERIFY, 0); +#else rv = CKR_ARGUMENTS_BAD; +#endif WOLFPKCS11_LEAVE("C_VerifyInit", rv); return rv; } @@ -6942,7 +7409,7 @@ static CK_RV wp11_C_Verify(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pData, default: (void)ulDataLen; (void)ulSignatureLen; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } WP11_Session_SetOpInitialized(session, 0); @@ -7075,11 +7542,11 @@ static CK_RV wp11_C_VerifyUpdate(CK_SESSION_HANDLE hSession, CK_BYTE_PTR pPart, #endif default: (void)ulPartLen; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } if (ret < 0) { - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_FUNCTION_FAILED; } @@ -7210,7 +7677,7 @@ static CK_RV wp11_C_VerifyFinal(CK_SESSION_HANDLE hSession, #endif default: (void)ulSignatureLen; - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); return CKR_MECHANISM_INVALID; } WP11_Session_SetOpInitialized(session, 0); @@ -7242,7 +7709,8 @@ CK_RV C_VerifyFinal(CK_SESSION_HANDLE hSession, * @param hKey [in] Handle to key object. * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. - * CKR_ARGUMENTS_BAD when pMechanism is NULL. + * CKR_ARGUMENTS_BAD when pMechanism is NULL; PKCS#11 v3.0 builds + * instead cancel the active operation and return CKR_OK. * CKR_OBJECT_HANDLE_INVALID when key object handle is not valid. * CKR_MECHANISM_INVALID when the mechanism is not supported with this * type of operation. @@ -7275,7 +7743,11 @@ static CK_RV wp11_C_VerifyRecoverInit(CK_SESSION_HANDLE hSession, return rv; } if (pMechanism == NULL) { +#ifdef WOLFPKCS11_PKCS11_V3_0 + rv = CancelOperation(session, WP11_OP_VERIFY, 1); +#else rv = CKR_ARGUMENTS_BAD; +#endif WOLFPKCS11_LEAVE("C_VerifyRecoverInit", rv); return rv; } @@ -8711,6 +9183,10 @@ static CK_RV wp11_C_WrapKey(CK_SESSION_HANDLE hSession, CK_OBJECT_CLASS keyClass = CKO_PRIVATE_KEY; word32 serialSize = 0; byte* serialBuff = NULL; +#ifndef NO_AES + byte* padBuff; + word32 padSize; +#endif CK_BBOOL getVar; CK_ULONG getVarLen = sizeof(CK_BBOOL); @@ -8843,6 +9319,31 @@ static CK_RV wp11_C_WrapKey(CK_SESSION_HANDLE hSession, rv = CKR_WRAPPING_KEY_TYPE_INCONSISTENT; goto err_out; } + /* AES-ECB zero-pads a secret key value to the block size. */ + if (pMechanism->mechanism == CKM_AES_ECB && + keyClass == CKO_SECRET_KEY && + (serialSize % AES_BLOCK_SIZE) != 0) { + padSize = serialSize + AES_BLOCK_SIZE - + (serialSize % AES_BLOCK_SIZE); + padBuff = (byte*)XMALLOC(padSize, NULL, + DYNAMIC_TYPE_TMP_BUFFER); + if (padBuff == NULL) { + rv = CKR_HOST_MEMORY; + goto err_out; + } + XMEMCPY(padBuff, serialBuff, serialSize); + XMEMSET(padBuff + serialSize, 0, padSize - serialSize); + wc_ForceZero(serialBuff, serialSize); + XFREE(serialBuff, NULL, DYNAMIC_TYPE_TMP_BUFFER); + serialBuff = padBuff; + serialSize = padSize; + } + /* Only a block aligned key encoding can be ECB wrapped. */ + if (pMechanism->mechanism == CKM_AES_ECB && + (serialSize % AES_BLOCK_SIZE) != 0) { + rv = CKR_KEY_SIZE_RANGE; + goto err_out; + } #ifdef WOLFSSL_STM32U5_DHUK if (WP11_Object_GetDevId(wrappingKey) == @@ -8872,7 +9373,7 @@ static CK_RV wp11_C_WrapKey(CK_SESSION_HANDLE hSession, /* C_WrapKey is single-part: a length query or a short buffer * leaves C_Encrypt's operation active, which would make the * caller's next C_WrapKey fail with CKR_OPERATION_ACTIVE. */ - WP11_Session_SetOpInitialized(session, 0); + WP11_Session_AbortOp(session); if (rv != CKR_OK) goto err_out; } @@ -9111,6 +9612,27 @@ static CK_RV wp11_C_UnwrapKey(CK_SESSION_HANDLE hSession, if (rv != CKR_OK) goto err_out; + /* ECB zero pads; a 32 byte AES value may be a padded AES-192 key. */ + if (pMechanism->mechanism == CKM_AES_ECB && + keyClass == CKO_SECRET_KEY) { + FindAttributeType(pTemplate, ulAttributeCount, CKA_VALUE_LEN, + &attr); + if (attr == NULL && (keyType != CKK_AES || + ulUnwrappedLen != AES_128_KEY_SIZE)) { + rv = CKR_TEMPLATE_INCOMPLETE; + goto err_out; + } + if (attr != NULL && (attr->pValue == NULL || + attr->ulValueLen != sizeof(CK_ULONG) || + *(CK_ULONG*)attr->pValue == 0 || + *(CK_ULONG*)attr->pValue > ulUnwrappedLen)) { + rv = CKR_TEMPLATE_INCONSISTENT; + goto err_out; + } + if (attr != NULL) + ulUnwrappedLen = *(CK_ULONG*)attr->pValue; + } + break; #endif #ifndef NO_RSA @@ -9222,7 +9744,7 @@ CK_RV C_UnwrapKey(CK_SESSION_HANDLE hSession, return rv; } -#if defined(HAVE_ECC) || !defined(NO_DH) +#ifdef WP11_DERIVE_SECRET_KEY /** * Determine the key length of the object. * @@ -9246,7 +9768,7 @@ static int SymmKeyLen(WP11_Object* obj, word32 len, CK_ULONG* symmKeyLen) switch (WP11_Object_GetType(obj)) { case CKK_AES: -#ifdef WOLFPKCS11_NSS +#if defined(WOLFPKCS11_NSS) && !defined(NO_AES) /* This is the only wrapping mechanism that we support. NSS chooses * the wrapping mechanism from the list in wrapMechanismList in * PK11_GetBestWrapMechanism. Unfortunately this relies on a default @@ -9277,7 +9799,9 @@ static int SymmKeyLen(WP11_Object* obj, word32 len, CK_ULONG* symmKeyLen) #ifdef WOLFSSL_HAVE_PRF static int SetKeyExtract(WP11_Session* session, byte* ptr, CK_ULONG length, CK_ATTRIBUTE_PTR pTemplate, CK_ULONG ulAttributeCount, - CK_BBOOL isMac, CK_OBJECT_HANDLE* handle) + CK_BBOOL isMac, CK_BBOOL baseAlwaysSensitive, + CK_BBOOL baseNeverExtractable, + CK_OBJECT_HANDLE* handle) { WP11_Object* secret = NULL; int ret; @@ -9304,66 +9828,61 @@ static int SetKeyExtract(WP11_Session* session, byte* ptr, CK_ULONG length, WP11_Object_Free(secret); return CKR_FUNCTION_FAILED; } - ret = (int)AddObject(session, secret, pTemplate, ulAttributeCount, - handle); + ret = (int)SetDerivedStates(secret, baseAlwaysSensitive, + baseNeverExtractable); if (ret != CKR_OK) { WP11_Object_Free(secret); - return ret; + return CKR_FUNCTION_FAILED; } } else { WP11_Object_Free(secret); - return ret; + return CKR_FUNCTION_FAILED; } if (isMac) { ret = WP11_Object_SetAttr(secret, CKA_KEY_TYPE, (byte*)&keyType, sizeof(keyType)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_DERIVE, &ckTrue, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_ENCRYPT, &ckFalse, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_DECRYPT, &ckFalse, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_SIGN, &ckTrue, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_VERIFY, &ckTrue, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_WRAP, &ckFalse, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; - - ret = WP11_Object_SetAttr(secret, CKA_UNWRAP, &ckFalse, - sizeof(CK_BBOOL)); - if (ret != CKR_OK) - return ret; + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_DERIVE, &ckTrue, + sizeof(CK_BBOOL)); + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_ENCRYPT, &ckFalse, + sizeof(CK_BBOOL)); + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_DECRYPT, &ckFalse, + sizeof(CK_BBOOL)); + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_SIGN, &ckTrue, + sizeof(CK_BBOOL)); + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_VERIFY, &ckTrue, + sizeof(CK_BBOOL)); + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_WRAP, &ckFalse, + sizeof(CK_BBOOL)); + if (ret == CKR_OK) + ret = WP11_Object_SetAttr(secret, CKA_UNWRAP, &ckFalse, + sizeof(CK_BBOOL)); + if (ret != CKR_OK) { + WP11_Object_Free(secret); + return CKR_FUNCTION_FAILED; + } } + /* Add only once the key is complete so a failure leaves no object. */ + ret = (int)AddObject(session, secret, pTemplate, ulAttributeCount, handle); + if (ret != CKR_OK) + WP11_Object_Free(secret); + return ret; } static int Tls12_Extract_Keys(WP11_Session* session, CK_TLS12_KEY_MAT_PARAMS* tlsParams, CK_ATTRIBUTE_PTR pTemplate, - CK_ULONG ulAttributeCount, byte* derivedKey) + CK_ULONG ulAttributeCount, byte* derivedKey, + CK_BBOOL baseAlwaysSensitive, + CK_BBOOL baseNeverExtractable) { int ret = 0; unsigned char* ptr = derivedKey; @@ -9376,7 +9895,8 @@ static int Tls12_Extract_Keys(WP11_Session* session, /* Client MAC key */ length = tlsParams->ulMacSizeInBits / 8; ret = SetKeyExtract(session, ptr, length, pTemplate, - ulAttributeCount, CK_TRUE, + ulAttributeCount, CK_TRUE, baseAlwaysSensitive, + baseNeverExtractable, &tlsParams->pReturnedKeyMaterial->hClientMacSecret); if (ret != 0) { return ret; @@ -9384,7 +9904,8 @@ static int Tls12_Extract_Keys(WP11_Session* session, ptr += length; /* Server MAC key */ ret = SetKeyExtract(session, ptr, length, pTemplate, - ulAttributeCount, CK_TRUE, + ulAttributeCount, CK_TRUE, baseAlwaysSensitive, + baseNeverExtractable, &tlsParams->pReturnedKeyMaterial->hServerMacSecret); if (ret != 0) { return ret; @@ -9393,7 +9914,8 @@ static int Tls12_Extract_Keys(WP11_Session* session, /* Client key */ length = tlsParams->ulKeySizeInBits / 8; ret = SetKeyExtract(session, ptr, length, pTemplate, - ulAttributeCount, CK_FALSE, + ulAttributeCount, CK_FALSE, baseAlwaysSensitive, + baseNeverExtractable, &tlsParams->pReturnedKeyMaterial->hClientKey); if (ret != 0) { return ret; @@ -9401,7 +9923,8 @@ static int Tls12_Extract_Keys(WP11_Session* session, ptr += length; /* Server key */ ret = SetKeyExtract(session, ptr, length, pTemplate, - ulAttributeCount, CK_FALSE, + ulAttributeCount, CK_FALSE, baseAlwaysSensitive, + baseNeverExtractable, &tlsParams->pReturnedKeyMaterial->hServerKey); if (ret != 0) { return ret; @@ -9457,7 +9980,7 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, CK_RV rv = CKR_OK; WP11_Session* session; WP11_Object* obj = NULL; -#if defined(HAVE_ECC) || !defined(NO_DH) || defined(WOLFPKCS11_HKDF) +#ifdef WP11_DERIVE_SECRET_KEY byte* derivedKey = NULL; word32 keyLen; CK_ULONG symmKeyLen; @@ -9469,6 +9992,16 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, CK_BBOOL baseExtractable = CK_TRUE; CK_ULONG bLen; #endif +#endif +#ifdef WP11_DERIVE_SECRET_KEY + CK_BBOOL baseAlwaysSensitive = CK_FALSE; + CK_BBOOL baseNeverExtractable = CK_FALSE; + CK_ULONG histLen; + CK_ATTRIBUTE* valueLenAttr = NULL; + CK_ULONG reqValueLen = 0; +#endif +#ifndef NO_DH + CK_ULONG dhPrimeLen = 0; #endif WOLFPKCS11_ENTER("C_DeriveKey"); @@ -9533,6 +10066,11 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, return ret; #endif + rv = CheckPrivateLogin(session, pTemplate, ulAttributeCount, + CKO_SECRET_KEY); + if (rv != CKR_OK) + return rv; + switch (pMechanism->mechanism) { #ifdef HAVE_ECC case CKM_ECDH1_DERIVE: { @@ -9549,8 +10087,13 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; if (params->ulPublicDataLen == 0) return CKR_MECHANISM_PARAM_INVALID; + /* Largest DER-wrapped uncompressed point of a supported curve. */ + if (params->ulPublicDataLen > (CK_ULONG)(2 * MAX_ECC_BYTES + 4)) + return CKR_MECHANISM_PARAM_INVALID; if (params->kdf != CKD_NULL) return CKR_MECHANISM_PARAM_INVALID; + if (params->pSharedData != NULL || params->ulSharedDataLen != 0) + return CKR_MECHANISM_PARAM_INVALID; keyLen = (word32)(params->ulPublicDataLen / 2); derivedKey = (byte*)XMALLOC(keyLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -9559,8 +10102,8 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, XMEMSET(derivedKey, 0, keyLen); ret = WP11_EC_Derive(params->pPublicData, - (int)params->ulPublicDataLen, derivedKey, - &keyLen, obj); + (word32)params->ulPublicDataLen, derivedKey, + &keyLen, obj); if (ret != 0) rv = CKR_FUNCTION_FAILED; break; @@ -9585,6 +10128,13 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, kdfParams = (CK_HKDF_PARAMS_PTR)pMechanism->pParameter; if (!kdfParams->bExpand && !kdfParams->bExtract) return CKR_MECHANISM_PARAM_INVALID; + if (kdfParams->bExtract && + kdfParams->ulSaltType == CKF_HKDF_SALT_DATA && + !CK_ULONG_FITS_WORD32(kdfParams->ulSaltLen)) + return CKR_MECHANISM_PARAM_INVALID; + if (kdfParams->bExpand && + !CK_ULONG_FITS_WORD32(kdfParams->ulInfoLen)) + return CKR_MECHANISM_PARAM_INVALID; FindAttributeType(pTemplate, ulAttributeCount, CKA_VALUE_LEN, &lenAttr); @@ -9614,7 +10164,10 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, ret = WP11_KDF_Derive(session, kdfParams, derivedKey, &keyLen, obj); - if (ret != 0) + /* A parameter check reports a CK_RV, not a wolfCrypt error. */ + if (ret == (int)CKR_MECHANISM_PARAM_INVALID) + rv = CKR_MECHANISM_PARAM_INVALID; + else if (ret != 0) rv = CKR_FUNCTION_FAILED; break; } @@ -9625,10 +10178,17 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, return CKR_KEY_TYPE_INCONSISTENT; if (pMechanism->pParameter == NULL) return CKR_MECHANISM_PARAM_INVALID; - if (pMechanism->ulParameterLen == 0) + if (pMechanism->ulParameterLen == 0 || + pMechanism->ulParameterLen > WP11_MAX_DH_KEY_SZ) return CKR_MECHANISM_PARAM_INVALID; + /* The secret is bounded by the prime, not by the peer value. */ + if (WP11_Object_GetAttr(obj, CKA_PRIME, NULL, &dhPrimeLen) != 0 || + dhPrimeLen > WP11_MAX_DH_KEY_SZ) + return CKR_FUNCTION_FAILED; keyLen = (word32)pMechanism->ulParameterLen; + if (dhPrimeLen > keyLen) + keyLen = (word32)dhPrimeLen; derivedKey = (byte*)XMALLOC(keyLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); if (derivedKey == NULL) return CKR_DEVICE_MEMORY; @@ -9653,7 +10213,7 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, sizeof(CK_AES_CBC_ENCRYPT_DATA_PARAMS)) return CKR_MECHANISM_PARAM_INVALID; params = (CK_AES_CBC_ENCRYPT_DATA_PARAMS*)pMechanism->pParameter; - if (params->length % 16) + if (!CK_ULONG_FITS_WORD32(params->length) || params->length % 16) return CKR_MECHANISM_PARAM_INVALID; keyLen = (word32)params->length; @@ -9682,7 +10242,9 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, sizeof(CK_TLS12_KEY_MAT_PARAMS)) return CKR_MECHANISM_PARAM_INVALID; tlsParams = (CK_TLS12_KEY_MAT_PARAMS*) pMechanism->pParameter; - if (tlsParams->pReturnedKeyMaterial == NULL) + if (tlsParams->pReturnedKeyMaterial == NULL || + tlsParams->RandomInfo.pClientRandom == NULL || + tlsParams->RandomInfo.pServerRandom == NULL) return CKR_MECHANISM_PARAM_INVALID; { @@ -9696,6 +10258,12 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, tlsParams->ulIVSizeInBits > ((CK_ULONG)0xFFFFFFFF / 2)) { return CKR_MECHANISM_PARAM_INVALID; } + /* Each field is extracted separately in whole bytes */ + if ((tlsParams->ulMacSizeInBits % 8) != 0 || + (tlsParams->ulKeySizeInBits % 8) != 0 || + (tlsParams->ulIVSizeInBits % 8) != 0) { + return CKR_MECHANISM_PARAM_INVALID; + } /* Check sum won't overflow on 32-bit */ if (a > (CK_ULONG)0xFFFFFFFF - b) return CKR_MECHANISM_PARAM_INVALID; @@ -9723,9 +10291,20 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, "key expansion", 13, derivedKey, keyLen, CK_FALSE, obj); - if (ret == 0) + if (ret == 0) { + histLen = sizeof(CK_BBOOL); + if (WP11_Object_GetAttr(obj, CKA_ALWAYS_SENSITIVE, + &baseAlwaysSensitive, &histLen) != 0) + baseAlwaysSensitive = CK_FALSE; + histLen = sizeof(CK_BBOOL); + if (WP11_Object_GetAttr(obj, CKA_NEVER_EXTRACTABLE, + &baseNeverExtractable, &histLen) != 0) + baseNeverExtractable = CK_FALSE; ret = Tls12_Extract_Keys(session, tlsParams, pTemplate, - ulAttributeCount, derivedKey); + ulAttributeCount, derivedKey, + baseAlwaysSensitive, + baseNeverExtractable); + } /* Freeing here so that we don't attempt to generate a key at the * end of the function */ @@ -9759,7 +10338,7 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; if ((prfParams->pVersion->major != 3) || (prfParams->pVersion->minor != 3)) - return CKR_MECHANISM_INVALID; + return CKR_MECHANISM_PARAM_INVALID; } keyLen = PRF_KEY_SIZE; @@ -9790,6 +10369,9 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_PARAM_INVALID; nssParams = (CK_NSS_TLS_EXTENDED_MASTER_KEY_DERIVE_PARAMS*) pMechanism->pParameter; + if (nssParams->pSessionHash == NULL || + !CK_ULONG_FITS_WORD32(nssParams->ulSessionHashLen)) + return CKR_MECHANISM_PARAM_INVALID; keyLen = PRF_KEY_SIZE; derivedKey = (byte*)XMALLOC(keyLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -9814,11 +10396,9 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, return CKR_MECHANISM_INVALID; } -#if defined(HAVE_ECC) || !defined(NO_DH) || defined(WOLFPKCS11_HKDF) || \ - (!defined(NO_AES) && defined(HAVE_AES_CBC)) +#ifdef WP11_DERIVE_SECRET_KEY if ((ret == 0) && (derivedKey != NULL)) { -#if (defined(HAVE_ECC) || !defined(NO_DH) || defined(WOLFPKCS11_HKDF)) && \ - !defined(WOLFPKCS11_LEGACY_DERIVE_NO_INHERIT) +#ifndef WOLFPKCS11_LEGACY_DERIVE_NO_INHERIT /* F-4533: read the base key's protection bits while `obj' still * refers to it, before CreateObject reuses `obj' for the new key. */ bLen = sizeof(CK_BBOOL); @@ -9829,11 +10409,37 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, &bLen) != 0) baseExtractable = CK_TRUE; #endif - rv = CreateObject(session, pTemplate, ulAttributeCount, &obj); + histLen = sizeof(CK_BBOOL); + if (WP11_Object_GetAttr(obj, CKA_ALWAYS_SENSITIVE, + &baseAlwaysSensitive, &histLen) != 0) + baseAlwaysSensitive = CK_FALSE; + histLen = sizeof(CK_BBOOL); + if (WP11_Object_GetAttr(obj, CKA_NEVER_EXTRACTABLE, + &baseNeverExtractable, &histLen) != 0) + baseNeverExtractable = CK_FALSE; + FindAttributeType(pTemplate, ulAttributeCount, CKA_VALUE_LEN, + &valueLenAttr); + if ((valueLenAttr != NULL) && (valueLenAttr->pValue != NULL) && + (valueLenAttr->ulValueLen == sizeof(CK_ULONG))) { + XMEMCPY(&reqValueLen, valueLenAttr->pValue, sizeof(CK_ULONG)); + } +#ifndef NO_DH + /* DH agreement drops leading zero bytes; restore up to the prime + * length on request. */ + if ((pMechanism->mechanism == CKM_DH_PKCS_DERIVE) && + (reqValueLen > keyLen) && (reqValueLen <= dhPrimeLen)) { + XMEMMOVE(derivedKey + (reqValueLen - keyLen), derivedKey, keyLen); + XMEMSET(derivedKey, 0, reqValueLen - keyLen); + keyLen = (word32)reqValueLen; + } +#endif + if (reqValueLen > keyLen) + rv = CKR_ATTRIBUTE_VALUE_INVALID; + if (rv == CKR_OK) + rv = CreateObject(session, pTemplate, ulAttributeCount, &obj); if (rv == CKR_OK) { /* obj now refers to the newly created derived key. */ -#if (defined(HAVE_ECC) || !defined(NO_DH) || defined(WOLFPKCS11_HKDF)) && \ - !defined(WOLFPKCS11_LEGACY_DERIVE_NO_INHERIT) +#ifndef WOLFPKCS11_LEGACY_DERIVE_NO_INHERIT /* F-4533: PKCS#11 v3.0 5.5.5 - a derived key must be at least as * protected as its base key. Force the inherited attributes after * the caller template has been applied so a weaker template @@ -9869,21 +10475,24 @@ static CK_RV wp11_C_DeriveKey(CK_SESSION_HANDLE hSession, rv = CKR_FUNCTION_FAILED; } if (ret == 0) { - rv = AddObject(session, obj, pTemplate, - ulAttributeCount, phKey); + /* Set before AddObject so a token store has them. */ + rv = SetDerivedStates(obj, baseAlwaysSensitive, + baseNeverExtractable); + if (rv == CKR_OK) { + rv = AddObject(session, obj, pTemplate, + ulAttributeCount, phKey); + } + if (rv != CKR_OK) + WP11_Object_Free(obj); } } else { WP11_Object_Free(obj); - rv = ret; + rv = CKR_FUNCTION_FAILED; } } } - if ((rv == CKR_OK) && (derivedKey != NULL)) { - rv = SetInitialStates(obj); - } - if (derivedKey != NULL) { wc_ForceZero(derivedKey, keyLen); XFREE(derivedKey, NULL, DYNAMIC_TYPE_TMP_BUFFER); diff --git a/src/internal.c b/src/internal.c index 7c61a3f6..29558b73 100644 --- a/src/internal.c +++ b/src/internal.c @@ -125,28 +125,15 @@ -/* Maximum size of storage for generated/derived DH key. */ -#ifdef WOLFPKCS11_NSS -#define WP11_MAX_DH_KEY_SZ (8192/8) -#else -#define WP11_MAX_DH_KEY_SZ (4096/8) -#endif - -/* Maximum size of storage for generated/derived symmetric key. */ -#ifdef WOLFPKCS11_NSS -#define WP11_MAX_SYM_KEY_SZ (2048) -#elif !defined(NO_DH) -#define WP11_MAX_SYM_KEY_SZ (4096/8) -#elif defined(HAVE_ECC) -#define WP11_MAX_SYM_KEY_SZ ((521+7)/8) -#else -#define WP11_MAX_SYM_KEY_SZ 64 -#endif - #ifndef WP11_MAX_CERT_SZ #define WP11_MAX_CERT_SZ 4096 #endif +/* Initial allocation when reading a variable-length array from storage. */ +#ifndef WP11_STORE_READ_CHUNK_SZ +#define WP11_STORE_READ_CHUNK_SZ 4096 +#endif + #define PKCS11_CHECK_VALUE_SIZE 3 /* Sizes for storage. */ @@ -177,7 +164,12 @@ /* Determine object id from object handle. */ #define OBJ_HANDLE_OBJ_ID(h) ((h) & 0xfffffff) -#ifdef SINGLE_THREADED +/* SINGLE_THREADED alters wolfSSL struct layouts; only wolfSSL sets it. */ +#if defined(SINGLE_THREADED) || defined(WOLFPKCS11_SINGLE_THREADED) + #define WP11_NO_LOCKING +#endif + +#ifdef WP11_NO_LOCKING /* Disable locking. */ typedef int WP11_Lock; @@ -298,6 +290,8 @@ struct WP11_Object { int keyDataLen; /* Length of encoded key data */ byte iv[GCM_NONCE_MID_SZ]; /* IV/nonce for encrypt/decrypt */ byte encoded:1; /* Key isn't in decoded form */ + byte undecoded:1; /* Loaded key data not yet decoded */ + byte unbound:1; /* Key data not bound to the policy */ #endif WP11_Session* session; /* Session object belongs to */ @@ -531,6 +525,7 @@ struct WP11_Session { byte* data; /* Held data for one-shot mechs */ word32 dataSz; /* Size of held data */ + word32 dataCap; /* Allocated size of held data */ int devId; WP11_Session* next; /* Next session for slot */ @@ -596,6 +591,9 @@ struct WP11_Slot { WP11_Object* discarded; /* Destroyed objects awaiting free */ WP11_Lock callLock; /* Lock for object call epochs */ byte reclaim:1; /* Unlinked objects await free */ +#ifndef WOLFPKCS11_NO_STORE + WP11_Lock storeLock; /* Held for a whole token store pass */ +#endif }; #if defined(HAVE_FIPS) && FIPS_VERSION_LT(6,0) @@ -656,7 +654,7 @@ static int libraryInitCount = 0; /* Lock for globals including global random. */ static WP11_Lock globalLock; -#if !defined(SINGLE_THREADED) && defined(WOLFSSL_MUTEX_INITIALIZER) && \ +#if !defined(WP11_NO_LOCKING) && defined(WOLFSSL_MUTEX_INITIALIZER) && \ defined(WOLFSSL_MUTEX_INITIALIZER_CLAUSE) /* Permanently-live mutex that serializes WP11_Library_Init, * WP11_Library_Final, and WP11_Library_IsInitialized. Needed because @@ -676,7 +674,7 @@ static wolfSSL_Mutex libraryInitLock #define WP11_HAVE_LIBRARY_INIT_LOCK #endif -#if !defined(SINGLE_THREADED) && defined(WOLFSSL_MUTEX_INITIALIZER) && \ +#if !defined(WP11_NO_LOCKING) && defined(WOLFSSL_MUTEX_INITIALIZER) && \ defined(WOLFSSL_MUTEX_INITIALIZER_CLAUSE) && \ !defined(WOLFPKCS11_TPM_STORE) && defined(WOLFPKCS11_NSS) /* Permanently-live leaf mutex serializing the module-global storeDir, which @@ -691,7 +689,7 @@ static wolfSSL_Mutex storeDirLock #endif -#ifndef SINGLE_THREADED +#ifndef WP11_NO_LOCKING /** * Initialize a lock. * @@ -963,6 +961,7 @@ static int wp11_Slot_AddSession(WP11_Slot* slot, WP11_Session** session) return ret; } +static void wp11_Session_FreeOp(WP11_Session* session); static void wp11_Slot_UnlinkObject(WP11_Slot* slot, WP11_Object* object, WP11_Object** list); static void wp11_Slot_FreeUnlinkedObjects(WP11_Slot* slot); @@ -1002,6 +1001,17 @@ static void wp11_Session_Final(WP11_Session* session) /* Finalize any find. */ WP11_Session_FindFinal(session); + wp11_Session_FreeOp(session); +} + +/** + * Release the resources owned by the session's current operation and mark no + * operation as initialized. + * + * @param session [in] Session object. + */ +static void wp11_Session_FreeOp(WP11_Session* session) +{ #if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) if (session->mechanism == CKM_RSA_PKCS_OAEP && session->params.oaep.label != NULL) { @@ -1030,8 +1040,11 @@ static void wp11_Session_Final(WP11_Session* session) session->params.gcm.aad = NULL; } if (session->params.gcm.enc != NULL) { + wc_ForceZero(session->params.gcm.enc, + (word32)session->params.gcm.encSz); XFREE(session->params.gcm.enc, NULL, DYNAMIC_TYPE_TMP_BUFFER); session->params.gcm.enc = NULL; + session->params.gcm.encSz = 0; } #ifdef WOLFSSL_AESGCM_STREAM if (session->params.gcm.streamInit) { @@ -1087,10 +1100,24 @@ static void wp11_Session_Final(WP11_Session* session) session->params.digest.hashType); session->init &= ~WP11_INIT_DIGEST_MASK; } + WP11_Session_FreeData(session); + /* Older wolfSSL versions do not zeroize the key schedule in wc_AesFree. */ + wc_ForceZero(&session->params, sizeof(session->params)); /* Ensure no stale bits remain after all cleanup. */ session->init = 0; } +/** + * Abort the active operation on the session, releasing what it owns. + * + * @param session [in] Session object. + */ +void WP11_Session_AbortOp(WP11_Session* session) +{ + if (session->init != 0) + wp11_Session_FreeOp(session); +} + #ifndef WOLFPKCS11_NO_STORE #ifndef WOLFPKCS11_CUSTOM_STORE @@ -1119,8 +1146,65 @@ typedef struct WP11_TpmStore { WOLFTPM2_DEV* dev; WOLFTPM2_NV nv; word32 offset; + byte* buf; /* write record staged until commit, follows context */ + word32 bufSz; + word32 nvIndex; + int writeFailed; } WP11_TpmStore; static WP11_TpmStore tpmStores[1]; /* maximum of 1 open store */ +static word32 tpmNvIndexMax = 0; /* TPM_PT_NV_INDEX_MAX, 0 when unknown */ +#ifdef DEBUG_WOLFPKCS11 +static int tpmStoreShortBy = 0; + +/** + * Test hook: stage every record in a buffer short by the given number of + * bytes, so a write runs past it. + * + * @param bytes [in] Bytes to leave out, or 0 to disable the hook. + */ +WP11_API void WP11_Test_TpmStoreShortBy(int bytes) +{ + tpmStoreShortBy = bytes; +} +#endif + +/** + * Read the largest NV index the TPM allows, so a record that cannot fit in + * one is refused before any NV index is touched. + * + * @return 0 on success. + * @return TPM RC or BAD_FUNC_ARG when the limit could not be read. + */ +static int wolfPKCS11_Store_TpmReadLimits(void) +{ + int rc; + GetCapability_In in; + GetCapability_Out out; + TPML_TAGGED_TPM_PROPERTY* props; + + XMEMSET(&in, 0, sizeof(in)); + XMEMSET(&out, 0, sizeof(out)); + in.capability = TPM_CAP_TPM_PROPERTIES; + in.property = TPM_PT_NV_INDEX_MAX; + in.propertyCount = 1; + rc = TPM2_GetCapability(&in, &out); + if (rc == TPM_RC_SUCCESS) { + props = &out.capabilityData.data.tpmProperties; + if (out.capabilityData.capability != TPM_CAP_TPM_PROPERTIES || + props->count < 1 || + props->tpmProperty[0].property != TPM_PT_NV_INDEX_MAX || + props->tpmProperty[0].value == 0) { + rc = BAD_FUNC_ARG; + } + else { + tpmNvIndexMax = props->tpmProperty[0].value; + } + } + if (rc != 0) + printf("TPM NV index size limit unknown: %d\n", rc); + + return rc; +} /* Internal function to get maximum size for each store type */ static int wolfPKCS11_Store_GetMaxSize(int type, int variableSz) @@ -1195,6 +1279,12 @@ static int wolfPKCS11_Store_GetMaxSize(int type, int variableSz) case WOLFPKCS11_STORE_DHKEY_PUB: case WOLFPKCS11_STORE_CERT: case WOLFPKCS11_STORE_TRUST: + case WOLFPKCS11_STORE_MLDSAKEY_PRIV: + case WOLFPKCS11_STORE_MLDSAKEY_PUB: + case WOLFPKCS11_STORE_MLKEMKEY_PRIV: + case WOLFPKCS11_STORE_MLKEMKEY_PUB: + case WOLFPKCS11_STORE_HSSKEY_PUB: + case WOLFPKCS11_STORE_XMSSKEY_PUB: maxSz = sizeof(word32) + variableSz; break; @@ -1210,9 +1300,9 @@ static int wolfPKCS11_Store_GetMaxSize(int type, int variableSz) #ifdef WOLFPKCS11_TPM_STORE static word32 wolfPKCS11_Store_Handle(int type, CK_ULONG id1, CK_ULONG id2) { - /* Build unique handle */ + /* Build unique handle. Store types run past 0x0F, so use five bits. */ word32 nvIndex = WOLFPKCS11_TPM_NV_BASE + - ((type & 0x0F) << 16) + + ((type & 0x1F) << 16) + (((word32)id1 & 0xFF) << 8) + ((word32)id2 & 0xFF); return nvIndex; @@ -1267,27 +1357,154 @@ static void wolfPKCS11_StoreAbortTemp(WP11_FileStoreCtx* ctx) } } -static int wolfPKCS11_StoreCommitTemp(WP11_FileStoreCtx* ctx) +#if defined(DEBUG_WOLFPKCS11) && !defined(_WIN32) && !defined(_MSC_VER) +static int wp11_storeExitAfter = 0; +static int wp11_storeCloseFailAfter = -1; + +/** + * Test hook: end the process right after the given number of store renames, + * standing in for a crash part way through a store pass. + * + * @param renames [in] Renames to allow, or 0 to disable the hook. + */ +WP11_API void WP11_Test_StoreExitAfterRenames(int renames) +{ + wp11_storeExitAfter = renames; +} + +/** + * Test hook: make the close of a written store record fail once, after the + * given number of closes succeed. + * + * @param closes [in] Closes to allow first, or -1 to disable the hook. + */ +WP11_API void WP11_Test_StoreCloseFailAfter(int closes) +{ + wp11_storeCloseFailAfter = closes; +} +#endif + +static int wolfPKCS11_StoreRename(const char* from, const char* to) { int ret = 0; - if (ctx == NULL || ctx->has_temp == 0) { - return 0; +#if defined(_WIN32) || defined(_MSC_VER) + if (!MoveFileExA(from, to, + MOVEFILE_REPLACE_EXISTING | MOVEFILE_WRITE_THROUGH)) { + ret = READ_ONLY_E; + } +#else + if (rename(from, to) != 0) { + ret = READ_ONLY_E; } +#endif +#if defined(DEBUG_WOLFPKCS11) && !defined(_WIN32) && !defined(_MSC_VER) + if (ret == 0 && wp11_storeExitAfter > 0 && --wp11_storeExitAfter == 0) + _exit(WP11_TEST_STORE_EXIT_CODE); +#endif + + return ret; +} +/** + * Flush a written store file to stable storage. + * + * @param [in] file Open file written by the store. + * @return 0 on success. + * @return READ_ONLY_E when the data may not have reached the device. + */ +static int wolfPKCS11_StoreSyncFile(XFILE file) +{ + int ret = 0; #if defined(_WIN32) || defined(_MSC_VER) - if (!MoveFileExA(ctx->temp_name, ctx->final_name, - MOVEFILE_REPLACE_EXISTING | MOVEFILE_WRITE_THROUGH)) { + intptr_t osHandle; +#endif + + if (XFFLUSH(file) != 0) ret = READ_ONLY_E; +#if defined(_WIN32) || defined(_MSC_VER) + if (ret == 0) { + osHandle = _get_osfhandle(_fileno(file)); + if (!FlushFileBuffers((HANDLE)osHandle)) + ret = READ_ONLY_E; } #else - if (rename(ctx->temp_name, ctx->final_name) != 0) { + if (ret == 0 && fsync(fileno(file)) != 0) + ret = READ_ONLY_E; +#endif + + return ret; +} + +/** + * Make the renames done in a store directory durable. Windows has no + * directory flush; its renames are done write-through instead. + * + * @param [in] dir Store directory. + * @return 0 on success. + * @return READ_ONLY_E when the directory could not be flushed. + */ +static int wolfPKCS11_StoreSyncDir(const char* dir) +{ + int ret = 0; +#if !defined(_WIN32) && !defined(_MSC_VER) + int fd; + + fd = open(dir, O_RDONLY); + if (fd < 0) { ret = READ_ONLY_E; } + else { + /* Filesystems without directory flushing answer EINVAL. */ + if (fsync(fd) != 0 && errno != EINVAL) + ret = READ_ONLY_E; + close(fd); + } +#else + (void)dir; #endif + return ret; +} + +/** + * Make a rename of a store file durable by flushing its directory. + * + * @param [in] name Renamed store file. + * @return 0 on success. + * @return READ_ONLY_E when the directory could not be flushed. + */ +static int wolfPKCS11_StoreSyncParent(const char* name) +{ + size_t len = XSTRLEN(name); + size_t i; + size_t sep = 0; + char dir[WP11_STORE_MAX_PATH]; + + for (i = 0; i < len; i++) { + if (name[i] == '/' || name[i] == '\\') + sep = i; + } + if (sep == 0 || sep >= sizeof(dir)) + return READ_ONLY_E; + XMEMCPY(dir, name, sep); + dir[sep] = '\0'; + + return wolfPKCS11_StoreSyncDir(dir); +} + +static int wolfPKCS11_StoreCommitTemp(WP11_FileStoreCtx* ctx) +{ + int ret = 0; + + if (ctx == NULL || ctx->has_temp == 0) { + return 0; + } + + ret = wolfPKCS11_StoreRename(ctx->temp_name, ctx->final_name); if (ret == 0) { ctx->has_temp = 0; + ret = wolfPKCS11_StoreSyncParent(ctx->final_name); } return ret; @@ -1432,14 +1649,14 @@ static int wolfPKCS11_Store_Name(int type, CK_ULONG id1, CK_ULONG id2, char* nam int nameLen) { int ret = 0; -#ifndef WOLFPKCS11_NO_ENV const char* str = NULL; -#endif /* Reserve enough space in the final filename for suffixes such as * "/wp11_rsakey_priv_%016lx_%016lx" (47 chars worst-case). */ enum { WP11_STORE_SUFFIX_RESERVE = 48 }; +#ifndef WOLFPKCS11_NO_ENV char homePath[256]; +#endif #ifdef WP11_HAVE_STORE_DIR_LOCK char storeDirCopy[WP11_STORE_MAX_PATH]; #endif @@ -1487,6 +1704,7 @@ static int wolfPKCS11_Store_Name(int type, CK_ULONG id1, CK_ULONG id2, char* nam } #endif +#ifndef WOLFPKCS11_NO_ENV if (str == NULL) { const char* homeDir = NULL; @@ -1510,6 +1728,7 @@ static int wolfPKCS11_Store_Name(int type, CK_ULONG id1, CK_ULONG id2, char* nam } #endif } +#endif /* 47 is maximum number of character to a filename and path separator. */ if (str == NULL) { @@ -1609,6 +1828,541 @@ static int wolfPKCS11_Store_Name(int type, CK_ULONG id1, CK_ULONG id2, char* nam return ret; } + +/* One token store pass commits as a unit: staged records, then a marker. */ +#define WP11_STORE_BATCH +#define WP11_STORE_STAGED_MAX (WP11_STORE_MAX_PATH + 8) + +#define WP11_BATCH_DISCARD 0 +#define WP11_BATCH_APPLY 1 +#define WP11_BATCH_CHECK 2 + +typedef struct WP11_StoreBatch { + WP11_Slot* slot; + XFILE intent; +#if defined(_WIN32) || defined(_MSC_VER) + HANDLE lockFile; +#else + int lockFd; +#endif + int locked; + int active; + int failed; + size_t dirLen; + char dir[WP11_STORE_MAX_PATH]; + char intentName[WP11_STORE_STAGED_MAX]; + char commitName[WP11_STORE_STAGED_MAX]; + char lockName[WP11_STORE_STAGED_MAX]; +} WP11_StoreBatch; + +static WP11_StoreBatch storeBatch; + +/** + * Work out the store directory and the batch file names for a token. + * + * @param [in] tokenId Id of token. + * @return 0 on success. + * @return READ_ONLY_E when the names cannot be built. + */ +static int wolfPKCS11_StoreBatchNames(int tokenId) +{ + int ret = 0; + int len; + int i; + int sep = -1; + + len = wolfPKCS11_Store_Name(WOLFPKCS11_STORE_TOKEN, (CK_ULONG)tokenId, 0, + storeBatch.dir, sizeof(storeBatch.dir)); + if (len <= 0 || len >= (int)sizeof(storeBatch.dir)) + ret = READ_ONLY_E; + for (i = 0; ret == 0 && i < len; i++) { + if (storeBatch.dir[i] == '/' || storeBatch.dir[i] == '\\') + sep = i; + } + if (ret == 0 && sep <= 0) + ret = READ_ONLY_E; + if (ret == 0) { + storeBatch.dir[sep] = '\0'; + storeBatch.dirLen = (size_t)sep; + len = XSNPRINTF(storeBatch.intentName, sizeof(storeBatch.intentName), + "%s/wp11_txn_%016lx_intent", storeBatch.dir, + (unsigned long)tokenId); + if (len <= 0 || len >= (int)sizeof(storeBatch.intentName)) + ret = READ_ONLY_E; + } + if (ret == 0) { + len = XSNPRINTF(storeBatch.commitName, sizeof(storeBatch.commitName), + "%s/wp11_txn_%016lx_commit", storeBatch.dir, + (unsigned long)tokenId); + if (len <= 0 || len >= (int)sizeof(storeBatch.commitName)) + ret = READ_ONLY_E; + } + if (ret == 0) { + len = XSNPRINTF(storeBatch.lockName, sizeof(storeBatch.lockName), + "%s/wp11_txn_%016lx_lock", storeBatch.dir, + (unsigned long)tokenId); + if (len <= 0 || len >= (int)sizeof(storeBatch.lockName)) + ret = READ_ONLY_E; + } + + return ret; +} + +/** + * Take or release the lock file that keeps other processes out of the batch + * files while a pass or a recovery runs. + * + * @param [in] lock 1 to take the lock, 0 to release it. + * @return 0 on success. + * @return READ_ONLY_E when the lock cannot be taken. + */ +static int wolfPKCS11_StoreBatchLockFile(int lock) +{ + int ret = 0; +#if defined(_WIN32) || defined(_MSC_VER) + OVERLAPPED ov; + + XMEMSET(&ov, 0, sizeof(ov)); + if (lock) { + storeBatch.lockFile = CreateFileA(storeBatch.lockName, + GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, + NULL, OPEN_ALWAYS, + FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OPEN_REPARSE_POINT, NULL); + if (storeBatch.lockFile == INVALID_HANDLE_VALUE) + ret = READ_ONLY_E; + if (ret == 0 && !LockFileEx(storeBatch.lockFile, + LOCKFILE_EXCLUSIVE_LOCK, 0, 1, 0, &ov)) { + CloseHandle(storeBatch.lockFile); + ret = READ_ONLY_E; + } + storeBatch.locked = (ret == 0); + } + else if (storeBatch.locked) { + (void)UnlockFileEx(storeBatch.lockFile, 0, 1, 0, &ov); + CloseHandle(storeBatch.lockFile); + storeBatch.locked = 0; + } +#else + struct flock fl; + + XMEMSET(&fl, 0, sizeof(fl)); + fl.l_whence = SEEK_SET; + if (lock) { + storeBatch.lockFd = open(storeBatch.lockName, + O_RDWR | O_CREAT | O_NOFOLLOW, S_IRUSR | S_IWUSR); + if (storeBatch.lockFd < 0) + ret = READ_ONLY_E; + fl.l_type = F_WRLCK; + while (ret == 0 && fcntl(storeBatch.lockFd, F_SETLKW, &fl) != 0) { + if (errno != EINTR) { + close(storeBatch.lockFd); + ret = READ_ONLY_E; + } + } + storeBatch.locked = (ret == 0); + } + else if (storeBatch.locked) { + fl.l_type = F_UNLCK; + (void)fcntl(storeBatch.lockFd, F_SETLK, &fl); + close(storeBatch.lockFd); + storeBatch.locked = 0; + } +#endif + + return ret; +} + +/** + * Check that a listed name is the basename of a store file. Lists hold + * basenames so a recovery does not depend on how the directory was spelled. + * + * @param [in] name Name read from a batch list. + * @return 1 when acceptable, 0 otherwise. + */ +static int wolfPKCS11_StoreBatchNameOk(const char* name) +{ + size_t len = XSTRLEN(name); + size_t i; + int ok; + + ok = len > 5 && XSTRNCMP(name, "wp11_", 5) == 0 && + storeBatch.dirLen + 1 + len < WP11_STORE_MAX_PATH; + for (i = 0; ok && i < len; i++) { + if (name[i] == '/' || name[i] == '\\') + ok = 0; + } + + return ok; +} + +/** + * Check that a final record name can be replaced by a rename. + * + * @param [in] name Final record name. + * @return 1 when absent or a regular file, 0 otherwise. + */ +static int wolfPKCS11_StoreBatchReplaceable(const char* name) +{ +#if defined(_WIN32) || defined(_MSC_VER) + DWORD attrs = GetFileAttributesA(name); + + return attrs == INVALID_FILE_ATTRIBUTES || + (attrs & (FILE_ATTRIBUTE_DIRECTORY | + FILE_ATTRIBUTE_REPARSE_POINT)) == 0; +#else + struct stat st; + + if (lstat(name, &st) != 0) + return errno == ENOENT; + return S_ISREG(st.st_mode); +#endif +} + +/** + * Act on each staged record a batch list names: remove it, rename it into + * place, or check that it can be renamed into place. + * + * @param [in] listName Intent or commit file. + * @param [in] mode WP11_BATCH_DISCARD, WP11_BATCH_APPLY or + * WP11_BATCH_CHECK. + * @return 0 on success. + * @return READ_ONLY_E when the list is unreadable or malformed, or a record + * cannot be put in place. Discarding never fails. + */ +static int wolfPKCS11_StoreBatchApply(const char* listName, int mode) +{ + int ret = 0; + XFILE list; + XFILE staged; + size_t len; + size_t i; + const char* base; + char line[WP11_STORE_MAX_PATH + 2]; + char finalName[WP11_STORE_MAX_PATH]; + char stagedName[WP11_STORE_STAGED_MAX]; + + list = XFOPEN(listName, "rb"); + if (list == XBADFILE) + return (mode == WP11_BATCH_DISCARD) ? 0 : READ_ONLY_E; + while (fgets(line, sizeof(line), list) != NULL) { + len = XSTRLEN(line); + /* The intent can end in a line a crash cut short; a marker cannot. */ + if (len == 0 || line[len - 1] != '\n') { + if (mode != WP11_BATCH_DISCARD) + ret = READ_ONLY_E; + continue; + } + line[len - 1] = '\0'; + /* A list written with full paths is read by its basenames too. */ + base = line; + for (i = 0; i + 1 < len; i++) { + if (line[i] == '/' || line[i] == '\\') + base = line + i + 1; + } + if (!wolfPKCS11_StoreBatchNameOk(base)) { + if (mode != WP11_BATCH_DISCARD) + ret = READ_ONLY_E; + continue; + } + len = (size_t)XSNPRINTF(finalName, sizeof(finalName), "%s/%s", + storeBatch.dir, base); + if (len < sizeof(finalName)) { + len = (size_t)XSNPRINTF(stagedName, sizeof(stagedName), "%s.new", + finalName); + } + if (len >= sizeof(finalName) || len >= sizeof(stagedName)) { + if (mode != WP11_BATCH_DISCARD) + ret = READ_ONLY_E; + continue; + } + if (mode == WP11_BATCH_DISCARD) { + (void)remove(stagedName); + } + else if (mode == WP11_BATCH_CHECK) { + if (!wolfPKCS11_StoreBatchReplaceable(finalName)) + ret = READ_ONLY_E; + } + else if (wolfPKCS11_StoreRename(stagedName, finalName) != 0) { + /* Already renamed by an earlier, interrupted roll forward. */ + staged = XFOPEN(stagedName, "rb"); + if (staged != XBADFILE) { + XFCLOSE(staged); + ret = READ_ONLY_E; + } + } + } + if (ferror(list) && mode != WP11_BATCH_DISCARD) + ret = READ_ONLY_E; + XFCLOSE(list); + + return ret; +} + +/** + * Finish a store pass that reached its commit point, or drop one that did + * not. Caller holds the batch locks and has set the batch names. + * + * @return 0 on success. + * @return READ_ONLY_E when a committed pass could not be finished; the + * marker is kept so a later attempt can finish it. + */ +static int wolfPKCS11_StoreBatchRecover(void) +{ + int ret = 0; + XFILE marker; + + marker = XFOPEN(storeBatch.commitName, "rb"); + if (marker != XBADFILE) { + XFCLOSE(marker); + ret = wolfPKCS11_StoreBatchApply(storeBatch.commitName, + WP11_BATCH_APPLY); + /* The marker outlives the records it names until they are durable. */ + if (ret == 0) + ret = wolfPKCS11_StoreSyncDir(storeBatch.dir); + if (ret == 0) + (void)remove(storeBatch.commitName); + } + else { + (void)wolfPKCS11_StoreBatchApply(storeBatch.intentName, + WP11_BATCH_DISCARD); + (void)remove(storeBatch.intentName); + } + + return ret; +} + +/** + * Bring the store for a token to a committed state before it is read. On + * return the batch locks are held, so the token and object records read + * until wolfPKCS11_StoreBatchRelease belong to one committed pass. + * + * @param [in] tokenId Id of token. + * @param [out] held Slot whose batch locks are held, or NULL. + * @return 0 on success. + * @return Other value when an interrupted pass could not be finished. + */ +static int wolfPKCS11_StoreBatchCheck(int tokenId, WP11_Slot** held) +{ + int ret = 0; + XFILE marker; + WP11_Slot* slot; + + *held = NULL; +#ifndef WOLFPKCS11_NO_ENV + /* Storage is turned off, so nothing is read and nothing is finished. */ + if (XGETENV("WOLFPKCS11_NO_STORE") != NULL) + return 0; +#endif + if (WP11_Slot_Get((CK_SLOT_ID)tokenId, &slot) != 0) + return 0; + WP11_Lock_LockRW(&slot->storeLock); + *held = slot; + if (wolfPKCS11_StoreBatchNames(tokenId) == 0) { + if (wolfPKCS11_StoreBatchLockFile(1) == 0) { + ret = wolfPKCS11_StoreBatchRecover(); + } + else { + /* A committed pass that cannot be finished must not be read. */ + marker = XFOPEN(storeBatch.commitName, "rb"); + if (marker != XBADFILE) { + XFCLOSE(marker); + ret = READ_ONLY_E; + } + } + } + + return ret; +} + +/** + * Release the batch locks held since wolfPKCS11_StoreBatchCheck. + * + * @param [in] held Slot returned by wolfPKCS11_StoreBatchCheck. + */ +static void wolfPKCS11_StoreBatchRelease(WP11_Slot* held) +{ + if (held != NULL) { + (void)wolfPKCS11_StoreBatchLockFile(0); + WP11_Lock_UnlockRW(&held->storeLock); + } +} + +/** + * Start a store pass: records closed from now on are staged, not committed. + * On success the batch locks are held until wolfPKCS11_StoreBatchEnd. When + * the pass cannot start, nothing is written and the error is returned. + * + * @param [in] tokenId Id of token being stored. + * @return 0 on success. + * @return Other value when an interrupted pass could not be finished. + */ +static int wolfPKCS11_StoreBatchBegin(int tokenId) +{ + int ret = 0; +#ifndef WOLFPKCS11_NO_ENV + const char* str; +#endif +#if defined(_WIN32) || defined(_MSC_VER) + HANDLE h; +#endif + int fd; + WP11_Slot* slot; + +#ifndef WOLFPKCS11_NO_ENV + /* Storage is turned off, so the pass writes nothing. */ + str = XGETENV("WOLFPKCS11_NO_STORE"); + if (str != NULL) + return 0; +#endif + if (WP11_Slot_Get((CK_SLOT_ID)tokenId, &slot) != 0) + return BAD_FUNC_ARG; + /* The batch state is only written while the store lock is held. */ + WP11_Lock_LockRW(&slot->storeLock); + storeBatch.slot = slot; + ret = wolfPKCS11_StoreBatchNames(tokenId); + if (ret == 0) + ret = wolfPKCS11_StoreEnsureDir(storeBatch.dir); + if (ret == 0) + ret = wolfPKCS11_StoreBatchLockFile(1); + if (ret == 0) + ret = wolfPKCS11_StoreBatchRecover(); + if (ret == 0) { + #if defined(_WIN32) || defined(_MSC_VER) + /* Create afresh without following a link planted at the name. */ + (void)DeleteFileA(storeBatch.intentName); + h = CreateFileA(storeBatch.intentName, GENERIC_WRITE, 0, NULL, + CREATE_NEW, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OPEN_REPARSE_POINT, + NULL); + fd = (h != INVALID_HANDLE_VALUE) ? + _open_osfhandle((intptr_t)h, _O_WRONLY | _O_BINARY) : -1; + if (h != INVALID_HANDLE_VALUE && fd < 0) + CloseHandle(h); + storeBatch.intent = (fd >= 0) ? _fdopen(fd, "wb") : XBADFILE; + if (fd >= 0 && storeBatch.intent == XBADFILE) + _close(fd); + #else + fd = open(storeBatch.intentName, + O_WRONLY | O_CREAT | O_TRUNC | O_NOFOLLOW, S_IRUSR | S_IWUSR); + storeBatch.intent = (fd >= 0) ? fdopen(fd, "wb") : XBADFILE; + if (fd >= 0 && storeBatch.intent == XBADFILE) + close(fd); + #endif + if (storeBatch.intent == XBADFILE) + ret = READ_ONLY_E; + } + if (ret == 0) { + storeBatch.active = 1; + storeBatch.failed = 0; + } + else { + (void)wolfPKCS11_StoreBatchLockFile(0); + storeBatch.slot = NULL; + WP11_Lock_UnlockRW(&slot->storeLock); + } + + return ret; +} + +/** + * Stage a written record as ".new" and list its basename in the intent + * file. The record must be in the store directory of the pass. + * + * @param [in, out] ctx File store context with a written temporary file. + * @return 0 on success. + * @return READ_ONLY_E when the record could not be staged. + */ +static int wolfPKCS11_StoreStageTemp(WP11_FileStoreCtx* ctx) +{ + int ret = 0; + int len; + const char* base; + char stagedName[WP11_STORE_STAGED_MAX]; + + base = ctx->final_name + storeBatch.dirLen + 1; + if (XSTRNCMP(ctx->final_name, storeBatch.dir, storeBatch.dirLen) != 0 || + ctx->final_name[storeBatch.dirLen] != '/' || + !wolfPKCS11_StoreBatchNameOk(base)) { + ret = READ_ONLY_E; + } + if (ret == 0) { + len = XSNPRINTF(stagedName, sizeof(stagedName), "%s.new", + ctx->final_name); + if (len <= 0 || len >= (int)sizeof(stagedName)) + ret = READ_ONLY_E; + } + if (ret == 0) + ret = wolfPKCS11_StoreRename(ctx->temp_name, stagedName); + if (ret == 0) { + ctx->has_temp = 0; + if (fputs(base, storeBatch.intent) < 0 || + fputc('\n', storeBatch.intent) == EOF || + fflush(storeBatch.intent) != 0) { + ret = READ_ONLY_E; + } + } + + return ret; +} + +/** + * End a store pass. When committing, the intent becomes the commit marker and + * the staged records are renamed into place; otherwise they are removed. + * Once the marker exists the pass is committed and reported as such, and any + * record left unrenamed is finished by the next store or load. + * Releases the batch locks taken by wolfPKCS11_StoreBatchBegin. + * + * @param [in] commit 1 when every record of the pass was written. + * @return 0 when the pass is committed or there was nothing to commit. + * @return READ_ONLY_E when the pass was not committed. + */ +static int wolfPKCS11_StoreBatchEnd(int commit) +{ + int ret = 0; + WP11_Slot* slot; + + if (storeBatch.active) { + storeBatch.active = 0; + if (commit && wolfPKCS11_StoreSyncFile(storeBatch.intent) != 0) + storeBatch.failed = 1; + if (XFCLOSE(storeBatch.intent) != 0) + storeBatch.failed = 1; + storeBatch.intent = XBADFILE; + if (storeBatch.failed) + ret = READ_ONLY_E; + if (ret == 0 && commit) { + ret = wolfPKCS11_StoreBatchApply(storeBatch.intentName, + WP11_BATCH_CHECK); + } + /* The staged records must be durable before the marker can be. */ + if (ret == 0 && commit) + ret = wolfPKCS11_StoreSyncDir(storeBatch.dir); + if (ret == 0 && commit) { + ret = wolfPKCS11_StoreRename(storeBatch.intentName, + storeBatch.commitName); + } + if (ret == 0 && commit) { + (void)wolfPKCS11_StoreSyncDir(storeBatch.dir); + if (wolfPKCS11_StoreBatchApply(storeBatch.commitName, + WP11_BATCH_APPLY) == 0 && + wolfPKCS11_StoreSyncDir(storeBatch.dir) == 0) { + (void)remove(storeBatch.commitName); + } + } + else { + (void)wolfPKCS11_StoreBatchApply(storeBatch.intentName, + WP11_BATCH_DISCARD); + (void)remove(storeBatch.intentName); + } + (void)wolfPKCS11_StoreBatchLockFile(0); + } + slot = storeBatch.slot; + if (slot != NULL) { + storeBatch.slot = NULL; + WP11_Lock_UnlockRW(&slot->storeLock); + } + + return ret; +} #endif int wolfPKCS11_Store_Remove(int type, CK_ULONG id1, CK_ULONG id2) @@ -1643,15 +2397,25 @@ int wolfPKCS11_Store_Remove(int type, CK_ULONG id1, CK_ULONG id2) XMEMSET(&parent, 0, sizeof(parent)); parent.hndl = WOLFPKCS11_TPM_AUTH_TYPE; ret = wolfTPM2_NVDeleteAuth(tpmStore->dev, &parent, nvIndex); - if (ret != 0) { + /* TPM_RC_HANDLE carries the handle number: nothing is stored there. */ + if (ret > 0 && (ret & (RC_FMT1 | 0x3F)) == TPM_RC_HANDLE) { + ret = 0; + } + else if (ret != 0) { printf("Error %d (%s) removing NV handle 0x%x: \n", ret, wolfTPM2_GetRCString(ret), nvIndex); + /* Callers take a negative value as failure, not a TPM RC. */ + ret = WC_HW_E; } #else /* remove file */ ret = wolfPKCS11_Store_Name(type, id1, id2, name, sizeof(name)); if (ret > 0 && ret < (int)sizeof(name)) { ret = remove(name); + if (ret < 0 && errno == ENOENT) { + /* Nothing was stored, so nothing is left behind. */ + ret = 0; + } if (ret < 0) { printf("remove(%s) failed: %d\n", name, ret); } @@ -1684,9 +2448,7 @@ int wolfPKCS11_Store_OpenSz(int type, CK_ULONG id1, CK_ULONG id2, int read, WP11_Slot* slot = &slotList[0]; WP11_TpmStore* tpmStore = &tpmStores[0]; word32 nvIndex; - word32 nvAttributes; int maxSz; - WOLFTPM2_HANDLE parent; #else WP11_FileStoreCtx* ctx = NULL; char name[WP11_STORE_MAX_PATH] = "\0"; @@ -1705,7 +2467,6 @@ int wolfPKCS11_Store_OpenSz(int type, CK_ULONG id1, CK_ULONG id2, int read, #endif #ifdef WOLFPKCS11_TPM_STORE - XMEMSET(&parent, 0, sizeof(parent)); XMEMSET(tpmStore, 0, sizeof(*tpmStore)); tpmStore->dev = &slot->tpmDev; @@ -1716,39 +2477,36 @@ int wolfPKCS11_Store_OpenSz(int type, CK_ULONG id1, CK_ULONG id2, int read, if (maxSz <= 0) { ret = NOT_AVAILABLE_E; } - if (ret == 0) { - TPMS_NV_PUBLIC nvPublic; - XMEMSET(&nvPublic, 0, sizeof(nvPublic)); - - /* Get NV attributes */ - parent.hndl = WOLFPKCS11_TPM_AUTH_TYPE; - (void)wolfTPM2_GetNvAttributesTemplate(parent.hndl, &nvAttributes); - - /* If write check if handle is large enough */ - if (!read) { - ret = wolfTPM2_NVReadPublic(tpmStore->dev, nvIndex, &nvPublic); - if (ret == 0 && nvPublic.dataSize < maxSz) { - /* NV is not large enough, delete and re-create */ - printf("Expanding TPM NV Handle 0x%x: %d -> %d\n", - nvIndex, nvPublic.dataSize, maxSz); - - ret = wolfTPM2_NVDeleteAuth(tpmStore->dev, &parent, nvIndex); - if (ret != 0) { - printf("Error %d (%s) removing NV handle 0x%x: \n", - ret, wolfTPM2_GetRCString(ret), nvIndex); - } - } - } + if (ret == 0 && read) { /* Try and open handle */ ret = wolfTPM2_NVOpen(tpmStore->dev, &tpmStore->nv, nvIndex, NULL, 0); if (ret != 0) { - if (!read) { - ret = wolfTPM2_NVCreateAuth(tpmStore->dev, &parent, - &tpmStore->nv, nvIndex, nvAttributes, maxSz, NULL, 0); - } - else { - ret = NOT_AVAILABLE_E; /* read for handle that doesn't exist */ - } + ret = NOT_AVAILABLE_E; /* read for handle that doesn't exist */ + } + } + else if (ret == 0 && tpmNvIndexMax != 0 && (word32)maxSz > tpmNvIndexMax) { + /* Too big for one NV index: refuse it before anything is written. */ + printf("Store record of %d bytes exceeds the TPM NV index limit of " + "%u\n", maxSz, (unsigned int)tpmNvIndexMax); + ret = BUFFER_E; + } + else if (ret == 0) { + #ifdef DEBUG_WOLFPKCS11 + if (tpmStoreShortBy > 0 && maxSz > tpmStoreShortBy) + maxSz -= tpmStoreShortBy; + #endif + /* Stage the record in its own context, NV is only touched on commit */ + tpmStore = (WP11_TpmStore*)XMALLOC(sizeof(*tpmStore) + (size_t)maxSz, + NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (tpmStore == NULL) { + ret = MEMORY_E; + } + else { + XMEMSET(tpmStore, 0, sizeof(*tpmStore)); + tpmStore->dev = &slot->tpmDev; + tpmStore->buf = (byte*)(tpmStore + 1); + tpmStore->bufSz = (word32)maxSz; + tpmStore->nvIndex = nvIndex; } } if (ret == 0) { @@ -1878,16 +2636,102 @@ int wolfPKCS11_Store_Open(int type, CK_ULONG id1, CK_ULONG id2, int read, return wolfPKCS11_Store_OpenSz(type, id1, id2, read, 0, store); } +#ifdef WOLFPKCS11_TPM_STORE +/** + * Writes the staged record to its NV index, growing the index when needed. + * + * @param [in] tpmStore TPM store context holding the staged record. + * @return 0 on success. + * @return WC_HW_E when the TPM reports an error. + * @return Other negative value when the record could not be written. + */ +static int wolfPKCS11_Store_TpmCommit(WP11_TpmStore* tpmStore) +{ + int ret; + int create = 0; + word32 nvAttributes = 0; + word32 readSz; + WOLFTPM2_HANDLE parent; + TPMS_NV_PUBLIC nvPublic; + + XMEMSET(&parent, 0, sizeof(parent)); + XMEMSET(&nvPublic, 0, sizeof(nvPublic)); + parent.hndl = WOLFPKCS11_TPM_AUTH_TYPE; + + ret = wolfTPM2_GetNvAttributesTemplate(parent.hndl, &nvAttributes); + if (ret == 0 && wolfTPM2_NVReadPublic(tpmStore->dev, tpmStore->nvIndex, + &nvPublic) != 0) { + /* No NV index holds this record yet. */ + create = 1; + } + else if (ret == 0 && nvPublic.dataSize < tpmStore->bufSz) { + /* NV is not large enough, delete and re-create */ + printf("Expanding TPM NV Handle 0x%x: %d -> %d\n", + tpmStore->nvIndex, (int)nvPublic.dataSize, (int)tpmStore->bufSz); + create = 1; + if (tpmStore->offset == 0) { + /* Nothing staged, so carry the stored record into the new index */ + ret = wolfTPM2_NVOpen(tpmStore->dev, &tpmStore->nv, + tpmStore->nvIndex, NULL, 0); + if (ret == 0) { + ret = wolfTPM2_SetAuthHandle(tpmStore->dev, 0, + &tpmStore->nv.handle); + } + if (ret == 0) { + readSz = nvPublic.dataSize; + ret = wolfTPM2_NVReadAuth(tpmStore->dev, &tpmStore->nv, + tpmStore->nvIndex, tpmStore->buf, &readSz, 0); + if (ret == 0) + tpmStore->offset = readSz; + } + if (ret == TPM_RC_NV_UNINITIALIZED) + ret = 0; + } + if (ret == 0) { + ret = wolfTPM2_NVDeleteAuth(tpmStore->dev, &parent, + tpmStore->nvIndex); + if (ret != 0) { + printf("Error %d (%s) removing NV handle 0x%x: \n", + ret, wolfTPM2_GetRCString(ret), tpmStore->nvIndex); + } + } + } + + if (ret == 0 && create) { + ret = wolfTPM2_NVCreateAuth(tpmStore->dev, &parent, &tpmStore->nv, + tpmStore->nvIndex, nvAttributes, tpmStore->bufSz, NULL, 0); + } + else if (ret == 0 && tpmStore->offset > 0) { + ret = wolfTPM2_NVOpen(tpmStore->dev, &tpmStore->nv, tpmStore->nvIndex, + NULL, 0); + } + if (ret == 0 && tpmStore->offset > 0) { + /* wolfTPM splits this into as few maximum sized NV writes as fit */ + ret = wolfTPM2_NVWriteAuth(tpmStore->dev, &tpmStore->nv, + tpmStore->nv.handle.hndl, tpmStore->buf, tpmStore->offset, 0); + } + /* Callers only treat a negative value as a failure. */ + if (ret > 0) + ret = WC_HW_E; + return ret; +} +#endif + /** * Closes access to location being read or written. * Any dynamic memory associated with the store is freed here. * - * @param [in] store Context for operation. + * @param [in] store Context for operation. + * @param [in] commit 1 to commit written data and 0 to discard it. + * @return 0 on success. + * @return Other value when written data could not be committed. */ -void wolfPKCS11_Store_Close(void* store) +static int wolfPKCS11_Store_CloseCommit(void* store, int commit) { + int ret = 0; #ifdef WOLFPKCS11_TPM_STORE WP11_TpmStore* tpmStore = (WP11_TpmStore*)store; + size_t ctxSz; #else WP11_FileStoreCtx* ctx = (WP11_FileStoreCtx*)store; #endif @@ -1897,20 +2741,54 @@ void wolfPKCS11_Store_Close(void* store) #endif #ifdef WOLFPKCS11_TPM_STORE - /* nothing to do for TPM */ - (void)tpmStore; + if (tpmStore != NULL && tpmStore->buf != NULL) { + if (tpmStore->writeFailed) { + ret = BUFFER_E; + } + else if (commit) { + ret = wolfPKCS11_Store_TpmCommit(tpmStore); + } + ctxSz = sizeof(*tpmStore) + tpmStore->bufSz; + wc_ForceZero(tpmStore, ctxSz); + XFREE(tpmStore, NULL, DYNAMIC_TYPE_TMP_BUFFER); + } #else if (ctx != NULL) { int commitRet = 0; if (ctx->file != XBADFILE && ctx->file != NULL) { - XFCLOSE(ctx->file); + if (ctx->is_write && commit && + wolfPKCS11_StoreSyncFile(ctx->file) != 0) { + commit = 0; + ret = BUFFER_E; + } + #if defined(DEBUG_WOLFPKCS11) && !defined(_WIN32) && !defined(_MSC_VER) + if (ctx->is_write && wp11_storeCloseFailAfter >= 0 && + wp11_storeCloseFailAfter-- == 0) { + /* Test hook: with the descriptor gone the close fails. */ + (void)close(fileno(ctx->file)); + } + #endif + if (XFCLOSE(ctx->file) != 0 && ctx->is_write) { + commit = 0; + ret = BUFFER_E; + } ctx->file = XBADFILE; } - if (ctx->is_write && ctx->has_temp) { - commitRet = wolfPKCS11_StoreCommitTemp(ctx); + if (ctx->is_write && ctx->has_temp && commit) { + #ifdef WP11_STORE_BATCH + if (storeBatch.active) + commitRet = wolfPKCS11_StoreStageTemp(ctx); + else + #endif + commitRet = wolfPKCS11_StoreCommitTemp(ctx); + ret = commitRet; if (commitRet != 0) { + #ifdef WP11_STORE_BATCH + if (storeBatch.active) + storeBatch.failed = 1; + #endif wolfPKCS11_StoreAbortTemp(ctx); #ifdef WOLFPKCS11_DEBUG_STORE printf("Store commit failed for %s (ret %d)\n", @@ -1926,6 +2804,18 @@ void wolfPKCS11_Store_Close(void* store) XFREE(ctx, NULL, DYNAMIC_TYPE_TMP_BUFFER); } #endif + return ret; +} + +/** + * Closes access to location being read or written. + * Any dynamic memory associated with the store is freed here. + * + * @param [in] store Context for operation. + */ +void wolfPKCS11_Store_Close(void* store) +{ + (void)wolfPKCS11_Store_CloseCommit(store, 1); } /** @@ -1979,6 +2869,36 @@ int wolfPKCS11_Store_Read(void* store, unsigned char* buffer, int len) return ret; } +#ifndef WOLFPKCS11_TPM_STORE +/** + * Get the number of bytes left to read from a store file. + * + * @param [in] store Context for operation. + * @param [out] left Number of bytes after the read position. + * @return 0 on success. + * @return BUFFER_E when the position cannot be worked out or restored. + */ +static int wolfPKCS11_Store_Left(void* store, long* left) +{ + WP11_FileStoreCtx* ctx = (WP11_FileStoreCtx*)store; + long pos = -1; + long end = -1; + + /* stdio directly: the file store's handles are FILE* (see fdopen()). */ + if (ctx != NULL && ctx->file != XBADFILE && ctx->file != NULL) { + pos = ftell(ctx->file); + if (pos >= 0 && fseek(ctx->file, 0, SEEK_END) == 0) + end = ftell(ctx->file); + if (pos >= 0 && fseek(ctx->file, pos, SEEK_SET) != 0) + end = -1; + } + if (pos < 0 || end < pos) + return BUFFER_E; + *left = end - pos; + return 0; +} +#endif + /** * Writes a specific number of bytes from buffer. * @@ -2002,11 +2922,16 @@ int wolfPKCS11_Store_Write(void* store, unsigned char* buffer, int len) #endif #ifdef WOLFPKCS11_TPM_STORE - ret = wolfTPM2_NVWriteAuth(tpmStore->dev, &tpmStore->nv, - tpmStore->nv.handle.hndl, buffer, len, tpmStore->offset); - if (ret == 0) { - tpmStore->offset += len; - ret = len; + if (tpmStore != NULL) { + if (tpmStore->buf == NULL || buffer == NULL || len < 0 || + (word32)len > tpmStore->bufSz - tpmStore->offset) { + tpmStore->writeFailed = 1; + } + else { + XMEMCPY(tpmStore->buf + tpmStore->offset, buffer, (size_t)len); + tpmStore->offset += (word32)len; + ret = len; + } } #else if (ctx != NULL && ctx->file != XBADFILE && ctx->file != NULL) { @@ -2080,6 +3005,48 @@ static void wp11_storage_close(void* storage) wolfPKCS11_Store_Close(storage); } +/* + * Closes access to location being written, committing the data only when + * writing succeeded. + * + * @param [in] storage Context for operation. + * @param [in] ret Result of writing the data. + * @return ret when writing failed. + * @return Result of committing the data otherwise. + */ +static int wp11_storage_close_write(void* storage, int ret) +{ +#ifdef WOLFPKCS11_CUSTOM_STORE + wolfPKCS11_Store_Close(storage); +#else + int closeRet = wolfPKCS11_Store_CloseCommit(storage, ret == 0); + + if (ret == 0) + ret = closeRet; +#endif + return ret; +} + +/** + * Get the number of bytes left to read, when the store can tell. + * + * @param [in] storage Context for operation. + * @param [out] left Number of bytes left to read. + * @return 0 on success. + * @return NOT_AVAILABLE_E when the store cannot tell. + * @return Other value on failure. + */ +static int wp11_storage_left(void* storage, long* left) +{ +#if !defined(WOLFPKCS11_CUSTOM_STORE) && !defined(WOLFPKCS11_TPM_STORE) + return wolfPKCS11_Store_Left(storage, left); +#else + (void)storage; + (void)left; + return NOT_AVAILABLE_E; +#endif +} + /** * Reads a specific number of bytes into buffer. * @@ -2115,6 +3082,39 @@ static int wp11_storage_read(void* storage, unsigned char* buffer, int len) return ret; } +#ifdef DEBUG_WOLFPKCS11 +static int storeWriteFailAfter = -1; +static int storeWriteFailOnce = 0; + +/** + * Test hook: make storage writes fail after a number of them succeed. + * Exposed only in debug builds so a test can interrupt a record mid-write. + * + * @param writes [in] Writes that succeed before every later write fails, + * or -1 to stop failing writes. + * @return 0 always. + */ +WP11_API int WP11_Test_StoreWriteFailAfter(int writes) +{ + storeWriteFailAfter = writes; + storeWriteFailOnce = 0; + return 0; +} + +/** + * Test hook: make one storage write fail after a number of them succeed. + * + * @param writes [in] Writes that succeed before the one that fails. + * @return 0 always. + */ +WP11_API int WP11_Test_StoreWriteFailOnce(int writes) +{ + storeWriteFailAfter = writes; + storeWriteFailOnce = 1; + return 0; +} +#endif + /** * Writes a specific number of bytes from buffer. * @@ -2129,6 +3129,16 @@ static int wp11_storage_write(void* storage, unsigned char* buffer, int len) int ret = 0; unsigned char* p = buffer; +#ifdef DEBUG_WOLFPKCS11 + if (storeWriteFailAfter == 0) { + if (storeWriteFailOnce) + storeWriteFailAfter = -1; + return BUFFER_E; + } + if (storeWriteFailAfter > 0) + storeWriteFailAfter--; +#endif + /* Keep writing until no data written, error or all written. */ while (len > 0) { ret = wolfPKCS11_Store_Write(storage, p, len); @@ -2462,29 +3472,78 @@ static int wp11_storage_write_array(void* storage, * @param [in] len Length of data to read. * @return 0 on success. * @return BUFFER_E to indicate failure. + * @return ASN_PARSE_E when the stored length is negative. * @return MEMORY_E when dynamic memory allocation fails. */ static int wp11_storage_read_alloc_array(void* storage, unsigned char** buffer, int* len) { int ret; + int have = 0; + int bufSz = 0; + int newSz; + int first = WP11_STORE_READ_CHUNK_SZ; + long left = 0; + unsigned char* buf = NULL; + unsigned char* tmp; /* Read length of array. */ ret = wp11_storage_read_int(storage, len); + if (ret == 0 && *len < 0) { + *len = 0; + /* Not BUFFER_E, which callers accept as an older record's end. */ + ret = ASN_PARSE_E; + } if (ret == 0 && *len > 0) { - /* Allocate buffer to hold data. */ - *buffer = (unsigned char*)XMALLOC(*len, NULL, DYNAMIC_TYPE_TMP_BUFFER); - if (*buffer == NULL) - ret = MEMORY_E; + /* When the store knows its size, allocate once if the data is there. */ + ret = wp11_storage_left(storage, &left); + if (ret == 0 && (long)*len > left) + ret = BUFFER_E; + else if (ret == 0) + first = *len; + else if (ret == NOT_AVAILABLE_E) + ret = 0; + } + /* Otherwise grow the buffer only as stored data is read, so a corrupt + * length cannot drive an allocation larger than the data present. */ + while (ret == 0 && have < *len) { + if (bufSz == 0) + newSz = first; + else if (bufSz <= *len / 2) + newSz = bufSz * 2; + else + newSz = *len; + if (newSz > *len) + newSz = *len; - if (ret == 0) { - /* Read array data into allocated buffer. */ - ret = wp11_storage_read(storage, *buffer, *len); - if (ret != 0) { - XFREE(*buffer, NULL, DYNAMIC_TYPE_TMP_BUFFER); - *buffer = NULL; + tmp = (unsigned char*)XMALLOC(newSz, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (tmp == NULL) { + ret = MEMORY_E; + } + else { + if (buf != NULL) { + XMEMCPY(tmp, buf, have); + wc_ForceZero(buf, (word32)have); + XFREE(buf, NULL, DYNAMIC_TYPE_TMP_BUFFER); } + buf = tmp; + bufSz = newSz; + /* Read array data into the newly allocated space. */ + ret = wp11_storage_read(storage, buf + have, bufSz - have); + if (ret == 0) + have = bufSz; + } + } + + if (ret != 0 && (buf != NULL || ret == MEMORY_E)) { + if (buf != NULL) { + wc_ForceZero(buf, (word32)bufSz); + XFREE(buf, NULL, DYNAMIC_TYPE_TMP_BUFFER); } + *buffer = NULL; + } + else if (buf != NULL) { + *buffer = buf; } return ret; @@ -2563,7 +3622,7 @@ static int wp11_Object_New(WP11_Slot* slot, CK_KEY_TYPE type, /* Type-specific allocation will be done later when object class is known */ - if (ret != 0) { + if (ret != 0 && obj != NULL) { WP11_Object_Free(obj); obj = NULL; } @@ -2869,6 +3928,14 @@ int WP11_Object_Copy(WP11_Object *src, WP11_Object *dest) if (src == NULL || dest == NULL) return BAD_FUNC_ARG; + /* Not copyable; reject before any field is copied. */ + if (src->objClass == CKO_CERTIFICATE) + return BAD_FUNC_ARG; +#ifdef WOLFPKCS11_NSS + if (src->objClass == CKO_NSS_TRUST) + return BAD_FUNC_ARG; +#endif + /* We save data copying for the last step */ /* Copy the common mutable fields, and a data object's payload, under the @@ -2882,9 +3949,12 @@ int WP11_Object_Copy(WP11_Object *src, WP11_Object *dest) OBJ_COPY_DATA(src, dest, keyData); XMEMCPY(dest->iv, src->iv, sizeof(dest->iv)); dest->encoded = src->encoded; + dest->undecoded = src->undecoded; + dest->unbound = src->unbound; #endif dest->objClass = src->objClass; dest->keyGenMech = src->keyGenMech; + dest->local = src->local; dest->opFlag = src->opFlag; XMEMCPY(dest->startDate, src->startDate, sizeof(dest->startDate)); XMEMCPY(dest->endDate, src->endDate, sizeof(dest->endDate)); @@ -2921,15 +3991,7 @@ int WP11_Object_Copy(WP11_Object *src, WP11_Object *dest) if (ret != 0) return ret; - if (src->objClass == CKO_CERTIFICATE) { - return BAD_FUNC_ARG; - } -#ifdef WOLFPKCS11_NSS - else if (src->objClass == CKO_NSS_TRUST) { - return BAD_FUNC_ARG; - } -#endif - else if (src->objClass == CKO_DATA) { + if (src->objClass == CKO_DATA) { /* Payload copied above under the source lock. */ } else { @@ -3314,74 +4376,209 @@ int WP11_Object_Copy(WP11_Object *src, WP11_Object *dest) return ret; } -#ifndef WOLFPKCS11_NO_STORE +#ifndef WOLFPKCS11_NO_STORE +/** + * Encrypt the data with AES-GCM. + * + * The tag is append to the encrypted data. + * + * @param [out] out Output buffer to hold encrypted data and tag. + * @param [in] data Data to be encrypted. + * @param [in] len Length of data. + * @param [in] key AES key. + * @param [in] keySz Length of AES key in bytes. + * @param [in] iv IV/nonce. + * @param [in] ivSz Length of IV in bytes. + * @param [in] aad Additional authenticated data. May be NULL. + * @param [in] aadSz Length of additional authenticated data in bytes. + * @return 0 on success. + * @return -ve on failure. + */ +static int wp11_EncryptData(byte* out, byte* data, int len, byte* key, + int keySz, byte* iv, int ivSz, const byte* aad, + word32 aadSz, int devId) +{ + Aes aes; + int ret; + + ret = wc_AesInit(&aes, NULL, devId); + if (ret == 0) { + ret = wc_AesGcmSetKey(&aes, key, keySz); + } + if (ret == 0) { + ret = wc_AesGcmEncrypt(&aes, out, data, len, iv, ivSz, out + len, + AES_BLOCK_SIZE, aad, aadSz); + } + wc_AesFree(&aes); + + return ret; +} + +/** + * Decrypt the data with AES-GCM. + * + * The tag is append to the encrypted data. + * + * @param [out] out Output buffer to hold decrypted data. + * @param [in] data Data and tag to be decrypted. + * @param [in] len Length of data. + * @param [in] key AES key. + * @param [in] keySz Length of AES key in bytes. + * @param [in] iv IV/nonce. + * @param [in] ivSz Length of IV in bytes. + * @param [in] aad Additional authenticated data. May be NULL. + * @param [in] aadSz Length of additional authenticated data in bytes. + * @return 0 on success. + * @return AES_GCM_AUTH_E when encrypted data could not be verified. + * @return Other -ve on failure. + */ +static int wp11_DecryptData(byte* out, byte* data, int len, byte* key, + int keySz, byte* iv, int ivSz, const byte* aad, + word32 aadSz, int devId) +{ + Aes aes; + int ret; + + ret = wc_AesInit(&aes, NULL, devId); + if (ret == 0) { + ret = wc_AesGcmSetKey(&aes, key, keySz); + } + if (ret == 0) { + ret = wc_AesGcmDecrypt(&aes, out, data, len, iv, ivSz, data + len, + AES_BLOCK_SIZE, aad, aadSz); + } + wc_AesFree(&aes); + + return ret; +} + +/* Left out of the bound policy: set by the SO or the library, not the user. */ +#ifdef WOLFPKCS11_TPM +#define WP11_FLAG_POLICY_EXCLUDED (WP11_FLAG_TRUSTED | WP11_FLAG_TPM) +#else +#define WP11_FLAG_POLICY_EXCLUDED WP11_FLAG_TRUSTED +#endif + +#define WP11_POLICY_AAD_VER 1 +#define WP11_POLICY_AAD_SZ (1 + 3 * sizeof(CK_ULONG) + 1 + sizeof(word32)) + +/** + * Serialize the object attributes that govern use of its key material. + * + * @param [in] object Key object. + * @param [out] aad Buffer of WP11_POLICY_AAD_SZ bytes. + */ +static void wp11_Object_PolicyAad(WP11_Object* object, byte* aad) +{ + CK_ULONG vals[3]; + word32 flags = object->opFlag & ~(word32)WP11_FLAG_POLICY_EXCLUDED; + int idx = 0; + int i; + int j; + + vals[0] = object->objClass; + vals[1] = object->type; + vals[2] = object->keyGenMech; + aad[idx++] = WP11_POLICY_AAD_VER; + for (i = 0; i < 3; i++) { + for (j = (int)sizeof(CK_ULONG) - 1; j >= 0; j--) + aad[idx++] = (byte)(vals[i] >> (j * 8)); + } + aad[idx++] = (byte)object->local; + for (j = (int)sizeof(word32) - 1; j >= 0; j--) + aad[idx++] = (byte)(flags >> (j * 8)); +} + /** - * Encrypt the data with AES-GCM. - * - * The tag is append to the encrypted data. + * Encrypt key material with the token key, bound to the object's policy. * - * @param [out] out Output buffer to hold encrypted data and tag. - * @param [in] data Data to be encrypted. - * @param [in] len Length of data. - * @param [in] key AES key. - * @param [in] keySz Length of AES key in bytes. - * @param [in] iv IV/nonce. - * @param [in] ivSz Length of IV in bytes. + * @param [in, out] object Key object. + * @param [out] out Buffer for encrypted data and tag. + * @param [in] data Key material. + * @param [in] len Length of key material. * @return 0 on success. * @return -ve on failure. */ -static int wp11_EncryptData(byte* out, byte* data, int len, byte* key, - int keySz, byte* iv, int ivSz, int devId) +static int wp11_Object_EncryptKeyData(WP11_Object* object, byte* out, + byte* data, int len) { - Aes aes; int ret; + byte aad[WP11_POLICY_AAD_SZ]; - ret = wc_AesInit(&aes, NULL, devId); - if (ret == 0) { - ret = wc_AesGcmSetKey(&aes, key, keySz); - } - if (ret == 0) { - ret = wc_AesGcmEncrypt(&aes, out, data, len, iv, ivSz, out + len, - AES_BLOCK_SIZE, NULL, 0); - } - wc_AesFree(&aes); + wp11_Object_PolicyAad(object, aad); + ret = wp11_EncryptData(out, data, len, object->slot->token.key, + sizeof(object->slot->token.key), object->iv, + sizeof(object->iv), aad, sizeof(aad), + object->devId); + if (ret == 0) + object->unbound = 0; return ret; } /** - * Decrypt the data with AES-GCM. - * - * The tag is append to the encrypted data. + * Decrypt key material with the token key, verifying the object's policy. + * Key data stored before the policy was bound is accepted and marked so it is + * bound when next stored. * - * @param [out] out Output buffer to hold decrypted data. - * @param [in] data Data and tag to be decrypted. - * @param [in] len Length of data. - * @param [in] key AES key. - * @param [in] keySz Length of AES key in bytes. - * @param [in] iv IV/nonce. - * @param [in] ivSz Length of IV in bytes. + * @param [in, out] object Key object. + * @param [out] out Buffer for decrypted key material. + * @param [in] data Encrypted data and tag. + * @param [in] len Length of encrypted data. * @return 0 on success. * @return AES_GCM_AUTH_E when encrypted data could not be verified. * @return Other -ve on failure. */ -static int wp11_DecryptData(byte* out, byte* data, int len, byte* key, - int keySz, byte* iv, int ivSz, int devId) +static int wp11_Object_DecryptKeyData(WP11_Object* object, byte* out, + byte* data, int len) { - Aes aes; int ret; + byte aad[WP11_POLICY_AAD_SZ]; + + wp11_Object_PolicyAad(object, aad); + ret = wp11_DecryptData(out, data, len, object->slot->token.key, + sizeof(object->slot->token.key), object->iv, + sizeof(object->iv), aad, sizeof(aad), + object->devId); + if (ret == AES_GCM_AUTH_E) { + ret = wp11_DecryptData(out, data, len, object->slot->token.key, + sizeof(object->slot->token.key), object->iv, + sizeof(object->iv), NULL, 0, object->devId); + if (ret == 0) + object->unbound = 1; + } - ret = wc_AesInit(&aes, NULL, devId); + return ret; +} + +static int wp11_Object_IsEncrypted(WP11_Object* object); +static int wp11_Object_Encode(WP11_Object* object, int protect); + +/** + * Finish encoding key data into a new buffer: on success release the previous + * record, on failure drop the new buffer and put the previous record back so + * the key still decodes at the next login. + * + * @param [in, out] object Key object. + * @param [in] oldData Key data buffer before the encode. + * @param [in] oldLen Length of key data before the encode. + * @param [in] newLen Size of the buffer allocated for the encode. + * @param [in] ret Result of the encode. + */ +static void wp11_Object_EncodeDone(WP11_Object* object, unsigned char* oldData, + int oldLen, word32 newLen, int ret) +{ if (ret == 0) { - ret = wc_AesGcmSetKey(&aes, key, keySz); + XFREE(oldData, NULL, DYNAMIC_TYPE_TMP_BUFFER); } - if (ret == 0) { - ret = wc_AesGcmDecrypt(&aes, out, data, len, iv, ivSz, data + len, - AES_BLOCK_SIZE, NULL, 0); + else { + if (object->keyData != oldData && object->keyData != NULL) { + wc_ForceZero(object->keyData, newLen); + XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); + } + object->keyData = oldData; + object->keyDataLen = oldLen; } - wc_AesFree(&aes); - - return ret; } /** @@ -3407,14 +4604,25 @@ static void wp11_Object_Decode_Cert(WP11_Object* object) * Trust is not encrypted. * * @param [in, out] object Trust object. + * @return 0 on success. + * @return BUFFER_E when the stored trust data is not the expected size. */ -static void wp11_Object_Decode_Trust(WP11_Object* object) +static int wp11_Object_Decode_Trust(WP11_Object* object) { + int ret = 0; + if (object->keyData != NULL) { - XMEMCPY((unsigned char*)&object->data.trust, object->keyData, - object->keyDataLen); + if (object->keyDataLen != (int)sizeof(WP11_Trust)) { + ret = BUFFER_E; + } + else { + XMEMCPY((unsigned char*)&object->data.trust, object->keyData, + object->keyDataLen); + } } - object->encoded = 0; + object->encoded = (ret != 0); + + return ret; } #endif @@ -3481,7 +4689,11 @@ static int wp11_Object_Load_Trust(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_read_alloc_array(storage, &object->keyData, &object->keyDataLen); wp11_storage_close(storage); - wp11_Object_Decode_Trust(object); + /* An empty record would leave the trust zeroed but decoded. */ + if (ret == 0 && object->keyData == NULL) + ret = BUFFER_E; + if (ret == 0) + ret = wp11_Object_Decode_Trust(object); } return ret; @@ -3492,7 +4704,7 @@ static int wp11_Object_Load_Data(WP11_Object* object, int tokenId, int objId) { int ret; void* storage = NULL; - int tempLen; + int tempLen = 0; /* Open access to data. */ ret = wp11_storage_open_readonly(WOLFPKCS11_STORE_DATA, tokenId, objId, @@ -3501,7 +4713,8 @@ static int wp11_Object_Load_Data(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_read_alloc_array(storage, &object->data.genericData.data, &tempLen); - object->data.genericData.dataLen = (word32)tempLen; + if (ret == 0) + object->data.genericData.dataLen = (word32)tempLen; } /* Read application length and application. */ @@ -3509,7 +4722,8 @@ static int wp11_Object_Load_Data(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_read_alloc_array(storage, &object->data.genericData.application, &tempLen); - object->data.genericData.applicationLen = (word32)tempLen; + if (ret == 0) + object->data.genericData.applicationLen = (word32)tempLen; } /* Read object ID length and object ID. */ @@ -3517,7 +4731,8 @@ static int wp11_Object_Load_Data(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_read_alloc_array(storage, &object->data.genericData.objectId, &tempLen); - object->data.genericData.objectIdLen = (word32)tempLen; + if (ret == 0) + object->data.genericData.objectIdLen = (word32)tempLen; } wp11_storage_close(storage); @@ -3822,7 +5037,7 @@ static int wp11_Object_Store_Cert(WP11_Object* object, int tokenId, int objId) /* Write cert to storage. */ ret = wp11_storage_write_array(storage, object->data.cert.data, object->data.cert.len); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } #ifdef WOLFSSL_MAXQ10XX_CRYPTO @@ -3861,7 +5076,7 @@ static int wp11_Object_Store_Trust(WP11_Object* object, int tokenId, int objId) /* Write trust to storage. */ ret = wp11_storage_write_array(storage, (unsigned char*)&object->data.trust, sizeof(WP11_Trust)); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; @@ -4028,15 +5243,15 @@ static int WP11_Object_EncodeTpmKey(WP11_Object* object, byte* keyData, if (keyData != NULL) { if (ret <= keyDataLen) { /* Write size marker for the public part */ - XMEMCPY(object->keyData, &pubAreaSize, + XMEMCPY(keyData, &pubAreaSize, sizeof(UINT16)); idx += sizeof(UINT16); /* Write the public part with bytes aligned */ - XMEMCPY(object->keyData + idx, pubAreaBuffer, + XMEMCPY(keyData + idx, pubAreaBuffer, sizeof(UINT16) + pubAreaSize); idx += sizeof(UINT16) + pubAreaSize; /* Write the private part, size marker is included */ - XMEMCPY(object->keyData + idx, &object->tpmKey->priv, + XMEMCPY(keyData + idx, &object->tpmKey->priv, sizeof(UINT16) + object->tpmKey->priv.size); idx += sizeof(UINT16) + object->tpmKey->priv.size; @@ -4090,7 +5305,7 @@ static int wp11_Object_Store_Data(WP11_Object* object, int tokenId, int objId) object->data.genericData.objectIdLen); } - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); return ret; } @@ -4123,32 +5338,36 @@ static int wp11_Object_Decode_RsaKey(WP11_Object* object) else #endif if (object->objClass == CKO_PRIVATE_KEY) { - unsigned char* der; + unsigned char* der = NULL; int len = object->keyDataLen - AES_BLOCK_SIZE; - der = (unsigned char*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); - if (der == NULL) { + if (object->keyDataLen <= AES_BLOCK_SIZE) + ret = BAD_FUNC_ARG; + if (ret == 0) + der = (unsigned char*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (ret == 0 && der == NULL) { ret = MEMORY_E; } if (ret == 0) { - ret = wp11_DecryptData(der, object->keyData, len, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_DecryptKeyData(object, der, object->keyData, len); } if (ret == 0) { /* Decode RSA private key. */ ret = wc_RsaPrivateKeyDecode(der, &idx, key, len); - wc_ForceZero(der, len); } - if (der != NULL) + if (der != NULL) { + wc_ForceZero(der, len); XFREE(der, NULL, DYNAMIC_TYPE_TMP_BUFFER); + } } else { /* Decode RSA public key. */ ret = wc_RsaPublicKeyDecode(object->keyData, &idx, key, object->keyDataLen); } + /* Drop partially decoded components and the init before a retry. */ + if (ret != 0) + wc_FreeRsaKey(key); object->encoded = (ret != 0); return ret; @@ -4166,6 +5385,8 @@ static int wp11_Object_Decode_RsaKey(WP11_Object* object) static int wp11_Object_Encode_RsaKey(WP11_Object* object) { int ret; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; #ifdef WOLFPKCS11_TPM ret = WP11_Object_EncodeTpmKey(object, NULL, 0); @@ -4202,7 +5423,6 @@ static int wp11_Object_Encode_RsaKey(WP11_Object* object) } if (ret == 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); /* Allocate buffer to hold encoded key. */ object->keyData = (unsigned char*)XMALLOC(object->keyDataLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -4211,7 +5431,10 @@ static int wp11_Object_Encode_RsaKey(WP11_Object* object) } #ifdef WOLFPKCS11_TPM - ret = WP11_Object_EncodeTpmKey(object, object->keyData, object->keyDataLen); + if (ret == 0) { + ret = WP11_Object_EncodeTpmKey(object, object->keyData, + object->keyDataLen); + } if (ret > 0) { ret = 0; } @@ -4223,10 +5446,8 @@ static int wp11_Object_Encode_RsaKey(WP11_Object* object) ret = wc_RsaKeyToDer(object->data.rsaKey, object->keyData, object->keyDataLen); if (ret >= 0) { - ret = wp11_EncryptData(object->keyData, object->keyData, ret, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_EncryptKeyData(object, object->keyData, + object->keyData, ret); } #else ret = NOT_COMPILED_IN; @@ -4241,11 +5462,8 @@ static int wp11_Object_Encode_RsaKey(WP11_Object* object) } } - if (ret != 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyData = NULL; - object->keyDataLen = 0; - } + wp11_Object_EncodeDone(object, oldData, oldLen, (word32)object->keyDataLen, + ret); return ret; } @@ -4436,7 +5654,7 @@ static int wp11_Object_Store_RsaKey(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; @@ -4471,32 +5689,36 @@ static int wp11_Object_Decode_EccKey(WP11_Object* object) else #endif if (object->objClass == CKO_PRIVATE_KEY) { - unsigned char* der; + unsigned char* der = NULL; int len = object->keyDataLen - AES_BLOCK_SIZE; - der = (unsigned char*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); - if (der == NULL) { + if (object->keyDataLen <= AES_BLOCK_SIZE) + ret = BAD_FUNC_ARG; + if (ret == 0) + der = (unsigned char*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (ret == 0 && der == NULL) { ret = MEMORY_E; } if (ret == 0) { - ret = wp11_DecryptData(der, object->keyData, len, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_DecryptKeyData(object, der, object->keyData, len); } if (ret == 0) { /* Decode ECC private key. */ ret = wc_EccPrivateKeyDecode(der, &idx, key, len); - wc_ForceZero(der, len); } - if (der != NULL) + if (der != NULL) { + wc_ForceZero(der, len); XFREE(der, NULL, DYNAMIC_TYPE_TMP_BUFFER); + } } else { /* Decode ECC public key. */ ret = wc_EccPublicKeyDecode(object->keyData, &idx, key, object->keyDataLen); } + /* Drop partially decoded components and the init before a retry. */ + if (ret != 0) + wc_ecc_free(key); object->encoded = (ret != 0); return ret; @@ -4514,6 +5736,8 @@ static int wp11_Object_Decode_EccKey(WP11_Object* object) static int wp11_Object_Encode_EccKey(WP11_Object* object) { int ret; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; #ifdef WOLFPKCS11_TPM ret = WP11_Object_EncodeTpmKey(object, NULL, 0); @@ -4541,7 +5765,6 @@ static int wp11_Object_Encode_EccKey(WP11_Object* object) } if (ret == 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); /* Allocate buffer to hold encoded key. */ object->keyData = (unsigned char*)XMALLOC(object->keyDataLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -4550,7 +5773,10 @@ static int wp11_Object_Encode_EccKey(WP11_Object* object) } #ifdef WOLFPKCS11_TPM - ret = WP11_Object_EncodeTpmKey(object, object->keyData, object->keyDataLen); + if (ret == 0) { + ret = WP11_Object_EncodeTpmKey(object, object->keyData, + object->keyDataLen); + } if (ret > 0) { ret = 0; } @@ -4561,10 +5787,8 @@ static int wp11_Object_Encode_EccKey(WP11_Object* object) ret = wc_EccPrivateKeyToDer(object->data.ecKey, object->keyData, object->keyDataLen); if (ret >= 0) { - ret = wp11_EncryptData(object->keyData, object->keyData, ret, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_EncryptKeyData(object, object->keyData, + object->keyData, ret); } } else if (ret == 0 && object->objClass == CKO_PUBLIC_KEY) { @@ -4576,11 +5800,8 @@ static int wp11_Object_Encode_EccKey(WP11_Object* object) } } - if (ret != 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyData = NULL; - object->keyDataLen = 0; - } + wp11_Object_EncodeDone(object, oldData, oldLen, (word32)object->keyDataLen, + ret); return ret; } @@ -4658,7 +5879,7 @@ static int wp11_Object_Store_EccKey(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; @@ -4711,18 +5932,18 @@ static int wp11_Object_Decode_MldsaKey(WP11_Object* object) int ret = 0; if (object->objClass == CKO_PRIVATE_KEY) { - unsigned char* der; + unsigned char* der = NULL; int len = object->keyDataLen - AES_BLOCK_SIZE; - der = (unsigned char*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); - if (der == NULL) { + if (object->keyDataLen <= AES_BLOCK_SIZE) + ret = BAD_FUNC_ARG; + if (ret == 0) + der = (unsigned char*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (ret == 0 && der == NULL) { ret = MEMORY_E; } if (ret == 0) { - ret = wp11_DecryptData(der, object->keyData, len, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_DecryptKeyData(object, der, object->keyData, len); } if (ret == 0) { /* Decode ML-DSA private key. */ @@ -4736,10 +5957,11 @@ static int wp11_Object_Decode_MldsaKey(WP11_Object* object) ret = MldsaKeyTryDecode(object->data.mldsaKey, WC_ML_DSA_87, der, len, object->objClass); } - wc_ForceZero(der, len); } - if (der != NULL) + if (der != NULL) { + wc_ForceZero(der, len); XFREE(der, NULL, DYNAMIC_TYPE_TMP_BUFFER); + } } else { /* Decode ML-DSA public key. */ @@ -4774,6 +5996,8 @@ static int wp11_Object_Decode_MldsaKey(WP11_Object* object) static int wp11_Object_Encode_MldsaKey(WP11_Object* object) { int ret; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; if (object->objClass == CKO_PRIVATE_KEY) { /* Get length of encoded private key. */ @@ -4793,7 +6017,6 @@ static int wp11_Object_Encode_MldsaKey(WP11_Object* object) } if (ret == 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); /* Allocate buffer to hold encoded key. */ object->keyData = (unsigned char*)XMALLOC(object->keyDataLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -4807,10 +6030,8 @@ static int wp11_Object_Encode_MldsaKey(WP11_Object* object) object->keyData, object->keyDataLen); if (ret >= 0) { - ret = wp11_EncryptData(object->keyData, object->keyData, ret, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_EncryptKeyData(object, object->keyData, + object->keyData, ret); } } else if (ret == 0 && object->objClass == CKO_PUBLIC_KEY) { @@ -4823,11 +6044,8 @@ static int wp11_Object_Encode_MldsaKey(WP11_Object* object) } } - if (ret != 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyData = NULL; - object->keyDataLen = 0; - } + wp11_Object_EncodeDone(object, oldData, oldLen, (word32)object->keyDataLen, + ret); return ret; } @@ -4906,7 +6124,7 @@ static int wp11_Object_Store_MldsaKey(WP11_Object* object, int tokenId, ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; @@ -4928,14 +6146,24 @@ static int wp11_Object_Decode_DhKey(WP11_Object* object) int ret = 0; if (object->objClass == CKO_PRIVATE_KEY) { - ret = wp11_DecryptData(object->data.dhKey->key, object->keyData, - object->keyDataLen - AES_BLOCK_SIZE, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + if (object->keyDataLen <= AES_BLOCK_SIZE) + ret = BAD_FUNC_ARG; + else if (object->keyDataLen - AES_BLOCK_SIZE > WP11_MAX_DH_KEY_SZ) + ret = BUFFER_E; + if (ret == 0) { + ret = wp11_Object_DecryptKeyData(object, object->data.dhKey->key, + object->keyData, object->keyDataLen - AES_BLOCK_SIZE); + if (ret != 0) { + wc_ForceZero(object->data.dhKey->key, + (word32)(object->keyDataLen - AES_BLOCK_SIZE)); + } + } if (ret == 0) object->data.dhKey->len = object->keyDataLen - AES_BLOCK_SIZE; } + else if (object->keyDataLen > WP11_MAX_DH_KEY_SZ) { + ret = BUFFER_E; + } else { XMEMCPY(object->data.dhKey->key, object->keyData, object->keyDataLen); object->data.dhKey->len = object->keyDataLen; @@ -4957,22 +6185,22 @@ static int wp11_Object_Decode_DhKey(WP11_Object* object) static int wp11_Object_Encode_DhKey(WP11_Object* object) { int ret = 0; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; + word32 newLen = object->data.dhKey->len + AES_BLOCK_SIZE; - object->keyDataLen = object->data.dhKey->len + AES_BLOCK_SIZE; - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); + object->keyDataLen = (int)newLen; /* Allocate buffer to hold encoded key. */ - object->keyData = (unsigned char*)XMALLOC(object->keyDataLen, NULL, - DYNAMIC_TYPE_TMP_BUFFER); + object->keyData = (unsigned char*)XMALLOC(newLen, NULL, + DYNAMIC_TYPE_TMP_BUFFER); if (object->keyData == NULL) ret = MEMORY_E; if (ret == 0) { if (object->objClass == CKO_PRIVATE_KEY) { - ret = wp11_EncryptData(object->keyData, object->data.dhKey->key, - object->data.dhKey->len, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_EncryptKeyData(object, object->keyData, + object->data.dhKey->key, + object->data.dhKey->len); if (ret == 0) object->keyDataLen = object->data.dhKey->len + AES_BLOCK_SIZE; } @@ -4982,6 +6210,7 @@ static int wp11_Object_Encode_DhKey(WP11_Object* object) object->keyDataLen = object->data.dhKey->len; } } + wp11_Object_EncodeDone(object, oldData, oldLen, newLen, ret); return ret; } @@ -5238,7 +6467,7 @@ static int wp11_Object_Store_DhKey(WP11_Object* object, int tokenId, int objId) ret = wp11_storage_write_array(storage, der, len); } - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } if (der != NULL) { XFREE(der, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -5296,10 +6525,7 @@ static int wp11_Object_Decode_MlKemKey(WP11_Object* object) ret = MEMORY_E; } if (ret == 0) { - ret = wp11_DecryptData(der, object->keyData, len, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_DecryptKeyData(object, der, object->keyData, len); } if (ret == 0) { ret = MlKemKeyTryDecode(object->data.mlKemKey, WC_ML_KEM_512, @@ -5353,6 +6579,8 @@ static int wp11_Object_Encode_MlKemKey(WP11_Object* object) { int ret; word32 keyLen = 0; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; if (object->objClass == CKO_PRIVATE_KEY) { ret = wc_MlKemKey_PrivateKeySize(object->data.mlKemKey, &keyLen); @@ -5368,7 +6596,6 @@ static int wp11_Object_Encode_MlKemKey(WP11_Object* object) } if (ret == 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); object->keyData = (unsigned char*)XMALLOC(object->keyDataLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); if (object->keyData == NULL) @@ -5379,10 +6606,8 @@ static int wp11_Object_Encode_MlKemKey(WP11_Object* object) ret = wc_MlKemKey_EncodePrivateKey(object->data.mlKemKey, object->keyData, keyLen); if (ret == 0) { - ret = wp11_EncryptData(object->keyData, object->keyData, keyLen, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_EncryptKeyData(object, object->keyData, + object->keyData, keyLen); } } else if (ret == 0 && object->objClass == CKO_PUBLIC_KEY) { @@ -5390,13 +6615,8 @@ static int wp11_Object_Encode_MlKemKey(WP11_Object* object) object->keyData, keyLen); } - if (ret != 0) { - if (object->keyData != NULL) - wc_ForceZero(object->keyData, object->keyDataLen); - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyData = NULL; - object->keyDataLen = 0; - } + wp11_Object_EncodeDone(object, oldData, oldLen, (word32)object->keyDataLen, + ret); return ret; } @@ -5468,7 +6688,7 @@ static int wp11_Object_Store_MlKemKey(WP11_Object* object, int tokenId, if (ret == 0) { ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; @@ -5575,13 +6795,14 @@ static int wp11_Object_Encode_HssKey(WP11_Object* object) { int ret; word32 pubLen = 0; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; if (object->objClass != CKO_PUBLIC_KEY) return NOT_AVAILABLE_E; /* verify-only: no private-key objects */ ret = wc_LmsKey_GetPubLen(object->data.lmsKey, &pubLen); if (ret == 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); object->keyData = (unsigned char*)XMALLOC(pubLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); if (object->keyData == NULL) @@ -5593,14 +6814,7 @@ static int wp11_Object_Encode_HssKey(WP11_Object* object) if (ret == 0) object->keyDataLen = (int)pubLen; } - /* On failure release the buffer so a later store does not skip re-encoding - * (keyData != NULL) and persist an uninitialised buffer with a stale - * length. This is public key material only, so no zeroization is needed. */ - if (ret != 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyData = NULL; - object->keyDataLen = 0; - } + wp11_Object_EncodeDone(object, oldData, oldLen, pubLen, ret); return ret; } @@ -5650,7 +6864,7 @@ static int wp11_Object_Store_HssKey(WP11_Object* object, int tokenId, int objId) if (ret == 0) { ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; } @@ -6020,13 +7234,14 @@ static int wp11_Object_Encode_XmssKey(WP11_Object* object) { int ret; word32 pubLen = 0; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; if (object->objClass != CKO_PUBLIC_KEY) return NOT_AVAILABLE_E; ret = wc_XmssKey_GetPubLen(object->data.xmssKey, &pubLen); if (ret == 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); object->keyData = (unsigned char*)XMALLOC(pubLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); if (object->keyData == NULL) @@ -6038,14 +7253,7 @@ static int wp11_Object_Encode_XmssKey(WP11_Object* object) if (ret == 0) object->keyDataLen = (int)pubLen; } - /* On failure release the buffer so a later store does not skip re-encoding - * (keyData != NULL) and persist an uninitialised buffer with a stale - * length. This is public key material only, so no zeroization is needed. */ - if (ret != 0) { - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyData = NULL; - object->keyDataLen = 0; - } + wp11_Object_EncodeDone(object, oldData, oldLen, pubLen, ret); return ret; } @@ -6096,7 +7304,7 @@ static int wp11_Object_Store_XmssKey(WP11_Object* object, int tokenId, if (ret == 0) { ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; } @@ -6318,11 +7526,13 @@ static int wp11_Object_Decode_SymmKey(WP11_Object* object) (word32)(object->keyDataLen - AES_BLOCK_SIZE) > WP11_MAX_SYM_KEY_SZ) ret = BUFFER_E; if (ret == 0) { - ret = wp11_DecryptData(object->data.symmKey->data, object->keyData, - object->keyDataLen - AES_BLOCK_SIZE, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + ret = wp11_Object_DecryptKeyData(object, object->data.symmKey->data, + object->keyData, + object->keyDataLen - AES_BLOCK_SIZE); + if (ret != 0) { + wc_ForceZero(object->data.symmKey->data, + (word32)(object->keyDataLen - AES_BLOCK_SIZE)); + } } if (ret == 0) object->data.symmKey->len = object->keyDataLen - AES_BLOCK_SIZE; @@ -6341,24 +7551,23 @@ static int wp11_Object_Decode_SymmKey(WP11_Object* object) static int wp11_Object_Encode_SymmKey(WP11_Object* object) { int ret = 0; + unsigned char* oldData = object->keyData; + int oldLen = object->keyDataLen; + word32 newLen = object->data.symmKey->len + AES_BLOCK_SIZE; - object->keyDataLen = object->data.symmKey->len + AES_BLOCK_SIZE; - XFREE(object->keyData, NULL, DYNAMIC_TYPE_TMP_BUFFER); + object->keyDataLen = (int)newLen; /* Allocate buffer to hold encoded key. */ - object->keyData = (unsigned char*)XMALLOC(object->keyDataLen, NULL, - DYNAMIC_TYPE_TMP_BUFFER); + object->keyData = (unsigned char*)XMALLOC(newLen, NULL, + DYNAMIC_TYPE_TMP_BUFFER); if (object->keyData == NULL) ret = MEMORY_E; if (ret == 0) { - ret = wp11_EncryptData(object->keyData, object->data.symmKey->data, - object->data.symmKey->len, - object->slot->token.key, - sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); - if (ret == 0) - object->keyDataLen = object->data.symmKey->len + AES_BLOCK_SIZE; + ret = wp11_Object_EncryptKeyData(object, object->keyData, + object->data.symmKey->data, + object->data.symmKey->len); } + wp11_Object_EncodeDone(object, oldData, oldLen, newLen, ret); return ret; } @@ -6421,7 +7630,7 @@ static int wp11_Object_Store_SymmKey(WP11_Object* object, int tokenId, ret = wp11_storage_write_array(storage, object->keyData, object->keyDataLen); - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; @@ -6623,6 +7832,8 @@ static int wp11_Object_Load(WP11_Object* object, int tokenId, int objId) } } } + if (ret == 0 && wp11_Object_IsEncrypted(object)) + object->undecoded = 1; return ret; } @@ -6722,7 +7933,7 @@ static int wp11_Object_Store_Object(WP11_Object* object, int tokenId, int objId) } #endif - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } return ret; } @@ -6740,10 +7951,15 @@ static int wp11_Object_Store_Object(WP11_Object* object, int tokenId, int objId) */ static int wp11_Object_Store(WP11_Object* object, int tokenId, int objId) { - int ret; + int ret = 0; + /* Bind decoded key data to the current policy before writing anything, so + * a failure leaves the stored record unchanged. */ + if (object->unbound && !object->encoded && wp11_Object_IsEncrypted(object)) + ret = wp11_Object_Encode(object, 0); /* Open access to key object. */ - ret = wp11_Object_Store_Object(object, tokenId, objId); + if (ret == 0) + ret = wp11_Object_Store_Object(object, tokenId, objId); if (ret == 0) { if (object->objClass == CKO_CERTIFICATE) { @@ -6834,8 +8050,7 @@ static int wp11_Object_Decode(WP11_Object* object) } #ifdef WOLFPKCS11_NSS else if (object->objClass == CKO_NSS_TRUST) { - wp11_Object_Decode_Trust(object); - ret = 0; + ret = wp11_Object_Decode_Trust(object); } #endif else if (object->objClass == CKO_DATA) { @@ -6894,6 +8109,8 @@ static int wp11_Object_Decode(WP11_Object* object) } } + if (ret == 0) + object->undecoded = 0; /* Authentication failure means this object isn't for this user. */ if (ret == AES_GCM_AUTH_E) ret = 0; @@ -7026,10 +8243,12 @@ static int wp11_Object_Encode(WP11_Object* object, int protect) { int ret = 0; int encrypt = wp11_Object_IsEncrypted(object); + byte oldIv[GCM_NONCE_MID_SZ]; /* Every AES-GCM encryption under the token key needs a fresh nonce. Do * this immediately before encoding, while the plaintext is still the * source of the ciphertext that will be persisted. */ + XMEMCPY(oldIv, object->iv, sizeof(oldIv)); if (encrypt) { WP11_Lock_LockRW(&object->slot->token.rngLock); ret = wc_RNG_GenerateBlock(&object->slot->token.rng, object->iv, @@ -7038,6 +8257,9 @@ static int wp11_Object_Encode(WP11_Object* object, int protect) } if (ret == 0) ret = wp11_Object_EncodeData(object, protect); + /* The previous record stays in use when the encode fails. */ + if (ret != 0) + XMEMCPY(object->iv, oldIv, sizeof(object->iv)); return ret; } @@ -7090,6 +8312,7 @@ static int wp11_Object_ReKey(WP11_Object* object, byte* oldKey, byte* newKey, int ret = 0; int len = 0; byte* plain = NULL; + byte aad[WP11_POLICY_AAD_SZ]; /* Key data is the encrypted payload followed by the AES-GCM tag. */ if (object->keyDataLen < AES_BLOCK_SIZE) @@ -7104,9 +8327,18 @@ static int wp11_Object_ReKey(WP11_Object* object, byte* oldKey, byte* newKey, ret = MEMORY_E; } if (ret == 0) { + wp11_Object_PolicyAad(object, aad); ret = wp11_DecryptData(plain, object->keyData, len, oldKey, sizeof(object->slot->token.key), object->iv, - sizeof(object->iv), object->devId); + sizeof(object->iv), aad, sizeof(aad), + object->devId); + /* Key data stored before policy binding has no AAD. */ + if (ret == AES_GCM_AUTH_E) { + ret = wp11_DecryptData(plain, object->keyData, len, oldKey, + sizeof(object->slot->token.key), + object->iv, sizeof(object->iv), NULL, 0, + object->devId); + } } if (ret == 0) { WP11_Lock_LockRW(&object->slot->token.rngLock); @@ -7117,7 +8349,8 @@ static int wp11_Object_ReKey(WP11_Object* object, byte* oldKey, byte* newKey, if (ret == 0) { ret = wp11_EncryptData(reKey->keyData, plain, len, newKey, sizeof(object->slot->token.key), reKey->iv, - sizeof(reKey->iv), object->devId); + sizeof(reKey->iv), aad, sizeof(aad), + object->devId); } if (plain != NULL) { @@ -7497,7 +8730,7 @@ static void wp11_Token_Final(WP11_Token* token) Rng_Free(&token->rng); WP11_Lock_Free(&token->rngLock); WP11_Lock_Free(&token->lock); - XMEMSET(token, 0, sizeof(*token)); + wc_ForceZero(token, sizeof(*token)); } #ifndef WOLFPKCS11_NO_STORE @@ -7637,7 +8870,35 @@ static int wp11_token_write_seed(void* storage, WP11_Token* token) } /** - * Load a token from storage. + * Check for an empty user PIN while loading, when the caller alone can reach + * the token, and cache the verdict as WP11_Slot_Has_Empty_Pin would. + * + * @param [in] slot Slot object. + * @param [in, out] token Token being loaded. + * @return 1 when the user PIN is set and empty, 0 otherwise. + */ +static int wp11_Token_HasEmptyPin(WP11_Slot* slot, WP11_Token* token) +{ + int ret = 0; + byte hash[PIN_HASH_SZ]; + + if ((token->tokenFlags & WP11_TOKEN_FLAG_USER_PIN_SET) != 0) { + if (HashPIN((char*)"", 0, token->userPinSeed, + sizeof(token->userPinSeed), hash, sizeof(hash), slot) == 0) { + ret = WP11_ConstantCompare(hash, token->userPin, + token->userPinLen) != 0; + } + token->userPinEmpty = ret ? 1 : 2; + wc_ForceZero(hash, sizeof(hash)); + } + + return ret; +} + +/** + * Load a token from storage. Runs without the slot lock from library + * initialization and under it from a token reset; neither its callees nor + * this function take the slot lock. * * @param [in, out] object Token object. * @param [in] tokenId Id of token this key belongs to. @@ -7655,10 +8916,20 @@ static int wp11_Token_Load(WP11_Slot* slot, int tokenId, WP11_Token* token) WP11_Object** current; int objCnt = 0; word32 len; +#ifdef WP11_STORE_BATCH + WP11_Slot* batchHeld = NULL; +#endif + ret = 0; +#ifdef WP11_STORE_BATCH + /* Finish or drop an interrupted pass, then keep passes out until read. */ + ret = wolfPKCS11_StoreBatchCheck(tokenId, &batchHeld); +#endif /* Open access to token object. */ - ret = wp11_storage_open_readonly(WOLFPKCS11_STORE_TOKEN, tokenId, 0, - &storage); + if (ret == 0) { + ret = wp11_storage_open_readonly(WOLFPKCS11_STORE_TOKEN, tokenId, 0, + &storage); + } if (ret == 0) { /* Read label for token. (32) */ ret = wp11_storage_read_string(storage, token->label, @@ -7754,9 +9025,9 @@ static int wp11_Token_Load(WP11_Slot* slot, int tokenId, WP11_Token* token) token->nextObjId = 1; ret = 0; } - else { + else if (ret == 0) { ret = wp11_storage_read_int(storage, &token->nextObjId); - if (ret == BUFFER_E || token->nextObjId == 0) { + if (ret == BUFFER_E || (ret == 0 && token->nextObjId == 0)) { token->nextObjId = 1; ret = 0; } @@ -7778,7 +9049,7 @@ static int wp11_Token_Load(WP11_Slot* slot, int tokenId, WP11_Token* token) } /* If there is no pin, there is no login, so decode now */ - if (WP11_Slot_Has_Empty_Pin(slot) && (ret == 0)) { + if (ret == 0 && wp11_Token_HasEmptyPin(slot, token)) { #ifndef WOLFPKCS11_NO_STORE /* Derive token->key from empty PIN + seed before decoding */ ret = HashPIN((char*)"", 0, token->seed, sizeof(token->seed), @@ -7799,6 +9070,9 @@ static int wp11_Token_Load(WP11_Slot* slot, int tokenId, WP11_Token* token) /* No data to read. */ ret = 0; } +#ifdef WP11_STORE_BATCH + wolfPKCS11_StoreBatchRelease(batchHeld); +#endif return ret; } @@ -7813,9 +9087,9 @@ static int wp11_Token_Load(WP11_Slot* slot, int tokenId, WP11_Token* token) * @return BUFFER_E when storing fails. * @return NOT_AVAILABLE_E when unable to write data. */ -static int wp11_Token_Store(WP11_Token* token, int tokenId) +static int wp11_Token_StorePass(WP11_Token* token, int tokenId) { - int ret; + int ret = 0; int i; void* storage = NULL; WP11_Object* object; @@ -7825,15 +9099,17 @@ static int wp11_Token_Store(WP11_Token* token, int tokenId) printf("wp11_Token_Store: tokenId %d, objCnt %d\n", tokenId, token->objCnt); #endif - /* Reserve space for token object */ +#ifdef WOLFPKCS11_TPM_STORE + /* Reserve NV space for token object before the objects use it. */ variableSz = token->userPinLen + token->soPinLen + (token->objCnt * FIELD_SIZE(WP11_Object, type)); ret = wp11_storage_open(WOLFPKCS11_STORE_TOKEN, tokenId, 0, variableSz, &storage); if (ret == 0) { - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); storage = NULL; } +#endif /* Store the objects */ object = token->object; @@ -7844,12 +9120,6 @@ static int wp11_Token_Store(WP11_Token* token, int tokenId) #ifdef DEBUG_WOLFPKCS11 printf("Failed to store object %d, token %d, ret: %d\n", i, tokenId, ret); - #endif - token->objCnt = i; /* mark number of objects actually stored */ - #ifdef WOLFPKCS11_TPM_STORE - if ((ret & RC_MAX_FM0) == TPM_RC_NV_SPACE) { - ret = 0; /* allow this error and continue */ - } #endif break; } @@ -7937,7 +9207,7 @@ static int wp11_Token_Store(WP11_Token* token, int tokenId) ret = wp11_storage_write_int(storage, token->nextObjId); } - wp11_storage_close(storage); + ret = wp11_storage_close_write(storage, ret); } else if (ret == NOT_AVAILABLE_E) { /* Not writing. */ @@ -7946,6 +9216,34 @@ static int wp11_Token_Store(WP11_Token* token, int tokenId) return ret; } + +/** + * Store the token and its objects as one pass that commits as a unit. + * + * @param [in] token Token object. + * @param [in] tokenId Id of token. + * @return 0 on success. + * @return Other value when the pass did not commit. + */ +static int wp11_Token_Store(WP11_Token* token, int tokenId) +{ + int ret; +#ifdef WP11_STORE_BATCH + int endRet; + + ret = wolfPKCS11_StoreBatchBegin(tokenId); + if (ret == 0) { + ret = wp11_Token_StorePass(token, tokenId); + endRet = wolfPKCS11_StoreBatchEnd(ret == 0); + if (ret == 0) + ret = endRet; + } +#else + ret = wp11_Token_StorePass(token, tokenId); +#endif + + return ret; +} #endif /* !WOLFPKCS11_NO_STORE */ /** @@ -7993,6 +9291,10 @@ static int wp11_TpmInit(WP11_Slot* slot) "FIPS 140-2 %d, CC-EAL4 %d\n", caps.mfgStr, caps.mfg, caps.vendorStr, caps.fwVerMajor, caps.fwVerMinor, caps.fwVerVendor, caps.fips140_2, caps.cc_eal4); + #ifdef WOLFPKCS11_TPM_STORE + /* Without the limit, an oversized record fails at commit instead. */ + (void)wolfPKCS11_Store_TpmReadLimits(); + #endif } if (ret == 0) { ret = wolfTPM2_SetCryptoDevCb(&slot->tpmDev, wolfTPM2_CryptoDevCb, @@ -8065,6 +9367,9 @@ static void wp11_Slot_Final(WP11_Slot* slot) wp11_Token_Final(&slot->token); #ifdef WOLFPKCS11_TPM wp11_TpmFinal(slot); +#endif +#ifndef WOLFPKCS11_NO_STORE + WP11_Lock_Free(&slot->storeLock); #endif WP11_Lock_Free(&slot->callLock); WP11_Lock_Free(&slot->lock); @@ -8100,6 +9405,15 @@ static int wp11_Slot_Init(WP11_Slot* slot, int id) WP11_Lock_Free(&slot->lock); } } +#ifndef WOLFPKCS11_NO_STORE + if (ret == 0) { + ret = WP11_Lock_Init(&slot->storeLock); + if (ret != 0) { + WP11_Lock_Free(&slot->callLock); + WP11_Lock_Free(&slot->lock); + } + } +#endif if (ret == 0) { #if defined(WOLFPKCS11_TPM) ret = wp11_TpmInit(slot); @@ -8174,6 +9488,10 @@ int WP11_Library_Init(void) { int ret = 0; int i; + int lockInit = 0; + int cryptInit = 0; + int rngInit = 0; + int slotInitCnt = 0; #ifdef WP11_HAVE_LIBRARY_INIT_LOCK /* Serialize the entire init sequence: both the libraryInitCount==0 @@ -8187,8 +9505,10 @@ int WP11_Library_Init(void) if (libraryInitCount == 0) { ret = WP11_Lock_Init(&globalLock); if (ret == 0) { - + lockInit = 1; ret = wolfCrypt_Init(); + if (ret == 0) + cryptInit = 1; #ifdef WC_RNG_SEED_CB if (ret == 0) { ret = wc_SetSeed_Cb(wc_GenerateSeed); @@ -8205,17 +9525,33 @@ int WP11_Library_Init(void) #else ret = wc_InitRng(&globalRandom); #endif + if (ret == 0) + rngInit = 1; } } for (i = 0; (ret == 0) && (i < slotCnt); i++) { ret = wp11_Slot_Init(&slotList[i], i + 1); + if (ret == 0) + slotInitCnt++; } #ifndef WOLFPKCS11_NO_STORE for (i = 0; (ret == 0) && (i < slotCnt); i++) { ret = wp11_Slot_Load(&slotList[i], i + 1); } #endif + if (ret != 0) { + /* Undo partial initialization so a retry starts clean. */ + for (i = 0; i < slotInitCnt; i++) + wp11_Slot_Final(&slotList[i]); + if (rngInit) + wc_FreeRng(&globalRandom); + if (cryptInit) + wolfCrypt_Cleanup(); + if (lockInit) { + WP11_Lock_Free(&globalLock); + } + } } if (ret == 0) { WP11_Lock_LockRW(&globalLock); @@ -8233,9 +9569,14 @@ int WP11_Library_Init(void) /** * Finalize the globals for the library. * Multiple finalizations allowed. + * + * @return 0 on success. + * @return BAD_MUTEX_E when locking fails. + * @return Other value when storing a token fails. Cleanup is still done. */ -void WP11_Library_Final(void) +int WP11_Library_Final(void) { + int ret = 0; int i; int cnt; @@ -8244,7 +9585,7 @@ void WP11_Library_Final(void) * WP11_Lock_Free so a racing WP11_Library_IsInitialized can't observe * count>0, then call WP11_Lock_LockRO on a freed globalLock. */ if (wc_LockMutex(&libraryInitLock) != 0) - return; + return BAD_MUTEX_E; #endif WP11_Lock_LockRW(&globalLock); @@ -8254,13 +9595,14 @@ void WP11_Library_Final(void) #ifndef WOLFPKCS11_NO_STORE /* Store the slots. */ for (i = 0; i < slotCnt; i++) { - int ret = wp11_Slot_Store(&slotList[i], i + 1); + int storeRet = wp11_Slot_Store(&slotList[i], i + 1); #ifdef DEBUG_WOLFPKCS11 - if (ret != 0) { - printf("Failed to store slot %d, ret: %d\n", i + 1, ret); + if (storeRet != 0) { + printf("Failed to store slot %d, ret: %d\n", i + 1, storeRet); } #endif - (void)ret; /* store failure cannot be returned, so log and ignore */ + if (ret == 0) + ret = storeRet; } #if !defined (WOLFPKCS11_CUSTOM_STORE) && defined(WOLFPKCS11_NSS) /* Serialize the free against a concurrent WP11_SetStoreDir / @@ -8293,6 +9635,8 @@ void WP11_Library_Final(void) #ifdef WP11_HAVE_LIBRARY_INIT_LOCK wc_UnLockMutex(&libraryInitLock); #endif + + return ret; } /** @@ -8667,6 +10011,9 @@ int WP11_Slot_TokenReset(WP11_Slot* slot, char* pin, int pinLen, char* label) { int ret; WP11_Token* token; +#ifdef WP11_STORE_BATCH + char emptyLabel[LABEL_SZ]; +#endif WP11_Lock_LockRW(&slot->lock); /* Zeroizes token. */ @@ -8681,8 +10028,25 @@ int WP11_Slot_TokenReset(WP11_Slot* slot, char* pin, int pinLen, char* label) WP11_Lock_UnlockRW(&slot->lock); /* Locking used in setting SO PIN. */ - if (ret == 0) + if (ret == 0) { ret = WP11_Slot_SetSOPin(slot, pin, pinLen); +#ifdef WP11_STORE_BATCH + /* Only a staged store pass is known to leave storage unchanged. */ + if (ret != 0) { + /* Reload under the lock so no session sees a half loaded token. */ + XMEMSET(emptyLabel, 0, sizeof(emptyLabel)); + WP11_Lock_LockRW(&slot->lock); + wp11_Token_Final(token); + if (wp11_Token_Init(token, emptyLabel) == 0 && + wp11_Slot_Load(slot, (int)slot->id) != 0) { + /* Storage still holds the token; leave a blank one here. */ + wp11_Token_Final(token); + (void)wp11_Token_Init(token, emptyLabel); + } + WP11_Lock_UnlockRW(&slot->lock); + } +#endif + } return ret; } @@ -9269,7 +10633,8 @@ void WP11_Slot_ObjectCallLeave(WP11_ObjectCalls* calls) * @param slot [in] Slot object. * @param pin [in] PIN to use to login. * @param pinLen [in] Length of PIN. - * @return READ_ONLY_E when there is a read-only session open. + * @return LOGGED_IN_E when the user is already logged in. + * LOGGED_IN_ANOTHER_E when the SO is logged in. * PIN_NOT_SET_E when the token is not initialized. * PIN_INVALID_E when the PIN is not correct or changed during login. * MEMORY_E when dynamic memory allocation fails. @@ -9448,30 +10813,63 @@ int WP11_Slot_SetSOPin(WP11_Slot* slot, char* pin, int pinLen) { int ret = 0; WP11_Token* token; + byte pinSeed[PIN_SEED_SZ]; + byte pinHash[PIN_HASH_SZ]; +#ifdef WP11_STORE_BATCH + byte oldPin[PIN_HASH_SZ]; + byte oldPinSeed[PIN_SEED_SZ]; + int oldPinLen; + int oldFlags; +#endif WP11_Lock_LockRW(&slot->lock); token = &slot->token; +#ifdef WP11_STORE_BATCH + XMEMCPY(oldPin, token->soPin, sizeof(oldPin)); + XMEMCPY(oldPinSeed, token->soPinSeed, sizeof(oldPinSeed)); + oldPinLen = token->soPinLen; + oldFlags = token->tokenFlags; +#endif /* New seed each time. */ WP11_Lock_LockRW(&slot->token.rngLock); - ret = wc_RNG_GenerateBlock(&slot->token.rng, token->soPinSeed, - sizeof(token->soPinSeed)); + ret = wc_RNG_GenerateBlock(&slot->token.rng, pinSeed, sizeof(pinSeed)); WP11_Lock_UnlockRW(&slot->token.rngLock); if (ret == 0) { WP11_Lock_UnlockRW(&slot->lock); /* Costly Operation done out of lock. */ - ret = HashPIN(pin, pinLen, token->soPinSeed, - sizeof(token->soPinSeed), token->soPin, - sizeof(token->soPin), slot); + ret = HashPIN(pin, pinLen, pinSeed, sizeof(pinSeed), pinHash, + sizeof(pinHash), slot); WP11_Lock_LockRW(&slot->lock); } if (ret == 0) { + #ifndef WOLFPKCS11_NO_STORE + /* Order is slot, token, storeLock: no store runs from here on. */ + WP11_Lock_LockRW(&token->lock); + #endif + XMEMCPY(token->soPinSeed, pinSeed, sizeof(token->soPinSeed)); + XMEMCPY(token->soPin, pinHash, sizeof(token->soPin)); token->soPinLen = sizeof(token->soPin); token->tokenFlags |= WP11_TOKEN_FLAG_SO_PIN_SET; #ifndef WOLFPKCS11_NO_STORE ret = wp11_Token_Store(token, (int)slot->id); + #ifdef WP11_STORE_BATCH + if (ret != 0 && (oldFlags & WP11_TOKEN_FLAG_SO_PIN_SET)) { + /* Keep the PIN that is still in storage in effect. */ + XMEMCPY(token->soPin, oldPin, sizeof(oldPin)); + XMEMCPY(token->soPinSeed, oldPinSeed, sizeof(oldPinSeed)); + token->soPinLen = oldPinLen; + token->tokenFlags = oldFlags; + } + #endif + WP11_Lock_UnlockRW(&token->lock); #endif } WP11_Lock_UnlockRW(&slot->lock); + wc_ForceZero(pinHash, sizeof(pinHash)); +#ifdef WP11_STORE_BATCH + wc_ForceZero(oldPin, sizeof(oldPin)); + wc_ForceZero(oldPinSeed, sizeof(oldPinSeed)); +#endif return ret; } @@ -9580,6 +10978,7 @@ static int wp11_Slot_SetUserPin(WP11_Slot* slot, char* oldPin, int oldPinLen, #ifndef WOLFPKCS11_NO_STORE WP11_ReKeyData* reKey = NULL; int reKeyCnt = 0; + WP11_Object* object; #endif #ifndef WOLFPKCS11_NO_TIME time_t now = 0; @@ -9689,6 +11088,15 @@ static int wp11_Slot_SetUserPin(WP11_Slot* slot, char* oldPin, int oldPinLen, sizeof(d->prev.seed)) != 0) { ret = BAD_STATE_E; } + /* Bind pending policy changes under the old key first, so every + * record re-keys from key data that verifies. */ + for (object = token->object; ret == 0 && object != NULL; + object = object->next) { + if (object->unbound && !object->encoded && + wp11_Object_IsEncrypted(object)) { + ret = wp11_Object_Encode(object, 0); + } + } if (ret == 0) { ret = wp11_Token_ReKeyPrepare(token, d->oldKey, d->key, &reKey, &reKeyCnt); @@ -9713,9 +11121,11 @@ static int wp11_Slot_SetUserPin(WP11_Slot* slot, char* oldPin, int oldPinLen, if (ret != 0) { wp11_Token_ReKeySwap(reKey, reKeyCnt); wp11_Token_RestoreUserPinState(token, &d->prev); + #ifndef WP11_STORE_BATCH /* Best effort to put the previous state back in storage. */ (void)wp11_Token_Store(token, (int)slot->id); token->objCnt = d->prev.objCnt; + #endif } #endif } @@ -9983,6 +11393,22 @@ void WP11_Slot_GetTokenLabel(WP11_Slot* slot, char* label) WP11_Lock_UnlockRO(&slot->lock); } +/** + * Replace NUL bytes in a fixed-length character field with blanks. + * + * @param field [in,out] Character field. + * @param len [in] Length of field in bytes. + */ +void WP11_BlankPad(CK_UTF8CHAR* field, int len) +{ + int i; + + for (i = 0; i < len; i++) { + if (field[i] == '\0') + field[i] = ' '; + } +} + /** * Check if token has been initialized. * @@ -10227,15 +11653,29 @@ static int wp11_init_get_op_category(int init) } /** - * Check whether the session has an active operation of the given category. - * Per PKCS#11 spec, only operations of the same type block re-initialization. + * Return whether the mechanism reads the key object each time data is + * processed rather than copying the key at initialization. + * + * @param mechanism [in] Mechanism of the operation. + * @return 1 when the key object is read during the operation. + * 0 otherwise. + */ +static int wp11_mech_reads_key(CK_MECHANISM_TYPE mechanism) +{ + return mechanism == CKM_AES_GCM || mechanism == CKM_AES_CCM || + mechanism == CKM_AES_ECB; +} + +/** + * Check whether the session holds operation state of the given category, even + * when the key the operation reads was destroyed. * * @param session [in] Session object. * @param opCategory [in] Operation category (WP11_OP_*). - * @return 1 when an operation of the same category is active. + * @return 1 when an operation of the category is initialized. * 0 otherwise. */ -int WP11_Session_IsOpCategoryActive(WP11_Session* session, int opCategory) +int WP11_Session_IsOpCategoryInit(WP11_Session* session, int opCategory) { int currentInit; @@ -10246,35 +11686,58 @@ int WP11_Session_IsOpCategoryActive(WP11_Session* session, int opCategory) return wp11_init_get_op_category(currentInit) == opCategory; } +/** + * Check whether the session has an active operation of the given category. + * Per PKCS#11 spec, only operations of the same type block re-initialization. + * + * @param session [in] Session object. + * @param opCategory [in] Operation category (WP11_OP_*). + * @return 1 when an operation of the same category is active. + * 0 otherwise. + */ +int WP11_Session_IsOpCategoryActive(WP11_Session* session, int opCategory) +{ + /* The key of this operation was destroyed so it can no longer run. */ + if (session->curr == NULL && wp11_mech_reads_key(session->mechanism)) + return 0; + + return WP11_Session_IsOpCategoryInit(session, opCategory); +} + int WP11_Session_UpdateData(WP11_Session *session, byte *data, word32 dataLen) { int ret = 0; byte* tmp; + word32 cap; /* Guard the cumulative word32 sum against silent wrap that would lead to * a tiny allocation followed by an oversized XMEMCPY past it. */ if (dataLen > (word32)0xFFFFFFFFu - session->dataSz) return MEMORY_E; -#ifdef XREALLOC - tmp = (byte*)XREALLOC(session->data, session->dataSz + dataLen, NULL, - DYNAMIC_TYPE_TMP_BUFFER); - if (tmp == NULL) - ret = MEMORY_E; -#else - tmp = (byte*)XMALLOC(session->dataSz + dataLen, NULL, - DYNAMIC_TYPE_TMP_BUFFER); - if (tmp == NULL) - ret = MEMORY_E; - if (ret == 0) { - if (session->data != NULL) - XMEMCPY(tmp, session->data, session->dataSz); - XFREE(session->data, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (session->data == NULL || dataLen > session->dataCap - session->dataSz) { + /* Doubling keeps the copying linear in the total input. */ + cap = session->dataSz + dataLen; + if (session->dataCap <= (word32)0x7FFFFFFFu && + cap < session->dataCap * 2) { + cap = session->dataCap * 2; + } + /* Grow by copying so the old buffer can be scrubbed before freeing. */ + tmp = (byte*)XMALLOC(cap, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (tmp == NULL) + ret = MEMORY_E; + if (ret == 0) { + if (session->data != NULL) { + XMEMCPY(tmp, session->data, session->dataSz); + wc_ForceZero(session->data, session->dataCap); + } + XFREE(session->data, NULL, DYNAMIC_TYPE_TMP_BUFFER); + session->data = tmp; + session->dataCap = cap; + } } -#endif /* !XREALLOC */ if (ret == 0) { - session->data = tmp; XMEMCPY(session->data + session->dataSz, data, dataLen); session->dataSz += dataLen; } @@ -10290,9 +11753,12 @@ void WP11_Session_GetData(WP11_Session *session, byte** data, word32* dataLen) void WP11_Session_FreeData(WP11_Session *session) { + if (session->data != NULL) + wc_ForceZero(session->data, session->dataCap); XFREE(session->data, NULL, DYNAMIC_TYPE_TMP_BUFFER); session->data = NULL; session->dataSz = 0; + session->dataCap = 0; session->init = 0; } @@ -10499,6 +11965,8 @@ CK_MECHANISM_TYPE WP11_Session_GetMechanism(WP11_Session* session) void WP11_Session_SetMechanism(WP11_Session* session, CK_MECHANISM_TYPE mechanism) { + /* Committing a new operation ends any operation it replaces. */ + WP11_Session_AbortOp(session); session->mechanism = mechanism; } @@ -10614,7 +12082,8 @@ static int wp11_mgf(CK_MECHANISM_TYPE mgfType, int *mgf) * @param labelSz [in] Length of data in bytes. * @return MEMORY_E when dynamic memory allocation fails. * BAD_FUNC_ARG when the digest algorithm id or the mask generation - * function id are not recognized. + * function id are not recognized, labelSz is negative, or label is + * NULL with a non-zero length. * 0 on success. */ int WP11_Session_SetOaepParams(WP11_Session* session, CK_MECHANISM_TYPE hashAlg, @@ -10622,14 +12091,24 @@ int WP11_Session_SetOaepParams(WP11_Session* session, CK_MECHANISM_TYPE hashAlg, { int ret; WP11_OaepParams* oaep = &session->params.oaep; + enum wc_HashType hashType; + int mgfType; + + ret = wp11_hash_type(hashAlg, &hashType); + if (ret == 0) + ret = wp11_mgf(mgf, &mgfType); + if (ret == 0 && (labelSz < 0 || (label == NULL && labelSz != 0))) + ret = BAD_FUNC_ARG; + if (ret != 0) + return ret; + WP11_Session_AbortOp(session); if (session->mechanism == CKM_RSA_PKCS_OAEP && oaep->label != NULL) { XFREE(oaep->label, NULL, DYNAMIC_TYPE_TMP_BUFFER); } XMEMSET(oaep, 0, sizeof(*oaep)); - ret = wp11_hash_type(hashAlg, &oaep->hashType); - if (ret == 0) - ret = wp11_mgf(mgf, &oaep->mgf); + oaep->hashType = hashType; + oaep->mgf = mgfType; if (ret == 0 && label == NULL) { oaep->label = NULL; oaep->labelSz = 0; @@ -10665,13 +12144,19 @@ int WP11_Session_SetPssParams(WP11_Session* session, CK_MECHANISM_TYPE hashAlg, { int ret; WP11_PssParams* pss = &session->params.pss; + enum wc_HashType hashType; + int mgfType; - XMEMSET(pss, 0, sizeof(*pss)); - ret = wp11_hash_type(hashAlg, &pss->hashType); - if (ret == 0) - ret = wp11_mgf(mgf, &pss->mgf); + ret = wp11_hash_type(hashAlg, &hashType); if (ret == 0) + ret = wp11_mgf(mgf, &mgfType); + if (ret == 0) { + WP11_Session_AbortOp(session); + XMEMSET(pss, 0, sizeof(*pss)); + pss->hashType = hashType; + pss->mgf = mgfType; pss->saltLen = sLen; + } return ret; } @@ -10692,7 +12177,8 @@ int WP11_Session_SetMldsaParams(WP11_Session* session, CK_VOID_PTR params, CK_ULONG paramsLen) { int ret = 0; - WP11_MldsaParams* mldsa = &session->params.mldsa; + WP11_MldsaParams newParams; + WP11_MldsaParams* mldsa = &newParams; XMEMSET(mldsa, 0, sizeof(*mldsa)); @@ -10754,6 +12240,11 @@ int WP11_Session_SetMldsaParams(WP11_Session* session, CK_VOID_PTR params, mldsa->ctxSz = 0; } + if (ret == 0) { + WP11_Session_AbortOp(session); + XMEMCPY(&session->params.mldsa, mldsa, sizeof(*mldsa)); + } + return ret; } #endif /* WOLFPKCS11_MLDSA */ @@ -10774,6 +12265,7 @@ int WP11_Session_SetCbcParams(WP11_Session* session, unsigned char* iv, int enc, WP11_Object* object) { int ret; + int aesInit = 0; WP11_CbcParams* cbc = &session->params.cbc; WP11_Data* key; @@ -10781,6 +12273,7 @@ int WP11_Session_SetCbcParams(WP11_Session* session, unsigned char* iv, * at allocation, so a prior operation can leave stale multi-part streaming * state here. Reset it before use (as the other Set*Params routines do) so * a fresh CBC operation cannot inherit a bogus partial-block count. */ + WP11_Session_AbortOp(session); cbc->partialSz = 0; cbc->finalReady = 0; XMEMSET(cbc->partial, 0, sizeof(cbc->partial)); @@ -10788,6 +12281,8 @@ int WP11_Session_SetCbcParams(WP11_Session* session, unsigned char* iv, /* AES object on session. */ ret = wc_AesInit(&cbc->aes, NULL, object->devId); + if (ret == 0) + aesInit = 1; #ifdef WOLFSSL_STM32U5_DHUK if (ret == 0 && object->dhukIvLen > 0) { ret = wc_Stm32_Aes_SetDHUK_IV(&cbc->aes, object->dhukIv, @@ -10804,6 +12299,9 @@ int WP11_Session_SetCbcParams(WP11_Session* session, unsigned char* iv, WP11_Lock_UnlockRO(object->lock); } + if (ret != 0 && aesInit) + wc_AesFree(&cbc->aes); + return ret; } #endif /* HAVE_AES_CBC */ @@ -10831,6 +12329,7 @@ int WP11_Session_SetCtrParams(WP11_Session* session, CK_ULONG ulCounterBits, if (ulCounterBits > 128 || ulCounterBits == 0) return BAD_FUNC_ARG; + WP11_Session_AbortOp(session); XMEMSET(ctr, 0, sizeof(*ctr)); ret = wc_AesInit(&ctr->aes, NULL, object->devId); if (ret == 0) { @@ -10840,6 +12339,8 @@ int WP11_Session_SetCtrParams(WP11_Session* session, CK_ULONG ulCounterBits, ret = wc_AesSetKey(&ctr->aes, key->data, key->len, cb, AES_ENCRYPTION); if (object->onToken) WP11_Lock_UnlockRO(object->lock); + if (ret != 0) + wc_AesFree(&ctr->aes); } if (ret == 0) { XMEMCPY(ctr->counter, cb, sizeof(ctr->counter)); @@ -10858,6 +12359,7 @@ int WP11_Session_SetAesWrapParams(WP11_Session* session, byte* iv, word32 ivLen, WP11_KeyWrapParams *wrap = &session->params.kw; WP11_Data *key; + WP11_Session_AbortOp(session); XMEMSET(wrap, 0, sizeof(*wrap)); ret = wc_AesInit(&wrap->aes, NULL, object->devId); if (ret == 0) { @@ -10868,6 +12370,8 @@ int WP11_Session_SetAesWrapParams(WP11_Session* session, byte* iv, word32 ivLen, enc ? AES_ENCRYPTION : AES_DECRYPTION); if (object->onToken) WP11_Lock_UnlockRO(object->lock); + if (ret != 0) + wc_AesFree(&wrap->aes); } if (ret == 0) { if (iv != NULL) @@ -10889,7 +12393,8 @@ int WP11_Session_SetAesWrapParams(WP11_Session* session, byte* iv, word32 ivLen, * @param aad [in] Additional authentication data. * @param aadSz [in] Length of additional authentication data. * @param tagBits [in] Number of bits to use as the authentication tag. - * @return BAD_FUNC_ARG if the IV/nonce or the tagBits are too big. + * @return BAD_FUNC_ARG if the IV/nonce, AAD length or tagBits are out of + * range. * Other -ve value on failure. * 0 on success. */ @@ -10900,7 +12405,8 @@ int WP11_Session_SetGcmParams(WP11_Session* session, unsigned char* iv, int ret = 0; WP11_GcmParams* gcm = &session->params.gcm; - if (tagBits > 128 || ivSz < 0 || ivSz > WP11_MAX_GCM_NONCE_SZ) + if (tagBits < 0 || tagBits > 128 || ivSz < 0 || + ivSz > WP11_MAX_GCM_NONCE_SZ || aadLen < 0) ret = BAD_FUNC_ARG; /* Caller-supplied IV pointer and length must agree: NULL pairs with 0 * and only with 0, and a non-NULL buffer must come with a positive @@ -10913,9 +12419,18 @@ int WP11_Session_SetGcmParams(WP11_Session* session, unsigned char* iv, ret = BAD_FUNC_ARG; if (ret == 0) { + WP11_Session_AbortOp(session); if (session->mechanism == CKM_AES_GCM && gcm->aad != NULL) { XFREE(gcm->aad, NULL, DYNAMIC_TYPE_TMP_BUFFER); } + if (session->mechanism == CKM_AES_GCM && gcm->enc != NULL) { + wc_ForceZero(gcm->enc, (word32)gcm->encSz); + XFREE(gcm->enc, NULL, DYNAMIC_TYPE_TMP_BUFFER); + } +#ifdef WOLFSSL_AESGCM_STREAM + if (session->mechanism == CKM_AES_GCM && gcm->streamInit) + wc_AesFree(&gcm->aes); +#endif XMEMSET(gcm, 0, sizeof(*gcm)); if (ivSz > 0) XMEMCPY(gcm->iv, iv, ivSz); @@ -10948,7 +12463,7 @@ int WP11_Session_SetGcmParams(WP11_Session* session, unsigned char* iv, * @param aad [in] Additional authentication data. * @param aadSz [in] Length of additional authentication data. * @param macSz [in] Length of the MAC in bytes. - * @return BAD_FUNC_ARG if the IV/nonce is too big. + * @return BAD_FUNC_ARG if the nonce, MAC or a length is out of range. * Other -ve value on failure. * 0 on success. */ @@ -10960,7 +12475,12 @@ int WP11_Session_SetCcmParams(WP11_Session* session, int dataSz, int ret = 0; WP11_CcmParams* ccm = &session->params.ccm; - if (ivSz < 0 || ivSz > WP11_MAX_GCM_NONCE_SZ) + if (ivSz < CCM_NONCE_MIN_SZ || ivSz > CCM_NONCE_MAX_SZ) + ret = BAD_FUNC_ARG; + /* CCM tags are an even number of bytes from 4 to 16. */ + if (ret == 0 && (macSz < 4 || macSz > AES_BLOCK_SIZE || (macSz & 1))) + ret = BAD_FUNC_ARG; + if (ret == 0 && (dataSz < 0 || aadSz < 0)) ret = BAD_FUNC_ARG; /* Caller-supplied IV pointer and length must agree: NULL pairs with 0 * and only with 0, and a non-NULL buffer must come with a positive @@ -10973,6 +12493,7 @@ int WP11_Session_SetCcmParams(WP11_Session* session, int dataSz, ret = BAD_FUNC_ARG; if (ret == 0) { + WP11_Session_AbortOp(session); if (session->mechanism == CKM_AES_CCM && ccm->aad != NULL) { XFREE(ccm->aad, NULL, DYNAMIC_TYPE_TMP_BUFFER); } @@ -11008,6 +12529,7 @@ int WP11_Session_SetCtsParams(WP11_Session* session, unsigned char* iv, WP11_CtsParams* cts = &session->params.cts; WP11_Data* key; + WP11_Session_AbortOp(session); /* AES object on session. */ ret = wc_AesInit(&cts->aes, NULL, object->devId); if (ret == 0) { @@ -11018,6 +12540,8 @@ int WP11_Session_SetCtsParams(WP11_Session* session, unsigned char* iv, enc ? AES_ENCRYPTION : AES_DECRYPTION); if (object->onToken) WP11_Lock_UnlockRO(object->lock); + if (ret != 0) + wc_AesFree(&cts->aes); } return ret; @@ -11139,6 +12663,11 @@ int WP11_Session_AddObject(WP11_Session* session, int onToken, object->handle = CK_INVALID_HANDLE; object->next = NULL; object->lock = NULL; + #ifdef WOLFPKCS11_TPM_STORE + /* Best effort: the failed store may have recreated the token + * NV index empty. The original error is returned. */ + (void)wp11_Slot_Store(session->slot, (int)session->slotId); + #endif } } #endif @@ -11212,6 +12741,9 @@ int WP11_Session_RemoveObjectByHandle(WP11_Session* session, WP11_Token* token = &session->slot->token; WP11_Session* owner = session; int id = 0; +#ifndef WOLFPKCS11_NO_STORE + int storeRet; +#endif /* The token lock guards every object list on the slot (session lists * included). Take it before touching any list or the object. Token objects @@ -11303,13 +12835,16 @@ int WP11_Session_RemoveObjectByHandle(WP11_Session* session, #ifndef WOLFPKCS11_NO_STORE /* remove any id's with higher value */ - ret = wp11_Object_Unstore(*curr, (int)session->slotId, id); + storeRet = wp11_Object_Unstore(*curr, (int)session->slotId, id); #ifdef DEBUG_WOLFPKCS11 - if (ret != 0) { + if (storeRet != 0) { printf("Failed to unstore slot %d, id: %d, ret: %d, continuing...\n", - (int)session->slotId, id, ret); + (int)session->slotId, id, storeRet); } #endif + /* NOT_AVAILABLE_E: storage is disabled, nothing to remove. */ + if (ret == 0 && storeRet != NOT_AVAILABLE_E) + ret = storeRet; #endif curr = &(*curr)->next; @@ -11318,21 +12853,26 @@ int WP11_Session_RemoveObjectByHandle(WP11_Session* session, } #ifndef WOLFPKCS11_NO_STORE - ret = wp11_Object_Unstore(object, (int)session->slotId, id); + storeRet = wp11_Object_Unstore(object, (int)session->slotId, id); #ifdef DEBUG_WOLFPKCS11 - if (ret != 0) { + if (storeRet != 0) { printf("Failed to unstore slot %d, id: %d, ret: %d\n", - (int)session->slotId, id, ret); + (int)session->slotId, id, storeRet); } #endif + if (ret == 0 && storeRet != NOT_AVAILABLE_E) + ret = storeRet; /* re-store the token */ - ret = wp11_Slot_Store(session->slot, (int)session->slotId); + storeRet = wp11_Slot_Store(session->slot, (int)session->slotId); #ifdef DEBUG_WOLFPKCS11 - if (ret != 0) { + if (storeRet != 0) { printf("Failed to store slot %d, ret: %d\n", - (int)session->slotId, ret); + (int)session->slotId, storeRet); } #endif + /* A token record left unwritten outranks a record left behind. */ + if (storeRet != 0) + ret = storeRet; #endif } else { @@ -11380,7 +12920,9 @@ void WP11_Session_RemoveObject(WP11_Session* session, WP11_Object* object) * operation context during their *Init and run to completion without referring * to the object again, so a caller may legitimately destroy the key before * finishing the operation (wolfCrypt's PKCS#11 layer does exactly this for - * multi-part HMAC). Those operations are left untouched. + * multi-part HMAC). Those operations are left untouched. AES-GCM, AES-CCM and + * AES-ECB read the key object when data is processed, so only their reference + * is dropped; the owning session releases the operation state itself. * * Asymmetric operations dereference the key object while running, so for those * key types drop the active reference in every session that holds it and reset @@ -11393,18 +12935,22 @@ void WP11_Slot_ClearActiveObject(WP11_Slot* slot, WP11_Object* object) { WP11_Session* curr; CK_KEY_TYPE keyType; + int symmetric; if (slot == NULL || object == NULL) return; keyType = WP11_Object_GetType(object); - if (keyType == CKK_AES || keyType == CKK_GENERIC_SECRET || - keyType == CKK_HKDF) - return; + symmetric = (keyType == CKK_AES || keyType == CKK_GENERIC_SECRET || + keyType == CKK_HKDF); WP11_Lock_LockRW(&slot->lock); for (curr = slot->session; curr != NULL; curr = curr->next) { - if (curr->curr == object) { + if (curr->curr == object && symmetric) { + if (wp11_mech_reads_key(curr->mechanism)) + curr->curr = NULL; + } + else if (curr->curr == object) { #if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) /* Release the operation-owned RSA-OAEP label before dropping the * operation state. Otherwise a later init for a different mechanism @@ -11432,6 +12978,11 @@ void WP11_Slot_ClearActiveObject(WP11_Slot* slot, WP11_Object* object) */ void WP11_Session_GetObject(WP11_Session* session, WP11_Object** object) { + /* The key was destroyed: release the state its operation still holds. */ + if (session->curr == NULL && wp11_mech_reads_key(session->mechanism) && + (session->init & ~WP11_INIT_DIGEST_MASK) != 0) { + WP11_Session_AbortOp(session); + } *object = session->curr; } @@ -11694,6 +13245,9 @@ int WP11_Session_IsFindActive(WP11_Session* session) void WP11_Object_Free(WP11_Object* object) { int certFreed = 0; + + if (object == NULL) + return; #ifdef WOLFPKCS11_TPM if (object->tpmKey != NULL) { wolfTPM2_UnloadHandle(&object->slot->tpmDev, &object->tpmKey->handle); @@ -12013,7 +13567,10 @@ int WP11_Object_SetRsaKey(WP11_Object* object, unsigned char** data, } } if (ret == 0) { - if (len[8] == sizeof(CK_ULONG)) + if (len[8] == sizeof(CK_ULONG) && + *(CK_ULONG*)data[8] > (CK_ULONG)RSA_MAX_SIZE) + ret = BAD_FUNC_ARG; + else if (len[8] == sizeof(CK_ULONG)) object->size = (word32)*(CK_ULONG*)data[8]; else if (len[8] != 0) ret = BUFFER_E; @@ -12172,6 +13729,11 @@ int WP11_Object_SetEcKey(WP11_Object* object, unsigned char** data, if (ret == 0) { if (ret == 0 && data[0] != NULL) ret = EcSetParams(key, data[0], (int)len[0]); + /* Key material is only meaningful on a known curve. */ + if (ret == 0 && key->dp == NULL && + (data[1] != NULL || data[2] != NULL)) { + ret = BAD_FUNC_ARG; + } if (ret == 0 && data[1] != NULL) { key->type = ECC_PRIVATEKEY_ONLY; #if defined(HAVE_FIPS_VERSION) && (HAVE_FIPS_VERSION <= 5) @@ -12377,13 +13939,20 @@ int WP11_Object_SetDhKey(WP11_Object* object, unsigned char** data, key = object->data.dhKey; ret = wc_InitDhKey_ex(&key->params, NULL, object->devId); if (ret == 0) { - if (data[0] != NULL && data[1] != NULL) + /* Parameters larger than any storable DH key are not supported. */ + if (data[0] != NULL && data[1] != NULL && + (len[0] > WP11_MAX_DH_KEY_SZ || len[1] > WP11_MAX_DH_KEY_SZ)) { + ret = BAD_FUNC_ARG; + } + else if (data[0] != NULL && data[1] != NULL) { ret = wc_DhSetKey(&key->params, data[0], (int)len[0], data[1], (int)len[1]); + } if (ret == 0 && data[2] != NULL) { if (len[2] > (int)sizeof(key->key)) ret = BAD_FUNC_ARG; else { + wc_ForceZero(key->key, sizeof(key->key)); XMEMCPY(key->key, data[2], len[2]); key->len = (word32)len[2]; } @@ -12532,13 +14101,12 @@ int WP11_Object_SetSecretKey(WP11_Object* object, unsigned char** data, int ret = 0; WP11_Data* key; CK_ULONG keyLen = 0; + word32 newLen = 0; if (object->onToken) WP11_Lock_LockRW(object->lock); key = object->data.symmKey; - key->len = 0; - wc_ForceZero(key->data, sizeof(key->data)); /* First item is the key's length. */ if (ret == 0 && data[0] != NULL && len[0] != (int)sizeof(CK_ULONG)) @@ -12562,28 +14130,27 @@ int WP11_Object_SetSecretKey(WP11_Object* object, unsigned char** data, } #endif if (ret == 0 && data[0] != NULL) - key->len = (word32)keyLen; + newLen = (word32)keyLen; /* Second item is the key data. */ if (ret == 0 && data[1] != NULL) { - if (key->len == 0) { + if (newLen == 0) { if (len[1] > WP11_MAX_SYM_KEY_SZ) ret = BUFFER_E; else - key->len = (word32)len[1]; + newLen = (word32)len[1]; } - else if (len[1] != (CK_ULONG)key->len) + else if (len[1] != (CK_ULONG)newLen) ret = BUFFER_E; } - if (ret == 0 && key->len > WP11_MAX_SYM_KEY_SZ) - ret = BUFFER_E; - if (ret == 0 && data[1] != NULL) - XMEMCPY(key->data, data[1], key->len); - /* On any error, record no length so later teardown does not zeroize or - * copy past the fixed key buffer. */ - if (ret != 0) - key->len = 0; + /* Replace the key only once the new value is known to be valid. */ + if (ret == 0) { + wc_ForceZero(key->data, sizeof(key->data)); + key->len = newLen; + if (data[1] != NULL) + XMEMCPY(key->data, data[1], newLen); + } if (object->onToken) WP11_Lock_UnlockRW(object->lock); @@ -12693,12 +14260,12 @@ int WP11_Object_SetCert(WP11_Object* object, unsigned char** data, { int ret = 0; WP11_Cert* cert; + byte* newData = NULL; if (object->onToken) WP11_Lock_LockRW(object->lock); cert = &object->data.cert; - cert->len = 0; /* First item is certificate type */ if (ret == 0 && data[0] != NULL && len[0] != (int)sizeof(CK_ULONG)) @@ -12706,21 +14273,28 @@ int WP11_Object_SetCert(WP11_Object* object, unsigned char** data, if (ret == 0 && data[0] != NULL) cert->type = (word32)*(CK_ULONG*)data[0]; - /* Second item is certificate data (CKA_VALUE) */ + /* Second item is certificate data (CKA_VALUE). Copy it before releasing + * the current value so a failed allocation leaves the object intact. */ if (ret == 0 && data[1] != NULL) { - cert->len = (word32)len[1]; + newData = (byte*)XMALLOC(len[1], NULL, DYNAMIC_TYPE_CERT); + if (newData == NULL) + ret = MEMORY_E; + else + XMEMCPY(newData, data[1], len[1]); } if (ret == 0 && data[1] != NULL) { +#ifndef WOLFPKCS11_NO_STORE + /* A loaded certificate's value is the storage buffer itself. */ + if (cert->data != NULL && cert->data == object->keyData) { + object->keyData = NULL; + object->keyDataLen = 0; + } +#endif if (cert->data != NULL) { XFREE(cert->data, NULL, DYNAMIC_TYPE_CERT); } - cert->data = (byte *)XMALLOC(cert->len, NULL, DYNAMIC_TYPE_CERT); - if (cert->data == NULL) { - ret = MEMORY_E; - } - } - if (ret == 0 && data[1] != NULL) { - XMEMCPY(cert->data, data[1], cert->len); + cert->data = newData; + cert->len = (word32)len[1]; } if (object->onToken) @@ -13058,6 +14632,9 @@ static int GetCertAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, byte* data, case CKA_SUBJECT: ret = GetData(object->subject, object->subjectLen, data, len); break; + case CKA_CERTIFICATE_CATEGORY: + ret = GetULong(object->category, data, len); + break; #ifdef WOLFPKCS11_NSS case CKA_NSS_EMAIL: ret = GetData(object->email, object->emailLen, data, len); @@ -13111,14 +14688,14 @@ static int GetTrustAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, switch (type) { case CKA_CERT_SHA1_HASH: - if (*len < WC_SHA_DIGEST_SIZE) + if (data != NULL && *len < WC_SHA_DIGEST_SIZE) return BUFFER_E; *len = WC_SHA_DIGEST_SIZE; if (data != NULL) XMEMCPY(data, &object->data.trust.sha1Hash, WC_SHA_DIGEST_SIZE); break; case CKA_CERT_MD5_HASH: - if (*len < WC_MD5_DIGEST_SIZE) + if (data != NULL && *len < WC_MD5_DIGEST_SIZE) return BUFFER_E; *len = WC_MD5_DIGEST_SIZE; if (data != NULL) @@ -13270,17 +14847,22 @@ static int GetEcParams(ecc_key* key, byte* data, CK_ULONG* len) { int ret = 0; #if defined(HAVE_OID_ENCODING) - word32 dataLen = (word32)*len; - byte* out = (data != NULL) ? (data + 2) : NULL; - ret = wc_EncodeObjectId(key->dp->oid, key->dp->oidSz, out, &dataLen); - if (ret != 0) { - return ret; - } - if (data != NULL) { - data[0] = ASN_OBJECT_ID; - data[1] = dataLen; + word32 dataLen = 0; + + ret = wc_EncodeObjectId(key->dp->oid, key->dp->oidSz, NULL, &dataLen); + if (ret == 0 && data != NULL) { + if (*len < (CK_ULONG)dataLen + 2) + ret = BUFFER_E; + else + ret = wc_EncodeObjectId(key->dp->oid, key->dp->oidSz, data + 2, + &dataLen); + if (ret == 0) { + data[0] = ASN_OBJECT_ID; + data[1] = (byte)dataLen; + } } - *len = dataLen + 2; + if (ret == 0) + *len = (CK_ULONG)dataLen + 2; #else const byte* oid = NULL; word32 oidSz = 0; @@ -13385,8 +14967,10 @@ static int EcObject_GetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, #endif break; case CKA_EC_POINT: - if (noPub) + if (noPub) { *len = CK_UNAVAILABLE_INFORMATION; + ret = NOT_AVAILABLE_E; + } else ret = GetEcPoint(object->data.ecKey, data, len); break; @@ -13429,8 +15013,10 @@ static int GetMldsaParams(wc_MlDsaKey* key, byte* data, CK_ULONG* len) *len = sizeof(CK_ML_DSA_PARAMETER_SET_TYPE); else if (*len < sizeof(CK_ML_DSA_PARAMETER_SET_TYPE)) ret = BUFFER_E; - else + else { XMEMCPY(data, ¶ms, sizeof(CK_ML_DSA_PARAMETER_SET_TYPE)); + *len = sizeof(CK_ML_DSA_PARAMETER_SET_TYPE); + } return ret; } @@ -13530,6 +15116,7 @@ static int MldsaObject_GetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, break; case CKA_SEED: *len = CK_UNAVAILABLE_INFORMATION; + ret = NOT_AVAILABLE_E; break; case CKA_VALUE: if (object->objClass == CKO_PRIVATE_KEY) { @@ -13541,8 +15128,10 @@ static int MldsaObject_GetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, ret = GetMldsaPrivateKey(object->data.mldsaKey, data, len); } else if (object->objClass == CKO_PUBLIC_KEY) { - if (noPub) + if (noPub) { *len = CK_UNAVAILABLE_INFORMATION; + ret = NOT_AVAILABLE_E; + } else ret = GetMldsaPublicKey(object->data.mldsaKey, data, len); } @@ -13717,6 +15306,7 @@ static int MlKemObject_GetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, break; case CKA_SEED: *len = CK_UNAVAILABLE_INFORMATION; + ret = NOT_AVAILABLE_E; break; case CKA_VALUE: if (object->objClass == CKO_PRIVATE_KEY) { @@ -13728,8 +15318,10 @@ static int MlKemObject_GetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, ret = GetMlKemPrivateKey(object->data.mlKemKey, data, len); } else if (object->objClass == CKO_PUBLIC_KEY) { - if (noPub) + if (noPub) { *len = CK_UNAVAILABLE_INFORMATION; + ret = NOT_AVAILABLE_E; + } else ret = GetMlKemPublicKey(object->data.mlKemKey, data, len); } @@ -14114,6 +15706,12 @@ int WP11_Object_GetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, byte* data, #endif else if (object->objClass == CKO_DATA) ret = GetDataAttr(object, type, data, len); + else if (type == CKA_SUBJECT && + (object->objClass == CKO_PUBLIC_KEY || + object->objClass == CKO_PRIVATE_KEY)) { + ret = GetData(object->subject, object->subjectLen, data, + len); + } else { switch (object->type) { #ifndef NO_RSA @@ -14205,26 +15803,32 @@ static void WP11_Object_SetOpFlag(WP11_Object* object, word32 flag, int set) * @param object [in] Object object. * @param keyId [in] Key identifier data. * @param keyIdLen [in] Length of key identifier in bytes. - * @return MEMORY_E when dynamic memory allocation fails. + * @return BAD_FUNC_ARG when keyId is NULL and keyIdLen is not zero. + * MEMORY_E when dynamic memory allocation fails. * 0 on success. */ static int WP11_Object_SetKeyId(WP11_Object* object, unsigned char* keyId, int keyIdLen) { int ret = 0; + unsigned char* newId = NULL; + + if (keyId == NULL && keyIdLen > 0) + return BAD_FUNC_ARG; - XFREE(object->keyId, NULL, DYNAMIC_TYPE_TMP_BUFFER); - object->keyId = NULL; - object->keyIdLen = 0; + /* Build the replacement first so a failure keeps the current value. */ if (keyIdLen > 0) { - object->keyId = (unsigned char*)XMALLOC(keyIdLen, NULL, + newId = (unsigned char*)XMALLOC(keyIdLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); - if (object->keyId == NULL) + if (newId == NULL) ret = MEMORY_E; - if (ret == 0) { - XMEMCPY(object->keyId, keyId, keyIdLen); - object->keyIdLen = keyIdLen; - } + else + XMEMCPY(newId, keyId, keyIdLen); + } + if (ret == 0) { + XFREE(object->keyId, NULL, DYNAMIC_TYPE_TMP_BUFFER); + object->keyId = newId; + object->keyIdLen = (newId != NULL) ? keyIdLen : 0; } return ret; @@ -14237,26 +15841,31 @@ static int WP11_Object_SetKeyId(WP11_Object* object, unsigned char* keyId, * @param attributeLen [out] Length of attribute in bytes. * @param data [in] Data to be set. * @param dataLen [in] Length of data in bytes. - * @return MEMORY_E when dynamic memory allocation fails. + * @return BAD_FUNC_ARG when data is NULL and dataLen is not zero. + * MEMORY_E when dynamic memory allocation fails. * 0 on success. */ static int WP11_Object_SetData(byte** attribute, int* attributeLen, byte* data, int dataLen) { int ret = 0; + byte* newData = NULL; - XFREE(*attribute, NULL, DYNAMIC_TYPE_TMP_BUFFER); - *attribute = NULL; - *attributeLen = 0; + if (data == NULL && dataLen > 0) + return BAD_FUNC_ARG; + + /* Build the replacement first so a failure keeps the current value. */ if (dataLen > 0) { - *attribute = (byte*)XMALLOC(dataLen, NULL, - DYNAMIC_TYPE_TMP_BUFFER); - if (*attribute == NULL) + newData = (byte*)XMALLOC(dataLen, NULL, DYNAMIC_TYPE_TMP_BUFFER); + if (newData == NULL) ret = MEMORY_E; - if (ret == 0) { - XMEMCPY(*attribute, data, dataLen); - *attributeLen = dataLen; - } + else + XMEMCPY(newData, data, dataLen); + } + if (ret == 0) { + XFREE(*attribute, NULL, DYNAMIC_TYPE_TMP_BUFFER); + *attribute = newData; + *attributeLen = (newData != NULL) ? dataLen : 0; } return ret; @@ -14267,7 +15876,7 @@ static int WP11_Object_SetData(byte** attribute, int* attributeLen, byte* data, * * @param object [in] Object object. * @param startDate [in] Start data as a string. - * @param len [in] Length of string. + * @param len [in] Length of string. Zero clears the date. * @return BUFFER_E when string is too small. * 0 on success. */ @@ -14276,9 +15885,11 @@ static int WP11_Object_SetStartDate(WP11_Object* object, char* startDate, { int ret = 0; - if (len != sizeof(object->startDate)) + if (len == 0) + XMEMSET(object->startDate, 0, sizeof(object->startDate)); + else if (len != sizeof(object->startDate)) ret = BUFFER_E; - if (ret == 0) + else XMEMCPY(object->startDate, startDate, sizeof(object->startDate)); return ret; @@ -14289,7 +15900,7 @@ static int WP11_Object_SetStartDate(WP11_Object* object, char* startDate, * * @param object [in] Object object. * @param endData [in] End data as a string. - * @param len [in] Length of string. + * @param len [in] Length of string. Zero clears the date. * @return BUFFER_E when string is too small. * 0 on success. */ @@ -14297,14 +15908,51 @@ static int WP11_Object_SetEndDate(WP11_Object* object, char* endDate, int len) { int ret = 0; - if (len != sizeof(object->endDate)) + if (len == 0) + XMEMSET(object->endDate, 0, sizeof(object->endDate)); + else if (len != sizeof(object->endDate)) ret = BUFFER_E; - if (ret == 0) + else XMEMCPY(object->endDate, endDate, sizeof(object->endDate)); return ret; } +#ifndef WOLFPKCS11_NO_STORE +/** + * Track a change to a token object's policy so its key data is bound to the + * new policy when next stored. The change is undone when the key data is not + * decoded, as it could not be bound. Caller holds the object's lock when it + * is on a token. + * + * @param object [in] Object object. + * @param prevClass [in] Object class before the change. + * @param prevFlag [in] Operational flags before the change. + * @return BAD_STATE_E when the key data is not decoded. + * 0 on success. + */ +static int wp11_Object_PolicyChanged(WP11_Object* object, + CK_OBJECT_CLASS prevClass, word32 prevFlag) +{ + int ret = 0; + + if (object->onToken && wp11_Object_IsEncrypted(object) && + (object->objClass != prevClass || + ((object->opFlag ^ prevFlag) & + ~(word32)WP11_FLAG_POLICY_EXCLUDED) != 0)) { + if (object->encoded || object->undecoded) { + object->objClass = prevClass; + object->opFlag = prevFlag; + ret = BAD_STATE_E; + } + else { + object->unbound = 1; + } + } + + return ret; +} +#endif /** * Set an attribute against the object. @@ -14315,15 +15963,24 @@ static int WP11_Object_SetEndDate(WP11_Object* object, char* endDate, int len) * @param len [in] Length of attribute data in bytes. * @return BUFFER_E when data is too small. * BAD_FUNC_ARG when type is not supported with object. + * BAD_STATE_E when a policy change cannot be bound to the stored key. * 0 on success. */ int WP11_Object_SetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, byte* data, CK_ULONG len) { int ret = 0; +#ifndef WOLFPKCS11_NO_STORE + CK_OBJECT_CLASS prevClass; + word32 prevFlag; +#endif if (object->onToken) WP11_Lock_LockRW(object->lock); +#ifndef WOLFPKCS11_NO_STORE + prevClass = object->objClass; + prevFlag = object->opFlag; +#endif switch (type) { case CKA_CLASS: @@ -14375,7 +16032,7 @@ int WP11_Object_SetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, byte* data, (int)len); break; case CKA_CERTIFICATE_CATEGORY: - object->category = *(word32*)data; + object->category = (word32)*(CK_ULONG*)data; break; case CKA_PRIVATE: WP11_Object_SetOpFlag(object, WP11_FLAG_PRIVATE, *(CK_BBOOL*)data); @@ -14543,12 +16200,23 @@ int WP11_Object_SetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, byte* data, /* Handled in layer above */ break; case CKA_ISSUER: - ret = WP11_Object_SetData(&object->issuer, &object->issuerLen, - data, (int)len); - break; case CKA_SERIAL_NUMBER: - ret = WP11_Object_SetData(&object->serial, &object->serialLen, - data, (int)len); + /* Certificate attributes; NSS trust objects carry them too. */ + if (object->objClass != CKO_CERTIFICATE +#ifdef WOLFPKCS11_NSS + && object->objClass != CKO_NSS_TRUST +#endif + ) { + ret = BAD_FUNC_ARG; + } + else if (type == CKA_ISSUER) { + ret = WP11_Object_SetData(&object->issuer, &object->issuerLen, + data, (int)len); + } + else { + ret = WP11_Object_SetData(&object->serial, &object->serialLen, + data, (int)len); + } break; case CKA_SUBJECT: ret = WP11_Object_SetData(&object->subject, &object->subjectLen, @@ -14698,6 +16366,10 @@ int WP11_Object_SetAttr(WP11_Object* object, CK_ATTRIBUTE_TYPE type, byte* data, ret = BAD_FUNC_ARG; break; } +#ifndef WOLFPKCS11_NO_STORE + if (ret == 0) + ret = wp11_Object_PolicyChanged(object, prevClass, prevFlag); +#endif if (object->onToken) WP11_Lock_UnlockRW(object->lock); @@ -16255,8 +17927,10 @@ int WP11_EC_Derive(unsigned char* point, word32 pointLen, unsigned char* key, WC_RNG rng; #endif - if (priv->onToken) - WP11_Lock_LockRO(priv->lock); + /* Exclusive: the temporary RNG is stored in the shared key. */ + if (priv->onToken) { + WP11_Lock_LockRW(priv->lock); + } /* Check if the point data is DER-encoded (starts with OCTET STRING tag) */ if (pointLen >= 3 && point[0] == ASN_OCTET_STRING) { @@ -16270,9 +17944,10 @@ int WP11_EC_Derive(unsigned char* point, word32 pointLen, unsigned char* key, /* Not valid DER encoding, treat as raw X9.63 data */ x963Data = point; x963Len = pointLen; + i = 0; } } - if (i < (int)pointLen) { + if (i > 0 && i < (int)pointLen) { expectedPointLen = priv->data.ecKey->dp->size * 2 + 1; if (pointLen > expectedPointLen) { x963Len = point[i++]; @@ -16287,6 +17962,16 @@ int WP11_EC_Derive(unsigned char* point, word32 pointLen, unsigned char* key, } ret = wc_ecc_init_ex(&pubKey, NULL, priv->devId); + if (ret == 0 && x963Len > pointLen - (word32)(x963Data - point)) + ret = BUFFER_E; + if (ret == 0 && priv->data.ecKey->dp != NULL) { + /* Compressed point is size + 1 bytes, uncompressed 2 * size + 1 */ + expectedPointLen = (word32)priv->data.ecKey->dp->size + 1; + if (x963Len > 0 && x963Data[0] == ECC_POINT_UNCOMP) + expectedPointLen += (word32)priv->data.ecKey->dp->size; + if (x963Len != expectedPointLen) + ret = ECC_BAD_ARG_E; + } if (ret == 0) { if (priv->data.ecKey->dp) { ret = wc_ecc_import_x963_ex(x963Data, x963Len, &pubKey, @@ -16347,12 +18032,14 @@ int WP11_EC_Derive(unsigned char* point, word32 pointLen, unsigned char* key, } #if defined(ECC_TIMING_RESISTANT) && (!defined(HAVE_FIPS) || \ (defined(HAVE_FIPS_VERSION) && (HAVE_FIPS_VERSION > 2))) + (void)wc_ecc_set_rng(priv->data.ecKey, NULL); Rng_Free(&rng); #endif } - if (priv->onToken) - WP11_Lock_UnlockRO(priv->lock); + if (priv->onToken) { + WP11_Lock_UnlockRW(priv->lock); + } wc_ecc_free(&pubKey); @@ -16649,6 +18336,15 @@ int WP11_KDF_Derive(WP11_Session* session, CK_HKDF_PARAMS_PTR params, ret = WP11_Object_Find(session, params->hSaltKey, &saltKey); if (ret != 0) return CKR_OBJECT_HANDLE_INVALID; + /* Only these secret key types hold their value in symmKey. */ + if (saltKey->objClass != CKO_SECRET_KEY || + (saltKey->type != CKK_GENERIC_SECRET && +#ifndef NO_AES + saltKey->type != CKK_AES && +#endif + saltKey->type != CKK_HKDF) || + saltKey->data.symmKey == NULL) + return CKR_MECHANISM_PARAM_INVALID; salt = saltKey->data.symmKey->data; saltLen = saltKey->data.symmKey->len; break; @@ -17749,6 +19445,8 @@ int WP11_AesCtr_Final(WP11_Session* session) WP11_CtrParams* ctr = &session->params.ctr; wc_AesFree(&ctr->aes); + /* Older wolfSSL versions do not zeroize the key schedule in wc_AesFree. */ + wc_ForceZero(ctr, sizeof(*ctr)); session->init = 0; return 0; @@ -17865,6 +19563,10 @@ static int wp11_AesGcm_BufferAppend(WP11_GcmParams* gcm, unsigned char* data, { unsigned char* newBuf; + /* The cached length is an int; never let it grow past INT_MAX. */ + if (dataSz > (word32)(INT_MAX - gcm->encSz)) + return BAD_FUNC_ARG; + #ifdef XREALLOC newBuf = (unsigned char*)XREALLOC(gcm->enc, gcm->encSz + dataSz, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -18161,6 +19863,10 @@ int WP11_AesGcm_DecryptUpdate(unsigned char* enc, word32 encSz, unsigned char* newEnc; WP11_GcmParams* gcm = &session->params.gcm; + /* The cached length is an int; never let it grow past INT_MAX. */ + if (encSz > (word32)(INT_MAX - gcm->encSz)) + return BAD_FUNC_ARG; + #ifdef XREALLOC newEnc = (unsigned char*)XREALLOC(gcm->enc, gcm->encSz + encSz, NULL, DYNAMIC_TYPE_TMP_BUFFER); @@ -18885,6 +20591,21 @@ int WP11_AesCts_DecryptFinal(unsigned char* dec, word32* decSz, return ret; } + +/** + * Abandon an AES-CTS operation, releasing the AES object. + * + * @param session [in] Session object holding Aes object. + */ +void WP11_AesCts_Final(WP11_Session* session) +{ + WP11_CtsParams* cts = &session->params.cts; + + wc_AesFree(&cts->aes); + /* Older wolfSSL versions do not zeroize the key schedule in wc_AesFree. */ + wc_ForceZero(cts, sizeof(*cts)); + session->init = 0; +} #endif /* HAVE_AESCTS */ #ifdef HAVE_AESCMAC @@ -18897,6 +20618,7 @@ int WP11_Aes_Cmac_Init(WP11_Object* secret, WP11_Session* session, if (sigLen > WC_CMAC_TAG_MAX_SZ || sigLen < WC_CMAC_TAG_MIN_SZ) return BAD_FUNC_ARG; + WP11_Session_AbortOp(session); cmac->sigLen = (byte)sigLen; if (secret->onToken) WP11_Lock_LockRO(secret->lock); @@ -19081,6 +20803,7 @@ int WP11_Digest_Init(CK_MECHANISM_TYPE mechanism, WP11_Session* session) ret = wp11_digest_hash_type(mechanism, &hashType); if (ret == 0) { + WP11_Session_AbortOp(session); digest->hashType = (enum wc_HashType)hashType; ret = wc_HashInit(&digest->hash, (enum wc_HashType)hashType); } @@ -19307,12 +21030,14 @@ int WP11_Hmac_Init(CK_MECHANISM_TYPE mechanism, WP11_Object* secret, WP11_Data* key; ret = wp11_hmac_hash_type(mechanism, &hashType); - if (ret == 0) - hmac->hmacSz = wc_HmacSizeByType(hashType); - if (ret == 0 && digestSize != 0 && hmac->hmacSz != (word32)digestSize) + if (ret == 0 && digestSize != 0 && + (word32)wc_HmacSizeByType(hashType) != (word32)digestSize) ret = BAD_FUNC_ARG; - if (ret == 0) + if (ret == 0) { + WP11_Session_AbortOp(session); + hmac->hmacSz = wc_HmacSizeByType(hashType); ret = wc_HmacInit(&hmac->hmac, NULL, secret->slot->devId); + } if (ret == 0) { if (secret->onToken) WP11_Lock_LockRO(secret->lock); @@ -19320,6 +21045,8 @@ int WP11_Hmac_Init(CK_MECHANISM_TYPE mechanism, WP11_Object* secret, ret = wc_HmacSetKey(&hmac->hmac, hashType, key->data, key->len); if (secret->onToken) WP11_Lock_UnlockRO(secret->lock); + if (ret != 0) + wc_HmacFree(&hmac->hmac); } return ret; @@ -19463,6 +21190,9 @@ int WP11_TLS_MAC_init(unsigned long hashType, unsigned long macLen, byte server, int ret = 0; WP11_TlsMacParams *mac = &session->params.tlsMac; + if (hashType != CKM_TLS_PRF && MechToMac(hashType) == no_mac) + return BAD_FUNC_ARG; + WP11_Session_AbortOp(session); XMEMSET(mac, 0, sizeof(*mac)); if (hashType == CKM_TLS_PRF) { mac->isTlsPrf = 1; @@ -19617,7 +21347,10 @@ int WP11_GetOperationState(WP11_Session* session, unsigned char* stateData, wc_Hashes* hashAlg; #endif - *stateDataLen = sizeof(session->mechanism); + if (session->init == 0) + return CKR_OPERATION_NOT_INITIALIZED; + if (!WP11_Session_IsOpInitialized(session, WP11_INIT_DIGEST)) + return CKR_STATE_UNSAVEABLE; switch (session->mechanism) { #ifndef NO_MD5 @@ -19653,7 +21386,7 @@ int WP11_GetOperationState(WP11_Session* session, unsigned char* stateData, default: return CKR_STATE_UNSAVEABLE; } - *stateDataLen += mechSize; + *stateDataLen = sizeof(session->mechanism) + mechSize; if (stateData == NULL) return CKR_OK; @@ -19713,17 +21446,18 @@ int WP11_SetOperationState(WP11_Session* session, unsigned char* stateData, unsigned long mechSize = 0; int hashType = WC_HASH_TYPE_NONE; int ret; + CK_MECHANISM_TYPE mechanism; #if defined(LIBWOLFSSL_VERSION_HEX) && LIBWOLFSSL_VERSION_HEX < 0x05007004 wc_HashAlg* hashAlg; #else wc_Hashes* hashAlg; #endif - if (stateDataLen < sizeof(session->mechanism)) + if (stateDataLen < sizeof(mechanism)) return CKR_SAVED_STATE_INVALID; - XMEMCPY(&session->mechanism, stateData, sizeof(session->mechanism)); - switch (session->mechanism) { + XMEMCPY(&mechanism, stateData, sizeof(mechanism)); + switch (mechanism) { #ifndef NO_MD5 case CKM_MD5: mechSize = sizeof(wc_Md5); @@ -19758,16 +21492,20 @@ int WP11_SetOperationState(WP11_Session* session, unsigned char* stateData, return CKR_SAVED_STATE_INVALID; } - if (stateDataLen < (sizeof(session->mechanism) + mechSize)) + if (stateDataLen < (sizeof(mechanism) + mechSize)) return CKR_SAVED_STATE_INVALID; - stateData += sizeof(session->mechanism); + stateData += sizeof(mechanism); - ret = wp11_digest_hash_type(session->mechanism, &hashType); + ret = wp11_digest_hash_type(mechanism, &hashType); if (ret != CKR_OK) return ret; + /* The saved state is valid: replace the current operation. */ + WP11_Session_AbortOp(session); + session->mechanism = mechanism; + session->params.digest.hashType = (enum wc_HashType)hashType; ret = wc_HashInit(&session->params.digest.hash, (enum wc_HashType)hashType); diff --git a/src/slot.c b/src/slot.c index 8c9aa1ce..55d9f3a7 100644 --- a/src/slot.c +++ b/src/slot.c @@ -138,6 +138,8 @@ CK_RV C_GetSlotInfo(CK_SLOT_ID slotID, CK_SLOT_INFO_PTR pInfo) } XMEMCPY(pInfo, &slotInfoTemplate, sizeof(slotInfoTemplate)); + WP11_BlankPad(pInfo->slotDescription, sizeof(pInfo->slotDescription)); + WP11_BlankPad(pInfo->manufacturerID, sizeof(pInfo->manufacturerID)); /* Put in the slot id value as two decimal digits. */ pInfo->slotDescription[SLOT_ID_IDX + 0] = ((slotID / 10) % 10) + '0'; pInfo->slotDescription[SLOT_ID_IDX + 1] = ((slotID ) % 10) + '0'; @@ -164,8 +166,8 @@ static CK_TOKEN_INFO tokenInfoTemplate = { "wolfpkcs11", "wolfpkcs11", { - 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, - 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 + '0', '0', '0', '0', '0', '0', '0', '0', '0', + '0', '0', '0', '0', '0', '0', '0' }, /* serialNumber */ CKF_RNG | CKF_CLOCK_ON_TOKEN | CKF_LOGIN_REQUIRED, WP11_SESSION_CNT_MAX, /* ulMaxSessionCount */ @@ -202,6 +204,7 @@ CK_RV C_GetTokenInfo(CK_SLOT_ID slotID, CK_TOKEN_INFO_PTR pInfo) #ifndef WOLFPKCS11_NO_TIME time_t now, expire; struct tm nowTM; + struct tm* utc; #endif WP11_Slot* slot; int cnt; @@ -230,6 +233,8 @@ CK_RV C_GetTokenInfo(CK_SLOT_ID slotID, CK_TOKEN_INFO_PTR pInfo) } XMEMCPY(pInfo, &tokenInfoTemplate, sizeof(tokenInfoTemplate)); + WP11_BlankPad(pInfo->manufacturerID, sizeof(pInfo->manufacturerID)); + WP11_BlankPad(pInfo->model, sizeof(pInfo->model)); WP11_Slot_GetTokenLabel(slot, (char*)pInfo->label); pInfo->serialNumber[14] = ((slotID / 10) % 10) + '0'; pInfo->serialNumber[15] = ((slotID / 1) % 10) + '0'; @@ -241,28 +246,33 @@ CK_RV C_GetTokenInfo(CK_SLOT_ID slotID, CK_TOKEN_INFO_PTR pInfo) #ifndef WOLFPKCS11_NO_TIME now = XTIME(0); XMEMSET(&nowTM, 0, sizeof(nowTM)); - if (XGMTIME(&now, &nowTM) != NULL) { - pInfo->utcTime[ 0] = (((1900 + nowTM.tm_year) / 1000) % 10) + '0'; - pInfo->utcTime[ 1] = (((1900 + nowTM.tm_year) / 100) % 10) + '0'; - pInfo->utcTime[ 2] = (((1900 + nowTM.tm_year) / 10) % 10) + '0'; - pInfo->utcTime[ 3] = (((1900 + nowTM.tm_year) / 1) % 10) + '0'; - pInfo->utcTime[ 4] = (((1 + nowTM.tm_mon) / 10) % 10) + '0'; - pInfo->utcTime[ 5] = (((1 + nowTM.tm_mon) / 1) % 10) + '0'; - pInfo->utcTime[ 6] = ((nowTM.tm_mday / 10) % 10) + '0'; - pInfo->utcTime[ 7] = ((nowTM.tm_mday / 1) % 10) + '0'; - pInfo->utcTime[ 8] = ((nowTM.tm_hour / 10) % 10) + '0'; - pInfo->utcTime[ 9] = ((nowTM.tm_hour / 1) % 10) + '0'; - pInfo->utcTime[10] = ((nowTM.tm_min / 10) % 10) + '0'; - pInfo->utcTime[11] = ((nowTM.tm_min / 1) % 10) + '0'; - pInfo->utcTime[12] = ((nowTM.tm_sec / 10) % 10) + '0'; - pInfo->utcTime[13] = ((nowTM.tm_sec / 1) % 10) + '0'; + /* XGMTIME may map to gmtime(), which ignores nowTM. */ + utc = NULL; + if (now != (time_t)-1) + utc = XGMTIME(&now, &nowTM); + if (utc != NULL) { + pInfo->utcTime[ 0] = (((1900 + utc->tm_year) / 1000) % 10) + '0'; + pInfo->utcTime[ 1] = (((1900 + utc->tm_year) / 100) % 10) + '0'; + pInfo->utcTime[ 2] = (((1900 + utc->tm_year) / 10) % 10) + '0'; + pInfo->utcTime[ 3] = (((1900 + utc->tm_year) / 1) % 10) + '0'; + pInfo->utcTime[ 4] = (((1 + utc->tm_mon) / 10) % 10) + '0'; + pInfo->utcTime[ 5] = (((1 + utc->tm_mon) / 1) % 10) + '0'; + pInfo->utcTime[ 6] = ((utc->tm_mday / 10) % 10) + '0'; + pInfo->utcTime[ 7] = ((utc->tm_mday / 1) % 10) + '0'; + pInfo->utcTime[ 8] = ((utc->tm_hour / 10) % 10) + '0'; + pInfo->utcTime[ 9] = ((utc->tm_hour / 1) % 10) + '0'; + pInfo->utcTime[10] = ((utc->tm_min / 10) % 10) + '0'; + pInfo->utcTime[11] = ((utc->tm_min / 1) % 10) + '0'; + pInfo->utcTime[12] = ((utc->tm_sec / 10) % 10) + '0'; + pInfo->utcTime[13] = ((utc->tm_sec / 1) % 10) + '0'; } else { - /* Set date to all zeros. */ - XMEMCPY(pInfo->utcTime, "00000000000000", 14); + pInfo->flags &= ~(CKF_CLOCK_ON_TOKEN); + XMEMSET(pInfo->utcTime, ' ', sizeof(pInfo->utcTime)); } #else - XMEMCPY(pInfo->utcTime, "00000000000000", 14); + pInfo->flags &= ~(CKF_CLOCK_ON_TOKEN); + XMEMSET(pInfo->utcTime, ' ', sizeof(pInfo->utcTime)); #endif cnt = WP11_Slot_TokenFailedLogin(slot, WP11_LOGIN_SO); @@ -476,16 +486,20 @@ static CK_MECHANISM_TYPE mechanismList[] = { #ifdef WOLFPKCS11_NSS /* NSS uses this as a target-key marker when unwrapping TLS secrets. */ CKM_SSL3_MASTER_KEY_DERIVE, +#ifndef NO_HMAC CKM_NSS_PKCS12_PBE_SHA224_HMAC_KEY_GEN, CKM_NSS_PKCS12_PBE_SHA256_HMAC_KEY_GEN, CKM_NSS_PKCS12_PBE_SHA384_HMAC_KEY_GEN, CKM_NSS_PKCS12_PBE_SHA512_HMAC_KEY_GEN, #endif +#endif #ifdef WOLFSSL_HAVE_PRF CKM_TLS_MAC, #endif CKM_GENERIC_SECRET_KEY_GEN, +#ifndef NO_HMAC CKM_PKCS5_PBKD2 +#endif }; /* Count of mechanisms in list. */ @@ -504,7 +518,6 @@ static int mechanismCnt = ((int)(sizeof(mechanismList)/sizeof(*mechanismList))); * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SLOT_ID_INVALID when no slot with id can be found. * CKR_ARGUMENTS_BAD when pulCount is NULL. - * CKR_BUFFER_TOO_SMALL when pulCount is NULL. * CKR_BUFFER_TOO_SMALL when there are more mechanisms than entries in * array. * CKR_OK on success. @@ -564,6 +577,14 @@ CK_RV C_GetMechanismList(CK_SLOT_ID slotID, return rv; } +/* Matches the key serialization C_WrapKey compiles in for RSA and secret + * keys; without any, every wrap fails with CKR_KEY_NOT_WRAPPABLE. */ +#if defined(WOLFSSL_STM32U5_DHUK) || !defined(WOLFPKCS11_NO_STORE) + #define WP11_MECH_WRAP_FLAG CKF_WRAP +#else + #define WP11_MECH_WRAP_FLAG 0 +#endif + #ifndef NO_RSA #ifdef WOLFSSL_KEY_GEN /* Info on RSA key generation mechanism. */ @@ -573,13 +594,13 @@ static CK_MECHANISM_INFO rsaKgMechInfo = { #endif /* Info on RSA X.509 mechanism. */ static CK_MECHANISM_INFO rsaX509MechInfo = { - 1024, 4096, CKF_ENCRYPT | CKF_DECRYPT | CKF_SIGN | CKF_VERIFY | CKF_WRAP | - CKF_UNWRAP | CKF_VERIFY_RECOVER + 1024, 4096, CKF_ENCRYPT | CKF_DECRYPT | CKF_SIGN | CKF_VERIFY | + CKF_VERIFY_RECOVER }; /* Info on RSA PKCS#1.5 mechanism. */ static CK_MECHANISM_INFO rsaPkcsMechInfo = { 1024, 4096, CKF_ENCRYPT | CKF_DECRYPT | CKF_SIGN | CKF_VERIFY | - CKF_VERIFY_RECOVER + CKF_VERIFY_RECOVER | WP11_MECH_WRAP_FLAG | CKF_UNWRAP }; #ifndef WC_NO_RSA_OAEP /* Info on RSA PKCS#1 OAEP mechanism. */ @@ -712,6 +733,7 @@ static CK_MECHANISM_INFO nssTls12MasterKeyDeriveDhInfo = { static CK_MECHANISM_INFO nssTls12MasterKeyDeriveInfo = { 48, 128, CKF_DERIVE }; +#ifndef NO_HMAC static CK_MECHANISM_INFO nssPkcs12PbeSha224HmacKeyGenMechInfo = { 224, 224, CKF_GENERATE }; @@ -724,6 +746,7 @@ static CK_MECHANISM_INFO nssPkcs12PbeSha384HmacKeyGenMechInfo = { static CK_MECHANISM_INFO nssPkcs12PbeSha512HmacKeyGenMechInfo = { 512, 512, CKF_GENERATE }; +#endif /* NSS requires this mechanism identifier when selecting the slot used to * unwrap cached TLS secrets. C_DeriveKey does not implement the mechanism, so * do not advertise CKF_DERIVE. */ @@ -743,7 +766,7 @@ static CK_MECHANISM_INFO aesKeyGenMechInfo = { }; #ifdef HAVE_AES_KEYWRAP static CK_MECHANISM_INFO aesKeyWrapMechInfo = { - 16, 32, CKF_ENCRYPT | CKF_DECRYPT | CKF_WRAP | CKF_UNWRAP + 16, 32, CKF_ENCRYPT | CKF_DECRYPT | WP11_MECH_WRAP_FLAG | CKF_UNWRAP }; #endif #ifdef HAVE_AES_CBC @@ -751,6 +774,9 @@ static CK_MECHANISM_INFO aesKeyWrapMechInfo = { static CK_MECHANISM_INFO aesCbcMechInfo = { 16, 32, CKF_ENCRYPT | CKF_DECRYPT }; +static CK_MECHANISM_INFO aesCbcPadMechInfo = { + 16, 32, CKF_ENCRYPT | CKF_DECRYPT | WP11_MECH_WRAP_FLAG | CKF_UNWRAP +}; static CK_MECHANISM_INFO aesCbcEncryptDataMechInfo = { 1, 32, CKF_DERIVE }; @@ -776,7 +802,7 @@ static CK_MECHANISM_INFO aesCcmMechInfo = { #ifdef HAVE_AESECB /* Info on AES-ECB mechanism. */ static CK_MECHANISM_INFO aesEcbMechInfo = { - 16, 32, CKF_ENCRYPT | CKF_DECRYPT + 16, 32, CKF_ENCRYPT | CKF_DECRYPT | WP11_MECH_WRAP_FLAG | CKF_UNWRAP }; #endif #ifdef HAVE_AESCTS @@ -872,13 +898,16 @@ static CK_MECHANISM_INFO sha3MechInfo = { }; #endif #endif +/* Key sizes in bits. */ static CK_MECHANISM_INFO genSecKeyGenMechInfo = { - 1, 32, CKF_GENERATE + 8, WP11_MAX_SYM_KEY_SZ * 8, CKF_GENERATE }; +#ifndef NO_HMAC static CK_MECHANISM_INFO pkcs5Pbkdf2MechInfo = { 1, 256, CKF_GENERATE }; +#endif /** * Get information on a mechanism. @@ -1073,6 +1102,8 @@ CK_RV C_GetMechanismInfo(CK_SLOT_ID slotID, CK_MECHANISM_TYPE type, #endif #ifdef HAVE_AES_CBC case CKM_AES_CBC_PAD: + XMEMCPY(pInfo, &aesCbcPadMechInfo, sizeof(CK_MECHANISM_INFO)); + break; case CKM_AES_CBC: XMEMCPY(pInfo, &aesCbcMechInfo, sizeof(CK_MECHANISM_INFO)); break; @@ -1221,6 +1252,7 @@ CK_RV C_GetMechanismInfo(CK_SLOT_ID slotID, CK_MECHANISM_TYPE type, XMEMCPY(pInfo, &nssTls12MasterKeyDeriveDhInfo, sizeof(CK_MECHANISM_INFO)); break; +#ifndef NO_HMAC case CKM_NSS_PKCS12_PBE_SHA224_HMAC_KEY_GEN: XMEMCPY(pInfo, &nssPkcs12PbeSha224HmacKeyGenMechInfo, sizeof(CK_MECHANISM_INFO)); @@ -1237,6 +1269,7 @@ CK_RV C_GetMechanismInfo(CK_SLOT_ID slotID, CK_MECHANISM_TYPE type, XMEMCPY(pInfo, &nssPkcs12PbeSha512HmacKeyGenMechInfo, sizeof(CK_MECHANISM_INFO)); break; +#endif case CKM_SSL3_MASTER_KEY_DERIVE: XMEMCPY(pInfo, &ssl3MasterKeyTargetInfo, sizeof(CK_MECHANISM_INFO)); @@ -1253,10 +1286,12 @@ CK_RV C_GetMechanismInfo(CK_SLOT_ID slotID, CK_MECHANISM_TYPE type, XMEMCPY(pInfo, &genSecKeyGenMechInfo, sizeof(CK_MECHANISM_INFO)); break; +#ifndef NO_HMAC case CKM_PKCS5_PBKD2: XMEMCPY(pInfo, &pkcs5Pbkdf2MechInfo, sizeof(CK_MECHANISM_INFO)); break; +#endif default: return CKR_MECHANISM_INVALID; } @@ -1274,8 +1309,8 @@ CK_RV C_GetMechanismInfo(CK_SLOT_ID slotID, CK_MECHANISM_TYPE type, * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SLOT_ID_INVALID when no slot with id can be found. * CKR_ARGUMENTS_BAD when pPin or pLabel is NULL. - * CKR_PIN_INCORRECT when length of PIN is not valid or PIN does not - * match initialized PIN. + * CKR_PIN_LEN_RANGE when length of PIN is not valid. + * CKR_PIN_INCORRECT when PIN does not match initialized PIN. * CKR_SESSION_EXISTS when a session is open on the token. * CKR_FUNCTION_FAILED when resetting token fails. * CKR_OK on success. @@ -1363,7 +1398,7 @@ CK_RV C_InitToken(CK_SLOT_ID slotID, CK_UTF8CHAR_PTR pPin, * CKR_SESSION_HANDLE_INVALID when session handle is not valid. * CKR_ARGUMENTS_BAD when pPin is NULL. * CKR_USER_NOT_LOGGED_IN when not logged in as Security Officer. - * CKR_PIN_INCORRECT when length of PIN is not valid. + * CKR_PIN_LEN_RANGE when length of PIN is not valid. * CKR_FUNCTION_FAILED when setting User PIN fails. * CKR_OK on success. */ @@ -1433,8 +1468,9 @@ CK_RV C_InitPIN(CK_SESSION_HANDLE hSession, CK_UTF8CHAR_PTR pPin, * @return CKR_CRYPTOKI_NOT_INITIALIZED when library not initialized. * CKR_SESSION_HANDLE_INVALID when session handle is not valid. * CKR_ARGUMENTS_BAD when pOldPin or pNewPin is NULL. - * CKR_PIN_INCORRECT when length of old or new PIN is not valid or - * old PIN does not verify. + * CKR_PIN_INCORRECT when length of old PIN is not valid or old PIN + * does not verify. + * CKR_PIN_LEN_RANGE when length of new PIN is not valid. * CKR_SESSION_READ_ONLY when session not read/write. * CKR_USER_PIN_NOT_INITIALIZED when no previous PIN set for user. * CKR_FUNCTION_FAILED when setting user PIN fails. @@ -1562,6 +1598,8 @@ CK_RV C_SetPIN(CK_SESSION_HANDLE hSession, CK_UTF8CHAR_PTR pOldPin, * CKR_SESSION_READ_WRITE_SO_EXISTS when there is an existing open * Security Officer session. * CKR_SESSION_COUNT when no more sessions can be opened on token. + * CKR_HOST_MEMORY when allocating the session fails. + * CKR_FUNCTION_FAILED when opening the session fails otherwise. * CKR_OK on success. */ CK_RV C_OpenSession(CK_SLOT_ID slotID, CK_FLAGS flags, @@ -1611,6 +1649,16 @@ CK_RV C_OpenSession(CK_SLOT_ID slotID, CK_FLAGS flags, WOLFPKCS11_LEAVE("C_OpenSession", rv); return rv; } + if (ret == MEMORY_E) { + rv = CKR_HOST_MEMORY; + WOLFPKCS11_LEAVE("C_OpenSession", rv); + return rv; + } + if (ret != 0) { + rv = CKR_FUNCTION_FAILED; + WOLFPKCS11_LEAVE("C_OpenSession", rv); + return rv; + } rv = CKR_OK; WOLFPKCS11_LEAVE("C_OpenSession", rv); @@ -1842,8 +1890,13 @@ CK_RV C_SetOperationState(CK_SESSION_HANDLE hSession, return rv; } - (void)hEncryptionKey; - (void)hAuthenticationKey; + /* Only keyless digest states can be saved, so no key is ever needed. */ + if (hEncryptionKey != CK_INVALID_HANDLE || + hAuthenticationKey != CK_INVALID_HANDLE) { + rv = CKR_KEY_NOT_NEEDED; + WOLFPKCS11_LEAVE("C_SetOperationState", rv); + return rv; + } rv = WP11_SetOperationState(session, pOperationState, ulOperationStateLen); WOLFPKCS11_LEAVE("C_SetOperationState", rv); diff --git a/src/wolfpkcs11.c b/src/wolfpkcs11.c index bc1e64d0..73369a85 100644 --- a/src/wolfpkcs11.c +++ b/src/wolfpkcs11.c @@ -522,8 +522,12 @@ CK_RV C_GetInterface(CK_UTF8CHAR_PTR pInterfaceName, CK_VERSION_PTR pVersion, /** * Initialize the Crypto-Ki library. * - * @param pInitArgs [out] Ignored. - * @return CKR_FUNCTION_FAILED when initializing fails. + * @param pInitArgs [in] Optional CK_C_INITIALIZE_ARGS. The mutex callbacks + * must be all set or all NULL and pReserved must be + * NULL. NSS builds read the configdir from + * LibraryParameters. The callbacks are not used. + * @return CKR_ARGUMENTS_BAD when pInitArgs is not valid. + * CKR_FUNCTION_FAILED when initializing fails. * CKR_OK on success. */ CK_RV C_Initialize(CK_VOID_PTR pInitArgs) @@ -586,7 +590,11 @@ CK_RV C_Initialize(CK_VOID_PTR pInitArgs) * Finalize the Crypto-Ki library. * * @param pReserved [out] Ignored. - * @return CKR_OK on success. + * @return CKR_FUNCTION_FAILED when token state could not be stored, in which + * case the library is finalized all the same, or when the + * initialization lock could not be taken, in which case it stays + * initialized and the call can be retried. + * CKR_OK on success. */ CK_RV C_Finalize(CK_VOID_PTR pReserved) { @@ -604,9 +612,7 @@ CK_RV C_Finalize(CK_VOID_PTR pReserved) return ret; } - WP11_Library_Final(); - - ret = CKR_OK; + ret = WP11_Library_Final() == 0 ? CKR_OK : CKR_FUNCTION_FAILED; WOLFPKCS11_LEAVE("C_Finalize", ret); return ret; } @@ -666,6 +672,9 @@ CK_RV C_GetInfo(CK_INFO_PTR pInfo) } XMEMCPY(pInfo, &wolfpkcs11Info, sizeof(wolfpkcs11Info)); + WP11_BlankPad(pInfo->manufacturerID, sizeof(pInfo->manufacturerID)); + WP11_BlankPad(pInfo->libraryDescription, + sizeof(pInfo->libraryDescription)); ret = CKR_OK; WOLFPKCS11_LEAVE("C_GetInfo", ret); return ret; @@ -680,6 +689,9 @@ CK_RV C_GetInfoV3_0(CK_INFO_PTR pInfo) return CKR_ARGUMENTS_BAD; XMEMCPY(pInfo, &wolfpkcs11Info_3_0, sizeof(wolfpkcs11Info_3_0)); + WP11_BlankPad(pInfo->manufacturerID, sizeof(pInfo->manufacturerID)); + WP11_BlankPad(pInfo->libraryDescription, + sizeof(pInfo->libraryDescription)); return CKR_OK; } @@ -694,6 +706,9 @@ CK_RV C_GetInfoV3_2(CK_INFO_PTR pInfo) return CKR_ARGUMENTS_BAD; XMEMCPY(pInfo, &wolfpkcs11Info_3_2, sizeof(wolfpkcs11Info_3_2)); + WP11_BlankPad(pInfo->manufacturerID, sizeof(pInfo->manufacturerID)); + WP11_BlankPad(pInfo->libraryDescription, + sizeof(pInfo->libraryDescription)); return CKR_OK; } diff --git a/tests/attribute_validation_test.c b/tests/attribute_validation_test.c new file mode 100644 index 00000000..49041955 --- /dev/null +++ b/tests/attribute_validation_test.c @@ -0,0 +1,909 @@ +/* attribute_validation_test.c + * + * Copyright (C) 2006-2025 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Attribute templates supplied when creating or updating objects are + * validated, and defaulted attributes behave as PKCS#11 specifies. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#define TEST_DIR "./store/attribute_validation_test" + +#if defined(USE_WOLFSSL_MEMORY) && !defined(WOLFSSL_STATIC_MEMORY) && \ + !defined(WOLFSSL_NO_MALLOC) && !defined(WOLFSSL_DEBUG_MEMORY) +#define ATTR_TEST_ALLOC_HOOK +/* Allocation of exactly this size fails while armed. */ +#define FAIL_ALLOC_SIZE 97 +static int failAllocArmed = 0; + +static void* test_malloc(size_t n) +{ + if (failAllocArmed && n == FAIL_ALLOC_SIZE) + return NULL; + return malloc(n); +} + +static void test_free(void* p) +{ + free(p); +} + +static void* test_realloc(void* p, size_t n) +{ + if (failAllocArmed && n == FAIL_ALLOC_SIZE) + return NULL; + return realloc(p, n); +} +#endif + +static CK_OBJECT_CLASS dataClass = CKO_DATA; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; +static const byte dataValue[] = "attribute-validation"; +static const byte dataLabel[] = "attribute-validation-label"; +static const byte newLabel[] = "replaced"; + +/* Check that the object's CKA_LABEL is exactly the expected bytes. */ +static void expect_label(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + const byte* expect, CK_ULONG expectLen, + const char* name) +{ + CK_RV rv; + byte buf[64]; + CK_ATTRIBUTE attr = { CKA_LABEL, buf, sizeof(buf) }; + + rv = funcList->C_GetAttributeValue(session, obj, &attr, 1); + CHECK_TRUE(rv == CKR_OK && attr.ulValueLen == expectLen && + XMEMCMP(buf, expect, expectLen) == 0, name); +} + +/* Check that a boolean attribute of the object has the expected value. */ +static void expect_bool(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_ATTRIBUTE_TYPE type, CK_BBOOL expect, + const char* name) +{ + CK_RV rv; + CK_BBOOL val = (CK_BBOOL)!expect; + CK_ATTRIBUTE attr = { type, &val, sizeof(val) }; + + rv = funcList->C_GetAttributeValue(session, obj, &attr, 1); + CHECK_TRUE(rv == CKR_OK && val == expect, name); +} + +static void destroy_obj(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* obj) +{ + if (*obj != CK_INVALID_HANDLE) { + funcList->C_DestroyObject(session, *obj); + *obj = CK_INVALID_HANDLE; + } +} + +/* Create a public, modifiable session data object. */ +static CK_RV create_data_object(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE* obj) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, (void*)dataLabel, sizeof(dataLabel) - 1 }, + { CKA_VALUE, (void*)dataValue, sizeof(dataValue) - 1 }, + }; + + *obj = CK_INVALID_HANDLE; + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), obj); +} + +/* CKA_TOKEN is validated as a CK_BBOOL like every other boolean attribute. */ +static void test_token_attr_is_bool(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE bad = CK_INVALID_HANDLE; + CK_BBOOL badBool = 5; + byte wide[2] = { 0, 0 }; + CK_ATTRIBUTE nullTmpl[] = { { CKA_TOKEN, NULL, sizeof(CK_BBOOL) } }; + CK_ATTRIBUTE wideTmpl[] = { { CKA_TOKEN, wide, sizeof(wide) } }; + CK_ATTRIBUTE badTmpl[] = { { CKA_TOKEN, &badBool, sizeof(badBool) } }; + CK_ATTRIBUTE createTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &badBool, sizeof(badBool) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)dataValue, sizeof(dataValue) - 1 }, + }; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_SetAttributeValue(session, obj, nullTmpl, 1); + CHECK_RV(rv, "set CKA_TOKEN with NULL value", CKR_ATTRIBUTE_VALUE_INVALID); + rv = funcList->C_SetAttributeValue(session, obj, wideTmpl, 1); + CHECK_RV(rv, "set CKA_TOKEN with wrong length", + CKR_ATTRIBUTE_VALUE_INVALID); + rv = funcList->C_SetAttributeValue(session, obj, badTmpl, 1); + CHECK_RV(rv, "set CKA_TOKEN with non-boolean value", + CKR_ATTRIBUTE_VALUE_INVALID); + + rv = funcList->C_CreateObject(session, createTmpl, + sizeof(createTmpl) / sizeof(*createTmpl), + &bad); + CHECK_RV(rv, "create object with non-boolean CKA_TOKEN", + CKR_ATTRIBUTE_VALUE_INVALID); + if (rv == CKR_OK) + destroy_obj(session, &bad); + + destroy_obj(session, &obj); +} + +/* Attribute counts beyond what the library can index are rejected rather + * than treated as an empty template. */ +static void test_set_attr_count_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE setTmpl[] = { + { CKA_LABEL, (void*)newLabel, sizeof(newLabel) - 1 }, + }; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_SetAttributeValue(session, obj, setTmpl, + (CK_ULONG)INT_MAX + 1); + CHECK_RV(rv, "set attributes with oversized count", CKR_ARGUMENTS_BAD); + expect_label(session, obj, dataLabel, sizeof(dataLabel) - 1, + "label unchanged after oversized count"); + + destroy_obj(session, &obj); +} + +static void test_get_attr_count_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte buf[64]; + CK_ATTRIBUTE getTmpl[] = { { CKA_LABEL, buf, sizeof(buf) } }; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_GetAttributeValue(session, obj, getTmpl, + (CK_ULONG)INT_MAX + 1); + CHECK_RV(rv, "get attributes with oversized count", CKR_ARGUMENTS_BAD); + + destroy_obj(session, &obj); +} + +/* A large result array returns the matching objects rather than none. */ +static void test_find_objects_large_max(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE found[8]; + CK_ULONG count = 0; + static const byte label[] = "attribute-validation-large-max"; + CK_ATTRIBUTE labelTmpl[] = { + { CKA_LABEL, (void*)label, sizeof(label) - 1 }, + }; + CK_ATTRIBUTE findTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, (void*)label, sizeof(label) - 1 }, + }; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv != CKR_OK) + return; + rv = funcList->C_SetAttributeValue(session, obj, labelTmpl, 1); + CHECK_RV(rv, "set a label only this test uses", CKR_OK); + + XMEMSET(found, 0, sizeof(found)); + rv = funcList->C_FindObjectsInit(session, findTmpl, + sizeof(findTmpl) / sizeof(*findTmpl)); + CHECK_RV(rv, "find init", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, found, (CK_ULONG)INT_MAX + 1, + &count); + CHECK_RV(rv, "find with large maximum", CKR_OK); + CHECK_TRUE(count == 1 && found[0] == obj, + "find with large maximum returns the object"); + funcList->C_FindObjectsFinal(session); + } + + destroy_obj(session, &obj); +} + +/* CKA_TOKEN, CKA_PRIVATE and CKA_MODIFIABLE only change through + * C_CopyObject; C_SetAttributeValue may only restate the current value. */ +static void test_copy_only_attrs_read_only(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE copy = CK_INVALID_HANDLE; + CK_ATTRIBUTE tokenTmpl[] = { { CKA_TOKEN, &ckTrue, 1 } }; + CK_ATTRIBUTE privTmpl[] = { { CKA_PRIVATE, &ckTrue, 1 } }; + CK_ATTRIBUTE modTmpl[] = { { CKA_MODIFIABLE, &ckFalse, 1 } }; + CK_ATTRIBUTE sameTmpl[] = { + { CKA_TOKEN, &ckFalse, 1 }, + { CKA_PRIVATE, &ckFalse, 1 }, + { CKA_MODIFIABLE, &ckTrue, 1 }, + }; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_SetAttributeValue(session, obj, tokenTmpl, 1); + CHECK_RV(rv, "set CKA_TOKEN after creation", CKR_ATTRIBUTE_READ_ONLY); + rv = funcList->C_SetAttributeValue(session, obj, privTmpl, 1); + CHECK_RV(rv, "set CKA_PRIVATE after creation", CKR_ATTRIBUTE_READ_ONLY); + rv = funcList->C_SetAttributeValue(session, obj, modTmpl, 1); + CHECK_RV(rv, "set CKA_MODIFIABLE after creation", + CKR_ATTRIBUTE_READ_ONLY); + expect_bool(session, obj, CKA_TOKEN, CK_FALSE, "CKA_TOKEN unchanged"); + expect_bool(session, obj, CKA_PRIVATE, CK_FALSE, "CKA_PRIVATE unchanged"); + expect_bool(session, obj, CKA_MODIFIABLE, CK_TRUE, + "CKA_MODIFIABLE unchanged"); + + rv = funcList->C_SetAttributeValue(session, obj, sameTmpl, + sizeof(sameTmpl) / sizeof(*sameTmpl)); + CHECK_RV(rv, "restate current CKA_TOKEN/PRIVATE/MODIFIABLE", CKR_OK); + + rv = funcList->C_CopyObject(session, obj, modTmpl, 1, ©); + CHECK_RV(rv, "copy with CKA_MODIFIABLE changed", CKR_OK); + if (rv == CKR_OK) { + expect_bool(session, copy, CKA_MODIFIABLE, CK_FALSE, + "copy has CKA_MODIFIABLE false"); + } + + destroy_obj(session, ©); + destroy_obj(session, &obj); +} + +#ifndef NO_AES +/* CKA_ALWAYS_SENSITIVE and CKA_NEVER_EXTRACTABLE are derived by the token and + * cannot be supplied when a key is created or generated. */ +static void test_derived_attrs_not_settable_at_create(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_ULONG keyLen = 16; + CK_MECHANISM mech = { CKM_AES_KEY_GEN, NULL, 0 }; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + byte keyData[16]; + CK_ATTRIBUTE createTmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_ALWAYS_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + }; + CK_ULONG createCnt = sizeof(createTmpl) / sizeof(*createTmpl); + CK_ATTRIBUTE genTmpl[] = { + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE_LEN, &keyLen, sizeof(keyLen) }, + { CKA_NEVER_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + }; + CK_ULONG genCnt = sizeof(genTmpl) / sizeof(*genTmpl); + + XMEMSET(keyData, 0x5a, sizeof(keyData)); + + rv = funcList->C_CreateObject(session, createTmpl, createCnt, &key); + CHECK_RV(rv, "create key with CKA_ALWAYS_SENSITIVE", + CKR_ATTRIBUTE_READ_ONLY); + destroy_obj(session, &key); + + createTmpl[createCnt - 1].type = CKA_NEVER_EXTRACTABLE; + rv = funcList->C_CreateObject(session, createTmpl, createCnt, &key); + CHECK_RV(rv, "create key with CKA_NEVER_EXTRACTABLE", + CKR_ATTRIBUTE_READ_ONLY); + destroy_obj(session, &key); + + rv = funcList->C_GenerateKey(session, &mech, genTmpl, genCnt, &key); + CHECK_RV(rv, "generate key with CKA_NEVER_EXTRACTABLE", + CKR_ATTRIBUTE_READ_ONLY); + destroy_obj(session, &key); +} +#endif + +#ifndef NO_AES +/* A creation template must name one object class; a conflicting duplicate is + * rejected rather than overriding the class the object was built for. */ +static void test_create_class_consistent(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + byte keyData[16]; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + }; + + XMEMSET(keyData, 0x3c, sizeof(keyData)); + rv = funcList->C_CreateObject(session, tmpl, sizeof(tmpl) / sizeof(*tmpl), + &key); + CHECK_RV(rv, "create key with conflicting CKA_CLASS", + CKR_TEMPLATE_INCONSISTENT); + destroy_obj(session, &key); +} +#endif + +#ifndef NO_AES +/* A rejected multi-attribute update leaves every attribute unchanged. */ +static void test_rejected_update_is_atomic(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte keyData[16]; + byte valueBuf[64]; + CK_ATTRIBUTE valueAttr = { CKA_VALUE, valueBuf, sizeof(valueBuf) }; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, (void*)dataLabel, sizeof(dataLabel) - 1 }, + { CKA_VALUE, keyData, sizeof(keyData) }, + }; + CK_ATTRIBUTE labelAndClass[] = { + { CKA_LABEL, (void*)newLabel, sizeof(newLabel) - 1 }, + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + }; + CK_ATTRIBUTE labelAndSensitive[] = { + { CKA_LABEL, (void*)newLabel, sizeof(newLabel) - 1 }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE valueAndClass[] = { + { CKA_VALUE, (void*)newLabel, sizeof(newLabel) - 1 }, + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + }; + + XMEMSET(keyData, 0x7e, sizeof(keyData)); + rv = funcList->C_CreateObject(session, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl), &key); + CHECK_RV(rv, "create labelled secret key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, key, labelAndClass, 2); + CHECK_RV(rv, "set label with read-only class", CKR_ATTRIBUTE_READ_ONLY); + expect_label(session, key, dataLabel, sizeof(dataLabel) - 1, + "label unchanged after rejected class update"); + + rv = funcList->C_SetAttributeValue(session, key, labelAndSensitive, 2); + CHECK_RV(rv, "set label with sensitive cleared", + CKR_ATTRIBUTE_READ_ONLY); + expect_label(session, key, dataLabel, sizeof(dataLabel) - 1, + "label unchanged after rejected sensitive update"); + } + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, obj, valueAndClass, 2); + CHECK_RV(rv, "set value with read-only class", CKR_ATTRIBUTE_READ_ONLY); + rv = funcList->C_GetAttributeValue(session, obj, &valueAttr, 1); + CHECK_TRUE(rv == CKR_OK && + valueAttr.ulValueLen == sizeof(dataValue) - 1 && + XMEMCMP(valueBuf, dataValue, sizeof(dataValue) - 1) == 0, + "value unchanged after rejected class update"); + } + + destroy_obj(session, &obj); + destroy_obj(session, &key); +} +#endif + +/* Attribute lengths beyond what the library stores are rejected before any + * value is read, for both generic and key component attributes. */ +static void test_attr_length_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_ULONG wrapLen = 0; + CK_ATTRIBUTE labelTmpl[] = { + { CKA_LABEL, (void*)newLabel, (CK_ULONG)INT_MAX + 1 }, + }; +#ifndef NO_RSA + CK_OBJECT_CLASS rsaClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE rsaType = CKK_RSA; + CK_ATTRIBUTE rsaTmpl[] = { + { CKA_CLASS, &rsaClass, sizeof(rsaClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; +#endif +#ifndef NO_DH + CK_OBJECT_CLASS dhClass = CKO_PUBLIC_KEY; + CK_KEY_TYPE dhType = CKK_DH; + CK_ATTRIBUTE dhTmpl[] = { + { CKA_CLASS, &dhClass, sizeof(dhClass) }, + { CKA_KEY_TYPE, &dhType, sizeof(dhType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_pub, sizeof(dh_2048_pub) }, + }; +#endif + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, obj, labelTmpl, 1); + CHECK_RV(rv, "set label with oversized length", + CKR_ATTRIBUTE_VALUE_INVALID); + expect_label(session, obj, dataLabel, sizeof(dataLabel) - 1, + "label unchanged after oversized length"); + } + destroy_obj(session, &obj); + + /* Only a 64-bit CK_ULONG can carry a length whose low 32 bits look + * valid. */ + if (sizeof(CK_ULONG) <= 4) + return; + wrapLen = (CK_ULONG)1 << 16 << 16; +#ifndef NO_RSA + rsaTmpl[6].ulValueLen += wrapLen; + rv = funcList->C_CreateObject(session, rsaTmpl, + sizeof(rsaTmpl) / sizeof(*rsaTmpl), &key); + CHECK_RV(rv, "create RSA key with oversized prime length", + CKR_ATTRIBUTE_VALUE_INVALID); + destroy_obj(session, &key); +#endif +#ifndef NO_DH + dhTmpl[4].ulValueLen += wrapLen; + rv = funcList->C_CreateObject(session, dhTmpl, + sizeof(dhTmpl) / sizeof(*dhTmpl), &key); + CHECK_RV(rv, "create DH key with oversized prime length", + CKR_ATTRIBUTE_VALUE_INVALID); + destroy_obj(session, &key); +#endif + (void)wrapLen; + (void)key; +} + +#ifndef NO_AES +/* Keys cannot be marked as requiring per-use authentication, which the token + * does not enforce. */ +static void test_always_authenticate_rejected(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + byte keyData[16]; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_ALWAYS_AUTHENTICATE, &ckTrue, sizeof(ckTrue) }, + }; + CK_ULONG keyCnt = sizeof(keyTmpl) / sizeof(*keyTmpl); + CK_ATTRIBUTE setTrue[] = { { CKA_ALWAYS_AUTHENTICATE, &ckTrue, 1 } }; + CK_ATTRIBUTE setFalse[] = { { CKA_ALWAYS_AUTHENTICATE, &ckFalse, 1 } }; + + XMEMSET(keyData, 0x42, sizeof(keyData)); + rv = funcList->C_CreateObject(session, keyTmpl, keyCnt, &key); + CHECK_RV(rv, "create key requiring per-use authentication", + CKR_ATTRIBUTE_VALUE_INVALID); + destroy_obj(session, &key); + + /* Create without the attribute, then try to turn it on. */ + rv = funcList->C_CreateObject(session, keyTmpl, keyCnt - 1, &key); + CHECK_RV(rv, "create key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, key, setTrue, 1); + CHECK_RV(rv, "set CKA_ALWAYS_AUTHENTICATE true", + CKR_ATTRIBUTE_VALUE_INVALID); + expect_bool(session, key, CKA_ALWAYS_AUTHENTICATE, CK_FALSE, + "CKA_ALWAYS_AUTHENTICATE remains false"); + rv = funcList->C_SetAttributeValue(session, key, setFalse, 1); + CHECK_RV(rv, "set CKA_ALWAYS_AUTHENTICATE false", CKR_OK); + } + destroy_obj(session, &key); +} +#endif + +/* A byte-string attribute with a length must come with a value. */ +static void test_data_attr_requires_value(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE bad = CK_INVALID_HANDLE; + CK_ATTRIBUTE labelTmpl[] = { { CKA_LABEL, NULL, 5 } }; + CK_ATTRIBUTE createTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)dataValue, sizeof(dataValue) - 1 }, + { CKA_ID, NULL, 4 }, + }; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, obj, labelTmpl, 1); + CHECK_RV(rv, "set label with length and no value", + CKR_ATTRIBUTE_VALUE_INVALID); + expect_label(session, obj, dataLabel, sizeof(dataLabel) - 1, + "label unchanged after missing value"); + } + destroy_obj(session, &obj); + + rv = funcList->C_CreateObject(session, createTmpl, + sizeof(createTmpl) / sizeof(*createTmpl), + &bad); + CHECK_RV(rv, "create object with CKA_ID length and no value", + CKR_ATTRIBUTE_VALUE_INVALID); + destroy_obj(session, &bad); +} + +/* An empty CKA_START_DATE/CKA_END_DATE, as read from an object without + * dates, can be written back and clears a set date. */ +static void check_date_roundtrip(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE obj, CK_ATTRIBUTE_TYPE type, + const char* name) +{ + CK_RV rv; + CK_DATE date = { {'2','0','3','0'}, {'0','6'}, {'1','5'} }; + CK_DATE got; + CK_ATTRIBUTE getAttr = { type, &got, sizeof(got) }; + CK_ATTRIBUTE emptyNull[] = { { type, NULL, 0 } }; + CK_ATTRIBUTE emptyBuf[] = { { type, &got, 0 } }; + CK_ATTRIBUTE setDate[] = { { type, &date, sizeof(date) } }; + + rv = funcList->C_GetAttributeValue(session, obj, &getAttr, 1); + CHECK_TRUE(rv == CKR_OK && getAttr.ulValueLen == 0, name); + + rv = funcList->C_SetAttributeValue(session, obj, emptyNull, 1); + CHECK_RV(rv, "write back empty date", CKR_OK); + + rv = funcList->C_SetAttributeValue(session, obj, setDate, 1); + CHECK_RV(rv, "set date", CKR_OK); + getAttr.ulValueLen = sizeof(got); + rv = funcList->C_GetAttributeValue(session, obj, &getAttr, 1); + CHECK_TRUE(rv == CKR_OK && getAttr.ulValueLen == sizeof(date) && + XMEMCMP(&got, &date, sizeof(date)) == 0, "date read back"); + + rv = funcList->C_SetAttributeValue(session, obj, emptyBuf, 1); + CHECK_RV(rv, "clear date with empty value", CKR_OK); + getAttr.ulValueLen = sizeof(got); + rv = funcList->C_GetAttributeValue(session, obj, &getAttr, 1); + CHECK_TRUE(rv == CKR_OK && getAttr.ulValueLen == 0, "date cleared"); +} + +static void test_empty_date_roundtrip(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv == CKR_OK) { + check_date_roundtrip(session, obj, CKA_START_DATE, + "unset CKA_START_DATE reads as empty"); + check_date_roundtrip(session, obj, CKA_END_DATE, + "unset CKA_END_DATE reads as empty"); + } + destroy_obj(session, &obj); +} + +/* CKA_CERTIFICATE_CATEGORY is consumed as a full CK_ULONG. */ +static void test_certificate_category_ulong(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS certClass = CKO_CERTIFICATE; + CK_CERTIFICATE_TYPE certType = CKC_X_509; + CK_ULONG category = 2; + word32 shortCategory = 2; + byte certData[] = { 0x30, 0x82, 0x01, 0x00 }; + CK_OBJECT_HANDLE cert = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, certData, sizeof(certData) }, + { CKA_CERTIFICATE_CATEGORY, &category, sizeof(category) }, + }; + CK_ULONG cnt = sizeof(tmpl) / sizeof(*tmpl); + + rv = funcList->C_CreateObject(session, tmpl, cnt, &cert); + CHECK_RV(rv, "create certificate with category", CKR_OK); + destroy_obj(session, &cert); + + if (sizeof(CK_ULONG) != sizeof(word32)) { + tmpl[cnt - 1].pValue = &shortCategory; + tmpl[cnt - 1].ulValueLen = sizeof(shortCategory); + rv = funcList->C_CreateObject(session, tmpl, cnt, &cert); + CHECK_TRUE(rv != CKR_OK, "certificate category shorter than CK_ULONG"); + destroy_obj(session, &cert); + } +} + +#ifdef ATTR_TEST_ALLOC_HOOK +/* A replacement value that cannot be stored leaves the previous one. */ +static void test_failed_replace_keeps_value(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte big[FAIL_ALLOC_SIZE]; + byte idBuf[16]; + CK_ATTRIBUTE labelTmpl[] = { { CKA_LABEL, big, sizeof(big) } }; + CK_ATTRIBUTE idTmpl[] = { { CKA_ID, big, sizeof(big) } }; + CK_ATTRIBUTE idSet[] = { { CKA_ID, (void*)newLabel, + sizeof(newLabel) - 1 } }; + CK_ATTRIBUTE idGet = { CKA_ID, idBuf, sizeof(idBuf) }; + + XMEMSET(big, 'x', sizeof(big)); + rv = create_data_object(session, &obj); + CHECK_RV(rv, "create data object", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, obj, idSet, 1); + CHECK_RV(rv, "set CKA_ID", CKR_OK); + + failAllocArmed = 1; + rv = funcList->C_SetAttributeValue(session, obj, labelTmpl, 1); + failAllocArmed = 0; + CHECK_RV(rv, "label replacement out of memory", CKR_DEVICE_MEMORY); + expect_label(session, obj, dataLabel, sizeof(dataLabel) - 1, + "label kept after failed replacement"); + + failAllocArmed = 1; + rv = funcList->C_SetAttributeValue(session, obj, idTmpl, 1); + failAllocArmed = 0; + CHECK_RV(rv, "CKA_ID replacement out of memory", CKR_DEVICE_MEMORY); + rv = funcList->C_GetAttributeValue(session, obj, &idGet, 1); + CHECK_TRUE(rv == CKR_OK && idGet.ulValueLen == sizeof(newLabel) - 1 && + XMEMCMP(idBuf, newLabel, sizeof(newLabel) - 1) == 0, + "CKA_ID kept after failed replacement"); + } + destroy_obj(session, &obj); +} +#endif + +#if defined(WOLFPKCS11_KEYPAIR_GEN_COMMON_LABEL) && \ + defined(HAVE_AES_KEYWRAP) && !defined(WOLFPKCS11_NO_STORE) && \ + !defined(NO_RSA) && !defined(NO_AES) && \ + (defined(WOLFSSL_KEY_GEN) || defined(OPENSSL_EXTRA)) +static CK_OBJECT_CLASS secretKeyClass = CKO_SECRET_KEY; +static CK_OBJECT_CLASS privKeyClass = CKO_PRIVATE_KEY; +static CK_OBJECT_CLASS pubKeyClass = CKO_PUBLIC_KEY; +static CK_KEY_TYPE aesKeyType = CKK_AES; +static CK_KEY_TYPE rsaKeyType = CKK_RSA; + +/* An unwrap template that omits CKA_TOKEN creates session objects, including + * the companion RSA public key. */ +static void test_unwrap_rsa_token_default(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM mech = { CKM_AES_KEY_WRAP_PAD, NULL, 0 }; + CK_OBJECT_HANDLE wrappingKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE privKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE unwrapped = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE pubKey = CK_INVALID_HANDLE; + byte wrapped[2048]; + CK_ULONG wrappedLen = sizeof(wrapped); + CK_ULONG count = 0; + CK_ATTRIBUTE aesTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &aesKeyType, sizeof(aesKeyType) }, + { CKA_WRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_UNWRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)aes_cbc_key, sizeof(aes_cbc_key) }, + }; + CK_ATTRIBUTE rsaTmpl[] = { + { CKA_CLASS, &privKeyClass, sizeof(privKeyClass) }, + { CKA_KEY_TYPE, &rsaKeyType, sizeof(rsaKeyType) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE unwrapTmpl[] = { + { CKA_CLASS, &privKeyClass, sizeof(privKeyClass) }, + { CKA_KEY_TYPE, &rsaKeyType, sizeof(rsaKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE findTmpl[] = { + { CKA_CLASS, &pubKeyClass, sizeof(pubKeyClass) }, + { CKA_KEY_TYPE, &rsaKeyType, sizeof(rsaKeyType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + + rv = funcList->C_CreateObject(session, aesTmpl, + sizeof(aesTmpl) / sizeof(*aesTmpl), + &wrappingKey); + CHECK_RV(rv, "create AES wrapping key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, rsaTmpl, + sizeof(rsaTmpl) / sizeof(*rsaTmpl), + &privKey); + CHECK_RV(rv, "create RSA private key", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_WrapKey(session, &mech, wrappingKey, privKey, + wrapped, &wrappedLen); + CHECK_RV(rv, "wrap RSA private key", CKR_OK); + } + destroy_obj(session, &privKey); + if (rv == CKR_OK) { + rv = funcList->C_UnwrapKey(session, &mech, wrappingKey, wrapped, + wrappedLen, unwrapTmpl, + sizeof(unwrapTmpl) / sizeof(*unwrapTmpl), + &unwrapped); + CHECK_RV(rv, "unwrap RSA private key without CKA_TOKEN", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_FindObjectsInit(session, findTmpl, + sizeof(findTmpl) / sizeof(*findTmpl)); + CHECK_RV(rv, "find companion public key init", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, &pubKey, 1, &count); + CHECK_RV(rv, "find companion public key", CKR_OK); + funcList->C_FindObjectsFinal(session); + } + CHECK_TRUE(count == 1, "companion public key is a session object"); + if (count != 1) + pubKey = CK_INVALID_HANDLE; + } + + destroy_obj(session, &pubKey); + destroy_obj(session, &unwrapped); + destroy_obj(session, &wrappingKey); +} +#endif + +static int run_test(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = 0; + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + + rv = pkcs11_open_session(&session); + CHECK_RV(rv, "open session", CKR_OK); + if (rv == CKR_OK) { + test_token_attr_is_bool(session); + test_set_attr_count_range(session); + test_get_attr_count_range(session); + test_find_objects_large_max(session); + test_copy_only_attrs_read_only(session); +#ifndef NO_AES + test_derived_attrs_not_settable_at_create(session); + test_create_class_consistent(session); + test_rejected_update_is_atomic(session); +#endif + test_attr_length_range(session); +#ifndef NO_AES + test_always_authenticate_rejected(session); +#endif + test_data_attr_requires_value(session); + test_empty_date_roundtrip(session); + test_certificate_category_ulong(session); +#ifdef ATTR_TEST_ALLOC_HOOK + test_failed_replace_keeps_value(session); +#endif +#if defined(WOLFPKCS11_KEYPAIR_GEN_COMMON_LABEL) && \ + defined(HAVE_AES_KEYWRAP) && !defined(WOLFPKCS11_NO_STORE) && \ + !defined(NO_RSA) && !defined(NO_AES) && \ + (defined(WOLFSSL_KEY_GEN) || defined(OPENSSL_EXTRA)) + test_unwrap_rsa_token_default(session); +#endif + } + + if (session != 0) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + + printf("=== wolfPKCS11 attribute validation test ===\n"); +#ifdef ATTR_TEST_ALLOC_HOOK + if (wolfSSL_SetAllocators(test_malloc, test_free, test_realloc) != 0) { + fprintf(stderr, "FAIL: wolfSSL_SetAllocators\n"); + return 1; + } +#endif + run_test(); + return pkcs11_test_summary(); +} diff --git a/tests/concurrent_destroy_object_test.c b/tests/concurrent_destroy_object_test.c index 7edc50c1..8b07a0bb 100644 --- a/tests/concurrent_destroy_object_test.c +++ b/tests/concurrent_destroy_object_test.c @@ -60,7 +60,8 @@ #include "testdata.h" -#if !defined(NO_AES) && !defined(SINGLE_THREADED) +#if !defined(NO_AES) && !defined(SINGLE_THREADED) && \ + !defined(WOLFPKCS11_SINGLE_THREADED) #include diff --git a/tests/ec_derive_test.c b/tests/ec_derive_test.c new file mode 100644 index 00000000..be8a9ee6 --- /dev/null +++ b/tests/ec_derive_test.c @@ -0,0 +1,1159 @@ +/* ec_derive_test.c + * + * Copyright (C) 2006-2025 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Tests for C_DeriveKey with CKM_ECDH1_DERIVE: mechanism parameter + * validation and the attributes of the derived key object. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#if defined(HAVE_ECC) && !defined(WOLFPKCS11_NO_STORE) && !defined(_WIN32) + #define EC_DERIVE_PERSIST_TEST + #include + #include +#endif + +#include "testdata.h" + +#if defined(HAVE_ECC) && !defined(SINGLE_THREADED) && \ + !defined(WOLFPKCS11_SINGLE_THREADED) + #define EC_DERIVE_THREAD_TEST + #include +#endif + +#define TEST_DIR "./store/ec_derive_test" + +static int test_passed = 0; +static int test_failed = 0; +static int test_skipped = 0; + +#ifndef HAVE_PKCS11_STATIC +static void* dlib; +#endif +static CK_FUNCTION_LIST* funcList; +static CK_SLOT_ID slot = 0; +static const char* tokenName = "wolfpkcs11"; +static byte* soPin = (byte*)"password123456"; +static int soPinLen = 14; +static byte* userPin = (byte*)"wolfpkcs11-test"; +static int userPinLen = 15; + +#define CHECK_RV(rv, exp, op) do { \ + if ((rv) != (exp)) { \ + fprintf(stderr, "FAIL: %s: got 0x%lx, expected 0x%lx\n", op, \ + (unsigned long)(rv), (unsigned long)(exp)); \ + test_failed++; \ + result = -1; \ + goto cleanup; \ + } \ + else { \ + printf("PASS: %s\n", op); \ + test_passed++; \ + } \ +} while (0) + +#define CHECK_CKR(rv, op) CHECK_RV(rv, CKR_OK, op) + +#define CHECK_FAILS(rv, op) do { \ + if ((rv) == CKR_OK) { \ + fprintf(stderr, "FAIL: %s: got CKR_OK\n", op); \ + test_failed++; \ + result = -1; \ + goto cleanup; \ + } \ + else { \ + printf("PASS: %s\n", op); \ + test_passed++; \ + } \ +} while (0) + +#ifdef HAVE_ECC +/* Uncompressed P-384 public point. */ +static byte ecc_p384_point[] = { + 0x04, 0x2b, 0x46, 0xc2, 0xd4, 0x00, 0xbd, 0xba, 0xae, 0xa5, 0xd1, 0x4e, + 0xcd, 0x6e, 0x06, 0x05, 0x02, 0x41, 0xec, 0x49, 0x2f, 0x3a, 0xf6, 0x90, + 0xf4, 0x77, 0x8d, 0x2c, 0x77, 0x2e, 0x43, 0x21, 0x63, 0x47, 0x13, 0xd0, + 0x9e, 0xaf, 0x8e, 0x6c, 0x02, 0x2e, 0xd0, 0x00, 0x80, 0x5d, 0x6b, 0x3c, + 0xb8, 0x86, 0x69, 0x22, 0x1e, 0x58, 0x40, 0xdf, 0x2a, 0x8c, 0x64, 0x06, + 0x41, 0x37, 0xa0, 0xdf, 0x77, 0xd7, 0x56, 0x0f, 0x5d, 0x92, 0x5e, 0x0d, + 0x51, 0xdc, 0x79, 0xca, 0xa1, 0x7a, 0x84, 0x00, 0x98, 0x74, 0xd6, 0x82, + 0xa2, 0x55, 0xc8, 0xc6, 0x00, 0x4f, 0xf7, 0x33, 0x30, 0xfb, 0x2f, 0x8a, + 0x4a +}; + +static CK_OBJECT_CLASS privKeyClass = CKO_PRIVATE_KEY; +static CK_OBJECT_CLASS secretKeyClass = CKO_SECRET_KEY; +static CK_KEY_TYPE eccKeyType = CKK_EC; +static CK_KEY_TYPE genericKeyType = CKK_GENERIC_SECRET; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; +static CK_ULONG secretLen = 32; + +/* Session P-256 private key usable for ECDH. */ +static CK_RV create_ec_base(CK_SESSION_HANDLE session, CK_BBOOL sensitive, + CK_BBOOL extractable, CK_OBJECT_HANDLE* obj) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &privKeyClass, sizeof(privKeyClass) }, + { CKA_KEY_TYPE, &eccKeyType, sizeof(eccKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &sensitive, sizeof(sensitive) }, + { CKA_EXTRACTABLE, &extractable, sizeof(extractable) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + CK_ULONG cnt = sizeof(tmpl) / sizeof(*tmpl); + + return funcList->C_CreateObject(session, tmpl, cnt, obj); +} + +static CK_RV ecdh_derive_tmpl(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE base, + CK_ECDH1_DERIVE_PARAMS* params, + CK_ATTRIBUTE* tmpl, CK_ULONG cnt, + CK_OBJECT_HANDLE* derived) +{ + CK_MECHANISM mech; + + mech.mechanism = CKM_ECDH1_DERIVE; + mech.pParameter = params; + mech.ulParameterLen = sizeof(*params); + + return funcList->C_DeriveKey(session, &mech, base, tmpl, cnt, derived); +} + +/* ECDH derive of a public, extractable generic secret. */ +static CK_RV ecdh_derive(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE base, + CK_ECDH1_DERIVE_PARAMS* params, + CK_OBJECT_HANDLE* derived) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &secretLen, sizeof(secretLen) }, + }; + CK_ULONG cnt = sizeof(tmpl) / sizeof(*tmpl); + + return ecdh_derive_tmpl(session, base, params, tmpl, cnt, derived); +} + +static void ecdh_params_init(CK_ECDH1_DERIVE_PARAMS* params, byte* point, + CK_ULONG pointLen) +{ + XMEMSET(params, 0, sizeof(*params)); + params->kdf = CKD_NULL; + params->pSharedData = NULL; + params->ulSharedDataLen = 0; + params->pPublicData = point; + params->ulPublicDataLen = pointLen; +} + +/* With CKD_NULL there is no KDF, so shared data must be absent. */ +static int test_null_kdf_shared_data(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + byte sharedData[4] = { 0x01, 0x02, 0x03, 0x04 }; + int result = 0; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create EC base key"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_CKR(ret, "CKD_NULL derive without shared data"); + funcList->C_DestroyObject(session, derived); + derived = CK_INVALID_HANDLE; + + params.pSharedData = sharedData; + params.ulSharedDataLen = sizeof(sharedData); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "CKD_NULL derive rejects shared data"); + + params.pSharedData = NULL; + params.ulSharedDataLen = sizeof(sharedData); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "CKD_NULL derive rejects shared data length"); + + params.pSharedData = sharedData; + params.ulSharedDataLen = 0; + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "CKD_NULL derive rejects shared data pointer"); + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} + +/* A well-formed point of the wrong size for the base key's curve is rejected + * raw and DER wrapped, as is a raw point one byte short or long. */ +static int test_wrong_size_point_rejected(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + byte wrapped[2 + sizeof(ecc_p384_point)]; + byte longPoint[sizeof(ecc_p256_point) + 1]; + int result = 0; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create EC base key"); + + ecdh_params_init(¶ms, ecc_p384_point, sizeof(ecc_p384_point)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_FAILS(ret, "raw P-384 point against a P-256 key rejected"); + + wrapped[0] = 0x04; + wrapped[1] = (byte)sizeof(ecc_p384_point); + XMEMCPY(wrapped + 2, ecc_p384_point, sizeof(ecc_p384_point)); + ecdh_params_init(¶ms, wrapped, sizeof(wrapped)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_FAILS(ret, "DER wrapped P-384 point against a P-256 key rejected"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point) - 1); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_FAILS(ret, "raw point one byte short rejected"); + + XMEMCPY(longPoint, ecc_p256_point, sizeof(ecc_p256_point)); + longPoint[sizeof(longPoint) - 1] = 0x00; + ecdh_params_init(¶ms, longPoint, sizeof(longPoint)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_FAILS(ret, "raw point one byte long rejected"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_CKR(ret, "correctly sized point still derives"); + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} + +/* A public data length beyond any encodable EC point is a parameter error. */ +static int test_public_data_length_bound(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + byte bigPoint[512]; + CK_ULONG wideLen; + int result = 0; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create EC base key"); + + if (sizeof(CK_ULONG) > sizeof(word32)) { + /* Upper bits set, low 32 bits equal to the real point length. */ + wideLen = ((CK_ULONG)2 << 16) << 16; + wideLen += sizeof(ecc_p256_point); + ecdh_params_init(¶ms, ecc_p256_point, wideLen); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "public data length above 32 bits rejected"); + } + + XMEMSET(bigPoint, 0, sizeof(bigPoint)); + XMEMCPY(bigPoint, ecc_p256_point, sizeof(ecc_p256_point)); + ecdh_params_init(¶ms, bigPoint, sizeof(bigPoint)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "oversized public data length rejected"); + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} + +/* An EC point wrapped with an unsupported DER length form is not accepted. */ +static int test_unsupported_der_length_form(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + CK_MECHANISM genMech = { CKM_EC_KEY_PAIR_GEN, NULL, 0 }; + byte p521Params[] = { 0x06, 0x05, 0x2B, 0x81, 0x04, 0x00, 0x23 }; + byte derPoint[2 * 66 + 4]; + byte badPoint[2 * 66 + 3]; + CK_ATTRIBUTE pointAttr = { CKA_EC_POINT, derPoint, sizeof(derPoint) }; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_EC_PARAMS, p521Params, sizeof(p521Params) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + }; + int result = 0; + + ret = funcList->C_GenerateKeyPair(session, &genMech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &pub, &priv); + if (ret != CKR_OK) { + printf("SKIP: P-521 key generation unavailable (0x%lx)\n", + (unsigned long)ret); + test_skipped = 1; + goto cleanup; + } + + ret = funcList->C_GetAttributeValue(session, pub, &pointAttr, 1); + CHECK_CKR(ret, "read P-521 EC point"); + if (pointAttr.ulValueLen != sizeof(derPoint) || derPoint[0] != 0x04 || + derPoint[1] != 0x81 || derPoint[2] != sizeof(badPoint) - 2) { + ret = CKR_GENERAL_ERROR; + } + CHECK_CKR(ret, "P-521 EC point is DER wrapped"); + + ecdh_params_init(¶ms, derPoint, sizeof(derPoint)); + ret = ecdh_derive(session, priv, ¶ms, &derived); + CHECK_CKR(ret, "derive with DER wrapped point"); + funcList->C_DestroyObject(session, derived); + derived = CK_INVALID_HANDLE; + + /* Same point with a long-form length byte that has no length octets. */ + badPoint[0] = 0x04; + badPoint[1] = (byte)(sizeof(badPoint) - 2); + XMEMCPY(badPoint + 2, derPoint + 3, sizeof(badPoint) - 2); + ecdh_params_init(¶ms, badPoint, sizeof(badPoint)); + ret = ecdh_derive(session, priv, ¶ms, &derived); + if (ret == CKR_OK) { + fprintf(stderr, "FAIL: malformed DER length form accepted\n"); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: malformed DER length form rejected\n"); + test_passed++; + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (priv != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, priv); + if (pub != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, pub); + return result; +} + +/* The DER length of a wrapped point must fit in the supplied public data. */ +static int test_der_length_within_public_data(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + byte wrapped[3 + sizeof(ecc_p256_point)]; + int result = 0; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create EC base key"); + + wrapped[0] = 0x04; + wrapped[1] = 0x81; + wrapped[2] = (byte)sizeof(ecc_p256_point); + XMEMCPY(wrapped + 3, ecc_p256_point, sizeof(ecc_p256_point)); + + ecdh_params_init(¶ms, wrapped, sizeof(wrapped)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_CKR(ret, "derive with complete DER wrapped point"); + funcList->C_DestroyObject(session, derived); + derived = CK_INVALID_HANDLE; + + /* Declared length runs past the end of the supplied public data. */ + ecdh_params_init(¶ms, wrapped, sizeof(wrapped) - 2); + ret = ecdh_derive(session, base, ¶ms, &derived); + if (ret == CKR_OK) { + fprintf(stderr, "FAIL: truncated DER wrapped point accepted\n"); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: truncated DER wrapped point rejected\n"); + test_passed++; + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} + +/* Repeated derives with one base key leave it usable and give one secret. */ +static int test_repeated_derive_same_key(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + byte secret[sizeof(ecc_secret_256)]; + CK_ATTRIBUTE valAttr = { CKA_VALUE, secret, sizeof(secret) }; + int i; + int result = 0; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create EC base key"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + for (i = 0; i < 3; i++) { + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_CKR(ret, "repeated derive with one base key"); + XMEMSET(secret, 0, sizeof(secret)); + valAttr.ulValueLen = sizeof(secret); + ret = funcList->C_GetAttributeValue(session, derived, &valAttr, 1); + CHECK_CKR(ret, "read derived secret"); + if (valAttr.ulValueLen != sizeof(ecc_secret_256) || + XMEMCMP(secret, ecc_secret_256, sizeof(secret)) != 0) { + ret = CKR_GENERAL_ERROR; + } + CHECK_CKR(ret, "derived secret matches expected value"); + ret = funcList->C_DestroyObject(session, derived); + derived = CK_INVALID_HANDLE; + CHECK_CKR(ret, "destroy derived secret"); + } + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} + +#ifdef EC_DERIVE_THREAD_TEST +#define DERIVE_THREADS 2 +#define DERIVES_PER_THREAD 200 + +typedef struct DeriveCtx { + CK_OBJECT_HANDLE base; + int failures; +} DeriveCtx; + +static pthread_mutex_t deriveGateLock = PTHREAD_MUTEX_INITIALIZER; +static pthread_cond_t deriveGateCond = PTHREAD_COND_INITIALIZER; +static int deriveGateReady = 0; +static int deriveGateOpen = 0; + +/* Hold each worker until every worker has arrived so their derives overlap. */ +static void derive_gate_wait(void) +{ + if (pthread_mutex_lock(&deriveGateLock) == 0) { + deriveGateReady++; + (void)pthread_cond_broadcast(&deriveGateCond); + while (!deriveGateOpen) { + if (pthread_cond_wait(&deriveGateCond, &deriveGateLock) != 0) + break; + } + (void)pthread_mutex_unlock(&deriveGateLock); + } +} + +static int derive_gate_open(int workers) +{ + int ret; + + ret = pthread_mutex_lock(&deriveGateLock); + if (ret == 0) { + while (ret == 0 && deriveGateReady < workers) + ret = pthread_cond_wait(&deriveGateCond, &deriveGateLock); + deriveGateOpen = 1; + if (pthread_cond_broadcast(&deriveGateCond) != 0) + ret = -1; + if (pthread_mutex_unlock(&deriveGateLock) != 0) + ret = -1; + } + return ret; +} + +static void* derive_worker(void* arg) +{ + DeriveCtx* ctx = (DeriveCtx*)arg; + CK_RV ret; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived; + CK_ECDH1_DERIVE_PARAMS params; + byte secret[sizeof(ecc_secret_256)]; + CK_ATTRIBUTE valAttr = { CKA_VALUE, secret, sizeof(secret) }; + int i; + + derive_gate_wait(); + ret = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &session); + if (ret != CKR_OK) { + ctx->failures = DERIVES_PER_THREAD; + return NULL; + } + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + for (i = 0; i < DERIVES_PER_THREAD; i++) { + derived = CK_INVALID_HANDLE; + ret = ecdh_derive(session, ctx->base, ¶ms, &derived); + if (ret == CKR_OK) { + valAttr.ulValueLen = sizeof(secret); + ret = funcList->C_GetAttributeValue(session, derived, &valAttr, 1); + } + if (ret == CKR_OK && (valAttr.ulValueLen != sizeof(ecc_secret_256) || + XMEMCMP(secret, ecc_secret_256, sizeof(secret)) != 0)) { + ret = CKR_GENERAL_ERROR; + } + if (derived != CK_INVALID_HANDLE && ret == CKR_OK) + ret = funcList->C_DestroyObject(session, derived); + else if (derived != CK_INVALID_HANDLE) + (void)funcList->C_DestroyObject(session, derived); + if (ret != CKR_OK) + ctx->failures++; + } + + if (funcList->C_CloseSession(session) != CKR_OK) + ctx->failures++; + return NULL; +} + +/* Derives that share one token key run concurrently and all succeed. */ +static int test_concurrent_derive_same_key(CK_SESSION_HANDLE session) +{ + CK_RV ret = CKR_OK; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &privKeyClass, sizeof(privKeyClass) }, + { CKA_KEY_TYPE, &eccKeyType, sizeof(eccKeyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + pthread_t threads[DERIVE_THREADS]; + DeriveCtx ctx[DERIVE_THREADS]; + int started = 0; + int failures = 0; + int i; + int result = 0; + + ret = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &base); + CHECK_CKR(ret, "create EC token base key"); + + for (i = 0; i < DERIVE_THREADS; i++) { + ctx[i].base = base; + ctx[i].failures = 0; + if (pthread_create(&threads[i], NULL, derive_worker, &ctx[i]) != 0) + break; + started++; + } + if (derive_gate_open(started) != 0) + failures++; + for (i = 0; i < started; i++) { + if (pthread_join(threads[i], NULL) != 0) + failures++; + failures += ctx[i].failures; + } + if (started != DERIVE_THREADS || failures != 0) { + fprintf(stderr, "%d of %d concurrent derives failed\n", failures, + DERIVE_THREADS * DERIVES_PER_THREAD); + ret = CKR_GENERAL_ERROR; + } + CHECK_CKR(ret, "concurrent derives with one token key all succeed"); + +cleanup: + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} +#endif /* EC_DERIVE_THREAD_TEST */ + +/* Deriving a private key object requires a user login. */ +static int test_private_derive_requires_login(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + CK_ATTRIBUTE privTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &secretLen, sizeof(secretLen) }, + }; +#ifndef WOLFPKCS11_LEGACY_PRIVATE_FALSE_DEFAULT + CK_ATTRIBUTE defTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_VALUE_LEN, &secretLen, sizeof(secretLen) }, + }; +#endif + int loggedOut = 0; + int result = 0; + + ret = funcList->C_Logout(session); + CHECK_CKR(ret, "C_Logout"); + loggedOut = 1; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create public EC base key without login"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + ret = ecdh_derive(session, base, ¶ms, &derived); + CHECK_CKR(ret, "derive public key without login"); + funcList->C_DestroyObject(session, derived); + derived = CK_INVALID_HANDLE; + + ret = ecdh_derive_tmpl(session, base, ¶ms, privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &derived); + CHECK_RV(ret, CKR_USER_NOT_LOGGED_IN, + "derive private key without login rejected"); + +#ifndef WOLFPKCS11_LEGACY_PRIVATE_FALSE_DEFAULT + ret = ecdh_derive_tmpl(session, base, ¶ms, defTmpl, + sizeof(defTmpl) / sizeof(*defTmpl), &derived); + CHECK_RV(ret, CKR_USER_NOT_LOGGED_IN, + "derive default-private secret key without login rejected"); +#endif + + ret = funcList->C_Login(session, CKU_USER, userPin, userPinLen); + CHECK_CKR(ret, "C_Login"); + loggedOut = 0; + + ret = ecdh_derive_tmpl(session, base, ¶ms, privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &derived); + CHECK_CKR(ret, "derive private key after login"); + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + if (loggedOut && + funcList->C_Login(session, CKU_USER, userPin, userPinLen) != + CKR_OK) { + result = -1; + } + return result; +} + +static CK_RV read_states(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_BBOOL* alwaysSensitive, CK_BBOOL* neverExtractable) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_ALWAYS_SENSITIVE, alwaysSensitive, sizeof(CK_BBOOL) }, + { CKA_NEVER_EXTRACTABLE, neverExtractable, sizeof(CK_BBOOL) }, + }; + + return funcList->C_GetAttributeValue(session, obj, tmpl, + sizeof(tmpl) / sizeof(*tmpl)); +} + +/* A derived key is only always-sensitive / never-extractable when its base + * key has those historical properties too. */ +static int test_derived_states_follow_base(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + CK_MECHANISM genMech = { CKM_EC_KEY_PAIR_GEN, NULL, 0 }; + CK_BBOOL alwaysSensitive = 0xAA; + CK_BBOOL neverExtractable = 0xAA; + byte point[2 * 32 + 3]; + CK_ATTRIBUTE pointAttr = { CKA_EC_POINT, point, sizeof(point) }; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + }; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE_LEN, &secretLen, sizeof(secretLen) }, + }; + int result = 0; + + /* An imported key was never always-sensitive or never-extractable. */ + ret = create_ec_base(session, CK_TRUE, CK_FALSE, &base); + CHECK_CKR(ret, "create imported EC base key"); + ret = read_states(session, base, &alwaysSensitive, &neverExtractable); + CHECK_CKR(ret, "read imported base historical flags"); + if (alwaysSensitive != CK_FALSE || neverExtractable != CK_FALSE) + ret = CKR_GENERAL_ERROR; + CHECK_CKR(ret, "imported base historical flags are FALSE"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + ret = ecdh_derive_tmpl(session, base, ¶ms, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &derived); + CHECK_CKR(ret, "derive from imported base key"); + ret = read_states(session, derived, &alwaysSensitive, &neverExtractable); + CHECK_CKR(ret, "read derived historical flags"); + if (alwaysSensitive != CK_FALSE || neverExtractable != CK_FALSE) { + fprintf(stderr, "FAIL: derived from imported base: " + "ALWAYS_SENSITIVE=%d NEVER_EXTRACTABLE=%d, expected FALSE\n", + (int)alwaysSensitive, (int)neverExtractable); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: derived key does not claim history its base lacks\n"); + test_passed++; + funcList->C_DestroyObject(session, derived); + derived = CK_INVALID_HANDLE; + funcList->C_DestroyObject(session, base); + base = CK_INVALID_HANDLE; + + /* A generated sensitive, non-extractable key passes its history on. */ + ret = funcList->C_GenerateKeyPair(session, &genMech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &pub, &base); + CHECK_CKR(ret, "generate EC base key"); + ret = funcList->C_GetAttributeValue(session, pub, &pointAttr, 1); + CHECK_CKR(ret, "read generated EC point"); + + ecdh_params_init(¶ms, point, pointAttr.ulValueLen); + ret = ecdh_derive_tmpl(session, base, ¶ms, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &derived); + CHECK_CKR(ret, "derive from generated base key"); + ret = read_states(session, derived, &alwaysSensitive, &neverExtractable); + CHECK_CKR(ret, "read derived historical flags"); + if (alwaysSensitive != CK_TRUE || neverExtractable != CK_TRUE) { + fprintf(stderr, "FAIL: derived from generated base: " + "ALWAYS_SENSITIVE=%d NEVER_EXTRACTABLE=%d, expected TRUE\n", + (int)alwaysSensitive, (int)neverExtractable); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: derived key keeps history its base has\n"); + test_passed++; + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + if (pub != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, pub); + return result; +} + +/* A derived key type that cannot hold the secret fails with a defined + * PKCS#11 return value. */ +static int test_unusable_key_type_defined_error(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &eccKeyType, sizeof(eccKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + int result = 0; + + ret = create_ec_base(session, CK_FALSE, CK_TRUE, &base); + CHECK_CKR(ret, "create EC base key"); + + ecdh_params_init(¶ms, ecc_p256_point, sizeof(ecc_p256_point)); + ret = ecdh_derive_tmpl(session, base, ¶ms, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &derived); + if (ret == CKR_OK || ret >= CKR_VENDOR_DEFINED) { + fprintf(stderr, "FAIL: derive to EC secret key returned 0x%lx\n", + (unsigned long)ret); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: derive to unusable key type returns 0x%lx\n", + (unsigned long)ret); + test_passed++; + +cleanup: + if (derived != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, derived); + if (base != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, base); + return result; +} +#endif /* HAVE_ECC */ + +static CK_RV pkcs11_init(void) +{ + CK_RV ret; + CK_C_INITIALIZE_ARGS args; + CK_SLOT_ID slotList[16]; + CK_ULONG slotCount = sizeof(slotList) / sizeof(slotList[0]); +#ifndef HAVE_PKCS11_STATIC + CK_C_GetFunctionList func; + + dlib = dlopen(WOLFPKCS11_DLL_FILENAME, RTLD_NOW | RTLD_LOCAL); + if (dlib == NULL) { + fprintf(stderr, "dlopen error: %s\n", dlerror()); + return -1; + } + func = (CK_C_GetFunctionList)dlsym(dlib, "C_GetFunctionList"); + if (func == NULL) { + dlclose(dlib); + dlib = NULL; + return -1; + } + ret = func(&funcList); + if (ret != CKR_OK) { + dlclose(dlib); + dlib = NULL; + return ret; + } +#else + ret = C_GetFunctionList(&funcList); + if (ret != CKR_OK) + return ret; +#endif + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + ret = funcList->C_Initialize(&args); + if (ret != CKR_OK) + return ret; + + ret = funcList->C_GetSlotList(CK_TRUE, slotList, &slotCount); + if (ret != CKR_OK) + return ret; + if (slotCount == 0) + return CKR_GENERAL_ERROR; + slot = slotList[0]; + return ret; +} + +static CK_RV pkcs11_final(void) +{ + if (funcList != NULL) { + funcList->C_Finalize(NULL); + funcList = NULL; + } +#ifndef HAVE_PKCS11_STATIC + if (dlib) { + dlclose(dlib); + dlib = NULL; + } +#endif + return CKR_OK; +} + +static CK_RV pkcs11_setup_token(void) +{ + CK_RV ret; + unsigned char label[32]; + CK_SESSION_HANDLE soSession; + int sessFlags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, tokenName, XSTRLEN(tokenName)); + ret = funcList->C_InitToken(slot, soPin, soPinLen, label); + if (ret != CKR_OK) + return ret; + + ret = funcList->C_OpenSession(slot, sessFlags, NULL, NULL, &soSession); + if (ret != CKR_OK) + return ret; + ret = funcList->C_Login(soSession, CKU_SO, soPin, soPinLen); + if (ret == CKR_OK) + ret = funcList->C_InitPIN(soSession, userPin, userPinLen); + funcList->C_Logout(soSession); + funcList->C_CloseSession(soSession); + return ret; +} + +static int ec_derive_test(void) +{ + CK_RV ret; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + int sessFlags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + int result = 0; + + printf("\n=== Testing C_DeriveKey with CKM_ECDH1_DERIVE ===\n"); + + ret = pkcs11_init(); + CHECK_CKR(ret, "C_Initialize"); + + ret = pkcs11_setup_token(); + CHECK_CKR(ret, "token setup"); + + ret = funcList->C_OpenSession(slot, sessFlags, NULL, NULL, &session); + CHECK_CKR(ret, "C_OpenSession"); + ret = funcList->C_Login(session, CKU_USER, userPin, userPinLen); + CHECK_CKR(ret, "C_Login"); + +#ifdef HAVE_ECC + if (test_null_kdf_shared_data(session) != 0) + result = -1; + if (test_wrong_size_point_rejected(session) != 0) + result = -1; + if (test_public_data_length_bound(session) != 0) + result = -1; + if (test_unsupported_der_length_form(session) != 0) + result = -1; + if (test_der_length_within_public_data(session) != 0) + result = -1; + if (test_repeated_derive_same_key(session) != 0) + result = -1; +#ifdef EC_DERIVE_THREAD_TEST + if (test_concurrent_derive_same_key(session) != 0) + result = -1; +#endif + if (test_private_derive_requires_login(session) != 0) + result = -1; + if (test_derived_states_follow_base(session) != 0) + result = -1; + if (test_unusable_key_type_defined_error(session) != 0) + result = -1; +#else + printf("ECC not available, skipping ECDH derive tests\n"); + test_skipped = 1; +#endif + +cleanup: + if (session != CK_INVALID_HANDLE) { + funcList->C_Logout(session); + funcList->C_CloseSession(session); + } + pkcs11_final(); + return result; +} + +#ifdef EC_DERIVE_PERSIST_TEST +static const char persistLabel[] = "ec-derive-persist"; + +static CK_RV open_user_session(CK_SESSION_HANDLE* session) +{ + CK_RV ret; + + ret = pkcs11_init(); + if (ret == CKR_OK) { + ret = funcList->C_OpenSession(slot, + CKF_SERIAL_SESSION | CKF_RW_SESSION, NULL, NULL, session); + } + if (ret == CKR_OK) + ret = funcList->C_Login(*session, CKU_USER, userPin, userPinLen); + return ret; +} + +/* Derive a sensitive, non-extractable token key from a generated base key. */ +static CK_RV derive_token_key(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_MECHANISM genMech = { CKM_EC_KEY_PAIR_GEN, NULL, 0 }; + CK_ECDH1_DERIVE_PARAMS params; + byte point[2 * 32 + 3]; + CK_ATTRIBUTE pointAttr = { CKA_EC_POINT, point, sizeof(point) }; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + }; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE_LEN, &secretLen, sizeof(secretLen) }, + { CKA_LABEL, (void*)persistLabel, sizeof(persistLabel) - 1 }, + }; + + ret = funcList->C_GenerateKeyPair(session, &genMech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &pub, &priv); + if (ret == CKR_OK) + ret = funcList->C_GetAttributeValue(session, pub, &pointAttr, 1); + if (ret == CKR_OK) { + ecdh_params_init(¶ms, point, pointAttr.ulValueLen); + ret = ecdh_derive_tmpl(session, priv, ¶ms, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &derived); + } + return ret; +} + +/* Historical protection flags of a derived token key survive a reload even + * when the process ends without C_Finalize. */ +static int test_token_derive_states_persist(void) +{ + CK_RV ret; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ULONG found = 0; + CK_BBOOL alwaysSensitive = CK_FALSE; + CK_BBOOL neverExtractable = CK_FALSE; + CK_ATTRIBUTE findTmpl[] = { + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, (void*)persistLabel, sizeof(persistLabel) - 1 }, + }; + CK_ATTRIBUTE stateTmpl[] = { + { CKA_ALWAYS_SENSITIVE, &alwaysSensitive, sizeof(CK_BBOOL) }, + { CKA_NEVER_EXTRACTABLE, &neverExtractable, sizeof(CK_BBOOL) }, + }; + pid_t pid; + int status = 0; + int result = 0; + + printf("\n=== Testing derived token key state persistence ===\n"); + + pid = fork(); + if (pid == 0) { + /* Exit without C_Finalize so only stores made by the derive count. */ + ret = open_user_session(&session); + if (ret == CKR_OK) + ret = derive_token_key(session); + _exit(ret == CKR_OK ? 0 : 1); + } + if (pid < 0) + ret = CKR_GENERAL_ERROR; + else if (waitpid(pid, &status, 0) != pid || !WIFEXITED(status) || + WEXITSTATUS(status) != 0) + ret = CKR_GENERAL_ERROR; + else + ret = CKR_OK; + CHECK_CKR(ret, "derive token key in child process"); + + ret = open_user_session(&session); + CHECK_CKR(ret, "reopen token"); + + ret = funcList->C_FindObjectsInit(session, findTmpl, + sizeof(findTmpl) / sizeof(*findTmpl)); + CHECK_CKR(ret, "C_FindObjectsInit"); + ret = funcList->C_FindObjects(session, &obj, 1, &found); + if (funcList->C_FindObjectsFinal(session) != CKR_OK && ret == CKR_OK) + ret = CKR_GENERAL_ERROR; + if (ret == CKR_OK && found != 1) + ret = CKR_GENERAL_ERROR; + CHECK_CKR(ret, "find derived token key"); + + ret = funcList->C_GetAttributeValue(session, obj, stateTmpl, + sizeof(stateTmpl) / sizeof(*stateTmpl)); + CHECK_CKR(ret, "read derived key historical flags"); + if (alwaysSensitive != CK_TRUE || neverExtractable != CK_TRUE) { + fprintf(stderr, "FAIL: reloaded derived key ALWAYS_SENSITIVE=%d " + "NEVER_EXTRACTABLE=%d, expected both TRUE\n", + (int)alwaysSensitive, (int)neverExtractable); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: reloaded derived key keeps its historical flags\n"); + test_passed++; + +cleanup: + if (obj != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, obj); + if (session != CK_INVALID_HANDLE) { + funcList->C_Logout(session); + funcList->C_CloseSession(session); + } + pkcs11_final(); + return result; +} +#endif /* EC_DERIVE_PERSIST_TEST */ + +static void print_results(void) +{ + printf("\n=== Test Results ===\n"); + printf("Tests passed: %d\n", test_passed); + printf("Tests failed: %d\n", test_failed); + if (test_skipped != 0) + printf("Tests skipped: %d\n", test_skipped); + if (test_failed == 0) + printf("ALL TESTS PASSED!\n"); + else + printf("SOME TESTS FAILED!\n"); +} + +int main(int argc, char* argv[]) +{ +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + (void)argc; + (void)argv; + + if (ec_derive_test() != 0 && test_failed == 0) + test_failed++; +#ifdef EC_DERIVE_PERSIST_TEST + if (test_token_derive_states_persist() != 0 && test_failed == 0) + test_failed++; +#endif + + print_results(); +#ifndef HAVE_ECC + if (test_failed == 0) + return 77; +#endif + return (test_failed == 0) ? 0 : 1; +} diff --git a/tests/encrypt_decrypt_params_test.c b/tests/encrypt_decrypt_params_test.c new file mode 100644 index 00000000..95318e8f --- /dev/null +++ b/tests/encrypt_decrypt_params_test.c @@ -0,0 +1,1467 @@ +/* encrypt_decrypt_params_test.c + * + * Copyright (C) 2026 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Checks that encrypt/decrypt operations report output lengths correctly and + * validate mechanism parameters and data lengths. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#define TEST_DIR "./store/encrypt_decrypt_params_test" + +#ifndef NO_AES +static CK_RV create_aes_key(CK_SESSION_HANDLE session, unsigned char* key, + CK_ULONG keyLen, CK_OBJECT_HANDLE* hKey) +{ + CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; + CK_KEY_TYPE aesType = CKK_AES; + CK_BBOOL ckTrue = CK_TRUE; + CK_BBOOL ckFalse = CK_FALSE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_VALUE, NULL, 0 }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + + tmpl[2].pValue = key; + tmpl[2].ulValueLen = keyLen; + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), hKey); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) +/* C_EncryptFinal must report the number of bytes it wrote, not leave the + * caller's buffer capacity in place. */ +static void test_cbc_pad_encrypt_final_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_MECHANISM mech; + byte iv[16]; + byte plain[20]; + byte enc[64]; + byte dec[64]; + CK_ULONG updLen; + CK_ULONG lastLen; + CK_ULONG decLen; + + XMEMSET(iv, 0x11, sizeof(iv)); + XMEMSET(plain, 0x22, sizeof(plain)); + XMEMSET(enc, 0, sizeof(enc)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "CBC-PAD: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + mech.mechanism = CKM_AES_CBC_PAD; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "CBC-PAD: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + updLen = sizeof(enc); + rv = funcList->C_EncryptUpdate(session, plain, sizeof(plain), enc, + &updLen); + CHECK_RV(rv, "CBC-PAD: C_EncryptUpdate", CKR_OK); + if (rv != CKR_OK) + return; + CHECK_TRUE(updLen == 16, "CBC-PAD: update emits one full block"); + + lastLen = sizeof(enc) - updLen; + rv = funcList->C_EncryptFinal(session, enc + updLen, &lastLen); + CHECK_RV(rv, "CBC-PAD: C_EncryptFinal (large buffer)", CKR_OK); + CHECK_TRUE(lastLen == 16, + "CBC-PAD: C_EncryptFinal reports the bytes written"); + if (rv != CKR_OK || lastLen != 16) + return; + + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "CBC-PAD: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + decLen = sizeof(dec); + rv = funcList->C_Decrypt(session, enc, updLen + lastLen, dec, &decLen); + CHECK_RV(rv, "CBC-PAD: C_Decrypt of multi-part output", CKR_OK); + CHECK_TRUE(decLen == sizeof(plain) && + XMEMCMP(dec, plain, sizeof(plain)) == 0, + "CBC-PAD: multi-part ciphertext round-trips"); +} +#endif + +#if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) +static CK_RV create_rsa_keys(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* pub, + CK_OBJECT_HANDLE* priv) +{ + CK_RV rv; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE rsaType = CKK_RSA; + CK_BBOOL ckTrue = CK_TRUE; + CK_BBOOL ckFalse = CK_FALSE; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + + rv = funcList->C_CreateObject(session, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), pub); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), + priv); + } + return rv; +} + +static void oaep_params_init(CK_RSA_PKCS_OAEP_PARAMS* params, + CK_MECHANISM* mech) +{ + params->hashAlg = CKM_SHA256; + params->mgf = CKG_MGF1_SHA256; + params->source = CKZ_DATA_SPECIFIED; + params->pSourceData = NULL; + params->ulSourceDataLen = 0; + mech->mechanism = CKM_RSA_PKCS_OAEP; + mech->pParameter = params; + mech->ulParameterLen = sizeof(*params); +} + +/* An OAEP encoding parameter with no source data must have zero length. */ +static void test_oaep_source_ptr_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_RSA_PKCS_OAEP_PARAMS params; + CK_MECHANISM mech; + byte plain[16]; + byte enc[256]; + byte dec[256]; + CK_ULONG encLen; + CK_ULONG decLen; + + XMEMSET(plain, 0x33, sizeof(plain)); + rv = create_rsa_keys(session, &pub, &priv); + CHECK_RV(rv, "OAEP: create RSA keys", CKR_OK); + if (rv != CKR_OK) + return; + + oaep_params_init(¶ms, &mech); + params.ulSourceDataLen = 1; + rv = funcList->C_EncryptInit(session, &mech, pub); + CHECK_RV(rv, "OAEP: C_EncryptInit NULL source with length", + CKR_MECHANISM_PARAM_INVALID); + rv = funcList->C_DecryptInit(session, &mech, priv); + CHECK_RV(rv, "OAEP: C_DecryptInit NULL source with length", + CKR_MECHANISM_PARAM_INVALID); + + oaep_params_init(¶ms, &mech); + rv = funcList->C_EncryptInit(session, &mech, pub); + CHECK_RV(rv, "OAEP: C_EncryptInit empty source", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "OAEP: C_Encrypt empty source", CKR_OK); + if (rv != CKR_OK) + return; + rv = funcList->C_DecryptInit(session, &mech, priv); + CHECK_RV(rv, "OAEP: C_DecryptInit empty source", CKR_OK); + if (rv != CKR_OK) + return; + decLen = sizeof(dec); + rv = funcList->C_Decrypt(session, enc, encLen, dec, &decLen); + CHECK_RV(rv, "OAEP: C_Decrypt empty source", CKR_OK); + CHECK_TRUE(decLen == sizeof(plain) && + XMEMCMP(dec, plain, sizeof(plain)) == 0, + "OAEP: empty source round-trips"); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESCCM) +static void ccm_params_init(CK_CCM_PARAMS* params, CK_MECHANISM* mech, + byte* iv, CK_ULONG ivLen) +{ + XMEMSET(params, 0, sizeof(*params)); + params->pIv = iv; + params->ulIvLen = ivLen; + params->ulMacLen = 16; + mech->mechanism = CKM_AES_CCM; + mech->pParameter = params; + mech->ulParameterLen = sizeof(*params); +} + +/* CCM nonces must be 7 to 13 bytes long (NIST SP 800-38C). */ +static void test_ccm_nonce_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_CCM_PARAMS params; + CK_MECHANISM mech; + byte iv[16]; + byte plain[16]; + byte enc[32]; + CK_ULONG encLen; + static const CK_ULONG badLens[] = { 1, 6, 14, 16 }; + int i; + + XMEMSET(iv, 0x44, sizeof(iv)); + XMEMSET(plain, 0x55, sizeof(plain)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "CCM: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + for (i = 0; i < (int)(sizeof(badLens) / sizeof(*badLens)); i++) { + ccm_params_init(¶ms, &mech, iv, badLens[i]); + printf("CCM: nonce length %lu\n", (unsigned long)badLens[i]); + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "CCM: C_EncryptInit rejects nonce length", + CKR_MECHANISM_PARAM_INVALID); + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "CCM: C_DecryptInit rejects nonce length", + CKR_MECHANISM_PARAM_INVALID); + } + + ccm_params_init(¶ms, &mech, iv, 7); + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "CCM: C_EncryptInit nonce length 7", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "CCM: C_Encrypt nonce length 7", CKR_OK); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESCTR) +/* A NULL output buffer to C_Decrypt is a length query that leaves the + * operation active. */ +static void test_ctr_decrypt_size_query(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_AES_CTR_PARAMS params; + CK_MECHANISM mech; + byte plain[32]; + byte enc[32]; + byte dec[32]; + CK_ULONG encLen; + CK_ULONG decLen; + + XMEMSET(¶ms, 0, sizeof(params)); + params.ulCounterBits = 32; + XMEMSET(params.cb, 0x66, sizeof(params.cb)); + XMEMSET(plain, 0x77, sizeof(plain)); + mech.mechanism = CKM_AES_CTR; + mech.pParameter = ¶ms; + mech.ulParameterLen = sizeof(params); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "CTR: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "CTR: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "CTR: C_Encrypt", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "CTR: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + decLen = 0; + rv = funcList->C_Decrypt(session, enc, encLen, NULL, &decLen); + CHECK_RV(rv, "CTR: C_Decrypt length query", CKR_OK); + CHECK_TRUE(decLen == encLen, "CTR: length query reports plaintext size"); + + decLen = sizeof(dec); + rv = funcList->C_Decrypt(session, enc, encLen, dec, &decLen); + CHECK_RV(rv, "CTR: C_Decrypt after length query", CKR_OK); + CHECK_TRUE(decLen == sizeof(plain) && + XMEMCMP(dec, plain, sizeof(plain)) == 0, + "CTR: decrypt after length query round-trips"); +} +#endif + +#if !defined(NO_AES) && (defined(HAVE_AES_CBC) || defined(HAVE_AESECB)) +/* Unpadded block modes reject data that is not a whole number of blocks with + * the PKCS#11 length-range codes, and the failed call ends the operation. */ +static void check_block_alignment(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE key, CK_MECHANISM* mech) +{ + CK_RV rv; + byte data[32]; + byte out[32]; + CK_ULONG outLen; + + XMEMSET(data, 0x5A, sizeof(data)); + + rv = funcList->C_EncryptInit(session, mech, key); + CHECK_RV(rv, "block align: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, data, 17, out, &outLen); + CHECK_RV(rv, "block align: C_Encrypt of 17 bytes", CKR_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, data, 16, out, &outLen); + CHECK_RV(rv, "block align: encrypt operation ended after length error", + CKR_OPERATION_NOT_INITIALIZED); + + rv = funcList->C_DecryptInit(session, mech, key); + CHECK_RV(rv, "block align: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + outLen = sizeof(out); + rv = funcList->C_Decrypt(session, data, 15, out, &outLen); + CHECK_RV(rv, "block align: C_Decrypt of 15 bytes", + CKR_ENCRYPTED_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_Decrypt(session, data, 16, out, &outLen); + CHECK_RV(rv, "block align: decrypt operation ended after length error", + CKR_OPERATION_NOT_INITIALIZED); +} + +static void test_block_mode_alignment(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_MECHANISM mech; +#ifdef HAVE_AES_CBC + byte iv[16]; +#endif + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "block align: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + +#ifdef HAVE_AES_CBC + XMEMSET(iv, 0x12, sizeof(iv)); + mech.mechanism = CKM_AES_CBC; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + printf("AES-CBC\n"); + check_block_alignment(session, key, &mech); +#endif +#ifdef HAVE_AESECB + mech.mechanism = CKM_AES_ECB; + mech.pParameter = NULL; + mech.ulParameterLen = 0; + printf("AES-ECB\n"); + check_block_alignment(session, key, &mech); +#endif +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESECB) +/* CKA_CHECK_VALUE of an AES key is the first three bytes of one zero block + * encrypted under the key, for every AES key size. */ +static void test_aes_check_value_key_sizes(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key; + CK_MECHANISM mech; + byte keyData[32]; + byte zero[16]; + byte enc[16]; + byte kcv[3]; + CK_ULONG encLen; + CK_ATTRIBUTE attr; + static const CK_ULONG keyLens[] = { 16, 24, 32 }; + int i; + + for (i = 0; i < (int)sizeof(keyData); i++) + keyData[i] = (byte)(0xA0 + i); + XMEMSET(zero, 0, sizeof(zero)); + mech.mechanism = CKM_AES_ECB; + mech.pParameter = NULL; + mech.ulParameterLen = 0; + + for (i = 0; i < (int)(sizeof(keyLens) / sizeof(*keyLens)); i++) { + printf("check value: AES key length %lu\n", (unsigned long)keyLens[i]); + key = CK_INVALID_HANDLE; + rv = create_aes_key(session, keyData, keyLens[i], &key); + CHECK_RV(rv, "check value: create AES key", CKR_OK); + if (rv != CKR_OK) + continue; + + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "check value: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + continue; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, zero, sizeof(zero), enc, &encLen); + CHECK_RV(rv, "check value: C_Encrypt zero block", CKR_OK); + if (rv != CKR_OK) + continue; + + XMEMSET(kcv, 0, sizeof(kcv)); + attr.type = CKA_CHECK_VALUE; + attr.pValue = kcv; + attr.ulValueLen = sizeof(kcv); + rv = funcList->C_GetAttributeValue(session, key, &attr, 1); + CHECK_RV(rv, "check value: C_GetAttributeValue", CKR_OK); + CHECK_TRUE(rv == CKR_OK && attr.ulValueLen == sizeof(kcv) && + XMEMCMP(kcv, enc, sizeof(kcv)) == 0, + "check value matches encrypted zero block"); + } +} +#endif + +typedef void (*test_fn)(CK_SESSION_HANDLE session); + +/* Each check runs in its own session so session objects and operation state + * do not carry over between checks. */ +static void run_in_session(CK_SLOT_ID slot, test_fn fn) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + + rv = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &session); + CHECK_RV(rv, "open test session", CKR_OK); + if (rv == CKR_OK) { + fn(session); + funcList->C_CloseSession(session); + } +} + +#if !defined(NO_AES) && defined(HAVE_AESGCM) +static void gcm_params_init(CK_GCM_PARAMS* params, CK_MECHANISM* mech, + byte* iv, byte* aad) +{ + XMEMSET(params, 0, sizeof(*params)); + params->pIv = iv; + params->ulIvLen = 12; + params->pAAD = aad; + params->ulAADLen = 16; + params->ulTagBits = 128; + mech->mechanism = CKM_AES_GCM; + mech->pParameter = params; + mech->ulParameterLen = sizeof(*params); +} + +static void check_gcm_params_rejected(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE key, CK_MECHANISM* mech) +{ + CK_RV rv; + + rv = funcList->C_EncryptInit(session, mech, key); + CHECK_RV(rv, "GCM: C_EncryptInit rejects parameter", + CKR_MECHANISM_PARAM_INVALID); + rv = funcList->C_DecryptInit(session, mech, key); + CHECK_RV(rv, "GCM: C_DecryptInit rejects parameter", + CKR_MECHANISM_PARAM_INVALID); +} + +/* CK_GCM_PARAMS lengths outside what the token supports are rejected rather + * than narrowed. */ +static void test_gcm_param_ranges(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_GCM_PARAMS params; + CK_MECHANISM mech; + byte iv[12]; + byte aad[16]; + byte plain[16]; + byte enc[32]; + CK_ULONG encLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + XMEMSET(iv, 0x21, sizeof(iv)); + XMEMSET(aad, 0x43, sizeof(aad)); + XMEMSET(plain, 0x65, sizeof(plain)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "GCM: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + printf("GCM: tag bits 0xFFFFFFFF\n"); + gcm_params_init(¶ms, &mech, iv, aad); + params.ulTagBits = 0xFFFFFFFFUL; + check_gcm_params_rejected(session, key, &mech); + + printf("GCM: tag bits 0x80000000\n"); + gcm_params_init(¶ms, &mech, iv, aad); + params.ulTagBits = 0x80000000UL; + check_gcm_params_rejected(session, key, &mech); + + printf("GCM: AAD length 0x80000000\n"); + gcm_params_init(¶ms, &mech, iv, aad); + params.ulAADLen = 0x80000000UL; + check_gcm_params_rejected(session, key, &mech); + + if (wrap != 0) { + printf("GCM: tag bits 2^32 + 128\n"); + gcm_params_init(¶ms, &mech, iv, aad); + params.ulTagBits = wrap + 128; + check_gcm_params_rejected(session, key, &mech); + + printf("GCM: IV length 2^32 + 12\n"); + gcm_params_init(¶ms, &mech, iv, aad); + params.ulIvLen = wrap + sizeof(iv); + check_gcm_params_rejected(session, key, &mech); + + printf("GCM: AAD length 2^32 + 16\n"); + gcm_params_init(¶ms, &mech, iv, aad); + params.ulAADLen = wrap + sizeof(aad); + check_gcm_params_rejected(session, key, &mech); + } + + gcm_params_init(¶ms, &mech, iv, aad); + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "GCM: C_EncryptInit valid parameters", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "GCM: C_Encrypt valid parameters", CKR_OK); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESCCM) +static void check_ccm_params_rejected(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE key, CK_MECHANISM* mech) +{ + CK_RV rv; + + rv = funcList->C_EncryptInit(session, mech, key); + CHECK_RV(rv, "CCM: C_EncryptInit rejects parameter", + CKR_MECHANISM_PARAM_INVALID); + rv = funcList->C_DecryptInit(session, mech, key); + CHECK_RV(rv, "CCM: C_DecryptInit rejects parameter", + CKR_MECHANISM_PARAM_INVALID); +} + +/* CK_CCM_PARAMS lengths must be in range for CCM and are rejected rather than + * narrowed. */ +static void test_ccm_param_ranges(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_CCM_PARAMS params; + CK_MECHANISM mech; + byte iv[13]; + byte aad[16]; + byte plain[16]; + byte enc[32]; + CK_ULONG encLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + static const CK_ULONG badMacLens[] = { 0, 2, 5, 15, 18 }; + int i; + + XMEMSET(iv, 0x31, sizeof(iv)); + XMEMSET(aad, 0x53, sizeof(aad)); + XMEMSET(plain, 0x75, sizeof(plain)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "CCM: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + for (i = 0; i < (int)(sizeof(badMacLens) / sizeof(*badMacLens)); i++) { + printf("CCM: MAC length %lu\n", (unsigned long)badMacLens[i]); + ccm_params_init(¶ms, &mech, iv, sizeof(iv)); + params.ulMacLen = badMacLens[i]; + check_ccm_params_rejected(session, key, &mech); + } + + printf("CCM: AAD length 0x80000000\n"); + ccm_params_init(¶ms, &mech, iv, sizeof(iv)); + params.pAAD = aad; + params.ulAADLen = 0x80000000UL; + check_ccm_params_rejected(session, key, &mech); + + printf("CCM: data length 0x80000000\n"); + ccm_params_init(¶ms, &mech, iv, sizeof(iv)); + params.ulDataLen = 0x80000000UL; + check_ccm_params_rejected(session, key, &mech); + + if (wrap != 0) { + printf("CCM: MAC length 2^32 + 4\n"); + ccm_params_init(¶ms, &mech, iv, sizeof(iv)); + params.ulMacLen = wrap + 4; + check_ccm_params_rejected(session, key, &mech); + + printf("CCM: AAD length 2^32 + 16\n"); + ccm_params_init(¶ms, &mech, iv, sizeof(iv)); + params.pAAD = aad; + params.ulAADLen = wrap + sizeof(aad); + check_ccm_params_rejected(session, key, &mech); + + printf("CCM: nonce length 2^32 + 13\n"); + ccm_params_init(¶ms, &mech, iv, wrap + sizeof(iv)); + check_ccm_params_rejected(session, key, &mech); + } + + ccm_params_init(¶ms, &mech, iv, sizeof(iv)); + params.pAAD = aad; + params.ulAADLen = sizeof(aad); + params.ulMacLen = 8; + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "CCM: C_EncryptInit valid parameters", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "CCM: C_Encrypt valid parameters", CKR_OK); + CHECK_TRUE(encLen == sizeof(plain) + 8, "CCM: output carries 8-byte MAC"); +} +#endif + +#if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) +static void check_oaep_params_rejected(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE pub, + CK_OBJECT_HANDLE priv, + CK_MECHANISM* mech) +{ + CK_RV rv; + + rv = funcList->C_EncryptInit(session, mech, pub); + CHECK_RV(rv, "OAEP: C_EncryptInit rejects source length", + CKR_MECHANISM_PARAM_INVALID); + rv = funcList->C_DecryptInit(session, mech, priv); + CHECK_RV(rv, "OAEP: C_DecryptInit rejects source length", + CKR_MECHANISM_PARAM_INVALID); +} + +/* An OAEP source length the token cannot represent is rejected rather than + * narrowed to a different label. */ +static void test_oaep_source_len_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_RSA_PKCS_OAEP_PARAMS params; + CK_MECHANISM mech; + byte label[16]; + byte plain[16]; + byte enc[256]; + byte dec[256]; + CK_ULONG encLen; + CK_ULONG decLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + XMEMSET(label, 0x4C, sizeof(label)); + XMEMSET(plain, 0x50, sizeof(plain)); + + rv = create_rsa_keys(session, &pub, &priv); + CHECK_RV(rv, "OAEP: create RSA keys", CKR_OK); + if (rv != CKR_OK) + return; + + printf("OAEP: source length 0x80000000\n"); + oaep_params_init(¶ms, &mech); + params.pSourceData = label; + params.ulSourceDataLen = 0x80000000UL; + check_oaep_params_rejected(session, pub, priv, &mech); + + if (wrap != 0) { + printf("OAEP: source length 2^32 + 16\n"); + oaep_params_init(¶ms, &mech); + params.pSourceData = label; + params.ulSourceDataLen = wrap + sizeof(label); + check_oaep_params_rejected(session, pub, priv, &mech); + + printf("OAEP: source length 2^32\n"); + oaep_params_init(¶ms, &mech); + params.pSourceData = label; + params.ulSourceDataLen = wrap; + check_oaep_params_rejected(session, pub, priv, &mech); + } + + oaep_params_init(¶ms, &mech); + params.pSourceData = label; + params.ulSourceDataLen = sizeof(label); + rv = funcList->C_EncryptInit(session, &mech, pub); + CHECK_RV(rv, "OAEP: C_EncryptInit with label", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "OAEP: C_Encrypt with label", CKR_OK); + if (rv != CKR_OK) + return; + rv = funcList->C_DecryptInit(session, &mech, priv); + CHECK_RV(rv, "OAEP: C_DecryptInit with label", CKR_OK); + if (rv != CKR_OK) + return; + decLen = sizeof(dec); + rv = funcList->C_Decrypt(session, enc, encLen, dec, &decLen); + CHECK_RV(rv, "OAEP: C_Decrypt with label", CKR_OK); + CHECK_TRUE(decLen == sizeof(plain) && + XMEMCMP(dec, plain, sizeof(plain)) == 0, + "OAEP: labelled ciphertext round-trips"); +} +#endif + +#if !defined(NO_AES) && (defined(HAVE_AESCTR) || defined(HAVE_AESCTS)) +/* Single-part calls reject a data length the token cannot represent and end + * the operation instead of processing a shorter length. */ +static void check_single_part_len_range(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE key, + CK_MECHANISM* mech) +{ + CK_RV rv; + byte data[32]; + byte out[32]; + CK_ULONG outLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + XMEMSET(data, 0x3C, sizeof(data)); + + rv = funcList->C_EncryptInit(session, mech, key); + CHECK_RV(rv, "single-part length: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + outLen = wrap + sizeof(out); + rv = funcList->C_Encrypt(session, data, wrap + sizeof(data), out, &outLen); + CHECK_RV(rv, "single-part length: C_Encrypt rejects oversized data", + CKR_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, data, sizeof(data), out, &outLen); + CHECK_RV(rv, "single-part length: encrypt operation ended", + CKR_OPERATION_NOT_INITIALIZED); + + rv = funcList->C_DecryptInit(session, mech, key); + CHECK_RV(rv, "single-part length: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + outLen = wrap + sizeof(out); + rv = funcList->C_Decrypt(session, data, wrap + sizeof(data), out, &outLen); + CHECK_RV(rv, "single-part length: C_Decrypt rejects oversized data", + CKR_ENCRYPTED_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_Decrypt(session, data, sizeof(data), out, &outLen); + CHECK_RV(rv, "single-part length: decrypt operation ended", + CKR_OPERATION_NOT_INITIALIZED); +} + +static void test_single_part_len_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_MECHANISM mech; +#ifdef HAVE_AESCTR + CK_AES_CTR_PARAMS ctrParams; +#endif +#ifdef HAVE_AESCTS + byte iv[16]; +#endif + + if ((((CK_ULONG)1 << 16) << 16) == 0) + return; + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "single-part length: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + +#ifdef HAVE_AESCTR + XMEMSET(&ctrParams, 0, sizeof(ctrParams)); + ctrParams.ulCounterBits = 32; + XMEMSET(ctrParams.cb, 0x18, sizeof(ctrParams.cb)); + mech.mechanism = CKM_AES_CTR; + mech.pParameter = &ctrParams; + mech.ulParameterLen = sizeof(ctrParams); + printf("AES-CTR\n"); + check_single_part_len_range(session, key, &mech); +#endif +#ifdef HAVE_AESCTS + XMEMSET(iv, 0x29, sizeof(iv)); + mech.mechanism = CKM_AES_CTS; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + printf("AES-CTS\n"); + check_single_part_len_range(session, key, &mech); +#endif +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_KEYWRAP) +/* Key unwrap rejects a ciphertext length the token cannot represent and ends + * the operation instead of unwrapping a shorter length. */ +static void test_key_wrap_decrypt_len_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_MECHANISM mech; + byte plain[16]; + byte wrapped[24]; + byte out[32]; + CK_ULONG wrappedLen; + CK_ULONG outLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + static const CK_MECHANISM_TYPE mechs[] = { + CKM_AES_KEY_WRAP, CKM_AES_KEY_WRAP_PAD + }; + int i; + + if (wrap == 0) + return; + + XMEMSET(plain, 0x6B, sizeof(plain)); + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "key unwrap length: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + for (i = 0; i < (int)(sizeof(mechs) / sizeof(*mechs)); i++) { + mech.mechanism = mechs[i]; + mech.pParameter = NULL; + mech.ulParameterLen = 0; + printf("key unwrap length: mechanism 0x%lx\n", (unsigned long)mechs[i]); + + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "key unwrap length: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + continue; + wrappedLen = sizeof(wrapped); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), wrapped, + &wrappedLen); + CHECK_RV(rv, "key unwrap length: C_Encrypt", CKR_OK); + if (rv != CKR_OK) + continue; + + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "key unwrap length: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + continue; + outLen = sizeof(out); + rv = funcList->C_Decrypt(session, wrapped, wrap + wrappedLen, out, + &outLen); + CHECK_RV(rv, "key unwrap length: C_Decrypt rejects oversized length", + CKR_ENCRYPTED_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_Decrypt(session, wrapped, wrappedLen, out, &outLen); + CHECK_RV(rv, "key unwrap length: decrypt operation ended", + CKR_OPERATION_NOT_INITIALIZED); + } +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESGCM) +/* C_DecryptFinal accepts any output buffer length that is large enough, + * including lengths that do not fit in 32 bits. */ +static void test_gcm_decrypt_final_large_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_GCM_PARAMS params; + CK_MECHANISM mech; + byte iv[12]; + byte aad[16]; + byte plain[16]; + byte enc[32]; + byte dec[32]; + CK_ULONG encLen; + CK_ULONG partLen; + CK_ULONG lastLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + if (wrap == 0) + return; + + XMEMSET(iv, 0x2D, sizeof(iv)); + XMEMSET(aad, 0x4E, sizeof(aad)); + XMEMSET(plain, 0x6F, sizeof(plain)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "GCM final length: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + gcm_params_init(¶ms, &mech, iv, aad); + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "GCM final length: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + encLen = sizeof(enc); + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + CHECK_RV(rv, "GCM final length: C_Encrypt", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "GCM final length: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + partLen = sizeof(dec); + rv = funcList->C_DecryptUpdate(session, enc, encLen, dec, &partLen); + CHECK_RV(rv, "GCM final length: C_DecryptUpdate", CKR_OK); + if (rv != CKR_OK) + return; + lastLen = wrap; + rv = funcList->C_DecryptFinal(session, dec, &lastLen); + CHECK_RV(rv, "GCM final length: C_DecryptFinal with large buffer length", + CKR_OK); + CHECK_TRUE(rv == CKR_OK && lastLen == sizeof(plain) && + XMEMCMP(dec, plain, sizeof(plain)) == 0, + "GCM final length: plaintext recovered"); +} +#endif + +#if !defined(NO_AES) && (defined(HAVE_AES_CBC) || defined(HAVE_AESGCM) || \ + defined(HAVE_AESCTS)) +/* C_EncryptFinal accepts an output buffer length that does not fit in 32 bits + * and reports the bytes it wrote. */ +static void check_encrypt_final_large_len(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE key, + CK_MECHANISM* mech, + CK_ULONG dataLen, CK_ULONG encTotal) +{ + CK_RV rv; + byte plain[32]; + byte enc[64]; + CK_ULONG updLen; + CK_ULONG lastLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + XMEMSET(plain, 0x7A, sizeof(plain)); + + rv = funcList->C_EncryptInit(session, mech, key); + CHECK_RV(rv, "encrypt final length: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + updLen = sizeof(enc); + rv = funcList->C_EncryptUpdate(session, plain, dataLen, enc, &updLen); + CHECK_RV(rv, "encrypt final length: C_EncryptUpdate", CKR_OK); + if (rv != CKR_OK) + return; + lastLen = wrap; + rv = funcList->C_EncryptFinal(session, enc + updLen, &lastLen); + CHECK_RV(rv, "encrypt final length: C_EncryptFinal with large buffer " + "length", CKR_OK); + CHECK_TRUE(rv == CKR_OK && updLen + lastLen == encTotal, + "encrypt final length: reports bytes written"); + if (rv == CKR_BUFFER_TOO_SMALL) { + /* End the still-active operation so the next mechanism can start. */ + lastLen = sizeof(enc) - updLen; + (void)funcList->C_EncryptFinal(session, enc + updLen, &lastLen); + } +} + +static void test_encrypt_final_large_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_MECHANISM mech; +#if defined(HAVE_AES_CBC) || defined(HAVE_AESCTS) + byte iv[16]; +#endif +#ifdef HAVE_AESGCM + CK_GCM_PARAMS params; + byte gcmIv[12]; + byte aad[16]; +#endif + + if ((((CK_ULONG)1 << 16) << 16) == 0) + return; + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "encrypt final length: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + +#ifdef HAVE_AES_CBC + XMEMSET(iv, 0x13, sizeof(iv)); + mech.mechanism = CKM_AES_CBC_PAD; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + printf("AES-CBC-PAD\n"); + check_encrypt_final_large_len(session, key, &mech, 20, 32); +#endif +#ifdef HAVE_AESGCM + XMEMSET(gcmIv, 0x24, sizeof(gcmIv)); + XMEMSET(aad, 0x35, sizeof(aad)); + gcm_params_init(¶ms, &mech, gcmIv, aad); + printf("AES-GCM\n"); + check_encrypt_final_large_len(session, key, &mech, 16, 32); +#endif +#ifdef HAVE_AESCTS + XMEMSET(iv, 0x46, sizeof(iv)); + mech.mechanism = CKM_AES_CTS; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + printf("AES-CTS\n"); + check_encrypt_final_large_len(session, key, &mech, 20, 20); +#endif +} +#endif + +#if !defined(NO_AES) && (defined(HAVE_AESCTR) || defined(HAVE_AESGCM) || \ + defined(HAVE_AESCTS)) +/* Multi-part encryption rejects a part length the token cannot represent and + * ends the operation instead of encrypting a shorter part. */ +static void check_encrypt_update_len_range(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE key, + CK_MECHANISM* mech) +{ + CK_RV rv; + byte data[32]; + byte out[64]; + CK_ULONG outLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + XMEMSET(data, 0x5D, sizeof(data)); + + rv = funcList->C_EncryptInit(session, mech, key); + CHECK_RV(rv, "encrypt update length: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + outLen = wrap + sizeof(out); + rv = funcList->C_EncryptUpdate(session, data, wrap + sizeof(data), out, + &outLen); + CHECK_RV(rv, "encrypt update length: C_EncryptUpdate rejects oversized " + "part", CKR_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, data, sizeof(data), out, &outLen); + CHECK_RV(rv, "encrypt update length: operation ended", + CKR_OPERATION_NOT_INITIALIZED); + if (rv == CKR_OK) { + /* End the still-active operation so the next mechanism can start. */ + outLen = sizeof(out); + (void)funcList->C_EncryptFinal(session, out, &outLen); + } +} + +static void test_encrypt_update_len_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_MECHANISM mech; +#ifdef HAVE_AESCTR + CK_AES_CTR_PARAMS ctrParams; +#endif +#ifdef HAVE_AESGCM + CK_GCM_PARAMS gcmParams; + byte gcmIv[12]; + byte aad[16]; +#endif +#ifdef HAVE_AESCTS + byte iv[16]; +#endif + + if ((((CK_ULONG)1 << 16) << 16) == 0) + return; + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "encrypt update length: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + +#ifdef HAVE_AESCTR + XMEMSET(&ctrParams, 0, sizeof(ctrParams)); + ctrParams.ulCounterBits = 32; + XMEMSET(ctrParams.cb, 0x57, sizeof(ctrParams.cb)); + mech.mechanism = CKM_AES_CTR; + mech.pParameter = &ctrParams; + mech.ulParameterLen = sizeof(ctrParams); + printf("AES-CTR\n"); + check_encrypt_update_len_range(session, key, &mech); +#endif +#ifdef HAVE_AESGCM + XMEMSET(gcmIv, 0x68, sizeof(gcmIv)); + XMEMSET(aad, 0x79, sizeof(aad)); + gcm_params_init(&gcmParams, &mech, gcmIv, aad); + printf("AES-GCM\n"); + check_encrypt_update_len_range(session, key, &mech); +#endif +#ifdef HAVE_AESCTS + XMEMSET(iv, 0x8A, sizeof(iv)); + mech.mechanism = CKM_AES_CTS; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + printf("AES-CTS\n"); + check_encrypt_update_len_range(session, key, &mech); +#endif +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESCTR) +/* Multi-part AES-CTR decryption rejects a part length the token cannot + * represent and accepts an output buffer length that does not fit in 32 bits. */ +static void test_ctr_decrypt_update_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_AES_CTR_PARAMS params; + CK_MECHANISM mech; + byte data[32]; + byte out[32]; + CK_ULONG partLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + if (wrap == 0) + return; + + XMEMSET(¶ms, 0, sizeof(params)); + params.ulCounterBits = 32; + XMEMSET(params.cb, 0x9B, sizeof(params.cb)); + XMEMSET(data, 0xAC, sizeof(data)); + mech.mechanism = CKM_AES_CTR; + mech.pParameter = ¶ms; + mech.ulParameterLen = sizeof(params); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "CTR decrypt update: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "CTR decrypt update: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + partLen = wrap + sizeof(out); + rv = funcList->C_DecryptUpdate(session, data, wrap + sizeof(data), out, + &partLen); + CHECK_RV(rv, "CTR decrypt update: rejects oversized part", + CKR_ENCRYPTED_DATA_LEN_RANGE); + partLen = sizeof(out); + rv = funcList->C_DecryptUpdate(session, data, sizeof(data), out, &partLen); + CHECK_RV(rv, "CTR decrypt update: operation ended", + CKR_OPERATION_NOT_INITIALIZED); + if (rv == CKR_OK) { + /* End the still-active operation so the next check can start. */ + partLen = sizeof(out); + (void)funcList->C_DecryptFinal(session, out, &partLen); + } + + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "CTR decrypt update: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + partLen = wrap + 8; + rv = funcList->C_DecryptUpdate(session, data, 16, out, &partLen); + CHECK_RV(rv, "CTR decrypt update: large output buffer length", CKR_OK); + CHECK_TRUE(rv == CKR_OK && partLen == 16, + "CTR decrypt update: reports bytes written"); + if (rv == CKR_OK) { + partLen = sizeof(out); + rv = funcList->C_DecryptFinal(session, out, &partLen); + CHECK_RV(rv, "CTR decrypt update: C_DecryptFinal", CKR_OK); + } +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESGCM) +/* Multi-part AES-GCM decryption rejects a part that does not fit in 32 bits or + * would grow the buffered ciphertext past INT_MAX, and ends the operation. */ +static void check_gcm_decrypt_update_rejected(CK_SESSION_HANDLE session, + byte* data, CK_ULONG dataLen, + CK_ULONG badLen) +{ + CK_RV rv; + byte out[64]; + CK_ULONG partLen; + + partLen = sizeof(out); + rv = funcList->C_DecryptUpdate(session, data, badLen, out, &partLen); + CHECK_RV(rv, "GCM decrypt update: rejects part length", + CKR_ENCRYPTED_DATA_LEN_RANGE); + partLen = sizeof(out); + rv = funcList->C_DecryptUpdate(session, data, dataLen, out, &partLen); + CHECK_RV(rv, "GCM decrypt update: operation ended", + CKR_OPERATION_NOT_INITIALIZED); + if (rv == CKR_OK) { + /* End the still-active operation so the next check can start. */ + partLen = sizeof(out); + (void)funcList->C_DecryptFinal(session, out, &partLen); + } +} + +static void test_gcm_decrypt_update_len_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_GCM_PARAMS params; + CK_MECHANISM mech; + byte iv[12]; + byte aad[16]; + byte data[32]; + byte out[32]; + CK_ULONG partLen; + CK_ULONG wrap = ((CK_ULONG)1 << 16) << 16; + + XMEMSET(iv, 0xBD, sizeof(iv)); + XMEMSET(aad, 0xCE, sizeof(aad)); + XMEMSET(data, 0xDF, sizeof(data)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "GCM decrypt update: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + gcm_params_init(¶ms, &mech, iv, aad); + + if (wrap != 0) { + printf("GCM decrypt update: part length 2^32 + 32\n"); + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "GCM decrypt update: C_DecryptInit", CKR_OK); + if (rv == CKR_OK) { + check_gcm_decrypt_update_rejected(session, data, sizeof(data), + wrap + sizeof(data)); + } + } + + printf("GCM decrypt update: buffered length past INT_MAX\n"); + rv = funcList->C_DecryptInit(session, &mech, key); + CHECK_RV(rv, "GCM decrypt update: C_DecryptInit", CKR_OK); + if (rv != CKR_OK) + return; + partLen = sizeof(out); + rv = funcList->C_DecryptUpdate(session, data, sizeof(data), out, &partLen); + CHECK_RV(rv, "GCM decrypt update: first part", CKR_OK); + if (rv != CKR_OK) + return; + check_gcm_decrypt_update_rejected(session, data, sizeof(data), + (CK_ULONG)INT_MAX - sizeof(data) + 1); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AESGCM) +static void check_gcm_encrypt_update_rejected(CK_SESSION_HANDLE session, + byte* data, CK_ULONG dataLen, + CK_ULONG badLen) +{ + CK_RV rv; + byte out[64]; + CK_ULONG outLen; + + outLen = badLen; + rv = funcList->C_EncryptUpdate(session, data, badLen, out, &outLen); + CHECK_RV(rv, "GCM encrypt update: rejects part length", + CKR_DATA_LEN_RANGE); + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, data, dataLen, out, &outLen); + CHECK_RV(rv, "GCM encrypt update: operation ended", + CKR_OPERATION_NOT_INITIALIZED); + if (rv == CKR_OK) { + /* End the still-active operation so the next check can start. */ + outLen = sizeof(out); + (void)funcList->C_EncryptFinal(session, out, &outLen); + } +} + +/* Multi-part AES-GCM encryption rejects a part that would take the operation + * past INT_MAX bytes, and ends the operation. */ +static void test_gcm_encrypt_update_total_len(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_GCM_PARAMS params; + CK_MECHANISM mech; + byte iv[12]; + byte aad[16]; + byte data[32]; +#ifndef WOLFSSL_AESGCM_STREAM + byte out[32]; + CK_ULONG outLen; +#endif + + XMEMSET(iv, 0xE1, sizeof(iv)); + XMEMSET(aad, 0xF2, sizeof(aad)); + XMEMSET(data, 0x03, sizeof(data)); + + rv = create_aes_key(session, aes_128_key, sizeof(aes_128_key), &key); + CHECK_RV(rv, "GCM encrypt update: create AES key", CKR_OK); + if (rv != CKR_OK) + return; + gcm_params_init(¶ms, &mech, iv, aad); + + printf("GCM encrypt update: part length 0x80000000\n"); + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "GCM encrypt update: C_EncryptInit", CKR_OK); + if (rv == CKR_OK) { + check_gcm_encrypt_update_rejected(session, data, sizeof(data), + (CK_ULONG)0x80000000UL); + } + +#ifndef WOLFSSL_AESGCM_STREAM + printf("GCM encrypt update: buffered length past INT_MAX\n"); + rv = funcList->C_EncryptInit(session, &mech, key); + CHECK_RV(rv, "GCM encrypt update: C_EncryptInit", CKR_OK); + if (rv != CKR_OK) + return; + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, data, sizeof(data), out, &outLen); + CHECK_RV(rv, "GCM encrypt update: first part", CKR_OK); + if (rv != CKR_OK) + return; + check_gcm_encrypt_update_rejected(session, data, sizeof(data), + (CK_ULONG)INT_MAX - sizeof(data) + 1); +#endif +} +#endif + +static int run_test(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = 0; + CK_SESSION_INFO info; + CK_SLOT_ID slot = 0; + + (void)run_in_session; + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + + rv = pkcs11_open_session(&session); + CHECK_RV(rv, "open session", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_GetSessionInfo(session, &info); + CHECK_RV(rv, "C_GetSessionInfo", CKR_OK); + } + if (rv == CKR_OK) { + slot = info.slotID; +#if !defined(NO_AES) && defined(HAVE_AES_CBC) + run_in_session(slot, test_cbc_pad_encrypt_final_len); +#endif +#if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) + run_in_session(slot, test_oaep_source_ptr_len); +#endif +#if !defined(NO_AES) && defined(HAVE_AESCCM) + run_in_session(slot, test_ccm_nonce_len); +#endif +#if !defined(NO_AES) && defined(HAVE_AESCTR) + run_in_session(slot, test_ctr_decrypt_size_query); +#endif +#if !defined(NO_AES) && (defined(HAVE_AES_CBC) || defined(HAVE_AESECB)) + run_in_session(slot, test_block_mode_alignment); +#endif +#if !defined(NO_AES) && defined(HAVE_AESECB) + run_in_session(slot, test_aes_check_value_key_sizes); +#endif +#if !defined(NO_AES) && defined(HAVE_AESGCM) + run_in_session(slot, test_gcm_param_ranges); +#endif +#if !defined(NO_AES) && defined(HAVE_AESCCM) + run_in_session(slot, test_ccm_param_ranges); +#endif +#if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) + run_in_session(slot, test_oaep_source_len_range); +#endif +#if !defined(NO_AES) && (defined(HAVE_AESCTR) || defined(HAVE_AESCTS)) + run_in_session(slot, test_single_part_len_range); +#endif +#if !defined(NO_AES) && defined(HAVE_AES_KEYWRAP) + run_in_session(slot, test_key_wrap_decrypt_len_range); +#endif +#if !defined(NO_AES) && defined(HAVE_AESGCM) + run_in_session(slot, test_gcm_decrypt_final_large_len); +#endif +#if !defined(NO_AES) && (defined(HAVE_AES_CBC) || defined(HAVE_AESGCM) || \ + defined(HAVE_AESCTS)) + run_in_session(slot, test_encrypt_final_large_len); +#endif +#if !defined(NO_AES) && (defined(HAVE_AESCTR) || defined(HAVE_AESGCM) || \ + defined(HAVE_AESCTS)) + run_in_session(slot, test_encrypt_update_len_range); +#endif +#if !defined(NO_AES) && defined(HAVE_AESCTR) + run_in_session(slot, test_ctr_decrypt_update_len); +#endif +#if !defined(NO_AES) && defined(HAVE_AESGCM) + run_in_session(slot, test_gcm_decrypt_update_len_range); + run_in_session(slot, test_gcm_encrypt_update_total_len); +#endif + } + + if (session != 0) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + + printf("=== wolfPKCS11 encrypt/decrypt parameter and length test ===\n"); + run_test(); + return pkcs11_test_summary(); +} diff --git a/tests/include.am b/tests/include.am index 584136e7..3c07f3f9 100644 --- a/tests/include.am +++ b/tests/include.am @@ -206,6 +206,16 @@ noinst_PROGRAMS += tests/derive_inherit_protection_test tests_derive_inherit_protection_test_SOURCES = tests/derive_inherit_protection_test.c tests_derive_inherit_protection_test_LDADD = +check_PROGRAMS += tests/ec_derive_test +noinst_PROGRAMS += tests/ec_derive_test +tests_ec_derive_test_SOURCES = tests/ec_derive_test.c +tests_ec_derive_test_LDADD = + +check_PROGRAMS += tests/kdf_tls_derive_test +noinst_PROGRAMS += tests/kdf_tls_derive_test +tests_kdf_tls_derive_test_SOURCES = tests/kdf_tls_derive_test.c +tests_kdf_tls_derive_test_LDADD = + check_PROGRAMS += tests/secret_extractable_default_test noinst_PROGRAMS += tests/secret_extractable_default_test tests_secret_extractable_default_test_SOURCES = tests/secret_extractable_default_test.c @@ -264,6 +274,10 @@ check_PROGRAMS += tests/token_reset_fail_test noinst_PROGRAMS += tests/token_reset_fail_test tests_token_reset_fail_test_SOURCES = tests/token_reset_fail_test.c tests_token_reset_fail_test_LDADD = +check_PROGRAMS += tests/token_store_test +noinst_PROGRAMS += tests/token_store_test +tests_token_store_test_SOURCES = tests/token_store_test.c +tests_token_store_test_LDADD = check_PROGRAMS += tests/nss_email_store_test noinst_PROGRAMS += tests/nss_email_store_test @@ -320,6 +334,47 @@ noinst_PROGRAMS += tests/login_pin_state_test tests_login_pin_state_test_SOURCES = tests/login_pin_state_test.c tests_login_pin_state_test_LDADD = +check_PROGRAMS += tests/object_store_decode_test +noinst_PROGRAMS += tests/object_store_decode_test +tests_object_store_decode_test_SOURCES = tests/object_store_decode_test.c +tests_object_store_decode_test_LDADD = + +check_PROGRAMS += tests/object_policy_integrity_test +noinst_PROGRAMS += tests/object_policy_integrity_test +tests_object_policy_integrity_test_SOURCES = \ + tests/object_policy_integrity_test.c +tests_object_policy_integrity_test_LDADD = + +check_PROGRAMS += tests/session_op_lifecycle_test +noinst_PROGRAMS += tests/session_op_lifecycle_test +tests_session_op_lifecycle_test_SOURCES = tests/session_op_lifecycle_test.c +tests_session_op_lifecycle_test_LDADD = + +check_PROGRAMS += tests/info_mech_table_test +noinst_PROGRAMS += tests/info_mech_table_test +tests_info_mech_table_test_SOURCES = tests/info_mech_table_test.c +tests_info_mech_table_test_LDADD = + +check_PROGRAMS += tests/attribute_validation_test +noinst_PROGRAMS += tests/attribute_validation_test +tests_attribute_validation_test_SOURCES = tests/attribute_validation_test.c +tests_attribute_validation_test_LDADD = + +check_PROGRAMS += tests/key_material_update_test +noinst_PROGRAMS += tests/key_material_update_test +tests_key_material_update_test_SOURCES = tests/key_material_update_test.c +tests_key_material_update_test_LDADD = + +check_PROGRAMS += tests/object_create_getattr_test +noinst_PROGRAMS += tests/object_create_getattr_test +tests_object_create_getattr_test_SOURCES = tests/object_create_getattr_test.c +tests_object_create_getattr_test_LDADD = + +check_PROGRAMS += tests/encrypt_decrypt_params_test +noinst_PROGRAMS += tests/encrypt_decrypt_params_test +tests_encrypt_decrypt_params_test_SOURCES = tests/encrypt_decrypt_params_test.c +tests_encrypt_decrypt_params_test_LDADD = + if BUILD_STATIC tests_pkcs11test_LDADD += src/libwolfpkcs11.la tests_pkcs11mtt_LDADD += src/libwolfpkcs11.la @@ -361,6 +416,8 @@ tests_private_object_empty_pin_test_LDADD += src/libwolfpkcs11.la tests_key_sensitive_default_test_LDADD += src/libwolfpkcs11.la tests_derive_disabled_key_test_LDADD += src/libwolfpkcs11.la tests_derive_inherit_protection_test_LDADD += src/libwolfpkcs11.la +tests_ec_derive_test_LDADD += src/libwolfpkcs11.la +tests_kdf_tls_derive_test_LDADD += src/libwolfpkcs11.la tests_secret_extractable_default_test_LDADD += src/libwolfpkcs11.la tests_rsa_private_usage_default_test_LDADD += src/libwolfpkcs11.la tests_copyobject_token_test_LDADD += src/libwolfpkcs11.la @@ -374,6 +431,7 @@ tests_hbs_persistence_test_LDADD += src/libwolfpkcs11.la tests_hkdf_persistence_test_LDADD += src/libwolfpkcs11.la tests_unique_id_test_LDADD += src/libwolfpkcs11.la tests_token_reset_fail_test_LDADD += src/libwolfpkcs11.la +tests_token_store_test_LDADD += src/libwolfpkcs11.la tests_nss_email_store_test_LDADD += src/libwolfpkcs11.la tests_generate_key_class_test_LDADD += src/libwolfpkcs11.la tests_generate_keypair_class_test_LDADD += src/libwolfpkcs11.la @@ -385,6 +443,14 @@ tests_set_pin_rekey_test_LDADD += src/libwolfpkcs11.la tests_nss_configdir_prefix_test_LDADD += src/libwolfpkcs11.la tests_logout_retire_test_LDADD += src/libwolfpkcs11.la tests_login_pin_state_test_LDADD += src/libwolfpkcs11.la +tests_object_store_decode_test_LDADD += src/libwolfpkcs11.la +tests_object_policy_integrity_test_LDADD += src/libwolfpkcs11.la +tests_session_op_lifecycle_test_LDADD += src/libwolfpkcs11.la +tests_info_mech_table_test_LDADD += src/libwolfpkcs11.la +tests_attribute_validation_test_LDADD += src/libwolfpkcs11.la +tests_key_material_update_test_LDADD += src/libwolfpkcs11.la +tests_object_create_getattr_test_LDADD += src/libwolfpkcs11.la +tests_encrypt_decrypt_params_test_LDADD += src/libwolfpkcs11.la else tests_object_id_uniqueness_test_LDADD += src/libwolfpkcs11.la tests_empty_pin_store_test_LDADD += src/libwolfpkcs11.la @@ -418,6 +484,8 @@ tests_private_object_empty_pin_test_LDADD += src/libwolfpkcs11.la tests_key_sensitive_default_test_LDADD += src/libwolfpkcs11.la tests_derive_disabled_key_test_LDADD += src/libwolfpkcs11.la tests_derive_inherit_protection_test_LDADD += src/libwolfpkcs11.la +tests_ec_derive_test_LDADD += src/libwolfpkcs11.la +tests_kdf_tls_derive_test_LDADD += src/libwolfpkcs11.la tests_secret_extractable_default_test_LDADD += src/libwolfpkcs11.la tests_rsa_private_usage_default_test_LDADD += src/libwolfpkcs11.la tests_copyobject_token_test_LDADD += src/libwolfpkcs11.la @@ -431,6 +499,7 @@ tests_hbs_persistence_test_LDADD += src/libwolfpkcs11.la tests_hkdf_persistence_test_LDADD += src/libwolfpkcs11.la tests_unique_id_test_LDADD += src/libwolfpkcs11.la tests_token_reset_fail_test_LDADD += src/libwolfpkcs11.la +tests_token_store_test_LDADD += src/libwolfpkcs11.la tests_nss_email_store_test_LDADD += src/libwolfpkcs11.la tests_generate_key_class_test_LDADD += src/libwolfpkcs11.la tests_generate_keypair_class_test_LDADD += src/libwolfpkcs11.la @@ -442,6 +511,14 @@ tests_set_pin_rekey_test_LDADD += src/libwolfpkcs11.la tests_nss_configdir_prefix_test_LDADD += src/libwolfpkcs11.la tests_logout_retire_test_LDADD += src/libwolfpkcs11.la tests_login_pin_state_test_LDADD += src/libwolfpkcs11.la +tests_object_store_decode_test_LDADD += src/libwolfpkcs11.la +tests_object_policy_integrity_test_LDADD += src/libwolfpkcs11.la +tests_session_op_lifecycle_test_LDADD += src/libwolfpkcs11.la +tests_info_mech_table_test_LDADD += src/libwolfpkcs11.la +tests_attribute_validation_test_LDADD += src/libwolfpkcs11.la +tests_key_material_update_test_LDADD += src/libwolfpkcs11.la +tests_object_create_getattr_test_LDADD += src/libwolfpkcs11.la +tests_encrypt_decrypt_params_test_LDADD += src/libwolfpkcs11.la endif EXTRA_DIST += tests/unit.h \ diff --git a/tests/info_mech_table_test.c b/tests/info_mech_table_test.c new file mode 100644 index 00000000..9a44cda3 --- /dev/null +++ b/tests/info_mech_table_test.c @@ -0,0 +1,1005 @@ +/* info_mech_table_test.c + * + * Copyright (C) 2006-2025 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Library, slot, token and mechanism information must match the PKCS#11 + * encoding rules and what the token actually implements. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#define TEST_DIR "./store/info_mech_table_test" + +#if defined(USE_WOLFSSL_MEMORY) && !defined(NO_WOLFSSL_MEMORY) && \ + !defined(WOLFSSL_STATIC_MEMORY) && !defined(WOLFSSL_DEBUG_MEMORY) + #define ALLOC_HOOKS +#endif + +#ifdef ALLOC_HOOKS +static int failAllocs = 0; + +static void* hook_malloc(size_t sz) +{ + if (failAllocs) + return NULL; + return malloc(sz); +} + +static void hook_free(void* ptr) +{ + free(ptr); +} + +static void* hook_realloc(void* ptr, size_t sz) +{ + if (failAllocs) + return NULL; + return realloc(ptr, sz); +} +#endif + +/* Fixed-length character fields are blank padded and never contain NUL. */ +static int blank_padded(const CK_UTF8CHAR* field, size_t len) +{ + size_t i; + + for (i = 0; i < len; i++) { + if (field[i] == '\0') + return 0; + } + return 1; +} + +static void check_info(CK_FUNCTION_LIST* list, const char* name) +{ + CK_RV rv; + CK_INFO info; + char msg[96]; + + XMEMSET(&info, 0, sizeof(info)); + rv = list->C_GetInfo(&info); + snprintf(msg, sizeof(msg), "C_GetInfo(%s)", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv != CKR_OK) + return; + snprintf(msg, sizeof(msg), "%s CK_INFO.manufacturerID blank padded", name); + CHECK_TRUE(blank_padded(info.manufacturerID, sizeof(info.manufacturerID)), + msg); + snprintf(msg, sizeof(msg), "%s CK_INFO.libraryDescription blank padded", + name); + CHECK_TRUE(blank_padded(info.libraryDescription, + sizeof(info.libraryDescription)), msg); +} + +static void test_info_blank_padding(CK_SLOT_ID slot) +{ + CK_RV rv; + CK_SLOT_INFO slotInfo; + CK_TOKEN_INFO tokenInfo; +#ifdef WOLFPKCS11_PKCS11_V3_0 + CK_INTERFACE interfaces[8]; + CK_ULONG count = sizeof(interfaces) / sizeof(interfaces[0]); + CK_ULONG i; + char name[32]; +#ifndef HAVE_PKCS11_STATIC + CK_C_GetInterfaceList getList; +#endif +#endif + + check_info(funcList, "v2"); + +#ifdef WOLFPKCS11_PKCS11_V3_0 +#ifndef HAVE_PKCS11_STATIC + getList = (CK_C_GetInterfaceList)dlsym(dlib, "C_GetInterfaceList"); + CHECK_TRUE(getList != NULL, "C_GetInterfaceList symbol"); + rv = (getList != NULL) ? getList(interfaces, &count) : CKR_GENERAL_ERROR; +#else + rv = C_GetInterfaceList(interfaces, &count); +#endif + CHECK_RV(rv, "C_GetInterfaceList", CKR_OK); + for (i = 0; rv == CKR_OK && i < count; i++) { + snprintf(name, sizeof(name), "interface %lu", (unsigned long)i); + check_info((CK_FUNCTION_LIST*)interfaces[i].pFunctionList, name); + } +#endif + + XMEMSET(&slotInfo, 0, sizeof(slotInfo)); + rv = funcList->C_GetSlotInfo(slot, &slotInfo); + CHECK_RV(rv, "C_GetSlotInfo", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(blank_padded(slotInfo.slotDescription, + sizeof(slotInfo.slotDescription)), + "CK_SLOT_INFO.slotDescription blank padded"); + CHECK_TRUE(blank_padded(slotInfo.manufacturerID, + sizeof(slotInfo.manufacturerID)), + "CK_SLOT_INFO.manufacturerID blank padded"); + } + + XMEMSET(&tokenInfo, 0, sizeof(tokenInfo)); + rv = funcList->C_GetTokenInfo(slot, &tokenInfo); + CHECK_RV(rv, "C_GetTokenInfo", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(blank_padded(tokenInfo.label, sizeof(tokenInfo.label)), + "CK_TOKEN_INFO.label blank padded"); + CHECK_TRUE(blank_padded(tokenInfo.manufacturerID, + sizeof(tokenInfo.manufacturerID)), + "CK_TOKEN_INFO.manufacturerID blank padded"); + CHECK_TRUE(blank_padded(tokenInfo.model, sizeof(tokenInfo.model)), + "CK_TOKEN_INFO.model blank padded"); + CHECK_TRUE(blank_padded(tokenInfo.serialNumber, + sizeof(tokenInfo.serialNumber)), + "CK_TOKEN_INFO.serialNumber has no NUL bytes"); + } +} + +static int two_digits(const CK_CHAR* p, int* val) +{ + if (p[0] < '0' || p[0] > '9' || p[1] < '0' || p[1] > '9') + return 0; + *val = (p[0] - '0') * 10 + (p[1] - '0'); + return 1; +} + +/* A token that claims a clock must report a valid UTC time. */ +static void test_token_clock(CK_SLOT_ID slot) +{ + CK_RV rv; + CK_TOKEN_INFO tokenInfo; + int century = 0, year = 0, month = 0, day = 0; + int hour = 0, minute = 0, second = 0; + int valid; + + XMEMSET(&tokenInfo, 0, sizeof(tokenInfo)); + rv = funcList->C_GetTokenInfo(slot, &tokenInfo); + CHECK_RV(rv, "C_GetTokenInfo(clock)", CKR_OK); + if (rv != CKR_OK) + return; + + if ((tokenInfo.flags & CKF_CLOCK_ON_TOKEN) == 0) { + CHECK_TRUE(XMEMCMP(tokenInfo.utcTime, " ", + sizeof(tokenInfo.utcTime)) == 0, + "utcTime is blank without a clock"); + return; + } + + valid = two_digits(&tokenInfo.utcTime[0], ¢ury) && + two_digits(&tokenInfo.utcTime[2], &year) && + two_digits(&tokenInfo.utcTime[4], &month) && + two_digits(&tokenInfo.utcTime[6], &day) && + two_digits(&tokenInfo.utcTime[8], &hour) && + two_digits(&tokenInfo.utcTime[10], &minute) && + two_digits(&tokenInfo.utcTime[12], &second); + CHECK_TRUE(valid && century * 100 + year >= 1970 && + month >= 1 && month <= 12 && day >= 1 && day <= 31 && + hour <= 23 && minute <= 59 && second <= 60, + "CKF_CLOCK_ON_TOKEN reports a valid utcTime"); +} + +/* C_WrapKey rejects every key before the mechanism check when it cannot + * serialize keys, as in a build without a store. */ +#if defined(WOLFPKCS11_NO_STORE) && !defined(WOLFSSL_STM32U5_DHUK) + #define WRAP_UNSUPPORTED_RV CKR_KEY_NOT_WRAPPABLE +#else + #define WRAP_UNSUPPORTED_RV CKR_MECHANISM_INVALID +#endif + +static CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; +static CK_KEY_TYPE genericType = CKK_GENERIC_SECRET; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; + +static CK_RV generate_generic(CK_SESSION_HANDLE session, CK_ULONG len) +{ + CK_RV rv; + CK_MECHANISM mech = { CKM_GENERIC_SECRET_KEY_GEN, NULL, 0 }; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE_LEN, &len, sizeof(len) }, + }; + + rv = funcList->C_GenerateKey(session, &mech, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &key); + if (key != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, key); + return rv; +} + +/* Generic secret key sizes are reported in bits and match what generates. */ +static void test_generic_secret_size(CK_SLOT_ID slot, + CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM_INFO info; + + rv = funcList->C_GetMechanismInfo(slot, CKM_GENERIC_SECRET_KEY_GEN, &info); + CHECK_RV(rv, "C_GetMechanismInfo(CKM_GENERIC_SECRET_KEY_GEN)", CKR_OK); + if (rv != CKR_OK) + return; + + CHECK_TRUE(info.ulMinKeySize >= 8 && info.ulMinKeySize % 8 == 0 && + info.ulMaxKeySize % 8 == 0 && + info.ulMaxKeySize >= info.ulMinKeySize, + "generic secret key sizes are whole bytes in bits"); + rv = generate_generic(session, info.ulMinKeySize / 8); + CHECK_RV(rv, "generate generic secret of minimum size", CKR_OK); + rv = generate_generic(session, info.ulMaxKeySize / 8); + CHECK_RV(rv, "generate generic secret of maximum size", CKR_OK); + rv = generate_generic(session, info.ulMaxKeySize / 8 + 1); + CHECK_TRUE(rv != CKR_OK, + "generic secret above maximum size is rejected"); +} + +/* Every listed key generation mechanism must be known to C_GenerateKey. */ +static void test_listed_keygen_supported(CK_SLOT_ID slot, + CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM_TYPE mechs[256]; + CK_ULONG count = sizeof(mechs) / sizeof(mechs[0]); + CK_ULONG i; + CK_MECHANISM_INFO info; + CK_MECHANISM mech; + CK_OBJECT_HANDLE key; + char msg[96]; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + + rv = funcList->C_GetMechanismList(slot, mechs, &count); + CHECK_RV(rv, "C_GetMechanismList", CKR_OK); + if (rv != CKR_OK) + return; + + for (i = 0; i < count; i++) { + rv = funcList->C_GetMechanismInfo(slot, mechs[i], &info); + snprintf(msg, sizeof(msg), "C_GetMechanismInfo(listed 0x%lx)", + (unsigned long)mechs[i]); + CHECK_RV(rv, msg, CKR_OK); + if (rv != CKR_OK || (info.flags & CKF_GENERATE) == 0) + continue; + + mech.mechanism = mechs[i]; + mech.pParameter = NULL; + mech.ulParameterLen = 0; + key = CK_INVALID_HANDLE; + rv = funcList->C_GenerateKey(session, &mech, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &key); + if (key != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, key); + snprintf(msg, sizeof(msg), "C_GenerateKey accepts listed 0x%lx", + (unsigned long)mechs[i]); + CHECK_TRUE(rv != CKR_MECHANISM_INVALID, msg); + } +} + +#ifdef ALLOC_HOOKS +/* A session is reported open only when a usable handle was returned. */ +static void test_open_session_alloc_failure(CK_SLOT_ID slot) +{ + CK_RV rv; + CK_SESSION_HANDLE handles[8]; + CK_SESSION_INFO sessInfo; + int opened = 0; + int failed = 0; + int i; + + for (i = 0; i < (int)(sizeof(handles) / sizeof(handles[0])); i++) { + handles[i] = CK_INVALID_HANDLE; + failAllocs = 1; + rv = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &handles[i]); + failAllocs = 0; + if (rv != CKR_OK) { + CHECK_RV(rv, "C_OpenSession without memory", CKR_HOST_MEMORY); + failed = 1; + break; + } + opened++; + rv = funcList->C_GetSessionInfo(handles[i], &sessInfo); + CHECK_RV(rv, "C_OpenSession returns a usable handle", CKR_OK); + if (rv != CKR_OK) + break; + } + CHECK_TRUE(failed, "C_OpenSession reports allocation failure"); + + for (i = 0; i < opened; i++) { + if (handles[i] != CK_INVALID_HANDLE) + funcList->C_CloseSession(handles[i]); + } +} +#endif + +typedef struct WrapCase { + CK_MECHANISM_TYPE mech; + int useIv; + int rsa; +} WrapCase; + +/* Wrap and unwrap must succeed exactly for mechanisms that advertise them. */ +static void test_wrap_flags(CK_SLOT_ID slot, CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM_INFO info; + CK_MECHANISM mech; + CK_OBJECT_HANDLE secret = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE wrapKey; + CK_OBJECT_HANDLE unwrapKey; + CK_OBJECT_HANDLE unwrapped; + CK_BYTE iv[16]; + CK_BYTE secretValue[32]; + CK_BYTE aesValue[16]; + CK_BYTE wrapped[512]; + CK_ULONG wrappedLen; + CK_BYTE* unwrapData; + CK_ULONG unwrapLen; + char msg[96]; + size_t i; +#ifndef NO_RSA + CK_OBJECT_HANDLE rsaPub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE rsaPriv = CK_INVALID_HANDLE; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE rsaType = CKK_RSA; + CK_ATTRIBUTE rsaPubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_WRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE rsaPrivTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_UNWRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; +#endif +#ifndef NO_AES + CK_KEY_TYPE aesType = CKK_AES; + CK_ATTRIBUTE aesTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_WRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_UNWRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesValue, sizeof(aesValue) }, + }; +#endif + CK_ATTRIBUTE secretTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, secretValue, sizeof(secretValue) }, + }; + CK_ULONG secretLen = sizeof(secretValue); + CK_ATTRIBUTE unwrapTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &secretLen, sizeof(secretLen) }, + }; + static const WrapCase cases[] = { + { CKM_AES_KEY_WRAP, 0, 0 }, + { CKM_AES_KEY_WRAP_PAD, 0, 0 }, + { CKM_AES_CBC, 1, 0 }, + { CKM_AES_CBC_PAD, 1, 0 }, + { CKM_AES_ECB, 0, 0 }, + { CKM_RSA_PKCS, 0, 1 }, + { CKM_RSA_X_509, 0, 1 }, + }; + + XMEMSET(iv, 0x11, sizeof(iv)); + XMEMSET(secretValue, 0x22, sizeof(secretValue)); + XMEMSET(aesValue, 0x33, sizeof(aesValue)); + + rv = funcList->C_CreateObject(session, secretTmpl, + sizeof(secretTmpl) / sizeof(*secretTmpl), + &secret); + CHECK_RV(rv, "C_CreateObject(secret to wrap)", CKR_OK); + if (rv != CKR_OK) + return; +#ifndef NO_AES + rv = funcList->C_CreateObject(session, aesTmpl, + sizeof(aesTmpl) / sizeof(*aesTmpl), &aesKey); + CHECK_RV(rv, "C_CreateObject(AES wrapping key)", CKR_OK); +#endif +#ifndef NO_RSA + rv = funcList->C_CreateObject(session, rsaPubTmpl, + sizeof(rsaPubTmpl) / sizeof(*rsaPubTmpl), + &rsaPub); + CHECK_RV(rv, "C_CreateObject(RSA wrapping key)", CKR_OK); + rv = funcList->C_CreateObject(session, rsaPrivTmpl, + sizeof(rsaPrivTmpl) / sizeof(*rsaPrivTmpl), + &rsaPriv); + CHECK_RV(rv, "C_CreateObject(RSA unwrapping key)", CKR_OK); +#endif + + for (i = 0; i < sizeof(cases) / sizeof(cases[0]); i++) { + rv = funcList->C_GetMechanismInfo(slot, cases[i].mech, &info); + if (rv == CKR_MECHANISM_INVALID) + continue; + snprintf(msg, sizeof(msg), "C_GetMechanismInfo(0x%lx)", + (unsigned long)cases[i].mech); + CHECK_RV(rv, msg, CKR_OK); + if (rv != CKR_OK) + continue; + + wrapKey = aesKey; + unwrapKey = aesKey; +#ifndef NO_RSA + if (cases[i].rsa) { + wrapKey = rsaPub; + unwrapKey = rsaPriv; + } +#endif + if (wrapKey == CK_INVALID_HANDLE || unwrapKey == CK_INVALID_HANDLE) + continue; + + mech.mechanism = cases[i].mech; + mech.pParameter = cases[i].useIv ? iv : NULL; + mech.ulParameterLen = cases[i].useIv ? sizeof(iv) : 0; + + wrappedLen = sizeof(wrapped); + rv = funcList->C_WrapKey(session, &mech, wrapKey, secret, wrapped, + &wrappedLen); + snprintf(msg, sizeof(msg), "C_WrapKey(0x%lx) matches CKF_WRAP", + (unsigned long)cases[i].mech); + CHECK_RV(rv, msg, ((info.flags & CKF_WRAP) != 0) ? + CKR_OK : WRAP_UNSUPPORTED_RV); + + unwrapData = wrapped; + unwrapLen = wrappedLen; + if (rv != CKR_OK) { + XMEMSET(wrapped, 0x44, sizeof(wrapped)); + unwrapLen = cases[i].rsa ? 256 : 48; + } + if (rv != CKR_OK && (info.flags & CKF_UNWRAP) != 0) { + /* Unwrap works without wrap support: encrypt the key directly. */ + wrappedLen = sizeof(wrapped); + rv = funcList->C_EncryptInit(session, &mech, wrapKey); + if (rv == CKR_OK) { + rv = funcList->C_Encrypt(session, secretValue, + sizeof(secretValue), wrapped, + &wrappedLen); + } + snprintf(msg, sizeof(msg), "C_Encrypt(0x%lx) key to unwrap", + (unsigned long)cases[i].mech); + CHECK_RV(rv, msg, CKR_OK); + unwrapLen = wrappedLen; + } + unwrapped = CK_INVALID_HANDLE; + /* AES-ECB needs CKA_VALUE_LEN to know the key length. */ + rv = funcList->C_UnwrapKey(session, &mech, unwrapKey, unwrapData, + unwrapLen, unwrapTmpl, + sizeof(unwrapTmpl) / sizeof(*unwrapTmpl) - + (cases[i].mech == CKM_AES_ECB ? 0 : 1), + &unwrapped); + snprintf(msg, sizeof(msg), "C_UnwrapKey(0x%lx) matches CKF_UNWRAP", + (unsigned long)cases[i].mech); + CHECK_RV(rv, msg, ((info.flags & CKF_UNWRAP) != 0) ? + CKR_OK : CKR_MECHANISM_INVALID); + if (unwrapped != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, unwrapped); + } +} + +#ifndef NO_AES +/* AES-ECB wraps a key that is not block aligned and unwraps it back. */ +static void test_ecb_wrap_short_key(CK_SLOT_ID slot, CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM_INFO info; + CK_MECHANISM mech = { CKM_AES_ECB, NULL, 0 }; + CK_OBJECT_HANDLE secret = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE unwrapped = CK_INVALID_HANDLE; + CK_KEY_TYPE aesType = CKK_AES; + CK_ULONG valueLen = 20; + CK_BYTE secretValue[20]; + CK_BYTE aesValue[16]; + CK_BYTE wrapped[64]; + CK_BYTE value[32]; + CK_ULONG wrappedLen = sizeof(wrapped); + CK_ATTRIBUTE secretTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, secretValue, sizeof(secretValue) }, + }; + CK_ATTRIBUTE aesTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_WRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_UNWRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesValue, sizeof(aesValue) }, + }; + CK_ATTRIBUTE unwrapTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &valueLen, sizeof(valueLen) }, + }; + CK_ATTRIBUTE valueAttr = { CKA_VALUE, value, sizeof(value) }; + + rv = funcList->C_GetMechanismInfo(slot, CKM_AES_ECB, &info); + if (rv != CKR_OK || (info.flags & CKF_WRAP) == 0) + return; + XMEMSET(secretValue, 0x5c, sizeof(secretValue)); + XMEMSET(aesValue, 0x3a, sizeof(aesValue)); + rv = funcList->C_CreateObject(session, secretTmpl, + sizeof(secretTmpl) / sizeof(*secretTmpl), &secret); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, aesTmpl, + sizeof(aesTmpl) / sizeof(*aesTmpl), &aesKey); + } + CHECK_RV(rv, "create 20 byte secret and AES wrapping key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_WrapKey(session, &mech, aesKey, secret, wrapped, + &wrappedLen); + CHECK_RV(rv, "C_WrapKey(AES-ECB) pads a 20 byte key", CKR_OK); + CHECK_TRUE(rv != CKR_OK || wrappedLen == 32, + "AES-ECB wrapped length is two blocks"); + } + if (rv == CKR_OK) { + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, + sizeof(unwrapTmpl) / sizeof(*unwrapTmpl), + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) with CKA_VALUE_LEN", CKR_OK); + } + if (rv == CKR_OK) { + CK_OBJECT_HANDLE noLen = CK_INVALID_HANDLE; + + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, + sizeof(unwrapTmpl) / sizeof(*unwrapTmpl) - 1, + &noLen); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) needs CKA_VALUE_LEN", + CKR_TEMPLATE_INCOMPLETE); + rv = CKR_OK; + } + if (rv == CKR_OK) { + rv = funcList->C_GetAttributeValue(session, unwrapped, &valueAttr, 1); + CHECK_TRUE(rv == CKR_OK && valueAttr.ulValueLen == sizeof(secretValue) + && XMEMCMP(value, secretValue, sizeof(secretValue)) == 0, + "AES-ECB unwrapped key has its original value"); + } + if (unwrapped != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, unwrapped); + if (aesKey != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, aesKey); + if (secret != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, secret); +} + +/* An AES-128 key unwraps under AES-ECB without CKA_VALUE_LEN, a two block + * value needs it since it may be a padded AES-192 key, and a CKA_VALUE_LEN + * that is zero or longer than the decrypted data is rejected. */ +static void test_ecb_unwrap_aes_key(CK_SLOT_ID slot, CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM_INFO info; + CK_MECHANISM mech = { CKM_AES_ECB, NULL, 0 }; + CK_OBJECT_HANDLE inner = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE unwrapped = CK_INVALID_HANDLE; + CK_KEY_TYPE aesType = CKK_AES; + CK_ULONG valueLen = 0; + CK_BYTE innerValue[16]; + CK_BYTE aes192Value[24]; + CK_BYTE aesValue[16]; + CK_BYTE wrapped[64]; + CK_BYTE value[32]; + CK_ULONG wrappedLen = sizeof(wrapped); + CK_ATTRIBUTE innerTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, innerValue, sizeof(innerValue) }, + }; + CK_ATTRIBUTE aesTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_WRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_UNWRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesValue, sizeof(aesValue) }, + }; + CK_ATTRIBUTE unwrapTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &valueLen, sizeof(valueLen) }, + }; + CK_ULONG unwrapCnt = sizeof(unwrapTmpl) / sizeof(*unwrapTmpl); + CK_ATTRIBUTE valueAttr = { CKA_VALUE, value, sizeof(value) }; + + rv = funcList->C_GetMechanismInfo(slot, CKM_AES_ECB, &info); + if (rv != CKR_OK || (info.flags & CKF_WRAP) == 0) + return; + XMEMSET(innerValue, 0x6d, sizeof(innerValue)); + XMEMSET(aes192Value, 0x7b, sizeof(aes192Value)); + XMEMSET(aesValue, 0x2e, sizeof(aesValue)); + rv = funcList->C_CreateObject(session, innerTmpl, + sizeof(innerTmpl) / sizeof(*innerTmpl), &inner); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, aesTmpl, + sizeof(aesTmpl) / sizeof(*aesTmpl), &aesKey); + } + CHECK_RV(rv, "create AES-128 key and AES wrapping key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_WrapKey(session, &mech, aesKey, inner, wrapped, + &wrappedLen); + CHECK_RV(rv, "C_WrapKey(AES-ECB) of an AES-128 key", CKR_OK); + CHECK_TRUE(rv != CKR_OK || wrappedLen == sizeof(innerValue), + "AES-ECB wrapped AES-128 key is one block"); + } + if (rv == CKR_OK) { + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, unwrapCnt - 1, + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) of an AES key without CKA_VALUE_LEN", + CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_GetAttributeValue(session, unwrapped, &valueAttr, 1); + CHECK_TRUE(rv == CKR_OK && valueAttr.ulValueLen == sizeof(innerValue) + && XMEMCMP(value, innerValue, sizeof(innerValue)) == 0, + "AES-ECB unwrapped AES key keeps the full value"); + funcList->C_DestroyObject(session, unwrapped); + unwrapped = CK_INVALID_HANDLE; + } + if (rv == CKR_OK) { + valueLen = 0; + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, unwrapCnt, + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) rejects a zero CKA_VALUE_LEN", + CKR_TEMPLATE_INCONSISTENT); + valueLen = wrappedLen + 1; + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, unwrapCnt, + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) rejects an oversized CKA_VALUE_LEN", + CKR_TEMPLATE_INCONSISTENT); + valueLen = wrappedLen; + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, unwrapCnt, + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) of an AES key with CKA_VALUE_LEN", + CKR_OK); + funcList->C_DestroyObject(session, unwrapped); + unwrapped = CK_INVALID_HANDLE; + } + if (rv == CKR_OK) { + funcList->C_DestroyObject(session, inner); + inner = CK_INVALID_HANDLE; + innerTmpl[5].pValue = aes192Value; + innerTmpl[5].ulValueLen = sizeof(aes192Value); + rv = funcList->C_CreateObject(session, innerTmpl, + sizeof(innerTmpl) / sizeof(*innerTmpl), &inner); + CHECK_RV(rv, "create AES-192 key", CKR_OK); + } + if (rv == CKR_OK) { + wrappedLen = sizeof(wrapped); + rv = funcList->C_WrapKey(session, &mech, aesKey, inner, wrapped, + &wrappedLen); + CHECK_RV(rv, "C_WrapKey(AES-ECB) pads an AES-192 key", CKR_OK); + CHECK_TRUE(rv != CKR_OK || wrappedLen == 32, + "AES-ECB wrapped AES-192 key is two blocks"); + } + if (rv == CKR_OK) { + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, unwrapCnt - 1, + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) of two blocks needs CKA_VALUE_LEN", + CKR_TEMPLATE_INCOMPLETE); + valueLen = sizeof(aes192Value); + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, unwrapCnt, + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) of an AES-192 key with CKA_VALUE_LEN", + CKR_OK); + } + if (rv == CKR_OK) { + valueAttr.ulValueLen = sizeof(value); + rv = funcList->C_GetAttributeValue(session, unwrapped, &valueAttr, 1); + CHECK_TRUE(rv == CKR_OK && valueAttr.ulValueLen == sizeof(aes192Value) + && XMEMCMP(value, aes192Value, sizeof(aes192Value)) == 0, + "AES-ECB unwrapped AES-192 key has its original value"); + } + if (unwrapped != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, unwrapped); + if (aesKey != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, aesKey); + if (inner != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, inner); +} +#endif + +#if !defined(NO_RSA) && !defined(NO_AES) && defined(HAVE_AES_CBC) && \ + !defined(WOLFPKCS11_NO_STORE) +/* AES-ECB refuses an RSA key whose encoding is not block aligned, and the + * zero padded encoding unwraps under AES-ECB without CKA_VALUE_LEN. */ +static void test_ecb_wrap_rsa_key(CK_SLOT_ID slot, CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM_INFO info; + CK_MECHANISM mech = { CKM_AES_ECB, NULL, 0 }; + CK_MECHANISM cbcMech = { CKM_AES_CBC_PAD, NULL, 0 }; + CK_OBJECT_HANDLE rsaPriv = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE unwrapped = CK_INVALID_HANDLE; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE rsaType = CKK_RSA; + CK_KEY_TYPE aesType = CKK_AES; + CK_BYTE aesValue[16]; + CK_BYTE iv[16]; + CK_BYTE wrapped[2048]; + CK_BYTE encoded[2048]; + CK_BYTE modulus[sizeof(rsa_2048_modulus)]; + CK_ULONG wrappedLen = sizeof(wrapped); + CK_ULONG encodedLen = sizeof(encoded); + CK_ULONG ecbLen = sizeof(encoded); + CK_ATTRIBUTE modAttr = { CKA_MODULUS, modulus, sizeof(modulus) }; + CK_ATTRIBUTE rsaPrivTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE aesTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_WRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_UNWRAP, &ckTrue, sizeof(ckTrue) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesValue, sizeof(aesValue) }, + }; + CK_ATTRIBUTE unwrapTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + + rv = funcList->C_GetMechanismInfo(slot, CKM_AES_ECB, &info); + if (rv != CKR_OK || (info.flags & CKF_WRAP) == 0) + return; + XMEMSET(aesValue, 0x4b, sizeof(aesValue)); + XMEMSET(iv, 0x1c, sizeof(iv)); + cbcMech.pParameter = iv; + cbcMech.ulParameterLen = sizeof(iv); + rv = funcList->C_CreateObject(session, rsaPrivTmpl, + sizeof(rsaPrivTmpl) / sizeof(*rsaPrivTmpl), &rsaPriv); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, aesTmpl, + sizeof(aesTmpl) / sizeof(*aesTmpl), &aesKey); + } + CHECK_RV(rv, "create RSA private key and AES wrapping key", CKR_OK); + /* Recover the key encoding through AES-CBC-PAD; it is zero padded below. */ + if (rv == CKR_OK) { + rv = funcList->C_WrapKey(session, &cbcMech, aesKey, rsaPriv, wrapped, + &wrappedLen); + if (rv == CKR_FUNCTION_FAILED) { + printf("Skipping AES-ECB RSA wrap: RSA key export unsupported\n"); + rv = CKR_GENERAL_ERROR; + } + else { + CHECK_RV(rv, "C_WrapKey(AES-CBC-PAD) of the RSA key", CKR_OK); + } + } + if (rv == CKR_OK) { + rv = funcList->C_WrapKey(session, &mech, aesKey, rsaPriv, encoded, + &ecbLen); + CHECK_RV(rv, "C_WrapKey(AES-ECB) refuses an unaligned RSA key", + CKR_KEY_SIZE_RANGE); + ecbLen = 0; + rv = funcList->C_WrapKey(session, &mech, aesKey, rsaPriv, NULL, + &ecbLen); + CHECK_RV(rv, "C_WrapKey(AES-ECB) length query refuses it too", + CKR_KEY_SIZE_RANGE); + rv = (rv == CKR_KEY_SIZE_RANGE) ? CKR_OK : CKR_GENERAL_ERROR; + } + if (rv == CKR_OK) { + rv = funcList->C_DecryptInit(session, &cbcMech, aesKey); + if (rv == CKR_OK) { + rv = funcList->C_Decrypt(session, wrapped, wrappedLen, encoded, + &encodedLen); + } + CHECK_RV(rv, "C_Decrypt(AES-CBC-PAD) of the wrapped RSA key", CKR_OK); + CHECK_TRUE(rv != CKR_OK || (encodedLen % 16) != 0, + "RSA key encoding is not block aligned"); + } + if (rv == CKR_OK) { + XMEMSET(encoded + encodedLen, 0, 16 - (encodedLen % 16)); + encodedLen += 16 - (encodedLen % 16); + wrappedLen = sizeof(wrapped); + rv = funcList->C_EncryptInit(session, &mech, aesKey); + if (rv == CKR_OK) { + rv = funcList->C_Encrypt(session, encoded, encodedLen, wrapped, + &wrappedLen); + } + CHECK_RV(rv, "C_Encrypt(AES-ECB) of the padded RSA key", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_UnwrapKey(session, &mech, aesKey, wrapped, + wrappedLen, unwrapTmpl, + sizeof(unwrapTmpl) / sizeof(*unwrapTmpl), + &unwrapped); + CHECK_RV(rv, "C_UnwrapKey(AES-ECB) of an RSA key without CKA_VALUE_LEN", + CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_GetAttributeValue(session, unwrapped, &modAttr, 1); + CHECK_TRUE(rv == CKR_OK && modAttr.ulValueLen == sizeof(modulus) && + XMEMCMP(modulus, rsa_2048_modulus, sizeof(modulus)) == 0, + "AES-ECB unwrapped RSA key has the original modulus"); + } + if (unwrapped != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, unwrapped); + if (aesKey != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, aesKey); + if (rsaPriv != CK_INVALID_HANDLE) + funcList->C_DestroyObject(session, rsaPriv); +} +#endif + +static int run_test(void) +{ + CK_RV rv; + CK_C_INITIALIZE_ARGS args; + CK_SLOT_ID slotList[16]; + CK_ULONG slotCount = sizeof(slotList) / sizeof(slotList[0]); + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + rv = funcList->C_Initialize(&args); + CHECK_RV(rv, "C_Initialize", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_GetSlotList(CK_TRUE, slotList, &slotCount); + CHECK_RV(rv, "C_GetSlotList", CKR_OK); + } + if (rv == CKR_OK && slotCount == 0) { + CHECK_TRUE(0, "at least one slot"); + rv = CKR_TOKEN_NOT_PRESENT; + } + if (rv == CKR_OK) { + test_info_blank_padding(slotList[0]); + test_token_clock(slotList[0]); + rv = funcList->C_OpenSession(slotList[0], + CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &session); + CHECK_RV(rv, "C_OpenSession", CKR_OK); + } + if (rv == CKR_OK) { + test_generic_secret_size(slotList[0], session); + test_listed_keygen_supported(slotList[0], session); +#ifdef ALLOC_HOOKS + test_open_session_alloc_failure(slotList[0]); +#endif + test_wrap_flags(slotList[0], session); +#ifndef NO_AES + test_ecb_wrap_short_key(slotList[0], session); + test_ecb_unwrap_aes_key(slotList[0], session); +#endif +#if !defined(NO_RSA) && !defined(NO_AES) && defined(HAVE_AES_CBC) && \ + !defined(WOLFPKCS11_NO_STORE) + test_ecb_wrap_rsa_key(slotList[0], session); +#endif + funcList->C_CloseSession(session); + } + funcList->C_Finalize(NULL); + + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + + printf("=== wolfPKCS11 info and mechanism table test ===\n"); +#ifdef ALLOC_HOOKS + if (wolfSSL_SetAllocators(hook_malloc, hook_free, hook_realloc) != 0) { + CHECK_TRUE(0, "install allocation hooks"); + return pkcs11_test_summary(); + } +#endif + run_test(); + return pkcs11_test_summary(); +} diff --git a/tests/kdf_tls_derive_test.c b/tests/kdf_tls_derive_test.c new file mode 100644 index 00000000..f619085e --- /dev/null +++ b/tests/kdf_tls_derive_test.c @@ -0,0 +1,1276 @@ +/* kdf_tls_derive_test.c + * + * Copyright (C) 2006-2025 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Tests for C_DeriveKey with the HKDF, DH, AES-CBC and TLS 1.2 mechanisms: + * mechanism parameter validation and the attributes of the derived keys. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#if defined(WOLFSSL_HAVE_PRF) && !defined(WOLFPKCS11_NO_STORE) && \ + !defined(_WIN32) + #define TLS_PERSIST_TEST + #include + #include +#endif + +#include "testdata.h" + +#define TEST_DIR "./store/kdf_tls_derive_test" + +/* CK_ULONG lengths above the 32-bit range can only be expressed on LP64. */ +#if ULONG_MAX > 0xFFFFFFFFUL + #define WIDE_CK_ULONG + #define LEN_ABOVE_WORD32(n) ((CK_ULONG)0xFFFFFFFFUL + 1 + (CK_ULONG)(n)) +#endif + +#if defined(WOLFSSL_HAVE_PRF) && defined(WOLFPKCS11_NSS) && \ + defined(WIDE_CK_ULONG) + #define NSS_EMS_WIDE_TEST +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) && defined(WIDE_CK_ULONG) + #define AES_CBC_WIDE_TEST +#endif + +#if defined(WOLFPKCS11_HKDF) || \ + defined(NSS_EMS_WIDE_TEST) || defined(WOLFSSL_HAVE_PRF) || \ + (!defined(NO_AES) && defined(HAVE_AES_CBC)) + #define SECRET_BASE_TESTS +#endif + +#if defined(SECRET_BASE_TESTS) || !defined(NO_DH) + #define DERIVE_TESTS +#endif + +static int test_passed = 0; +static int test_failed = 0; + +#ifndef HAVE_PKCS11_STATIC +static void* dlib; +#endif +static CK_FUNCTION_LIST* funcList; +static CK_SLOT_ID slot = 0; +static const char* tokenName = "wolfpkcs11"; +static byte* soPin = (byte*)"password123456"; +static int soPinLen = 14; +static byte* userPin = (byte*)"wolfpkcs11-test"; +static int userPinLen = 15; + +#define CHECK_RV(rv, exp, op) do { \ + if ((rv) != (exp)) { \ + fprintf(stderr, "FAIL: %s: got 0x%lx, expected 0x%lx\n", op, \ + (unsigned long)(rv), (unsigned long)(exp)); \ + test_failed++; \ + result = -1; \ + goto cleanup; \ + } \ + else { \ + printf("PASS: %s\n", op); \ + test_passed++; \ + } \ +} while (0) + +#define CHECK_CKR(rv, op) CHECK_RV(rv, CKR_OK, op) + +#ifdef DERIVE_TESTS +static CK_OBJECT_CLASS secretKeyClass = CKO_SECRET_KEY; +static CK_KEY_TYPE genericKeyType = CKK_GENERIC_SECRET; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; +#endif + +#ifdef SECRET_BASE_TESTS + +/* Not every SECRET_BASE_TESTS configuration uses the shared secret. */ +#if defined(__GNUC__) + #pragma GCC diagnostic push + #pragma GCC diagnostic ignored "-Wunused-variable" +#endif +static byte baseSecret[32] = { + 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, + 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, + 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, + 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b +}; +#if defined(__GNUC__) + #pragma GCC diagnostic pop +#endif + +/* Session secret key of the given type usable as a derive base. */ +static CK_RV create_secret_base(CK_SESSION_HANDLE session, CK_KEY_TYPE type, + byte* value, CK_ULONG valueLen, + CK_OBJECT_HANDLE* obj) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &type, sizeof(type) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, value, valueLen }, + }; + CK_ULONG cnt = sizeof(tmpl) / sizeof(*tmpl); + + return funcList->C_CreateObject(session, tmpl, cnt, obj); +} +#endif /* SECRET_BASE_TESTS */ + +#ifdef DERIVE_TESTS +/* Derive a public, extractable generic secret of the given length. */ +static CK_RV derive_generic(CK_SESSION_HANDLE session, CK_MECHANISM* mech, + CK_OBJECT_HANDLE base, CK_ULONG len, + CK_OBJECT_HANDLE* derived) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &len, sizeof(len) }, + }; + CK_ULONG cnt = sizeof(tmpl) / sizeof(*tmpl); + + *derived = CK_INVALID_HANDLE; + return funcList->C_DeriveKey(session, mech, base, tmpl, cnt, derived); +} + +static void destroy_obj(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* obj) +{ + if (*obj != CK_INVALID_HANDLE) { + funcList->C_DestroyObject(session, *obj); + *obj = CK_INVALID_HANDLE; + } +} +#endif /* DERIVE_TESTS */ + +#if defined(WOLFPKCS11_HKDF) && defined(WIDE_CK_ULONG) +/* HKDF salt and info lengths must be representable without truncation. */ +static int test_hkdf_lengths_fit_word32(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + byte salt[4] = { 0x00, 0x01, 0x02, 0x03 }; + byte info[4] = { 0xf0, 0xf1, 0xf2, 0xf3 }; + CK_HKDF_PARAMS params; + CK_MECHANISM mech = { CKM_HKDF_DERIVE, ¶ms, sizeof(params) }; + int result = 0; + + ret = create_secret_base(session, CKK_GENERIC_SECRET, baseSecret, + sizeof(baseSecret), &base); + CHECK_CKR(ret, "create HKDF base key"); + + XMEMSET(¶ms, 0, sizeof(params)); + params.bExtract = CK_TRUE; + params.bExpand = CK_TRUE; + params.prfHashMechanism = CKM_SHA256; + params.ulSaltType = CKF_HKDF_SALT_DATA; + params.pSalt = salt; + params.ulSaltLen = sizeof(salt); + params.pInfo = info; + params.ulInfoLen = sizeof(info); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_CKR(ret, "HKDF derive with in-range lengths"); + destroy_obj(session, &derived); + + params.ulSaltLen = LEN_ABOVE_WORD32(sizeof(salt)); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "HKDF salt length beyond 32 bits rejected"); + params.ulSaltLen = sizeof(salt); + + params.bExtract = CK_FALSE; + params.ulInfoLen = LEN_ABOVE_WORD32(sizeof(info)); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "HKDF info length beyond 32 bits rejected"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef NSS_EMS_WIDE_TEST +/* The extended master secret session hash length must not be truncated. */ +static int test_session_hash_length_fits_word32(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + byte sessionHash[32]; + CK_VERSION version = { 3, 3 }; + CK_NSS_TLS_EXTENDED_MASTER_KEY_DERIVE_PARAMS params; + CK_MECHANISM mech = { CKM_NSS_TLS_EXTENDED_MASTER_KEY_DERIVE, ¶ms, + sizeof(params) }; + int result = 0; + + XMEMSET(sessionHash, 0xAA, sizeof(sessionHash)); + ret = create_secret_base(session, CKK_GENERIC_SECRET, baseSecret, + sizeof(baseSecret), &base); + CHECK_CKR(ret, "create TLS base key"); + + XMEMSET(¶ms, 0, sizeof(params)); + params.prfHashMechanism = CKM_SHA256; + params.pSessionHash = sessionHash; + params.ulSessionHashLen = sizeof(sessionHash); + params.pVersion = &version; + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_CKR(ret, "extended master secret derive"); + destroy_obj(session, &derived); + + params.ulSessionHashLen = LEN_ABOVE_WORD32(sizeof(sessionHash)); + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "session hash length beyond 32 bits rejected"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifndef NO_DH +/* DH peer public value lengths are bounded by the largest supported prime. */ +static int test_dh_public_length_bound(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE dhType = CKK_DH; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_MECHANISM mech; + byte padded[8192 / 8 + 1]; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &dhType, sizeof(dhType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv) }, + }; + int result = 0; + + ret = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &base); + CHECK_CKR(ret, "create DH base key"); + + mech.mechanism = CKM_DH_PKCS_DERIVE; + mech.pParameter = dh_2048_peer; + mech.ulParameterLen = sizeof(dh_2048_peer); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_CKR(ret, "DH derive with peer public value"); + destroy_obj(session, &derived); + + XMEMSET(padded, 0, sizeof(padded)); + XMEMCPY(padded + sizeof(padded) - sizeof(dh_2048_peer), dh_2048_peer, + sizeof(dh_2048_peer)); + mech.pParameter = padded; + mech.ulParameterLen = sizeof(padded); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "DH public value longer than any prime rejected"); + +#ifdef WIDE_CK_ULONG + mech.pParameter = dh_2048_peer; + mech.ulParameterLen = LEN_ABOVE_WORD32(sizeof(dh_2048_peer)); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "DH public value length beyond 32 bits rejected"); +#endif + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef AES_CBC_WIDE_TEST +/* The AES-CBC encrypt-data length must be representable in 32 bits. */ +static int test_aes_cbc_data_length_fits_word32(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + byte key[sizeof(aes_cbc_key)]; + byte plain[sizeof(aes_cbc_plain)]; + CK_AES_CBC_ENCRYPT_DATA_PARAMS params; + CK_MECHANISM mech = { CKM_AES_CBC_ENCRYPT_DATA, ¶ms, sizeof(params) }; + int result = 0; + + XMEMCPY(key, aes_cbc_key, sizeof(key)); + XMEMCPY(plain, aes_cbc_plain, sizeof(plain)); + ret = create_secret_base(session, CKK_AES, key, sizeof(key), &base); + CHECK_CKR(ret, "create AES base key"); + + XMEMSET(¶ms, 0, sizeof(params)); + XMEMCPY(params.iv, aes_cbc_iv, sizeof(params.iv)); + params.pData = plain; + params.length = sizeof(plain); + ret = derive_generic(session, &mech, base, sizeof(plain), &derived); + CHECK_CKR(ret, "AES-CBC encrypt-data derive"); + destroy_obj(session, &derived); + + params.length = LEN_ABOVE_WORD32(sizeof(plain)); + ret = derive_generic(session, &mech, base, sizeof(plain), &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "AES-CBC data length beyond 32 bits rejected"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef WOLFSSL_HAVE_PRF +static byte clientRandom[32] = { + 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, + 0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f, 0x70, + 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, + 0x79, 0x7a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f, 0x80 +}; +static byte serverRandom[32] = { + 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, + 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f, 0x50, + 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, + 0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f, 0x60 +}; + +static void tls_key_mat_init(CK_TLS12_KEY_MAT_PARAMS* params, + CK_SSL3_KEY_MAT_OUT* out, CK_ULONG macBits, + CK_ULONG keyBits, CK_ULONG ivBits) +{ + XMEMSET(out, 0, sizeof(*out)); + out->hClientMacSecret = CK_INVALID_HANDLE; + out->hServerMacSecret = CK_INVALID_HANDLE; + out->hClientKey = CK_INVALID_HANDLE; + out->hServerKey = CK_INVALID_HANDLE; + XMEMSET(params, 0, sizeof(*params)); + params->ulMacSizeInBits = macBits; + params->ulKeySizeInBits = keyBits; + params->ulIVSizeInBits = ivBits; + params->bIsExport = CK_FALSE; + params->RandomInfo.pClientRandom = clientRandom; + params->RandomInfo.ulClientRandomLen = sizeof(clientRandom); + params->RandomInfo.pServerRandom = serverRandom; + params->RandomInfo.ulServerRandomLen = sizeof(serverRandom); + params->pReturnedKeyMaterial = out; + params->prfHashMechanism = CKM_SHA256; +} + +static CK_RV tls_key_mat_derive(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE base, + CK_TLS12_KEY_MAT_PARAMS* params, + CK_ATTRIBUTE* tmpl, CK_ULONG cnt) +{ + CK_MECHANISM mech; + CK_ATTRIBUTE defTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + }; + + if (tmpl == NULL) { + tmpl = defTmpl; + cnt = sizeof(defTmpl) / sizeof(*defTmpl); + } + mech.mechanism = CKM_TLS12_KEY_AND_MAC_DERIVE; + mech.pParameter = params; + mech.ulParameterLen = sizeof(*params); + + return funcList->C_DeriveKey(session, &mech, base, tmpl, cnt, NULL); +} + +static void destroy_key_mat(CK_SESSION_HANDLE session, + CK_SSL3_KEY_MAT_OUT* out) +{ + destroy_obj(session, &out->hClientMacSecret); + destroy_obj(session, &out->hServerMacSecret); + destroy_obj(session, &out->hClientKey); + destroy_obj(session, &out->hServerKey); +} + +/* Every TLS key material size must be a whole number of bytes. */ +static int test_tls_key_sizes_byte_aligned(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_TLS12_KEY_MAT_PARAMS params; + CK_SSL3_KEY_MAT_OUT out; + byte ivClient[16]; + byte ivServer[16]; + byte master[48]; + int result = 0; + + XMEMSET(master, 0x5c, sizeof(master)); + tls_key_mat_init(¶ms, &out, 256, 128, 128); + ret = create_secret_base(session, CKK_GENERIC_SECRET, master, + sizeof(master), &base); + CHECK_CKR(ret, "create TLS master secret"); + + out.pIVClient = ivClient; + out.pIVServer = ivServer; + ret = tls_key_mat_derive(session, base, ¶ms, NULL, 0); + CHECK_CKR(ret, "TLS key and MAC derive"); + destroy_key_mat(session, &out); + + tls_key_mat_init(¶ms, &out, 4, 4, 0); + ret = tls_key_mat_derive(session, base, ¶ms, NULL, 0); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "TLS MAC and key sizes not whole bytes rejected"); + + tls_key_mat_init(¶ms, &out, 256, 128, 4); + out.pIVClient = ivClient; + out.pIVServer = ivServer; + ret = tls_key_mat_derive(session, base, ¶ms, NULL, 0); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "TLS IV size not whole bytes rejected"); + +cleanup: + destroy_key_mat(session, &out); + destroy_obj(session, &base); + return result; +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) +/* AES-CBC encrypt-data derives the CBC encryption of the data. */ +static int test_aes_cbc_encrypt_data_value(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + byte key[sizeof(aes_cbc_key)]; + byte plain[sizeof(aes_cbc_plain)]; + byte value[sizeof(aes_cbc_cipher)]; + CK_ATTRIBUTE valueAttr = { CKA_VALUE, value, sizeof(value) }; + CK_AES_CBC_ENCRYPT_DATA_PARAMS params; + CK_MECHANISM mech = { CKM_AES_CBC_ENCRYPT_DATA, ¶ms, sizeof(params) }; + int result = 0; + + XMEMCPY(key, aes_cbc_key, sizeof(key)); + XMEMCPY(plain, aes_cbc_plain, sizeof(plain)); + ret = create_secret_base(session, CKK_AES, key, sizeof(key), &base); + CHECK_CKR(ret, "create AES base key"); + + XMEMSET(¶ms, 0, sizeof(params)); + XMEMCPY(params.iv, aes_cbc_iv, sizeof(params.iv)); + params.pData = plain; + params.length = sizeof(plain); + ret = derive_generic(session, &mech, base, sizeof(plain), &derived); + CHECK_CKR(ret, "AES-CBC encrypt-data derive"); + + ret = funcList->C_GetAttributeValue(session, derived, &valueAttr, 1); + CHECK_CKR(ret, "read derived key value"); + if (valueAttr.ulValueLen != sizeof(aes_cbc_cipher) || + XMEMCMP(value, aes_cbc_cipher, sizeof(aes_cbc_cipher)) != 0) + ret = CKR_GENERAL_ERROR; + CHECK_CKR(ret, "derived key is the CBC encryption of the data"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef WOLFSSL_HAVE_PRF +/* TLS key expansion requires both client and server random data. */ +static int test_tls_randoms_required(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_TLS12_KEY_MAT_PARAMS params; + CK_SSL3_KEY_MAT_OUT out; + byte master[48]; + int result = 0; + + XMEMSET(master, 0x5c, sizeof(master)); + tls_key_mat_init(¶ms, &out, 256, 128, 0); + ret = create_secret_base(session, CKK_GENERIC_SECRET, master, + sizeof(master), &base); + CHECK_CKR(ret, "create TLS master secret"); + + params.RandomInfo.pClientRandom = NULL; + ret = tls_key_mat_derive(session, base, ¶ms, NULL, 0); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "TLS key and MAC derive without client random rejected"); + + tls_key_mat_init(¶ms, &out, 256, 128, 0); + params.RandomInfo.pServerRandom = NULL; + ret = tls_key_mat_derive(session, base, ¶ms, NULL, 0); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "TLS key and MAC derive without server random rejected"); + +cleanup: + destroy_key_mat(session, &out); + destroy_obj(session, &base); + return result; +} +#endif + +#if defined(WOLFSSL_HAVE_PRF) && defined(WOLFPKCS11_NSS) +/* The extended master secret is bound to a session hash. */ +static int test_session_hash_required(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_VERSION version = { 3, 3 }; + CK_NSS_TLS_EXTENDED_MASTER_KEY_DERIVE_PARAMS params; + CK_MECHANISM mech = { CKM_NSS_TLS_EXTENDED_MASTER_KEY_DERIVE, ¶ms, + sizeof(params) }; + int result = 0; + + ret = create_secret_base(session, CKK_GENERIC_SECRET, baseSecret, + sizeof(baseSecret), &base); + CHECK_CKR(ret, "create TLS base key"); + + XMEMSET(¶ms, 0, sizeof(params)); + params.prfHashMechanism = CKM_SHA256; + params.pSessionHash = NULL; + params.ulSessionHashLen = 0; + params.pVersion = &version; + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "extended master secret without session hash rejected"); + + params.ulSessionHashLen = 32; + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "extended master secret with NULL session hash rejected"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef WOLFSSL_HAVE_PRF +/* An unsupported protocol version is a mechanism parameter error. */ +static int test_tls_master_version_param(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_VERSION version = { 3, 3 }; + CK_TLS12_MASTER_KEY_DERIVE_PARAMS params; + CK_MECHANISM mech = { CKM_TLS12_MASTER_KEY_DERIVE, ¶ms, + sizeof(params) }; + byte preMaster[48]; + int result = 0; + + XMEMSET(preMaster, 0x33, sizeof(preMaster)); + preMaster[0] = 3; + preMaster[1] = 3; + ret = create_secret_base(session, CKK_GENERIC_SECRET, preMaster, + sizeof(preMaster), &base); + CHECK_CKR(ret, "create TLS pre-master secret"); + + XMEMSET(¶ms, 0, sizeof(params)); + params.RandomInfo.pClientRandom = clientRandom; + params.RandomInfo.ulClientRandomLen = sizeof(clientRandom); + params.RandomInfo.pServerRandom = serverRandom; + params.RandomInfo.ulServerRandomLen = sizeof(serverRandom); + params.pVersion = &version; + params.prfHashMechanism = CKM_SHA256; + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_CKR(ret, "TLS 1.2 master secret derive"); + destroy_obj(session, &derived); + + version.minor = 1; + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "TLS master secret derive with other version rejected"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#if defined(WOLFPKCS11_HKDF) || !defined(NO_DH) || \ + (!defined(NO_AES) && defined(HAVE_AES_CBC)) +/* A derived key has the requested CKA_VALUE_LEN or the derive fails. */ +static int test_derived_length_matches_template(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_MECHANISM mech; +#ifdef WOLFPKCS11_HKDF + CK_HKDF_PARAMS hkdfParams; +#endif +#if !defined(NO_AES) && defined(HAVE_AES_CBC) + byte key[sizeof(aes_cbc_key)]; + byte plain[sizeof(aes_cbc_plain)]; + CK_AES_CBC_ENCRYPT_DATA_PARAMS cbcParams; +#endif +#ifndef NO_DH + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE dhType = CKK_DH; + byte peer[2 * sizeof(dh_ffdhe2048_p)]; + byte value[sizeof(dh_ffdhe2048_p)]; + byte expect[sizeof(dh_ffdhe2048_p)]; + CK_ATTRIBUTE valueAttr = { CKA_VALUE, value, sizeof(value) }; + CK_ATTRIBUTE dhTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &dhType, sizeof(dhType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv) }, + }; +#endif + int result = 0; + +#ifdef WOLFPKCS11_HKDF + ret = create_secret_base(session, CKK_GENERIC_SECRET, baseSecret, + sizeof(baseSecret), &base); + CHECK_CKR(ret, "create HKDF base key"); + XMEMSET(&hkdfParams, 0, sizeof(hkdfParams)); + hkdfParams.bExtract = CK_TRUE; + hkdfParams.bExpand = CK_FALSE; + hkdfParams.prfHashMechanism = CKM_SHA256; + hkdfParams.ulSaltType = CKF_HKDF_SALT_NULL; + mech.mechanism = CKM_HKDF_DERIVE; + mech.pParameter = &hkdfParams; + mech.ulParameterLen = sizeof(hkdfParams); + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_CKR(ret, "HKDF extract with digest-sized length"); + destroy_obj(session, &derived); + ret = derive_generic(session, &mech, base, 48, &derived); + CHECK_RV(ret, CKR_ATTRIBUTE_VALUE_INVALID, + "HKDF extract longer than the digest rejected"); + destroy_obj(session, &base); +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) + XMEMCPY(key, aes_cbc_key, sizeof(key)); + XMEMCPY(plain, aes_cbc_plain, sizeof(plain)); + ret = create_secret_base(session, CKK_AES, key, sizeof(key), &base); + CHECK_CKR(ret, "create AES base key"); + XMEMSET(&cbcParams, 0, sizeof(cbcParams)); + XMEMCPY(cbcParams.iv, aes_cbc_iv, sizeof(cbcParams.iv)); + cbcParams.pData = plain; + cbcParams.length = sizeof(plain); + mech.mechanism = CKM_AES_CBC_ENCRYPT_DATA; + mech.pParameter = &cbcParams; + mech.ulParameterLen = sizeof(cbcParams); + ret = derive_generic(session, &mech, base, 2 * sizeof(plain), &derived); + CHECK_RV(ret, CKR_ATTRIBUTE_VALUE_INVALID, + "AES-CBC encrypt-data longer than the data rejected"); + destroy_obj(session, &base); +#endif + +#ifndef NO_DH + ret = funcList->C_CreateObject(session, dhTmpl, + sizeof(dhTmpl) / sizeof(*dhTmpl), &base); + CHECK_CKR(ret, "create DH base key"); + /* This peer value gives a shared secret with a leading zero byte. */ + XMEMSET(peer, 0, sizeof(peer)); + peer[sizeof(peer) - 2] = 0x01; + peer[sizeof(peer) - 1] = 0x5f; + mech.mechanism = CKM_DH_PKCS_DERIVE; + mech.pParameter = peer + sizeof(dh_ffdhe2048_p); + mech.ulParameterLen = sizeof(dh_ffdhe2048_p); + ret = derive_generic(session, &mech, base, sizeof(dh_ffdhe2048_p), + &derived); + CHECK_CKR(ret, "DH derive of a prime-sized secret"); + ret = funcList->C_GetAttributeValue(session, derived, &valueAttr, 1); + CHECK_CKR(ret, "read DH derived value"); + if (valueAttr.ulValueLen != sizeof(dh_ffdhe2048_p) || value[0] != 0) + ret = CKR_GENERAL_ERROR; + CHECK_CKR(ret, "DH derived value keeps its leading zero byte"); + destroy_obj(session, &derived); + + /* The same peer value encoded without its leading zero bytes. */ + XMEMCPY(expect, value, sizeof(expect)); + mech.pParameter = peer + sizeof(peer) - 2; + mech.ulParameterLen = 2; + ret = derive_generic(session, &mech, base, sizeof(dh_ffdhe2048_p), + &derived); + CHECK_CKR(ret, "DH derive of a prime-sized secret from a short peer value"); + valueAttr.ulValueLen = sizeof(value); + ret = funcList->C_GetAttributeValue(session, derived, &valueAttr, 1); + CHECK_CKR(ret, "read DH derived value from a short peer value"); + if (valueAttr.ulValueLen != sizeof(expect) || + XMEMCMP(value, expect, sizeof(expect)) != 0) { + ret = CKR_GENERAL_ERROR; + } + CHECK_CKR(ret, "DH derived value matches the zero padded peer derive"); + destroy_obj(session, &derived); + + mech.pParameter = peer; + mech.ulParameterLen = sizeof(peer); + ret = derive_generic(session, &mech, base, sizeof(peer), &derived); + CHECK_RV(ret, CKR_ATTRIBUTE_VALUE_INVALID, + "DH derive longer than the prime rejected"); +#endif + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef WOLFSSL_HAVE_PRF +static CK_RV read_states(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_BBOOL* alwaysSensitive, CK_BBOOL* neverExtractable) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_ALWAYS_SENSITIVE, alwaysSensitive, sizeof(CK_BBOOL) }, + { CKA_NEVER_EXTRACTABLE, neverExtractable, sizeof(CK_BBOOL) }, + }; + + return funcList->C_GetAttributeValue(session, obj, tmpl, + sizeof(tmpl) / sizeof(*tmpl)); +} + +/* All four TLS keys must report exactly the expected historical flags. */ +static CK_RV check_key_mat_states(CK_SESSION_HANDLE session, + CK_SSL3_KEY_MAT_OUT* out, CK_BBOOL expected) +{ + CK_RV ret = CKR_OK; + CK_OBJECT_HANDLE keys[4]; + CK_BBOOL alwaysSensitive; + CK_BBOOL neverExtractable; + int i; + + keys[0] = out->hClientMacSecret; + keys[1] = out->hServerMacSecret; + keys[2] = out->hClientKey; + keys[3] = out->hServerKey; + for (i = 0; i < 4 && ret == CKR_OK; i++) { + alwaysSensitive = !expected; + neverExtractable = !expected; + ret = read_states(session, keys[i], &alwaysSensitive, + &neverExtractable); + if (ret == CKR_OK && (alwaysSensitive != expected || + neverExtractable != expected)) { + fprintf(stderr, "TLS key %d: ALWAYS_SENSITIVE=%d " + "NEVER_EXTRACTABLE=%d, expected %d\n", i, + (int)alwaysSensitive, (int)neverExtractable, + (int)expected); + ret = CKR_GENERAL_ERROR; + } + } + return ret; +} + +/* TLS key material keys carry the derive historical protection flags. */ +static int test_tls_key_states_follow_base(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_TLS12_KEY_MAT_PARAMS params; + CK_SSL3_KEY_MAT_OUT out; + CK_MECHANISM genMech = { CKM_GENERIC_SECRET_KEY_GEN, NULL, 0 }; + CK_ULONG masterLen = 48; + byte master[48]; + CK_ATTRIBUTE genTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &masterLen, sizeof(masterLen) }, + }; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE importTmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, master, sizeof(master) }, + }; + int result = 0; + + tls_key_mat_init(¶ms, &out, 256, 128, 0); + ret = funcList->C_GenerateKey(session, &genMech, genTmpl, + sizeof(genTmpl) / sizeof(*genTmpl), &base); + CHECK_CKR(ret, "generate TLS master secret"); + ret = tls_key_mat_derive(session, base, ¶ms, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl)); + CHECK_CKR(ret, "TLS key and MAC derive from generated secret"); + ret = check_key_mat_states(session, &out, CK_TRUE); + CHECK_CKR(ret, "TLS keys from generated secret keep history"); + destroy_key_mat(session, &out); + destroy_obj(session, &base); + + XMEMSET(master, 0x5c, sizeof(master)); + ret = funcList->C_CreateObject(session, importTmpl, + sizeof(importTmpl) / sizeof(*importTmpl), + &base); + CHECK_CKR(ret, "import TLS master secret"); + tls_key_mat_init(¶ms, &out, 256, 128, 0); + ret = tls_key_mat_derive(session, base, ¶ms, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl)); + CHECK_CKR(ret, "TLS key and MAC derive from imported secret"); + ret = check_key_mat_states(session, &out, CK_FALSE); + CHECK_CKR(ret, "TLS keys from imported secret have no history"); + +cleanup: + destroy_key_mat(session, &out); + destroy_obj(session, &base); + return result; +} +#endif + +#ifdef WOLFPKCS11_HKDF +/* An HKDF salt key handle must refer to a symmetric secret key. */ +static int test_hkdf_salt_key_is_secret(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE salt = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_OBJECT_CLASS dataClass = CKO_DATA; + byte saltValue[16]; + byte dataValue[64]; + CK_ATTRIBUTE dataTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, dataValue, sizeof(dataValue) }, + }; + CK_HKDF_PARAMS params; + CK_MECHANISM mech = { CKM_HKDF_DERIVE, ¶ms, sizeof(params) }; + int result = 0; + + XMEMSET(saltValue, 0x5a, sizeof(saltValue)); + XMEMSET(dataValue, 0xa5, sizeof(dataValue)); + ret = create_secret_base(session, CKK_GENERIC_SECRET, baseSecret, + sizeof(baseSecret), &base); + CHECK_CKR(ret, "create HKDF base key"); + ret = create_secret_base(session, CKK_GENERIC_SECRET, saltValue, + sizeof(saltValue), &salt); + CHECK_CKR(ret, "create HKDF salt key"); + + XMEMSET(¶ms, 0, sizeof(params)); + params.bExtract = CK_TRUE; + params.bExpand = CK_TRUE; + params.prfHashMechanism = CKM_SHA256; + params.ulSaltType = CKF_HKDF_SALT_KEY; + params.hSaltKey = salt; + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_CKR(ret, "HKDF derive with secret salt key"); + destroy_obj(session, &derived); + destroy_obj(session, &salt); + + ret = funcList->C_CreateObject(session, dataTmpl, + sizeof(dataTmpl) / sizeof(*dataTmpl), + &salt); + CHECK_CKR(ret, "create data object"); + params.hSaltKey = salt; + ret = derive_generic(session, &mech, base, 32, &derived); + CHECK_RV(ret, CKR_MECHANISM_PARAM_INVALID, + "HKDF salt key that is a data object rejected"); + +cleanup: + destroy_obj(session, &derived); + destroy_obj(session, &salt); + destroy_obj(session, &base); + return result; +} +#endif + +static CK_RV pkcs11_init(void) +{ + CK_RV ret; + CK_C_INITIALIZE_ARGS args; + CK_SLOT_ID slotList[16]; + CK_ULONG slotCount = sizeof(slotList) / sizeof(slotList[0]); +#ifndef HAVE_PKCS11_STATIC + CK_C_GetFunctionList func; + + dlib = dlopen(WOLFPKCS11_DLL_FILENAME, RTLD_NOW | RTLD_LOCAL); + if (dlib == NULL) { + fprintf(stderr, "dlopen error: %s\n", dlerror()); + return -1; + } + func = (CK_C_GetFunctionList)dlsym(dlib, "C_GetFunctionList"); + if (func == NULL) { + dlclose(dlib); + dlib = NULL; + return -1; + } + ret = func(&funcList); + if (ret != CKR_OK) { + dlclose(dlib); + dlib = NULL; + return ret; + } +#else + ret = C_GetFunctionList(&funcList); + if (ret != CKR_OK) + return ret; +#endif + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + ret = funcList->C_Initialize(&args); + if (ret != CKR_OK) + return ret; + + ret = funcList->C_GetSlotList(CK_TRUE, slotList, &slotCount); + if (ret != CKR_OK) + return ret; + if (slotCount == 0) + return CKR_GENERAL_ERROR; + slot = slotList[0]; + return ret; +} + +static CK_RV pkcs11_final(void) +{ + if (funcList != NULL) { + funcList->C_Finalize(NULL); + funcList = NULL; + } +#ifndef HAVE_PKCS11_STATIC + if (dlib) { + dlclose(dlib); + dlib = NULL; + } +#endif + return CKR_OK; +} + +static CK_RV pkcs11_setup_token(void) +{ + CK_RV ret; + unsigned char label[32]; + CK_SESSION_HANDLE soSession; + int sessFlags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, tokenName, XSTRLEN(tokenName)); + ret = funcList->C_InitToken(slot, soPin, soPinLen, label); + if (ret != CKR_OK) + return ret; + + ret = funcList->C_OpenSession(slot, sessFlags, NULL, NULL, &soSession); + if (ret != CKR_OK) + return ret; + ret = funcList->C_Login(soSession, CKU_SO, soPin, soPinLen); + if (ret == CKR_OK) + ret = funcList->C_InitPIN(soSession, userPin, userPinLen); + funcList->C_Logout(soSession); + funcList->C_CloseSession(soSession); + return ret; +} + +static int kdf_tls_derive_test(void) +{ + CK_RV ret; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + int sessFlags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + int result = 0; + + printf("\n=== Testing C_DeriveKey with KDF and TLS mechanisms ===\n"); + + ret = pkcs11_init(); + CHECK_CKR(ret, "C_Initialize"); + + ret = pkcs11_setup_token(); + CHECK_CKR(ret, "token setup"); + + ret = funcList->C_OpenSession(slot, sessFlags, NULL, NULL, &session); + CHECK_CKR(ret, "C_OpenSession"); + ret = funcList->C_Login(session, CKU_USER, userPin, userPinLen); + CHECK_CKR(ret, "C_Login"); + +#if defined(WOLFPKCS11_HKDF) && defined(WIDE_CK_ULONG) + if (test_hkdf_lengths_fit_word32(session) != 0) + result = -1; +#endif + +#ifdef WOLFPKCS11_HKDF + if (test_hkdf_salt_key_is_secret(session) != 0) + result = -1; +#endif +#ifdef NSS_EMS_WIDE_TEST + if (test_session_hash_length_fits_word32(session) != 0) + result = -1; +#endif + +#ifndef NO_DH + if (test_dh_public_length_bound(session) != 0) + result = -1; +#endif + +#ifdef AES_CBC_WIDE_TEST + if (test_aes_cbc_data_length_fits_word32(session) != 0) + result = -1; +#endif + +#ifdef WOLFSSL_HAVE_PRF + if (test_tls_key_sizes_byte_aligned(session) != 0) + result = -1; +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) + if (test_aes_cbc_encrypt_data_value(session) != 0) + result = -1; +#endif + +#ifdef WOLFSSL_HAVE_PRF + if (test_tls_randoms_required(session) != 0) + result = -1; +#endif +#ifdef WOLFSSL_HAVE_PRF + if (test_tls_master_version_param(session) != 0) + result = -1; +#endif +#ifdef WOLFSSL_HAVE_PRF + if (test_tls_key_states_follow_base(session) != 0) + result = -1; +#endif +#if defined(WOLFSSL_HAVE_PRF) && defined(WOLFPKCS11_NSS) + if (test_session_hash_required(session) != 0) + result = -1; +#endif + +#if defined(WOLFPKCS11_HKDF) || !defined(NO_DH) || \ + (!defined(NO_AES) && defined(HAVE_AES_CBC)) + if (test_derived_length_matches_template(session) != 0) + result = -1; +#endif + +cleanup: + if (session != CK_INVALID_HANDLE) { + funcList->C_Logout(session); + funcList->C_CloseSession(session); + } + pkcs11_final(); + return result; +} + +#ifdef TLS_PERSIST_TEST +static const char persistLabel[] = "tls-key-mat-persist"; + +static CK_RV open_user_session(CK_SESSION_HANDLE* session) +{ + CK_RV ret; + + ret = pkcs11_init(); + if (ret == CKR_OK) { + ret = funcList->C_OpenSession(slot, + CKF_SERIAL_SESSION | CKF_RW_SESSION, NULL, NULL, session); + } + if (ret == CKR_OK) + ret = funcList->C_Login(*session, CKU_USER, userPin, userPinLen); + return ret; +} + +/* Derive TLS key material into token keys whose template disallows signing. */ +static CK_RV derive_token_key_mat(CK_SESSION_HANDLE session) +{ + CK_RV ret; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_TLS12_KEY_MAT_PARAMS params; + CK_SSL3_KEY_MAT_OUT out; + byte master[48]; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_SIGN, &ckFalse, sizeof(ckFalse) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, (void*)persistLabel, sizeof(persistLabel) - 1 }, + }; + + XMEMSET(master, 0x5c, sizeof(master)); + ret = create_secret_base(session, CKK_GENERIC_SECRET, master, + sizeof(master), &base); + if (ret == CKR_OK) { + tls_key_mat_init(¶ms, &out, 256, 128, 0); + ret = tls_key_mat_derive(session, base, ¶ms, tmpl, + sizeof(tmpl) / sizeof(*tmpl)); + } + return ret; +} + +/* The MAC key usage attributes are part of the stored token object even when + * the process ends without C_Finalize. */ +static int test_token_mac_key_usage_persist(void) +{ + CK_RV ret; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE objs[8]; + CK_ULONG found = 0; + CK_ULONG i; + CK_ULONG macKeys = 0; + CK_BBOOL sign; + CK_BBOOL encrypt; + CK_ATTRIBUTE findTmpl[] = { + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, (void*)persistLabel, sizeof(persistLabel) - 1 }, + }; + CK_ATTRIBUTE usageTmpl[] = { + { CKA_SIGN, &sign, sizeof(CK_BBOOL) }, + { CKA_ENCRYPT, &encrypt, sizeof(CK_BBOOL) }, + }; + pid_t pid; + int status = 0; + int result = 0; + + printf("\n=== Testing TLS MAC key token persistence ===\n"); + + pid = fork(); + if (pid == 0) { + /* Exit without C_Finalize so only stores made by the derive count. */ + ret = open_user_session(&session); + if (ret == CKR_OK) + ret = derive_token_key_mat(session); + _exit(ret == CKR_OK ? 0 : 1); + } + if (pid < 0) + ret = CKR_GENERAL_ERROR; + else if (waitpid(pid, &status, 0) != pid || !WIFEXITED(status) || + WEXITSTATUS(status) != 0) + ret = CKR_GENERAL_ERROR; + else + ret = CKR_OK; + CHECK_CKR(ret, "derive token key material in child process"); + + ret = open_user_session(&session); + CHECK_CKR(ret, "reopen token"); + + ret = funcList->C_FindObjectsInit(session, findTmpl, + sizeof(findTmpl) / sizeof(*findTmpl)); + CHECK_CKR(ret, "C_FindObjectsInit"); + ret = funcList->C_FindObjects(session, objs, + sizeof(objs) / sizeof(*objs), &found); + if (funcList->C_FindObjectsFinal(session) != CKR_OK && ret == CKR_OK) + ret = CKR_GENERAL_ERROR; + if (ret == CKR_OK && found != 4) + ret = CKR_GENERAL_ERROR; + CHECK_CKR(ret, "find derived token keys"); + + for (i = 0; i < found && ret == CKR_OK; i++) { + ret = funcList->C_GetAttributeValue(session, objs[i], usageTmpl, + sizeof(usageTmpl) / sizeof(*usageTmpl)); + if (ret == CKR_OK && sign == CK_TRUE && encrypt == CK_FALSE) + macKeys++; + } + CHECK_CKR(ret, "read derived key usage"); + if (macKeys != 2) { + fprintf(stderr, "FAIL: reloaded MAC keys with MAC usage: %lu, " + "expected 2\n", (unsigned long)macKeys); + test_failed++; + result = -1; + goto cleanup; + } + printf("PASS: reloaded MAC keys keep MAC usage\n"); + test_passed++; + +cleanup: + if (session != CK_INVALID_HANDLE) { + funcList->C_Logout(session); + funcList->C_CloseSession(session); + } + pkcs11_final(); + return result; +} +#endif /* TLS_PERSIST_TEST */ + +static void print_results(void) +{ + printf("\n=== Test Results ===\n"); + printf("Tests passed: %d\n", test_passed); + printf("Tests failed: %d\n", test_failed); + if (test_failed == 0) + printf("ALL TESTS PASSED!\n"); + else + printf("SOME TESTS FAILED!\n"); +} + +int main(int argc, char* argv[]) +{ +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + (void)argc; + (void)argv; + + if (kdf_tls_derive_test() != 0 && test_failed == 0) + test_failed++; +#ifdef TLS_PERSIST_TEST + if (test_token_mac_key_usage_persist() != 0 && test_failed == 0) + test_failed++; +#endif + + print_results(); +#ifndef DERIVE_TESTS + if (test_failed == 0) + return 77; +#endif + return (test_failed == 0) ? 0 : 1; +} diff --git a/tests/key_material_update_test.c b/tests/key_material_update_test.c new file mode 100644 index 00000000..a41092f6 --- /dev/null +++ b/tests/key_material_update_test.c @@ -0,0 +1,963 @@ +/* key_material_update_test.c + * + * Copyright (C) 2006-2025 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Key material and certificate values are fixed once an object exists, and + * what the token persists matches the live object. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#define TEST_DIR "./store/key_material_update_test" + +#if defined(USE_WOLFSSL_MEMORY) && !defined(WOLFSSL_STATIC_MEMORY) && \ + !defined(WOLFSSL_NO_MALLOC) && !defined(WOLFSSL_DEBUG_MEMORY) +#define KMU_ALLOC_HOOK +/* Allocation of exactly this size fails while armed. */ +#define FAIL_ALLOC_SIZE 101 +static int failAllocArmed = 0; + +static void* test_malloc(size_t n) +{ + if (failAllocArmed && n == FAIL_ALLOC_SIZE) + return NULL; + return malloc(n); +} + +static void test_free(void* p) +{ + free(p); +} + +static void* test_realloc(void* p, size_t n) +{ + if (failAllocArmed && n == FAIL_ALLOC_SIZE) + return NULL; + return realloc(p, n); +} +#endif + +static CK_SLOT_ID slot = 0; +static const char* soPin = "password123456"; +static const char* userPin = "wolfpkcs11-test"; +static const char tokenLabel[] = "key-material-update"; + +static CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; +static CK_KEY_TYPE genericType = CKK_GENERIC_SECRET; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; + +static CK_RV token_setup(void) +{ + CK_C_INITIALIZE_ARGS args; + CK_SLOT_ID slotList[16]; + CK_ULONG slotCount = sizeof(slotList) / sizeof(slotList[0]); + CK_SESSION_HANDLE soSession = 0; + unsigned char label[32]; + CK_RV rv; + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + rv = funcList->C_Initialize(&args); + if (rv == CKR_OK) + rv = funcList->C_GetSlotList(CK_TRUE, slotList, &slotCount); + if (rv == CKR_OK && slotCount == 0) + rv = CKR_TOKEN_NOT_PRESENT; + if (rv == CKR_OK) { + slot = slotList[0]; + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, tokenLabel, XSTRLEN(tokenLabel)); + rv = funcList->C_InitToken(slot, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin), label); + } + if (rv == CKR_OK) { + rv = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &soSession); + } + if (rv == CKR_OK) { + rv = funcList->C_Login(soSession, CKU_SO, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin)); + if (rv == CKR_OK) { + rv = funcList->C_InitPIN(soSession, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + funcList->C_Logout(soSession); + } + funcList->C_CloseSession(soSession); + } + return rv; +} + +static CK_RV open_session(CK_SESSION_HANDLE* session, int login) +{ + CK_RV rv; + + rv = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, session); + if (rv == CKR_OK && login) { + rv = funcList->C_Login(*session, CKU_USER, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + } + return rv; +} + +#ifndef WOLFPKCS11_NO_STORE +/* Finalize so token objects are written out, then load them again. */ +static CK_RV reload_token(CK_SESSION_HANDLE* session, int login) +{ + CK_C_INITIALIZE_ARGS args; + CK_RV rv; + + funcList->C_CloseSession(*session); + *session = 0; + funcList->C_Finalize(NULL); + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + rv = funcList->C_Initialize(&args); + if (rv == CKR_OK) + rv = open_session(session, login); + return rv; +} + +static CK_OBJECT_HANDLE find_by_label(CK_SESSION_HANDLE session, + const char* label) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ULONG count = 0; + CK_ATTRIBUTE tmpl[] = { + { CKA_LABEL, (void*)label, (CK_ULONG)XSTRLEN(label) }, + }; + + rv = funcList->C_FindObjectsInit(session, tmpl, 1); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, &obj, 1, &count); + funcList->C_FindObjectsFinal(session); + } + if (rv != CKR_OK || count != 1) + obj = CK_INVALID_HANDLE; + return obj; +} +#endif + +/* Read an attribute value and compare it with the expected bytes. */ +static int attr_equals(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_ATTRIBUTE_TYPE type, const byte* expect, + CK_ULONG expectLen) +{ + CK_RV rv; + byte buf[512]; + CK_ATTRIBUTE attr; + + attr.type = type; + attr.pValue = buf; + attr.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, obj, &attr, 1); + return rv == CKR_OK && attr.ulValueLen == expectLen && + XMEMCMP(buf, expect, expectLen) == 0; +} + +static void destroy_obj(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* obj) +{ + if (*obj != CK_INVALID_HANDLE) { + funcList->C_DestroyObject(session, *obj); + *obj = CK_INVALID_HANDLE; + } +} + +#ifndef NO_RSA +/* A token public key's modulus cannot be replaced and the persisted key + * matches the live key. */ +static void check_token_modulus_fixed(CK_SESSION_HANDLE* session) +{ + CK_RV rv; + CK_OBJECT_HANDLE rsa = CK_INVALID_HANDLE; + static const char rsaLabel[] = "kmu-token-rsa"; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_KEY_TYPE rsaType = CKK_RSA; + byte newMod[sizeof(rsa_2048_modulus)]; + byte liveMod[sizeof(rsa_2048_modulus) + 16]; + CK_ATTRIBUTE liveAttr = { CKA_MODULUS, liveMod, sizeof(liveMod) }; + CK_ATTRIBUTE rsaTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, (void*)rsaLabel, sizeof(rsaLabel) - 1 }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE setMod[] = { { CKA_MODULUS, newMod, sizeof(newMod) } }; + + XMEMCPY(newMod, rsa_2048_modulus, sizeof(newMod)); + newMod[sizeof(newMod) - 1] ^= 0x02; + rv = funcList->C_CreateObject(*session, rsaTmpl, + sizeof(rsaTmpl) / sizeof(*rsaTmpl), &rsa); + CHECK_RV(rv, "create token RSA public key", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_SetAttributeValue(*session, rsa, setMod, 1); + CHECK_RV(rv, "set modulus on existing key", CKR_ATTRIBUTE_READ_ONLY); + rv = funcList->C_GetAttributeValue(*session, rsa, &liveAttr, 1); + CHECK_RV(rv, "read live modulus", CKR_OK); + CHECK_TRUE(liveAttr.ulValueLen == sizeof(rsa_2048_modulus) && + XMEMCMP(liveMod, rsa_2048_modulus, + sizeof(rsa_2048_modulus)) == 0, + "live modulus unchanged"); + +#ifndef WOLFPKCS11_NO_STORE + /* Token objects are decoded at login. */ + rv = reload_token(session, 1); + CHECK_RV(rv, "reload token", CKR_OK); + if (rv != CKR_OK) + return; + rsa = find_by_label(*session, rsaLabel); + CHECK_TRUE(rsa != CK_INVALID_HANDLE && + attr_equals(*session, rsa, CKA_MODULUS, liveMod, + liveAttr.ulValueLen), + "persisted modulus matches live modulus"); +#endif + destroy_obj(*session, &rsa); +} +#endif + +/* A token secret key's value cannot be replaced directly or through a copy + * template, and C_CopyObject still carries the key material over. */ +static void check_token_secret_fixed(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE copy = CK_INVALID_HANDLE; + static const char keyLabel[] = "kmu-token-key"; + static const char copyLabel[] = "kmu-token-key-copy"; + byte origVal[16]; + byte newVal[16]; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, (void*)keyLabel, sizeof(keyLabel) - 1 }, + { CKA_VALUE, origVal, sizeof(origVal) }, + }; + CK_ATTRIBUTE setValue[] = { { CKA_VALUE, newVal, sizeof(newVal) } }; + CK_ATTRIBUTE copyWithValue[] = { + { CKA_LABEL, (void*)copyLabel, sizeof(copyLabel) - 1 }, + { CKA_VALUE, newVal, sizeof(newVal) }, + }; + CK_ATTRIBUTE copyTmpl[] = { + { CKA_LABEL, (void*)copyLabel, sizeof(copyLabel) - 1 }, + }; + + XMEMSET(origVal, 0x11, sizeof(origVal)); + XMEMSET(newVal, 0x22, sizeof(newVal)); + + rv = funcList->C_CreateObject(session, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl), &key); + CHECK_RV(rv, "create token secret key", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_SetAttributeValue(session, key, setValue, 1); + CHECK_RV(rv, "set key value on existing key", CKR_ATTRIBUTE_READ_ONLY); + CHECK_TRUE(attr_equals(session, key, CKA_VALUE, origVal, sizeof(origVal)), + "live key value unchanged"); + + rv = funcList->C_CopyObject(session, key, copyWithValue, 2, ©); + CHECK_RV(rv, "copy with a new key value", CKR_ATTRIBUTE_READ_ONLY); + destroy_obj(session, ©); + + rv = funcList->C_CopyObject(session, key, copyTmpl, 1, ©); + CHECK_RV(rv, "copy token secret key", CKR_OK); + CHECK_TRUE(attr_equals(session, copy, CKA_VALUE, origVal, sizeof(origVal)), + "copy carries the key value"); + + destroy_obj(session, ©); + destroy_obj(session, &key); +} + +static void test_token_key_value_fixed(CK_SESSION_HANDLE* session) +{ + CK_RV rv; + +#ifndef NO_RSA + check_token_modulus_fixed(session); +#endif + rv = funcList->C_Login(*session, CKU_USER, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + if (rv == CKR_USER_ALREADY_LOGGED_IN) + rv = CKR_OK; + CHECK_RV(rv, "login user", CKR_OK); + if (rv == CKR_OK) + check_token_secret_fixed(*session); +} + +#if defined(HAVE_ECC) && !defined(NO_AES) && !defined(WOLFPKCS11_NSS) +/* A derive hands the new key to WP11_Object_SetSecretKey directly, so a + * length the key type cannot take is rejected by the setter itself. */ +static void test_derive_invalid_secret_length(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE ecType = CKK_EC; + CK_KEY_TYPE aesType = CKK_AES; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + CK_ECDH1_DERIVE_PARAMS params; + CK_MECHANISM mech = { CKM_ECDH1_DERIVE, NULL, 0 }; + CK_ULONG badLen = 17; + CK_ATTRIBUTE baseTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + CK_ATTRIBUTE deriveTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE_LEN, &badLen, sizeof(badLen) }, + }; + + XMEMSET(¶ms, 0, sizeof(params)); + params.kdf = CKD_NULL; + params.pPublicData = ecc_p256_point; + params.ulPublicDataLen = sizeof(ecc_p256_point); + mech.pParameter = ¶ms; + mech.ulParameterLen = sizeof(params); + + rv = funcList->C_CreateObject(session, baseTmpl, + sizeof(baseTmpl) / sizeof(*baseTmpl), &base); + CHECK_RV(rv, "create EC base key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_DeriveKey(session, &mech, base, deriveTmpl, + sizeof(deriveTmpl) / sizeof(*deriveTmpl), + &derived); + CHECK_RV(rv, "derive an AES key with an invalid length", + CKR_ATTRIBUTE_VALUE_INVALID); + if (rv == CKR_OK) + destroy_obj(session, &derived); + } + destroy_obj(session, &base); +} +#endif + +/* The read-only gate in SetAttributeValue rejects a secret key update before + * WP11_Object_SetSecretKey runs, so the existing key is left untouched. */ +static void test_rejected_secret_update_keeps_key(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + byte origVal[16]; + byte newVal[16]; + CK_ULONG shortLen = 8; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, origVal, sizeof(origVal) }, + }; + CK_ATTRIBUTE badUpdate[] = { + { CKA_VALUE_LEN, &shortLen, sizeof(shortLen) }, + { CKA_VALUE, newVal, sizeof(newVal) }, + }; +#ifndef NO_AES + CK_KEY_TYPE aesType = CKK_AES; + CK_ULONG badAesLen = 17; + CK_ATTRIBUTE badAesLenTmpl[] = { + { CKA_VALUE_LEN, &badAesLen, sizeof(badAesLen) }, + }; +#endif + + XMEMSET(origVal, 0x33, sizeof(origVal)); + XMEMSET(newVal, 0x44, sizeof(newVal)); + + rv = funcList->C_CreateObject(session, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl), &key); + CHECK_RV(rv, "create secret key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, key, badUpdate, 2); + CHECK_RV(rv, "inconsistent key update rejected", + CKR_ATTRIBUTE_READ_ONLY); + CHECK_TRUE(attr_equals(session, key, CKA_VALUE, origVal, + sizeof(origVal)), + "key kept after rejected update"); + } + destroy_obj(session, &key); + +#ifndef NO_AES + keyTmpl[1].pValue = &aesType; + rv = funcList->C_CreateObject(session, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl), &key); + CHECK_RV(rv, "create AES key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, key, badAesLenTmpl, 1); + CHECK_RV(rv, "invalid AES key length rejected", + CKR_ATTRIBUTE_READ_ONLY); + CHECK_TRUE(attr_equals(session, key, CKA_VALUE, origVal, + sizeof(origVal)), + "AES key kept after rejected length"); + } + destroy_obj(session, &key); +#endif +} + +#ifndef NO_DH +/* A DH private value cannot be replaced once the key exists. */ +static void test_dh_private_value_fixed(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE dhType = CKK_DH; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &dhType, sizeof(dhType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv) }, + }; + CK_ATTRIBUTE shorterValue[] = { + { CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv) / 2 }, + }; + + rv = funcList->C_CreateObject(session, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl), &key); + CHECK_RV(rv, "create DH private key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, key, shorterValue, 1); + CHECK_RV(rv, "set shorter DH private value", CKR_ATTRIBUTE_READ_ONLY); + CHECK_TRUE(attr_equals(session, key, CKA_VALUE, dh_2048_priv, + sizeof(dh_2048_priv)), + "DH private value unchanged"); + } + destroy_obj(session, &key); +} +#endif + +#if !defined(NO_RSA) || defined(HAVE_ECC) +/* Sign with the private key and verify with the public key. */ +static CK_RV sign_verify(CK_SESSION_HANDLE session, CK_MECHANISM_TYPE type, + CK_OBJECT_HANDLE priv, CK_OBJECT_HANDLE pub) +{ + CK_RV rv; + CK_MECHANISM mech = { 0, NULL, 0 }; + byte hash[32]; + byte sig[512]; + CK_ULONG sigLen = sizeof(sig); + + mech.mechanism = type; + XMEMSET(hash, 0x5a, sizeof(hash)); + rv = funcList->C_SignInit(session, &mech, priv); + if (rv == CKR_OK) + rv = funcList->C_Sign(session, hash, sizeof(hash), sig, &sigLen); + if (rv == CKR_OK) + rv = funcList->C_VerifyInit(session, &mech, pub); + if (rv == CKR_OK) + rv = funcList->C_Verify(session, hash, sizeof(hash), sig, sigLen); + return rv; +} +#endif + +/* Private components of an existing asymmetric key cannot be replaced, and + * the key still works after the rejected update. */ +static void test_asym_private_fixed(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + byte other[32]; +#ifndef NO_RSA + CK_KEY_TYPE rsaType = CKK_RSA; + CK_ATTRIBUTE rsaPrivTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE rsaPubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE setPrivExp[] = { + { CKA_PRIVATE_EXPONENT, other, sizeof(other) }, + }; +#endif +#ifdef HAVE_ECC + CK_KEY_TYPE ecType = CKK_EC; + CK_ATTRIBUTE ecPrivTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + CK_ATTRIBUTE ecPubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + }; + CK_ATTRIBUTE setEcValue[] = { { CKA_VALUE, other, sizeof(other) } }; +#endif + + XMEMSET(other, 0x07, sizeof(other)); +#ifndef NO_RSA + rv = funcList->C_CreateObject(session, rsaPrivTmpl, + sizeof(rsaPrivTmpl) / sizeof(*rsaPrivTmpl), + &priv); + CHECK_RV(rv, "create RSA private key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, rsaPubTmpl, + sizeof(rsaPubTmpl) / sizeof(*rsaPubTmpl), + &pub); + CHECK_RV(rv, "create RSA public key", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, priv, setPrivExp, 1); + CHECK_RV(rv, "set RSA private exponent", CKR_ATTRIBUTE_READ_ONLY); + rv = sign_verify(session, CKM_RSA_PKCS, priv, pub); + CHECK_RV(rv, "RSA key still signs after rejected update", CKR_OK); + } + destroy_obj(session, &pub); + destroy_obj(session, &priv); +#endif +#ifdef HAVE_ECC + rv = funcList->C_CreateObject(session, ecPrivTmpl, + sizeof(ecPrivTmpl) / sizeof(*ecPrivTmpl), + &priv); + CHECK_RV(rv, "create EC private key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, ecPubTmpl, + sizeof(ecPubTmpl) / sizeof(*ecPubTmpl), + &pub); + CHECK_RV(rv, "create EC public key", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, priv, setEcValue, 1); + CHECK_RV(rv, "set EC private value", CKR_ATTRIBUTE_READ_ONLY); + rv = sign_verify(session, CKM_ECDSA, priv, pub); + CHECK_RV(rv, "EC key still signs after rejected update", CKR_OK); + } + destroy_obj(session, &pub); + destroy_obj(session, &priv); +#endif + (void)rv; + (void)priv; + (void)pub; + (void)other; + (void)privClass; + (void)pubClass; +} + +#ifdef HAVE_ECC +/* EC key material is only accepted together with the curve, and an existing + * key's point cannot be replaced. */ +static void test_ec_point_needs_curve(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE ecType = CKK_EC; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + }; + CK_ATTRIBUTE privNoCurve[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + CK_ATTRIBUTE setPoint[] = { + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + }; + + /* Leave out CKA_EC_PARAMS. */ + rv = funcList->C_CreateObject(session, pubTmpl, 4, &key); + CHECK_RV(rv, "EC public key without curve rejected", + CKR_TEMPLATE_INCOMPLETE); + destroy_obj(session, &key); + rv = funcList->C_CreateObject(session, privNoCurve, + sizeof(privNoCurve) / sizeof(*privNoCurve), + &key); + CHECK_RV(rv, "EC private key without curve rejected", + CKR_TEMPLATE_INCOMPLETE); + destroy_obj(session, &key); + + rv = funcList->C_CreateObject(session, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), &key); + CHECK_RV(rv, "create EC public key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, key, setPoint, 1); + CHECK_RV(rv, "set EC point on existing key", CKR_ATTRIBUTE_READ_ONLY); + CHECK_TRUE(attr_equals(session, key, CKA_EC_PARAMS, ecc_p256_params, + sizeof(ecc_p256_params)), + "EC curve unchanged"); + } + destroy_obj(session, &key); +} +#endif + +#if !defined(NO_RSA) && defined(WOLFSSL_KEY_GEN) +/* A modulus size the key generator cannot honour is rejected rather than + * narrowed to a different size. */ +static void test_rsa_modulus_bits_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM mech = { CKM_RSA_PKCS_KEY_PAIR_GEN, NULL, 0 }; + CK_ULONG bits; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_MODULUS_BITS, &bits, sizeof(bits) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + }; + + /* Only a 64-bit CK_ULONG can carry a size whose low 32 bits look valid. */ + if (sizeof(CK_ULONG) <= 4) + return; + bits = ((CK_ULONG)1 << 16 << 16) + 2048; + rv = funcList->C_GenerateKeyPair(session, &mech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), + privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), + &pub, &priv); + CHECK_RV(rv, "oversized modulus bits rejected", CKR_ATTRIBUTE_VALUE_INVALID); + if (rv == CKR_OK) { + destroy_obj(session, &pub); + destroy_obj(session, &priv); + } +} +#endif + +/* Restating the certificate type keeps the certificate value, and the type + * cannot be changed. */ +static void test_cert_type_update_keeps_value(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS certClass = CKO_CERTIFICATE; + CK_CERTIFICATE_TYPE certType = CKC_X_509; + CK_OBJECT_HANDLE cert = CK_INVALID_HANDLE; + byte certVal[48]; + CK_ATTRIBUTE certTmpl[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, certVal, sizeof(certVal) }, + }; + CK_ATTRIBUTE setType[] = { + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + }; + CK_CERTIFICATE_TYPE otherType = CKC_X_509_ATTR_CERT; + CK_ATTRIBUTE setOtherType[] = { + { CKA_CERTIFICATE_TYPE, &otherType, sizeof(otherType) }, + }; + + XMEMSET(certVal, 0x30, sizeof(certVal)); + rv = funcList->C_CreateObject(session, certTmpl, + sizeof(certTmpl) / sizeof(*certTmpl), &cert); + CHECK_RV(rv, "create certificate", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, cert, setType, 1); + CHECK_RV(rv, "set certificate type", CKR_OK); + CHECK_TRUE(attr_equals(session, cert, CKA_VALUE, certVal, + sizeof(certVal)), + "certificate value kept after type update"); + rv = funcList->C_SetAttributeValue(session, cert, setOtherType, 1); + CHECK_RV(rv, "change certificate type", CKR_ATTRIBUTE_READ_ONLY); + } + destroy_obj(session, &cert); +} + +#ifdef KMU_ALLOC_HOOK +/* A certificate value that cannot be allocated at creation leaves no object, + * which reaches WP11_Object_SetCert directly. Replacing the value of an + * existing certificate stops at the read-only gate before the setter runs. */ +static void test_cert_value_failure_consistent(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS certClass = CKO_CERTIFICATE; + CK_CERTIFICATE_TYPE certType = CKC_X_509; + CK_OBJECT_HANDLE cert = CK_INVALID_HANDLE; + byte certVal[FAIL_ALLOC_SIZE]; + byte newVal[FAIL_ALLOC_SIZE]; + CK_ATTRIBUTE certTmpl[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, certVal, sizeof(certVal) }, + }; + CK_ATTRIBUTE setValue[] = { { CKA_VALUE, newVal, sizeof(newVal) } }; + + XMEMSET(certVal, 0x30, sizeof(certVal)); + XMEMSET(newVal, 0x31, sizeof(newVal)); + + failAllocArmed = 1; + rv = funcList->C_CreateObject(session, certTmpl, + sizeof(certTmpl) / sizeof(*certTmpl), &cert); + failAllocArmed = 0; + CHECK_RV(rv, "certificate create out of memory", CKR_DEVICE_MEMORY); + destroy_obj(session, &cert); + + rv = funcList->C_CreateObject(session, certTmpl, + sizeof(certTmpl) / sizeof(*certTmpl), &cert); + CHECK_RV(rv, "create certificate", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, cert, setValue, 1); + CHECK_RV(rv, "certificate value replacement stops at the gate", + CKR_ATTRIBUTE_READ_ONLY); + CHECK_TRUE(attr_equals(session, cert, CKA_VALUE, certVal, + sizeof(certVal)), + "certificate value kept"); + } + destroy_obj(session, &cert); +} +#endif + +#ifndef WOLFPKCS11_NO_STORE +/* The read-only gate rejects replacing a loaded certificate's value before + * WP11_Object_SetCert can touch the storage buffer, a copy of it is rejected, + * and the value persists unchanged. */ +static void test_loaded_cert_value_fixed(CK_SESSION_HANDLE* session) +{ + CK_RV rv; + CK_OBJECT_CLASS certClass = CKO_CERTIFICATE; + CK_CERTIFICATE_TYPE certType = CKC_X_509; + CK_OBJECT_HANDLE cert = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE copy = CK_INVALID_HANDLE; + static const char certLabel[] = "kmu-token-cert"; + byte certVal[64]; + byte newVal[80]; + CK_ATTRIBUTE certTmpl[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, (void*)certLabel, sizeof(certLabel) - 1 }, + { CKA_VALUE, certVal, sizeof(certVal) }, + }; + CK_ATTRIBUTE setValue[] = { { CKA_VALUE, newVal, sizeof(newVal) } }; + + XMEMSET(certVal, 0x30, sizeof(certVal)); + XMEMSET(newVal, 0x31, sizeof(newVal)); + rv = funcList->C_CreateObject(*session, certTmpl, + sizeof(certTmpl) / sizeof(*certTmpl), &cert); + CHECK_RV(rv, "create token certificate", CKR_OK); + if (rv != CKR_OK) + return; + + rv = reload_token(session, 1); + CHECK_RV(rv, "reload token with certificate", CKR_OK); + if (rv != CKR_OK) + return; + cert = find_by_label(*session, certLabel); + CHECK_TRUE(cert != CK_INVALID_HANDLE, "find loaded certificate"); + if (cert == CK_INVALID_HANDLE) + return; + + rv = funcList->C_SetAttributeValue(*session, cert, setValue, 1); + CHECK_RV(rv, "replace loaded certificate value", CKR_ATTRIBUTE_READ_ONLY); + rv = funcList->C_CopyObject(*session, cert, NULL, 0, ©); + CHECK_TRUE(rv != CKR_OK, "copy of certificate rejected"); + destroy_obj(*session, ©); + CHECK_TRUE(attr_equals(*session, cert, CKA_VALUE, certVal, + sizeof(certVal)), + "loaded certificate value kept"); + + rv = reload_token(session, 1); + CHECK_RV(rv, "reload token again", CKR_OK); + if (rv != CKR_OK) + return; + cert = find_by_label(*session, certLabel); + CHECK_TRUE(cert != CK_INVALID_HANDLE && + attr_equals(*session, cert, CKA_VALUE, certVal, + sizeof(certVal)), + "persisted certificate value kept"); + destroy_obj(*session, &cert); +} +#endif + +#ifndef NO_DH +/* Domain parameters larger than any supported DH key are rejected. */ +static void test_dh_param_size_range(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_KEY_TYPE dhType = CKK_DH; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + static byte bigPrime[2049]; + byte base[1] = { 0x02 }; + CK_ATTRIBUTE keyTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &dhType, sizeof(dhType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIME, bigPrime, sizeof(bigPrime) }, + { CKA_BASE, base, sizeof(base) }, + { CKA_VALUE, base, sizeof(base) }, + }; + + XMEMSET(bigPrime, 0xff, sizeof(bigPrime)); + rv = funcList->C_CreateObject(session, keyTmpl, + sizeof(keyTmpl) / sizeof(*keyTmpl), &key); + CHECK_RV(rv, "oversized DH prime rejected", CKR_ATTRIBUTE_VALUE_INVALID); + if (rv == CKR_OK) + destroy_obj(session, &key); +} +#endif + +static int run_test(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = 0; + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + + rv = token_setup(); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) { + rv = open_session(&session, 0); + CHECK_RV(rv, "open session", CKR_OK); + } + if (rv == CKR_OK) { + test_token_key_value_fixed(&session); +#if defined(HAVE_ECC) && !defined(NO_AES) && !defined(WOLFPKCS11_NSS) + test_derive_invalid_secret_length(session); +#endif + test_rejected_secret_update_keeps_key(session); +#ifndef NO_DH + test_dh_private_value_fixed(session); +#endif + test_asym_private_fixed(session); +#ifdef HAVE_ECC + test_ec_point_needs_curve(session); +#endif +#if !defined(NO_RSA) && defined(WOLFSSL_KEY_GEN) + test_rsa_modulus_bits_range(session); +#endif + test_cert_type_update_keeps_value(session); +#ifdef KMU_ALLOC_HOOK + test_cert_value_failure_consistent(session); +#endif +#ifndef WOLFPKCS11_NO_STORE + test_loaded_cert_value_fixed(&session); +#endif +#ifndef NO_DH + test_dh_param_size_range(session); +#endif + } + + if (session != 0) { + funcList->C_Logout(session); + funcList->C_CloseSession(session); + } + funcList->C_Finalize(NULL); + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + + printf("=== wolfPKCS11 key material update test ===\n"); +#ifdef KMU_ALLOC_HOOK + if (wolfSSL_SetAllocators(test_malloc, test_free, test_realloc) != 0) { + fprintf(stderr, "FAIL: wolfSSL_SetAllocators\n"); + return 1; + } +#endif + run_test(); + return pkcs11_test_summary(); +} diff --git a/tests/login_pin_state_test.c b/tests/login_pin_state_test.c index 9388178a..92dc169b 100644 --- a/tests/login_pin_state_test.c +++ b/tests/login_pin_state_test.c @@ -48,7 +48,11 @@ #include "testdata.h" #include "pkcs11_test_util.h" -#ifndef SINGLE_THREADED +#if !defined(SINGLE_THREADED) && !defined(WOLFPKCS11_SINGLE_THREADED) + #define LOGIN_TEST_THREADS +#endif + +#ifdef LOGIN_TEST_THREADS #include #include #endif @@ -361,7 +365,7 @@ static void logout_protects_all_objects_test(void) } #endif -#ifndef SINGLE_THREADED +#ifdef LOGIN_TEST_THREADS #define EMPTY_PIN_ROUNDS 8 static pthread_mutex_t infoMutex = PTHREAD_MUTEX_INITIALIZER; @@ -510,7 +514,7 @@ static void set_pin_lockout(CK_USER_TYPE type, const char* pin, int maxFails, } #endif -#if !defined(SINGLE_THREADED) && !defined(WOLFPKCS11_NO_TIME) +#if defined(LOGIN_TEST_THREADS) && !defined(WOLFPKCS11_NO_TIME) #define LOCKOUT_THREADS 6 #define LOCKOUT_ROUNDS 4 @@ -636,7 +640,7 @@ static void concurrent_login_lockout_test(void) } #endif -#ifndef SINGLE_THREADED +#ifdef LOGIN_TEST_THREADS #define CLOSE_OPEN_ROUNDS 10 static pthread_mutex_t openMutex = PTHREAD_MUTEX_INITIALIZER; @@ -727,7 +731,7 @@ static void close_last_session_race_test(void) } #endif -#ifndef SINGLE_THREADED +#ifdef LOGIN_TEST_THREADS #define CHURN_ROUNDS 200 typedef struct churn_ctx { @@ -808,7 +812,7 @@ static void find_during_object_churn_test(void) } #endif -#if !defined(SINGLE_THREADED) && !defined(WOLFPKCS11_NO_STORE) && \ +#if defined(LOGIN_TEST_THREADS) && !defined(WOLFPKCS11_NO_STORE) && \ !defined(NO_AES) #define FIND_LOGIN_ROUNDS 3 @@ -1043,7 +1047,7 @@ static void logout_drops_private_session_objects_test(void) } #endif -#if defined(DEBUG_WOLFPKCS11) && !defined(SINGLE_THREADED) && \ +#if defined(DEBUG_WOLFPKCS11) && defined(LOGIN_TEST_THREADS) && \ !defined(WOLFPKCS11_NO_STORE) && !defined(WOLFPKCS11_NSS) #define FIND_HOOK_WAIT_MS 200 @@ -1151,7 +1155,7 @@ static void find_with_racing_logout_test(void) } #endif -#if defined(DEBUG_WOLFPKCS11) && !defined(SINGLE_THREADED) && \ +#if defined(DEBUG_WOLFPKCS11) && defined(LOGIN_TEST_THREADS) && \ !defined(WOLFPKCS11_NO_TIME) extern void WP11_Slot_SetPinCheckHook(void (*hook)(void)); @@ -1287,33 +1291,33 @@ static int run_test(void) if (rv != CKR_OK) return -1; -#ifndef SINGLE_THREADED +#ifdef LOGIN_TEST_THREADS empty_pin_change_requires_login_test(); #endif #ifdef LOGIN_TEST_FILE_STORE failed_login_leaves_no_token_key_test(); #endif -#ifndef SINGLE_THREADED +#ifdef LOGIN_TEST_THREADS close_last_session_race_test(); #endif #ifndef WOLFPKCS11_NSS logout_drops_private_session_objects_test(); #endif -#if !defined(SINGLE_THREADED) && !defined(WOLFPKCS11_NO_TIME) +#if defined(LOGIN_TEST_THREADS) && !defined(WOLFPKCS11_NO_TIME) concurrent_login_lockout_test(); #endif -#ifndef SINGLE_THREADED -#if defined(DEBUG_WOLFPKCS11) && !defined(SINGLE_THREADED) && \ +#if defined(DEBUG_WOLFPKCS11) && defined(LOGIN_TEST_THREADS) && \ !defined(WOLFPKCS11_NO_TIME) guess_in_flight_test(); #endif +#ifdef LOGIN_TEST_THREADS find_during_object_churn_test(); #endif -#if !defined(SINGLE_THREADED) && !defined(WOLFPKCS11_NO_STORE) && \ +#if defined(LOGIN_TEST_THREADS) && !defined(WOLFPKCS11_NO_STORE) && \ !defined(NO_AES) find_during_login_changes_test(); #endif -#if defined(DEBUG_WOLFPKCS11) && !defined(SINGLE_THREADED) && \ +#if defined(DEBUG_WOLFPKCS11) && defined(LOGIN_TEST_THREADS) && \ !defined(WOLFPKCS11_NO_STORE) && !defined(WOLFPKCS11_NSS) find_with_racing_logout_test(); #endif diff --git a/tests/object_create_getattr_test.c b/tests/object_create_getattr_test.c new file mode 100644 index 00000000..ba582a00 --- /dev/null +++ b/tests/object_create_getattr_test.c @@ -0,0 +1,1040 @@ +/* object_create_getattr_test.c + * + * Copyright (C) 2006-2025 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Objects created with C_CreateObject carry the attributes PKCS#11 requires, + * and attributes read back with C_GetAttributeValue follow the PKCS#11 + * length and error conventions. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#define TEST_DIR "./store/object_create_getattr_test" + +static CK_BBOOL ckFalse = CK_FALSE; +static const byte certValue[] = { 0x30, 0x82, 0x01, 0x00 }; +static const byte certSubject[] = { 0x30, 0x00 }; +static const char certUrl[] = "http://example.com/cert.der"; +#ifdef HAVE_ECC +static const byte keySubject[] = { + 0x30, 0x0f, 0x31, 0x0d, 0x30, 0x0b, 0x06, 0x03, + 0x55, 0x04, 0x03, 0x0c, 0x04, 0x74, 0x65, 0x73, 0x74 +}; +static const byte keySerial[] = { 0x02, 0x01, 0x01 }; +#endif +static const byte dataValue[] = "object-create-data"; +static const byte dataLabel[] = "object-create-data-label"; +static const byte secretValue[] = { + 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, + 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff +}; + +static void destroy_obj(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* obj) +{ + if (*obj != CK_INVALID_HANDLE) { + funcList->C_DestroyObject(session, *obj); + *obj = CK_INVALID_HANDLE; + } +} + +/* Create an object, expecting rv, and destroy anything that was created. */ +static void expect_create(CK_SESSION_HANDLE session, CK_ATTRIBUTE* tmpl, + CK_ULONG cnt, CK_RV expected, const char* name) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + + rv = funcList->C_CreateObject(session, tmpl, cnt, &obj); + CHECK_RV(rv, name, expected); + if (rv == CKR_OK) + destroy_obj(session, &obj); +} + +/* A certificate object needs its encoding. */ +static void test_cert_requires_value(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS certClass = CKO_CERTIFICATE; + CK_CERTIFICATE_TYPE certType = CKC_X_509; + CK_CERTIFICATE_TYPE attrCertType = CKC_X_509_ATTR_CERT; + CK_ATTRIBUTE noValue[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SUBJECT, (void*)certSubject, sizeof(certSubject) }, + }; + CK_ATTRIBUTE attrCertNoValue[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &attrCertType, sizeof(attrCertType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE emptyValue[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SUBJECT, (void*)certSubject, sizeof(certSubject) }, + { CKA_VALUE, (void*)certValue, 0 }, + }; + CK_ATTRIBUTE urlOnly[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SUBJECT, (void*)certSubject, sizeof(certSubject) }, + { CKA_VALUE, (void*)certValue, 0 }, + { CKA_URL, (void*)certUrl, sizeof(certUrl) - 1 }, + }; + CK_ATTRIBUTE complete[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SUBJECT, (void*)certSubject, sizeof(certSubject) }, + { CKA_VALUE, (void*)certValue, sizeof(certValue) }, + }; + + expect_create(session, noValue, sizeof(noValue) / sizeof(*noValue), + CKR_TEMPLATE_INCOMPLETE, "certificate without CKA_VALUE"); + expect_create(session, attrCertNoValue, + sizeof(attrCertNoValue) / sizeof(*attrCertNoValue), + CKR_TEMPLATE_INCOMPLETE, + "attribute certificate without CKA_VALUE"); + expect_create(session, emptyValue, + sizeof(emptyValue) / sizeof(*emptyValue), + CKR_ATTRIBUTE_VALUE_INVALID, + "certificate with empty CKA_VALUE"); + expect_create(session, urlOnly, sizeof(urlOnly) / sizeof(*urlOnly), + CKR_ATTRIBUTE_VALUE_INVALID, + "certificate located only by CKA_URL"); + expect_create(session, complete, sizeof(complete) / sizeof(*complete), + CKR_OK, "complete certificate"); +} + +/* A secret key object needs its key value. */ +static void test_secret_key_requires_value(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; + CK_KEY_TYPE genericType = CKK_GENERIC_SECRET; + CK_ULONG valueLen = sizeof(secretValue); + CK_ATTRIBUTE noValue[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE lenOnly[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE_LEN, &valueLen, sizeof(valueLen) }, + }; + CK_ATTRIBUTE emptyValue[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)secretValue, 0 }, + }; + CK_ATTRIBUTE complete[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)secretValue, sizeof(secretValue) }, + }; +#ifndef NO_AES + CK_KEY_TYPE aesType = CKK_AES; + CK_ATTRIBUTE aesNoValue[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; +#endif + + expect_create(session, noValue, sizeof(noValue) / sizeof(*noValue), + CKR_TEMPLATE_INCOMPLETE, "secret key without CKA_VALUE"); + expect_create(session, lenOnly, sizeof(lenOnly) / sizeof(*lenOnly), + CKR_TEMPLATE_INCOMPLETE, + "secret key with only CKA_VALUE_LEN"); + expect_create(session, emptyValue, + sizeof(emptyValue) / sizeof(*emptyValue), + CKR_ATTRIBUTE_VALUE_INVALID, + "secret key with empty CKA_VALUE"); +#ifndef NO_AES + expect_create(session, aesNoValue, + sizeof(aesNoValue) / sizeof(*aesNoValue), + CKR_TEMPLATE_INCOMPLETE, "AES key without CKA_VALUE"); +#endif + expect_create(session, complete, sizeof(complete) / sizeof(*complete), + CKR_OK, "complete secret key"); +} + +/* An attribute count beyond what the library can index is rejected. */ +static void test_create_count_range(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; + CK_KEY_TYPE genericType = CKK_GENERIC_SECRET; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)secretValue, sizeof(secretValue) }, + }; + + expect_create(session, tmpl, (CK_ULONG)INT_MAX + 1, CKR_ARGUMENTS_BAD, + "create with oversized attribute count"); +} + +#if !defined(NO_RSA) || defined(HAVE_ECC) +/* A secret key object needs a secret key type, for session and token objects, + * and nothing is stored when it does not. */ +static void test_key_type_matches_class(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; + CK_BBOOL onToken = CK_FALSE; + CK_OBJECT_HANDLE found[2]; + CK_ULONG foundCnt = 0; + int i; +#ifndef NO_RSA + CK_KEY_TYPE rsaType = CKK_RSA; + CK_ATTRIBUTE rsaSecret[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &onToken, sizeof(onToken) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE rsaFind[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + }; +#endif +#ifdef HAVE_ECC + CK_KEY_TYPE ecType = CKK_EC; + CK_ATTRIBUTE ecSecret[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &onToken, sizeof(onToken) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + }; +#endif + + for (i = 0; i < 2; i++) { + onToken = (i == 0) ? CK_FALSE : CK_TRUE; +#ifndef NO_RSA + expect_create(session, rsaSecret, + sizeof(rsaSecret) / sizeof(*rsaSecret), + CKR_TEMPLATE_INCONSISTENT, "RSA key type as secret key"); +#endif +#ifdef HAVE_ECC + expect_create(session, ecSecret, sizeof(ecSecret) / sizeof(*ecSecret), + CKR_TEMPLATE_INCONSISTENT, "EC key type as secret key"); +#endif + } + +#ifndef NO_RSA + rv = funcList->C_FindObjectsInit(session, rsaFind, + sizeof(rsaFind) / sizeof(*rsaFind)); + CHECK_RV(rv, "find inconsistent objects init", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, found, 2, &foundCnt); + CHECK_TRUE(rv == CKR_OK && foundCnt == 0, + "no inconsistent object stored"); + funcList->C_FindObjectsFinal(session); + } +#else + (void)rv; + (void)found; + (void)foundCnt; +#endif +} +#endif + +#if !defined(NO_RSA) || defined(HAVE_ECC) +#define MAX_TMPL 16 + +/* The full template creates the object, and leaving out any one of the + * required attributes is reported as an incomplete template. */ +static void expect_required(CK_SESSION_HANDLE session, CK_ATTRIBUTE* tmpl, + CK_ULONG cnt, const CK_ATTRIBUTE_TYPE* required, + CK_ULONG reqCnt, const char* name) +{ + CK_ATTRIBUTE partial[MAX_TMPL]; + CK_ULONG i; + CK_ULONG j; + CK_ULONG n; + char msg[96]; + + if (cnt > MAX_TMPL) { + CHECK_TRUE(0, "template fits"); + return; + } + for (i = 0; i < reqCnt; i++) { + n = 0; + for (j = 0; j < cnt; j++) { + if (tmpl[j].type != required[i]) + partial[n++] = tmpl[j]; + } + snprintf(msg, sizeof(msg), "%s without attribute 0x%lx", name, + (unsigned long)required[i]); + expect_create(session, partial, n, CKR_TEMPLATE_INCOMPLETE, msg); + } + snprintf(msg, sizeof(msg), "complete %s", name); + expect_create(session, tmpl, cnt, CKR_OK, msg); +} +#endif + +#ifndef NO_RSA +/* An RSA key object needs the components that define the key. */ +static void test_rsa_key_requires_material(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE rsaType = CKK_RSA; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + CK_ATTRIBUTE primesTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + }; + static const CK_ATTRIBUTE_TYPE pubReq[] = { + CKA_MODULUS, CKA_PUBLIC_EXPONENT + }; + static const CK_ATTRIBUTE_TYPE privReq[] = { + CKA_MODULUS, CKA_PRIVATE_EXPONENT + }; + static const CK_ATTRIBUTE_TYPE primesReq[] = { + CKA_PRIME_1, CKA_PRIME_2, CKA_PRIVATE_EXPONENT + }; + + expect_required(session, pubTmpl, sizeof(pubTmpl) / sizeof(*pubTmpl), + pubReq, sizeof(pubReq) / sizeof(*pubReq), + "RSA public key"); + expect_required(session, privTmpl, sizeof(privTmpl) / sizeof(*privTmpl), + privReq, sizeof(privReq) / sizeof(*privReq), + "RSA private key"); + expect_required(session, primesTmpl, + sizeof(primesTmpl) / sizeof(*primesTmpl), primesReq, + sizeof(primesReq) / sizeof(*primesReq), + "RSA private key from primes"); +} +#endif + +#ifdef HAVE_ECC +/* An EC key object needs its curve and its point or private value. */ +static void test_ec_key_requires_material(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE ecType = CKK_EC; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + static const CK_ATTRIBUTE_TYPE pubReq[] = { + CKA_EC_PARAMS, CKA_EC_POINT + }; + static const CK_ATTRIBUTE_TYPE privReq[] = { + CKA_EC_PARAMS, CKA_VALUE + }; + + expect_required(session, pubTmpl, sizeof(pubTmpl) / sizeof(*pubTmpl), + pubReq, sizeof(pubReq) / sizeof(*pubReq), + "EC public key"); + expect_required(session, privTmpl, sizeof(privTmpl) / sizeof(*privTmpl), + privReq, sizeof(privReq) / sizeof(*privReq), + "EC private key"); +} +#endif + +/* CKA_VALUE of a data object is optional and defaults to empty. */ +static void test_data_object_value_optional(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS dataClass = CKO_DATA; + CK_BBOOL onToken = CK_FALSE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte buf[32]; + CK_ATTRIBUTE getValue = { CKA_VALUE, buf, sizeof(buf) }; + CK_ATTRIBUTE setValue = { CKA_VALUE, (void*)dataValue, + sizeof(dataValue) - 1 }; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &onToken, sizeof(onToken) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, (void*)dataLabel, sizeof(dataLabel) - 1 }, + }; + int i; + + for (i = 0; i < 2; i++) { + onToken = (i == 0) ? CK_FALSE : CK_TRUE; + rv = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); + CHECK_RV(rv, "create data object without CKA_VALUE", CKR_OK); + if (rv != CKR_OK) + continue; + + getValue.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, obj, &getValue, 1); + CHECK_TRUE(rv == CKR_OK && getValue.ulValueLen == 0, + "omitted data object value reads as empty"); + + rv = funcList->C_SetAttributeValue(session, obj, &setValue, 1); + CHECK_RV(rv, "set data object value later", CKR_OK); + getValue.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, obj, &getValue, 1); + CHECK_TRUE(rv == CKR_OK && + getValue.ulValueLen == sizeof(dataValue) - 1 && + XMEMCMP(buf, dataValue, sizeof(dataValue) - 1) == 0, + "data object value set after creation"); + + destroy_obj(session, &obj); + } +} + +#if defined(HAVE_ECC) || defined(WOLFPKCS11_MLDSA) || defined(WOLFPKCS11_MLKEM) +/* An attribute the object cannot provide is reported as an error, with the + * length set to CK_UNAVAILABLE_INFORMATION, for size queries and reads. */ +static void expect_unavailable(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_ATTRIBUTE_TYPE type, const char* name) +{ + CK_RV rv; + byte buf[64]; + CK_ATTRIBUTE query = { type, NULL, 0 }; + CK_ATTRIBUTE read = { type, buf, sizeof(buf) }; + char msg[96]; + + rv = funcList->C_GetAttributeValue(session, obj, &query, 1); + snprintf(msg, sizeof(msg), "%s size query fails", name); + CHECK_TRUE(rv != CKR_OK && + query.ulValueLen == CK_UNAVAILABLE_INFORMATION, msg); + rv = funcList->C_GetAttributeValue(session, obj, &read, 1); + snprintf(msg, sizeof(msg), "%s read fails", name); + CHECK_TRUE(rv != CKR_OK && + read.ulValueLen == CK_UNAVAILABLE_INFORMATION, msg); +} +#endif + +#if defined(WOLFPKCS11_MLDSA) || defined(WOLFPKCS11_MLKEM) +/* Generate a parameter-set based key pair as public session objects. */ +static CK_RV gen_pq_key_pair(CK_SESSION_HANDLE session, CK_MECHANISM_TYPE type, + CK_ULONG paramSet, CK_OBJECT_HANDLE* pub, + CK_OBJECT_HANDLE* priv) +{ + CK_MECHANISM mech = { type, NULL, 0 }; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_PARAMETER_SET, ¶mSet, sizeof(paramSet) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + }; + + return funcList->C_GenerateKeyPair(session, &mech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), pub, priv); +} + +/* Create a public key object carrying only its parameter set. */ +static CK_RV create_pq_params_only(CK_SESSION_HANDLE session, + CK_KEY_TYPE keyType, CK_ULONG paramSet, + CK_OBJECT_HANDLE* obj) +{ + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PARAMETER_SET, ¶mSet, sizeof(paramSet) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), obj); +} + +static void check_pq_unavailable(CK_SESSION_HANDLE session, + CK_MECHANISM_TYPE genMech, + CK_KEY_TYPE keyType, CK_ULONG paramSet, + const char* name) +{ + CK_RV rv; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE noPub = CK_INVALID_HANDLE; + char msg[96]; + + rv = gen_pq_key_pair(session, genMech, paramSet, &pub, &priv); + snprintf(msg, sizeof(msg), "%s generate key pair", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv == CKR_OK) { + snprintf(msg, sizeof(msg), "%s private key CKA_SEED", name); + expect_unavailable(session, priv, CKA_SEED, msg); + } + + rv = create_pq_params_only(session, keyType, paramSet, &noPub); + snprintf(msg, sizeof(msg), "%s public key without value", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv == CKR_OK) { + snprintf(msg, sizeof(msg), "%s missing public CKA_VALUE", name); + expect_unavailable(session, noPub, CKA_VALUE, msg); + } + + destroy_obj(session, &noPub); + destroy_obj(session, &priv); + destroy_obj(session, &pub); +} + +/* Reading CKA_PARAMETER_SET into a larger buffer reports the real length. */ +static void check_param_set_len(CK_SESSION_HANDLE session, + CK_MECHANISM_TYPE genMech, CK_ULONG paramSet, + const char* name) +{ + CK_RV rv; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_ULONG buf[4]; + CK_ATTRIBUTE get = { CKA_PARAMETER_SET, buf, sizeof(buf) }; + char msg[96]; + int i; + + rv = gen_pq_key_pair(session, genMech, paramSet, &pub, &priv); + snprintf(msg, sizeof(msg), "%s generate key pair", name); + CHECK_RV(rv, msg, CKR_OK); + for (i = 0; rv == CKR_OK && i < 2; i++) { + XMEMSET(buf, 0, sizeof(buf)); + get.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, (i == 0) ? pub : priv, + &get, 1); + snprintf(msg, sizeof(msg), "%s %s parameter set length", name, + (i == 0) ? "public" : "private"); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(CK_ULONG) && + buf[0] == paramSet, msg); + } + + destroy_obj(session, &priv); + destroy_obj(session, &pub); +} +#endif + +static void test_unavailable_attr_is_error(CK_SESSION_HANDLE session) +{ +#ifdef HAVE_ECC + CK_RV rv; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE ecType = CKK_EC; + CK_OBJECT_HANDLE ecPriv = CK_INVALID_HANDLE; + CK_ATTRIBUTE ecTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + + rv = funcList->C_CreateObject(session, ecTmpl, + sizeof(ecTmpl) / sizeof(*ecTmpl), &ecPriv); + CHECK_RV(rv, "create EC private key without point", CKR_OK); + if (rv == CKR_OK) + expect_unavailable(session, ecPriv, CKA_EC_POINT, "EC point"); + destroy_obj(session, &ecPriv); +#endif +#ifdef WOLFPKCS11_MLDSA + check_pq_unavailable(session, CKM_ML_DSA_KEY_PAIR_GEN, CKK_ML_DSA, + CKP_ML_DSA_44, "ML-DSA"); +#endif +#ifdef WOLFPKCS11_MLKEM + check_pq_unavailable(session, CKM_ML_KEM_KEY_PAIR_GEN, CKK_ML_KEM, + CKP_ML_KEM_512, "ML-KEM"); +#endif + (void)session; +} + +/* CKA_CERTIFICATE_CATEGORY reads back as supplied, and defaults to 0. */ +static void test_cert_category_roundtrip(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS certClass = CKO_CERTIFICATE; + CK_CERTIFICATE_TYPE certType = CKC_X_509; + CK_ULONG category = 2; + CK_ULONG got; + CK_OBJECT_HANDLE cert = CK_INVALID_HANDLE; + CK_ATTRIBUTE get = { CKA_CERTIFICATE_CATEGORY, NULL, 0 }; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &certClass, sizeof(certClass) }, + { CKA_CERTIFICATE_TYPE, &certType, sizeof(certType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_SUBJECT, (void*)certSubject, sizeof(certSubject) }, + { CKA_VALUE, (void*)certValue, sizeof(certValue) }, + { CKA_CERTIFICATE_CATEGORY, &category, sizeof(category) }, + }; + CK_ULONG cnt = sizeof(tmpl) / sizeof(*tmpl); + int i; + + for (i = 0; i < 2; i++) { + rv = funcList->C_CreateObject(session, tmpl, cnt - (CK_ULONG)i, &cert); + CHECK_RV(rv, "create certificate", CKR_OK); + if (rv != CKR_OK) + continue; + + get.pValue = NULL; + get.ulValueLen = 0; + rv = funcList->C_GetAttributeValue(session, cert, &get, 1); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(CK_ULONG), + "certificate category size"); + got = (CK_ULONG)-1; + get.pValue = &got; + get.ulValueLen = sizeof(got); + rv = funcList->C_GetAttributeValue(session, cert, &get, 1); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(got) && + got == ((i == 0) ? category : 0), + (i == 0) ? "certificate category reads back" : + "certificate category defaults to 0"); + destroy_obj(session, &cert); + } +} + +#ifdef HAVE_ECC +/* CKA_SUBJECT on a public or private key reads back as supplied, and the + * certificate-only CKA_ISSUER and CKA_SERIAL_NUMBER are not accepted. */ +static void test_key_subject_roundtrip(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_OBJECT_CLASS privClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE ecType = CKK_EC; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + byte buf[32]; + CK_ATTRIBUTE get = { CKA_SUBJECT, buf, sizeof(buf) }; + CK_ATTRIBUTE issuer = { CKA_ISSUER, (void*)keySubject, + sizeof(keySubject) }; + CK_ATTRIBUTE serial = { CKA_SERIAL_NUMBER, (void*)keySerial, + sizeof(keySerial) }; + byte labelBuf[8]; + CK_ATTRIBUTE getLabel = { CKA_LABEL, labelBuf, sizeof(labelBuf) }; + CK_ATTRIBUTE labelTmpl[] = { { CKA_LABEL, (void*)"A", 1 } }; + CK_ATTRIBUTE mixedTmpl[] = { + { CKA_LABEL, (void*)"B", 1 }, + { CKA_ISSUER, (void*)keySubject, sizeof(keySubject) }, + }; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + { CKA_SUBJECT, (void*)keySubject, sizeof(keySubject) }, + { CKA_ISSUER, (void*)keySubject, sizeof(keySubject) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_CLASS, &privClass, sizeof(privClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + { CKA_SUBJECT, (void*)keySubject, sizeof(keySubject) }, + { CKA_SERIAL_NUMBER, (void*)keySerial, sizeof(keySerial) }, + }; + CK_ULONG pubCnt = sizeof(pubTmpl) / sizeof(*pubTmpl); + CK_ULONG privCnt = sizeof(privTmpl) / sizeof(*privTmpl); + + rv = funcList->C_CreateObject(session, pubTmpl, pubCnt, &key); + CHECK_RV(rv, "public key with CKA_ISSUER rejected", + CKR_ATTRIBUTE_TYPE_INVALID); + destroy_obj(session, &key); + rv = funcList->C_CreateObject(session, privTmpl, privCnt, &key); + CHECK_RV(rv, "private key with CKA_SERIAL_NUMBER rejected", + CKR_ATTRIBUTE_TYPE_INVALID); + destroy_obj(session, &key); + + rv = funcList->C_CreateObject(session, pubTmpl, pubCnt - 1, &key); + CHECK_RV(rv, "create public key with subject", CKR_OK); + if (rv == CKR_OK) { + get.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, key, &get, 1); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(keySubject) && + XMEMCMP(buf, keySubject, sizeof(keySubject)) == 0, + "public key subject reads back"); + rv = funcList->C_SetAttributeValue(session, key, &issuer, 1); + CHECK_RV(rv, "set CKA_ISSUER on public key rejected", + CKR_ATTRIBUTE_TYPE_INVALID); + rv = funcList->C_SetAttributeValue(session, key, labelTmpl, 1); + CHECK_RV(rv, "set public key label", CKR_OK); + rv = funcList->C_SetAttributeValue(session, key, mixedTmpl, 2); + CHECK_RV(rv, "label with CKA_ISSUER on public key rejected", + CKR_ATTRIBUTE_TYPE_INVALID); + rv = funcList->C_GetAttributeValue(session, key, &getLabel, 1); + CHECK_TRUE(rv == CKR_OK && getLabel.ulValueLen == 1 && + labelBuf[0] == 'A', + "label unchanged after the rejected template"); + } + destroy_obj(session, &key); + + rv = funcList->C_CreateObject(session, privTmpl, privCnt - 1, &key); + CHECK_RV(rv, "create private key with subject", CKR_OK); + if (rv == CKR_OK) { + get.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, key, &get, 1); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(keySubject) && + XMEMCMP(buf, keySubject, sizeof(keySubject)) == 0, + "private key subject reads back"); + rv = funcList->C_SetAttributeValue(session, key, &serial, 1); + CHECK_RV(rv, "set CKA_SERIAL_NUMBER on private key rejected", + CKR_ATTRIBUTE_TYPE_INVALID); + } + destroy_obj(session, &key); +} +#endif + +#ifndef NO_AES +static CK_BBOOL get_local(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj) +{ + CK_BBOOL local = 2; + CK_ATTRIBUTE attr = { CKA_LOCAL, &local, sizeof(local) }; + + if (funcList->C_GetAttributeValue(session, obj, &attr, 1) != CKR_OK) + return 2; + return local; +} + +/* A copy keeps CKA_LOCAL of the key it was copied from. */ +static void test_copy_keeps_local(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; + CK_KEY_TYPE aesType = CKK_AES; + CK_ULONG keyLen = 16; + CK_MECHANISM mech = { CKM_AES_KEY_GEN, NULL, 0 }; + CK_OBJECT_HANDLE gen = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE imported = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE copy = CK_INVALID_HANDLE; + static const byte copyLabel[] = "copy"; + CK_ATTRIBUTE copyTmpl[] = { + { CKA_LABEL, (void*)copyLabel, sizeof(copyLabel) - 1 }, + }; + CK_ATTRIBUTE genTmpl[] = { + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE_LEN, &keyLen, sizeof(keyLen) }, + }; + CK_ATTRIBUTE importTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, (void*)secretValue, sizeof(secretValue) }, + }; + + rv = funcList->C_GenerateKey(session, &mech, genTmpl, + sizeof(genTmpl) / sizeof(*genTmpl), &gen); + CHECK_RV(rv, "generate AES key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(get_local(session, gen) == CK_TRUE, + "generated key is local"); + rv = funcList->C_CopyObject(session, gen, copyTmpl, 1, ©); + CHECK_RV(rv, "copy generated key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(get_local(session, copy) == CK_TRUE, + "copy of generated key is local"); + } + destroy_obj(session, ©); + } + + rv = funcList->C_CreateObject(session, importTmpl, + sizeof(importTmpl) / sizeof(*importTmpl), + &imported); + CHECK_RV(rv, "import AES key", CKR_OK); + if (rv == CKR_OK) { + rv = funcList->C_CopyObject(session, imported, copyTmpl, 1, ©); + CHECK_RV(rv, "copy imported key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(get_local(session, copy) == CK_FALSE, + "copy of imported key is not local"); + } + destroy_obj(session, ©); + } + + destroy_obj(session, &imported); + destroy_obj(session, &gen); +} +#endif + +#if defined(WOLFPKCS11_MLDSA) || defined(WOLFPKCS11_MLKEM) +static void test_param_set_len(CK_SESSION_HANDLE session) +{ +#ifdef WOLFPKCS11_MLDSA + check_param_set_len(session, CKM_ML_DSA_KEY_PAIR_GEN, CKP_ML_DSA_44, + "ML-DSA"); +#endif +#ifdef WOLFPKCS11_MLKEM + check_param_set_len(session, CKM_ML_KEM_KEY_PAIR_GEN, CKP_ML_KEM_512, + "ML-KEM"); +#endif +} +#endif + +#ifdef WOLFPKCS11_NSS +static void check_trust_hash(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_ATTRIBUTE_TYPE type, CK_ULONG hashLen, + const char* name) +{ + CK_RV rv; + byte buf[32]; + CK_ATTRIBUTE get = { type, NULL, 0 }; + char msg[96]; + + rv = funcList->C_GetAttributeValue(session, obj, &get, 1); + snprintf(msg, sizeof(msg), "%s size query", name); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == hashLen, msg); + + XMEMSET(buf, 0xAB, sizeof(buf)); + get.pValue = buf; + get.ulValueLen = hashLen - 1; + rv = funcList->C_GetAttributeValue(session, obj, &get, 1); + snprintf(msg, sizeof(msg), "%s undersized buffer", name); + CHECK_TRUE(rv == CKR_BUFFER_TOO_SMALL && buf[hashLen - 1] == 0xAB, msg); + + get.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, obj, &get, 1); + snprintf(msg, sizeof(msg), "%s read", name); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == hashLen, msg); +} + +/* Trust object hashes follow the size query convention. */ +static void test_trust_hash_size_query(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS trustClass = CKO_NSS_TRUST; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + static const byte issuer[] = "CN=Test,O=wolfSSL,C=US"; + static const byte serial[] = { 0x02, 0x01, 0x01 }; + static const byte sha1Hash[20] = { 0x01 }; + static const byte md5Hash[16] = { 0x02 }; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &trustClass, sizeof(trustClass) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_ISSUER, (void*)issuer, sizeof(issuer) - 1 }, + { CKA_SERIAL_NUMBER, (void*)serial, sizeof(serial) }, + { CKA_CERT_SHA1_HASH, (void*)sha1Hash, sizeof(sha1Hash) }, + { CKA_CERT_MD5_HASH, (void*)md5Hash, sizeof(md5Hash) }, + }; + + rv = funcList->C_CreateObject(session, tmpl, sizeof(tmpl) / sizeof(*tmpl), + &obj); + CHECK_RV(rv, "create trust object", CKR_OK); + if (rv == CKR_OK) { + check_trust_hash(session, obj, CKA_CERT_SHA1_HASH, sizeof(sha1Hash), + "trust SHA-1 hash"); + check_trust_hash(session, obj, CKA_CERT_MD5_HASH, sizeof(md5Hash), + "trust MD5 hash"); + } + destroy_obj(session, &obj); +} +#endif + +#ifdef HAVE_ECC +/* CKA_EC_PARAMS read into a buffer shorter than the encoding is reported as + * too small and nothing is written past the length given. */ +static void test_ec_params_short_buffer(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + CK_KEY_TYPE ecType = CKK_EC; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + byte buf[64]; + CK_ATTRIBUTE get = { CKA_EC_PARAMS, NULL, 0 }; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + }; + CK_ULONG need; + CK_ULONG shortLen; + CK_ULONG i; + int intact; + + rv = funcList->C_CreateObject(session, tmpl, sizeof(tmpl) / sizeof(*tmpl), + &key); + CHECK_RV(rv, "create EC public key", CKR_OK); + if (rv != CKR_OK) + return; + + rv = funcList->C_GetAttributeValue(session, key, &get, 1); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(ecc_p256_params), + "EC params size query"); + need = get.ulValueLen; + + for (shortLen = 0; rv == CKR_OK && shortLen < need; shortLen++) { + XMEMSET(buf, 0xAB, sizeof(buf)); + get.pValue = buf; + get.ulValueLen = shortLen; + rv = funcList->C_GetAttributeValue(session, key, &get, 1); + intact = 1; + for (i = shortLen; i < sizeof(buf); i++) { + if (buf[i] != 0xAB) + intact = 0; + } + if (rv != CKR_BUFFER_TOO_SMALL || !intact) { + CHECK_TRUE(0, "EC params short buffer rejected and untouched"); + rv = CKR_GENERAL_ERROR; + } + else { + rv = CKR_OK; + } + } + if (rv == CKR_OK) + CHECK_TRUE(1, "EC params short buffer rejected and untouched"); + + get.pValue = buf; + get.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, key, &get, 1); + CHECK_TRUE(rv == CKR_OK && get.ulValueLen == sizeof(ecc_p256_params) && + XMEMCMP(buf, ecc_p256_params, sizeof(ecc_p256_params)) == 0, + "EC params read"); + + destroy_obj(session, &key); +} +#endif + +static int run_test(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = 0; + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + + rv = pkcs11_open_session(&session); + CHECK_RV(rv, "open session", CKR_OK); + if (rv == CKR_OK) { + test_cert_requires_value(session); + test_secret_key_requires_value(session); + test_create_count_range(session); + test_data_object_value_optional(session); + test_unavailable_attr_is_error(session); + test_cert_category_roundtrip(session); +#ifdef HAVE_ECC + test_key_subject_roundtrip(session); +#endif +#ifndef NO_AES + test_copy_keeps_local(session); +#endif +#if defined(WOLFPKCS11_MLDSA) || defined(WOLFPKCS11_MLKEM) + test_param_set_len(session); +#endif +#ifdef WOLFPKCS11_NSS + test_trust_hash_size_query(session); +#endif +#ifdef HAVE_ECC + test_ec_params_short_buffer(session); +#endif +#if !defined(NO_RSA) || defined(HAVE_ECC) + test_key_type_matches_class(session); +#endif +#ifndef NO_RSA + test_rsa_key_requires_material(session); +#endif +#ifdef HAVE_ECC + test_ec_key_requires_material(session); +#endif + } + + if (session != 0) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + + printf("=== wolfPKCS11 object create and get attribute test ===\n"); + run_test(); + return pkcs11_test_summary(); +} diff --git a/tests/object_policy_integrity_test.c b/tests/object_policy_integrity_test.c new file mode 100644 index 00000000..dd98ba31 --- /dev/null +++ b/tests/object_policy_integrity_test.c @@ -0,0 +1,1900 @@ +/* object_policy_integrity_test.c + * + * Copyright (C) 2026 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * The policy attributes of a protected token key are only honored when they + * match the policy the key material was stored under. Legitimate attribute + * updates, PIN changes and stores written before the policy was bound must + * keep working. Each reload runs in a child process. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include +#include + +#ifndef HAVE_PKCS11_STATIC + #include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#if !defined(WOLFPKCS11_NO_STORE) && !defined(WOLFPKCS11_TPM_STORE) && \ + !defined(WOLFPKCS11_CUSTOM_STORE) && !defined(WOLFPKCS11_NO_ENV) && \ + !defined(_WIN32) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + defined(HAVE_AES_CBC) + +#include +#include +#include +#include + +#define TEST_DIR "./store/object_policy_integrity_test" +#define MAX_OBJ_ID 8 +#define MAX_FILE_SZ 16384 + +/* Offsets of the class and operational flags in a stored object record. */ +#define OBJ_CLASS_OFF (12 + sizeof(CK_ULONG)) +#define OBJ_FLAGS_OFF (12 + 3 * sizeof(CK_ULONG) + 1 + 1 + 4) +#define OBJ_IV_SZ 12 +#define TAG_SZ 16 +#define BLOCK_SZ 16 + +#define CHILD_PASS 0 +#define CHILD_FAIL 1 +#define CHILD_SETUP 2 + +static byte soPin[] = "password123456"; +static byte userPin[] = "wolfpkcs11-test"; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; + +static const char* storeKinds[] = { + "obj", "data", "symmkey", "rsakey_priv", "rsakey_pub", "ecckey_priv", + "ecckey_pub", "dhkey_priv", "dhkey_pub", "cert", "trust", "mldsakey_priv", + "mldsakey_pub", "mlkemkey_priv", "mlkemkey_pub" +}; + +static byte aesKeyValue[32] = { + 0x2f, 0x91, 0x0c, 0x6a, 0xd3, 0x48, 0x75, 0xbe, + 0x13, 0xe0, 0x5c, 0x87, 0x29, 0xfa, 0x64, 0x0b, + 0xc8, 0x3d, 0x9e, 0x51, 0x06, 0xb7, 0x7a, 0x22, + 0xe4, 0x1f, 0x98, 0x6d, 0x30, 0xab, 0x55, 0xc1 +}; +static byte refPlain[BLOCK_SZ] = { + 0x70, 0x6f, 0x6c, 0x69, 0x63, 0x79, 0x2d, 0x72, + 0x65, 0x66, 0x2d, 0x62, 0x6c, 0x6f, 0x63, 0x6b +}; +static byte refCipher[BLOCK_SZ]; + +/* Parameters of the current check, read by the child process. */ +static CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; +static CK_ATTRIBUTE_TYPE secretAttr = CKA_VALUE; +static const byte* secretNeedle = NULL; +static size_t secretNeedleSz = 0; +static int checkAesUse = 0; +static CK_MECHANISM_TYPE signMech = CKM_VENDOR_DEFINED; +static CK_BBOOL expSensitive = CK_TRUE; +static CK_BBOOL expExtractable = CK_FALSE; +static CK_BBOOL expDecrypt = CK_TRUE; + +static void store_path(char* path, size_t sz, const char* kind, int objId) +{ + if (objId < 0) { + (void)snprintf(path, sz, "%s/wp11_%s_%016lx", TEST_DIR, kind, 1UL); + } + else { + (void)snprintf(path, sz, "%s/wp11_%s_%016lx_%016lx", TEST_DIR, kind, + 1UL, (unsigned long)objId); + } +} + +static void remove_record(const char* path) +{ + char staged[280]; + + (void)remove(path); + (void)snprintf(staged, sizeof(staged), "%s.new", path); + (void)remove(staged); +} + +static void cleanup_test_files(void) +{ + static const char* txnKinds[] = { "lock", "intent", "commit" }; + char path[256]; + size_t k; + int i; + + for (k = 0; k < sizeof(storeKinds) / sizeof(*storeKinds); k++) { + for (i = 0; i < MAX_OBJ_ID; i++) { + store_path(path, sizeof(path), storeKinds[k], i); + remove_record(path); + } + } + store_path(path, sizeof(path), "token", -1); + remove_record(path); + for (k = 0; k < sizeof(txnKinds) / sizeof(*txnKinds); k++) { + (void)snprintf(path, sizeof(path), "%s/wp11_txn_%016lx_%s", TEST_DIR, + 1UL, txnKinds[k]); + if (remove(path) != 0) + (void)rmdir(path); + } +} + +/* Locate the stored file of the given kind for its single object. */ +static int find_store_file(const char* kind, char* path, size_t sz) +{ + FILE* f; + int i; + + for (i = 0; i < MAX_OBJ_ID; i++) { + store_path(path, sz, kind, i); + f = fopen(path, "rb"); + if (f != NULL) { + fclose(f); + return 0; + } + } + return -1; +} + +static int read_file(const char* path, byte* data, size_t sz, size_t* readSz) +{ + FILE* f = fopen(path, "rb"); + + if (f == NULL) + return -1; + *readSz = fread(data, 1, sz, f); + if (ferror(f) || !feof(f)) { + fclose(f); + return -1; + } + fclose(f); + return 0; +} + +static int write_file(const char* path, const byte* data, size_t sz) +{ + FILE* f = fopen(path, "wb"); + size_t written; + + if (f == NULL) + return -1; + written = fwrite(data, 1, sz, f); + if (fclose(f) != 0 || written != sz) + return -1; + return 0; +} + +static word32 get_be32(const byte* p) +{ + return ((word32)p[0] << 24) | ((word32)p[1] << 16) | + ((word32)p[2] << 8) | (word32)p[3]; +} + +static void put_be32(byte* p, word32 v) +{ + p[0] = (byte)(v >> 24); + p[1] = (byte)(v >> 16); + p[2] = (byte)(v >> 8); + p[3] = (byte)v; +} + +static CK_ULONG get_be_ulong(const byte* p) +{ + CK_ULONG v = 0; + size_t i; + + for (i = 0; i < sizeof(CK_ULONG); i++) + v = (v << 8) | p[i]; + return v; +} + +/* Rewrite the stored flags of the object of the checked class so they read as + * exportable. */ +static int mark_stored_key_exportable(void) +{ + static byte data[MAX_FILE_SZ]; + char path[256]; + size_t sz = 0; + word32 flags; + int i; + int found = 0; + + for (i = 0; i < MAX_OBJ_ID && !found; i++) { + store_path(path, sizeof(path), "obj", i); + if (read_file(path, data, sizeof(data), &sz) == 0 && + sz >= OBJ_FLAGS_OFF + 4 && + get_be_ulong(data + OBJ_CLASS_OFF) == keyClass) { + found = 1; + } + } + if (!found) + return -1; + flags = get_be32(data + OBJ_FLAGS_OFF); + if ((flags & WP11_FLAG_SENSITIVE) == 0 || + (flags & WP11_FLAG_EXTRACTABLE) != 0) + return -1; + flags &= ~(word32)WP11_FLAG_SENSITIVE; + flags |= WP11_FLAG_EXTRACTABLE; + put_be32(data + OBJ_FLAGS_OFF, flags); + return write_file(path, data, sz); +} + +static CK_RV init_slot(CK_SLOT_ID* slot) +{ + CK_RV rv; + CK_C_INITIALIZE_ARGS args; + CK_SLOT_ID slots[16]; + CK_ULONG count = sizeof(slots) / sizeof(*slots); + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + rv = funcList->C_Initialize(&args); + if (rv == CKR_OK) + rv = funcList->C_GetSlotList(CK_TRUE, slots, &count); + if (rv == CKR_OK && count == 0) + rv = CKR_TOKEN_NOT_PRESENT; + if (rv == CKR_OK) + *slot = slots[0]; + return rv; +} + +static CK_RV provision_token(CK_SLOT_ID slot) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_UTF8CHAR label[32]; + CK_FLAGS flags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, "wolfpkcs11", 10); + rv = funcList->C_InitToken(slot, soPin, sizeof(soPin) - 1, label); + if (rv == CKR_OK) + rv = funcList->C_OpenSession(slot, flags, NULL, NULL, &session); + if (rv == CKR_OK) + rv = funcList->C_Login(session, CKU_SO, soPin, sizeof(soPin) - 1); + if (rv == CKR_OK) + rv = funcList->C_InitPIN(session, userPin, sizeof(userPin) - 1); + if (session != CK_INVALID_HANDLE) { + (void)funcList->C_Logout(session); + (void)funcList->C_CloseSession(session); + } + return rv; +} + +static CK_RV open_session(CK_SLOT_ID slot, int login, + CK_SESSION_HANDLE* session) +{ + CK_RV rv; + CK_FLAGS flags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + + rv = funcList->C_OpenSession(slot, flags, NULL, NULL, session); + if (rv == CKR_OK && login) { + rv = funcList->C_Login(*session, CKU_USER, userPin, + (CK_ULONG)sizeof(userPin) - 1); + } + return rv; +} + +static CK_RV find_key(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* obj) +{ + CK_RV rv; + CK_ULONG count = 0; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + }; + + rv = funcList->C_FindObjectsInit(session, tmpl, 1); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, obj, 1, &count); + (void)funcList->C_FindObjectsFinal(session); + } + if (rv == CKR_OK && count != 1) + rv = CKR_OBJECT_HANDLE_INVALID; + return rv; +} + +static CK_RV aes_encrypt_block(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + byte* out) +{ + CK_RV rv; + byte iv[BLOCK_SZ]; + CK_MECHANISM mech; + CK_ULONG outSz = BLOCK_SZ; + + XMEMSET(iv, 0, sizeof(iv)); + mech.mechanism = CKM_AES_CBC; + mech.pParameter = iv; + mech.ulParameterLen = sizeof(iv); + rv = funcList->C_EncryptInit(session, &mech, obj); + if (rv == CKR_OK) + rv = funcList->C_Encrypt(session, refPlain, sizeof(refPlain), out, + &outSz); + if (rv == CKR_OK && outSz != BLOCK_SZ) + rv = CKR_FUNCTION_FAILED; + return rv; +} + +/* Sign a fixed digest-sized message with the key. */ +static CK_RV sign_block(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj) +{ + CK_RV rv; + CK_MECHANISM mech; + byte msg[32]; + static byte sig[8192]; + CK_ULONG sigSz = sizeof(sig); + + XMEMSET(msg, 0x5a, sizeof(msg)); + mech.mechanism = signMech; + mech.pParameter = NULL; + mech.ulParameterLen = 0; + rv = funcList->C_SignInit(session, &mech, obj); + if (rv == CKR_OK) + rv = funcList->C_Sign(session, msg, sizeof(msg), sig, &sigSz); + return rv; +} + +static CK_RV get_bool(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE obj, + CK_ATTRIBUTE_TYPE type, CK_BBOOL* val) +{ + CK_ATTRIBUTE attr; + + attr.type = type; + attr.pValue = val; + attr.ulValueLen = sizeof(*val); + return funcList->C_GetAttributeValue(session, obj, &attr, 1); +} + +typedef CK_RV (*create_fn)(CK_SESSION_HANDLE session); + +/* Provision a fresh token, create objects with 'create' and persist them. */ +static CK_RV store_objects(create_fn create) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + + cleanup_test_files(); + rv = pkcs11_load(); + if (rv != CKR_OK) + return rv; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = provision_token(slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = create(session); + if (session != CK_INVALID_HANDLE) + (void)funcList->C_CloseSession(session); + (void)funcList->C_Finalize(NULL); + pkcs11_unload(); + return rv; +} + +typedef int (*child_fn)(void); + +/* Run fn in a child process. Returns CHILD_PASS only when it exits normally + * reporting that the invariant held. */ +static int run_in_child(child_fn fn) +{ + pid_t pid; + int status = 0; + + fflush(stdout); + fflush(stderr); + pid = fork(); + if (pid < 0) + return CHILD_SETUP; + if (pid == 0) { + int res = fn(); + fflush(stdout); + fflush(stderr); + _exit(res); + } + if (waitpid(pid, &status, 0) != pid) + return CHILD_SETUP; + if (WIFSIGNALED(status)) { + fprintf(stderr, " reload terminated by signal %d\n", + WTERMSIG(status)); + return CHILD_FAIL; + } + if (!WIFEXITED(status)) + return CHILD_FAIL; + return WEXITSTATUS(status); +} + +/* After logging in, the key signs. */ +static int child_key_signs(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = sign_block(session, obj); + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " signing after reload failed: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static int holds_secret(const byte* buf, size_t sz) +{ + size_t i; + + if (secretNeedle == NULL || sz < secretNeedleSz) + return 0; + for (i = 0; i + secretNeedleSz <= sz; i++) { + if (XMEMCMP(buf + i, secretNeedle, secretNeedleSz) == 0) + return 1; + } + return 0; +} + +static CK_RV create_aes_key_ex(CK_SESSION_HANDLE session, CK_BBOOL priv, + CK_BBOOL sensitive, CK_BBOOL extractable) +{ + CK_RV rv; + CK_OBJECT_CLASS cls = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &cls, sizeof(cls) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &priv, sizeof(priv) }, + { CKA_SENSITIVE, &sensitive, sizeof(sensitive) }, + { CKA_EXTRACTABLE, &extractable, sizeof(extractable) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesKeyValue, sizeof(aesKeyValue) }, + }; + + rv = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, refCipher); + return rv; +} + +static CK_RV create_aes_key(CK_SESSION_HANDLE session) +{ + return create_aes_key_ex(session, CK_TRUE, CK_TRUE, CK_FALSE); +} + +#ifdef HAVE_ECC +static CK_RV create_ecc_private_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS cls = CKO_PRIVATE_KEY; + CK_KEY_TYPE keyType = CKK_EC; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &cls, sizeof(cls) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifndef NO_RSA +static CK_RV create_rsa_private_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS cls = CKO_PRIVATE_KEY; + CK_KEY_TYPE keyType = CKK_RSA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &cls, sizeof(cls) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifndef NO_DH +static CK_RV create_dh_private_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS cls = CKO_PRIVATE_KEY; + CK_KEY_TYPE keyType = CKK_DH; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &cls, sizeof(cls) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifdef WOLFPKCS11_MLDSA +static CK_RV create_mldsa_key_pair(CK_SESSION_HANDLE session) +{ + CK_MECHANISM mech = { CKM_ML_DSA_KEY_PAIR_GEN, NULL, 0 }; + CK_ULONG paramSet = CKP_ML_DSA_44; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_PARAMETER_SET, ¶mSet, sizeof(paramSet) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + }; + + return funcList->C_GenerateKeyPair(session, &mech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &pub, &priv); +} +#endif + +#if defined(WOLFPKCS11_MLDSA) || defined(WOLFPKCS11_MLKEM) +static byte pqValue[8192]; +static CK_ULONG pqValueSz = 0; +static CK_MECHANISM_TYPE pqGenMech = CKM_VENDOR_DEFINED; +static CK_ULONG pqParamSet = 0; + +/* Generate an exportable post-quantum key pair and record the private value. */ +static CK_RV create_pq_exportable_key_pair(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_MECHANISM mech; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_PARAMETER_SET, &pqParamSet, sizeof(pqParamSet) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + }; + CK_ATTRIBUTE attr; + + mech.mechanism = pqGenMech; + mech.pParameter = NULL; + mech.ulParameterLen = 0; + rv = funcList->C_GenerateKeyPair(session, &mech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &pub, &priv); + if (rv == CKR_OK) { + attr.type = CKA_VALUE; + attr.pValue = pqValue; + attr.ulValueLen = sizeof(pqValue); + rv = funcList->C_GetAttributeValue(session, priv, &attr, 1); + pqValueSz = attr.ulValueLen; + } + if (rv == CKR_OK && (pqValueSz == 0 || pqValueSz > sizeof(pqValue))) + rv = CKR_FUNCTION_FAILED; + return rv; +} + +/* After logging in, the private key value reads back unchanged. */ +static int child_pq_value_kept(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + static byte buf[sizeof(pqValue)]; + CK_ATTRIBUTE attr; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) { + attr.type = CKA_VALUE; + attr.pValue = buf; + attr.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, obj, &attr, 1); + } + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK || attr.ulValueLen != pqValueSz || + XMEMCMP(buf, pqValue, pqValueSz) != 0) { + fprintf(stderr, " private value not kept: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void check_pq_round_trip(const char* name, CK_MECHANISM_TYPE genMech, + CK_ULONG paramSet) +{ + CK_RV rv; + char msg[160]; + int res = CHILD_SETUP; + + keyClass = CKO_PRIVATE_KEY; + pqGenMech = genMech; + pqParamSet = paramSet; + rv = store_objects(create_pq_exportable_key_pair); + (void)snprintf(msg, sizeof(msg), "store %s", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv == CKR_OK) + res = run_in_child(child_pq_value_kept); + (void)snprintf(msg, sizeof(msg), "%s private value kept after reload", + name); + CHECK_TRUE(res == CHILD_PASS, msg); + cleanup_test_files(); +} + +static void test_pq_keys_round_trip(void) +{ + printf("\n--- post-quantum private keys round trip ---\n"); +#ifdef WOLFPKCS11_MLDSA + check_pq_round_trip("ML-DSA key pair", CKM_ML_DSA_KEY_PAIR_GEN, + CKP_ML_DSA_44); +#endif +#ifdef WOLFPKCS11_MLKEM + check_pq_round_trip("ML-KEM key pair", CKM_ML_KEM_KEY_PAIR_GEN, + CKP_ML_KEM_768); +#endif +} +#endif + +/* After logging in, the key must not be exported and, for an AES key, must not + * work as the original key. */ +static int child_key_not_exported(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + static byte buf[4096]; + byte out[BLOCK_SZ]; + CK_BBOOL derive = CK_FALSE; + CK_ATTRIBUTE attr; + int res = CHILD_PASS; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv != CKR_OK) { + fprintf(stderr, " login failed: 0x%lx\n", (unsigned long)rv); + res = CHILD_FAIL; + } + if (res == CHILD_PASS && find_key(session, &obj) == CKR_OK) { + attr.type = secretAttr; + attr.pValue = buf; + attr.ulValueLen = sizeof(buf); + rv = funcList->C_GetAttributeValue(session, obj, &attr, 1); + if (rv == CKR_OK && (holds_secret(buf, attr.ulValueLen) || + (secretNeedle == NULL && attr.ulValueLen > 0))) { + fprintf(stderr, " key value was returned\n"); + res = CHILD_FAIL; + } + if (checkAesUse && aes_encrypt_block(session, obj, out) == CKR_OK && + XMEMCMP(out, refCipher, sizeof(out)) == 0) { + fprintf(stderr, " key still operates as the original key\n"); + res = CHILD_FAIL; + } + if (signMech != CKM_VENDOR_DEFINED && + sign_block(session, obj) == CKR_OK) { + fprintf(stderr, " key still signs\n"); + res = CHILD_FAIL; + } + if (get_bool(session, obj, CKA_DERIVE, &derive) == CKR_OK) { + derive = (derive == CK_TRUE) ? CK_FALSE : CK_TRUE; + attr.type = CKA_DERIVE; + attr.pValue = &derive; + attr.ulValueLen = sizeof(derive); + rv = funcList->C_SetAttributeValue(session, obj, &attr, 1); + if (rv == CKR_OK || rv == CKR_USER_NOT_LOGGED_IN) { + fprintf(stderr, " policy update returned 0x%lx\n", + (unsigned long)rv); + res = CHILD_FAIL; + } + } + } + (void)funcList->C_Finalize(NULL); + return res; +} + +/* After logging in, the AES key works as the original key and reports the + * expected policy. */ +static int child_key_intact(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte out[BLOCK_SZ]; + byte value[sizeof(aesKeyValue)]; + CK_BBOOL sensitive = CK_FALSE; + CK_BBOOL extractable = CK_TRUE; + CK_BBOOL decrypt = CK_FALSE; + CK_ATTRIBUTE attr; + int res = CHILD_PASS; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, out); + if (rv == CKR_OK && XMEMCMP(out, refCipher, sizeof(out)) != 0) + rv = CKR_FUNCTION_FAILED; + if (rv == CKR_OK) + rv = get_bool(session, obj, CKA_SENSITIVE, &sensitive); + if (rv == CKR_OK) + rv = get_bool(session, obj, CKA_EXTRACTABLE, &extractable); + if (rv == CKR_OK) + rv = get_bool(session, obj, CKA_DECRYPT, &decrypt); + if (rv != CKR_OK) { + fprintf(stderr, " key unusable after reload: 0x%lx\n", + (unsigned long)rv); + res = CHILD_FAIL; + } + else if (sensitive != expSensitive || extractable != expExtractable || + decrypt != expDecrypt) { + fprintf(stderr, " policy not kept: sensitive %d extractable %d " + "decrypt %d\n", sensitive, extractable, decrypt); + res = CHILD_FAIL; + } + else if (sensitive == CK_TRUE) { + attr.type = CKA_VALUE; + attr.pValue = value; + attr.ulValueLen = sizeof(value); + if (funcList->C_GetAttributeValue(session, obj, &attr, 1) == CKR_OK) { + fprintf(stderr, " sensitive key value was returned\n"); + res = CHILD_FAIL; + } + } + (void)funcList->C_Finalize(NULL); + return res; +} + +static void set_aes_check(void) +{ + keyClass = CKO_SECRET_KEY; + secretAttr = CKA_VALUE; + secretNeedle = aesKeyValue + sizeof(aesKeyValue) - 16; + secretNeedleSz = 16; + checkAesUse = 1; + expSensitive = CK_TRUE; + expExtractable = CK_FALSE; + expDecrypt = CK_TRUE; +} + +static void check_stored_policy(const char* name, create_fn create, + CK_OBJECT_CLASS cls, CK_ATTRIBUTE_TYPE type, + const byte* secret, size_t secretSz, int aes, + CK_MECHANISM_TYPE sign) +{ + CK_RV rv; + char msg[160]; + int res = CHILD_SETUP; + + rv = store_objects(create); + (void)snprintf(msg, sizeof(msg), "store %s", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv != CKR_OK) + return; + keyClass = cls; + secretAttr = type; + secretNeedle = NULL; + secretNeedleSz = 0; + if (secret != NULL && secretSz >= 16) { + secretNeedle = secret + secretSz - 16; + secretNeedleSz = 16; + } + checkAesUse = aes; + signMech = sign; + if (sign != CKM_VENDOR_DEFINED) { + res = run_in_child(child_key_signs); + (void)snprintf(msg, sizeof(msg), "%s signs after reload", name); + CHECK_TRUE(res == CHILD_PASS, msg); + res = CHILD_SETUP; + } + if (mark_stored_key_exportable() == 0) + res = run_in_child(child_key_not_exported); + signMech = CKM_VENDOR_DEFINED; + (void)snprintf(msg, sizeof(msg), + "%s keeps the policy it was stored under", name); + CHECK_TRUE(res == CHILD_PASS, msg); + cleanup_test_files(); +} + +static void test_stored_policy_is_authenticated(void) +{ + printf("\n--- stored key policy is authenticated ---\n"); + check_stored_policy("AES key", create_aes_key, CKO_SECRET_KEY, CKA_VALUE, + aesKeyValue, sizeof(aesKeyValue), 1, CKM_VENDOR_DEFINED); +#ifdef HAVE_ECC + check_stored_policy("ECC private key", create_ecc_private_key, + CKO_PRIVATE_KEY, CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv), 0, + CKM_ECDSA); +#endif +#ifndef NO_RSA + check_stored_policy("RSA private key", create_rsa_private_key, + CKO_PRIVATE_KEY, CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, + sizeof(rsa_2048_priv_exp), 0, CKM_RSA_PKCS); +#endif +#ifndef NO_DH + check_stored_policy("DH private key", create_dh_private_key, + CKO_PRIVATE_KEY, CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv), 0, + CKM_VENDOR_DEFINED); +#endif +#ifdef WOLFPKCS11_MLDSA + check_stored_policy("ML-DSA private key", create_mldsa_key_pair, + CKO_PRIVATE_KEY, CKA_VALUE, NULL, 0, 0, CKM_ML_DSA); +#endif +} + +/* Tighten the policy of an exportable key while logged in. */ +static CK_RV create_and_restrict_aes_key(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + { CKA_DECRYPT, &ckFalse, sizeof(ckFalse) }, + }; + + rv = create_aes_key_ex(session, CK_TRUE, CK_FALSE, CK_TRUE); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, obj, tmpl, + sizeof(tmpl) / sizeof(*tmpl)); + } + return rv; +} + +/* Restrict the key while logged in, then persist the token by adding another + * object and stop without finalizing. */ +static int child_restrict_and_store(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE data = CK_INVALID_HANDLE; + CK_OBJECT_CLASS dataClass = CKO_DATA; + byte value[] = "stored"; + CK_ATTRIBUTE restrictTmpl[] = { + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE dataTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, value, sizeof(value) - 1 }, + }; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = funcList->C_SetAttributeValue(session, obj, restrictTmpl, 2); + if (rv == CKR_OK) + rv = funcList->C_CreateObject(session, dataTmpl, 3, &data); + if (rv != CKR_OK) { + fprintf(stderr, " restrict and store failed: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static CK_RV create_exportable_aes_key(CK_SESSION_HANDLE session) +{ + return create_aes_key_ex(session, CK_TRUE, CK_FALSE, CK_TRUE); +} + +static CK_RV create_public_aes_key(CK_SESSION_HANDLE session) +{ + return create_aes_key_ex(session, CK_FALSE, CK_TRUE, CK_FALSE); +} + +/* Without the user logged in, a policy change cannot be bound to the stored + * key and is refused, while a label change is accepted. */ +static int child_unbound_update_refused(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + char label[] = "renamed"; + CK_ATTRIBUTE policy[] = { + { CKA_DECRYPT, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE name[] = { + { CKA_LABEL, label, sizeof(label) - 1 }, + }; + int res = CHILD_PASS; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 0, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv != CKR_OK) { + fprintf(stderr, " public key not found: 0x%lx\n", (unsigned long)rv); + res = CHILD_FAIL; + } + if (res == CHILD_PASS) { + rv = funcList->C_SetAttributeValue(session, obj, policy, 1); + if (rv != CKR_USER_NOT_LOGGED_IN) { + fprintf(stderr, " policy change returned 0x%lx\n", + (unsigned long)rv); + res = CHILD_FAIL; + } + rv = funcList->C_SetAttributeValue(session, obj, name, 1); + if (rv != CKR_OK) { + fprintf(stderr, " label change returned 0x%lx\n", + (unsigned long)rv); + res = CHILD_FAIL; + } + } + (void)funcList->C_Finalize(NULL); + return res; +} + +static void test_legitimate_updates_round_trip(void) +{ + CK_RV rv; + int res = CHILD_SETUP; + + printf("\n--- legitimate policy updates survive a reload ---\n"); + set_aes_check(); + rv = store_objects(create_and_restrict_aes_key); + CHECK_RV(rv, "restrict AES key policy while logged in", CKR_OK); + if (rv == CKR_OK) { + expDecrypt = CK_FALSE; + res = run_in_child(child_key_intact); + } + CHECK_TRUE(res == CHILD_PASS, "restricted policy is kept after reload"); + res = CHILD_SETUP; + if (rv == CKR_OK && mark_stored_key_exportable() == 0) + res = run_in_child(child_key_not_exported); + CHECK_TRUE(res == CHILD_PASS, + "restricted policy cannot be undone in store"); + cleanup_test_files(); + + res = CHILD_SETUP; + set_aes_check(); + rv = store_objects(create_exportable_aes_key); + CHECK_RV(rv, "store exportable AES key", CKR_OK); + if (rv == CKR_OK) + res = run_in_child(child_restrict_and_store); + CHECK_TRUE(res == CHILD_PASS, "restrict key and store while logged in"); + if (res == CHILD_PASS) + res = run_in_child(child_key_intact); + CHECK_TRUE(res == CHILD_PASS, "policy stored while logged in is kept"); + if (res == CHILD_PASS && mark_stored_key_exportable() == 0) + res = run_in_child(child_key_not_exported); + else + res = CHILD_SETUP; + CHECK_TRUE(res == CHILD_PASS, "policy stored while logged in is bound"); + cleanup_test_files(); + + res = CHILD_SETUP; + set_aes_check(); + rv = store_objects(create_public_aes_key); + CHECK_RV(rv, "store public AES key", CKR_OK); + if (rv == CKR_OK) + res = run_in_child(child_unbound_update_refused); + CHECK_TRUE(res == CHILD_PASS, "policy change needs the user's key"); + res = CHILD_SETUP; + if (rv == CKR_OK) + res = run_in_child(child_key_intact); + CHECK_TRUE(res == CHILD_PASS, "public key usable after refused change"); + cleanup_test_files(); +} + +#if (defined(WOLFPKCS11_PBKDF2) || defined(HAVE_SCRYPT)) && \ + !defined(WOLFSSL_STM32U5_DHUK) && \ + !(defined(HAVE_FIPS) && defined(WOLFPKCS11_NSS)) +#define HAVE_LEGACY_STORE_TEST + +/* Derive the token storage key from the user PIN and the stored token seed, + * the way stores have always been keyed. */ +static int derive_token_key(byte* key, word32 keySz) +{ + static byte data[MAX_FILE_SZ]; + char path[256]; + size_t sz = 0; + size_t off = 32; + byte* seed; + + store_path(path, sizeof(path), "token", -1); + if (read_file(path, data, sizeof(data), &sz) != 0) + return -1; + /* label, SO PIN, SO seed, SO fail count and times */ + if (sz < off + 4) + return -1; + off += 4 + get_be32(data + off) + 16 + 4 + 2 * sizeof(time_t); + /* user PIN, user seed, user fail count and times */ + if (sz < off + 4) + return -1; + off += 4 + get_be32(data + off) + 16 + 4 + 2 * sizeof(time_t); + if (sz < off + 16) + return -1; + seed = data + off; +#ifdef WOLFPKCS11_PBKDF2 + return wc_PBKDF2(key, userPin, (int)sizeof(userPin) - 1, seed, 16, + PBKDF2_ITERATIONS, (int)keySz, WC_SHA256); +#else + return wc_scrypt(key, userPin, (int)sizeof(userPin) - 1, seed, 16, + WP11_HASH_PIN_COST, WP11_HASH_PIN_BLOCKSIZE, + WP11_HASH_PIN_PARALLEL, (int)keySz); +#endif +} + +static int read_stored_iv(byte* iv) +{ + static byte data[MAX_FILE_SZ]; + char path[256]; + size_t sz = 0; + + if (find_store_file("obj", path, sizeof(path)) != 0 || + read_file(path, data, sizeof(data), &sz) != 0 || sz < OBJ_IV_SZ) + return -1; + XMEMCPY(iv, data, OBJ_IV_SZ); + return 0; +} + +/* Replace the stored AES key with an encryption that binds no policy. */ +static int write_unbound_key_record(void) +{ + byte key[32]; + byte iv[OBJ_IV_SZ]; + byte rec[4 + sizeof(aesKeyValue) + TAG_SZ]; + char path[256]; + Aes aes; + int ret; + + ret = derive_token_key(key, sizeof(key)); + if (ret == 0) + ret = read_stored_iv(iv); + if (ret == 0) + ret = find_store_file("symmkey", path, sizeof(path)); + if (ret == 0) + ret = wc_AesInit(&aes, NULL, INVALID_DEVID); + if (ret == 0) { + ret = wc_AesGcmSetKey(&aes, key, sizeof(key)); + if (ret == 0) { + ret = wc_AesGcmEncrypt(&aes, rec + 4, aesKeyValue, + sizeof(aesKeyValue), iv, sizeof(iv), + rec + 4 + sizeof(aesKeyValue), TAG_SZ, NULL, 0); + } + wc_AesFree(&aes); + } + if (ret == 0) { + put_be32(rec, (word32)(sizeof(aesKeyValue) + TAG_SZ)); + ret = write_file(path, rec, sizeof(rec)); + } + wc_ForceZero(key, sizeof(key)); + return ret; +} + +/* Returns 1 when the stored AES key record still verifies without a policy. */ +static int stored_key_is_unbound(void) +{ + static byte data[MAX_FILE_SZ]; + byte key[32]; + byte iv[OBJ_IV_SZ]; + byte plain[sizeof(aesKeyValue)]; + char path[256]; + size_t sz = 0; + Aes aes; + int ret; + int unbound = 0; + + ret = derive_token_key(key, sizeof(key)); + if (ret == 0) + ret = read_stored_iv(iv); + if (ret == 0) + ret = find_store_file("symmkey", path, sizeof(path)); + if (ret == 0) + ret = read_file(path, data, sizeof(data), &sz); + if (ret == 0 && (sz != 4 + sizeof(aesKeyValue) + TAG_SZ || + get_be32(data) != sizeof(aesKeyValue) + TAG_SZ)) + ret = -1; + if (ret == 0) + ret = wc_AesInit(&aes, NULL, INVALID_DEVID); + if (ret == 0) { + ret = wc_AesGcmSetKey(&aes, key, sizeof(key)); + if (ret == 0) { + ret = wc_AesGcmDecrypt(&aes, plain, data + 4, sizeof(plain), iv, + sizeof(iv), data + 4 + sizeof(plain), TAG_SZ, NULL, 0); + unbound = (ret == 0); + } + wc_AesFree(&aes); + } + wc_ForceZero(key, sizeof(key)); + wc_ForceZero(plain, sizeof(plain)); + return unbound; +} + +static void test_unbound_store_upgraded(void) +{ + CK_RV rv; + int res = CHILD_SETUP; + + printf("\n--- store without bound policy loads and is upgraded ---\n"); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + if (rv == CKR_OK && write_unbound_key_record() == 0 && + stored_key_is_unbound()) { + set_aes_check(); + res = run_in_child(child_key_intact); + } + CHECK_TRUE(res == CHILD_PASS, "unbound key record still loads"); + CHECK_TRUE(res == CHILD_PASS && !stored_key_is_unbound(), + "key record is rebound to its policy after use"); + if (res == CHILD_PASS) + res = run_in_child(child_key_intact); + CHECK_TRUE(res == CHILD_PASS, "rebound key loads again"); + res = CHILD_SETUP; + if (mark_stored_key_exportable() == 0) + res = run_in_child(child_key_not_exported); + CHECK_TRUE(res == CHILD_PASS, "rebound key keeps its policy"); + cleanup_test_files(); +} +#endif + +static byte otherUserPin[] = "wolfpkcs11-other"; + +/* A directory at the lock file name makes every store pass fail. */ +static int block_store(char* lockPath, size_t sz) +{ + (void)snprintf(lockPath, sz, "%s/wp11_txn_%016lx_lock", TEST_DIR, 1UL); + (void)unlink(lockPath); + return (mkdir(lockPath, 0700) == 0) ? 0 : -1; +} + +/* Store passes that cannot start must leave the PIN and token in effect. */ +static int child_failed_store_keeps_state(void) +{ + CK_RV rv; + CK_RV setRv = CKR_OK; + CK_RV initRv = CKR_OK; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_UTF8CHAR label[32]; + byte out[BLOCK_SZ]; + char lockPath[256]; + + XMEMSET(label, ' ', sizeof(label)); + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK && block_store(lockPath, sizeof(lockPath)) != 0) + rv = CKR_GENERAL_ERROR; + if (rv == CKR_OK) { + setRv = funcList->C_SetPIN(session, userPin, sizeof(userPin) - 1, + otherUserPin, sizeof(otherUserPin) - 1); + (void)funcList->C_CloseSession(session); + session = CK_INVALID_HANDLE; + initRv = funcList->C_InitToken(slot, soPin, sizeof(soPin) - 1, label); + } + (void)rmdir(lockPath); + if (rv == CKR_OK && (setRv == CKR_OK || initRv == CKR_OK)) + rv = CKR_GENERAL_ERROR; + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, out); + if (rv == CKR_OK && XMEMCMP(out, refCipher, sizeof(out)) != 0) + rv = CKR_FUNCTION_FAILED; + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " old PIN or key lost after failed stores: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_failed_store_keeps_state(void) +{ + CK_RV rv; + + printf("\n--- failed PIN change and token init keep stored state ---\n"); + set_aes_check(); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(run_in_child(child_failed_store_keeps_state) == CHILD_PASS, + "old PIN and key work after failed stores"); + } + cleanup_test_files(); +} + +/* Create a public AES key on the token and change its policy so the next + * store pass has to rebind it. The key is only decoded in this process. */ +static CK_RV create_unbound_public_key(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS cls = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &cls, sizeof(cls) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesKeyValue, sizeof(aesKeyValue) }, + }; + CK_ATTRIBUTE policy[] = { + { CKA_DECRYPT, &ckFalse, sizeof(ckFalse) }, + }; + + rv = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); + if (rv == CKR_OK) + rv = funcList->C_SetAttributeValue(session, obj, policy, 1); + return rv; +} + +/* With the user's key never decoded in this process, a PIN change whose + * store fails must not touch the stored key. */ +static int child_undecoded_keys_survive_failed_pin_change(int asSO) +{ + CK_RV rv; + CK_RV setRv = CKR_OK; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + char lockPath[256]; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 0, &session); + if (rv == CKR_OK && asSO) + rv = funcList->C_Login(session, CKU_SO, soPin, sizeof(soPin) - 1); + if (rv == CKR_OK) + rv = create_unbound_public_key(session); + if (rv == CKR_OK && block_store(lockPath, sizeof(lockPath)) != 0) + rv = CKR_GENERAL_ERROR; + if (rv == CKR_OK && asSO) { + setRv = funcList->C_InitPIN(session, otherUserPin, + sizeof(otherUserPin) - 1); + } + else if (rv == CKR_OK) { + setRv = funcList->C_SetPIN(session, userPin, sizeof(userPin) - 1, + otherUserPin, sizeof(otherUserPin) - 1); + } + (void)rmdir(lockPath); + if (rv == CKR_OK && setRv == CKR_OK) + rv = CKR_GENERAL_ERROR; + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " forced PIN change failure not set up: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static int child_public_pin_change_fails(void) +{ + return child_undecoded_keys_survive_failed_pin_change(0); +} + +static int child_so_init_pin_fails(void) +{ + return child_undecoded_keys_survive_failed_pin_change(1); +} + +/* The user's key still works under the PIN that stayed in storage. */ +static int child_user_key_works(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte out[BLOCK_SZ]; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, out); + if (rv == CKR_OK && XMEMCMP(out, refCipher, sizeof(out)) != 0) + rv = CKR_FUNCTION_FAILED; + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " stored key lost by a failed PIN change: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_failed_store_keeps_undecoded_keys(void) +{ + CK_RV rv; + + printf("\n--- failed PIN change with undecoded keys keeps them ---\n"); + set_aes_check(); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(run_in_child(child_public_pin_change_fails) == CHILD_PASS, + "public session PIN change store failure is forced"); + CHECK_TRUE(run_in_child(child_user_key_works) == CHILD_PASS, + "key survives a failed public session PIN change"); + } + cleanup_test_files(); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(run_in_child(child_so_init_pin_fails) == CHILD_PASS, + "SO user PIN init store failure is forced"); + CHECK_TRUE(run_in_child(child_user_key_works) == CHILD_PASS, + "key survives a failed SO user PIN init"); + } + cleanup_test_files(); +} + +/* Change the user PIN and end without logging out or finalizing. */ +static int child_change_pin_and_exit(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) { + rv = funcList->C_SetPIN(session, userPin, sizeof(userPin) - 1, + otherUserPin, sizeof(otherUserPin) - 1); + } + if (rv != CKR_OK) { + fprintf(stderr, " PIN change failed: 0x%lx\n", (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +/* The key works after logging in with the new PIN. */ +static int child_key_under_new_pin(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte out[BLOCK_SZ]; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 0, &session); + if (rv == CKR_OK) { + rv = funcList->C_Login(session, CKU_USER, otherUserPin, + sizeof(otherUserPin) - 1); + } + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, out); + if (rv == CKR_OK && XMEMCMP(out, refCipher, sizeof(out)) != 0) + rv = CKR_FUNCTION_FAILED; + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " key not usable under the new PIN: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +/* The store pass of a PIN change must itself leave every key usable under + * the new PIN, without a later logout or finalize rewriting the records. */ +static void test_pin_change_is_complete_when_stored(void) +{ + CK_RV rv; + + printf("\n--- a stored PIN change needs no logout to take effect ---\n"); + set_aes_check(); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(run_in_child(child_change_pin_and_exit) == CHILD_PASS, + "PIN change stored"); + CHECK_TRUE(run_in_child(child_key_under_new_pin) == CHILD_PASS, + "key usable under the new PIN after an exit"); + } + cleanup_test_files(); +} + +/* A token init whose store fails leaves nothing behind, and a retry works. */ +static int child_init_token_retry(void) +{ + CK_RV rv; + CK_RV initRv = CKR_OK; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_TOKEN_INFO info; + char lockPath[256]; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK && block_store(lockPath, sizeof(lockPath)) != 0) + rv = CKR_GENERAL_ERROR; + if (rv == CKR_OK) { + initRv = funcList->C_InitToken(slot, soPin, sizeof(soPin) - 1, + (CK_UTF8CHAR*)"wolfpkcs11 " + " "); + (void)rmdir(lockPath); + if (initRv == CKR_OK) + rv = CKR_GENERAL_ERROR; + } + if (rv == CKR_OK) + rv = funcList->C_GetTokenInfo(slot, &info); + if (rv == CKR_OK && (info.flags & CKF_TOKEN_INITIALIZED) != 0) + rv = CKR_GENERAL_ERROR; + if (rv == CKR_OK) + rv = provision_token(slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " token init retry failed: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_init_token_failure_on_fresh_store(void) +{ + printf("\n--- failed token init on a fresh store can be retried ---\n"); + cleanup_test_files(); + CHECK_TRUE(run_in_child(child_init_token_retry) == CHILD_PASS, + "token init works after a failed init of a fresh store"); + cleanup_test_files(); +} + +#define ALT_TEST_DIR "store/./object_policy_integrity_test" + +static byte stagedRecord[] = "staged-record"; + +/* Load the token with the store directory spelled differently. */ +static int child_load_from_alt_dir(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + + XSETENV("WOLFPKCS11_TOKEN_PATH", ALT_TEST_DIR, 1); + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " load from the renamed store dir failed: 0x%lx\n", + (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +/* A committed pass left by a crash is finished by a process that reaches the + * store directory through a different path, whether the marker lists the + * records by basename or by the path the crashed process used. */ +static void test_recovery_ignores_dir_spelling(void) +{ + CK_RV rv; + char path[256]; + char fullPath[256]; + char staged[280]; + char marker[256]; + char lines[600]; + static byte data[MAX_FILE_SZ]; + size_t sz = 0; + int res; + + printf("\n--- committed pass is finished from any store dir path ---\n"); + set_aes_check(); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + store_path(path, sizeof(path), "data", MAX_OBJ_ID - 1); + store_path(fullPath, sizeof(fullPath), "data", MAX_OBJ_ID - 2); + (void)snprintf(marker, sizeof(marker), "%s/wp11_txn_%016lx_commit", + TEST_DIR, 1UL); + (void)snprintf(lines, sizeof(lines), "wp11_data_%016lx_%016lx\n%s\n", 1UL, + (unsigned long)(MAX_OBJ_ID - 1), fullPath); + if (rv == CKR_OK) { + (void)snprintf(staged, sizeof(staged), "%s.new", path); + CHECK_TRUE(write_file(staged, stagedRecord, sizeof(stagedRecord)) == 0, + "stage a record"); + (void)snprintf(staged, sizeof(staged), "%s.new", fullPath); + CHECK_TRUE(write_file(staged, stagedRecord, sizeof(stagedRecord)) == 0, + "stage a second record"); + CHECK_TRUE(write_file(marker, (const byte*)lines, XSTRLEN(lines)) == 0, + "write a commit marker naming both"); + res = run_in_child(child_load_from_alt_dir); + CHECK_TRUE(res == CHILD_PASS, "token loads through the other path"); + CHECK_TRUE(read_file(path, data, sizeof(data), &sz) == 0 && + sz == sizeof(stagedRecord) && + XMEMCMP(data, stagedRecord, sz) == 0, + "record listed by basename was put in place"); + CHECK_TRUE(read_file(fullPath, data, sizeof(data), &sz) == 0 && + sz == sizeof(stagedRecord) && + XMEMCMP(data, stagedRecord, sz) == 0, + "record listed by full path was put in place"); + (void)snprintf(staged, sizeof(staged), "%s.new", path); + CHECK_TRUE(access(staged, F_OK) != 0, "staged copy is gone"); + CHECK_TRUE(access(marker, F_OK) != 0, "commit marker is gone"); + } + cleanup_test_files(); +} + +#ifdef DEBUG_WOLFPKCS11 +/* Matches WP11_TEST_STORE_EXIT_CODE in wolfpkcs11/internal.h. */ +#define STORE_EXIT_CODE 75 +#define CRASH_MAX_RENAMES 40 + +extern void WP11_Test_StoreExitAfterRenames(int renames); + +static int crashAt = 0; + +/* Restrict the key and store the token, ending after crashAt renames. */ +static int child_restrict_and_crash(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE data = CK_INVALID_HANDLE; + CK_OBJECT_CLASS dataClass = CKO_DATA; + byte value[] = "stored"; + CK_ATTRIBUTE restrictTmpl[] = { + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + }; + CK_ATTRIBUTE dataTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, value, sizeof(value) - 1 }, + }; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = funcList->C_SetAttributeValue(session, obj, restrictTmpl, 2); + if (rv == CKR_OK) { + WP11_Test_StoreExitAfterRenames(crashAt); + rv = funcList->C_CreateObject(session, dataTmpl, 3, &data); + WP11_Test_StoreExitAfterRenames(0); + } + return (rv == CKR_OK) ? CHILD_PASS : CHILD_FAIL; +} + +/* The key loads and encrypts as before, under the old or the new policy. */ +static int child_key_old_or_new(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte out[BLOCK_SZ]; + CK_BBOOL sensitive = CK_FALSE; + CK_BBOOL extractable = CK_FALSE; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, out); + if (rv == CKR_OK && XMEMCMP(out, refCipher, sizeof(out)) != 0) + rv = CKR_FUNCTION_FAILED; + if (rv == CKR_OK) + rv = get_bool(session, obj, CKA_SENSITIVE, &sensitive); + if (rv == CKR_OK) + rv = get_bool(session, obj, CKA_EXTRACTABLE, &extractable); + if (rv == CKR_OK && sensitive == extractable) + rv = CKR_GENERAL_ERROR; + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " key not usable after crash at rename %d: 0x%lx\n", + crashAt, (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_store_pass_is_atomic(void) +{ + CK_RV rv; + int res; + int done = 0; + int bad = 0; + + printf("\n--- a store pass cut short at any rename keeps the key ---\n"); + for (crashAt = 1; !done && bad == 0 && crashAt <= CRASH_MAX_RENAMES; + crashAt++) { + set_aes_check(); + rv = store_objects(create_exportable_aes_key); + if (rv != CKR_OK) { + bad++; + break; + } + res = run_in_child(child_restrict_and_crash); + if (res == CHILD_PASS) + done = 1; + else if (res != STORE_EXIT_CODE) + bad++; + if (run_in_child(child_key_old_or_new) != CHILD_PASS) + bad++; + } + CHECK_TRUE(done, "store pass runs to completion without the hook"); + CHECK_TRUE(bad == 0, "key is old or new after a crash at every rename"); + cleanup_test_files(); +} + +static byte newUserPin[] = "wolfpkcs11-new1"; + +/* Restrict the key and change the user PIN, ending after crashAt renames. */ +static int child_rebind_and_change_pin(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE restrictTmpl[] = { + { CKA_SENSITIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_EXTRACTABLE, &ckFalse, sizeof(ckFalse) }, + }; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = funcList->C_SetAttributeValue(session, obj, restrictTmpl, 2); + if (rv == CKR_OK) { + WP11_Test_StoreExitAfterRenames(crashAt); + rv = funcList->C_SetPIN(session, userPin, sizeof(userPin) - 1, + newUserPin, sizeof(newUserPin) - 1); + WP11_Test_StoreExitAfterRenames(0); + } + return (rv == CKR_OK) ? CHILD_PASS : CHILD_FAIL; +} + +/* The key works after logging in with either the old or the new PIN. */ +static int child_key_under_either_pin(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + byte out[BLOCK_SZ]; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 0, &session); + if (rv == CKR_OK) { + rv = funcList->C_Login(session, CKU_USER, userPin, + sizeof(userPin) - 1); + if (rv == CKR_PIN_INCORRECT) { + rv = funcList->C_Login(session, CKU_USER, newUserPin, + sizeof(newUserPin) - 1); + } + } + if (rv == CKR_OK) + rv = find_key(session, &obj); + if (rv == CKR_OK) + rv = aes_encrypt_block(session, obj, out); + if (rv == CKR_OK && XMEMCMP(out, refCipher, sizeof(out)) != 0) + rv = CKR_FUNCTION_FAILED; + (void)funcList->C_Finalize(NULL); + if (rv != CKR_OK) { + fprintf(stderr, " key not usable after crash at rename %d: 0x%lx\n", + crashAt, (unsigned long)rv); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_pin_change_pass_is_atomic(void) +{ + CK_RV rv; + int res; + int done = 0; + int bad = 0; + + printf("\n--- a PIN change cut short at any rename keeps the key ---\n"); + for (crashAt = 1; !done && bad == 0 && crashAt <= CRASH_MAX_RENAMES; + crashAt++) { + set_aes_check(); + rv = store_objects(create_exportable_aes_key); + if (rv != CKR_OK) { + bad++; + break; + } + res = run_in_child(child_rebind_and_change_pin); + if (res == CHILD_PASS) + done = 1; + else if (res != STORE_EXIT_CODE) + bad++; + if (run_in_child(child_key_under_either_pin) != CHILD_PASS) + bad++; + } + CHECK_TRUE(done, "PIN change runs to completion without the hook"); + CHECK_TRUE(bad == 0, "key usable with old or new PIN after every crash"); + cleanup_test_files(); +} +#endif + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); + + printf("=== wolfPKCS11 stored object policy integrity test ===\n"); + test_stored_policy_is_authenticated(); + test_legitimate_updates_round_trip(); + test_failed_store_keeps_state(); + test_failed_store_keeps_undecoded_keys(); + test_pin_change_is_complete_when_stored(); + test_init_token_failure_on_fresh_store(); + test_recovery_ignores_dir_spelling(); +#ifdef DEBUG_WOLFPKCS11 + test_store_pass_is_atomic(); + test_pin_change_pass_is_atomic(); +#endif +#if defined(WOLFPKCS11_MLDSA) || defined(WOLFPKCS11_MLKEM) + test_pq_keys_round_trip(); +#endif +#ifdef HAVE_LEGACY_STORE_TEST + test_unbound_store_upgraded(); +#endif + + return pkcs11_test_summary(); +} + +#else + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + printf("File-backed token storage not available, skipping test\n"); + return 77; +} + +#endif diff --git a/tests/object_store_decode_test.c b/tests/object_store_decode_test.c new file mode 100644 index 00000000..74fcdb2f --- /dev/null +++ b/tests/object_store_decode_test.c @@ -0,0 +1,1280 @@ +/* object_store_decode_test.c + * + * Copyright (C) 2026 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Stored token objects whose recorded lengths do not describe valid data must + * be rejected cleanly when the token is loaded or logged into. Each reload runs + * in a child process so a failure is reported instead of ending the test. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC + #include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#if !defined(WOLFPKCS11_NO_STORE) && !defined(WOLFPKCS11_TPM_STORE) && \ + !defined(WOLFPKCS11_CUSTOM_STORE) && !defined(WOLFPKCS11_NO_ENV) && \ + !defined(_WIN32) + +#include +#include +#include + +#define TEST_DIR "./store/object_store_decode_test" +#define MAX_OBJ_ID 8 +#define MAX_FILE_SZ 16384 + +/* Offset of the first variable-length attribute in a stored object record. */ +#define OBJ_ID_LEN_OFF (12 + 3 * sizeof(CK_ULONG) + 1 + 1 + 4 + 4 + 8 + 8) +#define OBJ_LABEL_LEN_OFF (OBJ_ID_LEN_OFF + 4) +#define OBJ_ISSUER_LEN_OFF (OBJ_LABEL_LEN_OFF + 4) + +#define CHILD_PASS 0 +#define CHILD_FAIL 1 +#define CHILD_SETUP 2 + +static byte soPin[] = "password123456"; +static byte userPin[] = "wolfpkcs11-test"; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; +static byte dataValue[] = "stored data object value"; + +static const char* storeKinds[] = { + "obj", "data", "symmkey", "rsakey_priv", "rsakey_pub", "ecckey_priv", + "ecckey_pub", "dhkey_priv", "dhkey_pub", "cert", "trust", "mldsakey_priv", + "mldsakey_pub" +}; + +#if defined(USE_WOLFSSL_MEMORY) && !defined(WOLFSSL_STATIC_MEMORY) && \ + !defined(WOLFSSL_DEBUG_MEMORY) +#define HAVE_ALLOC_TRACKING +#define ALLOC_REFUSE_SZ ((size_t)64 * 1024 * 1024) + +#define TRACK_HDR_SZ 16 + +static size_t largestAlloc = 0; +/* Live bytes in blocks no larger than peakCap, and their peak. */ +static size_t peakCap = 0; +static size_t liveBytes = 0; +static size_t peakLive = 0; +static int refuseAllocs = 0; + +static void* track_malloc(size_t sz) +{ + unsigned char* p; + + if (sz > largestAlloc) + largestAlloc = sz; + if (sz >= ALLOC_REFUSE_SZ || refuseAllocs) + return NULL; + p = (unsigned char*)malloc(sz + TRACK_HDR_SZ); + if (p == NULL) + return NULL; + XMEMCPY(p, &sz, sizeof(sz)); + if (sz <= peakCap) { + liveBytes += sz; + if (liveBytes > peakLive) + peakLive = liveBytes; + } + return p + TRACK_HDR_SZ; +} + +static void track_free(void* ptr) +{ + unsigned char* p; + size_t sz; + + if (ptr == NULL) + return; + p = (unsigned char*)ptr - TRACK_HDR_SZ; + XMEMCPY(&sz, p, sizeof(sz)); + if (sz <= peakCap) + liveBytes -= sz; + free(p); +} + +static void* track_realloc(void* ptr, size_t sz) +{ + void* n; + size_t old; + + if (ptr == NULL) + return track_malloc(sz); + if (sz == 0) { + track_free(ptr); + return NULL; + } + n = track_malloc(sz); + if (n == NULL) + return NULL; + XMEMCPY(&old, (unsigned char*)ptr - TRACK_HDR_SZ, sizeof(old)); + XMEMCPY(n, ptr, (old < sz) ? old : sz); + track_free(ptr); + return n; +} +#endif + +static void store_path(char* path, size_t sz, const char* kind, int objId) +{ + if (objId < 0) { + (void)snprintf(path, sz, "%s/wp11_%s_%016lx", TEST_DIR, kind, 1UL); + } + else { + (void)snprintf(path, sz, "%s/wp11_%s_%016lx_%016lx", TEST_DIR, kind, + 1UL, (unsigned long)objId); + } +} + +static void cleanup_test_files(void) +{ + char path[256]; + size_t k; + int i; + + for (k = 0; k < sizeof(storeKinds) / sizeof(*storeKinds); k++) { + for (i = 0; i < MAX_OBJ_ID; i++) { + store_path(path, sizeof(path), storeKinds[k], i); + (void)remove(path); + } + } + store_path(path, sizeof(path), "token", -1); + (void)remove(path); +} + +/* Locate the stored file of the given kind for the single object of that kind. */ +static int find_store_file(const char* kind, char* path, size_t sz) +{ + FILE* f; + int i; + + for (i = 0; i < MAX_OBJ_ID; i++) { + store_path(path, sz, kind, i); + f = fopen(path, "rb"); + if (f != NULL) { + fclose(f); + return 0; + } + } + return -1; +} + +static int read_file(const char* path, byte* data, size_t sz, size_t* readSz) +{ + FILE* f = fopen(path, "rb"); + + if (f == NULL) + return -1; + *readSz = fread(data, 1, sz, f); + if (ferror(f) || !feof(f)) { + fclose(f); + return -1; + } + fclose(f); + return 0; +} + +static int write_file(const char* path, const byte* data, size_t sz) +{ + FILE* f = fopen(path, "wb"); + size_t written; + + if (f == NULL) + return -1; + written = fwrite(data, 1, sz, f); + if (fclose(f) != 0 || written != sz) + return -1; + return 0; +} + +static word32 get_be32(const byte* p) +{ + return ((word32)p[0] << 24) | ((word32)p[1] << 16) | + ((word32)p[2] << 8) | (word32)p[3]; +} + +static void put_be32(byte* p, word32 v) +{ + p[0] = (byte)(v >> 24); + p[1] = (byte)(v >> 16); + p[2] = (byte)(v >> 8); + p[3] = (byte)v; +} + +/* Overwrite a 4-byte length field that is expected to hold 'expect'. */ +static int patch_length(const char* path, size_t off, word32 expect, + word32 value) +{ + static byte data[MAX_FILE_SZ]; + size_t sz = 0; + + if (read_file(path, data, sizeof(data), &sz) != 0 || sz < off + 4) + return -1; + if (get_be32(data + off) != expect) + return -1; + put_be32(data + off, value); + return write_file(path, data, sz); +} + +static CK_RV init_slot(CK_SLOT_ID* slot) +{ + CK_RV rv; + CK_C_INITIALIZE_ARGS args; + CK_SLOT_ID slots[16]; + CK_ULONG count = sizeof(slots) / sizeof(*slots); + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + rv = funcList->C_Initialize(&args); + if (rv == CKR_OK) + rv = funcList->C_GetSlotList(CK_TRUE, slots, &count); + if (rv == CKR_OK && count == 0) + rv = CKR_TOKEN_NOT_PRESENT; + if (rv == CKR_OK) + *slot = slots[0]; + return rv; +} + +static CK_RV provision_token(CK_SLOT_ID slot) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_UTF8CHAR label[32]; + CK_FLAGS flags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, "wolfpkcs11", 10); + rv = funcList->C_InitToken(slot, soPin, sizeof(soPin) - 1, label); + if (rv == CKR_OK) + rv = funcList->C_OpenSession(slot, flags, NULL, NULL, &session); + if (rv == CKR_OK) + rv = funcList->C_Login(session, CKU_SO, soPin, sizeof(soPin) - 1); + if (rv == CKR_OK) + rv = funcList->C_InitPIN(session, userPin, sizeof(userPin) - 1); + if (session != CK_INVALID_HANDLE) { + (void)funcList->C_Logout(session); + (void)funcList->C_CloseSession(session); + } + return rv; +} + +static CK_RV open_session(CK_SLOT_ID slot, int login, + CK_SESSION_HANDLE* session) +{ + CK_RV rv; + CK_FLAGS flags = CKF_SERIAL_SESSION | CKF_RW_SESSION; + + rv = funcList->C_OpenSession(slot, flags, NULL, NULL, session); + if (rv == CKR_OK && login) { + rv = funcList->C_Login(*session, CKU_USER, userPin, + sizeof(userPin) - 1); + } + return rv; +} + +typedef CK_RV (*create_fn)(CK_SESSION_HANDLE session); + +/* Provision a fresh token, create objects with 'create' and persist them. */ +static CK_RV store_objects(create_fn create) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + + cleanup_test_files(); + rv = pkcs11_load(); + if (rv != CKR_OK) + return rv; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = provision_token(slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv == CKR_OK) + rv = create(session); + if (session != CK_INVALID_HANDLE) + (void)funcList->C_CloseSession(session); + (void)funcList->C_Finalize(NULL); + pkcs11_unload(); + return rv; +} + +typedef int (*child_fn)(void); + +/* Run fn in a child process. Returns CHILD_PASS only when it exits normally + * reporting that the invariant held. */ +static int run_in_child(child_fn fn) +{ + pid_t pid; + int status = 0; + + fflush(stdout); + fflush(stderr); + pid = fork(); + if (pid < 0) + return CHILD_SETUP; + if (pid == 0) { + int res = fn(); + fflush(stdout); + fflush(stderr); + _exit(res); + } + if (waitpid(pid, &status, 0) != pid) + return CHILD_SETUP; + if (WIFSIGNALED(status)) { + fprintf(stderr, " reload terminated by signal %d\n", + WTERMSIG(status)); + return CHILD_FAIL; + } + if (!WIFEXITED(status)) + return CHILD_FAIL; + return WEXITSTATUS(status); +} + +static CK_RV create_data_object(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS dataClass = CKO_DATA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + static byte app[] = "wolfpkcs11"; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_APPLICATION, app, sizeof(app) - 1 }, + { CKA_VALUE, dataValue, sizeof(dataValue) - 1 }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} + +/* Either the token refuses to load or the data object reports a real label + * length. */ +static int child_data_label_length(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_CLASS dataClass = CKO_DATA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ULONG count = 0; + CK_ATTRIBUTE findTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + }; + CK_ATTRIBUTE label = { CKA_LABEL, NULL, 0 }; + int res = CHILD_SETUP; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv != CKR_OK) + return CHILD_PASS; + + rv = open_session(slot, 0, &session); + if (rv == CKR_OK) + rv = funcList->C_FindObjectsInit(session, findTmpl, 1); + if (rv == CKR_OK) + rv = funcList->C_FindObjects(session, &obj, 1, &count); + if (rv == CKR_OK) + rv = funcList->C_FindObjectsFinal(session); + if (rv == CKR_OK && count == 1) + rv = funcList->C_GetAttributeValue(session, obj, &label, 1); + if (rv == CKR_OK && count == 1) { + res = (label.ulValueLen == CK_UNAVAILABLE_INFORMATION || + label.ulValueLen > 64) ? CHILD_FAIL : CHILD_PASS; + if (res != CHILD_PASS) { + fprintf(stderr, " loaded label length 0x%lx\n", + (unsigned long)label.ulValueLen); + } + } + (void)funcList->C_Finalize(NULL); + return res; +} + +/* A negative length must not pass as the end of an older record format. */ +static int child_token_not_loaded(void) +{ + CK_SLOT_ID slot = 0; + int res; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + res = (init_slot(&slot) != CKR_OK) ? CHILD_PASS : CHILD_FAIL; + (void)funcList->C_Finalize(NULL); + return res; +} + +static void test_negative_optional_length(void) +{ + CK_RV rv; + char path[256]; + int res = CHILD_SETUP; + + printf("\n--- stored issuer length must not be negative ---\n"); + rv = store_objects(create_data_object); + CHECK_RV(rv, "store data object", CKR_OK); + if (rv != CKR_OK) + return; + if (find_store_file("obj", path, sizeof(path)) == 0 && + patch_length(path, OBJ_ISSUER_LEN_OFF, 0, 0xFFFFFFFFUL) == 0) { + res = run_in_child(child_token_not_loaded); + } + CHECK_TRUE(res == CHILD_PASS, + "negative stored issuer length is not loaded"); + cleanup_test_files(); +} + +static void test_negative_attribute_length(void) +{ + CK_RV rv; + char path[256]; + int res = CHILD_SETUP; + + printf("\n--- stored attribute length must not be negative ---\n"); + rv = store_objects(create_data_object); + CHECK_RV(rv, "store data object", CKR_OK); + if (rv != CKR_OK) + return; + if (find_store_file("obj", path, sizeof(path)) == 0 && + patch_length(path, OBJ_LABEL_LEN_OFF, 0, 0xFFFFFFFFUL) == 0) { + res = run_in_child(child_data_label_length); + } + CHECK_TRUE(res == CHILD_PASS, + "negative stored attribute length is not loaded"); + cleanup_test_files(); +} + +static int truncate_file(const char* path, size_t keep) +{ + static byte data[MAX_FILE_SZ]; + size_t sz = 0; + + if (read_file(path, data, sizeof(data), &sz) != 0 || sz <= keep) + return -1; + return write_file(path, data, keep); +} + +/* Either the token refuses to load or the data object reads back its full + * stored value. */ +static int child_data_value_intact(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_CLASS dataClass = CKO_DATA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ULONG count = 0; + CK_ATTRIBUTE findTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + }; + byte value[64]; + CK_ATTRIBUTE attr = { CKA_VALUE, value, sizeof(value) }; + int res = CHILD_FAIL; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv != CKR_OK) + return CHILD_PASS; + + rv = open_session(slot, 0, &session); + if (rv == CKR_OK) + rv = funcList->C_FindObjectsInit(session, findTmpl, 1); + if (rv == CKR_OK) + rv = funcList->C_FindObjects(session, &obj, 1, &count); + if (rv == CKR_OK) + rv = funcList->C_FindObjectsFinal(session); + if (rv == CKR_OK && count == 1) + rv = funcList->C_GetAttributeValue(session, obj, &attr, 1); + if (rv == CKR_OK && count == 1 && + attr.ulValueLen == sizeof(dataValue) - 1 && + XMEMCMP(value, dataValue, sizeof(dataValue) - 1) == 0) { + res = CHILD_PASS; + } + (void)funcList->C_Finalize(NULL); + return res; +} + +static void test_truncated_data_record(void) +{ + CK_RV rv; + char path[256]; + int res = CHILD_SETUP; + + printf("\n--- truncated data object record is not loaded ---\n"); + rv = store_objects(create_data_object); + CHECK_RV(rv, "store data object", CKR_OK); + if (rv != CKR_OK) + return; + if (find_store_file("data", path, sizeof(path)) == 0 && + truncate_file(path, 2) == 0) { + res = run_in_child(child_data_value_intact); + } + CHECK_TRUE(res == CHILD_PASS, "truncated data object record is rejected"); + cleanup_test_files(); +} + +/* Replace the first length-prefixed array in a stored key record with newLen + * bytes, keeping the rest of the record unchanged. */ +static int rewrite_first_array(const char* path, word32 newLen) +{ + static byte data[MAX_FILE_SZ]; + static byte out[MAX_FILE_SZ]; + size_t sz = 0; + word32 oldLen; + size_t tail; + + if (read_file(path, data, sizeof(data), &sz) != 0 || sz < 4) + return -1; + oldLen = get_be32(data); + if (oldLen > sz - 4 || 4 + (size_t)newLen + (sz - 4 - oldLen) > sizeof(out)) + return -1; + tail = sz - 4 - oldLen; + put_be32(out, newLen); + XMEMSET(out + 4, 0x5A, newLen); + XMEMCPY(out + 4, data + 4, (newLen < oldLen) ? newLen : oldLen); + XMEMCPY(out + 4 + newLen, data + 4 + oldLen, tail); + return write_file(path, out, 4 + newLen + tail); +} + +/* Loading may succeed but logging in must reject the damaged key record. */ +static int child_login_rejected(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + int res = CHILD_FAIL; + + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv != CKR_OK) + return CHILD_PASS; + rv = open_session(slot, 1, &session); + if (rv != CKR_OK) + res = CHILD_PASS; + (void)funcList->C_Finalize(NULL); + return res; +} + +static void check_damaged_key_record(const char* name, create_fn create, + const char* kind, word32 newLen) +{ + CK_RV rv; + char path[256]; + char msg[128]; + int res = CHILD_SETUP; + + rv = store_objects(create); + (void)snprintf(msg, sizeof(msg), "store %s", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv != CKR_OK) + return; + if (find_store_file(kind, path, sizeof(path)) == 0 && + rewrite_first_array(path, newLen) == 0) { + res = run_in_child(child_login_rejected); + } + (void)snprintf(msg, sizeof(msg), "damaged %s record is rejected", name); + CHECK_TRUE(res == CHILD_PASS, msg); + cleanup_test_files(); +} + +#ifndef NO_DH +static CK_RV create_dh_private_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS keyClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE keyType = CKK_DH; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_priv, sizeof(dh_2048_priv) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifdef HAVE_ECC +static CK_RV create_ecc_private_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS keyClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE keyType = CKK_EC; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_VALUE, ecc_p256_priv, sizeof(ecc_p256_priv) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifndef NO_RSA +static CK_RV create_rsa_private_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS keyClass = CKO_PRIVATE_KEY; + CK_KEY_TYPE keyType = CKK_RSA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + { CKA_PRIVATE_EXPONENT, rsa_2048_priv_exp, sizeof(rsa_2048_priv_exp) }, + { CKA_PRIME_1, rsa_2048_p, sizeof(rsa_2048_p) }, + { CKA_PRIME_2, rsa_2048_q, sizeof(rsa_2048_q) }, + { CKA_EXPONENT_1, rsa_2048_dP, sizeof(rsa_2048_dP) }, + { CKA_EXPONENT_2, rsa_2048_dQ, sizeof(rsa_2048_dQ) }, + { CKA_COEFFICIENT, rsa_2048_u, sizeof(rsa_2048_u) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifdef WOLFPKCS11_MLDSA +static CK_RV create_mldsa_key_pair(CK_SESSION_HANDLE session) +{ + CK_MECHANISM mech = { CKM_ML_DSA_KEY_PAIR_GEN, NULL, 0 }; + CK_ULONG paramSet = CKP_ML_DSA_44; + CK_OBJECT_HANDLE pub = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE priv = CK_INVALID_HANDLE; + CK_ATTRIBUTE pubTmpl[] = { + { CKA_PARAMETER_SET, ¶mSet, sizeof(paramSet) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + }; + CK_ATTRIBUTE privTmpl[] = { + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + }; + + return funcList->C_GenerateKeyPair(session, &mech, pubTmpl, + sizeof(pubTmpl) / sizeof(*pubTmpl), privTmpl, + sizeof(privTmpl) / sizeof(*privTmpl), &pub, &priv); +} +#endif + +/* A stored private key record shorter than its authentication tag, or + * longer than the key it decrypts into. */ +static void test_short_private_key_record(void) +{ + printf("\n--- short private key record is rejected ---\n"); +#ifndef NO_DH + check_damaged_key_record("DH private key", create_dh_private_key, + "dhkey_priv", 8); + check_damaged_key_record("tag-only DH private key", create_dh_private_key, + "dhkey_priv", 16); + check_damaged_key_record("oversized DH private key", + create_dh_private_key, "dhkey_priv", 4096 + 16); +#endif +#ifdef HAVE_ECC + check_damaged_key_record("tag-only ECC private key", + create_ecc_private_key, "ecckey_priv", 16); +#endif +#ifndef NO_RSA + check_damaged_key_record("tag-only RSA private key", + create_rsa_private_key, "rsakey_priv", 16); +#endif +#ifdef WOLFPKCS11_MLDSA + check_damaged_key_record("tag-only ML-DSA private key", + create_mldsa_key_pair, "mldsakey_priv", 16); +#endif +} + +#ifndef NO_DH +static CK_RV create_dh_public_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS keyClass = CKO_PUBLIC_KEY; + CK_KEY_TYPE keyType = CKK_DH; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIME, dh_ffdhe2048_p, sizeof(dh_ffdhe2048_p) }, + { CKA_BASE, dh_ffdhe2048_g, sizeof(dh_ffdhe2048_g) }, + { CKA_VALUE, dh_2048_pub, sizeof(dh_2048_pub) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} + +/* A stored public key record longer than the largest supported key. */ +static void test_oversized_public_key_record(void) +{ + printf("\n--- oversized public key record is rejected ---\n"); + check_damaged_key_record("oversized DH public key", create_dh_public_key, + "dhkey_pub", 4096); +} +#endif + +#ifdef WOLFPKCS11_NSS +static CK_RV create_trust_object(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS trustClass = CKO_NSS_TRUST; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + static byte issuer[] = "CN=Test,O=wolfSSL,C=US"; + static byte serial[] = { 0x02, 0x05, 0x00, 0xC6, 0xA7, 0x91, 0x84 }; + static byte sha1Hash[20] = { 0 }; + static byte md5Hash[16] = { 0 }; + CK_ULONG trustValue = 0xCE534352; + CK_ATTRIBUTE tmpl[] = { + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_CLASS, &trustClass, sizeof(trustClass) }, + { CKA_ISSUER, issuer, sizeof(issuer) - 1 }, + { CKA_SERIAL_NUMBER, serial, sizeof(serial) }, + { CKA_CERT_SHA1_HASH, sha1Hash, sizeof(sha1Hash) }, + { CKA_CERT_MD5_HASH, md5Hash, sizeof(md5Hash) }, + { CKA_TRUST_SERVER_AUTH, &trustValue, sizeof(trustValue) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} + +/* Trust records have a fixed size and are stored unencrypted. */ +static void test_trust_record_length(void) +{ + printf("\n--- trust record length must match ---\n"); + check_damaged_key_record("oversized trust", create_trust_object, "trust", + 1024); + check_damaged_key_record("empty trust", create_trust_object, "trust", 0); +} +#endif + +#define LARGE_VALUE_SZ 300001 +static byte largeValue[LARGE_VALUE_SZ]; +static byte largeRead[LARGE_VALUE_SZ]; + +static CK_RV create_large_data_object(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS dataClass = CKO_DATA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, largeValue, sizeof(largeValue) }, + }; + size_t i; + + for (i = 0; i < sizeof(largeValue); i++) + largeValue[i] = (byte)(i * 7 + (i >> 8)); + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} + +static void test_large_stored_value_round_trip(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_CLASS dataClass = CKO_DATA; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ULONG count = 0; + CK_ATTRIBUTE findTmpl[] = { + { CKA_CLASS, &dataClass, sizeof(dataClass) }, + }; + CK_ATTRIBUTE value = { CKA_VALUE, largeRead, sizeof(largeRead) }; + + printf("\n--- large stored value reloads intact ---\n"); + rv = store_objects(create_large_data_object); + CHECK_RV(rv, "store large data object", CKR_OK); + if (rv != CKR_OK) + return; + rv = pkcs11_load(); + if (rv == CKR_OK) { + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 0, &session); + if (rv == CKR_OK) + rv = funcList->C_FindObjectsInit(session, findTmpl, 1); + if (rv == CKR_OK) + rv = funcList->C_FindObjects(session, &obj, 1, &count); + if (rv == CKR_OK) + rv = funcList->C_FindObjectsFinal(session); + if (rv == CKR_OK && count != 1) + rv = CKR_GENERAL_ERROR; + if (rv == CKR_OK) + rv = funcList->C_GetAttributeValue(session, obj, &value, 1); + (void)funcList->C_Finalize(NULL); + pkcs11_unload(); + } + CHECK_RV(rv, "reload large data object", CKR_OK); + CHECK_TRUE(rv == CKR_OK && value.ulValueLen == sizeof(largeValue) && + XMEMCMP(largeRead, largeValue, sizeof(largeValue)) == 0, + "large stored value is unchanged after reload"); + cleanup_test_files(); +} + +#ifdef HAVE_ALLOC_TRACKING +/* Loading must not size an allocation from a length the data cannot back. */ +static int child_load_allocation_bound(void) +{ + CK_SLOT_ID slot = 0; + + if (wolfSSL_SetAllocators(track_malloc, track_free, track_realloc) != 0) + return CHILD_SETUP; + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + if (init_slot(&slot) == CKR_OK) + (void)funcList->C_Finalize(NULL); + if (largestAlloc >= ALLOC_REFUSE_SZ) { + fprintf(stderr, " largest allocation while loading: %lu\n", + (unsigned long)largestAlloc); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_oversized_attribute_length(void) +{ + CK_RV rv; + char path[256]; + int res = CHILD_SETUP; + + printf("\n--- stored attribute length must be bounded ---\n"); + rv = store_objects(create_data_object); + CHECK_RV(rv, "store data object", CKR_OK); + if (rv != CKR_OK) + return; + if (find_store_file("obj", path, sizeof(path)) == 0 && + patch_length(path, OBJ_LABEL_LEN_OFF, 0, 0x7FFFFFF8UL) == 0) { + res = run_in_child(child_load_allocation_bound); + } + CHECK_TRUE(res == CHILD_PASS, + "oversized stored attribute length is not allocated"); + cleanup_test_files(); +} + +/* Loading a large stored value holds about one copy of it, not two. Larger + * blocks cannot be partial copies of it, so they are not counted. */ +static int child_large_value_peak(void) +{ + CK_SLOT_ID slot = 0; + + peakCap = LARGE_VALUE_SZ; + if (wolfSSL_SetAllocators(track_malloc, track_free, track_realloc) != 0) + return CHILD_SETUP; + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + if (init_slot(&slot) == CKR_OK) + (void)funcList->C_Finalize(NULL); + if (peakLive >= (size_t)LARGE_VALUE_SZ + LARGE_VALUE_SZ / 2) { + fprintf(stderr, " peak live bytes while loading: %lu\n", + (unsigned long)peakLive); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +static void test_large_stored_value_peak(void) +{ + CK_RV rv; + int res; + + printf("\n--- large stored value loads without a second copy ---\n"); + rv = store_objects(create_large_data_object); + CHECK_RV(rv, "store large data object", CKR_OK); + if (rv != CKR_OK) + return; + res = run_in_child(child_large_value_peak); + CHECK_TRUE(res == CHILD_PASS, + "peak memory while loading stays near one copy"); + cleanup_test_files(); +} +#endif + +#ifdef HAVE_ALLOC_TRACKING +#define MAX_TRACKED 8192 + +typedef struct TrackedAlloc { + void* ptr; + size_t sz; +} TrackedAlloc; + +static TrackedAlloc tracked[MAX_TRACKED]; +static const byte* secretNeedle = NULL; +static size_t secretNeedleSz = 0; +static int secretFreed = 0; + +static void track_add(void* ptr, size_t sz) +{ + int i; + + for (i = 0; ptr != NULL && i < MAX_TRACKED; i++) { + if (tracked[i].ptr == NULL) { + tracked[i].ptr = ptr; + tracked[i].sz = sz; + break; + } + } +} + +static int track_find(void* ptr) +{ + int i; + + for (i = 0; ptr != NULL && i < MAX_TRACKED; i++) { + if (tracked[i].ptr == ptr) + return i; + } + return -1; +} + +static int holds_secret(const byte* buf, size_t sz) +{ + size_t i; + + for (i = 0; sz >= secretNeedleSz && i <= sz - secretNeedleSz; i++) { + if (XMEMCMP(buf + i, secretNeedle, secretNeedleSz) == 0) + return 1; + } + return 0; +} + +static void* scan_malloc(size_t sz) +{ + void* ptr = malloc(sz); + + /* Zeroed so a scan only sees bytes the library wrote. */ + if (ptr != NULL) + XMEMSET(ptr, 0, sz); + track_add(ptr, sz); + return ptr; +} + +static void scan_free(void* ptr) +{ + int i = track_find(ptr); + + if (i >= 0) { + if (holds_secret((const byte*)ptr, tracked[i].sz)) + secretFreed++; + tracked[i].ptr = NULL; + } + free(ptr); +} + +static void* scan_realloc(void* ptr, size_t sz) +{ + int i = track_find(ptr); + int oldKnown = (ptr == NULL || i >= 0); + size_t oldSz = (i >= 0) ? tracked[i].sz : 0; + void* newPtr = realloc(ptr, sz); + + if (newPtr != NULL) { + if (i >= 0) + tracked[i].ptr = NULL; + if (oldKnown && sz > oldSz) + XMEMSET((byte*)newPtr + oldSz, 0, sz - oldSz); + track_add(newPtr, sz); + } + return newPtr; +} + +/* Logging in with a stored private key that fails authentication must not + * release decrypted key material to the heap. */ +static int child_no_key_material_freed(void) +{ + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + + if (wolfSSL_SetAllocators(scan_malloc, scan_free, scan_realloc) != 0) + return CHILD_SETUP; + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + if (init_slot(&slot) == CKR_OK) { + (void)open_session(slot, 1, &session); + (void)funcList->C_Finalize(NULL); + } + if (secretFreed != 0) { + fprintf(stderr, " %d freed buffer(s) held key material\n", + secretFreed); + return CHILD_FAIL; + } + return CHILD_PASS; +} + +/* Flip the last byte of the first length-prefixed array (the GCM tag). */ +static int corrupt_first_array_tag(const char* path) +{ + static byte data[MAX_FILE_SZ]; + size_t sz = 0; + word32 len; + + if (read_file(path, data, sizeof(data), &sz) != 0 || sz < 4) + return -1; + len = get_be32(data); + if (len == 0 || len > sz - 4) + return -1; + data[4 + len - 1] ^= 0x01; + return write_file(path, data, sz); +} + +static void check_failed_decrypt_scrubbed(const char* name, create_fn create, + const char* kind, + const byte* secret, size_t secretSz) +{ + CK_RV rv; + char path[256]; + char msg[128]; + int res = CHILD_SETUP; + + rv = store_objects(create); + (void)snprintf(msg, sizeof(msg), "store %s", name); + CHECK_RV(rv, msg, CKR_OK); + if (rv != CKR_OK) + return; + secretNeedle = secret; + secretNeedleSz = secretSz; + if (find_store_file(kind, path, sizeof(path)) == 0 && + corrupt_first_array_tag(path) == 0) { + res = run_in_child(child_no_key_material_freed); + } + (void)snprintf(msg, sizeof(msg), + "%s buffer is cleared after failed decrypt", name); + CHECK_TRUE(res == CHILD_PASS, msg); + cleanup_test_files(); +} + +static void test_failed_decrypt_scrubbed(void) +{ + printf("\n--- private key buffer cleared after failed decrypt ---\n"); +#ifdef HAVE_ECC + check_failed_decrypt_scrubbed("ECC private key", create_ecc_private_key, + "ecckey_priv", ecc_p256_priv, sizeof(ecc_p256_priv)); +#endif +#ifndef NO_RSA + check_failed_decrypt_scrubbed("RSA private key", create_rsa_private_key, + "rsakey_priv", rsa_2048_priv_exp, 32); +#endif +} + +#ifndef NO_AES +static byte aesKeyValue[32] = { + 0x5b, 0x1e, 0xc4, 0x27, 0x9a, 0x30, 0xe8, 0x71, + 0x0d, 0xb6, 0x43, 0xf9, 0x82, 0x2c, 0x95, 0x6e, + 0xa7, 0x14, 0xd0, 0x3b, 0x68, 0xcf, 0x51, 0x0a, + 0xe3, 0x9d, 0x26, 0x7c, 0xb1, 0x48, 0xf5, 0x13 +}; + +static CK_RV create_aes_key(CK_SESSION_HANDLE session) +{ + CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VALUE, aesKeyValue, sizeof(aesKeyValue) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); +} +#endif + +#ifndef NO_AES +/* Count the AES keys visible after a login and check one still encrypts. */ +static CK_RV use_stored_aes_key(CK_SESSION_HANDLE session) +{ + CK_RV rv; + CK_OBJECT_CLASS keyClass = CKO_SECRET_KEY; + CK_KEY_TYPE keyType = CKK_AES; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &keyClass, sizeof(keyClass) }, + { CKA_KEY_TYPE, &keyType, sizeof(keyType) }, + }; + CK_OBJECT_HANDLE key = CK_INVALID_HANDLE; + CK_ULONG count = 0; + byte iv[16]; + byte plain[16]; + byte enc[32]; + CK_ULONG encLen = sizeof(enc); + CK_MECHANISM mech = { CKM_AES_CBC, iv, sizeof(iv) }; + + XMEMSET(iv, 0, sizeof(iv)); + XMEMSET(plain, 0, sizeof(plain)); + rv = funcList->C_FindObjectsInit(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl)); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, &key, 1, &count); + (void)funcList->C_FindObjectsFinal(session); + } + if (rv == CKR_OK && count != 1) + rv = CKR_OBJECT_HANDLE_INVALID; + if (rv == CKR_OK) + rv = funcList->C_EncryptInit(session, &mech, key); + if (rv == CKR_OK) + rv = funcList->C_Encrypt(session, plain, sizeof(plain), enc, &encLen); + return rv; +} + +/* A logout whose re-encryption of a stored key fails must leave the stored + * record in place, so the next login still decodes it. */ +static int child_login_after_failed_encode(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + int res = CHILD_FAIL; + + if (wolfSSL_SetAllocators(track_malloc, track_free, track_realloc) != 0) + return CHILD_SETUP; + if (pkcs11_load() != CKR_OK) + return CHILD_SETUP; + rv = init_slot(&slot); + if (rv == CKR_OK) + rv = open_session(slot, 1, &session); + if (rv != CKR_OK) + return CHILD_SETUP; + refuseAllocs = 1; + (void)funcList->C_Logout(session); + refuseAllocs = 0; + rv = funcList->C_Login(session, CKU_USER, userPin, sizeof(userPin) - 1); + if (rv == CKR_OK) + rv = use_stored_aes_key(session); + if (rv == CKR_OK) + res = CHILD_PASS; + else + fprintf(stderr, " login or key use after failed encode: 0x%lx\n", + (unsigned long)rv); + (void)funcList->C_Finalize(NULL); + return res; +} + +static void test_login_after_failed_encode(void) +{ + CK_RV rv; + int res = CHILD_SETUP; + + printf("\n--- stored key survives a failed re-encryption at logout ---\n"); + rv = store_objects(create_aes_key); + CHECK_RV(rv, "store AES key", CKR_OK); + if (rv == CKR_OK) + res = run_in_child(child_login_after_failed_encode); + CHECK_TRUE(res == CHILD_PASS, "login decodes the key after a failed encode"); + cleanup_test_files(); +} +#endif + +static void test_failed_key_decrypt_scrubbed(void) +{ + printf("\n--- secret key storage cleared after failed decrypt ---\n"); +#ifndef NO_AES + check_failed_decrypt_scrubbed("AES key", create_aes_key, "symmkey", + aesKeyValue, sizeof(aesKeyValue)); +#endif +#ifndef NO_DH + check_failed_decrypt_scrubbed("DH private key value", create_dh_private_key, + "dhkey_priv", dh_2048_priv, sizeof(dh_2048_priv)); +#endif +} +#endif + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); + + printf("=== wolfPKCS11 stored object decode test ===\n"); + test_negative_optional_length(); + test_negative_attribute_length(); +#ifdef HAVE_ALLOC_TRACKING + test_oversized_attribute_length(); +#endif + test_large_stored_value_round_trip(); +#ifdef HAVE_ALLOC_TRACKING + test_large_stored_value_peak(); +#endif + test_truncated_data_record(); + test_short_private_key_record(); +#ifndef NO_DH + test_oversized_public_key_record(); +#endif +#ifdef WOLFPKCS11_NSS + test_trust_record_length(); +#endif +#ifdef HAVE_ALLOC_TRACKING + test_failed_decrypt_scrubbed(); + test_failed_key_decrypt_scrubbed(); +#ifndef NO_AES + test_login_after_failed_encode(); +#endif +#endif + + return pkcs11_test_summary(); +} + +#else + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + printf("File-backed token storage not available, skipping test\n"); + return 77; +} + +#endif diff --git a/tests/pkcs11mtt.c b/tests/pkcs11mtt.c index c82c9d00..0d3a4788 100644 --- a/tests/pkcs11mtt.c +++ b/tests/pkcs11mtt.c @@ -40,7 +40,8 @@ #include -#if defined(_POSIX_THREADS) && !defined(SINGLE_THREADED) +#if defined(_POSIX_THREADS) && !defined(SINGLE_THREADED) && \ + !defined(WOLFPKCS11_SINGLE_THREADED) #include #define TEST_MULTITHREADED @@ -238,8 +239,9 @@ static CK_RV test_object(void* args) { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, }; CK_ULONG tmplOnTokenCnt = sizeof(tmplOnToken) / sizeof(*tmplOnToken); + static byte copyLabel[] = "copy"; CK_ATTRIBUTE copyTmpl[] = { - { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_LABEL, copyLabel, sizeof(copyLabel)-1 }, }; CK_ULONG copyTmplCnt = sizeof(copyTmpl) / sizeof(*copyTmpl); CK_ULONG count; @@ -282,6 +284,18 @@ static CK_RV test_object(void* args) { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, { CKA_TOKEN, &ckTrue, 0 }, }; + CK_ATTRIBUTE tokenNullCreate[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_TOKEN, NULL, sizeof(CK_BBOOL) }, + }; + CK_ATTRIBUTE tokenBadLenCreate[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_TOKEN, &ckTrue, 0 }, + }; CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE; CK_OBJECT_HANDLE objOnToken = CK_INVALID_HANDLE; CK_OBJECT_HANDLE copyObj = CK_INVALID_HANDLE; @@ -343,14 +357,14 @@ static CK_RV test_object(void* args) "Create Object zero len key class"); } if (ret == CKR_OK) { - count = sizeof(tokenNull) / sizeof(*tokenNull); - ret = funcList->C_CreateObject(session, tokenNull, count, &obj); + count = sizeof(tokenNullCreate) / sizeof(*tokenNullCreate); + ret = funcList->C_CreateObject(session, tokenNullCreate, count, &obj); CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, "Create Object NULL token"); } if (ret == CKR_OK) { - count = sizeof(tokenBadLen) / sizeof(*tokenBadLen); - ret = funcList->C_CreateObject(session, tokenBadLen, count, &obj); + count = sizeof(tokenBadLenCreate) / sizeof(*tokenBadLenCreate); + ret = funcList->C_CreateObject(session, tokenBadLenCreate, count, &obj); CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, "Create Object zero token len"); } @@ -1054,8 +1068,12 @@ static CK_RV test_encrypt_decrypt(void* args) } if (ret == CKR_OK) { ret = funcList->C_EncryptInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "AES-CBC Encrypt Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "AES-CBC Encrypt Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_EncryptInit(session, &mech, CK_INVALID_HANDLE); @@ -1124,8 +1142,12 @@ static CK_RV test_encrypt_decrypt(void* args) } if (ret == CKR_OK) { ret = funcList->C_DecryptInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "AES-CBC Decrypt Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "AES-CBC Decrypt Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_DecryptInit(session, &mech, CK_INVALID_HANDLE); @@ -1231,7 +1253,11 @@ static CK_RV test_digest(void* args) } if (ret == CKR_OK) { ret = funcList->C_DigestInit(session, NULL); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "Digest Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "Digest Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_Digest(CK_INVALID_HANDLE, data, dataSz, hash, @@ -1331,7 +1357,11 @@ static CK_RV test_sign_verify(void* args) } if (ret == CKR_OK) { ret = funcList->C_SignInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "HMAC Sign Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "HMAC Sign Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_SignInit(session, &mech, CK_INVALID_HANDLE); @@ -1397,7 +1427,11 @@ static CK_RV test_sign_verify(void* args) } if (ret == CKR_OK) { ret = funcList->C_VerifyInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "HMAC Verify Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "HMAC Verify Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_VerifyInit(session, &mech, CK_INVALID_HANDLE); @@ -1534,8 +1568,12 @@ static CK_RV test_recover(void* args) } if (ret == CKR_OK) { ret = funcList->C_VerifyRecoverInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "Verify Recover Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "Verify Recover Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_VerifyRecoverInit(session, &mech, CK_INVALID_HANDLE); @@ -3725,7 +3763,9 @@ static CK_RV test_attributes_ecc(void* args) if (ret == CKR_OK) { ret = funcList->C_GetAttributeValue(session, priv, eccPrivTmpl, eccPrivTmplCnt); - CHECK_CKR(ret, "Get Attributes EC Private Key NULL values"); + /* No public point is stored, so CKA_EC_POINT is unavailable. */ + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_TYPE_INVALID, + "Get Attributes EC Private Key NULL values"); } if (ret == CKR_OK) { CHECK_COND(eccPrivTmpl[0].ulValueLen == sizeof(ecc_p256_params), ret, @@ -3744,7 +3784,9 @@ static CK_RV test_attributes_ecc(void* args) if (ret == CKR_OK) { ret = funcList->C_GetAttributeValue(session, priv, eccPrivTmpl, eccPrivTmplCnt); - CHECK_CKR(ret, "Get Attributes EC Private Key values"); + /* No public point is stored, so CKA_EC_POINT is unavailable. */ + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_TYPE_INVALID, + "Get Attributes EC Private Key values"); } funcList->C_DestroyObject(session, priv); @@ -4026,7 +4068,7 @@ static CK_RV test_ecc_create_key_fail(void* args) ecc_p256_priv_key[3].ulValueLen = sizeof(paramsBadOid); ret = funcList->C_CreateObject(session, ecc_p256_priv_key, ecc_p256_priv_key_cnt, &obj); - CHECK_CKR_FAIL(ret, CKR_FUNCTION_FAILED, + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, "EC Private Key Create Object - bad OID in params"); ecc_p256_priv_key[3].pValue = ecc_p256_params; ecc_p256_priv_key[3].ulValueLen = sizeof(ecc_p256_params); diff --git a/tests/pkcs11test.c b/tests/pkcs11test.c index dce1bb21..a597d1f1 100644 --- a/tests/pkcs11test.c +++ b/tests/pkcs11test.c @@ -61,6 +61,11 @@ #define PKCS11TEST_FUNC_SESS_DECL(func) \ PKCS11TEST_CASE(func, TEST_FLAG_INIT | TEST_FLAG_TOKEN | TEST_FLAG_SESSION) +#ifndef WOLFPKCS11_NO_TIME + #define TOKEN_CLOCK_FLAG CKF_CLOCK_ON_TOKEN +#else + #define TOKEN_CLOCK_FLAG 0 +#endif #ifndef HAVE_PKCS11_STATIC static void* dlib; @@ -468,7 +473,7 @@ static CK_RV test_no_token_init(void* args) * that has never had C_InitToken called. Pre-fix wp11_Token_Init * unconditionally marked the token state INITIALIZED, so the old * test included CKF_TOKEN_INITIALIZED in the expected mask. */ - CK_FLAGS expFlags = CKF_RNG | CKF_CLOCK_ON_TOKEN; + CK_FLAGS expFlags = CKF_RNG | TOKEN_CLOCK_FLAG; int flags = CKF_SERIAL_SESSION | CKF_RW_SESSION; ret = funcList->C_GetTokenInfo(slot, &tokenInfo); @@ -695,7 +700,7 @@ static CK_RV test_token(void* args) CK_SESSION_HANDLE session = *(CK_SESSION_HANDLE*)args; CK_RV ret; CK_TOKEN_INFO tokenInfo; - CK_FLAGS expFlags = CKF_RNG | CKF_CLOCK_ON_TOKEN | CKF_TOKEN_INITIALIZED; + CK_FLAGS expFlags = CKF_RNG | TOKEN_CLOCK_FLAG | CKF_TOKEN_INITIALIZED; unsigned char label[32]; int flags = CKF_SERIAL_SESSION | CKF_RW_SESSION; @@ -1002,7 +1007,7 @@ static CK_RV test_login_logout(void* args) CK_SESSION_HANDLE session = *(CK_SESSION_HANDLE*)args; CK_RV ret = 0; CK_TOKEN_INFO tokenInfo; - CK_FLAGS expFlags = CKF_RNG | CKF_CLOCK_ON_TOKEN | CKF_LOGIN_REQUIRED | + CK_FLAGS expFlags = CKF_RNG | TOKEN_CLOCK_FLAG | CKF_LOGIN_REQUIRED | CKF_TOKEN_INITIALIZED | CKF_USER_PIN_INITIALIZED; funcList->C_Logout(session); @@ -2021,8 +2026,9 @@ static CK_RV test_object(void* args) { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, }; CK_ULONG tmplOnTokenCnt = sizeof(tmplOnToken) / sizeof(*tmplOnToken); + static byte copyLabel[] = "copy"; CK_ATTRIBUTE copyTmpl[] = { - { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_LABEL, copyLabel, sizeof(copyLabel)-1 }, }; CK_ULONG copyTmplCnt = sizeof(copyTmpl) / sizeof(*copyTmpl); CK_ULONG count; @@ -2065,6 +2071,18 @@ static CK_RV test_object(void* args) { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, { CKA_TOKEN, &ckTrue, 0 }, }; + CK_ATTRIBUTE tokenNullCreate[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_TOKEN, NULL, sizeof(CK_BBOOL) }, + }; + CK_ATTRIBUTE tokenBadLenCreate[] = { + { CKA_CLASS, &secretKeyClass, sizeof(secretKeyClass) }, + { CKA_KEY_TYPE, &genericKeyType, sizeof(genericKeyType) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_TOKEN, &ckTrue, 0 }, + }; CK_OBJECT_HANDLE obj, objOnToken, copyObj; CK_ULONG size; @@ -2124,14 +2142,14 @@ static CK_RV test_object(void* args) "Create Object zero len key class"); } if (ret == CKR_OK) { - count = sizeof(tokenNull) / sizeof(*tokenNull); - ret = funcList->C_CreateObject(session, tokenNull, count, &obj); + count = sizeof(tokenNullCreate) / sizeof(*tokenNullCreate); + ret = funcList->C_CreateObject(session, tokenNullCreate, count, &obj); CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, "Create Object NULL token"); } if (ret == CKR_OK) { - count = sizeof(tokenBadLen) / sizeof(*tokenBadLen); - ret = funcList->C_CreateObject(session, tokenBadLen, count, &obj); + count = sizeof(tokenBadLenCreate) / sizeof(*tokenBadLenCreate); + ret = funcList->C_CreateObject(session, tokenBadLenCreate, count, &obj); CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, "Create Object zero token len"); } @@ -2451,12 +2469,24 @@ static CK_RV test_copy_object_deep_copy(void* args) } } - /* Test 3: Verify independence (deep copy) by modifying original object */ + /* Test 3: Verify independence (deep copy) by modifying original object. + * Key material is read-only after creation, so modify CKA_ID instead. */ if (ret == CKR_OK) { - CK_ATTRIBUTE modifyOriginalTmpl[] = { + CK_ATTRIBUTE modifyValueTmpl[] = { { CKA_VALUE, modifiedKeyData, sizeof(modifiedKeyData) }, }; + + ret = funcList->C_SetAttributeValue(session, originalObj, + modifyValueTmpl, 1); + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_READ_ONLY, + "Modify original object key value is read-only"); + } + if (ret == CKR_OK) { + CK_ATTRIBUTE modifyOriginalTmpl[] = { + { CKA_ID, modifiedKeyData, + sizeof(modifiedKeyData) }, + }; CK_ULONG modifyOriginalTmplCnt = sizeof(modifyOriginalTmpl) / sizeof(*modifyOriginalTmpl); @@ -2467,45 +2497,44 @@ static CK_RV test_copy_object_deep_copy(void* args) } if (ret == CKR_OK) { - /* Get the modified original object's value */ - XMEMSET(origValue, 0, sizeof(origValue)); - getOriginalAttrs[0].pValue = origValue; - getOriginalAttrs[0].ulValueLen = sizeof(origValue); + /* Get the modified original object's ID */ + XMEMSET(origId, 0, sizeof(origId)); + getOriginalAttrs[1].pValue = origId; + getOriginalAttrs[1].ulValueLen = sizeof(origId); ret = funcList->C_GetAttributeValue(session, originalObj, - getOriginalAttrs, 1); - CHECK_CKR(ret, "Get modified original object value"); + &getOriginalAttrs[1], 1); + CHECK_CKR(ret, "Get modified original object ID"); } if (ret == CKR_OK) { - /* Get the copied object's value (should be unchanged) */ - XMEMSET(copiedValue, 0, sizeof(copiedValue)); - getCopiedAttrs[0].pValue = copiedValue; - getCopiedAttrs[0].ulValueLen = sizeof(copiedValue); + /* Get the copied object's ID (should be unchanged) */ + XMEMSET(copiedId, 0, sizeof(copiedId)); + getCopiedAttrs[1].pValue = copiedId; + getCopiedAttrs[1].ulValueLen = sizeof(copiedId); - ret = funcList->C_GetAttributeValue(session, copiedObj, getCopiedAttrs, - 1); - CHECK_CKR(ret, "Get copied object value after original " + ret = funcList->C_GetAttributeValue(session, copiedObj, + &getCopiedAttrs[1], 1); + CHECK_CKR(ret, "Get copied object ID after original " "modification"); } if (ret == CKR_OK) { - /* Verify original object has modified value */ - if (getOriginalAttrs[0].ulValueLen != sizeof(modifiedKeyData) || - XMEMCMP(origValue, modifiedKeyData, + /* Verify original object has modified ID */ + if (getOriginalAttrs[1].ulValueLen != sizeof(modifiedKeyData) || + XMEMCMP(origId, modifiedKeyData, sizeof(modifiedKeyData)) != 0) { ret = -1; - CHECK_CKR(ret, "Original object should have modified value"); + CHECK_CKR(ret, "Original object should have modified ID"); } } if (ret == CKR_OK) { - /* Verify copied object still has original value (proving deep copy) */ - if (getCopiedAttrs[0].ulValueLen != sizeof(keyData) || - XMEMCMP(copiedValue, keyData, - sizeof(keyData)) != 0) { + /* Verify copied object still has original ID (proving deep copy) */ + if (getCopiedAttrs[1].ulValueLen != sizeof(keyId) || + XMEMCMP(copiedId, keyId, sizeof(keyId)) != 0) { ret = -1; - CHECK_CKR(ret, "Copied object should retain original value " + CHECK_CKR(ret, "Copied object should retain original ID " "(deep copy test)"); } } @@ -4838,8 +4867,12 @@ static CK_RV test_encrypt_decrypt(void* args) } if (ret == CKR_OK) { ret = funcList->C_EncryptInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "AES-CBC Encrypt Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "AES-CBC Encrypt Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_EncryptInit(session, &mech, CK_INVALID_HANDLE); @@ -4908,8 +4941,12 @@ static CK_RV test_encrypt_decrypt(void* args) } if (ret == CKR_OK) { ret = funcList->C_DecryptInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "AES-CBC Decrypt Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "AES-CBC Decrypt Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_DecryptInit(session, &mech, CK_INVALID_HANDLE); @@ -5382,7 +5419,11 @@ static CK_RV test_digest_fail(void* args) } if (ret == CKR_OK) { ret = funcList->C_DigestInit(session, NULL); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "Digest Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "Digest Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_Digest(CK_INVALID_HANDLE, data, dataSz, hash, @@ -5522,7 +5563,11 @@ static CK_RV test_sign_verify(void* args) } if (ret == CKR_OK) { ret = funcList->C_SignInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "HMAC Sign Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "HMAC Sign Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_SignInit(session, &mech, CK_INVALID_HANDLE); @@ -5588,7 +5633,11 @@ static CK_RV test_sign_verify(void* args) } if (ret == CKR_OK) { ret = funcList->C_VerifyInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "HMAC Verify Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "HMAC Verify Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_VerifyInit(session, &mech, CK_INVALID_HANDLE); @@ -5781,8 +5830,12 @@ static CK_RV test_recover(void* args) } if (ret == CKR_OK) { ret = funcList->C_VerifyRecoverInit(session, NULL, key); +#ifdef WOLFPKCS11_PKCS11_V3_0 + CHECK_CKR(ret, "Verify Recover Init no mechanism cancels"); +#else CHECK_CKR_FAIL(ret, CKR_ARGUMENTS_BAD, "Verify Recover Init no mechanism"); +#endif } if (ret == CKR_OK) { ret = funcList->C_VerifyRecoverInit(session, &mech, CK_INVALID_HANDLE); @@ -9275,7 +9328,9 @@ static CK_RV test_attributes_ecc(void* args) if (ret == CKR_OK) { ret = funcList->C_GetAttributeValue(session, priv, eccPrivTmpl, eccPrivTmplCnt); - CHECK_CKR(ret, "Get Attributes EC Private Key NULL values"); + /* No public point is stored, so CKA_EC_POINT is unavailable. */ + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_TYPE_INVALID, + "Get Attributes EC Private Key NULL values"); } if (ret == CKR_OK) { CHECK_COND(eccPrivTmpl[0].ulValueLen == sizeof(ecc_p256_params), ret, @@ -9294,7 +9349,9 @@ static CK_RV test_attributes_ecc(void* args) if (ret == CKR_OK) { ret = funcList->C_GetAttributeValue(session, priv, eccPrivTmpl, eccPrivTmplCnt); - CHECK_CKR(ret, "Get Attributes EC Private Key values"); + /* No public point is stored, so CKA_EC_POINT is unavailable. */ + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_TYPE_INVALID, + "Get Attributes EC Private Key values"); } funcList->C_DestroyObject(session, priv); @@ -9865,7 +9922,7 @@ static CK_RV test_ecc_create_key_fail(void* args) ecc_p256_priv_key[3].ulValueLen = sizeof(paramsBadOid); ret = funcList->C_CreateObject(session, ecc_p256_priv_key, ecc_p256_priv_key_cnt, &obj); - CHECK_CKR_FAIL(ret, CKR_FUNCTION_FAILED, + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, "EC Private Key Create Object - bad OID in params"); ecc_p256_priv_key[3].pValue = ecc_p256_params; ecc_p256_priv_key[3].ulValueLen = sizeof(ecc_p256_params); @@ -10541,8 +10598,8 @@ static CK_RV aes_cbc_encrypt_data_test(CK_SESSION_HANDLE session, word32 outSz = sizeof(out); CK_OBJECT_HANDLE secret; CK_KEY_TYPE keyType = CKK_GENERIC_SECRET; - CK_ULONG secSz = outSz; CK_BYTE data[16] = { 0 }; + CK_ULONG secSz = sizeof(data); CK_AES_CBC_ENCRYPT_DATA_PARAMS aesParams = { { 0 }, data, sizeof(data) }; @@ -11498,6 +11555,7 @@ static CK_RV test_aes_cbc_pad(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE key, if (ret == CKR_OK) { ret = funcList->C_EncryptFinal(session, pOut, &encSz); CHECK_CKR(ret, "AES-CBC Pad Encrypt Final"); + encSz += cumSz; } if (ret == CKR_OK) { @@ -16538,7 +16596,7 @@ static CK_RV test_derive_tls12_master_key(void* args) { ret = funcList->C_DeriveKey(session, &mechanism, hBaseKey, derivedKeyTemplate, ulDerivedKeyTemplateCount, &hDerivedKey); - CHECK_CKR_FAIL(ret, CKR_MECHANISM_INVALID, "Invalid version"); + CHECK_CKR_FAIL(ret, CKR_MECHANISM_PARAM_INVALID, "Invalid version"); version.major = 3; } @@ -17592,8 +17650,9 @@ static CK_RV test_create_session_obj_ro_session(void* args) CK_ULONG tmplOnTokenCnt = sizeof(tmplOnToken) / sizeof(*tmplOnToken); CK_OBJECT_HANDLE obj = CK_INVALID_HANDLE, objOnToken = CK_INVALID_HANDLE; CK_OBJECT_HANDLE copyObj = CK_INVALID_HANDLE, copyBad = CK_INVALID_HANDLE; + static byte copyLabel[] = "copy"; CK_ATTRIBUTE copyTmpl[] = { - { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_LABEL, copyLabel, sizeof(copyLabel)-1 }, }; CK_ULONG copyTmplCnt = sizeof(copyTmpl) / sizeof(*copyTmpl); char newLabel[] = "updated"; diff --git a/tests/pkcs11v3test.c b/tests/pkcs11v3test.c index 6b9abd50..a0bd054e 100644 --- a/tests/pkcs11v3test.c +++ b/tests/pkcs11v3test.c @@ -3560,9 +3560,8 @@ static CK_RV test_hss_paramset_rejected(void* args) return ret; } -/* A failed re-import via C_SetAttributeValue (a bad CKA_VALUE) must leave the - * existing key intact: the object still verifies its original signature. This - * exercises the atomic temporary-key swap in WP11_Object_SetHssKey. */ +/* A re-import via C_SetAttributeValue is rejected because key material is + * read-only after creation, and the existing key stays usable. */ static CK_RV test_hss_reimport_atomic(void* args) { CK_SESSION_HANDLE session = *(CK_SESSION_HANDLE*)args; @@ -3590,7 +3589,7 @@ static CK_RV test_hss_reimport_atomic(void* args) CHECK_CKR(ret, "HSS key import for re-import test"); if (ret == CKR_OK) { ret = funcList->C_SetAttributeValue(session, obj, badVal, 1); - CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_READ_ONLY, "HSS re-import of invalid CKA_VALUE rejected"); } if (ret == CKR_OK) { @@ -3876,12 +3875,12 @@ static CK_RV test_xmss_paramset_setattr_rejected(void* args) /* A wrong OID via C_SetAttributeValue must be rejected, not a no-op. */ if (ret == CKR_OK) { ret = funcList->C_SetAttributeValue(session, obj, setBadOnly, 1); - CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_READ_ONLY, "XMSS set wrong CKA_PARAMETER_SET rejected"); } if (ret == CKR_OK) { ret = funcList->C_SetAttributeValue(session, obj, setBadWithVal, 2); - CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_VALUE_INVALID, + CHECK_CKR_FAIL(ret, CKR_ATTRIBUTE_READ_ONLY, "XMSS set wrong CKA_PARAMETER_SET + value rejected"); } /* The key-derived OID must be unchanged by the rejected sets. */ diff --git a/tests/session_op_lifecycle_test.c b/tests/session_op_lifecycle_test.c new file mode 100644 index 00000000..80571e58 --- /dev/null +++ b/tests/session_op_lifecycle_test.c @@ -0,0 +1,1558 @@ +/* session_op_lifecycle_test.c + * + * Copyright (C) 2006-2026 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Every way a session operation ends (completion, error, replacement or + * session close) must release and scrub the state the operation owned. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#define TEST_DIR "./store/session_op_lifecycle_test" + +#if defined(USE_WOLFSSL_MEMORY) && !defined(NO_WOLFSSL_MEMORY) && \ + !defined(WOLFSSL_STATIC_MEMORY) && !defined(WOLFSSL_DEBUG_MEMORY) + #define TRACK_ALLOCS +#endif + +/* Fixtures and helpers are shared by tests that compile in different builds. */ +#if defined(_MSC_VER) + #pragma warning(push) + #pragma warning(disable: 4101 4189 4505) +#elif defined(__GNUC__) + #pragma GCC diagnostic push + #pragma GCC diagnostic ignored "-Wunused-variable" + #pragma GCC diagnostic ignored "-Wunused-function" +#endif + +static CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; +static CK_KEY_TYPE aesType = CKK_AES; +static CK_KEY_TYPE genericType = CKK_GENERIC_SECRET; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; + +/* Uniform key bytes so the value is recognisable in any key schedule. */ +static byte aesKeyValue[16] = { + 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, + 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, 0xC3, 0xC3 +}; +static byte macKeyValue[32] = { + 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, + 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, + 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, + 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C, 0x3C +}; +static byte gcmIv[12] = { 0 }; +static byte gcmAad[48] = { 0 }; +static byte plainMarker[32] = { + 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, + 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, + 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, + 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B, 0x6B +}; + +#ifdef TRACK_ALLOCS +/* Allocation hooks: count live blocks and look for a watched byte pattern in + * every block handed back to the allocator. */ +#define ALLOC_HDR_SZ 16 + +static long liveBlocks = 0; +static long allocCount = 0; +static const byte* watchData = NULL; +static size_t watchLen = 0; +static int watchHit = 0; + +static void track_scan(const byte* p, size_t sz) +{ + size_t i; + + if (watchData == NULL || watchLen == 0 || sz < watchLen) + return; + for (i = 0; i + watchLen <= sz; i++) { + if (XMEMCMP(p + i, watchData, watchLen) == 0) { + watchHit = 1; + return; + } + } +} + +static void* track_malloc(size_t sz) +{ + byte* p = (byte*)malloc(sz + ALLOC_HDR_SZ); + + if (p == NULL) + return NULL; + /* Zeroed so a scan only sees bytes the library wrote. */ + XMEMSET(p, 0, sz + ALLOC_HDR_SZ); + XMEMCPY(p, &sz, sizeof(sz)); + liveBlocks++; + allocCount++; + return p + ALLOC_HDR_SZ; +} + +static void track_free(void* ptr) +{ + byte* p; + size_t sz; + + if (ptr == NULL) + return; + p = (byte*)ptr - ALLOC_HDR_SZ; + XMEMCPY(&sz, p, sizeof(sz)); + track_scan((byte*)ptr, sz); + liveBlocks--; + free(p); +} + +static void* track_realloc(void* ptr, size_t sz) +{ + void* n; + size_t oldSz; + + if (ptr == NULL) + return track_malloc(sz); + if (sz == 0) { + track_free(ptr); + return NULL; + } + n = track_malloc(sz); + if (n == NULL) + return NULL; + XMEMCPY(&oldSz, (byte*)ptr - ALLOC_HDR_SZ, sizeof(oldSz)); + XMEMCPY(n, ptr, (oldSz < sz) ? oldSz : sz); + track_free(ptr); + return n; +} + +static void watch_start(const byte* data, size_t len) +{ + watchData = data; + watchLen = len; + watchHit = 0; +} + +static int watch_stop(void) +{ + watchData = NULL; + watchLen = 0; + return watchHit; +} +#endif /* TRACK_ALLOCS */ + +static CK_RV create_secret(CK_SESSION_HANDLE session, CK_KEY_TYPE* type, + byte* value, CK_ULONG valueLen, + CK_OBJECT_HANDLE* key) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, type, sizeof(*type) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_DECRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_SIGN, &ckTrue, sizeof(ckTrue) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_VALUE, value, valueLen }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), key); +} + +#ifndef NO_RSA +static CK_RV create_rsa_public(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE* key) +{ + static CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + static CK_KEY_TYPE rsaType = CKK_RSA; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &rsaType, sizeof(rsaType) }, + { CKA_ENCRYPT, &ckTrue, sizeof(ckTrue) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_VERIFY_RECOVER, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_MODULUS, rsa_2048_modulus, sizeof(rsa_2048_modulus) }, + { CKA_PUBLIC_EXPONENT, rsa_2048_pub_exp, sizeof(rsa_2048_pub_exp) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), key); +} +#endif + +#ifdef HAVE_ECC +static CK_RV create_ec_public(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE* key) +{ + static CK_OBJECT_CLASS pubClass = CKO_PUBLIC_KEY; + static CK_KEY_TYPE ecType = CKK_EC; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &pubClass, sizeof(pubClass) }, + { CKA_KEY_TYPE, &ecType, sizeof(ecType) }, + { CKA_VERIFY, &ckTrue, sizeof(ckTrue) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + { CKA_EC_PARAMS, ecc_p256_params, sizeof(ecc_p256_params) }, + { CKA_EC_POINT, ecc_p256_pub, sizeof(ecc_p256_pub) }, + }; + + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), key); +} +#endif + +static CK_RV init_library(CK_SLOT_ID* slot) +{ + CK_RV rv; + CK_C_INITIALIZE_ARGS args; + CK_ULONG slotCount = 1; + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + rv = funcList->C_Initialize(&args); + if (rv == CKR_OK) + rv = funcList->C_GetSlotList(CK_TRUE, slot, &slotCount); + if (rv == CKR_OK && slotCount == 0) + rv = CKR_TOKEN_NOT_PRESENT; + return rv; +} + +static CK_RV open_with_keys(CK_SLOT_ID slot, CK_SESSION_HANDLE* session, + CK_OBJECT_HANDLE* aesKey, + CK_OBJECT_HANDLE* macKey) +{ + CK_RV rv; + + rv = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, session); +#ifndef NO_AES + if (rv == CKR_OK) { + rv = create_secret(*session, &aesType, aesKeyValue, + sizeof(aesKeyValue), aesKey); + } +#else + (void)aesKey; +#endif + if (rv == CKR_OK) { + rv = create_secret(*session, &genericType, macKeyValue, + sizeof(macKeyValue), macKey); + } + return rv; +} + +#if defined(_MSC_VER) + #pragma warning(pop) +#elif defined(__GNUC__) + #pragma GCC diagnostic pop +#endif + +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + !defined(NO_HMAC) && !defined(NO_SHA256) +/* Starting an operation of another category must release the resources held + * by the operation it replaces. */ +static void test_cross_category_init_releases_state(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; + CK_MECHANISM hmacMech = { CKM_SHA256_HMAC, NULL, 0 }; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; +#if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) + CK_OBJECT_HANDLE rsaKey = CK_INVALID_HANDLE; + byte label[24] = { 0 }; + CK_RSA_PKCS_OAEP_PARAMS oaepParams = { + CKM_SHA256, CKG_MGF1_SHA256, CKZ_DATA_SPECIFIED, label, sizeof(label) + }; + CK_MECHANISM oaepMech = { CKM_RSA_PKCS_OAEP, &oaepParams, + sizeof(oaepParams) }; +#endif +#ifdef HAVE_ECC + CK_OBJECT_HANDLE ecKey = CK_INVALID_HANDLE; + CK_MECHANISM ecdsaMech = { CKM_ECDSA, NULL, 0 }; +#endif + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; + long before; + + printf("\n--- replacing an operation releases its state ---\n"); + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.pAAD = gcmAad; + gcmParams.ulAADLen = sizeof(gcmAad); + gcmParams.ulTagBits = 128; + + rv = init_library(&slot); + CHECK_RV(rv, "initialize library", CKR_OK); + if (rv != CKR_OK) + goto out; + before = liveBlocks; + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM with AAD)", CKR_OK); + rv = funcList->C_VerifyInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_VerifyInit(HMAC) over active encrypt", CKR_OK); + + rv = funcList->C_DecryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_DecryptInit(AES-GCM with AAD)", CKR_OK); + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit over active decrypt", CKR_OK); + + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC) over active digest", CKR_OK); + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM) over active sign", CKR_OK); + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC) over active encrypt", CKR_OK); + + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM) over active sign", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, plainMarker, sizeof(plainMarker), + out, &outLen); + CHECK_RV(rv, "C_EncryptUpdate(AES-GCM)", CKR_OK); + watch_start(plainMarker, sizeof(plainMarker)); + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC) over multi-part encrypt", CKR_OK); + CHECK_TRUE(!watch_stop(), + "replaced operation scrubs buffered input before release"); + +#ifdef HAVE_ECC + rv = create_ec_public(session, &ecKey); + CHECK_RV(rv, "C_CreateObject(EC public)", CKR_OK); + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM) over active sign", CKR_OK); + rv = funcList->C_VerifyInit(session, &ecdsaMech, ecKey); + CHECK_RV(rv, "C_VerifyInit(ECDSA) over active encrypt", CKR_OK); + rv = funcList->C_DestroyObject(session, ecKey); + CHECK_RV(rv, "C_DestroyObject(EC)", CKR_OK); +#endif + +#if !defined(NO_RSA) && !defined(WC_NO_RSA_OAEP) + rv = create_rsa_public(session, &rsaKey); + CHECK_RV(rv, "C_CreateObject(RSA public)", CKR_OK); + rv = funcList->C_EncryptInit(session, &oaepMech, rsaKey); + CHECK_RV(rv, "C_EncryptInit(RSA-OAEP with label)", CKR_OK); + rv = funcList->C_DecryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_DecryptInit(AES-GCM) over active encrypt", CKR_OK); + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC) over active decrypt", CKR_OK); + rv = funcList->C_DestroyObject(session, rsaKey); + CHECK_RV(rv, "C_DestroyObject(RSA)", CKR_OK); +#endif + + rv = funcList->C_DestroyObject(session, aesKey); + CHECK_RV(rv, "C_DestroyObject(AES)", CKR_OK); + rv = funcList->C_DestroyObject(session, macKey); + CHECK_RV(rv, "C_DestroyObject(HMAC)", CKR_OK); + funcList->C_CloseSession(session); + session = CK_INVALID_HANDLE; + CHECK_TRUE(liveBlocks == before, + "replaced operations leave no allocation behind"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) && defined(HAVE_ECC) +/* A request that is rejected before it can start must leave the operation + * already active on the session untouched. */ +static void test_rejected_init_keeps_active_operation(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_MECHANISM cbcMech = { CKM_AES_CBC, NULL, 0 }; + CK_MECHANISM ecdsaMech = { CKM_ECDSA, NULL, 0 }; + CK_MECHANISM badMech = { CKM_VENDOR_DEFINED, NULL, 0 }; +#if !defined(NO_HMAC) && !defined(NO_SHA256) + CK_ULONG badMacLen = 5; + CK_MECHANISM hmacMech = { CKM_SHA256_HMAC, &badMacLen, + sizeof(badMacLen) }; +#endif +#if !defined(NO_RSA) && defined(WC_RSA_PSS) + CK_OBJECT_HANDLE rsaKey = CK_INVALID_HANDLE; + CK_RSA_PKCS_PSS_PARAMS pssParams = { CKM_VENDOR_DEFINED, CKG_MGF1_SHA256, + 32 }; + CK_MECHANISM pssMech = { CKM_RSA_PKCS_PSS, &pssParams, + sizeof(pssParams) }; +#endif + byte iv[16] = { 0 }; + byte out[32]; + CK_ULONG outLen; + CK_SLOT_ID slot; + + printf("\n--- a rejected Init keeps the active operation ---\n"); + cbcMech.pParameter = iv; + cbcMech.ulParameterLen = sizeof(iv); + + rv = init_library(&slot); + CHECK_RV(rv, "initialize library", CKR_OK); + if (rv != CKR_OK) + goto out; + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_EncryptInit(session, &cbcMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-CBC)", CKR_OK); + rv = funcList->C_VerifyInit(session, &ecdsaMech, macKey); + CHECK_RV(rv, "C_VerifyInit(ECDSA) with a secret key", + CKR_KEY_TYPE_INCONSISTENT); + rv = funcList->C_SignInit(session, &badMech, macKey); + CHECK_RV(rv, "C_SignInit(unknown mechanism)", CKR_MECHANISM_INVALID); +#if !defined(NO_HMAC) && !defined(NO_SHA256) + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC with bad output length)", + CKR_MECHANISM_PARAM_INVALID); +#endif +#if !defined(NO_RSA) && defined(WC_RSA_PSS) + rv = create_rsa_public(session, &rsaKey); + CHECK_RV(rv, "C_CreateObject(RSA public)", CKR_OK); + rv = funcList->C_VerifyInit(session, &pssMech, rsaKey); + CHECK_RV(rv, "C_VerifyInit(RSA-PSS with bad hash)", + CKR_MECHANISM_PARAM_INVALID); +#endif + + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, plainMarker, sizeof(plainMarker), + out, &outLen); + CHECK_RV(rv, "C_EncryptUpdate after rejected Inits", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_EncryptFinal(session, out, &outLen); + CHECK_RV(rv, "C_EncryptFinal after rejected Inits", CKR_OK); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AES_CBC) && \ + !defined(NO_SHA256) +static int finalize_watched(CK_SESSION_HANDLE session) +{ + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); + return watch_stop(); +} + +/* An operation that ends on an error must release and scrub its keyed and + * buffered state, just as a completed operation does. */ +static void test_failed_operation_scrubs_state(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + byte iv[16] = { 0 }; + CK_MECHANISM cbcMech = { CKM_AES_CBC, NULL, 0 }; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; + CK_ULONG bigLen = (CK_ULONG)0xFFFFFFFFUL; + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; + int i; + + printf("\n--- an operation ending on error scrubs its state ---\n"); + cbcMech.pParameter = iv; + cbcMech.ulParameterLen = sizeof(iv); + if (sizeof(CK_ULONG) <= sizeof(word32)) { + printf("SKIP: lengths beyond 32 bits need a 64-bit CK_ULONG\n"); + return; + } + bigLen += 16; + + for (i = 0; i < 2; i++) { + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + rv = funcList->C_EncryptInit(session, &cbcMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-CBC)", CKR_OK); + outLen = sizeof(out); + /* Watch from the failing call: it may free the state itself. */ + watch_start(aesKeyValue, sizeof(aesKeyValue)); + if (i == 0) { + rv = funcList->C_EncryptUpdate(session, out, bigLen, out, &outLen); + CHECK_RV(rv, "C_EncryptUpdate(oversized)", CKR_DATA_LEN_RANGE); + } + else { + rv = funcList->C_Encrypt(session, out, bigLen, out, &outLen); + CHECK_RV(rv, "C_Encrypt(oversized)", CKR_DATA_LEN_RANGE); + } + CHECK_TRUE(!finalize_watched(session), + "failed AES operation leaves no key schedule behind"); + session = CK_INVALID_HANDLE; + } + + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit(SHA-256)", CKR_OK); + rv = funcList->C_DigestUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_DigestUpdate", CKR_OK); + watch_start(plainMarker, sizeof(plainMarker)); + rv = funcList->C_DigestKey(session, CK_INVALID_HANDLE); + CHECK_RV(rv, "C_DigestKey(invalid handle)", CKR_OBJECT_HANDLE_INVALID); + CHECK_TRUE(!finalize_watched(session), + "failed digest leaves no buffered input behind"); + session = CK_INVALID_HANDLE; + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) +/* Re-initializing AES-GCM must release, scrubbed, any data a previous GCM + * operation left buffered on the session. */ +static void test_gcm_reinit_releases_buffer(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; + long before; + int hit; + + printf("\n--- GCM re-init releases buffered data ---\n"); + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.ulTagBits = 128; + + rv = init_library(&slot); + CHECK_RV(rv, "initialize library", CKR_OK); + if (rv != CKR_OK) + goto out; + before = liveBlocks; + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, plainMarker, sizeof(plainMarker), + out, &outLen); + CHECK_RV(rv, "C_EncryptUpdate(AES-GCM)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, gcmIv, sizeof(gcmIv), out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-GCM) ends the operation", CKR_OK); + + watch_start(plainMarker, sizeof(plainMarker)); + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + hit = watch_stop(); + CHECK_RV(rv, "C_EncryptInit(AES-GCM) again", CKR_OK); + CHECK_TRUE(!hit, "GCM re-init scrubs buffered data before release"); + + rv = funcList->C_DestroyObject(session, aesKey); + CHECK_RV(rv, "C_DestroyObject(AES)", CKR_OK); + rv = funcList->C_DestroyObject(session, macKey); + CHECK_RV(rv, "C_DestroyObject(HMAC)", CKR_OK); + funcList->C_CloseSession(session); + session = CK_INVALID_HANDLE; + CHECK_TRUE(liveBlocks == before, + "GCM re-init leaves no allocation behind"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) +/* A key that the cipher rejects during setup must not leave a half-built + * context behind, and the session must stay usable. */ +static void test_failed_key_setup_keeps_session_usable(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE oddKey = CK_INVALID_HANDLE; + byte oddValue[20]; + byte iv[16] = { 0 }; + CK_MECHANISM cbcMech = { CKM_AES_CBC, NULL, 0 }; + byte out[32]; + CK_ULONG outLen; + CK_SLOT_ID slot; +#ifdef TRACK_ALLOCS + long before = 0; +#endif + + printf("\n--- a failed key setup leaves the session usable ---\n"); + cbcMech.pParameter = iv; + cbcMech.ulParameterLen = sizeof(iv); + XMEMSET(oddValue, 0x5A, sizeof(oddValue)); + + rv = init_library(&slot); + CHECK_RV(rv, "initialize library", CKR_OK); + if (rv != CKR_OK) + goto out; +#ifdef TRACK_ALLOCS + before = liveBlocks; +#endif + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = create_secret(session, &aesType, oddValue, sizeof(oddValue), &oddKey); + if (rv != CKR_OK) { + printf("SKIP: token rejects AES keys of invalid length\n"); + } + else { + rv = funcList->C_EncryptInit(session, &cbcMech, oddKey); + CHECK_TRUE(rv != CKR_OK, "C_EncryptInit with an unusable key fails"); + rv = funcList->C_EncryptInit(session, &cbcMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-CBC) after failed setup", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, iv, sizeof(iv), out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-CBC) after failed setup", CKR_OK); + funcList->C_DestroyObject(session, oddKey); + } + + funcList->C_DestroyObject(session, aesKey); + funcList->C_DestroyObject(session, macKey); + funcList->C_CloseSession(session); + session = CK_INVALID_HANDLE; +#ifdef TRACK_ALLOCS + CHECK_TRUE(liveBlocks == before, + "failed key setup leaves no allocation behind"); +#endif + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if !defined(NO_AES) && (defined(HAVE_AESGCM) || defined(HAVE_AESECB) || \ + defined(HAVE_AESCCM)) +/* Destroying a key must end the operations that still read it, while + * operations that copied the key at Init time may finish. */ +static void test_destroyed_key_ends_dependent_operation(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE tmpKey = CK_INVALID_HANDLE; +#ifdef HAVE_AESGCM + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; +#endif +#ifdef HAVE_AESECB + CK_MECHANISM ecbMech = { CKM_AES_ECB, NULL, 0 }; +#endif +#ifdef HAVE_AESCCM + CK_CCM_PARAMS ccmParams; + CK_MECHANISM ccmMech = { CKM_AES_CCM, &ccmParams, sizeof(ccmParams) }; +#endif +#ifdef HAVE_AES_CBC + byte iv[16] = { 0 }; + CK_MECHANISM cbcMech = { CKM_AES_CBC, NULL, 0 }; +#endif + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; + + printf("\n--- destroying a key ends operations that read it ---\n"); +#ifdef HAVE_AESGCM + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.ulTagBits = 128; +#endif +#ifdef HAVE_AESCCM + XMEMSET(&ccmParams, 0, sizeof(ccmParams)); + ccmParams.ulDataLen = 16; + ccmParams.pIv = gcmIv; + ccmParams.ulIvLen = sizeof(gcmIv); + ccmParams.pAAD = gcmAad; + ccmParams.ulAADLen = sizeof(gcmAad); + ccmParams.ulMacLen = 16; +#endif +#ifdef HAVE_AES_CBC + cbcMech.pParameter = iv; + cbcMech.ulParameterLen = sizeof(iv); +#endif + + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + +#ifdef HAVE_AESGCM + rv = create_secret(session, &aesType, aesKeyValue, sizeof(aesKeyValue), + &tmpKey); + CHECK_RV(rv, "C_CreateObject(AES)", CKR_OK); + rv = funcList->C_EncryptInit(session, &gcmMech, tmpKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM)", CKR_OK); + rv = funcList->C_DestroyObject(session, tmpKey); + CHECK_RV(rv, "C_DestroyObject(active AES-GCM key)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-GCM) after key destroyed", + CKR_OPERATION_NOT_INITIALIZED); + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM) with another key", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-GCM) with another key", CKR_OK); +#endif +#ifdef HAVE_AESCCM + rv = create_secret(session, &aesType, aesKeyValue, sizeof(aesKeyValue), + &tmpKey); + CHECK_RV(rv, "C_CreateObject(AES)", CKR_OK); + rv = funcList->C_EncryptInit(session, &ccmMech, tmpKey); + CHECK_RV(rv, "C_EncryptInit(AES-CCM)", CKR_OK); + rv = funcList->C_DestroyObject(session, tmpKey); + CHECK_RV(rv, "C_DestroyObject(active AES-CCM key)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-CCM) after key destroyed", + CKR_OPERATION_NOT_INITIALIZED); + rv = funcList->C_EncryptInit(session, &ccmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-CCM) with another key", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-CCM) with another key", CKR_OK); +#endif +#ifdef HAVE_AESECB + rv = create_secret(session, &aesType, aesKeyValue, sizeof(aesKeyValue), + &tmpKey); + CHECK_RV(rv, "C_CreateObject(AES)", CKR_OK); + rv = funcList->C_DecryptInit(session, &ecbMech, tmpKey); + CHECK_RV(rv, "C_DecryptInit(AES-ECB)", CKR_OK); + rv = funcList->C_DestroyObject(session, tmpKey); + CHECK_RV(rv, "C_DestroyObject(active AES-ECB key)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Decrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Decrypt(AES-ECB) after key destroyed", + CKR_OPERATION_NOT_INITIALIZED); +#endif +#ifdef HAVE_AES_CBC + rv = create_secret(session, &aesType, aesKeyValue, sizeof(aesKeyValue), + &tmpKey); + CHECK_RV(rv, "C_CreateObject(AES)", CKR_OK); + rv = funcList->C_EncryptInit(session, &cbcMech, tmpKey); + CHECK_RV(rv, "C_EncryptInit(AES-CBC)", CKR_OK); + rv = funcList->C_DestroyObject(session, tmpKey); + CHECK_RV(rv, "C_DestroyObject(active AES-CBC key)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-CBC) with key copied at Init", CKR_OK); +#endif + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(TRACK_ALLOCS) && defined(WOLFSSL_HAVE_PRF) && !defined(NO_SHA256) +static CK_RV tls_mac_sign_init(CK_SESSION_HANDLE session, CK_OBJECT_HANDLE key) +{ + CK_TLS_MAC_PARAMS params; + CK_MECHANISM mech = { CKM_TLS_MAC, ¶ms, sizeof(params) }; + + params.prfHashMechanism = CKM_SHA256; + params.ulMacLength = 12; + params.ulServerOrClient = 1; + return funcList->C_SignInit(session, &mech, key); +} + +/* Input accumulated for a one-shot MAC must be scrubbed whenever its buffer + * is grown or released. */ +static void test_mac_input_scrubbed_on_release(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + byte sig[64]; + CK_ULONG sigLen; + CK_SLOT_ID slot; + int hit; + + printf("\n--- accumulated MAC input is scrubbed ---\n"); + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = tls_mac_sign_init(session, macKey); + CHECK_RV(rv, "C_SignInit(TLS MAC)", CKR_OK); + rv = funcList->C_SignUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_SignUpdate(TLS MAC)", CKR_OK); + watch_start(plainMarker, sizeof(plainMarker)); + rv = funcList->C_SignUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_SignUpdate(TLS MAC) again", CKR_OK); + sigLen = sizeof(sig); + rv = funcList->C_SignFinal(session, sig, &sigLen); + hit = watch_stop(); + CHECK_RV(rv, "C_SignFinal(TLS MAC)", CKR_OK); + CHECK_TRUE(!hit, "MAC input is scrubbed when its buffer is released"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} + +#define MAC_SMALL_PARTS 1024 + +/* Many one-byte MAC parts must not reallocate the buffered input each time. */ +static void test_mac_input_grows_linearly(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + byte part = 0x5a; + CK_SLOT_ID slot; + long allocs; + int i; + + printf("\n--- many small MAC parts grow the input buffer linearly ---\n"); + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = tls_mac_sign_init(session, macKey); + CHECK_RV(rv, "C_SignInit(TLS MAC)", CKR_OK); + allocs = allocCount; + for (i = 0; rv == CKR_OK && i < MAC_SMALL_PARTS; i++) + rv = funcList->C_SignUpdate(session, &part, 1); + allocs = allocCount - allocs; + CHECK_RV(rv, "one-byte C_SignUpdate(TLS MAC) parts", CKR_OK); + if (allocs > 64) + fprintf(stderr, " %ld allocations for %d parts\n", allocs, i); + CHECK_TRUE(allocs <= 64, "buffered MAC input is not reallocated per part"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(TRACK_ALLOCS) && defined(WOLFSSL_HAVE_PRF) && !defined(NO_SHA256) +/* Closing a session in the middle of a one-shot MAC must release, scrubbed, + * the input accumulated so far. */ +static void test_mac_input_released_on_close(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_SLOT_ID slot; + long before; + int hit; + + printf("\n--- closing a session releases accumulated MAC input ---\n"); + rv = init_library(&slot); + CHECK_RV(rv, "initialize library", CKR_OK); + if (rv != CKR_OK) + goto out; + before = liveBlocks; + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = tls_mac_sign_init(session, macKey); + CHECK_RV(rv, "C_SignInit(TLS MAC)", CKR_OK); + rv = funcList->C_SignUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_SignUpdate(TLS MAC)", CKR_OK); + + watch_start(plainMarker, sizeof(plainMarker)); + funcList->C_CloseSession(session); + hit = watch_stop(); + session = CK_INVALID_HANDLE; + CHECK_TRUE(!hit, "closed session scrubs accumulated MAC input"); + CHECK_TRUE(liveBlocks == before, + "closed session leaves no allocation behind"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#ifndef NO_SHA256 +/* Operation state is only available while a digest is in progress, and a + * refused request leaves the caller's length untouched. */ +static void test_operation_state_requires_active_digest(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; +#ifndef NO_HMAC + CK_MECHANISM hmacMech = { CKM_SHA256_HMAC, NULL, 0 }; +#endif + byte digest[32]; + CK_ULONG digestLen; + CK_ULONG stateLen; + CK_SLOT_ID slot; + + printf("\n--- operation state needs an active digest ---\n"); + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit(SHA-256)", CKR_OK); + digestLen = sizeof(digest); + rv = funcList->C_Digest(session, plainMarker, sizeof(plainMarker), digest, + &digestLen); + CHECK_RV(rv, "C_Digest completes the operation", CKR_OK); + stateLen = 0; + rv = funcList->C_GetOperationState(session, NULL, &stateLen); + CHECK_RV(rv, "C_GetOperationState after digest completed", + CKR_OPERATION_NOT_INITIALIZED); + +#ifndef NO_HMAC + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC)", CKR_OK); + stateLen = 1234; + rv = funcList->C_GetOperationState(session, NULL, &stateLen); + CHECK_RV(rv, "C_GetOperationState during HMAC", CKR_STATE_UNSAVEABLE); + CHECK_TRUE(stateLen == 1234, + "refused C_GetOperationState leaves the length untouched"); +#endif + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + !defined(NO_SHA256) +/* Restoring a saved digest state must release the operation it replaces. */ +static void test_restored_state_releases_operation(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; + byte state[1024]; + CK_ULONG stateLen; + byte digest[32]; + CK_ULONG digestLen; + CK_SLOT_ID slot; + long before; + + printf("\n--- restoring a saved state releases the replaced op ---\n"); + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.pAAD = gcmAad; + gcmParams.ulAADLen = sizeof(gcmAad); + gcmParams.ulTagBits = 128; + + rv = init_library(&slot); + CHECK_RV(rv, "initialize library", CKR_OK); + if (rv != CKR_OK) + goto out; + before = liveBlocks; + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit(SHA-256)", CKR_OK); + rv = funcList->C_DigestUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_DigestUpdate", CKR_OK); + stateLen = sizeof(state); + rv = funcList->C_GetOperationState(session, state, &stateLen); + CHECK_RV(rv, "C_GetOperationState", CKR_OK); + digestLen = sizeof(digest); + rv = funcList->C_DigestFinal(session, digest, &digestLen); + CHECK_RV(rv, "C_DigestFinal", CKR_OK); + + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM with AAD)", CKR_OK); + rv = funcList->C_SetOperationState(session, state, stateLen, 0, 0); + CHECK_RV(rv, "C_SetOperationState over active encrypt", CKR_OK); + digestLen = sizeof(digest); + rv = funcList->C_DigestFinal(session, digest, &digestLen); + CHECK_RV(rv, "C_DigestFinal of restored state", CKR_OK); + + rv = funcList->C_DestroyObject(session, aesKey); + CHECK_RV(rv, "C_DestroyObject(AES)", CKR_OK); + rv = funcList->C_DestroyObject(session, macKey); + CHECK_RV(rv, "C_DestroyObject(HMAC)", CKR_OK); + funcList->C_CloseSession(session); + session = CK_INVALID_HANDLE; + CHECK_TRUE(liveBlocks == before, + "restored state leaves no allocation behind"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#ifndef NO_SHA256 +/* A saved state that is rejected must leave the session's digest exactly as + * it was. */ +static void test_rejected_state_keeps_session(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; + CK_MECHANISM_TYPE badMech; + byte state[1024]; + byte bogus[1024]; + byte after[1024]; + CK_ULONG stateLen; + CK_ULONG afterLen; + CK_SLOT_ID slot; + + printf("\n--- a rejected saved state keeps the session state ---\n"); + rv = init_library(&slot); + if (rv == CKR_OK) { + rv = funcList->C_OpenSession(slot, + CKF_SERIAL_SESSION | CKF_RW_SESSION, NULL, NULL, &session); + } + CHECK_RV(rv, "open session", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit(SHA-256)", CKR_OK); + rv = funcList->C_DigestUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_DigestUpdate", CKR_OK); + stateLen = sizeof(state); + rv = funcList->C_GetOperationState(session, state, &stateLen); + CHECK_RV(rv, "C_GetOperationState", CKR_OK); + if (rv != CKR_OK) + goto out; + + XMEMCPY(bogus, state, stateLen); + badMech = CKM_AES_CBC; + XMEMCPY(bogus, &badMech, sizeof(badMech)); + rv = funcList->C_SetOperationState(session, bogus, stateLen, 0, 0); + CHECK_RV(rv, "C_SetOperationState(unsupported mechanism)", + CKR_SAVED_STATE_INVALID); + badMech = CKM_SHA256; + XMEMCPY(bogus, &badMech, sizeof(badMech)); + rv = funcList->C_SetOperationState(session, bogus, sizeof(badMech) + 4, 0, + 0); + CHECK_RV(rv, "C_SetOperationState(truncated state)", + CKR_SAVED_STATE_INVALID); + + afterLen = sizeof(after); + rv = funcList->C_GetOperationState(session, after, &afterLen); + CHECK_RV(rv, "C_GetOperationState after rejected states", CKR_OK); + CHECK_TRUE(rv == CKR_OK && afterLen == stateLen && + XMEMCMP(after, state, stateLen) == 0, + "rejected saved state leaves the digest state unchanged"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#ifndef NO_SHA256 +/* Saved digest states need no key, so restoring one with a key handle is + * refused. */ +static void test_restore_with_key_is_refused(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; + byte state[1024]; + CK_ULONG stateLen; + CK_SLOT_ID slot; + + printf("\n--- restoring a digest state with a key is refused ---\n"); + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit(SHA-256)", CKR_OK); + rv = funcList->C_DigestUpdate(session, plainMarker, sizeof(plainMarker)); + CHECK_RV(rv, "C_DigestUpdate", CKR_OK); + stateLen = sizeof(state); + rv = funcList->C_GetOperationState(session, state, &stateLen); + CHECK_RV(rv, "C_GetOperationState", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_SetOperationState(session, state, stateLen, macKey, 0); + CHECK_RV(rv, "C_SetOperationState with encryption key", + CKR_KEY_NOT_NEEDED); + rv = funcList->C_SetOperationState(session, state, stateLen, 0, macKey); + CHECK_RV(rv, "C_SetOperationState with authentication key", + CKR_KEY_NOT_NEEDED); + rv = funcList->C_SetOperationState(session, state, stateLen, 0, 0); + CHECK_RV(rv, "C_SetOperationState without keys", CKR_OK); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(WOLFPKCS11_PKCS11_V3_0) && !defined(NO_AES) && \ + defined(HAVE_AES_CBC) && !defined(NO_HMAC) && !defined(NO_SHA256) +/* An Init call with a NULL mechanism terminates the active operation of that + * kind only, and a new operation of the same kind can then start. */ +static void test_null_mechanism_cancels_operation(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + byte iv[16] = { 0 }; + CK_MECHANISM cbcMech = { CKM_AES_CBC, NULL, 0 }; + CK_MECHANISM hmacMech = { CKM_SHA256_HMAC, NULL, 0 }; + CK_MECHANISM sha256Mech = { CKM_SHA256, NULL, 0 }; +#ifndef NO_RSA + CK_OBJECT_HANDLE rsaKey = CK_INVALID_HANDLE; + CK_MECHANISM rsaMech = { CKM_RSA_PKCS, NULL, 0 }; +#endif +#if defined(TRACK_ALLOCS) && defined(HAVE_AESGCM) + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; + long before; +#endif + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; + + printf("\n--- a NULL mechanism cancels the active operation ---\n"); + cbcMech.pParameter = iv; + cbcMech.ulParameterLen = sizeof(iv); + + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + + rv = funcList->C_EncryptInit(session, &cbcMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-CBC)", CKR_OK); + rv = funcList->C_EncryptInit(session, NULL, aesKey); + CHECK_RV(rv, "C_EncryptInit(NULL) cancels", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_EncryptUpdate after cancel", + CKR_OPERATION_NOT_INITIALIZED); + rv = funcList->C_EncryptInit(session, &cbcMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-CBC) after cancel", CKR_OK); + rv = funcList->C_EncryptInit(session, NULL, aesKey); + CHECK_RV(rv, "C_EncryptInit(NULL) cancels again", CKR_OK); + + rv = funcList->C_DecryptInit(session, &cbcMech, aesKey); + CHECK_RV(rv, "C_DecryptInit(AES-CBC)", CKR_OK); + rv = funcList->C_DecryptInit(session, NULL, aesKey); + CHECK_RV(rv, "C_DecryptInit(NULL) cancels", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_DecryptUpdate(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_DecryptUpdate after cancel", + CKR_OPERATION_NOT_INITIALIZED); + + rv = funcList->C_SignInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_SignInit(HMAC)", CKR_OK); + rv = funcList->C_SignInit(session, NULL, macKey); + CHECK_RV(rv, "C_SignInit(NULL) cancels", CKR_OK); + rv = funcList->C_SignUpdate(session, plainMarker, 16); + CHECK_RV(rv, "C_SignUpdate after cancel", CKR_OPERATION_NOT_INITIALIZED); + + rv = funcList->C_VerifyInit(session, &hmacMech, macKey); + CHECK_RV(rv, "C_VerifyInit(HMAC)", CKR_OK); +#ifndef NO_RSA + rv = funcList->C_VerifyRecoverInit(session, NULL, macKey); + CHECK_RV(rv, "C_VerifyRecoverInit(NULL) with a verify active", CKR_OK); + rv = funcList->C_VerifyUpdate(session, plainMarker, 16); + CHECK_RV(rv, "verify survives cancel of verify-recover", CKR_OK); +#endif + rv = funcList->C_VerifyInit(session, NULL, macKey); + CHECK_RV(rv, "C_VerifyInit(NULL) cancels", CKR_OK); + rv = funcList->C_VerifyUpdate(session, plainMarker, 16); + CHECK_RV(rv, "C_VerifyUpdate after cancel", + CKR_OPERATION_NOT_INITIALIZED); + +#ifndef NO_RSA + rv = create_rsa_public(session, &rsaKey); + CHECK_RV(rv, "C_CreateObject(RSA public)", CKR_OK); + rv = funcList->C_VerifyRecoverInit(session, &rsaMech, rsaKey); + CHECK_RV(rv, "C_VerifyRecoverInit(RSA)", CKR_OK); + rv = funcList->C_VerifyInit(session, NULL, rsaKey); + CHECK_RV(rv, "C_VerifyInit(NULL) with a verify-recover active", CKR_OK); + rv = funcList->C_VerifyRecoverInit(session, &rsaMech, rsaKey); + CHECK_RV(rv, "verify-recover survives cancel of verify", + CKR_OPERATION_ACTIVE); + rv = funcList->C_VerifyRecoverInit(session, NULL, rsaKey); + CHECK_RV(rv, "C_VerifyRecoverInit(NULL) cancels", CKR_OK); + rv = funcList->C_VerifyRecoverInit(session, &rsaMech, rsaKey); + CHECK_RV(rv, "C_VerifyRecoverInit(RSA) after cancel", CKR_OK); + rv = funcList->C_VerifyRecoverInit(session, NULL, rsaKey); + CHECK_RV(rv, "C_VerifyRecoverInit(NULL) cancels again", CKR_OK); +#endif + + rv = funcList->C_DigestInit(session, &sha256Mech); + CHECK_RV(rv, "C_DigestInit(SHA-256)", CKR_OK); + rv = funcList->C_EncryptInit(session, NULL, aesKey); + CHECK_RV(rv, "C_EncryptInit(NULL) with only a digest active", CKR_OK); + rv = funcList->C_DigestUpdate(session, plainMarker, 16); + CHECK_RV(rv, "digest survives cancel of another kind", CKR_OK); + rv = funcList->C_DigestInit(session, NULL); + CHECK_RV(rv, "C_DigestInit(NULL) cancels", CKR_OK); + rv = funcList->C_DigestUpdate(session, plainMarker, 16); + CHECK_RV(rv, "C_DigestUpdate after cancel", + CKR_OPERATION_NOT_INITIALIZED); + +#if defined(TRACK_ALLOCS) && defined(HAVE_AESGCM) + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.pAAD = gcmAad; + gcmParams.ulAADLen = sizeof(gcmAad); + gcmParams.ulTagBits = 128; + before = liveBlocks; + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM with AAD)", CKR_OK); + rv = funcList->C_EncryptInit(session, NULL, aesKey); + CHECK_RV(rv, "C_EncryptInit(NULL) cancels AES-GCM", CKR_OK); + CHECK_TRUE(liveBlocks == before, + "cancelled operation leaves no allocation behind"); +#endif + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(WOLFPKCS11_PKCS11_V3_0) && !defined(NO_AES) && defined(HAVE_AESGCM) +/* A NULL mechanism must still cancel, and release, an operation whose key + * was destroyed while the operation held buffered data. */ +static void test_null_mechanism_cancels_after_key_destroyed(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE tmpKey = CK_INVALID_HANDLE; + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; +#ifdef TRACK_ALLOCS + long before; + int hit; +#endif + + printf("\n--- a NULL mechanism cancels an operation with a destroyed key " + "---\n"); + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.ulTagBits = 128; + + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + CHECK_RV(rv, "open session with keys", CKR_OK); + if (rv != CKR_OK) + goto out; + +#ifdef TRACK_ALLOCS + before = liveBlocks; +#endif + rv = create_secret(session, &aesType, aesKeyValue, sizeof(aesKeyValue), + &tmpKey); + CHECK_RV(rv, "C_CreateObject(AES)", CKR_OK); + rv = funcList->C_EncryptInit(session, &gcmMech, tmpKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM)", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, plainMarker, sizeof(plainMarker), + out, &outLen); + CHECK_RV(rv, "C_EncryptUpdate(AES-GCM)", CKR_OK); + rv = funcList->C_DestroyObject(session, tmpKey); + CHECK_RV(rv, "C_DestroyObject(active AES-GCM key)", CKR_OK); + +#ifdef TRACK_ALLOCS + watch_start(plainMarker, sizeof(plainMarker)); +#endif + rv = funcList->C_EncryptInit(session, NULL, 0); + CHECK_RV(rv, "C_EncryptInit(NULL) after key destroyed", CKR_OK); +#ifdef TRACK_ALLOCS + hit = watch_stop(); + CHECK_TRUE(!hit, "cancel scrubs buffered plaintext before release"); + CHECK_TRUE(liveBlocks == before, + "cancel releases the operation of a destroyed key"); +#endif + + rv = funcList->C_EncryptInit(session, &gcmMech, aesKey); + CHECK_RV(rv, "C_EncryptInit(AES-GCM) with another key", CKR_OK); + outLen = sizeof(out); + rv = funcList->C_Encrypt(session, plainMarker, 16, out, &outLen); + CHECK_RV(rv, "C_Encrypt(AES-GCM) with another key", CKR_OK); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + !defined(WOLFSSL_AESGCM_STREAM) +/* Destroying the key of a multi-part AES-GCM operation must release the + * plaintext it buffered once the operation is next used. */ +static void test_destroyed_gcm_key_releases_input(void) +{ + CK_RV rv; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE aesKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE macKey = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE tmpKey = CK_INVALID_HANDLE; + CK_GCM_PARAMS gcmParams; + CK_MECHANISM gcmMech = { CKM_AES_GCM, &gcmParams, sizeof(gcmParams) }; + byte out[64]; + CK_ULONG outLen; + CK_SLOT_ID slot; + long before; + int hit; + + printf("\n--- destroying an AES-GCM key releases buffered input ---\n"); + XMEMSET(&gcmParams, 0, sizeof(gcmParams)); + gcmParams.pIv = gcmIv; + gcmParams.ulIvLen = sizeof(gcmIv); + gcmParams.ulIvBits = sizeof(gcmIv) * 8; + gcmParams.ulTagBits = 128; + + rv = init_library(&slot); + if (rv == CKR_OK) + rv = open_with_keys(slot, &session, &aesKey, &macKey); + if (rv == CKR_OK) { + rv = create_secret(session, &aesType, aesKeyValue, sizeof(aesKeyValue), + &tmpKey); + } + if (rv == CKR_OK) + rv = funcList->C_EncryptInit(session, &gcmMech, tmpKey); + if (rv == CKR_OK) { + outLen = sizeof(out); + rv = funcList->C_EncryptUpdate(session, plainMarker, + sizeof(plainMarker), out, &outLen); + } + if (rv == CKR_OK) + rv = funcList->C_DestroyObject(session, tmpKey); + CHECK_RV(rv, "buffer AES-GCM input and destroy its key", CKR_OK); + if (rv != CKR_OK) + goto out; + + before = liveBlocks; + watch_start(plainMarker, sizeof(plainMarker)); + outLen = sizeof(out); + rv = funcList->C_EncryptFinal(session, out, &outLen); + hit = watch_stop(); + CHECK_RV(rv, "C_EncryptFinal(AES-GCM) after key destroyed", + CKR_OPERATION_NOT_INITIALIZED); + CHECK_TRUE(liveBlocks < before, "buffered AES-GCM input is released"); + CHECK_TRUE(!hit, "buffered AES-GCM input is scrubbed"); + +out: + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} +#endif + +static int run_test(void) +{ + CK_RV rv; + +#ifdef TRACK_ALLOCS + if (wolfSSL_SetAllocators(track_malloc, track_free, track_realloc) != 0) { + CHECK_TRUE(0, "install allocation hooks"); + return -1; + } +#endif + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + !defined(NO_HMAC) && !defined(NO_SHA256) + test_cross_category_init_releases_state(); +#endif +#if !defined(NO_AES) && defined(HAVE_AES_CBC) && defined(HAVE_ECC) + test_rejected_init_keeps_active_operation(); +#endif +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AES_CBC) && \ + !defined(NO_SHA256) + test_failed_operation_scrubs_state(); +#endif +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) + test_gcm_reinit_releases_buffer(); +#endif +#if !defined(NO_AES) && defined(HAVE_AES_CBC) + test_failed_key_setup_keeps_session_usable(); +#endif +#if !defined(NO_AES) && (defined(HAVE_AESGCM) || defined(HAVE_AESECB) || \ + defined(HAVE_AESCCM)) + test_destroyed_key_ends_dependent_operation(); +#endif +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + !defined(WOLFSSL_AESGCM_STREAM) + test_destroyed_gcm_key_releases_input(); +#endif +#if defined(TRACK_ALLOCS) && defined(WOLFSSL_HAVE_PRF) && !defined(NO_SHA256) + test_mac_input_scrubbed_on_release(); + test_mac_input_grows_linearly(); + test_mac_input_released_on_close(); +#endif +#ifndef NO_SHA256 + test_operation_state_requires_active_digest(); + test_rejected_state_keeps_session(); + test_restore_with_key_is_refused(); +#endif +#if defined(WOLFPKCS11_PKCS11_V3_0) && !defined(NO_AES) && \ + defined(HAVE_AES_CBC) && !defined(NO_HMAC) && !defined(NO_SHA256) + test_null_mechanism_cancels_operation(); +#endif +#if defined(WOLFPKCS11_PKCS11_V3_0) && !defined(NO_AES) && defined(HAVE_AESGCM) + test_null_mechanism_cancels_after_key_destroyed(); +#endif +#if defined(TRACK_ALLOCS) && !defined(NO_AES) && defined(HAVE_AESGCM) && \ + !defined(NO_SHA256) + test_restored_state_releases_operation(); +#endif + + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + +#ifndef WOLFPKCS11_NO_ENV + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); +#endif + + printf("=== wolfPKCS11 session operation lifecycle test ===\n"); + run_test(); + return pkcs11_test_summary(); +} diff --git a/tests/set_pin_rekey_test.c b/tests/set_pin_rekey_test.c index 7adaadea..8686a048 100644 --- a/tests/set_pin_rekey_test.c +++ b/tests/set_pin_rekey_test.c @@ -32,7 +32,8 @@ #include #include #endif -#if defined(_POSIX_THREADS) && !defined(SINGLE_THREADED) && !defined(_WIN32) +#if defined(_POSIX_THREADS) && !defined(SINGLE_THREADED) && \ + !defined(WOLFPKCS11_SINGLE_THREADED) && !defined(_WIN32) #include #define SET_PIN_REKEY_THREADS #endif diff --git a/tests/token_store_test.c b/tests/token_store_test.c new file mode 100644 index 00000000..1be6d8bf --- /dev/null +++ b/tests/token_store_test.c @@ -0,0 +1,1335 @@ +/* token_store_test.c + * + * Copyright (C) 2006-2026 wolfSSL Inc. + * + * This file is part of wolfPKCS11. + * + * wolfPKCS11 is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * wolfPKCS11 is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA + * + * Token persistence must stay consistent with the in-memory token: every + * token object is persisted and reloaded, and a failure to persist token + * state is reported to the caller instead of being reported as success. + */ + +#ifdef HAVE_CONFIG_H + #include +#endif + +#include +#include +#include + +#ifndef WOLFSSL_USER_SETTINGS + #include +#endif +#include +#include + +#ifndef WOLFPKCS11_USER_SETTINGS + #include +#endif +#include + +#ifndef HAVE_PKCS11_STATIC +#include +#endif + +#include "testdata.h" +#include "pkcs11_test_util.h" + +#if !defined(WOLFPKCS11_NO_STORE) && !defined(WOLFPKCS11_TPM_STORE) && \ + !defined(WOLFPKCS11_CUSTOM_STORE) && !defined(WOLFPKCS11_NO_ENV) && \ + !defined(WOLFPKCS11_USER_ENV) && !defined(_WIN32) + #define TOKEN_STORE_FILE_TEST +#elif defined(WOLFPKCS11_TPM_STORE) && defined(DEBUG_WOLFPKCS11) && \ + !defined(WOLFPKCS11_NO_ENV) && !defined(WOLFPKCS11_USER_ENV) && \ + !defined(_WIN32) + #define TOKEN_STORE_TPM_TEST +#endif + +#if defined(TOKEN_STORE_FILE_TEST) || defined(TOKEN_STORE_TPM_TEST) + +#include +#include +#include + +#define TEST_DIR "./store/token_store_test" + +static const char* soPin = "password123456"; +static const char* userPin = "wolfpkcs11-test"; +static const char tokenLabel[] = "token-store-test"; +static CK_OBJECT_CLASS secretClass = CKO_SECRET_KEY; +static CK_KEY_TYPE genericType = CKK_GENERIC_SECRET; +static CK_BBOOL ckTrue = CK_TRUE; +static CK_BBOOL ckFalse = CK_FALSE; +static byte keyData[16] = { 0 }; + +#if defined(TOKEN_STORE_FILE_TEST) && !defined(WOLFSSL_STATIC_MEMORY) && \ + !defined(WOLFSSL_DEBUG_MEMORY) && !defined(NO_WOLFSSL_MEMORY) +#define TEST_ALLOC_FAILURE +static int failAllocs = 0; + +static void* test_malloc(size_t n) +{ + if (failAllocs) + return NULL; + return malloc(n); +} + +static void test_free(void* p) +{ + free(p); +} + +static void* test_realloc(void* p, size_t n) +{ + if (failAllocs) + return NULL; + return realloc(p, n); +} +#endif + +#ifdef TOKEN_STORE_FILE_TEST +static void clear_store_dir(void) +{ + DIR* dir; + struct dirent* ent; + char path[512]; + + (void)mkdir("./store", 0700); + (void)chmod(TEST_DIR, 0700); + dir = opendir(TEST_DIR); + if (dir == NULL) { + (void)mkdir(TEST_DIR, 0700); + return; + } + while ((ent = readdir(dir)) != NULL) { + if (strcmp(ent->d_name, ".") == 0 || strcmp(ent->d_name, "..") == 0) + continue; + snprintf(path, sizeof(path), "%s/%s", TEST_DIR, ent->d_name); + if (remove(path) != 0) + (void)rmdir(path); + } + closedir(dir); +} +#endif + +static CK_RV lib_init(void) +{ + CK_C_INITIALIZE_ARGS args; + + XMEMSET(&args, 0, sizeof(args)); + args.flags = CKF_OS_LOCKING_OK; + return funcList->C_Initialize(&args); +} + +static CK_RV first_slot(CK_SLOT_ID* slot) +{ + CK_RV rv; + CK_SLOT_ID slotList[16]; + CK_ULONG slotCount = sizeof(slotList) / sizeof(slotList[0]); + + rv = funcList->C_GetSlotList(CK_TRUE, slotList, &slotCount); + if (rv == CKR_OK && slotCount == 0) + rv = CKR_TOKEN_NOT_PRESENT; + if (rv == CKR_OK) + *slot = slotList[0]; + return rv; +} + +/* Initialize the library and provision a fresh token with a user PIN. */ +static CK_RV token_setup(CK_SLOT_ID* slot) +{ + CK_RV rv; + CK_SESSION_HANDLE soSession = CK_INVALID_HANDLE; + unsigned char label[32]; + +#ifdef TOKEN_STORE_FILE_TEST + clear_store_dir(); +#endif + rv = lib_init(); + if (rv == CKR_OK) + rv = first_slot(slot); + if (rv == CKR_OK) { + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, tokenLabel, XSTRLEN(tokenLabel)); + rv = funcList->C_InitToken(*slot, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin), label); + } + if (rv == CKR_OK) { + rv = funcList->C_OpenSession(*slot, + CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &soSession); + } + if (rv == CKR_OK) { + rv = funcList->C_Login(soSession, CKU_SO, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin)); + } + if (rv == CKR_OK) { + rv = funcList->C_InitPIN(soSession, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + } + if (soSession != CK_INVALID_HANDLE) { + funcList->C_Logout(soSession); + funcList->C_CloseSession(soSession); + } + return rv; +} + +static CK_RV user_session(CK_SLOT_ID slot, CK_SESSION_HANDLE* session) +{ + CK_RV rv; + + rv = funcList->C_OpenSession(slot, CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, session); + if (rv == CKR_OK) { + rv = funcList->C_Login(*session, CKU_USER, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + } + return rv; +} + +static void close_session(CK_SESSION_HANDLE session) +{ + if (session != CK_INVALID_HANDLE) { + funcList->C_Logout(session); + funcList->C_CloseSession(session); + } +} + +static CK_RV create_token_secret(CK_SESSION_HANDLE session, const char* label, + CK_OBJECT_HANDLE* obj) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, NULL, 0 }, + }; + + tmpl[5].pValue = (CK_VOID_PTR)label; + tmpl[5].ulValueLen = (CK_ULONG)XSTRLEN(label); + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), obj); +} + +static CK_ULONG count_label(CK_SESSION_HANDLE session, const char* label) +{ + CK_RV rv; + CK_OBJECT_HANDLE found[8]; + CK_ULONG foundCnt = 0; + CK_ATTRIBUTE tmpl[] = { + { CKA_LABEL, NULL, 0 }, + }; + + tmpl[0].pValue = (CK_VOID_PTR)label; + tmpl[0].ulValueLen = (CK_ULONG)XSTRLEN(label); + rv = funcList->C_FindObjectsInit(session, tmpl, 1); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, found, 8, &foundCnt); + funcList->C_FindObjectsFinal(session); + } + return (rv == CKR_OK) ? foundCnt : 0; +} + +#ifdef TOKEN_STORE_FILE_TEST +/* Path of the stored record of a token object. */ +static void object_record_path(CK_SLOT_ID slot, int objId, char* path, + size_t pathSz) +{ + snprintf(path, pathSz, "%s/wp11_obj_%016lx_%016lx", TEST_DIR, + (unsigned long)slot, (unsigned long)objId); +} + +/* Read a stored file into buf, returning its length or -1 on error. */ +static long read_store_file(const char* path, byte* buf, size_t bufSz) +{ + FILE* f; + long len; + + f = fopen(path, "rb"); + if (f == NULL) + return -1; + len = (long)fread(buf, 1, bufSz, f); + fclose(f); + return len; +} + +/* Replace a stored file with len bytes from buf. */ +static int write_store_file(const char* path, const byte* buf, size_t len) +{ + FILE* f; + size_t written; + + f = fopen(path, "wb"); + if (f == NULL) + return -1; + written = fwrite(buf, 1, len, f); + if (fclose(f) != 0 || written != len) + return -1; + return 0; +} + +/* Reload the token from storage and open a user session on it. */ +static CK_RV reload(CK_SLOT_ID slot, CK_SESSION_HANDLE* session) +{ + CK_RV rv; + + funcList->C_Finalize(NULL); + rv = lib_init(); + if (rv == CKR_OK) + rv = user_session(slot, session); + return rv; +} + +/* Token objects keep being persisted after storage is paused and resumed. */ +static void test_objects_persist_across_storage_pause(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + + printf("\n--- token objects persist across a storage pause ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "pause-first", &obj); + CHECK_RV(rv, "create first token object", CKR_OK); + } + if (rv == CKR_OK) { + XSETENV("WOLFPKCS11_NO_STORE", "1", 1); + rv = create_token_secret(session, "pause-second", &obj); + unsetenv("WOLFPKCS11_NO_STORE"); + CHECK_RV(rv, "create token object while storage is paused", CKR_OK); + } + if (rv == CKR_OK) { + rv = create_token_secret(session, "pause-third", &obj); + CHECK_RV(rv, "create token object after storage resumes", CKR_OK); + } + close_session(session); + session = CK_INVALID_HANDLE; + + if (rv == CKR_OK) { + rv = reload(slot, &session); + CHECK_RV(rv, "reload token", CKR_OK); + } + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "pause-first") == 1, + "object stored before the pause is reloaded"); + CHECK_TRUE(count_label(session, "pause-second") == 1, + "object created during the pause is reloaded"); + CHECK_TRUE(count_label(session, "pause-third") == 1, + "object created after the pause is reloaded"); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* A token object whose record cannot be committed is not reported created. */ +static void test_create_reports_commit_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + char path[512]; + int blocked = 0; + + printf("\n--- token object creation reports a commit failure ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "commit-first", &obj); + CHECK_RV(rv, "create first token object", CKR_OK); + } + if (rv == CKR_OK) { + object_record_path(slot, 1, path, sizeof(path)); + blocked = (mkdir(path, 0700) == 0); + CHECK_TRUE(blocked, "occupy the next object record path"); + } + if (blocked) { + rv = create_token_secret(session, "commit-second", &obj); + CHECK_TRUE(rv != CKR_OK, + "create fails when the object cannot be committed"); + CHECK_TRUE(count_label(session, "commit-second") == 0, + "uncommitted object is not in the token"); + CHECK_TRUE(count_label(session, "commit-first") == 1, + "committed object is still in the token"); + (void)rmdir(path); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +#if !defined(NO_AES) && defined(HAVE_AES_CBC) +static CK_KEY_TYPE aesType = CKK_AES; +static byte deriveBaseKey[16] = { + 0x4b, 0x1d, 0x8e, 0x3f, 0xa2, 0x57, 0xc0, 0x69, + 0x9d, 0x24, 0x6b, 0xf8, 0x13, 0xe5, 0x70, 0xac +}; +static byte deriveData[16] = { 0 }; + +/* Derive a token key from an AES base key with CKM_AES_CBC_ENCRYPT_DATA. */ +static CK_RV derive_token_secret(CK_SESSION_HANDLE session, + CK_OBJECT_HANDLE base, const char* label, + CK_OBJECT_HANDLE* obj) +{ + CK_MECHANISM mech; + CK_AES_CBC_ENCRYPT_DATA_PARAMS params; + CK_ULONG valueLen = sizeof(deriveData); + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_VALUE_LEN, &valueLen, sizeof(valueLen) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckFalse, sizeof(ckFalse) }, + { CKA_LABEL, NULL, 0 }, + }; + + XMEMSET(¶ms, 0, sizeof(params)); + params.pData = deriveData; + params.length = sizeof(deriveData); + mech.mechanism = CKM_AES_CBC_ENCRYPT_DATA; + mech.pParameter = ¶ms; + mech.ulParameterLen = sizeof(params); + tmpl[5].pValue = (CK_VOID_PTR)label; + tmpl[5].ulValueLen = (CK_ULONG)XSTRLEN(label); + return funcList->C_DeriveKey(session, &mech, base, tmpl, + sizeof(tmpl) / sizeof(*tmpl), obj); +} + +/* A derived token key whose record cannot be committed is not reported + * created, and deriving works again once it can be. */ +static void test_derive_reports_commit_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE base = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE derived = CK_INVALID_HANDLE; + char path[512]; + int blocked = 0; + CK_ATTRIBUTE baseTmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &aesType, sizeof(aesType) }, + { CKA_VALUE, deriveBaseKey, sizeof(deriveBaseKey) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_DERIVE, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, (CK_VOID_PTR)"derive-base", 11 }, + }; + + printf("\n--- derived token key reports a commit failure ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = funcList->C_CreateObject(session, baseTmpl, + sizeof(baseTmpl) / sizeof(*baseTmpl), + &base); + CHECK_RV(rv, "create base token key", CKR_OK); + } + if (rv == CKR_OK) { + object_record_path(slot, 1, path, sizeof(path)); + blocked = (mkdir(path, 0700) == 0); + CHECK_TRUE(blocked, "occupy the next object record path"); + } + if (blocked) { + rv = derive_token_secret(session, base, "derive-token", &derived); + CHECK_TRUE(rv != CKR_OK, + "derive fails when the key cannot be committed"); + CHECK_TRUE(count_label(session, "derive-token") == 0, + "uncommitted derived key is not in the token"); + CHECK_TRUE(count_label(session, "derive-base") == 1, + "base key is still in the token"); + (void)rmdir(path); + rv = derive_token_secret(session, base, "derive-token", &derived); + CHECK_RV(rv, "derive succeeds once the key can be committed", CKR_OK); + CHECK_TRUE(count_label(session, "derive-token") == 1, + "committed derived key is in the token"); + } + close_session(session); + funcList->C_Finalize(NULL); +} +#endif + +#ifdef DEBUG_WOLFPKCS11 +/* Debug-only hook exported by libwolfpkcs11. */ +extern void WP11_Test_StoreCloseFailAfter(int closes); + +/* A token object whose record cannot be closed is not reported created. */ +static void test_create_reports_close_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + + printf("\n--- token object creation reports a close failure ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "close-first", &obj); + CHECK_RV(rv, "create first token object", CKR_OK); + } + if (rv == CKR_OK) { + WP11_Test_StoreCloseFailAfter(0); + rv = create_token_secret(session, "close-second", &obj); + WP11_Test_StoreCloseFailAfter(-1); + CHECK_TRUE(rv != CKR_OK, + "create fails when the record cannot be closed"); + CHECK_TRUE(count_label(session, "close-second") == 0, + "unclosed object is not in the token"); + close_session(session); + session = CK_INVALID_HANDLE; + rv = reload(slot, &session); + CHECK_RV(rv, "reload token", CKR_OK); + } + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "close-first") == 1, + "stored object is reloaded"); + CHECK_TRUE(count_label(session, "close-second") == 0, + "unclosed object was not stored"); + rv = create_token_secret(session, "close-third", &obj); + CHECK_RV(rv, "create succeeds once records close", CKR_OK); + } + close_session(session); + funcList->C_Finalize(NULL); +} +#endif + +/* A failed token object store leaves the stored token record unchanged. */ +static void test_token_record_kept_on_failed_store(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + char path[512]; + char tokenPath[512]; + byte before[1024]; + byte after[1024]; + long beforeLen = -1; + long afterLen = -1; + int blocked = 0; + + printf("\n--- token record is kept when an object store fails ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "record-first", &obj); + CHECK_RV(rv, "create first token object", CKR_OK); + } + if (rv == CKR_OK) { + snprintf(tokenPath, sizeof(tokenPath), "%s/wp11_token_%016lx", + TEST_DIR, (unsigned long)slot); + beforeLen = read_store_file(tokenPath, before, sizeof(before)); + CHECK_TRUE(beforeLen > 0 && beforeLen < (long)sizeof(before), + "token record is stored"); + object_record_path(slot, 1, path, sizeof(path)); + blocked = (beforeLen > 0) && (mkdir(path, 0700) == 0); + } + if (blocked) { + rv = create_token_secret(session, "record-second", &obj); + CHECK_TRUE(rv != CKR_OK, + "create fails when the object cannot be committed"); + afterLen = read_store_file(tokenPath, after, sizeof(after)); + CHECK_TRUE(afterLen == beforeLen && + XMEMCMP(before, after, (size_t)beforeLen) == 0, + "token record is unchanged by the failed store"); + (void)rmdir(path); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* Some platforms, such as MSYS2, ignore a read-only mode on a directory. */ +static int block_store_dir(void) +{ + FILE* probe; + char path[256]; + + if (chmod(TEST_DIR, 0500) != 0) + return 0; + (void)snprintf(path, sizeof(path), "%s/wp11_probe", TEST_DIR); + probe = fopen(path, "wb"); + if (probe != NULL) { + fclose(probe); + (void)remove(path); + (void)chmod(TEST_DIR, 0700); + printf("Skipping: store directory cannot be made read-only\n"); + return 0; + } + return 1; +} + +/* A PIN change that cannot be stored leaves the stored PIN in effect. */ +static void test_pin_change_store_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + static const char* newPin = "wolfpkcs11-next"; + + if (geteuid() == 0) { + printf("\nSkipping PIN store failure test when run as root\n"); + return; + } + printf("\n--- PIN change that cannot be stored is not applied ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK && block_store_dir()) { + rv = funcList->C_SetPIN(session, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin), + (CK_UTF8CHAR_PTR)newPin, + (CK_ULONG)XSTRLEN(newPin)); + (void)chmod(TEST_DIR, 0700); + CHECK_TRUE(rv != CKR_OK, "PIN change reports the store failure"); + funcList->C_Logout(session); + rv = funcList->C_Login(session, CKU_USER, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + CHECK_RV(rv, "stored PIN still logs in", CKR_OK); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* A first user PIN that cannot be stored does not log in. */ +static void test_init_pin_store_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE soSession = CK_INVALID_HANDLE; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + unsigned char label[32]; + + if (geteuid() == 0) { + printf("\nSkipping user PIN store failure test when run as root\n"); + return; + } + printf("\n--- user PIN that cannot be stored is not applied ---\n"); + clear_store_dir(); + rv = lib_init(); + if (rv == CKR_OK) + rv = first_slot(&slot); + if (rv == CKR_OK) { + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, tokenLabel, XSTRLEN(tokenLabel)); + rv = funcList->C_InitToken(slot, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin), label); + } + if (rv == CKR_OK) { + rv = funcList->C_OpenSession(slot, + CKF_SERIAL_SESSION | CKF_RW_SESSION, + NULL, NULL, &soSession); + } + if (rv == CKR_OK) { + rv = funcList->C_Login(soSession, CKU_SO, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin)); + } + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK && block_store_dir()) { + rv = funcList->C_InitPIN(soSession, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin)); + (void)chmod(TEST_DIR, 0700); + CHECK_TRUE(rv != CKR_OK, "user PIN set reports the store failure"); + close_session(soSession); + soSession = CK_INVALID_HANDLE; + rv = user_session(slot, &session); + CHECK_TRUE(rv != CKR_OK, "unstored user PIN does not log in"); + } + close_session(soSession); + if (session != CK_INVALID_HANDLE) + funcList->C_CloseSession(session); + funcList->C_Finalize(NULL); +} + +/* A token reset that cannot be stored leaves the stored token in effect. */ +static void test_token_reset_store_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + unsigned char label[32]; + static const char* wrongPin = "not-the-so-pin"; + + if (geteuid() == 0) { + printf("\nSkipping token reset store failure test when run as root\n"); + return; + } + printf("\n--- token reset that cannot be stored is not applied ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "reset-keep", &obj); + CHECK_RV(rv, "create token object", CKR_OK); + } + close_session(session); + session = CK_INVALID_HANDLE; + XMEMSET(label, ' ', sizeof(label)); + XMEMCPY(label, tokenLabel, XSTRLEN(tokenLabel)); + if (rv == CKR_OK && block_store_dir()) { + rv = funcList->C_InitToken(slot, (CK_UTF8CHAR_PTR)soPin, + (CK_ULONG)XSTRLEN(soPin), label); + (void)chmod(TEST_DIR, 0700); + CHECK_TRUE(rv != CKR_OK, "token reset reports the store failure"); + rv = funcList->C_InitToken(slot, (CK_UTF8CHAR_PTR)wrongPin, + (CK_ULONG)XSTRLEN(wrongPin), label); + CHECK_TRUE(rv != CKR_OK, "token reset still requires the SO PIN"); + rv = user_session(slot, &session); + CHECK_RV(rv, "stored user PIN still logs in", CKR_OK); + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "reset-keep") == 1, + "stored token object is still present"); + } + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* C_Finalize reports a token that could not be stored. */ +static void test_finalize_reports_store_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + + if (geteuid() == 0) { + printf("\nSkipping finalize store failure test when run as root\n"); + return; + } + printf("\n--- finalize reports a token store failure ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "final-first", &obj); + CHECK_RV(rv, "create token object", CKR_OK); + } + close_session(session); + session = CK_INVALID_HANDLE; + if (rv == CKR_OK && block_store_dir()) { + rv = funcList->C_Finalize(NULL); + (void)chmod(TEST_DIR, 0700); + CHECK_RV(rv, "finalize reports the store failure", + CKR_FUNCTION_FAILED); + rv = lib_init(); + CHECK_RV(rv, "library initializes after the failed finalize", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "final-first") == 1, + "previously stored object is still present"); + } + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* The library initializes once an unreadable token record is repaired. */ +static void test_init_retry_after_load_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + char tokenPath[512]; + byte record[1024]; + long recordLen = -1; + + printf("\n--- library initializes after a failed initialization ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "retry-keep", &obj); + CHECK_RV(rv, "create token object", CKR_OK); + } + close_session(session); + session = CK_INVALID_HANDLE; + funcList->C_Finalize(NULL); + + if (rv == CKR_OK) { + snprintf(tokenPath, sizeof(tokenPath), "%s/wp11_token_%016lx", + TEST_DIR, (unsigned long)slot); + recordLen = read_store_file(tokenPath, record, sizeof(record)); + CHECK_TRUE(recordLen > 0 && recordLen < (long)sizeof(record), + "token record is stored"); + } + if (recordLen > 0) { + CHECK_TRUE(write_store_file(tokenPath, record, 4) == 0, + "truncate the token record"); + rv = lib_init(); + CHECK_TRUE(rv != CKR_OK, "initialize fails on a truncated record"); + if (rv == CKR_OK) + funcList->C_Finalize(NULL); + rv = write_store_file(tokenPath, record, (size_t)recordLen) == 0 ? + CKR_OK : CKR_GENERAL_ERROR; + if (rv == CKR_OK) + rv = lib_init(); + CHECK_RV(rv, "initialize succeeds once the record is restored", + CKR_OK); + if (rv == CKR_OK) { + rv = user_session(slot, &session); + CHECK_RV(rv, "user logs in after the retry", CKR_OK); + } + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "retry-keep") == 1, + "stored object is loaded after the retry"); + } + close_session(session); + funcList->C_Finalize(NULL); + } +} + +/* Destroying a token object reports stored data that cannot be removed. */ +static void test_destroy_reports_unstore_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE first; + CK_OBJECT_HANDLE second; + char keyPath[512]; + char innerPath[600]; + int blocked = 0; + + printf("\n--- destroy reports stored data that cannot be removed ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "destroy-first", &first); + CHECK_RV(rv, "create first token object", CKR_OK); + } + if (rv == CKR_OK) { + rv = create_token_secret(session, "destroy-second", &second); + CHECK_RV(rv, "create second token object", CKR_OK); + } + if (rv == CKR_OK) { + snprintf(keyPath, sizeof(keyPath), "%s/wp11_symmkey_%016lx_%016lx", + TEST_DIR, (unsigned long)slot, 1UL); + snprintf(innerPath, sizeof(innerPath), "%s/keep", keyPath); + blocked = remove(keyPath) == 0 && mkdir(keyPath, 0700) == 0 && + write_store_file(innerPath, keyData, sizeof(keyData)) == 0; + CHECK_TRUE(blocked, "make the stored key data unremovable"); + } + if (blocked) { + rv = funcList->C_DestroyObject(session, second); + CHECK_TRUE(rv != CKR_OK, + "destroy reports stored data that was not removed"); + CHECK_TRUE(count_label(session, "destroy-first") == 1, + "other token object is still present"); + (void)remove(innerPath); + (void)rmdir(keyPath); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* A token object never written to storage can still be destroyed. */ +static void test_destroy_unstored_object(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + + printf("\n--- destroy of an object created while paused ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + XSETENV("WOLFPKCS11_NO_STORE", "1", 1); + rv = create_token_secret(session, "paused-destroy", &obj); + unsetenv("WOLFPKCS11_NO_STORE"); + CHECK_RV(rv, "create token object while storage is paused", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_DestroyObject(session, obj); + CHECK_RV(rv, "destroy the never stored object", CKR_OK); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +#ifdef TEST_ALLOC_FAILURE +/* A library initialization that fails part way is undone and retried. */ +static void test_init_out_of_memory(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + + printf("\n--- library initializes after running out of memory ---\n"); + clear_store_dir(); + failAllocs = 1; + rv = lib_init(); + failAllocs = 0; + CHECK_TRUE(rv != CKR_OK, "initialize fails when allocation fails"); + if (rv == CKR_OK) + funcList->C_Finalize(NULL); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup once memory is available", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "init-retry", &obj); + CHECK_RV(rv, "create token object after the retry", CKR_OK); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* Object creation fails cleanly when memory cannot be allocated. */ +static void test_create_object_out_of_memory(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_VALUE, keyData, sizeof(keyData) }, + { CKA_TOKEN, &ckFalse, sizeof(ckFalse) }, + }; + + printf("\n--- object creation fails cleanly without memory ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + failAllocs = 1; + rv = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); + failAllocs = 0; + CHECK_TRUE(rv != CKR_OK, "create fails when allocation fails"); + rv = funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), &obj); + CHECK_RV(rv, "create succeeds once memory is available", CKR_OK); + } + close_session(session); + funcList->C_Finalize(NULL); +} +#endif + +/* A token record from before the trailing fields were added still loads. */ +static void test_load_record_without_trailing_fields(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + char tokenPath[512]; + byte record[1024]; + long recordLen = -1; + long dropLen; + + printf("\n--- token record without trailing fields loads ---\n"); + for (dropLen = 4; dropLen <= 8; dropLen += 4) { + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "legacy-keep", &obj); + CHECK_RV(rv, "create token object", CKR_OK); + } + close_session(session); + session = CK_INVALID_HANDLE; + funcList->C_Finalize(NULL); + if (rv == CKR_OK) { + snprintf(tokenPath, sizeof(tokenPath), "%s/wp11_token_%016lx", + TEST_DIR, (unsigned long)slot); + recordLen = read_store_file(tokenPath, record, sizeof(record)); + CHECK_TRUE(recordLen > dropLen && recordLen < (long)sizeof(record), + "token record is stored"); + } + if (rv == CKR_OK && recordLen > dropLen) { + CHECK_TRUE(write_store_file(tokenPath, record, + (size_t)(recordLen - dropLen)) == 0, + "drop trailing token record fields"); + rv = lib_init(); + CHECK_RV(rv, "initialize with the shorter record", CKR_OK); + if (rv == CKR_OK) { + rv = user_session(slot, &session); + CHECK_RV(rv, "user logs in", CKR_OK); + } + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "legacy-keep") == 1, + "stored object is loaded"); + } + close_session(session); + session = CK_INVALID_HANDLE; + funcList->C_Finalize(NULL); + } + } +} +#endif /* TOKEN_STORE_FILE_TEST */ + +#ifdef TOKEN_STORE_TPM_TEST +/* Debug-only hooks exported by libwolfpkcs11. */ +extern int WP11_Test_StoreWriteFailAfter(int writes); +extern int WP11_Test_StoreWriteFailOnce(int writes); +extern void WP11_Test_TpmStoreShortBy(int bytes); + +#define MAX_STORE_WRITES 500 + +static byte privateValue[16] = { + 0x5a, 0x0c, 0xe7, 0x31, 0x9b, 0x46, 0xd8, 0x2f, + 0x74, 0xa1, 0x1e, 0xc3, 0x68, 0xf5, 0x0d, 0xb2 +}; + +/* Create a private, readable token secret encrypted under the token key. */ +static CK_RV create_private_token_secret(CK_SESSION_HANDLE session, + const char* label, + CK_OBJECT_HANDLE* obj) +{ + CK_ATTRIBUTE tmpl[] = { + { CKA_CLASS, &secretClass, sizeof(secretClass) }, + { CKA_KEY_TYPE, &genericType, sizeof(genericType) }, + { CKA_VALUE, privateValue, sizeof(privateValue) }, + { CKA_TOKEN, &ckTrue, sizeof(ckTrue) }, + { CKA_PRIVATE, &ckTrue, sizeof(ckTrue) }, + { CKA_SENSITIVE, &ckFalse, sizeof(ckFalse) }, + { CKA_EXTRACTABLE, &ckTrue, sizeof(ckTrue) }, + { CKA_LABEL, NULL, 0 }, + }; + + tmpl[7].pValue = (CK_VOID_PTR)label; + tmpl[7].ulValueLen = (CK_ULONG)XSTRLEN(label); + return funcList->C_CreateObject(session, tmpl, + sizeof(tmpl) / sizeof(*tmpl), obj); +} + +/* The private secret with the label is found and its value decrypts. */ +static int private_secret_usable(CK_SESSION_HANDLE session, const char* label) +{ + CK_RV rv; + CK_OBJECT_HANDLE found[8]; + CK_ULONG foundCnt = 0; + byte value[sizeof(privateValue)]; + CK_ATTRIBUTE valueAttr = { CKA_VALUE, value, sizeof(value) }; + CK_ATTRIBUTE tmpl[] = { + { CKA_LABEL, NULL, 0 }, + }; + + tmpl[0].pValue = (CK_VOID_PTR)label; + tmpl[0].ulValueLen = (CK_ULONG)XSTRLEN(label); + rv = funcList->C_FindObjectsInit(session, tmpl, 1); + if (rv == CKR_OK) { + rv = funcList->C_FindObjects(session, found, 8, &foundCnt); + funcList->C_FindObjectsFinal(session); + } + if (rv == CKR_OK && foundCnt != 1) + rv = CKR_OBJECT_HANDLE_INVALID; + if (rv == CKR_OK) + rv = funcList->C_GetAttributeValue(session, found[0], &valueAttr, 1); + return rv == CKR_OK && valueAttr.ulValueLen == sizeof(privateValue) && + XMEMCMP(value, privateValue, sizeof(privateValue)) == 0; +} + +/* Reload the token from what storage holds after storing was made to fail. */ +static CK_RV reload_stored(CK_SLOT_ID slot, CK_SESSION_HANDLE* session) +{ + CK_RV rv; + + (void)funcList->C_Finalize(NULL); + (void)WP11_Test_StoreWriteFailAfter(-1); + rv = lib_init(); + if (rv == CKR_OK) + rv = user_session(slot, session); + return rv; +} + +/* An object create that fails at any storage write keeps the stored token. */ +static void test_interrupted_create_keeps_stored_token(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + int writes; + int intact = 1; + + printf("\n--- interrupted object create keeps the stored token ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "tpm-keep", &obj); + CHECK_RV(rv, "create stored token object", CKR_OK); + } + for (writes = 0; intact && rv == CKR_OK && writes < MAX_STORE_WRITES; + writes++) { + (void)WP11_Test_StoreWriteFailAfter(writes); + rv = create_token_secret(session, "tpm-new", &obj); + if (rv == CKR_OK) { + (void)WP11_Test_StoreWriteFailAfter(-1); + break; + } + close_session(session); + session = CK_INVALID_HANDLE; + rv = reload_stored(slot, &session); + if (rv != CKR_OK || count_label(session, "tpm-keep") != 1 || + count_label(session, "tpm-new") != 0) { + printf("Stored token changed by a store failing after %d " + "writes\n", writes); + intact = 0; + } + } + CHECK_TRUE(intact, "stored token is intact after each failed create"); + CHECK_TRUE(rv == CKR_OK && writes < MAX_STORE_WRITES, + "create succeeds once storing works"); + close_session(session); + funcList->C_Finalize(NULL); +} + +/* A PIN change that fails at any one storage write keeps the stored PIN. + * TPM records commit one at a time, so the old records are written back. */ +static void test_interrupted_pin_change_keeps_stored_pin(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + int writes; + int intact = 1; + static const char* newPin = "wolfpkcs11-next"; + + printf("\n--- interrupted PIN change keeps the stored PIN ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "tpm-pin-keep", &obj); + CHECK_RV(rv, "create stored token object", CKR_OK); + } + if (rv == CKR_OK) { + rv = create_private_token_secret(session, "tpm-pin-private", &obj); + CHECK_RV(rv, "create stored private token object", CKR_OK); + } + for (writes = 0; intact && rv == CKR_OK && writes < MAX_STORE_WRITES; + writes++) { + (void)WP11_Test_StoreWriteFailOnce(writes); + rv = funcList->C_SetPIN(session, (CK_UTF8CHAR_PTR)userPin, + (CK_ULONG)XSTRLEN(userPin), + (CK_UTF8CHAR_PTR)newPin, + (CK_ULONG)XSTRLEN(newPin)); + if (rv == CKR_OK) { + (void)WP11_Test_StoreWriteFailAfter(-1); + break; + } + close_session(session); + session = CK_INVALID_HANDLE; + rv = reload_stored(slot, &session); + /* Records under the token key must still open with the stored PIN. */ + if (rv != CKR_OK || count_label(session, "tpm-pin-keep") != 1 || + !private_secret_usable(session, "tpm-pin-private")) { + printf("Stored token changed by a store failing after %d " + "writes\n", writes); + intact = 0; + } + } + CHECK_TRUE(intact, "stored PIN and private key work after each failed " + "PIN change"); + CHECK_TRUE(rv == CKR_OK && writes < MAX_STORE_WRITES, + "PIN change succeeds once storing works"); + close_session(session); + funcList->C_Finalize(NULL); +} + +/* A record that outgrows its NV index is moved to a larger one with its + * data and reloads from there. */ +static void test_grown_record_reloads(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + char longLabel[200]; + CK_ATTRIBUTE labelAttr = { CKA_LABEL, longLabel, sizeof(longLabel) - 1 }; + + printf("\n--- record that outgrows its NV index reloads ---\n"); + XMEMSET(longLabel, 'g', sizeof(longLabel) - 1); + longLabel[sizeof(longLabel) - 1] = '\0'; + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "tpm-grow", &obj); + CHECK_RV(rv, "create stored token object", CKR_OK); + } + if (rv == CKR_OK) { + rv = funcList->C_SetAttributeValue(session, obj, &labelAttr, 1); + CHECK_RV(rv, "grow the object record past its NV index", CKR_OK); + } + close_session(session); + session = CK_INVALID_HANDLE; + if (rv == CKR_OK) { + rv = reload_stored(slot, &session); + CHECK_RV(rv, "reload token", CKR_OK); + } + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, longLabel) == 1, + "grown record reloads with its new data"); + CHECK_TRUE(count_label(session, "tpm-grow") == 0, + "old record is gone"); + } + close_session(session); + funcList->C_Finalize(NULL); +} + +/* A record written past its staging buffer is reported, not committed. */ +static void test_record_overflow_reports_failure(void) +{ + CK_RV rv; + CK_SLOT_ID slot = 0; + CK_SESSION_HANDLE session = CK_INVALID_HANDLE; + CK_OBJECT_HANDLE obj; + + printf("\n--- record overflowing its staging buffer is reported ---\n"); + rv = token_setup(&slot); + CHECK_RV(rv, "token setup", CKR_OK); + if (rv == CKR_OK) + rv = user_session(slot, &session); + if (rv == CKR_OK) { + rv = create_token_secret(session, "tpm-fit", &obj); + CHECK_RV(rv, "create stored token object", CKR_OK); + } + if (rv == CKR_OK) { + WP11_Test_TpmStoreShortBy(1); + rv = create_token_secret(session, "tpm-overflow", &obj); + WP11_Test_TpmStoreShortBy(0); + CHECK_TRUE(rv != CKR_OK, "create fails when a record overflows"); + close_session(session); + session = CK_INVALID_HANDLE; + rv = reload_stored(slot, &session); + CHECK_RV(rv, "reload token", CKR_OK); + } + if (rv == CKR_OK) { + CHECK_TRUE(count_label(session, "tpm-fit") == 1, + "stored object is still present"); + CHECK_TRUE(count_label(session, "tpm-overflow") == 0, + "overflowing object was not stored"); + rv = create_token_secret(session, "tpm-overflow", &obj); + CHECK_RV(rv, "create succeeds once records fit", CKR_OK); + } + close_session(session); + funcList->C_Finalize(NULL); +} +#endif /* TOKEN_STORE_TPM_TEST */ + +static int run_tests(void) +{ + CK_RV rv; + + rv = pkcs11_load(); + CHECK_RV(rv, "load library", CKR_OK); + if (rv != CKR_OK) + return -1; + +#ifdef TOKEN_STORE_FILE_TEST + test_objects_persist_across_storage_pause(); + test_create_reports_commit_failure(); +#if !defined(NO_AES) && defined(HAVE_AES_CBC) + test_derive_reports_commit_failure(); +#endif +#ifdef DEBUG_WOLFPKCS11 + test_create_reports_close_failure(); +#endif + test_token_record_kept_on_failed_store(); + test_pin_change_store_failure(); + test_init_pin_store_failure(); + test_token_reset_store_failure(); + test_finalize_reports_store_failure(); + test_init_retry_after_load_failure(); + test_destroy_reports_unstore_failure(); + test_destroy_unstored_object(); + test_load_record_without_trailing_fields(); +#ifdef TEST_ALLOC_FAILURE + test_init_out_of_memory(); + test_create_object_out_of_memory(); +#endif + + clear_store_dir(); +#else + test_interrupted_create_keeps_stored_token(); + test_interrupted_pin_change_keeps_stored_pin(); + test_grown_record_reloads(); + test_record_overflow_reports_failure(); +#endif + pkcs11_unload(); + return 0; +} + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + + XSETENV("WOLFPKCS11_TOKEN_PATH", TEST_DIR, 1); + unsetenv("WOLFPKCS11_NO_STORE"); + + printf("=== wolfPKCS11 token store test ===\n"); +#ifdef TEST_ALLOC_FAILURE + if (wolfSSL_SetAllocators(test_malloc, test_free, test_realloc) != 0) { + fprintf(stderr, "FAIL: wolfSSL_SetAllocators\n"); + return 1; + } +#endif + run_tests(); + return pkcs11_test_summary(); +} + +#else + +int main(int argc, char* argv[]) +{ + (void)argc; + (void)argv; + printf("Token store test requires the file-backed store or a debug " + "TPM store, skipping\n"); + return 77; +} + +#endif diff --git a/wolfpkcs11/internal.h b/wolfpkcs11/internal.h index 9d2aec7a..e222e1e4 100644 --- a/wolfpkcs11/internal.h +++ b/wolfpkcs11/internal.h @@ -348,6 +348,24 @@ C_EXTRA_FLAGS="-DWOLFSSL_PUBLIC_MP -DWC_RSA_DIRECT" #define WP11_MAX_PIN_LEN 32 #endif +/* Maximum size of storage for generated/derived symmetric key. */ +#ifdef WOLFPKCS11_NSS +#define WP11_MAX_SYM_KEY_SZ (2048) +#elif !defined(NO_DH) +#define WP11_MAX_SYM_KEY_SZ (4096/8) +#elif defined(HAVE_ECC) +#define WP11_MAX_SYM_KEY_SZ ((521+7)/8) +#else +#define WP11_MAX_SYM_KEY_SZ 64 +#endif + +/* Maximum size of storage for generated/derived DH key. */ +#ifdef WOLFPKCS11_NSS +#define WP11_MAX_DH_KEY_SZ (8192/8) +#else +#define WP11_MAX_DH_KEY_SZ (4096/8) +#endif + /* Login failure constants. */ #ifndef WP11_MAX_LOGIN_FAILS_SO #define WP11_MAX_LOGIN_FAILS_SO 3 @@ -388,7 +406,7 @@ typedef struct WP11_ObjectCalls WP11_ObjectCalls; WP11_LOCAL int WP11_Library_Init(void); -WP11_LOCAL void WP11_Library_Final(void); +WP11_LOCAL int WP11_Library_Final(void); WP11_LOCAL int WP11_Library_IsInitialized(void); WP11_LOCAL int WP11_SlotIdValid(CK_SLOT_ID slotId); @@ -424,6 +442,20 @@ WP11_API void WP11_Object_SetFindHook(void (*hook)(void)); WP11_API void WP11_Session_SetAddObjectHook(void (*hook)(void)); WP11_API void WP11_Slot_SetPinCheckHook(void (*hook)(void)); WP11_API void WP11_Test_FailTokenInit(int fail); +#if !defined(WOLFPKCS11_NO_STORE) && !defined(WOLFPKCS11_CUSTOM_STORE) && \ + !defined(WOLFPKCS11_TPM_STORE) && !defined(_WIN32) && !defined(_MSC_VER) +/* Exit status of a process ended by the store rename test hook. */ +#define WP11_TEST_STORE_EXIT_CODE 75 +WP11_API void WP11_Test_StoreExitAfterRenames(int renames); +WP11_API void WP11_Test_StoreCloseFailAfter(int closes); +#endif +#ifndef WOLFPKCS11_NO_STORE +WP11_API int WP11_Test_StoreWriteFailAfter(int writes); +WP11_API int WP11_Test_StoreWriteFailOnce(int writes); +#endif +#if defined(WOLFPKCS11_TPM_STORE) && !defined(WOLFPKCS11_NO_STORE) +WP11_API void WP11_Test_TpmStoreShortBy(int bytes); +#endif #if defined(WOLFPKCS11_TPM) && (!defined(NO_RSA) || defined(HAVE_ECC)) WP11_API int WP11_Test_DecodeTpmKey(CK_SLOT_ID slotId, unsigned char* keyData, int keyDataLen); @@ -436,6 +468,7 @@ WP11_LOCAL int WP11_Slot_ChangeUserPin(WP11_Slot* slot, char* oldPin, WP11_LOCAL int WP11_Slot_TokenReset(WP11_Slot* slot, char* pin, int pinLen, char* label); WP11_LOCAL void WP11_Slot_GetTokenLabel(WP11_Slot* slot, char* label); +WP11_LOCAL void WP11_BlankPad(CK_UTF8CHAR* field, int len); WP11_LOCAL int WP11_Slot_IsTokenInitialized(WP11_Slot* slot); WP11_LOCAL int WP11_Slot_TokenFailedLogin(WP11_Slot* slot, int login); WP11_LOCAL time_t WP11_Slot_TokenFailedExpire(WP11_Slot* slot, int login); @@ -445,6 +478,8 @@ WP11_LOCAL int WP11_Session_Get(CK_SESSION_HANDLE sessionHandle, WP11_Session** WP11_LOCAL int WP11_Session_GetState(WP11_Session* session); WP11_LOCAL int WP11_Session_IsRW(WP11_Session* session); WP11_LOCAL int WP11_Session_IsOpInitialized(WP11_Session* session, int init); +WP11_LOCAL int WP11_Session_IsOpCategoryInit(WP11_Session* session, + int opCategory); WP11_LOCAL int WP11_Session_IsOpCategoryActive(WP11_Session* session, int opCategory); WP11_LOCAL int WP11_Session_UpdateData(WP11_Session *session, byte *data, word32 dataLen); @@ -453,6 +488,7 @@ WP11_LOCAL void WP11_Session_FreeData(WP11_Session *session); WP11_LOCAL int WP11_Session_IsHashOpInitialized(WP11_Session* session, int mechanism); WP11_LOCAL enum wc_HashType WP11_Session_ToHashType(WP11_Session* session); WP11_LOCAL void WP11_Session_SetOpInitialized(WP11_Session* session, int init); +WP11_LOCAL void WP11_Session_AbortOp(WP11_Session* session); WP11_LOCAL WP11_Slot* WP11_Session_GetSlot(WP11_Session* session); WP11_LOCAL CK_SLOT_ID WP11_Session_GetSlotId(WP11_Session* session); WP11_LOCAL CK_MECHANISM_TYPE WP11_Session_GetMechanism(WP11_Session* session); @@ -780,6 +816,7 @@ WP11_LOCAL int WP11_AesCts_DecryptUpdate(unsigned char* enc, word32 encSz, WP11_Session* session); WP11_LOCAL int WP11_AesCts_DecryptFinal(unsigned char* dec, word32* decSz, WP11_Session* session); +WP11_LOCAL void WP11_AesCts_Final(WP11_Session* session); WP11_LOCAL int WP11_Aes_Cmac_Init(WP11_Object* secret, WP11_Session* session, word32 sigLen); diff --git a/wolfpkcs11/pkcs11.h b/wolfpkcs11/pkcs11.h index 3397af80..03bd305e 100644 --- a/wolfpkcs11/pkcs11.h +++ b/wolfpkcs11/pkcs11.h @@ -484,6 +484,7 @@ extern "C" { #define CKR_MUTEX_BAD 0x000001A0UL #define CKR_MUTEX_NOT_LOCKED 0x000001A1UL #define CKR_FUNCTION_REJECTED 0x00000200UL +#define CKR_OPERATION_CANCEL_FAILED 0x00000202UL #define CKR_VENDOR_DEFINED 0x80000000UL