Chromoting protocol layers to receive and send login messages

remoting/protocol/ref_counted_message.h

+// 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.
+
+// This is a wrapper class to help ref-counting a protobuf message.
+// This file should only be inclued on host_message_dispatcher.cc and
+// client_message_dispatche.cc.
+
+// 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.
+
+#ifndef REMOTING_PROTOCOL_REF_COUNTED_MESSAGE_H_
+#define REMOTING_PROTOCOL_REF_COUNTED_MESSAGE_H_
+
+#include "base/ref_counted.h"
+#include "base/task.h"
+
+namespace remoting {
+namespace protocol {
+
+template <typename T>
+class RefCountedMessage : public base::RefCounted<RefCountedMessage<T> > {
+ 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 protocol
+}  // namespace remoting
+
+#endif  // REMOTING_PROTOCOL_REF_COUNTED_MESSAGE_H_