OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/logging.h" | 5 #include "base/logging.h" |
6 #include "net/base/io_buffer.h" | 6 #include "net/base/io_buffer.h" |
7 #include "net/base/net_errors.h" | 7 #include "net/base/net_errors.h" |
8 #include "net/disk_cache/flash/cache_entry.h" | |
9 #include "net/disk_cache/flash/format.h" | 8 #include "net/disk_cache/flash/format.h" |
10 #include "net/disk_cache/flash/log_structured_store.h" | 9 #include "net/disk_cache/flash/log_store.h" |
| 10 #include "net/disk_cache/flash/log_store_entry.h" |
11 | 11 |
12 namespace disk_cache { | 12 namespace disk_cache { |
13 | 13 |
14 CacheEntry::CacheEntry(LogStructuredStore* store) | 14 LogStoreEntry::LogStoreEntry(LogStore* store) |
15 : store_(store), id_(-1), init_(false), closed_(false) { | 15 : store_(store), id_(-1), init_(false), closed_(false) { |
16 DCHECK(store); | 16 DCHECK(store); |
17 } | 17 } |
18 | 18 |
19 CacheEntry::CacheEntry(LogStructuredStore* store, int32 id) | 19 LogStoreEntry::LogStoreEntry(LogStore* store, int32 id) |
20 : store_(store), id_(id), init_(false), closed_(false) { | 20 : store_(store), id_(id), init_(false), closed_(false) { |
21 DCHECK(store); | 21 DCHECK(store); |
22 } | 22 } |
23 | 23 |
24 CacheEntry::~CacheEntry() { | 24 LogStoreEntry::~LogStoreEntry() { |
25 DCHECK(!init_ || closed_); | 25 DCHECK(!init_ || closed_); |
26 } | 26 } |
27 | 27 |
28 bool CacheEntry::Init() { | 28 bool LogStoreEntry::Init() { |
29 DCHECK(!init_); | 29 DCHECK(!init_); |
30 if (!OnDisk()) { | 30 if (!OnDisk()) { |
31 init_ = true; | 31 init_ = true; |
32 return true; | 32 return true; |
33 } | 33 } |
34 | 34 |
35 int32 stream_sizes[kFlashCacheEntryNumStreams]; | 35 int32 stream_sizes[kFlashLogStoreEntryNumStreams]; |
36 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashCacheEntryHeaderSize, | 36 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashLogStoreEntryHeaderSize, |
37 invalid_cache_entry_header_size); | 37 invalid_log_store_entry_header_size); |
38 | 38 |
39 if (!store_->OpenEntry(id_) || | 39 if (!store_->OpenEntry(id_) || |
40 !store_->ReadData(id_, stream_sizes, kFlashCacheEntryHeaderSize, 0)) { | 40 !store_->ReadData(id_, stream_sizes, kFlashLogStoreEntryHeaderSize, 0)) { |
41 return false; | 41 return false; |
42 } | 42 } |
43 for (int i = 0, offset = kFlashCacheEntryHeaderSize; | 43 for (int i = 0, offset = kFlashLogStoreEntryHeaderSize; |
44 i < kFlashCacheEntryNumStreams; ++i) { | 44 i < kFlashLogStoreEntryNumStreams; ++i) { |
45 streams_[i].offset = offset; | 45 streams_[i].offset = offset; |
46 streams_[i].size = stream_sizes[i]; | 46 streams_[i].size = stream_sizes[i]; |
47 offset += stream_sizes[i]; | 47 offset += stream_sizes[i]; |
48 } | 48 } |
49 init_ = true; | 49 init_ = true; |
50 return true; | 50 return true; |
51 } | 51 } |
52 | 52 |
53 bool CacheEntry::Close() { | 53 bool LogStoreEntry::Close() { |
54 DCHECK(init_ && !closed_); | 54 DCHECK(init_ && !closed_); |
55 if (OnDisk()) | 55 if (OnDisk()) |
56 store_->CloseEntry(id_); | 56 store_->CloseEntry(id_); |
57 else if (!Save()) | 57 else if (!Save()) |
58 return false; | 58 return false; |
59 closed_ = true; | 59 closed_ = true; |
60 return true; | 60 return true; |
61 } | 61 } |
62 | 62 |
63 int32 CacheEntry::id() const { | 63 int32 LogStoreEntry::id() const { |
64 DCHECK(init_); | 64 DCHECK(init_); |
65 return id_; | 65 return id_; |
66 } | 66 } |
67 | 67 |
68 int32 CacheEntry::GetDataSize(int index) const { | 68 int32 LogStoreEntry::GetDataSize(int index) const { |
69 DCHECK(init_); | 69 DCHECK(init_); |
70 return InvalidStream(index) ? 0 : streams_[index].size; | 70 return InvalidStream(index) ? 0 : streams_[index].size; |
71 } | 71 } |
72 | 72 |
73 int CacheEntry::ReadData(int index, int offset, net::IOBuffer* buf, | 73 int LogStoreEntry::ReadData(int index, int offset, net::IOBuffer* buf, |
74 int buf_len) { | 74 int buf_len) { |
75 DCHECK(init_); | 75 DCHECK(init_); |
76 if (InvalidStream(index)) | 76 if (InvalidStream(index)) |
77 return net::ERR_INVALID_ARGUMENT; | 77 return net::ERR_INVALID_ARGUMENT; |
78 | 78 |
79 int stream_size = streams_[index].size; | 79 int stream_size = streams_[index].size; |
80 if (offset >= stream_size || offset < 0 || buf_len == 0) | 80 if (offset >= stream_size || offset < 0 || buf_len == 0) |
81 return 0; | 81 return 0; |
82 if (offset + buf_len > stream_size) | 82 if (offset + buf_len > stream_size) |
83 buf_len = stream_size - offset; | 83 buf_len = stream_size - offset; |
84 | 84 |
85 if (OnDisk()) { | 85 if (OnDisk()) { |
86 offset += streams_[index].offset; | 86 offset += streams_[index].offset; |
87 if (store_->ReadData(id_, buf->data(), buf_len, offset)) | 87 if (store_->ReadData(id_, buf->data(), buf_len, offset)) |
88 return buf_len; | 88 return buf_len; |
89 return net::ERR_FAILED; | 89 return net::ERR_FAILED; |
90 } | 90 } |
91 memcpy(buf->data(), &streams_[index].write_buffer[offset], buf_len); | 91 memcpy(buf->data(), &streams_[index].write_buffer[offset], buf_len); |
92 return buf_len; | 92 return buf_len; |
93 } | 93 } |
94 | 94 |
95 int CacheEntry::WriteData(int index, int offset, net::IOBuffer* buf, | 95 int LogStoreEntry::WriteData(int index, int offset, net::IOBuffer* buf, |
96 int buf_len) { | 96 int buf_len) { |
97 DCHECK(init_ && !closed_); | 97 DCHECK(init_ && !closed_); |
98 if (InvalidStream(index)) | 98 if (InvalidStream(index)) |
99 return net::ERR_INVALID_ARGUMENT; | 99 return net::ERR_INVALID_ARGUMENT; |
100 | 100 |
101 DCHECK(offset >= 0 && buf_len >= 0); | 101 DCHECK(offset >= 0 && buf_len >= 0); |
102 Stream& stream = streams_[index]; | 102 Stream& stream = streams_[index]; |
103 size_t new_size = static_cast<size_t>(offset + buf_len); | 103 size_t new_size = static_cast<size_t>(offset + buf_len); |
104 if (new_size) { | 104 if (new_size) { |
105 // TODO(agayev): Currently, only append and overwrite is supported. Add | 105 // TODO(agayev): Currently, only append and overwrite is supported. Add |
106 // support for arbitrary writes. | 106 // support for arbitrary writes. |
107 DCHECK(!offset || offset == stream.size); | 107 DCHECK(!offset || offset == stream.size); |
108 if (stream.write_buffer.size() < new_size) | 108 if (stream.write_buffer.size() < new_size) |
109 stream.write_buffer.resize(new_size); | 109 stream.write_buffer.resize(new_size); |
110 memcpy(&streams_[index].write_buffer[offset], buf->data(), buf_len); | 110 memcpy(&streams_[index].write_buffer[offset], buf->data(), buf_len); |
111 } | 111 } |
112 stream.size = new_size; | 112 stream.size = new_size; |
113 return buf_len; | 113 return buf_len; |
114 } | 114 } |
115 | 115 |
116 bool CacheEntry::OnDisk() const { | 116 bool LogStoreEntry::OnDisk() const { |
117 return id_ != -1; | 117 return id_ != -1; |
118 } | 118 } |
119 | 119 |
120 bool CacheEntry::InvalidStream(int stream_index) const { | 120 bool LogStoreEntry::InvalidStream(int stream_index) const { |
121 return stream_index < 0 || stream_index >= kFlashCacheEntryNumStreams; | 121 return stream_index < 0 || stream_index >= kFlashLogStoreEntryNumStreams; |
122 } | 122 } |
123 | 123 |
124 int32 CacheEntry::Size() const { | 124 int32 LogStoreEntry::Size() const { |
125 DCHECK(init_); | 125 DCHECK(init_); |
126 int32 size = kFlashCacheEntryHeaderSize; | 126 int32 size = kFlashLogStoreEntryHeaderSize; |
127 for (int i = 0; i < kFlashCacheEntryNumStreams; ++i) | 127 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) |
128 size += streams_[i].size; | 128 size += streams_[i].size; |
129 DCHECK(size > 0 && size <= kFlashSegmentFreeSpace); | 129 DCHECK(size > 0 && size <= kFlashSegmentFreeSpace); |
130 return size; | 130 return size; |
131 } | 131 } |
132 | 132 |
133 bool CacheEntry::Save() { | 133 bool LogStoreEntry::Save() { |
134 DCHECK(init_ && !closed_ && !OnDisk()); | 134 DCHECK(init_ && !closed_ && !OnDisk()); |
135 int32 stream_sizes[kFlashCacheEntryNumStreams]; | 135 int32 stream_sizes[kFlashLogStoreEntryNumStreams]; |
136 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashCacheEntryHeaderSize, | 136 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashLogStoreEntryHeaderSize, |
137 invalid_cache_entry_header_size); | 137 invalid_log_store_entry_header_size); |
138 | 138 |
139 for (int i = 0; i < kFlashCacheEntryNumStreams; ++i) | 139 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) |
140 stream_sizes[i] = streams_[i].size; | 140 stream_sizes[i] = streams_[i].size; |
141 | 141 |
142 if (!store_->CreateEntry(Size(), &id_)) | 142 if (!store_->CreateEntry(Size(), &id_)) |
143 return false; | 143 return false; |
144 if (!store_->WriteData(stream_sizes, kFlashCacheEntryHeaderSize)) | 144 if (!store_->WriteData(stream_sizes, kFlashLogStoreEntryHeaderSize)) |
145 return false; | 145 return false; |
146 for (int i = 0; i < kFlashCacheEntryNumStreams; ++i) { | 146 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) { |
147 if (streams_[i].size > 0 && | 147 if (streams_[i].size > 0 && |
148 !store_->WriteData(&streams_[i].write_buffer[0], streams_[i].size)) { | 148 !store_->WriteData(&streams_[i].write_buffer[0], streams_[i].size)) { |
149 return false; | 149 return false; |
150 } | 150 } |
151 } | 151 } |
152 store_->CloseEntry(id_); | 152 store_->CloseEntry(id_); |
153 return true; | 153 return true; |
154 } | 154 } |
155 | 155 |
156 CacheEntry::Stream::Stream() : offset(0), size(0) { | 156 LogStoreEntry::Stream::Stream() : offset(0), size(0) { |
157 } | 157 } |
158 | 158 |
159 CacheEntry::Stream::~Stream() { | 159 LogStoreEntry::Stream::~Stream() { |
160 } | 160 } |
161 | 161 |
162 } // namespace disk_cache | 162 } // namespace disk_cache |
OLD | NEW |