don't panic on failed type assertion in PacketPacker

related to #93
2016-05-19 10:55:34 +07:00
parent d14a27f193
commit f751566f00
1 changed files with 5 additions and 1 deletions
--- a/packet_packer.go
+++ b/packet_packer.go
@@ -195,7 +195,11 @@ func (p *packetPacker) composeNextPacket(stopWaitingFrame *frames.StopWaitingFra

 	// remove the dataLen for the last StreamFrame in the packet
 	if hasStreamFrames {
-		payloadFrames[len(payloadFrames)-1].(*frames.StreamFrame).DataLenPresent = false
+		lastStreamFrame, ok := payloadFrames[len(payloadFrames)-1].(*frames.StreamFrame)
+		if !ok {
+			return nil, errors.New("PacketPacker BUG: StreamFrame type assertion failed")
+		}
+		lastStreamFrame.DataLenPresent = false
 		// payloadLength -= 2
 	}