90 lines
3.5 KiB
C++
90 lines
3.5 KiB
C++
//
|
|
// Copyright (C) 2017 Privacore ApS - https://www.privacore.com
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero 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 Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
//
|
|
// License TL;DR: If you change this file, you must publish your changes.
|
|
//
|
|
#include "SiteNumInlinks.h"
|
|
#include "Conf.h"
|
|
#include "GbUtil.h"
|
|
#include "Log.h"
|
|
|
|
// The protocol is very simple.
|
|
// The server Receives queries in the form
|
|
// <query-id>:v1|sitehash<NL>
|
|
//
|
|
// The server responses:
|
|
// <query-id>:site_inlink_count<NL>
|
|
|
|
SiteNumInlinks g_siteNumInlinks;
|
|
|
|
struct SiteNumInlinksRequest : public FxClientRequest {
|
|
SiteNumInlinksRequest(void *context, int timeout_ms, site_inlinks_count_callback_t callback, unsigned sitehash)
|
|
: FxClientRequest(context, timeout_ms)
|
|
, m_callback(callback)
|
|
, m_sitehash(sitehash) {
|
|
}
|
|
|
|
site_inlinks_count_callback_t m_callback;
|
|
unsigned m_sitehash;
|
|
};
|
|
|
|
// v1|sitehash
|
|
void SiteNumInlinks::convertRequestToWireFormat(IOBuffer *out_buffer, uint32_t seq, fxclient_request_ptr_t base_request) {
|
|
std::shared_ptr<SiteNumInlinksRequest> request = std::dynamic_pointer_cast<SiteNumInlinksRequest>(base_request);
|
|
|
|
out_buffer->reserve_extra(8 + 1 + 3 + 8 + 1);
|
|
|
|
sprintf(out_buffer->end(), "%08x", seq);
|
|
out_buffer->push_back(8);
|
|
out_buffer->end()[0] = ':';
|
|
out_buffer->push_back(1);
|
|
|
|
memcpy(out_buffer->end(), "v1|", 3);
|
|
out_buffer->push_back(3);
|
|
|
|
sprintf(out_buffer->end(), "%08x", request->m_sitehash);
|
|
out_buffer->push_back(8);
|
|
out_buffer->end()[0] = '\n';
|
|
out_buffer->push_back(1);
|
|
}
|
|
|
|
void SiteNumInlinks::processResponse(fxclient_request_ptr_t base_request, char *response) {
|
|
std::shared_ptr<SiteNumInlinksRequest> request = std::dynamic_pointer_cast<SiteNumInlinksRequest>(base_request);
|
|
logTrace(g_conf.m_logTraceSiteNumInlinks, "Got result='%s' for sitehash=%x", response, request->m_sitehash);
|
|
|
|
unsigned long site_num_inlinks = strtoul(response, nullptr, 10);
|
|
(request->m_callback)(request->m_context, site_num_inlinks);
|
|
}
|
|
|
|
void SiteNumInlinks::errorCallback(fxclient_request_ptr_t base_request) {
|
|
std::shared_ptr<SiteNumInlinksRequest> request = std::dynamic_pointer_cast<SiteNumInlinksRequest>(base_request);
|
|
request->m_callback(request->m_context, {});
|
|
}
|
|
|
|
bool SiteNumInlinks::initialize() {
|
|
return FxClient::initialize("site num inlinks", "sitenum", g_conf.m_siteNumInlinksServerName, g_conf.m_siteNumInlinksServerPort,
|
|
g_conf.m_maxOutstandingSiteNumInlinks, g_conf.m_logTraceSiteNumInlinks);
|
|
}
|
|
|
|
void SiteNumInlinks::reinitializeSettings() {
|
|
FxClient::reinitializeSettings(g_conf.m_siteNumInlinksServerName, g_conf.m_siteNumInlinksServerPort,
|
|
g_conf.m_maxOutstandingSiteNumInlinks, g_conf.m_logTraceSiteNumInlinks);
|
|
}
|
|
|
|
bool SiteNumInlinks::getSiteNumInlinks(void *context, site_inlinks_count_callback_t callback, unsigned sitehash) {
|
|
return sendRequest(std::static_pointer_cast<FxClientRequest>(std::make_shared<SiteNumInlinksRequest>(context, g_conf.m_siteNumInlinksTimeout, callback, sitehash)));
|
|
}
|