replace ast_build_string() with ast_str_*().
[asterisk/asterisk.git] / funcs / func_realtime.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2005-2006, BJ Weschke. All rights reserved.
5  * 
6  * BJ Weschke <bweschke@btwtech.com>
7  * 
8  * This code is released by the author with no restrictions on usage. 
9  *
10  * See http://www.asterisk.org for more information about
11  * the Asterisk project. Please do not directly contact
12  * any of the maintainers of this project for assistance;
13  * the project provides a web site, mailing lists and IRC
14  * channels for your use.
15  *
16  */
17
18 /*! \file
19  *
20  * \brief REALTIME dialplan function
21  * 
22  * \author BJ Weschke <bweschke@btwtech.com>
23  * 
24  * \ingroup functions
25  */
26
27 #include "asterisk.h"
28
29 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
30
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <unistd.h>
34 #include <string.h>
35 #include <sys/types.h>
36
37 #include "asterisk/file.h"
38 #include "asterisk/channel.h"
39 #include "asterisk/pbx.h"
40 #include "asterisk/options.h"
41 #include "asterisk/config.h"
42 #include "asterisk/module.h"
43 #include "asterisk/lock.h"
44 #include "asterisk/logger.h"
45 #include "asterisk/utils.h"
46 #include "asterisk/app.h"
47
48 static int function_realtime_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
49 {
50         struct ast_variable *var, *head;
51         struct ast_module_user *u;
52         struct ast_str *out;
53         size_t resultslen;
54         int n;
55         AST_DECLARE_APP_ARGS(args,
56                 AST_APP_ARG(family);
57                 AST_APP_ARG(fieldmatch);
58                 AST_APP_ARG(value);
59                 AST_APP_ARG(delim1);
60                 AST_APP_ARG(delim2);
61         );
62
63
64         if (ast_strlen_zero(data)) {
65                 ast_log(LOG_WARNING, "Syntax: REALTIME(family|fieldmatch[|value[|delim1[|delim2]]]) - missing argument!\n");
66                 return -1;
67         }
68
69         u = ast_module_user_add(chan);
70
71         AST_STANDARD_APP_ARGS(args, data);
72
73         if (!args.delim1)
74                 args.delim1 = "|";
75         if (!args.delim2)
76                 args.delim2 = "=";
77
78         head = ast_load_realtime(args.family, args.fieldmatch, args.value, NULL);
79
80         if (!head) {
81                 ast_module_user_remove(u);
82                 return -1;
83         }
84         resultslen = 0;
85         n = 0;
86         for (var = head; var; n++, var = var->next)
87                 resultslen += strlen(var->name) + strlen(var->value);
88         /* add space for delimiters and final '\0' */
89         resultslen += n * (strlen(args.delim1) + strlen(args.delim2)) + 1;
90
91         out = ast_str_alloca(resultslen);
92         for (var = head; var; var = var->next)
93                 ast_str_append(&out, 0, "%s%s%s%s", var->name, args.delim2, var->value, args.delim1);
94         ast_copy_string(buf, out->str, len);
95
96         ast_module_user_remove(u);
97
98         return 0;
99 }
100
101 static int function_realtime_write(struct ast_channel *chan, char *cmd, char *data, const char *value)
102 {
103         struct ast_module_user *u;
104         int res = 0;
105         AST_DECLARE_APP_ARGS(args,
106                 AST_APP_ARG(family);
107                 AST_APP_ARG(fieldmatch);
108                 AST_APP_ARG(value);
109                 AST_APP_ARG(field);
110         );
111
112         if (ast_strlen_zero(data)) {
113                 ast_log(LOG_WARNING, "Syntax: REALTIME(family|fieldmatch|value|newcol) - missing argument!\n");
114                 return -1;
115         }
116
117         u = ast_module_user_add(chan);
118
119         AST_STANDARD_APP_ARGS(args, data);
120
121         res = ast_update_realtime(args.family, args.fieldmatch, args.value, args.field, (char *)value, NULL);
122
123         if (res < 0) {
124                 ast_log(LOG_WARNING, "Failed to update. Check the debug log for possible data repository related entries.\n");
125         }
126
127         ast_module_user_remove(u);
128
129         return 0;
130 }
131
132 struct ast_custom_function realtime_function = {
133         .name = "REALTIME",
134         .synopsis = "RealTime Read/Write Functions",
135         .syntax = "REALTIME(family|fieldmatch[|value[|delim1[|delim2]]]) on read\n"
136                   "REALTIME(family|fieldmatch|value|field) on write\n",
137         .desc = "This function will read or write values from/to a RealTime repository.\n"
138                 "REALTIME(....) will read names/values from the repository, and \n"
139                 "REALTIME(....)= will write a new value/field to the repository. On a\n"
140                 "read, this function returns a delimited text string. The name/value \n"
141                 "pairs are delimited by delim1, and the name and value are delimited \n"
142                 "between each other with delim2. The default for delim1 is '=' and   \n"
143                 "the default for delim2 is '|'. If there is no match, NULL will be   \n"
144                 "returned by the function. On a write, this function will always     \n"
145                 "return NULL. \n",
146         .read = function_realtime_read,
147         .write = function_realtime_write,
148 };
149
150 static int unload_module(void)
151 {
152         int res = ast_custom_function_unregister(&realtime_function);
153
154         ast_module_user_hangup_all();
155
156         return res;
157 }
158
159 static int load_module(void)
160 {
161         int res = ast_custom_function_register(&realtime_function);
162
163         return res;
164 }
165
166 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Read/Write values from a RealTime repository");