websocket_extension_parser_test.cc 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  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
  3. // found in the LICENSE file.
  4. #include "net/websockets/websocket_extension_parser.h"
  5. #include <string>
  6. #include "net/websockets/websocket_extension.h"
  7. #include "testing/gtest/include/gtest/gtest.h"
  8. namespace net {
  9. namespace {
  10. TEST(WebSocketExtensionParserTest, ParseEmpty) {
  11. WebSocketExtensionParser parser;
  12. EXPECT_FALSE(parser.Parse("", 0));
  13. EXPECT_EQ(0U, parser.extensions().size());
  14. }
  15. TEST(WebSocketExtensionParserTest, ParseSimple) {
  16. WebSocketExtensionParser parser;
  17. WebSocketExtension expected("foo");
  18. EXPECT_TRUE(parser.Parse("foo"));
  19. ASSERT_EQ(1U, parser.extensions().size());
  20. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  21. }
  22. TEST(WebSocketExtensionParserTest, ParseMoreThanOnce) {
  23. WebSocketExtensionParser parser;
  24. WebSocketExtension expected("foo");
  25. EXPECT_TRUE(parser.Parse("foo"));
  26. ASSERT_EQ(1U, parser.extensions().size());
  27. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  28. EXPECT_FALSE(parser.Parse(""));
  29. EXPECT_EQ(0U, parser.extensions().size());
  30. EXPECT_TRUE(parser.Parse("foo"));
  31. ASSERT_EQ(1U, parser.extensions().size());
  32. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  33. }
  34. TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithoutValue) {
  35. WebSocketExtensionParser parser;
  36. WebSocketExtension expected("foo");
  37. expected.Add(WebSocketExtension::Parameter("bar"));
  38. EXPECT_TRUE(parser.Parse("\tfoo ; bar"));
  39. ASSERT_EQ(1U, parser.extensions().size());
  40. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  41. }
  42. TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithValue) {
  43. WebSocketExtensionParser parser;
  44. WebSocketExtension expected("foo");
  45. expected.Add(WebSocketExtension::Parameter("bar", "baz"));
  46. EXPECT_TRUE(parser.Parse("foo ; bar= baz\t"));
  47. ASSERT_EQ(1U, parser.extensions().size());
  48. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  49. }
  50. TEST(WebSocketExtensionParserTest, ParseOneExtensionWithParams) {
  51. WebSocketExtensionParser parser;
  52. WebSocketExtension expected("foo");
  53. expected.Add(WebSocketExtension::Parameter("bar", "baz"));
  54. expected.Add(WebSocketExtension::Parameter("hoge", "fuga"));
  55. EXPECT_TRUE(parser.Parse("foo ; bar= baz;\t \thoge\t\t=fuga"));
  56. ASSERT_EQ(1U, parser.extensions().size());
  57. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  58. }
  59. TEST(WebSocketExtensionParserTest, ParseTwoExtensions) {
  60. WebSocketExtensionParser parser;
  61. WebSocketExtension expected0("foo");
  62. expected0.Add(WebSocketExtension::Parameter("alpha", "x"));
  63. WebSocketExtension expected1("bar");
  64. expected1.Add(WebSocketExtension::Parameter("beta", "y"));
  65. EXPECT_TRUE(parser.Parse(" foo ; alpha = x , bar ; beta = y "));
  66. ASSERT_EQ(2U, parser.extensions().size());
  67. EXPECT_TRUE(expected0.Equals(parser.extensions()[0]));
  68. EXPECT_TRUE(expected1.Equals(parser.extensions()[1]));
  69. }
  70. TEST(WebSocketExtensionParserTest, InvalidPatterns) {
  71. const char* const patterns[] = {
  72. ",", // just a comma
  73. " , ", // just a comma with surrounding spaces
  74. "foo,", // second extension is incomplete (empty)
  75. "foo , ", // second extension is incomplete (space)
  76. "foo,;", // second extension is incomplete (semicolon)
  77. "foo;, bar", // first extension is incomplete
  78. "fo\ao", // control in extension name
  79. "fo\x01o", // control in extension name
  80. "fo<o", // separator in extension name
  81. "foo/", // separator in extension name
  82. ";bar", // empty extension name
  83. "foo bar", // missing ';'
  84. "foo;", // extension parameter without name and value
  85. "foo; b\ar", // control in parameter name
  86. "foo; b\x7fr", // control in parameter name
  87. "foo; b[r", // separator in parameter name
  88. "foo; ba:", // separator in parameter name
  89. "foo; =baz", // empty parameter name
  90. "foo; bar=", // empty parameter value
  91. "foo; =", // empty parameter name and value
  92. "foo; bar=b\x02z", // control in parameter value
  93. "foo; bar=b@z", // separator in parameter value
  94. "foo; bar=b\\z", // separator in parameter value
  95. "foo; bar=b?z", // separator in parameter value
  96. "\"foo\"", // quoted extension name
  97. "foo; \"bar\"", // quoted parameter name
  98. "foo; bar=\"\a2\"", // control in quoted parameter value
  99. "foo; bar=\"b@z\"", // separator in quoted parameter value
  100. "foo; bar=\"b\\\\z\"", // separator in quoted parameter value
  101. "foo; bar=\"\"", // quoted empty parameter value
  102. "foo; bar=\"baz", // unterminated quoted string
  103. "foo; bar=\"baz \"", // space in quoted string
  104. "foo; bar baz", // missing '='
  105. "foo; bar - baz", // '-' instead of '=' (note: "foo; bar-baz" is valid).
  106. "foo; bar=\r\nbaz", // CRNL not followed by a space
  107. "foo; bar=\r\n baz", // CRNL followed by a space
  108. "f\xFFpp", // 8-bit character in extension name
  109. "foo; b\xFFr=baz" // 8-bit character in parameter name
  110. "foo; bar=b\xFF" // 8-bit character in parameter value
  111. "foo; bar=\"b\xFF\"" // 8-bit character in quoted parameter value
  112. "foo; bar=\"baz\\" // ends with backslash
  113. };
  114. for (const auto* pattern : patterns) {
  115. WebSocketExtensionParser parser;
  116. EXPECT_FALSE(parser.Parse(pattern));
  117. EXPECT_EQ(0U, parser.extensions().size());
  118. }
  119. }
  120. TEST(WebSocketExtensionParserTest, QuotedParameterValue) {
  121. WebSocketExtensionParser parser;
  122. WebSocketExtension expected("foo");
  123. expected.Add(WebSocketExtension::Parameter("bar", "baz"));
  124. EXPECT_TRUE(parser.Parse("foo; bar = \"ba\\z\" "));
  125. ASSERT_EQ(1U, parser.extensions().size());
  126. EXPECT_TRUE(expected.Equals(parser.extensions()[0]));
  127. }
  128. // This is a regression test for crbug.com/647156
  129. TEST(WebSocketExtensionParserTest, InvalidToken) {
  130. static const char kInvalidInput[] = "\304;\304!*777\377=\377\254\377";
  131. WebSocketExtensionParser parser;
  132. EXPECT_FALSE(parser.Parse(kInvalidInput));
  133. }
  134. } // namespace
  135. } // namespace net