diff --git a/viproxy.go b/viproxy.go index 62bb687..7c420ef 100644 --- a/viproxy.go +++ b/viproxy.go @@ -56,6 +56,8 @@ func dial(addr net.Addr) (net.Conn, error) { conn, err = vsock.Dial(a.ContextID, a.Port, nil) case *net.TCPAddr: conn, err = net.DialTCP("tcp", nil, a) + case *net.UnixAddr: + conn, err = net.DialUnix(addr.Network(), nil, a) } if err != nil { return nil, err @@ -73,6 +75,8 @@ func listen(addr net.Addr) (net.Listener, error) { ln, err = vsock.ListenContextID(a.ContextID, a.Port, nil) case *net.TCPAddr: ln, err = net.ListenTCP(a.Network(), a) + case *net.UnixAddr: + ln, err = net.ListenUnix(addr.Network(), a) } if err != nil { return nil, err