diff --git a/handler/private.go b/handler/private.go index c550b47..d1aa93c 100644 --- a/handler/private.go +++ b/handler/private.go @@ -126,6 +126,9 @@ func HandshakeHandler(ctx *gin.Context) { CheckErr(err) randKey := params.Get("randkey") deviceId := params.Get("deviceid") + if deviceId == "" { + deviceId = ctx.Request.Header.Get("X-DEVICEID") + } // fmt.Println(randKey) // fmt.Println(deviceId)