Skip to content
Snippets Groups Projects
app_record.c 4.11 KiB
Newer Older
  • Learn to ignore specific revisions
  • Mark Spencer's avatar
    Mark Spencer committed
    /*
     * Asterisk -- A telephony toolkit for Linux.
     *
     * Trivial application to record a sound file
     * 
     * Copyright (C) 2001, Linux Support Services, Inc.
     *
     * Matthew Fredrickson <creslin@linux-support.net>
     *
     * This program is free software, distributed under the terms of
     * the GNU General Public License
     */
     
    
    Mark Spencer's avatar
    Mark Spencer committed
    #include <asterisk/lock.h>
    
    Mark Spencer's avatar
    Mark Spencer committed
    #include <asterisk/file.h>
    #include <asterisk/logger.h>
    #include <asterisk/channel.h>
    #include <asterisk/pbx.h>
    #include <asterisk/module.h>
    #include <asterisk/translate.h>
    #include <string.h>
    #include <stdlib.h>
    #include <pthread.h>
    
    static char *tdesc = "Trivial Record Application";
    
    static char *app = "Record";
    
    static char *synopsis = "Record to a file";
    
    static char *descrip = 
    
    Mark Spencer's avatar
    Mark Spencer committed
    "  Record(filename:extension): Records from the  channel into a given\n"
    "filename. If the file exists it will be overwritten. The 'extension'\n"
    "is the extension of the file type  to  be  recorded (wav, gsm, etc).\n"
    "Returns -1 when the user hangs up.\n";
    
    Mark Spencer's avatar
    Mark Spencer committed
    
    STANDARD_LOCAL_USER;
    
    LOCAL_USER_DECL;
    
    static int record_exec(struct ast_channel *chan, void *data)
    {
    	int res = 0;
    	int count = 0;
    	int percentflag = 0;
    	char fil[256];
    	char tmp[256];
    	char ext[10];
    	char * vdata;  /* Used so I don't have to typecast every use of *data */
    	int i = 0;
    	int j = 0;
    	
    	struct ast_filestream *s = '\0';
    	struct localuser *u;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	struct ast_frame *f = NULL;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	
    	vdata = data; /* explained above */
    	
    
    	/* The next few lines of code parse out the filename and header from the input string */
    	if (!vdata) { /* no data implies no filename or anything is present */
    		ast_log(LOG_WARNING, "Record requires an argument (filename)\n");
    		return -1;
    	}
    	
    	for (; vdata[i] && (vdata[i] != ':') ; i++ ) {
    		if ((vdata[i] == '%') && (vdata[i+1] == 'd')) {
    			percentflag = 1;                      /* the wildcard is used */
    		}
    		
    		if (i == strlen(vdata) ) {
    			ast_log(LOG_WARNING, "No extension found\n");
    			return -1;
    		}
    		fil[i] = vdata[i];
    	}
    	fil[i++] = '\0';
    
    	for (; j < 10 && i < strlen(data); i++, j++)
    		ext[j] = vdata[i];
    	ext[j] = '\0';
    	/* done parsing */
    	
    	
    	/* these are to allow the use of the %d in the config file for a wild card of sort to
    	  create a new file with the inputed name scheme */
    	if (percentflag) {
    		do {
    			snprintf(tmp, 256, fil, count);
    			count++;
    		} while ( ast_fileexists(tmp, ext, chan->language) != -1 );
    	} else
    
    Mark Spencer's avatar
    Mark Spencer committed
    		strncpy(tmp, fil, 256-1);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	/* end of routine mentioned */
    
    	LOCAL_USER_ADD(u);
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (chan->_state != AST_STATE_UP) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    		res = ast_answer(chan); /* Shouldn't need this, but checking to see if channel is already answered
    					 * Theoretically asterisk should already have answered before running the app */
    	}
    	
    	if (!res) {
    		/* Some code to play a nice little beep to signify the start of the record operation */
    		res = ast_streamfile(chan, "beep", chan->language);
    		if (!res) {
    			res = ast_waitstream(chan, "");
    		} else {
    			ast_log(LOG_WARNING, "ast_streamfile failed on %s\n", chan->name);
    		}
    		ast_stopstream(chan);
    		/* The end of beep code.  Now the recording starts */
    		s = ast_writefile( tmp, ext, NULL, O_CREAT|O_TRUNC|O_WRONLY , 0, 0644);
    	
    		if (s) {
    
    Mark Spencer's avatar
    Mark Spencer committed
    			while (ast_waitfor(chan, -1) > -1) {
    				f = ast_read(chan);
    				if (!f) {
    					res = -1;
    					break;
    				}
    
    Mark Spencer's avatar
    Mark Spencer committed
    				if (f->frametype == AST_FRAME_VOICE) {
    					res = ast_writestream(s, f);
    					
    					if (res) {
    						ast_log(LOG_WARNING, "Problem writing frame\n");
    						break;
    					}
    				}
    
    Mark Spencer's avatar
    Mark Spencer committed
    				if ((f->frametype == AST_FRAME_DTMF) &&
    					(f->subclass == '#')) {
    					ast_frfree(f);
    					break;
    				}
    				ast_frfree(f);
    
    Mark Spencer's avatar
    Mark Spencer committed
    			}
    			if (!f) {
    					ast_log(LOG_DEBUG, "Got hangup\n");
    					res = -1;
    			}
    			ast_closestream(s);
    		} else			
    			ast_log(LOG_WARNING, "Could not create file %s\n", fil);
    	} else
    		ast_log(LOG_WARNING, "Could not answer channel '%s'\n", chan->name);
    
    	LOCAL_USER_REMOVE(u);
    	return res;
    }
    
    int unload_module(void)
    {
    	STANDARD_HANGUP_LOCALUSERS;
    	return ast_unregister_application(app);
    }
    
    int load_module(void)
    {
    	return ast_register_application(app, record_exec, synopsis, descrip);
    }
    
    char *description(void)
    {
    	return tdesc;
    }
    
    int usecount(void)
    {
    	int res;
    	STANDARD_USECOUNT(res);
    	return res;
    }
    
    char *key()
    {
    	return ASTERISK_GPL_KEY;
    }