00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032 #include "asterisk.h"
00033
00034 ASTERISK_FILE_VERSION(__FILE__, "$Revision: 328209 $")
00035
00036 #include "asterisk/file.h"
00037 #include "asterisk/pbx.h"
00038 #include "asterisk/module.h"
00039 #include "asterisk/app.h"
00040 #include "asterisk/channel.h"
00041 #include "asterisk/dsp.h"
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114
00115
00116 static char *app = "Record";
00117
00118 enum {
00119 OPTION_APPEND = (1 << 0),
00120 OPTION_NOANSWER = (1 << 1),
00121 OPTION_QUIET = (1 << 2),
00122 OPTION_SKIP = (1 << 3),
00123 OPTION_STAR_TERMINATE = (1 << 4),
00124 OPTION_IGNORE_TERMINATE = (1 << 5),
00125 OPTION_KEEP = (1 << 6),
00126 FLAG_HAS_PERCENT = (1 << 7),
00127 OPTION_ANY_TERMINATE = (1 << 8),
00128 };
00129
00130 AST_APP_OPTIONS(app_opts,{
00131 AST_APP_OPTION('a', OPTION_APPEND),
00132 AST_APP_OPTION('k', OPTION_KEEP),
00133 AST_APP_OPTION('n', OPTION_NOANSWER),
00134 AST_APP_OPTION('q', OPTION_QUIET),
00135 AST_APP_OPTION('s', OPTION_SKIP),
00136 AST_APP_OPTION('t', OPTION_STAR_TERMINATE),
00137 AST_APP_OPTION('y', OPTION_ANY_TERMINATE),
00138 AST_APP_OPTION('x', OPTION_IGNORE_TERMINATE),
00139 });
00140
00141 static int record_exec(struct ast_channel *chan, const char *data)
00142 {
00143 int res = 0;
00144 int count = 0;
00145 char *ext = NULL, *opts[0];
00146 char *parse, *dir, *file;
00147 int i = 0;
00148 char tmp[256];
00149
00150 struct ast_filestream *s = NULL;
00151 struct ast_frame *f = NULL;
00152
00153 struct ast_dsp *sildet = NULL;
00154 int totalsilence = 0;
00155 int dspsilence = 0;
00156 int silence = 0;
00157 int gotsilence = 0;
00158 int maxduration = 0;
00159 int gottimeout = 0;
00160 int terminator = '#';
00161 int rfmt = 0;
00162 int ioflags;
00163 int waitres;
00164 struct ast_silence_generator *silgen = NULL;
00165 struct ast_flags flags = { 0, };
00166 AST_DECLARE_APP_ARGS(args,
00167 AST_APP_ARG(filename);
00168 AST_APP_ARG(silence);
00169 AST_APP_ARG(maxduration);
00170 AST_APP_ARG(options);
00171 );
00172
00173
00174 if (ast_strlen_zero(data)) {
00175 ast_log(LOG_WARNING, "Record requires an argument (filename)\n");
00176 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00177 return -1;
00178 }
00179
00180 parse = ast_strdupa(data);
00181 AST_STANDARD_APP_ARGS(args, parse);
00182 if (args.argc == 4)
00183 ast_app_parse_options(app_opts, &flags, opts, args.options);
00184
00185 if (!ast_strlen_zero(args.filename)) {
00186 if (strstr(args.filename, "%d"))
00187 ast_set_flag(&flags, FLAG_HAS_PERCENT);
00188 ext = strrchr(args.filename, '.');
00189 if (!ext)
00190 ext = strchr(args.filename, ':');
00191 if (ext) {
00192 *ext = '\0';
00193 ext++;
00194 }
00195 }
00196 if (!ext) {
00197 ast_log(LOG_WARNING, "No extension specified to filename!\n");
00198 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00199 return -1;
00200 }
00201 if (args.silence) {
00202 if ((sscanf(args.silence, "%30d", &i) == 1) && (i > -1)) {
00203 silence = i * 1000;
00204 } else if (!ast_strlen_zero(args.silence)) {
00205 ast_log(LOG_WARNING, "'%s' is not a valid silence duration\n", args.silence);
00206 }
00207 }
00208
00209 if (args.maxduration) {
00210 if ((sscanf(args.maxduration, "%30d", &i) == 1) && (i > -1))
00211
00212 maxduration = i * 1000;
00213 else if (!ast_strlen_zero(args.maxduration))
00214 ast_log(LOG_WARNING, "'%s' is not a valid maximum duration\n", args.maxduration);
00215 }
00216
00217 if (ast_test_flag(&flags, OPTION_STAR_TERMINATE))
00218 terminator = '*';
00219 if (ast_test_flag(&flags, OPTION_IGNORE_TERMINATE))
00220 terminator = '\0';
00221
00222
00223
00224
00225
00226 if (ast_test_flag(&flags, FLAG_HAS_PERCENT)) {
00227 AST_DECLARE_APP_ARGS(fname,
00228 AST_APP_ARG(piece)[100];
00229 );
00230 char *tmp2 = ast_strdupa(args.filename);
00231 char countstring[15];
00232 int idx;
00233
00234
00235 AST_NONSTANDARD_APP_ARGS(fname, tmp2, '%');
00236 do {
00237 int tmplen;
00238
00239 ast_copy_string(tmp, fname.piece[0], sizeof(tmp));
00240 tmplen = strlen(tmp);
00241 for (idx = 1; idx < fname.argc; idx++) {
00242 if (fname.piece[idx][0] == 'd') {
00243
00244 snprintf(countstring, sizeof(countstring), "%d", count);
00245 ast_copy_string(tmp + tmplen, countstring, sizeof(tmp) - tmplen);
00246 tmplen += strlen(countstring);
00247 } else if (tmplen + 2 < sizeof(tmp)) {
00248
00249 tmp[tmplen++] = '%';
00250 tmp[tmplen++] = fname.piece[idx][0];
00251 }
00252
00253 ast_copy_string(tmp + tmplen, &(fname.piece[idx][1]), sizeof(tmp) - tmplen);
00254 }
00255 count++;
00256 } while (ast_fileexists(tmp, ext, chan->language) > 0);
00257 pbx_builtin_setvar_helper(chan, "RECORDED_FILE", tmp);
00258 } else
00259 ast_copy_string(tmp, args.filename, sizeof(tmp));
00260
00261
00262 if (chan->_state != AST_STATE_UP) {
00263 if (ast_test_flag(&flags, OPTION_SKIP)) {
00264
00265 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "SKIP");
00266 return 0;
00267 } else if (!ast_test_flag(&flags, OPTION_NOANSWER)) {
00268
00269 res = ast_answer(chan);
00270 }
00271 }
00272
00273 if (res) {
00274 ast_log(LOG_WARNING, "Could not answer channel '%s'\n", chan->name);
00275 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00276 goto out;
00277 }
00278
00279 if (!ast_test_flag(&flags, OPTION_QUIET)) {
00280
00281 res = ast_streamfile(chan, "beep", chan->language);
00282 if (!res) {
00283 res = ast_waitstream(chan, "");
00284 } else {
00285 ast_log(LOG_WARNING, "ast_streamfile failed on %s\n", chan->name);
00286 }
00287 ast_stopstream(chan);
00288 }
00289
00290
00291
00292 if (silence > 0) {
00293 rfmt = chan->readformat;
00294 res = ast_set_read_format(chan, AST_FORMAT_SLINEAR);
00295 if (res < 0) {
00296 ast_log(LOG_WARNING, "Unable to set to linear mode, giving up\n");
00297 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00298 return -1;
00299 }
00300 sildet = ast_dsp_new();
00301 if (!sildet) {
00302 ast_log(LOG_WARNING, "Unable to create silence detector :(\n");
00303 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00304 return -1;
00305 }
00306 ast_dsp_set_threshold(sildet, ast_dsp_get_threshold_from_settings(THRESHOLD_SILENCE));
00307 }
00308
00309
00310 dir = ast_strdupa(tmp);
00311 if ((file = strrchr(dir, '/')))
00312 *file++ = '\0';
00313 ast_mkdir (dir, 0777);
00314
00315 ioflags = ast_test_flag(&flags, OPTION_APPEND) ? O_CREAT|O_APPEND|O_WRONLY : O_CREAT|O_TRUNC|O_WRONLY;
00316 s = ast_writefile(tmp, ext, NULL, ioflags, 0, AST_FILE_MODE);
00317
00318 if (!s) {
00319 ast_log(LOG_WARNING, "Could not create file %s\n", args.filename);
00320 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00321 goto out;
00322 }
00323
00324 if (ast_opt_transmit_silence)
00325 silgen = ast_channel_start_silence_generator(chan);
00326
00327
00328 ast_indicate(chan, AST_CONTROL_VIDUPDATE);
00329
00330 if (maxduration <= 0)
00331 maxduration = -1;
00332
00333 while ((waitres = ast_waitfor(chan, maxduration)) > -1) {
00334 if (maxduration > 0) {
00335 if (waitres == 0) {
00336 gottimeout = 1;
00337 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "TIMEOUT");
00338 break;
00339 }
00340 maxduration = waitres;
00341 }
00342
00343 f = ast_read(chan);
00344 if (!f) {
00345 res = -1;
00346 break;
00347 }
00348 if (f->frametype == AST_FRAME_VOICE) {
00349 res = ast_writestream(s, f);
00350
00351 if (res) {
00352 ast_log(LOG_WARNING, "Problem writing frame\n");
00353 ast_frfree(f);
00354 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00355 break;
00356 }
00357
00358 if (silence > 0) {
00359 dspsilence = 0;
00360 ast_dsp_silence(sildet, f, &dspsilence);
00361 if (dspsilence) {
00362 totalsilence = dspsilence;
00363 } else {
00364 totalsilence = 0;
00365 }
00366 if (totalsilence > silence) {
00367
00368 ast_frfree(f);
00369 gotsilence = 1;
00370 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "SILENCE");
00371 break;
00372 }
00373 }
00374 } else if (f->frametype == AST_FRAME_VIDEO) {
00375 res = ast_writestream(s, f);
00376
00377 if (res) {
00378 ast_log(LOG_WARNING, "Problem writing frame\n");
00379 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "ERROR");
00380 ast_frfree(f);
00381 break;
00382 }
00383 } else if ((f->frametype == AST_FRAME_DTMF) &&
00384 ((f->subclass.integer == terminator) ||
00385 (ast_test_flag(&flags, OPTION_ANY_TERMINATE)))) {
00386 ast_frfree(f);
00387 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "DTMF");
00388 break;
00389 }
00390 ast_frfree(f);
00391 }
00392 if (!f) {
00393 ast_debug(1, "Got hangup\n");
00394 res = -1;
00395 pbx_builtin_setvar_helper(chan, "RECORD_STATUS", "HANGUP");
00396 if (!ast_test_flag(&flags, OPTION_KEEP)) {
00397 ast_filedelete(args.filename, NULL);
00398 }
00399 }
00400
00401 if (gotsilence) {
00402 ast_stream_rewind(s, silence - 1000);
00403 ast_truncstream(s);
00404 } else if (!gottimeout) {
00405
00406 ast_stream_rewind(s, 250);
00407 ast_truncstream(s);
00408 }
00409 ast_closestream(s);
00410
00411 if (silgen)
00412 ast_channel_stop_silence_generator(chan, silgen);
00413
00414 out:
00415 if ((silence > 0) && rfmt) {
00416 res = ast_set_read_format(chan, rfmt);
00417 if (res)
00418 ast_log(LOG_WARNING, "Unable to restore read format on '%s'\n", chan->name);
00419 if (sildet)
00420 ast_dsp_free(sildet);
00421 }
00422
00423 return res;
00424 }
00425
00426 static int unload_module(void)
00427 {
00428 return ast_unregister_application(app);
00429 }
00430
00431 static int load_module(void)
00432 {
00433 return ast_register_application_xml(app, record_exec);
00434 }
00435
00436 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Trivial Record Application");