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_impl.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 GCMHandlerImpl::kCopresenceAppId[] = | |
38 "com.google.android.gms.location.copresence"; | |
39 const char GCMHandlerImpl::kCopresenceSenderId[] = "745476177629"; | |
40 const char GCMHandlerImpl::kGcmMessageKey[] = "PUSH_MESSAGE"; | |
41 | |
42 | |
43 // Public functions. | |
44 | |
45 GCMHandlerImpl::GCMHandlerImpl(gcm::GCMDriver* gcm_driver, | |
46 DirectiveHandler* directive_handler, | |
47 const MessagesCallback& new_messages_callback) | |
48 : driver_(gcm_driver), | |
49 directive_handler_(directive_handler), | |
50 new_messages_callback_(new_messages_callback), | |
51 registration_callback_(base::Bind(&GCMHandlerImpl::RegistrationComplete, | |
52 base::Unretained(this))) { | |
53 DCHECK(driver_); | |
54 DCHECK(directive_handler_); | |
55 | |
56 driver_->AddAppHandler(kCopresenceAppId, this); | |
57 driver_->Register(kCopresenceAppId, | |
58 std::vector<std::string>(1, kCopresenceSenderId), | |
59 registration_callback_.callback()); | |
60 } | |
61 | |
62 GCMHandlerImpl::~GCMHandlerImpl() { | |
63 if (driver_) | |
64 driver_->RemoveAppHandler(kCopresenceAppId); | |
65 } | |
66 | |
67 void GCMHandlerImpl::GetGcmId(const RegistrationCallback& callback) { | |
68 if (gcm_id_.empty()) { | |
69 pending_id_requests_.push_back(callback); | |
70 } else { | |
71 callback.Run(gcm_id_); | |
72 } | |
73 } | |
74 | |
75 void GCMHandlerImpl::ShutdownHandler() { | |
76 // The GCMDriver is going away. Make sure we don't try to contact it. | |
77 driver_ = nullptr; | |
78 } | |
79 | |
80 void GCMHandlerImpl::OnMessage(const std::string& app_id, | |
81 const gcm::IncomingMessage& message) { | |
82 DCHECK_EQ(kCopresenceAppId, app_id); | |
83 DVLOG(2) << "Incoming GCM message"; | |
84 | |
85 const auto& content = message.data.find(kGcmMessageKey); | |
86 if (content == message.data.end()) { | |
87 LOG(ERROR) << "GCM message missing data key"; | |
88 return; | |
89 } | |
90 | |
91 std::string serialized_message; | |
92 if (!Base64Decode(content->second, &serialized_message)) { | |
93 LOG(ERROR) << "Couldn't decode GCM message"; | |
94 return; | |
95 } | |
96 | |
97 PushMessage push_message; | |
98 if (!push_message.ParseFromString(serialized_message)) { | |
99 LOG(ERROR) << "GCM message contained invalid proto"; | |
100 return; | |
101 } | |
102 | |
103 if (push_message.type() != PushMessage::REPORT) { | |
104 DVLOG(2) << "Discarding non-report GCM message"; | |
105 return; | |
106 } | |
107 | |
108 DVLOG(3) << "Processing " << push_message.report().directive_size() | |
109 << " directive(s) from GCM message"; | |
110 for (const Directive& directive : push_message.report().directive()) { | |
111 // TODO(ckehoe): Use a callback here so GCMHandler | |
112 // is decoupled from DirectiveHandler. | |
113 directive_handler_->AddDirective(directive); | |
114 } | |
115 | |
116 new_messages_callback_.Run(push_message.report().subscribed_message()); | |
117 } | |
118 | |
119 void GCMHandlerImpl::OnMessagesDeleted(const std::string& app_id) { | |
120 DCHECK_EQ(kCopresenceAppId, app_id); | |
121 DVLOG(2) << "GCM message overflow reported"; | |
122 } | |
123 | |
124 void GCMHandlerImpl::OnSendError( | |
125 const std::string& /* app_id */, | |
126 const GCMClient::SendErrorDetails& /* send_error_details */) { | |
127 NOTREACHED() << "Copresence clients should not be sending GCM messages"; | |
128 } | |
129 | |
130 void GCMHandlerImpl::OnSendAcknowledged(const std::string& /* app_id */, | |
131 const std::string& /* message_id */) { | |
132 NOTREACHED() << "Copresence clients should not be sending GCM messages"; | |
133 } | |
134 | |
135 bool GCMHandlerImpl::CanHandle(const std::string& app_id) const { | |
136 return app_id == kCopresenceAppId; | |
137 } | |
138 | |
139 | |
140 // Private functions. | |
141 | |
142 void GCMHandlerImpl::RegistrationComplete(const std::string& registration_id, | |
143 GCMClient::Result result) { | |
144 if (result == GCMClient::SUCCESS) { | |
145 DVLOG(2) << "GCM registration successful. ID: " << registration_id; | |
146 gcm_id_ = registration_id; | |
147 } else { | |
148 LOG(ERROR) << "GCM registration failed with error " << result; | |
149 } | |
150 | |
151 for (const RegistrationCallback& callback : pending_id_requests_) { | |
152 callback.Run(result == GCMClient::SUCCESS ? | |
153 registration_id : std::string()); | |
154 } | |
155 pending_id_requests_.clear(); | |
156 } | |
157 | |
158 } // namespace copresence | |
OLD | NEW |