From b21398938be8893fa719712555367b189bcb778b Mon Sep 17 00:00:00 2001
From: Rolf Ahrenberg <Rolf.Ahrenberg@sci.fi>
Date: Fri, 21 Jul 2017 22:59:18 +0300
Subject: [PATCH] Add a server detection script.

---
 detectsatip.py | 70 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100755 detectsatip.py

diff --git a/detectsatip.py b/detectsatip.py
new file mode 100755
index 0000000..8efccb6
--- /dev/null
+++ b/detectsatip.py
@@ -0,0 +1,70 @@
+#!/usr/bin/env python2
+
+import json
+import socket
+import sys
+import urllib2
+import xml.etree.ElementTree as ET
+
+SSDP_ADDR = "239.255.255.250";
+SSDP_PORT = 1900;
+SSDP_MX = 1;
+SSDP_ST = "urn:ses-com:device:SatIPServer:1";
+
+ssdpRequest = "M-SEARCH * HTTP/1.1\r\n" + \
+                "HOST: %s:%d\r\n" % (SSDP_ADDR, SSDP_PORT) + \
+                "MAN: \"ssdp:discover\"\r\n" + \
+                "MX: %d\r\n" % (SSDP_MX, ) + \
+                "ST: %s\r\n" % (SSDP_ST, ) + "\r\n";
+
+data = None
+urls = list()
+devices = list()
+
+# find out available SAT>IP servers
+sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+sock.setblocking(0)
+sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+try:
+    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
+except:
+    pass
+sock.settimeout(1)
+sock.bind(("192.168.0.6", SSDP_PORT))
+sock.sendto(ssdpRequest, (SSDP_ADDR, SSDP_PORT))
+try:
+    while 1:
+        data = sock.recv(1024)
+        if data:
+            # find out their capabilities
+            for row in data.split("\r\n"):
+                if "LOCATION:" in row:
+                    url = row.replace("LOCATION:","").strip()
+                    if url in urls:
+                        continue
+                    urls.append(url)
+                    f = urllib2.urlopen(url, timeout=2)
+                    xml = f.read()
+                    if xml:
+                        root = ET.fromstring(xml)
+                        name = root.find(".//*/{urn:schemas-upnp-org:device-1-0}friendlyName")
+                        satipcap = root.find(".//*/{urn:ses-com:satip}X_SATIPCAP")
+                        caps = dict()
+                        for systems in satipcap.text.split(","):
+                            cap = systems.split("-")
+                            if cap:
+                                # prepare the output data
+                                count = int(cap[1])
+                                if cap[0] in caps:
+                                    count = count + caps[cap[0]]
+                                caps[cap[0]] = count
+                        devices.append({"name": name.text, "frontends": caps})
+        else:
+            break
+except:
+    pass
+sock.close()
+
+# print out the output data
+json.dump(devices, fp=sys.stdout, sort_keys=True, indent=2)