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/sshd/listener.go
+++ b/pkg/listener/sshd/listener.go
@ -7,6 +7,7 @@ import (
 	"strconv"
 	"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"
 	sshd_util "github.com/go-gost/gost/pkg/internal/util/sshd"
@ -59,6 +60,7 @@ func (l *sshdListener) 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{