修改语法层面的错误

This commit is contained in:
anlicheng 2026-01-27 22:36:41 +08:00
parent 21b8585d3c
commit 715fa6f491
4 changed files with 257 additions and 315 deletions

View File

@ -12,7 +12,6 @@ actor SDLPuncherActor {
private var coolingDown: Set<Data> = [] private var coolingDown: Set<Data> = []
private let cooldown: Duration = .seconds(5) private let cooldown: Duration = .seconds(5)
private var superClientActor: SDLSuperClientActor?
private var udpHoleActor: SDLUDPHoleActor? private var udpHoleActor: SDLUDPHoleActor?
// holer // holer
@ -28,10 +27,6 @@ actor SDLPuncherActor {
self.logger = logger self.logger = logger
} }
func setSuperClientActor(superClientActor: SDLSuperClientActor?) {
self.superClientActor = superClientActor
}
func setUDPHoleActor(udpHoleActor: SDLUDPHoleActor?) { func setUDPHoleActor(udpHoleActor: SDLUDPHoleActor?) {
self.udpHoleActor = udpHoleActor self.udpHoleActor = udpHoleActor
} }
@ -59,31 +54,31 @@ actor SDLPuncherActor {
} }
private func tryHole(request: RegisterRequest) async { private func tryHole(request: RegisterRequest) async {
var queryInfo = SDLQueryInfo() // var queryInfo = SDLQueryInfo()
queryInfo.dstMac = request.dstMac // queryInfo.dstMac = request.dstMac
guard let message = try? await self.superClientActor?.request(type: .queryInfo, data: try queryInfo.serializedData()) else { // guard let message = try? await self.superClientActor?.request(type: .queryInfo, data: try queryInfo.serializedData()) else {
return // return
} // }
//
switch message.packet { // switch message.packet {
case .empty: // case .empty:
self.logger.log("[SDLContext] hole query_info get empty: \(message)", level: .debug) // self.logger.log("[SDLContext] hole query_info get empty: \(message)", level: .debug)
case .peerInfo(let peerInfo): // case .peerInfo(let peerInfo):
if let remoteAddress = peerInfo.v4Info.socketAddress() { // if let remoteAddress = peerInfo.v4Info.socketAddress() {
self.logger.log("[SDLContext] hole sock address: \(remoteAddress)", level: .debug) // self.logger.log("[SDLContext] hole sock address: \(remoteAddress)", level: .debug)
// register // // register
var register = SDLRegister() // var register = SDLRegister()
register.networkID = request.networkId // register.networkID = request.networkId
register.srcMac = request.srcMac // register.srcMac = request.srcMac
register.dstMac = request.dstMac // register.dstMac = request.dstMac
//
await self.udpHoleActor?.send(type: .register, data: try! register.serializedData(), remoteAddress: remoteAddress) // await self.udpHoleActor?.send(type: .register, data: try! register.serializedData(), remoteAddress: remoteAddress)
} else { // } else {
self.logger.log("[SDLContext] hole sock address is invalid: \(peerInfo.v4Info)", level: .warning) // self.logger.log("[SDLContext] hole sock address is invalid: \(peerInfo.v4Info)", level: .warning)
} // }
default: // default:
self.logger.log("[SDLContext] hole query_info is packet: \(message)", level: .warning) // self.logger.log("[SDLContext] hole query_info is packet: \(message)", level: .warning)
} // }
} }
} }

View File

@ -44,8 +44,8 @@ actor SDLTunnelProviderActor {
} }
// //
func setNetworkSettings(devAddr: SDLDevAddr, dnsServer: String) async throws -> String { func setNetworkSettings(networkAddress: SDLConfiguration.NetworkAddress, dnsServer: String) async throws -> String {
let netAddress = SDLNetAddress(ip: devAddr.netAddr, maskLen: UInt8(devAddr.netBitLen)) let netAddress = SDLNetAddress(ip: networkAddress.ip, maskLen: networkAddress.maskLen)
let routes = [ let routes = [
Route(dstAddress: netAddress.networkAddress, subnetMask: netAddress.maskAddress), Route(dstAddress: netAddress.networkAddress, subnetMask: netAddress.maskAddress),
Route(dstAddress: dnsServer, subnetMask: "255.255.255.255") Route(dstAddress: dnsServer, subnetMask: "255.255.255.255")
@ -57,7 +57,7 @@ actor SDLTunnelProviderActor {
// DNS // DNS
let networkDomain = devAddr.networkDomain let networkDomain = networkAddress.networkDomain
let dnsSettings = NEDNSSettings(servers: [dnsServer]) let dnsSettings = NEDNSSettings(servers: [dnsServer])
dnsSettings.searchDomains = [networkDomain] dnsSettings.searchDomains = [networkDomain]
dnsSettings.matchDomains = [networkDomain] dnsSettings.matchDomains = [networkDomain]

View File

@ -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 let version: UInt8
@ -36,6 +44,8 @@ public class SDLConfiguration {
let noticePort: Int let noticePort: Int
let networkAddress: NetworkAddress = .init(networkId: 8, ip: 0, maskLen: 24, mac: "xyz".data(using: .utf8)!, networkDomain: "punchnet")
lazy var stunSocketAddress: SocketAddress = { lazy var stunSocketAddress: SocketAddress = {
let stunServer = stunServers[0] let stunServer = stunServers[0]
return try! SocketAddress.makeAddressResolvingHost(stunServer.host, port: stunServer.ports[0]) return try! SocketAddress.makeAddressResolvingHost(stunServer.host, port: stunServer.ports[0])
@ -69,4 +79,30 @@ public class SDLConfiguration {
self.hostname = hostname 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)
}
} }

View File

@ -30,9 +30,6 @@ public class SDLContext {
let config: SDLConfiguration let config: SDLConfiguration
// tun
var devAddr: SDLDevAddr
// nat, // nat,
//var natAddress: SDLNatAddress? //var natAddress: SDLNatAddress?
// nat // nat
@ -83,11 +80,6 @@ public class SDLContext {
self.rsaCipher = rsaCipher self.rsaCipher = rsaCipher
self.aesCipher = aesCipher self.aesCipher = aesCipher
// mac
var devAddr = SDLDevAddr()
devAddr.mac = Self.getMacAddress()
self.devAddr = devAddr
self.sessionManager = SessionManager() self.sessionManager = SessionManager()
self.arpServer = ArpServer(known_macs: [:]) self.arpServer = ArpServer(known_macs: [:])
self.providerActor = SDLTunnelProviderActor(provider: provider, logger: logger) self.providerActor = SDLTunnelProviderActor(provider: provider, logger: logger)
@ -119,18 +111,6 @@ 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 { group.addTask {
await self.startMonitor() await self.startMonitor()
} }
@ -180,28 +160,27 @@ public class SDLContext {
try await Task.sleep(nanoseconds: 5 * 1_000_000_000) try await Task.sleep(nanoseconds: 5 * 1_000_000_000)
try Task.checkCancellation() try Task.checkCancellation()
if let udpHoleActor = self.udpHoleActor { // TODO
let cookie = await udpHoleActor.getCookieId() // if let udpHoleActor = self.udpHoleActor {
var stunRequest = SDLStunRequest() // let cookie = await udpHoleActor.getCookieId()
stunRequest.cookie = cookie // var stunRequest = SDLStunRequest()
stunRequest.clientID = self.config.clientId // stunRequest.clientID = self.config.clientId
stunRequest.networkID = self.devAddr.networkID // stunRequest.networkID = self.devAddr.networkID
stunRequest.ip = self.devAddr.netAddr // stunRequest.ip = self.devAddr.netAddr
stunRequest.mac = self.devAddr.mac // stunRequest.mac = self.devAddr.mac
stunRequest.natType = UInt32(self.natType.rawValue) // stunRequest.natType = UInt32(self.natType.rawValue)
//
let remoteAddress = self.config.stunSocketAddress // let remoteAddress = self.config.stunSocketAddress
await udpHoleActor.send(type: .stunRequest, data: try stunRequest.serializedData(), remoteAddress: remoteAddress) // await udpHoleActor.send(type: .stunRequest, data: try stunRequest.serializedData(), remoteAddress: remoteAddress)
self.lastCookie = cookie // self.lastCookie = cookie
} // }
} }
} }
group.addTask { group.addTask {
if let eventFlow = self.udpHoleActor?.eventFlow { if let messageStream = self.udpHoleActor?.messageStream {
for try await event in eventFlow { for try await (remoteAddress, message) in messageStream {
try Task.checkCancellation() try Task.checkCancellation()
try await self.handleUDPEvent(event: event)
} }
} }
} }
@ -257,42 +236,32 @@ public class SDLContext {
} }
} }
private func handleSuperEvent(event: SDLSuperClientActor.SuperEvent) async throws { // private func handleMessage(remoteAddress: SocketAddress, message: SDLHoleInboundMessage) async throws {
switch event { // switch message {
case .ready: //// case .ready:
await self.puncherActor.setSuperClientActor(superClientActor: self.superClientActor) //// await self.puncherActor.setSuperClientActor(superClientActor: self.superClientActor)
////
self.logger.log("[SDLContext] get registerSuper, mac address: \(SDLUtil.formatMacAddress(mac: self.devAddr.mac))", level: .debug) //// self.logger.log("[SDLContext] get registerSuper, mac address: \(SDLUtil.formatMacAddress(mac: self.devAddr.mac))", level: .debug)
var registerSuper = SDLRegisterSuper() //// var registerSuper = SDLRegisterSuper()
registerSuper.version = UInt32(self.config.version) //// registerSuper.version = UInt32(self.config.version)
registerSuper.clientID = self.config.clientId //// registerSuper.clientID = self.config.clientId
registerSuper.devAddr = self.devAddr //// registerSuper.devAddr = self.devAddr
registerSuper.pubKey = self.rsaCipher.pubKey //// registerSuper.pubKey = self.rsaCipher.pubKey
registerSuper.token = self.config.token //// registerSuper.token = self.config.token
registerSuper.networkCode = self.config.networkCode //// registerSuper.networkCode = self.config.networkCode
registerSuper.hostname = self.config.hostname //// registerSuper.hostname = self.config.hostname
guard let message = try await self.superClientActor?.request(type: .registerSuper, data: try registerSuper.serializedData()) else { //// guard let message = try await self.superClientActor?.request(type: .registerSuper, data: try registerSuper.serializedData()) else {
return //// return
} //// }
///
switch message.packet { private func handleRegisterSuperAck(registerSuperAck: SDLRegisterSuperAck) async {
case .registerSuperAck(let registerSuperAck):
// rsa // rsa
let aesKey = try! self.rsaCipher.decode(data: Data(registerSuperAck.aesKey)) 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)
}
self.logger.log("[SDLContext] get registerSuperAck, aes_key len: \(aesKey.count)", level: .info)
// tun // tun
do { do {
let ipAddress = try await self.providerActor.setNetworkSettings(devAddr: self.devAddr, dnsServer: SDLDNSClientActor.Helper.dnsServer) let ipAddress = try await self.providerActor.setNetworkSettings(networkAddress: self.config.networkAddress, dnsServer: SDLDNSClientActor.Helper.dnsServer)
await self.noticeClient?.send(data: NoticeMessage.ipAdress(ip: ipAddress)) await self.noticeClient?.send(data: NoticeMessage.ipAdress(ip: ipAddress))
self.startReader() self.startReader()
@ -302,12 +271,9 @@ public class SDLContext {
} }
self.aesKey = aesKey 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): private func handleRegisterSuperNak(nakPacket: SDLRegisterSuperNak) async {
let errorMessage = nakPacket.errorMessage let errorMessage = nakPacket.errorMessage
guard let errorCode = SDLNAKErrorCode(rawValue: UInt8(nakPacket.errorCode)) else { guard let errorCode = SDLNAKErrorCode(rawValue: UInt8(nakPacket.errorCode)) else {
return return
@ -323,12 +289,11 @@ public class SDLContext {
await self.noticeClient?.send(data: alertNotice) await self.noticeClient?.send(data: alertNotice)
} }
self.logger.log("[SDLContext] Get a SuperNak message exit", level: .warning) self.logger.log("[SDLContext] Get a SuperNak message exit", level: .warning)
default:
()
} }
case .event(let evt): private func handleEvent(event: SDLEvent) async throws {
switch evt { switch event {
case .natChanged(let natChangedEvent): case .natChanged(let natChangedEvent):
let dstMac = natChangedEvent.mac let dstMac = natChangedEvent.mac
self.logger.log("[SDLContext] natChangedEvent, dstMac: \(dstMac)", level: .info) self.logger.log("[SDLContext] natChangedEvent, dstMac: \(dstMac)", level: .info)
@ -339,8 +304,8 @@ public class SDLContext {
if let remoteAddress = try? SocketAddress.makeAddressResolvingHost(address, port: Int(sendRegisterEvent.natPort)) { if let remoteAddress = try? SocketAddress.makeAddressResolvingHost(address, port: Int(sendRegisterEvent.natPort)) {
// register // register
var register = SDLRegister() var register = SDLRegister()
register.networkID = self.devAddr.networkID register.networkID = self.config.networkAddress.networkId
register.srcMac = self.devAddr.mac register.srcMac = self.config.networkAddress.mac
register.dstMac = sendRegisterEvent.dstMac register.dstMac = sendRegisterEvent.dstMac
await self.udpHoleActor?.send(type: .register, data: try register.serializedData(), remoteAddress: remoteAddress) await self.udpHoleActor?.send(type: .register, data: try register.serializedData(), remoteAddress: remoteAddress)
} }
@ -350,53 +315,25 @@ public class SDLContext {
await self.noticeClient?.send(data: alertNotice) await self.noticeClient?.send(data: alertNotice)
exit(-1) 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 //// case .ready:
//// await self.puncherActor.setUDPHoleActor(udpHoleActor: self.udpHoleActor)
//// //
//// self.natType = await getNatType()
//// self.logger.log("[SDLContext] broadcast is: \(self.natType)", level: .debug)
//
var commandAck = SDLCommandAck() private func handleRegister(remoteAddress: SocketAddress, register: SDLRegister) async throws {
commandAck.status = true let networkAddr = config.networkAddress
await self.superClientActor?.send(type: .commandAck, packetId: packetId, data: try commandAck.serializedData()) self.logger.log("register packet: \(register), network_address: \(networkAddr)", level: .debug)
}
}
}
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, // tun,
if register.dstMac == self.devAddr.mac && register.networkID == self.devAddr.networkID { if register.dstMac == networkAddr.mac && register.networkID == networkAddr.networkId {
// ack // ack
var registerAck = SDLRegisterAck() var registerAck = SDLRegisterAck()
registerAck.networkID = self.devAddr.networkID registerAck.networkID = networkAddr.networkId
registerAck.srcMac = self.devAddr.mac registerAck.srcMac = networkAddr.mac
registerAck.dstMac = register.srcMac registerAck.dstMac = register.srcMac
await self.udpHoleActor?.send(type: .registerAck, data: try registerAck.serializedData(), remoteAddress: remoteAddress) await self.udpHoleActor?.send(type: .registerAck, data: try registerAck.serializedData(), remoteAddress: remoteAddress)
@ -406,28 +343,33 @@ public class SDLContext {
} else { } else {
self.logger.log("SDLContext didReadRegister get a invalid packet, because dst_ip not matched: \(register.dstMac)", level: .warning) self.logger.log("SDLContext didReadRegister get a invalid packet, because dst_ip not matched: \(register.dstMac)", level: .warning)
} }
case .registerAck(let registerAck): }
private func handleRegisterAck(remoteAddress: SocketAddress, registerAck: SDLRegister) async {
// tun, // tun,
if registerAck.dstMac == self.devAddr.mac && registerAck.networkID == self.devAddr.networkID { let networkAddr = config.networkAddress
if registerAck.dstMac == networkAddr.mac && registerAck.networkID == networkAddr.networkId {
let session = Session(dstMac: registerAck.srcMac, natAddress: remoteAddress) let session = Session(dstMac: registerAck.srcMac, natAddress: remoteAddress)
await self.sessionManager.addSession(session: session) await self.sessionManager.addSession(session: session)
} else { } else {
self.logger.log("SDLContext didReadRegisterAck get a invalid packet, because dst_mac not matched: \(registerAck.dstMac)", level: .warning) 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): 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 mac = LayerPacket.MacAddress(data: data.dstMac)
guard (data.dstMac == self.devAddr.mac || mac.isBroadcast() || mac.isMulticast()) else {
let networkAddr = config.networkAddress
guard (data.dstMac == networkAddr.mac || mac.isBroadcast() || mac.isMulticast()) else {
return return
} }
@ -435,7 +377,6 @@ public class SDLContext {
return return
} }
do {
let layerPacket = try LayerPacket(layerData: decyptedData) let layerPacket = try LayerPacket(layerData: decyptedData)
await self.flowTracer.inc(num: decyptedData.count, type: .inbound) await self.flowTracer.inc(num: decyptedData.count, type: .inbound)
@ -444,24 +385,24 @@ public class SDLContext {
case .arp: case .arp:
// arp // arp
if let arpPacket = ARPPacket(data: layerPacket.data) { if let arpPacket = ARPPacket(data: layerPacket.data) {
if arpPacket.targetIP == self.devAddr.netAddr { if arpPacket.targetIP == networkAddr.ip {
switch arpPacket.opcode { switch arpPacket.opcode {
case .request: case .request:
self.logger.log("[SDLContext] get arp request packet", level: .debug) self.logger.log("[SDLContext] get arp request packet", level: .debug)
let response = ARPPacket.arpResponse(for: arpPacket, mac: self.devAddr.mac, ip: self.devAddr.netAddr) let response = ARPPacket.arpResponse(for: arpPacket, mac: networkAddr.mac, ip: networkAddr.ip)
await self.routeLayerPacket(dstMac: arpPacket.senderMAC, type: .arp, data: response.marshal()) await self.routeLayerPacket(dstMac: arpPacket.senderMAC, type: .arp, data: response.marshal())
case .response: case .response:
self.logger.log("[SDLContext] get arp response packet", level: .debug) self.logger.log("[SDLContext] get arp response packet", level: .debug)
await self.arpServer.append(ip: arpPacket.senderIP, mac: arpPacket.senderMAC) await self.arpServer.append(ip: arpPacket.senderIP, mac: arpPacket.senderMAC)
} }
} else { } 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) self.logger.log("[SDLContext] get invalid arp packet: \(arpPacket), target_ip: \(SDLUtil.int32ToIp(arpPacket.targetIP)), net ip: \(SDLUtil.int32ToIp(networkAddr.ip))", level: .debug)
} }
} else { } else {
self.logger.log("[SDLContext] get invalid arp packet", level: .debug) self.logger.log("[SDLContext] get invalid arp packet", level: .debug)
} }
case .ipv4: case .ipv4:
guard let ipPacket = IPPacket(layerPacket.data), ipPacket.header.destination == self.devAddr.netAddr else { guard let ipPacket = IPPacket(layerPacket.data), ipPacket.header.destination == networkAddr.ip else {
return return
} }
let packet = NEPacket(data: ipPacket.data, protocolFamily: 2) let packet = NEPacket(data: ipPacket.data, protocolFamily: 2)
@ -469,12 +410,8 @@ public class SDLContext {
default: default:
self.logger.log("[SDLContext] get invalid packet", level: .debug) self.logger.log("[SDLContext] get invalid packet", level: .debug)
} }
} catch let err {
self.logger.log("[SDLContext] didReadData err: \(err)", level: .warning)
}
} }
}
// //
// public func flowReportTask() { // public func flowReportTask() {
@ -512,6 +449,7 @@ public class SDLContext {
return return
} }
let networkAddr = self.config.networkAddress
if SDLDNSClientActor.Helper.isDnsRequestPacket(ipPacket: packet) { if SDLDNSClientActor.Helper.isDnsRequestPacket(ipPacket: packet) {
let destIp = packet.header.destination_ip let destIp = packet.header.destination_ip
self.logger.log("[DNSQuery] destIp: \(destIp), int: \(packet.header.destination.asIpAddress())", level: .debug) self.logger.log("[DNSQuery] destIp: \(destIp), int: \(packet.header.destination.asIpAddress())", level: .debug)
@ -521,7 +459,7 @@ public class SDLContext {
Task.detached { Task.detached {
let dstIp = packet.header.destination let dstIp = packet.header.destination
// , ip // , ip
if dstIp == self.devAddr.netAddr { if dstIp == networkAddr.ip {
let nePacket = NEPacket(data: packet.data, protocolFamily: 2) let nePacket = NEPacket(data: packet.data, protocolFamily: 2)
await self.providerActor.writePackets(packets: [nePacket]) await self.providerActor.writePackets(packets: [nePacket])
return return
@ -534,7 +472,7 @@ public class SDLContext {
else { else {
self.logger.log("[SDLContext] dstIp: \(dstIp.asIpAddress()) arp query not found, broadcast", level: .debug) self.logger.log("[SDLContext] dstIp: \(dstIp.asIpAddress()) arp query not found, broadcast", level: .debug)
// arp广 // 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()) 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 { private func routeLayerPacket(dstMac: Data, type: LayerPacket.PacketType, data: Data) async {
let networkAddr = self.config.networkAddress
// 2 // 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 { guard let encodedPacket = try? self.aesCipher.encrypt(aesKey: self.aesKey, data: layerPacket.marshal()) else {
return return
} }
// //
var dataPacket = SDLData() var dataPacket = SDLData()
dataPacket.networkID = self.devAddr.networkID dataPacket.networkID = networkAddr.networkId
dataPacket.srcMac = self.devAddr.mac dataPacket.srcMac = networkAddr.mac
dataPacket.dstMac = dstMac dataPacket.dstMac = dstMac
dataPacket.ttl = 255 dataPacket.ttl = 255
dataPacket.data = encodedPacket dataPacket.data = encodedPacket
@ -575,7 +514,7 @@ public class SDLContext {
// //
await self.flowTracer.inc(num: data.count, type: .forward) 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 { deinit {
self.rootTask?.cancel() self.rootTask?.cancel()
self.udpHoleActor = nil self.udpHoleActor = nil
self.superClientActor = nil
self.dnsClientActor = 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)
}
} }
// //