diff --git a/crates/proto/src/op/message.rs b/crates/proto/src/op/message.rs index b45d65f2ec..2d8f625a67 100644 --- a/crates/proto/src/op/message.rs +++ b/crates/proto/src/op/message.rs @@ -560,7 +560,7 @@ impl Message { pub fn update_counts(&mut self) -> &mut Self { self.header = update_header_counts( &self.header, - false, + self.truncated(), HeaderCounts { query_count: self.queries.len(), answer_count: self.answers.len(), @@ -797,7 +797,8 @@ where nameserver_count: nameserver_count.0, additional_count: additional_count.0, }; - let was_truncated = answer_count.1 || nameserver_count.1 || additional_count.1; + let was_truncated = + header.truncated() || answer_count.1 || nameserver_count.1 || additional_count.1; place.replace(encoder, update_header_counts(header, was_truncated, counts))?; Ok(())