X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/79b9da22a1f4b26279940d285c1bc28ce4e99252..7e6b461318c8a779d91381531435a68ee4e8b6ed:/OSX/libsecurity_cms/lib/CMSEncoder.cpp

diff --git a/OSX/libsecurity_cms/lib/CMSEncoder.cpp b/OSX/libsecurity_cms/lib/CMSEncoder.cpp
index de3b2c1e..27d6810d 100644
--- a/OSX/libsecurity_cms/lib/CMSEncoder.cpp
+++ b/OSX/libsecurity_cms/lib/CMSEncoder.cpp
@@ -25,8 +25,8 @@
  * CMSEncoder.cpp - encode, sign, and/or encrypt CMS messages. 
  */
  
-#include "CMSEncoder.h"
-#include "CMSPrivate.h"
+#include <Security/CMSEncoder.h>
+#include <Security/CMSPrivate.h>
 #include "CMSUtils.h"
 #include <Security/SecBase.h>
 #include <Security/SecCmsEncoder.h>
@@ -468,6 +468,9 @@ static OSStatus cmsSetupForSignedData(
 		case kCMSCertificateChainWithRoot:
 			chainMode = SecCmsCMCertChainWithRoot;
 			break;
+		case kCMSCertificateChainWithRootOrFail:
+			chainMode = SecCmsCMCertChainWithRootOrFail;
+			break;
 		default:
 			break;
 	}
@@ -1101,6 +1104,7 @@ OSStatus CMSEncoderSetCertificateChainMode(
 		case kCMSCertificateSignerOnly:
 		case kCMSCertificateChain:
 		case kCMSCertificateChainWithRoot:
+		case kCMSCertificateChainWithRootOrFail:
 			break;
 		default:
 			return errSecParam;