add current mount when selecting server

2 files changed, 28 insertions, 6 deletions

diff --git a/noncore/net/opie-smb/qsmb.cpp b/noncore/net/opie-smb/qsmb.cpp
index e2ddc7f..7f3ae89 100644
--- a/noncore/net/opie-smb/qsmb.cpp
+++ b/noncore/net/opie-smb/qsmb.cpp
@@ -1,416 +1,437 @@
 #include "qsmb.h"
 #include <qpushbutton.h>
 #include <qpe/qpeapplication.h>
 
 #include <string.h>
 #include <qstring.h>
 #include <qstringlist.h>
 #include <qdir.h>
 #include <qfileinfo.h>
 #include <qtabwidget.h>
 
 #include <qpe/process.h>
 #include <qlabel.h>
 #include <qlineedit.h>
 #include <qcombobox.h>
 #include <qstringlist.h>
 #include <qcheckbox.h>
 #include <qtextview.h>
 #include <qmessagebox.h>
 #include <qtextstream.h>
 
 
 #include <pthread.h>
 #include <signal.h>
 #include <ctype.h>
 
 
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <rpc/clnt.h>
 
 #include <sys/vfs.h>
 #include <mntent.h>
 
 // #include <opie2/odebug.h>
 // using namespace Opie::Core;
 
 
 Qsmb::Qsmb( QWidget* parent,  const char* name, WFlags fl )
    : FormQPESMBBase( parent, name, fl )
 {
    connect(CBHost, SIGNAL(activated(int)), this, SLOT(hostSelected(int)));
    connect(DoItBtn, SIGNAL(clicked()), this, SLOT(DoItClicked()));
    connect(UnmountBtn, SIGNAL(clicked()), this, SLOT(umountIt()));
    connect(BtnScan, SIGNAL(clicked()), this, SLOT(scanClicked()));
    connect(BtnClear, SIGNAL(clicked()), this, SLOT(clear()));
 
    mountpt->setEditable(true);
    mountpt->insertItem("/mnt/samba1",-1);
    mountpt->insertItem("/mnt/samba2",-1);
    mountpt->insertItem("/mnt/samba3",-1);
    
    setTabOrder(BtnScan, username);
    setTabOrder(username, password);
    setTabOrder(password, CBHost);
    setTabOrder(CBHost, TextViewOutput);
    setTabOrder(TextViewOutput, mountpt);
    setTabOrder(mountpt, DoItBtn);
    setTabOrder(DoItBtn, UnmountBtn);
    
    top_element = NULL;
    scanning = false;
 }
 
 Qsmb::~Qsmb()
 {
 }
 
 void Qsmb::clear() 
 {
    if (scanning) return;
    ListViewScan->clear();	
    TextViewOutput->setText("");
    CBHost->clear();
    top_element = NULL;	
 }
 
 void Qsmb::scanClicked() 
 {
    if (scanning) return;
    pthread_create(&tpid, NULL, runit, (void *)this);
 }
 
 void Qsmb::DoItClicked() 
 {
 
    if( !ListViewScan->selectedItem()) {
       QMessageBox::warning(this, tr("Error"),tr("<p>No share selected!</p>"));
       return;
    }
    if (scanning) return;
    pthread_create(&tpid, NULL, runitm, (void *)this);
 }
 
 void* runit(void* arg) 
 {
    Qsmb* caller = (Qsmb*)arg;
    caller->scan();
    return(0);
 }
 
 void* runitm(void* arg) 
 {
    Qsmb* caller = (Qsmb*)arg;
    caller->DoIt();
    return(0);
 }
 
 void Qsmb::scan() 
 {
    clear();
 //   if (scanning) return;
    scanning = true;
 
    QString match;
    QString cmd;
    LScan->setText("Scanning...");
    qApp->processEvents();
 
    sockaddr_in my_addr;
    get_myaddress( &my_addr);
 
    QString ip = inet_ntoa( my_addr.sin_addr);
    qWarning("IP Address : "+ip);
 
    match = ip.left(5);
 
    QStringList ccmd;
    TextViewOutput->append("smbfind");
 
    QFile lmhosts("/etc/samba/lmhosts");
    QTextStream lms(&lmhosts);
    lmhosts.open(IO_WriteOnly);
    lms << "127.0.0.1 localhost\n";
     
    /* parse output and display in ListViewScan */
    ccmd = "smbfind";
 	 runCommand(ccmd);
 
    QTextStream s(&out, IO_ReadOnly);
 
    while ( !s.atEnd() ) {
       QString ip_addr, host, output;
       QString tmp = s.readLine();
       bool ok;
       tmp.left(1).toInt( &ok, 10 );
       if(ok) {
          QStringList token = QStringList::split(' ',  tmp );
          ip_addr =  token[0];
          host = token[1];
          CBHost->insertItem( host, -1);
          lms <<  ip_addr+" "+host+"\n";
       }
    }
    lmhosts.close();
 
    TextViewOutput->append("\n\n============================================\n");
    LScan->setText("");
    scanning = false;
 }
 
 void Qsmb::hostSelected(int /*index*/ )
 {
    QListViewItem *element;
 //   QListViewItem *parent;
 
    QString text = CBHost->currentText();
    ListViewScan->clear();
 
    if (scanning) return;
    scanning = true;
 
    QString cmd;
    QStringList ccmd;
 
    LScan->setText("Scanning...");
 
    ccmd << "/usr/bin/smbclient";
    ccmd << "-L";
    ccmd << CBHost->currentText();
    ccmd << "-N";
    
    if(username->text().isEmpty()) {
       //do nothing
    } else {
       ccmd << "-U";
       ccmd << username->text()+"\%"+ password->text();
    }
    runCommand(ccmd);
    QTextStream s(&out, IO_ReadOnly);
 
    while ( !s.atEnd() ) {
       QString share;
       QString comment;
+      QString mount;
       QString tmp = s.readLine();
       
       if( tmp.find("$") == -1 && tmp.find("Disk") != -1) {
          QStringList token = QStringList::split(' ',  tmp );
          share = token[0];
          comment = token[2];
          share = share.stripWhiteSpace();
          comment = comment.stripWhiteSpace();
-         element = new QListViewItem(ListViewScan, share, comment);
+//         if(isMounted(share))
+
+         mount = getMount(share);
+         element = new QListViewItem(ListViewScan, share, comment, mount);
          element->setOpen(true);
 //             top_element = element;
 //             parent = element;
       }
 
    }
 //   owarn << "i="<< index << "cmd:" <<  cmd << oendl;
 
    TextViewOutput->append(cmd);
 
    /* run smbclient & read output */
 //    if ((pipe = popen(cmd.latin1(), "r")) == NULL) {
 //       snprintf(result, 256, "Error: Can't run %s", cmd.latin1());
 // //      cmd = "Error: Can't run "+cmd;
 //       TextViewOutput->append(result);
 //       return;
 //    }
 
    /* parse output and display in ListViewScan */
 //    while(fgets(result, 256, pipe) != NULL) {
 //       /* put result into TextViewOutput */
 //       TextViewOutput->append(result);
 
 //       if( strchr(result, '$') == NULL ) { 
 //          char share[256], *ptr1;
 
 //          strcpy(share,result);
 //          ptr1 = strchr(share,' ');
 //          share[ptr1 - share]='\0';
 
 //          owarn<< "add share: " << share << oendl;
 
 //          if(top_element != NULL) {
 //             bool found = false;
 //             element = top_element;
 
 //             while(element != NULL && !found) {
 //                if(strcmp( element->text(0).ascii(), share)==0) {
 //                   parent = element;
 //                   found = true;
 //                }
 //                element = element->nextSibling();
 //             }
 
 //             if(!found) {
 //                element = new QListViewItem(ListViewScan,share);
 //                element->setOpen(true);
 //                parent=element;
 //             }
 //          } else {
 //             element = new QListViewItem(ListViewScan,share);
 //             element->setOpen(true);
 //             top_element = element;
 //             parent = element;
 //          }
 //       }
 //    }
 
    TextViewOutput->append("\n\n============================================\n");
    LScan->setText("");
    scanning = false;
 }
 
 
 void Qsmb::DoIt()
 {
    
    QListViewItem *element;
    element = ListViewScan->selectedItem();
    if(!element) {
       return;
    }
 
    if (scanning) return;
    scanning = true;
 
    QString mount = mountpt->currentText();
    if(isMounted(mount)) {
       qWarning(mount +" is already mounted");
       TextViewOutput->append(mount +" is already mounted");
       return;
    }
    
    bool noerr = false;
 
    QString share;
    QString cmd;
    QString cmd2;
    QString text = mountpt->currentText();
    QStringList ccmd;
 
    LScan->setText("Mounting...");
    qApp->processEvents();
 
    if( !QFileInfo(text).exists()) {
       ccmd << "mkdir";
       ccmd << "-p";
       ccmd << text;
 
       qWarning( "cmd: "+ ccmd.join(" "));
       runCommand(ccmd);
    }
 
    share = element->text(0);
    qWarning("selected share is "+share);
 
    QString service = CBHost->currentText();
    service = service.stripWhiteSpace();
    if(mount.left(1) != "/")
       mount = QDir::currentDirPath()+"/"+mount;
    mount = mount.stripWhiteSpace();
    ccmd.clear();
    
    ccmd << "/usr/bin/smbmount";
    ccmd << "//"+ service+"/"+share;
    ccmd << mount;
    ccmd << "-o";
    ccmd << "username="+username->text()+",password="+password->text()+"";
 
-   TextViewOutput->append(ccmd.join(" ").latin1());
-
    if(onbootBtn->isChecked()) {
       qWarning("Saving Setting permanently...");
       QFile sambenv("/opt/QtPalmtop/etc/samba.env");
       QTextStream smbv(&sambenv);
       sambenv.open(IO_WriteOnly);
       smbv << ccmd.join(" ") ;
       sambenv.close();
    }
 
    noerr = runCommand(ccmd);
    
    LScan->setText("");
 
-   if(noerr) {
+   if(noerr && isMounted(mount)) {
       element->setText(2, mount);
       TextViewOutput->append("\n\n================CheckMounts==================\n");
       ccmd = "/bin/mount";
       runCommand(ccmd);
       TextViewOutput->append("\n\n============================================\n");
       qApp->processEvents();
    } else {
    //do nothing
  }
 
    scanning = false;
 }
 
 void Qsmb::umountIt()
 {
    QString mount = mountpt->currentText();
    if(!isMounted(mount)) {
       qWarning(mount +" is not mounted");
       TextViewOutput->append(mount +" is not mounted");
       return;
    }
 
    QStringList ccmd;
    QString share;
    QListViewItem *element;
    element = ListViewScan->selectedItem();
    share = element->text(0);
    qWarning("selected share is "+share);
 
    if(mount.left(1) != "/")
       mount = QDir::currentDirPath()+"/"+mount;
    mount = mount.stripWhiteSpace();
 
    ccmd << "/usr/bin/smbumount";
    ccmd << mount;
    runCommand(ccmd);
 
    element->setText(2, "");
 
    ccmd = "/bin/mount";
    runCommand(ccmd);
 }
 
 bool Qsmb::runCommand(const QStringList & command) {
    qWarning( "runCommand " + command.join(" ") );
    TextViewOutput->append(command.join(" "));
    out = "";
    Process ipkg_status( command);
    bool r = ipkg_status.exec("",out);
 
    qWarning("result is %d"+ r );
    qWarning("Output " + out );
    TextViewOutput->append(out);
 
 //very hacky
    if(out.find("failed") !=-1) {
       r = false;
    }
    return r;
 }
 
 
-bool Qsmb::isMounted(const QString &mountPoint) {
+bool Qsmb::isMounted(const QString &mountPoint)
+{
     struct mntent *me;
     bool mounted = false;
     FILE *mntfp = setmntent( "/etc/mtab", "r" );
     if ( mntfp ){
         while ( (me = getmntent( mntfp )) != 0 ) {
             QString deviceName = me->mnt_fsname;
             QString mountDir = me->mnt_dir;
             QString fsType = me->mnt_type;
-            if( fsType == "smbfs" && mountDir.find(mountPoint) !=-1)
+            if( fsType == "smbfs" && (mountDir.find(mountPoint) != -1 | deviceName.find(mountPoint) != -1))
                mounted = true;
         }
     }
     endmntent( mntfp );
     return mounted;
 }
+
+QString Qsmb::getMount(const QString &shareName)
+{
+   struct mntent *me;
+    QString mount;
+    FILE *mntfp = setmntent( "/etc/mtab", "r" );
+    if ( mntfp ){
+        while ( (me = getmntent( mntfp )) != 0 ) {
+            QString deviceName = me->mnt_fsname;
+            QString mountDir = me->mnt_dir;
+            QString fsType = me->mnt_type;
+            if( fsType == "smbfs" &&  deviceName.find(shareName) != -1)
+               mount = mountDir;
+        }
+    }
+    endmntent( mntfp );
+    return mount;
+}
diff --git a/noncore/net/opie-smb/qsmb.h b/noncore/net/opie-smb/qsmb.h
index fba94d0..abf27da 100644
--- a/noncore/net/opie-smb/qsmb.h
+++ b/noncore/net/opie-smb/qsmb.h
@@ -1,44 +1,45 @@
 #ifndef QSMB_H
 #define QSMB_H
 
 #include "qsmbbase.h"
 
 #include <stdlib.h>
 #include <stdlib.h>
 #include <stdio.h>
 
 #include <qlistview.h>
 #include <pthread.h>
 
 class Qsmb : public FormQPESMBBase
 { 
    Q_OBJECT
 
 public:
    static QString appName() { return QString::fromLatin1("opie-smb"); }
    Qsmb( QWidget* parent = 0, const char* name = 0, WFlags fl = 0 );
    ~Qsmb();
    void scan();
    void DoIt();
 
 private:
    QListViewItem *top_element;
    QComboBox *hosts;
    pthread_t tpid;
    bool scanning;
    bool isMounted(const QString &);
+   QString getMount(const QString &);
 
 public slots:
    void clear();
 	 void scanClicked();
    void hostSelected(int);
    void DoItClicked();
    void umountIt();
    QString out;
    bool runCommand(const QStringList &);
 
 };
 void* runit(void *arg);
 void* runitm(void *arg);
 
 #endif // QSMB_H