diff options
Diffstat (limited to 'mytasks.py')
| -rw-r--r-- | mytasks.py | 46 |
1 files changed, 41 insertions, 5 deletions
@@ -18,15 +18,51 @@ # You should have received a copy of the GNU General Public License # along with RaidGuessFS. If not, see <http://www.gnu.org/licenses/> -import multiprocessing, logging +import multiprocessing, binascii, logging import mydisks def do_find_files(d,state): - state['TODO'] = 'Not yet implemented' + logging.info("Enter do_find_files()") + try: + state['TODO'] = 'Not yet implemented' + state['progress'] = 100 + except Exception as e: + logging.exception(e) + logging.info("Exit. do_find_files()") def do_find_bootsect(d,state): - state['TODO'] = 'Not yet implemented' + logging.info("Enter do_find_bootsect()") + try: + 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 + 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: + raise Exception('Aborting after too many matches') + + if offset % one_percent == 0: + state['progress'] = state['progress'] + 1 + state['progress'] = 100 + except Exception as e: + logging.exception(e) + logging.info("Exit. do_find_bootsect()") @@ -84,11 +120,11 @@ class MyTasks(): if self.find_bootsect_process == None: return 'This task has never been started\n' else: - return 'TODO\n%s\n'%self.find_bootsect_state + 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 'TODO\n%s\n'%self.find_files_state + return '%s\n'%self.find_files_state |