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