add pointers to all Read() methods

This commit is contained in:
aler9
2020-12-16 11:33:49 +01:00
parent 2be3fac239
commit 0aaf4d45d8
4 changed files with 24 additions and 21 deletions

View File

@@ -7,36 +7,39 @@ import (
"strconv"
)
func contentRead(rb *bufio.Reader, header Header) ([]byte, error) {
type content []byte
func (c *content) read(rb *bufio.Reader, header Header) error {
cls, ok := header["Content-Length"]
if !ok || len(cls) != 1 {
return nil, nil
*c = nil
return nil
}
cl, err := strconv.ParseInt(cls[0], 10, 64)
if err != nil {
return nil, fmt.Errorf("invalid Content-Length")
return fmt.Errorf("invalid Content-Length")
}
if cl > rtspMaxContentLength {
return nil, fmt.Errorf("Content-Length exceeds %d", rtspMaxContentLength)
return fmt.Errorf("Content-Length exceeds %d", rtspMaxContentLength)
}
ret := make([]byte, cl)
n, err := io.ReadFull(rb, ret)
if err != nil && n != len(ret) {
return nil, err
*c = make([]byte, cl)
n, err := io.ReadFull(rb, *c)
if err != nil && n != len(*c) {
return err
}
return ret, nil
return nil
}
func contentWrite(bw *bufio.Writer, content []byte) error {
if len(content) == 0 {
func (c content) write(bw *bufio.Writer) error {
if len(c) == 0 {
return nil
}
_, err := bw.Write(content)
_, err := bw.Write(c)
if err != nil {
return err
}

View File

@@ -34,7 +34,9 @@ type HeaderValue []string
// Header is a RTSP reader, present in both Requests and Responses.
type Header map[string]HeaderValue
func (h Header) read(rb *bufio.Reader) error {
func (h *Header) read(rb *bufio.Reader) error {
*h = make(Header)
for {
byt, err := rb.ReadByte()
if err != nil {
@@ -50,7 +52,7 @@ func (h Header) read(rb *bufio.Reader) error {
break
}
if len(h) >= headerMaxEntryCount {
if len(*h) >= headerMaxEntryCount {
return fmt.Errorf("headers count exceeds %d", headerMaxEntryCount)
}
@@ -91,7 +93,7 @@ func (h Header) read(rb *bufio.Reader) error {
return err
}
h[key] = append(h[key], val)
(*h)[key] = append((*h)[key], val)
}
return nil

View File

@@ -94,13 +94,12 @@ func (req *Request) Read(rb *bufio.Reader) error {
return err
}
req.Header = make(Header)
err = req.Header.read(rb)
if err != nil {
return err
}
req.Content, err = contentRead(rb, req.Header)
err = (*content)(&req.Content).read(rb, req.Header)
if err != nil {
return err
}
@@ -125,7 +124,7 @@ func (req Request) Write(bw *bufio.Writer) error {
return err
}
err = contentWrite(bw, req.Content)
err = content(req.Content).write(bw)
if err != nil {
return err
}

View File

@@ -172,13 +172,12 @@ func (res *Response) Read(rb *bufio.Reader) error {
return err
}
res.Header = make(Header)
err = res.Header.read(rb)
if err != nil {
return err
}
res.Content, err = contentRead(rb, res.Header)
err = (*content)(&res.Content).read(rb, res.Header)
if err != nil {
return err
}
@@ -208,7 +207,7 @@ func (res Response) Write(bw *bufio.Writer) error {
return err
}
err = contentWrite(bw, res.Content)
err = content(res.Content).write(bw)
if err != nil {
return err
}