server.go 55 KB

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