mirror of
https://github.com/Tyrrrz/DiscordChatExporter.git
synced 2026-03-16 03:42:30 +00:00
Reorganize files
This commit is contained in:
@@ -0,0 +1,100 @@
|
||||
using System;
|
||||
using System.Security.Cryptography;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
using System.Text.Json.Serialization;
|
||||
using DiscordChatExporter.Gui.Utils.Extensions;
|
||||
|
||||
namespace DiscordChatExporter.Gui.Services;
|
||||
|
||||
public partial class SettingsService
|
||||
{
|
||||
private class TokenEncryptionConverter : JsonConverter<string?>
|
||||
{
|
||||
private const string Prefix = "enc:";
|
||||
|
||||
private static readonly Lazy<byte[]> Key = new(() =>
|
||||
Rfc2898DeriveBytes.Pbkdf2(
|
||||
Encoding.UTF8.GetBytes(Environment.TryGetMachineId() ?? string.Empty),
|
||||
Encoding.UTF8.GetBytes(ThisAssembly.Project.EncryptionSalt),
|
||||
iterations: 10_000,
|
||||
HashAlgorithmName.SHA256,
|
||||
outputLength: 16
|
||||
)
|
||||
);
|
||||
|
||||
public override string? Read(
|
||||
ref Utf8JsonReader reader,
|
||||
Type typeToConvert,
|
||||
JsonSerializerOptions options
|
||||
)
|
||||
{
|
||||
var value = reader.GetString();
|
||||
|
||||
// No prefix means the token is stored as plain text, which was
|
||||
// the case for older versions of the application.
|
||||
// Load it as is and encrypt it on next save.
|
||||
if (
|
||||
string.IsNullOrWhiteSpace(value)
|
||||
|| !value.StartsWith(Prefix, StringComparison.Ordinal)
|
||||
)
|
||||
return value;
|
||||
|
||||
try
|
||||
{
|
||||
var data = Convert.FromHexString(value[Prefix.Length..]);
|
||||
|
||||
// Layout: nonce (12 bytes) | paddingLength (1 byte) | tag (16 bytes) | cipher
|
||||
var nonce = data.AsSpan(0, 12);
|
||||
var paddingLength = data[12];
|
||||
var tag = data.AsSpan(13, 16);
|
||||
var cipher = data.AsSpan(29);
|
||||
|
||||
var decrypted = new byte[cipher.Length];
|
||||
using var aes = new AesGcm(Key.Value, 16);
|
||||
aes.Decrypt(nonce, cipher, tag, decrypted);
|
||||
|
||||
return Encoding.UTF8.GetString(decrypted.AsSpan(paddingLength));
|
||||
}
|
||||
catch (Exception ex)
|
||||
when (ex
|
||||
is FormatException
|
||||
or CryptographicException
|
||||
or ArgumentException
|
||||
or IndexOutOfRangeException
|
||||
)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
public override void Write(
|
||||
Utf8JsonWriter writer,
|
||||
string? value,
|
||||
JsonSerializerOptions options
|
||||
)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(value))
|
||||
{
|
||||
writer.WriteNullValue();
|
||||
return;
|
||||
}
|
||||
|
||||
var paddingLength = RandomNumberGenerator.GetInt32(1, 17);
|
||||
var tokenData = Encoding.UTF8.GetBytes(value);
|
||||
|
||||
// Layout: nonce (12 bytes) | paddingLength (1 byte) | tag (16 bytes) | cipher (paddingLength + tokenData.Length)
|
||||
var data = new byte[29 + paddingLength + tokenData.Length];
|
||||
RandomNumberGenerator.Fill(data.AsSpan(0, 12)); // nonce
|
||||
data[12] = (byte)paddingLength;
|
||||
var cipherSource = data.AsSpan(29);
|
||||
RandomNumberGenerator.Fill(cipherSource[..paddingLength]); // random padding
|
||||
tokenData.CopyTo(cipherSource[paddingLength..]); // token
|
||||
|
||||
using var aes = new AesGcm(Key.Value, 16);
|
||||
aes.Encrypt(data.AsSpan(0, 12), cipherSource, cipherSource, data.AsSpan(13, 16));
|
||||
|
||||
writer.WriteStringValue(Prefix + Convert.ToHexStringLower(data));
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,90 +0,0 @@
|
||||
using System;
|
||||
using System.Security.Cryptography;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
using System.Text.Json.Serialization;
|
||||
using DiscordChatExporter.Gui.Utils.Extensions;
|
||||
|
||||
namespace DiscordChatExporter.Gui.Services;
|
||||
|
||||
internal class TokenEncryptionConverter : JsonConverter<string?>
|
||||
{
|
||||
private const string Prefix = "enc:";
|
||||
|
||||
private static readonly Lazy<byte[]> Key = new(() =>
|
||||
Rfc2898DeriveBytes.Pbkdf2(
|
||||
Encoding.UTF8.GetBytes(Environment.TryGetMachineId() ?? string.Empty),
|
||||
Encoding.UTF8.GetBytes(ThisAssembly.Project.EncryptionSalt),
|
||||
iterations: 10_000,
|
||||
HashAlgorithmName.SHA256,
|
||||
outputLength: 16
|
||||
)
|
||||
);
|
||||
|
||||
public override string? Read(
|
||||
ref Utf8JsonReader reader,
|
||||
Type typeToConvert,
|
||||
JsonSerializerOptions options
|
||||
)
|
||||
{
|
||||
var value = reader.GetString();
|
||||
|
||||
// No prefix means the token is stored as plain text, which was
|
||||
// the case for older versions of the application.
|
||||
// Load it as is and encrypt it on next save.
|
||||
if (string.IsNullOrWhiteSpace(value) || !value.StartsWith(Prefix, StringComparison.Ordinal))
|
||||
return value;
|
||||
|
||||
try
|
||||
{
|
||||
var data = Convert.FromHexString(value[Prefix.Length..]);
|
||||
|
||||
// Layout: nonce (12 bytes) | paddingLength (1 byte) | tag (16 bytes) | cipher
|
||||
var nonce = data.AsSpan(0, 12);
|
||||
var paddingLength = data[12];
|
||||
var tag = data.AsSpan(13, 16);
|
||||
var cipher = data.AsSpan(29);
|
||||
|
||||
var decrypted = new byte[cipher.Length];
|
||||
using var aes = new AesGcm(Key.Value, 16);
|
||||
aes.Decrypt(nonce, cipher, tag, decrypted);
|
||||
|
||||
return Encoding.UTF8.GetString(decrypted.AsSpan(paddingLength));
|
||||
}
|
||||
catch (Exception ex)
|
||||
when (ex
|
||||
is FormatException
|
||||
or CryptographicException
|
||||
or ArgumentException
|
||||
or IndexOutOfRangeException
|
||||
)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
public override void Write(Utf8JsonWriter writer, string? value, JsonSerializerOptions options)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(value))
|
||||
{
|
||||
writer.WriteNullValue();
|
||||
return;
|
||||
}
|
||||
|
||||
var paddingLength = RandomNumberGenerator.GetInt32(1, 17);
|
||||
var tokenData = Encoding.UTF8.GetBytes(value);
|
||||
|
||||
// Layout: nonce (12 bytes) | paddingLength (1 byte) | tag (16 bytes) | cipher (paddingLength + tokenData.Length)
|
||||
var data = new byte[29 + paddingLength + tokenData.Length];
|
||||
RandomNumberGenerator.Fill(data.AsSpan(0, 12)); // nonce
|
||||
data[12] = (byte)paddingLength;
|
||||
var cipherSource = data.AsSpan(29);
|
||||
RandomNumberGenerator.Fill(cipherSource[..paddingLength]); // random padding
|
||||
tokenData.CopyTo(cipherSource[paddingLength..]); // token
|
||||
|
||||
using var aes = new AesGcm(Key.Value, 16);
|
||||
aes.Encrypt(data.AsSpan(0, 12), cipherSource, cipherSource, data.AsSpan(13, 16));
|
||||
|
||||
writer.WriteStringValue(Prefix + Convert.ToHexStringLower(data));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user