X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=src%2Fquery.rs;h=8c478e629274639388773cd5dc05a404672f2fa9;hb=cff93cd2340cb4e14c3f7e220d7c7bea0dd05e79;hp=37fbaefcc9d969ca50cc2b9b75b9dd8a575c453a;hpb=b2791c191a941208eff825a44440fbf605dcfb60;p=dnssec-prover

diff --git a/src/query.rs b/src/query.rs
index 37fbaef..8c478e6 100644
--- a/src/query.rs
+++ b/src/query.rs
@@ -34,7 +34,8 @@ pub struct QueryBuf {
 	len: u16,
 }
 impl QueryBuf {
-	fn new_zeroed(len: u16) -> Self {
+	/// Generates a new buffer of the given length, consisting of all zeros.
+	pub fn new_zeroed(len: u16) -> Self {
 		let heap_buf = if len > STACK_BUF_LIMIT { vec![0; len as usize] } else { Vec::new() };
 		Self {
 			buf: [0; STACK_BUF_LIMIT as usize],
@@ -42,7 +43,11 @@ impl QueryBuf {
 			len
 		}
 	}
-	pub(crate) fn extend_from_slice(&mut self, sl: &[u8]) {
+	/// Extends the size of this buffer by appending the given slice.
+	///
+	/// If the total length of this buffer exceeds [`u16::MAX`] after appending, the buffer's state
+	/// is undefined, however pushing data beyond [`u16::MAX`] will not panic.
+	pub fn extend_from_slice(&mut self, sl: &[u8]) {
 		let new_len = self.len.saturating_add(sl.len() as u16);
 		let was_heap = self.len > STACK_BUF_LIMIT;
 		let is_heap = new_len > STACK_BUF_LIMIT;
@@ -59,6 +64,14 @@ impl QueryBuf {
 		target.copy_from_slice(sl);
 		self.len = new_len;
 	}
+	/// Converts this query into its bytes on the heap
+	pub fn into_vec(self) -> Vec<u8> {
+		if self.len > STACK_BUF_LIMIT {
+			self.heap_buf
+		} else {
+			self.buf[..self.len as usize].to_vec()
+		}
+	}
 }
 impl ops::Deref for QueryBuf {
 	type Target = [u8];