171 lines
4.0 KiB
C++
171 lines
4.0 KiB
C++
/*
|
|
Copyright (C) 2010 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 "pdmlfileformat.h"
|
|
|
|
#include "ostprotolib.h"
|
|
#include "pdmlreader.h"
|
|
|
|
#include <QProcess>
|
|
#include <QTemporaryFile>
|
|
|
|
PdmlFileFormat pdmlFileFormat;
|
|
|
|
PdmlFileFormat::PdmlFileFormat()
|
|
{
|
|
}
|
|
|
|
PdmlFileFormat::~PdmlFileFormat()
|
|
{
|
|
}
|
|
|
|
bool PdmlFileFormat::open(const QString fileName,
|
|
OstProto::StreamConfigList &streams, QString &error)
|
|
{
|
|
bool isOk = false;
|
|
QFile file(fileName);
|
|
PdmlReader *reader = new PdmlReader(&streams);
|
|
|
|
if (!file.open(QIODevice::ReadOnly))
|
|
goto _open_fail;
|
|
|
|
connect(reader, SIGNAL(progress(int)), this, SIGNAL(progress(int)));
|
|
emit status("Reading PDML packets...");
|
|
emit target(100); // in percentage
|
|
|
|
isOk = reader->read(&file, NULL, &stop_);
|
|
|
|
if (stop_)
|
|
goto _user_cancel;
|
|
|
|
if (!isOk)
|
|
{
|
|
error.append(QString("Error processing PDML (%1, %2): %3\n")
|
|
.arg(reader->lineNumber())
|
|
.arg(reader->columnNumber())
|
|
.arg(reader->errorString()));
|
|
goto _exit;
|
|
}
|
|
|
|
goto _exit;
|
|
|
|
_open_fail:
|
|
isOk = false;
|
|
|
|
_user_cancel:
|
|
_exit:
|
|
delete reader;
|
|
|
|
return isOk;
|
|
}
|
|
|
|
bool PdmlFileFormat::save(const OstProto::StreamConfigList streams,
|
|
const QString fileName, QString &error)
|
|
{
|
|
bool isOk = false;
|
|
QTemporaryFile pcapFile;
|
|
StreamFileFormat *fmt = StreamFileFormat::fileFormatFromType("PCAP");
|
|
QProcess tshark;
|
|
|
|
Q_ASSERT(fmt);
|
|
|
|
if (!pcapFile.open())
|
|
{
|
|
error.append("Unable to open temporary file to create PCAP\n");
|
|
goto _fail;
|
|
}
|
|
|
|
qDebug("intermediate PCAP %s", qPrintable(pcapFile.fileName()));
|
|
|
|
connect(fmt, SIGNAL(target(int)), this, SIGNAL(target(int)));
|
|
connect(fmt, SIGNAL(progress(int)), this, SIGNAL(progress(int)));
|
|
|
|
emit status("Writing intermediate PCAP file...");
|
|
isOk = fmt->save(streams, pcapFile.fileName(), error);
|
|
|
|
qDebug("generating PDML %s", qPrintable(fileName));
|
|
emit status("Converting PCAP to PDML...");
|
|
emit target(0);
|
|
|
|
tshark.setStandardOutputFile(fileName);
|
|
tshark.start(OstProtoLib::tsharkPath(),
|
|
QStringList()
|
|
<< QString("-r%1").arg(pcapFile.fileName())
|
|
<< "-Tpdml");
|
|
if (!tshark.waitForStarted(-1))
|
|
{
|
|
error.append(QString("Unable to start tshark. Check path in preferences.\n"));
|
|
goto _fail;
|
|
}
|
|
|
|
if (!tshark.waitForFinished(-1))
|
|
{
|
|
error.append(QString("Error running tshark\n"));
|
|
goto _fail;
|
|
}
|
|
|
|
isOk = true;
|
|
_fail:
|
|
return isOk;
|
|
}
|
|
|
|
bool PdmlFileFormat::isMyFileFormat(const QString fileName)
|
|
{
|
|
bool ret = false;
|
|
QFile file(fileName);
|
|
QByteArray buf;
|
|
QXmlStreamReader xml;
|
|
|
|
if (!file.open(QIODevice::ReadOnly))
|
|
goto _exit;
|
|
|
|
xml.setDevice(&file);
|
|
|
|
xml.readNext();
|
|
if (xml.hasError() || !xml.isStartDocument())
|
|
goto _close_exit;
|
|
|
|
// skip everything until the start of the first element
|
|
while (!xml.isStartElement())
|
|
{
|
|
xml.readNext();
|
|
if (xml.hasError())
|
|
goto _close_exit;
|
|
}
|
|
|
|
if (!xml.hasError() && xml.isStartElement() && (xml.name() == "pdml"))
|
|
ret = true;
|
|
else
|
|
ret = false;
|
|
|
|
_close_exit:
|
|
xml.clear();
|
|
file.close();
|
|
_exit:
|
|
return ret;
|
|
}
|
|
|
|
bool PdmlFileFormat::isMyFileType(const QString fileType)
|
|
{
|
|
if (fileType.startsWith("PDML"))
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|