IGP handle crypto padding

VKCOM · Jan 14, 2025 · 433f996 · 433f996
1 parent f5837f0
commit 433f996
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/internal/vkgo/rpc/statshouse.go b/internal/vkgo/rpc/statshouse.go
@@ -221,16 +221,18 @@ func forwardPacketBody(dst, src *PacketConn, header *packetHeader, buf []byte) (
 		return res
 	}
 	// skip padding
-	if n := int(-header.length & 3); n != 0 {
-		if _, err := io.ReadFull(src.r, src.headerReadBuf[:n]); err != nil {
-			res.ReadErr = err
-			return res
-		}
-		for i := 0; i < n; i++ {
-			if src.headerReadBuf[i] != 0 {
-				res.ReadErr = errNonZeroPadding
+	if src.w.isEncrypted() {
+		if n := int(-header.length & 3); n != 0 {
+			if _, err := io.ReadFull(src.r, src.headerReadBuf[:n]); err != nil {
+				res.ReadErr = err
 				return res
 			}
+			for i := 0; i < n; i++ {
+				if src.headerReadBuf[i] != 0 {
+					res.ReadErr = errNonZeroPadding
+					return res
+				}
+			}
 		}
 	}
 	return res