1 /* metaflac - Command-line FLAC metadata editor
2 * Copyright (C) 2001,2002,2003,2004,2005,2006,2007,2008,2009 Josh Coalson
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
25 #include "FLAC/assert.h"
26 #include "share/grabbag.h" /* for grabbag__file_get_filesize() */
27 #include "share/utf8.h"
31 #include "operations_shorthand.h"
33 static FLAC__bool
remove_vc_all(const char *filename
, FLAC__StreamMetadata
*block
, FLAC__bool
*needs_write
);
34 static FLAC__bool
remove_vc_field(const char *filename
, FLAC__StreamMetadata
*block
, const char *field_name
, FLAC__bool
*needs_write
);
35 static FLAC__bool
remove_vc_firstfield(const char *filename
, FLAC__StreamMetadata
*block
, const char *field_name
, FLAC__bool
*needs_write
);
36 static FLAC__bool
set_vc_field(const char *filename
, FLAC__StreamMetadata
*block
, const Argument_VcField
*field
, FLAC__bool
*needs_write
, FLAC__bool raw
);
37 static FLAC__bool
import_vc_from(const char *filename
, FLAC__StreamMetadata
*block
, const Argument_String
*vc_filename
, FLAC__bool
*needs_write
, FLAC__bool raw
);
38 static FLAC__bool
export_vc_to(const char *filename
, FLAC__StreamMetadata
*block
, const Argument_String
*vc_filename
, FLAC__bool raw
);
40 FLAC__bool
do_shorthand_operation__vorbis_comment(const char *filename
, FLAC__bool prefix_with_filename
, FLAC__Metadata_Chain
*chain
, const Operation
*operation
, FLAC__bool
*needs_write
, FLAC__bool raw
)
42 FLAC__bool ok
= true, found_vc_block
= false;
43 FLAC__StreamMetadata
*block
= 0;
44 FLAC__Metadata_Iterator
*iterator
= FLAC__metadata_iterator_new();
47 die("out of memory allocating iterator");
49 FLAC__metadata_iterator_init(iterator
, chain
);
52 block
= FLAC__metadata_iterator_get_block(iterator
);
53 if(block
->type
== FLAC__METADATA_TYPE_VORBIS_COMMENT
)
54 found_vc_block
= true;
55 } while(!found_vc_block
&& FLAC__metadata_iterator_next(iterator
));
58 /* create a new block if necessary */
59 if(operation
->type
== OP__SET_VC_FIELD
|| operation
->type
== OP__IMPORT_VC_FROM
) {
60 block
= FLAC__metadata_object_new(FLAC__METADATA_TYPE_VORBIS_COMMENT
);
62 die("out of memory allocating VORBIS_COMMENT block");
63 while(FLAC__metadata_iterator_next(iterator
))
65 if(!FLAC__metadata_iterator_insert_block_after(iterator
, block
)) {
66 print_error_with_chain_status(chain
, "%s: ERROR: adding new VORBIS_COMMENT block to metadata", filename
);
69 /* iterator is left pointing to new block */
70 FLAC__ASSERT(FLAC__metadata_iterator_get_block(iterator
) == block
);
73 FLAC__metadata_iterator_delete(iterator
);
78 FLAC__ASSERT(0 != block
);
79 FLAC__ASSERT(block
->type
== FLAC__METADATA_TYPE_VORBIS_COMMENT
);
81 switch(operation
->type
) {
82 case OP__SHOW_VC_VENDOR
:
83 write_vc_field(prefix_with_filename
? filename
: 0, &block
->data
.vorbis_comment
.vendor_string
, raw
, stdout
);
85 case OP__SHOW_VC_FIELD
:
86 write_vc_fields(prefix_with_filename
? filename
: 0, operation
->argument
.vc_field_name
.value
, block
->data
.vorbis_comment
.comments
, block
->data
.vorbis_comment
.num_comments
, raw
, stdout
);
88 case OP__REMOVE_VC_ALL
:
89 ok
= remove_vc_all(filename
, block
, needs_write
);
91 case OP__REMOVE_VC_FIELD
:
92 ok
= remove_vc_field(filename
, block
, operation
->argument
.vc_field_name
.value
, needs_write
);
94 case OP__REMOVE_VC_FIRSTFIELD
:
95 ok
= remove_vc_firstfield(filename
, block
, operation
->argument
.vc_field_name
.value
, needs_write
);
97 case OP__SET_VC_FIELD
:
98 ok
= set_vc_field(filename
, block
, &operation
->argument
.vc_field
, needs_write
, raw
);
100 case OP__IMPORT_VC_FROM
:
101 ok
= import_vc_from(filename
, block
, &operation
->argument
.filename
, needs_write
, raw
);
103 case OP__EXPORT_VC_TO
:
104 ok
= export_vc_to(filename
, block
, &operation
->argument
.filename
, raw
);
112 FLAC__metadata_iterator_delete(iterator
);
120 FLAC__bool
remove_vc_all(const char *filename
, FLAC__StreamMetadata
*block
, FLAC__bool
*needs_write
)
122 FLAC__ASSERT(0 != block
);
123 FLAC__ASSERT(block
->type
== FLAC__METADATA_TYPE_VORBIS_COMMENT
);
124 FLAC__ASSERT(0 != needs_write
);
126 if(0 != block
->data
.vorbis_comment
.comments
) {
127 FLAC__ASSERT(block
->data
.vorbis_comment
.num_comments
> 0);
128 if(!FLAC__metadata_object_vorbiscomment_resize_comments(block
, 0)) {
129 fprintf(stderr
, "%s: ERROR: memory allocation failure\n", filename
);
135 FLAC__ASSERT(block
->data
.vorbis_comment
.num_comments
== 0);
141 FLAC__bool
remove_vc_field(const char *filename
, FLAC__StreamMetadata
*block
, const char *field_name
, FLAC__bool
*needs_write
)
145 FLAC__ASSERT(0 != needs_write
);
147 n
= FLAC__metadata_object_vorbiscomment_remove_entries_matching(block
, field_name
);
150 fprintf(stderr
, "%s: ERROR: memory allocation failure\n", filename
);
159 FLAC__bool
remove_vc_firstfield(const char *filename
, FLAC__StreamMetadata
*block
, const char *field_name
, FLAC__bool
*needs_write
)
163 FLAC__ASSERT(0 != needs_write
);
165 n
= FLAC__metadata_object_vorbiscomment_remove_entry_matching(block
, field_name
);
168 fprintf(stderr
, "%s: ERROR: memory allocation failure\n", filename
);
177 FLAC__bool
set_vc_field(const char *filename
, FLAC__StreamMetadata
*block
, const Argument_VcField
*field
, FLAC__bool
*needs_write
, FLAC__bool raw
)
179 FLAC__StreamMetadata_VorbisComment_Entry entry
;
182 FLAC__ASSERT(0 != block
);
183 FLAC__ASSERT(block
->type
== FLAC__METADATA_TYPE_VORBIS_COMMENT
);
184 FLAC__ASSERT(0 != field
);
185 FLAC__ASSERT(0 != needs_write
);
187 if(field
->field_value_from_file
) {
188 /* read the file into 'data' */
191 const off_t size
= grabbag__file_get_filesize(field
->field_value
);
193 fprintf(stderr
, "%s: ERROR: can't open file '%s' for '%s' tag value\n", filename
, field
->field_value
, field
->field_name
);
196 if(size
>= 0x100000) { /* magic arbitrary limit, actual format limit is near 16MB */
197 fprintf(stderr
, "%s: ERROR: file '%s' for '%s' tag value is too large\n", filename
, field
->field_value
, field
->field_name
);
200 if(0 == (data
= malloc(size
+1)))
201 die("out of memory allocating tag value");
203 if(0 == (f
= fopen(field
->field_value
, "rb")) || fread(data
, 1, size
, f
) != (size_t)size
) {
204 fprintf(stderr
, "%s: ERROR: while reading file '%s' for '%s' tag value: %s\n", filename
, field
->field_value
, field
->field_name
, strerror(errno
));
211 if(strlen(data
) != (size_t)size
) {
213 fprintf(stderr
, "%s: ERROR: file '%s' for '%s' tag value has embedded NULs\n", filename
, field
->field_value
, field
->field_name
);
217 /* move 'data' into 'converted', converting to UTF-8 if necessary */
221 else if(utf8_encode(data
, &converted
) >= 0) {
226 fprintf(stderr
, "%s: ERROR: converting file '%s' contents to UTF-8 for tag value\n", filename
, field
->field_value
);
230 /* create and entry and append it */
231 if(!FLAC__metadata_object_vorbiscomment_entry_from_name_value_pair(&entry
, field
->field_name
, converted
)) {
233 fprintf(stderr
, "%s: ERROR: file '%s' for '%s' tag value is not valid UTF-8\n", filename
, field
->field_value
, field
->field_name
);
237 if(!FLAC__metadata_object_vorbiscomment_append_comment(block
, entry
, /*copy=*/false)) {
238 fprintf(stderr
, "%s: ERROR: memory allocation failure\n", filename
);
246 FLAC__bool needs_free
= false;
248 entry
.entry
= (FLAC__byte
*)field
->field
;
250 else if(utf8_encode(field
->field
, &converted
) >= 0) {
251 entry
.entry
= (FLAC__byte
*)converted
;
255 fprintf(stderr
, "%s: ERROR: converting comment '%s' to UTF-8\n", filename
, field
->field
);
258 entry
.length
= strlen((const char *)entry
.entry
);
259 if(!FLAC__format_vorbiscomment_entry_is_legal(entry
.entry
, entry
.length
)) {
263 * our previous parsing has already established that the field
264 * name is OK, so it must be the field value
266 fprintf(stderr
, "%s: ERROR: tag value for '%s' is not valid UTF-8\n", filename
, field
->field_name
);
270 if(!FLAC__metadata_object_vorbiscomment_append_comment(block
, entry
, /*copy=*/true)) {
273 fprintf(stderr
, "%s: ERROR: memory allocation failure\n", filename
);
284 FLAC__bool
import_vc_from(const char *filename
, FLAC__StreamMetadata
*block
, const Argument_String
*vc_filename
, FLAC__bool
*needs_write
, FLAC__bool raw
)
290 if(0 == vc_filename
->value
|| strlen(vc_filename
->value
) == 0) {
291 fprintf(stderr
, "%s: ERROR: empty import file name\n", filename
);
294 if(0 == strcmp(vc_filename
->value
, "-"))
297 f
= fopen(vc_filename
->value
, "r");
300 fprintf(stderr
, "%s: ERROR: can't open import file %s: %s\n", filename
, vc_filename
->value
, strerror(errno
));
305 while(ret
&& !feof(f
) && fgets(line
, sizeof(line
), f
) != NULL
) {
307 char *p
= strchr(line
, '\n');
309 fprintf(stderr
, "%s: ERROR: line too long, aborting\n", vc_filename
->value
);
313 const char *violation
;
314 Argument_VcField field
;
316 memset(&field
, 0, sizeof(Argument_VcField
));
317 field
.field_value_from_file
= false;
318 if(!parse_vorbis_comment_field(line
, &field
.field
, &field
.field_name
, &field
.field_value
, &field
.field_value_length
, &violation
)) {
319 FLAC__ASSERT(0 != violation
);
320 fprintf(stderr
, "%s: ERROR: malformed vorbis comment field \"%s\",\n %s\n", vc_filename
->value
, line
, violation
);
324 ret
= set_vc_field(filename
, block
, &field
, needs_write
, raw
);
328 if(0 != field
.field_name
)
329 free(field
.field_name
);
330 if(0 != field
.field_value
)
331 free(field
.field_value
);
341 FLAC__bool
export_vc_to(const char *filename
, FLAC__StreamMetadata
*block
, const Argument_String
*vc_filename
, FLAC__bool raw
)
346 if(0 == vc_filename
->value
|| strlen(vc_filename
->value
) == 0) {
347 fprintf(stderr
, "%s: ERROR: empty export file name\n", filename
);
350 if(0 == strcmp(vc_filename
->value
, "-"))
353 f
= fopen(vc_filename
->value
, "w");
356 fprintf(stderr
, "%s: ERROR: can't open export file %s: %s\n", filename
, vc_filename
->value
, strerror(errno
));
362 write_vc_fields(0, 0, block
->data
.vorbis_comment
.comments
, block
->data
.vorbis_comment
.num_comments
, raw
, f
);