fix: clamp numMasksInChunk to prevent heap-buffer-overflow in EmbeddingExtractor#398
Merged
Alex-Wengg merged 1 commit intoFluidInference:mainfrom Mar 19, 2026
Conversation
…ngExtractor When audio.count > 160,000 samples (>10s at 16kHz), the numMasksInChunk formula `(firstMask.count * audio.count + 80_000) / 160_000` produces a value larger than firstMask.count. This causes vDSP_mmov in fillMaskBufferOptimized() to read past the mask buffer allocation. For example, with maskCount=100 and 20s audio (320k samples): buggy: (100 * 320000 + 80000) / 160000 = 200 — 2x overread fixed: min(200, 100) = 100 The fix clamps numMasksInChunk to firstMask.count with min(). Bug introduced in v0.8.0 (PR FluidInference#191, 2025-11-26). Affects v0.8.0–v0.12.4. Detected via AddressSanitizer: READ of size 3456 from 2388-byte buffer. Includes regression tests validating the formula and vDSP_mmov bounds. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
When audio.count > 160,000 samples (>10s at 16kHz), the numMasksInChunk formula
(firstMask.count * audio.count + 80_000) / 160_000produces a value larger than firstMask.count. This causes vDSP_mmov in fillMaskBufferOptimized() to read past the mask buffer allocation.For example, with maskCount=100 and 20s audio (320k samples):
buggy: (100 * 320000 + 80000) / 160000 = 200 — 2x overread
fixed: min(200, 100) = 100
The fix clamps numMasksInChunk to firstMask.count with min().
Bug introduced in v0.8.0 (PR #191, 2025-11-26). Affects v0.8.0–v0.12.4. Detected via AddressSanitizer: READ of size 3456 from 2388-byte buffer.
Includes regression tests validating the formula and vDSP_mmov bounds.