diff --git a/main.cpp b/main.cpp index 826c35b..d3c10b2 100644 --- a/main.cpp +++ b/main.cpp @@ -42,6 +42,13 @@ write_data(void* items, size_t item_size, size_t item_count, void* ctx){ return data_size; } +size_t header_callback(void *items, size_t item_size, size_t item_count, void *userdata){ + size_t data_size = item_size * item_count; + stringstream* buffer = reinterpret_cast(userdata); + buffer->write(reinterpret_cast(items), data_size); + return data_size; +} + Input download(const string& address){ stringstream buffer; @@ -51,6 +58,8 @@ download(const string& address){ curl_easy_setopt(curl, CURLOPT_URL, address.c_str()); curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer); + curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, header_callback); + curl_easy_setopt(curl, CURLOPT_HEADERDATA, &buffer); res = curl_easy_perform(curl); if (res != CURLE_OK) { curl_easy_strerror(res);