/src/protocols/ntp/test_ntp.cc

https://bitbucket.org/camp0/aiengine · C++ · 220 lines · 145 code · 52 blank · 23 comment · 87 complexity · 0594462be8c2f8c513043b6e7d3b3905 MD5 · raw file 

    

  1. /*
    2. 

  2.  * AIEngine a new generation network intrusion detection system.
    3. 

  3.  *
    4. 

  4.  * Copyright (C) 2013-2022  Luis Campo Giralte
    5. 

  5.  *
    6. 

  6.  * This library is free software; you can redistribute it and/or
    7. 

  7.  * modify it under the terms of the GNU Library General Public
    8. 

  8.  * License as published by the Free Software Foundation; either
    9. 

  9.  * version 2 of the License, or (at your option) any later version.
    10. 

  10.  *
    11. 

  11.  * This library is distributed in the hope that it will be useful,
    12. 

  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    13. 

  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    14. 

  14.  * Library General Public License for more details.
    15. 

  15.  *
    16. 

  16.  * You should have received a copy of the GNU Library General Public
    17. 

  17.  * License along with this library; if not, write to the
    18. 

  18.  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
    19. 

  19.  * Boston, MA  02110-1301, USA.
    20. 

  20.  *
    21. 

  21.  * Written by Luis Campo Giralte <luis.camp0.2009@gmail.com> 
    22. 

  22.  *
    23. 

  23.  */
    24. 

  24. #include "test_ntp.h"
    25. 

  25. 

  26. #define BOOST_TEST_DYN_LINK
    27. 

  27. #ifdef STAND_ALONE_TEST
    28. 

  28. #define BOOST_TEST_MODULE ntptest
    29. 

  29. #endif
    30. 

  30. #include <boost/test/unit_test.hpp>
    31. 

  31. 

  32. using namespace aiengine;
    33. 

  33. 

  34. BOOST_FIXTURE_TEST_SUITE(ntp_test_suite, StackNTPtest)
    35. 

  35. 

  36. BOOST_AUTO_TEST_CASE (test01)
    37. 

  37. {
    38. 

  38. 	Packet packet;
    39. 

  39. 

  40.         BOOST_CHECK(ntp->getTotalPackets() == 0);
    41. 

  41.         BOOST_CHECK(ntp->getTotalValidPackets() == 0);
    42. 

  42.         BOOST_CHECK(ntp->getTotalBytes() == 0);
    43. 

  43.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    44. 

  44. 	BOOST_CHECK(ntp->processPacket(packet) == true);
    45. 

  45. 	
    46. 

  46. 	CounterMap c = ntp->getCounters();
    47. 

  47. }
    48. 

  48. 

  49. BOOST_AUTO_TEST_CASE (test02)
    50. 

  50. {
    51. 

  51. 	Packet packet("../ntp/packets/packet01.pcap");
    52. 

  52. 

  53. 	inject(packet);
    54. 

  54. 

  55.         BOOST_CHECK(ip->getTotalPackets() == 1);
    56. 

  56.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    57. 

  57.         BOOST_CHECK(ip->getTotalBytes() == 96);
    58. 

  58.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    59. 

  59. 

  60.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    61. 

  61.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    62. 

  62.         BOOST_CHECK(ntp->getTotalBytes() == 68);
    63. 

  63.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    64. 

  64. 	BOOST_CHECK(ntp->getVersion() == 2);	
    65. 

  65. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_CLIENT);
    66. 

  66. }
    67. 

  67. 

  68. BOOST_AUTO_TEST_CASE (test03)
    69. 

  69. {
    70. 

  70. 	Packet packet("../ntp/packets/packet02.pcap");
    71. 

  71. 

  72. 	inject(packet);
    73. 

  73. 

  74.         BOOST_CHECK(ip->getTotalPackets() == 1);
    75. 

  75.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    76. 

  76.         BOOST_CHECK(ip->getTotalBytes() == 96);
    77. 

  77.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    78. 

  78. 

  79.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    80. 

  80.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    81. 

  81.         BOOST_CHECK(ntp->getTotalBytes() == 68);
    82. 

  82.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    83. 

  83. 

  84. 	BOOST_CHECK(ntp->getVersion() == 3);	
    85. 

  85. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_SERVER);
    86. 

  86. 

  87. 	Json j;
    88. 

  88. 

  89. 	ntp->statistics(j, 5);
    90. 

  90. 

  91. 	BOOST_CHECK(j["servers"] == 1);
    92. 

  92. 

  93. 	ntp->resetCounters();
    94. 

  94. 

  95. 	ntp->statistics(j, 5);
    96. 

  96. 

  97. 	BOOST_CHECK(j["servers"] == 0);
    98. 

  98. }
    99. 

  99. 

  100. BOOST_AUTO_TEST_CASE (test04)
    101. 

  101. {
    102. 

  102. 	Packet packet("../ntp/packets/packet03.pcap");
    103. 

  103. 

  104. 	inject(packet);
    105. 

  105. 

  106.         BOOST_CHECK(ip->getTotalPackets() == 1);
    107. 

  107.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    108. 

  108.         BOOST_CHECK(ip->getTotalBytes() == 76);
    109. 

  109.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    110. 

  110. 

  111.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    112. 

  112.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    113. 

  113.         BOOST_CHECK(ntp->getTotalBytes() == 48);
    114. 

  114.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    115. 

  115. 

  116.         BOOST_CHECK(ntp->getVersion() == 4);
    117. 

  117.         BOOST_CHECK(ntp->getMode() == NTP_MODE_CLIENT);
    118. 

  118. }
    119. 

  119. 

  120. BOOST_AUTO_TEST_CASE (test05)
    121. 

  121. {
    122. 

  122. 	Packet packet("../ntp/packets/packet04.pcap");
    123. 

  123. 

  124. 	inject(packet);
    125. 

  125. 

  126.         BOOST_CHECK(ip->getTotalPackets() == 1);
    127. 

  127.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    128. 

  128.         BOOST_CHECK(ip->getTotalBytes() == 76);
    129. 

  129.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    130. 

  130. 

  131.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    132. 

  132.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    133. 

  133.         BOOST_CHECK(ntp->getTotalBytes() == 48);
    134. 

  134.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    135. 

  135. 

  136. 	BOOST_CHECK(ntp->getVersion() == 3);	
    137. 

  137. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_SYM_ACT);
    138. 

  138. }
    139. 

  139. 

  140. BOOST_AUTO_TEST_CASE (test06)
    141. 

  141. {
    142. 

  142. 	Packet packet("../ntp/packets/packet05.pcap");
    143. 

  143. 

  144. 	inject(packet);
    145. 

  145. 

  146.         BOOST_CHECK(ip->getTotalPackets() == 1);
    147. 

  147.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    148. 

  148.         BOOST_CHECK(ip->getTotalBytes() == 76);
    149. 

  149.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    150. 

  150. 

  151.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    152. 

  152.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    153. 

  153.         BOOST_CHECK(ntp->getTotalBytes() == 48);
    154. 

  154.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    155. 

  155. 

  156. 	BOOST_CHECK(ntp->getVersion() == 3);	
    157. 

  157. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_SYM_PAS);
    158. 

  158. }
    159. 

  159. 

  160. BOOST_AUTO_TEST_CASE (test07)
    161. 

  161. {
    162. 

  162. 	Packet packet("../ntp/packets/packet06.pcap");
    163. 

  163. 

  164. 	inject(packet);
    165. 

  165. 

  166.         BOOST_CHECK(ip->getTotalPackets() == 1);
    167. 

  167.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    168. 

  168.         BOOST_CHECK(ip->getTotalBytes() == 468);
    169. 

  169.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    170. 

  170. 

  171.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    172. 

  172.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    173. 

  173.         BOOST_CHECK(ntp->getTotalBytes() == 440);
    174. 

  174.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    175. 

  175. 

  176. 	BOOST_CHECK(ntp->getVersion() == 2);	
    177. 

  177. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_RES2);
    178. 

  178. }
    179. 

  179. 

  180. BOOST_AUTO_TEST_CASE (test08) // ntp broadcast
    181. 

  181. {
    182. 

  182. 	Packet packet("../ntp/packets/packet07.pcap");
    183. 

  183. 

  184. 	inject(packet);
    185. 

  185. 

  186.         BOOST_CHECK(ip->getTotalPackets() == 1);
    187. 

  187.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    188. 

  188.         BOOST_CHECK(ip->getTotalBytes() == 76);
    189. 

  189.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    190. 

  190. 

  191.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    192. 

  192.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    193. 

  193.         BOOST_CHECK(ntp->getTotalBytes() == 48);
    194. 

  194.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    195. 

  195. 

  196. 	BOOST_CHECK(ntp->getVersion() == 3);	
    197. 

  197. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_BROADCAST);
    198. 

  198. }
    199. 

  199. 

  200. BOOST_AUTO_TEST_CASE (test09) // reserved unspec
    201. 

  201. {
    202. 

  202. 	Packet packet("../ntp/packets/packet08.pcap");
    203. 

  203. 

  204. 	inject(packet);
    205. 

  205. 

  206.         BOOST_CHECK(ip->getTotalPackets() == 1);
    207. 

  207.         BOOST_CHECK(ip->getTotalValidPackets() == 1);
    208. 

  208.         BOOST_CHECK(ip->getTotalBytes() == 76);
    209. 

  209.         BOOST_CHECK(ip->getTotalInvalidPackets() == 0);
    210. 

  210. 

  211.         BOOST_CHECK(ntp->getTotalPackets() == 1);
    212. 

  212.         BOOST_CHECK(ntp->getTotalValidPackets() == 1);
    213. 

  213.         BOOST_CHECK(ntp->getTotalBytes() == 48);
    214. 

  214.         BOOST_CHECK(ntp->getTotalInvalidPackets() == 0);
    215. 

  215. 

  216. 	BOOST_CHECK(ntp->getVersion() == 3);	
    217. 

  217. 	BOOST_CHECK(ntp->getMode() == NTP_MODE_UNSPEC);
    218. 

  218. }
    219. 

  219. 

  220. BOOST_AUTO_TEST_SUITE_END()
    221.