fix: prevent panic if query has no question

This commit is contained in:
Severin Buhler
2024-02-01 18:02:46 +01:00
parent dfcec7535f
commit 1e9e987c88
3 changed files with 7 additions and 3 deletions

2
Cargo.lock generated
View File

@@ -1008,7 +1008,7 @@ dependencies = [
[[package]] [[package]]
name = "pkdns" name = "pkdns"
version = "0.1.1" version = "0.1.2"
dependencies = [ dependencies = [
"any-dns", "any-dns",
"clap", "clap",

View File

@@ -1,6 +1,6 @@
[package] [package]
name = "pkdns" name = "pkdns"
version = "0.1.1" version = "0.1.2"
edition = "2021" edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

View File

@@ -46,7 +46,11 @@ impl PknamesResolver {
return pkarr_result; // It was a pkarr hostname return pkarr_result; // It was a pkarr hostname
} }
let question = original_query.questions.first().unwrap(); let question = original_query.questions.first();
if let None = question {
}
let question = original_query.questions.first().ok_or("Query does not include a question.")?;
let domain = question.qname.to_string(); let domain = question.qname.to_string();
let pkarr_domain = self.predict_pknames_domain(&domain)?; let pkarr_domain = self.predict_pknames_domain(&domain)?;