From 07dbdd80ecc33e276effb412ccb4e3133cd4ca0d Mon Sep 17 00:00:00 2001 From: LittleSheep Date: Sun, 12 May 2024 19:41:00 +0800 Subject: [PATCH] :bug: Fix key exchange cause echo --- .idea/workspace.xml | 54 ++++++++++++++++++++++----------------------- pkg/server/ws.go | 2 +- 2 files changed, 27 insertions(+), 29 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index cfe426d..b25ca93 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -4,10 +4,8 @@ - @@ -165,7 +162,8 @@ - true diff --git a/pkg/server/ws.go b/pkg/server/ws.go index ecdf629..ae1bf31 100644 --- a/pkg/server/ws.go +++ b/pkg/server/ws.go @@ -49,7 +49,7 @@ func listenWebsocket(c *websocket.Conn) { if len(req.RequestID) <= 0 || len(req.KeypairID) <= 0 || req.OwnerID <= 0 { message = lo.ToPtr(models.UnifiedCommandFromError(fmt.Errorf("invalid request"))) } - services.KexRequest(c, req.RequestID, req.KeypairID, user.ID, req.Deadline) + services.KexRequest(c, req.RequestID, req.KeypairID, req.OwnerID, req.Deadline) case "kex.provide": var req struct { RequestID string `json:"request_id"`