Version 0.1.10 from FTP

git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@373 65c4cc65-6c06-0410-ace0-fbb531ad65f3

cdr/cdr_csv.c

@@ -0,0 +1,253 @@
+/*
+ * Asterisk -- A telephony toolkit for Linux.
+ *
+ * Comma Separated Value CDR records.
+ * 
+ * Copyright (C) 1999, Mark Spencer
+ *
+ * Mark Spencer <markster@linux-support.net>
+ *
+ * This program is free software, distributed under the terms of
+ * the GNU General Public License.
+ *
+ * Includes code and algorithms from the Zapata library.
+ *
+ */
+
+#include <asterisk/channel.h>
+#include <asterisk/cdr.h>
+#include <asterisk/module.h>
+#include <asterisk/logger.h>
+#include "../asterisk.h"
+
+#define CSV_LOG_DIR AST_LOG_DIR "/cdr-csv"
+#define CSV_MASTER CSV_LOG_DIR "/Master.csv"
+
+#define DATE_FORMAT "%Y-%m-%d %T"
+
+#include <stdio.h>
+#include <string.h>
+
+#include <stdlib.h>
+#include <unistd.h>
+#include <time.h>
+
+/* The values are as follows:
+
+
+  "accountcode", 
+  "source",
+  "destination",
+  "destination context", 
+  "callerid",
+  "channel",
+  "destination channel",	(if applicable)
+  "last application",
+  "last app argument",
+  "start time", 
+  "answer time", 
+  "end time", 
+  duration, 
+  billable seconds, 
+  "disposition", 
+  "amaflags",
+
+*/
+
+static char *desc = "Comma Separated Values CDR Backend";
+
+static char *name = "csv";
+
+static FILE *mf = NULL;
+
+static int append_string(char *buf, char *s, int len)
+{
+	int pos = strlen(buf);
+	int spos = 0;
+	int error = 0;
+	if (pos >= len - 4)
+		return -1;
+	buf[pos++] = '\"';
+	error = -1;
+	while(pos < len - 3) {
+		if (!s[spos]) {
+			error = 0;
+			break;
+		}
+		if (s[spos] == '\"')
+			buf[pos++] = '\"';
+		buf[pos++] = s[spos];
+		spos++;
+	}
+	buf[pos++] = '\"';
+	buf[pos++] = ',';
+	buf[pos++] = '\0';
+	return error;
+}
+
+static int append_int(char *buf, int s, int len)
+{
+	char tmp[32];
+	int pos = strlen(buf);
+	snprintf(tmp, sizeof(tmp), "%d", s);
+	if (pos + strlen(tmp) > len - 3)
+		return -1;
+	strncat(buf, tmp, len);
+	pos = strlen(buf);
+	buf[pos++] = ',';
+	buf[pos++] = '\0';
+	return 0;
+}
+
+static int append_date(char *buf, struct timeval tv, int len)
+{
+	char tmp[80];
+	struct tm *tm;
+	time_t t;
+	t = tv.tv_sec;
+	if (strlen(buf) > len - 3)
+		return -1;
+	if (!tv.tv_sec && !tv.tv_usec) {
+		strncat(buf, ",", len);
+		return 0;
+	}
+	tm = localtime(&t);
+	strftime(tmp, sizeof(tmp), DATE_FORMAT, tm);
+	return append_string(buf, tmp, len);
+}
+
+static int build_csv_record(char *buf, int len, struct ast_cdr *cdr)
+{
+	buf[0] = '\0';
+	/* Account code */
+	append_string(buf, cdr->accountcode, len);
+	/* Source */
+	append_string(buf, cdr->src, len);
+	/* Destination */
+	append_string(buf, cdr->dst, len);
+	/* Destination context */
+	append_string(buf, cdr->dcontext, len);
+	/* Caller*ID */
+	append_string(buf, cdr->clid, len);
+	/* Channel */
+	append_string(buf, cdr->channel, len);
+	/* Destination Channel */
+	append_string(buf, cdr->dstchannel, len);
+	/* Last Application */
+	append_string(buf, cdr->lastapp, len);
+	/* Last Data */
+	append_string(buf, cdr->lastdata, len);
+	/* Start Time */
+	append_date(buf, cdr->start, len);
+	/* Answer Time */
+	append_date(buf, cdr->answer, len);
+	/* End Time */
+	append_date(buf, cdr->end, len);
+	/* Duration */
+	append_int(buf, cdr->duration, len);
+	/* Billable seconds */
+	append_int(buf, cdr->billsec, len);
+	/* Disposition */
+	append_string(buf, ast_cdr_disp2str(cdr->disposition), len);
+	/* AMA Flags */
+	append_string(buf, ast_cdr_flags2str(cdr->amaflags), len);
+
+	/* If we hit the end of our buffer, log an error */
+	if (strlen(buf) < len - 5) {
+		/* Trim off trailing comma */
+		buf[strlen(buf) - 1] = '\0';
+		strncat(buf, "\n", len);
+		return 0;
+	}
+	return -1;
+}
+
+static int writefile(char *s, char *acc)
+{
+	char tmp[256];
+	FILE *f;
+	if (strchr(acc, '/') || (acc[0] == '.')) {
+		ast_log(LOG_WARNING, "Account code '%s' insecure for writing file\n", acc);
+		return -1;
+	}
+	snprintf(tmp, sizeof(tmp), "%s/%s.csv", CSV_LOG_DIR, acc);
+	f = fopen(tmp, "a");
+	if (!f)
+		return -1;
+	fputs(s, f);
+	fclose(f);
+	return 0;
+}
+
+
+static int csv_log(struct ast_cdr *cdr)
+{
+	/* Make sure we have a big enough buf */
+	char buf[1024];
+#if 0
+	printf("[CDR] %s ('%s' -> '%s') Dur: %ds Bill: %ds Disp: %s Flags: %s Account: [%s]\n", cdr->channel, cdr->src, cdr->dst, cdr->duration, cdr->billsec, ast_cdr_disp2str(cdr->disposition), ast_cdr_flags2str(cdr->amaflags), cdr->accountcode);
+#endif
+	if (build_csv_record(buf, sizeof(buf), cdr)) {
+		ast_log(LOG_WARNING, "Unable to create CSV record in %d bytes.  CDR not recorded!\n", sizeof(buf));
+	} else {
+		/* because of the absolutely unconditional need for the
+		   highest reliability possible in writing billing records,
+		   we open write and close the log file each time */
+		mf = fopen(CSV_MASTER, "a");
+		if (!mf) {
+			ast_log(LOG_ERROR, "Unable to re-open master file %s\n", CSV_MASTER);
+		}
+		if (mf) {
+			fputs(buf, mf);
+			fflush(mf); /* be particularly anal here */
+			fclose(mf);
+			mf = NULL;
+		}
+		if (strlen(cdr->accountcode)) {
+			if (writefile(buf, cdr->accountcode))
+				ast_log(LOG_WARNING, "Unable to write CSV record to account file '%s'\n", cdr->accountcode);
+		}
+	}
+	return 0;
+}
+
+char *description(void)
+{
+	return desc;
+}
+
+int unload_module(void)
+{
+	if (mf)
+		fclose(mf);
+	ast_cdr_unregister(name);
+	return 0;
+}
+
+int load_module(void)
+{
+	int res;
+
+	res = ast_cdr_register(name, desc, csv_log);
+	if (res) {
+		ast_log(LOG_ERROR, "Unable to register CSV CDR handling\n");
+		if (mf)
+			fclose(mf);
+	}
+	return res;
+}
+
+int reload(void)
+{
+	return 0;
+}
+
+int usecount(void)
+{
+	return 0;
+}
+
+char *key()
+{
+	return ASTERISK_GPL_KEY;
+}