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
|
/*
* telak - A program that display pictures in root window
* (c) 2005 - Julien Danjou <julien@danjou.info>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2, as
* published by the Free Software Foundation.
*
*/
/* C stuff */
#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* cURL stuff */
#include <curl/curl.h>
#include <curl/types.h>
#include <curl/easy.h>
#include "telak.h"
#include "fetch.h"
#include "image.h"
int write_image(void *buffer, size_t size, size_t nmemb, void * data)
{
struct remote_img * img;
img = (struct remote_img *) data;
/* open file for writing */
if(!img->stream)
img->stream = fopen((char *) img->file, "wb");
if(!img->stream)
{
perror("Error creating file");
return -1; /* failure, can't open file to write */
}
return fwrite(buffer, size, nmemb, img->stream);
}
int
fetch(char *url, char *dest)
{
CURL *curl_handle;
struct remote_img img;
img.file = dest;
img.stream = NULL;
curl_global_init(CURL_GLOBAL_ALL);
curl_handle = curl_easy_init();
if(curl_handle)
{
/* specify URL to get */
curl_easy_setopt(curl_handle, CURLOPT_URL, url);
/* send all data to this function */
curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, write_image);
/* we pass our 'chunk' struct to the callback function */
curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *) &img);
/* 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, TELAK_USER_AGENT);
/* get it! */
printf("Downloading %s...\n", url);
curl_easy_perform(curl_handle);
/* cleanup curl stuff */
curl_easy_cleanup(curl_handle);
}
if(img.stream)
fclose(img.stream);
curl_global_cleanup();
return 0;
}
|
