| 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 #ifndef NET_DISK_CACHE_SPARSE_CONTROL_H_ | 5 #ifndef NET_DISK_CACHE_V3_SPARSE_CONTROL_V3_H_ |
| 6 #define NET_DISK_CACHE_SPARSE_CONTROL_H_ | 6 #define NET_DISK_CACHE_V3_SPARSE_CONTROL_V3_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 12 #include "base/compiler_specific.h" | 12 #include "base/compiler_specific.h" |
| 13 #include "net/base/completion_callback.h" | 13 #include "net/base/completion_callback.h" |
| 14 #include "net/disk_cache/bitmap.h" | 14 #include "net/disk_cache/bitmap.h" |
| 15 #include "net/disk_cache/disk_format.h" | 15 #include "net/disk_cache/disk_format_base.h" |
| 16 | 16 |
| 17 namespace net { | 17 namespace net { |
| 18 class IOBuffer; | 18 class IOBuffer; |
| 19 class DrainableIOBuffer; | 19 class DrainableIOBuffer; |
| 20 } | 20 } |
| 21 | 21 |
| 22 namespace disk_cache { | 22 namespace disk_cache { |
| 23 | 23 |
| 24 class Entry; | 24 class Entry; |
| 25 class EntryImpl; | 25 class EntryImplV3; |
| 26 | 26 |
| 27 // This class provides support for the sparse capabilities of the disk cache. | 27 // This class provides support for the sparse capabilities of the disk cache. |
| 28 // Basically, sparse IO is directed from EntryImpl to this class, and we split | 28 // Basically, sparse IO is directed from EntryImpl to this class, and we split |
| 29 // the operation into multiple small pieces, sending each one to the | 29 // the operation into multiple small pieces, sending each one to the |
| 30 // appropriate entry. An instance of this class is asociated with each entry | 30 // appropriate entry. An instance of this class is asociated with each entry |
| 31 // used directly for sparse operations (the entry passed in to the constructor). | 31 // used directly for sparse operations (the entry passed in to the constructor). |
| 32 class SparseControl { | 32 class SparseControlV3 { |
| 33 public: | 33 public: |
| 34 typedef net::CompletionCallback CompletionCallback; | 34 typedef net::CompletionCallback CompletionCallback; |
| 35 | 35 |
| 36 // The operation to perform. | 36 // The operation to perform. |
| 37 enum SparseOperation { | 37 enum SparseOperation { |
| 38 kNoOperation, | 38 kNoOperation, |
| 39 kReadOperation, | 39 kReadOperation, |
| 40 kWriteOperation, | 40 kWriteOperation, |
| 41 kGetRangeOperation | 41 kGetRangeOperation |
| 42 }; | 42 }; |
| 43 | 43 |
| 44 explicit SparseControl(EntryImpl* entry); | 44 explicit SparseControlV3(EntryImplV3* entry); |
| 45 ~SparseControl(); | 45 ~SparseControlV3(); |
| 46 |
| 47 void Close(); |
| 46 | 48 |
| 47 // Performs a quick test to see if the entry is sparse or not, without | 49 // Performs a quick test to see if the entry is sparse or not, without |
| 48 // generating disk IO (so the answer provided is only a best effort). | 50 // generating disk IO (so the answer provided is only a best effort). |
| 49 bool CouldBeSparse() const; | 51 bool CouldBeSparse() const; |
| 50 | 52 |
| 51 // Performs an actual sparse read or write operation for this entry. |op| is | 53 // Performs an actual sparse read or write operation for this entry. |op| is |
| 52 // the operation to perform, |offset| is the desired sparse offset, |buf| and | 54 // the operation to perform, |offset| is the desired sparse offset, |buf| and |
| 53 // |buf_len| specify the actual data to use and |callback| is the callback | 55 // |buf_len| specify the actual data to use and |callback| is the callback |
| 54 // to use for asynchronous operations. See the description of the Read / | 56 // to use for asynchronous operations. See the description of the Read / |
| 55 // WriteSparseData for details about the arguments. The return value is the | 57 // WriteSparseData for details about the arguments. The return value is the |
| 56 // number of bytes read or written, or a net error code. | 58 // number of bytes read or written, or a net error code. |
| 57 int StartIO(SparseOperation op, int64 offset, net::IOBuffer* buf, | 59 int StartIO(SparseOperation op, int64 offset, net::IOBuffer* buf, |
| 58 int buf_len, const CompletionCallback& callback); | 60 int buf_len, const CompletionCallback& callback); |
| 59 | 61 |
| 60 // Implements Entry::GetAvailableRange(). | 62 // Implements Entry::GetAvailableRange(). |
| 61 int GetAvailableRange(int64 offset, int len, int64* start); | 63 int GetAvailableRange(int64 offset, int len, int64* start, |
| 64 const CompletionCallback& callback); |
| 62 | 65 |
| 63 // Cancels the current sparse operation (if any). | 66 // Cancels the current sparse operation (if any). |
| 64 void CancelIO(); | 67 void CancelIO(); |
| 65 | 68 |
| 66 // Returns OK if the entry can be used for new IO or ERR_IO_PENDING if we are | 69 // Returns OK if the entry can be used for new IO or ERR_IO_PENDING if we are |
| 67 // busy. If the entry is busy, we'll invoke the callback when we are ready | 70 // busy. If the entry is busy, we'll invoke the callback when we are ready |
| 68 // again. See disk_cache::Entry::ReadyToUse() for more info. | 71 // again. See disk_cache::Entry::ReadyToUse() for more info. |
| 69 int ReadyToUse(const CompletionCallback& completion_callback); | 72 int ReadyToUse(const CompletionCallback& completion_callback); |
| 70 | 73 |
| 71 // Deletes the children entries of |entry|. | 74 // Deletes the children entries of |entry|. |
| 72 static void DeleteChildren(EntryImpl* entry); | 75 static void DeleteChildren(EntryImplV3* entry); |
| 73 | 76 |
| 74 private: | 77 private: |
| 75 // Initializes the object for the current entry. If this entry already stores | 78 enum State { |
| 76 // sparse data, or can be used to do it, it updates the relevant information | 79 STATE_NONE, |
| 77 // on disk and returns net::OK. Otherwise it returns a net error code. | 80 STATE_INIT, |
| 78 int Init(); | 81 STATE_CREATE_SPARSE_ENTRY, |
| 82 STATE_CREATE_SPARSE_ENTRY_COMPLETE, |
| 83 STATE_OPEN_SPARSE_ENTRY, |
| 84 STATE_OPEN_SPARSE_ENTRY_COMPLETE, |
| 85 STATE_READ_BITMAP_COMPLETE, |
| 86 STATE_GET_CHILD_KEY, |
| 87 STATE_OPEN_CHILD, |
| 88 STATE_OPEN_CHILD_COMPLETE, |
| 89 STATE_READ_SIGNATURE_COMPLETE, |
| 90 STATE_CLOSE_CHILD, |
| 91 STATE_CLOSE_CHILD_COMPLETE, |
| 92 STATE_CREATE_CHILD, |
| 93 STATE_CREATE_CHILD_COMPLETE, |
| 94 STATE_WRITE_BITMAP, |
| 95 STATE_WRITE_BITMAP_COMPLETE, |
| 96 STATE_DO_CHILD_IO, |
| 97 STATE_DO_CHILD_IO_COMPLETE, |
| 98 STATE_CLOSE |
| 99 }; |
| 79 | 100 |
| 80 // Creates a new sparse entry or opens an aready created entry from disk. | 101 int DoLoop(int result); |
| 81 // These methods just read / write the required info from disk for the current | 102 int DoInit(); |
| 82 // entry, and verify that everything is correct. The return value is a net | 103 int DoCreateSparseEntry(); |
| 83 // error code. | 104 int DoCreateSparseEntryComplete(int result); |
| 84 int CreateSparseEntry(); | 105 int DoOpenSparseEntry(int data_len); |
| 85 int OpenSparseEntry(int data_len); | 106 int DoOpenSparseEntryComplete(int result); |
| 86 | 107 int DoReadBitmapComplete(int result); |
| 87 // Opens and closes a child entry. A child entry is a regular EntryImpl object | 108 int DoGetChildKey(); |
| 88 // with a key derived from the key of the resource to store and the range | 109 int DoOpenChild(); |
| 89 // stored by that child. | 110 int DoOpenChildComplete(int result); |
| 90 bool OpenChild(); | 111 int DoReadSignatureComplete(int result); |
| 91 void CloseChild(); | 112 int DoCloseChild(); |
| 92 | 113 int DoCloseChildComplete(int result); |
| 93 // Continues the current operation (open) without a current child. | 114 int DoCreateChild(); |
| 94 bool ContinueWithoutChild(const std::string& key); | 115 int DoCreateChildComplete(int result); |
| 95 | 116 int DoWriteBitmap(); |
| 96 // Writes to disk the tracking information for this entry. | 117 int DoWriteBitmapComplete(int result); |
| 97 void WriteSparseData(); | 118 int DoChildIO(); |
| 98 | 119 int DoChildIOComplete(int result); |
| 99 // Performs a single operation with the current child. Returns true when we | 120 int DoClose(); |
| 100 // should move on to the next child and false when we should interrupt our | |
| 101 // work. | |
| 102 bool DoChildIO(); | |
| 103 | |
| 104 // Performs the required work after a single IO operations finishes. | |
| 105 void DoChildIOCompleted(int result); | |
| 106 | 121 |
| 107 std::string GenerateChildKey(); | 122 std::string GenerateChildKey(); |
| 108 | 123 |
| 109 // Deletes the current child and continues the current operation (open). | 124 // Deletes the current child and continues the current operation (open). |
| 110 bool KillChildAndContinue(const std::string& key, bool fatal); | 125 int KillChildAndContinue(); |
| 111 | 126 |
| 112 // Returns true if the required child is tracked by the parent entry, i.e. it | 127 // Returns true if the required child is tracked by the parent entry, i.e. it |
| 113 // was already created. | 128 // was already created. |
| 114 bool ChildPresent(); | 129 bool ChildPresent(); |
| 115 | 130 |
| 116 // Sets the bit for the current child to the provided |value|. In other words, | 131 // Sets the bit for the current child to the provided |value|. In other words, |
| 117 // starts or stops tracking this child. | 132 // starts or stops tracking this child. |
| 118 void SetChildBit(bool value); | 133 void SetChildBit(bool value); |
| 119 | 134 |
| 120 // Verify that the range to be accessed for the current child is appropriate. | 135 // Verify that the range to be accessed for the current child is appropriate. |
| 121 // Returns false if an error is detected or there is no need to perform the | 136 // Returns false if an error is detected or there is no need to perform the |
| 122 // current IO operation (for instance if the required range is not stored by | 137 // current IO operation (for instance if the required range is not stored by |
| 123 // the child). | 138 // the child). |
| 124 bool VerifyRange(); | 139 bool VerifyRange(); |
| 125 | 140 |
| 126 // Updates the contents bitmap for the current range, based on the result of | 141 // Updates the contents bitmap for the current range, based on the result of |
| 127 // the current operation. | 142 // the current operation. |
| 128 void UpdateRange(int result); | 143 void UpdateRange(int result); |
| 129 | 144 |
| 130 // Returns the number of bytes stored at |block_index|, if its allocation-bit | 145 // Returns the number of bytes stored at |block_index|, if its allocation-bit |
| 131 // is off (because it is not completely filled). | 146 // is off (because it is not completely filled). |
| 132 int PartialBlockLength(int block_index) const; | 147 int PartialBlockLength(int block_index) const; |
| 133 | 148 |
| 134 // Initializes the sparse info for the current child. | 149 // Initializes the sparse info for the current child. |
| 135 void InitChildData(); | 150 void InitChildData(); |
| 136 | 151 |
| 137 // Performs the required work for GetAvailableRange for one child. | 152 // Performs the required work for GetAvailableRange for one child. |
| 138 int DoGetAvailableRange(); | 153 int GetAvailableRangeImpl(); |
| 154 |
| 155 void LogChildOperationStart(); |
| 156 void LogChildOperationEnd(int result); |
| 157 int LogCompletion(int result); |
| 139 | 158 |
| 140 // Reports to the user that we are done. | 159 // Reports to the user that we are done. |
| 141 void DoUserCallback(); | 160 void HandleResult(int result); |
| 142 void DoAbortCallbacks(); | 161 void HanldeAbortCallbacks(); |
| 143 | 162 |
| 144 // Invoked by the callback of asynchronous operations. | 163 // Invoked by the callback of asynchronous operations. |
| 145 void OnChildIOCompleted(int result); | 164 void OnIOComplete(int result); |
| 146 | 165 |
| 147 EntryImpl* entry_; // The sparse entry. | 166 EntryImplV3* entry_; // The sparse entry. |
| 148 EntryImpl* child_; // The current child entry. | 167 Entry* child_; // The current child entry. |
| 149 SparseOperation operation_; | 168 SparseOperation operation_; |
| 150 bool pending_; // True if any child IO operation returned pending. | 169 State next_state_; |
| 151 bool finished_; | |
| 152 bool init_; | 170 bool init_; |
| 153 bool range_found_; // True if GetAvailableRange found something. | 171 bool range_found_; // True if GetAvailableRange found something. |
| 154 bool abort_; // True if we should abort the current operation ASAP. | 172 bool abort_; // True if we should abort the current operation ASAP. |
| 173 bool valid_; |
| 174 bool closing_; |
| 155 | 175 |
| 156 SparseHeader sparse_header_; // Data about the children of entry_. | 176 SparseHeader sparse_header_; // Data about the children of entry_. |
| 157 Bitmap children_map_; // The actual bitmap of children. | 177 Bitmap children_map_; // The actual bitmap of children. |
| 158 SparseData child_data_; // Parent and allocation map of child_. | 178 SparseData child_data_; // Parent and allocation map of child_. |
| 159 Bitmap child_map_; // The allocation map as a bitmap. | 179 Bitmap child_map_; // The allocation map as a bitmap. |
| 160 | 180 |
| 161 CompletionCallback user_callback_; | 181 CompletionCallback user_callback_; |
| 182 CompletionCallback callback_; |
| 162 std::vector<CompletionCallback> abort_callbacks_; | 183 std::vector<CompletionCallback> abort_callbacks_; |
| 163 int64 offset_; // Current sparse offset. | 184 int64 offset_; // Current sparse offset. |
| 164 scoped_refptr<net::DrainableIOBuffer> user_buf_; | 185 scoped_refptr<net::DrainableIOBuffer> user_buf_; |
| 186 scoped_refptr<net::IOBuffer> buf_; |
| 187 std::string key_; |
| 165 int buf_len_; // Bytes to read or write. | 188 int buf_len_; // Bytes to read or write. |
| 166 int child_offset_; // Offset to use for the current child. | 189 int child_offset_; // Offset to use for the current child. |
| 167 int child_len_; // Bytes to read or write for this child. | 190 int child_len_; // Bytes to read or write for this child. |
| 168 int result_; | 191 int result_; |
| 192 int64* range_start_; |
| 169 | 193 |
| 170 DISALLOW_COPY_AND_ASSIGN(SparseControl); | 194 DISALLOW_COPY_AND_ASSIGN(SparseControlV3); |
| 171 }; | 195 }; |
| 172 | 196 |
| 173 } // namespace disk_cache | 197 } // namespace disk_cache |
| 174 | 198 |
| 175 #endif // NET_DISK_CACHE_SPARSE_CONTROL_H_ | 199 #endif // NET_DISK_CACHE_V3_SPARSE_CONTROL_V3_H_ |
| OLD | NEW |