go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add client addr for auth plugin

Browse Source
This commit is contained in:
ginuerzh
2023-09-21 19:59:56 +08:00
parent 1760151500
commit ddc3c9392e
11 changed files with 51 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
auth/plugin.go
View File

@ -10,6 +10,7 @@ import (
"github.com/go-gost/core/auth"
"github.com/go-gost/core/logger"
"github.com/go-gost/plugin/auth/proto"
auth_util "github.com/go-gost/x/internal/util/auth"
"github.com/go-gost/x/internal/util/plugin"
"google.golang.org/grpc"
)
@ -57,6 +58,7 @@ func (p *grpcPluginAuthenticator) Authenticate(ctx context.Context, user, passwo
&proto.AuthenticateRequest{
Username: user,
Password: password,
Client: string(auth_util.ClientAddrFromContext(ctx)),
})
if err != nil {
p.log.Error(err)
@ -75,6 +77,7 @@ func (p *grpcPluginAuthenticator) Close() error {
type httpAutherRequest struct {
Username string `json:"username"`
Password string `json:"password"`
Client string `json:"client"`
}
type httpAutherResponse struct {
@ -115,6 +118,7 @@ func (p *httpPluginAuther) Authenticate(ctx context.Context, user, password stri
rb := httpAutherRequest{
Username: user,
Password: password,
Client: string(auth_util.ClientAddrFromContext(ctx)),
}
v, err := json.Marshal(&rb)
if err != nil {