X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/80e2389990082500d76eb566d4946be3e786c3ef..d8f41ccd20de16f8ebe2ccc84d47bf1cb2b26bbb:/Security/libsecurity_codesigning/lib/cdbuilder.h

diff --git a/Security/libsecurity_codesigning/lib/cdbuilder.h b/Security/libsecurity_codesigning/lib/cdbuilder.h
new file mode 100644
index 00000000..453a804a
--- /dev/null
+++ b/Security/libsecurity_codesigning/lib/cdbuilder.h
@@ -0,0 +1,98 @@
+/*
+ * Copyright (c) 2006-2012,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@
+ */
+
+//
+// cdbuilder - constructor for CodeDirectories
+//
+#ifndef _H_CDBUILDER
+#define _H_CDBUILDER
+
+#include "codedirectory.h"
+
+
+namespace Security {
+namespace CodeSigning {
+
+
+//
+// Builder can construct CodeDirectories from pieces:
+//	Builder builder(...);
+//	builder.variousSetters(withSuitableData);
+//  CodeDirectory *result = builder.build();
+// Builder is not reusable.
+//
+class CodeDirectory::Builder {
+public:
+	Builder(HashAlgorithm digestAlgorithm);
+	~Builder();
+	
+	void executable(string path, size_t pagesize, size_t offset, size_t length);
+	void reopen(string path, size_t offset, size_t length);
+
+	void specialSlot(SpecialSlot slot, CFDataRef data);
+	void identifier(const std::string &code) { mIdentifier = code; }
+	void teamID(const std::string &team) { mTeamID = team; }
+	void flags(uint32_t f) { mFlags = f; }
+	
+	Scatter *scatter(unsigned count);			// allocate that many scatter elements (w/o sentinel)
+	Scatter *scatter() { return mScatter; }		// return already allocated scatter vector
+	
+	size_t size(const uint32_t version);		// calculate size
+	CodeDirectory *build();						// build CodeDirectory and return it
+	const size_t fixedSize(const uint32_t version);	// calculate fixed size of the CodeDirectory
+
+	DynamicHash *getHash() const { return CodeDirectory::hashFor(this->mHashType); }
+	
+private:
+	Hashing::Byte *specialSlot(SpecialSlot slot)
+		{ assert(slot > 0 && slot <= cdSlotMax); return mSpecial + (slot - 1) * mDigestLength; }
+	Hashing::Byte *specialSlot(SpecialSlot slot) const
+		{ assert(slot > 0 && slot <= cdSlotMax); return mSpecial + (slot - 1) * mDigestLength; }
+	
+private:
+	Hashing::Byte *mSpecial;					// array of special slot hashes
+	UnixPlusPlus::AutoFileDesc mExec;			// main executable file
+	size_t mExecOffset;							// starting offset in mExec
+	size_t mExecLength;							// total bytes of file to sign
+	size_t mPageSize;							// page size of executable (bytes)
+	uint32_t mFlags;							// CodeDirectory flags
+	uint32_t mHashType;							// digest algorithm code
+	uint32_t mDigestLength;						// number of bytes in a single glue digest
+	std::string mIdentifier;					// canonical identifier
+	std::string mTeamID;                        // team identifier
+	
+	size_t mSpecialSlots;						// highest special slot set
+	size_t mCodeSlots;							// number of code pages (slots)
+	
+	Scatter *mScatter;							// scatter vector
+	size_t mScatterSize;						// number of scatter elements allocated (incl. sentinel)
+	
+	CodeDirectory *mDir;						// what we're building
+};
+
+
+}	// CodeSigning
+}	// Security
+
+
+#endif //_H_CDBUILDER