networkscanner.cpp
/*
* This file is part of the KDE libraries
* Copyright (c) 2001-2002 Michael Goffioul <kdeprint@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.
**/
#define USE_QSOCKET
#include "networkscanner.h"
#include <qprogressbar.h>
#include <kpushbutton.h>
#include <qlayout.h>
#include <qtimer.h>
#include <qlabel.h>
#include <qcombobox.h>
#include <qlineedit.h>
#include <qregexp.h>
#include <qsocket.h>
#include <klocale.h>
#include <kextendedsocket.h>
#include <kmessagebox.h>
#include <knumvalidator.h>
#include <kdebug.h>
#include <unistd.h>
class NetworkScanner::NetworkScannerPrivate
{
public:
int port;
QString prefixaddress;
int currentaddress;
int timeout;
bool scanning;
QPtrList<NetworkScanner::SocketInfo> printers;
QProgressBar *bar;
KPushButton *scan, *settings;
QLabel *subnetlab;
QTimer *timer;
#ifdef USE_QSOCKET
QSocket *socket;
#else
KExtendedSocket *socket;
#endif
NetworkScannerPrivate( int portvalue ) : port( portvalue )
{
prefixaddress = localPrefix();
currentaddress = 1;
timeout = 50;
scanning = false;
printers.setAutoDelete( true );
}
QString localPrefix();
QString scanString();
};
QString NetworkScanner::NetworkScannerPrivate::localPrefix()
{
char buf[256];
buf[0] = '\0';
if (!gethostname(buf, sizeof(buf)))
buf[sizeof(buf)-1] = '\0';
QPtrList<KAddressInfo> infos = KExtendedSocket::lookup(buf, QString::null);
infos.setAutoDelete(true);
if (infos.count() > 0)
{
QString IPstr = infos.first()->address()->nodeName();
int p = IPstr.findRev('.');
IPstr.truncate(p);
return IPstr;
}
return QString::null;
}
QString NetworkScanner::NetworkScannerPrivate::scanString()
{
QString s = prefixaddress + ".*";
if ( port != -1 )
s.append( ":" ).append( QString::number( port ) );
return s;
}
NetworkScanner::NetworkScanner( int port, QWidget *parent, const char *name )
: QWidget( parent, name )
{
d = new NetworkScannerPrivate( port );
d->bar = new QProgressBar( 256, this );
d->settings = new KPushButton( KGuiItem( i18n( "&Settings" ), "configure" ), this );
d->scan = new KPushButton( KGuiItem( i18n( "Sc&an" ), "viewmag" ), this );
d->timer = new QTimer( this );
#ifdef USE_QSOCKET
d->socket = new QSocket( this );
#else
d->socket = new KExtendedSocket();
#endif
QLabel *label = new QLabel( i18n( "Network scan:" ), this );
d->subnetlab = new QLabel( i18n( "Subnet: %1" ).arg( d->scanString() ), this );
QGridLayout *l0 = new QGridLayout( this, 4, 2, 0, 10 );
l0->addMultiCellWidget( label, 0, 0, 0, 1 );
l0->addMultiCellWidget( d->bar, 1, 1, 0, 1 );
l0->addMultiCellWidget( d->subnetlab, 2, 2, 0, 1 );
l0->addWidget( d->settings, 3, 0 );
l0->addWidget( d->scan, 3, 1 );
connect( d->timer, SIGNAL( timeout() ), SLOT( slotTimeout() ) );
connect( d->settings, SIGNAL( clicked() ), SLOT( slotSettingsClicked() ) );
connect( d->scan, SIGNAL( clicked() ), SLOT( slotScanClicked() ) );
#ifdef USE_QSOCKET
connect( d->socket, SIGNAL( connected() ), SLOT( slotConnectionSuccess() ) );
connect( d->socket, SIGNAL( error( int ) ), SLOT( slotConnectionFailed( int ) ) );
#else
connect( d->socket, SIGNAL( connectionSuccess() ), SLOT( slotConnectionSuccess() ) );
connect( d->socket, SIGNAL( connectionFailed( int ) ), SLOT( slotConnectionFailed( int ) ) );
#endif
}
NetworkScanner::~NetworkScanner()
{
#ifndef USE_QSOCKET
delete d->socket;
#endif
delete d;
}
void NetworkScanner::start()
{
if ( d->scanning )
return;
d->printers.clear();
emit scanStarted();
d->settings->setEnabled( false );
d->scan->setGuiItem( KGuiItem( i18n( "&Abort" ), "stop" ) );
d->currentaddress = -1;
d->scanning = true;
next();
}
void NetworkScanner::slotScanClicked()
{
if ( !d->scanning )
{
if ( d->localPrefix() == d->prefixaddress ||
KMessageBox::warningContinueCancel( this->parentWidget(),
i18n( "You are about to scan a subnet (%1.*) that does not "
"correspond to the current subnet of this computer (%2.*). Do you want "
"to scan the specified subnet anyway?" ).arg( d->prefixaddress ).arg( d->localPrefix() ),
QString::null, KGuiItem( i18n( "&Scan" ), "viewmag" ), "askForScan" ) == KMessageBox::Continue )
start();
}
else
{
#ifdef USE_QSOCKET
d->socket->close();
#else
d->socket->cancelAsyncConnect();
#endif
finish();
}
}
void NetworkScanner::finish()
{
if ( !d->scanning )
return;
d->settings->setEnabled( true );
d->scan->setGuiItem( KGuiItem( i18n( "Sc&an" ), "viewmag" ) );
d->bar->reset();
d->scanning = false;
emit scanFinished();
}
void NetworkScanner::slotSettingsClicked()
{
NetworkScannerConfig dlg( this );
dlg.exec();
}
void NetworkScanner::slotNext()
{
if ( !d->scanning )
return;
d->timer->stop();
#ifdef USE_QSOCKET
d->socket->connectToHost( d->prefixaddress + "." + QString::number( d->currentaddress ), d->port );
kdDebug() << "Address: " << d->socket->peerName() << ", Port: " << d->socket->peerPort() << endl;
#else
d->socket->setAddress( d->prefixaddress + "." + QString::number( d->currentaddress ), d->port );
d->socket->startAsyncLookup();
kdDebug() << "Address: " << d->socket->host() << ", Port: " << d->socket->port() << endl;
#endif
d->timer->start( d->timeout, true );
}
void NetworkScanner::next()
{
//kdDebug() << "Next" << endl;
d->currentaddress++;
if ( d->currentaddress >= 256 )
finish();
else
{
d->bar->setProgress( d->currentaddress );
QTimer::singleShot( 0, this, SLOT( slotNext() ) );
}
}
void NetworkScanner::slotTimeout()
{
kdDebug() << "Timeout" << endl;
if ( !d->scanning )
return;
#ifdef USE_QSOCKET
d->socket->close();
#else
d->socket->cancelAsyncConnect();
#endif
next();
}
void NetworkScanner::slotConnectionSuccess()
{
kdDebug() << "Success" << endl;
#ifdef USE_QSOCKET
KSocketAddress *addr = KExtendedSocket::peerAddress( d->socket->socket() );
#else
KSocketAddress *addr = const_cast<KSocketAddress*>( d->socket->peerAddress() );
#endif
kdDebug() << "Connection success: " << ( addr ? addr->pretty() : QString( "ERROR" ) ) << endl;
kdDebug() << "Socket: " << d->socket->socket() << endl;
if ( addr )
{
SocketInfo *info = new SocketInfo;
#ifdef USE_QSOCKET
info->IP = d->socket->peerName();
#else
info->IP = d->socket->host();
#endif
info->Port = d->port;
QString portname;
KExtendedSocket::resolve( addr, info->Name, portname );
d->printers.append( info );
d->socket->close();
delete addr;
}
else
kdDebug() << "Unconnected socket, skipping" << endl;
next();
}
void NetworkScanner::slotConnectionFailed( int )
{
kdDebug() << "Failure" << endl;
next();
}
const QPtrList<NetworkScanner::SocketInfo>* NetworkScanner::printerList()
{
return &( d->printers );
}
int NetworkScanner::timeout() const
{
return d->timeout;
}
void NetworkScanner::setTimeout( int to )
{
d->timeout = to;
}
QString NetworkScanner::subnet() const
{
return d->prefixaddress;
}
void NetworkScanner::setSubnet( const QString& sn )
{
d->prefixaddress = sn;
d->subnetlab->setText( i18n( "Subnet: %1" ).arg( d->scanString() ) );
}
int NetworkScanner::port() const
{
return d->port;
}
void NetworkScanner::setPort( int p )
{
d->port = p;
d->subnetlab->setText( i18n( "Subnet: %1" ).arg( d->scanString() ) );
}
bool NetworkScanner::checkPrinter( const QString& host, int port )
{
// try first to find it in the SocketInfo list
QPtrListIterator<NetworkScanner::SocketInfo> it( d->printers );
for ( ; it.current(); ++it )
{
if ( port == it.current()->Port && ( host == it.current()->IP ||
host == it.current()->Name ) )
return true;
}
// not found in SocketInfo list, try to establish connection
KExtendedSocket extsock( host, port );
extsock.setBlockingMode( false );
extsock.setTimeout( 0, d->timeout * 1000 );
return ( extsock.connect() == 0 );
}
NetworkScannerConfig::NetworkScannerConfig(NetworkScanner *scanner, const char *name)
: KDialogBase(scanner, name, true, QString::null, Ok|Cancel, Ok, true)
{
scanner_ = scanner;
QWidget *dummy = new QWidget(this);
setMainWidget(dummy);
KIntValidator *val = new KIntValidator( this );
QLabel *masklabel = new QLabel(i18n("&Subnetwork:"),dummy);
QLabel *portlabel = new QLabel(i18n("&Port:"),dummy);
QLabel *toutlabel = new QLabel(i18n("&Timeout (ms):"),dummy);
QLineEdit *mm = new QLineEdit(dummy);
mm->setText(QString::fromLatin1(".[0-255]"));
mm->setReadOnly(true);
mm->setFixedWidth(fontMetrics().width(mm->text())+10);
mask_ = new QLineEdit(dummy);
mask_->setAlignment(Qt::AlignRight);
port_ = new QComboBox(true,dummy);
if ( port_->lineEdit() )
port_->lineEdit()->setValidator( val );
tout_ = new QLineEdit(dummy);
tout_->setValidator( val );
masklabel->setBuddy(mask_);
portlabel->setBuddy(port_);
toutlabel->setBuddy(tout_);
mask_->setText(scanner_->subnet());
port_->insertItem("631");
port_->insertItem("9100");
port_->insertItem("9101");
port_->insertItem("9102");
port_->setEditText(QString::number(scanner_->port()));
tout_->setText(QString::number(scanner_->timeout()));
QGridLayout *main_ = new QGridLayout(dummy, 3, 2, 0, 10);
QHBoxLayout *lay1 = new QHBoxLayout(0, 0, 5);
main_->addWidget(masklabel, 0, 0);
main_->addWidget(portlabel, 1, 0);
main_->addWidget(toutlabel, 2, 0);
main_->addLayout(lay1, 0, 1);
main_->addWidget(port_, 1, 1);
main_->addWidget(tout_, 2, 1);
lay1->addWidget(mask_,1);
lay1->addWidget(mm,0);
resize(250,130);
setCaption(i18n("Scan Configuration"));
}
NetworkScannerConfig::~NetworkScannerConfig()
{
}
void NetworkScannerConfig::slotOk()
{
QString msg;
QRegExp re("(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})");
if (!re.exactMatch(mask_->text()))
msg = i18n("Wrong subnetwork specification.");
else
{
for (int i=1; i<=3; i++)
if (re.cap(i).toInt() >= 255)
{
msg = i18n("Wrong subnetwork specification.");
break;
}
}
bool ok(false);
int v = tout_->text().toInt(&ok);
if (!ok || v <= 0)
msg = i18n("Wrong timeout specification.");
v = port_->currentText().toInt(&ok);
if (!ok || v <= 0)
msg = i18n("Wrong port specification.");
if (!msg.isEmpty())
{
KMessageBox::error(this,msg);
return;
}
scanner_->setTimeout( tout_->text().toInt() );
scanner_->setSubnet( mask_->text() );
scanner_->setPort( port_->currentText().toInt() );
KDialogBase::slotOk();
}
#include "networkscanner.moc"

KDE Home | KDE Accessibility Home | Description of Access Keys