Checked the size of partial files #33705

[apt.git] / cmdline / apt-get.cc

diff --git a/cmdline/apt-get.cc b/cmdline/apt-get.cc
index da364e03c238f6ed673e8c2212b4df67ce216d72..73397e8aeab1446f688df3fd0a72cbbaaf4f1254 100644 (file)
--- a/cmdline/apt-get.cc
+++ b/cmdline/apt-get.cc
@@ -1,6 +1,6 @@
 // -*- mode: cpp; mode: fold -*-
 // Description                                                         /*{{{*/
-// $Id: apt-get.cc,v 1.43 1999/02/21 08:38:53 jgg Exp $
+// $Id: apt-get.cc,v 1.47 1999/03/27 03:02:39 jgg Exp $
 /* ######################################################################
    
    apt-get - Cover for dpkg
@@ -79,6 +79,18 @@ bool YnPrompt()
    return true;
 }
                                                                        /*}}}*/
+// AnalPrompt - Annoying Yes No Prompt.                                        /*{{{*/
+// ---------------------------------------------------------------------
+/* Returns true on a Yes.*/
+bool AnalPrompt(const char *Text)
+{
+   char Buf[1024];
+   cin.getline(Buf,sizeof(Buf));
+   if (strcmp(Buf,Text) == 0)
+      return true;
+   return false;
+}
+                                                                       /*}}}*/
 // ShowList - Show a list                                              /*{{{*/
 // ---------------------------------------------------------------------
 /* This prints out a string of space seperated words with a title and 
@@ -466,9 +478,10 @@ bool CacheFile::Open(bool AllowBroken)
 // ---------------------------------------------------------------------
 /* This displays the informative messages describing what is going to 
    happen and then calls the download routines */
-bool InstallPackages(CacheFile &Cache,bool ShwKept,bool Ask = true)
+bool InstallPackages(CacheFile &Cache,bool ShwKept,bool Ask = true,bool Saftey = true)
 {
    bool Fail = false;
+   bool Essential = false;
    
    // Show all the various warning indicators
    ShowDel(c1out,Cache);
@@ -478,7 +491,8 @@ bool InstallPackages(CacheFile &Cache,bool ShwKept,bool Ask = true)
    Fail |= !ShowHold(c1out,Cache);
    if (_config->FindB("APT::Get::Show-Upgraded",false) == true)
       ShowUpgraded(c1out,Cache);
-   Fail |= !ShowEssential(c1out,Cache);
+   Essential = !ShowEssential(c1out,Cache);
+   Fail |= Essential;
    Stats(c1out,Cache);
    
    // Sanity check
@@ -529,6 +543,7 @@ bool InstallPackages(CacheFile &Cache,bool ShwKept,bool Ask = true)
 
    // Display statistics
    unsigned long FetchBytes = Fetcher.FetchNeeded();
+   unsigned long FetchPBytes = Fetcher.PartialPresent();
    unsigned long DebBytes = Fetcher.TotalNeeded();
    if (DebBytes != Cache->DebSize())
    {
@@ -542,7 +557,7 @@ bool InstallPackages(CacheFile &Cache,bool ShwKept,bool Ask = true)
    if (statfs(OutputDir.c_str(),&Buf) != 0)
       return _error->Errno("statfs","Couldn't determine free space in %s",
                           OutputDir.c_str());
-   if (unsigned(Buf.f_bfree) < FetchBytes/Buf.f_bsize)
+   if (unsigned(Buf.f_bfree) < (FetchBytes - FetchPBytes)/Buf.f_bsize)
       return _error->Error("Sorry, you don't have enough free space in %s",
                           OutputDir.c_str());
    
@@ -565,23 +580,43 @@ bool InstallPackages(CacheFile &Cache,bool ShwKept,bool Ask = true)
       return false;
 
    // Fail safe check
-   if (_config->FindB("APT::Get::Assume-Yes",false) == true)
+   if (_config->FindI("quiet",0) >= 2 ||
+       _config->FindB("APT::Get::Assume-Yes",false) == true)
    {
       if (Fail == true && _config->FindB("APT::Get::Force-Yes",false) == false)
         return _error->Error("There are problems and -y was used without --force-yes");
    }         
-   
-   // Prompt to continue
-   if (Ask == true)
-   {            
-      if (_config->FindI("quiet",0) < 2 ||
-         _config->FindB("APT::Get::Assume-Yes",false) == false)
-        c2out << "Do you want to continue? [Y/n] " << flush;
 
-      if (YnPrompt() == false)
+   if (Essential == true && Saftey == true)
+   {
+      c2out << "You are about to do something potentially harmful" << endl;
+      c2out << "To continue type in the phrase 'Yes, I understand this is bad'" << endl;
+      c2out << " ?] " << flush;
+      if (AnalPrompt("Yes, I understand this is bad") == false)
+      {
+        c2out << "Abort." << endl;
         exit(1);
-   }      
-
+      }     
+   }
+   else
+   {
+      // Prompt to continue
+      if (Ask == true)
+      {            
+        if (_config->FindI("quiet",0) < 2 &&
+            _config->FindB("APT::Get::Assume-Yes",false) == false)
+        {
+           c2out << "Do you want to continue? [Y/n] " << flush;
+        
+           if (YnPrompt() == false)
+           {
+              c2out << "Abort." << endl;
+              exit(1);
+           }     
+        }       
+      }      
+   }
+   
    if (_config->FindB("APT::Get::Print-URIs") == true)
    {
       pkgAcquire::UriIterator I = Fetcher.UriBegin();
@@ -1174,6 +1209,10 @@ int main(int argc,const char *argv[])
        CmdL.FileSize() == 0)
       return ShowHelp(CmdL);
 
+   // Deal with stdout not being a tty
+   if (ttyname(STDOUT_FILENO) == 0 && _config->FindI("quiet",0) < 1)
+      _config->Set("quiet","1");
+   
    // Setup the output streams
    c0out.rdbuf(cout.rdbuf());
    c1out.rdbuf(cout.rdbuf());