tor-browser

pk11ectest.c (7723B)

---

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "blapi.h"
#include "nss.h"
#include "secutil.h"
#include "secitem.h"
#include "nspr.h"
#include "pk11pub.h"
#include <stdio.h>

typedef struct KeyLengthEntryStr {
   SECOidTag tag;
   unsigned int len;
   PRBool encoded;
} KeyLengthEntry;

const KeyLengthEntry keyLengthTable[] = {
   { SEC_OID_SECG_EC_SECP256R1, 65, PR_TRUE },
   { SEC_OID_SECG_EC_SECP384R1, 97, PR_TRUE },
   { SEC_OID_SECG_EC_SECP521R1, 133, PR_TRUE },
   { SEC_OID_CURVE25519, 32, PR_FALSE }
};

const KeyLengthEntry *
getKeyLengthEntry(SECOidTag tag)
{
   int i;

   for (i = 0; i < PR_ARRAY_SIZE(keyLengthTable); i++) {
       if (keyLengthTable[i].tag == tag) {
           return &keyLengthTable[i];
       }
   }
   return NULL;
}

void
printBuf(const SECItem *item)
{
   int i;
   if (!item || !item->len) {
       printf("(null)\n");
       return;
   }

   for (i = 0; i < item->len; i++) {
       printf("%02x", item->data[i]);
   }
   printf("\n");
}

void
PrintKey(PK11SymKey *symKey)
{
   char *name = PK11_GetSymKeyNickname(symKey);
   int len = PK11_GetKeyLength(symKey);
   int strength = PK11_GetKeyStrength(symKey, NULL);
   SECItem *value = NULL;
   CK_KEY_TYPE type = PK11_GetSymKeyType(symKey);
   (void)PK11_ExtractKeyValue(symKey);

   value = PK11_GetKeyData(symKey);
   printf("%s %3d   %4d   %s  ", name ? name : "no-name", len, strength,
          type == CKK_GENERIC_SECRET ? "generic" : "ERROR! UNKNOWN KEY TYPE");
   printBuf(value);

   PORT_Free(name);
}

SECStatus
ectest_curve_pkcs11(SECOidTag oid)
{
   SECKEYECParams pk_11_ecParams = { siBuffer, NULL, 0 };
   SECKEYPublicKey *pubKey = NULL;
   SECKEYPrivateKey *privKey = NULL;
   SECOidData *oidData = NULL;
   CK_MECHANISM_TYPE target = CKM_TLS12_MASTER_KEY_DERIVE_DH;
   PK11SymKey *symKey = NULL;
   SECStatus rv = SECFailure;
   const KeyLengthEntry *keyLengthEntry;
   SECItem point = { siBuffer, NULL, 0 };
   SECItem value = { siBuffer, NULL, 0 };
   PLArenaPool *arena = NULL;

   oidData = SECOID_FindOIDByTag(oid);
   if (oidData == NULL) {
       printf(" >>> SECOID_FindOIDByTag failed.\n");
       goto cleanup;
   }
   PORT_Assert(oidData->oid.len < 256);
   SECITEM_AllocItem(NULL, &pk_11_ecParams, (2 + oidData->oid.len));
   pk_11_ecParams.data[0] = SEC_ASN1_OBJECT_ID; /* we have to prepend 0x06 */
   pk_11_ecParams.data[1] = oidData->oid.len;
   memcpy(pk_11_ecParams.data + 2, oidData->oid.data, oidData->oid.len);

   privKey = SECKEY_CreateECPrivateKey(&pk_11_ecParams, &pubKey, NULL);
   if (!privKey || !pubKey) {
       printf(" >>> SECKEY_CreateECPrivateKey failed.\n");
       goto cleanup;
   }

   symKey = PK11_PubDeriveWithKDF(privKey, pubKey, PR_FALSE, NULL, NULL,
                                  CKM_ECDH1_DERIVE, target, CKA_DERIVE, 0,
                                  CKD_NULL, NULL, NULL);
   if (!symKey) {
       printf(" >>> PK11_PubDeriveWithKDF failed.\n");
       goto cleanup;
   }
   PrintKey(symKey);

   keyLengthEntry = getKeyLengthEntry(oid);
   /* this shouldn't happen unless new curves are added without adding them
    * to the keyLengthTable */
   PR_ASSERT(keyLengthEntry);

   /* make sure we are returning CKA_EC_POINT according to the PKCS #11 standard.
    * NSS itself can tolerate non-standard CKA_EC_POINT, so this is the only place
    * our test will detect incorrect behavior */
   rv = PK11_ReadRawAttribute(PK11_TypePubKey, pubKey, CKA_EC_POINT, &point);
   if (rv == SECFailure) {
       printf(" >>> Couldn't get CKA_EC_POINT from the ec pubKey.\n");
       goto cleanup;
   }
   rv = SECFailure;
   if (keyLengthEntry->encoded) {
       if (point.len == keyLengthEntry->len) {
           printf(" >>> Expected encoded CKA_EC_POINT and got a decoded value.\n");
           printBuf(&point);
           goto cleanup;
       }
       arena = PORT_NewArena(DER_DEFAULT_CHUNKSIZE);
       if (arena == NULL) {
           printf(" >>> arena alloc failed.\n");
           goto cleanup;
       }

       rv = SEC_QuickDERDecodeItem(arena, &value, SEC_ASN1_GET(SEC_OctetStringTemplate),
                                   &point);
       if (rv != SECSuccess) {
           printf(" >>> invalid endoded CKA_EC_POINT.\n");
           printBuf(&point);
           goto cleanup;
       }
       rv = SECFailure;
       if (value.len != keyLengthEntry->len) {
           printf(" >>> invalid decoded CKA_EC_POINT len (%d) expected %d.\n",
                  value.len, keyLengthEntry->len);
           printBuf(&value);
           goto cleanup;
       }
       if (value.data[0] != EC_POINT_FORM_UNCOMPRESSED) {
           printf(" >>> invalid CKA_EC_POINT format (%02x) expected %02x.\n",
                  value.data[0], EC_POINT_FORM_UNCOMPRESSED);
           printBuf(&value);
           goto cleanup;
       }
   } else {
       if (point.len != keyLengthEntry->len) {
           printf(" >>> invalid CKA_EC_POINT len (%d) expected %d.\n",
                  point.len, keyLengthEntry->len);
           printBuf(&point);
           goto cleanup;
       }
   }

   rv = SECSuccess;
cleanup:
   if (privKey) {
       SECKEY_DestroyPrivateKey(privKey);
   }
   if (pubKey) {
       SECKEY_DestroyPublicKey(pubKey);
   }
   if (symKey) {
       PK11_FreeSymKey(symKey);
   }
   if (arena) {
       PORT_FreeArena(arena, PR_TRUE);
   }
   SECITEM_FreeItem(&pk_11_ecParams, PR_FALSE);
   SECITEM_FreeItem(&point, PR_FALSE);

   return rv;
}

void
printUsage(char *prog)
{
   printf("Usage: %s [-fp] [-nd]\n"
          "\t-n: NIST curves\n"
          "\t-d: non-NIST curves\n"
          "You have to specify at at least one of n or d.\n"
          "By default no tests are executed.\n",
          prog);
}

/* Performs tests of elliptic curve cryptography over prime fields If
* tests fail, then it prints an error message, aborts, and returns an
* error code. Otherwise, returns 0. */
int
main(int argv, char **argc)
{
   SECStatus rv = SECSuccess;
   int i = 0;
   int nist = 0;
   int nonnist = 0;
   SECOidTag nistOids[3] = { SEC_OID_SECG_EC_SECP256R1,
                             SEC_OID_SECG_EC_SECP384R1,
                             SEC_OID_SECG_EC_SECP521R1 };

   for (i = 1; i < argv; i++) {
       if (PL_strcasecmp(argc[i], "-n") == 0) {
           nist = 1;
       } else if (PL_strcasecmp(argc[i], "-d") == 0) {
           nonnist = 1;
       } else {
           printUsage(argc[0]);
           return 1;
       }
   }
   if (!nist && !nonnist) {
       printUsage(argc[0]);
       return 1;
   }

   rv = NSS_NoDB_Init(NULL);
   if (rv != SECSuccess) {
       SECU_PrintError("Error:", "NSS_NoDB_Init");
       goto cleanup;
   }

   if (nonnist) {
       if (ectest_curve_pkcs11(SEC_OID_CURVE25519) != SECSuccess) {
           printf("not okay (OID %d) - PK11 test\n", SEC_OID_CURVE25519);
           rv = SECFailure;
       } else {
           printf("okay (OID %d) - PK11 test\n", SEC_OID_CURVE25519);
       }
   }
   if (nist) {
       for (i = 0; i < 3; ++i) {
           if (ectest_curve_pkcs11(nistOids[i]) != SECSuccess) {
               printf("not okay (OID %d) - PK11 test\n", nistOids[i]);
               rv = SECFailure;
           } else {
               printf("okay (OID %d) - PK11 test\n", nistOids[i]);
           }
       }
   }

cleanup:
   rv |= NSS_Shutdown();

   if (rv != SECSuccess) {
       printf("Error: exiting with error value\n");
   }
   return rv;
}