passepartoutvpn
/
wireguard-apple
mirror of https://github.com/passepartoutvpn/wireguard-apple.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NetworkExtension: check isTunnelClosed before packet existence

This commit is contained in:
Jason A. Donenfeld 2018-11-01 22:01:52 +01:00
parent 73f4a90dee
commit 2267079367
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
WireGuard/WireGuardNetworkExtension/PacketTunnelProvider.swift
View File

@ -163,9 +163,9 @@ class PacketTunnelProvider: NEPacketTunnelProvider {
guard let buf = buf else { return 0 } guard let buf = buf else { return 0 }
let wgContext = wgCtxPtr.bindMemory(to: WireGuardContext.self, capacity: 1).pointee let wgContext = wgCtxPtr.bindMemory(to: WireGuardContext.self, capacity: 1).pointee
var isTunnelClosed = false var isTunnelClosed = false
guard let packet = wgContext.readPacket(isTunnelClosed: &isTunnelClosed) else { return 0 } let packet = wgContext.readPacket(isTunnelClosed: &isTunnelClosed)
if isTunnelClosed { return -1 } if isTunnelClosed { return -1 }
let packetData = packet.data guard let packetData = packet?.data else { return 0 }
if packetData.count <= len { if packetData.count <= len {
packetData.copyBytes(to: buf, count: packetData.count) packetData.copyBytes(to: buf, count: packetData.count)
return packetData.count return packetData.count