Refactor UDP

Signed-off-by: Max Riveiro <[email protected]>
opavlyuk · Jan 14, 2017 · 5489088 · 5489088
1 parent b9e3ac0
commit 5489088
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tcp.go b/tcp.go
@@ -138,3 +138,8 @@ func NewReusablePortListener(proto, addr string) (l net.Listener, err error) {
 
 	return l, err
 }
+
+// Listen function is an alias for NewReusablePortListener.
+func Listen(proto, addr string) (l net.Listener, err error) {
+	return NewReusablePortListener(proto, addr)
+}
diff --git a/udp.go b/udp.go
@@ -46,6 +46,15 @@ func getUDPSockaddr(proto, addr string) (sa syscall.Sockaddr, soType int, err er
 			copy(sa.Addr[:], udp.IP) // copy all bytes of slice to array
 		}
 
+		if udp.Zone != "" {
+			iface, err := net.InterfaceByName(udp.Zone)
+			if err != nil {
+				return nil, -1, err
+			}
+
+			sa.ZoneId = uint32(iface.Index)
+		}
+
 		return sa, syscall.AF_INET6, nil
 	}
 
@@ -124,3 +133,8 @@ func NewReusablePortPacketConn(proto, addr string) (l net.PacketConn, err error)
 
 	return l, err
 }
+
+// ListenPacket is an alias for NewReusablePortPacketConn.
+func ListenPacket(proto, addr string) (l net.PacketConn, err error) {
+	return NewReusablePortPacketConn(proto, addr)
+}