Skip to content

Commit e0bd51c

Browse files
committed
IGP handle crypto padding
1 parent e5cbf86 commit e0bd51c

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

internal/vkgo/rpc/statshouse.go

+10-8
Original file line numberDiff line numberDiff line change
@@ -221,16 +221,18 @@ func forwardPacketBody(dst, src *PacketConn, header *packetHeader, buf []byte) (
221221
return res
222222
}
223223
// skip padding
224-
if n := int(-header.length & 3); n != 0 {
225-
if _, err := io.ReadFull(src.r, src.headerReadBuf[:n]); err != nil {
226-
res.ReadErr = err
227-
return res
228-
}
229-
for i := 0; i < n; i++ {
230-
if src.headerReadBuf[i] != 0 {
231-
res.ReadErr = errNonZeroPadding
224+
if src.w.isEncrypted() {
225+
if n := int(-header.length & 3); n != 0 {
226+
if _, err := io.ReadFull(src.r, src.headerReadBuf[:n]); err != nil {
227+
res.ReadErr = err
232228
return res
233229
}
230+
for i := 0; i < n; i++ {
231+
if src.headerReadBuf[i] != 0 {
232+
res.ReadErr = errNonZeroPadding
233+
return res
234+
}
235+
}
234236
}
235237
}
236238
return res

0 commit comments

Comments
 (0)