123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150 |
- // 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_SOCKET_FACTORY_H_
- #define REMOTING_TEST_FAKE_SOCKET_FACTORY_H_
- #include <stdint.h>
- #include <list>
- #include <memory>
- #include "base/callback_forward.h"
- #include "base/compiler_specific.h"
- #include "base/memory/weak_ptr.h"
- #include "base/time/time.h"
- #include "remoting/test/fake_network_dispatcher.h"
- #include "third_party/webrtc/api/packet_socket_factory.h"
- namespace remoting {
- class FakeNetworkDispatcher;
- class LeakyBucket;
- class FakePacketSocketFactory : public rtc::PacketSocketFactory,
- public FakeNetworkDispatcher::Node {
- public:
- // |dispatcher| must outlive the factory.
- explicit FakePacketSocketFactory(FakeNetworkDispatcher* dispatcher);
- FakePacketSocketFactory(const FakePacketSocketFactory&) = delete;
- FakePacketSocketFactory& operator=(const FakePacketSocketFactory&) = delete;
- ~FakePacketSocketFactory() override;
- void OnSocketDestroyed(int port);
- // |bandwidth| - simulated link bandwidth in bytes/second. 0 indicates that
- // bandwidth is unlimited.
- // |max_buffer| - size of buffers in bytes. Ignored when |bandwidth| is 0.
- void SetBandwidth(int bandwidth, int max_buffer);
- // Specifies parameters for simulated network latency. Latency is generated
- // with normal distribution around |average| with the given |stddev|. Random
- // latency calculated based on these parameters is added to the buffering
- // delay (which is calculated based on the parameters passed to
- // SetBandwidth()). I.e. total latency for each packet is calculated using the
- // following formula
- //
- // l = NormalRand(average, stddev) + bytes_buffered / bandwidth .
- //
- // Where bytes_buffered is the current level in the leaky bucket used to
- // control bandwidth.
- void SetLatency(base::TimeDelta average, base::TimeDelta stddev);
- void set_out_of_order_rate(double out_of_order_rate) {
- out_of_order_rate_ = out_of_order_rate;
- }
- void ResetStats();
- base::TimeDelta average_buffer_delay() {
- return total_packets_received_ > 0
- ? (total_buffer_delay_ / total_packets_received_)
- : base::TimeDelta();
- }
- base::TimeDelta max_buffer_delay() { return max_buffer_delay_; }
- double drop_rate() {
- return static_cast<double>(total_packets_dropped_) /
- (total_packets_received_ + total_packets_dropped_);
- }
- // rtc::PacketSocketFactory interface.
- rtc::AsyncPacketSocket* CreateUdpSocket(
- const rtc::SocketAddress& local_address,
- uint16_t min_port,
- uint16_t max_port) override;
- rtc::AsyncListenSocket* CreateServerTcpSocket(
- const rtc::SocketAddress& local_address,
- uint16_t min_port,
- uint16_t max_port,
- int opts) override;
- rtc::AsyncPacketSocket* CreateClientTcpSocket(
- const rtc::SocketAddress& local_address,
- const rtc::SocketAddress& remote_address,
- const rtc::ProxyInfo& proxy_info,
- const std::string& user_agent,
- const rtc::PacketSocketTcpOptions& opts) override;
- rtc::AsyncResolverInterface* CreateAsyncResolver() override;
- // FakeNetworkDispatcher::Node interface.
- const scoped_refptr<base::SingleThreadTaskRunner>& GetThread() const override;
- const rtc::IPAddress& GetAddress() const override;
- void ReceivePacket(const rtc::SocketAddress& from,
- const rtc::SocketAddress& to,
- const scoped_refptr<net::IOBuffer>& data,
- int data_size) override;
- private:
- struct PendingPacket {
- PendingPacket();
- PendingPacket(
- const rtc::SocketAddress& from,
- const rtc::SocketAddress& to,
- const scoped_refptr<net::IOBuffer>& data,
- int data_size);
- PendingPacket(const PendingPacket& other);
- ~PendingPacket();
- rtc::SocketAddress from;
- rtc::SocketAddress to;
- scoped_refptr<net::IOBuffer> data;
- int data_size;
- };
- using ReceiveCallback =
- base::RepeatingCallback<void(const rtc::SocketAddress& from,
- const rtc::SocketAddress& to,
- const scoped_refptr<net::IOBuffer>& data,
- int data_size)>;
- typedef std::map<uint16_t, ReceiveCallback> UdpSocketsMap;
- void DoReceivePacket();
- scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
- scoped_refptr<FakeNetworkDispatcher> dispatcher_;
- rtc::IPAddress address_;
- std::unique_ptr<LeakyBucket> leaky_bucket_;
- base::TimeDelta latency_average_;
- base::TimeDelta latency_stddev_;
- double out_of_order_rate_;
- UdpSocketsMap udp_sockets_;
- uint16_t next_port_;
- std::list<PendingPacket> pending_packets_;
- int total_packets_received_ = 0;
- int total_packets_dropped_ = 0;
- base::TimeDelta total_buffer_delay_;
- base::TimeDelta max_buffer_delay_;
- base::WeakPtrFactory<FakePacketSocketFactory> weak_factory_{this};
- };
- } // namespace remoting
- #endif // REMOTING_TEST_FAKE_SOCKET_FACTORY_H_
|