#include "mDNSCoreReceiveTest.h"
#include "unittest_common.h"
int InitmDNSCoreReceiveTest(void);
int ValidQueryReqTest(void);
int NullDstQueryReqTest(void);
int ReceiveArpLogMsgTest(void);
void InitmDNSStorage(mDNS *const m);
// This DNS message was gleaned from a uDNS query request packet that was captured with Wireshark.
uint8_t udns_query_request_message[28] = { // contains 1 question for www.f5.com
0x31, 0xca, // transaction id
0x01, 0x00, // flags
0x00, 0x01, // 1 question
0x00, 0x00, // no anwsers
0x00, 0x00, // no authoritative answers
0x00, 0x00, // no additionals
0x03, 0x77, 0x77, 0x77, 0x02, 0x66, 0x35, 0x03, 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00, 0x01
};
// This DNS message was gleaned from a uDNS query request packet that was captured with Wireshark.
// Then the header id (more specifically, the msg->h.id) was deliberately cleared to force code
// path to traverse regression case, <rdar://problem/28556513>.
uint8_t udns_query_request_message_with_invalid_id[28] = { // contains 1 question for www.f5.com, msg->h.id = 0
0x00, 0x00, // transaction id
0x01, 0x00, // flags
0x00, 0x01, // 1 question
0x00, 0x00, // no anwsers
0x00, 0x00, // no authoritative answers
0x00, 0x00, // no additionals
0x03, 0x77, 0x77, 0x77, 0x02, 0x66, 0x35, 0x03, 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00, 0x01
};
uint8_t arp_request_packet[28] = { // contains 1 question for www.f5.com, msg->h.id = 0
0x00, 0x01, // hardware type: enet
0x08, 0x00, // protocol type: IP
0x06, // hardware size
0x04, // Protcol size
0x00, 0x01, // opcode request
0x24, 0x01, 0xc7, 0x24, 0x35, 0x00, // Sender mac addr
0x11, 0xe2, 0x14, 0x01, // Sender ip addr
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, // target mac addr
0x11, 0xe2, 0x17, 0xbe // target ip addr
};
UNITTEST_HEADER(mDNSCoreReceiveTest)
UNITTEST_TEST(InitmDNSCoreReceiveTest)
UNITTEST_TEST(ValidQueryReqTest)
UNITTEST_TEST(NullDstQueryReqTest)
UNITTEST_TEST(ReceiveArpLogMsgTest)
UNITTEST_FOOTER
UNITTEST_HEADER(InitmDNSCoreReceiveTest)
mDNSPlatformTimeInit();
init_logging_ut();
mDNS_LoggingEnabled = 0;
mDNS_PacketLoggingEnabled = 0;
UNITTEST_FOOTER
UNITTEST_HEADER(ReceiveArpLogMsgTest)
// Init unit test environment and verify no error occurred.
mStatus result = init_mdns_environment(mDNStrue);
UNITTEST_ASSERT(result == mStatus_NoError);
UNITTEST_ASSERT(result == mStatus_NoError);
ArpLogMsgTest(&mDNSStorage, (const ARP_EthIP *) arp_request_packet, primary_interfaceID);
UNITTEST_ASSERT(result == mStatus_NoError);
UNITTEST_FOOTER
UNITTEST_HEADER(ValidQueryReqTest)
mDNS *const m = &mDNSStorage;
mDNSAddr srcaddr, dstaddr;
mDNSIPPort srcport, dstport;
DNSMessage * msg;
const mDNSu8 * end;
// This test case does not require setup of interfaces, the record's cache, or pending questions
// so m is initialized to all zeros.
InitmDNSStorage(m);
// Used random values for srcaddr and srcport
srcaddr.type = mDNSAddrType_IPv4;
srcaddr.ip.v4.b[0] = 192;
srcaddr.ip.v4.b[1] = 168;
srcaddr.ip.v4.b[2] = 1;
srcaddr.ip.v4.b[3] = 10;
srcport.NotAnInteger = swap16((mDNSu16)53);
// Used random values for dstaddr and dstport
dstaddr.type = mDNSAddrType_IPv4;
dstaddr.ip.v4.b[0] = 192;
dstaddr.ip.v4.b[1] = 168;
dstaddr.ip.v4.b[2] = 1;
dstaddr.ip.v4.b[3] = 20;
dstport.NotAnInteger = swap16((mDNSu16)49339);
// Set message to a DNS message (copied from a WireShark packet)
msg = (DNSMessage *)udns_query_request_message;
end = udns_query_request_message + sizeof(udns_query_request_message);
// Execute mDNSCoreReceive using a valid DNS message
mDNSCoreReceive(m, msg, end, &srcaddr, srcport, &dstaddr, dstport, if_nametoindex("en0"));
// Verify that mDNSCoreReceiveQuery traversed the normal code path
UNITTEST_ASSERT(m->mDNSStats.NormalQueries == 1);
UNITTEST_FOOTER
UNITTEST_HEADER(NullDstQueryReqTest)
mDNS *const m = &mDNSStorage;
mDNSAddr srcaddr;
mDNSIPPort srcport, dstport;
DNSMessage * msg;
const mDNSu8 * end;
// This test case does not require setup of interfaces, the record's cache, or pending questions
// so m is initialized to all zeros.
InitmDNSStorage(m);
// Used random values for srcaddr and srcport
srcaddr.type = mDNSAddrType_IPv4;
srcaddr.ip.v4.b[0] = 192;
srcaddr.ip.v4.b[1] = 168;
srcaddr.ip.v4.b[2] = 1;
srcaddr.ip.v4.b[3] = 10;
srcport.NotAnInteger = swap16((mDNSu16)53);
// Used random value for dstport
dstport.NotAnInteger = swap16((mDNSu16)49339);
// Set message to a DNS message (copied from a WireShark packet)
msg = (DNSMessage *)udns_query_request_message_with_invalid_id;
end = udns_query_request_message_with_invalid_id + sizeof(udns_query_request_message_with_invalid_id);
// Execute mDNSCoreReceive to regress <rdar://problem/28556513>
mDNSCoreReceive(m, msg, end, &srcaddr, srcport, NULL, dstport, if_nametoindex("en0"));
// Verify that mDNSCoreReceiveQuery was NOT traversed through the normal code path
UNITTEST_ASSERT(m->mDNSStats.NormalQueries == 0);
// Verify code path that previously crashed, in <rdar://problem/28556513>, now traverses successfully
// by checking a counter that was incremented on code path that crashed.
UNITTEST_ASSERT(m->MPktNum == 1);
UNITTEST_FOOTER
void InitmDNSStorage(mDNS *const m)
{
memset(m, 0, sizeof(mDNS));
}
