| Index: sync/notifier/p2p_invalidator.cc
|
| diff --git a/sync/notifier/p2p_invalidator.cc b/sync/notifier/p2p_invalidator.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f84f34ecab627695b7c251907801ad51fd892575
|
| --- /dev/null
|
| +++ b/sync/notifier/p2p_invalidator.cc
|
| @@ -0,0 +1,299 @@
|
| +// Copyright (c) 2012 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 "sync/notifier/p2p_invalidator.h"
|
| +
|
| +#include <algorithm>
|
| +#include <iterator>
|
| +
|
| +#include "base/json/json_reader.h"
|
| +#include "base/json/json_writer.h"
|
| +#include "base/logging.h"
|
| +#include "base/values.h"
|
| +#include "jingle/notifier/listener/push_client.h"
|
| +#include "sync/notifier/invalidation_handler.h"
|
| +#include "sync/notifier/invalidation_util.h"
|
| +#include "sync/notifier/object_id_invalidation_map.h"
|
| +
|
| +namespace syncer {
|
| +
|
| +const char kSyncP2PNotificationChannel[] = "http://www.google.com/chrome/sync";
|
| +
|
| +namespace {
|
| +
|
| +const char kNotifySelf[] = "notifySelf";
|
| +const char kNotifyOthers[] = "notifyOthers";
|
| +const char kNotifyAll[] = "notifyAll";
|
| +
|
| +const char kSenderIdKey[] = "senderId";
|
| +const char kNotificationTypeKey[] = "notificationType";
|
| +const char kInvalidationsKey[] = "invalidations";
|
| +
|
| +} // namespace
|
| +
|
| +std::string P2PNotificationTargetToString(P2PNotificationTarget target) {
|
| + switch (target) {
|
| + case NOTIFY_SELF:
|
| + return kNotifySelf;
|
| + case NOTIFY_OTHERS:
|
| + return kNotifyOthers;
|
| + case NOTIFY_ALL:
|
| + return kNotifyAll;
|
| + default:
|
| + NOTREACHED();
|
| + return std::string();
|
| + }
|
| +}
|
| +
|
| +P2PNotificationTarget P2PNotificationTargetFromString(
|
| + const std::string& target_str) {
|
| + if (target_str == kNotifySelf) {
|
| + return NOTIFY_SELF;
|
| + }
|
| + if (target_str == kNotifyOthers) {
|
| + return NOTIFY_OTHERS;
|
| + }
|
| + if (target_str == kNotifyAll) {
|
| + return NOTIFY_ALL;
|
| + }
|
| + LOG(WARNING) << "Could not parse " << target_str;
|
| + return NOTIFY_SELF;
|
| +}
|
| +
|
| +P2PNotificationData::P2PNotificationData()
|
| + : target_(NOTIFY_SELF) {}
|
| +
|
| +P2PNotificationData::P2PNotificationData(
|
| + const std::string& sender_id,
|
| + P2PNotificationTarget target,
|
| + const ObjectIdInvalidationMap& invalidation_map)
|
| + : sender_id_(sender_id),
|
| + target_(target),
|
| + invalidation_map_(invalidation_map) {}
|
| +
|
| +P2PNotificationData::~P2PNotificationData() {}
|
| +
|
| +bool P2PNotificationData::IsTargeted(const std::string& id) const {
|
| + switch (target_) {
|
| + case NOTIFY_SELF:
|
| + return sender_id_ == id;
|
| + case NOTIFY_OTHERS:
|
| + return sender_id_ != id;
|
| + case NOTIFY_ALL:
|
| + return true;
|
| + default:
|
| + NOTREACHED();
|
| + return false;
|
| + }
|
| +}
|
| +
|
| +const ObjectIdInvalidationMap&
|
| +P2PNotificationData::GetIdInvalidationMap() const {
|
| + return invalidation_map_;
|
| +}
|
| +
|
| +bool P2PNotificationData::Equals(const P2PNotificationData& other) const {
|
| + return
|
| + (sender_id_ == other.sender_id_) &&
|
| + (target_ == other.target_) &&
|
| + (invalidation_map_ == other.invalidation_map_);
|
| +}
|
| +
|
| +std::string P2PNotificationData::ToString() const {
|
| + scoped_ptr<base::DictionaryValue> dict(new base::DictionaryValue());
|
| + dict->SetString(kSenderIdKey, sender_id_);
|
| + dict->SetString(kNotificationTypeKey,
|
| + P2PNotificationTargetToString(target_));
|
| + dict->Set(kInvalidationsKey, invalidation_map_.ToValue().release());
|
| + std::string json;
|
| + base::JSONWriter::Write(dict.get(), &json);
|
| + return json;
|
| +}
|
| +
|
| +bool P2PNotificationData::ResetFromString(const std::string& str) {
|
| + scoped_ptr<base::Value> data_value(base::JSONReader::Read(str));
|
| + const base::DictionaryValue* data_dict = NULL;
|
| + if (!data_value.get() || !data_value->GetAsDictionary(&data_dict)) {
|
| + LOG(WARNING) << "Could not parse " << str << " as a dictionary";
|
| + return false;
|
| + }
|
| + if (!data_dict->GetString(kSenderIdKey, &sender_id_)) {
|
| + LOG(WARNING) << "Could not find string value for " << kSenderIdKey;
|
| + }
|
| + std::string target_str;
|
| + if (!data_dict->GetString(kNotificationTypeKey, &target_str)) {
|
| + LOG(WARNING) << "Could not find string value for "
|
| + << kNotificationTypeKey;
|
| + }
|
| + target_ = P2PNotificationTargetFromString(target_str);
|
| + const base::ListValue* invalidation_map_list = NULL;
|
| + if (!data_dict->GetList(kInvalidationsKey, &invalidation_map_list) ||
|
| + !invalidation_map_.ResetFromValue(*invalidation_map_list)) {
|
| + LOG(WARNING) << "Could not parse " << kInvalidationsKey;
|
| + }
|
| + return true;
|
| +}
|
| +
|
| +P2PInvalidator::P2PInvalidator(scoped_ptr<notifier::PushClient> push_client,
|
| + const std::string& invalidator_client_id,
|
| + P2PNotificationTarget send_notification_target)
|
| + : push_client_(push_client.Pass()),
|
| + invalidator_client_id_(invalidator_client_id),
|
| + logged_in_(false),
|
| + notifications_enabled_(false),
|
| + send_notification_target_(send_notification_target) {
|
| + DCHECK(send_notification_target_ == NOTIFY_OTHERS ||
|
| + send_notification_target_ == NOTIFY_ALL);
|
| + push_client_->AddObserver(this);
|
| +}
|
| +
|
| +P2PInvalidator::~P2PInvalidator() {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + push_client_->RemoveObserver(this);
|
| +}
|
| +
|
| +void P2PInvalidator::RegisterHandler(InvalidationHandler* handler) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + registrar_.RegisterHandler(handler);
|
| +}
|
| +
|
| +void P2PInvalidator::UpdateRegisteredIds(InvalidationHandler* handler,
|
| + const ObjectIdSet& ids) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + ObjectIdSet new_ids;
|
| + const ObjectIdSet& old_ids = registrar_.GetRegisteredIds(handler);
|
| + std::set_difference(ids.begin(), ids.end(),
|
| + old_ids.begin(), old_ids.end(),
|
| + std::inserter(new_ids, new_ids.end()),
|
| + ObjectIdLessThan());
|
| + registrar_.UpdateRegisteredIds(handler, ids);
|
| + const P2PNotificationData notification_data(
|
| + invalidator_client_id_,
|
| + send_notification_target_,
|
| + ObjectIdInvalidationMap::InvalidateAll(ids));
|
| + SendNotificationData(notification_data);
|
| +}
|
| +
|
| +void P2PInvalidator::UnregisterHandler(InvalidationHandler* handler) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + registrar_.UnregisterHandler(handler);
|
| +}
|
| +
|
| +InvalidatorState P2PInvalidator::GetInvalidatorState() const {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + return registrar_.GetInvalidatorState();
|
| +}
|
| +
|
| +void P2PInvalidator::UpdateCredentials(
|
| + const std::string& email, const std::string& token) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + notifier::Subscription subscription;
|
| + subscription.channel = kSyncP2PNotificationChannel;
|
| + // There may be some subtle issues around case sensitivity of the
|
| + // from field, but it doesn't matter too much since this is only
|
| + // used in p2p mode (which is only used in testing).
|
| + subscription.from = email;
|
| + push_client_->UpdateSubscriptions(
|
| + notifier::SubscriptionList(1, subscription));
|
| + // If already logged in, the new credentials will take effect on the
|
| + // next reconnection.
|
| + push_client_->UpdateCredentials(email, token);
|
| + logged_in_ = true;
|
| +}
|
| +
|
| +void P2PInvalidator::RequestDetailedStatus(
|
| + base::Callback<void(const base::DictionaryValue&)> callback) const {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + // TODO(mferreria): Make the P2P Invalidator work.
|
| + scoped_ptr<base::DictionaryValue> value(new base::DictionaryValue());
|
| + callback.Run(*value);
|
| +}
|
| +
|
| +void P2PInvalidator::SendInvalidation(const ObjectIdSet& ids) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + ObjectIdInvalidationMap invalidation_map =
|
| + ObjectIdInvalidationMap::InvalidateAll(ids);
|
| + const P2PNotificationData notification_data(
|
| + invalidator_client_id_, send_notification_target_, invalidation_map);
|
| + SendNotificationData(notification_data);
|
| +}
|
| +
|
| +void P2PInvalidator::OnNotificationsEnabled() {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + bool just_turned_on = (notifications_enabled_ == false);
|
| + notifications_enabled_ = true;
|
| + registrar_.UpdateInvalidatorState(INVALIDATIONS_ENABLED);
|
| + if (just_turned_on) {
|
| + const P2PNotificationData notification_data(
|
| + invalidator_client_id_,
|
| + NOTIFY_SELF,
|
| + ObjectIdInvalidationMap::InvalidateAll(
|
| + registrar_.GetAllRegisteredIds()));
|
| + SendNotificationData(notification_data);
|
| + }
|
| +}
|
| +
|
| +void P2PInvalidator::OnNotificationsDisabled(
|
| + notifier::NotificationsDisabledReason reason) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + registrar_.UpdateInvalidatorState(FromNotifierReason(reason));
|
| +}
|
| +
|
| +void P2PInvalidator::OnIncomingNotification(
|
| + const notifier::Notification& notification) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + DVLOG(1) << "Received notification " << notification.ToString();
|
| + if (!logged_in_) {
|
| + DVLOG(1) << "Not logged in yet -- not emitting notification";
|
| + return;
|
| + }
|
| + if (!notifications_enabled_) {
|
| + DVLOG(1) << "Notifications not on -- not emitting notification";
|
| + return;
|
| + }
|
| + if (notification.channel != kSyncP2PNotificationChannel) {
|
| + LOG(WARNING) << "Notification from unexpected source "
|
| + << notification.channel;
|
| + }
|
| + P2PNotificationData notification_data;
|
| + if (!notification_data.ResetFromString(notification.data)) {
|
| + LOG(WARNING) << "Could not parse notification data from "
|
| + << notification.data;
|
| + notification_data = P2PNotificationData(
|
| + invalidator_client_id_,
|
| + NOTIFY_ALL,
|
| + ObjectIdInvalidationMap::InvalidateAll(
|
| + registrar_.GetAllRegisteredIds()));
|
| + }
|
| + if (!notification_data.IsTargeted(invalidator_client_id_)) {
|
| + DVLOG(1) << "Not a target of the notification -- "
|
| + << "not emitting notification";
|
| + return;
|
| + }
|
| + registrar_.DispatchInvalidationsToHandlers(
|
| + notification_data.GetIdInvalidationMap());
|
| +}
|
| +
|
| +void P2PInvalidator::SendNotificationDataForTest(
|
| + const P2PNotificationData& notification_data) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + SendNotificationData(notification_data);
|
| +}
|
| +
|
| +void P2PInvalidator::SendNotificationData(
|
| + const P2PNotificationData& notification_data) {
|
| + DCHECK(thread_checker_.CalledOnValidThread());
|
| + if (notification_data.GetIdInvalidationMap().Empty()) {
|
| + DVLOG(1) << "Not sending XMPP notification with empty state map: "
|
| + << notification_data.ToString();
|
| + return;
|
| + }
|
| + notifier::Notification notification;
|
| + notification.channel = kSyncP2PNotificationChannel;
|
| + notification.data = notification_data.ToString();
|
| + DVLOG(1) << "Sending XMPP notification: " << notification.ToString();
|
| + push_client_->SendNotification(notification);
|
| +}
|
| +
|
| +} // namespace syncer
|
|
|