rmfakecloud-proxy/main.go

97 lines
1.9 KiB
Go
Raw Normal View History

2021-02-09 17:39:44 +01:00
//go:generate go run generate/versioninfo.go
2018-08-18 00:03:05 +08:00
// secure is a super simple TLS termination proxy
2018-08-17 00:39:01 +08:00
package main
import (
2018-08-18 00:03:05 +08:00
"context"
2018-08-17 00:39:01 +08:00
"flag"
"fmt"
2018-08-18 00:03:05 +08:00
"log"
2018-08-17 00:39:01 +08:00
"net/http"
2018-08-18 00:03:05 +08:00
"net/http/httputil"
"net/url"
2018-08-17 00:39:01 +08:00
"os"
2018-08-18 00:03:05 +08:00
"os/signal"
"path/filepath"
2018-08-17 00:39:01 +08:00
"syscall"
)
var (
certFile string
keyFile string
upstream string
addr string
2018-08-18 01:31:45 +08:00
version bool
2018-08-17 00:39:01 +08:00
)
func init() {
flag.StringVar(&addr, "addr", ":443", "listen address")
2018-08-18 00:03:05 +08:00
flag.StringVar(&certFile, "cert", "", "path to cert file")
flag.StringVar(&keyFile, "key", "", "path to key file")
2018-08-18 01:31:45 +08:00
flag.BoolVar(&version, "version", false, "print version string and exit")
2018-08-18 00:03:05 +08:00
flag.Usage = func() {
fmt.Fprintf(flag.CommandLine.Output(),
2018-08-18 01:31:45 +08:00
"usage: %s [-addr host:port] -cert certfile -key keyfile [-version] upstream\n",
2018-08-18 00:03:05 +08:00
filepath.Base(os.Args[0]))
flag.PrintDefaults()
fmt.Fprintln(flag.CommandLine.Output(), " upstream string\n \tupstream url")
}
2018-08-17 00:39:01 +08:00
}
func _main() error {
flag.Parse()
2018-08-18 01:31:45 +08:00
if version {
fmt.Fprintln(flag.CommandLine.Output(), Version)
os.Exit(0)
}
2018-08-18 00:03:05 +08:00
if flag.NArg() == 1 {
upstream = flag.Arg(0)
} else {
flag.Usage()
os.Exit(2)
}
2018-08-17 00:39:01 +08:00
u, err := url.Parse(upstream)
if err != nil {
return fmt.Errorf("invalid upstream address: %v", err)
}
rp := httputil.NewSingleHostReverseProxy(u)
srv := http.Server{
Handler: rp,
Addr: addr,
}
2018-08-18 00:03:05 +08:00
done := make(chan struct{})
2018-08-17 00:39:01 +08:00
go func() {
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt, syscall.SIGTERM)
fmt.Println(<-sig)
if err := srv.Shutdown(context.Background()); err != nil {
2018-08-18 00:03:05 +08:00
fmt.Printf("Shutdown: %v", err)
2018-08-17 00:39:01 +08:00
}
2018-08-18 00:03:05 +08:00
close(done)
2018-08-17 00:39:01 +08:00
}()
2018-08-18 00:03:05 +08:00
log.Printf("cert-file=%s key-file=%s listen-addr=%s upstream-url=%s", certFile, keyFile, srv.Addr, u.String())
2018-08-17 00:39:01 +08:00
if err := srv.ListenAndServeTLS(certFile, keyFile); err != http.ErrServerClosed {
return fmt.Errorf("ListenAndServeTLS: %v", err)
}
2018-08-18 00:03:05 +08:00
<-done
2018-08-17 00:39:01 +08:00
return nil
}
func main() {
err := _main()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
}