qiurui
/
uvgrtp-base
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tests: Add tests for other user managed key sizes

This commit is contained in:
Joni Räsänen 2022-09-15 10:59:53 +03:00
parent bb0cde0360
commit b0bc92b396
1 changed files with 31 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
test/test_5_srtp_zrtp.cpp
View File

@ -12,20 +12,34 @@ constexpr auto EXAMPLE_DURATION_S = std::chrono::seconds(2);
// encryption parameters of example // encryption parameters of example
enum Key_length { SRTP_128 = 128, SRTP_196 = 196, SRTP_256 = 256 }; enum Key_length { SRTP_128 = 128, SRTP_196 = 196, SRTP_256 = 256 };
constexpr Key_length KEY_SIZE = SRTP_128; // change this to change tested key length
constexpr int KEY_SIZE_BYTES = KEY_SIZE / 8;
constexpr int SALT_SIZE = 112; constexpr int SALT_SIZE = 112;
constexpr int SALT_SIZE_BYTES = SALT_SIZE / 8; constexpr int SALT_SIZE_BYTES = SALT_SIZE / 8;
void user_send_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES]); void user_send_func(uint8_t *key, uint8_t salt[SALT_SIZE_BYTES], uint8_t key_size);
void user_receive_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES]); void user_receive_func(uint8_t *key, uint8_t salt[SALT_SIZE_BYTES], uint8_t key_size);
void zrtp_sender_func(uvgrtp::session* sender_session, int sender_port, int receiver_port, unsigned int flags); void zrtp_sender_func(uvgrtp::session* sender_session, int sender_port, int receiver_port, unsigned int flags);
void zrtp_receive_func(uvgrtp::session* receiver_session, int sender_port, int receiver_port, unsigned int flags); void zrtp_receive_func(uvgrtp::session* receiver_session, int sender_port, int receiver_port, unsigned int flags);
void test_user_key(Key_length len);
// User key management test // User key management test
TEST(EncryptionTests, srtp_user_keys) TEST(EncryptionTests, srtp_user_key_128)
{
test_user_key(SRTP_128);
}
TEST(EncryptionTests, srtp_user_key_196)
{
test_user_key(SRTP_196);
}
TEST(EncryptionTests, srtp_user_key_256)
{
test_user_key(SRTP_256);
}
void test_user_key(Key_length len)
{ {
std::cout << "Starting ZRTP sender thread" << std::endl; std::cout << "Starting ZRTP sender thread" << std::endl;
uvgrtp::context ctx; uvgrtp::context ctx;
@ -37,18 +51,18 @@ TEST(EncryptionTests, srtp_user_keys)
return; return;
} }
uint8_t key[KEY_SIZE_BYTES] = { 0 }; uint8_t *key = new uint8_t[len];
uint8_t salt[SALT_SIZE_BYTES] = { 0 }; uint8_t salt[SALT_SIZE_BYTES] = { 0 };
// initialize SRTP key and salt with dummy values // initialize SRTP key and salt with dummy values
for (int i = 0; i < KEY_SIZE_BYTES; ++i) for (int i = 0; i < len; ++i)
key[i] = i; key[i] = i;
for (int i = 0; i < SALT_SIZE_BYTES; ++i) for (int i = 0; i < SALT_SIZE_BYTES; ++i)
salt[i] = i * 2; salt[i] = i * 2;
std::unique_ptr<std::thread> sender_thread = std::unique_ptr<std::thread>(new std::thread(user_send_func, key, salt)); std::unique_ptr<std::thread> sender_thread = std::unique_ptr<std::thread>(new std::thread(user_send_func, key, salt, len));
std::unique_ptr<std::thread> receiver_thread = std::unique_ptr<std::thread>(new std::thread(user_receive_func, key, salt)); std::unique_ptr<std::thread> receiver_thread = std::unique_ptr<std::thread>(new std::thread(user_receive_func, key, salt, len));
if (sender_thread && sender_thread->joinable()) if (sender_thread && sender_thread->joinable())
{ {
@ -59,9 +73,11 @@ TEST(EncryptionTests, srtp_user_keys)
{ {
receiver_thread->join(); receiver_thread->join();
} }
delete[] key;
} }
void user_send_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES]) void user_send_func(uint8_t* key, uint8_t salt[SALT_SIZE_BYTES], uint8_t key_size)
{ {
uvgrtp::context ctx; uvgrtp::context ctx;
uvgrtp::session* sender_session = nullptr; uvgrtp::session* sender_session = nullptr;
@ -71,11 +87,11 @@ void user_send_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES])
// Enable SRTP and let user manage the keys // Enable SRTP and let user manage the keys
unsigned flags = RCE_SRTP | RCE_SRTP_KMNGMNT_USER; unsigned flags = RCE_SRTP | RCE_SRTP_KMNGMNT_USER;
if (KEY_SIZE == 192) if (key_size == 192)
{ {
flags |= RCE_SRTP_KEYSIZE_192; flags |= RCE_SRTP_KEYSIZE_192;
} }
else if (KEY_SIZE == 256) else if (key_size == 256)
{ {
flags |= RCE_SRTP_KEYSIZE_256; flags |= RCE_SRTP_KEYSIZE_256;
} }
@ -103,7 +119,7 @@ void user_send_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES])
} }
} }
void user_receive_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES]) void user_receive_func(uint8_t *key, uint8_t salt[SALT_SIZE_BYTES], uint8_t key_size)
{ {
/* See sending.cc for more details */ /* See sending.cc for more details */
uvgrtp::context ctx; uvgrtp::context ctx;
@ -112,11 +128,11 @@ void user_receive_func(uint8_t key[KEY_SIZE_BYTES], uint8_t salt[SALT_SIZE_BYTES
/* Enable SRTP and let user manage keys */ /* Enable SRTP and let user manage keys */
unsigned flags = RCE_SRTP | RCE_SRTP_KMNGMNT_USER; unsigned flags = RCE_SRTP | RCE_SRTP_KMNGMNT_USER;
if (KEY_SIZE == 192) if (key_size == 192)
{ {
flags |= RCE_SRTP_KEYSIZE_192; flags |= RCE_SRTP_KEYSIZE_192;
} }
else if (KEY_SIZE == 256) else if (key_size == 256)
{ {
flags |= RCE_SRTP_KEYSIZE_256; flags |= RCE_SRTP_KEYSIZE_256;
} }