blob: c104d45be22c8df731158b5b906558e263a74f3e (
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
|
// 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.
#ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_WEB_SOCKET_PROXY_PRIVATE_API_H_
#define CHROME_BROWSER_EXTENSIONS_EXTENSION_WEB_SOCKET_PROXY_PRIVATE_API_H_
#pragma once
#include "base/timer.h"
#include "chrome/browser/extensions/extension_function.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
#include "net/base/address_list.h"
class IOThread;
namespace net {
class SingleRequestHostResolver;
}
// Base class for web socket proxy functions.
class WebSocketProxyPrivate
: public AsyncExtensionFunction, public content::NotificationObserver {
public:
WebSocketProxyPrivate();
virtual ~WebSocketProxyPrivate();
protected:
// Custom finalization.
virtual void CustomFinalize() = 0;
// ExtensionFunction implementation.
virtual bool RunImpl() OVERRIDE;
// content::NotificationObserver implementation.
virtual void Observe(
int type, const content::NotificationSource& source,
const content::NotificationDetails& details) OVERRIDE;
// Destination hostname.
std::string hostname_;
// Destination IP address.
net::AddressList addr_;
// Destination port.
int port_;
// Proxy accepts websocket connections on this port.
int listening_port_;
// Whether TLS should be used.
bool do_tls_;
// Requested parameters of connection.
std::map<std::string, std::string> map_;
private:
// Finalizes and sends respond. Overwrite 'CustomFinalize' in inherited
// classes.
void Finalize();
// Callback for DNS resolution.
void OnHostResolution(int result);
// Posts task to the IO thread, which will make dns resolution.
void ResolveHost();
// Executes on IO thread. Performs DNS resolution.
void ResolveHostIOPart(IOThread* io_thread);
// Used to signal timeout (when waiting for proxy initial launch).
base::OneShotTimer<WebSocketProxyPrivate> timer_;
// Used to register for notifications.
content::NotificationRegistrar registrar_;
// Used to cancel host resolution when out of scope.
scoped_ptr<net::SingleRequestHostResolver> resolver_;
// Callback which is called when host is resolved.
bool is_finalized_;
};
// New API function for web socket proxy, which should be used.
class WebSocketProxyPrivateGetURLForTCPFunction
: public WebSocketProxyPrivate {
public:
WebSocketProxyPrivateGetURLForTCPFunction();
virtual ~WebSocketProxyPrivateGetURLForTCPFunction();
private:
// ExtensionFunction implementation.
virtual bool RunImpl() OVERRIDE;
// WebSocketProxyPrivate implementation:
virtual void CustomFinalize() OVERRIDE;
DECLARE_EXTENSION_FUNCTION_NAME("webSocketProxyPrivate.getURLForTCP")
};
// Legacy API function for web socket proxy, to be eliminated.
class WebSocketProxyPrivateGetPassportForTCPFunction
: public WebSocketProxyPrivate {
public:
WebSocketProxyPrivateGetPassportForTCPFunction();
virtual ~WebSocketProxyPrivateGetPassportForTCPFunction();
private:
// WebSocketProxyPrivate implementation:
virtual void CustomFinalize() OVERRIDE;
DECLARE_EXTENSION_FUNCTION_NAME("webSocketProxyPrivate.getPassportForTCP")
};
#endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_WEB_SOCKET_PROXY_PRIVATE_API_H_
|
