Merge remote-tracking branch 'TRX/master' into add_coins

fbsobreira · fbsobreira · commit 42c4bd4332a2 · 2020-02-15T07:29:55.000-03:00
diff --git a/README.md b/README.md
@@ -93,6 +93,8 @@ npm install getdelta/wallet-address-validator
 
 * ZenCash/ZEN, `'zencash'` or `'ZEN'`
 
+* TRON/TRX, `'tron'` or `'trx'`
+
 
 ### Usage example
 
diff --git a/dist/wallet-address-validator.js b/dist/wallet-address-validator.js
@@ -6456,6 +6456,7 @@ var jsSHA = require('jssha/src/sha256');
 var Blake256 = require('./blake256');
 var keccak256 = require('./sha3')['keccak256'];
 var Blake2B = require('./blake2b');
+var base58 = require('./base58');
 
 function numberToHex (number) {
     var hex = Math.round(number).toString(16);
@@ -6474,6 +6475,72 @@ function hexToUint8(hexString) {
     return new Uint8Array(arr);
 }
 
+/* Convert a byte to string */
+function byte2hexStr(byte) {
+    var hexByteMap = "0123456789ABCDEF";
+    var str = "";
+    str += hexByteMap.charAt(byte >> 4);
+    str += hexByteMap.charAt(byte & 0x0f);
+    return str;
+}
+
+//byteArray格式数据转为16进制的ASCII字符串。
+function byteArray2hexStr(byteArray) {
+    var str = "";
+    for (var i = 0; i < (byteArray.length - 1); i++) {
+        str += byte2hexStr(byteArray[i]);
+    }
+    str += byte2hexStr(byteArray[i]);
+    return str;
+}
+
+/* Check if a char is hex char */
+function isHexChar(c) {
+    if ((c >= 'A' && c <= 'F') ||
+        (c >= 'a' && c <= 'f') ||
+        (c >= '0' && c <= '9')) {
+        return 1;
+    }
+    return 0;
+}
+
+/* Convert a hex char to value */
+function hexChar2byte(c) {
+    var d = 0;
+    if (c >= 'A' && c <= 'F') {
+        d = c.charCodeAt(0) - 'A'.charCodeAt(0) + 10;
+    }
+    else if (c >= 'a' && c <= 'f') {
+        d = c.charCodeAt(0) - 'a'.charCodeAt(0) + 10;
+    }
+    else if (c >= '0' && c <= '9') {
+        d = c.charCodeAt(0) - '0'.charCodeAt(0);
+    }
+    return d;
+}
+
+function hexStr2byteArray(str) {
+    var byteArray = Array();
+    var d = 0;
+    var i = 0;
+    var j = 0;
+    var k = 0;
+
+    for (i = 0; i < str.length; i++) {
+        var c = str.charAt(i);
+        if (isHexChar(c)) {
+            d <<= 4;
+            d += hexChar2byte(c);
+            j++;
+            if (0 === (j % 2)) {
+                byteArray[k++] = d;
+                d = 0;
+            }
+        }
+    }
+    return byteArray;
+}
+
 module.exports = {
     toHex: function (arrayOfBytes) {
         var hex = '';
@@ -6504,7 +6571,10 @@ module.exports = {
     },
     keccak256Checksum: function (payload) {
         return keccak256(payload).toString().substr(0, 8);
-    }
+    },
+    base58: base58.decode,
+    byteArray2hexStr: byteArray2hexStr,
+    hexStr2byteArray: hexStr2byteArray
 };
 
 },{"./blake256":38,"./blake2b":39,"./sha3":43,"jssha/src/sha256":30}],45:[function(require,module,exports){
diff --git a/src/crypto/utils.js b/src/crypto/utils.js
@@ -2,6 +2,7 @@ var jsSHA = require('jssha/src/sha256');
 var Blake256 = require('./blake256');
 var keccak256 = require('./sha3')['keccak256'];
 var Blake2B = require('./blake2b');
+var base58 = require('./base58');
 
 function numberToHex (number) {
     var hex = Math.round(number).toString(16);
@@ -20,6 +21,72 @@ function hexToUint8(hexString) {
     return new Uint8Array(arr);
 }
 
+/* Convert a byte to string */
+function byte2hexStr(byte) {
+    var hexByteMap = "0123456789ABCDEF";
+    var str = "";
+    str += hexByteMap.charAt(byte >> 4);
+    str += hexByteMap.charAt(byte & 0x0f);
+    return str;
+}
+
+//byteArray格式数据转为16进制的ASCII字符串。
+function byteArray2hexStr(byteArray) {
+    var str = "";
+    for (var i = 0; i < (byteArray.length - 1); i++) {
+        str += byte2hexStr(byteArray[i]);
+    }
+    str += byte2hexStr(byteArray[i]);
+    return str;
+}
+
+/* Check if a char is hex char */
+function isHexChar(c) {
+    if ((c >= 'A' && c <= 'F') ||
+        (c >= 'a' && c <= 'f') ||
+        (c >= '0' && c <= '9')) {
+        return 1;
+    }
+    return 0;
+}
+
+/* Convert a hex char to value */
+function hexChar2byte(c) {
+    var d = 0;
+    if (c >= 'A' && c <= 'F') {
+        d = c.charCodeAt(0) - 'A'.charCodeAt(0) + 10;
+    }
+    else if (c >= 'a' && c <= 'f') {
+        d = c.charCodeAt(0) - 'a'.charCodeAt(0) + 10;
+    }
+    else if (c >= '0' && c <= '9') {
+        d = c.charCodeAt(0) - '0'.charCodeAt(0);
+    }
+    return d;
+}
+
+function hexStr2byteArray(str) {
+    var byteArray = Array();
+    var d = 0;
+    var i = 0;
+    var j = 0;
+    var k = 0;
+
+    for (i = 0; i < str.length; i++) {
+        var c = str.charAt(i);
+        if (isHexChar(c)) {
+            d <<= 4;
+            d += hexChar2byte(c);
+            j++;
+            if (0 === (j % 2)) {
+                byteArray[k++] = d;
+                d = 0;
+            }
+        }
+    }
+    return byteArray;
+}
+
 module.exports = {
     toHex: function (arrayOfBytes) {
         var hex = '';
@@ -50,5 +117,8 @@ module.exports = {
     },
     keccak256Checksum: function (payload) {
         return keccak256(payload).toString().substr(0, 8);
-    }
+    },
+    base58: base58.decode,
+    byteArray2hexStr: byteArray2hexStr,
+    hexStr2byteArray: hexStr2byteArray
 };
diff --git a/src/currencies.js b/src/currencies.js
@@ -5,6 +5,7 @@ var XMRValidator = require('./monero_validator');
 var NANOValidator = require('./nano_validator');
 var XLMValidator = require('./stellar_validator');
 var ADAValidator = require('./cardano_validator');
+var TRXValidator = require('./tron_validator');
 
 // defines P2PKH and P2SH address types for standard (prod) and testnet networks
 var CURRENCIES = [{
@@ -224,6 +225,11 @@ var CURRENCIES = [{
     name: 'vechain',
     symbol: 'vet',
     validator: ETHValidator,
+},{
+    name: 'tron',
+    symbol: 'trx',
+    addressTypes: {prod: [0x41], testnet: [0xa0]},
+    validator: TRXValidator
 }];
 
 
diff --git a/src/tron_validator.js b/src/tron_validator.js
@@ -0,0 +1,61 @@
+var cryptoUtils = require('./crypto/utils');
+
+function decodeBase58Address(base58Sting) {
+    if (typeof (base58Sting) !== 'string') {
+        return false;
+    }
+    if (base58Sting.length <= 4) {
+        return false;
+    }
+
+    try {
+        var address = cryptoUtils.base58(base58Sting);
+    } catch (e) {
+        return false
+    }
+
+    if (base58Sting.length <= 4) {
+        return false;
+    }
+    var len = address.length;
+    var offset = len - 4;
+    var checkSum = address.slice(offset);
+    address = address.slice(0, offset);
+    var hash0 = cryptoUtils.sha256(cryptoUtils.byteArray2hexStr(address));
+    var hash1 = cryptoUtils.hexStr2byteArray(cryptoUtils.sha256(hash0));
+    var checkSum1 = hash1.slice(0, 4);
+    if (checkSum[0] === checkSum1[0] && checkSum[1] === checkSum1[1] && checkSum[2]
+        === checkSum1[2] && checkSum[3] === checkSum1[3]
+    ) {
+        return address;
+    }
+
+    return false;
+}
+
+function getEnv(currency, networkType) {
+    var evn = networkType || 'prod';
+
+    if (evn !== 'prod' && evn !== 'testnet') evn = 'prod';
+
+    return currency.addressTypes[evn][0]
+}
+
+module.exports = {
+    /**
+     * tron address validation
+     */
+    isValidAddress: function (mainAddress, currency, networkType) {
+        var address = decodeBase58Address(mainAddress);
+
+        if (!address) {
+            return false;
+        }
+
+        if (address.length !== 21) {
+            return false;
+        }
+
+        return getEnv(currency, networkType) === address[0];
+    }
+};
diff --git a/test/wallet_address_validator.js b/test/wallet_address_validator.js
@@ -407,6 +407,11 @@ describe('WAValidator.validate()', function () {
             valid('0x6bf76c6f76e0971130e72055b6445ee10eabf48b', 'vechain');
             valid('0x1671a3e4A2519A653E66E827ef6eaE690ee86729', 'vechain');
         });
+
+        it('should return true for correct trx addresses', function () {
+            valid('TNDzfERDpxLDS2w1q6yaFC7pzqaSQ3Bg3r', 'trx');
+            valid('27bLJCYjbH6MT8DBF9xcrK6yZnm43vx7MNQ', 'trx', 'testnet');
+        });
     });
 
     describe('invalid results', function () {
