src/third_party/WebKit/Source/modules/websockets/WebSocketDeflaterTest.cpp

   1 /*
   2  * Copyright (C) 2012 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * 1.  Redistributions of source code must retain the above copyright
   9  *     notice, this list of conditions and the following disclaimer.
  10  * 2.  Redistributions in binary form must reproduce the above copyright
  11  *     notice, this list of conditions and the following disclaimer in the
  12  *     documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  16  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  17  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  18  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  19  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  20  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  21  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "modules/websockets/WebSocketDeflater.h"
  28
  29 #include "wtf/Vector.h"
  30 #include <gtest/gtest.h>
  31
  32 namespace blink {
  33 namespace {
  34
  35 TEST(WebSocketDeflaterTest, TestCompressHello)
  36 {
  37     // Test the first example on section 4.3 of the specification.
  38     OwnPtr<WebSocketDeflater> deflater = WebSocketDeflater::create(15);
  39     ASSERT_TRUE(deflater->initialize());
  40     OwnPtr<WebSocketInflater> inflater = WebSocketInflater::create();
  41     ASSERT_TRUE(inflater->initialize());
  42     const char* inputData = "Hello";
  43     const size_t inputLength = strlen(inputData);
  44
  45     ASSERT_TRUE(deflater->addBytes(inputData, inputLength));
  46     ASSERT_TRUE(deflater->finish());
  47     const char* expectedFirst = "\xf2\x48\xcd\xc9\xc9\x07\x00";
  48     EXPECT_EQ(7U, deflater->size());
  49     EXPECT_EQ(0, memcmp(expectedFirst, deflater->data(), deflater->size()));
  50     ASSERT_TRUE(inflater->addBytes(deflater->data(), deflater->size()));
  51     ASSERT_TRUE(inflater->finish());
  52     EXPECT_EQ(inputLength, inflater->size());
  53     EXPECT_EQ(0, memcmp(inputData, inflater->data(), inflater->size()));
  54
  55     deflater->reset();
  56     inflater->reset();
  57
  58     ASSERT_TRUE(deflater->addBytes(inputData, inputLength));
  59     ASSERT_TRUE(deflater->finish());
  60     const char* expectedSecond = "\xf2\x00\x11\x00\x00";
  61     EXPECT_EQ(5U, deflater->size());
  62     EXPECT_EQ(0, memcmp(expectedSecond, deflater->data(), deflater->size()));
  63     ASSERT_TRUE(inflater->addBytes(deflater->data(), deflater->size()));
  64     ASSERT_TRUE(inflater->finish());
  65     EXPECT_EQ(inputLength, inflater->size());
  66     EXPECT_EQ(0, memcmp(inputData, inflater->data(), inflater->size()));
  67 }
  68
  69 TEST(WebSocketDeflaterTest, TestMultipleAddBytesCalls)
  70 {
  71     OwnPtr<WebSocketDeflater> deflater = WebSocketDeflater::create(15);
  72     ASSERT_TRUE(deflater->initialize());
  73     OwnPtr<WebSocketInflater> inflater = WebSocketInflater::create();
  74     ASSERT_TRUE(inflater->initialize());
  75     Vector<char> inputData(32);
  76     inputData.fill('a');
  77
  78     for (size_t i = 0; i < inputData.size(); ++i)
  79         ASSERT_TRUE(deflater->addBytes(inputData.data() + i, 1));
  80     ASSERT_TRUE(deflater->finish());
  81     for (size_t i = 0; i < deflater->size(); ++i)
  82         ASSERT_TRUE(inflater->addBytes(deflater->data() + i, 1));
  83     ASSERT_TRUE(inflater->finish());
  84     EXPECT_EQ(inputData.size(), inflater->size());
  85     EXPECT_EQ(0, memcmp(inputData.data(), inflater->data(), inflater->size()));
  86 }
  87
  88 TEST(WebSocketDeflaterTest, TestNoContextTakeOver)
  89 {
  90     OwnPtr<WebSocketDeflater> deflater = WebSocketDeflater::create(15, WebSocketDeflater::DoNotTakeOverContext);
  91     ASSERT_TRUE(deflater->initialize());
  92     OwnPtr<WebSocketInflater> inflater = WebSocketInflater::create();
  93     ASSERT_TRUE(inflater->initialize());
  94     const char* expected = "\xf2\x48\xcd\xc9\xc9\x07\x00";
  95     const char* inputData = "Hello";
  96     const size_t inputLength = strlen(inputData);
  97
  98     // If we don't take over context, the second result should be the identical
  99     // with the first one.
 100     for (size_t i = 0; i < 2; ++i) {
 101         ASSERT_TRUE(deflater->addBytes(inputData, inputLength));
 102         ASSERT_TRUE(deflater->finish());
 103         EXPECT_EQ(7U, deflater->size());
 104         EXPECT_EQ(0, memcmp(expected, deflater->data(), deflater->size()));
 105         ASSERT_TRUE(inflater->addBytes(deflater->data(), deflater->size()));
 106         ASSERT_TRUE(inflater->finish());
 107         EXPECT_EQ(inputLength, inflater->size());
 108         EXPECT_EQ(0, memcmp(inputData, inflater->data(), inflater->size()));
 109         deflater->reset();
 110         inflater->reset();
 111     }
 112 }
 113
 114 TEST(WebSocketDeflaterTest, TestWindowBits)
 115 {
 116     Vector<char> inputData(1024 + 64 * 2);
 117     inputData.fill('a');
 118     // Modify the head and tail of the inputData so that back-reference
 119     // can be used if the window size is sufficiently-large.
 120     for (size_t j = 0; j < 64; ++j) {
 121         inputData[j] = 'b';
 122         inputData[inputData.size() - j - 1] = 'b';
 123     }
 124
 125     OwnPtr<WebSocketDeflater> deflater = WebSocketDeflater::create(8);
 126     ASSERT_TRUE(deflater->initialize());
 127     ASSERT_TRUE(deflater->addBytes(inputData.data(), inputData.size()));
 128     ASSERT_TRUE(deflater->finish());
 129
 130     OwnPtr<WebSocketInflater> inflater = WebSocketInflater::create(8);
 131     ASSERT_TRUE(inflater->initialize());
 132     ASSERT_TRUE(inflater->addBytes(deflater->data(), deflater->size()));
 133     ASSERT_TRUE(inflater->finish());
 134     EXPECT_EQ(inputData.size(), inflater->size());
 135     EXPECT_EQ(0, memcmp(inputData.data(), inflater->data(), inflater->size()));
 136 }
 137
 138 TEST(WebSocketDeflaterTest, TestLargeData)
 139 {
 140     OwnPtr<WebSocketDeflater> deflater = WebSocketDeflater::create(15);
 141     ASSERT_TRUE(deflater->initialize());
 142     OwnPtr<WebSocketInflater> inflater = WebSocketInflater::create();
 143     ASSERT_TRUE(inflater->initialize());
 144     Vector<char> inputData(16 * 1024 * 1024);
 145     inputData.fill('a');
 146
 147     ASSERT_TRUE(deflater->addBytes(inputData.data(), inputData.size()));
 148     ASSERT_TRUE(deflater->finish());
 149     ASSERT_TRUE(inflater->addBytes(deflater->data(), deflater->size()));
 150     ASSERT_TRUE(inflater->finish());
 151     EXPECT_EQ(inputData.size(), inflater->size());
 152     EXPECT_EQ(0, memcmp(inputData.data(), inflater->data(), inflater->size()));
 153 }
 154
 155 } // namespace
 156 } // namespace blink