Contract Examples

my_contract/
├── Cargo.toml
├── Cargo.lock
├── .cargo/
│   └── config.toml
├── rust-toolchain.toml
├── build.sh
└── src/
    └── lib.rs

[package]
name = "my_contract"
version = "0.1.0"
edition = "2021"
 
[lib]
crate-type = ["cdylib"]
 
[dependencies]
tako-sdk = { git = "https://github.com/tos-network/tako" }
 
[profile.release]
opt-level = "z"
lto = true

[build]
target = "tbpfv3-tos-tos"
 
[target.tbpfv3-tos-tos]
rustflags = [
    "-C", "link-arg=-z", "-C", "link-arg=notext",
]

[toolchain]
channel = "1.89.0-dev"

#!/bin/bash
set -e
 
cargo build --release
 
# Copy output
cp target/tbpfv3-tos-tos/release/*.so ./

const KEY_INITIALIZED: &[u8] = b"init";
 
fn is_initialized() -> bool {
    let mut buf = [0u8; 1];
    storage_read(KEY_INITIALIZED, &mut buf) > 0 && buf[0] == 1
}
 
fn set_initialized() {
    let _ = storage_write(KEY_INITIALIZED, &[1u8]);
}
 
fn require_initialized() -> Result<(), u64> {
    if !is_initialized() {
        return Err(ERR_NOT_INITIALIZED);
    }
    Ok(())
}

const KEY_OWNER: &[u8] = b"owner";
 
fn get_owner() -> [u8; 32] {
    let mut owner = [0u8; 32];
    storage_read(KEY_OWNER, &mut owner);
    owner
}
 
fn require_owner() -> Result<(), u64> {
    let caller = get_tx_sender();
    let owner = get_owner();
    if caller != owner {
        return Err(ERR_UNAUTHORIZED);
    }
    Ok(())
}

const PREFIX_BALANCE: u8 = 0x10;
 
fn get_balance(account: &[u8; 32]) -> u64 {
    let mut key = [0u8; 33];
    key[0] = PREFIX_BALANCE;
    key[1..33].copy_from_slice(account);
 
    let mut buf = [0u8; 8];
    if storage_read(&key, &mut buf) == 8 {
        u64::from_le_bytes(buf)
    } else {
        0
    }
}
 
fn set_balance(account: &[u8; 32], amount: u64) {
    let mut key = [0u8; 33];
    key[0] = PREFIX_BALANCE;
    key[1..33].copy_from_slice(account);
    let _ = storage_write(&key, &amount.to_le_bytes());
}

#[no_mangle]
pub extern "C" fn entrypoint() -> u64 {
    let mut input = [0u8; 1024];
    let len = get_input_data(&mut input);
 
    if len == 0 {
        return ERR_INVALID_INPUT;
    }
 
    let opcode = input[0];
    let params = &input[1..len as usize];
 
    match opcode {
        0x00 => op_initialize(params),
        0x01 => op_transfer(params),
        0x02 => op_approve(params),
        0x10 => op_balance_of(params),
        _ => ERR_UNKNOWN_OPCODE,
    }
}

# Install platform tools (first time)
curl -sSf https://tos.network/install.sh | sh
 
# Build contract
cd my_contract
./build.sh
 
# Or manually
cargo build --release

# Run local devnet
tos daemon --network devnet
 
# Deploy contract
tos_wallet deploy_contract ./my_contract.so
 
# Call contract
tos_wallet call_contract <contract_address> <hex_input>