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
101
102
103
104
105
106
107
108
109
110
111
|
#include "http.h"
#include "log.h"
#include <assert.h>
CURL *http_create_req(const char *uri,
size_t (*write_fn)(char *, size_t, size_t, void *),
GString *download,
size_t (*read_fn)(char *, size_t, size_t, void *),
http_upload *upload,
const struct curl_slist *headers,
bool tls_verify,
CURLcode *errcode,
const char **errmsg)
{
CURLcode ret = CURLE_FAILED_INIT;
const char *err = "failed to create cURL object";
g_autoptr(CURL) c = curl_easy_init();
if (!c)
goto fail;
err = "setting CURLOPT_URL";
if ((ret = curl_easy_setopt(c, CURLOPT_URL, uri)) != CURLE_OK)
goto fail;
err = "setting CURLOPT_WRITEFUNCTION";
if ((ret = curl_easy_setopt(c, CURLOPT_WRITEFUNCTION, write_fn)) != CURLE_OK)
goto fail;
err = "setting CURLOPT_WRITEDATA";
if ((ret = curl_easy_setopt(c, CURLOPT_WRITEDATA, download)) != CURLE_OK)
goto fail;
err = "setting CURLOPT_READFUNCTION";
if ((ret = curl_easy_setopt(c, CURLOPT_READFUNCTION, read_fn)) != CURLE_OK)
goto fail;
err = "setting CURLOPT_READDATA";
if ((ret = curl_easy_setopt(c, CURLOPT_READDATA, upload)) != CURLE_OK)
goto fail;
/* allow redirects */
err = "setting CURLOPT_FOLLOWLOCATION";
if ((ret = curl_easy_setopt(c, CURLOPT_FOLLOWLOCATION, 1L)) != CURLE_OK)
goto fail;
/* max 5 redirects */
err = "setting CURLOPT_MAXREDIRS";
if ((ret = curl_easy_setopt(c, CURLOPT_MAXREDIRS, 5L)) != CURLE_OK)
goto fail;
/* add headers */
err = "setting CURLOPT_HTTPHEADER";
if ((ret = curl_easy_setopt(c, CURLOPT_HTTPHEADER, headers)) != CURLE_OK)
goto fail;
/* cert verify (enabled by default) */
if (!tls_verify) {
err = "setting CURLOPT_SSL_VERIFYPEER";
if ((ret = curl_easy_setopt(c, CURLOPT_SSL_VERIFYPEER, 0L)) != CURLE_OK)
goto fail;
}
// all ok
*errmsg = NULL;
*errcode = CURLE_OK;
CURL *o = c;
c = NULL; // prevent auto cleanup
return o;
fail:
*errmsg = err;
*errcode = ret;
return NULL;
}
void http_add_header(struct curl_slist **hdrs, const char *fmt, ...) {
va_list ap;
va_start(ap, fmt);
char *s = g_strdup_vprintf(fmt, ap);
*hdrs = curl_slist_append(*hdrs, s);
g_free(s);
va_end(ap);
}
size_t http_dummy_write(char *ptr, size_t size, size_t nmemb, void *userdata) {
return size * nmemb;
}
size_t http_dummy_read(char *ptr, size_t size, size_t nmemb, void *userdata) {
return 0;
}
size_t http_upload_read(char *ptr, size_t size, size_t nmemb, void *userdata) {
http_upload *u = userdata;
assert(size == 1); // as per docs
size_t len = MIN(nmemb, u->s.len);
memcpy(ptr, u->s.s, len);
str_shift(&u->s, len);
return len;
}
size_t http_download_write(char *ptr, size_t size, size_t nmemb, void *userdata) {
assert(size == 1); // as per docs
GString *s = userdata;
g_string_append_len(s, ptr, nmemb);
return nmemb;
}
|
