Enable pylint for qubes.tools.qmemmand
Adjust the code to make pylint pass. It's just one part of qmemman, the smalest one.
This commit is contained in:
parent
e7888ffc9b
commit
3b1e57dc73
@ -1,5 +1,3 @@
|
|||||||
# pylint: skip-file
|
|
||||||
|
|
||||||
#
|
#
|
||||||
# The Qubes OS Project, http://www.qubes-os.org
|
# The Qubes OS Project, http://www.qubes-os.org
|
||||||
#
|
#
|
||||||
@ -19,6 +17,8 @@
|
|||||||
# License along with this library; if not, see <https://www.gnu.org/licenses/>.
|
# License along with this library; if not, see <https://www.gnu.org/licenses/>.
|
||||||
#
|
#
|
||||||
#
|
#
|
||||||
|
# pylint: disable=global-statement
|
||||||
|
|
||||||
import configparser
|
import configparser
|
||||||
import socketserver
|
import socketserver
|
||||||
import logging
|
import logging
|
||||||
@ -27,8 +27,10 @@ import os
|
|||||||
import socket
|
import socket
|
||||||
import sys
|
import sys
|
||||||
import threading
|
import threading
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from typing import Callable, Any
|
||||||
|
|
||||||
import xen.lowlevel.xs
|
import xen.lowlevel.xs # pylint: disable=import-error
|
||||||
|
|
||||||
import qubes.qmemman
|
import qubes.qmemman
|
||||||
import qubes.qmemman.algo
|
import qubes.qmemman.algo
|
||||||
@ -38,7 +40,7 @@ SOCK_PATH = '/var/run/qubes/qmemman.sock'
|
|||||||
|
|
||||||
system_state = qubes.qmemman.SystemState()
|
system_state = qubes.qmemman.SystemState()
|
||||||
global_lock = threading.Lock()
|
global_lock = threading.Lock()
|
||||||
# If XS_Watcher will
|
# If XSWatcher will
|
||||||
# handle meminfo event before @introduceDomain, it will use
|
# handle meminfo event before @introduceDomain, it will use
|
||||||
# incomplete domain list for that and may redistribute memory
|
# incomplete domain list for that and may redistribute memory
|
||||||
# allocated to some VM, but not yet used (see #1389).
|
# allocated to some VM, but not yet used (see #1389).
|
||||||
@ -48,10 +50,10 @@ global_lock = threading.Lock()
|
|||||||
# this flag before processing other event.
|
# this flag before processing other event.
|
||||||
force_refresh_domain_list = False
|
force_refresh_domain_list = False
|
||||||
|
|
||||||
def only_in_first_list(l1, l2):
|
def only_in_first_list(list1, list2):
|
||||||
ret = []
|
ret = []
|
||||||
for i in l1:
|
for i in list1:
|
||||||
if not i in l2:
|
if i not in list2:
|
||||||
ret.append(i)
|
ret.append(i)
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
@ -59,21 +61,22 @@ def get_domain_meminfo_key(domain_id):
|
|||||||
return '/local/domain/'+domain_id+'/memory/meminfo'
|
return '/local/domain/'+domain_id+'/memory/meminfo'
|
||||||
|
|
||||||
|
|
||||||
class WatchType(object):
|
@dataclass
|
||||||
def __init__(self, fn, param):
|
class WatchType:
|
||||||
self.fn = fn
|
func: Callable
|
||||||
self.param = param
|
param: Any
|
||||||
|
|
||||||
class XS_Watcher(object):
|
|
||||||
|
class XSWatcher:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.log = logging.getLogger('qmemman.daemon.xswatcher')
|
self.log = logging.getLogger('qmemman.daemon.xswatcher')
|
||||||
self.log.debug('XS_Watcher()')
|
self.log.debug('XS_Watcher()')
|
||||||
|
|
||||||
self.handle = xen.lowlevel.xs.xs()
|
self.handle = xen.lowlevel.xs.xs()
|
||||||
self.handle.watch('@introduceDomain', WatchType(
|
self.handle.watch('@introduceDomain', WatchType(
|
||||||
XS_Watcher.domain_list_changed, False))
|
XSWatcher.domain_list_changed, False))
|
||||||
self.handle.watch('@releaseDomain', WatchType(
|
self.handle.watch('@releaseDomain', WatchType(
|
||||||
XS_Watcher.domain_list_changed, False))
|
XSWatcher.domain_list_changed, False))
|
||||||
self.watch_token_dict = {}
|
self.watch_token_dict = {}
|
||||||
|
|
||||||
def domain_list_changed(self, refresh_only=False):
|
def domain_list_changed(self, refresh_only=False):
|
||||||
@ -112,17 +115,18 @@ class XS_Watcher(object):
|
|||||||
|
|
||||||
for i in only_in_first_list(curr, self.watch_token_dict.keys()):
|
for i in only_in_first_list(curr, self.watch_token_dict.keys()):
|
||||||
# new domain has been created
|
# new domain has been created
|
||||||
watch = WatchType(XS_Watcher.meminfo_changed, i)
|
watch = WatchType(XSWatcher.meminfo_changed, i)
|
||||||
self.watch_token_dict[i] = watch
|
self.watch_token_dict[i] = watch
|
||||||
self.handle.watch(get_domain_meminfo_key(i), watch)
|
self.handle.watch(get_domain_meminfo_key(i), watch)
|
||||||
system_state.add_domain(i)
|
system_state.add_domain(i)
|
||||||
|
|
||||||
for i in only_in_first_list(self.watch_token_dict.keys(), curr):
|
for i in only_in_first_list(self.watch_token_dict.keys(), curr):
|
||||||
# domain destroyed
|
# domain destroyed
|
||||||
self.handle.unwatch(get_domain_meminfo_key(i), self.watch_token_dict[i])
|
self.handle.unwatch(get_domain_meminfo_key(i),
|
||||||
|
self.watch_token_dict[i])
|
||||||
self.watch_token_dict.pop(i)
|
self.watch_token_dict.pop(i)
|
||||||
system_state.del_domain(i)
|
system_state.del_domain(i)
|
||||||
except:
|
except: # pylint: disable=bare-except
|
||||||
self.log.exception('Updating domain list failed')
|
self.log.exception('Updating domain list failed')
|
||||||
finally:
|
finally:
|
||||||
if got_lock:
|
if got_lock:
|
||||||
@ -132,7 +136,7 @@ class XS_Watcher(object):
|
|||||||
if not refresh_only:
|
if not refresh_only:
|
||||||
try:
|
try:
|
||||||
system_state.do_balance()
|
system_state.do_balance()
|
||||||
except:
|
except: # pylint: disable=bare-except
|
||||||
self.log.exception('do_balance() failed')
|
self.log.exception('do_balance() failed')
|
||||||
|
|
||||||
|
|
||||||
@ -140,7 +144,7 @@ class XS_Watcher(object):
|
|||||||
self.log.debug('meminfo_changed(domain_id={!r})'.format(domain_id))
|
self.log.debug('meminfo_changed(domain_id={!r})'.format(domain_id))
|
||||||
untrusted_meminfo_key = self.handle.read(
|
untrusted_meminfo_key = self.handle.read(
|
||||||
'', get_domain_meminfo_key(domain_id))
|
'', get_domain_meminfo_key(domain_id))
|
||||||
if untrusted_meminfo_key == None or untrusted_meminfo_key == b'':
|
if untrusted_meminfo_key is None or untrusted_meminfo_key == b'':
|
||||||
return
|
return
|
||||||
|
|
||||||
self.log.debug('acquiring global_lock')
|
self.log.debug('acquiring global_lock')
|
||||||
@ -156,20 +160,19 @@ class XS_Watcher(object):
|
|||||||
return
|
return
|
||||||
|
|
||||||
system_state.refresh_meminfo(domain_id, untrusted_meminfo_key)
|
system_state.refresh_meminfo(domain_id, untrusted_meminfo_key)
|
||||||
except:
|
except: # pylint: disable=bare-except
|
||||||
self.log.exception('Updating meminfo for %d failed', domain_id)
|
self.log.exception('Updating meminfo for %d failed', domain_id)
|
||||||
finally:
|
finally:
|
||||||
global_lock.release()
|
global_lock.release()
|
||||||
self.log.debug('global_lock released')
|
self.log.debug('global_lock released')
|
||||||
|
|
||||||
|
|
||||||
def watch_loop(self):
|
def watch_loop(self):
|
||||||
self.log.debug('watch_loop()')
|
self.log.debug('watch_loop()')
|
||||||
while True:
|
while True:
|
||||||
result = self.handle.read_watch()
|
result = self.handle.read_watch()
|
||||||
self.log.debug('watch_loop result={!r}'.format(result))
|
self.log.debug('watch_loop result={!r}'.format(result))
|
||||||
token = result[1]
|
token = result[1]
|
||||||
token.fn(self, token.param)
|
token.func(self, token.param)
|
||||||
|
|
||||||
|
|
||||||
class QMemmanReqHandler(socketserver.BaseRequestHandler):
|
class QMemmanReqHandler(socketserver.BaseRequestHandler):
|
||||||
@ -276,7 +279,7 @@ def main():
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
os.unlink(SOCK_PATH)
|
os.unlink(SOCK_PATH)
|
||||||
except:
|
except FileNotFoundError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
log.debug('instantiating server')
|
log.debug('instantiating server')
|
||||||
@ -292,12 +295,12 @@ def main():
|
|||||||
nofity_socket = os.getenv('NOTIFY_SOCKET')
|
nofity_socket = os.getenv('NOTIFY_SOCKET')
|
||||||
if nofity_socket:
|
if nofity_socket:
|
||||||
log.debug('notifying systemd')
|
log.debug('notifying systemd')
|
||||||
s = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
|
sock = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
|
||||||
if nofity_socket.startswith('@'):
|
if nofity_socket.startswith('@'):
|
||||||
nofity_socket = '\0%s' % nofity_socket[1:]
|
nofity_socket = '\0%s' % nofity_socket[1:]
|
||||||
s.connect(nofity_socket)
|
sock.connect(nofity_socket)
|
||||||
s.sendall(b"READY=1")
|
sock.sendall(b"READY=1")
|
||||||
s.close()
|
sock.close()
|
||||||
|
|
||||||
threading.Thread(target=server.serve_forever).start()
|
threading.Thread(target=server.serve_forever).start()
|
||||||
XS_Watcher().watch_loop()
|
XSWatcher().watch_loop()
|
||||||
|
Loading…
Reference in New Issue
Block a user