15c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)/* 25c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Copyright (C) 2012 Google Inc. All rights reserved. 35c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * 45c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Redistribution and use in source and binary forms, with or without 55c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * modification, are permitted provided that the following conditions are 65c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * met: 75c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * 85c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * * Redistributions of source code must retain the above copyright 95c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * notice, this list of conditions and the following disclaimer. 105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * * Redistributions in binary form must reproduce the above 115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * copyright notice, this list of conditions and the following disclaimer 125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * in the documentation and/or other materials provided with the 135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * distribution. 145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * * Neither the name of Google Inc. nor the names of its 155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * contributors may be used to endorse or promote products derived from 165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * this software without specific prior written permission. 175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * 185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)#include "config.h" 325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 3353e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "modules/websockets/WebSocketExtensionParser.h" 345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 3553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "wtf/ASCIICType.h" 3653e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)#include "wtf/text/CString.h" 375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 38c1847b1379d12d0e05df27436bf19a9b1bf12deaTorne (Richard Coles)namespace blink { 395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 40591b958dee2cf159d33a0b931e6231072eaf38d5Ben MurdochWebSocketExtensionParser::ParserStateBackup::~ParserStateBackup() 41591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch{ 42591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch if (!m_isDisposed) { 43591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch m_parser->m_current = m_current; 44591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch } 45591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch} 46591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch 475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::finished() 485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return m_current >= m_end; 505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::parsedSuccessfully() 535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return m_current == m_end; 555c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)static bool isSeparator(char character) 585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 595c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) static const char* separatorCharacters = "()<>@,;:\\\"/[]?={} \t"; 605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) const char* p = strchr(separatorCharacters, character); 615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return p && *p; 625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 635c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 645c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)void WebSocketExtensionParser::skipSpaces() 655c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 665c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) while (m_current < m_end && (*m_current == ' ' || *m_current == '\t')) 675c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ++m_current; 685c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::consumeToken() 715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 72591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch ParserStateBackup backup(this); 735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) skipSpaces(); 745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) const char* start = m_current; 755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) while (m_current < m_end && isASCIIPrintable(*m_current) && !isSeparator(*m_current)) 765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ++m_current; 775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (start < m_current) { 785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) m_currentToken = String(start, m_current - start); 79591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch backup.dispose(); 805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return true; 815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::consumeQuotedString() 865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 87591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch ParserStateBackup backup(this); 885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) skipSpaces(); 895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (m_current >= m_end || *m_current != '"') 905c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 915c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 925c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) Vector<char> buffer; 935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ++m_current; 945c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) while (m_current < m_end && *m_current != '"') { 955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (*m_current == '\\' && ++m_current >= m_end) 965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 97591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch // RFC6455 requires that the value after quoted-string unescaping 98591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch // MUST conform to the 'token' ABNF. 99591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch if (!isASCIIPrintable(*m_current) || isSeparator(*m_current)) 100591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch return false; 1015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) buffer.append(*m_current); 1025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ++m_current; 1035c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 104591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch if (m_current >= m_end || *m_current != '"' || buffer.isEmpty()) 1055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 1065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) m_currentToken = String::fromUTF8(buffer.data(), buffer.size()); 1075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ++m_current; 108591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch backup.dispose(); 1095c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return true; 1105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 1115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::consumeQuotedStringOrToken() 1135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 114591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch // This is ok because consumeQuotedString() restores m_current 115591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch // on failure. 1165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return consumeQuotedString() || consumeToken(); 1175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 1185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::consumeCharacter(char character) 1205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 121591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch ParserStateBackup backup(this); 1225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) skipSpaces(); 1235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (m_current < m_end && *m_current == character) { 1245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ++m_current; 125591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch backup.dispose(); 1265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return true; 1275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 1295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 1305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)bool WebSocketExtensionParser::parseExtension(String& extensionToken, HashMap<String, String>& extensionParameters) 1325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles){ 133591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch ParserStateBackup backup(this); 1345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) // Parse extension-token. 1355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (!consumeToken()) 1365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 1375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) extensionToken = currentToken(); 1395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) // Parse extension-parameters if exists. 1415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) while (consumeCharacter(';')) { 1425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (!consumeToken()) 1435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 1445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) String parameterToken = currentToken(); 1465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (consumeCharacter('=')) { 1475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (consumeQuotedStringOrToken()) 1485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) extensionParameters.add(parameterToken, currentToken()); 1495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) else 1505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 151591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch } else { 1525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) extensionParameters.add(parameterToken, String()); 153591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch } 1545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 155591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch skipSpaces(); 1565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (!finished() && !consumeCharacter(',')) 1575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return false; 1585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 159591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch backup.dispose(); 1605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return true; 1615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 1625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 163c1847b1379d12d0e05df27436bf19a9b1bf12deaTorne (Richard Coles)} // namespace blink 164