|
|
|
@ -371,10 +371,11 @@ bool TryQuery(CFGRebuildState& state) { |
|
|
|
if (state.queries.empty()) { |
|
|
|
return false; |
|
|
|
} |
|
|
|
|
|
|
|
Query& q = state.queries.front(); |
|
|
|
const u32 block_index = state.registered[q.address]; |
|
|
|
BlockInfo& block = state.block_info[block_index]; |
|
|
|
// If the block is visted, check if the stacks match, else gather the ssy/pbk
|
|
|
|
// If the block is visited, check if the stacks match, else gather the ssy/pbk
|
|
|
|
// labels into the current stack and look if the branch at the end of the block
|
|
|
|
// consumes a label. Schedule new queries accordingly
|
|
|
|
if (block.visited) { |
|
|
|
@ -385,7 +386,8 @@ bool TryQuery(CFGRebuildState& state) { |
|
|
|
return all_okay; |
|
|
|
} |
|
|
|
block.visited = true; |
|
|
|
state.stacks[q.address] = BlockStack{q}; |
|
|
|
state.stacks.insert_or_assign(q.address, BlockStack{q}); |
|
|
|
|
|
|
|
Query q2(q); |
|
|
|
state.queries.pop_front(); |
|
|
|
gather_labels(q2.ssy_stack, state.ssy_labels, block); |
|
|
|
@ -394,6 +396,7 @@ bool TryQuery(CFGRebuildState& state) { |
|
|
|
q2.address = block.end + 1; |
|
|
|
state.queries.push_back(q2); |
|
|
|
} |
|
|
|
|
|
|
|
Query conditional_query{q2}; |
|
|
|
if (block.branch.is_sync) { |
|
|
|
if (block.branch.address == unassigned_branch) { |
|
|
|
@ -408,7 +411,7 @@ bool TryQuery(CFGRebuildState& state) { |
|
|
|
conditional_query.pbk_stack.pop(); |
|
|
|
} |
|
|
|
conditional_query.address = block.branch.address; |
|
|
|
state.queries.push_back(conditional_query); |
|
|
|
state.queries.push_back(std::move(conditional_query)); |
|
|
|
return true; |
|
|
|
} |
|
|
|
} // Anonymous namespace
|
|
|
|
@ -416,6 +419,7 @@ bool TryQuery(CFGRebuildState& state) { |
|
|
|
std::optional<ShaderCharacteristics> ScanFlow(const ProgramCode& program_code, |
|
|
|
std::size_t program_size, u32 start_address) { |
|
|
|
CFGRebuildState state{program_code, program_size, start_address}; |
|
|
|
|
|
|
|
// Inspect Code and generate blocks
|
|
|
|
state.labels.clear(); |
|
|
|
state.labels.emplace(start_address); |
|
|
|
@ -425,10 +429,9 @@ std::optional<ShaderCharacteristics> ScanFlow(const ProgramCode& program_code, |
|
|
|
return {}; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
// Decompile Stacks
|
|
|
|
Query start_query{}; |
|
|
|
start_query.address = state.start; |
|
|
|
state.queries.push_back(start_query); |
|
|
|
state.queries.push_back(Query{state.start, {}, {}}); |
|
|
|
bool decompiled = true; |
|
|
|
while (!state.queries.empty()) { |
|
|
|
if (!TryQuery(state)) { |
|
|
|
@ -436,14 +439,15 @@ std::optional<ShaderCharacteristics> ScanFlow(const ProgramCode& program_code, |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
// Sort and organize results
|
|
|
|
std::sort(state.block_info.begin(), state.block_info.end(), |
|
|
|
[](const BlockInfo& a, const BlockInfo& b) -> bool { return a.start < b.start; }); |
|
|
|
[](const BlockInfo& a, const BlockInfo& b) { return a.start < b.start; }); |
|
|
|
ShaderCharacteristics result_out{}; |
|
|
|
result_out.decompilable = decompiled; |
|
|
|
result_out.start = start_address; |
|
|
|
result_out.end = start_address; |
|
|
|
for (auto& block : state.block_info) { |
|
|
|
for (const auto& block : state.block_info) { |
|
|
|
ShaderBlock new_block{}; |
|
|
|
new_block.start = block.start; |
|
|
|
new_block.end = block.end; |
|
|
|
@ -458,8 +462,9 @@ std::optional<ShaderCharacteristics> ScanFlow(const ProgramCode& program_code, |
|
|
|
} |
|
|
|
if (result_out.decompilable) { |
|
|
|
result_out.labels = std::move(state.labels); |
|
|
|
return {result_out}; |
|
|
|
return {std::move(result_out)}; |
|
|
|
} |
|
|
|
|
|
|
|
// If it's not decompilable, merge the unlabelled blocks together
|
|
|
|
auto back = result_out.blocks.begin(); |
|
|
|
auto next = std::next(back); |
|
|
|
@ -472,6 +477,6 @@ std::optional<ShaderCharacteristics> ScanFlow(const ProgramCode& program_code, |
|
|
|
back = next; |
|
|
|
++next; |
|
|
|
} |
|
|
|
return {result_out}; |
|
|
|
return {std::move(result_out)}; |
|
|
|
} |
|
|
|
} // namespace VideoCommon::Shader
|