blob: c4bc17131ff5eed6319cddb44f7fe26ed6640424 [file] [log] [blame]
// 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_extension_parser.h"
#include <string>
#include "base/stl_util.h"
#include "net/websockets/websocket_extension.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net {
namespace {
TEST(WebSocketExtensionParserTest, ParseEmpty) {
WebSocketExtensionParser parser;
EXPECT_FALSE(parser.Parse("", 0));
EXPECT_EQ(0U, parser.extensions().size());
}
TEST(WebSocketExtensionParserTest, ParseSimple) {
WebSocketExtensionParser parser;
WebSocketExtension expected("foo");
EXPECT_TRUE(parser.Parse("foo"));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
}
TEST(WebSocketExtensionParserTest, ParseMoreThanOnce) {
WebSocketExtensionParser parser;
WebSocketExtension expected("foo");
EXPECT_TRUE(parser.Parse("foo"));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
EXPECT_FALSE(parser.Parse(""));
EXPECT_EQ(0U, parser.extensions().size());
EXPECT_TRUE(parser.Parse("foo"));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
}
TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithoutValue) {
WebSocketExtensionParser parser;
WebSocketExtension expected("foo");
expected.Add(WebSocketExtension::Parameter("bar"));
EXPECT_TRUE(parser.Parse("\tfoo ; bar"));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
}
TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithValue) {
WebSocketExtensionParser parser;
WebSocketExtension expected("foo");
expected.Add(WebSocketExtension::Parameter("bar", "baz"));
EXPECT_TRUE(parser.Parse("foo ; bar= baz\t"));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
}
TEST(WebSocketExtensionParserTest, ParseOneExtensionWithParams) {
WebSocketExtensionParser parser;
WebSocketExtension expected("foo");
expected.Add(WebSocketExtension::Parameter("bar", "baz"));
expected.Add(WebSocketExtension::Parameter("hoge", "fuga"));
EXPECT_TRUE(parser.Parse("foo ; bar= baz;\t \thoge\t\t=fuga"));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
}
TEST(WebSocketExtensionParserTest, ParseTwoExtensions) {
WebSocketExtensionParser parser;
WebSocketExtension expected0("foo");
expected0.Add(WebSocketExtension::Parameter("alpha", "x"));
WebSocketExtension expected1("bar");
expected1.Add(WebSocketExtension::Parameter("beta", "y"));
EXPECT_TRUE(parser.Parse(" foo ; alpha = x , bar ; beta = y "));
ASSERT_EQ(2U, parser.extensions().size());
EXPECT_TRUE(expected0.Equals(parser.extensions()[0]));
EXPECT_TRUE(expected1.Equals(parser.extensions()[1]));
}
TEST(WebSocketExtensionParserTest, InvalidPatterns) {
const char* const patterns[] = {
",", // just a comma
" , ", // just a comma with surrounding spaces
"foo,", // second extension is incomplete (empty)
"foo , ", // second extension is incomplete (space)
"foo,;", // second extension is incomplete (semicolon)
"foo;, bar", // first extension is incomplete
"fo\ao", // control in extension name
"fo\x01o", // control in extension name
"fo<o", // separator in extension name
"foo/", // separator in extension name
";bar", // empty extension name
"foo bar", // missing ';'
"foo;", // extension parameter without name and value
"foo; b\ar", // control in parameter name
"foo; b\x7fr", // control in parameter name
"foo; b[r", // separator in parameter name
"foo; ba:", // separator in parameter name
"foo; =baz", // empty parameter name
"foo; bar=", // empty parameter value
"foo; =", // empty parameter name and value
"foo; bar=b\x02z", // control in parameter value
"foo; bar=b@z", // separator in parameter value
"foo; bar=b\\z", // separator in parameter value
"foo; bar=b?z", // separator in parameter value
"\"foo\"", // quoted extension name
"foo; \"bar\"", // quoted parameter name
"foo; bar=\"\a2\"", // control in quoted parameter value
"foo; bar=\"b@z\"", // separator in quoted parameter value
"foo; bar=\"b\\\\z\"", // separator in quoted parameter value
"foo; bar=\"\"", // quoted empty parameter value
"foo; bar=\"baz", // unterminated quoted string
"foo; bar=\"baz \"", // space in quoted string
"foo; bar baz", // missing '='
"foo; bar - baz", // '-' instead of '=' (note: "foo; bar-baz" is valid).
"foo; bar=\r\nbaz", // CRNL not followed by a space
"foo; bar=\r\n baz", // CRNL followed by a space
"f\xFFpp", // 8-bit character in extension name
"foo; b\xFFr=baz" // 8-bit character in parameter name
"foo; bar=b\xFF" // 8-bit character in parameter value
"foo; bar=\"b\xFF\"" // 8-bit character in quoted parameter value
"foo; bar=\"baz\\" // ends with backslash
};
for (size_t i = 0; i < base::size(patterns); ++i) {
WebSocketExtensionParser parser;
EXPECT_FALSE(parser.Parse(patterns[i]));
EXPECT_EQ(0U, parser.extensions().size());
}
}
TEST(WebSocketExtensionParserTest, QuotedParameterValue) {
WebSocketExtensionParser parser;
WebSocketExtension expected("foo");
expected.Add(WebSocketExtension::Parameter("bar", "baz"));
EXPECT_TRUE(parser.Parse("foo; bar = \"ba\\z\" "));
ASSERT_EQ(1U, parser.extensions().size());
EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
}
// This is a regression test for crbug.com/647156
TEST(WebSocketExtensionParserTest, InvalidToken) {
static const char kInvalidInput[] = "\304;\304!*777\377=\377\254\377";
WebSocketExtensionParser parser;
EXPECT_FALSE(parser.Parse(kInvalidInput));
}
} // namespace
} // namespace net