10
0
Fork 0
mirror of https://github.com/ZeusWPI/ZNS.git synced 2024-10-30 13:34:26 +01:00

Fix: NXDOMAIN only when domain does not exist

This commit is contained in:
Xander Bil 2024-10-29 22:30:05 +01:00
parent 9c3d152618
commit a4bf483a82
No known key found for this signature in database
GPG key ID: EC9706B54A278598

View file

@ -35,20 +35,31 @@ impl ResponseHandler for QueryHandler {
match answers { match answers {
Ok(mut rrs) => { Ok(mut rrs) => {
if rrs.is_empty() { if rrs.is_empty() {
let domain_records = get_from_database(
&question.qname,
None,
question.qclass.clone(),
connection,
)?;
if domain_records.is_empty() && !question.qname.is_empty() {
rrs.extend(try_wildcard(question, connection)?); rrs.extend(try_wildcard(question, connection)?);
if rrs.is_empty() { }
if question.qtype == Type::Type(RRType::SOA)
if rrs.is_empty()
&& question.qtype == Type::Type(RRType::SOA)
&& Config::get().default_soa && Config::get().default_soa
{ {
rrs.extend([get_soa(&question.qname)?]) rrs.extend([get_soa(&question.qname)?])
} else { }
if rrs.is_empty() && domain_records.is_empty() {
return Err(ZNSError::NXDomain { return Err(ZNSError::NXDomain {
domain: question.qname.to_string(), domain: question.qname.to_string(),
qtype: question.qtype.clone(), qtype: question.qtype.clone(),
}); });
} }
} }
}
response.header.ancount += rrs.len() as u16; response.header.ancount += rrs.len() as u16;
response.answer.extend(rrs) response.answer.extend(rrs)
} }
@ -65,11 +76,6 @@ impl ResponseHandler for QueryHandler {
} }
fn try_wildcard(question: &Question, connection: &mut PgConnection) -> Result<Vec<RR>, ZNSError> { fn try_wildcard(question: &Question, connection: &mut PgConnection) -> Result<Vec<RR>, ZNSError> {
let records = get_from_database(&question.qname, None, question.qclass.clone(), connection)?;
if !records.is_empty() || question.qname.as_slice().is_empty() {
Ok(vec![])
} else {
let qname = question.qname.clone().to_vec(); let qname = question.qname.clone().to_vec();
qname.to_vec()[0] = String::from("*"); qname.to_vec()[0] = String::from("*");
Ok(get_from_database( Ok(get_from_database(
@ -84,7 +90,6 @@ fn try_wildcard(question: &Question, connection: &mut PgConnection) -> Result<Ve
rr rr
}) })
.collect()) .collect())
}
} }
fn get_soa(name: &LabelString) -> Result<RR, ZNSError> { fn get_soa(name: &LabelString) -> Result<RR, ZNSError> {