From 52307352229bb3db9b02a535b0543cad5dbdc8fa Mon Sep 17 00:00:00 2001
From: Niko <zero@imaskeleton.me>
Date: Wed, 3 Jan 2024 20:26:08 -0700
Subject: [PATCH 1/2] Fix possible segmentation fault

---
 src/error.rs         | 2 ++
 src/whisper_state.rs | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/src/error.rs b/src/error.rs
index 5807a90..8e3d3b0 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -39,6 +39,8 @@ pub enum WhisperError {
     InvalidText,
     /// Creating a state pointer failed. Check stderr for more information.
     FailedToCreateState,
+    /// No samples were provided.
+    NoSamples,
 }
 
 impl From<Utf8Error> for WhisperError {
diff --git a/src/whisper_state.rs b/src/whisper_state.rs
index c49e2b6..f336638 100644
--- a/src/whisper_state.rs
+++ b/src/whisper_state.rs
@@ -327,6 +327,11 @@ impl<'a> WhisperState<'a> {
     /// # C++ equivalent
     /// `int whisper_full(struct whisper_context * ctx, struct whisper_full_params params, const float * samples, int n_samples)`
     pub fn full(&mut self, params: FullParams, data: &[f32]) -> Result<c_int, WhisperError> {
+        if data.is_empty() {
+            // can randomly trigger segmentation faults if we don't check this
+            return Err(WhisperError::NoSamples);
+        }
+
         let ret = unsafe {
             whisper_rs_sys::whisper_full_with_state(
                 self.ctx,

From 937bc78f450a7d35613bf3e8ba51e129f28ea5c2 Mon Sep 17 00:00:00 2001
From: Niko <zero@imaskeleton.me>
Date: Wed, 3 Jan 2024 20:31:08 -0700
Subject: [PATCH 2/2] Fix clippy lints

---
 src/error.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/error.rs b/src/error.rs
index 8e3d3b0..51ab0bc 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -108,6 +108,7 @@ impl std::fmt::Display for WhisperError {
                 "Generic whisper error. Varies depending on the function. Error code: {}",
                 c_int
             ),
+            NoSamples => write!(f, "Input sample buffer was empty."),
         }
     }
 }