mirror of
https://github.com/gravitational/teleport.git
synced 2026-10-31 20:39:37 +08:00
rosstimothy
89606e07f0
Completes the service implementation of the transportv1.Service. Ensures that both SSH and SSH Agent protocols are multiplexed correctly over the gRPC stream via TestService_ProxySSH. Part of #19812
127 lines
3.5 KiB
Go
127 lines
3.5 KiB
Go
// Copyright 2022 Gravitational, Inc
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package peer
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/gravitational/trace"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/gravitational/teleport/api/client/proto"
|
|
"github.com/gravitational/teleport/api/types"
|
|
streamutils "github.com/gravitational/teleport/api/utils/grpc/stream"
|
|
"github.com/gravitational/teleport/lib/utils"
|
|
)
|
|
|
|
// proxyService implements the grpc ProxyService.
|
|
type proxyService struct {
|
|
clusterDialer ClusterDialer
|
|
log logrus.FieldLogger
|
|
}
|
|
|
|
// DialNode opens a bidirectional stream to the requested node.
|
|
func (s *proxyService) DialNode(stream proto.ProxyService_DialNodeServer) error {
|
|
frame, err := stream.Recv()
|
|
if err != nil {
|
|
return trace.Wrap(err)
|
|
}
|
|
|
|
// The first frame is always expected to be a dial request.
|
|
dial := frame.GetDialRequest()
|
|
if dial == nil {
|
|
return trace.BadParameter("invalid dial request: request must not be nil")
|
|
}
|
|
|
|
if dial.Source == nil || dial.Destination == nil {
|
|
return trace.BadParameter("invalid dial request: source and destination must not be nil")
|
|
}
|
|
|
|
log := s.log.WithFields(logrus.Fields{
|
|
"node": dial.NodeID,
|
|
"src": dial.Source.Addr,
|
|
"dst": dial.Destination.Addr,
|
|
})
|
|
log.Debugf("Dial request from peer.")
|
|
|
|
_, clusterName, err := splitServerID(dial.NodeID)
|
|
if err != nil {
|
|
return trace.Wrap(err)
|
|
}
|
|
|
|
source := &utils.NetAddr{
|
|
Addr: dial.Source.Addr,
|
|
AddrNetwork: dial.Source.Network,
|
|
}
|
|
destination := &utils.NetAddr{
|
|
Addr: dial.Destination.Addr,
|
|
AddrNetwork: dial.Destination.Network,
|
|
}
|
|
|
|
nodeConn, err := s.clusterDialer.Dial(clusterName, DialParams{
|
|
From: source,
|
|
To: destination,
|
|
ServerID: dial.NodeID,
|
|
ConnType: dial.TunnelType,
|
|
})
|
|
if err != nil {
|
|
return trace.Wrap(err)
|
|
}
|
|
|
|
err = stream.Send(&proto.Frame{
|
|
Message: &proto.Frame_ConnectionEstablished{
|
|
ConnectionEstablished: &proto.ConnectionEstablished{},
|
|
},
|
|
})
|
|
if err != nil {
|
|
return trace.Wrap(err)
|
|
}
|
|
|
|
streamRW, err := streamutils.NewReadWriter(frameStream{stream: stream})
|
|
if err != nil {
|
|
return trace.Wrap(err)
|
|
}
|
|
|
|
streamConn := utils.NewTrackingConn(streamutils.NewConn(streamRW, source, destination))
|
|
|
|
err = utils.ProxyConn(stream.Context(), streamConn, nodeConn)
|
|
sent, received := streamConn.Stat()
|
|
log.Debugf("Closing dial request from peer. sent: %d received %d", sent, received)
|
|
return trace.Wrap(err)
|
|
}
|
|
|
|
// splitServerID splits a server id in to a node id and cluster name.
|
|
func splitServerID(address string) (string, string, error) {
|
|
split := strings.Split(address, ".")
|
|
if len(split) == 0 || split[0] == "" {
|
|
return "", "", trace.BadParameter("invalid server id: \"%s\"", address)
|
|
}
|
|
|
|
return split[0], strings.Join(split[1:], "."), nil
|
|
}
|
|
|
|
// ClusterDialer dials a node in the given cluster.
|
|
type ClusterDialer interface {
|
|
Dial(clusterName string, request DialParams) (net.Conn, error)
|
|
}
|
|
|
|
type DialParams struct {
|
|
From *utils.NetAddr
|
|
To *utils.NetAddr
|
|
ServerID string
|
|
ConnType types.TunnelType
|
|
}
|