UDP下的SOCK5代理实现

Author:zfive5(zhaozidong)
Email :zfive5@yahoo.com.cn

    前些时间,以前同事叫我帮忙写一个upd下的sock5代理,由于一些原因
没有帮成他的忙,我在这里再次对他说抱歉! 关于sock5代理的包文和数据
格式可以参考以下url中的内容:
  http://www.vckbase.net/document/viewdoc/?id=853
  http://www.vckbase.net/document/viewdoc/?id=852
  http://www.china-pub.com/computers/eMook/emooknew/rfctxt/RFC1928.txt

    牙看来还是拔不了呀！医院的口腔科星期天居然没有人！！！大早晨在公交车
上一来一回8圆钱啊！

upd Sock5代理下的包数据:

 client:
      ver( 1 byte)+num method( 1 byte)+ methods(1 -255 byte)
      (05 02 00 02)

 server:
     
      ver( 1 byte)+ num method(1 byte)
      (05 00 ) or ( 05 02)

 *client( 当返回 02 时):
      
      ver(1  byte)+ name len(1 byte) +name (1-255 byte) +pass len(1 byte)+password(1 -255 byte)
      ( 01 01 61 01 61)

 *server( 当返回 02 时):
      ver(1 byte)+status(1 byte)
      (05 00  )or (05 XX)

  client:
     
      ver( 1 byte)+cmd(1 byte)+rsv( 1 byte)+address type( 1 byte)+ addrss( ?? byte) +port( 2 byte)

      (05 03 00 01 7f 00 00 01 0f A0)

  server:
      ver(1 byte)+reponse(1 byte)+rsv(1 byte)+address type(1 byte)+address(?? byte)+port( 2 byte)
      (05 00 00 01 7f 00 00 01 0f A0)

  client:
     rsv (2 byte) + fragment( 1 byte)+ address type (1 byte)+ addr(?? byte)+ port( 2 byte) +data(??)
     (00 00 00 01 7f 00 00 01 0f a0 .....)
  server:
     rsv (2 byte) + fragment( 1 byte)+ address type (1 byte)+ addr(?? byte)+ port( 2 byte) +data(??)
     (00 00 00 01 7f 00 00 01 0f a0 .....)
   ....

test server代码:
   
SOCKET s=socket(AF_INET,SOCK_DGRAM,0);
if(s==INVALID_SOCKET)
{
   closesocket(s);
   return 0;
}
struct sockaddr_in addr;
addr.sin_family=AF_INET;
addr.sin_addr.s_addr=ADDR_ANY;
addr.sin_port=htons(4000);
if(bind(s,(struct sockaddr *)&addr,sizeof(struct sockaddr))==SOCKET_ERROR)
{
   closesocket(s);
   return 0;
}

while(1==1)
{
   char sz_buf[1024];
   struct sockaddr in;
   int    n_len=sizeof(struct sockaddr);
   if(recvfrom(s,sz_buf,1024,0,(struct sockaddr *)&in,&n_len)==SOCKET_ERROR)
   {
      closesocket(s);
      return 0;
   }

   if(sendto(s,"Hello World",strlen("Hello World"),0,(struct sockaddr *)&in,sizeof(struct sockaddr_in))==SOCKET_ERROR)
   {
      closesocket(s);
      return 0;
   }
}

closesocket(s);
return 0;

test client:

SOCKET s=socket(AF_INET,SOCK_DGRAM,0);
if(s==INVALID_SOCKET)
{
   closesocket(s);
   return;
}

struct sockaddr_in addr1;
addr1.sin_family=AF_INET;
addr1.sin_addr.s_addr=ADDR_ANY;
addr1.sin_port=htons(4001);
if(bind(s,(struct sockaddr *)&addr1,sizeof(struct sockaddr_in))==SOCKET_ERROR)
{
   closesocket(s);
   return;
}

zf5_sock5_udp udp;
strcpy(udp.m_sz_name,"a");
strcpy(udp.m_sz_pass,"a");
if(udp.connect_sock5_udp("127.0.0.1",1080,&addr1)<0)
{
   closesocket(s);
   return;
}

struct sockaddr_in addr2;
addr2.sin_family=AF_INET;
addr2.sin_addr.s_addr=inet_addr("127.0.0.1");
addr2.sin_port=htons(4000);
if(udp.send_sock5_udp(s,"hello",strlen("hello"),&addr2)<0)
{
  closesocket(s);
  return;
}

char sz_buf[1024]="";
memset(sz_buf,0,1024);
struct sockaddr_in addr3;
if(udp.recv_sock5_udp(s,sz_buf,&addr3)<0)
{
   closesocket(s);
   return;
}

AfxMessageBox(sz_buf);
return;

zf5_sock5_udp.h文件:

#pragma once
#pragma comment(lib,"ws2_32.lib")

#include <winsock2.h>
#define ZF5_BUFFER_SIZE1 255
#define ZF5_BUFFER_SIZE2 1024
#define ZF5_BUFFER_SIZE3 8192
class zf5_sock5_udp
{
public:

  //构造函数
  zf5_sock5_udp(void);

  //析构函数
  virtual ~zf5_sock5_udp(void);

  //创建sock5的upd代理
  int  connect_sock5_udp(char *sz_addr_proxy,int n_port_proxy,struct sockaddr_in *p_addr);

  //通过sock5发送udp数据
  int  send_sock5_udp(int s_udp,char *sz_buf,int n_len,struct sockaddr_in *p_addr);

  //通过sock5接收udp数据
  int  recv_sock5_udp(int s_udp,char *sz_buf,struct sockaddr_in *p_addr);
 
  //关闭sock5的udp代理
  int  close_sock5_udp();
 
  //代理服务器用户名称
  char m_sz_name[ZF5_BUFFER_SIZE1+2];
 
  //代理服务器用户密码
  char m_sz_pass[ZF5_BUFFER_SIZE1+2];

private:
  //远端udp服务器地址
  char m_sz_udp_srv[ZF5_BUFFER_SIZE2+2];
 
  //远端udp服务器端口
  int  m_n_port;
 
  //地址类型
  int  m_n_addrtype;
 
  //sock5代理的tcp连接
  int  m_n_sock5;
};

zf5_sock5_udp.cpp文件:

#include "stdafx.h"
#include "./zf5_sock5_udp.h"

//构造函数
zf5_sock5_udp::zf5_sock5_udp(void)
{
 WSADATA wsa;
 WSAStartup(MAKEWORD(1,1),&wsa);
 m_n_addrtype=-1;
 m_n_port=0;
 m_n_sock5=INVALID_SOCKET;
 m_sz_name[0]=0;
 m_sz_pass[0]=0;
 m_sz_udp_srv[0]=0;

}

//析构函数
zf5_sock5_udp::~zf5_sock5_udp(void)
{
 close_sock5_udp();
 WSACleanup();
}

//创建sock5的upd代理
int  zf5_sock5_udp::connect_sock5_udp(char *sz_addr_proxy,int n_port_proxy,struct sockaddr_in *p_addr)
{
 SOCKET s_sock5_tcp=INVALID_SOCKET;

 s_sock5_tcp=socket(AF_INET,SOCK_STREAM,0);
 if(s_sock5_tcp==INVALID_SOCKET)
 {
  return -1;
 }
 
 timeval tv;
 tv.tv_sec=10;
 tv.tv_usec=0;
 if(setsockopt(s_sock5_tcp,SOL_SOCKET,SO_RCVTIMEO,(char *)&tv,sizeof(timeval))==SOCKET_ERROR)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 struct sockaddr_in sa_1;
 sa_1.sin_family=AF_INET;
 sa_1.sin_addr.s_addr=inet_addr(sz_addr_proxy);
 sa_1.sin_port=htons(n_port_proxy);
 if(sa_1.sin_addr.s_addr==INADDR_NONE)
 {
  struct hostent *p=gethostbyname(sz_addr_proxy);
  if(p==NULL)
  {
   closesocket(s_sock5_tcp);
   return -1;
  }
  sa_1.sin_addr.s_addr=((LPIN_ADDR)p->h_addr)->s_addr;
 }

 if(connect(s_sock5_tcp,(struct sockaddr *)&sa_1,sizeof(struct sockaddr_in))==SOCKET_ERROR)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 char sz_buf[ZF5_BUFFER_SIZE2+2];

 memset(sz_buf,0,ZF5_BUFFER_SIZE2);
 sz_buf[0]=0x05;
 sz_buf[1]=0x02;
 sz_buf[2]=0x00;
 sz_buf[3]=0x02;
 if(send(s_sock5_tcp,sz_buf,4,0)==SOCKET_ERROR)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 memset(sz_buf,0,ZF5_BUFFER_SIZE2);
 if(recv(s_sock5_tcp,sz_buf,ZF5_BUFFER_SIZE2,0)==SOCKET_ERROR)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

    if(sz_buf[0]==0x05&&sz_buf[1]==0x02)
 {
  int n_temp1=0;
  int n_temp2=0;
  memset(sz_buf,0,ZF5_BUFFER_SIZE2);
  n_temp1=strlen(m_sz_name);
  sz_buf[0]=0x01;
  sz_buf[1]=n_temp1;
  memcpy((void *)&(sz_buf[2]),(void *)&(m_sz_name[0]),n_temp1);
  n_temp2=strlen(m_sz_pass);
  sz_buf[1+n_temp1+1]=n_temp2;
  memcpy((void *)&(sz_buf[1+n_temp1+2]),(void *)&(m_sz_pass[0]),n_temp2);
  if(send(s_sock5_tcp,sz_buf,n_temp1+n_temp2+3,0)==SOCKET_ERROR)
  {
   closesocket(s_sock5_tcp);
   return -1;
  }

  memset(sz_buf,0,ZF5_BUFFER_SIZE2);
  if(recv(s_sock5_tcp,sz_buf,ZF5_BUFFER_SIZE2,0)==SOCKET_ERROR)
  {
   closesocket(s_sock5_tcp);
   return -1;
  }

  if(sz_buf[0]!=0x01||sz_buf[1]!=0x00)
  {
   closesocket(s_sock5_tcp);
   return -1;
  }
 }
 else if(sz_buf[0]!=0x05||sz_buf[1]!=0x00)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 memset(sz_buf,0,ZF5_BUFFER_SIZE2);
 sz_buf[0]=0x05;
 sz_buf[1]=0x03;
 sz_buf[2]=0x00;
 sz_buf[3]=0x01;

 *((int *)(&sz_buf[4]))=p_addr->sin_addr.s_addr;
 *((short *)(&(sz_buf[8])))=(short)p_addr->sin_port;

 if(send(s_sock5_tcp,sz_buf,10,0)==SOCKET_ERROR)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 memset(sz_buf,0,ZF5_BUFFER_SIZE2);
 if(recv(s_sock5_tcp,sz_buf,ZF5_BUFFER_SIZE2,0)==SOCKET_ERROR)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 if(sz_buf[0]!=0x05||sz_buf[1]!=0x00)
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 if(sz_buf[3]==0x01)
 {
  int n_ip=*((int *)&sz_buf[4]);
  struct in_addr in;
  in.S_un.S_addr=n_ip;
  memset((void *)m_sz_udp_srv,0,sizeof(m_sz_udp_srv));
  strcpy(m_sz_udp_srv,inet_ntoa(in));
  m_n_port=htons(*((short *)&(sz_buf[8])));
  m_n_addrtype=0x01;
 }
 else if(sz_buf[3]==0x03)
 {
  memset((void *)m_sz_udp_srv,0,sizeof(m_sz_udp_srv));
  strncpy(m_sz_udp_srv,&sz_buf[5],sz_buf[4]);
  m_n_port=htons(*((short *)&(sz_buf[5+sz_buf[4]])));
  m_n_addrtype=0x03;
 }
 else
 {
  closesocket(s_sock5_tcp);
  return -1;
 }

 m_n_sock5=s_sock5_tcp;

 return 0;
}

//通过sock5发送udp数据
int  zf5_sock5_udp::send_sock5_udp(int s_udp,char *sz_buf,int n_len,struct sockaddr_in *p_addr)
{
 char sz_buf1[ZF5_BUFFER_SIZE3+2];
 int  n_len1=0;
 sz_buf1[0]=0x00;
 sz_buf1[1]=0x00;
 sz_buf1[2]=0x00;
 sz_buf1[3]=0x01;
    memcpy((void *)&sz_buf1[4],(void *)&(p_addr->sin_addr.s_addr),4);
    memcpy((void *)&sz_buf1[8],(void *)&(p_addr->sin_port),2);
 struct sockaddr_in addr;
 addr.sin_family=AF_INET;
 addr.sin_addr.s_addr=inet_addr(m_sz_udp_srv);
 addr.sin_port=htons(m_n_port);
 if(addr.sin_addr.s_addr==INADDR_NONE)
 {
  struct hostent *p=gethostbyname(m_sz_udp_srv);
  if(p==NULL)
  {
   return -1;
  }
  addr.sin_addr.s_addr=((LPIN_ADDR)p->h_addr)->s_addr;
 }

 memcpy(&sz_buf1[10],sz_buf,n_len);

 int n_ret=-1;
 n_ret=sendto(s_udp,sz_buf1,n_len+10,0,(struct sockaddr *)&addr,sizeof(struct sockaddr_in ));

 return n_ret;
}

//通过sock5接收udp数据
int  zf5_sock5_udp::recv_sock5_udp(int s_udp,char *sz_buf,struct sockaddr_in *p_addr)
{
 char sz_buf1[ZF5_BUFFER_SIZE3+2];
 struct sockaddr_in addr;
 int n_ret=-1;
 int n_len1=0;
 int n_len2=sizeof(struct sockaddr_in);
 n_ret=recvfrom(s_udp,sz_buf1,ZF5_BUFFER_SIZE3,0,(struct sockaddr *)&addr,&n_len2);
 if(n_ret==SOCKET_ERROR)
 {
  return -1;
 }

 if(sz_buf1[3]==0x01)
 {
  p_addr->sin_family=AF_INET;
  p_addr->sin_addr.s_addr=*((int *)&(sz_buf1[4]));
  p_addr->sin_port=*((short *)&(sz_buf1[8]));
  n_ret-=10;
  memcpy(sz_buf,&sz_buf1[10],n_ret);
 }
 else if(sz_buf1[3]==0x03)
 {
  char sz_buf2[ZF5_BUFFER_SIZE2+2];
  memset(sz_buf2,0,ZF5_BUFFER_SIZE2);
  strncpy(sz_buf2,sz_buf2,sz_buf1[4]);
  struct hostent *p=gethostbyname(sz_buf2);
  if(p==NULL)
  {
   return -1;
  }
  n_len1=4+sz_buf1[4];
  p_addr->sin_family=AF_INET;
  p_addr->sin_addr.s_addr=((LPIN_ADDR)p->h_addr)->s_addr;
  p_addr->sin_port=*((short *)&(sz_buf1[n_len1]));
 
  n_ret-=(n_len1+2);
  memcpy((void *)sz_buf,(void *)&sz_buf1[n_len1],n_ret);

  n_ret-=(4+2+1+sz_buf1[4]);
 }
 else
 {
  return -1;
 }

 return n_ret;
}

//关闭sock5的udp代理
int  zf5_sock5_udp::close_sock5_udp()
{
 if(m_n_sock5!=INVALID_SOCKET)
 {
  closesocket(m_n_sock5);
  m_n_sock5=INVALID_SOCKET;
 }

 return 0;
}