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       "  -c=? Read this configuration file\n"
 221       "  -o=? Set an arbitrary configuration option, eg -o dir::cache=/tmp\n"
 222       "See fstab(5)\n";
 223    return 0;
 224 }
 225                                                                         /*}}}*/
 226 int main(int argc,const char *argv[])                                   /*{{{*/
 227 {
 228    CommandLine::Args Args[] = {
 229       {'h',"help","help",0},
 230       {'a',"auto-detect","Acquire::cdrom::AutoDetect",0},
 231       {'v',"version","version",0},
 232       {'d',"cdrom","Acquire::cdrom::mount",CommandLine::HasArg},
 233       {'r',"rename","APT::CDROM::Rename",0},
 234       {'m',"no-mount","APT::CDROM::NoMount",0},
 235       {'f',"fast","APT::CDROM::Fast",0},
 236       {'n',"just-print","APT::CDROM::NoAct",0},
 237       {'n',"recon","APT::CDROM::NoAct",0},
 238       {'n',"no-act","APT::CDROM::NoAct",0},
 239       {'a',"thorough","APT::CDROM::Thorough",0},
 240       {'c',"config-file",0,CommandLine::ConfigFile},
 241       {'o',"option",0,CommandLine::ArbItem},
 242       {0,0,0,0}};
 243    CommandLine::Dispatch Cmds[] = {
 244       {"add",&DoAdd},
 245       {"ident",&DoIdent},
 246       {0,0}};
 247
 248    // Set up gettext support
 249    setlocale(LC_ALL,"");
 250    textdomain(PACKAGE);
 251
 252    // Parse the command line and initialize the package library
 253    CommandLine CmdL(Args,_config);
 254    if (pkgInitConfig(*_config) == false ||
 255        CmdL.Parse(argc,argv) == false ||
 256        pkgInitSystem(*_config,_system) == false)
 257    {
 258       _error->DumpErrors();
 259       return 100;
 260    }
 261
 262    // See if the help should be shown
 263    if (_config->FindB("help") == true || _config->FindB("version") == true ||
 264        CmdL.FileSize() == 0)
 265       return ShowHelp();
 266
 267    // Deal with stdout not being a tty
 268    if (isatty(STDOUT_FILENO) && _config->FindI("quiet",0) < 1)
 269       _config->Set("quiet","1");
 270
 271    // Match the operation
 272    CmdL.DispatchArg(Cmds);
 273
 274    // Print any errors or warnings found during parsing
 275    if (_error->empty() == false)
 276    {
 277       bool Errors = _error->PendingError();
 278       _error->DumpErrors();
 279       return Errors == true?100:0;
 280    }
 281
 282    return 0;
 283 }
 284                                                                         /*}}}*/