diff --git a/GilRanClient/GilRanClient.cpp b/GilRanClient/GilRanClient.cpp index 8285419..33e29d6 100644 --- a/GilRanClient/GilRanClient.cpp +++ b/GilRanClient/GilRanClient.cpp @@ -2,10 +2,16 @@ #include #include #include + +#include +#include #include #include +#include +#include #include "../Common/Common.h" +#include "Utils.h" #define CLIENT_DEFAULT_REQUEST_COUNT 4 #define CLIENT_DEFAULT_THREAD_COUNT 4 @@ -14,6 +20,7 @@ typedef struct _THREAD_CONTEXT { HANDLE hCommunicationPort; HANDLE hIoCompletionPort; + std::list arrFolderPath; } THREAD_CONTEXT, *PTHREAD_CONTEXT; typedef struct _FILTER_MESSAGE { @@ -49,8 +56,6 @@ DWORD ClientWorker( break; } - printf("\n\n------------------------------ \nPID: %p\nVolumeName: %ws\nFileName: %ws\n", pFilterMessage->Request.ProcessID, pFilterMessage->Request.VolumeName, pFilterMessage->Request.FilePath); - PORT_RESPONSE Response; Response.Header.Status = 0; Response.Header.MessageId = pFilterMessage->Header.MessageId; @@ -86,7 +91,8 @@ int _cdecl main( { UNREFERENCED_PARAMETER(argc); UNREFERENCED_PARAMETER(argv); - + + THREAD_CONTEXT ThreadContext; HANDLE hCommunicationPort; HRESULT hResult = FilterConnectCommunicationPort( PORT_NAME, @@ -106,7 +112,6 @@ int _cdecl main( CLIENT_DEFAULT_THREAD_COUNT ); - THREAD_CONTEXT ThreadContext; ThreadContext.hCommunicationPort = hCommunicationPort; ThreadContext.hIoCompletionPort = hIoCompletionPort; @@ -159,10 +164,46 @@ int _cdecl main( } } } - WaitForMultipleObjectsEx(loop, hThread, TRUE, INFINITE, FALSE); + + int Command = TRUE; + while (Command) { + std::wcout << std::endl; + std::wcout << "================================================================================" << std::endl; + std::wcout << "| GilRan Ransomware Solution v0.1a |" << std::endl; + std::wcout << "================================================================================" << std::endl; + std::wcout << "| [1] DMZ Folder -- SHOW |" << std::endl; + std::wcout << "| [2] DMZ Folder -- ADD |" << std::endl; + std::wcout << "| [0] Exit |" << std::endl; + std::wcout << "================================================================================" << std::endl; + std::wcout << "Select: "; + + std::cin >> Command; + if (!std::cin) { + std::cin.clear(); + std::cin.ignore(INT_MAX, '\n'); + Command = TRUE; + } + + switch (Command) { + case 1: + for (auto FolderPath : ThreadContext.arrFolderPath) { + std::wcout << FolderPath << std::endl; + } + break; + case 2: + PWCHAR Path = new WCHAR[MAX_PATH]; + + std::wcout << L"Input folder path to make DMZ Folder" << std::endl; + std::wcout << L">"; + std::wcin >> Path; + + ThreadContext.arrFolderPath.push_back(Path); + break; + } + } CloseHandle(hCommunicationPort); CloseHandle(hIoCompletionPort); - + return S_OK; } \ No newline at end of file diff --git a/GilRanClient/GilRanClient.vcxproj b/GilRanClient/GilRanClient.vcxproj index 22ded8d..3bc7521 100644 --- a/GilRanClient/GilRanClient.vcxproj +++ b/GilRanClient/GilRanClient.vcxproj @@ -36,9 +36,11 @@ + + {AC5C0863-C3F8-4628-94E2-4F009FB92427} @@ -127,7 +129,7 @@ MultiThreadedDebugDLL - %(AdditionalDependencies);onecoreuap.lib + %(AdditionalDependencies);onecoreuap.lib;ole32.lib;shell32.lib;user32.lib @@ -135,7 +137,7 @@ WINAPI_FAMILY=WINAPI_FAMILY_DESKTOP_APP;WINAPI_PARTITION_DESKTOP=1;WINAPI_PARTITION_SYSTEM=1;WINAPI_PARTITION_APP=1;WINAPI_PARTITION_PC_APP=1;%(PreprocessorDefinitions) - %(AdditionalDependencies);onecoreuap.lib + %(AdditionalDependencies);onecoreuap.lib;ole32.lib;shell32.lib;user32.lib @@ -144,7 +146,7 @@ MultiThreadedDebugDLL - %(AdditionalDependencies);onecoreuap.lib + %(AdditionalDependencies);onecoreuap.lib;ole32.lib;shell32.lib;user32.lib @@ -152,7 +154,7 @@ WINAPI_FAMILY=WINAPI_FAMILY_DESKTOP_APP;WINAPI_PARTITION_DESKTOP=1;WINAPI_PARTITION_SYSTEM=1;WINAPI_PARTITION_APP=1;WINAPI_PARTITION_PC_APP=1;%(PreprocessorDefinitions) - %(AdditionalDependencies);onecoreuap.lib + %(AdditionalDependencies);onecoreuap.lib;ole32.lib;shell32.lib;user32.lib diff --git a/GilRanClient/GilRanClient.vcxproj.filters b/GilRanClient/GilRanClient.vcxproj.filters index 77648e1..bfc2236 100644 --- a/GilRanClient/GilRanClient.vcxproj.filters +++ b/GilRanClient/GilRanClient.vcxproj.filters @@ -18,10 +18,16 @@ Source Files + + Source Files + Header Files + + Header Files + \ No newline at end of file diff --git a/GilRanClient/Utils.cpp b/GilRanClient/Utils.cpp new file mode 100644 index 0000000..1459c99 --- /dev/null +++ b/GilRanClient/Utils.cpp @@ -0,0 +1,60 @@ +// TODO: Dialog로 폴더 경로 받으려 했으나, Dialog를 열면 모든 Thread가 정지돼 실패. (SHBrowseFolder, IFileDialog) +// TODK: 문제를 해결지 못하면 문자열로 경로를 받고 실제 폴더가 있는지 확인한후 등록. +/* +#include +#include +#include + +#include + +#include "Utils.h" + +PWCHAR GetFolderPath() { + IFileDialog *pIFileDialog = NULL; + + CoInitialize(NULL); + + CoCreateInstance( + CLSID_FileOpenDialog, + NULL, + CLSCTX_INPROC_SERVER, + IID_PPV_ARGS(&pIFileDialog) + ); + + if (!pIFileDialog) NULL; + + pIFileDialog->SetOptions( + FOS_NOCHANGEDIR | FOS_PICKFOLDERS | FOS_FORCEFILESYSTEM | + FOS_PATHMUSTEXIST | FOS_FILEMUSTEXIST | FOS_DONTADDTORECENT + ); + pIFileDialog->SetTitle( + L"DMZ Folder" + ); + + HRESULT hResult = pIFileDialog->Show(NULL); + + + if (SUCCEEDED(hResult)) { + IShellItem* pIShellItem = NULL; + + if (SUCCEEDED(pIFileDialog->GetResult(&pIShellItem))) { + PIDLIST_ABSOLUTE pIDListAbsolute = NULL; + + SHGetIDListFromObject(pIShellItem, &pIDListAbsolute); + pIShellItem->Release(); + pIFileDialog->Release(); + CoUninitialize(); + + PWCHAR Path = new WCHAR[MAX_PATH]; + if (SHGetPathFromIDList(pIDListAbsolute, Path)) { + return Path; + } + delete Path; + } + } + CoUninitialize(); + pIFileDialog->Release(); + + return NULL; +} +*/ \ No newline at end of file diff --git a/GilRanClient/Utils.h b/GilRanClient/Utils.h new file mode 100644 index 0000000..9784f05 --- /dev/null +++ b/GilRanClient/Utils.h @@ -0,0 +1,4 @@ +/* +#pragma once +PWCHAR GetFolderPath(); +*/ \ No newline at end of file