[wasm] support for recompilation if deserialization fails

third_party/WebKit/Source/bindings/core/v8/ScriptValueSerializer.cpp

 // 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 "bindings/core/v8/ScriptValueSerializer.h"
 
 #include "bindings/core/v8/Transferables.h"
 #include "bindings/core/v8/V8ArrayBuffer.h"
 #include "bindings/core/v8/V8ArrayBufferView.h"
 #include "bindings/core/v8/V8Blob.h"
@@ skipping 90 matching lines @@
 }
 
 void SerializedScriptValueWriter::writeFalse() {
   append(FalseTag);
 }
 
 void SerializedScriptValueWriter::writeBooleanObject(bool value) {
   append(value ? TrueObjectTag : FalseObjectTag);
 }
 
+void SerializedScriptValueWriter::writeRawStringBytes(
+    v8::Local<v8::String>& string) {
+  int rawLength = string->Length();
+  string->WriteOneByte(byteAt(m_position), 0, rawLength,
+                       v8StringWriteOptions());
+  m_position += rawLength;
+}
+
+void SerializedScriptValueWriter::writeUtf8String(
+    v8::Local<v8::String>& string) {
+  int utf8Length = string->Utf8Length();
+  char* buffer = reinterpret_cast<char*>(byteAt(m_position));
+  string->WriteUtf8(buffer, utf8Length, 0, v8StringWriteOptions());
+  m_position += utf8Length;
+}
+
 void SerializedScriptValueWriter::writeOneByteString(
     v8::Local<v8::String>& string) {
   int stringLength = string->Length();
   int utf8Length = string->Utf8Length();
   ASSERT(stringLength >= 0 && utf8Length >= 0);
 
   append(StringTag);
   doWriteUint32(static_cast<uint32_t>(utf8Length));
   ensureSpace(utf8Length);
 
   // ASCII fast path.
   if (stringLength == utf8Length) {
-    string->WriteOneByte(byteAt(m_position), 0, utf8Length,
-                         v8StringWriteOptions());
+    writeRawStringBytes(string);
   } else {
-    char* buffer = reinterpret_cast<char*>(byteAt(m_position));
-    string->WriteUtf8(buffer, utf8Length, 0, v8StringWriteOptions());
+    writeUtf8String(string);
   }
-  m_position += utf8Length;
 }
 
 void SerializedScriptValueWriter::writeUCharString(
     v8::Local<v8::String>& string) {
   int length = string->Length();
   ASSERT(length >= 0);
 
   int size = length * sizeof(UChar);
   int bytes = bytesNeededToWireEncode(static_cast<uint32_t>(size));
   if ((m_position + 1 + bytes) & 1)
@@ skipping 1109 matching lines @@
   return nullptr;
 }
 
 ScriptValueSerializer::StateBase*
 ScriptValueSerializer::writeWasmCompiledModule(v8::Local<v8::Object> object,
                                                StateBase* next) {
   CHECK(RuntimeEnabledFeatures::webAssemblySerializationEnabled());
   // TODO (mtrofin): explore mechanism avoiding data copying / buffer resizing.
   v8::Local<v8::WasmCompiledModule> wasmModule =
       object.As<v8::WasmCompiledModule>();
+  v8::Local<v8::String> wireBytes = wasmModule->GetWasmWireBytes();
+  DCHECK(wireBytes->IsOneByte());
+
   v8::WasmCompiledModule::SerializedModule data = wasmModule->Serialize();
   m_writer.append(WasmModuleTag);
+  uint32_t wireBytesLength = static_cast<uint32_t>(wireBytes->Length());
+  // We place a tag so we may evolve the format in which we store the
+  // wire bytes. We plan to move them to a blob.
+  // We want to control how we write the string, though, so we explicitly
+  // call writeRawStringBytes.
+  m_writer.append(RawBytesTag);
+  m_writer.doWriteUint32(wireBytesLength);
+  m_writer.ensureSpace(wireBytesLength);
+  m_writer.writeRawStringBytes(wireBytes);
   m_writer.doWriteUint32(static_cast<uint32_t>(data.second));
   m_writer.append(data.first.get(), static_cast<int>(data.second));
   return nullptr;
 }
 
 ScriptValueSerializer::StateBase*
 ScriptValueSerializer::writeAndGreyArrayBuffer(v8::Local<v8::Object> object,
                                                StateBase* next) {
   DOMArrayBuffer* arrayBuffer = V8ArrayBuffer::toImpl(object);
   if (!arrayBuffer)
@@ skipping 675 matching lines @@
   if (m_position + byteLength > m_length)
     return nullptr;
   const void* bufferStart = m_buffer + m_position;
   m_position += byteLength;
   return DOMArrayBuffer::create(bufferStart, byteLength);
 }
 
 bool SerializedScriptValueReader::readWasmCompiledModule(
     v8::Local<v8::Value>* value) {
   CHECK(RuntimeEnabledFeatures::webAssemblySerializationEnabled());
-  uint32_t size = 0;
-  if (!doReadUint32(&size))
+  // First, read the tag of the wire bytes.
+  SerializationTag wireBytesFormat = InvalidTag;
+  if (!readTag(&wireBytesFormat))
     return false;
-  if (m_position + size > m_length)
+  DCHECK(wireBytesFormat == RawBytesTag);
+  // Just like when writing, we don't rely on the default string serialization
+  // mechanics for the wire bytes. We don't even want a string, because
+  // that would lead to a memory copying API implementation on the V8 side.
+  uint32_t wireBytesSize = 0;
+  uint32_t compiledBytesSize = 0;
+  if (!doReadUint32(&wireBytesSize))
     return false;
-  const uint8_t* buf = m_buffer + m_position;
-  // TODO(mtrofin): simplify deserializer API. const uint8_t* + size_t should
-  // be sufficient.
-  v8::WasmCompiledModule::SerializedModule data = {
-      std::unique_ptr<const uint8_t[]>(buf), static_cast<size_t>(size)};
+  if (m_position + wireBytesSize > m_length)
+    return false;
+  const uint8_t* wireBytesStart = m_buffer + m_position;
+  m_position += wireBytesSize;
+
+  if (!doReadUint32(&compiledBytesSize))
+    return false;
+  if (m_position + compiledBytesSize > m_length)
+    return false;
+  const uint8_t* compiledBytesStart = m_buffer + m_position;
+  m_position += compiledBytesSize;
+
+  v8::WasmCompiledModule::CallerOwnedBuffer wireBytes = {
+      wireBytesStart, static_cast<size_t>(wireBytesSize)};
+
+  v8::WasmCompiledModule::CallerOwnedBuffer compiledBytes = {
+      compiledBytesStart, static_cast<size_t>(compiledBytesSize)};
+
   v8::MaybeLocal<v8::WasmCompiledModule> retval =
-      v8::WasmCompiledModule::Deserialize(isolate(), data);
-  data.first.release();
-  m_position += size;
+      v8::WasmCompiledModule::DeserializeOrCompile(isolate(), compiledBytes,
+                                                   wireBytes);
 
-  // TODO(mtrofin): right now, we'll return undefined if the deserialization
-  // fails, which is what may happen when v8's version changes. Update when
-  // spec settles. crbug.com/639090
   return retval.ToLocal(value);
 }
 
 bool SerializedScriptValueReader::readArrayBuffer(v8::Local<v8::Value>* value) {
   DOMArrayBuffer* arrayBuffer = doReadArrayBuffer();
   if (!arrayBuffer)
     return false;
   *value = toV8(arrayBuffer, m_scriptState->context()->Global(), isolate());
   return !value->IsEmpty();
 }
@@ skipping 651 matching lines @@
       m_openCompositeReferenceStack[m_openCompositeReferenceStack.size() - 1];
   m_openCompositeReferenceStack.shrink(m_openCompositeReferenceStack.size() -
                                        1);
   if (objectReference >= m_objectPool.size())
     return false;
   *object = m_objectPool[objectReference];
   return true;
 }
 
 }  // namespace blink