#ifndef MARKER_SDK_H
#define MARKER_SDK_H
#ifdef MARKER_SDK_EXPORTS
#define MARKER_API __declspec(dllexport)
#else
#define MARKER_API __declspec(dllimport)
#endif
// MARKER_SDK_EXPORTS
extern
"C"
{
MARKER_API
void
begin_encrypted
(
void
)
;
MARKER_API
void
end_encrypted
(
void
)
;
}
#endif
// MARKER_SDK_H

auto
find_sdk_imports
(
std
::
error_code
&
ec
)
{
auto
result
=
std
::
pair

{
}
;
// Получаем директории таблицы импортов
auto
imports_boundary
=
image_
->
get_directory
(
win
::
directory_entry_import
)
;
if
(
!
imports_boundary
)
{
ec
=
std
::
make_error_code
(
std
::
errc
::
no_such_file_or_directory
)
;
return
result
;
}
// Перебираем все импортируемые библиотеки
for
(
auto
library
=
reinterpret_cast

(
image_
->
rva_to_ptr
(
imports_boundary
->
rva
)
)
;
library
->
rva_name
;
++
library
)
{
// Получаем имя библиотеки
auto
library_name
=
std
::
string
{
reinterpret_cast

(
image_
->
rva_to_ptr
(
library
->
rva_name
)
)
}
;
if
(
library_name
!=
"Marker-SDK.dll"
)
continue
;
// Перебираем все импортируемые функции для библиотеки
for
(
auto
function
=
reinterpret_cast

(
image_
->
rva_to_ptr
(
library
->
rva_first_thunk
)
)
;
function
->
address
;
++
function
)
{
// Получаем имя функции
auto
function_name
=
std
::
string
{
reinterpret_cast

(
image_
->
rva_to_ptr
(
function
->
address
)
)
->
name
}
;
// Получаем смещение до функции в IAT
// Так же я конвертирую в смещение относительно начала буффера
auto
iat_address
=
image_
->
ptr_to_raw
(
function
)
;
if
(
function_name
==
"begin_encrypted"
)
{
result
.
first
=
iat_address
;
}
else
if
(
function_name
==
"end_encrypted"
)
{
result
.
second
=
iat_address
;
}
}
}
return
result
;
}