michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: #include "sechash.h"
michael@0: #include "secoidt.h"
michael@0: #include "secerr.h"
michael@0: #include "blapi.h"
michael@0: #include "pk11func.h"	/* for the PK11_ calls below. */
michael@0: 
michael@0: static void *
michael@0: null_hash_new_context(void)
michael@0: {
michael@0:     return NULL;
michael@0: }
michael@0: 
michael@0: static void *
michael@0: null_hash_clone_context(void *v)
michael@0: {
michael@0:     PORT_Assert(v == NULL);
michael@0:     return NULL;
michael@0: }
michael@0: 
michael@0: static void
michael@0: null_hash_begin(void *v)
michael@0: {
michael@0: }
michael@0: 
michael@0: static void
michael@0: null_hash_update(void *v, const unsigned char *input, unsigned int length)
michael@0: {
michael@0: }
michael@0: 
michael@0: static void
michael@0: null_hash_end(void *v, unsigned char *output, unsigned int *outLen,
michael@0: 	      unsigned int maxOut)
michael@0: {
michael@0:     *outLen = 0;
michael@0: }
michael@0: 
michael@0: static void
michael@0: null_hash_destroy_context(void *v, PRBool b)
michael@0: {
michael@0:     PORT_Assert(v == NULL);
michael@0: }
michael@0: 
michael@0: 
michael@0: static void *
michael@0: md2_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_MD2);
michael@0: }
michael@0: 
michael@0: static void *
michael@0: md5_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_MD5);
michael@0: }
michael@0: 
michael@0: static void *
michael@0: sha1_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_SHA1);
michael@0: }
michael@0: 
michael@0: static void *
michael@0: sha224_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_SHA224);
michael@0: }
michael@0: 
michael@0: static void *
michael@0: sha256_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_SHA256);
michael@0: }
michael@0: 
michael@0: static void *
michael@0: sha384_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_SHA384);
michael@0: }
michael@0: 
michael@0: static void *
michael@0: sha512_NewContext(void) {
michael@0: 	return (void *) PK11_CreateDigestContext(SEC_OID_SHA512);
michael@0: }
michael@0: 
michael@0: const SECHashObject SECHashObjects[] = {
michael@0:   { 0,
michael@0:     (void * (*)(void)) null_hash_new_context,
michael@0:     (void * (*)(void *)) null_hash_clone_context,
michael@0:     (void (*)(void *, PRBool)) null_hash_destroy_context,
michael@0:     (void (*)(void *)) null_hash_begin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) null_hash_update,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *,
michael@0: 	      unsigned int)) null_hash_end,
michael@0:     0,
michael@0:     HASH_AlgNULL
michael@0:   },
michael@0:   { MD2_LENGTH,
michael@0:     (void * (*)(void)) md2_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int)) 
michael@0: 							PK11_DigestFinal,
michael@0:     MD2_BLOCK_LENGTH,
michael@0:     HASH_AlgMD2
michael@0:   },
michael@0:   { MD5_LENGTH,
michael@0:     (void * (*)(void)) md5_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int)) 
michael@0: 							PK11_DigestFinal,
michael@0:     MD5_BLOCK_LENGTH,
michael@0:     HASH_AlgMD5
michael@0:   },
michael@0:   { SHA1_LENGTH,
michael@0:     (void * (*)(void)) sha1_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int)) 
michael@0: 							PK11_DigestFinal,
michael@0:     SHA1_BLOCK_LENGTH,
michael@0:     HASH_AlgSHA1
michael@0:   },
michael@0:   { SHA256_LENGTH,
michael@0:     (void * (*)(void)) sha256_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int)) 
michael@0: 							PK11_DigestFinal,
michael@0:     SHA256_BLOCK_LENGTH,
michael@0:     HASH_AlgSHA256
michael@0:   },
michael@0:   { SHA384_LENGTH,
michael@0:     (void * (*)(void)) sha384_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int)) 
michael@0: 							PK11_DigestFinal,
michael@0:     SHA384_BLOCK_LENGTH,
michael@0:     HASH_AlgSHA384
michael@0:   },
michael@0:   { SHA512_LENGTH,
michael@0:     (void * (*)(void)) sha512_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int)) 
michael@0: 							PK11_DigestFinal,
michael@0:     SHA512_BLOCK_LENGTH,
michael@0:     HASH_AlgSHA512
michael@0:   },
michael@0:   { SHA224_LENGTH,
michael@0:     (void * (*)(void)) sha224_NewContext,
michael@0:     (void * (*)(void *)) PK11_CloneContext,
michael@0:     (void (*)(void *, PRBool)) PK11_DestroyContext,
michael@0:     (void (*)(void *)) PK11_DigestBegin,
michael@0:     (void (*)(void *, const unsigned char *, unsigned int)) PK11_DigestOp,
michael@0:     (void (*)(void *, unsigned char *, unsigned int *, unsigned int))
michael@0: 							PK11_DigestFinal,
michael@0:     SHA224_BLOCK_LENGTH,
michael@0:     HASH_AlgSHA224
michael@0:   },
michael@0: };
michael@0: 
michael@0: const SECHashObject * 
michael@0: HASH_GetHashObject(HASH_HashType type)
michael@0: {
michael@0:     return &SECHashObjects[type];
michael@0: }
michael@0: 
michael@0: HASH_HashType
michael@0: HASH_GetHashTypeByOidTag(SECOidTag hashOid)
michael@0: {
michael@0:     HASH_HashType ht	= HASH_AlgNULL;
michael@0: 
michael@0:     switch(hashOid) {
michael@0:     case SEC_OID_MD2:	 ht = HASH_AlgMD2;    break;
michael@0:     case SEC_OID_MD5:	 ht = HASH_AlgMD5;    break;
michael@0:     case SEC_OID_SHA1:	 ht = HASH_AlgSHA1;   break;
michael@0:     case SEC_OID_SHA224: ht = HASH_AlgSHA224; break;
michael@0:     case SEC_OID_SHA256: ht = HASH_AlgSHA256; break;
michael@0:     case SEC_OID_SHA384: ht = HASH_AlgSHA384; break;
michael@0:     case SEC_OID_SHA512: ht = HASH_AlgSHA512; break;
michael@0:     default:             ht = HASH_AlgNULL;   
michael@0: 	PORT_SetError(SEC_ERROR_INVALID_ALGORITHM);
michael@0: 	break;
michael@0:     }
michael@0:     return ht;
michael@0: }
michael@0: 
michael@0: SECOidTag
michael@0: HASH_GetHashOidTagByHMACOidTag(SECOidTag hmacOid)
michael@0: {
michael@0:     SECOidTag hashOid = SEC_OID_UNKNOWN;
michael@0: 
michael@0:     switch(hmacOid) {
michael@0:     /* no oid exists for HMAC_MD2 */
michael@0:     /* NSS does not define a oid for HMAC_MD4 */
michael@0:     case SEC_OID_HMAC_SHA1:   hashOid = SEC_OID_SHA1;   break;
michael@0:     case SEC_OID_HMAC_SHA224: hashOid = SEC_OID_SHA224; break;
michael@0:     case SEC_OID_HMAC_SHA256: hashOid = SEC_OID_SHA256; break;
michael@0:     case SEC_OID_HMAC_SHA384: hashOid = SEC_OID_SHA384; break;
michael@0:     case SEC_OID_HMAC_SHA512: hashOid = SEC_OID_SHA512; break;
michael@0:     default:                  hashOid = SEC_OID_UNKNOWN;   
michael@0: 	PORT_SetError(SEC_ERROR_INVALID_ALGORITHM);
michael@0: 	break;
michael@0:     }
michael@0:     return hashOid;
michael@0: }
michael@0: 
michael@0: SECOidTag
michael@0: HASH_GetHMACOidTagByHashOidTag(SECOidTag hashOid)
michael@0: {
michael@0:     SECOidTag hmacOid = SEC_OID_UNKNOWN;
michael@0: 
michael@0:     switch(hashOid) {
michael@0:     /* no oid exists for HMAC_MD2 */
michael@0:     /* NSS does not define a oid for HMAC_MD4 */
michael@0:     case SEC_OID_SHA1:   hmacOid = SEC_OID_HMAC_SHA1;   break;
michael@0:     case SEC_OID_SHA224: hmacOid = SEC_OID_HMAC_SHA224; break;
michael@0:     case SEC_OID_SHA256: hmacOid = SEC_OID_HMAC_SHA256; break;
michael@0:     case SEC_OID_SHA384: hmacOid = SEC_OID_HMAC_SHA384; break;
michael@0:     case SEC_OID_SHA512: hmacOid = SEC_OID_HMAC_SHA512; break;
michael@0:     default:             hmacOid = SEC_OID_UNKNOWN;   
michael@0: 	PORT_SetError(SEC_ERROR_INVALID_ALGORITHM);
michael@0: 	break;
michael@0:     }
michael@0:     return hmacOid;
michael@0: }
michael@0: 
michael@0: const SECHashObject * 
michael@0: HASH_GetHashObjectByOidTag(SECOidTag hashOid)
michael@0: {
michael@0:     HASH_HashType ht	= HASH_GetHashTypeByOidTag(hashOid);
michael@0: 
michael@0:     return (ht == HASH_AlgNULL) ? NULL : &SECHashObjects[ht];
michael@0: }
michael@0: 
michael@0: /* returns zero for unknown hash OID */
michael@0: unsigned int
michael@0: HASH_ResultLenByOidTag(SECOidTag hashOid)
michael@0: {
michael@0:     const SECHashObject * hashObject = HASH_GetHashObjectByOidTag(hashOid);
michael@0:     unsigned int          resultLen = 0;
michael@0: 
michael@0:     if (hashObject)
michael@0:     	resultLen = hashObject->length;
michael@0:     return resultLen;
michael@0: }
michael@0: 
michael@0: /* returns zero if hash type invalid. */
michael@0: unsigned int
michael@0: HASH_ResultLen(HASH_HashType type)
michael@0: {
michael@0:     if ( ( type < HASH_AlgNULL ) || ( type >= HASH_AlgTOTAL ) ) {
michael@0: 	PORT_SetError(SEC_ERROR_INVALID_ALGORITHM);
michael@0: 	return(0);
michael@0:     }
michael@0:     
michael@0:     return(SECHashObjects[type].length);
michael@0: }
michael@0: 
michael@0: unsigned int
michael@0: HASH_ResultLenContext(HASHContext *context)
michael@0: {
michael@0:     return(context->hashobj->length);
michael@0: }
michael@0: 
michael@0: 
michael@0: 
michael@0: SECStatus
michael@0: HASH_HashBuf(HASH_HashType type,
michael@0: 	     unsigned char *dest,
michael@0: 	     const unsigned char *src,
michael@0: 	     PRUint32 src_len)
michael@0: {
michael@0:     HASHContext *cx;
michael@0:     unsigned int part;
michael@0:     
michael@0:     if ( ( type < HASH_AlgNULL ) || ( type >= HASH_AlgTOTAL ) ) {
michael@0: 	return(SECFailure);
michael@0:     }
michael@0:     
michael@0:     cx = HASH_Create(type);
michael@0:     if ( cx == NULL ) {
michael@0: 	return(SECFailure);
michael@0:     }
michael@0:     HASH_Begin(cx);
michael@0:     HASH_Update(cx, src, src_len);
michael@0:     HASH_End(cx, dest, &part, HASH_ResultLenContext(cx));
michael@0:     HASH_Destroy(cx);
michael@0: 
michael@0:     return(SECSuccess);
michael@0: }
michael@0: 
michael@0: HASHContext *
michael@0: HASH_Create(HASH_HashType type)
michael@0: {
michael@0:     void *hash_context = NULL;
michael@0:     HASHContext *ret = NULL;
michael@0:     
michael@0:     if ( ( type < HASH_AlgNULL ) || ( type >= HASH_AlgTOTAL ) ) {
michael@0: 	return(NULL);
michael@0:     }
michael@0:     
michael@0:     hash_context = (* SECHashObjects[type].create)();
michael@0:     if ( hash_context == NULL ) {
michael@0: 	goto loser;
michael@0:     }
michael@0: 
michael@0:     ret = (HASHContext *)PORT_Alloc(sizeof(HASHContext));
michael@0:     if ( ret == NULL ) {
michael@0: 	goto loser;
michael@0:     }
michael@0: 
michael@0:     ret->hash_context = hash_context;
michael@0:     ret->hashobj = &SECHashObjects[type];
michael@0:     
michael@0:     return(ret);
michael@0:     
michael@0: loser:
michael@0:     if ( hash_context != NULL ) {
michael@0: 	(* SECHashObjects[type].destroy)(hash_context, PR_TRUE);
michael@0:     }
michael@0:     
michael@0:     return(NULL);
michael@0: }
michael@0: 
michael@0: 
michael@0: HASHContext *
michael@0: HASH_Clone(HASHContext *context)
michael@0: {
michael@0:     void *hash_context = NULL;
michael@0:     HASHContext *ret = NULL;
michael@0:     
michael@0:     hash_context = (* context->hashobj->clone)(context->hash_context);
michael@0:     if ( hash_context == NULL ) {
michael@0: 	goto loser;
michael@0:     }
michael@0: 
michael@0:     ret = (HASHContext *)PORT_Alloc(sizeof(HASHContext));
michael@0:     if ( ret == NULL ) {
michael@0: 	goto loser;
michael@0:     }
michael@0: 
michael@0:     ret->hash_context = hash_context;
michael@0:     ret->hashobj = context->hashobj;
michael@0:     
michael@0:     return(ret);
michael@0:     
michael@0: loser:
michael@0:     if ( hash_context != NULL ) {
michael@0: 	(* context->hashobj->destroy)(hash_context, PR_TRUE);
michael@0:     }
michael@0:     
michael@0:     return(NULL);
michael@0: 
michael@0: }
michael@0: 
michael@0: void
michael@0: HASH_Destroy(HASHContext *context)
michael@0: {
michael@0:     (* context->hashobj->destroy)(context->hash_context, PR_TRUE);
michael@0:     PORT_Free(context);
michael@0:     return;
michael@0: }
michael@0: 
michael@0: 
michael@0: void
michael@0: HASH_Begin(HASHContext *context)
michael@0: {
michael@0:     (* context->hashobj->begin)(context->hash_context);
michael@0:     return;
michael@0: }
michael@0: 
michael@0: 
michael@0: void
michael@0: HASH_Update(HASHContext *context,
michael@0: 	    const unsigned char *src,
michael@0: 	    unsigned int len)
michael@0: {
michael@0:     (* context->hashobj->update)(context->hash_context, src, len);
michael@0:     return;
michael@0: }
michael@0: 
michael@0: void
michael@0: HASH_End(HASHContext *context,
michael@0: 	 unsigned char *result,
michael@0: 	 unsigned int *result_len,
michael@0: 	 unsigned int max_result_len)
michael@0: {
michael@0:     (* context->hashobj->end)(context->hash_context, result, result_len,
michael@0: 			      max_result_len);
michael@0:     return;
michael@0: }
michael@0: 
michael@0: HASH_HashType
michael@0: HASH_GetType(HASHContext *context)
michael@0: {
michael@0:     return(context->hashobj->type);
michael@0: }