mirror of https://github.com/tp4a/teleport
改进SSH命令记录处理，尚未完成。
Apex Liu
parent
1aa89aea9a
commit
a8629eaf60
|
|
@ -127,6 +127,7 @@ void TppSshRec::record_command(const ex_astr& cmd)
|
||||||
|
|
||||||
m_cmd_cache.append((ex_u8*)szTime, lenTime);
|
m_cmd_cache.append((ex_u8*)szTime, lenTime);
|
||||||
m_cmd_cache.append((ex_u8*)cmd.c_str(), cmd.length());
|
m_cmd_cache.append((ex_u8*)cmd.c_str(), cmd.length());
|
||||||
|
m_cmd_cache.append((ex_u8*)"\r\n", 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool TppSshRec::_save_to_info_file() {
|
bool TppSshRec::_save_to_info_file() {
|
||||||
|
|
|
||||||
|
|
@ -16,6 +16,13 @@ TP_SSH_CHANNEL_PAIR::TP_SSH_CHANNEL_PAIR() {
|
||||||
channel_id = 0;
|
channel_id = 0;
|
||||||
|
|
||||||
is_first_server_data = true;
|
is_first_server_data = true;
|
||||||
|
|
||||||
|
//last_client_char = TP_CHAR_NUL;
|
||||||
|
//cmd_flag = 0;
|
||||||
|
maybe_cmd = false;
|
||||||
|
process_srv = false;
|
||||||
|
|
||||||
|
cmd_char_pos = cmd_char_list.begin();
|
||||||
}
|
}
|
||||||
|
|
||||||
SshSession::SshSession(SshProxy *proxy, ssh_session sess_client) :
|
SshSession::SshSession(SshProxy *proxy, ssh_session sess_client) :
|
||||||
|
|
@ -864,48 +871,44 @@ void SshSession::_on_server_channel_close(ssh_session session, ssh_channel chann
|
||||||
|
|
||||||
void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const ex_u8* data, int len)
|
void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const ex_u8* data, int len)
|
||||||
{
|
{
|
||||||
|
if (len == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
// ls;ls
|
||||||
|
|
||||||
|
|
||||||
if (TP_SSH_CLIENT_SIDE == from)
|
if (TP_SSH_CLIENT_SIDE == from)
|
||||||
{
|
{
|
||||||
// 客户端输入回车时，可能时执行了一条命令，需要根据服务端返回的数据进行进一步判断
|
// 客户端输入回车时，可能时执行了一条命令，需要根据服务端返回的数据进行进一步判断
|
||||||
if (data[len - 1] == 0x0d)
|
cp->maybe_cmd = (data[len - 1] == 0x0d);
|
||||||
cp->cmd_flag = 1;
|
if(cp->maybe_cmd)
|
||||||
|
EXLOGD("[ssh] maybe cmd.\n");
|
||||||
|
|
||||||
|
cp->process_srv = false;
|
||||||
|
|
||||||
|
if (len == 1)
|
||||||
|
{
|
||||||
|
switch (data[0]) {
|
||||||
|
case 0x08:// 08=Backspace (回删一个字符)
|
||||||
|
if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
||||||
|
{
|
||||||
|
cp->cmd_char_pos--;
|
||||||
|
cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
case 0x09:// 09=TAB键，需要根据服务端返回的数据进一步判断
|
||||||
|
cp->process_srv = true;
|
||||||
|
return;
|
||||||
|
case 0x0d:// 0d=回车
|
||||||
|
return;
|
||||||
|
case 0x7f:// 7f=DEL (删除一个字符)
|
||||||
|
if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
{
|
||||||
|
cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
// cp->cmd_flag = 0;
|
|
||||||
//
|
|
||||||
// if (len == 3)
|
|
||||||
// {
|
|
||||||
// if ((data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x41) // key-up
|
|
||||||
// || (data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x42) // key-down
|
|
||||||
// )
|
|
||||||
// {
|
|
||||||
// cp->cmd_flag = 1;
|
|
||||||
// return;
|
|
||||||
// }
|
|
||||||
// else if (data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x43) // key-right
|
|
||||||
// {
|
|
||||||
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
|
||||||
// cp->cmd_char_pos++;
|
|
||||||
// return;
|
|
||||||
// }
|
|
||||||
// else if (data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x44) // key-left
|
|
||||||
// {
|
|
||||||
// if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
|
||||||
// cp->cmd_char_pos--;
|
|
||||||
// return;
|
|
||||||
// }
|
|
||||||
// else if (
|
|
||||||
// (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x41)
|
|
||||||
// || (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x42)
|
|
||||||
// || (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x43)
|
|
||||||
// || (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x44)
|
|
||||||
// )
|
|
||||||
// {
|
|
||||||
// // 编辑模式下的上下左右键
|
|
||||||
// return;
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// else if (len == 1)
|
|
||||||
// {
|
|
||||||
// if (data[0] == 0x08 || data[0] == 0x09) // 08=光标左移
|
// if (data[0] == 0x08 || data[0] == 0x09) // 08=光标左移
|
||||||
// {
|
// {
|
||||||
// cp->cmd_flag = 1;
|
// cp->cmd_flag = 1;
|
||||||
|
|
@ -925,24 +928,53 @@ void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const e
|
||||||
// // 按下 Esc 键
|
// // 按下 Esc 键
|
||||||
// return;
|
// return;
|
||||||
// }
|
// }
|
||||||
//
|
|
||||||
// if (data[0] != 0x0d && !isprint(data[0]))
|
if (!isprint(data[0]))
|
||||||
// return;
|
return;
|
||||||
// }
|
}
|
||||||
// else if (len > 3)
|
else if (len == 3)
|
||||||
// {
|
{
|
||||||
// if (data[0] == 0x1b && data[1] == 0x5b)
|
if ((data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x41) // key-up
|
||||||
// {
|
|| (data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x42) // key-down
|
||||||
// cp->cmd_flag = 1;
|
)
|
||||||
// //
|
{
|
||||||
// // // 不是命令行上的上下左右，也不是编辑模式下的上下左右，那么就忽略（应该是编辑模式下的其他输入）
|
// 上下键是 bash 的历史记录
|
||||||
// // m_cmd_char_list.clear();
|
cp->process_srv = true;
|
||||||
// // m_cmd_char_pos = m_cmd_char_list.begin();
|
return;
|
||||||
// return;
|
}
|
||||||
// }
|
else if (data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x43) // key-right
|
||||||
// }
|
{
|
||||||
//
|
if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
// int processed = 0;
|
cp->cmd_char_pos++;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (data[0] == 0x1b && data[1] == 0x5b && data[2] == 0x44) // key-left
|
||||||
|
{
|
||||||
|
if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
||||||
|
cp->cmd_char_pos--;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (
|
||||||
|
(data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x41)
|
||||||
|
|| (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x42)
|
||||||
|
|| (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x43)
|
||||||
|
|| (data[0] == 0x1b && data[1] == 0x4f && data[2] == 0x44)
|
||||||
|
)
|
||||||
|
{
|
||||||
|
// 编辑模式下的上下左右键
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (len > 3)
|
||||||
|
{
|
||||||
|
if (data[0] == 0x1b && data[1] == 0x5b)
|
||||||
|
{
|
||||||
|
cp->process_srv = true;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int processed = 0;
|
||||||
// for (int i = 0; i < len; i++)
|
// for (int i = 0; i < len; i++)
|
||||||
// {
|
// {
|
||||||
// if (data[i] == 0x0d)
|
// if (data[i] == 0x0d)
|
||||||
|
|
@ -955,46 +987,258 @@ void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const e
|
||||||
// cp->cmd_char_pos++;
|
// cp->cmd_char_pos++;
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
// if (cp->cmd_char_list.size() > 0)
|
|
||||||
// {
|
|
||||||
// ex_astr str(cp->cmd_char_list.begin(), cp->cmd_char_list.end());
|
|
||||||
// ex_replace_all(str, "\r", "");
|
|
||||||
// ex_replace_all(str, "\n", "");
|
|
||||||
// EXLOGD("[ssh] save cmd: [%s]\n", str.c_str());
|
|
||||||
// str += "\r\n";
|
|
||||||
// cp->rec.record_command(str);
|
|
||||||
// }
|
|
||||||
// cp->cmd_char_list.clear();
|
|
||||||
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
|
||||||
//
|
|
||||||
// processed = i + 1;
|
// processed = i + 1;
|
||||||
// }
|
// }
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
// if (processed < len)
|
if (processed < len)
|
||||||
// {
|
{
|
||||||
// for (int j = processed; j < len; ++j)
|
for (int j = processed; j < len; ++j)
|
||||||
|
{
|
||||||
|
cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, data[j]);
|
||||||
|
cp->cmd_char_pos++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// const ex_u8* d = data;
|
||||||
|
// int l = len;
|
||||||
|
// bool esc_mode = false;
|
||||||
|
// int esc_arg = 0;
|
||||||
|
//
|
||||||
|
// for (; l > 0; ) {
|
||||||
|
// ex_u8 ch = d[0];
|
||||||
|
//
|
||||||
|
// if (esc_mode)
|
||||||
// {
|
// {
|
||||||
// cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, data[j]);
|
// switch (ch)
|
||||||
// cp->cmd_char_pos++;
|
// {
|
||||||
|
// case '0':
|
||||||
|
// case '1':
|
||||||
|
// case '2':
|
||||||
|
// case '3':
|
||||||
|
// case '4':
|
||||||
|
// case '5':
|
||||||
|
// case '6':
|
||||||
|
// case '7':
|
||||||
|
// case '8':
|
||||||
|
// case '9':
|
||||||
|
// esc_arg = esc_arg * 10 + (ch - '0');
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case 0x3f:
|
||||||
|
// case ';':
|
||||||
|
// case '>':
|
||||||
|
// cp->cmd_char_list.clear();
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
// return;
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case 0x4b: // 'K'
|
||||||
|
// {
|
||||||
|
// if (0 == esc_arg)
|
||||||
|
// {
|
||||||
|
// // 删除光标到行尾的字符串
|
||||||
|
// cp->cmd_char_list.erase(cp->cmd_char_pos, cp->cmd_char_list.end());
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
|
// }
|
||||||
|
// else if (1 == esc_arg)
|
||||||
|
// {
|
||||||
|
// // 删除从开始到光标处的字符串
|
||||||
|
// cp->cmd_char_list.erase(cp->cmd_char_list.begin(), cp->cmd_char_pos);
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
|
// }
|
||||||
|
// else if (2 == esc_arg)
|
||||||
|
// {
|
||||||
|
// // 删除整行
|
||||||
|
// cp->cmd_char_list.clear();
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x43: // ^[C
|
||||||
|
// {
|
||||||
|
// // 光标右移
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
// cp->cmd_char_pos++;
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x44: // ^[D
|
||||||
|
// {
|
||||||
|
// // 光标左移
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
||||||
|
// cp->cmd_char_pos--;
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// case 0x50: // 'P' 删除指定数量的字符
|
||||||
|
// {
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// case 0x40: // '@' 插入指定数量的空白字符
|
||||||
|
// {
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, ' ');
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// default:
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// d += 1;
|
||||||
|
// l -= 1;
|
||||||
|
// continue;
|
||||||
// }
|
// }
|
||||||
|
//
|
||||||
|
// switch (ch)
|
||||||
|
// {
|
||||||
|
// case 0x07:
|
||||||
|
// {
|
||||||
|
// // 响铃
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x08:
|
||||||
|
// {
|
||||||
|
// // 光标左移
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
||||||
|
// cp->cmd_char_pos--;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x09:
|
||||||
|
// case 0x0d:
|
||||||
|
// {
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x1b:
|
||||||
|
// {
|
||||||
|
// //if (i + 1 < len)
|
||||||
|
// if (l > 1)
|
||||||
|
// {
|
||||||
|
// //if (data[i + 1] == 0x5b)
|
||||||
|
// if (d[1] == 0x5b)
|
||||||
|
// {
|
||||||
|
// esc_mode = true;
|
||||||
|
// esc_arg = 0;
|
||||||
|
//
|
||||||
|
// //i += 1;
|
||||||
|
// d += 1;
|
||||||
|
// l -= 1;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// default:
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, ch);
|
||||||
|
// cp->cmd_char_pos++;
|
||||||
|
// }
|
||||||
|
// else
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_list.push_back(ch);
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// d += 1;
|
||||||
|
// l -= 1;
|
||||||
// }
|
// }
|
||||||
|
|
||||||
}
|
}
|
||||||
else if (TP_SSH_SERVER_SIDE == from)
|
else if (TP_SSH_SERVER_SIDE == from)
|
||||||
{
|
{
|
||||||
if (cp->cmd_flag == 0)
|
// TODO: 不能直接判断第一个字节，有时候客户端输入得非常快就回车时，服务端的回显字符可能跟回车换行符一起发回来，因此
|
||||||
|
// 应该将判断放到后面的循环中去。另外，不能只判断一个0d，需要判断两个字节 0d0a，否则在编辑器模式（vim/nano等）下客户端输入也会记录
|
||||||
|
|
||||||
|
const ex_u8* d = data;
|
||||||
|
int l = len;
|
||||||
|
// 如果上一个客户端数据包的最后一个字节是 0d，且当前服务端数据包以 0d0a 打头，则当前命令缓存区中的数据就是一条命令了，需要记录
|
||||||
|
//if (len >= 2 && (data[0] == 0x0d && data[1] == 0x0a)) {
|
||||||
|
if(data[0] == 0x0d) {
|
||||||
|
if (cp->maybe_cmd)
|
||||||
|
EXLOGD("[ssh] maybe cmd.\n");
|
||||||
|
if (cp->maybe_cmd) {
|
||||||
|
if (cp->cmd_char_list.size() > 0)
|
||||||
|
{
|
||||||
|
ex_astr str(cp->cmd_char_list.begin(), cp->cmd_char_list.end());
|
||||||
|
EXLOGD("[ssh] --==--==-- save cmd: [%s]\n", str.c_str());
|
||||||
|
cp->rec.record_command(str);
|
||||||
|
}
|
||||||
|
|
||||||
|
cp->cmd_char_list.clear();
|
||||||
|
cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
cp->maybe_cmd = false;
|
||||||
|
}
|
||||||
|
//else {
|
||||||
|
// cp->cmd_char_list.clear();
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
//}
|
||||||
|
else {
|
||||||
|
cp->process_srv = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
//cp->maybe_cmd = false;
|
||||||
|
|
||||||
return;
|
return;
|
||||||
if (len != 2 || !(data[0] == 0x0d && data[1] == 0x0a))
|
}
|
||||||
|
|
||||||
|
if (cp->maybe_cmd) {
|
||||||
|
//if (!(len >= 2 && (data[0] == 0x0d && data[1] == 0x0a))) {
|
||||||
|
if(data[0] != 0x0d) {
|
||||||
|
cp->cmd_char_list.clear();
|
||||||
|
cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
cp->maybe_cmd = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// if (cp->cmd_flag == 0)
|
||||||
|
// return;
|
||||||
|
|
||||||
|
if (!cp->process_srv)
|
||||||
return;
|
return;
|
||||||
|
cp->process_srv = false;
|
||||||
|
|
||||||
bool esc_mode = false;
|
bool esc_mode = false;
|
||||||
int esc_arg = 0;
|
int esc_arg = 0;
|
||||||
|
for (; l > 0; ) {
|
||||||
|
ex_u8 ch = d[0];
|
||||||
|
|
||||||
for (int i = 0; i < len; i++)
|
|
||||||
{
|
|
||||||
if (esc_mode)
|
if (esc_mode)
|
||||||
{
|
{
|
||||||
switch (data[i])
|
switch (ch)
|
||||||
{
|
{
|
||||||
case '0':
|
case '0':
|
||||||
case '1':
|
case '1':
|
||||||
|
|
@ -1006,7 +1250,7 @@ void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const e
|
||||||
case '7':
|
case '7':
|
||||||
case '8':
|
case '8':
|
||||||
case '9':
|
case '9':
|
||||||
esc_arg = esc_arg * 10 + (data[i] - '0');
|
esc_arg = esc_arg * 10 + (ch - '0');
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case 0x3f:
|
case 0x3f:
|
||||||
|
|
@ -1098,10 +1342,12 @@ void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const e
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
d += 1;
|
||||||
|
l -= 1;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (data[i])
|
switch (ch)
|
||||||
{
|
{
|
||||||
case 0x07:
|
case 0x07:
|
||||||
{
|
{
|
||||||
|
|
@ -1117,15 +1363,15 @@ void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const e
|
||||||
}
|
}
|
||||||
case 0x1b:
|
case 0x1b:
|
||||||
{
|
{
|
||||||
if (i + 1 < len)
|
if (l > 1)
|
||||||
{
|
{
|
||||||
if (data[i + 1] == 0x5b)
|
if (d[1] == 0x5b)
|
||||||
{
|
{
|
||||||
esc_mode = true;
|
esc_mode = true;
|
||||||
esc_arg = 0;
|
esc_arg = 0;
|
||||||
|
|
||||||
i += 1;
|
d += 1;
|
||||||
break;
|
l -= 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -1142,16 +1388,184 @@ void SshSession::_process_ssh_command(TP_SSH_CHANNEL_PAIR* cp, int from, const e
|
||||||
if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
{
|
{
|
||||||
cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, data[i]);
|
cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, ch);
|
||||||
cp->cmd_char_pos++;
|
cp->cmd_char_pos++;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
cp->cmd_char_list.push_back(data[i]);
|
cp->cmd_char_list.push_back(ch);
|
||||||
cp->cmd_char_pos = cp->cmd_char_list.end();
|
cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
d += 1;
|
||||||
|
l -= 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// bool esc_mode = false;
|
||||||
|
// int esc_arg = 0;
|
||||||
|
// for (int i = 0; i < len; i++)
|
||||||
|
// {
|
||||||
|
// if (esc_mode)
|
||||||
|
// {
|
||||||
|
// switch (data[i])
|
||||||
|
// {
|
||||||
|
// case '0':
|
||||||
|
// case '1':
|
||||||
|
// case '2':
|
||||||
|
// case '3':
|
||||||
|
// case '4':
|
||||||
|
// case '5':
|
||||||
|
// case '6':
|
||||||
|
// case '7':
|
||||||
|
// case '8':
|
||||||
|
// case '9':
|
||||||
|
// esc_arg = esc_arg * 10 + (data[i] - '0');
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case 0x3f:
|
||||||
|
// case ';':
|
||||||
|
// case '>':
|
||||||
|
// cp->cmd_char_list.clear();
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
// return;
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case 0x4b: // 'K'
|
||||||
|
// {
|
||||||
|
// if (0 == esc_arg)
|
||||||
|
// {
|
||||||
|
// // 删除光标到行尾的字符串
|
||||||
|
// cp->cmd_char_list.erase(cp->cmd_char_pos, cp->cmd_char_list.end());
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
|
// }
|
||||||
|
// else if (1 == esc_arg)
|
||||||
|
// {
|
||||||
|
// // 删除从开始到光标处的字符串
|
||||||
|
// cp->cmd_char_list.erase(cp->cmd_char_list.begin(), cp->cmd_char_pos);
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
|
// }
|
||||||
|
// else if (2 == esc_arg)
|
||||||
|
// {
|
||||||
|
// // 删除整行
|
||||||
|
// cp->cmd_char_list.clear();
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x43: // ^[C
|
||||||
|
// {
|
||||||
|
// // 光标右移
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
// cp->cmd_char_pos++;
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x44: // ^[D
|
||||||
|
// {
|
||||||
|
// // 光标左移
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
||||||
|
// cp->cmd_char_pos--;
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// case 0x50: // 'P' 删除指定数量的字符
|
||||||
|
// {
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// case 0x40: // '@' 插入指定数量的空白字符
|
||||||
|
// {
|
||||||
|
// if (esc_arg == 0)
|
||||||
|
// esc_arg = 1;
|
||||||
|
// for (int j = 0; j < esc_arg; ++j)
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, ' ');
|
||||||
|
// }
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// default:
|
||||||
|
// esc_mode = false;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// continue;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// switch (data[i])
|
||||||
|
// {
|
||||||
|
// case 0x07:
|
||||||
|
// {
|
||||||
|
// // 响铃
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x08:
|
||||||
|
// {
|
||||||
|
// // 光标左移
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.begin())
|
||||||
|
// cp->cmd_char_pos--;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x1b:
|
||||||
|
// {
|
||||||
|
// if (i + 1 < len)
|
||||||
|
// {
|
||||||
|
// if (data[i + 1] == 0x5b)
|
||||||
|
// {
|
||||||
|
// esc_mode = true;
|
||||||
|
// esc_arg = 0;
|
||||||
|
//
|
||||||
|
// i += 1;
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// case 0x0d:
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_list.clear();
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.begin();
|
||||||
|
//
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
// default:
|
||||||
|
// if (cp->cmd_char_pos != cp->cmd_char_list.end())
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.erase(cp->cmd_char_pos);
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.insert(cp->cmd_char_pos, data[i]);
|
||||||
|
// cp->cmd_char_pos++;
|
||||||
|
// }
|
||||||
|
// else
|
||||||
|
// {
|
||||||
|
// cp->cmd_char_list.push_back(data[i]);
|
||||||
|
// cp->cmd_char_pos = cp->cmd_char_list.end();
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }
|
||||||
}
|
}
|
||||||
|
|
||||||
return;
|
return;
|
||||||
|
|
|
||||||
|
|
@ -23,6 +23,10 @@
|
||||||
class SshProxy;
|
class SshProxy;
|
||||||
class SshSession;
|
class SshSession;
|
||||||
|
|
||||||
|
#define TP_CHAR_NUL 0x00
|
||||||
|
#define TP_CHAR_CR 0x0D
|
||||||
|
#define TP_CHAR_TAB 0x09
|
||||||
|
|
||||||
class TP_SSH_CHANNEL_PAIR {
|
class TP_SSH_CHANNEL_PAIR {
|
||||||
|
|
||||||
friend class SshSession;
|
friend class SshSession;
|
||||||
|
|
@ -44,7 +48,10 @@ private:
|
||||||
bool is_first_server_data;
|
bool is_first_server_data;
|
||||||
|
|
||||||
// for ssh command record cache.
|
// for ssh command record cache.
|
||||||
int cmd_flag;
|
//ex_u8 last_client_char; // 最近一次收到的客户端数据的最后一个字节
|
||||||
|
bool maybe_cmd;
|
||||||
|
//int cmd_flag;
|
||||||
|
bool process_srv;
|
||||||
std::list<char> cmd_char_list;
|
std::list<char> cmd_char_list;
|
||||||
std::list<char>::iterator cmd_char_pos;
|
std::list<char>::iterator cmd_char_pos;
|
||||||
};
|
};
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue