1 概念
1.1 正向代理概念
正向代理是一个位于客户端和目标服务器之间的代理服务器（中间服务器）。为了从目标服务器取得内容，客户端向代理服务器发送一个请求，并且指定目标服务器，之后代理向目标服务器转发请求，将获得的内容返回给客户端。正向代理的情况下，客户端必须要进行一些特殊的设置才能使用。
1.2 示例
参考公安部门的网络关系，上下级直接联系，平级之间互不联系。如果A省A市公安局想访问B省B市服务器，那么就要通过多个节点进行中转。手工的多次执行SSH协议多跳几下也是可以，但是如果像访问B省B市内部的网站那就没办法了。这个问题可以用正向代理解决，正向代理就是报文转发。
2 特点
- 正向代理需要主动设置代理服务器ip或者域名进行访问，由设置的服务器ip或者域名去访问内容并返回
- 正向代理是代理客户端，为客户端收发请求，使真实客户端对服务器不可见。
3 正向代理功能实现
3.1 路由参数配置文件
正向代理服务器端维护一个路由参数配置文件，内容为：
每一列分别为：源端口，目标地址，目标端口；
每一行的意思是将5022端口转发到192.168.150.128地址的22端口。
正向代理需要实现的功能：根据路由配置文件，监听源端口，如果源端口有客户端链接上来，就按照路由配置文件查找目的ip和端口进行连接。然后A和C相互通信，就跟没有中间的B一样。
3.2 功能开发
正向代理的实现思路（水平触发+非阻塞）：
//帮助文档
//关闭信号和IO，处理信号
//打开日志//加载代理路由配置文件，加载到容器vroute中。
//初始化服务端用于监听源端口的socket：遍历vroute将源端口一个个监听，并设置非阻塞的。
//监听的socket全部加入epoll中。
while(true) //事件循环
{
int event = epoll_wait();
//循环判断是否是监听的socket
{
//连上来的socket是7
//向目标地址和端口发起tcp连接，socket是8
//那么把7和8的读事件加入epoll
//更新clientsocks数组中两端socket的值和活动时间
}
//如果客户端连接的socket有事件，表示有报文发过来或者连接已经断开。
//如果断开，即读取数据<=0
{
//关闭两端的socket
//清空clientsocks数组两边的值
}
//如果有报文传来，即成功读取了数据
{
//数据原封不动发送给对端
}
}
具体代码实现：
/*
* 程序名：inetd.cpp，正向网络代理服务程序。
* 作者：张咸武
*/
#include "_public.h"
using namespace idc;
// 代理路由参数的结构体。
struct st_route
{
int srcport; // 源端口。
char dstip[31]; // 目标主机的地址。
int dstport; // 目标主机的端口。
int listensock; // 源端口监听的socket。
}stroute;
vector<struct st_route> vroute; // 代理路由的容器。
bool loadroute(const char *inifile); // 把代理路由参数加载到vroute容器。
// 初始化服务端的监听端口。
int initserver(const int port);
int epollfd=0; // epoll的句柄。
int tfd=0; // 定时器的句柄。
#define MAXSOCK 1024 // 最大连接数。
int clientsocks[MAXSOCK]; // 存放每个socket连接对端的socket的值。
int clientatime[MAXSOCK]; // 存放每个socket连接最后一次收发报文的时间。
// 向目标地址和端口发起socket连接。
int conntodst(const char *ip,const int port);
void EXIT(int sig); // 进程退出函数。
clogfile logfile;
int main(int argc,char *argv[])
{
if (argc != 3)
{
printf("\n");
printf("Using :./inetd logfile inifile\n\n");
printf("Sample:./inetd /tmp/inetd.log /etc/inetd.conf\n\n");
printf(" /project/tools/bin/procctl 5 /project/tools/bin/inetd /tmp/inetd.log /etc/inetd.conf\n\n");
printf("本程序的功能是正向代理，如果用到了1024以下的端口，则必须由root用户启动。\n");
printf("logfile 本程序运行的日是志文件。\n");
printf("inifile 路由参数配置文件。\n");
return -1;
}
// 关闭全部的信号和输入输出。
// 设置信号,在shell状态下可用 "kill + 进程号" 正常终止些进程。
// 但请不要用 "kill -9 +进程号" 强行终止。
closeioandsignal(true); signal(SIGINT,EXIT); signal(SIGTERM,EXIT);
// 打开日志文件。
if (logfile.open(argv[1])==false)
{
printf("打开日志文件失败（%s）。\n",argv[1]); return -1;
}
// 把代理路由参数配置文件加载到vroute容器。
if (loadroute(argv[2])==false) return -1;
logfile.write("加载代理路由参数成功(%d)。\n",vroute.size());
// 初始化服务端用于监听的socket。
for (auto &aa:vroute)
{
if ( (aa.listensock=initserver(aa.srcport)) < 0 )
{
// 如果某一个socket初始化失败，忽略它。
logfile.write("initserver(%d) failed.\n",aa.srcport); continue;
}
// 把监听socket设置成非阻塞。
fcntl(aa.listensock,F_SETFL,fcntl(aa.listensock,F_GETFD,0)|O_NONBLOCK);
}
// 创建epoll句柄。
epollfd=epoll_create1(0);
struct epoll_event ev; // 声明事件的数据结构。
// 为监听的socket准备读事件。
for (auto aa:vroute)
{
if (aa.listensock<0) continue;
ev.events=EPOLLIN; // 读事件。
ev.data.fd=aa.listensock; // 指定事件的自定义数据，会随着epoll_wait()返回的事件一并返回。
epoll_ctl(epollfd,EPOLL_CTL_ADD,aa.listensock,&ev); // 把监听的socket的事件加入epollfd中。
}
struct epoll_event evs[10]; // 存放epoll返回的事件。
while (true) // 进入事件循环。
{
// 等待监视的socket有事件发生。
int infds=epoll_wait(epollfd,evs,10,-1);
// 返回失败。
if (infds < 0) { logfile.write("epoll() failed。\n"); EXIT(-1); }
// 遍历epoll返回的已发生事件的数组evs。
for (int ii=0;ii<infds;ii++)
{
logfile.write("已发生事件的socket=%d\n",evs[ii].data.fd);
// 如果发生事件的是listensock，表示有新的客户端连上来。
int jj=0;
for (jj=0;jj<vroute.size();jj++)
{
if (evs[ii].data.fd==vroute[jj].listensock) // 判断是哪个源端口有客户端连上来了。5058
{
// 从已连接队列中获取客户端连上来的socket。 // socket是7
struct sockaddr_in client;
socklen_t len = sizeof(client);
int srcsock = accept(vroute[jj].listensock,(struct sockaddr*)&client,&len);
if (srcsock<0) break;
if (srcsock>=MAXSOCK)
{
logfile.write("连接数已超过最大值%d。\n",MAXSOCK); close(srcsock); break;
}
// 向目标地址和端口发起连接，如果连接失败，会被epoll发现，将关闭通道。
int dstsock=conntodst(vroute[jj].dstip,vroute[jj].dstport); // socket是8
if (dstsock<0) { close(srcsock); break; }
if (dstsock>=MAXSOCK)
{
logfile.write("连接数已超过最大值%d。\n",MAXSOCK); close(srcsock); close(dstsock); break;
}
logfile.write("accept on port %d,client(%d,%d) ok。\n",vroute[jj].srcport,srcsock,dstsock);
// 为新连接的两个socket准备读事件，并添加到epoll中。
ev.data.fd=srcsock; ev.events=EPOLLIN;
epoll_ctl(epollfd,EPOLL_CTL_ADD,srcsock,&ev);
ev.data.fd=dstsock; ev.events=EPOLLIN;
epoll_ctl(epollfd,EPOLL_CTL_ADD,dstsock,&ev);
// 更新clientsocks数组中两端soccket的值和活动时间。
clientsocks[srcsock]=dstsock; clientatime[srcsock]=time(0);
clientsocks[dstsock]=srcsock; clientatime[dstsock]=time(0);
break;
}
}
// 如果jj<vroute.size()，表示事件在上面的for循环中已被处理，流程不必往下。
if (jj<vroute.size()) continue;
// 如果是客户端连接的socke有事件，表示有报文发过来或者连接已断开。
char buffer[5000]; // 存放从接收缓冲区中读取的数据。
int buflen=0; // 从接收缓冲区中读取的数据的大小。
// 从通道的一端读取数据。
if ( (buflen=recv(evs[ii].data.fd,buffer,sizeof(buffer),0)) <= 0 )
{
// 如果连接已断开，需要关闭通道两端的socket。
logfile.write("client(%d,%d) disconnected。\n",evs[ii].data.fd,clientsocks[evs[ii].data.fd]);
close(evs[ii].data.fd); // 关闭客户端的连接。
close(clientsocks[evs[ii].data.fd]); // 关闭客户端对端的连接。
clientsocks[clientsocks[evs[ii].data.fd]]=0; // 把数组中对端的socket置空，这一行代码和下一行代码的顺序不能乱。
clientsocks[evs[ii].data.fd]=0; // 把数组中本端的socket置空，这一行代码和上一行代码的顺序不能乱。
continue;
}
// 成功的读取到了数据，把接收到的报文内容原封不动的发给通道的对端。
logfile.write("from %d to %d,%d bytes。\n",evs[ii].data.fd,clientsocks[evs[ii].data.fd],buflen);
send(clientsocks[evs[ii].data.fd],buffer,buflen,0);
// 更新通道两端socket的活动时间。
clientatime[evs[ii].data.fd]=time(0);
clientatime[clientsocks[evs[ii].data.fd]]=time(0);
}
}
return 0;
}
// 初始化服务端的监听端口。
int initserver(const int port)
{
int sock = socket(AF_INET,SOCK_STREAM,0);
if (sock < 0)
{
logfile.write("socket(%d) failed.\n",port); return -1;
}
int opt = 1; unsigned int len = sizeof(opt);
setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,&opt,len);
struct sockaddr_in servaddr;
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(port);
if (bind(sock,(struct sockaddr *)&servaddr,sizeof(servaddr)) < 0 )
{
logfile.write("bind(%d) failed.\n",port); close(sock); return -1;
}
if (listen(sock,5) != 0 )
{
logfile.write("listen(%d) failed.\n",port); close(sock); return -1;
}
return sock;
}
// 把代理路由参数加载到vroute容器。
bool loadroute(const char *inifile)
{
cifile ifile;
if (ifile.open(inifile)==false)
{
logfile.write("打开代理路由参数文件(%s)失败。\n",inifile); return false;
}
string strbuffer;
ccmdstr cmdstr;
while (true)
{
if (ifile.readline(strbuffer)==false) break;
// 删除说明文字，#后面的部分。
auto pos=strbuffer.find("#");
if (pos!=string::npos) strbuffer.resize(pos);
replacestr(strbuffer," "," ",true); // 把两个空格替换成一个空格，注意第四个参数。
deletelrchr(strbuffer,' '); // 删除两边的空格。
// 拆分参数。
cmdstr.splittocmd(strbuffer," ");
if (cmdstr.size()!=3) continue;
memset(&stroute,0,sizeof(struct st_route));
cmdstr.getvalue(0,stroute.srcport); // 源端口。
cmdstr.getvalue(1,stroute.dstip); // 目标地址。
cmdstr.getvalue(2,stroute.dstport); // 目标端口。
vroute.push_back(stroute);
}
return true;
}
// 向目标地址和端口发起socket连接。
int conntodst(const char *ip,const int port)
{
// 第1步：创建客户端的socket。
int sockfd;
if ( (sockfd = socket(AF_INET,SOCK_STREAM,0))==-1) return -1;
// 第2步：向服务器发起连接请求。
struct hostent* h;
if ( (h = gethostbyname(ip)) == 0 ) { close(sockfd); return -1; }
struct sockaddr_in servaddr;
memset(&servaddr,0,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(port); // 指定服务端的通讯端口。
memcpy(&servaddr.sin_addr,h->h_addr,h->h_length);
// 把socket设置为非阻塞。
fcntl(sockfd,F_SETFL,fcntl(sockfd,F_GETFD,0)|O_NONBLOCK);
if (connect(sockfd, (struct sockaddr *)&servaddr,sizeof(servaddr))<0)
{
if (errno!=EINPROGRESS)
{
logfile.write("connect(%s,%d) failed.\n",ip,port); return -1;
}
}
return sockfd;
}
void EXIT(int sig)
{
logfile.write("程序退出，sig=%d。\n\n",sig);
// 关闭全部监听的socket。
for (auto &aa:vroute)
if (aa.listensock>0) close(aa.listensock);
// 关闭全部客户端的socket。
for (auto aa:clientsocks)
if (aa>0) close(aa);
close(epollfd); // 关闭epoll。
close(tfd); // 关闭定时器。
exit(0);
}
至此，正向代理模块基本功能已经实现。
扫一扫
5261
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
