Add Masquerade manager event which trips when a masquerade happens (issue #7840 repor...
[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  * \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
55 struct MemoryStruct {
56         char *memory;
57         size_t size;
58 };
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 int curl_internal(struct MemoryStruct *chunk, char *url, char *post)
86 {
87         CURL *curl;
88
89         curl = curl_easy_init();
90
91         if (!curl) {
92                 return -1;
93         }
94
95         curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1);
96         curl_easy_setopt(curl, CURLOPT_TIMEOUT, 180);
97         curl_easy_setopt(curl, CURLOPT_FRESH_CONNECT, 1);
98         curl_easy_setopt(curl, CURLOPT_FORBID_REUSE, 1);
99         curl_easy_setopt(curl, CURLOPT_URL, url);
100         curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
101         curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)chunk);
102         curl_easy_setopt(curl, CURLOPT_USERAGENT, "asterisk-libcurl-agent/1.0");
103
104         if (post) {
105                 curl_easy_setopt(curl, CURLOPT_POST, 1);
106                 curl_easy_setopt(curl, CURLOPT_POSTFIELDS, post);
107         }
108
109         curl_easy_perform(curl);
110         curl_easy_cleanup(curl);
111         return 0;
112 }
113
114 static int acf_curl_exec(struct ast_channel *chan, char *cmd, char *info, char *buf, size_t len)
115 {
116         struct ast_module_user *u;
117         struct MemoryStruct chunk = { NULL, 0 };
118         AST_DECLARE_APP_ARGS(args,
119                 AST_APP_ARG(url);
120                 AST_APP_ARG(postdata);
121         );
122
123         *buf = '\0';
124         
125         if (ast_strlen_zero(info)) {
126                 ast_log(LOG_WARNING, "CURL requires an argument (URL)\n");
127                 return -1;
128         }
129
130         u = ast_module_user_add(chan);
131
132         AST_STANDARD_APP_ARGS(args, info);      
133
134         if (!curl_internal(&chunk, args.url, args.postdata)) {
135                 if (chunk.memory) {
136                         chunk.memory[chunk.size] = '\0';
137                         if (chunk.memory[chunk.size - 1] == 10)
138                                 chunk.memory[chunk.size - 1] = '\0';
139
140                         ast_copy_string(buf, chunk.memory, len);
141                         free(chunk.memory);
142                 }
143         } else {
144                 ast_log(LOG_ERROR, "Cannot allocate curl structure\n");
145         }
146
147         ast_module_user_remove(u);
148
149         return 0;
150 }
151
152 struct ast_custom_function acf_curl = {
153         .name = "CURL",
154         .synopsis = "Retrieves the contents of a URL",
155         .syntax = "CURL(url[|post-data])",
156         .desc =
157         "  url       - URL to retrieve\n"
158         "  post-data - Optional data to send as a POST (GET is default action)\n",
159         .read = acf_curl_exec,
160 };
161
162 static int unload_module(void)
163 {
164         int res;
165
166         res = ast_custom_function_unregister(&acf_curl);
167
168         ast_module_user_hangup_all();
169
170         curl_global_cleanup();
171         
172         return res;
173 }
174
175 static int load_module(void)
176 {
177         int res;
178
179         if (curl_global_init(CURL_GLOBAL_ALL)) {
180                 ast_log(LOG_ERROR, "Unable to initialize the CURL library. Cannot load func_curl\n");
181                 return AST_MODULE_LOAD_DECLINE;
182         }       
183
184         res = ast_custom_function_register(&acf_curl);
185
186         return res;
187 }
188
189 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Load external URL");
190