Initial version
This commit is contained in:
commit
e5b6226a85
68 changed files with 7338 additions and 0 deletions
21
DataTransfer.Client/Services/BotCheckSolver.cs
Normal file
21
DataTransfer.Client/Services/BotCheckSolver.cs
Normal file
|
|
@ -0,0 +1,21 @@
|
|||
using DataTransfer.Client.Shared;
|
||||
using Microsoft.AspNetCore.Components;
|
||||
using Microsoft.JSInterop;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
/// <summary>Solves the server's proof-of-work bot check in web workers (<c>wwwroot/js/bot-check.js</c> of the server).</summary>
|
||||
public sealed class BotCheckSolver(IJSRuntime js, NavigationManager navigation)
|
||||
{
|
||||
/// <returns>The solution, or <c>-1</c> if none was found.</returns>
|
||||
public async Task<int> SolveAsync(BotCheckChallenge challenge)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(challenge);
|
||||
|
||||
// An absolute URL, so the module resolves relative to the app's base URI under any path base.
|
||||
await using var module = await js.InvokeAsync<IJSObjectReference>(
|
||||
"import",
|
||||
navigation.ToAbsoluteUri("js/bot-check.js").ToString());
|
||||
return await module.InvokeAsync<int>("solve", challenge.Salt, challenge.Hash, challenge.MaxNumber);
|
||||
}
|
||||
}
|
||||
36
DataTransfer.Client/Services/ChatEntry.cs
Normal file
36
DataTransfer.Client/Services/ChatEntry.cs
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
namespace DataTransfer.Client.Services;
|
||||
|
||||
public enum ChatEntryKind
|
||||
{
|
||||
Incoming,
|
||||
Outgoing,
|
||||
System
|
||||
}
|
||||
|
||||
public sealed class ChatEntry
|
||||
{
|
||||
public DateTimeOffset Time { get; } = DateTimeOffset.Now;
|
||||
|
||||
public bool IsFile => FileName is not null;
|
||||
|
||||
public bool IsTransferring => IsFile && Error is null && TransferredBytes < FileSize;
|
||||
|
||||
public int ProgressPercent => FileSize == 0 ? 100 : (int)(TransferredBytes * 100 / FileSize);
|
||||
public required ChatEntryKind Kind { get; init; }
|
||||
|
||||
public string? Text { get; init; }
|
||||
|
||||
public string? FileName { get; init; }
|
||||
|
||||
public long FileSize { get; init; }
|
||||
|
||||
public long TransferredBytes { get; set; }
|
||||
|
||||
/// <summary>Object URL of a completely received and decrypted file.</summary>
|
||||
public string? DownloadUrl { get; set; }
|
||||
|
||||
/// <summary>A received file whose type can run code or is often used for malware or phishing.</summary>
|
||||
public bool IsRiskyFile { get; init; }
|
||||
|
||||
public string? Error { get; set; }
|
||||
}
|
||||
32
DataTransfer.Client/Services/DisplayFormat.cs
Normal file
32
DataTransfer.Client/Services/DisplayFormat.cs
Normal file
|
|
@ -0,0 +1,32 @@
|
|||
using System.Globalization;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
public static class DisplayFormat
|
||||
{
|
||||
public static string Bytes(long bytes) => bytes switch
|
||||
{
|
||||
>= 1024L * 1024 * 1024 => string.Create(CultureInfo.InvariantCulture, $"{bytes / 1024d / 1024d / 1024d:0.#} GB"),
|
||||
>= 1024 * 1024 => string.Create(CultureInfo.InvariantCulture, $"{bytes / 1024d / 1024d:0.0} MB"),
|
||||
>= 1024 => string.Create(CultureInfo.InvariantCulture, $"{bytes / 1024d:0.0} KB"),
|
||||
_ => $"{bytes} B"
|
||||
};
|
||||
|
||||
public static string Duration(TimeSpan duration) =>
|
||||
duration switch
|
||||
{
|
||||
{ TotalHours: >= 1, Minutes: 0, Seconds: 0 } => string.Create(CultureInfo.InvariantCulture, $"{duration.TotalHours:0} h"),
|
||||
{ TotalMinutes: >= 1, Seconds: 0 } => string.Create(CultureInfo.InvariantCulture, $"{duration.TotalMinutes:0} min"),
|
||||
_ => string.Create(CultureInfo.InvariantCulture, $"{duration.TotalSeconds:0} s")
|
||||
};
|
||||
|
||||
public static string Countdown(TimeSpan remaining)
|
||||
{
|
||||
if (remaining < TimeSpan.Zero)
|
||||
{
|
||||
remaining = TimeSpan.Zero;
|
||||
}
|
||||
|
||||
return remaining.ToString(remaining.TotalHours >= 1 ? @"h\:mm\:ss" : @"mm\:ss", CultureInfo.InvariantCulture);
|
||||
}
|
||||
}
|
||||
54
DataTransfer.Client/Services/FileNameSanitizer.cs
Normal file
54
DataTransfer.Client/Services/FileNameSanitizer.cs
Normal file
|
|
@ -0,0 +1,54 @@
|
|||
using System.Globalization;
|
||||
using System.Text;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
public static class FileNameSanitizer
|
||||
{
|
||||
private const int MaxLength = 255;
|
||||
private const string Fallback = "file";
|
||||
|
||||
// Characters invalid in file names on Windows, macOS or Linux. Path.GetInvalidFileNameChars() returns only the
|
||||
// Unix set in WebAssembly, but the file is saved on any operating system.
|
||||
private const string InvalidCharacters = "<>:\"/\\|?*";
|
||||
|
||||
/// <summary>
|
||||
/// Makes a file name received from the peer safe to display and save. Invisible formatting characters are removed,
|
||||
/// so for example a right-to-left override can't disguise <c>invoice‮fdp.exe</c> as <c>invoiceexe.pdf</c>.
|
||||
/// </summary>
|
||||
public static string Sanitize(string? name)
|
||||
{
|
||||
if (string.IsNullOrEmpty(name))
|
||||
{
|
||||
return Fallback;
|
||||
}
|
||||
|
||||
var builder = new StringBuilder(Math.Min(name.Length, MaxLength));
|
||||
foreach (var rune in name.EnumerateRunes())
|
||||
{
|
||||
switch (Rune.GetUnicodeCategory(rune))
|
||||
{
|
||||
case UnicodeCategory.Format:
|
||||
continue;
|
||||
case UnicodeCategory.Control:
|
||||
Append(builder, new Rune('_'));
|
||||
continue;
|
||||
}
|
||||
|
||||
Append(builder, rune.IsAscii && InvalidCharacters.Contains((char)rune.Value) ? new Rune('_') : rune);
|
||||
}
|
||||
|
||||
// Windows silently drops trailing dots and spaces, which could change the extension.
|
||||
var cleaned = builder.ToString().Trim().TrimEnd('.');
|
||||
return cleaned.Length == 0 ? Fallback : cleaned;
|
||||
}
|
||||
|
||||
// Appends whole runes only, so truncation never splits a surrogate pair.
|
||||
private static void Append(StringBuilder builder, Rune rune)
|
||||
{
|
||||
if (builder.Length + rune.Utf16SequenceLength <= MaxLength)
|
||||
{
|
||||
builder.Append(rune.ToString());
|
||||
}
|
||||
}
|
||||
}
|
||||
101
DataTransfer.Client/Services/Frames.cs
Normal file
101
DataTransfer.Client/Services/Frames.cs
Normal file
|
|
@ -0,0 +1,101 @@
|
|||
using System.Text;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
/// <summary>Application frames. They are encrypted as a whole, so the relay server cannot see their type or content.</summary>
|
||||
public abstract record Frame;
|
||||
|
||||
public sealed record TextFrame(string Text) : Frame;
|
||||
|
||||
public sealed record FileStartFrame(int FileId, string Name, long Size) : Frame;
|
||||
|
||||
public sealed record FileChunkFrame(int FileId, ReadOnlyMemory<byte> Data) : Frame;
|
||||
|
||||
public sealed record FileCancelFrame(int FileId) : Frame;
|
||||
|
||||
/// <summary>Plaintext layout: version (byte) | type (byte) | sequence number (int64) | type-specific body.</summary>
|
||||
internal static class FrameCodec
|
||||
{
|
||||
private const byte Version = 1;
|
||||
|
||||
private enum FrameType : byte
|
||||
{
|
||||
Text = 1,
|
||||
FileStart = 2,
|
||||
FileChunk = 3,
|
||||
FileCancel = 4
|
||||
}
|
||||
|
||||
public static byte[] Encode(long sequence, Frame frame)
|
||||
{
|
||||
var capacity = frame is FileChunkFrame chunk ? chunk.Data.Length + 32 : 256;
|
||||
using var stream = new MemoryStream(capacity);
|
||||
using (var writer = new BinaryWriter(stream, Encoding.UTF8, true))
|
||||
{
|
||||
writer.Write(Version);
|
||||
switch (frame)
|
||||
{
|
||||
case TextFrame text:
|
||||
WriteHeader(writer, FrameType.Text, sequence);
|
||||
writer.Write(text.Text);
|
||||
break;
|
||||
case FileStartFrame start:
|
||||
WriteHeader(writer, FrameType.FileStart, sequence);
|
||||
writer.Write(start.FileId);
|
||||
writer.Write(start.Name);
|
||||
writer.Write(start.Size);
|
||||
break;
|
||||
case FileChunkFrame fileChunk:
|
||||
WriteHeader(writer, FrameType.FileChunk, sequence);
|
||||
writer.Write(fileChunk.FileId);
|
||||
writer.Write(fileChunk.Data.Span);
|
||||
break;
|
||||
case FileCancelFrame cancel:
|
||||
WriteHeader(writer, FrameType.FileCancel, sequence);
|
||||
writer.Write(cancel.FileId);
|
||||
break;
|
||||
default:
|
||||
throw new ArgumentOutOfRangeException(nameof(frame), frame, "Unknown frame.");
|
||||
}
|
||||
}
|
||||
|
||||
return stream.ToArray();
|
||||
}
|
||||
|
||||
/// <exception cref="InvalidDataException">The plaintext is not a valid frame.</exception>
|
||||
public static (long Sequence, Frame Frame) Decode(byte[] plaintext)
|
||||
{
|
||||
try
|
||||
{
|
||||
using var reader = new BinaryReader(new MemoryStream(plaintext, false), Encoding.UTF8);
|
||||
if (reader.ReadByte() != Version)
|
||||
{
|
||||
throw new InvalidDataException("Unsupported frame version.");
|
||||
}
|
||||
|
||||
var type = (FrameType)reader.ReadByte();
|
||||
var sequence = reader.ReadInt64();
|
||||
Frame frame = type switch
|
||||
{
|
||||
FrameType.Text => new TextFrame(reader.ReadString()),
|
||||
FrameType.FileStart => new FileStartFrame(reader.ReadInt32(), reader.ReadString(), reader.ReadInt64()),
|
||||
FrameType.FileChunk => new FileChunkFrame(reader.ReadInt32(), plaintext.AsMemory((int)reader.BaseStream.Position)),
|
||||
FrameType.FileCancel => new FileCancelFrame(reader.ReadInt32()),
|
||||
_ => throw new InvalidDataException("Unknown frame type.")
|
||||
};
|
||||
|
||||
return (sequence, frame);
|
||||
}
|
||||
catch (Exception ex) when (ex is IOException or FormatException)
|
||||
{
|
||||
// EndOfStreamException (truncated), IOException (negative string length), FormatException (bad 7-bit length).
|
||||
throw new InvalidDataException("Malformed frame.", ex);
|
||||
}
|
||||
}
|
||||
|
||||
private static void WriteHeader(BinaryWriter writer, FrameType type, long sequence)
|
||||
{
|
||||
writer.Write((byte)type);
|
||||
writer.Write(sequence);
|
||||
}
|
||||
}
|
||||
53
DataTransfer.Client/Services/HandshakeCodec.cs
Normal file
53
DataTransfer.Client/Services/HandshakeCodec.cs
Normal file
|
|
@ -0,0 +1,53 @@
|
|||
namespace DataTransfer.Client.Services;
|
||||
|
||||
/// <summary>
|
||||
/// Plaintext messages of the commit-then-reveal key exchange (see <see cref="SecureChannel" />). They are relayed
|
||||
/// like frames, but before the channel is encrypted. Layout: version (byte) | type (byte) | payload.
|
||||
/// </summary>
|
||||
internal static class HandshakeCodec
|
||||
{
|
||||
public const int CommitmentBytes = 32;
|
||||
|
||||
// A P-256 SPKI key has 91 bytes; anything much larger is not a key.
|
||||
private const int MaxPublicKeyBytes = 512;
|
||||
private const byte Version = 1;
|
||||
|
||||
private enum MessageType : byte
|
||||
{
|
||||
Commitment = 1,
|
||||
PublicKey = 2
|
||||
}
|
||||
|
||||
public static byte[] EncodeCommitment(byte[] commitment) => Encode(MessageType.Commitment, commitment);
|
||||
|
||||
public static byte[] EncodePublicKey(byte[] publicKey) => Encode(MessageType.PublicKey, publicKey);
|
||||
|
||||
/// <exception cref="InvalidDataException">The message is not a commitment.</exception>
|
||||
public static byte[] DecodeCommitment(byte[] message) =>
|
||||
Decode(message, MessageType.Commitment, CommitmentBytes, CommitmentBytes);
|
||||
|
||||
/// <exception cref="InvalidDataException">The message is not a public key.</exception>
|
||||
public static byte[] DecodePublicKey(byte[] message) => Decode(message, MessageType.PublicKey, 1, MaxPublicKeyBytes);
|
||||
|
||||
private static byte[] Encode(MessageType type, byte[] payload)
|
||||
{
|
||||
var message = new byte[payload.Length + 2];
|
||||
message[0] = Version;
|
||||
message[1] = (byte)type;
|
||||
payload.CopyTo(message, 2);
|
||||
return message;
|
||||
}
|
||||
|
||||
private static byte[] Decode(byte[] message, MessageType type, int minPayloadBytes, int maxPayloadBytes)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(message);
|
||||
|
||||
var payloadBytes = message.Length - 2;
|
||||
if (payloadBytes < minPayloadBytes || payloadBytes > maxPayloadBytes || message[0] != Version || message[1] != (byte)type)
|
||||
{
|
||||
throw new InvalidDataException($"Expected a valid key-exchange message of type {type}.");
|
||||
}
|
||||
|
||||
return message[2..];
|
||||
}
|
||||
}
|
||||
40
DataTransfer.Client/Services/RiskyFileTypes.cs
Normal file
40
DataTransfer.Client/Services/RiskyFileTypes.cs
Normal file
|
|
@ -0,0 +1,40 @@
|
|||
using System.Collections.Frozen;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
/// <summary>
|
||||
/// File types that can run code when opened, or are commonly used to deliver malware or phishing. Common archives
|
||||
/// (zip, 7z, rar) and documents are deliberately not listed, so the warning stays meaningful.
|
||||
/// </summary>
|
||||
public static class RiskyFileTypes
|
||||
{
|
||||
private static readonly FrozenSet<string> Extensions = new[]
|
||||
{
|
||||
// Windows programs and installers
|
||||
"exe", "com", "scr", "pif", "cpl", "msi", "msix", "msixbundle", "msp", "appx", "appxbundle", "appref-ms",
|
||||
"application", "gadget", "dll", "sys", "drv", "ocx",
|
||||
|
||||
// Scripts
|
||||
"bat", "cmd", "ps1", "psm1", "vbs", "vbe", "js", "jse", "mjs", "wsf", "wsh", "wsc", "sct", "hta", "jar", "py",
|
||||
"pyw", "sh", "bash", "command", "applescript", "scpt", "reg", "inf", "msc",
|
||||
|
||||
// Shortcuts and shell files that can launch programs
|
||||
"lnk", "url", "scf", "library-ms", "search-ms", "settingcontent-ms",
|
||||
|
||||
// macOS and Linux applications and packages
|
||||
"app", "pkg", "mpkg", "dmg", "workflow", "action", "deb", "rpm", "appimage", "run", "bin",
|
||||
|
||||
// Disk images, often used to bypass download protection
|
||||
"iso", "img", "vhd", "vhdx",
|
||||
|
||||
// Office documents with macros, add-ins and other formats abused for malware
|
||||
"docm", "dotm", "xlsm", "xltm", "xlam", "xll", "pptm", "potm", "ppam", "ppsm", "sldm", "one", "chm",
|
||||
|
||||
// Web pages opened from disk, a common way to deliver phishing
|
||||
"htm", "html", "xhtml", "mht", "mhtml"
|
||||
}.ToFrozenSet(StringComparer.OrdinalIgnoreCase);
|
||||
|
||||
/// <summary>Only the last extension counts, so <c>invoice.pdf.exe</c> is risky and <c>setup.exe.txt</c> is not.</summary>
|
||||
public static bool IsRisky(string? fileName) =>
|
||||
Path.GetExtension(fileName) is { Length: > 1 } extension && Extensions.Contains(extension[1..]);
|
||||
}
|
||||
264
DataTransfer.Client/Services/SecureChannel.cs
Normal file
264
DataTransfer.Client/Services/SecureChannel.cs
Normal file
|
|
@ -0,0 +1,264 @@
|
|||
using System.Globalization;
|
||||
using System.Security.Cryptography;
|
||||
using System.Text;
|
||||
using SpawnDev.BlazorJS.Cryptography;
|
||||
using static SpawnDev.BlazorJS.Cryptography.PortableCrypto;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
/// <summary>
|
||||
/// End-to-end encryption of one connection. Runs in the browser (WebAssembly + WebCrypto); keys never leave it.
|
||||
/// <list type="bullet">
|
||||
/// <item>
|
||||
/// Key agreement: ephemeral ECDH P-256; the private key is non-extractable and discarded after the
|
||||
/// agreement.
|
||||
/// </item>
|
||||
/// <item>
|
||||
/// Commit-then-reveal: the joiner first sends a hash commitment to its public key, the creator then reveals
|
||||
/// its key, and only afterwards the joiner reveals its key, which the creator checks against the commitment.
|
||||
/// A relay in the middle therefore has to fix the keys it shows each side before it learns the real keys, and
|
||||
/// can't search for keys that produce matching safety codes.
|
||||
/// </item>
|
||||
/// <item>
|
||||
/// One AES-256-GCM key per direction (PBKDF2-SHA-256 over the shared secret, salted with direction, code and
|
||||
/// both public keys), so a frame can't be reflected back to its sender.
|
||||
/// </item>
|
||||
/// <item>
|
||||
/// Every frame carries a sequence number inside the ciphertext: replayed, reordered or dropped frames are
|
||||
/// rejected.
|
||||
/// </item>
|
||||
/// <item>
|
||||
/// The safety code is a hash of both public keys; if both users see the same code, nobody intercepted the
|
||||
/// agreement.
|
||||
/// </item>
|
||||
/// </list>
|
||||
/// </summary>
|
||||
public sealed class SecureChannel : IDisposable
|
||||
{
|
||||
private const string Protocol = "DataTransfer/v3";
|
||||
private const int KeyDerivationIterations = 10_000;
|
||||
|
||||
private readonly PortableCrypto _crypto;
|
||||
private PortableECDHKey? _keyPair;
|
||||
private PortableAESGCMKey? _sendKey;
|
||||
private PortableAESGCMKey? _receiveKey;
|
||||
private long _sendSequence;
|
||||
private long _receiveSequence;
|
||||
|
||||
private SecureChannel(PortableCrypto crypto, PortableECDHKey keyPair, byte[] publicKey, byte[] commitment)
|
||||
{
|
||||
_crypto = crypto;
|
||||
_keyPair = keyPair;
|
||||
PublicKey = publicKey;
|
||||
Commitment = commitment;
|
||||
}
|
||||
|
||||
/// <summary>Ephemeral ECDH public key (SPKI) to send to the peer.</summary>
|
||||
public byte[] PublicKey { get; }
|
||||
|
||||
/// <summary>Hash commitment to <see cref="PublicKey" />; the joiner sends it before revealing its key.</summary>
|
||||
public byte[] Commitment { get; }
|
||||
|
||||
public bool IsEstablished => _sendKey is not null;
|
||||
|
||||
public static async Task<SecureChannel> CreateAsync(PortableCrypto crypto)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(crypto);
|
||||
|
||||
var keyPair = await crypto.GenerateECDHKey(NamedCurve.P256, false);
|
||||
try
|
||||
{
|
||||
var publicKey = await crypto.ExportPublicKeySpki(keyPair);
|
||||
var commitment = await ComputeCommitmentAsync(crypto, publicKey);
|
||||
return new SecureChannel(crypto, keyPair, publicKey, commitment);
|
||||
}
|
||||
catch
|
||||
{
|
||||
keyPair.Dispose();
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Completes the key agreement as creator and returns the safety code both users can compare.
|
||||
/// </summary>
|
||||
/// <param name="joinerCommitment">The commitment the joiner sent before the creator revealed its key.</param>
|
||||
/// <exception cref="CryptographicException">The joiner's key does not match its commitment.</exception>
|
||||
public async Task<string> EstablishAsCreatorAsync(byte[] joinerPublicKey, byte[] joinerCommitment, string code)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(joinerPublicKey);
|
||||
ArgumentNullException.ThrowIfNull(joinerCommitment);
|
||||
|
||||
var expected = await ComputeCommitmentAsync(_crypto, joinerPublicKey);
|
||||
if (!CryptographicOperations.FixedTimeEquals(expected, joinerCommitment))
|
||||
{
|
||||
throw new CryptographicException("The peer's public key does not match its commitment.");
|
||||
}
|
||||
|
||||
return await EstablishAsync(joinerPublicKey, true, code);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Completes the key agreement as joiner and returns the safety code. Must only be called after
|
||||
/// <see cref="Commitment" /> was sent, and <see cref="PublicKey" /> must only be revealed afterwards.
|
||||
/// </summary>
|
||||
public Task<string> EstablishAsJoinerAsync(byte[] creatorPublicKey, string code) =>
|
||||
EstablishAsync(creatorPublicKey, false, code);
|
||||
|
||||
/// <summary>Encrypts the next outgoing frame. Callers must send sealed frames in the order they were sealed.</summary>
|
||||
public Task<byte[]> SealAsync(Frame frame)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(frame);
|
||||
|
||||
var key = _sendKey ?? throw new InvalidOperationException("The channel is not established.");
|
||||
return _crypto.Encrypt(key, FrameCodec.Encode(++_sendSequence, frame));
|
||||
}
|
||||
|
||||
/// <summary>Decrypts and verifies the next incoming frame.</summary>
|
||||
/// <exception cref="CryptographicException">The frame was tampered with, replayed, reordered or dropped.</exception>
|
||||
public async Task<Frame> OpenAsync(byte[] ciphertext)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(ciphertext);
|
||||
|
||||
var key = _receiveKey ?? throw new InvalidOperationException("The channel is not established.");
|
||||
|
||||
byte[]? plaintext;
|
||||
try
|
||||
{
|
||||
plaintext = await _crypto.Decrypt(key, ciphertext);
|
||||
}
|
||||
catch (Exception ex) when (ex is not OperationCanceledException)
|
||||
{
|
||||
throw new CryptographicException("The frame could not be decrypted.", ex);
|
||||
}
|
||||
|
||||
if (plaintext is null)
|
||||
{
|
||||
throw new CryptographicException("The frame could not be decrypted.");
|
||||
}
|
||||
|
||||
long sequence;
|
||||
Frame frame;
|
||||
try
|
||||
{
|
||||
(sequence, frame) = FrameCodec.Decode(plaintext);
|
||||
}
|
||||
catch (InvalidDataException ex)
|
||||
{
|
||||
throw new CryptographicException("The frame is malformed.", ex);
|
||||
}
|
||||
|
||||
if (sequence != _receiveSequence + 1)
|
||||
{
|
||||
throw new CryptographicException("A frame was replayed, reordered or dropped.");
|
||||
}
|
||||
|
||||
_receiveSequence = sequence;
|
||||
return frame;
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
_keyPair?.Dispose();
|
||||
_sendKey?.Dispose();
|
||||
_receiveKey?.Dispose();
|
||||
_keyPair = null;
|
||||
_sendKey = null;
|
||||
_receiveKey = null;
|
||||
}
|
||||
|
||||
private static async Task<byte[]> ComputeCommitmentAsync(PortableCrypto crypto, byte[] publicKey) =>
|
||||
await crypto.Digest(HashName.SHA256, Concat(Encoding.UTF8.GetBytes($"{Protocol}|commitment|"), publicKey));
|
||||
|
||||
private static string FormatSafetyCode(byte[] digest)
|
||||
{
|
||||
var groups = new string[4];
|
||||
for (var group = 0; group < groups.Length; group++)
|
||||
{
|
||||
ulong value = 0;
|
||||
for (var i = 0; i < 5; i++)
|
||||
{
|
||||
value = (value << 8) | digest[group * 5 + i];
|
||||
}
|
||||
|
||||
groups[group] = (value % 100_000).ToString("D5", CultureInfo.InvariantCulture);
|
||||
}
|
||||
|
||||
return string.Join(' ', groups);
|
||||
}
|
||||
|
||||
private static byte[] Concat(params byte[][] parts)
|
||||
{
|
||||
var result = new byte[parts.Sum(part => part.Length)];
|
||||
var offset = 0;
|
||||
foreach (var part in parts)
|
||||
{
|
||||
part.CopyTo(result, offset);
|
||||
offset += part.Length;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
private async Task<string> EstablishAsync(byte[] peerPublicKey, bool isCreator, string code)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(peerPublicKey);
|
||||
ArgumentException.ThrowIfNullOrEmpty(code);
|
||||
|
||||
var keyPair = _keyPair;
|
||||
if (keyPair is null || IsEstablished)
|
||||
{
|
||||
throw new InvalidOperationException("The channel can only be established once.");
|
||||
}
|
||||
|
||||
if (peerPublicKey.AsSpan().SequenceEqual(PublicKey))
|
||||
{
|
||||
throw new CryptographicException("The peer presented our own public key.");
|
||||
}
|
||||
|
||||
_keyPair = null;
|
||||
byte[]? sharedSecret = null;
|
||||
try
|
||||
{
|
||||
using var peerKey = await _crypto.ImportECDHKey(peerPublicKey, NamedCurve.P256);
|
||||
sharedSecret = await _crypto.DeriveBits(keyPair, peerKey, 256);
|
||||
|
||||
var creatorKey = isCreator ? PublicKey : peerPublicKey;
|
||||
var joinerKey = isCreator ? peerPublicKey : PublicKey;
|
||||
var creatorToJoiner = await DeriveKeyAsync(sharedSecret, "creator->joiner", code, creatorKey, joinerKey);
|
||||
var joinerToCreator = await DeriveKeyAsync(sharedSecret, "joiner->creator", code, creatorKey, joinerKey);
|
||||
|
||||
_sendKey = isCreator ? creatorToJoiner : joinerToCreator;
|
||||
_receiveKey = isCreator ? joinerToCreator : creatorToJoiner;
|
||||
|
||||
var digest = await _crypto.Digest(HashName.SHA256, Concat(Encoding.UTF8.GetBytes($"{Protocol}|safety-code|"), creatorKey, joinerKey));
|
||||
return FormatSafetyCode(digest);
|
||||
}
|
||||
finally
|
||||
{
|
||||
if (sharedSecret is not null)
|
||||
{
|
||||
CryptographicOperations.ZeroMemory(sharedSecret);
|
||||
}
|
||||
|
||||
keyPair.Dispose();
|
||||
}
|
||||
}
|
||||
|
||||
private async Task<PortableAESGCMKey> DeriveKeyAsync(byte[] sharedSecret, string direction, string code, byte[] creatorKey, byte[] joinerKey)
|
||||
{
|
||||
var salt = await _crypto.Digest(
|
||||
HashName.SHA256,
|
||||
Concat(Encoding.UTF8.GetBytes($"{Protocol}|{direction}|{code}|"), creatorKey, joinerKey));
|
||||
|
||||
return await _crypto.GenerateAESGCMKey(
|
||||
sharedSecret,
|
||||
salt,
|
||||
KeyDerivationIterations,
|
||||
HashName.SHA256,
|
||||
32,
|
||||
16,
|
||||
12,
|
||||
false);
|
||||
}
|
||||
}
|
||||
63
DataTransfer.Client/Services/ServerStatusMonitor.cs
Normal file
63
DataTransfer.Client/Services/ServerStatusMonitor.cs
Normal file
|
|
@ -0,0 +1,63 @@
|
|||
using System.Net.Http.Json;
|
||||
using DataTransfer.Client.Shared;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
/// <summary>Periodically fetches the load of the relay server; shared by all components of the page.</summary>
|
||||
public sealed class ServerStatusMonitor(HttpClient http, ILogger<ServerStatusMonitor> logger) : IDisposable
|
||||
{
|
||||
private static readonly TimeSpan PollInterval = TimeSpan.FromSeconds(5);
|
||||
|
||||
private readonly CancellationTokenSource _stopping = new();
|
||||
private Task? _polling;
|
||||
|
||||
public event Action? Changed;
|
||||
|
||||
/// <summary>The latest status, or <c>null</c> while it is unknown.</summary>
|
||||
public ServerStatus? Current { get; private set; }
|
||||
|
||||
public void EnsureStarted() => _polling ??= PollAsync(_stopping.Token);
|
||||
|
||||
public async Task RefreshAsync(CancellationToken cancellationToken = default)
|
||||
{
|
||||
ServerStatus? status;
|
||||
try
|
||||
{
|
||||
status = await http.GetFromJsonAsync<ServerStatus>(TransferProtocol.StatusPath.TrimStart('/'), cancellationToken);
|
||||
}
|
||||
catch (OperationCanceledException) when (cancellationToken.IsCancellationRequested)
|
||||
{
|
||||
return;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
logger.LogDebug(ex, "Fetching the server status failed");
|
||||
status = null;
|
||||
}
|
||||
|
||||
if (status != Current)
|
||||
{
|
||||
Current = status;
|
||||
Changed?.Invoke();
|
||||
}
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
_stopping.Cancel();
|
||||
_stopping.Dispose();
|
||||
}
|
||||
|
||||
private async Task PollAsync(CancellationToken cancellationToken)
|
||||
{
|
||||
using var timer = new PeriodicTimer(PollInterval);
|
||||
try
|
||||
{
|
||||
do
|
||||
{
|
||||
await RefreshAsync(cancellationToken);
|
||||
} while (await timer.WaitForNextTickAsync(cancellationToken));
|
||||
}
|
||||
catch (OperationCanceledException) { }
|
||||
}
|
||||
}
|
||||
973
DataTransfer.Client/Services/TransferSession.cs
Normal file
973
DataTransfer.Client/Services/TransferSession.cs
Normal file
|
|
@ -0,0 +1,973 @@
|
|||
using System.Security.Cryptography;
|
||||
using DataTransfer.Client.Shared;
|
||||
using Microsoft.AspNetCore.Components;
|
||||
using Microsoft.AspNetCore.Components.Forms;
|
||||
using Microsoft.AspNetCore.SignalR.Client;
|
||||
using SpawnDev.BlazorJS.Cryptography;
|
||||
using SpawnDev.BlazorJS.JSObjects;
|
||||
|
||||
namespace DataTransfer.Client.Services;
|
||||
|
||||
public enum SessionState
|
||||
{
|
||||
Idle,
|
||||
|
||||
/// <summary>The server is at its connection limit; the connection is created once a slot becomes free.</summary>
|
||||
Queued,
|
||||
Joining,
|
||||
Waiting,
|
||||
Connected,
|
||||
Closed
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// One end of a transfer connection, running in the browser. Pairs via the relay hub, exchanges keys with the
|
||||
/// commit-then-reveal protocol of <see cref="SecureChannel" />, encrypts everything afterwards and keeps the decrypted
|
||||
/// chat history locally. Plaintext never leaves the browser.
|
||||
/// </summary>
|
||||
public sealed class TransferSession(
|
||||
NavigationManager navigation,
|
||||
PortableCrypto crypto,
|
||||
BotCheckSolver botCheckSolver,
|
||||
ILogger<TransferSession> logger) : IAsyncDisposable
|
||||
{
|
||||
private const string KeyExchangeFailedMessage = "The secure key exchange failed. Please start a new connection.";
|
||||
|
||||
private const string TooManyConnectionsMessage =
|
||||
"Your network already has the maximum number of open connections on this server. Please close other transfers or try again later.";
|
||||
|
||||
private const string TransferLimitMessage = "The connection was closed because it reached its transfer limit.";
|
||||
|
||||
private static readonly string ReceiveLimitMessage =
|
||||
$"The connection was closed because the other person sent more than {TransferProtocol.MaxItemsPerConnection:N0} " +
|
||||
$"messages and files or more than {DisplayFormat.Bytes(TransferProtocol.MaxFileBytesPerConnection)} in total.";
|
||||
|
||||
private static readonly string SendLimitMessage =
|
||||
$"This connection can transfer at most {TransferProtocol.MaxItemsPerConnection:N0} messages and files " +
|
||||
$"and {DisplayFormat.Bytes(TransferProtocol.MaxFileBytesPerConnection)} in total. Please start a new connection.";
|
||||
|
||||
// The creator's browser answers the joiner's commitment immediately; waiting longer means it's gone or misbehaving.
|
||||
private static readonly TimeSpan KeyExchangeTimeout = TimeSpan.FromSeconds(30);
|
||||
|
||||
private readonly List<ChatEntry> _entries = [];
|
||||
private readonly Dictionary<int, IncomingFile> _incomingFiles = [];
|
||||
private readonly List<string> _objectUrls = [];
|
||||
private readonly SemaphoreSlim _sendLock = new(1, 1);
|
||||
|
||||
private HubConnection? _hub;
|
||||
private SecureChannel? _channel;
|
||||
private KeyExchangeStep _keyExchange;
|
||||
private byte[]? _peerCommitment;
|
||||
private bool _peerJoined;
|
||||
private TaskCompletionSource _keyExchangeCompleted = CreateCompletionSource();
|
||||
private DateTimeOffset _lastActivity;
|
||||
private int _nextFileId;
|
||||
|
||||
// Per-connection totals, limited by TransferProtocol.MaxItemsPerConnection and MaxFileBytesPerConnection.
|
||||
private int _sentItems;
|
||||
private long _sentFileBytes;
|
||||
private int _receivedItems;
|
||||
private long _receivedFileBytes;
|
||||
|
||||
/// <summary>Steps of the commit-then-reveal key exchange; see <see cref="SecureChannel" />.</summary>
|
||||
private enum KeyExchangeStep
|
||||
{
|
||||
None,
|
||||
|
||||
/// <summary>Creator: waits for the joiner's commitment to its public key.</summary>
|
||||
AwaitingCommitment,
|
||||
|
||||
/// <summary>Creator: revealed its public key and waits for the joiner's.</summary>
|
||||
AwaitingJoinerKey,
|
||||
|
||||
/// <summary>Joiner: committed to its public key and waits for the creator's.</summary>
|
||||
AwaitingCreatorKey,
|
||||
|
||||
Completed
|
||||
}
|
||||
|
||||
public event Action? Changed;
|
||||
|
||||
public IReadOnlyList<ChatEntry> Entries => _entries;
|
||||
|
||||
public TimeSpan InactivityTimeout { get; private set; }
|
||||
|
||||
/// <summary>A new code expires unless someone joins within this period.</summary>
|
||||
public TimeSpan JoinTimeout { get; private set; }
|
||||
|
||||
/// <summary>The timeout <see cref="RemainingTime" /> counts down: joining for a fresh code, inactivity afterwards.</summary>
|
||||
public TimeSpan CountdownDuration =>
|
||||
State == SessionState.Waiting && !_peerJoined ? JoinTimeout : InactivityTimeout;
|
||||
|
||||
/// <summary>Time until the server removes the connection unless someone joins or something is transmitted.</summary>
|
||||
public TimeSpan? RemainingTime => State is SessionState.Waiting or SessionState.Connected
|
||||
? _lastActivity + CountdownDuration - DateTimeOffset.UtcNow
|
||||
: null;
|
||||
|
||||
public SessionState State { get; private set; }
|
||||
|
||||
/// <summary>The normalized connection code.</summary>
|
||||
public string? Code { get; private set; }
|
||||
|
||||
public string? SafetyCode { get; private set; }
|
||||
|
||||
public string? ClosedMessage { get; private set; }
|
||||
|
||||
/// <summary>1-based position in the server's queue while <see cref="SessionState.Queued" />.</summary>
|
||||
public int QueuePosition { get; private set; }
|
||||
|
||||
public bool HasPeer { get; private set; }
|
||||
|
||||
/// <summary>Someone joined the creator's connection and the key exchange is in progress.</summary>
|
||||
public bool IsPeerJoining => State == SessionState.Waiting && _peerJoined;
|
||||
|
||||
public bool IsSendingFiles { get; private set; }
|
||||
|
||||
/// <summary>The browser is solving the server's bot check before the connection is created.</summary>
|
||||
public bool IsVerifyingBrowser { get; private set; }
|
||||
|
||||
/// <returns>An error message for the user, or <c>null</c>.</returns>
|
||||
public async Task<string?> CreateAsync()
|
||||
{
|
||||
var (hub, _) = await StartAsync();
|
||||
var botCheckSolution = await SolveBotCheckAsync(hub);
|
||||
if (!IsCurrent(hub))
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
var result = await hub.InvokeAsync<CreateConnectionResult>(nameof(ITransferHub.CreateConnection), botCheckSolution);
|
||||
|
||||
// A ConnectionCreated message may have overtaken this result and already admitted the queued request.
|
||||
if (!IsCurrent(hub) || State != SessionState.Idle)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
switch (result)
|
||||
{
|
||||
case { Status: CreateStatus.Created, Code: not null }:
|
||||
Admit(result);
|
||||
return null;
|
||||
|
||||
case { Status: CreateStatus.Queued }:
|
||||
UpdateQueuePosition(result.QueuePosition);
|
||||
State = SessionState.Queued;
|
||||
NotifyChanged();
|
||||
return null;
|
||||
|
||||
case { Status: CreateStatus.TooManyConnections }:
|
||||
await ResetAsync();
|
||||
return TooManyConnectionsMessage;
|
||||
|
||||
case { Status: CreateStatus.BotCheckFailed }:
|
||||
await ResetAsync();
|
||||
return "The automatic check against bots failed. Please try again.";
|
||||
|
||||
default:
|
||||
await ResetAsync();
|
||||
return "The server is at its connection limit and the queue is full. Please try again later.";
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>Joins a connection and completes the key exchange.</summary>
|
||||
/// <returns>An error message for the user, or <c>null</c>.</returns>
|
||||
public async Task<string?> JoinAsync(string? input)
|
||||
{
|
||||
if (!ConnectionCode.TryNormalize(input, out var code))
|
||||
{
|
||||
return $"Please enter a valid {ConnectionCode.Length}-character code.";
|
||||
}
|
||||
|
||||
var (hub, channel) = await StartAsync();
|
||||
var keyExchangeCompleted = _keyExchangeCompleted.Task;
|
||||
Code = code;
|
||||
State = SessionState.Joining;
|
||||
NotifyChanged();
|
||||
|
||||
var result = await hub.InvokeAsync<JoinConnectionResult>(nameof(ITransferHub.JoinConnection), code);
|
||||
if (!IsCurrent(hub) || State != SessionState.Joining)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
if (result.Status != JoinStatus.Joined)
|
||||
{
|
||||
await ResetAsync();
|
||||
return result.Status switch
|
||||
{
|
||||
JoinStatus.InvalidCode => $"Please enter a valid {ConnectionCode.Length}-character code.",
|
||||
JoinStatus.AlreadyInUse => "This connection already has two participants.",
|
||||
JoinStatus.TooManyConnections => TooManyConnectionsMessage,
|
||||
_ => "No open connection with this code exists. It may have expired."
|
||||
};
|
||||
}
|
||||
|
||||
InactivityTimeout = result.InactivityTimeout;
|
||||
|
||||
// Commit to our key before the creator reveals its key; ours is revealed only after receiving it.
|
||||
_keyExchange = KeyExchangeStep.AwaitingCreatorKey;
|
||||
if (!await SendKeyExchangeMessageAsync(hub, HandshakeCodec.EncodeCommitment(channel.Commitment)))
|
||||
{
|
||||
await CloseAsync(hub, KeyExchangeFailedMessage);
|
||||
return null;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
// Completed by OnReceiveAsync once the creator's key arrived and ours was revealed.
|
||||
await keyExchangeCompleted.WaitAsync(KeyExchangeTimeout);
|
||||
}
|
||||
catch (TimeoutException)
|
||||
{
|
||||
await CloseAsync(hub, "The other person did not respond. Please ask for a new code.");
|
||||
}
|
||||
catch (OperationCanceledException)
|
||||
{
|
||||
// Closed or reset meanwhile; the closed message tells the user why.
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
/// <returns>An error message for the user, or <c>null</c>.</returns>
|
||||
public async Task<string?> SendTextAsync(string? text)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(text))
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
if (text.Length > TransferProtocol.MaxMessageLength)
|
||||
{
|
||||
return "The message is too long.";
|
||||
}
|
||||
|
||||
if (_sentItems >= TransferProtocol.MaxItemsPerConnection)
|
||||
{
|
||||
return SendLimitMessage;
|
||||
}
|
||||
|
||||
var error = await SendFrameAsync(new TextFrame(text));
|
||||
if (error is null)
|
||||
{
|
||||
_sentItems++;
|
||||
AddEntry(
|
||||
new ChatEntry
|
||||
{
|
||||
Kind = ChatEntryKind.Outgoing,
|
||||
Text = text
|
||||
});
|
||||
}
|
||||
|
||||
NotifyChanged();
|
||||
return error;
|
||||
}
|
||||
|
||||
public async Task SendFilesAsync(IReadOnlyList<IBrowserFile> files)
|
||||
{
|
||||
if (State != SessionState.Connected || IsSendingFiles || files.Count == 0)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
IsSendingFiles = true;
|
||||
NotifyChanged();
|
||||
try
|
||||
{
|
||||
foreach (var file in files)
|
||||
{
|
||||
if (State != SessionState.Connected)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
if (file.Size > TransferProtocol.MaxFileBytes)
|
||||
{
|
||||
AddSystemEntry($"\"{file.Name}\" is too large (max. {DisplayFormat.Bytes(TransferProtocol.MaxFileBytes)}).", true);
|
||||
continue;
|
||||
}
|
||||
|
||||
// The peer closes the connection beyond these limits, so stop before reaching them.
|
||||
if (_sentItems >= TransferProtocol.MaxItemsPerConnection
|
||||
|| _sentFileBytes + file.Size > TransferProtocol.MaxFileBytesPerConnection)
|
||||
{
|
||||
AddSystemEntry(SendLimitMessage, true);
|
||||
break;
|
||||
}
|
||||
|
||||
var entry = new ChatEntry
|
||||
{
|
||||
Kind = ChatEntryKind.Outgoing,
|
||||
FileName = file.Name,
|
||||
FileSize = file.Size
|
||||
};
|
||||
AddEntry(entry);
|
||||
NotifyChanged();
|
||||
|
||||
var error = await SendFileAsync(file, entry);
|
||||
if (error is not null)
|
||||
{
|
||||
entry.Error = error;
|
||||
}
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
IsSendingFiles = false;
|
||||
NotifyChanged();
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>Leaves the connection but keeps the chat history visible.</summary>
|
||||
public async Task DisconnectAsync()
|
||||
{
|
||||
if (_hub is not { } hub || State is not (SessionState.Waiting or SessionState.Connected))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
MarkClosed("You closed the connection.");
|
||||
try
|
||||
{
|
||||
await hub.InvokeAsync(nameof(ITransferHub.LeaveConnection));
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
// Includes the cancellation when a reset disposes the hub meanwhile.
|
||||
logger.LogDebug(ex, "Leaving the connection failed");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>Leaves any connection, forgets keys and history and returns to <see cref="SessionState.Idle" />.</summary>
|
||||
public async Task ResetAsync()
|
||||
{
|
||||
var hub = _hub;
|
||||
_hub = null;
|
||||
_keyExchangeCompleted.TrySetCanceled();
|
||||
_keyExchangeCompleted = CreateCompletionSource();
|
||||
_keyExchange = KeyExchangeStep.None;
|
||||
_peerCommitment = null;
|
||||
_peerJoined = false;
|
||||
_channel?.Dispose();
|
||||
_channel = null;
|
||||
|
||||
_incomingFiles.Clear();
|
||||
_entries.Clear();
|
||||
RevokeObjectUrls();
|
||||
|
||||
State = SessionState.Idle;
|
||||
Code = null;
|
||||
SafetyCode = null;
|
||||
ClosedMessage = null;
|
||||
QueuePosition = 0;
|
||||
HasPeer = false;
|
||||
IsVerifyingBrowser = false;
|
||||
_nextFileId = 0;
|
||||
_sentItems = 0;
|
||||
_sentFileBytes = 0;
|
||||
_receivedItems = 0;
|
||||
_receivedFileBytes = 0;
|
||||
|
||||
if (hub is not null)
|
||||
{
|
||||
try
|
||||
{
|
||||
// Disconnecting makes the server close the connection and notify the peer.
|
||||
await hub.DisposeAsync();
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
logger.LogDebug(ex, "Disposing the hub connection failed");
|
||||
}
|
||||
}
|
||||
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
public async ValueTask DisposeAsync()
|
||||
{
|
||||
Changed = null;
|
||||
await ResetAsync();
|
||||
}
|
||||
|
||||
private static TaskCompletionSource CreateCompletionSource() => new(TaskCreationOptions.RunContinuationsAsynchronously);
|
||||
|
||||
/// <summary>Proof of work against creating connections in bulk; see <see cref="BotCheckChallenge" />.</summary>
|
||||
/// <returns>The solution, or <c>-1</c> if the server doesn't require the check.</returns>
|
||||
private async Task<int> SolveBotCheckAsync(HubConnection hub)
|
||||
{
|
||||
var challenge = await hub.InvokeAsync<BotCheckChallenge?>(nameof(ITransferHub.GetBotCheckChallenge));
|
||||
if (challenge is null || !IsCurrent(hub))
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
IsVerifyingBrowser = true;
|
||||
NotifyChanged();
|
||||
try
|
||||
{
|
||||
return await botCheckSolver.SolveAsync(challenge);
|
||||
}
|
||||
finally
|
||||
{
|
||||
if (IsCurrent(hub))
|
||||
{
|
||||
IsVerifyingBrowser = false;
|
||||
NotifyChanged();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private async Task<(HubConnection Hub, SecureChannel Channel)> StartAsync()
|
||||
{
|
||||
await ResetAsync();
|
||||
|
||||
var channel = await SecureChannel.CreateAsync(crypto);
|
||||
var hub = new HubConnectionBuilder()
|
||||
// Relative to the base URI like all other requests, so the app also works under a path base.
|
||||
.WithUrl(navigation.ToAbsoluteUri(TransferProtocol.HubPath.TrimStart('/')))
|
||||
.Build();
|
||||
|
||||
// SignalR invokes these handlers one after another, in the order the server sent the messages.
|
||||
hub.On(nameof(ITransferHubClient.PeerJoined), () => OnPeerJoined(hub));
|
||||
hub.On<byte[]>(nameof(ITransferHubClient.Receive), frame => OnReceiveAsync(hub, frame));
|
||||
hub.On<ConnectionClosedReason>(nameof(ITransferHubClient.ConnectionClosed), reason => OnConnectionClosed(hub, reason));
|
||||
hub.On<CreateConnectionResult>(nameof(ITransferHubClient.ConnectionCreated), result => OnConnectionCreated(hub, result));
|
||||
hub.On<int>(nameof(ITransferHubClient.QueuePositionChanged), position => OnQueuePositionChanged(hub, position));
|
||||
hub.Closed += error => OnHubClosed(hub, error);
|
||||
|
||||
_hub = hub;
|
||||
_channel = channel;
|
||||
try
|
||||
{
|
||||
await hub.StartAsync();
|
||||
}
|
||||
catch
|
||||
{
|
||||
await ResetAsync();
|
||||
throw;
|
||||
}
|
||||
|
||||
return (hub, channel);
|
||||
}
|
||||
|
||||
private void OnPeerJoined(HubConnection hub)
|
||||
{
|
||||
if (IsCurrent(hub) && State == SessionState.Waiting)
|
||||
{
|
||||
// The server counts the join as activity, so the countdown switches to the inactivity timeout.
|
||||
_peerJoined = true;
|
||||
Touch();
|
||||
NotifyChanged();
|
||||
}
|
||||
}
|
||||
|
||||
private async Task OnReceiveAsync(HubConnection hub, byte[] frame)
|
||||
{
|
||||
if (!IsCurrent(hub) || _channel is not { } channel)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (_keyExchange != KeyExchangeStep.Completed)
|
||||
{
|
||||
await ContinueKeyExchangeAsync(hub, channel, frame);
|
||||
return;
|
||||
}
|
||||
|
||||
if (State != SessionState.Connected)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
var decrypted = await channel.OpenAsync(frame);
|
||||
if (!IsCurrent(hub) || State != SessionState.Connected)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
Touch();
|
||||
if (ExceedsReceiveLimits(decrypted))
|
||||
{
|
||||
logger.LogWarning("The peer exceeded the per-connection limits");
|
||||
await CloseAsync(hub, ReceiveLimitMessage);
|
||||
}
|
||||
else
|
||||
{
|
||||
Handle(decrypted);
|
||||
}
|
||||
}
|
||||
catch (Exception ex) when (ex is CryptographicException or InvalidDataException)
|
||||
{
|
||||
if (!IsCurrent(hub) || State != SessionState.Connected)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
logger.LogWarning(ex, "Rejected an incoming frame");
|
||||
AddSystemEntry("Received data failed verification and was discarded.", true);
|
||||
await CloseAsync(hub, "The connection was closed because received data failed verification.");
|
||||
}
|
||||
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
/// <summary>Handles the plaintext key-exchange messages that precede all encrypted frames.</summary>
|
||||
private async Task ContinueKeyExchangeAsync(HubConnection hub, SecureChannel channel, byte[] message)
|
||||
{
|
||||
try
|
||||
{
|
||||
switch (_keyExchange)
|
||||
{
|
||||
case KeyExchangeStep.AwaitingCommitment when State == SessionState.Waiting:
|
||||
{
|
||||
_peerCommitment = HandshakeCodec.DecodeCommitment(message);
|
||||
_peerJoined = true;
|
||||
_keyExchange = KeyExchangeStep.AwaitingJoinerKey;
|
||||
NotifyChanged();
|
||||
|
||||
// The joiner is bound to its key now, so ours can be revealed.
|
||||
if (!await SendKeyExchangeMessageAsync(hub, HandshakeCodec.EncodePublicKey(channel.PublicKey)))
|
||||
{
|
||||
await CloseAsync(hub, KeyExchangeFailedMessage);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case KeyExchangeStep.AwaitingJoinerKey
|
||||
when State == SessionState.Waiting && _peerCommitment is { } commitment && Code is { } code:
|
||||
{
|
||||
var joinerKey = HandshakeCodec.DecodePublicKey(message);
|
||||
var safetyCode = await channel.EstablishAsCreatorAsync(joinerKey, commitment, code);
|
||||
CompleteKeyExchange(hub, safetyCode);
|
||||
break;
|
||||
}
|
||||
|
||||
case KeyExchangeStep.AwaitingCreatorKey when State == SessionState.Joining && Code is { } code:
|
||||
{
|
||||
var creatorKey = HandshakeCodec.DecodePublicKey(message);
|
||||
var safetyCode = await channel.EstablishAsJoinerAsync(creatorKey, code);
|
||||
if (!IsCurrent(hub) || State != SessionState.Joining)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
// Revealed only now that the creator's key is fixed.
|
||||
if (!await SendKeyExchangeMessageAsync(hub, HandshakeCodec.EncodePublicKey(channel.PublicKey)))
|
||||
{
|
||||
await CloseAsync(hub, KeyExchangeFailedMessage);
|
||||
break;
|
||||
}
|
||||
|
||||
CompleteKeyExchange(hub, safetyCode);
|
||||
break;
|
||||
}
|
||||
|
||||
default:
|
||||
throw new InvalidDataException("Unexpected key-exchange message.");
|
||||
}
|
||||
}
|
||||
catch (Exception ex) when (ex is not OperationCanceledException)
|
||||
{
|
||||
logger.LogWarning(ex, "The key exchange failed");
|
||||
await CloseAsync(hub, KeyExchangeFailedMessage);
|
||||
}
|
||||
}
|
||||
|
||||
private void CompleteKeyExchange(HubConnection hub, string safetyCode)
|
||||
{
|
||||
if (!IsCurrent(hub) || State is not (SessionState.Waiting or SessionState.Joining))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
SafetyCode = safetyCode;
|
||||
_keyExchange = KeyExchangeStep.Completed;
|
||||
_peerCommitment = null;
|
||||
HasPeer = true;
|
||||
State = SessionState.Connected;
|
||||
Touch();
|
||||
_keyExchangeCompleted.TrySetResult();
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
private Task OnConnectionClosed(HubConnection hub, ConnectionClosedReason reason)
|
||||
{
|
||||
if (IsCurrent(hub) && State is SessionState.Joining or SessionState.Waiting or SessionState.Connected)
|
||||
{
|
||||
var inactivity = DisplayFormat.Duration(InactivityTimeout);
|
||||
MarkClosed(
|
||||
reason switch
|
||||
{
|
||||
ConnectionClosedReason.Inactivity when State == SessionState.Waiting && !_peerJoined =>
|
||||
$"The code expired because nobody joined within {DisplayFormat.Duration(JoinTimeout)}.",
|
||||
ConnectionClosedReason.Inactivity => $"The connection was closed because nothing was sent for {inactivity}.",
|
||||
ConnectionClosedReason.MaximumLifetime => "The connection was closed because it reached its maximum duration.",
|
||||
ConnectionClosedReason.TransferLimitReached => TransferLimitMessage,
|
||||
_ when !HasPeer => "The other person left before the connection was established.",
|
||||
_ => "The other person closed the connection."
|
||||
});
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
// Idle covers messages that arrive while the CreateConnection result is still on its way.
|
||||
private Task OnConnectionCreated(HubConnection hub, CreateConnectionResult result)
|
||||
{
|
||||
if (IsCurrent(hub) && State is SessionState.Idle or SessionState.Queued && result is { Status: CreateStatus.Created, Code: not null })
|
||||
{
|
||||
Admit(result);
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
private Task OnQueuePositionChanged(HubConnection hub, int position)
|
||||
{
|
||||
if (IsCurrent(hub) && State is SessionState.Idle or SessionState.Queued && position > 0)
|
||||
{
|
||||
UpdateQueuePosition(position);
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
private void Admit(CreateConnectionResult result)
|
||||
{
|
||||
Code = result.Code;
|
||||
JoinTimeout = result.JoinTimeout;
|
||||
InactivityTimeout = result.InactivityTimeout;
|
||||
QueuePosition = 0;
|
||||
_keyExchange = KeyExchangeStep.AwaitingCommitment;
|
||||
Touch();
|
||||
State = SessionState.Waiting;
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
// The position only ever decreases, so an older value that arrives late must not move the client back.
|
||||
private void UpdateQueuePosition(int position) =>
|
||||
QueuePosition = QueuePosition == 0 ? position : Math.Min(QueuePosition, position);
|
||||
|
||||
private Task OnHubClosed(HubConnection hub, Exception? error)
|
||||
{
|
||||
if (IsCurrent(hub) && State is SessionState.Queued or SessionState.Joining or SessionState.Waiting or SessionState.Connected)
|
||||
{
|
||||
logger.LogWarning(error, "Lost the connection to the server");
|
||||
MarkClosed(
|
||||
State == SessionState.Queued
|
||||
? "The connection to the server was lost while waiting in the queue."
|
||||
: "The connection to the server was lost.");
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Everything received stays in memory until the session is reset, so a peer must not be able to make this
|
||||
/// browser keep an unbounded number of entries or files.
|
||||
/// </summary>
|
||||
private bool ExceedsReceiveLimits(Frame frame) => frame switch
|
||||
{
|
||||
TextFrame => _receivedItems >= TransferProtocol.MaxItemsPerConnection,
|
||||
FileStartFrame start => _receivedItems >= TransferProtocol.MaxItemsPerConnection
|
||||
|| _receivedFileBytes + Math.Max(start.Size, 0) > TransferProtocol.MaxFileBytesPerConnection,
|
||||
_ => false
|
||||
};
|
||||
|
||||
private void Handle(Frame frame)
|
||||
{
|
||||
switch (frame)
|
||||
{
|
||||
case TextFrame text:
|
||||
if (text.Text.Length > TransferProtocol.MaxMessageLength)
|
||||
{
|
||||
throw new InvalidDataException("The message is too long.");
|
||||
}
|
||||
|
||||
_receivedItems++;
|
||||
AddEntry(
|
||||
new ChatEntry
|
||||
{
|
||||
Kind = ChatEntryKind.Incoming,
|
||||
Text = text.Text
|
||||
});
|
||||
break;
|
||||
|
||||
case FileStartFrame start:
|
||||
// Senders transmit one file at a time; this also bounds the memory a peer can make us allocate.
|
||||
if (start.Size is < 0 or > TransferProtocol.MaxFileBytes || _incomingFiles.Count > 0)
|
||||
{
|
||||
throw new InvalidDataException("Invalid file announcement.");
|
||||
}
|
||||
|
||||
_receivedItems++;
|
||||
_receivedFileBytes += start.Size;
|
||||
var fileName = FileNameSanitizer.Sanitize(start.Name);
|
||||
var entry = new ChatEntry
|
||||
{
|
||||
Kind = ChatEntryKind.Incoming,
|
||||
FileName = fileName,
|
||||
FileSize = start.Size,
|
||||
IsRiskyFile = RiskyFileTypes.IsRisky(fileName)
|
||||
};
|
||||
AddEntry(entry);
|
||||
|
||||
var incoming = new IncomingFile(entry, new byte[start.Size]);
|
||||
if (start.Size == 0)
|
||||
{
|
||||
CompleteFile(incoming);
|
||||
}
|
||||
else
|
||||
{
|
||||
_incomingFiles.Add(start.FileId, incoming);
|
||||
}
|
||||
|
||||
break;
|
||||
|
||||
case FileChunkFrame chunk:
|
||||
if (!_incomingFiles.TryGetValue(chunk.FileId, out var file)
|
||||
|| chunk.Data.Length == 0
|
||||
|| chunk.Data.Length > file.Buffer.Length - file.Entry.TransferredBytes)
|
||||
{
|
||||
throw new InvalidDataException("Unexpected file chunk.");
|
||||
}
|
||||
|
||||
chunk.Data.Span.CopyTo(file.Buffer.AsSpan((int)file.Entry.TransferredBytes));
|
||||
file.Entry.TransferredBytes += chunk.Data.Length;
|
||||
if (file.Entry.TransferredBytes == file.Buffer.Length)
|
||||
{
|
||||
_incomingFiles.Remove(chunk.FileId);
|
||||
CompleteFile(file);
|
||||
}
|
||||
|
||||
break;
|
||||
|
||||
case FileCancelFrame cancel:
|
||||
if (_incomingFiles.Remove(cancel.FileId, out var cancelled))
|
||||
{
|
||||
cancelled.Entry.Error = "The sender cancelled the transfer.";
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
private void CompleteFile(IncomingFile file)
|
||||
{
|
||||
try
|
||||
{
|
||||
// Always offered as a download (octet-stream); never rendered inline in this origin.
|
||||
using var blob = new Blob(new[] { file.Buffer }, new BlobOptions { Type = "application/octet-stream" });
|
||||
var url = URL.CreateObjectURL(blob);
|
||||
_objectUrls.Add(url);
|
||||
file.Entry.DownloadUrl = url;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
logger.LogError(ex, "Preparing a received file for download failed");
|
||||
file.Entry.Error = "The file could not be prepared for download.";
|
||||
}
|
||||
}
|
||||
|
||||
private async Task<string?> SendFileAsync(IBrowserFile file, ChatEntry entry)
|
||||
{
|
||||
var fileId = ++_nextFileId;
|
||||
var error = await SendFrameAsync(new FileStartFrame(fileId, file.Name, file.Size));
|
||||
if (error is not null)
|
||||
{
|
||||
return error;
|
||||
}
|
||||
|
||||
// Counted like the receiver does: the announced size, even if the transfer is cancelled later.
|
||||
_sentItems++;
|
||||
_sentFileBytes += file.Size;
|
||||
|
||||
try
|
||||
{
|
||||
await using var stream = file.OpenReadStream(TransferProtocol.MaxFileBytes);
|
||||
var buffer = new byte[TransferProtocol.FileChunkBytes];
|
||||
while (entry.TransferredBytes < file.Size)
|
||||
{
|
||||
var expected = (int)Math.Min(buffer.Length, file.Size - entry.TransferredBytes);
|
||||
var read = await stream.ReadAtLeastAsync(buffer.AsMemory(0, expected), expected, false);
|
||||
if (read != expected)
|
||||
{
|
||||
throw new IOException("The file changed while it was being sent.");
|
||||
}
|
||||
|
||||
error = await SendFrameAsync(new FileChunkFrame(fileId, buffer.AsMemory(0, read)));
|
||||
if (error is not null)
|
||||
{
|
||||
return error;
|
||||
}
|
||||
|
||||
entry.TransferredBytes += read;
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
// SendFrameAsync never throws, so this only covers reading the file (including browser errors).
|
||||
logger.LogWarning(ex, "Reading a file failed");
|
||||
await SendFrameAsync(new FileCancelFrame(fileId));
|
||||
return "The file could not be read.";
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Seals and sends one frame. Serialized, so frames reach the peer in sequence-number order. Never throws.
|
||||
/// </summary>
|
||||
/// <returns>An error message for the user, or <c>null</c>.</returns>
|
||||
private async Task<string?> SendFrameAsync(Frame frame)
|
||||
{
|
||||
await _sendLock.WaitAsync();
|
||||
try
|
||||
{
|
||||
if (State != SessionState.Connected || _hub is not { } hub || _channel is not { } channel)
|
||||
{
|
||||
return "The connection is closed.";
|
||||
}
|
||||
|
||||
return await SealAndSendAsync(hub, channel, frame);
|
||||
}
|
||||
finally
|
||||
{
|
||||
_sendLock.Release();
|
||||
}
|
||||
}
|
||||
|
||||
private async Task<string?> SealAndSendAsync(HubConnection hub, SecureChannel channel, Frame frame)
|
||||
{
|
||||
SendStatus status;
|
||||
try
|
||||
{
|
||||
var sealedFrame = await channel.SealAsync(frame);
|
||||
status = await hub.InvokeAsync<SendStatus>(nameof(ITransferHub.Send), sealedFrame);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
// Also covers the cancellation of pending invocations when a reset disposes the hub.
|
||||
if (!IsCurrent(hub))
|
||||
{
|
||||
return "The connection is closed.";
|
||||
}
|
||||
|
||||
logger.LogWarning(ex, "Sending a frame failed");
|
||||
await CloseAsync(hub, "The connection was closed because sending failed.");
|
||||
return "Sending failed.";
|
||||
}
|
||||
|
||||
if (status == SendStatus.Delivered)
|
||||
{
|
||||
Touch();
|
||||
return null;
|
||||
}
|
||||
|
||||
// A sealed but unsent frame leaves a gap in the sequence numbers, so the channel can't be used anymore.
|
||||
await CloseAsync(hub, status == SendStatus.TransferLimitReached ? TransferLimitMessage : "The connection was closed.");
|
||||
return status switch
|
||||
{
|
||||
SendStatus.InvalidSize => "The data is too large.",
|
||||
SendStatus.TransferLimitReached => "The transfer limit of this connection was reached.",
|
||||
_ => "The connection is closed."
|
||||
};
|
||||
}
|
||||
|
||||
/// <summary>Sends a plaintext key-exchange message; only public data is sent this way.</summary>
|
||||
private async Task<bool> SendKeyExchangeMessageAsync(HubConnection hub, byte[] message)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (await hub.InvokeAsync<SendStatus>(nameof(ITransferHub.Send), message) != SendStatus.Delivered)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
Touch();
|
||||
return true;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
logger.LogDebug(ex, "Sending a key-exchange message failed");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>Closes locally and asks the server to close the connection for the peer, without awaiting a response.</summary>
|
||||
private async Task CloseAsync(HubConnection hub, string message)
|
||||
{
|
||||
if (!IsCurrent(hub) || State is SessionState.Idle or SessionState.Closed)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
MarkClosed(message);
|
||||
try
|
||||
{
|
||||
await hub.SendAsync(nameof(ITransferHub.LeaveConnection));
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
logger.LogDebug(ex, "Leaving the connection failed");
|
||||
}
|
||||
}
|
||||
|
||||
private void MarkClosed(string message)
|
||||
{
|
||||
State = SessionState.Closed;
|
||||
ClosedMessage = message;
|
||||
_keyExchangeCompleted.TrySetCanceled();
|
||||
|
||||
foreach (var file in _incomingFiles.Values)
|
||||
{
|
||||
file.Entry.Error ??= "The transfer was interrupted.";
|
||||
}
|
||||
|
||||
_incomingFiles.Clear();
|
||||
NotifyChanged();
|
||||
}
|
||||
|
||||
private bool IsCurrent(HubConnection hub) => ReferenceEquals(_hub, hub);
|
||||
|
||||
private void Touch() => _lastActivity = DateTimeOffset.UtcNow;
|
||||
|
||||
private void AddEntry(ChatEntry entry) => _entries.Add(entry);
|
||||
|
||||
private void AddSystemEntry(string text, bool isError) =>
|
||||
_entries.Add(
|
||||
new ChatEntry
|
||||
{
|
||||
Kind = ChatEntryKind.System,
|
||||
Text = text,
|
||||
Error = isError ? text : null
|
||||
});
|
||||
|
||||
private void NotifyChanged() => Changed?.Invoke();
|
||||
|
||||
private void RevokeObjectUrls()
|
||||
{
|
||||
foreach (var url in _objectUrls)
|
||||
{
|
||||
try
|
||||
{
|
||||
URL.RevokeObjectURL(url);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
logger.LogDebug(ex, "Revoking an object URL failed");
|
||||
}
|
||||
}
|
||||
|
||||
_objectUrls.Clear();
|
||||
}
|
||||
|
||||
private sealed record IncomingFile(ChatEntry Entry, byte[] Buffer);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue