huin#48: add local address to the response

goupnp.go

@@ -22,6 +22,7 @@ import (
 	"net/url"
 	"time"
 
+	"github.com/huin/goupnp/httpu"
 	"github.com/huin/goupnp/ssdp"
 )
 
@@ -63,6 +64,9 @@ type MaybeRootDevice struct {
 	// the discovery of a device, regardless of if there was an error probing it.
 	Location *url.URL
 
+	// The address from which the device was discovered (if known - otherwise nil).
+	localAddr string
+
 	// Any error encountered probing a discovered device.
 	Err error
 }
@@ -99,6 +103,7 @@ func DiscoverDevices(searchTarget string) ([]MaybeRootDevice, error) {
 		} else {
 			maybe.Root = root
 		}
+		maybe.localAddr = response.Header.Get(httpu.LocalAddress)
 	}
 
 	return results, nil

httpu/httpu.go

@@ -143,9 +143,14 @@ func (httpu *HTTPUClient) Do(
 			continue
 		}
 
+		// Set the related local address used to discover the device.
+		response.Header.Add(LocalAddress, httpu.conn.LocalAddr().(*net.UDPAddr).IP.String())
+
 		responses = append(responses, response)
 	}
 
 	// Timeout reached - return discovered responses.
 	return responses, nil
 }
+
+const LocalAddress = "goupnp-local-address"

service_client.go

@@ -17,6 +17,7 @@ type ServiceClient struct {
 	RootDevice *RootDevice
 	Location   *url.URL
 	Service    *Service
+	localAddr  string
 }
 
 // NewServiceClients discovers services, and returns clients for them. err will
@@ -36,7 +37,7 @@ func NewServiceClients(searchTarget string) (clients []ServiceClient, errors []e
 			continue
 		}
 
-		deviceClients, err := NewServiceClientsFromRootDevice(maybeRootDevice.Root, maybeRootDevice.Location, searchTarget)
+		deviceClients, err := newServiceClientsFromRootDevice(maybeRootDevice.Root, maybeRootDevice.Location, searchTarget, maybeRootDevice.localAddr)
 		if err != nil {
 			errors = append(errors, err)
 			continue
@@ -61,6 +62,10 @@ func NewServiceClientsByURL(loc *url.URL, searchTarget string) ([]ServiceClient,
 // a given root device. The loc parameter is simply assigned to the
 // Location attribute of the returned ServiceClient(s).
 func NewServiceClientsFromRootDevice(rootDevice *RootDevice, loc *url.URL, searchTarget string) ([]ServiceClient, error) {
+	return newServiceClientsFromRootDevice(rootDevice, loc, searchTarget, "")
+}
+
+func newServiceClientsFromRootDevice(rootDevice *RootDevice, loc *url.URL, searchTarget string, lAddr string) ([]ServiceClient, error) {
 	device := &rootDevice.Device
 	srvs := device.FindService(searchTarget)
 	if len(srvs) == 0 {
@@ -75,6 +80,7 @@ func NewServiceClientsFromRootDevice(rootDevice *RootDevice, loc *url.URL, searc
 			RootDevice: rootDevice,
 			Location:   loc,
 			Service:    srv,
+			localAddr:  lAddr,
 		})
 	}
 	return clients, nil
@@ -86,3 +92,8 @@ func NewServiceClientsFromRootDevice(rootDevice *RootDevice, loc *url.URL, searc
 func (client *ServiceClient) GetServiceClient() *ServiceClient {
 	return client
 }
+
+// LocalAddr returns the address from which the device was discovered (if known - otherwise empty).
+func (client *ServiceClient) LocalAddr() string {
+	return client.localAddr
+}