1 /*
2
3 File: file_dxf.c
4
5 Copyright (C) 2009 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_STRING_H
27 #include <string.h>
28 #endif
29 #include <stdio.h>
30 #include "types.h"
31 #include "common.h"
32 #include "filegen.h"
33
34 static void register_header_check_dxf(file_stat_t *file_stat);
35 static data_check_t data_check_dxf(const unsigned char *buffer, const unsigned int buffer_size, file_recovery_t *file_recovery);
36 static void file_check_dxf(file_recovery_t *file_recovery);
37
38 const file_hint_t file_hint_dxf= {
39 .extension="dxf",
40 .description="Drawing Interchange File",
41 .max_filesize=PHOTOREC_MAX_FILE_SIZE,
42 .recover=1,
43 .enable_by_default=1,
44 .register_header_check=®ister_header_check_dxf
45 };
46
header_check_dxf(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)47 static int header_check_dxf(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)
48 {
49 reset_file_recovery(file_recovery_new);
50 file_recovery_new->extension=file_hint_dxf.extension;
51 file_recovery_new->file_check=&file_check_dxf;
52 if(file_recovery_new->blocksize >= 3)
53 {
54 file_recovery_new->data_check=&data_check_dxf;
55 }
56 return 1;
57 }
58
data_check_dxf(const unsigned char * buffer,const unsigned int buffer_size,file_recovery_t * file_recovery)59 static data_check_t data_check_dxf(const unsigned char *buffer, const unsigned int buffer_size, file_recovery_t *file_recovery)
60 {
61 unsigned int i;
62 for(i=(buffer_size/2)-3;i+4<buffer_size;i++)
63 {
64 if(buffer[i]=='\n' && buffer[i+1]=='E' && buffer[i+2]=='O' && buffer[i+3]=='F')
65 {
66 file_recovery->calculated_file_size=file_recovery->file_size+i+4-(buffer_size/2);
67 return DC_STOP;
68 }
69 }
70 file_recovery->calculated_file_size=file_recovery->file_size+(buffer_size/2);
71 return DC_CONTINUE;
72 }
73
file_check_dxf(file_recovery_t * file_recovery)74 static void file_check_dxf(file_recovery_t *file_recovery)
75 {
76 const unsigned char dxf_footer[4]= {'\n', 'E', 'O', 'F'};
77 file_search_footer(file_recovery, dxf_footer, sizeof(dxf_footer), 0);
78 file_allow_nl(file_recovery, NL_BARENL|NL_CRLF);
79 }
80
register_header_check_dxf(file_stat_t * file_stat)81 static void register_header_check_dxf(file_stat_t *file_stat)
82 {
83 static const unsigned char header_dxflib[10]= {'9', '9', '9', '\n',
84 'd', 'x', 'f', 'l', 'i', 'b'};
85 static const unsigned char header_dxflib_dos[11]= {'9', '9', '9', '\r', '\n',
86 'd', 'x', 'f', 'l', 'i', 'b'};
87 static const unsigned char header_dxf[11]= {' ', ' ', '0', '\n',
88 'S', 'E', 'C', 'T', 'I', 'O', 'N'};
89 static const unsigned char header_dxf_dos[12]= {' ', ' ', '0', '\r', '\n',
90 'S', 'E', 'C', 'T', 'I', 'O', 'N'};
91
92 register_header_check(0, header_dxf, sizeof(header_dxf), &header_check_dxf, file_stat);
93 register_header_check(0, header_dxf_dos, sizeof(header_dxf_dos), &header_check_dxf, file_stat);
94 register_header_check(0, header_dxflib, sizeof(header_dxflib), &header_check_dxf, file_stat);
95 register_header_check(0, header_dxflib_dos, sizeof(header_dxflib_dos), &header_check_dxf, file_stat);
96 }
97