143 lines
3.5 KiB
C++
143 lines
3.5 KiB
C++
#include "examples/socks4a/tunnel.h"
|
|
|
|
#include "muduo/net/Endian.h"
|
|
#include <stdio.h>
|
|
#include <netdb.h>
|
|
#include <unistd.h>
|
|
|
|
using namespace muduo;
|
|
using namespace muduo::net;
|
|
|
|
EventLoop* g_eventLoop;
|
|
std::map<string, TunnelPtr> g_tunnels;
|
|
|
|
void onServerConnection(const TcpConnectionPtr& conn)
|
|
{
|
|
LOG_DEBUG << conn->name() << (conn->connected() ? " UP" : " DOWN");
|
|
if (conn->connected())
|
|
{
|
|
conn->setTcpNoDelay(true);
|
|
}
|
|
else
|
|
{
|
|
std::map<string, TunnelPtr>::iterator it = g_tunnels.find(conn->name());
|
|
if (it != g_tunnels.end())
|
|
{
|
|
it->second->disconnect();
|
|
g_tunnels.erase(it);
|
|
}
|
|
}
|
|
}
|
|
|
|
void onServerMessage(const TcpConnectionPtr& conn, Buffer* buf, Timestamp)
|
|
{
|
|
LOG_DEBUG << conn->name() << " " << buf->readableBytes();
|
|
if (g_tunnels.find(conn->name()) == g_tunnels.end())
|
|
{
|
|
if (buf->readableBytes() > 128)
|
|
{
|
|
conn->shutdown();
|
|
}
|
|
else if (buf->readableBytes() > 8)
|
|
{
|
|
const char* begin = buf->peek() + 8;
|
|
const char* end = buf->peek() + buf->readableBytes();
|
|
const char* where = std::find(begin, end, '\0');
|
|
if (where != end)
|
|
{
|
|
char ver = buf->peek()[0];
|
|
char cmd = buf->peek()[1];
|
|
const void* port = buf->peek() + 2;
|
|
const void* ip = buf->peek() + 4;
|
|
|
|
sockaddr_in addr;
|
|
memZero(&addr, sizeof addr);
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = *static_cast<const in_port_t*>(port);
|
|
addr.sin_addr.s_addr = *static_cast<const uint32_t*>(ip);
|
|
|
|
bool socks4a = sockets::networkToHost32(addr.sin_addr.s_addr) < 256;
|
|
bool okay = false;
|
|
if (socks4a)
|
|
{
|
|
const char* endOfHostName = std::find(where+1, end, '\0');
|
|
if (endOfHostName != end)
|
|
{
|
|
string hostname = where+1;
|
|
where = endOfHostName;
|
|
LOG_INFO << "Socks4a host name " << hostname;
|
|
InetAddress tmp;
|
|
if (InetAddress::resolve(hostname, &tmp))
|
|
{
|
|
addr.sin_addr.s_addr = tmp.ipNetEndian();
|
|
okay = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
okay = true;
|
|
}
|
|
|
|
InetAddress serverAddr(addr);
|
|
if (ver == 4 && cmd == 1 && okay)
|
|
{
|
|
TunnelPtr tunnel(new Tunnel(g_eventLoop, serverAddr, conn));
|
|
tunnel->setup();
|
|
tunnel->connect();
|
|
g_tunnels[conn->name()] = tunnel;
|
|
buf->retrieveUntil(where+1);
|
|
char response[] = "\000\x5aUVWXYZ";
|
|
memcpy(response+2, &addr.sin_port, 2);
|
|
memcpy(response+4, &addr.sin_addr.s_addr, 4);
|
|
conn->send(response, 8);
|
|
}
|
|
else
|
|
{
|
|
char response[] = "\000\x5bUVWXYZ";
|
|
conn->send(response, 8);
|
|
conn->shutdown();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else if (!conn->getContext().empty())
|
|
{
|
|
const TcpConnectionPtr& clientConn
|
|
= boost::any_cast<const TcpConnectionPtr&>(conn->getContext());
|
|
clientConn->send(buf);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
if (argc < 2)
|
|
{
|
|
fprintf(stderr, "Usage: %s <listen_port>\n", argv[0]);
|
|
}
|
|
else
|
|
{
|
|
LOG_INFO << "pid = " << getpid() << ", tid = " << CurrentThread::tid();
|
|
|
|
uint16_t port = static_cast<uint16_t>(atoi(argv[1]));
|
|
InetAddress listenAddr(port);
|
|
|
|
EventLoop loop;
|
|
g_eventLoop = &loop;
|
|
|
|
TcpServer server(&loop, listenAddr, "Socks4");
|
|
|
|
server.setConnectionCallback(onServerConnection);
|
|
server.setMessageCallback(onServerMessage);
|
|
|
|
server.start();
|
|
|
|
loop.loop();
|
|
}
|
|
}
|
|
|