TCP to TCP 数据转发技术_C/C++_编程开发_程序员俱乐部

目的
    我们部门在内网有一台Web服务器A，用于部门文档信息管理，可分公司的员工，或出差的员工，都不能直接访问这台机器。
所以还得把发邮件来完成文档的提交。如何实现外地员工能访问Web服务器A呢？公司有一台可以访问外网的机器B，要是在B上运行
一个数据转发程序，固定地把外网发送到某个端口的数据转发到A上的Web端口，同时把A上回送的数据转发出去，不就一切OK了吗！


设计
//主程序
Main()
{
   beginthread(Main Thread());
}

//主线程
Main Thread(Param)
{
   创建"本地监听SOCK"
   while(true)
   {
      等待客户端连接
      等到后，创建"目标SOCK"
      连接到目的地
      连接成功后，创建"Wait Thread"
   }
}
//Wait 线程 
Wait Thread(Param)
{
  创建"连接SOCK"->"目标SOCK"数据转发Thread
  创建"目标SOCK"->"连接SOCK"数据转发Thread
  等待这两个线程结束
  释放SOCK资源
}
//TCP2TCP数据转发线程
TCP2TCP Thread(Param)
{
   SOCK FROM,TO
   while(read  >0)
   {
read(From,Buff);
Send(To,Buff);
   }
}

参考代码
参数结构
struct AGENTGATEWAYINFO{
int  LocalPort;  //本地Port
int  TargetPort;  //目标Port
int  TargetPort1; //目标Port2
char TargetIP[64]; //目标IP

char SpecialIP[64]; //特定的IP
int  SpecialTargetPort;//特定的目标port
char SpecialTargetIP[64];//特定的目标IP

int  ProxyType;  //代理类型
char ProxyIP[30]; //代理IP
int  ProxyPort;  //代理端口
char ProxyUser[50]; //代理用户
char ProxyPassword[50];//代理密码
};
//主线程
DWORD mainthread(LPVOID lpvoid)
{
//获取传人的参数
AGENTGATEWAYINFO *psi = (AGENTGATEWAYINFO *)lpvoid;
AGENTGATEWAYINFO  si;
memcpy(&si,psi,sizeof(AGENTGATEWAYINFO));
SOCKET listensock,sock[2];
unsigned long hostip;
char chostip[64]="0";
delete psi;

printf("\n************Start Agent Tcp GateWay %d==>%s:%d****************",si.LocalPort,si.TargetIP,si.TargetPort);
psi = NULL;
hostip=inet_addr(chostip);
//先建立本地监听socket，用来等待需要转发的连接
if((listensock = socket(AF_INET,SOCK_STREAM,0)) == NULL)
{
  printf("\nTcpGateWay:Can't create incoming socket.\n");
  return -1;
}
sockaddr_in incomingsin;
incomingsin.sin_family = AF_INET;
incomingsin.sin_port = htons(si.LocalPort);
if(hostip!=INADDR_NONE)
{
  incomingsin.sin_addr.s_addr=hostip;
}
else
{
  printf("\nTcpGateWay:Can't look up local ip.\n");
  return -1;
}

sockaddr_in clientaddr;
int iclientaddr = sizeof(clientaddr);

bind(listensock,(LPSOCKADDR)&incomingsin,sizeof(sockaddr));
listen(listensock,10);
printf("\nTcpGateWay:Listening at ip: %s , port: %d.\n",inet_ntoa(incomingsin.sin_addr) , si.LocalPort);
_AddSock(listensock);
while (1)
{
  //开始等待新的连接
  memset( &clientaddr, 0 , sizeof( clientaddr) );
  sock[0] = accept(listensock,(struct sockaddr *)&clientaddr,&iclientaddr);
 
  if(sock[0] == -1)
  {
   printf("client eq -1 ,break it!\n");
   Sleep(1000);
   continue;
  }
  else
  { //等到了
   SOCKETINFO *psgc = new SOCKETINFO();
   psgc->sock[0] = sock[0];
   strcpy(psgc->SockIP[0],inet_ntoa(clientaddr.sin_addr));
   psgc->SockPort[0] = ntohs(clientaddr.sin_port);
   printf("\nTcpGateWay:Incoming connection from %s:%d.",psgc->SockIP[0],psgc->SockPort[0]);

   strcpy(psgc->SockIP[1],si.TargetIP);
   psgc->SockPort[1] = si.TargetPort;
   //产生一个目标SOCK
   if((psgc->sock[1] = socket(AF_INET,SOCK_STREAM,0)) == NULL)
   {
    printf("\nTcpGateWay:Can't create sending socket.");
    delete psgc;
    continue;
   }
   CWSocket wsock0 ;
&nbp;  wsock0.Attach(psgc->sock[1]);

   try
   { //通过代理连接目标
    switch(si.ProxyType)
    {
    case 1://SOCK4
      wsock0.ConnectViaSocks4(psgc->SockIP[1], psgc->SockPort[1],si.ProxyIP, si.ProxyPort);
     break;
    case 2://SOCK5
     if(strlen(si.ProxyUser) > 0)
     {
      wsock0.ConnectViaSocks5(psgc->SockIP[1], psgc->SockPort[1],si.ProxyIP, si.ProxyPort,
       si.ProxyUser,
       si.ProxyPassword);
     }else
     {
      wsock0.ConnectViaSocks5(psgc->SockIP[1], psgc->SockPort[1],si.ProxyIP, si.ProxyPort);
     }

     break;
    case 3://HTTP11
     {
      CString sProxyResponse;
      if(strlen(si.ProxyUser) > 0)
      {
       wsock0.ConnectViaHTTPProxy(psgc->SockIP[1], psgc->SockPort[1],si.ProxyIP, si.ProxyPort,
        sProxyResponse,
        si.ProxyUser,
        si.ProxyPassword);
      }else
      {
       wsock0.ConnectViaHTTPProxy(psgc->SockIP[1], psgc->SockPort[1],si.ProxyIP, si.ProxyPort,sProxyResponse);
      }
     }
     break;
    default:
     wsock0.Connect(psgc->SockIP[1],psgc->SockPort[1]);
     break;
    }
   }
   catch(CWSocketException* pEx)
   {
    pEx->Delete();
    wsock0.Detach();
    _DelSock(psgc->sock[1]);
    _DelSock(psgc->sock[0]);
    printf("\nTcpGateWay:Can't forward to %s:%d.",si.TargetIP,si.TargetPort);
    delete psgc;
    continue;
   }
   wsock0.Detach();
   printf("\n%s:%d ==> %s:%d.\n",psgc->SockIP[0],psgc->SockPort[0],psgc->SockIP[1],psgc->SockPort[1]);
   DWORD threadid;
   //连接成功，创建监护线程，用来监控socket
   psgc->hWaitThread = CreateThread(NULL,0,waitthread,(LPVOID)psgc,NULL,&threadid);
   _AddThread(psgc->hWaitThread);
  }
}
_DelSock(listensock);
return 0;
}
//监护线程，用来监控socket
DWORD waitthread(LPVOID lpvoid)
{
SOCKETINFO *psgc  = (SOCKETINFO *)lpvoid;
HANDLE hWaitThread = psgc->hWaitThread;
_AddSock(psgc->sock[1]);
_AddSock(psgc->sock[0]);
//创建连接端->目的端数据转发线程
psgc->hThreads[0] = CreateThread(NULL,0,tcp2tcp,(LPVOID)psgc,NULL,&psgc->id[0]);
_AddThread(psgc->hThreads[0]);
//创建目的端->连接端数据转发线程
psgc->hThreads[1] = CreateThread(NULL,0,tcp2tcp,(LPVOID)psgc,NULL,&psgc->id[1]);
_AddThread(psgc->hThreads[1]);
//等待线程结束
DWORD  waiter;
waiter = WaitForMultipleObjects(MAX_SOCKS,psgc->hThreads,FALSE,INFINITE);

if(waiter == WAIT_OBJECT_0)
{
  _DelThread(psgc->hThreads[0]);
  _DelSock(psgc->sock[0]);
  Sleep(50);
  _DelSock(psgc->sock[1]);
  WaitForSingleObject(psgc->hThreads[1],2000);
  _DelThread(psgc->hThreads[1]);

}else if(waiter == WAIT_OBJECT_0+1)
{
  _DelThread(psgc->hThreads[1]);
  _DelSock(psgc->sock[1]);
  Sleep(50);
  _DelSock(psgc->sock[0]);
  WaitForSingleObject(psgc->hThreads[0],2000);
  _DelThread(psgc->hThreads[0]);
}else
{
  _DelSock(psgc->sock[1]);
  Sleep(50);
  _DelSock(psgc->sock[0]);
  _DelThread(psgc->hThreads[1]);
  _DelThread(psgc->hThreads[0]);
}
printf("\n%s:%d Socket closed.\n",psgc->SockIP[0],psgc->SockPort[0]);
delete psgc;
_PostDelThread(hWaitThread);
ExitThread(0);
return 0;
}
DWORD WINAPI tcp2tcp(LPVOID lpvoid)
{
SOCKETINFO *psgc  = (SOCKETINFO *)lpvoid;
DWORD threadid=GetCurrentThreadId();
SOCKET s[2];
HANDLE ReadEvent = NULL ;
BOOL isFirstRead=TRUE;
CString sHexDispData;
BOOL bIsDisplay0 = TRUE;
if(psgc->id[0] == threadid)
{
  s[0] = psgc->sock[0];
  s[1] = psgc->sock[1];
  ReadEvent = psgc->ReadEvent[0];
}else
{
  s[1] = psgc->sock[0];
  s[0] = psgc->sock[1];
  ReadEvent = psgc->ReadEvent[1];
  bIsDisplay0 = FALSE;
}
printf("\n***New TCP2TCP Thread %d recv %d send %d.\n",threadid,s[0],s[1]);
char incomingbuff[maxsize];
int read = 1;
while(read  >0)
{
  read  = recv(s[0],incomingbuff, maxsize,0);
  if(read <=0  ) break;

  printf("\nTCP2TCP Thread %d recv %d bytes.",threadid,read);
  int npos=0;
  while(npos < read)
  {
   int nsendcount = send(s[1],incomingbuff+npos,read-npos,0);
   if(nsendcount <= 0 ) return 0;
   printf("\nTCP2TCP Thread %d send %d bytes.",threadid,nsendcount);
   npos += nsendcount;
  }
}
printf("\nTCP2TCP Thread %d LastError %d",threadid,GetLastError());
return 0;
}

需要完整源代码请留言或和作者联系。