From 892dc502b4e4ed3fc29016601130345a598ebffb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Mon, 28 Mar 2022 17:20:43 +0800
Subject: [PATCH] fix: Close

---
 reg.go | 32 ++++++++++++--------------------
 1 file changed, 12 insertions(+), 20 deletions(-)

diff --git a/reg.go b/reg.go
index 82ac1d9..d6941cf 100644
--- a/reg.go
+++ b/reg.go
@@ -23,12 +23,11 @@ var (
 
 type Regedit struct {
 	sync.Mutex
-	conn   net.Conn
-	addr   string
-	tp     tea.TEA
-	ts     *tea.TEA
-	seq    byte
-	isopen bool
+	conn net.Conn
+	addr string
+	tp   tea.TEA
+	ts   *tea.TEA
+	seq  byte
 }
 
 func NewRegedit(addr, pwd, sps string) *Regedit {
@@ -55,38 +54,31 @@ func NewRegReader(addr, pwd string) *Regedit {
 }
 
 func (r *Regedit) Connect() (err error) {
-	if !r.isopen {
-		r.Lock()
+	r.Lock()
+	if r.conn == nil {
 		r.conn, err = net.Dial("tcp", r.addr)
-		if err == nil {
-			r.isopen = true
-		}
-		r.Unlock()
 	}
+	r.Unlock()
 	return
 }
 
 func (r *Regedit) ConnectIn(wait time.Duration) (err error) {
-	if !r.isopen {
-		r.Lock()
+	r.Lock()
+	if r.conn == nil {
 		r.conn, err = net.DialTimeout("tcp", r.addr, wait)
-		if err == nil {
-			r.isopen = true
-		}
-		r.Unlock()
 	}
+	r.Unlock()
 	return
 }
 
 func (r *Regedit) Close() (err error) {
 	r.Lock()
 	defer r.Unlock()
-	if r.isopen {
+	if r.conn != nil {
 		p := NewCmdPacket(CMDEND, []byte("fill"), &r.tp)
 		r.conn.Write(p.Encrypt(r.seq))
 		p.Put()
 		r.seq = 0
-		r.isopen = false
 		err = r.conn.Close()
 		r.conn = nil
 		return