// Copyright 2026 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "remoting/host/linux/login_session_server.h"

#include <systemd/sd-login.h>

#include <string>
#include <utility>
#include <vector>

#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/logging.h"
#include "base/posix/safe_strerror.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "components/named_mojo_ipc_server/connection_info.h"
#include "components/named_mojo_ipc_server/endpoint_options.h"
#include "remoting/base/logging.h"
#include "remoting/host/ipc_constants.h"
#include "remoting/host/mojo_caller_security_checker.h"

namespace remoting {
namespace {

named_mojo_ipc_server::EndpointOptions CreateEndpointOptions() {
  named_mojo_ipc_server::EndpointOptions options;
  options.server_name = GetLoginSessionServerName();
  options.message_pipe_id = kLoginSessionServerMessagePipeId;
  options.require_same_peer_user = false;
  return options;
}

bool IsRunningInGraphicalSession(base::ProcessId pid) {
  std::string environ_content;
  if (!base::ReadFileToString(
          base::FilePath(base::StringPrintf("/proc/%d/environ", pid)),
          &environ_content)) {
    return false;
  }

  std::vector<std::string> env_vars =
      base::SplitString(environ_content, std::string(1, '\0'),
                        base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
  for (const std::string& env_var : env_vars) {
    if (env_var == "XDG_SESSION_TYPE=x11" ||
        env_var == "XDG_SESSION_TYPE=wayland") {
      return true;
    }
  }
  return false;
}

std::string GetSessionIdForPid(base::ProcessId pid) {
  char* session_id = nullptr;
  int ret = sd_pid_get_session(pid, &session_id);
  if (ret < 0 && IsRunningInGraphicalSession(pid)) {
    // If the process is not directly in a session scope (e.g. running under
    // the systemd user manager), sd_pid_get_session() returns -ENODATA.
    // For modern GDM, each user can only have one graphical session, so it
    // falls back to the user's primary graphical display session.
    HOST_LOG << "Failed to get session ID for PID " << pid << ": "
             << base::safe_strerror(-ret)
             << ". Falling back to the user's display session.";
    uid_t uid;
    if (sd_pid_get_owner_uid(pid, &uid) == 0) {
      ret = sd_uid_get_display(uid, &session_id);
    }
  }
  if (ret < 0) {
    LOG(ERROR) << "Failed to get session ID for PID " << pid
               << ", error: " << base::safe_strerror(-ret);
    return {};
  }
  std::string session_id_string = session_id;
  free(session_id);
  return session_id_string;
}

}  // namespace

LoginSessionServer::LoginSessionServer(Delegate* delegate)
    : delegate_(delegate),
      ipc_server_(
          CreateEndpointOptions(),
          base::BindRepeating(IsTrustedMojoEndpoint)
              .Then(base::BindRepeating(
                  [](mojom::LoginSessionService* service, bool is_valid) {
                    return is_valid ? service : nullptr;
                  },
                  this))) {}

LoginSessionServer::~LoginSessionServer() = default;

void LoginSessionServer::StartServer() {
  ipc_server_.StartServer();
}

void LoginSessionServer::StopServer() {
  ipc_server_.StopServer();
}

void LoginSessionServer::IsRunningInCrdSession(
    IsRunningInCrdSessionCallback callback) {
  std::string session_id =
      GetSessionIdForPid(ipc_server_.current_connection_info().pid);
  if (session_id.empty()) {
    std::move(callback).Run(false);
    return;
  }
  delegate_->IsRunningInCrdSession(session_id, std::move(callback));
}

}  // namespace remoting