Просмотр исходного кода

Modified antireplay

tags/1.0^2
9seconds 6 лет назад
Родитель
Сommit
82ec3871a4
1 измененных файлов: 4 добавлений и 2 удалений
  1. 4
    2
      client/direct.go

+ 4
- 2
client/direct.go Просмотреть файл

45
 		return nil, nil, errors.Annotate(err, "Cannot parse obfuscated frame")
45
 		return nil, nil, errors.Annotate(err, "Cannot parse obfuscated frame")
46
 	}
46
 	}
47
 
47
 
48
-	if antiReplayCache.Has([]byte(frame)) {
48
+	var replayPart = []byte(frame)
49
+
50
+	if antiReplayCache.Has(replayPart[4:60]) {
49
 		return nil, nil, errors.New("Replay attack is detected")
51
 		return nil, nil, errors.New("Replay attack is detected")
50
 	}
52
 	}
51
-	antiReplayCache.Add([]byte(frame))
53
+	antiReplayCache.Add(replayPart[4:60])
52
 
54
 
53
 	connOpts.ConnectionProto = mtproto.ConnectionProtocolAny
55
 	connOpts.ConnectionProto = mtproto.ConnectionProtocolAny
54
 	connOpts.ClientAddr = conn.RemoteAddr()
56
 	connOpts.ClientAddr = conn.RemoteAddr()

Загрузка…
Отмена
Сохранить