Tue, 06 Jan 2015 21:39:09 +0100
Conditionally force memory storage according to privacy.thirdparty.isolate;
This solves Tor bug #9701, complying with disk avoidance documented in
https://www.torproject.org/projects/torbrowser/design/#disk-avoidance.
michael@0 | 1 | /* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*- |
michael@0 | 2 | * This Source Code Form is subject to the terms of the Mozilla Public |
michael@0 | 3 | * License, v. 2.0. If a copy of the MPL was not distributed with this |
michael@0 | 4 | * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
michael@0 | 5 | |
michael@0 | 6 | #ifndef MOZILLA_GFX_DRAWEVENTRECORDER_H_ |
michael@0 | 7 | #define MOZILLA_GFX_DRAWEVENTRECORDER_H_ |
michael@0 | 8 | |
michael@0 | 9 | #include "2D.h" |
michael@0 | 10 | #include "RecordedEvent.h" |
michael@0 | 11 | #include <ostream> |
michael@0 | 12 | #include <fstream> |
michael@0 | 13 | |
michael@0 | 14 | #if defined(_MSC_VER) |
michael@0 | 15 | #include <hash_set> |
michael@0 | 16 | #else |
michael@0 | 17 | #include <set> |
michael@0 | 18 | #endif |
michael@0 | 19 | |
michael@0 | 20 | namespace mozilla { |
michael@0 | 21 | namespace gfx { |
michael@0 | 22 | |
michael@0 | 23 | class PathRecording; |
michael@0 | 24 | |
michael@0 | 25 | class DrawEventRecorderPrivate : public DrawEventRecorder |
michael@0 | 26 | { |
michael@0 | 27 | public: |
michael@0 | 28 | MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(DrawEventRecorderPrivate) |
michael@0 | 29 | DrawEventRecorderPrivate(std::ostream *aStream); |
michael@0 | 30 | virtual ~DrawEventRecorderPrivate() { } |
michael@0 | 31 | |
michael@0 | 32 | void RecordEvent(const RecordedEvent &aEvent); |
michael@0 | 33 | void WritePath(const PathRecording *aPath); |
michael@0 | 34 | |
michael@0 | 35 | void AddStoredPath(const ReferencePtr aPath) { |
michael@0 | 36 | mStoredPaths.insert(aPath); |
michael@0 | 37 | } |
michael@0 | 38 | |
michael@0 | 39 | void RemoveStoredPath(const ReferencePtr aPath) { |
michael@0 | 40 | mStoredPaths.erase(aPath); |
michael@0 | 41 | } |
michael@0 | 42 | |
michael@0 | 43 | bool HasStoredPath(const ReferencePtr aPath) { |
michael@0 | 44 | if (mStoredPaths.find(aPath) != mStoredPaths.end()) { |
michael@0 | 45 | return true; |
michael@0 | 46 | } |
michael@0 | 47 | return false; |
michael@0 | 48 | } |
michael@0 | 49 | |
michael@0 | 50 | protected: |
michael@0 | 51 | std::ostream *mOutputStream; |
michael@0 | 52 | |
michael@0 | 53 | virtual void Flush() = 0; |
michael@0 | 54 | |
michael@0 | 55 | #if defined(_MSC_VER) |
michael@0 | 56 | typedef stdext::hash_set<const void*> ObjectSet; |
michael@0 | 57 | #else |
michael@0 | 58 | typedef std::set<const void*> ObjectSet; |
michael@0 | 59 | #endif |
michael@0 | 60 | |
michael@0 | 61 | ObjectSet mStoredPaths; |
michael@0 | 62 | ObjectSet mStoredScaledFonts; |
michael@0 | 63 | }; |
michael@0 | 64 | |
michael@0 | 65 | class DrawEventRecorderFile : public DrawEventRecorderPrivate |
michael@0 | 66 | { |
michael@0 | 67 | public: |
michael@0 | 68 | MOZ_DECLARE_REFCOUNTED_VIRTUAL_TYPENAME(DrawEventRecorderFile) |
michael@0 | 69 | DrawEventRecorderFile(const char *aFilename); |
michael@0 | 70 | ~DrawEventRecorderFile(); |
michael@0 | 71 | |
michael@0 | 72 | private: |
michael@0 | 73 | virtual void Flush(); |
michael@0 | 74 | |
michael@0 | 75 | std::ofstream mOutputFile; |
michael@0 | 76 | }; |
michael@0 | 77 | |
michael@0 | 78 | } |
michael@0 | 79 | } |
michael@0 | 80 | |
michael@0 | 81 | #endif /* MOZILLA_GFX_DRAWEVENTRECORDER_H_ */ |