OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/message_pump_libevent.h" | 5 #include "base/message_pump_libevent.h" |
6 | 6 |
7 #include <errno.h> | 7 #include <errno.h> |
8 #include <fcntl.h> | 8 #include <fcntl.h> |
9 #include <unistd.h> | 9 #include <unistd.h> |
10 | 10 |
(...skipping 126 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
137 } | 137 } |
138 if (wakeup_pipe_out_ >= 0) { | 138 if (wakeup_pipe_out_ >= 0) { |
139 if (HANDLE_EINTR(close(wakeup_pipe_out_)) < 0) | 139 if (HANDLE_EINTR(close(wakeup_pipe_out_)) < 0) |
140 DPLOG(ERROR) << "close"; | 140 DPLOG(ERROR) << "close"; |
141 } | 141 } |
142 event_base_free(event_base_); | 142 event_base_free(event_base_); |
143 } | 143 } |
144 | 144 |
145 bool MessagePumpLibevent::WatchFileDescriptor(int fd, | 145 bool MessagePumpLibevent::WatchFileDescriptor(int fd, |
146 bool persistent, | 146 bool persistent, |
147 Mode mode, | 147 int mode, |
148 FileDescriptorWatcher *controller, | 148 FileDescriptorWatcher *controller, |
149 Watcher *delegate) { | 149 Watcher *delegate) { |
150 DCHECK_GE(fd, 0); | 150 DCHECK_GE(fd, 0); |
151 DCHECK(controller); | 151 DCHECK(controller); |
152 DCHECK(delegate); | 152 DCHECK(delegate); |
153 DCHECK(mode == WATCH_READ || mode == WATCH_WRITE || mode == WATCH_READ_WRITE); | 153 DCHECK(mode == WATCH_READ || mode == WATCH_WRITE || mode == WATCH_READ_WRITE); |
154 // WatchFileDescriptor should be called on the pump thread. It is not | 154 // WatchFileDescriptor should be called on the pump thread. It is not |
155 // threadsafe, and your watcher may never be registered. | 155 // threadsafe, and your watcher may never be registered. |
156 DCHECK(watch_file_descriptor_caller_checker_.CalledOnValidThread()); | 156 DCHECK(watch_file_descriptor_caller_checker_.CalledOnValidThread()); |
157 | 157 |
158 int event_mask = persistent ? EV_PERSIST : 0; | 158 int event_mask = persistent ? EV_PERSIST : 0; |
159 if ((mode & WATCH_READ) != 0) { | 159 if (mode & WATCH_READ) { |
160 event_mask |= EV_READ; | 160 event_mask |= EV_READ; |
161 } | 161 } |
162 if ((mode & WATCH_WRITE) != 0) { | 162 if (mode & WATCH_WRITE) { |
163 event_mask |= EV_WRITE; | 163 event_mask |= EV_WRITE; |
164 } | 164 } |
165 | 165 |
166 scoped_ptr<event> evt(controller->ReleaseEvent()); | 166 scoped_ptr<event> evt(controller->ReleaseEvent()); |
167 if (evt.get() == NULL) { | 167 if (evt.get() == NULL) { |
168 // Ownership is transferred to the controller. | 168 // Ownership is transferred to the controller. |
169 evt.reset(new event); | 169 evt.reset(new event); |
170 } else { | 170 } else { |
171 // Make sure we don't pick up any funky internal libevent masks. | 171 // Make sure we don't pick up any funky internal libevent masks. |
172 int old_interest_mask = evt.get()->ev_events & | 172 int old_interest_mask = evt.get()->ev_events & |
(...skipping 10 matching lines...) Expand all Loading... |
183 if (EVENT_FD(evt.get()) != fd) { | 183 if (EVENT_FD(evt.get()) != fd) { |
184 NOTREACHED() << "FDs don't match" << EVENT_FD(evt.get()) << "!=" << fd; | 184 NOTREACHED() << "FDs don't match" << EVENT_FD(evt.get()) << "!=" << fd; |
185 return false; | 185 return false; |
186 } | 186 } |
187 } | 187 } |
188 | 188 |
189 // Set current interest mask and message pump for this event. | 189 // Set current interest mask and message pump for this event. |
190 event_set(evt.get(), fd, event_mask, OnLibeventNotification, controller); | 190 event_set(evt.get(), fd, event_mask, OnLibeventNotification, controller); |
191 | 191 |
192 // Tell libevent which message pump this socket will belong to when we add it. | 192 // Tell libevent which message pump this socket will belong to when we add it. |
193 if (event_base_set(event_base_, evt.get()) != 0) { | 193 if (event_base_set(event_base_, evt.get())) { |
194 return false; | 194 return false; |
195 } | 195 } |
196 | 196 |
197 // Add this socket to the list of monitored sockets. | 197 // Add this socket to the list of monitored sockets. |
198 if (event_add(evt.get(), NULL) != 0) { | 198 if (event_add(evt.get(), NULL)) { |
199 return false; | 199 return false; |
200 } | 200 } |
201 | 201 |
202 // Transfer ownership of evt to controller. | 202 // Transfer ownership of evt to controller. |
203 controller->Init(evt.release()); | 203 controller->Init(evt.release()); |
204 | 204 |
205 controller->set_watcher(delegate); | 205 controller->set_watcher(delegate); |
206 controller->set_pump(this); | 206 controller->set_pump(this); |
207 | 207 |
208 return true; | 208 return true; |
(...skipping 165 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
374 // Remove and discard the wakeup byte. | 374 // Remove and discard the wakeup byte. |
375 char buf; | 375 char buf; |
376 int nread = HANDLE_EINTR(read(socket, &buf, 1)); | 376 int nread = HANDLE_EINTR(read(socket, &buf, 1)); |
377 DCHECK_EQ(nread, 1); | 377 DCHECK_EQ(nread, 1); |
378 that->processed_io_events_ = true; | 378 that->processed_io_events_ = true; |
379 // Tell libevent to break out of inner loop. | 379 // Tell libevent to break out of inner loop. |
380 event_base_loopbreak(that->event_base_); | 380 event_base_loopbreak(that->event_base_); |
381 } | 381 } |
382 | 382 |
383 } // namespace base | 383 } // namespace base |
OLD | NEW |