1  #!/usr/bin/env python3
       2  
       3  import sys, os, subprocess, tempfile, shutil
       4  
       5  
       6  def cmd (command):
       7  	# https://stackoverflow.com/a/4408409 as we might have huge output sometimes
       8  	with tempfile.TemporaryFile () as tempf:
       9  		p = subprocess.Popen (command, stderr=tempf)
      10  
      11  		try:
      12  			p.wait ()
      13  			tempf.seek (0)
      14  			text = tempf.read ()
      15  
      16  			#TODO: Detect debug mode with a better way
      17  			is_debug_mode = b"SANITIZE" in text
      18  
      19  			return ("" if is_debug_mode else text.decode ("utf-8").strip ()), p.returncode
      20  		except subprocess.TimeoutExpired:
      21  			return 'error: timeout, ' + ' '.join (command), 1
      22  
      23  
      24  srcdir = os.getenv ("srcdir", ".")
      25  EXEEXT = os.getenv ("EXEEXT", "")
      26  top_builddir = os.getenv ("top_builddir", ".")
      27  hb_subset_fuzzer = os.path.join (top_builddir, "hb-subset-fuzzer" + EXEEXT)
      28  
      29  if not os.path.exists (hb_subset_fuzzer):
      30          if len (sys.argv) < 2 or not os.path.exists (sys.argv[1]):
      31                  sys.exit ("""Failed to find hb-subset-fuzzer binary automatically,
      32  please provide it as the first argument to the tool""")
      33  
      34          hb_subset_fuzzer = sys.argv[1]
      35  
      36  print ('hb_subset_fuzzer:', hb_subset_fuzzer)
      37  fails = 0
      38  
      39  valgrind = None
      40  if os.getenv ('RUN_VALGRIND', ''):
      41  	valgrind = shutil.which ('valgrind')
      42  	if valgrind is None:
      43  		sys.exit ("""Valgrind requested but not found.""")
      44  
      45  def run_dir (parent_path):
      46  	global fails
      47  	for file in os.listdir (parent_path):
      48  		path = os.path.join(parent_path, file)
      49  		# TODO: Run on all the fonts not just subset related ones
      50  		if "subset" not in path: continue
      51  
      52  		print ("running subset fuzzer against %s" % path)
      53  		if valgrind:
      54  			text, returncode = cmd ([valgrind, '--leak-check=full', '--error-exitcode=1', hb_subset_fuzzer, path])
      55  		else:
      56  			text, returncode = cmd ([hb_subset_fuzzer, path])
      57  			if 'error' in text:
      58  				returncode = 1
      59  
      60  		if (not valgrind or returncode) and text.strip ():
      61  			print (text)
      62  
      63  		if returncode != 0:
      64  			print ("failed for %s" % path)
      65  			fails = fails + 1
      66  
      67  
      68  run_dir (os.path.join (srcdir, "..", "subset", "data", "fonts"))
      69  run_dir (os.path.join (srcdir, "fonts"))
      70  
      71  if fails:
      72  	sys.exit ("%d subset fuzzer related tests failed." % fails)