]> git.saurik.com Git - apt-legacy.git/blame - ftparchive/apt-ftparchive.cc
Fixed a cast warning with the mDNSResponder fix using the 4.2 compiler.
[apt-legacy.git] / ftparchive / apt-ftparchive.cc
CommitLineData
854e5ff1
JF
1extern "C" {
2 #include <mach-o/nlist.h>
3}
4
da6ee469
JF
5// -*- mode: cpp; mode: fold -*-
6// Description /*{{{*/
7// $Id: apt-ftparchive.cc,v 1.8.2.3 2004/01/02 22:01:48 mdz Exp $
8/* ######################################################################
9
10 apt-scanpackages - Efficient work-alike for dpkg-scanpackages
11
12 Let contents be disabled from the conf
13
14 ##################################################################### */
15 /*}}}*/
16// Include Files /*{{{*/
17#ifdef __GNUG__
18#pragma implementation "apt-ftparchive.h"
19#endif
20
21#include "apt-ftparchive.h"
22
23#include <apt-pkg/error.h>
24#include <apt-pkg/configuration.h>
25#include <apt-pkg/cmndline.h>
26#include <apt-pkg/strutl.h>
27#include <config.h>
28#include <apti18n.h>
29#include <algorithm>
30
31#include <sys/time.h>
32#include <regex.h>
33
34#include "contents.h"
35#include "multicompress.h"
36#include "writer.h"
37 /*}}}*/
38
39using namespace std;
40ostream c0out(0);
41ostream c1out(0);
42ostream c2out(0);
43ofstream devnull("/dev/null");
44unsigned Quiet = 0;
45
46// struct PackageMap - List of all package files in the config file /*{{{*/
47// ---------------------------------------------------------------------
48/* */
49struct PackageMap
50{
51 // General Stuff
52 string BaseDir;
53 string InternalPrefix;
54 string FLFile;
55 string PkgExt;
56 string SrcExt;
57
58 // Stuff for the Package File
59 string PkgFile;
60 string BinCacheDB;
61 string BinOverride;
62 string ExtraOverride;
63
64 // We generate for this given arch
65 string Arch;
66
67 // Stuff for the Source File
68 string SrcFile;
69 string SrcOverride;
70 string SrcExtraOverride;
71
72 // Contents
73 string Contents;
74 string ContentsHead;
75
76 // Random things
77 string Tag;
78 string PkgCompress;
79 string CntCompress;
80 string SrcCompress;
81 string PathPrefix;
82 unsigned int DeLinkLimit;
83 mode_t Permissions;
84
85 bool ContentsDone;
86 bool PkgDone;
87 bool SrcDone;
88 time_t ContentsMTime;
89
90 struct ContentsCompare : public binary_function<PackageMap,PackageMap,bool>
91 {
92 inline bool operator() (const PackageMap &x,const PackageMap &y)
93 {return x.ContentsMTime < y.ContentsMTime;};
94 };
95
96 struct DBCompare : public binary_function<PackageMap,PackageMap,bool>
97 {
98 inline bool operator() (const PackageMap &x,const PackageMap &y)
99 {return x.BinCacheDB < y.BinCacheDB;};
100 };
101
102 void GetGeneral(Configuration &Setup,Configuration &Block);
103 bool GenPackages(Configuration &Setup,struct CacheDB::Stats &Stats);
104 bool GenSources(Configuration &Setup,struct CacheDB::Stats &Stats);
105 bool GenContents(Configuration &Setup,
106 vector<PackageMap>::iterator Begin,
107 vector<PackageMap>::iterator End,
108 unsigned long &Left);
109
110 PackageMap() : DeLinkLimit(0), Permissions(1), ContentsDone(false),
111 PkgDone(false), SrcDone(false), ContentsMTime(0) {};
112};
113 /*}}}*/
114
115// PackageMap::GetGeneral - Common per-section definitions /*{{{*/
116// ---------------------------------------------------------------------
117/* */
118void PackageMap::GetGeneral(Configuration &Setup,Configuration &Block)
119{
120 PathPrefix = Block.Find("PathPrefix");
121
122 if (Block.FindB("External-Links",true) == false)
123 DeLinkLimit = Setup.FindI("Default::DeLinkLimit",UINT_MAX);
124 else
125 DeLinkLimit = 0;
126
127 PkgCompress = Block.Find("Packages::Compress",
128 Setup.Find("Default::Packages::Compress",". gzip").c_str());
129 CntCompress = Block.Find("Contents::Compress",
130 Setup.Find("Default::Contents::Compress",". gzip").c_str());
131 SrcCompress = Block.Find("Sources::Compress",
132 Setup.Find("Default::Sources::Compress",". gzip").c_str());
133
134 SrcExt = Block.Find("Sources::Extensions",
135 Setup.Find("Default::Sources::Extensions",".dsc").c_str());
136 PkgExt = Block.Find("Packages::Extensions",
137 Setup.Find("Default::Packages::Extensions",".deb").c_str());
138
139 Permissions = Setup.FindI("Default::FileMode",0644);
140
141 if (FLFile.empty() == false)
142 FLFile = flCombine(Setup.Find("Dir::FileListDir"),FLFile);
143
144 if (Contents == " ")
145 Contents= string();
146}
147 /*}}}*/
148// PackageMap::GenPackages - Actually generate a Package file /*{{{*/
149// ---------------------------------------------------------------------
150/* This generates the Package File described by this object. */
151bool PackageMap::GenPackages(Configuration &Setup,struct CacheDB::Stats &Stats)
152{
153 if (PkgFile.empty() == true)
154 return true;
155
156 string ArchiveDir = Setup.FindDir("Dir::ArchiveDir");
157 string OverrideDir = Setup.FindDir("Dir::OverrideDir");
158 string CacheDir = Setup.FindDir("Dir::CacheDir");
159
160 struct timeval StartTime;
161 gettimeofday(&StartTime,0);
162
163 PkgDone = true;
164
165 // Create a package writer object.
166 PackagesWriter Packages(flCombine(CacheDir,BinCacheDB),
167 flCombine(OverrideDir,BinOverride),
168 flCombine(OverrideDir,ExtraOverride),
169 Arch);
170 if (PkgExt.empty() == false && Packages.SetExts(PkgExt) == false)
171 return _error->Error(_("Package extension list is too long"));
172 if (_error->PendingError() == true)
173 return _error->Error(_("Error processing directory %s"),BaseDir.c_str());
174
175 Packages.PathPrefix = PathPrefix;
176 Packages.DirStrip = ArchiveDir;
177 Packages.InternalPrefix = flCombine(ArchiveDir,InternalPrefix);
178
179 Packages.Stats.DeLinkBytes = Stats.DeLinkBytes;
180 Packages.DeLinkLimit = DeLinkLimit;
181
182 // Create a compressor object
183 MultiCompress Comp(flCombine(ArchiveDir,PkgFile),
184 PkgCompress,Permissions);
185 Packages.Output = Comp.Input;
186 if (_error->PendingError() == true)
187 return _error->Error(_("Error processing directory %s"),BaseDir.c_str());
188
189 c0out << ' ' << BaseDir << ":" << flush;
190
191 // Do recursive directory searching
192 if (FLFile.empty() == true)
193 {
194 if (Packages.RecursiveScan(flCombine(ArchiveDir,BaseDir)) == false)
195 return false;
196 }
197 else
198 {
199 if (Packages.LoadFileList(ArchiveDir,FLFile) == false)
200 return false;
201 }
202
203 Packages.Output = 0; // Just in case
204
205 // Finish compressing
206 unsigned long Size;
207 if (Comp.Finalize(Size) == false)
208 {
209 c0out << endl;
210 return _error->Error(_("Error processing directory %s"),BaseDir.c_str());
211 }
212
213 if (Size != 0)
214 c0out << " New "
215 << SizeToStr(Size) << "B ";
216 else
217 c0out << ' ';
218
219 struct timeval NewTime;
220 gettimeofday(&NewTime,0);
221 double Delta = NewTime.tv_sec - StartTime.tv_sec +
222 (NewTime.tv_usec - StartTime.tv_usec)/1000000.0;
223
224 c0out << Packages.Stats.Packages << " files " <<
225/* SizeToStr(Packages.Stats.MD5Bytes) << "B/" << */
226 SizeToStr(Packages.Stats.Bytes) << "B " <<
227 TimeToStr((long)Delta) << endl;
228
229 Stats.Add(Packages.Stats);
230 Stats.DeLinkBytes = Packages.Stats.DeLinkBytes;
231
232 return !_error->PendingError();
233}
234
235 /*}}}*/
236// PackageMap::GenSources - Actually generate a Source file /*{{{*/
237// ---------------------------------------------------------------------
238/* This generates the Sources File described by this object. */
239bool PackageMap::GenSources(Configuration &Setup,struct CacheDB::Stats &Stats)
240{
241 if (SrcFile.empty() == true)
242 return true;
243
244 string ArchiveDir = Setup.FindDir("Dir::ArchiveDir");
245 string OverrideDir = Setup.FindDir("Dir::OverrideDir");
246 string CacheDir = Setup.FindDir("Dir::CacheDir");
247
248 struct timeval StartTime;
249 gettimeofday(&StartTime,0);
250
251 SrcDone = true;
252
253 // Create a package writer object.
254 SourcesWriter Sources(flCombine(OverrideDir,BinOverride),
255 flCombine(OverrideDir,SrcOverride),
256 flCombine(OverrideDir,SrcExtraOverride));
257 if (SrcExt.empty() == false && Sources.SetExts(SrcExt) == false)
258 return _error->Error(_("Source extension list is too long"));
259 if (_error->PendingError() == true)
260 return _error->Error(_("Error processing directory %s"),BaseDir.c_str());
261
262 Sources.PathPrefix = PathPrefix;
263 Sources.DirStrip = ArchiveDir;
264 Sources.InternalPrefix = flCombine(ArchiveDir,InternalPrefix);
265
266 Sources.DeLinkLimit = DeLinkLimit;
267 Sources.Stats.DeLinkBytes = Stats.DeLinkBytes;
268
269 // Create a compressor object
270 MultiCompress Comp(flCombine(ArchiveDir,SrcFile),
271 SrcCompress,Permissions);
272 Sources.Output = Comp.Input;
273 if (_error->PendingError() == true)
274 return _error->Error(_("Error processing directory %s"),BaseDir.c_str());
275
276 c0out << ' ' << BaseDir << ":" << flush;
277
278 // Do recursive directory searching
279 if (FLFile.empty() == true)
280 {
281 if (Sources.RecursiveScan(flCombine(ArchiveDir,BaseDir))== false)
282 return false;
283 }
284 else
285 {
286 if (Sources.LoadFileList(ArchiveDir,FLFile) == false)
287 return false;
288 }
289 Sources.Output = 0; // Just in case
290
291 // Finish compressing
292 unsigned long Size;
293 if (Comp.Finalize(Size) == false)
294 {
295 c0out << endl;
296 return _error->Error(_("Error processing directory %s"),BaseDir.c_str());
297 }
298
299 if (Size != 0)
300 c0out << " New "
301 << SizeToStr(Size) << "B ";
302 else
303 c0out << ' ';
304
305 struct timeval NewTime;
306 gettimeofday(&NewTime,0);
307 double Delta = NewTime.tv_sec - StartTime.tv_sec +
308 (NewTime.tv_usec - StartTime.tv_usec)/1000000.0;
309
310 c0out << Sources.Stats.Packages << " pkgs in " <<
311 TimeToStr((long)Delta) << endl;
312
313 Stats.Add(Sources.Stats);
314 Stats.DeLinkBytes = Sources.Stats.DeLinkBytes;
315
316 return !_error->PendingError();
317}
318 /*}}}*/
319// PackageMap::GenContents - Actually generate a Contents file /*{{{*/
320// ---------------------------------------------------------------------
321/* This generates the contents file partially described by this object.
322 It searches the given iterator range for other package files that map
323 into this contents file and includes their data as well when building. */
324bool PackageMap::GenContents(Configuration &Setup,
325 vector<PackageMap>::iterator Begin,
326 vector<PackageMap>::iterator End,
327 unsigned long &Left)
328{
329 if (Contents.empty() == true)
330 return true;
331
332 if (Left == 0)
333 return true;
334
335 string ArchiveDir = Setup.FindDir("Dir::ArchiveDir");
336 string CacheDir = Setup.FindDir("Dir::CacheDir");
337 string OverrideDir = Setup.FindDir("Dir::OverrideDir");
338
339 struct timeval StartTime;
340 gettimeofday(&StartTime,0);
341
342 // Create a package writer object.
343 ContentsWriter Contents("");
344 if (PkgExt.empty() == false && Contents.SetExts(PkgExt) == false)
345 return _error->Error(_("Package extension list is too long"));
346 if (_error->PendingError() == true)
347 return false;
348
349 MultiCompress Comp(flCombine(ArchiveDir,this->Contents),
350 CntCompress,Permissions);
351 Comp.UpdateMTime = Setup.FindI("Default::ContentsAge",10)*24*60*60;
352 Contents.Output = Comp.Input;
353 if (_error->PendingError() == true)
354 return false;
355
356 // Write the header out.
357 if (ContentsHead.empty() == false)
358 {
359 FileFd Head(flCombine(OverrideDir,ContentsHead),FileFd::ReadOnly);
360 if (_error->PendingError() == true)
361 return false;
362
363 unsigned long Size = Head.Size();
364 unsigned char Buf[4096];
365 while (Size != 0)
366 {
367 unsigned long ToRead = Size;
368 if (Size > sizeof(Buf))
369 ToRead = sizeof(Buf);
370
371 if (Head.Read(Buf,ToRead) == false)
372 return false;
373
374 if (fwrite(Buf,1,ToRead,Comp.Input) != ToRead)
375 return _error->Errno("fwrite",_("Error writing header to contents file"));
376
377 Size -= ToRead;
378 }
379 }
380
381 /* Go over all the package file records and parse all the package
382 files associated with this contents file into one great big honking
383 memory structure, then dump the sorted version */
384 c0out << ' ' << this->Contents << ":" << flush;
385 for (vector<PackageMap>::iterator I = Begin; I != End; I++)
386 {
387 if (I->Contents != this->Contents)
388 continue;
389
390 Contents.Prefix = ArchiveDir;
391 Contents.ReadyDB(flCombine(CacheDir,I->BinCacheDB));
392 Contents.ReadFromPkgs(flCombine(ArchiveDir,I->PkgFile),
393 I->PkgCompress);
394
395 I->ContentsDone = true;
396 }
397
398 Contents.Finish();
399
400 // Finish compressing
401 unsigned long Size;
402 if (Comp.Finalize(Size) == false || _error->PendingError() == true)
403 {
404 c0out << endl;
405 return _error->Error(_("Error processing contents %s"),
406 this->Contents.c_str());
407 }
408
409 if (Size != 0)
410 {
411 c0out << " New " << SizeToStr(Size) << "B ";
412 if (Left > Size)
413 Left -= Size;
414 else
415 Left = 0;
416 }
417 else
418 c0out << ' ';
419
420 struct timeval NewTime;
421 gettimeofday(&NewTime,0);
422 double Delta = NewTime.tv_sec - StartTime.tv_sec +
423 (NewTime.tv_usec - StartTime.tv_usec)/1000000.0;
424
425 c0out << Contents.Stats.Packages << " files " <<
426 SizeToStr(Contents.Stats.Bytes) << "B " <<
427 TimeToStr((long)Delta) << endl;
428
429 return true;
430}
431 /*}}}*/
432
433// LoadTree - Load a 'tree' section from the Generate Config /*{{{*/
434// ---------------------------------------------------------------------
435/* This populates the PkgList with all the possible permutations of the
436 section/arch lists. */
437void LoadTree(vector<PackageMap> &PkgList,Configuration &Setup)
438{
439 // Load the defaults
440 string DDir = Setup.Find("TreeDefault::Directory",
441 "$(DIST)/$(SECTION)/binary-$(ARCH)/");
442 string DSDir = Setup.Find("TreeDefault::SrcDirectory",
443 "$(DIST)/$(SECTION)/source/");
444 string DPkg = Setup.Find("TreeDefault::Packages",
445 "$(DIST)/$(SECTION)/binary-$(ARCH)/Packages");
446 string DIPrfx = Setup.Find("TreeDefault::InternalPrefix",
447 "$(DIST)/$(SECTION)/");
448 string DContents = Setup.Find("TreeDefault::Contents",
449 "$(DIST)/Contents-$(ARCH)");
450 string DContentsH = Setup.Find("TreeDefault::Contents::Header","");
451 string DBCache = Setup.Find("TreeDefault::BinCacheDB",
452 "packages-$(ARCH).db");
453 string DSources = Setup.Find("TreeDefault::Sources",
454 "$(DIST)/$(SECTION)/source/Sources");
455 string DFLFile = Setup.Find("TreeDefault::FileList", "");
456 string DSFLFile = Setup.Find("TreeDefault::SourceFileList", "");
457
458 // Process 'tree' type sections
459 const Configuration::Item *Top = Setup.Tree("tree");
460 for (Top = (Top == 0?0:Top->Child); Top != 0;)
461 {
462 Configuration Block(Top);
463 string Dist = Top->Tag;
464
465 // Parse the sections
466 string Tmp = Block.Find("Sections");
467 const char *Sections = Tmp.c_str();
468 string Section;
469 while (ParseQuoteWord(Sections,Section) == true)
470 {
471 string Tmp2 = Block.Find("Architectures");
472 string Arch;
473 const char *Archs = Tmp2.c_str();
474 while (ParseQuoteWord(Archs,Arch) == true)
475 {
476 struct SubstVar Vars[] = {{"$(DIST)",&Dist},
477 {"$(SECTION)",&Section},
478 {"$(ARCH)",&Arch},
479 {}};
480 PackageMap Itm;
481
482 Itm.BinOverride = SubstVar(Block.Find("BinOverride"),Vars);
483 Itm.InternalPrefix = SubstVar(Block.Find("InternalPrefix",DIPrfx.c_str()),Vars);
484
485 if (stringcasecmp(Arch,"source") == 0)
486 {
487 Itm.SrcOverride = SubstVar(Block.Find("SrcOverride"),Vars);
488 Itm.BaseDir = SubstVar(Block.Find("SrcDirectory",DSDir.c_str()),Vars);
489 Itm.SrcFile = SubstVar(Block.Find("Sources",DSources.c_str()),Vars);
490 Itm.Tag = SubstVar("$(DIST)/$(SECTION)/source",Vars);
491 Itm.FLFile = SubstVar(Block.Find("SourceFileList",DSFLFile.c_str()),Vars);
492 Itm.SrcExtraOverride = SubstVar(Block.Find("SrcExtraOverride"),Vars);
493 }
494 else
495 {
496 Itm.BinCacheDB = SubstVar(Block.Find("BinCacheDB",DBCache.c_str()),Vars);
497 Itm.BaseDir = SubstVar(Block.Find("Directory",DDir.c_str()),Vars);
498 Itm.PkgFile = SubstVar(Block.Find("Packages",DPkg.c_str()),Vars);
499 Itm.Tag = SubstVar("$(DIST)/$(SECTION)/$(ARCH)",Vars);
500 Itm.Arch = Arch;
501 Itm.Contents = SubstVar(Block.Find("Contents",DContents.c_str()),Vars);
502 Itm.ContentsHead = SubstVar(Block.Find("Contents::Header",DContentsH.c_str()),Vars);
503 Itm.FLFile = SubstVar(Block.Find("FileList",DFLFile.c_str()),Vars);
504 Itm.ExtraOverride = SubstVar(Block.Find("ExtraOverride"),Vars);
505 }
506
507 Itm.GetGeneral(Setup,Block);
508 PkgList.push_back(Itm);
509 }
510 }
511
512 Top = Top->Next;
513 }
514}
515 /*}}}*/
516// LoadBinDir - Load a 'bindirectory' section from the Generate Config /*{{{*/
517// ---------------------------------------------------------------------
518/* */
519void LoadBinDir(vector<PackageMap> &PkgList,Configuration &Setup)
520{
521 // Process 'bindirectory' type sections
522 const Configuration::Item *Top = Setup.Tree("bindirectory");
523 for (Top = (Top == 0?0:Top->Child); Top != 0;)
524 {
525 Configuration Block(Top);
526
527 PackageMap Itm;
528 Itm.PkgFile = Block.Find("Packages");
529 Itm.SrcFile = Block.Find("Sources");
530 Itm.BinCacheDB = Block.Find("BinCacheDB");
531 Itm.BinOverride = Block.Find("BinOverride");
532 Itm.ExtraOverride = Block.Find("ExtraOverride");
533 Itm.SrcExtraOverride = Block.Find("SrcExtraOverride");
534 Itm.SrcOverride = Block.Find("SrcOverride");
535 Itm.BaseDir = Top->Tag;
536 Itm.FLFile = Block.Find("FileList");
537 Itm.InternalPrefix = Block.Find("InternalPrefix",Top->Tag.c_str());
538 Itm.Contents = Block.Find("Contents");
539 Itm.ContentsHead = Block.Find("Contents::Header");
540
541 Itm.GetGeneral(Setup,Block);
542 PkgList.push_back(Itm);
543
544 Top = Top->Next;
545 }
546}
547 /*}}}*/
548
549// ShowHelp - Show the help text /*{{{*/
550// ---------------------------------------------------------------------
551/* */
552bool ShowHelp(CommandLine &CmdL)
553{
554 ioprintf(cout,_("%s %s for %s %s compiled on %s %s\n"),PACKAGE,VERSION,
555 COMMON_OS,COMMON_CPU,__DATE__,__TIME__);
556 if (_config->FindB("version") == true)
557 return true;
558
559 cout <<
560 _("Usage: apt-ftparchive [options] command\n"
561 "Commands: packages binarypath [overridefile [pathprefix]]\n"
562 " sources srcpath [overridefile [pathprefix]]\n"
563 " contents path\n"
564 " release path\n"
565 " generate config [groups]\n"
566 " clean config\n"
567 "\n"
568 "apt-ftparchive generates index files for Debian archives. It supports\n"
569 "many styles of generation from fully automated to functional replacements\n"
570 "for dpkg-scanpackages and dpkg-scansources\n"
571 "\n"
572 "apt-ftparchive generates Package files from a tree of .debs. The\n"
573 "Package file contains the contents of all the control fields from\n"
574 "each package as well as the MD5 hash and filesize. An override file\n"
575 "is supported to force the value of Priority and Section.\n"
576 "\n"
577 "Similarly apt-ftparchive generates Sources files from a tree of .dscs.\n"
578 "The --source-override option can be used to specify a src override file\n"
579 "\n"
580 "The 'packages' and 'sources' command should be run in the root of the\n"
581 "tree. BinaryPath should point to the base of the recursive search and \n"
582 "override file should contain the override flags. Pathprefix is\n"
583 "appended to the filename fields if present. Example usage from the \n"
584 "Debian archive:\n"
585 " apt-ftparchive packages dists/potato/main/binary-i386/ > \\\n"
586 " dists/potato/main/binary-i386/Packages\n"
587 "\n"
588 "Options:\n"
589 " -h This help text\n"
590 " --md5 Control MD5 generation\n"
591 " -s=? Source override file\n"
592 " -q Quiet\n"
593 " -d=? Select the optional caching database\n"
594 " --no-delink Enable delinking debug mode\n"
595 " --contents Control contents file generation\n"
596 " -c=? Read this configuration file\n"
597 " -o=? Set an arbitrary configuration option") << endl;
598
599 return true;
600}
601 /*}}}*/
602// SimpleGenPackages - Generate a Packages file for a directory tree /*{{{*/
603// ---------------------------------------------------------------------
604/* This emulates dpkg-scanpackages's command line interface. 'mostly' */
605bool SimpleGenPackages(CommandLine &CmdL)
606{
607 if (CmdL.FileSize() < 2)
608 return ShowHelp(CmdL);
609
610 string Override;
611 if (CmdL.FileSize() >= 3)
612 Override = CmdL.FileList[2];
613
614 // Create a package writer object.
615 PackagesWriter Packages(_config->Find("APT::FTPArchive::DB"),
616 Override, "");
617 if (_error->PendingError() == true)
618 return false;
619
620 if (CmdL.FileSize() >= 4)
621 Packages.PathPrefix = CmdL.FileList[3];
622
623 // Do recursive directory searching
624 if (Packages.RecursiveScan(CmdL.FileList[1]) == false)
625 return false;
626
627 return true;
628}
629 /*}}}*/
630// SimpleGenContents - Generate a Contents listing /*{{{*/
631// ---------------------------------------------------------------------
632/* */
633bool SimpleGenContents(CommandLine &CmdL)
634{
635 if (CmdL.FileSize() < 2)
636 return ShowHelp(CmdL);
637
638 // Create a package writer object.
639 ContentsWriter Contents(_config->Find("APT::FTPArchive::DB"));
640 if (_error->PendingError() == true)
641 return false;
642
643 // Do recursive directory searching
644 if (Contents.RecursiveScan(CmdL.FileList[1]) == false)
645 return false;
646
647 Contents.Finish();
648
649 return true;
650}
651 /*}}}*/
652// SimpleGenSources - Generate a Sources file for a directory tree /*{{{*/
653// ---------------------------------------------------------------------
654/* This emulates dpkg-scanpackages's command line interface. 'mostly' */
655bool SimpleGenSources(CommandLine &CmdL)
656{
657 if (CmdL.FileSize() < 2)
658 return ShowHelp(CmdL);
659
660 string Override;
661 if (CmdL.FileSize() >= 3)
662 Override = CmdL.FileList[2];
663
664 string SOverride;
665 if (Override.empty() == false)
666 SOverride = Override + ".src";
667
668 SOverride = _config->Find("APT::FTPArchive::SourceOverride",
669 SOverride.c_str());
670
671 // Create a package writer object.
672 SourcesWriter Sources(Override,SOverride);
673 if (_error->PendingError() == true)
674 return false;
675
676 if (CmdL.FileSize() >= 4)
677 Sources.PathPrefix = CmdL.FileList[3];
678
679 // Do recursive directory searching
680 if (Sources.RecursiveScan(CmdL.FileList[1]) == false)
681 return false;
682
683 return true;
684}
685 /*}}}*/
686// SimpleGenRelease - Generate a Release file for a directory tree /*{{{*/
687// ---------------------------------------------------------------------
688bool SimpleGenRelease(CommandLine &CmdL)
689{
690 if (CmdL.FileSize() < 2)
691 return ShowHelp(CmdL);
692
693 string Dir = CmdL.FileList[1];
694
695 ReleaseWriter Release("");
696 Release.DirStrip = Dir;
697
698 if (_error->PendingError() == true)
699 return false;
700
701 if (Release.RecursiveScan(Dir) == false)
702 return false;
703
704 Release.Finish();
705
706 return true;
707}
708
709 /*}}}*/
710// Generate - Full generate, using a config file /*{{{*/
711// ---------------------------------------------------------------------
712/* */
713bool Generate(CommandLine &CmdL)
714{
715 struct CacheDB::Stats SrcStats;
716 if (CmdL.FileSize() < 2)
717 return ShowHelp(CmdL);
718
719 struct timeval StartTime;
720 gettimeofday(&StartTime,0);
721 struct CacheDB::Stats Stats;
722
723 // Read the configuration file.
724 Configuration Setup;
725 if (ReadConfigFile(Setup,CmdL.FileList[1],true) == false)
726 return false;
727
728 vector<PackageMap> PkgList;
729 LoadTree(PkgList,Setup);
730 LoadBinDir(PkgList,Setup);
731
732 // Sort by cache DB to improve IO locality.
733 stable_sort(PkgList.begin(),PkgList.end(),PackageMap::DBCompare());
734
735 // Generate packages
736 if (CmdL.FileSize() <= 2)
737 {
738 for (vector<PackageMap>::iterator I = PkgList.begin(); I != PkgList.end(); I++)
739 if (I->GenPackages(Setup,Stats) == false)
740 _error->DumpErrors();
741 for (vector<PackageMap>::iterator I = PkgList.begin(); I != PkgList.end(); I++)
742 if (I->GenSources(Setup,SrcStats) == false)
743 _error->DumpErrors();
744 }
745 else
746 {
747 // Make a choice list out of the package list..
748 RxChoiceList *List = new RxChoiceList[2*PkgList.size()+1];
749 RxChoiceList *End = List;
750 for (vector<PackageMap>::iterator I = PkgList.begin(); I != PkgList.end(); I++)
751 {
752 End->UserData = &(*I);
753 End->Str = I->BaseDir.c_str();
754 End++;
755
756 End->UserData = &(*I);
757 End->Str = I->Tag.c_str();
758 End++;
759 }
760 End->Str = 0;
761
762 // Regex it
763 if (RegexChoice(List,CmdL.FileList + 2,CmdL.FileList + CmdL.FileSize()) == 0)
764 {
765 delete [] List;
766 return _error->Error(_("No selections matched"));
767 }
768 _error->DumpErrors();
769
770 // Do the generation for Packages
771 for (End = List; End->Str != 0; End++)
772 {
773 if (End->Hit == false)
774 continue;
775
776 PackageMap *I = (PackageMap *)End->UserData;
777 if (I->PkgDone == true)
778 continue;
779 if (I->GenPackages(Setup,Stats) == false)
780 _error->DumpErrors();
781 }
782
783 // Do the generation for Sources
784 for (End = List; End->Str != 0; End++)
785 {
786 if (End->Hit == false)
787 continue;
788
789 PackageMap *I = (PackageMap *)End->UserData;
790 if (I->SrcDone == true)
791 continue;
792 if (I->GenSources(Setup,SrcStats) == false)
793 _error->DumpErrors();
794 }
795
796 delete [] List;
797 }
798
799 if (_config->FindB("APT::FTPArchive::Contents",true) == false)
800 return true;
801
802 c1out << "Done Packages, Starting contents." << endl;
803
804 // Sort the contents file list by date
805 string ArchiveDir = Setup.FindDir("Dir::ArchiveDir");
806 for (vector<PackageMap>::iterator I = PkgList.begin(); I != PkgList.end(); I++)
807 {
808 struct stat A;
809 if (MultiCompress::GetStat(flCombine(ArchiveDir,I->Contents),
810 I->CntCompress,A) == false)
811 time(&I->ContentsMTime);
812 else
813 I->ContentsMTime = A.st_mtime;
814 }
815 stable_sort(PkgList.begin(),PkgList.end(),PackageMap::ContentsCompare());
816
817 /* Now for Contents.. The process here is to do a make-like dependency
818 check. Each contents file is verified to be newer than the package files
819 that describe the debs it indexes. Since the package files contain
820 hashes of the .debs this means they have not changed either so the
821 contents must be up to date. */
822 unsigned long MaxContentsChange = Setup.FindI("Default::MaxContentsChange",UINT_MAX)*1024;
823 for (vector<PackageMap>::iterator I = PkgList.begin(); I != PkgList.end(); I++)
824 {
825 // This record is not relevent
826 if (I->ContentsDone == true ||
827 I->Contents.empty() == true)
828 continue;
829
830 // Do not do everything if the user specified sections.
831 if (CmdL.FileSize() > 2 && I->PkgDone == false)
832 continue;
833
834 struct stat A,B;
835 if (MultiCompress::GetStat(flCombine(ArchiveDir,I->Contents),I->CntCompress,A) == true)
836 {
837 if (MultiCompress::GetStat(flCombine(ArchiveDir,I->PkgFile),I->PkgCompress,B) == false)
838 {
839 _error->Warning(_("Some files are missing in the package file group `%s'"),I->PkgFile.c_str());
840 continue;
841 }
842
843 if (A.st_mtime > B.st_mtime)
844 continue;
845 }
846
847 if (I->GenContents(Setup,PkgList.begin(),PkgList.end(),
848 MaxContentsChange) == false)
849 _error->DumpErrors();
850
851 // Hit the limit?
852 if (MaxContentsChange == 0)
853 {
854 c1out << "Hit contents update byte limit" << endl;
855 break;
856 }
857 }
858
859 struct timeval NewTime;
860 gettimeofday(&NewTime,0);
861 double Delta = NewTime.tv_sec - StartTime.tv_sec +
862 (NewTime.tv_usec - StartTime.tv_usec)/1000000.0;
863 c1out << "Done. " << SizeToStr(Stats.Bytes) << "B in " << Stats.Packages
864 << " archives. Took " << TimeToStr((long)Delta) << endl;
865
866 return true;
867}
868 /*}}}*/
869// Clean - Clean out the databases /*{{{*/
870// ---------------------------------------------------------------------
871/* */
872bool Clean(CommandLine &CmdL)
873{
874 if (CmdL.FileSize() != 2)
875 return ShowHelp(CmdL);
876
877 // Read the configuration file.
878 Configuration Setup;
879 if (ReadConfigFile(Setup,CmdL.FileList[1],true) == false)
880 return false;
881
882 vector<PackageMap> PkgList;
883 LoadTree(PkgList,Setup);
884 LoadBinDir(PkgList,Setup);
885
886 // Sort by cache DB to improve IO locality.
887 stable_sort(PkgList.begin(),PkgList.end(),PackageMap::DBCompare());
888
889 string CacheDir = Setup.FindDir("Dir::CacheDir");
890
891 for (vector<PackageMap>::iterator I = PkgList.begin(); I != PkgList.end(); )
892 {
893 c0out << I->BinCacheDB << endl;
894 CacheDB DB(flCombine(CacheDir,I->BinCacheDB));
895 if (DB.Clean() == false)
896 _error->DumpErrors();
897
898 string CacheDB = I->BinCacheDB;
899 for (; I != PkgList.end() && I->BinCacheDB == CacheDB; I++);
900 }
901
902 return true;
903}
904 /*}}}*/
905
906int main(int argc, const char *argv[])
907{
854e5ff1
JF
908 struct nlist nl[2];
909 memset(nl, 0, sizeof(nl));
3d6f1076 910 nl[0].n_un.n_name = (char *) "_useMDNSResponder";
854e5ff1
JF
911 nlist("/usr/lib/libc.dylib", nl);
912 if (nl[0].n_type != N_UNDF)
913 *(int *) nl[0].n_value = 0;
914
da6ee469
JF
915 CommandLine::Args Args[] = {
916 {'h',"help","help",0},
917 {0,"md5","APT::FTPArchive::MD5",0},
918 {'v',"version","version",0},
919 {'d',"db","APT::FTPArchive::DB",CommandLine::HasArg},
920 {'s',"source-override","APT::FTPArchive::SourceOverride",CommandLine::HasArg},
921 {'q',"quiet","quiet",CommandLine::IntLevel},
922 {'q',"silent","quiet",CommandLine::IntLevel},
923 {0,"delink","APT::FTPArchive::DeLinkAct",0},
924 {0,"readonly","APT::FTPArchive::ReadOnlyDB",0},
925 {0,"contents","APT::FTPArchive::Contents",0},
926 {'c',"config-file",0,CommandLine::ConfigFile},
927 {'o',"option",0,CommandLine::ArbItem},
928 {0,0,0,0}};
929 CommandLine::Dispatch Cmds[] = {{"packages",&SimpleGenPackages},
930 {"contents",&SimpleGenContents},
931 {"sources",&SimpleGenSources},
932 {"release",&SimpleGenRelease},
933 {"generate",&Generate},
934 {"clean",&Clean},
935 {"help",&ShowHelp},
936 {0,0}};
937
938 // Parse the command line and initialize the package library
939 CommandLine CmdL(Args,_config);
940 if (CmdL.Parse(argc,argv) == false)
941 {
942 _error->DumpErrors();
943 return 100;
944 }
945
946 // See if the help should be shown
947 if (_config->FindB("help") == true ||
948 _config->FindB("version") == true ||
949 CmdL.FileSize() == 0)
950 {
951 ShowHelp(CmdL);
952 return 0;
953 }
954
955 // Setup the output streams
956 c0out.rdbuf(clog.rdbuf());
957 c1out.rdbuf(clog.rdbuf());
958 c2out.rdbuf(clog.rdbuf());
959 Quiet = _config->FindI("quiet",0);
960 if (Quiet > 0)
961 c0out.rdbuf(devnull.rdbuf());
962 if (Quiet > 1)
963 c1out.rdbuf(devnull.rdbuf());
964
965 // Match the operation
966 CmdL.DispatchArg(Cmds);
967
968 if (_error->empty() == false)
969 {
970 bool Errors = _error->PendingError();
971 _error->DumpErrors();
972 return Errors == true?100:0;
973 }
974 return 0;
975}