dom/quota/FileStreams.cpp

Sat, 03 Jan 2015 20:18:00 +0100

author
Michael Schloh von Bennewitz <michael@schloh.com>
date
Sat, 03 Jan 2015 20:18:00 +0100
branch
TOR_BUG_3246
changeset 7
129ffea94266
permissions
-rw-r--r--

Conditionally enable double key logic according to:
private browsing mode or privacy.thirdparty.isolate preference and
implement in GetCookieStringCommon and FindCookie where it counts...
With some reservations of how to convince FindCookie users to test
condition and pass a nullptr when disabling double key logic.

     1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
     2 /* vim: set ts=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/. */
     7 #include "FileStreams.h"
     9 #include "QuotaManager.h"
    10 #include "prio.h"
    12 USING_QUOTA_NAMESPACE
    14 template <class FileStreamBase>
    15 NS_IMETHODIMP
    16 FileQuotaStream<FileStreamBase>::SetEOF()
    17 {
    18   nsresult rv = FileStreamBase::SetEOF();
    19   NS_ENSURE_SUCCESS(rv, rv);
    21   if (mQuotaObject) {
    22     int64_t offset;
    23     nsresult rv = FileStreamBase::Tell(&offset);
    24     NS_ENSURE_SUCCESS(rv, rv);
    26     mQuotaObject->UpdateSize(offset);
    27   }
    29   return NS_OK;
    30 }
    32 template <class FileStreamBase>
    33 NS_IMETHODIMP
    34 FileQuotaStream<FileStreamBase>::Close()
    35 {
    36   nsresult rv = FileStreamBase::Close();
    37   NS_ENSURE_SUCCESS(rv, rv);
    39   mQuotaObject = nullptr;
    41   return NS_OK;
    42 }
    44 template <class FileStreamBase>
    45 nsresult
    46 FileQuotaStream<FileStreamBase>::DoOpen()
    47 {
    48   QuotaManager* quotaManager = QuotaManager::Get();
    49   NS_ASSERTION(quotaManager, "Shouldn't be null!");
    51   NS_ASSERTION(!mQuotaObject, "Creating quota object more than once?");
    52   mQuotaObject = quotaManager->GetQuotaObject(mPersistenceType, mGroup, mOrigin,
    53     FileStreamBase::mOpenParams.localFile);
    55   nsresult rv = FileStreamBase::DoOpen();
    56   NS_ENSURE_SUCCESS(rv, rv);
    58   if (mQuotaObject && (FileStreamBase::mOpenParams.ioFlags & PR_TRUNCATE)) {
    59     mQuotaObject->UpdateSize(0);
    60   }
    62   return NS_OK;
    63 }
    65 template <class FileStreamBase>
    66 NS_IMETHODIMP
    67 FileQuotaStreamWithWrite<FileStreamBase>::Write(const char* aBuf,
    68                                                 uint32_t aCount,
    69                                                 uint32_t* _retval)
    70 {
    71   nsresult rv;
    73   if (FileQuotaStreamWithWrite::mQuotaObject) {
    74     int64_t offset;
    75     rv = FileStreamBase::Tell(&offset);
    76     NS_ENSURE_SUCCESS(rv, rv);
    78     if (!FileQuotaStreamWithWrite::
    79          mQuotaObject->MaybeAllocateMoreSpace(offset, aCount)) {
    80       return NS_ERROR_FILE_NO_DEVICE_SPACE;
    81     }
    82   }
    84   rv = FileStreamBase::Write(aBuf, aCount, _retval);
    85   NS_ENSURE_SUCCESS(rv, rv);
    87   return NS_OK;
    88 }
    90 NS_IMPL_ISUPPORTS_INHERITED0(FileInputStream, nsFileInputStream)
    92 already_AddRefed<FileInputStream>
    93 FileInputStream::Create(PersistenceType aPersistenceType,
    94                         const nsACString& aGroup, const nsACString& aOrigin,
    95                         nsIFile* aFile, int32_t aIOFlags, int32_t aPerm,
    96                         int32_t aBehaviorFlags)
    97 {
    98   nsRefPtr<FileInputStream> stream =
    99     new FileInputStream(aPersistenceType, aGroup, aOrigin);
   100   nsresult rv = stream->Init(aFile, aIOFlags, aPerm, aBehaviorFlags);
   101   NS_ENSURE_SUCCESS(rv, nullptr);
   102   return stream.forget();
   103 }
   105 NS_IMPL_ISUPPORTS_INHERITED0(FileOutputStream, nsFileOutputStream)
   107 already_AddRefed<FileOutputStream>
   108 FileOutputStream::Create(PersistenceType aPersistenceType,
   109                          const nsACString& aGroup, const nsACString& aOrigin,
   110                          nsIFile* aFile, int32_t aIOFlags, int32_t aPerm,
   111                          int32_t aBehaviorFlags)
   112 {
   113   nsRefPtr<FileOutputStream> stream =
   114     new FileOutputStream(aPersistenceType, aGroup, aOrigin);
   115   nsresult rv = stream->Init(aFile, aIOFlags, aPerm, aBehaviorFlags);
   116   NS_ENSURE_SUCCESS(rv, nullptr);
   117   return stream.forget();
   118 }
   120 NS_IMPL_ISUPPORTS_INHERITED0(FileStream, nsFileStream)
   122 already_AddRefed<FileStream>
   123 FileStream::Create(PersistenceType aPersistenceType, const nsACString& aGroup,
   124                    const nsACString& aOrigin, nsIFile* aFile, int32_t aIOFlags,
   125                    int32_t aPerm, int32_t aBehaviorFlags)
   126 {
   127   nsRefPtr<FileStream> stream =
   128     new FileStream(aPersistenceType, aGroup, aOrigin);
   129   nsresult rv = stream->Init(aFile, aIOFlags, aPerm, aBehaviorFlags);
   130   NS_ENSURE_SUCCESS(rv, nullptr);
   131   return stream.forget();
   132 }

mercurial