apt-pkg/cdrom.h

   1 #ifndef PKGLIB_CDROM_H
   2 #define PKGLIB_CDROM_H
   3
   4 #include<string>
   5 #include<vector>
   6
   7 #ifndef APT_8_CLEANER_HEADERS
   8 #include <apt-pkg/init.h>
   9 using namespace std;
  10 #endif
  11
  12 class Configuration;
  13 class OpProgress;
  14
  15 class pkgCdromStatus                                                    /*{{{*/
  16 {
  17  protected:
  18    int totalSteps;
  19
  20  public:
  21    pkgCdromStatus() : totalSteps(0) {};
  22    virtual ~pkgCdromStatus() {};
  23
  24    // total steps
  25    virtual void SetTotal(int total) { totalSteps = total; };
  26    // update steps, will be called regularly as a "pulse"
  27    virtual void Update(std::string text="", int current=0) = 0;
  28
  29    // ask for cdrom insert
  30    virtual bool ChangeCdrom() = 0;
  31    // ask for cdrom name
  32    virtual bool AskCdromName(std::string &Name) = 0;
  33    // Progress indicator for the Index rewriter
  34    virtual OpProgress* GetOpProgress() {return NULL; };
  35 };
  36                                                                         /*}}}*/
  37 class pkgCdrom                                                          /*{{{*/
  38 {
  39  protected:
  40    enum {
  41       STEP_PREPARE = 1,
  42       STEP_UNMOUNT,
  43       STEP_WAIT,
  44       STEP_MOUNT,
  45       STEP_IDENT,
  46       STEP_SCAN,
  47       STEP_COPY,
  48       STEP_WRITE,
  49       STEP_UNMOUNT3,
  50       STEP_LAST
  51    };
  52
  53
  54    bool FindPackages(std::string CD,
  55                      std::vector<std::string> &List,
  56                      std::vector<std::string> &SList,
  57                      std::vector<std::string> &SigList,
  58                      std::vector<std::string> &TransList,
  59                      std::string &InfoDir, pkgCdromStatus *log,
  60                      unsigned int Depth = 0);
  61    bool DropBinaryArch(std::vector<std::string> &List);
  62    bool DropRepeats(std::vector<std::string> &List,const char *Name);
  63    bool DropTranslation(std::vector<std::string> &List);
  64    void ReduceSourcelist(std::string CD,std::vector<std::string> &List);
  65    bool WriteDatabase(Configuration &Cnf);
  66    bool WriteSourceList(std::string Name,std::vector<std::string> &List,bool Source);
  67    int Score(std::string Path);
  68
  69  public:
  70    bool Ident(std::string &ident, pkgCdromStatus *log);
  71    bool Add(pkgCdromStatus *log);
  72
  73  private:
  74    bool MountAndIdentCDROM(Configuration &Database, std::string &CDROM,
  75          std::string &ident, pkgCdromStatus * const log);
  76 };
  77                                                                         /*}}}*/
  78
  79
  80 // class that uses libudev to find cdrom/removable devices dynamically
  81 struct CdromDevice                                                      /*{{{*/
  82 {
  83    std::string DeviceName;
  84    bool Mounted;
  85    std::string MountPath;
  86 };
  87                                                                         /*}}}*/
  88 class pkgUdevCdromDevices                                               /*{{{*/
  89 {
  90  protected:
  91    // libudev dlopen stucture
  92    void *libudev_handle;
  93    struct udev* (*udev_new)(void);
  94    int (*udev_enumerate_add_match_property)(struct udev_enumerate *udev_enumerate, const char *property, const char *value);
  95    int (*udev_enumerate_scan_devices)(struct udev_enumerate *udev_enumerate);
  96    struct udev_list_entry* (*udev_enumerate_get_list_entry)(struct udev_enumerate *udev_enumerate);
  97    struct udev_device* (*udev_device_new_from_syspath)(struct udev *udev, const char *syspath);
  98    struct udev* (*udev_enumerate_get_udev)(struct udev_enumerate *udev_enumerate);
  99    const char* (*udev_list_entry_get_name)(struct udev_list_entry *list_entry);
 100    const char* (*udev_device_get_devnode)(struct udev_device *udev_device);
 101    struct udev_enumerate *(*udev_enumerate_new) (struct udev *udev);
 102    struct udev_list_entry *(*udev_list_entry_get_next)(struct udev_list_entry *list_entry);
 103    const char* (*udev_device_get_property_value)(struct udev_device *udev_device, const char *key);
 104    int (*udev_enumerate_add_match_sysattr)(struct udev_enumerate *udev_enumerate, const char *property, const char *value);
 105    // end libudev dlopen
 106
 107  public:
 108    pkgUdevCdromDevices();
 109    virtual ~pkgUdevCdromDevices();
 110
 111    // try to open
 112    bool Dlopen();
 113
 114    // convenience interface, this will just call ScanForRemovable
 115    // with "APT::cdrom::CdromOnly"
 116    std::vector<CdromDevice> Scan();
 117
 118    std::vector<CdromDevice> ScanForRemovable(bool CdromOnly);
 119 };
 120                                                                         /*}}}*/
 121
 122 #endif