smbview.cpp

/*
 *  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 <tdeprocess.h>
#include <tdetempfile.h>
#include <tqheader.h>
#include <tqapplication.h>
 
#include <kiconloader.h>
#include <tdelocale.h>
#include <kdebug.h>
#include <tdemessagebox.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,TQ_SIGNAL(processExited(TDEProcess*)),TQ_SLOT(slotProcessExited(TDEProcess*)));
    connect(m_proc,TQ_SIGNAL(receivedStdout(TDEProcess*,char*,int)),TQ_SLOT(slotReceivedStdout(TDEProcess*,char*,int)));
    connect(this,TQ_SIGNAL(selectionChanged(TQListViewItem*)),TQ_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"