methods/cdrom.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: cdrom.cc,v 1.9 1999/05/23 05:45:13 jgg Exp $
   4 /* ######################################################################
   5
   6    CDROM URI method for APT
   7
   8    ##################################################################### */
   9                                                                         /*}}}*/
  10 // Include Files                                                        /*{{{*/
  11 #include <apt-pkg/acquire-method.h>
  12 #include <apt-pkg/cdromutl.h>
  13 #include <apt-pkg/error.h>
  14 #include <apt-pkg/configuration.h>
  15 #include <apt-pkg/fileutl.h>
  16
  17 #include <sys/stat.h>
  18 #include <unistd.h>
  19                                                                         /*}}}*/
  20
  21 class CDROMMethod : public pkgAcqMethod
  22 {
  23    Configuration Database;
  24    bool DatabaseLoaded;
  25    string CurrentID;
  26
  27    virtual bool Fetch(FetchItem *Itm);
  28    string GetID(string Name);
  29
  30    public:
  31
  32    CDROMMethod();
  33 };
  34
  35 // CDROMMethod::CDROMethod - Constructor                                /*{{{*/
  36 // ---------------------------------------------------------------------
  37 /* */
  38 CDROMMethod::CDROMMethod() : pkgAcqMethod("1.0",SingleInstance | LocalOnly |
  39                                           SendConfig), DatabaseLoaded(false)
  40 {
  41 };
  42                                                                         /*}}}*/
  43 // CDROMMethod::GetID - Search the database for a matching string       /*{{{*/
  44 // ---------------------------------------------------------------------
  45 /* */
  46 string CDROMMethod::GetID(string Name)
  47 {
  48    // Search for an ID
  49    const Configuration::Item *Top = Database.Tree("CD");
  50    if (Top != 0)
  51       Top = Top->Child;
  52
  53    for (; Top != 0;)
  54    {
  55       if (Top->Value == Name)
  56          return Top->Tag;
  57
  58       Top = Top->Next;
  59    }
  60    return string();
  61 }
  62                                                                         /*}}}*/
  63 // CDROMMethod::Fetch - Fetch a file                                    /*{{{*/
  64 // ---------------------------------------------------------------------
  65 /* */
  66 bool CDROMMethod::Fetch(FetchItem *Itm)
  67 {
  68    URI Get = Itm->Uri;
  69    string File = Get.Path;
  70    FetchResult Res;
  71
  72    /* All IMS queries are returned as a hit, CDROMs are readonly so
  73       time stamps never change */
  74    if (Itm->LastModified != 0)
  75    {
  76       Res.LastModified = Itm->LastModified;
  77       Res.IMSHit = true;
  78       URIDone(Res);
  79       return true;
  80    }
  81
  82    // Load the database
  83    if (DatabaseLoaded == false)
  84    {
  85       // Read the database
  86       string DFile = _config->FindFile("Dir::State::cdroms");
  87       if (FileExists(DFile) == true)
  88       {
  89          if (ReadConfigFile(Database,DFile) == false)
  90             return _error->Error("Unable to read the cdrom database %s",
  91                           DFile.c_str());
  92       }
  93       DatabaseLoaded = true;
  94    }
  95
  96    // All non IMS queries for package files fail.
  97    if (Itm->IndexFile == true || GetID(Get.Host).empty() == true)
  98    {
  99       Fail("Please use apt-cdrom to make this CD recognized by APT."
 100            " apt-get update cannot be used to add new CDs");
 101       return true;
 102    }
 103
 104    // We already have a CD inserted, but it is the wrong one
 105    if (CurrentID.empty() == false && Database.Find("CD::" + CurrentID) != Get.Host)
 106    {
 107       Fail("Wrong CD",true);
 108       return true;
 109    }
 110
 111    string CDROM = _config->FindDir("Acquire::cdrom::mount","/cdrom/");
 112    if (CDROM[0] == '.')
 113       CDROM= SafeGetCWD() + '/' + CDROM;
 114    string NewID;
 115    while (1)
 116    {
 117       if (IdentCdrom(CDROM,NewID) == false)
 118          return false;
 119
 120       // A hit
 121       if (Database.Find("CD::" + NewID) == Get.Host)
 122          break;
 123
 124       // I suppose this should prompt somehow?
 125       if (UnmountCdrom(CDROM) == false)
 126          return _error->Error("Unable to unmount the CD-ROM in %s, it may still be in use.",
 127                               CDROM.c_str());
 128       if (MediaFail(Get.Host,CDROM) == false)
 129       {
 130          CurrentID = "FAIL";
 131          Fail("Wrong CD",true);
 132          return true;
 133       }
 134
 135       MountCdrom(CDROM);
 136    }
 137
 138    // Found a CD
 139    Res.Filename = CDROM + File;
 140    struct stat Buf;
 141    if (stat(Res.Filename.c_str(),&Buf) != 0)
 142       return _error->Error("File not found");
 143
 144    CurrentID = NewID;
 145    Res.LastModified = Buf.st_mtime;
 146    Res.IMSHit = true;
 147    Res.Size = Buf.st_size;
 148    URIDone(Res);
 149    return true;
 150 }
 151                                                                         /*}}}*/
 152
 153 int main()
 154 {
 155    CDROMMethod Mth;
 156    return Mth.Run();
 157 }