revert my pass through the tree to remove checks of the result of ast_strdupa
[asterisk/asterisk.git] / apps / app_curl.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C)  2004 - 2005, 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 STANDARD_LOCAL_USER;
54
55 LOCAL_USER_DECL;
56
57 struct MemoryStruct {
58         char *memory;
59         size_t size;
60 };
61
62 static void *myrealloc(void *ptr, size_t size)
63 {
64         /* There might be a realloc() out there that doesn't like reallocing
65            NULL pointers, so we take care of it here */
66         if (ptr)
67                 return ast_realloc(ptr, size);
68         else
69                 return ast_malloc(size);
70 }
71
72 static size_t WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
73 {
74         register int realsize = size * nmemb;
75         struct MemoryStruct *mem = (struct MemoryStruct *)data;
76
77         mem->memory = (char *)myrealloc(mem->memory, mem->size + realsize + 1);
78         if (mem->memory) {
79                 memcpy(&(mem->memory[mem->size]), ptr, realsize);
80                 mem->size += realsize;
81                 mem->memory[mem->size] = 0;
82         }
83         return realsize;
84 }
85
86 static int curl_internal(struct MemoryStruct *chunk, char *url, char *post)
87 {
88         CURL *curl;
89
90         curl_global_init(CURL_GLOBAL_ALL);
91         curl = curl_easy_init();
92
93         if (!curl) {
94                 return -1;
95         }
96
97         curl_easy_setopt(curl, CURLOPT_URL, url);
98         curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
99         curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)chunk);
100         curl_easy_setopt(curl, CURLOPT_USERAGENT, "asterisk-libcurl-agent/1.0");
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         curl_easy_cleanup(curl);
109         return 0;
110 }
111
112 static char *acf_curl_exec(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len)
113 {
114         struct localuser *u;
115         char *info;
116         struct MemoryStruct chunk = { NULL, 0 };
117         AST_DECLARE_APP_ARGS(args,
118                 AST_APP_ARG(url);
119                 AST_APP_ARG(postdata);
120         );
121
122         *buf = '\0';
123         
124         if (ast_strlen_zero(data)) {
125                 ast_log(LOG_WARNING, "CURL requires an argument (URL)\n");
126                 return buf;
127         }
128
129         LOCAL_USER_ACF_ADD(u);
130
131         info = ast_strdupa(data);
132         if (!info) {
133                 ast_log(LOG_ERROR, "Out of memory\n");
134                 LOCAL_USER_REMOVE(u);
135                 return buf;
136         }
137
138         AST_STANDARD_APP_ARGS(args, info);      
139         
140         if (! curl_internal(&chunk, args.url, args.postdata)) {
141                 if (chunk.memory) {
142                         chunk.memory[chunk.size] = '\0';
143                         if (chunk.memory[chunk.size - 1] == 10)
144                                 chunk.memory[chunk.size - 1] = '\0';
145
146                         ast_copy_string(buf, chunk.memory, len);
147                         free(chunk.memory);
148                 }
149         } else {
150                 ast_log(LOG_ERROR, "Cannot allocate curl structure\n");
151         }
152
153         LOCAL_USER_REMOVE(u);
154         return buf;
155 }
156
157 struct ast_custom_function acf_curl = {
158         .name = "CURL",
159         .synopsis = "Retrieves the contents of a URL",
160         .syntax = "CURL(url[|post-data])",
161         .desc =
162         "  url       - URL to retrieve\n"
163         "  post-data - Optional data to send as a POST (GET is default action)\n",
164         .read = acf_curl_exec,
165 };
166
167 int unload_module(void)
168 {
169         int res;
170
171         res = ast_custom_function_unregister(&acf_curl);
172
173         STANDARD_HANGUP_LOCALUSERS;
174         
175         return res;
176 }
177
178 int load_module(void)
179 {
180         int res;
181
182         res = ast_custom_function_register(&acf_curl);
183
184         return res;
185 }
186
187 char *description(void)
188 {
189         return tdesc;
190 }
191
192 int usecount(void)
193 {
194         int res;
195         STANDARD_USECOUNT(res);
196         return res;
197 }
198
199 char *key()
200 {
201         return ASTERISK_GPL_KEY;
202 }