summaryrefslogtreecommitdiffstats
path: root/src/file_wv.c
diff options
context:
space:
mode:
authorChristophe Grenier <grenier@cgsecurity.org>2008-11-13 09:19:30 +0100
committerChristophe Grenier <grenier@cgsecurity.org>2008-11-13 09:19:30 +0100
commit3f13734b7f60e28be661fa747c73f44cca873175 (patch)
tree6951198f7ca00834510fea75e13dd4b404d90444 /src/file_wv.c
parentcf9eaa3dbb96ce9f480e6e8243193523b01458be (diff)
PhotoRec: recover WavPack, Hybrid Lossless Wavefile Compressor, .wv files
Diffstat (limited to 'src/file_wv.c')
-rw-r--r--src/file_wv.c117
1 files changed, 117 insertions, 0 deletions
diff --git a/src/file_wv.c b/src/file_wv.c
new file mode 100644
index 0000000..80bbeb5
--- /dev/null
+++ b/src/file_wv.c
@@ -0,0 +1,117 @@
+/*
+
+ File: file_wv.c
+
+ Copyright (C) 2008 Christophe GRENIER <grenier@cgsecurity.org>
+
+ This software is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License along
+ with this program; if not, write the Free Software Foundation, Inc., 51
+ Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+#ifdef HAVE_STRING_H
+#include <string.h>
+#endif
+#include <stdio.h>
+#include "types.h"
+#include "filegen.h"
+#include "common.h"
+
+static void register_header_check_wv(file_stat_t *file_stat);
+static int header_check_wv(const unsigned char *buffer, const unsigned int buffer_size, const unsigned int safe_header_only, const file_recovery_t *file_recovery, file_recovery_t *file_recovery_new);
+static int data_check_wv(const unsigned char *buffer, const unsigned int buffer_size, file_recovery_t *file_recovery);
+
+const file_hint_t file_hint_wv= {
+ .extension="wv",
+ .description="WavPack, Hybrid Lossless Wavefile Compressor",
+ .min_header_distance=0,
+ .max_filesize=100*1024*1024,
+ .recover=1,
+ .enable_by_default=1,
+ .register_header_check=&register_header_check_wv
+};
+
+/* See http://www.wavpack.com/file_format.txt for file format description */
+
+static const unsigned char wv_header[4]= { 'w', 'v', 'p', 'k'};
+typedef struct {
+ char ckID [4]; // "wvpk"
+ uint32_t ckSize; // size of entire block (minus 8, of course)
+ uint16_t version; // 0x402 to 0x410 are currently valid for decode
+ unsigned char track_no; // track number (0 if not used, like now)
+ unsigned char index_no; // track sub-index (0 if not used, like now)
+ uint32_t total_samples; // total samples for entire file, but this is
+ // only valid if block_index == 0 and a value of
+ // -1 indicates unknown length
+ uint32_t block_index; // index of first sample in block relative to
+ // beginning of file (normally this would start
+ // at 0 for the first block)
+ uint32_t block_samples; // number of samples in this block (0 = no audio)
+ uint32_t flags; // various flags for id and decoding
+ uint32_t crc; // crc for actual decoded data
+} WavpackHeader;
+
+static void register_header_check_wv(file_stat_t *file_stat)
+{
+ register_header_check(0, wv_header, sizeof(wv_header), &header_check_wv, file_stat);
+}
+
+static int header_check_wv(const unsigned char *buffer, const unsigned int buffer_size, const unsigned int safe_header_only, const file_recovery_t *file_recovery, file_recovery_t *file_recovery_new)
+{
+ const WavpackHeader *wv=(const WavpackHeader*)buffer;
+ if(memcmp(buffer, wv_header, sizeof(wv_header))==0 && le32(wv->block_index)==0)
+ {
+ reset_file_recovery(file_recovery_new);
+ file_recovery_new->extension=file_hint_wv.extension;
+ file_recovery_new->data_check=&data_check_wv;
+ return 1;
+ }
+ return 0;
+}
+
+static int data_check_wv(const unsigned char *buffer, const unsigned int buffer_size, file_recovery_t *file_recovery)
+{
+ while(file_recovery->calculated_file_size + buffer_size/2 >= file_recovery->file_size &&
+ file_recovery->calculated_file_size + 8 < file_recovery->file_size + buffer_size/2)
+ {
+ const unsigned int i=file_recovery->calculated_file_size - file_recovery->file_size + buffer_size/2;
+ const WavpackHeader *wv=(const WavpackHeader*)&buffer[i];
+ if(memcmp(wv, wv_header, sizeof(wv_header))==0)
+ {
+ file_recovery->calculated_file_size+=le32(wv->ckSize)+8;
+ }
+ else if(buffer[i]=='A' && buffer[i+1]=='P' && buffer[i+2]=='E' && buffer[i+3]=='T' && buffer[i+4]=='A' && buffer[i+5]=='G' && buffer[i+6]=='E' && buffer[i+7]=='X')
+ { /* APE Tagv2 (APE Tagv1 has no header) http://wiki.hydrogenaudio.org/index.php?title=APE_Tags_Header */
+ const unsigned int ape_tag_size = (buffer[i+12] + (buffer[i+13]<<8) + (buffer[i+14]<<16) + (buffer[i+15]<<24))+32;
+ file_recovery->calculated_file_size+=ape_tag_size;
+ }
+ else if(buffer[i]=='T' && buffer[i+1]=='A' && buffer[i+2]=='G')
+ { /* http://www.id3.org/ID3v1 TAGv1 size = 128 bytes with header "TAG" */
+ file_recovery->calculated_file_size+=128;
+ }
+ else if(file_recovery->calculated_file_size > file_recovery->file_size)
+ {
+ return 1;
+ }
+ else
+ {
+ return 2;
+ }
+ }
+ return 1;
+}
+