c++模拟post和get请求

客官°小女子只卖身不卖艺 2022-06-03 06:30 570阅读 0赞

#include <winsock2.h>
    #include <string>
    #include <string.h>
    #include <iostream>
    #include <sstream>
    using namespace std;
    class UrlLib
    {
    public:
        UrlLib(string domainName,int port);
        string getHtml();
        string UtfToGbk(const char* utf8);
        void POST(string source,string data);
        void GET(string source);
        string getCookie(string header);
        void Init();
    private:
        string DomainName;
        string SourcePos;
        string Cookie;
        int Port;
        SOCKET Socket;
        void setSource(string source);
    };
    
    UrlLib::UrlLib(string domainName,int port)
    {
        this->Cookie = "";
        WSADATA WSAData;
        WSAStartup(0x202,&WSAData);
        this->DomainName = domainName;
        this->Port = port;
        this->Init();
    }
    
    void UrlLib::Init()
    {
        this->Socket = socket(AF_INET,SOCK_STREAM,0);
        hostent *host = gethostbyname(this->DomainName.c_str());
        sockaddr_in saddr;
        saddr.sin_family = AF_INET;
        saddr.sin_port = htons(this->Port);
        memcpy(&saddr.sin_addr,host->h_addr,4);
        if(connect(this->Socket,(sockaddr *)&saddr,sizeof(saddr))==-1)
        {
            cout<<"connect error"<<endl;
        }
        else
        {
            cout<<"connect succeed"<<endl;
        }
        int timeout = 3000; //3s                                               //设置发送数据和接收数据的最大延迟时间
        setsockopt(this->Socket,SOL_SOCKET,SO_SNDTIMEO,(char *)&timeout,sizeof(timeout));
        setsockopt(this->Socket,SOL_SOCKET,SO_RCVTIMEO,(char *)&timeout,sizeof(timeout));
    }
    
    
    void UrlLib::setSource(string source)
    {
        this->SourcePos = source;
    }
    
    string UrlLib::getHtml()
    {
        string html;
        char buf[1024];
        int n;
        int sum = 0;
        cout<<"正在获取html"<<endl;
        while((n=recv(this->Socket,buf,sizeof(buf)-sizeof(char),0))!=0)
        {
            cout<<n<<"字节"<<endl;
            for(int i = 0;i<n;i++)
            {
                html += buf[i];
            }
            sum ++;
            if(sum==10)
                break;
        }
    
        html = UtfToGbk(html.c_str());
        closesocket(this->Socket);
        return html;
    }
    
    string UrlLib::UtfToGbk(const char* utf8)   //由于windowscmd窗口支持的编码格式为gbk而网站默认的编码格式为utf8 所以要转换编码
    {
       int len = MultiByteToWideChar(CP_UTF8, 0, utf8, -1, NULL, 0);
       wchar_t* wstr = new wchar_t[len+1];
       memset(wstr, 0, len+1);
       MultiByteToWideChar(CP_UTF8, 0, utf8, -1, wstr, len);
       len = WideCharToMultiByte(CP_ACP, 0, wstr, -1, NULL, 0, NULL, NULL);
       char* str = new char[len+1];
       memset(str, 0, len+1);
       WideCharToMultiByte(CP_ACP, 0, wstr, -1, str, len, NULL, NULL);
       if(wstr) delete[] wstr;
       return str;
    }
    
    void UrlLib::POST(string source,string data)
    {
        this->Init();
        this->setSource(source);
        stringstream request;
        request <<"POST " <<this->SourcePos << " HTTP/1.1\r\n";
        request <<"Host:" <<this->DomainName <<"\r\n";
        request <<"Content-Length:"<<data.length()<<"\r\n"   ;//+ "\r\n";
        request <<"Connection:close\r\n";
        if(this->Cookie!="")
            request <<this->Cookie + "\r\n";
        request <<"Content-Type:application/x-www-form-urlencoded\r\n";
        //request <<"Referer:http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Login/login"<<"\r\n";
        request << "\r\n";
    
        request << data ;
        string Post = request.str();
        cout<<Post<<endl;
        send(this->Socket,Post.c_str(),Post.size(),0);
    }
    
    void UrlLib::GET(string source)
    {
        this->Init();
        //cout<<this->Cookie<<endl;
        this->setSource(source);
        string request = "GET "+this->SourcePos+" HTTP/1.1\r\n";
        request += "Host:" + this->DomainName + "\r\n";
        request += "connection:close\r\n";
        request += this->Cookie + "\r\n";
    
        request += "\r\n";
        cout<<request<<endl;
        if(send(this->Socket,request.c_str(),request.size(),0)>0)
        {
            cout<<"send succeed"<<endl;
        }
        else
        {
            cout<<"send error"<<endl;
    
        }
    }
    
    string UrlLib::getCookie(string header)
    {
        int begin = header.find("Cookie:");
        int end = header.find("path",begin);
        int len = end-begin;
        string cookie = header.substr(begin,len);
        this->Cookie = cookie;
        return this->Cookie;
    }
    
    int main()
    {
        string html;
        UrlLib urllib("acm.sdut.edu.cn",80);
        urllib.POST("/onlinejudge2/index.php/Home/Login/login","user_name=3272007939&password=3272007939");
        html = urllib.getHtml();
    
        urllib.getCookie(html);
        int i = 1000;
        while(true)
        {
            stringstream ss;
            ss<<i;
            string num;
            ss>>num;
            urllib.POST("/onlinejudge2/index.php/Home/Solution/submitsolution/pid/1000.html","pid="+num+"&lang=gcc&code=hello");
            string html2 = urllib.getHtml();
            //cout<<"-----"<<endl;
            cout<<html2;
            //Sleep(200);
            i++;
        }
    
    }