OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <sys/socket.h> | 5 #include <sys/socket.h> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/files/file_path.h" | 8 #include "base/files/file_path.h" |
9 #include "base/path_service.h" | 9 #include "base/path_service.h" |
10 #include "base/posix/eintr_wrapper.h" | 10 #include "base/posix/eintr_wrapper.h" |
11 #include "base/synchronization/waitable_event.h" | 11 #include "base/synchronization/waitable_event.h" |
12 #include "base/threading/thread.h" | 12 #include "base/threading/thread.h" |
13 #include "base/threading/thread_restrictions.h" | 13 #include "base/threading/thread_restrictions.h" |
14 #include "ipc/unix_domain_socket_util.h" | 14 #include "ipc/unix_domain_socket_util.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
16 | 16 |
17 namespace { | 17 namespace { |
18 | 18 |
19 class SocketAcceptor : public base::MessageLoopForIO::Watcher { | 19 class SocketAcceptor : public base::MessageLoopForIO::Watcher { |
20 public: | 20 public: |
21 SocketAcceptor(int fd, base::MessageLoopProxy* target_thread) | 21 SocketAcceptor(int fd, base::MessageLoopProxy* target_thread) |
22 : server_fd_(-1), | 22 : server_fd_(-1), |
23 target_thread_(target_thread), | 23 target_thread_(target_thread), |
24 started_watching_event_(false, false), | 24 started_watching_event_(false, false), |
25 accepted_event_(false, false) { | 25 accepted_event_(false, false) { |
26 target_thread->PostTask(FROM_HERE, | 26 target_thread->PostTask(FROM_HERE, |
27 base::Bind(&SocketAcceptor::StartWatching, base::Unretained(this), fd)); | 27 base::Bind(&SocketAcceptor::StartWatching, base::Unretained(this), fd)); |
28 } | 28 } |
29 | 29 |
30 virtual ~SocketAcceptor() { | 30 ~SocketAcceptor() override { Close(); } |
31 Close(); | |
32 } | |
33 | 31 |
34 int server_fd() const { return server_fd_; } | 32 int server_fd() const { return server_fd_; } |
35 | 33 |
36 void WaitUntilReady() { | 34 void WaitUntilReady() { |
37 started_watching_event_.Wait(); | 35 started_watching_event_.Wait(); |
38 } | 36 } |
39 | 37 |
40 void WaitForAccept() { | 38 void WaitForAccept() { |
41 accepted_event_.Wait(); | 39 accepted_event_.Wait(); |
42 } | 40 } |
(...skipping 10 matching lines...) Expand all Loading... |
53 void StartWatching(int fd) { | 51 void StartWatching(int fd) { |
54 watcher_.reset(new base::MessageLoopForIO::FileDescriptorWatcher); | 52 watcher_.reset(new base::MessageLoopForIO::FileDescriptorWatcher); |
55 base::MessageLoopForIO::current()->WatchFileDescriptor( | 53 base::MessageLoopForIO::current()->WatchFileDescriptor( |
56 fd, true, base::MessageLoopForIO::WATCH_READ, watcher_.get(), this); | 54 fd, true, base::MessageLoopForIO::WATCH_READ, watcher_.get(), this); |
57 started_watching_event_.Signal(); | 55 started_watching_event_.Signal(); |
58 } | 56 } |
59 void StopWatching(base::MessageLoopForIO::FileDescriptorWatcher* watcher) { | 57 void StopWatching(base::MessageLoopForIO::FileDescriptorWatcher* watcher) { |
60 watcher->StopWatchingFileDescriptor(); | 58 watcher->StopWatchingFileDescriptor(); |
61 delete watcher; | 59 delete watcher; |
62 } | 60 } |
63 virtual void OnFileCanReadWithoutBlocking(int fd) override { | 61 void OnFileCanReadWithoutBlocking(int fd) override { |
64 ASSERT_EQ(-1, server_fd_); | 62 ASSERT_EQ(-1, server_fd_); |
65 IPC::ServerAcceptConnection(fd, &server_fd_); | 63 IPC::ServerAcceptConnection(fd, &server_fd_); |
66 watcher_->StopWatchingFileDescriptor(); | 64 watcher_->StopWatchingFileDescriptor(); |
67 accepted_event_.Signal(); | 65 accepted_event_.Signal(); |
68 } | 66 } |
69 virtual void OnFileCanWriteWithoutBlocking(int fd) override {} | 67 void OnFileCanWriteWithoutBlocking(int fd) override {} |
70 | 68 |
71 int server_fd_; | 69 int server_fd_; |
72 base::MessageLoopProxy* target_thread_; | 70 base::MessageLoopProxy* target_thread_; |
73 scoped_ptr<base::MessageLoopForIO::FileDescriptorWatcher> watcher_; | 71 scoped_ptr<base::MessageLoopForIO::FileDescriptorWatcher> watcher_; |
74 base::WaitableEvent started_watching_event_; | 72 base::WaitableEvent started_watching_event_; |
75 base::WaitableEvent accepted_event_; | 73 base::WaitableEvent accepted_event_; |
76 | 74 |
77 DISALLOW_COPY_AND_ASSIGN(SocketAcceptor); | 75 DISALLOW_COPY_AND_ASSIGN(SocketAcceptor); |
78 }; | 76 }; |
79 | 77 |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
166 HANDLE_EINTR(send(connection.client_fd(), buffer, buf_len, 0)); | 164 HANDLE_EINTR(send(connection.client_fd(), buffer, buf_len, 0)); |
167 ASSERT_EQ(buf_len, sent_bytes); | 165 ASSERT_EQ(buf_len, sent_bytes); |
168 char recv_buf[sizeof(buffer)]; | 166 char recv_buf[sizeof(buffer)]; |
169 size_t received_bytes = | 167 size_t received_bytes = |
170 HANDLE_EINTR(recv(connection.server_fd(), recv_buf, buf_len, 0)); | 168 HANDLE_EINTR(recv(connection.server_fd(), recv_buf, buf_len, 0)); |
171 ASSERT_EQ(buf_len, received_bytes); | 169 ASSERT_EQ(buf_len, received_bytes); |
172 ASSERT_EQ(0, memcmp(recv_buf, buffer, buf_len)); | 170 ASSERT_EQ(0, memcmp(recv_buf, buffer, buf_len)); |
173 } | 171 } |
174 | 172 |
175 } // namespace | 173 } // namespace |
OLD | NEW |