1cea516cf7
- Implemented MachineLearning class with ExtractFeatures method - Updated project files to include new machine learning source and header files - Modified main executable to call feature extraction - Updated VSCode settings to include additional C++ headers - Commented out previous file dumping code in main function
32 lines
766 B
C++
32 lines
766 B
C++
#pragma once
|
|
#define _CRT_SECURE_NO_WARNINGS
|
|
#include <iostream>
|
|
#include <iostream>
|
|
#include <winternl.h>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <unordered_map>
|
|
#include <ntstatus.h>
|
|
#include <ctime>
|
|
|
|
|
|
#include "unicorn/include/unicorn/unicorn.h"
|
|
#include "capstone/include/capstone/capstone.h"
|
|
#pragma comment(lib, "unicorn/unicorn.lib")
|
|
#pragma comment(lib, "capstone/capstone.lib")
|
|
#include "libpeconv/include/peconv.h"
|
|
#include "native_struct.h"
|
|
struct BasicPeInfo {
|
|
std::string inputFilePath;
|
|
bool isX64;
|
|
uint64_t RecImageBase;
|
|
uint64_t entryPoint;
|
|
uint64_t imageEnd;
|
|
bool isRelocated;
|
|
uint8_t* peBuffer;
|
|
size_t peSize;
|
|
PIMAGE_NT_HEADERS ntHead64;
|
|
PIMAGE_NT_HEADERS32 ntHead32;
|
|
};
|
|
#include "sandbox.h"
|
|
#include "ml.h" |