mirror of
https://github.com/linuxkit/linuxkit.git
synced 2025-12-03 11:03:26 +00:00
d5bd7d690a
Previously the proxy would listen only on the vsock port, which is fine for accessing the port on the host, but if a container also wants to access the port (e.g. via `--net=host` and using the Moby IP) then we need to listen on the IP too. Related to [docker/pinata#2854] Signed-off-by: David Scott <dave.scott@docker.com>
73 lines
2.5 KiB
Go
73 lines
2.5 KiB
Go
// Package proxy provides a network Proxy interface and implementations for TCP
|
|
// and UDP.
|
|
package libproxy
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/djs55/vsock"
|
|
"net"
|
|
)
|
|
|
|
// Proxy defines the behavior of a proxy. It forwards traffic back and forth
|
|
// between two endpoints : the frontend and the backend.
|
|
// It can be used to do software port-mapping between two addresses.
|
|
// e.g. forward all traffic between the frontend (host) 127.0.0.1:3000
|
|
// to the backend (container) at 172.17.42.108:4000.
|
|
type Proxy interface {
|
|
// Run starts forwarding traffic back and forth between the front
|
|
// and back-end addresses.
|
|
Run()
|
|
// Close stops forwarding traffic and close both ends of the Proxy.
|
|
Close()
|
|
// FrontendAddr returns the address on which the proxy is listening.
|
|
FrontendAddr() net.Addr
|
|
// BackendAddr returns the proxied address.
|
|
BackendAddr() net.Addr
|
|
}
|
|
|
|
// NewVsockProxy creates a Proxy listening on Vsock
|
|
func NewVsockProxy(frontendAddr *vsock.VsockAddr, backendAddr net.Addr) (Proxy, error) {
|
|
switch backendAddr.(type) {
|
|
case *net.UDPAddr:
|
|
listener, err := vsock.Listen(frontendAddr.Port)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewUDPProxy(frontendAddr, NewUDPListener(listener), backendAddr.(*net.UDPAddr))
|
|
case *net.TCPAddr:
|
|
listener, err := vsock.Listen(frontendAddr.Port)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewTCPProxy(listener, backendAddr.(*net.TCPAddr))
|
|
default:
|
|
panic(fmt.Errorf("Unsupported protocol"))
|
|
}
|
|
}
|
|
|
|
// NewIPProxy creates a Proxy according to the specified frontendAddr and backendAddr.
|
|
func NewIPProxy(frontendAddr, backendAddr net.Addr) (Proxy, error) {
|
|
switch frontendAddr.(type) {
|
|
case *net.UDPAddr:
|
|
listener, err := net.ListenUDP("udp", frontendAddr.(*net.UDPAddr))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewUDPProxy(frontendAddr, listener, backendAddr.(*net.UDPAddr))
|
|
case *net.TCPAddr:
|
|
listener, err := net.Listen("tcp", frontendAddr.String())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewTCPProxy(listener, backendAddr.(*net.TCPAddr))
|
|
case *vsock.VsockAddr:
|
|
listener, err := vsock.Listen(frontendAddr.(*vsock.VsockAddr).Port)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewTCPProxy(listener, backendAddr.(*net.TCPAddr))
|
|
default:
|
|
panic(fmt.Errorf("Unsupported protocol"))
|
|
}
|
|
}
|