From ac186f89f1f64521b54d7728cf39110200120c13 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= <osd@foosel.net>
Date: Sun, 7 Sep 2014 01:57:59 +0200
Subject: [PATCH] discovery now has automatic location url building based on
 querying client

---
 src/octoprint/plugins/discovery/__init__.py | 39 ++++++++++++++++-----
 1 file changed, 30 insertions(+), 9 deletions(-)

diff --git a/src/octoprint/plugins/discovery/__init__.py b/src/octoprint/plugins/discovery/__init__.py
index eda3158c..d480d472 100644
--- a/src/octoprint/plugins/discovery/__init__.py
+++ b/src/octoprint/plugins/discovery/__init__.py
@@ -172,18 +172,36 @@ class DiscoveryPlugin(octoprint.plugin.types.StartupPlugin, octoprint.plugin.typ
 		self._ssdp_monitor_thread.daemon = True
 		self._ssdp_monitor_thread.start()
 
+	@classmethod
+	def interface_addresses(cls, family=None):
+		import netifaces
+		if not family:
+			family = netifaces.AF_INET
+
+		for interface in netifaces.interfaces():
+			ifaddresses = netifaces.ifaddresses(interface)
+			if family in ifaddresses:
+				for ifaddress in ifaddresses[family]:
+					yield ifaddress["addr"]
+
+	@classmethod
+	def address_for_client(cls, client):
+		import socket
+
+		for address in cls.interface_addresses():
+			try:
+				sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+				sock.bind((address, 0))
+				sock.connect(client)
+				return address
+			except Exception as e:
+				pass
+
 	def _ssdp_notify(self, port, alive=True):
 		import socket
 		import netifaces
 
-		def interface_addresses(family=netifaces.AF_INET):
-			for interface in netifaces.interfaces():
-				ifaddresses = netifaces.ifaddresses(interface)
-				if family in ifaddresses:
-					for ifaddress in ifaddresses[family]:
-						yield ifaddress["addr"]
-
-		for addr in interface_addresses():
+		for addr in self.__class__.interface_addresses():
 			try:
 				sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
 				sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
@@ -255,7 +273,10 @@ class DiscoveryPlugin(octoprint.plugin.types.StartupPlugin, octoprint.plugin.typ
 					data, address = sock.recvfrom(4096)
 					request = Request(data)
 					if not request.error_code and request.command == "M-SEARCH" and request.path == "*" and (request.headers["ST"] == "upnp:rootdevice" or request.headers["ST"] == "ssdp:all") and request.headers["MAN"] == '"ssdp:discover"':
-						message = location_message.format(uuid=UUID, location="http://192.168.1.3:5000/plugin/discovery/discovery.xml")
+						address_for_client = self.__class__.address_for_client(address)
+						if not address_for_client:
+							address_for_client = "192.168.1.3"
+						message = location_message.format(uuid=UUID, location="http://{host}:{port}/plugin/discovery/discovery.xml".format(host=address_for_client, port=port))
 						sock.sendto(message, address)
 						self.logger.info("Sent M-SEARCH reply for {path} and {st} to {address!r}".format(path=request.path, st=request.headers["ST"], address=address))
 				except socket.timeout: