diff options
-rw-r--r-- | tdecore/tdehardwaredevices.cpp | 83 | ||||
-rw-r--r-- | tdecore/tdehardwaredevices.h | 13 |
2 files changed, 55 insertions, 41 deletions
diff --git a/tdecore/tdehardwaredevices.cpp b/tdecore/tdehardwaredevices.cpp index cc1c90357..64809d474 100644 --- a/tdecore/tdehardwaredevices.cpp +++ b/tdecore/tdehardwaredevices.cpp @@ -21,6 +21,7 @@ #include <tqfile.h> #include <tqdir.h> #include <tqstringlist.h> +#include <tqsocketnotifier.h> #include <kglobal.h> #include <ktempfile.h> @@ -382,14 +383,13 @@ TDEHardwareDevices::TDEHardwareDevices() { udev_monitor_filter_add_match_subsystem_devtype(m_udevMonitorStruct, NULL, NULL); udev_monitor_enable_receiving(m_udevMonitorStruct); - m_devScanTimer = new TQTimer(); - connect( m_devScanTimer, TQT_SIGNAL(timeout()), this, TQT_SLOT(checkForHotPluggedHardware()) ); - m_devScanTimer->start(1, TRUE); + m_devScanNotifier = new TQSocketNotifier(udev_monitor_get_fd(m_udevMonitorStruct), TQSocketNotifier::Read, this); + connect( m_devScanNotifier, TQT_SIGNAL(activated(int)), this, TQT_SLOT(processHotPluggedHardware()) ); // Monitor for changed mounts - m_mountScanTimer = new TQTimer(); - connect( m_mountScanTimer, TQT_SIGNAL(timeout()), this, TQT_SLOT(checkForModifiedMounts()) ); - m_mountScanTimer->start(1, TRUE); + m_procMountsFd = open("/proc/mounts", O_RDONLY, 0); + m_mountScanNotifier = new TQSocketNotifier(m_procMountsFd, TQSocketNotifier::Exception, this); + connect( m_mountScanNotifier, TQT_SIGNAL(activated(int)), this, TQT_SLOT(processModifiedMounts()) ); // Update internal device information queryHardwareInformation(); @@ -397,13 +397,8 @@ TDEHardwareDevices::TDEHardwareDevices() { } TDEHardwareDevices::~TDEHardwareDevices() { - // Stop hardware scanning - m_devScanTimer->stop(); - delete m_devScanTimer; - // Stop mount scanning - m_mountScanTimer->stop(); - delete m_mountScanTimer; + close(m_procMountsFd); // Tear down udev interface udev_unref(m_udevStruct); @@ -434,11 +429,11 @@ TDEStorageDevice* TDEHardwareDevices::findDiskByUID(TQString uid) { return 0; } -void TDEHardwareDevices::checkForHotPluggedHardware() { +void TDEHardwareDevices::processHotPluggedHardware() { udev_device* dev = udev_monitor_receive_device(m_udevMonitorStruct); if (dev) { TQString actionevent(udev_device_get_action(dev)); - if (actionevent == "add") { + if (actionevent == "add") { TDEGenericDevice* device = classifyUnknownDevice(dev); // Make sure this device is not a duplicate @@ -462,36 +457,21 @@ void TDEHardwareDevices::checkForHotPluggedHardware() { TDEGenericDevice *hwdevice; for (hwdevice = m_deviceList.first(); hwdevice; hwdevice = m_deviceList.next()) { if (hwdevice->systemPath() == systempath) { + emit hardwareRemoved(*hwdevice); m_deviceList.remove(hwdevice); - emit hardwareAdded(*hwdevice); break; } } } } - - // Continue scanning for added/removed hardware - m_devScanTimer->start(1, TRUE); } -void TDEHardwareDevices::checkForModifiedMounts() { - int mfd = open("/proc/mounts", O_RDONLY, 0); - struct pollfd pfd; - int rv; - - pfd.fd = mfd; - pfd.events = POLLERR | POLLPRI; - pfd.revents = 0; - while ((rv = poll(&pfd, 1, 5)) >= 0) { - if (pfd.revents & POLLERR) { - emit mountTableModified(); - } - - pfd.revents = 0; - } +void TDEHardwareDevices::processModifiedMounts() { + // FIXME + // Detect what changed between the old mount table and the new one, + // and emit appropriate events - // Continue scanning for changed mounts - m_mountScanTimer->start(1, TRUE); + emit mountTableModified(); } TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { @@ -596,6 +576,13 @@ TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { disktype = disktype | TDEDiskDeviceType::Zip; } + if ((devicevendor.upper() == "APPLE") && (devicemodel.upper().contains("IPOD"))) { + disktype = disktype | TDEDiskDeviceType::MediaDevice; + } + if ((devicevendor.upper() == "SANDISK") && (devicemodel.upper().contains("SANSA"))) { + disktype = disktype | TDEDiskDeviceType::MediaDevice; + } + if (disktypestring.upper() == "FLOPPY") { disktype = disktype | TDEDiskDeviceType::Floppy; } @@ -751,13 +738,37 @@ TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { sdevice->setDiskType(disktype); sdevice->setDiskUUID(diskuuid); - sdevice->setDiskLabel(disklabel); sdevice->setDiskStatus(diskstatus); sdevice->setFileSystemName(filesystemtype); sdevice->setFileSystemUsage(filesystemusage); sdevice->setSlaveDevices(slaveDeviceNodes); sdevice->setHoldingDevices(holdingDeviceNodes); + // Clean up disk label + if ((sdevice->isDiskOfType(TDEDiskDeviceType::CDROM)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::CDRW)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::DVDROM)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::DVDRW)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::BDROM)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::BDRW)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::CDAudio)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::CDVideo)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::DVDVideo)) + || (sdevice->isDiskOfType(TDEDiskDeviceType::BDVideo)) + ) { + if (disklabel == "") { + // Read the volume label in via volname, since udev couldn't be bothered to do this on its own + FILE *exepipe = popen((TQString("volname %1").arg(devicenode).ascii()), "r"); + if (exepipe) { + char buffer[8092]; + disklabel = fgets(buffer, sizeof(buffer), exepipe); + pclose(exepipe); + } + } + } + + sdevice->setDiskLabel(disklabel); + printf("DISK DEVICE name: %s type: %s subsystem: %s vendor: %s model: %s bus: %s label: %s filesystem: %s disk type: %s disk type flags: 0x%08x media inserted: %d [Node Path: %s] [Syspath: %s]\n\r\n\r", devicename.ascii(), devicetype.ascii(), devicesubsystem.ascii(), devicevendor.ascii(), devicemodel.ascii(), devicebus.ascii(), disklabel.ascii(), filesystemtype.ascii(), disktypestring.ascii(), disktype, sdevice->mediaInserted(), devicenode.ascii(), udev_device_get_syspath(dev)); fflush(stdout); } else if (devicetype.isNull()) { diff --git a/tdecore/tdehardwaredevices.h b/tdecore/tdehardwaredevices.h index 11844fa5c..2b145b9e9 100644 --- a/tdecore/tdehardwaredevices.h +++ b/tdecore/tdehardwaredevices.h @@ -19,10 +19,10 @@ #define _TDEHARDWAREDEVICES_H // TDE includes +#include <tqobject.h> #include <tqstring.h> #include <tqptrlist.h> #include <tqstringlist.h> -#include <tqtimer.h> #include "tdelibs_export.h" // udev includes @@ -419,6 +419,8 @@ class TDECORE_EXPORT TDEStorageDevice : public TDEGenericDevice typedef TQPtrList<TDEGenericDevice> TDEGenericHardwareList; +class TQSocketNotifier; + class TDECORE_EXPORT TDEHardwareDevices : TQObject { Q_OBJECT @@ -470,8 +472,8 @@ class TDECORE_EXPORT TDEHardwareDevices : TQObject void mountTableModified(); private slots: - void checkForHotPluggedHardware(); - void checkForModifiedMounts(); + void processHotPluggedHardware(); + void processModifiedMounts(); private: TDEGenericDevice *classifyUnknownDevice(udev_device* dev); @@ -479,9 +481,10 @@ class TDECORE_EXPORT TDEHardwareDevices : TQObject struct udev *m_udevStruct; struct udev_monitor *m_udevMonitorStruct; TDEGenericHardwareList m_deviceList; + int m_procMountsFd; - TQTimer* m_devScanTimer; - TQTimer* m_mountScanTimer; + TQSocketNotifier* m_devScanNotifier; + TQSocketNotifier* m_mountScanNotifier; }; #endif
\ No newline at end of file |