Code/Editor/EditorEngineProcessFramework/EngineProcess/EngineProcessCommunicationChannel.cpp - dev - redgpu/ezEngine

Code/Editor/EditorEngineProcessFramework/EngineProcess/EngineProcessCommunicationChannel.cpp

84 строки3 KB

Sanakan8472

Improved IPC connection state handling (#1960)

09 июн 2026, 14:55

09 июн 2026, 14:55ac44884

#include <EditorEngineProcessFramework/EditorEngineProcessFrameworkPCH.h>

#include <EditorEngineProcessFramework/EngineProcess/EngineProcessApp.h>
#include <EditorEngineProcessFramework/EngineProcess/EngineProcessCommunicationChannel.h>
#include <Foundation/Communication/IpcChannel.h>
#include <Foundation/Communication/IpcProcessMessageProtocol.h>

#if EZ_ENABLED(EZ_PLATFORM_WINDOWS_DESKTOP)
#  include <Foundation/Platform/Win/Utils/IncludeWindows.h>
#elif EZ_ENABLED(EZ_PLATFORM_LINUX)
#  include <signal.h>
#endif

bool ezEngineProcessCommunicationChannel::IsHostAlive() const
{
  if (ezEditorEngineProcessApp::GetSingleton()->IsRemoteMode())
    return true;

  if (m_iHostPID == 0)
    return false;

  bool bValid = true;

#if EZ_ENABLED(EZ_PLATFORM_WINDOWS_DESKTOP)
  DWORD pid = static_cast<DWORD>(m_iHostPID);
  HANDLE hProcess = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pid);
  bValid = (hProcess != INVALID_HANDLE_VALUE) && (hProcess != nullptr);

  DWORD exitcode = 0;
  if (GetExitCodeProcess(hProcess, &exitcode) && exitcode != STILL_ACTIVE)
    bValid = false;

  CloseHandle(hProcess);
#elif EZ_ENABLED(EZ_PLATFORM_LINUX)
  // We send the signal 0 to the given PID (signal 0 is a no-op)
  // If this succeeds, the process with the given PID exists
  // if it fails, the process does not / no longer exist.
  if (kill(m_iHostPID, 0) < 0)
    bValid = false;
#else
#  error Not implemented
#endif

  return bValid;
}

ezResult ezEngineProcessCommunicationChannel::ConnectToHostProcess()
{
  EZ_ASSERT_DEV(m_pChannel == nullptr, "ProcessCommunication object already in use");

  ezResult res = EZ_SUCCESS;
  if (!ezEditorEngineProcessApp::GetSingleton()->IsRemoteMode())
  {
    if (ezCommandLineUtils::GetGlobalInstance()->GetStringOption("-IPC").IsEmpty())
    {
      EZ_REPORT_FAILURE("Command Line does not contain -IPC parameter");
      return EZ_FAILURE;
    }

    if (ezCommandLineUtils::GetGlobalInstance()->GetStringOption("-PID").IsEmpty())
    {
      EZ_REPORT_FAILURE("Command Line does not contain -PID parameter");
      return EZ_FAILURE;
    }

    m_iHostPID = 0;
    EZ_SUCCEED_OR_RETURN(ezConversionUtils::StringToInt64(ezCommandLineUtils::GetGlobalInstance()->GetStringOption("-PID"), m_iHostPID));

    ezLog::Debug("Host Process ID: {0}", m_iHostPID);

    res = CreateAndConnectChannel(ezIpcChannel::CreatePipeChannel(ezCommandLineUtils::GetGlobalInstance()->GetStringOption("-IPC"), ezIpcChannel::Mode::Client));
  }
  else
  {
    res = CreateAndConnectChannel(ezIpcChannel::CreateNetworkChannel("localhost:1050", ezIpcChannel::Mode::Server));
  }
  if (res.Failed())
  {
    ezLog::Error("IpcChannel: CreateAndConnectChannel failed");
    return EZ_FAILURE;
  }

  return WaitForConnection(ezTime::MakeFromSeconds(30));
}