cmdline/apt-cdrom.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: apt-cdrom.cc,v 1.45 2003/11/19 23:50:51 mdz Exp $
   4 /* ######################################################################
   5
   6    APT CDROM - Tool for handling APT's CDROM database.
   7
   8    Currently the only option is 'add' which will take the current CD
   9    in the drive and add it into the database.
  10
  11    ##################################################################### */
  12                                                                         /*}}}*/
  13 // Include Files                                                        /*{{{*/
  14 #include<config.h>
  15
  16 #include <apt-pkg/cmndline.h>
  17 #include <apt-pkg/error.h>
  18 #include <apt-pkg/init.h>
  19 #include <apt-pkg/fileutl.h>
  20 #include <apt-pkg/progress.h>
  21 #include <apt-pkg/cdromutl.h>
  22 #include <apt-pkg/strutl.h>
  23 #include <apt-pkg/cdrom.h>
  24 #include <apt-pkg/configuration.h>
  25 #include <apt-pkg/pkgsystem.h>
  26
  27 #include <iostream>
  28 #include <vector>
  29 #include <string>
  30 #include <sys/stat.h>
  31 #include <unistd.h>
  32
  33 #include <apt-private/private-cmndline.h>
  34 #include <apt-private/private-output.h>
  35 #include <apt-private/private-main.h>
  36
  37 #include <apti18n.h>
  38                                                                         /*}}}*/
  39
  40 using namespace std;
  41
  42 class pkgCdromTextStatus : public pkgCdromStatus                        /*{{{*/
  43 {
  44 protected:
  45    OpTextProgress Progress;
  46    void Prompt(const char *Text);
  47    string PromptLine(const char *Text);
  48    bool AskCdromName(string &name) APT_OVERRIDE;
  49
  50 public:
  51    virtual void Update(string text, int current) APT_OVERRIDE;
  52    virtual bool ChangeCdrom() APT_OVERRIDE;
  53    virtual OpProgress* GetOpProgress() APT_OVERRIDE;
  54 };
  55
  56 void pkgCdromTextStatus::Prompt(const char *Text)
  57 {
  58    char C;
  59    cout << Text << ' ' << flush;
  60    if (read(STDIN_FILENO,&C,1) < 0)
  61       _error->Errno("pkgCdromTextStatus::Prompt",
  62                     "Failed to read from standard input (not a terminal?)");
  63    if (C != '\n')
  64       cout << endl;
  65 }
  66
  67 string pkgCdromTextStatus::PromptLine(const char *Text)
  68 {
  69    cout << Text << ':' << endl;
  70
  71    string Res;
  72    getline(cin,Res);
  73    return Res;
  74 }
  75
  76 bool pkgCdromTextStatus::AskCdromName(string &name)
  77 {
  78    cout << _("Please provide a name for this Disc, such as 'Debian 5.0.3 Disk 1'") << flush;
  79    name = PromptLine("");
  80
  81    return true;
  82 }
  83
  84
  85 void pkgCdromTextStatus::Update(string text, int /*current*/)
  86 {
  87    if(text.size() > 0)
  88       cout << text << flush;
  89 }
  90
  91 bool pkgCdromTextStatus::ChangeCdrom()
  92 {
  93    Prompt(_("Please insert a Disc in the drive and press [Enter]"));
  94    return true;
  95 }
  96
  97 APT_CONST OpProgress* pkgCdromTextStatus::GetOpProgress()
  98 {
  99    return &Progress;
 100 }
 101                                                                         /*}}}*/
 102 // AddOrIdent - Add or Ident a CDROM                                    /*{{{*/
 103 static bool AddOrIdent(bool const Add)
 104 {
 105    pkgUdevCdromDevices UdevCdroms;
 106    pkgCdromTextStatus log;
 107    pkgCdrom cdrom;
 108
 109    bool oneSuccessful = false;
 110    bool AutoDetect = _config->FindB("Acquire::cdrom::AutoDetect", true);
 111    if (AutoDetect == true && UdevCdroms.Dlopen() == true)
 112    {
 113       bool const Debug = _config->FindB("Debug::Acquire::cdrom", false);
 114       std::string const CDMount = _config->Find("Acquire::cdrom::mount");
 115       bool const NoMount = _config->FindB("APT::CDROM::NoMount", false);
 116       if (NoMount == false)
 117          _config->Set("APT::CDROM::NoMount", true);
 118
 119       vector<struct CdromDevice> const v = UdevCdroms.Scan();
 120       for (std::vector<struct CdromDevice>::const_iterator cd = v.begin(); cd != v.end(); ++cd)
 121       {
 122          if (Debug)
 123             clog << "Looking at device:"
 124                << "\tDeviveName: '" << cd->DeviceName << "'"
 125                << "\tIsMounted: '" << cd->Mounted << "'"
 126                << "\tMountPoint: '" << cd->MountPath << "'"
 127                << endl;
 128
 129          std::string AptMountPoint;
 130          if (cd->Mounted)
 131             _config->Set("Acquire::cdrom::mount", cd->MountPath);
 132          else if (NoMount == true)
 133             continue;
 134          else
 135          {
 136             AptMountPoint = _config->FindDir("Dir::Media::MountPath");
 137             if (FileExists(AptMountPoint) == false)
 138                mkdir(AptMountPoint.c_str(), 0750);
 139             if(MountCdrom(AptMountPoint, cd->DeviceName) == false)
 140             {
 141                _error->Warning(_("Failed to mount '%s' to '%s'"), cd->DeviceName.c_str(), AptMountPoint.c_str());
 142                continue;
 143             }
 144             _config->Set("Acquire::cdrom::mount", AptMountPoint);
 145          }
 146
 147          _error->PushToStack();
 148          if (Add == true)
 149             oneSuccessful = cdrom.Add(&log);
 150          else
 151          {
 152             std::string id;
 153             oneSuccessful = cdrom.Ident(id, &log);
 154          }
 155          _error->MergeWithStack();
 156
 157          if (AptMountPoint.empty() == false)
 158             UnmountCdrom(AptMountPoint);
 159       }
 160       if (NoMount == false)
 161          _config->Set("APT::CDROM::NoMount", NoMount);
 162       _config->Set("Acquire::cdrom::mount", CDMount);
 163    }
 164
 165    // fallback if auto-detect didn't work
 166    if (oneSuccessful == false)
 167    {
 168       _error->PushToStack();
 169       if (Add == true)
 170          oneSuccessful = cdrom.Add(&log);
 171       else
 172       {
 173          std::string id;
 174          oneSuccessful = cdrom.Ident(id, &log);
 175       }
 176       _error->MergeWithStack();
 177    }
 178
 179    if (oneSuccessful == false)
 180       _error->Error("%s", _("No CD-ROM could be auto-detected or found using the default mount point.\n"
 181       "You may try the --cdrom option to set the CD-ROM mount point.\n"
 182       "See 'man apt-cdrom' for more information about the CD-ROM auto-detection and mount point."));
 183    else if (Add == true)
 184       cout << _("Repeat this process for the rest of the CDs in your set.") << endl;
 185
 186    return oneSuccessful;
 187 }
 188                                                                         /*}}}*/
 189 // DoAdd - Add a new CDROM                                              /*{{{*/
 190 // ---------------------------------------------------------------------
 191 /* This does the main add bit.. We show some status and things. The
 192    sequence is to mount/umount the CD, Ident it then scan it for package
 193    files and reduce that list. Then we copy over the package files and
 194    verify them. Then rewrite the database files */
 195 static bool DoAdd(CommandLine &)
 196 {
 197    return AddOrIdent(true);
 198 }
 199                                                                         /*}}}*/
 200 // DoIdent - Ident a CDROM                                              /*{{{*/
 201 static bool DoIdent(CommandLine &)
 202 {
 203    return AddOrIdent(false);
 204 }
 205                                                                         /*}}}*/
 206 static bool ShowHelp(CommandLine &)                                     /*{{{*/
 207 {
 208    std::cout <<
 209       _("Usage: apt-cdrom [options] command\n"
 210       "\n"
 211       "apt-cdrom is used to add CDROM's, USB flashdrives and other removable\n"
 212       "media types as package sources to APT. The mount point and device\n"
 213       "information is taken from apt.conf(5), udev(7) and fstab(5).\n");
 214    return true;
 215 }
 216                                                                         /*}}}*/
 217 static std::vector<aptDispatchWithHelp> GetCommands()                   /*{{{*/
 218 {
 219    return {
 220       {"add", &DoAdd, "Add a CDROM"},
 221       {"ident", &DoIdent, "Report the identity of a CDROM"},
 222       {nullptr, nullptr, nullptr}
 223    };
 224 }
 225                                                                         /*}}}*/
 226 int main(int argc,const char *argv[])                                   /*{{{*/
 227 {
 228    // Parse the command line and initialize the package library
 229    CommandLine CmdL;
 230    auto const Cmds = ParseCommandLine(CmdL, APT_CMD::APT_CDROM, &_config, &_system, argc, argv, &ShowHelp, &GetCommands);
 231
 232    InitOutput();
 233
 234    return DispatchCommandLine(CmdL, Cmds);
 235 }
 236                                                                         /*}}}*/