X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/bac41a7b9a0a9254fa30f8bb6e6038ab71a483e2..67c7378dcb8de24c86b7fedff90b4b496f2e474c:/AppleCSP/CryptKitCSP/cryptkitcsp.cpp?ds=sidebyside

diff --git a/AppleCSP/CryptKitCSP/cryptkitcsp.cpp b/AppleCSP/CryptKitCSP/cryptkitcsp.cpp
index 24827a03..ea97af75 100644
--- a/AppleCSP/CryptKitCSP/cryptkitcsp.cpp
+++ b/AppleCSP/CryptKitCSP/cryptkitcsp.cpp
@@ -31,6 +31,7 @@
 #include <CryptKit/falloc.h>
 #include <CryptKit/feeFunctions.h>
 #include <MiscCSPAlgs/SHA1_MD5_Object.h>
+#include <Security/digestobject.h>
 
 CssmAllocator *CryptKitFactory::normAllocator;
 CssmAllocator *CryptKitFactory::privAllocator;
@@ -107,6 +108,26 @@ bool CryptKitFactory::setup(
 								*privAllocator)));
 					}
 					return true;
+				case CSSM_ALGID_FEE:
+					if(cspCtx == NULL) {
+						cspCtx = new SignatureContext(session,
+							*(new NullDigest()),
+							*(new FEERawSigner(feeRandCallback, 
+								&session,
+								session,
+								*privAllocator)));
+					}
+					return true;
+				case CSSM_ALGID_ECDSA:
+					if(cspCtx == NULL) {
+						cspCtx = new SignatureContext(session,
+							*(new NullDigest()),
+							*(new FEEECDSASigner(feeRandCallback, 
+								&session,
+								session,
+								*privAllocator)));
+					}
+					return true;
 				default:
 					break;
 			}