server.go 57 KB

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