Udp (#34)

* add udp

* version bump

hookah.go

@@ -12,7 +12,7 @@ import (
 )
 
 // Version of hookah API.
-const Version = "2.1.0"
+const Version = "2.2.0"
 
 // API is an instance of the Hookah API.
 type API struct {
@@ -88,6 +88,9 @@ func (a *API) registerProtocols() {
 	a.RegisterProtocol("tcp-listen", "tcp-listen://address", protocols.TCPListen)
 	a.RegisterProtocol("tls", "tls://address?cert=path&insecure=false", protocols.TLS)
 	a.RegisterProtocol("tls-listen", "tls-listen://address?cert=path&key=path", protocols.TLSListen)
+	a.RegisterProtocol("udp", "udp://address", protocols.UDP)
+	a.RegisterProtocol("udp-listen", "udp-listen://address", protocols.UDPListen)
+	a.RegisterProtocol("udp-multicast", "udp-multicast://address", protocols.UDPMulticast)
 	a.RegisterProtocol("unix", "unix://path/to/sock", protocols.Unix)
 	a.RegisterProtocol("unix-listen", "unix-listen://path/to/sock", protocols.UnixListen)
 	a.RegisterProtocol("ws", "ws://address", protocols.WS)

internal/protocols/udp.go

@@ -0,0 +1,16 @@
+package protocols
+
+import (
+	"net"
+
+	"github.com/wybiral/hookah/pkg/node"
+)
+
+// UDP creates a UDP client node
+func UDP(addr string) (*node.Node, error) {
+	rwc, err := net.Dial("udp", addr)
+	if err != nil {
+		return nil, err
+	}
+	return node.New(rwc), nil
+}

internal/protocols/udplisten.go

@@ -0,0 +1,61 @@
+package protocols
+
+import (
+	"net"
+	"net/url"
+	"strings"
+
+	"github.com/wybiral/hookah/pkg/node"
+)
+
+// UDPListen creates a UDP listener Node
+func UDPListen(addr string) (*node.Node, error) {
+	a, err := net.ResolveUDPAddr("udp", addr)
+	if err != nil {
+		return nil, err
+	}
+	c, err := net.ListenUDP("udp", a)
+	if err != nil {
+		return nil, err
+	}
+	n := node.New(c)
+	n.W = nil
+	return n, nil
+}
+
+// UDPMulticast creates a UDP multicast listener Node
+func UDPMulticast(arg string) (*node.Node, error) {
+	var err error
+	var opts url.Values
+	// Parse options
+	addrOpts := strings.SplitN(arg, "?", 2)
+	addr := addrOpts[0]
+	if len(addrOpts) == 2 {
+		op, err := url.ParseQuery(addrOpts[1])
+		if err != nil {
+			return nil, err
+		}
+		opts = op
+	}
+	iface := opts.Get("iface")
+	var ifi *net.Interface
+	ifi = nil
+	if len(iface) > 0 {
+		// If interface is supplied, look it up
+		ifi, err = net.InterfaceByName(iface)
+		if err != nil {
+			return nil, err
+		}
+	}
+	a, err := net.ResolveUDPAddr("udp", addr)
+	if err != nil {
+		return nil, err
+	}
+	c, err := net.ListenMulticastUDP("udp", ifi, a)
+	if err != nil {
+		return nil, err
+	}
+	n := node.New(c)
+	n.W = nil
+	return n, nil
+}