From 823a4064ebcd2662f8e480f99326ac75dd6404a7 Mon Sep 17 00:00:00 2001 From: Ian Nartowicz Date: Wed, 4 Jun 2014 22:45:30 +0100 Subject: Full release of oggedit --- plugins/liboggedit/Makefile.am | 3 +- plugins/liboggedit/oggedit.c | 1212 --------------------------------- plugins/liboggedit/oggedit.h | 18 +- plugins/liboggedit/oggedit_art.c | 172 +++++ plugins/liboggedit/oggedit_flac.c | 213 ++++++ plugins/liboggedit/oggedit_internal.c | 524 ++++++++++++++ plugins/liboggedit/oggedit_internal.h | 59 ++ plugins/liboggedit/oggedit_opus.c | 197 ++++++ plugins/liboggedit/oggedit_utils.c | 125 ++++ plugins/liboggedit/oggedit_vorbis.c | 168 +++++ 10 files changed, 1471 insertions(+), 1220 deletions(-) delete mode 100644 plugins/liboggedit/oggedit.c create mode 100644 plugins/liboggedit/oggedit_art.c create mode 100644 plugins/liboggedit/oggedit_flac.c create mode 100644 plugins/liboggedit/oggedit_internal.c create mode 100644 plugins/liboggedit/oggedit_internal.h create mode 100644 plugins/liboggedit/oggedit_opus.c create mode 100644 plugins/liboggedit/oggedit_utils.c create mode 100644 plugins/liboggedit/oggedit_vorbis.c (limited to 'plugins/liboggedit') diff --git a/plugins/liboggedit/Makefile.am b/plugins/liboggedit/Makefile.am index cdf59124..2134718f 100644 --- a/plugins/liboggedit/Makefile.am +++ b/plugins/liboggedit/Makefile.am @@ -1,5 +1,6 @@ if HAVE_VORBIS noinst_LIBRARIES = liboggedit.a -liboggedit_a_SOURCES = oggedit.c +liboggedit_a_SOURCES = oggedit_internal.h oggedit.h \ + oggedit_internal.c oggedit_utils.c oggedit_art.c oggedit_opus.c oggedit_vorbis.c oggedit_flac.c AM_CFLAGS = $(VORBIS_CFLAGS) -fPIC -std=c99 endif diff --git a/plugins/liboggedit/oggedit.c b/plugins/liboggedit/oggedit.c deleted file mode 100644 index 66c4a358..00000000 --- a/plugins/liboggedit/oggedit.c +++ /dev/null @@ -1,1212 +0,0 @@ -/* - This file is part of Deadbeef Player source code - http://deadbeef.sourceforge.net - - DeaDBeeF Ogg Edit library - - Copyright (C) 2014 Ian Nartowicz - - This software is provided 'as-is', without any express or implied - warranty. In no event will the authors be held liable for any damages - arising from the use of this software. - - Permission is granted to anyone to use this software for any purpose, - including commercial applications, and to alter it and redistribute it - freely, subject to the following restrictions: - - 1. The origin of this software must not be misrepresented; you must not - claim that you wrote the original software. If you use this software - in a product, an acknowledgment in the product documentation would be - appreciated but is not required. - 2. Altered source versions must be plainly marked as such, and must not be - misrepresented as being the original software. - 3. This notice may not be removed or altered from any source distribution. - -*/ - -#ifdef HAVE_CONFIG_H -#include -#endif -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include "../../deadbeef.h" -#include "oggedit.h" - -#define CHUNKSIZE 4096 -#define MAXPAGE 65536 -#define MAXPAYLOAD 65025 - -/**************************** - * Ogg - ****************************/ -uint8_t *oggedit_vorbis_channel_map(const int channel_count) -{ - size_t map_size = channel_count * sizeof(uint8_t); - uint8_t *map = malloc(map_size); - if (!map) - return NULL; - switch(channel_count) { - case 3: - return memcpy(map, &(uint8_t[]){0,2,1}, map_size); - case 5: - return memcpy(map, &(uint8_t[]){0,2,1,3,4}, map_size); - case 6: - return memcpy(map, &(uint8_t[]){0,2,1,4,5,3}, map_size); - case 7: - return memcpy(map, &(uint8_t[]){0,2,1,4,5,6,3}, map_size); - case 8: - return memcpy(map, &(uint8_t[]){0,2,1,6,7,4,5,3}, map_size); - default: - free(map); - return NULL; - } -} - -const char *oggedit_map_tag(char *key, const char *in_or_out) -{ - typedef struct { - const char *tag; - const char *meta; - } key_t; - const key_t keys[] = { - /* Permanent named tags in DeaDBeef */ -// {.tag = "ARTIST", .meta = "artist"}, -// {.tag = "TITLE", .meta = "title"}, -// {.tag = "ALBUM", .meta = "album"}, - {.tag = "DATE", .meta = "year"}, - {.tag = "TRACKNUMBER", .meta = "track"}, - {.tag = "TRACKTOTAL", .meta = "numtracks"}, -// {.tag = "GENRE", .meta = "genre"}, -// {.tag = "COMPOSER", .meta = "composer"}, - {.tag = "DISCNUMBER", .meta = "disc"}, -// {.tag = "COMMENT", .meta = "comment"}, - /* Vorbis standard tags */ -// {.tag = "ARRANGER", .meta = "Arranger"}, -// {.tag = "AUTHOR", .meta = "Author"}, -// {.tag = "CONDUCTOR", .meta = "Conductor"}, -// {.tag = "ENSEMBLE", .meta = "Ensemble"}, -// {.tag = "LYRICIST", .meta = "Lyricist"}, -// {.tag = "PERFORMER", .meta = "Performer"}, -// {.tag = "PUBLISHER", .meta = "Publisher"}, -// {.tag = "DISCTOTAL", .meta = "Disctotal"}, -// {.tag = "OPUS", .meta = "Opus"}, -// {.tag = "PART", .meta = "Part"}, -// {.tag = "PARTNUMBER", .meta = "Partnumber"}, -// {.tag = "VERSION", .meta = "Version"}, -// {.tag = "DESCRIPTION", .meta = "Description"}, -// {.tag = "COPYRIGHT", .meta = "Copyright"}, -// {.tag = "LICENSE", .meta = "License"}, -// {.tag = "CONTACT", .meta = "Contact"}, -// {.tag = "ORGANIZATION", .meta = "Organization"}, -// {.tag = "LOCATION", .meta = "Location"}, -// {.tag = "EAN/UPN", .meta = "EAN/UPN"}, -// {.tag = "ISRC", .meta = "ISRC"}, -// {.tag = "LABEL", .meta = "Label"}, -// {.tag = "LABELNO", .meta = "Labelno"}, -// {.tag = "ENCODER", .meta = "Encoder"}, -// {.tag = "ENCODED-BY", .meta = "Encoded-by"}, -// {.tag = "ENCODING", .meta = "Encoding"}, - /* Other tags */ -// {.tag = "ALBUMARTIST", .meta = "Albumartist"}, -// {.tag = "ALBUM ARTIST", .meta = "Album artist"}, -// {.tag = "BAND", .meta = "Band"}, -// {.tag = "COMPILATION", .meta = "Compilation"}, -// {.tag = "TOTALTRACKS", .meta = "Totaltracks"}, -// {.tag = "ENCODED_BY", .meta = "Encoded_by"}, -// {.tag = "ENCODER_OPTIONS",.meta = "Encoder_options"}, - {.tag = NULL} - }; - - /* Mapping for special Deadbeef internal metadata */ - for (const key_t *match = keys; match->tag; match++) - if (!strcasecmp(*in_or_out == 't' ? match->tag : match->meta, key)) - return *in_or_out == 't' ? match->meta : match->tag; - - /* Upper-case all Vorbis Comment tag names */ - if (*in_or_out == 'm') - for (size_t i = 0; key[i]; i++) - key[i] = toupper(key[i]); - - return key; -} - -const char *oggedit_album_art_type(const int type) -{ - switch (type) { - case 1: - return "32x32 pixels file icon"; - case 2: - return "other file icon"; - case 3: - return "front cover"; - case 4: - return "back cover"; - case 5: - return "leaflet page"; - case 6: - return "media"; - case 7: - return "lead artist/lead performer/soloist"; - case 8: - return "artist/performer"; - case 9: - return "conductor"; - case 10: - return "band/orchestra"; - case 11: - return "composer"; - case 12: - return "lyricist/text writer"; - case 13: - return "recording location"; - case 14: - return "during recording"; - case 15: - return "during performance"; - case 16: - return "movie/video screen capture"; - case 17: - return "bright coloured fish"; - case 18: - return "illustration"; - case 19: - return "band/artist logotype"; - case 20: - return "publisher/studio logotype"; - default: - return "other"; - } -} - -static char *cat_string(char *dest, const char *src, const char *sep) -{ - char *more = realloc(dest, strlen(dest) + strlen(src) + strlen(sep) + 1); - if (!more) { - free(dest); - return NULL; - } - return strcat(strcat(more, sep), src); -} - -static const char *codec_name(ogg_page *og) -{ - typedef struct { - const size_t length; - const char *codec; - const char *magic; - } codec_t; - const codec_t codecs[] = { - {.codec = OPUSNAME, .length = 19, .magic = "OpusHead"}, - {.codec = VORBISNAME, .length = 30, .magic = "\1vorbis"}, - {.codec = FLACNAME, .length = 47, .magic = "\177FLAC"}, - {.codec = "Speex", .length = 80, .magic = "Speex "}, - {.codec = "Celt", .length = 80, .magic = "CELT"}, // obsolete - {.codec = "MIDI", .length = 13, .magic = "OggMIDI\0"}, - {.codec = "PCM", .length = 28, .magic = "PCM "}, - {.codec = "Theora", .length = 42, .magic = "\200theora"}, - {.codec = "Daala", .length = 38, .magic = "\200daala"}, - {.codec = "Dirac", .length = 5, .magic = "BBCD\0"}, - {.codec = "Skeleton", .length = 80, .magic = "fishead\0"}, - {.codec = "Kate", .length = 64, .magic = "\200kate\0\0\0"}, - {.codec = "CMML", .length = 29, .magic = "CMML\0\0\0\0"}, - {.codec = "YUV4MPEG", .length = 8, .magic = "YUV4Mpeg"}, - {.codec = "UVS", .length = 48, .magic = "UVS "}, - {.codec = "YUV", .length = 32, .magic = "\0YUV"}, - {.codec = "RGB", .length = 24, .magic = "\0RGB"}, - {.codec = "JNG", .length = 48, .magic = "\213JNG\r\n\032\n"}, - {.codec = "MNG", .length = 48, .magic = "\212MNG\r\n\032\n"}, - {.codec = "PNG", .length = 48, .magic = "\211PNG\r\n\032\n"}, - {.codec = "Spots", .length = 52, .magic = "SPOTS\0\0\0"}, - {.codec = NULL} - }; - - for (const codec_t *match = codecs; match->codec; match++) - if ((size_t)og->body_len >= match->length && !memcmp(og->body, match->magic, strlen(match->codec))) - return match->codec; - - return "unknown"; -} - -static char *btoa(const unsigned char *binary, const size_t binary_length) -{ - const char b64[64] = { - 'A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z', - 'a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z', - '0','1','2','3','4','5','6','7','8','9','+','/' - }; - - char *ascii_base64 = malloc((binary_length-1)/3 * 4 + 5); - if (!ascii_base64) - return NULL; - - char *p = ascii_base64; - const unsigned char *q = binary; - const unsigned char *binary_end = binary + binary_length; - while (q+2 < binary_end) { - uint32_t chunk = q[0]<<16 | q[1]<<8 | q[2]; - p[3] = b64[chunk & 0x3f]; - p[2] = b64[chunk>>6 & 0x3f]; - p[1] = b64[chunk>>12 & 0x3f]; - p[0] = b64[chunk>>18 & 0x3f]; - p+=4, q+=3; - } - - if (q < binary_end) { - uint16_t remainder = q[0]<<8 | (q+1 == binary_end ? '\0' : q[1]); - p[3] = '='; - p[2] = q+1 == binary_end ? '=' : b64[remainder<<2 & 0x3f]; - p[1] = b64[remainder>>4 & 0x3f]; - p[0] = b64[remainder>>10 & 0x3f]; - p+=4; - } - *p = '\0'; - - return ascii_base64; -} - -static void _oggpack_chars(oggpack_buffer *opb, const char *s, size_t length) -{ -// oggpack_writecopy(opb, s, length * 8); currently broken - while (length--) - oggpack_write(opb, *s++, 8); -} - -static void _oggpack_string(oggpack_buffer *opb, const char *s) -{ - oggpack_write(opb, strlen(s), 32); - _oggpack_chars(opb, s, strlen(s)); -} - -static void _oggpackB_string(oggpack_buffer *opb, const char *s) -{ - oggpackB_write(opb, strlen(s), 32); - _oggpack_chars(opb, s, strlen(s)); -} - -char *oggedit_album_art_tag(DB_FILE *fp, int *res) -{ - if (!fp) { - *res = OGGEDIT_FILE_NOT_OPEN; - return NULL; - } - - const int64_t data_length = fp->vfs->getlength(fp); - if (data_length < 50 || data_length > 10000000) { - fp->vfs->close(fp); - *res = OGGEDIT_BAD_FILE_LENGTH; - return NULL; - } - - char *data = malloc(data_length); - if (!data) { - fp->vfs->close(fp); - *res = OGGEDIT_ALLOCATION_FAILURE; - return NULL; - } - - const size_t data_read = fp->vfs->read(data, 1, data_length, fp); - fp->vfs->close(fp); - if (data_read != data_length) { - free(data); - *res = OGGEDIT_CANT_READ_IMAGE_FILE; - return NULL; - } - - oggpack_buffer opb; - oggpackB_writeinit(&opb); - oggpackB_write(&opb, 3, 32); - _oggpackB_string(&opb, memcmp(data, "\x89PNG\x0D\x0A\x1A\x0A", 8) ? "image/jpeg" : "image/png"); - _oggpackB_string(&opb, "Album art added from DeaDBeeF"); - oggpackB_write(&opb, 0, 32); - oggpackB_write(&opb, 0, 32); - oggpackB_write(&opb, 0, 32); - oggpackB_write(&opb, 0, 32); - oggpackB_write(&opb, data_length, 32); - _oggpack_chars(&opb, data, data_length); - free(data); - if (oggpack_writecheck(&opb)) { - *res = OGGEDIT_ALLOCATION_FAILURE; - return NULL; - } - - char *tag = btoa(oggpackB_get_buffer(&opb), oggpackB_bytes(&opb)); - oggpackB_writeclear(&opb); - return tag; -} - -static bool ensure_directory(const char *path) -{ - struct stat stat_struct; - if (!stat(path, &stat_struct)) - return !S_ISDIR(stat_struct.st_mode); - - if (errno != ENOENT) - return false; - - char* dir = strdup(path); - if (!dir) - return false; - - const int bad_dir = ensure_directory(dirname(dir)); - free(dir); - - return !bad_dir && !mkdir(path, 0777); -} - -static int finish_temp_file(const char *tempname, const char *fname) -{ - return rename(tempname, fname) ? OGGEDIT_RENAME_FAILED : 0; -} - -static int open_temp_file(const char *fname, char *tempname, FILE **out) -{ - snprintf(tempname, PATH_MAX, "%s.temp", fname); - unlink(tempname); - if (!(*out = freopen(tempname, "abx", *out))) - return OGGEDIT_CANNOT_OPEN_TEMPORARY_FILE; - - struct stat stat_struct; - if (!stat(fname, &stat_struct)) - chmod(tempname, stat_struct.st_mode); - - return 0; -} - -static FILE *open_new_file(const char *outname) -{ - char outpath[PATH_MAX]; - strcpy(outpath, outname); - if (!ensure_directory(dirname(outpath))) - return NULL; - - unlink(outname); - return fopen(outname, "wbx"); -} - -static off_t file_size(const char *fname) -{ - struct stat sb; - if (stat(fname, &sb)) - return OGGEDIT_STAT_FAILED; - - return sb.st_size; -} - -static void cleanup(DB_FILE *in, FILE *out, ogg_sync_state *oy, void *buffer) -{ - if (in) - in->vfs->close(in); - - if (out) - fclose(out); - - ogg_sync_clear(oy); - - if (buffer) - free(buffer); -} - -static int get_page(DB_FILE *in, ogg_sync_state *oy, ogg_page *og) -{ - uint16_t chunks_left = MAXPAGE / CHUNKSIZE; - while (ogg_sync_pageout(oy, og) != 1) { - char *buffer = ogg_sync_buffer(oy, CHUNKSIZE); - if (!in || !buffer || !chunks_left--) - return OGGEDIT_CANT_FIND_STREAM; - - const size_t bytes = in->vfs->read(buffer, 1, CHUNKSIZE, in); - if (!bytes) - return OGGEDIT_EOF; - - ogg_sync_wrote(oy, bytes); - } - - return ogg_page_serialno(og); -} - -static int skip_to_bos(DB_FILE *in, ogg_sync_state *oy, ogg_page *og, const off_t offset) -{ - if (!in) - return OGGEDIT_FILE_NOT_OPEN; - - if (in->vfs->seek(in, offset, SEEK_SET)) - return OGGEDIT_SEEK_FAILED; - - ogg_sync_reset(oy); - int serial; - do - serial = get_page(in, oy, og); - while (serial > OGGEDIT_EOF && !ogg_page_bos(og)); - - return serial; -} - -static int skip_to_codec(DB_FILE *in, ogg_sync_state *oy, ogg_page *og, const off_t offset, const char *codec) -{ - int serial = skip_to_bos(in, oy, og, offset); - while (serial > OGGEDIT_EOF && strcmp(codec_name(og), codec)) - serial = get_page(in, oy, og); - - return serial; -} - -static int skip_to_header(DB_FILE *in, ogg_sync_state *oy, ogg_page *og, int serial, const int codec_serial) -{ - while (serial > OGGEDIT_EOF && (ogg_page_bos(og) || serial != codec_serial)) - serial = get_page(in, oy, og); - - return serial; -} - -static off_t sync_tell(DB_FILE *in, ogg_sync_state *oy, ogg_page *og) -{ - return in->vfs->tell(in) - oy->fill + oy->returned - og->header_len - og->body_len; -} - -static bool write_page(FILE *out, ogg_page *og) -{ - if (fwrite(og->header, 1, og->header_len, out) != (size_t)og->header_len) - return false; - - if (fwrite(og->body, 1, og->body_len, out) != (size_t)og->body_len) - return false; - - return true; -} - -static int write_page_and_get_next(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og) -{ - if (!write_page(out, og)) - return OGGEDIT_WRITE_ERROR; - - return get_page(in, oy, og); -} - -static int copy_up_to_codec(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og, const off_t start_offset, const off_t link_offset, const char *codec) -{ - int serial = skip_to_bos(in, oy, og, start_offset); - - if (fseek(out, sync_tell(in, oy, og), SEEK_SET)) - return OGGEDIT_SEEK_FAILED; - - while (serial > OGGEDIT_EOF && (sync_tell(in, oy, og) < link_offset || !ogg_page_bos(og) || strcmp(codec_name(og), codec))) - serial = write_page_and_get_next(in, out, oy, og); - - return serial; -} - -static int copy_up_to_header(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og, const int codec_serial) -{ - int serial; - do - serial = write_page_and_get_next(in, out, oy, og); - while (serial > OGGEDIT_EOF && serial != codec_serial); - return serial; -} - -static long flush_stream(FILE *out, ogg_stream_state *os) -{ - ogg_page og; - while (ogg_stream_flush_fill(os, &og, MAXPAYLOAD)) - if (!write_page(out, &og)) - return OGGEDIT_WRITE_ERROR; - const long pageno = ogg_stream_check(os) ? OGGEDIT_FLUSH_FAILED : ogg_page_pageno(&og); - ogg_stream_clear(os); - return pageno; -} - -static char *codec_names(DB_FILE *in, ogg_sync_state *oy, const off_t link_offset, int *res) -{ - ogg_page og; - *res = skip_to_bos(in, oy, &og, link_offset); - char *codecs = strdup("Ogg"); - while (codecs && *res > OGGEDIT_EOF && ogg_page_bos(&og)) { - codecs = cat_string(codecs, codec_name(&og), strcmp(codecs, "Ogg") ? "/" : " "); - *res = get_page(in, oy, &og); - } - if (!*codecs) { - *res = OGGEDIT_ALLOCATION_FAILURE; - return NULL; - } - if (*res <= OGGEDIT_EOF) { - free(codecs); - return NULL; - } - - *res = OGGEDIT_OK; - return codecs; -} - -static off_t codec_stream_size(DB_FILE *in, ogg_sync_state *oy, const off_t start_offset, const off_t end_offset, const char *codec) -{ - /* Find codec serial and any other codecs */ - bool multiplex = false; - ogg_page og; - int codec_serial = -1; - int serial = skip_to_bos(in, oy, &og, start_offset); - while (serial > OGGEDIT_EOF && ogg_page_bos(&og)) { - if (strcmp(codec_name(&og), codec)) - multiplex = true; - else - codec_serial = serial; - serial = get_page(in, oy, &og); - } - - /* Skip to the first codec data page */ - while (serial > OGGEDIT_EOF && !(ogg_page_granulepos(&og) > 0 && serial == codec_serial)) - if ((serial = get_page(in, oy, &og)) <= OGGEDIT_EOF) - return serial; - - off_t stream_size = 0; - if (multiplex) { - /* Add up all the codec stream pages until EOF or a new link */ - while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) { - if (serial == codec_serial) - stream_size += og.header_len + og.body_len; - serial = get_page(in, oy, &og); - } - } - else { - /* Find the exact offsets of the start and end of the audio */ - stream_size -= sync_tell(in, oy, &og); - if (in->vfs->seek(in, end_offset, end_offset == 0 ? SEEK_END : SEEK_SET)) - return OGGEDIT_SEEK_FAILED; - stream_size += in->vfs->tell(in); - ogg_sync_reset(oy); - while ((serial = get_page(in, oy, &og)) > OGGEDIT_EOF && !ogg_page_bos(&og)) - stream_size += og.header_len + og.body_len; - } - if (serial < OGGEDIT_EOF) - return serial; - - return stream_size; -} - -static char *parse_vendor(const ogg_packet *op, const size_t magic_length) -{ - if (op->bytes < magic_length + 4) - return NULL; - - const uint8_t *p = op->packet + magic_length; - const uint32_t vendor_length = *p | *(p+1)<<8 | *(p+2)<<16 | *(p+3)<<24; - if (op->bytes < magic_length + 4 + vendor_length) - return NULL; - - char *vendor = calloc(vendor_length+1, 1); - if (vendor) - memcpy(vendor, p+4, vendor_length); - return vendor; -} - -static int init_read_stream(DB_FILE *in, ogg_sync_state *oy, ogg_stream_state *os, ogg_page *og, const off_t offset, const char *codec) -{ - int serial = skip_to_codec(in, oy, og, offset, codec); - serial = skip_to_header(in, oy, og, serial, serial); - if (serial <= OGGEDIT_EOF) - return serial; - - if (ogg_stream_init(os, serial)) - return OGGEDIT_FAILED_TO_INIT_STREAM; - - os->b_o_s = 1; - ogg_stream_pagein(os, og); - - return OGGEDIT_OK; -} - -static int read_packet(DB_FILE *in, ogg_sync_state *oy, ogg_stream_state *os, ogg_page *og, ogg_packet *header, int pages) -{ - ogg_packet op; - do { - while (ogg_stream_packetpeek(os, NULL) == 0) { - const int serial = get_page(in, oy, og); - if (serial <= OGGEDIT_EOF) { - return serial; - } - if (os->serialno == serial) { - pages++; - ogg_stream_pagein(os, og); - } - } - if (ogg_stream_check(os)) - return OGGEDIT_FAILED_TO_STREAM_PAGE_FOR_PACKET; - } while (ogg_stream_packetout(os, &op) != 1); - - memset(header, '\0', sizeof(*header)); - if (!header || !(header->packet = malloc(op.bytes))) { - free(header); - return OGGEDIT_ALLOCATION_FAILURE; - } - - header->bytes = op.bytes; - memcpy(header->packet, op.packet, op.bytes); - return pages; -} - -static ogg_packet *fill_vc_packet(const char *magic, const size_t magic_length, const char *vendor, const size_t num_tags, char **tags, const bool framing, const size_t padding, ogg_packet *op) -{ - oggpack_buffer opb; - oggpack_writeinit(&opb); - _oggpack_chars(&opb, magic, magic_length); - _oggpack_string(&opb, vendor); - oggpack_write(&opb, num_tags, 32); - for (size_t i = 0; i < num_tags; i++) - _oggpack_string(&opb, tags[i]); - if (framing) { - oggpack_write(&opb, 1, 1); - oggpack_writealign(&opb); - } - for (size_t i = 0; i < padding; i++) - oggpack_write(&opb, 0, 8); // use oggpack_writecopy when it is fixed - if (oggpack_writecheck(&opb)) - return NULL; - - if (op) { - memset(op, '\0', sizeof(*op)); - op->bytes = oggpack_bytes(&opb); - if (op->packet = malloc(op->bytes)) - memcpy(op->packet, oggpack_get_buffer(&opb), op->bytes); - } - oggpack_writeclear(&opb); - - if (!op->packet) - return NULL; - - return op; -} - -static size_t vc_size(const char *vendor, size_t num_tags, char **tags) -{ - size_t metadata_size = 4 + strlen(vendor) + 4; - for (size_t i = 0; i < num_tags; i++) - metadata_size += strlen(tags[i]) + 4; - return metadata_size; -} - -static int copy_remaining_pages(DB_FILE *in, FILE *out, ogg_sync_state *oy, const int codec_serial, uint32_t pageno) -{ - /* Skip past the codec header packets */ - ogg_page og; - int serial; - do - serial = get_page(in, oy, &og); - while (serial > OGGEDIT_EOF && serial == codec_serial && ogg_page_granulepos(&og) <= 0); - - /* Renumber the rest of this link */ - while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) { - if (serial == codec_serial && ogg_page_pageno(&og) != ++pageno) { - og.header[18] = pageno & 0xFF; - og.header[19] = pageno >> 8 & 0xFF; - og.header[20] = pageno >> 16 & 0xFF; - og.header[21] = pageno >> 24 & 0xFF; - ogg_page_checksum_set(&og); - } - serial = write_page_and_get_next(in, out, oy, &og); - } - - /* Blindly copy remaining links */ - while (serial > OGGEDIT_EOF) - serial = write_page_and_get_next(in, out, oy, &og); - - return serial; -} - -static int write_all_streams(DB_FILE *in, FILE *out, ogg_sync_state *oy, const off_t offset) -{ - /* Copy BOS page(s) */ - ogg_page og; - int serial = skip_to_bos(in, oy, &og, offset); - while (serial > OGGEDIT_EOF && ogg_page_bos(&og)) - serial = write_page_and_get_next(in, out, oy, &og); - if (serial <= OGGEDIT_EOF) - return serial; - - /* Copy all pages until EOF or next link */ - while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) - if ((serial = write_page_and_get_next(in, out, oy, &og)) < OGGEDIT_EOF) - return serial; - - return OGGEDIT_OK; -} - -static int write_one_stream(DB_FILE *in, FILE *out, ogg_sync_state *oy, const off_t offset, const char *codec) -{ - /* Find codec BOS page */ - ogg_page og; - const int codec_serial = skip_to_codec(in, oy, &og, offset, codec); - if (codec_serial <= OGGEDIT_EOF) - return codec_serial; - - /* Write it and skip the other BOS pages */ - int serial = write_page_and_get_next(in, out, oy, &og); - if ((serial = skip_to_header(in, oy, &og, serial, codec_serial)) <= OGGEDIT_EOF) - return serial; - - /* Copy all codec pages until EOF or next link */ - while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) { - if (serial == codec_serial && !write_page(out, &og)) - return OGGEDIT_WRITE_ERROR; - serial = get_page(in, oy, &og); - } - if (serial < OGGEDIT_EOF) - return serial; - - return OGGEDIT_OK; -} - -int oggedit_write_file(DB_FILE *in, const char *outname, const off_t offset, const char *codec) -{ - FILE *out = open_new_file(outname); - if (!out) - return OGGEDIT_CANNOT_OPEN_OUTPUT_FILE; - - ogg_sync_state oy; - ogg_sync_init(&oy); - - int res; - if (codec) - res = write_one_stream(in, out, &oy, offset, codec); - else - res = write_all_streams(in, out, &oy, offset); - - cleanup(in, out, &oy, NULL); - - if (res <= OGGEDIT_EOF) - unlink(outname); - - return res; -} - -off_t oggedit_stream_info(DB_FILE *in, const off_t start_offset, const off_t end_offset, const char *codec, char **codecs) -{ - int res; - ogg_sync_state oy; - ogg_sync_init(&oy); - *codecs = codec_names(in, &oy, start_offset, &res); - const off_t stream_size = codec_stream_size(in, &oy, start_offset, end_offset, codec); - cleanup(in, NULL, &oy, NULL); - return stream_size; -} - -/**************************** - * FLAC - ****************************/ -#define PADTYPE 0x01 -#define VCTYPE 0x04 -#define FLACLAST 0x80 - -static void clear_header_list(ogg_packet **headers) -{ - if (headers) { - for (ogg_packet **header = headers; *header; header++) { - ogg_packet_clear(*header); - free(*header); - } - free(headers); - } -} - -static ogg_packet **flac_headers_alloc(ogg_packet **headers, const size_t packets) -{ - ogg_packet **new_headers = realloc(headers, (packets+2) * sizeof(headers)); - if (!new_headers) { - clear_header_list(headers); - return NULL; - } - - if (!(new_headers[packets] = malloc(sizeof(ogg_packet)))) { - clear_header_list(new_headers); - return NULL; - } - - new_headers[packets + 1] = NULL; - return new_headers; -} - -static ogg_packet **metadata_block_packets(DB_FILE *in, ogg_sync_state *oy, const off_t offset, char **vendor, int *res) -{ - ogg_stream_state os; - ogg_page og; - if ((*res = init_read_stream(in, oy, &os, &og, offset, FLACNAME)) <= OGGEDIT_EOF) - return NULL; - - int pages = 1; - size_t packets = 0; - ogg_packet **headers = NULL; - while ((headers = flac_headers_alloc(headers, packets)) && - (pages = read_packet(in, oy, &os, &og, headers[packets], pages)) > OGGEDIT_EOF && - (headers[packets++]->packet[0] & FLACLAST) != FLACLAST); - - ogg_stream_clear(&os); - - if (!headers) - *res = OGGEDIT_ALLOCATION_FAILURE; - else if (!packets || headers[0]->packet[0] & 0x3F != VCTYPE) - *res = OGGEDIT_CANNOT_PARSE_HEADERS; - else - *res = pages; - if (*res <= OGGEDIT_EOF) { - clear_header_list(headers); - return NULL; - } - - *vendor = parse_vendor(headers[0], 4); - size_t bytes = 0; - for (size_t i = 0; i < packets; i++) - bytes += headers[i]->bytes; - if (bytes < MAXPAYLOAD * (pages-1)) - headers[0]->bytes = 4; - - *res = OGGEDIT_OK; - return headers; -} - -static long write_metadata_block_packets(FILE *out, const int serial, const char *vendor, const size_t num_tags, char **tags, size_t padding, ogg_packet **metadata) -{ - const size_t header_length = vc_size(vendor, num_tags, tags); - if (header_length > (2<<24)) - return OGGEDIT_ALLOCATION_FAILURE; - - char magic[4]; - magic[0] = VCTYPE; - magic[1] = header_length >> 16 & 0xFF; - magic[2] = header_length >> 8 & 0xFF; - magic[3] = header_length & 0xFF; - ogg_packet_clear(metadata[0]); - if (!fill_vc_packet(magic, 4, vendor, num_tags, tags, false, padding, metadata[0])) - return OGGEDIT_ALLOCATION_FAILURE; - - ogg_stream_state os; - if (ogg_stream_init(&os, serial)) - return OGGEDIT_FAILED_TO_INIT_STREAM; - os.b_o_s = 1; - os.pageno = 1; - - for (int i = 0; metadata[i]; i++) { - if (!metadata[i+1]) - metadata[i]->packet[0] |= FLACLAST; - ogg_stream_packetin(&os, metadata[i]); - } - - return flush_stream(out, &os); -} - -off_t oggedit_write_flac_metadata(DB_FILE *in, const char *fname, const off_t offset, const int num_tags, char **tags) -{ - off_t res; - char tempname[PATH_MAX] = ""; - ogg_packet **headers = NULL; - char *vendor = NULL; - ogg_sync_state oy; - ogg_sync_init(&oy); - - /* Original file must be writable whichever way we update it */ - FILE *out = fopen(fname, "r+b"); - if (!out) { - res = OGGEDIT_CANNOT_UPDATE_FILE; - goto cleanup; - } - - /* See if we can write the tags packet directly into the existing file ... */ - if (!(headers = metadata_block_packets(in, &oy, offset, &vendor, (int *)&res))) - goto cleanup; - const off_t stream_size_k = in->vfs->getlength(in) / 1000; // use file size for now - const size_t metadata_size = 4 + vc_size(vendor, num_tags, tags); - ptrdiff_t padding = headers[0]->bytes - metadata_size; - if (stream_size_k < 1000 || padding < 0 || headers[1] && padding > 0 || padding > stream_size_k+metadata_size) - if (res = open_temp_file(fname, tempname, &out)) - goto cleanup; - - /* Re-pad if writing the whole file */ - if (*tempname) { - size_t i = 0, j = 0; - while (headers[i]) { - headers[j++] = headers[i]; - while (headers[++i] && (headers[i]->packet[0] & 0x3F) == PADTYPE) { - ogg_packet_clear(headers[i]); - free(headers[i]); - } - } - headers[j] = NULL; - padding = headers[1] || stream_size_k < 900 ? 0 : stream_size_k < 10000 ? 1024 : stream_size_k < 100000 ? 8192 : 65536; - } - - /* Write pages until we reach the correct comment header */ - ogg_page og; - const int flac_serial = copy_up_to_codec(in, out, &oy, &og, *tempname ? 0 : offset, offset, FLACNAME); - if (flac_serial <= OGGEDIT_EOF) { - res = flac_serial; - goto cleanup; - } - if ((res = copy_up_to_header(in, out, &oy, &og, flac_serial)) <= OGGEDIT_EOF) - goto cleanup; - const long pageno = write_metadata_block_packets(out, flac_serial, vendor, num_tags, tags, padding, headers); - if (pageno < OGGEDIT_EOF) { - res = pageno; - goto cleanup; - } - - /* If we have tempfile, copy the remaining pages */ - if (*tempname) - if ((res = copy_remaining_pages(in, out, &oy, flac_serial, pageno)) < OGGEDIT_EOF || (res = finish_temp_file(tempname, fname))) - goto cleanup; - - res = file_size(fname); - -cleanup: - clear_header_list(headers); - cleanup(in, out, &oy, vendor); - if (res < OGGEDIT_OK) - unlink(tempname); - return res; -} - -/**************************** - * Vorbis - ****************************/ -#define VCMAGIC "\3vorbis" -#define CODEMAGIC "\5vorbis" - -static ptrdiff_t check_vorbis_headers(DB_FILE *in, ogg_sync_state *oy, const off_t offset, char **vendor, ogg_packet *codebooks) -{ - ogg_stream_state os; - ogg_page og; - const int serial = init_read_stream(in, oy, &os, &og, offset, VORBISNAME); - if (serial <= OGGEDIT_EOF) - return serial; - - ogg_packet vc; - int pages = read_packet(in, oy, &os, &og, &vc, 1); - if (pages > OGGEDIT_EOF) - pages = read_packet(in, oy, &os, &og, codebooks, pages); - ogg_stream_clear(&os); - if (pages <= OGGEDIT_EOF) - return pages; - - if (vc.bytes > strlen(VCMAGIC) && !memcmp(vc.packet, VCMAGIC, strlen(VCMAGIC)) && - codebooks->bytes > strlen(CODEMAGIC) && !memcmp(codebooks->packet, CODEMAGIC, strlen(CODEMAGIC))) - *vendor = parse_vendor(&vc, strlen(VCMAGIC)); - free(vc.packet); - if (!*vendor) - return OGGEDIT_CANNOT_PARSE_HEADERS; - - if ((vc.bytes + codebooks->bytes) < MAXPAYLOAD * (pages-1)) - return 4; // prevent in-place write if the packets are split over too many pages - - return vc.bytes; -} - -static long write_vorbis_tags(FILE *out, const int serial, const char *vendor, const size_t num_tags, char **tags, const size_t padding, ogg_packet *codebooks) -{ - ogg_packet op; - if (!fill_vc_packet(VCMAGIC, strlen(VCMAGIC), vendor, num_tags, tags, true, padding, &op)) - return OGGEDIT_ALLOCATION_FAILURE; - - ogg_stream_state os; - if (ogg_stream_init(&os, serial)) - return OGGEDIT_FAILED_TO_INIT_STREAM; - os.b_o_s = 1; - os.pageno = 1; - ogg_stream_packetin(&os, &op); - ogg_stream_packetin(&os, codebooks); - - ogg_packet_clear(&op); - return flush_stream(out, &os); -} - -off_t oggedit_write_vorbis_metadata(DB_FILE *in, const char *fname, const off_t offset, const size_t stream_size, const int num_tags, char **tags) -{ - off_t res; - char tempname[PATH_MAX] = ""; - char *vendor = NULL; - ogg_packet codebooks; - memset(&codebooks, '\0', sizeof(codebooks)); - ogg_sync_state oy; - ogg_sync_init(&oy); - - /* Original file must be writable whichever way we update it */ - FILE *out = fopen(fname, "r+b"); - if (!out) { - res = OGGEDIT_CANNOT_UPDATE_FILE; - goto cleanup; - } - - /* See if we can write the tags packet directly into the existing file ... */ - const ptrdiff_t tags_packet_size = check_vorbis_headers(in, &oy, offset, &vendor, &codebooks); - if (tags_packet_size <= OGGEDIT_EOF) { - res = tags_packet_size; - goto cleanup; - } - const size_t metadata_size = strlen(VCMAGIC) + vc_size(vendor, num_tags, tags) + 1; - ptrdiff_t padding = tags_packet_size - metadata_size; - const off_t file_size_k = in->vfs->getlength(in) / 1000; - const size_t stream_size_k = stream_size ? stream_size / 1000 : file_size_k; - if (file_size_k < 100 || padding < 0 || padding > file_size_k/10+stream_size_k+metadata_size) - if (res = open_temp_file(fname, tempname, &out)) - goto cleanup; - - /* Re-pad if writing the whole file */ - if (*tempname) - padding = stream_size_k < 90 ? 0 : stream_size_k < 1000 ? 128 : stream_size_k < 10000 ? 1024 : 8192; - - /* Write pages until the correct comment header */ - ogg_page og; - const int vorbis_serial = copy_up_to_codec(in, out, &oy, &og, *tempname ? 0 : offset, offset, VORBISNAME); - if (vorbis_serial <= OGGEDIT_EOF) { - res = vorbis_serial; - goto cleanup; - } - if ((res = copy_up_to_header(in, out, &oy, &og, vorbis_serial)) <= OGGEDIT_EOF) - goto cleanup; - const long pageno = write_vorbis_tags(out, vorbis_serial, vendor, num_tags, tags, padding, &codebooks); - if (pageno < OGGEDIT_EOF) { - res = pageno; - goto cleanup; - } - - /* If we have tempfile, copy the remaining pages */ - if (*tempname) - if ((res = copy_remaining_pages(in, out, &oy, vorbis_serial, pageno)) < OGGEDIT_EOF || (res = finish_temp_file(tempname, fname))) - goto cleanup; - - res = file_size(fname); - -cleanup: - ogg_packet_clear(&codebooks); - cleanup(in, out, &oy, vendor); - if (res < OGGEDIT_OK) - unlink(tempname); - return res; -} - -/**************************** - * Opus - ****************************/ -#define TAGMAGIC "OpusTags" - -static ptrdiff_t check_opus_header(DB_FILE *in, ogg_sync_state *oy, const off_t offset, char **vendor) -{ - ogg_stream_state os; - ogg_page og; - const int serial = init_read_stream(in, oy, &os, &og, offset, OPUSNAME); - if (serial <= OGGEDIT_EOF) - return serial; - - ogg_packet op; - const int pages = read_packet(in, oy, &os, &og, &op, 1); - ogg_stream_clear(&os); - if (pages <= OGGEDIT_EOF) - return pages; - - if (op.bytes > strlen(TAGMAGIC) && !memcmp(op.packet, TAGMAGIC, strlen(TAGMAGIC))) - *vendor = parse_vendor(&op, strlen(TAGMAGIC)); - free(op.packet); - if (!*vendor) - return OGGEDIT_CANNOT_PARSE_HEADERS; - - if (op.bytes < MAXPAYLOAD * (pages-1)) - return 4; // prevent in-place write if the packet is weirdly split into too many pages - - return op.bytes; -} - -static long write_opus_tags(FILE *out, const int serial, const char *vendor, const size_t num_tags, char **tags, const size_t padding) -{ - ogg_packet op; - if (!fill_vc_packet(TAGMAGIC, strlen(TAGMAGIC), vendor, num_tags, tags, false, padding, &op)) - return OGGEDIT_ALLOCATION_FAILURE; - - ogg_stream_state os; - if (ogg_stream_init(&os, serial)) - return OGGEDIT_FAILED_TO_INIT_STREAM; - os.b_o_s = 1; - os.pageno = 1; - ogg_stream_packetin(&os, &op); - - ogg_packet_clear(&op); - return flush_stream(out, &os); -} - -off_t oggedit_write_opus_metadata(DB_FILE *in, const char *fname, const off_t offset, const size_t stream_size, const int output_gain, const int num_tags, char **tags) -{ - off_t res; - char tempname[PATH_MAX] = ""; - char *vendor = NULL; - ogg_sync_state oy; - ogg_sync_init(&oy); - - /* Original file must be writable whichever way we update it */ - FILE *out = fopen(fname, "r+b"); - if (!out) { - res = OGGEDIT_CANNOT_UPDATE_FILE; - goto cleanup; - } - - /* Should we write the tags packet directly into the existing file ... */ - const ptrdiff_t tags_packet_size = check_opus_header(in, &oy, offset, &vendor); - if (tags_packet_size <= OGGEDIT_EOF) { - res = tags_packet_size; - goto cleanup; - } - const size_t metadata_size = strlen(TAGMAGIC) + vc_size(vendor, num_tags, tags); - ptrdiff_t padding = tags_packet_size - metadata_size; - const off_t file_size_k = in->vfs->getlength(in) / 1000; - const size_t stream_size_k = stream_size ? stream_size / 1000 : file_size_k; - if (file_size_k < 100 || padding < 0 || padding > file_size_k/10+stream_size_k+metadata_size) - if (res = open_temp_file(fname, tempname, &out)) - goto cleanup; - - /* Re-pad if writing the whole file */ - if (*tempname) - padding = stream_size_k < 90 ? 0 : stream_size_k < 1000 ? 128 : stream_size_k < 10000 ? 1024 : 8192; - - /* Write pages until we reach the correct OpusHead, then write OpusTags */ - ogg_page og; - const int opus_serial = copy_up_to_codec(in, out, &oy, &og, *tempname ? 0 : offset, offset, OPUSNAME); - if (opus_serial <= OGGEDIT_EOF) { - res = opus_serial; - goto cleanup; - } - if (output_gain > INT_MIN) { - og.body[16] = output_gain & 0xFF; - og.body[17] = output_gain >> 8 & 0xFF; - ogg_page_checksum_set(&og); - } - if ((res = copy_up_to_header(in, out, &oy, &og, opus_serial)) <= OGGEDIT_EOF) - goto cleanup; - const long pageno = write_opus_tags(out, opus_serial, vendor, num_tags, tags, (size_t)padding); - if (pageno < OGGEDIT_EOF) { - res = pageno; - goto cleanup; - } - - /* If we have tempfile, copy the remaining pages */ - if (*tempname) - if ((res = copy_remaining_pages(in, out, &oy, opus_serial, pageno)) < OGGEDIT_EOF || (res = finish_temp_file(tempname, fname))) - goto cleanup; - - res = file_size(fname); - -cleanup: - cleanup(in, out, &oy, vendor); - if (res < OGGEDIT_OK) - unlink(tempname); - return res; -} -/* -struct timeval timeval; -gettimeofday(&timeval, NULL); -int usecs = timeval.tv_sec* 1000000 + timeval.tv_usec; -gettimeofday(&timeval, NULL); -usecs = timeval.tv_sec* 1000000 + timeval.tv_usec - usecs; -fprintf(stderr, "%d micro-seconds\n", usecs); -*/ diff --git a/plugins/liboggedit/oggedit.h b/plugins/liboggedit/oggedit.h index 29ca7d13..b0a4d9b0 100644 --- a/plugins/liboggedit/oggedit.h +++ b/plugins/liboggedit/oggedit.h @@ -27,10 +27,6 @@ #ifndef __OGGEDIT_H #define __OGGEDIT_H -#define OPUSNAME "Opus" -#define VORBISNAME "Vorbis" -#define FLACNAME "Flac" - #define ALBUM_ART_KEY "METADATA_BLOCK_PICTURE" #define ALBUM_ART_META "metadata_block_picture" @@ -74,16 +70,24 @@ /* Cannot read data from image file */ #define OGGEDIT_CANT_READ_IMAGE_FILE -101 +/* oggedit_utils.c */ uint8_t *oggedit_vorbis_channel_map(const int channel_count); const char *oggedit_map_tag(char *key, const char *in_or_out); + +/* oggedit_art.c */ const char *oggedit_album_art_type(const int type); char *oggedit_album_art_tag(DB_FILE *fp, int *res); -int oggedit_write_file(DB_FILE *in, const char *outname, const off_t offset, const char *codec); -off_t oggedit_stream_info(DB_FILE *in, const off_t start_offset, const off_t end_offset, const char *codec, char **codecs); - +/* oggedit_flac.c */ off_t oggedit_write_flac_metadata(DB_FILE *in, const char *fname, const off_t offset, const int num_tags, char **tags); + +/* oggedit_vorbis.c */ +off_t oggedit_vorbis_stream_info(DB_FILE *in, const off_t start_offset, const off_t end_offset, char **codecs); off_t oggedit_write_vorbis_metadata(DB_FILE *in, const char *fname, const off_t offset, const size_t stream_size, const int num_tags, char **tags); + +/* oggedit_opus.c */ +int oggedit_write_opus_file(DB_FILE *in, const char *outname, const off_t offset, const bool all_streams); +off_t oggedit_opus_stream_info(DB_FILE *in, const off_t start_offset, const off_t end_offset, char **codecs); off_t oggedit_write_opus_metadata(DB_FILE *in, const char *fname, const off_t offset, const size_t stream_size, const int output_gain, const int num_tags, char **tags); #endif /* __OGGEDIT_H */ diff --git a/plugins/liboggedit/oggedit_art.c b/plugins/liboggedit/oggedit_art.c new file mode 100644 index 00000000..44e1c015 --- /dev/null +++ b/plugins/liboggedit/oggedit_art.c @@ -0,0 +1,172 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library album art functions + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifdef HAVE_CONFIG_H +#include +#endif +#include +#include +#include +#include +#include +#include "oggedit_internal.h" +#include "oggedit.h" + +const char *oggedit_album_art_type(const int type) +{ + switch (type) { + case 1: + return "32x32 pixels file icon"; + case 2: + return "other file icon"; + case 3: + return "front cover"; + case 4: + return "back cover"; + case 5: + return "leaflet page"; + case 6: + return "media"; + case 7: + return "lead artist/lead performer/soloist"; + case 8: + return "artist/performer"; + case 9: + return "conductor"; + case 10: + return "band/orchestra"; + case 11: + return "composer"; + case 12: + return "lyricist/text writer"; + case 13: + return "recording location"; + case 14: + return "during recording"; + case 15: + return "during performance"; + case 16: + return "movie/video screen capture"; + case 17: + return "bright coloured fish"; + case 18: + return "illustration"; + case 19: + return "band/artist logotype"; + case 20: + return "publisher/studio logotype"; + default: + return "other"; + } +} + +static char *btoa(const unsigned char *binary, const size_t binary_length) +{ + const char b64[64] = { + 'A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z', + 'a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z', + '0','1','2','3','4','5','6','7','8','9','+','/' + }; + + char *ascii_base64 = malloc((binary_length-1)/3 * 4 + 5); + if (!ascii_base64) + return NULL; + + char *p = ascii_base64; + const unsigned char *q = binary; + const unsigned char *binary_end = binary + binary_length; + while (q+2 < binary_end) { + uint32_t chunk = q[0]<<16 | q[1]<<8 | q[2]; + p[3] = b64[chunk & 0x3f]; + p[2] = b64[chunk>>6 & 0x3f]; + p[1] = b64[chunk>>12 & 0x3f]; + p[0] = b64[chunk>>18 & 0x3f]; + p+=4, q+=3; + } + + if (q < binary_end) { + uint16_t remainder = q[0]<<8 | (q+1 == binary_end ? '\0' : q[1]); + p[3] = '='; + p[2] = q+1 == binary_end ? '=' : b64[remainder<<2 & 0x3f]; + p[1] = b64[remainder>>4 & 0x3f]; + p[0] = b64[remainder>>10 & 0x3f]; + p+=4; + } + *p = '\0'; + + return ascii_base64; +} + +char *oggedit_album_art_tag(DB_FILE *fp, int *res) +{ + if (!fp) { + *res = OGGEDIT_FILE_NOT_OPEN; + return NULL; + } + + const int64_t data_length = fp->vfs->getlength(fp); + if (data_length < 50 || data_length > 10000000) { + fp->vfs->close(fp); + *res = OGGEDIT_BAD_FILE_LENGTH; + return NULL; + } + + char *data = malloc(data_length); + if (!data) { + fp->vfs->close(fp); + *res = OGGEDIT_ALLOCATION_FAILURE; + return NULL; + } + + const size_t data_read = fp->vfs->read(data, 1, data_length, fp); + fp->vfs->close(fp); + if (data_read != data_length) { + free(data); + *res = OGGEDIT_CANT_READ_IMAGE_FILE; + return NULL; + } + + oggpack_buffer opb; + oggpackB_writeinit(&opb); + oggpackB_write(&opb, 3, 32); + _oggpackB_string(&opb, memcmp(data, "\x89PNG\x0D\x0A\x1A\x0A", 8) ? "image/jpeg" : "image/png"); + _oggpackB_string(&opb, "Album art added from DeaDBeeF"); + oggpackB_write(&opb, 0, 32); + oggpackB_write(&opb, 0, 32); + oggpackB_write(&opb, 0, 32); + oggpackB_write(&opb, 0, 32); + oggpackB_write(&opb, data_length, 32); + _oggpack_chars(&opb, data, data_length); + free(data); + if (oggpack_writecheck(&opb)) { + *res = OGGEDIT_ALLOCATION_FAILURE; + return NULL; + } + + char *tag = btoa(oggpackB_get_buffer(&opb), oggpackB_bytes(&opb)); + oggpackB_writeclear(&opb); + return tag; +} diff --git a/plugins/liboggedit/oggedit_flac.c b/plugins/liboggedit/oggedit_flac.c new file mode 100644 index 00000000..6eec6ac9 --- /dev/null +++ b/plugins/liboggedit/oggedit_flac.c @@ -0,0 +1,213 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library Flac functions + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifdef HAVE_CONFIG_H +#include +#endif +#include +#include +#include +#include +#include +#include +#include "oggedit_internal.h" +#include "oggedit.h" + +#define PADTYPE 0x01 +#define VCTYPE 0x04 +#define LASTBLOCK 0x80 + +static void clear_header_list(ogg_packet **headers) +{ + if (headers) { + for (ogg_packet **header = headers; *header; header++) { + ogg_packet_clear(*header); + free(*header); + } + free(headers); + } +} + +static ogg_packet **headers_alloc(ogg_packet **headers, const size_t packets) +{ + ogg_packet **new_headers = realloc(headers, (packets+2) * sizeof(headers)); + if (!new_headers) { + clear_header_list(headers); + return NULL; + } + + if (!(new_headers[packets] = malloc(sizeof(ogg_packet)))) { + clear_header_list(new_headers); + return NULL; + } + + new_headers[packets + 1] = NULL; + return new_headers; +} + +static ogg_packet **metadata_block_packets(DB_FILE *in, ogg_sync_state *oy, const off_t offset, char **vendor, int *res) +{ + ogg_stream_state os; + ogg_page og; + if ((*res = init_read_stream(in, oy, &os, &og, offset, FLACNAME)) <= OGGEDIT_EOF) + return NULL; + + int pages = 1; + size_t packets = 0; + ogg_packet **headers = NULL; + while ((headers = headers_alloc(headers, packets)) && + (pages = read_packet(in, oy, &os, &og, headers[packets], pages)) > OGGEDIT_EOF && + (headers[packets++]->packet[0] & LASTBLOCK) != LASTBLOCK); + + ogg_stream_clear(&os); + + if (!headers) + *res = OGGEDIT_ALLOCATION_FAILURE; + else if (!packets || headers[0]->packet[0] & 0x3F != VCTYPE) + *res = OGGEDIT_CANNOT_PARSE_HEADERS; + else + *res = pages; + if (*res <= OGGEDIT_EOF) { + clear_header_list(headers); + return NULL; + } + + *vendor = parse_vendor(headers[0], 4); + size_t bytes = 0; + for (size_t i = 0; i < packets; i++) + bytes += headers[i]->bytes; + if (bytes < MAXPAYLOAD * (pages-1)) + headers[0]->bytes = 4; + + *res = OGGEDIT_OK; + return headers; +} + +static long write_metadata_block_packets(FILE *out, const int serial, const char *vendor, const size_t num_tags, char **tags, size_t padding, ogg_packet **metadata) +{ + const size_t header_length = vc_size(vendor, num_tags, tags); + if (header_length > (2<<24)) + return OGGEDIT_ALLOCATION_FAILURE; + + char magic[4]; + magic[0] = VCTYPE; + magic[1] = header_length >> 16 & 0xFF; + magic[2] = header_length >> 8 & 0xFF; + magic[3] = header_length & 0xFF; + ogg_packet_clear(metadata[0]); + if (!fill_vc_packet(magic, 4, vendor, num_tags, tags, false, padding, metadata[0])) + return OGGEDIT_ALLOCATION_FAILURE; + + ogg_stream_state os; + if (ogg_stream_init(&os, serial)) + return OGGEDIT_FAILED_TO_INIT_STREAM; + os.b_o_s = 1; + os.pageno = 1; + + for (int i = 0; metadata[i]; i++) { + if (!metadata[i+1]) + metadata[i]->packet[0] |= LASTBLOCK; + ogg_stream_packetin(&os, metadata[i]); + } + + return flush_stream(out, &os); +} + +off_t oggedit_write_flac_metadata(DB_FILE *in, const char *fname, const off_t offset, const int num_tags, char **tags) +{ + off_t res; + char tempname[PATH_MAX] = ""; + ogg_packet **headers = NULL; + char *vendor = NULL; + ogg_sync_state oy; + ogg_sync_init(&oy); + + /* Original file must be writable whichever way we update it */ + FILE *out = fopen(fname, "r+b"); + if (!out) { + res = OGGEDIT_CANNOT_UPDATE_FILE; + goto cleanup; + } + + /* See if we can write the tags packet directly into the existing file ... */ + if (!(headers = metadata_block_packets(in, &oy, offset, &vendor, (int *)&res))) + goto cleanup; + const off_t stream_size_k = in->vfs->getlength(in) / 1000; // use file size for now + const size_t metadata_size = 4 + vc_size(vendor, num_tags, tags); + ptrdiff_t padding = headers[0]->bytes - metadata_size; + if (stream_size_k < 1000 || padding < 0 || headers[1] && padding > 0 || padding > stream_size_k+metadata_size) + if (res = open_temp_file(fname, tempname, &out)) + goto cleanup; + + /* Re-pad if writing the whole file */ + if (*tempname) { + size_t i = 0, j = 0; + while (headers[i]) { + headers[j++] = headers[i]; + while (headers[++i] && (headers[i]->packet[0] & 0x3F) == PADTYPE) { + ogg_packet_clear(headers[i]); + free(headers[i]); + } + } + headers[j] = NULL; + padding = headers[1] || stream_size_k < 900 ? 0 : stream_size_k < 10000 ? 1024 : stream_size_k < 100000 ? 8192 : 65536; + } + + /* Write pages until we reach the correct comment header */ + ogg_page og; + const int flac_serial = copy_up_to_codec(in, out, &oy, &og, *tempname ? 0 : offset, offset, FLACNAME); + if (flac_serial <= OGGEDIT_EOF) { + res = flac_serial; + goto cleanup; + } + if ((res = copy_up_to_header(in, out, &oy, &og, flac_serial)) <= OGGEDIT_EOF) + goto cleanup; + const long pageno = write_metadata_block_packets(out, flac_serial, vendor, num_tags, tags, padding, headers); + if (pageno < OGGEDIT_EOF) { + res = pageno; + goto cleanup; + } + + /* If we have tempfile, copy the remaining pages */ + if (*tempname) { + if ((res = copy_remaining_pages(in, out, &oy, flac_serial, pageno)) < OGGEDIT_EOF) + goto cleanup; + if (rename(tempname, fname)) { + res = OGGEDIT_RENAME_FAILED; + goto cleanup; + } + } + + res = file_size(fname); + +cleanup: + clear_header_list(headers); + cleanup(in, out, &oy, vendor); + if (res < OGGEDIT_OK) + unlink(tempname); + return res; +} + diff --git a/plugins/liboggedit/oggedit_internal.c b/plugins/liboggedit/oggedit_internal.c new file mode 100644 index 00000000..a95fa1f8 --- /dev/null +++ b/plugins/liboggedit/oggedit_internal.c @@ -0,0 +1,524 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library internal functions + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifdef HAVE_CONFIG_H +#include +#endif +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "oggedit.h" +#include "oggedit_internal.h" + +static char *cat_string(char *dest, const char *src, const char *sep) +{ + char *more = realloc(dest, strlen(dest) + strlen(src) + strlen(sep) + 1); + if (!more) { + free(dest); + return NULL; + } + return strcat(strcat(more, sep), src); +} + +static const char *codec_name(ogg_page *og) +{ + typedef struct { + const size_t length; + const char *codec; + const char *magic; + } codec_t; + const codec_t codecs[] = { + {.codec = OPUSNAME, .length = 19, .magic = "OpusHead"}, + {.codec = VORBISNAME, .length = 30, .magic = "\1vorbis"}, + {.codec = FLACNAME, .length = 47, .magic = "\177FLAC"}, + {.codec = "Speex", .length = 80, .magic = "Speex "}, + {.codec = "Celt", .length = 80, .magic = "CELT"}, // obsolete + {.codec = "MIDI", .length = 13, .magic = "OggMIDI\0"}, + {.codec = "PCM", .length = 28, .magic = "PCM "}, + {.codec = "Theora", .length = 42, .magic = "\200theora"}, + {.codec = "Daala", .length = 38, .magic = "\200daala"}, + {.codec = "Dirac", .length = 5, .magic = "BBCD\0"}, + {.codec = "Skeleton", .length = 80, .magic = "fishead\0"}, + {.codec = "Kate", .length = 64, .magic = "\200kate\0\0\0"}, + {.codec = "CMML", .length = 29, .magic = "CMML\0\0\0\0"}, + {.codec = "YUV4MPEG", .length = 8, .magic = "YUV4Mpeg"}, + {.codec = "UVS", .length = 48, .magic = "UVS "}, + {.codec = "YUV", .length = 32, .magic = "\0YUV"}, + {.codec = "RGB", .length = 24, .magic = "\0RGB"}, + {.codec = "JNG", .length = 48, .magic = "\213JNG\r\n\032\n"}, + {.codec = "MNG", .length = 48, .magic = "\212MNG\r\n\032\n"}, + {.codec = "PNG", .length = 48, .magic = "\211PNG\r\n\032\n"}, + {.codec = "Spots", .length = 52, .magic = "SPOTS\0\0\0"}, + {.codec = NULL} + }; + + for (const codec_t *match = codecs; match->codec; match++) + if ((size_t)og->body_len >= match->length && !memcmp(og->body, match->magic, strlen(match->codec))) + return match->codec; + + return "unknown"; +} + +void _oggpack_chars(oggpack_buffer *opb, const char *s, size_t length) +{ +// oggpack_writecopy(opb, s, length * 8); currently broken + while (length--) + oggpack_write(opb, *s++, 8); +} + +void _oggpack_string(oggpack_buffer *opb, const char *s) +{ + oggpack_write(opb, strlen(s), 32); + _oggpack_chars(opb, s, strlen(s)); +} + +void _oggpackB_string(oggpack_buffer *opb, const char *s) +{ + oggpackB_write(opb, strlen(s), 32); + _oggpack_chars(opb, s, strlen(s)); +} + +static bool ensure_directory(const char *path) +{ + struct stat stat_struct; + if (!stat(path, &stat_struct)) + return !S_ISDIR(stat_struct.st_mode); + + if (errno != ENOENT) + return false; + + char* dir = strdup(path); + if (!dir) + return false; + + const int bad_dir = ensure_directory(dirname(dir)); + free(dir); + + return !bad_dir && !mkdir(path, 0777); +} + +FILE *open_new_file(const char *outname) +{ + char outpath[PATH_MAX]; + strcpy(outpath, outname); + if (!ensure_directory(dirname(outpath))) + return NULL; + + unlink(outname); + return fopen(outname, "wbx"); +} + +int open_temp_file(const char *fname, char *tempname, FILE **out) +{ + snprintf(tempname, PATH_MAX, "%s.temp", fname); + unlink(tempname); + if (!(*out = freopen(tempname, "abx", *out))) + return OGGEDIT_CANNOT_OPEN_TEMPORARY_FILE; + + struct stat stat_struct; + if (!stat(fname, &stat_struct)) + chmod(tempname, stat_struct.st_mode); + + return 0; +} + +off_t file_size(const char *fname) +{ + struct stat sb; + if (stat(fname, &sb)) + return OGGEDIT_STAT_FAILED; + + return sb.st_size; +} + +void cleanup(DB_FILE *in, FILE *out, ogg_sync_state *oy, void *buffer) +{ + if (in) + in->vfs->close(in); + + if (out) + fclose(out); + + ogg_sync_clear(oy); + + if (buffer) + free(buffer); +} + +static int get_page(DB_FILE *in, ogg_sync_state *oy, ogg_page *og) +{ + uint16_t chunks_left = MAXPAGE / CHUNKSIZE; + while (ogg_sync_pageout(oy, og) != 1) { + char *buffer = ogg_sync_buffer(oy, CHUNKSIZE); + if (!in || !buffer || !chunks_left--) + return OGGEDIT_CANT_FIND_STREAM; + + const size_t bytes = in->vfs->read(buffer, 1, CHUNKSIZE, in); + if (!bytes) + return OGGEDIT_EOF; + + ogg_sync_wrote(oy, bytes); + } + + return ogg_page_serialno(og); +} + +static int skip_to_bos(DB_FILE *in, ogg_sync_state *oy, ogg_page *og, const off_t offset) +{ + if (!in) + return OGGEDIT_FILE_NOT_OPEN; + + if (in->vfs->seek(in, offset, SEEK_SET)) + return OGGEDIT_SEEK_FAILED; + + ogg_sync_reset(oy); + int serial; + do + serial = get_page(in, oy, og); + while (serial > OGGEDIT_EOF && !ogg_page_bos(og)); + + return serial; +} + +static int skip_to_codec(DB_FILE *in, ogg_sync_state *oy, ogg_page *og, const off_t offset, const char *codec) +{ + int serial = skip_to_bos(in, oy, og, offset); + while (serial > OGGEDIT_EOF && strcmp(codec_name(og), codec)) + serial = get_page(in, oy, og); + + return serial; +} + +static int skip_to_header(DB_FILE *in, ogg_sync_state *oy, ogg_page *og, int serial, const int codec_serial) +{ + while (serial > OGGEDIT_EOF && (ogg_page_bos(og) || serial != codec_serial)) + serial = get_page(in, oy, og); + + return serial; +} + +static off_t sync_tell(DB_FILE *in, ogg_sync_state *oy, ogg_page *og) +{ + return in->vfs->tell(in) - oy->fill + oy->returned - og->header_len - og->body_len; +} + +static bool write_page(FILE *out, ogg_page *og) +{ + if (fwrite(og->header, 1, og->header_len, out) != (size_t)og->header_len) + return false; + + if (fwrite(og->body, 1, og->body_len, out) != (size_t)og->body_len) + return false; + + return true; +} + +static int write_page_and_get_next(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og) +{ + if (!write_page(out, og)) + return OGGEDIT_WRITE_ERROR; + + return get_page(in, oy, og); +} + +int copy_up_to_codec(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og, const off_t start_offset, const off_t link_offset, const char *codec) +{ + int serial = skip_to_bos(in, oy, og, start_offset); + + if (fseek(out, sync_tell(in, oy, og), SEEK_SET)) + return OGGEDIT_SEEK_FAILED; + + while (serial > OGGEDIT_EOF && (sync_tell(in, oy, og) < link_offset || !ogg_page_bos(og) || strcmp(codec_name(og), codec))) + serial = write_page_and_get_next(in, out, oy, og); + + return serial; +} + +int copy_up_to_header(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og, const int codec_serial) +{ + int serial; + do + serial = write_page_and_get_next(in, out, oy, og); + while (serial > OGGEDIT_EOF && serial != codec_serial); + return serial; +} + +long flush_stream(FILE *out, ogg_stream_state *os) +{ + ogg_page og; + while (ogg_stream_flush_fill(os, &og, MAXPAYLOAD)) + if (!write_page(out, &og)) + return OGGEDIT_WRITE_ERROR; + const long pageno = ogg_stream_check(os) ? OGGEDIT_FLUSH_FAILED : ogg_page_pageno(&og); + ogg_stream_clear(os); + return pageno; +} + +char *codec_names(DB_FILE *in, ogg_sync_state *oy, const off_t link_offset, int *res) +{ + ogg_page og; + *res = skip_to_bos(in, oy, &og, link_offset); + char *codecs = strdup("Ogg"); + while (codecs && *res > OGGEDIT_EOF && ogg_page_bos(&og)) { + codecs = cat_string(codecs, codec_name(&og), strcmp(codecs, "Ogg") ? "/" : " "); + *res = get_page(in, oy, &og); + } + if (!*codecs) { + *res = OGGEDIT_ALLOCATION_FAILURE; + return NULL; + } + if (*res <= OGGEDIT_EOF) { + free(codecs); + return NULL; + } + + *res = OGGEDIT_OK; + return codecs; +} + +off_t codec_stream_size(DB_FILE *in, ogg_sync_state *oy, const off_t start_offset, const off_t end_offset, const char *codec) +{ + /* Find codec serial and any other codecs */ + bool multiplex = false; + ogg_page og; + int codec_serial = -1; + int serial = skip_to_bos(in, oy, &og, start_offset); + while (serial > OGGEDIT_EOF && ogg_page_bos(&og)) { + if (strcmp(codec_name(&og), codec)) + multiplex = true; + else + codec_serial = serial; + serial = get_page(in, oy, &og); + } + + /* Skip to the first codec data page */ + while (serial > OGGEDIT_EOF && !(ogg_page_granulepos(&og) > 0 && serial == codec_serial)) + if ((serial = get_page(in, oy, &og)) <= OGGEDIT_EOF) + return serial; + + off_t stream_size = 0; + if (multiplex) { + /* Add up all the codec stream pages until EOF or a new link */ + while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) { + if (serial == codec_serial) + stream_size += og.header_len + og.body_len; + serial = get_page(in, oy, &og); + } + } + else { + /* Find the exact offsets of the start and end of the audio */ + stream_size -= sync_tell(in, oy, &og); + if (in->vfs->seek(in, end_offset, end_offset == 0 ? SEEK_END : SEEK_SET)) + return OGGEDIT_SEEK_FAILED; + stream_size += in->vfs->tell(in); + ogg_sync_reset(oy); + while ((serial = get_page(in, oy, &og)) > OGGEDIT_EOF && !ogg_page_bos(&og)) + stream_size += og.header_len + og.body_len; + } + if (serial < OGGEDIT_EOF) + return serial; + + return stream_size; +} + +char *parse_vendor(const ogg_packet *op, const size_t magic_length) +{ + if (op->bytes < magic_length + 4) + return NULL; + + const uint8_t *p = op->packet + magic_length; + const uint32_t vendor_length = *p | *(p+1)<<8 | *(p+2)<<16 | *(p+3)<<24; + if (op->bytes < magic_length + 4 + vendor_length) + return NULL; + + char *vendor = calloc(vendor_length+1, 1); + if (vendor) + memcpy(vendor, p+4, vendor_length); + return vendor; +} + +int init_read_stream(DB_FILE *in, ogg_sync_state *oy, ogg_stream_state *os, ogg_page *og, const off_t offset, const char *codec) +{ + int serial = skip_to_codec(in, oy, og, offset, codec); + serial = skip_to_header(in, oy, og, serial, serial); + if (serial <= OGGEDIT_EOF) + return serial; + + if (ogg_stream_init(os, serial)) + return OGGEDIT_FAILED_TO_INIT_STREAM; + + os->b_o_s = 1; + ogg_stream_pagein(os, og); + + return OGGEDIT_OK; +} + +int read_packet(DB_FILE *in, ogg_sync_state *oy, ogg_stream_state *os, ogg_page *og, ogg_packet *header, int pages) +{ + ogg_packet op; + do { + while (ogg_stream_packetpeek(os, NULL) == 0) { + const int serial = get_page(in, oy, og); + if (serial <= OGGEDIT_EOF) { + return serial; + } + if (os->serialno == serial) { + pages++; + ogg_stream_pagein(os, og); + } + } + if (ogg_stream_check(os)) + return OGGEDIT_FAILED_TO_STREAM_PAGE_FOR_PACKET; + } while (ogg_stream_packetout(os, &op) != 1); + + memset(header, '\0', sizeof(*header)); + if (!header || !(header->packet = malloc(op.bytes))) { + free(header); + return OGGEDIT_ALLOCATION_FAILURE; + } + + header->bytes = op.bytes; + memcpy(header->packet, op.packet, op.bytes); + return pages; +} + +ogg_packet *fill_vc_packet(const char *magic, const size_t magic_length, const char *vendor, const size_t num_tags, char **tags, const bool framing, const size_t padding, ogg_packet *op) +{ + oggpack_buffer opb; + oggpack_writeinit(&opb); + _oggpack_chars(&opb, magic, magic_length); + _oggpack_string(&opb, vendor); + oggpack_write(&opb, num_tags, 32); + for (size_t i = 0; i < num_tags; i++) + _oggpack_string(&opb, tags[i]); + if (framing) { + oggpack_write(&opb, 1, 1); + oggpack_writealign(&opb); + } + for (size_t i = 0; i < padding; i++) + oggpack_write(&opb, 0, 8); // use oggpack_writecopy when it is fixed + if (oggpack_writecheck(&opb)) + return NULL; + + if (op) { + memset(op, '\0', sizeof(*op)); + op->bytes = oggpack_bytes(&opb); + if (op->packet = malloc(op->bytes)) + memcpy(op->packet, oggpack_get_buffer(&opb), op->bytes); + } + oggpack_writeclear(&opb); + + if (!op->packet) + return NULL; + + return op; +} + +size_t vc_size(const char *vendor, size_t num_tags, char **tags) +{ + size_t metadata_size = 4 + strlen(vendor) + 4; + for (size_t i = 0; i < num_tags; i++) + metadata_size += strlen(tags[i]) + 4; + return metadata_size; +} + +int copy_remaining_pages(DB_FILE *in, FILE *out, ogg_sync_state *oy, const int codec_serial, uint32_t pageno) +{ + /* Skip past the codec header packets */ + ogg_page og; + int serial; + do + serial = get_page(in, oy, &og); + while (serial > OGGEDIT_EOF && serial == codec_serial && ogg_page_granulepos(&og) <= 0); + + /* Renumber the rest of this link */ + while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) { + if (serial == codec_serial && ogg_page_pageno(&og) != ++pageno) { + og.header[18] = pageno & 0xFF; + og.header[19] = pageno >> 8 & 0xFF; + og.header[20] = pageno >> 16 & 0xFF; + og.header[21] = pageno >> 24 & 0xFF; + ogg_page_checksum_set(&og); + } + serial = write_page_and_get_next(in, out, oy, &og); + } + + /* Blindly copy remaining links */ + while (serial > OGGEDIT_EOF) + serial = write_page_and_get_next(in, out, oy, &og); + + return serial; +} + +int write_all_streams(DB_FILE *in, FILE *out, ogg_sync_state *oy, const off_t offset) +{ + /* Copy BOS page(s) */ + ogg_page og; + int serial = skip_to_bos(in, oy, &og, offset); + while (serial > OGGEDIT_EOF && ogg_page_bos(&og)) + serial = write_page_and_get_next(in, out, oy, &og); + if (serial <= OGGEDIT_EOF) + return serial; + + /* Copy all pages until EOF or next link */ + while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) + if ((serial = write_page_and_get_next(in, out, oy, &og)) < OGGEDIT_EOF) + return serial; + + return OGGEDIT_OK; +} + +int write_one_stream(DB_FILE *in, FILE *out, ogg_sync_state *oy, const off_t offset, const char *codec) +{ + /* Find codec BOS page */ + ogg_page og; + const int codec_serial = skip_to_codec(in, oy, &og, offset, codec); + if (codec_serial <= OGGEDIT_EOF) + return codec_serial; + + /* Write it and skip the other BOS pages */ + int serial = write_page_and_get_next(in, out, oy, &og); + if ((serial = skip_to_header(in, oy, &og, serial, codec_serial)) <= OGGEDIT_EOF) + return serial; + + /* Copy all codec pages until EOF or next link */ + while (serial > OGGEDIT_EOF && !ogg_page_bos(&og)) { + if (serial == codec_serial && !write_page(out, &og)) + return OGGEDIT_WRITE_ERROR; + serial = get_page(in, oy, &og); + } + if (serial < OGGEDIT_EOF) + return serial; + + return OGGEDIT_OK; +} diff --git a/plugins/liboggedit/oggedit_internal.h b/plugins/liboggedit/oggedit_internal.h new file mode 100644 index 00000000..46a84345 --- /dev/null +++ b/plugins/liboggedit/oggedit_internal.h @@ -0,0 +1,59 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library internal headers + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifndef __OGGEDIT_INT_H +#define __OGGEDIT_INT_H + +#define OPUSNAME "Opus" +#define VORBISNAME "Vorbis" +#define FLACNAME "Flac" + +#define CHUNKSIZE 4096 +#define MAXPAGE 65536 +#define MAXPAYLOAD 65025 + +void _oggpack_chars(oggpack_buffer *opb, const char *s, size_t length); +void _oggpack_string(oggpack_buffer *opb, const char *s); +void _oggpackB_string(oggpack_buffer *opb, const char *s); +int open_temp_file(const char *fname, char *tempname, FILE **out); +FILE *open_new_file(const char *outname); +off_t file_size(const char *fname); +void cleanup(DB_FILE *in, FILE *out, ogg_sync_state *oy, void *buffer); +int copy_up_to_codec(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og, const off_t start_offset, const off_t link_offset, const char *codec); +int copy_up_to_header(DB_FILE *in, FILE *out, ogg_sync_state *oy, ogg_page *og, const int codec_serial); +long flush_stream(FILE *out, ogg_stream_state *os); +char *codec_names(DB_FILE *in, ogg_sync_state *oy, const off_t link_offset, int *res); +off_t codec_stream_size(DB_FILE *in, ogg_sync_state *oy, const off_t start_offset, const off_t end_offset, const char *codec); +char *parse_vendor(const ogg_packet *op, const size_t magic_length); +int init_read_stream(DB_FILE *in, ogg_sync_state *oy, ogg_stream_state *os, ogg_page *og, const off_t offset, const char *codec); +int read_packet(DB_FILE *in, ogg_sync_state *oy, ogg_stream_state *os, ogg_page *og, ogg_packet *header, int pages); +ogg_packet *fill_vc_packet(const char *magic, const size_t magic_length, const char *vendor, const size_t num_tags, char **tags, const bool framing, const size_t padding, ogg_packet *op); +size_t vc_size(const char *vendor, size_t num_tags, char **tags); +int copy_remaining_pages(DB_FILE *in, FILE *out, ogg_sync_state *oy, const int codec_serial, uint32_t pageno); +int write_all_streams(DB_FILE *in, FILE *out, ogg_sync_state *oy, const off_t offset); +int write_one_stream(DB_FILE *in, FILE *out, ogg_sync_state *oy, const off_t offset, const char *codec); + +#endif /* __OGGEDIT_INT_H */ diff --git a/plugins/liboggedit/oggedit_opus.c b/plugins/liboggedit/oggedit_opus.c new file mode 100644 index 00000000..cb52480d --- /dev/null +++ b/plugins/liboggedit/oggedit_opus.c @@ -0,0 +1,197 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library Opus functions + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifdef HAVE_CONFIG_H +#include +#endif +#include +#include +#include +#include +#include +#include +#include +#include "oggedit_internal.h" +#include "oggedit.h" + +#define TAGMAGIC "OpusTags" + +int oggedit_write_opus_file(DB_FILE *in, const char *outname, const off_t offset, const bool all_streams) +{ + FILE *out = open_new_file(outname); + if (!out) + return OGGEDIT_CANNOT_OPEN_OUTPUT_FILE; + + ogg_sync_state oy; + ogg_sync_init(&oy); + + int res; + if (all_streams) + res = write_one_stream(in, out, &oy, offset, OPUSNAME); + else + res = write_all_streams(in, out, &oy, offset); + + cleanup(in, out, &oy, NULL); + + if (res <= OGGEDIT_EOF) + unlink(outname); + + return res; +} + +off_t oggedit_opus_stream_info(DB_FILE *in, const off_t start_offset, const off_t end_offset, char **codecs) +{ + int res; + ogg_sync_state oy; + ogg_sync_init(&oy); + *codecs = codec_names(in, &oy, start_offset, &res); + const off_t stream_size = codec_stream_size(in, &oy, start_offset, end_offset, OPUSNAME); + cleanup(in, NULL, &oy, NULL); + return stream_size; +} + +static ptrdiff_t check_opus_header(DB_FILE *in, ogg_sync_state *oy, const off_t offset, char **vendor) +{ + ogg_stream_state os; + ogg_page og; + const int serial = init_read_stream(in, oy, &os, &og, offset, OPUSNAME); + if (serial <= OGGEDIT_EOF) + return serial; + + ogg_packet op; + const int pages = read_packet(in, oy, &os, &og, &op, 1); + ogg_stream_clear(&os); + if (pages <= OGGEDIT_EOF) + return pages; + + if (op.bytes > strlen(TAGMAGIC) && !memcmp(op.packet, TAGMAGIC, strlen(TAGMAGIC))) + *vendor = parse_vendor(&op, strlen(TAGMAGIC)); + free(op.packet); + if (!*vendor) + return OGGEDIT_CANNOT_PARSE_HEADERS; + + if (op.bytes < MAXPAYLOAD * (pages-1)) + return 4; // prevent in-place write if the packet is weirdly split into too many pages + + return op.bytes; +} + +static long write_opus_tags(FILE *out, const int serial, const char *vendor, const size_t num_tags, char **tags, const size_t padding) +{ + ogg_packet op; + if (!fill_vc_packet(TAGMAGIC, strlen(TAGMAGIC), vendor, num_tags, tags, false, padding, &op)) + return OGGEDIT_ALLOCATION_FAILURE; + + ogg_stream_state os; + if (ogg_stream_init(&os, serial)) + return OGGEDIT_FAILED_TO_INIT_STREAM; + os.b_o_s = 1; + os.pageno = 1; + ogg_stream_packetin(&os, &op); + + ogg_packet_clear(&op); + return flush_stream(out, &os); +} + +off_t oggedit_write_opus_metadata(DB_FILE *in, const char *fname, const off_t offset, const size_t stream_size, const int output_gain, const int num_tags, char **tags) +{ + off_t res; + char tempname[PATH_MAX] = ""; + char *vendor = NULL; + ogg_sync_state oy; + ogg_sync_init(&oy); + + /* Original file must be writable whichever way we update it */ + FILE *out = fopen(fname, "r+b"); + if (!out) { + res = OGGEDIT_CANNOT_UPDATE_FILE; + goto cleanup; + } + + /* Should we write the tags packet directly into the existing file ... */ + const ptrdiff_t tags_packet_size = check_opus_header(in, &oy, offset, &vendor); + if (tags_packet_size <= OGGEDIT_EOF) { + res = tags_packet_size; + goto cleanup; + } + const size_t metadata_size = strlen(TAGMAGIC) + vc_size(vendor, num_tags, tags); + ptrdiff_t padding = tags_packet_size - metadata_size; + const off_t file_size_k = in->vfs->getlength(in) / 1000; + const size_t stream_size_k = stream_size ? stream_size / 1000 : file_size_k; + if (file_size_k < 100 || padding < 0 || padding > file_size_k/10+stream_size_k+metadata_size) + if (res = open_temp_file(fname, tempname, &out)) + goto cleanup; + + /* Re-pad if writing the whole file */ + if (*tempname) + padding = stream_size_k < 90 ? 0 : stream_size_k < 1000 ? 128 : stream_size_k < 10000 ? 1024 : 8192; + + /* Write pages until we reach the correct OpusHead, then write OpusTags */ + ogg_page og; + const int opus_serial = copy_up_to_codec(in, out, &oy, &og, *tempname ? 0 : offset, offset, OPUSNAME); + if (opus_serial <= OGGEDIT_EOF) { + res = opus_serial; + goto cleanup; + } + if (output_gain > INT_MIN) { + og.body[16] = output_gain & 0xFF; + og.body[17] = output_gain >> 8 & 0xFF; + ogg_page_checksum_set(&og); + } + if ((res = copy_up_to_header(in, out, &oy, &og, opus_serial)) <= OGGEDIT_EOF) + goto cleanup; + const long pageno = write_opus_tags(out, opus_serial, vendor, num_tags, tags, (size_t)padding); + if (pageno < OGGEDIT_EOF) { + res = pageno; + goto cleanup; + } + + /* If we have tempfile, copy the remaining pages */ + if (*tempname) { + if ((res = copy_remaining_pages(in, out, &oy, opus_serial, pageno)) < OGGEDIT_EOF) + goto cleanup; + if (rename(tempname, fname)) { + res = OGGEDIT_RENAME_FAILED; + goto cleanup; + } + } + + res = file_size(fname); + +cleanup: + cleanup(in, out, &oy, vendor); + if (res < OGGEDIT_OK) + unlink(tempname); + return res; +} +/* +struct timeval timeval; +gettimeofday(&timeval, NULL); +int usecs = timeval.tv_sec* 1000000 + timeval.tv_usec; +gettimeofday(&timeval, NULL); +usecs = timeval.tv_sec* 1000000 + timeval.tv_usec - usecs; +fprintf(stderr, "%d micro-seconds\n", usecs); +*/ diff --git a/plugins/liboggedit/oggedit_utils.c b/plugins/liboggedit/oggedit_utils.c new file mode 100644 index 00000000..03e753a8 --- /dev/null +++ b/plugins/liboggedit/oggedit_utils.c @@ -0,0 +1,125 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library miscellaneous functions + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifdef HAVE_CONFIG_H +#include +#endif +#include +#include +#include +#include +#include + +uint8_t *oggedit_vorbis_channel_map(const int channel_count) +{ + size_t map_size = channel_count * sizeof(uint8_t); + uint8_t *map = malloc(map_size); + if (!map) + return NULL; + switch(channel_count) { + case 3: + return memcpy(map, &(uint8_t[]){0,2,1}, map_size); + case 5: + return memcpy(map, &(uint8_t[]){0,2,1,3,4}, map_size); + case 6: + return memcpy(map, &(uint8_t[]){0,2,1,4,5,3}, map_size); + case 7: + return memcpy(map, &(uint8_t[]){0,2,1,4,5,6,3}, map_size); + case 8: + return memcpy(map, &(uint8_t[]){0,2,1,6,7,4,5,3}, map_size); + default: + free(map); + return NULL; + } +} + +const char *oggedit_map_tag(char *key, const char *in_or_out) +{ + typedef struct { + const char *tag; + const char *meta; + } key_t; + const key_t keys[] = { + /* Permanent named tags in DeaDBeef */ +// {.tag = "ARTIST", .meta = "artist"}, +// {.tag = "TITLE", .meta = "title"}, +// {.tag = "ALBUM", .meta = "album"}, + {.tag = "DATE", .meta = "year"}, + {.tag = "TRACKNUMBER", .meta = "track"}, + {.tag = "TRACKTOTAL", .meta = "numtracks"}, +// {.tag = "GENRE", .meta = "genre"}, +// {.tag = "COMPOSER", .meta = "composer"}, + {.tag = "DISCNUMBER", .meta = "disc"}, +// {.tag = "COMMENT", .meta = "comment"}, + /* Vorbis standard tags */ +// {.tag = "ARRANGER", .meta = "Arranger"}, +// {.tag = "AUTHOR", .meta = "Author"}, +// {.tag = "CONDUCTOR", .meta = "Conductor"}, +// {.tag = "ENSEMBLE", .meta = "Ensemble"}, +// {.tag = "LYRICIST", .meta = "Lyricist"}, +// {.tag = "PERFORMER", .meta = "Performer"}, +// {.tag = "PUBLISHER", .meta = "Publisher"}, +// {.tag = "DISCTOTAL", .meta = "Disctotal"}, +// {.tag = "OPUS", .meta = "Opus"}, +// {.tag = "PART", .meta = "Part"}, +// {.tag = "PARTNUMBER", .meta = "Partnumber"}, +// {.tag = "VERSION", .meta = "Version"}, +// {.tag = "DESCRIPTION", .meta = "Description"}, +// {.tag = "COPYRIGHT", .meta = "Copyright"}, +// {.tag = "LICENSE", .meta = "License"}, +// {.tag = "CONTACT", .meta = "Contact"}, +// {.tag = "ORGANIZATION", .meta = "Organization"}, +// {.tag = "LOCATION", .meta = "Location"}, +// {.tag = "EAN/UPN", .meta = "EAN/UPN"}, +// {.tag = "ISRC", .meta = "ISRC"}, +// {.tag = "LABEL", .meta = "Label"}, +// {.tag = "LABELNO", .meta = "Labelno"}, +// {.tag = "ENCODER", .meta = "Encoder"}, +// {.tag = "ENCODED-BY", .meta = "Encoded-by"}, +// {.tag = "ENCODING", .meta = "Encoding"}, + /* Other tags */ +// {.tag = "ALBUMARTIST", .meta = "Albumartist"}, +// {.tag = "ALBUM ARTIST", .meta = "Album artist"}, +// {.tag = "BAND", .meta = "Band"}, +// {.tag = "COMPILATION", .meta = "Compilation"}, +// {.tag = "TOTALTRACKS", .meta = "Totaltracks"}, +// {.tag = "ENCODED_BY", .meta = "Encoded_by"}, +// {.tag = "ENCODER_OPTIONS",.meta = "Encoder_options"}, + {.tag = NULL} + }; + + /* Mapping for special Deadbeef internal metadata */ + for (const key_t *match = keys; match->tag; match++) + if (!strcasecmp(*in_or_out == 't' ? match->tag : match->meta, key)) + return *in_or_out == 't' ? match->meta : match->tag; + + /* Upper-case all Vorbis Comment tag names */ + if (*in_or_out == 'm') + for (size_t i = 0; key[i]; i++) + key[i] = toupper(key[i]); + + return key; +} diff --git a/plugins/liboggedit/oggedit_vorbis.c b/plugins/liboggedit/oggedit_vorbis.c new file mode 100644 index 00000000..2ff3fc09 --- /dev/null +++ b/plugins/liboggedit/oggedit_vorbis.c @@ -0,0 +1,168 @@ +/* + This file is part of Deadbeef Player source code + http://deadbeef.sourceforge.net + + DeaDBeeF Ogg Edit library Vorbis functions + + Copyright (C) 2014 Ian Nartowicz + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + +*/ + +#ifdef HAVE_CONFIG_H +#include +#endif +#include +#include +#include +#include +#include +#include +#include "oggedit_internal.h" +#include "oggedit.h" + +#define VCMAGIC "\3vorbis" +#define CODEMAGIC "\5vorbis" + +off_t oggedit_vorbis_stream_info(DB_FILE *in, const off_t start_offset, const off_t end_offset, char **codecs) +{ + int res; + ogg_sync_state oy; + ogg_sync_init(&oy); + *codecs = codec_names(in, &oy, start_offset, &res); + const off_t stream_size = codec_stream_size(in, &oy, start_offset, end_offset, VORBISNAME); + cleanup(in, NULL, &oy, NULL); + return stream_size; +} + +static ptrdiff_t check_vorbis_headers(DB_FILE *in, ogg_sync_state *oy, const off_t offset, char **vendor, ogg_packet *codebooks) +{ + ogg_stream_state os; + ogg_page og; + const int serial = init_read_stream(in, oy, &os, &og, offset, VORBISNAME); + if (serial <= OGGEDIT_EOF) + return serial; + + ogg_packet vc; + int pages = read_packet(in, oy, &os, &og, &vc, 1); + if (pages > OGGEDIT_EOF) + pages = read_packet(in, oy, &os, &og, codebooks, pages); + ogg_stream_clear(&os); + if (pages <= OGGEDIT_EOF) + return pages; + + if (vc.bytes > strlen(VCMAGIC) && !memcmp(vc.packet, VCMAGIC, strlen(VCMAGIC)) && + codebooks->bytes > strlen(CODEMAGIC) && !memcmp(codebooks->packet, CODEMAGIC, strlen(CODEMAGIC))) + *vendor = parse_vendor(&vc, strlen(VCMAGIC)); + free(vc.packet); + if (!*vendor) + return OGGEDIT_CANNOT_PARSE_HEADERS; + + if ((vc.bytes + codebooks->bytes) < MAXPAYLOAD * (pages-1)) + return 4; // prevent in-place write if the packets are split over too many pages + + return vc.bytes; +} + +static long write_vorbis_tags(FILE *out, const int serial, const char *vendor, const size_t num_tags, char **tags, const size_t padding, ogg_packet *codebooks) +{ + ogg_packet op; + if (!fill_vc_packet(VCMAGIC, strlen(VCMAGIC), vendor, num_tags, tags, true, padding, &op)) + return OGGEDIT_ALLOCATION_FAILURE; + + ogg_stream_state os; + if (ogg_stream_init(&os, serial)) + return OGGEDIT_FAILED_TO_INIT_STREAM; + os.b_o_s = 1; + os.pageno = 1; + ogg_stream_packetin(&os, &op); + ogg_stream_packetin(&os, codebooks); + + ogg_packet_clear(&op); + return flush_stream(out, &os); +} + +off_t oggedit_write_vorbis_metadata(DB_FILE *in, const char *fname, const off_t offset, const size_t stream_size, const int num_tags, char **tags) +{ + off_t res; + char tempname[PATH_MAX] = ""; + char *vendor = NULL; + ogg_packet codebooks; + memset(&codebooks, '\0', sizeof(codebooks)); + ogg_sync_state oy; + ogg_sync_init(&oy); + + /* Original file must be writable whichever way we update it */ + FILE *out = fopen(fname, "r+b"); + if (!out) { + res = OGGEDIT_CANNOT_UPDATE_FILE; + goto cleanup; + } + + /* See if we can write the tags packet directly into the existing file ... */ + const ptrdiff_t tags_packet_size = check_vorbis_headers(in, &oy, offset, &vendor, &codebooks); + if (tags_packet_size <= OGGEDIT_EOF) { + res = tags_packet_size; + goto cleanup; + } + const size_t metadata_size = strlen(VCMAGIC) + vc_size(vendor, num_tags, tags) + 1; + ptrdiff_t padding = tags_packet_size - metadata_size; + const off_t file_size_k = in->vfs->getlength(in) / 1000; + const size_t stream_size_k = stream_size ? stream_size / 1000 : file_size_k; + if (file_size_k < 100 || padding < 0 || padding > file_size_k/10+stream_size_k+metadata_size) + if (res = open_temp_file(fname, tempname, &out)) + goto cleanup; + + /* Re-pad if writing the whole file */ + if (*tempname) + padding = stream_size_k < 90 ? 0 : stream_size_k < 1000 ? 128 : stream_size_k < 10000 ? 1024 : 8192; + + /* Write pages until the correct comment header */ + ogg_page og; + const int vorbis_serial = copy_up_to_codec(in, out, &oy, &og, *tempname ? 0 : offset, offset, VORBISNAME); + if (vorbis_serial <= OGGEDIT_EOF) { + res = vorbis_serial; + goto cleanup; + } + if ((res = copy_up_to_header(in, out, &oy, &og, vorbis_serial)) <= OGGEDIT_EOF) + goto cleanup; + const long pageno = write_vorbis_tags(out, vorbis_serial, vendor, num_tags, tags, padding, &codebooks); + if (pageno < OGGEDIT_EOF) { + res = pageno; + goto cleanup; + } + + /* If we have tempfile, copy the remaining pages */ + if (*tempname) { + if ((res = copy_remaining_pages(in, out, &oy, vorbis_serial, pageno)) < OGGEDIT_EOF) + goto cleanup; + if (rename(tempname, fname)) { + res = OGGEDIT_RENAME_FAILED; + goto cleanup; + } + } + + res = file_size(fname); + +cleanup: + ogg_packet_clear(&codebooks); + cleanup(in, out, &oy, vendor); + if (res < OGGEDIT_OK) + unlink(tempname); + return res; +} -- cgit v1.2.3