112 lines
3.7 KiB
Python
112 lines
3.7 KiB
Python
|
"""ONVIF WS-Discovery 设备发现"""
|
|||
|
|
import logging
|
|||
|
|
import socket
|
|||
|
|
import struct
|
|||
|
|
import uuid
|
|||
|
|
from xml.etree import ElementTree as ET
|
|||
|
|
|
|||
|
|
logger = logging.getLogger("services.onvif")
|
|||
|
|
|
|||
|
|
WS_DISCOVERY_ADDR = "239.255.255.250"
|
|||
|
|
WS_DISCOVERY_PORT = 3702
|
|||
|
|
|
|||
|
|
PROBE_TEMPLATE = """<?xml version="1.0" encoding="UTF-8"?>
|
|||
|
|
<e:Envelope xmlns:e="http://www.w3.org/2003/05/soap-envelope"
|
|||
|
|
xmlns:w="http://schemas.xmlsoap.org/ws/2004/08/addressing"
|
|||
|
|
xmlns:d="http://schemas.xmlsoap.org/ws/2005/04/discovery"
|
|||
|
|
xmlns:dn="http://www.onvif.org/ver10/network/wsdl">
|
|||
|
|
<e:Header>
|
|||
|
|
<w:MessageID>uuid:{msg_id}</w:MessageID>
|
|||
|
|
<w:To e:mustUnderstand="true">urn:schemas-xmlsoap-org:ws:2005:04:discovery</w:To>
|
|||
|
|
<w:Action a:mustUnderstand="true" xmlns:a="http://www.w3.org/2003/05/soap-envelope">http://schemas.xmlsoap.org/ws/2005/04/discovery/Probe</w:Action>
|
|||
|
|
</e:Header>
|
|||
|
|
<e:Body>
|
|||
|
|
<d:Probe>
|
|||
|
|
<d:Types>dn:NetworkVideoTransmitter</d:Types>
|
|||
|
|
</d:Probe>
|
|||
|
|
</e:Body>
|
|||
|
|
</e:Envelope>"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _parse_probe_match(data):
|
|||
|
|
try:
|
|||
|
|
root = ET.fromstring(data)
|
|||
|
|
except Exception:
|
|||
|
|
return None
|
|||
|
|
ns = {
|
|||
|
|
"s": "http://www.w3.org/2003/05/soap-envelope",
|
|||
|
|
"wsa": "http://schemas.xmlsoap.org/ws/2004/08/addressing",
|
|||
|
|
"d": "http://schemas.xmlsoap.org/ws/2005/04/discovery",
|
|||
|
|
}
|
|||
|
|
scopes = ""
|
|||
|
|
xaddrs = ""
|
|||
|
|
for el in root.iter():
|
|||
|
|
tag = el.tag.split("}")[-1] if "}" in el.tag else el.tag
|
|||
|
|
if tag == "Scopes" and el.text:
|
|||
|
|
scopes = el.text.strip()
|
|||
|
|
if tag == "XAddrs" and el.text:
|
|||
|
|
xaddrs = el.text.strip()
|
|||
|
|
if not xaddrs:
|
|||
|
|
return None
|
|||
|
|
url = xaddrs.split()[0]
|
|||
|
|
name = scopes.split("/")[-1] if scopes else url
|
|||
|
|
return {"name": name, "xaddr": url, "scopes": scopes}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def discover_onvif(timeout=3.0):
|
|||
|
|
"""UDP 组播 WS-Discovery，返回 [{name, xaddr, scopes, ip}]"""
|
|||
|
|
results = []
|
|||
|
|
seen = set()
|
|||
|
|
msg = PROBE_TEMPLATE.format(msg_id=uuid.uuid4()).encode("utf-8")
|
|||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
|
|||
|
|
try:
|
|||
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|||
|
|
sock.settimeout(0.5)
|
|||
|
|
ttl = struct.pack("b", 2)
|
|||
|
|
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, ttl)
|
|||
|
|
sock.sendto(msg, (WS_DISCOVERY_ADDR, WS_DISCOVERY_PORT))
|
|||
|
|
import time
|
|||
|
|
deadline = time.time() + timeout
|
|||
|
|
while time.time() < deadline:
|
|||
|
|
try:
|
|||
|
|
data, addr = sock.recvfrom(65535)
|
|||
|
|
except socket.timeout:
|
|||
|
|
continue
|
|||
|
|
ip = addr[0]
|
|||
|
|
if ip in seen:
|
|||
|
|
continue
|
|||
|
|
item = _parse_probe_match(data)
|
|||
|
|
if item:
|
|||
|
|
item["ip"] = ip
|
|||
|
|
seen.add(ip)
|
|||
|
|
results.append(item)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning("ONVIF discovery 失败: %s", e)
|
|||
|
|
finally:
|
|||
|
|
sock.close()
|
|||
|
|
return results
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_rtsp_url_from_onvif(xaddr, username="", password=""):
|
|||
|
|
"""尝试通过 ONVIF 获取 RTSP 主码流地址"""
|
|||
|
|
try:
|
|||
|
|
from onvif import ONVIFCamera
|
|||
|
|
from urllib.parse import urlparse
|
|||
|
|
p = urlparse(xaddr if "://" in xaddr else "http://" + xaddr)
|
|||
|
|
host = p.hostname
|
|||
|
|
port = p.port or 80
|
|||
|
|
cam = ONVIFCamera(host, port, username or "admin", password or "admin")
|
|||
|
|
media = cam.create_media_service()
|
|||
|
|
profiles = media.GetProfiles()
|
|||
|
|
if not profiles:
|
|||
|
|
return ""
|
|||
|
|
token = profiles[0].token
|
|||
|
|
uri = media.GetStreamUri({
|
|||
|
|
"StreamSetup": {"Stream": "RTP-Unicast", "Transport": {"Protocol": "RTSP"}},
|
|||
|
|
"ProfileToken": token,
|
|||
|
|
})
|
|||
|
|
return uri.Uri if uri else ""
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.debug("ONVIF get stream uri: %s", e)
|
|||
|
|
return ""
|