diff options
Diffstat (limited to 'gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach')
7 files changed, 288 insertions, 0 deletions
diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/Makefile b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/Makefile new file mode 100644 index 00000000000..8a52f8fcd47 --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/Makefile @@ -0,0 +1,5 @@ +CXX_SOURCES := main.cpp + +EXE := ProcessAttach + +include Makefile.rules diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/TestProcessAttach.py b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/TestProcessAttach.py new file mode 100644 index 00000000000..b85d5713350 --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/TestProcessAttach.py @@ -0,0 +1,90 @@ +""" +Test process attach. +""" + + + +import os +import lldb +import shutil +from lldbsuite.test.decorators import * +from lldbsuite.test.lldbtest import * +from lldbsuite.test import lldbutil + +exe_name = "ProcessAttach" # Must match Makefile + + +class ProcessAttachTestCase(TestBase): + + mydir = TestBase.compute_mydir(__file__) + + NO_DEBUG_INFO_TESTCASE = True + + @skipIfiOSSimulator + @expectedFailureNetBSD + def test_attach_to_process_by_id(self): + """Test attach by process id""" + self.build() + exe = self.getBuildArtifact(exe_name) + + # Spawn a new process + popen = self.spawnSubprocess(exe) + self.addTearDownHook(self.cleanupSubprocesses) + + self.runCmd("process attach -p " + str(popen.pid)) + + target = self.dbg.GetSelectedTarget() + + process = target.GetProcess() + self.assertTrue(process, PROCESS_IS_VALID) + + @expectedFailureNetBSD + def test_attach_to_process_from_different_dir_by_id(self): + """Test attach by process id""" + newdir = self.getBuildArtifact("newdir") + try: + os.mkdir(newdir) + except OSError as e: + if e.errno != os.errno.EEXIST: + raise + testdir = self.getBuildDir() + exe = os.path.join(newdir, 'proc_attach') + self.buildProgram('main.cpp', exe) + self.addTearDownHook(lambda: shutil.rmtree(newdir)) + + # Spawn a new process + popen = self.spawnSubprocess(exe) + self.addTearDownHook(self.cleanupSubprocesses) + + os.chdir(newdir) + self.addTearDownHook(lambda: os.chdir(testdir)) + self.runCmd("process attach -p " + str(popen.pid)) + + target = self.dbg.GetSelectedTarget() + + process = target.GetProcess() + self.assertTrue(process, PROCESS_IS_VALID) + + @expectedFailureNetBSD + def test_attach_to_process_by_name(self): + """Test attach by process name""" + self.build() + exe = self.getBuildArtifact(exe_name) + + # Spawn a new process + popen = self.spawnSubprocess(exe) + self.addTearDownHook(self.cleanupSubprocesses) + + self.runCmd("process attach -n " + exe_name) + + target = self.dbg.GetSelectedTarget() + + process = target.GetProcess() + self.assertTrue(process, PROCESS_IS_VALID) + + def tearDown(self): + # Destroy process before TestBase.tearDown() + self.dbg.GetSelectedTarget().GetProcess().Destroy() + + # Call super's tearDown(). + TestBase.tearDown(self) diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/Makefile b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/Makefile new file mode 100644 index 00000000000..a694d36ac0d --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/Makefile @@ -0,0 +1,12 @@ +CXX_SOURCES := main.cpp + +EXE := AttachDenied + +all: AttachDenied sign + +include Makefile.rules + +sign: entitlements.plist AttachDenied +ifeq ($(OS),Darwin) + codesign -s - -f --entitlements $^ +endif diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/TestAttachDenied.py b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/TestAttachDenied.py new file mode 100644 index 00000000000..dcd73da42e9 --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/TestAttachDenied.py @@ -0,0 +1,45 @@ +""" +Test denied process attach. +""" + + + +import time +import lldb +from lldbsuite.test.decorators import * +from lldbsuite.test.lldbtest import * +from lldbsuite.test import lldbutil + +exe_name = 'AttachDenied' # Must match Makefile + + +class AttachDeniedTestCase(TestBase): + + mydir = TestBase.compute_mydir(__file__) + NO_DEBUG_INFO_TESTCASE = True + + @skipIfWindows + @skipIfiOSSimulator + @skipIfDarwinEmbedded # ptrace(ATTACH_REQUEST...) won't work on ios/tvos/etc + def test_attach_to_process_by_id_denied(self): + """Test attach by process id denied""" + self.build() + exe = self.getBuildArtifact(exe_name) + + # Use a file as a synchronization point between test and inferior. + pid_file_path = lldbutil.append_to_process_working_directory(self, + "pid_file_%d" % (int(time.time()))) + self.addTearDownHook( + lambda: self.run_platform_command( + "rm %s" % + (pid_file_path))) + + # Spawn a new process + popen = self.spawnSubprocess(exe, [pid_file_path]) + self.addTearDownHook(self.cleanupSubprocesses) + + pid = lldbutil.wait_for_file_on_target(self, pid_file_path) + + self.expect('process attach -p ' + pid, + startstr='error: attach failed:', + error=True) diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/entitlements.plist b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/entitlements.plist new file mode 100644 index 00000000000..3d60e8bd0b9 --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/entitlements.plist @@ -0,0 +1,8 @@ +<?xml version="1.0" encoding="UTF-8"?> +<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd"> +<plist version="1.0"> +<dict> + <key>com.apple.security.cs.debugger</key> + <true/> +</dict> +</plist> diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/main.cpp b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/main.cpp new file mode 100644 index 00000000000..ff1fccae4b1 --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/attach_denied/main.cpp @@ -0,0 +1,108 @@ +#include <errno.h> +#include <fcntl.h> +#include <signal.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + +#include <sys/types.h> +#include <sys/ptrace.h> +#include <sys/stat.h> +#include <sys/wait.h> + +#if defined(PTRACE_ATTACH) +#define ATTACH_REQUEST PTRACE_ATTACH +#define DETACH_REQUEST PTRACE_DETACH +#elif defined(PT_ATTACH) +#define ATTACH_REQUEST PT_ATTACH +#define DETACH_REQUEST PT_DETACH +#else +#error "Unsupported platform" +#endif + +bool writePid (const char* file_name, const pid_t pid) +{ + char *tmp_file_name = (char *)malloc(strlen(file_name) + 16); + strcpy(tmp_file_name, file_name); + strcat(tmp_file_name, "_tmp"); + int fd = open (tmp_file_name, O_CREAT | O_WRONLY | O_TRUNC, S_IRUSR | S_IWUSR); + if (fd == -1) + { + fprintf (stderr, "open(%s) failed: %s\n", tmp_file_name, strerror (errno)); + free(tmp_file_name); + return false; + } + char buffer[64]; + snprintf (buffer, sizeof(buffer), "%ld", (long)pid); + + bool res = true; + if (write (fd, buffer, strlen (buffer)) == -1) + { + fprintf (stderr, "write(%s) failed: %s\n", buffer, strerror (errno)); + res = false; + } + close (fd); + + if (rename (tmp_file_name, file_name) == -1) + { + fprintf (stderr, "rename(%s, %s) failed: %s\n", tmp_file_name, file_name, strerror (errno)); + res = false; + } + free(tmp_file_name); + + return res; +} + +void signal_handler (int) +{ +} + +int main (int argc, char const *argv[]) +{ + if (argc < 2) + { + fprintf (stderr, "invalid number of command line arguments\n"); + return 1; + } + + const pid_t pid = fork (); + if (pid == -1) + { + fprintf (stderr, "fork failed: %s\n", strerror (errno)); + return 1; + } + + if (pid > 0) + { + // Make pause call to return when a signal is received. Normally this happens when the + // test runner tries to terminate us. + signal (SIGHUP, signal_handler); + signal (SIGTERM, signal_handler); + if (ptrace (ATTACH_REQUEST, pid, NULL, 0) == -1) + { + fprintf (stderr, "ptrace(ATTACH) failed: %s\n", strerror (errno)); + } + else + { + if (writePid (argv[1], pid)) + pause (); // Waiting for the debugger trying attach to the child. + + if (ptrace (DETACH_REQUEST, pid, NULL, 0) != 0) + fprintf (stderr, "ptrace(DETACH) failed: %s\n", strerror (errno)); + } + + kill (pid, SIGTERM); + int status = 0; + if (waitpid (pid, &status, 0) == -1) + fprintf (stderr, "waitpid failed: %s\n", strerror (errno)); + } + else + { + // child inferior. + pause (); + } + + printf ("Exiting now\n"); + return 0; +} diff --git a/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/main.cpp b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/main.cpp new file mode 100644 index 00000000000..46ffacc0a84 --- /dev/null +++ b/gnu/llvm/lldb/packages/Python/lldbsuite/test/commands/process/attach/main.cpp @@ -0,0 +1,20 @@ +#include <stdio.h> + +#include <chrono> +#include <thread> + +int main(int argc, char const *argv[]) { + int temp; + lldb_enable_attach(); + + // Waiting to be attached by the debugger. + temp = 0; + + while (temp < 30) // Waiting to be attached... + { + std::this_thread::sleep_for(std::chrono::seconds(2)); + temp++; + } + + printf("Exiting now\n"); +} |