jrouter/atalk/rtmp/response.go

package rtmp

import (
	"bytes"
	"encoding/binary"
	"fmt"

	"github.com/sfiera/multitalk/pkg/ddp"
)

// ResponsePacket represents an RTMP Response packet.
type ResponsePacket struct {
	SenderAddr ddp.Addr
	Extended   bool
	RangeStart ddp.Network
	RangeEnd   ddp.Network
}

// Marshal marshals an RTMP Response packet.
func (rp *ResponsePacket) Marshal() ([]byte, error) {
	b := bytes.NewBuffer(nil)
	b.Grow(10)
	write16(b, rp.SenderAddr.Network)
	b.WriteByte(8)
	b.WriteByte(byte(rp.SenderAddr.Node))
	if !rp.Extended {
		return b.Bytes(), nil
	}
	write16(b, rp.RangeStart)
	b.WriteByte(0x80)
	write16(b, rp.RangeEnd)
	b.WriteByte(0x82)
	return b.Bytes(), nil
}

// UnmarshalResponsePacket unmarshals an RTMP Response packet.
func UnmarshalResponsePacket(data []byte) (*ResponsePacket, error) {
	if len(data) != 4 && len(data) != 10 {
		return nil, fmt.Errorf("invalid input length %d for RTMP Response packet", len(data))
	}
	if data[2] != 8 {
		return nil, fmt.Errorf("unsupported node ID length %d for RTMP Response packet", data[2])
	}
	rp := &ResponsePacket{
		SenderAddr: ddp.Addr{
			Network: ddp.Network(binary.BigEndian.Uint16(data[:2])),
			Node:    ddp.Node(data[3]),
		},
	}
	if len(data) == 4 {
		return rp, nil
	}

	rp.RangeStart = ddp.Network(binary.BigEndian.Uint16(data[4:6]))
	if data[6] != 0x80 {
		return nil, fmt.Errorf("invalid intermediate byte %x for RTMP Response packet", data[6])
	}
	rp.RangeEnd = ddp.Network(binary.BigEndian.Uint16(data[7:9]))
	if data[9] != 0x82 {
		return nil, fmt.Errorf("unsupported version %x for RTMP Response packet", data[9])
	}
	return rp, nil
}