--- /dev/null
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+#include "folly/io/async/EventHandler.h"
+#include "folly/io/async/EventBase.h"
+
+#include <assert.h>
+
+namespace folly {
+
+EventHandler::EventHandler(EventBase* eventBase, int fd) {
+ event_set(&event_, fd, 0, &EventHandler::libeventCallback, this);
+ if (eventBase != nullptr) {
+ setEventBase(eventBase);
+ } else {
+ // Callers must set the EventBase and fd before using this timeout.
+ // Set event_->ev_base to nullptr to ensure that this happens.
+ // (otherwise libevent will initialize it to the "default" event_base)
+ event_.ev_base = nullptr;
+ eventBase_ = nullptr;
+ }
+}
+
+EventHandler::~EventHandler() {
+ unregisterHandler();
+}
+
+bool EventHandler::registerImpl(uint16_t events, bool internal) {
+ assert(event_.ev_base != nullptr);
+
+ // We have to unregister the event before we can change the event flags
+ if (isHandlerRegistered()) {
+ // If the new events are the same are the same as the already registered
+ // flags, we don't have to do anything. Just return.
+ if (events == event_.ev_events &&
+ static_cast<bool>(event_.ev_flags & EVLIST_INTERNAL) == internal) {
+ return true;
+ }
+
+ event_del(&event_);
+ }
+
+ // Update the event flags
+ // Unfortunately, event_set() resets the event_base, so we have to remember
+ // it before hand, then pass it back into event_base_set() afterwards
+ struct event_base* evb = event_.ev_base;
+ event_set(&event_, event_.ev_fd, events,
+ &EventHandler::libeventCallback, this);
+ event_base_set(evb, &event_);
+
+ // Set EVLIST_INTERNAL if this is an internal event
+ if (internal) {
+ event_.ev_flags |= EVLIST_INTERNAL;
+ }
+
+ // Add the event.
+ //
+ // Although libevent allows events to wait on both I/O and a timeout,
+ // we intentionally don't allow an EventHandler to also use a timeout.
+ // Callers must maintain a separate AsyncTimeout object if they want a
+ // timeout.
+ //
+ // Otherwise, it is difficult to handle persistent events properly. (The I/O
+ // event and timeout may both fire together the same time around the event
+ // loop. Normally we would want to inform the caller of the I/O event first,
+ // then the timeout. However, it is difficult to do this properly since the
+ // I/O callback could delete the EventHandler.) Additionally, if a caller
+ // uses the same struct event for both I/O and timeout, and they just want to
+ // reschedule the timeout, libevent currently makes an epoll_ctl() call even
+ // if the I/O event flags haven't changed. Using a separate event struct is
+ // therefore slightly more efficient in this case (although it does take up
+ // more space).
+ if (event_add(&event_, nullptr) < 0) {
+ LOG(ERROR) << "EventBase: failed to register event handler for fd "
+ << event_.ev_fd << ": " << strerror(errno);
+ // Call event_del() to make sure the event is completely uninstalled
+ event_del(&event_);
+ return false;
+ }
+
+ return true;
+}
+
+void EventHandler::unregisterHandler() {
+ if (isHandlerRegistered()) {
+ event_del(&event_);
+ }
+}
+
+void EventHandler::attachEventBase(EventBase* eventBase) {
+ // attachEventBase() may only be called on detached handlers
+ assert(event_.ev_base == nullptr);
+ assert(!isHandlerRegistered());
+ // This must be invoked from the EventBase's thread
+ assert(eventBase->isInEventBaseThread());
+
+ setEventBase(eventBase);
+}
+
+void EventHandler::detachEventBase() {
+ ensureNotRegistered(__func__);
+ event_.ev_base = nullptr;
+}
+
+void EventHandler::changeHandlerFD(int fd) {
+ ensureNotRegistered(__func__);
+ // event_set() resets event_base.ev_base, so manually restore it afterwards
+ struct event_base* evb = event_.ev_base;
+ event_set(&event_, fd, 0, &EventHandler::libeventCallback, this);
+ event_.ev_base = evb; // don't use event_base_set(), since evb may be nullptr
+}
+
+void EventHandler::initHandler(EventBase* eventBase, int fd) {
+ ensureNotRegistered(__func__);
+ event_set(&event_, fd, 0, &EventHandler::libeventCallback, this);
+ setEventBase(eventBase);
+}
+
+void EventHandler::ensureNotRegistered(const char* fn) {
+ // Neither the EventBase nor file descriptor may be changed while the
+ // handler is registered. Treat it as a programmer bug and abort the program
+ // if this requirement is violated.
+ if (isHandlerRegistered()) {
+ LOG(ERROR) << fn << " called on registered handler; aborting";
+ abort();
+ }
+}
+
+void EventHandler::libeventCallback(int fd, short events, void* arg) {
+ EventHandler* handler = reinterpret_cast<EventHandler*>(arg);
+ assert(fd == handler->event_.ev_fd);
+
+ // this can't possibly fire if handler->eventBase_ is nullptr
+ (void) handler->eventBase_->bumpHandlingTime();
+
+ handler->handlerReady(events);
+}
+
+void EventHandler::setEventBase(EventBase* eventBase) {
+ event_base_set(eventBase->getLibeventBase(), &event_);
+ eventBase_ = eventBase;
+}
+
+bool EventHandler::isPending() {
+ if (event_.ev_flags & EVLIST_ACTIVE) {
+ if (event_.ev_res & EV_READ) {
+ return true;
+ }
+ }
+ return false;
+}
+
+} // folly