86 lines
2.7 KiB
C++
Raw Normal View History

2024-05-03 01:54:36 +08:00
#pragma comment(linker, "/subsystem:windows /entry:wmainCRTStartup")
2024-01-08 23:37:00 +08:00
2024-04-02 15:36:52 +08:00
int dllinjectwmain(int argc, wchar_t *argv[]);
int ntleaswmain(int argc, wchar_t *wargv[]);
2024-03-23 18:33:58 +08:00
bool checkisapatch();
2024-01-08 23:37:00 +08:00
#ifndef _WIN64
2024-04-02 15:36:52 +08:00
int LRwmain(int argc, wchar_t *argv[]);
int jbjwmain(int argc, wchar_t *argv[]);
int dreyewmain(int argc, wchar_t *argv[]);
int kingsoftwmain(int argc, wchar_t *argv[]);
int voiceroid2wmain(int argc, wchar_t *argv[]);
int lewmain(int argc, wchar_t *argv[]);
int neospeech(int argc, wchar_t *argv[]);
int neospeechlist(int argc, wchar_t *argv[]);
2024-01-08 23:37:00 +08:00
#else
#endif // !_WIN64
2024-04-02 15:36:52 +08:00
void listprocessmodule_1(std::ofstream &of, DWORD processPID)
{
DWORD need;
2024-01-08 23:37:00 +08:00
HMODULE modules1[1] = {};
HANDLE hProcess = ::OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, processPID);
if (hProcess)
{
2024-04-02 15:36:52 +08:00
if (EnumProcessModules(hProcess, modules1, sizeof(modules1), &need) == 0)
return;
2024-01-08 23:37:00 +08:00
auto modules = std::make_unique<HMODULE[]>(need / sizeof(HMODULE));
2024-04-02 15:36:52 +08:00
if (EnumProcessModules(hProcess, modules.get(), need, &need) == 0)
return;
for (int i = 0; i < need / sizeof(HMODULE); i++)
{
wchar_t fileName[MAX_PATH] = {0};
2024-01-08 23:37:00 +08:00
GetModuleFileNameExW(hProcess, modules.get()[i], fileName, sizeof(fileName));
2024-04-02 15:36:52 +08:00
auto s = std::wstring(fileName);
of.write((char *)s.c_str(), s.size() * 2);
of.write((char *)L"\n", 2);
2024-01-08 23:37:00 +08:00
}
2024-04-02 15:36:52 +08:00
}
2024-01-08 23:37:00 +08:00
}
2024-04-02 15:36:52 +08:00
int listprocessmodule(int argc, wchar_t *argv[])
{
std::ofstream of(argv[1], std::ios_base::binary);
for (int i = 2; i < argc; i++)
{
auto pid = std::stoi(argv[i]);
listprocessmodule_1(of, pid);
2024-01-08 23:37:00 +08:00
}
of.close();
return 0;
}
2024-04-02 15:36:52 +08:00
int wmain(int argc, wchar_t *argv[])
2024-01-08 23:37:00 +08:00
{
2024-04-02 15:36:52 +08:00
if (checkisapatch())
return 1;
2024-01-08 23:37:00 +08:00
auto argv0 = std::wstring(argv[1]);
if (argv0 == L"dllinject")
return dllinjectwmain(argc - 1, argv + 1);
if (argv0 == L"ntleas")
return ntleaswmain(argc - 1, argv + 1);
if (argv0 == L"listpm")
return listprocessmodule(argc - 1, argv + 1);
2024-04-02 15:36:52 +08:00
2024-01-08 23:37:00 +08:00
#ifndef _WIN64
else if (argv0 == L"LR")
return LRwmain(argc - 1, argv + 1);
else if (argv0 == L"le")
return lewmain(argc - 1, argv + 1);
else if (argv0 == L"jbj7")
return jbjwmain(argc - 1, argv + 1);
else if (argv0 == L"dreye")
return dreyewmain(argc - 1, argv + 1);
else if (argv0 == L"kingsoft")
return kingsoftwmain(argc - 1, argv + 1);
else if (argv0 == L"voiceroid2")
return voiceroid2wmain(argc - 1, argv + 1);
else if (argv0 == L"neospeech")
return neospeech(argc - 1, argv + 1);
2024-03-23 18:33:58 +08:00
else if (argv0 == L"neospeechlist")
return neospeechlist(argc - 1, argv + 1);
2024-01-08 23:37:00 +08:00
#else
2024-05-03 17:57:41 +08:00
2024-01-08 23:37:00 +08:00
#endif // !_WIN64
}