1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
#!/usr/bin/env python
# RaidGuessFS, a FUSE pseudo-filesystem to guess RAID parameters of a damaged device
# Copyright (C) 2015 Ludovic Pouzenc <ludovic@pouzenc.fr>
#
# This file is part of RaidGuessFS.
#
# RaidGuessFS 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 3 of the License, or
# (at your option) any later version.
#
# RaidGuessFS 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 RaidGuessFS. If not, see <http://www.gnu.org/licenses/>
import os, multiprocessing, binascii, logging
import mydisks
def do_find_files(d,state):
logging.info("Enter do_find_files()")
try:
state['state'] = 'initializing'
ref_paths = state['filepaths']
ref_count = len(ref_paths)
ref_fds = [None]*ref_count
ref_sizes = [None]*ref_count
ref_offset = [None]*ref_count
ref_cur_sect = [None]*ref_count
for ref_no in range(ref_count):
path = state['filepaths'][ref_no]
logging.debug("Try to open reffile '%s'"%path)
ref_offset[ref_no] = 0
ref_sizes[ref_no] = os.lstat(path).st_size
ref_fds[ref_no] = open(path, "r")
ref_fds[ref_no].seek(0)
ref_cur_sect[ref_no] = ref_fds[ref_no].read(512)
logging.debug("Opened reffile '%s'"%path)
start = 0
end = min(d.disks_size)
one_percent = (end - start) / 100
one_percent = one_percent + ( (-one_percent)%512 )
logging.debug("start/end/1pc : %i / %i / %i"%(start,end,one_percent))
state['found'] = []
state['progress'] = 0
state['state'] = 'searching'
for offset in range(start, end, 512):
for disk_no in range(d.disk_count):
d.disks[disk_no].seek(offset)
data = d.disks[disk_no].read(512)
for ref_no in range(ref_count):
#TODO : gerer les buffers plus court que 512
if data == ref_cur_sect[ref_no]:
#TODO : chercher la suite du fichier
f = state['found']
if len(f) < 200:
# TODO agreger les matches
f.append((ref_paths[ref_no],ref_offset[ref_no],disk_no,offset))
state['found'] = f
else:
state['state'] = 'aborted'
raise Exception('Aborting after too many matches')
if offset % one_percent == 0:
state['progress'] = state['progress'] + 1
state['state'] = 'finished'
state['progress'] = 100
except Exception as e:
logging.exception(e)
logging.info("Exit. do_find_files()")
def do_find_bootsect(d,state):
logging.info("Enter do_find_bootsect()")
try:
state['state'] = 'initializing'
ref_sig = binascii.unhexlify('55AA')
start = 0
end = min(d.disks_size)
one_percent = (end - start) / 100
one_percent = one_percent + ( (-one_percent)%512 )
logging.debug("start/end/1pc : %i / %i / %i"%(start,end,one_percent))
state['found'] = []
state['progress'] = 0
state['state'] = 'searching'
for offset in range(start, end, 512):
for disk_no in range(d.disk_count):
d.disks[disk_no].seek(offset)
data = d.disks[disk_no].read(512)
sig = data[510:]
if sig == ref_sig:
f = state['found']
if len(f) < 200:
f.append((disk_no,offset))
state['found'] = f
else:
state['state'] = 'aborted'
raise Exception('Aborting after too many matches')
if offset % one_percent == 0:
state['progress'] = state['progress'] + 1
state['progress'] = 100
state['state'] = 'finished'
except Exception as e:
logging.exception(e)
logging.info("Exit. do_find_bootsect()")
class MyTasks():
"""Auxiliary class, managing long or background tasks"""
TASK_NAMES = [ 'find_bootsect', 'find_files' ]
def __init__(self, mydisks):
self.tasks = []
self.d = mydisks
self.find_files_pathlist = []
m = multiprocessing.Manager()
self.find_bootsect_state = m.dict()
self.find_bootsect_process = None
self.find_files_state = m.dict()
self.find_files_process = None
def get_find_files_pathlist(self):
return self.find_files_pathlist
def get_find_files_pathlist_str(self):
return '\n'.join(self.find_files_pathlist)
def task_start(self, task_name):
if task_name == 'find_files':
self.find_files_state['filepaths'] = list(self.find_files_pathlist)
self.find_files_process = multiprocessing.Process(
target = do_find_files,
args = (self.d, self.find_files_state)
)
self.find_files_process.start()
elif task_name == 'find_bootsect':
self.find_bootsect_process = multiprocessing.Process(
target = do_find_bootsect,
args = (self.d, self.find_bootsect_state)
)
self.find_bootsect_process.start()
else:
raise ValueError('Valid task names are : %s'%','.join(MyTasks.TASK_NAMES))
def task_kill(self, task_name):
if task_name == 'find_bootsect':
if self.find_bootsect_process != None and self.find_bootsect_process.is_alive():
self.find_bootsect_process.terminate()
elif task_name == 'find_files':
if self.find_files_process != None and self.find_files_process.is_alive():
self.find_files_process.terminate()
else:
raise ValueError('Valid task names are : %s'%','.join(MyTasks.TASK_NAMES))
def append_find_files_pathlist(self, pathlist):
# TODO : should receive a list, make changes in raidguessfs.py
self.find_files_pathlist.extend(pathlist.split('\n'))
def set_find_files_pathlist(self, new_find_files_pathlist):
self.find_files_pathlist = new_find_files_pathlist
def read_find_bootsect(self):
if self.find_bootsect_process == None:
return 'This task has never been started\n'
else:
return '%s\n'%self.find_bootsect_state
def read_find_files(self):
if self.find_files_process == None:
return 'This task has never been started\n'
else:
return '%s\n'%self.find_files_state
|
