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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// Command atlas TODO.
package main
import (
"context"
"errors"
"fmt"
"net/http"
"os"
"os/signal"
"strings"
"syscall"
"net/http/pprof"
"github.com/hashicorp/go-envparse"
_ "github.com/mattn/go-sqlite3"
"github.com/r2northstar/atlas/pkg/atlas"
"github.com/spf13/pflag"
)
var opt struct {
Help bool
}
func init() {
pflag.BoolVarP(&opt.Help, "help", "h", false, "Show this help text")
}
func main() {
pflag.Parse()
if pflag.NArg() > 1 || opt.Help {
fmt.Printf("usage: %s [options] [env_file]\n\noptions:\n%s\nnote: if env_file is provided, config from the environment is ignored\n", os.Args[0], pflag.CommandLine.FlagUsages())
if opt.Help {
os.Exit(2)
}
os.Exit(0)
}
var e []string
if pflag.NArg() == 0 {
e = os.Environ()
} else {
if x, err := readEnv(pflag.Arg(0)); err == nil {
e = x
} else {
fmt.Fprintf(os.Stderr, "error: read env file: %v\n", err)
os.Exit(1)
}
if v, ok := os.LookupEnv("NOTIFY_SOCKET"); ok {
e = append(e, "NOTIFY_SOCKET="+v)
}
}
if dbgAddr, _ := getEnvList("INSECURE_DEBUG_SERVER_ADDR", e, os.Environ()); dbgAddr != "" {
go func() {
fmt.Fprintf(os.Stderr, "warning: running insecure debug server on %q\n", dbgAddr)
if err := runDebugServer(dbgAddr); err != nil {
fmt.Fprintf(os.Stderr, "warning: failed to start debug server: %v\n", err)
}
}()
}
var c atlas.Config
if err := c.UnmarshalEnv(e, false); err != nil {
fmt.Fprintf(os.Stderr, "error: parse config: %v\n", err)
os.Exit(1)
}
s, err := atlas.NewServer(&c)
if err != nil {
fmt.Fprintf(os.Stderr, "error: initialize server: %v\n", err)
os.Exit(1)
}
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer stop()
hch := make(chan os.Signal, 1)
signal.Notify(hch, syscall.SIGHUP)
go func() {
for range hch {
fmt.Println("got SIGHUP")
s.HandleSIGHUP()
}
}()
if err := s.Run(ctx); err != nil && !errors.Is(err, context.Canceled) {
fmt.Fprintf(os.Stderr, "error: run server: %v\n", err)
os.Exit(1)
}
}
func runDebugServer(addr string) error {
m := http.NewServeMux()
m.HandleFunc("/debug/pprof/", pprof.Index)
m.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
m.HandleFunc("/debug/pprof/profile", pprof.Profile)
m.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
m.HandleFunc("/debug/pprof/trace", pprof.Trace)
return http.ListenAndServe(addr, m)
}
func getEnvList(k string, e ...[]string) (string, bool) {
for _, l := range e {
for _, x := range l {
if xk, xv, ok := strings.Cut(x, "="); ok && xk == k {
return xv, true
}
}
}
return "", false
}
func readEnv(name string) ([]string, error) {
f, err := os.Open(name)
if err != nil {
return nil, err
}
defer f.Close()
m, err := envparse.Parse(f)
if err != nil {
return nil, err
}
var r []string
for k, v := range m {
r = append(r, k+"="+v)
}
return r, nil
}
|
