refactor: Remove unnecessary roomId parameter from decryptRoomEvent method

This commit is contained in:
Krille 2024-12-15 12:33:21 +01:00
parent 04a46226cb
commit bc7b987772
No known key found for this signature in database
GPG Key ID: E067ECD60F1A0652
9 changed files with 28 additions and 45 deletions

View File

@ -1,2 +1,2 @@
flutter_version=3.24.3 flutter_version=3.27.1
dart_version=3.5.3 dart_version=3.6.0

View File

@ -193,7 +193,7 @@ class Encryption {
} }
} }
Event decryptRoomEventSync(String roomId, Event event) { Event decryptRoomEventSync(Event event) {
if (event.type != EventTypes.Encrypted || event.redacted) { if (event.type != EventTypes.Encrypted || event.redacted) {
return event; return event;
} }
@ -214,7 +214,7 @@ class Encryption {
} }
final inboundGroupSession = final inboundGroupSession =
keyManager.getInboundGroupSession(roomId, sessionId); keyManager.getInboundGroupSession(event.room.id, sessionId);
if (!(inboundGroupSession?.isValid ?? false)) { if (!(inboundGroupSession?.isValid ?? false)) {
canRequestSession = true; canRequestSession = true;
throw DecryptException(DecryptException.unknownSession); throw DecryptException(DecryptException.unknownSession);
@ -249,7 +249,7 @@ class Encryption {
// ignore: discarded_futures // ignore: discarded_futures
?.updateInboundGroupSessionIndexes( ?.updateInboundGroupSessionIndexes(
json.encode(inboundGroupSession.indexes), json.encode(inboundGroupSession.indexes),
roomId, event.room.id,
sessionId, sessionId,
) )
// ignore: discarded_futures // ignore: discarded_futures
@ -260,14 +260,16 @@ class Encryption {
// alright, if this was actually by our own outbound group session, we might as well clear it // alright, if this was actually by our own outbound group session, we might as well clear it
if (exception.toString() != DecryptException.unknownSession && if (exception.toString() != DecryptException.unknownSession &&
(keyManager (keyManager
.getOutboundGroupSession(roomId) .getOutboundGroupSession(event.room.id)
?.outboundGroupSession ?.outboundGroupSession
?.session_id() ?? ?.session_id() ??
'') == '') ==
content.sessionId) { content.sessionId) {
runInRoot( runInRoot(
() async => () async => keyManager.clearOrUseOutboundGroupSession(
keyManager.clearOrUseOutboundGroupSession(roomId, wipe: true), event.room.id,
wipe: true,
),
); );
} }
if (canRequestSession) { if (canRequestSession) {
@ -308,7 +310,6 @@ class Encryption {
} }
Future<Event> decryptRoomEvent( Future<Event> decryptRoomEvent(
String roomId,
Event event, { Event event, {
bool store = false, bool store = false,
EventUpdateType updateType = EventUpdateType.timeline, EventUpdateType updateType = EventUpdateType.timeline,
@ -323,22 +324,22 @@ class Encryption {
sessionId != null && sessionId != null &&
!(keyManager !(keyManager
.getInboundGroupSession( .getInboundGroupSession(
roomId, event.room.id,
sessionId, sessionId,
) )
?.isValid ?? ?.isValid ??
false)) { false)) {
await keyManager.loadInboundGroupSession( await keyManager.loadInboundGroupSession(
roomId, event.room.id,
sessionId, sessionId,
); );
} }
event = decryptRoomEventSync(roomId, event); event = decryptRoomEventSync(event);
if (event.type == EventTypes.Encrypted && if (event.type == EventTypes.Encrypted &&
event.content['can_request_session'] == true && event.content['can_request_session'] == true &&
sessionId != null) { sessionId != null) {
keyManager.maybeAutoRequest( keyManager.maybeAutoRequest(
roomId, event.room.id,
sessionId, sessionId,
content.senderKey, content.senderKey,
); );
@ -350,7 +351,7 @@ class Encryption {
await client.database?.storeEventUpdate( await client.database?.storeEventUpdate(
EventUpdate( EventUpdate(
content: event.toJson(), content: event.toJson(),
roomID: roomId, roomID: event.room.id,
type: updateType, type: updateType,
), ),
client, client,

View File

@ -192,7 +192,7 @@ class KeyManager {
if (event != null && if (event != null &&
event.type == EventTypes.Encrypted && event.type == EventTypes.Encrypted &&
event.content['session_id'] == sessionId) { event.content['session_id'] == sessionId) {
final decrypted = encryption.decryptRoomEventSync(roomId, event); final decrypted = encryption.decryptRoomEventSync(event);
if (decrypted.type != EventTypes.Encrypted) { if (decrypted.type != EventTypes.Encrypted) {
// Update the last event in memory first // Update the last event in memory first
room.lastEvent = decrypted; room.lastEvent = decrypted;

View File

@ -1243,10 +1243,7 @@ class Client extends MatrixApi {
if (archivedRoom.encrypted && archivedRoom.client.encryptionEnabled) { if (archivedRoom.encrypted && archivedRoom.client.encryptionEnabled) {
if (timeline.events[i].type == EventTypes.Encrypted) { if (timeline.events[i].type == EventTypes.Encrypted) {
await archivedRoom.client.encryption! await archivedRoom.client.encryption!
.decryptRoomEvent( .decryptRoomEvent(timeline.events[i])
archivedRoom.id,
timeline.events[i],
)
.then( .then(
(decrypted) => timeline.events[i] = decrypted, (decrypted) => timeline.events[i] = decrypted,
); );
@ -1880,11 +1877,11 @@ class Client extends MatrixApi {
final encryption = this.encryption; final encryption = this.encryption;
if (event.type == EventTypes.Encrypted && encryption != null) { if (event.type == EventTypes.Encrypted && encryption != null) {
var decrypted = await encryption.decryptRoomEvent(roomId, event); var decrypted = await encryption.decryptRoomEvent(event);
if (decrypted.messageType == MessageTypes.BadEncrypted && if (decrypted.messageType == MessageTypes.BadEncrypted &&
prevBatch != null) { prevBatch != null) {
await oneShotSync(); await oneShotSync();
decrypted = await encryption.decryptRoomEvent(roomId, event); decrypted = await encryption.decryptRoomEvent(event);
} }
event = decrypted; event = decrypted;
} }

View File

@ -1460,10 +1460,7 @@ class Room {
for (var i = 0; i < events.length; i++) { for (var i = 0; i < events.length; i++) {
if (events[i].type == EventTypes.Encrypted && if (events[i].type == EventTypes.Encrypted &&
events[i].content['can_request_session'] == true) { events[i].content['can_request_session'] == true) {
events[i] = await client.encryption!.decryptRoomEvent( events[i] = await client.encryption!.decryptRoomEvent(events[i]);
id,
events[i],
);
} }
} }
} }
@ -1529,10 +1526,7 @@ class Room {
// Try to decrypt encrypted events but don't update the database. // Try to decrypt encrypted events but don't update the database.
if (encrypted && client.encryptionEnabled) { if (encrypted && client.encryptionEnabled) {
if (events[i].type == EventTypes.Encrypted) { if (events[i].type == EventTypes.Encrypted) {
events[i] = await client.encryption!.decryptRoomEvent( events[i] = await client.encryption!.decryptRoomEvent(events[i]);
id,
events[i],
);
} }
} }
} }
@ -1576,7 +1570,6 @@ class Room {
// for the fragmented timeline, we don't cache the decrypted // for the fragmented timeline, we don't cache the decrypted
//message in the database //message in the database
chunk.events[i] = await client.encryption!.decryptRoomEvent( chunk.events[i] = await client.encryption!.decryptRoomEvent(
id,
chunk.events[i], chunk.events[i],
); );
} else if (client.database != null) { } else if (client.database != null) {
@ -1585,7 +1578,6 @@ class Room {
for (var i = 0; i < chunk.events.length; i++) { for (var i = 0; i < chunk.events.length; i++) {
if (chunk.events[i].content['can_request_session'] == true) { if (chunk.events[i].content['can_request_session'] == true) {
chunk.events[i] = await client.encryption!.decryptRoomEvent( chunk.events[i] = await client.encryption!.decryptRoomEvent(
id,
chunk.events[i], chunk.events[i],
store: !isArchived, store: !isArchived,
updateType: EventUpdateType.history, updateType: EventUpdateType.history,
@ -1913,10 +1905,7 @@ class Room {
final event = Event.fromMatrixEvent(matrixEvent, this); final event = Event.fromMatrixEvent(matrixEvent, this);
if (event.type == EventTypes.Encrypted && client.encryptionEnabled) { if (event.type == EventTypes.Encrypted && client.encryptionEnabled) {
// attempt decryption // attempt decryption
return await client.encryption?.decryptRoomEvent( return await client.encryption?.decryptRoomEvent(event);
id,
event,
);
} }
return event; return event;
} on MatrixException catch (err) { } on MatrixException catch (err) {

View File

@ -281,7 +281,6 @@ class Timeline {
for (var i = 0; i < newEvents.length; i++) { for (var i = 0; i < newEvents.length; i++) {
if (newEvents[i].type == EventTypes.Encrypted) { if (newEvents[i].type == EventTypes.Encrypted) {
newEvents[i] = await room.client.encryption!.decryptRoomEvent( newEvents[i] = await room.client.encryption!.decryptRoomEvent(
room.id,
newEvents[i], newEvents[i],
); );
} }
@ -390,7 +389,6 @@ class Timeline {
events[i].messageType == MessageTypes.BadEncrypted && events[i].messageType == MessageTypes.BadEncrypted &&
events[i].content['session_id'] == sessionId) { events[i].content['session_id'] == sessionId) {
events[i] = await encryption.decryptRoomEvent( events[i] = await encryption.decryptRoomEvent(
room.id,
events[i], events[i],
store: true, store: true,
updateType: EventUpdateType.history, updateType: EventUpdateType.history,
@ -707,7 +705,7 @@ class Timeline {
for (final matrixEvent in resp.chunk) { for (final matrixEvent in resp.chunk) {
var event = Event.fromMatrixEvent(matrixEvent, room); var event = Event.fromMatrixEvent(matrixEvent, room);
if (event.type == EventTypes.Encrypted && encryption != null) { if (event.type == EventTypes.Encrypted && encryption != null) {
event = await encryption.decryptRoomEvent(room.id, event); event = await encryption.decryptRoomEvent(event);
if (event.type == EventTypes.Encrypted && if (event.type == EventTypes.Encrypted &&
event.messageType == MessageTypes.BadEncrypted && event.messageType == MessageTypes.BadEncrypted &&
event.content['can_request_session'] == true) { event.content['can_request_session'] == true) {

View File

@ -68,7 +68,6 @@ class EventUpdate {
} }
try { try {
final decrpytedEvent = await encryption.decryptRoomEvent( final decrpytedEvent = await encryption.decryptRoomEvent(
room.id,
Event.fromJson(content, room), Event.fromJson(content, room),
store: store, store: store,
updateType: type, updateType: type,

View File

@ -29,7 +29,7 @@ class PushNotification {
}); });
/// Generate a Push Notification object from JSON. It also supports a /// Generate a Push Notification object from JSON. It also supports a
/// <String, String> map which usually comes from Firebase Cloud Messaging. /// `map<String, String>` which usually comes from Firebase Cloud Messaging.
factory PushNotification.fromJson(Map<String, dynamic> json) => factory PushNotification.fromJson(Map<String, dynamic> json) =>
PushNotification( PushNotification(
content: json['content'] is Map content: json['content'] is Map

View File

@ -61,7 +61,7 @@ void main() {
senderId: client.userID!, senderId: client.userID!,
); );
final decryptedEvent = final decryptedEvent =
await client.encryption!.decryptRoomEvent(roomId, encryptedEvent); await client.encryption!.decryptRoomEvent(encryptedEvent);
expect(decryptedEvent.type, 'm.room.message'); expect(decryptedEvent.type, 'm.room.message');
expect(decryptedEvent.content['msgtype'], 'm.text'); expect(decryptedEvent.content['msgtype'], 'm.text');
expect(decryptedEvent.content['text'], 'Hello foxies!'); expect(decryptedEvent.content['text'], 'Hello foxies!');
@ -80,7 +80,7 @@ void main() {
senderId: client.userID!, senderId: client.userID!,
); );
final decryptedEvent = final decryptedEvent =
await client.encryption!.decryptRoomEvent(roomId, encryptedEvent); await client.encryption!.decryptRoomEvent(encryptedEvent);
expect(decryptedEvent.type, 'm.room.message'); expect(decryptedEvent.type, 'm.room.message');
expect(decryptedEvent.content['msgtype'], 'm.text'); expect(decryptedEvent.content['msgtype'], 'm.text');
expect(decryptedEvent.content['text'], 'Hello foxies!'); expect(decryptedEvent.content['text'], 'Hello foxies!');
@ -98,12 +98,11 @@ void main() {
senderId: '@alice:example.com', senderId: '@alice:example.com',
); );
final decryptedEvent = final decryptedEvent =
await client.encryption!.decryptRoomEvent(roomId, encryptedEvent); await client.encryption!.decryptRoomEvent(encryptedEvent);
expect(decryptedEvent.type, 'm.room.message'); expect(decryptedEvent.type, 'm.room.message');
expect(decryptedEvent.content['msgtype'], 'm.text'); expect(decryptedEvent.content['msgtype'], 'm.text');
expect(decryptedEvent.content['text'], 'Hello foxies!'); expect(decryptedEvent.content['text'], 'Hello foxies!');
await client.encryption! await client.encryption!.decryptRoomEvent(encryptedEvent, store: true);
.decryptRoomEvent(roomId, encryptedEvent, store: true);
}); });
test('dispose client', () async { test('dispose client', () async {