make http size check work

diff --cc apt-pkg/acquire-method.cc
Simple merge

diff --cc apt-pkg/acquire-method.h
Simple merge

diff --cc methods/http.cc
index 916fa46,f2a4a4d..b076e59
--- 1/methods/http.cc
--- 2/methods/http.cc
+++ b/methods/http.cc
@@@ -653,10 -651,6 +655,12 @@@ bool HttpServerState::Go(bool ToFile, F
         return _error->Errno("write",_("Error writing to output file"));
     }
  
-    if (ExpectedSize > 0 && In.TotalWriten > ExpectedSize)
++   if (ExpectedSize > 0 && File && File->Tell() > ExpectedSize)
++   {
 +      return _error->Error("Writing more data than expected (%llu > %llu)",
-                            In.TotalWriten, ExpectedSize);
++                           File->Tell(), ExpectedSize);
++   }
 +
     // Handle commands from APT
     if (FD_ISSET(STDIN_FILENO,&rfds))
     {

diff --cc methods/http.h
Simple merge

diff --cc methods/https.cc
Simple merge

diff --cc methods/https.h
index 2335559,45d1f7f..0387cb9
--- 1/methods/https.h
--- 2/methods/https.h
+++ b/methods/https.h
@@@ -70,10 -69,9 +70,9 @@@ class HttpsMethod : public pkgAcqMetho
  
     public:
     FileFd *File;
 -
 -   HttpsMethod() : pkgAcqMethod("1.2",Pipeline | SendConfig), Server(NULL), File(NULL)
 +      
-    HttpsMethod() : pkgAcqMethod("1.2",Pipeline | SendConfig), TotalWritten(0), File(NULL)
++   HttpsMethod() : pkgAcqMethod("1.2",Pipeline | SendConfig), Server(NULL), TotalWritten(0), File(NULL)
     {
-       File = 0;
        curl = curl_easy_init();
     };
  

diff --cc methods/server.cc
index 1b6511c,4a961f4..2237379
--- 1/methods/server.cc
--- 2/methods/server.cc
+++ b/methods/server.cc
@@@ -609,7 -605,7 +610,10 @@@ int ServerMethod::Loop(
                  QueueBack = Queue;
               }
               else
++               {
++                  Server->Close();
                  Fail(true);
++               }
            }
            break;
         }

diff --cc methods/server.h
Simple merge

diff --cc test/integration/test-apt-update-expected-size
index 0000000,0000000..7281233
new file mode 100755 (executable)
--- /dev/null
--- /dev/null
+++ b/test/integration/test-apt-update-expected-size
@@@ -1,0 -1,0 +1,27 @@@
++#!/bin/sh
++set -e
++
++TESTDIR=$(readlink -f $(dirname $0))
++. $TESTDIR/framework
++
++setupenvironment
++configarchitecture "i386"
++
++insertpackage 'unstable' 'apt' 'all' '1.0'
++
++setupaptarchive --no-update
++changetowebserver
++
++# normal update works fine
++testsuccess aptget update
++
++# append junk at the end of the Packages.gz/Packages
++SIZE="$(stat --printf=%s aptarchive/dists/unstable/main/binary-i386/Packages)"
++echo "1234567890" >> aptarchive/dists/unstable/main/binary-i386/Packages.gz
++echo "1234567890" >> aptarchive/dists/unstable/main/binary-i386/Packages
++NEW_SIZE="$(stat --printf=%s aptarchive/dists/unstable/main/binary-i386/Packages)"
++rm -f rootdir/var/lib/apt/lists/localhost*
++testequal "W: Failed to fetch http://localhost:8080/dists/unstable/main/binary-i386/Packages  Writing more data than expected ($NEW_SIZE > $SIZE) [IP: ::1 8080]
++
++E: Some index files failed to download. They have been ignored, or old ones used instead." aptget update -qq
++