folly/io/async/test/SocketPair.h

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements. See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership. The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License. You may obtain a copy of the License at
   9  *
  10  *   http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing,
  13  * software distributed under the License is distributed on an
  14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15  * KIND, either express or implied. See the License for the
  16  * specific language governing permissions and limitations
  17  * under the License.
  18  */
  19 #pragma once
  20
  21 namespace folly {
  22
  23 class SocketPair {
  24  public:
  25   enum Mode {
  26     BLOCKING,
  27     NONBLOCKING
  28   };
  29
  30   explicit SocketPair(Mode mode = NONBLOCKING);
  31   ~SocketPair();
  32
  33   int operator[](int index) const {
  34     return fds_[index];
  35   }
  36
  37   void closeFD0();
  38   void closeFD1();
  39
  40   int extractFD0() {
  41     return extractFD(0);
  42   }
  43   int extractFD1() {
  44     return extractFD(1);
  45   }
  46   int extractFD(int index) {
  47     int fd = fds_[index];
  48     fds_[index] = -1;
  49     return fd;
  50   }
  51
  52  private:
  53   int fds_[2];
  54 };
  55
  56 }