Implement a dummy host to do capturing and analysis only.

remoting/test/fake_connection_event_logger.cc

+// 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 <string>
+
+#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:
+  explicit MessageCounter(const char* name);
+
+  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);
+
+ private:
+  std::string name_;
+  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(const char* name)
+    : name_(name),
+      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) {
+  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:
+  CounterClientStub();
+
+ private:
+  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 {}
+};
+
+FakeConnectionEventLogger::CounterClientStub::CounterClientStub()
+    : MessageCounter("client") {}
+
+void FakeConnectionEventLogger::CounterClientStub::DeliverHostMessage(
+    const protocol::ExtensionMessage& message) {
+  LogMessage(message);
+}
+
+// Analyzes messages from DeliverClientMessage function.
+class FakeConnectionEventLogger::CounterHostStub
+    : public protocol::HostStub, public MessageCounter {
+ public:
+  CounterHostStub();
+
+ private:
+  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 {}
+};
+
+FakeConnectionEventLogger::CounterHostStub::CounterHostStub()
+    : MessageCounter("host") {}
+
+void FakeConnectionEventLogger::CounterHostStub::DeliverClientMessage(
+    const protocol::ExtensionMessage& message) {
+  LogMessage(message);
+}
+
+// Analyzes messages from ProcessAudioPacket function.
+class FakeConnectionEventLogger::CounterAudioStub
+    : public protocol::AudioStub, public MessageCounter {
+ public:
+  CounterAudioStub();
+
+ private:
+  void ProcessAudioPacket(std::unique_ptr<AudioPacket> audio_packet,
+                          const base::Closure& done) override;
+};
+
+FakeConnectionEventLogger::CounterAudioStub::CounterAudioStub()
+    : MessageCounter("audio") {}
+
+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);
+
+ private:
+  void ProcessVideoPacket(std::unique_ptr<VideoPacket> video_packet,
+                          const base::Closure& done) override;
+
+  protocol::FakeConnectionToClient* connection_ = nullptr;
+};
+
+FakeConnectionEventLogger::CounterVideoStub::CounterVideoStub(
+    protocol::FakeConnectionToClient* connection)
+    : MessageCounter("video"),
+      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);
+  logger.video_stub_->DisplayStatistics(os);
+  logger.client_stub_->DisplayStatistics(os);
+  logger.host_stub_->DisplayStatistics(os);
+  return os;
+}
+
+}  // namespace test
+}  // namespace remoting