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
AsyncSocketIoHandler()9 AsyncSocketIoHandler::AsyncSocketIoHandler()
10 : socket_(base::SyncSocket::kInvalidHandle),
11 context_(NULL),
12 is_pending_(false) {}
13
~AsyncSocketIoHandler()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.
OnIOCompleted(base::MessageLoopForIO::IOContext * context,DWORD bytes_transfered,DWORD error)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
Read(char * buffer,int buffer_len)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
Initialize(base::SyncSocket::Handle socket,const ReadCompleteCallback & callback)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.
78