Форум программистов [Powered by Invision Power Board]

Код

    class a_socket //объявление класса
    {
        public:
            a_socket(string needle_method, string needle_url, bool use_proxy, string proxy);
            ~a_socket();
            string get_content();
            int get_socket_data();
        private:
            string host, port, page, method, content, proxy_ip, proxy_port;
            bool is_set_proxy;
#ifdef WINDOWS
            SOCKET sock;  //Об"являем идентификатор сокета
#else
            int sock;     //Об"являем идентификатор сокета
#endif
    };

    a_socket::a_socket(string needle_method, string needle_url, bool use_proxy, string proxy) //конструктор
    {
        method=needle_method;  //needle_method

        string temp_string=needle_url;  //needle_url
        if(temp_string.substr(0,7).compare("http://")==0) temp_string=temp_string.substr(7);
        if(temp_string.find('/')==-1)
        {
            host=temp_string;
            page="/";
        }
        else
        {
            host=temp_string.substr(0,temp_string.find('/'));
            page=temp_string.substr(temp_string.find('/'));
        }
        if(host.find(':')==-1) port="80";
        else 
        {
            port=host.substr(host.find(':')+1);
            if(port.compare("")==0) port="80";
            host=host.substr(0,host.find(':'));
        }
        
        is_set_proxy=use_proxy; //proxy
        if(is_set_proxy)
        {
            if(proxy.find(':')==-1)
            {
                proxy_ip=proxy;
                proxy_port="80";
            }
            else
            {
                proxy_ip=proxy.substr(0,proxy.find(':'));
                proxy_port=proxy.substr(proxy.find(':')+1);
                if(proxy_port.compare("")==0) port="80";
            }
            if(inet_addr(&proxy_ip[0])==-1) is_set_proxy=FALSE;
        }
        
        content=""; //content
    }

    a_socket::~a_socket()
    {
        if(sock)
        {
            shutdown(sock, 0x02);
#ifdef WINDOWS
            closesocket(sock);
#else
            close(sock);
#endif
        }
    }

    string a_socket::get_content()
    {
        return content;
    }

    int a_socket::get_socket_data()
    {
        sockaddr_in server_addr;    // структура для адреса
        string send_text="";
        if(is_set_proxy)
        {
            if( ( server_addr.sin_addr.s_addr =inet_addr(proxy_ip.c_str()))==(-1))  // проверим, возможно host содержит просто ip
            {
                // нет, адрес не есть ip, тогда возможно это dns адрес
                hostent* server_host;
                server_host=gethostbyname(proxy_ip.c_str());
                if(server_host==NULL)
                {
                    return 1;  //ошибка, не возможно определить айпи-адрес сайта
                }
                memcpy(&server_addr.sin_addr.s_addr, server_host->h_addr_list[0], server_host->h_length);  //присваиваем айпишник
            }
            server_addr.sin_family = AF_INET;  //указываем что будем использовать айпишники
            server_addr.sin_port =  htons(atoi(proxy_port.c_str()));   //переводим порт в правильный вид
            if((sock = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1)   //создаем сокет
            {
                return 2;   //ошибка, не возможно приконектиться
            }
            if( connect( sock, (sockaddr *)&server_addr, sizeof(server_addr)) != 0)  //соединяемся
            {
                return 3;  //ошибка, не возможно соединиться с хостом
            }  
            send_text=method+" http://"+host+page+" HTTP/1.0\r\n"+
                     +"Host: "+host+"\r\n"+
                     +"User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.0.1) Gecko/20060111 Firefox/1.5.0.1\r\n"+
                     +"Accept: text/xml,application/xml,application/xhtml+xml,text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5\r\n"+
                     +"Accept-Language: en-us,en;q=0.8\r\n"+
                     +"Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.7\r\n"+
                     +"Referer: http://"+host+"\r\n"+
                     +"Connection: close\r\n\r\n";
        }
        else
        {
            if( ( server_addr.sin_addr.s_addr =inet_addr(host.c_str()))==(-1))  // проверим, возможно host содержит просто ip
            {
                // нет, адрес не есть ip, тогда возможно это dns адрес
                hostent* server_host;
                server_host=gethostbyname(host.c_str());
                if(server_host==NULL)
                {
                    return 1;  //ошибка, не возможно определить айпи-адрес сайта
                }
                memcpy(&server_addr.sin_addr.s_addr, server_host->h_addr_list[0], server_host->h_length);  //присваиваем айпишник
            }
            server_addr.sin_family = AF_INET;  //указываем что будм использовать айпишники
            server_addr.sin_port =  htons(atoi(port.c_str()));   //переводим порт в правильный вид
            if((sock = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1)   //создаем сокет
            {
                return 2;   //ошибка, не возможно приконектиться
            }
            if( connect( sock, (sockaddr *)&server_addr, sizeof(server_addr)) != 0)  //соединяемся
            {
                return 3;  //ошибка, не возможно соединиться с хостом
            }
            send_text=method+" "+page+" HTTP/1.0\r\n"+
                     +"Host: "+host+"\r\n"+
                     +"User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.0.1) Gecko/20060111 Firefox/1.5.0.1\r\n"+
                     +"Accept: text/xml,application/xml,application/xhtml+xml,text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5\r\n"+
                     +"Accept-Language: en-us,en;q=0.8\r\n"+
                     +"Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.7\r\n"+
                     +"Referer: http://"+host+"\r\n"+
                     +"Connection: close\r\n\r\n";
        }
        send(sock, send_text.c_str(), send_text.length(), 0);  //посылаем запрос
        char receive_buf[DEF_BUF_SIZE+5];   //буфер для приема
        int receive_buf_len = recv(sock, receive_buf, DEF_BUF_SIZE, 0);
        while(receive_buf_len>0)
        {
            receive_buf[receive_buf_len]='\0';
            content.append(receive_buf);
            receive_buf_len = recv( sock, receive_buf, DEF_BUF_SIZE, 0);
        }
        shutdown(sock, 0x02);
#ifdef WINDOWS
        closesocket(sock);
#else
        close(sock);
#endif
        return 0;
    }