-rw-r--r-- | noncore/net/opie-smb/qsmb.cpp | 80 |
1 files changed, 25 insertions, 55 deletions
diff --git a/noncore/net/opie-smb/qsmb.cpp b/noncore/net/opie-smb/qsmb.cpp index d6d1b23..c1ee9a9 100644 --- a/noncore/net/opie-smb/qsmb.cpp +++ b/noncore/net/opie-smb/qsmb.cpp @@ -1,270 +1,240 @@ #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 <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 <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(BtnScan, SIGNAL(clicked()), this, SLOT(scanClicked())); connect(BtnClear, SIGNAL(clicked()), this, SLOT(clear())); mountpt->insertItem("/mnt/samba1",-1); mountpt->insertItem("/mnt/samba2",-1); mountpt->insertItem("/mnt/samba3",-1); top_element = NULL; scanning = false; } Qsmb::~Qsmb() { } void Qsmb::clear() { if (scanning) return; ListViewScan->clear(); TextViewOutput->setText(""); 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() { int i; if (scanning) return; scanning = true; QString match; - char lmhosts[512]; QString cmd; - char result[256]; - - FILE *pipe, *pipe2; - LScan->setText("Scanning..."); qApp->processEvents(); sockaddr_in my_addr; get_myaddress( &my_addr); QString ip = inet_ntoa( my_addr.sin_addr); owarn << "IP Address : " << ip<< oendl; match = ip.left(5); - cmd = "smbfind"; - owarn <<"cmd: " << cmd << oendl; - - TextViewOutput->append(cmd); - - snprintf(lmhosts, 512, "echo '127.0.0.1 localhost'>/etc/samba/lmhosts"); - - if ((pipe2 = popen(lmhosts, "r")) == NULL) { - snprintf(result, 256, "Error: Can't run %s", lmhosts); - //TextViewOutput->append(result); - return; - } - - /* run command & read output */ - if ((pipe = popen(cmd.latin1(), "r")) == NULL) { - snprintf(result, 256, "Error: Can't run %s", cmd.latin1()); - TextViewOutput->append(result); - return; - } + 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 */ - while(fgets(result, 256, pipe) != NULL) { - /* put result into TextViewOutput */ - TextViewOutput->append(result); - - if( strstr(result, match.latin1()) != NULL ) { - char ip_addr[256], host[256], *ptr1; - - strcpy( ip_addr, result); - ptr1 = strchr(ip_addr,' '); - strcpy( host, ptr1); - ip_addr[ptr1 - ip_addr]='\0'; - - for(i = 0; i < 256; i++) { - if(host[i]!=' ') { - strcpy( host, host + i); - break; - } - } - ptr1 = strchr(host,' '); - host[ptr1 - host] = '\0'; - - owarn << "add host: " << host << oendl; - + ccmd = "smbfind"; + owarn <<"cmd: " << ccmd << oendl; + 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); - snprintf( lmhosts, 512, "echo '%s %s'>>/etc/samba/lmhosts", ip_addr,host); - - owarn << "lmhosts: " << lmhosts << oendl; - - if ((pipe2 = popen(lmhosts, "r")) == NULL) { - snprintf(result, 256, "Error: Can't run %s", lmhosts); - return; - } + lms << ip_addr+" "+host+"\n"; } } + lmhosts.close(); TextViewOutput->append("\n\n============================================\n"); LScan->setText(""); scanning = false; } void Qsmb::hostSelected(int index) { owarn << "hostSelected" << oendl; int i; QListViewItem *element; QListViewItem *parent; QString text = CBHost->currentText(); if (scanning) return; scanning = true; QString cmd; char result[256]; FILE *pipe; LScan->setText("Scanning..."); if(username->text().isEmpty()) cmd = "/usr/bin/smbclient -L //"+CBHost->currentText()+" -N 2>&1 |grep Disk"; else cmd = "/usr/bin/smbclient -L //"+CBHost->currentText()+" -N -U "+username->text()+"\%"+password->text()+" 2>&1 |grep Disk"; // for(i = 0; i < 512; i++) { // if(cmd[i]==':') { // cmd[i]='%'; // break; // } // if(cmd[i]=='\0') // break; // } 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; } |