check for PacketNumberLen when writing the PublicHeader packet number

This commit is contained in:
Marten Seemann
2016-11-07 10:42:56 +07:00
parent cc2d966d84
commit db8eb257c6
2 changed files with 26 additions and 6 deletions

View File

@@ -89,6 +89,10 @@ func (h *PublicHeader) Write(b *bytes.Buffer, version protocol.VersionNumber, pe
return nil
}
if h.PacketNumberLen != protocol.PacketNumberLen1 && h.PacketNumberLen != protocol.PacketNumberLen2 && h.PacketNumberLen != protocol.PacketNumberLen4 && h.PacketNumberLen != protocol.PacketNumberLen6 {
return errPacketNumberLenNotSet
}
switch h.PacketNumberLen {
case protocol.PacketNumberLen1:
b.WriteByte(uint8(h.PacketNumber))