- some error handling + sanity checks for repomd parsing
[apt.git] / apt-pkg / rpm / rpmindexfile.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description                                                          /*{{{*/
3 // $Id: rpmindexfile.cc,v 1.4 2002/11/27 16:22:40 niemeyer Exp $
4 /* ######################################################################
5
6    RPM Specific sources.list types and the three sorts of RPM
7    index files.
8    
9    ##################################################################### */
10                                                                         /*}}}*/
11 // Include Files                                                        /*{{{*/
12 #ifdef __GNUG__
13 #pragma implementation "apt-pkg/rpmindexfile.h"
14 #endif
15
16 #include <config.h>
17
18 #ifdef HAVE_RPM
19
20 #include <apt-pkg/rpmindexfile.h>
21 #include <apt-pkg/rpmsrcrecords.h>
22 #include <apt-pkg/rpmlistparser.h>
23 #include <apt-pkg/rpmrecords.h>
24 #include <apt-pkg/rpmsystem.h>
25 #include <apt-pkg/rpmhandler.h>
26 #include <apt-pkg/rpmpackagedata.h>
27 #include <apt-pkg/sourcelist.h>
28 #include <apt-pkg/configuration.h>
29 #include <apt-pkg/progress.h>
30 #include <apt-pkg/error.h>
31 #include <apt-pkg/strutl.h>
32 #include <apt-pkg/acquire-item.h>
33 #include <apt-pkg/repomd.h>
34
35 #include <apti18n.h>
36
37 #include <sys/stat.h>
38                                                                         /*}}}*/
39 vector<pkgRepository *> RepList;
40
41 // rpmListIndex::Release* - Return the URI to the release file          /*{{{*/
42 // ---------------------------------------------------------------------
43 /* */
44 inline string rpmListIndex::ReleaseFile(string Type) const
45 {
46    return URItoFileName(ReleaseURI(Type));
47 }
48
49 string rpmListIndex::ReleaseURI(string Type) const
50 {
51    RPMPackageData *rpmdata = RPMPackageData::Singleton();
52    string Res;
53    if (Dist[Dist.size() - 1] == '/')
54    {
55       if (Dist != "/")
56          Res = URI + Dist;
57       else
58          Res = URI;
59    }
60    else
61       Res = URI + Dist + "/base/";
62
63    Res += Type;
64    
65    if (rpmdata->HasIndexTranslation() == true)
66    {
67       map<string,string> Dict;
68       Dict["uri"] = URI;
69       Dict["dist"] = Dist;
70       Dict["sect"] = "";
71       Dict["type"] = Type;
72       rpmdata->TranslateIndex(Res, Dict);
73    }
74
75    return Res;
76 }
77                                                                         /*}}}*/
78 // rpmListIndex::ReleaseInfo - One liner describing the index URI       /*{{{*/
79 // ---------------------------------------------------------------------
80 /* */
81 string rpmListIndex::ReleaseInfo(string Type) const 
82 {
83    string Info = ::URI::SiteOnly(URI) + ' ';
84    if (Dist[Dist.size() - 1] == '/')
85    {
86       if (Dist != "/")
87          Info += Dist;
88    }
89    else
90       Info += Dist;   
91    Info += " ";
92    Info += Type;
93    return Info;
94 }
95                                                                         /*}}}*/
96 // rpmListIndex::GetReleases - Fetch the index files                    /*{{{*/
97 // ---------------------------------------------------------------------
98 /* */
99 bool rpmListIndex::GetReleases(pkgAcquire *Owner) const
100 {
101    if (!Repository->Acquire)
102       return true;
103    Repository->Acquire = false;
104    new pkgAcqIndexRel(Owner,Repository,ReleaseURI("release"),
105                       ReleaseInfo("release"), "release", true);
106    return true;
107 }
108                                                                         /*}}}*/
109 // rpmListIndex::Info - One liner describing the index URI              /*{{{*/
110 // ---------------------------------------------------------------------
111 /* */
112 string rpmListIndex::Info(string Type) const 
113 {
114    string Info = ::URI::SiteOnly(URI) + ' ';
115    if (Dist[Dist.size() - 1] == '/')
116    {
117       if (Dist != "/")
118          Info += Dist;
119    }
120    else
121       Info += Dist + '/' + Section;   
122    Info += " ";
123    Info += Type;
124    return Info;
125 }
126                                                                         /*}}}*/
127 // rpmListIndex::Index* - Return the URI to the index files             /*{{{*/
128 // ---------------------------------------------------------------------
129 /* */
130 inline string rpmListIndex::IndexFile(string Type) const
131 {
132    return _config->FindDir("Dir::State::lists")+URItoFileName(IndexURI(Type));
133 }
134
135
136 string rpmListIndex::IndexURI(string Type) const
137 {
138    RPMPackageData *rpmdata = RPMPackageData::Singleton();
139    string Res;
140    if (Dist[Dist.size() - 1] == '/')
141    {
142       if (Dist != "/")
143          Res = URI + Dist;
144       else 
145          Res = URI;
146    }
147    else
148       Res = URI + Dist + "/base/";
149    
150    Res += Type + '.' + Section;
151
152    if (rpmdata->HasIndexTranslation() == true)
153    {
154       map<string,string> Dict;
155       Dict["uri"] = URI;
156       Dict["dist"] = Dist; 
157       Dict["sect"] = Section;
158       Dict["type"] = Type;
159       rpmdata->TranslateIndex(Res, Dict);
160    }
161
162    return Res;
163 }
164                                                                         /*}}}*/
165 // rpmListIndex::Exists - Check if the index is available               /*{{{*/
166 // ---------------------------------------------------------------------
167 /* */
168 bool rpmListIndex::Exists() const
169 {
170    return FileExists(IndexPath());
171 }
172                                                                         /*}}}*/
173 // rpmListIndex::Size - Return the size of the index                    /*{{{*/
174 // ---------------------------------------------------------------------
175 /* */
176 unsigned long rpmListIndex::Size() const
177 {
178    struct stat S;
179    if (stat(IndexPath().c_str(),&S) != 0)
180       return 0;
181    return S.st_size;
182 }
183                                                                         /*}}}*/
184 // rpmListIndex::Describe - Give a descriptive path to the index        /*{{{*/
185 // ---------------------------------------------------------------------
186 /* */
187 string rpmListIndex::Describe(bool Short) const
188 {
189    char S[300];
190    if (Short == true)
191       snprintf(S,sizeof(S),"%s",Info(MainType()).c_str());
192    else
193       snprintf(S,sizeof(S),"%s (%s)",Info(MainType()).c_str(),
194          IndexFile(MainType()).c_str());
195    return S;
196 }
197                                                                         /*}}}*/
198
199 // SrcListIndex::SourceInfo - Short 1 liner describing a source         /*{{{*/
200 // ---------------------------------------------------------------------
201 string rpmSrcListIndex::SourceInfo(pkgSrcRecords::Parser const &Record,
202                                    pkgSrcRecords::File const &File) const
203 {
204    string Res;
205    Res = ::URI::SiteOnly(URI) + ' ';
206    if (Dist[Dist.size() - 1] == '/')
207    {
208       if (Dist != "/")
209          Res += Dist;
210    }      
211    else
212       Res += Dist + '/' + Section;
213    
214    Res += " ";
215    Res += Record.Package();
216    Res += " ";
217    Res += Record.Version();
218    if (File.Type.empty() == false)
219       Res += " (" + File.Type + ")";
220    return Res;
221 }
222                                                                         /*}}}*/
223 // SrcListIndex::ArchiveURI - URI for the archive                       /*{{{*/
224 // ---------------------------------------------------------------------
225 string rpmSrcListIndex::ArchiveURI(string File) const
226 {
227    RPMPackageData *rpmdata = RPMPackageData::Singleton();
228    string Res;
229    
230    if (Dist[Dist.size() - 1] == '/')
231    {
232       if (Dist != "/")
233          Res = URI + Dist;
234       else
235          Res = URI;
236    }
237    else
238       Res = URI + Dist;
239    
240    if (File.find("/") != string::npos)
241       Res += '/' + File;
242    else
243       Res += "/SRPMS."+Section + '/' + File;
244
245    if (rpmdata->HasSourceTranslation() == true)
246    {
247       map<string,string> Dict;
248       Dict["uri"] = URI;
249       Dict["dist"] = Dist; 
250       Dict["sect"] = Section;
251       string::size_type pos = File.rfind("/");
252       if (pos != string::npos)
253          Dict["file"] = string(File, pos+1);
254       else
255          Dict["file"] = File;
256       
257       rpmdata->TranslateSource(Res, Dict);
258    }
259          
260    return Res;
261 }
262                                                                         /*}}}*/
263 // SrcListIndex::CreateSrcParser - Get a parser for the source files    /*{{{*/
264 // ---------------------------------------------------------------------
265 /* */
266 pkgSrcRecords::Parser *rpmSrcListIndex::CreateSrcParser() const
267 {
268    return new rpmSrcRecordParser(IndexPath(), this);
269 }
270                                                                         /*}}}*/
271 // SrcListIndex::GetIndexes - Fetch the index files                     /*{{{*/
272 // ---------------------------------------------------------------------
273 /* */
274 bool rpmSrcListIndex::GetIndexes(pkgAcquire *Owner) const
275 {
276    // Ignore indexes for repositories that could not be authenticated
277    if (Repository->IsAuthenticated() == true && 
278        Repository->HasRelease() == false)
279       return true;
280    new pkgAcqIndex(Owner,Repository,IndexURI("srclist"),Info("srclist"),
281                    "srclist");
282    return true;
283 }
284                                                                         /*}}}*/
285
286 // PkgListIndex::ArchiveInfo - Short version of the archive url         /*{{{*/
287 // ---------------------------------------------------------------------
288 /* This is a shorter version that is designed to be < 60 chars or so */
289 string rpmPkgListIndex::ArchiveInfo(pkgCache::VerIterator Ver) const
290 {
291    string Res = ::URI::SiteOnly(URI) + ' ';
292    if (Dist[Dist.size() - 1] == '/')
293    {
294       if (Dist != "/")
295          Res += Dist;
296    }
297    else
298       Res += Dist + '/' + Section;
299    
300    Res += " ";
301    Res += Ver.ParentPkg().Name();
302    Res += " ";
303    Res += Ver.VerStr();
304    return Res;
305 }
306                                                                         /*}}}*/
307 // PkgListIndex::ArchiveURI - URI for the archive                       /*{{{*/
308 // ---------------------------------------------------------------------
309 string rpmPkgListIndex::ArchiveURI(string File) const
310 {
311    RPMPackageData *rpmdata = RPMPackageData::Singleton();
312    string Res;
313    if (Dist[Dist.size() - 1] == '/')
314    {
315       if (Dist != "/")
316          Res = URI + Dist;
317       else
318          Res = URI;
319    }
320    else
321       Res = URI + Dist;
322
323    if (File.find("/") != string::npos)
324       Res += '/' + File;
325    else
326       Res += "/RPMS." + Section + '/' + File;
327
328    if (rpmdata->HasBinaryTranslation() == true)
329    {
330       map<string,string> Dict;
331       Dict["uri"] = URI;
332       Dict["dist"] = Dist; 
333       Dict["sect"] = Section;
334       string::size_type pos = File.rfind("/");
335       if (pos != string::npos)
336          Dict["file"] = string(File, pos+1);
337       else
338          Dict["file"] = File;
339       rpmdata->TranslateBinary(Res, Dict);
340    }
341          
342    return Res;
343 }
344                                                                         /*}}}*/
345 // PkgListIndex::GetIndexes - Fetch the index files                     /*{{{*/
346 // ---------------------------------------------------------------------
347 /* */
348 bool rpmPkgListIndex::GetIndexes(pkgAcquire *Owner) const
349 {
350    // Ignore indexes for repositories that could not be authenticated
351    if (Repository->IsAuthenticated() == true && 
352        Repository->HasRelease() == false)
353       return true;
354    new pkgAcqIndex(Owner,Repository,IndexURI("pkglist"),Info("pkglist"),
355                    "pkglist");
356    new pkgAcqIndexRel(Owner,Repository,IndexURI("release"),Info("release"),
357                       "release");
358    return true;
359 }
360                                                                         /*}}}*/
361 // PkgListIndex::Merge - Load the index file into a cache               /*{{{*/
362 // ---------------------------------------------------------------------
363 /* */
364 bool rpmPkgListIndex::Merge(pkgCacheGenerator &Gen,OpProgress &Prog) const
365 {
366    string PackageFile = IndexPath();
367    RPMHandler *Handler = CreateHandler();
368
369    Prog.SubProgress(0,Info(MainType()));
370    ::URI Tmp(URI);
371    if (Gen.SelectFile(PackageFile,Tmp.Host,*this) == false)
372    {
373       delete Handler;
374       return _error->Error(_("Problem with SelectFile %s"),PackageFile.c_str());
375    }
376
377    // Store the IMS information
378    pkgCache::PkgFileIterator File = Gen.GetCurFile();
379    struct stat St;
380    if (stat(PackageFile.c_str(),&St) != 0) 
381    {
382       delete Handler;
383       return _error->Errno("stat",_("Failed to stat %s"), PackageFile.c_str());
384    }
385    File->Size = St.st_size;
386    File->mtime = St.st_mtime;
387    
388    rpmListParser Parser(Handler);
389    if (_error->PendingError() == true) 
390    {
391       delete Handler;
392       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
393    }
394    
395    if (Gen.MergeList(Parser) == false)
396    {
397       delete Handler;
398       return _error->Error(_("Problem with MergeList %s"),PackageFile.c_str());
399    }
400    
401    delete Handler;
402
403    // Check the release file
404    string RelFile = ReleasePath();
405    if (FileExists(RelFile) == true)
406    {
407       FileFd Rel(RelFile,FileFd::ReadOnly);
408       if (_error->PendingError() == true)
409          return false;
410       Parser.LoadReleaseInfo(File,Rel);
411       Rel.Seek(0);
412    }
413
414    return true;
415 }
416                                                                         /*}}}*/
417 // PkgListIndex::MergeFileProvides - Process file dependencies if any   /*{{{*/
418 // ---------------------------------------------------------------------
419 /* */
420 bool rpmPkgListIndex::MergeFileProvides(pkgCacheGenerator &Gen,
421                                         OpProgress &Prog) const
422 {
423    string PackageFile = IndexPath();
424    RPMHandler *Handler = CreateHandler();
425    rpmListParser Parser(Handler);
426    if (_error->PendingError() == true) {
427       delete Handler;
428       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
429    }
430    // We call SubProgress with Size(), since we won't call SelectFile() here.
431    Prog.SubProgress(Size(),Info("pkglist"));
432    if (Gen.MergeFileProvides(Parser) == false)
433       return _error->Error(_("Problem with MergeFileProvides %s"),
434                            PackageFile.c_str());
435    delete Handler;
436    return true;
437 }
438                                                                         /*}}}*/
439 // PkgListIndex::FindInCache - Find this index                          /*{{{*/
440 // ---------------------------------------------------------------------
441 /* */
442 pkgCache::PkgFileIterator rpmPkgListIndex::FindInCache(pkgCache &Cache) const
443 {
444    string FileName = IndexPath();
445    pkgCache::PkgFileIterator File = Cache.FileBegin();
446    for (; File.end() == false; File++)
447    {
448       if (FileName != File.FileName())
449          continue;
450       
451       struct stat St;
452       if (stat(File.FileName(),&St) != 0)
453          return pkgCache::PkgFileIterator(Cache);
454
455       if ((unsigned)St.st_size != File->Size || St.st_mtime != File->mtime)
456          return pkgCache::PkgFileIterator(Cache);
457       return File;
458    }
459    
460    return File;
461 }
462                                                                         /*}}}*/
463
464 // PkgDirIndex::Index* - Return the URI to the index files              /*{{{*/
465 // ---------------------------------------------------------------------
466 /* */
467 string rpmPkgDirIndex::IndexPath() const
468 {
469    return ::URI(ArchiveURI("")).Path;
470 }
471                                                                         /*}}}*/
472 // PkgDirIndex::Release* - Return the URI to the index files            /*{{{*/
473 // ---------------------------------------------------------------------
474 /* */
475 string rpmPkgDirIndex::ReleasePath() const
476 {
477    return ::URI(IndexURI("release")).Path;
478 }
479                                                                         /*}}}*/
480 // PkgDirIndex::Size - Return the size of the index                     /*{{{*/
481 // ---------------------------------------------------------------------
482 /* This is really only used for progress reporting. */
483 unsigned long rpmPkgDirIndex::Size() const
484 {
485    // XXX: Must optimize this somehow.
486    RPMHandler *Handler = CreateHandler();
487    unsigned long Res = Handler->Size();
488    delete Handler;
489    return Res;
490 }
491                                                                         /*}}}*/
492
493 // SrcDirIndex::Index* - Return the URI to the index files              /*{{{*/
494 // ---------------------------------------------------------------------
495 /* */
496 string rpmSrcDirIndex::IndexPath() const
497 {
498    return ::URI(ArchiveURI("")).Path;
499 }
500                                                                         /*}}}*/
501 // SrcDirIndex::Size - Return the size of the index                     /*{{{*/
502 // ---------------------------------------------------------------------
503 /* This is really only used for progress reporting. */
504 unsigned long rpmSrcDirIndex::Size() const
505 {
506    // XXX: Must optimize this somehow.
507    RPMHandler *Handler = CreateHandler();
508    unsigned long Res = Handler->Size();
509    delete Handler;
510    return Res;
511 }
512
513 // SinglePkgIndex::ArchiveURI - URI for the archive                     /*{{{*/
514 // ---------------------------------------------------------------------
515 string rpmSinglePkgIndex::ArchiveURI(string File) const
516 {
517    char *cwd = getcwd(NULL,0);
518    if (File[0] == '.' && File[1] == '/')
519       File = string(File, 2);
520    string URI = "file://"+flCombine(cwd, File);
521    free(cwd);
522    return URI;
523 }
524                                                                         /*}}}*/
525 // SinglePkgIndex::ArchiveURI - URI for the archive                     /*{{{*/
526 // ---------------------------------------------------------------------
527 string rpmSingleSrcIndex::ArchiveURI(string File) const
528 {
529    char *cwd = getcwd(NULL,0);
530    if (File[0] == '.' && File[1] == '/')
531       File = string(File, 2);
532    string URI = "file://"+flCombine(cwd, File);
533    free(cwd);
534    return URI;
535 }
536
537 string rpmRepomdIndex::ArchiveURI(string File) const
538 {
539    string Res;
540
541    Res += URI + '/' + Dist + '/' + File;
542    return Res;
543 }
544
545 string rpmRepomdIndex::ArchiveInfo(pkgCache::VerIterator Ver) const
546 {
547    string Res = ::URI::SiteOnly(URI) + ' ';
548    if (Dist[Dist.size() - 1] == '/')
549    {
550       if (Dist != "/")
551          Res += Dist;
552    }
553    else
554       Res += Dist + '/';
555    
556    Res += " ";
557    Res += Ver.ParentPkg().Name();
558    Res += " ";
559    Res += Ver.VerStr();
560    return Res;
561 }
562 pkgCache::PkgFileIterator rpmRepomdIndex::FindInCache(pkgCache &Cache) const
563 {
564    string FileName = IndexPath();
565    pkgCache::PkgFileIterator File = Cache.FileBegin();
566    for (; File.end() == false; File++)
567    {
568       if (FileName != File.FileName())
569          continue;
570       
571       struct stat St;
572       if (stat(File.FileName(),&St) != 0)
573          return pkgCache::PkgFileIterator(Cache);
574
575       if ((unsigned)St.st_size != File->Size || St.st_mtime != File->mtime)
576          return pkgCache::PkgFileIterator(Cache);
577       return File;
578    }
579    
580    return File;
581 }
582 string rpmRepomdIndex::ReleaseURI(string Type) const
583 {
584    string Res;
585    Res = URI + Dist + "/repodata/" + "repomd.xml";
586    return Res;
587 }
588
589 string rpmRepomdIndex::ReleaseInfo(string Type) const
590 {
591    string Info = ::URI::SiteOnly(URI) + ' ';
592    if (Dist[Dist.size() - 1] == '/')
593    {
594       if (Dist != "/")
595          Info += Dist;
596    }
597    else
598       Info += Dist;   
599    Info += " ";
600    Info += Type;
601    return Info;
602 };
603
604 string rpmRepomdIndex::Info(string Type) const 
605 {
606    string Info = ::URI::SiteOnly(URI) + ' ';
607    if (Dist[Dist.size() - 1] == '/')
608    {
609       if (Dist != "/")
610          Info += Dist;
611    }
612    else
613       Info += Dist + '/' ;   
614    Info += " ";
615    Info += Type;
616    return Info;
617 }
618
619 string rpmRepomdIndex::IndexURI(string Type) const
620 {
621    string Res = URI + Dist;
622    if (Dist[Dist.size() - 1] != '/') {
623          Res += "/";
624    }
625    // When called with the main type it's primary.xml we want, at other
626    // times it could be eg filelists.xml
627    if (Type == "repomd" || Type == "repomd-src")
628       Res += "repodata/primary.xml";
629    else
630       Res += "repodata/" + Type;
631    return Res;
632 }
633                                                                         /*}}}*/
634 bool rpmRepomdIndex::GetReleases(pkgAcquire *Owner) const
635 {
636    if (!Repository->Acquire)
637       return true;
638    Repository->Acquire = false;
639    // ignore for now - we need to parse the file for checksum and 
640    // optimally for location of other xml files as well
641    new pkgAcqIndexRel(Owner,Repository,ReleaseURI("repomd.xml"),
642                       ReleaseInfo("repomd.xml"), "repomd.xml", true);
643    return true;
644 }
645
646 bool rpmRepomdIndex::GetIndexes(pkgAcquire *Owner) const
647 {
648    new pkgAcqIndex(Owner,Repository,IndexURI("primary.xml"),
649                    Info("primary.xml"), "primary.xml");
650    new pkgAcqIndex(Owner,Repository,IndexURI("filelists.xml"),
651                    Info("filelists.xml"), "filelists.xml");
652 #if 0
653    new pkgAcqIndex(Owner,Repository,IndexURI("other.xml"),
654                    Info("other.xml"), "other.xml");
655 #endif
656    return true;
657 }
658
659 string rpmRepomdIndex::Describe(bool Short) const
660 {
661    char S[300];
662    if (Short == true)
663       snprintf(S,sizeof(S),"%s",Info(MainType()).c_str());
664    else
665       snprintf(S,sizeof(S),"%s (%s)",Info(MainType()).c_str(),
666          IndexFile(MainType()).c_str());
667    return S;
668 }
669
670 string rpmRepomdIndex::IndexFile(string Type) const
671 {
672    return _config->FindDir("Dir::State::lists")+URItoFileName(IndexURI(Type));
673 };
674
675
676 bool rpmRepomdIndex::Exists() const
677 {
678    return FileExists(IndexPath());
679 }
680
681 string rpmRepomdIndex::ReleasePath() const
682 {
683    return _config->FindDir("Dir::State::lists")+URItoFileName(ReleaseURI("release.xml"));
684 }
685
686 unsigned long rpmRepomdIndex::Size() const
687 {
688    // ugh.. all this just to get the package count ;(
689    RPMHandler *Handler = CreateHandler();
690    unsigned long Res = Handler->Size();
691    delete Handler;
692    return Res;
693 }
694
695 bool rpmRepomdIndex::Merge(pkgCacheGenerator &Gen,OpProgress &Prog) const
696 {
697    string PackageFile = IndexPath();
698    RPMHandler *Handler = CreateHandler();
699
700    Prog.SubProgress(0,Info(MainType()));
701    ::URI Tmp(URI);
702    if (Gen.SelectFile(PackageFile,Tmp.Host,*this) == false)
703    {
704       delete Handler;
705       return _error->Error(_("Problem with SelectFile %s"),PackageFile.c_str());
706    }
707
708    // Store the IMS information
709    pkgCache::PkgFileIterator File = Gen.GetCurFile();
710    struct stat St;
711    if (stat(PackageFile.c_str(),&St) != 0) 
712    {
713       delete Handler;
714       return _error->Errno("stat",_("Failed to stat %s"), PackageFile.c_str());
715    }
716    File->Size = St.st_size;
717    File->mtime = St.st_mtime;
718    
719    rpmRepomdParser Parser(Handler);
720    if (_error->PendingError() == true) 
721    {
722       delete Handler;
723       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
724    }
725    
726    if (Gen.MergeList(Parser) == false)
727    {
728       delete Handler;
729       return _error->Error(_("Problem with MergeList %s"),PackageFile.c_str());
730    }
731    
732    delete Handler;
733
734    // Check the release file
735    string RelFile = ReleasePath();
736    if (FileExists(RelFile) == true)
737    {
738       Parser.LoadReleaseInfo(File,RelFile);
739    }
740
741    return true;
742 }
743
744 bool rpmRepomdIndex::MergeFileProvides(pkgCacheGenerator &Gen,
745                                         OpProgress &Prog) const
746 {
747    string PackageFile = IndexPath();
748    RPMHandler *Handler = CreateHandler();
749    rpmListParser Parser(Handler);
750    if (_error->PendingError() == true) {
751       delete Handler;
752       return _error->Error(_("Problem opening %s"),PackageFile.c_str());
753    }
754    // We call SubProgress with Size(), since we won't call SelectFile() here.
755    Prog.SubProgress(Size(),Info("pkglist"));
756    if (Gen.MergeFileProvides(Parser) == false)
757       return _error->Error(_("Problem with MergeFileProvides %s"),
758                            PackageFile.c_str());
759    delete Handler;
760    return true;
761 }
762
763 pkgSrcRecords::Parser *rpmRepomdSrcIndex::CreateSrcParser() const
764 {
765    return new rpmSrcRecordParser(IndexPath(), this);
766 }
767
768 string rpmRepomdSrcIndex::SourceInfo(pkgSrcRecords::Parser const &Record,
769                                      pkgSrcRecords::File const &File) const
770 {
771    string Res;
772    Res = ::URI::SiteOnly(URI) + ' ';
773    if (Dist[Dist.size() - 1] == '/')
774    {
775       if (Dist != "/")
776          Res += Dist;
777    }      
778    else
779       Res += Dist + '/' + Section;
780    
781    Res += " ";
782    Res += Record.Package();
783    Res += " ";
784    Res += Record.Version();
785    if (File.Type.empty() == false)
786       Res += " (" + File.Type + ")";
787    return Res;
788 }
789
790 // DatabaseIndex::rpmDatabaseIndex - Constructor                        /*{{{*/
791 // ---------------------------------------------------------------------
792 /* */
793 rpmDatabaseIndex::rpmDatabaseIndex()
794 {
795 }
796                                                                         /*}}}*/
797 // DatabaseIndex::Size - Return the size of the index                   /*{{{*/
798 // ---------------------------------------------------------------------
799 /* */
800 unsigned long rpmDatabaseIndex::Size() const
801 {
802    return rpmSys.GetDBHandler()->Size();
803 }
804                                                                         /*}}}*/
805 // DatabaseIndex::CreateHandler - Create a RPMHandler for this file     /*{{{*/
806 // ---------------------------------------------------------------------
807 RPMHandler *rpmDatabaseIndex::CreateHandler() const
808 {
809    return rpmSys.GetDBHandler();
810 }
811                                                                         /*}}}*/
812 // DatabaseIndex::Merge - Load the index file into a cache              /*{{{*/
813 // ---------------------------------------------------------------------
814 /* */
815 bool rpmDatabaseIndex::Merge(pkgCacheGenerator &Gen,OpProgress &Prog) const
816 {
817    RPMDBHandler *Handler = rpmSys.GetDBHandler();
818    rpmListParser Parser(Handler);
819    if (_error->PendingError() == true)
820       return _error->Error(_("Problem opening RPM database"));
821    
822    Prog.SubProgress(0,"RPM Database");
823    if (Gen.SelectFile(Handler->DataPath(false),string(),*this,pkgCache::Flag::NotSource) == false)
824       return _error->Error(_("Problem with SelectFile RPM Database"));
825
826    // Store the IMS information
827    pkgCache::PkgFileIterator CFile = Gen.GetCurFile();
828    struct stat St;
829    if (stat(Handler->DataPath(false).c_str(),&St) != 0)
830       return _error->Errno("fstat",_("Failed to stat %s"), Handler->DataPath(false).c_str());
831    CFile->Size = St.st_size;
832    CFile->mtime = Handler->Mtime();
833    
834    if (Gen.MergeList(Parser) == false)
835       return _error->Error(_("Problem with MergeList %s"),
836                            Handler->DataPath(false).c_str());
837    return true;
838 }
839                                                                         /*}}}*/
840 // DatabaseIndex::MergeFileProvides - Process file dependencies if any  /*{{{*/
841 // ---------------------------------------------------------------------
842 /* */
843 bool rpmDatabaseIndex::MergeFileProvides(pkgCacheGenerator &Gen,
844                                          OpProgress &Prog) const
845 {
846    RPMDBHandler *Handler = rpmSys.GetDBHandler();
847    rpmListParser Parser(Handler);
848    if (_error->PendingError() == true)
849       return _error->Error(_("Problem opening RPM database"));
850    // We call SubProgress with Size(), since we won't call SelectFile() here.
851    Prog.SubProgress(Size(),"RPM Database");
852    if (Gen.MergeFileProvides(Parser) == false)
853       return _error->Error(_("Problem with MergeFileProvides %s"),
854                            Handler->DataPath(false).c_str());
855    return true;
856 }
857                                                                         /*}}}*/
858 // DatabaseIndex::FindInCache - Find this index                         /*{{{*/
859 // ---------------------------------------------------------------------
860 /* */
861 pkgCache::PkgFileIterator rpmDatabaseIndex::FindInCache(pkgCache &Cache) const
862 {
863    pkgCache::PkgFileIterator File = Cache.FileBegin();
864    for (; File.end() == false; File++)
865    {
866       if (rpmSys.GetDBHandler()->DataPath(false) != File.FileName())
867          continue;
868       struct stat St;
869       if (stat(File.FileName(),&St) != 0)
870          return pkgCache::PkgFileIterator(Cache);
871       if ((unsigned)St.st_size != File->Size || St.st_mtime != File->mtime)
872          return pkgCache::PkgFileIterator(Cache);
873       return File;
874    }   
875    return File;
876 }
877                                                                         /*}}}*/
878
879 // Source List types for rpm                                            /*{{{*/
880
881 class rpmSLTypeGen : public pkgSourceList::Type
882 {
883    public:
884
885    rpmSLTypeGen()
886    {
887       Name = "rpm";
888       Label = "Standard RPM source tree";
889    }   
890
891    pkgRepository *FindRepository(string URI,string Dist,
892                                  const pkgSourceList::Vendor *Vendor) const
893    {
894       for (vector<pkgRepository *>::const_iterator iter = RepList.begin();
895            iter != RepList.end(); iter++) 
896       {
897          if ((*iter)->URI == URI && (*iter)->Dist == Dist) 
898          {       
899             if (Vendor != NULL)
900                (*iter)->FingerPrint = Vendor->FingerPrint;
901             return *iter;
902          }
903       }
904       return NULL;
905    }
906
907    pkgRepository *GetRepository(string URI,string Dist,
908                                 const pkgSourceList::Vendor *Vendor) const
909    {
910       pkgRepository *Rep = FindRepository(URI,Dist,Vendor);
911       if (Rep != NULL)
912          return Rep;
913
914       string BaseURI;
915       if (Dist[Dist.size() - 1] == '/')
916       {
917          if (Dist != "/")
918             BaseURI = URI + Dist;
919          else 
920             BaseURI = URI + '/';
921       }
922       else
923          BaseURI = URI + Dist + '/';
924
925       Rep = new pkgRepository(URI,Dist,Vendor,BaseURI);
926       RepList.push_back(Rep);
927       return Rep;
928    }
929 };
930
931
932 class rpmSLTypeRpm : public rpmSLTypeGen
933 {
934    public:
935
936    bool CreateItem(vector<pkgIndexFile *> &List,
937                    string URI, string Dist, string Section,
938                    pkgSourceList::Vendor const *Vendor) const 
939    {
940       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
941       List.push_back(new rpmPkgListIndex(URI,Dist,Section,Rep));
942       return true;
943    };
944
945    rpmSLTypeRpm()
946    {
947       Name = "rpm";
948       Label = "Standard RPM binary tree";
949    }   
950 };
951
952 class rpmSLTypeSrpm : public rpmSLTypeGen
953 {
954    public:
955
956    bool CreateItem(vector<pkgIndexFile *> &List,
957                    string URI, string Dist, string Section,
958                    pkgSourceList::Vendor const *Vendor) const 
959    {
960       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
961       List.push_back(new rpmSrcListIndex(URI,Dist,Section,Rep));
962       return true;
963    };  
964    
965    rpmSLTypeSrpm()
966    {
967       Name = "rpm-src";
968       Label = "Standard RPM source tree";
969    }   
970 };
971
972 class rpmSLTypeRpmDir : public rpmSLTypeGen
973 {
974    public:
975
976    bool CreateItem(vector<pkgIndexFile *> &List,
977                    string URI, string Dist, string Section,
978                    pkgSourceList::Vendor const *Vendor) const 
979    {
980       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
981       List.push_back(new rpmPkgDirIndex(URI,Dist,Section,Rep));
982       return true;
983    };
984
985    rpmSLTypeRpmDir()
986    {
987       Name = "rpm-dir";
988       Label = "Local RPM directory tree";
989    }   
990 };
991
992 class rpmSLTypeSrpmDir : public rpmSLTypeGen
993 {
994    public:
995
996    bool CreateItem(vector<pkgIndexFile *> &List,
997                    string URI, string Dist, string Section,
998                    pkgSourceList::Vendor const *Vendor) const 
999    {
1000       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
1001       List.push_back(new rpmSrcDirIndex(URI,Dist,Section,Rep));
1002       return true;
1003    };
1004
1005    rpmSLTypeSrpmDir()
1006    {
1007       Name = "rpm-src-dir";
1008       Label = "Local SRPM directory tree";
1009    }   
1010 };
1011
1012 class rpmSLTypeRepomd : public rpmSLTypeGen
1013 {
1014    public:
1015
1016    pkgRepository *GetRepository(string URI,string Dist,
1017                                 const pkgSourceList::Vendor *Vendor) const
1018    {
1019       pkgRepository *Rep = FindRepository(URI,Dist,Vendor);
1020       if (Rep != NULL)
1021          return Rep;
1022
1023       string BaseURI;
1024       if (Dist[Dist.size() - 1] == '/')
1025       {
1026          if (Dist != "/")
1027             BaseURI = URI + Dist;
1028          else 
1029             BaseURI = URI + '/';
1030       }
1031       else
1032          BaseURI = URI + Dist + '/';
1033
1034       Rep = new repomdRepository(URI,Dist,Vendor,BaseURI);
1035       RepList.push_back(Rep);
1036       return Rep;
1037    }
1038
1039    bool CreateItem(vector<pkgIndexFile *> &List,
1040                    string URI, string Dist, string Section,
1041                    pkgSourceList::Vendor const *Vendor) const 
1042    {
1043       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
1044       List.push_back(new rpmRepomdPkgIndex(URI,Dist,Section,Rep));
1045       return true;
1046    };
1047
1048    bool ParseLine(vector<pkgIndexFile *> &List,
1049                   pkgSourceList::Vendor const *Vendor,
1050                   const char *Buffer,
1051                   unsigned long CurLine,string File) const
1052    {
1053       string URI;
1054       string Dist;
1055       if (ParseQuoteWord(Buffer,URI) == false)
1056          return _error->Error(_("Malformed line %lu in source list %s (URI)"),CurLine,File.c_str());
1057       if (ParseQuoteWord(Buffer,Dist) == false)
1058          return _error->Error(_("Malformed line %lu in source list %s (dist)"),CurLine,File.c_str());
1059
1060       if (FixupURI(URI) == false)
1061          return _error->Error(_("Malformed line %lu in source list %s (URI parse)"),CurLine,File.c_str());
1062       
1063       Dist = SubstVar(Dist,"$(ARCH)",_config->Find("APT::Architecture"));
1064       Dist = SubstVar(Dist,"$(VERSION)",_config->Find("APT::DistroVersion"));
1065
1066       if (CreateItem(List,URI,Dist,"",Vendor) == false)
1067          return false;
1068
1069       return true;
1070    };
1071
1072
1073    rpmSLTypeRepomd()
1074    {
1075       Name = "repomd";
1076       Label = "RepoMD tree";
1077    }   
1078 };
1079 class rpmSLTypeRepomdSrc : public rpmSLTypeRepomd
1080 {
1081    public:
1082
1083    bool CreateItem(vector<pkgIndexFile *> &List,
1084                    string URI, string Dist, string Section,
1085                    pkgSourceList::Vendor const *Vendor) const 
1086    {
1087       pkgRepository *Rep = GetRepository(URI,Dist,Vendor);
1088       List.push_back(new rpmRepomdSrcIndex(URI,Dist,Section,Rep));
1089       return true;
1090    };
1091
1092    rpmSLTypeRepomdSrc()
1093    {
1094       Name = "repomd-src";
1095       Label = "RepoMD src tree";
1096    }   
1097 };
1098
1099 rpmSLTypeRpm _apt_rpmType;
1100 rpmSLTypeSrpm _apt_rpmSrcType;
1101 rpmSLTypeRpmDir _apt_rpmDirType;
1102 rpmSLTypeSrpmDir _apt_rpmSrcDirType;
1103 rpmSLTypeRepomd _apt_repomdType;
1104 rpmSLTypeRepomdSrc _apt_repomdSrcType;
1105                                                                         /*}}}*/
1106 // Index File types for rpm                                             /*{{{*/
1107 class rpmIFTypeSrc : public pkgIndexFile::Type
1108 {
1109    public:
1110    
1111    rpmIFTypeSrc() {Label = "RPM Source Index";};
1112 };
1113 class rpmIFTypePkg : public pkgIndexFile::Type
1114 {
1115    public:
1116    
1117    virtual pkgRecords::Parser *CreatePkgParser(pkgCache::PkgFileIterator File) const
1118    {
1119       return new rpmRecordParser(File.FileName(),*File.Cache());
1120    };
1121    rpmIFTypePkg() {Label = "RPM Package Index";};
1122 };
1123 class rpmIFTypeDatabase : public pkgIndexFile::Type
1124 {
1125    public:
1126    
1127    virtual pkgRecords::Parser *CreatePkgParser(pkgCache::PkgFileIterator File) const
1128    {
1129       return new rpmRecordParser(File.FileName(),*File.Cache());
1130    };
1131    rpmIFTypeDatabase() {Label = "RPM Database";};
1132 };
1133 static rpmIFTypeSrc _apt_Src;
1134 static rpmIFTypePkg _apt_Pkg;
1135 static rpmIFTypeDatabase _apt_DB;
1136
1137 const pkgIndexFile::Type *rpmSrcListIndex::GetType() const
1138 {
1139    return &_apt_Src;
1140 }
1141 const pkgIndexFile::Type *rpmPkgListIndex::GetType() const
1142 {
1143    return &_apt_Pkg;
1144 }
1145 const pkgIndexFile::Type *rpmSrcDirIndex::GetType() const
1146 {
1147    return &_apt_Src;
1148 }
1149 const pkgIndexFile::Type *rpmPkgDirIndex::GetType() const
1150 {
1151    return &_apt_Pkg;
1152 }
1153 const pkgIndexFile::Type *rpmSinglePkgIndex::GetType() const
1154 {
1155    return &_apt_Pkg;
1156 }
1157 const pkgIndexFile::Type *rpmSingleSrcIndex::GetType() const
1158 {
1159    return &_apt_Src;
1160 }
1161 const pkgIndexFile::Type *rpmDatabaseIndex::GetType() const
1162 {
1163    return &_apt_DB;
1164 }
1165 const pkgIndexFile::Type *rpmRepomdPkgIndex::GetType() const
1166 {
1167    return &_apt_Pkg;
1168 }
1169 const pkgIndexFile::Type *rpmRepomdSrcIndex::GetType() const
1170 {
1171    return &_apt_Src;
1172 }
1173
1174                                                                         /*}}}*/
1175 #endif /* HAVE_RPM */
1176
1177 // vim:sts=3:sw=3