inject: Add support for alloc_page family of functions (iovisor#2114)

* inject: Add support for alloc_page family of functions * Add vim backup files to gitignore
tkuburic · Jan 7, 2019 · 6d92549 · 6d92549
1 parent aed9b1e
commit 6d92549
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -3,6 +3,7 @@
 *.swo
 *.pyc
 .idea
+*~
 
 # Build artifacts
 /build/

diff --git a/tools/inject.py b/tools/inject.py
@@ -41,6 +41,7 @@ class Probe:
  errno_mapping = {
  "kmalloc": "-ENOMEM",
  "bio": "-EIO",
+ "alloc_page" : "true",
  }
 
  @classmethod
@@ -327,14 +328,15 @@ class Tool:
  error_injection_mapping = {
  "kmalloc": "should_failslab(struct kmem_cache *s, gfp_t gfpflags)",
  "bio": "should_fail_bio(struct bio *bio)",
+ "alloc_page": "should_fail_alloc_page(gfp_t gfp_mask, unsigned int order)",
  }
 
  def __init__(self):
  parser = argparse.ArgumentParser(description="Fail specified kernel" +
  " functionality when call chain and predicates are met",
  formatter_class=argparse.RawDescriptionHelpFormatter,
  epilog=Tool.examples)
- parser.add_argument(dest="mode", choices=['kmalloc','bio'],
+ parser.add_argument(dest="mode", choices=["kmalloc", "bio", "alloc_page"],
  help="indicate which base kernel function to fail")
  parser.add_argument(metavar="spec", dest="spec",
  help="specify call chain")