修改语法层面的错误
This commit is contained in:
parent
21b8585d3c
commit
715fa6f491
@ -12,7 +12,6 @@ actor SDLPuncherActor {
|
||||
private var coolingDown: Set<Data> = []
|
||||
private let cooldown: Duration = .seconds(5)
|
||||
|
||||
private var superClientActor: SDLSuperClientActor?
|
||||
private var udpHoleActor: SDLUDPHoleActor?
|
||||
|
||||
// 处理holer
|
||||
@ -28,10 +27,6 @@ actor SDLPuncherActor {
|
||||
self.logger = logger
|
||||
}
|
||||
|
||||
func setSuperClientActor(superClientActor: SDLSuperClientActor?) {
|
||||
self.superClientActor = superClientActor
|
||||
}
|
||||
|
||||
func setUDPHoleActor(udpHoleActor: SDLUDPHoleActor?) {
|
||||
self.udpHoleActor = udpHoleActor
|
||||
}
|
||||
@ -59,31 +54,31 @@ actor SDLPuncherActor {
|
||||
}
|
||||
|
||||
private func tryHole(request: RegisterRequest) async {
|
||||
var queryInfo = SDLQueryInfo()
|
||||
queryInfo.dstMac = request.dstMac
|
||||
guard let message = try? await self.superClientActor?.request(type: .queryInfo, data: try queryInfo.serializedData()) else {
|
||||
return
|
||||
}
|
||||
|
||||
switch message.packet {
|
||||
case .empty:
|
||||
self.logger.log("[SDLContext] hole query_info get empty: \(message)", level: .debug)
|
||||
case .peerInfo(let peerInfo):
|
||||
if let remoteAddress = peerInfo.v4Info.socketAddress() {
|
||||
self.logger.log("[SDLContext] hole sock address: \(remoteAddress)", level: .debug)
|
||||
// 发送register包
|
||||
var register = SDLRegister()
|
||||
register.networkID = request.networkId
|
||||
register.srcMac = request.srcMac
|
||||
register.dstMac = request.dstMac
|
||||
|
||||
await self.udpHoleActor?.send(type: .register, data: try! register.serializedData(), remoteAddress: remoteAddress)
|
||||
} else {
|
||||
self.logger.log("[SDLContext] hole sock address is invalid: \(peerInfo.v4Info)", level: .warning)
|
||||
}
|
||||
default:
|
||||
self.logger.log("[SDLContext] hole query_info is packet: \(message)", level: .warning)
|
||||
}
|
||||
// var queryInfo = SDLQueryInfo()
|
||||
// queryInfo.dstMac = request.dstMac
|
||||
// guard let message = try? await self.superClientActor?.request(type: .queryInfo, data: try queryInfo.serializedData()) else {
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// switch message.packet {
|
||||
// case .empty:
|
||||
// self.logger.log("[SDLContext] hole query_info get empty: \(message)", level: .debug)
|
||||
// case .peerInfo(let peerInfo):
|
||||
// if let remoteAddress = peerInfo.v4Info.socketAddress() {
|
||||
// self.logger.log("[SDLContext] hole sock address: \(remoteAddress)", level: .debug)
|
||||
// // 发送register包
|
||||
// var register = SDLRegister()
|
||||
// register.networkID = request.networkId
|
||||
// register.srcMac = request.srcMac
|
||||
// register.dstMac = request.dstMac
|
||||
//
|
||||
// await self.udpHoleActor?.send(type: .register, data: try! register.serializedData(), remoteAddress: remoteAddress)
|
||||
// } else {
|
||||
// self.logger.log("[SDLContext] hole sock address is invalid: \(peerInfo.v4Info)", level: .warning)
|
||||
// }
|
||||
// default:
|
||||
// self.logger.log("[SDLContext] hole query_info is packet: \(message)", level: .warning)
|
||||
// }
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@ -44,8 +44,8 @@ actor SDLTunnelProviderActor {
|
||||
}
|
||||
|
||||
// 网络改变时需要重新配置网络信息
|
||||
func setNetworkSettings(devAddr: SDLDevAddr, dnsServer: String) async throws -> String {
|
||||
let netAddress = SDLNetAddress(ip: devAddr.netAddr, maskLen: UInt8(devAddr.netBitLen))
|
||||
func setNetworkSettings(networkAddress: SDLConfiguration.NetworkAddress, dnsServer: String) async throws -> String {
|
||||
let netAddress = SDLNetAddress(ip: networkAddress.ip, maskLen: networkAddress.maskLen)
|
||||
let routes = [
|
||||
Route(dstAddress: netAddress.networkAddress, subnetMask: netAddress.maskAddress),
|
||||
Route(dstAddress: dnsServer, subnetMask: "255.255.255.255")
|
||||
@ -57,7 +57,7 @@ actor SDLTunnelProviderActor {
|
||||
|
||||
// 设置网卡的DNS解析
|
||||
|
||||
let networkDomain = devAddr.networkDomain
|
||||
let networkDomain = networkAddress.networkDomain
|
||||
let dnsSettings = NEDNSSettings(servers: [dnsServer])
|
||||
dnsSettings.searchDomains = [networkDomain]
|
||||
dnsSettings.matchDomains = [networkDomain]
|
||||
|
||||
@ -20,6 +20,14 @@ public class SDLConfiguration {
|
||||
}
|
||||
}
|
||||
|
||||
public struct NetworkAddress {
|
||||
public let networkId: UInt32
|
||||
public let ip: UInt32
|
||||
public let maskLen: UInt8
|
||||
public let mac: Data
|
||||
public let networkDomain: String
|
||||
}
|
||||
|
||||
// 当前的客户端版本
|
||||
let version: UInt8
|
||||
|
||||
@ -36,6 +44,8 @@ public class SDLConfiguration {
|
||||
|
||||
let noticePort: Int
|
||||
|
||||
let networkAddress: NetworkAddress = .init(networkId: 8, ip: 0, maskLen: 24, mac: "xyz".data(using: .utf8)!, networkDomain: "punchnet")
|
||||
|
||||
lazy var stunSocketAddress: SocketAddress = {
|
||||
let stunServer = stunServers[0]
|
||||
return try! SocketAddress.makeAddressResolvingHost(stunServer.host, port: stunServer.ports[0])
|
||||
@ -69,4 +79,30 @@ public class SDLConfiguration {
|
||||
self.hostname = hostname
|
||||
}
|
||||
|
||||
// 获取mac地址
|
||||
public static func getMacAddress() -> Data {
|
||||
let key = "gMacAddress2"
|
||||
|
||||
let userDefaults = UserDefaults.standard
|
||||
if let mac = userDefaults.value(forKey: key) as? Data {
|
||||
return mac
|
||||
}
|
||||
else {
|
||||
let mac = generateMacAddress()
|
||||
userDefaults.setValue(mac, forKey: key)
|
||||
|
||||
return mac
|
||||
}
|
||||
}
|
||||
|
||||
// 随机生成mac地址
|
||||
private static func generateMacAddress() -> Data {
|
||||
var macAddress = [UInt8](repeating: 0, count: 6)
|
||||
for i in 0..<6 {
|
||||
macAddress[i] = UInt8.random(in: 0...255)
|
||||
}
|
||||
|
||||
return Data(macAddress)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@ -30,9 +30,6 @@ public class SDLContext {
|
||||
|
||||
let config: SDLConfiguration
|
||||
|
||||
// tun网络地址信息
|
||||
var devAddr: SDLDevAddr
|
||||
|
||||
// nat映射的相关信息, 暂时没有用处
|
||||
//var natAddress: SDLNatAddress?
|
||||
// nat的网络类型
|
||||
@ -83,11 +80,6 @@ public class SDLContext {
|
||||
self.rsaCipher = rsaCipher
|
||||
self.aesCipher = aesCipher
|
||||
|
||||
// 生成mac地址
|
||||
var devAddr = SDLDevAddr()
|
||||
devAddr.mac = Self.getMacAddress()
|
||||
self.devAddr = devAddr
|
||||
|
||||
self.sessionManager = SessionManager()
|
||||
self.arpServer = ArpServer(known_macs: [:])
|
||||
self.providerActor = SDLTunnelProviderActor(provider: provider, logger: logger)
|
||||
@ -118,19 +110,7 @@ public class SDLContext {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
group.addTask {
|
||||
while !Task.isCancelled {
|
||||
do {
|
||||
try await self.startSuperClient()
|
||||
} catch let err {
|
||||
self.logger.log("[SDLContext] SuperClient get error: \(err), will restart", level: .warning)
|
||||
await self.arpServer.clear()
|
||||
try await Task.sleep(for: .seconds(2))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
group.addTask {
|
||||
await self.startMonitor()
|
||||
}
|
||||
@ -180,28 +160,27 @@ public class SDLContext {
|
||||
try await Task.sleep(nanoseconds: 5 * 1_000_000_000)
|
||||
try Task.checkCancellation()
|
||||
|
||||
if let udpHoleActor = self.udpHoleActor {
|
||||
let cookie = await udpHoleActor.getCookieId()
|
||||
var stunRequest = SDLStunRequest()
|
||||
stunRequest.cookie = cookie
|
||||
stunRequest.clientID = self.config.clientId
|
||||
stunRequest.networkID = self.devAddr.networkID
|
||||
stunRequest.ip = self.devAddr.netAddr
|
||||
stunRequest.mac = self.devAddr.mac
|
||||
stunRequest.natType = UInt32(self.natType.rawValue)
|
||||
|
||||
let remoteAddress = self.config.stunSocketAddress
|
||||
await udpHoleActor.send(type: .stunRequest, data: try stunRequest.serializedData(), remoteAddress: remoteAddress)
|
||||
self.lastCookie = cookie
|
||||
}
|
||||
// TODO
|
||||
// if let udpHoleActor = self.udpHoleActor {
|
||||
// let cookie = await udpHoleActor.getCookieId()
|
||||
// var stunRequest = SDLStunRequest()
|
||||
// stunRequest.clientID = self.config.clientId
|
||||
// stunRequest.networkID = self.devAddr.networkID
|
||||
// stunRequest.ip = self.devAddr.netAddr
|
||||
// stunRequest.mac = self.devAddr.mac
|
||||
// stunRequest.natType = UInt32(self.natType.rawValue)
|
||||
//
|
||||
// let remoteAddress = self.config.stunSocketAddress
|
||||
// await udpHoleActor.send(type: .stunRequest, data: try stunRequest.serializedData(), remoteAddress: remoteAddress)
|
||||
// self.lastCookie = cookie
|
||||
// }
|
||||
}
|
||||
}
|
||||
|
||||
group.addTask {
|
||||
if let eventFlow = self.udpHoleActor?.eventFlow {
|
||||
for try await event in eventFlow {
|
||||
if let messageStream = self.udpHoleActor?.messageStream {
|
||||
for try await (remoteAddress, message) in messageStream {
|
||||
try Task.checkCancellation()
|
||||
try await self.handleUDPEvent(event: event)
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -257,225 +236,183 @@ public class SDLContext {
|
||||
}
|
||||
}
|
||||
|
||||
private func handleSuperEvent(event: SDLSuperClientActor.SuperEvent) async throws {
|
||||
switch event {
|
||||
case .ready:
|
||||
await self.puncherActor.setSuperClientActor(superClientActor: self.superClientActor)
|
||||
// private func handleMessage(remoteAddress: SocketAddress, message: SDLHoleInboundMessage) async throws {
|
||||
// switch message {
|
||||
//// case .ready:
|
||||
//// await self.puncherActor.setSuperClientActor(superClientActor: self.superClientActor)
|
||||
////
|
||||
//// self.logger.log("[SDLContext] get registerSuper, mac address: \(SDLUtil.formatMacAddress(mac: self.devAddr.mac))", level: .debug)
|
||||
//// var registerSuper = SDLRegisterSuper()
|
||||
//// registerSuper.version = UInt32(self.config.version)
|
||||
//// registerSuper.clientID = self.config.clientId
|
||||
//// registerSuper.devAddr = self.devAddr
|
||||
//// registerSuper.pubKey = self.rsaCipher.pubKey
|
||||
//// registerSuper.token = self.config.token
|
||||
//// registerSuper.networkCode = self.config.networkCode
|
||||
//// registerSuper.hostname = self.config.hostname
|
||||
//// guard let message = try await self.superClientActor?.request(type: .registerSuper, data: try registerSuper.serializedData()) else {
|
||||
//// return
|
||||
//// }
|
||||
///
|
||||
private func handleRegisterSuperAck(registerSuperAck: SDLRegisterSuperAck) async {
|
||||
// 需要对数据通过rsa的私钥解码
|
||||
let aesKey = try! self.rsaCipher.decode(data: Data(registerSuperAck.aesKey))
|
||||
|
||||
self.logger.log("[SDLContext] get registerSuperAck, aes_key len: \(aesKey.count)", level: .info)
|
||||
// 服务器分配的tun网卡信息
|
||||
do {
|
||||
let ipAddress = try await self.providerActor.setNetworkSettings(networkAddress: self.config.networkAddress, dnsServer: SDLDNSClientActor.Helper.dnsServer)
|
||||
await self.noticeClient?.send(data: NoticeMessage.ipAdress(ip: ipAddress))
|
||||
|
||||
self.logger.log("[SDLContext] get registerSuper, mac address: \(SDLUtil.formatMacAddress(mac: self.devAddr.mac))", level: .debug)
|
||||
var registerSuper = SDLRegisterSuper()
|
||||
registerSuper.version = UInt32(self.config.version)
|
||||
registerSuper.clientID = self.config.clientId
|
||||
registerSuper.devAddr = self.devAddr
|
||||
registerSuper.pubKey = self.rsaCipher.pubKey
|
||||
registerSuper.token = self.config.token
|
||||
registerSuper.networkCode = self.config.networkCode
|
||||
registerSuper.hostname = self.config.hostname
|
||||
guard let message = try await self.superClientActor?.request(type: .registerSuper, data: try registerSuper.serializedData()) else {
|
||||
return
|
||||
}
|
||||
|
||||
switch message.packet {
|
||||
case .registerSuperAck(let registerSuperAck):
|
||||
// 需要对数据通过rsa的私钥解码
|
||||
let aesKey = try! self.rsaCipher.decode(data: Data(registerSuperAck.aesKey))
|
||||
let upgradeType = SDLUpgradeType(rawValue: registerSuperAck.upgradeType)
|
||||
|
||||
self.logger.log("[SDLContext] get registerSuperAck, aes_key len: \(aesKey.count), network_id:\(registerSuperAck.devAddr.networkID)", level: .info)
|
||||
self.devAddr = registerSuperAck.devAddr
|
||||
|
||||
if upgradeType == .force {
|
||||
let forceUpgrade = NoticeMessage.upgrade(prompt: registerSuperAck.upgradePrompt, address: registerSuperAck.upgradeAddress)
|
||||
await self.noticeClient?.send(data: forceUpgrade)
|
||||
exit(-1)
|
||||
}
|
||||
|
||||
// 服务器分配的tun网卡信息
|
||||
do {
|
||||
let ipAddress = try await self.providerActor.setNetworkSettings(devAddr: self.devAddr, dnsServer: SDLDNSClientActor.Helper.dnsServer)
|
||||
await self.noticeClient?.send(data: NoticeMessage.ipAdress(ip: ipAddress))
|
||||
|
||||
self.startReader()
|
||||
} catch let err {
|
||||
self.logger.log("[SDLContext] setTunnelNetworkSettings get error: \(err)", level: .error)
|
||||
exit(-1)
|
||||
}
|
||||
|
||||
self.aesKey = aesKey
|
||||
if upgradeType == .normal {
|
||||
let normalUpgrade = NoticeMessage.upgrade(prompt: registerSuperAck.upgradePrompt, address: registerSuperAck.upgradeAddress)
|
||||
await self.noticeClient?.send(data: normalUpgrade)
|
||||
}
|
||||
|
||||
case .registerSuperNak(let nakPacket):
|
||||
let errorMessage = nakPacket.errorMessage
|
||||
guard let errorCode = SDLNAKErrorCode(rawValue: UInt8(nakPacket.errorCode)) else {
|
||||
return
|
||||
}
|
||||
|
||||
switch errorCode {
|
||||
case .invalidToken, .nodeDisabled:
|
||||
let alertNotice = NoticeMessage.alert(alert: errorMessage)
|
||||
await self.noticeClient?.send(data: alertNotice)
|
||||
exit(-1)
|
||||
case .noIpAddress, .networkFault, .internalFault:
|
||||
let alertNotice = NoticeMessage.alert(alert: errorMessage)
|
||||
await self.noticeClient?.send(data: alertNotice)
|
||||
}
|
||||
self.logger.log("[SDLContext] Get a SuperNak message exit", level: .warning)
|
||||
default:
|
||||
()
|
||||
}
|
||||
|
||||
case .event(let evt):
|
||||
switch evt {
|
||||
case .natChanged(let natChangedEvent):
|
||||
let dstMac = natChangedEvent.mac
|
||||
self.logger.log("[SDLContext] natChangedEvent, dstMac: \(dstMac)", level: .info)
|
||||
await sessionManager.removeSession(dstMac: dstMac)
|
||||
case .sendRegister(let sendRegisterEvent):
|
||||
self.logger.log("[SDLContext] sendRegisterEvent, ip: \(sendRegisterEvent)", level: .debug)
|
||||
let address = SDLUtil.int32ToIp(sendRegisterEvent.natIp)
|
||||
if let remoteAddress = try? SocketAddress.makeAddressResolvingHost(address, port: Int(sendRegisterEvent.natPort)) {
|
||||
// 发送register包
|
||||
var register = SDLRegister()
|
||||
register.networkID = self.devAddr.networkID
|
||||
register.srcMac = self.devAddr.mac
|
||||
register.dstMac = sendRegisterEvent.dstMac
|
||||
await self.udpHoleActor?.send(type: .register, data: try register.serializedData(), remoteAddress: remoteAddress)
|
||||
}
|
||||
|
||||
case .networkShutdown(let shutdownEvent):
|
||||
let alertNotice = NoticeMessage.alert(alert: shutdownEvent.message)
|
||||
await self.noticeClient?.send(data: alertNotice)
|
||||
exit(-1)
|
||||
}
|
||||
case .command(let packetId, let command):
|
||||
switch command {
|
||||
case .changeNetwork(let changeNetworkCommand):
|
||||
// 需要对数据通过rsa的私钥解码
|
||||
let aesKey = try! self.rsaCipher.decode(data: Data(changeNetworkCommand.aesKey))
|
||||
self.logger.log("[SDLContext] change network command get aes_key len: \(aesKey.count)", level: .info)
|
||||
self.devAddr = changeNetworkCommand.devAddr
|
||||
|
||||
// 服务器分配的tun网卡信息
|
||||
do {
|
||||
let ipAddress = try await self.providerActor.setNetworkSettings(devAddr: self.devAddr, dnsServer: SDLDNSClientActor.Helper.dnsServer)
|
||||
await self.noticeClient?.send(data: NoticeMessage.ipAdress(ip: ipAddress))
|
||||
|
||||
self.startReader()
|
||||
} catch let err {
|
||||
self.logger.log("[SDLContext] setTunnelNetworkSettings get error: \(err)", level: .error)
|
||||
exit(-1)
|
||||
}
|
||||
|
||||
self.aesKey = aesKey
|
||||
|
||||
var commandAck = SDLCommandAck()
|
||||
commandAck.status = true
|
||||
await self.superClientActor?.send(type: .commandAck, packetId: packetId, data: try commandAck.serializedData())
|
||||
}
|
||||
self.startReader()
|
||||
} catch let err {
|
||||
self.logger.log("[SDLContext] setTunnelNetworkSettings get error: \(err)", level: .error)
|
||||
exit(-1)
|
||||
}
|
||||
|
||||
self.aesKey = aesKey
|
||||
}
|
||||
|
||||
private func handleUDPEvent(event: SDLUDPHoleActor.UDPEvent) async throws {
|
||||
switch event {
|
||||
case .ready:
|
||||
await self.puncherActor.setUDPHoleActor(udpHoleActor: self.udpHoleActor)
|
||||
// 获取当前网络的类型
|
||||
self.natType = await getNatType()
|
||||
self.logger.log("[SDLContext] broadcast is: \(self.natType)", level: .debug)
|
||||
|
||||
case .message(let remoteAddress, let message):
|
||||
switch message {
|
||||
case .register(let register):
|
||||
self.logger.log("register packet: \(register), dev_addr: \(self.devAddr)", level: .debug)
|
||||
// 判断目标地址是否是tun的网卡地址, 并且是在同一个网络下
|
||||
if register.dstMac == self.devAddr.mac && register.networkID == self.devAddr.networkID {
|
||||
// 回复ack包
|
||||
var registerAck = SDLRegisterAck()
|
||||
registerAck.networkID = self.devAddr.networkID
|
||||
registerAck.srcMac = self.devAddr.mac
|
||||
registerAck.dstMac = register.srcMac
|
||||
|
||||
await self.udpHoleActor?.send(type: .registerAck, data: try registerAck.serializedData(), remoteAddress: remoteAddress)
|
||||
// 这里需要建立到来源的会话, 在复杂网络下，通过super-node查询到的nat地址不一定靠谱，需要通过udp包的来源地址作为nat地址
|
||||
let session = Session(dstMac: register.srcMac, natAddress: remoteAddress)
|
||||
await self.sessionManager.addSession(session: session)
|
||||
} else {
|
||||
self.logger.log("SDLContext didReadRegister get a invalid packet, because dst_ip not matched: \(register.dstMac)", level: .warning)
|
||||
}
|
||||
case .registerAck(let registerAck):
|
||||
// 判断目标地址是否是tun的网卡地址, 并且是在同一个网络下
|
||||
if registerAck.dstMac == self.devAddr.mac && registerAck.networkID == self.devAddr.networkID {
|
||||
let session = Session(dstMac: registerAck.srcMac, natAddress: remoteAddress)
|
||||
await self.sessionManager.addSession(session: session)
|
||||
} else {
|
||||
self.logger.log("SDLContext didReadRegisterAck get a invalid packet, because dst_mac not matched: \(registerAck.dstMac)", level: .warning)
|
||||
}
|
||||
case .stunReply(let stunReply):
|
||||
let cookie = stunReply.cookie
|
||||
if cookie == self.lastCookie {
|
||||
// 记录下当前在nat上的映射信息，暂时没有用；后续会用来判断网络类型
|
||||
//self.natAddress = stunReply.natAddress
|
||||
self.logger.log("[SDLContext] get a stunReply: \(try! stunReply.jsonString())", level: .debug)
|
||||
}
|
||||
default:
|
||||
()
|
||||
}
|
||||
|
||||
case .data(let data):
|
||||
let mac = LayerPacket.MacAddress(data: data.dstMac)
|
||||
guard (data.dstMac == self.devAddr.mac || mac.isBroadcast() || mac.isMulticast()) else {
|
||||
return
|
||||
}
|
||||
|
||||
guard let decyptedData = try? self.aesCipher.decypt(aesKey: self.aesKey, data: Data(data.data)) else {
|
||||
return
|
||||
}
|
||||
|
||||
do {
|
||||
let layerPacket = try LayerPacket(layerData: decyptedData)
|
||||
|
||||
await self.flowTracer.inc(num: decyptedData.count, type: .inbound)
|
||||
// 处理arp请求
|
||||
switch layerPacket.type {
|
||||
case .arp:
|
||||
// 判断如果收到的是arp请求
|
||||
if let arpPacket = ARPPacket(data: layerPacket.data) {
|
||||
if arpPacket.targetIP == self.devAddr.netAddr {
|
||||
switch arpPacket.opcode {
|
||||
case .request:
|
||||
self.logger.log("[SDLContext] get arp request packet", level: .debug)
|
||||
let response = ARPPacket.arpResponse(for: arpPacket, mac: self.devAddr.mac, ip: self.devAddr.netAddr)
|
||||
await self.routeLayerPacket(dstMac: arpPacket.senderMAC, type: .arp, data: response.marshal())
|
||||
case .response:
|
||||
self.logger.log("[SDLContext] get arp response packet", level: .debug)
|
||||
await self.arpServer.append(ip: arpPacket.senderIP, mac: arpPacket.senderMAC)
|
||||
}
|
||||
} else {
|
||||
self.logger.log("[SDLContext] get invalid arp packet: \(arpPacket), target_ip: \(SDLUtil.int32ToIp(arpPacket.targetIP)), net ip: \(SDLUtil.int32ToIp(self.devAddr.netAddr))", level: .debug)
|
||||
}
|
||||
} else {
|
||||
self.logger.log("[SDLContext] get invalid arp packet", level: .debug)
|
||||
}
|
||||
case .ipv4:
|
||||
guard let ipPacket = IPPacket(layerPacket.data), ipPacket.header.destination == self.devAddr.netAddr else {
|
||||
return
|
||||
}
|
||||
let packet = NEPacket(data: ipPacket.data, protocolFamily: 2)
|
||||
await self.providerActor.writePackets(packets: [packet])
|
||||
default:
|
||||
self.logger.log("[SDLContext] get invalid packet", level: .debug)
|
||||
}
|
||||
} catch let err {
|
||||
self.logger.log("[SDLContext] didReadData err: \(err)", level: .warning)
|
||||
}
|
||||
private func handleRegisterSuperNak(nakPacket: SDLRegisterSuperNak) async {
|
||||
let errorMessage = nakPacket.errorMessage
|
||||
guard let errorCode = SDLNAKErrorCode(rawValue: UInt8(nakPacket.errorCode)) else {
|
||||
return
|
||||
}
|
||||
|
||||
switch errorCode {
|
||||
case .invalidToken, .nodeDisabled:
|
||||
let alertNotice = NoticeMessage.alert(alert: errorMessage)
|
||||
await self.noticeClient?.send(data: alertNotice)
|
||||
exit(-1)
|
||||
case .noIpAddress, .networkFault, .internalFault:
|
||||
let alertNotice = NoticeMessage.alert(alert: errorMessage)
|
||||
await self.noticeClient?.send(data: alertNotice)
|
||||
}
|
||||
self.logger.log("[SDLContext] Get a SuperNak message exit", level: .warning)
|
||||
|
||||
}
|
||||
|
||||
private func handleEvent(event: SDLEvent) async throws {
|
||||
switch event {
|
||||
case .natChanged(let natChangedEvent):
|
||||
let dstMac = natChangedEvent.mac
|
||||
self.logger.log("[SDLContext] natChangedEvent, dstMac: \(dstMac)", level: .info)
|
||||
await sessionManager.removeSession(dstMac: dstMac)
|
||||
case .sendRegister(let sendRegisterEvent):
|
||||
self.logger.log("[SDLContext] sendRegisterEvent, ip: \(sendRegisterEvent)", level: .debug)
|
||||
let address = SDLUtil.int32ToIp(sendRegisterEvent.natIp)
|
||||
if let remoteAddress = try? SocketAddress.makeAddressResolvingHost(address, port: Int(sendRegisterEvent.natPort)) {
|
||||
// 发送register包
|
||||
var register = SDLRegister()
|
||||
register.networkID = self.config.networkAddress.networkId
|
||||
register.srcMac = self.config.networkAddress.mac
|
||||
register.dstMac = sendRegisterEvent.dstMac
|
||||
await self.udpHoleActor?.send(type: .register, data: try register.serializedData(), remoteAddress: remoteAddress)
|
||||
}
|
||||
|
||||
case .networkShutdown(let shutdownEvent):
|
||||
let alertNotice = NoticeMessage.alert(alert: shutdownEvent.message)
|
||||
await self.noticeClient?.send(data: alertNotice)
|
||||
exit(-1)
|
||||
}
|
||||
}
|
||||
|
||||
//// case .ready:
|
||||
//// await self.puncherActor.setUDPHoleActor(udpHoleActor: self.udpHoleActor)
|
||||
//// // 获取当前网络的类型
|
||||
//// self.natType = await getNatType()
|
||||
//// self.logger.log("[SDLContext] broadcast is: \(self.natType)", level: .debug)
|
||||
//
|
||||
|
||||
private func handleRegister(remoteAddress: SocketAddress, register: SDLRegister) async throws {
|
||||
let networkAddr = config.networkAddress
|
||||
self.logger.log("register packet: \(register), network_address: \(networkAddr)", level: .debug)
|
||||
|
||||
// 判断目标地址是否是tun的网卡地址, 并且是在同一个网络下
|
||||
if register.dstMac == networkAddr.mac && register.networkID == networkAddr.networkId {
|
||||
// 回复ack包
|
||||
var registerAck = SDLRegisterAck()
|
||||
registerAck.networkID = networkAddr.networkId
|
||||
registerAck.srcMac = networkAddr.mac
|
||||
registerAck.dstMac = register.srcMac
|
||||
|
||||
await self.udpHoleActor?.send(type: .registerAck, data: try registerAck.serializedData(), remoteAddress: remoteAddress)
|
||||
// 这里需要建立到来源的会话, 在复杂网络下，通过super-node查询到的nat地址不一定靠谱，需要通过udp包的来源地址作为nat地址
|
||||
let session = Session(dstMac: register.srcMac, natAddress: remoteAddress)
|
||||
await self.sessionManager.addSession(session: session)
|
||||
} else {
|
||||
self.logger.log("SDLContext didReadRegister get a invalid packet, because dst_ip not matched: \(register.dstMac)", level: .warning)
|
||||
}
|
||||
}
|
||||
|
||||
private func handleRegisterAck(remoteAddress: SocketAddress, registerAck: SDLRegister) async {
|
||||
// 判断目标地址是否是tun的网卡地址, 并且是在同一个网络下
|
||||
let networkAddr = config.networkAddress
|
||||
if registerAck.dstMac == networkAddr.mac && registerAck.networkID == networkAddr.networkId {
|
||||
let session = Session(dstMac: registerAck.srcMac, natAddress: remoteAddress)
|
||||
await self.sessionManager.addSession(session: session)
|
||||
} else {
|
||||
self.logger.log("SDLContext didReadRegisterAck get a invalid packet, because dst_mac not matched: \(registerAck.dstMac)", level: .warning)
|
||||
}
|
||||
}
|
||||
|
||||
private func handleStunReply(stunReply: SDLStunReply) async {
|
||||
// let cookie = stunReply.cookie
|
||||
// if cookie == self.lastCookie {
|
||||
// // 记录下当前在nat上的映射信息，暂时没有用；后续会用来判断网络类型
|
||||
// //self.natAddress = stunReply.natAddress
|
||||
// self.logger.log("[SDLContext] get a stunReply: \(try! stunReply.jsonString())", level: .debug)
|
||||
// }
|
||||
}
|
||||
|
||||
private func handleData(data: SDLData) async throws {
|
||||
let mac = LayerPacket.MacAddress(data: data.dstMac)
|
||||
|
||||
let networkAddr = config.networkAddress
|
||||
guard (data.dstMac == networkAddr.mac || mac.isBroadcast() || mac.isMulticast()) else {
|
||||
return
|
||||
}
|
||||
|
||||
guard let decyptedData = try? self.aesCipher.decypt(aesKey: self.aesKey, data: Data(data.data)) else {
|
||||
return
|
||||
}
|
||||
|
||||
let layerPacket = try LayerPacket(layerData: decyptedData)
|
||||
|
||||
await self.flowTracer.inc(num: decyptedData.count, type: .inbound)
|
||||
// 处理arp请求
|
||||
switch layerPacket.type {
|
||||
case .arp:
|
||||
// 判断如果收到的是arp请求
|
||||
if let arpPacket = ARPPacket(data: layerPacket.data) {
|
||||
if arpPacket.targetIP == networkAddr.ip {
|
||||
switch arpPacket.opcode {
|
||||
case .request:
|
||||
self.logger.log("[SDLContext] get arp request packet", level: .debug)
|
||||
let response = ARPPacket.arpResponse(for: arpPacket, mac: networkAddr.mac, ip: networkAddr.ip)
|
||||
await self.routeLayerPacket(dstMac: arpPacket.senderMAC, type: .arp, data: response.marshal())
|
||||
case .response:
|
||||
self.logger.log("[SDLContext] get arp response packet", level: .debug)
|
||||
await self.arpServer.append(ip: arpPacket.senderIP, mac: arpPacket.senderMAC)
|
||||
}
|
||||
} else {
|
||||
self.logger.log("[SDLContext] get invalid arp packet: \(arpPacket), target_ip: \(SDLUtil.int32ToIp(arpPacket.targetIP)), net ip: \(SDLUtil.int32ToIp(networkAddr.ip))", level: .debug)
|
||||
}
|
||||
} else {
|
||||
self.logger.log("[SDLContext] get invalid arp packet", level: .debug)
|
||||
}
|
||||
case .ipv4:
|
||||
guard let ipPacket = IPPacket(layerPacket.data), ipPacket.header.destination == networkAddr.ip else {
|
||||
return
|
||||
}
|
||||
let packet = NEPacket(data: ipPacket.data, protocolFamily: 2)
|
||||
await self.providerActor.writePackets(packets: [packet])
|
||||
default:
|
||||
self.logger.log("[SDLContext] get invalid packet", level: .debug)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// 流量统计
|
||||
// public func flowReportTask() {
|
||||
// Task {
|
||||
@ -512,6 +449,7 @@ public class SDLContext {
|
||||
return
|
||||
}
|
||||
|
||||
let networkAddr = self.config.networkAddress
|
||||
if SDLDNSClientActor.Helper.isDnsRequestPacket(ipPacket: packet) {
|
||||
let destIp = packet.header.destination_ip
|
||||
self.logger.log("[DNSQuery] destIp: \(destIp), int: \(packet.header.destination.asIpAddress())", level: .debug)
|
||||
@ -521,7 +459,7 @@ public class SDLContext {
|
||||
Task.detached {
|
||||
let dstIp = packet.header.destination
|
||||
// 本地通讯, 目标地址是本地服务器的ip地址
|
||||
if dstIp == self.devAddr.netAddr {
|
||||
if dstIp == networkAddr.ip {
|
||||
let nePacket = NEPacket(data: packet.data, protocolFamily: 2)
|
||||
await self.providerActor.writePackets(packets: [nePacket])
|
||||
return
|
||||
@ -534,7 +472,7 @@ public class SDLContext {
|
||||
else {
|
||||
self.logger.log("[SDLContext] dstIp: \(dstIp.asIpAddress()) arp query not found, broadcast", level: .debug)
|
||||
// 构造arp广播
|
||||
let arpReqeust = ARPPacket.arpRequest(senderIP: self.devAddr.netAddr, senderMAC: self.devAddr.mac, targetIP: dstIp)
|
||||
let arpReqeust = ARPPacket.arpRequest(senderIP: networkAddr.ip, senderMAC: networkAddr.mac, targetIP: dstIp)
|
||||
await self.routeLayerPacket(dstMac: ARPPacket.broadcastMac , type: .arp, data: arpReqeust.marshal())
|
||||
}
|
||||
}
|
||||
@ -542,16 +480,17 @@ public class SDLContext {
|
||||
}
|
||||
|
||||
private func routeLayerPacket(dstMac: Data, type: LayerPacket.PacketType, data: Data) async {
|
||||
let networkAddr = self.config.networkAddress
|
||||
// 将数据封装层2层的数据包
|
||||
let layerPacket = LayerPacket(dstMac: dstMac, srcMac: self.devAddr.mac, type: type, data: data)
|
||||
let layerPacket = LayerPacket(dstMac: dstMac, srcMac: networkAddr.mac, type: type, data: data)
|
||||
guard let encodedPacket = try? self.aesCipher.encrypt(aesKey: self.aesKey, data: layerPacket.marshal()) else {
|
||||
return
|
||||
}
|
||||
|
||||
// 构造数据包
|
||||
var dataPacket = SDLData()
|
||||
dataPacket.networkID = self.devAddr.networkID
|
||||
dataPacket.srcMac = self.devAddr.mac
|
||||
dataPacket.networkID = networkAddr.networkId
|
||||
dataPacket.srcMac = networkAddr.mac
|
||||
dataPacket.dstMac = dstMac
|
||||
dataPacket.ttl = 255
|
||||
dataPacket.data = encodedPacket
|
||||
@ -575,7 +514,7 @@ public class SDLContext {
|
||||
// 流量统计
|
||||
await self.flowTracer.inc(num: data.count, type: .forward)
|
||||
// 尝试打洞
|
||||
await self.puncherActor.submitRegisterRequest(request: .init(srcMac: self.devAddr.mac, dstMac: dstMac, networkId: self.devAddr.networkID))
|
||||
await self.puncherActor.submitRegisterRequest(request: .init(srcMac: networkAddr.mac, dstMac: dstMac, networkId: networkAddr.networkId))
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -583,36 +522,8 @@ public class SDLContext {
|
||||
deinit {
|
||||
self.rootTask?.cancel()
|
||||
self.udpHoleActor = nil
|
||||
self.superClientActor = nil
|
||||
self.dnsClientActor = nil
|
||||
}
|
||||
|
||||
// 获取mac地址
|
||||
public static func getMacAddress() -> Data {
|
||||
let key = "gMacAddress2"
|
||||
|
||||
let userDefaults = UserDefaults.standard
|
||||
if let mac = userDefaults.value(forKey: key) as? Data {
|
||||
return mac
|
||||
}
|
||||
else {
|
||||
let mac = generateMacAddress()
|
||||
userDefaults.setValue(mac, forKey: key)
|
||||
|
||||
return mac
|
||||
}
|
||||
}
|
||||
|
||||
// 随机生成mac地址
|
||||
private static func generateMacAddress() -> Data {
|
||||
var macAddress = [UInt8](repeating: 0, count: 6)
|
||||
for i in 0..<6 {
|
||||
macAddress[i] = UInt8.random(in: 0...255)
|
||||
}
|
||||
|
||||
return Data(macAddress)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// 网络类型探测
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user