Reachability.swift 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316
  1. /*
  2. Copyright (c) 2014, Ashley Mills
  3. All rights reserved.
  4. Redistribution and use in source and binary forms, with or without
  5. modification, are permitted provided that the following conditions are met:
  6. 1. Redistributions of source code must retain the above copyright notice, this
  7. list of conditions and the following disclaimer.
  8. 2. Redistributions in binary form must reproduce the above copyright notice,
  9. this list of conditions and the following disclaimer in the documentation
  10. and/or other materials provided with the distribution.
  11. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  12. AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  13. IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  14. ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  15. LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  16. CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  17. SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  18. INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  19. CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  20. ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  21. POSSIBILITY OF SUCH DAMAGE.
  22. */
  23. import SystemConfiguration
  24. import Foundation
  25. public enum ReachabilityError: Error {
  26. case FailedToCreateWithAddress(sockaddr_in)
  27. case FailedToCreateWithHostname(String)
  28. case UnableToSetCallback
  29. case UnableToSetDispatchQueue
  30. case UnableToGetInitialFlags
  31. }
  32. @available(*, unavailable, renamed: "Notification.Name.reachabilityChanged")
  33. public let ReachabilityChangedNotification = NSNotification.Name("ReachabilityChangedNotification")
  34. public extension Notification.Name {
  35. static let reachabilityChanged = Notification.Name("reachabilityChanged")
  36. }
  37. public class Reachability {
  38. public typealias NetworkReachable = (Reachability) -> ()
  39. public typealias NetworkUnreachable = (Reachability) -> ()
  40. @available(*, unavailable, renamed: "Connection")
  41. public enum NetworkStatus: CustomStringConvertible {
  42. case notReachable, reachableViaWiFi, reachableViaWWAN
  43. public var description: String {
  44. switch self {
  45. case .reachableViaWWAN: return "Cellular"
  46. case .reachableViaWiFi: return "WiFi"
  47. case .notReachable: return "No Connection"
  48. }
  49. }
  50. }
  51. public enum Connection: CustomStringConvertible {
  52. case none, wifi, cellular
  53. public var description: String {
  54. switch self {
  55. case .cellular: return "Cellular"
  56. case .wifi: return "WiFi"
  57. case .none: return "No Connection"
  58. }
  59. }
  60. }
  61. public var whenReachable: NetworkReachable?
  62. public var whenUnreachable: NetworkUnreachable?
  63. @available(*, deprecated, renamed: "allowsCellularConnection")
  64. public let reachableOnWWAN: Bool = true
  65. /// Set to `false` to force Reachability.connection to .none when on cellular connection (default value `true`)
  66. public var allowsCellularConnection: Bool
  67. // The notification center on which "reachability changed" events are being posted
  68. public var notificationCenter: NotificationCenter = NotificationCenter.default
  69. @available(*, deprecated, renamed: "connection.description")
  70. public var currentReachabilityString: String {
  71. return "\(connection)"
  72. }
  73. @available(*, unavailable, renamed: "connection")
  74. public var currentReachabilityStatus: Connection {
  75. return connection
  76. }
  77. public var connection: Connection {
  78. if flags == nil {
  79. try? setReachabilityFlags()
  80. }
  81. switch flags?.connection {
  82. case .none?, nil: return .none
  83. case .cellular?: return allowsCellularConnection ? .cellular : .none
  84. case .wifi?: return .wifi
  85. }
  86. }
  87. fileprivate var isRunningOnDevice: Bool = {
  88. #if targetEnvironment(simulator)
  89. return false
  90. #else
  91. return true
  92. #endif
  93. }()
  94. fileprivate var notifierRunning = false
  95. fileprivate let reachabilityRef: SCNetworkReachability
  96. fileprivate let reachabilitySerialQueue: DispatchQueue
  97. fileprivate(set) var flags: SCNetworkReachabilityFlags? {
  98. didSet {
  99. guard flags != oldValue else { return }
  100. reachabilityChanged()
  101. }
  102. }
  103. required public init(reachabilityRef: SCNetworkReachability, queueQoS: DispatchQoS = .default, targetQueue: DispatchQueue? = nil) {
  104. self.allowsCellularConnection = true
  105. self.reachabilityRef = reachabilityRef
  106. self.reachabilitySerialQueue = DispatchQueue(label: "uk.co.ashleymills.reachability", qos: queueQoS, target: targetQueue)
  107. }
  108. public convenience init?(hostname: String, queueQoS: DispatchQoS = .default, targetQueue: DispatchQueue? = nil) {
  109. guard let ref = SCNetworkReachabilityCreateWithName(nil, hostname) else { return nil }
  110. self.init(reachabilityRef: ref, queueQoS: queueQoS, targetQueue: targetQueue)
  111. }
  112. public convenience init?(queueQoS: DispatchQoS = .default, targetQueue: DispatchQueue? = nil) {
  113. var zeroAddress = sockaddr()
  114. zeroAddress.sa_len = UInt8(MemoryLayout<sockaddr>.size)
  115. zeroAddress.sa_family = sa_family_t(AF_INET)
  116. guard let ref = SCNetworkReachabilityCreateWithAddress(nil, &zeroAddress) else { return nil }
  117. self.init(reachabilityRef: ref, queueQoS: queueQoS, targetQueue: targetQueue)
  118. }
  119. deinit {
  120. stopNotifier()
  121. }
  122. }
  123. public extension Reachability {
  124. // MARK: - *** Notifier methods ***
  125. func startNotifier() throws {
  126. guard !notifierRunning else { return }
  127. let callback: SCNetworkReachabilityCallBack = { (reachability, flags, info) in
  128. guard let info = info else { return }
  129. let reachability = Unmanaged<Reachability>.fromOpaque(info).takeUnretainedValue()
  130. reachability.flags = flags
  131. }
  132. var context = SCNetworkReachabilityContext(version: 0, info: nil, retain: nil, release: nil, copyDescription: nil)
  133. context.info = UnsafeMutableRawPointer(Unmanaged<Reachability>.passUnretained(self).toOpaque())
  134. if !SCNetworkReachabilitySetCallback(reachabilityRef, callback, &context) {
  135. stopNotifier()
  136. throw ReachabilityError.UnableToSetCallback
  137. }
  138. if !SCNetworkReachabilitySetDispatchQueue(reachabilityRef, reachabilitySerialQueue) {
  139. stopNotifier()
  140. throw ReachabilityError.UnableToSetDispatchQueue
  141. }
  142. // Perform an initial check
  143. try setReachabilityFlags()
  144. notifierRunning = true
  145. }
  146. func stopNotifier() {
  147. defer { notifierRunning = false }
  148. SCNetworkReachabilitySetCallback(reachabilityRef, nil, nil)
  149. SCNetworkReachabilitySetDispatchQueue(reachabilityRef, nil)
  150. }
  151. // MARK: - *** Connection test methods ***
  152. @available(*, deprecated, message: "Please use `connection != .none`")
  153. var isReachable: Bool {
  154. return connection != .none
  155. }
  156. @available(*, deprecated, message: "Please use `connection == .cellular`")
  157. var isReachableViaWWAN: Bool {
  158. // Check we're not on the simulator, we're REACHABLE and check we're on WWAN
  159. return connection == .cellular
  160. }
  161. @available(*, deprecated, message: "Please use `connection == .wifi`")
  162. var isReachableViaWiFi: Bool {
  163. return connection == .wifi
  164. }
  165. var description: String {
  166. guard let flags = flags else { return "unavailable flags" }
  167. let W = isRunningOnDevice ? (flags.isOnWWANFlagSet ? "W" : "-") : "X"
  168. let R = flags.isReachableFlagSet ? "R" : "-"
  169. let c = flags.isConnectionRequiredFlagSet ? "c" : "-"
  170. let t = flags.isTransientConnectionFlagSet ? "t" : "-"
  171. let i = flags.isInterventionRequiredFlagSet ? "i" : "-"
  172. let C = flags.isConnectionOnTrafficFlagSet ? "C" : "-"
  173. let D = flags.isConnectionOnDemandFlagSet ? "D" : "-"
  174. let l = flags.isLocalAddressFlagSet ? "l" : "-"
  175. let d = flags.isDirectFlagSet ? "d" : "-"
  176. return "\(W)\(R) \(c)\(t)\(i)\(C)\(D)\(l)\(d)"
  177. }
  178. }
  179. fileprivate extension Reachability {
  180. func setReachabilityFlags() throws {
  181. try reachabilitySerialQueue.sync { [unowned self] in
  182. var flags = SCNetworkReachabilityFlags()
  183. if !SCNetworkReachabilityGetFlags(self.reachabilityRef, &flags) {
  184. self.stopNotifier()
  185. throw ReachabilityError.UnableToGetInitialFlags
  186. }
  187. self.flags = flags
  188. }
  189. }
  190. func reachabilityChanged() {
  191. let block = connection != .none ? whenReachable : whenUnreachable
  192. DispatchQueue.main.async { [weak self] in
  193. guard let self = self else { return }
  194. block?(self)
  195. self.notificationCenter.post(name: .reachabilityChanged, object: self)
  196. }
  197. }
  198. }
  199. extension SCNetworkReachabilityFlags {
  200. typealias Connection = Reachability.Connection
  201. var connection: Connection {
  202. guard isReachableFlagSet else { return .none }
  203. // If we're reachable, but not on an iOS device (i.e. simulator), we must be on WiFi
  204. #if targetEnvironment(simulator)
  205. return .wifi
  206. #else
  207. var connection = Connection.none
  208. if !isConnectionRequiredFlagSet {
  209. connection = .wifi
  210. }
  211. if isConnectionOnTrafficOrDemandFlagSet {
  212. if !isInterventionRequiredFlagSet {
  213. connection = .wifi
  214. }
  215. }
  216. if isOnWWANFlagSet {
  217. connection = .cellular
  218. }
  219. return connection
  220. #endif
  221. }
  222. var isOnWWANFlagSet: Bool {
  223. #if os(iOS)
  224. return contains(.isWWAN)
  225. #else
  226. return false
  227. #endif
  228. }
  229. var isReachableFlagSet: Bool {
  230. return contains(.reachable)
  231. }
  232. var isConnectionRequiredFlagSet: Bool {
  233. return contains(.connectionRequired)
  234. }
  235. var isInterventionRequiredFlagSet: Bool {
  236. return contains(.interventionRequired)
  237. }
  238. var isConnectionOnTrafficFlagSet: Bool {
  239. return contains(.connectionOnTraffic)
  240. }
  241. var isConnectionOnDemandFlagSet: Bool {
  242. return contains(.connectionOnDemand)
  243. }
  244. var isConnectionOnTrafficOrDemandFlagSet: Bool {
  245. return !intersection([.connectionOnTraffic, .connectionOnDemand]).isEmpty
  246. }
  247. var isTransientConnectionFlagSet: Bool {
  248. return contains(.transientConnection)
  249. }
  250. var isLocalAddressFlagSet: Bool {
  251. return contains(.isLocalAddress)
  252. }
  253. var isDirectFlagSet: Bool {
  254. return contains(.isDirect)
  255. }
  256. var isConnectionRequiredAndTransientFlagSet: Bool {
  257. return intersection([.connectionRequired, .transientConnection]) == [.connectionRequired, .transientConnection]
  258. }
  259. }