summaryrefslogtreecommitdiffstats
path: root/net/websockets/websocket_extension_parser_test.cc
blob: c19460cb88940fe5b002b5fe45ee278c223d4a8e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
// 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 "net/websockets/websocket_extension.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {

namespace {

TEST(WebSocketExtensionParserTest, ParseEmpty) {
  WebSocketExtensionParser parser;
  parser.Parse("", 0);

  EXPECT_TRUE(parser.has_error());
}

TEST(WebSocketExtensionParserTest, ParseSimple) {
  WebSocketExtensionParser parser;
  WebSocketExtension expected("foo");

  parser.Parse("foo");

  ASSERT_FALSE(parser.has_error());
  EXPECT_TRUE(expected.Equals(parser.extension()));
}

TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithoutValue) {
  WebSocketExtensionParser parser;
  WebSocketExtension expected("foo");
  expected.Add(WebSocketExtension::Parameter("bar"));

  parser.Parse("\tfoo ; bar");

  ASSERT_FALSE(parser.has_error());
  EXPECT_TRUE(expected.Equals(parser.extension()));
}

TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithValue) {
  WebSocketExtensionParser parser;
  WebSocketExtension expected("foo");
  expected.Add(WebSocketExtension::Parameter("bar", "baz"));

  parser.Parse("foo ; bar= baz\t");

  ASSERT_FALSE(parser.has_error());
  EXPECT_TRUE(expected.Equals(parser.extension()));
}

TEST(WebSocketExtensionParserTest, ParseOneExtensionWithParams) {
  WebSocketExtensionParser parser;
  WebSocketExtension expected("foo");
  expected.Add(WebSocketExtension::Parameter("bar", "baz"));
  expected.Add(WebSocketExtension::Parameter("hoge", "fuga"));

  parser.Parse("foo ; bar= baz;\t \thoge\t\t=fuga");

  ASSERT_FALSE(parser.has_error());
  EXPECT_TRUE(expected.Equals(parser.extension()));
}

TEST(WebSocketExtensionParserTest, InvalidPatterns) {
  const char* const patterns[] = {
    "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",  // mising '='
    "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
    "foo, bar"  // multiple extensions
  };

  for (size_t i = 0; i < arraysize(patterns); ++i) {
    WebSocketExtensionParser parser;
    parser.Parse(patterns[i]);
    EXPECT_TRUE(parser.has_error());
  }
}

TEST(WebSocketExtensionParserTest, QuotedParameterValue) {
  WebSocketExtensionParser parser;
  WebSocketExtension expected("foo");
  expected.Add(WebSocketExtension::Parameter("bar", "baz"));

  parser.Parse("foo; bar = \"ba\\z\" ");

  ASSERT_FALSE(parser.has_error());
  EXPECT_TRUE(expected.Equals(parser.extension()));
}

}  // namespace

}  // namespace net