Starknet-Security-Challenges-Repo/challenge11.cairo at main · devnet0x/Starknet-Security-Challenges-Repo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
%lang starknet
from starkware.cairo.common.cairo_builtins import HashBuiltin
from starkware.cairo.common.bool import FALSE, TRUE
from starkware.starknet.common.syscalls import get_caller_address,get_tx_info
from starkware.cairo.common.math import assert_not_equal


@storage_var
func owner() -> (value: felt) {
}

@storage_var
func is_complete() -> (value: felt) {
}

func get_owner{syscall_ptr: felt*, pedersen_ptr: HashBuiltin*, range_check_ptr}() -> felt {
        alloc_locals;
        let (res)=owner.read();
        return res;
}

@constructor
func constructor{syscall_ptr: felt*, pedersen_ptr: HashBuiltin*, range_check_ptr}() {
    alloc_locals;
    let (sender) = get_caller_address();
    owner.write(sender);
    is_complete.write(FALSE);
    return();
}


@external
func changeOwner{syscall_ptr: felt*, pedersen_ptr: HashBuiltin*, range_check_ptr}(
    _owner : felt) {
    alloc_locals;

    let (tx_info) = get_tx_info();
    let (sender) = get_caller_address();
    if (tx_info.account_contract_address != sender) {
      owner.write(_owner);
      is_complete.write(TRUE);
      return();
    }
    return();
}


@view
func isComplete{syscall_ptr: felt*, pedersen_ptr: HashBuiltin*, range_check_ptr}() -> (output:felt) {
    alloc_locals;
    let (output)=is_complete.read();
    return (output=output);
}