Imported existing sources
This commit is contained in:
0xFEEDC0DE64
22
DbHashCracker.pro
Normal file
22
DbHashCracker.pro
Normal file
@@ -0,0 +1,22 @@
|
||||
QT += core sql
|
||||
QT -= gui widgets
|
||||
|
||||
DBLIBS +=
|
||||
|
||||
TARGET = hashcracker
|
||||
|
||||
PROJECT_ROOT = ..
|
||||
|
||||
SOURCES += main.cpp \
|
||||
workerthread.cpp
|
||||
|
||||
HEADERS += \
|
||||
workerthread.h
|
||||
|
||||
FORMS +=
|
||||
|
||||
RESOURCES +=
|
||||
|
||||
TRANSLATIONS +=
|
||||
|
||||
include($${PROJECT_ROOT}/app.pri)
|
||||
271
main.cpp
Normal file
271
main.cpp
Normal file
@@ -0,0 +1,271 @@
|
||||
#include <QCoreApplication>
|
||||
#include <qlogging.h>
|
||||
#include <QCommandLineParser>
|
||||
#include <QCommandLineOption>
|
||||
#include <QString>
|
||||
#include <QDebug>
|
||||
#include <QStringList>
|
||||
#include <QFile>
|
||||
#include <QTextStream>
|
||||
#include <QSqlDatabase>
|
||||
#include <QSqlQuery>
|
||||
#include <QSqlError>
|
||||
#include <QVector>
|
||||
#include <QStringList>
|
||||
#include <QDateTime>
|
||||
#include <QTimer>
|
||||
|
||||
#include <algorithm>
|
||||
|
||||
#include "workerthread.h"
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
QCoreApplication app(argc, argv);
|
||||
|
||||
qSetMessagePattern(QStringLiteral("%{time HH:mm:ss.zzz} "
|
||||
"["
|
||||
"%{if-debug}D%{endif}"
|
||||
"%{if-info}I%{endif}"
|
||||
"%{if-warning}W%{endif}"
|
||||
"%{if-critical}C%{endif}"
|
||||
"%{if-fatal}F%{endif}"
|
||||
"] "
|
||||
"%{message}"));
|
||||
|
||||
QCommandLineParser parser;
|
||||
parser.addHelpOption();
|
||||
parser.addVersionOption();
|
||||
parser.addOptions({
|
||||
{
|
||||
{ QStringLiteral("f"), QStringLiteral("filename") },
|
||||
QCoreApplication::translate("main", "File containing the wordlist"),
|
||||
QCoreApplication::translate("main", "filename"),
|
||||
QStringLiteral("wordlist.txt")
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("n"), QStringLiteral("table-name") },
|
||||
QCoreApplication::translate("main", "Name of the table to be filled"),
|
||||
QCoreApplication::translate("main", "table"),
|
||||
QStringLiteral("Hashes")
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("d"), QStringLiteral("no-drop-table") },
|
||||
QCoreApplication::translate("main", "Don't drop the (old) hash table at the beginning"),
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("c"), QStringLiteral("no-create-table") },
|
||||
QCoreApplication::translate("main", "Don't create a table at the beginning"),
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("i"), QStringLiteral("no-index") },
|
||||
QCoreApplication::translate("main", "Don't add an index on hash column"),
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("p"), QStringLiteral("no-primary") },
|
||||
QCoreApplication::translate("main", "Don't add an primary key on value column"),
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("t"), QStringLiteral("threads") },
|
||||
QCoreApplication::translate("main", "Thread count"),
|
||||
QCoreApplication::translate("main", "threads"),
|
||||
QString::number(QThread::idealThreadCount())
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("b"), QStringLiteral("buffer-size") },
|
||||
QCoreApplication::translate("main", "Buffer size (per thread)"),
|
||||
QCoreApplication::translate("main", "buffer-size"),
|
||||
QString::number(100)
|
||||
},
|
||||
{
|
||||
{ QStringLiteral("r"), QStringLiteral("no-replace-into") },
|
||||
QCoreApplication::translate("main", "Use INSERT INTO instead of REPLACE INTO")
|
||||
},
|
||||
{
|
||||
QStringLiteral("no-delayed"),
|
||||
QCoreApplication::translate("main", "Dont add DELAYED keyword when inserting")
|
||||
}
|
||||
});
|
||||
|
||||
parser.process(app);
|
||||
bool ok;
|
||||
|
||||
const auto filename = parser.value(QStringLiteral("filename"));
|
||||
const auto tableName = parser.value(QStringLiteral("table-name"));
|
||||
const auto noDropTable = parser.isSet(QStringLiteral("no-drop-table"));
|
||||
const auto noCreateTable = parser.isSet(QStringLiteral("no-create-table"));
|
||||
const auto noIndex = parser.isSet(QStringLiteral("no-index"));
|
||||
const auto noPrimary = parser.isSet(QStringLiteral("no-primary"));
|
||||
|
||||
const auto threads = parser.value(QStringLiteral("threads")).toInt(&ok);
|
||||
if(!ok)
|
||||
{
|
||||
qFatal("could not parse threads");
|
||||
return -1;
|
||||
}
|
||||
|
||||
|
||||
const auto bufferSize = parser.value(QStringLiteral("buffer-size")).toInt(&ok);;
|
||||
if(!ok)
|
||||
{
|
||||
qFatal("could not parse bufferSize");
|
||||
return -2;
|
||||
}
|
||||
|
||||
const auto noReplace = parser.isSet(QStringLiteral("no-replace-into"));
|
||||
const auto noDelayed = parser.isSet(QStringLiteral("no-delayed"));
|
||||
|
||||
qDebug() << "filename =" << filename;
|
||||
qDebug() << "tableName =" << tableName;
|
||||
qDebug() << "noDropTable =" << noDropTable;
|
||||
qDebug() << "noCreateTable =" << noCreateTable;
|
||||
qDebug() << "noIndex =" << noIndex;
|
||||
qDebug() << "noPrimary =" << noPrimary;
|
||||
qDebug() << "threads =" << threads;
|
||||
qDebug() << "bufferSize =" << bufferSize;
|
||||
qDebug() << "noReplace =" << noReplace;
|
||||
qDebug() << "noDelayed =" << noDelayed;
|
||||
|
||||
qDebug() << "reading wordlist...";
|
||||
|
||||
QStringList wordlist;
|
||||
|
||||
{
|
||||
QFile file(filename);
|
||||
if(!file.open(QIODevice::ReadOnly | QIODevice::Text))
|
||||
{
|
||||
qCritical() << file.errorString();
|
||||
return -1;
|
||||
}
|
||||
|
||||
QTextStream textStream(&file);
|
||||
while(!textStream.atEnd())
|
||||
wordlist.append(textStream.readLine());
|
||||
}
|
||||
|
||||
{
|
||||
QSqlDatabase db = WorkerThread::getDatabase();
|
||||
|
||||
if(!noDropTable)
|
||||
{
|
||||
qDebug() << "dropping old table...";
|
||||
|
||||
auto query = db.exec(QStringLiteral("DROP TABLE IF EXISTS `%0`").arg(tableName));
|
||||
if(query.lastError().isValid())
|
||||
{
|
||||
qCritical() << query.lastError().text();
|
||||
qCritical() << query.lastError().databaseText();
|
||||
qFatal("could not delete old table");
|
||||
return -3;
|
||||
}
|
||||
}
|
||||
|
||||
if(!noCreateTable)
|
||||
{
|
||||
qDebug() << "creating table...";
|
||||
|
||||
QString indexSql;
|
||||
|
||||
if(noIndex)
|
||||
qWarning() << "will not add an index for hash";
|
||||
else
|
||||
indexSql.append(QStringLiteral(",INDEX(`Hash`)"));
|
||||
|
||||
if(noPrimary)
|
||||
qWarning() << "will not add an primary for value";
|
||||
else
|
||||
indexSql.append(QStringLiteral(",PRIMARY KEY(`Value`)"));
|
||||
|
||||
auto query = db.exec(QStringLiteral("CREATE TABLE `%0` ("
|
||||
" `Hash` CHAR(32) NOT NULL,"
|
||||
" `Value` VARCHAR(64) NOT NULL"
|
||||
" %1"
|
||||
")").arg(tableName).arg(indexSql));
|
||||
if(query.lastError().isValid())
|
||||
{
|
||||
qCritical() << query.lastError().text();
|
||||
qCritical() << query.lastError().databaseText();
|
||||
qFatal("could not create table");
|
||||
return -4;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
QVector<QStringList> prefixes(threads);
|
||||
|
||||
for(auto iter = wordlist.constBegin(); iter != wordlist.constEnd(); )
|
||||
for(int i = 0; i < threads && iter != wordlist.constEnd(); i++, iter++)
|
||||
prefixes[i].append(*iter);
|
||||
|
||||
const quint64 allcount = quint64(wordlist.count()) * quint64(wordlist.count());
|
||||
|
||||
QVector<WorkerThread*> workers(threads);
|
||||
|
||||
for(int i = 0; i < threads; i++)
|
||||
{
|
||||
workers[i] = new WorkerThread(prefixes.at(i), wordlist, tableName, bufferSize, noReplace, noDelayed);
|
||||
workers.at(i)->start();
|
||||
QObject::connect(workers.at(i), &QThread::finished, [&workers](){
|
||||
qDebug() << "thread finished";
|
||||
|
||||
const auto allFinished = std::all_of(workers.constBegin(), workers.constEnd(), [](const WorkerThread *worker){
|
||||
return worker->isFinished();
|
||||
});
|
||||
|
||||
if(allFinished)
|
||||
{
|
||||
qDeleteAll(workers);
|
||||
workers.clear();
|
||||
qApp->quit();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
const auto started = QDateTime::currentDateTime();
|
||||
|
||||
QTimer timer;
|
||||
timer.setInterval(1000);
|
||||
QObject::connect(&timer, &QTimer::timeout, [allcount, started](){
|
||||
try {
|
||||
auto formatTime = [](quint64 total){
|
||||
auto seconds = total % 60;
|
||||
total -= seconds;
|
||||
total /= 60;
|
||||
|
||||
auto minutes = total % 60;
|
||||
total -= minutes;
|
||||
total /= 60;
|
||||
|
||||
const auto hours = total;
|
||||
|
||||
return QString("%0:%1:%2")
|
||||
.arg(hours)
|
||||
.arg(minutes, 2, 10, QLatin1Char('0'))
|
||||
.arg(seconds, 2, 10, QLatin1Char('0'));
|
||||
};
|
||||
|
||||
const quint64 sum = WorkerThread::m_counter;
|
||||
const auto speed = WorkerThread::m_speed.fetchAndStoreOrdered(0);
|
||||
|
||||
const auto elapsed = started.secsTo(QDateTime::currentDateTime());
|
||||
const auto avgSpeed = elapsed == 0 ? 0 : sum / elapsed;
|
||||
|
||||
const auto remaining = allcount-sum;
|
||||
|
||||
qInfo().nospace().noquote() << formatTime(elapsed) << " | "
|
||||
<< sum << "/" << allcount
|
||||
<< " (" << QString::number(qreal(sum)/allcount*100., 'f', 2) << "%) | NOW "
|
||||
<< speed << "/sec "
|
||||
<< formatTime(speed == 0 ? 0 : remaining/speed) << " | AVG "
|
||||
<< avgSpeed << "/sec "
|
||||
<< formatTime(avgSpeed == 0 ? 0 : remaining/avgSpeed);
|
||||
} catch (...) {
|
||||
qCritical() << "exception while printing stats";
|
||||
}
|
||||
});
|
||||
timer.start();
|
||||
|
||||
const auto result = app.exec();
|
||||
qDebug() << "finished.";
|
||||
return result;
|
||||
}
|
||||
124
workerthread.cpp
Normal file
124
workerthread.cpp
Normal file
@@ -0,0 +1,124 @@
|
||||
#include "workerthread.h"
|
||||
|
||||
#include <QQueue>
|
||||
#include <QPair>
|
||||
#include <QCryptographicHash>
|
||||
#include <QDebug>
|
||||
#include <QProcessEnvironment>
|
||||
#include <QSqlError>
|
||||
|
||||
QAtomicInt WorkerThread::m_dbIdentifier;
|
||||
|
||||
QAtomicInteger<quint64> WorkerThread::m_counter;
|
||||
QAtomicInt WorkerThread::m_speed;
|
||||
|
||||
WorkerThread::WorkerThread(const QStringList &prefixes, const QStringList &suffixes, const QString &tableName,
|
||||
const int bufferSize, const bool noReplace, const bool noDelayed, QObject *parent) :
|
||||
QThread(parent), m_prefixes(prefixes), m_suffixes(suffixes), m_tableName(tableName),
|
||||
m_bufferSize(bufferSize), m_noReplace(noReplace), m_noDelayed(noDelayed)
|
||||
{
|
||||
}
|
||||
|
||||
void WorkerThread::run()
|
||||
{
|
||||
QSqlDatabase db = getDatabase(QStringLiteral("thread%0").arg(m_dbIdentifier++));
|
||||
if(!db.isOpen())
|
||||
return;
|
||||
|
||||
QSqlQuery query = getQuery(db, m_tableName, m_noReplace, m_noDelayed, m_bufferSize);
|
||||
//TODO error handling
|
||||
|
||||
QQueue<QPair<QString, QString> > queue;
|
||||
|
||||
for(const auto &prefix : m_prefixes)
|
||||
for(const auto &suffix : m_suffixes)
|
||||
{
|
||||
const auto word = prefix + suffix;
|
||||
queue.enqueue(qMakePair(QCryptographicHash::hash(word.toUtf8(), QCryptographicHash::Md5).toHex(), word));
|
||||
|
||||
m_counter++;
|
||||
m_speed++;
|
||||
|
||||
if(queue.count() == m_bufferSize)
|
||||
if(!bindAndExecute(query, queue))
|
||||
return;
|
||||
|
||||
if(isInterruptionRequested())
|
||||
return;
|
||||
}
|
||||
|
||||
if(!queue.isEmpty())
|
||||
{
|
||||
query = getQuery(db, m_tableName, m_noReplace, m_noDelayed, queue.count());
|
||||
//TODO error handling
|
||||
|
||||
if(!bindAndExecute(query, queue))
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
QSqlDatabase WorkerThread::getDatabase(const QString &connectionName)
|
||||
{
|
||||
qDebug() << "connecting to database for" << connectionName;
|
||||
|
||||
QSqlDatabase db(QSqlDatabase::addDatabase(QStringLiteral("QMYSQL"), connectionName));
|
||||
db.setHostName(QProcessEnvironment::systemEnvironment().value(QStringLiteral("MYSQL_HOSTNAME"), QStringLiteral("localhost")));
|
||||
db.setUserName(QProcessEnvironment::systemEnvironment().value(QStringLiteral("MYSQL_USERNAME"), QStringLiteral("hashcracker")));
|
||||
db.setPassword(QProcessEnvironment::systemEnvironment().value(QStringLiteral("MYSQL_PASSWORD"), QStringLiteral("topsecret")));
|
||||
db.setDatabaseName(QProcessEnvironment::systemEnvironment().value(QStringLiteral("MYSQL_DATABASE"), QStringLiteral("hashcracker")));
|
||||
|
||||
if(!db.open())
|
||||
{
|
||||
qCritical() << db.lastError().text();
|
||||
qCritical() << db.lastError().databaseText();
|
||||
qFatal("could not connect to database");
|
||||
}
|
||||
|
||||
return db;
|
||||
}
|
||||
|
||||
QSqlQuery WorkerThread::getQuery(const QSqlDatabase &db, const QString &tableName, const bool noReplace, const bool noDelayed, const int rowCount)
|
||||
{
|
||||
QSqlQuery query(db);
|
||||
|
||||
QString columns;
|
||||
for(int i = 0; i < rowCount; i++)
|
||||
{
|
||||
if(i > 0)
|
||||
columns.append(',');
|
||||
columns.append(QStringLiteral(" (:Hash%0, :Value%0)").arg(i));
|
||||
}
|
||||
|
||||
if(!query.prepare(QStringLiteral("%0%1 INTO `%2` (`Hash`, `Value`) VALUES%3")
|
||||
.arg(noReplace ? QStringLiteral("INSERT") : QStringLiteral("REPLACE"))
|
||||
.arg(noDelayed ? QString() : QStringLiteral(" DELAYED"))
|
||||
.arg(tableName)
|
||||
.arg(columns)))
|
||||
{
|
||||
qCritical() << query.lastError().text();
|
||||
qCritical() << query.lastError().databaseText();
|
||||
qFatal("could not prepare query");
|
||||
}
|
||||
|
||||
return query;
|
||||
}
|
||||
|
||||
bool WorkerThread::bindAndExecute(QSqlQuery &query, QQueue<QPair<QString, QString> > &queue)
|
||||
{
|
||||
for(int i = 0; !queue.isEmpty(); i++)
|
||||
{
|
||||
const auto pair = queue.dequeue();
|
||||
query.bindValue(QStringLiteral(":Hash%0").arg(i), pair.first);
|
||||
query.bindValue(QStringLiteral(":Value%0").arg(i), pair.second);
|
||||
}
|
||||
|
||||
const auto result = query.exec();
|
||||
if(!result)
|
||||
{
|
||||
qCritical() << query.lastError().text();
|
||||
qCritical() << query.lastError().databaseText();
|
||||
qFatal("could not execute query");
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
40
workerthread.h
Normal file
40
workerthread.h
Normal file
@@ -0,0 +1,40 @@
|
||||
#pragma once
|
||||
|
||||
#include <QThread>
|
||||
#include <QAtomicInteger>
|
||||
#include <QAtomicInt>
|
||||
#include <QStringList>
|
||||
#include <QString>
|
||||
#include <QSqlDatabase>
|
||||
#include <QSqlQuery>
|
||||
|
||||
template <class T> class QQueue;
|
||||
template <class T1, class T2> struct QPair;
|
||||
|
||||
class WorkerThread : public QThread
|
||||
{
|
||||
Q_OBJECT
|
||||
|
||||
static QAtomicInt m_dbIdentifier;
|
||||
|
||||
public:
|
||||
static QAtomicInteger<quint64> m_counter;
|
||||
static QAtomicInt m_speed;
|
||||
|
||||
explicit WorkerThread(const QStringList &prefixes, const QStringList &suffixes, const QString &tableName,
|
||||
const int bufferSize, const bool noReplace, const bool noDelayed, QObject *parent = Q_NULLPTR);
|
||||
|
||||
void run() Q_DECL_OVERRIDE;
|
||||
|
||||
static QSqlDatabase getDatabase(const QString &connectionName = QLatin1String(QSqlDatabase::defaultConnection));
|
||||
static QSqlQuery getQuery(const QSqlDatabase &db, const QString &tableName, const bool noReplace, const bool noDelayed, const int rowCount);
|
||||
static bool bindAndExecute(QSqlQuery &query, QQueue<QPair<QString, QString> > &queue);
|
||||
|
||||
private:
|
||||
const QStringList m_prefixes;
|
||||
const QStringList m_suffixes;
|
||||
const QString m_tableName;
|
||||
const int m_bufferSize;
|
||||
const bool m_noReplace;
|
||||
const bool m_noDelayed;
|
||||
};
|
||||
Reference in New Issue
Block a user