askiiart/tunsafe-clang15
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Add DPRINTF

Browse source
This commit is contained in:
Ludvig Strigeus 2018-12-16 17:07:34 +01:00
parent 983af5cf3e
commit 48b2604064
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tunsafe_types.h
View file

@ -68,4 +68,10 @@ void RERROR(const char *msg, ...);
void RINFO(const char *msg, ...);
void tunsafe_die(const char *msg);
#ifdef _DEBUG
#define DPRINTF RINFO
#else
#define DPRINTF(msg, ...)
#endif
#endif // TINYVPN_TYPES_H_

2
wireguard.cpp
View file

@ -1007,6 +1007,7 @@ WireguardProcessor::PacketResult WireguardProcessor::CheckIncomingHandshakeRateL
WgRateLimit::RateLimitResult rr = dev_.rate_limiter()->CheckRateLimit(GetIpForRateLimit(packet));
if ((overload && rr.is_rate_limited()) || !dev_.CheckCookieMac1(packet)) {
DPRINTF("Rate limited or cookie mac failed!");
stats_.invalid_packets_in++;
stats_.invalid_bytes_in += packet->size;
return kPacketResult_Free;
@ -1014,6 +1015,7 @@ WireguardProcessor::PacketResult WireguardProcessor::CheckIncomingHandshakeRateL
dev_.rate_limiter()->CommitResult(rr);
if (overload && !rr.is_first_ip() && !dev_.CheckCookieMac2(packet)) {
DPRINTF("Responding with cookie message");
dev_.CreateCookieMessage((MessageHandshakeCookie*)packet->data, packet, ((MessageHandshakeInitiation*)packet->data)->sender_key_id);
packet->size = sizeof(MessageHandshakeCookie);
PrepareOutgoingHandshakePacket(NULL, packet);