|
|
|
@ -37,7 +37,7 @@ using namespace std;
@@ -37,7 +37,7 @@ using namespace std;
|
|
|
|
|
|
|
|
|
|
// Settings
|
|
|
|
|
static proxyType proxyInfo[NET_MAX]; |
|
|
|
|
static proxyType nameproxyInfo; |
|
|
|
|
static CService nameProxy; |
|
|
|
|
static CCriticalSection cs_proxyInfos; |
|
|
|
|
int nConnectTimeout = 5000; |
|
|
|
|
bool fNameLookup = false; |
|
|
|
@ -214,50 +214,6 @@ bool LookupNumeric(const char *pszName, CService& addr, int portDefault)
@@ -214,50 +214,6 @@ bool LookupNumeric(const char *pszName, CService& addr, int portDefault)
|
|
|
|
|
return Lookup(pszName, addr, portDefault, false); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool static Socks4(const CService &addrDest, SOCKET& hSocket) |
|
|
|
|
{ |
|
|
|
|
LogPrintf("SOCKS4 connecting %s\n", addrDest.ToString()); |
|
|
|
|
if (!addrDest.IsIPv4()) |
|
|
|
|
{ |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
return error("Proxy destination is not IPv4"); |
|
|
|
|
} |
|
|
|
|
char pszSocks4IP[] = "\4\1\0\0\0\0\0\0user"; |
|
|
|
|
struct sockaddr_in addr; |
|
|
|
|
socklen_t len = sizeof(addr); |
|
|
|
|
if (!addrDest.GetSockAddr((struct sockaddr*)&addr, &len) || addr.sin_family != AF_INET) |
|
|
|
|
{ |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
return error("Cannot get proxy destination address"); |
|
|
|
|
} |
|
|
|
|
memcpy(pszSocks4IP + 2, &addr.sin_port, 2); |
|
|
|
|
memcpy(pszSocks4IP + 4, &addr.sin_addr, 4); |
|
|
|
|
char* pszSocks4 = pszSocks4IP; |
|
|
|
|
int nSize = sizeof(pszSocks4IP); |
|
|
|
|
|
|
|
|
|
int ret = send(hSocket, pszSocks4, nSize, MSG_NOSIGNAL); |
|
|
|
|
if (ret != nSize) |
|
|
|
|
{ |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
return error("Error sending to proxy"); |
|
|
|
|
} |
|
|
|
|
char pchRet[8]; |
|
|
|
|
if (recv(hSocket, pchRet, 8, 0) != 8) |
|
|
|
|
{ |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
return error("Error reading proxy response"); |
|
|
|
|
} |
|
|
|
|
if (pchRet[1] != 0x5a) |
|
|
|
|
{ |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
if (pchRet[1] != 0x5b) |
|
|
|
|
LogPrintf("ERROR: Proxy returned error %d\n", pchRet[1]); |
|
|
|
|
return false; |
|
|
|
|
} |
|
|
|
|
LogPrintf("SOCKS4 connected %s\n", addrDest.ToString()); |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool static Socks5(string strDest, int port, SOCKET& hSocket) |
|
|
|
|
{ |
|
|
|
|
LogPrintf("SOCKS5 connecting %s\n", strDest); |
|
|
|
@ -468,53 +424,49 @@ bool static ConnectSocketDirectly(const CService &addrConnect, SOCKET& hSocketRe
@@ -468,53 +424,49 @@ bool static ConnectSocketDirectly(const CService &addrConnect, SOCKET& hSocketRe
|
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool SetProxy(enum Network net, CService addrProxy, int nSocksVersion) { |
|
|
|
|
bool SetProxy(enum Network net, CService addrProxy) { |
|
|
|
|
assert(net >= 0 && net < NET_MAX); |
|
|
|
|
if (nSocksVersion != 0 && nSocksVersion != 4 && nSocksVersion != 5) |
|
|
|
|
return false; |
|
|
|
|
if (nSocksVersion != 0 && !addrProxy.IsValid()) |
|
|
|
|
if (!addrProxy.IsValid()) |
|
|
|
|
return false; |
|
|
|
|
LOCK(cs_proxyInfos); |
|
|
|
|
proxyInfo[net] = std::make_pair(addrProxy, nSocksVersion); |
|
|
|
|
proxyInfo[net] = addrProxy; |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool GetProxy(enum Network net, proxyType &proxyInfoOut) { |
|
|
|
|
assert(net >= 0 && net < NET_MAX); |
|
|
|
|
LOCK(cs_proxyInfos); |
|
|
|
|
if (!proxyInfo[net].second) |
|
|
|
|
if (!proxyInfo[net].IsValid()) |
|
|
|
|
return false; |
|
|
|
|
proxyInfoOut = proxyInfo[net]; |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool SetNameProxy(CService addrProxy, int nSocksVersion) { |
|
|
|
|
if (nSocksVersion != 0 && nSocksVersion != 5) |
|
|
|
|
return false; |
|
|
|
|
if (nSocksVersion != 0 && !addrProxy.IsValid()) |
|
|
|
|
bool SetNameProxy(CService addrProxy) { |
|
|
|
|
if (!addrProxy.IsValid()) |
|
|
|
|
return false; |
|
|
|
|
LOCK(cs_proxyInfos); |
|
|
|
|
nameproxyInfo = std::make_pair(addrProxy, nSocksVersion); |
|
|
|
|
nameProxy = addrProxy; |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool GetNameProxy(proxyType &nameproxyInfoOut) { |
|
|
|
|
bool GetNameProxy(CService &nameProxyOut) { |
|
|
|
|
LOCK(cs_proxyInfos); |
|
|
|
|
if (!nameproxyInfo.second) |
|
|
|
|
if(!nameProxy.IsValid()) |
|
|
|
|
return false; |
|
|
|
|
nameproxyInfoOut = nameproxyInfo; |
|
|
|
|
nameProxyOut = nameProxy; |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool HaveNameProxy() { |
|
|
|
|
LOCK(cs_proxyInfos); |
|
|
|
|
return nameproxyInfo.second != 0; |
|
|
|
|
return nameProxy.IsValid(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
bool IsProxy(const CNetAddr &addr) { |
|
|
|
|
LOCK(cs_proxyInfos); |
|
|
|
|
for (int i = 0; i < NET_MAX; i++) { |
|
|
|
|
if (proxyInfo[i].second && (addr == (CNetAddr)proxyInfo[i].first)) |
|
|
|
|
if (addr == (CNetAddr)proxyInfo[i]) |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
return false; |
|
|
|
@ -523,31 +475,18 @@ bool IsProxy(const CNetAddr &addr) {
@@ -523,31 +475,18 @@ bool IsProxy(const CNetAddr &addr) {
|
|
|
|
|
bool ConnectSocket(const CService &addrDest, SOCKET& hSocketRet, int nTimeout) |
|
|
|
|
{ |
|
|
|
|
proxyType proxy; |
|
|
|
|
|
|
|
|
|
// no proxy needed
|
|
|
|
|
// no proxy needed (none set for target network)
|
|
|
|
|
if (!GetProxy(addrDest.GetNetwork(), proxy)) |
|
|
|
|
return ConnectSocketDirectly(addrDest, hSocketRet, nTimeout); |
|
|
|
|
|
|
|
|
|
SOCKET hSocket = INVALID_SOCKET; |
|
|
|
|
|
|
|
|
|
// first connect to proxy server
|
|
|
|
|
if (!ConnectSocketDirectly(proxy.first, hSocket, nTimeout)) |
|
|
|
|
if (!ConnectSocketDirectly(proxy, hSocket, nTimeout)) |
|
|
|
|
return false; |
|
|
|
|
|
|
|
|
|
// do socks negotiation
|
|
|
|
|
switch (proxy.second) { |
|
|
|
|
case 4: |
|
|
|
|
if (!Socks4(addrDest, hSocket)) |
|
|
|
|
return false; |
|
|
|
|
break; |
|
|
|
|
case 5: |
|
|
|
|
if (!Socks5(addrDest.ToStringIP(), addrDest.GetPort(), hSocket)) |
|
|
|
|
return false; |
|
|
|
|
break; |
|
|
|
|
default: |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
return false; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
hSocketRet = hSocket; |
|
|
|
|
return true; |
|
|
|
@ -561,30 +500,25 @@ bool ConnectSocketByName(CService &addr, SOCKET& hSocketRet, const char *pszDest
@@ -561,30 +500,25 @@ bool ConnectSocketByName(CService &addr, SOCKET& hSocketRet, const char *pszDest
|
|
|
|
|
|
|
|
|
|
SOCKET hSocket = INVALID_SOCKET; |
|
|
|
|
|
|
|
|
|
proxyType nameproxy; |
|
|
|
|
GetNameProxy(nameproxy); |
|
|
|
|
CService nameProxy; |
|
|
|
|
GetNameProxy(nameProxy); |
|
|
|
|
|
|
|
|
|
CService addrResolved(CNetAddr(strDest, fNameLookup && !nameproxy.second), port); |
|
|
|
|
CService addrResolved(CNetAddr(strDest, fNameLookup && !HaveNameProxy()), port); |
|
|
|
|
if (addrResolved.IsValid()) { |
|
|
|
|
addr = addrResolved; |
|
|
|
|
return ConnectSocket(addr, hSocketRet, nTimeout); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
addr = CService("0.0.0.0:0"); |
|
|
|
|
if (!nameproxy.second) |
|
|
|
|
return false; |
|
|
|
|
if (!ConnectSocketDirectly(nameproxy.first, hSocket, nTimeout)) |
|
|
|
|
return false; |
|
|
|
|
|
|
|
|
|
switch(nameproxy.second) { |
|
|
|
|
default: |
|
|
|
|
case 4: |
|
|
|
|
closesocket(hSocket); |
|
|
|
|
if (!HaveNameProxy()) |
|
|
|
|
return false; |
|
|
|
|
case 5: |
|
|
|
|
if (!Socks5(strDest, port, hSocket)) |
|
|
|
|
// first connect to name proxy server
|
|
|
|
|
if (!ConnectSocketDirectly(nameProxy, hSocket, nTimeout)) |
|
|
|
|
return false; |
|
|
|
|
// do socks negotiation
|
|
|
|
|
if (!Socks5(strDest, (unsigned short)port, hSocket)) |
|
|
|
|
return false; |
|
|
|
|
break; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
hSocketRet = hSocket; |
|
|
|
|
return true; |
|
|
|
|