public void TestAddInt()
{
BlobBuilder builder = new BlobBuilder();
UInt32 value = 12345;
builder.AddUInt32(value);
Assert.That(builder.GetBlob(), Is.EqualTo(value.ToBytes()));
}
public void TestAddBytes()
{
BlobBuilder builder = new BlobBuilder();
byte[] value = { 0, 1, 2, 3, 4, 5 };
builder.AddBytes(value);
Assert.That(builder.GetBlob(), Is.EqualTo(value));
}
public void TestAddBlob()
{
BlobBuilder builder = new BlobBuilder();
byte[] blob = { 0, 1, 2, 3, 4, 5 };
builder.AddBlob(blob);
byte[] expected = new byte[blob.Length + 4];
Array.Copy(blob.Length.ToBytes(), expected, 4);
Array.Copy(blob, 0, expected, 4, blob.Length);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
}
public void TestAddString()
{
BlobBuilder builder = new BlobBuilder();
string value = "test string";
builder.AddStringBlob(value);
byte[] valueBytes = Encoding.UTF8.GetBytes(value);
byte[] expected = new byte[valueBytes.Length + 4];
Array.Copy(valueBytes.Length.ToBytes(), expected, 4);
Array.Copy(valueBytes, 0, expected, 4, valueBytes.Length);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
}
public void TestAddBigInt()
{
BlobBuilder builder = new BlobBuilder();
BigInteger value = new BigInteger("12398259028592293582039293420948023");
builder.AddBigIntBlob(value);
byte[] valueBytes = value.ToByteArrayUnsigned();
//Assert.That(valueBytes[0], Is.EqualTo(0));
byte[] expected = new byte[valueBytes.Length + 4];
Array.Copy(valueBytes.Length.ToBytes(), expected, 4);
Array.Copy(valueBytes, 0, expected, 4, valueBytes.Length);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
}
public void TestAddSsh1BigIntBlob()
{
BlobBuilder builder = new BlobBuilder();
BigInteger value = new BigInteger("12398259028592293582039293420948023");
builder.AddSsh1BigIntBlob(value);
byte[] valueBytes = value.ToByteArrayUnsigned();
byte[] expected = new byte[valueBytes.Length + 2];
ushort size = (ushort)(value.BitLength);
expected[0] = (byte)((size >> 8) & 0xFF);
expected[1] = (byte)(size & 0xFF);
Array.Copy(valueBytes, 0, expected, 2, valueBytes.Length);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
}
public void SendMessageTest()
{
// TODO: Need to modify this test so that it does not use PageantAgent
const string messageValue = "junk";
var builder = new BlobBuilder();
builder.AddStringBlob(messageValue);
var messageBytes = builder.GetBlob();
using (var agent = new PageantAgent()) {
var client = new PageantClient();
var reply = client.SendMessage(messageBytes);
var replyParser = new BlobParser(reply);
var replyHeader = replyParser.ReadHeader();
Assert.That(replyHeader.Message, Is.EqualTo(Agent.Message.SSH_AGENT_FAILURE));
}
}
public void TestInsertHeader()
{
BlobBuilder builder = new BlobBuilder();
builder.InsertHeader(Agent.Message.SSH_AGENT_SUCCESS);
byte[] expected = { 0, 0, 0, 1, (byte)Agent.Message.SSH_AGENT_SUCCESS };
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
builder = new BlobBuilder();
int value1 = 12345;
builder.InsertHeader(Agent.Message.SSH_AGENT_SUCCESS, value1);
expected = new byte[9];
Array.Copy((5).ToBytes(), expected, 4);
expected[4] = (byte)Agent.Message.SSH_AGENT_SUCCESS;
Array.Copy(value1.ToBytes(), 0, expected, 5, 4);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
builder = new BlobBuilder();
byte[] value2 = { 1, 2, 3, 4, 5 };
builder.AddBytes(value2);
builder.InsertHeader(Agent.Message.SSH_AGENT_SUCCESS);
expected = new byte[5 + value2.Length];
int length = value2.Length + 1;
Array.Copy(length.ToBytes(), expected, 4);
expected[4] = (byte)Agent.Message.SSH_AGENT_SUCCESS;
Array.Copy(value2, 0, expected, 5, value2.Length);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
builder = new BlobBuilder();
builder.AddBytes(value2);
builder.InsertHeader(Agent.Message.SSH_AGENT_SUCCESS, value1);
expected = new byte[9 + value2.Length];
Array.Copy((5 + value2.Length).ToBytes(), expected, 4);
expected[4] = (byte)Agent.Message.SSH_AGENT_SUCCESS;
Array.Copy(value1.ToBytes(), 0, expected, 5, 4);
Array.Copy(value2, 0, expected, 9, value2.Length);
Assert.That(builder.GetBlob(), Is.EqualTo(expected));
}
