From e536090331d0532b839fef0c841880d07ff8a51f Mon Sep 17 00:00:00 2001
From: ramdaron <terehovv@mail.ru>
Date: Tue, 16 Feb 2016 10:25:37 +0700
Subject: [PATCH] Add file with pcpu RCTL limits support

---
 lib/vm-rctl | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 lib/vm-rctl

diff --git a/lib/vm-rctl b/lib/vm-rctl
new file mode 100644
index 0000000..b73ddd4
--- /dev/null
+++ b/lib/vm-rctl
@@ -0,0 +1,55 @@
+#!/bin/sh
+# 'vm _rctl'
+# set limits to virtual machine
+# this is the background process
+#
+# @param string _name the name of the guest to enable limits
+#
+__vm_set_rctl_limits(){
+    local _name="$1"
+    local _conf="${vm_dir}/${_name}/${_name}.conf"
+
+    __config_load "${_conf}"
+    __config_get "_pcpu" "pcpu"
+
+    # it be available in FreeBSD 11.0
+#    __config_get "_rbps" "readbps"
+#    __config_get "_wbps" "writebps"
+#    __config_get "_riops" "readiops"
+#    __config_get "_wiops" "writeiops"
+
+    # wait a few seconds until the bhyve starts
+    sleep 1;
+    _pid=$(pgrep -fx "bhyve: ${_name}")
+
+    if [ -n "${_pid}" ]; then
+        if [ ! -z "${_pcpu}" -o ! -z "${_ior}" -o ! -z "${_iow}" ]; then
+            /usr/bin/rctl > /dev/null 2>&1
+            _exit=$?
+            [ ${_exit} -ne 0 ] && __err "RCTL support requested but RCTL not available"
+
+            if [ ! -z "${_pcpu}" ]; then
+                /usr/bin/rctl -a process:${_pid}:pcpu:deny=${_pcpu}
+                __log "guest" "${_name}" "limit rctl pcpu resource to ${_pcpu}"
+            fi
+
+            # it be available in FreeBSD 11.0
+            if [ ! -z ${_rbps} ]; then
+                /usr/bin/rctl -a process:${_pid}:readbps:throttle=${_rbps}
+                __log "guest" "limit rctl readbps resource to ${_rbps}"
+            fi
+            if [ ! -z ${_wbps} ]; then
+                /usr/bin/rctl -a process:${_pid}:writebps:throttle=${_wbps}
+                __log "guest" "limit rctl writebps resource to ${_wbps}"
+            fi
+            if [ ! -z ${_riops} ]; then
+                /usr/bin/rctl -a process:${_pid}:readiops:throttle=${_riops}
+                __log "guest" "limit rctl readiops resource to ${_riops}"
+            fi
+            if [ ! -z ${_wiops} ]; then
+                /usr/bin/rctl -a process:${_pid}:writeiops:throttle=${_wiops}
+                __log "guest" "limit rctl writeiops resource to ${_wiops}"
+            fi
+        fi
+    fi
+}