Fixed infinite loop inside verity write and --disable_fec_computation am: 701cb6f654 am: 0583240c61

Original change: https://android-review.googlesource.com/c/platform/system/update_engine/+/2467467

Change-Id: I1c39e6c4b05a74395bddbc8ca8a475859a125b39
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
diff --git a/payload_consumer/verity_writer_android.cc b/payload_consumer/verity_writer_android.cc
index 31d2875..4a476d2 100644
--- a/payload_consumer/verity_writer_android.cc
+++ b/payload_consumer/verity_writer_android.cc
@@ -151,6 +151,9 @@
 }
 
 double IncrementalEncodeFEC::ReportProgress() const {
+  if (num_rounds_ == 0) {
+    return 1.0;
+  }
   return static_cast<double>(current_round_) / num_rounds_;
 }
 
@@ -311,7 +314,8 @@
   return true;
 }
 bool VerityWriterAndroid::FECFinished() const {
-  if (encodeFEC_.Finished()) {
+  if ((encodeFEC_.Finished() || partition_->fec_size == 0) &&
+      hash_tree_written_) {
     return true;
   }
   return false;