formats/format_g729.c

   1 /*
   2  * Asterisk -- A telephony toolkit for Linux.
   3  *
   4  * Save to raw, headerless G729 data.
   5  *
   6  * Copyright (C) 1999, Mark Spencer
   7  *
   8  * Mark Spencer <markster@linux-support.net>
   9  *
  10  * This program is free software, distributed under the terms of
  11  * the GNU General Public License
  12  */
  13
  14 #include <asterisk/lock.h>
  15 #include <asterisk/channel.h>
  16 #include <asterisk/file.h>
  17 #include <asterisk/logger.h>
  18 #include <asterisk/sched.h>
  19 #include <asterisk/module.h>
  20 #include <arpa/inet.h>
  21 #include <stdlib.h>
  22 #include <sys/time.h>
  23 #include <stdio.h>
  24 #include <unistd.h>
  25 #include <errno.h>
  26 #include <string.h>
  27 #include <pthread.h>
  28 #ifdef __linux__
  29 #include <endian.h>
  30 #else
  31 #include <machine/endian.h>
  32 #endif
  33
  34 /* Some Ideas for this code came from makeg729e.c by Jeffery Chilton */
  35
  36 /* Portions of the conversion code are by guido@sienanet.it */
  37
  38 struct ast_filestream {
  39         void *reserved[AST_RESERVED_POINTERS];
  40         /* Believe it or not, we must decode/recode to account for the
  41            weird MS format */
  42         /* This is what a filestream means to us */
  43         int fd; /* Descriptor */
  44         struct ast_channel *owner;
  45         struct ast_frame fr;                            /* Frame information */
  46         char waste[AST_FRIENDLY_OFFSET];        /* Buffer for sending frames, etc */
  47         char empty;                                                     /* Empty character */
  48         unsigned char g729[20];                         /* Two Real G729 Frames */
  49         int lasttimeout;
  50         struct timeval last;
  51         int adj;
  52         struct ast_filestream *next;
  53 };
  54
  55
  56 static struct ast_filestream *glist = NULL;
  57 static pthread_mutex_t g729_lock = AST_MUTEX_INITIALIZER;
  58 static int glistcnt = 0;
  59
  60 static char *name = "g729";
  61 static char *desc = "Raw G729 data";
  62 static char *exts = "g729";
  63
  64 static struct ast_filestream *g729_open(int fd)
  65 {
  66         /* We don't have any header to read or anything really, but
  67            if we did, it would go here.  We also might want to check
  68            and be sure it's a valid file.  */
  69         struct ast_filestream *tmp;
  70         if ((tmp = malloc(sizeof(struct ast_filestream)))) {
  71                 memset(tmp, 0, sizeof(struct ast_filestream));
  72                 if (ast_pthread_mutex_lock(&g729_lock)) {
  73                         ast_log(LOG_WARNING, "Unable to lock g729 list\n");
  74                         free(tmp);
  75                         return NULL;
  76                 }
  77                 tmp->next = glist;
  78                 glist = tmp;
  79                 tmp->fd = fd;
  80                 tmp->owner = NULL;
  81                 tmp->fr.data = tmp->g729;
  82                 tmp->fr.frametype = AST_FRAME_VOICE;
  83                 tmp->fr.subclass = AST_FORMAT_G729A;
  84                 /* datalen will vary for each frame */
  85                 tmp->fr.src = name;
  86                 tmp->fr.mallocd = 0;
  87                 tmp->lasttimeout = -1;
  88                 glistcnt++;
  89                 ast_pthread_mutex_unlock(&g729_lock);
  90                 ast_update_use_count();
  91         }
  92         return tmp;
  93 }
  94
  95 static struct ast_filestream *g729_rewrite(int fd, char *comment)
  96 {
  97         /* We don't have any header to read or anything really, but
  98            if we did, it would go here.  We also might want to check
  99            and be sure it's a valid file.  */
 100         struct ast_filestream *tmp;
 101         if ((tmp = malloc(sizeof(struct ast_filestream)))) {
 102                 memset(tmp, 0, sizeof(struct ast_filestream));
 103                 if (ast_pthread_mutex_lock(&g729_lock)) {
 104                         ast_log(LOG_WARNING, "Unable to lock g729 list\n");
 105                         free(tmp);
 106                         return NULL;
 107                 }
 108                 tmp->next = glist;
 109                 glist = tmp;
 110                 tmp->fd = fd;
 111                 tmp->owner = NULL;
 112                 tmp->lasttimeout = -1;
 113                 glistcnt++;
 114                 ast_pthread_mutex_unlock(&g729_lock);
 115                 ast_update_use_count();
 116         } else
 117                 ast_log(LOG_WARNING, "Out of memory\n");
 118         return tmp;
 119 }
 120
 121 static struct ast_frame *g729_read(struct ast_filestream *s)
 122 {
 123         return NULL;
 124 }
 125
 126 static void g729_close(struct ast_filestream *s)
 127 {
 128         struct ast_filestream *tmp, *tmpl = NULL;
 129         if (ast_pthread_mutex_lock(&g729_lock)) {
 130                 ast_log(LOG_WARNING, "Unable to lock g729 list\n");
 131                 return;
 132         }
 133         tmp = glist;
 134         while(tmp) {
 135                 if (tmp == s) {
 136                         if (tmpl)
 137                                 tmpl->next = tmp->next;
 138                         else
 139                                 glist = tmp->next;
 140                         break;
 141                 }
 142                 tmpl = tmp;
 143                 tmp = tmp->next;
 144         }
 145         glistcnt--;
 146         if (s->owner) {
 147                 s->owner->stream = NULL;
 148                 if (s->owner->streamid > -1)
 149                         ast_sched_del(s->owner->sched, s->owner->streamid);
 150                 s->owner->streamid = -1;
 151         }
 152         ast_pthread_mutex_unlock(&g729_lock);
 153         ast_update_use_count();
 154         if (!tmp)
 155                 ast_log(LOG_WARNING, "Freeing a filestream we don't seem to own\n");
 156         close(s->fd);
 157         free(s);
 158         s = NULL;
 159 }
 160
 161 static int ast_read_callback(void *data)
 162 {
 163         int retval = 0;
 164         int res;
 165         int delay = 20;
 166         struct ast_filestream *s = data;
 167         struct timeval tv;
 168         /* Send a frame from the file to the appropriate channel */
 169
 170         s->fr.frametype = AST_FRAME_VOICE;
 171         s->fr.subclass = AST_FORMAT_G729A;
 172         s->fr.offset = AST_FRIENDLY_OFFSET;
 173         s->fr.samples = 160;
 174         s->fr.datalen = 20;
 175         s->fr.mallocd = 0;
 176         s->fr.data = s->g729;
 177         if ((res = read(s->fd, s->g729, 20)) != 20) {
 178                 if (res)
 179                         ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", res, strerror(errno));
 180                 s->owner->streamid = -1;
 181                 return 0;
 182         }
 183         /* Lastly, process the frame */
 184         if (ast_write(s->owner, &s->fr)) {
 185                 ast_log(LOG_WARNING, "Failed to write frame\n");
 186                 s->owner->streamid = -1;
 187                 return 0;
 188         }
 189         if (s->last.tv_usec || s->last.tv_usec) {
 190                 int ms;
 191                 gettimeofday(&tv, NULL);
 192                 ms = 1000 * (tv.tv_sec - s->last.tv_sec) +
 193                         (tv.tv_usec - s->last.tv_usec) / 1000;
 194                 s->last.tv_sec = tv.tv_sec;
 195                 s->last.tv_usec = tv.tv_usec;
 196                 if ((ms - delay) * (ms - delay) > 4) {
 197                         /* Compensate if we're more than 2 ms off */
 198                         s->adj -= (ms - delay);
 199                 }
 200 #if 0
 201                 fprintf(stdout, "Delay is %d, adjustment is %d, last was %d\n", delay, s->adj, ms);
 202 #endif
 203                 delay += s->adj;
 204                 if (delay < 1)
 205                         delay = 1;
 206         } else
 207                 gettimeofday(&s->last, NULL);
 208         if (s->lasttimeout != delay) {
 209                 /* We'll install the next timeout now. */
 210                 s->owner->streamid = ast_sched_add(s->owner->sched,
 211                                 delay, ast_read_callback, s);
 212                 s->lasttimeout = delay;
 213         } else {
 214                 /* Just come back again at the same time */
 215                 retval = -1;
 216         }
 217         return retval;
 218 }
 219
 220 static int g729_apply(struct ast_channel *c, struct ast_filestream *s)
 221 {
 222         /* Select our owner for this stream, and get the ball rolling. */
 223         s->owner = c;
 224         return 0;
 225 }
 226
 227 static int g729_play(struct ast_filestream *s)
 228 {
 229         ast_read_callback(s);
 230         return 0;
 231 }
 232
 233 static int g729_write(struct ast_filestream *fs, struct ast_frame *f)
 234 {
 235         int res;
 236         if (f->frametype != AST_FRAME_VOICE) {
 237                 ast_log(LOG_WARNING, "Asked to write non-voice frame!\n");
 238                 return -1;
 239         }
 240         if (f->subclass != AST_FORMAT_G729A) {
 241                 ast_log(LOG_WARNING, "Asked to write non-G729 frame (%d)!\n", f->subclass);
 242                 return -1;
 243         }
 244         if (f->datalen % 20) {
 245                 ast_log(LOG_WARNING, "Invalid data length, %d, should be multiple of 20\n", f->datalen);
 246                 return -1;
 247         }
 248         if ((res = write(fs->fd, f->data, f->datalen)) != f->datalen) {
 249                         ast_log(LOG_WARNING, "Bad write (%d/20): %s\n", res, strerror(errno));
 250                         return -1;
 251         }
 252         return 0;
 253 }
 254
 255 static char *g729_getcomment(struct ast_filestream *s)
 256 {
 257         return NULL;
 258 }
 259
 260 static int g729_seek(struct ast_filestream *fs, long sample_offset, int whence)
 261 {
 262         long bytes;
 263         off_t min,cur,max,offset;
 264         min = 0;
 265         cur = lseek(fs->fd, 0, SEEK_CUR);
 266         max = lseek(fs->fd, 0, SEEK_END);
 267
 268         bytes = 20 * (sample_offset / 160);
 269         if(whence == SEEK_SET)
 270                 offset = bytes;
 271         if(whence == SEEK_CUR)
 272                 offset = cur + bytes;
 273         if(whence == SEEK_END)
 274                 offset = max - bytes;
 275         offset = (offset > max)?max:offset;
 276         offset = (offset < min)?min:offset;
 277         if (lseek(fs->fd, offset, SEEK_SET) < 0)
 278                 return -1;
 279         return 0;
 280 }
 281
 282 static int g729_trunc(struct ast_filestream *fs)
 283 {
 284         /* Truncate file to current length */
 285         if (ftruncate(fs->fd, lseek(fs->fd, 0, SEEK_CUR)) < 0)
 286                 return -1;
 287         return 0;
 288 }
 289
 290 static long g729_tell(struct ast_filestream *fs)
 291 {
 292         off_t offset;
 293         offset = lseek(fs->fd, 0, SEEK_CUR);
 294         return (offset/20)*160;
 295 }
 296
 297 int load_module()
 298 {
 299         return ast_format_register(name, exts, AST_FORMAT_G729A,
 300                                                                 g729_open,
 301                                                                 g729_rewrite,
 302                                                                 g729_apply,
 303                                                                 g729_play,
 304                                                                 g729_write,
 305                                                                 g729_seek,
 306                                                                 g729_trunc,
 307                                                                 g729_tell,
 308                                                                 g729_read,
 309                                                                 g729_close,
 310                                                                 g729_getcomment);
 311
 312
 313 }
 314
 315 int unload_module()
 316 {
 317         struct ast_filestream *tmp, *tmpl;
 318         if (ast_pthread_mutex_lock(&g729_lock)) {
 319                 ast_log(LOG_WARNING, "Unable to lock g729 list\n");
 320                 return -1;
 321         }
 322         tmp = glist;
 323         while(tmp) {
 324                 if (tmp->owner)
 325                         ast_softhangup(tmp->owner, AST_SOFTHANGUP_APPUNLOAD);
 326                 tmpl = tmp;
 327                 tmp = tmp->next;
 328                 free(tmpl);
 329         }
 330         ast_pthread_mutex_unlock(&g729_lock);
 331         return ast_format_unregister(name);
 332 }
 333
 334 int usecount()
 335 {
 336         int res;
 337         if (ast_pthread_mutex_lock(&g729_lock)) {
 338                 ast_log(LOG_WARNING, "Unable to lock g729 list\n");
 339                 return -1;
 340         }
 341         res = glistcnt;
 342         ast_pthread_mutex_unlock(&g729_lock);
 343         return res;
 344 }
 345
 346 char *description()
 347 {
 348         return desc;
 349 }
 350
 351
 352 char *key()
 353 {
 354         return ASTERISK_GPL_KEY;
 355 }