From e09015638ece23f06f34a553d8615c86b3be0712 Mon Sep 17 00:00:00 2001 From: bloodstalker Date: Sun, 12 Nov 2017 01:16:20 +0330 Subject: added obfuscator, some minor changes made to load.py so it would be easier to call python scripts from lua, general changes for a new project --- obfuscator/compile_commands.json | 7 ++ obfuscator/makefile | 27 ++++++ obfuscator/obfuscator.cpp | 200 +++++++++++++++++++++++++++++++++++++++ 3 files changed, 234 insertions(+) create mode 100644 obfuscator/compile_commands.json create mode 100644 obfuscator/makefile create mode 100644 obfuscator/obfuscator.cpp (limited to 'obfuscator') diff --git a/obfuscator/compile_commands.json b/obfuscator/compile_commands.json new file mode 100644 index 0000000..5b4729c --- /dev/null +++ b/obfuscator/compile_commands.json @@ -0,0 +1,7 @@ +[ + { + "command": "c++ -c -I/home/bloodstalker/extra/llvm-clang-4/llvm/include -I/home/bloodstalker/extra/llvm-clang-4/build/include -fPIC -fvisibility-inlines-hidden -Werror=date-time -std=c++11 -Wall -W -Wno-unused-parameter -Wwrite-strings -Wcast-qual -Wno-missing-field-initializers -pedantic -Wno-long-long -Wno-maybe-uninitialized -Wdelete-non-virtual-dtor -Wno-comment -ffunction-sections -fdata-sections -O2 -fno-exceptions -D_GNU_SOURCE -D_DEBUG -D__STDC_CONSTANT_MACROS -D__STDC_FORMAT_MACROS -D__STDC_LIMIT_MACROS -I/home/bloodstalker/extra/llvm-clang-4/llvm/tools/clang/include -I/home/bloodstalker/extra/llvm-clang-4/build/tools/clang/include -std=c++1z -stdlib=libstdc++ -UNDEBUG -fexceptions -o obfuscator.o obfuscator.cpp", + "directory": "/home/bloodstalker/devi/hell2/obfuscator", + "file": "/home/bloodstalker/devi/hell2/obfuscator/obfuscator.cpp" + } +] \ No newline at end of file diff --git a/obfuscator/makefile b/obfuscator/makefile new file mode 100644 index 0000000..dee61e2 --- /dev/null +++ b/obfuscator/makefile @@ -0,0 +1,27 @@ + +######################################INCLUDES################################# +include ../macros.mk + +#######################################VARS#################################### +OBSC=obfuscator +######################################RULES#################################### +.DEFAULT: all + +.PHONY: all clean help $(OBSC) + +all: $(OBSC) + +.cpp.o: + $(CXX) $(CXX_FLAGS) -c $< -o $@ + +$(OBSC): $(OBSC).o ../mutator_aux.o + $(CXX) $^ $(LD_FLAGS) -o $@ + +clean: + rm -f *.o *~ $(OBSC) + +help: + @echo 'There is help.' + @echo 'All is the defualt target.' + @echo 'Clean runs clean.' + @echo 'For a more complete and detaild list of BUILD_MODE and other things look at the main makefiles help under project root.' diff --git a/obfuscator/obfuscator.cpp b/obfuscator/obfuscator.cpp new file mode 100644 index 0000000..ae8e9e7 --- /dev/null +++ b/obfuscator/obfuscator.cpp @@ -0,0 +1,200 @@ + +/***************************************************Project Mutator****************************************************/ +//-*-c++-*- +/*first line intentionally left blank.*/ +/*Copyright (C) 2017 Farzad Sadeghi + +This program is free software; you can redistribute it and/or +modify it under the terms of the GNU General Public License +as published by the Free Software Foundation; either version 2 +of the License, or (at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program; if not, write to the Free Software +Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.*/ +/*code structure inspired by Eli Bendersky's tutorial on Rewriters.*/ +/**********************************************************************************************************************/ +/*FIXME-all classes should use replacements.*/ +/**********************************************************************************************************************/ +/*included modules*/ +/*project headers*/ +#include "../mutator_aux.h" +/*standard headers*/ +#include +#include +#include +/*LLVM headers*/ +#include "clang/AST/AST.h" +#include "clang/AST/ASTConsumer.h" +#include "clang/ASTMatchers/ASTMatchers.h" +#include "clang/ASTMatchers/ASTMatchFinder.h" +#include "clang/Basic/LLVM.h" +#include "clang/CodeGen/CodeGenAction.h" +#include "clang/CodeGen/BackendUtil.h" +#include "clang/Frontend/CompilerInstance.h" +#include "clang/Frontend/FrontendActions.h" +#include "clang/Lex/Lexer.h" +#include "clang/Tooling/CommonOptionsParser.h" +#include "clang/Tooling/Tooling.h" +#include "clang/Rewrite/Core/Rewriter.h" +#include "llvm/ADT/ArrayRef.h" +#include "llvm/IR/IRBuilder.h" +#include "llvm/IR/LLVMContext.h" +#include "llvm/IR/Module.h" +#include "llvm/IR/BasicBlock.h" +#include "llvm/IR/Function.h" +#include "llvm/IR/Function.h" +#include "llvm/Support/raw_ostream.h" +#include "llvm/Linker/Linker.h" +/**********************************************************************************************************************/ +/*used namespaces*/ +using namespace llvm; +using namespace clang; +using namespace clang::ast_matchers; +using namespace clang::driver; +using namespace clang::tooling; +/**********************************************************************************************************************/ +/*global vars*/ + +static llvm::cl::OptionCategory MatcherSampleCategory("Matcher Sample"); +/**********************************************************************************************************************/ +class FuncDecl : public MatchFinder::MatchCallback +{ +public: + FuncDecl (Rewriter &Rewrite) : Rewrite (Rewrite) {} + + virtual void run(const MatchFinder::MatchResult &MR) + { + if (MR.Nodes.getNodeAs("funcdecl") != nullptr) { + const FunctionDecl* FD = MR.Nodes.getNodeAs("funcdecl"); + std::string funcname = FD->getNameInfo().getAsString(); + std::size_t hash = std::hash{}(funcname); + std::string newname = "ID" + std::to_string(hash); + std::cout << "Function name: " << funcname << " Hash: " << hash << " New ID: " << newname << "\n"; + + SourceRange SR = FD->getSourceRange(); + } + } + +private: + Rewriter &Rewrite; +}; +/**********************************************************************************************************************/ +class VDecl : public MatchFinder::MatchCallback +{ +public: + VDecl (Rewriter &Rewrite) : Rewrite (Rewrite) {} + + virtual void run(const MatchFinder::MatchResult &MR) + { + if (MR.Nodes.getNodeAs("vardecl") != nullptr) { + const VarDecl* VD = MR.Nodes.getNodeAs("vardecl"); + std::string varname = VD->getIdentifier()->getName().str(); + std::size_t hash = std::hash{}(varname); + std::string newname = "ID" + std::to_string(hash); + std::cout << "Var name: " << varname << " Hash: " << hash << " New ID: " << newname << "\n"; + + SourceRange SR = VD->getSourceRange(); + } + } + +private: + Rewriter &Rewrite; +}; +/**********************************************************************************************************************/ +class ClassDecl : public MatchFinder::MatchCallback { + public: + ClassDecl (Rewriter &Rewrite) : Rewrite(Rewrite) {} + + virtual void run(const MatchFinder::MatchResult &MR) { + if (MR.Nodes.getNodeAs("classdecl") != nullptr) { + const RecordDecl* RD = MR.Nodes.getNodeAs("classdecl"); + } + } + + private: + Rewriter &Rewrite; +}; +/**********************************************************************************************************************/ +class PPInclusion : public PPCallbacks +{ +public: + explicit PPInclusion (SourceManager *SM) : SM(*SM) {} + + virtual void MacroDefined(const Token &MacroNameTok, const MacroDirective *MD) { + const MacroInfo* MI = MD->getMacroInfo(); + + SourceLocation SL = MacroNameTok.getLocation(); + CheckSLValidity(SL); + std::string macroname = MacroNameTok.getIdentifierInfo()->getName().str(); + std::size_t hash = std::hash{}(macroname); + std::string newname = "ID" + std::to_string(hash); + std::cout << "Macro name: " << macroname << " Hash: " << hash << " New ID: " << newname << "\n"; + } + +private: + const SourceManager &SM; +}; +/**********************************************************************************************************************/ +class BlankDiagConsumer : public clang::DiagnosticConsumer +{ + public: + BlankDiagConsumer() = default; + virtual ~BlankDiagConsumer() {} + virtual void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel, const Diagnostic &Info) override {} +}; +/**********************************************************************************************************************/ +class MyASTConsumer : public ASTConsumer { +public: + MyASTConsumer(Rewriter &R) : funcDeclHandler(R), HandlerForVar(R), HandlerForClass(R) { + Matcher.addMatcher(functionDecl().bind("funcdecl"), &funcDeclHandler); + Matcher.addMatcher(varDecl().bind("vardecl"), &HandlerForVar); + Matcher.addMatcher(recordDecl(isClass()).bind("classdecl"), &HandlerForClass); + } + + void HandleTranslationUnit(ASTContext &Context) override { + Matcher.matchAST(Context); + } + +private: + FuncDecl funcDeclHandler; + VDecl HandlerForVar; + ClassDecl HandlerForClass; + MatchFinder Matcher; +}; +/**********************************************************************************************************************/ +class ObfFrontendAction : public ASTFrontendAction { +public: + ObfFrontendAction() {} + ~ObfFrontendAction() {} + void EndSourceFileAction() override { + //TheRewriter.getEditBuffer(TheRewriter.getSourceMgr().getMainFileID()).write(llvm::outs()); + } + + std::unique_ptr CreateASTConsumer(CompilerInstance &CI, StringRef file) override { + CI.getPreprocessor().addPPCallbacks(llvm::make_unique(&CI.getSourceManager())); + DiagnosticsEngine &DE = CI.getPreprocessor().getDiagnostics(); + DE.setClient(BDCProto, false); + TheRewriter.setSourceMgr(CI.getSourceManager(), CI.getLangOpts()); + return llvm::make_unique(TheRewriter); + } + +private: + BlankDiagConsumer* BDCProto = new BlankDiagConsumer; + Rewriter TheRewriter; +}; +/**********************************************************************************************************************/ +/*Main*/ +int main(int argc, const char **argv) { + CommonOptionsParser op(argc, argv, MatcherSampleCategory); + ClangTool Tool(op.getCompilations(), op.getSourcePathList()); + return Tool.run(newFrontendActionFactory().get()); +} +/**********************************************************************************************************************/ +/*last line intentionally left blank.*/ + -- cgit v1.2.3