/* This file is part of the KDE project Copyright (c) 2004 Kévin Ottens Parts of this file are Copyright 2003 Waldo Bastian This library is free software; you can redistribute it and/or modify it under the terms of the GNU Library General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License for more details. You should have received a copy of the GNU Library General Public License along with this library; see the file COPYING.LIB. If not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #ifdef __TDE_HAVE_TDEHWLIB #include #endif #include "dialog.h" #include "tdeio_media_mounthelper.h" const Medium MountHelper::findMedium(const TQString &device) { DCOPReply reply = m_mediamanager.call("properties", device); if (!reply.isValid()) { m_errorStr = i18n("The TDE mediamanager is not running.\n"); return Medium(TQString::null, TQString::null, TQString::null); } const Medium &medium = Medium::create(reply); return medium; } void MountHelper::mount(const Medium &medium) { if (medium.id().isEmpty()) { m_errorStr = i18n("Try to mount an unknown medium."); errorAndExit(); } TQString device = medium.deviceNode(); if (!medium.isMountable()) { m_errorStr = i18n("%1 is not a mountable media.").arg(device); errorAndExit(); } else if (medium.isMounted()) { m_errorStr = i18n("%1 is already mounted to %2.").arg(device).arg(medium.mountPoint()); errorAndExit(); } DCOPReply reply = m_mediamanager.call("mount", medium.id()); TQStringVariantMap mountResult; if (reply.isValid()) { reply.get(mountResult); } if (!mountResult.contains("result") || !mountResult["result"].toBool()) { m_errorStr = mountResult.contains("errStr") ? mountResult["errStr"].toString() : i18n("Unknown mount error."); errorAndExit(); } } void MountHelper::unmount(const Medium &medium) { if (medium.id().isEmpty()) { m_errorStr = i18n("Try to unmount an unknown medium."); errorAndExit(); } TQString device = medium.deviceNode(); if (!medium.isMountable()) { m_errorStr = i18n("%1 is not a mountable media.").arg(device); errorAndExit(); } else if (!medium.isMounted()) { m_errorStr = i18n("%1 is already unmounted.").arg(device); errorAndExit(); } DCOPReply reply = m_mediamanager.call("unmount", medium.id()); TQStringVariantMap unmountResult; if (reply.isValid()) { reply.get(unmountResult); } if (!unmountResult.contains("result") || !unmountResult["result"].toBool()) { m_errorStr = unmountResult.contains("errStr") ? unmountResult["errStr"].toString() : i18n("Unknown unmount error."); kdDebug() << "medium unmount " << m_errorStr << endl; errorAndExit(); } } void MountHelper::unlock(const Medium &medium) { if (medium.id().isEmpty()) { m_errorStr = i18n("Try to unlock an unknown medium."); errorAndExit(); } TQString device = medium.deviceNode(); if (!medium.isEncrypted()) { m_errorStr = i18n("%1 is not an encrypted media.").arg(device); errorAndExit(); } if (!medium.needUnlocking()) { m_errorStr = i18n("%1 is already unlocked.").arg(device); errorAndExit(); } TQString iconName = medium.iconName(); if (iconName.isEmpty()) { TQString mime = medium.mimeType(); iconName = KMimeType::mimeType(mime)->icon(mime, false); } m_mediumId = medium.id(); m_dialog = new Dialog(device, iconName); connect(m_dialog, TQT_SIGNAL(user1Clicked()), this, TQT_SLOT(slotSendPassword())); connect(m_dialog, TQT_SIGNAL(cancelClicked()), this, TQT_SLOT(slotCancel())); m_dialog->show(); } void MountHelper::lock(const Medium &medium) { if (medium.id().isEmpty()) { m_errorStr = i18n("Try to lock an unknown medium."); errorAndExit(); } TQString device = medium.deviceNode(); if (!medium.isEncrypted()) { m_errorStr = i18n("%1 is not an encrypted media.").arg(device); errorAndExit(); } if (medium.needUnlocking()) { m_errorStr = i18n("%1 is already locked.").arg(device); errorAndExit(); } // Release children devices releaseHolders(medium); DCOPReply reply = m_mediamanager.call("lock", medium.id()); TQStringVariantMap lockResult; if (reply.isValid()) { reply.get(lockResult); } if (!lockResult.contains("result") || !lockResult["result"].toBool()) { m_errorStr = lockResult.contains("errStr") ? lockResult["errStr"].toString() : i18n("Unknown lock error."); kdDebug() << "medium lock " << m_errorStr << endl; errorAndExit(); } } void MountHelper::eject(const TQString &device, bool quiet) { #ifdef __TDE_HAVE_TDEHWLIB // Try TDE HW library eject first... TDEHardwareDevices *hwdevices = TDEGlobal::hardwareDevices(); TDEGenericDevice *hwdevice = hwdevices->findByDeviceNode(device); if (hwdevice->type() == TDEGenericDeviceType::Disk) { TDEStorageDevice *sdevice = static_cast(hwdevice); TQStringVariantMap ejectResult = sdevice->ejectDrive(); if (ejectResult["result"].toBool() == true) { // Success! ::exit(0); } } #endif // Otherwise fall back to tdeeject TDEProcess *proc = new TDEProcess(TQT_TQOBJECT(this)); *proc << "tdeeject"; if (quiet) { *proc << "-q"; } *proc << device; connect(proc, TQT_SIGNAL(processExited(TDEProcess*)), this, TQT_SLOT(ejectFinished(TDEProcess*))); proc->start(); } void MountHelper::releaseHolders(const Medium &medium, bool handleThis) { #ifdef __TDE_HAVE_TDEHWLIB if (medium.id().isEmpty()) { m_errorStr = i18n("Try to release holders from an unknown medium."); return; } // Scan the holding devices and unmount/lock them if possible TDEHardwareDevices *hwdevices = TDEGlobal::hardwareDevices(); TDEStorageDevice *sdevice = hwdevices->findDiskByUID(medium.id()); if (sdevice) { TQStringList holdingDeviceList = sdevice->holdingDevices(); for (TQStringList::Iterator holdingDevIt = holdingDeviceList.begin(); holdingDevIt != holdingDeviceList.end(); ++holdingDevIt) { TDEGenericDevice *hwHolderDevice = hwdevices->findBySystemPath(*holdingDevIt); if (hwHolderDevice->type() == TDEGenericDeviceType::Disk) { TDEStorageDevice *holderSDevice = static_cast(hwHolderDevice); const Medium holderMedium = findMedium(holderSDevice->deviceNode()); if (!holderMedium.id().isEmpty()) { releaseHolders(holderMedium, true); } } } } if (handleThis) { // Unmount if necessary if (medium.isMountable() && medium.isMounted()) { unmount(medium); } // Lock if necessary. if (medium.isEncrypted() && !medium.isLocked()) { lock(medium); } } #endif } void MountHelper::safeRemoval(const Medium &medium) { /* * Safely remove will performs the following tasks: * 1) release children devices (if tdehw is available) * 2) if the medium is mounted, unmount it * 3) if the medium is encrypted and unlocked, lock it * 4) invoke eject to release the medium. * If any of the above steps fails, the procedure will interrupt and an * error message will be displayed to the user. * * Note: previously eject was invoked also in case of unmount failure. This * could lead to data loss and therefore the behaviour has been changed. * If a user really wants to eject the medium, he needs to either unmount it * first or invoke eject manually. */ if (medium.id().isEmpty()) { m_errorStr = i18n("Try to safe remove an unknown medium."); errorAndExit(); } // Release children devices releaseHolders(medium); TQStringVariantMap opResult; TQString device = medium.deviceNode(); // Unmount if necessary if (medium.isMountable() && medium.isMounted()) { unmount(medium); } // Lock if necessary. if (medium.isEncrypted() && !medium.isLocked()) { lock(medium); } } void MountHelper::openRealFolder(const Medium &medium) { Medium &m = const_cast(medium); if (!m.isMounted()) { // If the medium is not mounted, try mounting it first mount(m); m = findMedium(m.deviceNode()); } if (m.isMounted()) { system((TQString("kfmclient exec 'file://") + m.mountPoint()).local8Bit() + "'"); } else { m_errorStr = i18n("Try to open an unknown medium."); errorAndExit(); } } MountHelper::MountHelper() : TDEApplication(), m_mediamanager("kded", "mediamanager") { TDECmdLineArgs *args = TDECmdLineArgs::parsedArgs(); m_errorStr = TQString::null; const Medium medium = findMedium(args->arg(0)); if (medium.id().isEmpty()) { if (m_errorStr.isEmpty()) { m_errorStr+= i18n("%1 cannot be found.").arg(args->arg(0)); } errorAndExit(); } TQString device = medium.deviceNode(); if (!medium.isMountable() && !medium.isEncrypted() && !args->isSet("e") && !args->isSet("s")) { m_errorStr = i18n("%1 is not a mountable or encrypted media.").arg(device); errorAndExit(); } if (args->isSet("m")) { mount(medium); ::exit(0); } else if (args->isSet("u")) { unmount(medium); ::exit(0); } else if (args->isSet("k")) { unlock(medium); // No call to ::exit() here because this will open up the password dialog // ::exit() is handled in the invoked code. } else if (args->isSet("l")) { lock(medium); ::exit(0); } else if (args->isSet("e")) { eject(device, true); ::exit(0); } else if (args->isSet("s")) { safeRemoval(medium); eject(device, true); ::exit(0); } else if (args->isSet("f")) { openRealFolder(medium); ::exit(0); } else { TDECmdLineArgs::usage(); ::exit(0); } } MountHelper::~MountHelper() { if (m_dialog) { delete m_dialog; } } void MountHelper::ejectFinished(TDEProcess *proc) { //If eject failed, report the error stored in m_errorStr if (proc->normalExit() && proc->exitStatus() == 0) { ::exit(0); } else { if (!m_errorStr.isEmpty()) { errorAndExit(); } ::exit(0); } } void MountHelper::error() { TQString prettyErrorString = m_errorStr; if (m_errorStr.contains("<") && m_errorStr.contains(">")) { if (!m_errorStr.contains("")) { prettyErrorString = TQString("%1").arg(m_errorStr); } } KMessageBox::error(0, prettyErrorString); } void MountHelper::errorAndExit() { error(); ::exit(1); } void MountHelper::slotSendPassword() { DCOPReply reply = m_mediamanager.call("unlock", m_mediumId, m_dialog->getPassword()); TQStringVariantMap unlockResult; if (reply.isValid()) { reply.get(unlockResult); } if (unlockResult.contains("result") && unlockResult["result"].toBool()) { ::exit(0); } else { m_errorStr = unlockResult.contains("errStr") ? unlockResult["errStr"].toString() : i18n("Unknown unlock error."); kdDebug() << "medium unlock " << m_errorStr << endl; error(); } } void MountHelper::slotCancel() { exit(0); } static TDECmdLineOptions options[] = { { "m", I18N_NOOP("Mount given URL"), 0 }, { "u", I18N_NOOP("Unmount given URL"), 0 }, { "k", I18N_NOOP("Unlock given URL"), 0 }, { "l", I18N_NOOP("Lock given URL"), 0 }, { "e", I18N_NOOP("Eject given URL"), 0}, { "s", I18N_NOOP("Safely remove (unmount and eject) given URL"), 0}, { "f", I18N_NOOP("Open real medium folder"), 0}, {"!+URL", I18N_NOOP("media:/URL to mount/unmount/unlock/lock/eject/remove"), 0 }, TDECmdLineLastOption }; int main(int argc, char **argv) { TDECmdLineArgs::init(argc, argv, "tdeio_media_mounthelper", "tdeio_media_mounthelper", "tdeio_media_mounthelper", "0.1"); TDECmdLineArgs::addCmdLineOptions(options); TDEGlobal::locale()->setMainCatalogue("tdeio_media"); TDEApplication::addCmdLineOptions(); if (TDECmdLineArgs::parsedArgs()->count()==0) { TDECmdLineArgs::usage(); } TDEApplication *app = new MountHelper(); TDEStartupInfo::appStarted(); app->dcopClient()->attach(); return app->exec(); } #include "tdeio_media_mounthelper.moc"