lib/ldb/tests/python/crash.py

   1 #!/usr/bin/env python3
   2 #
   3 # Tests for crashing functions
   4
   5 import os
   6 from unittest import TestCase
   7 import os
   8 import sys
   9 import traceback
  10
  11 import ldb
  12
  13
  14 def segfault_detector(f):
  15     def wrapper(*args, **kwargs):
  16         pid = os.fork()
  17         if pid == 0:
  18             # child, crashing?
  19             try:
  20                 f(*args, **kwargs)
  21             except Exception as e:
  22                 traceback.print_exc()
  23             sys.stderr.flush()
  24             sys.stdout.flush()
  25             os._exit(0)
  26
  27         # parent, waiting
  28         pid2, status = os.waitpid(pid, 0)
  29         if os.WIFSIGNALED(status):
  30             signal = os.WTERMSIG(status)
  31             raise AssertionError("Failed with signal %d" % signal)
  32
  33     return wrapper
  34
  35
  36 class LdbDnCrashTests(TestCase):
  37     @segfault_detector
  38     def test_ldb_dn_explode_crash(self):
  39         for i in range(106, 150):
  40             dn = ldb.Dn(ldb.Ldb(), "a=b%s,c= " % (' ' * i))
  41             dn.validate()
  42
  43 if __name__ == '__main__':
  44     import unittest
  45     unittest.TestProgram()