mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-24 12:11:52 +08:00
ZacharyZcR
71b92d4408
## 架构重构
- 全局变量消除，迁移至 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）
663 lines
17 KiB
Go
663 lines
17 KiB
Go
package login
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/core"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/protocol/nla"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/protocol/pdu"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/protocol/sec"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/protocol/t125"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/protocol/tpkt"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/protocol/x224"
|
|
"golang.org/x/net/context"
|
|
"golang.org/x/net/proxy"
|
|
"image"
|
|
"image/color"
|
|
"image/draw"
|
|
"image/jpeg"
|
|
"log"
|
|
"math/rand"
|
|
"net"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
Socks5Proxy string = ""
|
|
LogLever glog.LEVEL = glog.NONE
|
|
OutputDir string
|
|
)
|
|
|
|
func init() {
|
|
}
|
|
|
|
func RdpConn(host, domain, user, password string, timeout int64, rdpProtocol uint32) (bool, error) {
|
|
g := NewClient(host, LogLever)
|
|
status, err, reconnectProtocol := g.ScreenShot(domain, user, password, timeout, rdpProtocol)
|
|
if status == true {
|
|
return true, err
|
|
} else {
|
|
if reconnectProtocol != rdpProtocol {
|
|
glog.Info("reconnect with protocol:", reconnectProtocol)
|
|
return RdpConn(host, domain, user, password, timeout, reconnectProtocol)
|
|
} else {
|
|
return status, err
|
|
}
|
|
}
|
|
}
|
|
|
|
func RdpCrack(host, domain, user, password string, timeout int64, rdpProtocol uint32) (bool, error) {
|
|
g := NewClient(host, LogLever)
|
|
status, err, reconnectProtocol := g.Crack(domain, user, password, timeout, rdpProtocol)
|
|
if status == true {
|
|
return true, err
|
|
} else {
|
|
if reconnectProtocol != rdpProtocol {
|
|
glog.Info("reconnect with protocol:", reconnectProtocol)
|
|
return RdpCrack(host, domain, user, password, timeout, reconnectProtocol)
|
|
} else {
|
|
return status, err
|
|
}
|
|
}
|
|
}
|
|
|
|
type Client struct {
|
|
Host string // ip:port
|
|
tpkt *tpkt.TPKT
|
|
x224 *x224.X224
|
|
mcs *t125.MCSClient
|
|
sec *sec.Client
|
|
pdu *pdu.Client
|
|
}
|
|
|
|
func NewClient(host string, logLevel glog.LEVEL) *Client {
|
|
glog.SetLevel(logLevel)
|
|
logger := log.New(os.Stdout, "", 0)
|
|
glog.SetLogger(logger)
|
|
return &Client{
|
|
Host: host,
|
|
}
|
|
}
|
|
|
|
func WrapperTcpWithTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
|
|
local_ip := "0.0.0.0"
|
|
net_ip := net.ParseIP(local_ip)
|
|
if net_ip == nil {
|
|
net_ip = net.ParseIP("0.0.0.0")
|
|
}
|
|
local_addr := &net.TCPAddr{
|
|
IP: net_ip, // 替换为你想要使用的本地IP地址
|
|
}
|
|
d := &net.Dialer{Timeout: timeout, LocalAddr: local_addr}
|
|
return WrapperTCP(network, address, d)
|
|
}
|
|
|
|
func WrapperTCP(network, address string, forward *net.Dialer) (net.Conn, error) {
|
|
//get conn
|
|
var conn net.Conn
|
|
if Socks5Proxy == "" {
|
|
var err error
|
|
conn, err = forward.Dial(network, address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
dailer, err := Socks5Dailer(forward)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conn, err = dailer.Dial(network, address)
|
|
if err != nil {
|
|
// fmt.Println(err)
|
|
return nil, err
|
|
}
|
|
|
|
}
|
|
|
|
timeout := forward.Timeout
|
|
if err := conn.SetWriteDeadline(time.Now().Add(timeout * 6)); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := conn.SetReadDeadline(time.Now().Add(timeout * 6)); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return conn, nil
|
|
|
|
}
|
|
|
|
func Socks5Dailer(forward *net.Dialer) (proxy.Dialer, error) {
|
|
u, err := url.Parse(Socks5Proxy)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if strings.ToLower(u.Scheme) != "socks5" {
|
|
return nil, errors.New("Only support socks5")
|
|
}
|
|
address := u.Host
|
|
var auth proxy.Auth
|
|
var dailer proxy.Dialer
|
|
if u.User.String() != "" {
|
|
auth = proxy.Auth{}
|
|
auth.User = u.User.Username()
|
|
password, _ := u.User.Password()
|
|
auth.Password = password
|
|
dailer, err = proxy.SOCKS5("tcp", address, &auth, forward)
|
|
} else {
|
|
dailer, err = proxy.SOCKS5("tcp", address, nil, forward)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return dailer, nil
|
|
}
|
|
|
|
type Bitmap struct {
|
|
DestLeft int `json:"destLeft"`
|
|
DestTop int `json:"destTop"`
|
|
DestRight int `json:"destRight"`
|
|
DestBottom int `json:"destBottom"`
|
|
Width int `json:"width"`
|
|
Height int `json:"height"`
|
|
BitsPerPixel int `json:"bitsPerPixel"`
|
|
IsCompress bool `json:"isCompress"`
|
|
Data []byte `json:"data"`
|
|
}
|
|
|
|
func Bpp(BitsPerPixel uint16) (pixel int) {
|
|
switch BitsPerPixel {
|
|
case 15:
|
|
pixel = 1
|
|
|
|
case 16:
|
|
pixel = 2
|
|
|
|
case 24:
|
|
pixel = 3
|
|
|
|
case 32:
|
|
pixel = 4
|
|
|
|
default:
|
|
glog.Error("-------------------------------------Bpp func. invalid bitmap data format")
|
|
}
|
|
return
|
|
}
|
|
|
|
func BitmapDecompress(bitmap *pdu.BitmapData) []byte {
|
|
return core.Decompress(bitmap.BitmapDataStream, int(bitmap.Width), int(bitmap.Height), Bpp(bitmap.BitsPerPixel))
|
|
}
|
|
|
|
func ToRGBA(pixel int, i int, data []byte) (r, g, b, a uint8) {
|
|
a = 255
|
|
switch pixel {
|
|
case 1:
|
|
rgb555 := core.Uint16BE(data[i], data[i+1])
|
|
r, g, b = core.RGB555ToRGB(rgb555)
|
|
case 2:
|
|
rgb565 := core.Uint16BE(data[i], data[i+1])
|
|
r, g, b = core.RGB565ToRGB(rgb565)
|
|
case 3, 4:
|
|
fallthrough
|
|
default:
|
|
r, g, b = data[i+2], data[i+1], data[i]
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (g *Client) ProbeOSInfo(host, domain, user, pwd string, timeout int64, rdpProtocol uint32) (info map[string]any) {
|
|
start := time.Now()
|
|
exitFlag := make(chan bool)
|
|
info = make(map[string]any)
|
|
|
|
targetSlice := strings.Split(g.Host, ":")
|
|
ip := targetSlice[0]
|
|
conn, err := WrapperTcpWithTimeout("tcp", g.Host, time.Duration(timeout)*time.Second)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
glog.Info(conn.LocalAddr().String())
|
|
|
|
g.tpkt = tpkt.New(core.NewSocketLayer(conn), nla.NewNTLMv2(domain, user, pwd))
|
|
g.x224 = x224.New(g.tpkt)
|
|
g.mcs = t125.NewMCSClient(g.x224)
|
|
g.sec = sec.NewClient(g.mcs)
|
|
g.pdu = pdu.NewClient(g.sec)
|
|
|
|
g.sec.SetUser(user)
|
|
g.sec.SetPwd(pwd)
|
|
g.sec.SetDomain(domain)
|
|
|
|
g.tpkt.SetFastPathListener(g.sec)
|
|
g.sec.SetFastPathListener(g.pdu)
|
|
g.pdu.SetFastPathSender(g.tpkt)
|
|
g.sec.SetChannelSender(g.mcs)
|
|
|
|
g.tpkt.On("os_info", func(infoMap map[string]any) {
|
|
glog.Debug("[+] callback, get os info ........................")
|
|
for k, v := range infoMap {
|
|
glog.Debugf("%s: %s\n", k, v)
|
|
}
|
|
info = infoMap
|
|
g.pdu.Emit("done")
|
|
})
|
|
|
|
g.x224.SetRequestedProtocol(rdpProtocol) //x224.PROTOCOL_SSL , x224.PROTOCOL_RDP , x224.PROTOCOL_HYBRID , x224.PROTOCOL_HYBRID_EX
|
|
g.x224.On("reconnect", func(protocol uint32) {
|
|
info["reconn"] = protocol
|
|
g.pdu.Emit("close")
|
|
exitFlag <- true
|
|
})
|
|
|
|
err = g.x224.Connect()
|
|
if err != nil {
|
|
info["err"] = err.Error()
|
|
return
|
|
}
|
|
glog.Info("wait connect ok")
|
|
|
|
g.pdu.On("error", func(e error) {
|
|
err = e
|
|
glog.Error("error", e)
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("close", func() {
|
|
err = errors.New("close")
|
|
glog.Info("on close")
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("success", func() {
|
|
glog.Debugf("===============login success %s===============", ip)
|
|
err = nil
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("ready", func() {
|
|
err = nil
|
|
glog.Debug("on ready")
|
|
})
|
|
g.pdu.On("bitmap", func(rectangles []pdu.BitmapData) {
|
|
})
|
|
g.pdu.On("done", func() {
|
|
glog.Debug("done信号触发")
|
|
exitFlag <- true
|
|
})
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeout*3)*time.Second)
|
|
defer cancel()
|
|
|
|
loop:
|
|
for {
|
|
select {
|
|
case <-time.After(time.Second * time.Duration(timeout)): //
|
|
break loop
|
|
case <-exitFlag:
|
|
break loop
|
|
case <-ctx.Done():
|
|
glog.Debug("总超时已达到，退出")
|
|
break loop
|
|
}
|
|
}
|
|
glog.Debug("循环结束，总时间过去了：", time.Since(start))
|
|
return info
|
|
}
|
|
|
|
func (g *Client) ScreenShot(domain, user, pwd string, timeout int64, rdpProtocol uint32) (status bool, err error, reconnProtocol uint32) {
|
|
//glog.SetLevel(glog.ERROR)
|
|
reconnProtocol = rdpProtocol
|
|
pic_length := 1280 //1280
|
|
pic_width := 800 //800
|
|
needReconnect := false
|
|
isScreenOK := false
|
|
refresh := make(chan bool)
|
|
exitFlag := make(chan bool)
|
|
start := time.Now()
|
|
now := start
|
|
screenImage := image.NewRGBA(image.Rect(0, 0, pic_length, pic_width))
|
|
|
|
index := 1
|
|
targetSlice := strings.Split(g.Host, ":")
|
|
ip := targetSlice[0]
|
|
port := targetSlice[1]
|
|
status = false
|
|
conn, err := WrapperTcpWithTimeout("tcp", g.Host, time.Duration(timeout)*time.Second)
|
|
if err != nil {
|
|
return status, fmt.Errorf("[dial err] %v", err), reconnProtocol
|
|
}
|
|
defer conn.Close()
|
|
glog.Info(conn.LocalAddr().String())
|
|
|
|
g.tpkt = tpkt.New(core.NewSocketLayer(conn), nla.NewNTLMv2(domain, user, pwd))
|
|
g.x224 = x224.New(g.tpkt)
|
|
g.mcs = t125.NewMCSClient(g.x224)
|
|
g.sec = sec.NewClient(g.mcs)
|
|
g.pdu = pdu.NewClient(g.sec)
|
|
|
|
g.sec.SetUser(user)
|
|
g.sec.SetPwd(pwd)
|
|
g.sec.SetDomain(domain)
|
|
//g.sec.SetClientAutoReconnect()
|
|
|
|
g.tpkt.SetFastPathListener(g.sec)
|
|
g.sec.SetFastPathListener(g.pdu)
|
|
g.pdu.SetFastPathSender(g.tpkt)
|
|
g.sec.SetChannelSender(g.mcs)
|
|
|
|
g.tpkt.On("os_info", func(info map[string]any) {
|
|
glog.Debug("[+] callback, get os info ........................")
|
|
for k, v := range info {
|
|
glog.Debugf("%s: %s\n", k, v)
|
|
}
|
|
})
|
|
|
|
g.x224.SetRequestedProtocol(rdpProtocol) //x224.PROTOCOL_SSL , x224.PROTOCOL_RDP , x224.PROTOCOL_HYBRID , x224.PROTOCOL_HYBRID_EX
|
|
g.x224.On("reconnect", func(protocol uint32) {
|
|
needReconnect = true
|
|
reconnProtocol = protocol
|
|
glog.Info("need reconnect with protocol:", protocol)
|
|
g.pdu.Emit("close")
|
|
exitFlag <- true
|
|
})
|
|
g.x224.On("more_timeout", func() {
|
|
timeout += 18 //如果是PROTOCOL_RDP协议，可以适当延长超时时间
|
|
})
|
|
|
|
err = g.x224.Connect()
|
|
if err != nil {
|
|
return status, fmt.Errorf("[x224 connect err] %v", err), reconnProtocol
|
|
}
|
|
glog.Info("wait connect ok")
|
|
|
|
g.pdu.On("error", func(e error) {
|
|
err = e
|
|
glog.Error("error", e)
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("close", func() {
|
|
err = errors.New("close")
|
|
glog.Info("on close")
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("success", func() {
|
|
glog.Debugf("===============login success %s===============", ip)
|
|
status = true
|
|
err = nil
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("ready", func() {
|
|
err = nil
|
|
glog.Debug("on ready")
|
|
//g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("bitmap", func(rectangles []pdu.BitmapData) {
|
|
now = time.Now()
|
|
// 发送一个鼠标事件，作用是与服务器保持联系不要断开
|
|
rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
randomNumber := rand.Intn(1000) + 10 // 10到1000之间的随机数
|
|
mouseX := 60 + randomNumber
|
|
mouseY := 10 + randomNumber
|
|
pevent := &pdu.PointerEvent{}
|
|
pevent.PointerFlags |= pdu.PTRFLAGS_MOVE
|
|
pevent.XPos = uint16(mouseX)
|
|
pevent.YPos = uint16(mouseY)
|
|
g.pdu.SendInputEvents(pdu.INPUT_EVENT_MOUSE, []pdu.InputEventsInterface{pevent})
|
|
|
|
glog.Debug("on update bitmap:", len(rectangles))
|
|
bs := make([]Bitmap, 0)
|
|
for _, v := range rectangles {
|
|
IsCompress := v.IsCompress()
|
|
data := v.BitmapDataStream
|
|
if IsCompress {
|
|
data = BitmapDecompress(&v)
|
|
IsCompress = false
|
|
}
|
|
b := Bitmap{int(v.DestLeft), int(v.DestTop), int(v.DestRight), int(v.DestBottom),
|
|
int(v.Width), int(v.Height), Bpp(v.BitsPerPixel), IsCompress, data}
|
|
bs = append(bs, b)
|
|
}
|
|
var (
|
|
pixel int
|
|
i int
|
|
r, g, b, a uint8
|
|
)
|
|
|
|
for _, bm := range bs {
|
|
i = 0
|
|
pixel = bm.BitsPerPixel
|
|
m := image.NewRGBA(image.Rect(0, 0, bm.Width, bm.Height))
|
|
for y := 0; y < bm.Height; y++ {
|
|
for x := 0; x < bm.Width; x++ {
|
|
r, g, b, a = ToRGBA(pixel, i, bm.Data)
|
|
c := color.RGBA{R: r, G: g, B: b, A: a}
|
|
i += pixel
|
|
m.Set(x, y, c)
|
|
}
|
|
}
|
|
draw.Draw(screenImage, screenImage.Bounds().Add(image.Pt(bm.DestLeft, bm.DestTop)), m, m.Bounds().Min, draw.Src)
|
|
}
|
|
// Encode to jpeg.
|
|
//var imageBuf bytes.Buffer
|
|
//err = jpeg.Encode(&imageBuf, screenImage, nil)
|
|
//
|
|
//if err != nil {
|
|
// glog.Info("trans bitmap to jpeg err:", err)
|
|
//}
|
|
|
|
// Write to file.
|
|
//fo, err := os.Create(fmt.Sprintf("img/%s-%d.jpg", ip, index))
|
|
//if err != nil {
|
|
// panic(err)
|
|
//}
|
|
//index += 1
|
|
//fw := bufio.NewWriter(fo)
|
|
//fw.Write(imageBuf.Bytes())
|
|
|
|
isScreenOK = true
|
|
refresh <- true
|
|
|
|
})
|
|
g.pdu.On("done", func() {
|
|
glog.Debug("done信号触发")
|
|
exitFlag <- true
|
|
})
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeout*6)*time.Second)
|
|
defer cancel()
|
|
|
|
loop:
|
|
for {
|
|
select {
|
|
case <-time.After(time.Second * time.Duration(timeout*3)): //
|
|
glog.Debug("本次获取bitmap超时了, 距离上次获取到图像过去了：", time.Since(now))
|
|
break loop
|
|
case <-refresh:
|
|
continue loop
|
|
case <-exitFlag:
|
|
break loop
|
|
case <-ctx.Done():
|
|
glog.Debug("总超时已达到，退出")
|
|
break loop
|
|
}
|
|
}
|
|
glog.Debug("循环结束，总时间过去了：", time.Since(start))
|
|
|
|
if g.x224.ServerChooseProtocol() == x224.PROTOCOL_HYBRID && g.x224.ServerChooseProtocol() == x224.PROTOCOL_HYBRID {
|
|
if err == nil {
|
|
status = true
|
|
}
|
|
}
|
|
|
|
if needReconnect {
|
|
return status, err, reconnProtocol
|
|
} else if isScreenOK {
|
|
glog.Info("get screen ok")
|
|
// Encode to jpeg.
|
|
var imageBuf bytes.Buffer
|
|
err = jpeg.Encode(&imageBuf, screenImage, nil)
|
|
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
// Write to file.
|
|
saveDate := time.Now().Format("2006_01_02_15_04_05")
|
|
fo, writeErr := os.Create(fmt.Sprintf("%s/%s_%s_%s.jpg", OutputDir, ip, port, saveDate))
|
|
index += 1
|
|
if writeErr != nil {
|
|
glog.Error("Can not create rdp screenshot file:", writeErr)
|
|
} else {
|
|
fw := bufio.NewWriter(fo)
|
|
_, writeErr := fw.Write(imageBuf.Bytes())
|
|
if writeErr != nil {
|
|
glog.Error("Can not write rdp screenshot file:", writeErr)
|
|
}
|
|
}
|
|
|
|
}
|
|
return status, err, reconnProtocol
|
|
|
|
}
|
|
|
|
func (g *Client) Crack(domain, user, pwd string, timeout int64, rdpProtocol uint32) (status bool, err error, reconnProtocol uint32) {
|
|
//glog.SetLevel(glog.ERROR)
|
|
reconnProtocol = rdpProtocol
|
|
needReconnect := false
|
|
refresh := make(chan bool)
|
|
exitFlag := make(chan bool)
|
|
start := time.Now()
|
|
now := start
|
|
|
|
targetSlice := strings.Split(g.Host, ":")
|
|
ip := targetSlice[0]
|
|
status = false
|
|
conn, err := WrapperTcpWithTimeout("tcp", g.Host, time.Duration(timeout)*time.Second)
|
|
if err != nil {
|
|
return status, fmt.Errorf("[dial err] %v", err), reconnProtocol
|
|
}
|
|
defer conn.Close()
|
|
glog.Info(conn.LocalAddr().String())
|
|
|
|
g.tpkt = tpkt.New(core.NewSocketLayer(conn), nla.NewNTLMv2(domain, user, pwd))
|
|
g.x224 = x224.New(g.tpkt)
|
|
g.mcs = t125.NewMCSClient(g.x224)
|
|
g.sec = sec.NewClient(g.mcs)
|
|
g.pdu = pdu.NewClient(g.sec)
|
|
|
|
g.sec.SetUser(user)
|
|
g.sec.SetPwd(pwd)
|
|
g.sec.SetDomain(domain)
|
|
//g.sec.SetClientAutoReconnect()
|
|
|
|
g.tpkt.SetFastPathListener(g.sec)
|
|
g.sec.SetFastPathListener(g.pdu)
|
|
g.pdu.SetFastPathSender(g.tpkt)
|
|
g.sec.SetChannelSender(g.mcs)
|
|
|
|
g.tpkt.On("os_info", func(info map[string]any) {
|
|
glog.Debug("[+] callback, get os info ........................")
|
|
for k, v := range info {
|
|
glog.Debugf("%s: %s\n", k, v)
|
|
}
|
|
})
|
|
|
|
g.x224.SetRequestedProtocol(rdpProtocol) //x224.PROTOCOL_SSL , x224.PROTOCOL_RDP , x224.PROTOCOL_HYBRID , x224.PROTOCOL_HYBRID_EX
|
|
g.x224.On("reconnect", func(protocol uint32) {
|
|
needReconnect = true
|
|
reconnProtocol = protocol
|
|
glog.Info("need reconnect with protocol:", protocol)
|
|
g.pdu.Emit("close")
|
|
exitFlag <- true
|
|
})
|
|
g.x224.On("more_timeout", func() {
|
|
timeout += 18 //如果是PROTOCOL_RDP协议，可以适当延长超时时间
|
|
})
|
|
|
|
err = g.x224.Connect()
|
|
if err != nil {
|
|
return status, fmt.Errorf("[x224 connect err] %v", err), reconnProtocol
|
|
}
|
|
glog.Info("wait connect ok")
|
|
|
|
g.pdu.On("error", func(e error) {
|
|
err = e
|
|
glog.Error("error", e)
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("close", func() {
|
|
err = errors.New("close")
|
|
glog.Info("on close")
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("success", func() {
|
|
glog.Debugf("===============login success %s===============", ip)
|
|
status = true
|
|
err = nil
|
|
g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("ready", func() {
|
|
err = nil
|
|
glog.Debug("on ready")
|
|
//g.pdu.Emit("done")
|
|
})
|
|
g.pdu.On("bitmap", func(rectangles []pdu.BitmapData) {
|
|
now = time.Now()
|
|
// 发送一个鼠标事件，作用是与服务器保持联系不要断开
|
|
rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
randomNumber := rand.Intn(1000) + 10 // 10到1000之间的随机数
|
|
mouseX := 60 + randomNumber
|
|
mouseY := 10 + randomNumber
|
|
pevent := &pdu.PointerEvent{}
|
|
pevent.PointerFlags |= pdu.PTRFLAGS_MOVE
|
|
pevent.XPos = uint16(mouseX)
|
|
pevent.YPos = uint16(mouseY)
|
|
g.pdu.SendInputEvents(pdu.INPUT_EVENT_MOUSE, []pdu.InputEventsInterface{pevent})
|
|
|
|
glog.Debug("on update bitmap:", len(rectangles))
|
|
refresh <- true
|
|
|
|
})
|
|
g.pdu.On("done", func() {
|
|
glog.Debug("done信号触发")
|
|
exitFlag <- true
|
|
})
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Duration(timeout*6)*time.Second)
|
|
defer cancel()
|
|
|
|
loop:
|
|
for {
|
|
select {
|
|
case <-time.After(time.Second * time.Duration(timeout*3)): //
|
|
glog.Debug("本次获取bitmap超时了, 距离上次获取到图像过去了：", time.Since(now))
|
|
break loop
|
|
case <-refresh:
|
|
continue loop
|
|
case <-exitFlag:
|
|
break loop
|
|
case <-ctx.Done():
|
|
glog.Debug("总超时已达到，退出")
|
|
break loop
|
|
}
|
|
}
|
|
glog.Debug("循环结束，总时间过去了：", time.Since(start))
|
|
|
|
if g.x224.ServerChooseProtocol() == x224.PROTOCOL_HYBRID && g.x224.ServerChooseProtocol() == x224.PROTOCOL_HYBRID {
|
|
if err == nil {
|
|
status = true
|
|
}
|
|
}
|
|
|
|
if needReconnect {
|
|
return status, err, reconnProtocol
|
|
}
|
|
return status, err, reconnProtocol
|
|
}
|