Просмотр исходного кода

prototype for libudev dlopen() type of cdrom detection

Michael Vogt лет назад: 17
Родитель
Сommit
8e372e79de
2 измененных файлов с 62 добавлено и 5 удалено
  1. 61 4
      methods/cdrom.cc
  2. 1 1
      methods/makefile

+ 61 - 4
methods/cdrom.cc

@@ -17,6 +17,7 @@
 
 #include <sys/stat.h>
 #include <unistd.h>
+#include <dlfcn.h>
 
 #include <iostream>
 #include <apti18n.h>
@@ -24,6 +25,23 @@
 
 using namespace std;
 
+struct udev;
+struct udev_list_entry;
+
+// libudev dlopen stucture
+struct udev_p {
+      struct udev* (*udev_new)(void);
+      int (*udev_enumerate_add_match_property)(struct udev_enumerate *udev_enumerate, const char *property, const char *value);
+      int (*udev_enumerate_scan_devices)(struct udev_enumerate *udev_enumerate);
+      struct udev_list_entry *(*udev_enumerate_get_list_entry)(struct udev_enumerate *udev_enumerate);
+      struct udev_device *(*udev_device_new_from_syspath)(struct udev *udev, const char *syspath);
+      struct udev *(*udev_enumerate_get_udev)(struct udev_enumerate *udev_enumerate);
+      const char *(*udev_list_entry_get_name)(struct udev_list_entry *list_entry);
+      const char *(*udev_device_get_devnode)(struct udev_device *udev_device);
+      struct udev_enumerate *(*udev_enumerate_new) (struct udev *udev);
+      struct udev_list_entry *(*udev_list_entry_get_next)(struct udev_list_entry *list_entry);
+};
+
 class CDROMMethod : public pkgAcqMethod
 {
    bool DatabaseLoaded;
@@ -31,11 +49,12 @@ class CDROMMethod : public pkgAcqMethod
    string CurrentID;
    string CDROM;
    bool MountedByApt;
-   
+   vector<string> CdromDevices;
+ 
    virtual bool Fetch(FetchItem *Itm);
    string GetID(string Name);
    virtual void Exit();
-   
+      
    public:
    
    CDROMMethod();
@@ -50,14 +69,52 @@ CDROMMethod::CDROMMethod() : pkgAcqMethod("1.0",SingleInstance | LocalOnly |
                                           DatabaseLoaded(false), 
                                           MountedByApt(false)
 {
+   // see if we can get libudev
+   void *h = dlopen("libudev.so.0", RTLD_LAZY);
+   if (h) {
+      // the pointers for the udev struct
+      struct udev_p p;
+      p.udev_new = (udev* (*)(void)) dlsym(h, "udev_new");
+      p.udev_enumerate_add_match_property = (int (*)(udev_enumerate*, const char*, const char*))dlsym(h, "udev_enumerate_add_match_property");
+      p.udev_enumerate_scan_devices = (int (*)(udev_enumerate*))dlsym(h, "udev_enumerate_scan_devices");
+      p.udev_enumerate_get_list_entry = (udev_list_entry* (*)(udev_enumerate*))dlsym(h, "udev_enumerate_get_list_entry");
+      p.udev_device_new_from_syspath = (udev_device* (*)(udev*, const char*))dlsym(h, "udev_device_new_from_syspath");
+      p.udev_enumerate_get_udev = (udev* (*)(udev_enumerate*))dlsym(h, "udev_enumerate_get_udev");
+      p.udev_list_entry_get_name = (const char* (*)(udev_list_entry*))dlsym(h, "udev_list_entry_get_name");
+      p.udev_device_get_devnode = (const char* (*)(udev_device*))dlsym(h, "udev_device_get_devnode");
+      p.udev_enumerate_new = (udev_enumerate* (*)(udev*))dlsym(h, "udev_enumerate_new");
+      p.udev_list_entry_get_next = (udev_list_entry* (*)(udev_list_entry*))dlsym(h, "udev_list_entry_get_next");
+      struct udev_enumerate *enumerate;
+      struct udev_list_entry *l, *devices;
+      struct udev *udev_ctx;
+
+      udev_ctx = p.udev_new();
+      enumerate = p.udev_enumerate_new (udev_ctx);
+      p.udev_enumerate_add_match_property(enumerate, "ID_CDROM", "1");
+
+      p.udev_enumerate_scan_devices (enumerate);
+      devices = p.udev_enumerate_get_list_entry (enumerate);
+      for (l = devices; l != NULL; l = p.udev_list_entry_get_next (l))
+      {
+	 struct udev_device *udevice;
+	 udevice = p.udev_device_new_from_syspath (p.udev_enumerate_get_udev (enumerate), p.udev_list_entry_get_name (l));
+	 if (udevice == NULL)
+	    continue;
+	 const char* devnode = p.udev_device_get_devnode(udevice);
+	 //std::cerr << devnode  << std::endl;
+	 CdromDevices.push_back(string(devnode));
+      } 
+   }
+
+
 };
 									/*}}}*/
 // CDROMMethod::Exit - Unmount the disc if necessary			/*{{{*/
 // ---------------------------------------------------------------------
 /* */
 void CDROMMethod::Exit()
-{
-   if (MountedByApt == true)
+{ 
+  if (MountedByApt == true)
       UnmountCdrom(CDROM);
 }
 									/*}}}*/

+ 1 - 1
methods/makefile

@@ -40,7 +40,7 @@ include $(PROGRAM_H)
 
 # The cdrom method
 PROGRAM=cdrom
-SLIBS = -lapt-pkg $(INTLLIBS)
+SLIBS = -lapt-pkg -ldl $(INTLLIBS)
 LIB_MAKES = apt-pkg/makefile
 SOURCE = cdrom.cc
 include $(PROGRAM_H)