public static UdpTrackerMessage DecodeMessage(byte[] buffer, int offset, int count, MessageType type)
{
UdpTrackerMessage m = null;
int action = type == MessageType.Request ? ReadInt(buffer, offset + 8) : ReadInt(buffer, offset);
switch (action)
{
case 0:
if (type == MessageType.Request)
m = new ConnectMessage();
else
m = new ConnectResponseMessage();
break;
case 1:
if (type == MessageType.Request)
m = new AnnounceMessage();
else
m = new AnnounceResponseMessage();
break;
case 2:
if (type == MessageType.Request)
m = new ScrapeMessage();
else
m = new ScrapeResponseMessage();
break;
case 3:
m = new ErrorMessage();
break;
default:
throw new ProtocolException(string.Format("Invalid udp message received: {0}", buffer[offset]));
}
try
{
m.Decode(buffer, offset, count);
}
catch
{
m = new ErrorMessage(0, "Couldn't decode the tracker response");
}
return m;
}
protected virtual void ReceiveConnect(ConnectMessage connectMessage)
{
UdpTrackerMessage m = new ConnectResponseMessage(connectMessage.TransactionId, CreateConnectionID());
var data = m.Encode();
listener.Send(data, data.Length, endpoint);
}
public void ConnectTest()
{
ConnectMessage m = new ConnectMessage();
ConnectMessage d = (ConnectMessage)UdpTrackerMessage.DecodeMessage(m.Encode(), 0, m.ByteLength, MessageType.Request);
Check(m, MessageType.Request);
Assert.AreEqual(0, m.Action, "#0");
Assert.AreEqual(m.Action, d.Action, "#1");
Assert.AreEqual(m.ConnectionId, d.ConnectionId, "#2");
Assert.AreEqual(m.TransactionId, d.TransactionId, "#3");
Assert.IsTrue(Toolbox.ByteMatch(m.Encode(), d.Encode()), "#4");
}
protected override void ReceiveConnect(ConnectMessage connectMessage)
{
if (!IgnoreConnects)
base.ReceiveConnect(connectMessage);
}
public void ConnectMessageTest()
{
var m = new ConnectMessage();
var d =
(ConnectMessage) UdpTrackerMessage.DecodeMessage(m.Encode(), 0, m.ByteLength, MessageType.Request);
Check(m, MessageType.Request);
Assert.Equal(0, m.Action);
Assert.Equal(m.Action, d.Action);
Assert.Equal(m.ConnectionId, d.ConnectionId);
Assert.Equal(m.TransactionId, d.TransactionId);
Assert.True(Toolbox.ByteMatch(m.Encode(), d.Encode()));
}
