Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(505)

Unified Diff: webkit/blob/blob_url_request_job.cc

Issue 12047012: Avoid integer overflows in BlobURLRequestJob. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: use kint64max Created 7 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webkit/blob/blob_url_request_job.h ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webkit/blob/blob_url_request_job.cc
diff --git a/webkit/blob/blob_url_request_job.cc b/webkit/blob/blob_url_request_job.cc
index ae3dd24c9961c6835c3159b1276ca28932b5df71..c9aefe43f2e87d842a4cd69984229fbd1a274d1e 100644
--- a/webkit/blob/blob_url_request_job.cc
+++ b/webkit/blob/blob_url_request_job.cc
@@ -4,6 +4,9 @@
#include "webkit/blob/blob_url_request_job.h"
+#include <limits>
+
+#include "base/basictypes.h"
#include "base/bind.h"
#include "base/compiler_specific.h"
#include "base/files/file_util_proxy.h"
@@ -183,6 +186,19 @@ void BlobURLRequestJob::DidStart() {
CountSize();
}
+bool BlobURLRequestJob::AddItemLength(size_t index, int64 item_length) {
+ if (item_length > kint64max - total_size_) {
+ NotifyFailure(net::ERR_FAILED);
+ return false;
+ }
+
+ // Cache the size and add it to the total size.
+ DCHECK_LT(index, item_length_list_.size());
+ item_length_list_[index] = item_length;
+ total_size_ += item_length;
+ return true;
+}
+
void BlobURLRequestJob::CountSize() {
error_ = false;
pending_get_file_info_count_ = 0;
@@ -198,10 +214,9 @@ void BlobURLRequestJob::CountSize() {
weak_factory_.GetWeakPtr(), i));
continue;
}
- // Cache the size and add it to the total size.
- int64 item_length = static_cast<int64>(item.length());
- item_length_list_[i] = item_length;
- total_size_ += item_length;
+
+ if (!AddItemLength(i, item.length()))
+ return;
}
if (pending_get_file_info_count_ == 0)
@@ -251,16 +266,28 @@ void BlobURLRequestJob::DidGetFileItemLength(size_t index, int64 result) {
const BlobData::Item& item = blob_data_->items().at(index);
DCHECK(IsFileType(item.type()));
+ uint64 file_length = result;
+ uint64 item_offset = item.offset();
+ uint64 item_length = item.length();
+
+ if (item_offset > file_length) {
+ NotifyFailure(net::ERR_FILE_NOT_FOUND);
+ return;
+ }
+
+ uint64 max_length = file_length - item_offset;
+
// If item length is -1, we need to use the file size being resolved
// in the real time.
- int64 item_length = static_cast<int64>(item.length());
- if (item_length == -1)
- item_length = result - item.offset();
+ if (item_length == static_cast<uint64>(-1)) {
+ item_length = max_length;
+ } else if (item_length > max_length) {
+ NotifyFailure(net::ERR_FILE_NOT_FOUND);
+ return;
+ }
- // Cache the size and add it to the total size.
- DCHECK_LT(index, item_length_list_.size());
- item_length_list_[index] = item_length;
- total_size_ += item_length;
+ if (!AddItemLength(index, item_length))
+ return;
if (--pending_get_file_info_count_ == 0)
DidCountSize(net::OK);
@@ -420,14 +447,18 @@ int BlobURLRequestJob::BytesReadCompleted() {
}
int BlobURLRequestJob::ComputeBytesToRead() const {
- int64 current_item_remaining_bytes =
- item_length_list_[current_item_index_] - current_item_offset_;
- int64 remaining_bytes = std::min(current_item_remaining_bytes,
- remaining_bytes_);
-
- return static_cast<int>(std::min(
- static_cast<int64>(read_buf_->BytesRemaining()),
- remaining_bytes));
+ int64 current_item_length = item_length_list_[current_item_index_];
+
+ int64 item_remaining = current_item_length - current_item_offset_;
+ int64 buf_remaining = read_buf_->BytesRemaining();
+ int64 max_remaining = std::numeric_limits<int>::max();
+
+ int64 min = std::min(std::min(std::min(item_remaining,
+ buf_remaining),
+ remaining_bytes_),
+ max_remaining);
+
+ return static_cast<int>(min);
}
bool BlobURLRequestJob::ReadLoop(int* bytes_read) {
« no previous file with comments | « webkit/blob/blob_url_request_job.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698