mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-22 15:04:44 +01:00
refactor: Move port mapping code to its own module
This commit does not change behavior.
This commit is contained in:
parent
4b8b71e630
commit
02ccf69dd6
8 changed files with 163 additions and 135 deletions
|
@ -152,6 +152,7 @@ BITCOIN_CORE_H = \
|
||||||
key_io.h \
|
key_io.h \
|
||||||
logging.h \
|
logging.h \
|
||||||
logging/timer.h \
|
logging/timer.h \
|
||||||
|
mapport.h \
|
||||||
memusage.h \
|
memusage.h \
|
||||||
merkleblock.h \
|
merkleblock.h \
|
||||||
miner.h \
|
miner.h \
|
||||||
|
@ -299,6 +300,7 @@ libbitcoin_server_a_SOURCES = \
|
||||||
index/blockfilterindex.cpp \
|
index/blockfilterindex.cpp \
|
||||||
index/txindex.cpp \
|
index/txindex.cpp \
|
||||||
init.cpp \
|
init.cpp \
|
||||||
|
mapport.cpp \
|
||||||
miner.cpp \
|
miner.cpp \
|
||||||
net.cpp \
|
net.cpp \
|
||||||
net_processing.cpp \
|
net_processing.cpp \
|
||||||
|
|
|
@ -26,6 +26,7 @@
|
||||||
#include <interfaces/chain.h>
|
#include <interfaces/chain.h>
|
||||||
#include <interfaces/node.h>
|
#include <interfaces/node.h>
|
||||||
#include <key.h>
|
#include <key.h>
|
||||||
|
#include <mapport.h>
|
||||||
#include <miner.h>
|
#include <miner.h>
|
||||||
#include <net.h>
|
#include <net.h>
|
||||||
#include <net_permissions.h>
|
#include <net_permissions.h>
|
||||||
|
|
137
src/mapport.cpp
Normal file
137
src/mapport.cpp
Normal file
|
@ -0,0 +1,137 @@
|
||||||
|
// Copyright (c) 2011-2020 The Bitcoin Core developers
|
||||||
|
// Distributed under the MIT software license, see the accompanying
|
||||||
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||||
|
|
||||||
|
#if defined(HAVE_CONFIG_H)
|
||||||
|
#include <config/bitcoin-config.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <mapport.h>
|
||||||
|
|
||||||
|
#include <clientversion.h>
|
||||||
|
#include <logging.h>
|
||||||
|
#include <net.h>
|
||||||
|
#include <netaddress.h>
|
||||||
|
#include <netbase.h>
|
||||||
|
#include <threadinterrupt.h>
|
||||||
|
#include <util/system.h>
|
||||||
|
|
||||||
|
#ifdef USE_UPNP
|
||||||
|
#include <miniupnpc/miniupnpc.h>
|
||||||
|
#include <miniupnpc/upnpcommands.h>
|
||||||
|
#include <miniupnpc/upnperrors.h>
|
||||||
|
// The minimum supported miniUPnPc API version is set to 10. This keeps compatibility
|
||||||
|
// with Ubuntu 16.04 LTS and Debian 8 libminiupnpc-dev packages.
|
||||||
|
static_assert(MINIUPNPC_API_VERSION >= 10, "miniUPnPc API version >= 10 assumed");
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <cassert>
|
||||||
|
#include <chrono>
|
||||||
|
#include <functional>
|
||||||
|
#include <string>
|
||||||
|
#include <thread>
|
||||||
|
|
||||||
|
#ifdef USE_UPNP
|
||||||
|
static CThreadInterrupt g_upnp_interrupt;
|
||||||
|
static std::thread g_upnp_thread;
|
||||||
|
static void ThreadMapPort()
|
||||||
|
{
|
||||||
|
std::string port = strprintf("%u", GetListenPort());
|
||||||
|
const char * multicastif = nullptr;
|
||||||
|
const char * minissdpdpath = nullptr;
|
||||||
|
struct UPNPDev * devlist = nullptr;
|
||||||
|
char lanaddr[64];
|
||||||
|
|
||||||
|
int error = 0;
|
||||||
|
#if MINIUPNPC_API_VERSION < 14
|
||||||
|
devlist = upnpDiscover(2000, multicastif, minissdpdpath, 0, 0, &error);
|
||||||
|
#else
|
||||||
|
devlist = upnpDiscover(2000, multicastif, minissdpdpath, 0, 0, 2, &error);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
struct UPNPUrls urls;
|
||||||
|
struct IGDdatas data;
|
||||||
|
int r;
|
||||||
|
|
||||||
|
r = UPNP_GetValidIGD(devlist, &urls, &data, lanaddr, sizeof(lanaddr));
|
||||||
|
if (r == 1)
|
||||||
|
{
|
||||||
|
if (fDiscover) {
|
||||||
|
char externalIPAddress[40];
|
||||||
|
r = UPNP_GetExternalIPAddress(urls.controlURL, data.first.servicetype, externalIPAddress);
|
||||||
|
if (r != UPNPCOMMAND_SUCCESS) {
|
||||||
|
LogPrintf("UPnP: GetExternalIPAddress() returned %d\n", r);
|
||||||
|
} else {
|
||||||
|
if (externalIPAddress[0]) {
|
||||||
|
CNetAddr resolved;
|
||||||
|
if (LookupHost(externalIPAddress, resolved, false)) {
|
||||||
|
LogPrintf("UPnP: ExternalIPAddress = %s\n", resolved.ToString());
|
||||||
|
AddLocal(resolved, LOCAL_UPNP);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
LogPrintf("UPnP: GetExternalIPAddress failed.\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string strDesc = PACKAGE_NAME " " + FormatFullVersion();
|
||||||
|
|
||||||
|
do {
|
||||||
|
r = UPNP_AddPortMapping(urls.controlURL, data.first.servicetype, port.c_str(), port.c_str(), lanaddr, strDesc.c_str(), "TCP", 0, "0");
|
||||||
|
|
||||||
|
if (r != UPNPCOMMAND_SUCCESS) {
|
||||||
|
LogPrintf("AddPortMapping(%s, %s, %s) failed with code %d (%s)\n", port, port, lanaddr, r, strupnperror(r));
|
||||||
|
} else {
|
||||||
|
LogPrintf("UPnP Port Mapping successful.\n");
|
||||||
|
}
|
||||||
|
} while (g_upnp_interrupt.sleep_for(std::chrono::minutes(20)));
|
||||||
|
|
||||||
|
r = UPNP_DeletePortMapping(urls.controlURL, data.first.servicetype, port.c_str(), "TCP", 0);
|
||||||
|
LogPrintf("UPNP_DeletePortMapping() returned: %d\n", r);
|
||||||
|
freeUPNPDevlist(devlist); devlist = nullptr;
|
||||||
|
FreeUPNPUrls(&urls);
|
||||||
|
} else {
|
||||||
|
LogPrintf("No valid UPnP IGDs found\n");
|
||||||
|
freeUPNPDevlist(devlist); devlist = nullptr;
|
||||||
|
if (r != 0)
|
||||||
|
FreeUPNPUrls(&urls);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void StartMapPort()
|
||||||
|
{
|
||||||
|
if (!g_upnp_thread.joinable()) {
|
||||||
|
assert(!g_upnp_interrupt);
|
||||||
|
g_upnp_thread = std::thread((std::bind(&TraceThread<void (*)()>, "upnp", &ThreadMapPort)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void InterruptMapPort()
|
||||||
|
{
|
||||||
|
if(g_upnp_thread.joinable()) {
|
||||||
|
g_upnp_interrupt();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void StopMapPort()
|
||||||
|
{
|
||||||
|
if(g_upnp_thread.joinable()) {
|
||||||
|
g_upnp_thread.join();
|
||||||
|
g_upnp_interrupt.reset();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
void StartMapPort()
|
||||||
|
{
|
||||||
|
// Intentionally left blank.
|
||||||
|
}
|
||||||
|
void InterruptMapPort()
|
||||||
|
{
|
||||||
|
// Intentionally left blank.
|
||||||
|
}
|
||||||
|
void StopMapPort()
|
||||||
|
{
|
||||||
|
// Intentionally left blank.
|
||||||
|
}
|
||||||
|
#endif
|
19
src/mapport.h
Normal file
19
src/mapport.h
Normal file
|
@ -0,0 +1,19 @@
|
||||||
|
// Copyright (c) 2011-2020 The Bitcoin Core developers
|
||||||
|
// Distributed under the MIT software license, see the accompanying
|
||||||
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||||
|
|
||||||
|
#ifndef BITCOIN_MAPPORT_H
|
||||||
|
#define BITCOIN_MAPPORT_H
|
||||||
|
|
||||||
|
/** -upnp default */
|
||||||
|
#ifdef USE_UPNP
|
||||||
|
static const bool DEFAULT_UPNP = USE_UPNP;
|
||||||
|
#else
|
||||||
|
static const bool DEFAULT_UPNP = false;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
void StartMapPort();
|
||||||
|
void InterruptMapPort();
|
||||||
|
void StopMapPort();
|
||||||
|
|
||||||
|
#endif // BITCOIN_MAPPORT_H
|
124
src/net.cpp
124
src/net.cpp
|
@ -33,15 +33,6 @@
|
||||||
#include <poll.h>
|
#include <poll.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef USE_UPNP
|
|
||||||
#include <miniupnpc/miniupnpc.h>
|
|
||||||
#include <miniupnpc/upnpcommands.h>
|
|
||||||
#include <miniupnpc/upnperrors.h>
|
|
||||||
// The minimum supported miniUPnPc API version is set to 10. This keeps compatibility
|
|
||||||
// with Ubuntu 16.04 LTS and Debian 8 libminiupnpc-dev packages.
|
|
||||||
static_assert(MINIUPNPC_API_VERSION >= 10, "miniUPnPc API version >= 10 assumed");
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
|
@ -1539,121 +1530,6 @@ void CConnman::WakeMessageHandler()
|
||||||
condMsgProc.notify_one();
|
condMsgProc.notify_one();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef USE_UPNP
|
|
||||||
static CThreadInterrupt g_upnp_interrupt;
|
|
||||||
static std::thread g_upnp_thread;
|
|
||||||
static void ThreadMapPort()
|
|
||||||
{
|
|
||||||
std::string port = strprintf("%u", GetListenPort());
|
|
||||||
const char * multicastif = nullptr;
|
|
||||||
const char * minissdpdpath = nullptr;
|
|
||||||
struct UPNPDev * devlist = nullptr;
|
|
||||||
char lanaddr[64];
|
|
||||||
|
|
||||||
int error = 0;
|
|
||||||
#if MINIUPNPC_API_VERSION < 14
|
|
||||||
devlist = upnpDiscover(2000, multicastif, minissdpdpath, 0, 0, &error);
|
|
||||||
#else
|
|
||||||
devlist = upnpDiscover(2000, multicastif, minissdpdpath, 0, 0, 2, &error);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
struct UPNPUrls urls;
|
|
||||||
struct IGDdatas data;
|
|
||||||
int r;
|
|
||||||
|
|
||||||
r = UPNP_GetValidIGD(devlist, &urls, &data, lanaddr, sizeof(lanaddr));
|
|
||||||
if (r == 1)
|
|
||||||
{
|
|
||||||
if (fDiscover) {
|
|
||||||
char externalIPAddress[40];
|
|
||||||
r = UPNP_GetExternalIPAddress(urls.controlURL, data.first.servicetype, externalIPAddress);
|
|
||||||
if (r != UPNPCOMMAND_SUCCESS) {
|
|
||||||
LogPrintf("UPnP: GetExternalIPAddress() returned %d\n", r);
|
|
||||||
} else {
|
|
||||||
if (externalIPAddress[0]) {
|
|
||||||
CNetAddr resolved;
|
|
||||||
if (LookupHost(externalIPAddress, resolved, false)) {
|
|
||||||
LogPrintf("UPnP: ExternalIPAddress = %s\n", resolved.ToString());
|
|
||||||
AddLocal(resolved, LOCAL_UPNP);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
LogPrintf("UPnP: GetExternalIPAddress failed.\n");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string strDesc = PACKAGE_NAME " " + FormatFullVersion();
|
|
||||||
|
|
||||||
do {
|
|
||||||
r = UPNP_AddPortMapping(urls.controlURL, data.first.servicetype, port.c_str(), port.c_str(), lanaddr, strDesc.c_str(), "TCP", 0, "0");
|
|
||||||
|
|
||||||
if (r != UPNPCOMMAND_SUCCESS) {
|
|
||||||
LogPrintf("AddPortMapping(%s, %s, %s) failed with code %d (%s)\n", port, port, lanaddr, r, strupnperror(r));
|
|
||||||
} else {
|
|
||||||
LogPrintf("UPnP Port Mapping successful.\n");
|
|
||||||
}
|
|
||||||
} while (g_upnp_interrupt.sleep_for(std::chrono::minutes(20)));
|
|
||||||
|
|
||||||
r = UPNP_DeletePortMapping(urls.controlURL, data.first.servicetype, port.c_str(), "TCP", 0);
|
|
||||||
LogPrintf("UPNP_DeletePortMapping() returned: %d\n", r);
|
|
||||||
freeUPNPDevlist(devlist); devlist = nullptr;
|
|
||||||
FreeUPNPUrls(&urls);
|
|
||||||
} else {
|
|
||||||
LogPrintf("No valid UPnP IGDs found\n");
|
|
||||||
freeUPNPDevlist(devlist); devlist = nullptr;
|
|
||||||
if (r != 0)
|
|
||||||
FreeUPNPUrls(&urls);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void StartMapPort()
|
|
||||||
{
|
|
||||||
if (!g_upnp_thread.joinable()) {
|
|
||||||
assert(!g_upnp_interrupt);
|
|
||||||
g_upnp_thread = std::thread((std::bind(&TraceThread<void (*)()>, "upnp", &ThreadMapPort)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void InterruptMapPort()
|
|
||||||
{
|
|
||||||
if(g_upnp_thread.joinable()) {
|
|
||||||
g_upnp_interrupt();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void StopMapPort()
|
|
||||||
{
|
|
||||||
if(g_upnp_thread.joinable()) {
|
|
||||||
g_upnp_thread.join();
|
|
||||||
g_upnp_interrupt.reset();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
void StartMapPort()
|
|
||||||
{
|
|
||||||
// Intentionally left blank.
|
|
||||||
}
|
|
||||||
void InterruptMapPort()
|
|
||||||
{
|
|
||||||
// Intentionally left blank.
|
|
||||||
}
|
|
||||||
void StopMapPort()
|
|
||||||
{
|
|
||||||
// Intentionally left blank.
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
void CConnman::ThreadDNSAddressSeed()
|
void CConnman::ThreadDNSAddressSeed()
|
||||||
{
|
{
|
||||||
FastRandomContext rng;
|
FastRandomContext rng;
|
||||||
|
|
|
@ -67,12 +67,6 @@ static const int MAX_BLOCK_RELAY_ONLY_CONNECTIONS = 2;
|
||||||
static const int MAX_FEELER_CONNECTIONS = 1;
|
static const int MAX_FEELER_CONNECTIONS = 1;
|
||||||
/** -listen default */
|
/** -listen default */
|
||||||
static const bool DEFAULT_LISTEN = true;
|
static const bool DEFAULT_LISTEN = true;
|
||||||
/** -upnp default */
|
|
||||||
#ifdef USE_UPNP
|
|
||||||
static const bool DEFAULT_UPNP = USE_UPNP;
|
|
||||||
#else
|
|
||||||
static const bool DEFAULT_UPNP = false;
|
|
||||||
#endif
|
|
||||||
/** The maximum number of peer connections to maintain. */
|
/** The maximum number of peer connections to maintain. */
|
||||||
static const unsigned int DEFAULT_MAX_PEER_CONNECTIONS = 125;
|
static const unsigned int DEFAULT_MAX_PEER_CONNECTIONS = 125;
|
||||||
/** The default for -maxuploadtarget. 0 = Unlimited */
|
/** The default for -maxuploadtarget. 0 = Unlimited */
|
||||||
|
@ -181,9 +175,6 @@ enum class ConnectionType {
|
||||||
};
|
};
|
||||||
|
|
||||||
void Discover();
|
void Discover();
|
||||||
void StartMapPort();
|
|
||||||
void InterruptMapPort();
|
|
||||||
void StopMapPort();
|
|
||||||
uint16_t GetListenPort();
|
uint16_t GetListenPort();
|
||||||
|
|
||||||
enum
|
enum
|
||||||
|
|
|
@ -12,6 +12,7 @@
|
||||||
#include <interfaces/handler.h>
|
#include <interfaces/handler.h>
|
||||||
#include <interfaces/node.h>
|
#include <interfaces/node.h>
|
||||||
#include <interfaces/wallet.h>
|
#include <interfaces/wallet.h>
|
||||||
|
#include <mapport.h>
|
||||||
#include <net.h>
|
#include <net.h>
|
||||||
#include <net_processing.h>
|
#include <net_processing.h>
|
||||||
#include <netaddress.h>
|
#include <netaddress.h>
|
||||||
|
|
|
@ -13,11 +13,12 @@
|
||||||
#include <qt/guiutil.h>
|
#include <qt/guiutil.h>
|
||||||
|
|
||||||
#include <interfaces/node.h>
|
#include <interfaces/node.h>
|
||||||
#include <validation.h> // For DEFAULT_SCRIPTCHECK_THREADS
|
#include <mapport.h>
|
||||||
#include <net.h>
|
#include <net.h>
|
||||||
#include <netbase.h>
|
#include <netbase.h>
|
||||||
#include <txdb.h> // for -dbcache defaults
|
#include <txdb.h> // for -dbcache defaults
|
||||||
#include <util/string.h>
|
#include <util/string.h>
|
||||||
|
#include <validation.h> // For DEFAULT_SCRIPTCHECK_THREADS
|
||||||
|
|
||||||
#include <QDebug>
|
#include <QDebug>
|
||||||
#include <QSettings>
|
#include <QSettings>
|
||||||
|
|
Loading…
Add table
Reference in a new issue