1
0

server.go 53 KB

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