123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156 |
- #!/usr/bin/python
- import SocketServer
- import thread
- import time
- import xen.lowlevel.xs
- import sys
- import os
- from qmemman import SystemState
- import qmemman_algo
- from ConfigParser import SafeConfigParser
- from optparse import OptionParser
- from qubesutils import parse_size
- config_path = '/etc/qubes/qmemman.conf'
- SOCK_PATH='/var/run/qubes/qmemman.sock'
- LOG_PATH='/var/log/qubes/qmemman.log'
- system_state = SystemState()
- global_lock = thread.allocate_lock()
- def only_in_first_list(l1, l2):
- ret=[]
- for i in l1:
- if not i in l2:
- ret.append(i)
- return ret
- def get_domain_meminfo_key(domain_id):
- return '/local/domain/'+domain_id+'/memory/meminfo'
-
- class WatchType:
- def __init__(self, fn, param):
- self.fn = fn
- self.param = param
- class XS_Watcher:
- def __init__(self):
- self.handle = xen.lowlevel.xs.xs()
- self.handle.watch('@introduceDomain', WatchType(XS_Watcher.domain_list_changed, None))
- self.handle.watch('@releaseDomain', WatchType(XS_Watcher.domain_list_changed, None))
- self.watch_token_dict = {}
- def domain_list_changed(self, param):
- curr = self.handle.ls('', '/local/domain')
- if curr == None:
- return
- global_lock.acquire()
- for i in only_in_first_list(curr, self.watch_token_dict.keys()):
- #new domain has been created
- watch = WatchType(XS_Watcher.meminfo_changed, i)
- self.watch_token_dict[i] = watch
- self.handle.watch(get_domain_meminfo_key(i), watch)
- system_state.add_domain(i)
- for i in only_in_first_list(self.watch_token_dict.keys(), curr):
- #domain destroyed
- self.handle.unwatch(get_domain_meminfo_key(i), self.watch_token_dict[i])
- self.watch_token_dict.pop(i)
- system_state.del_domain(i)
- global_lock.release()
- system_state.do_balance()
- def meminfo_changed(self, domain_id):
- untrusted_meminfo_key = self.handle.read('', get_domain_meminfo_key(domain_id))
- if untrusted_meminfo_key == None or untrusted_meminfo_key == '':
- return
- global_lock.acquire()
- system_state.refresh_meminfo(domain_id, untrusted_meminfo_key)
- global_lock.release()
- def watch_loop(self):
- # sys.stderr = file('/var/log/qubes/qfileexchgd.errors', 'a')
- while True:
- result = self.handle.read_watch()
- token = result[1]
- token.fn(self, token.param)
- class QMemmanReqHandler(SocketServer.BaseRequestHandler):
- """
- The RequestHandler class for our server.
- It is instantiated once per connection to the server, and must
- override the handle() method to implement communication to the
- client.
- """
- def handle(self):
- got_lock = False
- # self.request is the TCP socket connected to the client
- while True:
- self.data = self.request.recv(1024).strip()
- if len(self.data) == 0:
- print 'EOF'
- if got_lock:
- global_lock.release()
- return
- if got_lock:
- print 'Second request over qmemman.sock ?'
- return
- global_lock.acquire()
- got_lock = True
- if system_state.do_balloon(int(self.data)):
- resp = "OK\n"
- else:
- resp = "FAIL\n"
- self.request.send(resp)
- def start_server(server):
- server.serve_forever()
- class QMemmanServer:
- @staticmethod
- def main():
- usage = "usage: %prog [options]"
- parser = OptionParser(usage)
- parser.add_option("-c", "--config", action="store", dest="config", default=config_path)
- (options, args) = parser.parse_args()
- logfd = os.open(LOG_PATH, os.O_WRONLY|os.O_APPEND|os.O_CREAT, 0644)
- if logfd < 0:
- print sys.stderr, "ERROR: Failed to open log file (%s)" % LOG_PATH
- exit(1)
- # reinitialize python stdout/err
- sys.stdout.flush()
- sys.stderr.flush()
- os.dup2(logfd, 1)
- os.dup2(logfd, 2)
- os.close(logfd)
- devnull = os.open('/dev/null', os.O_RDONLY)
- os.dup2(devnull, 0)
- config = SafeConfigParser({
- 'vm-min-mem': str(qmemman_algo.MIN_PREFMEM),
- 'dom0-mem-boost': str(qmemman_algo.DOM0_MEM_BOOST),
- 'cache-margin-factor': str(qmemman_algo.CACHE_FACTOR)
- })
- config.read(options.config)
- if config.has_section('global'):
- qmemman_algo.MIN_PREFMEM = parse_size(config.get('global', 'vm-min-mem'))
- qmemman_algo.DOM0_MEM_BOOST = parse_size(config.get('global', 'dom0-mem-boost'))
- qmemman_algo.CACHE_FACTOR = config.getfloat('global', 'cache-margin-factor')
- print "values: %s, %s, %s" % (str(qmemman_algo.MIN_PREFMEM), str(qmemman_algo.DOM0_MEM_BOOST), str(qmemman_algo.CACHE_FACTOR))
- try:
- os.unlink(SOCK_PATH)
- except:
- pass
- os.umask(0)
- server = SocketServer.UnixStreamServer(SOCK_PATH, QMemmanReqHandler)
- os.umask(077)
- if os.fork() == 0:
- thread.start_new_thread(start_server, tuple([server]))
- XS_Watcher().watch_loop()
|