lib/mvfst
1
0
Fork 0
mirror of https://github.com/facebookincubator/mvfst.git synced 2025-08-06 22:22:38 +03:00
Code Activity

Folly SocketOptionMap with strings - separate cmsgs

Browse Source 
Summary: Splitting cmgs from socket options.

Reviewed By: mjoras

Differential Revision: D49557451

fbshipit-source-id: fa3005170d049557d1c845fd14b0ada73b58b377
This commit is contained in:
Andrii Vasylevskyi
2023-10-01 11:33:09 -07:00
committed by Facebook GitHub Bot
parent 1626c53f1e
commit c0e008c266
9 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
quic/api/QuicTransportBase.cpp
View File

@@ -3826,11 +3826,11 @@ void QuicTransportBase::notifyAppRateLimited() {
}
}
void QuicTransportBase::setCmsgs(const folly::SocketOptionMap& options) {
void QuicTransportBase::setCmsgs(const folly::SocketCmsgMap& options) {
socket_->setCmsgs(options);
}
void QuicTransportBase::appendCmsgs(const folly::SocketOptionMap& options) {
void QuicTransportBase::appendCmsgs(const folly::SocketCmsgMap& options) {
socket_->appendCmsgs(options);
}
@@ -3904,7 +3904,7 @@ QuicTransportBase::setStreamGroupRetransmissionPolicy(
}
void QuicTransportBase::updatePacketProcessorsPrewriteRequests() {
folly::SocketOptionMap cmsgs;
folly::SocketCmsgMap cmsgs;
for (const auto& pp : conn_->packetProcessors) {
// In case of overlapping cmsg keys, the priority is given to
// that were added to the QuicSocket first.
@@ -3921,7 +3921,7 @@ void QuicTransportBase::updatePacketProcessorsPrewriteRequests() {
conn_->socketCmsgsState.targetWriteCount = conn_->writeCount;
}
folly::Optional<folly::SocketOptionMap>
folly::Optional<folly::SocketCmsgMap>
QuicTransportBase::getAdditionalCmsgsForAsyncUDPSocket() {
if (conn_->socketCmsgsState.additionalCmsgs) {
// This callback should be happening for the target write

6
quic/api/QuicTransportBase.h
View File

@@ -660,9 +660,9 @@ class QuicTransportBase : public QuicSocket, QuicStreamPrioritiesObserver {
* specific transport and does not change the other sockets sharing the same
* fd.
*/
void setCmsgs(const folly::SocketOptionMap& options);
void setCmsgs(const folly::SocketCmsgMap& options);
void appendCmsgs(const folly::SocketOptionMap& options);
void appendCmsgs(const folly::SocketCmsgMap& options);
/**
* Sets the policy per stream group id.
@@ -852,7 +852,7 @@ class QuicTransportBase : public QuicSocket, QuicStreamPrioritiesObserver {
* The callback function for AsyncUDPSocket to provide the additional cmsgs
* required by this QuicSocket's packet processors.
*/
folly::Optional<folly::SocketOptionMap> getAdditionalCmsgsForAsyncUDPSocket();
folly::Optional<folly::SocketCmsgMap> getAdditionalCmsgsForAsyncUDPSocket();
std::atomic<QuicEventBase*> qEvbPtr_;
std::unique_ptr<QuicAsyncUDPSocketWrapper> socket_;

2
quic/api/test/QuicTransportBaseTest.cpp
View File

@@ -4212,7 +4212,7 @@ TEST_P(QuicTransportImplTestBase, ZeroLengthDatagramBufs) {
TEST_P(QuicTransportImplTestBase, Cmsgs) {
transport->setServerConnectionId();
folly::SocketOptionMap cmsgs;
folly::SocketCmsgMap cmsgs;
cmsgs[{IPPROTO_IP, IP_TOS}] = 123;
EXPECT_CALL(*socketPtr, setCmsgs(_)).Times(1);
transport->setCmsgs(cmsgs);

6
quic/api/test/QuicTypedTransportTest.cpp
View File

@@ -975,7 +975,7 @@ TYPED_TEST(
this->loopForWrites();
ASSERT_FALSE(this->getConn().outstandings.packets.empty());
folly::SocketOptionMap expectedCmsgs;
folly::SocketCmsgMap expectedCmsgs;
expectedCmsgs[{IPPROTO_IPV6, IPV6_HOPLIMIT}] = 255;
expectedCmsgs[{IPPROTO_IPV6, IPV6_DONTFRAG}] = 1;
auto pkt = this->getConn().outstandings.packets.rbegin();
@@ -1019,7 +1019,7 @@ TYPED_TEST(
ASSERT_EQ(this->getConn().outstandings.packets.size(), packetsSent);
// Verify both packets are marked
folly::SocketOptionMap expectedCmsgs;
folly::SocketCmsgMap expectedCmsgs;
expectedCmsgs[{IPPROTO_IPV6, IPV6_HOPLIMIT}] = 255;
auto pkt = this->getConn().outstandings.packets.rbegin();
ASSERT_TRUE(pkt->metadata.cmsgs);
@@ -1065,7 +1065,7 @@ TYPED_TEST(
ASSERT_EQ(this->getConn().outstandings.packets.size(), packetsSent);
// Verify the last two packets are marked
folly::SocketOptionMap expectedCmsgs;
folly::SocketCmsgMap expectedCmsgs;
expectedCmsgs[{IPPROTO_IPV6, IPV6_DONTFRAG}] = 1;
auto pkt = this->getConn().outstandings.packets.rbegin();
ASSERT_TRUE(pkt->metadata.cmsgs);

6
quic/common/QuicAsyncUDPSocketImpl.cpp
View File

@@ -112,7 +112,7 @@ int QuicAsyncUDPSocketImpl::writem(
}
void QuicAsyncUDPSocketImpl::setAdditionalCmsgsFunc(
folly::Function<folly::Optional<folly::SocketOptionMap>()>&&
folly::Function<folly::Optional<folly::SocketCmsgMap>()>&&
/* additionalCmsgsFunc */) {
LOG(WARNING)
<< "Setting an additional cmsgs function is not implemented for QuicAsyncUDPSocketImpl";
@@ -157,12 +157,12 @@ void QuicAsyncUDPSocketImpl::detachEventBase() {
}
void QuicAsyncUDPSocketImpl::setCmsgs(
const folly::SocketOptionMap& /* cmsgs */) {
const folly::SocketCmsgMap& /* cmsgs */) {
throw std::runtime_error("setCmsgs is not implemented.");
}
void QuicAsyncUDPSocketImpl::appendCmsgs(
const folly::SocketOptionMap& /* cmsgs */) {
const folly::SocketCmsgMap& /* cmsgs */) {
throw std::runtime_error("appendCmsgs is not implemented.");
}

6
quic/common/QuicAsyncUDPSocketImpl.h
View File

@@ -185,10 +185,10 @@ class QuicAsyncUDPSocketImpl {
/**
* Set extra control messages to send
*/
void setCmsgs(const folly::SocketOptionMap& /* cmsgs */);
void appendCmsgs(const folly::SocketOptionMap& /* cmsgs */);
void setCmsgs(const folly::SocketCmsgMap& /* cmsgs */);
void appendCmsgs(const folly::SocketCmsgMap& /* cmsgs */);
void setAdditionalCmsgsFunc(
folly::Function<folly::Optional<folly::SocketOptionMap>()>&&
folly::Function<folly::Optional<folly::SocketCmsgMap>()>&&
/* additionalCmsgsFunc */);
/*

2
quic/congestion_control/PacketProcessor.h
View File

@@ -17,7 +17,7 @@ namespace quic {
class PacketProcessor {
public:
struct PrewriteRequest {
folly::Optional<folly::SocketOptionMap> cmsgs;
folly::Optional<folly::SocketCmsgMap> cmsgs;
};
virtual ~PacketProcessor() = default;

2
quic/state/OutstandingPacket.h
View File

@@ -43,7 +43,7 @@ struct OutstandingPacketMetadata {
// tracks the number of writes on this socket.
uint64_t writeCount{0};
// Cmsgs added by the QuicSocket when this packet was written
folly::Optional<folly::SocketOptionMap> cmsgs;
folly::Optional<folly::SocketCmsgMap> cmsgs;
// Has value if the packet is lost by timeout. The value is the loss timeout
// dividend that was used to declare this packet.

2
quic/state/StateData.h
View File

@@ -667,7 +667,7 @@ struct QuicConnectionStateBase : public folly::DelayedDestruction {
retransmissionPolicies;
struct SocketCmsgsState {
folly::Optional<folly::SocketOptionMap> additionalCmsgs;
folly::Optional<folly::SocketCmsgMap> additionalCmsgs;
// The write count which this SocketCmsgs state is intended for.
// This is used to make sure this cmsgs list does not end up used
// for multiple writes.