Always forward grandchild's exit codes to its parent.  This fixes
KVM_ONE_VCPU_TEST().

Initial patch written by Sean Christopherson [1].

Cc: Jakub Kicinski <[email protected]>
Cc: Kees Cook <[email protected]>
Cc: Mark Brown <[email protected]>
Cc: Sean Christopherson <[email protected]>
Cc: Shuah Khan <[email protected]>
Cc: Will Drewry <[email protected]>
Link: https://lore.kernel.org/r/[email protected] [1]
Fixes: 0710a1a73fb4 ("selftests/harness: Merge TEST_F_FORK() into TEST_F()")
Signed-off-by: Mickaël Salaün <[email protected]>
Link: https://lore.kernel.org/r/[email protected]
---

Changes since v3:
* New patch mainly from Sean Christopherson.
---
 tools/testing/selftests/kselftest_harness.h | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/tools/testing/selftests/kselftest_harness.h 
b/tools/testing/selftests/kselftest_harness.h
index d3837a3a584e..77d4fac8d0c0 100644
--- a/tools/testing/selftests/kselftest_harness.h
+++ b/tools/testing/selftests/kselftest_harness.h
@@ -442,7 +442,7 @@ static inline pid_t clone3_vfork(void)
                                fixture_name##_setup(_metadata, self, 
variant->data); \
                                /* Let setup failure terminate early. */ \
                                if (_metadata->exit_code) \
-                                       _exit(0); \
+                                       _exit(_metadata->exit_code); \
                                _metadata->setup_completed = true; \
                                fixture_name##_##test_name(_metadata, self, 
variant->data); \
                        } else if (child < 0 || child != waitpid(child, 
&status, 0)) { \
@@ -454,7 +454,7 @@ static inline pid_t clone3_vfork(void)
                        if (_metadata->setup_completed && 
!fixture_name##_teardown_parent && \
                                        __sync_bool_compare_and_swap(teardown, 
false, true)) \
                                fixture_name##_teardown(_metadata, self, 
variant->data); \
-                       _exit(0); \
+                       _exit(_metadata->exit_code); \
                } \
                if (_metadata->setup_completed && 
fixture_name##_teardown_parent && \
                                __sync_bool_compare_and_swap(teardown, false, 
true)) \
@@ -462,8 +462,10 @@ static inline pid_t clone3_vfork(void)
                munmap(teardown, sizeof(*teardown)); \
                if (self && fixture_name##_teardown_parent) \
                        munmap(self, sizeof(*self)); \
-               if (!WIFEXITED(status) && WIFSIGNALED(status)) \
-                       /* Forward signal to __wait_for_test(). */ \
+               /* Forward exit codes and signals to __wait_for_test(). */ \
+               if (WIFEXITED(status)) \
+                       _exit(_metadata->exit_code); \
+               if (WIFSIGNALED(status)) \
                        kill(getpid(), WTERMSIG(status)); \
                __test_check_assert(_metadata); \
        } \
-- 
2.45.0


Reply via email to