Merge remote-tracking branch 'upstream/new-ddrescue-arguments' into new-ddrescue-arguments

scripts/wk/cfg/ddrescue.py

@@ -17,9 +17,10 @@ TMUX_LAYOUT = OrderedDict({
 # ddrescue
 AUTO_PASS_THRESHOLDS = {
   # NOTE: The scrape key is set to infinity to force a break
-  'read': 95,
-  'trim': 98,
-  'scrape': float('inf'),
+  'read-skip':  50,
+  'read-full':  95,
+  'trim':       98,
+  'scrape':     float('inf'),
   }
 DDRESCUE_MAP_TEMPLATE = '''# Mapfile. Created by {name}
 0x0 ? 1
@@ -36,10 +37,10 @@ DDRESCUE_SETTINGS = {
     '--max-error-rate':   {'Selected': True,  'Value': '100MiB',            },
     '--max-read-rate':    {'Selected': False, 'Value': '1MiB',              },
     '--min-read-rate':    {'Selected': True,  'Value': '64KiB',             },
-    '--reopen-on-error':  {'Selected': True,                                },
+    '--reopen-on-error':  {'Selected': False,                               },
     '--retry-passes':     {'Selected': True,  'Value': '0',                 },
     '--reverse':          {'Selected': False,                               },
-    '--skip-size':        {'Selected': True,  'Value': '0.0001,0.01',       },  # Percentages of source size
+    '--skip-size':        {'Selected': True,  'Value': '0.001,0.05',        },  # Percentages of source size
     '--test-mode':        {'Selected': False, 'Value': 'test.map',          },
     '--timeout':          {'Selected': True,  'Value': '30m',               },
     '-vvvv':              {'Selected': True,                'Hidden': True, },
@@ -47,17 +48,19 @@ DDRESCUE_SETTINGS = {
   'Fast': {
     '--max-error-rate':   {'Selected': True,  'Value': '32MiB',             },
     '--min-read-rate':    {'Selected': True,  'Value': '1MiB',              },
-    '--reopen-on-error':  {'Selected': False,                               },
     '--timeout':          {'Selected': True,  'Value': '5m',                },
     },
   'Safe': {
     '--max-read-rate':    {'Selected': True,  'Value': '64MiB',             },
     '--min-read-rate':    {'Selected': True,  'Value': '1KiB',              },
-    '--reopen-on-error':  {'Selected': True,                                },
-    '--skip-size':        {'Selected': True,  'Value': '0.001,0.05',        },  # Percentages of source size
     '--timeout':          {'Selected': False, 'Value': '30m',               },
     },
   }
+DDRESCUE_SPECIFIC_PASS_SETTINGS = {
+  'read-skip':  ['--no-scrape', '--no-trim', '--cpass=1,2'],
+  'read-full':  ['--no-scrape', '--no-trim'],
+  'trim':       ['--no-scrape'],
+  }
 DRIVE_POWEROFF_TIMEOUT = 90
 PARTITION_TYPES = {
   'GPT': {

scripts/wk/hw/ddrescue.py

@@ -22,7 +22,11 @@ import psutil
 import pytz
 
 from wk import cfg, debug, exe, io, log, net, osticket, std, tmux
-from wk.cfg.ddrescue import DDRESCUE_MAP_TEMPLATE, DDRESCUE_SETTINGS
+from wk.cfg.ddrescue import (
+  DDRESCUE_MAP_TEMPLATE,
+  DDRESCUE_SETTINGS,
+  DDRESCUE_SPECIFIC_PASS_SETTINGS,
+  )
 from wk.hw import obj as hw_obj
 
 
@@ -107,11 +111,11 @@ SETTING_PRESETS = (
   'Safe',
   )
 STATUS_COLORS = {
-  'Passed': 'GREEN',
-  'Aborted': 'YELLOW',
-  'Skipped': 'YELLOW',
-  'Working': 'YELLOW',
-  'ERROR': 'RED',
+  'Passed':   'GREEN',
+  'Aborted':  'YELLOW',
+  'Skipped':  'YELLOW',
+  'Working':  'YELLOW',
+  'ERROR':    'RED',
   }
 TIMEZONE = pytz.timezone(cfg.main.LINUX_TIME_ZONE)
 
@@ -133,9 +137,10 @@ class BlockPair():
     self.map_path = None
     self.size = source.details['size']
     self.status = OrderedDict({
-      'read': 'Pending',
-      'trim': 'Pending',
-      'scrape': 'Pending',
+      'read-skip':  'Pending',
+      'read-full':  'Pending',
+      'trim':       'Pending',
+      'scrape':     'Pending',
       })
     self.view_map = 'DISPLAY' in os.environ or 'WAYLAND_DISPLAY' in os.environ
 
@@ -326,10 +331,9 @@ class BlockPair():
 
     # Mark future passes as skipped if applicable
     if percent == 100:
-      if pass_name == 'read':
-        self.status['trim'] = 'Skipped'
-      if pass_name in ('read', 'trim'):
-        self.status['scrape'] = 'Skipped'
+      status_keys = list(self.status.keys())
+      for i in status_keys[status_keys.index(pass_name)+1:]:
+        self.status[status_keys[i]] = 'Skipped'
 
 
 class State():
@@ -1306,14 +1310,7 @@ def build_ddrescue_cmd(block_pair, pass_name, settings_menu):
   if (block_pair.destination.is_block_device()
       or block_pair.destination.is_char_device()):
     cmd.append('--force')
-  if pass_name == 'read':
-    cmd.extend(['--no-trim', '--no-scrape'])
-  elif pass_name == 'trim':
-    # Allow trimming
-    cmd.append('--no-scrape')
-  elif pass_name == 'scrape':
-    # Allow trimming and scraping
-    pass
+  cmd.extend(DDRESCUE_SPECIFIC_PASS_SETTINGS.get(pass_name, []))
 
   # Fix domain size based on starting position
   domain_size = block_pair.size
@@ -2313,7 +2310,7 @@ def run_recovery(state, main_menu, settings_menu, dry_run=True):
       )
 
   # Run pass(es)
-  for pass_name in ('read', 'trim', 'scrape'):
+  for pass_name in ('read-skip', 'read-full', 'trim', 'scrape'):
     abort = False
 
     # Skip to next pass