Web钱包私钥导出新旧两种地址格式（加班到深夜修改的一个功能点）
前言：
这个问题是中午的时候，被用户反馈，当时还不觉得有问题。
调到最后发现一个小坑是：私钥地址，带0x和不带0x是有区别的。我们的Web钱包不要带0x，否则会还原出另一个新地址。
1. jssdk修改
（1）用私钥还原地址
/** 新地址
* Get address from private key.
* @param {string} privateKeyHex the private key hexstring
* @param {string} prefix address prefix
* @return {string} address
*/
export const getAddressFromPrivateKey = (privateKeyHex, prefix) => {
return getAddressFromPubKey(getPubKeyHexFromPrivateKey(privateKeyHex), prefix)
}
// 旧地址 Legacy
export const getAddressFromPrivateKeyLegacy = (privateKeyHex, prefix) => {
return getAddressFromPublicKeyLegacy(getPubKeyHexFromPrivateKey(privateKeyHex), prefix)
}
/** 新方法
* Gets address from pubKey with hex format.
* @param {string} publicKey publicKey hexstring
* @param {string} prefix address prefix
* @return {string} address
*/
export const getAddressFromPubKey = (publicKey, prefix) => {
publicKey = publicKey.slice(0, 2) === '0x' ? publicKey.slice(2) : publicKey
const publicKey1 = Buffer.from(publicKey, 'hex')
publicKey = Buffer.from(secp256k1.publicKeyConvert(new Uint8Array(publicKey1), false)).slice(1)
const hash = createKeccakHash('keccak256').update(publicKey).digest()
return encodeAddressToBech32(hash.slice(-20).toString('hex'), prefix)
}
// 旧的方法Legacy
export const getAddressFromPublicKeyLegacy = (publicKeyHex, prefix) => {
// const pubKey = ec.keyFromPublic(publicKeyHex, "hex")
// const pubPoint = pubKey.getPublic()
// const compressed = pubPoint.encodeCompressed()
// const hexed = ab2hexstring(compressed)
const hash = sha256ripemd160(publicKeyHex) // https://git.io/fAn8N
const address = encodeAddress(hash, prefix)
return address
}
export const encodeAddress = (value, prefix = "ex", type = "hex") => {
const words = bech32.toWords(Buffer.from(value, type))
return bech32.encode(prefix, words)
}
（2）新旧两种地址在转账交易类型
设置账户信息，也要体现两种地址
/**
* @param {string} privateKey
* @param {string} prefix
* @return {OKEXChainClient}
*/
async setAccountInfo(privateKey, prefix = "ex", isPrivatekeyOld = 0) {
if(!privateKey) {
const address = await wallet.getAddress();
if (!address) throw new Error("invalid privateKey: " + privateKey)
await this.setAccountInfoByWallet(address);
return this;
}
if (privateKey !== this.privateKey) {
let address = crypto.getAddressFromPrivateKey(privateKey, prefix)
if (isPrivatekeyOld) {
address = crypto.getAddressFromPrivateKeyLegacy(privateKey, prefix)
}
if (!address) throw new Error("invalid privateKey: " + privateKey)
if (address === this.address) return this
this.privateKey = privateKey
this.address = address
const data = await this.getAccount(address)
this.account_number = this.getAccountNumberFromAccountInfo(data)
}
return this
}
构造两种交易："tendermint/PubKeySecp256k1” 和"ethermint/PubKeyEthSecp256k1"，及两种不同的签名。
// PrivatekeyHex
if (typeof privateKeyHexOrSigner === 'string') {
const jsonStr = JSON.stringify(signMsg)
const signBytes = Buffer.from(jsonStr)
const privateKey = Buffer.from(privateKeyHexOrSigner, "hex")
const signature = isPrivatekeyOldAddress ? crypto.signPrivateKeyOldAddress(signBytes.toString("hex"), privateKey)
: crypto.sign(signBytes.toString("hex"), privateKey)
const pubKey = crypto.encodePubKeyToCompressedBuffer(crypto.getPubKeyFromPrivateKey(privateKey))
signatures = [{
pub_key: {
type: isPrivatekeyOldAddress ? "tendermint/PubKeySecp256k1" : "ethermint/PubKeyEthSecp256k1",
value:pubKey},
signature: signature,
}]
}
/** 旧签名
* Generates a signature (64 byte <r,s>) for a transaction based on given private key.
* @param {string} signBytesHex - Unsigned transaction sign bytes hexstring.
* @param {string | Buffer} privateKey - The private key.
* @return {Buffer} Signature. Does not include tx.
*/
export const signPrivateKeyOldAddress = (signBytesHex, privateKey) => {
const msgHash = sha256(signBytesHex)
const msgHashHex = Buffer.from(msgHash, "hex")
const signature = ecc.sign(msgHashHex, Buffer.from(privateKey, "hex")) // enc ignored if buffer
return signature
}
/** 新签名
* Sign msg with privateKey and Msg in hex format.
* @param {string} msgHex msg in hex format.
* @param {string} privateKey - The private key in hex format.
* @return {Buffer} Signature.
*/
export const sign = (msgHex, privateKey) => {
const msg = Buffer.from(msgHex, "hex")
const msgHash = createKeccakHash('keccak256').update(msg).digest()
const signature = ecc.sign(msgHash, Buffer.from(privateKey, "hex")) // enc ignored if buffer
return signature
}
2. Web前端修改
（1）还原地址处
validatePrivateKey = async () => {
try {
let { privateKey, password } = this.state;
privateKey = privateKey.replace(/^(0x)/, '');
if (!/^[\d|a-f]{64}$/.test(privateKey)) {
throw new Error('not pass the reg');
}
const oldAddress = crypto.getAddressFromPrivateKeyLegacy(privateKey);
const newAddress = crypto.getAddressFromPrivateKey(privateKey);
const oldBalance = await this.fetchAcount(oldAddress);
const newBalance = await this.fetchAcount(newAddress);
this.setState({
isValidatedPrivateKey: true,
buttonLoading: false,
isNone: false,
oldAddress,
newAddress: crypto.convertBech32ToHex(newAddress)[0],
oldBalance: calc.showFloorTruncation(oldBalance),
newBalance: calc.showFloorTruncation(newBalance),
step: 2,
});
const keyStore = crypto.generateKeyStore(privateKey, password);
walletUtil.setUserInSessionStroage(privateKey, keyStore);
this.props.commonAction.setPrivateKey(privateKey);
// util.go(DesktopTypeMenu.current ? DesktopTypeMenu.current.url : void 0);
} catch (e) {
this.setState({
isValidatedPrivateKey: false,
buttonLoading: false,
isNone: false,
});
}
};
// 保留最后的尊严，哈哈哈
selectPathType = (type) => {
window.localStorage.setItem(env.envConfig.privatekeyPathType, type);
if (type === 'old') {
window.localStorage.setItem(
env.envConfig.privatekeyOldPathPrivatekey,
this.state.privateKey
);
} else {
window.localStorage.setItem(
env.envConfig.privatekeyOldPathPrivatekey,
''
);
}
this.setState({
pathType: type,
});
};
账户部分
setUserInSessionStroageByWalletConnect(addr) {
const user = {
addr,
};
window.localStorage.setItem(env.envConfig.dexUser, JSON.stringify(user));
window.OK_GLOBAL.senderAddr = addr;
window.OK_GLOBAL.isLogin = true;
},
// 重点是
setUserInSessionStroage(privateKey, keyStore) {
const privatekeyPathType = window.localStorage.getItem(
env.envConfig.privatekeyPathType
);
let addr = crypto.getAddressFromPrivateKey(privateKey);
if (privatekeyPathType === 'old') {
addr = crypto.getAddressFromPrivateKeyLegacy(privateKey);
}
const user = {
addr,
info: keyStore,
};
window.localStorage.setItem(env.envConfig.dexUser, JSON.stringify(user));
window.OK_GLOBAL.senderAddr = addr;
window.OK_GLOBAL.generalAddr = crypto.convertBech32ToHex(addr)[0];
window.OK_GLOBAL.isLogin = true;
},
还有URL部分，涉及到跨域。
export const DEFAULT = window.okGlobal.mainDomain || 'https://www.okx.com';
export function getApiUrl(apiUrl = DEFAULT) {
const protocol = window.location.protocol;
if (/file/.test(protocol) || window.location.hostname === '127.0.0.1')
return apiUrl;
return `${window.location.protocol}//${window.location.host}`;
}
export function getCurrentApiUrl() {
// let url = getApiUrl();
let url = 'https://www.okx.com';
const currentNode = storage.get('currentNode');
if (currentNode && currentNode.httpUrl) url = currentNode.httpUrl;
return url;
}
（2）发送交易处
const isPrivatekeyOld = oldPrivate ? 1 : 0;
okexchainClient
.setAccountInfo(privateKey, 'ex', isPrivatekeyOld)
.then(() => {
okexchainClient
.sendSendTransaction(
address,
amountStr,
symbol,
note,
null,
isPrivatekeyOld
)
.then((res) => {
if (res.result.code) {
this._transferErr(
toLocale(`error.code.${res.result.code}`) ||
toLocale('trans_fail')
);
} else {
this._transferSuccess(onSuccess);
}
})
.catch(() => {
this._transferErr(toLocale('trans_fail'));
});
});
