dbd82c941bd3f48655996df3f6ad82f7cc5cb4bc
[asterisk/asterisk.git] / funcs / func_rand.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2006, Digium, Inc.
5  * Copyright (C) 2006, Claude Patry
6  *
7  * See http://www.asterisk.org for more information about
8  * the Asterisk project. Please do not directly contact
9  * any of the maintainers of this project for assistance;
10  * the project provides a web site, mailing lists and IRC
11  * channels for your use.
12  *
13  * This program is free software, distributed under the terms of
14  * the GNU General Public License Version 2. See the LICENSE file
15  * at the top of the source tree.
16  */
17
18 /*! \file
19  *
20  * \brief Generate Random Number
21  * 
22  * \author Claude Patry <cpatry@gmail.com>
23  * \author Tilghman Lesher ( http://asterisk.drunkcoder.com/ )
24  */
25
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdio.h>
29 #include <sys/types.h>
30
31 #include "asterisk.h"
32
33 /* ASTERISK_FILE_VERSION(__FILE__, "$Revision: 7682 $") */
34
35 #include "asterisk/channel.h"
36 #include "asterisk/pbx.h"
37 #include "asterisk/logger.h"
38 #include "asterisk/utils.h"
39 #include "asterisk/app.h"
40 #ifndef BUILTIN_FUNC
41 #include "asterisk/module.h"
42 #endif
43
44
45 STANDARD_LOCAL_USER;
46
47 LOCAL_USER_DECL;
48
49
50 static char *acf_rand_exec(struct ast_channel *chan, char *cmd, char *data, char *buffer, size_t buflen)
51 {
52         struct localuser *u;
53         char *args[2] = { "", "" }, *s;
54         int min_int, response_int, max_int;
55
56         LOCAL_USER_ACF_ADD(u);
57
58         if (!(s = ast_strdupa(data))) {
59                 ast_log(LOG_WARNING, "Out of memory\n");
60                 *buffer = '\0';
61                 LOCAL_USER_REMOVE(u);
62                 return buffer;
63         }
64
65         ast_app_separate_args(s, '|', args, sizeof(args) / sizeof(args[0]));
66
67         if (ast_strlen_zero(args[0]) || sscanf(args[0], "%d", &min_int) != 1) {
68                 min_int = 0;
69         }
70
71         if (ast_strlen_zero(args[1]) || sscanf(args[1], "%d", &max_int) != 1) {
72                 max_int = RAND_MAX;
73         }
74
75         if (max_int < min_int) {
76                 int tmp = max_int;
77                 max_int = min_int;
78                 min_int = tmp;
79                 ast_log(LOG_DEBUG, "max<min\n");
80         }
81
82         response_int = min_int + (ast_random() % (max_int - min_int + 1));
83         ast_log(LOG_DEBUG, "%d was the lucky number in range [%d,%d]\n", response_int, min_int, max_int);
84         snprintf(buffer, buflen, "%d", response_int);
85
86         LOCAL_USER_REMOVE(u);
87         return buffer;
88 }
89
90 #ifndef BUILTIN_FUNC
91 static
92 #endif
93 struct ast_custom_function acf_rand = {
94         .name = "RAND",
95         .synopsis = "Choose a random number in a range",
96         .syntax = "RAND([min][,max])",
97         .desc =
98 "Choose a random number between min and max.  Min defaults to 0, if not\n"
99 "specified, while max defaults to RAND_MAX (2147483647 on many systems).\n"
100 "  Example:  Set(junky=${RAND(1,8)}); \n"
101 "  Sets junky to a random number between 1 and 8, inclusive.\n",
102         .read = acf_rand_exec,
103 };
104
105
106 #ifndef BUILTIN_FUNC
107
108 static char *tdesc = "Generate a random number";
109
110 int unload_module(void)
111 {
112         ast_custom_function_unregister(&acf_rand);
113
114         STANDARD_HANGUP_LOCALUSERS;
115
116         return 0;
117 }
118
119 int load_module(void)
120 {
121         return ast_custom_function_register(&acf_rand);
122 }
123
124 char *description(void)
125 {
126        return tdesc;
127 }
128
129 int usecount(void)
130 {
131         int res;
132
133         STANDARD_USECOUNT(res);
134
135         return res;
136 }
137
138 char *key()
139 {
140        return ASTERISK_GPL_KEY;
141 }
142
143 #endif