// ProxySocketSocksV5.h : header file
//

/*
使用Socks5代理協議的socket類，目前只支援UDP協議。

[email protected] 24/09/2002
*/
#if !defined(AFX_PROXYSOCKETSOCKSV5_H__07BD21EA_0867_4F91_A82A_72F3AF706228__INCLUDED_)
#define AFX_PROXYSOCKETSOCKSV5_H__07BD21EA_0867_4F91_A82A_72F3AF706228__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

//驗證方法選擇 請求
struct SOCKS5SELECTREQEST
{
BYTE ver; //版本號，必須是5
BYTE nmethods; //認證方法的數量
BYTE methods[255]; //認證方法
/*
· '00' 不需要認證
· '01'    GSSAPI
· '02'    使用者名稱/密碼
· '03' -- X'7F'   由IANA分配
· '80' -- X'FE'  為私人方法所保留的
· 'FF'      沒有可以接受的方法
*/
};

//驗證方法選擇 迴應
struct SOCKS5SELECTANSWER
{
BYTE ver; //版本號，必須是5
BYTE method; //驗證方法，是SOCKS5REQUEST1中methods[255]中的一個值
//具體見SOCKS5REQUEST1.methods[255]的說明
};

//驗證 請求
struct SOCKS5AUTHENTICATE
{
BYTE ver; //版本號，為1
BYTE ulen; //使用者名稱長度
BYTE user[255]; //使用者名稱
BYTE plen; //密碼長度
BYTE pass[255]; //密碼
};

//驗證 迴應
struct SOCKS5AUTHANSWER
{
BYTE ver; //版本號
BYTE state; //驗證狀態
};

//Socks5命令 請求
struct SOCKS5COMMANDREQUEST
{
BYTE ver; //版本號
BYTE cmd; //命令
/*
CONNECT：'01'
BIND：'02'
UDP ASSOCIATE：'03'
*/
BYTE rsv; //保留，必須為0
BYTE type; //表明其後的地址型別
/*
· IPV4：X'01'
· 域名：X'03'
· IPV6：X'04'
*/
ULONG addr; //目標地址，這裡只支援ipv4
/*取決於type
'01': 基於IPV4的IP地址，4個位元組長
'03': 基於域名的地址，地址欄位中的第一位元組是以位元組為單位的該域名的長度，沒有結尾的NUL位元組。
'04': 基於IPV6的IP地址，16個位元組長
*/
WORD port; //目標埠(網路位元組順序)
};

//SOCKS5命令 迴應
struct SOCK5COMMANDANSWER
{
BYTE ver; //版本號
BYTE rep; //迴應訊息
/*
· X'00'  成功
· X'01'  普通的SOCKS伺服器請求失敗
· X'02'  現有的規則不允許的連線
· X'03'  網路不可達
· X'04'  主機不可達
· X'05' 連線被拒
· X'06'  TTL超時
· X'07'  不支援的命令
· X'08'  不支援的地址型別
· X'09' - X'FF'  未定義
*/
BYTE rsv; //保留
BYTE type; //同SOCKS5COMMANDREQUEST
ULONG addr; //返回的地址
WORD port; //返回的埠(網路位元組順序)
};

//socks5規定的udp資料包頭
struct SOCKS5UDPHEADER
{
WORD rsv; //保留
BYTE frag; //分段標誌，如果不分段則為0
BYTE type; //同SOCKS5COMMANDREQUEST
ULONG addr; //目標地址，同SOCKS5COMMANDREQUEST
WORD port; //目標埠，同SOCKS5COMMANDREQUEST
};
/////////////////////////////////////////////////////////////////////////////
// CProxySocketSocksV5 command target

class CChildView;
class CProxySocketSocksV5 : public CAsyncSocket
{
SOCKET m_SocketCommand; //與Sock5伺服器互動的socket
BOOL m_bConnected;
ULONG m_AddrBinding;
WORD m_PortBinding;
// Attributes
public:
CChildView* m_pView;
char m_User[255]; //使用者名稱
char m_Pass[255]; //密碼
char m_Server[20]; //socks5 server ip
WORD m_Port;
// Operations
public:
CProxySocketSocksV5();
virtual ~CProxySocketSocksV5();

int SendTo(const void* lpBuf, int nBufLen,
UINT nHostPort, LPCTSTR lpszHostAddress = NULL, int nFlags = 0);
int SendTo(const void* lpBuf, int nBufLen,
const SOCKADDR* lpSockAddr, int nSockAddrLen, int nFlags = 0);
int ReceiveFrom(void* lpBuf, int nBufLen,
CString& rSocketAddress, UINT& rSocketPort, int nFlags = 0);
int ReceiveFrom(void* lpBuf, int nBufLen,
SOCKADDR* lpSockAddr, int* lpSockAddrLen, int nFlags = 0);
// Overrides
public:
void Close();
UINT ConnectToProxyServer(const char* server = NULL,WORD nPort = 0, char* user = NULL, char* pass = NULL);
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CProxySocketSocksV5)
public:
virtual void OnReceive(int nErrorCode);
//}}AFX_VIRTUAL

// Generated message map functions
//{{AFX_MSG(CProxySocketSocksV5)
// NOTE - the ClassWizard will add and remove member functions here.
//}}AFX_MSG

// Implementation
protected:
};

/////////////////////////////////////////////////////////////////////////////

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_PROXYSOCKETSOCKSV5_H__07BD21EA_0867_4F91_A82A_72F3AF706228__INCLUDED_)

/////////////////////////////////////////////////////////////////////////////
// CProxySocketSocksV5

CProxySocketSocksV5::CProxySocketSocksV5()
{
m_Port = 0;
ZeroMemory(m_User,255);
ZeroMemory(m_Pass,255);
ZeroMemory(m_Server,20);
m_bConnected = FALSE;
m_AddrBinding = 0;
m_PortBinding = 0;
}

CProxySocketSocksV5::~CProxySocketSocksV5()
{
}


// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(CProxySocketSocksV5, CAsyncSocket)
//{{AFX_MSG_MAP(CProxySocketSocksV5)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif // 0

/////////////////////////////////////////////////////////////////////////////
// CProxySocketSocksV5 member functions

//連線到socks5代理伺服器，在傳送或接收前先呼叫次函式
//函式返回0表示成功，返回其他數值表示失敗，具體意義見程式碼
UINT CProxySocketSocksV5::ConnectToProxyServer(const char* server,WORD nPort, char* user, char* pass)
{
if(server != NULL)
strcpy(m_Server,server);
if(nPort != 0)
m_Port = nPort;
if( user != NULL)
strcpy (m_User,user);
if(pass != NULL)
strcpy (m_Pass,pass);

char buf[1024];
ASSERT(m_hSocket != INVALID_SOCKET);

if(m_SocketCommand != INVALID_SOCKET)
closesocket(m_SocketCommand);

m_SocketCommand = socket(AF_INET,SOCK_STREAM,0);

SOCKADDR_IN addr;
addr.sin_addr.s_addr = inet_addr(m_Server);
addr.sin_family = AF_INET;
addr.sin_port = htons(m_Port);

if(m_SocketCommand == INVALID_SOCKET)
return 1; //Create Socket Error
if(SOCKET_ERROR == connect(m_SocketCommand,(sockaddr*)&addr,sizeof(addr)))
{
closesocket (m_SocketCommand);
return 2; //connect socks server error;
}

SOCKS5SELECTREQEST selectrequest;
ZeroMemory(&selectrequest,sizeof(selectrequest));
selectrequest.ver = 5;
selectrequest.nmethods = 2;
selectrequest.methods[0] = 0;
selectrequest.methods[1] = 2;

if(SOCKET_ERROR  == send(m_SocketCommand,(char*)&selectrequest,4,0))
{
closesocket (m_SocketCommand);
return 3; //send error
}

ZeroMemory(buf,1024);
if(SOCKET_ERROR == recv(m_SocketCommand,buf,1024,0))
{
closesocket (m_SocketCommand);
return 4; //recv error
}

SOCKS5SELECTANSWER *pselectanswer = (SOCKS5SELECTANSWER *)buf;
if((pselectanswer->method != 0 && pselectanswer->method != 2) || pselectanswer->ver != 5)
{
closesocket (m_SocketCommand);
return 5; //無合適的驗證方法
}

if(pselectanswer->method == 2)
{
SOCKS5AUTHENTICATE auth;
ZeroMemory(&auth,sizeof(auth));
auth.ver = 1;
auth.ulen = strlen(m_User)>254?254:strlen(m_User);

相關推薦