package main

import (
	"flag"
	"fmt"
	"log"
	"net"
	"os"
	"os/signal"
	"syscall"
)

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, localIP bool) {
	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")
		noLocalIP     = flag.Bool("no-local-ip", false, "bind only on the Host, not in the VM")
	)

	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)
	}
	localIP = !*noLocalIP
	return host, port, container, localIP
}

func handleStopSignals() {
	s := make(chan os.Signal, 10)
	signal.Notify(s, os.Interrupt, syscall.SIGTERM, syscall.SIGSTOP)

	for range s {
		os.Exit(0)
	}
}