add admission

pkg/listener/grpc/server.go

@@ -8,6 +8,7 @@ import (
 
 	pb "github.com/go-gost/gost/pkg/common/util/grpc/proto"
 	"github.com/go-gost/gost/pkg/logger"
+	"google.golang.org/grpc/peer"
 )
 
 type server struct {
@@ -24,6 +25,9 @@ func (s *server) Tunnel(srv pb.GostTunel_TunnelServer) error {
 		remoteAddr: &net.TCPAddr{},
 		closed:     make(chan struct{}),
 	}
+	if p, ok := peer.FromContext(srv.Context()); ok {
+		c.remoteAddr = p.Addr
+	}
 
 	select {
 	case s.cqueue <- c: