| 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", true); |
| 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", -1) == -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 | bool const Errors = _error->PendingError(); |
| 277 | if (_config->FindI("quiet",0) > 0) |
| 278 | _error->DumpErrors(); |
| 279 | else |
| 280 | _error->DumpErrors(GlobalError::DEBUG); |
| 281 | return Errors == true ? 100 : 0; |
| 282 | } |
| 283 | /*}}}*/ |