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
|
// Copyright (c) 2011 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.
//
// WebSocketHandshake*Handler handles WebSocket handshake request message
// from WebKit renderer process, and WebSocket handshake response message
// from WebSocket server.
// It modifies messages for the following reason:
// - We don't trust WebKit renderer process, so we'll not expose HttpOnly
// cookies to the renderer process, so handles HttpOnly cookies in
// browser process.
#ifndef NET_WEBSOCKETS_WEBSOCKET_HANDSHAKE_HANDLER_H_
#define NET_WEBSOCKETS_WEBSOCKET_HANDSHAKE_HANDLER_H_
#pragma once
#include <string>
#include <vector>
#include "base/memory/ref_counted.h"
#include "net/http/http_request_info.h"
#include "net/http/http_response_info.h"
#include "net/spdy/spdy_framer.h"
namespace net {
class WebSocketHandshakeRequestHandler {
public:
WebSocketHandshakeRequestHandler();
~WebSocketHandshakeRequestHandler() {}
// Parses WebSocket handshake request from renderer process.
// It assumes a WebSocket handshake request message is given at once, and
// no other data is added to the request message.
bool ParseRequest(const char* data, int length);
size_t original_length() const;
// Appends the header value pair for |name| and |value|, if |name| doesn't
// exist.
void AppendHeaderIfMissing(const std::string& name,
const std::string& value);
// Removes the headers that matches (case insensitive).
void RemoveHeaders(const char* const headers_to_remove[],
size_t headers_to_remove_len);
// Gets request info to open WebSocket connection.
// Also, fill challange data in |challenge|.
HttpRequestInfo GetRequestInfo(const GURL& url, std::string* challenge);
// Gets request as SpdyHeaderBlock.
// Also, fill challenge data in |challenge|.
bool GetRequestHeaderBlock(const GURL& url,
spdy::SpdyHeaderBlock* headers,
std::string* challenge);
// Gets WebSocket handshake raw request message to open WebSocket
// connection.
std::string GetRawRequest();
// Calling raw_length is valid only after GetRawRquest() call.
size_t raw_length() const;
private:
std::string status_line_;
std::string headers_;
std::string key3_;
int original_length_;
int raw_length_;
DISALLOW_COPY_AND_ASSIGN(WebSocketHandshakeRequestHandler);
};
class WebSocketHandshakeResponseHandler {
public:
WebSocketHandshakeResponseHandler();
~WebSocketHandshakeResponseHandler();
// Parses WebSocket handshake response from WebSocket server.
// Returns number of bytes in |data| used for WebSocket handshake response
// message, including response key. If it already got whole WebSocket
// handshake response message, returns zero. In other words,
// [data + returned value, data + length) will be WebSocket frame data
// after handshake response message.
// TODO(ukai): fail fast when response gives wrong status code.
size_t ParseRawResponse(const char* data, int length);
// Returns true if it already parses full handshake response message.
bool HasResponse() const;
// Parses WebSocket handshake response info given as HttpResponseInfo.
bool ParseResponseInfo(const HttpResponseInfo& response_info,
const std::string& challenge);
// Parses WebSocket handshake response as SpdyHeaderBlock.
bool ParseResponseHeaderBlock(const spdy::SpdyHeaderBlock& headers,
const std::string& challenge);
// Gets the headers value.
void GetHeaders(const char* const headers_to_get[],
size_t headers_to_get_len,
std::vector<std::string>* values);
// Removes the headers that matches (case insensitive).
void RemoveHeaders(const char* const headers_to_remove[],
size_t headers_to_remove_len);
// Gets raw WebSocket handshake response received from WebSocket server.
std::string GetRawResponse() const;
// Gets WebSocket handshake response message sent to renderer process.
std::string GetResponse();
private:
std::string original_;
int original_header_length_;
std::string status_line_;
std::string headers_;
std::string header_separator_;
std::string key_;
DISALLOW_COPY_AND_ASSIGN(WebSocketHandshakeResponseHandler);
};
} // namespace net
#endif // NET_WEBSOCKETS_WEBSOCKET_HANDSHAKE_HANDLER_H_
|
