mirror of
https://github.com/Nheko-Reborn/nheko.git
synced 2024-11-25 20:48:52 +03:00
Merge branch 'master' of ssh://github.com/Nheko-Reborn/nheko
This commit is contained in:
Joseph Donofry
commit
a6c89d1362
5 changed files with 68 additions and 10 deletions
|
|
@ -361,7 +361,7 @@ if(USE_BUNDLED_MTXCLIENT)
|
|||
FetchContent_Declare(
|
||||
MatrixClient
|
||||
GIT_REPOSITORY https://github.com/Nheko-Reborn/mtxclient.git
|
||||
GIT_TAG 808605299937203696a572c585a51509b1de28cf
|
||||
GIT_TAG 5b7654c5d4512abc38806a0f44efc199029ceef4
|
||||
)
|
||||
set(BUILD_LIB_EXAMPLES OFF CACHE INTERNAL "")
|
||||
set(BUILD_LIB_TESTS OFF CACHE INTERNAL "")
|
||||
|
|
|
|||
|
|
@ -148,7 +148,7 @@ modules:
|
|||
buildsystem: cmake-ninja
|
||||
name: mtxclient
|
||||
sources:
|
||||
- commit: 7194b4f058406b1c10d3741d83abcf2d8963d849
|
||||
- commit: 5b7654c5d4512abc38806a0f44efc199029ceef4
|
||||
type: git
|
||||
url: https://github.com/Nheko-Reborn/mtxclient.git
|
||||
- config-opts:
|
||||
|
|
|
|||
|
|
@ -1181,11 +1181,14 @@ Cache::saveState(const mtx::responses::Sync &res)
|
|||
|
||||
// Save joined rooms
|
||||
for (const auto &room : res.rooms.join) {
|
||||
auto statesdb = getStatesDb(txn, room.first);
|
||||
auto membersdb = getMembersDb(txn, room.first);
|
||||
auto statesdb = getStatesDb(txn, room.first);
|
||||
auto stateskeydb = getStatesKeyDb(txn, room.first);
|
||||
auto membersdb = getMembersDb(txn, room.first);
|
||||
|
||||
saveStateEvents(txn, statesdb, membersdb, room.first, room.second.state.events);
|
||||
saveStateEvents(txn, statesdb, membersdb, room.first, room.second.timeline.events);
|
||||
saveStateEvents(
|
||||
txn, statesdb, stateskeydb, membersdb, room.first, room.second.state.events);
|
||||
saveStateEvents(
|
||||
txn, statesdb, stateskeydb, membersdb, room.first, room.second.timeline.events);
|
||||
|
||||
saveTimelineMessages(txn, room.first, room.second.timeline);
|
||||
|
||||
|
|
|
|||
|
|
@ -273,6 +273,18 @@ public:
|
|||
return false;
|
||||
}
|
||||
|
||||
static int compare_state_key(const MDB_val *a, const MDB_val *b)
|
||||
{
|
||||
auto get_skey = [](const MDB_val *v) {
|
||||
return nlohmann::json::parse(
|
||||
std::string_view(static_cast<const char *>(v->mv_data),
|
||||
v->mv_size))
|
||||
.value("key", "");
|
||||
};
|
||||
|
||||
return get_skey(a).compare(get_skey(b));
|
||||
}
|
||||
|
||||
signals:
|
||||
void newReadReceipts(const QString &room_id, const std::vector<QString> &event_ids);
|
||||
void roomReadStatus(const std::map<QString, bool> &status);
|
||||
|
|
@ -323,17 +335,19 @@ private:
|
|||
template<class T>
|
||||
void saveStateEvents(lmdb::txn &txn,
|
||||
lmdb::dbi &statesdb,
|
||||
lmdb::dbi &stateskeydb,
|
||||
lmdb::dbi &membersdb,
|
||||
const std::string &room_id,
|
||||
const std::vector<T> &events)
|
||||
{
|
||||
for (const auto &e : events)
|
||||
saveStateEvent(txn, statesdb, membersdb, room_id, e);
|
||||
saveStateEvent(txn, statesdb, stateskeydb, membersdb, room_id, e);
|
||||
}
|
||||
|
||||
template<class T>
|
||||
void saveStateEvent(lmdb::txn &txn,
|
||||
lmdb::dbi &statesdb,
|
||||
lmdb::dbi &stateskeydb,
|
||||
lmdb::dbi &membersdb,
|
||||
const std::string &room_id,
|
||||
const T &event)
|
||||
|
|
@ -371,9 +385,22 @@ private:
|
|||
}
|
||||
|
||||
std::visit(
|
||||
[&txn, &statesdb](auto e) {
|
||||
if (isStateEvent(e) && e.type != EventType::Unsupported)
|
||||
statesdb.put(txn, to_string(e.type), json(e).dump());
|
||||
[&txn, &statesdb, &stateskeydb](auto e) {
|
||||
if constexpr (isStateEvent(e))
|
||||
if (e.type != EventType::Unsupported) {
|
||||
if (e.state_key.empty())
|
||||
statesdb.put(
|
||||
txn, to_string(e.type), json(e).dump());
|
||||
else
|
||||
stateskeydb.put(
|
||||
txn,
|
||||
to_string(e.type),
|
||||
json::object({
|
||||
{"key", e.state_key},
|
||||
{"id", e.event_id},
|
||||
})
|
||||
.dump());
|
||||
}
|
||||
},
|
||||
event);
|
||||
}
|
||||
|
|
@ -461,6 +488,14 @@ private:
|
|||
return lmdb::dbi::open(txn, std::string(room_id + "/state").c_str(), MDB_CREATE);
|
||||
}
|
||||
|
||||
lmdb::dbi getStatesKeyDb(lmdb::txn &txn, const std::string &room_id)
|
||||
{
|
||||
auto db =
|
||||
lmdb::dbi::open(txn, std::string(room_id + "/state_by_key").c_str(), MDB_CREATE);
|
||||
lmdb::dbi_set_dupsort(txn, db, compare_state_key);
|
||||
return db;
|
||||
}
|
||||
|
||||
lmdb::dbi getAccountDataDb(lmdb::txn &txn, const std::string &room_id)
|
||||
{
|
||||
return lmdb::dbi::open(
|
||||
|
|
|
|||
20
src/Olm.cpp
20
src/Olm.cpp
|
|
@ -359,6 +359,26 @@ handle_olm_message(const OlmMessage &msg)
|
|||
return;
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
auto otherUserDeviceKeys = cache::userKeys(msg.sender);
|
||||
|
||||
if (!otherUserDeviceKeys)
|
||||
return;
|
||||
|
||||
std::map<std::string, std::vector<std::string>> targets;
|
||||
for (auto [device_id, key] : otherUserDeviceKeys->device_keys) {
|
||||
if (key.keys.at("curve25519:" + device_id) == msg.sender_key)
|
||||
targets[msg.sender].push_back(device_id);
|
||||
}
|
||||
|
||||
send_encrypted_to_device_messages(
|
||||
targets, mtx::events::DeviceEvent<mtx::events::msg::Dummy>{}, true);
|
||||
nhlog::crypto()->info(
|
||||
"Recovering from broken olm channel with {}:{}", msg.sender, msg.sender_key);
|
||||
} catch (std::exception &e) {
|
||||
nhlog::crypto()->error("Failed to recover from broken olm sessions: {}", e.what());
|
||||
}
|
||||
}
|
||||
|
||||
nlohmann::json
|
||||
|
|
|
|||
Loading…
Reference in a new issue