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
|
// 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 CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_
#define CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_
#include <vector>
#include "base/memory/ref_counted.h"
#include "content/common/p2p_sockets.h"
#include "net/base/ip_endpoint.h"
namespace base {
class MessageLoopProxy;
} // namespace base
class P2PSocketDispatcher;
// P2P socket that rountes all calls over IPC.
//
// The object runs on two threads: IPC thread and delegate thread. The
// IPC thread is used to interact with P2PSocketDispatcher. All
// callbacks to the user of this class are called on the delegate
// thread which is specified in Init().
class P2PSocketClient : public base::RefCountedThreadSafe<P2PSocketClient> {
public:
// Delegate is called on the the same thread on the delegate thread.
class Delegate {
public:
virtual ~Delegate() { }
virtual void OnOpen(const net::IPEndPoint& address) = 0;
virtual void OnError() = 0;
virtual void OnDataReceived(const net::IPEndPoint& address,
const std::vector<char>& data) = 0;
};
explicit P2PSocketClient(P2PSocketDispatcher* dispatcher);
// Initialize socket of the specified |type| and connected to the
// specified |address|. |address| matters only when |type| is set to
// P2P_SOCKET_TCP_CLIENT.
void Init(P2PSocketType type, const net::IPEndPoint& address,
Delegate* delegate,
scoped_refptr<base::MessageLoopProxy> delegate_loop);
// Send the |data| to the |address|.
void Send(const net::IPEndPoint& address, const std::vector<char>& data);
// Must be called before the socket is destroyed. The delegate may
// not be called after |closed_task| is executed.
void Close();
int socket_id() const { return socket_id_; }
private:
enum State {
STATE_UNINITIALIZED,
STATE_OPENING,
STATE_OPEN,
STATE_CLOSED,
STATE_ERROR,
};
friend class P2PSocketDispatcher;
// Calls destructor.
friend class base::RefCountedThreadSafe<P2PSocketClient>;
virtual ~P2PSocketClient();
// Message handlers that run on IPC thread.
void OnSocketCreated(const net::IPEndPoint& address);
void OnError();
void OnDataReceived(const net::IPEndPoint& address,
const std::vector<char>& data);
// Proxy methods that deliver messages to the delegate thread.
void DeliverOnSocketCreated(const net::IPEndPoint& address);
void DeliverOnError();
void DeliverOnDataReceived(const net::IPEndPoint& address,
const std::vector<char>& data);
// Scheduled on the IPC thread to finish closing the connection.
void DoClose();
// Called by the dispatcher when it is destroyed.
void Detach();
P2PSocketDispatcher* dispatcher_;
scoped_refptr<base::MessageLoopProxy> ipc_message_loop_;
scoped_refptr<base::MessageLoopProxy> delegate_message_loop_;
int socket_id_;
Delegate* delegate_;
State state_;
DISALLOW_COPY_AND_ASSIGN(P2PSocketClient);
};
#endif // CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_
|
