二級(jí)考試:用VC++6.0編寫Proxy服務(wù)器

常用的代理服務(wù)器軟件有procy sever 2.0以及WinGate等等。利用VC 6.0強(qiáng)大功能也可以編寫出自己的代理服務(wù)器應(yīng)用軟件。本文介紹了具體的實(shí)現(xiàn)方法。
    一． 原理
    本程序的結(jié)構(gòu)原理如下：
    對(duì)于每一個(gè)用戶的請(qǐng)求（Internet 請(qǐng)求，由瀏覽器發(fā)出），本程序?qū)?dòng)兩個(gè)線程，一個(gè)把本地用戶的請(qǐng)求數(shù)據(jù)發(fā)送到遠(yuǎn)程的Internet主機(jī)，另一個(gè)線程把遠(yuǎn)程主機(jī)的回應(yīng)數(shù)據(jù)發(fā)送到本地請(qǐng)求用戶。
    二． 主要函數(shù)
    UserToProxyThread ( void * pParam )
    ：它是用來(lái)把本地用戶請(qǐng)求數(shù)據(jù)發(fā)送到遠(yuǎn)程主機(jī)的，起服務(wù)器線程角色。當(dāng)接到本地（局域網(wǎng)）用戶的請(qǐng)求，它就啟動(dòng)另一個(gè)自身線程，以偵聽別的用戶的請(qǐng)求，并讀出已接收到的請(qǐng)求數(shù)據(jù)，接著啟動(dòng)第二個(gè)線程ProxyToServer()（這個(gè)線程用來(lái)連接遠(yuǎn)程主機(jī)），當(dāng)遠(yuǎn)程主機(jī)連接成功后，它把已讀出的本地用戶請(qǐng)求數(shù)據(jù)發(fā)送到遠(yuǎn)程主機(jī)。
    ProxyToServer ( void * pParam)
    ，可以被當(dāng)作是客戶端服務(wù)，它把遠(yuǎn)程主機(jī)發(fā)送來(lái)的數(shù)據(jù)分發(fā)給本地請(qǐng)求用戶。
    三． 開發(fā)運(yùn)行環(huán)境
    本程序是在VC++6.0環(huán)境下開發(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ī)的端口
    }; //這個(gè)結(jié)構(gòu)用來(lái)PROXY SERVER與遠(yuǎn)程主機(jī)的信息交換.
    SOCKET gListen_Socket; //用來(lái)偵聽的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)偵聽
    return 1;
    }
    int CloseServer() //關(guān)閉服務(wù)
    {
    closesocket(gListen_Socket);
    WSACleanup();
    return 1;
    }
    //分析接收到的字符，得到遠(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ī)%C。