/// <summary>
/// 创建一个新的 HostEventArgs 对象.
/// </summary>
public HostEventArgs(Entity.Host host)
{
Host = host;
}
/// <summary>
/// 创建 <see cref="FileReceivedEventArgs" /> 的新实例
/// </summary>
public FileReceivedEventArgs(Entity.Host host, Entity.Message message, Entity.FileTaskInfo file)
{
Host = host;
Message = message;
File = file;
}
/// <summary>
/// 检测一个主机是否在黑名单中
/// </summary>
/// <param name="ip"></param>
/// <returns></returns>
public bool IsHostInBlockList(Entity.Host ip)
{
return(IsHostInBlockList(ip.HostSub.Ipv4Address.Address));
}
/// <summary>
/// 加密信息
/// </summary>
/// <param name="cmd">Commander</param>
/// <param name="e">事件参数</param>
void EncryptMessage(CommandExecutor cmd, Entity.MessageEventArgs e)
{
Entity.Host host = null;
if (cmd.LivedHost == null || (host = cmd.LivedHost.GetHost(e.Message.HostAddr.Address.ToString())) == null || !host.SupportEncrypt)
{
return;
}
if (host.PubKey == null)
{
//压入队列
if (host.QueuedMessage == null)
{
host.QueuedMessage = new Queue <FSLib.IPMessager.Entity.Message>();
}
host.QueuedMessage.Enqueue(e.Message);
e.IsHandled = true;
//请求公钥
GetPubKey(host);
}
else
{
if (e.Message.NormalMsgBytes == null)
{
System.Text.Encoding enc = e.Message.IsEncodingUnicode ? System.Text.Encoding.Unicode : System.Text.Encoding.Default;
if (!string.IsNullOrEmpty(e.Message.NormalMsg))
{
e.Message.NormalMsgBytes = enc.GetBytes(e.Message.NormalMsg);
}
else
{
e.Message.NormalMsgBytes = new byte[] { 0 }
};
if (!string.IsNullOrEmpty(e.Message.ExtendMessage))
{
e.Message.ExtendMessageBytes = enc.GetBytes(e.Message.ExtendMessage);
}
else
{
e.Message.ExtendMessageBytes = new byte[] { 0 }
};
}
if (e.Message.NormalMsgBytes == null)
{
return;
}
//加密
if (host.IsEncryptUseSmallKey)
{
//RSA512&&RC2
//暂时不支持。。。
//TODO:加密支持RC2
e.Message.IsEncrypt = false;
}
else
{
//RSA1024&BlowFish
byte[] key = new byte[8];
rcsp.GetBytes(key);
blowFish.Initialize(key, 0, key.Length);
byte[] content = new byte[(int)Math.Ceiling(e.Message.NormalMsgBytes.Length / 16.0) * 16]; //BlowFish加密必须是16字节的整数倍
e.Message.NormalMsgBytes.CopyTo(content, 0);
blowFish.Encrypt(content, 0, content, 0, content.Length);
blowFish.Invalidate();
//加密Key
RSAParameters p = new RSAParameters()
{
Modulus = host.PubKey,
Exponent = host.Exponent
};
RSA_1024_Encryptor.ImportParameters(p);
key = RSA_1024_Encryptor.Encrypt(key, false);
//Array.Reverse(key);
//组装消息
System.Text.StringBuilder sb = new StringBuilder();
sb.Append(EncryptNormalCapa.ToString("x"));
sb.Append(":");
sb.Append(key.ConvertToString());
sb.Append(":");
sb.Append(content.ConvertToString());
e.Message.NormalMsg = sb.ToString();
e.Message.NormalMsgBytes = null;
}
}
}
/// <summary>
/// 解析加密信息
/// </summary>
/// <param name="cmd">Commander</param>
/// <param name="e">事件参数</param>
void DecryptMessage(CommandExecutor cmd, Entity.MessageEventArgs e)
{
int index = 0;
int nextMatch = 0;
if ((nextMatch = e.Message.NormalMsg.IndexOf(':', index)) == -1)
{
return;
}
ulong capa;
if (!ulong.TryParse(e.Message.NormalMsg.Substring(index, nextMatch - index), System.Globalization.NumberStyles.AllowHexSpecifier, null, out capa))
{
return;
}
//查找密钥
index = nextMatch + 1;
if ((nextMatch = e.Message.NormalMsg.IndexOf(':', index)) == -1)
{
return;
}
byte[] buf = e.Message.NormalMsg.ConvertToBytes(index, nextMatch);
//消息加密内容
byte[] content = e.Message.NormalMsg.ConvertToBytes(nextMatch + 1, e.Message.NormalMsg.Length);
//解密密钥
bool isSuccess = false;
Define.Consts.Cmd_Encrpy_Option encOpt = (Define.Consts.Cmd_Encrpy_Option)capa;
if ((encOpt & Consts.Cmd_Encrpy_Option.RSA_1024) == Consts.Cmd_Encrpy_Option.RSA_1024)
{
//RSA1024
try
{
buf = this.RSA_1024_Decryptor.Decrypt(buf, false);
isSuccess = true;
}
catch (Exception) { }
}
else
{
try
{
buf = this.RSA_512_Decryptor.Decrypt(buf, false);
isSuccess = true;
}
catch (Exception) { }
}
if (!isSuccess)
{
if ((encOpt & Consts.Cmd_Encrpy_Option.RSA_1024) == Consts.Cmd_Encrpy_Option.RSA_1024)
{
cmd.SendCommand(e.Message.HostAddr, Consts.Commands.SendMsg, 0ul, "[AUTO=PUBKEY_EXP]\n\n公钥已经过期，无法解密信息，请刷新主机列表。", "", false, false, true, true, true, false, false, false);
e.IsHandled = true; //过滤掉消息
return;
}
else
{
cmd.SendCommand(e.Message.HostAddr, Consts.Commands.SendMsg, 0ul, "[AUTO=KEY_NOT_SUPPORT]\n\n您的客户端所使用的加密等级过低，无法支持。", "", false, false, true, true, true, false, false, false);
e.IsHandled = true; //过滤掉消息
return;
}
}
else
{
//解密文本
if ((encOpt & Consts.Cmd_Encrpy_Option.RC2_40) == Consts.Cmd_Encrpy_Option.RC2_40)
{
//TODO:RC2加密解决
}
else
{
//BlowFish
blowFish.Initialize(buf, 0, buf.Length);
blowFish.Decrypt(content, 0, content, 0, content.Length);
blowFish.Invalidate();
}
int endIndex = Array.FindIndex(content, (s) => s == 0);
if (endIndex == -1)
{
endIndex = content.Length;
}
if (Define.Consts.Check(e.Message.Options, Define.Consts.Cmd_Send_Option.Content_Unicode))
{
e.Message.NormalMsg = System.Text.Encoding.Unicode.GetString(content, 0, endIndex);
}
else
{
e.Message.NormalMsg = System.Text.Encoding.Default.GetString(content, 0, endIndex);
}
}
}
/// <summary>
/// 获得指定主机的公钥
/// </summary>
/// <param name="host">主机</param>
void GetPubKey(Host host)
{
if (!host.SupportEncrypt)
{
return;
}
Commander.SendCommand(host, Consts.Commands.GetPubKey, 0ul, EncryptCapa.ToString("x"), "", false, false, false, false, false, false, false, false);
}
/// <summary>
/// 加入一个主机到黑名单
/// </summary>
/// <param name="ip"></param>
public void BanHost(Entity.Host ip)
{
BanHost(ip.HostSub.Ipv4Address.Address);
}
