manager.go 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356
  1. package user
  2. import (
  3. "database/sql"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. _ "github.com/mattn/go-sqlite3" // SQLite driver
  8. "github.com/stripe/stripe-go/v74"
  9. "golang.org/x/crypto/bcrypt"
  10. "heckel.io/ntfy/log"
  11. "heckel.io/ntfy/util"
  12. "net/netip"
  13. "strings"
  14. "sync"
  15. "time"
  16. )
  17. const (
  18. tierIDPrefix = "ti_"
  19. tierIDLength = 8
  20. syncTopicPrefix = "st_"
  21. syncTopicLength = 16
  22. userIDPrefix = "u_"
  23. userIDLength = 12
  24. userAuthIntentionalSlowDownHash = "$2a$10$YFCQvqQDwIIwnJM1xkAYOeih0dg17UVGanaTStnrSzC8NCWxcLDwy" // Cost should match DefaultUserPasswordBcryptCost
  25. userHardDeleteAfterDuration = 7 * 24 * time.Hour
  26. tokenPrefix = "tk_"
  27. tokenLength = 32
  28. tokenMaxCount = 20 // Only keep this many tokens in the table per user
  29. tagManager = "user_manager"
  30. )
  31. // Default constants that may be overridden by configs
  32. const (
  33. DefaultUserStatsQueueWriterInterval = 33 * time.Second
  34. DefaultUserPasswordBcryptCost = 10
  35. )
  36. var (
  37. errNoTokenProvided = errors.New("no token provided")
  38. errTopicOwnedByOthers = errors.New("topic owned by others")
  39. errNoRows = errors.New("no rows found")
  40. )
  41. // Manager-related queries
  42. const (
  43. createTablesQueriesNoTx = `
  44. CREATE TABLE IF NOT EXISTS tier (
  45. id TEXT PRIMARY KEY,
  46. code TEXT NOT NULL,
  47. name TEXT NOT NULL,
  48. messages_limit INT NOT NULL,
  49. messages_expiry_duration INT NOT NULL,
  50. emails_limit INT NOT NULL,
  51. reservations_limit INT NOT NULL,
  52. attachment_file_size_limit INT NOT NULL,
  53. attachment_total_size_limit INT NOT NULL,
  54. attachment_expiry_duration INT NOT NULL,
  55. attachment_bandwidth_limit INT NOT NULL,
  56. stripe_price_id TEXT
  57. );
  58. CREATE UNIQUE INDEX idx_tier_code ON tier (code);
  59. CREATE UNIQUE INDEX idx_tier_price_id ON tier (stripe_price_id);
  60. CREATE TABLE IF NOT EXISTS user (
  61. id TEXT PRIMARY KEY,
  62. tier_id TEXT,
  63. user TEXT NOT NULL,
  64. pass TEXT NOT NULL,
  65. role TEXT CHECK (role IN ('anonymous', 'admin', 'user')) NOT NULL,
  66. prefs JSON NOT NULL DEFAULT '{}',
  67. sync_topic TEXT NOT NULL,
  68. stats_messages INT NOT NULL DEFAULT (0),
  69. stats_emails INT NOT NULL DEFAULT (0),
  70. stripe_customer_id TEXT,
  71. stripe_subscription_id TEXT,
  72. stripe_subscription_status TEXT,
  73. stripe_subscription_paid_until INT,
  74. stripe_subscription_cancel_at INT,
  75. created INT NOT NULL,
  76. deleted INT,
  77. FOREIGN KEY (tier_id) REFERENCES tier (id)
  78. );
  79. CREATE UNIQUE INDEX idx_user ON user (user);
  80. CREATE UNIQUE INDEX idx_user_stripe_customer_id ON user (stripe_customer_id);
  81. CREATE UNIQUE INDEX idx_user_stripe_subscription_id ON user (stripe_subscription_id);
  82. CREATE TABLE IF NOT EXISTS user_access (
  83. user_id TEXT NOT NULL,
  84. topic TEXT NOT NULL,
  85. read INT NOT NULL,
  86. write INT NOT NULL,
  87. owner_user_id INT,
  88. PRIMARY KEY (user_id, topic),
  89. FOREIGN KEY (user_id) REFERENCES user (id) ON DELETE CASCADE,
  90. FOREIGN KEY (owner_user_id) REFERENCES user (id) ON DELETE CASCADE
  91. );
  92. CREATE TABLE IF NOT EXISTS user_token (
  93. user_id TEXT NOT NULL,
  94. token TEXT NOT NULL,
  95. label TEXT NOT NULL,
  96. last_access INT NOT NULL,
  97. last_origin TEXT NOT NULL,
  98. expires INT NOT NULL,
  99. PRIMARY KEY (user_id, token),
  100. FOREIGN KEY (user_id) REFERENCES user (id) ON DELETE CASCADE
  101. );
  102. CREATE TABLE IF NOT EXISTS schemaVersion (
  103. id INT PRIMARY KEY,
  104. version INT NOT NULL
  105. );
  106. INSERT INTO user (id, user, pass, role, sync_topic, created)
  107. VALUES ('` + everyoneID + `', '*', '', 'anonymous', '', UNIXEPOCH())
  108. ON CONFLICT (id) DO NOTHING;
  109. `
  110. createTablesQueries = `BEGIN; ` + createTablesQueriesNoTx + ` COMMIT;`
  111. builtinStartupQueries = `
  112. PRAGMA foreign_keys = ON;
  113. `
  114. selectUserByIDQuery = `
  115. SELECT u.id, u.user, u.pass, u.role, u.prefs, u.sync_topic, u.stats_messages, u.stats_emails, u.stripe_customer_id, u.stripe_subscription_id, u.stripe_subscription_status, u.stripe_subscription_paid_until, u.stripe_subscription_cancel_at, deleted, t.id, t.code, t.name, t.messages_limit, t.messages_expiry_duration, t.emails_limit, t.reservations_limit, t.attachment_file_size_limit, t.attachment_total_size_limit, t.attachment_expiry_duration, t.attachment_bandwidth_limit, t.stripe_price_id
  116. FROM user u
  117. LEFT JOIN tier t on t.id = u.tier_id
  118. WHERE u.id = ?
  119. `
  120. selectUserByNameQuery = `
  121. SELECT u.id, u.user, u.pass, u.role, u.prefs, u.sync_topic, u.stats_messages, u.stats_emails, u.stripe_customer_id, u.stripe_subscription_id, u.stripe_subscription_status, u.stripe_subscription_paid_until, u.stripe_subscription_cancel_at, deleted, t.id, t.code, t.name, t.messages_limit, t.messages_expiry_duration, t.emails_limit, t.reservations_limit, t.attachment_file_size_limit, t.attachment_total_size_limit, t.attachment_expiry_duration, t.attachment_bandwidth_limit, t.stripe_price_id
  122. FROM user u
  123. LEFT JOIN tier t on t.id = u.tier_id
  124. WHERE user = ?
  125. `
  126. selectUserByTokenQuery = `
  127. SELECT u.id, u.user, u.pass, u.role, u.prefs, u.sync_topic, u.stats_messages, u.stats_emails, u.stripe_customer_id, u.stripe_subscription_id, u.stripe_subscription_status, u.stripe_subscription_paid_until, u.stripe_subscription_cancel_at, deleted, t.id, t.code, t.name, t.messages_limit, t.messages_expiry_duration, t.emails_limit, t.reservations_limit, t.attachment_file_size_limit, t.attachment_total_size_limit, t.attachment_expiry_duration, t.attachment_bandwidth_limit, t.stripe_price_id
  128. FROM user u
  129. JOIN user_token tk on u.id = tk.user_id
  130. LEFT JOIN tier t on t.id = u.tier_id
  131. WHERE tk.token = ? AND (tk.expires = 0 OR tk.expires >= ?)
  132. `
  133. selectUserByStripeCustomerIDQuery = `
  134. SELECT u.id, u.user, u.pass, u.role, u.prefs, u.sync_topic, u.stats_messages, u.stats_emails, u.stripe_customer_id, u.stripe_subscription_id, u.stripe_subscription_status, u.stripe_subscription_paid_until, u.stripe_subscription_cancel_at, deleted, t.id, t.code, t.name, t.messages_limit, t.messages_expiry_duration, t.emails_limit, t.reservations_limit, t.attachment_file_size_limit, t.attachment_total_size_limit, t.attachment_expiry_duration, t.attachment_bandwidth_limit, t.stripe_price_id
  135. FROM user u
  136. LEFT JOIN tier t on t.id = u.tier_id
  137. WHERE u.stripe_customer_id = ?
  138. `
  139. selectTopicPermsQuery = `
  140. SELECT read, write
  141. FROM user_access a
  142. JOIN user u ON u.id = a.user_id
  143. WHERE (u.user = ? OR u.user = ?) AND ? LIKE a.topic
  144. ORDER BY u.user DESC
  145. `
  146. insertUserQuery = `
  147. INSERT INTO user (id, user, pass, role, sync_topic, created)
  148. VALUES (?, ?, ?, ?, ?, ?)
  149. `
  150. selectUsernamesQuery = `
  151. SELECT user
  152. FROM user
  153. ORDER BY
  154. CASE role
  155. WHEN 'admin' THEN 1
  156. WHEN 'anonymous' THEN 3
  157. ELSE 2
  158. END, user
  159. `
  160. updateUserPassQuery = `UPDATE user SET pass = ? WHERE user = ?`
  161. updateUserRoleQuery = `UPDATE user SET role = ? WHERE user = ?`
  162. updateUserPrefsQuery = `UPDATE user SET prefs = ? WHERE user = ?`
  163. updateUserStatsQuery = `UPDATE user SET stats_messages = ?, stats_emails = ? WHERE id = ?`
  164. updateUserStatsResetAllQuery = `UPDATE user SET stats_messages = 0, stats_emails = 0`
  165. updateUserDeletedQuery = `UPDATE user SET deleted = ? WHERE id = ?`
  166. deleteUsersMarkedQuery = `DELETE FROM user WHERE deleted < ?`
  167. deleteUserQuery = `DELETE FROM user WHERE user = ?`
  168. upsertUserAccessQuery = `
  169. INSERT INTO user_access (user_id, topic, read, write, owner_user_id)
  170. VALUES ((SELECT id FROM user WHERE user = ?), ?, ?, ?, (SELECT IIF(?='',NULL,(SELECT id FROM user WHERE user=?))))
  171. ON CONFLICT (user_id, topic)
  172. DO UPDATE SET read=excluded.read, write=excluded.write, owner_user_id=excluded.owner_user_id
  173. `
  174. selectUserAccessQuery = `
  175. SELECT topic, read, write
  176. FROM user_access
  177. WHERE user_id = (SELECT id FROM user WHERE user = ?)
  178. ORDER BY write DESC, read DESC, topic
  179. `
  180. selectUserReservationsQuery = `
  181. SELECT a_user.topic, a_user.read, a_user.write, a_everyone.read AS everyone_read, a_everyone.write AS everyone_write
  182. FROM user_access a_user
  183. LEFT JOIN user_access a_everyone ON a_user.topic = a_everyone.topic AND a_everyone.user_id = (SELECT id FROM user WHERE user = ?)
  184. WHERE a_user.user_id = a_user.owner_user_id
  185. AND a_user.owner_user_id = (SELECT id FROM user WHERE user = ?)
  186. ORDER BY a_user.topic
  187. `
  188. selectUserReservationsCountQuery = `
  189. SELECT COUNT(*)
  190. FROM user_access
  191. WHERE user_id = owner_user_id AND owner_user_id = (SELECT id FROM user WHERE user = ?)
  192. `
  193. selectUserHasReservationQuery = `
  194. SELECT COUNT(*)
  195. FROM user_access
  196. WHERE user_id = owner_user_id
  197. AND owner_user_id = (SELECT id FROM user WHERE user = ?)
  198. AND topic = ?
  199. `
  200. selectOtherAccessCountQuery = `
  201. SELECT COUNT(*)
  202. FROM user_access
  203. WHERE (topic = ? OR ? LIKE topic)
  204. AND (owner_user_id IS NULL OR owner_user_id != (SELECT id FROM user WHERE user = ?))
  205. `
  206. deleteAllAccessQuery = `DELETE FROM user_access`
  207. deleteUserAccessQuery = `
  208. DELETE FROM user_access
  209. WHERE user_id = (SELECT id FROM user WHERE user = ?)
  210. OR owner_user_id = (SELECT id FROM user WHERE user = ?)
  211. `
  212. deleteTopicAccessQuery = `
  213. DELETE FROM user_access
  214. WHERE (user_id = (SELECT id FROM user WHERE user = ?) OR owner_user_id = (SELECT id FROM user WHERE user = ?))
  215. AND topic = ?
  216. `
  217. selectTokenCountQuery = `SELECT COUNT(*) FROM user_token WHERE user_id = ?`
  218. selectTokensQuery = `SELECT token, label, last_access, last_origin, expires FROM user_token WHERE user_id = ?`
  219. selectTokenQuery = `SELECT token, label, last_access, last_origin, expires FROM user_token WHERE user_id = ? AND token = ?`
  220. insertTokenQuery = `INSERT INTO user_token (user_id, token, label, last_access, last_origin, expires) VALUES (?, ?, ?, ?, ?, ?)`
  221. updateTokenExpiryQuery = `UPDATE user_token SET expires = ? WHERE user_id = ? AND token = ?`
  222. updateTokenLabelQuery = `UPDATE user_token SET label = ? WHERE user_id = ? AND token = ?`
  223. updateTokenLastAccessQuery = `UPDATE user_token SET last_access = ?, last_origin = ? WHERE token = ?`
  224. deleteTokenQuery = `DELETE FROM user_token WHERE user_id = ? AND token = ?`
  225. deleteAllTokenQuery = `DELETE FROM user_token WHERE user_id = ?`
  226. deleteExpiredTokensQuery = `DELETE FROM user_token WHERE expires > 0 AND expires < ?`
  227. deleteExcessTokensQuery = `
  228. DELETE FROM user_token
  229. WHERE (user_id, token) NOT IN (
  230. SELECT user_id, token
  231. FROM user_token
  232. WHERE user_id = ?
  233. ORDER BY expires DESC
  234. LIMIT ?
  235. )
  236. `
  237. insertTierQuery = `
  238. INSERT INTO tier (id, code, name, messages_limit, messages_expiry_duration, emails_limit, reservations_limit, attachment_file_size_limit, attachment_total_size_limit, attachment_expiry_duration, attachment_bandwidth_limit, stripe_price_id)
  239. VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
  240. `
  241. updateTierQuery = `
  242. UPDATE tier
  243. SET name = ?, messages_limit = ?, messages_expiry_duration = ?, emails_limit = ?, reservations_limit = ?, attachment_file_size_limit = ?, attachment_total_size_limit = ?, attachment_expiry_duration = ?, attachment_bandwidth_limit = ?, stripe_price_id = ?
  244. WHERE code = ?
  245. `
  246. selectTiersQuery = `
  247. SELECT id, code, name, messages_limit, messages_expiry_duration, emails_limit, reservations_limit, attachment_file_size_limit, attachment_total_size_limit, attachment_expiry_duration, attachment_bandwidth_limit, stripe_price_id
  248. FROM tier
  249. `
  250. selectTierByCodeQuery = `
  251. SELECT id, code, name, messages_limit, messages_expiry_duration, emails_limit, reservations_limit, attachment_file_size_limit, attachment_total_size_limit, attachment_expiry_duration, attachment_bandwidth_limit, stripe_price_id
  252. FROM tier
  253. WHERE code = ?
  254. `
  255. selectTierByPriceIDQuery = `
  256. SELECT id, code, name, messages_limit, messages_expiry_duration, emails_limit, reservations_limit, attachment_file_size_limit, attachment_total_size_limit, attachment_expiry_duration, attachment_bandwidth_limit, stripe_price_id
  257. FROM tier
  258. WHERE stripe_price_id = ?
  259. `
  260. updateUserTierQuery = `UPDATE user SET tier_id = (SELECT id FROM tier WHERE code = ?) WHERE user = ?`
  261. deleteUserTierQuery = `UPDATE user SET tier_id = null WHERE user = ?`
  262. deleteTierQuery = `DELETE FROM tier WHERE code = ?`
  263. updateBillingQuery = `
  264. UPDATE user
  265. SET stripe_customer_id = ?, stripe_subscription_id = ?, stripe_subscription_status = ?, stripe_subscription_paid_until = ?, stripe_subscription_cancel_at = ?
  266. WHERE user = ?
  267. `
  268. )
  269. // Schema management queries
  270. const (
  271. currentSchemaVersion = 2
  272. insertSchemaVersion = `INSERT INTO schemaVersion VALUES (1, ?)`
  273. updateSchemaVersion = `UPDATE schemaVersion SET version = ? WHERE id = 1`
  274. selectSchemaVersionQuery = `SELECT version FROM schemaVersion WHERE id = 1`
  275. // 1 -> 2 (complex migration!)
  276. migrate1To2RenameUserTableQueryNoTx = `
  277. ALTER TABLE user RENAME TO user_old;
  278. `
  279. migrate1To2SelectAllOldUsernamesNoTx = `SELECT user FROM user_old`
  280. migrate1To2InsertUserNoTx = `
  281. INSERT INTO user (id, user, pass, role, sync_topic, created)
  282. SELECT ?, user, pass, role, ?, UNIXEPOCH() FROM user_old WHERE user = ?
  283. `
  284. migrate1To2InsertFromOldTablesAndDropNoTx = `
  285. INSERT INTO user_access (user_id, topic, read, write)
  286. SELECT u.id, a.topic, a.read, a.write
  287. FROM user u
  288. JOIN access a ON u.user = a.user;
  289. DROP TABLE access;
  290. DROP TABLE user_old;
  291. `
  292. )
  293. // Manager is an implementation of Manager. It stores users and access control list
  294. // in a SQLite database.
  295. type Manager struct {
  296. db *sql.DB
  297. defaultAccess Permission // Default permission if no ACL matches
  298. statsQueue map[string]*Stats // "Queue" to asynchronously write user stats to the database (UserID -> Stats)
  299. tokenQueue map[string]*TokenUpdate // "Queue" to asynchronously write token access stats to the database (Token ID -> TokenUpdate)
  300. bcryptCost int // Makes testing easier
  301. mu sync.Mutex
  302. }
  303. var _ Auther = (*Manager)(nil)
  304. // NewManager creates a new Manager instance
  305. func NewManager(filename, startupQueries string, defaultAccess Permission, bcryptCost int, queueWriterInterval time.Duration) (*Manager, error) {
  306. db, err := sql.Open("sqlite3", filename)
  307. if err != nil {
  308. return nil, err
  309. }
  310. if err := setupDB(db); err != nil {
  311. return nil, err
  312. }
  313. if err := runStartupQueries(db, startupQueries); err != nil {
  314. return nil, err
  315. }
  316. manager := &Manager{
  317. db: db,
  318. defaultAccess: defaultAccess,
  319. statsQueue: make(map[string]*Stats),
  320. tokenQueue: make(map[string]*TokenUpdate),
  321. bcryptCost: bcryptCost,
  322. }
  323. go manager.asyncQueueWriter(queueWriterInterval)
  324. return manager, nil
  325. }
  326. // Authenticate checks username and password and returns a User if correct, and the user has not been
  327. // marked as deleted. The method returns in constant-ish time, regardless of whether the user exists or
  328. // the password is correct or incorrect.
  329. func (a *Manager) Authenticate(username, password string) (*User, error) {
  330. if username == Everyone {
  331. return nil, ErrUnauthenticated
  332. }
  333. user, err := a.User(username)
  334. if err != nil {
  335. log.Tag(tagManager).Field("user_name", username).Err(err).Trace("Authentication of user failed (1)")
  336. bcrypt.CompareHashAndPassword([]byte(userAuthIntentionalSlowDownHash), []byte("intentional slow-down to avoid timing attacks"))
  337. return nil, ErrUnauthenticated
  338. } else if user.Deleted {
  339. log.Tag(tagManager).Field("user_name", username).Trace("Authentication of user failed (2): user marked deleted")
  340. bcrypt.CompareHashAndPassword([]byte(userAuthIntentionalSlowDownHash), []byte("intentional slow-down to avoid timing attacks"))
  341. return nil, ErrUnauthenticated
  342. } else if err := bcrypt.CompareHashAndPassword([]byte(user.Hash), []byte(password)); err != nil {
  343. log.Tag(tagManager).Field("user_name", username).Err(err).Trace("Authentication of user failed (3)")
  344. return nil, ErrUnauthenticated
  345. }
  346. return user, nil
  347. }
  348. // AuthenticateToken checks if the token exists and returns the associated User if it does.
  349. // The method sets the User.Token value to the token that was used for authentication.
  350. func (a *Manager) AuthenticateToken(token string) (*User, error) {
  351. if len(token) != tokenLength {
  352. return nil, ErrUnauthenticated
  353. }
  354. user, err := a.userByToken(token)
  355. if err != nil {
  356. return nil, ErrUnauthenticated
  357. }
  358. user.Token = token
  359. return user, nil
  360. }
  361. // CreateToken generates a random token for the given user and returns it. The token expires
  362. // after a fixed duration unless ChangeToken is called. This function also prunes tokens for the
  363. // given user, if there are too many of them.
  364. func (a *Manager) CreateToken(userID, label string, expires time.Time, origin netip.Addr) (*Token, error) {
  365. token := util.RandomStringPrefix(tokenPrefix, tokenLength)
  366. tx, err := a.db.Begin()
  367. if err != nil {
  368. return nil, err
  369. }
  370. defer tx.Rollback()
  371. access := time.Now()
  372. if _, err := tx.Exec(insertTokenQuery, userID, token, label, access.Unix(), origin.String(), expires.Unix()); err != nil {
  373. return nil, err
  374. }
  375. rows, err := tx.Query(selectTokenCountQuery, userID)
  376. if err != nil {
  377. return nil, err
  378. }
  379. defer rows.Close()
  380. if !rows.Next() {
  381. return nil, errNoRows
  382. }
  383. var tokenCount int
  384. if err := rows.Scan(&tokenCount); err != nil {
  385. return nil, err
  386. }
  387. if tokenCount >= tokenMaxCount {
  388. // This pruning logic is done in two queries for efficiency. The SELECT above is a lookup
  389. // on two indices, whereas the query below is a full table scan.
  390. if _, err := tx.Exec(deleteExcessTokensQuery, userID, tokenMaxCount); err != nil {
  391. return nil, err
  392. }
  393. }
  394. if err := tx.Commit(); err != nil {
  395. return nil, err
  396. }
  397. return &Token{
  398. Value: token,
  399. Label: label,
  400. LastAccess: access,
  401. LastOrigin: origin,
  402. Expires: expires,
  403. }, nil
  404. }
  405. // Tokens returns all existing tokens for the user with the given user ID
  406. func (a *Manager) Tokens(userID string) ([]*Token, error) {
  407. rows, err := a.db.Query(selectTokensQuery, userID)
  408. if err != nil {
  409. return nil, err
  410. }
  411. defer rows.Close()
  412. tokens := make([]*Token, 0)
  413. for {
  414. token, err := a.readToken(rows)
  415. if err == ErrTokenNotFound {
  416. break
  417. } else if err != nil {
  418. return nil, err
  419. }
  420. tokens = append(tokens, token)
  421. }
  422. return tokens, nil
  423. }
  424. // Token returns a specific token for a user
  425. func (a *Manager) Token(userID, token string) (*Token, error) {
  426. rows, err := a.db.Query(selectTokenQuery, userID, token)
  427. if err != nil {
  428. return nil, err
  429. }
  430. defer rows.Close()
  431. return a.readToken(rows)
  432. }
  433. func (a *Manager) readToken(rows *sql.Rows) (*Token, error) {
  434. var token, label, lastOrigin string
  435. var lastAccess, expires int64
  436. if !rows.Next() {
  437. return nil, ErrTokenNotFound
  438. }
  439. if err := rows.Scan(&token, &label, &lastAccess, &lastOrigin, &expires); err != nil {
  440. return nil, err
  441. } else if err := rows.Err(); err != nil {
  442. return nil, err
  443. }
  444. lastOriginIP, err := netip.ParseAddr(lastOrigin)
  445. if err != nil {
  446. lastOriginIP = netip.IPv4Unspecified()
  447. }
  448. return &Token{
  449. Value: token,
  450. Label: label,
  451. LastAccess: time.Unix(lastAccess, 0),
  452. LastOrigin: lastOriginIP,
  453. Expires: time.Unix(expires, 0),
  454. }, nil
  455. }
  456. // ChangeToken updates a token's label and/or expiry date
  457. func (a *Manager) ChangeToken(userID, token string, label *string, expires *time.Time) (*Token, error) {
  458. if token == "" {
  459. return nil, errNoTokenProvided
  460. }
  461. tx, err := a.db.Begin()
  462. if err != nil {
  463. return nil, err
  464. }
  465. defer tx.Rollback()
  466. if label != nil {
  467. if _, err := tx.Exec(updateTokenLabelQuery, *label, userID, token); err != nil {
  468. return nil, err
  469. }
  470. }
  471. if expires != nil {
  472. if _, err := tx.Exec(updateTokenExpiryQuery, expires.Unix(), userID, token); err != nil {
  473. return nil, err
  474. }
  475. }
  476. if err := tx.Commit(); err != nil {
  477. return nil, err
  478. }
  479. return a.Token(userID, token)
  480. }
  481. // RemoveToken deletes the token defined in User.Token
  482. func (a *Manager) RemoveToken(userID, token string) error {
  483. if token == "" {
  484. return errNoTokenProvided
  485. }
  486. if _, err := a.db.Exec(deleteTokenQuery, userID, token); err != nil {
  487. return err
  488. }
  489. return nil
  490. }
  491. // RemoveExpiredTokens deletes all expired tokens from the database
  492. func (a *Manager) RemoveExpiredTokens() error {
  493. if _, err := a.db.Exec(deleteExpiredTokensQuery, time.Now().Unix()); err != nil {
  494. return err
  495. }
  496. return nil
  497. }
  498. // RemoveDeletedUsers deletes all users that have been marked deleted for
  499. func (a *Manager) RemoveDeletedUsers() error {
  500. if _, err := a.db.Exec(deleteUsersMarkedQuery, time.Now().Unix()); err != nil {
  501. return err
  502. }
  503. return nil
  504. }
  505. // ChangeSettings persists the user settings
  506. func (a *Manager) ChangeSettings(user *User) error {
  507. prefs, err := json.Marshal(user.Prefs)
  508. if err != nil {
  509. return err
  510. }
  511. if _, err := a.db.Exec(updateUserPrefsQuery, string(prefs), user.Name); err != nil {
  512. return err
  513. }
  514. return nil
  515. }
  516. // ResetStats resets all user stats in the user database. This touches all users.
  517. func (a *Manager) ResetStats() error {
  518. a.mu.Lock() // Includes database query to avoid races!
  519. defer a.mu.Unlock()
  520. if _, err := a.db.Exec(updateUserStatsResetAllQuery); err != nil {
  521. return err
  522. }
  523. a.statsQueue = make(map[string]*Stats)
  524. return nil
  525. }
  526. // EnqueueStats adds the user to a queue which writes out user stats (messages, emails, ..) in
  527. // batches at a regular interval
  528. func (a *Manager) EnqueueStats(userID string, stats *Stats) {
  529. a.mu.Lock()
  530. defer a.mu.Unlock()
  531. a.statsQueue[userID] = stats
  532. }
  533. // EnqueueTokenUpdate adds the token update to a queue which writes out token access times
  534. // in batches at a regular interval
  535. func (a *Manager) EnqueueTokenUpdate(tokenID string, update *TokenUpdate) {
  536. a.mu.Lock()
  537. defer a.mu.Unlock()
  538. a.tokenQueue[tokenID] = update
  539. }
  540. func (a *Manager) asyncQueueWriter(interval time.Duration) {
  541. ticker := time.NewTicker(interval)
  542. for range ticker.C {
  543. if err := a.writeUserStatsQueue(); err != nil {
  544. log.Tag(tagManager).Err(err).Warn("Writing user stats queue failed")
  545. }
  546. if err := a.writeTokenUpdateQueue(); err != nil {
  547. log.Tag(tagManager).Err(err).Warn("Writing token update queue failed")
  548. }
  549. }
  550. }
  551. func (a *Manager) writeUserStatsQueue() error {
  552. a.mu.Lock()
  553. if len(a.statsQueue) == 0 {
  554. a.mu.Unlock()
  555. log.Tag(tagManager).Trace("No user stats updates to commit")
  556. return nil
  557. }
  558. statsQueue := a.statsQueue
  559. a.statsQueue = make(map[string]*Stats)
  560. a.mu.Unlock()
  561. tx, err := a.db.Begin()
  562. if err != nil {
  563. return err
  564. }
  565. defer tx.Rollback()
  566. log.Tag(tagManager).Debug("Writing user stats queue for %d user(s)", len(statsQueue))
  567. for userID, update := range statsQueue {
  568. log.
  569. Tag(tagManager).
  570. Fields(log.Context{
  571. "user_id": userID,
  572. "messages_count": update.Messages,
  573. "emails_count": update.Emails,
  574. }).
  575. Trace("Updating stats for user %s", userID)
  576. if _, err := tx.Exec(updateUserStatsQuery, update.Messages, update.Emails, userID); err != nil {
  577. return err
  578. }
  579. }
  580. return tx.Commit()
  581. }
  582. func (a *Manager) writeTokenUpdateQueue() error {
  583. a.mu.Lock()
  584. if len(a.tokenQueue) == 0 {
  585. a.mu.Unlock()
  586. log.Tag(tagManager).Trace("No token updates to commit")
  587. return nil
  588. }
  589. tokenQueue := a.tokenQueue
  590. a.tokenQueue = make(map[string]*TokenUpdate)
  591. a.mu.Unlock()
  592. tx, err := a.db.Begin()
  593. if err != nil {
  594. return err
  595. }
  596. defer tx.Rollback()
  597. log.Tag(tagManager).Debug("Writing token update queue for %d token(s)", len(tokenQueue))
  598. for tokenID, update := range tokenQueue {
  599. log.Tag(tagManager).Trace("Updating token %s with last access time %v", tokenID, update.LastAccess.Unix())
  600. if _, err := tx.Exec(updateTokenLastAccessQuery, update.LastAccess.Unix(), update.LastOrigin.String(), tokenID); err != nil {
  601. return err
  602. }
  603. }
  604. return tx.Commit()
  605. }
  606. // Authorize returns nil if the given user has access to the given topic using the desired
  607. // permission. The user param may be nil to signal an anonymous user.
  608. func (a *Manager) Authorize(user *User, topic string, perm Permission) error {
  609. if user != nil && user.Role == RoleAdmin {
  610. return nil // Admin can do everything
  611. }
  612. username := Everyone
  613. if user != nil {
  614. username = user.Name
  615. }
  616. // Select the read/write permissions for this user/topic combo. The query may return two
  617. // rows (one for everyone, and one for the user), but prioritizes the user.
  618. rows, err := a.db.Query(selectTopicPermsQuery, Everyone, username, topic)
  619. if err != nil {
  620. return err
  621. }
  622. defer rows.Close()
  623. if !rows.Next() {
  624. return a.resolvePerms(a.defaultAccess, perm)
  625. }
  626. var read, write bool
  627. if err := rows.Scan(&read, &write); err != nil {
  628. return err
  629. } else if err := rows.Err(); err != nil {
  630. return err
  631. }
  632. return a.resolvePerms(NewPermission(read, write), perm)
  633. }
  634. func (a *Manager) resolvePerms(base, perm Permission) error {
  635. if perm == PermissionRead && base.IsRead() {
  636. return nil
  637. } else if perm == PermissionWrite && base.IsWrite() {
  638. return nil
  639. }
  640. return ErrUnauthorized
  641. }
  642. // AddUser adds a user with the given username, password and role
  643. func (a *Manager) AddUser(username, password string, role Role) error {
  644. if !AllowedUsername(username) || !AllowedRole(role) {
  645. return ErrInvalidArgument
  646. }
  647. hash, err := bcrypt.GenerateFromPassword([]byte(password), a.bcryptCost)
  648. if err != nil {
  649. return err
  650. }
  651. userID := util.RandomStringPrefix(userIDPrefix, userIDLength)
  652. syncTopic, now := util.RandomStringPrefix(syncTopicPrefix, syncTopicLength), time.Now().Unix()
  653. if _, err = a.db.Exec(insertUserQuery, userID, username, hash, role, syncTopic, now); err != nil {
  654. return err
  655. }
  656. return nil
  657. }
  658. // RemoveUser deletes the user with the given username. The function returns nil on success, even
  659. // if the user did not exist in the first place.
  660. func (a *Manager) RemoveUser(username string) error {
  661. if !AllowedUsername(username) {
  662. return ErrInvalidArgument
  663. }
  664. // Rows in user_access, user_token, etc. are deleted via foreign keys
  665. if _, err := a.db.Exec(deleteUserQuery, username); err != nil {
  666. return err
  667. }
  668. return nil
  669. }
  670. // MarkUserRemoved sets the deleted flag on the user, and deletes all access tokens. This prevents
  671. // successful auth via Authenticate. A background process will delete the user at a later date.
  672. func (a *Manager) MarkUserRemoved(user *User) error {
  673. if !AllowedUsername(user.Name) {
  674. return ErrInvalidArgument
  675. }
  676. tx, err := a.db.Begin()
  677. if err != nil {
  678. return err
  679. }
  680. defer tx.Rollback()
  681. if _, err := a.db.Exec(deleteUserAccessQuery, user.Name, user.Name); err != nil {
  682. return err
  683. }
  684. if _, err := tx.Exec(deleteAllTokenQuery, user.ID); err != nil {
  685. return err
  686. }
  687. if _, err := tx.Exec(updateUserDeletedQuery, time.Now().Add(userHardDeleteAfterDuration).Unix(), user.ID); err != nil {
  688. return err
  689. }
  690. return tx.Commit()
  691. }
  692. // Users returns a list of users. It always also returns the Everyone user ("*").
  693. func (a *Manager) Users() ([]*User, error) {
  694. rows, err := a.db.Query(selectUsernamesQuery)
  695. if err != nil {
  696. return nil, err
  697. }
  698. defer rows.Close()
  699. usernames := make([]string, 0)
  700. for rows.Next() {
  701. var username string
  702. if err := rows.Scan(&username); err != nil {
  703. return nil, err
  704. } else if err := rows.Err(); err != nil {
  705. return nil, err
  706. }
  707. usernames = append(usernames, username)
  708. }
  709. rows.Close()
  710. users := make([]*User, 0)
  711. for _, username := range usernames {
  712. user, err := a.User(username)
  713. if err != nil {
  714. return nil, err
  715. }
  716. users = append(users, user)
  717. }
  718. return users, nil
  719. }
  720. // User returns the user with the given username if it exists, or ErrUserNotFound otherwise.
  721. // You may also pass Everyone to retrieve the anonymous user and its Grant list.
  722. func (a *Manager) User(username string) (*User, error) {
  723. rows, err := a.db.Query(selectUserByNameQuery, username)
  724. if err != nil {
  725. return nil, err
  726. }
  727. return a.readUser(rows)
  728. }
  729. // UserByID returns the user with the given ID if it exists, or ErrUserNotFound otherwise
  730. func (a *Manager) UserByID(id string) (*User, error) {
  731. rows, err := a.db.Query(selectUserByIDQuery, id)
  732. if err != nil {
  733. return nil, err
  734. }
  735. return a.readUser(rows)
  736. }
  737. // UserByStripeCustomer returns the user with the given Stripe customer ID if it exists, or ErrUserNotFound otherwise.
  738. func (a *Manager) UserByStripeCustomer(stripeCustomerID string) (*User, error) {
  739. rows, err := a.db.Query(selectUserByStripeCustomerIDQuery, stripeCustomerID)
  740. if err != nil {
  741. return nil, err
  742. }
  743. return a.readUser(rows)
  744. }
  745. func (a *Manager) userByToken(token string) (*User, error) {
  746. rows, err := a.db.Query(selectUserByTokenQuery, token, time.Now().Unix())
  747. if err != nil {
  748. return nil, err
  749. }
  750. return a.readUser(rows)
  751. }
  752. func (a *Manager) readUser(rows *sql.Rows) (*User, error) {
  753. defer rows.Close()
  754. var id, username, hash, role, prefs, syncTopic string
  755. var stripeCustomerID, stripeSubscriptionID, stripeSubscriptionStatus, stripePriceID, tierID, tierCode, tierName sql.NullString
  756. var messages, emails int64
  757. var messagesLimit, messagesExpiryDuration, emailsLimit, reservationsLimit, attachmentFileSizeLimit, attachmentTotalSizeLimit, attachmentExpiryDuration, attachmentBandwidthLimit, stripeSubscriptionPaidUntil, stripeSubscriptionCancelAt, deleted sql.NullInt64
  758. if !rows.Next() {
  759. return nil, ErrUserNotFound
  760. }
  761. if err := rows.Scan(&id, &username, &hash, &role, &prefs, &syncTopic, &messages, &emails, &stripeCustomerID, &stripeSubscriptionID, &stripeSubscriptionStatus, &stripeSubscriptionPaidUntil, &stripeSubscriptionCancelAt, &deleted, &tierID, &tierCode, &tierName, &messagesLimit, &messagesExpiryDuration, &emailsLimit, &reservationsLimit, &attachmentFileSizeLimit, &attachmentTotalSizeLimit, &attachmentExpiryDuration, &attachmentBandwidthLimit, &stripePriceID); err != nil {
  762. return nil, err
  763. } else if err := rows.Err(); err != nil {
  764. return nil, err
  765. }
  766. user := &User{
  767. ID: id,
  768. Name: username,
  769. Hash: hash,
  770. Role: Role(role),
  771. Prefs: &Prefs{},
  772. SyncTopic: syncTopic,
  773. Stats: &Stats{
  774. Messages: messages,
  775. Emails: emails,
  776. },
  777. Billing: &Billing{
  778. StripeCustomerID: stripeCustomerID.String, // May be empty
  779. StripeSubscriptionID: stripeSubscriptionID.String, // May be empty
  780. StripeSubscriptionStatus: stripe.SubscriptionStatus(stripeSubscriptionStatus.String), // May be empty
  781. StripeSubscriptionPaidUntil: time.Unix(stripeSubscriptionPaidUntil.Int64, 0), // May be zero
  782. StripeSubscriptionCancelAt: time.Unix(stripeSubscriptionCancelAt.Int64, 0), // May be zero
  783. },
  784. Deleted: deleted.Valid,
  785. }
  786. if err := json.Unmarshal([]byte(prefs), user.Prefs); err != nil {
  787. return nil, err
  788. }
  789. if tierCode.Valid {
  790. // See readTier() when this is changed!
  791. user.Tier = &Tier{
  792. ID: tierID.String,
  793. Code: tierCode.String,
  794. Name: tierName.String,
  795. MessageLimit: messagesLimit.Int64,
  796. MessageExpiryDuration: time.Duration(messagesExpiryDuration.Int64) * time.Second,
  797. EmailLimit: emailsLimit.Int64,
  798. ReservationLimit: reservationsLimit.Int64,
  799. AttachmentFileSizeLimit: attachmentFileSizeLimit.Int64,
  800. AttachmentTotalSizeLimit: attachmentTotalSizeLimit.Int64,
  801. AttachmentExpiryDuration: time.Duration(attachmentExpiryDuration.Int64) * time.Second,
  802. AttachmentBandwidthLimit: attachmentBandwidthLimit.Int64,
  803. StripePriceID: stripePriceID.String, // May be empty
  804. }
  805. }
  806. return user, nil
  807. }
  808. // Grants returns all user-specific access control entries
  809. func (a *Manager) Grants(username string) ([]Grant, error) {
  810. rows, err := a.db.Query(selectUserAccessQuery, username)
  811. if err != nil {
  812. return nil, err
  813. }
  814. defer rows.Close()
  815. grants := make([]Grant, 0)
  816. for rows.Next() {
  817. var topic string
  818. var read, write bool
  819. if err := rows.Scan(&topic, &read, &write); err != nil {
  820. return nil, err
  821. } else if err := rows.Err(); err != nil {
  822. return nil, err
  823. }
  824. grants = append(grants, Grant{
  825. TopicPattern: fromSQLWildcard(topic),
  826. Allow: NewPermission(read, write),
  827. })
  828. }
  829. return grants, nil
  830. }
  831. // Reservations returns all user-owned topics, and the associated everyone-access
  832. func (a *Manager) Reservations(username string) ([]Reservation, error) {
  833. rows, err := a.db.Query(selectUserReservationsQuery, Everyone, username)
  834. if err != nil {
  835. return nil, err
  836. }
  837. defer rows.Close()
  838. reservations := make([]Reservation, 0)
  839. for rows.Next() {
  840. var topic string
  841. var ownerRead, ownerWrite bool
  842. var everyoneRead, everyoneWrite sql.NullBool
  843. if err := rows.Scan(&topic, &ownerRead, &ownerWrite, &everyoneRead, &everyoneWrite); err != nil {
  844. return nil, err
  845. } else if err := rows.Err(); err != nil {
  846. return nil, err
  847. }
  848. reservations = append(reservations, Reservation{
  849. Topic: topic,
  850. Owner: NewPermission(ownerRead, ownerWrite),
  851. Everyone: NewPermission(everyoneRead.Bool, everyoneWrite.Bool), // false if null
  852. })
  853. }
  854. return reservations, nil
  855. }
  856. // HasReservation returns true if the given topic access is owned by the user
  857. func (a *Manager) HasReservation(username, topic string) (bool, error) {
  858. rows, err := a.db.Query(selectUserHasReservationQuery, username, topic)
  859. if err != nil {
  860. return false, err
  861. }
  862. defer rows.Close()
  863. if !rows.Next() {
  864. return false, errNoRows
  865. }
  866. var count int64
  867. if err := rows.Scan(&count); err != nil {
  868. return false, err
  869. }
  870. return count > 0, nil
  871. }
  872. // ReservationsCount returns the number of reservations owned by this user
  873. func (a *Manager) ReservationsCount(username string) (int64, error) {
  874. rows, err := a.db.Query(selectUserReservationsCountQuery, username)
  875. if err != nil {
  876. return 0, err
  877. }
  878. defer rows.Close()
  879. if !rows.Next() {
  880. return 0, errNoRows
  881. }
  882. var count int64
  883. if err := rows.Scan(&count); err != nil {
  884. return 0, err
  885. }
  886. return count, nil
  887. }
  888. // ChangePassword changes a user's password
  889. func (a *Manager) ChangePassword(username, password string) error {
  890. hash, err := bcrypt.GenerateFromPassword([]byte(password), a.bcryptCost)
  891. if err != nil {
  892. return err
  893. }
  894. if _, err := a.db.Exec(updateUserPassQuery, hash, username); err != nil {
  895. return err
  896. }
  897. return nil
  898. }
  899. // ChangeRole changes a user's role. When a role is changed from RoleUser to RoleAdmin,
  900. // all existing access control entries (Grant) are removed, since they are no longer needed.
  901. func (a *Manager) ChangeRole(username string, role Role) error {
  902. if !AllowedUsername(username) || !AllowedRole(role) {
  903. return ErrInvalidArgument
  904. }
  905. if _, err := a.db.Exec(updateUserRoleQuery, string(role), username); err != nil {
  906. return err
  907. }
  908. if role == RoleAdmin {
  909. if _, err := a.db.Exec(deleteUserAccessQuery, username, username); err != nil {
  910. return err
  911. }
  912. }
  913. return nil
  914. }
  915. // ChangeTier changes a user's tier using the tier code. This function does not delete reservations, messages,
  916. // or attachments, even if the new tier has lower limits in this regard. That has to be done elsewhere.
  917. func (a *Manager) ChangeTier(username, tier string) error {
  918. if !AllowedUsername(username) {
  919. return ErrInvalidArgument
  920. }
  921. t, err := a.Tier(tier)
  922. if err != nil {
  923. return err
  924. } else if err := a.checkReservationsLimit(username, t.ReservationLimit); err != nil {
  925. return err
  926. }
  927. if _, err := a.db.Exec(updateUserTierQuery, tier, username); err != nil {
  928. return err
  929. }
  930. return nil
  931. }
  932. // ResetTier removes the tier from the given user
  933. func (a *Manager) ResetTier(username string) error {
  934. if !AllowedUsername(username) && username != Everyone && username != "" {
  935. return ErrInvalidArgument
  936. } else if err := a.checkReservationsLimit(username, 0); err != nil {
  937. return err
  938. }
  939. _, err := a.db.Exec(deleteUserTierQuery, username)
  940. return err
  941. }
  942. func (a *Manager) checkReservationsLimit(username string, reservationsLimit int64) error {
  943. u, err := a.User(username)
  944. if err != nil {
  945. return err
  946. }
  947. if u.Tier != nil && reservationsLimit < u.Tier.ReservationLimit {
  948. reservations, err := a.Reservations(username)
  949. if err != nil {
  950. return err
  951. } else if int64(len(reservations)) > reservationsLimit {
  952. return ErrTooManyReservations
  953. }
  954. }
  955. return nil
  956. }
  957. // CheckAllowAccess tests if a user may create an access control entry for the given topic.
  958. // If there are any ACL entries that are not owned by the user, an error is returned.
  959. // FIXME is this the same as HasReservation?
  960. func (a *Manager) CheckAllowAccess(username string, topic string) error {
  961. if (!AllowedUsername(username) && username != Everyone) || !AllowedTopic(topic) {
  962. return ErrInvalidArgument
  963. }
  964. rows, err := a.db.Query(selectOtherAccessCountQuery, topic, topic, username)
  965. if err != nil {
  966. return err
  967. }
  968. defer rows.Close()
  969. if !rows.Next() {
  970. return errNoRows
  971. }
  972. var otherCount int
  973. if err := rows.Scan(&otherCount); err != nil {
  974. return err
  975. }
  976. if otherCount > 0 {
  977. return errTopicOwnedByOthers
  978. }
  979. return nil
  980. }
  981. // AllowAccess adds or updates an entry in th access control list for a specific user. It controls
  982. // read/write access to a topic. The parameter topicPattern may include wildcards (*). The ACL entry
  983. // owner may either be a user (username), or the system (empty).
  984. func (a *Manager) AllowAccess(username string, topicPattern string, permission Permission) error {
  985. if !AllowedUsername(username) && username != Everyone {
  986. return ErrInvalidArgument
  987. } else if !AllowedTopicPattern(topicPattern) {
  988. return ErrInvalidArgument
  989. }
  990. owner := ""
  991. if _, err := a.db.Exec(upsertUserAccessQuery, username, toSQLWildcard(topicPattern), permission.IsRead(), permission.IsWrite(), owner, owner); err != nil {
  992. return err
  993. }
  994. return nil
  995. }
  996. // ResetAccess removes an access control list entry for a specific username/topic, or (if topic is
  997. // empty) for an entire user. The parameter topicPattern may include wildcards (*).
  998. func (a *Manager) ResetAccess(username string, topicPattern string) error {
  999. if !AllowedUsername(username) && username != Everyone && username != "" {
  1000. return ErrInvalidArgument
  1001. } else if !AllowedTopicPattern(topicPattern) && topicPattern != "" {
  1002. return ErrInvalidArgument
  1003. }
  1004. if username == "" && topicPattern == "" {
  1005. _, err := a.db.Exec(deleteAllAccessQuery, username)
  1006. return err
  1007. } else if topicPattern == "" {
  1008. _, err := a.db.Exec(deleteUserAccessQuery, username, username)
  1009. return err
  1010. }
  1011. _, err := a.db.Exec(deleteTopicAccessQuery, username, username, toSQLWildcard(topicPattern))
  1012. return err
  1013. }
  1014. // AddReservation creates two access control entries for the given topic: one with full read/write access for the
  1015. // given user, and one for Everyone with the permission passed as everyone. The user also owns the entries, and
  1016. // can modify or delete them.
  1017. func (a *Manager) AddReservation(username string, topic string, everyone Permission) error {
  1018. if !AllowedUsername(username) || username == Everyone || !AllowedTopic(topic) {
  1019. return ErrInvalidArgument
  1020. }
  1021. tx, err := a.db.Begin()
  1022. if err != nil {
  1023. return err
  1024. }
  1025. defer tx.Rollback()
  1026. if _, err := tx.Exec(upsertUserAccessQuery, username, topic, true, true, username, username); err != nil {
  1027. return err
  1028. }
  1029. if _, err := tx.Exec(upsertUserAccessQuery, Everyone, topic, everyone.IsRead(), everyone.IsWrite(), username, username); err != nil {
  1030. return err
  1031. }
  1032. return tx.Commit()
  1033. }
  1034. // RemoveReservations deletes the access control entries associated with the given username/topic, as
  1035. // well as all entries with Everyone/topic. This is the counterpart for AddReservation.
  1036. func (a *Manager) RemoveReservations(username string, topics ...string) error {
  1037. if !AllowedUsername(username) || username == Everyone || len(topics) == 0 {
  1038. return ErrInvalidArgument
  1039. }
  1040. for _, topic := range topics {
  1041. if !AllowedTopic(topic) {
  1042. return ErrInvalidArgument
  1043. }
  1044. }
  1045. tx, err := a.db.Begin()
  1046. if err != nil {
  1047. return err
  1048. }
  1049. defer tx.Rollback()
  1050. for _, topic := range topics {
  1051. if _, err := tx.Exec(deleteTopicAccessQuery, username, username, topic); err != nil {
  1052. return err
  1053. }
  1054. if _, err := tx.Exec(deleteTopicAccessQuery, Everyone, Everyone, topic); err != nil {
  1055. return err
  1056. }
  1057. }
  1058. return tx.Commit()
  1059. }
  1060. // DefaultAccess returns the default read/write access if no access control entry matches
  1061. func (a *Manager) DefaultAccess() Permission {
  1062. return a.defaultAccess
  1063. }
  1064. // AddTier creates a new tier in the database
  1065. func (a *Manager) AddTier(tier *Tier) error {
  1066. if tier.ID == "" {
  1067. tier.ID = util.RandomStringPrefix(tierIDPrefix, tierIDLength)
  1068. }
  1069. if _, err := a.db.Exec(insertTierQuery, tier.ID, tier.Code, tier.Name, tier.MessageLimit, int64(tier.MessageExpiryDuration.Seconds()), tier.EmailLimit, tier.ReservationLimit, tier.AttachmentFileSizeLimit, tier.AttachmentTotalSizeLimit, int64(tier.AttachmentExpiryDuration.Seconds()), tier.AttachmentBandwidthLimit, nullString(tier.StripePriceID)); err != nil {
  1070. return err
  1071. }
  1072. return nil
  1073. }
  1074. // UpdateTier updates a tier's properties in the database
  1075. func (a *Manager) UpdateTier(tier *Tier) error {
  1076. if _, err := a.db.Exec(updateTierQuery, tier.Name, tier.MessageLimit, int64(tier.MessageExpiryDuration.Seconds()), tier.EmailLimit, tier.ReservationLimit, tier.AttachmentFileSizeLimit, tier.AttachmentTotalSizeLimit, int64(tier.AttachmentExpiryDuration.Seconds()), tier.AttachmentBandwidthLimit, nullString(tier.StripePriceID), tier.Code); err != nil {
  1077. return err
  1078. }
  1079. return nil
  1080. }
  1081. // RemoveTier deletes the tier with the given code
  1082. func (a *Manager) RemoveTier(code string) error {
  1083. if !AllowedTier(code) {
  1084. return ErrInvalidArgument
  1085. }
  1086. // This fails if any user has this tier
  1087. if _, err := a.db.Exec(deleteTierQuery, code); err != nil {
  1088. return err
  1089. }
  1090. return nil
  1091. }
  1092. // ChangeBilling updates a user's billing fields, namely the Stripe customer ID, and subscription information
  1093. func (a *Manager) ChangeBilling(username string, billing *Billing) error {
  1094. if _, err := a.db.Exec(updateBillingQuery, nullString(billing.StripeCustomerID), nullString(billing.StripeSubscriptionID), nullString(string(billing.StripeSubscriptionStatus)), nullInt64(billing.StripeSubscriptionPaidUntil.Unix()), nullInt64(billing.StripeSubscriptionCancelAt.Unix()), username); err != nil {
  1095. return err
  1096. }
  1097. return nil
  1098. }
  1099. // Tiers returns a list of all Tier structs
  1100. func (a *Manager) Tiers() ([]*Tier, error) {
  1101. rows, err := a.db.Query(selectTiersQuery)
  1102. if err != nil {
  1103. return nil, err
  1104. }
  1105. defer rows.Close()
  1106. tiers := make([]*Tier, 0)
  1107. for {
  1108. tier, err := a.readTier(rows)
  1109. if err == ErrTierNotFound {
  1110. break
  1111. } else if err != nil {
  1112. return nil, err
  1113. }
  1114. tiers = append(tiers, tier)
  1115. }
  1116. return tiers, nil
  1117. }
  1118. // Tier returns a Tier based on the code, or ErrTierNotFound if it does not exist
  1119. func (a *Manager) Tier(code string) (*Tier, error) {
  1120. rows, err := a.db.Query(selectTierByCodeQuery, code)
  1121. if err != nil {
  1122. return nil, err
  1123. }
  1124. defer rows.Close()
  1125. return a.readTier(rows)
  1126. }
  1127. // TierByStripePrice returns a Tier based on the Stripe price ID, or ErrTierNotFound if it does not exist
  1128. func (a *Manager) TierByStripePrice(priceID string) (*Tier, error) {
  1129. rows, err := a.db.Query(selectTierByPriceIDQuery, priceID)
  1130. if err != nil {
  1131. return nil, err
  1132. }
  1133. defer rows.Close()
  1134. return a.readTier(rows)
  1135. }
  1136. func (a *Manager) readTier(rows *sql.Rows) (*Tier, error) {
  1137. var id, code, name string
  1138. var stripePriceID sql.NullString
  1139. var messagesLimit, messagesExpiryDuration, emailsLimit, reservationsLimit, attachmentFileSizeLimit, attachmentTotalSizeLimit, attachmentExpiryDuration, attachmentBandwidthLimit sql.NullInt64
  1140. if !rows.Next() {
  1141. return nil, ErrTierNotFound
  1142. }
  1143. if err := rows.Scan(&id, &code, &name, &messagesLimit, &messagesExpiryDuration, &emailsLimit, &reservationsLimit, &attachmentFileSizeLimit, &attachmentTotalSizeLimit, &attachmentExpiryDuration, &attachmentBandwidthLimit, &stripePriceID); err != nil {
  1144. return nil, err
  1145. } else if err := rows.Err(); err != nil {
  1146. return nil, err
  1147. }
  1148. // When changed, note readUser() as well
  1149. return &Tier{
  1150. ID: id,
  1151. Code: code,
  1152. Name: name,
  1153. MessageLimit: messagesLimit.Int64,
  1154. MessageExpiryDuration: time.Duration(messagesExpiryDuration.Int64) * time.Second,
  1155. EmailLimit: emailsLimit.Int64,
  1156. ReservationLimit: reservationsLimit.Int64,
  1157. AttachmentFileSizeLimit: attachmentFileSizeLimit.Int64,
  1158. AttachmentTotalSizeLimit: attachmentTotalSizeLimit.Int64,
  1159. AttachmentExpiryDuration: time.Duration(attachmentExpiryDuration.Int64) * time.Second,
  1160. AttachmentBandwidthLimit: attachmentBandwidthLimit.Int64,
  1161. StripePriceID: stripePriceID.String, // May be empty
  1162. }, nil
  1163. }
  1164. // Close closes the underlying database
  1165. func (a *Manager) Close() error {
  1166. return a.db.Close()
  1167. }
  1168. func toSQLWildcard(s string) string {
  1169. return strings.ReplaceAll(s, "*", "%")
  1170. }
  1171. func fromSQLWildcard(s string) string {
  1172. return strings.ReplaceAll(s, "%", "*")
  1173. }
  1174. func runStartupQueries(db *sql.DB, startupQueries string) error {
  1175. if _, err := db.Exec(startupQueries); err != nil {
  1176. return err
  1177. }
  1178. if _, err := db.Exec(builtinStartupQueries); err != nil {
  1179. return err
  1180. }
  1181. return nil
  1182. }
  1183. func setupDB(db *sql.DB) error {
  1184. // If 'schemaVersion' table does not exist, this must be a new database
  1185. rowsSV, err := db.Query(selectSchemaVersionQuery)
  1186. if err != nil {
  1187. return setupNewDB(db)
  1188. }
  1189. defer rowsSV.Close()
  1190. // If 'schemaVersion' table exists, read version and potentially upgrade
  1191. schemaVersion := 0
  1192. if !rowsSV.Next() {
  1193. return errors.New("cannot determine schema version: database file may be corrupt")
  1194. }
  1195. if err := rowsSV.Scan(&schemaVersion); err != nil {
  1196. return err
  1197. }
  1198. rowsSV.Close()
  1199. // Do migrations
  1200. if schemaVersion == currentSchemaVersion {
  1201. return nil
  1202. } else if schemaVersion == 1 {
  1203. return migrateFrom1(db)
  1204. }
  1205. return fmt.Errorf("unexpected schema version found: %d", schemaVersion)
  1206. }
  1207. func setupNewDB(db *sql.DB) error {
  1208. if _, err := db.Exec(createTablesQueries); err != nil {
  1209. return err
  1210. }
  1211. if _, err := db.Exec(insertSchemaVersion, currentSchemaVersion); err != nil {
  1212. return err
  1213. }
  1214. return nil
  1215. }
  1216. func migrateFrom1(db *sql.DB) error {
  1217. log.Tag(tagManager).Info("Migrating user database schema: from 1 to 2")
  1218. tx, err := db.Begin()
  1219. if err != nil {
  1220. return err
  1221. }
  1222. defer tx.Rollback()
  1223. // Rename user -> user_old, and create new tables
  1224. if _, err := tx.Exec(migrate1To2RenameUserTableQueryNoTx); err != nil {
  1225. return err
  1226. }
  1227. if _, err := tx.Exec(createTablesQueriesNoTx); err != nil {
  1228. return err
  1229. }
  1230. // Insert users from user_old into new user table, with ID and sync_topic
  1231. rows, err := tx.Query(migrate1To2SelectAllOldUsernamesNoTx)
  1232. if err != nil {
  1233. return err
  1234. }
  1235. defer rows.Close()
  1236. usernames := make([]string, 0)
  1237. for rows.Next() {
  1238. var username string
  1239. if err := rows.Scan(&username); err != nil {
  1240. return err
  1241. }
  1242. usernames = append(usernames, username)
  1243. }
  1244. if err := rows.Close(); err != nil {
  1245. return err
  1246. }
  1247. for _, username := range usernames {
  1248. userID := util.RandomStringPrefix(userIDPrefix, userIDLength)
  1249. syncTopic := util.RandomStringPrefix(syncTopicPrefix, syncTopicLength)
  1250. if _, err := tx.Exec(migrate1To2InsertUserNoTx, userID, syncTopic, username); err != nil {
  1251. return err
  1252. }
  1253. }
  1254. // Migrate old "access" table to "user_access" and drop "access" and "user_old"
  1255. if _, err := tx.Exec(migrate1To2InsertFromOldTablesAndDropNoTx); err != nil {
  1256. return err
  1257. }
  1258. if _, err := tx.Exec(updateSchemaVersion, 2); err != nil {
  1259. return err
  1260. }
  1261. if err := tx.Commit(); err != nil {
  1262. return err
  1263. }
  1264. return nil // Update this when a new version is added
  1265. }
  1266. func nullString(s string) sql.NullString {
  1267. if s == "" {
  1268. return sql.NullString{}
  1269. }
  1270. return sql.NullString{String: s, Valid: true}
  1271. }
  1272. func nullInt64(v int64) sql.NullInt64 {
  1273. if v == 0 {
  1274. return sql.NullInt64{}
  1275. }
  1276. return sql.NullInt64{Int64: v, Valid: true}
  1277. }