tera/Source/core/wallet.js

272 lines
8.1 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 fs = require('fs');
const crypto = require('crypto');
require("./library");
require("./crypto-library");
const WalletPath = "WALLET";
const DBRow = require("./db/db-row");
const CONFIG_NAME = GetDataPath(WalletPath + "/config.lst");
global.HIDDEN_ACC_PATH = GetDataPath(WalletPath + "/hidden.lst");
class CApp
{
constructor()
{
CheckCreateDir(GetDataPath(WalletPath))
var bReadOnly = (global.PROCESS_NAME !== "TX");
this.Password = ""
this.WalletOpen = undefined
var Params = LoadParams(CONFIG_NAME, undefined);
if(!Params)
{
Params = {}
if(global.TEST_NETWORK)
{
Params.Key = global.ARR_PUB_KEY[0]
}
else
{
Params.Key = GetHexFromArr(crypto.randomBytes(32))
}
Params.AccountMap = {}
Params.MiningAccount = 0
}
if(Params.MiningAccount)
global.GENERATE_BLOCK_ACCOUNT = Params.MiningAccount
this.AccountMap = Params.AccountMap
this.KeyPair = crypto.createECDH('secp256k1')
if(Params.Protect)
{
ToLog("Wallet protect by password")
this.KeyXOR = GetArrFromHex(Params.KeyXOR)
this.WalletOpen = false
this.SetPrivateKey(Params.PubKey)
}
else
{
this.SetPrivateKey(Params.Key)
}
}
SetMiningAccount(Account)
{
global.GENERATE_BLOCK_ACCOUNT = Account
this.SaveWallet()
}
AddTransaction(Tr)
{
if(!global.TX_PROCESS.Worker)
return 0;
var StrHex = GetHexFromArr(sha3(Tr.body));
global.TX_PROCESS.Worker.send({cmd:"FindTX", TX:StrHex})
return SERVER.AddTransaction(Tr, 1);
}
SetPrivateKey(KeyStr, bSetNew)
{
var bGo = 1;
if(this.WalletOpen === false)
{
bGo = 0
}
if(KeyStr && KeyStr.length === 64 && bGo)
{
this.KeyPair.setPrivateKey(GetArr32FromHex(KeyStr))
this.KeyPair.PubKeyArr = this.KeyPair.getPublicKey('', 'compressed')
this.KeyPair.PubKeyStr = GetHexFromArr(this.KeyPair.PubKeyArr)
this.KeyPair.PrivKeyStr = KeyStr.toUpperCase()
this.KeyPair.addrArr = this.KeyPair.PubKeyArr.slice(1)
this.KeyPair.addrStr = GetHexAddresFromPublicKey(this.KeyPair.addrArr)
this.KeyPair.addr = this.KeyPair.addrArr
this.KeyPair.WasInit = 1
this.PubKeyArr = this.KeyPair.PubKeyArr
}
else
{
this.KeyPair.WasInit = 0
if(KeyStr)
{
this.PubKeyArr = GetArrFromHex(KeyStr)
this.KeyPair.PubKeyStr = GetHexFromArr(this.PubKeyArr)
}
else
{
this.PubKeyArr = []
this.KeyPair.PubKeyStr = ""
}
this.KeyPair.PrivKeyStr = ""
}
if(bSetNew)
{
this.AccountMap = {}
}
this.FindMyAccounts(0)
if(bGo)
this.SaveWallet()
}
CloseWallet()
{
this.Password = ""
this.WalletOpen = false
this.KeyPair = crypto.createECDH('secp256k1')
this.SetPrivateKey(GetHexFromArr(this.PubKeyArr), false)
ToLogClient("Wallet close")
return 1;
}
OpenWallet(StrPassword)
{
if(this.WalletOpen !== false)
{
ToLogClient("Wallet was open")
}
var Hash = this.HashProtect(StrPassword);
var TestPrivKey = this.XORHash(this.KeyXOR, Hash, 32);
if(!IsZeroArr(TestPrivKey))
{
this.KeyPair.setPrivateKey(Buffer.from(TestPrivKey))
var TestPubKey = this.KeyPair.getPublicKey('', 'compressed');
if(CompareArr(TestPubKey, this.PubKeyArr) !== 0)
{
ToLogClient("Wrong password")
return 0;
}
this.Password = StrPassword
this.WalletOpen = true
this.SetPrivateKey(GetHexFromArr(TestPrivKey), false)
}
else
{
this.Password = StrPassword
this.WalletOpen = true
this.SetPrivateKey(GetHexFromArr(this.PubKeyArr), false)
}
ToLogClient("Wallet open")
return 1;
}
SetPasswordNew(StrPassword)
{
if(this.WalletOpen === false)
{
ToLogClient("Wallet is close by password")
return ;
}
this.Password = StrPassword
if(StrPassword)
this.WalletOpen = true
else
this.WalletOpen = undefined
this.SaveWallet()
}
HashProtect(Str)
{
var arr = shaarr(Str);
for(var i = 0; i < 10000; i++)
{
arr = shaarr(arr)
}
return arr;
}
XORHash(arr1, arr2, length)
{
var arr3 = [];
for(var i = 0; i < length; i++)
{
arr3[i] = arr1[i] ^ arr2[i]
}
return arr3;
}
SaveWallet()
{
if(this.WalletOpen === false)
{
return ;
}
var Params = {};
if(this.Password)
{
Params.Protect = true
var Hash = this.HashProtect(this.Password);
if(this.KeyPair.WasInit)
{
Params.KeyXOR = GetHexFromArr(this.XORHash(this.KeyPair.getPrivateKey(), Hash, 32))
}
else
{
var Key2 = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0];
Params.KeyXOR = GetHexFromArr(this.XORHash(Key2, Hash, 32))
}
Params.PubKey = GetHexFromArr(this.PubKeyArr)
this.KeyXOR = GetArrFromHex(Params.KeyXOR)
}
else
{
if(this.KeyPair.WasInit)
Params.Key = this.KeyPair.PrivKeyStr
else
Params.Key = GetHexFromArr(this.PubKeyArr)
}
Params.AccountMap = this.AccountMap
Params.MiningAccount = global.GENERATE_BLOCK_ACCOUNT
SaveParams(CONFIG_NAME, Params)
}
OnCreateAccount(Data)
{
this.AccountMap[Data.Num] = 0
}
FindMyAccounts(bClean)
{
if(IsZeroArr(this.PubKeyArr))
return ;
var HiddenMap = LoadParams(HIDDEN_ACC_PATH, {});
if(bClean)
this.AccountMap = {}
DApps.Accounts.FindAccounts([this.PubKeyArr], this.AccountMap, HiddenMap, 0)
}
GetAccountKey(Num)
{
if(this.KeyPair.WasInit && global.TestTestWaletMode)
{
}
return this.KeyPair;
}
GetPrivateKey(Num)
{
if(!this.KeyPair.WasInit)
return [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0];
var KeyPair;
if(Num)
{
KeyPair = this.GetAccountKey(Num)
}
else
{
KeyPair = this.KeyPair
}
return KeyPair.getPrivateKey();
}
GetSignFromArr(Arr, Num)
{
if(!this.KeyPair.WasInit)
return "00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000";
var PrivKey = this.GetPrivateKey(Num);
var sigObj = secp256k1.sign(SHA3BUF(Arr), Buffer.from(PrivKey));
return GetHexFromArr(sigObj.signature);
}
GetSignTransaction(TR)
{
if(!this.KeyPair.WasInit)
return "00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000";
var PrivKey = this.GetPrivateKey(this.AccountMap[TR.FromID]);
var Arr = DApps.Accounts.GetSignTransferTx(TR, PrivKey);
return GetHexFromArr(Arr);
}
};
global.WALLET = new CApp;