patch file cleanup... ick.. kaloz you owe me some love
[openwrt/svn-archive/archive.git] / net / asterisk / patches / 007-app_mysql.patch
1 diff -Nru asterisk-1.2.14.org/apps/app_sql_mysql.c asterisk-1.2.14/apps/app_sql_mysql.c
2 --- asterisk-1.2.14.org/apps/app_sql_mysql.c 1970-01-01 01:00:00.000000000 +0100
3 +++ asterisk-1.2.14/apps/app_sql_mysql.c 2006-12-27 09:00:04.000000000 +0100
4 @@ -0,0 +1,445 @@
5 +/*
6 + * Asterisk -- A telephony toolkit for Linux.
7 + *
8 + * Connect to MySQL
9 + *
10 + * Copyright (C) 2004, Constantine Filin and Christos Ricudis
11 + *
12 + * Christos Ricudis <ricudis@itc.auth.gr>
13 + * Constantine Filin <cf@intermedia.net>
14 + *
15 + * This program is free software, distributed under the terms of
16 + * the GNU General Public License
17 + */
18 +
19 +#include <stdlib.h>
20 +#include <unistd.h>
21 +#include <string.h>
22 +#include <stdlib.h>
23 +#include <sys/types.h>
24 +#include <stdio.h>
25 +#include <unistd.h>
26 +
27 +#include <mysql.h>
28 +
29 +#include <asterisk/file.h>
30 +#include <asterisk/logger.h>
31 +#include <asterisk/channel.h>
32 +#include <asterisk/pbx.h>
33 +#include <asterisk/module.h>
34 +#include <asterisk/linkedlists.h>
35 +#include <asterisk/chanvars.h>
36 +#include <asterisk/lock.h>
37 +
38 +#define EXTRA_LOG 0
39 +
40 +static char *tdesc = "Simple Mysql Interface";
41 +
42 +static char *app = "MYSQL";
43 +
44 +static char *synopsis = "Do several mySQLy things";
45 +
46 +static char *descrip =
47 +"MYSQL(): Do several mySQLy things\n"
48 +"Syntax:\n"
49 +" MYSQL(Connect connid dhhost dbuser dbpass dbname)\n"
50 +" Connects to a database. Arguments contain standard MySQL parameters\n"
51 +" passed to function mysql_real_connect. Connection identifer returned\n"
52 +" in ${var}\n"
53 +" MYSQL(Query resultid ${connid} query-string)\n"
54 +" Executes standard MySQL query contained in query-string using established\n"
55 +" connection identified by ${connection_identifier}. Result of query is\n"
56 +" is stored in ${var}.\n"
57 +" MYSQL(Fetch fetchid ${resultid} var1 var2 ... varN)\n"
58 +" Fetches a single row from a result set contained in ${result_identifier}.\n"
59 +" Assigns returned fields to ${var1} ... ${varn}. ${fetchid} is set TRUE\n"
60 +" if additional rows exist in result set.\n"
61 +" MYSQL(Clear ${resultid})\n"
62 +" Frees memory and datastructures associated with result set.\n"
63 +" MYSQL(Disconnect ${connid})\n"
64 +" Disconnects from named connection to MySQL.\n"
65 +" On exit, always returns 0. Sets MYSQL_STATUS to 0 on success and -1 on error.\n";
66 +
67 +/*
68 +EXAMPLES OF USE :
69 +
70 +exten => s,2,MYSQL(Connect connid localhost asterisk mypass credit)
71 +exten => s,3,MYSQL(Query resultid ${connid} SELECT username,credit FROM credit WHERE callerid=${CALLERIDNUM})
72 +exten => s,4,MYSQL(Fetch fetchid ${resultid} datavar1 datavar2)
73 +exten => s,5,GotoIf(${fetchid}?6:8)
74 +exten => s,6,Festival("User ${datavar1} currently has credit balance of ${datavar2} dollars.")
75 +exten => s,7,Goto(s,4)
76 +exten => s,8,MYSQL(Clear ${resultid})
77 +exten => s,9,MYSQL(Disconnect ${connid})
78 +*/
79 +
80 +STANDARD_LOCAL_USER;
81 +LOCAL_USER_DECL;
82 +
83 +AST_MUTEX_DEFINE_STATIC(_mysql_mutex);
84 +
85 +#define AST_MYSQL_ID_DUMMY 0
86 +#define AST_MYSQL_ID_CONNID 1
87 +#define AST_MYSQL_ID_RESID 2
88 +#define AST_MYSQL_ID_FETCHID 3
89 +
90 +struct ast_MYSQL_id {
91 + int identifier_type; /* 0=dummy, 1=connid, 2=resultid */
92 + int identifier;
93 + void *data;
94 + AST_LIST_ENTRY(ast_MYSQL_id) entries;
95 +} *ast_MYSQL_id;
96 +
97 +AST_LIST_HEAD(MYSQLidshead,ast_MYSQL_id) _mysql_ids_head;
98 +
99 +/* helpful procs */
100 +static void *find_identifier(int identifier,int identifier_type) {
101 + struct MYSQLidshead *headp;
102 + struct ast_MYSQL_id *i;
103 + void *res=NULL;
104 + int found=0;
105 +
106 + headp=&_mysql_ids_head;
107 +
108 + if (AST_LIST_LOCK(headp)) {
109 + ast_log(LOG_WARNING,"Unable to lock identifiers list\n");
110 + } else {
111 + AST_LIST_TRAVERSE(headp,i,entries) {
112 + if ((i->identifier==identifier) && (i->identifier_type==identifier_type)) {
113 + found=1;
114 + res=i->data;
115 + break;
116 + }
117 + }
118 + if (!found) {
119 + ast_log(LOG_WARNING,"Identifier %d, identifier_type %d not found in identifier list\n",identifier,identifier_type);
120 + }
121 + AST_LIST_UNLOCK(headp);
122 + }
123 +
124 + return res;
125 +}
126 +
127 +static int add_identifier(int identifier_type,void *data) {
128 + struct ast_MYSQL_id *i,*j;
129 + struct MYSQLidshead *headp;
130 + int maxidentifier=0;
131 +
132 + headp=&_mysql_ids_head;
133 + i=NULL;
134 + j=NULL;
135 +
136 + if (AST_LIST_LOCK(headp)) {
137 + ast_log(LOG_WARNING,"Unable to lock identifiers list\n");
138 + return(-1);
139 + } else {
140 + i=malloc(sizeof(struct ast_MYSQL_id));
141 + AST_LIST_TRAVERSE(headp,j,entries) {
142 + if (j->identifier>maxidentifier) {
143 + maxidentifier=j->identifier;
144 + }
145 + }
146 + i->identifier=maxidentifier+1;
147 + i->identifier_type=identifier_type;
148 + i->data=data;
149 + AST_LIST_INSERT_HEAD(headp,i,entries);
150 + AST_LIST_UNLOCK(headp);
151 + }
152 + return i->identifier;
153 +}
154 +
155 +static int del_identifier(int identifier,int identifier_type) {
156 + struct ast_MYSQL_id *i;
157 + struct MYSQLidshead *headp;
158 + int found=0;
159 +
160 + headp=&_mysql_ids_head;
161 +
162 + if (AST_LIST_LOCK(headp)) {
163 + ast_log(LOG_WARNING,"Unable to lock identifiers list\n");
164 + } else {
165 + AST_LIST_TRAVERSE(headp,i,entries) {
166 + if ((i->identifier==identifier) &&
167 + (i->identifier_type==identifier_type)) {
168 + AST_LIST_REMOVE(headp,i,entries);
169 + free(i);
170 + found=1;
171 + break;
172 + }
173 + }
174 + AST_LIST_UNLOCK(headp);
175 + }
176 +
177 + if (found==0) {
178 + ast_log(LOG_WARNING,"Could not find identifier %d, identifier_type %d in list to delete\n",identifier,identifier_type);
179 + return(-1);
180 + } else {
181 + return(0);
182 + }
183 +}
184 +
185 +static int set_asterisk_int(struct ast_channel *chan, char *varname, int id) {
186 + if( id>=0 ) {
187 + char s[100] = "";
188 + snprintf(s, sizeof(s)-1, "%d", id);
189 +#if EXTRA_LOG
190 + ast_log(LOG_WARNING,"MYSQL: setting var '%s' to value '%s'\n",varname,s);
191 +#endif
192 + pbx_builtin_setvar_helper(chan,varname,s);
193 + }
194 + return id;
195 +}
196 +
197 +static int add_identifier_and_set_asterisk_int(struct ast_channel *chan, char *varname, int identifier_type, void *data) {
198 + return set_asterisk_int(chan,varname,add_identifier(identifier_type,data));
199 +}
200 +
201 +static int safe_scan_int( char** data, char* delim, int def ) {
202 + char* end;
203 + int res = def;
204 + char* s = strsep(data,delim);
205 + if( s ) {
206 + res = strtol(s,&end,10);
207 + if (*end) res = def; /* not an integer */
208 + }
209 + return res;
210 +}
211 +
212 +/* MYSQL operations */
213 +static int aMYSQL_connect(struct ast_channel *chan, char *data) {
214 +
215 + MYSQL *mysql;
216 +
217 + char *connid_var;
218 + char *dbhost;
219 + char *dbuser;
220 + char *dbpass;
221 + char *dbname;
222 +
223 + strsep(&data," "); // eat the first token, we already know it :P
224 +
225 + connid_var=strsep(&data," ");
226 + dbhost=strsep(&data," ");
227 + dbuser=strsep(&data," ");
228 + dbpass=strsep(&data," ");
229 + dbname=strsep(&data,"\n");
230 +
231 + if( connid_var && dbhost && dbuser && dbpass && dbname ) {
232 + mysql = mysql_init(NULL);
233 + if (mysql) {
234 + if (mysql_real_connect(mysql,dbhost,dbuser,dbpass,dbname,0,NULL,0)) {
235 + add_identifier_and_set_asterisk_int(chan,connid_var,AST_MYSQL_ID_CONNID,mysql);
236 + return 0;
237 + }
238 + else {
239 + ast_log(LOG_WARNING,"mysql_real_connect(mysql,%s,%s,dbpass,%s,...) failed\n",dbhost,dbuser,dbname);
240 + }
241 + }
242 + else {
243 + ast_log(LOG_WARNING,"myslq_init returned NULL\n");
244 + }
245 + }
246 + else {
247 + ast_log(LOG_WARNING,"MYSQL(connect is missing some arguments\n");
248 + }
249 +
250 + return -1;
251 +}
252 +
253 +static int aMYSQL_query(struct ast_channel *chan, char *data) {
254 +
255 + MYSQL *mysql;
256 + MYSQL_RES *mysqlres;
257 +
258 + char *resultid_var;
259 + int connid;
260 + char *querystring;
261 +
262 + strsep(&data," "); // eat the first token, we already know it :P
263 +
264 + resultid_var = strsep(&data," ");
265 + connid = safe_scan_int(&data," ",-1);
266 + querystring = strsep(&data,"\n");
267 +
268 + if (resultid_var && (connid>=0) && querystring) {
269 + if ((mysql=find_identifier(connid,AST_MYSQL_ID_CONNID))!=NULL) {
270 + mysql_query(mysql,querystring);
271 + if ((mysqlres=mysql_use_result(mysql))!=NULL) {
272 + add_identifier_and_set_asterisk_int(chan,resultid_var,AST_MYSQL_ID_RESID,mysqlres);
273 + return 0;
274 + }
275 + else if( mysql_field_count(mysql)==0 ) {
276 + return 0; // See http://dev.mysql.com/doc/mysql/en/mysql_field_count.html
277 + }
278 + else {
279 + ast_log(LOG_WARNING,"aMYSQL_query: mysql_store_result() failed on query %s\n",querystring);
280 + }
281 + }
282 + else {
283 + ast_log(LOG_WARNING,"aMYSQL_query: Invalid connection identifier %d passed in aMYSQL_query\n",connid);
284 + }
285 + }
286 + else {
287 + ast_log(LOG_WARNING,"aMYSQL_query: missing some arguments\n");
288 + }
289 +
290 + return -1;
291 +}
292 +
293 +
294 +static int aMYSQL_fetch(struct ast_channel *chan, char *data) {
295 +
296 + MYSQL_RES *mysqlres;
297 + MYSQL_ROW mysqlrow;
298 +
299 + char *fetchid_var,*s5,*s6;
300 + int resultid,numFields,j;
301 +
302 + strsep(&data," "); // eat the first token, we already know it :P
303 +
304 + fetchid_var = strsep(&data," ");
305 + resultid = safe_scan_int(&data," ",-1);
306 +
307 + if (fetchid_var && (resultid>=0) ) {
308 + if ((mysqlres=find_identifier(resultid,AST_MYSQL_ID_RESID))!=NULL) {
309 + /* Grab the next row */
310 + if ((mysqlrow=mysql_fetch_row(mysqlres))!=NULL) {
311 + numFields=mysql_num_fields(mysqlres);
312 + for (j=0;j<numFields;j++) {
313 + s5=strsep(&data," ");
314 + if (s5==NULL) {
315 + ast_log(LOG_WARNING,"ast_MYSQL_fetch: More fields (%d) than variables (%d)\n",numFields,j);
316 + break;
317 + }
318 + s6=mysqlrow[j];
319 + pbx_builtin_setvar_helper(chan,s5, s6 ? s6 : "NULL");
320 + }
321 +#ifdef EXTRA_LOG
322 + ast_log(LOG_WARNING,"ast_MYSQL_fetch: numFields=%d\n",numFields);
323 +#endif
324 + set_asterisk_int(chan,fetchid_var,1); // try more rows
325 + } else {
326 +#if EXTRA_LOG
327 + ast_log(LOG_WARNING,"ast_MYSQL_fetch : EOF\n");
328 +#endif
329 + set_asterisk_int(chan,fetchid_var,0); // no more rows
330 + }
331 + return 0;
332 + }
333 + else {
334 + ast_log(LOG_WARNING,"aMYSQL_fetch: Invalid result identifier %d passed\n",resultid);
335 + }
336 + }
337 + else {
338 + ast_log(LOG_WARNING,"aMYSQL_fetch: missing some arguments\n");
339 + }
340 +
341 + return -1;
342 +}
343 +
344 +static int aMYSQL_clear(struct ast_channel *chan, char *data) {
345 +
346 + MYSQL_RES *mysqlres;
347 +
348 + int id;
349 + strsep(&data," "); // eat the first token, we already know it :P
350 + id = safe_scan_int(&data," \n",-1);
351 + if ((mysqlres=find_identifier(id,AST_MYSQL_ID_RESID))==NULL) {
352 + ast_log(LOG_WARNING,"Invalid result identifier %d passed in aMYSQL_clear\n",id);
353 + } else {
354 + mysql_free_result(mysqlres);
355 + del_identifier(id,AST_MYSQL_ID_RESID);
356 + }
357 +
358 + return 0;
359 +}
360 +
361 +static int aMYSQL_disconnect(struct ast_channel *chan, char *data) {
362 +
363 + MYSQL *mysql;
364 + int id;
365 + strsep(&data," "); // eat the first token, we already know it :P
366 +
367 + id = safe_scan_int(&data," \n",-1);
368 + if ((mysql=find_identifier(id,AST_MYSQL_ID_CONNID))==NULL) {
369 + ast_log(LOG_WARNING,"Invalid connection identifier %d passed in aMYSQL_disconnect\n",id);
370 + } else {
371 + mysql_close(mysql);
372 + del_identifier(id,AST_MYSQL_ID_CONNID);
373 + }
374 +
375 + return 0;
376 +}
377 +
378 +static int MYSQL_exec(struct ast_channel *chan, void *data)
379 +{
380 + struct localuser *u;
381 + int result;
382 + char sresult[10];
383 +
384 +#if EXTRA_LOG
385 + fprintf(stderr,"MYSQL_exec: data=%s\n",(char*)data);
386 +#endif
387 +
388 + if (!data) {
389 + ast_log(LOG_WARNING, "APP_MYSQL requires an argument (see manual)\n");
390 + return -1;
391 + }
392 +
393 + LOCAL_USER_ADD(u);
394 + result=0;
395 +
396 + ast_mutex_lock(&_mysql_mutex);
397 +
398 + if (strncasecmp("connect",data,strlen("connect"))==0) {
399 + result=aMYSQL_connect(chan,ast_strdupa(data));
400 + } else if (strncasecmp("query",data,strlen("query"))==0) {
401 + result=aMYSQL_query(chan,ast_strdupa(data));
402 + } else if (strncasecmp("fetch",data,strlen("fetch"))==0) {
403 + result=aMYSQL_fetch(chan,ast_strdupa(data));
404 + } else if (strncasecmp("clear",data,strlen("clear"))==0) {
405 + result=aMYSQL_clear(chan,ast_strdupa(data));
406 + } else if (strncasecmp("disconnect",data,strlen("disconnect"))==0) {
407 + result=aMYSQL_disconnect(chan,ast_strdupa(data));
408 + } else {
409 + ast_log(LOG_WARNING, "Unknown argument to MYSQL application : %s\n",(char *)data);
410 + result=-1;
411 + }
412 +
413 + ast_mutex_unlock(&_mysql_mutex);
414 +
415 + LOCAL_USER_REMOVE(u);
416 + snprintf(sresult, sizeof(sresult), "%d", result);
417 + pbx_builtin_setvar_helper(chan, "MYSQL_STATUS", sresult);
418 + return 0;
419 +}
420 +
421 +int unload_module(void)
422 +{
423 + STANDARD_HANGUP_LOCALUSERS;
424 + return ast_unregister_application(app);
425 +}
426 +
427 +int load_module(void)
428 +{
429 + struct MYSQLidshead *headp = &_mysql_ids_head;
430 + AST_LIST_HEAD_INIT(headp);
431 + return ast_register_application(app, MYSQL_exec, synopsis, descrip);
432 +}
433 +
434 +char *description(void)
435 +{
436 + return tdesc;
437 +}
438 +
439 +int usecount(void)
440 +{
441 + int res;
442 + STANDARD_USECOUNT(res);
443 + return res;
444 +}
445 +
446 +char *key()
447 +{
448 + return ASTERISK_GPL_KEY;
449 +}