305 lines
9.7 KiB
C++
305 lines
9.7 KiB
C++
// ShellDlg.cpp : 实现文件
|
||
//
|
||
|
||
#include "stdafx.h"
|
||
#include "2015Remote.h"
|
||
#include "ShellDlg.h"
|
||
#include "afxdialogex.h"
|
||
|
||
#define EDIT_MAXLENGTH 30000
|
||
|
||
BEGIN_MESSAGE_MAP(CAutoEndEdit, CEdit)
|
||
ON_WM_CHAR()
|
||
END_MESSAGE_MAP()
|
||
|
||
void CAutoEndEdit::OnChar(UINT nChar, UINT nRepCnt, UINT nFlags)
|
||
{
|
||
// 获取当前光标位置
|
||
int nStart, nEnd;
|
||
GetSel(nStart, nEnd);
|
||
|
||
// 如果光标在最小可编辑位置之前，移动到末尾
|
||
if (nStart < (int)m_nMinEditPos) {
|
||
int nLength = GetWindowTextLength();
|
||
SetSel(nLength, nLength);
|
||
}
|
||
if (nStart == 30000){
|
||
static int hasNotify = 0;
|
||
if (hasNotify++ % 10 == 0) {
|
||
THIS_APP->PostNotify(_TR("需要清理终端"), _TR("达到字符数限制时，需执行\"clear\"命令"));
|
||
}
|
||
}
|
||
|
||
// 调用父类处理输入字符
|
||
CEdit::OnChar(nChar, nRepCnt, nFlags);
|
||
}
|
||
|
||
// CShellDlg 对话框
|
||
|
||
IMPLEMENT_DYNAMIC(CShellDlg, CDialog)
|
||
|
||
CShellDlg::CShellDlg(CWnd* pParent, Server* IOCPServer, CONTEXT_OBJECT *ContextObject)
|
||
: DialogBase(CShellDlg::IDD, pParent, IOCPServer, ContextObject, IDI_ICON_SHELL)
|
||
{
|
||
m_brBackground.CreateSolidBrush(RGB(0, 0, 0)); // 黑色背景
|
||
}
|
||
|
||
CShellDlg::~CShellDlg()
|
||
{
|
||
}
|
||
|
||
void CShellDlg::DoDataExchange(CDataExchange* pDX)
|
||
{
|
||
__super::DoDataExchange(pDX);
|
||
DDX_Control(pDX, IDC_EDIT, m_Edit);
|
||
}
|
||
|
||
|
||
BEGIN_MESSAGE_MAP(CShellDlg, CDialog)
|
||
ON_WM_CLOSE()
|
||
ON_WM_CTLCOLOR()
|
||
ON_WM_SIZE()
|
||
END_MESSAGE_MAP()
|
||
|
||
|
||
// CShellDlg 消息处理程序
|
||
|
||
|
||
BOOL CShellDlg::OnInitDialog()
|
||
{
|
||
__super::OnInitDialog();
|
||
m_nCurSel = 0;
|
||
m_nReceiveLength = 0;
|
||
SetIcon(m_hIcon, TRUE);
|
||
SetIcon(m_hIcon,FALSE);
|
||
|
||
CString str;
|
||
str.FormatL("%s - 远程终端", m_IPAddress);
|
||
SetWindowText(str);
|
||
|
||
BYTE bToken = COMMAND_NEXT;
|
||
m_ContextObject->Send2Client(&bToken, sizeof(BYTE));
|
||
|
||
m_Edit.SetWindowTextA(">>");
|
||
m_nCurSel = m_Edit.GetWindowTextLengthA();
|
||
m_nReceiveLength = m_nCurSel;
|
||
m_Edit.m_nMinEditPos = m_nReceiveLength; // 设置最小可编辑位置
|
||
m_Edit.SetSel((int)m_nCurSel, (int)m_nCurSel);
|
||
m_Edit.PostMessage(EM_SETSEL, m_nCurSel, m_nCurSel);
|
||
m_Edit.SetLimitText(EDIT_MAXLENGTH);
|
||
|
||
return TRUE; // return TRUE unless you set the focus to a control
|
||
// 异常: OCX 属性页应返回 FALSE
|
||
}
|
||
|
||
|
||
VOID CShellDlg::OnReceiveComplete()
|
||
{
|
||
if (m_ContextObject==NULL) {
|
||
return;
|
||
}
|
||
|
||
AddKeyBoardData();
|
||
m_nReceiveLength = m_Edit.GetWindowTextLength();
|
||
m_Edit.m_nMinEditPos = m_nReceiveLength; // 更新最小可编辑位置
|
||
}
|
||
|
||
|
||
#include <regex>
|
||
std::string removeAnsiCodes(const std::string& input)
|
||
{
|
||
// Match all common ANSI escape sequences:
|
||
// CSI sequences: \x1B[...X where X is a letter
|
||
// OSC sequences: \x1B]...(\x07|\x1B\\)
|
||
// Simple escapes: \x1B[=>] or single char after \x1B
|
||
std::regex ansi_regex(
|
||
"\x1B\\[[0-9;?]*[A-Za-z]" // CSI: \x1B[...m, \x1B[...H, \x1B[...J, etc.
|
||
"|\x1B\\][^\x07]*\x07" // OSC: \x1B]...\x07
|
||
"|\x1B\\][^\x1B]*\x1B\\\\" // OSC: \x1B]...\x1B\\ [*]
|
||
"|\x1B[=>]" // \x1B= or \x1B>
|
||
"|\x1B[78]" // Save/restore cursor
|
||
"|\x1B\\([AB0-2]" // Character set selection
|
||
);
|
||
return std::regex_replace(input, ansi_regex, "");
|
||
}
|
||
|
||
// UTF-8 → ANSI(GBK) 转换，如果输入不是合法 UTF-8 则原样返回
|
||
static std::string Utf8ToLocal(const std::string& text)
|
||
{
|
||
if (text.empty()) return text;
|
||
// 尝试以 UTF-8 解码，MB_ERR_INVALID_CHARS 会让非法 UTF-8 失败
|
||
int wLen = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, text.c_str(), -1, NULL, 0);
|
||
if (wLen <= 0) return text; // 不是合法 UTF-8，原样返回（Windows 客户端 GBK 数据走这里）
|
||
std::wstring wstr(wLen, 0);
|
||
MultiByteToWideChar(CP_UTF8, 0, text.c_str(), -1, &wstr[0], wLen);
|
||
int aLen = WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), -1, NULL, 0, NULL, NULL);
|
||
if (aLen <= 0) return text;
|
||
std::string ansi(aLen, 0);
|
||
WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), -1, &ansi[0], aLen, NULL, NULL);
|
||
if (!ansi.empty() && ansi.back() == '\0') ansi.pop_back();
|
||
return ansi;
|
||
}
|
||
|
||
VOID CShellDlg::AddKeyBoardData(void)
|
||
{
|
||
// 最后填上0
|
||
|
||
//Shit\0
|
||
m_ContextObject->InDeCompressedBuffer.WriteBuffer((LPBYTE)"", 1); //从被控制端来的数据我们要加上一个\0
|
||
Buffer tmp = m_ContextObject->InDeCompressedBuffer.GetMyBuffer(0);
|
||
bool firstRecv = tmp.c_str() == std::string(">");
|
||
std::string cleaned = removeAnsiCodes(tmp.c_str());
|
||
std::string converted = Utf8ToLocal(cleaned); // Linux 客户端 UTF-8 → GBK；Windows 客户端原样通过
|
||
CString strResult = firstRecv ? "" : CString("\r\n") + converted.c_str();
|
||
|
||
//替换掉原来的换行符 可能cmd 的换行同w32下的编辑控件的换行符不一致 所有的回车换行
|
||
strResult.Replace("\n", "\r\n");
|
||
|
||
if (strResult.GetLength() + m_Edit.GetWindowTextLength() >= EDIT_MAXLENGTH) {
|
||
CString text;
|
||
m_Edit.GetWindowTextA(text);
|
||
auto n = EDIT_MAXLENGTH - strResult.GetLength() - 5; // 留5个字符输入clear清屏
|
||
if (n < 0) {
|
||
strResult = strResult.Right(strResult.GetLength() + n);
|
||
}
|
||
m_Edit.SetWindowTextA(text.Right(max(n, 0)));
|
||
}
|
||
|
||
//得到当前窗口的字符个数
|
||
int iLength = m_Edit.GetWindowTextLength(); //kdfjdjfdir
|
||
//1.txt
|
||
//2.txt
|
||
//dir\r\n
|
||
|
||
//将光标定位到该位置并选中指定个数的字符 也就是末尾 因为从被控端来的数据 要显示在 我们的 先前内容的后面
|
||
m_Edit.SetSel(iLength, iLength);
|
||
|
||
//用传递过来的数据替换掉该位置的字符 //显示
|
||
m_Edit.ReplaceSel(strResult);
|
||
|
||
//重新得到字符的大小
|
||
|
||
m_nCurSel = m_Edit.GetWindowTextLength();
|
||
|
||
//我们注意到，我们在使用远程终端时 ，发送的每一个命令行 都有一个换行符 就是一个回车
|
||
//要找到这个回车的处理我们就要到PreTranslateMessage函数的定义
|
||
}
|
||
|
||
void CShellDlg::OnClose()
|
||
{
|
||
CancelIO();
|
||
// 等待数据处理完毕
|
||
if (IsProcessing()) {
|
||
ShowWindow(SW_HIDE);
|
||
return;
|
||
}
|
||
|
||
DialogBase::OnClose();
|
||
}
|
||
|
||
|
||
CString ExtractAfterLastNewline(const CString& str)
|
||
{
|
||
int nPos = str.ReverseFind(_T('\n'));
|
||
if (nPos != -1) {
|
||
return str.Mid(nPos + 1);
|
||
}
|
||
|
||
nPos = str.ReverseFind(_T('\r'));
|
||
if (nPos != -1) {
|
||
return str.Mid(nPos + 1);
|
||
}
|
||
|
||
return str;
|
||
}
|
||
|
||
|
||
BOOL CShellDlg::PreTranslateMessage(MSG* pMsg)
|
||
{
|
||
if (pMsg->message == WM_KEYDOWN) {
|
||
// 屏蔽VK_ESCAPE、VK_DELETE
|
||
if (pMsg->wParam == VK_ESCAPE || pMsg->wParam == VK_DELETE)
|
||
return true;
|
||
//如果是可编辑框的回车键
|
||
if (pMsg->wParam == VK_RETURN && pMsg->hwnd == m_Edit.m_hWnd) {
|
||
//得到窗口的数据大小
|
||
int iLength = m_Edit.GetWindowTextLength();
|
||
CString str;
|
||
//得到窗口的字符数据
|
||
m_Edit.GetWindowText(str);
|
||
//加入换行符
|
||
str += "\r\n";
|
||
//得到整个的缓冲区的首地址再加上原有的字符的位置，其实就是用户当前输入的数据了
|
||
//然后将数据发送出去
|
||
LPBYTE pSrc = (LPBYTE)str.GetBuffer(0) + m_nCurSel;
|
||
#ifdef _DEBUG
|
||
TRACE("[Shell]=> %s", (char*)pSrc);
|
||
#endif
|
||
if (0 == strcmp((char*)pSrc, "exit\r\n")) { // 退出终端
|
||
return PostMessage(WM_CLOSE);
|
||
} else if (0 == strcmp((char*)pSrc, "clear\r\n")) { // 清理终端
|
||
str = ExtractAfterLastNewline(str.Left(str.GetLength() - 7));
|
||
m_Edit.SetWindowTextA(str);
|
||
m_nCurSel = m_Edit.GetWindowTextLength();
|
||
m_nReceiveLength = m_nCurSel;
|
||
m_Edit.m_nMinEditPos = m_nReceiveLength; // 更新最小可编辑位置
|
||
m_Edit.SetSel(m_nCurSel, m_nCurSel);
|
||
return TRUE;
|
||
}
|
||
int length = str.GetLength() - m_nCurSel;
|
||
m_ContextObject->Send2Client(pSrc, length);
|
||
m_nCurSel = m_Edit.GetWindowTextLength();
|
||
}
|
||
// 限制VK_BACK
|
||
if (pMsg->wParam == VK_BACK && pMsg->hwnd == m_Edit.m_hWnd) {
|
||
if (m_Edit.GetWindowTextLength() <= m_nReceiveLength)
|
||
return true;
|
||
}
|
||
// 限制VK_LEFT - 不能移动到历史输出区域
|
||
if (pMsg->wParam == VK_LEFT && pMsg->hwnd == m_Edit.m_hWnd) {
|
||
int nStart, nEnd;
|
||
m_Edit.GetSel(nStart, nEnd);
|
||
if (nStart <= (int)m_nReceiveLength)
|
||
return true;
|
||
}
|
||
// 限制VK_UP - 禁止向上移动到历史输出
|
||
if (pMsg->wParam == VK_UP && pMsg->hwnd == m_Edit.m_hWnd) {
|
||
return true;
|
||
}
|
||
// 限制VK_HOME - 移动到当前命令行开始位置而不是文本开头
|
||
if (pMsg->wParam == VK_HOME && pMsg->hwnd == m_Edit.m_hWnd) {
|
||
m_Edit.SetSel((int)m_nReceiveLength, (int)m_nReceiveLength);
|
||
return true;
|
||
}
|
||
}
|
||
|
||
return __super::PreTranslateMessage(pMsg);
|
||
}
|
||
|
||
|
||
HBRUSH CShellDlg::OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor)
|
||
{
|
||
if ((pWnd->GetDlgCtrlID() == IDC_EDIT) && (nCtlColor == CTLCOLOR_EDIT)) {
|
||
pDC->SetTextColor(RGB(255, 255, 255)); // 白色文本
|
||
pDC->SetBkColor(RGB(0, 0, 0)); // 黑色背景
|
||
return (HBRUSH)m_brBackground.GetSafeHandle();
|
||
}
|
||
return __super::OnCtlColor(pDC, pWnd, nCtlColor);
|
||
}
|
||
|
||
|
||
void CShellDlg::OnSize(UINT nType, int cx, int cy)
|
||
{
|
||
__super::OnSize(nType, cx, cy);
|
||
|
||
if (!m_Edit.GetSafeHwnd()) return; // 确保控件已创建
|
||
|
||
// 计算新位置和大小
|
||
CRect rc;
|
||
m_Edit.GetWindowRect(&rc);
|
||
ScreenToClient(&rc);
|
||
|
||
// 重新设置控件大小
|
||
m_Edit.MoveWindow(0, 0, cx, cy, TRUE);
|
||
}
|