core: Respect the probe search limit when probing (#233)

pdeljanov · Jan 15, 2024 · 3d206b2 · 3d206b2
1 parent 850b227
commit 3d206b2
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/symphonia-core/src/probe.rs b/symphonia-core/src/probe.rs
@@ -237,6 +237,10 @@ impl Probe {
 
             count += 1;
 
+            if count > Probe::PROBE_SEARCH_LIMIT {
+                break;
+            }
+
             if count % 4096 == 0 {
                 debug!(
                     "searching for format marker... {}+{} / {} bytes.",
@@ -291,8 +295,13 @@ impl Probe {
             }
         }
 
-        // Could not find any marker within the probe limit.
-        error!("reached probe limit of {} bytes.", Probe::PROBE_SEARCH_LIMIT);
+        if count < Probe::PROBE_SEARCH_LIMIT {
+            error!("probe reach EOF at {} bytes.", count);
+        }
+        else {
+            // Could not find any marker within the probe limit.
+            error!("reached probe limit of {} bytes.", Probe::PROBE_SEARCH_LIMIT);
+        }
 
         unsupported_error("core (probe): no suitable format reader found")
     }