mirror of
https://github.com/aler9/gortsplib
synced 2025-11-01 02:52:36 +08:00
55 lines
858 B
Go
55 lines
858 B
Go
package base
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
rtspMaxContentLength = 128 * 1024
|
|
)
|
|
|
|
type body []byte
|
|
|
|
func (b *body) unmarshal(header Header, rb *bufio.Reader) error {
|
|
cls, ok := header["Content-Length"]
|
|
if !ok || len(cls) != 1 {
|
|
*b = nil
|
|
return nil
|
|
}
|
|
|
|
cl, err := strconv.ParseUint(cls[0], 10, 64)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid Content-Length")
|
|
}
|
|
|
|
if cl > rtspMaxContentLength {
|
|
return fmt.Errorf("Content-Length exceeds %d (it's %d)",
|
|
rtspMaxContentLength, cl)
|
|
}
|
|
|
|
*b = make([]byte, cl)
|
|
n, err := io.ReadFull(rb, *b)
|
|
if err != nil && n != len(*b) {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b body) marshalSize() int {
|
|
return len(b)
|
|
}
|
|
|
|
func (b body) marshalTo(buf []byte) int {
|
|
return copy(buf, b)
|
|
}
|
|
|
|
func (b body) marshal() []byte {
|
|
buf := make([]byte, b.marshalSize())
|
|
b.marshalTo(buf)
|
|
return buf
|
|
}
|