Files
BlueSocket/Sources/SocketUtils.swift
T

532 lines
18 KiB
Swift

//
// SocketUtils.swift
// BlueSocket
//
// Created by Bill Abt on 11/19/15.
// Copyright © 2016 IBM. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
#if os(macOS) || os(iOS) || os(tvOS) || os(watchOS)
import Darwin
#elseif os(Linux)
import Glibc
#endif
import Foundation
// MARK: sockaddr_storage Extension
public extension sockaddr_storage {
///
/// Cast to sockaddr
///
/// - Returns: sockaddr
///
public func asAddr() -> sockaddr {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr.self).pointee
}
///
/// Cast to sockaddr_in
///
/// - Returns: sockaddr_in
///
public func asIPV4() -> sockaddr_in {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr_in.self).pointee
}
///
/// Cast to sockaddr_in6
///
/// - Returns: sockaddr_in6
///
public func asIPV6() -> sockaddr_in6 {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr_in6.self).pointee
}
///
/// Cast to sockaddr_un
///
/// - Returns: sockaddr_un
///
public func asUnix() -> sockaddr_un {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr_un.self).pointee
}
}
// MARK: sockaddr_in Extension
public extension sockaddr_in {
///
/// Cast to sockaddr
///
/// - Returns: sockaddr
///
public func asAddr() -> sockaddr {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr.self).pointee
}
}
// MARK: sockaddr_in6 Extension
public extension sockaddr_in6 {
///
/// Cast to sockaddr
///
/// - Returns: sockaddr
///
public func asAddr() -> sockaddr {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr.self).pointee
}
}
// MARK: sockaddr_un Extension
public extension sockaddr_un {
///
/// Cast to sockaddr
///
/// - Returns: sockaddr
///
public func asAddr() -> sockaddr {
var temp = self
let addr = withUnsafePointer(to: &temp) {
return UnsafeRawPointer($0)
}
return addr.assumingMemoryBound(to: sockaddr.self).pointee
}
}
#if arch(arm) && os(Linux)
public struct FD {
/// Replacement for FD_ZERO macro
public static func ZERO(set: inout fd_set) {
set.__fds_bits = (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
}
/// Replacement for FD_SET macro
public static func SET(fd: Int32, set: inout fd_set) {
let intOffset = Int(fd / 32)
let bitOffset: Int = Int(fd % 32)
let mask: Int = 1 << bitOffset
switch intOffset {
case 0: set.__fds_bits.0 = set.__fds_bits.0 | mask
case 1: set.__fds_bits.1 = set.__fds_bits.1 | mask
case 2: set.__fds_bits.2 = set.__fds_bits.2 | mask
case 3: set.__fds_bits.3 = set.__fds_bits.3 | mask
case 4: set.__fds_bits.4 = set.__fds_bits.4 | mask
case 5: set.__fds_bits.5 = set.__fds_bits.5 | mask
case 6: set.__fds_bits.6 = set.__fds_bits.6 | mask
case 7: set.__fds_bits.7 = set.__fds_bits.7 | mask
case 8: set.__fds_bits.8 = set.__fds_bits.8 | mask
case 9: set.__fds_bits.9 = set.__fds_bits.9 | mask
case 10: set.__fds_bits.10 = set.__fds_bits.10 | mask
case 11: set.__fds_bits.11 = set.__fds_bits.11 | mask
case 12: set.__fds_bits.12 = set.__fds_bits.12 | mask
case 13: set.__fds_bits.13 = set.__fds_bits.13 | mask
case 14: set.__fds_bits.14 = set.__fds_bits.14 | mask
case 15: set.__fds_bits.15 = set.__fds_bits.15 | mask
case 16: set.__fds_bits.16 = set.__fds_bits.16 | mask
case 17: set.__fds_bits.17 = set.__fds_bits.17 | mask
case 18: set.__fds_bits.18 = set.__fds_bits.18 | mask
case 19: set.__fds_bits.19 = set.__fds_bits.19 | mask
case 20: set.__fds_bits.20 = set.__fds_bits.20 | mask
case 21: set.__fds_bits.21 = set.__fds_bits.21 | mask
case 22: set.__fds_bits.22 = set.__fds_bits.22 | mask
case 23: set.__fds_bits.23 = set.__fds_bits.23 | mask
case 24: set.__fds_bits.24 = set.__fds_bits.24 | mask
case 25: set.__fds_bits.25 = set.__fds_bits.25 | mask
case 26: set.__fds_bits.26 = set.__fds_bits.26 | mask
case 27: set.__fds_bits.27 = set.__fds_bits.27 | mask
case 28: set.__fds_bits.28 = set.__fds_bits.28 | mask
case 29: set.__fds_bits.29 = set.__fds_bits.29 | mask
case 30: set.__fds_bits.20 = set.__fds_bits.30 | mask
case 31: set.__fds_bits.31 = set.__fds_bits.31 | mask
default: break
}
}
/// Replacement for FD_CLR macro
public static func CLR(fd: Int32, set: inout fd_set) {
let intOffset = Int(fd / 32)
let bitOffset: Int = Int(fd % 32)
let mask: Int = ~(1 << bitOffset)
switch intOffset {
case 0: set.__fds_bits.0 = set.__fds_bits.0 & mask
case 1: set.__fds_bits.1 = set.__fds_bits.1 & mask
case 2: set.__fds_bits.2 = set.__fds_bits.2 & mask
case 3: set.__fds_bits.3 = set.__fds_bits.3 & mask
case 4: set.__fds_bits.4 = set.__fds_bits.4 & mask
case 5: set.__fds_bits.5 = set.__fds_bits.5 & mask
case 6: set.__fds_bits.6 = set.__fds_bits.6 & mask
case 7: set.__fds_bits.7 = set.__fds_bits.7 & mask
case 8: set.__fds_bits.8 = set.__fds_bits.8 & mask
case 9: set.__fds_bits.9 = set.__fds_bits.9 & mask
case 10: set.__fds_bits.10 = set.__fds_bits.10 & mask
case 11: set.__fds_bits.11 = set.__fds_bits.11 & mask
case 12: set.__fds_bits.12 = set.__fds_bits.12 & mask
case 13: set.__fds_bits.13 = set.__fds_bits.13 & mask
case 14: set.__fds_bits.14 = set.__fds_bits.14 & mask
case 15: set.__fds_bits.15 = set.__fds_bits.15 & mask
case 16: set.__fds_bits.16 = set.__fds_bits.16 & mask
case 17: set.__fds_bits.17 = set.__fds_bits.17 & mask
case 18: set.__fds_bits.18 = set.__fds_bits.18 & mask
case 19: set.__fds_bits.19 = set.__fds_bits.19 & mask
case 20: set.__fds_bits.20 = set.__fds_bits.20 & mask
case 21: set.__fds_bits.21 = set.__fds_bits.21 & mask
case 22: set.__fds_bits.22 = set.__fds_bits.22 & mask
case 23: set.__fds_bits.23 = set.__fds_bits.23 & mask
case 24: set.__fds_bits.24 = set.__fds_bits.24 & mask
case 25: set.__fds_bits.25 = set.__fds_bits.25 & mask
case 26: set.__fds_bits.26 = set.__fds_bits.26 & mask
case 27: set.__fds_bits.27 = set.__fds_bits.27 & mask
case 28: set.__fds_bits.28 = set.__fds_bits.28 & mask
case 29: set.__fds_bits.29 = set.__fds_bits.29 & mask
case 30: set.__fds_bits.20 = set.__fds_bits.30 & mask
case 31: set.__fds_bits.31 = set.__fds_bits.31 & mask
default: break
}
}
/// Replacement for FD_ISSET macro
public static func ISSET(fd: Int32, set: inout fd_set) -> Bool {
let intOffset = Int(fd / 32)
let bitOffset = Int(fd % 32)
let mask: Int = 1 << bitOffset
switch intOffset {
case 0: return set.__fds_bits.0 & mask != 0
case 1: return set.__fds_bits.1 & mask != 0
case 2: return set.__fds_bits.2 & mask != 0
case 3: return set.__fds_bits.3 & mask != 0
case 4: return set.__fds_bits.4 & mask != 0
case 5: return set.__fds_bits.5 & mask != 0
case 6: return set.__fds_bits.6 & mask != 0
case 7: return set.__fds_bits.7 & mask != 0
case 8: return set.__fds_bits.8 & mask != 0
case 9: return set.__fds_bits.9 & mask != 0
case 10: return set.__fds_bits.10 & mask != 0
case 11: return set.__fds_bits.11 & mask != 0
case 12: return set.__fds_bits.12 & mask != 0
case 13: return set.__fds_bits.13 & mask != 0
case 14: return set.__fds_bits.14 & mask != 0
case 15: return set.__fds_bits.15 & mask != 0
case 16: return set.__fds_bits.16 & mask != 0
case 17: return set.__fds_bits.17 & mask != 0
case 18: return set.__fds_bits.18 & mask != 0
case 19: return set.__fds_bits.19 & mask != 0
case 20: return set.__fds_bits.20 & mask != 0
case 21: return set.__fds_bits.21 & mask != 0
case 22: return set.__fds_bits.22 & mask != 0
case 23: return set.__fds_bits.23 & mask != 0
case 24: return set.__fds_bits.24 & mask != 0
case 25: return set.__fds_bits.25 & mask != 0
case 26: return set.__fds_bits.26 & mask != 0
case 27: return set.__fds_bits.27 & mask != 0
case 28: return set.__fds_bits.28 & mask != 0
case 29: return set.__fds_bits.29 & mask != 0
case 30: return set.__fds_bits.30 & mask != 0
case 31: return set.__fds_bits.31 & mask != 0
default: return false
}
}
}
#elseif os(Linux)
public struct FD {
/// Replacement for FD_ZERO macro
public static func ZERO(set: inout fd_set) {
set.__fds_bits = (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
}
/// Replacement for FD_SET macro
public static func SET(fd: Int32, set: inout fd_set) {
let intOffset = Int(fd / 16)
let bitOffset: Int = Int(fd % 16)
let mask: Int = 1 << bitOffset
switch intOffset {
case 0: set.__fds_bits.0 = set.__fds_bits.0 | mask
case 1: set.__fds_bits.1 = set.__fds_bits.1 | mask
case 2: set.__fds_bits.2 = set.__fds_bits.2 | mask
case 3: set.__fds_bits.3 = set.__fds_bits.3 | mask
case 4: set.__fds_bits.4 = set.__fds_bits.4 | mask
case 5: set.__fds_bits.5 = set.__fds_bits.5 | mask
case 6: set.__fds_bits.6 = set.__fds_bits.6 | mask
case 7: set.__fds_bits.7 = set.__fds_bits.7 | mask
case 8: set.__fds_bits.8 = set.__fds_bits.8 | mask
case 9: set.__fds_bits.9 = set.__fds_bits.9 | mask
case 10: set.__fds_bits.10 = set.__fds_bits.10 | mask
case 11: set.__fds_bits.11 = set.__fds_bits.11 | mask
case 12: set.__fds_bits.12 = set.__fds_bits.12 | mask
case 13: set.__fds_bits.13 = set.__fds_bits.13 | mask
case 14: set.__fds_bits.14 = set.__fds_bits.14 | mask
case 15: set.__fds_bits.15 = set.__fds_bits.15 | mask
default: break
}
}
/// Replacement for FD_CLR macro
public static func CLR(fd: Int32, set: inout fd_set) {
let intOffset = Int(fd / 16)
let bitOffset: Int = Int(fd % 16)
let mask: Int = ~(1 << bitOffset)
switch intOffset {
case 0: set.__fds_bits.0 = set.__fds_bits.0 & mask
case 1: set.__fds_bits.1 = set.__fds_bits.1 & mask
case 2: set.__fds_bits.2 = set.__fds_bits.2 & mask
case 3: set.__fds_bits.3 = set.__fds_bits.3 & mask
case 4: set.__fds_bits.4 = set.__fds_bits.4 & mask
case 5: set.__fds_bits.5 = set.__fds_bits.5 & mask
case 6: set.__fds_bits.6 = set.__fds_bits.6 & mask
case 7: set.__fds_bits.7 = set.__fds_bits.7 & mask
case 8: set.__fds_bits.8 = set.__fds_bits.8 & mask
case 9: set.__fds_bits.9 = set.__fds_bits.9 & mask
case 10: set.__fds_bits.10 = set.__fds_bits.10 & mask
case 11: set.__fds_bits.11 = set.__fds_bits.11 & mask
case 12: set.__fds_bits.12 = set.__fds_bits.12 & mask
case 13: set.__fds_bits.13 = set.__fds_bits.13 & mask
case 14: set.__fds_bits.14 = set.__fds_bits.14 & mask
case 15: set.__fds_bits.15 = set.__fds_bits.15 & mask
default: break
}
}
/// Replacement for FD_ISSET macro
public static func ISSET(fd: Int32, set: inout fd_set) -> Bool {
let intOffset = Int(fd / 16)
let bitOffset = Int(fd % 16)
let mask: Int = 1 << bitOffset
switch intOffset {
case 0: return set.__fds_bits.0 & mask != 0
case 1: return set.__fds_bits.1 & mask != 0
case 2: return set.__fds_bits.2 & mask != 0
case 3: return set.__fds_bits.3 & mask != 0
case 4: return set.__fds_bits.4 & mask != 0
case 5: return set.__fds_bits.5 & mask != 0
case 6: return set.__fds_bits.6 & mask != 0
case 7: return set.__fds_bits.7 & mask != 0
case 8: return set.__fds_bits.8 & mask != 0
case 9: return set.__fds_bits.9 & mask != 0
case 10: return set.__fds_bits.10 & mask != 0
case 11: return set.__fds_bits.11 & mask != 0
case 12: return set.__fds_bits.12 & mask != 0
case 13: return set.__fds_bits.13 & mask != 0
case 14: return set.__fds_bits.14 & mask != 0
case 15: return set.__fds_bits.15 & mask != 0
default: return false
}
}
}
#else
public struct FD {
/// Replacement for FD_ZERO macro
public static func ZERO(set: inout fd_set) {
set.fds_bits = (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
}
/// Replacement for FD_SET macro
public static func SET(fd: Int32, set: inout fd_set) {
let intOffset = Int(fd / 32)
let bitOffset = fd % 32
let mask = 1 << bitOffset
switch intOffset {
case 0: set.fds_bits.0 = set.fds_bits.0 | mask
case 1: set.fds_bits.1 = set.fds_bits.1 | mask
case 2: set.fds_bits.2 = set.fds_bits.2 | mask
case 3: set.fds_bits.3 = set.fds_bits.3 | mask
case 4: set.fds_bits.4 = set.fds_bits.4 | mask
case 5: set.fds_bits.5 = set.fds_bits.5 | mask
case 6: set.fds_bits.6 = set.fds_bits.6 | mask
case 7: set.fds_bits.7 = set.fds_bits.7 | mask
case 8: set.fds_bits.8 = set.fds_bits.8 | mask
case 9: set.fds_bits.9 = set.fds_bits.9 | mask
case 10: set.fds_bits.10 = set.fds_bits.10 | mask
case 11: set.fds_bits.11 = set.fds_bits.11 | mask
case 12: set.fds_bits.12 = set.fds_bits.12 | mask
case 13: set.fds_bits.13 = set.fds_bits.13 | mask
case 14: set.fds_bits.14 = set.fds_bits.14 | mask
case 15: set.fds_bits.15 = set.fds_bits.15 | mask
case 16: set.fds_bits.16 = set.fds_bits.16 | mask
case 17: set.fds_bits.17 = set.fds_bits.17 | mask
case 18: set.fds_bits.18 = set.fds_bits.18 | mask
case 19: set.fds_bits.19 = set.fds_bits.19 | mask
case 20: set.fds_bits.20 = set.fds_bits.20 | mask
case 21: set.fds_bits.21 = set.fds_bits.21 | mask
case 22: set.fds_bits.22 = set.fds_bits.22 | mask
case 23: set.fds_bits.23 = set.fds_bits.23 | mask
case 24: set.fds_bits.24 = set.fds_bits.24 | mask
case 25: set.fds_bits.25 = set.fds_bits.25 | mask
case 26: set.fds_bits.26 = set.fds_bits.26 | mask
case 27: set.fds_bits.27 = set.fds_bits.27 | mask
case 28: set.fds_bits.28 = set.fds_bits.28 | mask
case 29: set.fds_bits.29 = set.fds_bits.29 | mask
case 30: set.fds_bits.30 = set.fds_bits.30 | mask
case 31: set.fds_bits.31 = set.fds_bits.31 | mask
default: break
}
}
/// Replacement for FD_CLR macro
public static func CLR(fd: Int32, set: inout fd_set) {
let intOffset = Int(fd / 32)
let bitOffset = fd % 32
let mask = ~(1 << bitOffset)
switch intOffset {
case 0: set.fds_bits.0 = set.fds_bits.0 & mask
case 1: set.fds_bits.1 = set.fds_bits.1 & mask
case 2: set.fds_bits.2 = set.fds_bits.2 & mask
case 3: set.fds_bits.3 = set.fds_bits.3 & mask
case 4: set.fds_bits.4 = set.fds_bits.4 & mask
case 5: set.fds_bits.5 = set.fds_bits.5 & mask
case 6: set.fds_bits.6 = set.fds_bits.6 & mask
case 7: set.fds_bits.7 = set.fds_bits.7 & mask
case 8: set.fds_bits.8 = set.fds_bits.8 & mask
case 9: set.fds_bits.9 = set.fds_bits.9 & mask
case 10: set.fds_bits.10 = set.fds_bits.10 & mask
case 11: set.fds_bits.11 = set.fds_bits.11 & mask
case 12: set.fds_bits.12 = set.fds_bits.12 & mask
case 13: set.fds_bits.13 = set.fds_bits.13 & mask
case 14: set.fds_bits.14 = set.fds_bits.14 & mask
case 15: set.fds_bits.15 = set.fds_bits.15 & mask
case 16: set.fds_bits.16 = set.fds_bits.16 & mask
case 17: set.fds_bits.17 = set.fds_bits.17 & mask
case 18: set.fds_bits.18 = set.fds_bits.18 & mask
case 19: set.fds_bits.19 = set.fds_bits.19 & mask
case 20: set.fds_bits.20 = set.fds_bits.20 & mask
case 21: set.fds_bits.21 = set.fds_bits.21 & mask
case 22: set.fds_bits.22 = set.fds_bits.22 & mask
case 23: set.fds_bits.23 = set.fds_bits.23 & mask
case 24: set.fds_bits.24 = set.fds_bits.24 & mask
case 25: set.fds_bits.25 = set.fds_bits.25 & mask
case 26: set.fds_bits.26 = set.fds_bits.26 & mask
case 27: set.fds_bits.27 = set.fds_bits.27 & mask
case 28: set.fds_bits.28 = set.fds_bits.28 & mask
case 29: set.fds_bits.29 = set.fds_bits.29 & mask
case 30: set.fds_bits.30 = set.fds_bits.30 & mask
case 31: set.fds_bits.31 = set.fds_bits.31 & mask
default: break
}
}
/// Replacement for FD_ISSET macro
public static func ISSET(fd: Int32, set: inout fd_set) -> Bool {
let intOffset = Int(fd / 32)
let bitOffset = fd % 32
let mask = 1 << bitOffset
switch intOffset {
case 0: return set.fds_bits.0 & mask != 0
case 1: return set.fds_bits.1 & mask != 0
case 2: return set.fds_bits.2 & mask != 0
case 3: return set.fds_bits.3 & mask != 0
case 4: return set.fds_bits.4 & mask != 0
case 5: return set.fds_bits.5 & mask != 0
case 6: return set.fds_bits.6 & mask != 0
case 7: return set.fds_bits.7 & mask != 0
case 8: return set.fds_bits.8 & mask != 0
case 9: return set.fds_bits.9 & mask != 0
case 10: return set.fds_bits.10 & mask != 0
case 11: return set.fds_bits.11 & mask != 0
case 12: return set.fds_bits.12 & mask != 0
case 13: return set.fds_bits.13 & mask != 0
case 14: return set.fds_bits.14 & mask != 0
case 15: return set.fds_bits.15 & mask != 0
case 16: return set.fds_bits.16 & mask != 0
case 17: return set.fds_bits.17 & mask != 0
case 18: return set.fds_bits.18 & mask != 0
case 19: return set.fds_bits.19 & mask != 0
case 20: return set.fds_bits.20 & mask != 0
case 21: return set.fds_bits.21 & mask != 0
case 22: return set.fds_bits.22 & mask != 0
case 23: return set.fds_bits.23 & mask != 0
case 24: return set.fds_bits.24 & mask != 0
case 25: return set.fds_bits.25 & mask != 0
case 26: return set.fds_bits.26 & mask != 0
case 27: return set.fds_bits.27 & mask != 0
case 28: return set.fds_bits.28 & mask != 0
case 29: return set.fds_bits.29 & mask != 0
case 30: return set.fds_bits.30 & mask != 0
case 31: return set.fds_bits.31 & mask != 0
default: return false
}
}
}
#endif