#!/bin/sh # Copyright (C) 2014-2019 Daniel Baumann <daniel.baumann@open-infrastructure.net> # # SPDX-License-Identifier: GPL-3.0+ # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. set -e PROJECT="open-infrastructure" PROGRAM="container" COMMAND="$(basename ${0})" HOOKS="/etc/${PROJECT}/${PROGRAM}/hooks" MACHINES="/var/lib/machines" Parameters () { GETOPT_LONGOPTIONS="name:,blockio-device-weight:,blockio-read-bandwidth:,blockio-weight:,blockio-write-bandwidth:,cpu-quota:,cpu-shares:,memory-limit:,tasks-max:," GETOPT_OPTIONS="n:b:c:m:t:," PARAMETERS="$(getopt --longoptions ${GETOPT_LONGOPTIONS} --name=${COMMAND} --options ${GETOPT_OPTIONS} --shell sh -- ${@})" if [ "${?}" != "0" ] then echo "'${COMMAND}': getopt exit" >&2 exit 1 fi eval set -- "${PARAMETERS}" while true do case "${1}" in -n|--name) NAME="${2}" shift 2 ;; -c|--cpu-quota) CPU_QUOTA="${2}" shift 2 ;; --cpu-shares) CPU_SHARES="${2}" shift 2 ;; -m|--memory-limit) MEMORY_LIMIT="${2}" shift 2 ;; -t|--tasks-max) TASKS_MAX="${2}" shift 2 ;; --blockio-device-weight) BLOCK_IO_DEVICE_WEIGHT="${2}" shift 2 ;; --blockio-read-bandwidth) BLOCK_IO_READ_BANDWIDTH="${2}" shift 2 ;; -b|--blockio-weight) BLOCK_IO_WEIGHT="${2}" shift 2 ;; --blockio-write-bandwidth) BLOCK_IO_WRITE_BANDWIDTH="${2}" shift 2 ;; --) shift 1 break ;; *) echo "'${COMMAND}': getopt error" >&2 exit 1 ;; esac done } Usage () { echo "Usage: ${PROGRAM} ${COMMAND} -n|--name NAME [--blockio-device-weight \"DEVICE WEIGHT\"] [--blockio-read-bandwidth \"DEVICE BYTES\"] [-b|--blockio-weight WEIGHT] [--blockio-write-bandwidth \"DEVICE BYTES\"] [-c|--cpu-quota QUOTA] [--cpu-shares SHARES] [-m|--memory-limit BYTES] [-t|--tasks-max NUMBER]" >&2 exit 1 } Parameters "${@}" if [ -z "${NAME}" ] then Usage fi if [ ! -e "${MACHINES}/${NAME}" ] then echo "'${NAME}': no such container" >&2 exit 1 fi STATE="$(machinectl show ${NAME} 2>&1 | awk -F= '/^State=/ { print $2 }')" case "${STATE}" in running) ;; *) echo "'${NAME}': container is not running" >&2 exit 1 ;; esac if [ -n "${BLOCK_IO_DEVICE_WEIGHT}" ] then BLOCK_IO_DEVICE_WEIGHT="BlockIODeviceWeight=${BLOCK_IO_DEVICE_WEIGHT}" SET_PROPERTY="true" fi if [ -n "${BLOCK_IO_READ_BANDWIDTH}" ] then BLOCK_IO_READ_BANDWIDTH="BlockIOReadBandwidth=${BLOCK_IO_READ_BANDWIDTH}" SET_PROPERTY="true" fi if [ -n "${BLOCK_IO_WEIGHT}" ] then BLOCK_IO_WEIGHT="BlockIOWeight=${BLOCK_IO_WEIGHT}" SET_PROPERTY="true" fi if [ -n "${BLOCK_IO_WRITE_BANDWIDTH}" ] then BLOCK_IO_WRITE_BANDWIDTH="BlockIOReadBandwidth=${BLOCK_IO_WRITE_BANDWIDTH}" SET_PROPERTY="true" fi if [ -n "${CPU_QUOTA}" ] then CPU_QUOTA="CPUQuota=${CPU_QUOTA}" SET_PROPERTY="true" fi if [ -n "${CPU_SHARES}" ] then CPU_SHARES="CPUShares=${CPU_SHARES}" SET_PROPERTY="true" fi if [ -n "${MEMORY_LIMIT}" ] then MEMORY_LIMIT="MemoryLimit=${MEMORY_LIMIT}" SET_PROPERTY="true" fi if [ -n "${TASKS_MAX}" ] then TASKS_MAX="TasksMax=${TASKS_MAX}" SET_PROPERTY="true" fi if [ "${SET_PROPERTY}" != "true" ] then Usage fi # Pre hooks for FILE in "${HOOKS}/pre-${COMMAND}".* "${HOOKS}/${NAME}.pre-${COMMAND}" do if [ -x "${FILE}" ] then "${FILE}" fi done # Run systemctl --runtime set-property ${NAME} ${BLOCK_IO_DEVICE_WEIGHT} ${BLOCK_IO_READ_BANDWIDTH} ${BLOCK_IO_WEIGHT} ${BLOCK_IO_WRITE_BANDWIDTH} ${CPU_QUOTA} ${CPU_SHARES} ${MEMORY_LIMIT} ${TASKS_MAX} # Post hooks for FILE in "${HOOKS}/post-${COMMAND}".* "${HOOKS}/${NAME}.post-${COMMAND}" do if [ -x "${FILE}" ] then "${FILE}" fi done