Use modern for-loops in WebCore/Modules - 2
[WebKit-https.git] / Source / WebCore / Modules / websockets / WebSocketExtensionDispatcher.cpp
1 /*
2  * Copyright (C) 2012 Google Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32
33 #if ENABLE(WEB_SOCKETS)
34
35 #include "WebSocketExtensionDispatcher.h"
36
37 #include "WebSocketExtensionParser.h"
38
39 #include <wtf/ASCIICType.h>
40 #include <wtf/HashMap.h>
41 #include <wtf/text/CString.h>
42 #include <wtf/text/StringHash.h>
43
44 namespace WebCore {
45
46 void WebSocketExtensionDispatcher::reset()
47 {
48     m_processors.clear();
49 }
50
51 void WebSocketExtensionDispatcher::addProcessor(std::unique_ptr<WebSocketExtensionProcessor> processor)
52 {
53     for (auto& extensionProcessor : m_processors) {
54         if (extensionProcessor->extensionToken() == processor->extensionToken())
55             return;
56     }
57     ASSERT(processor->handshakeString().length());
58     ASSERT(!processor->handshakeString().contains('\n'));
59     ASSERT(!processor->handshakeString().contains(static_cast<UChar>('\0')));
60     m_processors.append(WTF::move(processor));
61 }
62
63 const String WebSocketExtensionDispatcher::createHeaderValue() const
64 {
65     size_t numProcessors = m_processors.size();
66     if (!numProcessors)
67         return String();
68
69     StringBuilder builder;
70     builder.append(m_processors[0]->handshakeString());
71     for (size_t i = 1; i < numProcessors; ++i) {
72         builder.appendLiteral(", ");
73         builder.append(m_processors[i]->handshakeString());
74     }
75     return builder.toString();
76 }
77
78 void WebSocketExtensionDispatcher::appendAcceptedExtension(const String& extensionToken, HashMap<String, String>& extensionParameters)
79 {
80     if (!m_acceptedExtensionsBuilder.isEmpty())
81         m_acceptedExtensionsBuilder.appendLiteral(", ");
82     m_acceptedExtensionsBuilder.append(extensionToken);
83     // FIXME: Should use ListHashSet to keep the order of the parameters.
84     for (auto& parameter : extensionParameters) {
85         m_acceptedExtensionsBuilder.appendLiteral("; ");
86         m_acceptedExtensionsBuilder.append(parameter.key);
87         if (!parameter.value.isNull()) {
88             m_acceptedExtensionsBuilder.append('=');
89             m_acceptedExtensionsBuilder.append(parameter.value);
90         }
91     }
92 }
93
94 void WebSocketExtensionDispatcher::fail(const String& reason)
95 {
96     m_failureReason = reason;
97     m_acceptedExtensionsBuilder.clear();
98 }
99
100 bool WebSocketExtensionDispatcher::processHeaderValue(const String& headerValue)
101 {
102     if (!headerValue.length())
103         return true;
104
105     // If we don't send Sec-WebSocket-Extensions header, the server should not return the header.
106     if (!m_processors.size()) {
107         fail("Received unexpected Sec-WebSocket-Extensions header");
108         return false;
109     }
110
111     const CString headerValueData = headerValue.utf8();
112     WebSocketExtensionParser parser(headerValueData.data(), headerValueData.data() + headerValueData.length());
113     while (!parser.finished()) {
114         String extensionToken;
115         HashMap<String, String> extensionParameters;
116         if (!parser.parseExtension(extensionToken, extensionParameters)) {
117             fail("Sec-WebSocket-Extensions header is invalid");
118             return false;
119         }
120
121         size_t index = 0;
122         for (auto& processor : m_processors) {
123             if (extensionToken == processor->extensionToken()) {
124                 if (processor->processResponse(extensionParameters)) {
125                     appendAcceptedExtension(extensionToken, extensionParameters);
126                     break;
127                 }
128                 fail(processor->failureReason());
129                 return false;
130             }
131             ++index;
132         }
133         // There is no extension which can process the response.
134         if (index == m_processors.size()) {
135             fail("Received unexpected extension: " + extensionToken);
136             return false;
137         }
138     }
139     return parser.parsedSuccessfully();
140 }
141
142 String WebSocketExtensionDispatcher::acceptedExtensions() const
143 {
144     if (m_acceptedExtensionsBuilder.isEmpty())
145         return String();
146     return m_acceptedExtensionsBuilder.toStringPreserveCapacity();
147 }
148
149 String WebSocketExtensionDispatcher::failureReason() const
150 {
151     return m_failureReason;
152 }
153
154 } // namespace WebCore
155
156 #endif // ENABLE(WEB_SOCKETS)