51 lines
2.2 KiB
C#
51 lines
2.2 KiB
C#
|
using System.Buffers;
|
||
|
|
using System.Text.Json;
|
||
|
|
using VoiceCat.Crypto;
|
||
|
|
using VoiceCat.Protocol;
|
||
|
|
using Voicecat.V1;
|
||
|
|
|
||
|
|
namespace VoiceCat.Tests;
|
||
|
|
|
||
|
|
public class GoldenTests
|
||
|
|
{
|
||
|
|
[Fact]
|
||
|
|
public void EnvelopeMatchesCppFixture()
|
||
|
|
{
|
||
|
|
using var fixture = Load();
|
||
|
|
var expected = Convert.FromHexString(fixture.RootElement.GetProperty("envelope").GetString()!);
|
||
|
|
var envelope = new Envelope { RequestId = 42, ClientHello = new() { ProtoVersion = 1, ClientName = "test-client", ClientVersion = "0.0.1" } };
|
||
|
|
envelope.ClientHello.Features.Add("text");
|
||
|
|
var output = new ArrayBufferWriter<byte>();
|
||
|
|
ControlFraming.WriteEnvelope(output, envelope);
|
||
|
|
Assert.Equal(expected, output.WrittenSpan.ToArray());
|
||
|
|
}
|
||
|
|
|
||
|
|
[Theory]
|
||
|
|
[InlineData(false)]
|
||
|
|
[InlineData(true)]
|
||
|
|
public void MediaPacketsMatchCppFixtures(bool managed)
|
||
|
|
{
|
||
|
|
using var fixture = Load();
|
||
|
|
foreach (var vector in fixture.RootElement.GetProperty("media").EnumerateArray())
|
||
|
|
{
|
||
|
|
byte[] key = Convert.FromHexString(vector.GetProperty("key").GetString()!);
|
||
|
|
byte[] plaintext = Convert.FromHexString(vector.GetProperty("plaintext").GetString()!);
|
||
|
|
byte[] expected = Convert.FromHexString(vector.GetProperty("packet").GetString()!);
|
||
|
|
ulong sequence = vector.GetProperty("sequence").GetUInt64();
|
||
|
|
using var sender = new MediaEncryptor(key, managed, sequence);
|
||
|
|
using var receiver = new MediaDecryptor(key, managed);
|
||
|
|
var header = new VoiceFrameHeader(MediaFrameType.Voice, VoiceFrameFlags.Marker, 0, 0xcafebabe, 0, 960);
|
||
|
|
byte[] actual = new byte[expected.Length];
|
||
|
|
sender.Encrypt(header, plaintext, actual);
|
||
|
|
Assert.Equal(expected, actual);
|
||
|
|
byte[] decoded = new byte[plaintext.Length];
|
||
|
|
Assert.True(receiver.TryDecrypt(expected, decoded, out var parsed, out int written));
|
||
|
|
Assert.Equal(sequence, parsed.Sequence);
|
||
|
|
Assert.Equal(plaintext.Length, written);
|
||
|
|
Assert.Equal(plaintext, decoded);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private static JsonDocument Load() => JsonDocument.Parse(File.ReadAllText(Path.Combine(AppContext.BaseDirectory, "Fixtures", "cpp-wire.json")));
|
||
|
|
}
|