funcs/func_db.c

   1 /*
   2  * Asterisk -- A telephony toolkit for Linux.
   3  *
   4  * Functions for interaction with the Asterisk database
   5  *
   6  * Copyright (C) 2005, Russell Bryant <russelb@clemson.edu>
   7  *
   8  * func_db.c adapted from the old app_db.c, copyright by the following people
   9  * Copyright (C) 2005, Mark Spencer <markster@digium.com>
  10  * Copyright (C) 2003, Jefferson Noxon <jeff@debian.org>
  11  *
  12  * This program is free software, distributed under the terms of
  13  * the GNU General Public License
  14  */
  15
  16 #include <stdlib.h>
  17 #include <string.h>
  18 #include <sys/types.h>
  19 #include <regex.h>
  20
  21 #include "asterisk/channel.h"
  22 #include "asterisk/pbx.h"
  23 #include "asterisk/logger.h"
  24 #include "asterisk/options.h"
  25 #include "asterisk/utils.h"
  26 #include "asterisk/app.h"
  27 #include "asterisk/astdb.h"
  28
  29 static char *function_db_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len)
  30 {
  31         int argc;
  32         char *args;
  33         char *argv[2];
  34         char *family;
  35         char *key;
  36
  37         if (!data || ast_strlen_zero(data)) {
  38                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)\n");
  39                 return buf;
  40         }
  41
  42         args = ast_strdupa(data);
  43         argc = ast_separate_app_args(args, '/', argv, sizeof(argv) / sizeof(argv[0]));
  44
  45         if (argc > 1) {
  46                 family = argv[0];
  47                 key = argv[1];
  48         } else {
  49                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)\n");
  50                 return buf;
  51         }
  52
  53         if (ast_db_get(family, key, buf, len-1)) {
  54                 ast_log(LOG_WARNING, "DB: %s/%s not found in database.\n", family, key);
  55         }
  56
  57         return buf;
  58 }
  59
  60 static void function_db_write(struct ast_channel *chan, char *cmd, char *data, const char *value)
  61 {
  62         int argc;
  63         char *args;
  64         char *argv[2];
  65         char *family;
  66         char *key;
  67
  68         if (!data || ast_strlen_zero(data)) {
  69                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)=<value>\n");
  70                 return;
  71         }
  72
  73         args = ast_strdupa(data);
  74         argc = ast_separate_app_args(args, '/', argv, sizeof(argv) / sizeof(argv[0]));
  75
  76         if (argc > 1) {
  77                 family = argv[0];
  78                 key = argv[1];
  79         } else {
  80                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)=value\n");
  81                 return;
  82         }
  83
  84         if (ast_db_put(family, key, (char*)value)) {
  85                 ast_log(LOG_WARNING, "DB: Error writing value to database.\n");
  86         }
  87 }
  88
  89 #ifndef BUILTIN_FUNC
  90 static
  91 #endif
  92 struct ast_custom_function db_function = {
  93         .name = "DB",
  94         .synopsis = "Read or Write from/to the Asterisk database",
  95         .syntax = "DB(<family>/<key>)",
  96         .desc = "This function will read or write a value from/to the Asterisk database."
  97                 "DB(...) will read a value from the database, while DB(...)=value"
  98                 "will write a value to the database.  On a read, this function"
  99                 "returns the value from the datase, or NULL if it does not exist."
 100                 "On a write, this function will always return NULL.",
 101         .read = function_db_read,
 102         .write = function_db_write,
 103 };
 104
 105 static char *function_db_exists(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len)
 106 {
 107         int argc;
 108         char *args;
 109         char *argv[2];
 110         char *family;
 111         char *key;
 112
 113         if (!data || ast_strlen_zero(data)) {
 114                 ast_log(LOG_WARNING, "DB_EXISTS requires an argument, DB(<family>/<key>)\n");
 115                 return buf;
 116         }
 117
 118         args = ast_strdupa(data);
 119         argc = ast_separate_app_args(args, '/', argv, sizeof(argv) / sizeof(argv[0]));
 120
 121         if (argc > 1) {
 122                 family = argv[0];
 123                 key = argv[1];
 124         } else {
 125                 ast_log(LOG_WARNING, "DB_EXISTS requires an argument, DB(<family>/<key>)\n");
 126                 return buf;
 127         }
 128
 129         if (ast_db_get(family, key, buf, len-1))
 130                 ast_copy_string(buf, "0", len);
 131         else
 132                 ast_copy_string(buf, "1", len);
 133
 134         return buf;
 135 }
 136
 137 #ifndef BUILTIN_FUNC
 138 static
 139 #endif
 140 struct ast_custom_function db_exists_function = {
 141         .name = "DB_EXISTS",
 142         .synopsis = "Check to see if a key exists in the Asterisk database",
 143         .syntax = "DB_EXISTS(<family>/<key>)",
 144         .desc = "This function will check to see if a key exists in the Asterisk\n"
 145                 "database. If it exists, the function will return \"1\". If not,\n"
 146                 "it will return \"0\".",
 147         .read = function_db_exists,
 148 };