OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_ | 5 #ifndef REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_ |
6 #define REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_ | 6 #define REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_ |
7 | 7 |
8 #include <list> | 8 #include <list> |
9 #include <queue> | 9 #include <queue> |
10 #include <string> | 10 #include <string> |
11 | 11 |
12 #include "base/macros.h" | 12 #include "base/macros.h" |
13 #include "base/memory/weak_ptr.h" | 13 #include "base/memory/weak_ptr.h" |
14 #include "base/observer_list.h" | 14 #include "base/observer_list.h" |
15 #include "base/threading/non_thread_safe.h" | 15 #include "base/threading/non_thread_safe.h" |
16 #include "remoting/signaling/iq_sender.h" | 16 #include "remoting/signaling/iq_sender.h" |
17 #include "remoting/signaling/signal_strategy.h" | 17 #include "remoting/signaling/signal_strategy.h" |
| 18 #include "remoting/signaling/signaling_address.h" |
18 | 19 |
19 namespace base { | 20 namespace base { |
20 class SingleThreadTaskRunner; | 21 class SingleThreadTaskRunner; |
21 } // namespace base | 22 } // namespace base |
22 | 23 |
23 namespace remoting { | 24 namespace remoting { |
24 | 25 |
25 class FakeSignalStrategy : public SignalStrategy, | 26 class FakeSignalStrategy : public SignalStrategy, |
26 public base::NonThreadSafe { | 27 public base::NonThreadSafe { |
27 public: | 28 public: |
28 // Calls ConenctTo() to connect |peer1| and |peer2|. Both |peer1| and |peer2| | 29 // Calls ConenctTo() to connect |peer1| and |peer2|. Both |peer1| and |peer2| |
29 // must belong to the current thread. | 30 // must belong to the current thread. |
30 static void Connect(FakeSignalStrategy* peer1, FakeSignalStrategy* peer2); | 31 static void Connect(FakeSignalStrategy* peer1, FakeSignalStrategy* peer2); |
31 | 32 |
32 FakeSignalStrategy(const std::string& jid); | 33 FakeSignalStrategy(const SignalingAddress& address); |
33 ~FakeSignalStrategy() override; | 34 ~FakeSignalStrategy() override; |
34 | 35 |
35 const std::list<buzz::XmlElement*>& received_messages() { | 36 const std::list<buzz::XmlElement*>& received_messages() { |
36 return received_messages_; | 37 return received_messages_; |
37 } | 38 } |
38 | 39 |
39 void set_send_delay(base::TimeDelta delay) { | 40 void set_send_delay(base::TimeDelta delay) { |
40 send_delay_ = delay; | 41 send_delay_ = delay; |
41 } | 42 } |
42 | 43 |
43 // Connects current FakeSignalStrategy to receive messages from |peer|. | 44 // Connects current FakeSignalStrategy to receive messages from |peer|. |
44 void ConnectTo(FakeSignalStrategy* peer); | 45 void ConnectTo(FakeSignalStrategy* peer); |
45 | 46 |
46 void SetLocalJid(const std::string& jid); | 47 void SetLocalAddress(const SignalingAddress& address); |
47 | 48 |
48 // Simulate IQ messages re-ordering by swapping the delivery order of | 49 // Simulate IQ messages re-ordering by swapping the delivery order of |
49 // next pair of messages. | 50 // next pair of messages. |
50 void SimulateMessageReordering(); | 51 void SimulateMessageReordering(); |
51 | 52 |
52 // SignalStrategy interface. | 53 // SignalStrategy interface. |
53 void Connect() override; | 54 void Connect() override; |
54 void Disconnect() override; | 55 void Disconnect() override; |
55 State GetState() const override; | 56 State GetState() const override; |
56 Error GetError() const override; | 57 Error GetError() const override; |
57 std::string GetLocalJid() const override; | 58 const SignalingAddress& GetLocalAddress() const override; |
58 void AddListener(Listener* listener) override; | 59 void AddListener(Listener* listener) override; |
59 void RemoveListener(Listener* listener) override; | 60 void RemoveListener(Listener* listener) override; |
60 bool SendStanza(std::unique_ptr<buzz::XmlElement> stanza) override; | 61 bool SendStanza(std::unique_ptr<buzz::XmlElement> stanza) override; |
61 std::string GetNextId() override; | 62 std::string GetNextId() override; |
62 | 63 |
63 private: | 64 private: |
64 typedef base::Callback<void(std::unique_ptr<buzz::XmlElement> message)> | 65 typedef base::Callback<void(std::unique_ptr<buzz::XmlElement> message)> |
65 PeerCallback; | 66 PeerCallback; |
66 | 67 |
67 static void DeliverMessageOnThread( | 68 static void DeliverMessageOnThread( |
68 scoped_refptr<base::SingleThreadTaskRunner> thread, | 69 scoped_refptr<base::SingleThreadTaskRunner> thread, |
69 base::WeakPtr<FakeSignalStrategy> target, | 70 base::WeakPtr<FakeSignalStrategy> target, |
70 std::unique_ptr<buzz::XmlElement> stanza); | 71 std::unique_ptr<buzz::XmlElement> stanza); |
71 | 72 |
72 // Called by the |peer_|. Takes ownership of |stanza|. | 73 // Called by the |peer_|. Takes ownership of |stanza|. |
73 void OnIncomingMessage(std::unique_ptr<buzz::XmlElement> stanza); | 74 void OnIncomingMessage(std::unique_ptr<buzz::XmlElement> stanza); |
74 void NotifyListeners(std::unique_ptr<buzz::XmlElement> stanza); | 75 void NotifyListeners(std::unique_ptr<buzz::XmlElement> stanza); |
75 void SetPeerCallback(const PeerCallback& peer_callback); | 76 void SetPeerCallback(const PeerCallback& peer_callback); |
76 | 77 |
77 scoped_refptr<base::SingleThreadTaskRunner> main_thread_; | 78 scoped_refptr<base::SingleThreadTaskRunner> main_thread_; |
78 | 79 |
79 std::string jid_; | 80 SignalingAddress address_; |
80 PeerCallback peer_callback_; | 81 PeerCallback peer_callback_; |
81 base::ObserverList<Listener, true> listeners_; | 82 base::ObserverList<Listener, true> listeners_; |
82 | 83 |
83 int last_id_; | 84 int last_id_; |
84 | 85 |
85 base::TimeDelta send_delay_; | 86 base::TimeDelta send_delay_; |
86 | 87 |
87 bool simulate_reorder_ = false; | 88 bool simulate_reorder_ = false; |
88 std::unique_ptr<buzz::XmlElement> pending_stanza_; | 89 std::unique_ptr<buzz::XmlElement> pending_stanza_; |
89 | 90 |
90 // All received messages, includes thouse still in |pending_messages_|. | 91 // All received messages, includes thouse still in |pending_messages_|. |
91 std::list<buzz::XmlElement*> received_messages_; | 92 std::list<buzz::XmlElement*> received_messages_; |
92 | 93 |
93 base::WeakPtrFactory<FakeSignalStrategy> weak_factory_; | 94 base::WeakPtrFactory<FakeSignalStrategy> weak_factory_; |
94 | 95 |
95 DISALLOW_COPY_AND_ASSIGN(FakeSignalStrategy); | 96 DISALLOW_COPY_AND_ASSIGN(FakeSignalStrategy); |
96 }; | 97 }; |
97 | 98 |
98 } // namespace remoting | 99 } // namespace remoting |
99 | 100 |
100 #endif // REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_ | 101 #endif // REMOTING_SIGNALING_FAKE_SIGNAL_STRATEGY_H_ |
OLD | NEW |