summaryrefslogtreecommitdiff
path: root/m4p/discover.go
blob: a63dde284e6d688a2264907f333ceab6fcad94cb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package m4p

import (
	"log"
	"net"
)

// Discoverer magic4pc servers.
type Discoverer struct {
	ln     *net.UDPConn
	device chan DeviceInfo
}

// NewDiscover returns a new Discoverer that listens on the broadcast port.
func NewDiscoverer(broadcastPort int) (*Discoverer, error) {
	addr := net.UDPAddr{
		Port: broadcastPort,
		IP:   net.ParseIP("0.0.0.0"),
	}

	ln, err := net.ListenUDP("udp", &addr)
	if err != nil {
		panic(err)
	}

	d := &Discoverer{
		ln:     ln,
		device: make(chan DeviceInfo), // Unbuffered, discard when nobody is listening.
	}
	go d.discover()

	return d, nil
}

func (d *Discoverer) discover() {
	var buf [1024]byte
	for {
		n, addr, err := d.ln.ReadFromUDP(buf[:])
		if err != nil {
			log.Printf("m4p: Discoverer: discover: read udp packet failed: %v", err)
			continue
		}

		m, err := decode(buf[:n])
		if err != nil {
			log.Printf("m4p: Discoverer: discover: decode failed: %v", err)
		}

		switch m.Type {
		case Magic4PCAdMessage:
			dev := m.DeviceInfo
			dev.IPAddr = addr.IP.String()
			log.Printf("m4p: Discoverer: discover: found device: %#v", dev)

			select {
			case d.device <- *dev:
			default:
			}

		default:
			log.Printf("m4p: Discoverer: discover: unknown message: %s", m.Type)
		}
	}
}

// NextDevice returns a newly discovered magic4pc server.
func (d *Discoverer) NextDevice() <-chan DeviceInfo {
	return d.device
}

// Close the Discoverer and stop listening for broadcasts.
func (d *Discoverer) Close() error {
	return d.ln.Close()
}