mirror of
https://github.com/linuxkit/linuxkit.git
synced 2025-12-03 12:03:18 +00:00
45bb435b51
This allows the proxy to be run easily from a terminal or other script without requiring fd 3 to be open and writable. Signed-off-by: David Scott <dave.scott@docker.com>
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"proxy/libproxy"
|
|
)
|
|
|
|
var interactiveMode bool
|
|
|
|
|
|
// sendError signals the error to the parent and quits the process.
|
|
func sendError(err error) {
|
|
if interactiveMode {
|
|
log.Fatal("Failed to set up proxy", err)
|
|
}
|
|
f := os.NewFile(3, "signal-parent")
|
|
|
|
fmt.Fprintf(f, "1\n%s", err)
|
|
f.Close()
|
|
os.Exit(1)
|
|
}
|
|
|
|
// sendOK signals the parent that the forward is running.
|
|
func sendOK() {
|
|
if interactiveMode {
|
|
log.Println("Proxy running")
|
|
return
|
|
}
|
|
f := os.NewFile(3, "signal-parent")
|
|
fmt.Fprint(f, "0\n")
|
|
f.Close()
|
|
}
|
|
|
|
// Map dynamic TCP ports onto vsock ports over this offset
|
|
var vSockTCPPortOffset = 0x10000
|
|
|
|
// Map dynamic UDP ports onto vsock ports over this offset
|
|
var vSockUDPPortOffset = 0x20000
|
|
|
|
// From docker/libnetwork/portmapper/proxy.go:
|
|
|
|
// parseHostContainerAddrs parses the flags passed on reexec to create the TCP or UDP
|
|
// net.Addrs to map the host and container ports
|
|
func parseHostContainerAddrs() (host net.Addr, port int, container net.Addr) {
|
|
var (
|
|
proto = flag.String("proto", "tcp", "proxy protocol")
|
|
hostIP = flag.String("host-ip", "", "host ip")
|
|
hostPort = flag.Int("host-port", -1, "host port")
|
|
containerIP = flag.String("container-ip", "", "container ip")
|
|
containerPort = flag.Int("container-port", -1, "container port")
|
|
interactive = flag.Bool("i", false, "print success/failure to stdout/stderr")
|
|
)
|
|
|
|
flag.Parse()
|
|
interactiveMode = *interactive
|
|
|
|
switch *proto {
|
|
case "tcp":
|
|
host = &net.TCPAddr{IP: net.ParseIP(*hostIP), Port: *hostPort}
|
|
port = vSockTCPPortOffset + *hostPort
|
|
container = &net.TCPAddr{IP: net.ParseIP(*containerIP), Port: *containerPort}
|
|
case "udp":
|
|
host = &net.UDPAddr{IP: net.ParseIP(*hostIP), Port: *hostPort}
|
|
port = vSockUDPPortOffset + *hostPort
|
|
container = &net.UDPAddr{IP: net.ParseIP(*containerIP), Port: *containerPort}
|
|
default:
|
|
log.Fatalf("unsupported protocol %s", *proto)
|
|
}
|
|
|
|
return host, port, container
|
|
}
|
|
|
|
func handleStopSignals(p libproxy.Proxy) {
|
|
s := make(chan os.Signal, 10)
|
|
signal.Notify(s, os.Interrupt, syscall.SIGTERM, syscall.SIGSTOP)
|
|
|
|
for range s {
|
|
os.Exit(0)
|
|
// The vsock proxy cannot be shutdown the same way as the TCP one:
|
|
//p.Close()
|
|
}
|
|
}
|