Bug 5440 - Add STAT() dialplan function
[asterisk/asterisk.git] / funcs / func_env.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * See http://www.asterisk.org for more information about
7  * the Asterisk project. Please do not directly contact
8  * any of the maintainers of this project for assistance;
9  * the project provides a web site, mailing lists and IRC
10  * channels for your use.
11  *
12  * This program is free software, distributed under the terms of
13  * the GNU General Public License Version 2. See the LICENSE file
14  * at the top of the source tree.
15  */
16
17 /*! \file
18  *
19  * \brief Environment related dialplan functions
20  * 
21  */
22
23 #include <stdlib.h>
24 #include <string.h>
25 #include <sys/types.h>
26 #include <sys/stat.h>
27
28 #include "asterisk.h"
29
30 /* ASTERISK_FILE_VERSION(__FILE__, "$Revision$") */
31
32 #include "asterisk/channel.h"
33 #include "asterisk/pbx.h"
34 #include "asterisk/logger.h"
35 #include "asterisk/utils.h"
36 #include "asterisk/app.h"
37
38 static char *builtin_function_env_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
39 {
40         char *ret = "";
41
42         if (data) {
43                 ret = getenv(data);
44                 if (!ret)
45                         ret = "";
46         }
47         ast_copy_string(buf, ret, len);
48
49         return buf;
50 }
51
52 static void builtin_function_env_write(struct ast_channel *chan, char *cmd, char *data, const char *value) 
53 {
54         if (!ast_strlen_zero(data)) {
55                 if (!ast_strlen_zero(value)) {
56                         setenv(data, value, 1);
57                 } else {
58                         unsetenv(data);
59                 }
60         }
61 }
62
63 static char *builtin_function_stat_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len)
64 {
65         char *action;
66         struct stat s;
67
68         ast_copy_string(buf, "0", len);
69         if (!data) {
70                 ast_log(LOG_ERROR, "Out of memory\n");
71                 return buf;
72         }
73
74         action = strsep(&data, "|");
75         if (stat(data, &s)) {
76                 return buf;
77         } else {
78                 switch (*action) {
79                 case 'e':
80                         ast_copy_string(buf, "1", len);
81                         break;
82                 case 's':
83                         snprintf(buf, len, "%d", (unsigned int)s.st_size);
84                         break;
85                 case 'f':
86                         snprintf(buf, len, "%d", S_ISREG(s.st_mode) ? 1 : 0);
87                         break;
88                 case 'd':
89                         snprintf(buf, len, "%d", S_ISDIR(s.st_mode) ? 1 : 0);
90                         break;
91                 case 'M':
92                         snprintf(buf, len, "%d", (int)s.st_mtime);
93                         break;
94                 case 'A':
95                         snprintf(buf, len, "%d", (int)s.st_mtime);
96                         break;
97                 case 'C':
98                         snprintf(buf, len, "%d", (int)s.st_ctime);
99                         break;
100                 case 'm':
101                         snprintf(buf, len, "%o", (int)s.st_mode);
102                         break;
103                 }
104         }
105         return buf;
106 }
107
108 #ifndef BUILTIN_FUNC
109 static
110 #endif
111 struct ast_custom_function env_function = {
112         .name = "ENV",
113         .synopsis = "Gets or sets the environment variable specified",
114         .syntax = "ENV(<envname>)",
115         .read = builtin_function_env_read,
116         .write = builtin_function_env_write,
117 };
118
119 #ifndef BUILTIN_FUNC
120 static
121 #endif
122 struct ast_custom_function stat_function = {
123         .name = "STAT",
124         .synopsis = "Does a check on the specified file",
125         .syntax = "STAT(<flag>,<filename>)",
126         .read = builtin_function_stat_read,
127         .desc =
128 "flag may be one of the following:\n"
129 "  d - Checks if the file is a directory\n"
130 "  e - Checks if the file exists\n"
131 "  f - Checks if the file is a regular file\n"
132 "  m - Returns the file mode (in octal)\n"
133 "  s - Returns the size (in bytes) of the file\n"
134 "  A - Returns the epoch at which the file was last accessed\n"
135 "  C - Returns the epoch at which the inode was last changed\n"
136 "  M - Returns the epoch at which the file was last modified\n",
137 };
138