summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/Makefile.am1
-rw-r--r--src/file_list.c2
-rw-r--r--src/file_pdb.c98
3 files changed, 101 insertions, 0 deletions
diff --git a/src/Makefile.am b/src/Makefile.am
index 532c578..ad8554a 100644
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -237,6 +237,7 @@ file_C = filegen.c \
file_pcb.c \
file_pct.c \
file_pcx.c \
+ file_pdb.c \
file_pdf.c \
file_pds.c \
file_pf.c \
diff --git a/src/file_list.c b/src/file_list.c
index 4fd428f..d8b9a56 100644
--- a/src/file_list.c
+++ b/src/file_list.c
@@ -240,6 +240,7 @@ extern const file_hint_t file_hint_pcap;
extern const file_hint_t file_hint_pcb;
extern const file_hint_t file_hint_pct;
extern const file_hint_t file_hint_pcx;
+extern const file_hint_t file_hint_pdb;
extern const file_hint_t file_hint_pdf;
extern const file_hint_t file_hint_pds;
extern const file_hint_t file_hint_pf;
@@ -574,6 +575,7 @@ file_enable_t list_file_enable[]=
{ .enable=0, .file_hint=&file_hint_pcb },
{ .enable=0, .file_hint=&file_hint_pct },
{ .enable=0, .file_hint=&file_hint_pcx },
+ { .enable=0, .file_hint=&file_hint_pdb },
{ .enable=0, .file_hint=&file_hint_pdf },
{ .enable=0, .file_hint=&file_hint_pds },
{ .enable=0, .file_hint=&file_hint_pf },
diff --git a/src/file_pdb.c b/src/file_pdb.c
new file mode 100644
index 0000000..4a9ec33
--- /dev/null
+++ b/src/file_pdb.c
@@ -0,0 +1,98 @@
+/*
+
+ File: file_pdb.c
+
+ Copyright (C) 2018 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"
+
+static void register_header_check_pdb(file_stat_t *file_stat);
+
+const file_hint_t file_hint_pdb= {
+ .extension="pdb",
+ .description="Protein Data Bank data",
+ .max_filesize=PHOTOREC_MAX_FILE_SIZE,
+ .recover=1,
+ .enable_by_default=1,
+ .register_header_check=&register_header_check_pdb
+};
+
+static data_check_t data_check_pdb(const unsigned char *buffer, const unsigned int buffer_size, file_recovery_t *file_recovery)
+{
+ unsigned int i;
+ for(i=buffer_size/2; i<buffer_size; i++)
+ if(buffer[i]==0)
+ {
+ file_recovery->calculated_file_size+=i;
+ return DC_STOP;
+ }
+ file_recovery->calculated_file_size+=buffer_size/2;
+ return DC_CONTINUE;
+}
+
+static void file_check_pdb(file_recovery_t *file_recovery)
+{
+ unsigned char buffer[512];
+ if(my_fseek(file_recovery->handle, 0, SEEK_SET) < 0 ||
+ fread(&buffer, 1, sizeof(buffer), file_recovery->handle) < 82)
+ return ;
+ if(buffer[80]=='\r' && buffer[81]=='\n')
+ file_recovery->file_size=file_recovery->calculated_file_size/82*82;
+ else if(buffer[80]=='\n')
+ file_recovery->file_size=file_recovery->calculated_file_size/81*81;
+ else
+ file_recovery->file_size=0;
+}
+
+static int header_check_pdb(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)
+{
+ /* Check date */
+ if(buffer[0x32] < '0' || buffer[0x32] > '9' || buffer[0x33] < '0' || buffer[0x33] > '9')
+ return 0;
+ if(buffer[0x34]!='-')
+ return 0;
+ if(buffer[0x35] < 'A' || buffer[0x35] > 'Z' || buffer[0x36] < 'A' || buffer[0x36] > 'Z' || buffer[0x37] < 'A' || buffer[0x37] > 'Z')
+ return 0;
+ if(buffer[0x38]!='-')
+ return 0;
+ if(buffer[0x39] < '0' || buffer[0x39] > '9' || buffer[0x3a] < '0' || buffer[0x3a] > '9')
+ return 0;
+ /* Check space */
+ if(buffer[59]!=' ' || buffer[60]!=' ' || buffer[61]!=' ' || buffer[66]!=' ' || buffer[67]!=' ' || buffer[68]!=' ' || buffer[69]!=' ')
+ return 0;
+ reset_file_recovery(file_recovery_new);
+ file_recovery_new->extension=file_hint_pdb.extension;
+ file_recovery_new->data_check=&data_check_pdb;
+ file_recovery_new->file_check=&file_check_pdb;
+ file_recovery_new->min_filesize=80;
+ return 1;
+}
+
+static void register_header_check_pdb(file_stat_t *file_stat)
+{
+ register_header_check(0, "HEADER ", 10, &header_check_pdb, file_stat);
+}