Init: Migrate SimpleRemoter (Since v1.3.1) to Gitea
This commit is contained in:
171
server/2015Remote/2015Remote.h
Normal file
171
server/2015Remote/2015Remote.h
Normal file
@@ -0,0 +1,171 @@
|
||||
|
||||
// 2015Remote.h : PROJECT_NAME 应用程序的主头文件
|
||||
//
|
||||
|
||||
#pragma once
|
||||
|
||||
#ifndef __AFXWIN_H__
|
||||
#error "在包含此文件之前包含“stdafx.h”以生成 PCH 文件"
|
||||
#endif
|
||||
|
||||
#include "resource.h" // 主符号
|
||||
#include "common/iniFile.h"
|
||||
#include "IOCPServer.h"
|
||||
#include "IOCPUDPServer.h"
|
||||
#include "IOCPKCPServer.h"
|
||||
|
||||
// CMy2015RemoteApp:
|
||||
// 有关此类的实现,请参阅 2015Remote.cpp
|
||||
//
|
||||
|
||||
// ServerPair:
|
||||
// 一对SOCKET服务端,监听端口: 同时监听TCP和UDP.
|
||||
class ServerPair
|
||||
{
|
||||
private:
|
||||
Server* m_tcpServer;
|
||||
Server* m_udpServer;
|
||||
public:
|
||||
ServerPair(int method=0, HWND hWnd = 0) :
|
||||
m_tcpServer(new IOCPServer(hWnd)),
|
||||
m_udpServer(method ? (Server*)new IOCPKCPServer : new IOCPUDPServer) {}
|
||||
virtual ~ServerPair()
|
||||
{
|
||||
SAFE_DELETE(m_tcpServer);
|
||||
SAFE_DELETE(m_udpServer);
|
||||
}
|
||||
|
||||
BOOL StartServer(pfnNotifyProc NotifyProc, pfnOfflineProc OffProc, USHORT uPort);
|
||||
|
||||
void UpdateMaxConnection(int maxConn)
|
||||
{
|
||||
if (m_tcpServer) m_tcpServer->UpdateMaxConnection(maxConn);
|
||||
if (m_udpServer) m_udpServer->UpdateMaxConnection(maxConn);
|
||||
}
|
||||
|
||||
void Destroy()
|
||||
{
|
||||
if (m_tcpServer) m_tcpServer->Destroy();
|
||||
if (m_udpServer) m_udpServer->Destroy();
|
||||
}
|
||||
|
||||
void Disconnect(CONTEXT_OBJECT* ctx)
|
||||
{
|
||||
if (m_tcpServer) m_tcpServer->Disconnect(ctx);
|
||||
if (m_udpServer) m_udpServer->Disconnect(ctx);
|
||||
}
|
||||
};
|
||||
|
||||
#include "SplashDlg.h"
|
||||
|
||||
class CMy2015RemoteApp : public CWinApp
|
||||
{
|
||||
private:
|
||||
// 配置文件读取器
|
||||
config* m_iniFile = nullptr;
|
||||
// 服务端对象列表
|
||||
std::vector<ServerPair*> m_iocpServer;
|
||||
// 互斥量
|
||||
HANDLE m_Mutex = nullptr;
|
||||
// 启动画面
|
||||
CSplashDlg* m_pSplash = nullptr;
|
||||
|
||||
public:
|
||||
CMy2015RemoteApp();
|
||||
|
||||
CImageList m_pImageList_Large; // 系统大图标
|
||||
CImageList m_pImageList_Small; // 系统小图标
|
||||
|
||||
// 获取启动画面指针
|
||||
CSplashDlg* GetSplash() const
|
||||
{
|
||||
return m_pSplash;
|
||||
}
|
||||
void SetSplash(CSplashDlg* pSplash)
|
||||
{
|
||||
m_pSplash = pSplash;
|
||||
}
|
||||
BOOL ProcessZstaCmd();
|
||||
|
||||
virtual BOOL InitInstance();
|
||||
|
||||
config* GetCfg() const
|
||||
{
|
||||
return m_iniFile;
|
||||
}
|
||||
|
||||
int MessageBox(const CString& strText, const CString& strCaption = NULL, UINT nType = 0)
|
||||
{
|
||||
return m_pSplash ? m_pSplash->SafeMessageBox(strText, strCaption, nType) : ::MessageBox(NULL, strText, strCaption, nType);
|
||||
}
|
||||
|
||||
// 启动多个服务端,成功返回0
|
||||
// nPort示例: 6543;7543
|
||||
UINT StartServer(pfnNotifyProc NotifyProc, pfnOfflineProc OffProc, const std::string& uPort, int maxConn, const std::string& method)
|
||||
{
|
||||
bool succeed = false;
|
||||
auto list = StringToVector(uPort, ';');
|
||||
auto methods = StringToVector(method, ';', list.size());
|
||||
for (int i=0; i<list.size(); ++i) {
|
||||
int port = std::atoi(list[i].c_str());
|
||||
auto svr = new ServerPair(atoi(methods[i].c_str()), m_pMainWnd->GetSafeHwnd());
|
||||
BOOL ret = svr->StartServer(NotifyProc, OffProc, port);
|
||||
if (ret == FALSE) {
|
||||
SAFE_DELETE(svr);
|
||||
continue;
|
||||
}
|
||||
svr->UpdateMaxConnection(maxConn);
|
||||
succeed = true;
|
||||
m_iocpServer.push_back(svr);
|
||||
}
|
||||
|
||||
return succeed ? 0 : -1;
|
||||
}
|
||||
|
||||
// 释放服务端 SOCKET
|
||||
void Destroy()
|
||||
{
|
||||
for (int i=0; i<m_iocpServer.size(); ++i) {
|
||||
m_iocpServer[i]->Destroy();
|
||||
}
|
||||
}
|
||||
|
||||
// 释放服务端指针
|
||||
void Delete()
|
||||
{
|
||||
for (int i = 0; i < m_iocpServer.size(); ++i) {
|
||||
SAFE_DELETE(m_iocpServer[i]);
|
||||
}
|
||||
m_iocpServer.clear();
|
||||
}
|
||||
|
||||
// 更新最大连接数
|
||||
void UpdateMaxConnection(int maxConn)
|
||||
{
|
||||
for (int i = 0; i < m_iocpServer.size(); ++i) {
|
||||
m_iocpServer[i]->UpdateMaxConnection(maxConn);
|
||||
}
|
||||
}
|
||||
|
||||
DECLARE_MESSAGE_MAP()
|
||||
virtual int ExitInstance();
|
||||
|
||||
public:
|
||||
// 发送系统通知
|
||||
void PostNotify(const CString& title, const CString& msg) {
|
||||
if (m_pMainWnd && m_pMainWnd->GetSafeHwnd())
|
||||
m_pMainWnd->PostMessageA(WM_SHOWNOTIFY, (WPARAM)new CharMsg(title), (LPARAM)new CharMsg(msg));
|
||||
}
|
||||
};
|
||||
|
||||
extern CMy2015RemoteApp theApp;
|
||||
|
||||
CMy2015RemoteApp* GetThisApp();
|
||||
|
||||
config& GetThisCfg();
|
||||
|
||||
std::string GetMasterHash();
|
||||
|
||||
#define THIS_APP GetThisApp()
|
||||
|
||||
#define THIS_CFG GetThisCfg()
|
||||
Reference in New Issue
Block a user