285 lines
7.3 KiB
C++
285 lines
7.3 KiB
C++
// ProxyMapDlg.cpp : implementation file
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include "Client.h"
|
||
#include "ProxyMapDlg.h"
|
||
|
||
#ifdef _DEBUG
|
||
#define new DEBUG_NEW
|
||
#undef THIS_FILE
|
||
static char THIS_FILE[] = __FILE__;
|
||
#endif
|
||
|
||
/////////////////////////////////////////////////////////////////////////////
|
||
// CProxyMapDlg dialog
|
||
|
||
CProxyMapDlg *g_pProxyMap;
|
||
|
||
CProxyMapDlg::CProxyMapDlg(CWnd* pParent, CIOCPServer* pIOCPServer, ClientContext *pContext)
|
||
: CDialog(CProxyMapDlg::IDD, pParent)
|
||
{
|
||
//{{AFX_DATA_INIT(CProxyMapDlg)
|
||
// NOTE: the ClassWizard will add member initialization here
|
||
//}}AFX_DATA_INIT
|
||
m_iocpServer = pIOCPServer;
|
||
m_pContext = pContext;
|
||
m_iocpLocal = NULL;
|
||
g_pProxyMap = this;
|
||
}
|
||
|
||
|
||
void CProxyMapDlg::DoDataExchange(CDataExchange* pDX)
|
||
{
|
||
CDialog::DoDataExchange(pDX);
|
||
//{{AFX_DATA_MAP(CProxyMapDlg)
|
||
DDX_Control(pDX, IDC_EDIT, m_edit);
|
||
//}}AFX_DATA_MAP
|
||
}
|
||
#define WM_NOTIFYPROC (WM_USER+109)
|
||
|
||
BEGIN_MESSAGE_MAP(CProxyMapDlg, CDialog)
|
||
//{{AFX_MSG_MAP(CProxyMapDlg)
|
||
ON_WM_CLOSE()
|
||
ON_MESSAGE(WM_NOTIFYPROC, OnNotifyProc)
|
||
ON_WM_SIZE()
|
||
//}}AFX_MSG_MAP
|
||
END_MESSAGE_MAP()
|
||
|
||
/////////////////////////////////////////////////////////////////////////////
|
||
// CProxyMapDlg message handlers
|
||
|
||
|
||
BOOL CProxyMapDlg::OnInitDialog()
|
||
{
|
||
CDialog::OnInitDialog();
|
||
|
||
// TODO: Add extra initialization here
|
||
m_iocpLocal = new CIOCPLOCAL;
|
||
|
||
if (m_iocpLocal == NULL)
|
||
{
|
||
AfxMessageBox("CIOCPLOCAL == NULL");
|
||
return FALSE;
|
||
}
|
||
|
||
m_edit.SetLimitText(MAXDWORD); // 设置最大长度
|
||
|
||
CString str;
|
||
DWORD nPort;
|
||
// 开启IPCP服务器
|
||
if (m_iocpLocal->Initialize(NotifyProc, 100000, 0))
|
||
{
|
||
sockaddr_in sockAddr;
|
||
memset(&sockAddr, 0, sizeof(sockAddr));
|
||
int nSockAddrLen = sizeof(sockAddr);
|
||
getsockname(m_iocpLocal->m_socListen, (SOCKADDR*)&sockAddr, &nSockAddrLen);
|
||
nPort = ntohs(sockAddr.sin_port) ;
|
||
|
||
BOOL bResult = getpeername(m_pContext->m_Socket,(SOCKADDR*)&sockAddr, &nSockAddrLen);
|
||
m_IPAddress = bResult != INVALID_SOCKET ? inet_ntoa(sockAddr.sin_addr) : "";
|
||
|
||
|
||
str.Format(_T("\\\\%s 代理服务器 端口: %d \r\n"),m_IPAddress, nPort);
|
||
SetWindowText(str);
|
||
/*
|
||
WriteRegEx(HKEY_CURRENT_USER, _T("Software\\Permeo Technologies\\SocksCap32\\Connections"),
|
||
_T("SocksPort"), REG_DWORD, (TCHAR *)&nPort, nPort, 1);
|
||
WriteRegEx(HKEY_CURRENT_USER, _T("Software\\Permeo Technologies\\SocksCap32\\Connections"),
|
||
_T("SocksServer"),REG_SZ,_T("127.0.0.1"),0,1);*/
|
||
|
||
str.Format(_T("已设置sockscap32，使用其他socks代理软件请设置服务器为:127.0.0.1, 端口为:%d \r\n"),nPort);
|
||
AddLog(str.GetBuffer(0));
|
||
}
|
||
else
|
||
{
|
||
str.Format(_T("\\\\代理服务器 端口绑定失败 \r\n"));
|
||
SetWindowText(str);
|
||
}
|
||
return TRUE; // return TRUE unless you set the focus to a control
|
||
// EXCEPTION: OCX Property Pages should return FALSE
|
||
}
|
||
|
||
void CProxyMapDlg::OnClose()
|
||
{
|
||
// TODO: Add your message handler code here and/or call default
|
||
closesocket(m_pContext->m_Socket);
|
||
|
||
m_iocpLocal->Shutdown();
|
||
delete m_iocpLocal;
|
||
|
||
CDialog::OnClose();
|
||
}
|
||
|
||
LRESULT CProxyMapDlg::OnNotifyProc(WPARAM wParam, LPARAM lParam)
|
||
{
|
||
ClientContext * pContext=(ClientContext *)lParam;
|
||
DWORD index = pContext->dwID;
|
||
TCHAR szMsg[200]={0};
|
||
try
|
||
{
|
||
switch (wParam)
|
||
{
|
||
case NC_CLIENT_CONNECT:
|
||
pContexts[index]=pContext;
|
||
wsprintf(szMsg,_T("%d new connection\r\n"),index);
|
||
break;
|
||
case NC_CLIENT_DISCONNECT:
|
||
if(pContext->m_bProxyConnected)
|
||
{
|
||
BYTE lpData[5];
|
||
lpData[0]=COMMAND_PROXY_CLOSE;
|
||
memcpy(lpData+1,&index,sizeof(DWORD));
|
||
m_iocpServer->Send(m_pContext,lpData,5);
|
||
}
|
||
wsprintf(szMsg,_T("%d disconnect\r\n"),index);
|
||
break;
|
||
case NC_TRANSMIT:
|
||
break;
|
||
case NC_RECEIVE:
|
||
if(pContext->m_bProxyConnected==2)
|
||
{
|
||
m_iocpServer->Send(m_pContext,pContext->m_CompressionBuffer.GetBuffer(),
|
||
pContext->m_CompressionBuffer.GetBufferLen());
|
||
wsprintf(szMsg,_T("%d <==发 %d bytes\r\n"),index,pContext->m_CompressionBuffer.GetBufferLen()-5);
|
||
}
|
||
else if(pContext->m_bProxyConnected==0)
|
||
{
|
||
BYTE buf[2];
|
||
LPBYTE lpData=pContext->m_CompressionBuffer.GetBuffer(5);
|
||
buf[0]=5;
|
||
buf[1]=0;
|
||
pContext->m_bProxyConnected=1;
|
||
m_iocpLocal->Send(pContext,buf,2);
|
||
wsprintf(szMsg,_T("%d 返回标示 %d %d %d\r\n"),index,lpData[0],lpData[1],lpData[2]);
|
||
}
|
||
else if(pContext->m_bProxyConnected==1)
|
||
{
|
||
LPBYTE lpData=pContext->m_CompressionBuffer.GetBuffer(5);
|
||
BYTE buf[11];
|
||
if(lpData[0]==5 && lpData[1]==1 && lpData[3]==1 && pContext->m_CompressionBuffer.GetBufferLen()>10)
|
||
{
|
||
buf[0]=COMMAND_PROXY_CONNECT;
|
||
memcpy(buf+1,&index,4);
|
||
memcpy(buf+5,lpData+4,6);
|
||
m_iocpServer->Send(m_pContext,buf,sizeof(buf));
|
||
in_addr inaddr;
|
||
inaddr.s_addr=*(DWORD *)(buf+5);
|
||
|
||
wsprintf(szMsg,_T("%d connecting %s:%d...\r\n"),index,inet_ntoa(inaddr),ntohs(*(USHORT *)(buf+9)));
|
||
|
||
}
|
||
else
|
||
{
|
||
buf[0]=5;
|
||
buf[1]=7;
|
||
buf[2]=0;
|
||
buf[3]=lpData[3];
|
||
m_iocpLocal->Send(pContext,buf,sizeof(buf));
|
||
closesocket(pContext->m_Socket );
|
||
wsprintf(szMsg,_T("%d 不符要求,断开 %d %d %d\r\n"),index,lpData[0],lpData[1],lpData[3]);
|
||
}
|
||
}
|
||
break;
|
||
}
|
||
}catch(...){}
|
||
if (szMsg[0])
|
||
AddLog(szMsg);
|
||
return 0;
|
||
}
|
||
|
||
|
||
void CProxyMapDlg::OnReceiveComplete()
|
||
{
|
||
if(m_iocpLocal == NULL)
|
||
return;
|
||
|
||
LPBYTE buf=m_pContext->m_DeCompressionBuffer.GetBuffer(0);
|
||
DWORD index=*(DWORD*)&buf[1];
|
||
TCHAR szMsg[200];
|
||
switch (buf[0])
|
||
{
|
||
case TOKEN_PROXY_CONNECT_RESULT:
|
||
BYTE sendbuf[10];
|
||
sendbuf[0]=5;
|
||
sendbuf[1]=(buf[9] || buf[10])? 0 : 5;
|
||
sendbuf[2]=0;
|
||
sendbuf[3]=1;
|
||
memcpy(&sendbuf[4],&buf[5],6);
|
||
if (sendbuf[1]==0)
|
||
{
|
||
pContexts[index]->m_bProxyConnected =2;
|
||
wsprintf(szMsg,_T("%d 连接成功\r\n"),index);
|
||
}else
|
||
wsprintf(szMsg,_T("%d 连接失败\r\n"),index);
|
||
m_iocpLocal->Send(pContexts[index],sendbuf,sizeof(sendbuf));
|
||
AddLog(szMsg);
|
||
break;
|
||
case TOKEN_PROXY_BIND_RESULT:
|
||
break;
|
||
case TOKEN_PROXY_CLOSE:
|
||
wsprintf(szMsg,_T("%d TOKEN_PROXY_CLOSE\r\n"),index);
|
||
AddLog(szMsg);
|
||
if(pContexts[index]->m_Socket && pContexts[index]->m_Socket!=INVALID_SOCKET)
|
||
{
|
||
LINGER lingerStruct;
|
||
lingerStruct.l_onoff = 1;
|
||
lingerStruct.l_linger = 1;
|
||
setsockopt(pContexts[index]->m_Socket, SOL_SOCKET, SO_LINGER,
|
||
(char *)&lingerStruct, sizeof(lingerStruct) );
|
||
shutdown(pContexts[index]->m_Socket, SD_SEND);
|
||
closesocket(pContexts[index]->m_Socket);
|
||
pContexts[index]->m_Socket=0;
|
||
}
|
||
break;
|
||
case TOKEN_PROXY_DATA:
|
||
m_iocpLocal->Send(pContexts[index],&buf[5],m_pContext->m_DeCompressionBuffer.GetBufferLen()-5);
|
||
wsprintf(szMsg,_T("%d ==>收 %d bytes\r\n"),index,m_pContext->m_DeCompressionBuffer.GetBufferLen()-5);
|
||
AddLog(szMsg);
|
||
break;
|
||
default:
|
||
// 传输发生异常数据
|
||
break;
|
||
}
|
||
}
|
||
|
||
void CALLBACK CProxyMapDlg::NotifyProc(ClientContext *pContext, UINT nCode)
|
||
{
|
||
#ifdef _DEBUG
|
||
g_pProxyMap->SendMessage(WM_NOTIFYPROC,nCode,(LPARAM)pContext);
|
||
#else
|
||
g_pProxyMap->OnNotifyProc(nCode,(LPARAM)pContext);
|
||
#endif
|
||
}
|
||
|
||
void CProxyMapDlg::AddLog(TCHAR * lpText)
|
||
{
|
||
m_edit.SetSel(-1, -1);
|
||
m_edit.ReplaceSel(lpText);
|
||
}
|
||
|
||
void CProxyMapDlg::OnSize(UINT nType, int cx, int cy)
|
||
{
|
||
CDialog::OnSize(nType, cx, cy);
|
||
|
||
// TODO: Add your message handler code here
|
||
if (!IsWindowVisible())
|
||
return;
|
||
|
||
RECT rectClient;
|
||
RECT rectEdit;
|
||
GetClientRect(&rectClient);
|
||
rectEdit.left = 0;
|
||
rectEdit.top = 0;
|
||
rectEdit.right = rectClient.right;
|
||
rectEdit.bottom = rectClient.bottom;
|
||
m_edit.MoveWindow(&rectEdit);
|
||
}
|
||
|
||
void CProxyMapDlg::PostNcDestroy()
|
||
{
|
||
// TODO: Add your specialized code here and/or call the base class
|
||
delete this;
|
||
CDialog::PostNcDestroy();
|
||
}
|