diff options
Diffstat (limited to 'function.cpp')
| -rw-r--r-- | function.cpp | 290 |
1 files changed, 290 insertions, 0 deletions
diff --git a/function.cpp b/function.cpp index a7e4c154..5b1d2ba4 100644 --- a/function.cpp +++ b/function.cpp @@ -1260,6 +1260,126 @@ set<SSAVariable> Function::GetHighLevelILSSAVariables() } +set<Variable> Function::GetMediumLevelILVariablesIfAvailable() +{ + Ref<MediumLevelILFunction> mlil = this->GetMediumLevelILIfAvailable(); + if (!mlil) + return {}; + + size_t count; + BNVariable* vars = BNGetMediumLevelILVariables(mlil->GetObject(), &count); + + set<Variable> result; + for (size_t i = 0; i < count; ++i) + result.emplace(vars[i]); + + BNFreeVariableList(vars); + return result; +} + + +set<Variable> Function::GetMediumLevelILAliasedVariablesIfAvailable() +{ + Ref<MediumLevelILFunction> mlil = this->GetMediumLevelILIfAvailable(); + if (!mlil) + return {}; + + size_t count; + BNVariable* vars = BNGetMediumLevelILAliasedVariables(mlil->GetObject(), &count); + + set<Variable> result; + for (size_t i = 0; i < count; ++i) + result.emplace(vars[i]); + + BNFreeVariableList(vars); + return result; +} + + +set<SSAVariable> Function::GetMediumLevelILSSAVariablesIfAvailable() +{ + Ref<MediumLevelILFunction> mlil = this->GetMediumLevelILIfAvailable(); + if (!mlil) + return {}; + + size_t count; + BNVariable* vars = BNGetMediumLevelILVariables(mlil->GetObject(), &count); + + set<SSAVariable> result; + for (size_t i = 0; i < count; ++i) + { + size_t versionCount; + size_t* versions = BNGetMediumLevelILVariableSSAVersions(mlil->GetObject(), &vars[i], &versionCount); + for (size_t j = 0; j < versionCount; ++j) + result.emplace(vars[i], versions[j]); + BNFreeILInstructionList(versions); + } + + BNFreeVariableList(vars); + return result; +} + + +set<Variable> Function::GetHighLevelILVariablesIfAvailable() +{ + Ref<HighLevelILFunction> hlil = this->GetHighLevelILIfAvailable(); + if (!hlil) + return {}; + + size_t count; + BNVariable* vars = BNGetHighLevelILVariables(hlil->GetObject(), &count); + + set<Variable> result; + for (size_t i = 0; i < count; ++i) + result.emplace(vars[i]); + + BNFreeVariableList(vars); + return result; +} + + +set<Variable> Function::GetHighLevelILAliasedVariablesIfAvailable() +{ + Ref<HighLevelILFunction> hlil = this->GetHighLevelILIfAvailable(); + if (!hlil) + return {}; + + size_t count; + BNVariable* vars = BNGetHighLevelILAliasedVariables(hlil->GetObject(), &count); + + set<Variable> result; + for (size_t i = 0; i < count; ++i) + result.emplace(vars[i]); + + BNFreeVariableList(vars); + return result; +} + + +set<SSAVariable> Function::GetHighLevelILSSAVariablesIfAvailable() +{ + Ref<HighLevelILFunction> hlil = this->GetHighLevelILIfAvailable(); + if (!hlil) + return {}; + + size_t count; + BNVariable* vars = BNGetHighLevelILVariables(hlil->GetObject(), &count); + + set<SSAVariable> result; + for (size_t i = 0; i < count; ++i) + { + size_t versionCount; + size_t* versions = BNGetHighLevelILVariableSSAVersions(hlil->GetObject(), &vars[i], &versionCount); + for (size_t j = 0; j < versionCount; ++j) + result.emplace(vars[i], versions[j]); + BNFreeILInstructionList(versions); + } + + BNFreeVariableList(vars); + return result; +} + + void Function::CreateAutoVariable(const Variable& var, const Confidence<Ref<Type>>& type, const string& name, bool ignoreDisjointUses) { @@ -2484,6 +2604,176 @@ vector<VariableReferenceSource> Function::GetHighLevelILVariableReferencesInRang } +vector<ILReferenceSource> Function::GetMediumLevelILVariableReferencesIfAvailable(const Variable& var) +{ + size_t count; + + BNVariable varData; + varData.type = var.type; + varData.index = var.index; + varData.storage = var.storage; + + BNILReferenceSource* refs = BNGetMediumLevelILVariableReferencesIfAvailable(m_object, &varData, &count); + + vector<ILReferenceSource> result; + result.reserve(count); + for (size_t i = 0; i < count; i++) + { + ILReferenceSource src; + src.func = new Function(BNNewFunctionReference(refs[i].func)); + src.arch = new CoreArchitecture(refs[i].arch); + src.addr = refs[i].addr; + src.type = refs[i].type; + src.exprId = refs[i].exprId; + result.push_back(src); + } + + BNFreeILReferences(refs, count); + return result; +} + + +vector<VariableReferenceSource> Function::GetMediumLevelILVariableReferencesFromIfAvailable(Architecture* arch, uint64_t addr) +{ + size_t count; + BNVariableReferenceSource* refs = BNGetMediumLevelILVariableReferencesFromIfAvailable(m_object, arch->GetObject(), addr, &count); + + vector<VariableReferenceSource> result; + result.reserve(count); + for (size_t i = 0; i < count; i++) + { + VariableReferenceSource src; + src.var.index = refs[i].var.index; + src.var.storage = refs[i].var.storage; + src.var.type = refs[i].var.type; + + src.source.func = new Function(BNNewFunctionReference(refs[i].source.func)); + src.source.arch = new CoreArchitecture(refs[i].source.arch); + src.source.addr = refs[i].source.addr; + src.source.type = refs[i].source.type; + src.source.exprId = refs[i].source.exprId; + + result.push_back(src); + } + + BNFreeVariableReferenceSourceList(refs, count); + return result; +} + + +vector<VariableReferenceSource> Function::GetMediumLevelILVariableReferencesInRangeIfAvailable(Architecture* arch, uint64_t addr, uint64_t len) +{ + size_t count; + BNVariableReferenceSource* refs = BNGetMediumLevelILVariableReferencesInRangeIfAvailable(m_object, arch->GetObject(), addr, len, &count); + + vector<VariableReferenceSource> result; + result.reserve(count); + for (size_t i = 0; i < count; i++) + { + VariableReferenceSource src; + src.var.index = refs[i].var.index; + src.var.storage = refs[i].var.storage; + src.var.type = refs[i].var.type; + + src.source.func = new Function(BNNewFunctionReference(refs[i].source.func)); + src.source.arch = new CoreArchitecture(refs[i].source.arch); + src.source.addr = refs[i].source.addr; + src.source.type = refs[i].source.type; + src.source.exprId = refs[i].source.exprId; + + result.push_back(src); + } + + BNFreeVariableReferenceSourceList(refs, count); + return result; +} + + +vector<ILReferenceSource> Function::GetHighLevelILVariableReferencesIfAvailable(const Variable& var) +{ + size_t count; + + BNVariable varData; + varData.type = var.type; + varData.index = var.index; + varData.storage = var.storage; + + BNILReferenceSource* refs = BNGetHighLevelILVariableReferencesIfAvailable(m_object, &varData, &count); + + vector<ILReferenceSource> result; + result.reserve(count); + for (size_t i = 0; i < count; i++) + { + ILReferenceSource src; + src.func = new Function(BNNewFunctionReference(refs[i].func)); + src.arch = new CoreArchitecture(refs[i].arch); + src.addr = refs[i].addr; + src.type = refs[i].type; + src.exprId = refs[i].exprId; + result.push_back(src); + } + + BNFreeILReferences(refs, count); + return result; +} + + +vector<VariableReferenceSource> Function::GetHighLevelILVariableReferencesFromIfAvailable(Architecture* arch, uint64_t addr) +{ + size_t count; + BNVariableReferenceSource* refs = BNGetHighLevelILVariableReferencesFromIfAvailable(m_object, arch->GetObject(), addr, &count); + + vector<VariableReferenceSource> result; + result.reserve(count); + for (size_t i = 0; i < count; i++) + { + VariableReferenceSource src; + src.var.index = refs[i].var.index; + src.var.storage = refs[i].var.storage; + src.var.type = refs[i].var.type; + + src.source.func = new Function(BNNewFunctionReference(refs[i].source.func)); + src.source.arch = new CoreArchitecture(refs[i].source.arch); + src.source.addr = refs[i].source.addr; + src.source.type = refs[i].source.type; + src.source.exprId = refs[i].source.exprId; + + result.push_back(src); + } + + BNFreeVariableReferenceSourceList(refs, count); + return result; +} + + +vector<VariableReferenceSource> Function::GetHighLevelILVariableReferencesInRangeIfAvailable(Architecture* arch, uint64_t addr, uint64_t len) +{ + size_t count; + BNVariableReferenceSource* refs = BNGetHighLevelILVariableReferencesInRangeIfAvailable(m_object, arch->GetObject(), addr, len, &count); + + vector<VariableReferenceSource> result; + result.reserve(count); + for (size_t i = 0; i < count; i++) + { + VariableReferenceSource src; + src.var.index = refs[i].var.index; + src.var.storage = refs[i].var.storage; + src.var.type = refs[i].var.type; + + src.source.func = new Function(BNNewFunctionReference(refs[i].source.func)); + src.source.arch = new CoreArchitecture(refs[i].source.arch); + src.source.addr = refs[i].source.addr; + src.source.type = refs[i].source.type; + src.source.exprId = refs[i].source.exprId; + + result.push_back(src); + } + + BNFreeVariableReferenceSourceList(refs, count); + return result; +} + + uint64_t Function::GetHighestAddress() { return BNGetFunctionHighestAddress(m_object); |