Udp (#10)

* add udp protocols

* add udp multicast

* udp examples

Author: wybiral

examples_test.go

@@ -47,6 +47,18 @@ func ExampleNewInput_tcpListen() {
 	r, err = hookah.NewInput("tcp-listen://localhost:8080")
 }
 
+// Create UDP listen input.
+func ExampleNewInput_udpListen() {
+	r, err = hookah.NewInput("udp-listen://localhost:8080")
+}
+
+// Create UDP multicast listen input.
+// Here the network interface is set to eth0 but can be ommitted to use the
+// default.
+func ExampleNewInput_udpMulticast() {
+	r, err = hookah.NewInput("udp-multicast://eth0,localhost:8080")
+}
+
 // Create Unix client input.
 func ExampleNewInput_unixClient() {
 	r, err = hookah.NewInput("unix://path/to/sock")
@@ -109,6 +121,11 @@ func ExampleNewOutput_tcpListen() {
 	w, err = hookah.NewOutput("tcp-listen://localhost:8080")
 }
 
+// Create UDP client output.
+func ExampleNewOutput_udpClient() {
+	w, err = hookah.NewOutput("udp://localhost:8080")
+}
+
 // Create Unix client output.
 func ExampleNewOutput_unixClient() {
 	w, err = hookah.NewOutput("unix://path/to/sock")

pkg/input/input.go

@@ -63,6 +63,16 @@ func New(opts string) (io.ReadCloser, error) {
 			return nil, errors.New("tcp-listen: no address supplied")
 		}
 		return tcpListen(arg)
+	case "udp-listen", "udp-server":
+		if arg == "" {
+			return nil, errors.New("udp-listen: no address supplied")
+		}
+		return udpListen(arg)
+	case "udp-multicast":
+		if arg == "" {
+			return nil, errors.New("udp-multicast: no address supplied")
+		}
+		return udpMulticast(arg)
 	case "unix":
 		if arg == "" {
 			return nil, errors.New("unix: no address supplied")

pkg/input/udplisten.go

@@ -0,0 +1,15 @@
+package input
+
+import (
+	"io"
+	"net"
+)
+
+// Create a UDP listener and return as ReadCloser
+func udpListen(addr string) (io.ReadCloser, error) {
+	a, err := net.ResolveUDPAddr("udp", addr)
+	if err != nil {
+		return nil, err
+	}
+	return net.ListenUDP("udp", a)
+}

pkg/input/udpmulticast.go

@@ -0,0 +1,30 @@
+package input
+
+import (
+	"io"
+	"net"
+	"strings"
+)
+
+// Create a UDP multicast listener and return as ReadCloser
+func udpMulticast(arg string) (io.ReadCloser, error) {
+	var addr string
+	var err error
+	var iface *net.Interface
+	parts := strings.SplitN(arg, ",", 2)
+	if len(parts) == 1 {
+		addr = parts[0]
+	} else {
+		addr = parts[1]
+		// If interface is supplied, look it up
+		iface, err = net.InterfaceByName(parts[0])
+		if err != nil {
+			return nil, err
+		}
+	}
+	a, err := net.ResolveUDPAddr("udp", addr)
+	if err != nil {
+		return nil, err
+	}
+	return net.ListenMulticastUDP("udp", iface, a)
+}

pkg/output/output.go

@@ -65,6 +65,11 @@ func New(opts string) (io.WriteCloser, error) {
 			return nil, errors.New("tcp-listen: no address supplied")
 		}
 		return tcpListen(arg)
+	case "udp":
+		if arg == "" {
+			return nil, errors.New("udp: no address supplied")
+		}
+		return udpClient(arg)
 	case "unix":
 		if arg == "" {
 			return nil, errors.New("unix: no address supplied")

pkg/output/udpclient.go

@@ -0,0 +1,11 @@
+package output
+
+import (
+	"io"
+	"net"
+)
+
+// Create a UDP client and return as WriteCloser
+func udpClient(addr string) (io.WriteCloser, error) {
+	return net.Dial("udp", addr)
+}