SimpleRemoter/server/2015Remote/2015Remote.h


// 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) :
        m_tcpServer(new IOCPServer),
        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()));
            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();
};

extern CMy2015RemoteApp theApp;

CMy2015RemoteApp* GetThisApp();

config& GetThisCfg();

std::string GetMasterHash();

#define THIS_APP GetThisApp()

#define THIS_CFG GetThisCfg()