| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "components/copresence/handlers/gcm_handler.h" | |
| 6 | |
| 7 #include "base/base64.h" | |
| 8 #include "base/bind.h" | |
| 9 #include "base/logging.h" | |
| 10 #include "base/strings/string_util.h" | |
| 11 #include "components/copresence/handlers/directive_handler.h" | |
| 12 #include "components/copresence/proto/push_message.pb.h" | |
| 13 #include "components/gcm_driver/gcm_driver.h" | |
| 14 | |
| 15 using gcm::GCMClient; | |
| 16 | |
| 17 namespace { | |
| 18 | |
| 19 // TODO(ckehoe): Move this to a common library. | |
| 20 bool Base64Decode(std::string data, std::string* out) { | |
| 21 // Convert from URL-safe. | |
| 22 base::ReplaceChars(data, "-", "+", &data); | |
| 23 base::ReplaceChars(data, "_", "/", &data); | |
| 24 | |
| 25 // Add padding if needed. | |
| 26 while (data.size() % 4) | |
| 27 data.push_back('='); | |
| 28 | |
| 29 // Decode. | |
| 30 return base::Base64Decode(data, out); | |
| 31 } | |
| 32 | |
| 33 } // namespace | |
| 34 | |
| 35 namespace copresence { | |
| 36 | |
| 37 const char GCMHandler::kCopresenceAppId[] = | |
| 38 "com.google.android.gms.location.copresence"; | |
| 39 const char GCMHandler::kCopresenceSenderId[] = "745476177629"; | |
| 40 const char GCMHandler::kGcmMessageKey[] = "PUSH_MESSAGE"; | |
| 41 | |
| 42 | |
| 43 // Public functions. | |
| 44 | |
| 45 GCMHandler::GCMHandler(gcm::GCMDriver* gcm_driver, | |
| 46 DirectiveHandler* directive_handler) | |
| 47 : driver_(gcm_driver), | |
| 48 directive_handler_(directive_handler), | |
| 49 registration_callback_(base::Bind(&GCMHandler::RegistrationComplete, | |
| 50 AsWeakPtr())) { | |
| 51 DCHECK(driver_); | |
| 52 DCHECK(directive_handler_); | |
| 53 | |
| 54 driver_->AddAppHandler(kCopresenceAppId, this); | |
| 55 driver_->Register(kCopresenceAppId, | |
| 56 std::vector<std::string>(1, kCopresenceSenderId), | |
| 57 registration_callback_); | |
| 58 } | |
| 59 | |
| 60 GCMHandler::~GCMHandler() { | |
| 61 if (driver_) | |
| 62 driver_->RemoveAppHandler(kCopresenceAppId); | |
| 63 } | |
| 64 | |
| 65 void GCMHandler::GetGcmId(const RegistrationCallback& callback) { | |
| 66 if (gcm_id_.empty()) { | |
| 67 pending_id_requests_.push_back(callback); | |
| 68 } else { | |
| 69 callback.Run(gcm_id_); | |
| 70 } | |
| 71 } | |
| 72 | |
| 73 void GCMHandler::ShutdownHandler() { | |
| 74 // The GCMDriver is going away. Make sure we don't try to contact it. | |
| 75 driver_ = nullptr; | |
| 76 } | |
| 77 | |
| 78 void GCMHandler::OnMessage(const std::string& app_id, | |
| 79 const GCMClient::IncomingMessage& message) { | |
| 80 DCHECK_EQ(kCopresenceAppId, app_id); | |
| 81 DVLOG(2) << "Incoming GCM message"; | |
| 82 | |
| 83 const auto& content = message.data.find(kGcmMessageKey); | |
| 84 if (content == message.data.end()) { | |
| 85 LOG(ERROR) << "GCM message missing data key"; | |
| 86 return; | |
| 87 } | |
| 88 | |
| 89 std::string serialized_message; | |
| 90 if (!Base64Decode(content->second, &serialized_message)) { | |
| 91 LOG(ERROR) << "Couldn't decode GCM message"; | |
| 92 return; | |
| 93 } | |
| 94 | |
| 95 PushMessage push_message; | |
| 96 if (!push_message.ParseFromString(serialized_message)) { | |
| 97 LOG(ERROR) << "GCM message contained invalid proto"; | |
| 98 return; | |
| 99 } | |
| 100 | |
| 101 if (push_message.type() != PushMessage::REPORT) { | |
| 102 DVLOG(2) << "Discarding non-report GCM message"; | |
| 103 return; | |
| 104 } | |
| 105 | |
| 106 DVLOG(3) << "Processing " << push_message.report().directive_size() | |
| 107 << " directive(s) from GCM message"; | |
| 108 for (const Directive& directive : push_message.report().directive()) | |
| 109 directive_handler_->AddDirective(directive); | |
| 110 | |
| 111 int message_count = push_message.report().subscribed_message_size(); | |
| 112 LOG_IF(WARNING, message_count > 0) | |
| 113 << "Discarding " << message_count << " copresence messages sent via GCM"; | |
| 114 } | |
| 115 | |
| 116 void GCMHandler::OnMessagesDeleted(const std::string& app_id) { | |
| 117 DCHECK_EQ(kCopresenceAppId, app_id); | |
| 118 DVLOG(2) << "GCM message overflow reported"; | |
| 119 } | |
| 120 | |
| 121 void GCMHandler::OnSendError( | |
| 122 const std::string& /* app_id */, | |
| 123 const GCMClient::SendErrorDetails& /* send_error_details */) { | |
| 124 NOTREACHED() << "Copresence clients should not be sending GCM messages"; | |
| 125 } | |
| 126 | |
| 127 void GCMHandler::OnSendAcknowledged(const std::string& /* app_id */, | |
| 128 const std::string& /* message_id */) { | |
| 129 NOTREACHED() << "Copresence clients should not be sending GCM messages"; | |
| 130 } | |
| 131 | |
| 132 bool GCMHandler::CanHandle(const std::string& app_id) const { | |
| 133 return app_id == kCopresenceAppId; | |
| 134 } | |
| 135 | |
| 136 | |
| 137 // Private functions. | |
| 138 | |
| 139 void GCMHandler::RegistrationComplete(const std::string& registration_id, | |
| 140 GCMClient::Result result) { | |
| 141 if (result == GCMClient::SUCCESS) { | |
| 142 DVLOG(2) << "GCM registration successful. ID: " << registration_id; | |
| 143 gcm_id_ = registration_id; | |
| 144 } else { | |
| 145 LOG(ERROR) << "GCM registration failed with error " << result; | |
| 146 } | |
| 147 | |
| 148 for (const RegistrationCallback& callback : pending_id_requests_) { | |
| 149 callback.Run(result == GCMClient::SUCCESS ? | |
| 150 registration_id : std::string()); | |
| 151 } | |
| 152 pending_id_requests_.clear(); | |
| 153 } | |
| 154 | |
| 155 } // namespace copresence | |
| OLD | NEW |