mojo: Sketch a profile application.

content/browser/browser_context.cc

 // 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 "content/public/browser/browser_context.h"
 
 #include <stddef.h>
 #include <stdint.h>
+#include <algorithm>
+#include <limits>
 #include <utility>
 
+#include "base/lazy_instance.h"
+#include "base/rand_util.h"
 #include "build/build_config.h"
 
 #if !defined(OS_IOS)
+#include "components/profile_service/profile_app.h"
 #include "content/browser/download/download_manager_impl.h"
 #include "content/browser/fileapi/chrome_blob_storage_context.h"
 #include "content/browser/indexed_db/indexed_db_context_impl.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
 #include "content/browser/push_messaging/push_messaging_router.h"
 #include "content/browser/storage_partition_impl_map.h"
 #include "content/common/child_process_host_impl.h"
 #include "content/public/browser/blob_handle.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/site_instance.h"
 #include "net/cookies/cookie_store.h"
 #include "net/ssl/channel_id_service.h"
 #include "net/ssl/channel_id_store.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "storage/browser/database/database_tracker.h"
 #include "storage/browser/fileapi/external_mount_points.h"
 #endif // !OS_IOS
 
 using base::UserDataAdapter;
 
 namespace content {
 
 // Only ~BrowserContext() is needed on iOS.
 #if !defined(OS_IOS)
 namespace {
 
+base::LazyInstance<std::set<uint32_t>> g_used_user_ids =
+    LAZY_INSTANCE_INITIALIZER;
+base::LazyInstance<std::vector<std::pair<BrowserContext*, uint32_t>>>
+g_context_to_user_id = LAZY_INSTANCE_INITIALIZER;
+
 // Key names on BrowserContext.
 const char kDownloadManagerKeyName[] = "download_manager";
 const char kStoragePartitionMapKeyName[] = "content_storage_partition_map";
 
+const char kMojoWasInitialized[] = "mojo-was-initialized";
+
 #if defined(OS_CHROMEOS)
 const char kMountPointsKey[] = "mount_points";
 #endif  // defined(OS_CHROMEOS)
 
 StoragePartitionImplMap* GetStoragePartitionMap(
     BrowserContext* browser_context) {
   StoragePartitionImplMap* partition_map =
       static_cast<StoragePartitionImplMap*>(
           browser_context->GetUserData(kStoragePartitionMapKeyName));
   if (!partition_map) {
@@ skipping 258 matching lines @@
                    make_scoped_refptr(indexed_db_context_impl)));
   }
 }
 
 void BrowserContext::SetDownloadManagerForTesting(
     BrowserContext* browser_context,
     DownloadManager* download_manager) {
   SetDownloadManager(browser_context, download_manager);
 }
 
+void BrowserContext::Initialize(
+    BrowserContext* browser_context,
+    const base::FilePath& path) {
+  // Associate a random unsigned 32 bit number with |browser_context|. This
+  // becomes the mojo user id for this BrowserContext.
+  uint32_t new_id = static_cast<uint32_t>(base::RandGenerator(UINT32_MAX));
+  while ((new_id == 0) ||
+         (g_used_user_ids.Get().find(new_id) != g_used_user_ids.Get().end())) {
+    new_id = static_cast<uint32_t>(base::RandGenerator(UINT32_MAX));
+  }
+
+  g_used_user_ids.Get().insert(new_id);
+  g_context_to_user_id.Get().push_back(std::make_pair(browser_context, new_id));
+
+  profile::ProfileApp::AssociateMojoUserIDWithProfileDir(new_id, path);
+  browser_context->SetUserData(kMojoWasInitialized,
+                               new base::SupportsUserData::Data);
+}
+
+uint32_t BrowserContext::GetMojoUserIdFor(BrowserContext* browser_context) {
+  CHECK(browser_context->GetUserData(kMojoWasInitialized))
+      << "Attempting to get the mojo user id for a BrowserContext that was "
+      << "never Initialize()ed.";
+
+  auto it = std::find_if(
+      g_context_to_user_id.Get().begin(),
+      g_context_to_user_id.Get().end(),
+      [&browser_context](const std::pair<BrowserContext*, uint32_t>& p) {
+        return p.first == browser_context; });
+  CHECK(it != g_context_to_user_id.Get().end());
+  return it->second;
+}
+
 #endif  // !OS_IOS
 
 BrowserContext::~BrowserContext() {
+  CHECK(GetUserData(kMojoWasInitialized))
+      << "Attempting to destroy a BrowserContext that never called "
+      << "Initialize()";
+
 #if !defined(OS_IOS)
   if (GetUserData(kDownloadManagerKeyName))
     GetDownloadManager(this)->Shutdown();
 #endif
 }
 
 }  // namespace content