Automatically create new buddy upon reception of a presence stanza of
[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 struct MemoryStruct {
54         char *memory;
55         size_t size;
56 };
57
58 /* There might be a realloc() out there that doesn't like reallocing
59  * NULL pointers, so we take care of it here
60  */
61 static void *myrealloc(void *ptr, size_t size)
62 {
63         return (ptr ? ast_realloc(ptr, size) : ast_malloc(size));
64 }
65
66 static size_t WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
67 {
68         register int realsize = size * nmemb;
69         struct MemoryStruct *mem = (struct MemoryStruct *)data;
70
71         if ((mem->memory = (char *)myrealloc(mem->memory, mem->size + realsize + 1))) {
72                 memcpy(&(mem->memory[mem->size]), ptr, realsize);
73                 mem->size += realsize;
74                 mem->memory[mem->size] = 0;
75         }
76
77         return realsize;
78 }
79
80 static const char *global_useragent = "asterisk-libcurl-agent/1.0";
81
82 static int curl_instance_init(void *data)
83 {
84         CURL **curl = data;
85
86         if (!(*curl = curl_easy_init()))
87                 return -1;
88
89         curl_easy_setopt(*curl, CURLOPT_NOSIGNAL, 1);
90         curl_easy_setopt(*curl, CURLOPT_TIMEOUT, 180);
91         curl_easy_setopt(*curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
92         curl_easy_setopt(*curl, CURLOPT_USERAGENT, global_useragent);
93
94         return 0;
95 }
96
97 static void curl_instance_cleanup(void *data)
98 {
99         CURL **curl = data;
100
101         curl_easy_cleanup(*curl);
102 }
103
104 AST_THREADSTORAGE_CUSTOM(curl_instance, curl_instance_init, curl_instance_cleanup);
105
106 static int curl_internal(struct MemoryStruct *chunk, char *url, char *post)
107 {
108         CURL **curl;
109
110         if (!(curl = ast_threadstorage_get(&curl_instance, sizeof(*curl))))
111                 return -1;
112
113         curl_easy_setopt(*curl, CURLOPT_URL, url);
114         curl_easy_setopt(*curl, CURLOPT_WRITEDATA, (void *) chunk);
115
116         if (post) {
117                 curl_easy_setopt(*curl, CURLOPT_POST, 1);
118                 curl_easy_setopt(*curl, CURLOPT_POSTFIELDS, post);
119         }
120
121         curl_easy_perform(*curl);
122
123         if (post)
124                 curl_easy_setopt(*curl, CURLOPT_POST, 0);
125
126         return 0;
127 }
128
129 static int acf_curl_exec(struct ast_channel *chan, const char *cmd, char *info, char *buf, size_t len)
130 {
131         struct MemoryStruct chunk = { NULL, 0 };
132         AST_DECLARE_APP_ARGS(args,
133                 AST_APP_ARG(url);
134                 AST_APP_ARG(postdata);
135         );
136
137         *buf = '\0';
138         
139         if (ast_strlen_zero(info)) {
140                 ast_log(LOG_WARNING, "CURL requires an argument (URL)\n");
141                 return -1;
142         }
143
144         AST_STANDARD_APP_ARGS(args, info);      
145
146         if (chan)
147                 ast_autoservice_start(chan);
148
149         if (!curl_internal(&chunk, args.url, args.postdata)) {
150                 if (chunk.memory) {
151                         chunk.memory[chunk.size] = '\0';
152                         if (chunk.memory[chunk.size - 1] == 10)
153                                 chunk.memory[chunk.size - 1] = '\0';
154
155                         ast_copy_string(buf, chunk.memory, len);
156                         ast_free(chunk.memory);
157                 }
158         } else {
159                 ast_log(LOG_ERROR, "Cannot allocate curl structure\n");
160         }
161
162         if (chan)
163                 ast_autoservice_stop(chan);
164         
165         return 0;
166 }
167
168 struct ast_custom_function acf_curl = {
169         .name = "CURL",
170         .synopsis = "Retrieves the contents of a URL",
171         .syntax = "CURL(url[,post-data])",
172         .desc =
173         "  url       - URL to retrieve\n"
174         "  post-data - Optional data to send as a POST (GET is default action)\n",
175         .read = acf_curl_exec,
176 };
177
178 static int unload_module(void)
179 {
180         int res;
181
182         res = ast_custom_function_unregister(&acf_curl);
183
184         curl_global_cleanup();
185         
186         return res;
187 }
188
189 static int load_module(void)
190 {
191         int res;
192
193         if (curl_global_init(CURL_GLOBAL_ALL)) {
194                 ast_log(LOG_ERROR, "Unable to initialize the CURL library. Cannot load func_curl\n");
195                 return AST_MODULE_LOAD_DECLINE;
196         }       
197
198         res = ast_custom_function_register(&acf_curl);
199
200         return res;
201 }
202
203 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Load external URL");
204