| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/websockets/websocket_extension_parser.h" | 5 #include "net/websockets/websocket_extension_parser.h" |
| 6 | 6 |
| 7 #include "base/strings/string_util.h" | 7 #include "base/strings/string_util.h" |
| 8 | 8 |
| 9 namespace net { | 9 namespace net { |
| 10 | 10 |
| 11 WebSocketExtensionParser::WebSocketExtensionParser() {} | 11 WebSocketExtensionParser::WebSocketExtensionParser() {} |
| 12 | 12 |
| 13 WebSocketExtensionParser::~WebSocketExtensionParser() {} | 13 WebSocketExtensionParser::~WebSocketExtensionParser() {} |
| 14 | 14 |
| 15 void WebSocketExtensionParser::Parse(const char* data, size_t size) { | 15 bool WebSocketExtensionParser::Parse(const char* data, size_t size) { |
| 16 current_ = data; | 16 current_ = data; |
| 17 end_ = data + size; | 17 end_ = data + size; |
| 18 has_error_ = false; | |
| 19 extensions_.clear(); | 18 extensions_.clear(); |
| 20 | 19 |
| 20 bool failed = false; |
| 21 |
| 21 while (true) { | 22 while (true) { |
| 22 WebSocketExtension extension; | 23 WebSocketExtension extension; |
| 23 ConsumeExtension(&extension); | 24 if (!ConsumeExtension(&extension)) { |
| 24 if (has_error_) | 25 failed = true; |
| 25 break; | 26 break; |
| 27 } |
| 26 extensions_.push_back(extension); | 28 extensions_.push_back(extension); |
| 27 | 29 |
| 28 ConsumeSpaces(); | 30 ConsumeSpaces(); |
| 29 DCHECK(!has_error_); | |
| 30 | 31 |
| 31 if (!ConsumeIfMatch(',')) { | 32 if (!ConsumeIfMatch(',')) { |
| 32 break; | 33 break; |
| 33 } | 34 } |
| 34 } | 35 } |
| 35 | 36 |
| 36 has_error_ = has_error_ || current_ != end_; | 37 if (!failed && current_ == end_) |
| 37 if (has_error_) | 38 return true; |
| 38 extensions_.clear(); | 39 |
| 40 extensions_.clear(); |
| 41 return false; |
| 39 } | 42 } |
| 40 | 43 |
| 41 void WebSocketExtensionParser::Consume(char c) { | 44 bool WebSocketExtensionParser::Consume(char c) { |
| 42 DCHECK(!has_error_); | |
| 43 ConsumeSpaces(); | 45 ConsumeSpaces(); |
| 44 DCHECK(!has_error_); | |
| 45 if (current_ == end_ || c != current_[0]) { | 46 if (current_ == end_ || c != current_[0]) { |
| 46 has_error_ = true; | 47 return false; |
| 47 return; | |
| 48 } | 48 } |
| 49 ++current_; | 49 ++current_; |
| 50 return true; |
| 50 } | 51 } |
| 51 | 52 |
| 52 void WebSocketExtensionParser::ConsumeExtension(WebSocketExtension* extension) { | 53 bool WebSocketExtensionParser::ConsumeExtension(WebSocketExtension* extension) { |
| 53 DCHECK(!has_error_); | |
| 54 base::StringPiece name; | 54 base::StringPiece name; |
| 55 ConsumeToken(&name); | 55 if (!ConsumeToken(&name)) |
| 56 if (has_error_) return; | 56 return false; |
| 57 *extension = WebSocketExtension(name.as_string()); | 57 *extension = WebSocketExtension(name.as_string()); |
| 58 | 58 |
| 59 while (ConsumeIfMatch(';')) { | 59 while (ConsumeIfMatch(';')) { |
| 60 WebSocketExtension::Parameter parameter((std::string())); | 60 WebSocketExtension::Parameter parameter((std::string())); |
| 61 ConsumeExtensionParameter(¶meter); | 61 if (!ConsumeExtensionParameter(¶meter)) |
| 62 if (has_error_) return; | 62 return false; |
| 63 extension->Add(parameter); | 63 extension->Add(parameter); |
| 64 } | 64 } |
| 65 |
| 66 return true; |
| 65 } | 67 } |
| 66 | 68 |
| 67 void WebSocketExtensionParser::ConsumeExtensionParameter( | 69 bool WebSocketExtensionParser::ConsumeExtensionParameter( |
| 68 WebSocketExtension::Parameter* parameter) { | 70 WebSocketExtension::Parameter* parameter) { |
| 69 DCHECK(!has_error_); | |
| 70 base::StringPiece name, value; | 71 base::StringPiece name, value; |
| 71 std::string value_string; | 72 std::string value_string; |
| 72 | 73 |
| 73 ConsumeToken(&name); | 74 if (!ConsumeToken(&name)) |
| 74 if (has_error_) return; | 75 return false; |
| 76 |
| 75 if (!ConsumeIfMatch('=')) { | 77 if (!ConsumeIfMatch('=')) { |
| 76 *parameter = WebSocketExtension::Parameter(name.as_string()); | 78 *parameter = WebSocketExtension::Parameter(name.as_string()); |
| 77 return; | 79 return true; |
| 78 } | 80 } |
| 79 | 81 |
| 80 if (Lookahead('\"')) { | 82 if (Lookahead('\"')) { |
| 81 ConsumeQuotedToken(&value_string); | 83 if (!ConsumeQuotedToken(&value_string)) |
| 84 return false; |
| 82 } else { | 85 } else { |
| 83 ConsumeToken(&value); | 86 if (!ConsumeToken(&value)) |
| 87 return false; |
| 84 value_string = value.as_string(); | 88 value_string = value.as_string(); |
| 85 } | 89 } |
| 86 if (has_error_) return; | |
| 87 *parameter = WebSocketExtension::Parameter(name.as_string(), value_string); | 90 *parameter = WebSocketExtension::Parameter(name.as_string(), value_string); |
| 91 return true; |
| 88 } | 92 } |
| 89 | 93 |
| 90 void WebSocketExtensionParser::ConsumeToken(base::StringPiece* token) { | 94 bool WebSocketExtensionParser::ConsumeToken(base::StringPiece* token) { |
| 91 DCHECK(!has_error_); | |
| 92 ConsumeSpaces(); | 95 ConsumeSpaces(); |
| 93 DCHECK(!has_error_); | |
| 94 const char* head = current_; | 96 const char* head = current_; |
| 95 while (current_ < end_ && | 97 while (current_ < end_ && |
| 96 !IsControl(current_[0]) && !IsSeparator(current_[0])) | 98 !IsControl(current_[0]) && !IsSeparator(current_[0])) |
| 97 ++current_; | 99 ++current_; |
| 98 if (current_ == head) { | 100 if (current_ == head) { |
| 99 has_error_ = true; | 101 return false; |
| 100 return; | |
| 101 } | 102 } |
| 102 *token = base::StringPiece(head, current_ - head); | 103 *token = base::StringPiece(head, current_ - head); |
| 104 return true; |
| 103 } | 105 } |
| 104 | 106 |
| 105 void WebSocketExtensionParser::ConsumeQuotedToken(std::string* token) { | 107 bool WebSocketExtensionParser::ConsumeQuotedToken(std::string* token) { |
| 106 DCHECK(!has_error_); | 108 if (!Consume('"')) |
| 107 Consume('"'); | 109 return false; |
| 108 if (has_error_) return; | 110 |
| 109 *token = ""; | 111 *token = ""; |
| 110 while (current_ < end_ && !IsControl(current_[0])) { | 112 while (current_ < end_ && !IsControl(current_[0])) { |
| 111 if (UnconsumedBytes() >= 2 && current_[0] == '\\') { | 113 if (UnconsumedBytes() >= 2 && current_[0] == '\\') { |
| 112 char next = current_[1]; | 114 char next = current_[1]; |
| 113 if (IsControl(next) || IsSeparator(next)) break; | 115 if (IsControl(next) || IsSeparator(next)) break; |
| 114 *token += next; | 116 *token += next; |
| 115 current_ += 2; | 117 current_ += 2; |
| 116 } else if (IsSeparator(current_[0])) { | 118 } else if (IsSeparator(current_[0])) { |
| 117 break; | 119 break; |
| 118 } else { | 120 } else { |
| 119 *token += current_[0]; | 121 *token += current_[0]; |
| 120 ++current_; | 122 ++current_; |
| 121 } | 123 } |
| 122 } | 124 } |
| 123 // We can't use Consume here because we don't want to consume spaces. | 125 // We can't use Consume here because we don't want to consume spaces. |
| 124 if (current_ < end_ && current_[0] == '"') | 126 if (current_ >= end_ || current_[0] != '"') |
| 125 ++current_; | 127 return false; |
| 126 else | 128 |
| 127 has_error_ = true; | 129 ++current_; |
| 128 has_error_ = has_error_ || token->empty(); | 130 |
| 131 return !token->empty(); |
| 129 } | 132 } |
| 130 | 133 |
| 131 void WebSocketExtensionParser::ConsumeSpaces() { | 134 void WebSocketExtensionParser::ConsumeSpaces() { |
| 132 DCHECK(!has_error_); | |
| 133 while (current_ < end_ && (current_[0] == ' ' || current_[0] == '\t')) | 135 while (current_ < end_ && (current_[0] == ' ' || current_[0] == '\t')) |
| 134 ++current_; | 136 ++current_; |
| 135 return; | 137 return; |
| 136 } | 138 } |
| 137 | 139 |
| 138 bool WebSocketExtensionParser::Lookahead(char c) { | 140 bool WebSocketExtensionParser::Lookahead(char c) { |
| 139 DCHECK(!has_error_); | |
| 140 const char* head = current_; | 141 const char* head = current_; |
| 141 | 142 bool result = Consume(c); |
| 142 Consume(c); | |
| 143 bool result = !has_error_; | |
| 144 current_ = head; | 143 current_ = head; |
| 145 has_error_ = false; | |
| 146 return result; | 144 return result; |
| 147 } | 145 } |
| 148 | 146 |
| 149 bool WebSocketExtensionParser::ConsumeIfMatch(char c) { | 147 bool WebSocketExtensionParser::ConsumeIfMatch(char c) { |
| 150 DCHECK(!has_error_); | |
| 151 const char* head = current_; | 148 const char* head = current_; |
| 152 | 149 if (!Consume(c)) { |
| 153 Consume(c); | |
| 154 if (has_error_) { | |
| 155 current_ = head; | 150 current_ = head; |
| 156 has_error_ = false; | |
| 157 return false; | 151 return false; |
| 158 } | 152 } |
| 153 |
| 159 return true; | 154 return true; |
| 160 } | 155 } |
| 161 | 156 |
| 162 // static | 157 // static |
| 163 bool WebSocketExtensionParser::IsControl(char c) { | 158 bool WebSocketExtensionParser::IsControl(char c) { |
| 164 return (0 <= c && c <= 31) || c == 127; | 159 return (0 <= c && c <= 31) || c == 127; |
| 165 } | 160 } |
| 166 | 161 |
| 167 // static | 162 // static |
| 168 bool WebSocketExtensionParser::IsSeparator(char c) { | 163 bool WebSocketExtensionParser::IsSeparator(char c) { |
| 169 const char separators[] = "()<>@,;:\\\"/[]?={} \t"; | 164 const char separators[] = "()<>@,;:\\\"/[]?={} \t"; |
| 170 return strchr(separators, c) != NULL; | 165 return strchr(separators, c) != NULL; |
| 171 } | 166 } |
| 172 | 167 |
| 173 } // namespace net | 168 } // namespace net |
| OLD | NEW |