Adding GCM support to the copresence component.

components/copresence/handlers/gcm_handler.cc

Index: components/copresence/handlers/gcm_handler.cc
diff --git a/components/copresence/handlers/gcm_handler.cc b/components/copresence/handlers/gcm_handler.cc
new file mode 100644
index 0000000000000000000000000000000000000000..bacc02c2f933b674a74cb38242855ad5743e7a21
--- /dev/null
+++ b/components/copresence/handlers/gcm_handler.cc
@@ -0,0 +1,157 @@
+// Copyright 2014 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 "components/copresence/handlers/gcm_handler.h"
+
+#include "base/base64.h"
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/strings/string_util.h"
+#include "components/copresence/handlers/directive_handler.h"
+#include "components/copresence/proto/push_message.pb.h"
+#include "components/gcm_driver/gcm_driver.h"
+
+using gcm::GCMClient;
+
+namespace {
+
+// TODO(ckehoe): Move this to a common library.
+bool Base64Decode(std::string data, std::string* out) {
+  // Convert from URL-safe.
+  base::ReplaceChars(data, "-", "+", &data);
+  base::ReplaceChars(data, "_", "/", &data);
+
+  // Add padding if needed.
+  while (data.size() % 4)
+    data.push_back('=');
+
+  // Decode.
+  return base::Base64Decode(data, out);
+}
+
+}  // namespace
+
+namespace copresence {
+
+const char GCMHandler::kCopresenceAppId[] =
+    "com.google.android.gms.location.copresence";
+const char GCMHandler::kCopresenceSenderId[] = "745476177629";
+const char GCMHandler::kGcmMessageKey[] = "PUSH_MESSAGE";
+
+
+// Public functions.
+
+GCMHandler::GCMHandler(gcm::GCMDriver* gcm_driver,
+                       DirectiveHandler* directive_handler)
+    : driver_(gcm_driver),
+      directive_handler_(directive_handler),
+      registration_callback_(base::Bind(&GCMHandler::RegistrationComplete,
+                                        base::Unretained(this))) {
+  DCHECK(driver_);
+  DCHECK(directive_handler_);
+
+  driver_->AddAppHandler(kCopresenceAppId, this);
+  driver_->Register(kCopresenceAppId,
+                    std::vector<std::string>(1, kCopresenceSenderId),
+                    // We cancel this callback on destruction.
+                    registration_callback_.callback());
+}
+
+GCMHandler::~GCMHandler() {
+  if (driver_)
+    driver_->RemoveAppHandler(kCopresenceAppId);
+  registration_callback_.Cancel();

[rkc, 2014/11/07 20:24:15]

Looking at the cancelable callback code, I *think* that you don't need to do
this. The weak ptr in the object clears itself on destruction. You should ping
jhawkins@ or groby@ to confirm.

(If this is true, we probably should add this info to the cancelable callback
documentation - not in this CL though, that'll require a //base owners review).

[Charlie, 2014/11/07 20:50:12]

Done.

+}
+
+void GCMHandler::GetGcmId(const RegistrationCallback& callback) {
+  if (gcm_id_.empty()) {
+    pending_id_requests_.push_back(callback);
+  } else {
+    callback.Run(gcm_id_);
+  }
+}
+
+void GCMHandler::ShutdownHandler() {
+  // The GCMDriver is going away. Make sure we don't try to contact it.
+  driver_ = nullptr;
+}
+
+void GCMHandler::OnMessage(const std::string& app_id,
+                           const GCMClient::IncomingMessage& message) {
+  DCHECK_EQ(kCopresenceAppId, app_id);
+  DVLOG(2) << "Incoming GCM message";
+
+  const auto& content = message.data.find(kGcmMessageKey);
+  if (content == message.data.end()) {
+    LOG(ERROR) << "GCM message missing data key";
+    return;
+  }
+
+  std::string serialized_message;
+  if (!Base64Decode(content->second, &serialized_message)) {
+    LOG(ERROR) << "Couldn't decode GCM message";
+    return;
+  }
+
+  PushMessage push_message;
+  if (!push_message.ParseFromString(serialized_message)) {
+    LOG(ERROR) << "GCM message contained invalid proto";
+    return;
+  }
+
+  if (push_message.type() != PushMessage::REPORT) {
+    DVLOG(2) << "Discarding non-report GCM message";
+    return;
+  }
+
+  DVLOG(3) << "Processing " << push_message.report().directive_size()
+           << " directive(s) from GCM message";
+  for (const Directive& directive : push_message.report().directive())
+    directive_handler_->AddDirective(directive);
+
+  int message_count = push_message.report().subscribed_message_size();
+  LOG_IF(WARNING, message_count > 0)
+      << "Discarding " << message_count << " copresence messages sent via GCM";
+}
+
+void GCMHandler::OnMessagesDeleted(const std::string& app_id) {
+  DCHECK_EQ(kCopresenceAppId, app_id);

[rkc, 2014/11/07 20:24:15]

app_id will become an unused argument on release builds.

[Charlie, 2014/11/07 20:50:12]

I think this is ok, since it's an argument as opposed to a local variable. We
can let the bots decide.

+  DVLOG(2) << "GCM message overflow reported";
+}
+
+void GCMHandler::OnSendError(
+    const std::string& app_id,
+    const GCMClient::SendErrorDetails& send_error_details) {

[rkc, 2014/11/07 20:24:15]

/* app_id */
/* send_error_details */
here and all the functions below where the arguments aren't being used.

[Charlie, 2014/11/07 20:50:11]

Done.

+  NOTREACHED() << "Copresence clients should not be sending GCM messages";
+}
+
+void GCMHandler::OnSendAcknowledged(const std::string& app_id,
+                                    const std::string& message_id) {
+  NOTREACHED() << "Copresence clients should not be sending GCM messages";
+}
+
+bool GCMHandler::CanHandle(const std::string& app_id) const {
+  return app_id == kCopresenceAppId;
+}
+
+
+// Private functions.
+
+void GCMHandler::RegistrationComplete(const std::string& registration_id,
+                                      GCMClient::Result result) {
+  if (result == GCMClient::SUCCESS) {
+    DVLOG(2) << "GCM registration successful. ID: " << registration_id;
+    gcm_id_ = registration_id;
+  } else {
+    LOG(ERROR) << "GCM registration failed with error " << result;
+  }
+
+  for (const RegistrationCallback& callback : pending_id_requests_) {
+    callback.Run(result == GCMClient::SUCCESS ?
+        registration_id : std::string());
+  }
+  pending_id_requests_.clear();
+}
+
+}  // namespace copresence