[downloader] Allow underlying downloads to provide detailed job progress

Signed-off-by: Michael Brown <mcb30@ipxe.org>
ipxe · Sep 5, 2017 · a258b08 · a258b08
1 parent e30cc5e
commit a258b08
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/core/downloader.c b/src/core/downloader.c
@@ -111,13 +111,20 @@ static void downloader_finished ( struct downloader *downloader, int rc ) {
  */
 static int downloader_progress ( struct downloader *downloader,
 				 struct job_progress *progress ) {
+	int rc;
+
+	/* Allow data transfer to provide an accurate description */
+	if ( ( rc = job_progress ( &downloader->xfer, progress ) ) != 0 )
+		return rc;
 
 	/* This is not entirely accurate, since downloaded data may
 	 * arrive out of order (e.g. with multicast protocols), but
 	 * it's a reasonable first approximation.
 	 */
-	progress->completed = downloader->buffer.pos;
-	progress->total = downloader->buffer.len;
+	if ( ! progress->total ) {
+		progress->completed = downloader->buffer.pos;
+		progress->total = downloader->buffer.len;
+	}
 
 	return 0;
 }