1 /*
2
3 File: file_tib.c
4
5 Copyright (C) 2007-2010 Christophe GRENIER <grenier@cgsecurity.org>
6
7 This software is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
11
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License along
18 with this program; if not, write the Free Software Foundation, Inc., 51
19 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20
21 */
22
23 #ifdef HAVE_CONFIG_H
24 #include <config.h>
25 #endif
26 #ifdef HAVE_STDLIB_H
27 #include <stdlib.h>
28 #endif
29 #ifdef HAVE_STRING_H
30 #include <string.h>
31 #endif
32 #include <stdio.h>
33 #include "types.h"
34 #include "filegen.h"
35 #include "common.h"
36
37 static void register_header_check_tib(file_stat_t *file_stat);
38 static int header_check_tib(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);
39
40 const file_hint_t file_hint_tib= {
41 .extension="tib",
42 .description="Acronis True Image",
43 .max_filesize=PHOTOREC_MAX_FILE_SIZE,
44 .recover=1,
45 .enable_by_default=1,
46 .register_header_check=®ister_header_check_tib
47 };
48
49 static const unsigned char tib2_footer[7]= {0x00, 0x00, 0x20, 0xa2, 0xb9, 0x24, 0xce};
50
data_check_tib2(const unsigned char * buffer,const unsigned int buffer_size,file_recovery_t * file_recovery)51 static data_check_t data_check_tib2(const unsigned char *buffer, const unsigned int buffer_size, file_recovery_t *file_recovery)
52 {
53 while(file_recovery->calculated_file_size + buffer_size/2 >= file_recovery->file_size &&
54 file_recovery->calculated_file_size + 512 <= file_recovery->file_size + buffer_size/2)
55 {
56 const unsigned int i=file_recovery->calculated_file_size - file_recovery->file_size + buffer_size/2;
57 file_recovery->calculated_file_size+=512;
58 if(memcmp(&buffer[i + 512 - sizeof(tib2_footer)], tib2_footer, sizeof(tib2_footer))==0)
59 return DC_STOP;
60 }
61 return DC_CONTINUE;
62 }
63
file_check_tib2(file_recovery_t * file_recovery)64 static void file_check_tib2(file_recovery_t *file_recovery)
65 {
66 unsigned char*buffer=(unsigned char*)MALLOC(512);
67 int64_t file_size=file_recovery->calculated_file_size-512;
68 file_recovery->file_size = file_recovery->calculated_file_size;
69 if(my_fseek(file_recovery->handle, file_size, SEEK_SET) < 0 ||
70 fread(buffer, 1, 512, file_recovery->handle) != 512)
71 {
72 free(buffer);
73 file_recovery->file_size=0;
74 return;
75 }
76 if(memcmp(&buffer[512 - sizeof(tib2_footer)], tib2_footer, sizeof(tib2_footer))==0)
77 {
78 free(buffer);
79 return;
80 }
81
82 for(; file_size>0; file_size-=512)
83 {
84 unsigned int i;
85 if(my_fseek(file_recovery->handle, file_size, SEEK_SET) < 0 ||
86 fread(buffer, 1, 512, file_recovery->handle) != 512)
87 {
88 free(buffer);
89 file_recovery->file_size=0;
90 return;
91 }
92 for(i=0; i<512 && buffer[i]==0; i++);
93 if(i!=512)
94 {
95 file_recovery->file_size=file_size + 512;
96 free(buffer);
97 return ;
98 }
99 }
100 free(buffer);
101 }
102
header_check_tib(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)103 static int header_check_tib(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)
104 {
105 reset_file_recovery(file_recovery_new);
106 file_recovery_new->extension=file_hint_tib.extension;
107 return 1;
108 }
109
header_check_tib2(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)110 static int header_check_tib2(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)
111 {
112 reset_file_recovery(file_recovery_new);
113 file_recovery_new->extension=file_hint_tib.extension;
114 if(file_recovery_new->blocksize < 512)
115 return 1;
116 file_recovery_new->file_check=&file_check_tib2;
117 file_recovery_new->data_check=&data_check_tib2;
118 return 1;
119 }
120
register_header_check_tib(file_stat_t * file_stat)121 static void register_header_check_tib(file_stat_t *file_stat)
122 {
123 static const unsigned char tib_header[4]= { 0xb4, 0x6e, 0x68, 0x44};
124 static const unsigned char tib2_header[7]= { 0xce, 0x24, 0xb9, 0xa2, 0x20, 0x00, 0x00};
125 register_header_check(0, tib_header,sizeof(tib_header), &header_check_tib, file_stat);
126 register_header_check(0, tib2_header,sizeof(tib2_header), &header_check_tib2, file_stat);
127 }
128