mirror of https://github.com/kholia/OSX-KVM.git
added boot-usb-passthrough
this script detect when usb is connected and starts macos performing usb passthrough note: start the script as sudo
This commit is contained in:
0Franky
GitHub
parent
422bb3b713
commit
7a4033e72d
|
|
@ -0,0 +1,157 @@
|
|||
import subprocess
|
||||
import os
|
||||
import signal
|
||||
import time
|
||||
import re
|
||||
|
||||
verbose_log = False
|
||||
|
||||
def initialize():
|
||||
# Check if the /tmp/udevadm_output file exists and delete it if necessary
|
||||
if os.path.exists("/tmp/udevadm_output"):
|
||||
os.remove("/tmp/udevadm_output")
|
||||
|
||||
# Reload udev rules
|
||||
# subprocess.run(["udevadm", "control", "--reload-rules"])
|
||||
# subprocess.run(["udevadm", "trigger"])
|
||||
|
||||
def scan_udevadm():
|
||||
return subprocess.Popen(["udevadm", "monitor", "--environment", "--subsystem-match=usb", "--property"],
|
||||
stdout=open("/tmp/udevadm_output", "w"))
|
||||
|
||||
def show_added_usb_devices(udevadm_pid):
|
||||
global verbose_log
|
||||
|
||||
input("Now, insert the USB device. Press Enter when done.")
|
||||
time.sleep(3)
|
||||
os.kill(udevadm_pid, signal.SIGINT)
|
||||
|
||||
# Show results in the desired format
|
||||
if verbose_log:
|
||||
print("Results of udevadm scan:")
|
||||
with open("/tmp/udevadm_output", "r") as file:
|
||||
print(file.read())
|
||||
|
||||
def extract_information():
|
||||
global verbose_log
|
||||
|
||||
add_action = "ACTION=add"
|
||||
usb_devices = set()
|
||||
additional_settings = ""
|
||||
|
||||
# Extract information blocks about the new USB device
|
||||
with open("/tmp/udevadm_output", "r") as file:
|
||||
info_blocks = file.read()
|
||||
|
||||
if info_blocks:
|
||||
info_blocks = info_blocks.split("\n\n")
|
||||
info_blocks = [block for block in info_blocks if add_action in block and 'UDEV' in block]
|
||||
|
||||
# Print complete blocks if necessary
|
||||
if verbose_log:
|
||||
print("Complete blocks of information about the inserted USB device:")
|
||||
print(info_blocks)
|
||||
|
||||
print("\nInformation about the inserted USB devices:")
|
||||
for i, new_usb in enumerate(info_blocks):
|
||||
if 'DEVNAME' in new_usb:
|
||||
devname = [item.split('=')[1] for item in new_usb.split() if 'DEVNAME' in item][0]
|
||||
|
||||
# Extract additional information
|
||||
hostbus = [item.split('=')[1] for item in new_usb.split() if 'BUSNUM' in item][0]
|
||||
hostaddr = [item.split('=')[1] for item in new_usb.split() if 'DEVNUM' in item][0]
|
||||
id_model = [item.split('=')[1] for item in new_usb.split() if 'ID_MODEL' in item]
|
||||
id_vendor = [item.split('=')[1] for item in new_usb.split() if 'ID_VENDOR' in item]
|
||||
id_fs_size = [item.split('=')[1] for item in new_usb.split() if 'ID_FS_SIZE' in item]
|
||||
|
||||
# Print general information about the USB device
|
||||
print(f"==== Device #{i} ====")
|
||||
print(f"USB Host {hostbus}:{hostaddr}")
|
||||
if id_model:
|
||||
print(f"Model: {id_model[0]}")
|
||||
if id_vendor:
|
||||
print(f"Manufacturer: {id_vendor[0]}")
|
||||
if id_fs_size:
|
||||
print(f"Size: {id_fs_size[0]}")
|
||||
|
||||
print("\n")
|
||||
|
||||
# Add the line to be added to the boot-macOS.sh script
|
||||
usb_devices.add(f"-device usb-host,hostdevice={devname}")
|
||||
|
||||
additional_settings = "-device qemu-xhci,id=xhci-usb\n" + "\n".join(usb_devices)
|
||||
|
||||
print("additionalSettings to pass to qemu:")
|
||||
print(additional_settings)
|
||||
else:
|
||||
print("No information available about the new USB device.")
|
||||
|
||||
return additional_settings
|
||||
|
||||
def check_open_core_boot_file(additional_settings):
|
||||
global verbose_log
|
||||
|
||||
file_path = "OpenCore-Boot.sh"
|
||||
# file_path = "boot-passthrough.sh"
|
||||
|
||||
if os.path.exists(file_path):
|
||||
choice = input(f"\n\nThe file {file_path} exists. Start it with USB Passthrough? (Y/n)")
|
||||
if choice == "" or choice.lower().startswith("y"):
|
||||
with open(file_path, "r") as file:
|
||||
start_script = file.read()
|
||||
|
||||
if start_script:
|
||||
# Insert additional_settings before the closing parenthesis of args
|
||||
start_script = re.sub(r'(\n\))', rf'\n {additional_settings}\1', start_script, flags=re.DOTALL)
|
||||
|
||||
if verbose_log:
|
||||
print(start_script)
|
||||
|
||||
print("Starting macOS...\n\n\n")
|
||||
execute_bash_script(start_script)
|
||||
|
||||
else:
|
||||
print(f"The file {file_path} does not exist.")
|
||||
|
||||
def execute_bash_script(script):
|
||||
# Create a process to execute the bash script
|
||||
process = subprocess.Popen(['bash'], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
||||
|
||||
# Write the script to the process's stdin
|
||||
stdout, stderr = process.communicate(input=script)
|
||||
|
||||
# Print the output and any potential errors
|
||||
print("==== QEMU Output ====")
|
||||
print(stdout)
|
||||
|
||||
if stderr:
|
||||
print("\n\n\n==== QEMU ERROR ====")
|
||||
print(stderr)
|
||||
|
||||
def clean_up():
|
||||
print("\n\nRemoving temporary files...")
|
||||
os.remove("/tmp/udevadm_output")
|
||||
|
||||
def main():
|
||||
print("Preparation phase...")
|
||||
|
||||
# Initialization
|
||||
initialize()
|
||||
|
||||
# Perform the initial udevadm scan
|
||||
udevadm_pid = scan_udevadm().pid
|
||||
|
||||
# Display a message and wait for user input
|
||||
show_added_usb_devices(udevadm_pid)
|
||||
|
||||
# Extract information and print the line for the boot-macOS.sh script
|
||||
additional_settings = extract_information()
|
||||
|
||||
# Execute exit procedures
|
||||
clean_up()
|
||||
|
||||
check_open_core_boot_file(additional_settings)
|
||||
|
||||
print("\n\nExiting... Goodbye!")
|
||||
|
||||
main()
|
||||
Loading…
Reference in New Issue