perf: 优化FindNet.go的代码,添加注释,规范输出

This commit is contained in:
ZacharyZcR 2024-12-18 23:39:13 +08:00
parent 5ad4c1a580
commit 23fea2c290

View File

@ -12,113 +12,151 @@ import (
) )
var ( var (
// RPC请求数据包
bufferV1, _ = hex.DecodeString("05000b03100000004800000001000000b810b810000000000100000000000100c4fefc9960521b10bbcb00aa0021347a00000000045d888aeb1cc9119fe808002b10486002000000") bufferV1, _ = hex.DecodeString("05000b03100000004800000001000000b810b810000000000100000000000100c4fefc9960521b10bbcb00aa0021347a00000000045d888aeb1cc9119fe808002b10486002000000")
bufferV2, _ = hex.DecodeString("050000031000000018000000010000000000000000000500") bufferV2, _ = hex.DecodeString("050000031000000018000000010000000000000000000500")
bufferV3, _ = hex.DecodeString("0900ffff0000") bufferV3, _ = hex.DecodeString("0900ffff0000")
) )
// Findnet 探测Windows网络主机信息的入口函数
func Findnet(info *Config.HostInfo) error { func Findnet(info *Config.HostInfo) error {
err := FindnetScan(info) return FindnetScan(info)
return err
} }
// FindnetScan 通过RPC协议扫描网络主机信息
func FindnetScan(info *Config.HostInfo) error { func FindnetScan(info *Config.HostInfo) error {
realhost := fmt.Sprintf("%s:%v", info.Host, 135) // 连接目标RPC端口
conn, err := Common.WrapperTcpWithTimeout("tcp", realhost, time.Duration(Common.Timeout)*time.Second) target := fmt.Sprintf("%s:%v", info.Host, 135)
conn, err := Common.WrapperTcpWithTimeout("tcp", target, time.Duration(Common.Timeout)*time.Second)
if err != nil { if err != nil {
return err return fmt.Errorf("连接RPC端口失败: %v", err)
} }
defer conn.Close() defer conn.Close()
err = conn.SetDeadline(time.Now().Add(time.Duration(Common.Timeout) * time.Second))
if err != nil { // 设置连接超时
return err if err = conn.SetDeadline(time.Now().Add(time.Duration(Common.Timeout) * time.Second)); err != nil {
return fmt.Errorf("设置超时失败: %v", err)
} }
_, err = conn.Write(bufferV1)
if err != nil { // 发送第一个RPC请求
return err if _, err = conn.Write(bufferV1); err != nil {
return fmt.Errorf("发送RPC请求1失败: %v", err)
} }
// 读取响应
reply := make([]byte, 4096) reply := make([]byte, 4096)
_, err = conn.Read(reply) if _, err = conn.Read(reply); err != nil {
if err != nil { return fmt.Errorf("读取RPC响应1失败: %v", err)
return err
} }
_, err = conn.Write(bufferV2)
if err != nil { // 发送第二个RPC请求
return err if _, err = conn.Write(bufferV2); err != nil {
return fmt.Errorf("发送RPC请求2失败: %v", err)
} }
if n, err := conn.Read(reply); err != nil || n < 42 {
return err // 读取并检查响应
n, err := conn.Read(reply)
if err != nil || n < 42 {
return fmt.Errorf("读取RPC响应2失败: %v", err)
} }
// 解析响应数据
text := reply[42:] text := reply[42:]
flag := true found := false
for i := 0; i < len(text)-5; i++ { for i := 0; i < len(text)-5; i++ {
if bytes.Equal(text[i:i+6], bufferV3) { if bytes.Equal(text[i:i+6], bufferV3) {
text = text[:i-4] text = text[:i-4]
flag = false found = true
break break
} }
} }
if flag {
return err if !found {
return fmt.Errorf("未找到有效的响应标记")
} }
err = read(text, info.Host)
return err // 解析主机信息
return read(text, info.Host)
} }
// HexUnicodeStringToString 将16进制Unicode字符串转换为可读字符串
func HexUnicodeStringToString(src string) string { func HexUnicodeStringToString(src string) string {
sText := "" // 确保输入长度是4的倍数
if len(src)%4 != 0 { if len(src)%4 != 0 {
src += src[:len(src)-len(src)%4] src += src[:len(src)-len(src)%4]
} }
for i := 0; i < len(src); i = i + 4 {
sText += "\\u" + src[i+2:i+4] + src[i:i+2] // 转换为标准Unicode格式
var sText string
for i := 0; i < len(src); i += 4 {
sText += "\\u" + src[i+2:i+4] + src[i:i+2] // 调整字节顺序
} }
textUnquoted := sText // 解析每个Unicode字符
sUnicodev := strings.Split(textUnquoted, "\\u") unicodeChars := strings.Split(sText, "\\u")
var context string var result string
for _, v := range sUnicodev {
if len(v) < 1 { for _, char := range unicodeChars {
// 跳过空字符
if len(char) < 1 {
continue continue
} }
temp, err := strconv.ParseInt(v, 16, 32)
// 将16进制转换为整数
codePoint, err := strconv.ParseInt(char, 16, 32)
if err != nil { if err != nil {
return "" return ""
} }
context += fmt.Sprintf("%c", temp)
// 转换为实际字符
result += fmt.Sprintf("%c", codePoint)
} }
return context
return result
} }
// read 解析并显示主机网络信息
func read(text []byte, host string) error { func read(text []byte, host string) error {
// 将原始数据转换为16进制字符串
encodedStr := hex.EncodeToString(text) encodedStr := hex.EncodeToString(text)
hn := "" // 解析主机名
for i := 0; i < len(encodedStr)-4; i = i + 4 { var hostName string
for i := 0; i < len(encodedStr)-4; i += 4 {
if encodedStr[i:i+4] == "0000" { if encodedStr[i:i+4] == "0000" {
break break
} }
hn += encodedStr[i : i+4] hostName += encodedStr[i : i+4]
} }
var name string // 转换主机名为可读字符串
name = HexUnicodeStringToString(hn) name := HexUnicodeStringToString(hostName)
hostnames := strings.Replace(encodedStr, "0700", "", -1) // 解析网络信息
hostname := strings.Split(hostnames, "000000") netInfo := strings.Replace(encodedStr, "0700", "", -1)
result := "[*] NetInfo \n[*]" + host hosts := strings.Split(netInfo, "000000")
hosts = hosts[1:] // 跳过第一个空元素
// 构造输出结果
result := fmt.Sprintf("[*] NetInfo\n[*] %s", host)
if name != "" { if name != "" {
result += "\n [->]" + name result += fmt.Sprintf("\n [->] %s", name)
} }
hostname = hostname[1:]
for i := 0; i < len(hostname); i++ { // 解析每个网络主机信息
hostname[i] = strings.Replace(hostname[i], "00", "", -1) for _, h := range hosts {
host, err := hex.DecodeString(hostname[i]) // 移除填充字节
h = strings.Replace(h, "00", "", -1)
// 解码主机信息
hostInfo, err := hex.DecodeString(h)
if err != nil { if err != nil {
return err return fmt.Errorf("解码主机信息失败: %v", err)
} }
result += "\n [->]" + string(host) result += fmt.Sprintf("\n [->] %s", string(hostInfo))
} }
// 输出结果
Common.LogSuccess(result) Common.LogSuccess(result)
return nil return nil
} }