Add DPRINTF
This commit is contained in:
parent
983af5cf3e
commit
48b2604064
|
@ -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_
|
||||
|
|
|
@ -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);
|
||||
|
|
Loading…
Reference in a new issue