base::Bind: Convert Socket::Read.

jingle/glue/pseudotcp_adapter_unittest.cc

 // Copyright (c) 2011 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 "jingle/glue/pseudotcp_adapter.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/compiler_specific.h"
 #include "jingle/glue/thread_wrapper.h"
+#include "net/base/completion_callback.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/base/test_completion_callback.h"
 #include "net/udp/udp_socket.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 
 namespace jingle_glue {
 namespace {
@@ skipping 50 matching lines @@
  private:
   double volume_;
   double rate_;
   double level_;
   base::TimeTicks last_update_;
 };
 
 class FakeSocket : public net::Socket {
  public:
   FakeSocket()
-      : read_callback_(NULL),
+      : old_read_callback_(NULL),
         rate_limiter_(NULL),
         latency_ms_(0) {
   }
   virtual ~FakeSocket() { }
 
   void AppendInputPacket(const std::vector<char>& data) {
     if (rate_limiter_ && rate_limiter_->DropNextPacket())
       return;  // Lose the packet.
 
-    if (read_callback_) {
+    if (old_read_callback_ || !read_callback_.is_null()) {
       int size = std::min(read_buffer_size_, static_cast<int>(data.size()));
       memcpy(read_buffer_->data(), &data[0], data.size());
-      net::OldCompletionCallback* cb = read_callback_;
-      read_callback_ = NULL;
-      read_buffer_ = NULL;
-      cb->Run(size);
+      if (old_read_callback_) {
+        net::OldCompletionCallback* cb = old_read_callback_;
+        old_read_callback_ = NULL;
+        read_buffer_ = NULL;
+        cb->Run(size);
+      } else {
+        net::CompletionCallback cb = read_callback_;
+        read_callback_.Reset();
+        read_buffer_ = NULL;
+        cb.Run(size);
+      }
     } else {
       incoming_packets_.push_back(data);
     }
   }
 
   void Connect(FakeSocket* peer_socket) {
     peer_socket_ = peer_socket;
   }
 
   void set_rate_limiter(RateLimiter* rate_limiter) {
     rate_limiter_ = rate_limiter;
   };
 
   void set_latency(int latency_ms) { latency_ms_ = latency_ms; };
 
-  // net::Socket interface.
+  // net::Socket implementation.
   virtual int Read(net::IOBuffer* buf, int buf_len,
                    net::OldCompletionCallback* callback) {
-    CHECK(!read_callback_);
+    CHECK(!old_read_callback_ && read_callback_.is_null());
     CHECK(buf);
 
     if (incoming_packets_.size() > 0) {
+      scoped_refptr<net::IOBuffer> buffer(buf);
+      int size = std::min(
+          static_cast<int>(incoming_packets_.front().size()), buf_len);
+      memcpy(buffer->data(), &*incoming_packets_.front().begin(), size);
+      incoming_packets_.pop_front();
+      return size;
+    } else {
+      old_read_callback_ = callback;
+      read_buffer_ = buf;
+      read_buffer_size_ = buf_len;
+      return net::ERR_IO_PENDING;
+    }
+  }
+  virtual int Read(net::IOBuffer* buf, int buf_len,
+                   const net::CompletionCallback& callback) {
+    CHECK(!old_read_callback_ && read_callback_.is_null());
+    CHECK(buf);
+
+    if (incoming_packets_.size() > 0) {
       scoped_refptr<net::IOBuffer> buffer(buf);
       int size = std::min(
           static_cast<int>(incoming_packets_.front().size()), buf_len);
       memcpy(buffer->data(), &*incoming_packets_.front().begin(), size);
       incoming_packets_.pop_front();
       return size;
     } else {
       read_callback_ = callback;
       read_buffer_ = buf;
       read_buffer_size_ = buf_len;
@@ skipping 21 matching lines @@
     return false;
   }
   virtual bool SetSendBufferSize(int32 size) OVERRIDE {
     NOTIMPLEMENTED();
     return false;
   }
 
  private:
   scoped_refptr<net::IOBuffer> read_buffer_;
   int read_buffer_size_;
-  net::OldCompletionCallback* read_callback_;
+  net::OldCompletionCallback* old_read_callback_;
+  net::CompletionCallback read_callback_;
 
   std::deque<std::vector<char> > incoming_packets_;
 
   FakeSocket* peer_socket_;
   RateLimiter* rate_limiter_;
   int latency_ms_;
 };
 
 class TCPChannelTester : public base::RefCountedThreadSafe<TCPChannelTester> {
  public:
@@ skipping 234 matching lines @@
   host_pseudotcp_->Connect(&host_connect_cb);
   client_pseudotcp_->Connect(&client_connect_cb);
   message_loop_.Run();
 
   ASSERT_EQ(NULL, host_pseudotcp_.get());
 }
 
 }  // namespace
 
 }  // namespace jingle_glue