63 lines
1.4 KiB
C++
63 lines
1.4 KiB
C++
|
#include "Challenge.hpp"
|
||
|
|
|
||
|
|
#include "Crypto.hpp"
|
||
|
|
|
||
|
|
#include <fmt/format.h>
|
||
|
|
#include <glaze/glaze.hpp>
|
||
|
|
|
||
|
|
constexpr const uint64_t CHALLENGE_VERSION = 1;
|
||
|
|
|
||
|
|
CChallenge::CChallenge(const std::string& fingerprint, const std::string& challenge, int difficulty) :
|
||
|
|
m_fingerprint(fingerprint), m_challenge(challenge), m_difficulty(difficulty) {
|
||
|
|
std::string toSign = getSigString();
|
||
|
|
|
||
|
|
m_sig = g_pCrypto->sign(toSign);
|
||
|
|
|
||
|
|
m_valid = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
CChallenge::CChallenge(const std::string& jsonResponse) {
|
||
|
|
auto json = glz::read_json<SChallengeJSON>(jsonResponse);
|
||
|
|
|
||
|
|
if (!json.has_value())
|
||
|
|
return;
|
||
|
|
|
||
|
|
SChallengeJSON s = json.value();
|
||
|
|
|
||
|
|
m_challenge = s.challenge;
|
||
|
|
m_fingerprint = s.fingerprint;
|
||
|
|
m_sig = s.sig;
|
||
|
|
|
||
|
|
if (!g_pCrypto->verifySignature(getSigString(), m_sig))
|
||
|
|
return;
|
||
|
|
|
||
|
|
const auto SHA = g_pCrypto->sha256(m_challenge + std::to_string(s.solution));
|
||
|
|
|
||
|
|
for (size_t i = 0; i < m_difficulty; ++i) {
|
||
|
|
if (SHA.at(i) != '0')
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
m_valid = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string CChallenge::fingerprint() const {
|
||
|
|
return m_fingerprint;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string CChallenge::challenge() const {
|
||
|
|
return m_challenge;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string CChallenge::signature() const {
|
||
|
|
return m_sig;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool CChallenge::valid() const {
|
||
|
|
return m_valid;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string CChallenge::getSigString() {
|
||
|
|
return fmt::format("{}-{},{}", CHALLENGE_VERSION, m_fingerprint, m_challenge);
|
||
|
|
}
|