diff options
Diffstat (limited to 'curl_tomem.c')
-rw-r--r-- | curl_tomem.c | 100 |
1 files changed, 100 insertions, 0 deletions
diff --git a/curl_tomem.c b/curl_tomem.c new file mode 100644 index 0000000..e30cfae --- /dev/null +++ b/curl_tomem.c @@ -0,0 +1,100 @@ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <curl/curl.h> + +/* dump curl result into a memory instead of FILE * */ +struct curl_inmem +{ + char *memory; + size_t size; /* size is used as an index */ +}; +typedef struct curl_inmem curl_inmem_t; + +void dump_curlmem(void *data) +{ + curl_inmem_t *mem = (curl_inmem_t *) data; + size_t iter = 0; + while (iter < mem->size) + putchar(mem->memory[iter++]); +} + +size_t write_data(char *ptr, size_t size, size_t nmemb, void *userdata) +{ + size_t realsize = size * nmemb; + curl_inmem_t *mem = (curl_inmem_t *) userdata; + + mem->memory = realloc(mem->memory, mem->size + realsize + 1); + if (!mem->memory) + { + perror("failed to resize curl in memory"); + exit(EXIT_FAILURE); + } + + memcpy(&(mem->memory[mem->size]), ptr, realsize); + mem->size += realsize; + mem->memory[mem->size] = 0; + + return realsize; +} + +int main(int argc, char **argv) +{ + if (argc != 2) + { + fprintf(stderr, "usage: %s <url>\n", argv[0]); + return 1; + } + + CURL *curl_handle; + CURLcode ret; + + curl_global_init(CURL_GLOBAL_ALL); + curl_handle = curl_easy_init(); + if (curl_handle) + { + curl_easy_setopt(curl_handle, CURLOPT_URL, argv[1]); + curl_easy_setopt(curl_handle, CURLOPT_HEADER, 1L); + /* some servers don't like requests that are made without a user-agent + * field, so we provide one */ + curl_easy_setopt(curl_handle, CURLOPT_USERAGENT, "libcurl-agent/1.0"); + char err_buff[CURL_ERROR_SIZE]; + curl_easy_setopt(curl_handle, CURLOPT_ERRORBUFFER, err_buff); + + /* by default passes NULL aka stdout */ + curl_inmem_t chunk = { 0, 0 }; + chunk.memory = (char *) malloc(sizeof(char)); + if (!chunk.memory) + perror("failed to create a chunk"); + else + { + curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *) &chunk); + /* send all data to this function */ + curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, write_data); + } + + if ((ret = curl_easy_perform(curl_handle)) != CURLE_OK) + fprintf(stderr, "%s\n", err_buff); + else + { + /* ask for the content-type */ + char *ct; + ret = curl_easy_getinfo(curl_handle, CURLINFO_CONTENT_TYPE, &ct); + if ((CURLE_OK == ret) && ct) + printf("Content-Type: %s\n", ct); + } + + /* print out */ + dump_curlmem(&chunk); + + /* always cleanup */ + curl_easy_cleanup(curl_handle); + if (chunk.memory) + free(chunk.memory); + /* we're done with libcurl, so clean it up */ + curl_global_cleanup(); + } + + return 0; +} + |