* cmdline/apt-mark:

[apt.git] / apt-pkg / packagemanager.h

diff --git a/apt-pkg/packagemanager.h b/apt-pkg/packagemanager.h
index da1f21e6ce90d6db3bb86b1a7d5fbd7b710c1db4..48f53576cf981517abd93f20be92a01e9ad686cf 100644 (file)
--- a/apt-pkg/packagemanager.h
+++ b/apt-pkg/packagemanager.h
@@ -28,6 +28,7 @@
 #endif
 
 #include <string>
+#include <iostream>
 #include <apt-pkg/pkgcache.h>
 #include <apt-pkg/depcache.h>
 
@@ -82,12 +83,7 @@ class pkgPackageManager : protected pkgCache::Namespace
                    pkgRecords *Recs);
 
    // Do the installation 
-   OrderResult DoInstall() {
-      if(DoInstallPreFork() == Failed)
-        return Failed;
-
-      return DoInstallPostFork();
-   }
+   OrderResult DoInstall(int statusFd=-1);
 
    // stuff that needs to be done before the fork() of a library that
    // uses apt
@@ -101,11 +97,11 @@ class pkgPackageManager : protected pkgCache::Namespace
       bool goResult = Go(statusFd);
       if(goResult == false) 
         return Failed;
-      
+
       // if all was fine update the state file
-      if(Res == Completed)
+      if(Res == Completed) {
         Cache.writeStateFile(NULL);
-
+      }
       return Res;
    };