You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdelibs/tdeprint/management/smbview.cpp

293 lines
7.9 KiB

/*
* This file is part of the KDE libraries
* Copyright (c) 2001 Michael Goffioul <tdeprint@swing.be>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License version 2 as published by the Free Software Foundation.
*
* 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 "smbview.h"
#include <kprocess.h>
#include <ktempfile.h>
#include <tqheader.h>
#include <tqapplication.h>
#include <kiconloader.h>
#include <klocale.h>
#include <kdebug.h>
#include <kmessagebox.h>
#include <kcursor.h>
#include <tqfile.h>
#include <tqtextstream.h>
#include <cstdlib>
//*********************************************************************************************
SmbView::SmbView(TQWidget *parent, const char *name)
: TDEListView(parent,name)
{
addColumn(i18n("Printer"));
addColumn(i18n("Comment"));
setFrameStyle(TQFrame::WinPanel|TQFrame::Sunken);
setLineWidth(1);
setAllColumnsShowFocus(true);
setRootIsDecorated(true);
m_state = Idle;
m_current = 0;
m_proc = new TDEProcess();
m_proc->setUseShell(true);
m_passwdFile = 0;
connect(m_proc,TQT_SIGNAL(processExited(TDEProcess*)),TQT_SLOT(slotProcessExited(TDEProcess*)));
connect(m_proc,TQT_SIGNAL(receivedStdout(TDEProcess*,char*,int)),TQT_SLOT(slotReceivedStdout(TDEProcess*,char*,int)));
connect(this,TQT_SIGNAL(selectionChanged(TQListViewItem*)),TQT_SLOT(slotSelectionChanged(TQListViewItem*)));
}
SmbView::~SmbView()
{
delete m_proc;
delete m_passwdFile;
}
void SmbView::setLoginInfos(const TQString& login, const TQString& password)
{
m_login = login;
m_password = password;
// We can't pass the password via the command line or the environment
// because the command line is publically accessible on most OSes and
// the environment is publically accessible on some OSes.
// Therefor we write the password to a file and pass that file to
// smbclient with the -A option
delete m_passwdFile;
m_passwdFile = new KTempFile;
m_passwdFile->setAutoDelete(true);
TQTextStream *passwdFile = m_passwdFile->textStream();
if (!passwdFile) return; // Error
(*passwdFile) << "username = " << m_login << endl;
(*passwdFile) << "password = " << m_password << endl;
// (*passwdFile) << "domain = " << ???? << endl;
m_passwdFile->close();
}
void SmbView::startProcess(int state)
{
m_buffer = TQString::null;
m_state = state;
TQApplication::setOverrideCursor(KCursor::waitCursor());
m_proc->start(TDEProcess::NotifyOnExit,TDEProcess::Stdout);
emit running(true);
}
void SmbView::endProcess()
{
switch (m_state)
{
case GroupListing:
processGroups();
break;
case ServerListing:
processServers();
break;
case ShareListing:
processShares();
break;
default:
break;
}
m_state = Idle;
TQApplication::restoreOverrideCursor();
emit running(false);
// clean up for future usage
m_proc->clearArguments();
}
void SmbView::slotProcessExited(TDEProcess*)
{
endProcess();
}
void SmbView::slotReceivedStdout(TDEProcess*, char *buf, int len)
{
m_buffer.append(TQString::fromLocal8Bit(buf,len));
}
void SmbView::init()
{
// Open Samba configuration file and check if a WINS server is defined
m_wins_server = TQString::null;
TQString wins_keyword("wins server");
TQFile smb_conf ("/etc/samba/smb.conf");
if (smb_conf.exists () && smb_conf.open (IO_ReadOnly))
{
TQTextStream smb_stream (&smb_conf);
while (!smb_stream.atEnd ())
{
TQString smb_line = smb_stream.readLine ();
if (smb_line.contains (wins_keyword, FALSE) > 0)
{
TQString key = smb_line.section ('=', 0, 0);
key = key.stripWhiteSpace();
if (key.lower() == wins_keyword)
{
continue;
}
m_wins_server = smb_line.section ('=', 1, 1);
// take only the first declared WINS server
m_wins_server = m_wins_server.section(',', 0, 0);
m_wins_server = m_wins_server.stripWhiteSpace ();
m_wins_server = m_wins_server.section(' ', 0, 0);
// strip any server tag (see man smb.conf(5))
if (m_wins_server.section(':', 1, 1) != NULL)
{
m_wins_server = m_wins_server.section(':', 1, 1);
}
break;
}
}
smb_conf.close ();
}
m_wins_server = m_wins_server.isEmpty ()? " " : " -U " + m_wins_server + " ";
TQString cmd ("nmblookup" + m_wins_server +
"-M -- - | grep '<01>' | awk '{print $1}' | xargs nmblookup -A | grep '<1d>'");
*m_proc << cmd;
startProcess(GroupListing);
}
void SmbView::setOpen(TQListViewItem *item, bool on)
{
if (on && item->childCount() == 0)
{
if (item->depth() == 0)
{ // opening group
m_current = item;
*m_proc << "nmblookup"+m_wins_server+"-M ";
*m_proc << TDEProcess::quote(item->text(0));
*m_proc << " -S";
startProcess(ServerListing);
}
else if (item->depth() == 1)
{ // opening server
char *krb5ccname = getenv ("KRB5CCNAME");
m_current = item;
if (krb5ccname)
{
*m_proc << "smbclient -k -N -L ";
}
else
{
*m_proc << "smbclient -N -L ";
}
*m_proc << TDEProcess::quote (item->text (0));
*m_proc << " -W ";
*m_proc << TDEProcess::quote (item->parent ()->
text (0));
if (!krb5ccname)
{
*m_proc << " -A ";
*m_proc << TDEProcess::
quote (m_passwdFile->name ());
}
startProcess(ShareListing);
}
}
TQListView::setOpen(item,on);
}
void SmbView::processGroups()
{
TQStringList grps = TQStringList::split('\n',m_buffer,false);
clear();
for (TQStringList::ConstIterator it=grps.begin(); it!=grps.end(); ++it)
{
int p = (*it).find("<1d>");
if (p == -1)
continue;
TQListViewItem *item = new TQListViewItem(this,(*it).left(p).stripWhiteSpace());
item->setExpandable(true);
item->setPixmap(0,SmallIcon("network"));
}
}
void SmbView::processServers()
{
TQStringList lines = TQStringList::split('\n',m_buffer,true);
TQString line;
uint index(0);
while (index < lines.count())
{
line = lines[index++].stripWhiteSpace();
if (line.isEmpty())
break;
TQStringList words = TQStringList::split(' ',line,false);
if (words[1] != "<00>" || words[3] == "<GROUP>")
continue;
TQListViewItem *item = new TQListViewItem(m_current,words[0]);
item->setExpandable(true);
item->setPixmap(0,SmallIcon("tdeprint_computer"));
}
}
void SmbView::processShares()
{
TQStringList lines = TQStringList::split('\n',m_buffer,true);
TQString line;
uint index(0);
for (;index < lines.count();index++)
if (lines[index].stripWhiteSpace().startsWith("Sharename"))
break;
index += 2;
while (index < lines.count())
{
line = lines[index++].stripWhiteSpace();
if (line.isEmpty())
break;
else if ( line.startsWith( "Error returning" ) )
{
KMessageBox::error( this, line );
break;
}
TQString typestr(line.mid(15, 10).stripWhiteSpace());
//TQStringList words = TQStringList::split(' ',line,false);
//if (words[1] == "Printer")
if (typestr == "Printer")
{
TQString comm(line.mid(25).stripWhiteSpace()), sharen(line.mid(0, 15).stripWhiteSpace());
//for (uint i=2; i<words.count(); i++)
// comm += (words[i]+" ");
//TQListViewItem *item = new TQListViewItem(m_current,words[0],comm);
TQListViewItem *item = new TQListViewItem(m_current,sharen,comm);
item->setPixmap(0,SmallIcon("tdeprint_printer"));
}
}
}
void SmbView::slotSelectionChanged(TQListViewItem *item)
{
if (item && item->depth() == 2)
emit printerSelected(item->parent()->parent()->text(0),item->parent()->text(0),item->text(0));
}
void SmbView::abort()
{
if (m_proc->isRunning())
m_proc->kill();
}
#include "smbview.moc"