mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-23 03:31:53 +08:00
ZacharyZcR
90c7f9c27e
- 移除 screen.go 中错误的认证结果覆盖逻辑 - 启用 NLA 协议的 ErrorCode 字段检测 - 添加 PubKeyAuth 验证确保认证真正成功 - 修复 io.go 中错误被静默忽略的问题 - 修复 socket.go/io.go 中可能导致 panic 的代码 - 修复 screen.go 中文件句柄泄漏和 log.Panic
155 lines
3.0 KiB
Go
155 lines
3.0 KiB
Go
package core
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"github.com/shadow1ng/fscan/mylib/grdp/glog"
|
|
"io"
|
|
)
|
|
|
|
type ReadBytesComplete func(result []byte, err error)
|
|
|
|
func StartReadBytes(len int, r io.Reader, cb ReadBytesComplete) {
|
|
glog.Debug("create len:", len)
|
|
b := make([]byte, len)
|
|
go func() {
|
|
_, err := io.ReadFull(r, b)
|
|
//glog.Debug("StartReadBytes Get", n, "Bytes:", hex.EncodeToString(b))
|
|
cb(b, err)
|
|
}()
|
|
}
|
|
|
|
func ReadBytes(len int, r io.Reader) ([]byte, error) {
|
|
b := make([]byte, len)
|
|
length, err := io.ReadFull(r, b)
|
|
return b[:length], err
|
|
}
|
|
|
|
func ReadByte(r io.Reader) (byte, error) {
|
|
b, err := ReadBytes(1, r)
|
|
if err != nil || len(b) == 0 {
|
|
return 0, err
|
|
}
|
|
return b[0], nil
|
|
}
|
|
|
|
func ReadUInt8(r io.Reader) (uint8, error) {
|
|
b, err := ReadBytes(1, r)
|
|
if err != nil {
|
|
return uint8(0), err
|
|
} else {
|
|
return uint8(b[0]), err
|
|
}
|
|
}
|
|
|
|
func ReadUint16LE(r io.Reader) (uint16, error) {
|
|
b := make([]byte, 2)
|
|
_, err := io.ReadFull(r, b)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return binary.LittleEndian.Uint16(b), nil
|
|
}
|
|
|
|
func ReadUint16BE(r io.Reader) (uint16, error) {
|
|
b := make([]byte, 2)
|
|
_, err := io.ReadFull(r, b)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return binary.BigEndian.Uint16(b), nil
|
|
}
|
|
|
|
func ReadUInt32LE(r io.Reader) (uint32, error) {
|
|
b := make([]byte, 4)
|
|
_, err := io.ReadFull(r, b)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return binary.LittleEndian.Uint32(b), nil
|
|
}
|
|
|
|
func ReadUInt32BE(r io.Reader) (uint32, error) {
|
|
b := make([]byte, 4)
|
|
_, err := io.ReadFull(r, b)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return binary.BigEndian.Uint32(b), nil
|
|
}
|
|
|
|
func WriteByte(data byte, w io.Writer) (int, error) {
|
|
b := make([]byte, 1)
|
|
b[0] = byte(data)
|
|
return w.Write(b)
|
|
}
|
|
|
|
func WriteBytes(data []byte, w io.Writer) (int, error) {
|
|
return w.Write(data)
|
|
}
|
|
|
|
func WriteUInt8(data uint8, w io.Writer) (int, error) {
|
|
b := make([]byte, 1)
|
|
b[0] = byte(data)
|
|
return w.Write(b)
|
|
}
|
|
|
|
func WriteUInt16BE(data uint16, w io.Writer) (int, error) {
|
|
b := make([]byte, 2)
|
|
binary.BigEndian.PutUint16(b, data)
|
|
return w.Write(b)
|
|
}
|
|
|
|
func WriteUInt16LE(data uint16, w io.Writer) (int, error) {
|
|
b := make([]byte, 2)
|
|
binary.LittleEndian.PutUint16(b, data)
|
|
return w.Write(b)
|
|
}
|
|
|
|
func WriteUInt32LE(data uint32, w io.Writer) (int, error) {
|
|
b := make([]byte, 4)
|
|
binary.LittleEndian.PutUint32(b, data)
|
|
return w.Write(b)
|
|
}
|
|
|
|
func WriteUInt32BE(data uint32, w io.Writer) (int, error) {
|
|
b := make([]byte, 4)
|
|
binary.BigEndian.PutUint32(b, data)
|
|
return w.Write(b)
|
|
}
|
|
|
|
func PutUint16BE(data uint16) (uint8, uint8) {
|
|
b := make([]byte, 2)
|
|
binary.BigEndian.PutUint16(b, data)
|
|
return uint8(b[0]), uint8(b[1])
|
|
}
|
|
|
|
func Uint16BE(d0, d1 uint8) uint16 {
|
|
b := make([]byte, 2)
|
|
b[0] = d0
|
|
b[1] = d1
|
|
|
|
return binary.BigEndian.Uint16(b)
|
|
}
|
|
|
|
func RGB565ToRGB(data uint16) (r, g, b uint8) {
|
|
r = uint8((data & 0xF800) >> 8)
|
|
r |= r >> 5
|
|
g = uint8((data & 0x07E0) >> 3)
|
|
g |= g >> 6
|
|
b = uint8((data & 0x001F) << 3)
|
|
b |= b >> 5
|
|
|
|
return
|
|
}
|
|
|
|
func RGB555ToRGB(data uint16) (r, g, b uint8) {
|
|
r = uint8((data & 0x7C00) >> 7)
|
|
r |= r >> 5
|
|
g = uint8((data & 0x03E0) >> 2)
|
|
g |= g >> 5
|
|
b = uint8((data & 0x001F) << 3)
|
|
b |= b >> 5
|
|
|
|
return
|
|
}
|