2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Show partition info in 4K alignment check

Browse source
This commit is contained in:
2Shirt 2023-03-25 20:24:17 -07:00
parent 4f6a07c449
commit d5bc74d21b
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 35 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
scripts/wk/os/win.py
View file

@ -6,6 +6,7 @@ import logging
import os import os
import pathlib import pathlib
import platform import platform
import re
from contextlib import suppress from contextlib import suppress
import psutil import psutil
@ -71,6 +72,9 @@ KNOWN_HIVE_NAMES = {
RAM_OK = 5.5 * 1024**3 # ~6 GiB assuming a bit of shared memory RAM_OK = 5.5 * 1024**3 # ~6 GiB assuming a bit of shared memory
RAM_WARNING = 3.5 * 1024**3 # ~4 GiB assuming a bit of shared memory RAM_WARNING = 3.5 * 1024**3 # ~4 GiB assuming a bit of shared memory
REG_MSISERVER = r'HKLM\SYSTEM\CurrentControlSet\Control\SafeBoot\Network\MSIServer' REG_MSISERVER = r'HKLM\SYSTEM\CurrentControlSet\Control\SafeBoot\Network\MSIServer'
REGEX_4K_ALIGNMENT = re.compile(
r'^(?P<description>.*?)\s+(?P<size>\d+)\s+(?P<offset>\d+)',
)
SLMGR = pathlib.Path(f'{os.environ.get("SYSTEMROOT")}/System32/slmgr.vbs') SLMGR = pathlib.Path(f'{os.environ.get("SYSTEMROOT")}/System32/slmgr.vbs')
SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE') SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE')
@ -163,19 +167,41 @@ def set_timezone(zone):
# Info Functions # Info Functions
def check_4k_alignment(show_alert=False): def check_4k_alignment(show_alert=False):
"""Check if all partitions are 4K aligned, returns book.""" """Check if all partitions are 4K aligned, returns book."""
cmd = ['WMIC', 'partition', 'get', 'StartingOffset'] cmd = ['WMIC', 'partition', 'get', 'Caption,Size,StartingOffset']
report = []
show_alert = False
# Check offsets # Check offsets
proc = run_program(cmd) proc = run_program(cmd)
for offset in proc.stdout.splitlines(): for line in proc.stdout.splitlines():
offset = offset.strip() line = line.strip()
if not offset.isnumeric(): if not line or not line.startswith('Disk'):
continue continue
if int(offset) % 4096 != 0: match = REGEX_4K_ALIGNMENT.match(line)
# Not aligned if not match:
if show_alert: LOG.error('Failed to parse partition info for: %s', line)
show_alert_box('One or more partitions are not 4K aligned') continue
raise GenericError('One or more partitions are not 4K aligned') if int(match.group('offset')) % 4096 == 0:
report.append(
color_string(
f'{match.group("description")}'
f' ({bytes_to_string(match.group("size"), decimals=1)})'
,
'RED'
)
)
# Show alert
if show_alert:
show_alert_box('One or more partitions not 4K aligned')
# Done
if report:
report.insert(
0,
color_string('One or more partitions not 4K aligned', 'YELLOW'),
)
return report
def export_bitlocker_info(): def export_bitlocker_info():