| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/async_socket_io_handler.h" | |
| 6 | |
| 7 namespace base { | |
| 8 | |
| 9 AsyncSocketIoHandler::AsyncSocketIoHandler() | |
| 10 : socket_(base::SyncSocket::kInvalidHandle), | |
| 11 context_(NULL), | |
| 12 is_pending_(false) {} | |
| 13 | |
| 14 AsyncSocketIoHandler::~AsyncSocketIoHandler() { | |
| 15 // We need to be deleted on the correct thread to avoid racing with the | |
| 16 // message loop thread. | |
| 17 DCHECK(CalledOnValidThread()); | |
| 18 | |
| 19 if (context_) { | |
| 20 if (is_pending_) { | |
| 21 // Make the context be deleted by the message pump when done. | |
| 22 context_->handler = NULL; | |
| 23 } else { | |
| 24 delete context_; | |
| 25 } | |
| 26 } | |
| 27 } | |
| 28 | |
| 29 // Implementation of IOHandler on Windows. | |
| 30 void AsyncSocketIoHandler::OnIOCompleted( | |
| 31 base::MessageLoopForIO::IOContext* context, | |
| 32 DWORD bytes_transfered, | |
| 33 DWORD error) { | |
| 34 DCHECK(CalledOnValidThread()); | |
| 35 DCHECK_EQ(context_, context); | |
| 36 DCHECK(!read_complete_.is_null()); | |
| 37 is_pending_ = false; | |
| 38 read_complete_.Run(error == ERROR_SUCCESS ? bytes_transfered : 0); | |
| 39 } | |
| 40 | |
| 41 bool AsyncSocketIoHandler::Read(char* buffer, int buffer_len) { | |
| 42 DCHECK(CalledOnValidThread()); | |
| 43 DCHECK(!read_complete_.is_null()); | |
| 44 DCHECK(!is_pending_); | |
| 45 DCHECK_NE(socket_, base::SyncSocket::kInvalidHandle); | |
| 46 | |
| 47 DWORD bytes_read = 0; | |
| 48 BOOL ok = ::ReadFile(socket_, buffer, buffer_len, &bytes_read, | |
| 49 &context_->overlapped); | |
| 50 // The completion port will be signaled regardless of completing the read | |
| 51 // straight away or asynchronously (ERROR_IO_PENDING). OnIOCompleted() will | |
| 52 // be called regardless and we don't need to explicitly run the callback | |
| 53 // in the case where ok is FALSE and GLE==ERROR_IO_PENDING. | |
| 54 is_pending_ = !ok && (GetLastError() == ERROR_IO_PENDING); | |
| 55 return ok || is_pending_; | |
| 56 } | |
| 57 | |
| 58 bool AsyncSocketIoHandler::Initialize(base::SyncSocket::Handle socket, | |
| 59 const ReadCompleteCallback& callback) { | |
| 60 DCHECK(!context_); | |
| 61 DCHECK_EQ(socket_, base::SyncSocket::kInvalidHandle); | |
| 62 | |
| 63 DetachFromThread(); | |
| 64 | |
| 65 socket_ = socket; | |
| 66 read_complete_ = callback; | |
| 67 | |
| 68 base::MessageLoopForIO::current()->RegisterIOHandler(socket, this); | |
| 69 | |
| 70 context_ = new base::MessageLoopForIO::IOContext(); | |
| 71 context_->handler = this; | |
| 72 memset(&context_->overlapped, 0, sizeof(context_->overlapped)); | |
| 73 | |
| 74 return true; | |
| 75 } | |
| 76 | |
| 77 } // namespace base. | |
| OLD | NEW |