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/format.h" | 8 #include "net/disk_cache/flash/format.h" |
9 #include "net/disk_cache/flash/log_store.h" | 9 #include "net/disk_cache/flash/log_store.h" |
10 #include "net/disk_cache/flash/log_store_entry.h" | 10 #include "net/disk_cache/flash/log_store_entry.h" |
11 | 11 |
12 namespace disk_cache { | 12 namespace disk_cache { |
13 | 13 |
14 LogStoreEntry::LogStoreEntry(LogStore* store) | 14 LogStoreEntry::LogStoreEntry(LogStore* store) |
15 : store_(store), id_(-1), init_(false), closed_(false), deleted_(false) { | 15 : store_(store), id_(-1), init_(false), closed_(false), deleted_(false) { |
16 DCHECK(store); | 16 DCHECK(store); |
17 } | 17 } |
18 | 18 |
19 LogStoreEntry::LogStoreEntry(LogStore* store, int32 id) | 19 LogStoreEntry::LogStoreEntry(LogStore* store, int32 id) |
20 : store_(store), id_(id), init_(false), closed_(false), deleted_(false) { | 20 : store_(store), id_(id), init_(false), closed_(false), deleted_(false) { |
21 DCHECK(store); | 21 DCHECK(store); |
22 } | 22 } |
23 | 23 |
24 LogStoreEntry::~LogStoreEntry() { | 24 LogStoreEntry::~LogStoreEntry() { |
25 DCHECK(!init_ || closed_); | 25 DCHECK(!init_ || closed_); |
26 } | 26 } |
27 | 27 |
28 bool LogStoreEntry::Init() { | 28 bool LogStoreEntry::Init() { |
29 DCHECK(!init_); | 29 DCHECK(!init_); |
30 if (!ReadOnly()) { | 30 if (IsNew()) { |
31 init_ = true; | 31 init_ = true; |
32 return true; | 32 return true; |
33 } | 33 } |
34 | 34 |
35 int32 stream_sizes[kFlashLogStoreEntryNumStreams]; | 35 int32 stream_sizes[kFlashLogStoreEntryNumStreams]; |
36 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashLogStoreEntryHeaderSize, | 36 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashLogStoreEntryHeaderSize, |
37 invalid_log_store_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, kFlashLogStoreEntryHeaderSize, 0)) { | 40 !store_->ReadData(id_, stream_sizes, kFlashLogStoreEntryHeaderSize, 0)) { |
41 return false; | 41 return false; |
42 } | 42 } |
43 for (int i = 0, offset = kFlashLogStoreEntryHeaderSize; | 43 for (int i = 0, offset = kFlashLogStoreEntryHeaderSize; |
44 i < kFlashLogStoreEntryNumStreams; ++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 LogStoreEntry::Close() { | 53 bool LogStoreEntry::Close() { |
54 DCHECK(init_ && !closed_); | 54 DCHECK(init_ && !closed_); |
55 | 55 |
56 if (ReadOnly()) { | 56 if (IsNew()) { |
| 57 closed_ = deleted_ ? true : Save(); |
| 58 } else { |
57 store_->CloseEntry(id_); | 59 store_->CloseEntry(id_); |
58 if (deleted_) | 60 if (deleted_) |
59 store_->DeleteEntry(id_, Size()); | 61 store_->DeleteEntry(id_, Size()); |
60 closed_ = true; | 62 closed_ = true; |
61 } else { | |
62 closed_ = deleted_ ? true : Save(); | |
63 } | 63 } |
64 return closed_; | 64 return closed_; |
65 } | 65 } |
66 | 66 |
67 int32 LogStoreEntry::id() const { | 67 int32 LogStoreEntry::id() const { |
68 DCHECK(init_); | 68 DCHECK(init_); |
69 return id_; | 69 return id_; |
70 } | 70 } |
71 | 71 |
72 int32 LogStoreEntry::GetDataSize(int index) const { | 72 int32 LogStoreEntry::GetDataSize(int index) const { |
73 DCHECK(init_); | 73 DCHECK(init_); |
74 return InvalidStream(index) ? 0 : streams_[index].size; | 74 return InvalidStream(index) ? 0 : streams_[index].size; |
75 } | 75 } |
76 | 76 |
77 int LogStoreEntry::ReadData(int index, int offset, net::IOBuffer* buf, | 77 int LogStoreEntry::ReadData(int index, int offset, net::IOBuffer* buf, |
78 int buf_len) { | 78 int buf_len) { |
79 DCHECK(init_); | 79 DCHECK(init_); |
80 if (InvalidStream(index)) | 80 if (InvalidStream(index)) |
81 return net::ERR_INVALID_ARGUMENT; | 81 return net::ERR_INVALID_ARGUMENT; |
82 | 82 |
83 int stream_size = streams_[index].size; | 83 int stream_size = streams_[index].size; |
84 if (offset >= stream_size || offset < 0 || buf_len == 0) | 84 if (offset >= stream_size || offset < 0 || buf_len == 0) |
85 return 0; | 85 return 0; |
86 if (offset + buf_len > stream_size) | 86 if (offset + buf_len > stream_size) |
87 buf_len = stream_size - offset; | 87 buf_len = stream_size - offset; |
88 | 88 |
89 if (ReadOnly()) { | 89 if (!IsNew()) { |
90 offset += streams_[index].offset; | 90 offset += streams_[index].offset; |
91 if (store_->ReadData(id_, buf->data(), buf_len, offset)) | 91 if (store_->ReadData(id_, buf->data(), buf_len, offset)) |
92 return buf_len; | 92 return buf_len; |
93 return net::ERR_FAILED; | 93 return net::ERR_FAILED; |
94 } | 94 } |
95 memcpy(buf->data(), &streams_[index].write_buffer[offset], buf_len); | 95 memcpy(buf->data(), &streams_[index].write_buffer[offset], buf_len); |
96 return buf_len; | 96 return buf_len; |
97 } | 97 } |
98 | 98 |
99 int LogStoreEntry::WriteData(int index, int offset, net::IOBuffer* buf, | 99 int LogStoreEntry::WriteData(int index, int offset, net::IOBuffer* buf, |
(...skipping 15 matching lines...) Expand all Loading... |
115 } | 115 } |
116 stream.size = new_size; | 116 stream.size = new_size; |
117 return buf_len; | 117 return buf_len; |
118 } | 118 } |
119 | 119 |
120 void LogStoreEntry::Delete() { | 120 void LogStoreEntry::Delete() { |
121 DCHECK(init_ && !closed_); | 121 DCHECK(init_ && !closed_); |
122 deleted_ = true; | 122 deleted_ = true; |
123 } | 123 } |
124 | 124 |
125 bool LogStoreEntry::ReadOnly() const { | 125 bool LogStoreEntry::IsNew() const { |
126 return id_ != -1; | 126 return id_ == -1; |
127 } | 127 } |
128 | 128 |
129 bool LogStoreEntry::InvalidStream(int stream_index) const { | 129 bool LogStoreEntry::InvalidStream(int stream_index) const { |
130 return stream_index < 0 || stream_index >= kFlashLogStoreEntryNumStreams; | 130 return stream_index < 0 || stream_index >= kFlashLogStoreEntryNumStreams; |
131 } | 131 } |
132 | 132 |
133 int32 LogStoreEntry::Size() const { | 133 int32 LogStoreEntry::Size() const { |
134 DCHECK(init_); | 134 DCHECK(init_); |
135 int32 size = kFlashLogStoreEntryHeaderSize; | 135 int32 size = kFlashLogStoreEntryHeaderSize; |
136 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) | 136 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) |
137 size += streams_[i].size; | 137 size += streams_[i].size; |
138 DCHECK(size > 0 && size <= kFlashSegmentFreeSpace); | 138 DCHECK(size > 0 && size <= kFlashSegmentFreeSpace); |
139 return size; | 139 return size; |
140 } | 140 } |
141 | 141 |
142 bool LogStoreEntry::Save() { | 142 bool LogStoreEntry::Save() { |
143 DCHECK(init_ && !closed_ && !deleted_ && !ReadOnly()); | 143 DCHECK(init_ && !closed_ && !deleted_ && IsNew()); |
144 int32 stream_sizes[kFlashLogStoreEntryNumStreams]; | 144 int32 stream_sizes[kFlashLogStoreEntryNumStreams]; |
145 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashLogStoreEntryHeaderSize, | 145 COMPILE_ASSERT(sizeof(stream_sizes) == kFlashLogStoreEntryHeaderSize, |
146 invalid_log_store_entry_header_size); | 146 invalid_log_store_entry_header_size); |
147 | 147 |
148 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) | 148 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) |
149 stream_sizes[i] = streams_[i].size; | 149 stream_sizes[i] = streams_[i].size; |
150 | 150 |
151 if (!store_->CreateEntry(Size(), &id_)) | 151 if (!store_->CreateEntry(Size(), &id_)) |
152 return false; | 152 return false; |
153 if (!store_->WriteData(stream_sizes, kFlashLogStoreEntryHeaderSize)) | 153 if (!store_->WriteData(stream_sizes, kFlashLogStoreEntryHeaderSize)) |
154 return false; | 154 return false; |
155 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) { | 155 for (int i = 0; i < kFlashLogStoreEntryNumStreams; ++i) { |
156 if (streams_[i].size > 0 && | 156 if (streams_[i].size > 0 && |
157 !store_->WriteData(&streams_[i].write_buffer[0], streams_[i].size)) { | 157 !store_->WriteData(&streams_[i].write_buffer[0], streams_[i].size)) { |
158 return false; | 158 return false; |
159 } | 159 } |
160 } | 160 } |
161 store_->CloseEntry(id_); | 161 store_->CloseEntry(id_); |
162 return true; | 162 return true; |
163 } | 163 } |
164 | 164 |
165 LogStoreEntry::Stream::Stream() : offset(0), size(0) { | 165 LogStoreEntry::Stream::Stream() : offset(0), size(0) { |
166 } | 166 } |
167 | 167 |
168 LogStoreEntry::Stream::~Stream() { | 168 LogStoreEntry::Stream::~Stream() { |
169 } | 169 } |
170 | 170 |
171 } // namespace disk_cache | 171 } // namespace disk_cache |
OLD | NEW |