refactor: mylib 重命名为 libs

更清晰的目录命名，libs/ 存放内嵌的独立协议实现库。

libs/grdp/core/io.go

@@ -0,0 +1,132 @@
+package core
+
+import (
+	"encoding/binary"
+	"github.com/shadow1ng/fscan/libs/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)
+}

libs/grdp/core/socket.go

@@ -0,0 +1,81 @@
+package core
+
+import (
+	"crypto/rsa"
+	"math/big"
+
+	"github.com/huin/asn1ber"
+
+	//"crypto/tls"
+	"errors"
+	"github.com/icodeface/tls"
+	"net"
+)
+
+type SocketLayer struct {
+	conn    net.Conn
+	tlsConn *tls.Conn
+}
+
+func NewSocketLayer(conn net.Conn) *SocketLayer {
+	l := &SocketLayer{
+		conn:    conn,
+		tlsConn: nil,
+	}
+	return l
+}
+
+func (s *SocketLayer) Read(b []byte) (n int, err error) {
+	if s.tlsConn != nil {
+		return s.tlsConn.Read(b)
+	}
+	return s.conn.Read(b)
+}
+
+func (s *SocketLayer) Write(b []byte) (n int, err error) {
+	if s.tlsConn != nil {
+		return s.tlsConn.Write(b)
+	}
+	return s.conn.Write(b)
+}
+
+func (s *SocketLayer) Close() error {
+	if s.tlsConn != nil {
+		err := s.tlsConn.Close()
+		if err != nil {
+			return err
+		}
+	}
+	return s.conn.Close()
+}
+
+func (s *SocketLayer) StartTLS() error {
+	config := &tls.Config{
+		InsecureSkipVerify:       true,
+		MinVersion:               tls.VersionTLS10,
+		MaxVersion:               tls.VersionTLS13,
+		PreferServerCipherSuites: true,
+	}
+	s.tlsConn = tls.Client(s.conn, config)
+	return s.tlsConn.Handshake()
+}
+
+type PublicKey struct {
+	N *big.Int `asn1:"explicit,tag:0"` // modulus
+	E int      `asn1:"explicit,tag:1"` // public exponent
+}
+
+func (s *SocketLayer) TlsPubKey() ([]byte, error) {
+	if s.tlsConn == nil {
+		return nil, errors.New("TLS conn does not exist")
+	}
+	certs := s.tlsConn.ConnectionState().PeerCertificates
+	if len(certs) == 0 {
+		return nil, errors.New("no peer certificates")
+	}
+	pub, ok := certs[0].PublicKey.(*rsa.PublicKey)
+	if !ok {
+		return nil, errors.New("invalid public key type")
+	}
+	return asn1ber.Marshal(*pub)
+}

libs/grdp/core/types.go

@@ -0,0 +1,25 @@
+package core
+
+import "github.com/shadow1ng/fscan/libs/grdp/emission"
+
+type Transport interface {
+	Read(b []byte) (n int, err error)
+	Write(b []byte) (n int, err error)
+	Close() error
+
+	On(event, listener interface{}) *emission.Emitter
+	Once(event, listener interface{}) *emission.Emitter
+	Emit(event interface{}, arguments ...interface{}) *emission.Emitter
+}
+
+type FastPathListener interface {
+	RecvFastPath(secFlag byte, s []byte)
+}
+
+type FastPathSender interface {
+	SendFastPath(secFlag byte, s []byte) (int, error)
+}
+
+type ChannelSender interface {
+	SendToChannel(channel string, s []byte) (int, error)
+}

libs/grdp/core/util.go

@@ -0,0 +1,67 @@
+package core
+
+import (
+	"bytes"
+	"crypto/rand"
+	"encoding/binary"
+	"unicode/utf16"
+)
+
+func Reverse(s []byte) []byte {
+	for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
+		s[i], s[j] = s[j], s[i]
+	}
+	return s
+}
+
+func Random(n int) []byte {
+	const alpha = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
+	var bytes = make([]byte, n)
+	rand.Read(bytes)
+	for i, b := range bytes {
+		bytes[i] = alpha[b%byte(len(alpha))]
+	}
+	return bytes
+}
+
+func UTF16ToLittleEndianBytes(u []uint16) []byte {
+	b := make([]byte, 2*len(u))
+	for index, value := range u {
+		binary.LittleEndian.PutUint16(b[index*2:], value)
+	}
+	return b
+}
+
+func LittleEndianBytesToUTF16(u []byte) []uint16 {
+	b := make([]uint16, 0, len(u)/2)
+	n := make([]byte, 2)
+	for i, v := range u {
+		if i%2 == 0 {
+			n[0] = v
+		} else {
+			n[1] = v
+			b = append(b, binary.LittleEndian.Uint16(n))
+		}
+	}
+	return b
+}
+
+// s.encode('utf-16le')
+func UnicodeEncode(p string) []byte {
+	return UTF16ToLittleEndianBytes(utf16.Encode([]rune(p)))
+}
+
+func UnicodeDecode(p []byte) string {
+	r := bytes.NewReader(p)
+	n := make([]uint16, 0, 100)
+	for r.Len() > 0 {
+		a, _ := ReadUint16LE(r)
+		n = append(n, a)
+	}
+	//n := LittleEndianBytesToUTF16(p)
+	return string(utf16.Decode(n))
+}
+
+func BytesToUint64(b []byte) uint64 {
+	return binary.LittleEndian.Uint64(b)
+}