Separate the global initialization routines for cURL into its own separate
[asterisk/asterisk.git] / funcs / func_curl.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C)  2004 - 2006, Tilghman Lesher
5  *
6  * Tilghman Lesher <curl-20050919@the-tilghman.com>
7  * and Brian Wilkins <bwilkins@cfl.rr.com> (Added POST option)
8  *
9  * app_curl.c is distributed with no restrictions on usage or
10  * redistribution.
11  *
12  * See http://www.asterisk.org for more information about
13  * the Asterisk project. Please do not directly contact
14  * any of the maintainers of this project for assistance;
15  * the project provides a web site, mailing lists and IRC
16  * channels for your use.
17  *
18  */
19
20 /*! \file
21  * 
22  * \brief Curl - Load a URL
23  *
24  * \author Tilghman Lesher <curl-20050919@the-tilghman.com>
25  *
26  * \note Brian Wilkins <bwilkins@cfl.rr.com> (Added POST option) 
27  *
28  * \extref Depends on the CURL library  - http://curl.haxx.se/
29  * 
30  * \ingroup functions
31  */
32  
33 /*** MODULEINFO
34         <depend>curl</depend>
35  ***/
36
37 #include "asterisk.h"
38
39 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
40
41 #include <curl/curl.h>
42
43 #include "asterisk/lock.h"
44 #include "asterisk/file.h"
45 #include "asterisk/channel.h"
46 #include "asterisk/pbx.h"
47 #include "asterisk/cli.h"
48 #include "asterisk/module.h"
49 #include "asterisk/app.h"
50 #include "asterisk/utils.h"
51 #include "asterisk/threadstorage.h"
52
53 static size_t WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
54 {
55         register int realsize = size * nmemb;
56         struct ast_str **str = (struct ast_str **)data;
57
58         if (ast_str_make_space(str, (*str)->used + realsize + 1) == 0) {
59                 memcpy(&(*str)->str[(*str)->used], ptr, realsize);
60                 (*str)->used += realsize;
61         }
62
63         return realsize;
64 }
65
66 static const char *global_useragent = "asterisk-libcurl-agent/1.0";
67
68 static int curl_instance_init(void *data)
69 {
70         CURL **curl = data;
71
72         if (!(*curl = curl_easy_init()))
73                 return -1;
74
75         curl_easy_setopt(*curl, CURLOPT_NOSIGNAL, 1);
76         curl_easy_setopt(*curl, CURLOPT_TIMEOUT, 180);
77         curl_easy_setopt(*curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
78         curl_easy_setopt(*curl, CURLOPT_USERAGENT, global_useragent);
79
80         return 0;
81 }
82
83 static void curl_instance_cleanup(void *data)
84 {
85         CURL **curl = data;
86
87         curl_easy_cleanup(*curl);
88 }
89
90 AST_THREADSTORAGE_CUSTOM(curl_instance, curl_instance_init, curl_instance_cleanup);
91
92 static int curl_internal(struct ast_str **chunk, char *url, char *post)
93 {
94         CURL **curl;
95
96         if (!(curl = ast_threadstorage_get(&curl_instance, sizeof(*curl))))
97                 return -1;
98
99         curl_easy_setopt(*curl, CURLOPT_URL, url);
100         curl_easy_setopt(*curl, CURLOPT_WRITEDATA, (void *) chunk);
101
102         if (post) {
103                 curl_easy_setopt(*curl, CURLOPT_POST, 1);
104                 curl_easy_setopt(*curl, CURLOPT_POSTFIELDS, post);
105         }
106
107         curl_easy_perform(*curl);
108
109         if (post)
110                 curl_easy_setopt(*curl, CURLOPT_POST, 0);
111
112         return 0;
113 }
114
115 static int acf_curl_exec(struct ast_channel *chan, const char *cmd, char *info, char *buf, size_t len)
116 {
117         struct ast_str *str = ast_str_create(16);
118         AST_DECLARE_APP_ARGS(args,
119                 AST_APP_ARG(url);
120                 AST_APP_ARG(postdata);
121         );
122
123         *buf = '\0';
124         
125         if (ast_strlen_zero(info)) {
126                 ast_log(LOG_WARNING, "CURL requires an argument (URL)\n");
127                 ast_free(str);
128                 return -1;
129         }
130
131         AST_STANDARD_APP_ARGS(args, info);      
132
133         if (chan)
134                 ast_autoservice_start(chan);
135
136         if (!curl_internal(&str, args.url, args.postdata)) {
137                 if (str->used) {
138                         str->str[str->used] = '\0';
139                         if (str->str[str->used - 1] == '\n') {
140                                 str->str[str->used - 1] = '\0';
141                         }
142
143                         ast_copy_string(buf, str->str, len);
144                 }
145         } else {
146                 ast_log(LOG_ERROR, "Cannot allocate curl structure\n");
147         }
148         ast_free(str);
149
150         if (chan)
151                 ast_autoservice_stop(chan);
152         
153         return 0;
154 }
155
156 struct ast_custom_function acf_curl = {
157         .name = "CURL",
158         .synopsis = "Retrieves the contents of a URL",
159         .syntax = "CURL(url[,post-data])",
160         .desc =
161         "  url       - URL to retrieve\n"
162         "  post-data - Optional data to send as a POST (GET is default action)\n",
163         .read = acf_curl_exec,
164 };
165
166 static int unload_module(void)
167 {
168         int res;
169
170         res = ast_custom_function_unregister(&acf_curl);
171
172         return res;
173 }
174
175 static int load_module(void)
176 {
177         int res;
178
179         if (!ast_module_check("res_curl.so")) {
180                 if (ast_load_resource("res_curl.so") != AST_MODULE_LOAD_SUCCESS) {
181                         ast_log(LOG_ERROR, "Cannot load res_curl, so func_curl cannot be loaded\n");
182                         return AST_MODULE_LOAD_DECLINE;
183                 }
184         }
185
186         res = ast_custom_function_register(&acf_curl);
187
188         return res;
189 }
190
191 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Load external URL");
192