From d2c54b1e74ea52013d7e9fe87a568492b4708e21 Mon Sep 17 00:00:00 2001
From: TommyLike <tommylikehu@gmail.com>
Date: Thu, 13 Jul 2023 22:44:43 +0800
Subject: [PATCH] Check file content is empty before sending request to remote
 server

---
 src/client/worker/signer.rs | 4 ++++
 src/util/error.rs           | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/src/client/worker/signer.rs b/src/client/worker/signer.rs
index 815e8fe..d00842c 100644
--- a/src/client/worker/signer.rs
+++ b/src/client/worker/signer.rs
@@ -68,6 +68,10 @@ impl SignHandler for RemoteSigner {
                     key_id: item.key_id.clone(),
                 });
             }
+            if sign_segments.is_empty() {
+                *item.error.borrow_mut() = Err(Error::FileContentEmpty);
+                return item
+            }
             let result = self.client.sign_stream(
                 tokio_stream::iter(sign_segments)).await;
             match result {
diff --git a/src/util/error.rs b/src/util/error.rs
index 270b76c..c05b35e 100644
--- a/src/util/error.rs
+++ b/src/util/error.rs
@@ -125,6 +125,8 @@ pub enum Error {
     PartialSuccessError,
     #[error("Error in sign or parse EFI image")]
     EFIError(String),
+    #[error("file content is empty")]
+    FileContentEmpty,
 }
 
 #[derive(Deserialize, Serialize, ToSchema)]
-- 
Gitee