mirror of
https://github.com/anrieff/libcpuid
synced 2025-05-06 09:03:36 +00:00
Utils: make create_test_from_instlatx64.sh more user-friendly
This commit is contained in:
The Tumultuous Unicorn Of Darkness
parent
87bc719460
commit
28c88e8f0a
1 changed files with 93 additions and 39 deletions
|
|
@ -2,62 +2,116 @@
|
|||
|
||||
set -euo pipefail
|
||||
|
||||
# Environment variables
|
||||
export LIBCPUID_NO_WARN=1
|
||||
GIT_ROOT_DIR="$(git rev-parse --show-toplevel)"
|
||||
CPUID_TOOL="${CPUID_TOOL:-"$GIT_ROOT_DIR/build/cpuid_tool/cpuid_tool"}"
|
||||
CREATE_TEST="${CREATE_TEST:-"$GIT_ROOT_DIR/tests/create_test.py"}"
|
||||
RAW_FILE="${1:-"/tmp/raw.txt"}"
|
||||
OUTPUT_DIR="${2:-}"
|
||||
REPORT_FILE="$(mktemp --tmpdir libcpuid-report.XXXXXX)"
|
||||
|
||||
is_tmp_raw=false
|
||||
# Download remote file if string contains URL
|
||||
if [[ "$RAW_FILE" =~ ^https?:\/\/ ]]; then
|
||||
is_tmp_raw=true
|
||||
raw_file_tmp="$(mktemp --tmpdir libcpuid-raw.XXXXXX)"
|
||||
echo -e "\033[1mDownloading '$RAW_FILE' raw file as '$raw_file_tmp'...\033[0m"
|
||||
wget --quiet "$RAW_FILE" --output-document="$raw_file_tmp"
|
||||
RAW_FILE="$raw_file_tmp"
|
||||
# Constants
|
||||
SHORT_OPTS="r:o:h"
|
||||
LONG_OPTS="raw-file:,output:,cpuid-tool:,create-test:,help"
|
||||
GIT_ROOT_DIR="$(git rev-parse --show-toplevel)"
|
||||
REPORT_FILE="$(mktemp --tmpdir libcpuid-report.XXXXXX)"
|
||||
MAX_UNCOMPRESSED_SIZE_KB=128
|
||||
|
||||
# Variables
|
||||
raw_file=""
|
||||
raw_file_tmp_download=""
|
||||
raw_file_tmp_fix=""
|
||||
output_dir=""
|
||||
output_file="/dev/stdout"
|
||||
cpuid_tool="$GIT_ROOT_DIR/build/cpuid_tool/cpuid_tool"
|
||||
create_test="$GIT_ROOT_DIR/tests/create_test.py"
|
||||
|
||||
# Cleanup on exit
|
||||
on_exit() {
|
||||
rm "$REPORT_FILE"
|
||||
[[ -f "$raw_file_tmp_download" ]] && rm "$raw_file_tmp_download"
|
||||
[[ -f "$raw_file_tmp_fix" ]] && rm "$raw_file_tmp_fix"
|
||||
}
|
||||
|
||||
# Display usage
|
||||
print_help() {
|
||||
echo -e "$(basename "$0") [OPTIONS] -r RAW_FILE [-o OUTPUT]\n"
|
||||
echo -e "This script helps to download raw file from InstLatx64 and create tests with standardized names.\n"
|
||||
echo -e "OPTIONS:"
|
||||
echo -e " -r, --raw-file\tPath to a local raw file OR an URL to a raw file"
|
||||
echo -e " -o, --output\t\tOutput test directory name (for semi-automatic file name) OR output test file name"
|
||||
echo -e " --cpuid-tool\tPath to the cpuid_tool binary (default: $cpuid_tool)"
|
||||
echo -e " --create-test\tPath to the create_test.py script (default: $create_test)"
|
||||
echo -e " -h, --help\t\tThis help"
|
||||
}
|
||||
|
||||
trap on_exit EXIT
|
||||
|
||||
# Parse options
|
||||
eval set -- "$(getopt --name="$0" --options="$SHORT_OPTS" --longoptions="$LONG_OPTS" -- "$@")"
|
||||
while true; do
|
||||
case "$1" in
|
||||
-r|--raw-file) raw_file="$2"; shift 2;;
|
||||
-o|--output) [[ -d "$2" ]] && output_dir="$2" || output_file="$2"; shift 2;;
|
||||
--cpuid-tool) cpuid_tool="$2"; shift 2;;
|
||||
--create-test) create_test="$2"; shift 2;;
|
||||
-h|--help) print_help; exit 0;;
|
||||
--) shift; break;;
|
||||
*) error "'$1' is not handled."; exit 1;;
|
||||
esac
|
||||
done
|
||||
|
||||
if [[ -z "$raw_file" ]]; then
|
||||
echo -e "\033[31m$(basename "$0"): --raw-file is mandatory.\033[0m"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Download a raw file from an URL
|
||||
if [[ "$raw_file" =~ ^https?:\/\/ ]]; then
|
||||
raw_file_tmp_download="$(mktemp --tmpdir libcpuid-raw.XXXXXX)"
|
||||
echo -e "\033[34mDownloading '$raw_file' raw file as '$raw_file_tmp_download'...\033[0m"
|
||||
wget --quiet "$raw_file" --output-document="$raw_file_tmp_download"
|
||||
raw_file="$raw_file_tmp_download"
|
||||
fi
|
||||
|
||||
# Fix raw file if there is no header for logical cores
|
||||
if [[ -z "$(grep -Ev "CPUID [0-9]{8}" "$RAW_FILE" | xargs)" ]]; then
|
||||
is_tmp_raw=true
|
||||
raw_file_tmp="$(mktemp --tmpdir libcpuid-raw.XXXXXX)"
|
||||
if [[ -z "$(grep -Ev "CPUID [0-9A-F]{8}" "$raw_file" | xargs)" ]]; then
|
||||
raw_file_tmp_fix="$(mktemp --tmpdir libcpuid-raw.XXXXXX)"
|
||||
echo -e "\033[34mFixing '$raw_file' raw file as '$raw_file_tmp_fix'...\033[0m"
|
||||
logical_cpu=0
|
||||
echo "------[ Logical CPU #$logical_cpu ]------" > "$raw_file_tmp"
|
||||
echo "------[ Logical CPU #$logical_cpu ]------" > "$raw_file_tmp_fix"
|
||||
while IFS= read -r line || [[ -n "$line" ]]; do
|
||||
if [[ -z "$line" ]]; then
|
||||
logical_cpu=$((logical_cpu + 1))
|
||||
echo -e "\n------[ Logical CPU #$logical_cpu ]------" >> "$raw_file_tmp"
|
||||
echo -e "\n------[ Logical CPU #$logical_cpu ]------" >> "$raw_file_tmp_fix"
|
||||
else
|
||||
echo "$line" >> "$raw_file_tmp"
|
||||
echo "$line" >> "$raw_file_tmp_fix"
|
||||
fi
|
||||
done < "$RAW_FILE"
|
||||
RAW_FILE="$raw_file_tmp"
|
||||
done < "$raw_file"
|
||||
raw_file="$raw_file_tmp_fix"
|
||||
fi
|
||||
|
||||
echo -e "\033[1mLoading '$RAW_FILE' raw file...\033[0m"
|
||||
"$CPUID_TOOL" --load="$RAW_FILE" --report
|
||||
echo -e "\033[34mLoading '$raw_file' raw file...\033[0m"
|
||||
"$cpuid_tool" --load="$raw_file" --report
|
||||
|
||||
echo -e "\033[1mCreating '$REPORT_FILE' report file...\033[0m"
|
||||
"$CPUID_TOOL" --load="$RAW_FILE" --outfile="$REPORT_FILE" --report
|
||||
echo -e "\033[34mCreating '$REPORT_FILE' report file...\033[0m"
|
||||
"$cpuid_tool" --load="$raw_file" --outfile="$REPORT_FILE" --report
|
||||
|
||||
if [[ -d "$OUTPUT_DIR" ]]; then
|
||||
test_file="$("$CPUID_TOOL" --load="$RAW_FILE" --brandstr | head -n1 | sed -e 's/([^)]*)//g' -e 's/,//g' | cut -d'@' -f1 | xargs | sed -r 's/\s+/-/g' | tr '[:upper:]' '[:lower:]')"
|
||||
if [[ -z "$test_file" ]]; then
|
||||
read -r -p "There is no brand string available, please enter filename manually: " test_file
|
||||
# Generate test file name
|
||||
if [[ -n "$output_dir" ]]; then
|
||||
test_file="$("$cpuid_tool" --load="$raw_file" --brandstr | head -n1 | sed -e 's/([^)]*)//g' -e 's/,//g' | cut -d'@' -f1 | xargs | sed -r 's/\s+/-/g' | tr '[:upper:]' '[:lower:]')"
|
||||
if [[ -n "$test_file" ]]; then
|
||||
read -r -p "'$test_file' will be the test file name, you can enter a different one if needed (without extension): " test_file_prompt
|
||||
test_file="${test_file_prompt:-$test_file}"
|
||||
else
|
||||
read -r -p "There is no brand string available, please enter file name manually (without extension): " test_file
|
||||
fi
|
||||
output_file="${OUTPUT_DIR}/${test_file}.test"
|
||||
echo -e "\033[1mCreating '$output_file' test file...\033[0m"
|
||||
else
|
||||
output_file="/dev/stdout"
|
||||
if [[ -z "$test_file" ]]; then
|
||||
echo -e "\033[31m$(basename "$0"): no test file name provided.\033[0m"
|
||||
exit 1
|
||||
fi
|
||||
output_file="${output_dir}/${test_file}.test"
|
||||
echo -e "\033[34mCreating '$output_file' test file...\033[0m"
|
||||
fi
|
||||
"$CREATE_TEST" "$RAW_FILE" "$REPORT_FILE" > "$output_file"
|
||||
rm "$REPORT_FILE"
|
||||
$is_tmp_raw && rm "$raw_file_tmp"
|
||||
"$create_test" "$raw_file" "$REPORT_FILE" > "$output_file"
|
||||
|
||||
if [[ -f "$output_file" ]] && [[ $(stat -c %s "$output_file") -gt $(( 1024*128 )) ]]; then
|
||||
echo -e "\033[1mCompressing '$output_file' test file...\033[0m"
|
||||
# Compress big test
|
||||
if [[ -f "$output_file" ]] && [[ $(stat -c %s "$output_file") -gt $(( 1024 * MAX_UNCOMPRESSED_SIZE_KB )) ]]; then
|
||||
echo -e "\033[34mCompressing '$output_file' test file...\033[0m"
|
||||
xz --force --compress "$output_file"
|
||||
fi
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue