From fbc6c9339552d252b3e560e655c3f485acca05d1 Mon Sep 17 00:00:00 2001 From: wweir Date: Thu, 29 Jul 2021 08:51:46 +0800 Subject: [PATCH] extend sower protocol checksum length --- transport/sower/sower.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/transport/sower/sower.go b/transport/sower/sower.go index 70a911d..cd393f6 100644 --- a/transport/sower/sower.go +++ b/transport/sower/sower.go @@ -19,7 +19,7 @@ var headSize = binary.Size(new(Head)) // data(HTTP, first byte < 0x7F) type Head struct { Cmd byte - Checksum byte + Checksum uint64 Port uint16 TgtAddr [maxDomainLength]byte } @@ -74,6 +74,8 @@ func (s *Sower) Wrap(conn net.Conn, tgtHost string, tgtPort uint16) error { }) } -func sumChecksum(target [maxDomainLength]byte, password []byte) byte { - return md5.Sum(append(target[:], password...))[0] +func sumChecksum(target [maxDomainLength]byte, password []byte) uint64 { + checksum := md5.Sum(append(target[:], password...)) + checksumVal, _ := binary.Uvarint(checksum[:]) + return checksumVal }