Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
@GCdePaula
GCdePaula committed Dec 22, 2024
1 parent c56671d commit 4709a6e
Showing 1 changed file with 112 additions and 112 deletions.
224 changes: 112 additions & 112 deletions prt/tests/rollups/prt_rollups.lua
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ local FAST_FORWARD_TIME = 20
-- delay time for blockchain node to be ready
local NODE_DELAY = 3
-- number of fake commitment to make
local FAKE_COMMITMENT_COUNT = 5
local FAKE_COMMITMENT_COUNT = 1
-- number of idle players
local IDLE_PLAYER_COUNT = 0
-- consensus contract address in anvil deployment
Expand Down Expand Up @@ -53,119 +53,119 @@ local ROOT_LEAFS_QUERY =
[[sqlite3 /compute_data/%s/db 'select level,base_cycle,compute_leaf_index,repetitions,HEX(compute_leaf)
from compute_leafs where level=0 ORDER BY compute_leaf_index ASC']]
local function build_root_commitment_from_db(machine_path, root_tournament)
local builder = MerkleBuilder:new()
local machine = Machine:new_from_path(machine_path)
local initial_state = machine:state()

local handle = io.popen(string.format(ROOT_LEAFS_QUERY, root_tournament))
assert(handle)
local rows = handle:read "*a"
handle:close()

if rows:find "Error" then
error(string.format("Read leafs failed:\n%s", rows))
end

-- Iterate over each line in the input data
for line in rows:gmatch("[^\n]+") do
local _, _, _, repetitions, compute_leaf = line:match(
"([^|]+)|([^|]+)|([^|]+)|([^|]+)|([^|]+)")
-- Convert values to appropriate types
repetitions = tonumber(repetitions)
compute_leaf = Hash:from_digest_hex("0x" .. compute_leaf)

builder:add(compute_leaf, repetitions)
end

return builder:build(initial_state.root_hash)
local builder = MerkleBuilder:new()
local machine = Machine:new_from_path(machine_path)
local initial_state = machine:state()

local handle = io.popen(string.format(ROOT_LEAFS_QUERY, root_tournament))
assert(handle)
local rows = handle:read "*a"
handle:close()

if rows:find "Error" then
error(string.format("Read leafs failed:\n%s", rows))
end

-- Iterate over each line in the input data
for line in rows:gmatch("[^\n]+") do
local _, _, _, repetitions, compute_leaf = line:match(
"([^|]+)|([^|]+)|([^|]+)|([^|]+)|([^|]+)")
-- Convert values to appropriate types
repetitions = tonumber(repetitions)
compute_leaf = Hash:from_digest_hex("0x" .. compute_leaf)

builder:add(compute_leaf, repetitions)
end

return builder:build(initial_state.root_hash)
end

local INPUTS_QUERY =
[[sqlite3 /compute_data/%s/db 'select HEX(input)
from inputs ORDER BY input_index ASC']]
local function get_inputs_from_db(root_tournament)
local handle = io.popen(string.format(INPUTS_QUERY, root_tournament))
assert(handle)
local rows = handle:read "*a"
handle:close()

if rows:find "Error" then
error(string.format("Read inputs failed:\n%s", rows))
end

local inputs = {}
-- Iterate over each line in the input data
for line in rows:gmatch("[^\n]+") do
local input = line:match("([^|]+)")
table.insert(inputs, "0x" .. input)
end

return inputs
local handle = io.popen(string.format(INPUTS_QUERY, root_tournament))
assert(handle)
local rows = handle:read "*a"
handle:close()

if rows:find "Error" then
error(string.format("Read inputs failed:\n%s", rows))
end

local inputs = {}
-- Iterate over each line in the input data
for line in rows:gmatch("[^\n]+") do
local input = line:match("([^|]+)")
table.insert(inputs, "0x" .. input)
end

return inputs
end

-- Function to setup players
local function setup_players(root_tournament, machine_path)
local player_coroutines = {}
local player_index = 1
print("Calculating root commitment...")
local root_commitment = build_root_commitment_from_db(machine_path, root_tournament)
local inputs = get_inputs_from_db(root_tournament)

if FAKE_COMMITMENT_COUNT > 0 then
print(string.format("Setting up dishonest player with %d fake commitments", FAKE_COMMITMENT_COUNT))
local scoped_require = new_scoped_require(_ENV)
local start_sybil = scoped_require "runners.sybil_runner"
player_coroutines[player_index] = start_sybil(player_index + 1, machine_path, root_commitment, root_tournament,
FAKE_COMMITMENT_COUNT, inputs)
player_index = player_index + 1
end

if IDLE_PLAYER_COUNT > 0 then
print(string.format("Setting up %d idle players", IDLE_PLAYER_COUNT))
local scoped_require = new_scoped_require(_ENV)
local start_idle = scoped_require "runners.idle_runner"
for _ = 1, IDLE_PLAYER_COUNT do
player_coroutines[player_index] = start_idle(player_index + 1, machine_path, root_tournament)
player_index = player_index + 1
local player_coroutines = {}
local player_index = 1
print("Calculating root commitment...")
local root_commitment = build_root_commitment_from_db(machine_path, root_tournament)
local inputs = get_inputs_from_db(root_tournament)

if FAKE_COMMITMENT_COUNT > 0 then
print(string.format("Setting up dishonest player with %d fake commitments", FAKE_COMMITMENT_COUNT))
local scoped_require = new_scoped_require(_ENV)
local start_sybil = scoped_require "runners.sybil_runner"
player_coroutines[player_index] = start_sybil(player_index + 1, machine_path, root_commitment, root_tournament,
FAKE_COMMITMENT_COUNT, inputs)
player_index = player_index + 1
end
end

return player_coroutines
if IDLE_PLAYER_COUNT > 0 then
print(string.format("Setting up %d idle players", IDLE_PLAYER_COUNT))
local scoped_require = new_scoped_require(_ENV)
local start_idle = scoped_require "runners.idle_runner"
for _ = 1, IDLE_PLAYER_COUNT do
player_coroutines[player_index] = start_idle(player_index + 1, machine_path, root_tournament)
player_index = player_index + 1
end
end

return player_coroutines
end

-- Function to run players
local function run_players(player_coroutines)
while true do
local idle = true
local has_live_coroutine = false
for i, c in ipairs(player_coroutines) do
if c then
local success, ret = coroutine.resume(c)
local status = coroutine.status(c)

if status == "dead" then
player_coroutines[i] = false
end
if not success then
print(string.format("coroutine %d fail to resume with error: %s", i, ret))
elseif ret then
has_live_coroutine = true
idle = idle and ret.idle
while true do
local idle = true
local has_live_coroutine = false
for i, c in ipairs(player_coroutines) do
if c then
local success, ret = coroutine.resume(c)
local status = coroutine.status(c)

if status == "dead" then
player_coroutines[i] = false
end
if not success then
print(string.format("coroutine %d fail to resume with error: %s", i, ret))
elseif ret then
has_live_coroutine = true
idle = idle and ret.idle
end
end
end
end
end

if not has_live_coroutine then
print("No active players, ending attack...")
break
end
if not has_live_coroutine then
print("No active players, ending attack...")
break
end

if idle then
print(string.format("All players idle, fastforward blockchain for %d seconds...", FAST_FORWARD_TIME))
blockchain_utils.advance_time(FAST_FORWARD_TIME, blockchain_constants.endpoint)
if idle then
print(string.format("All players idle, fastforward blockchain for %d seconds...", FAST_FORWARD_TIME))
blockchain_utils.advance_time(FAST_FORWARD_TIME, blockchain_constants.endpoint)
end
time.sleep(SLEEP_TIME)
end
time.sleep(SLEEP_TIME)
end
end

-- Main Execution
Expand All @@ -192,27 +192,27 @@ print("Hello from Dave rollups lua prototype!")
local input_index = 1

while true do
local sealed_epochs = reader:read_epochs_sealed(CONSENSUS_ADDRESS)
local sealed_epochs = reader:read_epochs_sealed(CONSENSUS_ADDRESS)

if #sealed_epochs > 0 then
local last_sealed_epoch = sealed_epochs[#sealed_epochs]
for _ = input_index, input_index + 2 do
sender:tx_add_input(INPUT_BOX_ADDRESS, APP_ADDRESS, ECHO_MSG)
end
if #sealed_epochs > 0 then
local last_sealed_epoch = sealed_epochs[#sealed_epochs]
for _ = input_index, input_index + 2 do
sender:tx_add_input(INPUT_BOX_ADDRESS, APP_ADDRESS, ECHO_MSG)
end

-- react to last sealed epoch
local root_tournament = sealed_epochs[#sealed_epochs].tournament
local work_path = string.format("/compute_data/%s", root_tournament)
if helper.exists(work_path) then
print(string.format("sybil player attacking epoch %d",
last_sealed_epoch.epoch_number))
local epoch_machine_path = string.format("/rollups_data/%d/0", last_sealed_epoch.epoch_number)
local player_coroutines = setup_players(root_tournament, epoch_machine_path)
run_players(player_coroutines)
-- react to last sealed epoch
local root_tournament = sealed_epochs[#sealed_epochs].tournament
local work_path = string.format("/compute_data/%s", root_tournament)
if helper.exists(work_path) then
print(string.format("sybil player attacking epoch %d",
last_sealed_epoch.epoch_number))
local epoch_machine_path = string.format("/rollups_data/%d/0", last_sealed_epoch.epoch_number)
local player_coroutines = setup_players(root_tournament, epoch_machine_path)
run_players(player_coroutines)
end
end
end
-- blockchain_utils.advance_time(EPOCH_TIME, blockchain_constants.endpoint)
time.sleep(SLEEP_TIME)
-- blockchain_utils.advance_time(EPOCH_TIME, blockchain_constants.endpoint)
time.sleep(SLEEP_TIME)
end

print("Good-bye, world!")

0 comments on commit 4709a6e

Please sign in to comment.