mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-24 12:11:52 +08:00
v2.1.2 核心优化与多架构发布 (#561)
* feat: v2.1.0 核心重构与功能增强
## 架构重构
- 全局变量消除，迁移至 Config/State 对象
- SMB 插件融合（smb/smb2/smbghost/smbinfo）
- 服务探测重构，实现 Nmap 风格 fallback 机制
- 输出系统重构，TXT 实时刷盘 + 双写机制
- i18n 框架升级至 go-i18n
## 性能优化
- 正则表达式预编译
- 内存优化 map[string]struct{}
- 并发指纹匹配
- SOCKS5 连接复用
- 滑动窗口调度 + 自适应线程池
## 新功能
- Web 管理界面
- 多格式 POC 适配（xray/afrog）
- 增强指纹库（3139条）
- Favicon hash 指纹识别
- 插件选择性编译（Build Tags）
- fscan-lab 靶场环境
- 默认端口扩展（62→133）
## 构建系统
- 添加 no_local tag 支持排除本地插件
- 多版本构建：fscan/fscan-nolocal/fscan-web
- CI 添加 snapshot 模式支持仅测试构建
## Bug 修复
- 修复 120+ 个问题，包括 RDP panic、批量扫描漏报、
JSON 输出格式、Redis 检测、Context 超时等
## 测试增强
- 单元测试覆盖率 74-100%
- 并发安全测试
- 集成测试（Web/端口/服务/SSH/ICMP）
* fix(ci): 移除 PR 对 Project 自动化的触发
* fix: Elasticsearch未授权检测优先于爆破 (#554)
* fix: 修复RDP爆破高误报率问题 (#555)
- 移除 screen.go 中错误的认证结果覆盖逻辑
- 启用 NLA 协议的 ErrorCode 字段检测
- 添加 PubKeyAuth 验证确保认证真正成功
- 修复 io.go 中错误被静默忽略的问题
- 修复 socket.go/io.go 中可能导致 panic 的代码
- 修复 screen.go 中文件句柄泄漏和 log.Panic
* fix: 修复-user/-pwd凭据参数不生效的问题
问题原因：
- Parse()解析凭据后更新globalConfig
- 但BuildConfigFromFlags()创建新Config时使用默认字典
- 导致解析的UserPassPairs等凭据信息被丢弃
修复内容：
1. initialize.go: 将Parse解析的凭据结果应用到新Config
2. credential.go: 单用户密码对时创建UserPassPairs
3. rdp.go: 单凭据测试时跳过指纹识别，减少连接次数
* feat: RDP使用NLA仅验证模式，避免挤掉已登录用户
- 添加ErrNLAAuthSuccess标志用于NLA验证成功信号
- tpkt层支持nlaAuthOnly模式，验证成功后不建立完整会话
- x224层正确传播NLA验证结果
- rdpCrack改用NlaAuth进行凭据验证
* fix: 修复进度条在Windows终端满屏重复输出的问题
- 添加终端宽度检测，动态调整进度条长度
- 使用空格覆盖清除旧内容，避免残留
- 简化进度条格式，确保不超过终端宽度
* feat: 优化日志颜色方案，区分漏洞和普通信息
- 新增 LogVuln 级别（红色），用于漏洞和重要发现
- 密码爆破成功、未授权访问、POC漏洞等改用红色显示
- 普通信息（扫描统计等）改为白色
- Web指纹保持绿色
* refactor: 精简化输出，移除冗余启动信息
- 移除showParseSummary开局配置输出
- 移除LogPluginInfo/LogPluginInfoWithPort插件信息输出
- 移除alive_scanner冗余统计输出
- 移除port_scan_start扫描开始提示
- 移除handleUDPPorts SNMP死代码
- 移除相关i18n条目
* chore: 版本号更新为2.1.1
* fix: 降级依赖版本以保持Go 1.20兼容性
* feat(ldap): 添加NTLM Hash认证支持 (#433)
* chore: 清理无用的 replace 指令
* fix(ping): 修复 TTL expired 导致主机误判为存活的问题
在 ExecCommandPing 中增加错误关键词检测，当 ping 输出包含
TTL expired、Destination unreachable 等错误信息时，不再将
目标主机标记为存活。
Fixes #454
* fix(proxy): 修复透明代理导致输出全端口的问题
在代理初始化时主动探测代理行为，通过连接 RFC 5737 保留的
测试地址来检测是否存在"全回显"问题。如果探测到代理不可靠，
则在端口扫描时跳过所有端口，避免误报。
- 新增 proxyReliable 标志位标记代理可靠性
- 新增 ProbeProxyBehavior 函数探测代理行为
- 端口扫描前检查代理可靠性并输出警告
Fixes #495
* refactor: 移动debug模块到common/debug子包
* fix(web): 修复-u模式下Web插件未执行的问题
* fix: 优化输出格式和颜色显示
- 网段统计格式改为 10.253.0.0/16 网段存活: 26
- WebTitle基础信息改为白色，指纹识别单独绿色输出
- 移除重复的端口数量输出
* fix: URL解析自动补全协议头
-uf 文件中 192.168.1.1:8080 自动转为 http://192.168.1.1:8080
* fix: 修复-u/-uf模式下URLs丢失导致0目标扫描的问题
Parse阶段将URLs设置到全局状态，但Initialize随后创建新状态
并覆盖了全局状态，导致URLs数据丢失。现在在创建新状态前
先保存并迁移Parse阶段设置的URLs和HostPorts数据。
* fix: 智能检测HTTP/HTTPS协议并优化URL显示
- 修复-u/-uf模式URLs丢失导致0目标扫描问题
- detectProtocol改为主动TLS握手检测，不依赖服务名
- WebTitle输出显示完整协议(http/https)
- 隐藏标准端口(80/443)使输出更简洁
* refactor: 精简parsers包，统一配置构建入口
- 删除冗余的中间层（XXXInput、XXXParser类）
- 新增 config_builder.go 统一配置构建
- parsers包从3000+行精简至~540行
- 保留核心函数：ParseIP、ParsePort、文件读取、凭据解析
* test: 扩展parsers单元测试覆盖边缘情况
- 新增内网简写解析测试(192/172/10)
- 新增完整IP范围和无效CIDR测试
- 新增Windows行尾(CRLF)处理测试
- 新增凭据和哈希文件解析测试
- 新增端口解析边缘情况测试
- 测试覆盖率达到94.2%
* refactor: 优化控制台输出格式
- 去掉时间戳，保留[*][+]前缀
- Web输出合并WebTitle和WebFinger为一行
- 有指纹显示绿色[+]，无指纹显示白色[*]
- 格式: code:xxx len:xxx title:xxx server:xxx [指纹]
- 服务探测格式: [Product:xxx ||Version:xxx] Banner:(xxx)
- 字段对齐，输出更清爽
* feat: 添加凭据测试未发现弱密码的提示
- credential_tester.go: 失败时设置 Type=ResultTypeCredential
- scanner.go: 根据结果类型在 error 级别输出'未发现弱密码'提示
- 新增 i18n 翻译 brute_no_weak_pass
使用 -log all 或 -log error 可看到此提示
* refactor(logging): 重构日志级别为层级过滤设计
- LogLevel 从 string 改为 int 类型，支持层级比较
- 层级设计：Debug(0) < Base(1) < Info(2) < Success(3) < Vuln(4) < Error(5)
- 设置一个级别后，显示该级别及以上的日志
- Error 级别始终显示，不会被配置过滤掉
- 保留向后兼容别名（LevelAll, LevelInfoSuccess 等）
- 更新测试以匹配新的层级过滤行为
* style(logging): Error级别日志改为黄色显示
* style(findnet): NetInfo输出改为每行一个IP
* refactor(ms17010): 优化错误提示，明确指出SMBv1不支持等情况
* fix(credential): 修复凭据测试结果不一致的问题
问题原因：
1. 未知错误类型不重试，导致服务端限流时跳过正确密码
2. SSH 错误分类不够准确，某些临时错误未被识别
修复内容：
1. 未知错误改为可重试（可能是临时问题）
2. 增加 SSH 特有的网络错误识别（handshake failed, disconnect 等）
* fix(portfinger): 修复SMB2服务指纹识别和NetInfo输出问题
- 添加SMB2ProgNeg探针支持现代Windows的SMB2协议
- 修复Go regexp对高位字节的UTF-8兼容问题，使用Latin-1转换
- 修复探针失败后连接重建逻辑
- 修复vendor_product字段名不匹配问题
- 修复NetInfo多行输出被其他日志打断的问题
* fix(config): 从默认端口移除9100，避免触发打印机打印 (#517)
* feat(proxy): 增强代理端口扫描的深度验证机制
- 新增4阶段深度验证：Banner读取→探测发送→响应等待→最终判定
- 新增SOCKS5错误码和代理错误文本检测
- 优化ProbeProxyBehavior探测逻辑，发送数据验证连接可达性
- 解决透明代理/全回显代理导致的假阳性问题
* fix(proxy): 修复代理深度验证的若干问题
- detector.go: 修复 AutoConfigureProxy 覆盖探测结果的问题
只有未探测过时才设置默认 proxyReliable 值
- port_scan.go: 改进深度验证机制
- 使用带 Host header 的 HTTP GET 请求替代 OPTIONS
- 延长响应等待超时至 2s 以适配慢速服务器
- 正确重置连接 deadline 避免影响后续操作
* refactor: 统一 common 包文件命名风格
Flag.go -> flag.go
* refactor(proxy): 删除自定义 contains() 函数，改用标准库
- 用 strings.Contains() 替代手写的 contains()
- 删除过时的注释
* fix(parsers): 修复带横杠域名被误识别为IP范围的问题
如 111-555.sss.com 这类域名因包含 - 被错误解析为 IP 范围，
添加 looksLikeIPRange() 检查，只有 - 前是有效 IP 才走范围解析
* fix(proxy): 修复代理模式下服务识别错误和端口漏扫问题
- port_scan.go: 验证通过后重建干净连接，避免HTTP GET探测污染服务识别
- port_scan.go: 优化验证策略，用轻量CRLF探测替代HTTP GET，超时从2.2s降至0.6s
- manager.go: 修正ProbeProxyBehavior判断逻辑，超时应视为代理正常转发
* fix(pool): 移除线程池预分配，优化大规模扫描内存占用
WithPreAlloc(true) 会预先创建所有 worker goroutine，
在大规模扫描(如 25域名×65535端口)时可能导致内存问题
* refactor(logging): 统一日志前缀，删除废弃的 LogBase
- 删除 LogBase 函数，所有调用迁移到 LogInfo/LogError
- 新增 PrefixDebug ([.]) 前缀，所有日志级别现在都有前缀
- 修复日志输出缩进不一致的问题
- 删除未使用的 PrefixDefault 常量
* perf(icmp): 实现自适应等待算法优化存活检测性能
- 新增 waitAdaptive 函数，监控响应增量实现智能提前结束
- 算法保守原则：最小等待1s + 连续500ms无新响应才提前结束
- 添加100ms检查间隔避免CPU空转
- 保留原有最大等待时间(3s/6s)作为兜底
- 添加完整单元测试覆盖各种场景
优化效果：
- 全部响应：~100ms (原3s)
- 无响应：~1s (原3s)
- 部分响应后稳定：~1.5s (原3s)
* perf(scan): 实现启发式优化提升扫描体验
1. 端口优先级排序：高价值端口(80,443,22,3389等)优先扫描
- 用户能更快看到有意义的结果
- 不影响端口喷洒策略
2. TCP 补充探测：ICMP 响应率<10%时自动启用
- 对未响应主机用 TCP 80/443/22/445 补充探测
- 解决防火墙过滤 ICMP 导致漏检的问题
* refactor(grdp): 精简RDP库，删除认证检测不需要的代码
- 删除 VNC 协议支持 (protocol/rfb, client/rfb.go)
- 删除完整客户端框架 (client/)
- 删除 RemoteApp 等插件 (plugin/)
- 删除 RLE 图形解压 (core/rle.go)
- 删除绘图指令处理 (pdu/orders.go, pdu/gdi.go)
- 精简 screen.go，移除截图和完整会话功能
- 移除未使用的 RGB 转换函数
grdp 代码从 13,044 行精简至 7,581 行，削减 42%
* refactor(common): 删除死代码，优化代码风格
- 删除未使用的 joinStrings/joinInts 函数
- 删除未使用的 memStats 字段和 getMemoryInfo 方法
- 简化 parsePasswords 中的循环为 append(...) 形式
* refactor(services): 统一数据库插件的DBWrapper
4个数据库插件(MySQL、PostgreSQL、MSSQL、Oracle)都有相同的sql.DB包装代码，
合并为通用的SQLDBWrapper，减少重复。
* refactor(core,grdp): 删除未使用的死代码
- 移除 BaseScanStrategy.LogPluginInfoWithPort 方法（无调用者）
- 移除 mcs.go 中被注释的旧 connect 函数实现
* refactor: 删除 deadcode 检测出的未使用函数
- proxy/detector.go: 删除 IsSOCKS5Standard, IsProxyInitialized
- findnet.go: 删除 NetworkInfo.OneLine, TreeFormat 方法
- port_scan.go: 删除 estimateScanTime 函数
- web_scanner.go: 删除 GetFingerprints 函数
- 清理相关测试代码
* refactor: 删除更多未使用的死代码
- parse.go: 删除 RemoveDuplicate 函数及其测试
- parsers.go: 删除 excludeHosts, removeDuplicates 别名函数
- 更新测试使用真正的函数名
* fix(test): 修复 TestParseIP_InvalidIPRange 测试用例
- 删除不合理的测试用例（无效IP被当作普通主机名处理是设计行为）
- 修复测试逻辑，只在真正通过时输出"正确"
* fix(scan): 移除域名预解析，保留原始域名进行扫描
域名预解析会将域名转换为IP，导致虚拟主机场景下HTTP访问失败
（Host头变成IP而非域名，无法正确路由）
* fix(scan): 修复 -hf 参数无法单独使用的问题
* fix(proxy): 修复透明代理环境下 SOCKS5 代理全端口误报问题
问题：在透明代理(TUN模式)环境下使用 SOCKS5 代理扫描时，
会出现全端口开放的误报，因为代理可靠性检测被透明代理污染。
修复方案（参考 fscanx）：
1. 将探针从 CRLF 改为 HTTP GET，更有效检测真实连接状态
2. 删除 "uncertain" 状态，无响应一律判定为端口关闭
3. 调整超时时间以适应代理链路延迟
Fixes #524
* feat(telnet): 新增 telnetd RCE 命令执行验证，修复未授权访问日志级别
* fix: 修复 i18n.Tr vet 报错、Unicode 测试用例，移除过期域名
- 移除 i18n.Tr 中错误的 fmt.Sprintf fallback，消除 go vet 误报
- 修复 match_engine_test Unicode 测试用例与 Latin-1 转换逻辑不匹配
- README 移除过期的 fscan.club 域名
- 添加 .gitattributes 统一换行符为 LF
* refactor: 统一控制台输出风格，使用统一的日志函数
手动合并 PR #558 的改动，适配重构后的代码路径
* fix(ci): 修复版本注入和CI触发配置
- goreleaser ldflags 指向正确的包路径 common.version/commit/date
- version 改为 var 支持 ldflags 注入，banner 显示 commit 和构建日期
- test-build 触发分支增加 dev-* 通配
* fix(ci): 修复 Windows 产物 .exe.exe 双后缀问题
* feat(ci): 扩展构建架构支持 MIPS/ARM/FreeBSD/Solaris
This commit is contained in:
ZacharyZcR
@@ -0,0 +1,183 @@
|
||||
package lic
|
||||
|
||||
import (
|
||||
"io"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
)
|
||||
|
||||
const (
|
||||
LICENSE_REQUEST = 0x01
|
||||
PLATFORM_CHALLENGE = 0x02
|
||||
NEW_LICENSE = 0x03
|
||||
UPGRADE_LICENSE = 0x04
|
||||
LICENSE_INFO = 0x12
|
||||
NEW_LICENSE_REQUEST = 0x13
|
||||
PLATFORM_CHALLENGE_RESPONSE = 0x15
|
||||
ERROR_ALERT = 0xFF
|
||||
)
|
||||
|
||||
// error code
|
||||
const (
|
||||
ERR_INVALID_SERVER_CERTIFICATE = 0x00000001
|
||||
ERR_NO_LICENSE = 0x00000002
|
||||
ERR_INVALID_SCOPE = 0x00000004
|
||||
ERR_NO_LICENSE_SERVER = 0x00000006
|
||||
STATUS_VALID_CLIENT = 0x00000007
|
||||
ERR_INVALID_CLIENT = 0x00000008
|
||||
ERR_INVALID_PRODUCTID = 0x0000000B
|
||||
ERR_INVALID_MESSAGE_LEN = 0x0000000C
|
||||
ERR_INVALID_MAC = 0x00000003
|
||||
)
|
||||
|
||||
// state transition
|
||||
const (
|
||||
ST_TOTAL_ABORT = 0x00000001
|
||||
ST_NO_TRANSITION = 0x00000002
|
||||
ST_RESET_PHASE_TO_START = 0x00000003
|
||||
ST_RESEND_LAST_MESSAGE = 0x00000004
|
||||
)
|
||||
|
||||
/*
|
||||
"""
|
||||
@summary: Binary blob data type
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc240481.aspx
|
||||
"""
|
||||
*/
|
||||
type BinaryBlobType uint16
|
||||
|
||||
const (
|
||||
BB_ANY_BLOB = 0x0000
|
||||
BB_DATA_BLOB = 0x0001
|
||||
BB_RANDOM_BLOB = 0x0002
|
||||
BB_CERTIFICATE_BLOB = 0x0003
|
||||
BB_ERROR_BLOB = 0x0004
|
||||
BB_ENCRYPTED_DATA_BLOB = 0x0009
|
||||
BB_KEY_EXCHG_ALG_BLOB = 0x000D
|
||||
BB_SCOPE_BLOB = 0x000E
|
||||
BB_CLIENT_USER_NAME_BLOB = 0x000F
|
||||
BB_CLIENT_MACHINE_NAME_BLOB = 0x0010
|
||||
)
|
||||
|
||||
type ErrorMessage struct {
|
||||
DwErrorCode uint32
|
||||
DwStateTransaction uint32
|
||||
Blob []byte
|
||||
}
|
||||
|
||||
func readErrorMessage(r io.Reader) *ErrorMessage {
|
||||
m := &ErrorMessage{}
|
||||
m.DwErrorCode, _ = core.ReadUInt32LE(r)
|
||||
m.DwStateTransaction, _ = core.ReadUInt32LE(r)
|
||||
return m
|
||||
}
|
||||
|
||||
type LicensePacket struct {
|
||||
BMsgtype uint8
|
||||
Flag uint8
|
||||
WMsgSize uint16
|
||||
LicensingMessage interface{}
|
||||
}
|
||||
|
||||
func ReadLicensePacket(r io.Reader) *LicensePacket {
|
||||
l := &LicensePacket{}
|
||||
l.BMsgtype, _ = core.ReadUInt8(r)
|
||||
l.Flag, _ = core.ReadUInt8(r)
|
||||
l.WMsgSize, _ = core.ReadUint16LE(r)
|
||||
|
||||
switch l.BMsgtype {
|
||||
case ERROR_ALERT:
|
||||
l.LicensingMessage = readErrorMessage(r)
|
||||
default:
|
||||
l.LicensingMessage, _ = core.ReadBytes(int(l.WMsgSize-4), r)
|
||||
}
|
||||
|
||||
return l
|
||||
}
|
||||
|
||||
/*
|
||||
"""
|
||||
@summary: Blob use by license manager to exchange security data
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc240481.aspx
|
||||
"""
|
||||
*/
|
||||
type LicenseBinaryBlob struct {
|
||||
WBlobType uint16 `struc:"little"`
|
||||
WBlobLen uint16 `struc:"little"`
|
||||
BlobData []byte `struc:"sizefrom=WBlobLen"`
|
||||
}
|
||||
|
||||
func NewLicenseBinaryBlob(WBlobType uint16) *LicenseBinaryBlob {
|
||||
return &LicenseBinaryBlob{}
|
||||
}
|
||||
|
||||
/*
|
||||
"""
|
||||
@summary: License server product information
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241915.aspx
|
||||
"""
|
||||
*/
|
||||
type ProductInformation struct {
|
||||
DwVersion uint32 `struc:"little"`
|
||||
CbCompanyName uint32 `struc:"little"`
|
||||
//may contain "Microsoft Corporation" from server microsoft
|
||||
PbCompanyName []byte `struc:"sizefrom=CbCompanyName"`
|
||||
CbProductId uint32 `struc:"little"`
|
||||
//may contain "A02" from microsoft license server
|
||||
PbProductId []byte `struc:"sizefrom=CbProductId"`
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: Send by server to signal license request
|
||||
|
||||
server -> client
|
||||
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241914.aspx
|
||||
*/
|
||||
type ServerLicenseRequest struct {
|
||||
ServerRandom []byte `struc:"[32]byte"`
|
||||
ProductInfo ProductInformation `struc:"little"`
|
||||
KeyExchangeList LicenseBinaryBlob `struc:"little"`
|
||||
ServerCertificate LicenseBinaryBlob `struc:"little"`
|
||||
//ScopeList ScopeList
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: Send by client to ask new license for client.
|
||||
RDPY doesn'support license reuse, need it in futur version
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241918.aspx
|
||||
#RSA and must be only RSA
|
||||
#pure microsoft client ;-)
|
||||
#http://msdn.microsoft.com/en-us/library/1040af38-c733-4fb3-acd1-8db8cc979eda#id10
|
||||
*/
|
||||
|
||||
type ClientNewLicenseRequest struct {
|
||||
PreferredKeyExchangeAlg uint32 `struc:"little"`
|
||||
PlatformId uint32 `struc:"little"`
|
||||
ClientRandom []byte `struc:"[32]byte"`
|
||||
EncryptedPreMasterSecret LicenseBinaryBlob `struc:"little"`
|
||||
ClientUserName LicenseBinaryBlob `struc:"little"`
|
||||
ClientMachineName LicenseBinaryBlob `struc:"little"`
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: challenge send from server to client
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241921.aspx
|
||||
*/
|
||||
type ServerPlatformChallenge struct {
|
||||
ConnectFlags uint32
|
||||
EncryptedPlatformChallenge LicenseBinaryBlob
|
||||
MACData [16]byte
|
||||
}
|
||||
|
||||
/*
|
||||
"""
|
||||
@summary: client challenge response
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241922.aspx
|
||||
"""
|
||||
*/
|
||||
type ClientPLatformChallengeResponse struct {
|
||||
EncryptedPlatformChallengeResponse LicenseBinaryBlob
|
||||
EncryptedHWID LicenseBinaryBlob
|
||||
MACData []byte //[16]byte
|
||||
}
|
||||
@@ -0,0 +1,99 @@
|
||||
package nla
|
||||
|
||||
import (
|
||||
"encoding/asn1"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
)
|
||||
|
||||
type NegoToken struct {
|
||||
Data []byte `asn1:"explicit,tag:0"`
|
||||
}
|
||||
|
||||
type TSRequest struct {
|
||||
Version int `asn1:"explicit,tag:0"`
|
||||
NegoTokens []NegoToken `asn1:"optional,explicit,tag:1"`
|
||||
AuthInfo []byte `asn1:"optional,explicit,tag:2"`
|
||||
PubKeyAuth []byte `asn1:"optional,explicit,tag:3"`
|
||||
ErrorCode int `asn1:"optional,explicit,tag:4"`
|
||||
}
|
||||
|
||||
type TSCredentials struct {
|
||||
CredType int `asn1:"explicit,tag:0"`
|
||||
Credentials []byte `asn1:"explicit,tag:1"`
|
||||
}
|
||||
|
||||
type TSPasswordCreds struct {
|
||||
DomainName []byte `asn1:"explicit,tag:0"`
|
||||
UserName []byte `asn1:"explicit,tag:1"`
|
||||
Password []byte `asn1:"explicit,tag:2"`
|
||||
}
|
||||
|
||||
type TSCspDataDetail struct {
|
||||
KeySpec int `asn1:"explicit,tag:0"`
|
||||
CardName string `asn1:"explicit,tag:1"`
|
||||
ReaderName string `asn1:"explicit,tag:2"`
|
||||
ContainerName string `asn1:"explicit,tag:3"`
|
||||
CspName string `asn1:"explicit,tag:4"`
|
||||
}
|
||||
|
||||
type TSSmartCardCreds struct {
|
||||
Pin string `asn1:"explicit,tag:0"`
|
||||
CspData []TSCspDataDetail `asn1:"explicit,tag:1"`
|
||||
UserHint string `asn1:"explicit,tag:2"`
|
||||
DomainHint string `asn1:"explicit,tag:3"`
|
||||
}
|
||||
|
||||
func EncodeDERTRequest(msgs []Message, authInfo []byte, pubKeyAuth []byte) []byte {
|
||||
req := TSRequest{
|
||||
Version: 2,
|
||||
}
|
||||
|
||||
if len(msgs) > 0 {
|
||||
req.NegoTokens = make([]NegoToken, 0, len(msgs))
|
||||
}
|
||||
|
||||
for _, msg := range msgs {
|
||||
token := NegoToken{msg.Serialize()}
|
||||
req.NegoTokens = append(req.NegoTokens, token)
|
||||
}
|
||||
|
||||
if len(authInfo) > 0 {
|
||||
req.AuthInfo = authInfo
|
||||
}
|
||||
|
||||
if len(pubKeyAuth) > 0 {
|
||||
req.PubKeyAuth = pubKeyAuth
|
||||
}
|
||||
|
||||
result, err := asn1.Marshal(req)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func DecodeDERTRequest(s []byte) (*TSRequest, error) {
|
||||
treq := &TSRequest{}
|
||||
_, err := asn1.Unmarshal(s, treq)
|
||||
return treq, err
|
||||
}
|
||||
func EncodeDERTCredentials(domain, username, password []byte) []byte {
|
||||
tpas := TSPasswordCreds{domain, username, password}
|
||||
result, err := asn1.Marshal(tpas)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
}
|
||||
tcre := TSCredentials{1, result}
|
||||
result, err = asn1.Marshal(tcre)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func DecodeDERTCredentials(s []byte) (*TSCredentials, error) {
|
||||
tcre := &TSCredentials{}
|
||||
_, err := asn1.Unmarshal(s, tcre)
|
||||
return tcre, err
|
||||
}
|
||||
@@ -0,0 +1,46 @@
|
||||
package nla
|
||||
|
||||
import (
|
||||
"crypto/hmac"
|
||||
"crypto/md5"
|
||||
"crypto/rc4"
|
||||
"strings"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"golang.org/x/crypto/md4"
|
||||
)
|
||||
|
||||
func MD4(data []byte) []byte {
|
||||
h := md4.New()
|
||||
h.Write(data)
|
||||
return h.Sum(nil)
|
||||
}
|
||||
|
||||
func MD5(data []byte) []byte {
|
||||
h := md5.New()
|
||||
h.Write(data)
|
||||
return h.Sum(nil)
|
||||
}
|
||||
|
||||
func HMAC_MD5(key, data []byte) []byte {
|
||||
h := hmac.New(md5.New, key)
|
||||
h.Write(data)
|
||||
return h.Sum(nil)
|
||||
}
|
||||
|
||||
// Version 2 of NTLM hash function
|
||||
func NTOWFv2(password, user, domain string) []byte {
|
||||
return HMAC_MD5(MD4(core.UnicodeEncode(password)), core.UnicodeEncode(strings.ToUpper(user)+domain))
|
||||
}
|
||||
|
||||
// Same as NTOWFv2
|
||||
func LMOWFv2(password, user, domain string) []byte {
|
||||
return NTOWFv2(password, user, domain)
|
||||
}
|
||||
|
||||
func RC4K(key, src []byte) []byte {
|
||||
result := make([]byte, len(src))
|
||||
rc4obj, _ := rc4.NewCipher(key)
|
||||
rc4obj.XORKeyStream(result, src)
|
||||
return result
|
||||
}
|
||||
@@ -0,0 +1,515 @@
|
||||
package nla
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/md5"
|
||||
"crypto/rc4"
|
||||
"encoding/binary"
|
||||
"encoding/hex"
|
||||
"time"
|
||||
|
||||
"github.com/lunixbochs/struc"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
)
|
||||
|
||||
const (
|
||||
WINDOWS_MINOR_VERSION_0 = 0x00
|
||||
WINDOWS_MINOR_VERSION_1 = 0x01
|
||||
WINDOWS_MINOR_VERSION_2 = 0x02
|
||||
WINDOWS_MINOR_VERSION_3 = 0x03
|
||||
|
||||
WINDOWS_MAJOR_VERSION_5 = 0x05
|
||||
WINDOWS_MAJOR_VERSION_6 = 0x06
|
||||
NTLMSSP_REVISION_W2K3 = 0x0F
|
||||
)
|
||||
|
||||
const (
|
||||
MsvAvEOL = 0x0000
|
||||
MsvAvNbComputerName = 0x0001
|
||||
MsvAvNbDomainName = 0x0002
|
||||
MsvAvDnsComputerName = 0x0003
|
||||
MsvAvDnsDomainName = 0x0004
|
||||
MsvAvDnsTreeName = 0x0005
|
||||
MsvAvFlags = 0x0006
|
||||
MsvAvTimestamp = 0x0007
|
||||
MsvAvSingleHost = 0x0008
|
||||
MsvAvTargetName = 0x0009
|
||||
MsvChannelBindings = 0x000A
|
||||
)
|
||||
|
||||
type AVPair struct {
|
||||
Id uint16 `struc:"little"`
|
||||
Len uint16 `struc:"little,sizeof=Value"`
|
||||
Value []byte `struc:"little"`
|
||||
}
|
||||
|
||||
const (
|
||||
NTLMSSP_NEGOTIATE_56 = 0x80000000
|
||||
NTLMSSP_NEGOTIATE_KEY_EXCH = 0x40000000
|
||||
NTLMSSP_NEGOTIATE_128 = 0x20000000
|
||||
NTLMSSP_NEGOTIATE_VERSION = 0x02000000
|
||||
NTLMSSP_NEGOTIATE_TARGET_INFO = 0x00800000
|
||||
NTLMSSP_REQUEST_NON_NT_SESSION_KEY = 0x00400000
|
||||
NTLMSSP_NEGOTIATE_IDENTIFY = 0x00100000
|
||||
NTLMSSP_NEGOTIATE_EXTENDED_SESSIONSECURITY = 0x00080000
|
||||
NTLMSSP_TARGET_TYPE_SERVER = 0x00020000
|
||||
NTLMSSP_TARGET_TYPE_DOMAIN = 0x00010000
|
||||
NTLMSSP_NEGOTIATE_ALWAYS_SIGN = 0x00008000
|
||||
NTLMSSP_NEGOTIATE_OEM_WORKSTATION_SUPPLIED = 0x00002000
|
||||
NTLMSSP_NEGOTIATE_OEM_DOMAIN_SUPPLIED = 0x00001000
|
||||
NTLMSSP_NEGOTIATE_NTLM = 0x00000200
|
||||
NTLMSSP_NEGOTIATE_LM_KEY = 0x00000080
|
||||
NTLMSSP_NEGOTIATE_DATAGRAM = 0x00000040
|
||||
NTLMSSP_NEGOTIATE_SEAL = 0x00000020
|
||||
NTLMSSP_NEGOTIATE_SIGN = 0x00000010
|
||||
NTLMSSP_REQUEST_TARGET = 0x00000004
|
||||
NTLM_NEGOTIATE_OEM = 0x00000002
|
||||
NTLMSSP_NEGOTIATE_UNICODE = 0x00000001
|
||||
)
|
||||
|
||||
type NVersion struct {
|
||||
ProductMajorVersion uint8 `struc:"little"`
|
||||
ProductMinorVersion uint8 `struc:"little"`
|
||||
ProductBuild uint16 `struc:"little"`
|
||||
Reserved [3]byte `struc:"little"`
|
||||
NTLMRevisionCurrent uint8 `struc:"little"`
|
||||
}
|
||||
|
||||
func NewNVersion() NVersion {
|
||||
return NVersion{
|
||||
ProductMajorVersion: WINDOWS_MAJOR_VERSION_6,
|
||||
ProductMinorVersion: WINDOWS_MINOR_VERSION_0,
|
||||
ProductBuild: 6002,
|
||||
NTLMRevisionCurrent: NTLMSSP_REVISION_W2K3,
|
||||
}
|
||||
}
|
||||
|
||||
type Message interface {
|
||||
Serialize() []byte
|
||||
}
|
||||
|
||||
type NegotiateMessage struct {
|
||||
Signature [8]byte `struc:"little"`
|
||||
MessageType uint32 `struc:"little"`
|
||||
NegotiateFlags uint32 `struc:"little"`
|
||||
DomainNameLen uint16 `struc:"little"`
|
||||
DomainNameMaxLen uint16 `struc:"little"`
|
||||
DomainNameBufferOffset uint32 `struc:"little"`
|
||||
WorkstationLen uint16 `struc:"little"`
|
||||
WorkstationMaxLen uint16 `struc:"little"`
|
||||
WorkstationBufferOffset uint32 `struc:"little"`
|
||||
Version NVersion `struc:"little"`
|
||||
Payload [32]byte `struc:"skip"`
|
||||
}
|
||||
|
||||
func NewNegotiateMessage() *NegotiateMessage {
|
||||
return &NegotiateMessage{
|
||||
Signature: [8]byte{'N', 'T', 'L', 'M', 'S', 'S', 'P', 0x00},
|
||||
MessageType: 0x00000001,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *NegotiateMessage) Serialize() []byte {
|
||||
if (m.NegotiateFlags & NTLMSSP_NEGOTIATE_VERSION) != 0 {
|
||||
m.Version = NewNVersion()
|
||||
}
|
||||
buff := &bytes.Buffer{}
|
||||
struc.Pack(buff, m)
|
||||
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type ChallengeMessage struct {
|
||||
Signature []byte `struc:"[8]byte"`
|
||||
MessageType uint32 `struc:"little"`
|
||||
TargetNameLen uint16 `struc:"little"`
|
||||
TargetNameMaxLen uint16 `struc:"little"`
|
||||
TargetNameBufferOffset uint32 `struc:"little"`
|
||||
NegotiateFlags uint32 `struc:"little"`
|
||||
ServerChallenge [8]byte `struc:"little"`
|
||||
Reserved [8]byte `struc:"little"`
|
||||
TargetInfoLen uint16 `struc:"little"`
|
||||
TargetInfoMaxLen uint16 `struc:"little"`
|
||||
TargetInfoBufferOffset uint32 `struc:"little"`
|
||||
Version NVersion `struc:"skip"`
|
||||
Payload []byte `struc:"skip"`
|
||||
}
|
||||
|
||||
func (m *ChallengeMessage) Serialize() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
struc.Pack(buff, m)
|
||||
if (m.NegotiateFlags & NTLMSSP_NEGOTIATE_VERSION) != 0 {
|
||||
struc.Pack(buff, m.Version)
|
||||
}
|
||||
buff.Write(m.Payload)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
func NewChallengeMessage() *ChallengeMessage {
|
||||
return &ChallengeMessage{
|
||||
Signature: []byte{'N', 'T', 'L', 'M', 'S', 'S', 'P', 0x00},
|
||||
MessageType: 0x00000002,
|
||||
}
|
||||
}
|
||||
|
||||
// total len - payload len
|
||||
func (m *ChallengeMessage) BaseLen() uint32 {
|
||||
return 56
|
||||
}
|
||||
|
||||
func (m *ChallengeMessage) getTargetInfo() []byte {
|
||||
if m.TargetInfoLen == 0 {
|
||||
return make([]byte, 0)
|
||||
}
|
||||
offset := m.BaseLen()
|
||||
start := m.TargetInfoBufferOffset - offset
|
||||
return m.Payload[start : start+uint32(m.TargetInfoLen)]
|
||||
}
|
||||
func (m *ChallengeMessage) getTargetName() []byte {
|
||||
if m.TargetNameLen == 0 {
|
||||
return make([]byte, 0)
|
||||
}
|
||||
offset := m.BaseLen()
|
||||
start := m.TargetNameBufferOffset - offset
|
||||
return m.Payload[start : start+uint32(m.TargetNameLen)]
|
||||
}
|
||||
func (m *ChallengeMessage) getTargetInfoTimestamp(data []byte) []byte {
|
||||
r := bytes.NewReader(data)
|
||||
for r.Len() > 0 {
|
||||
avPair := &AVPair{}
|
||||
struc.Unpack(r, avPair)
|
||||
if avPair.Id == MsvAvTimestamp {
|
||||
return avPair.Value
|
||||
}
|
||||
|
||||
if avPair.Id == MsvAvEOL {
|
||||
break
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type AuthenticateMessage struct {
|
||||
Signature [8]byte
|
||||
MessageType uint32 `struc:"little"`
|
||||
LmChallengeResponseLen uint16 `struc:"little"`
|
||||
LmChallengeResponseMaxLen uint16 `struc:"little"`
|
||||
LmChallengeResponseBufferOffset uint32 `struc:"little"`
|
||||
NtChallengeResponseLen uint16 `struc:"little"`
|
||||
NtChallengeResponseMaxLen uint16 `struc:"little"`
|
||||
NtChallengeResponseBufferOffset uint32 `struc:"little"`
|
||||
DomainNameLen uint16 `struc:"little"`
|
||||
DomainNameMaxLen uint16 `struc:"little"`
|
||||
DomainNameBufferOffset uint32 `struc:"little"`
|
||||
UserNameLen uint16 `struc:"little"`
|
||||
UserNameMaxLen uint16 `struc:"little"`
|
||||
UserNameBufferOffset uint32 `struc:"little"`
|
||||
WorkstationLen uint16 `struc:"little"`
|
||||
WorkstationMaxLen uint16 `struc:"little"`
|
||||
WorkstationBufferOffset uint32 `struc:"little"`
|
||||
EncryptedRandomSessionLen uint16 `struc:"little"`
|
||||
EncryptedRandomSessionMaxLen uint16 `struc:"little"`
|
||||
EncryptedRandomSessionBufferOffset uint32 `struc:"little"`
|
||||
NegotiateFlags uint32 `struc:"little"`
|
||||
Version NVersion `struc:"little"`
|
||||
MIC [16]byte `struc:"little"`
|
||||
Payload []byte `struc:"skip"`
|
||||
}
|
||||
|
||||
func (m *AuthenticateMessage) BaseLen() uint32 {
|
||||
return 88
|
||||
}
|
||||
|
||||
func NewAuthenticateMessage(negFlag uint32, domain, user, workstation []byte,
|
||||
lmchallResp, ntchallResp, enRandomSessKey []byte) *AuthenticateMessage {
|
||||
msg := &AuthenticateMessage{
|
||||
Signature: [8]byte{'N', 'T', 'L', 'M', 'S', 'S', 'P', 0x00},
|
||||
MessageType: 0x00000003,
|
||||
NegotiateFlags: negFlag,
|
||||
}
|
||||
payloadBuff := &bytes.Buffer{}
|
||||
|
||||
msg.LmChallengeResponseLen = uint16(len(lmchallResp))
|
||||
msg.LmChallengeResponseMaxLen = msg.LmChallengeResponseLen
|
||||
msg.LmChallengeResponseBufferOffset = msg.BaseLen()
|
||||
payloadBuff.Write(lmchallResp)
|
||||
|
||||
msg.NtChallengeResponseLen = uint16(len(ntchallResp))
|
||||
msg.NtChallengeResponseMaxLen = msg.NtChallengeResponseLen
|
||||
msg.NtChallengeResponseBufferOffset = msg.LmChallengeResponseBufferOffset + uint32(msg.LmChallengeResponseLen)
|
||||
payloadBuff.Write(ntchallResp)
|
||||
|
||||
msg.DomainNameLen = uint16(len(domain))
|
||||
msg.DomainNameMaxLen = msg.DomainNameLen
|
||||
msg.DomainNameBufferOffset = msg.NtChallengeResponseBufferOffset + uint32(msg.NtChallengeResponseLen)
|
||||
payloadBuff.Write(domain)
|
||||
|
||||
msg.UserNameLen = uint16(len(user))
|
||||
msg.UserNameMaxLen = msg.UserNameLen
|
||||
msg.UserNameBufferOffset = msg.DomainNameBufferOffset + uint32(msg.DomainNameLen)
|
||||
payloadBuff.Write(user)
|
||||
|
||||
msg.WorkstationLen = uint16(len(workstation))
|
||||
msg.WorkstationMaxLen = msg.WorkstationLen
|
||||
msg.WorkstationBufferOffset = msg.UserNameBufferOffset + uint32(msg.UserNameLen)
|
||||
payloadBuff.Write(workstation)
|
||||
|
||||
msg.EncryptedRandomSessionLen = uint16(len(enRandomSessKey))
|
||||
msg.EncryptedRandomSessionMaxLen = msg.EncryptedRandomSessionLen
|
||||
msg.EncryptedRandomSessionBufferOffset = msg.WorkstationBufferOffset + uint32(msg.WorkstationLen)
|
||||
payloadBuff.Write(enRandomSessKey)
|
||||
|
||||
if (msg.NegotiateFlags & NTLMSSP_NEGOTIATE_VERSION) != 0 {
|
||||
msg.Version = NewNVersion()
|
||||
}
|
||||
msg.Payload = payloadBuff.Bytes()
|
||||
|
||||
return msg
|
||||
}
|
||||
|
||||
func (m *AuthenticateMessage) Serialize() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
struc.Pack(buff, m)
|
||||
buff.Write(m.Payload)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type NTLMv2 struct {
|
||||
domain string
|
||||
user string
|
||||
password string
|
||||
respKeyNT []byte
|
||||
respKeyLM []byte
|
||||
negotiateMessage *NegotiateMessage
|
||||
challengeMessage *ChallengeMessage
|
||||
authenticateMessage *AuthenticateMessage
|
||||
enableUnicode bool
|
||||
}
|
||||
|
||||
func NewNTLMv2(domain, user, password string) *NTLMv2 {
|
||||
return &NTLMv2{
|
||||
domain: domain,
|
||||
user: user,
|
||||
password: password,
|
||||
respKeyNT: NTOWFv2(password, user, domain),
|
||||
respKeyLM: LMOWFv2(password, user, domain),
|
||||
}
|
||||
}
|
||||
|
||||
// generate first handshake messgae
|
||||
func (n *NTLMv2) GetNegotiateMessage() *NegotiateMessage {
|
||||
negoMsg := NewNegotiateMessage()
|
||||
negoMsg.NegotiateFlags = NTLMSSP_NEGOTIATE_KEY_EXCH |
|
||||
NTLMSSP_NEGOTIATE_128 |
|
||||
NTLMSSP_NEGOTIATE_EXTENDED_SESSIONSECURITY |
|
||||
NTLMSSP_NEGOTIATE_ALWAYS_SIGN |
|
||||
NTLMSSP_NEGOTIATE_NTLM |
|
||||
NTLMSSP_NEGOTIATE_SEAL |
|
||||
NTLMSSP_NEGOTIATE_SIGN |
|
||||
NTLMSSP_REQUEST_TARGET |
|
||||
NTLMSSP_NEGOTIATE_UNICODE
|
||||
n.negotiateMessage = negoMsg
|
||||
return n.negotiateMessage
|
||||
}
|
||||
|
||||
// process NTLMv2 Authenticate hash
|
||||
func (n *NTLMv2) ComputeResponseV2(respKeyNT, respKeyLM, serverChallenge, clientChallenge,
|
||||
timestamp, serverInfo []byte) (ntChallResp, lmChallResp, SessBaseKey []byte) {
|
||||
|
||||
tempBuff := &bytes.Buffer{}
|
||||
tempBuff.Write([]byte{0x01, 0x01}) // Responser version, HiResponser version
|
||||
tempBuff.Write([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00})
|
||||
tempBuff.Write(timestamp)
|
||||
tempBuff.Write(clientChallenge)
|
||||
tempBuff.Write([]byte{0x00, 0x00, 0x00, 0x00})
|
||||
tempBuff.Write(serverInfo)
|
||||
tempBuff.Write([]byte{0x00, 0x00, 0x00, 0x00})
|
||||
|
||||
ntBuf := bytes.NewBuffer(serverChallenge)
|
||||
ntBuf.Write(tempBuff.Bytes())
|
||||
ntProof := HMAC_MD5(respKeyNT, ntBuf.Bytes())
|
||||
|
||||
ntChallResp = make([]byte, 0, len(ntProof)+tempBuff.Len())
|
||||
ntChallResp = append(ntChallResp, ntProof...)
|
||||
ntChallResp = append(ntChallResp, tempBuff.Bytes()...)
|
||||
|
||||
lmBuf := bytes.NewBuffer(serverChallenge)
|
||||
lmBuf.Write(clientChallenge)
|
||||
lmChallResp = HMAC_MD5(respKeyLM, lmBuf.Bytes())
|
||||
lmChallResp = append(lmChallResp, clientChallenge...)
|
||||
|
||||
SessBaseKey = HMAC_MD5(respKeyNT, ntProof)
|
||||
return
|
||||
}
|
||||
|
||||
func MIC(exportedSessionKey []byte, negotiateMessage, challengeMessage, authenticateMessage Message) []byte {
|
||||
buff := bytes.Buffer{}
|
||||
buff.Write(negotiateMessage.Serialize())
|
||||
buff.Write(challengeMessage.Serialize())
|
||||
buff.Write(authenticateMessage.Serialize())
|
||||
return HMAC_MD5(exportedSessionKey, buff.Bytes())
|
||||
}
|
||||
|
||||
func concat(bs ...[]byte) []byte {
|
||||
return bytes.Join(bs, nil)
|
||||
}
|
||||
|
||||
var (
|
||||
clientSigning = concat([]byte("session key to client-to-server signing key magic constant"), []byte{0x00})
|
||||
serverSigning = concat([]byte("session key to server-to-client signing key magic constant"), []byte{0x00})
|
||||
clientSealing = concat([]byte("session key to client-to-server sealing key magic constant"), []byte{0x00})
|
||||
serverSealing = concat([]byte("session key to server-to-client sealing key magic constant"), []byte{0x00})
|
||||
)
|
||||
|
||||
func (n *NTLMv2) GetAuthenticateMessage(s []byte) (*AuthenticateMessage, *NTLMv2Security) {
|
||||
challengeMsg := &ChallengeMessage{}
|
||||
r := bytes.NewReader(s)
|
||||
err := struc.Unpack(r, challengeMsg)
|
||||
if err != nil {
|
||||
glog.Error("read challengeMsg", err)
|
||||
return nil, nil
|
||||
}
|
||||
if challengeMsg.NegotiateFlags&NTLMSSP_NEGOTIATE_VERSION != 0 {
|
||||
version := NVersion{}
|
||||
err := struc.Unpack(r, &version)
|
||||
if err != nil {
|
||||
glog.Error("read version", err)
|
||||
return nil, nil
|
||||
}
|
||||
challengeMsg.Version = version
|
||||
}
|
||||
challengeMsg.Payload, _ = core.ReadBytes(r.Len(), r)
|
||||
n.challengeMessage = challengeMsg
|
||||
glog.Debugf("challengeMsg:%+v", challengeMsg)
|
||||
|
||||
serverName := challengeMsg.getTargetName()
|
||||
serverInfo := challengeMsg.getTargetInfo()
|
||||
timestamp := challengeMsg.getTargetInfoTimestamp(serverInfo)
|
||||
computeMIC := false
|
||||
if timestamp == nil {
|
||||
ft := uint64(time.Now().UnixNano()) / 100
|
||||
ft += 116444736000000000 // add time between unix & windows offset
|
||||
timestamp = make([]byte, 8)
|
||||
binary.LittleEndian.PutUint64(timestamp, ft)
|
||||
} else {
|
||||
computeMIC = true
|
||||
}
|
||||
glog.Infof("serverName=%+v", core.UnicodeDecode(serverName))
|
||||
serverChallenge := challengeMsg.ServerChallenge[:]
|
||||
clientChallenge := core.Random(8)
|
||||
ntChallengeResponse, lmChallengeResponse, SessionBaseKey := n.ComputeResponseV2(
|
||||
n.respKeyNT, n.respKeyLM, serverChallenge, clientChallenge, timestamp, serverInfo)
|
||||
|
||||
exchangeKey := SessionBaseKey
|
||||
exportedSessionKey := core.Random(16)
|
||||
EncryptedRandomSessionKey := make([]byte, len(exportedSessionKey))
|
||||
rc, _ := rc4.NewCipher(exchangeKey)
|
||||
rc.XORKeyStream(EncryptedRandomSessionKey, exportedSessionKey)
|
||||
|
||||
if challengeMsg.NegotiateFlags&NTLMSSP_NEGOTIATE_UNICODE != 0 {
|
||||
n.enableUnicode = true
|
||||
}
|
||||
glog.Infof("user: %s, passwd:%s", n.user, n.password)
|
||||
domain, user, _ := n.GetEncodedCredentials()
|
||||
|
||||
n.authenticateMessage = NewAuthenticateMessage(challengeMsg.NegotiateFlags,
|
||||
domain, user, []byte(""), lmChallengeResponse, ntChallengeResponse, EncryptedRandomSessionKey)
|
||||
|
||||
if computeMIC {
|
||||
copy(n.authenticateMessage.MIC[:], MIC(exportedSessionKey, n.negotiateMessage, n.challengeMessage, n.authenticateMessage)[:16])
|
||||
}
|
||||
|
||||
md := md5.New()
|
||||
//ClientSigningKey
|
||||
a := concat(exportedSessionKey, clientSigning)
|
||||
md.Write(a)
|
||||
ClientSigningKey := md.Sum(nil)
|
||||
//ServerSigningKey
|
||||
md.Reset()
|
||||
a = concat(exportedSessionKey, serverSigning)
|
||||
md.Write(a)
|
||||
ServerSigningKey := md.Sum(nil)
|
||||
//ClientSealingKey
|
||||
md.Reset()
|
||||
a = concat(exportedSessionKey, clientSealing)
|
||||
md.Write(a)
|
||||
ClientSealingKey := md.Sum(nil)
|
||||
//ServerSealingKey
|
||||
md.Reset()
|
||||
a = concat(exportedSessionKey, serverSealing)
|
||||
md.Write(a)
|
||||
ServerSealingKey := md.Sum(nil)
|
||||
|
||||
glog.Debugf("ClientSigningKey:%s", hex.EncodeToString(ClientSigningKey))
|
||||
glog.Debugf("ServerSigningKey:%s", hex.EncodeToString(ServerSigningKey))
|
||||
glog.Debugf("ClientSealingKey:%s", hex.EncodeToString(ClientSealingKey))
|
||||
glog.Debugf("ServerSealingKey:%s", hex.EncodeToString(ServerSealingKey))
|
||||
|
||||
encryptRC4, _ := rc4.NewCipher(ClientSealingKey)
|
||||
decryptRC4, _ := rc4.NewCipher(ServerSealingKey)
|
||||
|
||||
ntlmSec := &NTLMv2Security{encryptRC4, decryptRC4, ClientSigningKey, ServerSigningKey, 0}
|
||||
|
||||
return n.authenticateMessage, ntlmSec
|
||||
}
|
||||
|
||||
func (n *NTLMv2) GetEncodedCredentials() ([]byte, []byte, []byte) {
|
||||
if n.enableUnicode {
|
||||
return core.UnicodeEncode(n.domain), core.UnicodeEncode(n.user), core.UnicodeEncode(n.password)
|
||||
}
|
||||
return []byte(n.domain), []byte(n.user), []byte(n.password)
|
||||
}
|
||||
|
||||
type NTLMv2Security struct {
|
||||
EncryptRC4 *rc4.Cipher
|
||||
DecryptRC4 *rc4.Cipher
|
||||
SigningKey []byte
|
||||
VerifyKey []byte
|
||||
SeqNum uint32
|
||||
}
|
||||
|
||||
func (n *NTLMv2Security) GssEncrypt(s []byte) []byte {
|
||||
p := make([]byte, len(s))
|
||||
n.EncryptRC4.XORKeyStream(p, s)
|
||||
b := &bytes.Buffer{}
|
||||
|
||||
//signature
|
||||
core.WriteUInt32LE(n.SeqNum, b)
|
||||
core.WriteBytes(s, b)
|
||||
s1 := HMAC_MD5(n.SigningKey, b.Bytes())[:8]
|
||||
checksum := make([]byte, 8)
|
||||
n.EncryptRC4.XORKeyStream(checksum, s1)
|
||||
b.Reset()
|
||||
core.WriteUInt32LE(0x00000001, b)
|
||||
core.WriteBytes(checksum, b)
|
||||
core.WriteUInt32LE(n.SeqNum, b)
|
||||
|
||||
core.WriteBytes(p, b)
|
||||
|
||||
n.SeqNum++
|
||||
|
||||
return b.Bytes()
|
||||
}
|
||||
func (n *NTLMv2Security) GssDecrypt(s []byte) []byte {
|
||||
r := bytes.NewReader(s)
|
||||
core.ReadUInt32LE(r) //version
|
||||
checksum, _ := core.ReadBytes(8, r)
|
||||
seqNum, _ := core.ReadUInt32LE(r)
|
||||
data, _ := core.ReadBytes(r.Len(), r)
|
||||
|
||||
p := make([]byte, len(data))
|
||||
n.DecryptRC4.XORKeyStream(p, data)
|
||||
|
||||
check := make([]byte, len(checksum))
|
||||
n.DecryptRC4.XORKeyStream(check, checksum)
|
||||
|
||||
b := &bytes.Buffer{}
|
||||
core.WriteUInt32LE(seqNum, b)
|
||||
core.WriteBytes(p, b)
|
||||
verify := HMAC_MD5(n.VerifyKey, b.Bytes())
|
||||
if string(verify) != string(check) {
|
||||
return nil
|
||||
}
|
||||
return p
|
||||
}
|
||||
@@ -0,0 +1,760 @@
|
||||
package pdu
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
|
||||
"github.com/lunixbochs/struc"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/gcc"
|
||||
)
|
||||
|
||||
type CapsType uint16
|
||||
|
||||
const (
|
||||
CAPSTYPE_GENERAL CapsType = 0x0001
|
||||
CAPSTYPE_BITMAP = 0x0002
|
||||
CAPSTYPE_ORDER = 0x0003
|
||||
CAPSTYPE_BITMAPCACHE = 0x0004
|
||||
CAPSTYPE_CONTROL = 0x0005
|
||||
CAPSTYPE_ACTIVATION = 0x0007
|
||||
CAPSTYPE_POINTER = 0x0008
|
||||
CAPSTYPE_SHARE = 0x0009
|
||||
CAPSTYPE_COLORCACHE = 0x000A
|
||||
CAPSTYPE_SOUND = 0x000C
|
||||
CAPSTYPE_INPUT = 0x000D
|
||||
CAPSTYPE_FONT = 0x000E
|
||||
CAPSTYPE_BRUSH = 0x000F
|
||||
CAPSTYPE_GLYPHCACHE = 0x0010
|
||||
CAPSTYPE_OFFSCREENCACHE = 0x0011
|
||||
CAPSTYPE_BITMAPCACHE_HOSTSUPPORT = 0x0012
|
||||
CAPSTYPE_BITMAPCACHE_REV2 = 0x0013
|
||||
CAPSTYPE_VIRTUALCHANNEL = 0x0014
|
||||
CAPSTYPE_DRAWNINEGRIDCACHE = 0x0015
|
||||
CAPSTYPE_DRAWGDIPLUS = 0x0016
|
||||
CAPSTYPE_RAIL = 0x0017
|
||||
CAPSTYPE_WINDOW = 0x0018
|
||||
CAPSETTYPE_COMPDESK = 0x0019
|
||||
CAPSETTYPE_MULTIFRAGMENTUPDATE = 0x001A
|
||||
CAPSETTYPE_LARGE_POINTER = 0x001B
|
||||
CAPSETTYPE_SURFACE_COMMANDS = 0x001C
|
||||
CAPSETTYPE_BITMAP_CODECS = 0x001D
|
||||
CAPSSETTYPE_FRAME_ACKNOWLEDGE = 0x001E
|
||||
)
|
||||
|
||||
func (c CapsType) String() string {
|
||||
switch c {
|
||||
case CAPSTYPE_GENERAL:
|
||||
return "CAPSTYPE_GENERAL"
|
||||
case CAPSTYPE_BITMAP:
|
||||
return "CAPSTYPE_BITMAP"
|
||||
case CAPSTYPE_ORDER:
|
||||
return "CAPSTYPE_ORDER"
|
||||
case CAPSTYPE_BITMAPCACHE:
|
||||
return "CAPSTYPE_BITMAPCACHE"
|
||||
case CAPSTYPE_CONTROL:
|
||||
return "CAPSTYPE_CONTROL"
|
||||
case CAPSTYPE_ACTIVATION:
|
||||
return "CAPSTYPE_ACTIVATION"
|
||||
case CAPSTYPE_POINTER:
|
||||
return "CAPSTYPE_POINTER"
|
||||
case CAPSTYPE_SHARE:
|
||||
return "CAPSTYPE_SHARE"
|
||||
case CAPSTYPE_COLORCACHE:
|
||||
return "CAPSTYPE_COLORCACHE"
|
||||
case CAPSTYPE_SOUND:
|
||||
return "CAPSTYPE_SOUND"
|
||||
case CAPSTYPE_INPUT:
|
||||
return "CAPSTYPE_INPUT"
|
||||
case CAPSTYPE_FONT:
|
||||
return "CAPSTYPE_FONT"
|
||||
case CAPSTYPE_BRUSH:
|
||||
return "CAPSTYPE_BRUSH"
|
||||
case CAPSTYPE_GLYPHCACHE:
|
||||
return "CAPSTYPE_GLYPHCACHE"
|
||||
case CAPSTYPE_OFFSCREENCACHE:
|
||||
return "CAPSTYPE_OFFSCREENCACHE"
|
||||
case CAPSTYPE_BITMAPCACHE_HOSTSUPPORT:
|
||||
return "CAPSTYPE_BITMAPCACHE_HOSTSUPPORT"
|
||||
case CAPSTYPE_BITMAPCACHE_REV2:
|
||||
return "CAPSTYPE_BITMAPCACHE_REV2"
|
||||
case CAPSTYPE_VIRTUALCHANNEL:
|
||||
return "CAPSTYPE_VIRTUALCHANNEL"
|
||||
case CAPSTYPE_DRAWNINEGRIDCACHE:
|
||||
return "CAPSTYPE_DRAWNINEGRIDCACHE"
|
||||
case CAPSTYPE_DRAWGDIPLUS:
|
||||
return "CAPSTYPE_DRAWGDIPLUS"
|
||||
case CAPSTYPE_RAIL:
|
||||
return "CAPSTYPE_RAIL"
|
||||
case CAPSTYPE_WINDOW:
|
||||
return "CAPSTYPE_WINDOW"
|
||||
case CAPSETTYPE_COMPDESK:
|
||||
return "CAPSETTYPE_COMPDESK"
|
||||
case CAPSETTYPE_MULTIFRAGMENTUPDATE:
|
||||
return "CAPSETTYPE_MULTIFRAGMENTUPDATE"
|
||||
case CAPSETTYPE_LARGE_POINTER:
|
||||
return "CAPSETTYPE_LARGE_POINTER"
|
||||
case CAPSETTYPE_SURFACE_COMMANDS:
|
||||
return "CAPSETTYPE_SURFACE_COMMANDS"
|
||||
case CAPSETTYPE_BITMAP_CODECS:
|
||||
return "CAPSETTYPE_BITMAP_CODECS"
|
||||
case CAPSSETTYPE_FRAME_ACKNOWLEDGE:
|
||||
return "CAPSSETTYPE_FRAME_ACKNOWLEDGE"
|
||||
}
|
||||
|
||||
return "Unknown"
|
||||
}
|
||||
|
||||
type MajorType uint16
|
||||
|
||||
const (
|
||||
OSMAJORTYPE_UNSPECIFIED MajorType = 0x0000
|
||||
OSMAJORTYPE_WINDOWS = 0x0001
|
||||
OSMAJORTYPE_OS2 = 0x0002
|
||||
OSMAJORTYPE_MACINTOSH = 0x0003
|
||||
OSMAJORTYPE_UNIX = 0x0004
|
||||
OSMAJORTYPE_IOS = 0x0005
|
||||
OSMAJORTYPE_OSX = 0x0006
|
||||
OSMAJORTYPE_ANDROID = 0x0007
|
||||
)
|
||||
|
||||
type MinorType uint16
|
||||
|
||||
const (
|
||||
OSMINORTYPE_UNSPECIFIED MinorType = 0x0000
|
||||
OSMINORTYPE_WINDOWS_31X = 0x0001
|
||||
OSMINORTYPE_WINDOWS_95 = 0x0002
|
||||
OSMINORTYPE_WINDOWS_NT = 0x0003
|
||||
OSMINORTYPE_OS2_V21 = 0x0004
|
||||
OSMINORTYPE_POWER_PC = 0x0005
|
||||
OSMINORTYPE_MACINTOSH = 0x0006
|
||||
OSMINORTYPE_NATIVE_XSERVER = 0x0007
|
||||
OSMINORTYPE_PSEUDO_XSERVER = 0x0008
|
||||
OSMINORTYPE_WINDOWS_RT = 0x0009
|
||||
)
|
||||
|
||||
const (
|
||||
FASTPATH_OUTPUT_SUPPORTED uint16 = 0x0001
|
||||
NO_BITMAP_COMPRESSION_HDR = 0x0400
|
||||
LONG_CREDENTIALS_SUPPORTED = 0x0004
|
||||
AUTORECONNECT_SUPPORTED = 0x0008
|
||||
ENC_SALTED_CHECKSUM = 0x0010
|
||||
)
|
||||
|
||||
type OrderFlag uint16
|
||||
|
||||
const (
|
||||
NEGOTIATEORDERSUPPORT OrderFlag = 0x0002
|
||||
ZEROBOUNDSDELTASSUPPORT = 0x0008
|
||||
COLORINDEXSUPPORT = 0x0020
|
||||
SOLIDPATTERNBRUSHONLY = 0x0040
|
||||
ORDERFLAGS_EXTRA_FLAGS = 0x0080
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240556.aspx
|
||||
*/
|
||||
type Order uint8
|
||||
|
||||
const (
|
||||
TS_NEG_DSTBLT_INDEX Order = 0x00
|
||||
TS_NEG_PATBLT_INDEX = 0x01
|
||||
TS_NEG_SCRBLT_INDEX = 0x02
|
||||
TS_NEG_MEMBLT_INDEX = 0x03
|
||||
TS_NEG_MEM3BLT_INDEX = 0x04
|
||||
TS_NEG_DRAWNINEGRID_INDEX = 0x07
|
||||
TS_NEG_LINETO_INDEX = 0x08
|
||||
TS_NEG_MULTI_DRAWNINEGRID_INDEX = 0x09
|
||||
TS_NEG_SAVEBITMAP_INDEX = 0x0B
|
||||
TS_NEG_MULTIDSTBLT_INDEX = 0x0F
|
||||
TS_NEG_MULTIPATBLT_INDEX = 0x10
|
||||
TS_NEG_MULTISCRBLT_INDEX = 0x11
|
||||
TS_NEG_MULTIOPAQUERECT_INDEX = 0x12
|
||||
TS_NEG_FAST_INDEX_INDEX = 0x13
|
||||
TS_NEG_POLYGON_SC_INDEX = 0x14
|
||||
TS_NEG_POLYGON_CB_INDEX = 0x15
|
||||
TS_NEG_POLYLINE_INDEX = 0x16
|
||||
TS_NEG_FAST_GLYPH_INDEX = 0x18
|
||||
TS_NEG_ELLIPSE_SC_INDEX = 0x19
|
||||
TS_NEG_ELLIPSE_CB_INDEX = 0x1A
|
||||
TS_NEG_GLYPH_INDEX_INDEX = 0x1B
|
||||
)
|
||||
|
||||
type OrderEx uint16
|
||||
|
||||
const (
|
||||
ORDERFLAGS_EX_CACHE_BITMAP_REV3_SUPPORT OrderEx = 0x0002
|
||||
ORDERFLAGS_EX_ALTSEC_FRAME_MARKER_SUPPORT = 0x0004
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240563.aspx
|
||||
*/
|
||||
|
||||
const (
|
||||
INPUT_FLAG_SCANCODES uint16 = 0x0001
|
||||
INPUT_FLAG_MOUSEX = 0x0004
|
||||
INPUT_FLAG_FASTPATH_INPUT = 0x0008
|
||||
INPUT_FLAG_UNICODE = 0x0010
|
||||
INPUT_FLAG_FASTPATH_INPUT2 = 0x0020
|
||||
INPUT_FLAG_UNUSED1 = 0x0040
|
||||
INPUT_FLAG_UNUSED2 = 0x0080
|
||||
INPUT_FLAG_MOUSE_HWHEEL = 0x0100
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240564.aspx
|
||||
*/
|
||||
type BrushSupport uint32
|
||||
|
||||
const (
|
||||
BRUSH_DEFAULT BrushSupport = 0x00000000
|
||||
BRUSH_COLOR_8x8 = 0x00000001
|
||||
BRUSH_COLOR_FULL = 0x00000002
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240565.aspx
|
||||
*/
|
||||
type GlyphSupport uint16
|
||||
|
||||
const (
|
||||
GLYPH_SUPPORT_NONE GlyphSupport = 0x0000
|
||||
GLYPH_SUPPORT_PARTIAL = 0x0001
|
||||
GLYPH_SUPPORT_FULL = 0x0002
|
||||
GLYPH_SUPPORT_ENCODE = 0x0003
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240550.aspx
|
||||
*/
|
||||
type OffscreenSupportLevel uint32
|
||||
|
||||
const (
|
||||
OSL_FALSE OffscreenSupportLevel = 0x00000000
|
||||
OSL_TRUE = 0x00000001
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240551.aspx
|
||||
*/
|
||||
type VirtualChannelCompressionFlag uint32
|
||||
|
||||
const (
|
||||
VCCAPS_NO_COMPR VirtualChannelCompressionFlag = 0x00000000
|
||||
VCCAPS_COMPR_SC = 0x00000001
|
||||
VCCAPS_COMPR_CS_8K = 0x00000002
|
||||
)
|
||||
|
||||
type SoundFlag uint16
|
||||
|
||||
const (
|
||||
SOUND_NONE SoundFlag = 0x0000
|
||||
SOUND_BEEPS_FLAG = 0x0001
|
||||
)
|
||||
|
||||
type RailsupportLevel uint32
|
||||
|
||||
const (
|
||||
RAIL_LEVEL_SUPPORTED = 0x00000001
|
||||
RAIL_LEVEL_DOCKED_LANGBAR_SUPPORTED = 0x00000002
|
||||
RAIL_LEVEL_SHELL_INTEGRATION_SUPPORTED = 0x00000004
|
||||
RAIL_LEVEL_LANGUAGE_IME_SYNC_SUPPORTED = 0x00000008
|
||||
RAIL_LEVEL_SERVER_TO_CLIENT_IME_SYNC_SUPPORTED = 0x00000010
|
||||
RAIL_LEVEL_HIDE_MINIMIZED_APPS_SUPPORTED = 0x00000020
|
||||
RAIL_LEVEL_WINDOW_CLOAKING_SUPPORTED = 0x00000040
|
||||
RAIL_LEVEL_HANDSHAKE_EX_SUPPORTED = 0x00000080
|
||||
)
|
||||
|
||||
const (
|
||||
INPUT_EVENT_SYNC = 0x0000
|
||||
INPUT_EVENT_UNUSED = 0x0002
|
||||
INPUT_EVENT_SCANCODE = 0x0004
|
||||
INPUT_EVENT_UNICODE = 0x0005
|
||||
INPUT_EVENT_MOUSE = 0x8001
|
||||
INPUT_EVENT_MOUSEX = 0x8002
|
||||
)
|
||||
|
||||
const (
|
||||
PTRFLAGS_HWHEEL = 0x0400
|
||||
PTRFLAGS_WHEEL = 0x0200
|
||||
PTRFLAGS_WHEEL_NEGATIVE = 0x0100
|
||||
WheelRotationMask = 0x01FF
|
||||
PTRFLAGS_MOVE = 0x0800
|
||||
PTRFLAGS_DOWN = 0x8000
|
||||
PTRFLAGS_BUTTON1 = 0x1000
|
||||
PTRFLAGS_BUTTON2 = 0x2000
|
||||
PTRFLAGS_BUTTON3 = 0x4000
|
||||
)
|
||||
|
||||
const (
|
||||
KBDFLAGS_EXTENDED = 0x0100
|
||||
KBDFLAGS_DOWN = 0x4000
|
||||
KBDFLAGS_RELEASE = 0x8000
|
||||
)
|
||||
|
||||
type SurfaceCmdFlags uint32
|
||||
|
||||
const (
|
||||
SURFCMDS_SET_SURFACE_BITS = 0x00000002
|
||||
SURFCMDS_FRAME_MARKER = 0x00000010
|
||||
SURFCMDS_STREAM_SURFACE_BITS = 0x00000040
|
||||
)
|
||||
|
||||
type Capability interface {
|
||||
Type() CapsType
|
||||
}
|
||||
|
||||
type GeneralCapability struct {
|
||||
// 010018000100030000020000000015040000000000000000
|
||||
OSMajorType MajorType `struc:"little"`
|
||||
OSMinorType MinorType `struc:"little"`
|
||||
ProtocolVersion uint16 `struc:"little"`
|
||||
Pad2octetsA uint16 `struc:"little"`
|
||||
GeneralCompressionTypes uint16 `struc:"little"`
|
||||
ExtraFlags uint16 `struc:"little"`
|
||||
UpdateCapabilityFlag uint16 `struc:"little"`
|
||||
RemoteUnshareFlag uint16 `struc:"little"`
|
||||
GeneralCompressionLevel uint16 `struc:"little"`
|
||||
RefreshRectSupport uint8 `struc:"little"`
|
||||
SuppressOutputSupport uint8 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*GeneralCapability) Type() CapsType {
|
||||
return CAPSTYPE_GENERAL
|
||||
}
|
||||
|
||||
type BitmapCapability struct {
|
||||
// 02001c00180001000100010000052003000000000100000001000000
|
||||
PreferredBitsPerPixel gcc.HighColor `struc:"little"`
|
||||
Receive1BitPerPixel uint16 `struc:"little"`
|
||||
Receive4BitsPerPixel uint16 `struc:"little"`
|
||||
Receive8BitsPerPixel uint16 `struc:"little"`
|
||||
DesktopWidth uint16 `struc:"little"`
|
||||
DesktopHeight uint16 `struc:"little"`
|
||||
Pad2octets uint16 `struc:"little"`
|
||||
DesktopResizeFlag uint16 `struc:"little"`
|
||||
BitmapCompressionFlag uint16 `struc:"little"`
|
||||
HighColorFlags uint8 `struc:"little"`
|
||||
DrawingFlags uint8 `struc:"little"`
|
||||
MultipleRectangleSupport uint16 `struc:"little"`
|
||||
Pad2octetsB uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*BitmapCapability) Type() CapsType {
|
||||
return CAPSTYPE_BITMAP
|
||||
}
|
||||
|
||||
type BitmapCacheCapability struct {
|
||||
// 04002800000000000000000000000000000000000000000000000000000000000000000000000000
|
||||
Pad1 uint32 `struc:"little"`
|
||||
Pad2 uint32 `struc:"little"`
|
||||
Pad3 uint32 `struc:"little"`
|
||||
Pad4 uint32 `struc:"little"`
|
||||
Pad5 uint32 `struc:"little"`
|
||||
Pad6 uint32 `struc:"little"`
|
||||
Cache0Entries uint16 `struc:"little"`
|
||||
Cache0MaximumCellSize uint16 `struc:"little"`
|
||||
Cache1Entries uint16 `struc:"little"`
|
||||
Cache1MaximumCellSize uint16 `struc:"little"`
|
||||
Cache2Entries uint16 `struc:"little"`
|
||||
Cache2MaximumCellSize uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*BitmapCacheCapability) Type() CapsType {
|
||||
return CAPSTYPE_BITMAPCACHE
|
||||
}
|
||||
|
||||
type OrderCapability struct {
|
||||
// 030058000000000000000000000000000000000000000000010014000000010000000a0000000000000000000000000000000000000000000000000000000000000000000000000000000000008403000000000000000000
|
||||
TerminalDescriptor [16]byte
|
||||
Pad4octetsA uint32 `struc:"little"`
|
||||
DesktopSaveXGranularity uint16 `struc:"little"`
|
||||
DesktopSaveYGranularity uint16 `struc:"little"`
|
||||
Pad2octetsA uint16 `struc:"little"`
|
||||
MaximumOrderLevel uint16 `struc:"little"`
|
||||
NumberFonts uint16 `struc:"little"`
|
||||
OrderFlags OrderFlag `struc:"little"`
|
||||
OrderSupport [32]byte
|
||||
TextFlags uint16 `struc:"little"`
|
||||
OrderSupportExFlags uint16 `struc:"little"`
|
||||
Pad4octetsB uint32 `struc:"little"`
|
||||
DesktopSaveSize uint32 `struc:"little"`
|
||||
Pad2octetsC uint16 `struc:"little"`
|
||||
Pad2octetsD uint16 `struc:"little"`
|
||||
TextANSICodePage uint16 `struc:"little"`
|
||||
Pad2octetsE uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*OrderCapability) Type() CapsType {
|
||||
return CAPSTYPE_ORDER
|
||||
}
|
||||
|
||||
type PointerCapability struct {
|
||||
ColorPointerFlag uint16 `struc:"little"`
|
||||
ColorPointerCacheSize uint16 `struc:"little"`
|
||||
// old version of rdp doesn't support ...
|
||||
PointerCacheSize uint16 `struc:"little"` // only server need
|
||||
}
|
||||
|
||||
func (*PointerCapability) Type() CapsType {
|
||||
return CAPSTYPE_POINTER
|
||||
}
|
||||
|
||||
type InputCapability struct {
|
||||
// 0d005c001500000009040000040000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000c000000
|
||||
Flags uint16 `struc:"little"`
|
||||
Pad2octetsA uint16 `struc:"little"`
|
||||
// same value as gcc.ClientCoreSettings.kbdLayout
|
||||
KeyboardLayout gcc.KeyboardLayout `struc:"little"`
|
||||
// same value as gcc.ClientCoreSettings.keyboardType
|
||||
KeyboardType uint32 `struc:"little"`
|
||||
// same value as gcc.ClientCoreSettings.keyboardSubType
|
||||
KeyboardSubType uint32 `struc:"little"`
|
||||
// same value as gcc.ClientCoreSettings.keyboardFnKeys
|
||||
KeyboardFunctionKey uint32 `struc:"little"`
|
||||
// same value as gcc.ClientCoreSettingrrs.imeFileName
|
||||
ImeFileName [64]byte
|
||||
//need add 0c000000 in the end
|
||||
}
|
||||
|
||||
func (*InputCapability) Type() CapsType {
|
||||
return CAPSTYPE_INPUT
|
||||
}
|
||||
|
||||
type BrushCapability struct {
|
||||
// 0f00080000000000
|
||||
SupportLevel BrushSupport `struc:"little"`
|
||||
}
|
||||
|
||||
func (*BrushCapability) Type() CapsType {
|
||||
return CAPSTYPE_BRUSH
|
||||
}
|
||||
|
||||
type cacheEntry struct {
|
||||
Entries uint16 `struc:"little"`
|
||||
MaximumCellSize uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
type GlyphCapability struct {
|
||||
// 10003400000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000
|
||||
GlyphCache [10]cacheEntry `struc:"little"`
|
||||
FragCache uint32 `struc:"little"`
|
||||
SupportLevel GlyphSupport `struc:"little"`
|
||||
Pad2octets uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*GlyphCapability) Type() CapsType {
|
||||
return CAPSTYPE_GLYPHCACHE
|
||||
}
|
||||
|
||||
type OffscreenBitmapCacheCapability struct {
|
||||
// 11000c000000000000000000
|
||||
SupportLevel OffscreenSupportLevel `struc:"little"`
|
||||
CacheSize uint16 `struc:"little"`
|
||||
CacheEntries uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*OffscreenBitmapCacheCapability) Type() CapsType {
|
||||
return CAPSTYPE_OFFSCREENCACHE
|
||||
}
|
||||
|
||||
type BitmapCache2Capability struct {
|
||||
BitmapCachePersist uint16 `struc:"little"`
|
||||
Pad2octets uint8 `struc:"little"`
|
||||
CachesNum uint8 `struc:"little"`
|
||||
BmpC0Cells uint32 `struc:"little"`
|
||||
BmpC1Cells uint32 `struc:"little"`
|
||||
BmpC2Cells uint32 `struc:"little"`
|
||||
BmpC3Cells uint32 `struc:"little"`
|
||||
BmpC4Cells uint32 `struc:"little"`
|
||||
Pad2octets1 [12]byte `struc:"little"`
|
||||
}
|
||||
|
||||
func (*BitmapCache2Capability) Type() CapsType {
|
||||
return CAPSTYPE_BITMAPCACHE_REV2
|
||||
}
|
||||
|
||||
type VirtualChannelCapability struct {
|
||||
// 14000c000000000000000000
|
||||
Flags VirtualChannelCompressionFlag `struc:"little"`
|
||||
VCChunkSize uint32 `struc:"little"` // optional
|
||||
}
|
||||
|
||||
func (*VirtualChannelCapability) Type() CapsType {
|
||||
return CAPSTYPE_VIRTUALCHANNEL
|
||||
}
|
||||
|
||||
type SoundCapability struct {
|
||||
// 0c00080000000000
|
||||
Flags SoundFlag `struc:"little"`
|
||||
Pad2octets uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*SoundCapability) Type() CapsType {
|
||||
return CAPSTYPE_SOUND
|
||||
}
|
||||
|
||||
type ControlCapability struct {
|
||||
ControlFlags uint16 `struc:"little"`
|
||||
RemoteDetachFlag uint16 `struc:"little"`
|
||||
ControlInterest uint16 `struc:"little"`
|
||||
DetachInterest uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*ControlCapability) Type() CapsType {
|
||||
return CAPSTYPE_CONTROL
|
||||
}
|
||||
|
||||
type WindowActivationCapability struct {
|
||||
HelpKeyFlag uint16 `struc:"little"`
|
||||
HelpKeyIndexFlag uint16 `struc:"little"`
|
||||
HelpExtendedKeyFlag uint16 `struc:"little"`
|
||||
WindowManagerKeyFlag uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*WindowActivationCapability) Type() CapsType {
|
||||
return CAPSTYPE_ACTIVATION
|
||||
}
|
||||
|
||||
type FontCapability struct {
|
||||
SupportFlags uint16 `struc:"little"`
|
||||
Pad2octets uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*FontCapability) Type() CapsType {
|
||||
return CAPSTYPE_FONT
|
||||
}
|
||||
|
||||
type ColorCacheCapability struct {
|
||||
CacheSize uint16 `struc:"little"`
|
||||
Pad2octets uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*ColorCacheCapability) Type() CapsType {
|
||||
return CAPSTYPE_COLORCACHE
|
||||
}
|
||||
|
||||
type ShareCapability struct {
|
||||
NodeId uint16 `struc:"little"`
|
||||
Pad2octets uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*ShareCapability) Type() CapsType {
|
||||
return CAPSTYPE_SHARE
|
||||
}
|
||||
|
||||
type MultiFragmentUpdate struct {
|
||||
// 1a00080000000000
|
||||
MaxRequestSize uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*MultiFragmentUpdate) Type() CapsType {
|
||||
return CAPSETTYPE_MULTIFRAGMENTUPDATE
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpegdi/52635737-d144-4f47-9c88-b48ceaf3efb4
|
||||
|
||||
type DrawGDIPlusCapability struct {
|
||||
SupportLevel uint32
|
||||
GdipVersion uint32
|
||||
CacheLevel uint32
|
||||
GdipCacheEntries [10]byte
|
||||
GdipCacheChunkSize [8]byte
|
||||
GdipImageCacheProperties [6]byte
|
||||
}
|
||||
|
||||
func (*DrawGDIPlusCapability) Type() CapsType {
|
||||
return CAPSTYPE_DRAWGDIPLUS
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/86507fed-a0ee-4242-b802-237534a8f65e
|
||||
type BitmapCodec struct {
|
||||
GUID [16]byte
|
||||
ID uint8
|
||||
PropertiesLength uint16 `struc:"little,sizeof=Properties"`
|
||||
Properties []byte
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/408b1878-9f6e-4106-8329-1af42219ba6a
|
||||
type BitmapCodecS struct {
|
||||
Count uint8 `struc:"sizeof=Array"`
|
||||
Array []BitmapCodec
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/17e80f50-d163-49de-a23b-fd6456aa472f
|
||||
type BitmapCodecsCapability struct {
|
||||
SupportedBitmapCodecs BitmapCodecS // A variable-length field containing a TS_BITMAPCODECS structure (section 2.2.7.2.10.1).
|
||||
}
|
||||
|
||||
func (*BitmapCodecsCapability) Type() CapsType {
|
||||
return CAPSETTYPE_BITMAP_CODECS
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/fc05c385-46c3-42cb-9ed2-c475a3990e0b
|
||||
type BitmapCacheHostSupportCapability struct {
|
||||
CacheVersion uint8
|
||||
Pad1 uint8
|
||||
Pad2 uint16
|
||||
}
|
||||
|
||||
func (*BitmapCacheHostSupportCapability) Type() CapsType {
|
||||
return CAPSTYPE_BITMAPCACHE_HOSTSUPPORT
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/41323437-c753-460e-8108-495a6fdd68a8
|
||||
type LargePointerCapability struct {
|
||||
SupportFlags uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*LargePointerCapability) Type() CapsType {
|
||||
return CAPSETTYPE_LARGE_POINTER
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdperp/36a25e21-25e1-4954-aae8-09aaf6715c79
|
||||
type RemoteProgramsCapability struct {
|
||||
RailSupportLevel uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*RemoteProgramsCapability) Type() CapsType {
|
||||
return CAPSTYPE_RAIL
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdperp/82ec7a69-f7e3-4294-830d-666178b35d15
|
||||
type WindowListCapability struct {
|
||||
WndSupportLevel uint32 `struc:"little"`
|
||||
NumIconCaches uint8
|
||||
NumIconCacheEntries uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*WindowListCapability) Type() CapsType {
|
||||
return CAPSTYPE_WINDOW
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/9132002f-f133-4a0f-ba2f-2dc48f1e7f93
|
||||
type DesktopCompositionCapability struct {
|
||||
CompDeskSupportLevel uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*DesktopCompositionCapability) Type() CapsType {
|
||||
return CAPSETTYPE_COMPDESK
|
||||
}
|
||||
|
||||
// see https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-rdpbcgr/aa953018-c0a8-4761-bb12-86586c2cd56a
|
||||
type SurfaceCommandsCapability struct {
|
||||
CmdFlags uint32 `struc:"little"`
|
||||
Reserved uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*SurfaceCommandsCapability) Type() CapsType {
|
||||
return CAPSETTYPE_SURFACE_COMMANDS
|
||||
}
|
||||
|
||||
type FrameAcknowledgeCapability struct {
|
||||
FrameCount uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*FrameAcknowledgeCapability) Type() CapsType {
|
||||
return CAPSSETTYPE_FRAME_ACKNOWLEDGE
|
||||
}
|
||||
|
||||
type DrawNineGridCapability struct {
|
||||
SupportLevel uint32 `struc:"little"`
|
||||
CacheSize uint16 `struc:"little"`
|
||||
CacheEntries uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func (*DrawNineGridCapability) Type() CapsType {
|
||||
return CAPSTYPE_DRAWNINEGRIDCACHE
|
||||
}
|
||||
|
||||
func readCapability(r io.Reader) (Capability, error) {
|
||||
capType, err := core.ReadUint16LE(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
capLen, err := core.ReadUint16LE(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if int(capLen)-4 <= 0 {
|
||||
return nil, errors.New(fmt.Sprintf("Capability length expected %d", capLen))
|
||||
}
|
||||
|
||||
capBytes, err := core.ReadBytes(int(capLen)-4, r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
capReader := bytes.NewReader(capBytes)
|
||||
var c Capability
|
||||
glog.Debugf("Capability type 0x%04x", capType)
|
||||
switch CapsType(capType) {
|
||||
case CAPSTYPE_GENERAL:
|
||||
c = &GeneralCapability{}
|
||||
case CAPSTYPE_BITMAP:
|
||||
c = &BitmapCapability{}
|
||||
case CAPSTYPE_ORDER:
|
||||
c = &OrderCapability{}
|
||||
case CAPSTYPE_BITMAPCACHE:
|
||||
c = &BitmapCacheCapability{}
|
||||
case CAPSTYPE_POINTER:
|
||||
c = &PointerCapability{}
|
||||
case CAPSTYPE_INPUT:
|
||||
c = &InputCapability{}
|
||||
case CAPSTYPE_BRUSH:
|
||||
c = &BrushCapability{}
|
||||
case CAPSTYPE_GLYPHCACHE:
|
||||
c = &GlyphCapability{}
|
||||
case CAPSTYPE_OFFSCREENCACHE:
|
||||
c = &OffscreenBitmapCacheCapability{}
|
||||
case CAPSTYPE_VIRTUALCHANNEL:
|
||||
c = &VirtualChannelCapability{}
|
||||
case CAPSTYPE_SOUND:
|
||||
c = &SoundCapability{}
|
||||
case CAPSTYPE_CONTROL:
|
||||
c = &ControlCapability{}
|
||||
case CAPSTYPE_ACTIVATION:
|
||||
c = &WindowActivationCapability{}
|
||||
case CAPSTYPE_FONT:
|
||||
c = &FontCapability{}
|
||||
case CAPSTYPE_COLORCACHE:
|
||||
c = &ColorCacheCapability{}
|
||||
case CAPSTYPE_SHARE:
|
||||
c = &ShareCapability{}
|
||||
case CAPSETTYPE_MULTIFRAGMENTUPDATE:
|
||||
c = &MultiFragmentUpdate{}
|
||||
case CAPSTYPE_DRAWGDIPLUS:
|
||||
c = &DrawGDIPlusCapability{}
|
||||
case CAPSETTYPE_BITMAP_CODECS:
|
||||
c = &BitmapCodecsCapability{}
|
||||
case CAPSTYPE_BITMAPCACHE_HOSTSUPPORT:
|
||||
c = &BitmapCacheHostSupportCapability{}
|
||||
case CAPSETTYPE_LARGE_POINTER:
|
||||
c = &LargePointerCapability{}
|
||||
case CAPSTYPE_RAIL:
|
||||
c = &RemoteProgramsCapability{}
|
||||
case CAPSTYPE_WINDOW:
|
||||
c = &WindowListCapability{}
|
||||
case CAPSETTYPE_COMPDESK:
|
||||
c = &DesktopCompositionCapability{}
|
||||
case CAPSETTYPE_SURFACE_COMMANDS:
|
||||
c = &SurfaceCommandsCapability{}
|
||||
case CAPSSETTYPE_FRAME_ACKNOWLEDGE:
|
||||
c = &FrameAcknowledgeCapability{}
|
||||
default:
|
||||
err := errors.New(fmt.Sprintf("unsupported Capability type 0x%04x", capType))
|
||||
glog.Error(err)
|
||||
return nil, err
|
||||
}
|
||||
if err := struc.Unpack(capReader, c); err != nil {
|
||||
glog.Error("Capability unpack error", err, fmt.Sprintf("0x%04x", capType), hex.EncodeToString(capBytes))
|
||||
return nil, err
|
||||
}
|
||||
glog.Debugf("Capability<%s>: %+v", c.Type(), c)
|
||||
return c, nil
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,465 @@
|
||||
package pdu
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/hex"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/emission"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/gcc"
|
||||
)
|
||||
|
||||
type PDULayer struct {
|
||||
emission.Emitter
|
||||
transport core.Transport
|
||||
sharedId uint32
|
||||
userId uint16
|
||||
channelId uint16
|
||||
serverCapabilities map[CapsType]Capability
|
||||
clientCapabilities map[CapsType]Capability
|
||||
fastPathSender core.FastPathSender
|
||||
demandActivePDU *DemandActivePDU
|
||||
}
|
||||
|
||||
func NewPDULayer(t core.Transport) *PDULayer {
|
||||
p := &PDULayer{
|
||||
Emitter: *emission.NewEmitter(),
|
||||
transport: t,
|
||||
sharedId: 0x103EA,
|
||||
serverCapabilities: map[CapsType]Capability{
|
||||
CAPSTYPE_GENERAL: &GeneralCapability{
|
||||
ProtocolVersion: 0x0200,
|
||||
},
|
||||
CAPSTYPE_BITMAP: &BitmapCapability{
|
||||
Receive1BitPerPixel: 0x0001,
|
||||
Receive4BitsPerPixel: 0x0001,
|
||||
Receive8BitsPerPixel: 0x0001,
|
||||
BitmapCompressionFlag: 0x0001,
|
||||
MultipleRectangleSupport: 0x0001,
|
||||
},
|
||||
CAPSTYPE_ORDER: &OrderCapability{
|
||||
DesktopSaveXGranularity: 1,
|
||||
DesktopSaveYGranularity: 20,
|
||||
MaximumOrderLevel: 1,
|
||||
OrderFlags: NEGOTIATEORDERSUPPORT,
|
||||
DesktopSaveSize: 480 * 480,
|
||||
},
|
||||
CAPSTYPE_POINTER: &PointerCapability{ColorPointerCacheSize: 20},
|
||||
CAPSTYPE_INPUT: &InputCapability{},
|
||||
CAPSTYPE_VIRTUALCHANNEL: &VirtualChannelCapability{},
|
||||
CAPSTYPE_FONT: &FontCapability{SupportFlags: 0x0001},
|
||||
CAPSTYPE_COLORCACHE: &ColorCacheCapability{CacheSize: 0x0006},
|
||||
CAPSTYPE_SHARE: &ShareCapability{},
|
||||
},
|
||||
clientCapabilities: map[CapsType]Capability{
|
||||
CAPSTYPE_GENERAL: &GeneralCapability{
|
||||
ProtocolVersion: 0x0200,
|
||||
},
|
||||
CAPSTYPE_BITMAP: &BitmapCapability{
|
||||
Receive1BitPerPixel: 0x0001,
|
||||
Receive4BitsPerPixel: 0x0001,
|
||||
Receive8BitsPerPixel: 0x0001,
|
||||
BitmapCompressionFlag: 0x0001,
|
||||
MultipleRectangleSupport: 0x0001,
|
||||
},
|
||||
CAPSTYPE_ORDER: &OrderCapability{
|
||||
DesktopSaveXGranularity: 1,
|
||||
DesktopSaveYGranularity: 20,
|
||||
MaximumOrderLevel: 1,
|
||||
OrderFlags: NEGOTIATEORDERSUPPORT,
|
||||
DesktopSaveSize: 480 * 480,
|
||||
TextANSICodePage: 0x4e4,
|
||||
},
|
||||
CAPSTYPE_CONTROL: &ControlCapability{0, 0, 2, 2},
|
||||
CAPSTYPE_ACTIVATION: &WindowActivationCapability{},
|
||||
CAPSTYPE_POINTER: &PointerCapability{1, 20, 20},
|
||||
CAPSTYPE_SHARE: &ShareCapability{},
|
||||
CAPSTYPE_COLORCACHE: &ColorCacheCapability{6, 0},
|
||||
CAPSTYPE_SOUND: &SoundCapability{0x0001, 0},
|
||||
CAPSTYPE_INPUT: &InputCapability{},
|
||||
CAPSTYPE_FONT: &FontCapability{0x0001, 0},
|
||||
CAPSTYPE_BRUSH: &BrushCapability{BRUSH_COLOR_8x8},
|
||||
CAPSTYPE_GLYPHCACHE: &GlyphCapability{},
|
||||
CAPSETTYPE_BITMAP_CODECS: &BitmapCodecsCapability{},
|
||||
CAPSTYPE_BITMAPCACHE_REV2: &BitmapCache2Capability{
|
||||
BitmapCachePersist: 2,
|
||||
CachesNum: 5,
|
||||
BmpC0Cells: 0x258,
|
||||
BmpC1Cells: 0x258,
|
||||
BmpC2Cells: 0x800,
|
||||
BmpC3Cells: 0x1000,
|
||||
BmpC4Cells: 0x800,
|
||||
},
|
||||
CAPSTYPE_VIRTUALCHANNEL: &VirtualChannelCapability{0, 1600},
|
||||
CAPSETTYPE_MULTIFRAGMENTUPDATE: &MultiFragmentUpdate{65535},
|
||||
CAPSTYPE_RAIL: &RemoteProgramsCapability{
|
||||
RailSupportLevel: RAIL_LEVEL_SUPPORTED |
|
||||
RAIL_LEVEL_SHELL_INTEGRATION_SUPPORTED |
|
||||
RAIL_LEVEL_LANGUAGE_IME_SYNC_SUPPORTED |
|
||||
RAIL_LEVEL_SERVER_TO_CLIENT_IME_SYNC_SUPPORTED |
|
||||
RAIL_LEVEL_HIDE_MINIMIZED_APPS_SUPPORTED |
|
||||
RAIL_LEVEL_WINDOW_CLOAKING_SUPPORTED |
|
||||
RAIL_LEVEL_HANDSHAKE_EX_SUPPORTED |
|
||||
RAIL_LEVEL_DOCKED_LANGBAR_SUPPORTED,
|
||||
},
|
||||
CAPSETTYPE_LARGE_POINTER: &LargePointerCapability{1},
|
||||
CAPSETTYPE_SURFACE_COMMANDS: &SurfaceCommandsCapability{
|
||||
CmdFlags: SURFCMDS_SET_SURFACE_BITS | SURFCMDS_STREAM_SURFACE_BITS | SURFCMDS_FRAME_MARKER,
|
||||
},
|
||||
CAPSSETTYPE_FRAME_ACKNOWLEDGE: &FrameAcknowledgeCapability{2},
|
||||
},
|
||||
}
|
||||
|
||||
t.On("close", func() {
|
||||
p.Emit("close")
|
||||
}).On("error", func(err error) {
|
||||
p.Emit("error", err)
|
||||
})
|
||||
return p
|
||||
}
|
||||
|
||||
func (p *PDULayer) sendPDU(message PDUMessage) {
|
||||
pdu := NewPDU(p.userId, message)
|
||||
p.transport.Write(pdu.serialize())
|
||||
}
|
||||
|
||||
func (p *PDULayer) sendDataPDU(message DataPDUData) {
|
||||
dataPdu := NewDataPDU(message, p.sharedId)
|
||||
p.sendPDU(dataPdu)
|
||||
}
|
||||
|
||||
func (p *PDULayer) SetFastPathSender(f core.FastPathSender) {
|
||||
p.fastPathSender = f
|
||||
}
|
||||
|
||||
type Client struct {
|
||||
*PDULayer
|
||||
clientCoreData *gcc.ClientCoreData
|
||||
buff *bytes.Buffer
|
||||
}
|
||||
|
||||
func NewClient(t core.Transport) *Client {
|
||||
c := &Client{
|
||||
PDULayer: NewPDULayer(t),
|
||||
buff: &bytes.Buffer{},
|
||||
}
|
||||
c.transport.Once("connect", c.connect)
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *Client) connect(data *gcc.ClientCoreData, userId uint16, channelId uint16) {
|
||||
glog.Debug("pdu connect:", userId, ",", channelId)
|
||||
c.clientCoreData = data
|
||||
c.userId = userId
|
||||
c.channelId = channelId
|
||||
c.transport.Once("data", c.recvDemandActivePDU)
|
||||
}
|
||||
|
||||
func (c *Client) recvDemandActivePDU(s []byte) {
|
||||
glog.Trace("PDU recvDemandActivePDU", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
pdu, err := readPDU(r)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
return
|
||||
}
|
||||
if pdu.ShareCtrlHeader.PDUType != PDUTYPE_DEMANDACTIVEPDU {
|
||||
glog.Info("PDU ignore message during connection sequence, type is", pdu.ShareCtrlHeader.PDUType)
|
||||
c.transport.Once("data", c.recvDemandActivePDU)
|
||||
return
|
||||
}
|
||||
|
||||
c.sharedId = pdu.Message.(*DemandActivePDU).SharedId
|
||||
c.demandActivePDU = pdu.Message.(*DemandActivePDU)
|
||||
for _, caps := range c.demandActivePDU.CapabilitySets {
|
||||
glog.Debugf("serverCapabilities<%s>: %+v", caps.Type(), caps)
|
||||
c.serverCapabilities[caps.Type()] = caps
|
||||
}
|
||||
|
||||
c.sendConfirmActivePDU()
|
||||
c.sendClientFinalizeSynchronizePDU()
|
||||
c.transport.Once("data", c.recvServerSynchronizePDU)
|
||||
}
|
||||
|
||||
func (c *Client) sendConfirmActivePDU() {
|
||||
glog.Debug("PDU start sendConfirmActivePDU")
|
||||
|
||||
pdu := NewConfirmActivePDU()
|
||||
generalCapa := c.clientCapabilities[CAPSTYPE_GENERAL].(*GeneralCapability)
|
||||
generalCapa.OSMajorType = OSMAJORTYPE_WINDOWS
|
||||
generalCapa.OSMinorType = OSMINORTYPE_WINDOWS_NT
|
||||
generalCapa.ExtraFlags = LONG_CREDENTIALS_SUPPORTED | NO_BITMAP_COMPRESSION_HDR |
|
||||
FASTPATH_OUTPUT_SUPPORTED | AUTORECONNECT_SUPPORTED
|
||||
generalCapa.RefreshRectSupport = 0
|
||||
generalCapa.SuppressOutputSupport = 0
|
||||
|
||||
bitmapCapa := c.clientCapabilities[CAPSTYPE_BITMAP].(*BitmapCapability)
|
||||
bitmapCapa.PreferredBitsPerPixel = c.clientCoreData.HighColorDepth
|
||||
bitmapCapa.DesktopWidth = c.clientCoreData.DesktopWidth
|
||||
bitmapCapa.DesktopHeight = c.clientCoreData.DesktopHeight
|
||||
bitmapCapa.DesktopResizeFlag = 0x0001
|
||||
|
||||
orderCapa := c.clientCapabilities[CAPSTYPE_ORDER].(*OrderCapability)
|
||||
orderCapa.OrderFlags = NEGOTIATEORDERSUPPORT | ZEROBOUNDSDELTASSUPPORT | COLORINDEXSUPPORT | ORDERFLAGS_EXTRA_FLAGS
|
||||
orderCapa.OrderSupportExFlags |= ORDERFLAGS_EX_ALTSEC_FRAME_MARKER_SUPPORT
|
||||
orderCapa.OrderSupport[TS_NEG_DSTBLT_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_PATBLT_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_SCRBLT_INDEX] = 1
|
||||
//orderCapa.OrderSupport[TS_NEG_LINETO_INDEX] = 1
|
||||
//orderCapa.OrderSupport[TS_NEG_MEMBLT_INDEX] = 1
|
||||
//orderCapa.OrderSupport[TS_NEG_MEM3BLT_INDEX] = 1
|
||||
//orderCapa.OrderSupport[TS_NEG_POLYLINE_INDEX] = 1
|
||||
/*orderCapa.OrderSupport[TS_NEG_MULTIOPAQUERECT_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_GLYPH_INDEX_INDEX] = 1
|
||||
//orderCapa.OrderSupport[TS_NEG_DRAWNINEGRID_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_SAVEBITMAP_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_POLYGON_SC_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_POLYGON_CB_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_ELLIPSE_SC_INDEX] = 1
|
||||
orderCapa.OrderSupport[TS_NEG_ELLIPSE_CB_INDEX] = 1*/
|
||||
orderCapa.OrderSupport[TS_NEG_FAST_GLYPH_INDEX] = 1
|
||||
|
||||
inputCapa := c.clientCapabilities[CAPSTYPE_INPUT].(*InputCapability)
|
||||
inputCapa.Flags = INPUT_FLAG_SCANCODES | INPUT_FLAG_MOUSEX | INPUT_FLAG_UNICODE
|
||||
inputCapa.KeyboardLayout = c.clientCoreData.KbdLayout
|
||||
inputCapa.KeyboardType = c.clientCoreData.KeyboardType
|
||||
inputCapa.KeyboardSubType = c.clientCoreData.KeyboardSubType
|
||||
inputCapa.KeyboardFunctionKey = c.clientCoreData.KeyboardFnKeys
|
||||
inputCapa.ImeFileName = c.clientCoreData.ImeFileName
|
||||
|
||||
glyphCapa := c.clientCapabilities[CAPSTYPE_GLYPHCACHE].(*GlyphCapability)
|
||||
/*glyphCapa.GlyphCache[0] = cacheEntry{254, 4}
|
||||
glyphCapa.GlyphCache[1] = cacheEntry{254, 4}
|
||||
glyphCapa.GlyphCache[2] = cacheEntry{254, 8}
|
||||
glyphCapa.GlyphCache[3] = cacheEntry{254, 8}
|
||||
glyphCapa.GlyphCache[4] = cacheEntry{254, 16}
|
||||
glyphCapa.GlyphCache[5] = cacheEntry{254, 32}
|
||||
glyphCapa.GlyphCache[6] = cacheEntry{254, 64}
|
||||
glyphCapa.GlyphCache[7] = cacheEntry{254, 128}
|
||||
glyphCapa.GlyphCache[8] = cacheEntry{254, 256}
|
||||
glyphCapa.GlyphCache[9] = cacheEntry{64, 2048}
|
||||
glyphCapa.FragCache = 0x01000100*/
|
||||
glyphCapa.SupportLevel = GLYPH_SUPPORT_NONE
|
||||
|
||||
pdu.SharedId = c.sharedId
|
||||
for _, v := range c.clientCapabilities {
|
||||
glog.Debugf("clientCapabilities<%s>: %+v", v.Type(), v)
|
||||
pdu.CapabilitySets = append(pdu.CapabilitySets, v)
|
||||
}
|
||||
pdu.NumberCapabilities = uint16(len(pdu.CapabilitySets))
|
||||
pdu.LengthSourceDescriptor = c.demandActivePDU.LengthSourceDescriptor
|
||||
pdu.SourceDescriptor = c.demandActivePDU.SourceDescriptor
|
||||
pdu.LengthCombinedCapabilities = c.demandActivePDU.LengthCombinedCapabilities
|
||||
|
||||
c.sendPDU(pdu)
|
||||
}
|
||||
|
||||
func (c *Client) sendClientFinalizeSynchronizePDU() {
|
||||
glog.Debug("PDU start sendClientFinalizeSynchronizePDU")
|
||||
c.sendDataPDU(NewSynchronizeDataPDU(c.channelId))
|
||||
c.sendDataPDU(&ControlDataPDU{Action: CTRLACTION_COOPERATE})
|
||||
c.sendDataPDU(&ControlDataPDU{Action: CTRLACTION_REQUEST_CONTROL})
|
||||
//c.sendDataPDU(&PersistKeyPDU{BBitMask: 0x03})
|
||||
c.sendDataPDU(&FontListDataPDU{ListFlags: 0x0003, EntrySize: 0x0032})
|
||||
}
|
||||
|
||||
func (c *Client) recvServerSynchronizePDU(s []byte) {
|
||||
glog.Debug("PDU recvServerSynchronizePDU")
|
||||
r := bytes.NewReader(s)
|
||||
pdu, err := readPDU(r)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
return
|
||||
}
|
||||
dataPdu, ok := pdu.Message.(*DataPDU)
|
||||
if !ok || dataPdu.Header.PDUType2 != PDUTYPE2_SYNCHRONIZE {
|
||||
if ok {
|
||||
glog.Error("recvServerSynchronizePDU ignore datapdu type2", dataPdu.Header.PDUType2)
|
||||
} else {
|
||||
glog.Error("recvServerSynchronizePDU ignore message type", pdu.ShareCtrlHeader.PDUType)
|
||||
}
|
||||
glog.Infof("%+v", dataPdu)
|
||||
c.transport.Once("data", c.recvServerSynchronizePDU)
|
||||
return
|
||||
}
|
||||
c.transport.Once("data", c.recvServerControlCooperatePDU)
|
||||
}
|
||||
|
||||
func (c *Client) recvServerControlCooperatePDU(s []byte) {
|
||||
glog.Debug("PDU recvServerControlCooperatePDU")
|
||||
r := bytes.NewReader(s)
|
||||
pdu, err := readPDU(r)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
dataPdu, ok := pdu.Message.(*DataPDU)
|
||||
|
||||
if !ok || dataPdu.Header.PDUType2 != PDUTYPE2_CONTROL {
|
||||
if ok {
|
||||
glog.Error("recvServerControlCooperatePDU ignore datapdu type2", dataPdu.Header.PDUType2)
|
||||
} else {
|
||||
glog.Error("recvServerControlCooperatePDU ignore message type", pdu.ShareCtrlHeader.PDUType)
|
||||
}
|
||||
c.transport.Once("data", c.recvServerControlCooperatePDU)
|
||||
return
|
||||
}
|
||||
glog.Debugf("-------------PDUType2 = %02x\n", dataPdu.Header.PDUType2)
|
||||
if dataPdu.Header.PDUType2 == PDUTYPE2_SAVE_SESSION_INFO {
|
||||
c.Emit("success")
|
||||
}
|
||||
|
||||
if dataPdu.Data.(*ControlDataPDU).Action != CTRLACTION_COOPERATE {
|
||||
glog.Error("recvServerControlCooperatePDU ignore action", dataPdu.Data.(*ControlDataPDU).Action)
|
||||
c.transport.Once("data", c.recvServerControlCooperatePDU)
|
||||
return
|
||||
}
|
||||
c.transport.Once("data", c.recvServerControlGrantedPDU)
|
||||
}
|
||||
|
||||
func (c *Client) recvServerControlGrantedPDU(s []byte) {
|
||||
glog.Debug("PDU recvServerControlGrantedPDU")
|
||||
r := bytes.NewReader(s)
|
||||
pdu, err := readPDU(r)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
return
|
||||
}
|
||||
dataPdu, ok := pdu.Message.(*DataPDU)
|
||||
if !ok || dataPdu.Header.PDUType2 != PDUTYPE2_CONTROL {
|
||||
if ok {
|
||||
glog.Error("recvServerControlGrantedPDU ignore datapdu type2", dataPdu.Header.PDUType2)
|
||||
} else {
|
||||
glog.Error("recvServerControlGrantedPDU ignore message type", pdu.ShareCtrlHeader.PDUType)
|
||||
}
|
||||
c.transport.Once("data", c.recvServerControlGrantedPDU)
|
||||
return
|
||||
}
|
||||
if dataPdu.Data.(*ControlDataPDU).Action != CTRLACTION_GRANTED_CONTROL {
|
||||
glog.Error("recvServerControlGrantedPDU ignore action", dataPdu.Data.(*ControlDataPDU).Action)
|
||||
c.transport.Once("data", c.recvServerControlGrantedPDU)
|
||||
return
|
||||
}
|
||||
c.transport.Once("data", c.recvServerFontMapPDU)
|
||||
}
|
||||
|
||||
func (c *Client) recvServerFontMapPDU(s []byte) {
|
||||
glog.Debug("PDU recvServerFontMapPDU")
|
||||
r := bytes.NewReader(s)
|
||||
pdu, err := readPDU(r)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
return
|
||||
}
|
||||
dataPdu, ok := pdu.Message.(*DataPDU)
|
||||
if !ok || dataPdu.Header.PDUType2 != PDUTYPE2_FONTMAP {
|
||||
if ok {
|
||||
glog.Error("recvServerFontMapPDU ignore datapdu type2", dataPdu.Header.PDUType2)
|
||||
} else {
|
||||
glog.Error("recvServerFontMapPDU ignore message type", pdu.ShareCtrlHeader.PDUType)
|
||||
}
|
||||
return
|
||||
}
|
||||
c.transport.On("data", c.recvPDU)
|
||||
c.Emit("ready")
|
||||
}
|
||||
|
||||
func (c *Client) recvPDU(s []byte) {
|
||||
glog.Trace("PDU recvPDU", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
if r.Len() > 0 {
|
||||
p, err := readPDU(r)
|
||||
if err != nil {
|
||||
glog.Error(err)
|
||||
return
|
||||
}
|
||||
if p.ShareCtrlHeader.PDUType == PDUTYPE_DEACTIVATEALLPDU {
|
||||
c.transport.Once("data", c.recvDemandActivePDU)
|
||||
} else if p.ShareCtrlHeader.PDUType == PDUTYPE_DATAPDU {
|
||||
d := p.Message.(*DataPDU)
|
||||
if d.Header.PDUType2 == PDUTYPE2_UPDATE {
|
||||
up := d.Data.(*UpdateDataPDU)
|
||||
p := up.Udata
|
||||
if up.UpdateType == FASTPATH_UPDATETYPE_BITMAP {
|
||||
c.Emit("bitmap", p.(*BitmapUpdateDataPDU).Rectangles)
|
||||
}
|
||||
}
|
||||
if d.Header.PDUType2 == PDUTYPE2_SAVE_SESSION_INFO {
|
||||
c.Emit("success")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) RecvFastPath(secFlag byte, s []byte) {
|
||||
glog.Trace("PDU RecvFastPath", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
for r.Len() > 0 {
|
||||
updateHeader, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
updateCode := updateHeader & 0x0f
|
||||
fragmentation := updateHeader & 0x30
|
||||
compression := updateHeader & 0xC0
|
||||
|
||||
var compressionFlags uint8 = 0
|
||||
if compression == FASTPATH_OUTPUT_COMPRESSION_USED {
|
||||
compressionFlags, err = core.ReadUInt8(r)
|
||||
}
|
||||
|
||||
size, err := core.ReadUint16LE(r)
|
||||
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
glog.Trace("Code:", FastPathUpdateType(updateCode),
|
||||
"compressionFlags:", compressionFlags,
|
||||
"fragmentation:", fragmentation,
|
||||
"size:", size, "len:", r.Len())
|
||||
if compressionFlags&RDP_MPPC_COMPRESSED != 0 {
|
||||
glog.Info("RDP_MPPC_COMPRESSED")
|
||||
}
|
||||
if fragmentation != FASTPATH_FRAGMENT_SINGLE {
|
||||
if fragmentation == FASTPATH_FRAGMENT_FIRST {
|
||||
c.buff.Reset()
|
||||
}
|
||||
b, _ := core.ReadBytes(r.Len(), r)
|
||||
c.buff.Write(b)
|
||||
if fragmentation != FASTPATH_FRAGMENT_LAST {
|
||||
return
|
||||
}
|
||||
r = bytes.NewReader(c.buff.Bytes())
|
||||
}
|
||||
|
||||
p, err := readFastPathUpdatePDU(r, updateCode)
|
||||
if err != nil || p == nil || p.Data == nil {
|
||||
glog.Debug("readFastPathUpdatePDU:", err)
|
||||
return
|
||||
}
|
||||
|
||||
if updateCode == FASTPATH_UPDATETYPE_BITMAP {
|
||||
c.Emit("bitmap", p.Data.(*FastPathBitmapUpdateDataPDU).Rectangles)
|
||||
} else if updateCode == FASTPATH_UPDATETYPE_COLOR {
|
||||
c.Emit("color", p.Data.(*FastPathColorPdu))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type InputEventsInterface interface {
|
||||
Serialize() []byte
|
||||
}
|
||||
|
||||
func (c *Client) SendInputEvents(msgType uint16, events []InputEventsInterface) {
|
||||
p := &ClientInputEventPDU{}
|
||||
p.NumEvents = uint16(len(events))
|
||||
p.SlowPathInputEvents = make([]SlowPathInputEvent, 0, p.NumEvents)
|
||||
for _, in := range events {
|
||||
seria := in.Serialize()
|
||||
s := SlowPathInputEvent{0, msgType, len(seria), seria}
|
||||
p.SlowPathInputEvents = append(p.SlowPathInputEvents, s)
|
||||
}
|
||||
|
||||
c.sendDataPDU(p)
|
||||
}
|
||||
@@ -0,0 +1,913 @@
|
||||
package sec
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/md5"
|
||||
"crypto/rand"
|
||||
"crypto/rc4"
|
||||
"crypto/rsa"
|
||||
"crypto/sha1"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"io"
|
||||
"unicode/utf16"
|
||||
|
||||
"github.com/lunixbochs/struc"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/nla"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/emission"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/lic"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/gcc"
|
||||
)
|
||||
|
||||
/**
|
||||
* SecurityFlag
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240579.aspx
|
||||
*/
|
||||
const (
|
||||
EXCHANGE_PKT uint16 = 0x0001
|
||||
TRANSPORT_REQ = 0x0002
|
||||
TRANSPORT_RSP = 0x0004
|
||||
ENCRYPT = 0x0008
|
||||
RESET_SEQNO = 0x0010
|
||||
IGNORE_SEQNO = 0x0020
|
||||
INFO_PKT = 0x0040
|
||||
LICENSE_PKT = 0x0080
|
||||
LICENSE_ENCRYPT_CS = 0x0200
|
||||
LICENSE_ENCRYPT_SC = 0x0200
|
||||
REDIRECTION_PKT = 0x0400
|
||||
SECURE_CHECKSUM = 0x0800
|
||||
AUTODETECT_REQ = 0x1000
|
||||
AUTODETECT_RSP = 0x2000
|
||||
HEARTBEAT = 0x4000
|
||||
FLAGSHI_VALID = 0x8000
|
||||
)
|
||||
|
||||
const (
|
||||
INFO_MOUSE uint32 = 0x00000001
|
||||
INFO_DISABLECTRLALTDEL = 0x00000002
|
||||
INFO_AUTOLOGON = 0x00000008
|
||||
INFO_UNICODE = 0x00000010
|
||||
INFO_MAXIMIZESHELL = 0x00000020
|
||||
INFO_LOGONNOTIFY = 0x00000040
|
||||
INFO_COMPRESSION = 0x00000080
|
||||
INFO_ENABLEWINDOWSKEY = 0x00000100
|
||||
INFO_REMOTECONSOLEAUDIO = 0x00002000
|
||||
INFO_FORCE_ENCRYPTED_CS_PDU = 0x00004000
|
||||
INFO_RAIL = 0x00008000
|
||||
INFO_LOGONERRORS = 0x00010000
|
||||
INFO_MOUSE_HAS_WHEEL = 0x00020000
|
||||
INFO_PASSWORD_IS_SC_PIN = 0x00040000
|
||||
INFO_NOAUDIOPLAYBACK = 0x00080000
|
||||
INFO_USING_SAVED_CREDS = 0x00100000
|
||||
INFO_AUDIOCAPTURE = 0x00200000
|
||||
INFO_VIDEO_DISABLE = 0x00400000
|
||||
INFO_CompressionTypeMask = 0x00001E00
|
||||
)
|
||||
|
||||
const (
|
||||
AF_INET uint16 = 0x00002
|
||||
AF_INET6 = 0x0017
|
||||
)
|
||||
|
||||
const (
|
||||
PERF_DISABLE_WALLPAPER uint32 = 0x00000001
|
||||
PERF_DISABLE_FULLWINDOWDRAG = 0x00000002
|
||||
PERF_DISABLE_MENUANIMATIONS = 0x00000004
|
||||
PERF_DISABLE_THEMING = 0x00000008
|
||||
PERF_DISABLE_CURSOR_SHADOW = 0x00000020
|
||||
PERF_DISABLE_CURSORSETTINGS = 0x00000040
|
||||
PERF_ENABLE_FONT_SMOOTHING = 0x00000080
|
||||
PERF_ENABLE_DESKTOP_COMPOSITION = 0x00000100
|
||||
)
|
||||
|
||||
const (
|
||||
FASTPATH_OUTPUT_SECURE_CHECKSUM = 0x1
|
||||
FASTPATH_OUTPUT_ENCRYPTED = 0x2
|
||||
)
|
||||
|
||||
type ClientAutoReconnect struct {
|
||||
CbAutoReconnectLen uint16
|
||||
CbLen uint32
|
||||
Version uint32
|
||||
LogonId uint32
|
||||
SecVerifier []byte
|
||||
}
|
||||
|
||||
func NewClientAutoReconnect(id uint32, random []byte) *ClientAutoReconnect {
|
||||
return &ClientAutoReconnect{
|
||||
CbAutoReconnectLen: 28,
|
||||
CbLen: 28,
|
||||
Version: 1,
|
||||
LogonId: id,
|
||||
SecVerifier: nla.HMAC_MD5(random, random),
|
||||
}
|
||||
}
|
||||
|
||||
type RDPExtendedInfo struct {
|
||||
ClientAddressFamily uint16 `struc:"little"`
|
||||
CbClientAddress uint16 `struc:"little,sizeof=ClientAddress"`
|
||||
ClientAddress []byte `struc:"[]byte"`
|
||||
CbClientDir uint16 `struc:"little,sizeof=ClientDir"`
|
||||
ClientDir []byte `struc:"[]byte"`
|
||||
ClientTimeZone []byte `struc:"[172]byte"`
|
||||
ClientSessionId uint32 `struc:"litttle"`
|
||||
PerformanceFlags uint32 `struc:"little"`
|
||||
AutoReconnect *ClientAutoReconnect
|
||||
}
|
||||
|
||||
func NewExtendedInfo(auto *ClientAutoReconnect) *RDPExtendedInfo {
|
||||
return &RDPExtendedInfo{
|
||||
ClientAddressFamily: AF_INET,
|
||||
ClientAddress: []byte{0, 0},
|
||||
ClientDir: []byte{0, 0},
|
||||
ClientTimeZone: make([]byte, 172),
|
||||
ClientSessionId: 0,
|
||||
AutoReconnect: auto,
|
||||
}
|
||||
}
|
||||
|
||||
func (o *RDPExtendedInfo) Serialize() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt16LE(o.ClientAddressFamily, buff)
|
||||
core.WriteUInt16LE(uint16(len(o.ClientAddress)), buff)
|
||||
core.WriteBytes(o.ClientAddress, buff)
|
||||
core.WriteUInt16LE(uint16(len(o.ClientDir)), buff)
|
||||
core.WriteBytes(o.ClientDir, buff)
|
||||
core.WriteBytes(o.ClientTimeZone, buff)
|
||||
core.WriteUInt32LE(o.ClientSessionId, buff)
|
||||
core.WriteUInt32LE(o.PerformanceFlags, buff)
|
||||
|
||||
if o.AutoReconnect != nil {
|
||||
core.WriteUInt16LE(o.AutoReconnect.CbAutoReconnectLen, buff)
|
||||
core.WriteUInt32LE(o.AutoReconnect.CbLen, buff)
|
||||
core.WriteUInt32LE(o.AutoReconnect.Version, buff)
|
||||
core.WriteUInt32LE(o.AutoReconnect.LogonId, buff)
|
||||
core.WriteBytes(o.AutoReconnect.SecVerifier, buff)
|
||||
}
|
||||
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type RDPInfo struct {
|
||||
CodePage uint32
|
||||
Flag uint32
|
||||
CbDomain uint16
|
||||
CbUserName uint16
|
||||
CbPassword uint16
|
||||
CbAlternateShell uint16
|
||||
CbWorkingDir uint16
|
||||
Domain []byte
|
||||
UserName []byte
|
||||
Password []byte
|
||||
AlternateShell []byte
|
||||
WorkingDir []byte
|
||||
ExtendedInfo *RDPExtendedInfo
|
||||
}
|
||||
|
||||
func NewRDPInfo() *RDPInfo {
|
||||
info := &RDPInfo{
|
||||
Flag: INFO_MOUSE | INFO_UNICODE | INFO_MAXIMIZESHELL |
|
||||
INFO_ENABLEWINDOWSKEY | INFO_DISABLECTRLALTDEL | INFO_MOUSE_HAS_WHEEL |
|
||||
INFO_FORCE_ENCRYPTED_CS_PDU | INFO_AUTOLOGON,
|
||||
Domain: []byte{0, 0},
|
||||
UserName: []byte{0, 0},
|
||||
Password: []byte{0, 0},
|
||||
AlternateShell: []byte{0, 0},
|
||||
WorkingDir: []byte{0, 0},
|
||||
ExtendedInfo: NewExtendedInfo(nil),
|
||||
}
|
||||
return info
|
||||
}
|
||||
|
||||
func (o *RDPInfo) SetClientAutoReconnect(auto *ClientAutoReconnect) {
|
||||
o.ExtendedInfo.AutoReconnect = auto
|
||||
}
|
||||
|
||||
func (o *RDPInfo) SetClientInfo() {
|
||||
o.Flag |= INFO_LOGONNOTIFY | INFO_LOGONERRORS
|
||||
}
|
||||
|
||||
func (o *RDPInfo) Serialize(hasExtended bool) []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt32LE(o.CodePage, buff) // 0000000
|
||||
core.WriteUInt32LE(o.Flag, buff) // 0530101
|
||||
core.WriteUInt16LE(uint16(len(o.Domain)-2), buff) // 001c
|
||||
core.WriteUInt16LE(uint16(len(o.UserName)-2), buff) // 0008
|
||||
core.WriteUInt16LE(uint16(len(o.Password)-2), buff) //000c
|
||||
core.WriteUInt16LE(uint16(len(o.AlternateShell)-2), buff) //0000
|
||||
core.WriteUInt16LE(uint16(len(o.WorkingDir)-2), buff) //0000
|
||||
core.WriteBytes(o.Domain, buff)
|
||||
core.WriteBytes(o.UserName, buff)
|
||||
core.WriteBytes(o.Password, buff)
|
||||
core.WriteBytes(o.AlternateShell, buff)
|
||||
core.WriteBytes(o.WorkingDir, buff)
|
||||
if hasExtended {
|
||||
core.WriteBytes(o.ExtendedInfo.Serialize(), buff)
|
||||
}
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type SecurityHeader struct {
|
||||
securityFlag uint16
|
||||
securityFlagHi uint16
|
||||
}
|
||||
|
||||
func readSecurityHeader(r io.Reader) *SecurityHeader {
|
||||
s := &SecurityHeader{}
|
||||
s.securityFlag, _ = core.ReadUint16LE(r)
|
||||
s.securityFlagHi, _ = core.ReadUint16LE(r)
|
||||
return s
|
||||
}
|
||||
|
||||
type SEC struct {
|
||||
emission.Emitter
|
||||
transport core.Transport
|
||||
info *RDPInfo
|
||||
machineName string
|
||||
clientData []interface{}
|
||||
serverData []interface{}
|
||||
|
||||
enableEncryption bool
|
||||
//Enable Secure Mac generation
|
||||
enableSecureCheckSum bool
|
||||
//counter before update
|
||||
nbEncryptedPacket int
|
||||
nbDecryptedPacket int
|
||||
|
||||
currentDecrytKey []byte
|
||||
currentEncryptKey []byte
|
||||
|
||||
//current rc4 tab
|
||||
decryptRc4 *rc4.Cipher
|
||||
encryptRc4 *rc4.Cipher
|
||||
|
||||
macKey []byte
|
||||
macSalt []byte
|
||||
}
|
||||
|
||||
func NewSEC(t core.Transport) *SEC {
|
||||
sec := &SEC{
|
||||
*emission.NewEmitter(),
|
||||
t,
|
||||
NewRDPInfo(),
|
||||
"",
|
||||
nil,
|
||||
nil,
|
||||
false,
|
||||
false,
|
||||
0,
|
||||
0,
|
||||
nil,
|
||||
nil,
|
||||
nil,
|
||||
nil,
|
||||
nil,
|
||||
nil,
|
||||
}
|
||||
|
||||
t.On("close", func() {
|
||||
sec.Emit("close")
|
||||
}).On("error", func(err error) {
|
||||
sec.Emit("error", err)
|
||||
})
|
||||
return sec
|
||||
}
|
||||
|
||||
func (s *SEC) Read(data []byte) (n int, err error) {
|
||||
return s.transport.Read(data)
|
||||
}
|
||||
|
||||
func (s *SEC) Write(b []byte) (n int, err error) {
|
||||
if !s.enableEncryption {
|
||||
return s.transport.Write(b)
|
||||
}
|
||||
data := s.encrytData(b)
|
||||
return s.transport.Write(data)
|
||||
}
|
||||
|
||||
func (s *SEC) Close() error {
|
||||
return s.transport.Close()
|
||||
}
|
||||
|
||||
func (s *SEC) sendFlagged(flag uint16, data []byte) (n int, err error) {
|
||||
glog.Trace("sendFlagged:", hex.EncodeToString(data))
|
||||
b := s.encryt(flag, data)
|
||||
return s.transport.Write(b)
|
||||
}
|
||||
|
||||
/*
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241995.aspx
|
||||
@param macSaltKey: {str} mac key
|
||||
@param data: {str} data to sign
|
||||
@return: {str} signature
|
||||
*/
|
||||
func macData(macSaltKey, data []byte) []byte {
|
||||
sha1Digest := sha1.New()
|
||||
md5Digest := md5.New()
|
||||
|
||||
b := &bytes.Buffer{}
|
||||
core.WriteUInt32LE(uint32(len(data)), b)
|
||||
|
||||
sha1Digest.Write(macSaltKey)
|
||||
for i := 0; i < 40; i++ {
|
||||
sha1Digest.Write([]byte("\x36"))
|
||||
}
|
||||
|
||||
sha1Digest.Write(b.Bytes())
|
||||
sha1Digest.Write(data)
|
||||
|
||||
sha1Sig := sha1Digest.Sum(nil)
|
||||
|
||||
md5Digest.Write(macSaltKey)
|
||||
for i := 0; i < 48; i++ {
|
||||
md5Digest.Write([]byte("\x5c"))
|
||||
}
|
||||
|
||||
md5Digest.Write(sha1Sig)
|
||||
|
||||
return md5Digest.Sum(nil)
|
||||
}
|
||||
func (s *SEC) readEncryptedPayload(data []byte, checkSum bool) []byte {
|
||||
r := bytes.NewReader(data)
|
||||
sign, _ := core.ReadBytes(8, r)
|
||||
glog.Debug("read sign:", sign)
|
||||
encryptedPayload, _ := core.ReadBytes(r.Len(), r)
|
||||
if s.decryptRc4 == nil {
|
||||
s.decryptRc4, _ = rc4.NewCipher(s.currentDecrytKey)
|
||||
}
|
||||
s.nbDecryptedPacket++
|
||||
glog.Debug("nbDecryptedPacket:", s.nbDecryptedPacket)
|
||||
plaintext := make([]byte, len(encryptedPayload))
|
||||
s.decryptRc4.XORKeyStream(plaintext, encryptedPayload)
|
||||
|
||||
return plaintext
|
||||
|
||||
}
|
||||
func (s *SEC) writeEncryptedPayload(data []byte, checkSum bool) []byte {
|
||||
if s.nbEncryptedPacket == 4096 {
|
||||
|
||||
}
|
||||
|
||||
if checkSum {
|
||||
glog.Debug("need checkSum")
|
||||
return []byte{}
|
||||
}
|
||||
|
||||
s.nbEncryptedPacket++
|
||||
glog.Debug("nbEncryptedPacket:", s.nbEncryptedPacket)
|
||||
b := &bytes.Buffer{}
|
||||
|
||||
sign := macData(s.macKey, data)[:8]
|
||||
//sign := macData(s.macSalt, data)[:8]
|
||||
if s.encryptRc4 == nil {
|
||||
s.encryptRc4, _ = rc4.NewCipher(s.currentEncryptKey)
|
||||
}
|
||||
|
||||
plaintext := make([]byte, len(data))
|
||||
s.encryptRc4.XORKeyStream(plaintext, data)
|
||||
b.Write(sign)
|
||||
b.Write(plaintext)
|
||||
glog.Debug("sign:", hex.EncodeToString(sign), "plaintext:", hex.EncodeToString(plaintext))
|
||||
return b.Bytes()
|
||||
}
|
||||
|
||||
func (s *SEC) encryt(flag uint16, b []byte) []byte {
|
||||
data := b
|
||||
if flag&ENCRYPT != 0 {
|
||||
data = s.writeEncryptedPayload(b, flag&SECURE_CHECKSUM != 0)
|
||||
}
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt16LE(flag, buff)
|
||||
core.WriteUInt16LE(0, buff)
|
||||
core.WriteBytes(data, buff)
|
||||
|
||||
return buff.Bytes()
|
||||
}
|
||||
func (s *SEC) encrytData(b []byte) []byte {
|
||||
if !s.enableEncryption {
|
||||
return b
|
||||
}
|
||||
|
||||
var flag uint16 = ENCRYPT
|
||||
if s.enableSecureCheckSum {
|
||||
flag |= SECURE_CHECKSUM
|
||||
}
|
||||
return s.encryt(flag, b)
|
||||
}
|
||||
|
||||
func (s *SEC) decrytData(b []byte) []byte {
|
||||
if !s.enableEncryption {
|
||||
return b
|
||||
}
|
||||
|
||||
r := bytes.NewReader(b)
|
||||
securityFlag, _ := core.ReadUint16LE(r)
|
||||
_, _ = core.ReadUint16LE(r) //securityFlagHi
|
||||
data, _ := core.ReadBytes(r.Len(), r)
|
||||
if securityFlag&ENCRYPT != 0 {
|
||||
data = s.readEncryptedPayload(data, securityFlag&SECURE_CHECKSUM != 0)
|
||||
}
|
||||
return data
|
||||
}
|
||||
|
||||
type Client struct {
|
||||
*SEC
|
||||
userId uint16
|
||||
channelId uint16
|
||||
//initialise decrypt and encrypt keys
|
||||
initialDecrytKey []byte
|
||||
initialEncryptKey []byte
|
||||
|
||||
fastPathListener core.FastPathListener
|
||||
channelSender core.ChannelSender
|
||||
}
|
||||
|
||||
func NewClient(t core.Transport) *Client {
|
||||
c := &Client{
|
||||
SEC: NewSEC(t),
|
||||
}
|
||||
t.On("connect", c.connect)
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *Client) SetClientAutoReconnect(id uint32, random []byte) {
|
||||
auto := NewClientAutoReconnect(id, random)
|
||||
c.info.SetClientAutoReconnect(auto)
|
||||
}
|
||||
|
||||
func (c *Client) SetAlternateShell(shell string) {
|
||||
buff := &bytes.Buffer{}
|
||||
for _, ch := range utf16.Encode([]rune(shell)) {
|
||||
core.WriteUInt16LE(ch, buff)
|
||||
}
|
||||
core.WriteUInt16LE(0, buff)
|
||||
c.info.AlternateShell = buff.Bytes()
|
||||
c.info.Flag |= INFO_RAIL
|
||||
}
|
||||
|
||||
func (c *Client) SetUser(user string) {
|
||||
buff := &bytes.Buffer{}
|
||||
for _, ch := range utf16.Encode([]rune(user)) {
|
||||
core.WriteUInt16LE(ch, buff)
|
||||
}
|
||||
core.WriteUInt16LE(0, buff)
|
||||
c.info.UserName = buff.Bytes()
|
||||
}
|
||||
|
||||
func (c *Client) SetPwd(pwd string) {
|
||||
buff := &bytes.Buffer{}
|
||||
for _, ch := range utf16.Encode([]rune(pwd)) {
|
||||
core.WriteUInt16LE(ch, buff)
|
||||
}
|
||||
core.WriteUInt16LE(0, buff)
|
||||
c.info.Password = buff.Bytes()
|
||||
}
|
||||
|
||||
func (c *Client) SetDomain(domain string) {
|
||||
buff := &bytes.Buffer{}
|
||||
for _, ch := range utf16.Encode([]rune(domain)) {
|
||||
core.WriteUInt16LE(ch, buff)
|
||||
}
|
||||
core.WriteUInt16LE(0, buff)
|
||||
c.info.Domain = buff.Bytes()
|
||||
}
|
||||
|
||||
func (c *Client) connect(clientData []interface{}, serverData []interface{}, userId uint16, channels []t125.MCSChannelInfo) {
|
||||
glog.Debug("sec on connect:", clientData)
|
||||
glog.Debug("sec on connect:", serverData)
|
||||
glog.Debug("sec on connect:", userId)
|
||||
glog.Debug("sec on connect:", channels)
|
||||
c.clientData = clientData
|
||||
c.serverData = serverData
|
||||
c.userId = userId
|
||||
for _, channel := range channels {
|
||||
glog.Infof("channel: %s <%d>:", channel.Name, channel.ID)
|
||||
if channel.Name == t125.GLOBAL_CHANNEL_NAME {
|
||||
c.channelId = channel.ID
|
||||
//break
|
||||
}
|
||||
}
|
||||
c.enableEncryption = c.ClientCoreData().ServerSelectedProtocol == 0
|
||||
|
||||
if c.enableEncryption {
|
||||
c.sendClientRandom()
|
||||
}
|
||||
|
||||
c.sendInfoPkt()
|
||||
c.transport.Once("sec", c.recvLicenceInfo)
|
||||
}
|
||||
|
||||
func (c *Client) ClientCoreData() *gcc.ClientCoreData {
|
||||
return c.clientData[0].(*gcc.ClientCoreData)
|
||||
}
|
||||
func (c *Client) ClientSecurityData() *gcc.ClientSecurityData {
|
||||
return c.clientData[1].(*gcc.ClientSecurityData)
|
||||
}
|
||||
func (c *Client) ClientNetworkData() *gcc.ClientNetworkData {
|
||||
return c.clientData[2].(*gcc.ClientNetworkData)
|
||||
}
|
||||
|
||||
func (c *Client) ServerSecurityData() *gcc.ServerSecurityData {
|
||||
return c.serverData[1].(*gcc.ServerSecurityData)
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: generate 40 bits data from 128 bits data
|
||||
@param data: {str} 128 bits data
|
||||
@return: {str} 40 bits data
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc240785.aspx
|
||||
*/
|
||||
func gen40bits(data []byte) []byte {
|
||||
return append([]byte("\xd1\x26\x9e"), data[3:8]...)
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: generate 56 bits data from 128 bits data
|
||||
@param data: {str} 128 bits data
|
||||
@return: {str} 56 bits data
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc240785.aspx
|
||||
*/
|
||||
func gen56bits(data []byte) []byte {
|
||||
return append([]byte("\xd1"), data[1:8]...)
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: Generate particular signature from combination of sha1 and md5
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241992.aspx
|
||||
@param inputData: strange input (see doc)
|
||||
@param salt: salt for context call
|
||||
@param salt1: another salt (ex : client random)
|
||||
@param salt2: another another salt (ex: server random)
|
||||
@return : MD5(Salt + SHA1(Input + Salt + Salt1 + Salt2))
|
||||
*/
|
||||
func saltedHash(inputData, salt, salt1, salt2 []byte) []byte {
|
||||
sha1Digest := sha1.New()
|
||||
md5Digest := md5.New()
|
||||
|
||||
sha1Digest.Write(inputData)
|
||||
sha1Digest.Write(salt[:48])
|
||||
sha1Digest.Write(salt1)
|
||||
sha1Digest.Write(salt2)
|
||||
sha1Sig := sha1Digest.Sum(nil)
|
||||
|
||||
md5Digest.Write(salt[:48])
|
||||
md5Digest.Write(sha1Sig)
|
||||
|
||||
return md5Digest.Sum(nil)[:16]
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: MD5(in0[:16] + in1[:32] + in2[:32])
|
||||
@param key: in 16
|
||||
@param random1: in 32
|
||||
@param random2: in 32
|
||||
@return MD5(in0[:16] + in1[:32] + in2[:32])
|
||||
*/
|
||||
func finalHash(key, random1, random2 []byte) []byte {
|
||||
md5Digest := md5.New()
|
||||
md5Digest.Write(key)
|
||||
md5Digest.Write(random1)
|
||||
md5Digest.Write(random2)
|
||||
return md5Digest.Sum(nil)
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: Generate master secret
|
||||
@param secret: {str} secret
|
||||
@param clientRandom : {str} client random
|
||||
@param serverRandom : {str} server random
|
||||
@see: http://msdn.microsoft.com/en-us/library/cc241992.aspx
|
||||
*/
|
||||
func masterSecret(secret, random1, random2 []byte) []byte {
|
||||
sh1 := saltedHash([]byte("A"), secret, random1, random2)
|
||||
sh2 := saltedHash([]byte("BB"), secret, random1, random2)
|
||||
sh3 := saltedHash([]byte("CCC"), secret, random1, random2)
|
||||
ms := bytes.NewBuffer(nil)
|
||||
ms.Write(sh1)
|
||||
ms.Write(sh2)
|
||||
ms.Write(sh3)
|
||||
return ms.Bytes()
|
||||
}
|
||||
|
||||
/*
|
||||
@summary: Generate master secret
|
||||
@param secret: secret
|
||||
@param clientRandom : client random
|
||||
@param serverRandom : server random
|
||||
*/
|
||||
func sessionKeyBlob(secret, random1, random2 []byte) []byte {
|
||||
sh1 := saltedHash([]byte("X"), secret, random1, random2)
|
||||
sh2 := saltedHash([]byte("YY"), secret, random1, random2)
|
||||
sh3 := saltedHash([]byte("ZZZ"), secret, random1, random2)
|
||||
ms := bytes.NewBuffer(nil)
|
||||
ms.Write(sh1)
|
||||
ms.Write(sh2)
|
||||
ms.Write(sh3)
|
||||
return ms.Bytes()
|
||||
|
||||
}
|
||||
func generateKeys(clientRandom, serverRandom []byte, method uint32) ([]byte, []byte, []byte) {
|
||||
b := &bytes.Buffer{}
|
||||
b.Write(clientRandom[:24])
|
||||
b.Write(serverRandom[:24])
|
||||
preMasterHash := b.Bytes()
|
||||
glog.Debug("preMasterHash:", hex.EncodeToString(preMasterHash))
|
||||
|
||||
masterHash := masterSecret(preMasterHash, clientRandom, serverRandom)
|
||||
glog.Debug("masterHash:", hex.EncodeToString(masterHash))
|
||||
|
||||
sessionKey := sessionKeyBlob(masterHash, clientRandom, serverRandom)
|
||||
glog.Debug("sessionKey:", hex.EncodeToString(sessionKey))
|
||||
|
||||
macKey128 := sessionKey[:16]
|
||||
initialFirstKey128 := finalHash(sessionKey[16:32], clientRandom, serverRandom)
|
||||
initialSecondKey128 := finalHash(sessionKey[32:48], clientRandom, serverRandom)
|
||||
|
||||
glog.Debug("macKey128:", hex.EncodeToString(macKey128))
|
||||
glog.Debug("FirstKey128:", hex.EncodeToString(initialFirstKey128))
|
||||
glog.Debug("SecondKey128:", hex.EncodeToString(initialSecondKey128))
|
||||
//generate valid key
|
||||
if method == gcc.ENCRYPTION_FLAG_40BIT {
|
||||
return gen40bits(macKey128), gen40bits(initialFirstKey128), gen40bits(initialSecondKey128)
|
||||
} else if method == gcc.ENCRYPTION_FLAG_56BIT {
|
||||
return gen56bits(macKey128), gen56bits(initialFirstKey128), gen56bits(initialSecondKey128)
|
||||
}
|
||||
// method == gcc.ENCRYPTION_FLAG_128BIT
|
||||
return macKey128, initialFirstKey128, initialSecondKey128
|
||||
|
||||
}
|
||||
|
||||
type ClientSecurityExchangePDU struct {
|
||||
Length uint32 `struc:"little"`
|
||||
EncryptedClientRandom []byte `struc:"little"`
|
||||
Padding []byte `struc:"[8]byte"`
|
||||
}
|
||||
|
||||
func (e *ClientSecurityExchangePDU) serialize() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt32LE(e.Length, buff)
|
||||
core.WriteBytes(e.EncryptedClientRandom, buff)
|
||||
core.WriteBytes(e.Padding, buff)
|
||||
|
||||
return buff.Bytes()
|
||||
}
|
||||
func (c *Client) sendClientRandom() {
|
||||
glog.Debug("send Client Random")
|
||||
|
||||
clientRandom := core.Random(32)
|
||||
glog.Debug("clientRandom:", hex.EncodeToString(clientRandom))
|
||||
|
||||
serverRandom := c.ServerSecurityData().ServerRandom
|
||||
glog.Debug("ServerRandom:", hex.EncodeToString(serverRandom))
|
||||
|
||||
c.macKey, c.initialDecrytKey, c.initialEncryptKey = generateKeys(clientRandom,
|
||||
serverRandom, c.ServerSecurityData().EncryptionMethod)
|
||||
|
||||
//initialize keys
|
||||
c.currentDecrytKey = c.initialDecrytKey
|
||||
c.currentEncryptKey = c.initialEncryptKey
|
||||
|
||||
//verify certificate
|
||||
if !c.ServerSecurityData().ServerCertificate.CertData.Verify() {
|
||||
glog.Warn("Cannot verify server identity")
|
||||
}
|
||||
|
||||
serverPubKey, err := c.ServerSecurityData().ServerCertificate.CertData.GetPublicKey()
|
||||
if err != nil || serverPubKey == nil {
|
||||
glog.Error("GetPublicKey failed:", err)
|
||||
c.Emit("error", errors.New("failed to get server public key"))
|
||||
return
|
||||
}
|
||||
ret, err := rsa.EncryptPKCS1v15(rand.Reader, serverPubKey, core.Reverse(clientRandom))
|
||||
if err != nil {
|
||||
glog.Error("EncryptPKCS1v15 err:", err)
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
message := ClientSecurityExchangePDU{}
|
||||
message.EncryptedClientRandom = core.Reverse(ret)
|
||||
message.Length = uint32(len(message.EncryptedClientRandom) + 8)
|
||||
message.Padding = make([]byte, 8)
|
||||
|
||||
glog.Debug("message:", message)
|
||||
|
||||
c.sendFlagged(EXCHANGE_PKT, message.serialize())
|
||||
}
|
||||
func (c *Client) sendInfoPkt() {
|
||||
var secFlag uint16 = INFO_PKT
|
||||
if c.enableEncryption {
|
||||
secFlag |= ENCRYPT
|
||||
}
|
||||
|
||||
glog.Debug("RdpVersion:", c.ClientCoreData().RdpVersion, ":", gcc.RDP_VERSION_5_PLUS)
|
||||
c.sendFlagged(secFlag, c.info.Serialize(c.ClientCoreData().RdpVersion == gcc.RDP_VERSION_5_PLUS))
|
||||
}
|
||||
|
||||
func (c *Client) recvLicenceInfo(channel string, s []byte) {
|
||||
glog.Debug("sec recvLicenceInfo", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
h := readSecurityHeader(r)
|
||||
if (h.securityFlag & LICENSE_PKT) == 0 {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_PDU_SEC_BAD_LICENSE_HEADER"))
|
||||
return
|
||||
}
|
||||
|
||||
p := lic.ReadLicensePacket(r)
|
||||
switch p.BMsgtype {
|
||||
case lic.NEW_LICENSE:
|
||||
glog.Info("sec NEW_LICENSE")
|
||||
c.Emit("success")
|
||||
goto connect
|
||||
case lic.ERROR_ALERT:
|
||||
message := p.LicensingMessage.(*lic.ErrorMessage)
|
||||
glog.Info("sec ERROR_ALERT and ErrorCode:", message.DwErrorCode)
|
||||
if message.DwErrorCode == lic.STATUS_VALID_CLIENT && message.DwStateTransaction == lic.ST_NO_TRANSITION {
|
||||
goto connect
|
||||
}
|
||||
goto retry
|
||||
case lic.LICENSE_REQUEST:
|
||||
glog.Info("sec LICENSE_REQUEST")
|
||||
c.sendClientNewLicenseRequest(p.LicensingMessage.([]byte))
|
||||
goto retry
|
||||
case lic.PLATFORM_CHALLENGE:
|
||||
glog.Info("sec PLATFORM_CHALLENGE")
|
||||
c.sendClientChallengeResponse(p.LicensingMessage.([]byte))
|
||||
goto retry
|
||||
default:
|
||||
glog.Error("Not a valid license packet")
|
||||
c.Emit("error", errors.New("Not a valid license packet"))
|
||||
return
|
||||
}
|
||||
|
||||
connect:
|
||||
c.transport.On("sec", c.recvData)
|
||||
c.Emit("connect", c.clientData[0].(*gcc.ClientCoreData), c.userId, c.channelId)
|
||||
return
|
||||
|
||||
retry:
|
||||
c.transport.Once("sec", c.recvLicenceInfo)
|
||||
return
|
||||
}
|
||||
|
||||
func (c *Client) sendClientNewLicenseRequest(data []byte) {
|
||||
var req lic.ServerLicenseRequest
|
||||
struc.Unpack(bytes.NewReader(data), &req)
|
||||
|
||||
var sc gcc.ServerCertificate
|
||||
if c.ServerSecurityData().ServerCertificate.DwVersion != 0 {
|
||||
sc = c.ServerSecurityData().ServerCertificate
|
||||
} else {
|
||||
rd := bytes.NewReader(req.ServerCertificate.BlobData)
|
||||
err := sc.Unpack(rd)
|
||||
if err != nil {
|
||||
glog.Error("read serverCertificate err:", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
serverRandom := req.ServerRandom
|
||||
clientRandom := core.Random(32)
|
||||
preMasterSecret := core.Random(48)
|
||||
masSecret := masterSecret(preMasterSecret, clientRandom, serverRandom)
|
||||
sessionKeyBlob := masterSecret(masSecret, serverRandom, clientRandom)
|
||||
//c.macKey = sessionKeyBlob[:16]
|
||||
c.macSalt = sessionKeyBlob[:16]
|
||||
c.initialDecrytKey = finalHash(sessionKeyBlob[16:32], clientRandom, serverRandom)
|
||||
|
||||
//format message
|
||||
message := &lic.ClientNewLicenseRequest{}
|
||||
message.PreferredKeyExchangeAlg = 0x00000001
|
||||
message.PlatformId = 0x04000000 | 0x00010000
|
||||
message.ClientRandom = clientRandom
|
||||
|
||||
buff := &bytes.Buffer{}
|
||||
|
||||
serverPubKey, err := sc.CertData.GetPublicKey()
|
||||
if err != nil {
|
||||
glog.Error("GetPublicKey failed:", err)
|
||||
return
|
||||
}
|
||||
ret, err := rsa.EncryptPKCS1v15(rand.Reader, serverPubKey, core.Reverse(preMasterSecret))
|
||||
if err != nil {
|
||||
glog.Error("EncryptPKCS1v15 failed:", err)
|
||||
return
|
||||
}
|
||||
|
||||
buff.Write(core.Reverse(ret))
|
||||
buff.Write([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00})
|
||||
message.EncryptedPreMasterSecret.BlobData = buff.Bytes()
|
||||
message.EncryptedPreMasterSecret.WBlobLen = uint16(buff.Len())
|
||||
message.EncryptedPreMasterSecret.WBlobType = lic.BB_RANDOM_BLOB
|
||||
|
||||
buff.Reset()
|
||||
buff.Write(c.info.UserName)
|
||||
buff.Write([]byte{0x00})
|
||||
message.ClientUserName.BlobData = buff.Bytes()
|
||||
message.ClientUserName.WBlobLen = uint16(buff.Len())
|
||||
message.ClientUserName.WBlobType = lic.BB_CLIENT_USER_NAME_BLOB
|
||||
|
||||
buff.Reset()
|
||||
buff.Write(c.ClientCoreData().ClientName[:])
|
||||
buff.Write([]byte{0x00})
|
||||
message.ClientMachineName.BlobData = buff.Bytes()
|
||||
message.ClientMachineName.WBlobLen = uint16(buff.Len())
|
||||
message.ClientMachineName.WBlobType = lic.BB_CLIENT_MACHINE_NAME_BLOB
|
||||
|
||||
buff.Reset()
|
||||
err = struc.Pack(buff, message)
|
||||
if err != nil {
|
||||
glog.Error("err:", err)
|
||||
}
|
||||
|
||||
c.sendFlagged(LICENSE_PKT, buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *Client) sendClientChallengeResponse(data []byte) {
|
||||
var pc lic.ServerPlatformChallenge
|
||||
struc.Unpack(bytes.NewReader(data), &pc)
|
||||
|
||||
serverEncryptedChallenge := pc.EncryptedPlatformChallenge.BlobData
|
||||
//decrypt server challenge
|
||||
//it should be TEST word in unicode format
|
||||
rc, _ := rc4.NewCipher(c.initialDecrytKey)
|
||||
serverChallenge := make([]byte, 20)
|
||||
rc.XORKeyStream(serverChallenge, serverEncryptedChallenge)
|
||||
//if serverChallenge != "T\x00E\x00S\x00T\x00\x00\x00":
|
||||
//raise InvalidExpectedDataException("bad license server challenge")
|
||||
|
||||
//generate hwid
|
||||
b := &bytes.Buffer{}
|
||||
b.Write(c.ClientCoreData().ClientName[:])
|
||||
b.Write(c.info.UserName)
|
||||
for i := 0; i < 2; i++ {
|
||||
b.Write([]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00})
|
||||
}
|
||||
hwid := b.Bytes()[:20]
|
||||
|
||||
encryptedHWID := make([]byte, 20)
|
||||
rc.XORKeyStream(encryptedHWID, hwid)
|
||||
|
||||
b.Reset()
|
||||
b.Write(serverChallenge)
|
||||
b.Write(hwid)
|
||||
|
||||
message := &lic.ClientPLatformChallengeResponse{}
|
||||
message.EncryptedPlatformChallengeResponse.BlobData = serverEncryptedChallenge
|
||||
message.EncryptedHWID.BlobData = encryptedHWID
|
||||
//message.MACData = macData(c.macKey, b.Bytes())[:16]
|
||||
message.MACData = macData(c.macSalt, b.Bytes())[:16]
|
||||
|
||||
b.Reset()
|
||||
struc.Pack(b, message)
|
||||
c.sendFlagged(LICENSE_PKT, b.Bytes())
|
||||
}
|
||||
|
||||
func (c *Client) recvData(channel string, s []byte) {
|
||||
glog.Trace("sec recvData", hex.EncodeToString(s))
|
||||
glog.Debugf("channel<%s> data len: %d", channel, len(s))
|
||||
data := c.decrytData(s)
|
||||
if channel != t125.GLOBAL_CHANNEL_NAME {
|
||||
c.Emit("channel", channel, data)
|
||||
return
|
||||
}
|
||||
c.Emit("data", data)
|
||||
}
|
||||
func (c *Client) SetFastPathListener(f core.FastPathListener) {
|
||||
c.fastPathListener = f
|
||||
}
|
||||
|
||||
func (c *Client) RecvFastPath(secFlag byte, s []byte) {
|
||||
data := s
|
||||
if c.enableEncryption && secFlag&FASTPATH_OUTPUT_ENCRYPTED != 0 {
|
||||
data = c.readEncryptedPayload(s, secFlag&FASTPATH_OUTPUT_SECURE_CHECKSUM != 0)
|
||||
}
|
||||
c.fastPathListener.RecvFastPath(secFlag, data)
|
||||
}
|
||||
|
||||
func (c *Client) SetChannelSender(f core.ChannelSender) {
|
||||
c.channelSender = f
|
||||
}
|
||||
|
||||
func (c *Client) SendToChannel(channel string, b []byte) (int, error) {
|
||||
if !c.enableEncryption {
|
||||
glog.Debug("Sec Client write", hex.EncodeToString(b))
|
||||
return c.channelSender.SendToChannel(channel, b)
|
||||
}
|
||||
var flag uint16 = ENCRYPT
|
||||
if c.enableSecureCheckSum {
|
||||
flag |= SECURE_CHECKSUM
|
||||
}
|
||||
data := c.writeEncryptedPayload(b, c.enableSecureCheckSum)
|
||||
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt16LE(flag, buff)
|
||||
core.WriteUInt16LE(0, buff)
|
||||
core.WriteBytes(data, buff)
|
||||
glog.Debug("Sec Client write", channel, hex.EncodeToString(buff.Bytes()))
|
||||
return c.channelSender.SendToChannel(channel, buff.Bytes())
|
||||
}
|
||||
@@ -0,0 +1,189 @@
|
||||
package ber
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
)
|
||||
|
||||
const (
|
||||
CLASS_MASK uint8 = 0xC0
|
||||
CLASS_UNIV = 0x00
|
||||
CLASS_APPL = 0x40
|
||||
CLASS_CTXT = 0x80
|
||||
CLASS_PRIV = 0xC0
|
||||
)
|
||||
|
||||
const (
|
||||
PC_MASK uint8 = 0x20
|
||||
PC_PRIMITIVE = 0x00
|
||||
PC_CONSTRUCT = 0x20
|
||||
)
|
||||
|
||||
const (
|
||||
TAG_MASK uint8 = 0x1F
|
||||
TAG_BOOLEAN = 0x01
|
||||
TAG_INTEGER = 0x02
|
||||
TAG_BIT_STRING = 0x03
|
||||
TAG_OCTET_STRING = 0x04
|
||||
TAG_OBJECT_IDENFIER = 0x06
|
||||
TAG_ENUMERATED = 0x0A
|
||||
TAG_SEQUENCE = 0x10
|
||||
TAG_SEQUENCE_OF = 0x10
|
||||
)
|
||||
|
||||
func berPC(pc bool) uint8 {
|
||||
if pc {
|
||||
return PC_CONSTRUCT
|
||||
}
|
||||
return PC_PRIMITIVE
|
||||
}
|
||||
|
||||
func ReadEnumerated(r io.Reader) (uint8, error) {
|
||||
if !ReadUniversalTag(TAG_ENUMERATED, false, r) {
|
||||
return 0, errors.New("invalid ber tag")
|
||||
}
|
||||
length, err := ReadLength(r)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if length != 1 {
|
||||
return 0, errors.New(fmt.Sprintf("enumerate size is wrong, get %v, expect 1", length))
|
||||
}
|
||||
return core.ReadUInt8(r)
|
||||
}
|
||||
|
||||
func ReadUniversalTag(tag uint8, pc bool, r io.Reader) bool {
|
||||
bb, _ := core.ReadUInt8(r)
|
||||
return bb == (CLASS_UNIV|berPC(pc))|(TAG_MASK&tag)
|
||||
}
|
||||
|
||||
func WriteUniversalTag(tag uint8, pc bool, w io.Writer) {
|
||||
core.WriteUInt8((CLASS_UNIV|berPC(pc))|(TAG_MASK&tag), w)
|
||||
}
|
||||
|
||||
func ReadLength(r io.Reader) (int, error) {
|
||||
ret := 0
|
||||
size, _ := core.ReadUInt8(r)
|
||||
if size&0x80 > 0 {
|
||||
size = size &^ 0x80
|
||||
if size == 1 {
|
||||
r, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
ret = int(r)
|
||||
} else if size == 2 {
|
||||
r, err := core.ReadUint16BE(r)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
ret = int(r)
|
||||
} else {
|
||||
return 0, errors.New("BER length may be 1 or 2")
|
||||
}
|
||||
} else {
|
||||
ret = int(size)
|
||||
}
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func WriteLength(size int, w io.Writer) {
|
||||
if size > 0x7f {
|
||||
core.WriteUInt8(0x82, w)
|
||||
core.WriteUInt16BE(uint16(size), w)
|
||||
} else {
|
||||
core.WriteUInt8(uint8(size), w)
|
||||
}
|
||||
}
|
||||
|
||||
func ReadInteger(r io.Reader) (int, error) {
|
||||
if !ReadUniversalTag(TAG_INTEGER, false, r) {
|
||||
return 0, errors.New("Bad integer tag")
|
||||
}
|
||||
size, _ := ReadLength(r)
|
||||
switch size {
|
||||
case 1:
|
||||
num, _ := core.ReadUInt8(r)
|
||||
return int(num), nil
|
||||
case 2:
|
||||
num, _ := core.ReadUint16BE(r)
|
||||
return int(num), nil
|
||||
case 3:
|
||||
integer1, _ := core.ReadUInt8(r)
|
||||
integer2, _ := core.ReadUint16BE(r)
|
||||
return int(integer2) + (int(integer1) << 16), nil
|
||||
case 4:
|
||||
num, _ := core.ReadUInt32BE(r)
|
||||
return int(num), nil
|
||||
default:
|
||||
return 0, errors.New("wrong size")
|
||||
}
|
||||
}
|
||||
|
||||
func WriteInteger(n int, w io.Writer) {
|
||||
WriteUniversalTag(TAG_INTEGER, false, w)
|
||||
if n <= 0xff {
|
||||
WriteLength(1, w)
|
||||
core.WriteUInt8(uint8(n), w)
|
||||
} else if n <= 0xffff {
|
||||
WriteLength(2, w)
|
||||
core.WriteUInt16BE(uint16(n), w)
|
||||
} else {
|
||||
WriteLength(4, w)
|
||||
core.WriteUInt32BE(uint32(n), w)
|
||||
}
|
||||
}
|
||||
|
||||
func WriteOctetstring(str string, w io.Writer) {
|
||||
WriteUniversalTag(TAG_OCTET_STRING, false, w)
|
||||
WriteLength(len(str), w)
|
||||
core.WriteBytes([]byte(str), w)
|
||||
}
|
||||
|
||||
func WriteBoolean(b bool, w io.Writer) {
|
||||
bb := uint8(0)
|
||||
if b {
|
||||
bb = uint8(0xff)
|
||||
}
|
||||
WriteUniversalTag(TAG_BOOLEAN, false, w)
|
||||
WriteLength(1, w)
|
||||
core.WriteUInt8(bb, w)
|
||||
}
|
||||
|
||||
func ReadApplicationTag(tag uint8, r io.Reader) (int, error) {
|
||||
bb, _ := core.ReadUInt8(r)
|
||||
if tag > 30 {
|
||||
if bb != (CLASS_APPL|PC_CONSTRUCT)|TAG_MASK {
|
||||
return 0, errors.New("ReadApplicationTag invalid data")
|
||||
}
|
||||
bb, _ := core.ReadUInt8(r)
|
||||
if bb != tag {
|
||||
return 0, errors.New("ReadApplicationTag bad tag")
|
||||
}
|
||||
} else {
|
||||
if bb != (CLASS_APPL|PC_CONSTRUCT)|(TAG_MASK&tag) {
|
||||
return 0, errors.New("ReadApplicationTag invalid data2")
|
||||
}
|
||||
}
|
||||
return ReadLength(r)
|
||||
}
|
||||
|
||||
func WriteApplicationTag(tag uint8, size int, w io.Writer) {
|
||||
if tag > 30 {
|
||||
core.WriteUInt8((CLASS_APPL|PC_CONSTRUCT)|TAG_MASK, w)
|
||||
core.WriteUInt8(tag, w)
|
||||
WriteLength(size, w)
|
||||
} else {
|
||||
core.WriteUInt8((CLASS_APPL|PC_CONSTRUCT)|(TAG_MASK&tag), w)
|
||||
WriteLength(size, w)
|
||||
}
|
||||
}
|
||||
|
||||
func WriteEncodedDomainParams(data []byte, w io.Writer) {
|
||||
WriteUniversalTag(TAG_SEQUENCE, true, w)
|
||||
WriteLength(len(data), w)
|
||||
core.WriteBytes(data, w)
|
||||
}
|
||||
@@ -0,0 +1,642 @@
|
||||
package gcc
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/rsa"
|
||||
"crypto/x509"
|
||||
"crypto/x509/pkix"
|
||||
"encoding/asn1"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"math/big"
|
||||
"os"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
|
||||
"github.com/lunixbochs/struc"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/per"
|
||||
)
|
||||
|
||||
var t124_02_98_oid = []byte{0, 0, 20, 124, 0, 1}
|
||||
var h221_cs_key = "Duca"
|
||||
var h221_sc_key = "McDn"
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240509.aspx
|
||||
*/
|
||||
type Message uint16
|
||||
|
||||
const (
|
||||
//server -> client
|
||||
SC_CORE Message = 0x0C01
|
||||
SC_SECURITY = 0x0C02
|
||||
SC_NET = 0x0C03
|
||||
//client -> server
|
||||
CS_CORE = 0xC001
|
||||
CS_SECURITY = 0xC002
|
||||
CS_NET = 0xC003
|
||||
CS_CLUSTER = 0xC004
|
||||
CS_MONITOR = 0xC005
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240510.aspx
|
||||
*/
|
||||
type ColorDepth uint16
|
||||
|
||||
const (
|
||||
RNS_UD_COLOR_8BPP ColorDepth = 0xCA01
|
||||
RNS_UD_COLOR_16BPP_555 = 0xCA02
|
||||
RNS_UD_COLOR_16BPP_565 = 0xCA03
|
||||
RNS_UD_COLOR_24BPP = 0xCA04
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240510.aspx
|
||||
*/
|
||||
type HighColor uint16
|
||||
|
||||
const (
|
||||
HIGH_COLOR_4BPP HighColor = 0x0004
|
||||
HIGH_COLOR_8BPP = 0x0008
|
||||
HIGH_COLOR_15BPP = 0x000f
|
||||
HIGH_COLOR_16BPP = 0x0010
|
||||
HIGH_COLOR_24BPP = 0x0018
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240510.aspx
|
||||
*/
|
||||
type Support uint16
|
||||
|
||||
const (
|
||||
RNS_UD_24BPP_SUPPORT uint16 = 0x0001
|
||||
RNS_UD_16BPP_SUPPORT = 0x0002
|
||||
RNS_UD_15BPP_SUPPORT = 0x0004
|
||||
RNS_UD_32BPP_SUPPORT = 0x0008
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240510.aspx
|
||||
*/
|
||||
type CapabilityFlag uint16
|
||||
|
||||
const (
|
||||
RNS_UD_CS_SUPPORT_ERRINFO_PDU uint16 = 0x0001
|
||||
RNS_UD_CS_WANT_32BPP_SESSION = 0x0002
|
||||
RNS_UD_CS_SUPPORT_STATUSINFO_PDU = 0x0004
|
||||
RNS_UD_CS_STRONG_ASYMMETRIC_KEYS = 0x0008
|
||||
RNS_UD_CS_UNUSED = 0x0010
|
||||
RNS_UD_CS_VALID_CONNECTION_TYPE = 0x0020
|
||||
RNS_UD_CS_SUPPORT_MONITOR_LAYOUT_PDU = 0x0040
|
||||
RNS_UD_CS_SUPPORT_NETCHAR_AUTODETECT = 0x0080
|
||||
RNS_UD_CS_SUPPORT_DYNVC_GFX_PROTOCOL = 0x0100
|
||||
RNS_UD_CS_SUPPORT_DYNAMIC_TIME_ZONE = 0x0200
|
||||
RNS_UD_CS_SUPPORT_HEARTBEAT_PDU = 0x0400
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240510.aspx
|
||||
*/
|
||||
type ConnectionType uint8
|
||||
|
||||
const (
|
||||
CONNECTION_TYPE_MODEM ConnectionType = 0x01
|
||||
CONNECTION_TYPE_BROADBAND_LOW = 0x02
|
||||
CONNECTION_TYPE_SATELLITEV = 0x03
|
||||
CONNECTION_TYPE_BROADBAND_HIGH = 0x04
|
||||
CONNECTION_TYPE_WAN = 0x05
|
||||
CONNECTION_TYPE_LAN = 0x06
|
||||
CONNECTION_TYPE_AUTODETECT = 0x07
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240510.aspx
|
||||
*/
|
||||
type VERSION uint32
|
||||
|
||||
const (
|
||||
RDP_VERSION_4 VERSION = 0x00080001
|
||||
RDP_VERSION_5_PLUS = 0x00080004
|
||||
)
|
||||
|
||||
type Sequence uint16
|
||||
|
||||
const (
|
||||
RNS_UD_SAS_DEL Sequence = 0xAA03
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240511.aspx
|
||||
*/
|
||||
type EncryptionMethod uint32
|
||||
|
||||
const (
|
||||
ENCRYPTION_FLAG_40BIT uint32 = 0x00000001
|
||||
ENCRYPTION_FLAG_128BIT = 0x00000002
|
||||
ENCRYPTION_FLAG_56BIT = 0x00000008
|
||||
FIPS_ENCRYPTION_FLAG = 0x00000010
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240518.aspx
|
||||
*/
|
||||
type EncryptionLevel uint32
|
||||
|
||||
const (
|
||||
ENCRYPTION_LEVEL_NONE EncryptionLevel = 0x00000000
|
||||
ENCRYPTION_LEVEL_LOW = 0x00000001
|
||||
ENCRYPTION_LEVEL_CLIENT_COMPATIBLE = 0x00000002
|
||||
ENCRYPTION_LEVEL_HIGH = 0x00000003
|
||||
ENCRYPTION_LEVEL_FIPS = 0x00000004
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240513.aspx
|
||||
*/
|
||||
type ChannelOptions uint32
|
||||
|
||||
const (
|
||||
CHANNEL_OPTION_INITIALIZED ChannelOptions = 0x80000000
|
||||
CHANNEL_OPTION_ENCRYPT_RDP = 0x40000000
|
||||
CHANNEL_OPTION_ENCRYPT_SC = 0x20000000
|
||||
CHANNEL_OPTION_ENCRYPT_CS = 0x10000000
|
||||
CHANNEL_OPTION_PRI_HIGH = 0x08000000
|
||||
CHANNEL_OPTION_PRI_MED = 0x04000000
|
||||
CHANNEL_OPTION_PRI_LOW = 0x02000000
|
||||
CHANNEL_OPTION_COMPRESS_RDP = 0x00800000
|
||||
CHANNEL_OPTION_COMPRESS = 0x00400000
|
||||
CHANNEL_OPTION_SHOW_PROTOCOL = 0x00200000
|
||||
REMOTE_CONTROL_PERSISTENT = 0x00100000
|
||||
)
|
||||
|
||||
/**
|
||||
* IBM_101_102_KEYS is the most common keyboard type
|
||||
*/
|
||||
type KeyboardType uint32
|
||||
|
||||
const (
|
||||
KT_IBM_PC_XT_83_KEY KeyboardType = 0x00000001
|
||||
KT_OLIVETTI = 0x00000002
|
||||
KT_IBM_PC_AT_84_KEY = 0x00000003
|
||||
KT_IBM_101_102_KEYS = 0x00000004
|
||||
KT_NOKIA_1050 = 0x00000005
|
||||
KT_NOKIA_9140 = 0x00000006
|
||||
KT_JAPANESE = 0x00000007
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://technet.microsoft.com/en-us/library/cc766503%28WS.10%29.aspx
|
||||
*/
|
||||
type KeyboardLayout uint32
|
||||
|
||||
const (
|
||||
ARABIC KeyboardLayout = 0x00000401
|
||||
BULGARIAN = 0x00000402
|
||||
CHINESE_US_KEYBOARD = 0x00000404
|
||||
CZECH = 0x00000405
|
||||
DANISH = 0x00000406
|
||||
GERMAN = 0x00000407
|
||||
GREEK = 0x00000408
|
||||
US = 0x00000409
|
||||
SPANISH = 0x0000040a
|
||||
FINNISH = 0x0000040b
|
||||
FRENCH = 0x0000040c
|
||||
HEBREW = 0x0000040d
|
||||
HUNGARIAN = 0x0000040e
|
||||
ICELANDIC = 0x0000040f
|
||||
ITALIAN = 0x00000410
|
||||
JAPANESE = 0x00000411
|
||||
KOREAN = 0x00000412
|
||||
DUTCH = 0x00000413
|
||||
NORWEGIAN = 0x00000414
|
||||
)
|
||||
|
||||
/**
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240521.aspx
|
||||
*/
|
||||
type CertificateType uint32
|
||||
|
||||
const (
|
||||
CERT_CHAIN_VERSION_1 CertificateType = 0x00000001
|
||||
CERT_CHAIN_VERSION_2 = 0x00000002
|
||||
)
|
||||
|
||||
type ChannelDef struct {
|
||||
Name string `struc:"little"`
|
||||
Options uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
type ClientCoreData struct {
|
||||
RdpVersion VERSION `struc:"uint32,little"`
|
||||
DesktopWidth uint16 `struc:"little"`
|
||||
DesktopHeight uint16 `struc:"little"`
|
||||
ColorDepth ColorDepth `struc:"little"`
|
||||
SasSequence Sequence `struc:"little"`
|
||||
KbdLayout KeyboardLayout `struc:"little"`
|
||||
ClientBuild uint32 `struc:"little"`
|
||||
ClientName [32]byte `struc:"[32]byte"`
|
||||
KeyboardType uint32 `struc:"little"`
|
||||
KeyboardSubType uint32 `struc:"little"`
|
||||
KeyboardFnKeys uint32 `struc:"little"`
|
||||
ImeFileName [64]byte `struc:"[64]byte"`
|
||||
PostBeta2ColorDepth ColorDepth `struc:"little"`
|
||||
ClientProductId uint16 `struc:"little"`
|
||||
SerialNumber uint32 `struc:"little"`
|
||||
HighColorDepth HighColor `struc:"little"`
|
||||
SupportedColorDepths uint16 `struc:"little"`
|
||||
EarlyCapabilityFlags uint16 `struc:"little"`
|
||||
ClientDigProductId [64]byte `struc:"[64]byte"`
|
||||
ConnectionType uint8 `struc:"uint8"`
|
||||
Pad1octet uint8 `struc:"uint8"`
|
||||
ServerSelectedProtocol uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func NewClientCoreData() *ClientCoreData {
|
||||
name, _ := os.Hostname()
|
||||
var ClientName [32]byte
|
||||
copy(ClientName[:], core.UnicodeEncode(name)[:])
|
||||
return &ClientCoreData{
|
||||
RDP_VERSION_5_PLUS, 1280, 800, RNS_UD_COLOR_8BPP,
|
||||
RNS_UD_SAS_DEL, US, 3790, ClientName, KT_IBM_101_102_KEYS,
|
||||
0, 12, [64]byte{}, RNS_UD_COLOR_8BPP, 1, 0, HIGH_COLOR_24BPP,
|
||||
RNS_UD_15BPP_SUPPORT | RNS_UD_16BPP_SUPPORT | RNS_UD_24BPP_SUPPORT | RNS_UD_32BPP_SUPPORT,
|
||||
RNS_UD_CS_SUPPORT_ERRINFO_PDU, [64]byte{}, 0, 0, 0}
|
||||
}
|
||||
|
||||
func (data *ClientCoreData) Pack() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt16LE(CS_CORE, buff) // 01C0
|
||||
core.WriteUInt16LE(0xd8, buff) // d800
|
||||
struc.Pack(buff, data)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type ClientNetworkData struct {
|
||||
ChannelCount uint32
|
||||
ChannelDefArray []ChannelDef
|
||||
}
|
||||
|
||||
func NewClientNetworkData() *ClientNetworkData {
|
||||
n := &ClientNetworkData{ChannelDefArray: make([]ChannelDef, 0, 100)}
|
||||
|
||||
/*var d1 ChannelDef
|
||||
d1.Name = plugin.RDPDR_SVC_CHANNEL_NAME
|
||||
d1.Options = uint32(CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP |
|
||||
CHANNEL_OPTION_COMPRESS_RDP)
|
||||
n.ChannelDefArray = append(n.ChannelDefArray, d1)
|
||||
|
||||
var d2 ChannelDef
|
||||
d2.Name = plugin.RDPSND_SVC_CHANNEL_NAME
|
||||
d2.Options = uint32(CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP |
|
||||
CHANNEL_OPTION_COMPRESS_RDP | CHANNEL_OPTION_SHOW_PROTOCOL)
|
||||
n.ChannelDefArray = append(n.ChannelDefArray, d2)*/
|
||||
|
||||
return n
|
||||
}
|
||||
|
||||
func (n *ClientNetworkData) AddVirtualChannel(name string, option uint32) {
|
||||
var d ChannelDef
|
||||
d.Name = name
|
||||
d.Options = option
|
||||
n.ChannelDefArray = append(n.ChannelDefArray, d)
|
||||
n.ChannelCount++
|
||||
}
|
||||
|
||||
func (n *ClientNetworkData) Pack() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt16LE(CS_NET, buff) // type
|
||||
length := uint16(n.ChannelCount*12 + 8)
|
||||
core.WriteUInt16LE(length, buff) // len 8
|
||||
core.WriteUInt32LE(n.ChannelCount, buff)
|
||||
for i := 0; i < int(n.ChannelCount); i++ {
|
||||
v := n.ChannelDefArray[i]
|
||||
name := make([]byte, 8)
|
||||
copy(name, []byte(v.Name))
|
||||
core.WriteBytes(name[:], buff)
|
||||
core.WriteUInt32LE(v.Options, buff)
|
||||
}
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type ClientSecurityData struct {
|
||||
EncryptionMethods uint32
|
||||
ExtEncryptionMethods uint32
|
||||
}
|
||||
|
||||
func NewClientSecurityData() *ClientSecurityData {
|
||||
return &ClientSecurityData{
|
||||
ENCRYPTION_FLAG_40BIT | ENCRYPTION_FLAG_56BIT | ENCRYPTION_FLAG_128BIT,
|
||||
00}
|
||||
}
|
||||
|
||||
func (d *ClientSecurityData) Pack() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt16LE(CS_SECURITY, buff) // type
|
||||
core.WriteUInt16LE(0x0c, buff) // len 12
|
||||
core.WriteUInt32LE(d.EncryptionMethods, buff)
|
||||
core.WriteUInt32LE(d.ExtEncryptionMethods, buff)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type RSAPublicKey struct {
|
||||
Magic uint32 `struc:"little"` //0x31415352
|
||||
Keylen uint32 `struc:"little,sizeof=Modulus"`
|
||||
Bitlen uint32 `struc:"little"`
|
||||
Datalen uint32 `struc:"little"`
|
||||
PubExp uint32 `struc:"little"`
|
||||
Modulus []byte `struc:"little"`
|
||||
Padding []byte `struc:"[8]byte"`
|
||||
}
|
||||
|
||||
type ProprietaryServerCertificate struct {
|
||||
DwSigAlgId uint32 `struc:"little"` //0x00000001
|
||||
DwKeyAlgId uint32 `struc:"little"` //0x00000001
|
||||
PublicKeyBlobType uint16 `struc:"little"` //0x0006
|
||||
PublicKeyBlobLen uint16 `struc:"little,sizeof=PublicKeyBlob"`
|
||||
PublicKeyBlob RSAPublicKey `struc:"little"`
|
||||
SignatureBlobType uint16 `struc:"little"` //0x0008
|
||||
SignatureBlobLen uint16 `struc:"little,sizeof=SignatureBlob"`
|
||||
SignatureBlob []byte `struc:"little"`
|
||||
//PaddingLen uint16 `struc:"little,sizeof=Padding,skip"`
|
||||
Padding []byte `struc:"[8]byte"`
|
||||
}
|
||||
|
||||
func (p *ProprietaryServerCertificate) GetPublicKey() (*rsa.PublicKey, error) {
|
||||
b := new(big.Int).SetBytes(core.Reverse(p.PublicKeyBlob.Modulus))
|
||||
e := new(big.Int).SetInt64(int64(p.PublicKeyBlob.PubExp))
|
||||
return &rsa.PublicKey{N: b, E: int(e.Int64())}, nil
|
||||
}
|
||||
func (p *ProprietaryServerCertificate) Verify() bool {
|
||||
return true
|
||||
}
|
||||
func (p *ProprietaryServerCertificate) Encrypt() []byte {
|
||||
//todo
|
||||
return nil
|
||||
}
|
||||
func (p *ProprietaryServerCertificate) Unpack(r io.Reader) error {
|
||||
p.DwSigAlgId, _ = core.ReadUInt32LE(r)
|
||||
p.DwKeyAlgId, _ = core.ReadUInt32LE(r)
|
||||
p.PublicKeyBlobType, _ = core.ReadUint16LE(r)
|
||||
p.PublicKeyBlobLen, _ = core.ReadUint16LE(r)
|
||||
var b RSAPublicKey
|
||||
b.Magic, _ = core.ReadUInt32LE(r)
|
||||
b.Keylen, _ = core.ReadUInt32LE(r)
|
||||
b.Bitlen, _ = core.ReadUInt32LE(r)
|
||||
b.Datalen, _ = core.ReadUInt32LE(r)
|
||||
b.PubExp, _ = core.ReadUInt32LE(r)
|
||||
b.Modulus, _ = core.ReadBytes(int(b.Keylen)-8, r)
|
||||
b.Padding, _ = core.ReadBytes(8, r)
|
||||
p.PublicKeyBlob = b
|
||||
p.SignatureBlobType, _ = core.ReadUint16LE(r)
|
||||
p.SignatureBlobLen, _ = core.ReadUint16LE(r)
|
||||
p.SignatureBlob, _ = core.ReadBytes(int(p.SignatureBlobLen)-8, r)
|
||||
p.Padding, _ = core.ReadBytes(8, r)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type CertBlob struct {
|
||||
CbCert uint32 `struc:"little,sizeof=AbCert"`
|
||||
AbCert []byte `struc:"little"`
|
||||
}
|
||||
type X509CertificateChain struct {
|
||||
NumCertBlobs uint32 `struc:"little,sizeof=CertBlobArray"`
|
||||
CertBlobArray []CertBlob `struc:"little"`
|
||||
Padding []byte `struc:"[12]byte"`
|
||||
}
|
||||
|
||||
func (x *X509CertificateChain) GetPublicKey() (*rsa.PublicKey, error) {
|
||||
if len(x.CertBlobArray) == 0 {
|
||||
return nil, errors.New("empty certificate chain")
|
||||
}
|
||||
data := x.CertBlobArray[len(x.CertBlobArray)-1].AbCert
|
||||
cert, err := x509.ParseCertificate(data)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("parse certificate: %w", err)
|
||||
}
|
||||
if cert.PublicKey == nil {
|
||||
var pubKeyInfo struct {
|
||||
Algorithm pkix.AlgorithmIdentifier
|
||||
SubjectPublicKey asn1.BitString
|
||||
}
|
||||
_, err = asn1.Unmarshal(cert.RawSubjectPublicKeyInfo, &pubKeyInfo)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("unmarshal public key info: %w", err)
|
||||
}
|
||||
rsaPublicKey, err := x509.ParsePKCS1PublicKey(pubKeyInfo.SubjectPublicKey.Bytes)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("parse PKCS1 public key: %w", err)
|
||||
}
|
||||
return rsaPublicKey, nil
|
||||
}
|
||||
rsaPublicKey, ok := cert.PublicKey.(*rsa.PublicKey)
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("unsupported public key type: %T", cert.PublicKey)
|
||||
}
|
||||
return rsaPublicKey, nil
|
||||
}
|
||||
func (x *X509CertificateChain) Verify() bool {
|
||||
return true
|
||||
}
|
||||
func (x *X509CertificateChain) Encrypt() []byte {
|
||||
//todo
|
||||
return nil
|
||||
}
|
||||
func (x *X509CertificateChain) Unpack(r io.Reader) error {
|
||||
return struc.Unpack(r, x)
|
||||
}
|
||||
|
||||
type ServerCoreData struct {
|
||||
RdpVersion VERSION `struc:"uint32,little"`
|
||||
ClientRequestedProtocol uint32 `struc:"little"`
|
||||
EarlyCapabilityFlags uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func NewServerCoreData() *ServerCoreData {
|
||||
return &ServerCoreData{
|
||||
RDP_VERSION_5_PLUS, 0, 0}
|
||||
}
|
||||
|
||||
func (d *ServerCoreData) Serialize() []byte {
|
||||
return []byte{}
|
||||
}
|
||||
|
||||
func (d *ServerCoreData) ScType() Message {
|
||||
return SC_CORE
|
||||
}
|
||||
func (d *ServerCoreData) Unpack(r io.Reader) error {
|
||||
version, _ := core.ReadUInt32LE(r)
|
||||
d.RdpVersion = VERSION(version)
|
||||
d.ClientRequestedProtocol, _ = core.ReadUInt32LE(r)
|
||||
d.EarlyCapabilityFlags, _ = core.ReadUInt32LE(r)
|
||||
|
||||
return nil
|
||||
//return struc.Unpack(r, d)
|
||||
}
|
||||
|
||||
type ServerNetworkData struct {
|
||||
MCSChannelId uint16 `struc:"little"`
|
||||
ChannelCount uint16 `struc:"little,sizeof=ChannelIdArray"`
|
||||
ChannelIdArray []uint16 `struc:"little"`
|
||||
}
|
||||
|
||||
func NewServerNetworkData() *ServerNetworkData {
|
||||
return &ServerNetworkData{}
|
||||
}
|
||||
func (d *ServerNetworkData) ScType() Message {
|
||||
return SC_NET
|
||||
}
|
||||
func (d *ServerNetworkData) Unpack(r io.Reader) error {
|
||||
return struc.Unpack(r, d)
|
||||
}
|
||||
|
||||
type CertData interface {
|
||||
GetPublicKey() (*rsa.PublicKey, error)
|
||||
Verify() bool
|
||||
Unpack(io.Reader) error
|
||||
}
|
||||
type ServerCertificate struct {
|
||||
DwVersion uint32
|
||||
CertData CertData
|
||||
}
|
||||
|
||||
func (sc *ServerCertificate) Unpack(r io.Reader) error {
|
||||
sc.DwVersion, _ = core.ReadUInt32LE(r)
|
||||
var cd CertData
|
||||
switch CertificateType(sc.DwVersion & 0x7fffffff) {
|
||||
case CERT_CHAIN_VERSION_1:
|
||||
glog.Debug("ProprietaryServerCertificate")
|
||||
cd = &ProprietaryServerCertificate{}
|
||||
case CERT_CHAIN_VERSION_2:
|
||||
glog.Debug("X509CertificateChain")
|
||||
cd = &X509CertificateChain{}
|
||||
default:
|
||||
glog.Error("Unsupported version:", sc.DwVersion&0x7fffffff)
|
||||
return errors.New("Unsupported version")
|
||||
}
|
||||
if cd != nil {
|
||||
err := cd.Unpack(r)
|
||||
if err != nil {
|
||||
glog.Error("Unpack:", err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
sc.CertData = cd
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type ServerSecurityData struct {
|
||||
EncryptionMethod uint32 `struc:"little"`
|
||||
EncryptionLevel uint32 `struc:"little"`
|
||||
ServerRandomLen uint32 //0x00000020
|
||||
ServerCertLen uint32
|
||||
ServerRandom []byte
|
||||
ServerCertificate ServerCertificate
|
||||
}
|
||||
|
||||
func NewServerSecurityData() *ServerSecurityData {
|
||||
return &ServerSecurityData{
|
||||
0, 0, 0x00000020, 0, []byte{}, ServerCertificate{}}
|
||||
}
|
||||
func (d *ServerSecurityData) ScType() Message {
|
||||
return SC_SECURITY
|
||||
}
|
||||
func (s *ServerSecurityData) Unpack(r io.Reader) error {
|
||||
s.EncryptionMethod, _ = core.ReadUInt32LE(r)
|
||||
s.EncryptionLevel, _ = core.ReadUInt32LE(r)
|
||||
if !(s.EncryptionMethod == 0 && s.EncryptionLevel == 0) {
|
||||
s.ServerRandomLen, _ = core.ReadUInt32LE(r)
|
||||
s.ServerCertLen, _ = core.ReadUInt32LE(r)
|
||||
s.ServerRandom, _ = core.ReadBytes(int(s.ServerRandomLen), r)
|
||||
var sc ServerCertificate
|
||||
data, _ := core.ReadBytes(int(s.ServerCertLen), r)
|
||||
rd := bytes.NewReader(data)
|
||||
err := sc.Unpack(rd)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
s.ServerCertificate = sc
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func MakeConferenceCreateRequest(userData []byte) []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
per.WriteChoice(0, buff) // 00
|
||||
per.WriteObjectIdentifier(t124_02_98_oid, buff) // 05:00:14:7c:00:01
|
||||
per.WriteLength(len(userData)+14, buff)
|
||||
per.WriteChoice(0, buff) // 00
|
||||
per.WriteSelection(0x08, buff) // 08
|
||||
per.WriteNumericString("1", 1, buff) // 00 10
|
||||
per.WritePadding(1, buff) // 00
|
||||
per.WriteNumberOfSet(1, buff) // 01
|
||||
per.WriteChoice(0xc0, buff) // c0
|
||||
per.WriteOctetStream(h221_cs_key, 4, buff) // 00 44:75:63:61
|
||||
per.WriteOctetStream(string(userData), 0, buff)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
type ScData interface {
|
||||
ScType() Message
|
||||
Unpack(io.Reader) error
|
||||
}
|
||||
|
||||
func ReadConferenceCreateResponse(data []byte) []interface{} {
|
||||
ret := make([]interface{}, 0, 3)
|
||||
|
||||
r := bytes.NewReader(data)
|
||||
per.ReadChoice(r)
|
||||
if !per.ReadObjectIdentifier(r, t124_02_98_oid) {
|
||||
glog.Error("NODE_RDP_PROTOCOL_T125_GCC_BAD_OBJECT_IDENTIFIER_T124")
|
||||
return ret
|
||||
}
|
||||
per.ReadLength(r)
|
||||
per.ReadChoice(r)
|
||||
per.ReadInteger16(r)
|
||||
per.ReadInteger(r)
|
||||
per.ReadEnumerates(r)
|
||||
per.ReadNumberOfSet(r)
|
||||
per.ReadChoice(r)
|
||||
|
||||
if !per.ReadOctetStream(r, h221_sc_key, 4) {
|
||||
glog.Error("NODE_RDP_PROTOCOL_T125_GCC_BAD_H221_SC_KEY")
|
||||
return ret
|
||||
}
|
||||
|
||||
ln, _ := per.ReadLength(r)
|
||||
for ln > 0 {
|
||||
t, _ := core.ReadUint16LE(r)
|
||||
glog.Debugf("Message type 0x%x,ln:%v", t, ln)
|
||||
l, _ := core.ReadUint16LE(r)
|
||||
dataBytes, _ := core.ReadBytes(int(l)-4, r)
|
||||
ln = ln - l
|
||||
var d ScData
|
||||
switch Message(t) {
|
||||
case SC_CORE:
|
||||
d = &ServerCoreData{}
|
||||
case SC_SECURITY:
|
||||
d = &ServerSecurityData{}
|
||||
case SC_NET:
|
||||
d = &ServerNetworkData{}
|
||||
default:
|
||||
glog.Error("Unknown type", t)
|
||||
continue
|
||||
}
|
||||
|
||||
if d != nil {
|
||||
r := bytes.NewReader(dataBytes)
|
||||
err := d.Unpack(r)
|
||||
if err != nil {
|
||||
glog.Warn("Unpack:", err)
|
||||
}
|
||||
ret = append(ret, d)
|
||||
}
|
||||
}
|
||||
|
||||
return ret
|
||||
}
|
||||
@@ -0,0 +1,558 @@
|
||||
package t125
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"reflect"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/emission"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/ber"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/gcc"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125/per"
|
||||
)
|
||||
|
||||
// take idea from https://github.com/Madnikulin50/gordp
|
||||
|
||||
// Multiple Channel Service layer
|
||||
|
||||
type MCSMessage uint8
|
||||
|
||||
const (
|
||||
MCS_TYPE_CONNECT_INITIAL MCSMessage = 0x65
|
||||
MCS_TYPE_CONNECT_RESPONSE = 0x66
|
||||
)
|
||||
|
||||
type MCSDomainPDU uint16
|
||||
|
||||
const (
|
||||
ERECT_DOMAIN_REQUEST MCSDomainPDU = 1
|
||||
DISCONNECT_PROVIDER_ULTIMATUM = 8
|
||||
ATTACH_USER_REQUEST = 10
|
||||
ATTACH_USER_CONFIRM = 11
|
||||
CHANNEL_JOIN_REQUEST = 14
|
||||
CHANNEL_JOIN_CONFIRM = 15
|
||||
SEND_DATA_REQUEST = 25
|
||||
SEND_DATA_INDICATION = 26
|
||||
)
|
||||
|
||||
const (
|
||||
MCS_GLOBAL_CHANNEL_ID uint16 = 1003
|
||||
MCS_USERCHANNEL_BASE = 1001
|
||||
)
|
||||
|
||||
const (
|
||||
GLOBAL_CHANNEL_NAME = "global"
|
||||
)
|
||||
|
||||
/**
|
||||
* Format MCS PDULayer header packet
|
||||
* @param mcsPdu {integer}
|
||||
* @param options {integer}
|
||||
* @returns {type.UInt8} headers
|
||||
*/
|
||||
func writeMCSPDUHeader(mcsPdu MCSDomainPDU, options uint8, w io.Writer) {
|
||||
core.WriteUInt8((uint8(mcsPdu)<<2)|options, w)
|
||||
}
|
||||
|
||||
func readMCSPDUHeader(options uint8, mcsPdu MCSDomainPDU) bool {
|
||||
return (options >> 2) == uint8(mcsPdu)
|
||||
}
|
||||
|
||||
type DomainParameters struct {
|
||||
MaxChannelIds int
|
||||
MaxUserIds int
|
||||
MaxTokenIds int
|
||||
NumPriorities int
|
||||
MinThoughput int
|
||||
MaxHeight int
|
||||
MaxMCSPDUsize int
|
||||
ProtocolVersion int
|
||||
}
|
||||
|
||||
/**
|
||||
* @see http://www.itu.int/rec/T-REC-T.125-199802-I/en page 25
|
||||
* @returns {asn1.univ.Sequence}
|
||||
*/
|
||||
func NewDomainParameters(
|
||||
maxChannelIds int,
|
||||
maxUserIds int,
|
||||
maxTokenIds int,
|
||||
numPriorities int,
|
||||
minThoughput int,
|
||||
maxHeight int,
|
||||
maxMCSPDUsize int,
|
||||
protocolVersion int) *DomainParameters {
|
||||
return &DomainParameters{maxChannelIds, maxUserIds, maxTokenIds,
|
||||
numPriorities, minThoughput, maxHeight, maxMCSPDUsize, protocolVersion}
|
||||
}
|
||||
|
||||
func (d *DomainParameters) BER() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
ber.WriteInteger(d.MaxChannelIds, buff)
|
||||
ber.WriteInteger(d.MaxUserIds, buff)
|
||||
ber.WriteInteger(d.MaxTokenIds, buff)
|
||||
ber.WriteInteger(1, buff)
|
||||
ber.WriteInteger(0, buff)
|
||||
ber.WriteInteger(1, buff)
|
||||
ber.WriteInteger(d.MaxMCSPDUsize, buff)
|
||||
ber.WriteInteger(2, buff)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
func ReadDomainParameters(r io.Reader) (*DomainParameters, error) {
|
||||
if !ber.ReadUniversalTag(ber.TAG_SEQUENCE, true, r) {
|
||||
return nil, errors.New("bad BER tags")
|
||||
}
|
||||
d := &DomainParameters{}
|
||||
ber.ReadLength(r)
|
||||
|
||||
d.MaxChannelIds, _ = ber.ReadInteger(r)
|
||||
d.MaxUserIds, _ = ber.ReadInteger(r)
|
||||
d.MaxTokenIds, _ = ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
d.MaxMCSPDUsize, _ = ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
return d, nil
|
||||
}
|
||||
|
||||
/**
|
||||
* @see http://www.itu.int/rec/T-REC-T.125-199802-I/en page 25
|
||||
* @param userData {Buffer}
|
||||
* @returns {asn1.univ.Sequence}
|
||||
*/
|
||||
type ConnectInitial struct {
|
||||
CallingDomainSelector []byte
|
||||
CalledDomainSelector []byte
|
||||
UpwardFlag bool
|
||||
TargetParameters DomainParameters
|
||||
MinimumParameters DomainParameters
|
||||
MaximumParameters DomainParameters
|
||||
UserData []byte
|
||||
}
|
||||
|
||||
func NewConnectInitial(userData []byte) ConnectInitial {
|
||||
return ConnectInitial{[]byte{0x1},
|
||||
[]byte{0x1},
|
||||
true,
|
||||
*NewDomainParameters(34, 2, 0, 1, 0, 1, 0xffff, 2),
|
||||
*NewDomainParameters(1, 1, 1, 1, 0, 1, 0x420, 2),
|
||||
*NewDomainParameters(0xffff, 0xfc17, 0xffff, 1, 0, 1, 0xffff, 2),
|
||||
userData}
|
||||
}
|
||||
|
||||
func (c *ConnectInitial) BER() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
ber.WriteOctetstring(string(c.CallingDomainSelector), buff)
|
||||
ber.WriteOctetstring(string(c.CalledDomainSelector), buff)
|
||||
ber.WriteBoolean(c.UpwardFlag, buff)
|
||||
ber.WriteEncodedDomainParams(c.TargetParameters.BER(), buff)
|
||||
ber.WriteEncodedDomainParams(c.MinimumParameters.BER(), buff)
|
||||
ber.WriteEncodedDomainParams(c.MaximumParameters.BER(), buff)
|
||||
ber.WriteOctetstring(string(c.UserData), buff)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
/**
|
||||
* @see http://www.itu.int/rec/T-REC-T.125-199802-I/en page 25
|
||||
* @returns {asn1.univ.Sequence}
|
||||
*/
|
||||
|
||||
type ConnectResponse struct {
|
||||
result uint8
|
||||
calledConnectId int
|
||||
domainParameters *DomainParameters
|
||||
userData []byte
|
||||
}
|
||||
|
||||
func NewConnectResponse(userData []byte) *ConnectResponse {
|
||||
return &ConnectResponse{0,
|
||||
0,
|
||||
NewDomainParameters(22, 3, 0, 1, 0, 1, 0xfff8, 2),
|
||||
userData}
|
||||
}
|
||||
|
||||
func ReadConnectResponse(r io.Reader) (*ConnectResponse, error) {
|
||||
c := &ConnectResponse{}
|
||||
var err error
|
||||
_, err = ber.ReadApplicationTag(MCS_TYPE_CONNECT_RESPONSE, r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
c.result, err = ber.ReadEnumerated(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c.calledConnectId, err = ber.ReadInteger(r)
|
||||
c.domainParameters, err = ReadDomainParameters(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !ber.ReadUniversalTag(ber.TAG_OCTET_STRING, false, r) {
|
||||
return nil, errors.New("invalid expected BER tag")
|
||||
}
|
||||
dataLen, _ := ber.ReadLength(r)
|
||||
c.userData, err = core.ReadBytes(dataLen, r)
|
||||
return c, err
|
||||
}
|
||||
|
||||
type MCSChannelInfo struct {
|
||||
ID uint16
|
||||
Name string
|
||||
}
|
||||
|
||||
type MCS struct {
|
||||
emission.Emitter
|
||||
transport core.Transport
|
||||
recvOpCode MCSDomainPDU
|
||||
sendOpCode MCSDomainPDU
|
||||
channels []MCSChannelInfo
|
||||
}
|
||||
|
||||
func NewMCS(t core.Transport, recvOpCode MCSDomainPDU, sendOpCode MCSDomainPDU) *MCS {
|
||||
m := &MCS{
|
||||
*emission.NewEmitter(),
|
||||
t,
|
||||
recvOpCode,
|
||||
sendOpCode,
|
||||
[]MCSChannelInfo{{MCS_GLOBAL_CHANNEL_ID, GLOBAL_CHANNEL_NAME}},
|
||||
}
|
||||
|
||||
m.transport.On("close", func() {
|
||||
m.Emit("close")
|
||||
}).On("error", func(err error) {
|
||||
m.Emit("error", err)
|
||||
})
|
||||
return m
|
||||
}
|
||||
|
||||
func (x *MCS) Read(b []byte) (n int, err error) {
|
||||
return x.transport.Read(b)
|
||||
}
|
||||
|
||||
func (x *MCS) Write(b []byte) (n int, err error) {
|
||||
return x.transport.Write(b)
|
||||
}
|
||||
|
||||
func (m *MCS) Close() error {
|
||||
return m.transport.Close()
|
||||
}
|
||||
|
||||
type MCSClient struct {
|
||||
*MCS
|
||||
clientCoreData *gcc.ClientCoreData
|
||||
clientNetworkData *gcc.ClientNetworkData
|
||||
clientSecurityData *gcc.ClientSecurityData
|
||||
|
||||
serverCoreData *gcc.ServerCoreData
|
||||
serverNetworkData *gcc.ServerNetworkData
|
||||
serverSecurityData *gcc.ServerSecurityData
|
||||
|
||||
channelsConnected int
|
||||
userId uint16
|
||||
nbChannelRequested int
|
||||
}
|
||||
|
||||
func NewMCSClient(t core.Transport) *MCSClient {
|
||||
c := &MCSClient{
|
||||
MCS: NewMCS(t, SEND_DATA_INDICATION, SEND_DATA_REQUEST),
|
||||
clientCoreData: gcc.NewClientCoreData(),
|
||||
clientNetworkData: gcc.NewClientNetworkData(),
|
||||
clientSecurityData: gcc.NewClientSecurityData(),
|
||||
userId: 1 + MCS_USERCHANNEL_BASE,
|
||||
}
|
||||
c.transport.On("connect", c.connect)
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *MCSClient) SetClientCoreData(width, height uint16) {
|
||||
c.clientCoreData.DesktopWidth = width
|
||||
c.clientCoreData.DesktopHeight = height
|
||||
}
|
||||
|
||||
func (c *MCSClient) connect(selectedProtocol uint32) {
|
||||
glog.Debug("mcs client on connect", selectedProtocol)
|
||||
c.clientCoreData.ServerSelectedProtocol = selectedProtocol
|
||||
|
||||
glog.Debugf("clientCoreData:%+v", c.clientCoreData)
|
||||
glog.Debugf("clientNetworkData:%+v", c.clientNetworkData)
|
||||
glog.Debugf("clientSecurityData:%+v", c.clientSecurityData)
|
||||
// sendConnectclientCoreDataInitial
|
||||
userDataBuff := bytes.Buffer{}
|
||||
userDataBuff.Write(c.clientCoreData.Pack())
|
||||
userDataBuff.Write(c.clientNetworkData.Pack())
|
||||
userDataBuff.Write(c.clientSecurityData.Pack())
|
||||
|
||||
ccReq := gcc.MakeConferenceCreateRequest(userDataBuff.Bytes())
|
||||
connectInitial := NewConnectInitial(ccReq)
|
||||
connectInitialBerEncoded := connectInitial.BER()
|
||||
|
||||
dataBuff := &bytes.Buffer{}
|
||||
ber.WriteApplicationTag(uint8(MCS_TYPE_CONNECT_INITIAL), len(connectInitialBerEncoded), dataBuff)
|
||||
dataBuff.Write(connectInitialBerEncoded)
|
||||
|
||||
_, err := c.transport.Write(dataBuff.Bytes())
|
||||
if err != nil {
|
||||
c.Emit("error", errors.New(fmt.Sprintf("mcs sendConnectInitial write error %v", err)))
|
||||
return
|
||||
}
|
||||
glog.Debug("mcs wait for data event")
|
||||
c.transport.Once("data", c.recvConnectResponse)
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvConnectResponse(s []byte) {
|
||||
glog.Debug("mcs recvConnectResponse", hex.EncodeToString(s))
|
||||
cResp, err := ReadConnectResponse(bytes.NewReader(s))
|
||||
if err != nil {
|
||||
c.Emit("error", errors.New(fmt.Sprintf("ReadConnectResponse %v", err)))
|
||||
return
|
||||
}
|
||||
// record server gcc block
|
||||
serverSettings := gcc.ReadConferenceCreateResponse(cResp.userData)
|
||||
for _, v := range serverSettings {
|
||||
switch v.(type) {
|
||||
case *gcc.ServerSecurityData:
|
||||
c.serverSecurityData = v.(*gcc.ServerSecurityData)
|
||||
|
||||
case *gcc.ServerCoreData:
|
||||
c.serverCoreData = v.(*gcc.ServerCoreData)
|
||||
|
||||
case *gcc.ServerNetworkData:
|
||||
c.serverNetworkData = v.(*gcc.ServerNetworkData)
|
||||
|
||||
default:
|
||||
err := errors.New(fmt.Sprintf("unhandle server gcc block %v", reflect.TypeOf(v)))
|
||||
glog.Error(err)
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
glog.Debugf("serverSecurityData: %+v", c.serverSecurityData)
|
||||
glog.Debugf("serverCoreData: %+v", c.serverCoreData)
|
||||
glog.Info("version", c.serverCoreData.RdpVersion, c.serverCoreData.ClientRequestedProtocol)
|
||||
glog.Debugf("serverNetworkData: %+v", c.serverNetworkData)
|
||||
glog.Debug("mcs sendErectDomainRequest")
|
||||
c.sendErectDomainRequest()
|
||||
|
||||
glog.Debug("mcs sendAttachUserRequest")
|
||||
c.sendAttachUserRequest()
|
||||
|
||||
c.transport.Once("data", c.recvAttachUserConfirm)
|
||||
}
|
||||
|
||||
func (c *MCSClient) sendErectDomainRequest() {
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(ERECT_DOMAIN_REQUEST, 0, buff)
|
||||
per.WriteInteger(0, buff)
|
||||
per.WriteInteger(0, buff)
|
||||
c.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *MCSClient) sendAttachUserRequest() {
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(ATTACH_USER_REQUEST, 0, buff)
|
||||
c.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvAttachUserConfirm(s []byte) {
|
||||
glog.Debug("mcs recvAttachUserConfirm", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
|
||||
option, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
|
||||
if !readMCSPDUHeader(option, ATTACH_USER_CONFIRM) {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_BAD_HEADER"))
|
||||
return
|
||||
}
|
||||
|
||||
e, err := per.ReadEnumerates(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
if e != 0 {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_SERVER_REJECT_USER'"))
|
||||
return
|
||||
}
|
||||
|
||||
userId, _ := per.ReadInteger16(r)
|
||||
userId += MCS_USERCHANNEL_BASE
|
||||
c.userId = userId
|
||||
|
||||
c.channels = append(c.channels, MCSChannelInfo{userId, "user"})
|
||||
c.connectChannels()
|
||||
}
|
||||
|
||||
func (c *MCSClient) connectChannels() {
|
||||
glog.Debug("mcs connectChannels:", c.channelsConnected, ":", len(c.channels))
|
||||
if c.channelsConnected == len(c.channels) && c.serverNetworkData != nil {
|
||||
if c.nbChannelRequested < int(c.serverNetworkData.ChannelCount) {
|
||||
//static virtual channel
|
||||
chanId := c.serverNetworkData.ChannelIdArray[c.nbChannelRequested]
|
||||
c.nbChannelRequested++
|
||||
c.sendChannelJoinRequest(chanId)
|
||||
c.transport.Once("data", c.recvChannelJoinConfirm)
|
||||
return
|
||||
}
|
||||
c.transport.On("data", c.recvData)
|
||||
// send client and sever gcc informations callback to sec
|
||||
clientData := make([]interface{}, 0)
|
||||
clientData = append(clientData, c.clientCoreData)
|
||||
clientData = append(clientData, c.clientSecurityData)
|
||||
clientData = append(clientData, c.clientNetworkData)
|
||||
|
||||
serverData := make([]interface{}, 0)
|
||||
serverData = append(serverData, c.serverCoreData)
|
||||
serverData = append(serverData, c.serverSecurityData)
|
||||
glog.Debug("msc connectChannels callback to sec")
|
||||
c.Emit("connect", clientData, serverData, c.userId, c.channels)
|
||||
return
|
||||
}
|
||||
|
||||
// sendChannelJoinRequest
|
||||
glog.Debug("sendChannelJoinRequest:", c.channels[c.channelsConnected].Name)
|
||||
c.sendChannelJoinRequest(c.channels[c.channelsConnected].ID)
|
||||
|
||||
c.transport.Once("data", c.recvChannelJoinConfirm)
|
||||
}
|
||||
|
||||
func (c *MCSClient) sendChannelJoinRequest(channelId uint16) {
|
||||
glog.Debug("mcs sendChannelJoinRequest", channelId)
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(CHANNEL_JOIN_REQUEST, 0, buff)
|
||||
per.WriteInteger16(c.userId-MCS_USERCHANNEL_BASE, buff)
|
||||
per.WriteInteger16(channelId, buff)
|
||||
c.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvData(s []byte) {
|
||||
glog.Debug("msc on data recvData:", hex.EncodeToString(s))
|
||||
|
||||
r := bytes.NewReader(s)
|
||||
option, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
|
||||
if readMCSPDUHeader(option, DISCONNECT_PROVIDER_ULTIMATUM) {
|
||||
c.Emit("error", errors.New("MCS DISCONNECT_PROVIDER_ULTIMATUM"))
|
||||
c.transport.Close()
|
||||
return
|
||||
} else if !readMCSPDUHeader(option, c.recvOpCode) {
|
||||
c.Emit("error", errors.New("Invalid expected MCS opcode receive data"))
|
||||
return
|
||||
}
|
||||
|
||||
userId, _ := per.ReadInteger16(r)
|
||||
userId += MCS_USERCHANNEL_BASE
|
||||
|
||||
channelId, _ := per.ReadInteger16(r)
|
||||
per.ReadEnumerates(r)
|
||||
size, _ := per.ReadLength(r)
|
||||
// channel ID doesn't match a requested layer
|
||||
found := false
|
||||
channelName := ""
|
||||
for _, channel := range c.channels {
|
||||
if channel.ID == channelId {
|
||||
found = true
|
||||
channelName = channel.Name
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
glog.Error("mcs receive data for an unconnected layer")
|
||||
return
|
||||
}
|
||||
left, err := core.ReadBytes(int(size), r)
|
||||
if err != nil {
|
||||
c.Emit("error", errors.New(fmt.Sprintf("mcs recvData get data error %v", err)))
|
||||
return
|
||||
}
|
||||
glog.Debugf("mcs emit channel<%s>:%v", channelName, left)
|
||||
c.Emit("sec", channelName, left)
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvChannelJoinConfirm(s []byte) {
|
||||
glog.Debug("mcs recvChannelJoinConfirm", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
option, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
|
||||
if !readMCSPDUHeader(option, CHANNEL_JOIN_CONFIRM) {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_WAIT_CHANNEL_JOIN_CONFIRM"))
|
||||
return
|
||||
}
|
||||
|
||||
confirm, _ := per.ReadEnumerates(r)
|
||||
userId, _ := per.ReadInteger16(r)
|
||||
userId += MCS_USERCHANNEL_BASE
|
||||
|
||||
if c.userId != userId {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_INVALID_USER_ID"))
|
||||
return
|
||||
}
|
||||
|
||||
channelId, _ := per.ReadInteger16(r)
|
||||
if (confirm != 0) && (channelId == uint16(MCS_GLOBAL_CHANNEL_ID) || channelId == c.userId) {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_SERVER_MUST_CONFIRM_STATIC_CHANNEL"))
|
||||
return
|
||||
}
|
||||
glog.Debug("Confirm channelId:", channelId)
|
||||
if confirm == 0 && c.serverNetworkData != nil {
|
||||
for i := 0; i < int(c.serverNetworkData.ChannelCount); i++ {
|
||||
if channelId == c.serverNetworkData.ChannelIdArray[i] {
|
||||
var t MCSChannelInfo
|
||||
t.ID = channelId
|
||||
t.Name = string(c.clientNetworkData.ChannelDefArray[i].Name[:])
|
||||
c.channels = append(c.channels, t)
|
||||
}
|
||||
}
|
||||
}
|
||||
c.channelsConnected++
|
||||
c.connectChannels()
|
||||
}
|
||||
|
||||
func (c *MCSClient) Pack(data []byte, channelId uint16) []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(c.sendOpCode, 0, buff)
|
||||
per.WriteInteger16(c.userId-MCS_USERCHANNEL_BASE, buff)
|
||||
per.WriteInteger16(channelId, buff)
|
||||
core.WriteUInt8(0x70, buff)
|
||||
per.WriteLength(len(data), buff)
|
||||
core.WriteBytes(data, buff)
|
||||
glog.Debug("MCSClient write", channelId, ":", hex.EncodeToString(buff.Bytes()))
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
func (c *MCSClient) Write(data []byte) (n int, err error) {
|
||||
data = c.Pack(data, c.channels[0].ID)
|
||||
return c.transport.Write(data)
|
||||
}
|
||||
|
||||
func (c *MCSClient) SendToChannel(channel string, data []byte) (n int, err error) {
|
||||
channelId := c.channels[0].ID
|
||||
for _, ch := range c.channels {
|
||||
if channel == ch.Name {
|
||||
channelId = ch.ID
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
data = c.Pack(data, channelId)
|
||||
return c.transport.Write(data)
|
||||
}
|
||||
@@ -0,0 +1,574 @@
|
||||
package t125
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"reflect"
|
||||
|
||||
"github.com/xxx/wscan/mylib/grdp/plugin/rail"
|
||||
|
||||
"github.com/xxx/wscan/mylib/grdp/plugin/drdynvc"
|
||||
|
||||
"github.com/xxx/wscan/mylib/grdp/core"
|
||||
"github.com/xxx/wscan/mylib/grdp/emission"
|
||||
"github.com/xxx/wscan/mylib/grdp/glog"
|
||||
"github.com/xxx/wscan/mylib/grdp/protocol/t125/ber"
|
||||
"github.com/xxx/wscan/mylib/grdp/protocol/t125/gcc"
|
||||
"github.com/xxx/wscan/mylib/grdp/protocol/t125/per"
|
||||
)
|
||||
|
||||
// take idea from https://github.com/Madnikulin50/gordp
|
||||
|
||||
// Multiple Channel Service layer
|
||||
|
||||
type MCSMessage uint8
|
||||
|
||||
const (
|
||||
MCS_TYPE_CONNECT_INITIAL MCSMessage = 0x65
|
||||
MCS_TYPE_CONNECT_RESPONSE = 0x66
|
||||
)
|
||||
|
||||
type MCSDomainPDU uint16
|
||||
|
||||
const (
|
||||
ERECT_DOMAIN_REQUEST MCSDomainPDU = 1
|
||||
DISCONNECT_PROVIDER_ULTIMATUM = 8
|
||||
ATTACH_USER_REQUEST = 10
|
||||
ATTACH_USER_CONFIRM = 11
|
||||
CHANNEL_JOIN_REQUEST = 14
|
||||
CHANNEL_JOIN_CONFIRM = 15
|
||||
SEND_DATA_REQUEST = 25
|
||||
SEND_DATA_INDICATION = 26
|
||||
)
|
||||
|
||||
const (
|
||||
MCS_GLOBAL_CHANNEL_ID uint16 = 1003
|
||||
MCS_USERCHANNEL_BASE = 1001
|
||||
)
|
||||
|
||||
const (
|
||||
GLOBAL_CHANNEL_NAME = "global"
|
||||
)
|
||||
|
||||
/**
|
||||
* Format MCS PDULayer header packet
|
||||
* @param mcsPdu {integer}
|
||||
* @param options {integer}
|
||||
* @returns {type.UInt8} headers
|
||||
*/
|
||||
func writeMCSPDUHeader(mcsPdu MCSDomainPDU, options uint8, w io.Writer) {
|
||||
core.WriteUInt8((uint8(mcsPdu)<<2)|options, w)
|
||||
}
|
||||
|
||||
func readMCSPDUHeader(options uint8, mcsPdu MCSDomainPDU) bool {
|
||||
return (options >> 2) == uint8(mcsPdu)
|
||||
}
|
||||
|
||||
type DomainParameters struct {
|
||||
MaxChannelIds int
|
||||
MaxUserIds int
|
||||
MaxTokenIds int
|
||||
NumPriorities int
|
||||
MinThoughput int
|
||||
MaxHeight int
|
||||
MaxMCSPDUsize int
|
||||
ProtocolVersion int
|
||||
}
|
||||
|
||||
/**
|
||||
* @see http://www.itu.int/rec/T-REC-T.125-199802-I/en page 25
|
||||
* @returns {asn1.univ.Sequence}
|
||||
*/
|
||||
func NewDomainParameters(
|
||||
maxChannelIds int,
|
||||
maxUserIds int,
|
||||
maxTokenIds int,
|
||||
numPriorities int,
|
||||
minThoughput int,
|
||||
maxHeight int,
|
||||
maxMCSPDUsize int,
|
||||
protocolVersion int) *DomainParameters {
|
||||
return &DomainParameters{maxChannelIds, maxUserIds, maxTokenIds,
|
||||
numPriorities, minThoughput, maxHeight, maxMCSPDUsize, protocolVersion}
|
||||
}
|
||||
|
||||
func (d *DomainParameters) BER() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
ber.WriteInteger(d.MaxChannelIds, buff)
|
||||
ber.WriteInteger(d.MaxUserIds, buff)
|
||||
ber.WriteInteger(d.MaxTokenIds, buff)
|
||||
ber.WriteInteger(1, buff)
|
||||
ber.WriteInteger(0, buff)
|
||||
ber.WriteInteger(1, buff)
|
||||
ber.WriteInteger(d.MaxMCSPDUsize, buff)
|
||||
ber.WriteInteger(2, buff)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
func ReadDomainParameters(r io.Reader) (*DomainParameters, error) {
|
||||
if !ber.ReadUniversalTag(ber.TAG_SEQUENCE, true, r) {
|
||||
return nil, errors.New("bad BER tags")
|
||||
}
|
||||
d := &DomainParameters{}
|
||||
ber.ReadLength(r)
|
||||
|
||||
d.MaxChannelIds, _ = ber.ReadInteger(r)
|
||||
d.MaxUserIds, _ = ber.ReadInteger(r)
|
||||
d.MaxTokenIds, _ = ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
d.MaxMCSPDUsize, _ = ber.ReadInteger(r)
|
||||
ber.ReadInteger(r)
|
||||
return d, nil
|
||||
}
|
||||
|
||||
/**
|
||||
* @see http://www.itu.int/rec/T-REC-T.125-199802-I/en page 25
|
||||
* @param userData {Buffer}
|
||||
* @returns {asn1.univ.Sequence}
|
||||
*/
|
||||
type ConnectInitial struct {
|
||||
CallingDomainSelector []byte
|
||||
CalledDomainSelector []byte
|
||||
UpwardFlag bool
|
||||
TargetParameters DomainParameters
|
||||
MinimumParameters DomainParameters
|
||||
MaximumParameters DomainParameters
|
||||
UserData []byte
|
||||
}
|
||||
|
||||
func NewConnectInitial(userData []byte) ConnectInitial {
|
||||
return ConnectInitial{[]byte{0x1},
|
||||
[]byte{0x1},
|
||||
true,
|
||||
*NewDomainParameters(34, 2, 0, 1, 0, 1, 0xffff, 2),
|
||||
*NewDomainParameters(1, 1, 1, 1, 0, 1, 0x420, 2),
|
||||
*NewDomainParameters(0xffff, 0xfc17, 0xffff, 1, 0, 1, 0xffff, 2),
|
||||
userData}
|
||||
}
|
||||
|
||||
func (c *ConnectInitial) BER() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
ber.WriteOctetstring(string(c.CallingDomainSelector), buff)
|
||||
ber.WriteOctetstring(string(c.CalledDomainSelector), buff)
|
||||
ber.WriteBoolean(c.UpwardFlag, buff)
|
||||
ber.WriteEncodedDomainParams(c.TargetParameters.BER(), buff)
|
||||
ber.WriteEncodedDomainParams(c.MinimumParameters.BER(), buff)
|
||||
ber.WriteEncodedDomainParams(c.MaximumParameters.BER(), buff)
|
||||
ber.WriteOctetstring(string(c.UserData), buff)
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
/**
|
||||
* @see http://www.itu.int/rec/T-REC-T.125-199802-I/en page 25
|
||||
* @returns {asn1.univ.Sequence}
|
||||
*/
|
||||
|
||||
type ConnectResponse struct {
|
||||
result uint8
|
||||
calledConnectId int
|
||||
domainParameters *DomainParameters
|
||||
userData []byte
|
||||
}
|
||||
|
||||
func NewConnectResponse(userData []byte) *ConnectResponse {
|
||||
return &ConnectResponse{0,
|
||||
0,
|
||||
NewDomainParameters(22, 3, 0, 1, 0, 1, 0xfff8, 2),
|
||||
userData}
|
||||
}
|
||||
|
||||
func ReadConnectResponse(r io.Reader) (*ConnectResponse, error) {
|
||||
c := &ConnectResponse{}
|
||||
var err error
|
||||
_, err = ber.ReadApplicationTag(MCS_TYPE_CONNECT_RESPONSE, r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
c.result, err = ber.ReadEnumerated(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c.calledConnectId, err = ber.ReadInteger(r)
|
||||
c.domainParameters, err = ReadDomainParameters(r)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !ber.ReadUniversalTag(ber.TAG_OCTET_STRING, false, r) {
|
||||
return nil, errors.New("invalid expected BER tag")
|
||||
}
|
||||
dataLen, _ := ber.ReadLength(r)
|
||||
c.userData, err = core.ReadBytes(dataLen, r)
|
||||
return c, err
|
||||
}
|
||||
|
||||
type MCSChannelInfo struct {
|
||||
ID uint16
|
||||
Name string
|
||||
}
|
||||
|
||||
type MCS struct {
|
||||
emission.Emitter
|
||||
transport core.Transport
|
||||
recvOpCode MCSDomainPDU
|
||||
sendOpCode MCSDomainPDU
|
||||
channels []MCSChannelInfo
|
||||
}
|
||||
|
||||
func NewMCS(t core.Transport, recvOpCode MCSDomainPDU, sendOpCode MCSDomainPDU) *MCS {
|
||||
m := &MCS{
|
||||
*emission.NewEmitter(),
|
||||
t,
|
||||
recvOpCode,
|
||||
sendOpCode,
|
||||
[]MCSChannelInfo{{MCS_GLOBAL_CHANNEL_ID, GLOBAL_CHANNEL_NAME}},
|
||||
}
|
||||
|
||||
m.transport.On("close", func() {
|
||||
m.Emit("close")
|
||||
}).On("error", func(err error) {
|
||||
m.Emit("error", err)
|
||||
})
|
||||
return m
|
||||
}
|
||||
|
||||
func (x *MCS) Read(b []byte) (n int, err error) {
|
||||
return x.transport.Read(b)
|
||||
}
|
||||
|
||||
func (x *MCS) Write(b []byte) (n int, err error) {
|
||||
return x.transport.Write(b)
|
||||
}
|
||||
|
||||
func (m *MCS) Close() error {
|
||||
return m.transport.Close()
|
||||
}
|
||||
|
||||
type MCSClient struct {
|
||||
*MCS
|
||||
clientCoreData *gcc.ClientCoreData
|
||||
clientNetworkData *gcc.ClientNetworkData
|
||||
clientSecurityData *gcc.ClientSecurityData
|
||||
|
||||
serverCoreData *gcc.ServerCoreData
|
||||
serverNetworkData *gcc.ServerNetworkData
|
||||
serverSecurityData *gcc.ServerSecurityData
|
||||
|
||||
channelsConnected int
|
||||
userId uint16
|
||||
nbChannelRequested int
|
||||
}
|
||||
|
||||
func NewMCSClient(t core.Transport) *MCSClient {
|
||||
c := &MCSClient{
|
||||
MCS: NewMCS(t, SEND_DATA_INDICATION, SEND_DATA_REQUEST),
|
||||
clientCoreData: gcc.NewClientCoreData(),
|
||||
clientNetworkData: gcc.NewClientNetworkData(),
|
||||
clientSecurityData: gcc.NewClientSecurityData(),
|
||||
userId: 1 + MCS_USERCHANNEL_BASE,
|
||||
}
|
||||
c.transport.On("connect", c.connect)
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *MCSClient) SetClientDesktop(width, height uint16) {
|
||||
c.clientCoreData.DesktopWidth = width
|
||||
c.clientCoreData.DesktopHeight = height
|
||||
}
|
||||
|
||||
func (c *MCSClient) SetClientDynvcProtocol() {
|
||||
c.clientCoreData.EarlyCapabilityFlags = gcc.RNS_UD_CS_SUPPORT_DYNVC_GFX_PROTOCOL
|
||||
c.clientNetworkData.AddVirtualChannel(drdynvc.ChannelName, drdynvc.ChannelOption)
|
||||
}
|
||||
|
||||
func (c *MCSClient) SetClientRemoteProgram() {
|
||||
c.clientNetworkData.AddVirtualChannel(rail.ChannelName, rail.ChannelOption)
|
||||
}
|
||||
|
||||
func (c *MCSClient) SetClientCliprdr() {
|
||||
//c.clientNetworkData.AddVirtualChannel(cliprdr.ChannelName, cliprdr.ChannelOption)
|
||||
}
|
||||
|
||||
func (c *MCSClient) connect(selectedProtocol uint32) {
|
||||
glog.Debug("mcs client on connect", selectedProtocol)
|
||||
c.clientCoreData.ServerSelectedProtocol = selectedProtocol
|
||||
|
||||
glog.Debugf("clientCoreData:%+v", c.clientCoreData)
|
||||
glog.Debugf("clientNetworkData:%+v", c.clientNetworkData)
|
||||
glog.Debugf("clientSecurityData:%+v", c.clientSecurityData)
|
||||
// sendConnectclientCoreDataInitial
|
||||
userDataBuff := bytes.Buffer{}
|
||||
userDataBuff.Write(c.clientCoreData.Pack())
|
||||
userDataBuff.Write(c.clientNetworkData.Pack())
|
||||
userDataBuff.Write(c.clientSecurityData.Pack())
|
||||
|
||||
ccReq := gcc.MakeConferenceCreateRequest(userDataBuff.Bytes())
|
||||
connectInitial := NewConnectInitial(ccReq)
|
||||
connectInitialBerEncoded := connectInitial.BER()
|
||||
|
||||
dataBuff := &bytes.Buffer{}
|
||||
ber.WriteApplicationTag(uint8(MCS_TYPE_CONNECT_INITIAL), len(connectInitialBerEncoded), dataBuff)
|
||||
dataBuff.Write(connectInitialBerEncoded)
|
||||
|
||||
_, err := c.transport.Write(dataBuff.Bytes())
|
||||
if err != nil {
|
||||
c.Emit("error", errors.New(fmt.Sprintf("mcs sendConnectInitial write error %v", err)))
|
||||
return
|
||||
}
|
||||
glog.Debug("mcs wait for data event")
|
||||
c.transport.Once("data", c.recvConnectResponse)
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvConnectResponse(s []byte) {
|
||||
glog.Trace("mcs recvConnectResponse", hex.EncodeToString(s))
|
||||
cResp, err := ReadConnectResponse(bytes.NewReader(s))
|
||||
if err != nil {
|
||||
c.Emit("error", errors.New(fmt.Sprintf("ReadConnectResponse %v", err)))
|
||||
return
|
||||
}
|
||||
// record server gcc block
|
||||
serverSettings := gcc.ReadConferenceCreateResponse(cResp.userData)
|
||||
for _, v := range serverSettings {
|
||||
switch v.(type) {
|
||||
case *gcc.ServerSecurityData:
|
||||
c.serverSecurityData = v.(*gcc.ServerSecurityData)
|
||||
|
||||
case *gcc.ServerCoreData:
|
||||
c.serverCoreData = v.(*gcc.ServerCoreData)
|
||||
|
||||
case *gcc.ServerNetworkData:
|
||||
c.serverNetworkData = v.(*gcc.ServerNetworkData)
|
||||
|
||||
default:
|
||||
err := errors.New(fmt.Sprintf("unhandle server gcc block %v", reflect.TypeOf(v)))
|
||||
glog.Error(err)
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
glog.Debugf("serverSecurityData: %+v", c.serverSecurityData)
|
||||
glog.Debugf("serverCoreData: %+v", c.serverCoreData)
|
||||
glog.Debugf("serverNetworkData: %+v", c.serverNetworkData)
|
||||
glog.Debug("mcs sendErectDomainRequest")
|
||||
c.sendErectDomainRequest()
|
||||
|
||||
glog.Debug("mcs sendAttachUserRequest")
|
||||
c.sendAttachUserRequest()
|
||||
|
||||
c.transport.Once("data", c.recvAttachUserConfirm)
|
||||
}
|
||||
|
||||
func (c *MCSClient) sendErectDomainRequest() {
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(ERECT_DOMAIN_REQUEST, 0, buff)
|
||||
per.WriteInteger(0, buff)
|
||||
per.WriteInteger(0, buff)
|
||||
c.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *MCSClient) sendAttachUserRequest() {
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(ATTACH_USER_REQUEST, 0, buff)
|
||||
c.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvAttachUserConfirm(s []byte) {
|
||||
glog.Debug("mcs recvAttachUserConfirm", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
|
||||
option, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
|
||||
if !readMCSPDUHeader(option, ATTACH_USER_CONFIRM) {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_BAD_HEADER"))
|
||||
return
|
||||
}
|
||||
|
||||
e, err := per.ReadEnumerates(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
if e != 0 {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_SERVER_REJECT_USER'"))
|
||||
return
|
||||
}
|
||||
|
||||
userId, _ := per.ReadInteger16(r)
|
||||
userId += MCS_USERCHANNEL_BASE
|
||||
c.userId = userId
|
||||
|
||||
c.channels = append(c.channels, MCSChannelInfo{userId, "user"})
|
||||
c.connectChannels()
|
||||
}
|
||||
|
||||
func (c *MCSClient) connectChannels() {
|
||||
glog.Debug("mcs connectChannels:", c.channelsConnected, ":", len(c.channels))
|
||||
if c.channelsConnected == len(c.channels) {
|
||||
if c.nbChannelRequested < int(c.serverNetworkData.ChannelCount) {
|
||||
//static virtual channel
|
||||
chanId := c.serverNetworkData.ChannelIdArray[c.nbChannelRequested]
|
||||
c.nbChannelRequested++
|
||||
c.sendChannelJoinRequest(chanId)
|
||||
c.transport.Once("data", c.recvChannelJoinConfirm)
|
||||
return
|
||||
}
|
||||
c.transport.On("data", c.recvData)
|
||||
// send client and sever gcc informations callback to sec
|
||||
clientData := make([]interface{}, 0)
|
||||
clientData = append(clientData, c.clientCoreData)
|
||||
clientData = append(clientData, c.clientSecurityData)
|
||||
clientData = append(clientData, c.clientNetworkData)
|
||||
|
||||
serverData := make([]interface{}, 0)
|
||||
serverData = append(serverData, c.serverCoreData)
|
||||
serverData = append(serverData, c.serverSecurityData)
|
||||
glog.Debug("msc connectChannels callback to sec")
|
||||
c.Emit("connect", clientData, serverData, c.userId, c.channels)
|
||||
return
|
||||
}
|
||||
|
||||
// sendChannelJoinRequest
|
||||
glog.Debug("sendChannelJoinRequest:", c.channels[c.channelsConnected].Name)
|
||||
c.sendChannelJoinRequest(c.channels[c.channelsConnected].ID)
|
||||
|
||||
c.transport.Once("data", c.recvChannelJoinConfirm)
|
||||
}
|
||||
|
||||
func (c *MCSClient) sendChannelJoinRequest(channelId uint16) {
|
||||
glog.Debug("mcs sendChannelJoinRequest", channelId)
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(CHANNEL_JOIN_REQUEST, 0, buff)
|
||||
per.WriteInteger16(c.userId-MCS_USERCHANNEL_BASE, buff)
|
||||
per.WriteInteger16(channelId, buff)
|
||||
c.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvData(s []byte) {
|
||||
glog.Trace("msc on data recvData:", hex.EncodeToString(s))
|
||||
|
||||
r := bytes.NewReader(s)
|
||||
option, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
|
||||
if readMCSPDUHeader(option, DISCONNECT_PROVIDER_ULTIMATUM) {
|
||||
c.Emit("error", errors.New("MCS DISCONNECT_PROVIDER_ULTIMATUM"))
|
||||
c.transport.Close()
|
||||
return
|
||||
} else if !readMCSPDUHeader(option, c.recvOpCode) {
|
||||
c.Emit("error", errors.New("Invalid expected MCS opcode receive data"))
|
||||
return
|
||||
}
|
||||
|
||||
userId, _ := per.ReadInteger16(r)
|
||||
userId += MCS_USERCHANNEL_BASE
|
||||
|
||||
channelId, _ := per.ReadInteger16(r)
|
||||
per.ReadEnumerates(r)
|
||||
size, _ := per.ReadLength(r)
|
||||
// channel ID doesn't match a requested layer
|
||||
found := false
|
||||
channelName := ""
|
||||
for _, channel := range c.channels {
|
||||
if channel.ID == channelId {
|
||||
found = true
|
||||
channelName = channel.Name
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
glog.Error("mcs receive data for an unconnected layer")
|
||||
return
|
||||
}
|
||||
left, err := core.ReadBytes(int(size), r)
|
||||
if err != nil {
|
||||
c.Emit("error", errors.New(fmt.Sprintf("mcs recvData get data error %v", err)))
|
||||
return
|
||||
}
|
||||
glog.Debugf("mcs emit channel<%s>", channelName)
|
||||
c.Emit("sec", channelName, left)
|
||||
}
|
||||
|
||||
func (c *MCSClient) recvChannelJoinConfirm(s []byte) {
|
||||
glog.Debug("mcs recvChannelJoinConfirm", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
option, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
c.Emit("error", err)
|
||||
return
|
||||
}
|
||||
|
||||
if !readMCSPDUHeader(option, CHANNEL_JOIN_CONFIRM) {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_WAIT_CHANNEL_JOIN_CONFIRM"))
|
||||
return
|
||||
}
|
||||
|
||||
confirm, _ := per.ReadEnumerates(r)
|
||||
userId, _ := per.ReadInteger16(r)
|
||||
userId += MCS_USERCHANNEL_BASE
|
||||
|
||||
if c.userId != userId {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_INVALID_USER_ID"))
|
||||
return
|
||||
}
|
||||
|
||||
channelId, _ := per.ReadInteger16(r)
|
||||
if (confirm != 0) && (channelId == uint16(MCS_GLOBAL_CHANNEL_ID) || channelId == c.userId) {
|
||||
c.Emit("error", errors.New("NODE_RDP_PROTOCOL_T125_MCS_SERVER_MUST_CONFIRM_STATIC_CHANNEL"))
|
||||
return
|
||||
}
|
||||
glog.Debug("Confirm channelId:", channelId)
|
||||
if confirm == 0 {
|
||||
for i := 0; i < int(c.serverNetworkData.ChannelCount); i++ {
|
||||
if channelId == c.serverNetworkData.ChannelIdArray[i] {
|
||||
var t MCSChannelInfo
|
||||
t.ID = channelId
|
||||
t.Name = string(c.clientNetworkData.ChannelDefArray[i].Name[:])
|
||||
c.channels = append(c.channels, t)
|
||||
}
|
||||
}
|
||||
}
|
||||
c.channelsConnected++
|
||||
c.connectChannels()
|
||||
}
|
||||
|
||||
func (c *MCSClient) Pack(data []byte, channelId uint16) []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
writeMCSPDUHeader(c.sendOpCode, 0, buff)
|
||||
per.WriteInteger16(c.userId-MCS_USERCHANNEL_BASE, buff)
|
||||
per.WriteInteger16(channelId, buff)
|
||||
core.WriteUInt8(0x70, buff)
|
||||
per.WriteLength(len(data), buff)
|
||||
core.WriteBytes(data, buff)
|
||||
glog.Trace("MCSClient write", channelId, ":", hex.EncodeToString(buff.Bytes()))
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
func (c *MCSClient) Write(data []byte) (n int, err error) {
|
||||
data = c.Pack(data, c.channels[0].ID)
|
||||
return c.transport.Write(data)
|
||||
}
|
||||
|
||||
func (c *MCSClient) SendToChannel(channel string, data []byte) (n int, err error) {
|
||||
channelId := c.channels[0].ID
|
||||
for _, ch := range c.channels {
|
||||
if channel == ch.Name {
|
||||
channelId = ch.ID
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
data = c.Pack(data, channelId)
|
||||
return c.transport.Write(data)
|
||||
}
|
||||
@@ -0,0 +1,203 @@
|
||||
package per
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
)
|
||||
|
||||
func ReadEnumerates(r io.Reader) (uint8, error) {
|
||||
return core.ReadUInt8(r)
|
||||
}
|
||||
|
||||
func WriteInteger(n int, w io.Writer) {
|
||||
if n <= 0xff {
|
||||
WriteLength(1, w)
|
||||
core.WriteUInt8(uint8(n), w)
|
||||
} else if n <= 0xffff {
|
||||
WriteLength(2, w)
|
||||
core.WriteUInt16BE(uint16(n), w)
|
||||
} else {
|
||||
WriteLength(4, w)
|
||||
core.WriteUInt32BE(uint32(n), w)
|
||||
}
|
||||
}
|
||||
|
||||
func ReadInteger16(r io.Reader) (uint16, error) {
|
||||
return core.ReadUint16BE(r)
|
||||
}
|
||||
|
||||
func WriteInteger16(value uint16, w io.Writer) {
|
||||
core.WriteUInt16BE(value, w)
|
||||
}
|
||||
|
||||
/**
|
||||
* @param choice {integer}
|
||||
* @returns {type.UInt8} choice per encoded
|
||||
*/
|
||||
func WriteChoice(choice uint8, w io.Writer) {
|
||||
core.WriteUInt8(choice, w)
|
||||
}
|
||||
|
||||
/**
|
||||
* @param value {raw} value to convert to per format
|
||||
* @returns type objects per encoding value
|
||||
*/
|
||||
func WriteLength(value int, w io.Writer) {
|
||||
if value > 0x7f {
|
||||
core.WriteUInt16BE(uint16(value|0x8000), w)
|
||||
} else {
|
||||
core.WriteUInt8(uint8(value), w)
|
||||
}
|
||||
}
|
||||
|
||||
func ReadLength(r io.Reader) (uint16, error) {
|
||||
b, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
return 0, nil
|
||||
}
|
||||
var size uint16
|
||||
if b&0x80 > 0 {
|
||||
b = b &^ 0x80
|
||||
size = uint16(b) << 8
|
||||
left, _ := core.ReadUInt8(r)
|
||||
size += uint16(left)
|
||||
} else {
|
||||
size = uint16(b)
|
||||
}
|
||||
return size, nil
|
||||
}
|
||||
|
||||
/**
|
||||
* @param oid {array} oid to write
|
||||
* @returns {type.Component} per encoded object identifier
|
||||
*/
|
||||
func WriteObjectIdentifier(oid []byte, w io.Writer) {
|
||||
core.WriteUInt8(5, w)
|
||||
core.WriteByte((oid[0]<<4)&(oid[1]&0x0f), w)
|
||||
core.WriteByte(oid[2], w)
|
||||
core.WriteByte(oid[3], w)
|
||||
core.WriteByte(oid[4], w)
|
||||
core.WriteByte(oid[5], w)
|
||||
}
|
||||
|
||||
/**
|
||||
* @param selection {integer}
|
||||
* @returns {type.UInt8} per encoded selection
|
||||
*/
|
||||
func WriteSelection(selection uint8, w io.Writer) {
|
||||
core.WriteUInt8(selection, w)
|
||||
}
|
||||
|
||||
func WriteNumericString(s string, minValue int, w io.Writer) {
|
||||
length := len(s)
|
||||
mLength := minValue
|
||||
if length >= minValue {
|
||||
mLength = length - minValue
|
||||
}
|
||||
buff := &bytes.Buffer{}
|
||||
for i := 0; i < length; i += 2 {
|
||||
c1 := int(s[i])
|
||||
c2 := 0x30
|
||||
if i+1 < length {
|
||||
c2 = int(s[i+1])
|
||||
}
|
||||
c1 = (c1 - 0x30) % 10
|
||||
c2 = (c2 - 0x30) % 10
|
||||
core.WriteUInt8(uint8((c1<<4)|c2), buff)
|
||||
}
|
||||
WriteLength(mLength, w)
|
||||
w.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func WritePadding(length int, w io.Writer) {
|
||||
b := make([]byte, length)
|
||||
w.Write(b)
|
||||
}
|
||||
|
||||
func WriteNumberOfSet(n int, w io.Writer) {
|
||||
core.WriteUInt8(uint8(n), w)
|
||||
}
|
||||
|
||||
/**
|
||||
* @param oStr {String}
|
||||
* @param minValue {integer} default 0
|
||||
* @returns {type.Component} per encoded octet stream
|
||||
*/
|
||||
func WriteOctetStream(oStr string, minValue int, w io.Writer) {
|
||||
length := len(oStr)
|
||||
mlength := minValue
|
||||
|
||||
if length-minValue >= 0 {
|
||||
mlength = length - minValue
|
||||
}
|
||||
WriteLength(mlength, w)
|
||||
w.Write([]byte(oStr)[:length])
|
||||
}
|
||||
|
||||
func ReadChoice(r io.Reader) uint8 {
|
||||
choice, _ := core.ReadUInt8(r)
|
||||
return choice
|
||||
}
|
||||
func ReadNumberOfSet(r io.Reader) uint8 {
|
||||
choice, _ := core.ReadUInt8(r)
|
||||
return choice
|
||||
}
|
||||
func ReadInteger(r io.Reader) uint32 {
|
||||
size, _ := ReadLength(r)
|
||||
switch size {
|
||||
case 1:
|
||||
ret, _ := core.ReadUInt8(r)
|
||||
return uint32(ret)
|
||||
case 2:
|
||||
ret, _ := core.ReadUint16BE(r)
|
||||
return uint32(ret)
|
||||
case 4:
|
||||
ret, _ := core.ReadUInt32BE(r)
|
||||
return ret
|
||||
default:
|
||||
glog.Info("ReadInteger")
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func ReadObjectIdentifier(r io.Reader, oid []byte) bool {
|
||||
size, _ := ReadLength(r)
|
||||
if size != 5 {
|
||||
return false
|
||||
}
|
||||
|
||||
a_oid := []byte{0, 0, 0, 0, 0, 0}
|
||||
t12, _ := core.ReadByte(r)
|
||||
a_oid[0] = t12 >> 4
|
||||
a_oid[1] = t12 & 0x0f
|
||||
a_oid[2], _ = core.ReadByte(r)
|
||||
a_oid[3], _ = core.ReadByte(r)
|
||||
a_oid[4], _ = core.ReadByte(r)
|
||||
a_oid[5], _ = core.ReadByte(r)
|
||||
|
||||
for i, _ := range oid {
|
||||
if oid[i] != a_oid[i] {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
func ReadOctetStream(r io.Reader, s string, min int) bool {
|
||||
ln, _ := ReadLength(r)
|
||||
size := int(ln) + min
|
||||
if size != len(s) {
|
||||
return false
|
||||
}
|
||||
for i := 0; i < size; i++ {
|
||||
b, _ := core.ReadByte(r)
|
||||
if b != s[i] {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
@@ -0,0 +1,482 @@
|
||||
package tpkt
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/binary"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/emission"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/nla"
|
||||
)
|
||||
|
||||
// take idea from https://github.com/Madnikulin50/gordp
|
||||
|
||||
/**
|
||||
* Type of tpkt packet
|
||||
* Fastpath is use to shortcut RDP stack
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240621.aspx
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240589.aspx
|
||||
*/
|
||||
const (
|
||||
FASTPATH_ACTION_FASTPATH = 0x0
|
||||
FASTPATH_ACTION_X224 = 0x3
|
||||
)
|
||||
|
||||
/**
|
||||
* TPKT layer of rdp stack
|
||||
*/
|
||||
type TPKT struct {
|
||||
emission.Emitter
|
||||
Conn *core.SocketLayer
|
||||
ntlm *nla.NTLMv2
|
||||
secFlag byte
|
||||
lastShortLength int
|
||||
fastPathListener core.FastPathListener
|
||||
ntlmSec *nla.NTLMv2Security
|
||||
nlaAuthOnly bool // NLA仅验证模式：验证成功后立即断开，不建立会话
|
||||
}
|
||||
|
||||
var OsVersion = map[string]string{
|
||||
"3.10.511": "Windows NT 3.1",
|
||||
"3.50.807": "Windows NT 3.5",
|
||||
"3.10.528": "Windows NT 3.1, Service Pack 3",
|
||||
"3.51.1057": "Windows NT 3.51",
|
||||
"4.00.950": "Windows 95",
|
||||
"4.0.1381": "Windows NT 4.0",
|
||||
"4.10.1998": "Windows 98",
|
||||
"4.10.2222": "Windows 98 Second Edition (SE)",
|
||||
"5.0.2195": "Windows 2000",
|
||||
"4.90.3000": "Windows Me",
|
||||
"5.1.2600": "Windows XP/Windows XP, Service Pack 3",
|
||||
"5.1.2600.1105": "Windows XP, Service Pack 1",
|
||||
"5.2.3790": "Windows Server 2003/Windows Server 2003 R2/Windows Server 2003, Service Pack 2",
|
||||
"5.1.2600.2180": "Windows XP, Service Pack 2",
|
||||
"5.2.3790.1180": "Windows Server 2003, Service Pack 1",
|
||||
"6.0.6000": "Windows Vista",
|
||||
"5.2.4500": "Windows Home Server",
|
||||
"6.0.6001": "Windows Vista, Service Pack 1/Windows Server 2008",
|
||||
"6.0.6002": "Windows Vista, Service Pack 2/Windows Server 2008, Service Pack 2",
|
||||
"6.1.7600": "Windows 7/Windows Server 2008 R2",
|
||||
"6.1.7601": "Windows 7, Service Pack 1/Windows Server 2008 R2, Service Pack 1",
|
||||
"6.1.8400": "Windows Home Server 2011",
|
||||
"6.2.9200": "Windows Server 2012/Windows 8",
|
||||
"6.3.9600": "Windows 8.1/Windows Server 2012 R2",
|
||||
"10.0.10240": "Windows 10, Version 1507",
|
||||
"10.0.10586": "Windows 10, Version 1511",
|
||||
"10.0.14393": "Windows 10, Version 1607/Windows Server 2016, Version 1607",
|
||||
"10.0.15063": "Windows 10, Version 1703",
|
||||
"10.0.16299": "Windows 10, Version 1709",
|
||||
"10.0.17134": "Windows 10, Version 1803",
|
||||
"10.0.17763": "Windows Server 2019, Version 1809/Windows 10, Version 1809",
|
||||
"6.0.6003": "Windows Server 2008, Service Pack 2, Rollup KB4489887",
|
||||
"10.0.18362": "Windows 10, Version 1903",
|
||||
"10.0.18363": "Windows 10, Version 1909/Windows Server, Version 1909",
|
||||
"10.0.19041": "Windows 10, Version 2004/Windows Server, Version 2004",
|
||||
"10.0.19042": "Windows 10, Version 20H2/Windows Server, Version 20H2",
|
||||
"10.0.19043": "Windows 10, Version 21H1",
|
||||
"10.0.20348": "Windows Server 2022",
|
||||
"10.0.22000": "Windows 11, Version 21H2",
|
||||
"10.0.19044": "Windows 10, Version 21H2",
|
||||
}
|
||||
|
||||
func New(s *core.SocketLayer, ntlm *nla.NTLMv2) *TPKT {
|
||||
t := &TPKT{
|
||||
Emitter: *emission.NewEmitter(),
|
||||
Conn: s,
|
||||
secFlag: 0,
|
||||
ntlm: ntlm}
|
||||
core.StartReadBytes(2, s, t.recvHeader)
|
||||
return t
|
||||
}
|
||||
|
||||
func (t *TPKT) StartTLS() error {
|
||||
return t.Conn.StartTLS()
|
||||
}
|
||||
|
||||
// SetNLAAuthOnly 设置NLA仅验证模式
|
||||
// 启用后，NLA认证成功即返回，不发送credentials建立会话，不会挤掉已登录用户
|
||||
func (t *TPKT) SetNLAAuthOnly(authOnly bool) {
|
||||
t.nlaAuthOnly = authOnly
|
||||
}
|
||||
|
||||
func (t *TPKT) StartNLA() error {
|
||||
err := t.StartTLS()
|
||||
if err != nil {
|
||||
glog.Info("start tls failed", err)
|
||||
return err
|
||||
}
|
||||
req := nla.EncodeDERTRequest([]nla.Message{t.ntlm.GetNegotiateMessage()}, nil, nil)
|
||||
_, err = t.Conn.Write(req)
|
||||
if err != nil {
|
||||
glog.Info("send NegotiateMessage", err)
|
||||
return err
|
||||
}
|
||||
|
||||
resp := make([]byte, 1024)
|
||||
n, err := t.Conn.Read(resp)
|
||||
if err != nil {
|
||||
return fmt.Errorf("read %s", err)
|
||||
} else {
|
||||
glog.Debug("StartNLA Read success")
|
||||
}
|
||||
return t.recvChallenge(resp[:n])
|
||||
}
|
||||
|
||||
func (t *TPKT) recvChallenge(data []byte) error {
|
||||
//own add
|
||||
glog.Debug("start recv challenge......")
|
||||
info := make(map[string]any)
|
||||
type NTLMChallenge struct {
|
||||
Signature [8]byte
|
||||
MessageType uint32
|
||||
TargetNameLen uint16
|
||||
TargetNameMaxLen uint16
|
||||
TargetNameBufferOffset uint32
|
||||
NegotiateFlags uint32
|
||||
ServerChallenge uint64
|
||||
Reserved uint64
|
||||
TargetInfoLen uint16
|
||||
TargetInfoMaxLen uint16
|
||||
TargetInfoBufferOffset uint32
|
||||
Version [8]byte
|
||||
// Payload (variable)
|
||||
}
|
||||
var challengeLen = 56
|
||||
|
||||
challengeStartOffset := bytes.Index(data, []byte{'N', 'T', 'L', 'M', 'S', 'S', 'P', 0})
|
||||
if challengeStartOffset == -1 {
|
||||
}
|
||||
if len(data) < challengeStartOffset+challengeLen {
|
||||
return nil
|
||||
}
|
||||
var responseData NTLMChallenge
|
||||
response := data[challengeStartOffset:]
|
||||
responseBuf := bytes.NewBuffer(response)
|
||||
err := binary.Read(responseBuf, binary.LittleEndian, &responseData)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// Check if valid NTLM challenge response message structure
|
||||
if responseData.MessageType != 0x00000002 ||
|
||||
responseData.Reserved != 0 ||
|
||||
!reflect.DeepEqual(responseData.Version[4:], []byte{0, 0, 0, 0xF}) {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Parse: Version
|
||||
type version struct {
|
||||
MajorVersion byte
|
||||
MinorVersion byte
|
||||
BuildNumber uint16
|
||||
}
|
||||
var versionData version
|
||||
versionBuf := bytes.NewBuffer(responseData.Version[:4])
|
||||
err = binary.Read(versionBuf, binary.LittleEndian, &versionData)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
ProductVersion := fmt.Sprintf("%d.%d.%d", versionData.MajorVersion,
|
||||
versionData.MinorVersion,
|
||||
versionData.BuildNumber)
|
||||
glog.Debug("get product version: Windows", ProductVersion)
|
||||
info["ProductVersion"] = ProductVersion
|
||||
|
||||
v, ok := OsVersion[ProductVersion]
|
||||
if ok {
|
||||
info["OsVerion"] = v
|
||||
glog.Debug("get os version:", v)
|
||||
} else {
|
||||
if versionData.BuildNumber >= 22000 {
|
||||
info["OsVerion"] = fmt.Sprintf("Windows 11, version:%s", ProductVersion)
|
||||
} else {
|
||||
info["OsVerion"] = fmt.Sprintf("Windows %s", ProductVersion)
|
||||
}
|
||||
}
|
||||
|
||||
// Parse: TargetName
|
||||
targetNameLen := int(responseData.TargetNameLen)
|
||||
if targetNameLen > 0 {
|
||||
startIdx := int(responseData.TargetNameBufferOffset)
|
||||
endIdx := startIdx + targetNameLen
|
||||
targetName := strings.ReplaceAll(string(response[startIdx:endIdx]), "\x00", "")
|
||||
info["TargetName"] = targetName
|
||||
glog.Debug("target Name = ", targetName)
|
||||
}
|
||||
|
||||
// Parse: TargetInfo
|
||||
AvIDMap := map[uint16]string{
|
||||
1: "NetBIOSComputerName",
|
||||
2: "NetBIOSDomainName",
|
||||
3: "FQDN", // DNS Computer Name
|
||||
4: "DNSDomainName",
|
||||
5: "DNSTreeName",
|
||||
7: "Timestamp",
|
||||
9: "MsvAvTargetName",
|
||||
}
|
||||
|
||||
type AVPair struct {
|
||||
AvID uint16
|
||||
AvLen uint16
|
||||
// Value (variable)
|
||||
}
|
||||
var avPairLen = 4
|
||||
targetInfoLen := int(responseData.TargetInfoLen)
|
||||
if targetInfoLen > 0 {
|
||||
startIdx := int(responseData.TargetInfoBufferOffset)
|
||||
if startIdx+targetInfoLen > len(response) {
|
||||
return fmt.Errorf("Invalid TargetInfoLen value")
|
||||
}
|
||||
var avPair AVPair
|
||||
avPairBuf := bytes.NewBuffer(response[startIdx : startIdx+avPairLen])
|
||||
err = binary.Read(avPairBuf, binary.LittleEndian, &avPair)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
currIdx := startIdx
|
||||
for avPair.AvID != 0 {
|
||||
if field, exists := AvIDMap[avPair.AvID]; exists {
|
||||
var value string
|
||||
r := response[currIdx+avPairLen : currIdx+avPairLen+int(avPair.AvLen)]
|
||||
if avPair.AvID == 7 {
|
||||
unixStamp := binary.LittleEndian.Uint64(r)/10000000 - 11644473600
|
||||
tm := time.Unix(int64(unixStamp), 0)
|
||||
value = tm.Format("2006-01-02 15:04:05")
|
||||
} else {
|
||||
value = strings.ReplaceAll(string(r), "\x00", "")
|
||||
}
|
||||
info[field] = value
|
||||
}
|
||||
currIdx += avPairLen + int(avPair.AvLen)
|
||||
if currIdx+avPairLen > startIdx+targetInfoLen {
|
||||
return fmt.Errorf("Invalid AV_PAIR list")
|
||||
}
|
||||
avPairBuf = bytes.NewBuffer(response[currIdx : currIdx+avPairLen])
|
||||
err = binary.Read(avPairBuf, binary.LittleEndian, &avPair)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
glog.Info("get os info by NLA done !")
|
||||
glog.Info("=======================================")
|
||||
for key, value := range info {
|
||||
glog.Info(key, ":", value)
|
||||
}
|
||||
glog.Info("=======================================")
|
||||
|
||||
//判断是否存在windows域
|
||||
if netBiosDomainName, exists := info["NetBIOSDomainName"]; exists {
|
||||
if netBiosComputerName, exists := info["NetBIOSComputerName"]; exists {
|
||||
if netBiosDomainName == netBiosComputerName {
|
||||
info["DNSDomainName"], info["NetBIOSDomainName"] = "WORKGROUP", "WORKGROUP"
|
||||
//delete(info, "FQDN")
|
||||
} else {
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
t.Emit("os_info", info)
|
||||
|
||||
// end
|
||||
glog.Trace("recvChallenge", hex.EncodeToString(data))
|
||||
tsreq, err := nla.DecodeDERTRequest(data)
|
||||
if err != nil {
|
||||
glog.Info("DecodeDERTRequest", err)
|
||||
return err
|
||||
}
|
||||
glog.Debugf("tsreq:%+v", tsreq)
|
||||
// get pubkey
|
||||
pubkey, err := t.Conn.TlsPubKey()
|
||||
glog.Debugf("pubkey=%+v", pubkey)
|
||||
|
||||
authMsg, ntlmSec := t.ntlm.GetAuthenticateMessage(tsreq.NegoTokens[0].Data)
|
||||
t.ntlmSec = ntlmSec
|
||||
|
||||
encryptPubkey := ntlmSec.GssEncrypt(pubkey)
|
||||
req := nla.EncodeDERTRequest([]nla.Message{authMsg}, nil, encryptPubkey)
|
||||
_, err = t.Conn.Write(req)
|
||||
if err != nil {
|
||||
glog.Info("send AuthenticateMessage", err)
|
||||
return err
|
||||
}
|
||||
resp := make([]byte, 1024)
|
||||
n, err := t.Conn.Read(resp)
|
||||
if err != nil {
|
||||
glog.Error("Read:", err)
|
||||
return fmt.Errorf("read %s", err)
|
||||
} else {
|
||||
glog.Debug("recvChallenge Read success")
|
||||
}
|
||||
return t.recvPubKeyInc(resp[:n])
|
||||
}
|
||||
|
||||
// ErrNLAAuthSuccess 表示NLA仅验证模式下认证成功（非真正错误）
|
||||
var ErrNLAAuthSuccess = fmt.Errorf("NLA_AUTH_SUCCESS")
|
||||
|
||||
func (t *TPKT) recvPubKeyInc(data []byte) error {
|
||||
glog.Trace("recvPubKeyInc", hex.EncodeToString(data))
|
||||
|
||||
tsreq, err := nla.DecodeDERTRequest(data)
|
||||
if err != nil {
|
||||
glog.Info("DecodeDERTRequest", err)
|
||||
return err
|
||||
}
|
||||
|
||||
// 检查服务器是否返回错误码（认证失败）
|
||||
// 常见错误码: 0xC000006D = STATUS_LOGON_FAILURE (密码错误)
|
||||
if tsreq.ErrorCode != 0 {
|
||||
glog.Error("NLA authentication failed with error code:", tsreq.ErrorCode)
|
||||
return fmt.Errorf("NLA auth failed: error code %d (0x%X)", tsreq.ErrorCode, uint32(tsreq.ErrorCode))
|
||||
}
|
||||
|
||||
// 验证 PubKeyAuth 不为空（认证成功的标志）
|
||||
if len(tsreq.PubKeyAuth) == 0 {
|
||||
glog.Error("NLA authentication failed: empty PubKeyAuth")
|
||||
return fmt.Errorf("NLA auth failed: empty PubKeyAuth")
|
||||
}
|
||||
|
||||
glog.Trace("PubKeyAuth:", tsreq.PubKeyAuth)
|
||||
|
||||
// 尝试解密验证公钥，但不作为强制失败条件
|
||||
// 因为某些Windows版本的响应格式可能略有不同
|
||||
pubkey := t.ntlmSec.GssDecrypt(tsreq.PubKeyAuth)
|
||||
if pubkey == nil {
|
||||
glog.Debug("GssDecrypt returned nil, but continuing since no ErrorCode was returned")
|
||||
}
|
||||
|
||||
// NLA仅验证模式：凭据已验证成功，不发送credentials，直接返回
|
||||
// 这样不会建立RDP会话，不会挤掉已登录用户
|
||||
if t.nlaAuthOnly {
|
||||
glog.Info("NLA auth-only mode: credentials verified, skipping session establishment")
|
||||
return ErrNLAAuthSuccess
|
||||
}
|
||||
|
||||
domain, username, password := t.ntlm.GetEncodedCredentials()
|
||||
credentials := nla.EncodeDERTCredentials(domain, username, password)
|
||||
authInfo := t.ntlmSec.GssEncrypt(credentials)
|
||||
req := nla.EncodeDERTRequest(nil, authInfo, nil)
|
||||
_, err = t.Conn.Write(req)
|
||||
if err != nil {
|
||||
glog.Info("send AuthenticateMessage", err)
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *TPKT) Read(b []byte) (n int, err error) {
|
||||
return t.Conn.Read(b)
|
||||
}
|
||||
|
||||
func (t *TPKT) Write(data []byte) (n int, err error) {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt8(FASTPATH_ACTION_X224, buff)
|
||||
core.WriteUInt8(0, buff)
|
||||
core.WriteUInt16BE(uint16(len(data)+4), buff)
|
||||
buff.Write(data)
|
||||
glog.Trace("tpkt Write", hex.EncodeToString(buff.Bytes()))
|
||||
return t.Conn.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (t *TPKT) Close() error {
|
||||
return t.Conn.Close()
|
||||
}
|
||||
|
||||
func (t *TPKT) SetFastPathListener(f core.FastPathListener) {
|
||||
t.fastPathListener = f
|
||||
}
|
||||
|
||||
func (t *TPKT) SendFastPath(secFlag byte, data []byte) (n int, err error) {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt8(FASTPATH_ACTION_FASTPATH|((secFlag&0x3)<<6), buff)
|
||||
core.WriteUInt16BE(uint16(len(data)+3)|0x8000, buff)
|
||||
buff.Write(data)
|
||||
glog.Trace("TPTK SendFastPath", hex.EncodeToString(buff.Bytes()))
|
||||
return t.Conn.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (t *TPKT) recvHeader(s []byte, err error) {
|
||||
glog.Trace("tpkt recvHeader", hex.EncodeToString(s), err)
|
||||
if err != nil {
|
||||
t.Emit("error", err)
|
||||
return
|
||||
}
|
||||
r := bytes.NewReader(s)
|
||||
version, _ := core.ReadUInt8(r)
|
||||
if version == FASTPATH_ACTION_X224 {
|
||||
glog.Debug("tptk recvHeader FASTPATH_ACTION_X224, wait for recvExtendedHeader")
|
||||
core.StartReadBytes(2, t.Conn, t.recvExtendedHeader)
|
||||
} else {
|
||||
glog.Debug("[-] !!!! version is not FASTPATH_ACTION_X224, version=", version)
|
||||
t.secFlag = (version >> 6) & 0x3
|
||||
length, _ := core.ReadUInt8(r)
|
||||
t.lastShortLength = int(length)
|
||||
glog.Debug("last read len:", length)
|
||||
if t.lastShortLength&0x80 != 0 {
|
||||
core.StartReadBytes(1, t.Conn, t.recvExtendedFastPathHeader)
|
||||
} else {
|
||||
//core.StartReadBytes(1, t.Conn, t.recvExtendedFastPathHeader)
|
||||
if t.lastShortLength >= 2 {
|
||||
core.StartReadBytes(t.lastShortLength-2, t.Conn, t.recvFastPath)
|
||||
} else {
|
||||
glog.Debug("lastShortLength = 0")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (t *TPKT) recvExtendedHeader(s []byte, err error) {
|
||||
glog.Trace("tpkt recvExtendedHeader", hex.EncodeToString(s), err)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
r := bytes.NewReader(s)
|
||||
size, _ := core.ReadUint16BE(r)
|
||||
glog.Debug("tpkt wait recvData:", size)
|
||||
core.StartReadBytes(int(size-4), t.Conn, t.recvData)
|
||||
}
|
||||
|
||||
func (t *TPKT) recvData(s []byte, err error) {
|
||||
glog.Trace("tpkt recvData", hex.EncodeToString(s), err)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
t.Emit("data", s)
|
||||
core.StartReadBytes(2, t.Conn, t.recvHeader)
|
||||
}
|
||||
|
||||
func (t *TPKT) recvExtendedFastPathHeader(s []byte, err error) {
|
||||
glog.Trace("tpkt recvExtendedFastPathHeader", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
rightPart, err := core.ReadUInt8(r)
|
||||
if err != nil {
|
||||
glog.Error("TPTK recvExtendedFastPathHeader", err)
|
||||
return
|
||||
}
|
||||
|
||||
leftPart := t.lastShortLength & ^0x80
|
||||
packetSize := (leftPart << 8) + int(rightPart)
|
||||
if packetSize == 0 {
|
||||
fmt.Println("get packetSize,rightPart=", packetSize, rightPart)
|
||||
t.Emit("close")
|
||||
} else {
|
||||
core.StartReadBytes(packetSize-3, t.Conn, t.recvFastPath)
|
||||
}
|
||||
}
|
||||
|
||||
func (t *TPKT) recvFastPath(s []byte, err error) {
|
||||
glog.Trace("tpkt recvFastPath")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
t.fastPathListener.RecvFastPath(t.secFlag, s)
|
||||
core.StartReadBytes(2, t.Conn, t.recvHeader)
|
||||
}
|
||||
@@ -0,0 +1,431 @@
|
||||
package x224
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
||||
|
||||
"github.com/lunixbochs/struc"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/emission"
|
||||
"github.com/shadow1ng/fscan/mylib/grdp/protocol/tpkt"
|
||||
)
|
||||
|
||||
// take idea from https://github.com/Madnikulin50/gordp
|
||||
|
||||
/**
|
||||
* Message type present in X224 packet header
|
||||
*/
|
||||
type MessageType byte
|
||||
|
||||
const (
|
||||
TPDU_CONNECTION_REQUEST MessageType = 0xE0
|
||||
TPDU_CONNECTION_CONFIRM = 0xD0
|
||||
TPDU_DISCONNECT_REQUEST = 0x80
|
||||
TPDU_DATA = 0xF0
|
||||
TPDU_ERROR = 0x70
|
||||
)
|
||||
|
||||
/**
|
||||
* Type of negotiation present in negotiation packet
|
||||
*/
|
||||
type NegotiationType byte
|
||||
|
||||
const (
|
||||
TYPE_RDP_NEG_REQ NegotiationType = 0x01
|
||||
TYPE_RDP_NEG_RSP = 0x02
|
||||
TYPE_RDP_NEG_FAILURE = 0x03
|
||||
)
|
||||
|
||||
/**
|
||||
* Protocols available for x224 layer
|
||||
*/
|
||||
|
||||
const (
|
||||
PROTOCOL_RDP uint32 = 0x00000000
|
||||
PROTOCOL_SSL = 0x00000001
|
||||
PROTOCOL_HYBRID = 0x00000002
|
||||
PROTOCOL_HYBRID_EX = 0x00000008
|
||||
)
|
||||
|
||||
/**
|
||||
* Use to negotiate security layer of RDP stack
|
||||
* In node-rdpjs only ssl is available
|
||||
* @param opt {object} component type options
|
||||
* @see request -> http://msdn.microsoft.com/en-us/library/cc240500.aspx
|
||||
* @see response -> http://msdn.microsoft.com/en-us/library/cc240506.aspx
|
||||
* @see failure ->http://msdn.microsoft.com/en-us/library/cc240507.aspx
|
||||
*/
|
||||
type Negotiation struct {
|
||||
Type NegotiationType `struc:"byte"`
|
||||
Flag uint8 `struc:"uint8"`
|
||||
Length uint16 `struc:"little"`
|
||||
Result uint32 `struc:"little"`
|
||||
}
|
||||
|
||||
func NewNegotiation() *Negotiation {
|
||||
return &Negotiation{0, 0, 0x0008 /*constant*/, PROTOCOL_RDP}
|
||||
}
|
||||
|
||||
const (
|
||||
//The server requires that the client support Enhanced RDP Security (section 5.4) with either TLS 1.0, 1.1 or 1.2 (section 5.4.5.1) or CredSSP (section 5.4.5.2). If only CredSSP was requested then the server only supports TLS.
|
||||
SSL_REQUIRED_BY_SERVER = 0x00000001
|
||||
|
||||
//The server is configured to only use Standard RDP Security mechanisms (section 5.3) and does not support any External Security Protocols (section 5.4.5).
|
||||
SSL_NOT_ALLOWED_BY_SERVER = 0x00000002
|
||||
|
||||
//The server does not possess a valid authentication certificate and cannot initialize the External Security Protocol Provider (section 5.4.5).
|
||||
SSL_CERT_NOT_ON_SERVER = 0x00000003
|
||||
|
||||
//The list of requested security protocols is not consistent with the current security protocol in effect. This error is only possible when the Direct Approach (sections 5.4.2.2 and 1.3.1.2) is used and an External Security Protocol (section 5.4.5) is already being used.
|
||||
INCONSISTENT_FLAGS = 0x00000004
|
||||
|
||||
//The server requires that the client support Enhanced RDP Security (section 5.4) with CredSSP (section 5.4.5.2).
|
||||
HYBRID_REQUIRED_BY_SERVER = 0x00000005
|
||||
|
||||
//The server requires that the client support Enhanced RDP Security (section 5.4) with TLS 1.0, 1.1 or 1.2 (section 5.4.5.1) and certificate-based client authentication.<4>
|
||||
SSL_WITH_USER_AUTH_REQUIRED_BY_SERVER = 0x00000006
|
||||
)
|
||||
|
||||
/**
|
||||
* X224 client connection request
|
||||
* @param opt {object} component type options
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240470.aspx
|
||||
*/
|
||||
type ClientConnectionRequestPDU struct {
|
||||
Len uint8
|
||||
Code MessageType
|
||||
Padding1 uint16
|
||||
Padding2 uint16
|
||||
Padding3 uint8
|
||||
Cookie []byte
|
||||
requestedProtocol uint32
|
||||
ProtocolNeg *Negotiation
|
||||
}
|
||||
|
||||
func NewClientConnectionRequestPDU(cookie []byte, requestedProtocol uint32) *ClientConnectionRequestPDU {
|
||||
x := ClientConnectionRequestPDU{0, TPDU_CONNECTION_REQUEST, 0, 0, 0,
|
||||
cookie, requestedProtocol, NewNegotiation()}
|
||||
|
||||
x.Len = 6
|
||||
if len(cookie) > 0 {
|
||||
x.Len += uint8(len(cookie) + 2)
|
||||
}
|
||||
if x.requestedProtocol > PROTOCOL_RDP {
|
||||
x.Len += 8
|
||||
}
|
||||
|
||||
return &x
|
||||
}
|
||||
|
||||
func (x *ClientConnectionRequestPDU) Serialize() []byte {
|
||||
buff := &bytes.Buffer{}
|
||||
core.WriteUInt8(x.Len, buff)
|
||||
core.WriteUInt8(uint8(x.Code), buff)
|
||||
core.WriteUInt16BE(x.Padding1, buff)
|
||||
core.WriteUInt16BE(x.Padding2, buff)
|
||||
core.WriteUInt8(x.Padding3, buff)
|
||||
|
||||
if len(x.Cookie) > 0 {
|
||||
buff.Write(x.Cookie)
|
||||
core.WriteUInt8(0x0D, buff)
|
||||
core.WriteUInt8(0x0A, buff)
|
||||
}
|
||||
|
||||
if x.requestedProtocol > PROTOCOL_RDP {
|
||||
struc.Pack(buff, x.ProtocolNeg)
|
||||
}
|
||||
|
||||
return buff.Bytes()
|
||||
}
|
||||
|
||||
/**
|
||||
* X224 Server connection confirm
|
||||
* @param opt {object} component type options
|
||||
* @see http://msdn.microsoft.com/en-us/library/cc240506.aspx
|
||||
*/
|
||||
type ServerConnectionConfirm struct {
|
||||
Len uint8
|
||||
Code MessageType
|
||||
Padding1 uint16
|
||||
Padding2 uint16
|
||||
Padding3 uint8
|
||||
ProtocolNeg *Negotiation
|
||||
}
|
||||
|
||||
/**
|
||||
* Header of each data message from x224 layer
|
||||
* @returns {type.Component}
|
||||
*/
|
||||
type DataHeader struct {
|
||||
Header uint8 `struc:"little"`
|
||||
MessageType MessageType `struc:"uint8"`
|
||||
Separator uint8 `struc:"little"`
|
||||
}
|
||||
|
||||
func NewDataHeader() *DataHeader {
|
||||
return &DataHeader{2, TPDU_DATA /* constant */, 0x80 /*constant*/}
|
||||
}
|
||||
|
||||
/**
|
||||
* Common X224 Automata
|
||||
* @param presentation {Layer} presentation layer
|
||||
*/
|
||||
type X224 struct {
|
||||
emission.Emitter
|
||||
transport core.Transport
|
||||
requestedProtocol uint32
|
||||
selectedProtocol uint32
|
||||
dataHeader *DataHeader
|
||||
}
|
||||
|
||||
func New(t core.Transport) *X224 {
|
||||
x := &X224{
|
||||
*emission.NewEmitter(),
|
||||
t,
|
||||
PROTOCOL_RDP | PROTOCOL_SSL | PROTOCOL_HYBRID,
|
||||
PROTOCOL_SSL,
|
||||
NewDataHeader(),
|
||||
}
|
||||
|
||||
t.On("close", func() {
|
||||
x.Emit("close")
|
||||
}).On("error", func(err error) {
|
||||
x.Emit("error", err)
|
||||
})
|
||||
|
||||
return x
|
||||
}
|
||||
|
||||
func (x *X224) ServerChooseProtocol() uint32 {
|
||||
return x.selectedProtocol
|
||||
}
|
||||
|
||||
func (x *X224) Read(b []byte) (n int, err error) {
|
||||
return x.transport.Read(b)
|
||||
}
|
||||
|
||||
func (x *X224) Write(b []byte) (n int, err error) {
|
||||
buff := &bytes.Buffer{}
|
||||
err = struc.Pack(buff, x.dataHeader)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
buff.Write(b)
|
||||
|
||||
glog.Trace("x224 write:", hex.EncodeToString(buff.Bytes()))
|
||||
return x.transport.Write(buff.Bytes())
|
||||
}
|
||||
|
||||
func (x *X224) Close() error {
|
||||
return x.transport.Close()
|
||||
}
|
||||
|
||||
func (x *X224) SetRequestedProtocol(p uint32) {
|
||||
x.requestedProtocol = p
|
||||
}
|
||||
|
||||
func (x *X224) Connect() error {
|
||||
if x.transport == nil {
|
||||
return errors.New("no transport")
|
||||
}
|
||||
cookie := "Cookie: mstshash=bob"
|
||||
message := NewClientConnectionRequestPDU([]byte(cookie), x.requestedProtocol)
|
||||
message.ProtocolNeg.Type = TYPE_RDP_NEG_REQ
|
||||
message.ProtocolNeg.Result = uint32(x.requestedProtocol)
|
||||
|
||||
glog.Debug("x224 sendConnectionRequest", hex.EncodeToString(message.Serialize()))
|
||||
_, err := x.transport.Write(message.Serialize())
|
||||
x.transport.Once("data", x.recvConnectionConfirm)
|
||||
return err
|
||||
}
|
||||
|
||||
func (x *X224) recvConnectionConfirm(s []byte) {
|
||||
/*
|
||||
在Windows的远程桌面协议（RDP）交互过程中，NLA是指网络级别身份验证（Network Level Authentication）。NLA是一种用于增强远程桌面连接安全性的机制。在启用了NLA的情况下，客户端必须在建立RDP会话之前通过网络级别的身份验证，这样可以防止未经授权的用户连接到远程桌面服务器。
|
||||
|
||||
NLA的优点
|
||||
提高安全性：在建立RDP会话之前进行身份验证，确保只有经过验证的用户才能连接。
|
||||
减少资源消耗：因为身份验证是在连接建立之前完成的，可以减少未授权用户消耗的系统资源。
|
||||
RDP协议中的不同连接类型
|
||||
RDP协议有几种不同的连接类型，它们在使用NLA方面有所不同：
|
||||
|
||||
PROTOCOL_RDP：标准的RDP连接方式。这是最早期的RDP连接类型，不使用任何额外的安全层。
|
||||
|
||||
PROTOCOL_SSL：使用SSL/TLS加密的RDP连接方式。这种方式可以增强连接的安全性。
|
||||
|
||||
PROTOCOL_HYBRID：混合连接方式，通常指的是使用NLA和TLS结合的连接方式。它先进行网络级别的身份验证（NLA），然后使用TLS加密传输数据。
|
||||
|
||||
PROTOCOL_HYBRID_EX：这是PROTOCOL_HYBRID的扩展版本，可能包含额外的安全特性或增强功能，具体细节通常会在相关文档中描述。
|
||||
|
||||
NLA与不同连接类型的关系
|
||||
PROTOCOL_RDP：不使用NLA，因为这是最基本的连接方式。
|
||||
PROTOCOL_SSL：可以与NLA结合使用。首先通过NLA进行身份验证，然后使用SSL/TLS加密数据传输。
|
||||
PROTOCOL_HYBRID：使用NLA进行身份验证，然后通过TLS加密数据传输。因此，NLA在这种连接类型中是必需的。
|
||||
PROTOCOL_HYBRID_EX：作为PROTOCOL_HYBRID的扩展版本，也可以使用NLA进行身份验证，并结合其他安全增强特性。
|
||||
|
||||
总的来说，除了最基本的PROTOCOL_RDP之外，其他连接类型（PROTOCOL_SSL、PROTOCOL_HYBRID、PROTOCOL_HYBRID_EX）都可以使用或要求使用NLA来提高连接的安全性。
|
||||
*/
|
||||
|
||||
/*
|
||||
总体而言，较早的Windows版本（如Windows 2000、Windows XP、Windows Server 2003等）默认使用基本的RDP协议（PROTOCOL_RDP），而现代的Windows版本（如Windows 7及之后的版本）默认启用网络级别身份验证（NLA）并支持SSL/TLS加密，以提高连接的安全性。具体的默认协议如下：
|
||||
|
||||
Windows 2000、Windows XP、Windows Server 2003： PROTOCOL_RDP
|
||||
Windows Vista、Windows Server 2008： PROTOCOL_RDP（NLA可配置）
|
||||
Windows 7、Windows Server 2008 R2： PROTOCOL_HYBRID（默认启用NLA）
|
||||
Windows 8、Windows Server 2012： PROTOCOL_HYBRID（默认启用NLA）
|
||||
Windows 8.1、Windows Server 2012 R2： PROTOCOL_HYBRID（默认启用NLA）
|
||||
Windows 10、Windows Server 2016： PROTOCOL_HYBRID（默认启用NLA）
|
||||
Windows 10（1809及以上版本）、Windows Server 2019：PROTOCOL_HYBRID（默认启用NLA）
|
||||
Windows 11、Windows Server 2022： PROTOCOL_HYBRID（默认启用NLA）
|
||||
|
||||
*/
|
||||
glog.Debug("x224 recvConnectionConfirm ", hex.EncodeToString(s))
|
||||
r := bytes.NewReader(s)
|
||||
ln, _ := core.ReadUInt8(r)
|
||||
|
||||
if ln > 6 {
|
||||
message := &ServerConnectionConfirm{}
|
||||
if err := struc.Unpack(bytes.NewReader(s), message); err != nil {
|
||||
glog.Error("ReadServerConnectionConfirm err", err)
|
||||
return
|
||||
}
|
||||
glog.Debugf("message: %+v", *message.ProtocolNeg)
|
||||
|
||||
if message.ProtocolNeg.Type == TYPE_RDP_NEG_FAILURE {
|
||||
glog.Error(fmt.Sprintf("NODE_RDP_PROTOCOL_X224_NEG_FAILURE with code: %d,see https://msdn.microsoft.com/en-us/library/cc240507.aspx",
|
||||
message.ProtocolNeg.Result))
|
||||
//only use Standard RDP Security mechanisms
|
||||
if message.ProtocolNeg.Result == 2 {
|
||||
glog.Info("Only use Standard RDP Security mechanisms, Reconnect with Standard RDP")
|
||||
}
|
||||
switch message.ProtocolNeg.Result {
|
||||
case SSL_REQUIRED_BY_SERVER:
|
||||
// mean need use PROTOCOL_SSL
|
||||
glog.Info("The server requires that the client support Enhanced RDP Security")
|
||||
x.Emit("reconnect", PROTOCOL_SSL)
|
||||
case SSL_NOT_ALLOWED_BY_SERVER:
|
||||
// mean need to use PROTOCOL_RDP only
|
||||
glog.Info("The server is configured to only use Standard RDP Security mechanisms")
|
||||
x.Emit("reconnect", PROTOCOL_RDP)
|
||||
|
||||
case SSL_CERT_NOT_ON_SERVER:
|
||||
glog.Info("The server does not possess a valid authentication certificate and cannot initialize the External Security Protocol Provider")
|
||||
case INCONSISTENT_FLAGS:
|
||||
glog.Info("The list of requested security protocols is not consistent with the current security protocol in effect. This error is only possible when the Direct Approach")
|
||||
case HYBRID_REQUIRED_BY_SERVER:
|
||||
glog.Info("The server requires that the client support Enhanced RDP Security (section 5.4) with CredSSP (section 5.4.5.2).")
|
||||
x.Emit("reconnect", PROTOCOL_HYBRID)
|
||||
case SSL_WITH_USER_AUTH_REQUIRED_BY_SERVER:
|
||||
glog.Info("The server requires that the client support Enhanced RDP Security (section 5.4) with TLS 1.0, 1.1 or 1.2 (section 5.4.5.1) and certificate-based client authentication.<4>")
|
||||
x.Emit("reconnect", PROTOCOL_SSL)
|
||||
////The server requires that the client support Enhanced RDP Security (section 5.4) with either TLS 1.0, 1.1 or 1.2 (section 5.4.5.1) or CredSSP (section 5.4.5.2). If only CredSSP was requested then the server only supports TLS.
|
||||
// SSL_REQUIRED_BY_SERVER = 0x00000001
|
||||
//
|
||||
// //The server is configured to only use Standard RDP Security mechanisms (section 5.3) and does not support any External Security Protocols (section 5.4.5).
|
||||
// SSL_NOT_ALLOWED_BY_SERVER = 0x00000002
|
||||
//
|
||||
// //The server does not possess a valid authentication certificate and cannot initialize the External Security Protocol Provider (section 5.4.5).
|
||||
// SSL_CERT_NOT_ON_SERVER = 0x00000003
|
||||
//
|
||||
// //The list of requested security protocols is not consistent with the current security protocol in effect. This error is only possible when the Direct Approach (sections 5.4.2.2 and 1.3.1.2) is used and an External Security Protocol (section 5.4.5) is already being used.
|
||||
// INCONSISTENT_FLAGS = 0x00000004
|
||||
//
|
||||
// //The server requires that the client support Enhanced RDP Security (section 5.4) with CredSSP (section 5.4.5.2).
|
||||
// HYBRID_REQUIRED_BY_SERVER = 0x00000005
|
||||
//
|
||||
// //The server requires that the client support Enhanced RDP Security (section 5.4) with TLS 1.0, 1.1 or 1.2 (section 5.4.5.1) and certificate-based client authentication.<4>
|
||||
// SSL_WITH_USER_AUTH_REQUIRED_BY_SERVER = 0x00000006
|
||||
}
|
||||
|
||||
x.Close()
|
||||
return
|
||||
}
|
||||
|
||||
if message.ProtocolNeg.Type == TYPE_RDP_NEG_RSP {
|
||||
glog.Info("TYPE_RDP_NEG_RSP", message.ProtocolNeg.Result)
|
||||
x.selectedProtocol = message.ProtocolNeg.Result
|
||||
}
|
||||
} else {
|
||||
x.selectedProtocol = PROTOCOL_RDP
|
||||
}
|
||||
|
||||
serverChooseProtocol := "other not support protocol"
|
||||
switch x.selectedProtocol {
|
||||
case PROTOCOL_RDP:
|
||||
serverChooseProtocol = "PROTOCOL_RDP"
|
||||
x.Emit("more_timeout")
|
||||
case PROTOCOL_SSL:
|
||||
serverChooseProtocol = "PROTOCOL_SSL"
|
||||
case PROTOCOL_HYBRID:
|
||||
serverChooseProtocol = "PROTOCOL_HYBRID"
|
||||
case PROTOCOL_HYBRID_EX:
|
||||
serverChooseProtocol = "PROTOCOL_HYBRID_EX"
|
||||
}
|
||||
glog.Info("Server choose protocol:", serverChooseProtocol)
|
||||
|
||||
//if x.selectedProtocol == PROTOCOL_HYBRID_EX {
|
||||
// glog.Error("NODE_RDP_PROTOCOL_HYBRID_EX_NOT_SUPPORTED")
|
||||
// return
|
||||
//}
|
||||
|
||||
if x.selectedProtocol == PROTOCOL_HYBRID_EX {
|
||||
glog.Info("*** NLA Security selected ***")
|
||||
err := x.transport.(*tpkt.TPKT).StartNLA()
|
||||
glog.Debug("nla end, err?:", err)
|
||||
if err != nil {
|
||||
x.transport.Emit("close")
|
||||
glog.Error("start NLA failed:", err)
|
||||
return
|
||||
}
|
||||
x.Emit("connect", uint32(x.selectedProtocol))
|
||||
return
|
||||
}
|
||||
|
||||
x.transport.On("data", x.recvData)
|
||||
|
||||
if x.selectedProtocol == PROTOCOL_RDP {
|
||||
glog.Info("*** RDP security selected ***")
|
||||
x.Emit("connect", x.selectedProtocol)
|
||||
return
|
||||
}
|
||||
|
||||
if x.selectedProtocol == PROTOCOL_SSL {
|
||||
glog.Info("*** SSL security selected ***")
|
||||
err := x.transport.(*tpkt.TPKT).StartTLS()
|
||||
if err != nil {
|
||||
glog.Error("start tls failed:", err)
|
||||
return
|
||||
}
|
||||
x.Emit("connect", x.selectedProtocol)
|
||||
return
|
||||
}
|
||||
|
||||
if x.selectedProtocol == PROTOCOL_HYBRID {
|
||||
glog.Info("*** NLA Security selected ***")
|
||||
err := x.transport.(*tpkt.TPKT).StartNLA()
|
||||
glog.Debug("nla end, err?:", err)
|
||||
if err != nil {
|
||||
// 检查是否是NLA仅验证模式的成功返回
|
||||
if err == tpkt.ErrNLAAuthSuccess {
|
||||
glog.Info("NLA auth-only mode: credentials verified successfully")
|
||||
x.Emit("error", err) // 通过 error 事件传播成功信号
|
||||
return
|
||||
}
|
||||
glog.Error("start NLA failed:", err)
|
||||
x.Emit("error", err)
|
||||
return
|
||||
}
|
||||
x.Emit("connect", x.selectedProtocol)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func (x *X224) recvData(s []byte) {
|
||||
glog.Trace("x224 recvData", hex.EncodeToString(s), "emit data")
|
||||
// x224 header takes 3 bytes
|
||||
x.Emit("data", s[3:])
|
||||
}
|
||||
Reference in New Issue
Block a user