Differential D225 Diff 657 src/tests/crypto/key-export-test.cpp

Changeset View

Standalone View

src/tests/crypto/key-export-test.cpp

/*		/*
* This file is part of libkazv.		* This file is part of libkazv.
* SPDX-FileCopyrightText: 2025 tusooa <tusooa@kazv.moe>		* SPDX-FileCopyrightText: 2025 tusooa <tusooa@kazv.moe>
* SPDX-License-Identifier: AGPL-3.0-or-later		* SPDX-License-Identifier: AGPL-3.0-or-later
*/		*/

#include <libkazv-config.hpp>		#include <libkazv-config.hpp>
#include <catch2/catch_test_macros.hpp>		#include <catch2/catch_test_macros.hpp>
#include <key-export.hpp>		#include <key-export.hpp>
		#include <crypto.hpp>

using namespace Kazv;		using namespace Kazv;

static const std::string password = "test";		static const std::string password = "test";
static const std::string salt = "\x5e\x91\xe3\x75\xcc\x7b\xaf\x04\x91\xfe\xe3\x07\x89\x58\x1c\xb8";		static const std::string salt = "\x5e\x91\xe3\x75\xcc\x7b\xaf\x04\x91\xfe\xe3\x07\x89\x58\x1c\xb8";
static const std::size_t iterations = 1000000;		static const std::size_t iterations = 1000000;
static const ByteArray iv(16, 0);		static const ByteArray iv(16, 0);
static const ByteArray expEncryptionKey = {5, 60, 69, 180, 233, 146, 45, 161, 158, 115, 248, 156, 6, 242, 138, 45, 103, 156, 226, 255, 176, 37, 30, 182, 138, 0, 240, 157, 217, 116, 141, 128};		static const ByteArray expEncryptionKey = {5, 60, 69, 180, 233, 146, 45, 161, 158, 115, 248, 156, 6, 242, 138, 45, 103, 156, 226, 255, 176, 37, 30, 182, 138, 0, 240, 157, 217, 116, 141, 128};
▲ Show 20 Lines • Show All 93 Lines • ▼ Show 20 Lines	WHEN("does not pass HMAC")
REQUIRE(res.reason() == DecryptKeyExportErrorCodes::HMAC_FAILED);		REQUIRE(res.reason() == DecryptKeyExportErrorCodes::HMAC_FAILED);
}		}

WHEN("decrypted content is not json")		WHEN("decrypted content is not json")
{		{
auto res = decryptKeyExport(nonJsonFile, password);		auto res = decryptKeyExport(nonJsonFile, password);
REQUIRE(!res.has_value());		REQUIRE(!res.has_value());
REQUIRE(res.reason() == DecryptKeyExportErrorCodes::NOT_JSON);		REQUIRE(res.reason() == DecryptKeyExportErrorCodes::NOT_JSON);
		}
		}

		TEST_CASE("Crypto::importInboundGroupSessions()")
		{
		Crypto a(RandomTag{}, genRandomData(Crypto::constructRandomSize()));
		std::string roomId = "!room:example.org";
		auto sessionKey = a.rotateMegOlmSessionWithRandom(
		genRandomData(a.rotateMegOlmSessionRandomSize()),
		1000,
		roomId
		);
		// encrypt to get the session id
		auto plainText = R"({
		"content": {},
		"type": "m.room.message",
		"room_id": "!room:example.org"
		})"_json;
		auto encryptedContent = a.encryptMegOlm(plainText);
		auto sessionId = encryptedContent["session_id"].template get<std::string>();
		auto exportJson = nlohmann::json{
		{
		{"algorithm", "m.megolm.v1.aes-sha2"},
		{"forwarding_curve25519_key_chain", {}},
		{"room_id", roomId},
		{"sender_claimed_keys", {
		{"ed25519", a.ed25519IdentityKey()},
		}},
		{"sender_key", a.curve25519IdentityKey()},
		{"session_id", sessionId},
		{"session_key", sessionKey},
		},
		};

		Crypto b(RandomTag{}, genRandomData(Crypto::constructRandomSize()));
		WHEN("importing one")
		{
		auto res = b.importInboundGroupSessions(exportJson);
		REQUIRE(res == 1);
		}

		WHEN("importing empty array")
		{
		auto res = b.importInboundGroupSessions(nlohmann::json::array());
		REQUIRE(res == 0);
		}

		WHEN("importing non-array")
		{
		auto res = b.importInboundGroupSessions(nlohmann::json(nullptr));
		REQUIRE(res == 0);
		}

		WHEN("importing one valid and other invalid")
		{
		exportJson.push_back("a");
		auto invalidKey = exportJson[0];
		invalidKey["session_id"] = 1;
		exportJson.insert(exportJson.begin(), invalidKey);
		auto res = b.importInboundGroupSessions(exportJson);
		REQUIRE(res == 1);
		}

		WHEN("importing bad session key")
		{
		exportJson[0]["session_key"] = "bad_session_key";
		auto res = b.importInboundGroupSessions(exportJson);
		REQUIRE(res == 0);
}		}
}		}