Handle out of memory in GrowableIOBuffer more gracefully....

net/websockets/websocket.cc

 // Copyright (c) 2009 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 <algorithm>
 #include <limits>
 
 #include "net/websockets/websocket.h"
 
 #include "base/message_loop.h"
@@ skipping 88 matching lines @@
   }
 }
 
 void WebSocket::OnConnected(SocketStream* socket_stream,
                             int max_pending_send_allowed) {
   DCHECK(socket_stream == socket_stream_);
   max_pending_send_allowed_ = max_pending_send_allowed;
 
   // Use |max_pending_send_allowed| as hint for initial size of read buffer.
   current_read_buf_ = new GrowableIOBuffer();
-  current_read_buf_->set_capacity(max_pending_send_allowed_);
+  current_read_buf_->SetCapacity(max_pending_send_allowed_);
   read_consumed_len_ = 0;
 
   DCHECK(!current_write_buf_);
   pending_write_bufs_.push_back(CreateClientHandshakeMessage());
   origin_loop_->PostTask(FROM_HERE,
                          NewRunnableMethod(this, &WebSocket::SendPending));
 }
 
 void WebSocket::OnSentData(SocketStream* socket_stream, int amount_sent) {
   DCHECK(socket_stream == socket_stream_);
@@ skipping 274 matching lines @@
       }
     }
   }
   SkipReadBuffer(next_frame - start_frame);
 }
 
 void WebSocket::AddToReadBuffer(const char* data, int len) {
   DCHECK(current_read_buf_);
   // Check if |current_read_buf_| has enough space to store |len| of |data|.
   if (len >= current_read_buf_->RemainingCapacity()) {
-    current_read_buf_->set_capacity(
+    current_read_buf_->SetCapacity(
         current_read_buf_->offset() + len);
   }
 
   DCHECK(current_read_buf_->RemainingCapacity() >= len);
   memcpy(current_read_buf_->data(), data, len);
   current_read_buf_->set_offset(current_read_buf_->offset() + len);
 }
 
 void WebSocket::SkipReadBuffer(int len) {
   if (len == 0)
@@ skipping 25 matching lines @@
   WebSocketDelegate* delegate = delegate_;
   delegate_ = NULL;
   ready_state_ = CLOSED;
   if (!socket_stream_)
     return;
   socket_stream_ = NULL;
   delegate->OnClose(this);
 }
 
 }  // namespace net