* the GNU General Public License
*/
+#include <asterisk/lock.h>
#include <asterisk/channel.h>
#include <asterisk/channel_pvt.h>
#include <asterisk/logger.h>
#include <asterisk/translate.h>
#include <asterisk/options.h>
+#include <asterisk/frame.h>
+#include <asterisk/sched.h>
+#include <asterisk/cli.h>
+#include <asterisk/term.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>
-static char *type = "Trans";
+/* Uncomment the EXPERIMENTAL_TRANSLATION to enable a more complicated, but probably more
+ correct way of handling full duplex translation */
-static pthread_mutex_t list_lock = PTHREAD_MUTEX_INITIALIZER;
+/*
+#define EXPERIMENTAL_TRANSLATION
+*/
+
+/* This could all be done more efficiently *IF* we chained packets together
+ by default, but it would also complicate virtually every application. */
+
+static ast_mutex_t list_lock = AST_MUTEX_INITIALIZER;
static struct ast_translator *list = NULL;
struct ast_translator_dir {
int cost; /* Complete cost to destination */
};
+struct ast_frame_delivery {
+ struct ast_frame *f;
+ struct ast_channel *chan;
+ int fd;
+ struct translator_pvt *owner;
+ struct ast_frame_delivery *prev;
+ struct ast_frame_delivery *next;
+};
+
static struct ast_translator_dir tr_matrix[MAX_FORMAT][MAX_FORMAT];
struct ast_trans_pvt {
return -1;
}
-struct translator_pvt {
- /* Sockets for communication */
- int comm[2];
- struct ast_trans_pvt *system;
- struct ast_trans_pvt *rsystem;
- struct timeval lastpass;
- pthread_t threadid;
-};
-
-static int translator_hangup(struct ast_channel *chan)
-{
- ast_log(LOG_WARNING, "Explicit hangup on '%s' not recommended! Call translator_destroy() instead.\n", chan->name);
- chan->master->trans = NULL;
- ast_hangup(chan->master);
- chan->master = NULL;
- return 0;
-}
-
-static int translator_send_digit(struct ast_channel *chan, char digit)
-{
- /* Pass digits right along */
- if (chan->master->pvt->send_digit)
- return chan->master->pvt->send_digit(chan->master, digit);
- return -1;
-}
-
-static int translator_call(struct ast_channel *chan, char *addr, int timeout)
-{
- if (chan->master->pvt->call)
- return chan->master->pvt->call(chan->master, addr, timeout);
- return -1;
-}
-
-static int translator_answer(struct ast_channel *chan)
-{
- if (chan->master->pvt->answer)
- return chan->master->pvt->answer(chan->master);
- return -1;
-}
-
void ast_translator_free_path(struct ast_trans_pvt *p)
{
struct ast_trans_pvt *pl;
}
}
-static void ast_translator_free(struct translator_pvt *pvt)
-{
- ast_translator_free_path(pvt->system);
- ast_translator_free_path(pvt->rsystem);
- if (pvt->comm[0] > -1)
- close(pvt->comm[0]);
- if (pvt->comm[1] > -1)
- close(pvt->comm[1]);
- free(pvt);
-}
-
-struct ast_trans_pvt *ast_translator_build_path(int source, int dest)
+struct ast_trans_pvt *ast_translator_build_path(int dest, int source)
{
struct ast_trans_pvt *tmpr = NULL, *tmp = NULL;
/* One of the hardest parts: Build a set of translators based upon
tmp->step = tr_matrix[source][dest].step;
tmp->state = tmp->step->new();
if (!tmp->state) {
+ ast_log(LOG_WARNING, "Failed to build translator step from %d to %d\n", source, dest);
free(tmp);
tmp = NULL;
+ return NULL;
}
/* Set the root, if it doesn't exist yet... */
if (!tmpr)
ast_log(LOG_WARNING, "Out of memory\n");
return NULL;
}
- }
- }
- return tmpr;
-}
-
-static struct ast_frame *fd_read(int fd)
-{
- char buf[4096];
- int res;
- struct ast_frame *f = (struct ast_frame *)buf;
- /* Read a frame directly from there. They're always in the
- right format. */
-
- if (read(fd, buf, sizeof(struct ast_frame))
- == sizeof(struct ast_frame)) {
- /* read the frame header */
- f->mallocd = 0;
- /* Re-write data position */
- f->data = buf + sizeof(struct ast_frame) + AST_FRIENDLY_OFFSET;
- f->offset = AST_FRIENDLY_OFFSET;
- /* Forget about being mallocd */
- f->mallocd = 0;
- /* Re-write the source */
- f->src = __FUNCTION__;
- if (f->datalen > sizeof(buf) - sizeof(struct ast_frame) - AST_FRIENDLY_OFFSET) {
- /* Really bad read */
- ast_log(LOG_WARNING, "Strange read (%d bytes)\n", f->datalen);
+ } else {
+ /* We shouldn't have allocated any memory */
+ ast_log(LOG_WARNING, "No translator path from %s to %s\n",
+ ast_getformatname(source), ast_getformatname(dest));
return NULL;
}
- if (f->datalen) {
- if ((res = read(fd, f->data, f->datalen)) != f->datalen) {
- /* Bad read */
- ast_log(LOG_WARNING, "How very strange, expected %d, got %d\n", f->datalen, res);
- return NULL;
- }
- }
- return ast_frisolate(f);
- } else if (option_debug)
- ast_log(LOG_DEBUG, "NULL or invalid header\n");
- /* Null if there was an error */
- return NULL;
-}
-
-static struct ast_frame *translator_read(struct ast_channel *chan)
-{
- return fd_read(chan->fd);
-}
-
-static int fd_write(int fd, struct ast_frame *frame)
-{
- /* Write the frame exactly */
- if (write(fd, frame, sizeof(struct ast_frame)) != sizeof(struct ast_frame)) {
- ast_log(LOG_WARNING, "Write error\n");
- return -1;
- }
- if (write(fd, frame->data, frame->datalen) != frame->datalen) {
- ast_log(LOG_WARNING, "Write error\n");
- return -1;
- }
- return 0;
-}
-
-static int translator_write(struct ast_channel *chan, struct ast_frame *frame)
-{
- return fd_write(chan->fd, frame);
-}
-
-struct ast_frame_chain *ast_translate(struct ast_trans_pvt *path, struct ast_frame *f)
-{
- struct ast_trans_pvt *p;
- struct ast_frame *out;
- struct ast_frame_chain *outc = NULL, *prev = NULL, *cur;
- p = path;
- /* Feed the first frame into the first translator */
- p->step->framein(p->state, f);
- while(p) {
- /* Read all the frames from the current translator */
- while((out = p->step->frameout(p->state))) {
- if (p->next) {
- /* Feed to next layer */
- p->next->step->framein(p->next->state, out);
- } else {
- /* Last layer -- actually do something */
- cur = malloc(sizeof(struct ast_frame_chain));
- if (!cur) {
- /* XXX Leak majorly on a problem XXX */
- ast_log(LOG_WARNING, "Out of memory\n");
- return NULL;
- }
- if (prev)
- prev->next = cur;
- else
- outc = cur;
- cur->fr = ast_frisolate(out);
- cur->next = NULL;
- if (prev)
- prev = prev->next;
- else
- prev = outc;
- }
- }
- p = p->next;
}
- return outc;
+ return tmpr;
}
-#define FUDGE 2
-
-static void translator_apply(struct ast_trans_pvt *path, struct ast_frame *f, int fd, struct ast_channel *c, struct timeval *last)
+struct ast_frame *ast_translate(struct ast_trans_pvt *path, struct ast_frame *f, int consume)
{
struct ast_trans_pvt *p;
struct ast_frame *out;
- struct timeval tv;
- int ms;
p = path;
/* Feed the first frame into the first translator */
p->step->framein(p->state, f);
+ if (consume)
+ ast_frfree(f);
while(p) {
- /* Read all the frames from the current translator */
- while((out = p->step->frameout(p->state))) {
- if (p->next) {
- /* Feed to next layer */
- p->next->step->framein(p->next->state, out);
- } else {
- /* Delay if needed */
- if (last->tv_sec || last->tv_usec) {
- gettimeofday(&tv, NULL);
- ms = 1000 * (tv.tv_sec - last->tv_sec) +
- (tv.tv_usec - last->tv_usec) / 1000;
- if (ms + FUDGE < out->timelen)
- usleep((out->timelen - ms - FUDGE) * 1000);
- last->tv_sec = tv.tv_sec;
- last->tv_usec = tv.tv_usec;
- }
- if (c)
- ast_write(c, out);
- else
- fd_write(fd, out);
- }
- ast_frfree(out);
- }
+ out = p->step->frameout(p->state);
+ /* If we get nothing out, return NULL */
+ if (!out)
+ return NULL;
+ /* If there is a next state, feed it in there. If not,
+ return this frame */
+ if (p->next)
+ p->next->step->framein(p->next->state, out);
+ else
+ return out;
p = p->next;
}
-}
-
-static void *translator_thread(void *data)
-{
- struct ast_channel *real = data;
- struct ast_frame *f;
- int ms = -1;
- struct translator_pvt *pvt = NULL;
- int fd = -1;
- int fds[2];
- int res;
- /* Read from the real, translate, write as necessary to the fake */
- for(;;) {
- /* Break here if need be */
- pthread_testcancel();
- if (!real->trans) {
- ast_log(LOG_WARNING, "No translator anymore\n");
- break;
- }
- pvt = real->trans->pvt->pvt;
- fd = pvt->comm[1];
- fds[0] = fd;
- fds[1] = real->fd;
- CHECK_BLOCKING(real);
- res = ast_waitfor_n_fd(fds, 2, &ms);
- real->blocking = 0;
- /* Or we can die here, that's fine too */
- pthread_testcancel();
- if (res >= 0) {
- if (res == real->fd) {
- f = ast_read(real);
- if (!f) {
- if (option_debug)
- ast_log(LOG_DEBUG, "Empty frame\n");
- break;
- }
- if (f->frametype == AST_FRAME_VOICE) {
- if (pvt->system)
- translator_apply(pvt->system, f, fd, NULL, &pvt->lastpass);
- } else {
- /* If it's not voice, just pass it along */
- fd_write(fd, f);
- }
- ast_frfree(f);
- } else {
- f = fd_read(res);
- if (!f) {
- if (option_debug)
- ast_log(LOG_DEBUG, "Empty (hangup) frame\n");
- break;
- }
-
- if (f->frametype == AST_FRAME_VOICE) {
- if (pvt->rsystem)
- translator_apply(pvt->rsystem, f, -1, real, &pvt->lastpass);
- } else {
- ast_write(real, f);
- }
- ast_frfree(f);
- }
- } else {
- ast_log(LOG_DEBUG, "Waitfor returned non-zero\n");
- break;
- }
- }
- if (pvt)
- pvt->comm[1] = -1;
- if (fd > -1) {
- /* Write a bogus frame */
- write(fd, data, 1);
- close(fd);
- }
+ ast_log(LOG_WARNING, "I should never get here...\n");
return NULL;
}
-struct ast_channel *ast_translator_create(struct ast_channel *real, int format, int direction)
-{
- struct ast_channel *tmp;
- struct translator_pvt *pvt;
- if (real->trans) {
- ast_log(LOG_WARNING, "Translator already exists on '%s'\n", real->name);
- return NULL;
- }
- if (!(pvt = malloc(sizeof(struct translator_pvt)))) {
- ast_log(LOG_WARNING, "Unable to allocate private translator on '%s'\n", real->name);
- return NULL;
- }
- pvt->comm[0] = -1;
- pvt->comm[1] = -1;
- pvt->lastpass.tv_usec = 0;
- pvt->lastpass.tv_sec = 0;
- if (socketpair(AF_UNIX, SOCK_DGRAM, 0, pvt->comm)) {
- ast_log(LOG_WARNING, "Unable to create UNIX domain socket on '%s'\n", real->name);
- ast_translator_free(pvt);
- }
- /* In to the system */
- if (direction & AST_DIRECTION_IN)
- pvt->system = ast_translator_build_path(real->format, format);
- else
- pvt->system = NULL;
- /* Out from the system */
- if (direction & AST_DIRECTION_OUT)
- pvt->rsystem = ast_translator_build_path(format, real->format);
- else
- pvt->rsystem = NULL;
- if (!pvt->system && !pvt->rsystem) {
- ast_log(LOG_WARNING, "Unable to build a translation path for %s (%d to %d)\n", real->name, real->format, format);
- ast_translator_free(pvt);
- return NULL;
- }
- if (!pvt->system && (direction & AST_DIRECTION_IN)) {
- ast_log(LOG_WARNING, "Translation path for '%s' is one-way (reverse)\n", real->name);
- ast_translator_free(pvt);
- return NULL;
- }
- if (!pvt->rsystem && (direction & AST_DIRECTION_OUT)) {
- ast_log(LOG_WARNING, "Translation path for '%s' is one-way (forward)\n", real->name);
- ast_translator_free(pvt);
- return NULL;
- }
- if ((tmp = ast_channel_alloc())) {
- snprintf(tmp->name, sizeof(tmp->name), "%s/Translate:%d", real->name, format);
- tmp->type = type;
- tmp->fd = pvt->comm[0];
- tmp->format = format;
- tmp->state = real->state;
- tmp->rings = 0;
- tmp->pvt->pvt = pvt;
- tmp->master = real;
- tmp->pvt->send_digit = translator_send_digit;
- tmp->pvt->call = translator_call;
- tmp->pvt->hangup = translator_hangup;
- tmp->pvt->answer = translator_answer;
- tmp->pvt->read = translator_read;
- tmp->pvt->write = translator_write;
- tmp->pvt->pvt = pvt;
- real->trans = tmp;
- if (option_verbose > 2)
- ast_verbose(VERBOSE_PREFIX_3 "Created translator %s\n", tmp->name);
- if (pthread_create(&pvt->threadid, NULL, translator_thread, real) < 0) {
- ast_translator_destroy(tmp);
- tmp = NULL;
- ast_log(LOG_WARNING, "Failed to start thread\n");
- }
- } else {
- ast_translator_free(pvt);
- ast_log(LOG_WARNING, "Unable to allocate channel\n");
- }
- return tmp;
-}
-
-static void rebuild_matrix()
+static void rebuild_matrix(void)
{
struct ast_translator *t;
int changed;
tr_matrix[x][z].cost = tr_matrix[x][y].cost +
tr_matrix[y][z].cost;
if (option_debug)
- ast_log(LOG_DEBUG, "Discovered %d cost path from %d to %d, via %d\n", tr_matrix[x][z].cost, x, z, y);
+ ast_log(LOG_DEBUG, "Discovered %d cost path from %s to %s, via %d\n", tr_matrix[x][z].cost, ast_getformatname(x), ast_getformatname(z), y);
changed++;
}
}
gettimeofday(&start, NULL);
/* Call the encoder until we've processed one second of time */
- while(sofar < 1000) {
+ while(sofar < 8000) {
f = t->sample();
if (!f) {
ast_log(LOG_WARNING, "Translator '%s' failed to produce a sample frame.\n", t->name);
t->framein(pvt, f);
ast_frfree(f);
while((out = t->frameout(pvt))) {
- sofar += out->timelen;
+ sofar += out->samples;
ast_frfree(out);
}
}
t->destroy(pvt);
cost = (finish.tv_sec - start.tv_sec) * 1000 + (finish.tv_usec - start.tv_usec) / 1000;
t->cost = cost;
+ if (!t->cost)
+ t->cost = 1;
+}
+
+static int show_translation(int fd, int argc, char *argv[])
+{
+#define SHOW_TRANS 11
+ int x,y;
+ char line[80];
+ if (argc != 2)
+ return RESULT_SHOWUSAGE;
+ ast_cli(fd, " Translation times between formats (in milliseconds)\n");
+ ast_cli(fd, " Source Format (Rows) Destination Format(Columns)\n\n");
+ ast_mutex_lock(&list_lock);
+ for (x=-1;x<SHOW_TRANS; x++) {
+ strcpy(line, " ");
+ for (y=-1;y<SHOW_TRANS;y++) {
+ if (x >= 0 && y >= 0 && tr_matrix[x][y].step)
+ snprintf(line + strlen(line), sizeof(line) - strlen(line), " %5d", tr_matrix[x][y].cost >= 99999 ? tr_matrix[x][y].cost-99999 : tr_matrix[x][y].cost);
+ else
+ if (((x == -1 && y >= 0) || (y == -1 && x >= 0))) {
+ snprintf(line + strlen(line), sizeof(line) - strlen(line),
+ " %5s", ast_getformatname(1<<(x+y+1)) );
+ } else if (x != -1 && y != -1) {
+ snprintf(line + strlen(line), sizeof(line) - strlen(line), " -");
+ } else {
+ snprintf(line + strlen(line), sizeof(line) - strlen(line), " ");
+ }
+ }
+ snprintf(line + strlen(line), sizeof(line) - strlen(line), "\n");
+ ast_cli(fd, line);
+ }
+ ast_mutex_unlock(&list_lock);
+ return RESULT_SUCCESS;
}
+static int added_cli = 0;
+
+static char show_trans_usage[] =
+"Usage: show translation\n"
+" Displays known codec translators and the cost associated\n"
+"with each conversion.\n";
+
+static struct ast_cli_entry show_trans =
+{ { "show", "translation", NULL }, show_translation, "Display translation matrix", show_trans_usage };
+
int ast_register_translator(struct ast_translator *t)
{
+ char tmp[80];
t->srcfmt = powerof(t->srcfmt);
t->dstfmt = powerof(t->dstfmt);
if ((t->srcfmt >= MAX_FORMAT) || (t->dstfmt >= MAX_FORMAT)) {
- ast_log(LOG_WARNING, "Format %d is larger than MAX_FORMAT\n", t->srcfmt);
+ ast_log(LOG_WARNING, "Format %s is larger than MAX_FORMAT\n", ast_getformatname(t->srcfmt));
return -1;
}
calc_cost(t);
if (option_verbose > 1)
- ast_verbose(VERBOSE_PREFIX_2 "Registered translator '%s' from format %d to %d, cost %d\n", t->name, t->srcfmt, t->dstfmt, t->cost);
- pthread_mutex_lock(&list_lock);
+ ast_verbose(VERBOSE_PREFIX_2 "Registered translator '%s' from format %s to %s, cost %d\n", term_color(tmp, t->name, COLOR_MAGENTA, COLOR_BLACK, sizeof(tmp)), ast_getformatname(1 << t->srcfmt), ast_getformatname(1 << t->dstfmt), t->cost);
+ ast_mutex_lock(&list_lock);
+ if (!added_cli) {
+ ast_cli_register(&show_trans);
+ added_cli++;
+ }
t->next = list;
list = t;
rebuild_matrix();
- pthread_mutex_unlock(&list_lock);
+ ast_mutex_unlock(&list_lock);
return 0;
}
int ast_unregister_translator(struct ast_translator *t)
{
struct ast_translator *u, *ul = NULL;
- pthread_mutex_lock(&list_lock);
+ ast_mutex_lock(&list_lock);
u = list;
while(u) {
if (u == t) {
list = u->next;
break;
}
+ ul = u;
u = u->next;
}
rebuild_matrix();
- pthread_mutex_unlock(&list_lock);
+ ast_mutex_unlock(&list_lock);
return (u ? 0 : -1);
}
-void ast_translator_destroy(struct ast_channel *trans)
-{
- struct translator_pvt *pvt;
- if (!trans->master) {
- ast_log(LOG_WARNING, "Translator is not part of a real channel?\n");
- return;
- }
- if (trans->master->trans != trans) {
- ast_log(LOG_WARNING, "Translator is not the right one!?!?\n");
- return;
- }
- trans->master->trans = NULL;
- pvt = trans->pvt->pvt;
- /* Cancel the running translator thread */
- pthread_cancel(pvt->threadid);
- pthread_join(pvt->threadid, NULL);
- ast_translator_free(pvt);
- trans->pvt->pvt = NULL;
- if (option_verbose > 2)
- ast_verbose(VERBOSE_PREFIX_3 "Destroyed translator %s\n", trans->name);
- ast_channel_free(trans);
-}
-
-int ast_translator_best_choice(int dst, int srcs)
+int ast_translator_best_choice(int *dst, int *srcs)
{
/* Calculate our best source format, given costs, and a desired destination */
- int x;
+ int x,y;
int best=-1;
+ int bestdst=0;
+ int cur = 1;
int besttime=999999999;
- dst = powerof(dst);
- pthread_mutex_lock(&list_lock);
- for (x=0;x<MAX_FORMAT;x++) {
- if (tr_matrix[x][dst].step && /* There's a step */
- (tr_matrix[x][dst].cost < besttime) && /* We're better than what exists now */
- (srcs & (1 << x))) /* x is a valid source format */
- {
- best = 1 << x;
- besttime = tr_matrix[x][dst].cost;
+ ast_mutex_lock(&list_lock);
+ for (y=0;y<MAX_FORMAT;y++) {
+ if ((cur & *dst) && (cur & *srcs)) {
+ /* This is a common format to both. Pick it if we don't have one already */
+ besttime=0;
+ bestdst = cur;
+ best = cur;
+ break;
+ }
+ if (cur & *dst)
+ for (x=0;x<MAX_FORMAT;x++) {
+ if (tr_matrix[x][y].step && /* There's a step */
+ (tr_matrix[x][y].cost < besttime) && /* We're better than what exists now */
+ (*srcs & (1 << x))) /* x is a valid source format */
+ {
+ best = 1 << x;
+ bestdst = cur;
+ besttime = tr_matrix[x][y].cost;
+ }
}
-
+ cur = cur << 1;
+ }
+ if (best > -1) {
+ *srcs = best;
+ *dst = bestdst;
+ best = 0;
}
- pthread_mutex_unlock(&list_lock);
+ ast_mutex_unlock(&list_lock);
return best;
}