mirror of
https://github.com/gravitational/teleport.git
synced 2026-10-30 18:29:21 +08:00
Convert lib/srv/transport to use slog (#49914)
This commit is contained in:
@@ -4932,7 +4932,7 @@ func (process *TeleportProcess) initProxyEndpoint(conn *Connector) error {
|
||||
|
||||
transportService, err := transportv1.NewService(transportv1.ServerConfig{
|
||||
FIPS: cfg.FIPS,
|
||||
Logger: process.log.WithField(teleport.ComponentKey, "transport"),
|
||||
Logger: process.logger.With(teleport.ComponentKey, "transport"),
|
||||
Dialer: proxyRouter,
|
||||
SignerFn: func(authzCtx *authz.Context, clusterName string) agentless.SignerCreator {
|
||||
return agentless.SignerFromAuthzContext(authzCtx, accessPoint, clusterName)
|
||||
|
||||
@@ -22,12 +22,12 @@ import (
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
"log/slog"
|
||||
"net"
|
||||
"net/netip"
|
||||
"sync"
|
||||
|
||||
"github.com/gravitational/trace"
|
||||
"github.com/sirupsen/logrus"
|
||||
"golang.org/x/crypto/ssh/agent"
|
||||
"google.golang.org/grpc/codes"
|
||||
"google.golang.org/grpc/credentials"
|
||||
@@ -43,6 +43,7 @@ import (
|
||||
"github.com/gravitational/teleport/lib/services"
|
||||
"github.com/gravitational/teleport/lib/teleagent"
|
||||
"github.com/gravitational/teleport/lib/utils"
|
||||
logutils "github.com/gravitational/teleport/lib/utils/log"
|
||||
)
|
||||
|
||||
// Dialer is the interface that groups basic dialing methods.
|
||||
@@ -63,7 +64,7 @@ type ServerConfig struct {
|
||||
// to run in FIPS mode.
|
||||
FIPS bool
|
||||
// Logger provides a mechanism to log output.
|
||||
Logger logrus.FieldLogger
|
||||
Logger *slog.Logger
|
||||
// Dialer is used to establish remote connections.
|
||||
Dialer Dialer
|
||||
// SignerFn is used to create an [ssh.Signer] for an authenticated connection.
|
||||
@@ -93,7 +94,7 @@ func (c *ServerConfig) CheckAndSetDefaults() error {
|
||||
}
|
||||
|
||||
if c.Logger == nil {
|
||||
c.Logger = utils.NewLogger().WithField(teleport.ComponentKey, "transport")
|
||||
c.Logger = slog.With(teleport.ComponentKey, "transport")
|
||||
}
|
||||
|
||||
if c.agentGetterFn == nil {
|
||||
@@ -247,7 +248,7 @@ func (s *Service) ProxySSH(stream transportv1pb.TransportService_ProxySSHServer)
|
||||
req, err := stream.Recv()
|
||||
if err != nil {
|
||||
if !utils.IsOKNetworkError(err) && !errors.Is(err, context.Canceled) && status.Code(err) != codes.Canceled {
|
||||
s.cfg.Logger.Errorf("ssh stream terminated unexpectedly: %v", err)
|
||||
s.cfg.Logger.ErrorContext(ctx, "ssh stream terminated unexpectedly", "error", err)
|
||||
}
|
||||
|
||||
return
|
||||
@@ -262,7 +263,7 @@ func (s *Service) ProxySSH(stream transportv1pb.TransportService_ProxySSHServer)
|
||||
case *transportv1pb.ProxySSHRequest_Agent:
|
||||
agentStream.incomingC <- frame.Agent.Payload
|
||||
default:
|
||||
s.cfg.Logger.Errorf("received unexpected ssh frame: %T", frame)
|
||||
s.cfg.Logger.ErrorContext(ctx, "received unexpected ssh frame", "frame", logutils.TypeAttr(frame))
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
@@ -256,7 +256,7 @@ func TestService_GetClusterDetails(t *testing.T) {
|
||||
t.Parallel()
|
||||
srv := newServer(t, ServerConfig{
|
||||
Dialer: fakeDialer{},
|
||||
Logger: utils.NewLoggerForTests(),
|
||||
Logger: utils.NewSlogLoggerForTests(),
|
||||
FIPS: test.FIPS,
|
||||
SignerFn: fakeSigner,
|
||||
ConnectionMonitor: fakeMonitor{},
|
||||
@@ -339,7 +339,7 @@ func TestService_ProxyCluster(t *testing.T) {
|
||||
cluster: conn,
|
||||
},
|
||||
},
|
||||
Logger: utils.NewLoggerForTests(),
|
||||
Logger: utils.NewSlogLoggerForTests(),
|
||||
SignerFn: fakeSigner,
|
||||
ConnectionMonitor: fakeMonitor{},
|
||||
LocalAddr: utils.MustParseAddr("127.0.0.1:4242"),
|
||||
@@ -491,7 +491,7 @@ func TestService_ProxySSH_Errors(t *testing.T) {
|
||||
},
|
||||
SignerFn: fakeSigner,
|
||||
ConnectionMonitor: fakeMonitor{},
|
||||
Logger: utils.NewLoggerForTests(),
|
||||
Logger: utils.NewSlogLoggerForTests(),
|
||||
LocalAddr: utils.MustParseAddr("127.0.0.1:4242"),
|
||||
authzContextFn: func(info credentials.AuthInfo) (*authz.Context, error) {
|
||||
checker, err := test.checkerFn(info)
|
||||
@@ -554,7 +554,7 @@ func TestService_ProxySSH(t *testing.T) {
|
||||
srv := newServer(t, ServerConfig{
|
||||
Dialer: sshSrv,
|
||||
SignerFn: fakeSigner,
|
||||
Logger: utils.NewLoggerForTests(),
|
||||
Logger: utils.NewSlogLoggerForTests(),
|
||||
LocalAddr: utils.MustParseAddr("127.0.0.1:4242"),
|
||||
ConnectionMonitor: fakeMonitor{},
|
||||
agentGetterFn: func(rw io.ReadWriter) teleagent.Getter {
|
||||
|
||||
@@ -8308,7 +8308,7 @@ func createProxy(ctx context.Context, t *testing.T, proxyID string, node *regula
|
||||
|
||||
transportService, err := transportv1.NewService(transportv1.ServerConfig{
|
||||
FIPS: false,
|
||||
Logger: utils.NewLoggerForTests(),
|
||||
Logger: utils.NewSlogLoggerForTests(),
|
||||
Dialer: router,
|
||||
SignerFn: func(authzCtx *authz.Context, clusterName string) agentless.SignerCreator {
|
||||
return agentless.SignerFromAuthzContext(authzCtx, client, clusterName)
|
||||
|
||||
Reference in New Issue
Block a user