tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

DataStorageManager.cpp (2550B)

      1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
      2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
      3 /* This Source Code Form is subject to the terms of the Mozilla Public
      4 * License, v. 2.0. If a copy of the MPL was not distributed with this
      5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
      6 
      7 #include "DataStorageManager.h"
      8 #include "MainThreadUtils.h"
      9 #include "nsIMemoryReporter.h"
     10 #include "nsString.h"
     11 
     12 using VoidPtrToSizeFn = uintptr_t (*)(const void* ptr);
     13 
     14 extern "C" nsresult make_data_storage(const nsAString* basename,
     15                                      size_t valueLength,
     16                                      VoidPtrToSizeFn sizeOfOp,
     17                                      VoidPtrToSizeFn enclosingSizeOfOp,
     18                                      nsIDataStorage** result);
     19 
     20 MOZ_DEFINE_MALLOC_SIZE_OF(DataStorageMallocSizeOf)
     21 MOZ_DEFINE_MALLOC_ENCLOSING_SIZE_OF(DataStorageMallocEnclosingSizeOf)
     22 
     23 namespace mozilla {
     24 
     25 NS_IMPL_ISUPPORTS(DataStorageManager, nsIDataStorageManager)
     26 
     27 NS_IMETHODIMP
     28 DataStorageManager::Get(nsIDataStorageManager::DataStorage aDataStorage,
     29                        nsIDataStorage** aResult) {
     30  if (!NS_IsMainThread()) {
     31    return NS_ERROR_NOT_SAME_THREAD;
     32  }
     33  nsAutoString filename;
     34  size_t valueLength = 1024;
     35  switch (aDataStorage) {
     36    case nsIDataStorageManager::AlternateServices:
     37      if (mAlternateServicesCreated) {
     38        return NS_ERROR_ALREADY_INITIALIZED;
     39      }
     40      mAlternateServicesCreated = true;
     41      filename.Assign(u"AlternateServices"_ns);
     42      break;
     43    case nsIDataStorageManager::ClientAuthRememberList:
     44      if (mClientAuthRememberListCreated) {
     45        return NS_ERROR_ALREADY_INITIALIZED;
     46      }
     47      mClientAuthRememberListCreated = true;
     48      filename.Assign(u"ClientAuthRememberList"_ns);
     49      break;
     50    case nsIDataStorageManager::SiteSecurityServiceState:
     51      if (mSiteSecurityServiceStateCreated) {
     52        return NS_ERROR_ALREADY_INITIALIZED;
     53      }
     54      mSiteSecurityServiceStateCreated = true;
     55      filename.Assign(u"SiteSecurityServiceState"_ns);
     56      // For most nsIDataStorage use cases, values can be quite long (1024
     57      // bytes by default). For HSTS, much less information is stored, so save
     58      // space by limiting values to 24 bytes.
     59      valueLength = 24;
     60      break;
     61    default:
     62      return NS_ERROR_INVALID_ARG;
     63  }
     64  return make_data_storage(&filename, valueLength, &DataStorageMallocSizeOf,
     65                           &DataStorageMallocEnclosingSizeOf, aResult);
     66 }
     67 
     68 }  // namespace mozilla