11
11
#include " DnsLayer.h"
12
12
#include " SystemUtils.h"
13
13
14
+ using pcpp_tests::utils::createPacketAndBufferFromHexResource;
14
15
using pcpp_tests::utils::createPacketFromHexResource;
15
16
16
17
PTF_TEST_CASE (DnsLayerParsingTest)
@@ -222,12 +223,11 @@ PTF_TEST_CASE(DnsLayerParsingTest)
222
223
223
224
PTF_TEST_CASE (DnsLayerQueryCreationTest)
224
225
{
225
- timeval time;
226
- gettimeofday (&time, nullptr );
226
+ auto rawPacketAndBuf2 = createPacketAndBufferFromHexResource (" PacketExamples/DnsEdit2.dat" );
227
+ auto & resource2 = rawPacketAndBuf2.resourceBuffer ;
228
+ auto & rawPacket2 = rawPacketAndBuf2.packet ;
227
229
228
- READ_FILE_AND_CREATE_PACKET (2 , " PacketExamples/DnsEdit2.dat" );
229
-
230
- pcpp::Packet dnsEdit2RefPacket (&rawPacket2);
230
+ pcpp::Packet dnsEdit2RefPacket (rawPacket2.get ());
231
231
232
232
pcpp::EthLayer ethLayer2 (*dnsEdit2RefPacket.getLayerOfType <pcpp::EthLayer>());
233
233
pcpp::IPv4Layer ipLayer2 (*dnsEdit2RefPacket.getLayerOfType <pcpp::IPv4Layer>());
@@ -250,13 +250,14 @@ PTF_TEST_CASE(DnsLayerQueryCreationTest)
250
250
251
251
dnsEdit2Packet.computeCalculateFields ();
252
252
253
- PTF_ASSERT_EQUAL (bufferLength2, dnsEdit2Packet.getRawPacket ()->getRawDataLen ());
254
-
255
- PTF_ASSERT_BUF_COMPARE (dnsEdit2Packet.getRawPacket ()->getRawData (), buffer2, bufferLength2);
253
+ PTF_ASSERT_EQUAL (dnsEdit2Packet.getRawPacket ()->getRawDataLen (), resource2.length );
254
+ PTF_ASSERT_BUF_COMPARE (dnsEdit2Packet.getRawPacket ()->getRawData (), resource2.data .get (), resource2.length );
256
255
257
- READ_FILE_AND_CREATE_PACKET (1 , " PacketExamples/DnsEdit1.dat" );
256
+ auto rawPacketAndBuf1 = createPacketAndBufferFromHexResource (" PacketExamples/DnsEdit1.dat" );
257
+ auto & resource1 = rawPacketAndBuf1.resourceBuffer ;
258
+ auto & rawPacket1 = rawPacketAndBuf1.packet ;
258
259
259
- pcpp::Packet dnsEdit1RefPacket (& rawPacket1);
260
+ pcpp::Packet dnsEdit1RefPacket (rawPacket1. get () );
260
261
261
262
pcpp::EthLayer ethLayer1 (*dnsEdit1RefPacket.getLayerOfType <pcpp::EthLayer>());
262
263
pcpp::IPv4Layer ipLayer1 (*dnsEdit1RefPacket.getLayerOfType <pcpp::IPv4Layer>());
@@ -285,19 +286,17 @@ PTF_TEST_CASE(DnsLayerQueryCreationTest)
285
286
286
287
dnsEdit1Packet.computeCalculateFields ();
287
288
288
- PTF_ASSERT_EQUAL (dnsEdit1Packet.getRawPacket ()->getRawDataLen (), bufferLength1);
289
-
290
- PTF_ASSERT_BUF_COMPARE (dnsEdit1Packet.getRawPacket ()->getRawData (), buffer1, bufferLength1);
289
+ PTF_ASSERT_EQUAL (dnsEdit1Packet.getRawPacket ()->getRawDataLen (), resource1.length );
290
+ PTF_ASSERT_BUF_COMPARE (dnsEdit1Packet.getRawPacket ()->getRawData (), resource1.data .get (), resource1.length );
291
291
} // DnsLayerQueryCreationTest
292
292
293
293
PTF_TEST_CASE (DnsLayerResourceCreationTest)
294
294
{
295
- timeval time;
296
- gettimeofday (&time, nullptr );
295
+ auto rawPacketAndBuf4 = createPacketAndBufferFromHexResource (" PacketExamples/DnsEdit4.dat" );
296
+ auto & resource4 = rawPacketAndBuf4.resourceBuffer ;
297
+ auto & rawPacket4 = rawPacketAndBuf4.packet ;
297
298
298
- READ_FILE_AND_CREATE_PACKET (4 , " PacketExamples/DnsEdit4.dat" );
299
-
300
- pcpp::Packet dnsEdit4RefPacket (&rawPacket4);
299
+ pcpp::Packet dnsEdit4RefPacket (rawPacket4.get ());
301
300
302
301
pcpp::EthLayer ethLayer4 (*dnsEdit4RefPacket.getLayerOfType <pcpp::EthLayer>());
303
302
pcpp::IPv4Layer ipLayer4 (*dnsEdit4RefPacket.getLayerOfType <pcpp::IPv4Layer>());
@@ -356,13 +355,14 @@ PTF_TEST_CASE(DnsLayerResourceCreationTest)
356
355
357
356
dnsEdit4Packet.computeCalculateFields ();
358
357
359
- PTF_ASSERT_EQUAL (dnsEdit4Packet.getRawPacket ()->getRawDataLen (), bufferLength4);
360
-
361
- PTF_ASSERT_BUF_COMPARE (dnsEdit4Packet.getRawPacket ()->getRawData (), buffer4, bufferLength4);
358
+ PTF_ASSERT_EQUAL (dnsEdit4Packet.getRawPacket ()->getRawDataLen (), resource4.length );
359
+ PTF_ASSERT_BUF_COMPARE (dnsEdit4Packet.getRawPacket ()->getRawData (), resource4.data .get (), resource4.length );
362
360
363
- READ_FILE_AND_CREATE_PACKET (6 , " PacketExamples/DnsEdit6.dat" );
361
+ auto rawPacketAndBuf6 = createPacketAndBufferFromHexResource (" PacketExamples/DnsEdit6.dat" );
362
+ auto & resource6 = rawPacketAndBuf6.resourceBuffer ;
363
+ auto & rawPacket6 = rawPacketAndBuf6.packet ;
364
364
365
- pcpp::Packet dnsEdit6RefPacket (& rawPacket6);
365
+ pcpp::Packet dnsEdit6RefPacket (rawPacket6. get () );
366
366
367
367
pcpp::EthLayer ethLayer6 (*dnsEdit6RefPacket.getLayerOfType <pcpp::EthLayer>());
368
368
pcpp::IPv6Layer ipLayer6 (*dnsEdit6RefPacket.getLayerOfType <pcpp::IPv6Layer>());
@@ -419,13 +419,14 @@ PTF_TEST_CASE(DnsLayerResourceCreationTest)
419
419
420
420
dnsEdit6Packet.computeCalculateFields ();
421
421
422
- PTF_ASSERT_EQUAL (dnsEdit6Packet.getRawPacket ()->getRawDataLen (), bufferLength6);
422
+ PTF_ASSERT_EQUAL (dnsEdit6Packet.getRawPacket ()->getRawDataLen (), resource6.length );
423
+ PTF_ASSERT_BUF_COMPARE (dnsEdit6Packet.getRawPacket ()->getRawData (), resource6.data .get (), resource6.length );
423
424
424
- PTF_ASSERT_BUF_COMPARE (dnsEdit6Packet.getRawPacket ()->getRawData (), buffer6, bufferLength6);
425
+ auto rawPacketAndBuf7 = createPacketAndBufferFromHexResource (" PacketExamples/DnsEdit7.dat" );
426
+ auto & resource7 = rawPacketAndBuf7.resourceBuffer ;
427
+ auto & rawPacket7 = rawPacketAndBuf7.packet ;
425
428
426
- READ_FILE_AND_CREATE_PACKET (7 , " PacketExamples/DnsEdit7.dat" );
427
-
428
- pcpp::Packet dnsEdit7RefPacket (&rawPacket7);
429
+ pcpp::Packet dnsEdit7RefPacket (rawPacket7.get ());
429
430
430
431
pcpp::EthLayer ethLayer7 (*dnsEdit7RefPacket.getLayerOfType <pcpp::EthLayer>());
431
432
pcpp::IPv4Layer ipLayer7 (*dnsEdit7RefPacket.getLayerOfType <pcpp::IPv4Layer>());
@@ -467,9 +468,8 @@ PTF_TEST_CASE(DnsLayerResourceCreationTest)
467
468
468
469
dnsEdit7Packet.computeCalculateFields ();
469
470
470
- PTF_ASSERT_EQUAL (dnsEdit7Packet.getRawPacket ()->getRawDataLen (), bufferLength7);
471
-
472
- PTF_ASSERT_BUF_COMPARE (dnsEdit7Packet.getRawPacket ()->getRawData (), buffer7, bufferLength7);
471
+ PTF_ASSERT_EQUAL (dnsEdit7Packet.getRawPacket ()->getRawDataLen (), resource7.length );
472
+ PTF_ASSERT_BUF_COMPARE (dnsEdit7Packet.getRawPacket ()->getRawData (), resource7.data .get (), resource7.length );
473
473
} // DnsLayerResourceCreationTest
474
474
475
475
PTF_TEST_CASE (DnsLayerAddDnsKeyTest)
@@ -620,8 +620,6 @@ PTF_TEST_CASE(DnsLayerRemoveResourceTest)
620
620
621
621
PTF_TEST_CASE (DnsOverTcpParsingTest)
622
622
{
623
- using pcpp_tests::utils::createPacketFromHexResource;
624
-
625
623
auto rawPacket1 = createPacketFromHexResource (" PacketExamples/dns_over_tcp_query.dat" );
626
624
pcpp::Packet dnsPacket (rawPacket1.get ());
627
625
@@ -701,10 +699,9 @@ PTF_TEST_CASE(DnsOverTcpParsingTest)
701
699
702
700
PTF_TEST_CASE (DnsOverTcpCreationTest)
703
701
{
704
- timeval time;
705
- gettimeofday (&time, nullptr );
706
-
707
- READ_FILE_AND_CREATE_PACKET (1 , " PacketExamples/dns_over_tcp_answer2.dat" );
702
+ auto rawPacketAndBuf = createPacketAndBufferFromHexResource (" PacketExamples/dns_over_tcp_answer2.dat" );
703
+ auto & resource1 = rawPacketAndBuf.resourceBuffer ;
704
+ auto & rawPacket1 = rawPacketAndBuf.packet ;
708
705
709
706
pcpp::DnsOverTcpLayer newDnsLayer;
710
707
newDnsLayer.getDnsHeader ()->transactionID = htobe16 (0x38 );
@@ -715,14 +712,14 @@ PTF_TEST_CASE(DnsOverTcpCreationTest)
715
712
pcpp::IPv4DnsResourceData ipv4Answer (" 192.30.255.113" );
716
713
newDnsLayer.addAnswer (" github.com" , pcpp::DNS_TYPE_A, pcpp::DNS_CLASS_IN, 32 , &ipv4Answer);
717
714
718
- pcpp::Packet dnsPacket (& rawPacket1);
715
+ pcpp::Packet dnsPacket (rawPacket1. get () );
719
716
dnsPacket.removeLayer (pcpp::DNS);
720
717
721
718
dnsPacket.addLayer (&newDnsLayer);
722
719
dnsPacket.computeCalculateFields ();
723
720
724
- PTF_ASSERT_EQUAL (bufferLength1 , dnsPacket.getRawPacket ()->getRawDataLen ());
725
- PTF_ASSERT_BUF_COMPARE (dnsPacket.getRawPacket ()->getRawData (), buffer1, bufferLength1 );
721
+ PTF_ASSERT_EQUAL (resource1. length , dnsPacket.getRawPacket ()->getRawDataLen ());
722
+ PTF_ASSERT_BUF_COMPARE (dnsPacket.getRawPacket ()->getRawData (), resource1. data . get (), resource1. length );
726
723
} // DnsOverTcpCreationTest
727
724
728
725
PTF_TEST_CASE (DnsNXDomainTest)
0 commit comments