File volumes are started NAND exported

So add a lock to ensure this.
This commit is contained in:
Demi Marie Obenour 2020-11-25 13:25:57 -05:00
parent 14e9154e4e
commit e4854df42f
No known key found for this signature in database
GPG Key ID: 28A45C93B0B5B6E0

View File

@ -176,11 +176,14 @@ class FilePool(qubes.storage.Pool):
class FileVolume(qubes.storage.Volume): class FileVolume(qubes.storage.Volume):
''' Parent class for the xen volumes implementation which expects a ''' Parent class for the xen volumes implementation which expects a
`target_dir` param on initialization. ''' `target_dir` param on initialization. '''
_marker_running = object()
_marker_exported = object()
def __init__(self, dir_path, **kwargs): def __init__(self, dir_path, **kwargs):
self.dir_path = dir_path self.dir_path = dir_path
assert self.dir_path, "dir_path not specified" assert self.dir_path, "dir_path not specified"
self._revisions_to_keep = 0 self._revisions_to_keep = 0
self._locked = None
super().__init__(**kwargs) super().__init__(**kwargs)
if self.snap_on_start: if self.snap_on_start:
@ -265,10 +268,18 @@ class FileVolume(qubes.storage.Volume):
return self return self
def export(self): def export(self):
if self.is_dirty(): if self._locked is not None:
assert self._locked is FileVolume._marker_running, \
'nested calls to export()'
self._not_implemented('exporting a dirty volume') self._not_implemented('exporting a dirty volume')
self._locked = FileVolume._marker_exported
return self.path return self.path
def export_end(self, path):
assert self._locked is FileVolume._marker_exported, \
'cannot end an export that never began'
self._locked = None
@asyncio.coroutine @asyncio.coroutine
def import_volume(self, src_volume): def import_volume(self, src_volume):
if src_volume.snap_on_start: if src_volume.snap_on_start:
@ -311,6 +322,11 @@ class FileVolume(qubes.storage.Volume):
return self return self
def start(self): def start(self):
if self._locked is not None:
assert self._locked is FileVolume._marker_exported, \
'nested calls to start()'
self._not_implemented('starting a VM with an exported volume')
self._locked = FileVolume._marker_running
if not self.save_on_stop and not self.snap_on_start: if not self.save_on_stop and not self.snap_on_start:
self.reset() self.reset()
else: else:
@ -328,12 +344,15 @@ class FileVolume(qubes.storage.Volume):
return self return self
def stop(self): def stop(self):
assert self._locked is FileVolume._marker_running, \
'cannot stop a volume that has not been started'
if self.save_on_stop: if self.save_on_stop:
self.commit() self.commit()
elif self.snap_on_start: elif self.snap_on_start:
_remove_if_exists(self.path_cow) _remove_if_exists(self.path_cow)
else: else:
_remove_if_exists(self.path) _remove_if_exists(self.path)
self._locked = None
return self return self
@property @property