server.go 57 KB

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