#include #ifdef _WIN32 int main(){ using namespace std; cout << "Rerun the code until you do not see this message anymore. Why? Because Sololearn use both linux and windows os to run c++ codes but this one is made for unix based systems so some system libraries will not work." << endl; return 0; } #else #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include using namespace std; // to only see how to use, go to line 148 namespace HTTP { class RequestAnswer; class RequestBody{ public: RequestBody(){} RequestBody(const char * url, const char * header, const char * content = nullptr); RequestBody& content(const char * text); RequestAnswer get() const; RequestAnswer put() const; RequestAnswer send(const char * method) const; operator string() const; private: string content_; string url_; string header_; }; ostream& operator<<(ostream& os, const RequestBody& body){os << string(body); return os;} class RequestHeader { public: // header chain RequestHeader& setUrl(const char * url); RequestHeader& accept(const char * contentType); RequestHeader& headerValue(const char * key, const char * value); RequestBody body(const char * content = nullptr) const; // other methods const string& getUrl() const {return url_;} operator string() const; // end of chain RequestAnswer get() const; RequestAnswer put() const; RequestAnswer send(const char * method) const; private: string url_; map attributes_; }; ostream& operator<<(ostream& os, const RequestHeader& header){os << string(header); return os;} class RequestAnswer{ public: RequestAnswer(const string& answer); const RequestBody& body() const{return body_;} const RequestHeader& head() const{return head_;} private: RequestBody body_; RequestHeader head_; }; class Request{ public: Request(const char * url); // url chain Request& addToQuery(const char * key, const char * value); // begin of header chain RequestHeader init() const; RequestBody body(const char * content = nullptr) const; // end of chain RequestAnswer get() const; RequestAnswer put() const; RequestAnswer send(const char * method) const; private: string url_; map attributes_; }; Request http(const char * url){return Request(url);} class Exception : public exception { public: Exception(const char * msg) : msg_(msg){} const char * what() const noexcept {return msg_.c_str();} private: string msg_; }; class DNSException : public Exception{ public: DNSException(const char * msg) : Exception(msg){} }; class ConnectionException : public Exception{ public: ConnectionException(const char * msg) : Exception(msg){} }; }; template void test(Func f){ using namespace HTTP; try{ cout << f() << endl << endl; }catch(const DNSException& e){ cout << "HTTP DNS error: " << e.what() << endl; } catch(const ConnectionException& e){ cout << "HTTP connection error: " << e.what() << endl; } catch(const Exception& e){ cout << "HTTP error: " << e.what() << endl; } } int main(){ using namespace HTTP; test([](){ return http("www.google.com") .get() .head(); }); test([](){ return http("www.google.com") .addToQuery("search", "toto") .get() .head(); }); test([](){ return http("www.google.com?search=toto") .get() .head(); }); test([](){ return http("www.google.com") .get() .body(); }); return 0; } // ----- utils static regex r_http("^https?:"); pair splitUrl(string url){ pair pair; if(regex_search(url, r_http)){ size_t id = url.find_first_of(':'); url = url.substr(id + 1, url.size() - id - 1); } size_t i = url.find_first_of('/'); size_t j = url.find_first_of('?'); if(i != string::npos){ pair.first = url.substr(0, i); pair.second = url.substr(i, url.size() - i); }else if(j == string::npos){ pair.first = url; pair.second = "/"; }else{ pair.first = url.substr(0, j); pair.second = "/" + url.substr(j, url.size() - j); } return pair; } // ----- RequestHeader HTTP::RequestHeader& HTTP::RequestHeader::setUrl(const char * url){ url_ = url; return *this; } HTTP::RequestHeader& HTTP::RequestHeader::accept(const char * contentType){ headerValue("Accept", contentType); return *this; } HTTP::RequestHeader& HTTP::RequestHeader::headerValue(const char * key, const char * value){ if(key && *key && value) attributes_[key] = value; return *this; } HTTP::RequestBody HTTP::RequestHeader::body(const char * content) const{ return RequestBody(url_.c_str(), string(*this).c_str(), content); } HTTP::RequestHeader::operator string() const{ tuple splittedUrl = splitUrl(url_); stringstream ss; ss << "GET " << std::get<1>(splittedUrl) << " HTTP/1.0\r\n" << "Host: " << std::get<0>(splittedUrl) << "\r\n"; for(pair p : attributes_) ss << p.first << ": " << p.second << "\r\n"; ss << "\r\n"; return ss.str(); } HTTP::RequestAnswer HTTP::RequestHeader::get() const{ return body().get(); } HTTP::RequestAnswer HTTP::RequestHeader::put() const{ return body().put(); } HTTP::RequestAnswer HTTP::RequestHeader::send(const char * method) const{ return body().send(method); } // ---- RequestBody HTTP::RequestBody::RequestBody(const char * url, const char * header, const char * content) :content_() ,url_(url) ,header_(header) { if(content) content_ = content; } HTTP::RequestBody& HTTP::RequestBody::content(const char * text){ if(text) content_ = text; return *this; } HTTP::RequestAnswer HTTP::RequestBody::get() const{ return send("GET"); } HTTP::RequestAnswer HTTP::RequestBody::put() const{ return send("PUT"); } HTTP::RequestAnswer HTTP::RequestBody::send(const char * method) const{ int sock; struct sockaddr_in client; int PORT = 80; tuple splittedUrl = splitUrl(url_); struct hostent * host = gethostbyname(std::get<0>(splittedUrl).c_str()); if((host == NULL) || (host->h_addr == NULL)) throw HTTP::DNSException(("Error retrieving DNS information. " + url_).c_str()); bzero(&client, sizeof(client)); client.sin_family = AF_INET; client.sin_port = htons( PORT ); memcpy(&client.sin_addr, host->h_addr, host->h_length); sock = socket(AF_INET, SOCK_STREAM, 0); if(sock < 0) throw ConnectionException("Error creating socket."); if(connect(sock, (struct sockaddr *)&client, sizeof(client)) < 0 ) { close(sock); throw ConnectionException("Could not connect"); } string request(*this); request = header_ + "\r\n\r\n" + request; if(::send(sock, request.c_str(), request.length(), 0) != (int)request.length()) throw ConnectionException("Error sending request."); char cur; ostringstream oss; while ( read(sock, &cur, 1) > 0 ) { oss << cur; } return RequestAnswer(oss.str()); } HTTP::RequestBody::operator string() const{ return content_; } // ----- Request HTTP::Request::Request(const char * url) :url_(url) ,attributes_() { size_t i = url_.find("?"); if(i != string::npos){ string query = url_.substr(i + 1, url_.size() - i - 1); url_ = url_.substr(0, i); i = 0; size_t j = 0; size_t oldI = 0; do{ oldI = i; i = query.find("=", j + 1); if(i == string::npos) break; j = query.find("&", i + 1); if(j == string::npos) j = query.size(); addToQuery( query.substr(oldI, i - oldI).c_str(), query.substr(i + 1, j - i).c_str() ); }while(1); } } HTTP::Request& HTTP::Request::addToQuery(const char * key, const char * value){ if(key && value && *key) attributes_.insert(make_pair(key, value)); return *this; } HTTP::RequestHeader HTTP::Request::init() const{ string url = url_; if(attributes_.size()){ url += "?"; for_each(attributes_.cbegin(), attributes_.cend(), [&](const pair& p){ url += p.first + "=" + p.second + "&"; }); url = url.substr(0, url.size() - 1); } return HTTP::RequestHeader().setUrl(url.c_str()); } HTTP::RequestBody HTTP::Request::body(const char * content) const{ return init().body(content); } HTTP::RequestAnswer HTTP::Request::get() const{ return send("GET"); } HTTP::RequestAnswer HTTP::Request::put() const{ return send("PUT"); } HTTP::RequestAnswer HTTP::Request::send(const char * method) const{ return init().send(method); } // ---- RequestAnswer HTTP::RequestAnswer::RequestAnswer(const string& content){ size_t l = content.find("\r\n\r\n"); string head; if(l == string::npos){ head = content; }else{ head = content.substr(0, l); } const char * s = "\r\n"; size_t i = head.find(s); if(i != string::npos){ i = head.find(s, i + 2); while(i != string::npos){ size_t j = head.find(": ", i + 2); if(j != string::npos && j > i + 2){ size_t k = head.find(s, i + 2); size_t res = k; if(k == string::npos) k = head.size(); head_.headerValue( head.substr(i + 2, j - i - 2).c_str(), head.substr(j + 2, k - j - 2).c_str()); i = res; }else{ i = head.find(s, i + 2); } } } if(l != string::npos){ body_ = RequestBody(head_.getUrl().c_str(), head.c_str(), content.substr(l + 4, content.size() - l - 5).c_str()); } } #endif

http test