00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035 #include "asterisk.h"
00036
00037 ASTERISK_FILE_VERSION(__FILE__, "$Revision: 146799 $")
00038
00039 #include <stdio.h>
00040 #include <stdlib.h>
00041 #include <string.h>
00042 #include <curl/curl.h>
00043
00044 #include "asterisk/lock.h"
00045 #include "asterisk/file.h"
00046 #include "asterisk/logger.h"
00047 #include "asterisk/channel.h"
00048 #include "asterisk/pbx.h"
00049 #include "asterisk/cli.h"
00050 #include "asterisk/options.h"
00051 #include "asterisk/module.h"
00052 #include "asterisk/app.h"
00053 #include "asterisk/utils.h"
00054 #include "asterisk/threadstorage.h"
00055
00056 struct MemoryStruct {
00057 char *memory;
00058 size_t size;
00059 };
00060
00061 static void *myrealloc(void *ptr, size_t size)
00062 {
00063
00064
00065 if (ptr)
00066 return ast_realloc(ptr, size);
00067 else
00068 return ast_malloc(size);
00069 }
00070
00071 static size_t WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
00072 {
00073 register int realsize = size * nmemb;
00074 struct MemoryStruct *mem = (struct MemoryStruct *)data;
00075
00076 mem->memory = (char *)myrealloc(mem->memory, mem->size + realsize + 1);
00077 if (mem->memory) {
00078 memcpy(&(mem->memory[mem->size]), ptr, realsize);
00079 mem->size += realsize;
00080 mem->memory[mem->size] = 0;
00081 }
00082 return realsize;
00083 }
00084
00085 static const char *global_useragent = "asterisk-libcurl-agent/1.0";
00086
00087 static void curl_instance_cleanup(void *data)
00088 {
00089 CURL **curl = data;
00090
00091 curl_easy_cleanup(*curl);
00092
00093 free(data);
00094 }
00095
00096 AST_THREADSTORAGE_CUSTOM(curl_instance, curl_instance_init, curl_instance_cleanup);
00097
00098 static int curl_internal(struct MemoryStruct *chunk, char *url, char *post)
00099 {
00100 int ret;
00101 CURL **curl;
00102
00103 if (!(curl = ast_threadstorage_get(&curl_instance, sizeof(*curl))))
00104 return -1;
00105
00106 if (!*curl) {
00107 if (!(*curl = curl_easy_init()))
00108 return -1;
00109 curl_easy_setopt(*curl, CURLOPT_NOSIGNAL, 1);
00110 curl_easy_setopt(*curl, CURLOPT_TIMEOUT, 180);
00111 curl_easy_setopt(*curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
00112 curl_easy_setopt(*curl, CURLOPT_USERAGENT, global_useragent);
00113 }
00114
00115 curl_easy_setopt(*curl, CURLOPT_URL, url);
00116 curl_easy_setopt(*curl, CURLOPT_WRITEDATA, (void *) chunk);
00117
00118 if (post) {
00119 curl_easy_setopt(*curl, CURLOPT_POST, 1);
00120 curl_easy_setopt(*curl, CURLOPT_POSTFIELDS, post);
00121 }
00122
00123 ret = curl_easy_perform(*curl);
00124
00125 if (post)
00126 curl_easy_setopt(*curl, CURLOPT_POST, 0);
00127
00128 return ret ? -1 : 0;
00129 }
00130
00131 static int acf_curl_exec(struct ast_channel *chan, char *cmd, char *info, char *buf, size_t len)
00132 {
00133 int ret = -1;
00134 struct ast_module_user *u;
00135 struct MemoryStruct chunk = { NULL, 0 };
00136 AST_DECLARE_APP_ARGS(args,
00137 AST_APP_ARG(url);
00138 AST_APP_ARG(postdata);
00139 );
00140
00141 *buf = '\0';
00142
00143 if (ast_strlen_zero(info)) {
00144 ast_log(LOG_WARNING, "CURL requires an argument (URL)\n");
00145 return -1;
00146 }
00147
00148 u = ast_module_user_add(chan);
00149
00150 AST_STANDARD_APP_ARGS(args, info);
00151
00152 if (chan)
00153 ast_autoservice_start(chan);
00154
00155 if (!curl_internal(&chunk, args.url, args.postdata)) {
00156 if (chunk.memory) {
00157 chunk.memory[chunk.size] = '\0';
00158 if (chunk.memory[chunk.size - 1] == 10)
00159 chunk.memory[chunk.size - 1] = '\0';
00160
00161 ast_copy_string(buf, chunk.memory, len);
00162 free(chunk.memory);
00163 }
00164 ret = 0;
00165 } else {
00166 ast_log(LOG_ERROR, "Cannot allocate curl structure\n");
00167 }
00168
00169 if (chan)
00170 ast_autoservice_stop(chan);
00171
00172 ast_module_user_remove(u);
00173
00174 return ret;
00175 }
00176
00177 struct ast_custom_function acf_curl = {
00178 .name = "CURL",
00179 .synopsis = "Retrieves the contents of a URL",
00180 .syntax = "CURL(url[|post-data])",
00181 .desc =
00182 " url - URL to retrieve\n"
00183 " post-data - Optional data to send as a POST (GET is default action)\n",
00184 .read = acf_curl_exec,
00185 };
00186
00187 static int unload_module(void)
00188 {
00189 int res;
00190
00191 res = ast_custom_function_unregister(&acf_curl);
00192
00193 ast_module_user_hangup_all();
00194
00195 curl_global_cleanup();
00196
00197 return res;
00198 }
00199
00200 static int load_module(void)
00201 {
00202 int res;
00203
00204 if (curl_global_init(CURL_GLOBAL_ALL)) {
00205 ast_log(LOG_ERROR, "Unable to initialize the CURL library. Cannot load func_curl\n");
00206 return AST_MODULE_LOAD_DECLINE;
00207 }
00208
00209 res = ast_custom_function_register(&acf_curl);
00210
00211 return res;
00212 }
00213
00214 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Load external URL");
00215