X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-remote.git/blobdiff_plain/8b723d5c2cad29205e1f48142a9f5d4350c9fd2a..6c6f2af1d327dba9883f988b8bc23dd2804bd686:/host/usr/sbin/invirt-availability

diff --git a/host/usr/sbin/invirt-availability b/host/usr/sbin/invirt-availability
index 1dcf9c8..3ff23a8 100644
--- a/host/usr/sbin/invirt-availability
+++ b/host/usr/sbin/invirt-availability
@@ -8,12 +8,6 @@ import sys
 import yaml
 import os
 
-# return the amount of memory in kibibytes represented by s
-def parseUnits(s):
-    num, unit = s.split(' ')
-    assert unit == 'kB', "unexpected unit"
-    return int(num)
-
 def main(argv):
     """
     Calculate the amount of memory available for new VMs
@@ -40,10 +34,8 @@ def main(argv):
 
     free_memory = int(xminfo['free_memory']) * 1024
 
-    ballooninfo = yaml.load(open('/proc/xen/balloon', 'r').read())
-    currentallocation = parseUnits(ballooninfo['Current allocation'])
-    minimumtarget = parseUnits(ballooninfo['Minimum target'])
-
+    currentallocation = open('/sys/devices/system/xen_memory/xen_memory0/info/current_kb', 'r').read()
+    minimumtarget = open('/sys/devices/system/xen_memory/xen_memory0/target_kb', 'r').read()
     p = Popen(['/usr/sbin/xm', 'info', '-c'], stdout=PIPE)
     output = p.communicate()[0]
     if p.returncode != 0: