mirror/transmission
1
0
Fork 0
mirror of https://github.com/transmission/transmission synced 2025-02-22 06:00:41 +00:00
Code Issues Releases Wiki Activity

refactor: remove raw ptr variant of tr_peerMgrCompactToPex() (#2556)

Browse source
This commit is contained in:
Charles Kerr 2022-01-31 21:11:47 -06:00 committed by GitHub
parent b645c4cfc5
commit cf2c32e1ce
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 18 additions and 98 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

68
libtransmission/peer-mgr.cc
View file

@ -7,9 +7,8 @@
#include <cerrno> /* error codes ERANGE, ... */ #include <cerrno> /* error codes ERANGE, ... */
#include <climits> /* INT_MAX */ #include <climits> /* INT_MAX */
#include <cstdlib> /* qsort */ #include <cstdlib> /* qsort */
#include <cstring> /* memcpy, memcmp, strstr */ #include <ctime> // time_t
#include <ctime> #include <iterator> // std::back_inserter
#include <iterator>
#include <vector> #include <vector>
#include <event2/event.h> #include <event2/event.h>
@ -20,6 +19,7 @@
#define LIBTRANSMISSION_PEER_MODULE #define LIBTRANSMISSION_PEER_MODULE
#include "transmission.h" #include "transmission.h"
#include "announcer.h" #include "announcer.h"
#include "bandwidth.h" #include "bandwidth.h"
#include "blocklist.h" #include "blocklist.h"
@ -31,9 +31,9 @@
#include "log.h" #include "log.h"
#include "net.h" #include "net.h"
#include "peer-io.h" #include "peer-io.h"
#include "peer-mgr.h"
#include "peer-mgr-active-requests.h" #include "peer-mgr-active-requests.h"
#include "peer-mgr-wishlist.h" #include "peer-mgr-wishlist.h"
#include "peer-mgr.h"
#include "peer-msgs.h" #include "peer-msgs.h"
#include "ptrarray.h" #include "ptrarray.h"
#include "session.h" #include "session.h"
@ -1171,35 +1171,6 @@ size_t tr_peerMgrAddPex(tr_torrent* tor, uint8_t from, tr_pex const* pex, size_t
return n_used; return n_used;
} }
tr_pex* tr_peerMgrCompactToPex(
void const* compact,
size_t compactLen,
uint8_t const* added_f,
size_t added_f_len,
size_t* pexCount)
{
size_t n = compactLen / 6;
auto const* walk = static_cast<uint8_t const*>(compact);
auto* const pex = tr_new0(tr_pex, n);
for (size_t i = 0; i < n; ++i)
{
pex[i].addr.type = TR_AF_INET;
memcpy(&pex[i].addr.addr, walk, 4);
walk += 4;
memcpy(&pex[i].port, walk, 2);
walk += 2;
if (added_f != nullptr && n == added_f_len)
{
pex[i].flags = added_f[i];
}
}
*pexCount = n;
return pex;
}
std::vector<tr_pex> tr_peerMgrCompactToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len) std::vector<tr_pex> tr_peerMgrCompactToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len)
{ {
size_t n = compactLen / 6; size_t n = compactLen / 6;
@ -1223,35 +1194,6 @@ std::vector<tr_pex> tr_peerMgrCompactToPex(void const* compact, size_t compactLe
return pex; return pex;
} }
tr_pex* tr_peerMgrCompact6ToPex(
void const* compact,
size_t compactLen,
uint8_t const* added_f,
size_t added_f_len,
size_t* pexCount)
{
size_t n = compactLen / 18;
auto const* walk = static_cast<uint8_t const*>(compact);
auto* const pex = tr_new0(tr_pex, n);
for (size_t i = 0; i < n; ++i)
{
pex[i].addr.type = TR_AF_INET6;
memcpy(&pex[i].addr.addr.addr6.s6_addr, walk, 16);
walk += 16;
memcpy(&pex[i].port, walk, 2);
walk += 2;
if (added_f != nullptr && n == added_f_len)
{
pex[i].flags = added_f[i];
}
}
*pexCount = n;
return pex;
}
std::vector<tr_pex> tr_peerMgrCompact6ToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len) std::vector<tr_pex> tr_peerMgrCompact6ToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len)
{ {
size_t n = compactLen / 18; size_t n = compactLen / 18;
@ -1612,7 +1554,7 @@ void tr_peerMgrTorrentAvailability(tr_torrent const* tor, int8_t* tab, unsigned
TR_ASSERT(tab != nullptr); TR_ASSERT(tab != nullptr);
TR_ASSERT(tabCount > 0); TR_ASSERT(tabCount > 0);
memset(tab, 0, tabCount); std::fill_n(tab, tabCount, int8_t{});
if (tor->hasMetadata()) if (tor->hasMetadata())
{ {

14
libtransmission/peer-mgr.h
View file

@ -102,20 +102,6 @@ size_t tr_peerMgrCountActiveRequestsToPeer(tr_torrent const* torrent, tr_peer co
void tr_peerMgrAddIncoming(tr_peerMgr* manager, tr_address const* addr, tr_port port, struct tr_peer_socket const socket); void tr_peerMgrAddIncoming(tr_peerMgr* manager, tr_address const* addr, tr_port port, struct tr_peer_socket const socket);
tr_pex* tr_peerMgrCompactToPex(
void const* compact,
size_t compactLen,
uint8_t const* added_f,
size_t added_f_len,
size_t* setme_pex_count);
tr_pex* tr_peerMgrCompact6ToPex(
void const* compact,
size_t compactLen,
uint8_t const* added_f,
size_t added_f_len,
size_t* pexCount);
std::vector<tr_pex> tr_peerMgrCompactToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len); std::vector<tr_pex> tr_peerMgrCompactToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len);
std::vector<tr_pex> tr_peerMgrCompact6ToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len); std::vector<tr_pex> tr_peerMgrCompact6ToPex(void const* compact, size_t compactLen, uint8_t const* added_f, size_t added_f_len);

22
libtransmission/peer-msgs.cc
View file

@ -125,7 +125,7 @@ static auto constexpr RequestBufSecs = int{ 10 };
namespace namespace
{ {
constexpr int MAX_PEX_PEER_COUNT = 50; auto constexpr MaxPexPeerCount = size_t{ 50 };
} // unnamed namespace } // unnamed namespace
@ -1323,11 +1323,9 @@ static void parseUtPex(tr_peerMsgsImpl* msgs, uint32_t msglen, struct evbuffer*
added_f = nullptr; added_f = nullptr;
} }
auto n = size_t{}; auto pex = tr_peerMgrCompactToPex(added, added_len, added_f, added_f_len);
tr_pex* const pex = tr_peerMgrCompactToPex(added, added_len, added_f, added_f_len, &n); pex.resize(std::min(MaxPexPeerCount, std::size(pex)));
n = std::min(n, size_t{ MAX_PEX_PEER_COUNT }); tr_peerMgrAddPex(tor, TR_PEER_FROM_PEX, std::data(pex), std::size(pex));
tr_peerMgrAddPex(tor, TR_PEER_FROM_PEX, pex, n);
tr_free(pex);
} }
if (tr_variantDictFindRaw(&val, TR_KEY_added6, &added, &added_len)) if (tr_variantDictFindRaw(&val, TR_KEY_added6, &added, &added_len))
@ -1340,11 +1338,9 @@ static void parseUtPex(tr_peerMsgsImpl* msgs, uint32_t msglen, struct evbuffer*
added_f = nullptr; added_f = nullptr;
} }
auto n = size_t{}; auto pex = tr_peerMgrCompact6ToPex(added, added_len, added_f, added_f_len);
tr_pex* const pex = tr_peerMgrCompact6ToPex(added, added_len, added_f, added_f_len, &n); pex.resize(std::min(MaxPexPeerCount, std::size(pex)));
n = std::min(n, size_t{ MAX_PEX_PEER_COUNT }); tr_peerMgrAddPex(tor, TR_PEER_FROM_PEX, std::data(pex), std::size(pex));
tr_peerMgrAddPex(tor, TR_PEER_FROM_PEX, pex, n);
tr_free(pex);
} }
tr_variantFree(&val); tr_variantFree(&val);
@ -2483,8 +2479,8 @@ static void sendPex(tr_peerMsgsImpl* msgs)
PexDiffs diffs6; PexDiffs diffs6;
tr_pex* newPex = nullptr; tr_pex* newPex = nullptr;
tr_pex* newPex6 = nullptr; tr_pex* newPex6 = nullptr;
int const newCount = tr_peerMgrGetPeers(msgs->torrent, &newPex, TR_AF_INET, TR_PEERS_CONNECTED, MAX_PEX_PEER_COUNT); int const newCount = tr_peerMgrGetPeers(msgs->torrent, &newPex, TR_AF_INET, TR_PEERS_CONNECTED, MaxPexPeerCount);
int const newCount6 = tr_peerMgrGetPeers(msgs->torrent, &newPex6, TR_AF_INET6, TR_PEERS_CONNECTED, MAX_PEX_PEER_COUNT); int const newCount6 = tr_peerMgrGetPeers(msgs->torrent, &newPex6, TR_AF_INET6, TR_PEERS_CONNECTED, MaxPexPeerCount);
/* build the diffs */ /* build the diffs */
diffs.added = tr_new(tr_pex, newCount); diffs.added = tr_new(tr_pex, newCount);

12
libtransmission/tr-dht.cc
View file

@ -609,14 +609,10 @@ static void callback(void* /*ignore*/, int event, unsigned char const* info_hash
{ {
if (tor != nullptr && tor->allowsDht()) if (tor != nullptr && tor->allowsDht())
{ {
size_t n = 0; auto const pex = event == DHT_EVENT_VALUES ? tr_peerMgrCompactToPex(data, data_len, nullptr, 0) :
tr_pex* const pex = event == DHT_EVENT_VALUES ? tr_peerMgrCompactToPex(data, data_len, nullptr, 0, &n) : tr_peerMgrCompact6ToPex(data, data_len, nullptr, 0);
tr_peerMgrCompact6ToPex(data, data_len, nullptr, 0, &n); tr_peerMgrAddPex(tor, TR_PEER_FROM_DHT, std::data(pex), std::size(pex));
tr_logAddTorDbg(tor, "Learned %zu %s peers from DHT", std::size(pex), event == DHT_EVENT_VALUES6 ? "IPv6" : "IPv4");
tr_peerMgrAddPex(tor, TR_PEER_FROM_DHT, pex, n);
tr_free(pex);
tr_logAddTorDbg(tor, "Learned %d %s peers from DHT", (int)n, event == DHT_EVENT_VALUES6 ? "IPv6" : "IPv4");
} }
} }
else if (event == DHT_EVENT_SEARCH_DONE || event == DHT_EVENT_SEARCH_DONE6) else if (event == DHT_EVENT_SEARCH_DONE || event == DHT_EVENT_SEARCH_DONE6)