manager.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774
  1. package user
  2. import (
  3. "database/sql"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. _ "github.com/mattn/go-sqlite3" // SQLite driver
  8. "golang.org/x/crypto/bcrypt"
  9. "heckel.io/ntfy/log"
  10. "heckel.io/ntfy/util"
  11. "strings"
  12. "sync"
  13. "time"
  14. )
  15. const (
  16. bcryptCost = 10
  17. intentionalSlowDownHash = "$2a$10$YFCQvqQDwIIwnJM1xkAYOeih0dg17UVGanaTStnrSzC8NCWxcLDwy" // Cost should match bcryptCost
  18. userStatsQueueWriterInterval = 33 * time.Second
  19. tokenLength = 32
  20. tokenExpiryDuration = 72 * time.Hour // Extend tokens by this much
  21. tokenMaxCount = 10 // Only keep this many tokens in the table per user
  22. )
  23. var (
  24. errNoTokenProvided = errors.New("no token provided")
  25. errTopicOwnedByOthers = errors.New("topic owned by others")
  26. errNoRows = errors.New("no rows found")
  27. )
  28. // Manager-related queries
  29. const (
  30. createTablesQueriesNoTx = `
  31. CREATE TABLE IF NOT EXISTS plan (
  32. id INT NOT NULL,
  33. code TEXT NOT NULL,
  34. messages_limit INT NOT NULL,
  35. emails_limit INT NOT NULL,
  36. topics_limit INT NOT NULL,
  37. attachment_file_size_limit INT NOT NULL,
  38. attachment_total_size_limit INT NOT NULL,
  39. PRIMARY KEY (id)
  40. );
  41. CREATE TABLE IF NOT EXISTS user (
  42. id INTEGER PRIMARY KEY AUTOINCREMENT,
  43. plan_id INT,
  44. user TEXT NOT NULL,
  45. pass TEXT NOT NULL,
  46. role TEXT NOT NULL,
  47. messages INT NOT NULL DEFAULT (0),
  48. emails INT NOT NULL DEFAULT (0),
  49. settings JSON,
  50. FOREIGN KEY (plan_id) REFERENCES plan (id)
  51. );
  52. CREATE UNIQUE INDEX idx_user ON user (user);
  53. CREATE TABLE IF NOT EXISTS user_access (
  54. user_id INT NOT NULL,
  55. topic TEXT NOT NULL,
  56. read INT NOT NULL,
  57. write INT NOT NULL,
  58. owner_user_id INT,
  59. PRIMARY KEY (user_id, topic),
  60. FOREIGN KEY (user_id) REFERENCES user (id) ON DELETE CASCADE,
  61. FOREIGN KEY (owner_user_id) REFERENCES user (id) ON DELETE CASCADE
  62. );
  63. CREATE TABLE IF NOT EXISTS user_token (
  64. user_id INT NOT NULL,
  65. token TEXT NOT NULL,
  66. expires INT NOT NULL,
  67. PRIMARY KEY (user_id, token),
  68. FOREIGN KEY (user_id) REFERENCES user (id) ON DELETE CASCADE
  69. );
  70. CREATE TABLE IF NOT EXISTS schemaVersion (
  71. id INT PRIMARY KEY,
  72. version INT NOT NULL
  73. );
  74. INSERT INTO user (id, user, pass, role) VALUES (1, '*', '', 'anonymous') ON CONFLICT (id) DO NOTHING;
  75. `
  76. createTablesQueries = `BEGIN; ` + createTablesQueriesNoTx + ` COMMIT;`
  77. builtinStartupQueries = `
  78. PRAGMA foreign_keys = ON;
  79. `
  80. selectUserByNameQuery = `
  81. SELECT u.user, u.pass, u.role, u.messages, u.emails, u.settings, p.code, p.messages_limit, p.emails_limit, p.topics_limit, p.attachment_file_size_limit, p.attachment_total_size_limit
  82. FROM user u
  83. LEFT JOIN plan p on p.id = u.plan_id
  84. WHERE user = ?
  85. `
  86. selectUserByTokenQuery = `
  87. SELECT u.user, u.pass, u.role, u.messages, u.emails, u.settings, p.code, p.messages_limit, p.emails_limit, p.topics_limit, p.attachment_file_size_limit, p.attachment_total_size_limit
  88. FROM user u
  89. JOIN user_token t on u.id = t.user_id
  90. LEFT JOIN plan p on p.id = u.plan_id
  91. WHERE t.token = ? AND t.expires >= ?
  92. `
  93. selectTopicPermsQuery = `
  94. SELECT read, write
  95. FROM user_access a
  96. JOIN user u ON u.id = a.user_id
  97. WHERE (u.user = ? OR u.user = ?) AND ? LIKE a.topic
  98. ORDER BY u.user DESC
  99. `
  100. insertUserQuery = `INSERT INTO user (user, pass, role) VALUES (?, ?, ?)`
  101. selectUsernamesQuery = `
  102. SELECT user
  103. FROM user
  104. ORDER BY
  105. CASE role
  106. WHEN 'admin' THEN 1
  107. WHEN 'anonymous' THEN 3
  108. ELSE 2
  109. END, user
  110. `
  111. updateUserPassQuery = `UPDATE user SET pass = ? WHERE user = ?`
  112. updateUserRoleQuery = `UPDATE user SET role = ? WHERE user = ?`
  113. updateUserSettingsQuery = `UPDATE user SET settings = ? WHERE user = ?`
  114. updateUserStatsQuery = `UPDATE user SET messages = ?, emails = ? WHERE user = ?`
  115. deleteUserQuery = `DELETE FROM user WHERE user = ?`
  116. upsertUserAccessQuery = `
  117. INSERT INTO user_access (user_id, topic, read, write, owner_user_id)
  118. VALUES ((SELECT id FROM user WHERE user = ?), ?, ?, ?, (SELECT IIF(?='',NULL,(SELECT id FROM user WHERE user=?))))
  119. ON CONFLICT (user_id, topic)
  120. DO UPDATE SET read=excluded.read, write=excluded.write, owner_user_id=excluded.owner_user_id
  121. `
  122. selectUserAccessQuery = `
  123. SELECT topic, read, write
  124. FROM user_access
  125. WHERE user_id = (SELECT id FROM user WHERE user = ?)
  126. ORDER BY write DESC, read DESC, topic
  127. `
  128. selectUserReservationsQuery = `
  129. SELECT a_user.topic, a_user.read, a_user.write, a_everyone.read AS everyone_read, a_everyone.write AS everyone_write
  130. FROM user_access a_user
  131. LEFT JOIN user_access a_everyone ON a_user.topic = a_everyone.topic AND a_everyone.user_id = (SELECT id FROM user WHERE user = ?)
  132. WHERE a_user.user_id = a_user.owner_user_id
  133. AND a_user.owner_user_id = (SELECT id FROM user WHERE user = ?)
  134. ORDER BY a_user.topic
  135. `
  136. selectOtherAccessCountQuery = `
  137. SELECT COUNT(*)
  138. FROM user_access
  139. WHERE (topic = ? OR ? LIKE topic)
  140. AND (owner_user_id IS NULL OR owner_user_id != (SELECT id FROM user WHERE user = ?))
  141. `
  142. deleteAllAccessQuery = `DELETE FROM user_access`
  143. deleteUserAccessQuery = `DELETE FROM user_access WHERE user_id = (SELECT id FROM user WHERE user = ?)`
  144. deleteTopicAccessQuery = `DELETE FROM user_access WHERE user_id = (SELECT id FROM user WHERE user = ?) AND topic = ?`
  145. selectTokenCountQuery = `SELECT COUNT(*) FROM user_token WHERE (SELECT id FROM user WHERE user = ?)`
  146. insertTokenQuery = `INSERT INTO user_token (user_id, token, expires) VALUES ((SELECT id FROM user WHERE user = ?), ?, ?)`
  147. updateTokenExpiryQuery = `UPDATE user_token SET expires = ? WHERE user_id = (SELECT id FROM user WHERE user = ?) AND token = ?`
  148. deleteTokenQuery = `DELETE FROM user_token WHERE user_id = (SELECT id FROM user WHERE user = ?) AND token = ?`
  149. deleteExpiredTokensQuery = `DELETE FROM user_token WHERE expires < ?`
  150. deleteExcessTokensQuery = `
  151. DELETE FROM user_token
  152. WHERE (user_id, token) NOT IN (
  153. SELECT user_id, token
  154. FROM user_token
  155. WHERE user_id = (SELECT id FROM user WHERE user = ?)
  156. ORDER BY expires DESC
  157. LIMIT ?
  158. )
  159. ;
  160. `
  161. )
  162. // Schema management queries
  163. const (
  164. currentSchemaVersion = 2
  165. insertSchemaVersion = `INSERT INTO schemaVersion VALUES (1, ?)`
  166. updateSchemaVersion = `UPDATE schemaVersion SET version = ? WHERE id = 1`
  167. selectSchemaVersionQuery = `SELECT version FROM schemaVersion WHERE id = 1`
  168. // 1 -> 2 (complex migration!)
  169. migrate1To2RenameUserTableQueryNoTx = `
  170. ALTER TABLE user RENAME TO user_old;
  171. `
  172. migrate1To2InsertFromOldTablesAndDropNoTx = `
  173. INSERT INTO user (user, pass, role)
  174. SELECT user, pass, role FROM user_old;
  175. INSERT INTO user_access (user_id, topic, read, write)
  176. SELECT u.id, a.topic, a.read, a.write
  177. FROM user u
  178. JOIN access a ON u.user = a.user;
  179. DROP TABLE access;
  180. DROP TABLE user_old;
  181. `
  182. )
  183. // Manager is an implementation of Manager. It stores users and access control list
  184. // in a SQLite database.
  185. type Manager struct {
  186. db *sql.DB
  187. defaultAccess Permission // Default permission if no ACL matches
  188. statsQueue map[string]*User // Username -> User, for "unimportant" user updates
  189. mu sync.Mutex
  190. }
  191. var _ Auther = (*Manager)(nil)
  192. // NewManager creates a new Manager instance
  193. func NewManager(filename, startupQueries string, defaultAccess Permission) (*Manager, error) {
  194. return newManager(filename, startupQueries, defaultAccess, userStatsQueueWriterInterval)
  195. }
  196. // NewManager creates a new Manager instance
  197. func newManager(filename, startupQueries string, defaultAccess Permission, statsWriterInterval time.Duration) (*Manager, error) {
  198. db, err := sql.Open("sqlite3", filename)
  199. if err != nil {
  200. return nil, err
  201. }
  202. if err := setupDB(db); err != nil {
  203. return nil, err
  204. }
  205. if err := runStartupQueries(db, startupQueries); err != nil {
  206. return nil, err
  207. }
  208. manager := &Manager{
  209. db: db,
  210. defaultAccess: defaultAccess,
  211. statsQueue: make(map[string]*User),
  212. }
  213. go manager.userStatsQueueWriter(statsWriterInterval)
  214. return manager, nil
  215. }
  216. // Authenticate checks username and password and returns a User if correct. The method
  217. // returns in constant-ish time, regardless of whether the user exists or the password is
  218. // correct or incorrect.
  219. func (a *Manager) Authenticate(username, password string) (*User, error) {
  220. if username == Everyone {
  221. return nil, ErrUnauthenticated
  222. }
  223. user, err := a.User(username)
  224. if err != nil {
  225. log.Trace("authentication of user %s failed (1): %s", username, err.Error())
  226. bcrypt.CompareHashAndPassword([]byte(intentionalSlowDownHash), []byte("intentional slow-down to avoid timing attacks"))
  227. return nil, ErrUnauthenticated
  228. }
  229. if err := bcrypt.CompareHashAndPassword([]byte(user.Hash), []byte(password)); err != nil {
  230. log.Trace("authentication of user %s failed (2): %s", username, err.Error())
  231. return nil, ErrUnauthenticated
  232. }
  233. return user, nil
  234. }
  235. // AuthenticateToken checks if the token exists and returns the associated User if it does.
  236. // The method sets the User.Token value to the token that was used for authentication.
  237. func (a *Manager) AuthenticateToken(token string) (*User, error) {
  238. if len(token) != tokenLength {
  239. return nil, ErrUnauthenticated
  240. }
  241. user, err := a.userByToken(token)
  242. if err != nil {
  243. return nil, ErrUnauthenticated
  244. }
  245. user.Token = token
  246. return user, nil
  247. }
  248. // CreateToken generates a random token for the given user and returns it. The token expires
  249. // after a fixed duration unless ExtendToken is called. This function also prunes tokens for the
  250. // given user, if there are too many of them.
  251. func (a *Manager) CreateToken(user *User) (*Token, error) {
  252. token, expires := util.RandomString(tokenLength), time.Now().Add(tokenExpiryDuration)
  253. tx, err := a.db.Begin()
  254. if err != nil {
  255. return nil, err
  256. }
  257. defer tx.Rollback()
  258. if _, err := tx.Exec(insertTokenQuery, user.Name, token, expires.Unix()); err != nil {
  259. return nil, err
  260. }
  261. rows, err := tx.Query(selectTokenCountQuery, user.Name)
  262. if err != nil {
  263. return nil, err
  264. }
  265. defer rows.Close()
  266. if !rows.Next() {
  267. return nil, errNoRows
  268. }
  269. var tokenCount int
  270. if err := rows.Scan(&tokenCount); err != nil {
  271. return nil, err
  272. }
  273. if tokenCount >= tokenMaxCount {
  274. // This pruning logic is done in two queries for efficiency. The SELECT above is a lookup
  275. // on two indices, whereas the query below is a full table scan.
  276. if _, err := tx.Exec(deleteExcessTokensQuery, user.Name, tokenMaxCount); err != nil {
  277. return nil, err
  278. }
  279. }
  280. if err := tx.Commit(); err != nil {
  281. return nil, err
  282. }
  283. return &Token{
  284. Value: token,
  285. Expires: expires,
  286. }, nil
  287. }
  288. // ExtendToken sets the new expiry date for a token, thereby extending its use further into the future.
  289. func (a *Manager) ExtendToken(user *User) (*Token, error) {
  290. if user.Token == "" {
  291. return nil, errNoTokenProvided
  292. }
  293. newExpires := time.Now().Add(tokenExpiryDuration)
  294. if _, err := a.db.Exec(updateTokenExpiryQuery, newExpires.Unix(), user.Name, user.Token); err != nil {
  295. return nil, err
  296. }
  297. return &Token{
  298. Value: user.Token,
  299. Expires: newExpires,
  300. }, nil
  301. }
  302. // RemoveToken deletes the token defined in User.Token
  303. func (a *Manager) RemoveToken(user *User) error {
  304. if user.Token == "" {
  305. return ErrUnauthorized
  306. }
  307. if _, err := a.db.Exec(deleteTokenQuery, user.Name, user.Token); err != nil {
  308. return err
  309. }
  310. return nil
  311. }
  312. // RemoveExpiredTokens deletes all expired tokens from the database
  313. func (a *Manager) RemoveExpiredTokens() error {
  314. if _, err := a.db.Exec(deleteExpiredTokensQuery, time.Now().Unix()); err != nil {
  315. return err
  316. }
  317. return nil
  318. }
  319. // ChangeSettings persists the user settings
  320. func (a *Manager) ChangeSettings(user *User) error {
  321. settings, err := json.Marshal(user.Prefs)
  322. if err != nil {
  323. return err
  324. }
  325. if _, err := a.db.Exec(updateUserSettingsQuery, string(settings), user.Name); err != nil {
  326. return err
  327. }
  328. return nil
  329. }
  330. // EnqueueStats adds the user to a queue which writes out user stats (messages, emails, ..) in
  331. // batches at a regular interval
  332. func (a *Manager) EnqueueStats(user *User) {
  333. a.mu.Lock()
  334. defer a.mu.Unlock()
  335. a.statsQueue[user.Name] = user
  336. }
  337. func (a *Manager) userStatsQueueWriter(interval time.Duration) {
  338. ticker := time.NewTicker(interval)
  339. for range ticker.C {
  340. if err := a.writeUserStatsQueue(); err != nil {
  341. log.Warn("UserManager: Writing user stats queue failed: %s", err.Error())
  342. }
  343. }
  344. }
  345. func (a *Manager) writeUserStatsQueue() error {
  346. a.mu.Lock()
  347. if len(a.statsQueue) == 0 {
  348. a.mu.Unlock()
  349. log.Trace("UserManager: No user stats updates to commit")
  350. return nil
  351. }
  352. statsQueue := a.statsQueue
  353. a.statsQueue = make(map[string]*User)
  354. a.mu.Unlock()
  355. tx, err := a.db.Begin()
  356. if err != nil {
  357. return err
  358. }
  359. defer tx.Rollback()
  360. log.Debug("UserManager: Writing user stats queue for %d user(s)", len(statsQueue))
  361. for username, u := range statsQueue {
  362. log.Trace("UserManager: Updating stats for user %s: messages=%d, emails=%d", username, u.Stats.Messages, u.Stats.Emails)
  363. if _, err := tx.Exec(updateUserStatsQuery, u.Stats.Messages, u.Stats.Emails, username); err != nil {
  364. return err
  365. }
  366. }
  367. return tx.Commit()
  368. }
  369. // Authorize returns nil if the given user has access to the given topic using the desired
  370. // permission. The user param may be nil to signal an anonymous user.
  371. func (a *Manager) Authorize(user *User, topic string, perm Permission) error {
  372. if user != nil && user.Role == RoleAdmin {
  373. return nil // Admin can do everything
  374. }
  375. username := Everyone
  376. if user != nil {
  377. username = user.Name
  378. }
  379. // Select the read/write permissions for this user/topic combo. The query may return two
  380. // rows (one for everyone, and one for the user), but prioritizes the user.
  381. rows, err := a.db.Query(selectTopicPermsQuery, Everyone, username, topic)
  382. if err != nil {
  383. return err
  384. }
  385. defer rows.Close()
  386. if !rows.Next() {
  387. return a.resolvePerms(a.defaultAccess, perm)
  388. }
  389. var read, write bool
  390. if err := rows.Scan(&read, &write); err != nil {
  391. return err
  392. } else if err := rows.Err(); err != nil {
  393. return err
  394. }
  395. return a.resolvePerms(NewPermission(read, write), perm)
  396. }
  397. func (a *Manager) resolvePerms(base, perm Permission) error {
  398. if perm == PermissionRead && base.IsRead() {
  399. return nil
  400. } else if perm == PermissionWrite && base.IsWrite() {
  401. return nil
  402. }
  403. return ErrUnauthorized
  404. }
  405. // AddUser adds a user with the given username, password and role
  406. func (a *Manager) AddUser(username, password string, role Role) error {
  407. if !AllowedUsername(username) || !AllowedRole(role) {
  408. return ErrInvalidArgument
  409. }
  410. hash, err := bcrypt.GenerateFromPassword([]byte(password), bcryptCost)
  411. if err != nil {
  412. return err
  413. }
  414. if _, err = a.db.Exec(insertUserQuery, username, hash, role); err != nil {
  415. return err
  416. }
  417. return nil
  418. }
  419. // RemoveUser deletes the user with the given username. The function returns nil on success, even
  420. // if the user did not exist in the first place.
  421. func (a *Manager) RemoveUser(username string) error {
  422. if !AllowedUsername(username) {
  423. return ErrInvalidArgument
  424. }
  425. // Rows in user_access, user_token, etc. are deleted via foreign keys
  426. if _, err := a.db.Exec(deleteUserQuery, username); err != nil {
  427. return err
  428. }
  429. return nil
  430. }
  431. // Users returns a list of users. It always also returns the Everyone user ("*").
  432. func (a *Manager) Users() ([]*User, error) {
  433. rows, err := a.db.Query(selectUsernamesQuery)
  434. if err != nil {
  435. return nil, err
  436. }
  437. defer rows.Close()
  438. usernames := make([]string, 0)
  439. for rows.Next() {
  440. var username string
  441. if err := rows.Scan(&username); err != nil {
  442. return nil, err
  443. } else if err := rows.Err(); err != nil {
  444. return nil, err
  445. }
  446. usernames = append(usernames, username)
  447. }
  448. rows.Close()
  449. users := make([]*User, 0)
  450. for _, username := range usernames {
  451. user, err := a.User(username)
  452. if err != nil {
  453. return nil, err
  454. }
  455. users = append(users, user)
  456. }
  457. return users, nil
  458. }
  459. // User returns the user with the given username if it exists, or ErrNotFound otherwise.
  460. // You may also pass Everyone to retrieve the anonymous user and its Grant list.
  461. func (a *Manager) User(username string) (*User, error) {
  462. rows, err := a.db.Query(selectUserByNameQuery, username)
  463. if err != nil {
  464. return nil, err
  465. }
  466. return a.readUser(rows)
  467. }
  468. func (a *Manager) userByToken(token string) (*User, error) {
  469. rows, err := a.db.Query(selectUserByTokenQuery, token, time.Now().Unix())
  470. if err != nil {
  471. return nil, err
  472. }
  473. return a.readUser(rows)
  474. }
  475. func (a *Manager) readUser(rows *sql.Rows) (*User, error) {
  476. defer rows.Close()
  477. var username, hash, role string
  478. var settings, planCode sql.NullString
  479. var messages, emails int64
  480. var messagesLimit, emailsLimit, topicsLimit, attachmentFileSizeLimit, attachmentTotalSizeLimit sql.NullInt64
  481. if !rows.Next() {
  482. return nil, ErrNotFound
  483. }
  484. if err := rows.Scan(&username, &hash, &role, &messages, &emails, &settings, &planCode, &messagesLimit, &emailsLimit, &topicsLimit, &attachmentFileSizeLimit, &attachmentTotalSizeLimit); err != nil {
  485. return nil, err
  486. } else if err := rows.Err(); err != nil {
  487. return nil, err
  488. }
  489. user := &User{
  490. Name: username,
  491. Hash: hash,
  492. Role: Role(role),
  493. Stats: &Stats{
  494. Messages: messages,
  495. Emails: emails,
  496. },
  497. }
  498. if settings.Valid {
  499. user.Prefs = &Prefs{}
  500. if err := json.Unmarshal([]byte(settings.String), user.Prefs); err != nil {
  501. return nil, err
  502. }
  503. }
  504. if planCode.Valid {
  505. user.Plan = &Plan{
  506. Code: planCode.String,
  507. Upgradeable: false,
  508. MessagesLimit: messagesLimit.Int64,
  509. EmailsLimit: emailsLimit.Int64,
  510. TopicsLimit: topicsLimit.Int64,
  511. AttachmentFileSizeLimit: attachmentFileSizeLimit.Int64,
  512. AttachmentTotalSizeLimit: attachmentTotalSizeLimit.Int64,
  513. }
  514. }
  515. return user, nil
  516. }
  517. // Grants returns all user-specific access control entries
  518. func (a *Manager) Grants(username string) ([]Grant, error) {
  519. rows, err := a.db.Query(selectUserAccessQuery, username)
  520. if err != nil {
  521. return nil, err
  522. }
  523. defer rows.Close()
  524. grants := make([]Grant, 0)
  525. for rows.Next() {
  526. var topic string
  527. var read, write bool
  528. if err := rows.Scan(&topic, &read, &write); err != nil {
  529. return nil, err
  530. } else if err := rows.Err(); err != nil {
  531. return nil, err
  532. }
  533. grants = append(grants, Grant{
  534. TopicPattern: fromSQLWildcard(topic),
  535. Allow: NewPermission(read, write),
  536. })
  537. }
  538. return grants, nil
  539. }
  540. // Reservations returns all user-owned topics, and the associated everyone-access
  541. func (a *Manager) Reservations(username string) ([]Reservation, error) {
  542. rows, err := a.db.Query(selectUserReservationsQuery, Everyone, username)
  543. if err != nil {
  544. return nil, err
  545. }
  546. defer rows.Close()
  547. reservations := make([]Reservation, 0)
  548. for rows.Next() {
  549. var topic string
  550. var ownerRead, ownerWrite bool
  551. var everyoneRead, everyoneWrite sql.NullBool
  552. if err := rows.Scan(&topic, &ownerRead, &ownerWrite, &everyoneRead, &everyoneWrite); err != nil {
  553. return nil, err
  554. } else if err := rows.Err(); err != nil {
  555. return nil, err
  556. }
  557. reservations = append(reservations, Reservation{
  558. Topic: topic,
  559. Owner: NewPermission(ownerRead, ownerWrite),
  560. Everyone: NewPermission(everyoneRead.Bool, everyoneWrite.Bool), // false if null
  561. })
  562. }
  563. return reservations, nil
  564. }
  565. // ChangePassword changes a user's password
  566. func (a *Manager) ChangePassword(username, password string) error {
  567. hash, err := bcrypt.GenerateFromPassword([]byte(password), bcryptCost)
  568. if err != nil {
  569. return err
  570. }
  571. if _, err := a.db.Exec(updateUserPassQuery, hash, username); err != nil {
  572. return err
  573. }
  574. return nil
  575. }
  576. // ChangeRole changes a user's role. When a role is changed from RoleUser to RoleAdmin,
  577. // all existing access control entries (Grant) are removed, since they are no longer needed.
  578. func (a *Manager) ChangeRole(username string, role Role) error {
  579. if !AllowedUsername(username) || !AllowedRole(role) {
  580. return ErrInvalidArgument
  581. }
  582. if _, err := a.db.Exec(updateUserRoleQuery, string(role), username); err != nil {
  583. return err
  584. }
  585. if role == RoleAdmin {
  586. if _, err := a.db.Exec(deleteUserAccessQuery, username); err != nil {
  587. return err
  588. }
  589. }
  590. return nil
  591. }
  592. // CheckAllowAccess tests if a user may create an access control entry for the given topic.
  593. // If there are any ACL entries that are not owned by the user, an error is returned.
  594. func (a *Manager) CheckAllowAccess(username string, topic string) error {
  595. if (!AllowedUsername(username) && username != Everyone) || !AllowedTopic(topic) {
  596. return ErrInvalidArgument
  597. }
  598. rows, err := a.db.Query(selectOtherAccessCountQuery, topic, topic, username)
  599. if err != nil {
  600. return err
  601. }
  602. defer rows.Close()
  603. if !rows.Next() {
  604. return errNoRows
  605. }
  606. var otherCount int
  607. if err := rows.Scan(&otherCount); err != nil {
  608. return err
  609. }
  610. if otherCount > 0 {
  611. return errTopicOwnedByOthers
  612. }
  613. return nil
  614. }
  615. // AllowAccess adds or updates an entry in th access control list for a specific user. It controls
  616. // read/write access to a topic. The parameter topicPattern may include wildcards (*). The ACL entry
  617. // owner may either be a user (username), or the system (empty).
  618. func (a *Manager) AllowAccess(owner, username string, topicPattern string, read bool, write bool) error {
  619. if !AllowedUsername(username) && username != Everyone {
  620. return ErrInvalidArgument
  621. } else if owner != "" && !AllowedUsername(owner) {
  622. return ErrInvalidArgument
  623. } else if !AllowedTopicPattern(topicPattern) {
  624. return ErrInvalidArgument
  625. }
  626. if _, err := a.db.Exec(upsertUserAccessQuery, username, toSQLWildcard(topicPattern), read, write, owner, owner); err != nil {
  627. return err
  628. }
  629. return nil
  630. }
  631. // ResetAccess removes an access control list entry for a specific username/topic, or (if topic is
  632. // empty) for an entire user. The parameter topicPattern may include wildcards (*).
  633. func (a *Manager) ResetAccess(username string, topicPattern string) error {
  634. if !AllowedUsername(username) && username != Everyone && username != "" {
  635. return ErrInvalidArgument
  636. } else if !AllowedTopicPattern(topicPattern) && topicPattern != "" {
  637. return ErrInvalidArgument
  638. }
  639. if username == "" && topicPattern == "" {
  640. _, err := a.db.Exec(deleteAllAccessQuery, username)
  641. return err
  642. } else if topicPattern == "" {
  643. _, err := a.db.Exec(deleteUserAccessQuery, username)
  644. return err
  645. }
  646. _, err := a.db.Exec(deleteTopicAccessQuery, username, toSQLWildcard(topicPattern))
  647. return err
  648. }
  649. // DefaultAccess returns the default read/write access if no access control entry matches
  650. func (a *Manager) DefaultAccess() Permission {
  651. return a.defaultAccess
  652. }
  653. func toSQLWildcard(s string) string {
  654. return strings.ReplaceAll(s, "*", "%")
  655. }
  656. func fromSQLWildcard(s string) string {
  657. return strings.ReplaceAll(s, "%", "*")
  658. }
  659. func runStartupQueries(db *sql.DB, startupQueries string) error {
  660. if _, err := db.Exec(startupQueries); err != nil {
  661. return err
  662. }
  663. if _, err := db.Exec(builtinStartupQueries); err != nil {
  664. return err
  665. }
  666. return nil
  667. }
  668. func setupDB(db *sql.DB) error {
  669. // If 'schemaVersion' table does not exist, this must be a new database
  670. rowsSV, err := db.Query(selectSchemaVersionQuery)
  671. if err != nil {
  672. return setupNewDB(db)
  673. }
  674. defer rowsSV.Close()
  675. // If 'schemaVersion' table exists, read version and potentially upgrade
  676. schemaVersion := 0
  677. if !rowsSV.Next() {
  678. return errors.New("cannot determine schema version: database file may be corrupt")
  679. }
  680. if err := rowsSV.Scan(&schemaVersion); err != nil {
  681. return err
  682. }
  683. rowsSV.Close()
  684. // Do migrations
  685. if schemaVersion == currentSchemaVersion {
  686. return nil
  687. } else if schemaVersion == 1 {
  688. return migrateFrom1(db)
  689. }
  690. return fmt.Errorf("unexpected schema version found: %d", schemaVersion)
  691. }
  692. func setupNewDB(db *sql.DB) error {
  693. if _, err := db.Exec(createTablesQueries); err != nil {
  694. return err
  695. }
  696. if _, err := db.Exec(insertSchemaVersion, currentSchemaVersion); err != nil {
  697. return err
  698. }
  699. return nil
  700. }
  701. func migrateFrom1(db *sql.DB) error {
  702. log.Info("Migrating user database schema: from 1 to 2")
  703. tx, err := db.Begin()
  704. if err != nil {
  705. return err
  706. }
  707. defer tx.Rollback()
  708. if _, err := tx.Exec(migrate1To2RenameUserTableQueryNoTx); err != nil {
  709. return err
  710. }
  711. if _, err := tx.Exec(createTablesQueriesNoTx); err != nil {
  712. return err
  713. }
  714. if _, err := tx.Exec(migrate1To2InsertFromOldTablesAndDropNoTx); err != nil {
  715. return err
  716. }
  717. if _, err := tx.Exec(updateSchemaVersion, 2); err != nil {
  718. return err
  719. }
  720. if err := tx.Commit(); err != nil {
  721. return err
  722. }
  723. return nil // Update this when a new version is added
  724. }