Improve socket.write() to make sure all data will be sent out.

chrome/browser/extensions/api/socket/tcp_socket.h

 // Copyright (c) 2012 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.
 
 #ifndef CHROME_BROWSER_EXTENSIONS_API_SOCKET_TCP_SOCKET_H_
 #define CHROME_BROWSER_EXTENSIONS_API_SOCKET_TCP_SOCKET_H_
 #pragma once
 
 #include <string>
 
@@ skipping 16 matching lines @@
   explicit TCPSocket(APIResourceEventNotifier* event_notifier);
   virtual ~TCPSocket();
 
   virtual void Connect(const std::string& address,
                        int port,
                        const CompletionCallback& callback) OVERRIDE;
   virtual void Disconnect() OVERRIDE;
   virtual int Bind(const std::string& address, int port) OVERRIDE;
   virtual void Read(int count,
                     const ReadCompletionCallback& callback) OVERRIDE;
-  virtual void Write(scoped_refptr<net::IOBuffer> io_buffer,
-                     int bytes,
-                     const CompletionCallback& callback) OVERRIDE;
   virtual void RecvFrom(int count,
                         const RecvFromCompletionCallback& callback) OVERRIDE;
   virtual void SendTo(scoped_refptr<net::IOBuffer> io_buffer,
                       int byte_count,
                       const std::string& address,
                       int port,
                       const CompletionCallback& callback) OVERRIDE;
 
   static TCPSocket* CreateSocketForTesting(
       net::TCPClientSocket* tcp_client_socket,
       APIResourceEventNotifier* event_notifier);
 
+ protected:
+  virtual int WriteImpl(net::IOBuffer* io_buffer,
+                        int io_buffer_size,
+                        const net::CompletionCallback& callback);
+
  private:
   virtual void OnConnectComplete(int result);
   virtual void OnReadComplete(scoped_refptr<net::IOBuffer> io_buffer,
                               int result);
-  virtual void OnWriteComplete(int result);
 
   TCPSocket(net::TCPClientSocket* tcp_client_socket,
             APIResourceEventNotifier* event_notifier);
 
   scoped_ptr<net::TCPClientSocket> socket_;
 
   CompletionCallback connect_callback_;
 
   ReadCompletionCallback read_callback_;
-
-  CompletionCallback write_callback_;
 };
 
 }  //  namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_SOCKET_TCP_SOCKET_H_