LVM support added

1 files changed, 32 insertions, 14 deletions

diff --git a/cumount.py b/cumount.py
index 16aa2f3..e2fb32f 100755
--- a/cumount.py
+++ b/cumount.py
@@ -20,20 +20,38 @@ def main():
             config_dict = json.load(fp)
         cstm_error = 'Invalid / nonexistent mapping: {0}'.format(sys.argv[1])
         data = config_dict[sys.argv[1]]
-        subprocess.check_call(['/bin/umount',
-                               '/dev/mapper/' + data['mapper_name']])
-        subprocess.check_call(['/sbin/cryptsetup',
-                               'luksClose',
-                               data['mapper_name']])
-        if data.get('type') == 'efile':
-            cstm_error = 'Unable to destroy loop device'
-            loop_device = subprocess.check_output(['/usr/sbin/losetup',
-                                                   '--associated',
-                                                   data['device']])
-            loop_device = loop_device.split(':')[0]
-            subprocess.check_call(['/usr/sbin/losetup',
-                                   '-d',
-                                   loop_device])
+        if data.get('type') == 'lvm':
+            if not isinstance(data.get('pv_sources'), dict):
+                cstm_error = 'Invalid config for lvm-type'
+                raise Exception()
+            cstm_error = 'umount error'
+            subprocess.check_call(['/bin/umount',
+                                   data['device']])
+            cstm_error = 'lvchange error'
+            subprocess.check_call(['/sbin/lvchange',
+                                   '-a',
+                                   'n',
+                                   data['device']])
+            cstm_error = 'pv_sources error'
+            for mapper_name in data['pv_sources'].values():
+                subprocess.check_call(['/sbin/cryptsetup',
+                                       'luksClose',
+                                       mapper_name])
+        else:
+            subprocess.check_call(['/bin/umount',
+                                   '/dev/mapper/' + data['mapper_name']])
+            subprocess.check_call(['/sbin/cryptsetup',
+                                   'luksClose',
+                                   data['mapper_name']])
+            if data.get('type') == 'efile':
+                cstm_error = 'Unable to destroy loop device'
+                loop_device = subprocess.check_output(['/usr/sbin/losetup',
+                                                       '--associated',
+                                                       data['device']])
+                loop_device = loop_device.split(':')[0]
+                subprocess.check_call(['/usr/sbin/losetup',
+                                       '-d',
+                                       loop_device])
     except subprocess.CalledProcessError as e:
         sys.stderr.write('Execution error: {0}\n'.format(e))
         sys.stderr.flush()