func_env.c File Reference

Environment related dialplan functions. More...

#include "asterisk.h"
#include <sys/stat.h>
#include <libgen.h>
#include "asterisk/module.h"
#include "asterisk/channel.h"
#include "asterisk/pbx.h"
#include "asterisk/utils.h"
#include "asterisk/app.h"
#include "asterisk/file.h"

Include dependency graph for func_env.c:

Go to the source code of this file.

Macros
#define	LINE_COUNTER(cptr, term, counter)

Enumerations
enum	file_format { FF_UNKNOWN = -1 , FF_UNIX , FF_DOS , FF_MAC }

Functions
static void	__reg_module (void)
static void	__unreg_module (void)
struct ast_module *	AST_MODULE_SELF_SYM (void)
static int64_t	count_lines (const char *filename, enum file_format newline_format)
static int	env_read (struct ast_channel *chan, const char *cmd, char *data, char *buf, size_t len)
static int	env_write (struct ast_channel *chan, const char *cmd, char *data, const char *value)
static enum file_format	file2format (const char *filename)
static int	file_basename (struct ast_channel *chan, const char *cmd, char *data, char *buf, size_t len)
static int	file_count_line (struct ast_channel *chan, const char *cmd, char *data, struct ast_str **buf, ssize_t len)
static int	file_dirname (struct ast_channel *chan, const char *cmd, char *data, char *buf, size_t len)
static int	file_format (struct ast_channel *chan, const char *cmd, char *data, struct ast_str **buf, ssize_t len)
static int	file_read (struct ast_channel *chan, const char *cmd, char *data, struct ast_str **buf, ssize_t len)
static int	file_write (struct ast_channel *chan, const char *cmd, char *data, const char *value)
const char *	format2term (enum file_format f)
static int	load_module (void)
static int	stat_read (struct ast_channel *chan, const char *cmd, char *data, char *buf, size_t len)
static int	unload_module (void)

Variables
static struct ast_module_info	__mod_info = { .name = AST_MODULE, .flags = AST_MODFLAG_LOAD_ORDER , .description = "Environment/filesystem dialplan functions" , .key = "This paragraph is copyright (c) 2006 by Digium, Inc. \In order for your module to load, it must return this \key via a function called \"key\". Any code which \includes this paragraph must be licensed under the GNU \General Public License version 2 or later (at your \option). In addition to Digium's general reservations \of rights, Digium expressly reserves the right to \allow other parties to license this paragraph under \different terms. Any use of Digium, Inc. trademarks or \logos (including \"Asterisk\" or \"Digium\") without \express written permission of Digium, Inc. is prohibited.\n" , .buildopt_sum = AST_BUILDOPT_SUM, .load = load_module, .unload = unload_module, .load_pri = AST_MODPRI_DEFAULT, .support_level = AST_MODULE_SUPPORT_CORE, }
static const struct ast_module_info *	ast_module_info = &__mod_info
static struct ast_custom_function	env_function
static struct ast_custom_function	file_basename_function
static struct ast_custom_function	file_count_line_function
static struct ast_custom_function	file_dirname_function
static struct ast_custom_function	file_format_function
static struct ast_custom_function	file_function
static struct ast_custom_function	stat_function

Detailed Description

Environment related dialplan functions.

Definition in file func_env.c.

Macro Definition Documentation

LINE_COUNTER

#define LINE_COUNTER	(		cptr,
			term,
			counter
	)

Definition at line 547 of file func_env.c.

Enumeration Type Documentation

file_format

enum file_format

Enumerator
FF_UNKNOWN
FF_UNIX
FF_DOS
FF_MAC

Definition at line 440 of file func_env.c.

                 {
    FF_UNKNOWN = -1,
    FF_UNIX,
    FF_DOS,
    FF_MAC,
};

Function Documentation

__reg_module()

static void __reg_module ( void  )

static

Definition at line 1461 of file func_env.c.

__unreg_module()

static void __unreg_module ( void  )

static

Definition at line 1461 of file func_env.c.

AST_MODULE_SELF_SYM()

struct ast_module * AST_MODULE_SELF_SYM

(

void

)

Definition at line 1461 of file func_env.c.

count_lines()

static int64_t count_lines ( const char *  filename, enum file_format  newline_format  )

static

Definition at line 447 of file func_env.c.

{
    int count = 0;
    char fbuf[4096];
    FILE *ff;
 
    if (!(ff = fopen(filename, "r"))) {
        ast_log(LOG_ERROR, "Unable to open '%s': %s\n", filename, strerror(errno));
        return -1;
    }
 
    while (fgets(fbuf, sizeof(fbuf), ff)) {
        char *next = fbuf, *first_cr = NULL, *first_nl = NULL;
 
        /* Must do it this way, because if the fileformat is FF_MAC, then Unix
         * assumptions about line-format will not come into play. */
        while (next) {
            if (newline_format == FF_DOS || newline_format == FF_MAC || newline_format == FF_UNKNOWN) {
                first_cr = strchr(next, '\r');
            }
            if (newline_format == FF_UNIX || newline_format == FF_UNKNOWN) {
                first_nl = strchr(next, '\n');
            }
 
            /* No terminators found in buffer */
            if (!first_cr && !first_nl) {
                break;
            }
 
            if (newline_format == FF_UNKNOWN) {
                if ((first_cr && !first_nl) || (first_cr && first_cr < first_nl)) {
                    if (first_nl && first_nl == first_cr + 1) {
                        newline_format = FF_DOS;
                    } else if (first_cr && first_cr == &fbuf[sizeof(fbuf) - 2]) {
                        /* Get it on the next pass */
                        fseek(ff, -1, SEEK_CUR);
                        break;
                    } else {
                        newline_format = FF_MAC;
                        first_nl = NULL;
                    }
                } else {
                    newline_format = FF_UNIX;
                    first_cr = NULL;
                }
                /* Jump down into next section */
            }
 
            if (newline_format == FF_DOS) {
                if (first_nl && first_cr && first_nl == first_cr + 1) {
                    next = first_nl + 1;
                    count++;
                } else if (first_cr == &fbuf[sizeof(fbuf) - 2]) {
                    /* Get it on the next pass */
                    fseek(ff, -1, SEEK_CUR);
                    break;
                }
            } else if (newline_format == FF_MAC) {
                if (first_cr) {
                    next = first_cr + 1;
                    count++;
                }
            } else if (newline_format == FF_UNIX) {
                if (first_nl) {
                    next = first_nl + 1;
                    count++;
                }
            }
        }
    }
    fclose(ff);
 
    return count;
}

References ast_log, errno, FF_DOS, FF_MAC, FF_UNIX, FF_UNKNOWN, LOG_ERROR, and NULL.

Referenced by file_count_line().

env_read()

static int env_read ( struct ast_channel *  chan, const char *  cmd, char *  data, char *  buf, size_t  len  )

static

Definition at line 367 of file func_env.c.

{
    char *ret = NULL;
 
    *buf = '\0';
 
    if (data)
        ret = getenv(data);
 
    if (ret)
        ast_copy_string(buf, ret, len);
 
    return 0;
}

References ast_copy_string(), buf, len(), and NULL.

env_write()

static int env_write ( struct ast_channel *  chan, const char *  cmd, char *  data, const char *  value  )

static

Definition at line 383 of file func_env.c.

{
    if (!ast_strlen_zero(data) && strncmp(data, "AST_", 4)) {
        if (!ast_strlen_zero(value)) {
            setenv(data, value, 1);
        } else {
            unsetenv(data);
        }
    }
 
    return 0;
}

References ast_strlen_zero(), setenv(), unsetenv(), and value.

file2format()

static enum file_format file2format ( const char *  filename )

static

Definition at line 561 of file func_env.c.

{
    FILE *ff;
    char fbuf[4096];
    char *first_cr, *first_nl;
    enum file_format newline_format = FF_UNKNOWN;
 
    if (!(ff = fopen(filename, "r"))) {
        ast_log(LOG_ERROR, "Cannot open '%s': %s\n", filename, strerror(errno));
        return -1;
    }
 
    while (fgets(fbuf, sizeof(fbuf), ff)) {
        first_cr = strchr(fbuf, '\r');
        first_nl = strchr(fbuf, '\n');
 
        if (!first_cr && !first_nl) {
            continue;
        }
 
        if ((first_cr && !first_nl) || (first_cr && first_cr < first_nl)) {
 
            if (first_nl && first_nl == first_cr + 1) {
                newline_format = FF_DOS;
            } else if (first_cr && first_cr == &fbuf[sizeof(fbuf) - 2]) {
                /* Edge case: get it on the next pass */
                fseek(ff, -1, SEEK_CUR);
                continue;
            } else {
                newline_format = FF_MAC;
            }
        } else {
            newline_format = FF_UNIX;
        }
        break;
    }
    fclose(ff);
    return newline_format;
}

References ast_log, errno, FF_DOS, FF_MAC, FF_UNIX, FF_UNKNOWN, and LOG_ERROR.

Referenced by file_format(), file_read(), and file_write().

file_basename()

static int file_basename ( struct ast_channel *  chan, const char *  cmd, char *  data, char *  buf, size_t  len  )

static

Definition at line 625 of file func_env.c.

{
    char *ret = NULL;
 
    *buf = '\0';
 
    if (data) {
        ret = basename(data);
    }
 
    if (ret) {
        ast_copy_string(buf, ret, len);
    }
 
    return 0;
}

References ast_copy_string(), buf, len(), and NULL.

file_count_line()

static int file_count_line ( struct ast_channel *  chan, const char *  cmd, char *  data, struct ast_str **  buf, ssize_t  len  )

static

Definition at line 522 of file func_env.c.

{
    enum file_format newline_format = FF_UNKNOWN;
    int64_t count;
    AST_DECLARE_APP_ARGS(args,
        AST_APP_ARG(filename);
        AST_APP_ARG(format);
    );
 
    AST_STANDARD_APP_ARGS(args, data);
    if (args.argc > 1) {
        if (tolower(args.format[0]) == 'd') {
            newline_format = FF_DOS;
        } else if (tolower(args.format[0]) == 'm') {
            newline_format = FF_MAC;
        } else if (tolower(args.format[0]) == 'u') {
            newline_format = FF_UNIX;
        }
    }
 
    count = count_lines(args.filename, newline_format);
    ast_str_set(buf, len, "%" PRId64, count);
    return 0;
}

References args, AST_APP_ARG, AST_DECLARE_APP_ARGS, AST_STANDARD_APP_ARGS, ast_str_set(), buf, count_lines(), FF_DOS, FF_MAC, FF_UNIX, FF_UNKNOWN, and len().

file_dirname()

static int file_dirname ( struct ast_channel *  chan, const char *  cmd, char *  data, char *  buf, size_t  len  )

static

Definition at line 608 of file func_env.c.

{
    char *ret = NULL;
 
    *buf = '\0';
 
    if (data) {
        ret = dirname(data);
    }
 
    if (ret) {
        ast_copy_string(buf, ret, len);
    }
 
    return 0;
}

References ast_copy_string(), buf, len(), and NULL.

file_format()

static int file_format ( struct ast_channel *  chan, const char *  cmd, char *  data, struct ast_str **  buf, ssize_t  len  )

static

Definition at line 601 of file func_env.c.

{
    enum file_format newline_format = file2format(data);
    ast_str_set(buf, len, "%c", newline_format == FF_UNIX ? 'u' : newline_format == FF_DOS ? 'd' : newline_format == FF_MAC ? 'm' : 'x');
    return 0;
}

References ast_str_set(), buf, FF_DOS, FF_MAC, FF_UNIX, file2format(), and len().

file_read()

static int file_read ( struct ast_channel *  chan, const char *  cmd, char *  data, struct ast_str **  buf, ssize_t  len  )

static

Definition at line 642 of file func_env.c.

{
    FILE *ff;
    int64_t offset = 0, length = LLONG_MAX;
    enum file_format format = FF_UNKNOWN;
    char fbuf[4096];
    int64_t flength, i; /* iterator needs to be signed, so it can go negative and terminate the loop */
    int64_t offset_offset = -1, length_offset = -1;
    char dos_state = 0;
    AST_DECLARE_APP_ARGS(args,
        AST_APP_ARG(filename);
        AST_APP_ARG(offset);
        AST_APP_ARG(length);
        AST_APP_ARG(options);
        AST_APP_ARG(fileformat);
    );
 
    AST_STANDARD_APP_ARGS(args, data);
 
    if (args.argc > 1) {
        sscanf(args.offset, "%" SCNd64, &offset);
    }
    if (args.argc > 2) {
        sscanf(args.length, "%" SCNd64, &length);
    }
 
    if (args.argc < 4 || !strchr(args.options, 'l')) {
        /* Character-based mode */
        off_t off_i;
 
        if (!(ff = fopen(args.filename, "r"))) {
            ast_log(LOG_WARNING, "Cannot open file '%s' for reading: %s\n", args.filename, strerror(errno));
            return 0;
        }
 
        if (fseeko(ff, 0, SEEK_END) < 0) {
            ast_log(LOG_ERROR, "Cannot seek to end of '%s': %s\n", args.filename, strerror(errno));
            fclose(ff);
            return -1;
        }
        flength = ftello(ff);
 
        if (offset < 0) {
            fseeko(ff, offset, SEEK_END);
            if ((offset = ftello(ff)) < 0) {
                ast_log(AST_LOG_ERROR, "Cannot determine offset position of '%s': %s\n", args.filename, strerror(errno));
                fclose(ff);
                return -1;
            }
        }
        if (length < 0) {
            fseeko(ff, length, SEEK_END);
            if ((length = ftello(ff)) - offset < 0) {
                /* Eliminates all results */
                fclose(ff);
                return -1;
            }
        } else if (length == LLONG_MAX) {
            fseeko(ff, 0, SEEK_END);
            length = ftello(ff);
        }
 
        ast_str_reset(*buf);
 
        fseeko(ff, offset, SEEK_SET);
        for (off_i = ftello(ff); off_i < flength && off_i < offset + length; off_i += sizeof(fbuf)) {
            /* Calculate if we need to retrieve just a portion of the file in memory */
            size_t toappend = sizeof(fbuf);
 
            if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                ast_log(LOG_ERROR, "Short read?!!\n");
                break;
            }
 
            /* Don't go past the length requested */
            if (off_i + toappend > offset + length) {
                toappend = MIN(offset + length - off_i, flength - off_i);
            }
 
            ast_str_append_substr(buf, len, fbuf, toappend);
        }
        fclose(ff);
        return 0;
    }
 
    /* Line-based read */
    if (args.argc == 5) {
        if (tolower(args.fileformat[0]) == 'd') {
            format = FF_DOS;
        } else if (tolower(args.fileformat[0]) == 'm') {
            format = FF_MAC;
        } else if (tolower(args.fileformat[0]) == 'u') {
            format = FF_UNIX;
        }
    }
 
    if (format == FF_UNKNOWN) {
        if ((format = file2format(args.filename)) == FF_UNKNOWN) {
            ast_log(LOG_WARNING, "'%s' is not a line-based file\n", args.filename);
            return -1;
        }
    }
 
    if (offset < 0 && length <= offset) {
        /* Length eliminates all content */
        return -1;
    } else if (offset == 0) {
        offset_offset = 0;
    }
 
    if (!(ff = fopen(args.filename, "r"))) {
        ast_log(LOG_ERROR, "Cannot open '%s': %s\n", args.filename, strerror(errno));
        return -1;
    }
 
    if (fseek(ff, 0, SEEK_END)) {
        ast_log(LOG_ERROR, "Cannot seek to end of file '%s': %s\n", args.filename, strerror(errno));
        fclose(ff);
        return -1;
    }
 
    flength = ftello(ff);
 
    if (length == LLONG_MAX) {
        length_offset = flength;
    }
 
    /* For negative offset and/or negative length */
    if (offset < 0 || length < 0) {
        int64_t count = 0;
        /* Start with an even multiple of fbuf, so at the end of reading with a
         * 0 offset, we don't try to go past the beginning of the file. */
        for (i = (flength / sizeof(fbuf)) * sizeof(fbuf); i >= 0; i -= sizeof(fbuf)) {
            size_t end;
            char *pos;
            if (fseeko(ff, i, SEEK_SET)) {
                ast_log(LOG_ERROR, "Cannot seek to offset %" PRId64 ": %s\n", i, strerror(errno));
            }
            end = fread(fbuf, 1, sizeof(fbuf), ff);
            for (pos = (end < sizeof(fbuf) ? fbuf + end - 1 : fbuf + sizeof(fbuf) - 1); pos >= fbuf; pos--) {
                LINE_COUNTER(pos, format, count);
 
                if (length < 0 && count * -1 == length) {
                    length_offset = i + (pos - fbuf);
                } else if (offset < 0 && count * -1 == (offset - 1)) {
                    /* Found our initial offset.  We're done with reverse motion! */
                    if (format == FF_DOS) {
                        offset_offset = i + (pos - fbuf) + 2;
                    } else {
                        offset_offset = i + (pos - fbuf) + 1;
                    }
                    break;
                }
            }
            if ((offset < 0 && offset_offset >= 0) || (offset >= 0 && length_offset >= 0)) {
                break;
            }
        }
        /* We're at the beginning, and the negative offset indicates the exact number of lines in the file */
        if (offset < 0 && offset_offset < 0 && offset == count * -1) {
            offset_offset = 0;
        }
    }
 
    /* Positve line offset */
    if (offset > 0) {
        int64_t count = 0;
        fseek(ff, 0, SEEK_SET);
        for (i = 0; i < flength; i += sizeof(fbuf)) {
            char *pos;
            if (i + sizeof(fbuf) <= flength) {
                /* Don't let previous values influence current counts, due to short reads */
                memset(fbuf, 0, sizeof(fbuf));
            }
            if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                ast_log(LOG_ERROR, "Short read?!!\n");
                fclose(ff);
                return -1;
            }
            for (pos = fbuf; pos < fbuf + sizeof(fbuf); pos++) {
                LINE_COUNTER(pos, format, count);
 
                if (count == offset) {
                    offset_offset = i + (pos - fbuf) + 1;
                    break;
                }
            }
            if (offset_offset >= 0) {
                break;
            }
        }
    }
 
    if (offset_offset < 0) {
        ast_log(LOG_ERROR, "Offset '%s' refers to before the beginning of the file!\n", args.offset);
        fclose(ff);
        return -1;
    }
 
    ast_str_reset(*buf);
    if (fseeko(ff, offset_offset, SEEK_SET)) {
        ast_log(LOG_ERROR, "fseeko failed: %s\n", strerror(errno));
    }
 
    /* If we have both offset_offset and length_offset, then grabbing the
     * buffer is simply a matter of just retrieving the file and adding it
     * to buf.  Otherwise, we need to run byte-by-byte forward until the
     * length is complete. */
    if (length_offset >= 0) {
        ast_debug(3, "offset=%" PRId64 ", length=%" PRId64 ", offset_offset=%" PRId64 ", length_offset=%" PRId64 "\n", offset, length, offset_offset, length_offset);
        for (i = offset_offset; i < length_offset; i += sizeof(fbuf)) {
            if (fread(fbuf, 1, i + sizeof(fbuf) > flength ? flength - i : sizeof(fbuf), ff) < (i + sizeof(fbuf) > flength ? flength - i : sizeof(fbuf))) {
                ast_log(LOG_ERROR, "Short read?!!\n");
            }
            ast_debug(3, "Appending first %" PRId64" bytes of fbuf=%s\n", (int64_t)(i + sizeof(fbuf) > length_offset ? length_offset - i : sizeof(fbuf)), fbuf);
            ast_str_append_substr(buf, len, fbuf, i + sizeof(fbuf) > length_offset ? length_offset - i : sizeof(fbuf));
        }
    } else if (length == 0) {
        /* Nothing to do */
    } else {
        /* Positive line offset */
        int64_t current_length = 0;
        char dos_state = 0;
        ast_debug(3, "offset=%" PRId64 ", length=%" PRId64 ", offset_offset=%" PRId64 ", length_offset=%" PRId64 "\n", offset, length, offset_offset, length_offset);
        for (i = offset_offset; i < flength; i += sizeof(fbuf)) {
            char *pos;
            size_t bytes_read;
            if ((bytes_read = fread(fbuf, 1, sizeof(fbuf), ff)) < sizeof(fbuf) && !feof(ff)) {
                ast_log(LOG_ERROR, "Short read?!!\n");
                fclose(ff);
                return -1;
            }
            for (pos = fbuf; pos < fbuf + bytes_read; pos++) {
                LINE_COUNTER(pos, format, current_length);
 
                if (current_length == length) {
                    length_offset = i + (pos - fbuf) + 1;
                    break;
                }
            }
            ast_debug(3, "length_offset=%" PRId64 ", length_offset - i=%" PRId64 "\n", length_offset, length_offset - i);
            ast_str_append_substr(buf, len, fbuf, (length_offset >= 0) ? length_offset - i : (flength > i + sizeof(fbuf)) ? sizeof(fbuf) : flength - i);
 
            if (length_offset >= 0) {
                break;
            }
        }
    }
 
    fclose(ff);
    return 0;
}

References args, AST_APP_ARG, ast_debug, AST_DECLARE_APP_ARGS, ast_log, AST_LOG_ERROR, AST_STANDARD_APP_ARGS, ast_str_append_substr(), ast_str_reset(), buf, end, errno, FF_DOS, FF_MAC, FF_UNIX, FF_UNKNOWN, file2format(), len(), LINE_COUNTER, LLONG_MAX, LOG_ERROR, LOG_WARNING, MIN, and options.

file_write()

static int file_write ( struct ast_channel *  chan, const char *  cmd, char *  data, const char *  value  )

static

Definition at line 902 of file func_env.c.

{
    AST_DECLARE_APP_ARGS(args,
        AST_APP_ARG(filename);
        AST_APP_ARG(offset);
        AST_APP_ARG(length);
        AST_APP_ARG(options);
        AST_APP_ARG(format);
    );
    int64_t offset = 0, length = LLONG_MAX;
    off_t flength, vlength;
    size_t foplen = 0;
    FILE *ff;
 
    AST_STANDARD_APP_ARGS(args, data);
 
    if (args.argc > 1) {
        sscanf(args.offset, "%" SCNd64, &offset);
    }
    if (args.argc > 2) {
        sscanf(args.length, "%" SCNd64, &length);
    }
 
    vlength = strlen(value);
 
    if (args.argc < 4 || !strchr(args.options, 'l')) {
        /* Character-based mode */
 
        if (args.argc > 3 && strchr(args.options, 'a')) {
            /* Append mode */
            if (!(ff = fopen(args.filename, "a"))) {
                ast_log(LOG_WARNING, "Cannot open file '%s' for appending: %s\n", args.filename, strerror(errno));
                return 0;
            }
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            fclose(ff);
            return 0;
        } else if (offset == 0 && length == LLONG_MAX) {
            if (!(ff = fopen(args.filename, "w"))) {
                ast_log(LOG_WARNING, "Cannot open file '%s' for writing: %s\n", args.filename, strerror(errno));
                return 0;
            }
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            fclose(ff);
            return 0;
        }
 
        if (!(ff = fopen(args.filename, "r+"))) {
            ast_log(LOG_WARNING, "Cannot open file '%s' for modification: %s\n", args.filename, strerror(errno));
            return 0;
        }
        fseeko(ff, 0, SEEK_END);
        flength = ftello(ff);
 
        if (offset < 0) {
            if (fseeko(ff, offset, SEEK_END)) {
                ast_log(LOG_ERROR, "Cannot seek to offset of '%s': %s\n", args.filename, strerror(errno));
                fclose(ff);
                return -1;
            }
            if ((offset = ftello(ff)) < 0) {
                ast_log(AST_LOG_ERROR, "Cannot determine offset position of '%s': %s\n", args.filename, strerror(errno));
                fclose(ff);
                return -1;
            }
        }
 
        if (length < 0) {
            length = flength - offset + length;
            if (length < 0) {
                ast_log(LOG_ERROR, "Length '%s' exceeds the file length.  No data will be written.\n", args.length);
                fclose(ff);
                return -1;
            }
        }
 
        fseeko(ff, offset, SEEK_SET);
 
        ast_debug(3, "offset=%s/%" PRId64 ", length=%s/%" PRId64 ", vlength=%" PRId64 ", flength=%" PRId64 "\n",
            S_OR(args.offset, "(null)"), offset, S_OR(args.length, "(null)"), length, vlength, flength);
 
        if (length == vlength) {
            /* Simplest case, a straight replace */
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            fclose(ff);
        } else if (length == LLONG_MAX) {
            /* Simple truncation */
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            fclose(ff);
            if (truncate(args.filename, offset + vlength)) {
                ast_log(LOG_ERROR, "Unable to truncate the file: %s\n", strerror(errno));
            }
        } else if (length > vlength) {
            /* More complex -- need to close a gap */
            char fbuf[4096];
            off_t cur;
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            fseeko(ff, length - vlength, SEEK_CUR);
            while ((cur = ftello(ff)) < flength) {
                if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                    ast_log(LOG_ERROR, "Short read?!!\n");
                }
                fseeko(ff, cur + vlength - length, SEEK_SET);
                if (fwrite(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf)) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                }
                /* Seek to where we stopped reading */
                if (fseeko(ff, cur + sizeof(fbuf), SEEK_SET) < 0) {
                    /* Only reason for seek to fail is EOF */
                    break;
                }
            }
            fclose(ff);
            if (truncate(args.filename, flength - (length - vlength))) {
                ast_log(LOG_ERROR, "Unable to truncate the file: %s\n", strerror(errno));
            }
        } else {
            /* Most complex -- need to open a gap */
            char fbuf[4096];
            off_t lastwritten = flength + vlength - length;
 
            /* Start reading exactly the buffer size back from the end. */
            fseeko(ff, flength - sizeof(fbuf), SEEK_SET);
            while (offset < ftello(ff)) {
                if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf)) {
                    ast_log(LOG_ERROR, "Short read?!!\n");
                    fclose(ff);
                    return -1;
                }
                /* Since the read moved our file ptr forward, we reverse, but
                 * seek an offset equal to the amount we want to extend the
                 * file by */
                fseeko(ff, vlength - length - sizeof(fbuf), SEEK_CUR);
 
                /* Note the location of this buffer -- we must not overwrite this position. */
                lastwritten = ftello(ff);
 
                if (fwrite(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf)) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                    fclose(ff);
                    return -1;
                }
 
                if (lastwritten < offset + sizeof(fbuf)) {
                    break;
                }
                /* Our file pointer is now either pointing to the end of the
                 * file (new position) or a multiple of the fbuf size back from
                 * that point.  Move back to where we want to start reading
                 * again.  We never actually try to read beyond the end of the
                 * file, so we don't have do deal with short reads, as we would
                 * when we're shortening the file. */
                fseeko(ff, 2 * sizeof(fbuf) + vlength - length, SEEK_CUR);
            }
 
            /* Last part of the file that we need to preserve */
            if (fseeko(ff, offset + length, SEEK_SET)) {
                ast_log(LOG_WARNING, "Unable to seek to %" PRId64 " + %" PRId64 " != %" PRId64 "?)\n", offset, length, ftello(ff));
            }
 
            /* Doesn't matter how much we read -- just need to restrict the write */
            ast_debug(1, "Reading at %" PRId64 "\n", ftello(ff));
            if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                ast_log(LOG_ERROR, "Short read?!!\n");
            }
            fseek(ff, offset, SEEK_SET);
            /* Write out the value, then write just up until where we last moved some data */
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            } else {
                off_t curpos = ftello(ff);
                foplen = lastwritten - curpos;
                if (fwrite(fbuf, 1, foplen, ff) < foplen) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                }
            }
            fclose(ff);
        }
    } else {
        enum file_format newline_format = FF_UNKNOWN;
 
        /* Line mode */
        if (args.argc == 5) {
            if (tolower(args.format[0]) == 'u') {
                newline_format = FF_UNIX;
            } else if (tolower(args.format[0]) == 'm') {
                newline_format = FF_MAC;
            } else if (tolower(args.format[0]) == 'd') {
                newline_format = FF_DOS;
            }
        }
        if (newline_format == FF_UNKNOWN && (newline_format = file2format(args.filename)) == FF_UNKNOWN) {
            ast_log(LOG_ERROR, "File '%s' not in line format\n", args.filename);
            return -1;
        }
 
        if (strchr(args.options, 'a')) {
            /* Append to file */
            if (!(ff = fopen(args.filename, "a"))) {
                ast_log(LOG_ERROR, "Unable to open '%s' for appending: %s\n", args.filename, strerror(errno));
                return -1;
            }
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            } else if (!strchr(args.options, 'd') && fwrite(format2term(newline_format), 1, strlen(format2term(newline_format)), ff) < strlen(format2term(newline_format))) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            fclose(ff);
        } else if (offset == 0 && length == LLONG_MAX) {
            /* Overwrite file */
            off_t truncsize;
            if (!(ff = fopen(args.filename, "w"))) {
                ast_log(LOG_ERROR, "Unable to open '%s' for writing: %s\n", args.filename, strerror(errno));
                return -1;
            }
            if (fwrite(value, 1, vlength, ff) < vlength) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            } else if (!strchr(args.options, 'd') && fwrite(format2term(newline_format), 1, strlen(format2term(newline_format)), ff) < strlen(format2term(newline_format))) {
                ast_log(LOG_ERROR, "Short write?!!\n");
            }
            if ((truncsize = ftello(ff)) < 0) {
                ast_log(AST_LOG_ERROR, "Unable to determine truncate position of '%s': %s\n", args.filename, strerror(errno));
            }
            fclose(ff);
            if (truncsize >= 0 && truncate(args.filename, truncsize)) {
                ast_log(LOG_ERROR, "Unable to truncate file '%s': %s\n", args.filename, strerror(errno));
                return -1;
            }
        } else {
            int64_t offset_offset = (offset == 0 ? 0 : -1), length_offset = -1, flength, i, current_length = 0;
            char dos_state = 0, fbuf[4096];
 
            if (offset < 0 && length < offset) {
                /* Nonsense! */
                ast_log(LOG_ERROR, "Length cannot specify a position prior to the offset\n");
                return -1;
            }
 
            if (!(ff = fopen(args.filename, "r+"))) {
                ast_log(LOG_ERROR, "Cannot open '%s' for modification: %s\n", args.filename, strerror(errno));
                return -1;
            }
 
            if (fseek(ff, 0, SEEK_END)) {
                ast_log(LOG_ERROR, "Cannot seek to end of file '%s': %s\n", args.filename, strerror(errno));
                fclose(ff);
                return -1;
            }
            if ((flength = ftello(ff)) < 0) {
                ast_log(AST_LOG_ERROR, "Cannot determine end position of file '%s': %s\n", args.filename, strerror(errno));
                fclose(ff);
                return -1;
            }
 
            /* For negative offset and/or negative length */
            if (offset < 0 || length < 0) {
                int64_t count = 0;
                for (i = (flength / sizeof(fbuf)) * sizeof(fbuf); i >= 0; i -= sizeof(fbuf)) {
                    char *pos;
                    if (fseeko(ff, i, SEEK_SET)) {
                        ast_log(LOG_ERROR, "Cannot seek to offset %" PRId64 ": %s\n", i, strerror(errno));
                    }
                    if (i + sizeof(fbuf) >= flength) {
                        memset(fbuf, 0, sizeof(fbuf));
                    }
                    if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                        ast_log(LOG_ERROR, "Short read: %s\n", strerror(errno));
                        fclose(ff);
                        return -1;
                    }
                    for (pos = fbuf + sizeof(fbuf) - 1; pos >= fbuf; pos--) {
                        LINE_COUNTER(pos, newline_format, count);
 
                        if (length < 0 && count * -1 == length) {
                            length_offset = i + (pos - fbuf);
                        } else if (offset < 0 && count * -1 == (offset - 1)) {
                            /* Found our initial offset.  We're done with reverse motion! */
                            if (newline_format == FF_DOS) {
                                offset_offset = i + (pos - fbuf) + 2;
                            } else {
                                offset_offset = i + (pos - fbuf) + 1;
                            }
                            break;
                        }
                    }
                    if ((offset < 0 && offset_offset >= 0) || (offset >= 0 && length_offset >= 0)) {
                        break;
                    }
                }
                /* We're at the beginning, and the negative offset indicates the exact number of lines in the file */
                if (offset < 0 && offset_offset < 0 && offset == count * -1) {
                    offset_offset = 0;
                }
            }
 
            /* Positve line offset */
            if (offset > 0) {
                int64_t count = 0;
                fseek(ff, 0, SEEK_SET);
                for (i = 0; i < flength; i += sizeof(fbuf)) {
                    char *pos;
                    if (i + sizeof(fbuf) >= flength) {
                        memset(fbuf, 0, sizeof(fbuf));
                    }
                    if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                        ast_log(LOG_ERROR, "Short read?!!\n");
                        fclose(ff);
                        return -1;
                    }
                    for (pos = fbuf; pos < fbuf + sizeof(fbuf); pos++) {
                        LINE_COUNTER(pos, newline_format, count);
 
                        if (count == offset) {
                            offset_offset = i + (pos - fbuf) + 1;
                            break;
                        }
                    }
                    if (offset_offset >= 0) {
                        break;
                    }
                }
            }
 
            if (offset_offset < 0) {
                ast_log(LOG_ERROR, "Offset '%s' refers to before the beginning of the file!\n", args.offset);
                fclose(ff);
                return -1;
            }
 
            if (length == 0) {
                length_offset = offset_offset;
            } else if (length == LLONG_MAX) {
                length_offset = flength;
            }
 
            /* Positive line length */
            if (length_offset < 0) {
                fseeko(ff, offset_offset, SEEK_SET);
                for (i = offset_offset; i < flength; i += sizeof(fbuf)) {
                    char *pos;
                    if (i + sizeof(fbuf) >= flength) {
                        memset(fbuf, 0, sizeof(fbuf));
                    }
                    if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                        ast_log(LOG_ERROR, "Short read?!!\n");
                        fclose(ff);
                        return -1;
                    }
                    for (pos = fbuf; pos < fbuf + sizeof(fbuf); pos++) {
                        LINE_COUNTER(pos, newline_format, current_length);
 
                        if (current_length == length) {
                            length_offset = i + (pos - fbuf) + 1;
                            break;
                        }
                    }
                    if (length_offset >= 0) {
                        break;
                    }
                }
                if (length_offset < 0) {
                    /* Exceeds length of file */
                    ast_debug(3, "Exceeds length of file? length=%" PRId64 ", count=%" PRId64 ", flength=%" PRId64 "\n", length, current_length, flength);
                    length_offset = flength;
                }
            }
 
            /* Have offset_offset and length_offset now */
            if (length_offset - offset_offset == vlength + (strchr(args.options, 'd') ? 0 : strlen(format2term(newline_format)))) {
                /* Simple case - replacement of text inline */
                fseeko(ff, offset_offset, SEEK_SET);
                if (fwrite(value, 1, vlength, ff) < vlength) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                } else if (!strchr(args.options, 'd') && fwrite(format2term(newline_format), 1, strlen(format2term(newline_format)), ff) < strlen(format2term(newline_format))) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                }
                fclose(ff);
            } else if (length_offset - offset_offset > vlength + (strchr(args.options, 'd') ? 0 : strlen(format2term(newline_format)))) {
                /* More complex case - need to shorten file */
                off_t cur;
                int64_t length_length = length_offset - offset_offset;
                size_t vlen = vlength + (strchr(args.options, 'd') ? 0 : strlen(format2term(newline_format)));
 
                ast_debug(3, "offset=%s/%" PRId64 ", length=%s/%" PRId64 " (%" PRId64 "), vlength=%" PRId64 ", flength=%" PRId64 "\n",
                    args.offset, offset_offset, args.length, length_offset, length_length, vlength, flength);
 
                fseeko(ff, offset_offset, SEEK_SET);
                if (fwrite(value, 1, vlength, ff) < vlength) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                    fclose(ff);
                    return -1;
                } else if (!strchr(args.options, 'd') && fwrite(format2term(newline_format), 1, vlen - vlength, ff) < vlen - vlength) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                    fclose(ff);
                    return -1;
                }
                while ((cur = ftello(ff)) < flength) {
                    if (cur < 0) {
                        ast_log(AST_LOG_ERROR, "Unable to determine last write position for '%s': %s\n", args.filename, strerror(errno));
                        fclose(ff);
                        return -1;
                    }
                    fseeko(ff, length_length - vlen, SEEK_CUR);
                    if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                        ast_log(LOG_ERROR, "Short read?!!\n");
                        fclose(ff);
                        return -1;
                    }
                    /* Seek to where we last stopped writing */
                    fseeko(ff, cur, SEEK_SET);
                    if (fwrite(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf)) {
                        ast_log(LOG_ERROR, "Short write?!!\n");
                        fclose(ff);
                        return -1;
                    }
                }
                fclose(ff);
                if (truncate(args.filename, flength - (length_length - vlen))) {
                    ast_log(LOG_ERROR, "Truncation of file failed: %s\n", strerror(errno));
                }
            } else {
                /* Most complex case - need to lengthen file */
                size_t vlen = vlength + (strchr(args.options, 'd') ? 0 : strlen(format2term(newline_format)));
                int64_t origlen = length_offset - offset_offset;
                off_t lastwritten = flength + vlen - origlen;
 
                ast_debug(3, "offset=%s/%" PRId64 ", length=%s/%" PRId64 ", vlength=%" PRId64 ", flength=%" PRId64 "\n",
                    args.offset, offset_offset, args.length, length_offset, vlength, flength);
 
                fseeko(ff, flength - sizeof(fbuf), SEEK_SET);
                while (offset_offset + sizeof(fbuf) < ftello(ff)) {
                    if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf)) {
                        ast_log(LOG_ERROR, "Short read?!!\n");
                        fclose(ff);
                        return -1;
                    }
                    fseeko(ff, sizeof(fbuf) - vlen - origlen, SEEK_CUR);
                    if (fwrite(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf)) {
                        ast_log(LOG_ERROR, "Short write?!!\n");
                        fclose(ff);
                        return -1;
                    }
                    if ((lastwritten = ftello(ff) - sizeof(fbuf)) < offset_offset + sizeof(fbuf)) {
                        break;
                    }
                    fseeko(ff, 2 * sizeof(fbuf) + vlen - origlen, SEEK_CUR);
                }
                fseek(ff, length_offset, SEEK_SET);
                if (fread(fbuf, 1, sizeof(fbuf), ff) < sizeof(fbuf) && !feof(ff)) {
                    ast_log(LOG_ERROR, "Short read?!!\n");
                    fclose(ff);
                    return -1;
                }
                fseek(ff, offset_offset, SEEK_SET);
                if (fwrite(value, 1, vlength, ff) < vlength) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                    fclose(ff);
                    return -1;
                } else if (!strchr(args.options, 'd') && fwrite(format2term(newline_format), 1, strlen(format2term(newline_format)), ff) < strlen(format2term(newline_format))) {
                    ast_log(LOG_ERROR, "Short write?!!\n");
                    fclose(ff);
                    return -1;
                } else {
                    off_t curpos = ftello(ff);
                    foplen = lastwritten - curpos;
                    if (fwrite(fbuf, 1, foplen, ff) < foplen) {
                        ast_log(LOG_ERROR, "Short write?!!\n");
                    }
                }
                fclose(ff);
            }
        }
    }
 
    return 0;
}

References args, AST_APP_ARG, ast_debug, AST_DECLARE_APP_ARGS, ast_log, AST_LOG_ERROR, AST_STANDARD_APP_ARGS, errno, FF_DOS, FF_MAC, FF_UNIX, FF_UNKNOWN, file2format(), format2term(), LINE_COUNTER, LLONG_MAX, LOG_ERROR, LOG_WARNING, options, S_OR, and value.

format2term()

const char * format2term

(

enum file_format

f

)

Definition at line 896 of file func_env.c.

{
    const char *term[] = { "", "\n", "\r\n", "\r" };
    return term[f + 1];
}

Referenced by file_write().

load_module()

static int load_module ( void  )

static

Definition at line 1446 of file func_env.c.

{
    int res = 0;
 
    res |= ast_custom_function_register(&env_function);
    res |= ast_custom_function_register_escalating(&stat_function, AST_CFE_READ);
    res |= ast_custom_function_register_escalating(&file_function, AST_CFE_BOTH);
    res |= ast_custom_function_register_escalating(&file_count_line_function, AST_CFE_READ);
    res |= ast_custom_function_register_escalating(&file_format_function, AST_CFE_READ);
    res |= ast_custom_function_register(&file_dirname_function);
    res |= ast_custom_function_register(&file_basename_function);
 
    return res;
}

References AST_CFE_BOTH, AST_CFE_READ, ast_custom_function_register, ast_custom_function_register_escalating, env_function, file_basename_function, file_count_line_function, file_dirname_function, file_format_function, file_function, and stat_function.

stat_read()

static int stat_read ( struct ast_channel *  chan, const char *  cmd, char *  data, char *  buf, size_t  len  )

static

Definition at line 397 of file func_env.c.

{
    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, "%u", (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", (unsigned int) s.st_mode);
            break;
        }
    }
 
    return 0;
}

References ast_copy_string(), buf, len(), and strsep().

unload_module()

static int unload_module ( void  )

static

Definition at line 1431 of file func_env.c.

{
    int res = 0;
 
    res |= ast_custom_function_unregister(&env_function);
    res |= ast_custom_function_unregister(&stat_function);
    res |= ast_custom_function_unregister(&file_function);
    res |= ast_custom_function_unregister(&file_count_line_function);
    res |= ast_custom_function_unregister(&file_format_function);
    res |= ast_custom_function_unregister(&file_dirname_function);
    res |= ast_custom_function_unregister(&file_basename_function);
 
    return res;
}

References ast_custom_function_unregister(), env_function, file_basename_function, file_count_line_function, file_dirname_function, file_format_function, file_function, and stat_function.

Variable Documentation

__mod_info

struct ast_module_info __mod_info = { .name = AST_MODULE, .flags = AST_MODFLAG_LOAD_ORDER , .description = "Environment/filesystem dialplan functions" , .key = "This paragraph is copyright (c) 2006 by Digium, Inc. \In order for your module to load, it must return this \key via a function called \"key\". Any code which \includes this paragraph must be licensed under the GNU \General Public License version 2 or later (at your \option). In addition to Digium's general reservations \of rights, Digium expressly reserves the right to \allow other parties to license this paragraph under \different terms. Any use of Digium, Inc. trademarks or \logos (including \"Asterisk\" or \"Digium\") without \express written permission of Digium, Inc. is prohibited.\n" , .buildopt_sum = AST_BUILDOPT_SUM, .load = load_module, .unload = unload_module, .load_pri = AST_MODPRI_DEFAULT, .support_level = AST_MODULE_SUPPORT_CORE, }

static

Definition at line 1461 of file func_env.c.

ast_module_info

const struct ast_module_info* ast_module_info = &__mod_info

static

Definition at line 1461 of file func_env.c.

env_function

struct ast_custom_function env_function

static

Initial value:

= {
    .name = "ENV",
    .read = env_read,
    .write = env_write
}

Definition at line 1389 of file func_env.c.

Referenced by load_module(), and unload_module().

file_basename_function

struct ast_custom_function file_basename_function

static

Initial value:

= {
    .name = "BASENAME",
    .read = file_basename,
    .read_max = 12,
}

Definition at line 1425 of file func_env.c.

Referenced by load_module(), and unload_module().

file_count_line_function

struct ast_custom_function file_count_line_function

static

Initial value:

= {
    .name = "FILE_COUNT_LINE",
    .read2 = file_count_line,
    .read_max = 12,
}

Definition at line 1407 of file func_env.c.

Referenced by load_module(), and unload_module().

file_dirname_function

struct ast_custom_function file_dirname_function

static

Initial value:

= {
    .name = "DIRNAME",
    .read = file_dirname,
    .read_max = 12,
}

Definition at line 1419 of file func_env.c.

Referenced by load_module(), and unload_module().

file_format_function

struct ast_custom_function file_format_function

static

Initial value:

= {
    .name = "FILE_FORMAT",
    .read2 = file_format,
    .read_max = 2,
}

Definition at line 1413 of file func_env.c.

Referenced by load_module(), and unload_module().

file_function

struct ast_custom_function file_function

static

Initial value:

= {
    .name = "FILE",
    .read2 = file_read,
    .write = file_write,
}

Definition at line 1401 of file func_env.c.

Referenced by load_module(), and unload_module().

stat_function

struct ast_custom_function stat_function

static

Initial value:

= {
    .name = "STAT",
    .read = stat_read,
    .read_max = 12,
}

Definition at line 1395 of file func_env.c.

Referenced by load_module(), and unload_module().