Method handleConnection

internal/sshserver/serverImpl.go:508–642

(conn net.Conn)

}

func (s *serverImpl) handleConnection(conn net.Conn) {
	addr := conn.RemoteAddr().(*net.TCPAddr)
	connectionID := GenerateConnectionID()
	logger := s.logger.
		WithLabel("remoteAddr", addr.IP.String()).
		WithLabel("connectionId", connectionID)
	connectionMeta := metadata.ConnectionMetadata{
		RemoteAddress: metadata.RemoteAddress(*addr),
		ConnectionID:  connectionID,
		Metadata:      map[string]metadata.Value{},
		Environment:   map[string]metadata.Value{},
		Files:         map[string]metadata.BinaryValue{},
	}

	handlerNetworkConnection, connectionMeta, err := s.handler.OnNetworkConnection(connectionMeta)
	if err != nil {
		logger.Info(err)
		_ = conn.Close()
		s.wg.Done()
		return
	}
	shutdownHandlerID := fmt.Sprintf("network-%s", connectionID)
	s.shutdownHandlers.Register(shutdownHandlerID, handlerNetworkConnection)

	logger.Debug(
		messageCodes.NewMessage(
			messageCodes.MSSHConnected, "Client connected",
		),
	)

	// HACK: check HACKS.md "OnHandshakeSuccess conformanceTestHandler"
	wrapper := networkConnectionWrapper{
		NetworkConnectionHandler: handlerNetworkConnection,
	}

	sshConn, channels, globalRequests, err := ssh.NewServerConn(
		conn,
		s.createConfiguration(connectionMeta, &wrapper, logger),
	)
	abortCleanup := func() {
		logger.Info(messageCodes.Wrap(err, messageCodes.ESSHHandshakeFailed, "SSH handshake failed"))
		handlerNetworkConnection.OnHandshakeFailed(connectionMeta, err)
		s.shutdownHandlers.Unregister(shutdownHandlerID)
		logger.Debug(messageCodes.NewMessage(messageCodes.MSSHDisconnected, "Client disconnected"))
		handlerNetworkConnection.OnDisconnect()
		_ = conn.Close()
		s.wg.Done()
	}
	if err != nil {
		abortCleanup()
		return
	}
	var authenticatedMetadata metadata.ConnectionAuthenticatedMetadata
	marshaledMetadata, ok := sshConn.Permissions.Extensions["containerssh-metadata"]
	if !ok {
		abortCleanup()
		return
	}