diff --git a/main.cpp b/main.cpp index 952cd9c..d1f063e 100644 --- a/main.cpp +++ b/main.cpp @@ -34,6 +34,15 @@ Input input_data(istream& stream, bool prompt) { stream >> in.bin_count; return in; } + +size_t +write_data(void* items, size_t item_size, size_t item_count, void* ctx) { + size_t data_size = item_size * item_count; + stringstream* buffer = reinterpret_cast(ctx); + (*buffer).write(reinterpret_cast(items), data_size); + return data_size; +} + Input download(const string& address) { stringstream buffer; @@ -41,6 +50,8 @@ download(const string& address) { if(curl) { CURLcode res; curl_easy_setopt(curl, CURLOPT_URL, address.c_str()); + curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); + curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer); res = curl_easy_perform(curl); if (res!=CURLE_OK){ cerr << curl_easy_strerror(res);