ServerTrustEvaluation.swift 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728
  1. //
  2. // ServerTrustPolicy.swift
  3. //
  4. // Copyright (c) 2014-2016 Alamofire Software Foundation (http://alamofire.org/)
  5. //
  6. // Permission is hereby granted, free of charge, to any person obtaining a copy
  7. // of this software and associated documentation files (the "Software"), to deal
  8. // in the Software without restriction, including without limitation the rights
  9. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. // copies of the Software, and to permit persons to whom the Software is
  11. // furnished to do so, subject to the following conditions:
  12. //
  13. // The above copyright notice and this permission notice shall be included in
  14. // all copies or substantial portions of the Software.
  15. //
  16. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22. // THE SOFTWARE.
  23. //
  24. import Foundation
  25. /// Responsible for managing the mapping of `ServerTrustEvaluating` values to given hosts.
  26. open class ServerTrustManager {
  27. /// Determines whether all hosts for this `ServerTrustManager` must be evaluated. `true` by default.
  28. public let allHostsMustBeEvaluated: Bool
  29. /// The dictionary of policies mapped to a particular host.
  30. public let evaluators: [String: ServerTrustEvaluating]
  31. /// Initializes the `ServerTrustManager` instance with the given evaluators.
  32. ///
  33. /// Since different servers and web services can have different leaf certificates, intermediate and even root
  34. /// certificates, it is important to have the flexibility to specify evaluation policies on a per host basis. This
  35. /// allows for scenarios such as using default evaluation for host1, certificate pinning for host2, public key
  36. /// pinning for host3 and disabling evaluation for host4.
  37. ///
  38. /// - Parameters:
  39. /// - allHostsMustBeEvaluated: The value determining whether all hosts for this instance must be evaluated. `true`
  40. /// by default.
  41. /// - evaluators: A dictionary of evaluators mapped to hosts.
  42. public init(allHostsMustBeEvaluated: Bool = true, evaluators: [String: ServerTrustEvaluating]) {
  43. self.allHostsMustBeEvaluated = allHostsMustBeEvaluated
  44. self.evaluators = evaluators
  45. }
  46. #if !(os(Linux) || os(Windows))
  47. /// Returns the `ServerTrustEvaluating` value for the given host, if one is set.
  48. ///
  49. /// By default, this method will return the policy that perfectly matches the given host. Subclasses could override
  50. /// this method and implement more complex mapping implementations such as wildcards.
  51. ///
  52. /// - Parameter host: The host to use when searching for a matching policy.
  53. ///
  54. /// - Returns: The `ServerTrustEvaluating` value for the given host if found, `nil` otherwise.
  55. /// - Throws: `AFError.serverTrustEvaluationFailed` if `allHostsMustBeEvaluated` is `true` and no matching
  56. /// evaluators are found.
  57. open func serverTrustEvaluator(forHost host: String) throws -> ServerTrustEvaluating? {
  58. guard let evaluator = evaluators[host] else {
  59. if allHostsMustBeEvaluated {
  60. throw AFError.serverTrustEvaluationFailed(reason: .noRequiredEvaluator(host: host))
  61. }
  62. return nil
  63. }
  64. return evaluator
  65. }
  66. #endif
  67. }
  68. /// A protocol describing the API used to evaluate server trusts.
  69. public protocol ServerTrustEvaluating {
  70. #if os(Linux) || os(Windows)
  71. // Implement this once Linux/Windows has API for evaluating server trusts.
  72. #else
  73. /// Evaluates the given `SecTrust` value for the given `host`.
  74. ///
  75. /// - Parameters:
  76. /// - trust: The `SecTrust` value to evaluate.
  77. /// - host: The host for which to evaluate the `SecTrust` value.
  78. ///
  79. /// - Returns: A `Bool` indicating whether the evaluator considers the `SecTrust` value valid for `host`.
  80. func evaluate(_ trust: SecTrust, forHost host: String) throws
  81. #endif
  82. }
  83. // MARK: - Server Trust Evaluators
  84. #if !(os(Linux) || os(Windows))
  85. /// An evaluator which uses the default server trust evaluation while allowing you to control whether to validate the
  86. /// host provided by the challenge. Applications are encouraged to always validate the host in production environments
  87. /// to guarantee the validity of the server's certificate chain.
  88. public final class DefaultTrustEvaluator: ServerTrustEvaluating {
  89. private let validateHost: Bool
  90. /// Creates a `DefaultTrustEvaluator`.
  91. ///
  92. /// - Parameter validateHost: Determines whether or not the evaluator should validate the host. `true` by default.
  93. public init(validateHost: Bool = true) {
  94. self.validateHost = validateHost
  95. }
  96. public func evaluate(_ trust: SecTrust, forHost host: String) throws {
  97. if validateHost {
  98. try trust.af.performValidation(forHost: host)
  99. }
  100. try trust.af.performDefaultValidation(forHost: host)
  101. }
  102. }
  103. /// An evaluator which Uses the default and revoked server trust evaluations allowing you to control whether to validate
  104. /// the host provided by the challenge as well as specify the revocation flags for testing for revoked certificates.
  105. /// Apple platforms did not start testing for revoked certificates automatically until iOS 10.1, macOS 10.12 and tvOS
  106. /// 10.1 which is demonstrated in our TLS tests. Applications are encouraged to always validate the host in production
  107. /// environments to guarantee the validity of the server's certificate chain.
  108. public final class RevocationTrustEvaluator: ServerTrustEvaluating {
  109. /// Represents the options to be use when evaluating the status of a certificate.
  110. /// Only Revocation Policy Constants are valid, and can be found in [Apple's documentation](https://developer.apple.com/documentation/security/certificate_key_and_trust_services/policies/1563600-revocation_policy_constants).
  111. public struct Options: OptionSet {
  112. /// Perform revocation checking using the CRL (Certification Revocation List) method.
  113. public static let crl = Options(rawValue: kSecRevocationCRLMethod)
  114. /// Consult only locally cached replies; do not use network access.
  115. public static let networkAccessDisabled = Options(rawValue: kSecRevocationNetworkAccessDisabled)
  116. /// Perform revocation checking using OCSP (Online Certificate Status Protocol).
  117. public static let ocsp = Options(rawValue: kSecRevocationOCSPMethod)
  118. /// Prefer CRL revocation checking over OCSP; by default, OCSP is preferred.
  119. public static let preferCRL = Options(rawValue: kSecRevocationPreferCRL)
  120. /// Require a positive response to pass the policy. If the flag is not set, revocation checking is done on a
  121. /// "best attempt" basis, where failure to reach the server is not considered fatal.
  122. public static let requirePositiveResponse = Options(rawValue: kSecRevocationRequirePositiveResponse)
  123. /// Perform either OCSP or CRL checking. The checking is performed according to the method(s) specified in the
  124. /// certificate and the value of `preferCRL`.
  125. public static let any = Options(rawValue: kSecRevocationUseAnyAvailableMethod)
  126. /// The raw value of the option.
  127. public let rawValue: CFOptionFlags
  128. /// Creates an `Options` value with the given `CFOptionFlags`.
  129. ///
  130. /// - Parameter rawValue: The `CFOptionFlags` value to initialize with.
  131. public init(rawValue: CFOptionFlags) {
  132. self.rawValue = rawValue
  133. }
  134. }
  135. private let performDefaultValidation: Bool
  136. private let validateHost: Bool
  137. private let options: Options
  138. /// Creates a `RevocationTrustEvaluator` using the provided parameters.
  139. ///
  140. /// - Note: Default and host validation will fail when using this evaluator with self-signed certificates. Use
  141. /// `PinnedCertificatesTrustEvaluator` if you need to use self-signed certificates.
  142. ///
  143. /// - Parameters:
  144. /// - performDefaultValidation: Determines whether default validation should be performed in addition to
  145. /// evaluating the pinned certificates. `true` by default.
  146. /// - validateHost: Determines whether or not the evaluator should validate the host, in addition to
  147. /// performing the default evaluation, even if `performDefaultValidation` is `false`.
  148. /// `true` by default.
  149. /// - options: The `Options` to use to check the revocation status of the certificate. `.any` by
  150. /// default.
  151. public init(performDefaultValidation: Bool = true, validateHost: Bool = true, options: Options = .any) {
  152. self.performDefaultValidation = performDefaultValidation
  153. self.validateHost = validateHost
  154. self.options = options
  155. }
  156. public func evaluate(_ trust: SecTrust, forHost host: String) throws {
  157. if performDefaultValidation {
  158. try trust.af.performDefaultValidation(forHost: host)
  159. }
  160. if validateHost {
  161. try trust.af.performValidation(forHost: host)
  162. }
  163. if #available(iOS 12, macOS 10.14, tvOS 12, watchOS 5, *) {
  164. try trust.af.evaluate(afterApplying: SecPolicy.af.revocation(options: options))
  165. } else {
  166. try trust.af.validate(policy: SecPolicy.af.revocation(options: options)) { status, result in
  167. AFError.serverTrustEvaluationFailed(reason: .revocationCheckFailed(output: .init(host, trust, status, result), options: options))
  168. }
  169. }
  170. }
  171. }
  172. extension ServerTrustEvaluating where Self == RevocationTrustEvaluator {
  173. /// Provides a default `RevocationTrustEvaluator` instance.
  174. public static var revocationChecking: RevocationTrustEvaluator { RevocationTrustEvaluator() }
  175. /// Creates a `RevocationTrustEvaluator` using the provided parameters.
  176. ///
  177. /// - Note: Default and host validation will fail when using this evaluator with self-signed certificates. Use
  178. /// `PinnedCertificatesTrustEvaluator` if you need to use self-signed certificates.
  179. ///
  180. /// - Parameters:
  181. /// - performDefaultValidation: Determines whether default validation should be performed in addition to
  182. /// evaluating the pinned certificates. `true` by default.
  183. /// - validateHost: Determines whether or not the evaluator should validate the host, in addition
  184. /// to performing the default evaluation, even if `performDefaultValidation` is
  185. /// `false`. `true` by default.
  186. /// - options: The `Options` to use to check the revocation status of the certificate. `.any`
  187. /// by default.
  188. /// - Returns: The `RevocationTrustEvaluator`.
  189. public static func revocationChecking(performDefaultValidation: Bool = true,
  190. validateHost: Bool = true,
  191. options: RevocationTrustEvaluator.Options = .any) -> RevocationTrustEvaluator {
  192. RevocationTrustEvaluator(performDefaultValidation: performDefaultValidation,
  193. validateHost: validateHost,
  194. options: options)
  195. }
  196. }
  197. /// Uses the pinned certificates to validate the server trust. The server trust is considered valid if one of the pinned
  198. /// certificates match one of the server certificates. By validating both the certificate chain and host, certificate
  199. /// pinning provides a very secure form of server trust validation mitigating most, if not all, MITM attacks.
  200. /// Applications are encouraged to always validate the host and require a valid certificate chain in production
  201. /// environments.
  202. public final class PinnedCertificatesTrustEvaluator: ServerTrustEvaluating {
  203. private let certificates: [SecCertificate]
  204. private let acceptSelfSignedCertificates: Bool
  205. private let performDefaultValidation: Bool
  206. private let validateHost: Bool
  207. /// Creates a `PinnedCertificatesTrustEvaluator` from the provided parameters.
  208. ///
  209. /// - Parameters:
  210. /// - certificates: The certificates to use to evaluate the trust. All `cer`, `crt`, and `der`
  211. /// certificates in `Bundle.main` by default.
  212. /// - acceptSelfSignedCertificates: Adds the provided certificates as anchors for the trust evaluation, allowing
  213. /// self-signed certificates to pass. `false` by default. THIS SETTING SHOULD BE
  214. /// FALSE IN PRODUCTION!
  215. /// - performDefaultValidation: Determines whether default validation should be performed in addition to
  216. /// evaluating the pinned certificates. `true` by default.
  217. /// - validateHost: Determines whether or not the evaluator should validate the host, in addition
  218. /// to performing the default evaluation, even if `performDefaultValidation` is
  219. /// `false`. `true` by default.
  220. public init(certificates: [SecCertificate] = Bundle.main.af.certificates,
  221. acceptSelfSignedCertificates: Bool = false,
  222. performDefaultValidation: Bool = true,
  223. validateHost: Bool = true) {
  224. self.certificates = certificates
  225. self.acceptSelfSignedCertificates = acceptSelfSignedCertificates
  226. self.performDefaultValidation = performDefaultValidation
  227. self.validateHost = validateHost
  228. }
  229. public func evaluate(_ trust: SecTrust, forHost host: String) throws {
  230. guard !certificates.isEmpty else {
  231. throw AFError.serverTrustEvaluationFailed(reason: .noCertificatesFound)
  232. }
  233. if acceptSelfSignedCertificates {
  234. try trust.af.setAnchorCertificates(certificates)
  235. }
  236. if performDefaultValidation {
  237. try trust.af.performDefaultValidation(forHost: host)
  238. }
  239. if validateHost {
  240. try trust.af.performValidation(forHost: host)
  241. }
  242. let serverCertificatesData = Set(trust.af.certificateData)
  243. let pinnedCertificatesData = Set(certificates.af.data)
  244. let pinnedCertificatesInServerData = !serverCertificatesData.isDisjoint(with: pinnedCertificatesData)
  245. if !pinnedCertificatesInServerData {
  246. throw AFError.serverTrustEvaluationFailed(reason: .certificatePinningFailed(host: host,
  247. trust: trust,
  248. pinnedCertificates: certificates,
  249. serverCertificates: trust.af.certificates))
  250. }
  251. }
  252. }
  253. extension ServerTrustEvaluating where Self == PinnedCertificatesTrustEvaluator {
  254. /// Provides a default `PinnedCertificatesTrustEvaluator` instance.
  255. public static var pinnedCertificates: PinnedCertificatesTrustEvaluator { PinnedCertificatesTrustEvaluator() }
  256. /// Creates a `PinnedCertificatesTrustEvaluator` using the provided parameters.
  257. ///
  258. /// - Parameters:
  259. /// - certificates: The certificates to use to evaluate the trust. All `cer`, `crt`, and `der`
  260. /// certificates in `Bundle.main` by default.
  261. /// - acceptSelfSignedCertificates: Adds the provided certificates as anchors for the trust evaluation, allowing
  262. /// self-signed certificates to pass. `false` by default. THIS SETTING SHOULD BE
  263. /// FALSE IN PRODUCTION!
  264. /// - performDefaultValidation: Determines whether default validation should be performed in addition to
  265. /// evaluating the pinned certificates. `true` by default.
  266. /// - validateHost: Determines whether or not the evaluator should validate the host, in addition
  267. /// to performing the default evaluation, even if `performDefaultValidation` is
  268. /// `false`. `true` by default.
  269. public static func pinnedCertificates(certificates: [SecCertificate] = Bundle.main.af.certificates,
  270. acceptSelfSignedCertificates: Bool = false,
  271. performDefaultValidation: Bool = true,
  272. validateHost: Bool = true) -> PinnedCertificatesTrustEvaluator {
  273. PinnedCertificatesTrustEvaluator(certificates: certificates,
  274. acceptSelfSignedCertificates: acceptSelfSignedCertificates,
  275. performDefaultValidation: performDefaultValidation,
  276. validateHost: validateHost)
  277. }
  278. }
  279. /// Uses the pinned public keys to validate the server trust. The server trust is considered valid if one of the pinned
  280. /// public keys match one of the server certificate public keys. By validating both the certificate chain and host,
  281. /// public key pinning provides a very secure form of server trust validation mitigating most, if not all, MITM attacks.
  282. /// Applications are encouraged to always validate the host and require a valid certificate chain in production
  283. /// environments.
  284. public final class PublicKeysTrustEvaluator: ServerTrustEvaluating {
  285. private let keys: [SecKey]
  286. private let performDefaultValidation: Bool
  287. private let validateHost: Bool
  288. /// Creates a `PublicKeysTrustEvaluator` from the provided parameters.
  289. ///
  290. /// - Note: Default and host validation will fail when using this evaluator with self-signed certificates. Use
  291. /// `PinnedCertificatesTrustEvaluator` if you need to use self-signed certificates.
  292. ///
  293. /// - Parameters:
  294. /// - keys: The `SecKey`s to use to validate public keys. Defaults to the public keys of all
  295. /// certificates included in the main bundle.
  296. /// - performDefaultValidation: Determines whether default validation should be performed in addition to
  297. /// evaluating the pinned certificates. `true` by default.
  298. /// - validateHost: Determines whether or not the evaluator should validate the host, in addition to
  299. /// performing the default evaluation, even if `performDefaultValidation` is `false`.
  300. /// `true` by default.
  301. public init(keys: [SecKey] = Bundle.main.af.publicKeys,
  302. performDefaultValidation: Bool = true,
  303. validateHost: Bool = true) {
  304. self.keys = keys
  305. self.performDefaultValidation = performDefaultValidation
  306. self.validateHost = validateHost
  307. }
  308. public func evaluate(_ trust: SecTrust, forHost host: String) throws {
  309. guard !keys.isEmpty else {
  310. throw AFError.serverTrustEvaluationFailed(reason: .noPublicKeysFound)
  311. }
  312. if performDefaultValidation {
  313. try trust.af.performDefaultValidation(forHost: host)
  314. }
  315. if validateHost {
  316. try trust.af.performValidation(forHost: host)
  317. }
  318. let pinnedKeysInServerKeys: Bool = {
  319. for serverPublicKey in trust.af.publicKeys {
  320. for pinnedPublicKey in keys {
  321. if serverPublicKey == pinnedPublicKey {
  322. return true
  323. }
  324. }
  325. }
  326. return false
  327. }()
  328. if !pinnedKeysInServerKeys {
  329. throw AFError.serverTrustEvaluationFailed(reason: .publicKeyPinningFailed(host: host,
  330. trust: trust,
  331. pinnedKeys: keys,
  332. serverKeys: trust.af.publicKeys))
  333. }
  334. }
  335. }
  336. extension ServerTrustEvaluating where Self == PublicKeysTrustEvaluator {
  337. /// Provides a default `PublicKeysTrustEvaluator` instance.
  338. public static var publicKeys: PublicKeysTrustEvaluator { PublicKeysTrustEvaluator() }
  339. /// Creates a `PublicKeysTrustEvaluator` from the provided parameters.
  340. ///
  341. /// - Note: Default and host validation will fail when using this evaluator with self-signed certificates. Use
  342. /// `PinnedCertificatesTrustEvaluator` if you need to use self-signed certificates.
  343. ///
  344. /// - Parameters:
  345. /// - keys: The `SecKey`s to use to validate public keys. Defaults to the public keys of all
  346. /// certificates included in the main bundle.
  347. /// - performDefaultValidation: Determines whether default validation should be performed in addition to
  348. /// evaluating the pinned certificates. `true` by default.
  349. /// - validateHost: Determines whether or not the evaluator should validate the host, in addition to
  350. /// performing the default evaluation, even if `performDefaultValidation` is `false`.
  351. /// `true` by default.
  352. public static func publicKeys(keys: [SecKey] = Bundle.main.af.publicKeys,
  353. performDefaultValidation: Bool = true,
  354. validateHost: Bool = true) -> PublicKeysTrustEvaluator {
  355. PublicKeysTrustEvaluator(keys: keys, performDefaultValidation: performDefaultValidation, validateHost: validateHost)
  356. }
  357. }
  358. /// Uses the provided evaluators to validate the server trust. The trust is only considered valid if all of the
  359. /// evaluators consider it valid.
  360. public final class CompositeTrustEvaluator: ServerTrustEvaluating {
  361. private let evaluators: [ServerTrustEvaluating]
  362. /// Creates a `CompositeTrustEvaluator` from the provided evaluators.
  363. ///
  364. /// - Parameter evaluators: The `ServerTrustEvaluating` values used to evaluate the server trust.
  365. public init(evaluators: [ServerTrustEvaluating]) {
  366. self.evaluators = evaluators
  367. }
  368. public func evaluate(_ trust: SecTrust, forHost host: String) throws {
  369. try evaluators.evaluate(trust, forHost: host)
  370. }
  371. }
  372. extension ServerTrustEvaluating where Self == CompositeTrustEvaluator {
  373. /// Creates a `CompositeTrustEvaluator` from the provided evaluators.
  374. ///
  375. /// - Parameter evaluators: The `ServerTrustEvaluating` values used to evaluate the server trust.
  376. public static func composite(evaluators: [ServerTrustEvaluating]) -> CompositeTrustEvaluator {
  377. CompositeTrustEvaluator(evaluators: evaluators)
  378. }
  379. }
  380. /// Disables all evaluation which in turn will always consider any server trust as valid.
  381. ///
  382. /// - Note: Instead of disabling server trust evaluation, it's a better idea to configure systems to properly trust test
  383. /// certificates, as outlined in [this Apple tech note](https://developer.apple.com/library/archive/qa/qa1948/_index.html).
  384. ///
  385. /// **THIS EVALUATOR SHOULD NEVER BE USED IN PRODUCTION!**
  386. @available(*, deprecated, renamed: "DisabledTrustEvaluator", message: "DisabledEvaluator has been renamed DisabledTrustEvaluator.")
  387. public typealias DisabledEvaluator = DisabledTrustEvaluator
  388. /// Disables all evaluation which in turn will always consider any server trust as valid.
  389. ///
  390. ///
  391. /// - Note: Instead of disabling server trust evaluation, it's a better idea to configure systems to properly trust test
  392. /// certificates, as outlined in [this Apple tech note](https://developer.apple.com/library/archive/qa/qa1948/_index.html).
  393. ///
  394. /// **THIS EVALUATOR SHOULD NEVER BE USED IN PRODUCTION!**
  395. public final class DisabledTrustEvaluator: ServerTrustEvaluating {
  396. /// Creates an instance.
  397. public init() {}
  398. public func evaluate(_ trust: SecTrust, forHost host: String) throws {}
  399. }
  400. // MARK: - Extensions
  401. extension Array where Element == ServerTrustEvaluating {
  402. #if os(Linux) || os(Windows)
  403. // Add this same convenience method for Linux/Windows.
  404. #else
  405. /// Evaluates the given `SecTrust` value for the given `host`.
  406. ///
  407. /// - Parameters:
  408. /// - trust: The `SecTrust` value to evaluate.
  409. /// - host: The host for which to evaluate the `SecTrust` value.
  410. ///
  411. /// - Returns: Whether or not the evaluator considers the `SecTrust` value valid for `host`.
  412. public func evaluate(_ trust: SecTrust, forHost host: String) throws {
  413. for evaluator in self {
  414. try evaluator.evaluate(trust, forHost: host)
  415. }
  416. }
  417. #endif
  418. }
  419. extension Bundle: AlamofireExtended {}
  420. extension AlamofireExtension where ExtendedType: Bundle {
  421. /// Returns all valid `cer`, `crt`, and `der` certificates in the bundle.
  422. public var certificates: [SecCertificate] {
  423. paths(forResourcesOfTypes: [".cer", ".CER", ".crt", ".CRT", ".der", ".DER"]).compactMap { path in
  424. guard
  425. let certificateData = try? Data(contentsOf: URL(fileURLWithPath: path)) as CFData,
  426. let certificate = SecCertificateCreateWithData(nil, certificateData) else { return nil }
  427. return certificate
  428. }
  429. }
  430. /// Returns all public keys for the valid certificates in the bundle.
  431. public var publicKeys: [SecKey] {
  432. certificates.af.publicKeys
  433. }
  434. /// Returns all pathnames for the resources identified by the provided file extensions.
  435. ///
  436. /// - Parameter types: The filename extensions locate.
  437. ///
  438. /// - Returns: All pathnames for the given filename extensions.
  439. public func paths(forResourcesOfTypes types: [String]) -> [String] {
  440. Array(Set(types.flatMap { type.paths(forResourcesOfType: $0, inDirectory: nil) }))
  441. }
  442. }
  443. extension SecTrust: AlamofireExtended {}
  444. extension AlamofireExtension where ExtendedType == SecTrust {
  445. /// Evaluates `self` after applying the `SecPolicy` value provided.
  446. ///
  447. /// - Parameter policy: The `SecPolicy` to apply to `self` before evaluation.
  448. ///
  449. /// - Throws: Any `Error` from applying the `SecPolicy` or from evaluation.
  450. @available(iOS 12, macOS 10.14, tvOS 12, watchOS 5, *)
  451. public func evaluate(afterApplying policy: SecPolicy) throws {
  452. try apply(policy: policy).af.evaluate()
  453. }
  454. /// Attempts to validate `self` using the `SecPolicy` provided and transforming any error produced using the closure passed.
  455. ///
  456. /// - Parameters:
  457. /// - policy: The `SecPolicy` used to evaluate `self`.
  458. /// - errorProducer: The closure used transform the failed `OSStatus` and `SecTrustResultType`.
  459. /// - Throws: Any `Error` from applying the `policy`, or the result of `errorProducer` if validation fails.
  460. @available(iOS, introduced: 10, deprecated: 12, renamed: "evaluate(afterApplying:)")
  461. @available(macOS, introduced: 10.12, deprecated: 10.14, renamed: "evaluate(afterApplying:)")
  462. @available(tvOS, introduced: 10, deprecated: 12, renamed: "evaluate(afterApplying:)")
  463. @available(watchOS, introduced: 3, deprecated: 5, renamed: "evaluate(afterApplying:)")
  464. public func validate(policy: SecPolicy, errorProducer: (_ status: OSStatus, _ result: SecTrustResultType) -> Error) throws {
  465. try apply(policy: policy).af.validate(errorProducer: errorProducer)
  466. }
  467. /// Applies a `SecPolicy` to `self`, throwing if it fails.
  468. ///
  469. /// - Parameter policy: The `SecPolicy`.
  470. ///
  471. /// - Returns: `self`, with the policy applied.
  472. /// - Throws: An `AFError.serverTrustEvaluationFailed` instance with a `.policyApplicationFailed` reason.
  473. public func apply(policy: SecPolicy) throws -> SecTrust {
  474. let status = SecTrustSetPolicies(type, policy)
  475. guard status.af.isSuccess else {
  476. throw AFError.serverTrustEvaluationFailed(reason: .policyApplicationFailed(trust: type,
  477. policy: policy,
  478. status: status))
  479. }
  480. return type
  481. }
  482. /// Evaluate `self`, throwing an `Error` if evaluation fails.
  483. ///
  484. /// - Throws: `AFError.serverTrustEvaluationFailed` with reason `.trustValidationFailed` and associated error from
  485. /// the underlying evaluation.
  486. @available(iOS 12, macOS 10.14, tvOS 12, watchOS 5, *)
  487. public func evaluate() throws {
  488. var error: CFError?
  489. let evaluationSucceeded = SecTrustEvaluateWithError(type, &error)
  490. if !evaluationSucceeded {
  491. throw AFError.serverTrustEvaluationFailed(reason: .trustEvaluationFailed(error: error))
  492. }
  493. }
  494. /// Validate `self`, passing any failure values through `errorProducer`.
  495. ///
  496. /// - Parameter errorProducer: The closure used to transform the failed `OSStatus` and `SecTrustResultType` into an
  497. /// `Error`.
  498. /// - Throws: The `Error` produced by the `errorProducer` closure.
  499. @available(iOS, introduced: 10, deprecated: 12, renamed: "evaluate()")
  500. @available(macOS, introduced: 10.12, deprecated: 10.14, renamed: "evaluate()")
  501. @available(tvOS, introduced: 10, deprecated: 12, renamed: "evaluate()")
  502. @available(watchOS, introduced: 3, deprecated: 5, renamed: "evaluate()")
  503. public func validate(errorProducer: (_ status: OSStatus, _ result: SecTrustResultType) -> Error) throws {
  504. var result = SecTrustResultType.invalid
  505. let status = SecTrustEvaluate(type, &result)
  506. guard status.af.isSuccess && result.af.isSuccess else {
  507. throw errorProducer(status, result)
  508. }
  509. }
  510. /// Sets a custom certificate chain on `self`, allowing full validation of a self-signed certificate and its chain.
  511. ///
  512. /// - Parameter certificates: The `SecCertificate`s to add to the chain.
  513. /// - Throws: Any error produced when applying the new certificate chain.
  514. public func setAnchorCertificates(_ certificates: [SecCertificate]) throws {
  515. // Add additional anchor certificates.
  516. let status = SecTrustSetAnchorCertificates(type, certificates as CFArray)
  517. guard status.af.isSuccess else {
  518. throw AFError.serverTrustEvaluationFailed(reason: .settingAnchorCertificatesFailed(status: status,
  519. certificates: certificates))
  520. }
  521. // Trust only the set anchor certs.
  522. let onlyStatus = SecTrustSetAnchorCertificatesOnly(type, true)
  523. guard onlyStatus.af.isSuccess else {
  524. throw AFError.serverTrustEvaluationFailed(reason: .settingAnchorCertificatesFailed(status: onlyStatus,
  525. certificates: certificates))
  526. }
  527. }
  528. /// The public keys contained in `self`.
  529. public var publicKeys: [SecKey] {
  530. certificates.af.publicKeys
  531. }
  532. /// The `SecCertificate`s contained in `self`.
  533. public var certificates: [SecCertificate] {
  534. #if swift(>=5.5.1) // Xcode 13.1 / 2021 SDKs.
  535. if #available(iOS 15, macOS 12, tvOS 15, watchOS 8, *) {
  536. return (SecTrustCopyCertificateChain(type) as? [SecCertificate]) ?? []
  537. } else {
  538. return (0..<SecTrustGetCertificateCount(type)).compactMap { index in
  539. SecTrustGetCertificateAtIndex(type, index)
  540. }
  541. }
  542. #else
  543. (0..<SecTrustGetCertificateCount(type)).compactMap { index in
  544. SecTrustGetCertificateAtIndex(type, index)
  545. }
  546. #endif
  547. }
  548. /// The `Data` values for all certificates contained in `self`.
  549. public var certificateData: [Data] {
  550. certificates.af.data
  551. }
  552. /// Validates `self` after applying `SecPolicy.af.default`. This evaluation does not validate the hostname.
  553. ///
  554. /// - Parameter host: The hostname, used only in the error output if validation fails.
  555. /// - Throws: An `AFError.serverTrustEvaluationFailed` instance with a `.defaultEvaluationFailed` reason.
  556. public func performDefaultValidation(forHost host: String) throws {
  557. if #available(iOS 12, macOS 10.14, tvOS 12, watchOS 5, *) {
  558. try evaluate(afterApplying: SecPolicy.af.default)
  559. } else {
  560. try validate(policy: SecPolicy.af.default) { status, result in
  561. AFError.serverTrustEvaluationFailed(reason: .defaultEvaluationFailed(output: .init(host, type, status, result)))
  562. }
  563. }
  564. }
  565. /// Validates `self` after applying `SecPolicy.af.hostname(host)`, which performs the default validation as well as
  566. /// hostname validation.
  567. ///
  568. /// - Parameter host: The hostname to use in the validation.
  569. /// - Throws: An `AFError.serverTrustEvaluationFailed` instance with a `.defaultEvaluationFailed` reason.
  570. public func performValidation(forHost host: String) throws {
  571. if #available(iOS 12, macOS 10.14, tvOS 12, watchOS 5, *) {
  572. try evaluate(afterApplying: SecPolicy.af.hostname(host))
  573. } else {
  574. try validate(policy: SecPolicy.af.hostname(host)) { status, result in
  575. AFError.serverTrustEvaluationFailed(reason: .hostValidationFailed(output: .init(host, type, status, result)))
  576. }
  577. }
  578. }
  579. }
  580. extension SecPolicy: AlamofireExtended {}
  581. extension AlamofireExtension where ExtendedType == SecPolicy {
  582. /// Creates a `SecPolicy` instance which will validate server certificates but not require a host name match.
  583. public static let `default` = SecPolicyCreateSSL(true, nil)
  584. /// Creates a `SecPolicy` instance which will validate server certificates and much match the provided hostname.
  585. ///
  586. /// - Parameter hostname: The hostname to validate against.
  587. ///
  588. /// - Returns: The `SecPolicy`.
  589. public static func hostname(_ hostname: String) -> SecPolicy {
  590. SecPolicyCreateSSL(true, hostname as CFString)
  591. }
  592. /// Creates a `SecPolicy` which checks the revocation of certificates.
  593. ///
  594. /// - Parameter options: The `RevocationTrustEvaluator.Options` for evaluation.
  595. ///
  596. /// - Returns: The `SecPolicy`.
  597. /// - Throws: An `AFError.serverTrustEvaluationFailed` error with reason `.revocationPolicyCreationFailed`
  598. /// if the policy cannot be created.
  599. public static func revocation(options: RevocationTrustEvaluator.Options) throws -> SecPolicy {
  600. guard let policy = SecPolicyCreateRevocation(options.rawValue) else {
  601. throw AFError.serverTrustEvaluationFailed(reason: .revocationPolicyCreationFailed)
  602. }
  603. return policy
  604. }
  605. }
  606. extension Array: AlamofireExtended {}
  607. extension AlamofireExtension where ExtendedType == [SecCertificate] {
  608. /// All `Data` values for the contained `SecCertificate`s.
  609. public var data: [Data] {
  610. type.map { SecCertificateCopyData($0) as Data }
  611. }
  612. /// All public `SecKey` values for the contained `SecCertificate`s.
  613. public var publicKeys: [SecKey] {
  614. type.compactMap(\.af.publicKey)
  615. }
  616. }
  617. extension SecCertificate: AlamofireExtended {}
  618. extension AlamofireExtension where ExtendedType == SecCertificate {
  619. /// The public key for `self`, if it can be extracted.
  620. ///
  621. /// - Note: On 2020 OSes and newer, only RSA and ECDSA keys are supported.
  622. ///
  623. public var publicKey: SecKey? {
  624. let policy = SecPolicyCreateBasicX509()
  625. var trust: SecTrust?
  626. let trustCreationStatus = SecTrustCreateWithCertificates(type, policy, &trust)
  627. guard let createdTrust = trust, trustCreationStatus == errSecSuccess else { return nil }
  628. if #available(iOS 14, macOS 11, tvOS 14, watchOS 7, *) {
  629. return SecTrustCopyKey(createdTrust)
  630. } else {
  631. return SecTrustCopyPublicKey(createdTrust)
  632. }
  633. }
  634. }
  635. extension OSStatus: AlamofireExtended {}
  636. extension AlamofireExtension where ExtendedType == OSStatus {
  637. /// Returns whether `self` is `errSecSuccess`.
  638. public var isSuccess: Bool { type == errSecSuccess }
  639. }
  640. extension SecTrustResultType: AlamofireExtended {}
  641. extension AlamofireExtension where ExtendedType == SecTrustResultType {
  642. /// Returns whether `self is `.unspecified` or `.proceed`.
  643. public var isSuccess: Bool {
  644. type == .unspecified || type == .proceed
  645. }
  646. }
  647. #endif