Fix out-of-memory crashes related to ArrayBuffer allocation

third_party/WebKit/Source/modules/websockets/DocumentWebSocketChannelTest.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 "config.h"
 #include "modules/websockets/WebSocketChannel.h"
 
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/Document.h"
 #include "core/fileapi/Blob.h"
@@ skipping 332 matching lines @@
 {
     connect();
     {
         InSequence s;
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("foo", 3), 3));
     }
 
     handleClient()->didReceiveFlowControl(handle(), 16);
     EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_)).Times(AnyNumber());
 
-    RefPtr<DOMArrayBuffer> fooBuffer = DOMArrayBuffer::create("foo", 3);
+    RefPtr<DOMArrayBuffer> fooBuffer = DOMArrayBuffer::deprecatedCreateOrCrash("foo", 3);
     channel()->send(*fooBuffer, 0, 3);
 
     EXPECT_EQ(3ul, m_sumOfConsumedBufferedAmount);
 }
 
 TEST_F(DocumentWebSocketChannelTest, sendBinaryInArrayBufferPartial)
 {
     connect();
     {
         InSequence s;
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("foo", 3), 3));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("bar", 3), 3));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("baz", 3), 3));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("a", 1), 1));
     }
 
     handleClient()->didReceiveFlowControl(handle(), 16);
     EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_)).Times(AnyNumber());
 
-    RefPtr<DOMArrayBuffer> foobarBuffer = DOMArrayBuffer::create("foobar", 6);
-    RefPtr<DOMArrayBuffer> qbazuxBuffer = DOMArrayBuffer::create("qbazux", 6);
+    RefPtr<DOMArrayBuffer> foobarBuffer = DOMArrayBuffer::deprecatedCreateOrCrash("foobar", 6);
+    RefPtr<DOMArrayBuffer> qbazuxBuffer = DOMArrayBuffer::deprecatedCreateOrCrash("qbazux", 6);
     channel()->send(*foobarBuffer, 0, 3);
     channel()->send(*foobarBuffer, 3, 3);
     channel()->send(*qbazuxBuffer, 1, 3);
     channel()->send(*qbazuxBuffer, 2, 1);
 
     EXPECT_EQ(10ul, m_sumOfConsumedBufferedAmount);
 }
 
 TEST_F(DocumentWebSocketChannelTest, sendBinaryInArrayBufferWithNullBytes)
 {
     connect();
     {
         InSequence s;
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("\0ar", 3), 3));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("b\0z", 3), 3));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("qu\0", 3), 3));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("\0\0\0", 3), 3));
     }
 
     handleClient()->didReceiveFlowControl(handle(), 16);
     EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_)).Times(AnyNumber());
 
     {
-        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("\0ar", 3);
+        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("\0ar", 3);
         channel()->send(*b, 0, 3);
     }
     {
-        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("b\0z", 3);
+        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("b\0z", 3);
         channel()->send(*b, 0, 3);
     }
     {
-        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("qu\0", 3);
+        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("qu\0", 3);
         channel()->send(*b, 0, 3);
     }
     {
-        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("\0\0\0", 3);
+        RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("\0\0\0", 3);
         channel()->send(*b, 0, 3);
     }
 
     EXPECT_EQ(12ul, m_sumOfConsumedBufferedAmount);
 }
 
 TEST_F(DocumentWebSocketChannelTest, sendBinaryInArrayBufferNonLatin1UTF8)
 {
     connect();
     EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("\xe7\x8b\x90", 3), 3));
 
     handleClient()->didReceiveFlowControl(handle(), 16);
     EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_)).Times(AnyNumber());
 
-    RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("\xe7\x8b\x90", 3);
+    RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("\xe7\x8b\x90", 3);
     channel()->send(*b, 0, 3);
 
     EXPECT_EQ(3ul, m_sumOfConsumedBufferedAmount);
 }
 
 TEST_F(DocumentWebSocketChannelTest, sendBinaryInArrayBufferNonUTF8)
 {
     connect();
     EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeBinary, MemEq("\x80\xff\xe7", 3), 3));
 
     handleClient()->didReceiveFlowControl(handle(), 16);
     EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_)).Times(AnyNumber());
 
-    RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("\x80\xff\xe7", 3);
+    RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("\x80\xff\xe7", 3);
     channel()->send(*b, 0, 3);
 
     EXPECT_EQ(3ul, m_sumOfConsumedBufferedAmount);
 }
 
 TEST_F(DocumentWebSocketChannelTest, sendBinaryInArrayBufferNonLatin1UTF8Continuation)
 {
     connect();
     Checkpoint checkpoint;
     {
         InSequence s;
         EXPECT_CALL(*handle(), send(false, WebSocketHandle::MessageTypeBinary, MemEq("\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7", 16), 16));
         EXPECT_CALL(checkpoint, Call(1));
         EXPECT_CALL(*handle(), send(true, WebSocketHandle::MessageTypeContinuation, MemEq("\x8b\x90", 2), 2));
     }
 
     handleClient()->didReceiveFlowControl(handle(), 16);
     EXPECT_CALL(*channelClient(), didConsumeBufferedAmount(_)).Times(AnyNumber());
 
-    RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::create("\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90", 18);
+    RefPtr<DOMArrayBuffer> b = DOMArrayBuffer::deprecatedCreateOrCrash("\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90\xe7\x8b\x90", 18);
     channel()->send(*b, 0, 18);
     checkpoint.Call(1);
 
     handleClient()->didReceiveFlowControl(handle(), 16);
 
     EXPECT_EQ(18ul, m_sumOfConsumedBufferedAmount);
 }
 
 // FIXME: Add tests for WebSocketChannel::send(PassRefPtr<BlobDataHandle>)
 
@@ skipping 211 matching lines @@
         EXPECT_CALL(*channelClient(), didError());
         EXPECT_CALL(*channelClient(), didClose(WebSocketChannelClient::ClosingHandshakeIncomplete, WebSocketChannel::CloseEventCodeAbnormalClosure, String()));
     }
 
     channel()->fail("fail message from WebSocket", ErrorMessageLevel, "sourceURL", 1234);
 }
 
 } // namespace
 
 } // namespace blink