mirror of
https://github.com/facebook/sapling.git
synced 2024-10-05 14:28:17 +03:00
36089152d7
Summary: I'm planning on re-using this thread pool in both PrjFS and FUSE, let's thus move it to a more generic location. Reviewed By: genevievehelsel Differential Revision: D51302395 fbshipit-source-id: 39a503fd31a7b7464fc0a38f8d4152e33808293f
113 lines
3.2 KiB
C++
113 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This software may be used and distributed according to the terms of the
|
|
* GNU General Public License version 2.
|
|
*/
|
|
|
|
#include "eden/fs/nfs/NfsServer.h"
|
|
|
|
#include "eden/fs/nfs/Nfsd3.h"
|
|
#include "eden/fs/nfs/portmap/Rpcbindd.h"
|
|
|
|
namespace facebook::eden {
|
|
|
|
NfsServer::NfsServer(
|
|
PrivHelper* privHelper,
|
|
folly::EventBase* evb,
|
|
std::shared_ptr<folly::Executor> threadPool,
|
|
bool shouldRunOurOwnRpcbindServer,
|
|
const std::shared_ptr<StructuredLogger>& structuredLogger)
|
|
: privHelper_{privHelper},
|
|
evb_(evb),
|
|
threadPool_{std::move(threadPool)},
|
|
rpcbindd_(
|
|
shouldRunOurOwnRpcbindServer
|
|
? std::make_shared<Rpcbindd>(evb_, threadPool_, structuredLogger)
|
|
: nullptr),
|
|
mountd_(evb_, threadPool_, structuredLogger) {}
|
|
|
|
void NfsServer::initialize(
|
|
folly::SocketAddress addr,
|
|
bool registerMountdWithRpcbind) {
|
|
mountd_.initialize(addr, registerMountdWithRpcbind);
|
|
if (rpcbindd_) {
|
|
rpcbindd_->initialize();
|
|
}
|
|
auto registeredAddr = mountd_.getAddr();
|
|
// we can't register uds sockets with our portmapper (portmapper v2
|
|
// does not support those).
|
|
if (registeredAddr.isFamilyInet()) {
|
|
recordPortNumber(
|
|
mountd_.getProgramNumber(),
|
|
mountd_.getProgramVersion(),
|
|
registeredAddr.getPort());
|
|
}
|
|
}
|
|
|
|
void NfsServer::initialize(folly::File socket) {
|
|
mountd_.initialize(std::move(socket));
|
|
if (rpcbindd_) {
|
|
rpcbindd_->initialize();
|
|
}
|
|
// TODO: we should register the mountd server on takeover too. but
|
|
// we only transfer the connected socket not the listening socket.
|
|
// the listening one is the one we wanna register. So we need to
|
|
// transfer that socket to be able to register it.
|
|
}
|
|
|
|
NfsServer::NfsMountInfo NfsServer::registerMount(
|
|
AbsolutePathPiece path,
|
|
InodeNumber rootIno,
|
|
std::unique_ptr<NfsDispatcher> dispatcher,
|
|
const folly::Logger* straceLogger,
|
|
std::shared_ptr<ProcessInfoCache> processInfoCache,
|
|
std::shared_ptr<FsEventLogger> fsEventLogger,
|
|
const std::shared_ptr<StructuredLogger>& structuredLogger,
|
|
folly::Duration requestTimeout,
|
|
std::shared_ptr<Notifier> notifier,
|
|
CaseSensitivity caseSensitive,
|
|
uint32_t iosize,
|
|
size_t traceBusCapacity) {
|
|
auto nfsd = std::unique_ptr<Nfsd3, FsChannelDeleter>{new Nfsd3{
|
|
privHelper_,
|
|
AbsolutePath{path},
|
|
evb_,
|
|
threadPool_,
|
|
std::move(dispatcher),
|
|
straceLogger,
|
|
std::move(processInfoCache),
|
|
std::move(fsEventLogger),
|
|
structuredLogger,
|
|
requestTimeout,
|
|
std::move(notifier),
|
|
caseSensitive,
|
|
iosize,
|
|
traceBusCapacity}};
|
|
mountd_.registerMount(path, rootIno);
|
|
|
|
return {std::move(nfsd), mountd_.getAddr()};
|
|
}
|
|
|
|
/**
|
|
* Registers an RPC service running a certain protocol version on port.
|
|
*/
|
|
void NfsServer::recordPortNumber(
|
|
uint32_t protocol,
|
|
uint32_t version,
|
|
uint32_t port) {
|
|
if (rpcbindd_) {
|
|
rpcbindd_->recordPortNumber(protocol, version, port);
|
|
}
|
|
}
|
|
|
|
void NfsServer::unregisterMount(AbsolutePathPiece path) {
|
|
mountd_.unregisterMount(path);
|
|
}
|
|
|
|
folly::SemiFuture<folly::File> NfsServer::takeoverStop() {
|
|
return mountd_.takeoverStop();
|
|
}
|
|
|
|
} // namespace facebook::eden
|