proton: Better handle broken symlinks

This commit is contained in:
Andrew Eikum 2022-02-09 11:11:11 -06:00 committed by Arkadiusz Hiler
parent 54c2347e15
commit 265c5ffaed

19
proton
View File

@ -79,6 +79,9 @@ def file_is_wine_builtin_dll(path):
def makedirs(path): def makedirs(path):
try: try:
#replace broken symlinks with a new directory
if os.path.islink(path) and not file_exists(path, follow_symlinks=True):
os.remove(path)
os.makedirs(path) os.makedirs(path)
except OSError: except OSError:
#already exists #already exists
@ -362,7 +365,7 @@ class CompatData:
dirs = [] dirs = []
for f in tracked_files: for f in tracked_files:
path = self.prefix_dir + f.strip() path = self.prefix_dir + f.strip()
if file_exists(path, follow_symlinks=True): if file_exists(path, follow_symlinks=False):
if os.path.isfile(path) or os.path.islink(path): if os.path.isfile(path) or os.path.islink(path):
os.remove(path) os.remove(path)
else: else:
@ -527,8 +530,8 @@ class CompatData:
if len(rel_dir) > 0: if len(rel_dir) > 0:
rel_dir = rel_dir + "/" rel_dir = rel_dir + "/"
dst_dir = src_dir.replace(g_proton.default_pfx_dir, self.prefix_dir, 1) dst_dir = src_dir.replace(g_proton.default_pfx_dir, self.prefix_dir, 1)
if not file_exists(dst_dir, follow_symlinks=False): if not file_exists(dst_dir, follow_symlinks=True):
os.makedirs(dst_dir) makedirs(dst_dir)
tracked_files.write(rel_dir + "\n") tracked_files.write(rel_dir + "\n")
for dir_ in dirs: for dir_ in dirs:
src_file = os.path.join(src_dir, dir_) src_file = os.path.join(src_dir, dir_)
@ -554,8 +557,8 @@ class CompatData:
if len(rel_dir) > 0: if len(rel_dir) > 0:
rel_dir = rel_dir + "/" rel_dir = rel_dir + "/"
dst_dir = src_dir.replace(g_proton.default_pfx_dir, self.prefix_dir, 1) dst_dir = src_dir.replace(g_proton.default_pfx_dir, self.prefix_dir, 1)
if not file_exists(dst_dir, follow_symlinks=False): if not file_exists(dst_dir, follow_symlinks=True):
os.makedirs(dst_dir) makedirs(dst_dir)
tracked_files.write(rel_dir + "\n") tracked_files.write(rel_dir + "\n")
for file_ in files: for file_ in files:
src_file = os.path.join(src_dir, file_) src_file = os.path.join(src_dir, file_)
@ -820,9 +823,9 @@ class CompatData:
else: else:
nvapi64_dll = self.prefix_dir + "drive_c/windows/system32/nvapi64.dll" nvapi64_dll = self.prefix_dir + "drive_c/windows/system32/nvapi64.dll"
nvapi32_dll = self.prefix_dir + "drive_c/windows/syswow64/nvapi.dll" nvapi32_dll = self.prefix_dir + "drive_c/windows/syswow64/nvapi.dll"
if file_exists(nvapi64_dll, follow_symlinks=True): if file_exists(nvapi64_dll, follow_symlinks=False):
os.unlink(nvapi64_dll) os.unlink(nvapi64_dll)
if file_exists(nvapi32_dll, follow_symlinks=True): if file_exists(nvapi32_dll, follow_symlinks=False):
os.unlink(nvapi32_dll) os.unlink(nvapi32_dll)
# Try to detect known DLLs that ship with the NVIDIA Linux Driver # Try to detect known DLLs that ship with the NVIDIA Linux Driver
@ -962,7 +965,7 @@ class Session:
lfile_path = basedir + "/steam-" + os.environ["SteamGameId"] + ".log" lfile_path = basedir + "/steam-" + os.environ["SteamGameId"] + ".log"
if file_exists(lfile_path, follow_symlinks=True): if file_exists(lfile_path, follow_symlinks=False):
os.remove(lfile_path) os.remove(lfile_path)
makedirs(basedir) makedirs(basedir)