/*
- * Copyright 2017 Facebook, Inc.
+ * Copyright 2014-present Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
#include <glog/logging.h>
#include <folly/FileUtil.h>
+#include <folly/Singleton.h>
#include <folly/portability/Sockets.h>
namespace folly {
+namespace {
+struct PrivateTag {};
+folly::Singleton<folly::ShutdownSocketSet, PrivateTag> singleton;
+} // namespace
+
ShutdownSocketSet::ShutdownSocketSet(int maxFd)
- : maxFd_(maxFd),
- data_(static_cast<std::atomic<uint8_t>*>(
- folly::checkedCalloc(maxFd, sizeof(std::atomic<uint8_t>)))),
- nullFile_("/dev/null", O_RDWR) {
+ : maxFd_(maxFd),
+ data_(static_cast<std::atomic<uint8_t>*>(
+ folly::checkedCalloc(size_t(maxFd), sizeof(std::atomic<uint8_t>)))),
+ nullFile_("/dev/null", O_RDWR) {}
+
+std::shared_ptr<ShutdownSocketSet> ShutdownSocketSet::getInstance() {
+ return singleton.try_get();
}
void ShutdownSocketSet::add(int fd) {
return;
}
- auto& sref = data_[fd];
+ auto& sref = data_[size_t(fd)];
uint8_t prevState = FREE;
- CHECK(sref.compare_exchange_strong(prevState,
- IN_USE,
- std::memory_order_acq_rel))
- << "Invalid prev state for fd " << fd << ": " << int(prevState);
+ CHECK(sref.compare_exchange_strong(
+ prevState, IN_USE, std::memory_order_relaxed))
+ << "Invalid prev state for fd " << fd << ": " << int(prevState);
}
void ShutdownSocketSet::remove(int fd) {
return;
}
- auto& sref = data_[fd];
+ auto& sref = data_[size_t(fd)];
uint8_t prevState = 0;
-retry_load:
prevState = sref.load(std::memory_order_relaxed);
-
-retry:
- switch (prevState) {
- case IN_SHUTDOWN:
- std::this_thread::sleep_for(std::chrono::milliseconds(1));
- goto retry_load;
- case FREE:
- LOG(FATAL) << "Invalid prev state for fd " << fd << ": " << int(prevState);
- }
-
- if (!sref.compare_exchange_weak(prevState,
- FREE,
- std::memory_order_acq_rel)) {
- goto retry;
- }
+ do {
+ switch (prevState) {
+ case IN_SHUTDOWN:
+ std::this_thread::sleep_for(std::chrono::milliseconds(1));
+ prevState = sref.load(std::memory_order_relaxed);
+ continue;
+ case FREE:
+ LOG(FATAL) << "Invalid prev state for fd " << fd << ": "
+ << int(prevState);
+ }
+ } while (
+ !sref.compare_exchange_weak(prevState, FREE, std::memory_order_relaxed));
}
int ShutdownSocketSet::close(int fd) {
return folly::closeNoInt(fd);
}
- auto& sref = data_[fd];
+ auto& sref = data_[size_t(fd)];
uint8_t prevState = sref.load(std::memory_order_relaxed);
uint8_t newState = 0;
-retry:
- switch (prevState) {
- case IN_USE:
- case SHUT_DOWN:
- newState = FREE;
- break;
- case IN_SHUTDOWN:
- newState = MUST_CLOSE;
- break;
- default:
- LOG(FATAL) << "Invalid prev state for fd " << fd << ": " << int(prevState);
- }
-
- if (!sref.compare_exchange_weak(prevState,
- newState,
- std::memory_order_acq_rel)) {
- goto retry;
- }
+ do {
+ switch (prevState) {
+ case IN_USE:
+ case SHUT_DOWN:
+ newState = FREE;
+ break;
+ case IN_SHUTDOWN:
+ newState = MUST_CLOSE;
+ break;
+ default:
+ LOG(FATAL) << "Invalid prev state for fd " << fd << ": "
+ << int(prevState);
+ }
+ } while (!sref.compare_exchange_weak(
+ prevState, newState, std::memory_order_relaxed));
return newState == FREE ? folly::closeNoInt(fd) : 0;
}
return;
}
- auto& sref = data_[fd];
+ auto& sref = data_[size_t(fd)];
uint8_t prevState = IN_USE;
- if (!sref.compare_exchange_strong(prevState,
- IN_SHUTDOWN,
- std::memory_order_acq_rel)) {
+ if (!sref.compare_exchange_strong(
+ prevState, IN_SHUTDOWN, std::memory_order_relaxed)) {
return;
}
doShutdown(fd, abortive);
prevState = IN_SHUTDOWN;
- if (sref.compare_exchange_strong(prevState,
- SHUT_DOWN,
- std::memory_order_acq_rel)) {
+ if (sref.compare_exchange_strong(
+ prevState, SHUT_DOWN, std::memory_order_relaxed)) {
return;
}
folly::closeNoInt(fd); // ignore errors, nothing to do
- CHECK(sref.compare_exchange_strong(prevState,
- FREE,
- std::memory_order_acq_rel))
- << "Invalid prev state for fd " << fd << ": " << int(prevState);
+ CHECK(
+ sref.compare_exchange_strong(prevState, FREE, std::memory_order_relaxed))
+ << "Invalid prev state for fd " << fd << ": " << int(prevState);
}
void ShutdownSocketSet::shutdownAll(bool abortive) {
for (int i = 0; i < maxFd_; ++i) {
- auto& sref = data_[i];
- if (sref.load(std::memory_order_acquire) == IN_USE) {
+ auto& sref = data_[size_t(i)];
+ if (sref.load(std::memory_order_relaxed) == IN_USE) {
shutdown(i, abortive);
}
}
folly::dup2NoInt(nullFile_.fd(), fd);
}
-} // namespaces
+} // namespace folly