src/test/DoS_tests.cpp

   1 //
   2 // Unit tests for denial-of-service detection/prevention code
   3 //
   4 #include <stdint.h>
   5
   6 #include <boost/assign/list_of.hpp> // for 'map_list_of()'
   7 #include <boost/test/unit_test.hpp>
   8 #include <boost/foreach.hpp>
   9
  10 #include "main.h"
  11 #include "wallet.h"
  12 #include "net.h"
  13 #include "util.h"
  14
  15 using namespace std;
  16
  17 BOOST_AUTO_TEST_SUITE(DoS_tests)
  18
  19 BOOST_AUTO_TEST_CASE(DoS_banning)
  20 {
  21     CNode::ClearBanned();
  22     CAddress addr1(0xa0b0c001);
  23     CNode dummyNode1(INVALID_SOCKET, addr1, true);
  24     dummyNode1.Misbehaving(100); // Should get banned
  25     BOOST_CHECK(CNode::IsBanned(addr1.ip));
  26     BOOST_CHECK(!CNode::IsBanned(addr1.ip|0x0000ff00)); // Different ip, not banned
  27
  28     CAddress addr2(0xa0b0c002);
  29     CNode dummyNode2(INVALID_SOCKET, addr2, true);
  30     dummyNode2.Misbehaving(50);
  31     BOOST_CHECK(!CNode::IsBanned(addr2.ip)); // 2 not banned yet...
  32     BOOST_CHECK(CNode::IsBanned(addr1.ip));  // ... but 1 still should be
  33     dummyNode2.Misbehaving(50);
  34     BOOST_CHECK(CNode::IsBanned(addr2.ip));
  35 }
  36
  37 BOOST_AUTO_TEST_CASE(DoS_banscore)
  38 {
  39     CNode::ClearBanned();
  40     mapArgs["-banscore"] = "111"; // because 11 is my favorite number
  41     CAddress addr1(0xa0b0c001);
  42     CNode dummyNode1(INVALID_SOCKET, addr1, true);
  43     dummyNode1.Misbehaving(100);
  44     BOOST_CHECK(!CNode::IsBanned(addr1.ip));
  45     dummyNode1.Misbehaving(10);
  46     BOOST_CHECK(!CNode::IsBanned(addr1.ip));
  47     dummyNode1.Misbehaving(1);
  48     BOOST_CHECK(CNode::IsBanned(addr1.ip));
  49     mapArgs["-banscore"] = "100";
  50 }
  51
  52 BOOST_AUTO_TEST_CASE(DoS_bantime)
  53 {
  54     CNode::ClearBanned();
  55     int64_t nStartTime = GetTime();
  56     SetMockTime(nStartTime); // Overrides future calls to GetTime()
  57
  58     CAddress addr(0xa0b0c001);
  59     CNode dummyNode(INVALID_SOCKET, addr, true);
  60
  61     dummyNode.Misbehaving(100);
  62     BOOST_CHECK(CNode::IsBanned(addr.ip));
  63
  64     SetMockTime(nStartTime+60*60);
  65     BOOST_CHECK(CNode::IsBanned(addr.ip));
  66
  67     SetMockTime(nStartTime+60*60*24+1);
  68     BOOST_CHECK(!CNode::IsBanned(addr.ip));
  69 }
  70
  71 static bool CheckNBits(unsigned int nbits1, int64_t time1, unsigned int nbits2, int64_t time2)
  72 {
  73     if (time1 > time2)
  74         return CheckNBits(nbits2, time2, nbits1, time1);
  75     int64_t deltaTime = time2-time1;
  76
  77     CBigNum required;
  78     required.SetCompact(ComputeMinWork(nbits1, deltaTime));
  79     CBigNum have;
  80     have.SetCompact(nbits2);
  81     return (have <= required);
  82 }
  83
  84 BOOST_AUTO_TEST_CASE(DoS_checknbits)
  85 {
  86     using namespace boost::assign; // for 'map_list_of()'
  87
  88     // Timestamps,nBits from the bitcoin blockchain.
  89     // These are the block-chain checkpoint blocks
  90     typedef std::map<int64_t, unsigned int> BlockData;
  91     BlockData chainData =
  92         map_list_of(1239852051,486604799)(1262749024,486594666)
  93         (1279305360,469854461)(1280200847,469830746)(1281678674,469809688)
  94         (1296207707,453179945)(1302624061,453036989)(1309640330,437004818)
  95         (1313172719,436789733);
  96
  97     // Make sure CheckNBits considers every combination of block-chain-lock-in-points
  98     // "sane":
  99     BOOST_FOREACH(const BlockData::value_type& i, chainData)
 100     {
 101         BOOST_FOREACH(const BlockData::value_type& j, chainData)
 102         {
 103             BOOST_CHECK(CheckNBits(i.second, i.first, j.second, j.first));
 104         }
 105     }
 106
 107     // Test a couple of insane combinations:
 108     BlockData::value_type firstcheck = *(chainData.begin());
 109     BlockData::value_type lastcheck = *(chainData.rbegin());
 110
 111     // First checkpoint difficulty at or a while after the last checkpoint time should fail when
 112     // compared to last checkpoint
 113     BOOST_CHECK(!CheckNBits(firstcheck.second, lastcheck.first+60*10, lastcheck.second, lastcheck.first));
 114     BOOST_CHECK(!CheckNBits(firstcheck.second, lastcheck.first+60*60*24*14, lastcheck.second, lastcheck.first));
 115
 116     // ... but OK if enough time passed for difficulty to adjust downward:
 117     BOOST_CHECK(CheckNBits(firstcheck.second, lastcheck.first+60*60*24*365*4, lastcheck.second, lastcheck.first));
 118
 119 }
 120
 121 BOOST_AUTO_TEST_SUITE_END()