Fix a bug in the CLI reverbification, as pointed out by ZX81 in #asterisk-dev
[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 <stdio.h>
34 #include <string.h>
35 #include <stdlib.h>
36
37 #include "asterisk/channel.h"
38 #include "asterisk/logger.h"
39 #include "asterisk/module.h"
40 #include "asterisk/cli.h"
41 #include "asterisk/file.h"
42
43 /*! \brief Split the filename to basename and extension */
44 static int split_ext(char *filename, char **name, char **ext)
45 {
46         *name = *ext = filename;
47         
48         strsep(ext, ".");
49
50         if (ast_strlen_zero(*name) || ast_strlen_zero(*ext))
51                 return -1;
52
53         return 0;
54 }
55
56 /*! \brief Convert a file from one format to another */
57 static int cli_audio_convert_deprecated(int fd, int argc, char *argv[])
58 {
59         int ret = RESULT_FAILURE;
60         struct ast_filestream *fs_in = NULL, *fs_out = NULL;
61         struct ast_frame *f;
62         struct timeval start;
63         int cost;
64         char *file_in = NULL, *file_out = NULL;
65         char *name_in, *ext_in, *name_out, *ext_out;
66         
67         /* ugly, can be removed when CLI entries have ast_module pointers */
68         ast_module_ref(ast_module_info->self);
69
70         if (argc != 3 || ast_strlen_zero(argv[1]) || ast_strlen_zero(argv[2])) {
71                 ret = RESULT_SHOWUSAGE;
72                 goto fail_out;  
73         }
74
75         file_in = ast_strdupa(argv[1]);
76         file_out = ast_strdupa(argv[2]);
77
78         if (split_ext(file_in, &name_in, &ext_in)) {
79                 ast_cli(fd, "'%s' is an invalid filename!\n", argv[1]);
80                 goto fail_out;
81         }
82         if (!(fs_in = ast_readfile(name_in, ext_in, NULL, O_RDONLY, 0, 0))) {
83                 ast_cli(fd, "Unable to open input file: %s\n", argv[1]);
84                 goto fail_out;
85         }
86         
87         if (split_ext(file_out, &name_out, &ext_out)) {
88                 ast_cli(fd, "'%s' is an invalid filename!\n", argv[2]);
89                 goto fail_out;
90         }
91         if (!(fs_out = ast_writefile(name_out, ext_out, NULL, O_CREAT|O_TRUNC|O_WRONLY, 0, 0644))) {
92                 ast_cli(fd, "Unable to open output file: %s\n", argv[2]);
93                 goto fail_out;
94         }
95
96         start = ast_tvnow();
97         
98         while ((f = ast_readframe(fs_in))) {
99                 if (ast_writestream(fs_out, f)) {
100                         ast_cli(fd, "Failed to convert %s.%s to %s.%s!\n", name_in, ext_in, name_out, ext_out);
101                         goto fail_out;
102                 }
103         }
104
105         cost = ast_tvdiff_ms(ast_tvnow(), start);
106         ast_cli(fd, "Converted %s.%s to %s.%s in %dms\n", name_in, ext_in, name_out, ext_out, cost);
107         ret = RESULT_SUCCESS;
108
109 fail_out:
110         if (fs_out) {
111                 ast_closestream(fs_out);
112                 if (ret != RESULT_SUCCESS)
113                         ast_filedelete(name_out, ext_out);
114         }
115
116         if (fs_in) 
117                 ast_closestream(fs_in);
118
119         ast_module_unref(ast_module_info->self);
120
121         return ret;
122 }
123
124 static int cli_audio_convert(int fd, int argc, char *argv[])
125 {
126         int ret = RESULT_FAILURE;
127         struct ast_filestream *fs_in = NULL, *fs_out = NULL;
128         struct ast_frame *f;
129         struct timeval start;
130         int cost;
131         char *file_in = NULL, *file_out = NULL;
132         char *name_in, *ext_in, *name_out, *ext_out;
133         
134         /* ugly, can be removed when CLI entries have ast_module pointers */
135         ast_module_ref(ast_module_info->self);
136
137         if (argc != 4 || ast_strlen_zero(argv[2]) || ast_strlen_zero(argv[3])) {
138                 ret = RESULT_SHOWUSAGE;
139                 goto fail_out;  
140         }
141
142         file_in = ast_strdupa(argv[2]);
143         file_out = ast_strdupa(argv[3]);
144
145         if (split_ext(file_in, &name_in, &ext_in)) {
146                 ast_cli(fd, "'%s' is an invalid filename!\n", argv[2]);
147                 goto fail_out;
148         }
149         if (!(fs_in = ast_readfile(name_in, ext_in, NULL, O_RDONLY, 0, 0))) {
150                 ast_cli(fd, "Unable to open input file: %s\n", argv[2]);
151                 goto fail_out;
152         }
153         
154         if (split_ext(file_out, &name_out, &ext_out)) {
155                 ast_cli(fd, "'%s' is an invalid filename!\n", argv[3]);
156                 goto fail_out;
157         }
158         if (!(fs_out = ast_writefile(name_out, ext_out, NULL, O_CREAT|O_TRUNC|O_WRONLY, 0, 0644))) {
159                 ast_cli(fd, "Unable to open output file: %s\n", argv[3]);
160                 goto fail_out;
161         }
162
163         start = ast_tvnow();
164         
165         while ((f = ast_readframe(fs_in))) {
166                 if (ast_writestream(fs_out, f)) {
167                         ast_cli(fd, "Failed to convert %s.%s to %s.%s!\n", name_in, ext_in, name_out, ext_out);
168                         goto fail_out;
169                 }
170         }
171
172         cost = ast_tvdiff_ms(ast_tvnow(), start);
173         ast_cli(fd, "Converted %s.%s to %s.%s in %dms\n", name_in, ext_in, name_out, ext_out, cost);
174         ret = RESULT_SUCCESS;
175
176 fail_out:
177         if (fs_out) {
178                 ast_closestream(fs_out);
179                 if (ret != RESULT_SUCCESS)
180                         ast_filedelete(name_out, ext_out);
181         }
182
183         if (fs_in) 
184                 ast_closestream(fs_in);
185
186         ast_module_unref(ast_module_info->self);
187
188         return ret;
189 }
190
191 static char usage_audio_convert[] =
192 "Usage: file convert <file_in> <file_out>\n"
193 "    Convert from file_in to file_out. If an absolute path is not given, the\n"
194 "default Asterisk sounds directory will be used.\n\n"
195 "Example:\n"
196 "    file convert tt-weasels.gsm tt-weasels.ulaw\n";
197
198 static struct ast_cli_entry cli_convert_deprecated = {
199         { "convert" , NULL },
200         cli_audio_convert_deprecated, NULL,
201         NULL };
202
203 static struct ast_cli_entry cli_convert[] = {
204         { { "file", "convert" , NULL },
205         cli_audio_convert, "Convert audio file",
206         usage_audio_convert, NULL, &cli_convert_deprecated },
207 };
208
209 static int unload_module(void)
210 {
211         ast_cli_unregister_multiple(cli_convert, sizeof(cli_convert) / sizeof(struct ast_cli_entry));
212         return 0;
213 }
214
215 static int load_module(void)
216 {
217         ast_cli_register_multiple(cli_convert, sizeof(cli_convert) / sizeof(struct ast_cli_entry));
218         return 0;
219 }
220
221 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "File format conversion CLI command");