introduced lambda in launchExecutor

Author: kr-2003

clang/include/clang/Interpreter/RemoteJITUtils.h

@@ -35,8 +35,7 @@
 llvm::Expected<std::unique_ptr<llvm::orc::SimpleRemoteEPC>>
 launchExecutor(llvm::StringRef ExecutablePath, bool UseSharedMemory,
                llvm::StringRef SlabAllocateSizeString,
-               int stdin_fd = STDIN_FILENO, int stdout_fd = STDOUT_FILENO,
-               int stderr_fd = STDERR_FILENO);
+               std::function<void()> CustomizeFork = nullptr);
 
 /// Create a JITLinkExecutor that connects to the given network address
 /// through a TCP socket. A valid NetworkAddress provides hostname and port,

clang/lib/Interpreter/RemoteJITUtils.cpp

@@ -95,8 +95,8 @@ createSharedMemoryManager(SimpleRemoteEPC &SREPC,
 
 Expected<std::unique_ptr<SimpleRemoteEPC>>
 launchExecutor(StringRef ExecutablePath, bool UseSharedMemory,
-               llvm::StringRef SlabAllocateSizeString, int stdin_fd,
-               int stdout_fd, int stderr_fd) {
+               llvm::StringRef SlabAllocateSizeString,
+               std::function<void()> CustomizeFork) {
 #ifndef LLVM_ON_UNIX
   // FIXME: Add support for Windows.
   return make_error<StringError>("-" + ExecutablePath +
@@ -139,22 +139,8 @@ launchExecutor(StringRef ExecutablePath, bool UseSharedMemory,
     close(ToExecutor[WriteEnd]);
     close(FromExecutor[ReadEnd]);
 
-    if (stdin_fd != STDIN_FILENO) {
-      dup2(stdin_fd, STDIN_FILENO);
-      close(stdin_fd);
-    }
-
-    if (stdout_fd != STDOUT_FILENO) {
-      dup2(stdout_fd, STDOUT_FILENO);
-      close(stdout_fd);
-      setvbuf(stdout, NULL, _IONBF, 0);
-    }
-
-    if (stderr_fd != STDERR_FILENO) {
-      dup2(stderr_fd, STDERR_FILENO);
-      close(stderr_fd);
-      setvbuf(stderr, NULL, _IONBF, 0);
-    }
+    if (CustomizeFork)
+      CustomizeFork();
 
     // Execute the child process.
     std::unique_ptr<char[]> ExecutorPath, FDSpecifier;