]>
git.saurik.com Git - apt.git/blob - apt-pkg/pkgcachegen.cc
1 // -*- mode: cpp; mode: fold -*-
3 // $Id: pkgcachegen.cc,v 1.24 1998/12/07 07:26:21 jgg Exp $
4 /* ######################################################################
6 Package Cache Generator - Generator for the cache structure.
8 This builds the cache structure from the abstract package list parser.
10 ##################################################################### */
12 // Include Files /*{{{*/
14 #pragma implementation "apt-pkg/pkgcachegen.h"
17 #include <apt-pkg/pkgcachegen.h>
18 #include <apt-pkg/error.h>
19 #include <apt-pkg/version.h>
20 #include <apt-pkg/progress.h>
21 #include <apt-pkg/sourcelist.h>
22 #include <apt-pkg/configuration.h>
23 #include <apt-pkg/deblistparser.h>
31 // CacheGenerator::pkgCacheGenerator - Constructor /*{{{*/
32 // ---------------------------------------------------------------------
33 /* We set the diry flag and make sure that is written to the disk */
34 pkgCacheGenerator::pkgCacheGenerator(DynamicMMap
&Map
,OpProgress
&Prog
) :
35 Map(Map
), Cache(Map
), Progress(Prog
)
37 if (_error
->PendingError() == true)
42 Map
.RawAllocate(sizeof(pkgCache::Header
));
43 *Cache
.HeaderP
= pkgCache::Header();
45 Cache
.HeaderP
->Dirty
= true;
46 Map
.Sync(0,sizeof(pkgCache::Header
));
47 Map
.UsePools(*Cache
.HeaderP
->Pools
,sizeof(Cache
.HeaderP
->Pools
)/sizeof(Cache
.HeaderP
->Pools
[0]));
50 // CacheGenerator::~pkgCacheGenerator - Destructor /*{{{*/
51 // ---------------------------------------------------------------------
52 /* We sync the data then unset the dirty flag in two steps so as to
53 advoid a problem during a crash */
54 pkgCacheGenerator::~pkgCacheGenerator()
56 if (_error
->PendingError() == true)
58 if (Map
.Sync() == false)
61 Cache
.HeaderP
->Dirty
= false;
62 Map
.Sync(0,sizeof(pkgCache::Header
));
65 // CacheGenerator::MergeList - Merge the package list /*{{{*/
66 // ---------------------------------------------------------------------
67 /* This provides the generation of the entries in the cache. Each loop
68 goes through a single package record from the underlying parse engine. */
69 bool pkgCacheGenerator::MergeList(ListParser
&List
)
74 while (List
.Step() == true)
76 // Get a pointer to the package structure
77 string PackageName
= List
.Package();
78 pkgCache::PkgIterator Pkg
;
79 if (NewPackage(Pkg
,PackageName
) == false)
80 return _error
->Error("Error occured while processing %s (NewPackage)",PackageName
.c_str());
82 if (Counter
% 100 == 0)
83 Progress
.Progress(List
.Offset());
85 /* Get a pointer to the version structure. We know the list is sorted
86 so we use that fact in the search. Insertion of new versions is
87 done with correct sorting */
88 string Version
= List
.Version();
89 if (Version
.empty() == true)
91 if (List
.UsePackage(Pkg
,pkgCache::VerIterator(Cache
)) == false)
92 return _error
->Error("Error occured while processing %s (UsePackage1)",PackageName
.c_str());
96 pkgCache::VerIterator Ver
= Pkg
.VersionList();
97 __apt_ptrloc
*Last
= &Pkg
->VersionList
;
99 for (; Ver
.end() == false; Last
= &Ver
->NextVer
, Ver
++)
101 Res
= pkgVersionCompare(Version
.begin(),Version
.end(),Ver
.VerStr(),
102 Ver
.VerStr() + strlen(Ver
.VerStr()));
107 /* We already have a version for this item, record that we
111 if (List
.UsePackage(Pkg
,Ver
) == false)
112 return _error
->Error("Error occured while processing %s (UsePackage2)",PackageName
.c_str());
114 if (NewFileVer(Ver
,List
) == false)
115 return _error
->Error("Error occured while processing %s (NewFileVer1)",PackageName
.c_str());
121 *Last
= NewVersion(Ver
,Version
,*Last
);
122 Ver
->ParentPkg
= Pkg
.Index();
123 if (List
.NewVersion(Ver
) == false)
124 return _error
->Error("Error occured while processing %s (NewVersion1)",PackageName
.c_str());
126 if (List
.UsePackage(Pkg
,Ver
) == false)
127 return _error
->Error("Error occured while processing %s (UsePackage3)",PackageName
.c_str());
129 if (NewFileVer(Ver
,List
) == false)
130 return _error
->Error("Error occured while processing %s (NewVersion2)",PackageName
.c_str());
136 // CacheGenerator::NewPackage - Add a new package /*{{{*/
137 // ---------------------------------------------------------------------
138 /* This creates a new package structure and adds it to the hash table */
139 bool pkgCacheGenerator::NewPackage(pkgCache::PkgIterator
&Pkg
,string Name
)
141 Pkg
= Cache
.FindPkg(Name
);
142 if (Pkg
.end() == false)
146 unsigned long Package
= Map
.Allocate(sizeof(pkgCache::Package
));
150 Pkg
= pkgCache::PkgIterator(Cache
,Cache
.PkgP
+ Package
);
152 // Insert it into the hash table
153 unsigned long Hash
= Cache
.Hash(Name
);
154 Pkg
->NextPackage
= Cache
.HeaderP
->HashTable
[Hash
];
155 Cache
.HeaderP
->HashTable
[Hash
] = Package
;
157 // Set the name and the ID
158 Pkg
->Name
= Map
.WriteString(Name
);
161 Pkg
->ID
= Cache
.HeaderP
->PackageCount
++;
166 // CacheGenerator::NewFileVer - Create a new File<->Version association /*{{{*/
167 // ---------------------------------------------------------------------
169 bool pkgCacheGenerator::NewFileVer(pkgCache::VerIterator
&Ver
,
173 unsigned long VerFile
= Map
.Allocate(sizeof(pkgCache::VerFile
));
177 pkgCache::VerFileIterator
VF(Cache
,Cache
.VerFileP
+ VerFile
);
178 VF
->File
= CurrentFile
- Cache
.PkgFileP
;
180 // Link it to the end of the list
181 __apt_ptrloc
*Last
= &Ver
->FileList
;
182 for (pkgCache::VerFileIterator V
= Ver
.FileList(); V
.end() == false; V
++)
184 VF
->NextFile
= *Last
;
187 VF
->Offset
= List
.Offset();
188 VF
->Size
= List
.Size();
189 if (Cache
.HeaderP
->MaxVerFileSize
< VF
->Size
)
190 Cache
.HeaderP
->MaxVerFileSize
= VF
->Size
;
194 // CacheGenerator::NewVersion - Create a new Version /*{{{*/
195 // ---------------------------------------------------------------------
196 /* This puts a version structure in the linked list */
197 unsigned long pkgCacheGenerator::NewVersion(pkgCache::VerIterator
&Ver
,
202 unsigned long Version
= Map
.Allocate(sizeof(pkgCache::Version
));
207 Ver
= pkgCache::VerIterator(Cache
,Cache
.VerP
+ Version
);
209 Ver
->ID
= Cache
.HeaderP
->VersionCount
++;
210 Ver
->VerStr
= Map
.WriteString(VerStr
);
211 if (Ver
->VerStr
== 0)
217 // ListParser::NewDepends - Create a dependency element /*{{{*/
218 // ---------------------------------------------------------------------
219 /* This creates a dependency element in the tree. It is linked to the
220 version and to the package that it is pointing to. */
221 bool pkgCacheGenerator::ListParser::NewDepends(pkgCache::VerIterator Ver
,
227 pkgCache
&Cache
= Owner
->Cache
;
230 unsigned long Dependency
= Owner
->Map
.Allocate(sizeof(pkgCache::Dependency
));
235 pkgCache::DepIterator
Dep(Cache
,Cache
.DepP
+ Dependency
);
236 Dep
->ParentVer
= Ver
.Index();
239 Dep
->ID
= Cache
.HeaderP
->DependsCount
++;
241 // Locate the target package
242 pkgCache::PkgIterator Pkg
;
243 if (Owner
->NewPackage(Pkg
,PackageName
) == false)
246 // Probe the reverse dependency list for a version string that matches
247 if (Version
.empty() == false)
249 for (pkgCache::DepIterator I
= Pkg
.RevDependsList(); I
.end() == false; I
++)
250 if (I
->Version
!= 0 && I
.TargetVer() == Version
)
251 Dep
->Version
= I
->Version
;
252 if (Dep
->Version
== 0)
253 if ((Dep
->Version
= WriteString(Version
)) == 0)
257 // Link it to the package
258 Dep
->Package
= Pkg
.Index();
259 Dep
->NextRevDepends
= Pkg
->RevDepends
;
260 Pkg
->RevDepends
= Dep
.Index();
262 /* Link it to the version (at the end of the list)
263 Caching the old end point speeds up generation substantially */
264 static pkgCache::VerIterator
OldVer(Cache
);
265 static __apt_ptrloc
*OldLast
;
268 OldLast
= &Ver
->DependsList
;
269 for (pkgCache::DepIterator D
= Ver
.DependsList(); D
.end() == false; D
++)
270 OldLast
= &D
->NextDepends
;
274 Dep
->NextDepends
= *OldLast
;
275 *OldLast
= Dep
.Index();
276 OldLast
= &Dep
->NextDepends
;
281 // ListParser::NewProvides - Create a Provides element /*{{{*/
282 // ---------------------------------------------------------------------
284 bool pkgCacheGenerator::ListParser::NewProvides(pkgCache::VerIterator Ver
,
288 pkgCache
&Cache
= Owner
->Cache
;
290 // We do not add self referencing provides
291 if (Ver
.ParentPkg().Name() == PackageName
)
295 unsigned long Provides
= Owner
->Map
.Allocate(sizeof(pkgCache::Provides
));
300 pkgCache::PrvIterator
Prv(Cache
,Cache
.ProvideP
+ Provides
,Cache
.PkgP
);
301 Prv
->Version
= Ver
.Index();
302 Prv
->NextPkgProv
= Ver
->ProvidesList
;
303 Ver
->ProvidesList
= Prv
.Index();
304 if (Version
.empty() == false && (Prv
->Version
= WriteString(Version
)) == 0)
307 // Locate the target package
308 pkgCache::PkgIterator Pkg
;
309 if (Owner
->NewPackage(Pkg
,PackageName
) == false)
312 // Link it to the package
313 Prv
->ParentPkg
= Pkg
.Index();
314 Prv
->NextProvides
= Pkg
->ProvidesList
;
315 Pkg
->ProvidesList
= Prv
.Index();
320 // CacheGenerator::SelectFile - Select the current file being parsed /*{{{*/
321 // ---------------------------------------------------------------------
322 /* This is used to select which file is to be associated with all newly
324 bool pkgCacheGenerator::SelectFile(string File
,unsigned long Flags
)
327 if (stat(File
.c_str(),&Buf
) == -1)
328 return _error
->Errno("stat","Couldn't stat ",File
.c_str());
330 // Get some space for the structure
331 CurrentFile
= Cache
.PkgFileP
+ Map
.Allocate(sizeof(*CurrentFile
));
332 if (CurrentFile
== Cache
.PkgFileP
)
336 CurrentFile
->FileName
= Map
.WriteString(File
);
337 CurrentFile
->Size
= Buf
.st_size
;
338 CurrentFile
->mtime
= Buf
.st_mtime
;
339 CurrentFile
->NextFile
= Cache
.HeaderP
->FileList
;
340 CurrentFile
->Flags
= Flags
;
341 CurrentFile
->ID
= Cache
.HeaderP
->PackageFileCount
;
343 Cache
.HeaderP
->FileList
= CurrentFile
- Cache
.PkgFileP
;
344 Cache
.HeaderP
->PackageFileCount
++;
346 if (CurrentFile
->FileName
== 0)
349 Progress
.SubProgress(Buf
.st_size
);
353 // CacheGenerator::WriteUniqueString - Insert a unique string /*{{{*/
354 // ---------------------------------------------------------------------
355 /* This is used to create handles to strings. Given the same text it
356 always returns the same number */
357 unsigned long pkgCacheGenerator::WriteUniqString(const char *S
,
360 // Search for an insertion point
361 pkgCache::StringItem
*I
= Cache
.StringItemP
+ Cache
.HeaderP
->StringList
;
363 __apt_ptrloc
*Last
= &Cache
.HeaderP
->StringList
;
364 for (; I
!= Cache
.StringItemP
; Last
= &I
->NextItem
,
365 I
= Cache
.StringItemP
+ I
->NextItem
)
367 Res
= stringcmp(S
,S
+Size
,Cache
.StrP
+ I
->String
);
377 unsigned long Item
= Map
.Allocate(sizeof(pkgCache::StringItem
));
381 // Fill in the structure
382 pkgCache::StringItem
*ItemP
= Cache
.StringItemP
+ Item
;
383 ItemP
->NextItem
= I
- Cache
.StringItemP
;
385 ItemP
->String
= Map
.WriteString(S
,Size
);
386 if (ItemP
->String
== 0)
389 return ItemP
->String
;
393 // SrcCacheCheck - Check if the source package cache is uptodate /*{{{*/
394 // ---------------------------------------------------------------------
395 /* The source cache is checked against the source list and the files
396 on disk, any difference results in a false. */
397 bool pkgSrcCacheCheck(pkgSourceList
&List
)
399 if (_error
->PendingError() == true)
402 string CacheFile
= _config
->FindFile("Dir::Cache::srcpkgcache");
403 string ListDir
= _config
->FindDir("Dir::State::lists");
405 // Count the number of missing files
407 for (pkgSourceList::const_iterator I
= List
.begin(); I
!= List
.end(); I
++)
409 string File
= ListDir
+ URItoFileName(I
->PackagesURI());
411 if (stat(File
.c_str(),&Buf
) != 0)
413 _error
->WarningE("stat","Couldn't stat source package list '%s' (%s)",
414 I
->PackagesInfo().c_str(),File
.c_str());
419 // Open the source package cache
420 if (FileExists(CacheFile
) == false)
423 FileFd
CacheF(CacheFile
,FileFd::ReadOnly
);
424 if (_error
->PendingError() == true)
430 MMap
Map(CacheF
,MMap::Public
| MMap::ReadOnly
);
431 if (_error
->PendingError() == true || Map
.Size() == 0)
438 if (_error
->PendingError() == true)
444 // They are certianly out of sync
445 if (Cache
.Head().PackageFileCount
!= List
.size() - Missing
)
448 for (pkgCache::PkgFileIterator
F(Cache
); F
.end() == false; F
++)
450 // Search for a match in the source list
452 for (pkgSourceList::const_iterator I
= List
.begin();
453 I
!= List
.end(); I
++)
455 string File
= ListDir
+ URItoFileName(I
->PackagesURI());
456 if (F
.FileName() == File
)
463 // Check if the file matches what was cached
472 // PkgCacheCheck - Check if the package cache is uptodate /*{{{*/
473 // ---------------------------------------------------------------------
474 /* This does a simple check of all files used to compose the cache */
475 bool pkgPkgCacheCheck(string CacheFile
)
477 if (_error
->PendingError() == true)
480 // Open the source package cache
481 if (FileExists(CacheFile
) == false)
484 FileFd
CacheF(CacheFile
,FileFd::ReadOnly
);
485 if (_error
->PendingError() == true)
491 MMap
Map(CacheF
,MMap::Public
| MMap::ReadOnly
);
492 if (_error
->PendingError() == true || Map
.Size() == 0)
499 if (_error
->PendingError() == true)
505 // Status files that must be in the cache
507 Status
[0] = _config
->FindFile("Dir::State::xstatus");
508 Status
[1]= _config
->FindFile("Dir::State::userstatus");
509 Status
[2] = _config
->FindFile("Dir::State::status");
512 for (pkgCache::PkgFileIterator
F(Cache
); F
.end() == false; F
++)
514 if (F
.IsOk() == false)
517 // See if this is one of the status files
518 for (int I
= 0; I
!= 3; I
++)
519 if (F
.FileName() == Status
[I
])
520 Status
[I
] = string();
523 // Make sure all the status files are loaded.
524 for (int I
= 0; I
!= 3; I
++)
526 if (Status
[I
].empty() == false && FileExists(Status
[I
]) == true)
533 // AddSourcesSize - Add the size of the status files /*{{{*/
534 // ---------------------------------------------------------------------
535 /* This adds the size of all the status files to the size counter */
536 static bool pkgAddSourcesSize(unsigned long &TotalSize
)
538 // Grab the file names
539 string xstatus
= _config
->FindFile("Dir::State::xstatus");
540 string userstatus
= _config
->FindFile("Dir::State::userstatus");
541 string status
= _config
->FindFile("Dir::State::status");
545 if (stat(xstatus
.c_str(),&Buf
) == 0)
546 TotalSize
+= Buf
.st_size
;
547 if (stat(userstatus
.c_str(),&Buf
) == 0)
548 TotalSize
+= Buf
.st_size
;
549 if (stat(status
.c_str(),&Buf
) != 0)
550 return _error
->Errno("stat","Couldn't stat the status file %s",status
.c_str());
551 TotalSize
+= Buf
.st_size
;
556 // MergeStatus - Add the status files to the cache /*{{{*/
557 // ---------------------------------------------------------------------
558 /* This adds the status files to the map */
559 static bool pkgMergeStatus(OpProgress
&Progress
,pkgCacheGenerator
&Gen
,
560 unsigned long &CurrentSize
,unsigned long TotalSize
)
562 // Grab the file names
564 Status
[0] = _config
->FindFile("Dir::State::xstatus");
565 Status
[1]= _config
->FindFile("Dir::State::userstatus");
566 Status
[2] = _config
->FindFile("Dir::State::status");
568 for (int I
= 0; I
!= 3; I
++)
570 // Check if the file exists and it is not the primary status file.
571 string File
= Status
[I
];
572 if (I
!= 2 && FileExists(File
) == false)
575 FileFd
Pkg(File
,FileFd::ReadOnly
);
576 debListParser
Parser(Pkg
);
577 Progress
.OverallProgress(CurrentSize
,TotalSize
,Pkg
.Size(),"Reading Package Lists");
578 if (_error
->PendingError() == true)
579 return _error
->Error("Problem opening %s",File
.c_str());
580 CurrentSize
+= Pkg
.Size();
582 Progress
.SubProgress(0,"Local Package State - " + flNotDir(File
));
583 if (Gen
.SelectFile(File
,pkgCache::Flag::NotSource
) == false)
584 return _error
->Error("Problem with SelectFile %s",File
.c_str());
586 if (Gen
.MergeList(Parser
) == false)
587 return _error
->Error("Problem with MergeList %s",File
.c_str());
588 Progress
.Progress(Pkg
.Size());
594 // MakeStatusCache - Generates a cache that includes the status files /*{{{*/
595 // ---------------------------------------------------------------------
596 /* This copies the package source cache and then merges the status and
597 xstatus files into it. */
598 bool pkgMakeStatusCache(pkgSourceList
&List
,OpProgress
&Progress
)
600 Progress
.OverallProgress(0,1,1,"Reading Package Lists");
602 string CacheFile
= _config
->FindFile("Dir::Cache::pkgcache");
603 bool SrcOk
= pkgSrcCacheCheck(List
);
604 bool PkgOk
= SrcOk
&& pkgPkgCacheCheck(CacheFile
);
606 // Rebuild the source and package caches
609 string SCacheFile
= _config
->FindFile("Dir::Cache::srcpkgcache");
610 string ListDir
= _config
->FindDir("Dir::State::lists");
611 FileFd
SCacheF(SCacheFile
,FileFd::WriteEmpty
);
612 FileFd
CacheF(CacheFile
,FileFd::WriteEmpty
);
613 DynamicMMap
Map(CacheF
,MMap::Public
);
614 if (_error
->PendingError() == true)
617 pkgCacheGenerator
Gen(Map
,Progress
);
619 // Prepare the progress indicator
620 unsigned long TotalSize
= 0;
622 for (pkgSourceList::const_iterator I
= List
.begin(); I
!= List
.end(); I
++)
624 string File
= ListDir
+ URItoFileName(I
->PackagesURI());
625 if (stat(File
.c_str(),&Buf
) != 0)
627 TotalSize
+= Buf
.st_size
;
630 if (pkgAddSourcesSize(TotalSize
) == false)
633 // Generate the pkg source cache
634 unsigned long CurrentSize
= 0;
635 for (pkgSourceList::const_iterator I
= List
.begin(); I
!= List
.end(); I
++)
637 string File
= ListDir
+ URItoFileName(I
->PackagesURI());
639 if (stat(File
.c_str(),&Buf
) != 0)
642 FileFd
Pkg(File
,FileFd::ReadOnly
);
643 debListParser
Parser(Pkg
);
644 Progress
.OverallProgress(CurrentSize
,TotalSize
,Pkg
.Size(),"Reading Package Lists");
645 if (_error
->PendingError() == true)
646 return _error
->Error("Problem opening %s",File
.c_str());
647 CurrentSize
+= Pkg
.Size();
649 Progress
.SubProgress(0,I
->PackagesInfo());
650 if (Gen
.SelectFile(File
) == false)
651 return _error
->Error("Problem with SelectFile %s",File
.c_str());
653 if (Gen
.MergeList(Parser
) == false)
654 return _error
->Error("Problem with MergeList %s",File
.c_str());
657 // Write the src cache
658 Gen
.GetCache().HeaderP
->Dirty
= false;
659 if (SCacheF
.Write(Map
.Data(),Map
.Size()) == false)
660 return _error
->Error("IO Error saving source cache");
661 Gen
.GetCache().HeaderP
->Dirty
= true;
663 // Merge in the source caches
664 return pkgMergeStatus(Progress
,Gen
,CurrentSize
,TotalSize
);
669 Progress
.OverallProgress(1,1,1,"Reading Package Lists");
673 // We use the source cache to generate the package cache
674 string SCacheFile
= _config
->FindFile("Dir::Cache::srcpkgcache");
676 FileFd
SCacheF(SCacheFile
,FileFd::ReadOnly
);
677 FileFd
CacheF(CacheFile
,FileFd::WriteEmpty
);
678 DynamicMMap
Map(CacheF
,MMap::Public
);
679 if (_error
->PendingError() == true)
682 // Preload the map with the source cache
683 if (SCacheF
.Read((unsigned char *)Map
.Data() + Map
.RawAllocate(SCacheF
.Size()),
684 SCacheF
.Size()) == false)
687 pkgCacheGenerator
Gen(Map
,Progress
);
689 // Compute the progress
690 unsigned long TotalSize
= 0;
691 if (pkgAddSourcesSize(TotalSize
) == false)
694 unsigned long CurrentSize
= 0;
695 return pkgMergeStatus(Progress
,Gen
,CurrentSize
,TotalSize
);