network/parser_stream.go

package network

import (
	"fmt"
	coder "git.hpds.cc/Component/mq_coder"
	"git.hpds.cc/Component/network/frame"
	"io"
)

// ParseFrame parses the frame from QUIC stream.
func ParseFrame(stream io.Reader) (frame.Frame, error) {
	buf, err := coder.ReadPacket(stream)
	if err != nil {
		return nil, err
	}

	frameType := buf[0]
	switch frameType {
	case 0x80 | byte(frame.TagOfHandshakeFrame):
		return frame.DecodeToHandshakeFrame(buf)
	case 0x80 | byte(frame.TagOfDataFrame):
		return frame.DecodeToDataFrame(buf)
	case 0x80 | byte(frame.TagOfAcceptedFrame):
		return frame.DecodeToAcceptedFrame(buf)
	case 0x80 | byte(frame.TagOfRejectedFrame):
		return frame.DecodeToRejectedFrame(buf)
	case 0x80 | byte(frame.TagOfGoawayFrame):
		return frame.DecodeToGoawayFrame(buf)
	case 0x80 | byte(frame.TagOfBackFlowFrame):
		return frame.DecodeToBackFlowFrame(buf)
	case 0x80 | byte(frame.TagOfHandshakeAckFrame):
		return frame.DecodeToHandshakeAckFrame(buf)
	case 0x80 | byte(frame.TagOfAuthenticationFrame):
		return frame.DecodeToAuthenticationFrame(buf)
	case 0x80 | byte(frame.TagOfAuthenticationAckFrame):
		return frame.DecodeToAuthenticationRespFrame(buf)
	case 0x80 | (byte(frame.TagOfCloseStreamFrame)):
		return frame.DecodeToCloseStreamFrame(buf)
	default:
		return nil, fmt.Errorf("unknown frame type, buf[0]=%#x", buf[0])
	}
}