1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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;
}
|
