Skip to content
This repository was archived by the owner on Apr 15, 2025. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add Reverse fix for TCP and xorptrpos overflow fix for UDP
  • Loading branch information
tmthecoder committed May 25, 2022
commit f57cc6cd66680bf8cef0b4820da9f21a4cad549f
17 changes: 12 additions & 5 deletions Sources/CTunnelKitOpenVPNProtocol/PacketStream.m
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ @implementation PacketStream

+ (void)xormask:(uint8_t *)dst src:(uint8_t *)src xorMask:(NSData *)xorMask length:(int)length
{
if (((uint8_t *)(xorMask.bytes))[0] != 0) {
if (xorMask.length > 0) {
for (int i = 0; i < length; ++i) {
dst[i] = src[i] ^ ((uint8_t *)(xorMask.bytes))[i % xorMask.length];
}
Expand All @@ -49,12 +49,19 @@ + (void)xorptrpos:(uint8_t *)dst src:(uint8_t *)src length:(int)length

+ (void)reverse:(uint8_t *)dst src:(uint8_t *)src length:(int)length
{
int start = 1;
int end = length - 1;
uint8_t temp = 0;
dst[0] = src[0];
for (int i = 1; i < length/2; ++i) {
temp = dst[length - 1 - i];
dst[length - 1 - i] = src[i];
dst[i] = temp;
while (start < end) {
temp = src[start];
dst[start] = src[end];
dst[end] = temp;
start++;
end--;
}
if (start == end) {
dst[start] = src[start];
}
}

Expand Down
8 changes: 4 additions & 4 deletions Sources/TunnelKitOpenVPNAppExtension/NEUDPLink.swift
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ class NEUDPLink: LinkInterface {
return
}
var packetsToUse: [Data]?
if let packets = packets, [UInt8](self.xorMask)[0] != 0 {
if let packets = packets, self.xorMethod != 0 {
packetsToUse = packets.map { packet in
self.xorPacket(packet: packet, mode: .read)
}
Expand All @@ -85,7 +85,7 @@ class NEUDPLink: LinkInterface {

func writePackets(_ packets: [Data], completionHandler: ((Error?) -> Void)?) {
var packetsToUse: [Data]
if [UInt8](xorMask)[0] != 0 {
if xorMethod != 0 {
packetsToUse = packets.map { packet in
xorPacket(packet: packet, mode: .write)
}
Expand Down Expand Up @@ -119,7 +119,7 @@ class NEUDPLink: LinkInterface {
}

private func xormask(packet: Data) -> Data {
if [UInt8](xorMask)[0] != 0 {
if xorMask.count == 0 {
return packet
}
return Data(packet.enumerated().map { (index, byte) in
Expand All @@ -129,7 +129,7 @@ class NEUDPLink: LinkInterface {

private func xorptrpos(packet: Data) -> Data {
return Data(packet.enumerated().map { (index, byte) in
byte ^ UInt8((index + 1))
byte ^ UInt8(truncatingIfNeeded: index &+ 1)
})
}

Expand Down