c4fc4a630061b720f3d63b621463b30c46af0347
[asterisk/asterisk.git] / res / res_convert.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2005, 2006, Digium, Inc.
5  *
6  * redice li <redice_li@yahoo.com>
7  * Russell Bryant <russell@digium.com>
8  *
9  * See http://www.asterisk.org for more information about
10  * the Asterisk project. Please do not directly contact
11  * any of the maintainers of this project for assistance;
12  * the project provides a web site, mailing lists and IRC
13  * channels for your use.
14  *
15  * This program is free software, distributed under the terms of
16  * the GNU General Public License Version 2. See the LICENSE file
17  * at the top of the source tree.
18  */
19
20 /*! \file
21  * 
22  * \brief file format conversion CLI command using Asterisk formats and translators
23  *
24  * \author redice li <redice_li@yahoo.com>
25  * \author Russell Bryant <russell@digium.com>
26  *
27  */ 
28
29 #include "asterisk.h"
30
31 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
32
33 #include "asterisk/channel.h"
34 #include "asterisk/logger.h"
35 #include "asterisk/module.h"
36 #include "asterisk/cli.h"
37 #include "asterisk/file.h"
38
39 /*! \brief Split the filename to basename and extension */
40 static int split_ext(char *filename, char **name, char **ext)
41 {
42         *name = *ext = filename;
43         
44         if ((*ext = strrchr(filename, '.'))) {
45                 **ext = '\0';
46                 (*ext)++;
47         }
48
49         if (ast_strlen_zero(*name) || ast_strlen_zero(*ext))
50                 return -1;
51
52         return 0;
53 }
54
55 /*! 
56  * \brief Convert a file from one format to another 
57  * \param e CLI entry
58  * \param cmd command number
59  * \param a list of cli arguments
60  * \retval CLI_SUCCESS on success.
61  * \retval CLI_SHOWUSAGE or CLI_FAILURE on failure.
62 */
63 static char *handle_cli_file_convert(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
64 {
65         char *ret = CLI_FAILURE;
66         struct ast_filestream *fs_in = NULL, *fs_out = NULL;
67         struct ast_frame *f;
68         struct timeval start;
69         int cost;
70         char *file_in = NULL, *file_out = NULL;
71         char *name_in, *ext_in, *name_out, *ext_out;
72
73         switch (cmd) {
74         case CLI_INIT:
75                 e->command = "file convert";
76                 e->usage =
77                         "Usage: file convert <file_in> <file_out>\n"
78                         "       Convert from file_in to file_out. If an absolute path\n"
79                         "       is not given, the default Asterisk sounds directory\n"
80                         "       will be used.\n\n"
81                         "       Example:\n"
82                         "           file convert tt-weasels.gsm tt-weasels.ulaw\n";
83                 return NULL;
84         case CLI_GENERATE:
85                 return NULL;
86         }
87         
88         /* ugly, can be removed when CLI entries have ast_module pointers */
89         ast_module_ref(ast_module_info->self);
90
91         if (a->argc != 4 || ast_strlen_zero(a->argv[2]) || ast_strlen_zero(a->argv[3])) {
92                 ret = CLI_SHOWUSAGE;
93                 goto fail_out;  
94         }
95
96         file_in = ast_strdupa(a->argv[2]);
97         file_out = ast_strdupa(a->argv[3]);
98
99         if (split_ext(file_in, &name_in, &ext_in)) {
100                 ast_cli(a->fd, "'%s' is an invalid filename!\n", a->argv[2]);
101                 goto fail_out;
102         }
103         if (!(fs_in = ast_readfile(name_in, ext_in, NULL, O_RDONLY, 0, 0))) {
104                 ast_cli(a->fd, "Unable to open input file: %s\n", a->argv[2]);
105                 goto fail_out;
106         }
107         
108         if (split_ext(file_out, &name_out, &ext_out)) {
109                 ast_cli(a->fd, "'%s' is an invalid filename!\n", a->argv[3]);
110                 goto fail_out;
111         }
112         if (!(fs_out = ast_writefile(name_out, ext_out, NULL, O_CREAT|O_TRUNC|O_WRONLY, 0, AST_FILE_MODE))) {
113                 ast_cli(a->fd, "Unable to open output file: %s\n", a->argv[3]);
114                 goto fail_out;
115         }
116
117         start = ast_tvnow();
118         
119         while ((f = ast_readframe(fs_in))) {
120                 if (ast_writestream(fs_out, f)) {
121                         ast_cli(a->fd, "Failed to convert %s.%s to %s.%s!\n", name_in, ext_in, name_out, ext_out);
122                         goto fail_out;
123                 }
124         }
125
126         cost = ast_tvdiff_ms(ast_tvnow(), start);
127         ast_cli(a->fd, "Converted %s.%s to %s.%s in %dms\n", name_in, ext_in, name_out, ext_out, cost);
128         ret = CLI_SUCCESS;
129
130 fail_out:
131         if (fs_out) {
132                 ast_closestream(fs_out);
133                 if (ret != CLI_SUCCESS)
134                         ast_filedelete(name_out, ext_out);
135         }
136
137         if (fs_in) 
138                 ast_closestream(fs_in);
139
140         ast_module_unref(ast_module_info->self);
141
142         return ret;
143 }
144
145 static struct ast_cli_entry cli_convert[] = {
146         AST_CLI_DEFINE(handle_cli_file_convert, "Convert audio file")
147 };
148
149 static int unload_module(void)
150 {
151         ast_cli_unregister_multiple(cli_convert, sizeof(cli_convert) / sizeof(struct ast_cli_entry));
152         return 0;
153 }
154
155 static int load_module(void)
156 {
157         ast_cli_register_multiple(cli_convert, sizeof(cli_convert) / sizeof(struct ast_cli_entry));
158         return AST_MODULE_LOAD_SUCCESS;
159 }
160
161 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "File format conversion CLI command");