Reachability.swift 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317
  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 { [weak self] in
  182. guard let self = self else { return }
  183. var flags = SCNetworkReachabilityFlags()
  184. if !SCNetworkReachabilityGetFlags(self.reachabilityRef, &flags) {
  185. self.stopNotifier()
  186. throw ReachabilityError.UnableToGetInitialFlags
  187. }
  188. self.flags = flags
  189. }
  190. }
  191. func reachabilityChanged() {
  192. let block = connection != .none ? whenReachable : whenUnreachable
  193. DispatchQueue.main.async { [weak self] in
  194. guard let self = self else { return }
  195. block?(self)
  196. self.notificationCenter.post(name: .reachabilityChanged, object: self)
  197. }
  198. }
  199. }
  200. extension SCNetworkReachabilityFlags {
  201. typealias Connection = Reachability.Connection
  202. var connection: Connection {
  203. guard isReachableFlagSet else { return .none }
  204. // If we're reachable, but not on an iOS device (i.e. simulator), we must be on WiFi
  205. #if targetEnvironment(simulator)
  206. return .wifi
  207. #else
  208. var connection = Connection.none
  209. if !isConnectionRequiredFlagSet {
  210. connection = .wifi
  211. }
  212. if isConnectionOnTrafficOrDemandFlagSet {
  213. if !isInterventionRequiredFlagSet {
  214. connection = .wifi
  215. }
  216. }
  217. if isOnWWANFlagSet {
  218. connection = .cellular
  219. }
  220. return connection
  221. #endif
  222. }
  223. var isOnWWANFlagSet: Bool {
  224. #if os(iOS)
  225. return contains(.isWWAN)
  226. #else
  227. return false
  228. #endif
  229. }
  230. var isReachableFlagSet: Bool {
  231. return contains(.reachable)
  232. }
  233. var isConnectionRequiredFlagSet: Bool {
  234. return contains(.connectionRequired)
  235. }
  236. var isInterventionRequiredFlagSet: Bool {
  237. return contains(.interventionRequired)
  238. }
  239. var isConnectionOnTrafficFlagSet: Bool {
  240. return contains(.connectionOnTraffic)
  241. }
  242. var isConnectionOnDemandFlagSet: Bool {
  243. return contains(.connectionOnDemand)
  244. }
  245. var isConnectionOnTrafficOrDemandFlagSet: Bool {
  246. return !intersection([.connectionOnTraffic, .connectionOnDemand]).isEmpty
  247. }
  248. var isTransientConnectionFlagSet: Bool {
  249. return contains(.transientConnection)
  250. }
  251. var isLocalAddressFlagSet: Bool {
  252. return contains(.isLocalAddress)
  253. }
  254. var isDirectFlagSet: Bool {
  255. return contains(.isDirect)
  256. }
  257. var isConnectionRequiredAndTransientFlagSet: Bool {
  258. return intersection([.connectionRequired, .transientConnection]) == [.connectionRequired, .transientConnection]
  259. }
  260. }