基本适应win7,win10,win server 08,win server 12,win server 16的截图

grdp/core/socket.go

@@ -0,0 +1,74 @@
+package core
+
+import (
+	"crypto/rsa"
+	"math/big"
+
+	"github.com/huin/asn1ber"
+
+	//"crypto/tls"
+	"errors"
+	"net"
+	"github.com/icodeface/tls"
+)
+
+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")
+	}
+	pub := s.tlsConn.ConnectionState().PeerCertificates[0].PublicKey.(*rsa.PublicKey)
+	return asn1ber.Marshal(*pub)
+}