merged from lp:~donkult/apt/sid
[ntk/apt.git] / ftparchive / writer.cc
CommitLineData
b2e465d6
AL
1// -*- mode: cpp; mode: fold -*-
2// Description /*{{{*/
5200ec6f 3// $Id: writer.cc,v 1.14 2004/03/24 01:40:43 mdz Exp $
b2e465d6
AL
4/* ######################################################################
5
6 Writer
7
8 The file writer classes. These write various types of output, sources,
9 packages and contents.
10
11 ##################################################################### */
12 /*}}}*/
13// Include Files /*{{{*/
ea542140
DK
14#include <config.h>
15
b2e465d6
AL
16#include <apt-pkg/strutl.h>
17#include <apt-pkg/error.h>
18#include <apt-pkg/configuration.h>
03bef784 19#include <apt-pkg/aptconfiguration.h>
b2e465d6 20#include <apt-pkg/md5.h>
8c4e1f97 21#include <apt-pkg/hashes.h>
b2e465d6
AL
22#include <apt-pkg/deblistparser.h>
23
24#include <sys/types.h>
25#include <unistd.h>
98953965 26#include <ctime>
b2e465d6 27#include <ftw.h>
98953965 28#include <fnmatch.h>
8c58f506 29#include <iostream>
bf99a6d3 30#include <sstream>
4f333a8b 31#include <memory>
ea542140
DK
32
33#include "writer.h"
b2e465d6
AL
34#include "cachedb.h"
35#include "apt-ftparchive.h"
36#include "multicompress.h"
ea542140
DK
37
38#include <apti18n.h>
b2e465d6 39 /*}}}*/
8c58f506 40using namespace std;
b2e465d6
AL
41FTWScanner *FTWScanner::Owner;
42
64177f17
AL
43// SetTFRewriteData - Helper for setting rewrite lists /*{{{*/
44// ---------------------------------------------------------------------
45/* */
46inline void SetTFRewriteData(struct TFRewriteData &tfrd,
47 const char *tag,
48 const char *rewrite,
49 const char *newtag = 0)
50{
51 tfrd.Tag = tag;
52 tfrd.Rewrite = rewrite;
53 tfrd.NewTag = newtag;
54}
55 /*}}}*/
56
b2e465d6
AL
57// FTWScanner::FTWScanner - Constructor /*{{{*/
58// ---------------------------------------------------------------------
59/* */
31981076 60FTWScanner::FTWScanner(string const &Arch): Arch(Arch)
b2e465d6
AL
61{
62 ErrorPrinted = false;
63 NoLinkAct = !_config->FindB("APT::FTPArchive::DeLinkAct",true);
3c54407f
DK
64
65 DoMD5 = _config->FindB("APT::FTPArchive::MD5",true);
66 DoSHA1 = _config->FindB("APT::FTPArchive::SHA1",true);
67 DoSHA256 = _config->FindB("APT::FTPArchive::SHA256",true);
9abccf4a 68 DoSHA512 = _config->FindB("APT::FTPArchive::SHA512",true);
b2e465d6
AL
69}
70 /*}}}*/
71// FTWScanner::Scanner - FTW Scanner /*{{{*/
72// ---------------------------------------------------------------------
73/* This is the FTW scanner, it processes each directory element in the
74 directory tree. */
cde41ae8 75int FTWScanner::ScannerFTW(const char *File,const struct stat *sb,int Flag)
b2e465d6
AL
76{
77 if (Flag == FTW_DNR)
78 {
79 Owner->NewLine(1);
dc738e7a 80 ioprintf(c1out, _("W: Unable to read directory %s\n"), File);
b2e465d6
AL
81 }
82 if (Flag == FTW_NS)
83 {
84 Owner->NewLine(1);
dc738e7a 85 ioprintf(c1out, _("W: Unable to stat %s\n"), File);
b2e465d6
AL
86 }
87 if (Flag != FTW_F)
88 return 0;
89
cde41ae8
MV
90 return ScannerFile(File, true);
91}
92 /*}}}*/
93// FTWScanner::ScannerFile - File Scanner /*{{{*/
94// ---------------------------------------------------------------------
95/* */
9209ec47 96int FTWScanner::ScannerFile(const char *File, bool const &ReadLink)
cde41ae8 97{
98953965 98 const char *LastComponent = strrchr(File, '/');
ab3846c0
MV
99 char *RealPath = NULL;
100
98953965
AL
101 if (LastComponent == NULL)
102 LastComponent = File;
103 else
104 LastComponent++;
105
9209ec47 106 vector<string>::const_iterator I;
98953965
AL
107 for(I = Owner->Patterns.begin(); I != Owner->Patterns.end(); ++I)
108 {
109 if (fnmatch((*I).c_str(), LastComponent, 0) == 0)
110 break;
111 }
112 if (I == Owner->Patterns.end())
b2e465d6
AL
113 return 0;
114
115 /* Process it. If the file is a link then resolve it into an absolute
116 name.. This works best if the directory components the scanner are
117 given are not links themselves. */
118 char Jnk[2];
119 Owner->OriginalPath = File;
ab3846c0 120 if (ReadLink &&
cde41ae8 121 readlink(File,Jnk,sizeof(Jnk)) != -1 &&
ab3846c0
MV
122 (RealPath = realpath(File,NULL)) != 0)
123 {
124 Owner->DoPackage(RealPath);
125 free(RealPath);
126 }
b2e465d6
AL
127 else
128 Owner->DoPackage(File);
129
130 if (_error->empty() == false)
131 {
132 // Print any errors or warnings found
133 string Err;
134 bool SeenPath = false;
135 while (_error->empty() == false)
136 {
137 Owner->NewLine(1);
138
9209ec47 139 bool const Type = _error->PopMessage(Err);
b2e465d6 140 if (Type == true)
dc738e7a 141 cerr << _("E: ") << Err << endl;
b2e465d6 142 else
dc738e7a 143 cerr << _("W: ") << Err << endl;
b2e465d6
AL
144
145 if (Err.find(File) != string::npos)
146 SeenPath = true;
147 }
148
149 if (SeenPath == false)
dc738e7a 150 cerr << _("E: Errors apply to file ") << "'" << File << "'" << endl;
b2e465d6
AL
151 return 0;
152 }
153
154 return 0;
155}
156 /*}}}*/
157// FTWScanner::RecursiveScan - Just scan a directory tree /*{{{*/
158// ---------------------------------------------------------------------
159/* */
9209ec47 160bool FTWScanner::RecursiveScan(string const &Dir)
b2e465d6 161{
ab3846c0 162 char *RealPath = NULL;
b2e465d6
AL
163 /* If noprefix is set then jam the scan root in, so we don't generate
164 link followed paths out of control */
165 if (InternalPrefix.empty() == true)
166 {
ab3846c0 167 if ((RealPath = realpath(Dir.c_str(),NULL)) == 0)
dc738e7a 168 return _error->Errno("realpath",_("Failed to resolve %s"),Dir.c_str());
ab3846c0
MV
169 InternalPrefix = RealPath;
170 free(RealPath);
b2e465d6
AL
171 }
172
173 // Do recursive directory searching
174 Owner = this;
9209ec47 175 int const Res = ftw(Dir.c_str(),ScannerFTW,30);
b2e465d6
AL
176
177 // Error treewalking?
178 if (Res != 0)
179 {
180 if (_error->PendingError() == false)
dc738e7a 181 _error->Errno("ftw",_("Tree walking failed"));
b2e465d6
AL
182 return false;
183 }
184
185 return true;
186}
187 /*}}}*/
188// FTWScanner::LoadFileList - Load the file list from a file /*{{{*/
189// ---------------------------------------------------------------------
190/* This is an alternative to using FTW to locate files, it reads the list
191 of files from another file. */
9209ec47 192bool FTWScanner::LoadFileList(string const &Dir, string const &File)
b2e465d6 193{
ab3846c0 194 char *RealPath = NULL;
b2e465d6
AL
195 /* If noprefix is set then jam the scan root in, so we don't generate
196 link followed paths out of control */
197 if (InternalPrefix.empty() == true)
198 {
ab3846c0 199 if ((RealPath = realpath(Dir.c_str(),NULL)) == 0)
dc738e7a 200 return _error->Errno("realpath",_("Failed to resolve %s"),Dir.c_str());
b2e465d6 201 InternalPrefix = RealPath;
ab3846c0 202 free(RealPath);
b2e465d6
AL
203 }
204
205 Owner = this;
206 FILE *List = fopen(File.c_str(),"r");
207 if (List == 0)
dc738e7a 208 return _error->Errno("fopen",_("Failed to open %s"),File.c_str());
b2e465d6
AL
209
210 /* We are a tad tricky here.. We prefix the buffer with the directory
211 name, that way if we need a full path with just use line.. Sneaky and
212 fully evil. */
213 char Line[1000];
214 char *FileStart;
215 if (Dir.empty() == true || Dir.end()[-1] != '/')
216 FileStart = Line + snprintf(Line,sizeof(Line),"%s/",Dir.c_str());
217 else
218 FileStart = Line + snprintf(Line,sizeof(Line),"%s",Dir.c_str());
219 while (fgets(FileStart,sizeof(Line) - (FileStart - Line),List) != 0)
220 {
221 char *FileName = _strstrip(FileStart);
222 if (FileName[0] == 0)
223 continue;
224
225 if (FileName[0] != '/')
226 {
227 if (FileName != FileStart)
228 memmove(FileStart,FileName,strlen(FileStart));
229 FileName = Line;
230 }
231
cde41ae8 232#if 0
b2e465d6
AL
233 struct stat St;
234 int Flag = FTW_F;
235 if (stat(FileName,&St) != 0)
236 Flag = FTW_NS;
cde41ae8 237#endif
b2e465d6 238
cde41ae8 239 if (ScannerFile(FileName, false) != 0)
b2e465d6
AL
240 break;
241 }
242
243 fclose(List);
244 return true;
245}
246 /*}}}*/
247// FTWScanner::Delink - Delink symlinks /*{{{*/
248// ---------------------------------------------------------------------
249/* */
250bool FTWScanner::Delink(string &FileName,const char *OriginalPath,
650faab0
DK
251 unsigned long long &DeLinkBytes,
252 unsigned long long const &FileSize)
b2e465d6
AL
253{
254 // See if this isn't an internaly prefix'd file name.
255 if (InternalPrefix.empty() == false &&
256 InternalPrefix.length() < FileName.length() &&
257 stringcmp(FileName.begin(),FileName.begin() + InternalPrefix.length(),
258 InternalPrefix.begin(),InternalPrefix.end()) != 0)
259 {
260 if (DeLinkLimit != 0 && DeLinkBytes/1024 < DeLinkLimit)
261 {
262 // Tidy up the display
263 if (DeLinkBytes == 0)
264 cout << endl;
265
266 NewLine(1);
dc738e7a 267 ioprintf(c1out, _(" DeLink %s [%s]\n"), (OriginalPath + InternalPrefix.length()),
cde41ae8 268 SizeToStr(FileSize).c_str());
dc738e7a 269 c1out << flush;
b2e465d6
AL
270
271 if (NoLinkAct == false)
272 {
273 char OldLink[400];
274 if (readlink(OriginalPath,OldLink,sizeof(OldLink)) == -1)
dc738e7a 275 _error->Errno("readlink",_("Failed to readlink %s"),OriginalPath);
b2e465d6
AL
276 else
277 {
278 if (unlink(OriginalPath) != 0)
dc738e7a 279 _error->Errno("unlink",_("Failed to unlink %s"),OriginalPath);
b2e465d6
AL
280 else
281 {
282 if (link(FileName.c_str(),OriginalPath) != 0)
283 {
284 // Panic! Restore the symlink
285 symlink(OldLink,OriginalPath);
dc738e7a 286 return _error->Errno("link",_("*** Failed to link %s to %s"),
b2e465d6
AL
287 FileName.c_str(),
288 OriginalPath);
289 }
290 }
291 }
292 }
293
cde41ae8 294 DeLinkBytes += FileSize;
b2e465d6 295 if (DeLinkBytes/1024 >= DeLinkLimit)
dc738e7a 296 ioprintf(c1out, _(" DeLink limit of %sB hit.\n"), SizeToStr(DeLinkBytes).c_str());
b2e465d6
AL
297 }
298
299 FileName = OriginalPath;
300 }
301
302 return true;
303}
304 /*}}}*/
b2e465d6
AL
305
306// PackagesWriter::PackagesWriter - Constructor /*{{{*/
307// ---------------------------------------------------------------------
308/* */
9209ec47 309PackagesWriter::PackagesWriter(string const &DB,string const &Overrides,string const &ExtOverrides,
31981076 310 string const &Arch) :
66905344 311 FTWScanner(Arch), Db(DB), Stats(Db.Stats), TransWriter(NULL)
b2e465d6
AL
312{
313 Output = stdout;
31981076 314 SetExts(".deb .udeb");
b2e465d6 315 DeLinkLimit = 0;
3cb3fe76 316
b2e465d6 317 // Process the command line options
3c54407f
DK
318 DoMD5 = _config->FindB("APT::FTPArchive::Packages::MD5",DoMD5);
319 DoSHA1 = _config->FindB("APT::FTPArchive::Packages::SHA1",DoSHA1);
320 DoSHA256 = _config->FindB("APT::FTPArchive::Packages::SHA256",DoSHA256);
8e5d47a3 321 DoSHA512 = _config->FindB("APT::FTPArchive::Packages::SHA512",DoSHA512);
ff574e76 322 DoAlwaysStat = _config->FindB("APT::FTPArchive::AlwaysStat", false);
b2e465d6
AL
323 DoContents = _config->FindB("APT::FTPArchive::Contents",true);
324 NoOverride = _config->FindB("APT::FTPArchive::NoOverrideMsg",false);
9c24493f 325 LongDescription = _config->FindB("APT::FTPArchive::LongDescription",true);
b2e465d6
AL
326
327 if (Db.Loaded() == false)
328 DoContents = false;
66905344 329
b2e465d6
AL
330 // Read the override file
331 if (Overrides.empty() == false && Over.ReadOverride(Overrides) == false)
332 return;
333 else
334 NoOverride = true;
64177f17
AL
335
336 if (ExtOverrides.empty() == false)
337 Over.ReadExtraOverride(ExtOverrides);
338
b2e465d6
AL
339 _error->DumpErrors();
340}
98953965
AL
341 /*}}}*/
342// FTWScanner::SetExts - Set extensions to support /*{{{*/
343// ---------------------------------------------------------------------
344/* */
9209ec47 345bool FTWScanner::SetExts(string const &Vals)
98953965
AL
346{
347 ClearPatterns();
348 string::size_type Start = 0;
d6689735 349 while (Start <= Vals.length()-1)
98953965 350 {
31981076
DK
351 string::size_type const Space = Vals.find(' ',Start);
352 string::size_type const Length = ((Space == string::npos) ? Vals.length() : Space) - Start;
353 if ( Arch.empty() == false )
98953965 354 {
31981076
DK
355 AddPattern(string("*_") + Arch + Vals.substr(Start, Length));
356 AddPattern(string("*_all") + Vals.substr(Start, Length));
98953965 357 }
d6689735 358 else
31981076
DK
359 AddPattern(string("*") + Vals.substr(Start, Length));
360
d6689735 361 Start += Length + 1;
98953965
AL
362 }
363
364 return true;
365}
366
b2e465d6
AL
367 /*}}}*/
368// PackagesWriter::DoPackage - Process a single package /*{{{*/
369// ---------------------------------------------------------------------
370/* This method takes a package and gets its control information and
cde41ae8
MV
371 MD5, SHA1 and SHA256 then writes out a control record with the proper fields
372 rewritten and the path/size/hash appended. */
b2e465d6
AL
373bool PackagesWriter::DoPackage(string FileName)
374{
b2e465d6 375 // Pull all the data we need form the DB
9a961efc 376 if (Db.GetFileInfo(FileName, true, DoContents, true, DoMD5, DoSHA1, DoSHA256, DoSHA512, DoAlwaysStat)
cde41ae8
MV
377 == false)
378 {
b2e465d6 379 return false;
cde41ae8 380 }
b2e465d6 381
650faab0 382 unsigned long long FileSize = Db.GetFileSize();
cde41ae8 383 if (Delink(FileName,OriginalPath,Stats.DeLinkBytes,FileSize) == false)
b2e465d6
AL
384 return false;
385
386 // Lookup the overide information
387 pkgTagSection &Tags = Db.Control.Section;
388 string Package = Tags.FindS("Package");
0b41e0e7
MV
389 string Architecture;
390 // if we generate a Packages file for a given arch, we use it to
391 // look for overrides. if we run in "simple" mode without the
392 // "Architecures" variable in the config we use the architecure value
393 // from the deb file
394 if(Arch != "")
395 Architecture = Arch;
396 else
397 Architecture = Tags.FindS("Architecture");
398 auto_ptr<Override::Item> OverItem(Over.GetItem(Package,Architecture));
b2e465d6
AL
399
400 if (Package.empty() == true)
dc738e7a 401 return _error->Error(_("Archive had no package field"));
0b41e0e7 402
b2e465d6 403 // If we need to do any rewriting of the header do it now..
0b41e0e7 404 if (OverItem.get() == 0)
b2e465d6
AL
405 {
406 if (NoOverride == false)
407 {
408 NewLine(1);
dc738e7a 409 ioprintf(c1out, _(" %s has no override entry\n"), Package.c_str());
b2e465d6
AL
410 }
411
0b41e0e7
MV
412 OverItem = auto_ptr<Override::Item>(new Override::Item);
413 OverItem->FieldOverride["Section"] = Tags.FindS("Section");
414 OverItem->Priority = Tags.FindS("Priority");
b2e465d6
AL
415 }
416
417 char Size[40];
650faab0 418 sprintf(Size,"%llu", (unsigned long long) FileSize);
b2e465d6
AL
419
420 // Strip the DirStrip prefix from the FileName and add the PathPrefix
421 string NewFileName;
422 if (DirStrip.empty() == false &&
423 FileName.length() > DirStrip.length() &&
424 stringcmp(FileName.begin(),FileName.begin() + DirStrip.length(),
425 DirStrip.begin(),DirStrip.end()) == 0)
426 NewFileName = string(FileName.begin() + DirStrip.length(),FileName.end());
427 else
428 NewFileName = FileName;
429 if (PathPrefix.empty() == false)
430 NewFileName = flCombine(PathPrefix,NewFileName);
9c24493f
DK
431
432 /* Configuration says we don't want to include the long Description
433 in the package file - instead we want to ship a separated file */
434 string desc;
435 if (LongDescription == false) {
436 desc = Tags.FindS("Description").append("\n");
437 OverItem->FieldOverride["Description"] = desc.substr(0, desc.find('\n')).c_str();
438 }
439
b2e465d6 440 // This lists all the changes to the fields we are going to make.
64177f17 441 // (7 hardcoded + maintainer + suggests + end marker)
9c24493f 442 TFRewriteData Changes[6+2+OverItem->FieldOverride.size()+1+1];
64177f17 443
b2e465d6 444 unsigned int End = 0;
64177f17 445 SetTFRewriteData(Changes[End++], "Size", Size);
3c54407f
DK
446 if (DoMD5 == true)
447 SetTFRewriteData(Changes[End++], "MD5sum", Db.MD5Res.c_str());
448 if (DoSHA1 == true)
449 SetTFRewriteData(Changes[End++], "SHA1", Db.SHA1Res.c_str());
450 if (DoSHA256 == true)
451 SetTFRewriteData(Changes[End++], "SHA256", Db.SHA256Res.c_str());
12cd178d
MV
452 if (DoSHA512 == true)
453 SetTFRewriteData(Changes[End++], "SHA512", Db.SHA512Res.c_str());
64177f17
AL
454 SetTFRewriteData(Changes[End++], "Filename", NewFileName.c_str());
455 SetTFRewriteData(Changes[End++], "Priority", OverItem->Priority.c_str());
456 SetTFRewriteData(Changes[End++], "Status", 0);
457 SetTFRewriteData(Changes[End++], "Optional", 0);
458
9c24493f
DK
459 string DescriptionMd5;
460 if (LongDescription == false) {
461 MD5Summation descmd5;
462 descmd5.Add(desc.c_str());
463 DescriptionMd5 = descmd5.Result().Value();
464 SetTFRewriteData(Changes[End++], "Description-md5", DescriptionMd5.c_str());
66905344
DK
465 if (TransWriter != NULL)
466 TransWriter->DoPackage(Package, desc, DescriptionMd5);
9c24493f
DK
467 }
468
b2e465d6
AL
469 // Rewrite the maintainer field if necessary
470 bool MaintFailed;
471 string NewMaint = OverItem->SwapMaint(Tags.FindS("Maintainer"),MaintFailed);
472 if (MaintFailed == true)
473 {
474 if (NoOverride == false)
475 {
476 NewLine(1);
dc738e7a
AL
477 ioprintf(c1out, _(" %s maintainer is %s not %s\n"),
478 Package.c_str(), Tags.FindS("Maintainer").c_str(), OverItem->OldMaint.c_str());
b2e465d6
AL
479 }
480 }
481
482 if (NewMaint.empty() == false)
64177f17 483 SetTFRewriteData(Changes[End++], "Maintainer", NewMaint.c_str());
b2e465d6
AL
484
485 /* Get rid of the Optional tag. This is an ugly, ugly, ugly hack that
c6474fb6 486 dpkg-scanpackages does. Well sort of. dpkg-scanpackages just does renaming
b2e465d6
AL
487 but dpkg does this append bit. So we do the append bit, at least that way the
488 status file and package file will remain similar. There are other transforms
489 but optional is the only legacy one still in use for some lazy reason. */
490 string OptionalStr = Tags.FindS("Optional");
491 if (OptionalStr.empty() == false)
492 {
493 if (Tags.FindS("Suggests").empty() == false)
494 OptionalStr = Tags.FindS("Suggests") + ", " + OptionalStr;
64177f17 495 SetTFRewriteData(Changes[End++], "Suggests", OptionalStr.c_str());
b2e465d6 496 }
64177f17 497
9209ec47 498 for (map<string,string>::const_iterator I = OverItem->FieldOverride.begin();
f7f0d6c7 499 I != OverItem->FieldOverride.end(); ++I)
64177f17
AL
500 SetTFRewriteData(Changes[End++],I->first.c_str(),I->second.c_str());
501
502 SetTFRewriteData(Changes[End++], 0, 0);
503
b2e465d6
AL
504 // Rewrite and store the fields.
505 if (TFRewrite(Output,Tags,TFRewritePackageOrder,Changes) == false)
506 return false;
507 fprintf(Output,"\n");
508
509 return Db.Finish();
510}
511 /*}}}*/
512
66905344
DK
513// TranslationWriter::TranslationWriter - Constructor /*{{{*/
514// ---------------------------------------------------------------------
515/* Create a Translation-Master file for this Packages file */
34f1d96c
DK
516TranslationWriter::TranslationWriter(string const &File, string const &TransCompress,
517 mode_t const &Permissions) : Output(NULL),
66905344
DK
518 RefCounter(0)
519{
520 if (File.empty() == true)
521 return;
522
34f1d96c
DK
523 Comp = new MultiCompress(File, TransCompress, Permissions);
524 Output = Comp->Input;
66905344
DK
525}
526 /*}}}*/
527// TranslationWriter::DoPackage - Process a single package /*{{{*/
528// ---------------------------------------------------------------------
529/* Create a Translation-Master file for this Packages file */
530bool TranslationWriter::DoPackage(string const &Pkg, string const &Desc,
531 string const &MD5)
532{
533 if (Output == NULL)
534 return true;
535
536 // Different archs can include different versions and therefore
537 // different descriptions - so we need to check for both name and md5.
538 string const Record = Pkg + ":" + MD5;
539
540 if (Included.find(Record) != Included.end())
541 return true;
542
543 fprintf(Output, "Package: %s\nDescription-md5: %s\nDescription-en: %s\n",
544 Pkg.c_str(), MD5.c_str(), Desc.c_str());
545
546 Included.insert(Record);
547 return true;
548}
549 /*}}}*/
550// TranslationWriter::~TranslationWriter - Destructor /*{{{*/
551// ---------------------------------------------------------------------
552/* */
553TranslationWriter::~TranslationWriter()
554{
34f1d96c
DK
555 if (Comp == NULL)
556 return;
557
558 delete Comp;
66905344
DK
559}
560 /*}}}*/
561
b2e465d6
AL
562// SourcesWriter::SourcesWriter - Constructor /*{{{*/
563// ---------------------------------------------------------------------
564/* */
9209ec47
DK
565SourcesWriter::SourcesWriter(string const &BOverrides,string const &SOverrides,
566 string const &ExtOverrides)
b2e465d6
AL
567{
568 Output = stdout;
98953965 569 AddPattern("*.dsc");
b2e465d6
AL
570 DeLinkLimit = 0;
571 Buffer = 0;
572 BufSize = 0;
573
574 // Process the command line options
3c54407f
DK
575 DoMD5 = _config->FindB("APT::FTPArchive::Sources::MD5",DoMD5);
576 DoSHA1 = _config->FindB("APT::FTPArchive::Sources::SHA1",DoSHA1);
577 DoSHA256 = _config->FindB("APT::FTPArchive::Sources::SHA256",DoSHA256);
b2e465d6
AL
578 NoOverride = _config->FindB("APT::FTPArchive::NoOverrideMsg",false);
579
580 // Read the override file
581 if (BOverrides.empty() == false && BOver.ReadOverride(BOverrides) == false)
582 return;
583 else
584 NoOverride = true;
64177f17 585
cde41ae8
MV
586 // WTF?? The logic above: if we can't read binary overrides, don't even try
587 // reading source overrides. if we can read binary overrides, then say there
588 // are no overrides. THIS MAKES NO SENSE! -- ajt@d.o, 2006/02/28
589
64177f17
AL
590 if (ExtOverrides.empty() == false)
591 SOver.ReadExtraOverride(ExtOverrides);
b2e465d6 592
64177f17
AL
593 if (SOverrides.empty() == false && FileExists(SOverrides) == true)
594 SOver.ReadOverride(SOverrides,true);
b2e465d6
AL
595}
596 /*}}}*/
597// SourcesWriter::DoPackage - Process a single package /*{{{*/
598// ---------------------------------------------------------------------
599/* */
600bool SourcesWriter::DoPackage(string FileName)
601{
602 // Open the archive
603 FileFd F(FileName,FileFd::ReadOnly);
604 if (_error->PendingError() == true)
605 return false;
606
607 // Stat the file for later
608 struct stat St;
609 if (fstat(F.Fd(),&St) != 0)
610 return _error->Errno("fstat","Failed to stat %s",FileName.c_str());
611
612 if (St.st_size > 128*1024)
613 return _error->Error("DSC file '%s' is too large!",FileName.c_str());
614
650faab0 615 if (BufSize < (unsigned long long)St.st_size+1)
b2e465d6
AL
616 {
617 BufSize = St.st_size+1;
618 Buffer = (char *)realloc(Buffer,St.st_size+1);
619 }
620
621 if (F.Read(Buffer,St.st_size) == false)
622 return false;
623
624 // Hash the file
625 char *Start = Buffer;
626 char *BlkEnd = Buffer + St.st_size;
f99da908 627
3c54407f 628 MD5Summation MD5;
f99da908
DK
629 SHA1Summation SHA1;
630 SHA256Summation SHA256;
12cd178d 631 SHA256Summation SHA512;
3c54407f
DK
632
633 if (DoMD5 == true)
634 MD5.Add((unsigned char *)Start,BlkEnd - Start);
635 if (DoSHA1 == true)
636 SHA1.Add((unsigned char *)Start,BlkEnd - Start);
637 if (DoSHA256 == true)
638 SHA256.Add((unsigned char *)Start,BlkEnd - Start);
12cd178d
MV
639 if (DoSHA512 == true)
640 SHA512.Add((unsigned char *)Start,BlkEnd - Start);
f99da908 641
b2e465d6
AL
642 // Add an extra \n to the end, just in case
643 *BlkEnd++ = '\n';
644
645 /* Remove the PGP trailer. Some .dsc's have this without a blank line
646 before */
647 const char *Key = "-----BEGIN PGP SIGNATURE-----";
648 for (char *MsgEnd = Start; MsgEnd < BlkEnd - strlen(Key) -1; MsgEnd++)
649 {
650 if (*MsgEnd == '\n' && strncmp(MsgEnd+1,Key,strlen(Key)) == 0)
651 {
652 MsgEnd[1] = '\n';
653 break;
654 }
655 }
656
657 /* Read records until we locate the Source record. This neatly skips the
658 GPG header (which is RFC822 formed) without any trouble. */
659 pkgTagSection Tags;
660 do
661 {
662 unsigned Pos;
663 if (Tags.Scan(Start,BlkEnd - Start) == false)
664 return _error->Error("Could not find a record in the DSC '%s'",FileName.c_str());
665 if (Tags.Find("Source",Pos) == true)
666 break;
667 Start += Tags.size();
668 }
669 while (1);
670 Tags.Trim();
671
672 // Lookup the overide information, finding first the best priority.
673 string BestPrio;
b2e465d6 674 string Bins = Tags.FindS("Binary");
5200ec6f 675 char Buffer[Bins.length() + 1];
0b41e0e7 676 auto_ptr<Override::Item> OverItem(0);
5200ec6f 677 if (Bins.empty() == false)
b2e465d6
AL
678 {
679 strcpy(Buffer,Bins.c_str());
680
681 // Ignore too-long errors.
682 char *BinList[400];
683 TokSplitString(',',Buffer,BinList,sizeof(BinList)/sizeof(BinList[0]));
684
685 // Look at all the binaries
686 unsigned char BestPrioV = pkgCache::State::Extra;
687 for (unsigned I = 0; BinList[I] != 0; I++)
688 {
0b41e0e7
MV
689 auto_ptr<Override::Item> Itm(BOver.GetItem(BinList[I]));
690 if (Itm.get() == 0)
b2e465d6 691 continue;
b2e465d6
AL
692
693 unsigned char NewPrioV = debListParser::GetPrio(Itm->Priority);
694 if (NewPrioV < BestPrioV || BestPrio.empty() == true)
695 {
696 BestPrioV = NewPrioV;
697 BestPrio = Itm->Priority;
698 }
7524e348
MV
699
700 if (OverItem.get() == 0)
701 OverItem = Itm;
b2e465d6
AL
702 }
703 }
704
705 // If we need to do any rewriting of the header do it now..
0b41e0e7 706 if (OverItem.get() == 0)
b2e465d6
AL
707 {
708 if (NoOverride == false)
709 {
710 NewLine(1);
dc738e7a 711 ioprintf(c1out, _(" %s has no override entry\n"), Tags.FindS("Source").c_str());
b2e465d6
AL
712 }
713
0b41e0e7 714 OverItem = auto_ptr<Override::Item>(new Override::Item);
b2e465d6
AL
715 }
716
0b41e0e7 717 auto_ptr<Override::Item> SOverItem(SOver.GetItem(Tags.FindS("Source")));
cde41ae8 718 // const auto_ptr<Override::Item> autoSOverItem(SOverItem);
0b41e0e7 719 if (SOverItem.get() == 0)
b2e465d6 720 {
cde41ae8 721 ioprintf(c1out, _(" %s has no source override entry\n"), Tags.FindS("Source").c_str());
0b41e0e7
MV
722 SOverItem = auto_ptr<Override::Item>(BOver.GetItem(Tags.FindS("Source")));
723 if (SOverItem.get() == 0)
724 {
cde41ae8 725 ioprintf(c1out, _(" %s has no binary override entry either\n"), Tags.FindS("Source").c_str());
0b41e0e7
MV
726 SOverItem = auto_ptr<Override::Item>(new Override::Item);
727 *SOverItem = *OverItem;
728 }
b2e465d6
AL
729 }
730
731 // Add the dsc to the files hash list
f99da908 732 string const strippedName = flNotDir(FileName);
bf99a6d3 733 std::ostringstream ostreamFiles;
3c54407f 734 if (DoMD5 == true && Tags.Exists("Files"))
2a19ec29
MV
735 ostreamFiles << "\n " << string(MD5.Result()) << " " << St.st_size << " "
736 << strippedName << "\n " << Tags.FindS("Files");
bf99a6d3
DK
737 string const Files = ostreamFiles.str();
738
739 std::ostringstream ostreamSha1;
3c54407f 740 if (DoSHA1 == true && Tags.Exists("Checksums-Sha1"))
2a19ec29
MV
741 ostreamSha1 << "\n " << string(SHA1.Result()) << " " << St.st_size << " "
742 << strippedName << "\n " << Tags.FindS("Checksums-Sha1");
bf99a6d3
DK
743 string const ChecksumsSha1 = ostreamSha1.str();
744
745 std::ostringstream ostreamSha256;
3c54407f 746 if (DoSHA256 == true && Tags.Exists("Checksums-Sha256"))
2a19ec29
MV
747 ostreamSha256 << "\n " << string(SHA256.Result()) << " " << St.st_size << " "
748 << strippedName << "\n " << Tags.FindS("Checksums-Sha256");
bf99a6d3 749 string const ChecksumsSha256 = ostreamSha256.str();
f99da908 750
9a961efc
MV
751 std::ostringstream ostreamSha512;
752 if (Tags.Exists("Checksums-Sha512"))
753 ostreamSha512 << "\n " << string(SHA512.Result()) << " " << St.st_size << " "
754 << strippedName << "\n " << Tags.FindS("Checksums-Sha512");
755 string const ChecksumsSha512 = ostreamSha512.str();
756
b2e465d6
AL
757 // Strip the DirStrip prefix from the FileName and add the PathPrefix
758 string NewFileName;
759 if (DirStrip.empty() == false &&
760 FileName.length() > DirStrip.length() &&
8c58f506 761 stringcmp(DirStrip,OriginalPath,OriginalPath + DirStrip.length()) == 0)
b2e465d6
AL
762 NewFileName = string(OriginalPath + DirStrip.length());
763 else
764 NewFileName = OriginalPath;
765 if (PathPrefix.empty() == false)
766 NewFileName = flCombine(PathPrefix,NewFileName);
171c45bc 767
b2e465d6
AL
768 string Directory = flNotFile(OriginalPath);
769 string Package = Tags.FindS("Source");
171c45bc 770
b2e465d6
AL
771 // Perform the delinking operation over all of the files
772 string ParseJnk;
bf99a6d3 773 const char *C = Files.c_str();
ab3846c0 774 char *RealPath = NULL;
b2e465d6
AL
775 for (;isspace(*C); C++);
776 while (*C != 0)
777 {
778 // Parse each of the elements
779 if (ParseQuoteWord(C,ParseJnk) == false ||
780 ParseQuoteWord(C,ParseJnk) == false ||
781 ParseQuoteWord(C,ParseJnk) == false)
782 return _error->Error("Error parsing file record");
783
784 char Jnk[2];
785 string OriginalPath = Directory + ParseJnk;
ab3846c0
MV
786 if (readlink(OriginalPath.c_str(),Jnk,sizeof(Jnk)) != -1 &&
787 (RealPath = realpath(OriginalPath.c_str(),NULL)) != 0)
b2e465d6
AL
788 {
789 string RP = RealPath;
ab3846c0 790 free(RealPath);
cde41ae8 791 if (Delink(RP,OriginalPath.c_str(),Stats.DeLinkBytes,St.st_size) == false)
b2e465d6
AL
792 return false;
793 }
794 }
795
796 Directory = flNotFile(NewFileName);
797 if (Directory.length() > 2)
798 Directory.erase(Directory.end()-1);
171c45bc 799
b2e465d6 800 // This lists all the changes to the fields we are going to make.
f99da908
DK
801 // (5 hardcoded + checksums + maintainer + end marker)
802 TFRewriteData Changes[5+2+1+SOverItem->FieldOverride.size()+1];
64177f17 803
b2e465d6 804 unsigned int End = 0;
64177f17 805 SetTFRewriteData(Changes[End++],"Source",Package.c_str(),"Package");
3c54407f
DK
806 if (Files.empty() == false)
807 SetTFRewriteData(Changes[End++],"Files",Files.c_str());
808 if (ChecksumsSha1.empty() == false)
809 SetTFRewriteData(Changes[End++],"Checksums-Sha1",ChecksumsSha1.c_str());
810 if (ChecksumsSha256.empty() == false)
811 SetTFRewriteData(Changes[End++],"Checksums-Sha256",ChecksumsSha256.c_str());
12cd178d
MV
812 if (ChecksumsSha512.empty() == false)
813 SetTFRewriteData(Changes[End++],"Checksums-Sha512",ChecksumsSha512.c_str());
171c45bc
AL
814 if (Directory != "./")
815 SetTFRewriteData(Changes[End++],"Directory",Directory.c_str());
64177f17
AL
816 SetTFRewriteData(Changes[End++],"Priority",BestPrio.c_str());
817 SetTFRewriteData(Changes[End++],"Status",0);
b2e465d6
AL
818
819 // Rewrite the maintainer field if necessary
820 bool MaintFailed;
821 string NewMaint = OverItem->SwapMaint(Tags.FindS("Maintainer"),MaintFailed);
822 if (MaintFailed == true)
823 {
824 if (NoOverride == false)
825 {
826 NewLine(1);
dc738e7a
AL
827 ioprintf(c1out, _(" %s maintainer is %s not %s\n"), Package.c_str(),
828 Tags.FindS("Maintainer").c_str(), OverItem->OldMaint.c_str());
b2e465d6
AL
829 }
830 }
831 if (NewMaint.empty() == false)
64177f17
AL
832 SetTFRewriteData(Changes[End++], "Maintainer", NewMaint.c_str());
833
9209ec47 834 for (map<string,string>::const_iterator I = SOverItem->FieldOverride.begin();
f7f0d6c7 835 I != SOverItem->FieldOverride.end(); ++I)
64177f17
AL
836 SetTFRewriteData(Changes[End++],I->first.c_str(),I->second.c_str());
837
838 SetTFRewriteData(Changes[End++], 0, 0);
b2e465d6
AL
839
840 // Rewrite and store the fields.
841 if (TFRewrite(Output,Tags,TFRewriteSourceOrder,Changes) == false)
842 return false;
843 fprintf(Output,"\n");
844
845 Stats.Packages++;
846
847 return true;
848}
849 /*}}}*/
850
851// ContentsWriter::ContentsWriter - Constructor /*{{{*/
852// ---------------------------------------------------------------------
853/* */
31981076
DK
854ContentsWriter::ContentsWriter(string const &DB, string const &Arch) :
855 FTWScanner(Arch), Db(DB), Stats(Db.Stats)
b2e465d6
AL
856
857{
31981076 858 SetExts(".deb");
b2e465d6
AL
859 Output = stdout;
860}
861 /*}}}*/
862// ContentsWriter::DoPackage - Process a single package /*{{{*/
863// ---------------------------------------------------------------------
864/* If Package is the empty string the control record will be parsed to
865 determine what the package name is. */
9209ec47 866bool ContentsWriter::DoPackage(string FileName, string Package)
b2e465d6 867{
ff574e76 868 if (!Db.GetFileInfo(FileName, Package.empty(), true, false, false, false, false, false))
cde41ae8 869 {
b2e465d6 870 return false;
cde41ae8 871 }
b2e465d6
AL
872
873 // Parse the package name
874 if (Package.empty() == true)
875 {
b2e465d6
AL
876 Package = Db.Control.Section.FindS("Package");
877 }
878
879 Db.Contents.Add(Gen,Package);
880
881 return Db.Finish();
882}
883 /*}}}*/
884// ContentsWriter::ReadFromPkgs - Read from a packages file /*{{{*/
885// ---------------------------------------------------------------------
886/* */
9209ec47 887bool ContentsWriter::ReadFromPkgs(string const &PkgFile,string const &PkgCompress)
b2e465d6
AL
888{
889 MultiCompress Pkgs(PkgFile,PkgCompress,0,false);
890 if (_error->PendingError() == true)
891 return false;
12d1f5b3 892
b2e465d6 893 // Open the package file
12d1f5b3
DK
894 FileFd Fd;
895 if (Pkgs.OpenOld(Fd) == false)
b2e465d6 896 return false;
12d1f5b3 897
b2e465d6
AL
898 pkgTagFile Tags(&Fd);
899 if (_error->PendingError() == true)
b2e465d6 900 return false;
12d1f5b3 901
b2e465d6
AL
902 // Parse.
903 pkgTagSection Section;
904 while (Tags.Step(Section) == true)
905 {
906 string File = flCombine(Prefix,Section.FindS("FileName"));
907 string Package = Section.FindS("Section");
908 if (Package.empty() == false && Package.end()[-1] != '/')
909 {
910 Package += '/';
911 Package += Section.FindS("Package");
912 }
913 else
914 Package += Section.FindS("Package");
915
916 DoPackage(File,Package);
917 if (_error->empty() == false)
918 {
919 _error->Error("Errors apply to file '%s'",File.c_str());
920 _error->DumpErrors();
921 }
922 }
12d1f5b3 923
b2e465d6 924 // Tidy the compressor
12d1f5b3
DK
925 Fd.Close();
926
b2e465d6
AL
927 return true;
928}
98953965 929
b2e465d6 930 /*}}}*/
98953965
AL
931
932// ReleaseWriter::ReleaseWriter - Constructor /*{{{*/
933// ---------------------------------------------------------------------
934/* */
9209ec47 935ReleaseWriter::ReleaseWriter(string const &DB)
98953965 936{
3cb3fe76
DK
937 if (_config->FindB("APT::FTPArchive::Release::Default-Patterns", true) == true)
938 {
939 AddPattern("Packages");
940 AddPattern("Packages.gz");
941 AddPattern("Packages.bz2");
942 AddPattern("Packages.lzma");
b7080ced 943 AddPattern("Packages.xz");
8e3900d0 944 AddPattern("Translation-*");
3cb3fe76
DK
945 AddPattern("Sources");
946 AddPattern("Sources.gz");
947 AddPattern("Sources.bz2");
948 AddPattern("Sources.lzma");
b7080ced 949 AddPattern("Sources.xz");
3cb3fe76 950 AddPattern("Release");
0baf849d 951 AddPattern("Contents-*");
8d16c617 952 AddPattern("Index");
3cb3fe76
DK
953 AddPattern("md5sum.txt");
954 }
955 AddPatterns(_config->FindVector("APT::FTPArchive::Release::Patterns"));
187492a6 956
98953965 957 Output = stdout;
9209ec47 958 time_t const now = time(NULL);
cb12d0a6
DK
959
960 setlocale(LC_TIME, "C");
961
98953965
AL
962 char datestr[128];
963 if (strftime(datestr, sizeof(datestr), "%a, %d %b %Y %H:%M:%S UTC",
964 gmtime(&now)) == 0)
965 {
966 datestr[0] = '\0';
967 }
968
c99e48ec 969 time_t const validuntil = now + _config->FindI("APT::FTPArchive::Release::ValidTime", 0);
cdb623ed 970 char validstr[128];
c99e48ec
DK
971 if (now == validuntil ||
972 strftime(validstr, sizeof(validstr), "%a, %d %b %Y %H:%M:%S UTC",
973 gmtime(&validuntil)) == 0)
974 {
cdb623ed 975 validstr[0] = '\0';
c99e48ec
DK
976 }
977
cb12d0a6
DK
978 setlocale(LC_TIME, "");
979
98953965
AL
980 map<string,string> Fields;
981 Fields["Origin"] = "";
982 Fields["Label"] = "";
983 Fields["Suite"] = "";
984 Fields["Version"] = "";
985 Fields["Codename"] = "";
986 Fields["Date"] = datestr;
c99e48ec 987 Fields["Valid-Until"] = validstr;
98953965
AL
988 Fields["Architectures"] = "";
989 Fields["Components"] = "";
990 Fields["Description"] = "";
991
992 for(map<string,string>::const_iterator I = Fields.begin();
993 I != Fields.end();
994 ++I)
995 {
996 string Config = string("APT::FTPArchive::Release::") + (*I).first;
997 string Value = _config->Find(Config, (*I).second.c_str());
998 if (Value == "")
999 continue;
1000
1001 fprintf(Output, "%s: %s\n", (*I).first.c_str(), Value.c_str());
1002 }
3c54407f
DK
1003
1004 DoMD5 = _config->FindB("APT::FTPArchive::Release::MD5",DoMD5);
1005 DoSHA1 = _config->FindB("APT::FTPArchive::Release::SHA1",DoSHA1);
1006 DoSHA256 = _config->FindB("APT::FTPArchive::Release::SHA256",DoSHA256);
98953965
AL
1007}
1008 /*}}}*/
1009// ReleaseWriter::DoPackage - Process a single package /*{{{*/
1010// ---------------------------------------------------------------------
1011bool ReleaseWriter::DoPackage(string FileName)
1012{
1013 // Strip the DirStrip prefix from the FileName and add the PathPrefix
1014 string NewFileName;
1015 if (DirStrip.empty() == false &&
1016 FileName.length() > DirStrip.length() &&
1017 stringcmp(FileName.begin(),FileName.begin() + DirStrip.length(),
1018 DirStrip.begin(),DirStrip.end()) == 0)
c0eb6bc6 1019 {
98953965 1020 NewFileName = string(FileName.begin() + DirStrip.length(),FileName.end());
c0eb6bc6 1021 while (NewFileName[0] == '/')
9202409d 1022 NewFileName = string(NewFileName.begin() + 1,NewFileName.end());
c0eb6bc6 1023 }
98953965
AL
1024 else
1025 NewFileName = FileName;
c0eb6bc6 1026
98953965
AL
1027 if (PathPrefix.empty() == false)
1028 NewFileName = flCombine(PathPrefix,NewFileName);
1029
1030 FileFd fd(FileName, FileFd::ReadOnly);
1031
1032 if (!fd.IsOpen())
1033 {
1034 return false;
1035 }
1036
c0eb6bc6 1037 CheckSums[NewFileName].size = fd.Size();
f7291f62 1038
8c4e1f97 1039 Hashes hs;
109eb151 1040 hs.AddFD(fd, 0, DoMD5, DoSHA1, DoSHA256, DoSHA512);
3c54407f 1041 if (DoMD5 == true)
8c4e1f97 1042 CheckSums[NewFileName].MD5 = hs.MD5.Result();
3c54407f 1043 if (DoSHA1 == true)
8c4e1f97 1044 CheckSums[NewFileName].SHA1 = hs.SHA1.Result();
3c54407f 1045 if (DoSHA256 == true)
8c4e1f97 1046 CheckSums[NewFileName].SHA256 = hs.SHA256.Result();
9abccf4a 1047 if (DoSHA512 == true)
8c4e1f97 1048 CheckSums[NewFileName].SHA512 = hs.SHA512.Result();
98953965 1049 fd.Close();
8c4e1f97 1050
98953965
AL
1051 return true;
1052}
f7291f62
AL
1053
1054 /*}}}*/
1055// ReleaseWriter::Finish - Output the checksums /*{{{*/
1056// ---------------------------------------------------------------------
1057void ReleaseWriter::Finish()
1058{
3c54407f 1059 if (DoMD5 == true)
f7291f62 1060 {
3c54407f
DK
1061 fprintf(Output, "MD5Sum:\n");
1062 for(map<string,struct CheckSum>::const_iterator I = CheckSums.begin();
1063 I != CheckSums.end(); ++I)
1064 {
650faab0 1065 fprintf(Output, " %s %16llu %s\n",
3c54407f
DK
1066 (*I).second.MD5.c_str(),
1067 (*I).second.size,
1068 (*I).first.c_str());
1069 }
f7291f62 1070 }
3c54407f 1071 if (DoSHA1 == true)
f7291f62 1072 {
3c54407f
DK
1073 fprintf(Output, "SHA1:\n");
1074 for(map<string,struct CheckSum>::const_iterator I = CheckSums.begin();
1075 I != CheckSums.end(); ++I)
1076 {
650faab0 1077 fprintf(Output, " %s %16llu %s\n",
3c54407f
DK
1078 (*I).second.SHA1.c_str(),
1079 (*I).second.size,
1080 (*I).first.c_str());
1081 }
f7291f62 1082 }
3c54407f 1083 if (DoSHA256 == true)
cde41ae8 1084 {
3c54407f
DK
1085 fprintf(Output, "SHA256:\n");
1086 for(map<string,struct CheckSum>::const_iterator I = CheckSums.begin();
1087 I != CheckSums.end(); ++I)
1088 {
650faab0 1089 fprintf(Output, " %s %16llu %s\n",
3c54407f
DK
1090 (*I).second.SHA256.c_str(),
1091 (*I).second.size,
1092 (*I).first.c_str());
1093 }
cde41ae8 1094 }
9a961efc
MV
1095
1096 fprintf(Output, "SHA512:\n");
1097 for(map<string,struct CheckSum>::const_iterator I = CheckSums.begin();
1098 I != CheckSums.end();
1099 ++I)
1100 {
650faab0 1101 fprintf(Output, " %s %16llu %s\n",
9a961efc
MV
1102 (*I).second.SHA512.c_str(),
1103 (*I).second.size,
1104 (*I).first.c_str());
1105 }
1106
f7291f62 1107}