| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/websockets/websocket_frame_parser.h" | 5 #include "net/websockets/websocket_frame_parser.h" |
| 6 | 6 |
| 7 #include <stdint.h> |
| 7 #include <algorithm> | 8 #include <algorithm> |
| 8 #include <vector> | 9 #include <vector> |
| 9 | 10 |
| 10 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 11 #include "base/memory/scoped_vector.h" | 12 #include "base/memory/scoped_vector.h" |
| 12 #include "base/port.h" | |
| 13 #include "net/base/io_buffer.h" | 13 #include "net/base/io_buffer.h" |
| 14 #include "net/websockets/websocket_frame.h" | 14 #include "net/websockets/websocket_frame.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 16 |
| 17 namespace net { | 17 namespace net { |
| 18 | 18 |
| 19 namespace { | 19 namespace { |
| 20 | 20 |
| 21 const char kHello[] = "Hello, world!"; | 21 const char kHello[] = "Hello, world!"; |
| 22 const uint64 kHelloLength = arraysize(kHello) - 1; | 22 const uint64 kHelloLength = arraysize(kHello) - 1; |
| 23 const char kHelloFrame[] = "\x81\x0DHello, world!"; | 23 const char kHelloFrame[] = "\x81\x0DHello, world!"; |
| 24 const uint64 kHelloFrameLength = arraysize(kHelloFrame) - 1; | 24 const uint64 kHelloFrameLength = arraysize(kHelloFrame) - 1; |
| 25 const char kMaskedHelloFrame[] = | 25 const char kMaskedHelloFrame[] = |
| 26 "\x81\x8D\xDE\xAD\xBE\xEF" | 26 "\x81\x8D\xDE\xAD\xBE\xEF" |
| 27 "\x96\xC8\xD2\x83\xB1\x81\x9E\x98\xB1\xDF\xD2\x8B\xFF"; | 27 "\x96\xC8\xD2\x83\xB1\x81\x9E\x98\xB1\xDF\xD2\x8B\xFF"; |
| 28 const uint64 kMaskedHelloFrameLength = arraysize(kMaskedHelloFrame) - 1; | 28 const uint64 kMaskedHelloFrameLength = arraysize(kMaskedHelloFrame) - 1; |
| 29 | 29 |
| 30 struct FrameHeaderTestCase { | 30 struct FrameHeaderTestCase { |
| 31 const char* frame_header; | 31 const char* frame_header; |
| 32 size_t frame_header_length; | 32 size_t frame_header_length; |
| 33 uint64 frame_length; | 33 uint64 frame_length; |
| 34 WebSocketError error_code; | 34 WebSocketError error_code; |
| 35 }; | 35 }; |
| 36 | 36 |
| 37 const FrameHeaderTestCase kFrameHeaderTests[] = { | 37 const FrameHeaderTestCase kFrameHeaderTests[] = { |
| 38 { "\x81\x00", 2, GG_UINT64_C(0), kWebSocketNormalClosure }, | 38 { "\x81\x00", 2, UINT64_C(0), kWebSocketNormalClosure }, |
| 39 { "\x81\x7D", 2, GG_UINT64_C(125), kWebSocketNormalClosure }, | 39 { "\x81\x7D", 2, UINT64_C(125), kWebSocketNormalClosure }, |
| 40 { "\x81\x7E\x00\x7E", 4, GG_UINT64_C(126), kWebSocketNormalClosure }, | 40 { "\x81\x7E\x00\x7E", 4, UINT64_C(126), kWebSocketNormalClosure }, |
| 41 { "\x81\x7E\xFF\xFF", 4, GG_UINT64_C(0xFFFF), kWebSocketNormalClosure }, | 41 { "\x81\x7E\xFF\xFF", 4, UINT64_C(0xFFFF), kWebSocketNormalClosure }, |
| 42 { "\x81\x7F\x00\x00\x00\x00\x00\x01\x00\x00", 10, GG_UINT64_C(0x10000), | 42 { "\x81\x7F\x00\x00\x00\x00\x00\x01\x00\x00", 10, UINT64_C(0x10000), |
| 43 kWebSocketNormalClosure }, | 43 kWebSocketNormalClosure }, |
| 44 { "\x81\x7F\x00\x00\x00\x00\x7F\xFF\xFF\xFF", 10, GG_UINT64_C(0x7FFFFFFF), | 44 { "\x81\x7F\x00\x00\x00\x00\x7F\xFF\xFF\xFF", 10, UINT64_C(0x7FFFFFFF), |
| 45 kWebSocketNormalClosure }, | 45 kWebSocketNormalClosure }, |
| 46 { "\x81\x7F\x00\x00\x00\x00\x80\x00\x00\x00", 10, GG_UINT64_C(0x80000000), | 46 { "\x81\x7F\x00\x00\x00\x00\x80\x00\x00\x00", 10, UINT64_C(0x80000000), |
| 47 kWebSocketErrorMessageTooBig }, | 47 kWebSocketErrorMessageTooBig }, |
| 48 { "\x81\x7F\x7F\xFF\xFF\xFF\xFF\xFF\xFF\xFF", 10, | 48 { "\x81\x7F\x7F\xFF\xFF\xFF\xFF\xFF\xFF\xFF", 10, |
| 49 GG_UINT64_C(0x7FFFFFFFFFFFFFFF), kWebSocketErrorMessageTooBig } | 49 UINT64_C(0x7FFFFFFFFFFFFFFF), kWebSocketErrorMessageTooBig } |
| 50 }; | 50 }; |
| 51 const int kNumFrameHeaderTests = arraysize(kFrameHeaderTests); | 51 const int kNumFrameHeaderTests = arraysize(kFrameHeaderTests); |
| 52 | 52 |
| 53 TEST(WebSocketFrameParserTest, DecodeNormalFrame) { | 53 TEST(WebSocketFrameParserTest, DecodeNormalFrame) { |
| 54 WebSocketFrameParser parser; | 54 WebSocketFrameParser parser; |
| 55 | 55 |
| 56 ScopedVector<WebSocketFrameChunk> frames; | 56 ScopedVector<WebSocketFrameChunk> frames; |
| 57 EXPECT_TRUE(parser.Decode(kHelloFrame, kHelloFrameLength, &frames)); | 57 EXPECT_TRUE(parser.Decode(kHelloFrame, kHelloFrameLength, &frames)); |
| 58 EXPECT_EQ(kWebSocketNormalClosure, parser.websocket_error()); | 58 EXPECT_EQ(kWebSocketNormalClosure, parser.websocket_error()); |
| 59 ASSERT_EQ(1u, frames.size()); | 59 ASSERT_EQ(1u, frames.size()); |
| (...skipping 507 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 567 EXPECT_EQ(reserved3, header->reserved3); | 567 EXPECT_EQ(reserved3, header->reserved3); |
| 568 EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, header->opcode); | 568 EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, header->opcode); |
| 569 EXPECT_FALSE(header->masked); | 569 EXPECT_FALSE(header->masked); |
| 570 EXPECT_EQ(0u, header->payload_length); | 570 EXPECT_EQ(0u, header->payload_length); |
| 571 } | 571 } |
| 572 } | 572 } |
| 573 | 573 |
| 574 } // Unnamed namespace | 574 } // Unnamed namespace |
| 575 | 575 |
| 576 } // namespace net | 576 } // namespace net |
| OLD | NEW |