add support for Zaptel transcoders
authorKevin P. Fleming <kpfleming@digium.com>
Fri, 23 Jun 2006 16:33:17 +0000 (16:33 +0000)
committerKevin P. Fleming <kpfleming@digium.com>
Fri, 23 Jun 2006 16:33:17 +0000 (16:33 +0000)
git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@35745 65c4cc65-6c06-0410-ace0-fbb531ad65f3

codecs/codec_zap.c [new file with mode: 0644]

diff --git a/codecs/codec_zap.c b/codecs/codec_zap.c
new file mode 100644 (file)
index 0000000..cad97c4
--- /dev/null
@@ -0,0 +1,406 @@
+/*
+ * Asterisk -- An open source telephony toolkit.
+ *
+ * Zaptel native transcoding support
+ *
+ * Copyright (C) 1999 - 2006, Digium, Inc.
+ *
+ * Mark Spencer <markster@digium.com>
+ * Kevin P. Fleming <kpfleming@digium.com>
+ *
+ * See http://www.asterisk.org for more information about
+ * the Asterisk project. Please do not directly contact
+ * any of the maintainers of this project for assistance;
+ * the project provides a web site, mailing lists and IRC
+ * channels for your use.
+ *
+ * This program is free software, distributed under the terms of
+ * the GNU General Public License Version 2. See the LICENSE file
+ * at the top of the source tree.
+ */
+
+/*! \file
+ *
+ * \brief Translate between various formats natively through Zaptel transcoding
+ *
+ * \ingroup codecs
+ */
+
+/*** MODULEINFO
+       <depend>zaptel</depend>
+ ***/
+
+#include "asterisk.h"
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include <fcntl.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <netinet/in.h>
+#include <string.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <errno.h>
+#include <sys/mman.h>
+#include <linux/zaptel.h>
+
+#include "asterisk/lock.h"
+#include "asterisk/translate.h"
+#include "asterisk/config.h"
+#include "asterisk/options.h"
+#include "asterisk/module.h"
+#include "asterisk/logger.h"
+#include "asterisk/channel.h"
+#include "asterisk/utils.h"
+#include "asterisk/linkedlists.h"
+
+#define BUFFER_SAMPLES 8000
+
+static unsigned int global_useplc = 0;
+
+struct format_map {
+       unsigned int map[32][32];
+};
+
+static struct format_map global_format_map = { { { 0 } } };
+
+struct translator {
+       struct ast_translator t;
+       AST_LIST_ENTRY(translator) entry;
+};
+
+static AST_LIST_HEAD_STATIC(translators, translator);
+
+struct pvt {
+       int fd;
+       int fake;
+       struct zt_transcode_header *hdr;
+       struct ast_frame f;
+};
+
+static int zap_framein(struct ast_trans_pvt *pvt, struct ast_frame *f)
+{
+       struct pvt *ztp = pvt->pvt;
+       struct zt_transcode_header *hdr = ztp->hdr;
+
+       if (!f->subclass) {
+               /* Fake a return frame for calculation purposes */
+               ztp->fake = 2;
+               pvt->samples = f->samples;
+               return 0;
+       }
+
+       if (!hdr->srclen)
+               /* Copy at front of buffer */
+               hdr->srcoffset = 0;
+
+       if (hdr->srclen + f->datalen > sizeof(hdr->srcdata)) {
+               ast_log(LOG_WARNING, "Out of space for codec translation!\n");
+               return -1;
+       }
+
+       if (hdr->srclen + f->datalen + hdr->srcoffset > sizeof(hdr->srcdata)) {
+               /* Very unlikely */
+               memmove(hdr->srcdata, hdr->srcdata + hdr->srcoffset, hdr->srclen);
+               hdr->srcoffset = 0;
+       }
+
+       memcpy(hdr->srcdata + hdr->srcoffset, f->data, f->datalen);
+       hdr->srclen += f->datalen;
+       pvt->samples += f->samples;
+
+       return -1;
+}
+
+static struct ast_frame *zap_frameout(struct ast_trans_pvt *pvt)
+{
+       struct pvt *ztp = pvt->pvt;
+       struct zt_transcode_header *hdr = ztp->hdr;
+       unsigned int x;
+
+       if (ztp->fake == 2) {
+               ztp->fake = 1;
+               ztp->f.frametype = AST_FRAME_VOICE;
+               ztp->f.subclass = 0;
+               ztp->f.samples = 160;
+               ztp->f.data = NULL;
+               ztp->f.offset = 0;
+               ztp->f.datalen = 0;
+               ztp->f.mallocd = 0;
+               pvt->samples = 0;
+       } else if (ztp->fake == 1) {
+               return NULL;
+       } else if (!hdr->srclen) {
+               return NULL;
+       } else {
+               hdr->dstoffset = 0;
+               hdr->dstlen = 0;
+               x = ZT_TCOP_TRANSCODE;
+               if (ioctl(ztp->fd, ZT_TRANSCODE_OP, &x))
+                       ast_log(LOG_WARNING, "Failed to transcode: %s\n", strerror(errno));
+               if (!hdr->dstlen)
+                       return NULL;
+               ztp->f.frametype = AST_FRAME_VOICE;
+               ztp->f.subclass = hdr->dstfmt;
+               ztp->f.samples = hdr->dstsamples;
+               ztp->f.data = hdr->dstdata + hdr->dstoffset;
+               ztp->f.offset = hdr->dstoffset;
+               ztp->f.datalen = hdr->dstlen;
+               ztp->f.mallocd = 0;
+               pvt->samples -= ztp->f.samples;
+       }
+
+       return &ztp->f;
+}
+
+static void zap_destroy(struct ast_trans_pvt *pvt)
+{
+       struct pvt *ztp = pvt->pvt;
+
+       munmap(ztp->hdr, sizeof(*ztp->hdr));
+       close(ztp->fd);
+}
+
+static struct ast_trans_pvt *zap_translate(struct ast_trans_pvt *pvt, int dest, int source)
+{
+       /* Request translation through zap if possible */
+       int fd;
+       unsigned int x = ZT_TCOP_RESET;
+       struct pvt *ztp = pvt->pvt;
+       struct zt_transcode_header *hdr;
+       
+       if ((fd = open("/dev/zap/transcode", O_RDWR)) < 0)
+               return NULL;
+
+       if ((hdr = mmap(NULL, sizeof(*hdr), PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0)) == MAP_FAILED) {
+               ast_log(LOG_ERROR, "Memory Map failed for transcoding (%s)\n", strerror(errno));
+               close(fd);
+
+               return NULL;
+       }
+
+       if (hdr->magic != ZT_TRANSCODE_MAGIC) {
+               ast_log(LOG_ERROR, "Transcoder header (%08x) wasn't magic.  Abandoning\n", hdr->magic);
+               munmap(hdr, sizeof(*hdr));
+               close(fd);
+
+               return NULL;
+       }
+       
+       hdr->srcfmt = (1 << source);
+       hdr->dstfmt = (1 << dest);
+       if (ioctl(fd, ZT_TRANSCODE_OP, &x)) {
+               ast_log(LOG_ERROR, "Unable to attach transcoder: %s\n", strerror(errno));
+               munmap(hdr, sizeof(*hdr));
+               close(fd);
+
+               return NULL;
+       }
+
+       ztp = pvt->pvt;
+       ztp->fd = fd;
+       ztp->hdr = hdr;
+
+       return pvt;
+}
+
+static void *zap_new(struct ast_trans_pvt *pvt)
+{
+       return zap_translate(pvt, pvt->t->dstfmt, pvt->t->srcfmt);
+}
+
+static struct ast_frame *fakesrc_sample(void)
+{
+       /* Don't bother really trying to test hardware ones. */
+       static struct ast_frame f = {
+               .frametype = AST_FRAME_VOICE,
+               .samples = 160,
+               .src = __PRETTY_FUNCTION__
+       };
+
+       return &f;
+}
+
+static int register_translator(int dst, int src, void *mod)
+{
+       struct translator *zt;
+       int res;
+
+       if (!(zt = ast_calloc(1, sizeof(*zt))))
+               return -1;
+
+       snprintf((char *) (zt->t.name), sizeof(zt->t.name), "zap%sto%s", 
+                ast_getformatname((1 << src)), ast_getformatname((1 << dst)));
+       zt->t.srcfmt = (1 << src);
+       zt->t.dstfmt = (1 << dst);
+       zt->t.newpvt = zap_new;
+       zt->t.framein = zap_framein;
+       zt->t.frameout = zap_frameout;
+       zt->t.destroy = zap_destroy;
+       zt->t.sample = fakesrc_sample;
+       zt->t.useplc = global_useplc;
+       zt->t.buf_size = BUFFER_SAMPLES * 2;
+       zt->t.desc_size = sizeof(struct pvt);
+       if ((res = ast_register_translator(&zt->t, mod))) {
+               free(zt);
+               return -1;
+       }
+
+       AST_LIST_LOCK(&translators);
+       AST_LIST_INSERT_HEAD(&translators, zt, entry);
+       AST_LIST_UNLOCK(&translators);
+
+       global_format_map.map[dst][src] = 1;
+
+       return res;
+}
+
+static void drop_translator(int dst, int src)
+{
+       struct translator *cur;
+
+       AST_LIST_LOCK(&translators);
+       AST_LIST_TRAVERSE_SAFE_BEGIN(&translators, cur, entry) {
+               if (cur->t.srcfmt != src)
+                       continue;
+
+               if (cur->t.dstfmt != dst)
+                       continue;
+
+               AST_LIST_REMOVE_CURRENT(&translators, entry);
+               ast_unregister_translator(&cur->t);
+               free(cur);
+               global_format_map.map[dst][src] = 0;
+               break;
+       }
+       AST_LIST_TRAVERSE_SAFE_END;
+       AST_LIST_UNLOCK(&translators);
+}
+
+static void unregister_translators(void)
+{
+       struct translator *cur;
+
+       AST_LIST_LOCK(&translators);
+       while ((cur = AST_LIST_REMOVE_HEAD(&translators, entry))) {
+               ast_unregister_translator(&cur->t);
+               free(cur);
+       }
+       AST_LIST_UNLOCK(&translators);
+}
+
+static void parse_config(void)
+{
+       struct ast_variable *var;
+       struct ast_config *cfg = ast_config_load("codecs.conf");
+
+       if (!cfg)
+               return;
+
+       for (var = ast_variable_browse(cfg, "plc"); var; var = var->next) {
+              if (!strcasecmp(var->name, "genericplc")) {
+                      global_useplc = ast_true(var->value);
+                      if (option_verbose > 2)
+                              ast_verbose(VERBOSE_PREFIX_3 "codec_zap: %susing generic PLC\n",
+                                          global_useplc ? "" : "not ");
+              }
+       }
+
+       ast_config_destroy(cfg);
+}
+
+static void build_translators(void *mod, struct format_map *map, unsigned int dstfmts, unsigned int srcfmts)
+{
+       unsigned int src, dst;
+
+       for (src = 0; src < 32; src++) {
+               for (dst = 0; dst < 32; dst++) {
+                       if (!(srcfmts & (1 << src)))
+                               continue;
+
+                       if (!(dstfmts & (1 << dst)))
+                               continue;
+
+                       if (global_format_map.map[dst][src])
+                               continue;
+
+                       if (!register_translator(dst, src, mod))
+                               map->map[dst][src] = 1;
+               }
+       }
+}
+
+static int find_transcoders(void *mod)
+{
+       struct zt_transcode_info info = { 0, };
+       struct format_map map = { { { 0 } } };
+       int fd, res;
+       unsigned int x, y;
+
+       info.op = ZT_TCOP_GETINFO;
+       if ((fd = open("/dev/zap/transcode", O_RDWR)) < 0) {
+               ast_log(LOG_NOTICE, "No Zaptel transcoder support!\n");
+               return 0;
+       }
+       for (info.tcnum = 0; !(res = ioctl(fd, ZT_TRANSCODE_OP, &info)); info.tcnum++) {
+               if (option_verbose > 1)
+                       ast_verbose(VERBOSE_PREFIX_2 "Found transcoder '%s'.\n", info.name);
+               build_translators(mod, &map, info.dstfmts, info.srcfmts);
+       }
+       close(fd);
+
+       if (!info.tcnum && (option_verbose > 1))
+               ast_verbose(VERBOSE_PREFIX_2 "No hardware transcoders found.\n");
+
+       for (x = 0; x < 32; x++) {
+               for (y = 0; y < 32; y++) {
+                       if (!map.map[x][y] && global_format_map.map[x][y])
+                               drop_translator(x, y);
+               }
+       }
+
+       return 0;
+}
+
+static int reload(void *mod)
+{
+       struct translator *cur;
+
+       parse_config();
+       find_transcoders(mod);
+
+       AST_LIST_LOCK(&translators);
+       AST_LIST_TRAVERSE(&translators, cur, entry)
+               cur->t.useplc = global_useplc;
+       AST_LIST_UNLOCK(&translators);
+
+       return 0;
+}
+
+static int unload_module(void *mod)
+{
+       unregister_translators();
+
+       return 0;
+}
+
+static int load_module(void *mod)
+{
+       parse_config();
+       find_transcoders(mod);
+
+       return 0;
+}
+
+static const char *description(void)
+{
+       return "Generic Zaptel Transcoder Codec Translator";
+}
+
+static const char *key(void)
+{
+       return ASTERISK_GPL_KEY;
+}
+
+STD_MOD(MOD_1, reload, NULL, NULL);