Restore simple onvif client logic
This commit is contained in:
+7
-59
@@ -1,13 +1,10 @@
|
|||||||
package onvif
|
package onvif
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"bytes"
|
"bytes"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
|
||||||
"html"
|
"html"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"regexp"
|
"regexp"
|
||||||
@@ -177,73 +174,24 @@ func (c *Client) MediaRequest(operation string) ([]byte, error) {
|
|||||||
return c.Request(c.mediaURL, operation)
|
return c.Request(c.mediaURL, operation)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Client) Request(rawUrl, body string) ([]byte, error) {
|
func (c *Client) Request(url, body string) ([]byte, error) {
|
||||||
if rawUrl == "" {
|
if url == "" {
|
||||||
return nil, errors.New("onvif: unsupported service")
|
return nil, errors.New("onvif: unsupported service")
|
||||||
}
|
}
|
||||||
|
|
||||||
u, err := url.Parse(rawUrl)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
host := u.Host
|
|
||||||
if u.Port() == "" {
|
|
||||||
host += ":80"
|
|
||||||
}
|
|
||||||
|
|
||||||
const timeout = 5 * time.Second
|
|
||||||
|
|
||||||
conn, err := net.DialTimeout("tcp", host, timeout)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
|
|
||||||
e := NewEnvelopeWithUser(c.url.User)
|
e := NewEnvelopeWithUser(c.url.User)
|
||||||
e.Append(body)
|
e.Append(body)
|
||||||
buf := e.Bytes()
|
|
||||||
|
|
||||||
req := &http.Request{
|
client := &http.Client{Timeout: time.Second * 5000}
|
||||||
Method: "POST",
|
res, err := client.Post(url, `application/soap+xml;charset=utf-8`, bytes.NewReader(e.Bytes()))
|
||||||
URL: u,
|
if err != nil {
|
||||||
Proto: "HTTP/1.1",
|
|
||||||
Header: http.Header{"Content-Type": {"application/soap+xml;charset=utf-8"}},
|
|
||||||
Body: io.NopCloser(bytes.NewReader(buf)),
|
|
||||||
ContentLength: int64(len(buf)),
|
|
||||||
Close: true,
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = conn.SetWriteDeadline(time.Now().Add(timeout))
|
|
||||||
if err = req.Write(conn); err != nil {
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
defer res.Body.Close()
|
||||||
rd := bufio.NewReaderSize(conn, 16*1024)
|
|
||||||
|
|
||||||
_ = conn.SetReadDeadline(time.Now().Add(timeout))
|
|
||||||
res, err := http.ReadResponse(rd, req)
|
|
||||||
if err != nil {
|
|
||||||
// Try to fix broken response https://github.com/AlexxIT/go2rtc/pull/1589
|
|
||||||
if buf, err = io.ReadAll(rd); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Look for XML in complete response
|
|
||||||
if i := bytes.Index(buf, []byte("<?xml")); i > 0 {
|
|
||||||
return buf[i:], nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil, fmt.Errorf("onvif: broken response: %.100s", buf)
|
|
||||||
}
|
|
||||||
|
|
||||||
if res.StatusCode != http.StatusOK {
|
if res.StatusCode != http.StatusOK {
|
||||||
return nil, errors.New("onvif: wrong response " + res.Status)
|
return nil, errors.New("onvif: wrong response " + res.Status)
|
||||||
}
|
}
|
||||||
|
|
||||||
if buf, err = io.ReadAll(res.Body); err != nil {
|
return io.ReadAll(res.Body)
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return buf, nil
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user