mirror of
https://github.com/eosswedenorg/antelope-keygen
synced 2026-06-18 04:00:03 +02:00
125 lines
2.9 KiB
C++
125 lines
2.9 KiB
C++
/**
|
|
* MIT License
|
|
*
|
|
* Copyright (c) 2019-2021 EOS Sw/eden
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
#include <libeosio/ec.hpp>
|
|
#include <libeosio/WIF.hpp>
|
|
#include <eoskeygen/config.hpp>
|
|
#include <eoskeygen/core/string.hpp>
|
|
#include <eoskeygen/key_search_result.hpp>
|
|
#include <eoskeygen/key_search.hpp>
|
|
|
|
namespace eoskeygen {
|
|
|
|
KeySearch::KeySearch() :
|
|
m_max (0),
|
|
m_count (0),
|
|
#ifdef EOSIOKEYGEN_HAVE_THREADS
|
|
m_threads (0),
|
|
#endif
|
|
m_callback (NULL)
|
|
{
|
|
}
|
|
|
|
void KeySearch::addWord(const std::string& str)
|
|
{
|
|
std::string tmp = str;
|
|
strtolower(tmp);
|
|
m_words.push_back(tmp);
|
|
}
|
|
|
|
void KeySearch::addList(const strlist_t& list)
|
|
{
|
|
for(const std::string& item : list) {
|
|
addWord(item);
|
|
}
|
|
}
|
|
|
|
const strlist_t& KeySearch::getList()
|
|
{
|
|
return m_words;
|
|
}
|
|
|
|
void KeySearch::clear()
|
|
{
|
|
m_words.clear();
|
|
}
|
|
|
|
void KeySearch::setCallback(IKeySearchResult* callback)
|
|
{
|
|
m_callback = callback;
|
|
}
|
|
|
|
void KeySearch::_search_linear()
|
|
{
|
|
struct libeosio::ec_keypair pair;
|
|
|
|
while (m_count < m_max) {
|
|
struct result res;
|
|
libeosio::ec_generate_key(&pair);
|
|
if (_contains_word(&pair, res)) {
|
|
m_callback->onResult(&pair, res);
|
|
m_count++;
|
|
}
|
|
}
|
|
}
|
|
|
|
void KeySearch::abort()
|
|
{
|
|
// exit find() operation by setting m_max to zero.
|
|
m_max = 0;
|
|
}
|
|
|
|
void KeySearch::find(size_t num_results)
|
|
{
|
|
m_count = 0;
|
|
m_max = num_results;
|
|
|
|
#ifdef EOSIOKEYGEN_HAVE_THREADS
|
|
// Only do multithread if number of threads makes sense.
|
|
if (m_threads >= 2) {
|
|
_search_mt();
|
|
return;
|
|
}
|
|
#endif /* HAVE_THREADS */
|
|
|
|
_search_linear();
|
|
}
|
|
|
|
bool KeySearch::_contains_word(const struct libeosio::ec_keypair* key, struct result& result) {
|
|
|
|
// skip first 3 chars, as those are always "EOS"
|
|
std::string pubstr = libeosio::wif_pub_encode(key->pub).substr(3);
|
|
strtolower(pubstr);
|
|
|
|
for(auto const& w: m_words) {
|
|
size_t p = pubstr.find(w);
|
|
if (p != std::string::npos) {
|
|
result.pos = p + 3;
|
|
result.len = w.length();
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
} // namespace eoskeygen
|