Toggle navigation
Toggle navigation
This project
Loading...
Sign in
2020-1-capstone-design1
/
JJS_Project1
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Snippets
Network
Create a new issue
Builds
Commits
Issue Boards
Authored by
이한솔
2020-06-20 23:42:14 +0900
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Commit
ea70b69a02ea5e8f14987a7e42e59e38577030ae
ea70b69a
1 parent
3e93963c
보고서 및 발표자료 업로드
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
754 additions
and
279 deletions
src/llvm/lib/Transforms/Obfuscation/PreProcess.cpp
src/llvm/lib/Transforms/Obfuscation/ReturnObfuscation.cpp
진행보고서/[최종발표자료]_LLVM_기반의_Baremetal_IoT_디바이스_펌웨어_난독화.pptx
진행보고서/[최종보고서]_LLVM_기반의_Baremetal_IoT_디바이스_펌웨어_난독화.docx
src/llvm/lib/Transforms/Obfuscation/PreProcess.cpp
View file @
ea70b69
...
...
@@ -14,32 +14,36 @@ namespace {
static
char
ID
;
PreProcess
()
:
FunctionPass
(
ID
)
{
}
bool
runOnFunction
(
Function
&
F
)
override
{
bool
runOnFunction
(
Function
&
F
)
override
{
Module
*
mod
=
F
.
getParent
();
std
::
vector
<
Instruction
*>
instructions
;
std
::
vector
<
BasicBlock
*>
RetBlocks
;
bool
inserted
=
false
;
std
::
ofstream
functionFile
(
"functions.txt"
,
std
::
ios_base
::
app
);
if
(
functionFile
.
is_open
())
{
functionFile
<<
F
.
getName
().
str
()
<<
"
\n
"
;
if
(
!
F
.
getName
().
contains
(
"__cxx"
)
&&
!
F
.
getName
().
contains
(
"_GLOBAL"
))
functionFile
<<
F
.
getName
().
str
()
<<
"
\n
"
;
functionFile
.
close
();
}
for
(
auto
&
BB
:
F
)
{
for
(
auto
&
I
:
BB
)
{
if
(
I
.
getOpcode
()
==
Instruction
::
Ret
)
{
instructions
.
push_back
(
&
I
);
if
(
!
F
.
getName
().
contains
(
"__cxx"
)
&&
!
F
.
getName
().
contains
(
"_GLOBAL"
))
{
for
(
auto
&
BB
:
F
)
{
for
(
auto
&
I
:
BB
)
{
if
(
I
.
getOpcode
()
==
Instruction
::
Ret
)
{
instructions
.
push_back
(
&
I
);
}
}
}
}
for
(
auto
&
I
:
instructions
)
{
BasicBlock
*
BB
=
I
->
getParent
();
// One Instruction Basic Block has only one ret instructions
if
(
!
BB
->
size
()
<
2
)
{
BasicBlock
*
retblock
=
BB
->
splitBasicBlock
(
I
->
getIterator
(),
"obfuscatedreturn"
);
}
else
{
BB
->
setName
(
"obfuscatedreturn"
);
for
(
auto
&
I
:
instructions
)
{
BasicBlock
*
BB
=
I
->
getParent
();
// One Instruction Basic Block has only one ret instructions
if
(
!
BB
->
size
()
<
2
)
{
BasicBlock
*
retblock
=
BB
->
splitBasicBlock
(
I
->
getIterator
(),
"obfuscatedreturn"
);
}
else
{
BB
->
setName
(
"obfuscatedreturn"
);
}
}
}
return
true
;
}
...
...
src/llvm/lib/Transforms/Obfuscation/ReturnObfuscation.cpp
View file @
ea70b69
#include "llvm/
Pass
.h"
#include "llvm/
IR/CFG
.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/Module.h"
#include "llvm/Pass.h"
#include "llvm/Support/Alignment.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/IR/CFG.h"
#include <fstream>
#include <string>
using
namespace
llvm
;
namespace
{
struct
ReturnObfuscation
:
public
FunctionPass
{
static
char
ID
;
ReturnObfuscation
()
:
FunctionPass
(
ID
)
{
}
bool
runOnFunction
(
Function
&
F
)
override
{
size_t
num_retblocks
;
Module
*
mod
=
F
.
getParent
();
std
::
vector
<
Constant
*>
retblocks
;
// 함수 가져오기
std
::
ifstream
function_list
;
function_list
.
open
(
"functions.txt"
);
std
::
vector
<
Function
*>
functions
;
std
::
string
line
;
while
(
getline
(
function_list
,
line
))
{
functions
.
push_back
(
mod
->
getFunction
(
line
));
}
// 함수 별로 벡터에 집어넣기
for
(
auto
&
Fn
:
functions
)
{
for
(
auto
&
BB
:
(
*
Fn
))
{
if
(
BB
.
getName
().
equals
(
"obfuscatedreturn"
)){
Constant
*
retBlockAddress
=
BlockAddress
::
get
(
&
BB
);
retblocks
.
push_back
(
retBlockAddress
);
}
}
}
struct
ReturnObfuscation
:
public
FunctionPass
{
static
char
ID
;
ReturnObfuscation
()
:
FunctionPass
(
ID
)
{}
bool
runOnFunction
(
Function
&
F
)
override
{
int
num_this_function
;
int
count
;
size_t
num_retblocks
;
Module
*
mod
=
F
.
getParent
();
std
::
vector
<
Constant
*>
retblocks
;
// 함수 가져오기
std
::
ifstream
function_list
;
function_list
.
open
(
"functions.txt"
);
std
::
vector
<
Function
*>
functions
;
std
::
vector
<
Function
*>
functions2
;
std
::
string
line
;
while
(
getline
(
function_list
,
line
))
{
functions
.
push_back
(
mod
->
getFunction
(
line
));
}
count
=
0
;
bool
inserted
=
false
;
// 함수 별로 벡터에 집어넣기
for
(
auto
&
Fn
:
functions
)
{
inserted
=
false
;
for
(
auto
&
BB
:
(
*
Fn
))
{
if
(
BB
.
getName
().
equals
(
"obfuscatedreturn"
))
{
Constant
*
retBlockAddress
=
BlockAddress
::
get
(
&
BB
);
retblocks
.
push_back
(
retBlockAddress
);
if
(
!
inserted
)
functions2
.
push_back
(
Fn
);
inserted
=
true
;
}
}
}
for
(
auto
&
Fn
:
functions2
)
{
if
(
Fn
->
getName
().
equals
(
F
.
getName
()))
{
errs
()
<<
"GOOD Num this function "
<<
Fn
->
getName
()
<<
"
\n
"
;
num_this_function
=
count
;
}
count
++
;
}
num_retblocks
=
retblocks
.
size
();
ArrayType
*
array_in
=
ArrayType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
8
),
30
);
ArrayType
*
array_out
=
ArrayType
::
get
(
array_in
,
num_retblocks
);
ArrayType
*
array_retblock
=
ArrayType
::
get
(
PointerType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
8
),
0
),
num_retblocks
);
PointerType
*
array_ptr
=
PointerType
::
get
(
array_out
,
0
);
ConstantInt
*
const_int_0
=
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
32
,
StringRef
(
std
::
to_string
(
0
)),
10
));
ConstantInt
*
const_int_1
=
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
32
,
StringRef
(
std
::
to_string
(
1
)),
10
));
ConstantInt
*
const_int_array_out
=
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
32
,
StringRef
(
std
::
to_string
(
num_retblocks
)),
10
));
ConstantInt
*
const_int_array_in
=
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
32
,
StringRef
(
std
::
to_string
(
30
)),
10
));
ConstantInt
*
const_int_xordata
=
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
32
,
StringRef
(
std
::
to_string
(
0x33
)),
10
));
std
::
vector
<
Type
*>
Func_deobfus_type_args
;
FunctionType
*
Func_deobfus_type
=
FunctionType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
32
),
Func_deobfus_type_args
,
false
);
Function
*
Func_deobfus
=
mod
->
getFunction
(
"func_deobfus"
);
if
(
!
Func_deobfus
)
{
Func_deobfus
=
Function
::
Create
(
Func_deobfus_type
,
GlobalValue
::
ExternalLinkage
,
"func_deobfus"
,
mod
);
Func_deobfus
->
setCallingConv
(
CallingConv
::
C
);
}
AttributeList
Func_deobfus_att_list
;
SmallVector
<
AttributeList
,
4
>
Attrs
;
AttributeList
PAS
;
AttrBuilder
B
;
B
.
addAttribute
(
Attribute
::
NoInline
);
B
.
addAttribute
(
Attribute
::
NoRecurse
);
B
.
addAttribute
(
Attribute
::
NoUnwind
);
B
.
addAttribute
(
Attribute
::
OptimizeNone
);
PAS
=
AttributeList
::
get
(
mod
->
getContext
(),
~
0U
,
B
);
Attrs
.
push_back
(
PAS
);
Func_deobfus_att_list
=
AttributeList
::
get
(
mod
->
getContext
(),
Attrs
);
Func_deobfus
->
setAttributes
(
Func_deobfus_att_list
);
if
(
Func_deobfus
->
size
()
==
0
)
{
PointerType
*
ret_func_ptr
[
10000
];
AllocaInst
*
ptr_this_ret
[
10000
];
StoreInst
*
void_17
[
10000
];
GlobalVariable
*
gvar_ret_inst_list
=
new
GlobalVariable
(
*
mod
,
array_out
,
false
,
GlobalValue
::
CommonLinkage
,
0
,
"ret_inst_list"
);
gvar_ret_inst_list
->
setAlignment
(
MaybeAlign
(
16
));
ConstantAggregateZero
*
const_array_6
=
ConstantAggregateZero
::
get
(
array_out
);
gvar_ret_inst_list
->
setInitializer
(
const_array_6
);
// TODO : 없애기
/*
GlobalVariable *gvar_array_retblock =
new GlobalVariable(*mod, array_retblock, false,
GlobalValue::ExternalLinkage, 0, "array_retblock");
gvar_array_retblock->setAlignment(MaybeAlign(16));
*/
GlobalVariable
*
gvar_array_retblock
=
new
GlobalVariable
(
*
mod
,
array_retblock
,
false
,
GlobalValue
::
CommonLinkage
,
0
,
"array_retblock"
);
gvar_array_retblock
->
setAlignment
(
MaybeAlign
(
16
));
//TODO 고치기
ConstantAggregateZero
*
const_array_7
=
ConstantAggregateZero
::
get
(
array_retblock
);
gvar_array_retblock
->
setInitializer
(
const_array_7
);
BasicBlock
*
obfus_entry
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"entry"
,
Func_deobfus
);
BasicBlock
*
for_i_cond
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"i_cond"
,
Func_deobfus
);
BasicBlock
*
for_j_init
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"j_init"
,
Func_deobfus
);
BasicBlock
*
for_j_cond
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"j_cond"
,
Func_deobfus
);
BasicBlock
*
for_i_j
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"i_j"
,
Func_deobfus
);
BasicBlock
*
for_j_add
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"j_add"
,
Func_deobfus
);
BasicBlock
*
for_i_add
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"i_add"
,
Func_deobfus
);
BasicBlock
*
for_i_end
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"i_end"
,
Func_deobfus
);
std
::
vector
<
Constant
*>
const_ptr_blocks
;
for
(
size_t
i
=
0
;
i
<
num_retblocks
;
i
++
)
{
// Constant* const_blockaddress = BlockAddress::get(ret)
// TODO: 꼭해야함
//const_ptr_blocks.push_back(retblocks[i]);
std
::
vector
<
Constant
*>
const_ptr_16_indices
;
const_ptr_16_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
const_ptr_16_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
std
::
to_string
(
i
)),
10
)));
Constant
*
const_ptr_16
=
ConstantExpr
::
getGetElementPtr
(
array_retblock
,
gvar_array_retblock
,
const_ptr_16_indices
);
StoreInst
*
store_good
=
new
StoreInst
(
retblocks
[
i
],
const_ptr_16
,
false
,
obfus_entry
);
store_good
->
setAlignment
(
MaybeAlign
(
16
));
/*
std::vector<Value *> get_retblock_array_indices;
get_retblock_array_indices.push_back(const_int_0);
get_retblock_array_indices.push_back(ConstantInt::get(
mod->getContext(), APInt(32, StringRef(std::to_string(i)), 10)));
Instruction *get_retblock = GetElementPtrInst::Create(
array_retblock, gvar_array_retblock, get_retblock_array_indices, "",
obfus_entry);
LoadInst *load_ret_block_arr =
new LoadInst(get_retblock, "", false, obfus_entry);
*/
/*
ret_func_ptr[i] =
PointerType::get(IntegerType::get(mod->getContext(), 8), 0);
ptr_this_ret[i] =
new AllocaInst(ret_func_ptr[i], NULL, "ptr", obfus_entry);
;
*/
// void_17[i] =
// new StoreInst(retblocks[i], get_retblock, false, obfus_entry);
}
// TODO : 없애기
/*
Constant* array_ret_blocks_const = ConstantArray::get(array_retblock, const_ptr_blocks);
gvar_array_retblock->setInitializer(array_ret_blocks_const);
*/
// %i = alloca i32, align 4
AllocaInst
*
ptr_i
=
new
AllocaInst
(
IntegerType
::
get
(
mod
->
getContext
(),
32
),
NULL
,
"i"
,
obfus_entry
);
// %j = alloca i32, align 4
AllocaInst
*
ptr_j
=
new
AllocaInst
(
IntegerType
::
get
(
mod
->
getContext
(),
32
),
NULL
,
"j"
,
obfus_entry
);
StoreInst
*
str_i_0_1
=
new
StoreInst
(
const_int_0
,
ptr_i
,
false
,
obfus_entry
);
str_i_0_1
->
setAlignment
(
MaybeAlign
(
4
));
// br label %i_cond
BranchInst
::
Create
(
for_i_cond
,
obfus_entry
);
// for_i_cond
// %4 = load i32* %i, align 4
LoadInst
*
load_i_1
=
new
LoadInst
(
ptr_i
,
""
,
false
,
for_i_cond
);
load_i_1
->
setAlignment
(
MaybeAlign
(
4
));
// %5 = icmp slt i32 %4, 50
ICmpInst
*
i_cmp_1
=
new
ICmpInst
(
*
for_i_cond
,
ICmpInst
::
ICMP_SLT
,
load_i_1
,
const_int_array_out
,
""
);
// br i1 %5, label %for_j_init, for_i_end
BranchInst
::
Create
(
for_j_init
,
for_i_end
,
i_cmp_1
,
for_i_cond
);
// for_j_init
StoreInst
*
str_j_0_1
=
new
StoreInst
(
const_int_0
,
ptr_j
,
false
,
for_j_init
);
str_j_0_1
->
setAlignment
(
MaybeAlign
(
4
));
BranchInst
::
Create
(
for_j_cond
,
for_j_init
);
// for_j_cond
LoadInst
*
load_j_1
=
new
LoadInst
(
ptr_j
,
""
,
false
,
for_j_cond
);
load_j_1
->
setAlignment
(
MaybeAlign
(
4
));
// %5 = icmp slt i32 %4, 50
ICmpInst
*
j_cmp_1
=
new
ICmpInst
(
*
for_j_cond
,
ICmpInst
::
ICMP_SLT
,
load_j_1
,
const_int_array_in
,
""
);
// br i1 %5, label %for_j_init, for_i_end
BranchInst
::
Create
(
for_i_j
,
for_i_add
,
j_cmp_1
,
for_j_cond
);
// for_i_j
LoadInst
*
load_i_addc
=
new
LoadInst
(
ptr_i
,
""
,
false
,
for_i_j
);
load_i_addc
->
setAlignment
(
MaybeAlign
(
4
));
CastInst
*
casted_i_addc
=
new
SExtInst
(
load_i_addc
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
std
::
vector
<
Value
*>
get_retblock_index_ptr_indices
;
get_retblock_index_ptr_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_retblock_index_ptr_indices
.
push_back
(
casted_i_addc
);
Instruction
*
get_retblock_index_ptr
=
GetElementPtrInst
::
Create
(
array_retblock
,
gvar_array_retblock
,
get_retblock_index_ptr_indices
,
""
,
for_i_j
);
LoadInst
*
load_retblock_ptr
=
new
LoadInst
(
get_retblock_index_ptr
,
""
,
false
,
for_i_j
);
load_retblock_ptr
->
setAlignment
(
MaybeAlign
(
8
));
LoadInst
*
load_j_addc
=
new
LoadInst
(
ptr_j
,
""
,
false
,
for_i_j
);
load_j_addc
->
setAlignment
(
MaybeAlign
(
4
));
CastInst
*
casted_j_addc
=
new
SExtInst
(
load_j_addc
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
std
::
vector
<
Value
*>
get_retblock_index_ptr_value_indices
;
get_retblock_index_ptr_value_indices
.
push_back
(
casted_j_addc
);
Instruction
*
get_retblock_index_ptr_value
=
GetElementPtrInst
::
Create
(
IntegerType
::
get
(
mod
->
getContext
(),
8
),
load_retblock_ptr
,
get_retblock_index_ptr_value_indices
,
""
,
for_i_j
);
LoadInst
*
load_retblock_index_ptr_value
=
new
LoadInst
(
get_retblock_index_ptr_value
,
""
,
false
,
for_i_j
);
load_retblock_index_ptr_value
->
setAlignment
(
MaybeAlign
(
1
));
// %5
LoadInst
*
load_i_kkbc
=
new
LoadInst
(
ptr_i
,
""
,
false
,
for_i_j
);
load_i_kkbc
->
setAlignment
(
MaybeAlign
(
4
));
// 6
CastInst
*
casted_i_kkbc
=
new
SExtInst
(
load_i_kkbc
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
// idxprom6
std
::
vector
<
Value
*>
get_ret_inst_ptr_indices
;
get_ret_inst_ptr_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_ret_inst_ptr_indices
.
push_back
(
casted_i_kkbc
);
Instruction
*
get_ret_inst_ptr
=
GetElementPtrInst
::
Create
(
array_out
,
gvar_ret_inst_list
,
get_ret_inst_ptr_indices
,
""
,
for_i_j
);
//arrayidx7
LoadInst
*
load_j_kkbc
=
new
LoadInst
(
ptr_j
,
""
,
false
,
for_i_j
);
load_j_kkbc
->
setAlignment
(
MaybeAlign
(
4
));
// 7
CastInst
*
casted_j_kkbc
=
new
SExtInst
(
load_j_kkbc
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
// idxprom8
std
::
vector
<
Value
*>
get_ret_inst_ptr_value_indices
;
get_ret_inst_ptr_value_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_ret_inst_ptr_value_indices
.
push_back
(
casted_j_kkbc
);
Instruction
*
get_ret_inst_ptr_value
=
GetElementPtrInst
::
Create
(
array_in
,
get_ret_inst_ptr
,
get_ret_inst_ptr_value_indices
,
""
,
for_i_j
);
//arrayidx9
StoreInst
*
store_data
=
new
StoreInst
(
load_retblock_index_ptr_value
,
get_ret_inst_ptr_value
,
false
,
for_i_j
);
store_data
->
setAlignment
(
MaybeAlign
(
1
));
LoadInst
*
load_i_arrd
=
new
LoadInst
(
ptr_i
,
""
,
false
,
for_i_j
);
load_i_arrd
->
setAlignment
(
MaybeAlign
(
4
));
// 8
CastInst
*
casted_i_arrd
=
new
SExtInst
(
load_i_arrd
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
// idxprom10
std
::
vector
<
Value
*>
get_ret_inst_ptr_indices2
;
get_ret_inst_ptr_indices2
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_ret_inst_ptr_indices2
.
push_back
(
casted_i_arrd
);
Instruction
*
get_ret_inst_ptr2
=
GetElementPtrInst
::
Create
(
array_out
,
gvar_ret_inst_list
,
get_ret_inst_ptr_indices2
,
""
,
for_i_j
);
//arrayidx7
LoadInst
*
load_j_arrd
=
new
LoadInst
(
ptr_j
,
""
,
false
,
for_i_j
);
load_j_arrd
->
setAlignment
(
MaybeAlign
(
4
));
// 9
CastInst
*
casted_j_arrd
=
new
SExtInst
(
load_j_arrd
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
// idxprom12
std
::
vector
<
Value
*>
get_ret_inst_ptr_value_indices2
;
get_ret_inst_ptr_value_indices2
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_ret_inst_ptr_value_indices2
.
push_back
(
casted_j_arrd
);
Instruction
*
get_ret_inst_ptr_value2
=
GetElementPtrInst
::
Create
(
array_in
,
get_ret_inst_ptr2
,
get_ret_inst_ptr_value_indices2
,
""
,
for_i_j
);
//arrayidx13
LoadInst
*
load_ret_inst_value
=
new
LoadInst
(
get_ret_inst_ptr_value2
,
""
,
false
,
for_i_j
);
load_ret_inst_value
->
setAlignment
(
MaybeAlign
(
1
));
// 10
CastInst
*
cast_value
=
new
SExtInst
(
load_ret_inst_value
,
IntegerType
::
get
(
mod
->
getContext
(),
32
),
""
,
for_i_j
);
// conv
BinaryOperator
*
xor_1
=
BinaryOperator
::
Create
(
Instruction
::
Xor
,
cast_value
,
const_int_xordata
,
""
,
for_i_j
);
// xor
CastInst
*
trunc_value
=
new
TruncInst
(
xor_1
,
IntegerType
::
get
(
mod
->
getContext
(),
8
),
""
,
for_i_j
);
// conv14
LoadInst
*
load_i_kera
=
new
LoadInst
(
ptr_i
,
""
,
false
,
for_i_j
);
load_i_kera
->
setAlignment
(
MaybeAlign
(
4
));
// 11
CastInst
*
casted_i_kera
=
new
SExtInst
(
load_i_kera
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
// idxprom15
std
::
vector
<
Value
*>
get_ret_inst_ptr_indices3
;
get_ret_inst_ptr_indices3
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_ret_inst_ptr_indices3
.
push_back
(
casted_i_kera
);
Instruction
*
get_ret_inst_ptr3
=
GetElementPtrInst
::
Create
(
array_out
,
gvar_ret_inst_list
,
get_ret_inst_ptr_indices3
,
""
,
for_i_j
);
//arrayidx16
LoadInst
*
load_j_kera
=
new
LoadInst
(
ptr_j
,
""
,
false
,
for_i_j
);
load_j_kera
->
setAlignment
(
MaybeAlign
(
4
));
// 12
CastInst
*
casted_j_kera
=
new
SExtInst
(
load_j_kera
,
IntegerType
::
get
(
mod
->
getContext
(),
64
),
""
,
for_i_j
);
// idxprom17
std
::
vector
<
Value
*>
get_ret_inst_ptr_value_indices3
;
get_ret_inst_ptr_value_indices3
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
get_ret_inst_ptr_value_indices3
.
push_back
(
casted_j_kera
);
Instruction
*
get_ret_inst_ptr_value3
=
GetElementPtrInst
::
Create
(
array_in
,
get_ret_inst_ptr3
,
get_ret_inst_ptr_value_indices3
,
""
,
for_i_j
);
//arrayidx18
StoreInst
*
store_xor_data
=
new
StoreInst
(
trunc_value
,
get_ret_inst_ptr_value3
,
false
,
for_i_j
);
store_xor_data
->
setAlignment
(
MaybeAlign
(
1
));
/*
std::vector<Value *> ptr_41_indices;
ptr_41_indices.push_back(const_int_0);
ptr_41_indices.push_back(casted_j_64_1);
Instruction *ptr_41 = GetElementPtrInst::Create(
array_in, ptr_40, ptr_41_indices, "", for_i_j);
LoadInst *load_j_2 = new LoadInst(ptr_j, "", false, for_i_j);
load_j_2->setAlignment(MaybeAlign(4));
CastInst *casted_j_64_1 = new SExtInst(
load_j_2, IntegerType::get(mod->getContext(), 64), "", for_i_j);
LoadInst *load_i_2 = new LoadInst(ptr_i, "", false, for_i_j);
load_i_2->setAlignment(MaybeAlign(4));
CastInst *casted_i_64_1 = new SExtInst(
load_i_2, IntegerType::get(mod->getContext(), 64), "", for_i_j);
std::vector<Value *> get_retblock_array_indices;
get_retblock_array_indices.push_back(const_int_0);
get_retblock_array_indices.push_back(casted_i_64_1);
Instruction *get_retblock =
GetElementPtrInst::Create(array_retblock, gvar_array_retblock,
get_retblock_array_indices, "", for_i_j);
LoadInst *load_array = new LoadInst(get_retblock, "", false, for_i_j);
load_array->setAlignment(MaybeAlign(4));
Instruction *get_retblock_element = GetElementPtrInst::Create(
cast<PointerType>(load_array->getType()->getScalarType())
->getElementType(),
load_array, load_j_2, "arrayidx1", for_i_j);
LoadInst *load_elem =
new LoadInst(get_retblock_element, "", false, for_i_j);
load_elem->setAlignment(MaybeAlign(1));
std::vector<Value *> ptr_40_indices;
ptr_40_indices.push_back(const_int_0);
ptr_40_indices.push_back(casted_i_64_1);
Instruction *ptr_40 = GetElementPtrInst::Create(
array_out, gvar_ret_inst_list, ptr_40_indices, "", for_i_j);
std::vector<Value *> ptr_41_indices;
ptr_41_indices.push_back(const_int_0);
ptr_41_indices.push_back(casted_j_64_1);
Instruction *ptr_41 = GetElementPtrInst::Create(
array_in, ptr_40, ptr_41_indices, "", for_i_j);
StoreInst *store_elem = new StoreInst(load_elem, ptr_41, false, for_i_j);
store_elem->setAlignment(MaybeAlign(1));
LoadInst *int8_42 = new LoadInst(ptr_41, "", false, for_i_j);
int8_42->setAlignment(MaybeAlign(1));
CastInst *int32_43 = new SExtInst(
int8_42, IntegerType::get(mod->getContext(), 32), "", for_i_j);
BinaryOperator *int32_44 = BinaryOperator::Create(
Instruction::Xor, int32_43, const_int_xordata, "", for_i_j);
CastInst *int8_45 = new TruncInst(
int32_44, IntegerType::get(mod->getContext(), 8), "", for_i_j);
LoadInst *int32_46 = new LoadInst(ptr_j, "", false, for_i_j);
int32_46->setAlignment(MaybeAlign(4));
CastInst *int64_47 = new SExtInst(
int32_46, IntegerType::get(mod->getContext(), 64), "", for_i_j);
LoadInst *int32_48 = new LoadInst(ptr_i, "", false, for_i_j);
int32_48->setAlignment(MaybeAlign(4));
CastInst *int64_49 = new SExtInst(
int32_48, IntegerType::get(mod->getContext(), 64), "", for_i_j);
std::vector<Value *> ptr_50_indices;
ptr_50_indices.push_back(const_int_0);
ptr_50_indices.push_back(int64_49);
Instruction *ptr_50 = GetElementPtrInst::Create(
array_out, gvar_ret_inst_list, ptr_50_indices, "", for_i_j);
std::vector<Value *> ptr_51_indices;
ptr_51_indices.push_back(const_int_0);
ptr_51_indices.push_back(int64_47);
Instruction *ptr_51 = GetElementPtrInst::Create(
array_in, ptr_50, ptr_51_indices, "", for_i_j);
StoreInst *void_52 = new StoreInst(int8_45, ptr_51, false, for_i_j);
void_52->setAlignment(MaybeAlign(4));
*/
BranchInst
::
Create
(
for_j_add
,
for_i_j
);
// for_j_add
LoadInst
*
load_j_5
=
new
LoadInst
(
ptr_j
,
""
,
false
,
for_j_add
);
load_j_5
->
setAlignment
(
MaybeAlign
(
4
));
BinaryOperator
*
add_j_1
=
BinaryOperator
::
Create
(
Instruction
::
Add
,
load_j_5
,
const_int_1
,
""
,
for_j_add
);
StoreInst
*
void_56
=
new
StoreInst
(
add_j_1
,
ptr_j
,
false
,
for_j_add
);
void_56
->
setAlignment
(
MaybeAlign
(
4
));
BranchInst
::
Create
(
for_j_cond
,
for_j_add
);
// for_i_add
LoadInst
*
load_i_5
=
new
LoadInst
(
ptr_i
,
""
,
false
,
for_i_add
);
load_i_5
->
setAlignment
(
MaybeAlign
(
4
));
BinaryOperator
*
add_i_1
=
BinaryOperator
::
Create
(
Instruction
::
Add
,
load_i_5
,
const_int_1
,
""
,
for_i_add
);
StoreInst
*
stor_i_2
=
new
StoreInst
(
add_i_1
,
ptr_i
,
false
,
for_i_add
);
stor_i_2
->
setAlignment
(
MaybeAlign
(
4
));
BranchInst
::
Create
(
for_i_cond
,
for_i_add
);
/*
Function *Func_Reset_Handler = mod->getFunction("Func_Reset_Handler");
if (!Func_Reset_Handler) {
Func_Reset_Handler =
Function::Create(Func_deobfus_type, GlobalValue::ExternalLinkage,
"Reset_Handler", mod);
Func_Reset_Handler->setCallingConv(CallingConv::C);
}
CallInst *call_handler =
CallInst::Create(Func_Reset_Handler, "", for_i_end);
call_handler->setCallingConv(CallingConv::C);
call_handler->setTailCall(false);
AttributeList int32_25_PAL;
call_handler->setAttributes(int32_25_PAL); */
// for_i_end
ReturnInst
::
Create
(
mod
->
getContext
(),
const_int_0
,
for_i_end
);
}
/* else {
GlobalVariable *gvar_ret_inst_list =
new GlobalVariable(*mod, array_out, false,
GlobalValue::ExternalLinkage, 0, "ret_inst_list");
gvar_ret_inst_list->setAlignment(MaybeAlign(16));
} */
if
(
F
.
getName
().
equals
(
"main"
))
{
for
(
auto
&
BB
:
F
)
{
for
(
auto
&
I
:
BB
)
{
CallInst
*
int32_25
=
CallInst
::
Create
(
Func_deobfus
,
""
,
&
I
);
int32_25
->
setCallingConv
(
CallingConv
::
C
);
int32_25
->
setTailCall
(
false
);
AttributeList
int32_25_PAL
;
int32_25
->
setAttributes
(
int32_25_PAL
);
break
;
}
break
;
}
}
errs
()
<<
F
.
getName
()
<<
"
\n
"
;
if
(
!
F
.
getName
().
contains
(
"func_deobfus"
)
&&
!
F
.
getName
().
contains
(
"__cxx"
)
&&
!
F
.
getName
().
contains
(
"_GLOBAL"
))
{
errs
()
<<
F
.
getName
()
<<
"
\n
"
;
for
(
auto
&
BB
:
F
)
{
if
(
BB
.
getName
().
equals
(
"obfuscatedreturn"
))
{
GlobalVariable
*
gvar_ret_inst_list
=
mod
->
getGlobalVariable
(
"ret_inst_list"
);
PointerType
*
PointerTy_31
=
PointerType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
8
),
0
);
BasicBlock
*
jmp_to
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"jmp_to"
,
&
F
,
&
BB
);
BasicBlock
*
ret_jmp
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"ret_jmp"
,
&
F
,
&
BB
);
//AllocaInst *ptr_array =
// new AllocaInst(array_out, NULL, "ptr_ret_array", jmp_to);
AllocaInst
*
ptr_ptr3
=
new
AllocaInst
(
PointerTy_31
,
NULL
,
"ret_ptr_jmp"
,
jmp_to
);
ptr_ptr3
->
setAlignment
(
MaybeAlign
(
8
));
for
(
BasicBlock
*
preds
:
predecessors
(
&
BB
))
{
preds
->
getTerminator
()
->
eraseFromParent
();
BranchInst
::
Create
(
jmp_to
,
preds
);
}
std
::
vector
<
Constant
*>
const_ptr_14_indices
;
const_ptr_14_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
const_ptr_14_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
std
::
to_string
(
num_this_function
)),
10
)));
const_ptr_14_indices
.
push_back
(
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
64
,
StringRef
(
"0"
),
10
)));
Constant
*
const_ptr_14
=
ConstantExpr
::
getGetElementPtr
(
array_out
,
gvar_ret_inst_list
,
const_ptr_14_indices
);
/*
std::vector<Value *> ptr_175_indices;
ptr_175_indices.push_back(const_int_0);
ptr_175_indices.push_back(ConstantInt::get(
mod->getContext(),
APInt(32, StringRef(std::to_string(num_this_function)), 10)));
errs() << num_this_function << "\n";
Instruction *ptr_175 = GetElementPtrInst::Create(
array_out, gvar_ret_inst_list, ptr_175_indices, "", jmp_to);
std::vector<Value *> ptr_176_indices;
ptr_176_indices.push_back(const_int_0);
ptr_176_indices.push_back(const_int_0);
Instruction *ptr_176 = GetElementPtrInst::Create(
array_in, ptr_175, ptr_176_indices, "", jmp_to);
*/
StoreInst
*
void_177
=
new
StoreInst
(
const_ptr_14
,
ptr_ptr3
,
false
,
jmp_to
);
void_177
->
setAlignment
(
MaybeAlign
(
8
));
LoadInst
*
ptr_178
=
new
LoadInst
(
ptr_ptr3
,
""
,
false
,
jmp_to
);
ptr_178
->
setAlignment
(
MaybeAlign
(
8
));
BranchInst
::
Create
(
ret_jmp
,
jmp_to
);
PHINode
*
ptr_181
=
PHINode
::
Create
(
PointerTy_31
,
1
,
""
,
ret_jmp
);
ptr_181
->
addIncoming
(
ptr_178
,
jmp_to
);
IndirectBrInst
*
void_182
=
IndirectBrInst
::
Create
(
ptr_181
,
1
,
ret_jmp
);
void_182
->
addDestination
(
&
BB
);
}
}
}
/*
Module* mod = F.getParent();
ArrayType* return_array = ArrayType::get(IntegerType::get(mod->getContext(),
8), 12); PointerType* return_array_ptr = PointerType::get(return_array, 0);
PointerType* ret_func_ptr =
PointerType::get(IntegerType::get(mod->getContext(), 8), 0); ConstantInt*
const_int_1 = ConstantInt::get(mod->getContext(), APInt(32, StringRef("1"),
10)); ConstantInt* const_int_0 = ConstantInt::get(mod->getContext(),
APInt(32, StringRef("0"), 10)); ConstantInt* const_int_20 =
ConstantInt::get(mod->getContext(), APInt(32, StringRef("12"), 10));
ConstantInt* const_int32_133 = ConstantInt::get(mod->getContext(), APInt(32,
StringRef("133"), 10)); AllocaInst* ptr_ret_array; AllocaInst* ptr_this_ret;
AllocaInst* ret_array_ptr;
AllocaInst* ptr_i;
std::vector<Instruction *> instructions;
std::vector<BasicBlock *> RetBlocks;
bool inserted = false;
bool splitted = false;
for (auto &BB : F) {
for (auto &I : BB) {
if(!inserted) {
ptr_ret_array = new AllocaInst(return_array, NULL, "ret_ptr",
&I); ptr_ret_array->setAlignment(MaybeAlign(1)); ptr_this_ret = new
AllocaInst(ret_func_ptr, NULL, "ptr", &I); ret_array_ptr = new
AllocaInst(ret_func_ptr, NULL, "ptr2", &I); ptr_i = new
AllocaInst(IntegerType::get(mod->getContext(), 32), NULL, "i", &I);
inserted=true;
IndirectBrInst *a;
num_retblocks
=
retblocks
.
size
();
ArrayType
*
array_in
=
ArrayType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
8
),
20
);
ArrayType
*
array_out
=
ArrayType
::
get
(
array_in
,
50
);
PointerType
*
array_ptr
=
PointerType
::
get
(
array_out
,
0
);
ConstantInt
*
const_int_0
=
ConstantInt
::
get
(
mod
->
getContext
(),
APInt
(
32
,
StringRef
(
"0"
),
10
));
std
::
vector
<
Type
*>
Func_deobfus_type_args
;
FunctionType
*
Func_deobfus_type
=
FunctionType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
32
),
Func_deobfus_type_args
,
false
);
Function
*
Func_deobfus
=
mod
->
getFunction
(
"func_deobfus"
);
if
(
!
Func_deobfus
)
{
Func_deobfus
=
Function
::
Create
(
Func_deobfus_type
,
GlobalValue
::
ExternalLinkage
,
"func_deobfus"
,
mod
);
Func_deobfus
->
setCallingConv
(
CallingConv
::
C
);
AttributeList
Func_deobfus_att_list
;
SmallVector
<
AttributeList
,
4
>
Attrs
;
AttributeList
PAS
;
AttrBuilder
B
;
B
.
addAttribute
(
Attribute
::
NoInline
);
B
.
addAttribute
(
Attribute
::
NoRecurse
);
B
.
addAttribute
(
Attribute
::
NoUnwind
);
B
.
addAttribute
(
Attribute
::
OptimizeNone
);
PAS
=
AttributeList
::
get
(
mod
->
getContext
(),
~
0U
,
B
);
Attrs
.
push_back
(
PAS
);
Func_deobfus_att_list
=
AttributeList
::
get
(
mod
->
getContext
(),
Attrs
);
Func_deobfus
->
setAttributes
(
Func_deobfus_att_list
);
if
(
Func_deobfus
->
size
()
==
0
)
{
GlobalVariable
*
gvar_ret_inst_list
=
new
GlobalVariable
(
*
mod
,
array_out
,
false
,
GlobalValue
::
ExternalLinkage
,
0
,
"ret_inst_list"
);
gvar_ret_inst_list
->
setAlignment
(
MaybeAlign
(
16
));
BasicBlock
*
obfus_entry
=
BasicBlock
::
Create
(
mod
->
getContext
(),
"entry"
,
Func_deobfus
);
for
(
size_t
i
=
0
;
i
<
num_retblocks
;
i
++
)
{
PointerType
*
ret_func_ptr
=
PointerType
::
get
(
IntegerType
::
get
(
mod
->
getContext
(),
8
),
0
);
AllocaInst
*
ptr_this_ret
=
new
AllocaInst
(
ret_func_ptr
,
NULL
,
"ptr"
,
obfus_entry
);;
StoreInst
*
void_17
=
new
StoreInst
(
retblocks
[
i
],
ptr_this_ret
,
false
,
obfus_entry
);
}
ReturnInst
::
Create
(
mod
->
getContext
(),
const_int_0
,
obfus_entry
);
}
}
/*
Module* mod = F.getParent();
ArrayType* return_array = ArrayType::get(IntegerType::get(mod->getContext(), 8), 12);
PointerType* return_array_ptr = PointerType::get(return_array, 0);
PointerType* ret_func_ptr = PointerType::get(IntegerType::get(mod->getContext(), 8), 0);
ConstantInt* const_int_1 = ConstantInt::get(mod->getContext(), APInt(32, StringRef("1"), 10));
ConstantInt* const_int_0 = ConstantInt::get(mod->getContext(), APInt(32, StringRef("0"), 10));
ConstantInt* const_int_20 = ConstantInt::get(mod->getContext(), APInt(32, StringRef("12"), 10));
ConstantInt* const_int32_133 = ConstantInt::get(mod->getContext(), APInt(32, StringRef("133"), 10));
AllocaInst* ptr_ret_array;
AllocaInst* ptr_this_ret;
AllocaInst* ret_array_ptr;
AllocaInst* ptr_i;
std::vector<Instruction *> instructions;
std::vector<BasicBlock *> RetBlocks;
bool inserted = false;
bool splitted = false;
for (auto &BB : F) {
for (auto &I : BB) {
if(!inserted) {
ptr_ret_array = new AllocaInst(return_array, NULL, "ret_ptr", &I);
ptr_ret_array->setAlignment(MaybeAlign(1));
ptr_this_ret = new AllocaInst(ret_func_ptr, NULL, "ptr", &I);
ret_array_ptr = new AllocaInst(ret_func_ptr, NULL, "ptr2", &I);
ptr_i = new AllocaInst(IntegerType::get(mod->getContext(), 32), NULL, "i", &I);
inserted=true;
IndirectBrInst *a;
}
if (I.getOpcode() == Instruction::Ret) {
instructions.push_back(&I);
}
}
if (I.getOpcode() == Instruction::Ret) {
instructions.push_back(&I);
}
for (auto &I : instructions) {
BasicBlock *BB = I->getParent();
// One Instruction Basic Block has only one ret instructions
if (!BB->size() < 2)
{
BasicBlock *retblock = BB->splitBasicBlock(I->getIterator(), BB->getName() + ".RetBlock");
RetBlocks.push_back(retblock);
} else {
RetBlocks.push_back(BB);
}
}
}
for (auto &I : instructions) {
BasicBlock *BB = I->getParent();
// One Instruction Basic Block has only one ret instructions
if (!BB->size() < 2)
{
BasicBlock *retblock = BB->splitBasicBlock(I->getIterator(),
BB->getName() + ".RetBlock"); RetBlocks.push_back(retblock); } else {
RetBlocks.push_back(BB);
}
}
for (auto &BB : RetBlocks) {
Constant* retBlockAddress = BlockAddress::get(BB);
Module* M = F.getParent();
for (auto curFref = M->getFunctionList().begin(),
endFref = M->getFunctionList().end();
curFref != endFref; ++curFref) {
for (auto& B: curFref->getBasicBlockList()) {
StoreInst* asdf = new StoreInst(retBlockAddress, ptr_this_ret,
false, &B); asdf->setAlignment(MaybeAlign(4)); break;
}
for (auto &BB : RetBlocks) {
Constant* retBlockAddress = BlockAddress::get(BB);
Module* M = F.getParent();
for (auto curFref = M->getFunctionList().begin(),
endFref = M->getFunctionList().end();
curFref != endFref; ++curFref) {
for (auto& B: curFref->getBasicBlockList()) {
StoreInst* asdf = new StoreInst(retBlockAddress, ptr_this_ret, false, &B);
asdf->setAlignment(MaybeAlign(4));
break;
}
}
BasicBlock* decrypt_start = BasicBlock::Create(mod->getContext(), "dec_start", &F, BB);
for (BasicBlock* preds : predecessors(BB)) {
preds->getTerminator()->eraseFromParent();
BranchInst::Create(decrypt_start, preds);
}
std::vector<Value*> ptr_to_retarray_indices;
ptr_to_retarray_indices.push_back(const_int_0);
ptr_to_retarray_indices.push_back(const_int_0);
GetElementPtrInst* ptr_to_retarray = GetElementPtrInst::Create(return_array, ptr_ret_array, ptr_to_retarray_indices, "arrayidx", decrypt_start);
ptr_to_retarray->setIsInBounds(true);
StoreInst* store_to_ret_ptr = new StoreInst(ptr_to_retarray, ret_array_ptr, false, decrypt_start);
store_to_ret_ptr->setAlignment(MaybeAlign(4));
StoreInst* void_17 = new StoreInst(retBlockAddress, ptr_this_ret, false, decrypt_start);
ptr_this_ret->setAlignment(MaybeAlign(4));
ret_array_ptr->setAlignment(MaybeAlign(4));
ptr_i->setAlignment(MaybeAlign(4));
void_17->setAlignment(MaybeAlign(4));
StoreInst* store_i_0 = new StoreInst(const_int_0, ptr_i, false, decrypt_start);
store_i_0->setAlignment(MaybeAlign(4));
BasicBlock* decrypt_cond = BasicBlock::Create(mod->getContext(), "dec_cond", &F, BB);
BranchInst::Create(decrypt_cond, decrypt_start);
LoadInst* ldr_i_data = new LoadInst(ptr_i, "", false, decrypt_cond);
ldr_i_data->setAlignment(MaybeAlign(4));
ICmpInst* cmp_i_with_20 = new ICmpInst(*decrypt_cond, ICmpInst::ICMP_SLT, ldr_i_data, const_int_20, "cmp");
BasicBlock* decrypt_ing = BasicBlock::Create(mod->getContext(), "dec_ing", &F, BB);
BasicBlock* decrypt_add = BasicBlock::Create(mod->getContext(), "dec_add", &F, BB);
BasicBlock* decrypt_end = BasicBlock::Create(mod->getContext(), "dec_end", &F, BB);
BranchInst::Create(decrypt_ing, decrypt_end, cmp_i_with_20, decrypt_cond);
LoadInst* ldr_i_data_2 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_2->setAlignment(MaybeAlign(4));
LoadInst* ldr_ptr_this_ret = new LoadInst(ptr_this_ret, "", false, decrypt_ing);
ldr_ptr_this_ret->setAlignment(MaybeAlign(4));
GetElementPtrInst* get_func_ptr_idx = GetElementPtrInst::Create(cast<PointerType>(ldr_ptr_this_ret->getType()->getScalarType())->getElementType(), ldr_ptr_this_ret, ldr_i_data_2, "arrayidx1", decrypt_ing);
get_func_ptr_idx->setIsInBounds(true);
LoadInst* ldr_func_ptr_idx = new LoadInst(get_func_ptr_idx, "", false, decrypt_ing);
ldr_func_ptr_idx->setAlignment(MaybeAlign(1));
LoadInst* ldr_i_data_3 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_3->setAlignment(MaybeAlign(4));
std::vector<Value*> ptr_retn_array_indices;
ptr_retn_array_indices.push_back(const_int_0);
ptr_retn_array_indices.push_back(ldr_i_data_3);
GetElementPtrInst* get_retn_array_data_idx = GetElementPtrInst::Create(return_array, ptr_ret_array, ptr_retn_array_indices, "arrayidx2", decrypt_ing);
get_retn_array_data_idx->setIsInBounds(true);
StoreInst* str_retn_array_data_idx = new StoreInst(ldr_func_ptr_idx, get_retn_array_data_idx, false, decrypt_ing);
str_retn_array_data_idx->setAlignment(MaybeAlign(1));
LoadInst* ldr_i_data_4 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_4->setAlignment(MaybeAlign(4));
std::vector<Value*> ptr_retn_array_indices2;
ptr_retn_array_indices2.push_back(const_int_0);
ptr_retn_array_indices2.push_back(ldr_i_data_4);
GetElementPtrInst* get_retn_array_data_idx2 = GetElementPtrInst::Create(return_array, ptr_ret_array, ptr_retn_array_indices2, "arrayidx3", decrypt_ing);
get_retn_array_data_idx2->setIsInBounds(true);
LoadInst* ldr_retn_array_data_idx2 = new LoadInst(get_retn_array_data_idx2, "", false, decrypt_ing);
ldr_retn_array_data_idx2->setAlignment(MaybeAlign(1));
CastInst* cast_retn_array_data_idx2 = new ZExtInst(ldr_retn_array_data_idx2, IntegerType::get(mod->getContext(), 32), "conv", decrypt_ing);
BinaryOperator* xor_retn_array_data_idx2 = BinaryOperator::Create(Instruction::Xor, cast_retn_array_data_idx2, const_int32_133, "xor", decrypt_ing);
CastInst* trun_retn_array_data_idx2 = new TruncInst(xor_retn_array_data_idx2, IntegerType::get(mod->getContext(), 8), "conv4", decrypt_ing);
LoadInst* ldr_i_data_5 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_5->setAlignment(MaybeAlign(4));
std::vector<Value*> ptr_retn_array_indices4;
ptr_retn_array_indices4.push_back(const_int_0);
ptr_retn_array_indices4.push_back(ldr_i_data_5);
GetElementPtrInst* get_retn_array_data_idx4 = GetElementPtrInst::Create(return_array, ptr_ret_array, ptr_retn_array_indices4, "arrayidx5", decrypt_ing);
get_retn_array_data_idx4->setIsInBounds(true);
StoreInst* str_retn_array_data_idx4 = new StoreInst(trun_retn_array_data_idx2, get_retn_array_data_idx4, false, decrypt_ing);
str_retn_array_data_idx4->setAlignment(MaybeAlign(1));
BranchInst::Create(decrypt_add, decrypt_ing);
LoadInst* ldr_i_data_6 = new LoadInst(ptr_i, "", false, decrypt_add);
ldr_i_data_6->setAlignment(MaybeAlign(4));
BinaryOperator* add_i_data_4 = BinaryOperator::Create(Instruction::Add, ldr_i_data_6, const_int_1, "", decrypt_add);
StoreInst* str_i_data_4 = new StoreInst(add_i_data_4, ptr_i, false, decrypt_add);
str_i_data_4->setAlignment(MaybeAlign(4));
BranchInst::Create(decrypt_cond, decrypt_add);
LoadInst* ldr_ret_array = new LoadInst(ret_array_ptr, "", false, decrypt_end);
ldr_ret_array->setAlignment(MaybeAlign(4));
BasicBlock* dec_jmp = BasicBlock::Create(mod->getContext(), "dec_jmp", &F, BB);
BranchInst::Create(dec_jmp, decrypt_end);
}
BasicBlock* decrypt_start = BasicBlock::Create(mod->getContext(),
"dec_start", &F, BB); for (BasicBlock* preds : predecessors(BB)) {
preds->getTerminator()->eraseFromParent();
BranchInst::Create(decrypt_start, preds);
}
PHINode* ptr_40 = PHINode::Create(ret_func_ptr, 1, "", dec_jmp);
ptr_40->addIncoming(ldr_ret_array, decrypt_end);
IndirectBrInst *void_41 = IndirectBrInst::Create(ldr_ret_array, 1, dec_jmp);
void_41->addDestination(BB);
errs().write_escaped(F.getName()) << " " << F.getParent()->getName() << '\n';
}
*/
return
true
;
}
};
// end of struct Hello
}
// end of anonymous namespace
std::vector<Value*> ptr_to_retarray_indices;
ptr_to_retarray_indices.push_back(const_int_0);
ptr_to_retarray_indices.push_back(const_int_0);
GetElementPtrInst* ptr_to_retarray =
GetElementPtrInst::Create(return_array, ptr_ret_array,
ptr_to_retarray_indices, "arrayidx", decrypt_start);
ptr_to_retarray->setIsInBounds(true);
StoreInst* store_to_ret_ptr = new StoreInst(ptr_to_retarray,
ret_array_ptr, false, decrypt_start);
store_to_ret_ptr->setAlignment(MaybeAlign(4));
StoreInst* void_17 = new StoreInst(retBlockAddress, ptr_this_ret, false,
decrypt_start);
ptr_this_ret->setAlignment(MaybeAlign(4));
ret_array_ptr->setAlignment(MaybeAlign(4));
ptr_i->setAlignment(MaybeAlign(4));
void_17->setAlignment(MaybeAlign(4));
StoreInst* store_i_0 = new StoreInst(const_int_0, ptr_i, false,
decrypt_start); store_i_0->setAlignment(MaybeAlign(4));
BasicBlock* decrypt_cond = BasicBlock::Create(mod->getContext(),
"dec_cond", &F, BB);
BranchInst::Create(decrypt_cond, decrypt_start);
LoadInst* ldr_i_data = new LoadInst(ptr_i, "", false, decrypt_cond);
ldr_i_data->setAlignment(MaybeAlign(4));
ICmpInst* cmp_i_with_20 = new ICmpInst(*decrypt_cond,
ICmpInst::ICMP_SLT, ldr_i_data, const_int_20, "cmp");
BasicBlock* decrypt_ing = BasicBlock::Create(mod->getContext(),
"dec_ing", &F, BB); BasicBlock* decrypt_add =
BasicBlock::Create(mod->getContext(), "dec_add", &F, BB); BasicBlock*
decrypt_end = BasicBlock::Create(mod->getContext(), "dec_end", &F, BB);
BranchInst::Create(decrypt_ing, decrypt_end, cmp_i_with_20,
decrypt_cond);
LoadInst* ldr_i_data_2 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_2->setAlignment(MaybeAlign(4));
LoadInst* ldr_ptr_this_ret = new LoadInst(ptr_this_ret, "", false,
decrypt_ing); ldr_ptr_this_ret->setAlignment(MaybeAlign(4));
GetElementPtrInst* get_func_ptr_idx =
GetElementPtrInst::Create(cast<PointerType>(ldr_ptr_this_ret->getType()->getScalarType())->getElementType(),
ldr_ptr_this_ret, ldr_i_data_2, "arrayidx1", decrypt_ing);
get_func_ptr_idx->setIsInBounds(true);
LoadInst* ldr_func_ptr_idx = new LoadInst(get_func_ptr_idx, "", false,
decrypt_ing); ldr_func_ptr_idx->setAlignment(MaybeAlign(1));
LoadInst* ldr_i_data_3 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_3->setAlignment(MaybeAlign(4));
std::vector<Value*> ptr_retn_array_indices;
ptr_retn_array_indices.push_back(const_int_0);
ptr_retn_array_indices.push_back(ldr_i_data_3);
GetElementPtrInst* get_retn_array_data_idx =
GetElementPtrInst::Create(return_array, ptr_ret_array,
ptr_retn_array_indices, "arrayidx2", decrypt_ing);
get_retn_array_data_idx->setIsInBounds(true);
StoreInst* str_retn_array_data_idx = new StoreInst(ldr_func_ptr_idx,
get_retn_array_data_idx, false, decrypt_ing);
str_retn_array_data_idx->setAlignment(MaybeAlign(1));
LoadInst* ldr_i_data_4 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_4->setAlignment(MaybeAlign(4));
std::vector<Value*> ptr_retn_array_indices2;
ptr_retn_array_indices2.push_back(const_int_0);
ptr_retn_array_indices2.push_back(ldr_i_data_4);
GetElementPtrInst* get_retn_array_data_idx2 =
GetElementPtrInst::Create(return_array, ptr_ret_array,
ptr_retn_array_indices2, "arrayidx3", decrypt_ing);
get_retn_array_data_idx2->setIsInBounds(true);
LoadInst* ldr_retn_array_data_idx2 = new
LoadInst(get_retn_array_data_idx2, "", false, decrypt_ing);
ldr_retn_array_data_idx2->setAlignment(MaybeAlign(1));
CastInst* cast_retn_array_data_idx2 = new
ZExtInst(ldr_retn_array_data_idx2, IntegerType::get(mod->getContext(), 32),
"conv", decrypt_ing); BinaryOperator* xor_retn_array_data_idx2 =
BinaryOperator::Create(Instruction::Xor, cast_retn_array_data_idx2,
const_int32_133, "xor", decrypt_ing);
CastInst* trun_retn_array_data_idx2 = new
TruncInst(xor_retn_array_data_idx2, IntegerType::get(mod->getContext(), 8),
"conv4", decrypt_ing);
LoadInst* ldr_i_data_5 = new LoadInst(ptr_i, "", false, decrypt_ing);
ldr_i_data_5->setAlignment(MaybeAlign(4));
std::vector<Value*> ptr_retn_array_indices4;
ptr_retn_array_indices4.push_back(const_int_0);
ptr_retn_array_indices4.push_back(ldr_i_data_5);
GetElementPtrInst* get_retn_array_data_idx4 =
GetElementPtrInst::Create(return_array, ptr_ret_array,
ptr_retn_array_indices4, "arrayidx5", decrypt_ing);
get_retn_array_data_idx4->setIsInBounds(true);
StoreInst* str_retn_array_data_idx4 = new
StoreInst(trun_retn_array_data_idx2, get_retn_array_data_idx4, false,
decrypt_ing); str_retn_array_data_idx4->setAlignment(MaybeAlign(1));
BranchInst::Create(decrypt_add, decrypt_ing);
LoadInst* ldr_i_data_6 = new LoadInst(ptr_i, "", false, decrypt_add);
ldr_i_data_6->setAlignment(MaybeAlign(4));
BinaryOperator* add_i_data_4 = BinaryOperator::Create(Instruction::Add,
ldr_i_data_6, const_int_1, "", decrypt_add); StoreInst* str_i_data_4 = new
StoreInst(add_i_data_4, ptr_i, false, decrypt_add);
str_i_data_4->setAlignment(MaybeAlign(4));
BranchInst::Create(decrypt_cond, decrypt_add);
LoadInst* ldr_ret_array = new LoadInst(ret_array_ptr, "", false,
decrypt_end); ldr_ret_array->setAlignment(MaybeAlign(4));
BasicBlock* dec_jmp = BasicBlock::Create(mod->getContext(), "dec_jmp",
&F, BB); BranchInst::Create(dec_jmp, decrypt_end);
PHINode* ptr_40 = PHINode::Create(ret_func_ptr, 1, "", dec_jmp);
ptr_40->addIncoming(ldr_ret_array, decrypt_end);
IndirectBrInst *void_41 = IndirectBrInst::Create(ldr_ret_array, 1,
dec_jmp); void_41->addDestination(BB); errs().write_escaped(F.getName()) <<
" " << F.getParent()->getName() << '\n';
}
*/
return
true
;
}
};
// end of struct Hello
}
// end of anonymous namespace
char
ReturnObfuscation
::
ID
=
0
;
static
RegisterPass
<
ReturnObfuscation
>
X
(
"rof"
,
"Hello World Pass"
,
false
/* Only looks at CFG */
,
false
/* Analysis Pass */
);
\ No newline at end of file
false
/* Only looks at CFG */
,
false
/* Analysis Pass */
);
\ No newline at end of file
...
...
진행보고서/[최종발표자료]_LLVM_기반의_Baremetal_IoT_디바이스_펌웨어_난독화.pptx
0 → 100644
View file @
ea70b69
No preview for this file type
진행보고서/[최종보고서]_LLVM_기반의_Baremetal_IoT_디바이스_펌웨어_난독화.docx
0 → 100644
View file @
ea70b69
No preview for this file type
Please
register
or
login
to post a comment