hoernschen/dendrite
1
0
Fork 0
mirror of https://github.com/hoernschen/dendrite.git synced 2025-04-21 11:13:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Check per-user

Browse source
This commit is contained in:
Neil Alexander 2021-07-30 10:23:36 +01:00
parent 0571b7eb1d
commit 1fed507679
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
keyserver/internal/internal.go
View file

@ -227,17 +227,13 @@ func (a *KeyInternalAPI) QueryKeys(ctx context.Context, req *api.QueryKeysReques
res.Failures = make(map[string]interface{})
// get cross-signing keys from the database
crossSigningSatisfiedLocally := true
if err := a.crossSigningKeys(ctx, req, res); err != nil {
util.GetLogger(ctx).WithError(err).Error("Failed to retrieve cross-signing keys from database")
crossSigningSatisfiedLocally = false
}
if len(res.MasterKeys) == 0 || len(res.SelfSigningKeys) == 0 {
crossSigningSatisfiedLocally = false
}
// make a map from domain to device keys
domainToDeviceKeys := make(map[string]map[string][]string)
domainToCrossSigningKeys := make(map[string]struct{})
for userID, deviceIDs := range req.UserToDevices {
_, serverName, err := gomatrixserverlib.SplitID('@', userID)
if err != nil {
@ -288,11 +284,21 @@ func (a *KeyInternalAPI) QueryKeys(ctx context.Context, req *api.QueryKeysReques
domainToDeviceKeys[domain] = make(map[string][]string)
domainToDeviceKeys[domain][userID] = append(domainToDeviceKeys[domain][userID], deviceIDs...)
}
// work out if our cross-signing request for this user was
// satisfied
if _, ok := domainToCrossSigningKeys[userID]; !ok {
if _, ok := res.MasterKeys[userID]; !ok {
domainToCrossSigningKeys[userID] = struct{}{}
}
if _, ok := res.SelfSigningKeys[userID]; !ok {
domainToCrossSigningKeys[userID] = struct{}{}
}
}
}
// attempt to satisfy key queries from the local database first as we should get device updates pushed to us
domainToDeviceKeys = a.remoteKeysFromDatabase(ctx, res, domainToDeviceKeys)
if len(domainToDeviceKeys) == 0 && crossSigningSatisfiedLocally {
if len(domainToDeviceKeys) == 0 && len(domainToCrossSigningKeys) == 0 {
return // nothing to query
}