asterisk/funcs/func_env.c

/*
 * Asterisk -- An open source telephony toolkit.
 *
 * Copyright (C) 1999 - 2005, 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
 * 
 */

#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>

#include "asterisk.h"

/* ASTERISK_FILE_VERSION(__FILE__, "$Revision$") */

#include "asterisk/channel.h"
#include "asterisk/pbx.h"
#include "asterisk/logger.h"
#include "asterisk/utils.h"
#include "asterisk/app.h"

static char *builtin_function_env_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
{
	char *ret = "";

	if (data) {
		ret = getenv(data);
		if (!ret)
			ret = "";
	}
	ast_copy_string(buf, ret, len);

	return buf;
}

static void builtin_function_env_write(struct ast_channel *chan, char *cmd, char *data, const char *value) 
{
	if (!ast_strlen_zero(data)) {
		if (!ast_strlen_zero(value)) {
			setenv(data, value, 1);
		} else {
			unsetenv(data);
		}
	}
}

static char *builtin_function_stat_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len)
{
	char *action;
	struct stat s;

	ast_copy_string(buf, "0", len);
	if (!data) {
		ast_log(LOG_ERROR, "Out of memory\n");
		return buf;
	}

	action = strsep(&data, "|");
	if (stat(data, &s)) {
		return buf;
	} else {
		switch (*action) {
		case 'e':
			ast_copy_string(buf, "1", len);
			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 buf;
}

#ifndef BUILTIN_FUNC
static
#endif
struct ast_custom_function env_function = {
	.name = "ENV",
	.synopsis = "Gets or sets the environment variable specified",
	.syntax = "ENV(<envname>)",
	.read = builtin_function_env_read,
	.write = builtin_function_env_write,
};

#ifndef BUILTIN_FUNC
static
#endif
struct ast_custom_function stat_function = {
	.name = "STAT",
	.synopsis = "Does a check on the specified file",
	.syntax = "STAT(<flag>,<filename>)",
	.read = builtin_function_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",
};