serve.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378
  1. //go:build !noserver
  2. package cmd
  3. import (
  4. "errors"
  5. "fmt"
  6. "github.com/stripe/stripe-go/v74"
  7. "heckel.io/ntfy/user"
  8. "io/fs"
  9. "math"
  10. "net"
  11. "net/netip"
  12. "os"
  13. "os/signal"
  14. "strings"
  15. "syscall"
  16. "time"
  17. "heckel.io/ntfy/log"
  18. "github.com/urfave/cli/v2"
  19. "github.com/urfave/cli/v2/altsrc"
  20. "heckel.io/ntfy/server"
  21. "heckel.io/ntfy/util"
  22. )
  23. func init() {
  24. commands = append(commands, cmdServe)
  25. }
  26. const (
  27. defaultServerConfigFile = "/etc/ntfy/server.yml"
  28. )
  29. var flagsServe = append(
  30. flagsDefault,
  31. &cli.StringFlag{Name: "config", Aliases: []string{"c"}, EnvVars: []string{"NTFY_CONFIG_FILE"}, Value: defaultServerConfigFile, DefaultText: defaultServerConfigFile, Usage: "config file"},
  32. altsrc.NewStringFlag(&cli.StringFlag{Name: "base-url", Aliases: []string{"base_url", "B"}, EnvVars: []string{"NTFY_BASE_URL"}, Usage: "externally visible base URL for this host (e.g. https://ntfy.sh)"}),
  33. altsrc.NewStringFlag(&cli.StringFlag{Name: "listen-http", Aliases: []string{"listen_http", "l"}, EnvVars: []string{"NTFY_LISTEN_HTTP"}, Value: server.DefaultListenHTTP, Usage: "ip:port used to as HTTP listen address"}),
  34. altsrc.NewStringFlag(&cli.StringFlag{Name: "listen-https", Aliases: []string{"listen_https", "L"}, EnvVars: []string{"NTFY_LISTEN_HTTPS"}, Usage: "ip:port used to as HTTPS listen address"}),
  35. altsrc.NewStringFlag(&cli.StringFlag{Name: "listen-unix", Aliases: []string{"listen_unix", "U"}, EnvVars: []string{"NTFY_LISTEN_UNIX"}, Usage: "listen on unix socket path"}),
  36. altsrc.NewIntFlag(&cli.IntFlag{Name: "listen-unix-mode", Aliases: []string{"listen_unix_mode"}, EnvVars: []string{"NTFY_LISTEN_UNIX_MODE"}, DefaultText: "system default", Usage: "file permissions of unix socket, e.g. 0700"}),
  37. altsrc.NewStringFlag(&cli.StringFlag{Name: "key-file", Aliases: []string{"key_file", "K"}, EnvVars: []string{"NTFY_KEY_FILE"}, Usage: "private key file, if listen-https is set"}),
  38. altsrc.NewStringFlag(&cli.StringFlag{Name: "cert-file", Aliases: []string{"cert_file", "E"}, EnvVars: []string{"NTFY_CERT_FILE"}, Usage: "certificate file, if listen-https is set"}),
  39. altsrc.NewStringFlag(&cli.StringFlag{Name: "firebase-key-file", Aliases: []string{"firebase_key_file", "F"}, EnvVars: []string{"NTFY_FIREBASE_KEY_FILE"}, Usage: "Firebase credentials file; if set additionally publish to FCM topic"}),
  40. altsrc.NewStringFlag(&cli.StringFlag{Name: "cache-file", Aliases: []string{"cache_file", "C"}, EnvVars: []string{"NTFY_CACHE_FILE"}, Usage: "cache file used for message caching"}),
  41. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "cache-duration", Aliases: []string{"cache_duration", "b"}, EnvVars: []string{"NTFY_CACHE_DURATION"}, Value: server.DefaultCacheDuration, Usage: "buffer messages for this time to allow `since` requests"}),
  42. altsrc.NewIntFlag(&cli.IntFlag{Name: "cache-batch-size", Aliases: []string{"cache_batch_size"}, EnvVars: []string{"NTFY_BATCH_SIZE"}, Usage: "max size of messages to batch together when writing to message cache (if zero, writes are synchronous)"}),
  43. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "cache-batch-timeout", Aliases: []string{"cache_batch_timeout"}, EnvVars: []string{"NTFY_CACHE_BATCH_TIMEOUT"}, Usage: "timeout for batched async writes to the message cache (if zero, writes are synchronous)"}),
  44. altsrc.NewStringFlag(&cli.StringFlag{Name: "cache-startup-queries", Aliases: []string{"cache_startup_queries"}, EnvVars: []string{"NTFY_CACHE_STARTUP_QUERIES"}, Usage: "queries run when the cache database is initialized"}),
  45. altsrc.NewStringFlag(&cli.StringFlag{Name: "auth-file", Aliases: []string{"auth_file", "H"}, EnvVars: []string{"NTFY_AUTH_FILE"}, Usage: "auth database file used for access control"}),
  46. altsrc.NewStringFlag(&cli.StringFlag{Name: "auth-startup-queries", Aliases: []string{"auth_startup_queries"}, EnvVars: []string{"NTFY_AUTH_STARTUP_QUERIES"}, Usage: "queries run when the auth database is initialized"}),
  47. altsrc.NewStringFlag(&cli.StringFlag{Name: "auth-default-access", Aliases: []string{"auth_default_access", "p"}, EnvVars: []string{"NTFY_AUTH_DEFAULT_ACCESS"}, Value: "read-write", Usage: "default permissions if no matching entries in the auth database are found"}),
  48. altsrc.NewStringFlag(&cli.StringFlag{Name: "attachment-cache-dir", Aliases: []string{"attachment_cache_dir"}, EnvVars: []string{"NTFY_ATTACHMENT_CACHE_DIR"}, Usage: "cache directory for attached files"}),
  49. altsrc.NewStringFlag(&cli.StringFlag{Name: "attachment-total-size-limit", Aliases: []string{"attachment_total_size_limit", "A"}, EnvVars: []string{"NTFY_ATTACHMENT_TOTAL_SIZE_LIMIT"}, DefaultText: "5G", Usage: "limit of the on-disk attachment cache"}),
  50. altsrc.NewStringFlag(&cli.StringFlag{Name: "attachment-file-size-limit", Aliases: []string{"attachment_file_size_limit", "Y"}, EnvVars: []string{"NTFY_ATTACHMENT_FILE_SIZE_LIMIT"}, DefaultText: "15M", Usage: "per-file attachment size limit (e.g. 300k, 2M, 100M)"}),
  51. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "attachment-expiry-duration", Aliases: []string{"attachment_expiry_duration", "X"}, EnvVars: []string{"NTFY_ATTACHMENT_EXPIRY_DURATION"}, Value: server.DefaultAttachmentExpiryDuration, DefaultText: "3h", Usage: "duration after which uploaded attachments will be deleted (e.g. 3h, 20h)"}),
  52. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "keepalive-interval", Aliases: []string{"keepalive_interval", "k"}, EnvVars: []string{"NTFY_KEEPALIVE_INTERVAL"}, Value: server.DefaultKeepaliveInterval, Usage: "interval of keepalive messages"}),
  53. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "manager-interval", Aliases: []string{"manager_interval", "m"}, EnvVars: []string{"NTFY_MANAGER_INTERVAL"}, Value: server.DefaultManagerInterval, Usage: "interval of for message pruning and stats printing"}),
  54. altsrc.NewStringFlag(&cli.StringFlag{Name: "web-root", Aliases: []string{"web_root"}, EnvVars: []string{"NTFY_WEB_ROOT"}, Value: "app", Usage: "sets web root to landing page (home), web app (app) or disabled (disable)"}),
  55. altsrc.NewBoolFlag(&cli.BoolFlag{Name: "enable-signup", Aliases: []string{"enable_signup"}, EnvVars: []string{"NTFY_ENABLE_SIGNUP"}, Value: false, Usage: "allows users to sign up via the web app, or API"}),
  56. altsrc.NewBoolFlag(&cli.BoolFlag{Name: "enable-login", Aliases: []string{"enable_login"}, EnvVars: []string{"NTFY_ENABLE_LOGIN"}, Value: false, Usage: "allows users to log in via the web app, or API"}),
  57. altsrc.NewBoolFlag(&cli.BoolFlag{Name: "enable-reservations", Aliases: []string{"enable_reservations"}, EnvVars: []string{"NTFY_ENABLE_RESERVATIONS"}, Value: false, Usage: "allows users to reserve topics (if their tier allows it)"}),
  58. altsrc.NewStringFlag(&cli.StringFlag{Name: "upstream-base-url", Aliases: []string{"upstream_base_url"}, EnvVars: []string{"NTFY_UPSTREAM_BASE_URL"}, Value: "", Usage: "forward poll request to an upstream server, this is needed for iOS push notifications for self-hosted servers"}),
  59. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-sender-addr", Aliases: []string{"smtp_sender_addr"}, EnvVars: []string{"NTFY_SMTP_SENDER_ADDR"}, Usage: "SMTP server address (host:port) for outgoing emails"}),
  60. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-sender-user", Aliases: []string{"smtp_sender_user"}, EnvVars: []string{"NTFY_SMTP_SENDER_USER"}, Usage: "SMTP user (if e-mail sending is enabled)"}),
  61. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-sender-pass", Aliases: []string{"smtp_sender_pass"}, EnvVars: []string{"NTFY_SMTP_SENDER_PASS"}, Usage: "SMTP password (if e-mail sending is enabled)"}),
  62. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-sender-from", Aliases: []string{"smtp_sender_from"}, EnvVars: []string{"NTFY_SMTP_SENDER_FROM"}, Usage: "SMTP sender address (if e-mail sending is enabled)"}),
  63. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-server-listen", Aliases: []string{"smtp_server_listen"}, EnvVars: []string{"NTFY_SMTP_SERVER_LISTEN"}, Usage: "SMTP server address (ip:port) for incoming emails, e.g. :25"}),
  64. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-server-domain", Aliases: []string{"smtp_server_domain"}, EnvVars: []string{"NTFY_SMTP_SERVER_DOMAIN"}, Usage: "SMTP domain for incoming e-mail, e.g. ntfy.sh"}),
  65. altsrc.NewStringFlag(&cli.StringFlag{Name: "smtp-server-addr-prefix", Aliases: []string{"smtp_server_addr_prefix"}, EnvVars: []string{"NTFY_SMTP_SERVER_ADDR_PREFIX"}, Usage: "SMTP email address prefix for topics to prevent spam (e.g. 'ntfy-')"}),
  66. altsrc.NewIntFlag(&cli.IntFlag{Name: "global-topic-limit", Aliases: []string{"global_topic_limit", "T"}, EnvVars: []string{"NTFY_GLOBAL_TOPIC_LIMIT"}, Value: server.DefaultTotalTopicLimit, Usage: "total number of topics allowed"}),
  67. altsrc.NewIntFlag(&cli.IntFlag{Name: "visitor-subscription-limit", Aliases: []string{"visitor_subscription_limit"}, EnvVars: []string{"NTFY_VISITOR_SUBSCRIPTION_LIMIT"}, Value: server.DefaultVisitorSubscriptionLimit, Usage: "number of subscriptions per visitor"}),
  68. altsrc.NewStringFlag(&cli.StringFlag{Name: "visitor-attachment-total-size-limit", Aliases: []string{"visitor_attachment_total_size_limit"}, EnvVars: []string{"NTFY_VISITOR_ATTACHMENT_TOTAL_SIZE_LIMIT"}, Value: "100M", Usage: "total storage limit used for attachments per visitor"}),
  69. altsrc.NewStringFlag(&cli.StringFlag{Name: "visitor-attachment-daily-bandwidth-limit", Aliases: []string{"visitor_attachment_daily_bandwidth_limit"}, EnvVars: []string{"NTFY_VISITOR_ATTACHMENT_DAILY_BANDWIDTH_LIMIT"}, Value: "500M", Usage: "total daily attachment download/upload bandwidth limit per visitor"}),
  70. altsrc.NewIntFlag(&cli.IntFlag{Name: "visitor-request-limit-burst", Aliases: []string{"visitor_request_limit_burst"}, EnvVars: []string{"NTFY_VISITOR_REQUEST_LIMIT_BURST"}, Value: server.DefaultVisitorRequestLimitBurst, Usage: "initial limit of requests per visitor"}),
  71. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "visitor-request-limit-replenish", Aliases: []string{"visitor_request_limit_replenish"}, EnvVars: []string{"NTFY_VISITOR_REQUEST_LIMIT_REPLENISH"}, Value: server.DefaultVisitorRequestLimitReplenish, Usage: "interval at which burst limit is replenished (one per x)"}),
  72. altsrc.NewStringFlag(&cli.StringFlag{Name: "visitor-request-limit-exempt-hosts", Aliases: []string{"visitor_request_limit_exempt_hosts"}, EnvVars: []string{"NTFY_VISITOR_REQUEST_LIMIT_EXEMPT_HOSTS"}, Value: "", Usage: "hostnames and/or IP addresses of hosts that will be exempt from the visitor request limit"}),
  73. altsrc.NewIntFlag(&cli.IntFlag{Name: "visitor-email-limit-burst", Aliases: []string{"visitor_email_limit_burst"}, EnvVars: []string{"NTFY_VISITOR_EMAIL_LIMIT_BURST"}, Value: server.DefaultVisitorEmailLimitBurst, Usage: "initial limit of e-mails per visitor"}),
  74. altsrc.NewDurationFlag(&cli.DurationFlag{Name: "visitor-email-limit-replenish", Aliases: []string{"visitor_email_limit_replenish"}, EnvVars: []string{"NTFY_VISITOR_EMAIL_LIMIT_REPLENISH"}, Value: server.DefaultVisitorEmailLimitReplenish, Usage: "interval at which burst limit is replenished (one per x)"}),
  75. altsrc.NewBoolFlag(&cli.BoolFlag{Name: "behind-proxy", Aliases: []string{"behind_proxy", "P"}, EnvVars: []string{"NTFY_BEHIND_PROXY"}, Value: false, Usage: "if set, use X-Forwarded-For header to determine visitor IP address (for rate limiting)"}),
  76. altsrc.NewStringFlag(&cli.StringFlag{Name: "stripe-key", Aliases: []string{"stripe_key"}, EnvVars: []string{"NTFY_STRIPE_KEY"}, Value: "", Usage: "xxxxxxxxxxxxx"}),
  77. altsrc.NewStringFlag(&cli.StringFlag{Name: "stripe-webhook-key", Aliases: []string{"stripe_webhook_key"}, EnvVars: []string{"NTFY_STRIPE_WEBHOOK_KEY"}, Value: "", Usage: "xxxxxxxxxxxx"}),
  78. )
  79. var cmdServe = &cli.Command{
  80. Name: "serve",
  81. Usage: "Run the ntfy server",
  82. UsageText: "ntfy serve [OPTIONS..]",
  83. Action: execServe,
  84. Category: categoryServer,
  85. Flags: flagsServe,
  86. Before: initConfigFileInputSourceFunc("config", flagsServe, initLogFunc),
  87. Description: `Run the ntfy server and listen for incoming requests
  88. The command will load the configuration from /etc/ntfy/server.yml. Config options can
  89. be overridden using the command line options.
  90. Examples:
  91. ntfy serve # Starts server in the foreground (on port 80)
  92. ntfy serve --listen-http :8080 # Starts server with alternate port`,
  93. }
  94. func execServe(c *cli.Context) error {
  95. if c.NArg() > 0 {
  96. return errors.New("no arguments expected, see 'ntfy serve --help' for help")
  97. }
  98. // Read all the options
  99. config := c.String("config")
  100. baseURL := c.String("base-url")
  101. listenHTTP := c.String("listen-http")
  102. listenHTTPS := c.String("listen-https")
  103. listenUnix := c.String("listen-unix")
  104. listenUnixMode := c.Int("listen-unix-mode")
  105. keyFile := c.String("key-file")
  106. certFile := c.String("cert-file")
  107. firebaseKeyFile := c.String("firebase-key-file")
  108. cacheFile := c.String("cache-file")
  109. cacheDuration := c.Duration("cache-duration")
  110. cacheStartupQueries := c.String("cache-startup-queries")
  111. cacheBatchSize := c.Int("cache-batch-size")
  112. cacheBatchTimeout := c.Duration("cache-batch-timeout")
  113. authFile := c.String("auth-file")
  114. authStartupQueries := c.String("auth-startup-queries")
  115. authDefaultAccess := c.String("auth-default-access")
  116. attachmentCacheDir := c.String("attachment-cache-dir")
  117. attachmentTotalSizeLimitStr := c.String("attachment-total-size-limit")
  118. attachmentFileSizeLimitStr := c.String("attachment-file-size-limit")
  119. attachmentExpiryDuration := c.Duration("attachment-expiry-duration")
  120. keepaliveInterval := c.Duration("keepalive-interval")
  121. managerInterval := c.Duration("manager-interval")
  122. webRoot := c.String("web-root")
  123. enableSignup := c.Bool("enable-signup")
  124. enableLogin := c.Bool("enable-login")
  125. enableReservations := c.Bool("enable-reservations")
  126. upstreamBaseURL := c.String("upstream-base-url")
  127. smtpSenderAddr := c.String("smtp-sender-addr")
  128. smtpSenderUser := c.String("smtp-sender-user")
  129. smtpSenderPass := c.String("smtp-sender-pass")
  130. smtpSenderFrom := c.String("smtp-sender-from")
  131. smtpServerListen := c.String("smtp-server-listen")
  132. smtpServerDomain := c.String("smtp-server-domain")
  133. smtpServerAddrPrefix := c.String("smtp-server-addr-prefix")
  134. totalTopicLimit := c.Int("global-topic-limit")
  135. visitorSubscriptionLimit := c.Int("visitor-subscription-limit")
  136. visitorAttachmentTotalSizeLimitStr := c.String("visitor-attachment-total-size-limit")
  137. visitorAttachmentDailyBandwidthLimitStr := c.String("visitor-attachment-daily-bandwidth-limit")
  138. visitorRequestLimitBurst := c.Int("visitor-request-limit-burst")
  139. visitorRequestLimitReplenish := c.Duration("visitor-request-limit-replenish")
  140. visitorRequestLimitExemptHosts := util.SplitNoEmpty(c.String("visitor-request-limit-exempt-hosts"), ",")
  141. visitorEmailLimitBurst := c.Int("visitor-email-limit-burst")
  142. visitorEmailLimitReplenish := c.Duration("visitor-email-limit-replenish")
  143. behindProxy := c.Bool("behind-proxy")
  144. stripeKey := c.String("stripe-key")
  145. stripeWebhookKey := c.String("stripe-webhook-key")
  146. // Check values
  147. if firebaseKeyFile != "" && !util.FileExists(firebaseKeyFile) {
  148. return errors.New("if set, FCM key file must exist")
  149. } else if keepaliveInterval < 5*time.Second {
  150. return errors.New("keepalive interval cannot be lower than five seconds")
  151. } else if managerInterval < 5*time.Second {
  152. return errors.New("manager interval cannot be lower than five seconds")
  153. } else if cacheDuration > 0 && cacheDuration < managerInterval {
  154. return errors.New("cache duration cannot be lower than manager interval")
  155. } else if keyFile != "" && !util.FileExists(keyFile) {
  156. return errors.New("if set, key file must exist")
  157. } else if certFile != "" && !util.FileExists(certFile) {
  158. return errors.New("if set, certificate file must exist")
  159. } else if listenHTTPS != "" && (keyFile == "" || certFile == "") {
  160. return errors.New("if listen-https is set, both key-file and cert-file must be set")
  161. } else if smtpSenderAddr != "" && (baseURL == "" || smtpSenderUser == "" || smtpSenderPass == "" || smtpSenderFrom == "") {
  162. return errors.New("if smtp-sender-addr is set, base-url, smtp-sender-user, smtp-sender-pass and smtp-sender-from must also be set")
  163. } else if smtpServerListen != "" && smtpServerDomain == "" {
  164. return errors.New("if smtp-server-listen is set, smtp-server-domain must also be set")
  165. } else if attachmentCacheDir != "" && baseURL == "" {
  166. return errors.New("if attachment-cache-dir is set, base-url must also be set")
  167. } else if baseURL != "" && !strings.HasPrefix(baseURL, "http://") && !strings.HasPrefix(baseURL, "https://") {
  168. return errors.New("if set, base-url must start with http:// or https://")
  169. } else if baseURL != "" && strings.HasSuffix(baseURL, "/") {
  170. return errors.New("if set, base-url must not end with a slash (/)")
  171. } else if !util.Contains([]string{"app", "home", "disable"}, webRoot) {
  172. return errors.New("if set, web-root must be 'home' or 'app'")
  173. } else if upstreamBaseURL != "" && !strings.HasPrefix(upstreamBaseURL, "http://") && !strings.HasPrefix(upstreamBaseURL, "https://") {
  174. return errors.New("if set, upstream-base-url must start with http:// or https://")
  175. } else if upstreamBaseURL != "" && strings.HasSuffix(upstreamBaseURL, "/") {
  176. return errors.New("if set, upstream-base-url must not end with a slash (/)")
  177. } else if upstreamBaseURL != "" && baseURL == "" {
  178. return errors.New("if upstream-base-url is set, base-url must also be set")
  179. } else if upstreamBaseURL != "" && baseURL != "" && baseURL == upstreamBaseURL {
  180. return errors.New("base-url and upstream-base-url cannot be identical, you'll likely want to set upstream-base-url to https://ntfy.sh, see https://ntfy.sh/docs/config/#ios-instant-notifications")
  181. } else if authFile == "" && (enableSignup || enableLogin || enableReservations || stripeKey != "") {
  182. return errors.New("cannot set enable-signup, enable-login, enable-reserve-topics, or stripe-key if auth-file is not set")
  183. } else if enableSignup && !enableLogin {
  184. return errors.New("cannot set enable-signup without also setting enable-login")
  185. } else if stripeKey != "" && (stripeWebhookKey == "" || baseURL == "") {
  186. return errors.New("if stripe-key is set, stripe-webhook-key and base-url must also be set")
  187. }
  188. webRootIsApp := webRoot == "app"
  189. enableWeb := webRoot != "disable"
  190. enablePayments := stripeKey != ""
  191. // Default auth permissions
  192. authDefault, err := user.ParsePermission(authDefaultAccess)
  193. if err != nil {
  194. return errors.New("if set, auth-default-access must start set to 'read-write', 'read-only', 'write-only' or 'deny-all'")
  195. }
  196. // Special case: Unset default
  197. if listenHTTP == "-" {
  198. listenHTTP = ""
  199. }
  200. // Convert sizes to bytes
  201. attachmentTotalSizeLimit, err := parseSize(attachmentTotalSizeLimitStr, server.DefaultAttachmentTotalSizeLimit)
  202. if err != nil {
  203. return err
  204. }
  205. attachmentFileSizeLimit, err := parseSize(attachmentFileSizeLimitStr, server.DefaultAttachmentFileSizeLimit)
  206. if err != nil {
  207. return err
  208. }
  209. visitorAttachmentTotalSizeLimit, err := parseSize(visitorAttachmentTotalSizeLimitStr, server.DefaultVisitorAttachmentTotalSizeLimit)
  210. if err != nil {
  211. return err
  212. }
  213. visitorAttachmentDailyBandwidthLimit, err := parseSize(visitorAttachmentDailyBandwidthLimitStr, server.DefaultVisitorAttachmentDailyBandwidthLimit)
  214. if err != nil {
  215. return err
  216. } else if visitorAttachmentDailyBandwidthLimit > math.MaxInt {
  217. return fmt.Errorf("config option visitor-attachment-daily-bandwidth-limit must be lower than %d", math.MaxInt)
  218. }
  219. // Resolve hosts
  220. visitorRequestLimitExemptIPs := make([]netip.Prefix, 0)
  221. for _, host := range visitorRequestLimitExemptHosts {
  222. ips, err := parseIPHostPrefix(host)
  223. if err != nil {
  224. log.Warn("cannot resolve host %s: %s, ignoring visitor request exemption", host, err.Error())
  225. continue
  226. }
  227. visitorRequestLimitExemptIPs = append(visitorRequestLimitExemptIPs, ips...)
  228. }
  229. // Stripe things
  230. if stripeKey != "" {
  231. stripe.Key = stripeKey
  232. }
  233. // Run server
  234. conf := server.NewConfig()
  235. conf.BaseURL = baseURL
  236. conf.ListenHTTP = listenHTTP
  237. conf.ListenHTTPS = listenHTTPS
  238. conf.ListenUnix = listenUnix
  239. conf.ListenUnixMode = fs.FileMode(listenUnixMode)
  240. conf.KeyFile = keyFile
  241. conf.CertFile = certFile
  242. conf.FirebaseKeyFile = firebaseKeyFile
  243. conf.CacheFile = cacheFile
  244. conf.CacheDuration = cacheDuration
  245. conf.CacheStartupQueries = cacheStartupQueries
  246. conf.CacheBatchSize = cacheBatchSize
  247. conf.CacheBatchTimeout = cacheBatchTimeout
  248. conf.AuthFile = authFile
  249. conf.AuthStartupQueries = authStartupQueries
  250. conf.AuthDefault = authDefault
  251. conf.AttachmentCacheDir = attachmentCacheDir
  252. conf.AttachmentTotalSizeLimit = attachmentTotalSizeLimit
  253. conf.AttachmentFileSizeLimit = attachmentFileSizeLimit
  254. conf.AttachmentExpiryDuration = attachmentExpiryDuration
  255. conf.KeepaliveInterval = keepaliveInterval
  256. conf.ManagerInterval = managerInterval
  257. conf.WebRootIsApp = webRootIsApp
  258. conf.UpstreamBaseURL = upstreamBaseURL
  259. conf.SMTPSenderAddr = smtpSenderAddr
  260. conf.SMTPSenderUser = smtpSenderUser
  261. conf.SMTPSenderPass = smtpSenderPass
  262. conf.SMTPSenderFrom = smtpSenderFrom
  263. conf.SMTPServerListen = smtpServerListen
  264. conf.SMTPServerDomain = smtpServerDomain
  265. conf.SMTPServerAddrPrefix = smtpServerAddrPrefix
  266. conf.TotalTopicLimit = totalTopicLimit
  267. conf.VisitorSubscriptionLimit = visitorSubscriptionLimit
  268. conf.VisitorAttachmentTotalSizeLimit = visitorAttachmentTotalSizeLimit
  269. conf.VisitorAttachmentDailyBandwidthLimit = int(visitorAttachmentDailyBandwidthLimit)
  270. conf.VisitorRequestLimitBurst = visitorRequestLimitBurst
  271. conf.VisitorRequestLimitReplenish = visitorRequestLimitReplenish
  272. conf.VisitorRequestExemptIPAddrs = visitorRequestLimitExemptIPs
  273. conf.VisitorEmailLimitBurst = visitorEmailLimitBurst
  274. conf.VisitorEmailLimitReplenish = visitorEmailLimitReplenish
  275. conf.BehindProxy = behindProxy
  276. conf.StripeKey = stripeKey
  277. conf.StripeWebhookKey = stripeWebhookKey
  278. conf.EnableWeb = enableWeb
  279. conf.EnableSignup = enableSignup
  280. conf.EnableLogin = enableLogin
  281. conf.EnablePayments = enablePayments
  282. conf.EnableReservations = enableReservations
  283. conf.Version = c.App.Version
  284. // Set up hot-reloading of config
  285. go sigHandlerConfigReload(config)
  286. // Run server
  287. s, err := server.New(conf)
  288. if err != nil {
  289. log.Fatal(err)
  290. } else if err := s.Run(); err != nil {
  291. log.Fatal(err)
  292. }
  293. log.Info("Exiting.")
  294. return nil
  295. }
  296. func parseSize(s string, defaultValue int64) (v int64, err error) {
  297. if s == "" {
  298. return defaultValue, nil
  299. }
  300. v, err = util.ParseSize(s)
  301. if err != nil {
  302. return 0, err
  303. }
  304. return v, nil
  305. }
  306. func sigHandlerConfigReload(config string) {
  307. sigs := make(chan os.Signal, 1)
  308. signal.Notify(sigs, syscall.SIGHUP)
  309. for range sigs {
  310. log.Info("Partially hot reloading configuration ...")
  311. inputSource, err := newYamlSourceFromFile(config, flagsServe)
  312. if err != nil {
  313. log.Warn("Hot reload failed: %s", err.Error())
  314. continue
  315. }
  316. reloadLogLevel(inputSource)
  317. }
  318. }
  319. func parseIPHostPrefix(host string) (prefixes []netip.Prefix, err error) {
  320. // Try parsing as prefix, e.g. 10.0.1.0/24
  321. prefix, err := netip.ParsePrefix(host)
  322. if err == nil {
  323. prefixes = append(prefixes, prefix.Masked())
  324. return prefixes, nil
  325. }
  326. // Not a prefix, parse as host or IP (LookupHost passes through an IP as is)
  327. ips, err := net.LookupHost(host)
  328. if err != nil {
  329. return nil, err
  330. }
  331. for _, ipStr := range ips {
  332. ip, err := netip.ParseAddr(ipStr)
  333. if err == nil {
  334. prefix, err := ip.Prefix(ip.BitLen())
  335. if err != nil {
  336. return nil, fmt.Errorf("%s successfully parsed but unable to make prefix: %s", ip.String(), err.Error())
  337. }
  338. prefixes = append(prefixes, prefix.Masked())
  339. }
  340. }
  341. return
  342. }
  343. func reloadLogLevel(inputSource altsrc.InputSourceContext) {
  344. newLevelStr, err := inputSource.String("log-level")
  345. if err != nil {
  346. log.Warn("Cannot load log level: %s", err.Error())
  347. return
  348. }
  349. newLevel := log.ToLevel(newLevelStr)
  350. log.SetLevel(newLevel)
  351. log.Info("Log level is %s", newLevel.String())
  352. }