mirror of
https://github.com/hoernschen/dendrite.git
synced 2025-08-03 22:52:47 +00:00
✅ Test 7 Pushers created with a the same access token are not deleted on password change
This commit is contained in:
Dan Peleg
parent
4c4cf8020a
commit
763354e371
11 changed files with 335 additions and 71 deletions
|
|
@ -20,6 +20,7 @@ import (
|
|||
|
||||
"github.com/matrix-org/dendrite/internal/sqlutil"
|
||||
"github.com/matrix-org/dendrite/userapi/api"
|
||||
"github.com/sirupsen/logrus"
|
||||
|
||||
"github.com/matrix-org/dendrite/clientapi/userutil"
|
||||
"github.com/matrix-org/gomatrixserverlib"
|
||||
|
|
@ -28,32 +29,36 @@ import (
|
|||
const pushersSchema = `
|
||||
-- Stores data about pushers.
|
||||
CREATE TABLE IF NOT EXISTS pusher_pushers (
|
||||
localpart TEXT PRIMARY KEY,
|
||||
pushkey VARCHAR(512),
|
||||
localpart TEXT,
|
||||
session_id BIGINT,
|
||||
profile_tag TEXT,
|
||||
kind TEXT,
|
||||
app_id VARCHAR(64),
|
||||
app_display_name TEXT,
|
||||
device_display_name TEXT,
|
||||
profile_tag TEXT,
|
||||
pushkey VARCHAR(512),
|
||||
lang TEXT,
|
||||
url TEXT,
|
||||
format TEXT,
|
||||
|
||||
UNIQUE (localpart, pushkey)
|
||||
UNIQUE (app_id, pushkey, localpart)
|
||||
);
|
||||
`
|
||||
const insertPusherSQL = "" +
|
||||
"INSERT INTO pusher_pushers (localpart, pushkey, kind, app_id, app_display_name, device_display_name, profile_tag, lang, url, format)" +
|
||||
" VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9. $10)"
|
||||
"INSERT INTO pusher_pushers (localpart, session_id, pushkey, kind, app_id, app_display_name, device_display_name, profile_tag, lang, url, format)" +
|
||||
" VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)"
|
||||
|
||||
const selectPushersByLocalpartSQL = "" +
|
||||
"SELECT pushkey, kind, app_id, app_display_name, device_display_name, profile_tag, lang, url, format FROM pusher_pushers WHERE localpart = $1"
|
||||
"SELECT session_id, pushkey, kind, app_id, app_display_name, device_display_name, profile_tag, lang, url, format FROM pusher_pushers WHERE localpart = $1"
|
||||
|
||||
const selectPusherByPushkeySQL = "" +
|
||||
"SELECT pushkey, kind, app_id, app_display_name, device_display_name, profile_tag, lang, url, format FROM pusher_pushers WHERE localpart = $1 AND pushkey = $2"
|
||||
"SELECT session_id, pushkey, kind, app_id, app_display_name, device_display_name, profile_tag, lang, url, format FROM pusher_pushers WHERE localpart = $1 AND pushkey = $2"
|
||||
|
||||
const updatePusherSQL = "" +
|
||||
"UPDATE pusher_pushers SET kind = $1, app_id = $2, app_display_name = $3, device_display_name = $4, profile_tag = $5, lang = $6, url = $7, format = $8 WHERE localpart = $9 AND pushkey = $10"
|
||||
|
||||
const deletePusherSQL = "" +
|
||||
"DELETE FROM pusher_pushers WHERE pushkey = $1 AND localpart = $2"
|
||||
"DELETE FROM pusher_pushers WHERE app_id = $1 AND pushkey = $2 AND localpart = $3"
|
||||
|
||||
type pushersStatements struct {
|
||||
db *sql.DB
|
||||
|
|
@ -61,6 +66,7 @@ type pushersStatements struct {
|
|||
insertPusherStmt *sql.Stmt
|
||||
selectPushersByLocalpartStmt *sql.Stmt
|
||||
selectPusherByPushkeyStmt *sql.Stmt
|
||||
updatePusherStmt *sql.Stmt
|
||||
deletePusherStmt *sql.Stmt
|
||||
serverName gomatrixserverlib.ServerName
|
||||
}
|
||||
|
|
@ -82,6 +88,9 @@ func (s *pushersStatements) prepare(db *sql.DB, writer sqlutil.Writer, server go
|
|||
if s.selectPusherByPushkeyStmt, err = db.Prepare(selectPusherByPushkeySQL); err != nil {
|
||||
return
|
||||
}
|
||||
if s.updatePusherStmt, err = db.Prepare(updatePusherSQL); err != nil {
|
||||
return
|
||||
}
|
||||
if s.deletePusherStmt, err = db.Prepare(deletePusherSQL); err != nil {
|
||||
return
|
||||
}
|
||||
|
|
@ -93,10 +102,12 @@ func (s *pushersStatements) prepare(db *sql.DB, writer sqlutil.Writer, server go
|
|||
// Returns an error if the user already has a pusher with the given pusher pushkey.
|
||||
// Returns nil error success.
|
||||
func (s *pushersStatements) insertPusher(
|
||||
ctx context.Context, txn *sql.Tx, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart string,
|
||||
ctx context.Context, txn *sql.Tx, session_id int64,
|
||||
pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart string,
|
||||
) error {
|
||||
stmt := sqlutil.TxStmt(txn, s.insertPusherStmt)
|
||||
_, err := stmt.ExecContext(ctx, localpart, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format)
|
||||
_, err := stmt.ExecContext(ctx, localpart, session_id, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format)
|
||||
logrus.Debugf("🥳 Created pusher %d", session_id)
|
||||
return err
|
||||
}
|
||||
|
||||
|
|
@ -111,12 +122,17 @@ func (s *pushersStatements) selectPushersByLocalpart(
|
|||
}
|
||||
|
||||
for rows.Next() {
|
||||
logrus.Debug("Next pusher row...")
|
||||
var pusher api.Pusher
|
||||
var sessionid sql.NullInt64
|
||||
var pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format sql.NullString
|
||||
err = rows.Scan(&pushkey, &kind, &appid, &appdisplayname, &devicedisplayname, &profiletag, &lang, &url, &format)
|
||||
err = rows.Scan(&sessionid, &pushkey, &kind, &appid, &appdisplayname, &devicedisplayname, &profiletag, &lang, &url, &format)
|
||||
if err != nil {
|
||||
return pushers, err
|
||||
}
|
||||
if sessionid.Valid {
|
||||
pusher.SessionID = sessionid.Int64
|
||||
}
|
||||
if pushkey.Valid {
|
||||
pusher.PushKey = pushkey.String
|
||||
}
|
||||
|
|
@ -149,6 +165,7 @@ func (s *pushersStatements) selectPushersByLocalpart(
|
|||
pushers = append(pushers, pusher)
|
||||
}
|
||||
|
||||
logrus.Debugf("🤓 Database returned %d pushers", len(pushers))
|
||||
return pushers, nil
|
||||
}
|
||||
|
||||
|
|
@ -194,10 +211,18 @@ func (s *pushersStatements) selectPusherByPushkey(
|
|||
return &pusher, err
|
||||
}
|
||||
|
||||
func (s *pushersStatements) deletePusher(
|
||||
ctx context.Context, txn *sql.Tx, id, localpart string,
|
||||
func (s *pushersStatements) updatePusher(
|
||||
ctx context.Context, txn *sql.Tx, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart string,
|
||||
) error {
|
||||
stmt := sqlutil.TxStmt(txn, s.deletePusherStmt)
|
||||
_, err := stmt.ExecContext(ctx, id, localpart)
|
||||
stmt := sqlutil.TxStmt(txn, s.updatePusherStmt)
|
||||
_, err := stmt.ExecContext(ctx, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart, pushkey)
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *pushersStatements) deletePusher(
|
||||
ctx context.Context, txn *sql.Tx, appid, pushkey, localpart string,
|
||||
) error {
|
||||
stmt := sqlutil.TxStmt(txn, s.deletePusherStmt)
|
||||
_, err := stmt.ExecContext(ctx, appid, pushkey, localpart)
|
||||
return err
|
||||
}
|
||||
|
|
|
|||
|
|
@ -22,6 +22,7 @@ import (
|
|||
"github.com/matrix-org/dendrite/setup/config"
|
||||
"github.com/matrix-org/dendrite/userapi/api"
|
||||
"github.com/matrix-org/gomatrixserverlib"
|
||||
"github.com/sirupsen/logrus"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
|
@ -58,9 +59,10 @@ func NewDatabase(dbProperties *config.DatabaseOptions, serverName gomatrixserver
|
|||
}
|
||||
|
||||
func (d *Database) CreatePusher(
|
||||
ctx context.Context, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart string,
|
||||
ctx context.Context, session_id int64,
|
||||
pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart string,
|
||||
) error {
|
||||
return d.pushers.insertPusher(ctx, nil, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart)
|
||||
return d.pushers.insertPusher(ctx, nil, session_id, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart)
|
||||
}
|
||||
|
||||
// GetPushersByLocalpart returns the pushers matching the given localpart.
|
||||
|
|
@ -77,16 +79,29 @@ func (d *Database) GetPusherByPushkey(
|
|||
return d.pushers.selectPusherByPushkey(ctx, pushkey, localpart)
|
||||
}
|
||||
|
||||
// UpdatePusher updates the given device with the display name.
|
||||
// Returns SQL error if there are problems and nil on success.
|
||||
func (d *Database) UpdatePusher(
|
||||
ctx context.Context, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart string,
|
||||
) error {
|
||||
return d.writer.Do(d.db, nil, func(txn *sql.Tx) error {
|
||||
return d.pushers.updatePusher(ctx, txn, pushkey, kind, appid, appdisplayname, devicedisplayname, profiletag, lang, url, format, localpart)
|
||||
})
|
||||
}
|
||||
|
||||
// RemovePusher revokes a pusher by deleting the entry in the database
|
||||
// matching with the given pusher pushkey and user ID localpart.
|
||||
// If the pusher doesn't exist, it will not return an error
|
||||
// If something went wrong during the deletion, it will return the SQL error.
|
||||
func (d *Database) RemovePusher(
|
||||
ctx context.Context, pushkey, localpart string,
|
||||
ctx context.Context, appid, pushkey, localpart string,
|
||||
) error {
|
||||
return d.writer.Do(d.db, nil, func(txn *sql.Tx) error {
|
||||
if err := d.pushers.deletePusher(ctx, txn, pushkey, localpart); err != sql.ErrNoRows {
|
||||
if err := d.pushers.deletePusher(ctx, txn, appid, pushkey, localpart); err != sql.ErrNoRows {
|
||||
logrus.WithError(err).Debug("RemovePusher Yes Error")
|
||||
return err
|
||||
} else {
|
||||
logrus.WithError(err).Debug("RemovePusher No Error")
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue