| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #ifndef WEBRTC_BASE_VIRTUALSOCKETSERVER_H_ | 11 #ifndef WEBRTC_BASE_VIRTUALSOCKETSERVER_H_ |
| 12 #define WEBRTC_BASE_VIRTUALSOCKETSERVER_H_ | 12 #define WEBRTC_BASE_VIRTUALSOCKETSERVER_H_ |
| 13 | 13 |
| 14 #include <deque> | 14 #include <deque> |
| 15 #include <map> | 15 #include <map> |
| 16 | 16 |
| 17 #include "webrtc/base/checks.h" | 17 #include "webrtc/base/checks.h" |
| 18 #include "webrtc/base/constructormagic.h" | 18 #include "webrtc/base/constructormagic.h" |
| 19 #include "webrtc/base/event.h" | 19 #include "webrtc/base/event.h" |
| 20 #include "webrtc/base/fakeclock.h" |
| 20 #include "webrtc/base/messagequeue.h" | 21 #include "webrtc/base/messagequeue.h" |
| 21 #include "webrtc/base/socketserver.h" | 22 #include "webrtc/base/socketserver.h" |
| 22 | 23 |
| 23 namespace rtc { | 24 namespace rtc { |
| 24 | 25 |
| 25 class Packet; | 26 class Packet; |
| 26 class VirtualSocket; | 27 class VirtualSocket; |
| 27 class SocketAddressPair; | 28 class SocketAddressPair; |
| 28 | 29 |
| 29 // Simulates a network in the same manner as a loopback interface. The | 30 // Simulates a network in the same manner as a loopback interface. The |
| 30 // interface can create as many addresses as you want. All of the sockets | 31 // interface can create as many addresses as you want. All of the sockets |
| 31 // created by this network will be able to communicate with one another, unless | 32 // created by this network will be able to communicate with one another, unless |
| 32 // they are bound to addresses from incompatible families. | 33 // they are bound to addresses from incompatible families. |
| 33 class VirtualSocketServer : public SocketServer, public sigslot::has_slots<> { | 34 class VirtualSocketServer : public SocketServer, public sigslot::has_slots<> { |
| 34 public: | 35 public: |
| 35 VirtualSocketServer(); | 36 VirtualSocketServer(); |
| 37 // This constructor needs to be used if the test uses a fake clock and |
| 38 // ProcessMessagesUntilIdle, since ProcessMessagesUntilIdle needs a way of |
| 39 // advancing time. |
| 40 explicit VirtualSocketServer(FakeClock* fake_clock); |
| 36 ~VirtualSocketServer() override; | 41 ~VirtualSocketServer() override; |
| 37 | 42 |
| 38 // The default route indicates which local address to use when a socket is | 43 // The default route indicates which local address to use when a socket is |
| 39 // bound to the 'any' address, e.g. 0.0.0.0. | 44 // bound to the 'any' address, e.g. 0.0.0.0. |
| 40 IPAddress GetDefaultRoute(int family); | 45 IPAddress GetDefaultRoute(int family); |
| 41 void SetDefaultRoute(const IPAddress& from_addr); | 46 void SetDefaultRoute(const IPAddress& from_addr); |
| 42 | 47 |
| 43 // Limits the network bandwidth (maximum bytes per second). Zero means that | 48 // Limits the network bandwidth (maximum bytes per second). Zero means that |
| 44 // all sends occur instantly. Defaults to 0. | 49 // all sends occur instantly. Defaults to 0. |
| 45 uint32_t bandwidth() const { return bandwidth_; } | 50 uint32_t bandwidth() const { return bandwidth_; } |
| (...skipping 189 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 235 | 240 |
| 236 private: | 241 private: |
| 237 friend class VirtualSocket; | 242 friend class VirtualSocket; |
| 238 | 243 |
| 239 // Sending was previously blocked, but now isn't. | 244 // Sending was previously blocked, but now isn't. |
| 240 sigslot::signal0<> SignalReadyToSend; | 245 sigslot::signal0<> SignalReadyToSend; |
| 241 | 246 |
| 242 typedef std::map<SocketAddress, VirtualSocket*> AddressMap; | 247 typedef std::map<SocketAddress, VirtualSocket*> AddressMap; |
| 243 typedef std::map<SocketAddressPair, VirtualSocket*> ConnectionMap; | 248 typedef std::map<SocketAddressPair, VirtualSocket*> ConnectionMap; |
| 244 | 249 |
| 250 // May be null if the test doesn't use a fake clock, or it does but doesn't |
| 251 // use ProcessMessagesUntilIdle. |
| 252 FakeClock* fake_clock_ = nullptr; |
| 253 |
| 245 // Used to implement Wait/WakeUp. | 254 // Used to implement Wait/WakeUp. |
| 246 Event wakeup_; | 255 Event wakeup_; |
| 247 MessageQueue* msg_queue_; | 256 MessageQueue* msg_queue_; |
| 248 bool stop_on_idle_; | 257 bool stop_on_idle_; |
| 249 in_addr next_ipv4_; | 258 in_addr next_ipv4_; |
| 250 in6_addr next_ipv6_; | 259 in6_addr next_ipv6_; |
| 251 uint16_t next_port_; | 260 uint16_t next_port_; |
| 252 AddressMap* bindings_; | 261 AddressMap* bindings_; |
| 253 ConnectionMap* connections_; | 262 ConnectionMap* connections_; |
| 254 | 263 |
| (...skipping 127 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 382 | 391 |
| 383 // Store the options that are set | 392 // Store the options that are set |
| 384 OptionsMap options_map_; | 393 OptionsMap options_map_; |
| 385 | 394 |
| 386 friend class VirtualSocketServer; | 395 friend class VirtualSocketServer; |
| 387 }; | 396 }; |
| 388 | 397 |
| 389 } // namespace rtc | 398 } // namespace rtc |
| 390 | 399 |
| 391 #endif // WEBRTC_BASE_VIRTUALSOCKETSERVER_H_ | 400 #endif // WEBRTC_BASE_VIRTUALSOCKETSERVER_H_ |
| OLD | NEW |