From aa39d5a1c96aeedaa796ef3124828228830ebf6e 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, 21 Nov 2022 18:54:40 +0800
Subject: [PATCH] fix

---
 reg.go | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/reg.go b/reg.go
index dbec412..a06bde5 100644
--- a/reg.go
+++ b/reg.go
@@ -50,14 +50,16 @@ func NewRegedit(addr, stor, pwd, sps string) *Regedit {
 	copy(tp[:], pwd)
 	copy(ts[:], sps)
 	s := tea.NewTeaCipherLittleEndian(ts[:])
-	f, err := os.Open(stor)
-	if err != nil {
-		f, err = os.Create(stor)
+	if stor != "" {
+		f, err := os.Open(stor)
 		if err != nil {
-			panic(err)
+			f, err = os.Create(stor)
+			if err != nil {
+				panic(err)
+			}
 		}
+		_ = f.Close()
 	}
-	_ = f.Close()
 	return &Regedit{addr: addr, stor: stor, tp: tea.NewTeaCipherLittleEndian(tp[:]), ts: &s}
 }
 
@@ -67,14 +69,16 @@ func NewRegReader(addr, stor, pwd string) *Regedit {
 		pwd = pwd[:15]
 	}
 	copy(tp[:], pwd)
-	f, err := os.Open(stor)
-	if err != nil {
-		f, err = os.Create(stor)
+	if stor != "" {
+		f, err := os.Open(stor)
 		if err != nil {
-			panic(err)
+			f, err = os.Create(stor)
+			if err != nil {
+				panic(err)
+			}
 		}
+		_ = f.Close()
 	}
-	_ = f.Close()
 	return &Regedit{addr: addr, stor: stor, tp: tea.NewTeaCipherLittleEndian(tp[:])}
 }