增加支持trx doge dash bch bsv导入地址

GalaxySciTech · Mar 19, 2021 · d0f2e7e · d0f2e7e
1 parent ba574fa
commit d0f2e7e
Showing 1 changed file with 49 additions and 39 deletions.
diff --git a/src/main/java/org/consenlabs/tokencore/wallet/keystore/V3Keystore.java b/src/main/java/org/consenlabs/tokencore/wallet/keystore/V3Keystore.java
@@ -8,6 +8,7 @@
 import org.consenlabs.tokencore.wallet.address.BitcoinAddressCreator;
 import org.consenlabs.tokencore.wallet.address.EthereumAddressCreator;
 import org.consenlabs.tokencore.wallet.address.SegWitBitcoinAddressCreator;
+import org.consenlabs.tokencore.wallet.address.TronAddressCreator;
 import org.consenlabs.tokencore.wallet.model.ChainType;
 import org.consenlabs.tokencore.wallet.model.Metadata;
 import org.consenlabs.tokencore.wallet.model.TokenException;
@@ -21,50 +22,59 @@
  */
 
 public class V3Keystore extends IMTKeystore implements ExportableKeystore {
-  public static final int VERSION = 3;
+    public static final int VERSION = 3;
 
-  public V3Keystore() {
-  }
+    public V3Keystore() {
+    }
 
-  public static V3Keystore create(Metadata metadata, String password, String prvKeyHex) {
-    return new V3Keystore(metadata, password, prvKeyHex, "");
-  }
+    public static V3Keystore create(Metadata metadata, String password, String prvKeyHex) {
+        return new V3Keystore(metadata, password, prvKeyHex, "");
+    }
 
-  public V3Keystore(Metadata metadata, String password, String prvKeyHex, String id) {
-    byte[] prvKeyBytes;
-    if (metadata.getChainType().equals(ChainType.BITCOIN)||metadata.getChainType().equals(ChainType.LITECOIN)) {
-      NetworkParameters network = MetaUtil.getNetWork(metadata);
-      prvKeyBytes = prvKeyHex.getBytes();
-      new WIFValidator(prvKeyHex, network).validate();
-      if (Metadata.P2WPKH.equals(metadata.getSegWit())) {
-        this.address = new SegWitBitcoinAddressCreator(network).fromPrivateKey(prvKeyHex);
-      } else {
-        this.address = new BitcoinAddressCreator(network).fromPrivateKey(prvKeyHex);
-      }
-    } else if (metadata.getChainType().equals(ChainType.ETHEREUM)) {
-      prvKeyBytes = NumericUtil.hexToBytes(prvKeyHex);
-      new PrivateKeyValidator(prvKeyHex).validate();
-      this.address = new EthereumAddressCreator().fromPrivateKey(prvKeyBytes);
-    } else {
-      throw new TokenException("Can't init eos keystore in this constructor");
+    public V3Keystore(Metadata metadata, String password, String prvKeyHex, String id) {
+        byte[] prvKeyBytes;
+        if (metadata.getChainType().equals(ChainType.BITCOIN) ||
+                metadata.getChainType().equals(ChainType.LITECOIN) ||
+                metadata.getChainType().equals(ChainType.DASH) ||
+                metadata.getChainType().equals(ChainType.DOGECOIN) ||
+                metadata.getChainType().equals(ChainType.BITCOINCASH) ||
+                metadata.getChainType().equals(ChainType.BITCOINSV)) {
+            NetworkParameters network = MetaUtil.getNetWork(metadata);
+            prvKeyBytes = prvKeyHex.getBytes();
+            new WIFValidator(prvKeyHex, network).validate();
+            if (Metadata.P2WPKH.equals(metadata.getSegWit())) {
+                this.address = new SegWitBitcoinAddressCreator(network).fromPrivateKey(prvKeyHex);
+            } else {
+                this.address = new BitcoinAddressCreator(network).fromPrivateKey(prvKeyHex);
+            }
+        } else if (metadata.getChainType().equals(ChainType.ETHEREUM)) {
+            prvKeyBytes = NumericUtil.hexToBytes(prvKeyHex);
+            new PrivateKeyValidator(prvKeyHex).validate();
+            this.address = new EthereumAddressCreator().fromPrivateKey(prvKeyBytes);
+        } else if (metadata.getChainType().equals(ChainType.TRON)) {
+            prvKeyBytes = NumericUtil.hexToBytes(prvKeyHex);
+            new PrivateKeyValidator(prvKeyHex).validate();
+            this.address = new TronAddressCreator().fromPrivateKey(prvKeyBytes);
+        }else {
+            throw new TokenException("Can't init eos keystore in this constructor");
+        }
+        this.crypto = Crypto.createPBKDF2Crypto(password, prvKeyBytes);
+        metadata.setWalletType(Metadata.V3);
+        this.metadata = metadata;
+        this.version = VERSION;
+        this.id = Strings.isNullOrEmpty(id) ? UUID.randomUUID().toString() : id;
     }
-    this.crypto = Crypto.createPBKDF2Crypto(password, prvKeyBytes);
-    metadata.setWalletType(Metadata.V3);
-    this.metadata = metadata;
-    this.version = VERSION;
-    this.id = Strings.isNullOrEmpty(id) ? UUID.randomUUID().toString() : id;
-  }
 
 
-  @Override
-  public Keystore changePassword(String oldPassword, String newPassword) {
-    byte[] decrypted = this.crypto.decrypt(oldPassword);
-    String prvKeyHex;
-    if (Metadata.FROM_WIF.equals(getMetadata().getSource())) {
-      prvKeyHex = new String(decrypted);
-    } else {
-      prvKeyHex = NumericUtil.bytesToHex(decrypted);
+    @Override
+    public Keystore changePassword(String oldPassword, String newPassword) {
+        byte[] decrypted = this.crypto.decrypt(oldPassword);
+        String prvKeyHex;
+        if (Metadata.FROM_WIF.equals(getMetadata().getSource())) {
+            prvKeyHex = new String(decrypted);
+        } else {
+            prvKeyHex = NumericUtil.bytesToHex(decrypted);
+        }
+        return new V3Keystore(metadata, newPassword, prvKeyHex, this.id);
     }
-    return new V3Keystore(metadata, newPassword, prvKeyHex, this.id);
-  }
 }