Klamscan: handle block devices correctly.

This resolves issue #12.

Signed-off-by: Mavridis Philippe <mavridisf@gmail.com>

1 files changed, 21 insertions, 1 deletions

diff --git a/src/klamscan.cpp b/src/klamscan.cpp
index 57245c4..0015088 100644
--- a/src/klamscan.cpp
+++ b/src/klamscan.cpp
@@ -33,6 +33,9 @@
 #include <dcopref.h>
 #include <dcopclient.h>
 
+
+#include <sys/stat.h> // TDEIO does not seem to have what we need
+
 using namespace KlamAV;
 
 Klamscan::Klamscan(TQWidget *parent, const char *name)
@@ -477,9 +480,26 @@ TQStringList Klamscan::pruneSelectedDirs(){
     listOfUrlsToScan.sort();
     TQString prev;
     TQStringList prevdirs;
+    struct stat sb;
     for (TQStringList::Iterator it = listOfUrlsToScan.begin(); it != listOfUrlsToScan.end(); it++ ){
         //kdDebug() << "dir: " << (*it) << endl;
-        (*it) = (*it).stripWhiteSpace() + "/";
+        (*it) = (*it).stripWhiteSpace();
+
+	// replace block devices with mountpoints
+	lstat( (*it), &sb );
+	if ( (sb.st_mode & S_IFMT) == S_IFBLK ) {
+		// This is actually from directorylist.cpp
+		DCOPRef mediamanager("kded", "mediamanager");
+		DCOPReply reply = mediamanager.call( "properties", (*it) );
+
+		TQStringList properties;
+		reply.get( properties, "TQStringList" );
+
+		(*it) = * (properties.at(7) );
+	} else {
+		(*it) = (*it) + "/";
+	}
+
         if (prevdirs.isEmpty()){
             //kdDebug() << (*it) << endl;
             filepattern.append(*it);