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 <net.h>
#include <err.h>
#include <assert.h>
#include <string.h>
#include <curl/curl.h>
#include <stdlib.h>
size_t write_cb(char *ptr, size_t size, size_t nmemb, void *userdata)
{
assert(ptr != NULL);
assert(userdata != NULL);
struct net_write_data *chunk = (struct net_write_data*)userdata;
size_t nsize = chunk->size + nmemb;
chunk->buffer = realloc(chunk->buffer, nsize);
if (nsize > chunk->size)
memcpy(&chunk->buffer[chunk->size], ptr, nmemb);
chunk->size = nsize;
return nmemb;
}
size_t write_file_cb(char *ptr, size_t size, size_t nmemb, void *userdata)
{
assert(ptr != NULL);
assert(ptr != NULL);
struct net_file_write_data * user =
(struct net_file_write_data*)userdata;
if (user->file == NULL) {
fprintf(stderr, "net: Write failed, "
"file stream could not be found!\n");
return 0;
}
return fwrite(ptr, size, nmemb, user->file);
}
int net_init()
{
CURLcode result = curl_global_init(CURL_GLOBAL_ALL);
if (result != CURLE_OK)
return -NETINITERR;
return SUCCESS;
}
void net_shutdown()
{
curl_global_cleanup();
}
int net_send_request(char *url, int write_opts, void* userdata)
{
assert(url != NULL);
assert(userdata != NULL);
CURL *curl;
curl = curl_easy_init();
if (!curl)
return -CONINITERR;
curl_easy_setopt(curl, CURLOPT_URL, url);
if (write_opts == WRITE_OPT_MEMORY) {
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_cb);
} else if (write_opts == WRITE_OPT_FILE) {
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_file_cb);
} else {
fprintf(stderr, "net: a valid write_opts must be specified! "
"current: %d",
write_opts);
return -CONINITERR;
}
curl_easy_setopt(curl, CURLOPT_WRITEDATA, userdata);
curl_easy_setopt(curl, CURLOPT_CA_CACHE_TIMEOUT, 604800L);
CURLcode result;
result = curl_easy_perform(curl);
if (result != CURLE_OK) {
fprintf(stderr, "curl_easy_perform() failed: %s\n",
curl_easy_strerror(result));
return -(CONREQERR + result);
}
curl_easy_cleanup(curl);
return SUCCESS;
}
int net_download(const char *url, const char *dst)
{
struct net_file_write_data fwdata;
fwdata.file = fopen(dst, "w"); // change to tmp dir
if (!fwdata.file) {
return -PKGNOTFND;
}
net_send_request(url, WRITE_OPT_FILE, (void*)&fwdata);
fclose(fwdata.file);
return SUCCESS;
}
|
