Chromoting protocol layers to receive and send login messages

remoting/protocol/client_message_dispatcher.cc

+// Copyright (c) 2010 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 "base/ref_counted.h"
+#include "net/base/io_buffer.h"
+#include "remoting/proto/control.pb.h"
+#include "remoting/proto/event.pb.h"
+#include "remoting/proto/internal.pb.h"
+#include "remoting/protocol/client_message_dispatcher.h"
+#include "remoting/protocol/client_stub.h"
+#include "remoting/protocol/input_stub.h"
+#include "remoting/protocol/message_reader.h"
+#include "remoting/protocol/session.h"
+
+namespace {
+
+// A single protobuf can contain multiple messages that will be handled by
+// different message handlers.  We use this wrapper to ensure that the
+// protobuf is only deleted after all the handlers have finished executing.
+template <typename T>
+class RefCountedMessage : public base::RefCounted<RefCountedMessage<T> > {

[awong, 2010/12/23 01:15:10]

Is this cut/paste code?  Can we factor it out?

[Alpha Left Google, 2010/12/23 02:17:58]

Done.

+ public:
+  RefCountedMessage(T* message) : message_(message) { }
+
+  T* message() { return message_.get(); }
+
+ private:
+  scoped_ptr<T> message_;
+};
+
+// Dummy methods to destroy messages.
+template <class T>
+static void DeleteMessage(scoped_refptr<T> message) { }
+
+template <class T>
+static Task* NewDeleteTask(scoped_refptr<T> message) {
+  return NewRunnableFunction(&DeleteMessage<T>, message);
+}
+
+}  // namespace
+
+namespace remoting {
+namespace protocol {
+
+ClientMessageDispatcher::ClientMessageDispatcher() : client_stub_(NULL) {
+}
+
+ClientMessageDispatcher::~ClientMessageDispatcher() {
+}
+
+bool ClientMessageDispatcher::Initialize(
+    protocol::Session* session, ClientStub* client_stub) {
+  if (!session || !client_stub || !session->control_channel()) {
+    return false;
+  }
+
+  control_message_reader_.reset(new MessageReader());
+  client_stub_ = client_stub;
+
+  control_message_reader_->Init<ControlMessage>(
+      session->control_channel(),
+      NewCallback(this, &ClientMessageDispatcher::OnControlMessageReceived));
+  return true;
+}
+
+void ClientMessageDispatcher::OnControlMessageReceived(
+    ControlMessage* message) {
+  scoped_refptr<RefCountedMessage<ControlMessage> > ref_msg =
+      new RefCountedMessage<ControlMessage>(message);
+  if (message->has_notify_resolution()) {
+    client_stub_->NotifyResolution(
+        &message->notify_resolution(), NewDeleteTask(ref_msg));
+  } else if (message->has_begin_session_response()) {
+    client_stub_->BeginSessionResponse(
+        &message->begin_session_response().login_status(),
+        NewDeleteTask(ref_msg));
+  } else {
+    NOTREACHED() << "Invalid control message received";
+  }
+}
+
+}  // namespace protocol
+}  // namespace remoting