add admission option for listener

2022-03-13 00:08:16 +08:00
parent 4007e80811
commit bfc1f8472c
22 changed files with 348 additions and 5 deletions
--- a/pkg/listener/ssh/listener.go
+++ b/pkg/listener/ssh/listener.go
@ -5,6 +5,7 @@ import (
 	"net"
 	"time"

+	"github.com/go-gost/gost/pkg/common/admission"
 	"github.com/go-gost/gost/pkg/common/metrics"
 	ssh_util "github.com/go-gost/gost/pkg/internal/util/ssh"
 	"github.com/go-gost/gost/pkg/listener"
@ -50,6 +51,7 @@ func (l *sshListener) Init(md md.Metadata) (err error) {
 	}

 	ln = metrics.WrapListener(l.options.Service, ln)
+	ln = admission.WrapListener(l.options.Admission, ln)
 	l.Listener = ln

 	config := &ssh.ServerConfig{