140 lines · bash
1#!/bin/bash2# SPDX-License-Identifier: GPL-2.0+3#4# Transform a qemu-cmd file to allow reuse.5#6# Usage: kvm-transform.sh bzImage console.log jitter_dir seconds [ bootargs ] < qemu-cmd-in > qemu-cmd-out7#8# bzImage: Kernel and initrd from the same prior kvm.sh run.9# console.log: File into which to place console output.10# jitter_dir: Jitter directory for TORTURE_JITTER_START and11# TORTURE_JITTER_STOP environment variables.12# seconds: Run duaration for *.shutdown_secs module parameter.13# bootargs: New kernel boot parameters. Beware of Robert Tables.14#15# The original qemu-cmd file is provided on standard input.16# The transformed qemu-cmd file is on standard output.17# The transformation assumes that the qemu command is confined to a18# single line. It also assumes no whitespace in filenames.19#20# Copyright (C) 2020 Facebook, Inc.21#22# Authors: Paul E. McKenney <paulmck@kernel.org>23 24T=`mktemp -d /tmp/kvm-transform.sh.XXXXXXXXXX`25trap 'rm -rf $T' 0 226 27image="$1"28if test -z "$image"29then30 echo Need kernel image file.31 exit 132fi33consolelog="$2"34if test -z "$consolelog"35then36 echo "Need console log file name."37 exit 138fi39jitter_dir="$3"40if test -z "$jitter_dir" || ! test -d "$jitter_dir"41then42 echo "Need valid jitter directory: '$jitter_dir'"43 exit 144fi45seconds="$4"46if test -n "$seconds" && echo $seconds | grep -q '[^0-9]'47then48 echo "Invalid duration, should be numeric in seconds: '$seconds'"49 exit 150fi51bootargs="$5"52 53# Build awk program.54echo "BEGIN {" > $T/bootarg.awk55echo $bootargs | tr -s ' ' '\012' |56 awk -v dq='"' '/./ { print "\tbootarg[" NR "] = " dq $1 dq ";" }' >> $T/bootarg.awk57echo $bootargs | tr -s ' ' '\012' | sed -e 's/=.*$//' |58 awk -v dq='"' '/./ { print "\tbootpar[" NR "] = " dq $1 dq ";" }' >> $T/bootarg.awk59cat >> $T/bootarg.awk << '___EOF___'60}61 62/^# seconds=/ {63 if (seconds == "")64 print $0;65 else66 print "# seconds=" seconds;67 next;68}69 70/^# TORTURE_JITTER_START=/ {71 print "# TORTURE_JITTER_START=\". jitterstart.sh " $4 " " jitter_dir " " $6 " " $7;72 next;73}74 75/^# TORTURE_JITTER_STOP=/ {76 print "# TORTURE_JITTER_STOP=\". jitterstop.sh " " " jitter_dir " " $5;77 next;78}79 80/^#/ {81 print $0;82 next;83}84 85{86 line = "";87 for (i = 1; i <= NF; i++) {88 if (line == "") {89 line = $i;90 } else {91 line = line " " $i;92 }93 if ($i == "-serial") {94 i++;95 line = line " file:" consolelog;96 } else if ($i == "-kernel") {97 i++;98 line = line " " image;99 } else if ($i == "-append") {100 for (i++; i <= NF; i++) {101 arg = $i;102 lq = "";103 rq = "";104 if ("" seconds != "" && $i ~ /\.shutdown_secs=[0-9]*$/)105 sub(/[0-9]*$/, seconds, arg);106 if (arg ~ /^"/) {107 lq = substr(arg, 1, 1);108 arg = substr(arg, 2);109 }110 if (arg ~ /"$/) {111 rq = substr(arg, length($i), 1);112 arg = substr(arg, 1, length($i) - 1);113 }114 par = arg;115 gsub(/=.*$/, "", par);116 j = 1;117 while (bootpar[j] != "") {118 if (bootpar[j] == par) {119 arg = "";120 break;121 }122 j++;123 }124 if (line == "")125 line = lq arg;126 else127 line = line " " lq arg;128 }129 for (j in bootarg)130 line = line " " bootarg[j];131 line = line rq;132 }133 }134 print line;135}136___EOF___137 138awk -v image="$image" -v consolelog="$consolelog" -v jitter_dir="$jitter_dir" \139 -v seconds="$seconds" -f $T/bootarg.awk140