| 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..5071297dba9872da7b747b2b45f2845c9a780bf8
|
| --- /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 talk_base::IPAddress& GetAddress() const = 0;
|
| +
|
| + // Deliver a packet sent by a different node.
|
| + virtual void ReceivePacket(const talk_base::SocketAddress& from,
|
| + const talk_base::SocketAddress& to,
|
| + const scoped_refptr<net::IOBuffer>& data,
|
| + int data_size) = 0;
|
| + };
|
| +
|
| + FakeNetworkDispatcher();
|
| +
|
| + talk_base::IPAddress AllocateAddress();
|
| +
|
| + // Must be called on the thread that the |node| works on.
|
| + void AddNode(Node* node);
|
| + void RemoveNode(Node* node);
|
| +
|
| + void DeliverPacket(const talk_base::SocketAddress& from,
|
| + const talk_base::SocketAddress& to,
|
| + const scoped_refptr<net::IOBuffer>& data,
|
| + int data_size);
|
| +
|
| + private:
|
| + typedef std::map<talk_base::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_
|
|
|