protected override Task OnMessageReceivedAsync(IWebSocketContext context,
byte[] rxBuffer,
IWebSocketReceiveResult rxResult)
{
var user_id = context.HttpContextId;
TripleDESCryptoServiceProvider crypto;
UserProfile profile;
if (API.TripleDES.UserDict.TryGetValue(user_id, out profile))
{
// ----------------------------------------
crypto = profile.crypto;
MsgRequest req = new MsgRequest(crypto, rxBuffer);
if (req.MsgId != profile.next_msg_id)
{
return(Task.CompletedTask);
}
// ----------------------------------------
if (req.Type == "str")
{
string content = API.Encoding.ByteArrayToString(req.Data);
if (API.Keyboard.EnableAutoTyping)
{
API.Keyboard.SendString(content);
}
if (API.Keyboard.EnableClipboard)
{
API.Keyboard.SetClipboard(content);
}
}
// ----------------------------------------
string pongEncrypted = req.GetResponse(profile);
SendToOthersAsync(context, req.GetRepeat(profile));
return(SendAsync(context, pongEncrypted));
}
else
{
// ----------------------------------------
byte[] bytes = API.RSA.RSADecrypt(rxBuffer);
if (bytes == null)
{
return(Task.CompletedTask);
}
string[] splitted = API.Encoding.ByteArrayToString(bytes).Split(';');
string type = splitted[0], key = splitted[1], iv = splitted[2];
// ----------------------------------------
if (type != "reg" || key.Length != 32 || iv.Length != 16)
{
return(Task.CompletedTask);
}
crypto = API.TripleDES.createDESCrypto(key, iv);
string next_msg_id = API.Util.GetRandomHexString(4);
string helloRaw = "acc;" + user_id + ";" + next_msg_id;
string helloEncrypted = API.Util.StringToEncryptHex(helloRaw, crypto);
API.TripleDES.UserDict.Add(user_id, new UserProfile()
{
userid = user_id, crypto = crypto, next_msg_id = next_msg_id
});
return(SendAsync(context, helloEncrypted));
}
}
