From e428644d86c505207079914f592d5275205493d1 Mon Sep 17 00:00:00 2001 From: Timothy Pearson Date: Fri, 30 Mar 2012 15:44:16 -0500 Subject: Add device mapper functions to TDE hardware library --- tdecore/tdehardwaredevices.cpp | 95 +++++++++++++++++++++++++++++++++++++++--- tdecore/tdehardwaredevices.h | 26 ++++++++++++ 2 files changed, 115 insertions(+), 6 deletions(-) (limited to 'tdecore') diff --git a/tdecore/tdehardwaredevices.cpp b/tdecore/tdehardwaredevices.cpp index 1874c65d7..6124f0173 100644 --- a/tdecore/tdehardwaredevices.cpp +++ b/tdecore/tdehardwaredevices.cpp @@ -19,6 +19,7 @@ #include #include +#include #include #include @@ -152,6 +153,22 @@ void TDEStorageDevice::setDiskUUID(TQString id) { m_diskUUID = id; } +TQStringList &TDEStorageDevice::holdingDevices() { + return m_holdingDevices; +} + +void TDEStorageDevice::setHoldingDevices(TQStringList hd) { + m_slaveDevices = hd; +} + +TQStringList &TDEStorageDevice::slaveDevices() { + return m_slaveDevices; +} + +void TDEStorageDevice::setSlaveDevices(TQStringList sd) { + m_slaveDevices = sd; +} + TQString TDEStorageDevice::mountPath() { // See if this device node is mounted // This requires parsing /proc/mounts, looking for deviceNode() @@ -305,6 +322,47 @@ TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { } fclose(fp); } + + // See if any other devices are exclusively using this device, such as the Device Mapper| + TQStringList holdingDeviceNodes; + TQString holdersnodename = udev_device_get_syspath(dev); + holdersnodename.append("/holders/"); + TQDir holdersdir(holdersnodename); + holdersdir.setFilter(TQDir::All); + const TQFileInfoList *dirlist = holdersdir.entryInfoList(); + if (dirlist) { + TQFileInfoListIterator holdersdirit(*dirlist); + TQFileInfo *dirfi; + while ( (dirfi = holdersdirit.current()) != 0 ) { + if (dirfi->isSymLink()) { + char* collapsedPath = realpath((holdersnodename + dirfi->readLink()).ascii(), NULL); + holdingDeviceNodes.append(TQString(collapsedPath)); + free(collapsedPath); + } + ++holdersdirit; + } + } + + // See if any other physical devices underlie this device, for example when the Device Mapper is in use + TQStringList slaveDeviceNodes; + TQString slavesnodename = udev_device_get_syspath(dev); + slavesnodename.append("/slaves/"); + TQDir slavedir(slavesnodename); + slavedir.setFilter(TQDir::All); + dirlist = slavedir.entryInfoList(); + if (dirlist) { + TQFileInfoListIterator slavedirit(*dirlist); + TQFileInfo *dirfi; + while ( (dirfi = slavedirit.current()) != 0 ) { + if (dirfi->isSymLink()) { + char* collapsedPath = realpath((slavesnodename + dirfi->readLink()).ascii(), NULL); + slaveDeviceNodes.append(TQString(collapsedPath)); + free(collapsedPath); + } + ++slavedirit; + } + } + device = new TDEStorageDevice(TDEGenericDeviceType::Disk); // Determine generic disk information @@ -432,10 +490,10 @@ TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { diskstatus = diskstatus | TDEDiskDeviceStatus::Removable; } - if (disktypestring.upper() == "CRYPTO_LUKS") { + if (filesystemtype.upper() == "CRYPTO_LUKS") { disktype = disktype | TDEDiskDeviceType::LUKS; } - else if (disktypestring.upper() == "CRYPTO") { + else if (filesystemtype.upper() == "CRYPTO") { disktype = disktype | TDEDiskDeviceType::OtherCrypted; } @@ -452,11 +510,32 @@ TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { if ((!disktypestring.upper().isNull()) && (disktype & TDEDiskDeviceType::HDD)) { diskstatus = diskstatus & ~TDEDiskDeviceStatus::Mountable; } - if (sdevice->mediaInserted()) { - diskstatus = diskstatus | TDEDiskDeviceStatus::Inserted; + if (removable) { + if (sdevice->mediaInserted()) { + diskstatus = diskstatus | TDEDiskDeviceStatus::Inserted; + } + else { + diskstatus = diskstatus & ~TDEDiskDeviceStatus::Mountable; + } } - else { - diskstatus = diskstatus & ~TDEDiskDeviceStatus::Mountable; + + if (holdingDeviceNodes.count() > 0) { + diskstatus = diskstatus | TDEDiskDeviceStatus::UsedByDevice; + } + + if (slaveDeviceNodes.count() > 0) { + diskstatus = diskstatus | TDEDiskDeviceStatus::UsesDevice; + } + + // See if any slaves were crypted + for ( TQStringList::Iterator slaveit = slaveDeviceNodes.begin(); slaveit != slaveDeviceNodes.end(); ++slaveit ) { + struct udev_device *slavedev; + slavedev = udev_device_new_from_syspath(m_udevStruct, (*slaveit).ascii()); + TQString slavediskfstype(udev_device_get_property_value(slavedev, "ID_FS_TYPE")); + if ((slavediskfstype.upper() == "CRYPTO_LUKS") || (slavediskfstype.upper() == "CRYPTO")) { + disktype = disktype | TDEDiskDeviceType::UnlockedCrypt; + } + udev_device_unref(slavedev); } sdevice->setDiskType(disktype); @@ -465,6 +544,8 @@ TDEGenericDevice* TDEHardwareDevices::classifyUnknownDevice(udev_device* dev) { sdevice->setDiskStatus(diskstatus); sdevice->setFileSystemName(filesystemtype); sdevice->setFileSystemUsage(filesystemusage); + sdevice->setSlaveDevices(slaveDeviceNodes); + sdevice->setHoldingDevices(holdingDeviceNodes); 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); } @@ -564,6 +645,8 @@ bool TDEHardwareDevices::queryHardwareInformation() { if (device) { m_deviceList.append(device); } + + udev_device_unref(dev); } // Free the enumerator object diff --git a/tdecore/tdehardwaredevices.h b/tdecore/tdehardwaredevices.h index e56217cbe..531106c36 100644 --- a/tdecore/tdehardwaredevices.h +++ b/tdecore/tdehardwaredevices.h @@ -21,6 +21,7 @@ // TDE includes #include #include +#include #include #include "tdelibs_export.h" @@ -98,6 +99,7 @@ enum TDEDiskDeviceType { MemoryStick = 0x04000000, SmartMedia = 0x08000000, SDMMC = 0x10000000, + UnlockedCrypt = 0x20000000, Other = 0x80000000 }; @@ -123,6 +125,8 @@ enum TDEDiskDeviceStatus { Removable = 0x00000002, Inserted = 0x00000004, Blank = 0x00000008, + UsedByDevice = 0x00000010, + UsesDevice = 0x00000020, Other = 0x80000000 }; @@ -323,6 +327,26 @@ class TDECORE_EXPORT TDEStorageDevice : public TDEGenericDevice */ void setFileSystemUsage(TQString fu); + /** + * @return a TQStringList containing system paths to all devices with a lock on this device, if any + */ + TQStringList &holdingDevices(); + + /** + * @param a TQStringList containing system paths to all devices with a lock on this device, if any + */ + void setHoldingDevices(TQStringList hd); + + /** + * @return a TQStringList containing system paths to all devices locked by this device, if any + */ + TQStringList &slaveDevices(); + + /** + * @param a TQStringList containing system paths to all devices locked by this device, if any + */ + void setSlaveDevices(TQStringList sd); + /** * @return a TQString with the mount path, if mounted */ @@ -337,6 +361,8 @@ class TDECORE_EXPORT TDEStorageDevice : public TDEGenericDevice TQString m_fileSystemUsage; bool m_mediaInserted; TQString m_mountPath; + TQStringList m_holdingDevices; + TQStringList m_slaveDevices; }; typedef TQPtrList TDEGenericHardwareList; -- cgit v1.2.1