mirror of
https://github.com/1f349/dendrite.git
synced 2024-11-12 14:51:40 +00:00
114 lines
3.4 KiB
Go
114 lines
3.4 KiB
Go
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package routing
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/matrix-org/dendrite/userapi/api"
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
"github.com/matrix-org/gomatrixserverlib/fclient"
|
|
"github.com/matrix-org/gomatrixserverlib/spec"
|
|
"github.com/matrix-org/util"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
// GetUserDevices for the given user id
|
|
func GetUserDevices(
|
|
req *http.Request,
|
|
keyAPI api.FederationKeyAPI,
|
|
userID string,
|
|
) util.JSONResponse {
|
|
var res api.QueryDeviceMessagesResponse
|
|
if err := keyAPI.QueryDeviceMessages(req.Context(), &api.QueryDeviceMessagesRequest{
|
|
UserID: userID,
|
|
}, &res); err != nil {
|
|
return util.ErrorResponse(err)
|
|
}
|
|
if res.Error != nil {
|
|
util.GetLogger(req.Context()).WithError(res.Error).Error("keyAPI.QueryDeviceMessages failed")
|
|
return util.JSONResponse{
|
|
Code: http.StatusInternalServerError,
|
|
JSON: spec.InternalServerError{},
|
|
}
|
|
}
|
|
|
|
sigReq := &api.QuerySignaturesRequest{
|
|
TargetIDs: map[string][]gomatrixserverlib.KeyID{
|
|
userID: {},
|
|
},
|
|
}
|
|
sigRes := &api.QuerySignaturesResponse{}
|
|
for _, dev := range res.Devices {
|
|
sigReq.TargetIDs[userID] = append(sigReq.TargetIDs[userID], gomatrixserverlib.KeyID(dev.DeviceID))
|
|
}
|
|
keyAPI.QuerySignatures(req.Context(), sigReq, sigRes)
|
|
|
|
response := fclient.RespUserDevices{
|
|
UserID: userID,
|
|
StreamID: res.StreamID,
|
|
Devices: []fclient.RespUserDevice{},
|
|
}
|
|
|
|
if masterKey, ok := sigRes.MasterKeys[userID]; ok {
|
|
response.MasterKey = &masterKey
|
|
}
|
|
if selfSigningKey, ok := sigRes.SelfSigningKeys[userID]; ok {
|
|
response.SelfSigningKey = &selfSigningKey
|
|
}
|
|
|
|
for _, dev := range res.Devices {
|
|
var key fclient.RespUserDeviceKeys
|
|
err := json.Unmarshal(dev.DeviceKeys.KeyJSON, &key)
|
|
if err != nil {
|
|
util.GetLogger(req.Context()).WithError(err).Warnf("malformed device key: %s", string(dev.DeviceKeys.KeyJSON))
|
|
continue
|
|
}
|
|
|
|
displayName := dev.DisplayName
|
|
if displayName == "" {
|
|
displayName = gjson.GetBytes(dev.DeviceKeys.KeyJSON, "unsigned.device_display_name").Str
|
|
}
|
|
|
|
device := fclient.RespUserDevice{
|
|
DeviceID: dev.DeviceID,
|
|
DisplayName: displayName,
|
|
Keys: key,
|
|
}
|
|
|
|
if targetUser, ok := sigRes.Signatures[userID]; ok {
|
|
if targetKey, ok := targetUser[gomatrixserverlib.KeyID(dev.DeviceID)]; ok {
|
|
for sourceUserID, forSourceUser := range targetKey {
|
|
for sourceKeyID, sourceKey := range forSourceUser {
|
|
if device.Keys.Signatures == nil {
|
|
device.Keys.Signatures = map[string]map[gomatrixserverlib.KeyID]spec.Base64Bytes{}
|
|
}
|
|
if _, ok := device.Keys.Signatures[sourceUserID]; !ok {
|
|
device.Keys.Signatures[sourceUserID] = map[gomatrixserverlib.KeyID]spec.Base64Bytes{}
|
|
}
|
|
device.Keys.Signatures[sourceUserID][sourceKeyID] = sourceKey
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
response.Devices = append(response.Devices, device)
|
|
}
|
|
|
|
return util.JSONResponse{
|
|
Code: 200,
|
|
JSON: response,
|
|
}
|
|
}
|