TCP to TCP 数据转发

TCP to TCP 数据转发

MSN：yiqianfeng@hotmail.com
需要转载请联系作者

目的
    我们部门在内网有一台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 ;
   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;
}

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