IndexedDB: Make getAll() requests fail if result exceeds IPC limits

content/browser/indexed_db/mock_browsertest_indexed_db_class_factory.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <string>
 
 #include "base/logging.h"
 #include "content/browser/indexed_db/indexed_db_transaction.h"
 #include "content/browser/indexed_db/leveldb/leveldb_iterator_impl.h"
 #include "content/browser/indexed_db/leveldb/leveldb_transaction.h"
@@ skipping 23 matching lines @@
   std::string class_name_;
   std::string method_name_;
   int instance_count_;
   int current_call_num_;
 };
 
 }  // namespace
 
 namespace content {
 
+class IndexedDBTestDatabase : public IndexedDBDatabase {
+ public:
+  IndexedDBTestDatabase(const base::string16& name,
+                        IndexedDBBackingStore* backing_store,
+                        IndexedDBFactory* factory,
+                        const IndexedDBDatabase::Identifier& unique_identifier)
+      : IndexedDBDatabase(name, backing_store, factory, unique_identifier) {}
+
+ protected:
+  ~IndexedDBTestDatabase() override {}
+
+  size_t GetMaxMessageSizeInBytes() const override {
+    return 10 * 1024 * 1024;  // 10MB
+  }
+};
+
 class IndexedDBTestTransaction : public IndexedDBTransaction {
  public:
   IndexedDBTestTransaction(
       int64 id,
       scoped_refptr<IndexedDBDatabaseCallbacks> callbacks,
       const std::set<int64>& scope,
       blink::WebIDBTransactionMode mode,
       IndexedDBDatabase* db,
       IndexedDBBackingStore::Transaction* backing_store_transaction)
       : IndexedDBTransaction(id,
@@ skipping 173 matching lines @@
 
 MockBrowserTestIndexedDBClassFactory::MockBrowserTestIndexedDBClassFactory()
     : failure_class_(FAIL_CLASS_NOTHING),
       failure_method_(FAIL_METHOD_NOTHING),
       only_trace_calls_(false) {
 }
 
 MockBrowserTestIndexedDBClassFactory::~MockBrowserTestIndexedDBClassFactory() {
 }
 
+IndexedDBDatabase*
+MockBrowserTestIndexedDBClassFactory::CreateIndexedDBDatabase(
+    const base::string16& name,
+    IndexedDBBackingStore* backing_store,
+    IndexedDBFactory* factory,
+    const IndexedDBDatabase::Identifier& unique_identifier) {
+  return new IndexedDBTestDatabase(name, backing_store, factory,
+                                   unique_identifier);
+}
+
 IndexedDBTransaction*
 MockBrowserTestIndexedDBClassFactory::CreateIndexedDBTransaction(
     int64 id,
     scoped_refptr<IndexedDBDatabaseCallbacks> callbacks,
     const std::set<int64>& scope,
     blink::WebIDBTransactionMode mode,
     IndexedDBDatabase* db,
     IndexedDBBackingStore::Transaction* backing_store_transaction) {
   return new IndexedDBTestTransaction(id, callbacks, scope, mode, db,
                                       backing_store_transaction);
@@ skipping 62 matching lines @@
 
 void MockBrowserTestIndexedDBClassFactory::Reset() {
   failure_class_ = FAIL_CLASS_NOTHING;
   failure_method_ = FAIL_METHOD_NOTHING;
   instance_count_.clear();
   fail_on_instance_num_.clear();
   fail_on_call_num_.clear();
 }
 
 }  // namespace content