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 ""
|