add -nocheck command line option

2 files changed, 60 insertions, 42 deletions

diff --git a/noncore/net/wellenreiter/ChangeLog b/noncore/net/wellenreiter/ChangeLog
index 55e13fc..755ed76 100644
--- a/noncore/net/wellenreiter/ChangeLog
+++ b/noncore/net/wellenreiter/ChangeLog
@@ -2,2 +2,3 @@
 
+	* Added command line option "-nocheck" to skip non-root and dhcp tests
 	* Improved the speed reading the manufacturer list
diff --git a/noncore/net/wellenreiter/gui/main.cpp b/noncore/net/wellenreiter/gui/main.cpp
index 55a5260..dd757b5 100644
--- a/noncore/net/wellenreiter/gui/main.cpp
+++ b/noncore/net/wellenreiter/gui/main.cpp
@@ -54,45 +54,58 @@ int main( int argc, char **argv )
     int result = -1;
+    static int killed = false;
 
-    // root check
-    if ( getuid() )
+    bool check = true;
+    for ( int i = 1; i < argc; ++i )
     {
-        qWarning( QObject::tr( "Wellenreiter: trying to run as non-root!" ) );
-        result = QMessageBox::warning( w, " - Wellenreiter II -  (non-root)", QObject::tr( "You have started Wellenreiter II\n"
-          "as non-root. You will have\nonly limited functionality.\nProceed anyway?" ),
-          QMessageBox::Yes, QMessageBox::No );
-        if ( result == QMessageBox::No ) return -1;
+        if ( !strcmp( "-nocheck", argv[i] ) )
+        {
+            qDebug( "-nocheck found" );
+            check = false;
+            break;
+        }
     }
 
-    // dhcp check - NOT HERE! This really belongs as a static member to OProcess
-    // and I want to call it like that: if ( OProcess::isRunning( QString& ) ) ...
+    if ( check )
+    {
+        // root check
+        if ( getuid() )
+        {
+            qWarning( QObject::tr( "Wellenreiter: trying to run as non-root!" ) );
+            result = QMessageBox::warning( w, " - Wellenreiter II -  (non-root)", QObject::tr( "You have started Wellenreiter II\n"
+            "as non-root. You will have\nonly limited functionality.\nProceed anyway?" ),
+            QMessageBox::Yes, QMessageBox::No );
+            if ( result == QMessageBox::No ) return -1;
+        }
 
-    static int killed = false;
+        // dhcp check - NOT HERE! This really belongs as a static member to OProcess
+        // and I want to call it like that: if ( OProcess::isRunning( QString& ) ) ...
 
-    QString line;
-    QDir d = QDir( "/proc" );
-    QStringList dirs = d.entryList( QDir::Dirs );
-    QStringList::Iterator it;
-    for ( it = dirs.begin(); it != dirs.end(); ++it )
-    {
-        //qDebug( "next entry: %s", (const char*) *it );
-        QFile file( "/proc/"+*it+"/cmdline" );
-        file.open( IO_ReadOnly );
-        if ( !file.isOpen() ) continue;
-        QTextStream t( &file );
-        line = t.readLine();
-        //qDebug( "cmdline = %s", (const char*) line );
-        if ( line.contains( "dhcp" ) ) break;
-    }
-    if ( line.contains( "dhcp" ) )
-    {
-        qWarning( "Wellenreiter: found dhcp process #%d", (*it).toInt() );
-        result = QMessageBox::warning( w, " - Wellenreiter II -  (dhcp)", QObject::tr( "You have a dhcp client running.\n"
-          "This can severly limit scanning!\nShould I kill it for you?" ),
-          QMessageBox::Yes, QMessageBox::No );
-        if ( result == QMessageBox::Yes )
+        QString line;
+        QDir d = QDir( "/proc" );
+        QStringList dirs = d.entryList( QDir::Dirs );
+        QStringList::Iterator it;
+        for ( it = dirs.begin(); it != dirs.end(); ++it )
+        {
+            //qDebug( "next entry: %s", (const char*) *it );
+            QFile file( "/proc/"+*it+"/cmdline" );
+            file.open( IO_ReadOnly );
+            if ( !file.isOpen() ) continue;
+            QTextStream t( &file );
+            line = t.readLine();
+            //qDebug( "cmdline = %s", (const char*) line );
+            if ( line.contains( "dhcp" ) ) break;
+        }
+        if ( line.contains( "dhcp" ) )
         {
-            if ( -1 == ::kill( (*it).toInt(), SIGTERM ) )
-                qWarning( "Wellenreiter: can't kill process #%d (%s)", result, strerror( errno ) );
-            else
-                killed = true;
+            qWarning( "Wellenreiter: found dhcp process #%d", (*it).toInt() );
+            result = QMessageBox::warning( w, " - Wellenreiter II -  (dhcp)", QObject::tr( "You have a dhcp client running.\n"
+            "This can severly limit scanning!\nShould I kill it for you?" ),
+            QMessageBox::Yes, QMessageBox::No );
+            if ( result == QMessageBox::Yes )
+            {
+                if ( -1 == ::kill( (*it).toInt(), SIGTERM ) )
+                    qWarning( "Wellenreiter: can't kill process #%d (%s)", result, strerror( errno ) );
+                else
+                    killed = true;
+            }
         }
@@ -102,12 +115,16 @@ int main( int argc, char **argv )
 
-    if ( killed )
+    if ( check )
     {
-        result = QMessageBox::warning( w, " - Wellenreiter II -  (dhcp)", QObject::tr( "Restart your dhcp client?" ), QMessageBox::Yes, QMessageBox::No );
-        if ( result == QMessageBox::Yes )
+
+        if ( killed )
         {
-            system( QString().sprintf( "dhclient &; udhcpcd &; dhcpcd &" ) );
+            result = QMessageBox::warning( w, " - Wellenreiter II -  (dhcp)", QObject::tr( "Restart your dhcp client?" ), QMessageBox::Yes, QMessageBox::No );
+            if ( result == QMessageBox::Yes )
+            {
+                system( QString().sprintf( "dhclient &; udhcpcd &; dhcpcd &" ) );
+            }
         }
-    }
 
-    delete w;
+        delete w;
+    }
     return 0;