#!/bin/bash
# blockio_test.sh - Don Yang (uguu.org)
#
# 06/03/12

OUTPUT=test_output

set -e
mkdir -p $OUTPUT

function encrypt_and_compare() {
   ./blockio e $OUTPUT/blockio.in $OUTPUT/blockio.enc
   ./blockio d $OUTPUT/blockio.enc $OUTPUT/blockio.dec
   cmp $OUTPUT/blockio.in $OUTPUT/blockio.dec
}

function run_small_test() {
   echo -n "$1" > $OUTPUT/blockio.in
   encrypt_and_compare
}

function run_large_test() {
   perl -e "for(\$i=0;\$i<$1;\$i++){print 'X';}" > $OUTPUT/blockio.in
   encrypt_and_compare
}

function run_large_random_test() {
   perl -e "for(\$i=0;\$i<$1;\$i++){print chr(int(rand 256));}" > \
      $OUTPUT/blockio.in
   encrypt_and_compare
}

run_small_test "0"
run_small_test "111111111111111"
run_small_test "1111111111111111"
run_small_test "11111111111111112"
run_small_test "11111111111111112222222222222222"
run_small_test "1111111111111111222222222222222233"
run_small_test "111111111111111122222222222222223333"
run_small_test "11111111111111112222222222222222333333"
run_small_test "1111111111111111222222222222222233333333333333334444"

run_small_test "0000000011111111222222223333333"
run_small_test "00000000111111112222222233333333"
run_small_test "0000111122223333444455556666777"
run_small_test "00001111222233334444555566667777"
run_small_test "00112233445566778899aabbccddeef"
run_small_test "00112233445566778899aabbccddeeff"
run_small_test "0123456789abcdefFEDCBA987654321"
run_small_test "0123456789abcdefFEDCBA9876543210"

run_large_test 0x07f
run_large_test 0x080
run_large_test 0x081
run_large_test 0x100
run_large_test 0x101

run_large_random_test 0x07f
run_large_random_test 0x080
run_large_random_test 0x081
run_large_random_test 0x100
run_large_random_test 0x101