1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
# Copyright (C) 2010, Nokia <jean-luc.lamadon@nokia.com>
# Copyright (C) 2019, Sam Thursfield <sam@afuera.me.uk>
#
# 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 2
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.
#
"""
Sandbox environment for running tests.
The sandbox is essentially a private D-Bus daemon.
"""
import atexit
import logging
import os
import signal
from . import dbusdaemon
from . import dconf
from . import psutil_mini as psutil
log = logging.getLogger(__name__)
TRACKER_DBUS_PREFIX = 'org.freedesktop.Tracker3'
TRACKER_MINER_FS_BUSNAME = 'org.freedesktop.Tracker3.Miner.Files'
_process_list = []
def _cleanup_processes():
for process in _process_list:
log.debug("helpers._cleanup_processes: stopping %s", process)
process.stop()
atexit.register(_cleanup_processes)
class TrackerSandbox:
"""
Private D-Bus session bus which executes a sandboxed Tracker instance.
"""
def __init__(self, dbus_daemon_config_file, extra_env=None):
self.dbus_daemon_config_file = dbus_daemon_config_file
self.extra_env = extra_env or {}
self.daemon = dbusdaemon.DBusDaemon()
def start(self, new_session=False):
env = os.environ
env.update(self.extra_env)
env['G_MESSAGES_PREFIXED'] = 'all'
# This avoids an issue where gvfsd-fuse can start up while the bus is
# shutting down. If it fails to connect to the bus, it continues to
# run anyway which leads to our dbus-daemon failing to shut down.
#
# Since https://gitlab.gnome.org/GNOME/gvfs/issues/323 was implemented
# in GVFS 1.42 this problem may have gone away.
env['GVFS_DISABLE_FUSE'] = '1'
# Precreate runtime dir, to avoid this warning from dbus-daemon:
#
# Unable to set up transient service directory: XDG_RUNTIME_DIR "/home/sam/tracker-tests/tmp_59i3ev1/run" not available: No such file or directory
#
xdg_runtime_dir = env.get('XDG_RUNTIME_DIR')
if xdg_runtime_dir:
os.makedirs(xdg_runtime_dir, exist_ok=True)
log.info("Starting D-Bus daemon for sandbox.")
log.debug("Added environment variables: %s", self.extra_env)
self.daemon.start(self.dbus_daemon_config_file, env=env, new_session=new_session)
def stop(self):
tracker_processes = []
log.info("Looking for active Tracker processes on the bus")
for busname in self.daemon.list_names_sync():
if busname.startswith(TRACKER_DBUS_PREFIX):
pid = self.daemon.get_connection_unix_process_id_sync(busname)
if pid is not None:
tracker_processes.append(pid)
log.info("Terminating %i Tracker processes", len(tracker_processes))
for pid in tracker_processes:
os.kill(pid, signal.SIGTERM)
log.info("Waiting for %i Tracker processes", len(tracker_processes))
for pid in tracker_processes:
psutil.wait_pid(pid)
# We need to wait until Tracker processes have stopped before we
# terminate the D-Bus daemon, otherwise lots of criticals like this
# appear in the log output:
#
# (tracker-miner-fs:14955): GLib-GIO-CRITICAL **: 11:38:40.386: Error while sending AddMatch() message: The connection is closed
log.info("Stopping D-Bus daemon for sandbox.")
self.daemon.stop()
def stop_daemon(self, busname):
"""Stops the daemon that owns 'busname'.
This can be used if you want to force the miner-fs to exit, for
example.
"""
log.info("Stopping daemon process that owns %s.", busname)
pid = self.daemon.get_connection_unix_process_id_sync(busname)
if pid:
os.kill(pid, signal.SIGTERM)
psutil.wait_pid(pid)
log.info("Process %i has stopped.", pid)
else:
log.info("Couldn't find a process owning %s.", busname)
def get_connection(self):
return self.daemon.get_connection()
def get_session_bus_address(self):
return self.daemon.get_address()
def set_config(self, schema_config_dict):
"""Set config values in multiple GSettings schemas.
Example input:
set_all({
'org.freedesktop.Tracker3.Miner.Files': {
'enable-writeback': GLib.Variant.new_boolean(False),
}
})
"""
for schema_name, contents in schema_config_dict.items():
dconfclient = dconf.DConfClient(self)
for key, value in contents.items():
dconfclient.write(schema_name, key, value)
|
