From 02c4f9e3d1b055db6184fde0a280e15a2400325a Mon Sep 17 00:00:00 2001
From: mklee <mklee@action2quare.com>
Date: Wed, 23 Aug 2023 18:53:24 +0900
Subject: [PATCH] =?UTF-8?q?[=EC=9D=B4=EB=AF=BC=EA=B6=8C]=20link,=20unlink?=
 =?UTF-8?q?=20=EC=9D=B4=EC=8A=88=20=EC=88=98=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 core/service.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/core/service.go b/core/service.go
index c9e2d48..123eae2 100644
--- a/core/service.go
+++ b/core/service.go
@@ -329,6 +329,13 @@ func (sh *serviceDescription) link(w http.ResponseWriter, r *http.Request) {
 	// fmt.Println(oldAuth.Uid)
 	// fmt.Println("=================")
 
+	oldType, oldId, err := sh.getProviderInfo(oldType, oldId)
+	if err != nil {
+		logger.Error("getProviderInfo failed :", err)
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
 	//if oldAuth.Token != oldToken || oldAuth.Uid != oldId || oldAuth.Platform != oldType {
 	if oldAuth.Uid != oldId || oldAuth.Platform != oldType {
 		logger.Println("link failed. session key is not correct :", *oldAuth, queryvals)
@@ -452,6 +459,13 @@ func (sh *serviceDescription) unlink(w http.ResponseWriter, r *http.Request) {
 	// fmt.Println(authInfo.Uid)
 	// fmt.Println("=================")
 
+	sType, sId, err := sh.getProviderInfo(sType, sId)
+	if err != nil {
+		logger.Error("getProviderInfo failed :", err)
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
 	if authInfo.Uid != sId || authInfo.Platform != sType {
 		logger.Println("unlink failed. session key is not correct :", *authInfo, queryvals)
 		w.WriteHeader(http.StatusBadRequest)