From 10ca7c9c8510f3150c4a315cc1dba74bbbb6af32 Mon Sep 17 00:00:00 2001 From: bjdgyc Date: Fri, 23 Sep 2022 17:26:19 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=BF=94=E5=9B=9E=E9=94=99?= =?UTF-8?q?=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/handler/dtls.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/server/handler/dtls.go b/server/handler/dtls.go index 441c18a..7501246 100644 --- a/server/handler/dtls.go +++ b/server/handler/dtls.go @@ -4,6 +4,7 @@ import ( "context" "crypto/tls" "encoding/hex" + "errors" "net" "time" @@ -82,12 +83,12 @@ func (ms *sessionStore) Set(key []byte, s dtls.Session) error { func (ms *sessionStore) Get(key []byte) (dtls.Session, error) { k := hex.EncodeToString(key) secret := sessdata.Dtls2MasterSecret(k) - if secret != "" { - masterSecret, _ := hex.DecodeString(secret) - return dtls.Session{ID: key, Secret: masterSecret}, nil + if secret == "" { + return dtls.Session{}, errors.New("Dtls2MasterSecret is nil") } - return dtls.Session{}, nil + masterSecret, _ := hex.DecodeString(secret) + return dtls.Session{ID: key, Secret: masterSecret}, nil } func (ms *sessionStore) Del(key []byte) error {