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