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/acquire.h>
  24 #include <apt-pkg/acquire-item.h>
  25 #include <apt-pkg/cdrom.h>
  26 #include <apt-pkg/configuration.h>
  27 #include <apt-pkg/pkgsystem.h>
  28
  29 #include <locale.h>
  30 #include <iostream>
  31 #include <fstream>
  32 #include <vector>
  33 #include <algorithm>
  34 #include <sys/stat.h>
  35 #include <fcntl.h>
  36 #include <dirent.h>
  37 #include <unistd.h>
  38 #include <stdio.h>
  39
  40 #include <apti18n.h>
  41                                                                         /*}}}*/
  42 static const char *W_NO_CDROM_FOUND = \
  43    N_("No CD-ROM could be auto-detected or found using "
  44       "the default mount point.\n"
  45       "You may try the --cdrom option to set the CD-ROM mount point. "
  46       "See 'man apt-cdrom' for more "
  47       "information about the CD-ROM auto-detection and mount point.");
  48
  49 using namespace std;
  50
  51 class pkgCdromTextStatus : public pkgCdromStatus                        /*{{{*/
  52 {
  53 protected:
  54    OpTextProgress Progress;
  55    void Prompt(const char *Text);
  56    string PromptLine(const char *Text);
  57    bool AskCdromName(string &name);
  58
  59 public:
  60    virtual void Update(string text, int current);
  61    virtual bool ChangeCdrom();
  62    virtual OpProgress* GetOpProgress();
  63 };
  64
  65 void pkgCdromTextStatus::Prompt(const char *Text)
  66 {
  67    char C;
  68    cout << Text << ' ' << flush;
  69    if (read(STDIN_FILENO,&C,1) < 0)
  70       _error->Errno("pkgCdromTextStatus::Prompt",
  71                     "Failed to read from standard input (not a terminal?)");
  72    if (C != '\n')
  73       cout << endl;
  74 }
  75
  76 string pkgCdromTextStatus::PromptLine(const char *Text)
  77 {
  78    cout << Text << ':' << endl;
  79
  80    string Res;
  81    getline(cin,Res);
  82    return Res;
  83 }
  84
  85 bool pkgCdromTextStatus::AskCdromName(string &name)
  86 {
  87    cout << _("Please provide a name for this Disc, such as 'Debian 5.0.3 Disk 1'") << flush;
  88    name = PromptLine("");
  89
  90    return true;
  91 }
  92
  93
  94 void pkgCdromTextStatus::Update(string text, int current)
  95 {
  96    if(text.size() > 0)
  97       cout << text << flush;
  98 }
  99
 100 bool pkgCdromTextStatus::ChangeCdrom()
 101 {
 102    Prompt(_("Please insert a Disc in the drive and press enter"));
 103    return true;
 104 }
 105
 106 OpProgress* pkgCdromTextStatus::GetOpProgress()
 107 {
 108    return &Progress;
 109 };
 110                                                                         /*}}}*/
 111 // SetupAutoDetect                                                      /*{{{*/
 112 bool AutoDetectCdrom(pkgUdevCdromDevices &UdevCdroms, unsigned int &i)
 113 {
 114    bool Debug =  _config->FindB("Debug::Acquire::cdrom", false);
 115
 116    vector<struct CdromDevice> v = UdevCdroms.Scan();
 117    if (i >= v.size())
 118       return false;
 119
 120    if (Debug)
 121       clog << "Looking at devce " << i
 122            << " DeviveName: " << v[i].DeviceName
 123            << " IsMounted: '" << v[i].Mounted << "'"
 124            << " MountPoint: '" << v[i].MountPath << "'"
 125            << endl;
 126
 127    if (v[i].Mounted)
 128    {
 129       // set the right options
 130       _config->Set("Acquire::cdrom::mount", v[i].MountPath);
 131       _config->Set("APT::CDROM::NoMount", true);
 132    } else {
 133       string AptMountPoint = _config->FindDir("Dir::Media::MountPath");
 134       if (!FileExists(AptMountPoint))
 135          mkdir(AptMountPoint.c_str(), 0750);
 136       if(MountCdrom(AptMountPoint, v[i].DeviceName) == false)
 137          _error->Warning(_("Failed to mount '%s' to '%s'"), v[i].DeviceName.c_str(), AptMountPoint.c_str());
 138       _config->Set("Acquire::cdrom::mount", AptMountPoint);
 139       _config->Set("APT::CDROM::NoMount", true);
 140    }
 141    i++;
 142
 143    return true;
 144 }
 145                                                                         /*}}}*/
 146 // DoAdd - Add a new CDROM                                              /*{{{*/
 147 // ---------------------------------------------------------------------
 148 /* This does the main add bit.. We show some status and things. The
 149    sequence is to mount/umount the CD, Ident it then scan it for package
 150    files and reduce that list. Then we copy over the package files and
 151    verify them. Then rewrite the database files */
 152 bool DoAdd(CommandLine &)
 153 {
 154    pkgUdevCdromDevices UdevCdroms;
 155    pkgCdromTextStatus log;
 156    pkgCdrom cdrom;
 157    bool res = true;
 158
 159    bool AutoDetect = _config->FindB("Acquire::cdrom::AutoDetect", true);
 160    unsigned int count = 0;
 161    if (AutoDetect && UdevCdroms.Dlopen())
 162       while (AutoDetectCdrom(UdevCdroms, count))
 163          res &= cdrom.Add(&log);
 164    if (count == 0) {
 165       res = cdrom.Add(&log);
 166       if (res == false) {
 167          _error->Error("%s", _(W_NO_CDROM_FOUND));
 168       }
 169    }
 170
 171    if(res)
 172       cout << _("Repeat this process for the rest of the CDs in your set.") << endl;
 173
 174    return res;
 175 }
 176                                                                         /*}}}*/
 177 // DoIdent - Ident a CDROM                                              /*{{{*/
 178 // ---------------------------------------------------------------------
 179 /* */
 180 bool DoIdent(CommandLine &)
 181 {
 182    pkgUdevCdromDevices UdevCdroms;
 183    string ident;
 184    pkgCdromTextStatus log;
 185    pkgCdrom cdrom;
 186    bool res = true;
 187
 188    bool AutoDetect = _config->FindB("Acquire::cdrom::AutoDetect");
 189
 190    unsigned int count = 0;
 191    if (AutoDetect && UdevCdroms.Dlopen())
 192       while (AutoDetectCdrom(UdevCdroms, count))
 193          res &= cdrom.Ident(ident, &log);
 194    if (count == 0) {
 195       res = cdrom.Ident(ident, &log);
 196       if (res == false) {
 197          _error->Error("%s", _(W_NO_CDROM_FOUND));
 198       }
 199    }
 200    return res;
 201 }
 202                                                                         /*}}}*/
 203 // ShowHelp - Show the help screen                                      /*{{{*/
 204 // ---------------------------------------------------------------------
 205 /* */
 206 int ShowHelp()
 207 {
 208    ioprintf(cout,_("%s %s for %s compiled on %s %s\n"),PACKAGE,PACKAGE_VERSION,
 209             COMMON_ARCH,__DATE__,__TIME__);
 210    if (_config->FindB("version") == true)
 211       return 0;
 212
 213    cout <<
 214       "Usage: apt-cdrom [options] command\n"
 215       "\n"
 216       "apt-cdrom is a tool to add CDROM's to APT's source list. The\n"
 217       "CDROM mount point and device information is taken from apt.conf\n"
 218       "and /etc/fstab.\n"
 219       "\n"
 220       "Commands:\n"
 221       "   add - Add a CDROM\n"
 222       "   ident - Report the identity of a CDROM\n"
 223       "\n"
 224       "Options:\n"
 225       "  -h   This help text\n"
 226       "  -d   CD-ROM mount point\n"
 227       "  -r   Rename a recognized CD-ROM\n"
 228       "  -m   No mounting\n"
 229       "  -f   Fast mode, don't check package files\n"
 230       "  -a   Thorough scan mode\n"
 231       "  --no-auto-detect Do not try to auto detect drive and mount point\n"
 232       "  -c=? Read this configuration file\n"
 233       "  -o=? Set an arbitrary configuration option, eg -o dir::cache=/tmp\n"
 234       "See fstab(5)\n";
 235    return 0;
 236 }
 237                                                                         /*}}}*/
 238 int main(int argc,const char *argv[])                                   /*{{{*/
 239 {
 240    CommandLine::Args Args[] = {
 241       {'h',"help","help",0},
 242       {  0,"auto-detect","Acquire::cdrom::AutoDetect", CommandLine::Boolean},
 243       {'v',"version","version",0},
 244       {'d',"cdrom","Acquire::cdrom::mount",CommandLine::HasArg},
 245       {'r',"rename","APT::CDROM::Rename",0},
 246       {'m',"no-mount","APT::CDROM::NoMount",0},
 247       {'f',"fast","APT::CDROM::Fast",0},
 248       {'n',"just-print","APT::CDROM::NoAct",0},
 249       {'n',"recon","APT::CDROM::NoAct",0},
 250       {'n',"no-act","APT::CDROM::NoAct",0},
 251       {'a',"thorough","APT::CDROM::Thorough",0},
 252       {'c',"config-file",0,CommandLine::ConfigFile},
 253       {'o',"option",0,CommandLine::ArbItem},
 254       {0,0,0,0}};
 255    CommandLine::Dispatch Cmds[] = {
 256       {"add",&DoAdd},
 257       {"ident",&DoIdent},
 258       {0,0}};
 259
 260    // Set up gettext support
 261    setlocale(LC_ALL,"");
 262    textdomain(PACKAGE);
 263
 264    // Parse the command line and initialize the package library
 265    CommandLine CmdL(Args,_config);
 266    if (pkgInitConfig(*_config) == false ||
 267        CmdL.Parse(argc,argv) == false ||
 268        pkgInitSystem(*_config,_system) == false)
 269    {
 270       _error->DumpErrors();
 271       return 100;
 272    }
 273
 274    // See if the help should be shown
 275    if (_config->FindB("help") == true || _config->FindB("version") == true ||
 276        CmdL.FileSize() == 0)
 277       return ShowHelp();
 278
 279    // Deal with stdout not being a tty
 280    if (isatty(STDOUT_FILENO) && _config->FindI("quiet", -1) == -1)
 281       _config->Set("quiet","1");
 282
 283    // Match the operation
 284    CmdL.DispatchArg(Cmds);
 285
 286    // Print any errors or warnings found during parsing
 287    bool const Errors = _error->PendingError();
 288    if (_config->FindI("quiet",0) > 0)
 289       _error->DumpErrors();
 290    else
 291       _error->DumpErrors(GlobalError::DEBUG);
 292    return Errors == true ? 100 : 0;
 293 }
 294                                                                         /*}}}*/