SocketPair.h 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. /*
  2. * Copyright 2014-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. #pragma once
  17. namespace folly {
  18. class SocketPair {
  19. public:
  20. enum Mode { BLOCKING, NONBLOCKING };
  21. explicit SocketPair(Mode mode = NONBLOCKING);
  22. ~SocketPair();
  23. int operator[](int index) const {
  24. return fds_[index];
  25. }
  26. void closeFD0();
  27. void closeFD1();
  28. int extractFD0() {
  29. return extractFD(0);
  30. }
  31. int extractFD1() {
  32. return extractFD(1);
  33. }
  34. int extractFD(int index) {
  35. int fd = fds_[index];
  36. fds_[index] = -1;
  37. return fd;
  38. }
  39. private:
  40. int fds_[2];
  41. };
  42. } // namespace folly