#!/bin/bash

# Copyright 2010 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# Generate test cases for use for the RSA verify benchmark.

# Load common constants and variables.
. "$(dirname "$0")/common.sh"

TEST_FILE=${TESTCASE_DIR}/test_file
TEST_FILE_SIZE=1000000

# Generate public key signatures on an input file for various combinations
# of message digest algorithms and RSA key sizes.
function generate_test_signatures {
  echo "Generating test signatures..."
  algorithmcounter=0
  for keylen in "${key_lengths[@]}"
  do
    for hashalgo in "${hash_algos[@]}"
    do
      openssl dgst "-${hashalgo}" -binary "${TEST_FILE}" > \
        "${TEST_FILE}.${hashalgo}.digest"
      "${BIN_DIR}/signature_digest_utility" "$algorithmcounter"  \
        "${TEST_FILE}" | openssl rsautl \
        -sign -pkcs -inkey "${TESTKEY_DIR}/key_rsa${keylen}.pem" \
        > "${TEST_FILE}.rsa${keylen}_${hashalgo}.sig"
      algorithmcounter=$((algorithmcounter + 1))
    done
  done
}

# Generate a file with random bytes for signature tests.
function generate_test_file {
  echo "Generating test file..."
  if [ -f "${TEST_FILE}" ]; then
    echo "(skipping, file already exists)"
    return
  fi
  dd if=/dev/urandom of="${TEST_FILE}" bs="${TEST_FILE_SIZE}" count=1
}

mkdir -p "${TESTCASE_DIR}"
check_test_keys
generate_test_file
generate_test_signatures