The Tor Browser / file revision

 /* 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/. */

 /*

  * CMS contentInfo methods.

  */

 #include "cmslocal.h"

 #include "pk11func.h"

 #include "secitem.h"

 #include "secoid.h"

 #include "secerr.h"

 /*

  * NSS_CMSContentInfo_Create - create a content info

  *

  * version is set in the _Finalize procedures for each content type

  */

 SECStatus

 NSS_CMSContentInfo_Private_Init(NSSCMSContentInfo *cinfo)

 {

     if (cinfo->privateInfo) {

 	return SECSuccess;

     }

     cinfo->privateInfo = PORT_ZNew(NSSCMSContentInfoPrivate);

     return (cinfo->privateInfo) ? SECSuccess : SECFailure;

 }

 static void

 nss_cmsContentInfo_private_destroy(NSSCMSContentInfoPrivate *privateInfo)

 {

     if (privateInfo->digcx) {

 	/* must destroy digest objects */

 	NSS_CMSDigestContext_Cancel(privateInfo->digcx);

 	privateInfo->digcx = NULL;

     }

     if (privateInfo->ciphcx) {

 	NSS_CMSCipherContext_Destroy(privateInfo->ciphcx);

 	privateInfo->ciphcx = NULL;

     }

     PORT_Free(privateInfo);

 }

 /*

  * NSS_CMSContentInfo_Destroy - destroy a CMS contentInfo and all of its sub-pieces.

  */

 void

 NSS_CMSContentInfo_Destroy(NSSCMSContentInfo *cinfo)

 {

     SECOidTag kind;

     kind = NSS_CMSContentInfo_GetContentTypeTag(cinfo);

     switch (kind) {

     case SEC_OID_PKCS7_ENVELOPED_DATA:

 	NSS_CMSEnvelopedData_Destroy(cinfo->content.envelopedData);

 	break;

       case SEC_OID_PKCS7_SIGNED_DATA:

 	NSS_CMSSignedData_Destroy(cinfo->content.signedData);

 	break;

       case SEC_OID_PKCS7_ENCRYPTED_DATA:

 	NSS_CMSEncryptedData_Destroy(cinfo->content.encryptedData);

 	break;

       case SEC_OID_PKCS7_DIGESTED_DATA:

 	NSS_CMSDigestedData_Destroy(cinfo->content.digestedData);

 	break;

       default:

 	NSS_CMSGenericWrapperData_Destroy(kind, cinfo->content.genericData);

 	/* XXX Anything else that needs to be "manually" freed/destroyed? */

 	break;

     }

     if (cinfo->privateInfo) {

 	nss_cmsContentInfo_private_destroy(cinfo->privateInfo);

 	cinfo->privateInfo = NULL;

     }

     if (cinfo->bulkkey) {

 	PK11_FreeSymKey(cinfo->bulkkey);

     }

 }

 /*

  * NSS_CMSContentInfo_GetChildContentInfo - get content's contentInfo (if it exists)

  */

 NSSCMSContentInfo *

 NSS_CMSContentInfo_GetChildContentInfo(NSSCMSContentInfo *cinfo)

 {

     NSSCMSContentInfo * ccinfo  = NULL;

     SECOidTag tag = NSS_CMSContentInfo_GetContentTypeTag(cinfo);

     switch (tag) {

     case SEC_OID_PKCS7_SIGNED_DATA:

 	if (cinfo->content.signedData != NULL) {

 	    ccinfo = &(cinfo->content.signedData->contentInfo);

 	}

 	break;

     case SEC_OID_PKCS7_ENVELOPED_DATA:

 	if (cinfo->content.envelopedData != NULL) {

 	    ccinfo = &(cinfo->content.envelopedData->contentInfo);

 	}

 	break;

     case SEC_OID_PKCS7_DIGESTED_DATA:

 	if (cinfo->content.digestedData != NULL) {

 	    ccinfo = &(cinfo->content.digestedData->contentInfo);

 	}

 	break;

     case SEC_OID_PKCS7_ENCRYPTED_DATA:

 	if (cinfo->content.encryptedData != NULL) {

 	    ccinfo = &(cinfo->content.encryptedData->contentInfo);

 	}

 	break;

     case SEC_OID_PKCS7_DATA:

     default:

 	if (NSS_CMSType_IsWrapper(tag)) {

 	    if (cinfo->content.genericData != NULL) {

 	       ccinfo = &(cinfo->content.genericData->contentInfo);

 	    }

 	}

 	break;

     }

     if (ccinfo && !ccinfo->privateInfo) {

 	NSS_CMSContentInfo_Private_Init(ccinfo);

     }

     return ccinfo;

 }

 SECStatus

 NSS_CMSContentInfo_SetDontStream(NSSCMSContentInfo *cinfo, PRBool dontStream)

 {

    SECStatus rv;

    rv = NSS_CMSContentInfo_Private_Init(cinfo);

    if (rv != SECSuccess) {

 	/* default is streaming, failure to get ccinfo will not effect this */

 	return dontStream ? SECFailure :  SECSuccess ;

    }

    cinfo->privateInfo->dontStream = dontStream;

    return SECSuccess;

 }

 /*

  * NSS_CMSContentInfo_SetContent - set content type & content

  */

 SECStatus

 NSS_CMSContentInfo_SetContent(NSSCMSMessage *cmsg, NSSCMSContentInfo *cinfo, SECOidTag type, void *ptr)

 {

     SECStatus rv;

     cinfo->contentTypeTag = SECOID_FindOIDByTag(type);

     if (cinfo->contentTypeTag == NULL)

 	return SECFailure;

     /* do not copy the oid, just create a reference */

     rv = SECITEM_CopyItem (cmsg->poolp, &(cinfo->contentType), &(cinfo->contentTypeTag->oid));

     if (rv != SECSuccess)

 	return SECFailure;

     cinfo->content.pointer = ptr;

     if (NSS_CMSType_IsData(type) && ptr) {

 	cinfo->rawContent = ptr;

     } else {

 	/* as we always have some inner data,

 	 * we need to set it to something, just to fool the encoder enough to work on it

 	 * and get us into nss_cms_encoder_notify at that point */

 	cinfo->rawContent = SECITEM_AllocItem(cmsg->poolp, NULL, 1);

 	if (cinfo->rawContent == NULL) {

 	    PORT_SetError(SEC_ERROR_NO_MEMORY);

 	    return SECFailure;

 	}

     }

     return SECSuccess;

 }

 /*

  * NSS_CMSContentInfo_SetContent_XXXX - typesafe wrappers for NSS_CMSContentInfo_SetContent

  */

 /*

  * data == NULL -> pass in data via NSS_CMSEncoder_Update

  * data != NULL -> take this data

  */

 SECStatus

 NSS_CMSContentInfo_SetContent_Data(NSSCMSMessage *cmsg, NSSCMSContentInfo *cinfo, SECItem *data, PRBool detached)

 {

     if (NSS_CMSContentInfo_SetContent(cmsg, cinfo, SEC_OID_PKCS7_DATA, (void *)data) != SECSuccess)

 	return SECFailure;

     if (detached) {

         cinfo->rawContent = NULL;

     }

     return SECSuccess;

 }

 SECStatus

 NSS_CMSContentInfo_SetContent_SignedData(NSSCMSMessage *cmsg, NSSCMSContentInfo *cinfo, NSSCMSSignedData *sigd)

 {

     return NSS_CMSContentInfo_SetContent(cmsg, cinfo, SEC_OID_PKCS7_SIGNED_DATA, (void *)sigd);

 }

 SECStatus

 NSS_CMSContentInfo_SetContent_EnvelopedData(NSSCMSMessage *cmsg, NSSCMSContentInfo *cinfo, NSSCMSEnvelopedData *envd)

 {

     return NSS_CMSContentInfo_SetContent(cmsg, cinfo, SEC_OID_PKCS7_ENVELOPED_DATA, (void *)envd);

 }

 SECStatus

 NSS_CMSContentInfo_SetContent_DigestedData(NSSCMSMessage *cmsg, NSSCMSContentInfo *cinfo, NSSCMSDigestedData *digd)

 {

     return NSS_CMSContentInfo_SetContent(cmsg, cinfo, SEC_OID_PKCS7_DIGESTED_DATA, (void *)digd);

 }

 SECStatus

 NSS_CMSContentInfo_SetContent_EncryptedData(NSSCMSMessage *cmsg, NSSCMSContentInfo *cinfo, NSSCMSEncryptedData *encd)

 {

     return NSS_CMSContentInfo_SetContent(cmsg, cinfo, SEC_OID_PKCS7_ENCRYPTED_DATA, (void *)encd);

 }

 /*

  * NSS_CMSContentInfo_GetContent - get pointer to inner content

  *

  * needs to be casted...

  */

 void *

 NSS_CMSContentInfo_GetContent(NSSCMSContentInfo *cinfo)

 {

     SECOidTag tag = (cinfo && cinfo->contentTypeTag) 

 	                ? cinfo->contentTypeTag->offset 

 	                : SEC_OID_UNKNOWN;

     switch (tag) {

     case SEC_OID_PKCS7_DATA:

     case SEC_OID_PKCS7_SIGNED_DATA:

     case SEC_OID_PKCS7_ENVELOPED_DATA:

     case SEC_OID_PKCS7_DIGESTED_DATA:

     case SEC_OID_PKCS7_ENCRYPTED_DATA:

 	return cinfo->content.pointer;

     default:

 	return NSS_CMSType_IsWrapper(tag) ? cinfo->content.pointer : (NSS_CMSType_IsData(tag) ? cinfo->rawContent : NULL);

     }

 }

 /* 

  * NSS_CMSContentInfo_GetInnerContent - get pointer to innermost content

  *

  * this is typically only called by NSS_CMSMessage_GetContent()

  */

 SECItem *

 NSS_CMSContentInfo_GetInnerContent(NSSCMSContentInfo *cinfo)

 {

     NSSCMSContentInfo *ccinfo;

     SECOidTag          tag;

     SECItem           *pItem = NULL;

     tag = NSS_CMSContentInfo_GetContentTypeTag(cinfo);

     if (NSS_CMSType_IsData(tag)) {

 	pItem = cinfo->content.data; 

     } else if (NSS_CMSType_IsWrapper(tag)) {

 	ccinfo = NSS_CMSContentInfo_GetChildContentInfo(cinfo);

 	if (ccinfo != NULL) {

 	    pItem = NSS_CMSContentInfo_GetContent(ccinfo);

 	}

     } else {

 	PORT_Assert(0);

     }

     return pItem;

 }

 /*

  * NSS_CMSContentInfo_GetContentType{Tag,OID} - find out (saving pointer to lookup result

  * for future reference) and return the inner content type.

  */

 SECOidTag

 NSS_CMSContentInfo_GetContentTypeTag(NSSCMSContentInfo *cinfo)

 {

     if (cinfo->contentTypeTag == NULL)

 	cinfo->contentTypeTag = SECOID_FindOID(&(cinfo->contentType));

     if (cinfo->contentTypeTag == NULL)

 	return SEC_OID_UNKNOWN;

     return cinfo->contentTypeTag->offset;

 }

 SECItem *

 NSS_CMSContentInfo_GetContentTypeOID(NSSCMSContentInfo *cinfo)

 {

     if (cinfo->contentTypeTag == NULL)

 	cinfo->contentTypeTag = SECOID_FindOID(&(cinfo->contentType));

     if (cinfo->contentTypeTag == NULL)

 	return NULL;

     return &(cinfo->contentTypeTag->oid);

 }

 /*

  * NSS_CMSContentInfo_GetContentEncAlgTag - find out (saving pointer to lookup result

  * for future reference) and return the content encryption algorithm tag.

  */

 SECOidTag

 NSS_CMSContentInfo_GetContentEncAlgTag(NSSCMSContentInfo *cinfo)

 {

     if (cinfo->contentEncAlgTag == SEC_OID_UNKNOWN)

 	cinfo->contentEncAlgTag = SECOID_GetAlgorithmTag(&(cinfo->contentEncAlg));

     return cinfo->contentEncAlgTag;

 }

 /*

  * NSS_CMSContentInfo_GetContentEncAlg - find out and return the content encryption algorithm tag.

  */

 SECAlgorithmID *

 NSS_CMSContentInfo_GetContentEncAlg(NSSCMSContentInfo *cinfo)

 {

     return &(cinfo->contentEncAlg);

 }

 SECStatus

 NSS_CMSContentInfo_SetContentEncAlg(PLArenaPool *poolp, NSSCMSContentInfo *cinfo,

 				    SECOidTag bulkalgtag, SECItem *parameters, int keysize)

 {

     SECStatus rv;

     rv = SECOID_SetAlgorithmID(poolp, &(cinfo->contentEncAlg), bulkalgtag, parameters);

     if (rv != SECSuccess)

 	return SECFailure;

     cinfo->keysize = keysize;

     return SECSuccess;

 }

 SECStatus

 NSS_CMSContentInfo_SetContentEncAlgID(PLArenaPool *poolp, NSSCMSContentInfo *cinfo,

 				    SECAlgorithmID *algid, int keysize)

 {

     SECStatus rv;

     rv = SECOID_CopyAlgorithmID(poolp, &(cinfo->contentEncAlg), algid);

     if (rv != SECSuccess)

 	return SECFailure;

     if (keysize >= 0)

 	cinfo->keysize = keysize;

     return SECSuccess;

 }

 void

 NSS_CMSContentInfo_SetBulkKey(NSSCMSContentInfo *cinfo, PK11SymKey *bulkkey)

 {

     cinfo->bulkkey = PK11_ReferenceSymKey(bulkkey);

     cinfo->keysize = PK11_GetKeyStrength(cinfo->bulkkey, &(cinfo->contentEncAlg));

 }

 PK11SymKey *

 NSS_CMSContentInfo_GetBulkKey(NSSCMSContentInfo *cinfo)

 {

     if (cinfo->bulkkey == NULL)

 	return NULL;

     return PK11_ReferenceSymKey(cinfo->bulkkey);

 }

 int

 NSS_CMSContentInfo_GetBulkKeySize(NSSCMSContentInfo *cinfo)

 {

     return cinfo->keysize;

 }