aheinecke@100: #include "cinstprocesstest.h"
aheinecke@100: #include "certificatelist.h"
aheinecke@100: #include "errorcodes.h"
aheinecke@100: 
aheinecke@100: #include <QDebug>
aheinecke@100: #include <QDir>
aheinecke@100: #include <QFile>
aheinecke@100: #include <QProcess>
aheinecke@100: 
aheinecke@294: #define CINST_PATH_CANDIDATES "../../cinst/cinst" << \
aheinecke@294:     "cinst" << "../../cinst/cinst.exe" << "cinst.exe";
aheinecke@100: 
aheinecke@294: QProcess *CinstProcessTest::startCinstProcess(const QStringList& args) {
aheinecke@294:     QStringList cinstCandidates;
aheinecke@294:     cinstCandidates << CINST_PATH_CANDIDATES;
aheinecke@294:     QString processPath;
aheinecke@294:     foreach (const QString& candidate, cinstCandidates) {
aheinecke@294:         QFileInfo fi(candidate);
aheinecke@294:         if (fi.isExecutable()) {
aheinecke@294:             processPath = candidate;
aheinecke@294:             break;
aheinecke@294:         }
aheinecke@294:     }
aheinecke@294: 
aheinecke@100:     QProcess *installerProcess = new QProcess();
aheinecke@294:     installerProcess->setArguments(args);
aheinecke@294:     installerProcess->setProgram(processPath);
aheinecke@100:     installerProcess->start();
aheinecke@100:     installerProcess->waitForStarted();
aheinecke@100:     return installerProcess;
aheinecke@100: }
aheinecke@100: 
aheinecke@151: #define VERIFY_PROC_DEBUG(x) \
aheinecke@151:     if (! x ) { \
aheinecke@151:         qDebug() << "Stdout:" << proc->readAllStandardOutput(); \
aheinecke@151:         qDebug() << "Stderr:" << proc->readAllStandardError(); \
aheinecke@151:         qDebug() << "Exit code: " << proc->exitCode(); \
aheinecke@151:     } \
aheinecke@151:     QVERIFY(x)
aheinecke@151: 
aheinecke@100: void finishVerify(QProcess *proc, int exitCode) {
aheinecke@100:     proc->closeWriteChannel();
aheinecke@100:     proc->waitForFinished();
aheinecke@151:     VERIFY_PROC_DEBUG(proc->exitStatus() == QProcess::NormalExit);
aheinecke@151:     VERIFY_PROC_DEBUG(proc->exitCode() == exitCode);
aheinecke@100:     delete proc;
aheinecke@100: }
aheinecke@100: 
aheinecke@100: void CinstProcessTest::testValidInput() {
aheinecke@295:     QStringList args;
aheinecke@295:     args << "list=" + validListFile.fileName();
aheinecke@100: 
aheinecke@295:     QTemporaryFile instructions;
aheinecke@295:     instructions.open();
aheinecke@295:     foreach (const Certificate &cert, validList.getCertificates()) {
aheinecke@295:         instructions.write(cert.base64Line().toLatin1());
aheinecke@295:         instructions.write("\n");
aheinecke@295:     }
aheinecke@295:     instructions.close();
aheinecke@100: 
aheinecke@295:     args << "instructions=" + instructions.fileName();
aheinecke@100: 
aheinecke@295:     QProcess* installerProcess = startCinstProcess(args);
aheinecke@159:     finishVerify(installerProcess, ERR_NO_ERROR);
aheinecke@100: }
aheinecke@100: 
aheinecke@100: void CinstProcessTest::initTestCase() {
aheinecke@295:     QFile valid(":/list-valid-signed.txt");
aheinecke@295:     valid.open(QIODevice::ReadOnly);
aheinecke@295:     validListFile.open();
aheinecke@295:     validListFile.write(valid.readAll());
aheinecke@295:     valid.close();
aheinecke@295:     validListFile.close();
aheinecke@295:     validList = CertificateList(validListFile.fileName().toLocal8Bit().data());
aheinecke@295: 
aheinecke@295:     QVERIFY(validList.isValid());
aheinecke@295: 
aheinecke@295:     QFile invalid(":/list-invalid-signed.txt");
aheinecke@295:     invalid.open(QIODevice::ReadOnly);
aheinecke@295:     invalidListFile.open();
aheinecke@295:     invalidListFile.write(invalid.readAll());
aheinecke@295:     invalid.close();
aheinecke@295:     invalidListFile.close();
aheinecke@295:     invalidList = CertificateList(invalidListFile.fileName().toLocal8Bit().data());
aheinecke@295: 
aheinecke@295:     QVERIFY(!invalidList.isValid());
aheinecke@295: 
aheinecke@295:     QFile other(":/list-valid-other-signature.txt");
aheinecke@295:     other.open(QIODevice::ReadOnly);
aheinecke@295:     otherListFile.open();
aheinecke@295:     otherListFile.write(other.readAll());
aheinecke@295:     other.close();
aheinecke@295:     otherListFile.close();
aheinecke@295:     otherList = CertificateList(otherListFile.fileName().toLocal8Bit().data());
aheinecke@295: 
aheinecke@295:     QVERIFY(!otherList.isValid());
aheinecke@100: }
aheinecke@100: 
aheinecke@100: void CinstProcessTest::testNoList() {
aheinecke@100:     /* No list */
aheinecke@296:     QTemporaryFile emptyFile;
aheinecke@296:     emptyFile.open();
aheinecke@296:     emptyFile.close();
aheinecke@100: 
aheinecke@296:     QStringList args;
aheinecke@296:     args << "list=" + emptyFile.fileName();
aheinecke@296: 
aheinecke@296:     QTemporaryFile instructions;
aheinecke@296:     instructions.open();
aheinecke@249:     foreach (const Certificate &cert, validList.getCertificates()) {
aheinecke@296:         instructions.write(cert.base64Line().toLatin1());
aheinecke@296:         instructions.write("\n");
aheinecke@100:     }
aheinecke@296:     instructions.close();
aheinecke@296: 
aheinecke@296:     args << "instructions=" + instructions.fileName();
aheinecke@296: 
aheinecke@296:     QProcess* installerProcess = startCinstProcess(args);
aheinecke@100:     finishVerify(installerProcess, ERR_INVALID_INPUT_NO_LIST);
aheinecke@100: }
aheinecke@100: 
aheinecke@100: void CinstProcessTest::testGarbageInput() {
aheinecke@100:     QProcess* installerProcess = startCinstProcess();
aheinecke@100:     QVERIFY(installerProcess->state() == QProcess::Running);
aheinecke@100:     /* Garbage */
aheinecke@100:     installerProcess = startCinstProcess();
aheinecke@100:     installerProcess->write("-----BEGIN CERTIFICATE LIST-----\r\n");
aheinecke@100:     int retval=0;
aheinecke@100:     int bytesWritten=0;
aheinecke@100:     do {
aheinecke@100:         char garbage[1030];
aheinecke@100:         for (int i = 0; i < 1030; i++) {
aheinecke@100:             garbage[i] = (char) qrand() % 255;
aheinecke@100:         }
aheinecke@100:         retval = installerProcess->write(garbage, 1030);
aheinecke@100:         bytesWritten += retval;
aheinecke@115:     } while (retval != -1 && bytesWritten < 15 *1024 *1024 );
aheinecke@100: 
aheinecke@115:     finishVerify(installerProcess, ERR_INVALID_INPUT);
aheinecke@100: }
aheinecke@100: 
aheinecke@100: void CinstProcessTest::testNoInput() {
aheinecke@100:     QProcess* installerProcess = startCinstProcess();
aheinecke@100:     QVERIFY(installerProcess->state() == QProcess::Running);
aheinecke@100: 
aheinecke@100:     /* Nothing */
aheinecke@100:     installerProcess = startCinstProcess();
aheinecke@100:     finishVerify(installerProcess, ERR_INVALID_INPUT_NO_LIST);
aheinecke@100: }
aheinecke@100: 
aheinecke@100: 
aheinecke@100: void CinstProcessTest::testNoInstructions() {
aheinecke@100:     /* No instructions */
aheinecke@100:     QProcess* installerProcess = startCinstProcess();
aheinecke@100:     QVERIFY(installerProcess->state() == QProcess::Running);
aheinecke@100:     installerProcess->write("-----BEGIN CERTIFICATE LIST-----\r\n");
aheinecke@100:     installerProcess->write(validList.rawData().toLatin1());
aheinecke@100:     installerProcess->write("-----END CERTIFICATE LIST-----\r\n");
aheinecke@100: 
aheinecke@100:     finishVerify(installerProcess, ERR_NO_INSTRUCTIONS);
aheinecke@100: }
aheinecke@100: 
aheinecke@100: void CinstProcessTest::testInvalidInstruction() {
aheinecke@100:     QProcess* installerProcess = startCinstProcess();
aheinecke@100:     QVERIFY(installerProcess->state() == QProcess::Running);
aheinecke@100: 
aheinecke@100:     installerProcess->write("-----BEGIN CERTIFICATE LIST-----\r\n");
aheinecke@100:     installerProcess->write(validList.rawData().toLatin1());
aheinecke@100:     installerProcess->write("-----END CERTIFICATE LIST-----\r\n");
aheinecke@100: 
aheinecke@150:     installerProcess->write("I:ABCDEF\r\n");
aheinecke@100: 
aheinecke@100:     finishVerify(installerProcess, ERR_INVALID_INSTRUCTIONS);
aheinecke@100: }
aheinecke@100: 
aheinecke@151: void CinstProcessTest::testUninstall() {
aheinecke@151:     QProcess* installerProcess = startCinstProcess();
aheinecke@151:     QVERIFY(installerProcess->state() == QProcess::Running);
aheinecke@151: 
aheinecke@151:     installerProcess->write("-----BEGIN CERTIFICATE LIST-----\r\n");
aheinecke@151:     installerProcess->write(validList.rawData().toLatin1());
aheinecke@151:     installerProcess->write("-----END CERTIFICATE LIST-----\r\n");
aheinecke@151: 
aheinecke@151:     installerProcess->write("UNINSTALL\r\n");
aheinecke@151: 
aheinecke@159:     finishVerify(installerProcess, ERR_NO_ERROR);
aheinecke@151: }
aheinecke@151: 
aheinecke@100: QTEST_GUILESS_MAIN (CinstProcessTest);