1
0

server.go 54 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627
  1. package server
  2. import (
  3. "bytes"
  4. "context"
  5. "crypto/sha256"
  6. "embed"
  7. "encoding/base64"
  8. "encoding/json"
  9. "errors"
  10. "fmt"
  11. "heckel.io/ntfy/user"
  12. "io"
  13. "net"
  14. "net/http"
  15. "net/netip"
  16. "net/url"
  17. "os"
  18. "path"
  19. "path/filepath"
  20. "regexp"
  21. "sort"
  22. "strconv"
  23. "strings"
  24. "sync"
  25. "time"
  26. "unicode/utf8"
  27. "heckel.io/ntfy/log"
  28. "github.com/emersion/go-smtp"
  29. "github.com/gorilla/websocket"
  30. "golang.org/x/sync/errgroup"
  31. "heckel.io/ntfy/util"
  32. )
  33. /*
  34. TODO
  35. Limits & rate limiting:
  36. login/account endpoints
  37. purge accounts that were not logged int o in X
  38. reset daily Limits for users
  39. Make sure account endpoints make sense for admins
  40. UI:
  41. - Align size of message bar and upgrade banner
  42. - flicker of upgrade banner
  43. - JS constants
  44. - useContext for account
  45. Sync:
  46. - "account topic" sync mechanism
  47. - "mute" setting
  48. - figure out what settings are "web" or "phone"
  49. Delete visitor when tier is changed to refresh rate limiters
  50. Tests:
  51. - Change tier from higher to lower tier (delete reservations)
  52. - Message rate limiting and reset tests
  53. Docs:
  54. - "expires" field in message
  55. Refactor:
  56. - rename /access -> /reservation
  57. Later:
  58. - Password reset
  59. - Pricing
  60. - change email
  61. */
  62. // Server is the main server, providing the UI and API for ntfy
  63. type Server struct {
  64. config *Config
  65. httpServer *http.Server
  66. httpsServer *http.Server
  67. unixListener net.Listener
  68. smtpServer *smtp.Server
  69. smtpServerBackend *smtpBackend
  70. smtpSender mailer
  71. topics map[string]*topic
  72. visitors map[string]*visitor // ip:<ip> or user:<user>
  73. firebaseClient *firebaseClient
  74. messages int64
  75. userManager *user.Manager // Might be nil!
  76. messageCache *messageCache
  77. fileCache *fileCache
  78. closeChan chan bool
  79. mu sync.Mutex
  80. }
  81. // handleFunc extends the normal http.HandlerFunc to be able to easily return errors
  82. type handleFunc func(http.ResponseWriter, *http.Request, *visitor) error
  83. var (
  84. // If changed, don't forget to update Android App and auth_sqlite.go
  85. topicRegex = regexp.MustCompile(`^[-_A-Za-z0-9]{1,64}$`) // No /!
  86. topicPathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}$`) // Regex must match JS & Android app!
  87. externalTopicPathRegex = regexp.MustCompile(`^/[^/]+\.[^/]+/[-_A-Za-z0-9]{1,64}$`) // Extended topic path, for web-app, e.g. /example.com/mytopic
  88. jsonPathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}(,[-_A-Za-z0-9]{1,64})*/json$`)
  89. ssePathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}(,[-_A-Za-z0-9]{1,64})*/sse$`)
  90. rawPathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}(,[-_A-Za-z0-9]{1,64})*/raw$`)
  91. wsPathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}(,[-_A-Za-z0-9]{1,64})*/ws$`)
  92. authPathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}(,[-_A-Za-z0-9]{1,64})*/auth$`)
  93. publishPathRegex = regexp.MustCompile(`^/[-_A-Za-z0-9]{1,64}/(publish|send|trigger)$`)
  94. webConfigPath = "/config.js"
  95. healthPath = "/v1/health"
  96. accountPath = "/v1/account"
  97. accountTokenPath = "/v1/account/token"
  98. accountPasswordPath = "/v1/account/password"
  99. accountSettingsPath = "/v1/account/settings"
  100. accountSubscriptionPath = "/v1/account/subscription"
  101. accountAccessPath = "/v1/account/access"
  102. accountAccessSingleRegex = regexp.MustCompile(`/v1/account/access/([-_A-Za-z0-9]{1,64})$`)
  103. accountSubscriptionSingleRegex = regexp.MustCompile(`^/v1/account/subscription/([-_A-Za-z0-9]{16})$`)
  104. matrixPushPath = "/_matrix/push/v1/notify"
  105. staticRegex = regexp.MustCompile(`^/static/.+`)
  106. docsRegex = regexp.MustCompile(`^/docs(|/.*)$`)
  107. fileRegex = regexp.MustCompile(`^/file/([-_A-Za-z0-9]{1,64})(?:\.[A-Za-z0-9]{1,16})?$`)
  108. disallowedTopics = []string{"docs", "static", "file", "app", "account", "settings", "pricing", "signup", "login", "reset-password"} // If updated, also update in Android and web app
  109. urlRegex = regexp.MustCompile(`^https?://`)
  110. //go:embed site
  111. webFs embed.FS
  112. webFsCached = &util.CachingEmbedFS{ModTime: time.Now(), FS: webFs}
  113. webSiteDir = "/site"
  114. webHomeIndex = "/home.html" // Landing page, only if "web-root: home"
  115. webAppIndex = "/app.html" // React app
  116. //go:embed docs
  117. docsStaticFs embed.FS
  118. docsStaticCached = &util.CachingEmbedFS{ModTime: time.Now(), FS: docsStaticFs}
  119. )
  120. const (
  121. firebaseControlTopic = "~control" // See Android if changed
  122. firebasePollTopic = "~poll" // See iOS if changed
  123. emptyMessageBody = "triggered" // Used if message body is empty
  124. newMessageBody = "New message" // Used in poll requests as generic message
  125. defaultAttachmentMessage = "You received a file: %s" // Used if message body is empty, and there is an attachment
  126. encodingBase64 = "base64"
  127. )
  128. // WebSocket constants
  129. const (
  130. wsWriteWait = 2 * time.Second
  131. wsBufferSize = 1024
  132. wsReadLimit = 64 // We only ever receive PINGs
  133. wsPongWait = 15 * time.Second
  134. )
  135. // New instantiates a new Server. It creates the cache and adds a Firebase
  136. // subscriber (if configured).
  137. func New(conf *Config) (*Server, error) {
  138. var mailer mailer
  139. if conf.SMTPSenderAddr != "" {
  140. mailer = &smtpSender{config: conf}
  141. }
  142. messageCache, err := createMessageCache(conf)
  143. if err != nil {
  144. return nil, err
  145. }
  146. topics, err := messageCache.Topics()
  147. if err != nil {
  148. return nil, err
  149. }
  150. var fileCache *fileCache
  151. if conf.AttachmentCacheDir != "" {
  152. fileCache, err = newFileCache(conf.AttachmentCacheDir, conf.AttachmentTotalSizeLimit)
  153. if err != nil {
  154. return nil, err
  155. }
  156. }
  157. var userManager *user.Manager
  158. if conf.AuthFile != "" {
  159. userManager, err = user.NewManager(conf.AuthFile, conf.AuthStartupQueries, conf.AuthDefault)
  160. if err != nil {
  161. return nil, err
  162. }
  163. }
  164. var firebaseClient *firebaseClient
  165. if conf.FirebaseKeyFile != "" {
  166. sender, err := newFirebaseSender(conf.FirebaseKeyFile)
  167. if err != nil {
  168. return nil, err
  169. }
  170. firebaseClient = newFirebaseClient(sender, userManager)
  171. }
  172. return &Server{
  173. config: conf,
  174. messageCache: messageCache,
  175. fileCache: fileCache,
  176. firebaseClient: firebaseClient,
  177. smtpSender: mailer,
  178. topics: topics,
  179. userManager: userManager,
  180. visitors: make(map[string]*visitor),
  181. }, nil
  182. }
  183. func createMessageCache(conf *Config) (*messageCache, error) {
  184. if conf.CacheDuration == 0 {
  185. return newNopCache()
  186. } else if conf.CacheFile != "" {
  187. return newSqliteCache(conf.CacheFile, conf.CacheStartupQueries, conf.CacheDuration, conf.CacheBatchSize, conf.CacheBatchTimeout, false)
  188. }
  189. return newMemCache()
  190. }
  191. // Run executes the main server. It listens on HTTP (+ HTTPS, if configured), and starts
  192. // a manager go routine to print stats and prune messages.
  193. func (s *Server) Run() error {
  194. var listenStr string
  195. if s.config.ListenHTTP != "" {
  196. listenStr += fmt.Sprintf(" %s[http]", s.config.ListenHTTP)
  197. }
  198. if s.config.ListenHTTPS != "" {
  199. listenStr += fmt.Sprintf(" %s[https]", s.config.ListenHTTPS)
  200. }
  201. if s.config.ListenUnix != "" {
  202. listenStr += fmt.Sprintf(" %s[unix]", s.config.ListenUnix)
  203. }
  204. if s.config.SMTPServerListen != "" {
  205. listenStr += fmt.Sprintf(" %s[smtp]", s.config.SMTPServerListen)
  206. }
  207. log.Info("Listening on%s, ntfy %s, log level is %s", listenStr, s.config.Version, log.CurrentLevel().String())
  208. mux := http.NewServeMux()
  209. mux.HandleFunc("/", s.handle)
  210. errChan := make(chan error)
  211. s.mu.Lock()
  212. s.closeChan = make(chan bool)
  213. if s.config.ListenHTTP != "" {
  214. s.httpServer = &http.Server{Addr: s.config.ListenHTTP, Handler: mux}
  215. go func() {
  216. errChan <- s.httpServer.ListenAndServe()
  217. }()
  218. }
  219. if s.config.ListenHTTPS != "" {
  220. s.httpsServer = &http.Server{Addr: s.config.ListenHTTPS, Handler: mux}
  221. go func() {
  222. errChan <- s.httpsServer.ListenAndServeTLS(s.config.CertFile, s.config.KeyFile)
  223. }()
  224. }
  225. if s.config.ListenUnix != "" {
  226. go func() {
  227. var err error
  228. s.mu.Lock()
  229. os.Remove(s.config.ListenUnix)
  230. s.unixListener, err = net.Listen("unix", s.config.ListenUnix)
  231. if err != nil {
  232. s.mu.Unlock()
  233. errChan <- err
  234. return
  235. }
  236. defer s.unixListener.Close()
  237. if s.config.ListenUnixMode > 0 {
  238. if err := os.Chmod(s.config.ListenUnix, s.config.ListenUnixMode); err != nil {
  239. s.mu.Unlock()
  240. errChan <- err
  241. return
  242. }
  243. }
  244. s.mu.Unlock()
  245. httpServer := &http.Server{Handler: mux}
  246. errChan <- httpServer.Serve(s.unixListener)
  247. }()
  248. }
  249. if s.config.SMTPServerListen != "" {
  250. go func() {
  251. errChan <- s.runSMTPServer()
  252. }()
  253. }
  254. s.mu.Unlock()
  255. go s.runManager()
  256. go s.runDelayedSender()
  257. go s.runFirebaseKeepaliver()
  258. return <-errChan
  259. }
  260. // Stop stops HTTP (+HTTPS) server and all managers
  261. func (s *Server) Stop() {
  262. s.mu.Lock()
  263. defer s.mu.Unlock()
  264. if s.httpServer != nil {
  265. s.httpServer.Close()
  266. }
  267. if s.httpsServer != nil {
  268. s.httpsServer.Close()
  269. }
  270. if s.unixListener != nil {
  271. s.unixListener.Close()
  272. }
  273. if s.smtpServer != nil {
  274. s.smtpServer.Close()
  275. }
  276. close(s.closeChan)
  277. }
  278. func (s *Server) handle(w http.ResponseWriter, r *http.Request) {
  279. v, err := s.visitor(r) // Note: Always returns v, even when error is returned
  280. if err == nil {
  281. log.Debug("%s Dispatching request", logHTTPPrefix(v, r))
  282. if log.IsTrace() {
  283. log.Trace("%s Entire request (headers and body):\n%s", logHTTPPrefix(v, r), renderHTTPRequest(r))
  284. }
  285. err = s.handleInternal(w, r, v)
  286. }
  287. if err != nil {
  288. if websocket.IsWebSocketUpgrade(r) {
  289. isNormalError := strings.Contains(err.Error(), "i/o timeout")
  290. if isNormalError {
  291. log.Debug("%s WebSocket error (this error is okay, it happens a lot): %s", logHTTPPrefix(v, r), err.Error())
  292. } else {
  293. log.Info("%s WebSocket error: %s", logHTTPPrefix(v, r), err.Error())
  294. }
  295. return // Do not attempt to write to upgraded connection
  296. }
  297. if matrixErr, ok := err.(*errMatrix); ok {
  298. writeMatrixError(w, r, v, matrixErr)
  299. return
  300. }
  301. httpErr, ok := err.(*errHTTP)
  302. if !ok {
  303. httpErr = errHTTPInternalError
  304. }
  305. isNormalError := httpErr.HTTPCode == http.StatusNotFound || httpErr.HTTPCode == http.StatusBadRequest
  306. if isNormalError {
  307. log.Debug("%s Connection closed with HTTP %d (ntfy error %d): %s", logHTTPPrefix(v, r), httpErr.HTTPCode, httpErr.Code, err.Error())
  308. } else {
  309. log.Info("%s Connection closed with HTTP %d (ntfy error %d): %s", logHTTPPrefix(v, r), httpErr.HTTPCode, httpErr.Code, err.Error())
  310. }
  311. w.Header().Set("Content-Type", "application/json")
  312. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  313. w.WriteHeader(httpErr.HTTPCode)
  314. io.WriteString(w, httpErr.JSON()+"\n")
  315. }
  316. }
  317. func (s *Server) handleInternal(w http.ResponseWriter, r *http.Request, v *visitor) error {
  318. if r.Method == http.MethodGet && r.URL.Path == "/" {
  319. return s.ensureWebEnabled(s.handleHome)(w, r, v)
  320. } else if r.Method == http.MethodHead && r.URL.Path == "/" {
  321. return s.ensureWebEnabled(s.handleEmpty)(w, r, v)
  322. } else if r.Method == http.MethodGet && r.URL.Path == healthPath {
  323. return s.handleHealth(w, r, v)
  324. } else if r.Method == http.MethodGet && r.URL.Path == webConfigPath {
  325. return s.ensureWebEnabled(s.handleWebConfig)(w, r, v)
  326. } else if r.Method == http.MethodPost && r.URL.Path == accountPath {
  327. return s.ensureUserManager(s.handleAccountCreate)(w, r, v)
  328. } else if r.Method == http.MethodPost && r.URL.Path == accountTokenPath {
  329. return s.ensureUser(s.handleAccountTokenIssue)(w, r, v)
  330. } else if r.Method == http.MethodGet && r.URL.Path == accountPath {
  331. return s.handleAccountGet(w, r, v) // Allowed by anonymous
  332. } else if r.Method == http.MethodDelete && r.URL.Path == accountPath {
  333. return s.ensureUser(s.handleAccountDelete)(w, r, v)
  334. } else if r.Method == http.MethodPost && r.URL.Path == accountPasswordPath {
  335. return s.ensureUser(s.handleAccountPasswordChange)(w, r, v)
  336. } else if r.Method == http.MethodPatch && r.URL.Path == accountTokenPath {
  337. return s.ensureUser(s.handleAccountTokenExtend)(w, r, v)
  338. } else if r.Method == http.MethodDelete && r.URL.Path == accountTokenPath {
  339. return s.ensureUser(s.handleAccountTokenDelete)(w, r, v)
  340. } else if r.Method == http.MethodPatch && r.URL.Path == accountSettingsPath {
  341. return s.ensureUser(s.handleAccountSettingsChange)(w, r, v)
  342. } else if r.Method == http.MethodPost && r.URL.Path == accountSubscriptionPath {
  343. return s.ensureUser(s.handleAccountSubscriptionAdd)(w, r, v)
  344. } else if r.Method == http.MethodPatch && accountSubscriptionSingleRegex.MatchString(r.URL.Path) {
  345. return s.ensureUser(s.handleAccountSubscriptionChange)(w, r, v)
  346. } else if r.Method == http.MethodDelete && accountSubscriptionSingleRegex.MatchString(r.URL.Path) {
  347. return s.ensureUser(s.handleAccountSubscriptionDelete)(w, r, v)
  348. } else if r.Method == http.MethodPost && r.URL.Path == accountAccessPath {
  349. return s.ensureUser(s.handleAccountAccessAdd)(w, r, v)
  350. } else if r.Method == http.MethodDelete && accountAccessSingleRegex.MatchString(r.URL.Path) {
  351. return s.ensureUser(s.handleAccountAccessDelete)(w, r, v)
  352. } else if r.Method == http.MethodGet && r.URL.Path == matrixPushPath {
  353. return s.handleMatrixDiscovery(w)
  354. } else if r.Method == http.MethodGet && staticRegex.MatchString(r.URL.Path) {
  355. return s.ensureWebEnabled(s.handleStatic)(w, r, v)
  356. } else if r.Method == http.MethodGet && docsRegex.MatchString(r.URL.Path) {
  357. return s.ensureWebEnabled(s.handleDocs)(w, r, v)
  358. } else if (r.Method == http.MethodGet || r.Method == http.MethodHead) && fileRegex.MatchString(r.URL.Path) && s.config.AttachmentCacheDir != "" {
  359. return s.limitRequests(s.handleFile)(w, r, v)
  360. } else if r.Method == http.MethodOptions {
  361. return s.ensureWebEnabled(s.handleOptions)(w, r, v)
  362. } else if (r.Method == http.MethodPut || r.Method == http.MethodPost) && r.URL.Path == "/" {
  363. return s.limitRequests(s.transformBodyJSON(s.authorizeTopicWrite(s.handlePublish)))(w, r, v)
  364. } else if r.Method == http.MethodPost && r.URL.Path == matrixPushPath {
  365. return s.limitRequests(s.transformMatrixJSON(s.authorizeTopicWrite(s.handlePublishMatrix)))(w, r, v)
  366. } else if (r.Method == http.MethodPut || r.Method == http.MethodPost) && topicPathRegex.MatchString(r.URL.Path) {
  367. return s.limitRequests(s.authorizeTopicWrite(s.handlePublish))(w, r, v)
  368. } else if r.Method == http.MethodGet && publishPathRegex.MatchString(r.URL.Path) {
  369. return s.limitRequests(s.authorizeTopicWrite(s.handlePublish))(w, r, v)
  370. } else if r.Method == http.MethodGet && jsonPathRegex.MatchString(r.URL.Path) {
  371. return s.limitRequests(s.authorizeTopicRead(s.handleSubscribeJSON))(w, r, v)
  372. } else if r.Method == http.MethodGet && ssePathRegex.MatchString(r.URL.Path) {
  373. return s.limitRequests(s.authorizeTopicRead(s.handleSubscribeSSE))(w, r, v)
  374. } else if r.Method == http.MethodGet && rawPathRegex.MatchString(r.URL.Path) {
  375. return s.limitRequests(s.authorizeTopicRead(s.handleSubscribeRaw))(w, r, v)
  376. } else if r.Method == http.MethodGet && wsPathRegex.MatchString(r.URL.Path) {
  377. return s.limitRequests(s.authorizeTopicRead(s.handleSubscribeWS))(w, r, v)
  378. } else if r.Method == http.MethodGet && authPathRegex.MatchString(r.URL.Path) {
  379. return s.limitRequests(s.authorizeTopicRead(s.handleTopicAuth))(w, r, v)
  380. } else if r.Method == http.MethodGet && (topicPathRegex.MatchString(r.URL.Path) || externalTopicPathRegex.MatchString(r.URL.Path)) {
  381. return s.ensureWebEnabled(s.handleTopic)(w, r, v)
  382. }
  383. return errHTTPNotFound
  384. }
  385. func (s *Server) handleHome(w http.ResponseWriter, r *http.Request, v *visitor) error {
  386. if s.config.WebRootIsApp {
  387. r.URL.Path = webAppIndex
  388. } else {
  389. r.URL.Path = webHomeIndex
  390. }
  391. return s.handleStatic(w, r, v)
  392. }
  393. func (s *Server) handleTopic(w http.ResponseWriter, r *http.Request, v *visitor) error {
  394. unifiedpush := readBoolParam(r, false, "x-unifiedpush", "unifiedpush", "up") // see PUT/POST too!
  395. if unifiedpush {
  396. w.Header().Set("Content-Type", "application/json")
  397. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  398. _, err := io.WriteString(w, `{"unifiedpush":{"version":1}}`+"\n")
  399. return err
  400. }
  401. r.URL.Path = webAppIndex
  402. return s.handleStatic(w, r, v)
  403. }
  404. func (s *Server) handleEmpty(_ http.ResponseWriter, _ *http.Request, _ *visitor) error {
  405. return nil
  406. }
  407. func (s *Server) handleTopicAuth(w http.ResponseWriter, _ *http.Request, _ *visitor) error {
  408. w.Header().Set("Content-Type", "application/json")
  409. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  410. _, err := io.WriteString(w, `{"success":true}`+"\n")
  411. return err
  412. }
  413. func (s *Server) handleHealth(w http.ResponseWriter, _ *http.Request, _ *visitor) error {
  414. response := &apiHealthResponse{
  415. Healthy: true,
  416. }
  417. w.Header().Set("Content-Type", "text/json")
  418. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  419. if err := json.NewEncoder(w).Encode(response); err != nil {
  420. return err
  421. }
  422. return nil
  423. }
  424. func (s *Server) handleWebConfig(w http.ResponseWriter, _ *http.Request, _ *visitor) error {
  425. appRoot := "/"
  426. if !s.config.WebRootIsApp {
  427. appRoot = "/app"
  428. }
  429. response := &apiConfigResponse{
  430. BaseURL: "", // Will translate to window.location.origin
  431. AppRoot: appRoot,
  432. EnableLogin: s.config.EnableLogin,
  433. EnableSignup: s.config.EnableSignup,
  434. EnablePasswordReset: s.config.EnablePasswordReset,
  435. EnablePayments: s.config.EnablePayments,
  436. EnableReserveTopics: s.config.EnableReserveTopics,
  437. DisallowedTopics: disallowedTopics,
  438. }
  439. b, err := json.Marshal(response)
  440. if err != nil {
  441. return err
  442. }
  443. w.Header().Set("Content-Type", "text/javascript")
  444. _, err = io.WriteString(w, fmt.Sprintf("// Generated server configuration\nvar config = %s;\n", string(b)))
  445. return err
  446. }
  447. func (s *Server) handleStatic(w http.ResponseWriter, r *http.Request, _ *visitor) error {
  448. r.URL.Path = webSiteDir + r.URL.Path
  449. util.Gzip(http.FileServer(http.FS(webFsCached))).ServeHTTP(w, r)
  450. return nil
  451. }
  452. func (s *Server) handleDocs(w http.ResponseWriter, r *http.Request, _ *visitor) error {
  453. util.Gzip(http.FileServer(http.FS(docsStaticCached))).ServeHTTP(w, r)
  454. return nil
  455. }
  456. func (s *Server) handleFile(w http.ResponseWriter, r *http.Request, v *visitor) error {
  457. if s.config.AttachmentCacheDir == "" {
  458. return errHTTPInternalError
  459. }
  460. matches := fileRegex.FindStringSubmatch(r.URL.Path)
  461. if len(matches) != 2 {
  462. return errHTTPInternalErrorInvalidPath
  463. }
  464. messageID := matches[1]
  465. file := filepath.Join(s.config.AttachmentCacheDir, messageID)
  466. stat, err := os.Stat(file)
  467. if err != nil {
  468. return errHTTPNotFound
  469. }
  470. if r.Method == http.MethodGet {
  471. if err := v.BandwidthLimiter().Allow(stat.Size()); err != nil {
  472. return errHTTPTooManyRequestsLimitAttachmentBandwidth
  473. }
  474. }
  475. w.Header().Set("Content-Length", fmt.Sprintf("%d", stat.Size()))
  476. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  477. if r.Method == http.MethodGet {
  478. f, err := os.Open(file)
  479. if err != nil {
  480. return err
  481. }
  482. defer f.Close()
  483. _, err = io.Copy(util.NewContentTypeWriter(w, r.URL.Path), f)
  484. return err
  485. }
  486. return nil
  487. }
  488. func (s *Server) handleMatrixDiscovery(w http.ResponseWriter) error {
  489. if s.config.BaseURL == "" {
  490. return errHTTPInternalErrorMissingBaseURL
  491. }
  492. return writeMatrixDiscoveryResponse(w)
  493. }
  494. func (s *Server) handlePublishWithoutResponse(r *http.Request, v *visitor) (*message, error) {
  495. t, err := s.topicFromPath(r.URL.Path)
  496. if err != nil {
  497. return nil, err
  498. }
  499. if err := v.MessageAllowed(); err != nil {
  500. return nil, errHTTPTooManyRequestsLimitMessages
  501. }
  502. body, err := util.Peek(r.Body, s.config.MessageLimit)
  503. if err != nil {
  504. return nil, err
  505. }
  506. m := newDefaultMessage(t.ID, "")
  507. cache, firebase, email, unifiedpush, err := s.parsePublishParams(r, v, m)
  508. if err != nil {
  509. return nil, err
  510. }
  511. if m.PollID != "" {
  512. m = newPollRequestMessage(t.ID, m.PollID)
  513. }
  514. if v.user != nil {
  515. m.User = v.user.Name
  516. }
  517. m.Expires = time.Now().Add(v.Limits().MessagesExpiryDuration).Unix()
  518. if err := s.handlePublishBody(r, v, m, body, unifiedpush); err != nil {
  519. return nil, err
  520. }
  521. if m.Message == "" {
  522. m.Message = emptyMessageBody
  523. }
  524. delayed := m.Time > time.Now().Unix()
  525. log.Debug("%s Received message: event=%s, user=%s, body=%d byte(s), delayed=%t, firebase=%t, cache=%t, up=%t, email=%s",
  526. logMessagePrefix(v, m), m.Event, m.User, len(m.Message), delayed, firebase, cache, unifiedpush, email)
  527. if log.IsTrace() {
  528. log.Trace("%s Message body: %s", logMessagePrefix(v, m), util.MaybeMarshalJSON(m))
  529. }
  530. if !delayed {
  531. if err := t.Publish(v, m); err != nil {
  532. return nil, err
  533. }
  534. if s.firebaseClient != nil && firebase {
  535. go s.sendToFirebase(v, m)
  536. }
  537. if s.smtpSender != nil && email != "" {
  538. v.IncrEmails()
  539. go s.sendEmail(v, m, email)
  540. }
  541. if s.config.UpstreamBaseURL != "" {
  542. go s.forwardPollRequest(v, m)
  543. }
  544. } else {
  545. log.Debug("%s Message delayed, will process later", logMessagePrefix(v, m))
  546. }
  547. if cache {
  548. log.Debug("%s Adding message to cache", logMessagePrefix(v, m))
  549. if err := s.messageCache.AddMessage(m); err != nil {
  550. return nil, err
  551. }
  552. }
  553. v.IncrMessages()
  554. if s.userManager != nil && v.user != nil {
  555. s.userManager.EnqueueStats(v.user)
  556. }
  557. s.mu.Lock()
  558. s.messages++
  559. s.mu.Unlock()
  560. return m, nil
  561. }
  562. func (s *Server) handlePublish(w http.ResponseWriter, r *http.Request, v *visitor) error {
  563. m, err := s.handlePublishWithoutResponse(r, v)
  564. if err != nil {
  565. return err
  566. }
  567. w.Header().Set("Content-Type", "application/json")
  568. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  569. if err := json.NewEncoder(w).Encode(m); err != nil {
  570. return err
  571. }
  572. return nil
  573. }
  574. func (s *Server) handlePublishMatrix(w http.ResponseWriter, r *http.Request, v *visitor) error {
  575. _, err := s.handlePublishWithoutResponse(r, v)
  576. if err != nil {
  577. return &errMatrix{pushKey: r.Header.Get(matrixPushKeyHeader), err: err}
  578. }
  579. return writeMatrixSuccess(w)
  580. }
  581. func (s *Server) sendToFirebase(v *visitor, m *message) {
  582. log.Debug("%s Publishing to Firebase", logMessagePrefix(v, m))
  583. if err := s.firebaseClient.Send(v, m); err != nil {
  584. if err == errFirebaseTemporarilyBanned {
  585. log.Debug("%s Unable to publish to Firebase: %v", logMessagePrefix(v, m), err.Error())
  586. } else {
  587. log.Warn("%s Unable to publish to Firebase: %v", logMessagePrefix(v, m), err.Error())
  588. }
  589. }
  590. }
  591. func (s *Server) sendEmail(v *visitor, m *message, email string) {
  592. log.Debug("%s Sending email to %s", logMessagePrefix(v, m), email)
  593. if err := s.smtpSender.Send(v, m, email); err != nil {
  594. log.Warn("%s Unable to send email to %s: %v", logMessagePrefix(v, m), email, err.Error())
  595. }
  596. }
  597. func (s *Server) forwardPollRequest(v *visitor, m *message) {
  598. topicURL := fmt.Sprintf("%s/%s", s.config.BaseURL, m.Topic)
  599. topicHash := fmt.Sprintf("%x", sha256.Sum256([]byte(topicURL)))
  600. forwardURL := fmt.Sprintf("%s/%s", s.config.UpstreamBaseURL, topicHash)
  601. log.Debug("%s Publishing poll request to %s", logMessagePrefix(v, m), forwardURL)
  602. req, err := http.NewRequest("POST", forwardURL, strings.NewReader(""))
  603. if err != nil {
  604. log.Warn("%s Unable to publish poll request: %v", logMessagePrefix(v, m), err.Error())
  605. return
  606. }
  607. req.Header.Set("X-Poll-ID", m.ID)
  608. var httpClient = &http.Client{
  609. Timeout: time.Second * 10,
  610. }
  611. response, err := httpClient.Do(req)
  612. if err != nil {
  613. log.Warn("%s Unable to publish poll request: %v", logMessagePrefix(v, m), err.Error())
  614. return
  615. } else if response.StatusCode != http.StatusOK {
  616. log.Warn("%s Unable to publish poll request, unexpected HTTP status: %d", logMessagePrefix(v, m), response.StatusCode)
  617. return
  618. }
  619. }
  620. func (s *Server) parsePublishParams(r *http.Request, v *visitor, m *message) (cache bool, firebase bool, email string, unifiedpush bool, err error) {
  621. cache = readBoolParam(r, true, "x-cache", "cache")
  622. firebase = readBoolParam(r, true, "x-firebase", "firebase")
  623. m.Title = readParam(r, "x-title", "title", "t")
  624. m.Click = readParam(r, "x-click", "click")
  625. icon := readParam(r, "x-icon", "icon")
  626. filename := readParam(r, "x-filename", "filename", "file", "f")
  627. attach := readParam(r, "x-attach", "attach", "a")
  628. if attach != "" || filename != "" {
  629. m.Attachment = &attachment{}
  630. }
  631. if filename != "" {
  632. m.Attachment.Name = filename
  633. }
  634. if attach != "" {
  635. if !urlRegex.MatchString(attach) {
  636. return false, false, "", false, errHTTPBadRequestAttachmentURLInvalid
  637. }
  638. m.Attachment.URL = attach
  639. if m.Attachment.Name == "" {
  640. u, err := url.Parse(m.Attachment.URL)
  641. if err == nil {
  642. m.Attachment.Name = path.Base(u.Path)
  643. if m.Attachment.Name == "." || m.Attachment.Name == "/" {
  644. m.Attachment.Name = ""
  645. }
  646. }
  647. }
  648. if m.Attachment.Name == "" {
  649. m.Attachment.Name = "attachment"
  650. }
  651. }
  652. if icon != "" {
  653. if !urlRegex.MatchString(icon) {
  654. return false, false, "", false, errHTTPBadRequestIconURLInvalid
  655. }
  656. m.Icon = icon
  657. }
  658. email = readParam(r, "x-email", "x-e-mail", "email", "e-mail", "mail", "e")
  659. if email != "" {
  660. if err := v.EmailAllowed(); err != nil {
  661. return false, false, "", false, errHTTPTooManyRequestsLimitEmails
  662. }
  663. }
  664. if s.smtpSender == nil && email != "" {
  665. return false, false, "", false, errHTTPBadRequestEmailDisabled
  666. }
  667. messageStr := strings.ReplaceAll(readParam(r, "x-message", "message", "m"), "\\n", "\n")
  668. if messageStr != "" {
  669. m.Message = messageStr
  670. }
  671. m.Priority, err = util.ParsePriority(readParam(r, "x-priority", "priority", "prio", "p"))
  672. if err != nil {
  673. return false, false, "", false, errHTTPBadRequestPriorityInvalid
  674. }
  675. tagsStr := readParam(r, "x-tags", "tags", "tag", "ta")
  676. if tagsStr != "" {
  677. m.Tags = make([]string, 0)
  678. for _, s := range util.SplitNoEmpty(tagsStr, ",") {
  679. m.Tags = append(m.Tags, strings.TrimSpace(s))
  680. }
  681. }
  682. delayStr := readParam(r, "x-delay", "delay", "x-at", "at", "x-in", "in")
  683. if delayStr != "" {
  684. if !cache {
  685. return false, false, "", false, errHTTPBadRequestDelayNoCache
  686. }
  687. if email != "" {
  688. return false, false, "", false, errHTTPBadRequestDelayNoEmail // we cannot store the email address (yet)
  689. }
  690. delay, err := util.ParseFutureTime(delayStr, time.Now())
  691. if err != nil {
  692. return false, false, "", false, errHTTPBadRequestDelayCannotParse
  693. } else if delay.Unix() < time.Now().Add(s.config.MinDelay).Unix() {
  694. return false, false, "", false, errHTTPBadRequestDelayTooSmall
  695. } else if delay.Unix() > time.Now().Add(s.config.MaxDelay).Unix() {
  696. return false, false, "", false, errHTTPBadRequestDelayTooLarge
  697. }
  698. m.Time = delay.Unix()
  699. m.Sender = v.ip // Important for rate limiting
  700. }
  701. actionsStr := readParam(r, "x-actions", "actions", "action")
  702. if actionsStr != "" {
  703. m.Actions, err = parseActions(actionsStr)
  704. if err != nil {
  705. return false, false, "", false, wrapErrHTTP(errHTTPBadRequestActionsInvalid, err.Error())
  706. }
  707. }
  708. unifiedpush = readBoolParam(r, false, "x-unifiedpush", "unifiedpush", "up") // see GET too!
  709. if unifiedpush {
  710. firebase = false
  711. unifiedpush = true
  712. }
  713. m.PollID = readParam(r, "x-poll-id", "poll-id")
  714. if m.PollID != "" {
  715. unifiedpush = false
  716. cache = false
  717. email = ""
  718. }
  719. return cache, firebase, email, unifiedpush, nil
  720. }
  721. // handlePublishBody consumes the PUT/POST body and decides whether the body is an attachment or the message.
  722. //
  723. // 1. curl -X POST -H "Poll: 1234" ntfy.sh/...
  724. // If a message is flagged as poll request, the body does not matter and is discarded
  725. // 2. curl -T somebinarydata.bin "ntfy.sh/mytopic?up=1"
  726. // If body is binary, encode as base64, if not do not encode
  727. // 3. curl -H "Attach: http://example.com/file.jpg" ntfy.sh/mytopic
  728. // Body must be a message, because we attached an external URL
  729. // 4. curl -T short.txt -H "Filename: short.txt" ntfy.sh/mytopic
  730. // Body must be attachment, because we passed a filename
  731. // 5. curl -T file.txt ntfy.sh/mytopic
  732. // If file.txt is <= 4096 (message limit) and valid UTF-8, treat it as a message
  733. // 6. curl -T file.txt ntfy.sh/mytopic
  734. // If file.txt is > message limit, treat it as an attachment
  735. func (s *Server) handlePublishBody(r *http.Request, v *visitor, m *message, body *util.PeekedReadCloser, unifiedpush bool) error {
  736. if m.Event == pollRequestEvent { // Case 1
  737. return s.handleBodyDiscard(body)
  738. } else if unifiedpush {
  739. return s.handleBodyAsMessageAutoDetect(m, body) // Case 2
  740. } else if m.Attachment != nil && m.Attachment.URL != "" {
  741. return s.handleBodyAsTextMessage(m, body) // Case 3
  742. } else if m.Attachment != nil && m.Attachment.Name != "" {
  743. return s.handleBodyAsAttachment(r, v, m, body) // Case 4
  744. } else if !body.LimitReached && utf8.Valid(body.PeekedBytes) {
  745. return s.handleBodyAsTextMessage(m, body) // Case 5
  746. }
  747. return s.handleBodyAsAttachment(r, v, m, body) // Case 6
  748. }
  749. func (s *Server) handleBodyDiscard(body *util.PeekedReadCloser) error {
  750. _, err := io.Copy(io.Discard, body)
  751. _ = body.Close()
  752. return err
  753. }
  754. func (s *Server) handleBodyAsMessageAutoDetect(m *message, body *util.PeekedReadCloser) error {
  755. if utf8.Valid(body.PeekedBytes) {
  756. m.Message = string(body.PeekedBytes) // Do not trim
  757. } else {
  758. m.Message = base64.StdEncoding.EncodeToString(body.PeekedBytes)
  759. m.Encoding = encodingBase64
  760. }
  761. return nil
  762. }
  763. func (s *Server) handleBodyAsTextMessage(m *message, body *util.PeekedReadCloser) error {
  764. if !utf8.Valid(body.PeekedBytes) {
  765. return errHTTPBadRequestMessageNotUTF8
  766. }
  767. if len(body.PeekedBytes) > 0 { // Empty body should not override message (publish via GET!)
  768. m.Message = strings.TrimSpace(string(body.PeekedBytes)) // Truncates the message to the peek limit if required
  769. }
  770. if m.Attachment != nil && m.Attachment.Name != "" && m.Message == "" {
  771. m.Message = fmt.Sprintf(defaultAttachmentMessage, m.Attachment.Name)
  772. }
  773. return nil
  774. }
  775. func (s *Server) handleBodyAsAttachment(r *http.Request, v *visitor, m *message, body *util.PeekedReadCloser) error {
  776. if s.fileCache == nil || s.config.BaseURL == "" || s.config.AttachmentCacheDir == "" {
  777. return errHTTPBadRequestAttachmentsDisallowed
  778. }
  779. vinfo, err := v.Info()
  780. if err != nil {
  781. return err
  782. }
  783. attachmentExpiry := time.Now().Add(vinfo.Limits.AttachmentExpiryDuration).Unix()
  784. if m.Time > attachmentExpiry {
  785. return errHTTPBadRequestAttachmentsExpiryBeforeDelivery
  786. }
  787. contentLengthStr := r.Header.Get("Content-Length")
  788. if contentLengthStr != "" { // Early "do-not-trust" check, hard limit see below
  789. contentLength, err := strconv.ParseInt(contentLengthStr, 10, 64)
  790. if err == nil && (contentLength > vinfo.Stats.AttachmentTotalSizeRemaining || contentLength > vinfo.Limits.AttachmentFileSizeLimit) {
  791. return errHTTPEntityTooLargeAttachment
  792. }
  793. }
  794. if m.Attachment == nil {
  795. m.Attachment = &attachment{}
  796. }
  797. var ext string
  798. m.Sender = v.ip // Important for attachment rate limiting
  799. m.Attachment.Expires = attachmentExpiry
  800. m.Attachment.Type, ext = util.DetectContentType(body.PeekedBytes, m.Attachment.Name)
  801. m.Attachment.URL = fmt.Sprintf("%s/file/%s%s", s.config.BaseURL, m.ID, ext)
  802. if m.Attachment.Name == "" {
  803. m.Attachment.Name = fmt.Sprintf("attachment%s", ext)
  804. }
  805. if m.Message == "" {
  806. m.Message = fmt.Sprintf(defaultAttachmentMessage, m.Attachment.Name)
  807. }
  808. limiters := []util.Limiter{
  809. v.BandwidthLimiter(),
  810. util.NewFixedLimiter(vinfo.Limits.AttachmentFileSizeLimit),
  811. util.NewFixedLimiter(vinfo.Stats.AttachmentTotalSizeRemaining),
  812. }
  813. m.Attachment.Size, err = s.fileCache.Write(m.ID, body, limiters...)
  814. if err == util.ErrLimitReached {
  815. return errHTTPEntityTooLargeAttachment
  816. } else if err != nil {
  817. return err
  818. }
  819. return nil
  820. }
  821. func (s *Server) handleSubscribeJSON(w http.ResponseWriter, r *http.Request, v *visitor) error {
  822. encoder := func(msg *message) (string, error) {
  823. var buf bytes.Buffer
  824. if err := json.NewEncoder(&buf).Encode(&msg); err != nil {
  825. return "", err
  826. }
  827. return buf.String(), nil
  828. }
  829. return s.handleSubscribeHTTP(w, r, v, "application/x-ndjson", encoder)
  830. }
  831. func (s *Server) handleSubscribeSSE(w http.ResponseWriter, r *http.Request, v *visitor) error {
  832. encoder := func(msg *message) (string, error) {
  833. var buf bytes.Buffer
  834. if err := json.NewEncoder(&buf).Encode(&msg); err != nil {
  835. return "", err
  836. }
  837. if msg.Event != messageEvent {
  838. return fmt.Sprintf("event: %s\ndata: %s\n", msg.Event, buf.String()), nil // Browser's .onmessage() does not fire on this!
  839. }
  840. return fmt.Sprintf("data: %s\n", buf.String()), nil
  841. }
  842. return s.handleSubscribeHTTP(w, r, v, "text/event-stream", encoder)
  843. }
  844. func (s *Server) handleSubscribeRaw(w http.ResponseWriter, r *http.Request, v *visitor) error {
  845. encoder := func(msg *message) (string, error) {
  846. if msg.Event == messageEvent { // only handle default events
  847. return strings.ReplaceAll(msg.Message, "\n", " ") + "\n", nil
  848. }
  849. return "\n", nil // "keepalive" and "open" events just send an empty line
  850. }
  851. return s.handleSubscribeHTTP(w, r, v, "text/plain", encoder)
  852. }
  853. func (s *Server) handleSubscribeHTTP(w http.ResponseWriter, r *http.Request, v *visitor, contentType string, encoder messageEncoder) error {
  854. log.Debug("%s HTTP stream connection opened", logHTTPPrefix(v, r))
  855. defer log.Debug("%s HTTP stream connection closed", logHTTPPrefix(v, r))
  856. if err := v.SubscriptionAllowed(); err != nil {
  857. return errHTTPTooManyRequestsLimitSubscriptions
  858. }
  859. defer v.RemoveSubscription()
  860. topics, topicsStr, err := s.topicsFromPath(r.URL.Path)
  861. if err != nil {
  862. return err
  863. }
  864. poll, since, scheduled, filters, err := parseSubscribeParams(r)
  865. if err != nil {
  866. return err
  867. }
  868. var wlock sync.Mutex
  869. defer func() {
  870. // Hack: This is the fix for a horrible data race that I have not been able to figure out in quite some time.
  871. // It appears to be happening when the Go HTTP code reads from the socket when closing the request (i.e. AFTER
  872. // this function returns), and causes a data race with the ResponseWriter. Locking wlock here silences the
  873. // data race detector. See https://github.com/binwiederhier/ntfy/issues/338#issuecomment-1163425889.
  874. wlock.TryLock()
  875. }()
  876. sub := func(v *visitor, msg *message) error {
  877. if !filters.Pass(msg) {
  878. return nil
  879. }
  880. m, err := encoder(msg)
  881. if err != nil {
  882. return err
  883. }
  884. wlock.Lock()
  885. defer wlock.Unlock()
  886. if _, err := w.Write([]byte(m)); err != nil {
  887. return err
  888. }
  889. if fl, ok := w.(http.Flusher); ok {
  890. fl.Flush()
  891. }
  892. return nil
  893. }
  894. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  895. w.Header().Set("Content-Type", contentType+"; charset=utf-8") // Android/Volley client needs charset!
  896. if poll {
  897. return s.sendOldMessages(topics, since, scheduled, v, sub)
  898. }
  899. subscriberIDs := make([]int, 0)
  900. for _, t := range topics {
  901. subscriberIDs = append(subscriberIDs, t.Subscribe(sub))
  902. }
  903. defer func() {
  904. for i, subscriberID := range subscriberIDs {
  905. topics[i].Unsubscribe(subscriberID) // Order!
  906. }
  907. }()
  908. if err := sub(v, newOpenMessage(topicsStr)); err != nil { // Send out open message
  909. return err
  910. }
  911. if err := s.sendOldMessages(topics, since, scheduled, v, sub); err != nil {
  912. return err
  913. }
  914. for {
  915. select {
  916. case <-r.Context().Done():
  917. return nil
  918. case <-time.After(s.config.KeepaliveInterval):
  919. log.Trace("%s Sending keepalive message", logHTTPPrefix(v, r))
  920. v.Keepalive()
  921. if err := sub(v, newKeepaliveMessage(topicsStr)); err != nil { // Send keepalive message
  922. return err
  923. }
  924. }
  925. }
  926. }
  927. func (s *Server) handleSubscribeWS(w http.ResponseWriter, r *http.Request, v *visitor) error {
  928. if strings.ToLower(r.Header.Get("Upgrade")) != "websocket" {
  929. return errHTTPBadRequestWebSocketsUpgradeHeaderMissing
  930. }
  931. if err := v.SubscriptionAllowed(); err != nil {
  932. return errHTTPTooManyRequestsLimitSubscriptions
  933. }
  934. defer v.RemoveSubscription()
  935. log.Debug("%s WebSocket connection opened", logHTTPPrefix(v, r))
  936. defer log.Debug("%s WebSocket connection closed", logHTTPPrefix(v, r))
  937. topics, topicsStr, err := s.topicsFromPath(r.URL.Path)
  938. if err != nil {
  939. return err
  940. }
  941. poll, since, scheduled, filters, err := parseSubscribeParams(r)
  942. if err != nil {
  943. return err
  944. }
  945. upgrader := &websocket.Upgrader{
  946. ReadBufferSize: wsBufferSize,
  947. WriteBufferSize: wsBufferSize,
  948. CheckOrigin: func(r *http.Request) bool {
  949. return true // We're open for business!
  950. },
  951. }
  952. conn, err := upgrader.Upgrade(w, r, nil)
  953. if err != nil {
  954. return err
  955. }
  956. defer conn.Close()
  957. var wlock sync.Mutex
  958. g, ctx := errgroup.WithContext(context.Background())
  959. g.Go(func() error {
  960. pongWait := s.config.KeepaliveInterval + wsPongWait
  961. conn.SetReadLimit(wsReadLimit)
  962. if err := conn.SetReadDeadline(time.Now().Add(pongWait)); err != nil {
  963. return err
  964. }
  965. conn.SetPongHandler(func(appData string) error {
  966. log.Trace("%s Received WebSocket pong", logHTTPPrefix(v, r))
  967. return conn.SetReadDeadline(time.Now().Add(pongWait))
  968. })
  969. for {
  970. _, _, err := conn.NextReader()
  971. if err != nil {
  972. return err
  973. }
  974. }
  975. })
  976. g.Go(func() error {
  977. ping := func() error {
  978. wlock.Lock()
  979. defer wlock.Unlock()
  980. if err := conn.SetWriteDeadline(time.Now().Add(wsWriteWait)); err != nil {
  981. return err
  982. }
  983. log.Trace("%s Sending WebSocket ping", logHTTPPrefix(v, r))
  984. return conn.WriteMessage(websocket.PingMessage, nil)
  985. }
  986. for {
  987. select {
  988. case <-ctx.Done():
  989. return nil
  990. case <-time.After(s.config.KeepaliveInterval):
  991. v.Keepalive()
  992. if err := ping(); err != nil {
  993. return err
  994. }
  995. }
  996. }
  997. })
  998. sub := func(v *visitor, msg *message) error {
  999. if !filters.Pass(msg) {
  1000. return nil
  1001. }
  1002. wlock.Lock()
  1003. defer wlock.Unlock()
  1004. if err := conn.SetWriteDeadline(time.Now().Add(wsWriteWait)); err != nil {
  1005. return err
  1006. }
  1007. return conn.WriteJSON(msg)
  1008. }
  1009. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  1010. if poll {
  1011. return s.sendOldMessages(topics, since, scheduled, v, sub)
  1012. }
  1013. subscriberIDs := make([]int, 0)
  1014. for _, t := range topics {
  1015. subscriberIDs = append(subscriberIDs, t.Subscribe(sub))
  1016. }
  1017. defer func() {
  1018. for i, subscriberID := range subscriberIDs {
  1019. topics[i].Unsubscribe(subscriberID) // Order!
  1020. }
  1021. }()
  1022. if err := sub(v, newOpenMessage(topicsStr)); err != nil { // Send out open message
  1023. return err
  1024. }
  1025. if err := s.sendOldMessages(topics, since, scheduled, v, sub); err != nil {
  1026. return err
  1027. }
  1028. err = g.Wait()
  1029. if err != nil && websocket.IsCloseError(err, websocket.CloseNormalClosure, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
  1030. log.Trace("%s WebSocket connection closed: %s", logHTTPPrefix(v, r), err.Error())
  1031. return nil // Normal closures are not errors; note: "1006 (abnormal closure)" is treated as normal, because people disconnect a lot
  1032. }
  1033. return err
  1034. }
  1035. func parseSubscribeParams(r *http.Request) (poll bool, since sinceMarker, scheduled bool, filters *queryFilter, err error) {
  1036. poll = readBoolParam(r, false, "x-poll", "poll", "po")
  1037. scheduled = readBoolParam(r, false, "x-scheduled", "scheduled", "sched")
  1038. since, err = parseSince(r, poll)
  1039. if err != nil {
  1040. return
  1041. }
  1042. filters, err = parseQueryFilters(r)
  1043. if err != nil {
  1044. return
  1045. }
  1046. return
  1047. }
  1048. // sendOldMessages selects old messages from the messageCache and calls sub for each of them. It uses since as the
  1049. // marker, returning only messages that are newer than the marker.
  1050. func (s *Server) sendOldMessages(topics []*topic, since sinceMarker, scheduled bool, v *visitor, sub subscriber) error {
  1051. if since.IsNone() {
  1052. return nil
  1053. }
  1054. messages := make([]*message, 0)
  1055. for _, t := range topics {
  1056. topicMessages, err := s.messageCache.Messages(t.ID, since, scheduled)
  1057. if err != nil {
  1058. return err
  1059. }
  1060. messages = append(messages, topicMessages...)
  1061. }
  1062. sort.Slice(messages, func(i, j int) bool {
  1063. return messages[i].Time < messages[j].Time
  1064. })
  1065. for _, m := range messages {
  1066. if err := sub(v, m); err != nil {
  1067. return err
  1068. }
  1069. }
  1070. return nil
  1071. }
  1072. // parseSince returns a timestamp identifying the time span from which cached messages should be received.
  1073. //
  1074. // Values in the "since=..." parameter can be either a unix timestamp or a duration (e.g. 12h), or
  1075. // "all" for all messages.
  1076. func parseSince(r *http.Request, poll bool) (sinceMarker, error) {
  1077. since := readParam(r, "x-since", "since", "si")
  1078. // Easy cases (empty, all, none)
  1079. if since == "" {
  1080. if poll {
  1081. return sinceAllMessages, nil
  1082. }
  1083. return sinceNoMessages, nil
  1084. } else if since == "all" {
  1085. return sinceAllMessages, nil
  1086. } else if since == "none" {
  1087. return sinceNoMessages, nil
  1088. }
  1089. // ID, timestamp, duration
  1090. if validMessageID(since) {
  1091. return newSinceID(since), nil
  1092. } else if s, err := strconv.ParseInt(since, 10, 64); err == nil {
  1093. return newSinceTime(s), nil
  1094. } else if d, err := time.ParseDuration(since); err == nil {
  1095. return newSinceTime(time.Now().Add(-1 * d).Unix()), nil
  1096. }
  1097. return sinceNoMessages, errHTTPBadRequestSinceInvalid
  1098. }
  1099. func (s *Server) handleOptions(w http.ResponseWriter, _ *http.Request, _ *visitor) error {
  1100. w.Header().Set("Access-Control-Allow-Methods", "GET, PUT, POST, PATCH, DELETE")
  1101. w.Header().Set("Access-Control-Allow-Origin", "*") // CORS, allow cross-origin requests
  1102. w.Header().Set("Access-Control-Allow-Headers", "*") // CORS, allow auth via JS // FIXME is this terrible?
  1103. return nil
  1104. }
  1105. func (s *Server) topicFromPath(path string) (*topic, error) {
  1106. parts := strings.Split(path, "/")
  1107. if len(parts) < 2 {
  1108. return nil, errHTTPBadRequestTopicInvalid
  1109. }
  1110. topics, err := s.topicsFromIDs(parts[1])
  1111. if err != nil {
  1112. return nil, err
  1113. }
  1114. return topics[0], nil
  1115. }
  1116. func (s *Server) topicsFromPath(path string) ([]*topic, string, error) {
  1117. parts := strings.Split(path, "/")
  1118. if len(parts) < 2 {
  1119. return nil, "", errHTTPBadRequestTopicInvalid
  1120. }
  1121. topicIDs := util.SplitNoEmpty(parts[1], ",")
  1122. topics, err := s.topicsFromIDs(topicIDs...)
  1123. if err != nil {
  1124. return nil, "", errHTTPBadRequestTopicInvalid
  1125. }
  1126. return topics, parts[1], nil
  1127. }
  1128. func (s *Server) topicsFromIDs(ids ...string) ([]*topic, error) {
  1129. s.mu.Lock()
  1130. defer s.mu.Unlock()
  1131. topics := make([]*topic, 0)
  1132. for _, id := range ids {
  1133. if util.Contains(disallowedTopics, id) {
  1134. return nil, errHTTPBadRequestTopicDisallowed
  1135. }
  1136. if _, ok := s.topics[id]; !ok {
  1137. if len(s.topics) >= s.config.TotalTopicLimit {
  1138. return nil, errHTTPTooManyRequestsLimitTotalTopics
  1139. }
  1140. s.topics[id] = newTopic(id)
  1141. }
  1142. topics = append(topics, s.topics[id])
  1143. }
  1144. return topics, nil
  1145. }
  1146. func (s *Server) execManager() {
  1147. log.Debug("Manager: Starting")
  1148. defer log.Debug("Manager: Finished")
  1149. // WARNING: Make sure to only selectively lock with the mutex, and be aware that this
  1150. // there is no mutex for the entire function.
  1151. // Expire visitors from rate visitors map
  1152. s.mu.Lock()
  1153. staleVisitors := 0
  1154. for ip, v := range s.visitors {
  1155. if v.Stale() {
  1156. log.Trace("Deleting stale visitor %s", v.ip)
  1157. delete(s.visitors, ip)
  1158. staleVisitors++
  1159. }
  1160. }
  1161. s.mu.Unlock()
  1162. log.Debug("Manager: Deleted %d stale visitor(s)", staleVisitors)
  1163. // Delete expired user tokens
  1164. if s.userManager != nil {
  1165. if err := s.userManager.RemoveExpiredTokens(); err != nil {
  1166. log.Warn("Error expiring user tokens: %s", err.Error())
  1167. }
  1168. }
  1169. // Delete expired attachments
  1170. if s.fileCache != nil {
  1171. ids, err := s.messageCache.AttachmentsExpired()
  1172. if err != nil {
  1173. log.Warn("Manager: Error retrieving expired attachments: %s", err.Error())
  1174. } else if len(ids) > 0 {
  1175. if log.IsDebug() {
  1176. log.Debug("Manager: Deleting attachments %s", strings.Join(ids, ", "))
  1177. }
  1178. if err := s.fileCache.Remove(ids...); err != nil {
  1179. log.Warn("Manager: Error deleting attachments: %s", err.Error())
  1180. }
  1181. if err := s.messageCache.MarkAttachmentsDeleted(ids...); err != nil {
  1182. log.Warn("Manager: Error marking attachments deleted: %s", err.Error())
  1183. }
  1184. } else {
  1185. log.Debug("Manager: No expired attachments to delete")
  1186. }
  1187. }
  1188. // DeleteMessages message cache
  1189. log.Debug("Manager: Pruning messages")
  1190. expiredMessages, err := s.messageCache.MessagesExpired()
  1191. if err != nil {
  1192. log.Warn("Manager: Error retrieving expired messages: %s", err.Error())
  1193. } else if len(expiredMessages) > 0 {
  1194. ids := make([]string, 0)
  1195. for _, m := range expiredMessages {
  1196. ids = append(ids, m.ID)
  1197. }
  1198. if err := s.fileCache.Remove(ids...); err != nil {
  1199. log.Warn("Manager: Error deleting attachments for expired messages: %s", err.Error())
  1200. }
  1201. if err := s.messageCache.DeleteMessages(ids...); err != nil {
  1202. log.Warn("Manager: Error marking attachments deleted: %s", err.Error())
  1203. }
  1204. } else {
  1205. log.Debug("Manager: No expired messages to delete")
  1206. }
  1207. // Message count per topic
  1208. var messages int
  1209. messageCounts, err := s.messageCache.MessageCounts()
  1210. if err != nil {
  1211. log.Warn("Manager: Cannot get message counts: %s", err.Error())
  1212. messageCounts = make(map[string]int) // Empty, so we can continue
  1213. }
  1214. for _, count := range messageCounts {
  1215. messages += count
  1216. }
  1217. // Remove subscriptions without subscribers
  1218. s.mu.Lock()
  1219. var subscribers int
  1220. for _, t := range s.topics {
  1221. subs := t.SubscribersCount()
  1222. msgs, exists := messageCounts[t.ID]
  1223. if subs == 0 && (!exists || msgs == 0) {
  1224. log.Trace("Deleting empty topic %s", t.ID)
  1225. delete(s.topics, t.ID)
  1226. continue
  1227. }
  1228. subscribers += subs
  1229. }
  1230. s.mu.Unlock()
  1231. // Mail stats
  1232. var receivedMailTotal, receivedMailSuccess, receivedMailFailure int64
  1233. if s.smtpServerBackend != nil {
  1234. receivedMailTotal, receivedMailSuccess, receivedMailFailure = s.smtpServerBackend.Counts()
  1235. }
  1236. var sentMailTotal, sentMailSuccess, sentMailFailure int64
  1237. if s.smtpSender != nil {
  1238. sentMailTotal, sentMailSuccess, sentMailFailure = s.smtpSender.Counts()
  1239. }
  1240. // Print stats
  1241. s.mu.Lock()
  1242. messagesCount, topicsCount, visitorsCount := s.messages, len(s.topics), len(s.visitors)
  1243. s.mu.Unlock()
  1244. log.Info("Stats: %d messages published, %d in cache, %d topic(s) active, %d subscriber(s), %d visitor(s), %d mails received (%d successful, %d failed), %d mails sent (%d successful, %d failed)",
  1245. messagesCount, messages, topicsCount, subscribers, visitorsCount,
  1246. receivedMailTotal, receivedMailSuccess, receivedMailFailure,
  1247. sentMailTotal, sentMailSuccess, sentMailFailure)
  1248. }
  1249. func (s *Server) runSMTPServer() error {
  1250. s.smtpServerBackend = newMailBackend(s.config, s.handle)
  1251. s.smtpServer = smtp.NewServer(s.smtpServerBackend)
  1252. s.smtpServer.Addr = s.config.SMTPServerListen
  1253. s.smtpServer.Domain = s.config.SMTPServerDomain
  1254. s.smtpServer.ReadTimeout = 10 * time.Second
  1255. s.smtpServer.WriteTimeout = 10 * time.Second
  1256. s.smtpServer.MaxMessageBytes = 1024 * 1024 // Must be much larger than message size (headers, multipart, etc.)
  1257. s.smtpServer.MaxRecipients = 1
  1258. s.smtpServer.AllowInsecureAuth = true
  1259. return s.smtpServer.ListenAndServe()
  1260. }
  1261. func (s *Server) runManager() {
  1262. for {
  1263. select {
  1264. case <-time.After(s.config.ManagerInterval):
  1265. s.execManager()
  1266. case <-s.closeChan:
  1267. return
  1268. }
  1269. }
  1270. }
  1271. func (s *Server) runFirebaseKeepaliver() {
  1272. if s.firebaseClient == nil {
  1273. return
  1274. }
  1275. v := newVisitor(s.config, s.messageCache, s.userManager, netip.IPv4Unspecified(), nil) // Background process, not a real visitor, uses IP 0.0.0.0
  1276. for {
  1277. select {
  1278. case <-time.After(s.config.FirebaseKeepaliveInterval):
  1279. s.sendToFirebase(v, newKeepaliveMessage(firebaseControlTopic))
  1280. case <-time.After(s.config.FirebasePollInterval):
  1281. s.sendToFirebase(v, newKeepaliveMessage(firebasePollTopic))
  1282. case <-s.closeChan:
  1283. return
  1284. }
  1285. }
  1286. }
  1287. func (s *Server) runDelayedSender() {
  1288. for {
  1289. select {
  1290. case <-time.After(s.config.DelayedSenderInterval):
  1291. if err := s.sendDelayedMessages(); err != nil {
  1292. log.Warn("Error sending delayed messages: %s", err.Error())
  1293. }
  1294. case <-s.closeChan:
  1295. return
  1296. }
  1297. }
  1298. }
  1299. func (s *Server) sendDelayedMessages() error {
  1300. messages, err := s.messageCache.MessagesDue()
  1301. if err != nil {
  1302. return err
  1303. }
  1304. for _, m := range messages {
  1305. var v *visitor
  1306. if s.userManager != nil && m.User != "" {
  1307. user, err := s.userManager.User(m.User)
  1308. if err != nil {
  1309. log.Warn("%s Error sending delayed message: %s", logMessagePrefix(v, m), err.Error())
  1310. continue
  1311. }
  1312. v = s.visitorFromUser(user, m.Sender)
  1313. } else {
  1314. v = s.visitorFromIP(m.Sender)
  1315. }
  1316. if err := s.sendDelayedMessage(v, m); err != nil {
  1317. log.Warn("%s Error sending delayed message: %s", logMessagePrefix(v, m), err.Error())
  1318. }
  1319. }
  1320. return nil
  1321. }
  1322. func (s *Server) sendDelayedMessage(v *visitor, m *message) error {
  1323. log.Debug("%s Sending delayed message", logMessagePrefix(v, m))
  1324. s.mu.Lock()
  1325. t, ok := s.topics[m.Topic] // If no subscribers, just mark message as published
  1326. s.mu.Unlock()
  1327. if ok {
  1328. go func() {
  1329. // We do not rate-limit messages here, since we've rate limited them in the PUT/POST handler
  1330. if err := t.Publish(v, m); err != nil {
  1331. log.Warn("%s Unable to publish message: %v", logMessagePrefix(v, m), err.Error())
  1332. }
  1333. }()
  1334. }
  1335. if s.firebaseClient != nil { // Firebase subscribers may not show up in topics map
  1336. go s.sendToFirebase(v, m)
  1337. }
  1338. if s.config.UpstreamBaseURL != "" {
  1339. go s.forwardPollRequest(v, m)
  1340. }
  1341. if err := s.messageCache.MarkPublished(m); err != nil {
  1342. return err
  1343. }
  1344. return nil
  1345. }
  1346. func (s *Server) limitRequests(next handleFunc) handleFunc {
  1347. return func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1348. if util.ContainsIP(s.config.VisitorRequestExemptIPAddrs, v.ip) {
  1349. return next(w, r, v)
  1350. } else if err := v.RequestAllowed(); err != nil {
  1351. return errHTTPTooManyRequestsLimitRequests
  1352. }
  1353. return next(w, r, v)
  1354. }
  1355. }
  1356. func (s *Server) ensureWebEnabled(next handleFunc) handleFunc {
  1357. return func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1358. if !s.config.EnableWeb {
  1359. return errHTTPNotFound
  1360. }
  1361. return next(w, r, v)
  1362. }
  1363. }
  1364. func (s *Server) ensureUserManager(next handleFunc) handleFunc {
  1365. return func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1366. if s.userManager == nil {
  1367. return errHTTPNotFound
  1368. }
  1369. return next(w, r, v)
  1370. }
  1371. }
  1372. func (s *Server) ensureUser(next handleFunc) handleFunc {
  1373. return s.ensureUserManager(func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1374. if v.user == nil {
  1375. return errHTTPUnauthorized
  1376. }
  1377. return next(w, r, v)
  1378. })
  1379. }
  1380. // transformBodyJSON peeks the request body, reads the JSON, and converts it to headers
  1381. // before passing it on to the next handler. This is meant to be used in combination with handlePublish.
  1382. func (s *Server) transformBodyJSON(next handleFunc) handleFunc {
  1383. return func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1384. m, err := readJSONWithLimit[publishMessage](r.Body, s.config.MessageLimit)
  1385. if err != nil {
  1386. return err
  1387. }
  1388. if !topicRegex.MatchString(m.Topic) {
  1389. return errHTTPBadRequestTopicInvalid
  1390. }
  1391. if m.Message == "" {
  1392. m.Message = emptyMessageBody
  1393. }
  1394. r.URL.Path = "/" + m.Topic
  1395. r.Body = io.NopCloser(strings.NewReader(m.Message))
  1396. if m.Title != "" {
  1397. r.Header.Set("X-Title", m.Title)
  1398. }
  1399. if m.Priority != 0 {
  1400. r.Header.Set("X-Priority", fmt.Sprintf("%d", m.Priority))
  1401. }
  1402. if m.Tags != nil && len(m.Tags) > 0 {
  1403. r.Header.Set("X-Tags", strings.Join(m.Tags, ","))
  1404. }
  1405. if m.Attach != "" {
  1406. r.Header.Set("X-Attach", m.Attach)
  1407. }
  1408. if m.Filename != "" {
  1409. r.Header.Set("X-Filename", m.Filename)
  1410. }
  1411. if m.Click != "" {
  1412. r.Header.Set("X-Click", m.Click)
  1413. }
  1414. if m.Icon != "" {
  1415. r.Header.Set("X-Icon", m.Icon)
  1416. }
  1417. if len(m.Actions) > 0 {
  1418. actionsStr, err := json.Marshal(m.Actions)
  1419. if err != nil {
  1420. return errHTTPBadRequestMessageJSONInvalid
  1421. }
  1422. r.Header.Set("X-Actions", string(actionsStr))
  1423. }
  1424. if m.Email != "" {
  1425. r.Header.Set("X-Email", m.Email)
  1426. }
  1427. if m.Delay != "" {
  1428. r.Header.Set("X-Delay", m.Delay)
  1429. }
  1430. return next(w, r, v)
  1431. }
  1432. }
  1433. func (s *Server) transformMatrixJSON(next handleFunc) handleFunc {
  1434. return func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1435. newRequest, err := newRequestFromMatrixJSON(r, s.config.BaseURL, s.config.MessageLimit)
  1436. if err != nil {
  1437. return err
  1438. }
  1439. if err := next(w, newRequest, v); err != nil {
  1440. return &errMatrix{pushKey: newRequest.Header.Get(matrixPushKeyHeader), err: err}
  1441. }
  1442. return nil
  1443. }
  1444. }
  1445. func (s *Server) authorizeTopicWrite(next handleFunc) handleFunc {
  1446. return s.autorizeTopic(next, user.PermissionWrite)
  1447. }
  1448. func (s *Server) authorizeTopicRead(next handleFunc) handleFunc {
  1449. return s.autorizeTopic(next, user.PermissionRead)
  1450. }
  1451. func (s *Server) autorizeTopic(next handleFunc, perm user.Permission) handleFunc {
  1452. return func(w http.ResponseWriter, r *http.Request, v *visitor) error {
  1453. if s.userManager == nil {
  1454. return next(w, r, v)
  1455. }
  1456. topics, _, err := s.topicsFromPath(r.URL.Path)
  1457. if err != nil {
  1458. return err
  1459. }
  1460. for _, t := range topics {
  1461. if err := s.userManager.Authorize(v.user, t.ID, perm); err != nil {
  1462. log.Info("unauthorized: %s", err.Error())
  1463. return errHTTPForbidden
  1464. }
  1465. }
  1466. return next(w, r, v)
  1467. }
  1468. }
  1469. // visitor creates or retrieves a rate.Limiter for the given visitor.
  1470. // Note that this function will always return a visitor, even if an error occurs.
  1471. func (s *Server) visitor(r *http.Request) (v *visitor, err error) {
  1472. ip := extractIPAddress(r, s.config.BehindProxy)
  1473. var u *user.User // may stay nil if no auth header!
  1474. if u, err = s.authenticate(r); err != nil {
  1475. log.Debug("authentication failed: %s", err.Error())
  1476. err = errHTTPUnauthorized // Always return visitor, even when error occurs!
  1477. }
  1478. if u != nil {
  1479. v = s.visitorFromUser(u, ip)
  1480. } else {
  1481. v = s.visitorFromIP(ip)
  1482. }
  1483. v.mu.Lock()
  1484. v.user = u
  1485. v.mu.Unlock()
  1486. return v, err // Always return visitor, even when error occurs!
  1487. }
  1488. // authenticate a user based on basic auth username/password (Authorization: Basic ...), or token auth (Authorization: Bearer ...).
  1489. // The Authorization header can be passed as a header or the ?auth=... query param. The latter is required only to
  1490. // support the WebSocket JavaScript class, which does not support passing headers during the initial request. The auth
  1491. // query param is effectively double base64 encoded. Its format is base64(Basic base64(user:pass)).
  1492. func (s *Server) authenticate(r *http.Request) (user *user.User, err error) {
  1493. value := strings.TrimSpace(r.Header.Get("Authorization"))
  1494. queryParam := readQueryParam(r, "authorization", "auth")
  1495. if queryParam != "" {
  1496. a, err := base64.RawURLEncoding.DecodeString(queryParam)
  1497. if err != nil {
  1498. return nil, err
  1499. }
  1500. value = strings.TrimSpace(string(a))
  1501. }
  1502. if value == "" {
  1503. return nil, nil
  1504. } else if s.userManager == nil {
  1505. return nil, errHTTPUnauthorized
  1506. }
  1507. if strings.HasPrefix(value, "Bearer") {
  1508. return s.authenticateBearerAuth(value)
  1509. }
  1510. return s.authenticateBasicAuth(r, value)
  1511. }
  1512. func (s *Server) authenticateBasicAuth(r *http.Request, value string) (user *user.User, err error) {
  1513. r.Header.Set("Authorization", value)
  1514. username, password, ok := r.BasicAuth()
  1515. if !ok {
  1516. return nil, errors.New("invalid basic auth")
  1517. }
  1518. return s.userManager.Authenticate(username, password)
  1519. }
  1520. func (s *Server) authenticateBearerAuth(value string) (user *user.User, err error) {
  1521. token := strings.TrimSpace(strings.TrimPrefix(value, "Bearer"))
  1522. return s.userManager.AuthenticateToken(token)
  1523. }
  1524. func (s *Server) visitorFromID(visitorID string, ip netip.Addr, user *user.User) *visitor {
  1525. s.mu.Lock()
  1526. defer s.mu.Unlock()
  1527. v, exists := s.visitors[visitorID]
  1528. if !exists {
  1529. s.visitors[visitorID] = newVisitor(s.config, s.messageCache, s.userManager, ip, user)
  1530. return s.visitors[visitorID]
  1531. }
  1532. v.Keepalive()
  1533. return v
  1534. }
  1535. func (s *Server) visitorFromIP(ip netip.Addr) *visitor {
  1536. return s.visitorFromID(fmt.Sprintf("ip:%s", ip.String()), ip, nil)
  1537. }
  1538. func (s *Server) visitorFromUser(user *user.User, ip netip.Addr) *visitor {
  1539. return s.visitorFromID(fmt.Sprintf("user:%s", user.Name), ip, user)
  1540. }