Git-Mirrors/keepassxc
1
0
Fork 0
mirror of https://github.com/keepassxreboot/keepassxc.git synced 2024-12-27 00:09:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

Make challenge() member thread-safe

Browse Source
This commit is contained in:
Janek Bevendorff 2017-02-24 00:15:57 +01:00
parent 093fe5c7ef
commit 70816f90b2
No known key found for this signature in database
GPG Key ID: CFEC2F6850BFFA53
2 changed files with 26 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/keys/drivers/YubiKey.cpp
View File

@ -107,8 +107,10 @@ void YubiKey::detect()
QByteArray resp;
result = challenge(i, false, rand, resp);
if (result != YubiKey::ERROR) {
if (result == YubiKey::ALREADY_RUNNING) {
emit alreadyRunning();
return;
} else if (result != YubiKey::ERROR) {
emit detected(i, result == YubiKey::WOULDBLOCK ? true : false);
return;
}
@ -141,13 +143,18 @@ static inline QString printByteArray(const QByteArray& a)
}
#endif
YubiKey::ChallengeResult YubiKey::challenge(int slot, bool mayBlock, const QByteArray& chal, QByteArray& resp) const
YubiKey::ChallengeResult YubiKey::challenge(int slot, bool mayBlock, const QByteArray& chal, QByteArray& resp)
{
if (!m_mutex.tryLock()) {
return ALREADY_RUNNING;
}
int yk_cmd = (slot == 1) ? SLOT_CHAL_HMAC1 : SLOT_CHAL_HMAC2;
QByteArray paddedChal = chal;
/* Ensure that YubiKey::init() succeeded */
// ensure that YubiKey::init() succeeded
if (m_yk == NULL) {
m_mutex.unlock();
return ERROR;
}
@ -171,13 +178,12 @@ YubiKey::ChallengeResult YubiKey::challenge(int slot, bool mayBlock, const QByte
r = reinterpret_cast<unsigned char*>(resp.data());
#ifdef QT_DEBUG
qDebug().nospace() << __func__ << "(" << slot << ") c = "
<< printByteArray(paddedChal);
qDebug().nospace() << __func__ << "(" << slot << ") c = " << printByteArray(paddedChal);
#endif
int ret = yk_challenge_response(m_yk, yk_cmd, mayBlock,
paddedChal.size(), c,
resp.size(), r);
int ret = yk_challenge_response(m_yk, yk_cmd, mayBlock, paddedChal.size(), c, resp.size(), r);
m_mutex.unlock();
if (!ret) {
if (yk_errno == YK_EWOULDBLOCK) {
@ -206,8 +212,7 @@ YubiKey::ChallengeResult YubiKey::challenge(int slot, bool mayBlock, const QByte
resp.resize(20);
#ifdef QT_DEBUG
qDebug().nospace() << __func__ << "(" << slot << ") r = "
<< printByteArray(resp) << ", ret = " << ret;
qDebug().nospace() << __func__ << "(" << slot << ") r = " << printByteArray(resp) << ", ret = " << ret;
#endif
return SUCCESS;

12
src/keys/drivers/YubiKey.h
View File

@ -19,6 +19,7 @@
#define KEEPASSX_YUBIKEY_H
#include <QObject>
#include <QMutex>
/**
* Singleton class to manage the interface to the hardware
@ -28,7 +29,7 @@ class YubiKey : public QObject
Q_OBJECT
public:
enum ChallengeResult { ERROR = -1, SUCCESS = 0, WOULDBLOCK };
enum ChallengeResult { ERROR = -1, SUCCESS = 0, WOULDBLOCK, ALREADY_RUNNING };
/**
* @brief YubiKey::instance - get instance of singleton
@ -64,7 +65,7 @@ public:
*/
ChallengeResult challenge(int slot, bool mayBlock,
const QByteArray& chal,
QByteArray& resp) const;
QByteArray& resp);
/**
* @brief YubiKey::getSerial - serial number of YubiKey
@ -92,6 +93,11 @@ Q_SIGNALS:
*/
void notFound();
/**
* Emitted when detection is already running.
*/
void alreadyRunning();
private:
explicit YubiKey();
static YubiKey* m_instance;
@ -100,6 +106,8 @@ private:
void* m_yk_void;
void* m_ykds_void;
QMutex m_mutex;
Q_DISABLE_COPY(YubiKey)
};