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