104 linhas
2.7 KiB
C++
104 linhas
2.7 KiB
C++
|
#include <QFile>
|
||
|
|
#include <QFileInfo>
|
||
|
|
#include <QDataStream>
|
||
|
|
#include "sandboxedprocessor.h"
|
||
|
|
#include "pdfprocessor.h"
|
||
|
|
#include "defaulttextprocessor.h"
|
||
|
|
#include "tagstripperprocessor.h"
|
||
|
|
#include "nothingprocessor.h"
|
||
|
|
#include "odtprocessor.h"
|
||
|
|
#include "odsprocessor.h"
|
||
|
|
#include "../submodules/qssb.h/qssb.h"
|
||
|
|
#include "logger.h"
|
||
|
|
|
||
|
|
static DefaultTextProcessor *defaultTextProcessor = new DefaultTextProcessor();
|
||
|
|
static TagStripperProcessor *tagStripperProcessor = new TagStripperProcessor();
|
||
|
|
static NothingProcessor *nothingProcessor = new NothingProcessor();
|
||
|
|
static OdtProcessor *odtProcessor = new OdtProcessor();
|
||
|
|
static OdsProcessor *odsProcessor = new OdsProcessor();
|
||
|
|
|
||
|
|
static QMap<QString, Processor *> processors{
|
||
|
|
{"pdf", new PdfProcessor()}, {"txt", defaultTextProcessor}, {"md", defaultTextProcessor},
|
||
|
|
{"py", defaultTextProcessor}, {"xml", nothingProcessor}, {"html", tagStripperProcessor},
|
||
|
|
{"java", defaultTextProcessor}, {"js", defaultTextProcessor}, {"cpp", defaultTextProcessor},
|
||
|
|
{"c", defaultTextProcessor}, {"sql", defaultTextProcessor}, {"odt", odtProcessor},
|
||
|
|
{"ods", odsProcessor}};
|
||
|
|
|
||
|
|
void SandboxedProcessor::enableSandbox(QString readablePath)
|
||
|
|
{
|
||
|
|
struct qssb_policy *policy = qssb_init_policy();
|
||
|
|
|
||
|
|
policy->namespace_options = QSSB_UNSHARE_NETWORK | QSSB_UNSHARE_USER;
|
||
|
|
|
||
|
|
if(!readablePath.isEmpty())
|
||
|
|
{
|
||
|
|
std::string readablePathLocation = readablePath.toStdString();
|
||
|
|
qssb_append_path_policy(policy, QSSB_FS_ALLOW_READ, readablePathLocation.c_str());
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
policy->no_fs = 1;
|
||
|
|
}
|
||
|
|
int ret = qssb_enable_policy(policy);
|
||
|
|
if(ret != 0)
|
||
|
|
{
|
||
|
|
qDebug() << "Failed to establish sandbox: " << ret;
|
||
|
|
exit(EXIT_FAILURE);
|
||
|
|
}
|
||
|
|
qssb_free_policy(policy);
|
||
|
|
}
|
||
|
|
|
||
|
|
void SandboxedProcessor::printResults(const QVector<PageData> &pageData)
|
||
|
|
{
|
||
|
|
QFile fsstdout;
|
||
|
|
fsstdout.open(stdout, QIODevice::WriteOnly);
|
||
|
|
QDataStream stream(&fsstdout);
|
||
|
|
|
||
|
|
for(const PageData &data : pageData)
|
||
|
|
{
|
||
|
|
stream << data;
|
||
|
|
// fsstdout.flush();
|
||
|
|
}
|
||
|
|
|
||
|
|
fsstdout.close();
|
||
|
|
}
|
||
|
|
|
||
|
|
int SandboxedProcessor::process()
|
||
|
|
{
|
||
|
|
QFileInfo fileInfo(this->filePath);
|
||
|
|
Processor *processor = processors.value(fileInfo.suffix(), nothingProcessor);
|
||
|
|
|
||
|
|
if(processor == nothingProcessor)
|
||
|
|
{
|
||
|
|
/* Nothing to do */
|
||
|
|
return NOTHING_PROCESSED;
|
||
|
|
}
|
||
|
|
|
||
|
|
QVector<PageData> pageData;
|
||
|
|
QString absPath = fileInfo.absoluteFilePath();
|
||
|
|
|
||
|
|
try
|
||
|
|
{
|
||
|
|
if(processor->PREFERED_DATA_SOURCE == FILEPATH)
|
||
|
|
{
|
||
|
|
/* Read access to FS needed... doh..*/
|
||
|
|
enableSandbox(absPath);
|
||
|
|
pageData = processor->process(absPath);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
QByteArray data = Utils::readFile(absPath);
|
||
|
|
enableSandbox();
|
||
|
|
pageData = processor->process(data);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch(LooqsGeneralException &e)
|
||
|
|
{
|
||
|
|
Logger::error() << "Error while processing" << absPath << ":" << e.message << Qt::endl;
|
||
|
|
return 3 /* PROCESSFAIL */;
|
||
|
|
}
|
||
|
|
|
||
|
|
printResults(pageData);
|
||
|
|
return 0;
|
||
|
|
}
|