Junxiao Shi
142 lines
4.9 KiB
C++
142 lines
4.9 KiB
C++
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
|
|
/*
|
|
* Copyright (c) 2014-2023, The University of Memphis
|
|
*
|
|
* This file is part of PSync.
|
|
* See AUTHORS.md for complete list of PSync authors and contributors.
|
|
*
|
|
* PSync is free software: you can redistribute it and/or modify it under the terms
|
|
* of the GNU Lesser General Public License as published by the Free Software Foundation,
|
|
* either version 3 of the License, or (at your option) any later version.
|
|
*
|
|
* PSync is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
|
|
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
* PURPOSE. See the GNU Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License along with
|
|
* PSync, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
|
|
**/
|
|
|
|
#include "PSync/partial-producer.hpp"
|
|
|
|
#include "tests/boost-test.hpp"
|
|
#include "tests/key-chain-fixture.hpp"
|
|
|
|
#include <ndn-cxx/mgmt/nfd/control-parameters.hpp>
|
|
#include <ndn-cxx/util/dummy-client-face.hpp>
|
|
|
|
namespace psync {
|
|
|
|
using ndn::Interest;
|
|
using ndn::Name;
|
|
|
|
class PartialProducerFixture : public tests::KeyChainFixture
|
|
{
|
|
protected:
|
|
ndn::DummyClientFace m_face{m_keyChain, {true, true}};
|
|
};
|
|
|
|
BOOST_FIXTURE_TEST_SUITE(TestPartialProducer, PartialProducerFixture)
|
|
|
|
BOOST_AUTO_TEST_CASE(RegisterPrefix)
|
|
{
|
|
Name syncPrefix("/psync"), userNode("/testUser");
|
|
PartialProducer producer(m_face, m_keyChain, 40, syncPrefix, userNode);
|
|
|
|
m_face.processEvents(-1_ms);
|
|
|
|
BOOST_REQUIRE_EQUAL(m_face.sentInterests.size(), 1);
|
|
auto interest = m_face.sentInterests.front();
|
|
BOOST_CHECK_EQUAL(interest.getName().at(3), Name::Component("register"));
|
|
ndn::nfd::ControlParameters params(interest.getName().at(4).blockFromValue());
|
|
BOOST_CHECK_EQUAL(params.getName(), syncPrefix);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(PublishName)
|
|
{
|
|
Name syncPrefix("/psync"), userNode("/testUser"), nonUser("/testUser2");
|
|
PartialProducer producer(m_face, m_keyChain, 40, syncPrefix, userNode);
|
|
|
|
BOOST_CHECK_EQUAL(producer.getSeqNo(userNode).value_or(-1), 0);
|
|
producer.publishName(userNode);
|
|
BOOST_CHECK_EQUAL(producer.getSeqNo(userNode).value_or(-1), 1);
|
|
|
|
producer.publishName(userNode);
|
|
BOOST_CHECK_EQUAL(producer.getSeqNo(userNode).value_or(-1), 2);
|
|
|
|
producer.publishName(userNode, 10);
|
|
BOOST_CHECK_EQUAL(producer.getSeqNo(userNode).value_or(-1), 10);
|
|
|
|
producer.publishName(nonUser);
|
|
BOOST_CHECK_EQUAL(producer.getSeqNo(nonUser).value_or(-1), -1);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(SameSyncInterest)
|
|
{
|
|
Name syncPrefix("/psync"), userNode("/testUser");
|
|
PartialProducer producer(m_face, m_keyChain, 40, syncPrefix, userNode);
|
|
|
|
Name syncInterestName(syncPrefix);
|
|
syncInterestName.append("sync");
|
|
Name syncInterestPrefix = syncInterestName;
|
|
|
|
detail::BloomFilter bf(20, 0.001);
|
|
bf.appendToName(syncInterestName);
|
|
|
|
producer.m_iblt.appendToName(syncInterestName);
|
|
|
|
Interest syncInterest(syncInterestName);
|
|
syncInterest.setInterestLifetime(1_s);
|
|
syncInterest.setNonce(1);
|
|
BOOST_CHECK_NO_THROW(producer.onSyncInterest(syncInterestPrefix, syncInterest));
|
|
m_face.processEvents(10_ms);
|
|
BOOST_CHECK_EQUAL(producer.m_pendingEntries.size(), 1);
|
|
|
|
m_face.processEvents(500_ms);
|
|
|
|
// Same interest again - size of pending interest should remain same, but expirationEvent should change
|
|
syncInterest.setNonce(2);
|
|
BOOST_CHECK_NO_THROW(producer.onSyncInterest(syncInterestPrefix, syncInterest));
|
|
m_face.processEvents(10_ms);
|
|
BOOST_CHECK_EQUAL(producer.m_pendingEntries.size(), 1);
|
|
|
|
m_face.processEvents(500_ms);
|
|
BOOST_CHECK_EQUAL(producer.m_pendingEntries.size(), 1);
|
|
|
|
m_face.processEvents(500_ms);
|
|
BOOST_CHECK_EQUAL(producer.m_pendingEntries.size(), 0);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(OnSyncInterest)
|
|
{
|
|
Name syncPrefix("/psync"), userNode("/testUser");
|
|
PartialProducer producer(m_face, m_keyChain, 40, syncPrefix, userNode);
|
|
|
|
// Sync interest with no bloom filter attached
|
|
Name syncInterestName(syncPrefix);
|
|
syncInterestName.append("sync");
|
|
producer.m_iblt.appendToName(syncInterestName);
|
|
BOOST_CHECK_NO_THROW(producer.onSyncInterest(syncInterestName, Interest(syncInterestName)));
|
|
|
|
// Sync interest with malicious bloom filter
|
|
syncInterestName = syncPrefix;
|
|
syncInterestName.append("sync");
|
|
syncInterestName.appendNumber(20); // count of bloom filter
|
|
syncInterestName.appendNumber(1); // false positive probability * 1000 of bloom filter
|
|
syncInterestName.append("fake-name");
|
|
producer.m_iblt.appendToName(syncInterestName);
|
|
BOOST_CHECK_NO_THROW(producer.onSyncInterest(syncInterestName, Interest(syncInterestName)));
|
|
|
|
// Sync interest with malicious IBF
|
|
syncInterestName = syncPrefix;
|
|
syncInterestName.append("sync");
|
|
detail::BloomFilter bf(20, 0.001);
|
|
bf.appendToName(syncInterestName);
|
|
syncInterestName.append("fake-name");
|
|
BOOST_CHECK_NO_THROW(producer.onSyncInterest(syncInterestName, Interest(syncInterestName)));
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|
|
|
|
} // namespace psync
|