mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.5 KiB
C++
61 lines
1.5 KiB
C++
#include <crypto/encrypted_frame.hpp>
|
|
|
|
#include <crypto/crypto.hpp>
|
|
#include <crypto/crypto_libsodium.hpp>
|
|
#include <llarp_test.hpp>
|
|
#include <messages/relay_commit.hpp>
|
|
|
|
#include <test_util.hpp>
|
|
|
|
#include <catch2/catch.hpp>
|
|
|
|
using namespace ::llarp;
|
|
using namespace ::testing;
|
|
|
|
using EncryptedFrame = EncryptedFrame;
|
|
using SecretKey = SecretKey;
|
|
using PubKey = PubKey;
|
|
using LRCR = LR_CommitRecord;
|
|
|
|
class FrameTest : public test::LlarpTest<>
|
|
{
|
|
public:
|
|
SecretKey alice, bob;
|
|
};
|
|
|
|
TEST_CASE_METHOD(FrameTest, "Frame crypto")
|
|
{
|
|
EncryptedFrame f(256);
|
|
f.Fill(0);
|
|
LRCR record;
|
|
record.nextHop.Fill(1);
|
|
record.tunnelNonce.Fill(2);
|
|
record.rxid.Fill(3);
|
|
record.txid.Fill(4);
|
|
|
|
auto buf = f.Buffer();
|
|
buf->cur = buf->base + EncryptedFrameOverheadSize;
|
|
|
|
REQUIRE(record.BEncode(buf));
|
|
|
|
EXPECT_CALL(m_crypto, randbytes(_, _)).WillOnce(Invoke(&test::randbytes_impl));
|
|
|
|
EXPECT_CALL(m_crypto, dh_client(_, _, alice, _)).WillOnce(Return(true));
|
|
EXPECT_CALL(m_crypto, xchacha20(_, _, _)).Times(2).WillRepeatedly(Return(true));
|
|
EXPECT_CALL(m_crypto, hmac(_, _, _)).Times(2).WillRepeatedly(Return(true));
|
|
|
|
// rewind buffer
|
|
buf->cur = buf->base + EncryptedFrameOverheadSize;
|
|
// encrypt to alice
|
|
REQUIRE(f.EncryptInPlace(alice, bob.toPublic()));
|
|
|
|
EXPECT_CALL(m_crypto, dh_server(_, _, _, _)).WillOnce(Return(true));
|
|
|
|
// decrypt from alice
|
|
REQUIRE(f.DecryptInPlace(bob));
|
|
|
|
LRCR otherRecord;
|
|
REQUIRE(otherRecord.BDecode(buf));
|
|
REQUIRE(otherRecord == record);
|
|
}
|