methods/cdrom.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: cdrom.cc,v 1.20.2.1 2004/01/16 18:58:50 mdz 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 #include <apt-pkg/hashes.h>
  17
  18 #include <sys/stat.h>
  19 #include <unistd.h>
  20
  21 #include <iostream>
  22 #include <apti18n.h>
  23                                                                         /*}}}*/
  24
  25 using namespace std;
  26
  27 class CDROMMethod : public pkgAcqMethod
  28 {
  29    bool DatabaseLoaded;
  30    ::Configuration Database;
  31    string CurrentID;
  32    string CDROM;
  33    bool MountedByApt;
  34
  35    virtual bool Fetch(FetchItem *Itm);
  36    string GetID(string Name);
  37    virtual void Exit();
  38
  39    public:
  40
  41    CDROMMethod();
  42 };
  43
  44 // CDROMMethod::CDROMethod - Constructor                                /*{{{*/
  45 // ---------------------------------------------------------------------
  46 /* */
  47 CDROMMethod::CDROMMethod() : pkgAcqMethod("1.0",SingleInstance | LocalOnly |
  48                                           SendConfig | NeedsCleanup |
  49                                           Removable),
  50                                           DatabaseLoaded(false),
  51                                           MountedByApt(false)
  52 {
  53 };
  54                                                                         /*}}}*/
  55 // CDROMMethod::Exit - Unmount the disc if necessary                    /*{{{*/
  56 // ---------------------------------------------------------------------
  57 /* */
  58 void CDROMMethod::Exit()
  59 {
  60    if (MountedByApt == true)
  61       UnmountCdrom(CDROM);
  62 }
  63                                                                         /*}}}*/
  64 // CDROMMethod::GetID - Search the database for a matching string       /*{{{*/
  65 // ---------------------------------------------------------------------
  66 /* */
  67 string CDROMMethod::GetID(string Name)
  68 {
  69    // Search for an ID
  70    const Configuration::Item *Top = Database.Tree("CD");
  71    if (Top != 0)
  72       Top = Top->Child;
  73
  74    for (; Top != 0;)
  75    {
  76       if (Top->Value == Name)
  77          return Top->Tag;
  78
  79       Top = Top->Next;
  80    }
  81    return string();
  82 }
  83                                                                         /*}}}*/
  84 // CDROMMethod::Fetch - Fetch a file                                    /*{{{*/
  85 // ---------------------------------------------------------------------
  86 /* */
  87 bool CDROMMethod::Fetch(FetchItem *Itm)
  88 {
  89    URI Get = Itm->Uri;
  90    string File = Get.Path;
  91    FetchResult Res;
  92
  93    bool Debug = _config->FindB("Debug::Acquire::cdrom",false);
  94
  95    /* All IMS queries are returned as a hit, CDROMs are readonly so
  96       time stamps never change */
  97    if (Itm->LastModified != 0)
  98    {
  99       Res.LastModified = Itm->LastModified;
 100       Res.IMSHit = true;
 101       Res.Filename = Itm->DestFile;
 102       URIDone(Res);
 103       return true;
 104    }
 105
 106    // Load the database
 107    if (DatabaseLoaded == false)
 108    {
 109       // Read the database
 110       string DFile = _config->FindFile("Dir::State::cdroms");
 111       if (FileExists(DFile) == true)
 112       {
 113          if (ReadConfigFile(Database,DFile) == false)
 114             return _error->Error(_("Unable to read the cdrom database %s"),
 115                           DFile.c_str());
 116       }
 117       DatabaseLoaded = true;
 118    }
 119
 120    // All non IMS queries for package files fail.
 121    if (Itm->IndexFile == true || GetID(Get.Host).empty() == true)
 122    {
 123       Fail(_("Please use apt-cdrom to make this CD-ROM recognized by APT."
 124            " apt-get update cannot be used to add new CD-ROMs"));
 125       return true;
 126    }
 127
 128    // We already have a CD inserted, but it is the wrong one
 129    if (CurrentID.empty() == false && Database.Find("CD::" + CurrentID) != Get.Host)
 130    {
 131       Fail(_("Wrong CD-ROM"),true);
 132       return true;
 133    }
 134
 135    CDROM = _config->FindDir("Acquire::cdrom::mount","/cdrom/");
 136    if (CDROM[0] == '.')
 137       CDROM= SafeGetCWD() + '/' + CDROM;
 138    string NewID;
 139    while (CurrentID.empty() == true)
 140    {
 141       bool Hit = false;
 142       if(!IsMounted(CDROM))
 143          MountedByApt = MountCdrom(CDROM);
 144       for (unsigned int Version = 2; Version != 0; Version--)
 145       {
 146          if (IdentCdrom(CDROM,NewID,Version) == false)
 147             return false;
 148
 149          if (Debug == true)
 150             clog << "ID " << Version << " " << NewID << endl;
 151
 152          // A hit
 153          if (Database.Find("CD::" + NewID) == Get.Host)
 154          {
 155             Hit = true;
 156             break;
 157          }
 158       }
 159
 160       if (Hit == true)
 161          break;
 162
 163       // I suppose this should prompt somehow?
 164       if (_config->FindB("APT::CDROM::NoMount",false) == false &&
 165           UnmountCdrom(CDROM) == false)
 166          return _error->Error(_("Unable to unmount the CD-ROM in %s, it may still be in use."),
 167                               CDROM.c_str());
 168       if (MediaFail(Get.Host,CDROM) == false)
 169       {
 170          CurrentID = "FAIL";
 171          return _error->Error(_("Disk not found."));
 172       }
 173    }
 174
 175    // Found a CD
 176    Res.Filename = CDROM + File;
 177    struct stat Buf;
 178    if (stat(Res.Filename.c_str(),&Buf) != 0)
 179       return _error->Error(_("File not found"));
 180
 181    if (NewID.empty() == false)
 182       CurrentID = NewID;
 183    Res.LastModified = Buf.st_mtime;
 184    Res.Size = Buf.st_size;
 185
 186    Hashes Hash;
 187    FileFd Fd(Res.Filename, FileFd::ReadOnly);
 188    Hash.AddFD(Fd.Fd(), Fd.Size());
 189    Res.TakeHashes(Hash);
 190
 191    URIDone(Res);
 192    return true;
 193 }
 194                                                                         /*}}}*/
 195
 196 int main()
 197 {
 198    setlocale(LC_ALL, "");
 199
 200    CDROMMethod Mth;
 201    return Mth.Run();
 202 }