handle non-utf8 from libfuzzer stderr (#379)

microsoft · Dec 10, 2020 · 7f5673e · 7f5673e
1 parent 56090cb
commit 7f5673e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/agent/onefuzz-agent/src/tasks/fuzz/libfuzzer_fuzz.rs b/src/agent/onefuzz-agent/src/tasks/fuzz/libfuzzer_fuzz.rs
@@ -143,11 +143,16 @@ impl LibFuzzerFuzzTask {
             .stderr
             .as_mut()
             .ok_or_else(|| format_err!("stderr not captured"))?;
-        let stderr = BufReader::new(stderr);
+        let mut stderr = BufReader::new(stderr);
 
         let mut libfuzzer_output = Vec::new();
-        let mut lines = stderr.lines();
-        while let Some(line) = lines.next_line().await? {
+        loop {
+            let mut buf = vec![];
+            let bytes_read = stderr.read_until(b'\n', &mut buf).await?;
+            if bytes_read == 0 && buf.is_empty() {
+                break;
+            }
+            let line = String::from_utf8_lossy(&buf).to_string();
             if let Some(stats_sender) = stats_sender {
                 if let Err(err) = try_report_iter_update(stats_sender, worker_id, run_id, &line) {
                     error!("could not parse fuzzing interation update: {}", err);