blob: 7013f7bfed66015d8ad03bc97fa6f8ecd58eb597 [file] [log] [blame]
Frank Farzand5e36312012-01-13 14:34:03 -08001# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Chris Masone5e06f182010-03-23 08:29:52 -07002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
Eric Caruso8dc40982018-03-20 17:05:19 -07005import dbus, gobject, logging, os, random, re, shutil, string, sys, time
Will Drewry9e440792013-12-11 17:18:35 -06006from dbus.mainloop.glib import DBusGMainLoop
barfab@chromium.orgb6d29932012-04-11 09:46:43 +02007
Hsinyu Chaoe0b08e62015-08-11 10:50:37 +00008import common, constants
barfab@chromium.org5c374632012-04-05 16:50:56 +02009from autotest_lib.client.bin import utils
Chris Masone5e06f182010-03-23 08:29:52 -070010from autotest_lib.client.common_lib import error
Will Drewry9e440792013-12-11 17:18:35 -060011from autotest_lib.client.cros.cros_disks import DBusClient
Eric Lic4d8f4a2010-12-10 09:49:23 -080012
Sean Oe5d8fd02010-09-30 10:44:44 +020013CRYPTOHOME_CMD = '/usr/sbin/cryptohome'
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -040014GUEST_USER_NAME = '$guest'
Kris Rambishbb5258c2014-12-16 16:51:17 -080015UNAVAILABLE_ACTION = 'Unknown action or no action given.'
Alexis Saveryccb16be2017-02-01 16:23:15 -080016MOUNT_RETRY_COUNT = 20
Dan Spaidbe9789c2017-05-19 15:18:42 +090017TEMP_MOUNT_PATTERN = '/home/.shadow/%s/temporary_mount'
18VAULT_PATH_PATTERN = '/home/.shadow/%s/vault'
Sean Oe5d8fd02010-09-30 10:44:44 +020019
Eric Caruso8dc40982018-03-20 17:05:19 -070020DBUS_PROTOS_DEP = 'dbus_protos'
21
22
Chris Masone5d010aa2013-05-06 14:38:42 -070023class ChromiumOSError(error.TestError):
Sean Oe5d8fd02010-09-30 10:44:44 +020024 """Generic error for ChromiumOS-specific exceptions."""
25 pass
26
Sean Oe5d8fd02010-09-30 10:44:44 +020027def __run_cmd(cmd):
28 return utils.system_output(cmd + ' 2>&1', retain_output=True,
29 ignore_status=True).strip()
30
Sean Oe5d8fd02010-09-30 10:44:44 +020031def get_user_hash(user):
barfab@chromium.org5c374632012-04-05 16:50:56 +020032 """Get the user hash for the given user."""
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -040033 return utils.system_output(['cryptohome', '--action=obfuscate_user',
34 '--user=%s' % user])
Sean Oe5d8fd02010-09-30 10:44:44 +020035
36
barfab@chromium.org5c374632012-04-05 16:50:56 +020037def user_path(user):
38 """Get the user mount point for the given user."""
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -040039 return utils.system_output(['cryptohome-path', 'user', user])
barfab@chromium.org5c374632012-04-05 16:50:56 +020040
41
42def system_path(user):
43 """Get the system mount point for the given user."""
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -040044 return utils.system_output(['cryptohome-path', 'system', user])
barfab@chromium.org5c374632012-04-05 16:50:56 +020045
46
Dan Spaidbe9789c2017-05-19 15:18:42 +090047def temporary_mount_path(user):
48 """Get the vault mount path used during crypto-migration for the user.
49
50 @param user: user the temporary mount should be for
51 """
52 return TEMP_MOUNT_PATTERN % (get_user_hash(user))
53
54
55def vault_path(user):
56 """ Get the vault path for the given user.
57
58 @param user: The user who's vault path should be returned.
59 """
60 return VAULT_PATH_PATTERN % (get_user_hash(user))
61
62
Chris Masone5d010aa2013-05-06 14:38:42 -070063def ensure_clean_cryptohome_for(user, password=None):
64 """Ensure a fresh cryptohome exists for user.
65
66 @param user: user who needs a shiny new cryptohome.
67 @param password: if unset, a random password will be used.
68 """
69 if not password:
70 password = ''.join(random.sample(string.ascii_lowercase, 6))
Eric Caruso272193b2018-02-05 14:02:11 -080071 unmount_vault(user)
Chris Masone5d010aa2013-05-06 14:38:42 -070072 remove_vault(user)
73 mount_vault(user, password, create=True)
74
75
Frank Farzand5e36312012-01-13 14:34:03 -080076def get_tpm_status():
77 """Get the TPM status.
78
79 Returns:
80 A TPM status dictionary, for example:
81 { 'Enabled': True,
82 'Owned': True,
83 'Being Owned': False,
84 'Ready': True,
85 'Password': ''
86 }
87 """
88 out = __run_cmd(CRYPTOHOME_CMD + ' --action=tpm_status')
89 status = {}
90 for field in ['Enabled', 'Owned', 'Being Owned', 'Ready']:
91 match = re.search('TPM %s: (true|false)' % field, out)
92 if not match:
93 raise ChromiumOSError('Invalid TPM status: "%s".' % out)
94 status[field] = match.group(1) == 'true'
95 match = re.search('TPM Password: (\w*)', out)
96 status['Password'] = ''
97 if match:
98 status['Password'] = match.group(1)
99 return status
100
101
Kris Rambish82ee1c02014-12-10 17:02:39 -0800102def get_tpm_more_status():
103 """Get more of the TPM status.
104
105 Returns:
106 A TPM more status dictionary, for example:
107 { 'dictionary_attack_lockout_in_effect': False,
108 'attestation_prepared': False,
109 'boot_lockbox_finalized': False,
110 'enabled': True,
111 'owned': True,
Kris Rambishbe132592014-12-17 14:26:06 -0800112 'owner_password': ''
Kris Rambish82ee1c02014-12-10 17:02:39 -0800113 'dictionary_attack_counter': 0,
114 'dictionary_attack_lockout_seconds_remaining': 0,
115 'dictionary_attack_threshold': 10,
116 'attestation_enrolled': False,
117 'initialized': True,
118 'verified_boot_measured': False,
119 'install_lockbox_finalized': True
120 }
Kris Rambishbb5258c2014-12-16 16:51:17 -0800121 An empty dictionary is returned if the command is not supported.
Kris Rambish82ee1c02014-12-10 17:02:39 -0800122 """
Kris Rambish82ee1c02014-12-10 17:02:39 -0800123 status = {}
Kris Rambishbb5258c2014-12-16 16:51:17 -0800124 out = __run_cmd(CRYPTOHOME_CMD + ' --action=tpm_more_status | grep :')
125 if out.startswith(UNAVAILABLE_ACTION):
126 # --action=tpm_more_status only exists >= 41.
127 logging.info('Method not supported!')
128 return status
Kris Rambish82ee1c02014-12-10 17:02:39 -0800129 for line in out.splitlines():
130 items = line.strip().split(':')
131 if items[1].strip() == 'false':
132 value = False
133 elif items[1].strip() == 'true':
134 value = True
Kris Rambishbe132592014-12-17 14:26:06 -0800135 elif items[1].strip().isdigit():
Kris Rambish82ee1c02014-12-10 17:02:39 -0800136 value = int(items[1].strip())
Kris Rambishbe132592014-12-17 14:26:06 -0800137 else:
138 value = items[1].strip(' "')
Kris Rambish82ee1c02014-12-10 17:02:39 -0800139 status[items[0]] = value
140 return status
141
142
Mary Ruthven9a0ce562017-05-30 13:01:47 -0700143def get_fwmp(cleared_fwmp=False):
144 """Get the firmware management parameters.
145
146 Args:
147 cleared_fwmp: True if the space should not exist.
148
149 Returns:
150 The dictionary with the FWMP contents, for example:
151 { 'flags': 0xbb41,
152 'developer_key_hash':
153 "\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\
154 000\000\000\000\000\000\000\000\000\000\000",
155 }
156 or a dictionary with the Error if the FWMP doesn't exist and
157 cleared_fwmp is True
158 { 'error': 'CRYPTOHOME_ERROR_FIRMWARE_MANAGEMENT_PARAMETERS_INVALID' }
159
160 Raises:
161 ChromiumOSError if any expected field is not found in the cryptohome
162 output. This would typically happen when FWMP state does not match
163 'clreared_fwmp'
164 """
165 out = __run_cmd(CRYPTOHOME_CMD +
166 ' --action=get_firmware_management_parameters')
167
168 if cleared_fwmp:
169 fields = ['error']
170 else:
171 fields = ['flags', 'developer_key_hash']
172
173 status = {}
174 for field in fields:
175 match = re.search('%s: (\S+)\n' % field, out)
176 if not match:
177 raise ChromiumOSError('Invalid FWMP field %s: "%s".' %
178 (field, out))
179 status[field] = match.group(1)
180 return status
181
182
183def set_fwmp(flags, developer_key_hash=None):
184 """Set the firmware management parameter contents.
185
186 Args:
187 developer_key_hash: a string with the developer key hash
188
189 Raises:
190 ChromiumOSError cryptohome cannot set the FWMP contents
191 """
192 cmd = (CRYPTOHOME_CMD +
193 ' --action=set_firmware_management_parameters '
194 '--flags=' + flags)
195 if developer_key_hash:
196 cmd += ' --developer_key_hash=' + developer_key_hash
197
198 out = __run_cmd(cmd)
199 if 'SetFirmwareManagementParameters success' not in out:
200 raise ChromiumOSError('failed to set FWMP: %s' % out)
201
202
Kris Rambish82ee1c02014-12-10 17:02:39 -0800203def is_tpm_lockout_in_effect():
204 """Returns true if the TPM lockout is in effect; false otherwise."""
205 status = get_tpm_more_status()
Christopher Wiley94fd6b32014-12-13 18:52:03 -0800206 return status.get('dictionary_attack_lockout_in_effect', None)
Kris Rambish82ee1c02014-12-10 17:02:39 -0800207
208
David Pursell2a2ef342014-10-17 10:34:56 -0700209def get_login_status():
210 """Query the login status
211
212 Returns:
213 A login status dictionary containing:
214 { 'owner_user_exists': True|False,
215 'boot_lockbox_finalized': True|False
216 }
217 """
218 out = __run_cmd(CRYPTOHOME_CMD + ' --action=get_login_status')
219 status = {}
220 for field in ['owner_user_exists', 'boot_lockbox_finalized']:
221 match = re.search('%s: (true|false)' % field, out)
222 if not match:
223 raise ChromiumOSError('Invalid login status: "%s".' % out)
224 status[field] = match.group(1) == 'true'
225 return status
226
227
Darren Krahn5f880f62012-10-02 15:17:59 -0700228def get_tpm_attestation_status():
229 """Get the TPM attestation status. Works similar to get_tpm_status().
230 """
231 out = __run_cmd(CRYPTOHOME_CMD + ' --action=tpm_attestation_status')
232 status = {}
233 for field in ['Prepared', 'Enrolled']:
234 match = re.search('Attestation %s: (true|false)' % field, out)
235 if not match:
236 raise ChromiumOSError('Invalid attestation status: "%s".' % out)
237 status[field] = match.group(1) == 'true'
238 return status
239
240
Eric Caruso6da07a02018-02-07 16:02:41 -0800241def take_tpm_ownership(wait_for_ownership=True):
Frank Farzand5e36312012-01-13 14:34:03 -0800242 """Take TPM owernship.
243
Eric Caruso6da07a02018-02-07 16:02:41 -0800244 Args:
245 wait_for_ownership: block until TPM is owned if true
Frank Farzand5e36312012-01-13 14:34:03 -0800246 """
247 __run_cmd(CRYPTOHOME_CMD + ' --action=tpm_take_ownership')
Eric Caruso6da07a02018-02-07 16:02:41 -0800248 if wait_for_ownership:
Maksim Ivanov21c967a2018-03-22 21:24:41 +0100249 # Note that waiting for the 'Ready' flag is more correct than waiting
250 # for the 'Owned' flag, as the latter is set by cryptohomed before some
251 # of the ownership tasks are completed.
252 utils.poll_for_condition(
253 lambda: get_tpm_status()['Ready'],
254 timeout=300,
255 exception=error.TestError('Timeout waiting for TPM ownership'))
Frank Farzand5e36312012-01-13 14:34:03 -0800256
257
Darren Krahn0e73e7f2012-09-05 15:35:15 -0700258def verify_ek():
259 """Verify the TPM endorsement key.
260
261 Returns true if EK is valid.
262 """
263 cmd = CRYPTOHOME_CMD + ' --action=tpm_verify_ek'
264 return (utils.system(cmd, ignore_status=True) == 0)
265
266
Sean Oe5d8fd02010-09-30 10:44:44 +0200267def remove_vault(user):
barfab@chromium.org5c374632012-04-05 16:50:56 +0200268 """Remove the given user's vault from the shadow directory."""
Sean Oe5d8fd02010-09-30 10:44:44 +0200269 logging.debug('user is %s', user)
270 user_hash = get_user_hash(user)
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700271 logging.debug('Removing vault for user %s with hash %s', user, user_hash)
Sean Oe5d8fd02010-09-30 10:44:44 +0200272 cmd = CRYPTOHOME_CMD + ' --action=remove --force --user=%s' % user
273 __run_cmd(cmd)
barfab@chromium.org5c374632012-04-05 16:50:56 +0200274 # Ensure that the vault does not exist.
275 if os.path.exists(os.path.join(constants.SHADOW_ROOT, user_hash)):
Darren Krahne6c44b92014-03-31 12:11:08 -0700276 raise ChromiumOSError('Cryptohome could not remove the user\'s vault.')
Sean Oe5d8fd02010-09-30 10:44:44 +0200277
278
barfab@chromium.orgcf2151e2012-04-04 15:39:34 +0200279def remove_all_vaults():
280 """Remove any existing vaults from the shadow directory.
281
282 This function must be run with root privileges.
283 """
barfab@chromium.org5c374632012-04-05 16:50:56 +0200284 for item in os.listdir(constants.SHADOW_ROOT):
285 abs_item = os.path.join(constants.SHADOW_ROOT, item)
barfab@chromium.orgcf2151e2012-04-04 15:39:34 +0200286 if os.path.isdir(os.path.join(abs_item, 'vault')):
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700287 logging.debug('Removing vault for user with hash %s', item)
barfab@chromium.orgcf2151e2012-04-04 15:39:34 +0200288 shutil.rmtree(abs_item)
289
290
Eric Caruso22acd672018-02-01 17:55:28 -0800291def mount_vault(user, password, create=False, key_label='bar'):
292 """Mount the given user's vault. Mounts should be created by calling this
293 function with create=True, and can be used afterwards with create=False.
294 Only try to mount existing vaults created with this function.
295
296 """
297 args = [CRYPTOHOME_CMD, '--action=mount_ex', '--user=%s' % user,
Chris Masone3543e512013-11-04 13:09:30 -0800298 '--password=%s' % password, '--async']
Sean Oe5d8fd02010-09-30 10:44:44 +0200299 if create:
Eric Caruso22acd672018-02-01 17:55:28 -0800300 args += ['--key_label=%s' % key_label, '--create']
Chris Masone3543e512013-11-04 13:09:30 -0800301 logging.info(__run_cmd(' '.join(args)))
barfab@chromium.org5c374632012-04-05 16:50:56 +0200302 # Ensure that the vault exists in the shadow directory.
Sean Oe5d8fd02010-09-30 10:44:44 +0200303 user_hash = get_user_hash(user)
barfab@chromium.org5c374632012-04-05 16:50:56 +0200304 if not os.path.exists(os.path.join(constants.SHADOW_ROOT, user_hash)):
Alexis Saveryccb16be2017-02-01 16:23:15 -0800305 retry = 0
306 mounted = False
307 while retry < MOUNT_RETRY_COUNT and not mounted:
308 time.sleep(1)
309 logging.info("Retry " + str(retry + 1))
310 __run_cmd(' '.join(args))
311 # TODO: Remove this additional call to get_user_hash(user) when
312 # crbug.com/690994 is fixed
313 user_hash = get_user_hash(user)
314 if os.path.exists(os.path.join(constants.SHADOW_ROOT, user_hash)):
315 mounted = True
316 retry += 1
317 if not mounted:
318 raise ChromiumOSError('Cryptohome vault not found after mount.')
barfab@chromium.org5c374632012-04-05 16:50:56 +0200319 # Ensure that the vault is mounted.
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700320 if not is_permanent_vault_mounted(user=user, allow_fail=True):
barfab@chromium.org5c374632012-04-05 16:50:56 +0200321 raise ChromiumOSError('Cryptohome created a vault but did not mount.')
Sean Oe5d8fd02010-09-30 10:44:44 +0200322
323
Chris Masone5d010aa2013-05-06 14:38:42 -0700324def mount_guest():
Sergey Poromov533008f2017-10-13 14:07:43 +0200325 """Mount the guest vault."""
Greg Kerrd7cdc132018-06-08 11:55:40 -0700326 args = [CRYPTOHOME_CMD, '--action=mount_guest_ex']
Chris Masone3543e512013-11-04 13:09:30 -0800327 logging.info(__run_cmd(' '.join(args)))
Sergey Poromov533008f2017-10-13 14:07:43 +0200328 # Ensure that the guest vault is mounted.
Chris Masone5d010aa2013-05-06 14:38:42 -0700329 if not is_guest_vault_mounted(allow_fail=True):
Sergey Poromov533008f2017-10-13 14:07:43 +0200330 raise ChromiumOSError('Cryptohome did not mount guest vault.')
Chris Masone5d010aa2013-05-06 14:38:42 -0700331
332
Sean Oe5d8fd02010-09-30 10:44:44 +0200333def test_auth(user, password):
Eric Carusode07cf82018-02-12 15:34:02 -0800334 cmd = [CRYPTOHOME_CMD, '--action=check_key_ex', '--user=%s' % user,
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -0400335 '--password=%s' % password, '--async']
Eric Carusode07cf82018-02-12 15:34:02 -0800336 out = __run_cmd(' '.join(cmd))
337 logging.info(out)
338 return 'Key authenticated.' in out
Sean Oe5d8fd02010-09-30 10:44:44 +0200339
340
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -0400341def unmount_vault(user):
barfab@chromium.org5c374632012-04-05 16:50:56 +0200342 """Unmount the given user's vault.
343
344 Once unmounting for a specific user is supported, the user parameter will
345 name the target user. See crosbug.com/20778.
Elly Jones686c2f42011-10-24 16:45:07 -0400346 """
Chris Masone3543e512013-11-04 13:09:30 -0800347 __run_cmd(CRYPTOHOME_CMD + ' --action=unmount')
barfab@chromium.org5c374632012-04-05 16:50:56 +0200348 # Ensure that the vault is not mounted.
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -0400349 if is_vault_mounted(user, allow_fail=True):
Sean Oe5d8fd02010-09-30 10:44:44 +0200350 raise ChromiumOSError('Cryptohome did not unmount the user.')
351
352
barfab@chromium.org5c374632012-04-05 16:50:56 +0200353def __get_mount_info(mount_point, allow_fail=False):
354 """Get information about the active mount at a given mount point."""
beeps569f8672013-08-07 10:18:51 -0700355 cryptohomed_path = '/proc/$(pgrep cryptohomed)/mounts'
356 try:
Daniel Erat2ec32792017-01-31 18:26:59 -0700357 logging.debug("Active cryptohome mounts:\n" +
358 utils.system_output('cat %s' % cryptohomed_path))
beeps569f8672013-08-07 10:18:51 -0700359 mount_line = utils.system_output(
360 'grep %s %s' % (mount_point, cryptohomed_path),
361 ignore_status=allow_fail)
362 except Exception as e:
363 logging.error(e)
364 raise ChromiumOSError('Could not get info about cryptohome vault '
365 'through %s. See logs for complete mount-point.'
366 % os.path.dirname(str(mount_point)))
Sourav Poddar574bd622010-05-26 14:22:26 +0530367 return mount_line.split()
368
369
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -0400370def __get_user_mount_info(user, allow_fail=False):
barfab@chromium.org5c374632012-04-05 16:50:56 +0200371 """Get information about the active mounts for a given user.
372
373 Returns the active mounts at the user's user and system mount points. If no
374 user is given, the active mount at the shared mount point is returned
375 (regular users have a bind-mount at this mount point for backwards
376 compatibility; the guest user has a mount at this mount point only).
377 """
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -0400378 return [__get_mount_info(mount_point=user_path(user),
379 allow_fail=allow_fail),
380 __get_mount_info(mount_point=system_path(user),
381 allow_fail=allow_fail)]
Jim Hebertf08f88d2011-04-22 10:33:49 -0700382
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700383def is_vault_mounted(user, regexes=None, allow_fail=False):
barfab@chromium.org5c374632012-04-05 16:50:56 +0200384 """Check whether a vault is mounted for the given user.
385
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700386 user: If no user is given, the shared mount point is checked, determining
387 whether a vault is mounted for any user.
388 regexes: dictionary of regexes to matches against the mount information.
389 The mount filesystem for the user's user and system mounts point must
390 match one of the keys.
391 The mount source point must match the selected device regex.
392
393 In addition, if mounted over ext4, we check the directory is encrypted.
barfab@chromium.org5c374632012-04-05 16:50:56 +0200394 """
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700395 if regexes is None:
396 regexes = {
397 constants.CRYPTOHOME_FS_REGEX_ANY :
398 constants.CRYPTOHOME_DEV_REGEX_ANY
399 }
barfab@chromium.org5c374632012-04-05 16:50:56 +0200400 user_mount_info = __get_user_mount_info(user=user, allow_fail=allow_fail)
401 for mount_info in user_mount_info:
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700402 # Look at each /proc/../mount lines that match mount point for a given
403 # user user/system mount (/home/user/.... /home/root/...)
404
405 # We should have at least 3 arguments (source, mount, type of mount)
406 if len(mount_info) < 3:
barfab@chromium.org5c374632012-04-05 16:50:56 +0200407 return False
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700408
409 device_regex = None
410 for fs_regex in regexes.keys():
411 if re.match(fs_regex, mount_info[2]):
412 device_regex = regexes[fs_regex]
413 break
414
415 if not device_regex:
416 # The thrid argument in not the expectd mount point type.
417 return False
418
419 # Check if the mount source match the device regex: it can be loose,
420 # (anything) or stricter if we expect guest filesystem.
421 if not re.match(device_regex, mount_info[0]):
422 return False
423
Sergey Poromov533008f2017-10-13 14:07:43 +0200424 if (re.match(constants.CRYPTOHOME_FS_REGEX_EXT4, mount_info[2])
425 and not(re.match(constants.CRYPTOHOME_DEV_REGEX_LOOP_DEVICE,
426 mount_info[0]))):
427 # Ephemeral cryptohome uses ext4 mount from a loop device,
428 # otherwise it should be ext4 crypto. Check there is an encryption
429 # key for that directory.
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700430 find_key_cmd_list = ['e4crypt get_policy %s' % (mount_info[1]),
431 'cut -d \' \' -f 2']
432 key = __run_cmd(' | ' .join(find_key_cmd_list))
433 cmd_list = ['keyctl show @s',
434 'grep %s' % (key),
435 'wc -l']
436 out = __run_cmd(' | '.join(cmd_list))
437 if int(out) != 1:
438 return False
barfab@chromium.org5c374632012-04-05 16:50:56 +0200439 return True
Sourav Poddar574bd622010-05-26 14:22:26 +0530440
441
barfab@chromium.org5c374632012-04-05 16:50:56 +0200442def is_guest_vault_mounted(allow_fail=False):
Sergey Poromov533008f2017-10-13 14:07:43 +0200443 """Check whether a vault is mounted for the guest user.
Sergey Poromovd85dce52017-12-27 11:10:51 +0100444 It should be a mount of an ext4 partition on a loop device
445 or be backed by tmpfs.
Sergey Poromov533008f2017-10-13 14:07:43 +0200446 """
barfab@chromium.org5c374632012-04-05 16:50:56 +0200447 return is_vault_mounted(
Elly Fong-Jones6cb26ad2013-05-21 12:09:23 -0400448 user=GUEST_USER_NAME,
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700449 regexes={
Sergey Poromovd85dce52017-12-27 11:10:51 +0100450 # Remove tmpfs support when it becomes unnecessary as all guest
451 # modes will use ext4 on a loop device.
Sergey Poromov533008f2017-10-13 14:07:43 +0200452 constants.CRYPTOHOME_FS_REGEX_EXT4 :
453 constants.CRYPTOHOME_DEV_REGEX_LOOP_DEVICE,
Sergey Poromovd85dce52017-12-27 11:10:51 +0100454 constants.CRYPTOHOME_FS_REGEX_TMPFS :
455 constants.CRYPTOHOME_DEV_REGEX_GUEST,
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700456 },
barfab@chromium.org5c374632012-04-05 16:50:56 +0200457 allow_fail=allow_fail)
458
Gwendal Grignou6bad6722016-06-09 16:36:04 -0700459def is_permanent_vault_mounted(user, allow_fail=False):
460 """Check if user is mounted over ecryptfs or ext4 crypto. """
461 return is_vault_mounted(
462 user=user,
463 regexes={
464 constants.CRYPTOHOME_FS_REGEX_ECRYPTFS :
465 constants.CRYPTOHOME_DEV_REGEX_REGULAR_USER_SHADOW,
466 constants.CRYPTOHOME_FS_REGEX_EXT4 :
467 constants.CRYPTOHOME_DEV_REGEX_REGULAR_USER_DEVICE,
468 },
469 allow_fail=allow_fail)
barfab@chromium.org5c374632012-04-05 16:50:56 +0200470
Kazuhiro Inabaa3bf6452017-02-08 11:41:50 +0900471def get_mounted_vault_path(user, allow_fail=False):
472 """Get the path where the decrypted data for the user is located."""
473 return os.path.join(constants.SHADOW_ROOT, get_user_hash(user), 'mount')
Nirnimesh66814492011-06-27 18:00:33 -0700474
475
476def canonicalize(credential):
barfab@chromium.org5c374632012-04-05 16:50:56 +0200477 """Perform basic canonicalization of |email_address|.
Nirnimesh66814492011-06-27 18:00:33 -0700478
barfab@chromium.org5c374632012-04-05 16:50:56 +0200479 Perform basic canonicalization of |email_address|, taking into account that
480 gmail does not consider '.' or caps inside a username to matter. It also
481 ignores everything after a '+'. For example,
482 c.masone+abc@gmail.com == cMaSone@gmail.com, per
Nirnimesh66814492011-06-27 18:00:33 -0700483 http://mail.google.com/support/bin/answer.py?hl=en&ctx=mail&answer=10313
484 """
485 if not credential:
486 return None
487
488 parts = credential.split('@')
489 if len(parts) != 2:
barfab@chromium.org5c374632012-04-05 16:50:56 +0200490 raise error.TestError('Malformed email: ' + credential)
Nirnimesh66814492011-06-27 18:00:33 -0700491
492 (name, domain) = parts
493 name = name.partition('+')[0]
barfab@chromium.org5c374632012-04-05 16:50:56 +0200494 if (domain == constants.SPECIAL_CASE_DOMAIN):
Nirnimesh66814492011-06-27 18:00:33 -0700495 name = name.replace('.', '')
496 return '@'.join([name, domain]).lower()
Elly Jones686c2f42011-10-24 16:45:07 -0400497
barfab@chromium.orgcf2151e2012-04-04 15:39:34 +0200498
Will Drewryd2fed972013-12-05 16:35:51 -0600499def crash_cryptohomed():
Will Drewrydc2b0dd2013-12-10 16:41:04 -0600500 # Try to kill cryptohomed so we get something to work with.
501 pid = __run_cmd('pgrep cryptohomed')
502 try:
Will Drewry9e440792013-12-11 17:18:35 -0600503 pid = int(pid)
Will Drewrydc2b0dd2013-12-10 16:41:04 -0600504 except ValueError, e: # empty or invalid string
Will Drewry9e440792013-12-11 17:18:35 -0600505 raise error.TestError('Cryptohomed was not running')
Will Drewrydc2b0dd2013-12-10 16:41:04 -0600506 utils.system('kill -ABRT %d' % pid)
507 # CONT just in case cryptohomed had a spurious STOP.
508 utils.system('kill -CONT %d' % pid)
509 utils.poll_for_condition(
510 lambda: utils.system('ps -p %d' % pid,
511 ignore_status=True) != 0,
Will Drewry934d1532014-01-30 16:23:17 -0600512 timeout=180,
Will Drewrydc2b0dd2013-12-10 16:41:04 -0600513 exception=error.TestError(
514 'Timeout waiting for cryptohomed to coredump'))
515
Will Drewryd2fed972013-12-05 16:35:51 -0600516
Dan Spaidbe9789c2017-05-19 15:18:42 +0900517def create_ecryptfs_homedir(user, password):
518 """Creates a new home directory as ecryptfs.
519
520 If a home directory for the user exists already, it will be removed.
521 The resulting home directory will be mounted.
522
523 @param user: Username to create the home directory for.
524 @param password: Password to use when creating the home directory.
525 """
526 unmount_vault(user)
527 remove_vault(user)
528 args = [
529 CRYPTOHOME_CMD,
530 '--action=mount_ex',
531 '--user=%s' % user,
532 '--password=%s' % password,
533 '--key_label=foo',
534 '--ecryptfs',
535 '--create']
536 logging.info(__run_cmd(' '.join(args)))
537 if not is_vault_mounted(user, regexes={
538 constants.CRYPTOHOME_FS_REGEX_ECRYPTFS :
539 constants.CRYPTOHOME_DEV_REGEX_REGULAR_USER_SHADOW
540 }, allow_fail=True):
541 raise ChromiumOSError('Ecryptfs home could not be created')
542
543
544def do_dircrypto_migration(user, password, timeout=600):
545 """Start dircrypto migration for the user.
546
547 @param user: The user to migrate.
548 @param password: The password used to mount the users vault
549 @param timeout: How long in seconds to wait for the migration to finish
550 before failing.
551 """
552 unmount_vault(user)
553 args = [
554 CRYPTOHOME_CMD,
555 '--action=mount_ex',
556 '--to_migrate_from_ecryptfs',
557 '--user=%s' % user,
558 '--password=%s' % password]
559 logging.info(__run_cmd(' '.join(args)))
560 if not __get_mount_info(temporary_mount_path(user), allow_fail=True):
561 raise ChromiumOSError('Failed to mount home for migration')
562 args = [CRYPTOHOME_CMD, '--action=migrate_to_dircrypto', '--user=%s' % user]
563 logging.info(__run_cmd(' '.join(args)))
564 utils.poll_for_condition(
565 lambda: not __get_mount_info(
566 temporary_mount_path(user), allow_fail=True),
567 timeout=timeout,
568 exception=error.TestError(
569 'Timeout waiting for dircrypto migration to finish'))
570
571
Eric Caruso014d5ed2018-02-01 16:24:41 -0800572def change_password(user, password, new_password):
573 args = [
574 CRYPTOHOME_CMD,
Greg Kerr98aa75c2018-06-05 15:27:12 -0700575 '--action=migrate_key_ex',
Eric Caruso014d5ed2018-02-01 16:24:41 -0800576 '--user=%s' % user,
577 '--old_password=%s' % password,
578 '--password=%s' % new_password]
Eric Caruso43ec57e2018-02-05 15:54:49 -0800579 out = __run_cmd(' '.join(args))
580 logging.info(out)
581 if 'Key migration succeeded.' not in out:
582 raise ChromiumOSError('Key migration failed.')
Eric Caruso014d5ed2018-02-01 16:24:41 -0800583
584
Will Drewry9e440792013-12-11 17:18:35 -0600585class CryptohomeProxy(DBusClient):
586 """A DBus proxy client for testing the Cryptohome DBus server.
587 """
588 CRYPTOHOME_BUS_NAME = 'org.chromium.Cryptohome'
589 CRYPTOHOME_OBJECT_PATH = '/org/chromium/Cryptohome'
590 CRYPTOHOME_INTERFACE = 'org.chromium.CryptohomeInterface'
591 ASYNC_CALL_STATUS_SIGNAL = 'AsyncCallStatus'
592 ASYNC_CALL_STATUS_SIGNAL_ARGUMENTS = (
593 'async_id', 'return_status', 'return_code'
594 )
595 DBUS_PROPERTIES_INTERFACE = 'org.freedesktop.DBus.Properties'
596
Lutz Justenb9275782018-06-20 18:42:22 +0200597 # Default timeout in seconds for the D-Bus connection.
598 DEFAULT_DBUS_TIMEOUT = 30
Chris Masone19e305e2014-03-14 15:13:46 -0700599
Lutz Justenb9275782018-06-20 18:42:22 +0200600 def __init__(self, bus_loop=None, autodir=None, job=None,
601 timeout=DEFAULT_DBUS_TIMEOUT):
Eric Caruso8dc40982018-03-20 17:05:19 -0700602 if autodir and job:
603 # Install D-Bus protos necessary for some methods.
604 dep_dir = os.path.join(autodir, 'deps', DBUS_PROTOS_DEP)
605 job.install_pkg(DBUS_PROTOS_DEP, 'dep', dep_dir)
606 sys.path.append(dep_dir)
607
608 # Set up D-Bus main loop and interface.
Will Drewry9e440792013-12-11 17:18:35 -0600609 self.main_loop = gobject.MainLoop()
Will Drewry78db9dc2014-04-01 16:34:23 -0500610 if bus_loop is None:
Chris Masone64170f82014-03-14 15:47:05 -0700611 bus_loop = DBusGMainLoop(set_as_default=True)
Will Drewry9e440792013-12-11 17:18:35 -0600612 self.bus = dbus.SystemBus(mainloop=bus_loop)
613 super(CryptohomeProxy, self).__init__(self.main_loop, self.bus,
614 self.CRYPTOHOME_BUS_NAME,
Lutz Justen1c6be452018-05-29 13:37:00 +0200615 self.CRYPTOHOME_OBJECT_PATH,
616 timeout)
Will Drewry9e440792013-12-11 17:18:35 -0600617 self.iface = dbus.Interface(self.proxy_object,
618 self.CRYPTOHOME_INTERFACE)
619 self.properties = dbus.Interface(self.proxy_object,
620 self.DBUS_PROPERTIES_INTERFACE)
621 self.handle_signal(self.CRYPTOHOME_INTERFACE,
622 self.ASYNC_CALL_STATUS_SIGNAL,
623 self.ASYNC_CALL_STATUS_SIGNAL_ARGUMENTS)
Elly Jones2f0ebba2011-10-27 13:43:20 -0400624
Chris Masone19e305e2014-03-14 15:13:46 -0700625
Will Drewryd2fed972013-12-05 16:35:51 -0600626 # Wrap all proxied calls to catch cryptohomed failures.
627 def __call(self, method, *args):
628 try:
Chris Masonef59d9df2014-03-14 12:05:32 -0700629 return method(*args, timeout=180)
Will Drewryd2fed972013-12-05 16:35:51 -0600630 except dbus.exceptions.DBusException, e:
631 if e.get_dbus_name() == 'org.freedesktop.DBus.Error.NoReply':
632 logging.error('Cryptohome is not responding. Sending ABRT')
633 crash_cryptohomed()
634 raise ChromiumOSError('cryptohomed aborted. Check crashes!')
635 raise e
636
Chris Masone19e305e2014-03-14 15:13:46 -0700637
Will Drewry9e440792013-12-11 17:18:35 -0600638 def __wait_for_specific_signal(self, signal, data):
639 """Wait for the |signal| with matching |data|
640 Returns the resulting dict on success or {} on error.
641 """
Will Drewryc4de5ff2014-02-03 13:26:57 -0600642 # Do not bubble up the timeout here, just return {}.
643 result = {}
644 try:
645 result = self.wait_for_signal(signal)
646 except utils.TimeoutError:
647 return {}
Will Drewry9e440792013-12-11 17:18:35 -0600648 for k in data.keys():
649 if not result.has_key(k) or result[k] != data[k]:
650 return {}
651 return result
652
Chris Masone19e305e2014-03-14 15:13:46 -0700653
Will Drewry9e440792013-12-11 17:18:35 -0600654 # Perform a data-less async call.
655 # TODO(wad) Add __async_data_call.
656 def __async_call(self, method, *args):
Will Drewryfef135a2014-05-23 16:02:14 -0500657 # Clear out any superfluous async call signals.
658 self.clear_signal_content(self.ASYNC_CALL_STATUS_SIGNAL)
Will Drewry9e440792013-12-11 17:18:35 -0600659 out = self.__call(method, *args)
660 logging.debug('Issued call ' + str(method) +
661 ' with async_id ' + str(out))
662 result = {}
663 try:
Will Drewry934d1532014-01-30 16:23:17 -0600664 # __wait_for_specific_signal has a 10s timeout
Will Drewry9e440792013-12-11 17:18:35 -0600665 result = utils.poll_for_condition(
666 lambda: self.__wait_for_specific_signal(
667 self.ASYNC_CALL_STATUS_SIGNAL, {'async_id' : out}),
Will Drewry934d1532014-01-30 16:23:17 -0600668 timeout=180,
Will Drewry9e440792013-12-11 17:18:35 -0600669 desc='matching %s signal' % self.ASYNC_CALL_STATUS_SIGNAL)
670 except utils.TimeoutError, e:
671 logging.error('Cryptohome timed out. Sending ABRT.')
672 crash_cryptohomed()
673 raise ChromiumOSError('cryptohomed aborted. Check crashes!')
674 return result
675
Chris Masone19e305e2014-03-14 15:13:46 -0700676
Eric Caruso8dc40982018-03-20 17:05:19 -0700677 def mount(self, user, password, create=False, async=True, key_label='bar'):
Elly Jones2f0ebba2011-10-27 13:43:20 -0400678 """Mounts a cryptohome.
679
680 Returns True if the mount succeeds or False otherwise.
Elly Jones2f0ebba2011-10-27 13:43:20 -0400681 """
Eric Caruso8dc40982018-03-20 17:05:19 -0700682 import rpc_pb2
683
684 acc = rpc_pb2.AccountIdentifier()
685 acc.account_id = user
686
687 auth = rpc_pb2.AuthorizationRequest()
688 auth.key.secret = password
689 auth.key.data.label = key_label
690
691 mount_req = rpc_pb2.MountRequest()
692 if create:
693 mount_req.create.copy_authorization_key = True
694
695 out = self.__call(self.iface.MountEx, acc.SerializeToString(),
696 auth.SerializeToString(), mount_req.SerializeToString())
697 parsed_out = rpc_pb2.BaseReply()
698 parsed_out.ParseFromString(''.join(map(chr, out)))
699 return parsed_out.error == rpc_pb2.CRYPTOHOME_ERROR_NOT_SET
Elly Jones2f0ebba2011-10-27 13:43:20 -0400700
Chris Masone19e305e2014-03-14 15:13:46 -0700701
Elly Jones2f0ebba2011-10-27 13:43:20 -0400702 def unmount(self, user):
703 """Unmounts a cryptohome.
704
705 Returns True if the unmount suceeds or false otherwise.
706 TODO(ellyjones): Once there's a per-user unmount method, use it. See
707 <crosbug.com/20778>.
708 """
Will Drewryd2fed972013-12-05 16:35:51 -0600709 return self.__call(self.iface.Unmount)
Elly Jones2f0ebba2011-10-27 13:43:20 -0400710
Chris Masone19e305e2014-03-14 15:13:46 -0700711
Elly Jones2f0ebba2011-10-27 13:43:20 -0400712 def is_mounted(self, user):
713 """Tests whether a user's cryptohome is mounted."""
714 return (utils.is_mountpoint(user_path(user))
715 and utils.is_mountpoint(system_path(user)))
716
Chris Masone19e305e2014-03-14 15:13:46 -0700717
Elly Jones2f0ebba2011-10-27 13:43:20 -0400718 def require_mounted(self, user):
719 """Raises a test failure if a user's cryptohome is not mounted."""
720 utils.require_mountpoint(user_path(user))
721 utils.require_mountpoint(system_path(user))
Elly Jones4458f442012-04-16 15:42:56 -0400722
Chris Masone19e305e2014-03-14 15:13:46 -0700723
Will Drewry9e440792013-12-11 17:18:35 -0600724 def migrate(self, user, oldkey, newkey, async=True):
Elly Jones4458f442012-04-16 15:42:56 -0400725 """Migrates the specified user's cryptohome from one key to another."""
Will Drewry9e440792013-12-11 17:18:35 -0600726 if async:
727 return self.__async_call(self.iface.AsyncMigrateKey,
728 user, oldkey, newkey)['return_status']
Will Drewryd2fed972013-12-05 16:35:51 -0600729 return self.__call(self.iface.MigrateKey, user, oldkey, newkey)
Elly Jones4458f442012-04-16 15:42:56 -0400730
Chris Masone19e305e2014-03-14 15:13:46 -0700731
Will Drewry9e440792013-12-11 17:18:35 -0600732 def remove(self, user, async=True):
733 if async:
734 return self.__async_call(self.iface.AsyncRemove,
735 user)['return_status']
Will Drewryd2fed972013-12-05 16:35:51 -0600736 return self.__call(self.iface.Remove, user)
Chris Masone19e305e2014-03-14 15:13:46 -0700737
738
739 def ensure_clean_cryptohome_for(self, user, password=None):
740 """Ensure a fresh cryptohome exists for user.
741
742 @param user: user who needs a shiny new cryptohome.
743 @param password: if unset, a random password will be used.
744 """
745 if not password:
746 password = ''.join(random.sample(string.ascii_lowercase, 6))
747 self.remove(user)
748 self.mount(user, password, create=True)
Roman Sorokina45273e2017-12-20 12:03:27 +0100749
750 def lock_install_attributes(self, attrs):
751 """Set and lock install attributes for the device.
752
753 @param attrs: dict of install attributes.
754 """
Roman Sorokin0a228d12018-01-23 12:36:45 +0100755 take_tpm_ownership()
Roman Sorokina45273e2017-12-20 12:03:27 +0100756 for key, value in attrs.items():
757 if not self.__call(self.iface.InstallAttributesSet, key,
758 dbus.ByteArray(value + '\0')):
759 return False
760 return self.__call(self.iface.InstallAttributesFinalize)