aboutsummaryrefslogtreecommitdiffstats
path: root/trunk/funcs/func_env.c
diff options
context:
space:
mode:
Diffstat (limited to 'trunk/funcs/func_env.c')
-rw-r--r--trunk/funcs/func_env.c215
1 files changed, 215 insertions, 0 deletions
diff --git a/trunk/funcs/func_env.c b/trunk/funcs/func_env.c
new file mode 100644
index 000000000..4eba211bb
--- /dev/null
+++ b/trunk/funcs/func_env.c
@@ -0,0 +1,215 @@
+/*
+ * Asterisk -- An open source telephony toolkit.
+ *
+ * Copyright (C) 1999 - 2006, Digium, Inc.
+ *
+ * See http://www.asterisk.org for more information about
+ * the Asterisk project. Please do not directly contact
+ * any of the maintainers of this project for assistance;
+ * the project provides a web site, mailing lists and IRC
+ * channels for your use.
+ *
+ * This program is free software, distributed under the terms of
+ * the GNU General Public License Version 2. See the LICENSE file
+ * at the top of the source tree.
+ */
+
+/*! \file
+ *
+ * \brief Environment related dialplan functions
+ *
+ * \ingroup functions
+ */
+
+#include "asterisk.h"
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include <sys/stat.h>
+
+#include "asterisk/module.h"
+#include "asterisk/channel.h"
+#include "asterisk/pbx.h"
+#include "asterisk/utils.h"
+#include "asterisk/app.h"
+
+static int env_read(struct ast_channel *chan, const char *cmd, char *data,
+ char *buf, size_t len)
+{
+ char *ret = NULL;
+
+ *buf = '\0';
+
+ if (data)
+ ret = getenv(data);
+
+ if (ret)
+ ast_copy_string(buf, ret, len);
+
+ return 0;
+}
+
+static int env_write(struct ast_channel *chan, const char *cmd, char *data,
+ const char *value)
+{
+ if (!ast_strlen_zero(data)) {
+ if (!ast_strlen_zero(value)) {
+ setenv(data, value, 1);
+ } else {
+ unsetenv(data);
+ }
+ }
+
+ return 0;
+}
+
+static int stat_read(struct ast_channel *chan, const char *cmd, char *data,
+ char *buf, size_t len)
+{
+ char *action;
+ struct stat s;
+
+ ast_copy_string(buf, "0", len);
+
+ action = strsep(&data, ",");
+ if (stat(data, &s)) {
+ return 0;
+ } else {
+ switch (*action) {
+ case 'e':
+ strcpy(buf, "1");
+ break;
+ case 's':
+ snprintf(buf, len, "%d", (unsigned int) s.st_size);
+ break;
+ case 'f':
+ snprintf(buf, len, "%d", S_ISREG(s.st_mode) ? 1 : 0);
+ break;
+ case 'd':
+ snprintf(buf, len, "%d", S_ISDIR(s.st_mode) ? 1 : 0);
+ break;
+ case 'M':
+ snprintf(buf, len, "%d", (int) s.st_mtime);
+ break;
+ case 'A':
+ snprintf(buf, len, "%d", (int) s.st_mtime);
+ break;
+ case 'C':
+ snprintf(buf, len, "%d", (int) s.st_ctime);
+ break;
+ case 'm':
+ snprintf(buf, len, "%o", (int) s.st_mode);
+ break;
+ }
+ }
+
+ return 0;
+}
+
+static int file_read(struct ast_channel *chan, const char *cmd, char *data, char *buf, size_t len)
+{
+ AST_DECLARE_APP_ARGS(args,
+ AST_APP_ARG(filename);
+ AST_APP_ARG(offset);
+ AST_APP_ARG(length);
+ );
+ int offset = 0, length;
+ char *contents;
+
+ AST_STANDARD_APP_ARGS(args, data);
+ if (args.argc > 1)
+ offset = atoi(args.offset);
+
+ if (args.argc > 2) {
+ if ((length = atoi(args.length)) < 1) {
+ ast_log(LOG_WARNING, "Invalid length '%s'. Returning the max (%d)\n", args.length, (int)len);
+ length = len;
+ } else if (length > len) {
+ ast_log(LOG_WARNING, "Length %d is greater than the max (%d). Truncating output.\n", length, (int)len);
+ length = len;
+ }
+ } else
+ length = len;
+
+ if (!(contents = ast_read_textfile(args.filename)))
+ return -1;
+
+ if (offset >= 0)
+ ast_copy_string(buf, &contents[offset], length);
+ else {
+ size_t tmp = strlen(contents);
+ if (offset * -1 > tmp) {
+ ast_log(LOG_WARNING, "Offset is larger than the file size.\n");
+ offset = tmp * -1;
+ }
+ ast_copy_string(buf, &contents[tmp + offset], length);
+ }
+ ast_free(contents);
+
+ return 0;
+}
+
+static struct ast_custom_function env_function = {
+ .name = "ENV",
+ .synopsis = "Gets or sets the environment variable specified",
+ .syntax = "ENV(<envname>)",
+ .read = env_read,
+ .write = env_write,
+};
+
+static struct ast_custom_function stat_function = {
+ .name = "STAT",
+ .synopsis = "Does a check on the specified file",
+ .syntax = "STAT(<flag>,<filename>)",
+ .read = stat_read,
+ .desc =
+ "flag may be one of the following:\n"
+ " d - Checks if the file is a directory\n"
+ " e - Checks if the file exists\n"
+ " f - Checks if the file is a regular file\n"
+ " m - Returns the file mode (in octal)\n"
+ " s - Returns the size (in bytes) of the file\n"
+ " A - Returns the epoch at which the file was last accessed\n"
+ " C - Returns the epoch at which the inode was last changed\n"
+ " M - Returns the epoch at which the file was last modified\n",
+};
+
+static struct ast_custom_function file_function = {
+ .name = "FILE",
+ .synopsis = "Obtains the contents of a file",
+ .syntax = "FILE(<filename>,<offset>,<length>)",
+ .read = file_read,
+ /*
+ * Some enterprising programmer could probably add write functionality
+ * to FILE(), although I'm not sure how useful it would be. Hence why
+ * it's called FILE and not READFILE (like the app was).
+ */
+ .desc =
+"<offset> may be specified as any number. If negative, <offset> specifies\n"
+" the number of bytes back from the end of the file.\n"
+"<length>, if specified, will limit the length of the data read to that size.\n",
+};
+
+static int unload_module(void)
+{
+ int res = 0;
+
+ res |= ast_custom_function_unregister(&env_function);
+ res |= ast_custom_function_unregister(&stat_function);
+ res |= ast_custom_function_unregister(&file_function);
+
+ return res;
+}
+
+static int load_module(void)
+{
+ int res = 0;
+
+ res |= ast_custom_function_register(&env_function);
+ res |= ast_custom_function_register(&stat_function);
+ res |= ast_custom_function_register(&file_function);
+
+ return res;
+}
+
+AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Environment/filesystem dialplan functions");