X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/5dd5f9ec28f304ca377c42fd7f711d6cf12b90e1..5c19dc3ae3bd8e40a9c028b0deddd50ff337692c:/OSX/libsecurity_cdsa_utilities/lib/digestobject.h

diff --git a/OSX/libsecurity_cdsa_utilities/lib/digestobject.h b/OSX/libsecurity_cdsa_utilities/lib/digestobject.h
new file mode 100644
index 00000000..0698974b
--- /dev/null
+++ b/OSX/libsecurity_cdsa_utilities/lib/digestobject.h
@@ -0,0 +1,133 @@
+/*
+ * Copyright (c) 2000-2004,2006,2011,2013-2014 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@
+ */
+
+
+/*
+ * digestobject.h - generic virtual Digest base class 
+ */
+
+#ifndef	_DIGEST_OBJECT_H_
+#define _DIGEST_OBJECT_H_
+
+#include <security_cdsa_utilities/cssmalloc.h>
+
+/* common virtual digest class */
+class DigestObject {
+public:
+	DigestObject() : mInitFlag(false), mIsDone(false) { }
+	virtual ~DigestObject() { }
+	
+	/* 
+	 * The remaining functions must be implemented by subclass. 
+	 */
+	/* init is reusable */
+	virtual void digestInit() = 0;
+
+	/* add some data */
+	virtual void digestUpdate(
+		const void *data, 
+		size_t 		len) = 0;
+	
+	/* obtain digest (once only per init, update, ... cycle) */
+	virtual void digestFinal(
+		void 		*digest) = 0;  	/* RETURNED, alloc'd by caller */
+	
+	/* sublass-specific copy */
+	virtual DigestObject *digestClone() const = 0;
+	
+	virtual size_t digestSizeInBytes() const = 0;
+
+protected:
+	bool			mInitFlag;
+	bool			mIsDone;	
+			
+	bool			initFlag() 				{ return mInitFlag; }
+	void			setInitFlag(bool flag) 	{ mInitFlag = flag; }
+	bool			isDone() 				{ return mIsDone; }
+	void			setIsDone(bool done) 	{ mIsDone = done; }
+};
+
+/*
+ * NullDigest.h - nop digest for use with raw signature algorithms.
+ *				  NullDigest(someData) = someData.
+ */
+class NullDigest : public DigestObject
+{
+public:
+	NullDigest() : mInBuf(NULL), mInBufSize(0) 
+	{ 
+	}
+
+	void digestInit() 
+	{ 
+		/* reusable - reset */
+		if(mInBufSize) {
+			assert(mInBuf != NULL);
+			memset(mInBuf, 0, mInBufSize);
+			Allocator::standard().free(mInBuf);
+			mInBufSize = 0;
+			mInBuf = NULL;
+		}
+	}
+
+	~NullDigest()
+	{
+		digestInit();
+	}
+
+	void digestUpdate(
+		const void *data, 
+		size_t 		len) 
+	{
+		mInBuf = Allocator::standard().realloc(mInBuf, mInBufSize + len);
+		memmove((uint8 *)mInBuf + mInBufSize, data, len);
+		mInBufSize += len;
+	}
+	
+	virtual void digestFinal(
+		void 		*digest)
+	{
+		memmove(digest, mInBuf, mInBufSize);
+	}
+										
+	virtual DigestObject *digestClone() const
+	{
+		NullDigest *cloned = new NullDigest;
+		cloned->digestUpdate(mInBuf, mInBufSize);
+		return cloned;
+	}
+	
+	/* unique to NullDigest - just obtain current data ptr, no copy */
+	virtual const void *digestPtr() { return mInBuf; }
+	
+	size_t digestSizeInBytes() const
+	{ 
+		return mInBufSize;
+	}
+
+private:
+	void		*mInBuf;
+	size_t		mInBufSize;
+};
+
+#endif	/* _DIGEST_OBJECT_H_ */