| 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 |