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
|
package main
import (
"flag"
"log"
"os"
"strings"
)
// VERSION stores the information about the semantic version of application
var VERSION = "dev"
// REVISION stores the information about the git revision of application
var REVISION = "HEAD"
var (
pagesRootCert = flag.String("root-cert", "", "The default path to file certificate to serve static pages")
pagesRootKey = flag.String("root-key", "", "The default path to file certificate to serve static pages")
redirectHTTP = flag.Bool("redirect-http", false, "Redirect pages from HTTP to HTTPS")
useHTTP2 = flag.Bool("use-http2", true, "Enable HTTP2 support")
pagesRoot = flag.String("pages-root", "shared/pages", "The directory where pages are stored")
pagesDomain = flag.String("pages-domain", "gitlab-example.com", "The domain to serve static pages")
metricsAddress = flag.String("metrics-address", "", "The address to listen on for metrics requests")
daemonUID = flag.Uint("daemon-uid", 0, "Drop privileges to this user")
daemonGID = flag.Uint("daemon-gid", 0, "Drop privileges to this group")
)
func configFromFlags() appConfig {
var config appConfig
config.Domain = strings.ToLower(*pagesDomain)
config.RedirectHTTP = *redirectHTTP
config.HTTP2 = *useHTTP2
if *pagesRootCert != "" {
config.RootCertificate = readFile(*pagesRootCert)
}
if *pagesRootKey != "" {
config.RootKey = readFile(*pagesRootKey)
}
return config
}
func appMain() {
var showVersion = flag.Bool("version", false, "Show version")
var listenHTTP, listenHTTPS, listenProxy MultiStringFlag
flag.Var(&listenHTTP, "listen-http", "The address(es) to listen on for HTTP requests")
flag.Var(&listenHTTPS, "listen-https", "The address(es) to listen on for HTTPS requests")
flag.Var(&listenProxy, "listen-proxy", "The address(es) to listen on for proxy requests")
flag.Parse()
printVersion(*showVersion, VERSION)
log.Printf("GitLab Pages Daemon %s (%s)", VERSION, REVISION)
log.Printf("URL: https://gitlab.com/gitlab-org/gitlab-pages\n")
err := os.Chdir(*pagesRoot)
if err != nil {
log.Fatalln(err)
}
config := configFromFlags()
for _, addr := range listenHTTP {
l, fd := createSocket(addr)
defer l.Close()
config.ListenHTTP = append(config.ListenHTTP, fd)
}
for _, addr := range listenHTTPS {
l, fd := createSocket(addr)
defer l.Close()
config.ListenHTTPS = append(config.ListenHTTPS, fd)
}
for _, addr := range listenProxy {
l, fd := createSocket(addr)
defer l.Close()
config.ListenProxy = append(config.ListenProxy, fd)
}
if *metricsAddress != "" {
l, fd := createSocket(*metricsAddress)
defer l.Close()
config.ListenMetrics = fd
}
if *daemonUID != 0 || *daemonGID != 0 {
daemonize(config, *daemonUID, *daemonGID)
return
}
runApp(config)
}
func printVersion(showVersion bool, version string) {
if showVersion {
log.SetFlags(0)
log.Printf(version)
os.Exit(0)
}
}
func main() {
log.SetOutput(os.Stderr)
daemonMain()
appMain()
}
|