cryptospore / rpms / qemu-kvm

Forked from rpms/qemu-kvm 2 years ago
Clone
76daa3
#!/bin/bash
76daa3
#
76daa3
# Copyright 2009 Red Hat, Inc. and/or its affiliates.
76daa3
# Released under the GPL
76daa3
#
76daa3
# Author:      Dan Kenigsberg <danken@redhat.com>
76daa3
#
76daa3
# ksmtuned - a simple script that controls whether (and with what vigor) ksm
76daa3
# should search for duplicated pages.
76daa3
#
76daa3
# starts ksm when memory commited to qemu processes exceeds a threshold, and
76daa3
# make ksm work harder and harder untill memory load falls below that
76daa3
# threshold.
76daa3
#
76daa3
# send SIGUSR1 to this process right after a new qemu process is started, or
76daa3
# following its death, to retune ksm accordingly
76daa3
#
76daa3
# needs testing and ironing. contact danken@redhat.com if something breaks.
76daa3
76daa3
if [ -f /etc/ksmtuned.conf ]; then
76daa3
    . /etc/ksmtuned.conf
76daa3
fi
76daa3
76daa3
debug() {
76daa3
    if [ -n "$DEBUG" ]; then
76daa3
        s="`/bin/date`: $*"
76daa3
        [ -n "$LOGFILE" ] && echo "$s" >> "$LOGFILE" || echo "$s"
76daa3
    fi
76daa3
}
76daa3
76daa3
76daa3
KSM_MONITOR_INTERVAL=${KSM_MONITOR_INTERVAL:-60}
76daa3
KSM_NPAGES_BOOST=${KSM_NPAGES_BOOST:-300}
76daa3
KSM_NPAGES_DECAY=${KSM_NPAGES_DECAY:--50}
76daa3
76daa3
KSM_NPAGES_MIN=${KSM_NPAGES_MIN:-64}
76daa3
KSM_NPAGES_MAX=${KSM_NPAGES_MAX:-1250}
76daa3
# millisecond sleep between ksm scans for 16Gb server. Smaller servers sleep
76daa3
# more, bigger sleep less.
76daa3
KSM_SLEEP_MSEC=${KSM_SLEEP_MSEC:-10}
76daa3
76daa3
KSM_THRES_COEF=${KSM_THRES_COEF:-20}
76daa3
KSM_THRES_CONST=${KSM_THRES_CONST:-2048}
76daa3
76daa3
total=`awk '/^MemTotal:/ {print $2}' /proc/meminfo`
76daa3
debug total $total
76daa3
76daa3
npages=0
76daa3
sleep=$[KSM_SLEEP_MSEC * 16 * 1024 * 1024 / total]
76daa3
[ $sleep -le 10 ] && sleep=10
76daa3
debug sleep $sleep
76daa3
thres=$[total * KSM_THRES_COEF / 100]
76daa3
if [ $KSM_THRES_CONST -gt $thres ]; then
76daa3
    thres=$KSM_THRES_CONST
76daa3
fi
76daa3
debug thres $thres
76daa3
76daa3
KSMCTL () {
76daa3
    case x$1 in
76daa3
        xstop)
76daa3
            echo 0 > /sys/kernel/mm/ksm/run
76daa3
            ;;
76daa3
        xstart)
76daa3
            echo $2 > /sys/kernel/mm/ksm/pages_to_scan
76daa3
            echo $3 > /sys/kernel/mm/ksm/sleep_millisecs
76daa3
            echo 1 > /sys/kernel/mm/ksm/run
76daa3
            ;;
76daa3
    esac
76daa3
}
76daa3
76daa3
committed_memory () {
76daa3
    local pidlist
76daa3
    pidlist=$(pgrep -d ' ' -- '^qemu(-kvm|:.{1,11})$')
76daa3
    if [ -n "$pidlist" ]; then
76daa3
        ps -p "$pidlist" -o rsz=
76daa3
    fi | awk '{ sum += $1 }; END { print 0+sum }'
76daa3
}
76daa3
76daa3
free_memory () {
76daa3
    awk '/^(MemFree|Buffers|Cached):/ {free += $2}; END {print free}' \
76daa3
                /proc/meminfo
76daa3
}
76daa3
76daa3
increase_npages() {
76daa3
    local delta
76daa3
    delta=${1:-0}
76daa3
    npages=$[npages + delta]
76daa3
    if [ $npages -lt $KSM_NPAGES_MIN ]; then
76daa3
        npages=$KSM_NPAGES_MIN
76daa3
    elif [ $npages -gt $KSM_NPAGES_MAX ]; then
76daa3
        npages=$KSM_NPAGES_MAX
76daa3
    fi
76daa3
    echo $npages
76daa3
}
76daa3
76daa3
76daa3
adjust () {
76daa3
    local free committed
76daa3
    free=`free_memory`
76daa3
    committed=`committed_memory`
76daa3
    debug committed $committed free $free
76daa3
    if [ $[committed + thres] -lt $total -a $free -gt $thres ]; then
76daa3
        KSMCTL stop
76daa3
        debug "$[committed + thres] < $total and free > $thres, stop ksm"
76daa3
        return 1
76daa3
    fi
76daa3
    debug "$[committed + thres] > $total, start ksm"
76daa3
    if [ $free -lt $thres ]; then
76daa3
        npages=`increase_npages $KSM_NPAGES_BOOST`
76daa3
        debug "$free < $thres, boost"
76daa3
    else
76daa3
        npages=`increase_npages $KSM_NPAGES_DECAY`
76daa3
        debug "$free > $thres, decay"
76daa3
    fi
76daa3
    KSMCTL start $npages $sleep
76daa3
    debug "KSMCTL start $npages $sleep"
76daa3
    return 0
76daa3
}
76daa3
76daa3
function nothing () {
76daa3
    :
76daa3
}
76daa3
76daa3
loop () {
76daa3
    trap nothing SIGUSR1
76daa3
    while true
76daa3
    do
76daa3
        sleep $KSM_MONITOR_INTERVAL &
76daa3
        wait $!
76daa3
        adjust
76daa3
    done
76daa3
}
76daa3
76daa3
PIDFILE=${PIDFILE-/var/run/ksmtune.pid}
76daa3
if touch "$PIDFILE"; then
76daa3
  loop &
76daa3
  echo $! > "$PIDFILE"
76daa3
fi