1 /*
2 
3     File: file_drw.c
4 
5     Copyright (C) 2008 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 <ctype.h>
31 #include "types.h"
32 #include "filegen.h"
33 
34 static void register_header_check_drw(file_stat_t *file_stat);
35 
36 const file_hint_t file_hint_drw= {
37   .extension="drw",
38   .description="Pro/ENGINEER Drawing",
39   .max_filesize=PHOTOREC_MAX_FILE_SIZE,
40   .recover=1,
41   .enable_by_default=1,
42   .register_header_check=&register_header_check_drw
43 };
44 
file_check_drw(file_recovery_t * file_recovery)45 static void file_check_drw(file_recovery_t *file_recovery)
46 {
47   const unsigned char drw_footer[11]= {
48     '#', 'E', 'N', 'D', '_', 'O', 'F', '_',
49     'U', 'G', 'C'};
50   file_search_footer(file_recovery, drw_footer, sizeof(drw_footer), 1);
51 }
52 
header_check_drw(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)53 static int header_check_drw(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)
54 {
55   if(!isprint(buffer[14]) || !isprint(buffer[15]) || !isprint(buffer[16]) || !isprint(buffer[17]))
56     return 0;
57   reset_file_recovery(file_recovery_new);
58   file_recovery_new->file_check=&file_check_drw;
59   file_recovery_new->extension=file_hint_drw.extension;
60   return 1;
61 }
62 
register_header_check_drw(file_stat_t * file_stat)63 static void register_header_check_drw(file_stat_t *file_stat)
64 {
65   static const unsigned char drw_header[14]= {
66     '#', 'U', 'G', 'C', ':', '2', ' ', 'D',
67     'R', 'A', 'W', 'I', 'N', 'G'};
68   register_header_check(0, drw_header,sizeof(drw_header), &header_check_drw, file_stat);
69 }
70