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
135
136
137
138
139
140
141
142
143
144
145
146
147
|
package main
import (
"errors"
"log"
"net/url"
"os"
"strings"
"github.com/namsral/flag"
)
// 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")
artifactsServer = flag.String("artifacts-server", "", "API URL to proxy artifact requests to, e.g.: 'https://gitlab.com/api/v4'")
artifactsServerTimeout = flag.Int("artifacts-server-timeout", 10, "Timeout (in seconds) for a proxied request to the artifacts server")
pagesStatus = flag.String("pages-status", "", "The url path for a status page, e.g., /@status")
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")
disableCrossOriginRequests = flag.Bool("disable-cross-origin-requests", false, "Disable cross-origin requests")
errArtifactSchemaUnsupported = errors.New("artifacts-server scheme must be either http:// or https://")
errArtifactsServerTimeoutValue = errors.New("artifacts-server-timeout must be greater than or equal to 1")
)
func configFromFlags() appConfig {
var config appConfig
config.Domain = strings.ToLower(*pagesDomain)
config.RedirectHTTP = *redirectHTTP
config.HTTP2 = *useHTTP2
config.DisableCrossOriginRequests = *disableCrossOriginRequests
config.StatusPath = *pagesStatus
if *pagesRootCert != "" {
config.RootCertificate = readFile(*pagesRootCert)
}
if *pagesRootKey != "" {
config.RootKey = readFile(*pagesRootKey)
}
if *artifactsServerTimeout < 1 {
log.Fatal(errArtifactsServerTimeoutValue)
}
if *artifactsServer != "" {
u, err := url.Parse(*artifactsServer)
if err != nil {
log.Fatal(err)
}
// url.Parse ensures that the Scheme arttribute is always lower case.
if u.Scheme != "http" && u.Scheme != "https" {
log.Fatal(errArtifactSchemaUnsupported)
}
if *artifactsServerTimeout < 1 {
log.Fatal(errArtifactsServerTimeoutValue)
}
config.ArtifactsServerTimeout = *artifactsServerTimeout
config.ArtifactsServer = *artifactsServer
}
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.String(flag.DefaultConfigFlagname, "", "path to config file")
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.Split() {
l, fd := createSocket(addr)
defer l.Close()
config.ListenHTTP = append(config.ListenHTTP, fd)
}
for _, addr := range listenHTTPS.Split() {
l, fd := createSocket(addr)
defer l.Close()
config.ListenHTTPS = append(config.ListenHTTPS, fd)
}
for _, addr := range listenProxy.Split() {
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()
}
|
