| // Copyright 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "net/websockets/websocket_deflater.h" |
| |
| #include <string> |
| |
| #include "base/memory/ref_counted.h" |
| #include "net/base/io_buffer.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace net { |
| |
| namespace { |
| |
| std::string ToString(IOBufferWithSize* buffer) { |
| return std::string(buffer->data(), buffer->size()); |
| } |
| |
| TEST(WebSocketDeflaterTest, Construct) { |
| WebSocketDeflater deflater(WebSocketDeflater::TAKE_OVER_CONTEXT); |
| deflater.Initialize(8); |
| ASSERT_EQ(0u, deflater.CurrentOutputSize()); |
| ASSERT_TRUE(deflater.Finish()); |
| scoped_refptr<IOBufferWithSize> actual = |
| deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\00", 1), ToString(actual.get())); |
| ASSERT_EQ(0u, deflater.CurrentOutputSize()); |
| } |
| |
| TEST(WebSocketDeflaterTest, DeflateHelloTakeOverContext) { |
| WebSocketDeflater deflater(WebSocketDeflater::TAKE_OVER_CONTEXT); |
| deflater.Initialize(15); |
| scoped_refptr<IOBufferWithSize> actual1, actual2; |
| |
| ASSERT_TRUE(deflater.AddBytes("Hello", 5)); |
| ASSERT_TRUE(deflater.Finish()); |
| actual1 = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00", 7), |
| ToString(actual1.get())); |
| |
| ASSERT_TRUE(deflater.AddBytes("Hello", 5)); |
| ASSERT_TRUE(deflater.Finish()); |
| actual2 = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\xf2\x00\x11\x00\x00", 5), ToString(actual2.get())); |
| } |
| |
| TEST(WebSocketDeflaterTest, DeflateHelloDoNotTakeOverContext) { |
| WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT); |
| deflater.Initialize(15); |
| scoped_refptr<IOBufferWithSize> actual1, actual2; |
| |
| ASSERT_TRUE(deflater.AddBytes("Hello", 5)); |
| ASSERT_TRUE(deflater.Finish()); |
| actual1 = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00", 7), |
| ToString(actual1.get())); |
| |
| ASSERT_TRUE(deflater.AddBytes("Hello", 5)); |
| ASSERT_TRUE(deflater.Finish()); |
| actual2 = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00", 7), |
| ToString(actual2.get())); |
| } |
| |
| TEST(WebSocketDeflaterTest, MultipleAddBytesCalls) { |
| WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT); |
| deflater.Initialize(15); |
| std::string input(32, 'a'); |
| scoped_refptr<IOBufferWithSize> actual; |
| |
| for (size_t i = 0; i < input.size(); ++i) { |
| ASSERT_TRUE(deflater.AddBytes(&input[i], 1)); |
| } |
| ASSERT_TRUE(deflater.Finish()); |
| actual = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\x4a\x4c\xc4\x0f\x00\x00", 6), ToString(actual.get())); |
| } |
| |
| TEST(WebSocketDeflaterTest, GetMultipleDeflatedOutput) { |
| WebSocketDeflater deflater(WebSocketDeflater::TAKE_OVER_CONTEXT); |
| deflater.Initialize(15); |
| scoped_refptr<IOBufferWithSize> actual; |
| |
| ASSERT_TRUE(deflater.AddBytes("Hello", 5)); |
| ASSERT_TRUE(deflater.Finish()); |
| deflater.PushSyncMark(); |
| ASSERT_TRUE(deflater.Finish()); |
| deflater.PushSyncMark(); |
| ASSERT_TRUE(deflater.AddBytes("Hello", 5)); |
| ASSERT_TRUE(deflater.Finish()); |
| |
| actual = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("\xf2\x48\xcd\xc9\xc9\x07\x00\x00\x00\xff\xff" |
| "\x00\x00\x00\xff\xff" |
| "\xf2\x00\x11\x00\x00", 21), |
| ToString(actual.get())); |
| ASSERT_EQ(0u, deflater.CurrentOutputSize()); |
| } |
| |
| TEST(WebSocketDeflaterTest, WindowBits8) { |
| WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT); |
| deflater.Initialize(8); |
| // Set the head and tail of |input| so that back-reference |
| // can be used if the window size is sufficiently-large. |
| const std::string word = "Chromium"; |
| std::string input = word + std::string(256, 'a') + word; |
| scoped_refptr<IOBufferWithSize> actual; |
| |
| ASSERT_TRUE(deflater.AddBytes(input.data(), input.size())); |
| ASSERT_TRUE(deflater.Finish()); |
| actual = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ(std::string("r\xce(\xca\xcf\xcd,\xcdM\x1c\xe1\xc0\x39\xa3" |
| "(?7\xb3\x34\x17\x00", 21), |
| ToString(actual.get())); |
| } |
| |
| TEST(WebSocketDeflaterTest, WindowBits10) { |
| WebSocketDeflater deflater(WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT); |
| deflater.Initialize(10); |
| // Set the head and tail of |input| so that back-reference |
| // can be used if the window size is sufficiently-large. |
| const std::string word = "Chromium"; |
| std::string input = word + std::string(256, 'a') + word; |
| scoped_refptr<IOBufferWithSize> actual; |
| |
| ASSERT_TRUE(deflater.AddBytes(input.data(), input.size())); |
| ASSERT_TRUE(deflater.Finish()); |
| actual = deflater.GetOutput(deflater.CurrentOutputSize()); |
| EXPECT_EQ( |
| std::string("r\xce(\xca\xcf\xcd,\xcdM\x1c\xe1\xc0\x19\x1a\x0e\0\0", 17), |
| ToString(actual.get())); |
| } |
| |
| } // namespace |
| |
| } // namespace net |