-
Notifications
You must be signed in to change notification settings - Fork 96
/
run-kvm.sh
executable file
·739 lines (670 loc) · 25.9 KB
/
run-kvm.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
#!/bin/bash
# This script assumes the existence of a lot of supporting scripts
DEFAULT_CONFIG=config
DIRNAME=`dirname $0`
SCRIPTDIR=`cd "$DIRNAME" && pwd`
export PATH="$SCRIPTDIR/bin:$PATH:$SCRIPTDIR/bin-virt"
export EXPECT_UNBUFFER=$SCRIPTDIR/bin/unbuffer
. $SCRIPTDIR/shellpacks/common.sh
. $SCRIPTDIR/shellpacks/common-config.sh
. $SCRIPTDIR/shellpacks/monitors.sh
MMTESTS_SSH_CONFIG_OPTIONS+="-o StrictHostKeyChecking=no -o ForwardAgent=no -o ForwardX11=no"
MMTESTS_SSH_OPTIONS+=" $SSH_CONFIG_OPTIONS"
MMTESTS_PSSH_OPTIONS+=" -t 0 $(echo $MMTESTS_SSH_CONFIG_OPTIONS|sed s/-o/-O/g)"
#for i in $MMTESTS_SSH_OPTIONS ; do
# MMTESTS_PSSH_OPTIONS+=" -x $i"
#done
if [ "$MARVIN_KVM_DOMAIN" = "" ]; then
export MARVIN_KVM_DOMAIN="marvin-mmtests"
fi
usage() {
echo "$0 [-pkonmh] [-C CONFIG_HOST] [--vm VMNAME[,VMNAME][,...]] run-mmtests-options"
echo
echo "-h|--help Prints this help."
echo "-p|--performance Force performance CPUFreq governor on the host before starting the tests"
echo "-L|--host-logs Collect logs and hardware info about the host"
echo "-k|--keep-kernel Use whatever kernel the VM currently has."
echo "-o|--offline-iothreads Take down some VM's CPUs and use for IOthreads."
echo "-m|--run-monitor Force enable monitoring on the host."
echo "-n|--no-monitor Force disable monitoring on the host."
echo "-C|--config-host CFG Use CFG as config file for the host."
echo "--vm VMNAME[,VMNAME] Name(s) of existing, and already known to 'virsh', VM(s)."
echo " If not specified, use \$MARVIN_KVM_DOMAIN as VM name."
echo " If that is not defined, use 'marvin-mmtests'."
echo "run-mmtests-options Parameters for run-mmtests.sh inside the VM (check them"
echo " with ./run-mmtests.sh -h)."
echo ""
echo "NOTE that 'run-mmtests-options', i.e., the parameters that will be used to execute"
echo "run-mmtests.sh inside the VMs, must always follow all the parameters intended for"
echo "run-kvm.sh itself."
}
# Parameters handling. Note that our own parmeters (i.e., run-kvm.sh
# parameters) must always come *before* the parameters we want MMTests
# inside the VM to use.
#
# There may be params that are valid arguments for both run-kvm.sh and
# run-mmtests.sh. We need to make sure that they are parsed only once.
# In fact, if we any of them is actually present twice, and we parse both
# the occurrences in here, then run-mmtests.sh, when run inside the VMs(s),
# will not see them.
#
# This is why this code looks different than "traditional" parameter handling,
# but it is either this, or we mandate that there can't be parameters with the
# same names in run-kvm.sh and run-mmtests.sh.
declare -a CONFIGS
export CONFIGS
while true; do
case "$1" in
-p|--performance)
if [ -z $FORCE_HOST_PERFORMANCE_SETUP ]; then
export FORCE_HOST_PERFORMANCE_SETUP="yes"
shift
else
break
fi
;;
-L|--host-logs)
HOST_LOGS="yes"
shift
;;
-k|--keep-kernel)
KEEP_KERNEL="yes"
shift
;;
-o|--offline-iothreads)
OFFLINE_IOTHREADS="yes"
shift
;;
-m|--run-monitor)
if [ -z $FORCE_RUN_MONITOR ]; then
export FORCE_RUN_MONITOR="yes"
shift
else
break
fi
;;
-n|--no-monitor)
if [ -z $FORCE_RUN_MONITOR ]; then
export FORCE_RUN_MONITOR="no"
shift
else
break
fi
;;
-C|--config-host)
shift
CONFIGS+=( "$1" )
shift
;;
--vm|--vms)
shift
VMS_LIST="yes"
VMS=$1
shift
;;
-h|--help)
usage
exit 0
;;
*)
break
;;
esac
done
# CURRENT_TEST is used only by monitors. Just define it to 'monitor' for now.
# We will change this, if/when we will do per-test monitoring.
export CURRENT_TEST="monitor"
# We want to read the config(s) that run-mmtests.sh will use inside the guests.
# They are in the set of parameters that we have not parsed above, so retrieve
# them here, without "consuming" them.
declare -a MMTESTS_CONFIGS
for (( i=0; i < $#; i++ ))
do
if [ "${!i}" = "-c" ] || [ "${!i}" = "--config" ]; then
i=$((i+1))
MMTESTS_CONFIGS+=( "${!i}" )
fi
done
# No config file specified for guests, so they'll use the default.
[ ! -z $MMTESTS_CONFIGS ] || MMTESTS_CONFIGS=( "$DEFAULT_CONFIG" )
# If we have an host config, we use that one here. That's rather handy if,
# for instance, we want different monitors or topology related tuning
# on the host and in the guests.
#
# If, OTOH, we don't have an host config file, let's import the config
# file(s) of the guests and use them for the host as well.
[ ! -z $CONFIGS ] || CONFIGS=( "$MMTESTS_CONFIGS" )
import_configs
# Command line has priority. However, if there wasn't any `--vm` param, check
# if we have a list of VMs to use in the config files. If there's nothing
# there either, default to $MARVIN_KVM_DOMAIN
if [ -z $VMS ] && [ "$MMTESTS_VMS" != "" ]; then
VMS_LIST=yes
# MMTESTS_VMS is space separated, we want VMS to be comma separated
VMS=$(echo ${MMTESTS_VMS// /,})
fi
if [ -z $VMS ]; then
VMS=$MARVIN_KVM_DOMAIN
fi
# If MMTESTS_HOST_IP is defined (e.g., in the configs we've imported), it
# means we are running as a "standalone virtualization bench suite". And we
# need to install some packages to be able to do so.
#
# We also need to check if, for instance, MMTESTS_HOST_IP is defined in
# whatever we are using as host config file. If it is, it must be there in
# the guests' configs as well, or we'll get stuck (because it's the fact
# that this var exists that tells guests that they need to contact the host
# for coordination of the test runs). So, we add it (and while there, add
# AUTO_PACKAGE_INSTALL too).
if [ ! -z $MMTESTS_HOST_IP ]; then
install-depends expect netcat-openbsd iputils
install-depends gnu_parallel
if ! command -v parallel &> /dev/null ; then
echo "WARNING: No package for GNU parallel found. We can work without it, but it would be better to have it."
echo "If you are on [open]SUSE, you can try this (but no guarantees it's still there!):"
echo " zypper ar https://download.opensuse.org/repositories/home:/tange/openSUSE_Tumbleweed/home:tange.repo"
echo " zypper ref && zypper in parallel"
fi
for c in ${MMTESTS_CONFIGS[@]}; do
if [ "`grep MMTESTS_HOST_IP ${c}`" = "" ] ; then
echo "export MMTESTS_HOST_IP=${MMTESTS_HOST_IP}" >> ${c}
fi
if [ "`grep AUTO_PACKAGE_INSTALL ${c}`" = "" ] ; then
echo "export AUTO_PACKAGE_INSTALL=\"yes\"" >> ${c}
fi
done
fi
install-depends time
# Check host monitors
if [ "$FORCE_RUN_MONITOR" != "" ]; then
RUN_MONITOR=$FORCE_RUN_MONITOR
fi
if [ "$RUN_MONITOR" = "no" ] || [ "$HOST_LOGS" != "yes" ]; then
# Disable monitor
unset MONITORS_GZIP
unset MONITORS_WITH_LATENCY
unset MONITORS_TRACER
# If we don't collect host logs, we don't want to monitor it either,
# not even with 'always on' monitors. In fact, we don't have a place
# where we could store the data!
[ "$HOST_LOGS" != "yes" ] && unset MONITORS_ALWAYS
else
# Check at least one monitor is enabled
if [ "$MONITORS_ALWAYS" = "" -a "$MONITORS_GZIP" = "" -a "$MONITORS_WITH_LATENCY" = "" -a "$MONITORS_TRACER" = "" ]; then
echo WARNING: Monitors enabled but none configured
fi
fi
export STAP_USED=
export MONITOR_STAP=
check_monitor_stap
if [ "$STAP_USED" != "" ]; then
fixup_stap
fi
install-depends openssh-clients
install_numad
install_tuned
# NB: 'runname' is the last of our parameters, as it is the last
# parameter of run-mmtests.sh.
RUNNAME=${@:$#}
# For now, we do not support multiple iterations on the host-side. It is,
# however, convenient to have a common structure of the activity log between
# run-kvm and run-mmtests (that supports iterations), so we at least define
# the counter and add log its value, so comparing the output of the monitors
# on the host, between different runs, will work.
MMTEST_HOST_ITERATION=0
activity_log "run-kvm: Iteration $MMTEST_HOST_ITERATION start"
# We only collect logs if the '-L' parameter was present.
if [ "$HOST_LOGS" = "yes" ]; then
export SHELLPACK_LOG=$SHELLPACK_LOG_BASE/$RUNNAME-host/iter-$MMTEST_HOST_ITERATION
# Delete old runs
rm -rf $SHELLPACK_LOG &>/dev/null
mkdir -p $SHELLPACK_LOG
export SHELLPACK_ACTIVITY="$SHELLPACK_LOG/tests-activity"
export SHELLPACK_LOGFILE="$SHELLPACK_LOG/tests-timestamp"
export SHELLPACK_SYSSTATEFILE="$SHELLPACK_LOG/tests-sysstate"
rm -f $SHELLPACK_ACTIVITY $SHELLPACK_LOGFILE $SHELLPACK_SYSSTATEFILE
fi
activity_log "run-kvm: Start"
start_numad
start_tuned
teststate_log "start :: `date +%s`"
# We need to be able to reach the guest at the port we use for guest-host
# communication, even if a firewall is up. This should work fine if with
# firewalld/firewall-cmd.
firewall_whitelist_ip() {
local IP=$1
if command -v firewall-cmd &> /dev/null && [ "$(firewall-cmd --state)" = "running" ]; then
firewall-cmd --zone=trusted --add-source=${IP}
fi
}
# Arrays where we store, for each VM, the IP and a VM-specific
# runname. The latter, in particular, is necessary because otherwise,
# when running the same benchmark in several VMs with different names,
# results would overwrite each other.
declare -a GUEST_IP
declare -a VM_RUNNAME
if [ "$MMTESTS_VMS_IP" != "" ]; then
# MMTESTS_VMS_IP is space separated, we want it to be comma separated
IPS=$(echo ${MMTESTS_VMS_IP// /,})
i=1
for IP in $(tr ',' '\n' <<< "$IPS")
do
GUEST_IP[$i]=$IP
i=$(( $i + 1 ))
done
v=1
for VM in $(tr ',' '\n' <<< "$VMS")
do
echo "checking VM: $VM at IP: ${GUEST_IP[$v]}"
wait_ssh_available ${GUEST_IP[$v]}
SSH_HOST="root@${GUEST_IP[$v]}"
PSSH_HOSTS+=" -H $SSH_HOST"
echo "VM ready: $VM IP: ${GUEST_IP[$v]}"
firewall_whitelist_ip ${GUEST_IP[$v]}
activity_log "run-kvm: VM $VM IP ${GUEST_IP[$v]}"
VM_RUNNAME[$v]="$RUNNAME-$VM"
v=$(( $v + 1 ))
done
[ $v -eq $i ] || die "MMTESTS_VMS and MMTESTS_VMS_IP mismatch"
else
echo "Booting the VM(s)"
activity_log "run-kvm: Booting VMs"
kvm-start --vm $VMS || die "Failed to boot VM(s)"
teststate_log "VMs up :: `date +%s`"
v=1
for VM in $(tr ',' '\n' <<< "$VMS")
do
GUEST_IP[$v]=`kvm-ip-address --vm $VM`
echo "VM ready: $VM IP: ${GUEST_IP[$v]}"
SSH_HOST="root@${GUEST_IP[$v]}"
PSSH_HOSTS+=" -H $SSH_HOST"
if [ "$HOST_LOGS" = "yes" ]; then
virsh dumpxml $VM > $SHELLPACK_LOG/$VM.xml
fi
firewall_whitelist_ip ${GUEST_IP[$v]}
activity_log "run-kvm: VM $VM IP ${GUEST_IP[$v]}"
VM_RUNNAME[$v]="$RUNNAME-$VM"
v=$(( $v + 1 ))
done
fi
VMCOUNT=$(( $v - 1 ))
# if we're not using firewall-cmd, let's just (desperately) try something with
# iptables, but I can't be sure it'll work equally well.
if [ ! -z $MMTESTS_HOST_IP ] && ! command -v firewall-cmd &> /dev/null; then
iptables -A INPUT -p tcp --dport ${MMTESTS_HOST_PORT:-1234} -j ACCEPT
fi
[ $VMCOUNT -lt 1 ] && die "ERROR: No VM specified?"
if [ $VMCOUNT -eq 1 ]; then
PSCP=scp
PSSH=ssh
# Of course, using $SSH_HOST like this makes sense only because we know
# that there is only 1 VM.
SSH_TARGET="$SSH_HOST"
SCP_TARGET="$SSH_TARGET:~"
PSSH_OPTS="$MMTESTS_SSH_OPTIONS"
elif [ -z $MMTESTS_HOST_IP ]; then
# When using more than 1 VMs, we need MMTESTS_HOST_IP to be explicitly
# defined, so that we know that we should follow the lockstep protocol,
# and not just let it/them run.
#
# TODO: If more than 1 VM is used, and MMTESTS_HOST_IP is not defined, we
# can try to automatically figure it out, and let things proceed...
die "ERROR: When using more than 1 VM, define MMTESTS_HOST_IP!"
else
# With more than 2 VMs, we need parallel SSH/SCP. However, the package
# may not be available in all distros. Or maybe it is, but the programs
# have different names, like it is, e.g., in openSUSE and in Debian. So,
# let's handle things in a bit of a special way...
install-depends pssh
PSCP=pscp ; PSSH=pssh
# If install-depends worked, either pscp or parallel-scp should be available now
if command -v parallel-scp &> /dev/null; then
PSCP=parallel-scp
PSSH=parallel-ssh
elif ! command -v pscp &> /dev/null; then
# Ok, this means it's not available in packages!
# But we *need* it, so let's try some extreme measures...
if [ "$AUTO_PACKAGE_INSTALL" != "yes" ] && [ ! -e "${HOME}/.mmtests-auto-package-install" ]; then
echo -n "MMTests would like to try to install PSSH from pip (https://pypi.org/project/pssh/). Can we proceed (YES/no/)? "
read -r answer
else
answer="yes"
fi
case "${answer}" in
y | Y | Yes | yes | YES)
install-depends python3-pip
pip install git+https://github.com/lilydjwg/pssh
PIP_UNINSTALL_PSSH="yes"
;;
*)
die "ERROR: We cannot continue without pssh/pscp!"
;;
esac
fi
command -v $PSCP &> /dev/null || \
die "ERROR: pscp not available. Cannot continue!"
command -v $PSSH &> /dev/null || \
die "pscp is there, but not pssh? Too weird to continue!"
PSSH_OPTS="$PSSH_OPTS $PSSH_HOSTS $MMTESTS_PSSH_OPTIONS -p $(( $VMCOUNT * 2 ))"
SSH_TARGET="" # All we need is already in PSSH_OPTS!
SCP_TARGET="~" # We need just the path(s)"
fi
# If $MMTESTS_PSSH_OUT_DIR contains a valid path, ask `pssh` to create there
# one file for each VM (name will be like root@<VM_IP>), were we can watch,
# live, the output of run-mmtests.sh, from inside each VM. This can be quite
# handy, especialy for debugging.
if [ ! -z $MMTESTS_PSSH_OUTDIR ]; then
mkdir -p $MMTESTS_PSSH_OUTDIR
PSSH_OPTS+=" -o $MMTESTS_PSSH_OUTDIR"
fi
teststate_log "vms ready :: `date +%s`"
echo Creating archive
NAME=`basename $SCRIPTDIR`
cd ..
tar -czf ${NAME}.tar.gz --exclude=${NAME}/work* --exclude=${NAME}/.git ${NAME} || die Failed to create mmtests archive
mv ${NAME}.tar.gz ${NAME}/
cd ${NAME}
# SCP_TARGET is just the path (i.e., '~') where to copy the archive on within
# the various VMs, if we have more than 1, or just "root@GUEST_IP:~"
echo Uploading and extracting new mmtests
$PSCP $PSSH_OPTS ${NAME}.tar.gz $SCP_TARGET || die Failed to upload ${NAME}.tar.gz
# SSH_TARGET is "", if we have more than 2 VMs and are using `pssh`(and
# all the targets are in PSSH_OPTS already) or "root@GUEST_IP", if we have
# only one VM.
$PSSH $PSSH_OPTS $SSH_TARGET "mkdir -p git-private && rm -rf git-private/${NAME} && tar -C git-private -xf ${NAME}.tar.gz" || die Failed to extract ${NAME}.tar.gz
rm ${NAME}.tar.gz
# We'll be running benchmarks with [P]SSH, without a terminal, etc. We *must*
# be absolutely sure that packages are automatically installed.
$PSSH $PSSH_OPTS $SSH_TARGET "touch ~/.mmtests-auto-package-install"
# Booting the current host kernel in VMs is, currently, not supported in
# "standaolne MMTests" (i.e., when ./bin-virt/kvm-boot` does not exist) or
# if we're using more than one VM.
if [ "$KEEP_KERNEL" != "yes" ] && [ $VMCOUNT -lt 2 ] && [ -e $SCRIPTDIR/bin-virt/kvm-boot ]; then
echo Booting current kernel `uname -r` $MORE_BOOT_ARGS on the guest
kvm-boot `uname -r` $MORE_BOOT_ARGS || die Failed to boot `uname -r`
fi
if [ "$OFFLINE_IOTHREADS" = "yes" ]; then
offline_cpus=`virsh dumpxml marvin-mmtests | grep -c iothreadpin`
if [ "$offline_cpus" != "" ]; then
echo Taking $offline_cpus offline for pinned io threads
for PHYS_CPU in `virsh dumpxml marvin-mmtests | grep iothreadpin | sed -e "s/.* cpuset='\([0-9]\+\)'.*/\1/"`; do
VIRT_CPU=`virsh dumpxml marvin-mmtests | grep vcpupin | grep "cpuset='$PHYS_CPU'" | sed -e "s/.* vcpu='\([0-9]\+\)'.*/\1/"`
ssh root@$GUEST_IP "echo 0 > /sys/devices/system/cpu/cpu$VIRT_CPU/online"
echo o Virt $VIRT_CPU phys $PHYS_CPU
done
fi
fi
# Set performance governor on the host, if wanted
if [ "$FORCE_HOST_PERFORMANCE_SETUP" = "yes" ]; then
FORCE_HOST_PERFORMANCE_SCALINGGOV_BASE=`cat /sys/devices/system/cpu/cpu0/cpufreq/scaling_governor`
NOTURBO="/sys/devices/system/cpu/intel_pstate/no_turbo"
[ -f $NOTURBO ] && FORCE_HOST_PERFORMANCE_NOTURBO_BASE=`cat $NOTURBO`
force_performance_setup
fi
sysstate_log_basic_info
collect_hardware_info
collect_kernel_info
collect_sysconfig_info
sync
start_monitors
echo "Executing mmtests in $VMCOUNT guest(s)"
activity_log "run-kvm: begin run-mmtests in VMs"
teststate_log "test begin :: `date +%s`"
activity_log "run-kvm: begin $CURRENT_TEST"
sysstate_log_proc_files "start"
/usr/bin/time -f "time :: $CURRENT_TEST %U user %S system %e elapsed" -o $SHELLPACK_LOG/timestamp \
$PSSH $PSSH_OPTS $SSH_TARGET "cd git-private/$NAME && ./run-mmtests.sh $@" &
PSSHPID=$!
# This variable can be used to provide additional options to `nc`, as it is
# used both here and within the shellpacks. This is mostly intended for
# debugging, e.g., adding "-v" to have more output.
#export _NCV="-v"
# If MMTESTS_HOST_IP is defined, we need to coordinate run-mmtests.sh
# execution phases inside the various VMs.
#
# When each VM reach one of such phases, it will send a message, letting
# us know what state it has actually reached, and wait for a poke. What we
# need to do here, is making sure that all VMs have reached one state.
# As soon as we have collected as many tokens as there are VMs, it means
# we've reached that point, and we poke every VM so they can proceed.
#
# What the states are, and how transitioning between them occurs, is
# explained in the following diagram:
#
# +-------+ +---------------+
# | START +------->| mmtests_start |<-----+
# +-------+ +-------+-------+ |
# |test_do/ |NO
# | tokens++ |
# v |
# +-------------------+ |
# |tokens == VMCOUNT ?+----+
# +---------+---------+
# |YES/
# | tokens=0
# v
# +---------+
# +-------->| test_do |<--------+
# | +----+----+ |
# | |test_do/ |
# | | tokens++ |NO
# | v |
# test_do/ | +-------------------+ |
# tokens=1| |tokens == VMCOUNT ?+----+
# | +---------+---------+
# | |YES/
# | | tokens=0
# | v mmtests_end/
# +--------+-----------+ tokens=1
# +------------->| test_do2 +----------------------+
# | +--------->+-----+-----+------------------+ |
# | | |iteration_begin/ | |
# | | | tokens=1 | |
# | | v | |
# | | +-----------------+ | |
# | | | iteration_begin |<--------+ | |
# | | +--------+--------+ | | |
# | | |iterations_begin/ |NO | |
# | | | tokens++ | | |
# | | v | | |
# | | +-------------------+ | | |
# | | |tokens == VMCOUNT ?+--------+ | |
# | | +---------+---------+ | |
# | | |YES/ test_done| |
# | | | tokens=0 tokens=1| |
# | | v | |
# | | +---------------+ | |
# | | | iteration_end |<--------+ | |
# | | +-------+-------+ | | |
# | |YES/ |iterations_end/ |NO | |
# | | tokens=0 | tokens++ | | |
# | | v | | |
# | | +-------------------+ | | |
# | +------+tokens == VMCOUNT ?+-------+ | |
# | +-------------------+ | |
# | | |
# | +-----------+ | |
# | | test_done |<-----------------+ |
# | +-----+-----+<-------+ |
# |YES/ |test_done/ |NO |
# | tokens=0 | tokens++ | |
# | v | |
# | +-------------------+ | |
# +----------+tokens == VMCOUNT ?+----+ |
# +-------------------+ |
# |
# +-------------+<--------------------+
# | mmtests_end |<------+
# +------+------+ |
# |mmtests_end/ |
# | tokens++ |NO
# v |
# +-------+ +-------------------+ |
# | QUIT |<-----+tokens == VMCOUNT ?+----+
# +-------+ +-------------------+
#
# For figuring out when VMs send the tokens for any give state,
# check run-mmtests.sh and the shellpacks rewriting code.
#
# Token exchanging happens (currently) over the network, via `nc`.
#
# TODO: likely, this can be re-implemented using, for instance, something
# like gRPC (either here, with https://github.com/fullstorydev/grpcurl) or
# by putting together some service program.
#
log_state() {
echo -ne "`date +%H:%M:%S` $1"
[ "$1" == "test_do" ] && echo -ne "\t"
echo -ne " "
}
if [ ! -z $MMTESTS_HOST_IP ]; then
[ $VMCOUNT -ne 1 ] && echo "TIME STATE VMs"
STATE="mmtests_start"
tokens=0
NCFILE=`mktemp`
nc $_NCV -n -4 -l -k $MMTESTS_HOST_IP $MMTESTS_HOST_PORT > $NCFILE &
NCPID=$!
tail -f $NCFILE | while [ "$STATE" != "QUIT" ] && read TOKEN
do
teststate_log "recvd token :: \"$TOKEN\" `date +%s`"
# With only 1 VM, there is not much to be synched. We just need
# to reply with the very same token we receive, in order to
# unblock each phase of run-mmtests.sh, inside the VM itself.
if [ $VMCOUNT -eq 1 ]; then
case "$TOKEN" in
"mmtests_start"|"test_do"|"iteration_begin"|"iteration_end"|"test_done")
mmtests_signal_token "$TOKEN" ${GUEST_IP[@]}
teststate_log "sent token :: \"$TOKEN\" `date +%s`"
;;
"mmtests_end")
mmtests_signal_token "mmtests_end" ${GUEST_IP[@]}
teststate_log "sent token :: \"$TOKEN\" `date +%s`"
STATE="QUIT"
;;
*)
echo "ERROR: unknown token (\'$TOKEN\') received!"
STATE="QUIT"
kill $PSSHPID
;;
esac
else
case "$STATE" in
"mmtests_start"|"test_do"|"iteration_begin"|"iteration_end"|"test_done"|"mmtests_end")
if [ $tokens -eq 0 ]; then
# DEBUG: not very useful info to print, unless we're debugging
#echo "run-kvm --> run-mmtests: state = $STATE"
log_state $STATE
teststate_log "enter state :: \"$STATE\" `date +%s`"
activity_log "run-kvm: state \"$STATE\""
fi
if [ "$TOKEN" != "$STATE" ]; then
echo "ERROR: wrong token (\'$TOKEN\') received while in state \'$STATE\'!"
STATE="QUIT"
kill $PSSHPID
else
echo -n 'X'
tokens=$(( $tokens + 1 ))
fi
if [ $tokens -eq $VMCOUNT ]; then
tokens=0
if [ "$STATE" = "mmtests_start" ]; then
STATE="test_do"
elif [ "$STATE" = "test_do" ] || [ "$STATE" = "iteration_end" ] || [ $"$STATE" = "test_done" ]; then
STATE="test_do2"
elif [ "$STATE" = "iteration_begin" ]; then
STATE="iteration_end"
elif [ "$STATE" = "test_done" ]; then
STATE="test_do2"
elif [ "$STATE" = "mmtests_end" ]; then
STATE="QUIT"
fi
echo " Done!"
activity_log "run-kvm: sending token \"$TOKEN\""
mmtests_signal_token "$TOKEN" ${GUEST_IP[@]}
teststate_log "sent token :: \"$TOKEN\" `date +%s`"
fi
;;
"test_do2")
tokens=1
if [ "$TOKEN" = "test_do" ]; then
STATE="test_do"
elif [ "$TOKEN" = "test_done" ]; then
STATE="test_done"
elif [ "$TOKEN" = "iteration_begin" ]; then
STATE="iteration_begin"
elif [ "$TOKEN" = "mmtests_end" ]; then
STATE="mmtests_end"
else
echo "ERROR: wrong token (\'$TOKEN\') received while in state \'$STATE\'!"
STATE="QUIT"
kill $PSSHPID
fi
# DEBUG: not very useful info to print, unless we're debugging
#echo "run-kvm --> run-mmtests: state = $STATE"
log_state $STATE ; echo -ne 'X'
teststate_log "enter state :: \"$STATE\" `date +%s`"
activity_log "run-kvm: state \"$STATE\""
;;
*)
echo "ERROR: unknown token (\'$TOKEN\') received!"
STATE="QUIT"
kill $PSSHPID
;;
esac
fi
done
kill $NCPID
rm -f $NCFILE
fi
wait $PSSHPID
RETVAL=$?
stop_monitors
sysstate_log_proc_files "end"
teststate_log "test end :: `date +%s`"
activity_log "run-kvm: run-mmtests in VMs end"
if [ "$FORCE_HOST_PERFORMANCE_SETUP" = "yes" ]; then
restore_performance_setup $FORCE_HOST_PERFORMANCE_SCALINGGOV_BASE $FORCE_HOST_PERFORMANCE_NOTURBO_BASE
fi
echo Syncing $SHELLPACK_LOG_BASE_SUBDIR
v=1
for VM in $(tr ',' '\n' <<< "$VMS")
do
# TODO: these two can probably be replaced with `pssh` and `pslurp`...
ssh root@${GUEST_IP[$v]} "cd git-private/$NAME && tar -czf work-${VM_RUNNAME[$v]}.tar.gz $SHELLPACK_LOG_BASE_SUBDIR" || die Failed to archive $SHELLPACK_LOG_BASE_SUBDIR
scp root@${GUEST_IP[$v]}:git-private/$NAME/work-${VM_RUNNAME[$v]}.tar.gz . || die Failed to download work.tar.gz
# Do not change behavior, file names, etc, if no VM list is specified.
# That, in fact, is how currently Marvin works, and we don't want to
# break it.
NEW_RUNNAME=$RUNNAME
if [ "$VMS_LIST" = "yes" ]; then
NEW_RUNNAME=${VM_RUNNAME[$v]}
fi
# Store the results of benchmark named `FOO`, done in VM 'bar' in
# a directory called 'bar-FOO.
tar --transform="s|$RUNNAME|$NEW_RUNNAME|" -xf work-${VM_RUNNAME[$v]}.tar.gz || die Failed to extract work.tar.gz
v=$(( $v + 1 ))
done
if [ "$MMTESTS_VMS_IP" != "" ]; then
echo "Leaving the VM(s) up"
else
echo "Shutting down the VM(s)"
activity_log "run-kvm: Shutoff VMs"
kvm-stop --vm $VMS
teststate_log "VMs down :: `date +%s`"
fi
teststate_log "`cat $SHELLPACK_LOG/timestamp`"
teststate_log "finish :: `date +%s`"
if [ "$HOST_LOGS" = "yes" ]; then
dmesg > $SHELLPACK_LOG/dmesg
gzip -f $SHELLPACK_LOG/dmesg
gzip -f $SHELLPACK_SYSSTATEFILE
fi
shutdown_numad
shutdown_tuned
activity_log "run-kvm: Iteration $MMTEST_HOST_ITERATION end"
if [ "$PIP_UNINSTALL_PSSH" = "yes" ]; then
pip uninstall -y pssh
fi
activity_log "run-kvm: End"
teststate_log "status :: $RETVAL"
exit $RETVAL