github/Qv2ray
5
0
Fork 0
mirror of https://github.com/Qv2ray/Qv2ray.git synced 2025-05-20 02:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix windows build

Browse Source
This commit is contained in:
DuckVador 2020-07-25 20:57:36 +08:00
parent c84dc794fa
commit aaff3de0a7
No known key found for this signature in database
GPG Key ID: 9B0DED785B37110E
8 changed files with 136 additions and 142 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
makespec/BUILDVERSION
View File

@ -1 +1 @@
5808
5809

246
src/components/latency/DNSBase.hpp
View File

@ -1,116 +1,130 @@
#pragma once
#include "LatencyTest.hpp"
#include "coroutine.hpp"
#include "uvw.hpp"
namespace Qv2ray::components::latency
{
template<typename T>
class DNSBase
: public coroutine
, public std::enable_shared_from_this<T>
{
public:
DNSBase(const std::shared_ptr<uvw::Loop> &loopin, LatencyTestRequest &req, LatencyTestHost *testHost)
: loop(loopin), req(std::move(req)), testHost(testHost)
{
}
virtual ~DNSBase();
protected:
template<typename E, typename H>
void async_DNS_lookup(E &&e, H &&h)
{
co_enter(*this)
{
if (getAddrHandle)
{
getAddrHandle->once<uvw::ErrorEvent>(coro(async_DNS_lookup));
getAddrHandle->once<uvw::AddrInfoEvent>(coro(async_DNS_lookup));
co_yield return getAddrHandle->addrInfo(req.host.toStdString(), digitBuffer);
co_yield if constexpr (std::is_same_v<uvw::AddrInfoEvent, std::remove_reference_t<E>>)
{
if (getAddrInfoRes(e) != 0)
{
data.errorMessage = QObject::tr("DNS not resolved");
data.avg = LATENCY_TEST_VALUE_ERROR;
testHost->OnLatencyTestCompleted(req.id, data);
h.clear();
return;
}
}
else
{
if constexpr (std::is_same_v<uvw::ErrorEvent, std::remove_reference_t<E>>)
{
data.errorMessage = QObject::tr("DNS not resolved");
data.avg = LATENCY_TEST_VALUE_ERROR;
testHost->OnLatencyTestCompleted(req.id, data);
h.clear();
return;
}
}
}
}
ping();
if (getAddrHandle)
getAddrHandle->clear();
}
int getAddrInfoRes(uvw::AddrInfoEvent &e)
{
struct addrinfo *rp = nullptr;
for (rp = e.data.get(); rp != nullptr; rp = rp->ai_next)
if (rp->ai_family == AF_INET)
{
if (rp->ai_family == AF_INET)
{
af = AF_INET;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in));
}
else if (rp->ai_family == AF_INET6)
{
af = AF_INET6;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in6));
}
break;
}
if (rp == nullptr)
{
// fallback: if we can't find prefered AF, then we choose alternative.
for (rp = e.data.get(); rp != nullptr; rp = rp->ai_next)
{
if (rp->ai_family == AF_INET)
{
af = AF_INET;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in));
}
else if (rp->ai_family == AF_INET6)
{
af = AF_INET6;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in6));
}
break;
}
}
if (rp)
return 0;
return -1;
}
virtual void ping() = 0;
protected:
int af = AF_INET;
int successCount = 0;
LatencyTestRequest req;
LatencyTestResult data;
LatencyTestHost *testHost;
struct sockaddr_storage storage;
char digitBuffer[20] = { 0 };
std::shared_ptr<uvw::Loop> loop;
std::shared_ptr<uvw::GetAddrInfoReq> getAddrHandle;
};
template<typename T>
DNSBase<T>::~DNSBase()
{
}
} // namespace Qv2ray::components::latency
#pragma once
#include "LatencyTest.hpp"
#include "coroutine.hpp"
#include "uvw.hpp"
namespace Qv2ray::components::latency
{
template<typename T>
class DNSBase
: public coroutine
, public std::enable_shared_from_this<T>
{
public:
DNSBase(const std::shared_ptr<uvw::Loop> &loopin, LatencyTestRequest &req, LatencyTestHost *testHost)
: loop(loopin), req(std::move(req)), testHost(testHost)
{
}
virtual ~DNSBase();
protected:
int isAddr()
{
auto host = req.host.toStdString();
if (uv_ip4_addr(host.data(), req.port, reinterpret_cast<sockaddr_in *>(&storage)) == 0)
{
return AF_INET;
}
if (uv_ip6_addr(host.data(), req.port, reinterpret_cast<sockaddr_in6 *>(&storage)) == 0)
{
return AF_INET6;
}
return -1;
}
template<typename E, typename H>
void async_DNS_lookup(E &&e, H &&h)
{
co_enter(*this)
{
if (getAddrHandle)
{
getAddrHandle->once<uvw::ErrorEvent>(coro(async_DNS_lookup));
getAddrHandle->once<uvw::AddrInfoEvent>(coro(async_DNS_lookup));
co_yield return getAddrHandle->addrInfo(req.host.toStdString(), digitBuffer);
co_yield if constexpr (std::is_same_v<uvw::AddrInfoEvent, std::remove_reference_t<E>>)
{
if (getAddrInfoRes(e) != 0)
{
data.errorMessage = QObject::tr("DNS not resolved");
data.avg = LATENCY_TEST_VALUE_ERROR;
testHost->OnLatencyTestCompleted(req.id, data);
h.clear();
return;
}
}
else
{
if constexpr (std::is_same_v<uvw::ErrorEvent, std::remove_reference_t<E>>)
{
data.errorMessage = QObject::tr("DNS not resolved");
data.avg = LATENCY_TEST_VALUE_ERROR;
testHost->OnLatencyTestCompleted(req.id, data);
h.clear();
return;
}
}
}
}
ping();
if (getAddrHandle)
getAddrHandle->clear();
}
int getAddrInfoRes(uvw::AddrInfoEvent &e)
{
struct addrinfo *rp = nullptr;
for (rp = e.data.get(); rp != nullptr; rp = rp->ai_next)
if (rp->ai_family == AF_INET)
{
if (rp->ai_family == AF_INET)
{
af = AF_INET;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in));
}
else if (rp->ai_family == AF_INET6)
{
af = AF_INET6;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in6));
}
break;
}
if (rp == nullptr)
{
// fallback: if we can't find prefered AF, then we choose alternative.
for (rp = e.data.get(); rp != nullptr; rp = rp->ai_next)
{
if (rp->ai_family == AF_INET)
{
af = AF_INET;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in));
}
else if (rp->ai_family == AF_INET6)
{
af = AF_INET6;
memcpy(&storage, rp->ai_addr, sizeof(struct sockaddr_in6));
}
break;
}
}
if (rp)
return 0;
return -1;
}
virtual void ping() = 0;
protected:
int af = AF_INET;
int successCount = 0;
LatencyTestRequest req;
LatencyTestResult data;
LatencyTestHost *testHost;
struct sockaddr_storage storage;
char digitBuffer[20] = { 0 };
std::shared_ptr<uvw::Loop> loop;
std::shared_ptr<uvw::GetAddrInfoReq> getAddrHandle;
};
template<typename T>
DNSBase<T>::~DNSBase()
{
}
} // namespace Qv2ray::components::latency

13
src/components/latency/LatencyTest.cpp
View File

@ -2,22 +2,9 @@
#include "LatencyTestThread.hpp"
#include "core/handler/ConfigHandler.hpp"
#include "uvw.hpp"
namespace Qv2ray::components::latency
{
int isAddr(const char *host, int port, struct sockaddr_storage *storage, int ipv6first)
{
if (uv_ip4_addr(host, port, reinterpret_cast<sockaddr_in *>(storage)) == 0)
{
return AF_INET;
}
if (uv_ip6_addr(host, port, reinterpret_cast<sockaddr_in6 *>(storage)) == 0)
{
return AF_INET6;
}
return -1;
}
LatencyTestHost::LatencyTestHost(const int defaultCount, QObject *parent) : QObject(parent)
{
qRegisterMetaType<ConnectionId>();

2
src/components/latency/LatencyTest.hpp
View File

@ -27,8 +27,6 @@ namespace Qv2ray::components::latency
Qv2rayLatencyTestingMethod method;
};
int isAddr(const char *host, int port, struct sockaddr_storage *storage, int ipv6first);
class LatencyTestHost : public QObject
{
Q_OBJECT

2
src/components/latency/TCPing.cpp
View File

@ -36,7 +36,7 @@ namespace Qv2ray::components::latency::tcping
data.failedCount = 0;
data.worst = 0;
data.avg = 0;
af = isAddr(req.host.toStdString().data(), req.port, &storage, 0);
af = isAddr();
if (af == -1)
{
getAddrHandle = loop->resource<uvw::GetAddrInfoReq>();

2
src/components/latency/unix/ICMPPing.cpp
View File

@ -73,7 +73,7 @@ namespace Qv2ray::components::latency::icmping
data.failedCount = 0;
data.worst = 0;
data.avg = 0;
if (isAddr(req.host.toStdString().data(), req.port, &storage, 0) == -1)
if (isAddr() == -1)
{
getAddrHandle = loop->resource<uvw::GetAddrInfoReq>();
sprintf(digitBuffer, "%d", req.port);

5
src/components/latency/win/ICMPPing.cpp
View File

@ -11,10 +11,11 @@ typedef struct _IO_STATUS_BLOCK
} IO_STATUS_BLOCK, *PIO_STATUS_BLOCK;
typedef VOID(NTAPI *PIO_APC_ROUTINE)(IN PVOID ApcContext, IN PIO_STATUS_BLOCK IoStatusBlock, IN ULONG Reserved);
#define PIO_APC_ROUTINE_DEFINED
#include <IcmpAPI.h>
#include <WS2tcpip.h>
#include <Windows.h>
#include <iphlpapi.h>
#include <IcmpAPI.h>
#include <QString>
namespace Qv2ray::components::latency::icmping
{
@ -129,7 +130,7 @@ namespace Qv2ray::components::latency::icmping
data.failedCount = 0;
data.worst = 0;
data.avg = 0;
af = isAddr(req.host.toStdString().data(), req.port, &storage, 0);
af = isAddr();
if (af == -1)
{
getAddrHandle = loop->resource<uvw::GetAddrInfoReq>();

6
src/components/latency/win/ICMPPing.hpp
View File

@ -3,14 +3,8 @@
#include <QtGlobal>
#ifdef Q_OS_WIN
#include "components/latency/LatencyTest.hpp"
#include "../DNSBase.hpp"
#include <QPair>
#include <QString>
#include <memory>
#include <optional>
#include <utility>
namespace Qv2ray::components::latency::icmping
{