folly/portability/Event.h

   1 /*
   2  * Copyright 2016-present Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #pragma once
  18
  19 #ifdef _MSC_VER
  20 // This needs to be before the libevent include.
  21 #include <folly/portability/Windows.h>
  22 #endif
  23
  24 #include <event.h>
  25
  26 #ifdef _MSC_VER
  27 #include <event2/event_compat.h> // @manual
  28 #include <folly/portability/Fcntl.h>
  29 #endif
  30
  31 namespace folly {
  32 #ifdef _MSC_VER
  33 using libevent_fd_t = evutil_socket_t;
  34 #else
  35 using libevent_fd_t = int;
  36 #endif
  37
  38 inline libevent_fd_t getLibeventFd(int fd) {
  39 #ifdef _MSC_VER
  40   if (fd == -1) {
  41     return (libevent_fd_t)INVALID_HANDLE_VALUE;
  42   }
  43   return _get_osfhandle(fd);
  44 #else
  45   return fd;
  46 #endif
  47 }
  48
  49 inline int libeventFdToFd(libevent_fd_t fd) {
  50 #ifdef _MSC_VER
  51   if (fd == (libevent_fd_t)INVALID_HANDLE_VALUE) {
  52     return -1;
  53   }
  54   return _open_osfhandle((intptr_t)fd, O_RDWR | O_BINARY);
  55 #else
  56   return fd;
  57 #endif
  58 }
  59
  60 using EventSetCallback = void (*)(libevent_fd_t, short, void*);
  61 inline void
  62 folly_event_set(event* e, int fd, short s, EventSetCallback f, void* arg) {
  63   auto lfd = getLibeventFd(fd);
  64   event_set(e, lfd, s, f, arg);
  65 }
  66 } // namespace folly