shaba/systemd
1
0
Fork 0
mirror of https://github.com/systemd/systemd.git synced 2024-10-27 10:25:37 +03:00
Code

resolved: tests for dns_packet_set_flags()

Browse Source
This commit is contained in:
James Coglan 2024-06-10 10:16:01 +01:00
parent ca93c4fe75
commit 6bfbacc9b4
2 changed files with 145 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/resolve/meson.build
View File

@ -165,6 +165,9 @@ executables += [
resolve_test_template + {
'sources' : files('test-dns-packet.c'),
},
resolve_test_template + {
'sources' : files('test-dns-packet-append.c'),
},
resolve_test_template + {
'sources' : files('test-dns-packet-extract.c'),
},

142
src/resolve/test-dns-packet-append.c Normal file
View File

@ -0,0 +1,142 @@
/* SPDX-License-Identifier: LGPL-2.1-or-later */
#include "dns-type.h"
#include "resolved-dns-packet.h"
#include "resolved-dns-rr.h"
#include "log.h"
#include "tests.h"
#define BIT_QR (1 << 7)
#define BIT_AA (1 << 2)
#define BIT_TC (1 << 1)
#define BIT_RD (1 << 0)
#define BIT_RA (1 << 7)
#define BIT_AD (1 << 5)
#define BIT_CD (1 << 4)
/* ================================================================
* dns_packet_set_flags()
* ================================================================ */
TEST(packet_set_flags_dns_checking_enabled) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new(&packet, DNS_PROTOCOL_DNS, 0, DNS_PACKET_SIZE_MAX));
ASSERT_NOT_NULL(packet);
dns_packet_set_flags(packet, /* dnssec_checking_disabled= */ false, /* truncated= */ false);
ASSERT_EQ(DNS_PACKET_QR(packet), 0);
ASSERT_EQ(DNS_PACKET_OPCODE(packet), 0);
ASSERT_EQ(DNS_PACKET_AA(packet), 0);
ASSERT_EQ(DNS_PACKET_TC(packet), 0);
ASSERT_EQ(DNS_PACKET_RD(packet), 1);
ASSERT_EQ(DNS_PACKET_RA(packet), 0);
ASSERT_EQ(DNS_PACKET_AD(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 0);
ASSERT_EQ(DNS_PACKET_RCODE(packet), 0);
}
TEST(packet_set_flags_dns_checking_disabled) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new(&packet, DNS_PROTOCOL_DNS, 0, DNS_PACKET_SIZE_MAX));
ASSERT_NOT_NULL(packet);
dns_packet_set_flags(packet, /* dnssec_checking_disabled= */ true, /* truncated= */ false);
ASSERT_EQ(DNS_PACKET_QR(packet), 0);
ASSERT_EQ(DNS_PACKET_OPCODE(packet), 0);
ASSERT_EQ(DNS_PACKET_AA(packet), 0);
ASSERT_EQ(DNS_PACKET_TC(packet), 0);
ASSERT_EQ(DNS_PACKET_RD(packet), 1);
ASSERT_EQ(DNS_PACKET_RA(packet), 0);
ASSERT_EQ(DNS_PACKET_AD(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 1);
ASSERT_EQ(DNS_PACKET_RCODE(packet), 0);
}
TEST(packet_set_flags_llmnr) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new(&packet, DNS_PROTOCOL_LLMNR, 0, DNS_PACKET_SIZE_MAX));
ASSERT_NOT_NULL(packet);
dns_packet_set_flags(packet, /* dnssec_checking_disabled= */ true, /* truncated= */ false);
ASSERT_EQ(DNS_PACKET_QR(packet), 0);
ASSERT_EQ(DNS_PACKET_OPCODE(packet), 0);
ASSERT_EQ(DNS_PACKET_AA(packet), 0);
ASSERT_EQ(DNS_PACKET_TC(packet), 0);
ASSERT_EQ(DNS_PACKET_RD(packet), 0);
ASSERT_EQ(DNS_PACKET_RA(packet), 0);
ASSERT_EQ(DNS_PACKET_AD(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 0);
ASSERT_EQ(DNS_PACKET_RCODE(packet), 0);
}
TEST(packet_set_flags_mdns_not_truncated) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new(&packet, DNS_PROTOCOL_MDNS, 0, DNS_PACKET_SIZE_MAX));
ASSERT_NOT_NULL(packet);
dns_packet_set_flags(packet, /* dnssec_checking_disabled= */ true, /* truncated= */ false);
ASSERT_EQ(DNS_PACKET_QR(packet), 0);
ASSERT_EQ(DNS_PACKET_OPCODE(packet), 0);
ASSERT_EQ(DNS_PACKET_AA(packet), 0);
ASSERT_EQ(DNS_PACKET_TC(packet), 0);
ASSERT_EQ(DNS_PACKET_RD(packet), 0);
ASSERT_EQ(DNS_PACKET_RA(packet), 0);
ASSERT_EQ(DNS_PACKET_AD(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 0);
ASSERT_EQ(DNS_PACKET_RCODE(packet), 0);
}
TEST(packet_set_flags_mdns_truncated) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new(&packet, DNS_PROTOCOL_MDNS, 0, DNS_PACKET_SIZE_MAX));
ASSERT_NOT_NULL(packet);
dns_packet_set_flags(packet, /* dnssec_checking_disabled= */ true, /* truncated= */ true);
ASSERT_EQ(DNS_PACKET_QR(packet), 0);
ASSERT_EQ(DNS_PACKET_OPCODE(packet), 0);
ASSERT_EQ(DNS_PACKET_AA(packet), 0);
ASSERT_EQ(DNS_PACKET_TC(packet), 1);
ASSERT_EQ(DNS_PACKET_RD(packet), 0);
ASSERT_EQ(DNS_PACKET_RA(packet), 0);
ASSERT_EQ(DNS_PACKET_AD(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 0);
ASSERT_EQ(DNS_PACKET_RCODE(packet), 0);
}
/* ================================================================
* dns_packet_new_query()
* ================================================================ */
TEST(packet_new_query_checking_enabled) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new_query(&packet, DNS_PROTOCOL_DNS, 0, false));
ASSERT_NOT_NULL(packet);
ASSERT_EQ(DNS_PACKET_TC(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 0);
}
TEST(packet_new_query_checking_disabled) {
_cleanup_(dns_packet_unrefp) DnsPacket *packet = NULL;
ASSERT_OK(dns_packet_new_query(&packet, DNS_PROTOCOL_DNS, 0, true));
ASSERT_NOT_NULL(packet);
ASSERT_EQ(DNS_PACKET_TC(packet), 0);
ASSERT_EQ(DNS_PACKET_CD(packet), 1);
}
DEFINE_TEST_MAIN(LOG_DEBUG)