153 lines
6.3 KiB
Python
153 lines
6.3 KiB
Python
# Copyright (c) 2013 The Johns Hopkins University/Applied Physics Laboratory
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
|
|
import binascii
|
|
import copy
|
|
import mock
|
|
import six
|
|
|
|
from castellan.common.objects import symmetric_key as key
|
|
from castellan.tests.unit.key_manager import fake
|
|
from os_brick.encryptors import cryptsetup
|
|
from os_brick import exception
|
|
from os_brick.tests.encryptors import test_base
|
|
|
|
|
|
def fake__get_key(context):
|
|
raw = bytes(binascii.unhexlify('0' * 32))
|
|
|
|
symmetric_key = key.SymmetricKey('AES', len(raw) * 8, raw)
|
|
return symmetric_key
|
|
|
|
|
|
@mock.patch('os_brick.executor.encodeutils.safe_decode',
|
|
lambda x, errors=None: x)
|
|
class CryptsetupEncryptorTestCase(test_base.VolumeEncryptorTestCase):
|
|
@mock.patch('os.path.exists', return_value=False)
|
|
def _create(self, mock_exists, root_helper,
|
|
connection_info, keymgr, execute):
|
|
return cryptsetup.CryptsetupEncryptor(root_helper=root_helper,
|
|
connection_info=connection_info,
|
|
keymgr=keymgr,
|
|
execute=execute)
|
|
|
|
def setUp(self):
|
|
super(CryptsetupEncryptorTestCase, self).setUp()
|
|
|
|
self.dev_path = self.connection_info['data']['device_path']
|
|
self.dev_name = 'crypt-%s' % self.dev_path.split('/')[-1]
|
|
|
|
self.symlink_path = self.dev_path
|
|
|
|
def test__open_volume(self):
|
|
self.encryptor._open_volume("passphrase")
|
|
|
|
self.assert_exec_has_calls([
|
|
mock.call('cryptsetup', 'create', '--key-file=-', self.dev_name,
|
|
self.dev_path, process_input='passphrase',
|
|
run_as_root=True,
|
|
root_helper=self.root_helper,
|
|
check_exit_code=True),
|
|
])
|
|
|
|
def test_attach_volume(self):
|
|
self.encryptor._get_key = mock.MagicMock()
|
|
self.encryptor._get_key.return_value = fake__get_key(None)
|
|
|
|
self.encryptor.attach_volume(None)
|
|
|
|
self.assert_exec_has_calls([
|
|
mock.call('cryptsetup', 'create', '--key-file=-', self.dev_name,
|
|
self.dev_path, process_input='0' * 32,
|
|
root_helper=self.root_helper,
|
|
run_as_root=True, check_exit_code=True),
|
|
mock.call('ln', '--symbolic', '--force',
|
|
'/dev/mapper/%s' % self.dev_name, self.symlink_path,
|
|
root_helper=self.root_helper,
|
|
run_as_root=True, check_exit_code=True),
|
|
])
|
|
|
|
def test__close_volume(self):
|
|
self.encryptor.detach_volume()
|
|
|
|
self.assert_exec_has_calls([
|
|
mock.call('cryptsetup', 'remove', self.dev_name,
|
|
root_helper=self.root_helper,
|
|
run_as_root=True, check_exit_code=True),
|
|
])
|
|
|
|
def test_detach_volume(self):
|
|
self.encryptor.detach_volume()
|
|
|
|
self.assert_exec_has_calls([
|
|
mock.call('cryptsetup', 'remove', self.dev_name,
|
|
root_helper=self.root_helper,
|
|
run_as_root=True, check_exit_code=True),
|
|
])
|
|
|
|
def test_init_volume_encryption_not_supported(self):
|
|
# Tests that creating a CryptsetupEncryptor fails if there is no
|
|
# device_path key.
|
|
type = 'unencryptable'
|
|
data = dict(volume_id='a194699b-aa07-4433-a945-a5d23802043e')
|
|
connection_info = dict(driver_volume_type=type, data=data)
|
|
exc = self.assertRaises(exception.VolumeEncryptionNotSupported,
|
|
cryptsetup.CryptsetupEncryptor,
|
|
root_helper=self.root_helper,
|
|
connection_info=connection_info,
|
|
keymgr=fake.fake_api())
|
|
self.assertIn(type, six.text_type(exc))
|
|
|
|
@mock.patch('os.path.exists', return_value=True)
|
|
def test_init_volume_encryption_with_old_name(self,
|
|
mock_exists):
|
|
# If an old name crypt device exists, dev_path should be the old name.
|
|
old_dev_name = self.dev_path.split('/')[-1]
|
|
encryptor = cryptsetup.CryptsetupEncryptor(
|
|
root_helper=self.root_helper,
|
|
connection_info=self.connection_info,
|
|
keymgr=fake.fake_api(),
|
|
execute=self.mock_execute)
|
|
self.assertFalse(encryptor.dev_name.startswith('crypt-'))
|
|
self.assertEqual(old_dev_name, encryptor.dev_name)
|
|
self.assertEqual(self.dev_path, encryptor.dev_path)
|
|
self.assertEqual(self.symlink_path, encryptor.symlink_path)
|
|
mock_exists.assert_called_once_with('/dev/mapper/%s' % old_dev_name)
|
|
self.mock_execute.assert_called_once_with(
|
|
'cryptsetup', 'status', old_dev_name, run_as_root=True)
|
|
|
|
@mock.patch('os.path.exists', side_effect=[False, True])
|
|
def test_init_volume_encryption_with_wwn(self, mock_exists):
|
|
# If an wwn name crypt device exists, dev_path should be based on wwn.
|
|
old_dev_name = self.dev_path.split('/')[-1]
|
|
wwn = 'fake_wwn'
|
|
connection_info = copy.deepcopy(self.connection_info)
|
|
connection_info['data']['multipath_id'] = wwn
|
|
encryptor = cryptsetup.CryptsetupEncryptor(
|
|
root_helper=self.root_helper,
|
|
connection_info=connection_info,
|
|
keymgr=fake.fake_api(),
|
|
execute=self.mock_execute)
|
|
self.assertFalse(encryptor.dev_name.startswith('crypt-'))
|
|
self.assertEqual(wwn, encryptor.dev_name)
|
|
self.assertEqual(self.dev_path, encryptor.dev_path)
|
|
self.assertEqual(self.symlink_path, encryptor.symlink_path)
|
|
mock_exists.assert_has_calls([
|
|
mock.call('/dev/mapper/%s' % old_dev_name),
|
|
mock.call('/dev/mapper/%s' % wwn)])
|
|
self.mock_execute.assert_called_once_with(
|
|
'cryptsetup', 'status', wwn, run_as_root=True)
|