update handle associate

bepass-org · Aug 6, 2023 · 03046ec · 03046ec
1 parent c05d344
commit 03046ec
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bepass-cli/socks5/handle.go b/bepass-cli/socks5/handle.go
@@ -172,17 +172,17 @@ func (sf *Server) handleAssociate(ctx context.Context, writer io.Writer, request
 	if dest.FQDN != "" {
 		ctx, dest.IP, err = sf.resolver.Resolve(ctx, dest.FQDN)
 		if err != nil {
-			if err := SendReply(write, statute.RepHostUnreachable, nil); err != nil {
+			if err := SendReply(writer, statute.RepHostUnreachable, nil); err != nil {
 				return fmt.Errorf("failed to send reply, %v", err)
 			}
 			return fmt.Errorf("failed to resolve destination[%v], %v", dest.FQDN, err)
 		}
 	}
 
 	// Apply any address rewrites
-	req.DestAddr = req.RawDestAddr
+	request.DestAddr = request.RawDestAddr
 	if sf.rewriter != nil {
-		ctx, req.DestAddr = sf.rewriter.Rewrite(ctx, req)
+		ctx, request.DestAddr = sf.rewriter.Rewrite(ctx, request)
 	}
 	// Attempt to connect
 	dial := sf.dial