From 4fdcab32db126307db9f7c19deb3003de51fc8b2 Mon Sep 17 00:00:00 2001
From: number571 <number571gg@mail.ru>
Date: Thu, 24 Oct 2024 14:06:46 +0700
Subject: [PATCH] update

---
 pkg/client/client.go            | 9 +++++----
 test/result/badge_codelines.svg | 2 +-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/pkg/client/client.go b/pkg/client/client.go
index 39b61736..51890eb6 100644
--- a/pkg/client/client.go
+++ b/pkg/client/client.go
@@ -94,7 +94,7 @@ func (p *sClient) encryptWithParams(
 
 	data := joiner.NewBytesJoiner32([][]byte{pMsg, rand.GetBytes(pPadd)})
 	hash := hashing.NewHMACHasher(salt, bytes.Join(
-		[][]byte{pkid, pRecv.ToBytes(), data},
+		[][]byte{pkid, pRecv.GetHasher().ToBytes(), data},
 		[]byte{},
 	)).ToBytes()
 
@@ -146,6 +146,7 @@ func (p *sClient) DecryptMessage(pMapPubKeys asymmetric.IMapPubKeys, pMsg []byte
 		data = decSlice[4]
 	)
 
+	// Get public key from map by pkid (hash)
 	sPubKey := pMapPubKeys.GetPubKey(pkid)
 	if sPubKey == nil {
 		return nil, nil, ErrDecodePublicKey
@@ -154,13 +155,13 @@ func (p *sClient) DecryptMessage(pMapPubKeys asymmetric.IMapPubKeys, pMsg []byte
 	// Validate received hash with generated hash.
 	check := hashing.NewHMACHasher(salt, bytes.Join(
 		[][]byte{
-			sPubKey.ToBytes(),
-			p.fPrivKey.GetPubKey().ToBytes(),
+			sPubKey.GetHasher().ToBytes(),
+			p.fPrivKey.GetPubKey().GetHasher().ToBytes(),
 			data,
 		},
 		[]byte{},
 	)).ToBytes()
-	if bytes.Equal(check, hash) {
+	if !bytes.Equal(check, hash) {
 		return nil, nil, ErrInvalidDataHash
 	}
 
diff --git a/test/result/badge_codelines.svg b/test/result/badge_codelines.svg
index 6751f476..d8a72e99 100644
--- a/test/result/badge_codelines.svg
+++ b/test/result/badge_codelines.svg
@@ -1 +1 @@
-<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="96" height="20" role="img" aria-label="code lines: 12546"><title>code lines: 12546</title><linearGradient id="s" x2="0" y2="100%"><stop offset="0" stop-color="#bbb" stop-opacity=".1"/><stop offset="1" stop-opacity=".1"/></linearGradient><clipPath id="r"><rect width="96" height="20" rx="3" fill="#fff"/></clipPath><g clip-path="url(#r)"><rect width="57" height="20" fill="#555"/><rect x="57" width="40" height="20" fill="#4682b4"/><rect width="96" height="20" fill="url(#s)"/></g><g fill="#fff" text-anchor="middle" font-family="Verdana,Geneva,DejaVu Sans,sans-serif" text-rendering="geometricPrecision" font-size="110"><text aria-hidden="true" x="295" y="150" fill="#010101" fill-opacity=".3" transform="scale(.1)" textLength="510">code lines</text><text x="295" y="140" transform="scale(.1)" fill="#fff" textLength="510">code lines</text><text aria-hidden="true" x="770" y="150" fill="#010101" fill-opacity=".3" transform="scale(.1)" textLength="320">12546</text><text x="770" y="140" transform="scale(.1)" fill="#fff" textLength="320">12546</text></g></svg>
\ No newline at end of file
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" width="96" height="20" role="img" aria-label="code lines: 12547"><title>code lines: 12547</title><linearGradient id="s" x2="0" y2="100%"><stop offset="0" stop-color="#bbb" stop-opacity=".1"/><stop offset="1" stop-opacity=".1"/></linearGradient><clipPath id="r"><rect width="96" height="20" rx="3" fill="#fff"/></clipPath><g clip-path="url(#r)"><rect width="57" height="20" fill="#555"/><rect x="57" width="40" height="20" fill="#4682b4"/><rect width="96" height="20" fill="url(#s)"/></g><g fill="#fff" text-anchor="middle" font-family="Verdana,Geneva,DejaVu Sans,sans-serif" text-rendering="geometricPrecision" font-size="110"><text aria-hidden="true" x="295" y="150" fill="#010101" fill-opacity=".3" transform="scale(.1)" textLength="510">code lines</text><text x="295" y="140" transform="scale(.1)" fill="#fff" textLength="510">code lines</text><text aria-hidden="true" x="770" y="150" fill="#010101" fill-opacity=".3" transform="scale(.1)" textLength="320">12547</text><text x="770" y="140" transform="scale(.1)" fill="#fff" textLength="320">12547</text></g></svg>
\ No newline at end of file