Compare commits
3 Commits
a529e82912
...
3b83e60352
| Author | SHA1 | Date |
|---|---|---|
 laptoy
|
3b83e60352 | 6 months ago |
|
laptoy
|
f6130b2d1f | 6 months ago |
|
laptoy
|
e7778c32b7 | 6 months ago |
@ -0,0 +1,575 @@
|
|||||||
|
/****************************************************************************
|
||||||
|
*
|
||||||
|
* (c) 2019 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
*
|
||||||
|
* QGroundControl is licensed according to the terms in the file
|
||||||
|
* COPYING.md in the root of the source code directory.
|
||||||
|
*
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
#include "PairingManager.h"
|
||||||
|
#include "SettingsManager.h"
|
||||||
|
#include "MicrohardManager.h"
|
||||||
|
#include "QGCApplication.h"
|
||||||
|
#include "QGCCorePlugin.h"
|
||||||
|
|
||||||
|
#include <QSettings>
|
||||||
|
#include <QJsonObject>
|
||||||
|
#include <QStandardPaths>
|
||||||
|
#include <QMutexLocker>
|
||||||
|
|
||||||
|
QGC_LOGGING_CATEGORY(PairingManagerLog, "PairingManagerLog")
|
||||||
|
|
||||||
|
static const char* jsonFileName = "pairing.json";
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
static QString
|
||||||
|
random_string(uint length)
|
||||||
|
{
|
||||||
|
auto randchar = []() -> char
|
||||||
|
{
|
||||||
|
const char charset[] =
|
||||||
|
"0123456789"
|
||||||
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||||
|
"abcdefghijklmnopqrstuvwxyz";
|
||||||
|
const uint max_index = (sizeof(charset) - 1);
|
||||||
|
return charset[static_cast<uint>(rand()) % max_index];
|
||||||
|
};
|
||||||
|
std::string str(length, 0);
|
||||||
|
std::generate_n(str.begin(), length, randchar);
|
||||||
|
return QString::fromStdString(str);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
PairingManager::PairingManager(QGCApplication* app, QGCToolbox* toolbox)
|
||||||
|
: QGCTool(app, toolbox)
|
||||||
|
, _aes("J6+KuWh9K2!hG(F'", 0x368de30e8ec063ce)
|
||||||
|
{
|
||||||
|
_jsonFileName = QDir::temp().filePath(jsonFileName);
|
||||||
|
connect(this, &PairingManager::parsePairingJson, this, &PairingManager::_parsePairingJson);
|
||||||
|
connect(this, &PairingManager::setPairingStatus, this, &PairingManager::_setPairingStatus);
|
||||||
|
connect(this, &PairingManager::startUpload, this, &PairingManager::_startUpload);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
PairingManager::~PairingManager()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
void
|
||||||
|
PairingManager::setToolbox(QGCToolbox *toolbox)
|
||||||
|
{
|
||||||
|
QGCTool::setToolbox(toolbox);
|
||||||
|
_updatePairedDeviceNameList();
|
||||||
|
emit pairedListChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_pairingCompleted(QString name)
|
||||||
|
{
|
||||||
|
_writeJson(_jsonDoc, _pairingCacheFile(name));
|
||||||
|
_remotePairingMap["NM"] = name;
|
||||||
|
_lastPaired = name;
|
||||||
|
_updatePairedDeviceNameList();
|
||||||
|
emit pairedListChanged();
|
||||||
|
emit pairedVehicleChanged();
|
||||||
|
//_app->informationMessageBoxOnMainThread("", tr("Paired with %1").arg(name));
|
||||||
|
setPairingStatus(PairingSuccess, tr("Pairing Successfull"));
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_connectionCompleted(QString /*name*/)
|
||||||
|
{
|
||||||
|
//QString pwd = _remotePairingMap["PWD"].toString();
|
||||||
|
//_toolbox->microhardManager()->switchToConnectionEncryptionKey(pwd);
|
||||||
|
//_app->informationMessageBoxOnMainThread("", tr("Connected to %1").arg(name));
|
||||||
|
setPairingStatus(PairingConnected, tr("Connection Successfull"));
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_startUpload(QString pairURL, QJsonDocument jsonDoc)
|
||||||
|
{
|
||||||
|
QMutexLocker lock(&_uploadMutex);
|
||||||
|
if (_uploadManager != nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
_uploadManager = new QNetworkAccessManager(this);
|
||||||
|
|
||||||
|
QString str = jsonDoc.toJson(QJsonDocument::JsonFormat::Compact);
|
||||||
|
qCDebug(PairingManagerLog) << "Starting upload to: " << pairURL << " " << str;
|
||||||
|
_uploadData = QString::fromStdString(_aes.encrypt(str.toStdString()));
|
||||||
|
_uploadURL = pairURL;
|
||||||
|
_startUploadRequest();
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_startUploadRequest()
|
||||||
|
{
|
||||||
|
QNetworkRequest req;
|
||||||
|
req.setUrl(QUrl(_uploadURL));
|
||||||
|
req.setHeader(QNetworkRequest::ContentTypeHeader, "application/x-www-form-urlencoded");
|
||||||
|
QNetworkReply *reply = _uploadManager->post(req, _uploadData.toUtf8());
|
||||||
|
connect(reply, &QNetworkReply::finished, this, &PairingManager::_uploadFinished);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_stopUpload()
|
||||||
|
{
|
||||||
|
QMutexLocker lock(&_uploadMutex);
|
||||||
|
if (_uploadManager != nullptr) {
|
||||||
|
delete _uploadManager;
|
||||||
|
_uploadManager = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_uploadFinished()
|
||||||
|
{
|
||||||
|
QMutexLocker lock(&_uploadMutex);
|
||||||
|
QNetworkReply* reply = qobject_cast<QNetworkReply*>(QObject::sender());
|
||||||
|
if (reply) {
|
||||||
|
if (_uploadManager != nullptr) {
|
||||||
|
if (reply->error() == QNetworkReply::NoError) {
|
||||||
|
qCDebug(PairingManagerLog) << "Upload finished.";
|
||||||
|
QByteArray bytes = reply->readAll();
|
||||||
|
QString str = QString::fromUtf8(bytes.data(), bytes.size());
|
||||||
|
qCDebug(PairingManagerLog) << "Reply: " << str;
|
||||||
|
auto a = str.split(QRegExp("\\s+"));
|
||||||
|
if (a[0] == "Accepted" && a.length() > 1) {
|
||||||
|
_pairingCompleted(a[1]);
|
||||||
|
} else if (a[0] == "Connected" && a.length() > 1) {
|
||||||
|
_connectionCompleted(a[1]);
|
||||||
|
} else if (a[0] == "Connection" && a.length() > 1) {
|
||||||
|
setPairingStatus(PairingConnectionRejected, tr("Connection Rejected"));
|
||||||
|
qCDebug(PairingManagerLog) << "Connection error: " << str;
|
||||||
|
} else {
|
||||||
|
setPairingStatus(PairingRejected, tr("Pairing Rejected"));
|
||||||
|
qCDebug(PairingManagerLog) << "Pairing error: " << str;
|
||||||
|
}
|
||||||
|
_uploadManager->deleteLater();
|
||||||
|
_uploadManager = nullptr;
|
||||||
|
} else {
|
||||||
|
if(++_pairRetryCount > 3) {
|
||||||
|
qCDebug(PairingManagerLog) << "Giving up";
|
||||||
|
setPairingStatus(PairingError, tr("No Response From Vehicle"));
|
||||||
|
_uploadManager->deleteLater();
|
||||||
|
_uploadManager = nullptr;
|
||||||
|
} else {
|
||||||
|
qCDebug(PairingManagerLog) << "Upload error: " + reply->errorString();
|
||||||
|
_startUploadRequest();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_parsePairingJsonFile()
|
||||||
|
{
|
||||||
|
QFile file(_jsonFileName);
|
||||||
|
file.open(QIODevice::ReadOnly | QIODevice::Text);
|
||||||
|
QString json = file.readAll();
|
||||||
|
file.remove();
|
||||||
|
file.close();
|
||||||
|
|
||||||
|
jsonReceived(json);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::connectToPairedDevice(QString name)
|
||||||
|
{
|
||||||
|
setPairingStatus(PairingConnecting, tr("Connecting to %1").arg(name));
|
||||||
|
QFile file(_pairingCacheFile(name));
|
||||||
|
file.open(QIODevice::ReadOnly | QIODevice::Text);
|
||||||
|
QString json = file.readAll();
|
||||||
|
jsonReceived(json);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::removePairedDevice(QString name)
|
||||||
|
{
|
||||||
|
QFile file(_pairingCacheFile(name));
|
||||||
|
file.remove();
|
||||||
|
_updatePairedDeviceNameList();
|
||||||
|
emit pairedListChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_updatePairedDeviceNameList()
|
||||||
|
{
|
||||||
|
_deviceList.clear();
|
||||||
|
QDirIterator it(_pairingCacheDir().absolutePath(), QDir::Files);
|
||||||
|
while (it.hasNext()) {
|
||||||
|
QFileInfo fileInfo(it.next());
|
||||||
|
_deviceList.append(fileInfo.fileName());
|
||||||
|
qCDebug(PairingManagerLog) << "Listing: " << fileInfo.fileName();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QString
|
||||||
|
PairingManager::_assumeMicrohardPairingJson()
|
||||||
|
{
|
||||||
|
QJsonDocument json;
|
||||||
|
QJsonObject jsonObject;
|
||||||
|
|
||||||
|
jsonObject.insert("LT", "MH");
|
||||||
|
jsonObject.insert("IP", "192.168.168.10");
|
||||||
|
jsonObject.insert("AIP", _toolbox->microhardManager()->remoteIPAddr());
|
||||||
|
jsonObject.insert("CU", _toolbox->microhardManager()->configUserName());
|
||||||
|
jsonObject.insert("CP", _toolbox->microhardManager()->configPassword());
|
||||||
|
jsonObject.insert("EK", _toolbox->microhardManager()->encryptionKey());
|
||||||
|
json.setObject(jsonObject);
|
||||||
|
|
||||||
|
return QString(json.toJson(QJsonDocument::Compact));
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_parsePairingJson(QString jsonEnc)
|
||||||
|
{
|
||||||
|
QString json = QString::fromStdString(_aes.decrypt(jsonEnc.toStdString()));
|
||||||
|
if (json == "") {
|
||||||
|
json = jsonEnc;
|
||||||
|
}
|
||||||
|
qCDebug(PairingManagerLog) << "Parsing JSON: " << json;
|
||||||
|
|
||||||
|
_jsonDoc = QJsonDocument::fromJson(json.toUtf8());
|
||||||
|
|
||||||
|
if (_jsonDoc.isNull()) {
|
||||||
|
setPairingStatus(PairingError, tr("Invalid Pairing File"));
|
||||||
|
qCDebug(PairingManagerLog) << "Failed to create Pairing JSON doc.";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!_jsonDoc.isObject()) {
|
||||||
|
setPairingStatus(PairingError, tr("Error Parsing Pairing File"));
|
||||||
|
qCDebug(PairingManagerLog) << "Pairing JSON is not an object.";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
QJsonObject jsonObj = _jsonDoc.object();
|
||||||
|
|
||||||
|
if (jsonObj.isEmpty()) {
|
||||||
|
setPairingStatus(PairingError, tr("Error Parsing Pairing File"));
|
||||||
|
qCDebug(PairingManagerLog) << "Pairing JSON object is empty.";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_remotePairingMap = jsonObj.toVariantMap();
|
||||||
|
QString linkType = _remotePairingMap["LT"].toString();
|
||||||
|
QString pport = _remotePairingMap["PP"].toString();
|
||||||
|
if (pport.length()==0) {
|
||||||
|
pport = "29351";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (linkType.length()==0) {
|
||||||
|
setPairingStatus(PairingError, tr("Error Parsing Pairing File"));
|
||||||
|
qCDebug(PairingManagerLog) << "Pairing JSON is malformed.";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_toolbox->microhardManager()->switchToPairingEncryptionKey();
|
||||||
|
|
||||||
|
QString pairURL = "http://" + _remotePairingMap["IP"].toString() + ":" + pport;
|
||||||
|
bool connecting = jsonObj.contains("PWD");
|
||||||
|
QJsonDocument jsonDoc;
|
||||||
|
|
||||||
|
if (!connecting) {
|
||||||
|
pairURL += + "/pair";
|
||||||
|
QString pwd = random_string(8);
|
||||||
|
// TODO generate certificates
|
||||||
|
QString cert1 = "";
|
||||||
|
QString cert2 = "";
|
||||||
|
jsonObj.insert("PWD", pwd);
|
||||||
|
jsonObj.insert("CERT1", cert1);
|
||||||
|
jsonObj.insert("CERT2", cert2);
|
||||||
|
_jsonDoc.setObject(jsonObj);
|
||||||
|
if (linkType == "ZT") {
|
||||||
|
jsonDoc = _createZeroTierPairingJson(cert1);
|
||||||
|
} else if (linkType == "MH") {
|
||||||
|
jsonDoc = _createMicrohardPairingJson(pwd, cert1);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
pairURL += + "/connect";
|
||||||
|
QString cert2 = _remotePairingMap["CERT2"].toString();
|
||||||
|
if (linkType == "ZT") {
|
||||||
|
jsonDoc = _createZeroTierConnectJson(cert2);
|
||||||
|
} else if (linkType == "MH") {
|
||||||
|
jsonDoc = _createMicrohardConnectJson(cert2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (linkType == "ZT") {
|
||||||
|
_toolbox->settingsManager()->appSettings()->enableMicrohard()->setRawValue(false);
|
||||||
|
_toolbox->settingsManager()->appSettings()->enableTaisync()->setRawValue(false);
|
||||||
|
emit startUpload(pairURL, jsonDoc);
|
||||||
|
} else if (linkType == "MH") {
|
||||||
|
_toolbox->settingsManager()->appSettings()->enableMicrohard()->setRawValue(true);
|
||||||
|
_toolbox->settingsManager()->appSettings()->enableTaisync()->setRawValue(false);
|
||||||
|
if (_remotePairingMap.contains("AIP")) {
|
||||||
|
_toolbox->microhardManager()->setRemoteIPAddr(_remotePairingMap["AIP"].toString());
|
||||||
|
}
|
||||||
|
if (_remotePairingMap.contains("CU")) {
|
||||||
|
_toolbox->microhardManager()->setConfigUserName(_remotePairingMap["CU"].toString());
|
||||||
|
}
|
||||||
|
if (_remotePairingMap.contains("CP")) {
|
||||||
|
_toolbox->microhardManager()->setConfigPassword(_remotePairingMap["CP"].toString());
|
||||||
|
}
|
||||||
|
if (_remotePairingMap.contains("EK") && !connecting) {
|
||||||
|
_toolbox->microhardManager()->setEncryptionKey(_remotePairingMap["EK"].toString());
|
||||||
|
}
|
||||||
|
_toolbox->microhardManager()->updateSettings();
|
||||||
|
emit startUpload(pairURL, jsonDoc);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QString
|
||||||
|
PairingManager::_getLocalIPInNetwork(QString remoteIP, int num)
|
||||||
|
{
|
||||||
|
QStringList pieces = remoteIP.split(".");
|
||||||
|
QString ipPrefix = "";
|
||||||
|
for (int i = 0; i<num && i<pieces.length(); i++) {
|
||||||
|
ipPrefix += pieces[i] + ".";
|
||||||
|
}
|
||||||
|
|
||||||
|
const QHostAddress &localhost = QHostAddress(QHostAddress::LocalHost);
|
||||||
|
for (const QHostAddress &address: QNetworkInterface::allAddresses()) {
|
||||||
|
if (address.protocol() == QAbstractSocket::IPv4Protocol && address != localhost) {
|
||||||
|
if (address.toString().startsWith(ipPrefix)) {
|
||||||
|
return address.toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QDir
|
||||||
|
PairingManager::_pairingCacheDir()
|
||||||
|
{
|
||||||
|
const QString spath(QFileInfo(QSettings().fileName()).dir().absolutePath());
|
||||||
|
QDir dir = spath + QDir::separator() + "PairingCache";
|
||||||
|
if (!dir.exists()) {
|
||||||
|
dir.mkpath(".");
|
||||||
|
}
|
||||||
|
|
||||||
|
return dir;
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QString
|
||||||
|
PairingManager::_pairingCacheFile(QString uavName)
|
||||||
|
{
|
||||||
|
return _pairingCacheDir().filePath(uavName);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_writeJson(QJsonDocument &jsonDoc, QString fileName)
|
||||||
|
{
|
||||||
|
QString val = jsonDoc.toJson(QJsonDocument::JsonFormat::Compact);
|
||||||
|
qCDebug(PairingManagerLog) << "Write json " << val;
|
||||||
|
QString enc = QString::fromStdString(_aes.encrypt(val.toStdString()));
|
||||||
|
|
||||||
|
QFile file(fileName);
|
||||||
|
file.open(QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate);
|
||||||
|
file.write(enc.toUtf8());
|
||||||
|
file.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QJsonDocument
|
||||||
|
PairingManager::_createZeroTierPairingJson(QString cert1)
|
||||||
|
{
|
||||||
|
QString localIP = _getLocalIPInNetwork(_remotePairingMap["IP"].toString(), 2);
|
||||||
|
|
||||||
|
QJsonObject jsonObj;
|
||||||
|
jsonObj.insert("LT", "ZT");
|
||||||
|
jsonObj.insert("IP", localIP);
|
||||||
|
jsonObj.insert("P", 14550);
|
||||||
|
jsonObj.insert("CERT1", cert1);
|
||||||
|
return QJsonDocument(jsonObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QJsonDocument
|
||||||
|
PairingManager::_createMicrohardPairingJson(QString pwd, QString cert1)
|
||||||
|
{
|
||||||
|
QString localIP = _getLocalIPInNetwork(_remotePairingMap["IP"].toString(), 3);
|
||||||
|
|
||||||
|
QJsonObject jsonObj;
|
||||||
|
jsonObj.insert("LT", "MH");
|
||||||
|
jsonObj.insert("IP", localIP);
|
||||||
|
jsonObj.insert("P", 14550);
|
||||||
|
jsonObj.insert("PWD", pwd);
|
||||||
|
jsonObj.insert("CERT1", cert1);
|
||||||
|
return QJsonDocument(jsonObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QJsonDocument
|
||||||
|
PairingManager::_createZeroTierConnectJson(QString cert2)
|
||||||
|
{
|
||||||
|
QString localIP = _getLocalIPInNetwork(_remotePairingMap["IP"].toString(), 2);
|
||||||
|
|
||||||
|
QJsonObject jsonObj;
|
||||||
|
jsonObj.insert("LT", "ZT");
|
||||||
|
jsonObj.insert("IP", localIP);
|
||||||
|
jsonObj.insert("P", 14550);
|
||||||
|
jsonObj.insert("CERT2", cert2);
|
||||||
|
return QJsonDocument(jsonObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QJsonDocument
|
||||||
|
PairingManager::_createMicrohardConnectJson(QString cert2)
|
||||||
|
{
|
||||||
|
QString localIP = _getLocalIPInNetwork(_remotePairingMap["IP"].toString(), 3);
|
||||||
|
|
||||||
|
QJsonObject jsonObj;
|
||||||
|
jsonObj.insert("LT", "MH");
|
||||||
|
jsonObj.insert("IP", localIP);
|
||||||
|
jsonObj.insert("P", 14550);
|
||||||
|
jsonObj.insert("CERT2", cert2);
|
||||||
|
return QJsonDocument(jsonObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
QStringList
|
||||||
|
PairingManager::pairingLinkTypeStrings()
|
||||||
|
{
|
||||||
|
//-- Must follow same order as enum LinkType in LinkConfiguration.h
|
||||||
|
static QStringList list;
|
||||||
|
int i = 0;
|
||||||
|
if (!list.size()) {
|
||||||
|
#if defined QGC_ENABLE_QTNFC
|
||||||
|
list += tr("NFC");
|
||||||
|
_nfcIndex = i++;
|
||||||
|
#endif
|
||||||
|
#if defined QGC_GST_MICROHARD_ENABLED
|
||||||
|
list += tr("Microhard");
|
||||||
|
_microhardIndex = i++;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
return list;
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::_setPairingStatus(PairingStatus status, QString statusStr)
|
||||||
|
{
|
||||||
|
_status = status;
|
||||||
|
_statusString = statusStr;
|
||||||
|
emit pairingStatusChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
QString
|
||||||
|
PairingManager::pairingStatusStr() const
|
||||||
|
{
|
||||||
|
return _statusString;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if QGC_GST_MICROHARD_ENABLED
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::startMicrohardPairing()
|
||||||
|
{
|
||||||
|
stopPairing();
|
||||||
|
_pairRetryCount = 0;
|
||||||
|
setPairingStatus(PairingActive, tr("Pairing..."));
|
||||||
|
_parsePairingJson(_assumeMicrohardPairingJson());
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::stopPairing()
|
||||||
|
{
|
||||||
|
#if defined QGC_ENABLE_QTNFC
|
||||||
|
pairingNFC.stop();
|
||||||
|
#endif
|
||||||
|
_stopUpload();
|
||||||
|
setPairingStatus(PairingIdle, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
#if defined QGC_ENABLE_QTNFC
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingManager::startNFCScan()
|
||||||
|
{
|
||||||
|
stopPairing();
|
||||||
|
setPairingStatus(PairingActive, tr("Pairing..."));
|
||||||
|
pairingNFC.start();
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
#ifdef __android__
|
||||||
|
static const char kJniClassName[] {"org/mavlink/qgroundcontrol/QGCActivity"};
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
static void jniNFCTagReceived(JNIEnv *envA, jobject thizA, jstring messageA)
|
||||||
|
{
|
||||||
|
Q_UNUSED(thizA);
|
||||||
|
|
||||||
|
const char *stringL = envA->GetStringUTFChars(messageA, nullptr);
|
||||||
|
QString ndef = QString::fromUtf8(stringL);
|
||||||
|
envA->ReleaseStringUTFChars(messageA, stringL);
|
||||||
|
if (envA->ExceptionCheck())
|
||||||
|
envA->ExceptionClear();
|
||||||
|
qCDebug(PairingManagerLog) << "NDEF Tag Received: " << ndef;
|
||||||
|
qgcApp()->toolbox()->pairingManager()->jsonReceived(ndef);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void PairingManager::setNativeMethods(void)
|
||||||
|
{
|
||||||
|
// REGISTER THE C++ FUNCTION WITH JNI
|
||||||
|
JNINativeMethod javaMethods[] {
|
||||||
|
{"nativeNFCTagReceived", "(Ljava/lang/String;)V", reinterpret_cast<void *>(jniNFCTagReceived)}
|
||||||
|
};
|
||||||
|
|
||||||
|
QAndroidJniEnvironment jniEnv;
|
||||||
|
if (jniEnv->ExceptionCheck()) {
|
||||||
|
jniEnv->ExceptionDescribe();
|
||||||
|
jniEnv->ExceptionClear();
|
||||||
|
}
|
||||||
|
|
||||||
|
jclass objectClass = jniEnv->FindClass(kJniClassName);
|
||||||
|
if(!objectClass) {
|
||||||
|
qWarning() << "Couldn't find class:" << kJniClassName;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
jint val = jniEnv->RegisterNatives(objectClass, javaMethods, sizeof(javaMethods) / sizeof(javaMethods[0]));
|
||||||
|
|
||||||
|
if (val < 0) {
|
||||||
|
qWarning() << "Error registering methods: " << val;
|
||||||
|
} else {
|
||||||
|
qCDebug(PairingManagerLog) << "Native Functions Registered";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (jniEnv->ExceptionCheck()) {
|
||||||
|
jniEnv->ExceptionDescribe();
|
||||||
|
jniEnv->ExceptionClear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
@ -0,0 +1,153 @@
|
|||||||
|
/****************************************************************************
|
||||||
|
*
|
||||||
|
* (c) 2019 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
*
|
||||||
|
* QGroundControl is licensed according to the terms in the file
|
||||||
|
* COPYING.md in the root of the source code directory.
|
||||||
|
*
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <QJsonDocument>
|
||||||
|
#include <QMutex>
|
||||||
|
#include <QNetworkReply>
|
||||||
|
#include <QTimer>
|
||||||
|
#include <QTime>
|
||||||
|
#include <QVariantMap>
|
||||||
|
|
||||||
|
#include "aes.h"
|
||||||
|
#include "QGCToolbox.h"
|
||||||
|
#include "QGCLoggingCategory.h"
|
||||||
|
#include "Fact.h"
|
||||||
|
#if defined QGC_ENABLE_QTNFC
|
||||||
|
#include "QtNFC.h"
|
||||||
|
#endif
|
||||||
|
#ifdef __android__
|
||||||
|
#include <jni.h>
|
||||||
|
#include <QtAndroidExtras/QtAndroidExtras>
|
||||||
|
#include <QtAndroidExtras/QAndroidJniObject>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
Q_DECLARE_LOGGING_CATEGORY(PairingManagerLog)
|
||||||
|
|
||||||
|
class AppSettings;
|
||||||
|
class QGCApplication;
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
class PairingManager : public QGCTool
|
||||||
|
{
|
||||||
|
Q_OBJECT
|
||||||
|
public:
|
||||||
|
explicit PairingManager (QGCApplication* app, QGCToolbox* toolbox);
|
||||||
|
~PairingManager () override;
|
||||||
|
|
||||||
|
// Override from QGCTool
|
||||||
|
virtual void setToolbox(QGCToolbox *toolbox) override;
|
||||||
|
|
||||||
|
enum PairingStatus {
|
||||||
|
PairingIdle,
|
||||||
|
PairingActive,
|
||||||
|
PairingSuccess,
|
||||||
|
PairingConnecting,
|
||||||
|
PairingConnected,
|
||||||
|
PairingRejected,
|
||||||
|
PairingConnectionRejected,
|
||||||
|
PairingError
|
||||||
|
};
|
||||||
|
|
||||||
|
Q_ENUM(PairingStatus)
|
||||||
|
|
||||||
|
QStringList pairingLinkTypeStrings ();
|
||||||
|
QString pairingStatusStr () const;
|
||||||
|
QStringList pairedDeviceNameList () { return _deviceList; }
|
||||||
|
PairingStatus pairingStatus () { return _status; }
|
||||||
|
QString pairedVehicle () { return _lastPaired; }
|
||||||
|
int nfcIndex () { return _nfcIndex; }
|
||||||
|
int microhardIndex () { return _microhardIndex; }
|
||||||
|
bool firstBoot () { return _firstBoot; }
|
||||||
|
bool errorState () { return _status == PairingRejected || _status == PairingConnectionRejected || _status == PairingError; }
|
||||||
|
void setStatusMessage (PairingStatus status, QString statusStr) { emit setPairingStatus(status, statusStr); }
|
||||||
|
void jsonReceived (QString json) { emit parsePairingJson(json); }
|
||||||
|
void setFirstBoot (bool set) { _firstBoot = set; emit firstBootChanged(); }
|
||||||
|
#ifdef __android__
|
||||||
|
static void setNativeMethods (void);
|
||||||
|
#endif
|
||||||
|
Q_INVOKABLE void connectToPairedDevice (QString name);
|
||||||
|
Q_INVOKABLE void removePairedDevice (QString name);
|
||||||
|
|
||||||
|
#if defined defined QGC_ENABLE_QTNFC
|
||||||
|
Q_INVOKABLE void startNFCScan();
|
||||||
|
#endif
|
||||||
|
#if QGC_GST_MICROHARD_ENABLED
|
||||||
|
Q_INVOKABLE void startMicrohardPairing();
|
||||||
|
#endif
|
||||||
|
Q_INVOKABLE void stopPairing();
|
||||||
|
|
||||||
|
Q_PROPERTY(QString pairingStatusStr READ pairingStatusStr NOTIFY pairingStatusChanged)
|
||||||
|
Q_PROPERTY(PairingStatus pairingStatus READ pairingStatus NOTIFY pairingStatusChanged)
|
||||||
|
Q_PROPERTY(QStringList pairedDeviceNameList READ pairedDeviceNameList NOTIFY pairedListChanged)
|
||||||
|
Q_PROPERTY(QStringList pairingLinkTypeStrings READ pairingLinkTypeStrings CONSTANT)
|
||||||
|
Q_PROPERTY(QString pairedVehicle READ pairedVehicle NOTIFY pairedVehicleChanged)
|
||||||
|
Q_PROPERTY(bool errorState READ errorState NOTIFY pairingStatusChanged)
|
||||||
|
Q_PROPERTY(int nfcIndex READ nfcIndex CONSTANT)
|
||||||
|
Q_PROPERTY(int microhardIndex READ microhardIndex CONSTANT)
|
||||||
|
Q_PROPERTY(bool firstBoot READ firstBoot WRITE setFirstBoot NOTIFY firstBootChanged)
|
||||||
|
|
||||||
|
signals:
|
||||||
|
void startUpload (QString pairURL, QJsonDocument);
|
||||||
|
void closeConnection ();
|
||||||
|
void pairingConfigurationsChanged ();
|
||||||
|
void nameListChanged ();
|
||||||
|
void pairingStatusChanged ();
|
||||||
|
void parsePairingJson (QString json);
|
||||||
|
void setPairingStatus (PairingStatus status, QString pairingStatus);
|
||||||
|
void pairedListChanged ();
|
||||||
|
void pairedVehicleChanged ();
|
||||||
|
void firstBootChanged ();
|
||||||
|
|
||||||
|
private slots:
|
||||||
|
void _startUpload (QString pairURL, QJsonDocument);
|
||||||
|
void _stopUpload ();
|
||||||
|
void _startUploadRequest ();
|
||||||
|
void _parsePairingJson (QString jsonEnc);
|
||||||
|
void _setPairingStatus (PairingStatus status, QString pairingStatus);
|
||||||
|
|
||||||
|
private:
|
||||||
|
QString _statusString;
|
||||||
|
QString _jsonFileName;
|
||||||
|
QString _lastPaired;
|
||||||
|
QVariantMap _remotePairingMap;
|
||||||
|
int _nfcIndex = -1;
|
||||||
|
int _microhardIndex = -1;
|
||||||
|
int _pairRetryCount = 0;
|
||||||
|
PairingStatus _status = PairingIdle;
|
||||||
|
AES _aes;
|
||||||
|
QJsonDocument _jsonDoc{};
|
||||||
|
QMutex _uploadMutex{};
|
||||||
|
QNetworkAccessManager* _uploadManager = nullptr;
|
||||||
|
QString _uploadURL{};
|
||||||
|
QString _uploadData{};
|
||||||
|
bool _firstBoot = true;
|
||||||
|
QStringList _deviceList;
|
||||||
|
|
||||||
|
void _parsePairingJsonFile ();
|
||||||
|
QJsonDocument _createZeroTierConnectJson (QString cert2);
|
||||||
|
QJsonDocument _createMicrohardConnectJson (QString cert2);
|
||||||
|
QJsonDocument _createZeroTierPairingJson (QString cert1);
|
||||||
|
QJsonDocument _createMicrohardPairingJson (QString pwd, QString cert1);
|
||||||
|
QString _assumeMicrohardPairingJson ();
|
||||||
|
void _writeJson (QJsonDocument &jsonDoc, QString fileName);
|
||||||
|
QString _getLocalIPInNetwork (QString remoteIP, int num);
|
||||||
|
void _uploadFinished ();
|
||||||
|
void _uploadError (QNetworkReply::NetworkError code);
|
||||||
|
void _pairingCompleted (QString name);
|
||||||
|
void _connectionCompleted (QString name);
|
||||||
|
QDir _pairingCacheDir ();
|
||||||
|
QString _pairingCacheFile (QString uavName);
|
||||||
|
void _updatePairedDeviceNameList ();
|
||||||
|
|
||||||
|
#if defined QGC_ENABLE_QTNFC
|
||||||
|
PairingNFC pairingNFC;
|
||||||
|
#endif
|
||||||
|
};
|
||||||
@ -0,0 +1,135 @@
|
|||||||
|
/****************************************************************************
|
||||||
|
*
|
||||||
|
* (c) 2019 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
*
|
||||||
|
* QGroundControl is licensed according to the terms in the file
|
||||||
|
* COPYING.md in the root of the source code directory.
|
||||||
|
*
|
||||||
|
****************************************************************************/
|
||||||
|
#include "PairingManager.h"
|
||||||
|
#include "QtNFC.h"
|
||||||
|
#include "QGCApplication.h"
|
||||||
|
#include <QSoundEffect>
|
||||||
|
|
||||||
|
QGC_LOGGING_CATEGORY(PairingNFCLog, "PairingNFCLog")
|
||||||
|
|
||||||
|
#include <QNdefNfcTextRecord>
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
PairingNFC::PairingNFC()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::start()
|
||||||
|
{
|
||||||
|
if (manager != nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
qgcApp()->toolbox()->pairingManager()->setStatusMessage(tr("Waiting for NFC connection"));
|
||||||
|
qCDebug(PairingNFCLog) << "Waiting for NFC connection";
|
||||||
|
|
||||||
|
manager = new QNearFieldManager(this);
|
||||||
|
if (!manager->isAvailable()) {
|
||||||
|
qWarning() << "NFC not available";
|
||||||
|
delete manager;
|
||||||
|
manager = nullptr;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
QNdefFilter filter;
|
||||||
|
filter.setOrderMatch(false);
|
||||||
|
filter.appendRecord<QNdefNfcTextRecord>(1, UINT_MAX);
|
||||||
|
// type parameter cannot specify substring so filter for "image/" below
|
||||||
|
filter.appendRecord(QNdefRecord::Mime, QByteArray(), 0, 1);
|
||||||
|
|
||||||
|
int result = manager->registerNdefMessageHandler(filter, this, SLOT(handleMessage(QNdefMessage, QNearFieldTarget*)));
|
||||||
|
|
||||||
|
if (result < 0)
|
||||||
|
qWarning() << "Platform does not support NDEF message handler registration";
|
||||||
|
|
||||||
|
manager->startTargetDetection();
|
||||||
|
connect(manager, &QNearFieldManager::targetDetected, this, &PairingNFC::targetDetected);
|
||||||
|
connect(manager, &QNearFieldManager::targetLost, this, &PairingNFC::targetLost);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::stop()
|
||||||
|
{
|
||||||
|
if (manager != nullptr) {
|
||||||
|
qgcApp()->toolbox()->pairingManager()->setStatusMessage("");
|
||||||
|
qCDebug(PairingNFCLog) << "NFC: Stop";
|
||||||
|
manager->stopTargetDetection();
|
||||||
|
delete manager;
|
||||||
|
manager = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::targetDetected(QNearFieldTarget *target)
|
||||||
|
{
|
||||||
|
if (!target) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
qgcApp()->toolbox()->pairingManager()->setStatusMessage(tr("Device detected"));
|
||||||
|
qCDebug(PairingNFCLog) << "NFC: Device detected";
|
||||||
|
connect(target, &QNearFieldTarget::ndefMessageRead, this, &PairingNFC::handlePolledNdefMessage);
|
||||||
|
connect(target, SIGNAL(error(QNearFieldTarget::Error,QNearFieldTarget::RequestId)),
|
||||||
|
this, SLOT(targetError(QNearFieldTarget::Error,QNearFieldTarget::RequestId)));
|
||||||
|
connect(target, &QNearFieldTarget::requestCompleted, this, &PairingNFC::handleRequestCompleted);
|
||||||
|
|
||||||
|
manager->setTargetAccessModes(QNearFieldManager::NdefReadTargetAccess);
|
||||||
|
QNearFieldTarget::RequestId id = target->readNdefMessages();
|
||||||
|
if (target->waitForRequestCompleted(id)) {
|
||||||
|
qCDebug(PairingNFCLog) << "requestCompleted ";
|
||||||
|
QVariant res = target->requestResponse(id);
|
||||||
|
qCDebug(PairingNFCLog) << "Response: " << res.toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::handleRequestCompleted(const QNearFieldTarget::RequestId& id)
|
||||||
|
{
|
||||||
|
Q_UNUSED(id);
|
||||||
|
qCDebug(PairingNFCLog) << "handleRequestCompleted ";
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::targetError(QNearFieldTarget::Error error, const QNearFieldTarget::RequestId& id)
|
||||||
|
{
|
||||||
|
Q_UNUSED(id);
|
||||||
|
qCDebug(PairingNFCLog) << "Error: " << error;
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::targetLost(QNearFieldTarget *target)
|
||||||
|
{
|
||||||
|
qgcApp()->toolbox()->pairingManager()->setStatusMessage(tr("Device removed"));
|
||||||
|
qCDebug(PairingNFCLog) << "NFC: Device removed";
|
||||||
|
if (target) {
|
||||||
|
target->deleteLater();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
void
|
||||||
|
PairingNFC::handlePolledNdefMessage(QNdefMessage message)
|
||||||
|
{
|
||||||
|
qCDebug(PairingNFCLog) << "NFC: Handle NDEF message";
|
||||||
|
// QNearFieldTarget *target = qobject_cast<QNearFieldTarget *>(sender());
|
||||||
|
for (const QNdefRecord &record : message) {
|
||||||
|
if (record.isRecordType<QNdefNfcTextRecord>()) {
|
||||||
|
QNdefNfcTextRecord textRecord(record);
|
||||||
|
qgcApp()->toolbox()->pairingManager()->jsonReceived(textRecord.text());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
@ -0,0 +1,45 @@
|
|||||||
|
/****************************************************************************
|
||||||
|
*
|
||||||
|
* (c) 2019 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
|
||||||
|
*
|
||||||
|
* QGroundControl is licensed according to the terms in the file
|
||||||
|
* COPYING.md in the root of the source code directory.
|
||||||
|
*
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <QObject>
|
||||||
|
#include <QNdefMessage>
|
||||||
|
#include <QNearFieldManager>
|
||||||
|
#include <QNearFieldTarget>
|
||||||
|
|
||||||
|
#include "QGCLoggingCategory.h"
|
||||||
|
|
||||||
|
Q_DECLARE_LOGGING_CATEGORY(PairingNFCLog)
|
||||||
|
|
||||||
|
class PairingNFC : public QObject
|
||||||
|
{
|
||||||
|
Q_OBJECT
|
||||||
|
|
||||||
|
public:
|
||||||
|
PairingNFC();
|
||||||
|
|
||||||
|
void start();
|
||||||
|
|
||||||
|
void stop();
|
||||||
|
|
||||||
|
signals:
|
||||||
|
void parsePairingJson(QString json);
|
||||||
|
|
||||||
|
public slots:
|
||||||
|
void targetDetected(QNearFieldTarget *target);
|
||||||
|
void targetLost(QNearFieldTarget *target);
|
||||||
|
void handlePolledNdefMessage(QNdefMessage message);
|
||||||
|
void targetError(QNearFieldTarget::Error error, const QNearFieldTarget::RequestId& id);
|
||||||
|
void handleRequestCompleted(const QNearFieldTarget::RequestId& id);
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool _exitThread = false; ///< true: signal thread to exit
|
||||||
|
QNearFieldManager *manager = nullptr;
|
||||||
|
};
|
||||||
@ -0,0 +1,154 @@
|
|||||||
|
#include "aes.h"
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
#include <openssl/aes.h>
|
||||||
|
#include <openssl/bio.h>
|
||||||
|
#include <zlib.h>
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
AES::AES(std::string password, unsigned long long salt)
|
||||||
|
{
|
||||||
|
int nrounds = 5;
|
||||||
|
unsigned char key[32], iv[32];
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Gen key & IV for AES 256 CBC mode. A SHA1 digest is used to hash the supplied key material.
|
||||||
|
* nrounds is the number of times the we hash the material. More rounds are more secure but
|
||||||
|
* slower.
|
||||||
|
*/
|
||||||
|
EVP_BytesToKey(EVP_aes_256_cbc(), EVP_sha1(),
|
||||||
|
reinterpret_cast<const unsigned char*>(&salt),
|
||||||
|
reinterpret_cast<const unsigned char*>(password.c_str()),
|
||||||
|
static_cast<int>(password.length()),
|
||||||
|
nrounds, key, iv);
|
||||||
|
|
||||||
|
#if OPENSSL_VERSION_NUMBER >= 0x1010000fL
|
||||||
|
encCipherContext = EVP_CIPHER_CTX_new();
|
||||||
|
decCipherContext = EVP_CIPHER_CTX_new();
|
||||||
|
|
||||||
|
EVP_CIPHER_CTX_init(encCipherContext);
|
||||||
|
EVP_EncryptInit_ex(encCipherContext, EVP_aes_256_cbc(), nullptr, key, iv);
|
||||||
|
EVP_CIPHER_CTX_init(decCipherContext);
|
||||||
|
EVP_DecryptInit_ex(decCipherContext, EVP_aes_256_cbc(), nullptr, key, iv);
|
||||||
|
#else
|
||||||
|
EVP_CIPHER_CTX_init(&encCipherContext);
|
||||||
|
EVP_EncryptInit_ex(&encCipherContext, EVP_aes_256_cbc(), nullptr, key, iv);
|
||||||
|
EVP_CIPHER_CTX_init(&decCipherContext);
|
||||||
|
EVP_DecryptInit_ex(&decCipherContext, EVP_aes_256_cbc(), nullptr, key, iv);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
AES::~AES()
|
||||||
|
{
|
||||||
|
#if OPENSSL_VERSION_NUMBER >= 0x1010000fL
|
||||||
|
EVP_CIPHER_CTX_free(encCipherContext);
|
||||||
|
EVP_CIPHER_CTX_free(decCipherContext);
|
||||||
|
#else
|
||||||
|
EVP_CIPHER_CTX_cleanup(&encCipherContext);
|
||||||
|
EVP_CIPHER_CTX_cleanup(&decCipherContext);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
std::string
|
||||||
|
AES::encrypt(std::string plainText)
|
||||||
|
{
|
||||||
|
unsigned long sourceLen = static_cast<unsigned long>(plainText.length() + 1);
|
||||||
|
unsigned long destLen = sourceLen * 2;
|
||||||
|
unsigned char* compressed = new unsigned char[destLen];
|
||||||
|
int err = compress2(compressed, &destLen,
|
||||||
|
reinterpret_cast<const unsigned char *>(plainText.c_str()),
|
||||||
|
sourceLen, 9);
|
||||||
|
if (err != Z_OK) {
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
int pLen = static_cast<int>(destLen);
|
||||||
|
int cLen = pLen + AES_BLOCK_SIZE;
|
||||||
|
int fLen = 0;
|
||||||
|
unsigned char* cipherText = new unsigned char[cLen];
|
||||||
|
|
||||||
|
#if OPENSSL_VERSION_NUMBER >= 0x1010000fL
|
||||||
|
EVP_EncryptInit_ex(encCipherContext, nullptr, nullptr, nullptr, nullptr);
|
||||||
|
EVP_EncryptUpdate(encCipherContext, cipherText, &cLen, compressed, pLen);
|
||||||
|
EVP_EncryptFinal_ex(encCipherContext, cipherText + cLen, &fLen);
|
||||||
|
#else
|
||||||
|
EVP_EncryptInit_ex(&encCipherContext, nullptr, nullptr, nullptr, nullptr);
|
||||||
|
EVP_EncryptUpdate(&encCipherContext, cipherText, &cLen, compressed, pLen);
|
||||||
|
EVP_EncryptFinal_ex(&encCipherContext, cipherText + cLen, &fLen);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::vector<unsigned char> data(cipherText, cipherText + cLen + fLen);
|
||||||
|
std::string res = base64Encode(data);
|
||||||
|
delete[] cipherText;
|
||||||
|
delete[] compressed;
|
||||||
|
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
std::string
|
||||||
|
AES::decrypt(std::string cipherText)
|
||||||
|
{
|
||||||
|
int fLen = 0;
|
||||||
|
std::vector<unsigned char> text = base64Decode(cipherText);
|
||||||
|
int pLen = static_cast<int>(text.size());
|
||||||
|
unsigned char* plainText = new unsigned char[pLen];
|
||||||
|
|
||||||
|
#if OPENSSL_VERSION_NUMBER >= 0x1010000fL
|
||||||
|
EVP_DecryptInit_ex(decCipherContext, nullptr, nullptr, nullptr, nullptr);
|
||||||
|
EVP_DecryptUpdate(decCipherContext, plainText, &pLen, text.data(), pLen);
|
||||||
|
EVP_DecryptFinal_ex(decCipherContext, plainText + pLen, &fLen);
|
||||||
|
#else
|
||||||
|
EVP_DecryptInit_ex(&decCipherContext, nullptr, nullptr, nullptr, nullptr);
|
||||||
|
EVP_DecryptUpdate(&decCipherContext, plainText, &pLen, text.data(), pLen);
|
||||||
|
EVP_DecryptFinal_ex(&decCipherContext, plainText + pLen, &fLen);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
unsigned long destLen = static_cast<unsigned long>((pLen + fLen) * 2);
|
||||||
|
unsigned char* uncompressed = new unsigned char[destLen];
|
||||||
|
int err = uncompress(uncompressed, &destLen, plainText, static_cast<unsigned long>(pLen + fLen));
|
||||||
|
if (err != Z_OK) {
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string res(reinterpret_cast<char*>(uncompressed));
|
||||||
|
delete[] uncompressed;
|
||||||
|
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
struct BIOFreeAll { void operator()(BIO* p) { BIO_free_all(p); } };
|
||||||
|
|
||||||
|
std::string
|
||||||
|
AES::base64Encode(const std::vector<unsigned char>& binary)
|
||||||
|
{
|
||||||
|
std::unique_ptr<BIO, BIOFreeAll> b64(BIO_new(BIO_f_base64()));
|
||||||
|
BIO_set_flags(b64.get(), BIO_FLAGS_BASE64_NO_NL);
|
||||||
|
BIO* sink = BIO_new(BIO_s_mem());
|
||||||
|
BIO_push(b64.get(), sink);
|
||||||
|
BIO_write(b64.get(), binary.data(), static_cast<int>(binary.size()));
|
||||||
|
BIO_ctrl(b64.get(), BIO_CTRL_FLUSH, 0, nullptr);
|
||||||
|
const char* encoded;
|
||||||
|
const unsigned long len = static_cast<unsigned long>(BIO_ctrl(sink, BIO_CTRL_INFO, 0, &encoded));
|
||||||
|
|
||||||
|
return std::string(encoded, len);
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
std::vector<unsigned char>
|
||||||
|
AES::base64Decode(std::string encoded)
|
||||||
|
{
|
||||||
|
std::unique_ptr<BIO, BIOFreeAll> b64(BIO_new(BIO_f_base64()));
|
||||||
|
BIO_set_flags(b64.get(), BIO_FLAGS_BASE64_NO_NL);
|
||||||
|
BIO* source = BIO_new_mem_buf(encoded.c_str(), -1); // read-only source
|
||||||
|
BIO_push(b64.get(), source);
|
||||||
|
const unsigned long maxlen = encoded.length() / 4 * 3 + 1;
|
||||||
|
std::vector<unsigned char> decoded(maxlen);
|
||||||
|
const unsigned long len = static_cast<unsigned long>(BIO_read(b64.get(), decoded.data(), static_cast<int>(maxlen)));
|
||||||
|
decoded.resize(len);
|
||||||
|
return decoded;
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
@ -0,0 +1,35 @@
|
|||||||
|
#ifndef AES_H
|
||||||
|
#define AES_H
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <openssl/evp.h>
|
||||||
|
|
||||||
|
class AES
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
AES(std::string password, unsigned long long salt);
|
||||||
|
|
||||||
|
~AES();
|
||||||
|
|
||||||
|
std::string encrypt(std::string plainText);
|
||||||
|
|
||||||
|
std::string decrypt(std::string cipherText);
|
||||||
|
|
||||||
|
private:
|
||||||
|
#if OPENSSL_VERSION_NUMBER >= 0x1010000fL
|
||||||
|
EVP_CIPHER_CTX *encCipherContext = nullptr;
|
||||||
|
EVP_CIPHER_CTX *decCipherContext = nullptr;
|
||||||
|
#else
|
||||||
|
EVP_CIPHER_CTX encCipherContext;
|
||||||
|
EVP_CIPHER_CTX decCipherContext;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::string base64Encode(const std::vector<unsigned char>& binary);
|
||||||
|
|
||||||
|
std::vector<unsigned char> base64Decode(std::string encoded);
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // AES_H
|
||||||
Loading…
Reference in new issue