mirror of
https://github.com/stellarbear/YaraSharp
synced 2026-06-08 17:36:09 +00:00
stellarbear
110 lines
3.6 KiB
C++
110 lines
3.6 KiB
C++
#include "Stdafx.h"
|
|
|
|
// CCompiler
|
|
namespace YaraSharp
|
|
{
|
|
// Êîíñòðóêòîð
|
|
CCompiler::CCompiler(Dictionary<String^, Object^>^ ExternalVariables)
|
|
{
|
|
YR_COMPILER* TestCompiler;
|
|
ErrorUtility::ThrowOnError(yr_compiler_create(&TestCompiler));
|
|
Compiler = TestCompiler;
|
|
|
|
Errors = gcnew List<String^>();
|
|
|
|
SetCompilerExternals(ExternalVariables);
|
|
SetCompilerCallback();
|
|
}
|
|
// Äåñòðóêòîð
|
|
CCompiler::~CCompiler() { if (Compiler) yr_compiler_destroy(Compiler); }
|
|
|
|
|
|
void CCompiler::SetCompilerCallback()
|
|
{
|
|
YaraCompilerCallback^ CompilerCallback = gcnew YaraCompilerCallback(this, &CCompiler::HandleCompilerCallback);
|
|
GCHandle CallbackHandle = GCHandle::Alloc(CompilerCallback);
|
|
YR_COMPILER_CALLBACK_FUNC CallbackPointer = (YR_COMPILER_CALLBACK_FUNC)(Marshal::GetFunctionPointerForDelegate(CompilerCallback)).ToPointer();
|
|
yr_compiler_set_callback(Compiler, CallbackPointer, NULL);
|
|
}
|
|
// Äîáàâëåíèå ïðàâèë â êîëëåêöèþ èç ôàéëà
|
|
int CCompiler::AddFile(String^ FilePath)
|
|
{
|
|
FILE* File;
|
|
Errors->Clear();
|
|
auto NativePath = marshal_as<std::string>(FilePath);
|
|
|
|
auto FileOpenError = fopen_s(&File, NativePath.c_str(), "r");
|
|
|
|
if (FileOpenError)
|
|
ErrorUtility::ThrowOnError(String::Format("Îøèáêà îòêðûòèÿ ôàéëà: {0}", FileOpenError));
|
|
|
|
auto errors = yr_compiler_add_file(Compiler, File, nullptr, NativePath.c_str());
|
|
|
|
if (File) fclose(File);
|
|
|
|
return errors;
|
|
}
|
|
// Äîáàâëåíèå íåñêîëüêèõ ôàéëîâ
|
|
void CCompiler::AddFiles(List<String^>^ FilePathList)
|
|
{
|
|
for each (auto FilePath in FilePathList)
|
|
AddFile(FilePath);
|
|
}
|
|
// Ïîëó÷åíèå ñïèñêà ïðàâèë
|
|
CRules^ CCompiler::GetRules()
|
|
{
|
|
YR_RULES* Rules;
|
|
|
|
ErrorUtility::ThrowOnError(
|
|
yr_compiler_get_rules(Compiler, &Rules));
|
|
|
|
return gcnew CRules(Rules);
|
|
}
|
|
// Ïîëó÷åíèå ñïèñêà îøèáîê
|
|
List<String^>^ CCompiler::GetErrors()
|
|
{
|
|
return this->Errors;
|
|
}
|
|
// Óñòàíîâêà externals
|
|
void CCompiler::SetCompilerExternals(Dictionary<String^, Object^>^ ExternalVariables)
|
|
{
|
|
if (ExternalVariables)
|
|
{
|
|
marshal_context CTX;
|
|
|
|
for each (auto ExternalVariable in ExternalVariables)
|
|
{
|
|
const char* VariablePointer = CTX.marshal_as<const char*>(ExternalVariable.Key);
|
|
Type^ VariableType = ExternalVariable.Value->GetType();
|
|
int ExternalError = ERROR_SUCCESS;
|
|
|
|
if (VariableType == Boolean::typeid)
|
|
ExternalError = yr_compiler_define_boolean_variable(Compiler, VariablePointer, (bool)ExternalVariable.Value);
|
|
else if (VariableType == Double::typeid)
|
|
ExternalError = yr_compiler_define_float_variable(Compiler, VariablePointer, (double)ExternalVariable.Value);
|
|
else if (VariableType == Int64::typeid || VariableType == Int32::typeid)
|
|
ExternalError = yr_compiler_define_integer_variable(Compiler, VariablePointer, (Int64)ExternalVariable.Value);
|
|
else if (VariableType == String::typeid)
|
|
ExternalError = yr_compiler_define_string_variable(Compiler, VariablePointer, CTX.marshal_as<const char*>((String^)ExternalVariable.Value));
|
|
else
|
|
throw gcnew NotSupportedException(String::Format("Íåïîääåðæèâàåìûé òèï external variable: '{0}'", VariableType->Name));
|
|
|
|
if (ExternalError != ERROR_SUCCESS)
|
|
ErrorUtility::ThrowOnError("(Compiler) Íå óäàëîñü èíèöèàëèçèðîâàòü external variable");
|
|
}
|
|
}
|
|
}
|
|
|
|
// Îáðàáîòêà îòâåòà
|
|
void CCompiler::HandleCompilerCallback(int ErrorLevel, const char* Filename, int LineNumber, const char* Message, void* UserData)
|
|
{
|
|
UNREFERENCED_PARAMETER(ErrorLevel);
|
|
UNREFERENCED_PARAMETER(UserData);
|
|
|
|
auto msg = String::Format("{0} â ñòðîêå {1} â ôàéëå: {2}",
|
|
marshal_as<String^>(Message), LineNumber,
|
|
Filename ? marshal_as<String^>(Filename) : "[none]");
|
|
|
|
Errors->Add(msg);
|
|
}
|
|
} |