Beta Shell v2.0

File Manager

Name	Size	Perms	Modified	Actions
[ .. / .. ]
COFFPlatform.h	8.61 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
COFFVCRuntimeSupport.h	3.49 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
CompileOnDemandLayer.h	4.48 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
CompileUtils.h	3.25 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
Core.h	80.86 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
DebugObjectManagerPlugin.h	4.16 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
DebugUtils.h	4.74 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
Debugging	-	drwxr-xr-x	2025-05-28 10:12:49	Del
ELFNixPlatform.h	12.55 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCDebugObjectRegistrar.h	2.41 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCDynamicLibrarySearchGenerator.h	3.17 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCEHFrameRegistrar.h	2.25 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCGenericDylibManager.h	3.08 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCGenericJITLinkMemoryManager.h	3.21 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCGenericMemoryAccess.h	3.35 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCGenericRTDyldMemoryManager.h	4.53 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
EPCIndirectionUtils.h	8.46 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
ExecutionUtils.h	13.63 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
ExecutorProcessControl.h	21.14 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
IRCompileLayer.h	2.17 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
IRTransformLayer.h	1.78 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
IndirectionUtils.h	20.3 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
JITTargetMachineBuilder.h	5.44 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
LLJIT.h	22.71 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
Layer.h	7.44 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
LazyReexports.h	6.52 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
LookupAndRecordAddrs.h	2.7 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
MachOBuilder.h	19.35 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
MachOPlatform.h	17.04 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
Mangling.h	2.04 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
MapperJITLinkMemoryManager.h	2.48 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
MemoryMapper.h	5.21 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
ObjectFileInterface.h	1.45 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
ObjectLinkingLayer.h	9.46 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
ObjectTransformLayer.h	1.54 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
OrcABISupport.h	18.74 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
RTDyldObjectLinkingLayer.h	6.25 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
SectCreate.h	1.95 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
Shared	-	drwxr-xr-x	2025-05-28 10:12:49	Del
SimpleRemoteEPC.h	5.47 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
SpeculateAnalyses.h	2.85 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
Speculation.h	7.18 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
SymbolStringPool.h	10.4 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
TargetProcess	-	drwxr-xr-x	2025-05-28 10:12:49	Del
TaskDispatch.h	3.97 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del
ThreadSafeModule.h	5.88 KB	-rw-r--r--	2025-01-14 09:41:02	Edit Del

Editing: SimpleRemoteEPC.h (5.47 KB)

//===---- SimpleRemoteEPC.h - Simple remote executor control ----*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// Simple remote executor process control.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_EXECUTIONENGINE_ORC_SIMPLEREMOTEEPC_H
#define LLVM_EXECUTIONENGINE_ORC_SIMPLEREMOTEEPC_H

#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/FunctionExtras.h"
#include "llvm/ExecutionEngine/Orc/EPCGenericDylibManager.h"
#include "llvm/ExecutionEngine/Orc/EPCGenericJITLinkMemoryManager.h"
#include "llvm/ExecutionEngine/Orc/EPCGenericMemoryAccess.h"
#include "llvm/ExecutionEngine/Orc/ExecutorProcessControl.h"
#include "llvm/ExecutionEngine/Orc/Shared/SimpleRemoteEPCUtils.h"
#include "llvm/Support/Error.h"
#include "llvm/Support/MSVCErrorWorkarounds.h"

#include <future>

namespace llvm {
namespace orc {

class SimpleRemoteEPC : public ExecutorProcessControl,
                        public SimpleRemoteEPCTransportClient {
public:
  /// A setup object containing callbacks to construct a memory manager and
  /// memory access object. Both are optional. If not specified,
  /// EPCGenericJITLinkMemoryManager and EPCGenericMemoryAccess will be used.
  struct Setup {
    using CreateMemoryManagerFn =
        Expected<std::unique_ptr<jitlink::JITLinkMemoryManager>>(
            SimpleRemoteEPC &);
    using CreateMemoryAccessFn =
        Expected<std::unique_ptr<MemoryAccess>>(SimpleRemoteEPC &);

    unique_function<CreateMemoryManagerFn> CreateMemoryManager;
    unique_function<CreateMemoryAccessFn> CreateMemoryAccess;
  };

  /// Create a SimpleRemoteEPC using the given transport type and args.
  template <typename TransportT, typename... TransportTCtorArgTs>
  static Expected<std::unique_ptr<SimpleRemoteEPC>>
  Create(std::unique_ptr<TaskDispatcher> D, Setup S,
         TransportTCtorArgTs &&...TransportTCtorArgs) {
    std::unique_ptr<SimpleRemoteEPC> SREPC(
        new SimpleRemoteEPC(std::make_shared<SymbolStringPool>(),
                            std::move(D)));
    auto T = TransportT::Create(
        *SREPC, std::forward<TransportTCtorArgTs>(TransportTCtorArgs)...);
    if (!T)
      return T.takeError();
    SREPC->T = std::move(*T);
    if (auto Err = SREPC->setup(std::move(S)))
      return joinErrors(std::move(Err), SREPC->disconnect());
    return std::move(SREPC);
  }

  SimpleRemoteEPC(const SimpleRemoteEPC &) = delete;
  SimpleRemoteEPC &operator=(const SimpleRemoteEPC &) = delete;
  SimpleRemoteEPC(SimpleRemoteEPC &&) = delete;
  SimpleRemoteEPC &operator=(SimpleRemoteEPC &&) = delete;
  ~SimpleRemoteEPC();

  Expected<tpctypes::DylibHandle> loadDylib(const char *DylibPath) override;

  void lookupSymbolsAsync(ArrayRef<LookupRequest> Request,
                          SymbolLookupCompleteFn F) override;

  Expected<int32_t> runAsMain(ExecutorAddr MainFnAddr,
                              ArrayRef<std::string> Args) override;

  Expected<int32_t> runAsVoidFunction(ExecutorAddr VoidFnAddr) override;

  Expected<int32_t> runAsIntFunction(ExecutorAddr IntFnAddr, int Arg) override;

  void callWrapperAsync(ExecutorAddr WrapperFnAddr,
                        IncomingWFRHandler OnComplete,
                        ArrayRef<char> ArgBuffer) override;

  Error disconnect() override;

  Expected<HandleMessageAction>
  handleMessage(SimpleRemoteEPCOpcode OpC, uint64_t SeqNo, ExecutorAddr TagAddr,
                SimpleRemoteEPCArgBytesVector ArgBytes) override;

  void handleDisconnect(Error Err) override;

private:
  SimpleRemoteEPC(std::shared_ptr<SymbolStringPool> SSP,
                  std::unique_ptr<TaskDispatcher> D)
    : ExecutorProcessControl(std::move(SSP), std::move(D)) {}

  static Expected<std::unique_ptr<jitlink::JITLinkMemoryManager>>
  createDefaultMemoryManager(SimpleRemoteEPC &SREPC);
  static Expected<std::unique_ptr<MemoryAccess>>
  createDefaultMemoryAccess(SimpleRemoteEPC &SREPC);

  Error sendMessage(SimpleRemoteEPCOpcode OpC, uint64_t SeqNo,
                    ExecutorAddr TagAddr, ArrayRef<char> ArgBytes);

  Error handleSetup(uint64_t SeqNo, ExecutorAddr TagAddr,
                    SimpleRemoteEPCArgBytesVector ArgBytes);
  Error setup(Setup S);

  Error handleResult(uint64_t SeqNo, ExecutorAddr TagAddr,
                     SimpleRemoteEPCArgBytesVector ArgBytes);
  void handleCallWrapper(uint64_t RemoteSeqNo, ExecutorAddr TagAddr,
                         SimpleRemoteEPCArgBytesVector ArgBytes);
  Error handleHangup(SimpleRemoteEPCArgBytesVector ArgBytes);

  uint64_t getNextSeqNo() { return NextSeqNo++; }
  void releaseSeqNo(uint64_t SeqNo) {}

  using PendingCallWrapperResultsMap =
    DenseMap<uint64_t, IncomingWFRHandler>;

  std::mutex SimpleRemoteEPCMutex;
  std::condition_variable DisconnectCV;
  bool Disconnected = false;
  Error DisconnectErr = Error::success();

  std::unique_ptr<SimpleRemoteEPCTransport> T;
  std::unique_ptr<jitlink::JITLinkMemoryManager> OwnedMemMgr;
  std::unique_ptr<MemoryAccess> OwnedMemAccess;

  std::unique_ptr<EPCGenericDylibManager> DylibMgr;
  ExecutorAddr RunAsMainAddr;
  ExecutorAddr RunAsVoidFunctionAddr;
  ExecutorAddr RunAsIntFunctionAddr;

  uint64_t NextSeqNo = 0;
  PendingCallWrapperResultsMap PendingCallWrapperResults;
};

} // end namespace orc
} // end namespace llvm

#endif // LLVM_EXECUTIONENGINE_ORC_SIMPLEREMOTEEPC_H