mirror of
https://github.com/anrieff/libcpuid
synced 2025-10-03 11:01:30 +00:00
117 lines
4.2 KiB
Bash
Executable file
117 lines
4.2 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
# Environment variables
|
|
export LIBCPUID_NO_WARN=1
|
|
|
|
# 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-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_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_fix"
|
|
else
|
|
echo "$line" >> "$raw_file_tmp_fix"
|
|
fi
|
|
done < "$raw_file"
|
|
raw_file="$raw_file_tmp_fix"
|
|
fi
|
|
|
|
echo -e "\033[34mLoading '$raw_file' raw file...\033[0m"
|
|
"$cpuid_tool" --load="$raw_file" --report
|
|
|
|
echo -e "\033[34mCreating '$REPORT_FILE' report file...\033[0m"
|
|
"$cpuid_tool" --load="$raw_file" --outfile="$REPORT_FILE" --report
|
|
|
|
# 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' -e 's|w/|with|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
|
|
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"
|
|
|
|
# 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
|