ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-08-31 21:34:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1 from 9seconds/master

Browse Source 
Modified antireplay
This commit is contained in:
Savely Krasovsky
2019-08-29 10:33:56 +03:00
committed by GitHub
parent 2748fbd1e5 82ec3871a4
commit 389f6fd4bf
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client/direct.go
+4 -2
View File
@@ -45,10 +45,12 @@ func DirectInit(ctx context.Context, cancel context.CancelFunc, socket net.Conn,
return nil, nil, errors.Annotate(err, "Cannot parse obfuscated frame") return nil, nil, errors.Annotate(err, "Cannot parse obfuscated frame")
} }
if antiReplayCache.Has([]byte(frame)) { var replayPart = []byte(frame)
if antiReplayCache.Has(replayPart[4:60]) {
return nil, nil, errors.New("Replay attack is detected") return nil, nil, errors.New("Replay attack is detected")
} }
antiReplayCache.Add([]byte(frame)) antiReplayCache.Add(replayPart[4:60])
connOpts.ConnectionProto = mtproto.ConnectionProtocolAny connOpts.ConnectionProto = mtproto.ConnectionProtocolAny
connOpts.ClientAddr = conn.RemoteAddr() connOpts.ClientAddr = conn.RemoteAddr()