167 lines
4.5 KiB
C++
167 lines
4.5 KiB
C++
/*
|
|
Copyright (C) 2010-2012 Srivats P.
|
|
|
|
This file is part of "Ostinato"
|
|
|
|
This is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
|
|
#include "portmanager.h"
|
|
|
|
#include "bsdport.h"
|
|
#include "linuxport.h"
|
|
#include "pcapport.h"
|
|
#include "settings.h"
|
|
#include "winpcapport.h"
|
|
|
|
#include <QStringList>
|
|
#include <pcap.h>
|
|
|
|
PortManager *PortManager::instance_ = NULL;
|
|
|
|
PortManager::PortManager()
|
|
{
|
|
int i;
|
|
pcap_if_t *deviceList;
|
|
pcap_if_t *device;
|
|
char errbuf[PCAP_ERRBUF_SIZE];
|
|
AbstractPort::Accuracy txRateAccuracy;
|
|
|
|
qDebug("PCAP Lib: %s", pcap_lib_version());
|
|
qDebug("Retrieving the device list from the local machine\n");
|
|
|
|
if (pcap_findalldevs(&deviceList, errbuf) == -1)
|
|
qDebug("Error in pcap_findalldevs_ex: %s\n", errbuf);
|
|
|
|
txRateAccuracy = rateAccuracy();
|
|
|
|
for(device = deviceList, i = 0; device != NULL; device = device->next, i++)
|
|
{
|
|
AbstractPort *port;
|
|
|
|
qDebug("%d. %s", i, device->name);
|
|
if (device->description)
|
|
qDebug(" (%s)\n", device->description);
|
|
|
|
#if defined(Q_OS_WIN32)
|
|
if (!filterAcceptsPort(device->description))
|
|
#else
|
|
if (!filterAcceptsPort(device->name))
|
|
#endif
|
|
{
|
|
qDebug("%s (%s) rejected by filter. Skipping!",
|
|
device->name, device->description);
|
|
i--;
|
|
continue;
|
|
}
|
|
|
|
#if defined(Q_OS_WIN32)
|
|
port = new WinPcapPort(i, device->name);
|
|
#elif defined(Q_OS_LINUX)
|
|
port = new LinuxPort(i, device->name);
|
|
#elif defined(Q_OS_BSD4)
|
|
port = new BsdPort(i, device->name);
|
|
#else
|
|
port = new PcapPort(i, device->name);
|
|
#endif
|
|
|
|
if (!port->isUsable())
|
|
{
|
|
qDebug("%s: unable to open %s. Skipping!", __FUNCTION__,
|
|
device->name);
|
|
delete port;
|
|
i--;
|
|
continue;
|
|
}
|
|
|
|
if (!port->setRateAccuracy(txRateAccuracy))
|
|
qWarning("failed to set rateAccuracy (%d)", txRateAccuracy);
|
|
|
|
portList_.append(port);
|
|
}
|
|
|
|
pcap_freealldevs(deviceList);
|
|
|
|
foreach(AbstractPort *port, portList_)
|
|
port->init();
|
|
|
|
return;
|
|
}
|
|
|
|
PortManager::~PortManager()
|
|
{
|
|
while (!portList_.isEmpty())
|
|
delete portList_.takeFirst();
|
|
}
|
|
|
|
PortManager* PortManager::instance()
|
|
{
|
|
if (!instance_)
|
|
instance_ = new PortManager;
|
|
|
|
return instance_;
|
|
}
|
|
|
|
AbstractPort::Accuracy PortManager::rateAccuracy()
|
|
{
|
|
QString rateAccuracy = appSettings->value(kRateAccuracyKey,
|
|
kRateAccuracyDefaultValue).toString();
|
|
if (rateAccuracy == "High")
|
|
return AbstractPort::kHighAccuracy;
|
|
else if (rateAccuracy == "Low")
|
|
return AbstractPort::kLowAccuracy;
|
|
else
|
|
qWarning("Unsupported RateAccuracy setting - %s",
|
|
qPrintable(rateAccuracy));
|
|
|
|
return AbstractPort::kHighAccuracy;
|
|
}
|
|
|
|
bool PortManager::filterAcceptsPort(const char *name)
|
|
{
|
|
QRegExp pattern;
|
|
QStringList includeList = appSettings->value(kPortListIncludeKey)
|
|
.toStringList();
|
|
QStringList excludeList = appSettings->value(kPortListExcludeKey)
|
|
.toStringList();
|
|
|
|
pattern.setPatternSyntax(QRegExp::Wildcard);
|
|
|
|
// An empty (or missing) includeList accepts all ports
|
|
// NOTE: A blank "IncludeList=" is read as a stringlist with one
|
|
// string which is empty => treat it same as an empty stringlist
|
|
if (includeList.isEmpty()
|
|
|| (includeList.size() == 1 && includeList.at(0).isEmpty()))
|
|
goto _include_pass;
|
|
|
|
foreach (QString str, includeList) {
|
|
pattern.setPattern(str);
|
|
if (pattern.exactMatch(name))
|
|
goto _include_pass;
|
|
}
|
|
|
|
// IncludeList is not empty and port did not match a pattern
|
|
return false;
|
|
|
|
_include_pass:
|
|
foreach (QString str, excludeList) {
|
|
pattern.setPattern(str);
|
|
if (pattern.exactMatch(name))
|
|
return false;
|
|
}
|
|
|
|
// Port did not match a pattern in ExcludeList
|
|
return true;
|
|
}
|