Wrap large IndexedDB values into Blobs before writing to LevelDB.

third_party/WebKit/Source/modules/indexeddb/IDBValueWrappingTest.cpp

+// Copyright 2014 The Chromium Authors. All rights reserved.

[jsbell, 2017/05/25 18:49:42]

nit: 2017

[pwnall, 2017/05/25 20:01:11]

Done.
Thank you for catching!

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "modules/indexeddb/IDBValueWrapping.h"
+
+#include "bindings/core/v8/V8BindingForTesting.h"
+#include "modules/indexeddb/IDBValue.h"
+#include "platform/wtf/PtrUtil.h"
+#include "platform/wtf/RefPtr.h"
+#include "platform/wtf/Vector.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "v8/include/v8.h"
+
+namespace blink {
+
+TEST(IDBValueUnwrapperTest, IsWrapped) {
+  V8TestingScope scope;
+  NonThrowableExceptionState non_throwable_exception_state;
+  v8::Local<v8::Value> v8_true = v8::True(scope.GetIsolate());
+  IDBValueWrapper wrapper(scope.GetIsolate(), v8_true,
+                          SerializedScriptValue::SerializeOptions::kSerialize,
+                          non_throwable_exception_state);
+  wrapper.WrapIfBiggerThan(0);
+  Vector<RefPtr<BlobDataHandle>> blob_data_handles;
+  wrapper.ExtractBlobDataHandles(&blob_data_handles);
+  Vector<WebBlobInfo>& blob_infos = wrapper.WrappedBlobInfo();
+  RefPtr<SharedBuffer> wrapped_marker_buffer = wrapper.ExtractWireBytes();
+
+  RefPtr<IDBValue> wrapped_value = IDBValue::Create(
+      wrapped_marker_buffer,
+      WTF::MakeUnique<Vector<RefPtr<BlobDataHandle>>>(blob_data_handles),
+      WTF::MakeUnique<Vector<WebBlobInfo>>(blob_infos));
+  EXPECT_TRUE(IDBValueUnwrapper::IsWrapped(wrapped_value.Get()));
+
+  Vector<char> wrapped_marker_bytes(wrapped_marker_buffer->size());
+  wrapped_marker_buffer->GetAsBytes(wrapped_marker_bytes.data(),
+                                    wrapped_marker_bytes.size());
+
+  // IsWrapped() looks at the first 3 bytes in the value's byte array.
+  // Truncating the array to fewer than 3 bytes should cause IsWrapped() to
+  // return false.
+  ASSERT_LT(3U, wrapped_marker_bytes.size());
+  for (size_t i = 0; i < 3; ++i) {
+    RefPtr<IDBValue> mutant_value = IDBValue::Create(
+        SharedBuffer::Create(wrapped_marker_bytes.data(), i),
+        WTF::MakeUnique<Vector<RefPtr<BlobDataHandle>>>(blob_data_handles),
+        WTF::MakeUnique<Vector<WebBlobInfo>>(blob_infos));
+
+    EXPECT_FALSE(IDBValueUnwrapper::IsWrapped(mutant_value.Get()));
+  }
+
+  // IsWrapped() looks at the first 3 bytes in the value. Flipping any bit in
+  // these 3 bytes should cause IsWrapped() to return false.
+  ASSERT_LT(3U, wrapped_marker_bytes.size());
+  for (size_t i = 0; i < 3; ++i) {
+    for (int j = 0; j < 8; ++j) {
+      char mask = 1 << j;
+      wrapped_marker_bytes[i] ^= mask;
+      RefPtr<IDBValue> mutant_value = IDBValue::Create(
+          SharedBuffer::Create(wrapped_marker_bytes.data(),
+                               wrapped_marker_bytes.size()),
+          WTF::MakeUnique<Vector<RefPtr<BlobDataHandle>>>(blob_data_handles),
+          WTF::MakeUnique<Vector<WebBlobInfo>>(blob_infos));
+      EXPECT_FALSE(IDBValueUnwrapper::IsWrapped(mutant_value.Get()));

[jsbell, 2017/05/25 18:49:42]

All of the IsWrapped tests are either TRUE because it is wrapped, or false
because the value is "invalid" (artificial and too short, or corrupted). It
seems like we're missing the case where it's valid but not wrapped?

[pwnall, 2017/05/25 20:01:10]

Yes, I need more tests.

I prioritized these tests because I was most worried about accessing memory out
of bounds, and the layout tests have decent coverage for valid unwrapped values.
I plan to add more tests, and I'd much rather do that in follow-ups. This CL is
already too big (to fail?)

+
+      wrapped_marker_bytes[i] ^= mask;
+    }
+  }
+}
+
+}  // namespace blink