From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from firstgate.proxmox.com (firstgate.proxmox.com [212.224.123.68]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by lists.proxmox.com (Postfix) with ESMTPS id C024C98973 for ; Wed, 15 Nov 2023 16:48:58 +0100 (CET) Received: from firstgate.proxmox.com (localhost [127.0.0.1]) by firstgate.proxmox.com (Proxmox) with ESMTP id A220894F3 for ; Wed, 15 Nov 2023 16:48:28 +0100 (CET) Received: from proxmox-new.maurer-it.com (proxmox-new.maurer-it.com [94.136.29.106]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by firstgate.proxmox.com (Proxmox) with ESMTPS for ; Wed, 15 Nov 2023 16:48:28 +0100 (CET) Received: from proxmox-new.maurer-it.com (localhost.localdomain [127.0.0.1]) by proxmox-new.maurer-it.com (Proxmox) with ESMTP id E4413432B1 for ; Wed, 15 Nov 2023 16:48:27 +0100 (CET) From: Christian Ebner To: pbs-devel@lists.proxmox.com Date: Wed, 15 Nov 2023 16:47:48 +0100 Message-Id: <20231115154813.281564-4-c.ebner@proxmox.com> X-Mailer: git-send-email 2.39.2 In-Reply-To: <20231115154813.281564-1-c.ebner@proxmox.com> References: <20231115154813.281564-1-c.ebner@proxmox.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-SPAM-LEVEL: Spam detection results: 0 AWL 0.060 Adjusted score from AWL reputation of From: address BAYES_00 -1.9 Bayes spam probability is 0 to 1% DMARC_MISSING 0.1 Missing DMARC policy KAM_DMARC_STATUS 0.01 Test Rule for DKIM or SPF Failure with Strict Alignment SPF_HELO_NONE 0.001 SPF: HELO does not publish an SPF Record SPF_PASS -0.001 SPF: sender matches SPF record T_SCC_BODY_TEXT_LINE -0.01 - Subject: [pbs-devel] [PATCH v5 pxar 3/28] fix #3174: encoder: calc filename + metadata byte size X-BeenThere: pbs-devel@lists.proxmox.com X-Mailman-Version: 2.1.29 Precedence: list List-Id: Proxmox Backup Server development discussion List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-List-Received-Date: Wed, 15 Nov 2023 15:48:58 -0000 Introduce SeqSink and impl SeqWrite in order to create an encoder implementation which instead of writing data to a stream, consumes the encoded stream and returns the consumed bytes for that stream. Based on this, implement a helper function `byte_len` which returns the byte size of the filename entry and metadata entry as encoded by the archive. Signed-off-by: Christian Ebner --- Changes since v4: - Refactor and rename `bytes_len` to `encoded_size` Changes since v3: - no changes Changes since v2: - no changes Changes since v1: - Instead of calculating the metadata size based on the known encoding sizes, implement an Encoder instance which counts the encoded bytes. src/encoder/mod.rs | 35 +++++++++++++++++++++++++++++++++++ src/encoder/sync.rs | 5 +++++ 2 files changed, 40 insertions(+) diff --git a/src/encoder/mod.rs b/src/encoder/mod.rs index 0d342ec..860c21f 100644 --- a/src/encoder/mod.rs +++ b/src/encoder/mod.rs @@ -85,6 +85,24 @@ where } } +#[derive(Default)] +/// Sink to consume sequential byte stream +pub struct SeqSink; + +impl SeqWrite for SeqSink { + fn poll_seq_write( + self: Pin<&mut Self>, + _cx: &mut Context, + buf: &[u8], + ) -> Poll> { + Poll::Ready(Ok(buf.len())) + } + + fn poll_flush(self: Pin<&mut Self>, _cx: &mut Context) -> Poll> { + Poll::Ready(Ok(())) + } +} + /// awaitable verison of `poll_seq_write`. async fn seq_write( output: &mut T, @@ -183,6 +201,23 @@ where seq_write_pxar_entry(output, htype, buf, position).await } +/// Calculate the encoded byte len of filename and metadata struct +pub async fn encoded_size(filename: &std::ffi::CStr, metadata: &Metadata) -> io::Result { + let mut this = EncoderImpl { + output: EncoderOutput::Owned(SeqSink::default()), + state: EncoderState::default(), + parent: None, + finished: false, + file_copy_buffer: Arc::new(Mutex::new(unsafe { + crate::util::vec_new_uninitialized(1024 * 1024) + })), + }; + + this.start_file_do(Some(metadata), filename.to_bytes()) + .await?; + Ok(this.position()) +} + /// Error conditions caused by wrong usage of this crate. #[derive(Clone, Copy, Debug, Eq, PartialEq)] pub enum EncodeError { diff --git a/src/encoder/sync.rs b/src/encoder/sync.rs index 1ec91b8..93c3b2c 100644 --- a/src/encoder/sync.rs +++ b/src/encoder/sync.rs @@ -228,3 +228,8 @@ impl SeqWrite for StandardWriter { Poll::Ready(self.pin_to_inner().and_then(|inner| inner.flush())) } } + +/// Calculate the encoded byte len of filename and metadata struct +pub fn encoded_size(filename: &std::ffi::CStr, metadata: &Metadata) -> io::Result { + poll_result_once(crate::encoder::encoded_size(filename, metadata)) +} -- 2.39.2