make https honor ExpectedSize as well

1 files changed, 6 insertions, 0 deletions

diff --git a/methods/https.cc b/methods/https.cc
index e0348ab58..cacd8a6bc 100644
--- a/methods/https.cc
+++ b/methods/https.cc
@@ -81,6 +81,12 @@ HttpsMethod::write_data(void *buffer, size_t size, size_t nmemb, void *userp)
    if(me->File->Write(buffer, size*nmemb) != true)
       return false;
 
+   me->TotalWritten += size*nmemb;
+   if(me->TotalWritten > me->Queue->ExpectedSize)
+      return _error->Error("Writing more data than expected (%llu > %llu)",
+                           me->TotalWritten, me->Queue->ExpectedSize);
+
+
    return size*nmemb;
 }