| 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 |