Austin/BililiveRecorder
1
0
Fork 0
mirror of https://github.com/BililiveRecorder/BililiveRecorder.git synced 2024-11-16 03:32:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

use safe WhiteAllText

Browse Source
This commit is contained in:
Genteure 2021-01-03 11:38:20 +08:00
parent 131264af9c
commit 65fd63e858
1 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
BililiveRecorder.Core/Config/ConfigParser.cs
View File

@ -1,5 +1,6 @@
using System;
using System.IO;
using System.Linq;
using System.Text;
using Newtonsoft.Json;
using NLog;
@ -84,7 +85,7 @@ namespace BililiveRecorder.Core.Config
var filepath = Path.Combine(directory, CONFIG_FILE_NAME);
if (json is not null)
File.WriteAllText(filepath, json, Encoding.UTF8);
WriteAllTextWithBackup(filepath, json);
return true;
}
@ -108,5 +109,31 @@ namespace BililiveRecorder.Core.Config
return null;
}
}
// https://stackoverflow.com/q/25366534 with modification
private static void WriteAllTextWithBackup(string path, string contents)
{
var ext = Path.GetExtension(path);
var tempPath = Path.Combine(Path.GetDirectoryName(path), Path.ChangeExtension(path, RandomString(6) + ext));
var backupPath = Path.ChangeExtension(path, "backup" + ext);
// delete any existing backups
if (File.Exists(backupPath))
File.Delete(backupPath);
// get the bytes
var data = Encoding.UTF8.GetBytes(contents);
// write the data to a temp file
using (var tempFile = File.Create(tempPath, 4096, FileOptions.WriteThrough))
tempFile.Write(data, 0, data.Length);
// replace the contents
File.Replace(tempPath, path, backupPath);
}
private static readonly Random random = new Random();
private static string RandomString(int length) => new string(Enumerable.Repeat("ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789", length).Select(s => s[random.Next(s.Length)]).ToArray());
}
}