This commit is contained in:
Bolke de Bruin 2024-03-18 15:27:30 +01:00
parent f70348c18a
commit 46620c87b7
2 changed files with 11 additions and 4 deletions

View file

@ -3,8 +3,9 @@ package main
import ( import (
"context" "context"
"errors" "errors"
"fmt"
"github.com/bolkedebruin/rdpgw/shared/auth" "github.com/bolkedebruin/rdpgw/shared/auth"
"github.com/msteinert/pam" "github.com/msteinert/pam/v2"
"github.com/thought-machine/go-flags" "github.com/thought-machine/go-flags"
"google.golang.org/grpc" "google.golang.org/grpc"
"log" "log"
@ -52,7 +53,13 @@ func (s *AuthServiceImpl) Authenticate(ctx context.Context, message *auth.UserPa
r.Error = err.Error() r.Error = err.Error()
return r, err return r, err
} }
defer func() {
err := t.End()
if err != nil {
fmt.Fprintf(os.Stderr, "end: %v\n", err)
os.Exit(1)
}
}()
if err = t.Authenticate(0); err != nil { if err = t.Authenticate(0); err != nil {
log.Printf("Authentication for user: %s failed due to: %s", message.Username, err) log.Printf("Authentication for user: %s failed due to: %s", message.Username, err)
r.Error = err.Error() r.Error = err.Error()

4
go.mod
View file

@ -18,10 +18,10 @@ require (
github.com/knadh/koanf/providers/env v0.1.0 github.com/knadh/koanf/providers/env v0.1.0
github.com/knadh/koanf/providers/file v0.1.0 github.com/knadh/koanf/providers/file v0.1.0
github.com/knadh/koanf/v2 v2.1.0 github.com/knadh/koanf/v2 v2.1.0
github.com/msteinert/pam v1.2.0 github.com/msteinert/pam/v2 v2.0.0
github.com/patrickmn/go-cache v2.1.0+incompatible github.com/patrickmn/go-cache v2.1.0+incompatible
github.com/prometheus/client_golang v1.19.0 github.com/prometheus/client_golang v1.19.0
github.com/stretchr/testify v1.8.1 github.com/stretchr/testify v1.9.0
github.com/thought-machine/go-flags v1.6.3 github.com/thought-machine/go-flags v1.6.3
golang.org/x/crypto v0.21.0 golang.org/x/crypto v0.21.0
golang.org/x/oauth2 v0.18.0 golang.org/x/oauth2 v0.18.0