1 /* 2 * libjingle 3 * Copyright 2013 Google Inc. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright notice, 11 * this list of conditions and the following disclaimer in the documentation 12 * and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 #include "talk/app/webrtc/sctputils.h" 29 #include "webrtc/base/bytebuffer.h" 30 #include "webrtc/base/gunit.h" 31 32 class SctpUtilsTest : public testing::Test { 33 public: 34 void VerifyOpenMessageFormat(const rtc::Buffer& packet, 35 const std::string& label, 36 const webrtc::DataChannelInit& config) { 37 uint8_t message_type; 38 uint8_t channel_type; 39 uint32_t reliability; 40 uint16_t priority; 41 uint16_t label_length; 42 uint16_t protocol_length; 43 44 rtc::ByteBuffer buffer(packet.data(), packet.length()); 45 ASSERT_TRUE(buffer.ReadUInt8(&message_type)); 46 EXPECT_EQ(0x03, message_type); 47 48 ASSERT_TRUE(buffer.ReadUInt8(&channel_type)); 49 if (config.ordered) { 50 EXPECT_EQ(config.maxRetransmits > -1 ? 51 0x01 : (config.maxRetransmitTime > -1 ? 0x02 : 0), 52 channel_type); 53 } else { 54 EXPECT_EQ(config.maxRetransmits > -1 ? 55 0x81 : (config.maxRetransmitTime > -1 ? 0x82 : 0x80), 56 channel_type); 57 } 58 59 ASSERT_TRUE(buffer.ReadUInt16(&priority)); 60 61 ASSERT_TRUE(buffer.ReadUInt32(&reliability)); 62 if (config.maxRetransmits > -1 || config.maxRetransmitTime > -1) { 63 EXPECT_EQ(config.maxRetransmits > -1 ? 64 config.maxRetransmits : config.maxRetransmitTime, 65 static_cast<int>(reliability)); 66 } 67 68 ASSERT_TRUE(buffer.ReadUInt16(&label_length)); 69 ASSERT_TRUE(buffer.ReadUInt16(&protocol_length)); 70 EXPECT_EQ(label.size(), label_length); 71 EXPECT_EQ(config.protocol.size(), protocol_length); 72 73 std::string label_output; 74 ASSERT_TRUE(buffer.ReadString(&label_output, label_length)); 75 EXPECT_EQ(label, label_output); 76 std::string protocol_output; 77 ASSERT_TRUE(buffer.ReadString(&protocol_output, protocol_length)); 78 EXPECT_EQ(config.protocol, protocol_output); 79 } 80 }; 81 82 TEST_F(SctpUtilsTest, WriteParseOpenMessageWithOrderedReliable) { 83 webrtc::DataChannelInit config; 84 std::string label = "abc"; 85 config.protocol = "y"; 86 87 rtc::Buffer packet; 88 ASSERT_TRUE(webrtc::WriteDataChannelOpenMessage(label, config, &packet)); 89 90 VerifyOpenMessageFormat(packet, label, config); 91 92 std::string output_label; 93 webrtc::DataChannelInit output_config; 94 ASSERT_TRUE(webrtc::ParseDataChannelOpenMessage( 95 packet, &output_label, &output_config)); 96 97 EXPECT_EQ(label, output_label); 98 EXPECT_EQ(config.protocol, output_config.protocol); 99 EXPECT_EQ(config.ordered, output_config.ordered); 100 EXPECT_EQ(config.maxRetransmitTime, output_config.maxRetransmitTime); 101 EXPECT_EQ(config.maxRetransmits, output_config.maxRetransmits); 102 } 103 104 TEST_F(SctpUtilsTest, WriteParseOpenMessageWithMaxRetransmitTime) { 105 webrtc::DataChannelInit config; 106 std::string label = "abc"; 107 config.ordered = false; 108 config.maxRetransmitTime = 10; 109 config.protocol = "y"; 110 111 rtc::Buffer packet; 112 ASSERT_TRUE(webrtc::WriteDataChannelOpenMessage(label, config, &packet)); 113 114 VerifyOpenMessageFormat(packet, label, config); 115 116 std::string output_label; 117 webrtc::DataChannelInit output_config; 118 ASSERT_TRUE(webrtc::ParseDataChannelOpenMessage( 119 packet, &output_label, &output_config)); 120 121 EXPECT_EQ(label, output_label); 122 EXPECT_EQ(config.protocol, output_config.protocol); 123 EXPECT_EQ(config.ordered, output_config.ordered); 124 EXPECT_EQ(config.maxRetransmitTime, output_config.maxRetransmitTime); 125 EXPECT_EQ(-1, output_config.maxRetransmits); 126 } 127 128 TEST_F(SctpUtilsTest, WriteParseOpenMessageWithMaxRetransmits) { 129 webrtc::DataChannelInit config; 130 std::string label = "abc"; 131 config.maxRetransmits = 10; 132 config.protocol = "y"; 133 134 rtc::Buffer packet; 135 ASSERT_TRUE(webrtc::WriteDataChannelOpenMessage(label, config, &packet)); 136 137 VerifyOpenMessageFormat(packet, label, config); 138 139 std::string output_label; 140 webrtc::DataChannelInit output_config; 141 ASSERT_TRUE(webrtc::ParseDataChannelOpenMessage( 142 packet, &output_label, &output_config)); 143 144 EXPECT_EQ(label, output_label); 145 EXPECT_EQ(config.protocol, output_config.protocol); 146 EXPECT_EQ(config.ordered, output_config.ordered); 147 EXPECT_EQ(config.maxRetransmits, output_config.maxRetransmits); 148 EXPECT_EQ(-1, output_config.maxRetransmitTime); 149 } 150 151 TEST_F(SctpUtilsTest, WriteParseAckMessage) { 152 rtc::Buffer packet; 153 webrtc::WriteDataChannelOpenAckMessage(&packet); 154 155 uint8_t message_type; 156 rtc::ByteBuffer buffer(packet.data(), packet.length()); 157 ASSERT_TRUE(buffer.ReadUInt8(&message_type)); 158 EXPECT_EQ(0x02, message_type); 159 160 EXPECT_TRUE(webrtc::ParseDataChannelOpenAckMessage(packet)); 161 } 162 163 TEST_F(SctpUtilsTest, TestIsOpenMessage) { 164 rtc::ByteBuffer open; 165 open.WriteUInt8(0x03); 166 EXPECT_TRUE(webrtc::IsOpenMessage(open)); 167 168 rtc::ByteBuffer openAck; 169 openAck.WriteUInt8(0x02); 170 EXPECT_FALSE(webrtc::IsOpenMessage(open)); 171 172 rtc::ByteBuffer invalid; 173 openAck.WriteUInt8(0x01); 174 EXPECT_FALSE(webrtc::IsOpenMessage(invalid)); 175 176 rtc::ByteBuffer empty; 177 EXPECT_FALSE(webrtc::IsOpenMessage(empty)); 178 } 179
