Fix attaching to running process, i.e. omnitrace -p <PID> (#60)

* Fix attaching to a process

- e.g. omnitrace -p <PID>

* Update /proc/sys/kernel/yama/ptrace_scope in CI

* Query /proc/sys/kernel/yama/ptrace_scope

* Use AUTHOR_WARNING instead of WARNING for ptrace_scope

[ROCm/rocprofiler-systems commit: fe3d15a6c9]

projects/rocprofiler-systems/tests/CMakeLists.txt

@@ -98,6 +98,22 @@ set(_python_environment
     "LD_LIBRARY_PATH=${PROJECT_BINARY_DIR}:${OMNITRACE_DYNINST_API_RT_DIR}:$ENV{LD_LIBRARY_PATH}"
     "PYTHONPATH=${PROJECT_BINARY_DIR}/lib/python/site-packages")
 
+set(_attach_environment
+    "OMNITRACE_USE_PERFETTO=ON"
+    "OMNITRACE_USE_TIMEMORY=ON"
+    "OMNITRACE_USE_SAMPLING=OFF"
+    "OMNITRACE_USE_THREAD_SAMPLING=ON"
+    "OMNITRACE_USE_CRITICAL_TRACE=ON"
+    "OMNITRACE_USE_OMPT=ON"
+    "OMNITRACE_USE_KOKKOSP=ON"
+    "OMNITRACE_TIME_OUTPUT=OFF"
+    "OMNITRACE_USE_PID=OFF"
+    "OMP_PROC_BIND=spread"
+    "OMP_PLACES=threads"
+    "OMP_NUM_THREADS=2"
+    "LD_LIBRARY_PATH=${PROJECT_BINARY_DIR}:${OMNITRACE_DYNINST_API_RT_DIR}:$ENV{LD_LIBRARY_PATH}"
+    )
+
 # -------------------------------------------------------------------------------------- #
 
 set(MPIEXEC_EXECUTABLE_ARGS)
@@ -804,6 +820,52 @@ omnitrace_add_test(
     RUNTIME_PASS_REGEX "(\\\[[0-9]+\\\]) function coverage ::  66.67%"
     REWRITE_RUN_PASS_REGEX "(\\\[[0-9]+\\\]) function coverage ::  66.67%")
 
+# -------------------------------------------------------------------------------------- #
+#
+# attach tests
+#
+# -------------------------------------------------------------------------------------- #
+
+set(_VALID_PTRACE_SCOPE OFF)
+if(EXISTS "/proc/sys/kernel/yama/ptrace_scope")
+    file(READ "/proc/sys/kernel/yama/ptrace_scope" _PTRACE_SCOPE LIMIT 1)
+    if("${_PTRACE_SCOPE}" EQUAL 0)
+        set(_VALID_PTRACE_SCOPE ON)
+    endif()
+else()
+    omnitrace_message(
+        AUTHOR_WARNING
+        "Disabling attach tests. Run 'echo 0 | sudo tee /proc/sys/kernel/yama/ptrace_scope' to enable attaching to process"
+        )
+endif()
+
+if(TARGET parallel-overhead AND _VALID_PTRACE_SCOPE)
+    add_test(
+        NAME parallel-overhead-attach
+        COMMAND
+            ${CMAKE_CURRENT_LIST_DIR}/run-omnitrace-pid.sh $<TARGET_FILE:omnitrace-exe>
+            -ME "\.c$" -e -v 1 --label return args file -l --
+            $<TARGET_FILE:parallel-overhead> 30 8 1000
+        WORKING_DIRECTORY ${CMAKE_BINARY_DIR})
+
+    set(_parallel_overhead_attach_environ
+        "${_attach_environment}" "OMNITRACE_OUTPUT_PATH=omnitrace-tests-output"
+        "OMNITRACE_OUTPUT_PREFIX=parallel-overhead-attach/")
+
+    set_tests_properties(
+        parallel-overhead-attach
+        PROPERTIES ENVIRONMENT
+                   "${_parallel_overhead_attach_environ}"
+                   TIMEOUT
+                   300
+                   LABELS
+                   "parallel-overhead;attach"
+                   PASS_REGULAR_EXPRESSION
+                   "Outputting.*(perfetto-trace.proto).*Outputting.*(wall_clock.txt)"
+                   FAIL_REGULAR_EXPRESSION
+                   "Dyninst was unable to attach to the specified process")
+endif()
+
 # -------------------------------------------------------------------------------------- #
 #
 # python tests

projects/rocprofiler-systems/tests/run-omnitrace-pid.sh

@@ -0,0 +1,45 @@
+#!/bin/bash
+
+cleanup()
+{
+    kill -s 9 ${_PID}
+}
+
+trap cleanup SIGABRT SIGQUIT
+
+OMNITRACE_COMMAND=""
+APPLICATION_COMMAND=""
+
+while [[ $# -gt 0 ]]
+do
+    if [ "${1}" == "--" ]; then
+        shift
+        break
+    else
+        OMNITRACE_COMMAND="${OMNITRACE_COMMAND}${1} "
+        shift
+    fi
+done
+
+${@} &
+_PID=$!
+
+echo ""
+ps ax | grep "${1}"
+echo ""
+
+if [ "${_PID}" = "" ]; then
+    echo "Error! No PID for \"${@}\""
+    exit -1
+fi
+
+echo "PID: ${_PID}"
+echo ""
+
+${OMNITRACE_COMMAND} -p ${_PID} -- ${@}
+RET=$?
+
+killall $(basename ${1}) &> /dev/null
+
+echo "Exiting with code: ${RET}"
+exit ${RET}