mirror of
https://github.com/aler9/gortsplib
synced 2025-10-04 14:52:46 +08:00
49 lines
729 B
Go
49 lines
729 B
Go
package base
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
)
|
|
|
|
type payload []byte
|
|
|
|
func (c *payload) read(rb *bufio.Reader, header Header) error {
|
|
cls, ok := header["Content-Length"]
|
|
if !ok || len(cls) != 1 {
|
|
*c = nil
|
|
return nil
|
|
}
|
|
|
|
cl, err := strconv.ParseInt(cls[0], 10, 64)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid Content-Length")
|
|
}
|
|
|
|
if cl > rtspMaxContentLength {
|
|
return fmt.Errorf("Content-Length exceeds %d", rtspMaxContentLength)
|
|
}
|
|
|
|
*c = make([]byte, cl)
|
|
n, err := io.ReadFull(rb, *c)
|
|
if err != nil && n != len(*c) {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c payload) write(bw *bufio.Writer) error {
|
|
if len(c) == 0 {
|
|
return nil
|
|
}
|
|
|
|
_, err := bw.Write(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|