harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix UB in WhisperVadSegments

Browse source
This commit is contained in:
Niko 2025-09-10 15:28:40 -07:00
parent 5f054c94b7
commit 7e116dfb29
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/whisper_vad.rs
View file

@ -251,24 +251,24 @@ impl WhisperVadSegments {
} }
pub fn index_in_bounds(&self, idx: c_int) -> bool { pub fn index_in_bounds(&self, idx: c_int) -> bool {
idx < 0 || idx > self.segment_count idx >= 0 && idx < self.segment_count
} }
/// Return the start timestamp of this segment in centiseconds (10s of milliseconds). /// Return the start timestamp of this segment in centiseconds (10s of milliseconds).
pub fn get_segment_start_timestamp(&self, idx: c_int) -> Option<f32> { pub fn get_segment_start_timestamp(&self, idx: c_int) -> Option<f32> {
if self.index_in_bounds(idx) { if self.index_in_bounds(idx) {
None
} else {
Some(unsafe { whisper_vad_segments_get_segment_t0(self.ptr, idx) }) Some(unsafe { whisper_vad_segments_get_segment_t0(self.ptr, idx) })
} else {
None
} }
} }
/// Return the end timestamp of this segment in centiseconds (10s of milliseconds). /// Return the end timestamp of this segment in centiseconds (10s of milliseconds).
pub fn get_segment_end_timestamp(&self, idx: c_int) -> Option<f32> { pub fn get_segment_end_timestamp(&self, idx: c_int) -> Option<f32> {
if self.index_in_bounds(idx) { if self.index_in_bounds(idx) {
None
} else {
Some(unsafe { whisper_vad_segments_get_segment_t1(self.ptr, idx) }) Some(unsafe { whisper_vad_segments_get_segment_t1(self.ptr, idx) })
} else {
None
} }
} }