X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/80e2389990082500d76eb566d4946be3e786c3ef..d8f41ccd20de16f8ebe2ccc84d47bf1cb2b26bbb:/Security/libsecurity_asn1/lib/SecNssCoder.cpp

diff --git a/Security/libsecurity_asn1/lib/SecNssCoder.cpp b/Security/libsecurity_asn1/lib/SecNssCoder.cpp
new file mode 100644
index 00000000..86bb8d2e
--- /dev/null
+++ b/Security/libsecurity_asn1/lib/SecNssCoder.cpp
@@ -0,0 +1,194 @@
+/*
+ * Copyright (c) 2003-2006,2008,2010-2012 Apple Inc. All Rights Reserved.
+ * 
+ * @APPLE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_LICENSE_HEADER_END@
+ *
+ * SecNssCoder.cpp: simple C++ wrapper for PLArenaPool and the
+ * high-level ANS1 encode/decode routines.
+ */
+
+#include <TargetConditionals.h>
+#if (TARGET_OS_MAC && !(TARGET_OS_EMBEDDED || TARGET_OS_IPHONE))
+
+#include "SecNssCoder.h"
+#include <Security/cssmerr.h>
+#include <security_utilities/utilities.h>
+#include <security_asn1/secasn1.h>
+#include <string.h>
+#include <assert.h>
+
+#ifdef	NDEBUG
+#define THROW_ENABLE	1
+#else
+/* disable link against Security framework when true */
+#define THROW_ENABLE	0
+#endif
+
+#if		THROW_ENABLE
+#define THROW_ERROR	Security::CssmError::throwMe(CSSMERR_CSSM_MEMORY_ERROR)
+#else
+#define THROW_ERROR
+#endif
+
+SecNssCoder::SecNssCoder(PRUint32 chunkSize /* = SNC_CHUNKSIZE_DEF */)
+	: mPool(NULL)
+{
+	mPool = PORT_NewArena(chunkSize);
+	if(mPool == NULL) {
+		THROW_ERROR;
+	}
+}
+
+SecNssCoder::~SecNssCoder()
+{
+	if(mPool != NULL) {
+		/*
+		 * Note: we're asking for a memory zero here, but 
+		 * PORT_FreeArena doesn't do that (yet).
+		 */
+		PORT_FreeArena(mPool, PR_TRUE);
+		mPool = NULL;
+	}
+}
+
+PRErrorCode	SecNssCoder::decode(
+	const void				*src,		// BER-encoded source
+	size_t				len,
+	const SecAsn1Template 	*templ,	
+	void					*dest)
+{
+	SECStatus prtn;
+	
+	assert(mPool != NULL);
+	prtn = SEC_ASN1Decode(mPool, dest, templ, (const char *)src, len);
+	if(prtn) {
+		return PR_GetError();
+	}
+	else {
+		return 0;
+	}
+}
+
+PRErrorCode SecNssCoder::encodeItem(
+	const void				*src,
+	const SecAsn1Template 	*templ,	
+	SECItem					&dest)
+{
+	assert(mPool != NULL);
+
+	dest.Data = NULL;
+	dest.Length = 0;
+	
+	SECItem *rtnItem = SEC_ASN1EncodeItem(mPool, &dest, src, templ);
+	if(rtnItem == NULL) {
+		return PR_GetError();
+	}
+	else {
+		return 0;
+	}
+}
+
+void *SecNssCoder::malloc(size_t len)
+{
+	assert(mPool != NULL);
+	void *rtn = PORT_ArenaAlloc(mPool, len);
+	if(rtn == NULL) {
+		THROW_ERROR;
+	}
+	return rtn;
+}
+
+/* malloc item.Data, set item.Length */
+void SecNssCoder::allocItem(
+	SECItem					&item,
+	size_t					len)
+{
+	item.Data = (uint8 *)malloc(len);
+	item.Length = len;
+}
+
+/* malloc and copy */
+void SecNssCoder::allocCopyItem(
+	const void				*src,
+	size_t					len,
+	SECItem					&dest)
+{
+	allocItem(dest, len);
+	memmove(dest.Data, src, len);
+}
+
+/*
+ * This is pretty much a copy of SEC_ASN1EncodeItem, with a Allocator
+ * malloc replacing the sec_asn1e_allocate_item to alloc the output data.
+ */
+PRErrorCode SecNssEncodeItem(
+	const void				*src,
+	const SecAsn1Template 	*templ,	
+	Security::Allocator	&alloc,
+	SECItem					&dest)
+{
+    unsigned long encoding_length = 0;
+    SECStatus rv;
+
+	dest.Data = NULL;
+	dest.Length = 0;
+
+    rv = SEC_ASN1Encode (src, templ,
+			 sec_asn1e_encode_item_count, &encoding_length);
+    if (rv != SECSuccess) {
+		return PR_GetError();
+	}
+
+	/* replace this...
+    dest = sec_asn1e_allocate_item (poolp, dest, encoding_length);
+    if (dest == NULL)
+	return NULL;
+	... with this: */
+	dest.Data = (uint8 *)alloc.malloc(encoding_length);
+	dest.Length = 0;
+	/* end replacement */
+	
+    rv = SEC_ASN1Encode (src, templ, sec_asn1e_encode_item_store, &dest);
+    if (rv != SECSuccess) {
+		return PR_GetError();
+	}
+
+    assert(encoding_length == dest.Length);
+    return 0;
+}
+
+PRErrorCode SecNssEncodeItemOdata(
+	const void				*src,
+	const SecAsn1Template 	*templ,	
+	CssmOwnedData			&odata)
+{
+	Allocator &alloc = odata.allocator;
+	SECItem sitem;
+	PRErrorCode prtn;
+	
+	prtn = SecNssEncodeItem(src, templ, alloc, sitem);
+	if(prtn) {
+		return prtn;
+	}
+	odata.set(sitem.Data, sitem.Length);
+	return 0;
+}
+
+#endif /* TARGET_OS_MAC */