PDF: Fix progress bar for files of unknown size.

BUG=666663 Review-Url: https://codereview.chromium.org/2516453003 Cr-Commit-Position: refs/heads/master@{#433330}
feiyue1206 · Nov 18, 2016 · 2c81990 · 2c81990
1 parent 7cd55da
commit 2c81990
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/pdf/pdfium/pdfium_engine.cc b/pdf/pdfium/pdfium_engine.cc
@@ -1214,12 +1214,15 @@ void PDFiumEngine::OnPendingRequestComplete() {
 }
 
 void PDFiumEngine::OnNewDataAvailable() {
-  const float progress = doc_loader_->GetProgress();
-  if (progress < 0.001) {
-    client_->DocumentLoadProgress(0, 0);
-  } else {
-    client_->DocumentLoadProgress(progress * 10000, 10000);
+  if (!doc_loader_->GetDocumentSize()) {
+    client_->DocumentLoadProgress(doc_loader_->count_of_bytes_received(), 0);
+    return;
   }
+
+  const float progress = doc_loader_->GetProgress();
+  DCHECK_GE(progress, 0.0);
+  DCHECK_LE(progress, 1.0);
+  client_->DocumentLoadProgress(progress * 10000, 10000);
 }
 
 void PDFiumEngine::OnDocumentComplete() {