This repository has been archived by the owner on Jun 20, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 493
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
335 additions
and
63 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,228 @@ | ||
# SPDX-License-Identifier: MIT | ||
# OpenZeppelin Contracts for Cairo v0.1.0 (account/Account.cairo) | ||
|
||
%lang starknet | ||
|
||
from starkware.cairo.common.registers import get_fp_and_pc | ||
from starkware.starknet.common.syscalls import get_contract_address | ||
from starkware.cairo.common.signature import verify_ecdsa_signature | ||
from starkware.cairo.common.cairo_builtins import HashBuiltin, SignatureBuiltin | ||
from starkware.cairo.common.alloc import alloc | ||
from starkware.cairo.common.memcpy import memcpy | ||
from starkware.cairo.common.bool import TRUE | ||
from starkware.starknet.common.syscalls import call_contract, get_caller_address, get_tx_info | ||
|
||
# | ||
# Storage Var | ||
# | ||
@storage_var | ||
func Account_current_nonce() -> (res: felt): | ||
end | ||
@storage_var | ||
func Account_public_key() -> (res: felt): | ||
end | ||
# | ||
# Structs | ||
# | ||
struct Call: | ||
member to: felt | ||
member selector: felt | ||
member calldata_len: felt | ||
member calldata: felt* | ||
end | ||
# Tmp struct introduced while we wait for Cairo | ||
# to support passing `[AccountCall]` to __execute__ | ||
struct AccountCallArray: | ||
member to: felt | ||
member selector: felt | ||
member data_offset: felt | ||
member data_len: felt | ||
end | ||
# | ||
# Constructor | ||
# | ||
@constructor | ||
func constructor{ | ||
syscall_ptr : felt*, | ||
pedersen_ptr : HashBuiltin*, | ||
range_check_ptr | ||
}(public_key: felt): | ||
Account_public_key.write(public_key) | ||
return () | ||
end | ||
|
||
# | ||
# Getters | ||
# | ||
|
||
@view | ||
func get_public_key{ | ||
syscall_ptr : felt*, | ||
pedersen_ptr : HashBuiltin*, | ||
range_check_ptr | ||
}() -> (res: felt): | ||
let (res) = Account_public_key.read() | ||
return (res=res) | ||
end | ||
|
||
@view | ||
func get_nonce{ | ||
syscall_ptr : felt*, | ||
pedersen_ptr : HashBuiltin*, | ||
range_check_ptr | ||
}() -> (res: felt): | ||
let (res) = Account_current_nonce.read() | ||
return (res=res) | ||
end | ||
|
||
# | ||
# Setters | ||
# | ||
|
||
@external | ||
func set_public_key{ | ||
syscall_ptr : felt*, | ||
pedersen_ptr : HashBuiltin*, | ||
range_check_ptr | ||
}(new_public_key: felt): | ||
Account_public_key.write(new_public_key) | ||
return () | ||
end | ||
|
||
# | ||
# Business logic | ||
# | ||
|
||
@view | ||
func is_valid_signature{ | ||
syscall_ptr : felt*, | ||
pedersen_ptr : HashBuiltin*, | ||
range_check_ptr, | ||
ecdsa_ptr: SignatureBuiltin* | ||
}( | ||
hash: felt, | ||
signature_len: felt, | ||
signature: felt* | ||
) -> (is_valid: felt): | ||
let (_public_key) = Account_public_key.read() | ||
|
||
# This interface expects a signature pointer and length to make | ||
# no assumption about signature validation schemes. | ||
# But this implementation does, and it expects a (sig_r, sig_s) pair. | ||
let sig_r = signature[0] | ||
let sig_s = signature[1] | ||
|
||
verify_ecdsa_signature( | ||
message=hash, | ||
public_key=_public_key, | ||
signature_r=sig_r, | ||
signature_s=sig_s) | ||
|
||
return (is_valid=TRUE) | ||
end | ||
|
||
@external | ||
func __execute__{ | ||
syscall_ptr : felt*, | ||
pedersen_ptr : HashBuiltin*, | ||
range_check_ptr, | ||
ecdsa_ptr: SignatureBuiltin* | ||
}( | ||
call_array_len: felt, | ||
call_array: AccountCallArray*, | ||
calldata_len: felt, | ||
calldata: felt*, | ||
nonce: felt | ||
) -> (response_len: felt, response: felt*): | ||
alloc_locals | ||
|
||
let (caller) = get_caller_address() | ||
with_attr error_message("Account: no reentrant call"): | ||
assert caller = 0 | ||
end | ||
let (__fp__, _) = get_fp_and_pc() | ||
let (tx_info) = get_tx_info() | ||
let (_current_nonce) = Account_current_nonce.read() | ||
# validate nonce | ||
with_attr error_message("Account: nonce is invalid"): | ||
assert _current_nonce = nonce | ||
end | ||
# TMP: Convert `AccountCallArray` to 'Call'. | ||
let (calls : Call*) = alloc() | ||
_from_call_array_to_call(call_array_len, call_array, calldata, calls) | ||
let calls_len = call_array_len | ||
# validate transaction | ||
# let (is_valid) = is_valid_signature(tx_info.transaction_hash, tx_info.signature_len, tx_info.signature) | ||
# with_attr error_message("Account: invalid signature"): | ||
# assert is_valid = TRUE | ||
# end | ||
# bump nonce | ||
Account_current_nonce.write(_current_nonce + 1) | ||
# execute call | ||
let (response : felt*) = alloc() | ||
let (response_len) = _execute_list(calls_len, calls, response) | ||
return (response_len=response_len, response=response) | ||
end | ||
func _execute_list{syscall_ptr: felt*}( | ||
calls_len: felt, | ||
calls: Call*, | ||
response: felt* | ||
) -> (response_len: felt): | ||
alloc_locals | ||
|
||
# if no more calls | ||
if calls_len == 0: | ||
return (0) | ||
end | ||
|
||
# do the current call | ||
let this_call: Call = [calls] | ||
let res = call_contract( | ||
contract_address=this_call.to, | ||
function_selector=this_call.selector, | ||
calldata_size=this_call.calldata_len, | ||
calldata=this_call.calldata | ||
) | ||
# copy the result in response | ||
memcpy(response, res.retdata, res.retdata_size) | ||
# do the next calls recursively | ||
let (response_len) = _execute_list(calls_len - 1, calls + Call.SIZE, response + res.retdata_size) | ||
return (response_len + res.retdata_size) | ||
end | ||
|
||
func _from_call_array_to_call{syscall_ptr: felt*}( | ||
call_array_len: felt, | ||
call_array: AccountCallArray*, | ||
calldata: felt*, | ||
calls: Call* | ||
): | ||
# if no more calls | ||
if call_array_len == 0: | ||
return () | ||
end | ||
|
||
# parse the current call | ||
assert [calls] = Call( | ||
to=[call_array].to, | ||
selector=[call_array].selector, | ||
calldata_len=[call_array].data_len, | ||
calldata=calldata + [call_array].data_offset | ||
) | ||
# parse the remaining calls recursively | ||
_from_call_array_to_call(call_array_len - 1, call_array + AccountCallArray.SIZE, calldata, calls + Call.SIZE) | ||
return () | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
import os | ||
|
||
CAIRO_PATH = os.getenv("CAIRO_PATH") | ||
|
||
GOERLI_NODE=os.getenv("GOERLI_URL") | ||
GOERLI_CORE_CONTRACTS=0xde29d060D45901Fb19ED6C6e959EB22d8626708e | ||
GOERLI_VERIFIER=0xAB43bA48c9edF4C2C4bB01237348D1D7B28ef168 | ||
GOERLI_MEMORY_REGISTRY=0x743789ff2fF82Bfb907009C9911a7dA636D34FA7 | ||
|
||
FEEDER_URL = "https://alpha4.starknet.io/feeder_gateway" | ||
|
||
ESTIMATE_FEE_ENDPOINT = FEEDER_URL+"/estimate_fee" | ||
TRANSACTION_TRACE_ENDPOINT = FEEDER_URL+"/get_transaction_trace?transactionHash=" | ||
STATE_UPDATE_ENDPOINT = FEEDER_URL+"/get_state_update?blockNumber=" | ||
CALL_CONTRACT_ENDPOINT = FEEDER_URL+"/call_contract" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.