Fix load of cdr_custom
[asterisk/asterisk.git] / cdr / cdr_custom.c
1 /*
2  * Asterisk -- A telephony toolkit for Linux.
3  *
4  * Comma Separated Value CDR records.
5  * 
6  * Copyright (C) 1999 - 2005, Digium, inc 
7  *
8  * Mark Spencer <markster@digium.com>
9  *
10  * This program is free software, distributed under the terms of
11  * the GNU General Public License.
12  *
13  * Includes code and algorithms from the Zapata library.
14  *
15  */
16
17 #include <sys/types.h>
18 #include <asterisk/channel.h>
19 #include <asterisk/cdr.h>
20 #include <asterisk/module.h>
21 #include <asterisk/config.h>
22 #include <asterisk/pbx.h>
23 #include <asterisk/logger.h>
24 #include <asterisk/utils.h>
25 #include "../asterisk.h"
26 #include "../astconf.h"
27
28 #define CUSTOM_LOG_DIR "/cdr_custom"
29
30 #define DATE_FORMAT "%Y-%m-%d %T"
31
32 #include <stdio.h>
33 #include <string.h>
34 #include <errno.h>
35
36 #include <stdlib.h>
37 #include <unistd.h>
38 #include <time.h>
39
40 AST_MUTEX_DEFINE_STATIC(lock);
41
42 static char *desc = "Customizable Comma Separated Values CDR Backend";
43
44 static char *name = "cdr-custom";
45
46 static FILE *mf = NULL;
47
48 static char master[AST_CONFIG_MAX_PATH];
49 static char format[1024]="";
50
51 static int load_config(int reload) 
52 {
53         struct ast_config *cfg;
54         struct ast_variable *var;
55         int res = -1;
56
57         strcpy(format, "");
58         strcpy(master, "");
59         if((cfg = ast_config_load("cdr_custom.conf"))) {
60                 var = ast_variable_browse(cfg, "mappings");
61                 while(var) {
62                         ast_mutex_lock(&lock);
63                         if (!ast_strlen_zero(var->name) && !ast_strlen_zero(var->value)) {
64                                 strncpy(format, var->value, sizeof(format) - 2);
65                                 strcat(format,"\n");
66                                 snprintf(master, sizeof(master),"%s/%s/%s", ast_config_AST_LOG_DIR, name, var->name);
67                                 ast_mutex_unlock(&lock);
68                         } else
69                                 ast_log(LOG_NOTICE, "Mapping must have both filename and format at line %d\n", var->lineno);
70                         if (var->next)
71                                 ast_log(LOG_NOTICE, "Sorry, only one mapping is supported at this time, mapping '%s' will be ignored at line %d.\n", var->next->name, var->next->lineno); 
72                         var = var->next;
73                 }
74                 ast_config_destroy(cfg);
75         }
76         
77         return res;
78 }
79
80
81
82 static int custom_log(struct ast_cdr *cdr)
83 {
84         /* Make sure we have a big enough buf */
85         char buf[2048];
86         struct ast_channel dummy;
87
88         /* Abort if no master file is specified */
89         if (ast_strlen_zero(master))
90                 return 0;
91
92         memset(buf, 0 , sizeof(buf));
93         /* Quite possibly the first use of a static struct ast_channel, we need it so the var funcs will work */
94         memset(&dummy, 0, sizeof(dummy));
95         dummy.cdr = cdr;
96         pbx_substitute_variables_helper(&dummy, format, buf, sizeof(buf) - 1);
97
98         /* because of the absolutely unconditional need for the
99            highest reliability possible in writing billing records,
100            we open write and close the log file each time */
101         mf = fopen(master, "a");
102         if (!mf) {
103                 ast_log(LOG_ERROR, "Unable to re-open master file %s : %s\n", master, strerror(errno));
104         }
105         if (mf) {
106                 fputs(buf, mf);
107                 fflush(mf); /* be particularly anal here */
108                 fclose(mf);
109                 mf = NULL;
110         }
111         return 0;
112 }
113
114 char *description(void)
115 {
116         return desc;
117 }
118
119 int unload_module(void)
120 {
121         if (mf)
122                 fclose(mf);
123         ast_cdr_unregister(name);
124         return 0;
125 }
126
127 int load_module(void)
128 {
129         int res = 0;
130
131         if (!load_config(0)) {
132                 res = ast_cdr_register(name, desc, custom_log);
133                 if (res)
134                         ast_log(LOG_ERROR, "Unable to register custom CDR handling\n");
135                 if (mf)
136                         fclose(mf);
137         }
138         return res;
139 }
140
141 int reload(void)
142 {
143         return load_config(1);
144 }
145
146 int usecount(void)
147 {
148         return 0;
149 }
150
151 char *key()
152 {
153         return ASTERISK_GPL_KEY;
154 }