79903cbdff
1、在主对话框清理子窗口的资源（原先在各自的OnClose函数），通过CLOSE_DELETE_DLG控制。 2、修正CFileManagerDlg的构造函数调用SHGetFileInfo和FromHandle方法，解决多次打开崩溃。 3、更新服务端zlib版本为V1.2.11。（与客户端不同，因inflate_fast 崩溃，没有采用汇编）
399 lines
11 KiB
C++
399 lines
11 KiB
C++
// SystemDlg.cpp : 实现文件
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "2015Remote.h"
|
|
#include "SystemDlg.h"
|
|
#include "afxdialogex.h"
|
|
|
|
|
|
// CSystemDlg 对话框
|
|
|
|
enum
|
|
{
|
|
COMMAND_WINDOW_CLOSE, //关闭窗口
|
|
COMMAND_WINDOW_TEST, //操作窗口
|
|
};
|
|
IMPLEMENT_DYNAMIC(CSystemDlg, CDialog)
|
|
|
|
CSystemDlg::CSystemDlg(CWnd* pParent, IOCPServer* IOCPServer, CONTEXT_OBJECT *ContextObject)
|
|
: CDialog(CSystemDlg::IDD, pParent)
|
|
{
|
|
m_ContextObject = ContextObject;
|
|
m_iocpServer = IOCPServer;
|
|
|
|
char *lpBuffer = (char *)(m_ContextObject->InDeCompressedBuffer.GetBuffer(0)); //被控端传回的数据
|
|
m_bHow=lpBuffer[0];
|
|
}
|
|
|
|
CSystemDlg::~CSystemDlg()
|
|
{
|
|
}
|
|
|
|
void CSystemDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
DDX_Control(pDX, IDC_LIST_SYSTEM, m_ControlList);
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CSystemDlg, CDialog)
|
|
ON_WM_CLOSE()
|
|
ON_NOTIFY(NM_RCLICK, IDC_LIST_SYSTEM, &CSystemDlg::OnNMRClickListSystem)
|
|
ON_COMMAND(ID_PLIST_KILL, &CSystemDlg::OnPlistKill)
|
|
ON_COMMAND(ID_PLIST_REFRESH, &CSystemDlg::OnPlistRefresh)
|
|
ON_COMMAND(ID_WLIST_REFRESH, &CSystemDlg::OnWlistRefresh)
|
|
ON_COMMAND(ID_WLIST_CLOSE, &CSystemDlg::OnWlistClose)
|
|
ON_COMMAND(ID_WLIST_HIDE, &CSystemDlg::OnWlistHide)
|
|
ON_COMMAND(ID_WLIST_RECOVER, &CSystemDlg::OnWlistRecover)
|
|
ON_COMMAND(ID_WLIST_MAX, &CSystemDlg::OnWlistMax)
|
|
ON_COMMAND(ID_WLIST_MIN, &CSystemDlg::OnWlistMin)
|
|
END_MESSAGE_MAP()
|
|
|
|
|
|
// CSystemDlg 消息处理程序
|
|
|
|
|
|
BOOL CSystemDlg::OnInitDialog()
|
|
{
|
|
CDialog::OnInitDialog();
|
|
|
|
CString str;
|
|
sockaddr_in ClientAddr;
|
|
memset(&ClientAddr, 0, sizeof(ClientAddr));
|
|
int iSockAddrLength = sizeof(ClientAddr);
|
|
BOOL bResult = getpeername(m_ContextObject->sClientSocket, (SOCKADDR*)&ClientAddr, &iSockAddrLength);
|
|
m_bHow==TOKEN_PSLIST
|
|
? str.Format("%s - 进程管理", bResult != INVALID_SOCKET ? inet_ntoa(ClientAddr.sin_addr) : "")
|
|
:str.Format("%s - 窗口管理", bResult != INVALID_SOCKET ? inet_ntoa(ClientAddr.sin_addr) : "");
|
|
SetWindowText(str);//设置对话框标题
|
|
|
|
m_ControlList.SetExtendedStyle(LVS_EX_FLATSB | LVS_EX_FULLROWSELECT);
|
|
if (m_bHow==TOKEN_PSLIST) //进程管理初始化列表
|
|
{
|
|
m_ControlList.InsertColumn(0, "映像名称", LVCFMT_LEFT, 180);
|
|
m_ControlList.InsertColumn(1, "PID", LVCFMT_LEFT, 70);
|
|
m_ControlList.InsertColumn(2, "程序路径", LVCFMT_LEFT, 320);
|
|
ShowProcessList(); //由于第一个发送来的消息后面紧跟着进程的数据所以把数据显示到列表当中\0\0
|
|
}else if (m_bHow==TOKEN_WSLIST)//窗口管理初始化列表
|
|
{
|
|
//初始化 窗口管理的列表
|
|
m_ControlList.InsertColumn(0, "句柄", LVCFMT_LEFT, 80);
|
|
m_ControlList.InsertColumn(1, "窗口名称", LVCFMT_LEFT, 420);
|
|
m_ControlList.InsertColumn(2, "窗口状态", LVCFMT_LEFT, 200);
|
|
ShowWindowsList();
|
|
}
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
// 异常: OCX 属性页应返回 FALSE
|
|
}
|
|
|
|
void CSystemDlg::ShowWindowsList(void)
|
|
{
|
|
char *szBuffer = (char *)(m_ContextObject->InDeCompressedBuffer.GetBuffer(1));
|
|
DWORD dwOffset = 0;
|
|
char *szTitle = NULL;
|
|
bool isDel=false;
|
|
|
|
m_ControlList.DeleteAllItems();
|
|
CString str;
|
|
int i ;
|
|
for ( i = 0; dwOffset <m_ContextObject->InDeCompressedBuffer.GetBufferLength() - 1; i++)
|
|
{
|
|
LPDWORD lpPID = LPDWORD(szBuffer + dwOffset); //窗口句柄
|
|
szTitle = (char *)szBuffer + dwOffset + sizeof(DWORD); //窗口标题
|
|
str.Format("%5u", *lpPID);
|
|
m_ControlList.InsertItem(i, str);
|
|
m_ControlList.SetItemText(i, 1, szTitle);
|
|
m_ControlList.SetItemText(i, 2, "显示"); //(d) 将窗口状态显示为 "显示"
|
|
// ItemData 为窗口句柄
|
|
m_ControlList.SetItemData(i, *lpPID); //(d)
|
|
dwOffset += sizeof(DWORD) + lstrlen(szTitle) + 1;
|
|
}
|
|
str.Format("窗口名称 窗口个数【%d】", i); //修改CtrlList
|
|
LVCOLUMN lvc;
|
|
lvc.mask = LVCF_TEXT;
|
|
lvc.pszText = str.GetBuffer(0);
|
|
lvc.cchTextMax = str.GetLength();
|
|
m_ControlList.SetColumn(1, &lvc);
|
|
}
|
|
|
|
void CSystemDlg::ShowProcessList(void)
|
|
{
|
|
char *szBuffer = (char *)(m_ContextObject->InDeCompressedBuffer.GetBuffer(1)); //xiaoxi[][][][][]
|
|
char *szExeFile;
|
|
char *szProcessFullPath;
|
|
DWORD dwOffset = 0;
|
|
CString str;
|
|
m_ControlList.DeleteAllItems();
|
|
//遍历发送来的每一个字符别忘了他的数据结构啊 Id+进程名+0+完整名+0
|
|
int i;
|
|
for (i = 0; dwOffset < m_ContextObject->InDeCompressedBuffer.GetBufferLength() - 1; i++)
|
|
{
|
|
LPDWORD PID = LPDWORD(szBuffer + dwOffset); //这里得到进程ID
|
|
szExeFile = szBuffer + dwOffset + sizeof(DWORD); //进程名就是ID之后的啦
|
|
szProcessFullPath = szExeFile + lstrlen(szExeFile) + 1; //完整名就是进程名之后的啦
|
|
//他的数据结构的构建很巧妙
|
|
|
|
m_ControlList.InsertItem(i, szExeFile); //将得到的数据加入到列表当中
|
|
str.Format("%5u", *PID);
|
|
m_ControlList.SetItemText(i, 1, str);
|
|
m_ControlList.SetItemText(i, 2, szProcessFullPath);
|
|
// ItemData 为进程ID
|
|
m_ControlList.SetItemData(i, *PID);
|
|
|
|
dwOffset += sizeof(DWORD) + lstrlen(szExeFile) + lstrlen(szProcessFullPath) + 2; //跳过这个数据结构 进入下一个循环
|
|
}
|
|
|
|
str.Format("程序个数 / %d", i);
|
|
LVCOLUMN lvc;
|
|
lvc.mask = LVCF_TEXT;
|
|
lvc.pszText = str.GetBuffer(0);
|
|
lvc.cchTextMax = str.GetLength();
|
|
m_ControlList.SetColumn(2, &lvc); //在列表中显示有多少个进程
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnClose()
|
|
{
|
|
// TODO: 在此添加消息处理程序代码和/或调用默认值
|
|
#if CLOSE_DELETE_DLG
|
|
m_ContextObject->v1 = 0;
|
|
#endif
|
|
CancelIo((HANDLE)m_ContextObject->sClientSocket);
|
|
closesocket(m_ContextObject->sClientSocket);
|
|
CDialog::OnClose();
|
|
#if CLOSE_DELETE_DLG
|
|
delete this;
|
|
#endif
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnNMRClickListSystem(NMHDR *pNMHDR, LRESULT *pResult)
|
|
{
|
|
LPNMITEMACTIVATE pNMItemActivate = reinterpret_cast<LPNMITEMACTIVATE>(pNMHDR);
|
|
CMenu Menu;
|
|
if (m_bHow==TOKEN_PSLIST) //进程管理初始化列表
|
|
{
|
|
Menu.LoadMenu(IDR_PROCESS_LIST);
|
|
}else if (m_bHow==TOKEN_WSLIST)
|
|
{
|
|
Menu.LoadMenu(IDR_WINDOW_LIST);
|
|
}
|
|
CMenu* SubMenu = Menu.GetSubMenu(0);
|
|
CPoint Point;
|
|
GetCursorPos(&Point);
|
|
SubMenu->TrackPopupMenu(TPM_LEFTALIGN, Point.x, Point.y, this);
|
|
|
|
*pResult = 0;
|
|
}
|
|
|
|
void CSystemDlg::OnPlistKill()
|
|
{
|
|
CListCtrl *ListCtrl = NULL;
|
|
if (m_ControlList.IsWindowVisible())
|
|
ListCtrl = &m_ControlList;
|
|
else
|
|
return;
|
|
|
|
//[KILL][ID][ID][iD][ID]
|
|
//非配缓冲区
|
|
LPBYTE szBuffer = (LPBYTE)LocalAlloc(LPTR, 1 + (ListCtrl->GetSelectedCount() * 4));//1.exe 4 ID Handle
|
|
//加入结束进程的数据头
|
|
szBuffer[0] = COMMAND_KILLPROCESS;
|
|
//显示警告信息
|
|
char *szTips = "警告: 终止进程会导致不希望发生的结果，\n"
|
|
"包括数据丢失和系统不稳定。在被终止前，\n"
|
|
"进程将没有机会保存其状态和数据。";
|
|
CString str;
|
|
if (ListCtrl->GetSelectedCount() > 1)
|
|
{
|
|
str.Format("%s确实\n想终止这%d项进程吗?", szTips, ListCtrl->GetSelectedCount());
|
|
}
|
|
else
|
|
{
|
|
str.Format("%s确实\n想终止该项进程吗?", szTips);
|
|
}
|
|
if (::MessageBox(m_hWnd, str, "进程结束警告", MB_YESNO|MB_ICONQUESTION) == IDNO)
|
|
return;
|
|
|
|
DWORD dwOffset = 1;
|
|
POSITION Pos = ListCtrl->GetFirstSelectedItemPosition();
|
|
//得到要结束哪个进程
|
|
while(Pos)
|
|
{
|
|
int nItem = ListCtrl->GetNextSelectedItem(Pos);
|
|
DWORD dwProcessID = ListCtrl->GetItemData(nItem);
|
|
memcpy(szBuffer + dwOffset, &dwProcessID, sizeof(DWORD)); //sdkfj101112
|
|
dwOffset += sizeof(DWORD);
|
|
}
|
|
//发送数据到被控端在被控端中查找COMMAND_KILLPROCESS这个数据头
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, szBuffer, LocalSize(szBuffer));
|
|
LocalFree(szBuffer);
|
|
|
|
Sleep(100);
|
|
|
|
OnPlistRefresh();
|
|
}
|
|
|
|
|
|
VOID CSystemDlg::OnPlistRefresh()
|
|
{
|
|
if (m_ControlList.IsWindowVisible())
|
|
{
|
|
m_ControlList.DeleteAllItems();
|
|
GetProcessList();
|
|
ShowProcessList();
|
|
}
|
|
}
|
|
|
|
|
|
VOID CSystemDlg::GetProcessList(void)
|
|
{
|
|
BYTE bToken = COMMAND_PSLIST;
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, 1);
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnWlistRefresh()
|
|
{
|
|
GetWindowsList();
|
|
}
|
|
|
|
|
|
void CSystemDlg::GetWindowsList(void)
|
|
{
|
|
BYTE bToken = COMMAND_WSLIST;
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, &bToken, 1);
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnReceiveComplete(void)
|
|
{
|
|
switch (m_ContextObject->InDeCompressedBuffer.GetBuffer(0)[0])
|
|
{
|
|
case TOKEN_PSLIST:
|
|
{
|
|
ShowProcessList();
|
|
|
|
break;
|
|
}
|
|
case TOKEN_WSLIST:
|
|
{
|
|
ShowWindowsList();
|
|
break;
|
|
}
|
|
|
|
default:
|
|
// 传输发生异常数据
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnWlistClose()
|
|
{
|
|
BYTE lpMsgBuf[20];
|
|
CListCtrl *pListCtrl = NULL;
|
|
pListCtrl = &m_ControlList;
|
|
|
|
int nItem = pListCtrl->GetSelectionMark();
|
|
if (nItem>=0)
|
|
{
|
|
|
|
ZeroMemory(lpMsgBuf,20);
|
|
lpMsgBuf[0]=COMMAND_WINDOW_CLOSE; //注意这个就是我们的数据头
|
|
|
|
DWORD hwnd = pListCtrl->GetItemData(nItem); //得到窗口的句柄一同发送 4 djfkdfj dkfjf 4
|
|
memcpy(lpMsgBuf+1,&hwnd,sizeof(DWORD)); //1 4
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, lpMsgBuf, sizeof(lpMsgBuf));
|
|
|
|
}
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnWlistHide()
|
|
{
|
|
BYTE lpMsgBuf[20];
|
|
CListCtrl *pListCtrl = NULL;
|
|
pListCtrl = &m_ControlList;
|
|
|
|
int nItem = pListCtrl->GetSelectionMark();
|
|
if (nItem>=0)
|
|
{
|
|
ZeroMemory(lpMsgBuf,20);
|
|
lpMsgBuf[0]=COMMAND_WINDOW_TEST; //窗口处理数据头
|
|
DWORD hwnd = pListCtrl->GetItemData(nItem); //得到窗口的句柄一同发送
|
|
pListCtrl->SetItemText(nItem,2,"隐藏"); //注意这时将列表中的显示状态为"隐藏"
|
|
//这样在删除列表条目时就不删除该项了 如果删除该项窗口句柄会丢失 就永远也不能显示了
|
|
memcpy(lpMsgBuf+1,&hwnd,sizeof(DWORD)); //得到窗口的句柄一同发送
|
|
DWORD dHow=SW_HIDE; //窗口处理参数 0
|
|
memcpy(lpMsgBuf+1+sizeof(hwnd),&dHow,sizeof(DWORD));
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, lpMsgBuf, sizeof(lpMsgBuf));
|
|
}
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnWlistRecover()
|
|
{
|
|
BYTE lpMsgBuf[20];
|
|
CListCtrl *pListCtrl = NULL;
|
|
pListCtrl = &m_ControlList;
|
|
|
|
int nItem = pListCtrl->GetSelectionMark();
|
|
if (nItem>=0)
|
|
{
|
|
ZeroMemory(lpMsgBuf,20);
|
|
lpMsgBuf[0]=COMMAND_WINDOW_TEST;
|
|
DWORD hwnd = pListCtrl->GetItemData(nItem);
|
|
pListCtrl->SetItemText(nItem,2,"显示");
|
|
memcpy(lpMsgBuf+1,&hwnd,sizeof(DWORD));
|
|
DWORD dHow=SW_NORMAL;
|
|
memcpy(lpMsgBuf+1+sizeof(hwnd),&dHow,sizeof(DWORD));
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, lpMsgBuf, sizeof(lpMsgBuf));
|
|
}
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnWlistMax()
|
|
{
|
|
BYTE lpMsgBuf[20];
|
|
CListCtrl *pListCtrl = NULL;
|
|
pListCtrl = &m_ControlList;
|
|
|
|
int nItem = pListCtrl->GetSelectionMark();
|
|
if (nItem>=0)
|
|
{
|
|
ZeroMemory(lpMsgBuf,20);
|
|
lpMsgBuf[0]=COMMAND_WINDOW_TEST;
|
|
DWORD hwnd = pListCtrl->GetItemData(nItem);
|
|
pListCtrl->SetItemText(nItem,2,"显示");
|
|
memcpy(lpMsgBuf+1,&hwnd,sizeof(DWORD));
|
|
DWORD dHow=SW_MAXIMIZE;
|
|
memcpy(lpMsgBuf+1+sizeof(hwnd),&dHow,sizeof(DWORD));
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, lpMsgBuf, sizeof(lpMsgBuf));
|
|
}
|
|
}
|
|
|
|
|
|
void CSystemDlg::OnWlistMin()
|
|
{
|
|
BYTE lpMsgBuf[20];
|
|
CListCtrl *pListCtrl = NULL;
|
|
pListCtrl = &m_ControlList;
|
|
|
|
int nItem = pListCtrl->GetSelectionMark();
|
|
if (nItem>=0)
|
|
{
|
|
ZeroMemory(lpMsgBuf,20);
|
|
lpMsgBuf[0]=COMMAND_WINDOW_TEST;
|
|
DWORD hwnd = pListCtrl->GetItemData(nItem);
|
|
pListCtrl->SetItemText(nItem,2,"显示");
|
|
memcpy(lpMsgBuf+1,&hwnd,sizeof(DWORD));
|
|
DWORD dHow=SW_MINIMIZE;
|
|
memcpy(lpMsgBuf+1+sizeof(hwnd),&dHow,sizeof(DWORD));
|
|
m_iocpServer->OnClientPreSending(m_ContextObject, lpMsgBuf, sizeof(lpMsgBuf));
|
|
|
|
}// TODO: 在此添加命令处理程序代码
|
|
}
|