private static void CheckInstallPathPermissions(string path)
{
if (!PathTools.DirectoryHasWritePermission(path) ||
!PathTools.DirectoryHasWritePermission(Path.Combine(path, "mods")) ||
!PathTools.DirectoryHasWritePermission(Path.Combine(path, "userdata")))
{
if (MessageBox.Show("KK Manager doesn't have write permissions to the game directory! This can cause issues for both KK Manager and the game itself.\n\nDo you want KK Manager to fix permissions of the entire Koikatu folder?",
"No write access to game directory", MessageBoxButtons.YesNo, MessageBoxIcon.Warning) == DialogResult.Yes)
{
ProcessTools.FixPermissions(path)?.WaitForExit();
}
}
}
public static async Task <FileInfo> GetTempDownloadFilename()
{
var tempPath = Path.Combine(InstallDirectoryHelper.GameDirectory.FullName, "temp\\KKManager_downloads");
retryCreate:
try
{
Directory.CreateDirectory(tempPath);
retry:
var fileName = Path.Combine(tempPath, Path.GetRandomFileName());
if (File.Exists(fileName))
{
goto retry;
}
return(new FileInfo(fileName));
}
catch (IOException ex)
{
if (await ProcessWaiter.CheckForProcessesBlockingKoiDir() != true)
{
throw new IOException($"Failed to create file in directory {tempPath} because of an IO issue - {ex.Message}", ex);
}
goto retryCreate;
}
catch (SecurityException ex)
{
if (MessageBox.Show($"Failed to create file in directory {tempPath} because of a security issue - {ex.Message}\n\nDo you want KK Manager to attempt to fix the issue? Click cancel if you want to abort.",
"Could not apply update", MessageBoxButtons.OKCancel, MessageBoxIcon.Error) == DialogResult.OK)
{
var fixPermissions = ProcessTools.FixPermissions(InstallDirectoryHelper.GameDirectory.FullName);
if (fixPermissions == null)
{
throw new IOException($"Failed to create file in directory {tempPath} because of a security issue - {ex.Message}", ex);
}
fixPermissions.WaitForExit();
goto retryCreate;
}
throw;
}
}
private void fixFileAndFolderPermissionsToolStripMenuItem_Click(object sender, EventArgs e)
{
ProcessTools.FixPermissions(InstallDirectoryHelper.KoikatuDirectory.FullName)?.WaitForExit();
}
/// <exception cref="IOException">Failed to apply the update.</exception>
public async Task Update(Progress <double> progressCallback, CancellationToken cancellationToken)
{
var downloadTarget = await GetTempDownloadFilename();
// Need to store the filename because MoveTo changes it to the new filename
var downloadFilename = downloadTarget.FullName;
if (RemoteFile != null)
{
Console.WriteLine($"Attempting download of {TargetPath.Name} from source {RemoteFile.Source.Origin}");
await RetryHelper.RetryOnExceptionAsync(async() => await RemoteFile.Download(downloadTarget, progressCallback, cancellationToken), 2, TimeSpan.FromSeconds(10), cancellationToken);
downloadTarget.Refresh();
if (!downloadTarget.Exists || downloadTarget.Length != RemoteFile.ItemSize)
{
throw new IOException($"Failed to download the update file {RemoteFile.Name} - the downloaded file doesn't exist or is corrupted");
}
Console.WriteLine($"Downloaded {downloadTarget.Length} bytes successfully");
}
retryDelete:
try
{
Directory.CreateDirectory(Path.GetDirectoryName(TargetPath.FullName));
try
{
if (TargetPath.Exists)
{
Console.WriteLine($"Deleting old file {TargetPath.FullName}");
// Prevent issues removing readonly files
TargetPath.Attributes = FileAttributes.Normal;
TargetPath.Delete();
// Make sure the file gets deleted before continuing
await Task.Delay(200, cancellationToken);
}
if (RemoteFile != null)
{
downloadTarget.MoveTo(TargetPath.FullName);
}
}
catch (IOException)
{
if (RemoteFile != null)
{
await Task.Delay(1000, cancellationToken);
downloadTarget.Replace(TargetPath.FullName, TargetPath.FullName + ".old", true);
await Task.Delay(1000, cancellationToken);
File.Delete(TargetPath.FullName + ".old");
}
else
{
throw;
}
}
}
catch (IOException ex)
{
if (await ProcessWaiter.CheckForProcessesBlockingKoiDir() != true)
{
throw RetryHelper.DoNotAttemptToRetry(new IOException($"Failed to apply update {TargetPath.FullName} because of an IO issue - {ex.Message}", ex));
}
goto retryDelete;
}
catch (SecurityException ex)
{
if (MessageBox.Show($"Failed to apply update {TargetPath.FullName} because of a security issue - {ex.Message}\n\nDo you want KK Manager to attempt to fix the issue? Click cancel if you want to abort.", "Could not apply update", MessageBoxButtons.OKCancel, MessageBoxIcon.Error) != DialogResult.OK)
{
throw;
}
var fixPermissions = ProcessTools.FixPermissions(InstallDirectoryHelper.GameDirectory.FullName);
if (fixPermissions == null)
{
throw RetryHelper.DoNotAttemptToRetry(new IOException($"Failed to create file in directory {TargetPath.FullName} because of a security issue - {ex.Message}", ex));
}
fixPermissions.WaitForExit();
goto retryDelete;
}
finally
{
try { File.Delete(downloadFilename); }
catch (SystemException ex) { Console.WriteLine(ex); }
}
}