tera/Source/core/transfer-msg.js
2019-07-12 20:45:46 +08:00

259 lines
7.6 KiB
JavaScript

/*
* @project: TERA
* @version: Development (beta)
* @license: MIT (not for evil)
* @copyright: Yuriy Ivanov (Vtools) 2017-2019 [progr76@gmail.com]
* Web: https://terafoundation.org
* Twitter: https://twitter.com/terafoundation
* Telegram: https://t.me/terafoundation
*/
"use strict";
const MAX_MESSAGE_COUNT = 1000;
module.exports = class CMessages extends require("./transaction-validator")
{
constructor(SetKeyPair, RunIP, RunPort, UseRNDHeader, bVirtual)
{
super(SetKeyPair, RunIP, RunPort, UseRNDHeader, bVirtual)
this.MemPoolMsg = []
for(var i = 0; i <= MAX_LEVEL_SPECIALIZATION; i++)
this.MemPoolMsg[i] = new RBTree(CompareItemTimePow)
}
AddMsgToQuote(Msg)
{
var Tree = this.MemPoolMsg[Msg.Level];
if(Tree)
{
if(Tree.insert(Msg))
{
if(Tree.size > MEM_POOL_MSG_COUNT)
{
var maxitem = Tree.max();
Tree.remove(maxitem)
if(maxitem === Msg)
return 0;
}
return 1;
}
else
{
return 3;
}
}
return 0;
}
IsValidMsg(Msg)
{
this.CheckCreateMsgHASH(Msg)
if(Msg.power < MIN_POWER_POW_MSG)
return - 1;
if(Msg.time > this.CurrentBlockNum)
return - 1;
return 1;
}
CheckCreateMsgHASH(Msg)
{
if(!Msg.HashPow)
{
Msg.HASH = sha3(Msg.body)
Msg.HashPow = GetHashWithValues(Msg.HASH, Msg.nonce, Msg.time)
Msg.power = GetPowPower(Msg.HashPow)
Msg.TimePow = Msg.time + Msg.power - Math.log2(Msg.body.length / 128)
Msg.Level = AddrLevelArr(this.addrArr, Msg.addrArr)
if(Msg.Level >= MAX_LEVEL_SPECIALIZATION)
Msg.Level = MAX_LEVEL_SPECIALIZATION
}
}
CreateMsgFromBody(Body, ToAddr)
{
var HASH = sha3(Body);
var Msg = {HASH:HASH, body:Body, addrArr:ToAddr, nonce:CreateNoncePOWExtern(HASH, this.CurrentBlockNum, 3 * (1 << MIN_POWER_POW_MSG)),
time:this.CurrentBlockNum, };
this.CheckCreateMsgHASH(Msg)
return Msg;
}
SendMessage(Body, ToAddr)
{
var Msg = this.CreateMsgFromBody(Body, ToAddr);
this.SendMessageNext(Msg)
}
SendMessageNext(Msg)
{
var CountNodes = 3;
var LevelStart = Msg.Level;
if(CompareArr(this.addrArr, Msg.addrArr) === 0)
return false;
for(var L = LevelStart; L >= 0; L--)
if(this.LevelNodes[L] && this.LevelNodes[L].length)
{
var arr = this.LevelNodes[L];
for(var j = 0; arr && j < arr.length; j++)
{
var Node = arr[j];
this.SendF(Node, {"Method":"MESSAGE", "Data":{Arr:[Msg]}})
CountNodes--
if(CountNodes <= 0)
break;
}
}
return true;
}
static
MESSAGE_F()
{
return "{Arr:[{addrArr:hash,body:tr,nonce:uint,time:uint}]}";
}
MESSAGE(Info, CurTime)
{
var Data = this.DataFromF(Info);
var arr = Data.Arr;
for(var i = 0; i < arr.length; i++)
{
var Msg = arr[i];
if(this.IsValidMsg(Msg))
{
if(CompareArr(this.addrArr, Msg.addrArr) === 0)
{
var App = DAppByType[Msg.body[0]];
if(App)
{
App.OnMessage(Msg, BlockNum, i)
}
}
else
{
if(this.AddMsgToQuote(Msg) === 1)
{
this.SendMessageNext(Msg)
}
}
}
}
}
SendGetMessage(Node)
{
var Context = {"SendGetMessage":1};
this.Send(Node, {"Method":"GETMESSAGE", "Context":Context, "Data":undefined})
}
GETMESSAGE(Info, CurTime)
{
var arr = [];
var BufLength = 300;
var Level = AddrLevelArr(this.addrArr, Info.Node.addrArr);
var Tree = this.MemPoolMsg[Level];
if(Tree)
{
var it = Tree.iterator(), Item;
while((Item = it.next()) !== null)
{
if(arr.length >= MAX_MESSAGE_COUNT)
break;
arr.push(Item)
BufLength += Item.body.length + 50
}
}
this.SendF(Info.Node, {"Method":"MESSAGE", "Context":Info.Context, "Data":{Arr:arr}}, BufLength)
}
AddTransaction(Tr, ToAll)
{
Tr.ToAll = ToAll
var Res = this.IsValidTransaction(Tr, this.CurrentBlockNum);
if(Res <= 0 && Res !== - 3)
return Res;
if(Tr.num < this.CurrentBlockNum)
return - 3;
var delta = Tr.num - this.CurrentBlockNum;
if(delta > 3)
{
if(delta < 15)
{
let TR = Tr;
let SELF = this;
setTimeout(function ()
{
var Res = SELF.AddTransaction(TR, TR.ToAll);
if(TR.ToAll)
ToLogClient("#3 Added " + TrName(TR) + " for block: " + TR.num + " on timer Res=" + Res)
}, (delta - 3) * 1000)
if(Tr.ToAll)
ToLogClient("#2 Added " + TrName(Tr) + " for block: " + Tr.num + " to timer. Send transaction after " + (delta - 3) + " sec")
return 4;
}
return - 3;
}
var Block = this.GetBlockContext(Tr.num);
if(!Block)
return - 5;
if(Block.Active)
{
Res = - 3
}
else
{
Res = this.AddTrToBlockQuote(Block, Tr)
if(Tr.ToAll)
this.SendTransaction(Tr)
}
ToLogContext("#1 Add " + TrName(Tr) + " for Block: " + Tr.num + " Res=" + Res)
return Res;
}
SendTransaction(Tr)
{
if(!Tr.ToAll)
return ;
var CurTime = GetCurrentTime(0) - 0;
var Count;
if(GrayConnect())
Count = Math.trunc(MAX_GRAY_CONNECTIONS_TO_SERVER / 2)
else
Count = Math.min(this.ActualNodes.size, 16)
if(Count < 2)
Count = 2
var ArrNodes = this.GetActualNodes();
for(var i = 0; i < ArrNodes.length; i++)
{
var Node = ArrNodes[i];
if(!Node)
continue;
if(Node.TaskLastSend)
{
var Delta = CurTime - Node.TaskLastSend;
if(Delta < global.PERIOD_GET_BLOCK || Node.StopGetBlock)
{
continue;
}
}
Node.TaskLastSend = CurTime
this.SendF(Node, {"Method":"TRANSACTION", "Data":Tr}, Tr.body.length + 1000)
ToLogContext("Send " + TrName(Tr) + " to " + NodeName(Node))
Count--
if(Count <= 0)
break;
}
}
static
TRANSACTION_F()
{
return "{body:tr}";
}
TRANSACTION(Info, CurTime)
{
var Tr = this.DataFromF(Info);
ToLogContext("Receive " + TrName(Tr) + " from " + NodeName(Info.Node))
this.AddTransaction(Tr, 0)
}
};
function ToLogContext(Str)
{
};
function TrName(Tr)
{
if(!Tr.HASH)
SERVER.CheckCreateTransactionObject(Tr);
var Str = GetHexFromArr(Tr.HASH);
return "Tx:" + Str.substr(0, 8);
};
global.TrName = TrName;