130 lines
5.5 KiB
C++
130 lines
5.5 KiB
C++
/* XMRig
|
|
* Copyright (c) 2018-2021 SChernykh <https://github.com/SChernykh>
|
|
* Copyright (c) 2016-2021 XMRig <https://github.com/xmrig>, <support@xmrig.com>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef XMRIG_BENCHCLIENT_H
|
|
#define XMRIG_BENCHCLIENT_H
|
|
|
|
|
|
#include "backend/common/interfaces/IBenchListener.h"
|
|
#include "base/kernel/interfaces/IDnsListener.h"
|
|
#include "base/kernel/interfaces/IHttpListener.h"
|
|
#include "base/net/stratum/Client.h"
|
|
|
|
|
|
namespace xmrig {
|
|
|
|
|
|
class BenchClient : public IClient, public IHttpListener, public IBenchListener, public IDnsListener
|
|
{
|
|
public:
|
|
XMRIG_DISABLE_COPY_MOVE_DEFAULT(BenchClient)
|
|
|
|
BenchClient(const std::shared_ptr<BenchConfig> &benchmark, IClientListener* listener);
|
|
~BenchClient() override;
|
|
|
|
inline bool disconnect() override { return true; }
|
|
inline bool hasExtension(Extension) const noexcept override { return false; }
|
|
inline bool isEnabled() const override { return true; }
|
|
inline bool isTLS() const override { return false; }
|
|
inline const char *mode() const override { return "benchmark"; }
|
|
inline const char *tlsFingerprint() const override { return nullptr; }
|
|
inline const char *tlsVersion() const override { return nullptr; }
|
|
inline const Job &job() const override { return m_job; }
|
|
inline const Pool &pool() const override { return m_pool; }
|
|
inline const String &ip() const override { return m_ip; }
|
|
inline int id() const override { return 0; }
|
|
inline int64_t send(const rapidjson::Value &, Callback) override { return 0; }
|
|
inline int64_t send(const rapidjson::Value &) override { return 0; }
|
|
inline int64_t sequence() const override { return 0; }
|
|
inline int64_t submit(const JobResult &) override { return 0; }
|
|
inline void connect(const Pool &pool) override { setPool(pool); }
|
|
inline void deleteLater() override { delete this; }
|
|
inline void setAlgo(const Algorithm &algo) override {}
|
|
inline void setEnabled(bool enabled) override {}
|
|
inline void setProxy(const ProxyUrl &proxy) override {}
|
|
inline void setQuiet(bool quiet) override {}
|
|
inline void setRetries(int retries) override {}
|
|
inline void setRetryPause(uint64_t ms) override {}
|
|
inline void tick(uint64_t now) override {}
|
|
|
|
const char *tag() const override;
|
|
void connect() override;
|
|
void setPool(const Pool &pool) override;
|
|
|
|
protected:
|
|
void onBenchDone(uint64_t result, uint64_t diff, uint64_t ts) override;
|
|
void onBenchReady(uint64_t ts, uint32_t threads, const IBackend *backend) override;
|
|
void onHttpData(const HttpData &data) override;
|
|
void onResolved(const DnsRecords &records, int status, const char *error) override;
|
|
|
|
private:
|
|
enum Mode : uint32_t {
|
|
STATIC_BENCH,
|
|
ONLINE_BENCH,
|
|
STATIC_VERIFY,
|
|
ONLINE_VERIFY
|
|
};
|
|
|
|
enum Request : uint32_t {
|
|
NO_REQUEST,
|
|
GET_BENCH,
|
|
CREATE_BENCH,
|
|
START_BENCH,
|
|
DONE_BENCH
|
|
};
|
|
|
|
bool setSeed(const char *seed);
|
|
uint64_t referenceHash() const;
|
|
void printExit() const;
|
|
void start();
|
|
|
|
# ifdef XMRIG_FEATURE_HTTP
|
|
void onCreateReply(const rapidjson::Value &value);
|
|
void onDoneReply(const rapidjson::Value &value);
|
|
void onGetReply(const rapidjson::Value &value);
|
|
void resolve();
|
|
void send(Request request);
|
|
void setError(const char *message, const char *label = nullptr);
|
|
void update(const rapidjson::Value &body);
|
|
# endif
|
|
|
|
const IBackend *m_backend = nullptr;
|
|
IClientListener* m_listener;
|
|
Job m_job;
|
|
Mode m_mode = STATIC_BENCH;
|
|
Pool m_pool;
|
|
Request m_request = NO_REQUEST;
|
|
std::shared_ptr<BenchConfig> m_benchmark;
|
|
std::shared_ptr<DnsRequest> m_dns;
|
|
std::shared_ptr<IHttpListener> m_httpListener;
|
|
String m_ip;
|
|
String m_token;
|
|
uint32_t m_threads = 0;
|
|
uint64_t m_diff = 0;
|
|
uint64_t m_doneTime = 0;
|
|
uint64_t m_hash = 0;
|
|
uint64_t m_readyTime = 0;
|
|
uint64_t m_result = 0;
|
|
uint64_t m_startTime = 0;
|
|
};
|
|
|
|
|
|
} /* namespace xmrig */
|
|
|
|
|
|
#endif /* XMRIG_BENCHCLIENT_H */
|