lil-org/tokenary
1
1
Fork 0
mirror of https://github.com/lil-org/tokenary.git synced 2024-12-15 15:05:15 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update signing messages

Browse Source
This commit is contained in:
Ivan Grachyov 2021-08-13 22:03:04 +03:00
parent 82b02997dd
commit 8740b4a7ef
2 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Encrypted Ink/Ethereum/Ethereum.swift
View File

@ -19,19 +19,20 @@ struct Ethereum {
static let shared = Ethereum()
func sign(message: String, wallet: InkWallet) throws -> String {
return try sign(message: message, wallet: wallet, addPrefix: false)
func sign(data: Data, wallet: InkWallet) throws -> String {
return try sign(data: data, wallet: wallet, addPrefix: false)
}
func signPersonal(message: String, wallet: InkWallet) throws -> String {
return try sign(message: message, wallet: wallet, addPrefix: true)
func signPersonalMessage(data: Data, wallet: InkWallet) throws -> String {
return try sign(data: data, wallet: wallet, addPrefix: true)
}
private func sign(message: String, wallet: InkWallet, addPrefix: Bool) throws -> String {
private func sign(data: Data, wallet: InkWallet, addPrefix: Bool) throws -> String {
guard let ethereumPrivateKey = wallet.ethereumPrivateKey else { throw Error.keyNotFound }
let withPrefixIfNeeded = (addPrefix ? "\u{19}Ethereum Signed Message:\n" + String(message.count) : "") + message
guard let data = withPrefixIfNeeded.data(using: .utf8) else { throw Error.invalidInputData }
guard var signed = ethereumPrivateKey.sign(digest: Hash.keccak256(data: data), curve: CoinType.ethereum.curve) else { throw Error.failedToSign }
let prefixString = addPrefix ? "\u{19}Ethereum Signed Message:\n" + String(data.count) : ""
guard let prefixData = prefixString.data(using: .utf8) else { throw Error.failedToSign }
let digest = Hash.keccak256(data: prefixData + data)
guard var signed = ethereumPrivateKey.sign(digest: digest, curve: CoinType.ethereum.curve) else { throw Error.failedToSign }
signed[64] += 27
return signed.toPrefixedHexString()
}

23
Encrypted Ink/WalletConnect.swift
View File

@ -150,10 +150,10 @@ class WalletConnect {
let title: String
switch payload {
case let .sign(data: data, raw: _):
message = String(data: data, encoding: .utf8)
message = String(data: data, encoding: .utf8) ?? data.hexString
title = "Sign Message"
case let .personalSign(data: data, raw: _):
message = String(data: data, encoding: .utf8)
message = String(data: data, encoding: .utf8) ?? data.hexString
title = "Sign Personal Message"
case let .signTypeData(id: _, data: _, raw: raw):
title = "Sign Typed Data"
@ -165,7 +165,7 @@ class WalletConnect {
let peer = getPeerOfInteractor(interactor)
Agent.shared.showApprove(title: title, meta: message ?? "", peerMeta: peer) { [weak self, weak interactor] approved in
if approved {
self?.sign(id: id, message: message, payload: payload, walletId: walletId, interactor: interactor)
self?.sign(id: id, payload: payload, walletId: walletId, interactor: interactor)
} else {
self?.rejectRequest(id: id, interactor: interactor, message: "Cancelled")
}
@ -188,19 +188,20 @@ class WalletConnect {
interactor?.approveRequest(id: requestId, result: hash).cauterize()
}
private func sign(id: Int64, message: String?, payload: WCEthereumSignPayload, walletId: String, interactor: WCInteractor?) {
guard let message = message, let wallet = walletsManager.getWallet(id: walletId) else {
private func sign(id: Int64, payload: WCEthereumSignPayload, walletId: String, interactor: WCInteractor?) {
guard let wallet = walletsManager.getWallet(id: walletId) else {
rejectRequest(id: id, interactor: interactor, message: "Something went wrong.")
return
}
var signed: String?
switch payload {
case .personalSign:
signed = try? ethereum.signPersonal(message: message, wallet: wallet)
case .signTypeData:
signed = try? ethereum.sign(typedData: message, wallet: wallet)
case .sign:
signed = try? ethereum.sign(message: message, wallet: wallet)
case let .personalSign(data: data, raw: _):
signed = try? ethereum.signPersonalMessage(data: data, wallet: wallet)
case let .signTypeData(id: _, data: _, raw: raw):
let typedData = raw.count >= 2 ? raw[1] : ""
signed = try? ethereum.sign(typedData: typedData, wallet: wallet)
case let .sign(data: data, raw: _):
signed = try? ethereum.sign(data: data, wallet: wallet)
}
guard let result = signed else {
rejectRequest(id: id, interactor: interactor, message: "Something went wrong.")