diff options
| -rw-r--r-- | src/virtual/ogOperations.py | 41 |
1 files changed, 26 insertions, 15 deletions
diff --git a/src/virtual/ogOperations.py b/src/virtual/ogOperations.py index 0a49b2f..5096614 100644 --- a/src/virtual/ogOperations.py +++ b/src/virtual/ogOperations.py @@ -18,6 +18,7 @@ import hivex import pathlib import re import math +import sys class OgQMP: class State(enum.Enum): @@ -61,6 +62,14 @@ class OgVirtualOperations: self.IP = '127.0.0.1' self.VIRTUAL_PORT = 4444 self.USABLE_DISK = 0.75 + self.OG_PATH = os.path.dirname(os.path.realpath(sys.argv[0])) + self.OG_IMAGES_PATH = f'{self.OG_PATH}/images' + self.OG_PARTITIONS_PATH = f'{self.OG_PATH}/partitions' + + if not os.path.exists(self.OG_IMAGES_PATH): + os.mkdir(self.OG_IMAGES_PATH, mode=0o755) + if not os.path.exists(self.OG_PARTITIONS_PATH): + os.mkdir(self.OG_PARTITIONS_PATH, mode=0o755) def poweroff(self): qmp = OgQMP(self.IP, self.VIRTUAL_PORT) @@ -93,14 +102,13 @@ class OgVirtualOperations: # Calculate the lower power of 2 amout of RAM memory for the VM. vm_ram_mib = 2 ** math.floor(math.log(available_ram_mib) / math.log(2)) - cmd = (f'qemu-system-x86_64 -hda disk{disk}_part{partition}.qcow2 ' + cmd = (f'qemu-system-x86_64 -hda {self.OG_PARTITIONS_PATH}/disk{disk}_part{partition}.qcow2 ' f'-qmp tcp:localhost:4444,server,nowait --enable-kvm ' - f'-display gtk -cpu host -m {vm_ram_mib}M -boot c') + f'-display gtk -cpu host -m {vm_ram_mib}M -boot c -full-screen') subprocess.Popen([cmd], shell=True) def refresh(self, ogRest): - path = 'partitions.json' - temp_mount_dir = 'mnt' + path = f'{self.OG_PATH}/partitions.json' try: with open(path, 'r+') as f: data = json.loads(f.read()) @@ -158,8 +166,8 @@ class OgVirtualOperations: return data def setup(self, request, ogRest): - path = 'partitions.json' - refresh(ogRest) + path = f'{self.OG_PATH}/partitions.json' + self.refresh(ogRest) part_setup = request.getPartitionSetup() disk = request.getDisk() @@ -195,7 +203,7 @@ class OgVirtualOperations: f.write(json.dumps(data, indent=4)) f.truncate() - return refresh(ogRest) + return self.refresh(ogRest) def image_create(self, path, request, ogRest): disk = request.getDisk() @@ -209,16 +217,20 @@ class OgVirtualOperations: qmp.disconnect() return None - refresh(ogRest) + self.refresh(ogRest) + + drive_path = f'{self.OG_PARTITIONS_PATH}/disk{disk}_part{partition}.qcow2' drive_path = f'disk{disk}_part{partition}.qcow2' repo_path = f'images' try: - shutil.copy(drive_path, f'{repo_path}/{name}') + shutil.copy(drive_path, f'{self.OG_IMAGES_PATH}/{name}') except: return None + subprocess.run([f'umount {self.OG_IMAGES_PATH}'], shell=True) + return True def image_restore(self, request, ogRest): @@ -237,16 +249,15 @@ class OgVirtualOperations: qmp.disconnect() return None - refresh(ogRest) + self.refresh(ogRest) - drive_path = f'disk{disk}_part{partition}.qcow2' - repo_path = f'images' + drive_path = f'{self.OG_PARTITIONS_PATH}/disk{disk}_part{partition}.qcow2' if os.path.exists(drive_path): os.remove(drive_path) try: - shutil.copy(f'{repo_path}/{name}', drive_path) + shutil.copy(f'{self.OG_IMAGES_PATH}/{name}', drive_path) except: return None @@ -257,7 +268,7 @@ class OgVirtualOperations: disk = request.getDisk() partition = request.getPartition() - drive_path = f'disk{disk}_part{partition}.qcow2' + drive_path = f'{self.OG_PARTITIONS_PATH}/disk{disk}_part{partition}.qcow2' g = guestfs.GuestFS(python_return_dict=True) g.add_drive_opts(drive_path, readonly=1) g.launch() @@ -356,7 +367,7 @@ class OgVirtualOperations: qmp.send(str({"execute": "query-pci"})) data = json.loads(qmp.recv()) data = data['return'][0]['devices'] - pci_list = parse_pci() + pci_list = self.parse_pci() device_names = {} for device in data: vendor_id = hex(device['id']['vendor'])[2:] |