neil/proxyTest
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added intermediate proxy, Telegram works good

Browse Source
This commit is contained in:
Vladislav Khorev 2017-08-07 00:53:25 +03:00
parent d22850c3b6
commit db9a8c90c4
1 changed files with 219 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

239
main.cpp
View File

@ -74,19 +74,23 @@ std::string ConnectRequestRecord::getRequestData()
return data;
}
class proxyClient
class ProxyClient
{
public:
proxyClient(boost::asio::io_service& ioServiceIn, boost::asio::ssl::context& context, boost::asio::ip::tcp::resolver::iterator endpointIn)
ProxyClient(boost::asio::io_service& ioServiceIn, boost::asio::ssl::context& context, boost::asio::ip::tcp::resolver& resolverIn)
: ioService(ioServiceIn)
, socket(ioServiceIn, context)
, resolver(resolverIn)
{
socket.set_verify_mode(boost::asio::ssl::verify_peer);
socket.set_verify_callback(
std::bind(&proxyClient::verify_certificate, this, std::placeholders::_1, std::placeholders::_2));
std::bind(&ProxyClient::verify_certificate, this, std::placeholders::_1, std::placeholders::_2));
}
doConnect(endpointIn);
void start()
{
doConnect();
}
void close()
@ -94,22 +98,32 @@ public:
ioService.post([this]() { lowerSocket().close(); });
}
private:
std::function<void()> onReady;
boost::asio::ssl::stream<boost::asio::ip::tcp::socket>& getSocket()
{
return socket;
}
boost::asio::ssl::stream<boost::asio::ip::tcp::socket>::lowest_layer_type& lowerSocket()
{
return socket.lowest_layer();
}
void doConnect(boost::asio::ip::tcp::resolver::iterator endpointIterator)
private:
void doConnect()
{
boost::asio::ip::tcp::resolver::iterator endpointIterator(resolver.resolve({ "127.0.0.1", "8043" }));
boost::asio::async_connect(lowerSocket(), endpointIterator,
[this](boost::system::error_code ec, boost::asio::ip::tcp::resolver::iterator)
{
if (!ec)
{
//sendVersion();
doHandshake();
}
});
@ -124,7 +138,8 @@ private:
if (!error)
{
sendVersion();
onReady();
//sendVersion();
}
else
{
@ -298,7 +313,14 @@ private:
{
//doWriteSomething();
connectToWebAddress("www.google.com");
/*
if (onReady)
{
onReady();
}*/
//connectToWebAddress("www.google.com");
}
else
{
@ -490,6 +512,189 @@ private:
boost::asio::io_service& ioService;
boost::asio::ssl::stream<boost::asio::ip::tcp::socket> socket;
boost::asio::ip::tcp::resolver& resolver;
};
class ProxyClientSession : public std::enable_shared_from_this<ProxyClientSession> {
public:
ProxyClientSession(boost::asio::ip::tcp::socket socket, boost::asio::io_service& ioServiceIn, boost::asio::ssl::context& context, boost::asio::ip::tcp::resolver& resolverIn)
: socket(std::move(socket))
, resolver(resolverIn)
, proxyClient(ioServiceIn, context, resolverIn)
{
std::cout << "ProxyClientSession create" << std::endl;
}
~ProxyClientSession()
{
std::cout << "ProxyClientSession destroy" << std::endl;
}
void start()
{
auto self(shared_from_this());
proxyClient.onReady = [self]() {self->onProxyReady(); };
proxyClient.start();
}
protected:
unsigned char forwardChar;
unsigned char backwardChar;
void onProxyReady()
{
transferDataForward();
transferDataBackward();
}
void transferDataForward()
{
auto self(shared_from_this());
boost::asio::async_read(socket,
boost::asio::buffer(&forwardChar, 1),
[this, self](boost::system::error_code ec, std::size_t /*length*/)
{
if (!ec)
{
std::cout << static_cast<unsigned int>(forwardChar) << " ";
boost::asio::async_write(proxyClient.getSocket(),
boost::asio::buffer(&forwardChar, 1),
[this, self](boost::system::error_code ec, std::size_t length)
{
if (!ec)
{
transferDataForward();
}
else
{
std::cout << "transferDataForward write error" << std::endl;
proxyClient.lowerSocket().close();
socket.close();
}
});
}
else
{
std::cout << "transferDataForward read error" << std::endl;
proxyClient.lowerSocket().close();
socket.close();
}
});
}
void transferDataBackward()
{
auto self(shared_from_this());
boost::asio::async_read(proxyClient.getSocket(),
boost::asio::buffer(&backwardChar, 1),
[this, self](boost::system::error_code ec, std::size_t /*length*/)
{
if (!ec)
{
boost::asio::async_write(socket,
boost::asio::buffer(&backwardChar, 1),
[this, self](boost::system::error_code ec, std::size_t length)
{
if (!ec)
{
transferDataBackward();
}
else
{
std::cout << "transferDataBackward write error" << std::endl;
proxyClient.lowerSocket().close();
socket.close();
}
});
}
else
{
std::cout << "transferDataBackward read error" << std::endl;
proxyClient.lowerSocket().close();
socket.close();
}
});
}
boost::asio::ip::tcp::socket socket;
boost::asio::ip::tcp::resolver& resolver;
ProxyClient proxyClient;
};
class IntermediateServer
{
public:
IntermediateServer(boost::asio::io_service& io_service,
const boost::asio::ip::tcp::endpoint& endpoint)
: ioService(io_service)
, acceptor(io_service, endpoint)
, socket(io_service)
, resolver(io_service)
//, endpointIterator(resolver.resolve({ "127.0.0.1", "8043" })) //resolver.resolve({ "telegram-proxy.fishrungames.com", "8043" });
, ctx(boost::asio::ssl::context::sslv23)
{
ctx.load_verify_file("rootca.crt");
doAccept();
}
protected:
void doAccept()
{
acceptor.async_accept(socket,
[this](boost::system::error_code ec)
{
if (!ec)
{
std::make_shared<ProxyClientSession>(std::move(socket), ioService, ctx, resolver)->start();
}
doAccept();
});
}
boost::asio::io_service& ioService;
boost::asio::ip::tcp::acceptor acceptor;
boost::asio::ip::tcp::socket socket;
boost::asio::ip::tcp::resolver resolver;
//boost::asio::ip::tcp::resolver::iterator endpointIterator;
boost::asio::ssl::context ctx;
};
@ -501,16 +706,14 @@ int main()
boost::asio::io_service ioService;
//boost::asio::io_service::work work(ioService);
boost::asio::ip::tcp::resolver resolver(ioService);
//auto endpointIterator = resolver.resolve({ "telegram-proxy.fishrungames.com", "8043" });
auto endpointIterator = resolver.resolve({ "127.0.0.1", "8043" });
//Create intermediate:
boost::asio::ip::tcp::endpoint endpoint(boost::asio::ip::tcp::v4(), 8042);
IntermediateServer intermediateServer(ioService, endpoint);
boost::asio::ssl::context ctx(boost::asio::ssl::context::sslv23);
ctx.load_verify_file("rootca.crt");
proxyClient c(ioService, ctx, endpointIterator);
//ProxyClient c(ioService, ctx, endpointIterator);
std::thread t([&ioService]() { ioService.run(); });