generate-afl-tests.sh 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. #!/bin/sh
  2. # This script splits the data test files containing the test cases into
  3. # individual files (one test case per file) suitable for use with afl
  4. # (American Fuzzy Lop). http://lcamtuf.coredump.cx/afl/
  5. #
  6. # Usage: generate-afl-tests.sh <test data file path>
  7. # <test data file path> - should be the path to one of the test suite files
  8. # such as 'test_suite_mpi.data'
  9. # Abort on errors
  10. set -e
  11. if [ -z $1 ]
  12. then
  13. echo " [!] No test file specified" >&2
  14. echo "Usage: $0 <test data file>" >&2
  15. exit 1
  16. fi
  17. SRC_FILEPATH=$(dirname $1)/$(basename $1)
  18. TESTSUITE=$(basename $1 .data)
  19. THIS_DIR=$(basename $PWD)
  20. if [ -d ../library -a -d ../include -a -d ../tests -a $THIS_DIR == "tests" ];
  21. then :;
  22. else
  23. echo " [!] Must be run from mbed TLS tests directory" >&2
  24. exit 1
  25. fi
  26. DEST_TESTCASE_DIR=$TESTSUITE-afl-tests
  27. DEST_OUTPUT_DIR=$TESTSUITE-afl-out
  28. echo " [+] Creating output directories" >&2
  29. if [ -e $DEST_OUTPUT_DIR/* ];
  30. then :
  31. echo " [!] Test output files already exist." >&2
  32. exit 1
  33. else
  34. mkdir -p $DEST_OUTPUT_DIR
  35. fi
  36. if [ -e $DEST_TESTCASE_DIR/* ];
  37. then :
  38. echo " [!] Test output files already exist." >&2
  39. else
  40. mkdir -p $DEST_TESTCASE_DIR
  41. fi
  42. echo " [+] Creating test cases" >&2
  43. cd $DEST_TESTCASE_DIR
  44. split -p '^\s*$' ../$SRC_FILEPATH
  45. for f in *;
  46. do
  47. # Strip out any blank lines (no trim on OS X)
  48. sed '/^\s*$/d' $f >testcase_$f
  49. rm $f
  50. done
  51. cd ..
  52. echo " [+] Test cases in $DEST_TESTCASE_DIR" >&2