1
0
Fork 0
mirror of https://github.com/anrieff/libcpuid synced 2025-06-07 00:51:40 +00:00

Utils: allow to download and fix files in create_test_from_instlatx64.sh

This commit is contained in:
The Tumultuous Unicorn Of Darkness 2025-05-03 12:46:10 +02:00
parent 50ec270f40
commit 1cdc2bbd28
No known key found for this signature in database
GPG key ID: 1E55EE2EFF18BC1A

View file

@ -2,6 +2,7 @@
set -euo pipefail
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"}"
@ -9,6 +10,33 @@ 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"
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)"
logical_cpu=0
echo "------[ Logical CPU #$logical_cpu ]------" > "$raw_file_tmp"
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"
else
echo "$line" >> "$raw_file_tmp"
fi
done < "$RAW_FILE"
RAW_FILE="$raw_file_tmp"
fi
echo -e "\033[1mLoading '$RAW_FILE' raw file...\033[0m"
"$CPUID_TOOL" --load="$RAW_FILE" --report
@ -17,6 +45,9 @@ echo -e "\033[1mCreating '$REPORT_FILE' report file...\033[0m"
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
fi
output_file="${OUTPUT_DIR}/${test_file}.test"
echo -e "\033[1mCreating '$output_file' test file...\033[0m"
else
@ -24,7 +55,9 @@ else
fi
"$CREATE_TEST" "$RAW_FILE" "$REPORT_FILE" > "$output_file"
rm "$REPORT_FILE"
$is_tmp_raw && rm "$raw_file_tmp"
if [[ -f "$output_file" ]] && [[ $(stat -c %s "$output_file") -gt $(( 1024*128 )) ]]; then
echo -e "\033[1mCompressing '$output_file' test file...\033[0m"
xz --force --compress "$output_file"
fi