封装http操作类

      近期在对http数据包的研究过程中感觉自己应该封装个http类，虽然ms有wininet，功能也很强大，但我们自己也试着写个吧，练练手。

#include <iostream> #include <string> #include "mysocket.h" #include <windows.h> using namespace std; class http{ public: http(); ~http(); void addheader(char*,char*); void setmethod(char *); void putdata(char*,int len); void send(char*); int getdata(char*,int); void getheader(char*,char *); int getstatus(); void openurl(char *); void close(); private: string header; char* data; int datalen; string method; mysocket s; }; http::http(){ header=""; data=""; method="GET"; data=0; datalen=0; } http::~http(){ s.close(); delete[] data; } void http::addheader(char* key,char* v){ header+=key; header+=":"; header+=v; header+="/r/n"; } void http::setmethod(char *m){ method=strupr(m); } void http::putdata(char* d,int len){ data=new char[len]; datalen=len; for(int i=0;i<len;i++){ data[i]=d[i]; } } void http::send(char* url){ char *o=strstr(url,"http://"); string host; char *t=(o-url==0)?o+7:url; while(*t!='/'){ host+=*(t++); } string name=t; string str=method; str+=" "; str+=name; str+=" HTTP/1.1"; str+="/r/n"; str+="HOST:"; str+=host; str+="/r/n"; str+=header; str+="/r/n"; char temp[1024]; memset(temp,0,1024); s.init(); s.connect(s.gethostbyname((char*)host.data()),80); s.send((char*)str.data(),str.length()); for(int i=0;i<datalen;){ if(i+1024>datalen){ memcpy(temp,data+i,datalen-i); s.send(temp,datalen-i); }else{ memcpy(temp,data+i,1024); s.send(temp,1024); } i+=1024; memset(temp,0,1024); } char ch; str=""; while(1){ s.recv(&ch,1); str+=ch; if(str.length()>4 && str[str.length()-1]=='/n' && str[str.length()-2]=='/r' && str[str.length()-3]=='/n' && str[str.length()-4]=='/r'){ header=str; break; } } } int http::getdata(char* buf,int len){ /* if(l==0){ char buf[1024]; while(1){ data=new char[1024]; memset(buf,0,1024); int l=s.recv(buf,1024); if(l<=0){ break; } cout<<buf<<endl; } }else{ }*/ return s.recv(buf,len); } void http::getheader(char* key,char *v){ string temp="/r/n"; temp+=key; temp+=": "; char *l=strstr(header.data(),temp.data()); if(l==0){ strcpy(v,""); }else{ char *t=l+temp.length(); string val; while(*t!='/r'){ val+=*(t++); } strcpy(v,val.data()); } } int http::getstatus(){ char s[10]={0}; char *str=strstr(header.data()," ")+1; int i=0; while(*str!=' '){ s[i++]=*(str++); } return atoi(s); } void http::openurl(char *url){ addheader("Accept","*/*"); addheader("User-Agent","sx"); addheader("Connection","close"); addheader("Cache-Control","no-cache"); send(url); } void http::close(){ s.close(); delete[] data; header=""; data=""; method="GET"; data=0; datalen=0; }

      下面给出个例子：

main(){
 WSADATA data;  
 WORD v=MAKEWORD(2,2);  
 WSAStartup(v,&data);
 http h;
h.openurl("http://www.baidu.com/");
 char buf[1025];
 string str;
  while(1){
   memset(buf,0,1025);
   int l=h.getdata(buf,1024);
   if(l<=0){
    break;
   }
   cout<<buf;
   //system("pause");
  }
 
}

      其实openurl是addheader和send的简化形式。代码应该不难理解，我本来想在getdata这个方法里完成对数据的一次性读取，后来发现不可行，便放弃了。

      本文有不足之处，还望大家多多指教。