145 lines
4.5 KiB
Python
145 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
RDP 断开检测测试工具
|
||
专门测试 RDP 断开后的检测逻辑
|
||
"""
|
||
|
||
import subprocess
|
||
import os
|
||
|
||
|
||
def check_query_user():
|
||
"""检查 query user 输出"""
|
||
print("\n" + "=" * 60)
|
||
print(" query user 命令输出")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
result = subprocess.run(
|
||
['query', 'user'],
|
||
capture_output=True,
|
||
text=True,
|
||
shell=True,
|
||
timeout=5
|
||
)
|
||
|
||
print(f"\n{result.stdout}")
|
||
|
||
if result.stderr:
|
||
print(f"错误：{result.stderr}")
|
||
|
||
# 详细分析
|
||
print("\n" + "=" * 60)
|
||
print(" 详细分析")
|
||
print("=" * 60)
|
||
|
||
lines = result.stdout.strip().split('\n')
|
||
|
||
for i, line in enumerate(lines):
|
||
line_stripped = line.strip()
|
||
line_lower = line_stripped.lower()
|
||
|
||
if not line_stripped:
|
||
continue
|
||
|
||
print(f"\n第 {i+1} 行：{line_stripped}")
|
||
|
||
# 检查标记
|
||
markers = []
|
||
if '>' in line_stripped:
|
||
markers.append("✓ 当前会话")
|
||
if 'rdp' in line_lower:
|
||
markers.append("RDP 连接")
|
||
if 'tcp' in line_lower:
|
||
markers.append("TCP 连接")
|
||
if 'active' in line_lower:
|
||
markers.append("活跃状态")
|
||
if 'disc' in line_lower:
|
||
markers.append("已断开")
|
||
if 'Console' in line_stripped or 'console' in line_lower:
|
||
markers.append("控制台")
|
||
|
||
if markers:
|
||
print(f" 标记：{', '.join(markers)}")
|
||
|
||
# 判断
|
||
if '>' in line_stripped:
|
||
if 'active' in line_lower and ('rdp' in line_lower or 'tcp' in line_lower):
|
||
print(f" → 结论：当前是活跃的 RDP 连接")
|
||
elif 'disc' in line_lower and ('rdp' in line_lower or 'tcp' in line_lower):
|
||
print(f" → 结论：RDP 已断开，应恢复本地音量")
|
||
elif 'Console' in line_stripped:
|
||
print(f" → 结论：本地控制台会话")
|
||
else:
|
||
print(f" → 结论：未知状态")
|
||
|
||
return True
|
||
|
||
except FileNotFoundError:
|
||
print("✗ query 命令不存在（仅在 Windows 上可用）")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 执行失败：{e}")
|
||
return False
|
||
|
||
|
||
def check_sessionname():
|
||
"""检查 SESSIONNAME 环境变量"""
|
||
print("\n" + "=" * 60)
|
||
print(" 环境变量检查")
|
||
print("=" * 60)
|
||
|
||
session_name = os.environ.get('SESSIONNAME', '(未设置)')
|
||
username = os.environ.get('USERNAME', '(未设置)')
|
||
|
||
print(f"\nSESSIONNAME: {session_name}")
|
||
print(f"USERNAME: {username}")
|
||
|
||
if session_name.startswith('RDP'):
|
||
print(f"\n⚠ SESSIONNAME 以 RDP 开头")
|
||
print(f" 但这可能是已断开的会话，需要结合 query user 判断")
|
||
elif session_name == 'Console':
|
||
print(f"\n✓ SESSIONNAME 是 Console，本地会话")
|
||
else:
|
||
print(f"\n? SESSIONNAME 未知格式")
|
||
|
||
return session_name
|
||
|
||
|
||
def main():
|
||
print("\n")
|
||
print("╔" + "═" * 58 + "╗")
|
||
print("║" + " " * 12 + "RDP 断开检测诊断工具" + " " * 12 + "║")
|
||
print("╚" + "═" * 58 + "╝")
|
||
|
||
print("\n此工具用于诊断 RDP 断开后的检测问题")
|
||
print("适用于：断开 RDP 后程序仍显示远程连接的情况")
|
||
|
||
# 检查
|
||
session_name = check_sessionname()
|
||
check_query_user()
|
||
|
||
# 总结
|
||
print("\n" + "=" * 60)
|
||
print(" 诊断总结")
|
||
print("=" * 60)
|
||
|
||
print("\n📋 判断规则：")
|
||
print(" 1. 当前会话（带 >）+ active + RDP/TCP = 活跃远程连接")
|
||
print(" 2. 当前会话（带 >）+ disc + RDP/TCP = 已断开，应恢复音量")
|
||
print(" 3. 当前会话（带 >）+ Console = 本地会话")
|
||
print(" 4. 无活跃 RDP 会话 = 本地状态")
|
||
|
||
print("\n💡 如果断开 RDP 后仍显示远程连接：")
|
||
print(" - 检查是否有 'disc' 标记被误判为 'active'")
|
||
print(" - 检查是否有多个会话（一个断开 + 一个活跃）")
|
||
print(" - 查看上方详细分析，确认哪一行被判定为远程")
|
||
|
||
print("\n✅ 测试完成！")
|
||
print("\n")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|