]> git.saurik.com Git - apt.git/blobdiff - apt-pkg/pkgcachegen.cc
More cache usage counters
[apt.git] / apt-pkg / pkgcachegen.cc
index e0f8f3a197c6ffbbae51193da2f283170656c749..7cf332b68da4ed4a192aed8f57dda3439ab0e34c 100644 (file)
@@ -1,6 +1,6 @@
 // -*- mode: cpp; mode: fold -*-
 // Description                                                         /*{{{*/
-// $Id: pkgcachegen.cc,v 1.9 1998/07/12 23:58:34 jgg Exp $
+// $Id: pkgcachegen.cc,v 1.26 1998/12/14 03:39:15 jgg Exp $
 /* ######################################################################
    
    Package Cache Generator - Generator for the cache structure.
 #include <apt-pkg/pkgcachegen.h>
 #include <apt-pkg/error.h>
 #include <apt-pkg/version.h>
+#include <apt-pkg/progress.h>
+#include <apt-pkg/sourcelist.h>
+#include <apt-pkg/configuration.h>
+#include <apt-pkg/deblistparser.h>
+
 #include <strutl.h>
 
 #include <sys/stat.h>
@@ -26,7 +31,8 @@
 // CacheGenerator::pkgCacheGenerator - Constructor                     /*{{{*/
 // ---------------------------------------------------------------------
 /* We set the diry flag and make sure that is written to the disk */
-pkgCacheGenerator::pkgCacheGenerator(DynamicMMap &Map) : Map(Map), Cache(Map)
+pkgCacheGenerator::pkgCacheGenerator(DynamicMMap &Map,OpProgress &Prog) :
+                    Map(Map), Cache(Map), Progress(Prog)
 {
    if (_error->PendingError() == true)
       return;
@@ -64,18 +70,18 @@ bool pkgCacheGenerator::MergeList(ListParser &List)
 {
    List.Owner = this;
 
+   int Counter = 0;
    while (List.Step() == true)
    {
       // Get a pointer to the package structure
       string PackageName = List.Package();
       pkgCache::PkgIterator Pkg;
-      Cache.FindPkg(PackageName);
-      if (Pkg.end() == true)
-      {
-        if (NewPackage(Pkg,PackageName) == false)
-           return false;
-      }
-      
+      if (NewPackage(Pkg,PackageName) == false)
+        return _error->Error("Error occured while processing %s (NewPackage)",PackageName.c_str());
+      Counter++;
+      if (Counter % 100 == 0)
+        Progress.Progress(List.Offset());
+
       /* Get a pointer to the version structure. We know the list is sorted
          so we use that fact in the search. Insertion of new versions is
         done with correct sorting */
@@ -83,12 +89,12 @@ bool pkgCacheGenerator::MergeList(ListParser &List)
       if (Version.empty() == true)
       {
         if (List.UsePackage(Pkg,pkgCache::VerIterator(Cache)) == false)
-           return false;
+           return _error->Error("Error occured while processing %s (UsePackage1)",PackageName.c_str());
         continue;
       }
 
       pkgCache::VerIterator Ver = Pkg.VersionList();
-      unsigned long *Last = &Pkg->VersionList;
+      __apt_ptrloc *Last = &Pkg->VersionList;
       int Res = 1;
       for (; Ver.end() == false; Last = &Ver->NextVer, Ver++)
       {
@@ -103,10 +109,10 @@ bool pkgCacheGenerator::MergeList(ListParser &List)
       if (Res == 0)
       {
         if (List.UsePackage(Pkg,Ver) == false)
-           return false;
+           return _error->Error("Error occured while processing %s (UsePackage2)",PackageName.c_str());
         
         if (NewFileVer(Ver,List) == false)
-           return false;
+           return _error->Error("Error occured while processing %s (NewFileVer1)",PackageName.c_str());
         
         continue;
       }      
@@ -115,13 +121,13 @@ bool pkgCacheGenerator::MergeList(ListParser &List)
       *Last = NewVersion(Ver,Version,*Last);
       Ver->ParentPkg = Pkg.Index();
       if (List.NewVersion(Ver) == false)
-        return false;
+        return _error->Error("Error occured while processing %s (NewVersion1)",PackageName.c_str());
 
       if (List.UsePackage(Pkg,Ver) == false)
-        return false;
+        return _error->Error("Error occured while processing %s (UsePackage3)",PackageName.c_str());
       
       if (NewFileVer(Ver,List) == false)
-        return false;
+        return _error->Error("Error occured while processing %s (NewVersion2)",PackageName.c_str());
    }
 
    return true;
@@ -132,6 +138,10 @@ bool pkgCacheGenerator::MergeList(ListParser &List)
 /* This creates a new package structure and adds it to the hash table */
 bool pkgCacheGenerator::NewPackage(pkgCache::PkgIterator &Pkg,string Name)
 {
+   Pkg = Cache.FindPkg(Name);
+   if (Pkg.end() == false)
+      return true;
+       
    // Get a structure
    unsigned long Package = Map.Allocate(sizeof(pkgCache::Package));
    if (Package == 0)
@@ -166,11 +176,20 @@ bool pkgCacheGenerator::NewFileVer(pkgCache::VerIterator &Ver,
    
    pkgCache::VerFileIterator VF(Cache,Cache.VerFileP + VerFile);
    VF->File = CurrentFile - Cache.PkgFileP;
-   VF->NextFile = Ver->FileList;
-   Ver->FileList = VF.Index();
+   
+   // Link it to the end of the list
+   __apt_ptrloc *Last = &Ver->FileList;
+   for (pkgCache::VerFileIterator V = Ver.FileList(); V.end() == false; V++)
+      Last = &V->NextFile;
+   VF->NextFile = *Last;
+   *Last = VF.Index();
+   
    VF->Offset = List.Offset();
    VF->Size = List.Size();
-      
+   if (Cache.HeaderP->MaxVerFileSize < VF->Size)
+      Cache.HeaderP->MaxVerFileSize = VF->Size;
+   Cache.HeaderP->VerFileCount++;
+   
    return true;
 }
                                                                        /*}}}*/
@@ -222,10 +241,9 @@ bool pkgCacheGenerator::ListParser::NewDepends(pkgCache::VerIterator Ver,
    Dep->ID = Cache.HeaderP->DependsCount++;
    
    // Locate the target package
-   pkgCache::PkgIterator Pkg = Cache.FindPkg(PackageName);
-   if (Pkg.end() == true)
-      if (Owner->NewPackage(Pkg,PackageName) == false)
-        return false;
+   pkgCache::PkgIterator Pkg;
+   if (Owner->NewPackage(Pkg,PackageName) == false)
+      return false;
    
    // Probe the reverse dependency list for a version string that matches
    if (Version.empty() == false)
@@ -237,19 +255,28 @@ bool pkgCacheGenerator::ListParser::NewDepends(pkgCache::VerIterator Ver,
         if ((Dep->Version = WriteString(Version)) == 0)
            return false;
    }
-   
+      
    // Link it to the package
    Dep->Package = Pkg.Index();
    Dep->NextRevDepends = Pkg->RevDepends;
    Pkg->RevDepends = Dep.Index();
    
-   // Link it to the version (at the end of the list)
-   unsigned long *Last = &Ver->DependsList;
-   for (pkgCache::DepIterator D = Ver.DependsList(); D.end() == false; D++)
-      Last = &D->NextDepends;
-   Dep->NextDepends = *Last;
-   *Last = Dep.Index();
+   /* Link it to the version (at the end of the list)
+      Caching the old end point speeds up generation substantially */
+   static pkgCache::VerIterator OldVer(Cache);
+   static __apt_ptrloc *OldLast;
+   if (OldVer != Ver)
+   {
+      OldLast = &Ver->DependsList;
+      for (pkgCache::DepIterator D = Ver.DependsList(); D.end() == false; D++)
+        OldLast = &D->NextDepends;
+      OldVer = Ver;
+   }
    
+   Dep->NextDepends = *OldLast;
+   *OldLast = Dep.Index();
+   OldLast = &Dep->NextDepends;
+
    return true;
 }
                                                                        /*}}}*/
@@ -257,15 +284,20 @@ bool pkgCacheGenerator::ListParser::NewDepends(pkgCache::VerIterator Ver,
 // ---------------------------------------------------------------------
 /* */
 bool pkgCacheGenerator::ListParser::NewProvides(pkgCache::VerIterator Ver,
-                                              string PackageName,
+                                               string PackageName,
                                                string Version)
 {
    pkgCache &Cache = Owner->Cache;
+
+   // We do not add self referencing provides
+   if (Ver.ParentPkg().Name() == PackageName)
+      return true;
    
    // Get a structure
    unsigned long Provides = Owner->Map.Allocate(sizeof(pkgCache::Provides));
    if (Provides == 0)
       return false;
+   Cache.HeaderP->ProvidesCount++;
    
    // Fill it in
    pkgCache::PrvIterator Prv(Cache,Cache.ProvideP + Provides,Cache.PkgP);
@@ -276,10 +308,9 @@ bool pkgCacheGenerator::ListParser::NewProvides(pkgCache::VerIterator Ver,
       return false;
    
    // Locate the target package
-   pkgCache::PkgIterator Pkg = Cache.FindPkg(PackageName);
-   if (Pkg.end() == true)
-      if (Owner->NewPackage(Pkg,PackageName) == false)
-        return false;
+   pkgCache::PkgIterator Pkg;
+   if (Owner->NewPackage(Pkg,PackageName) == false)
+      return false;
    
    // Link it to the package
    Prv->ParentPkg = Pkg.Index();
@@ -310,10 +341,16 @@ bool pkgCacheGenerator::SelectFile(string File,unsigned long Flags)
    CurrentFile->mtime = Buf.st_mtime;
    CurrentFile->NextFile = Cache.HeaderP->FileList;
    CurrentFile->Flags = Flags;
+   CurrentFile->ID = Cache.HeaderP->PackageFileCount;
    PkgFileName = File;
-
+   Cache.HeaderP->FileList = CurrentFile - Cache.PkgFileP;
+   Cache.HeaderP->PackageFileCount++;
+      
    if (CurrentFile->FileName == 0)
       return false;
+   
+   Progress.SubProgress(Buf.st_size);
+   return true;
 }
                                                                        /*}}}*/
 // CacheGenerator::WriteUniqueString - Insert a unique string          /*{{{*/
@@ -326,7 +363,7 @@ unsigned long pkgCacheGenerator::WriteUniqString(const char *S,
    // Search for an insertion point
    pkgCache::StringItem *I = Cache.StringItemP + Cache.HeaderP->StringList;
    int Res = 1;
-   unsigned long *Last = &Cache.HeaderP->StringList;
+   __apt_ptrloc *Last = &Cache.HeaderP->StringList;
    for (; I != Cache.StringItemP; Last = &I->NextItem, 
         I = Cache.StringItemP + I->NextItem)
    {
@@ -355,3 +392,319 @@ unsigned long pkgCacheGenerator::WriteUniqString(const char *S,
    return ItemP->String;
 }
                                                                        /*}}}*/
+
+// SrcCacheCheck - Check if the source package cache is uptodate       /*{{{*/
+// ---------------------------------------------------------------------
+/* The source cache is checked against the source list and the files 
+   on disk, any difference results in a false. */
+bool pkgSrcCacheCheck(pkgSourceList &List)
+{
+   if (_error->PendingError() == true)
+      return false;
+
+   string CacheFile = _config->FindFile("Dir::Cache::srcpkgcache");
+   string ListDir = _config->FindDir("Dir::State::lists");
+
+   // Count the number of missing files
+   int Missing = 0;
+   for (pkgSourceList::const_iterator I = List.begin(); I != List.end(); I++)
+   {
+      string File = ListDir + URItoFileName(I->PackagesURI());
+      struct stat Buf;
+      if (stat(File.c_str(),&Buf) != 0)
+      {
+        _error->WarningE("stat","Couldn't stat source package list '%s' (%s)",
+                         I->PackagesInfo().c_str(),File.c_str());       
+        Missing++;
+      }      
+   }
+      
+   // Open the source package cache
+   if (FileExists(CacheFile) == false)
+      return false;
+   
+   FileFd CacheF(CacheFile,FileFd::ReadOnly);
+   if (_error->PendingError() == true)
+   {
+      _error->Discard();
+      return false;
+   }
+   
+   MMap Map(CacheF,MMap::Public | MMap::ReadOnly);
+   if (_error->PendingError() == true || Map.Size() == 0)
+   {
+      _error->Discard();
+      return false;
+   }
+   
+   pkgCache Cache(Map);
+   if (_error->PendingError() == true)
+   {
+      _error->Discard();
+      return false;
+   }
+
+   // They are certianly out of sync
+   if (Cache.Head().PackageFileCount != List.size() - Missing)
+      return false;
+   
+   for (pkgCache::PkgFileIterator F(Cache); F.end() == false; F++)
+   {
+      // Search for a match in the source list
+      bool Bad = true;
+      for (pkgSourceList::const_iterator I = List.begin(); 
+          I != List.end(); I++)
+      {
+        string File = ListDir + URItoFileName(I->PackagesURI());
+        if (F.FileName() == File)
+        {
+           Bad = false;
+           break;
+        }
+      }
+      
+      // Check if the file matches what was cached
+      Bad |= !F.IsOk();
+      if (Bad == true)
+        return false;
+   }
+   
+   return true;
+}
+                                                                       /*}}}*/
+// PkgCacheCheck - Check if the package cache is uptodate              /*{{{*/
+// ---------------------------------------------------------------------
+/* This does a simple check of all files used to compose the cache */
+bool pkgPkgCacheCheck(string CacheFile)
+{
+   if (_error->PendingError() == true)
+      return false;
+   
+   // Open the source package cache
+   if (FileExists(CacheFile) == false)
+      return false;
+   
+   FileFd CacheF(CacheFile,FileFd::ReadOnly);
+   if (_error->PendingError() == true)
+   {
+      _error->Discard();
+      return false;
+   }
+   
+   MMap Map(CacheF,MMap::Public | MMap::ReadOnly);
+   if (_error->PendingError() == true || Map.Size() == 0)
+   {
+      _error->Discard();
+      return false;
+   }
+   
+   pkgCache Cache(Map);
+   if (_error->PendingError() == true)
+   {
+      _error->Discard();
+      return false;
+   }
+
+   // Status files that must be in the cache
+   string Status[3];
+   Status[0] = _config->FindFile("Dir::State::xstatus");
+   Status[1]= _config->FindFile("Dir::State::userstatus");
+   Status[2] = _config->FindFile("Dir::State::status");
+   
+   // Cheack each file
+   for (pkgCache::PkgFileIterator F(Cache); F.end() == false; F++)
+   {
+      if (F.IsOk() == false)
+        return false;
+      
+      // See if this is one of the status files
+      for (int I = 0; I != 3; I++)
+        if (F.FileName() == Status[I])
+           Status[I] = string();
+   }
+   
+   // Make sure all the status files are loaded.
+   for (int I = 0; I != 3; I++)
+   {
+      if (Status[I].empty() == false && FileExists(Status[I]) == true)
+        return false;
+   }   
+   
+   return true;
+}
+                                                                       /*}}}*/
+// AddSourcesSize - Add the size of the status files                   /*{{{*/
+// ---------------------------------------------------------------------
+/* This adds the size of all the status files to the size counter */
+static bool pkgAddSourcesSize(unsigned long &TotalSize)
+{
+   // Grab the file names
+   string xstatus = _config->FindFile("Dir::State::xstatus");
+   string userstatus = _config->FindFile("Dir::State::userstatus");
+   string status = _config->FindFile("Dir::State::status");
+   
+   // Grab the sizes
+   struct stat Buf;
+   if (stat(xstatus.c_str(),&Buf) == 0)
+      TotalSize += Buf.st_size;
+   if (stat(userstatus.c_str(),&Buf) == 0)
+      TotalSize += Buf.st_size;
+   if (stat(status.c_str(),&Buf) != 0)
+      return _error->Errno("stat","Couldn't stat the status file %s",status.c_str());
+   TotalSize += Buf.st_size;
+   
+   return true;
+}
+                                                                       /*}}}*/
+// MergeStatus - Add the status files to the cache                     /*{{{*/
+// ---------------------------------------------------------------------
+/* This adds the status files to the map */
+static bool pkgMergeStatus(OpProgress &Progress,pkgCacheGenerator &Gen,
+                          unsigned long &CurrentSize,unsigned long TotalSize)
+{
+   // Grab the file names   
+   string Status[3];
+   Status[0] = _config->FindFile("Dir::State::xstatus");
+   Status[1]= _config->FindFile("Dir::State::userstatus");
+   Status[2] = _config->FindFile("Dir::State::status");
+   
+   for (int I = 0; I != 3; I++)
+   {
+      // Check if the file exists and it is not the primary status file.
+      string File = Status[I];
+      if (I != 2 && FileExists(File) == false)
+        continue;
+        
+      FileFd Pkg(File,FileFd::ReadOnly);
+      debListParser Parser(Pkg);
+      Progress.OverallProgress(CurrentSize,TotalSize,Pkg.Size(),"Reading Package Lists");
+      if (_error->PendingError() == true)
+        return _error->Error("Problem opening %s",File.c_str());
+      CurrentSize += Pkg.Size();
+
+      Progress.SubProgress(0,"Local Package State - " + flNotDir(File));
+      if (Gen.SelectFile(File,pkgCache::Flag::NotSource) == false)
+        return _error->Error("Problem with SelectFile %s",File.c_str());
+      
+      if (Gen.MergeList(Parser) == false)
+        return _error->Error("Problem with MergeList %s",File.c_str());
+      Progress.Progress(Pkg.Size());
+   }
+   
+   return true;
+}
+                                                                       /*}}}*/
+// MakeStatusCache - Generates a cache that includes the status files  /*{{{*/
+// ---------------------------------------------------------------------
+/* This copies the package source cache and then merges the status and 
+   xstatus files into it. */
+bool pkgMakeStatusCache(pkgSourceList &List,OpProgress &Progress)
+{
+   Progress.OverallProgress(0,1,1,"Reading Package Lists");
+   
+   string CacheFile = _config->FindFile("Dir::Cache::pkgcache");
+   bool SrcOk = pkgSrcCacheCheck(List);
+   bool PkgOk = SrcOk && pkgPkgCacheCheck(CacheFile);
+   
+   // Rebuild the source and package caches   
+   if (SrcOk == false)
+   {      
+      string SCacheFile = _config->FindFile("Dir::Cache::srcpkgcache");
+      string ListDir = _config->FindDir("Dir::State::lists");
+      FileFd SCacheF(SCacheFile,FileFd::WriteEmpty);
+      FileFd CacheF(CacheFile,FileFd::WriteEmpty);
+      DynamicMMap Map(CacheF,MMap::Public);
+      if (_error->PendingError() == true)
+        return false;
+      
+      pkgCacheGenerator Gen(Map,Progress);
+      
+      // Prepare the progress indicator
+      unsigned long TotalSize = 0;
+      struct stat Buf;
+      for (pkgSourceList::const_iterator I = List.begin(); I != List.end(); I++)
+      {
+        string File = ListDir + URItoFileName(I->PackagesURI());
+        if (stat(File.c_str(),&Buf) != 0)
+           continue;
+        TotalSize += Buf.st_size;
+      }
+      
+      if (pkgAddSourcesSize(TotalSize) == false)
+        return false;
+      
+      // Generate the pkg source cache
+      unsigned long CurrentSize = 0;
+      for (pkgSourceList::const_iterator I = List.begin(); I != List.end(); I++)
+      {
+        string File = ListDir + URItoFileName(I->PackagesURI());
+        
+        if (FileExists(File) == false)
+           continue;
+        
+        FileFd Pkg(File,FileFd::ReadOnly);
+        debListParser Parser(Pkg);
+        Progress.OverallProgress(CurrentSize,TotalSize,Pkg.Size(),"Reading Package Lists");
+        if (_error->PendingError() == true)
+           return _error->Error("Problem opening %s",File.c_str());
+        CurrentSize += Pkg.Size();
+
+        Progress.SubProgress(0,I->PackagesInfo());
+        if (Gen.SelectFile(File) == false)
+           return _error->Error("Problem with SelectFile %s",File.c_str());
+        
+        if (Gen.MergeList(Parser) == false)
+           return _error->Error("Problem with MergeList %s",File.c_str());
+
+        // Check the release file
+        string RFile = ListDir + URItoFileName(I->ReleaseURI());
+        if (FileExists(RFile) == true)
+        {
+           FileFd Rel(RFile,FileFd::ReadOnly);
+           if (_error->PendingError() == true)
+              return false;
+           Parser.LoadReleaseInfo(Gen.GetCurFile(),Rel);
+        }
+      }               
+      
+      // Write the src cache
+      Gen.GetCache().HeaderP->Dirty = false;
+      if (SCacheF.Write(Map.Data(),Map.Size()) == false)
+        return _error->Error("IO Error saving source cache");
+      Gen.GetCache().HeaderP->Dirty = true;
+      
+      // Merge in the source caches
+      return pkgMergeStatus(Progress,Gen,CurrentSize,TotalSize);
+   }
+
+   if (PkgOk == true)
+   {
+      Progress.OverallProgress(1,1,1,"Reading Package Lists");      
+      return true;
+   }
+   
+   // We use the source cache to generate the package cache
+   string SCacheFile = _config->FindFile("Dir::Cache::srcpkgcache");
+
+   FileFd SCacheF(SCacheFile,FileFd::ReadOnly);
+   FileFd CacheF(CacheFile,FileFd::WriteEmpty);
+   DynamicMMap Map(CacheF,MMap::Public);
+   if (_error->PendingError() == true)
+      return false;
+   
+   // Preload the map with the source cache
+   if (SCacheF.Read((unsigned char *)Map.Data() + Map.RawAllocate(SCacheF.Size()),
+                   SCacheF.Size()) == false)
+      return false;
+      
+   pkgCacheGenerator Gen(Map,Progress);
+   
+   // Compute the progress
+   unsigned long TotalSize = 0;
+   if (pkgAddSourcesSize(TotalSize) == false)
+      return false;
+
+   unsigned long CurrentSize = 0;
+   return pkgMergeStatus(Progress,Gen,CurrentSize,TotalSize);
+}
+                                                                       /*}}}*/