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