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 "webkit/common/blob/shareable_file_reference.h"
6
7 #include <map>
8
9 #include "base/lazy_instance.h"
10 #include "base/message_loop/message_loop_proxy.h"
11 #include "base/task_runner.h"
12 #include "base/threading/non_thread_safe.h"
13
14 namespace webkit_blob {
15
16 namespace {
17
18 // A shareable file map with enforcement of thread checker.
19 class ShareableFileMap : public base::NonThreadSafe {
20 public:
21 typedef std::map<base::FilePath, ShareableFileReference*> FileMap;
22 typedef FileMap::iterator iterator;
23 typedef FileMap::key_type key_type;
24 typedef FileMap::value_type value_type;
25
ShareableFileMap()26 ShareableFileMap() {}
27
~ShareableFileMap()28 ~ShareableFileMap() {
29 DetachFromThread();
30 }
31
Find(key_type key)32 iterator Find(key_type key) {
33 DCHECK(CalledOnValidThread());
34 return file_map_.find(key);
35 }
36
End()37 iterator End() {
38 DCHECK(CalledOnValidThread());
39 return file_map_.end();
40 }
41
Insert(value_type value)42 std::pair<iterator, bool> Insert(value_type value) {
43 DCHECK(CalledOnValidThread());
44 return file_map_.insert(value);
45 }
46
Erase(key_type key)47 void Erase(key_type key) {
48 DCHECK(CalledOnValidThread());
49 file_map_.erase(key);
50 }
51
52 private:
53 FileMap file_map_;
54 DISALLOW_COPY_AND_ASSIGN(ShareableFileMap);
55 };
56
57 base::LazyInstance<ShareableFileMap> g_file_map = LAZY_INSTANCE_INITIALIZER;
58
59 } // namespace
60
61 // static
Get(const base::FilePath & path)62 scoped_refptr<ShareableFileReference> ShareableFileReference::Get(
63 const base::FilePath& path) {
64 ShareableFileMap::iterator found = g_file_map.Get().Find(path);
65 ShareableFileReference* reference =
66 (found == g_file_map.Get().End()) ? NULL : found->second;
67 return scoped_refptr<ShareableFileReference>(reference);
68 }
69
70 // static
GetOrCreate(const base::FilePath & path,FinalReleasePolicy policy,base::TaskRunner * file_task_runner)71 scoped_refptr<ShareableFileReference> ShareableFileReference::GetOrCreate(
72 const base::FilePath& path,
73 FinalReleasePolicy policy,
74 base::TaskRunner* file_task_runner) {
75 return GetOrCreate(
76 ScopedFile(path, static_cast<ScopedFile::ScopeOutPolicy>(policy),
77 file_task_runner));
78 }
79
80 // static
GetOrCreate(ScopedFile scoped_file)81 scoped_refptr<ShareableFileReference> ShareableFileReference::GetOrCreate(
82 ScopedFile scoped_file) {
83 if (scoped_file.path().empty())
84 return scoped_refptr<ShareableFileReference>();
85
86 typedef std::pair<ShareableFileMap::iterator, bool> InsertResult;
87 // Required for VS2010: http://connect.microsoft.com/VisualStudio/feedback/details/520043/error-converting-from-null-to-a-pointer-type-in-std-pair
88 webkit_blob::ShareableFileReference* null_reference = NULL;
89 InsertResult result = g_file_map.Get().Insert(
90 ShareableFileMap::value_type(scoped_file.path(), null_reference));
91 if (result.second == false) {
92 scoped_file.Release();
93 return scoped_refptr<ShareableFileReference>(result.first->second);
94 }
95
96 // Wasn't in the map, create a new reference and store the pointer.
97 scoped_refptr<ShareableFileReference> reference(
98 new ShareableFileReference(scoped_file.Pass()));
99 result.first->second = reference.get();
100 return reference;
101 }
102
AddFinalReleaseCallback(const FinalReleaseCallback & callback)103 void ShareableFileReference::AddFinalReleaseCallback(
104 const FinalReleaseCallback& callback) {
105 DCHECK(g_file_map.Get().CalledOnValidThread());
106 scoped_file_.AddScopeOutCallback(callback, NULL);
107 }
108
ShareableFileReference(ScopedFile scoped_file)109 ShareableFileReference::ShareableFileReference(ScopedFile scoped_file)
110 : scoped_file_(scoped_file.Pass()) {
111 DCHECK(g_file_map.Get().Find(path())->second == NULL);
112 }
113
~ShareableFileReference()114 ShareableFileReference::~ShareableFileReference() {
115 DCHECK(g_file_map.Get().Find(path())->second == this);
116 g_file_map.Get().Erase(path());
117 }
118
119 } // namespace webkit_blob
120