mirror of
https://github.com/lifting-bits/remill
synced 2026-06-21 13:56:07 +00:00
Peter Goodman
99df2e19d4
* Running clang-format on files with some additional custom scripts for my style * Fix missing unique_ptr in remill/BC/Optimizer.h * Fixes and selective disabling of clang-format
97 lines
3.0 KiB
C++
97 lines
3.0 KiB
C++
/*
|
|
* Copyright (c) 2019 Trail of Bits, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "Annotate.h"
|
|
|
|
namespace remill {
|
|
|
|
const std::string BaseFunction::metadata_value = "base";
|
|
|
|
const std::string BaseFunction::metadata_kind = "remill.function.type";
|
|
|
|
const std::string LiftedFunction::metadata_value =
|
|
BaseFunction::metadata_value + "." + "lifted";
|
|
|
|
const std::string EntrypointFunction::metadata_value =
|
|
BaseFunction::metadata_value + "." + "entrypoint";
|
|
|
|
const std::string ExternalFunction::metadata_value =
|
|
BaseFunction::metadata_value + "." + "external";
|
|
|
|
const std::string AbiLibraries::metadata_value =
|
|
ExternalFunction::metadata_value + "." + "abilibraries";
|
|
|
|
const std::string CFGExternal::metadata_value =
|
|
ExternalFunction::metadata_value + "." + "cfgexternal";
|
|
|
|
const std::string ExtWrapper::metadata_value =
|
|
ExternalFunction::metadata_value + "." + "extwrapper";
|
|
|
|
|
|
const std::string Helper::metadata_value =
|
|
BaseFunction::metadata_value + "." + "helper";
|
|
|
|
|
|
const std::string RemillHelper::metadata_value =
|
|
Helper::metadata_value + "." + "remill";
|
|
|
|
const std::string McSemaHelper::metadata_value =
|
|
Helper::metadata_value + "." + "mcsema";
|
|
|
|
const std::string Semantics::metadata_value =
|
|
Helper::metadata_value + "." + "semantics";
|
|
|
|
#if LLVM_VERSION_NUMBER >= LLVM_VERSION(4, 0)
|
|
|
|
llvm::MDNode *TieFunction(llvm::Function *first, llvm::Function *second,
|
|
const std::string &kind) {
|
|
auto &C = first->getContext();
|
|
auto node = llvm::MDNode::get(C, llvm::ConstantAsMetadata::get(second));
|
|
first->setMetadata(kind, node);
|
|
return node;
|
|
}
|
|
|
|
std::pair<llvm::MDNode *, llvm::MDNode *>
|
|
TieFunctions(llvm::Function *first, llvm::Function *second,
|
|
const std::string &kind) {
|
|
LOG_IF(FATAL, IsTied(first, kind) || IsTied(second, kind))
|
|
<< "Tried to tie already tied functions " << first->getName().str()
|
|
<< " to " << second->getName().str();
|
|
|
|
return {TieFunction(first, second, kind), TieFunction(second, first, kind)};
|
|
}
|
|
|
|
|
|
llvm::Function *GetTied(llvm::Function *func, const std::string &kind) {
|
|
auto node = GetTieNode(func, kind);
|
|
if (!node || node->getNumOperands() != 1) {
|
|
return nullptr;
|
|
}
|
|
|
|
auto casted = llvm::dyn_cast<llvm::ConstantAsMetadata>(node->getOperand(0));
|
|
|
|
if (!casted) {
|
|
DLOG(INFO) << func->getName().str() << " with kind " << kind
|
|
<< " was tied to nullptr";
|
|
return nullptr;
|
|
}
|
|
|
|
return llvm::dyn_cast<llvm::Function>(casted->getValue());
|
|
}
|
|
|
|
#endif
|
|
} // namespace remill
|