src/sync/tools/sync_client.cc

   1 // Copyright (c) 2012 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 <cstddef>
   6 #include <cstdio>
   7 #include <string>
   8
   9 #include "base/at_exit.h"
  10 #include "base/command_line.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/debug/stack_trace.h"
  13 #include "base/files/scoped_temp_dir.h"
  14 #include "base/json/json_writer.h"
  15 #include "base/logging.h"
  16 #include "base/memory/ref_counted.h"
  17 #include "base/memory/scoped_ptr.h"
  18 #include "base/memory/weak_ptr.h"
  19 #include "base/message_loop/message_loop.h"
  20 #include "base/rand_util.h"
  21 #include "base/task_runner.h"
  22 #include "base/threading/thread.h"
  23 #include "jingle/notifier/base/notification_method.h"
  24 #include "jingle/notifier/base/notifier_options.h"
  25 #include "net/base/host_port_pair.h"
  26 #include "net/base/network_change_notifier.h"
  27 #include "net/dns/host_resolver.h"
  28 #include "net/http/transport_security_state.h"
  29 #include "net/url_request/url_request_test_util.h"
  30 #include "sync/internal_api/public/base/cancelation_signal.h"
  31 #include "sync/internal_api/public/base/model_type.h"
  32 #include "sync/internal_api/public/base_node.h"
  33 #include "sync/internal_api/public/engine/passive_model_worker.h"
  34 #include "sync/internal_api/public/http_bridge.h"
  35 #include "sync/internal_api/public/internal_components_factory_impl.h"
  36 #include "sync/internal_api/public/read_node.h"
  37 #include "sync/internal_api/public/sync_manager.h"
  38 #include "sync/internal_api/public/sync_manager_factory.h"
  39 #include "sync/internal_api/public/util/report_unrecoverable_error_function.h"
  40 #include "sync/internal_api/public/util/unrecoverable_error_handler.h"
  41 #include "sync/internal_api/public/util/weak_handle.h"
  42 #include "sync/js/js_event_details.h"
  43 #include "sync/js/js_event_handler.h"
  44 #include "sync/notifier/invalidation_state_tracker.h"
  45 #include "sync/notifier/non_blocking_invalidator.h"
  46 #include "sync/test/fake_encryptor.h"
  47 #include "sync/tools/null_invalidation_state_tracker.h"
  48
  49 #if defined(OS_MACOSX)
  50 #include "base/mac/scoped_nsautorelease_pool.h"
  51 #endif
  52
  53 // This is a simple utility that initializes a sync client and
  54 // prints out any events.
  55
  56 // TODO(akalin): Refactor to combine shared code with
  57 // sync_listen_notifications.
  58 namespace syncer {
  59 namespace {
  60
  61 const char kEmailSwitch[] = "email";
  62 const char kTokenSwitch[] = "token";
  63 const char kXmppHostPortSwitch[] = "xmpp-host-port";
  64 const char kXmppTrySslTcpFirstSwitch[] = "xmpp-try-ssltcp-first";
  65 const char kXmppAllowInsecureConnectionSwitch[] =
  66     "xmpp-allow-insecure-connection";
  67
  68 // Needed to use a real host resolver.
  69 class MyTestURLRequestContext : public net::TestURLRequestContext {
  70  public:
  71   MyTestURLRequestContext() : TestURLRequestContext(true) {
  72     context_storage_.set_host_resolver(
  73         net::HostResolver::CreateDefaultResolver(NULL));
  74     context_storage_.set_transport_security_state(
  75         new net::TransportSecurityState());
  76     Init();
  77   }
  78
  79   virtual ~MyTestURLRequestContext() {}
  80 };
  81
  82 class MyTestURLRequestContextGetter : public net::TestURLRequestContextGetter {
  83  public:
  84   explicit MyTestURLRequestContextGetter(
  85       const scoped_refptr<base::MessageLoopProxy>& io_message_loop_proxy)
  86       : TestURLRequestContextGetter(io_message_loop_proxy) {}
  87
  88   virtual net::TestURLRequestContext* GetURLRequestContext() OVERRIDE {
  89     // Construct |context_| lazily so it gets constructed on the right
  90     // thread (the IO thread).
  91     if (!context_)
  92       context_.reset(new MyTestURLRequestContext());
  93     return context_.get();
  94   }
  95
  96  private:
  97   virtual ~MyTestURLRequestContextGetter() {}
  98
  99   scoped_ptr<MyTestURLRequestContext> context_;
 100 };
 101
 102 // TODO(akalin): Use system encryptor once it's moved to sync/.
 103 class NullEncryptor : public Encryptor {
 104  public:
 105   virtual ~NullEncryptor() {}
 106
 107   virtual bool EncryptString(const std::string& plaintext,
 108                              std::string* ciphertext) OVERRIDE {
 109     *ciphertext = plaintext;
 110     return true;
 111   }
 112
 113   virtual bool DecryptString(const std::string& ciphertext,
 114                              std::string* plaintext) OVERRIDE {
 115     *plaintext = ciphertext;
 116     return true;
 117   }
 118 };
 119
 120 std::string ValueToString(const base::Value& value) {
 121   std::string str;
 122   base::JSONWriter::Write(&value, &str);
 123   return str;
 124 }
 125
 126 class LoggingChangeDelegate : public SyncManager::ChangeDelegate {
 127  public:
 128   virtual ~LoggingChangeDelegate() {}
 129
 130   virtual void OnChangesApplied(
 131       ModelType model_type,
 132       int64 model_version,
 133       const BaseTransaction* trans,
 134       const ImmutableChangeRecordList& changes) OVERRIDE {
 135     LOG(INFO) << "Changes applied for "
 136               << ModelTypeToString(model_type);
 137     size_t i = 1;
 138     size_t change_count = changes.Get().size();
 139     for (ChangeRecordList::const_iterator it =
 140              changes.Get().begin(); it != changes.Get().end(); ++it) {
 141       scoped_ptr<base::DictionaryValue> change_value(it->ToValue());
 142       LOG(INFO) << "Change (" << i << "/" << change_count << "): "
 143                 << ValueToString(*change_value);
 144       if (it->action != ChangeRecord::ACTION_DELETE) {
 145         ReadNode node(trans);
 146         CHECK_EQ(node.InitByIdLookup(it->id), BaseNode::INIT_OK);
 147         scoped_ptr<base::DictionaryValue> details(node.ToValue());
 148         VLOG(1) << "Details: " << ValueToString(*details);
 149       }
 150       ++i;
 151     }
 152   }
 153
 154   virtual void OnChangesComplete(ModelType model_type) OVERRIDE {
 155     LOG(INFO) << "Changes complete for "
 156               << ModelTypeToString(model_type);
 157   }
 158 };
 159
 160 class LoggingUnrecoverableErrorHandler
 161     : public UnrecoverableErrorHandler {
 162  public:
 163   virtual ~LoggingUnrecoverableErrorHandler() {}
 164
 165   virtual void OnUnrecoverableError(const tracked_objects::Location& from_here,
 166                                     const std::string& message) OVERRIDE {
 167     if (LOG_IS_ON(ERROR)) {
 168       logging::LogMessage(from_here.file_name(), from_here.line_number(),
 169                           logging::LOG_ERROR).stream()
 170           << message;
 171     }
 172   }
 173 };
 174
 175 class LoggingJsEventHandler
 176     : public JsEventHandler,
 177       public base::SupportsWeakPtr<LoggingJsEventHandler> {
 178  public:
 179   virtual ~LoggingJsEventHandler() {}
 180
 181   virtual void HandleJsEvent(
 182       const std::string& name,
 183       const JsEventDetails& details) OVERRIDE {
 184     VLOG(1) << name << ": " << details.ToString();
 185   }
 186 };
 187
 188 void LogUnrecoverableErrorContext() {
 189   base::debug::StackTrace().Print();
 190 }
 191
 192 notifier::NotifierOptions ParseNotifierOptions(
 193     const CommandLine& command_line,
 194     const scoped_refptr<net::URLRequestContextGetter>&
 195         request_context_getter) {
 196   notifier::NotifierOptions notifier_options;
 197   notifier_options.request_context_getter = request_context_getter;
 198   notifier_options.auth_mechanism = "X-OAUTH2";
 199
 200   if (command_line.HasSwitch(kXmppHostPortSwitch)) {
 201     notifier_options.xmpp_host_port =
 202         net::HostPortPair::FromString(
 203             command_line.GetSwitchValueASCII(kXmppHostPortSwitch));
 204     LOG(INFO) << "Using " << notifier_options.xmpp_host_port.ToString()
 205               << " for test sync notification server.";
 206   }
 207
 208   notifier_options.try_ssltcp_first =
 209       command_line.HasSwitch(kXmppTrySslTcpFirstSwitch);
 210   LOG_IF(INFO, notifier_options.try_ssltcp_first)
 211       << "Trying SSL/TCP port before XMPP port for notifications.";
 212
 213   notifier_options.allow_insecure_connection =
 214       command_line.HasSwitch(kXmppAllowInsecureConnectionSwitch);
 215   LOG_IF(INFO, notifier_options.allow_insecure_connection)
 216       << "Allowing insecure XMPP connections.";
 217
 218   return notifier_options;
 219 }
 220
 221 void StubNetworkTimeUpdateCallback(const base::Time&,
 222                                    const base::TimeDelta&,
 223                                    const base::TimeDelta&) {
 224 }
 225
 226 int SyncClientMain(int argc, char* argv[]) {
 227 #if defined(OS_MACOSX)
 228   base::mac::ScopedNSAutoreleasePool pool;
 229 #endif
 230   base::AtExitManager exit_manager;
 231   CommandLine::Init(argc, argv);
 232   logging::LoggingSettings settings;
 233   settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
 234   logging::InitLogging(settings);
 235
 236   base::MessageLoop sync_loop;
 237   base::Thread io_thread("IO thread");
 238   base::Thread::Options options;
 239   options.message_loop_type = base::MessageLoop::TYPE_IO;
 240   io_thread.StartWithOptions(options);
 241
 242   // Parse command line.
 243   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
 244   SyncCredentials credentials;
 245   credentials.email = command_line.GetSwitchValueASCII(kEmailSwitch);
 246   credentials.sync_token = command_line.GetSwitchValueASCII(kTokenSwitch);
 247   // TODO(akalin): Write a wrapper script that gets a token for an
 248   // email and password and passes that in to this utility.
 249   if (credentials.email.empty() || credentials.sync_token.empty()) {
 250     std::printf("Usage: %s --%s=foo@bar.com --%s=token\n"
 251                 "[--%s=host:port] [--%s] [--%s]\n"
 252                 "Run chrome and set a breakpoint on\n"
 253                 "syncer::SyncManagerImpl::UpdateCredentials() "
 254                 "after logging into\n"
 255                 "sync to get the token to pass into this utility.\n",
 256                 argv[0],
 257                 kEmailSwitch, kTokenSwitch, kXmppHostPortSwitch,
 258                 kXmppTrySslTcpFirstSwitch,
 259                 kXmppAllowInsecureConnectionSwitch);
 260     return -1;
 261   }
 262
 263   // Set up objects that monitor the network.
 264   scoped_ptr<net::NetworkChangeNotifier> network_change_notifier(
 265       net::NetworkChangeNotifier::Create());
 266
 267   // Set up sync notifier factory.
 268   const scoped_refptr<MyTestURLRequestContextGetter> context_getter =
 269       new MyTestURLRequestContextGetter(io_thread.message_loop_proxy());
 270   const notifier::NotifierOptions& notifier_options =
 271       ParseNotifierOptions(command_line, context_getter);
 272   syncer::NetworkChannelCreator network_channel_creator =
 273       syncer::NonBlockingInvalidator::MakePushClientChannelCreator(
 274           notifier_options);
 275   const char kClientInfo[] = "standalone_sync_client";
 276   std::string invalidator_id = base::RandBytesAsString(8);
 277   NullInvalidationStateTracker null_invalidation_state_tracker;
 278   scoped_ptr<Invalidator> invalidator(new NonBlockingInvalidator(
 279       network_channel_creator,
 280       invalidator_id,
 281       null_invalidation_state_tracker.GetSavedInvalidations(),
 282       null_invalidation_state_tracker.GetBootstrapData(),
 283       WeakHandle<InvalidationStateTracker>(
 284           null_invalidation_state_tracker.AsWeakPtr()),
 285       kClientInfo,
 286       notifier_options.request_context_getter));
 287
 288   // Set up database directory for the syncer.
 289   base::ScopedTempDir database_dir;
 290   CHECK(database_dir.CreateUniqueTempDir());
 291
 292   // Developers often add types to ModelTypeSet::All() before the server
 293   // supports them.  We need to be explicit about which types we want here.
 294   ModelTypeSet model_types;
 295   model_types.Put(BOOKMARKS);
 296   model_types.Put(PREFERENCES);
 297   model_types.Put(PASSWORDS);
 298   model_types.Put(AUTOFILL);
 299   model_types.Put(THEMES);
 300   model_types.Put(TYPED_URLS);
 301   model_types.Put(EXTENSIONS);
 302   model_types.Put(NIGORI);
 303   model_types.Put(SEARCH_ENGINES);
 304   model_types.Put(SESSIONS);
 305   model_types.Put(APPS);
 306   model_types.Put(AUTOFILL_PROFILE);
 307   model_types.Put(APP_SETTINGS);
 308   model_types.Put(EXTENSION_SETTINGS);
 309   model_types.Put(APP_NOTIFICATIONS);
 310   model_types.Put(HISTORY_DELETE_DIRECTIVES);
 311   model_types.Put(SYNCED_NOTIFICATIONS);
 312   model_types.Put(SYNCED_NOTIFICATION_APP_INFO);
 313   model_types.Put(DEVICE_INFO);
 314   model_types.Put(EXPERIMENTS);
 315   model_types.Put(PRIORITY_PREFERENCES);
 316   model_types.Put(DICTIONARY);
 317   model_types.Put(FAVICON_IMAGES);
 318   model_types.Put(FAVICON_TRACKING);
 319
 320   ModelSafeRoutingInfo routing_info;
 321   for (ModelTypeSet::Iterator it = model_types.First();
 322        it.Good(); it.Inc()) {
 323     routing_info[it.Get()] = GROUP_PASSIVE;
 324   }
 325   scoped_refptr<PassiveModelWorker> passive_model_safe_worker =
 326       new PassiveModelWorker(&sync_loop, NULL);
 327   std::vector<scoped_refptr<ModelSafeWorker> > workers;
 328   workers.push_back(passive_model_safe_worker);
 329
 330   // Set up sync manager.
 331   SyncManagerFactory sync_manager_factory;
 332   scoped_ptr<SyncManager> sync_manager =
 333       sync_manager_factory.CreateSyncManager("sync_client manager");
 334   LoggingJsEventHandler js_event_handler;
 335   const char kSyncServerAndPath[] = "clients4.google.com/chrome-sync/dev";
 336   int kSyncServerPort = 443;
 337   bool kUseSsl = true;
 338   // Used only by InitialProcessMetadata(), so it's okay to leave this as NULL.
 339   const scoped_refptr<base::TaskRunner> blocking_task_runner = NULL;
 340   const char kUserAgent[] = "sync_client";
 341   // TODO(akalin): Replace this with just the context getter once
 342   // HttpPostProviderFactory is removed.
 343   CancelationSignal factory_cancelation_signal;
 344   scoped_ptr<HttpPostProviderFactory> post_factory(
 345       new HttpBridgeFactory(context_getter.get(),
 346                             base::Bind(&StubNetworkTimeUpdateCallback),
 347                             &factory_cancelation_signal));
 348   post_factory->Init(kUserAgent);
 349   // Used only when committing bookmarks, so it's okay to leave this
 350   // as NULL.
 351   ExtensionsActivity* extensions_activity = NULL;
 352   LoggingChangeDelegate change_delegate;
 353   const char kRestoredKeyForBootstrapping[] = "";
 354   const char kRestoredKeystoreKeyForBootstrapping[] = "";
 355   NullEncryptor null_encryptor;
 356   InternalComponentsFactoryImpl::Switches factory_switches = {
 357       InternalComponentsFactory::ENCRYPTION_KEYSTORE,
 358       InternalComponentsFactory::BACKOFF_NORMAL
 359   };
 360   CancelationSignal scm_cancelation_signal;
 361
 362   sync_manager->Init(database_dir.path(),
 363                     WeakHandle<JsEventHandler>(
 364                         js_event_handler.AsWeakPtr()),
 365                     kSyncServerAndPath,
 366                     kSyncServerPort,
 367                     kUseSsl,
 368                     post_factory.Pass(),
 369                     workers,
 370                     extensions_activity,
 371                     &change_delegate,
 372                     credentials,
 373                     invalidator_id,
 374                     kRestoredKeyForBootstrapping,
 375                     kRestoredKeystoreKeyForBootstrapping,
 376                     new InternalComponentsFactoryImpl(factory_switches),
 377                     &null_encryptor,
 378                     scoped_ptr<UnrecoverableErrorHandler>(
 379                         new LoggingUnrecoverableErrorHandler).Pass(),
 380                     &LogUnrecoverableErrorContext,
 381                     &scm_cancelation_signal);
 382   // TODO(akalin): Avoid passing in model parameters multiple times by
 383   // organizing handling of model types.
 384   invalidator->UpdateCredentials(credentials.email, credentials.sync_token);
 385   invalidator->RegisterHandler(sync_manager.get());
 386   invalidator->UpdateRegisteredIds(
 387       sync_manager.get(), ModelTypeSetToObjectIdSet(model_types));
 388   sync_manager->StartSyncingNormally(routing_info);
 389
 390   sync_loop.Run();
 391
 392   io_thread.Stop();
 393   return 0;
 394 }
 395
 396 }  // namespace
 397 }  // namespace syncer
 398
 399 int main(int argc, char* argv[]) {
 400   return syncer::SyncClientMain(argc, argv);
 401 }