Index: talk/session/media/srtpfilter_unittest.cc |
diff --git a/talk/session/media/srtpfilter_unittest.cc b/talk/session/media/srtpfilter_unittest.cc |
index 8122455205b68d2e3c4bfc6d0e24403387c3c670..11874380e2f4b6227ff77ce222f1d790c175b9bd 100644 |
--- a/talk/session/media/srtpfilter_unittest.cc |
+++ b/talk/session/media/srtpfilter_unittest.cc |
@@ -508,21 +508,17 @@ TEST_F(SrtpFilterTest, TestDisableEncryption) { |
// Test directly setting the params with AES_CM_128_HMAC_SHA1_80 |
TEST_F(SrtpFilterTest, TestProtect_SetParamsDirect_AES_CM_128_HMAC_SHA1_80) { |
- EXPECT_TRUE(f1_.SetRtpParams(CS_AES_CM_128_HMAC_SHA1_80, |
- kTestKey1, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_80, |
+ EXPECT_TRUE(f1_.SetRtpParams(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_80, |
kTestKey2, kTestKeyLen)); |
- EXPECT_TRUE(f2_.SetRtpParams(CS_AES_CM_128_HMAC_SHA1_80, |
- kTestKey2, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_80, |
+ EXPECT_TRUE(f2_.SetRtpParams(rtc::SRTP_AES128_CM_SHA1_80, kTestKey2, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_80, |
kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(f1_.SetRtcpParams(CS_AES_CM_128_HMAC_SHA1_80, |
- kTestKey1, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_80, |
+ EXPECT_TRUE(f1_.SetRtcpParams(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_80, |
kTestKey2, kTestKeyLen)); |
- EXPECT_TRUE(f2_.SetRtcpParams(CS_AES_CM_128_HMAC_SHA1_80, |
- kTestKey2, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_80, |
+ EXPECT_TRUE(f2_.SetRtcpParams(rtc::SRTP_AES128_CM_SHA1_80, kTestKey2, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_80, |
kTestKey1, kTestKeyLen)); |
EXPECT_TRUE(f1_.IsActive()); |
EXPECT_TRUE(f2_.IsActive()); |
@@ -531,21 +527,17 @@ TEST_F(SrtpFilterTest, TestProtect_SetParamsDirect_AES_CM_128_HMAC_SHA1_80) { |
// Test directly setting the params with AES_CM_128_HMAC_SHA1_32 |
TEST_F(SrtpFilterTest, TestProtect_SetParamsDirect_AES_CM_128_HMAC_SHA1_32) { |
- EXPECT_TRUE(f1_.SetRtpParams(CS_AES_CM_128_HMAC_SHA1_32, |
- kTestKey1, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_32, |
+ EXPECT_TRUE(f1_.SetRtpParams(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_32, |
kTestKey2, kTestKeyLen)); |
- EXPECT_TRUE(f2_.SetRtpParams(CS_AES_CM_128_HMAC_SHA1_32, |
- kTestKey2, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_32, |
+ EXPECT_TRUE(f2_.SetRtpParams(rtc::SRTP_AES128_CM_SHA1_32, kTestKey2, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_32, |
kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(f1_.SetRtcpParams(CS_AES_CM_128_HMAC_SHA1_32, |
- kTestKey1, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_32, |
+ EXPECT_TRUE(f1_.SetRtcpParams(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_32, |
kTestKey2, kTestKeyLen)); |
- EXPECT_TRUE(f2_.SetRtcpParams(CS_AES_CM_128_HMAC_SHA1_32, |
- kTestKey2, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_32, |
+ EXPECT_TRUE(f2_.SetRtcpParams(rtc::SRTP_AES128_CM_SHA1_32, kTestKey2, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_32, |
kTestKey1, kTestKeyLen)); |
EXPECT_TRUE(f1_.IsActive()); |
EXPECT_TRUE(f2_.IsActive()); |
@@ -554,25 +546,21 @@ TEST_F(SrtpFilterTest, TestProtect_SetParamsDirect_AES_CM_128_HMAC_SHA1_32) { |
// Test directly setting the params with bogus keys |
TEST_F(SrtpFilterTest, TestSetParamsKeyTooShort) { |
- EXPECT_FALSE(f1_.SetRtpParams(CS_AES_CM_128_HMAC_SHA1_80, |
- kTestKey1, kTestKeyLen - 1, |
- CS_AES_CM_128_HMAC_SHA1_80, |
+ EXPECT_FALSE(f1_.SetRtpParams(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, |
+ kTestKeyLen - 1, rtc::SRTP_AES128_CM_SHA1_80, |
kTestKey1, kTestKeyLen - 1)); |
- EXPECT_FALSE(f1_.SetRtcpParams(CS_AES_CM_128_HMAC_SHA1_80, |
- kTestKey1, kTestKeyLen - 1, |
- CS_AES_CM_128_HMAC_SHA1_80, |
+ EXPECT_FALSE(f1_.SetRtcpParams(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, |
+ kTestKeyLen - 1, rtc::SRTP_AES128_CM_SHA1_80, |
kTestKey1, kTestKeyLen - 1)); |
} |
#if defined(ENABLE_EXTERNAL_AUTH) |
TEST_F(SrtpFilterTest, TestGetSendAuthParams) { |
- EXPECT_TRUE(f1_.SetRtpParams(CS_AES_CM_128_HMAC_SHA1_32, |
- kTestKey1, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_32, |
+ EXPECT_TRUE(f1_.SetRtpParams(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_32, |
kTestKey2, kTestKeyLen)); |
- EXPECT_TRUE(f1_.SetRtcpParams(CS_AES_CM_128_HMAC_SHA1_32, |
- kTestKey1, kTestKeyLen, |
- CS_AES_CM_128_HMAC_SHA1_32, |
+ EXPECT_TRUE(f1_.SetRtcpParams(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, |
+ kTestKeyLen, rtc::SRTP_AES128_CM_SHA1_32, |
kTestKey2, kTestKeyLen)); |
uint8_t* auth_key = NULL; |
int auth_key_len = 0, auth_tag_len = 0; |
@@ -629,28 +617,30 @@ class SrtpSessionTest : public testing::Test { |
// Test that we can set up the session and keys properly. |
TEST_F(SrtpSessionTest, TestGoodSetup) { |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
} |
// Test that we can't change the keys once set. |
TEST_F(SrtpSessionTest, TestBadSetup) { |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_FALSE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey2, kTestKeyLen)); |
- EXPECT_FALSE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey2, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_FALSE( |
+ s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey2, kTestKeyLen)); |
+ EXPECT_FALSE( |
+ s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey2, kTestKeyLen)); |
} |
// Test that we fail keys of the wrong length. |
TEST_F(SrtpSessionTest, TestKeysTooShort) { |
- EXPECT_FALSE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, 1)); |
- EXPECT_FALSE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, 1)); |
+ EXPECT_FALSE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, 1)); |
+ EXPECT_FALSE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, 1)); |
} |
// Test that we can encrypt and decrypt RTP/RTCP using AES_CM_128_HMAC_SHA1_80. |
TEST_F(SrtpSessionTest, TestProtect_AES_CM_128_HMAC_SHA1_80) { |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
TestProtectRtp(CS_AES_CM_128_HMAC_SHA1_80); |
TestProtectRtcp(CS_AES_CM_128_HMAC_SHA1_80); |
TestUnprotectRtp(CS_AES_CM_128_HMAC_SHA1_80); |
@@ -659,8 +649,8 @@ TEST_F(SrtpSessionTest, TestProtect_AES_CM_128_HMAC_SHA1_80) { |
// Test that we can encrypt and decrypt RTP/RTCP using AES_CM_128_HMAC_SHA1_32. |
TEST_F(SrtpSessionTest, TestProtect_AES_CM_128_HMAC_SHA1_32) { |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_32, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_32, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, kTestKeyLen)); |
TestProtectRtp(CS_AES_CM_128_HMAC_SHA1_32); |
TestProtectRtcp(CS_AES_CM_128_HMAC_SHA1_32); |
TestUnprotectRtp(CS_AES_CM_128_HMAC_SHA1_32); |
@@ -668,7 +658,7 @@ TEST_F(SrtpSessionTest, TestProtect_AES_CM_128_HMAC_SHA1_32) { |
} |
TEST_F(SrtpSessionTest, TestGetSendStreamPacketIndex) { |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_32, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_32, kTestKey1, kTestKeyLen)); |
int64_t index; |
int out_len = 0; |
EXPECT_TRUE(s1_.ProtectRtp(rtp_packet_, rtp_len_, |
@@ -681,8 +671,8 @@ TEST_F(SrtpSessionTest, TestGetSendStreamPacketIndex) { |
// Test that we fail to unprotect if someone tampers with the RTP/RTCP paylaods. |
TEST_F(SrtpSessionTest, TestTamperReject) { |
int out_len; |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
TestProtectRtp(CS_AES_CM_128_HMAC_SHA1_80); |
TestProtectRtcp(CS_AES_CM_128_HMAC_SHA1_80); |
rtp_packet_[0] = 0x12; |
@@ -694,8 +684,8 @@ TEST_F(SrtpSessionTest, TestTamperReject) { |
// Test that we fail to unprotect if the payloads are not authenticated. |
TEST_F(SrtpSessionTest, TestUnencryptReject) { |
int out_len; |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
EXPECT_FALSE(s2_.UnprotectRtp(rtp_packet_, rtp_len_, &out_len)); |
EXPECT_FALSE(s2_.UnprotectRtcp(rtcp_packet_, rtcp_len_, &out_len)); |
} |
@@ -703,7 +693,7 @@ TEST_F(SrtpSessionTest, TestUnencryptReject) { |
// Test that we fail when using buffers that are too small. |
TEST_F(SrtpSessionTest, TestBuffersTooSmall) { |
int out_len; |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
EXPECT_FALSE(s1_.ProtectRtp(rtp_packet_, rtp_len_, |
sizeof(rtp_packet_) - 10, &out_len)); |
EXPECT_FALSE(s1_.ProtectRtcp(rtcp_packet_, rtcp_len_, |
@@ -717,8 +707,8 @@ TEST_F(SrtpSessionTest, TestReplay) { |
static const uint16_t replay_window = 1024; |
int out_len; |
- EXPECT_TRUE(s1_.SetSend(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
- EXPECT_TRUE(s2_.SetRecv(CS_AES_CM_128_HMAC_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s1_.SetSend(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
+ EXPECT_TRUE(s2_.SetRecv(rtc::SRTP_AES128_CM_SHA1_80, kTestKey1, kTestKeyLen)); |
// Initial sequence number. |
rtc::SetBE16(reinterpret_cast<uint8_t*>(rtp_packet_) + 2, seqnum_big); |