二級考試:用VC++6.0編寫Proxy服務器

常用的代理服務器軟件有procy sever 2.0以及WinGate等等。利用VC 6.0強大功能也可以編寫出自己的代理服務器應用軟件。本文介紹了具體的實現(xiàn)方法。
    一． 原理
    本程序的結構原理如下：
    對于每一個用戶的請求（Internet 請求，由瀏覽器發(fā)出），本程序將啟動兩個線程，一個把本地用戶的請求數(shù)據(jù)發(fā)送到遠程的Internet主機，另一個線程把遠程主機的回應數(shù)據(jù)發(fā)送到本地請求用戶。
    二． 主要函數(shù)
    UserToProxyThread ( void * pParam )
    ：它是用來把本地用戶請求數(shù)據(jù)發(fā)送到遠程主機的，起服務器線程角色。當接到本地（局域網(wǎng)）用戶的請求，它就啟動另一個自身線程，以偵聽別的用戶的請求，并讀出已接收到的請求數(shù)據(jù)，接著啟動第二個線程ProxyToServer()（這個線程用來連接遠程主機），當遠程主機連接成功后，它把已讀出的本地用戶請求數(shù)據(jù)發(fā)送到遠程主機。
    ProxyToServer ( void * pParam)
    ，可以被當作是客戶端服務，它把遠程主機發(fā)送來的數(shù)據(jù)分發(fā)給本地請求用戶。
    三． 開發(fā)運行環(huán)境
    本程序是在VC++6.0環(huán)境下開發(fā)的，在Win95 和 WinNT4.0下運行正常。
    四． 詳細代碼
    #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 : 本地機器到PROXY 服務機
    SOCKET proxy_server; //socket : PROXY 服務機到遠程主機
    BOOL IsUser_ProxyClosed; // 本地機器到PROXY 服務機狀態(tài)
    BOOL IsProxy_ServerClosed; // PROXY 服務機到遠程主機狀態(tài)
    };
    struct ProxyParam{
    char Address[256]; // 遠程主機地址
    HANDLE User_SvrOK; // PROXY 服務機到遠程主機的聯(lián)結狀態(tài)
    SocketPair *pPair; // 維護一組SOCKET的指針
    int Port; // 用來聯(lián)結遠程主機的端口
    }; //這個結構用來PROXY SERVER與遠程主機的信息交換.
    SOCKET gListen_Socket; //用來偵聽的SOCKET。
    int StartServer() //啟動服務
    {
    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); //啟動偵聽
    return 1;
    }
    int CloseServer() //關閉服務
    {
    closesocket(gListen_Socket);
    WSACleanup();
    return 1;
    }
    //分析接收到的字符，得到遠程主機地址
    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)=` `; //去掉遠程主機%C。