|
楊冶軍
本人在用Oicq聊天時(shí)，經(jīng)常收到一些好友發(fā)給我的用文本符號(hào)描繪的圖像，覺(jué)得好羨慕啊，于是一想何
我們一般常用的Internet代理服務(wù)器是用微軟的Proxy Server 2.0 。但我們可以自己動(dòng)手編寫(xiě)一個(gè)簡(jiǎn)
單、小型的Proxy Server 。下面介紹具體的實(shí)現(xiàn)方法。
一． 原理
本程序的結(jié)構(gòu)原理如下：
對(duì)于每一個(gè)用戶(hù)的請(qǐng)求（Internet 請(qǐng)求，由瀏覽器發(fā)出），本程序?qū)?dòng)兩個(gè)線程，一個(gè)把本地用戶(hù)的請(qǐng)求
數(shù)據(jù)發(fā)送到遠(yuǎn)程的Internet主機(jī)，另一個(gè)線程把遠(yuǎn)程主機(jī)的回應(yīng)數(shù)據(jù)發(fā)送到本地請(qǐng)求用戶(hù)。
二． 主要函數(shù)
UserToProxyThread ( void * pParam ) ：它是用來(lái)把本地用戶(hù)請(qǐng)求數(shù)據(jù)發(fā)送到遠(yuǎn)程主機(jī)的，起服務(wù)器線
程角色。當(dāng)接到本地（局域網(wǎng)）用戶(hù)的請(qǐng)求，它就啟動(dòng)另一個(gè)自身線程，以偵聽(tīng)別的用戶(hù)的請(qǐng)求，并讀出已接
收到的請(qǐng)求數(shù)據(jù)，接著啟動(dòng)第二個(gè)線程ProxyToServer()（這個(gè)線程用來(lái)連接遠(yuǎn)程主機(jī)），當(dāng)遠(yuǎn)程主機(jī)連接成
功后，它把已讀出的本地用戶(hù)請(qǐng)求數(shù)據(jù)發(fā)送到遠(yuǎn)程主機(jī)。
ProxyToServer ( void * pParam) ，可以被當(dāng)作是客戶(hù)端服務(wù)，它把遠(yuǎn)程主機(jī)發(fā)送來(lái)的數(shù)據(jù)分發(fā)給本地請(qǐng)
求用戶(hù)。
三． 開(kāi)發(fā)運(yùn)行環(huán)境
本程序是在VC++6.0環(huán)境下開(kāi)發(fā)的，在Win95 和 WinNT4.0下運(yùn)行正常。
四． 詳細(xì)代碼
#include "stdafx.h"
#include "Proxy.h"
#include < winsock2.h > //WINSOCKET API 2。0
#include < stdlib.h >
#include < stdio.h >
#include < string.h >
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
////////////////////////////////////////////////////////////////
#define HTTP "http://"
#define FTP "ftp://"
#define PROXYPORT 5001 //Proxy 端口
#define BUFSIZE 10240 //緩沖區(qū)大小
CWinApp theApp;
using namespace std;
UINT ProxyToServer(LPVOID pParam);
UINT UserToProxyThread(void *pParam);
struct SocketPair{
SOCKET user_proxy; //socket : 本地機(jī)器到PROXY 服務(wù)機(jī)
SOCKET proxy_server; //socket : PROXY 服務(wù)機(jī)到遠(yuǎn)程主機(jī)
BOOL IsUser_ProxyClosed; // 本地機(jī)器到PROXY 服務(wù)機(jī)狀態(tài)
BOOL IsProxy_ServerClosed; // PROXY 服務(wù)機(jī)到遠(yuǎn)程主機(jī)狀態(tài)
};
struct ProxyParam{
char Address[256]; // 遠(yuǎn)程主機(jī)地址
HANDLE User_SvrOK; // PROXY 服務(wù)機(jī)到遠(yuǎn)程主機(jī)的聯(lián)結(jié)狀態(tài)
SocketPair *pPair; // 維護(hù)一組SOCKET的指針
int Port; // 用來(lái)聯(lián)結(jié)遠(yuǎn)程主機(jī)的端口
}; file://這個(gè)結(jié)構(gòu)用來(lái)PROXY SERVER與遠(yuǎn)程主機(jī)的信息交換.
SOCKET gListen_Socket; file://用來(lái)偵聽(tīng)的SOCKET。
int StartServer() //啟動(dòng)服務(wù)
{
WSADATA wsaData;
sockaddr_in local;
SOCKET listen_socket;
if(::WSAStartup(0x202,&wsaData)!=0)
{printf("\nError in Startup session.\n");WSACleanup();return -1;};
local.sin_family=AF_INET;
local.sin_addr.s_addr=INADDR_ANY;
local.sin_port=htons(PROXYPORT);
listen_socket=socket(AF_INET,SOCK_STREAM,0);
if(listen_socket==INVALID_SOCKET)
{printf("\nError in New a Socket.");WSACleanup();return -2;}
if(::bind(listen_socket,(sockaddr *)&local,sizeof(local))!=0)
{printf("\n Error in Binding socket."); WSACleanup();return -3; };
if(::listen(listen_socket,5)!=0)
{printf("\n Error in Listen."); WSACleanup(); return -4;}
gListen_Socket=listen_socket;
AfxBeginThread(UserToProxyThread,NULL); //啟動(dòng)偵聽(tīng)
return 1;
}
int CloseServer() //關(guān)閉服務(wù)
{
closesocket(gListen_Socket);
WSACleanup();
return 1;
}
file://分析接收到的字符，得到遠(yuǎn)程主機(jī)地址
int GetAddressAndPort( char * str, char *address, int * port)
{
char buf[BUFSIZE], command[512], proto[128], *p;
int j;
sscanf(str,"%s%s%s",command,buf,proto);
p=strstr(buf,HTTP);
//HTTP
if(p)
{
p+=strlen(HTTP);
for(int i=0;i< strlen(p);i++)
if( *(p+i)==`/`) break;
*(p+i)=0;
strcpy(address,p);
p=strstr(str,HTTP);
for(int j=0;j< i+strlen(HTTP);j++)
*(p+j)=` `; //去掉遠(yuǎn)程主機(jī)名: GET http://www.njust.edu.cn/
HTTP1.1 == > GET / HTTP1.1
*port=80; //缺省的 http 端口
}
else
{//FTP, 不支持, 下面的代碼可以省略.
p=strstr(buf,FTP);
if(!p) return 0;
p+=strlen(FTP);
for(int i=0;i< strlen(p);i++)
if( *(p+i)==`/`) break; //Get The Remote Host
*(p+i)=0;
for(j=0;j< strlen(p);j++)
if(*(p+j)==`:`)
{*port=atoi(p+j+1); //Get The Port
*(p+j)=0;
}
else *port=21;
strcpy(address,p);
p=strstr(str,FTP);
for(j=0;j< i+strlen(FTP);j++)
*(p+j)=` `;
}
return 1;
}
// 取到本地的數(shù)據(jù)，發(fā)往遠(yuǎn)程主機(jī)
UINT UserToProxyThread(void *pParam)
{
char Buffer[BUFSIZE];
int Len;
sockaddr_in from;
SOCKET msg_socket;
int fromlen,retval;
SocketPair SPair;
ProxyParam ProxyP;
CWinThread *pChildThread;
fromlen=sizeof(from);
msg_socket=accept(gListen_Socket,(struct sockaddr*)&from,&fromlen);
AfxBeginThread(UserToProxyThread,pParam); //啟動(dòng)另一偵聽(tīng).
if( msg_socket==INVALID_SOCKET)
{ printf( "\nError in accept "); return -5;}
//讀客戶(hù)的第一行數(shù)據(jù)
SPair.IsUser_ProxyClosed=FALSE;
SPair.IsProxy_ServerClosed=TRUE;
SPair.user_proxy=msg_socket;
retval=recv(SPair.user_proxy,Buffer,sizeof(Buffer),0);
if(retval==SOCKET_ERROR)
{ printf("\nError Recv");
if(SPair.IsUser_ProxyClosed==FALSE)
{closesocket(SPair.user_proxy);
SPair.IsUser_ProxyClosed=TRUE;
}
}
if(retval==0)
{printf("Client Close connection\n");
if(SPair.IsUser_ProxyClosed==FALSE)
{closesocket(SPair.user_proxy);
SPair.IsUser_ProxyClosed=TRUE;
}
}
Len=retval;
#ifdef _DEBUG
Buffer[Len]=0;
printf("\n Received %d bytes,data[%s]from client\n",retval,Buffer);
#endif
//
SPair.IsUser_ProxyClosed=FALSE;
SPair.IsProxy_ServerClosed=TRUE;
SPair.user_proxy=msg_socket;
ProxyP.pPair=&SPair;
ProxyP.User_SvrOK=CreateEvent(NULL,TRUE,FALSE,NULL);
GetAddressAndPort( Buffer,ProxyP.Address,&ProxyP.Port);
pChildThread=AfxBeginThread(ProxyToServer,(LPVOID)&ProxyP);
::WaitForSingleObject(ProxyP.User_SvrOK,60000); //等待聯(lián)結(jié)
::CloseHandle(ProxyP.User_SvrOK);
while(SPair.IsProxy_ServerClosed ==FALSE && SPair.IsUser_ProxyClosed==FALSE)
{
retval=send(SPair.proxy_server,Buffer,Len,0);
if(retval==SOCKET_ERROR)
{ printf("\n send() failed:error%d\n",WSAGetLastError());
if(SPair.IsProxy_ServerClosed==FALSE)
{
closesocket(SPair.proxy_server);
SPair.IsProxy_ServerClosed=TRUE;
}
continue;
}
retval=recv(SPair.user_proxy,Buffer,sizeof(Buffer),0);
if(retval==SOCKET_ERROR)
{ printf("\nError Recv");
if(SPair.IsUser_ProxyClosed==FALSE)
{closesocket(SPair.user_proxy);
SPair.IsUser_ProxyClosed=TRUE;
}
continue;
}
if(retval==0)
{printf("Client Close connection\n");
if(SPair.IsUser_ProxyClosed==FALSE)
{closesocket(SPair.user_proxy);
SPair.IsUser_ProxyClosed=TRUE;
}
break;
}
Len=retval;
#ifdef _DEBUG
Buffer[Len]=0;
printf("\n Received %d bytes,data[%s]from client\n",retval,Buffer);
#endif
} End While
if(SPair.IsProxy_ServerClosed==FALSE)
{
closesocket(SPair.proxy_server);
SPair.IsProxy_ServerClosed=TRUE;
}
if(SPair.IsUser_ProxyClosed==FALSE)
{closesocket(SPair.user_proxy);
SPair.IsUser_ProxyClosed=TRUE;
}
::WaitForSingleObject(pChildThread- >m_hThread,20000); //Should check the
return value
return 0;
}
// 讀取遠(yuǎn)程主機(jī)數(shù)據(jù)，并發(fā)往本地客戶(hù)機(jī)
UINT ProxyToServer(LPVOID pParam){
ProxyParam * pPar=(ProxyParam*)pParam;
char Buffer[BUFSIZE];
char *server_name= "localhost";
unsigned short port ;
int retval,Len;
unsigned int addr;
int socket_type ;
struct sockaddr_in server;
struct hostent *hp;
SOCKET conn_socket;
socket_type = SOCK_STREAM;
server_name = pPar- >Address;
port = pPar- >Port;
if (isalpha(server_name[0])) { /* server address is a name */
hp = gethostbyname(server_name);
}
else { /* Convert nnn.nnn address to a usable one */
addr = inet_addr(server_name);
hp = gethostbyaddr((char *)&addr,4,AF_INET);
}
if (hp == NULL ) {
fprintf(stderr,"Client: Cannot resolve address [%s]: Error %d\n",
server_name,WSAGetLastError());
::SetEvent(pPar- >User_SvrOK);
return 0;
}
//
// Copy the resolved information into the sockaddr_in structure
//
memset(&server,0,sizeof(server));
memcpy(&(server.sin_addr),hp- >h_addr,hp- >h_length);
server.sin_family = hp- >h_addrtype;
server.sin_port = htons(port);
conn_socket = socket(AF_INET,socket_type,0); /* 打開(kāi)一個(gè) socket */
if (conn_socket < 0 ) {
fprintf(stderr,"Client: Error Opening socket: Error %d\n",
WSAGetLastError());
pPar- >pPair- >IsProxy_ServerClosed=TRUE;
::SetEvent(pPar- >User_SvrOK);
return -1;
}
#ifdef _DEBUG
printf("Client connecting to: %s\n",hp- >h_name);
#endif
if (connect(conn_socket,(struct sockaddr*)&server,sizeof(server))
== SOCKET_ERROR) {
fprintf(stderr,"connect() failed: %d\n",WSAGetLastError());
pPar- >pPair- >IsProxy_ServerClosed=TRUE;
::SetEvent(pPar- >User_SvrOK);
return -1;
}
pPar- >pPair- >proxy_server=conn_socket;
pPar- >pPair- >IsProxy_ServerClosed=FALSE;
::SetEvent(pPar- >User_SvrOK);
// cook up a string to send
while(!pPar- >pPair- >IsProxy_ServerClosed &&!pPar- >pPair->IsUser_ProxyClosed)
{
retval = recv(conn_socket,Buffer,sizeof (Buffer),0 );
if (retval == SOCKET_ERROR ) {
fprintf(stderr,"recv() failed: error %d\n",WSAGetLastError());
closesocket(conn_socket);
pPar- >pPair- >IsProxy_ServerClosed=TRUE;
break;
}
Len=retval;
if (retval == 0) {
printf("Server closed connection\n");
closesocket(conn_socket);
pPar- >pPair- >IsProxy_ServerClosed=TRUE;
break;
}
retval = send(pPar- >pPair- >user_proxy,Buffer,Len,0);
if (retval == SOCKET_ERROR) {
fprintf(stderr,"send() failed: error %d\n",WSAGetLastError());
closesocket(pPar- >pPair- >user_proxy);
pPar- >pPair- >IsUser_ProxyClosed=TRUE;
break;
}
#ifdef _DEBUG
Buffer[Len]=0;
printf("Received %d bytes, data [%s] from server\n",retval,Buffer);
#endif
}
if(pPar- >pPair- >IsProxy_ServerClosed==FALSE)
{
closesocket(pPar- >pPair- >proxy_server);
pPar- >pPair- >IsProxy_ServerClosed=TRUE;
}
if(pPar- >pPair- >IsUser_ProxyClosed==FALSE)
{closesocket(pPar- >pPair- >user_proxy);
pPar- >pPair- >IsUser_ProxyClosed=TRUE;
}
return 1;
}
int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
int nRetCode = 0;
// 初始化SOCKET
if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
{
// 錯(cuò)誤處理
cerr < < _T("Fatal Error: MFC initialization failed") < < endl;
nRetCode = 1;
}
else
{
// 主程序開(kāi)始.
StartServer();
while(1)
if(getchar()==`q`) break;
CloseServer();
}
return nRetCode;
}
|
