Chromium Code Reviews| Index: remoting/test/fake_connection_event_logger.cc |
| diff --git a/remoting/test/fake_connection_event_logger.cc b/remoting/test/fake_connection_event_logger.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..2fc972eca6b7fe461b1fc902f27e3ade590aba8c |
| --- /dev/null |
| +++ b/remoting/test/fake_connection_event_logger.cc |
| @@ -0,0 +1,310 @@ |
| +// Copyright 2016 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. |
| + |
| +#include "remoting/test/fake_connection_event_logger.h" |
| + |
| +#include "base/atomicops.h" |
| +#include "base/callback.h" |
| +#include "base/macros.h" |
| +#include "base/time/time.h" |
| +#include "remoting/proto/audio.pb.h" |
| +#include "remoting/proto/control.pb.h" |
| +#include "remoting/proto/video.pb.h" |
| + |
| +#ifndef ARCH_CPU_64_BITS |
| +#include "base/synchronization/lock.h" |
| +#endif |
| + |
| +namespace remoting { |
| +namespace test { |
| +namespace { |
| + |
| +template <typename T> |
| +class NoBarrierAtomic { |
| + public: |
| + T operator++() { |
| + return base::subtle::NoBarrier_AtomicIncrement(&i_, 1) - 1; |
| + } |
| + |
| + T operator++(int) { |
| + return base::subtle::NoBarrier_AtomicIncrement(&i_, 1); |
| + } |
| + |
| + T operator--() { |
| + return base::subtle::NoBarrier_AtomicIncrement(&i_, -1) - 1; |
| + } |
| + |
| + T operator--(int) { |
| + return base::subtle::NoBarrier_AtomicIncrement(&i_, -1); |
| + } |
| + |
| + T operator+=(T other) { |
| + return base::subtle::NoBarrier_AtomicIncrement(&i_, other); |
| + } |
| + |
| + T operator-=(T other) { |
| + return base::subtle::NoBarrier_AtomicIncrement(&i_, -other); |
| + } |
| + |
| + T operator*() const { |
| + return base::subtle::NoBarrier_Load(&i_); |
| + } |
| + |
| + private: |
| + volatile T i_; |
| +}; |
| + |
| +class NoBarrierAtomicInt32 : public NoBarrierAtomic<base::subtle::Atomic32> {}; |
| +#ifdef ARCH_CPU_64_BITS |
| +class NoBarrierAtomicInt64 : public NoBarrierAtomic<base::subtle::Atomic64> {}; |
| +#else // ifdef ARCH_CPU_64_BITS |
| + |
| +using base::AutoLock; |
| + |
| +// A barriered, lock based implementation |
| +class NoBarrierAtomicInt64 { |
| + public: |
| + int64_t operator++(); |
| + int64_t operator++(int); |
| + int64_t operator--(); |
| + int64_t operator--(int); |
| + int64_t operator+=(int64_t other); |
| + int64_t operator-=(int64_t other); |
| + int64_t operator*() const; |
| + |
| + private: |
| + volatile int64_t i_ = 0; |
| + mutable base::Lock lock_; // field is used in operator*() const |
| +}; |
| + |
| +int64_t NoBarrierAtomicInt64::operator++() { |
| + AutoLock l(lock_); |
| + return i_++; |
| +} |
| + |
| +int64_t NoBarrierAtomicInt64::operator++(int) { |
| + AutoLock l(lock_); |
| + return ++i_; |
| +} |
| + |
| +int64_t NoBarrierAtomicInt64::operator--() { |
| + AutoLock l(lock_); |
| + return i_--; |
| +} |
| + |
| +int64_t NoBarrierAtomicInt64::operator--(int) { |
| + AutoLock l(lock_); |
| + return --i_; |
| +} |
| + |
| +int64_t NoBarrierAtomicInt64::operator+=(int64_t other) { |
| + AutoLock l(lock_); |
| + return (i_ += other); |
| +} |
| + |
| +int64_t NoBarrierAtomicInt64::operator-=(int64_t other) { |
| + AutoLock l(lock_); |
| + return (i_ -= other); |
| +} |
| + |
| +int64_t NoBarrierAtomicInt64::operator*() const { |
| + AutoLock l(lock_); |
| + return i_; |
| +} |
| + |
| +#endif // ifdef ARCH_CPU_64_BITS |
| + |
| +class MessageCounter { |
| + public: |
| + MessageCounter(); |
| + |
| + int message_count() const { return *count_; } |
| + int64_t message_size() const { return *size_; } |
| + int last_message_size() const { return last_size_; } |
| + double DurationSeconds() const; |
| + double MessagesPerSecond() const; |
| + double SizePerSecond() const; |
| + double AverageMessageSize() const; |
| + void LogMessage(const ::google::protobuf::MessageLite& message); |
| + void DisplayStatistics(std::ostream& os, const char* name); |
| + |
| + private: |
| + NoBarrierAtomicInt32 count_; |
| + NoBarrierAtomicInt64 size_; |
| + int last_size_ = 0; |
| + base::Time start_time_; |
| + |
| + // Copy or assign the start_time_ of a MessageCounter is senseless. |
| + DISALLOW_COPY_AND_ASSIGN(MessageCounter); |
| +}; |
| + |
| +MessageCounter::MessageCounter() |
| + : count_(), |
| + size_(), |
| + start_time_(base::Time::Now()) {} |
| + |
| +double MessageCounter::DurationSeconds() const { |
| + return (base::Time::Now() - start_time_).InSecondsF(); |
| +} |
| + |
| +double MessageCounter::MessagesPerSecond() const { |
| + return static_cast<double>(message_count()) / DurationSeconds(); |
| +} |
| +double MessageCounter::SizePerSecond() const { |
| + return static_cast<double>(message_size()) / DurationSeconds(); |
| +} |
| + |
| +double MessageCounter::AverageMessageSize() const { |
| + return static_cast<double>(message_size()) / message_count(); |
| +} |
| + |
| +void MessageCounter::LogMessage( |
| + const ::google::protobuf::MessageLite& message) { |
| + count_++; |
| + last_size_ = message.ByteSize(); |
| + size_ += message.ByteSize(); |
| +} |
| + |
| +void MessageCounter::DisplayStatistics(std::ostream& os, const char* name) { |
|
joedow
2016/05/04 22:58:49
Could this be passed in via the c'tor of the class
Hzj_jie
2016/05/05 00:47:36
Done.
|
| + os << name |
| + << ": " |
| + << message_size() |
| + << " bytes in " |
| + << message_count() |
| + << " packages, last package " |
| + << last_message_size() |
| + << " bytes, " |
| + << AverageMessageSize() |
| + << " bytes/package, " |
| + << MessagesPerSecond() |
| + << " packages/sec, " |
| + << SizePerSecond() |
| + << " bytes/sec" |
| + << std::endl; |
| +} |
| + |
| +} // namespace |
| + |
| +// Analyzes messages from DeliverHostMessage function. |
| +class FakeConnectionEventLogger::CounterClientStub |
| + : public protocol::ClientStub, public MessageCounter { |
| + public: |
|
joedow
2016/05/04 22:58:49
I think these interface methods can all be private
Hzj_jie
2016/05/05 00:47:36
Done.
|
| + void DeliverHostMessage(const protocol::ExtensionMessage& message) override; |
| + void InjectClipboardEvent(const protocol::ClipboardEvent& event) override {} |
| + void SetCapabilities(const protocol::Capabilities& capabilities) override {} |
| + void SetCursorShape(const protocol::CursorShapeInfo& cursor_shape) override {} |
| + void SetPairingResponse(const protocol::PairingResponse& response) override {} |
| + void SetVideoLayout(const protocol::VideoLayout& video_layout) override {} |
| +}; |
| + |
| +void FakeConnectionEventLogger::CounterClientStub::DeliverHostMessage( |
| + const protocol::ExtensionMessage& message) { |
| + LogMessage(message); |
| +} |
| + |
| +// Analyzes messages from DeliverClientMessage function. |
| +class FakeConnectionEventLogger::CounterHostStub |
| + : public protocol::HostStub, public MessageCounter { |
| + public: |
| + void ControlAudio(const protocol::AudioControl& audio_control) override {} |
| + void ControlVideo(const protocol::VideoControl& video_control) override {} |
| + void DeliverClientMessage(const protocol::ExtensionMessage& message) override; |
| + void NotifyClientResolution( |
| + const protocol::ClientResolution& resolution) override {} |
| + void RequestPairing( |
| + const protocol::PairingRequest& pairing_request) override {} |
| + void SetCapabilities(const protocol::Capabilities& capabilities) override {} |
| +}; |
| + |
| +void FakeConnectionEventLogger::CounterHostStub::DeliverClientMessage( |
| + const protocol::ExtensionMessage& message) { |
| + LogMessage(message); |
| +} |
| + |
| +// Analyzes messages from ProcessAudioPacket function. |
| +class FakeConnectionEventLogger::CounterAudioStub |
| + : public protocol::AudioStub, public MessageCounter { |
| + public: |
| + void ProcessAudioPacket(std::unique_ptr<AudioPacket> audio_packet, |
| + const base::Closure& done) override; |
| +}; |
| + |
| +void FakeConnectionEventLogger::CounterAudioStub::ProcessAudioPacket( |
| + std::unique_ptr<AudioPacket> audio_packet, |
| + const base::Closure& done) { |
| + if (audio_packet) { |
| + LogMessage(*audio_packet); |
| + } |
| + done.Run(); |
| +} |
| + |
| +// Analyzes messages from ProcessVideoPacket function. |
| +class FakeConnectionEventLogger::CounterVideoStub |
| + : public protocol::VideoStub, public MessageCounter { |
| + public: |
| + CounterVideoStub(protocol::FakeConnectionToClient* connection); |
| + |
| + void ProcessVideoPacket(std::unique_ptr<VideoPacket> video_packet, |
| + const base::Closure& done) override; |
| + |
| + private: |
| + protocol::FakeConnectionToClient* connection_ = nullptr; |
| +}; |
| + |
| +FakeConnectionEventLogger::CounterVideoStub::CounterVideoStub( |
| + protocol::FakeConnectionToClient* connection) |
| + : connection_(connection) {} |
| + |
| +void FakeConnectionEventLogger::CounterVideoStub::ProcessVideoPacket( |
| + std::unique_ptr<VideoPacket> video_packet, |
| + const base::Closure& done) { |
| + if (video_packet && video_packet->has_capture_overhead_time_ms()) { |
| + // Not a keepalive packet. |
| + if (connection_ && |
| + connection_->video_feedback_stub()) { |
| + std::unique_ptr<VideoAck> ack(new VideoAck()); |
| + ack->set_frame_id(video_packet->frame_id()); |
| + connection_->video_feedback_stub()->ProcessVideoAck(std::move(ack)); |
| + } |
| + LogMessage(*video_packet); |
| + } |
| + done.Run(); |
| +} |
| + |
| +FakeConnectionEventLogger::FakeConnectionEventLogger( |
| + protocol::FakeConnectionToClient* connection) |
| + : client_stub_(new CounterClientStub()), |
| + host_stub_(new CounterHostStub()), |
| + audio_stub_(new CounterAudioStub()), |
| + video_stub_(new CounterVideoStub(connection)) {} |
| + |
| +FakeConnectionEventLogger::~FakeConnectionEventLogger() {} |
| + |
| +protocol::ClientStub* FakeConnectionEventLogger::client_stub() { |
| + return client_stub_.get(); |
| +} |
| + |
| +protocol::HostStub* FakeConnectionEventLogger::host_stub() { |
| + return host_stub_.get(); |
| +} |
| + |
| +protocol::AudioStub* FakeConnectionEventLogger::audio_stub() { |
| + return audio_stub_.get(); |
| +} |
| + |
| +protocol::VideoStub* FakeConnectionEventLogger::video_stub() { |
| + return video_stub_.get(); |
| +} |
| + |
| +std::ostream& operator<<(std::ostream& os, |
| + const FakeConnectionEventLogger& logger) { |
| + logger.audio_stub_->DisplayStatistics(os, "audio"); |
| + logger.video_stub_->DisplayStatistics(os, "video"); |
| + logger.client_stub_->DisplayStatistics(os, "client"); |
| + logger.host_stub_->DisplayStatistics(os, "host"); |
| + return os; |
| +} |
| + |
| +} // namespace test |
| +} // namespace remoting |