file.cc 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. // -*- mode: cpp; mode: fold -*-
  2. // Description /*{{{*/
  3. // $Id: file.cc,v 1.9.2.1 2004/01/16 18:58:50 mdz Exp $
  4. /* ######################################################################
  5. File URI method for APT
  6. This simply checks that the file specified exists, if so the relevant
  7. information is returned. If a .gz filename is specified then the file
  8. name with .gz removed will also be checked and information about it
  9. will be returned in Alt-*
  10. ##################################################################### */
  11. /*}}}*/
  12. // Include Files /*{{{*/
  13. #include <config.h>
  14. #include <apt-pkg/acquire-method.h>
  15. #include <apt-pkg/aptconfiguration.h>
  16. #include <apt-pkg/error.h>
  17. #include <apt-pkg/hashes.h>
  18. #include <apt-pkg/fileutl.h>
  19. #include <apt-pkg/strutl.h>
  20. #include <string>
  21. #include <sys/stat.h>
  22. #include <apti18n.h>
  23. /*}}}*/
  24. class FileMethod : public pkgAcqMethod
  25. {
  26. virtual bool Fetch(FetchItem *Itm);
  27. public:
  28. FileMethod() : pkgAcqMethod("1.0",SingleInstance | SendConfig | LocalOnly) {};
  29. };
  30. // FileMethod::Fetch - Fetch a file /*{{{*/
  31. // ---------------------------------------------------------------------
  32. /* */
  33. bool FileMethod::Fetch(FetchItem *Itm)
  34. {
  35. URI Get = Itm->Uri;
  36. std::string File = Get.Path;
  37. FetchResult Res;
  38. if (Get.Host.empty() == false)
  39. return _error->Error(_("Invalid URI, local URIS must not start with //"));
  40. // See if the file exists
  41. struct stat Buf;
  42. if (stat(File.c_str(),&Buf) == 0)
  43. {
  44. Res.Size = Buf.st_size;
  45. Res.Filename = File;
  46. Res.LastModified = Buf.st_mtime;
  47. Res.IMSHit = false;
  48. if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
  49. Res.IMSHit = true;
  50. }
  51. // See if the uncompressed file exists and reuse it
  52. std::vector<std::string> extensions = APT::Configuration::getCompressorExtensions();
  53. for (std::vector<std::string>::const_iterator ext = extensions.begin(); ext != extensions.end(); ++ext)
  54. {
  55. if (APT::String::Endswith(File, *ext) == true)
  56. {
  57. std::string const unfile = File.substr(0, File.length() - ext->length() - 1);
  58. if (stat(unfile.c_str(),&Buf) == 0)
  59. {
  60. FetchResult AltRes;
  61. AltRes.Size = Buf.st_size;
  62. AltRes.Filename = unfile;
  63. AltRes.LastModified = Buf.st_mtime;
  64. AltRes.IMSHit = false;
  65. if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
  66. AltRes.IMSHit = true;
  67. URIDone(Res,&AltRes);
  68. return true;
  69. }
  70. // no break here as we could have situations similar to '.gz' vs '.tar.gz' here
  71. }
  72. }
  73. if (Res.Filename.empty() == true)
  74. return _error->Error(_("File not found"));
  75. Hashes Hash(Itm->ExpectedHashes);
  76. FileFd Fd(Res.Filename, FileFd::ReadOnly);
  77. Hash.AddFD(Fd);
  78. Res.TakeHashes(Hash);
  79. URIDone(Res);
  80. return true;
  81. }
  82. /*}}}*/
  83. int main()
  84. {
  85. setlocale(LC_ALL, "");
  86. FileMethod Mth;
  87. return Mth.Run();
  88. }