Index: remoting/test/fake_network_dispatcher.h |
diff --git a/remoting/test/fake_network_dispatcher.h b/remoting/test/fake_network_dispatcher.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..44491a4d7bb0e7e9270d1aaaf806f2220f28243c |
--- /dev/null |
+++ b/remoting/test/fake_network_dispatcher.h |
@@ -0,0 +1,74 @@ |
+// Copyright 2014 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 REMOTING_TEST_FAKE_NETWORK_DISPATCHER_H_ |
+#define REMOTING_TEST_FAKE_NETWORK_DISPATCHER_H_ |
+ |
+#include <map> |
+ |
+#include "base/compiler_specific.h" |
+#include "base/memory/ref_counted.h" |
+#include "base/synchronization/lock.h" |
+#include "third_party/libjingle/source/talk/p2p/base/packetsocketfactory.h" |
+ |
+namespace base { |
+class SingleThreadTaskRunner; |
+} // namespace base |
+ |
+namespace net { |
+class IOBuffer; |
+} // namespace net |
+ |
+namespace remoting { |
+ |
+class FakeNetworkDispatcher |
+ : public base::RefCountedThreadSafe<FakeNetworkDispatcher> { |
+ public: |
+ class Node { |
+ public: |
+ virtual ~Node() {}; |
+ |
+ // Return thread on which ReceivePacket() should be called. |
+ virtual const scoped_refptr<base::SingleThreadTaskRunner>& GetThread() |
+ const = 0; |
+ virtual const rtc::IPAddress& GetAddress() const = 0; |
+ |
+ // Deliver a packet sent by a different node. |
+ virtual void ReceivePacket(const rtc::SocketAddress& from, |
+ const rtc::SocketAddress& to, |
+ const scoped_refptr<net::IOBuffer>& data, |
+ int data_size) = 0; |
+ }; |
+ |
+ FakeNetworkDispatcher(); |
+ |
+ rtc::IPAddress AllocateAddress(); |
+ |
+ // Must be called on the thread that the |node| works on. |
+ void AddNode(Node* node); |
+ void RemoveNode(Node* node); |
+ |
+ void DeliverPacket(const rtc::SocketAddress& from, |
+ const rtc::SocketAddress& to, |
+ const scoped_refptr<net::IOBuffer>& data, |
+ int data_size); |
+ |
+ private: |
+ typedef std::map<rtc::IPAddress, Node*> NodesMap; |
+ |
+ friend class base::RefCountedThreadSafe<FakeNetworkDispatcher>; |
+ virtual ~FakeNetworkDispatcher(); |
+ |
+ NodesMap nodes_; |
+ base::Lock nodes_lock_; |
+ |
+ // A counter used to allocate unique addresses in AllocateAddress(). |
+ int allocated_address_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FakeNetworkDispatcher); |
+}; |
+ |
+} // namespace remoting |
+ |
+#endif // REMOTING_TEST_FAKE_NETWORK_DISPATCHER_H_ |