BOOST_AUTO_TEST_SUITE(mplexer)
-BOOST_AUTO_TEST_CASE(test_MPlexer)
+BOOST_AUTO_TEST_CASE(test_getMultiplexerSilent)
{
auto mplexer = std::unique_ptr<FDMultiplexer>(FDMultiplexer::getMultiplexerSilent());
BOOST_REQUIRE(mplexer != nullptr);
- struct timeval now;
+ struct timeval now{0,0};
int ready = mplexer->run(&now, 100);
BOOST_CHECK_EQUAL(ready, 0);
+ BOOST_CHECK(now.tv_sec != 0);
+}
- std::vector<int> readyFDs;
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_CHECK_EQUAL(readyFDs.size(), 0U);
-
- auto timeouts = mplexer->getTimeouts(now);
- BOOST_CHECK_EQUAL(timeouts.size(), 0U);
-
- int pipes[2];
- int res = pipe(pipes);
- BOOST_REQUIRE_EQUAL(res, 0);
- BOOST_REQUIRE_EQUAL(setNonBlocking(pipes[0]), true);
- BOOST_REQUIRE_EQUAL(setNonBlocking(pipes[1]), true);
-
- /* let's declare a TTD that expired 5s ago */
- struct timeval ttd = now;
- ttd.tv_sec -= 5;
-
- bool writeCBCalled = false;
- auto writeCB = [](int fd, FDMultiplexer::funcparam_t param) {
- auto calledPtr = boost::any_cast<bool*>(param);
- BOOST_REQUIRE(calledPtr != nullptr);
- *calledPtr = true;
- };
- mplexer->addWriteFD(pipes[1],
- writeCB,
- &writeCBCalled,
- &ttd);
- /* we can't add it twice */
- BOOST_CHECK_THROW(mplexer->addWriteFD(pipes[1],
- writeCB,
- &writeCBCalled,
- &ttd),
- FDMultiplexerException);
-
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
- BOOST_CHECK_EQUAL(readyFDs.at(0), pipes[1]);
-
- ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 1);
- BOOST_CHECK_EQUAL(writeCBCalled, true);
-
- /* no read timeouts */
- timeouts = mplexer->getTimeouts(now, false);
- BOOST_CHECK_EQUAL(timeouts.size(), 0U);
- /* but we should have a write one */
- timeouts = mplexer->getTimeouts(now, true);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
- BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[1]);
-
- /* can't remove from the wrong type of FD */
- BOOST_CHECK_THROW(mplexer->removeReadFD(pipes[1]), FDMultiplexerException);
- mplexer->removeWriteFD(pipes[1]);
- /* can't remove a non-existing FD */
- BOOST_CHECK_THROW(mplexer->removeWriteFD(pipes[0]), FDMultiplexerException);
- BOOST_CHECK_THROW(mplexer->removeWriteFD(pipes[1]), FDMultiplexerException);
-
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 0U);
-
- ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 0);
+BOOST_AUTO_TEST_CASE(test_MPlexer)
+{
+ for (const auto& entry : FDMultiplexer::getMultiplexerMap()) {
+ auto mplexer = std::unique_ptr<FDMultiplexer>(entry.second());
+ BOOST_REQUIRE(mplexer != nullptr);
+ //cerr<<"Testing multiplexer "<<mplexer->getName()<<endl;
+
+ struct timeval now{0,0};
+ int ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 0);
+ BOOST_CHECK(now.tv_sec != 0);
+
+ std::vector<int> readyFDs;
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_CHECK_EQUAL(readyFDs.size(), 0U);
+
+ auto timeouts = mplexer->getTimeouts(now);
+ BOOST_CHECK_EQUAL(timeouts.size(), 0U);
+
+ int pipes[2];
+ int res = pipe(pipes);
+ BOOST_REQUIRE_EQUAL(res, 0);
+ BOOST_REQUIRE_EQUAL(setNonBlocking(pipes[0]), true);
+ BOOST_REQUIRE_EQUAL(setNonBlocking(pipes[1]), true);
+
+ /* let's declare a TTD that expired 5s ago */
+ struct timeval ttd = now;
+ ttd.tv_sec -= 5;
+
+ bool writeCBCalled = false;
+ auto writeCB = [](int fd, FDMultiplexer::funcparam_t param) {
+ auto calledPtr = boost::any_cast<bool*>(param);
+ BOOST_REQUIRE(calledPtr != nullptr);
+ *calledPtr = true;
+ };
+ mplexer->addWriteFD(pipes[1],
+ writeCB,
+ &writeCBCalled,
+ &ttd);
+ /* we can't add it twice */
+ BOOST_CHECK_THROW(mplexer->addWriteFD(pipes[1],
+ writeCB,
+ &writeCBCalled,
+ &ttd),
+ FDMultiplexerException);
+
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
+ BOOST_CHECK_EQUAL(readyFDs.at(0), pipes[1]);
- bool readCBCalled = false;
- auto readCB = [](int fd, FDMultiplexer::funcparam_t param) {
- auto calledPtr = boost::any_cast<bool*>(param);
- BOOST_REQUIRE(calledPtr != nullptr);
- *calledPtr = true;
- };
- mplexer->addReadFD(pipes[0],
- readCB,
- &readCBCalled,
- &ttd);
-
- /* not ready for reading yet */
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 0U);
-
- ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 0);
- BOOST_CHECK_EQUAL(readCBCalled, false);
-
- /* let's make the pipe readable */
- BOOST_REQUIRE_EQUAL(write(pipes[1], "0", 1), 1);
-
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
- BOOST_CHECK_EQUAL(readyFDs.at(0), pipes[0]);
-
- ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 1);
- BOOST_CHECK_EQUAL(readCBCalled, true);
-
- /* add back the write FD */
- mplexer->addWriteFD(pipes[1],
- writeCB,
- &writeCBCalled,
- &ttd);
-
- /* both should be available */
- readCBCalled = false;
- writeCBCalled = false;
- readyFDs.clear();
-
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_GT(readyFDs.size(), 0U);
- if (readyFDs.size() == 2) {
- ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 2);
- }
- else if (readyFDs.size() == 1) {
- /* under high pressure (lots of existing pipes on the system, for example,
- the pipe might only have room for one 'buffer' and will not be writable
- after our write of 1 byte, we need to read it so that the pipe becomes
- writable again */
- /* make sure the pipe is readable, otherwise something is off */
- BOOST_REQUIRE_EQUAL(readyFDs.at(0), pipes[0]);
ready = mplexer->run(&now, 100);
BOOST_CHECK_EQUAL(ready, 1);
- BOOST_CHECK_EQUAL(readCBCalled, true);
- BOOST_CHECK_EQUAL(writeCBCalled, false);
- char buffer[1];
- ssize_t got = read(pipes[0], &buffer[0], sizeof(buffer));
- BOOST_CHECK_EQUAL(got, 1U);
+ BOOST_CHECK_EQUAL(writeCBCalled, true);
+
+ /* no read timeouts */
+ timeouts = mplexer->getTimeouts(now, false);
+ BOOST_CHECK_EQUAL(timeouts.size(), 0U);
+ /* but we should have a write one */
+ timeouts = mplexer->getTimeouts(now, true);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
+ BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[1]);
+
+ /* can't remove from the wrong type of FD */
+ BOOST_CHECK_THROW(mplexer->removeReadFD(pipes[1]), FDMultiplexerException);
+ mplexer->removeWriteFD(pipes[1]);
+ /* can't remove a non-existing FD */
+ BOOST_CHECK_THROW(mplexer->removeWriteFD(pipes[0]), FDMultiplexerException);
+ BOOST_CHECK_THROW(mplexer->removeWriteFD(pipes[1]), FDMultiplexerException);
- /* ok, the pipe should be writable now, but not readable */
readyFDs.clear();
mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_CHECK_EQUAL(readyFDs.size(), 1U);
- BOOST_REQUIRE_EQUAL(readyFDs.at(0), pipes[1]);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 0U);
+
+ ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 0);
+
+ bool readCBCalled = false;
+ auto readCB = [](int fd, FDMultiplexer::funcparam_t param) {
+ auto calledPtr = boost::any_cast<bool*>(param);
+ BOOST_REQUIRE(calledPtr != nullptr);
+ *calledPtr = true;
+ };
+ mplexer->addReadFD(pipes[0],
+ readCB,
+ &readCBCalled,
+ &ttd);
+
+ /* not ready for reading yet */
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 0U);
+
+ ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 0);
+ BOOST_CHECK_EQUAL(readCBCalled, false);
+
+ /* let's make the pipe readable */
+ BOOST_REQUIRE_EQUAL(write(pipes[1], "0", 1), 1);
+
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
+ BOOST_CHECK_EQUAL(readyFDs.at(0), pipes[0]);
ready = mplexer->run(&now, 100);
BOOST_CHECK_EQUAL(ready, 1);
- }
+ BOOST_CHECK_EQUAL(readCBCalled, true);
+
+ /* add back the write FD */
+ mplexer->addWriteFD(pipes[1],
+ writeCB,
+ &writeCBCalled,
+ &ttd);
+
+ /* both should be available */
+ readCBCalled = false;
+ writeCBCalled = false;
+ readyFDs.clear();
- BOOST_CHECK_EQUAL(readCBCalled, true);
- BOOST_CHECK_EQUAL(writeCBCalled, true);
-
- /* both the read and write FD should be reported */
- timeouts = mplexer->getTimeouts(now, false);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
- BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[0]);
- timeouts = mplexer->getTimeouts(now, true);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
- BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[1]);
-
- struct timeval past = ttd;
- /* so five seconds before the actual TTD */
- past.tv_sec -= 5;
-
- /* no read timeouts */
- timeouts = mplexer->getTimeouts(past, false);
- BOOST_CHECK_EQUAL(timeouts.size(), 0U);
- /* and we should not have a write one either */
- timeouts = mplexer->getTimeouts(past, true);
- BOOST_CHECK_EQUAL(timeouts.size(), 0U);
-
- /* update the timeouts to now, they should not be reported anymore */
- mplexer->setReadTTD(pipes[0], now, 0);
- mplexer->setWriteTTD(pipes[1], now, 0);
- timeouts = mplexer->getTimeouts(now, false);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 0U);
- timeouts = mplexer->getTimeouts(now, true);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 0U);
-
- /* put it back into the past */
- mplexer->setReadTTD(pipes[0], now, -5);
- mplexer->setWriteTTD(pipes[1], now, -5);
- timeouts = mplexer->getTimeouts(now, false);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
- BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[0]);
- timeouts = mplexer->getTimeouts(now, true);
- BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
- BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[1]);
-
- mplexer->removeReadFD(pipes[0]);
- mplexer->removeWriteFD(pipes[1]);
-
- /* clean up */
- close(pipes[0]);
- close(pipes[1]);
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_GT(readyFDs.size(), 0U);
+ if (readyFDs.size() == 2) {
+ ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 2);
+ }
+ else if (readyFDs.size() == 1) {
+ /* under high pressure (lots of existing pipes on the system, for example,
+ the pipe might only have room for one 'buffer' and will not be writable
+ after our write of 1 byte, we need to read it so that the pipe becomes
+ writable again */
+ /* make sure the pipe is readable, otherwise something is off */
+ BOOST_REQUIRE_EQUAL(readyFDs.at(0), pipes[0]);
+ ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 1);
+ BOOST_CHECK_EQUAL(readCBCalled, true);
+ BOOST_CHECK_EQUAL(writeCBCalled, false);
+ char buffer[1];
+ ssize_t got = read(pipes[0], &buffer[0], sizeof(buffer));
+ BOOST_CHECK_EQUAL(got, 1U);
+
+ /* ok, the pipe should be writable now, but not readable */
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_CHECK_EQUAL(readyFDs.size(), 1U);
+ BOOST_REQUIRE_EQUAL(readyFDs.at(0), pipes[1]);
+
+ ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 1);
+ }
+
+ BOOST_CHECK_EQUAL(readCBCalled, true);
+ BOOST_CHECK_EQUAL(writeCBCalled, true);
+
+ /* both the read and write FD should be reported */
+ timeouts = mplexer->getTimeouts(now, false);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
+ BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[0]);
+ timeouts = mplexer->getTimeouts(now, true);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
+ BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[1]);
+
+ struct timeval past = ttd;
+ /* so five seconds before the actual TTD */
+ past.tv_sec -= 5;
+
+ /* no read timeouts */
+ timeouts = mplexer->getTimeouts(past, false);
+ BOOST_CHECK_EQUAL(timeouts.size(), 0U);
+ /* and we should not have a write one either */
+ timeouts = mplexer->getTimeouts(past, true);
+ BOOST_CHECK_EQUAL(timeouts.size(), 0U);
+
+ /* update the timeouts to now, they should not be reported anymore */
+ mplexer->setReadTTD(pipes[0], now, 0);
+ mplexer->setWriteTTD(pipes[1], now, 0);
+ timeouts = mplexer->getTimeouts(now, false);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 0U);
+ timeouts = mplexer->getTimeouts(now, true);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 0U);
+
+ /* put it back into the past */
+ mplexer->setReadTTD(pipes[0], now, -5);
+ mplexer->setWriteTTD(pipes[1], now, -5);
+ timeouts = mplexer->getTimeouts(now, false);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
+ BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[0]);
+ timeouts = mplexer->getTimeouts(now, true);
+ BOOST_REQUIRE_EQUAL(timeouts.size(), 1U);
+ BOOST_CHECK_EQUAL(timeouts.at(0).first, pipes[1]);
+
+ mplexer->removeReadFD(pipes[0]);
+ mplexer->removeWriteFD(pipes[1]);
+
+ /* clean up */
+ close(pipes[0]);
+ close(pipes[1]);
+ }
}
BOOST_AUTO_TEST_CASE(test_MPlexer_ReadAndWrite)
{
- auto mplexer = std::unique_ptr<FDMultiplexer>(FDMultiplexer::getMultiplexerSilent());
- BOOST_REQUIRE(mplexer != nullptr);
+ for (const auto& entry : FDMultiplexer::getMultiplexerMap()) {
+ auto mplexer = std::unique_ptr<FDMultiplexer>(entry.second());
+ BOOST_REQUIRE(mplexer != nullptr);
+ //cerr<<"Testing multiplexer "<<mplexer->getName()<<" for read AND write"<<endl;
+
+ int sockets[2];
+ int res = socketpair(AF_UNIX, SOCK_STREAM, 0, sockets);
+ BOOST_REQUIRE_EQUAL(res, 0);
+ BOOST_REQUIRE_EQUAL(setNonBlocking(sockets[0]), true);
+ BOOST_REQUIRE_EQUAL(setNonBlocking(sockets[1]), true);
+
+ struct timeval now;
+ std::vector<int> readyFDs;
+ struct timeval ttd = now;
+ ttd.tv_sec += 5;
+
+ bool readCBCalled = false;
+ bool writeCBCalled = false;
+ auto readCB = [](int fd, FDMultiplexer::funcparam_t param) {
+ auto calledPtr = boost::any_cast<bool*>(param);
+ BOOST_REQUIRE(calledPtr != nullptr);
+ *calledPtr = true;
+ };
+ auto writeCB = [](int fd, FDMultiplexer::funcparam_t param) {
+ auto calledPtr = boost::any_cast<bool*>(param);
+ BOOST_REQUIRE(calledPtr != nullptr);
+ *calledPtr = true;
+ };
+ mplexer->addReadFD(sockets[0],
+ readCB,
+ &readCBCalled,
+ &ttd);
+ mplexer->addWriteFD(sockets[0],
+ writeCB,
+ &writeCBCalled,
+ &ttd);
+
+ /* not ready for reading yet, but should be writable */
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
+ BOOST_CHECK_EQUAL(readyFDs.at(0), sockets[0]);
+
+ /* let's make the socket readable */
+ BOOST_REQUIRE_EQUAL(write(sockets[1], "0", 1), 1);
- int sockets[2];
- int res = socketpair(AF_UNIX, SOCK_STREAM, 0, sockets);
- BOOST_REQUIRE_EQUAL(res, 0);
- BOOST_REQUIRE_EQUAL(setNonBlocking(sockets[0]), true);
- BOOST_REQUIRE_EQUAL(setNonBlocking(sockets[1]), true);
-
- struct timeval now;
- std::vector<int> readyFDs;
- struct timeval ttd = now;
- ttd.tv_sec += 5;
-
- bool readCBCalled = false;
- bool writeCBCalled = false;
- auto readCB = [](int fd, FDMultiplexer::funcparam_t param) {
- auto calledPtr = boost::any_cast<bool*>(param);
- BOOST_REQUIRE(calledPtr != nullptr);
- *calledPtr = true;
- };
- auto writeCB = [](int fd, FDMultiplexer::funcparam_t param) {
- auto calledPtr = boost::any_cast<bool*>(param);
- BOOST_REQUIRE(calledPtr != nullptr);
- *calledPtr = true;
- };
- mplexer->addReadFD(sockets[0],
- readCB,
- &readCBCalled,
- &ttd);
- mplexer->addWriteFD(sockets[0],
- writeCB,
- &writeCBCalled,
- &ttd);
-
- /* not ready for reading yet, but should be writable */
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
- BOOST_CHECK_EQUAL(readyFDs.at(0), sockets[0]);
-
- /* let's make the socket readable */
- BOOST_REQUIRE_EQUAL(write(sockets[1], "0", 1), 1);
-
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
- BOOST_CHECK_EQUAL(readyFDs.at(0), sockets[0]);
-
- auto ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 2);
- BOOST_CHECK_EQUAL(readCBCalled, true);
- BOOST_CHECK_EQUAL(writeCBCalled, true);
-
- /* check that the write cb remains when we remove the read one */
- mplexer->removeReadFD(sockets[0]);
-
- readCBCalled = false;
- writeCBCalled = false;
- readyFDs.clear();
- mplexer->getAvailableFDs(readyFDs, 0);
- BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
- BOOST_CHECK_EQUAL(readyFDs.at(0), sockets[0]);
- ready = mplexer->run(&now, 100);
- BOOST_CHECK_EQUAL(ready, 1);
- BOOST_CHECK_EQUAL(readCBCalled, false);
- BOOST_CHECK_EQUAL(writeCBCalled, true);
-
- mplexer->removeWriteFD(sockets[0]);
-
- /* clean up */
- close(sockets[0]);
- close(sockets[1]);
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
+ BOOST_CHECK_EQUAL(readyFDs.at(0), sockets[0]);
+
+ auto ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 2);
+ BOOST_CHECK_EQUAL(readCBCalled, true);
+ BOOST_CHECK_EQUAL(writeCBCalled, true);
+
+ /* check that the write cb remains when we remove the read one */
+ mplexer->removeReadFD(sockets[0]);
+
+ readCBCalled = false;
+ writeCBCalled = false;
+ readyFDs.clear();
+ mplexer->getAvailableFDs(readyFDs, 0);
+ BOOST_REQUIRE_EQUAL(readyFDs.size(), 1U);
+ BOOST_CHECK_EQUAL(readyFDs.at(0), sockets[0]);
+ ready = mplexer->run(&now, 100);
+ BOOST_CHECK_EQUAL(ready, 1);
+ BOOST_CHECK_EQUAL(readCBCalled, false);
+ BOOST_CHECK_EQUAL(writeCBCalled, true);
+
+ mplexer->removeWriteFD(sockets[0]);
+
+ /* clean up */
+ close(sockets[0]);
+ close(sockets[1]);
+ }
}
BOOST_AUTO_TEST_SUITE_END()
projects / thirdparty / pdns.git / commitdiff
