From 24dd419a9c03c689ca5ff7d8917315a865e1c284 Mon Sep 17 00:00:00 2001
From: deatil <2217957370@qq.com>
Date: Wed, 7 Feb 2024 12:52:16 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gm/sm2/sm2.go | 25 +++++++++++++------------
 1 file changed, 13 insertions(+), 12 deletions(-)

diff --git a/gm/sm2/sm2.go b/gm/sm2/sm2.go
index d590878c..a2f41831 100644
--- a/gm/sm2/sm2.go
+++ b/gm/sm2/sm2.go
@@ -340,13 +340,14 @@ func encrypt(random io.Reader, pub *PublicKey, data []byte) ([]byte, error) {
         c = append(c, x1Buf...) // x分量
         c = append(c, y1Buf...) // y分量
 
-        tm := []byte{}
-        tm = append(tm, x2Buf...)
-        tm = append(tm, data...)
-        tm = append(tm, y2Buf...)
+        md := sm3.New()
+        md.Write(x2Buf)
+        md.Write(data)
+        md.Write(y2Buf)
 
-        h := sm3.Sum(tm)
-        c = append(c, h[:]...)
+        h := md.Sum(nil)
+
+        c = append(c, h...)
 
         // 生成密钥 / make key
         ct, ok := kdf(length, x2Buf, y2Buf)
@@ -392,14 +393,14 @@ func decrypt(priv *PrivateKey, data []byte) ([]byte, error) {
     // 解密密文 / decrypt data
     subtle.XORBytes(c, c, data)
 
-    tm := []byte{}
-    tm = append(tm, x2Buf...)
-    tm = append(tm, c...)
-    tm = append(tm, y2Buf...)
+    md := sm3.New()
+    md.Write(x2Buf)
+    md.Write(c)
+    md.Write(y2Buf)
 
-    h := sm3.Sum(tm)
+    h := md.Sum(nil)
 
-    if bytes.Compare(h[:], hash) != 0 {
+    if bytes.Compare(h, hash) != 0 {
         return c, errors.New("cryptobin/sm2: failed to decrypt")
     }