1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// SPDX-FileCopyrightText: V <v@anomalous.eu>
// SPDX-FileCopyrightText: edef <edef@edef.eu>
// SPDX-License-Identifier: OSL-3.0
package main // import "go.anomalous.eu/loxy"
import (
"context"
"crypto/tls"
"flag"
"io/ioutil"
"log"
"net"
"net/http"
"os"
"os/signal"
"syscall"
)
var dbpath, certpath string
var addr interface {
flag.Value
Listeners() ([]net.Listener, error)
}
func init() {
if os.Getenv("LISTEN_FDS") == "" {
a := listenAddress(net.JoinHostPort("::1", "3893"))
addr = &a
} else {
addr = activationSocket{}
}
flag.Var(addr, "addr", "listen address")
flag.StringVar(&dbpath, "db", "loxy.db", "`path` to database")
flag.StringVar(&certpath, "cert", "", "`path` to client certificate")
log.SetFlags(log.Lshortfile)
}
func main() {
flag.Parse()
if flag.NArg() != 0 {
flag.Usage()
os.Exit(1)
}
tlsConfig := &tls.Config{}
if certpath != "" {
pem, err := ioutil.ReadFile(certpath)
if err != nil {
log.Fatal(err)
}
cert, err := tls.X509KeyPair(pem, pem)
if err != nil {
log.Fatal(err)
}
tlsConfig.Certificates = []tls.Certificate{cert}
}
proxy := NewProxy(OpenStore(dbpath), tlsConfig)
server := &http.Server{Handler: proxy}
ctx, cancel := context.WithCancel(context.Background())
server.BaseContext = func(net.Listener) context.Context { return ctx }
server.RegisterOnShutdown(cancel)
sig := make(chan os.Signal, 1)
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
listeners, err := addr.Listeners()
if err != nil {
log.Fatal(err)
}
serve := make(chan error, len(listeners))
for _, ln := range listeners {
go func(ln net.Listener) { serve <- server.Serve(ln) }(ln)
}
select {
case err = <-serve:
log.Printf("http.ListenAndServe: %v", err)
case sig := <-sig:
log.Printf("caught %v, shutting down", sig)
server.Shutdown(context.Background())
}
proxy.Shutdown()
if err != nil {
os.Exit(1)
}
}
|