Quelle  kvm-assign-cpus.sh   Sprache: Shell

#!/bin/sh
# SPDX-License-Identifier: GPL-2.0+
#
# Produce awk statements roughly depicting the system's CPU and cache
# layout.  If the required information is not available, produce
# error messages as awk comments.  Successful exit regardless.
#
# Usage: kvm-assign-cpus.sh /path/to/sysfs

T="`mktemp -d ${TMPDIR-/tmp}/kvm-assign-cpus.sh.XXXXXX`"
trap 'rm -rf $T' 0 2

sysfsdir=${1-/sys/devices/system/node}
if ! cd "$sysfsdir" > $T/msg 2>&1
then
 sed -e 's/^/# /' < $T/msg
 exit 0
fi
nodelist="`ls -d node*`"
for i in node*
do
 if ! test -d $i/
 then
  echo "# Not a directory: $sysfsdir/node*"
  exit 0
 fi
 for j in $i/cpu*/cache/index*
 do
  if ! test -d $j/
  then
   echo "# Not a directory: $sysfsdir/$j"
   exit 0
  else
   break
  fi
 done
 indexlist="`ls -d $i/cpu* | grep 'cpu[0-9][0-9]*' | head -1 | sed -e 's,^.*$,ls -d &/cache/index*,' | sh | sed -e 's,^.*/,,'`"
 break
done
for i in node*/cpu*/cache/index*/shared_cpu_list
do
 if ! test -f $i
 then
  echo "# Not a file: $sysfsdir/$i"
  exit 0
 else
  break
 fi
done
firstshared=
for i in $indexlist
do
 rm -f $T/cpulist
 for n in node*
 do
  f="$n/cpu*/cache/$i/shared_cpu_list"
  if ! cat $f > $T/msg 2>&1
  then
   sed -e 's/^/# /' < $T/msg
   exit 0
  fi
  cat $f >> $T/cpulist
 done
 if grep -q '[-,]' $T/cpulist
 then
  if test -z "$firstshared"
  then
   firstshared="$i"
  fi
 fi
done
if test -z "$firstshared"
then
 splitindex="`echo $indexlist | sed -e 's/ .*$//'`"
else
 splitindex="$firstshared"
fi
nodenum=0
for n in node*
do
 cat $n/cpu*/cache/$splitindex/shared_cpu_list | sort -u -k1n |
 awk -v nodenum="$nodenum" '
 BEGIN {
  idx = 0;
 }

 {
  nlists = split($0, cpulists, ",");
  for (i = 1; i <= nlists; i++) {
   listsize = split(cpulists[i], cpus, "-");
   if (listsize == 1)
    cpus[2] = cpus[1];
   for (j = cpus[1]; j <= cpus[2]; j++) {
    print "cpu[" nodenum "][" idx "] = " j ";";
    idx++;
   }
  }
 }

 END {
  print "nodecpus[" nodenum "] = " idx ";";
 }'
 nodenum=`expr $nodenum + 1`
done
echo "numnodes = $nodenum;"