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

Select correct IP version identifier based on packet contents.

This commit is contained in:
Jeroen Leenarts 2018-08-27 13:52:02 +02:00
parent acb7caa1f8
commit b730dcf6c0
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
WireGuardNetworkExtension/WireGuardGoWrapper.m
View File

@ -171,9 +171,19 @@ static ssize_t do_write(const void *ctx, const unsigned char *buf, size_t len)
os_log_debug([WireGuardGoWrapper log], "do_write - start");
WireGuardGoWrapper *wrapper = (__bridge WireGuardGoWrapper *)ctx;
//TODO: determine IPv4 or IPv6 status.
//determine IPv4 or IPv6 status.
NSInteger ipVersionBits = (buf[0] & 0xf0) >> 4;
NSNumber *ipVersion = NSDecimalNumber.zero;
if (ipVersionBits == 4) {
ipVersion = @AF_INET;
} else if (ipVersionBits == 6) {
ipVersion = @AF_INET6;
} else {
return 0;
}
NSData *packet = [[NSData alloc] initWithBytes:buf length:len];
[wrapper.packetFlow writePackets:@[packet] withProtocols:@[@AF_INET]];
[wrapper.packetFlow writePackets:@[packet] withProtocols:@[ipVersion]];
return len;
}
}