1 files changed, 7 insertions, 0 deletions

diff --git a/lib/fault-inject.c b/lib/fault-inject.c
index 4ff157159a0d..09ac73c177fd 100644
--- a/lib/fault-inject.c
+++ b/lib/fault-inject.c
@@ -107,6 +107,12 @@ static inline bool fail_stacktrace(struct fault_attr *attr)
 
 bool should_fail(struct fault_attr *attr, ssize_t size)
 {
+        if (in_task() && current->fail_nth) {
+                if (--current->fail_nth == 0)
+                        goto fail;
+                return false;
+        }
+
         /* No need to check any other properties if the probability is 0 */
         if (attr->probability == 0)
                 return false;
@@ -134,6 +140,7 @@ bool should_fail(struct fault_attr *attr, ssize_t size)
         if (!fail_stacktrace(attr))
                 return false;
 
+fail:
         fail_dump(attr);
 
         if (atomic_read(&attr->times) != -1)