1b5daea771
In some (most) cases VM needs to be started to complete resize operation. This may be unexpected, so make it clear and do not start the VM when the user did not explicitly allow that. Fixes QubesOS/qubes-issues#1268
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
#!/usr/bin/python2
|
|
# -*- encoding: utf8 -*-
|
|
#
|
|
# The Qubes OS Project, http://www.qubes-os.org
|
|
#
|
|
# This program 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 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program 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 this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
#
|
|
|
|
from __future__ import (
|
|
absolute_import,
|
|
division,
|
|
print_function,
|
|
unicode_literals,
|
|
)
|
|
|
|
from qubes.qubes import (
|
|
register_qubes_vm_class,
|
|
QubesException,
|
|
QubesVm,
|
|
)
|
|
from time import sleep
|
|
|
|
|
|
class QubesResizableVm(QubesVm):
|
|
|
|
def resize_root_img(self, size, allow_start=False):
|
|
if self.template:
|
|
raise QubesException("Cannot resize root.img of template-based VM"
|
|
". Resize the root.img of the template "
|
|
"instead.")
|
|
|
|
if self.is_running():
|
|
raise QubesException("Cannot resize root.img of running VM")
|
|
|
|
if size < self.get_root_img_sz():
|
|
raise QubesException(
|
|
"For your own safety shringing of root.img is disabled. If "
|
|
"you really know what you are doing, use 'truncate' manually.")
|
|
|
|
f_root = open(self.root_img, "a+b")
|
|
f_root.truncate(size)
|
|
f_root.close()
|
|
|
|
|
|
class QubesResizableVmWithResize2fs(QubesResizableVm):
|
|
|
|
def resize_root_img(self, size, allow_start=False):
|
|
super(QubesResizableVmWithResize2fs, self).\
|
|
resize_root_img(size, allow_start=allow_start)
|
|
if not allow_start:
|
|
raise QubesException("VM start required to complete the "
|
|
"operation, but not allowed. Either run the "
|
|
"operation again allowing VM start this "
|
|
"time, or run resize2fs in the VM manually.")
|
|
self.start(start_guid=False)
|
|
self.run("resize2fs /dev/mapper/dmroot", user="root", wait=True,
|
|
gui=False)
|
|
self.shutdown()
|
|
while self.is_running():
|
|
sleep(1)
|
|
|
|
|
|
register_qubes_vm_class(QubesResizableVm)
|
|
register_qubes_vm_class(QubesResizableVmWithResize2fs)
|