HTTP自定义Header-(SOCKET-TCP)

来源:互联网 发布:如何找淘宝达人推广 编辑:程序博客网 时间:2024/06/03 15:32

  HTTP自定义Header-TCP

    前几天弄一些东西,需要在发送http请求的时候自定义http头,找了几个库用着很不爽。有的把Cookie直接干掉了,还自己在头里加了版权,最后终于忍不了了。在网上一把梭了一个TCP的,这个比较灵活,没有那么多事,THHP下面终归还是TCP:

一把梭的代码如下:

#include "HttpConnect.h"#ifdef WIN32#pragma comment(lib,"ws2_32.lib")#endifHttpConnect::HttpConnect(){#ifdef WIN32    //此处一定要初始化一下,否则gethostbyname返回一直为空    WSADATA wsa = { 0 };    WSAStartup(MAKEWORD(2, 2), &wsa);#endif}HttpConnect::~HttpConnect(){}void HttpConnect::socketHttp(std::string host, std::string request){    int sockfd;    struct sockaddr_in address;    struct hostent *server;    sockfd = socket(AF_INET,SOCK_STREAM,0);    address.sin_family = AF_INET;    address.sin_port = htons(80);    server = gethostbyname(host.c_str());    memcpy((char *)&address.sin_addr.s_addr,(char*)server->h_addr, server->h_length);    if(-1 == connect(sockfd,(struct sockaddr *)&address,sizeof(address))){        DBG <<"connection error!"<<std::endl;        return;    }    DBG << request << std::endl;#ifdef WIN32    send(sockfd, request.c_str(),request.size(),0);#else    write(sockfd,request.c_str(),request.size());#endif    char buf[1024*1024] = {0};    int offset = 0;    int rc;#ifdef WIN32    while(rc = recv(sockfd, buf+offset, 1024,0))#else    while(rc = read(sockfd, buf+offset, 1024))#endif    {        offset += rc;    }#ifdef WIN32    closesocket(sockfd);#else    close(sockfd);#endif    buf[offset] = 0;    DBG << buf << std::endl;}void HttpConnect::postData(std::string host, std::string path, std::string post_content){    //POST请求方式    std::stringstream stream;    stream << "POST " << path;    stream << " HTTP/1.0\r\n";    stream << "Host: "<< host << "\r\n";    stream << "User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9.2.3) Gecko/20100401 Firefox/3.6.3\r\n";    stream << "Content-Type:application/x-www-form-urlencoded\r\n";    stream << "Content-Length:" << post_content.length()<<"\r\n";    stream << "Connection:close\r\n\r\n";    stream << post_content.c_str();    socketHttp(host, stream.str());}void HttpConnect::getData(std::string host, std::string path, std::string get_content){    //GET请求方式    std::stringstream stream;    stream << "GET " << path << "?" << get_content;    stream << " HTTP/1.0\r\n";    stream << "Host: " << host << "\r\n";    stream <<"User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9.2.3) Gecko/20100401 Firefox/3.6.3\r\n";    stream <<"Connection:close\r\n\r\n";    socketHttp(host, stream.str());} 调用方法:    HttpConnect *http = new HttpConnect();    http->getData("127.0.0.1", "/login", "id=liukang&pw=123");    http->postData("127.0.0.1", "/login","id=liukang&pw=123");
    当时用的是GET 自动以Header,就顺便改了下,代码如下。用的时候可以根据实际情况,改改上面代码就行了。TCP的话在C++里控制Header就比较方便了。
HHHHHH.h#pragma once#define WIN32_LEAN_AND_MEAN#include <windows.h>#include <winsock2.h>#include <ws2tcpip.h>#include <stdio.h>#include <string>#include <iostream>#include <sstream>#pragma comment(lib,"ws2_32.lib")#pragma warning(disable:4996)class HttpConnect{public:HttpConnect();~HttpConnect();std::string socketHttp(std::string host, std::string request);std::string getData(std::string strHost, std::string strPath, std::string strValue,std::string strCookie, std::string strReferer);};CPPPPPP.cpp#include "stdafx.h"#include "HttpConnect.h"HttpConnect::HttpConnect(){#ifdef WIN32WSADATA wsa = { 0 };WSAStartup(MAKEWORD(2, 2), &wsa);#endif}HttpConnect::~HttpConnect(){}char buf[1024 * 1024] = { 0 };//放外面,数组太大,放函数里,栈可能不够大。std::string HttpConnect::socketHttp(std::string host, std::string request){int sockfd;struct sockaddr_in address;struct hostent *server;sockfd = socket(AF_INET, SOCK_STREAM, 0);address.sin_family = AF_INET;address.sin_port = htons(80);server = gethostbyname(host.c_str());memcpy((char *)&address.sin_addr.s_addr, (char*)server->h_addr, server->h_length);if (-1 == connect(sockfd, (struct sockaddr *)&address, sizeof(address))) {//DBG << "connection error!" << std::endl;return "";}//MessageBoxA(NULL, request.c_str(), host.c_str(), MB_OK);#ifdef WIN32send(sockfd, request.c_str(), request.size(), 0);#elsewrite(sockfd, request.c_str(), request.size());#endif//char buf[1024 * 1024] = { 0 };int offset = 0;int rc;#ifdef WIN32while (rc = recv(sockfd, buf + offset, 1024, 0))#elsewhile (rc = read(sockfd, buf + offset, 1024))#endif{offset += rc;}#ifdef WIN32closesocket(sockfd);#elseclose(sockfd);#endifbuf[offset] = 0;//MessageBoxA(NULL , buf ,"Hi" ,MB_OK);return std::string(buf);}std::string HttpConnect::getData(std::string strHost, std::string strPath, std::string strValue,                              std::string strCookie , std::string strReferer) {//GET请求方式std::string stream = "";stream += "GET " + strPath + "?" + strValue;stream += " HTTP/1.0\r\n";stream += "Host: " + strHost + "\r\n";stream += "Cookie: "+ strCookie +"\r\n";stream += "Referer: "+ strReferer +"\r\n";stream += "Connection:close\r\n\r\n";//MessageBoxA(NULL , stream.c_str() ,"Hi" ,MB_OK);return socketHttp(strHost, stream);}



0 0
原创粉丝点击